diff --git a/common/rfb/CConnection.h b/common/rfb/CConnection.h
index 8bd886a..0109fe8 100644
--- a/common/rfb/CConnection.h
+++ b/common/rfb/CConnection.h
@@ -136,6 +136,9 @@
   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 da38b3e..fd6d263 100644
--- a/common/rfb/SConnection.h
+++ b/common/rfb/SConnection.h
@@ -184,6 +184,9 @@
   protected:
     void setState(stateEnum s) { state_ = s; }
 
+    void setReader(SMsgReader *r) { reader_ = r; }
+    void setWriter(SMsgWriter *w) { writer_ = w; }
+
   private:
     void writeFakeColourMap(void);