From ad2b3c4aa3db056b83f5d91564400e03f3239835 Mon Sep 17 00:00:00 2001
From: Pierre Ossman <ossman@cendio.se>
Date: Fri, 21 Sep 2018 15:31:11 +0200
Subject: Strongly bind security objects to connection object

There is already an implicit connection between them, so let's make
it explicit. This allows easy lookup outside of the processMsg() function.
---
 common/rfb/CSecurity.h | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'common/rfb/CSecurity.h')

diff --git a/common/rfb/CSecurity.h b/common/rfb/CSecurity.h
index 4bf4b38e..2e703c6d 100644
--- a/common/rfb/CSecurity.h
+++ b/common/rfb/CSecurity.h
@@ -44,8 +44,9 @@ namespace rfb {
   class CConnection;
   class CSecurity {
   public:
+    CSecurity(CConnection* cc) { this->cc = cc; }
     virtual ~CSecurity() {}
-    virtual bool processMsg(CConnection* cc)=0;
+    virtual bool processMsg() = 0;
     virtual int getType() const = 0;
     virtual const char* description() const = 0;
     virtual bool isSecure() const { return false; }
@@ -55,6 +56,9 @@ namespace rfb {
      * It MUST be set by viewer.
      */
     static UserPasswdGetter *upg;
+
+  protected:
+    CConnection* cc;
   };
 }
 #endif
-- 
cgit v1.2.3