完善虚拟机连接

VM/novnc/tests/fake.websocket.js

@@ -0,0 +1,92 @@
+import Base64 from '../core/base64.js';
+
+export default class FakeWebSocket {
+    constructor(uri, protocols) {
+        this.url = uri;
+        this.binaryType = "arraybuffer";
+        this.extensions = "";
+
+        this.onerror = null;
+        this.onmessage = null;
+        this.onopen = null;
+
+        if (!protocols || typeof protocols === 'string') {
+            this.protocol = protocols;
+        } else {
+            this.protocol = protocols[0];
+        }
+
+        this._sendQueue = new Uint8Array(20000);
+
+        this.readyState = FakeWebSocket.CONNECTING;
+        this.bufferedAmount = 0;
+
+        this._isFake = true;
+    }
+
+    close(code, reason) {
+        this.readyState = FakeWebSocket.CLOSED;
+        if (this.onclose) {
+            this.onclose(new CloseEvent("close", { 'code': code, 'reason': reason, 'wasClean': true }));
+        }
+    }
+
+    send(data) {
+        if (this.protocol == 'base64') {
+            data = Base64.decode(data);
+        } else {
+            data = new Uint8Array(data);
+        }
+        this._sendQueue.set(data, this.bufferedAmount);
+        this.bufferedAmount += data.length;
+    }
+
+    _getSentData() {
+        const res = this._sendQueue.slice(0, this.bufferedAmount);
+        this.bufferedAmount = 0;
+        return res;
+    }
+
+    _open() {
+        this.readyState = FakeWebSocket.OPEN;
+        if (this.onopen) {
+            this.onopen(new Event('open'));
+        }
+    }
+
+    _receiveData(data) {
+        if (data.length < 4096) {
+            // Break apart the data to expose bugs where we assume data is
+            // neatly packaged
+            for (let i = 0;i < data.length;i++) {
+                let buf = data.slice(i, i+1);
+                this.onmessage(new MessageEvent("message", { 'data': buf.buffer }));
+            }
+        } else {
+            this.onmessage(new MessageEvent("message", { 'data': data.buffer }));
+        }
+    }
+}
+
+FakeWebSocket.OPEN = WebSocket.OPEN;
+FakeWebSocket.CONNECTING = WebSocket.CONNECTING;
+FakeWebSocket.CLOSING = WebSocket.CLOSING;
+FakeWebSocket.CLOSED = WebSocket.CLOSED;
+
+FakeWebSocket._isFake = true;
+
+FakeWebSocket.replace = () => {
+    if (!WebSocket._isFake) {
+        const realVersion = WebSocket;
+        // eslint-disable-next-line no-global-assign
+        WebSocket = FakeWebSocket;
+        FakeWebSocket._realVersion = realVersion;
+    }
+};
+
+FakeWebSocket.restore = () => {
+    if (WebSocket._isFake) {
+        // eslint-disable-next-line no-global-assign
+        WebSocket = WebSocket._realVersion;
+    }
+};