1 files changed, 19 insertions, 1 deletions
diff --git a/src/modules/ssl.h b/src/modules/ssl.h
index e636aad46..4d303502e 100644
--- a/src/modules/ssl.h
+++ b/src/modules/ssl.h
@@ -152,13 +152,31 @@ class ssl_cert
 class ISHRequest : public Request
 {
  public:
-	const InspSocket* Sock;
+	InspSocket* Sock;
 
 	ISHRequest(Module* Me, Module* Target, const char* rtype, InspSocket* sock) : Request(Me, Target, rtype), Sock(sock)
 	{
 	}
 };
 
+class InspSocketAttachCertRequest : public ISHRequest
+{
+ public:
+	/** Initialize the request as an attach cert message */
+	InspSocketAttachCertRequest(InspSocket* is, Module* Me, Module* Target) : ISHRequest(Me, Target, "IS_ATTACH", is)
+	{
+	}
+};
+
+class InspSocketHSCompleteRequest : public ISHRequest
+{
+ public:
+	/** Initialize the request as a 'handshake complete?' message */
+	InspSocketHSCompleteRequest(InspSocket* is, Module* Me, Module* Target) : ISHRequest(Me, Target, "IS_HSDONE", is)
+	{
+	}
+};
+
 class InspSocketHookRequest : public ISHRequest
 {
  public: