Add StreamSocket::GetModHook() for obtaining the IOHook belonging to a given module

Use it to simplify logic in all modules using or providing IOHooks

1 files changed, 10 insertions, 0 deletions

diff --git a/src/inspsocket.cpp b/src/inspsocket.cpp
index dcc455482..0b0507f7c 100644
--- a/src/inspsocket.cpp
+++ b/src/inspsocket.cpp
@@ -434,3 +434,13 @@ void StreamSocket::CheckError(BufferedSocketError errcode)
 		OnError(errcode);
 	}
 }
+
+IOHook* StreamSocket::GetModHook(Module* mod) const
+{
+	if (iohook)
+	{
+		if (iohook->prov->creator == mod)
+			return iohook;
+	}
+	return NULL;
+}