Sfoglia il codice sorgente

Add Javascript variable container to record data.

- This allows the recorded data to be immediately usable by
  tests/vnc_playback.html
Joel Martin 15 anni fa
parent
commit
fb007628d6
2 ha cambiato i file con 10 aggiunte e 3 eliminazioni
  1. 5 1
      tests/vnc_playback.html
  2. 5 2
      utils/wsproxy.py

+ 5 - 1
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++;
             }
 

+ 5 - 2
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__':