Store the type of a StreamSocket within itself.

Similar to with IOHooks this allows you to convert StreamSocket to
a UserIOHandler quickly.

1 files changed, 13 insertions, 1 deletions

diff --git a/include/inspsocket.h b/include/inspsocket.h
index d88c350ba..e432f9c16 100644
--- a/include/inspsocket.h
+++ b/include/inspsocket.h
@@ -219,6 +219,13 @@ class CoreExport StreamSocket : public EventHandler
 		size_t nbytes;
 	};
 
+	/** The type of socket this IOHook represents. */
+	enum Type
+	{
+		SS_UNKNOWN,
+		SS_USER
+	};
+
  private:
 	/** The IOHook that handles raw I/O for this socket, or NULL */
 	IOHook* iohook;
@@ -264,7 +271,12 @@ class CoreExport StreamSocket : public EventHandler
  protected:
 	std::string recvq;
  public:
-	StreamSocket() : iohook(NULL) { }
+	const Type type;
+	StreamSocket(Type sstype = SS_UNKNOWN)
+		: iohook(NULL)
+		, type(sstype)
+	{
+	}
 	IOHook* GetIOHook() const;
 	void AddIOHook(IOHook* hook);
 	void DelIOHook();