Add Javascript variable container to record data.

- This allows the recorded data to be immediately usable by
  tests/vnc_playback.html

diff --git a/tests/vnc_playback.html b/tests/vnc_playback.html
index 4c579415493df712b807a8b8cfbd8614c6d53953..06de8377911bca3ab9b256be351a4dfb6e8d0b04 100644 (file)
--- a/tests/vnc_playback.html
+++ b/tests/vnc_playback.html
@@ -140,7 +140,11 @@
 
             //message("Processing frame: " + frame_idx);
             if (frame) {
-                rfb.recv_message({'data' : frame.slice(frame.indexOf('{', 1)+1)});
+                if (frame === 'EOF') {
+                    //message("Found EOF");
+                } else {
+                    rfb.recv_message({'data' : frame.slice(frame.indexOf('{', 1)+1)});
+                }
                 frame_idx++;
             }
 

diff --git a/utils/wsproxy.py b/utils/wsproxy.py
index cf4c94a5d677963d0fde2f9cb0820b91441930fe..c2a7e9906fccb027eebbc1fc9c43c708524d1e48 100755 (executable)
--- a/utils/wsproxy.py
+++ b/utils/wsproxy.py
@@ -105,7 +105,8 @@ def proxy_handler(client):
 
     if settings['record']:
         print "Opening record file: %s" % settings['record']
-        rec = open(settings['record'], 'a')
+        rec = open(settings['record'], 'w+')
+        rec.write("var VNC_frame_data = [\n")
 
     print "Connecting to: %s:%s" % (target_host, target_port)
     tsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@@ -117,7 +118,9 @@ def proxy_handler(client):
         do_proxy(client, tsock)
     except:
         if tsock: tsock.close()
-        if rec: rec.close()
+        if rec:
+            rec.write("'EOF']\n")
+            rec.close()
         raise
 
 if __name__ == '__main__':