From fb007628d68b401fa25fb0f308fc46585274bdee Mon Sep 17 00:00:00 2001
From: Joel Martin <github@martintribe.org>
Date: Thu, 12 Aug 2010 09:50:49 -0500
Subject: [PATCH] Add Javascript variable container to record data.

- This allows the recorded data to be immediately usable by
  tests/vnc_playback.html
---
 tests/vnc_playback.html | 6 +++++-
 utils/wsproxy.py        | 7 +++++--
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/tests/vnc_playback.html b/tests/vnc_playback.html
index 4c579415..06de8377 100644
--- 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 cf4c94a5..c2a7e990 100755
--- 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__':