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__':