48deea531c
This commit converts the send queue to use typed arrays, and converts message creation functions in 'rfb.js' to create messages directly into the socket's send queue. This commit also removes the separate mouse array, which is no longer needed.
92 lines
2.6 KiB
JavaScript
92 lines
2.6 KiB
JavaScript
var FakeWebSocket;
|
|
|
|
(function () {
|
|
// PhantomJS can't create Event objects directly, so we need to use this
|
|
function make_event(name, props) {
|
|
var evt = document.createEvent('Event');
|
|
evt.initEvent(name, true, true);
|
|
if (props) {
|
|
for (var prop in props) {
|
|
evt[prop] = props[prop];
|
|
}
|
|
}
|
|
return evt;
|
|
}
|
|
|
|
FakeWebSocket = function (uri, protocols) {
|
|
this.url = uri;
|
|
this.binaryType = "arraybuffer";
|
|
this.extensions = "";
|
|
|
|
if (!protocols || typeof protocols === 'string') {
|
|
this.protocol = protocols;
|
|
} else {
|
|
this.protocol = protocols[0];
|
|
}
|
|
|
|
this._send_queue = new Uint8Array(20000);
|
|
|
|
this.readyState = FakeWebSocket.CONNECTING;
|
|
this.bufferedAmount = 0;
|
|
|
|
this.__is_fake = true;
|
|
};
|
|
|
|
FakeWebSocket.prototype = {
|
|
close: function (code, reason) {
|
|
this.readyState = FakeWebSocket.CLOSED;
|
|
if (this.onclose) {
|
|
this.onclose(make_event("close", { 'code': code, 'reason': reason, 'wasClean': true }));
|
|
}
|
|
},
|
|
|
|
send: function (data) {
|
|
if (this.protocol == 'base64') {
|
|
data = Base64.decode(data);
|
|
} else {
|
|
data = new Uint8Array(data);
|
|
}
|
|
this._send_queue.set(data, this.bufferedAmount);
|
|
this.bufferedAmount += data.length;
|
|
},
|
|
|
|
_get_sent_data: function () {
|
|
var res = new Uint8Array(this._send_queue.buffer, 0, this.bufferedAmount);
|
|
this.bufferedAmount = 0;
|
|
return res;
|
|
},
|
|
|
|
_open: function (data) {
|
|
this.readyState = FakeWebSocket.OPEN;
|
|
if (this.onopen) {
|
|
this.onopen(make_event('open'));
|
|
}
|
|
},
|
|
|
|
_receive_data: function (data) {
|
|
this.onmessage(make_event("message", { 'data': data }));
|
|
}
|
|
};
|
|
|
|
FakeWebSocket.OPEN = WebSocket.OPEN;
|
|
FakeWebSocket.CONNECTING = WebSocket.CONNECTING;
|
|
FakeWebSocket.CLOSING = WebSocket.CLOSING;
|
|
FakeWebSocket.CLOSED = WebSocket.CLOSED;
|
|
|
|
FakeWebSocket.__is_fake = true;
|
|
|
|
FakeWebSocket.replace = function () {
|
|
if (!WebSocket.__is_fake) {
|
|
var real_version = WebSocket;
|
|
WebSocket = FakeWebSocket;
|
|
FakeWebSocket.__real_version = real_version;
|
|
}
|
|
};
|
|
|
|
FakeWebSocket.restore = function () {
|
|
if (WebSocket.__is_fake) {
|
|
WebSocket = WebSocket.__real_version;
|
|
}
|
|
};
|
|
})();
|