--- a/src/main/java/com/glavsoft/rfb/protocol/Protocol.java	Wed Jan 21 11:53:35 2015 +0900
+++ b/src/main/java/com/glavsoft/rfb/protocol/Protocol.java	Wed Jan 21 19:05:21 2015 +0900
@@ -66,9 +66,10 @@
     private String protocolVersion;
     private byte[] initData;
     private boolean isRetina = false;
+    private TreeRFBProto rfb;
 
     public Protocol(Reader reader, Writer writer,
-            IPasswordRetriever passwordRetriever, ProtocolSettings settings) {
+                    IPasswordRetriever passwordRetriever, ProtocolSettings settings, TreeRFBProto myRfb) {
         this.reader = reader;
         this.writer = writer;
         this.passwordRetriever = passwordRetriever;
@@ -77,6 +78,7 @@
         decoders.instantiateDecodersWhenNeeded(settings.encodings);
         state = new HandshakeState(this);
         logger = Logger.getLogger(getClass().getName());
+        this.rfb = myRfb;
     }
 
 
@@ -327,6 +329,11 @@
     }
 
     @Override
+    public TreeRFBProto getRfb() {
+        return rfb;
+    }
+
+    @Override
     public void setTight(boolean isTight) {
         this.isTight = isTight;
     }