Give subclasses the ability to overwrite the reader and writer

Needed for some tests.

diff --git a/common/rfb/CConnection.h b/common/rfb/CConnection.h
index 8bd886aa34d77f25558acfa4d4ba502c02701382..0109fe863ad1f693a54d7d8b01cf542bdad456c3 100644 (file)
--- a/common/rfb/CConnection.h
+++ b/common/rfb/CConnection.h
@@ -136,6 +136,9 @@ namespace rfb {
   protected:
     void setState(stateEnum s) { state_ = s; }
 
+    void setReader(CMsgReader *r) { reader_ = r; }
+    void setWriter(CMsgWriter *w) { writer_ = w; }
+
   private:
     // This is a default implementation of fences that automatically
     // responds to requests, stating no support for synchronisation.

diff --git a/common/rfb/SConnection.h b/common/rfb/SConnection.h
index da38b3e098406ede6e9b8e04a52d5e2b5f4737b4..fd6d26348a1600850cfa4b2c40e16b865400f9ca 100644 (file)
--- a/common/rfb/SConnection.h
+++ b/common/rfb/SConnection.h
@@ -184,6 +184,9 @@ namespace rfb {
   protected:
     void setState(stateEnum s) { state_ = s; }
 
+    void setReader(SMsgReader *r) { reader_ = r; }
+    void setWriter(SMsgWriter *w) { writer_ = w; }
+
   private:
     void writeFakeColourMap(void);