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, 6 insertions, 0 deletions

diff --git a/include/inspsocket.h b/include/inspsocket.h
index 53eca2e91..72fb03d58 100644
--- a/include/inspsocket.h
+++ b/include/inspsocket.h
@@ -284,6 +284,12 @@ class CoreExport StreamSocket : public EventHandler
 	virtual void Close();
 	/** This ensures that close is called prior to destructor */
 	virtual CullResult cull();
+
+	/** Get the IOHook of a module attached to this socket
+	 * @param mod Module whose IOHook to return
+	 * @return IOHook belonging to the module or NULL if the module haven't attached an IOHook to this socket
+	 */
+	IOHook* GetModHook(Module* mod) const;
 };
 /**
  * BufferedSocket is an extendable socket class which modules