e9f08c3da9
* If focus was set to document.body during archive, focus left pane * Shortcut Guide: Add space between text and shortcut highlight * Ensure that draft attachment can be closed with click on X button * Move to keyDown event for user idle checking * Additional resiliency around avatars; check for them on on-disk * Increase timeouts to preserve websocket connection * On startup, be resilient to malformed JSON in log files * Don't crash if shell.openExternal returns an error * Whenever we request a contact/group sync, also request block list * Avatar popup: Ensure styling is mouse- and keyboard-appropriate * MainHeader: Create popperRoot on demand, not on mount * CompositionInput: Disable default Ctrl-/ shortcut * Update libphonenumber
243 lines
7.1 KiB
JavaScript
243 lines
7.1 KiB
JavaScript
/* global window, dcodeIO, Event, textsecure, FileReader, WebSocketResource */
|
|
|
|
// eslint-disable-next-line func-names
|
|
(function() {
|
|
/*
|
|
* WebSocket-Resources
|
|
*
|
|
* Create a request-response interface over websockets using the
|
|
* WebSocket-Resources sub-protocol[1].
|
|
*
|
|
* var client = new WebSocketResource(socket, function(request) {
|
|
* request.respond(200, 'OK');
|
|
* });
|
|
*
|
|
* client.sendRequest({
|
|
* verb: 'PUT',
|
|
* path: '/v1/messages',
|
|
* body: '{ some: "json" }',
|
|
* success: function(message, status, request) {...},
|
|
* error: function(message, status, request) {...}
|
|
* });
|
|
*
|
|
* 1. https://github.com/signalapp/WebSocket-Resources
|
|
*
|
|
*/
|
|
|
|
const Request = function Request(options) {
|
|
this.verb = options.verb || options.type;
|
|
this.path = options.path || options.url;
|
|
this.headers = options.headers;
|
|
this.body = options.body || options.data;
|
|
this.success = options.success;
|
|
this.error = options.error;
|
|
this.id = options.id;
|
|
|
|
if (this.id === undefined) {
|
|
const bits = new Uint32Array(2);
|
|
window.crypto.getRandomValues(bits);
|
|
this.id = dcodeIO.Long.fromBits(bits[0], bits[1], true);
|
|
}
|
|
|
|
if (this.body === undefined) {
|
|
this.body = null;
|
|
}
|
|
};
|
|
|
|
const IncomingWebSocketRequest = function IncomingWebSocketRequest(options) {
|
|
const request = new Request(options);
|
|
const { socket } = options;
|
|
|
|
this.verb = request.verb;
|
|
this.path = request.path;
|
|
this.body = request.body;
|
|
this.headers = request.headers;
|
|
|
|
this.respond = (status, message) => {
|
|
socket.send(
|
|
new textsecure.protobuf.WebSocketMessage({
|
|
type: textsecure.protobuf.WebSocketMessage.Type.RESPONSE,
|
|
response: { id: request.id, message, status },
|
|
})
|
|
.encode()
|
|
.toArrayBuffer()
|
|
);
|
|
};
|
|
};
|
|
|
|
const outgoing = {};
|
|
const OutgoingWebSocketRequest = function OutgoingWebSocketRequest(
|
|
options,
|
|
socket
|
|
) {
|
|
const request = new Request(options);
|
|
outgoing[request.id] = request;
|
|
socket.send(
|
|
new textsecure.protobuf.WebSocketMessage({
|
|
type: textsecure.protobuf.WebSocketMessage.Type.REQUEST,
|
|
request: {
|
|
verb: request.verb,
|
|
path: request.path,
|
|
body: request.body,
|
|
headers: request.headers,
|
|
id: request.id,
|
|
},
|
|
})
|
|
.encode()
|
|
.toArrayBuffer()
|
|
);
|
|
};
|
|
|
|
window.WebSocketResource = function WebSocketResource(socket, opts = {}) {
|
|
let { handleRequest } = opts;
|
|
if (typeof handleRequest !== 'function') {
|
|
handleRequest = request => request.respond(404, 'Not found');
|
|
}
|
|
this.sendRequest = options => new OutgoingWebSocketRequest(options, socket);
|
|
|
|
// eslint-disable-next-line no-param-reassign
|
|
socket.onmessage = socketMessage => {
|
|
const blob = socketMessage.data;
|
|
const handleArrayBuffer = buffer => {
|
|
const message = textsecure.protobuf.WebSocketMessage.decode(buffer);
|
|
if (
|
|
message.type === textsecure.protobuf.WebSocketMessage.Type.REQUEST
|
|
) {
|
|
handleRequest(
|
|
new IncomingWebSocketRequest({
|
|
verb: message.request.verb,
|
|
path: message.request.path,
|
|
body: message.request.body,
|
|
headers: message.request.headers,
|
|
id: message.request.id,
|
|
socket,
|
|
})
|
|
);
|
|
} else if (
|
|
message.type === textsecure.protobuf.WebSocketMessage.Type.RESPONSE
|
|
) {
|
|
const { response } = message;
|
|
const request = outgoing[response.id];
|
|
if (request) {
|
|
request.response = response;
|
|
let callback = request.error;
|
|
if (response.status >= 200 && response.status < 300) {
|
|
callback = request.success;
|
|
}
|
|
|
|
if (typeof callback === 'function') {
|
|
callback(response.message, response.status, request);
|
|
}
|
|
} else {
|
|
throw new Error(
|
|
`Received response for unknown request ${message.response.id}`
|
|
);
|
|
}
|
|
}
|
|
};
|
|
|
|
if (blob instanceof ArrayBuffer) {
|
|
handleArrayBuffer(blob);
|
|
} else {
|
|
const reader = new FileReader();
|
|
reader.onload = () => handleArrayBuffer(reader.result);
|
|
reader.readAsArrayBuffer(blob);
|
|
}
|
|
};
|
|
|
|
if (opts.keepalive) {
|
|
this.keepalive = new KeepAlive(this, {
|
|
path: opts.keepalive.path,
|
|
disconnect: opts.keepalive.disconnect,
|
|
});
|
|
const resetKeepAliveTimer = this.keepalive.reset.bind(this.keepalive);
|
|
socket.addEventListener('open', resetKeepAliveTimer);
|
|
socket.addEventListener('message', resetKeepAliveTimer);
|
|
socket.addEventListener(
|
|
'close',
|
|
this.keepalive.stop.bind(this.keepalive)
|
|
);
|
|
}
|
|
|
|
socket.addEventListener('close', () => {
|
|
this.closed = true;
|
|
});
|
|
|
|
this.close = (code = 3000, reason) => {
|
|
if (this.closed) {
|
|
return;
|
|
}
|
|
|
|
window.log.info('WebSocketResource.close()');
|
|
if (this.keepalive) {
|
|
this.keepalive.stop();
|
|
}
|
|
|
|
socket.close(code, reason);
|
|
// eslint-disable-next-line no-param-reassign
|
|
socket.onmessage = null;
|
|
|
|
// On linux the socket can wait a long time to emit its close event if we've
|
|
// lost the internet connection. On the order of minutes. This speeds that
|
|
// process up.
|
|
setTimeout(() => {
|
|
if (this.closed) {
|
|
return;
|
|
}
|
|
this.closed = true;
|
|
|
|
window.log.warn('Dispatching our own socket close event');
|
|
const ev = new Event('close');
|
|
ev.code = code;
|
|
ev.reason = reason;
|
|
this.dispatchEvent(ev);
|
|
}, 5000);
|
|
};
|
|
};
|
|
window.WebSocketResource.prototype = new textsecure.EventTarget();
|
|
|
|
function KeepAlive(websocketResource, opts = {}) {
|
|
if (websocketResource instanceof WebSocketResource) {
|
|
this.path = opts.path;
|
|
if (this.path === undefined) {
|
|
this.path = '/';
|
|
}
|
|
this.disconnect = opts.disconnect;
|
|
if (this.disconnect === undefined) {
|
|
this.disconnect = true;
|
|
}
|
|
this.wsr = websocketResource;
|
|
} else {
|
|
throw new TypeError('KeepAlive expected a WebSocketResource');
|
|
}
|
|
}
|
|
|
|
KeepAlive.prototype = {
|
|
constructor: KeepAlive,
|
|
stop() {
|
|
clearTimeout(this.keepAliveTimer);
|
|
clearTimeout(this.disconnectTimer);
|
|
},
|
|
reset() {
|
|
clearTimeout(this.keepAliveTimer);
|
|
clearTimeout(this.disconnectTimer);
|
|
this.keepAliveTimer = setTimeout(() => {
|
|
if (this.disconnect) {
|
|
// automatically disconnect if server doesn't ack
|
|
this.disconnectTimer = setTimeout(() => {
|
|
clearTimeout(this.keepAliveTimer);
|
|
this.wsr.close(3001, 'No response to keepalive request');
|
|
}, 10000);
|
|
} else {
|
|
this.reset();
|
|
}
|
|
window.log.info('Sending a keepalive message');
|
|
this.wsr.sendRequest({
|
|
verb: 'GET',
|
|
path: this.path,
|
|
success: this.reset.bind(this),
|
|
});
|
|
}, 55000);
|
|
},
|
|
};
|
|
})();
|