You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1946 lines
63 KiB
JavaScript
1946 lines
63 KiB
JavaScript
/* global window: false */
|
|
/* global callWorker: false */
|
|
/* global textsecure: false */
|
|
/* global StringView: false */
|
|
/* global libloki: false */
|
|
/* global libsignal: false */
|
|
/* global WebSocket: false */
|
|
/* global Event: false */
|
|
/* global dcodeIO: false */
|
|
/* global _: false */
|
|
/* global HttpResource: false */
|
|
/* global ContactBuffer: false */
|
|
/* global GroupBuffer: false */
|
|
/* global WebSocketResource: false */
|
|
/* global lokiPublicChatAPI: false */
|
|
/* global lokiMessageAPI: false */
|
|
/* global feeds: false */
|
|
/* global Whisper: false */
|
|
/* global lokiFileServerAPI: false */
|
|
/* global WebAPI: false */
|
|
/* global ConversationController: false */
|
|
/* global Signal: false */
|
|
|
|
/* eslint-disable more/no-then */
|
|
/* eslint-disable no-unreachable */
|
|
|
|
let openGroupBound = false;
|
|
|
|
function MessageReceiver(username, password, signalingKey, options = {}) {
|
|
this.count = 0;
|
|
|
|
this.signalingKey = signalingKey;
|
|
this.username = username;
|
|
this.password = password;
|
|
this.server = WebAPI.connect({ username, password });
|
|
|
|
if (!options.serverTrustRoot) {
|
|
throw new Error('Server trust root is required!');
|
|
}
|
|
this.serverTrustRoot = window.Signal.Crypto.base64ToArrayBuffer(
|
|
options.serverTrustRoot
|
|
);
|
|
|
|
const address = libsignal.SignalProtocolAddress.fromString(username);
|
|
this.number = address.getName();
|
|
this.deviceId = address.getDeviceId();
|
|
|
|
this.pending = Promise.resolve();
|
|
|
|
if (options.retryCached) {
|
|
this.pending = this.queueAllCached();
|
|
}
|
|
|
|
// only do this once to prevent duplicates
|
|
if (lokiPublicChatAPI) {
|
|
window.log.info('Binding open group events handler', openGroupBound);
|
|
if (!openGroupBound) {
|
|
// clear any previous binding
|
|
lokiPublicChatAPI.removeAllListeners('publicMessage');
|
|
// we only need one MR in the system handling these
|
|
// bind events
|
|
lokiPublicChatAPI.on(
|
|
'publicMessage',
|
|
this.handleUnencryptedMessage.bind(this)
|
|
);
|
|
openGroupBound = true;
|
|
}
|
|
} else {
|
|
window.log.error('Can not handle open group data, API is not available');
|
|
}
|
|
}
|
|
|
|
MessageReceiver.stringToArrayBuffer = string =>
|
|
Promise.resolve(dcodeIO.ByteBuffer.wrap(string, 'binary').toArrayBuffer());
|
|
MessageReceiver.arrayBufferToString = arrayBuffer =>
|
|
Promise.resolve(dcodeIO.ByteBuffer.wrap(arrayBuffer).toString('binary'));
|
|
|
|
MessageReceiver.stringToArrayBufferBase64 = string =>
|
|
callWorker('stringToArrayBufferBase64', string);
|
|
MessageReceiver.arrayBufferToStringBase64 = arrayBuffer =>
|
|
callWorker('arrayBufferToStringBase64', arrayBuffer);
|
|
|
|
MessageReceiver.prototype = new textsecure.EventTarget();
|
|
MessageReceiver.prototype.extend({
|
|
constructor: MessageReceiver,
|
|
connect() {
|
|
if (this.calledClose) {
|
|
return;
|
|
}
|
|
|
|
this.count = 0;
|
|
if (this.hasConnected) {
|
|
const ev = new Event('reconnect');
|
|
this.dispatchEvent(ev);
|
|
}
|
|
|
|
this.hasConnected = true;
|
|
this.httpPollingResource = new HttpResource(lokiMessageAPI, {
|
|
handleRequest: this.handleRequest.bind(this),
|
|
});
|
|
this.httpPollingResource.pollServer();
|
|
|
|
// start polling all open group rooms you have registered
|
|
// if not registered yet, they'll get started when they're created
|
|
if (lokiPublicChatAPI) {
|
|
lokiPublicChatAPI.open();
|
|
}
|
|
// set up pollers for any RSS feeds
|
|
feeds.forEach(feed => {
|
|
feed.on('rssMessage', this.handleUnencryptedMessage.bind(this));
|
|
});
|
|
|
|
// TODO: Rework this socket stuff to work with online messaging
|
|
const useWebSocket = false;
|
|
if (useWebSocket) {
|
|
if (this.socket && this.socket.readyState !== WebSocket.CLOSED) {
|
|
this.socket.close();
|
|
this.wsr.close();
|
|
}
|
|
// initialize the socket and start listening for messages
|
|
this.socket = this.server.getMessageSocket();
|
|
this.socket.onclose = this.onclose.bind(this);
|
|
this.socket.onerror = this.onerror.bind(this);
|
|
this.socket.onopen = this.onopen.bind(this);
|
|
this.wsr = new WebSocketResource(this.socket, {
|
|
handleRequest: this.handleRequest.bind(this),
|
|
keepalive: {
|
|
path: '/v1/keepalive',
|
|
disconnect: true,
|
|
},
|
|
});
|
|
|
|
// Because sometimes the socket doesn't properly emit its close event
|
|
this._onClose = this.onclose.bind(this);
|
|
this.wsr.addEventListener('close', this._onClose);
|
|
}
|
|
|
|
// Ensures that an immediate 'empty' event from the websocket will fire only after
|
|
// all cached envelopes are processed.
|
|
this.incoming = [this.pending];
|
|
},
|
|
async handleUnencryptedMessage({ message }) {
|
|
const isMe = message.source === textsecure.storage.user.getNumber();
|
|
if (!isMe && message.message.profile) {
|
|
const conversation = await window.ConversationController.getOrCreateAndWait(
|
|
message.source,
|
|
'private'
|
|
);
|
|
await this.updateProfile(
|
|
conversation,
|
|
message.message.profile,
|
|
message.message.profileKey
|
|
);
|
|
}
|
|
|
|
const ev = new Event('message');
|
|
ev.confirm = function confirmTerm() {};
|
|
ev.data = message;
|
|
this.dispatchAndWait(ev);
|
|
},
|
|
stopProcessing() {
|
|
window.log.info('MessageReceiver: stopProcessing requested');
|
|
this.stoppingProcessing = true;
|
|
return this.close();
|
|
},
|
|
shutdown() {
|
|
if (this.socket) {
|
|
this.socket.onclose = null;
|
|
this.socket.onerror = null;
|
|
this.socket.onopen = null;
|
|
this.socket = null;
|
|
}
|
|
|
|
if (this.wsr) {
|
|
this.wsr.removeEventListener('close', this._onClose);
|
|
this.wsr = null;
|
|
}
|
|
},
|
|
async close() {
|
|
window.log.info('MessageReceiver.close()');
|
|
this.calledClose = true;
|
|
|
|
// Our WebSocketResource instance will close the socket and emit a 'close' event
|
|
// if the socket doesn't emit one quickly enough.
|
|
if (this.wsr) {
|
|
this.wsr.close(3000, 'called close');
|
|
}
|
|
|
|
// stop polling all open group rooms
|
|
if (lokiPublicChatAPI) {
|
|
await lokiPublicChatAPI.close();
|
|
}
|
|
|
|
if (this.httpPollingResource) {
|
|
this.httpPollingResource.close();
|
|
}
|
|
|
|
return this.drain();
|
|
},
|
|
onopen() {
|
|
window.log.info('websocket open');
|
|
},
|
|
onerror() {
|
|
window.log.error('websocket error');
|
|
},
|
|
dispatchAndWait(event) {
|
|
const promise = this.appPromise || Promise.resolve();
|
|
const appJobPromise = Promise.all(this.dispatchEvent(event));
|
|
const job = () => appJobPromise;
|
|
|
|
this.appPromise = promise.then(job, job);
|
|
|
|
return Promise.resolve();
|
|
},
|
|
onclose(ev) {
|
|
window.log.info(
|
|
'websocket closed',
|
|
ev.code,
|
|
ev.reason || '',
|
|
'calledClose:',
|
|
this.calledClose
|
|
);
|
|
// TODO: handle properly
|
|
// this.shutdown();
|
|
|
|
// if (this.calledClose) {
|
|
// return Promise.resolve();
|
|
// }
|
|
// if (ev.code === 3000) {
|
|
// return Promise.resolve();
|
|
// }
|
|
// if (ev.code === 3001) {
|
|
// this.onEmpty();
|
|
// }
|
|
// // possible 403 or network issue. Make an request to confirm
|
|
// return this.server
|
|
// .getDevices(this.number)
|
|
// .then(this.connect.bind(this)) // No HTTP error? Reconnect
|
|
// .catch(e => {
|
|
// const event = new Event('error');
|
|
// event.error = e;
|
|
// return this.dispatchAndWait(event);
|
|
// });
|
|
},
|
|
handleRequest(request, options) {
|
|
const { onSuccess, onFailure } = options;
|
|
this.incoming = this.incoming || [];
|
|
const lastPromise = _.last(this.incoming);
|
|
|
|
// We do the message decryption here, instead of in the ordered pending queue,
|
|
// to avoid exposing the time it took us to process messages through the time-to-ack.
|
|
|
|
if (request.path !== '/api/v1/message') {
|
|
window.log.info('got request', request.verb, request.path);
|
|
request.respond(200, 'OK');
|
|
|
|
if (request.verb === 'PUT' && request.path === '/api/v1/queue/empty') {
|
|
this.onEmpty();
|
|
}
|
|
return;
|
|
}
|
|
|
|
const promise = Promise.resolve(request.body.toArrayBuffer()) // textsecure.crypto
|
|
.then(plaintext => {
|
|
const envelope = textsecure.protobuf.Envelope.decode(plaintext);
|
|
// After this point, decoding errors are not the server's
|
|
// fault, and we should handle them gracefully and tell the
|
|
// user they received an invalid message
|
|
|
|
if (this.isBlocked(envelope.source)) {
|
|
return request.respond(200, 'OK');
|
|
}
|
|
|
|
envelope.id = envelope.serverGuid || window.getGuid();
|
|
envelope.serverTimestamp = envelope.serverTimestamp
|
|
? envelope.serverTimestamp.toNumber()
|
|
: null;
|
|
|
|
return this.addToCache(envelope, plaintext).then(
|
|
async () => {
|
|
request.respond(200, 'OK');
|
|
|
|
// To ensure that we queue in the same order we receive messages
|
|
await lastPromise;
|
|
this.queueEnvelope(envelope, onSuccess, onFailure);
|
|
},
|
|
error => {
|
|
request.respond(500, 'Failed to cache message');
|
|
window.log.error(
|
|
'handleRequest error trying to add message to cache:',
|
|
error && error.stack ? error.stack : error
|
|
);
|
|
}
|
|
);
|
|
})
|
|
.catch(e => {
|
|
request.respond(500, 'Bad encrypted websocket message');
|
|
window.log.error(
|
|
'Error handling incoming message:',
|
|
e && e.stack ? e.stack : e
|
|
);
|
|
const ev = new Event('error');
|
|
ev.error = e;
|
|
return this.dispatchAndWait(ev);
|
|
});
|
|
|
|
this.incoming.push(promise);
|
|
},
|
|
addToQueue(task) {
|
|
this.count += 1;
|
|
this.pending = this.pending.then(task, task);
|
|
|
|
const { count, pending } = this;
|
|
|
|
const cleanup = () => {
|
|
this.updateProgress(count);
|
|
// We want to clear out the promise chain whenever possible because it could
|
|
// lead to large memory usage over time:
|
|
// https://github.com/nodejs/node/issues/6673#issuecomment-244331609
|
|
if (this.pending === pending) {
|
|
this.pending = Promise.resolve();
|
|
}
|
|
};
|
|
|
|
pending.then(cleanup, cleanup);
|
|
|
|
return pending;
|
|
},
|
|
onEmpty() {
|
|
const { incoming } = this;
|
|
this.incoming = [];
|
|
|
|
const emitEmpty = () => {
|
|
window.log.info("MessageReceiver: emitting 'empty' event");
|
|
const ev = new Event('empty');
|
|
this.dispatchAndWait(ev);
|
|
};
|
|
|
|
const waitForApplication = async () => {
|
|
window.log.info(
|
|
"MessageReceiver: finished processing messages after 'empty', now waiting for application"
|
|
);
|
|
const promise = this.appPromise || Promise.resolve();
|
|
this.appPromise = Promise.resolve();
|
|
|
|
// We don't await here because we don't this to gate future message processing
|
|
promise.then(emitEmpty, emitEmpty);
|
|
};
|
|
|
|
const waitForEmptyQueue = () => {
|
|
// resetting count to zero so everything queued after this starts over again
|
|
this.count = 0;
|
|
|
|
this.addToQueue(waitForApplication);
|
|
};
|
|
|
|
// We first wait for all recently-received messages (this.incoming) to be queued,
|
|
// then we queue a task to wait for the application to finish its processing, then
|
|
// finally we emit the 'empty' event to the queue.
|
|
Promise.all(incoming).then(waitForEmptyQueue, waitForEmptyQueue);
|
|
},
|
|
drain() {
|
|
const { incoming } = this;
|
|
this.incoming = [];
|
|
|
|
const queueDispatch = () =>
|
|
this.addToQueue(() => {
|
|
window.log.info('drained');
|
|
});
|
|
|
|
// This promise will resolve when there are no more messages to be processed.
|
|
return Promise.all(incoming).then(queueDispatch, queueDispatch);
|
|
},
|
|
updateProgress(count) {
|
|
// count by 10s
|
|
if (count % 10 !== 0) {
|
|
return;
|
|
}
|
|
const ev = new Event('progress');
|
|
ev.count = count;
|
|
this.dispatchEvent(ev);
|
|
},
|
|
async queueAllCached() {
|
|
const items = await this.getAllFromCache();
|
|
for (let i = 0, max = items.length; i < max; i += 1) {
|
|
// eslint-disable-next-line no-await-in-loop
|
|
await this.queueCached(items[i]);
|
|
}
|
|
},
|
|
async queueCached(item) {
|
|
try {
|
|
let envelopePlaintext = item.envelope;
|
|
|
|
if (item.version === 2) {
|
|
envelopePlaintext = await MessageReceiver.stringToArrayBufferBase64(
|
|
envelopePlaintext
|
|
);
|
|
}
|
|
|
|
if (typeof envelopePlaintext === 'string') {
|
|
envelopePlaintext = await MessageReceiver.stringToArrayBuffer(
|
|
envelopePlaintext
|
|
);
|
|
}
|
|
const envelope = textsecure.protobuf.Envelope.decode(envelopePlaintext);
|
|
envelope.id = envelope.serverGuid || item.id;
|
|
envelope.source = envelope.source || item.source;
|
|
envelope.sourceDevice = envelope.sourceDevice || item.sourceDevice;
|
|
envelope.serverTimestamp =
|
|
envelope.serverTimestamp || item.serverTimestamp;
|
|
envelope.preKeyBundleMessage =
|
|
envelope.preKeyBundleMessage || item.preKeyBundleMessage;
|
|
|
|
const { decrypted } = item;
|
|
if (decrypted) {
|
|
let payloadPlaintext = decrypted;
|
|
|
|
if (item.version === 2) {
|
|
payloadPlaintext = await MessageReceiver.stringToArrayBufferBase64(
|
|
payloadPlaintext
|
|
);
|
|
}
|
|
|
|
if (typeof payloadPlaintext === 'string') {
|
|
payloadPlaintext = await MessageReceiver.stringToArrayBuffer(
|
|
payloadPlaintext
|
|
);
|
|
}
|
|
|
|
// Convert preKeys to array buffer
|
|
if (typeof envelope.preKeyBundleMessage === 'string') {
|
|
envelope.preKeyBundleMessage = await MessageReceiver.stringToArrayBuffer(
|
|
envelope.preKeyBundleMessage
|
|
);
|
|
}
|
|
this.queueDecryptedEnvelope(envelope, payloadPlaintext);
|
|
} else {
|
|
this.queueEnvelope(envelope);
|
|
}
|
|
} catch (error) {
|
|
window.log.error(
|
|
'queueCached error handling item',
|
|
item.id,
|
|
'removing it. Error:',
|
|
error && error.stack ? error.stack : error
|
|
);
|
|
|
|
try {
|
|
const { id } = item;
|
|
await textsecure.storage.unprocessed.remove(id);
|
|
} catch (deleteError) {
|
|
window.log.error(
|
|
'queueCached error deleting item',
|
|
item.id,
|
|
'Error:',
|
|
deleteError && deleteError.stack ? deleteError.stack : deleteError
|
|
);
|
|
}
|
|
}
|
|
},
|
|
getEnvelopeId(envelope) {
|
|
if (envelope.source) {
|
|
return `${envelope.source}.${
|
|
envelope.sourceDevice
|
|
} ${envelope.timestamp.toNumber()} (${envelope.id})`;
|
|
}
|
|
|
|
return envelope.id;
|
|
},
|
|
async getAllFromCache() {
|
|
window.log.info('getAllFromCache');
|
|
const count = await textsecure.storage.unprocessed.getCount();
|
|
|
|
if (count > 1500) {
|
|
await textsecure.storage.unprocessed.removeAll();
|
|
window.log.warn(
|
|
`There were ${count} messages in cache. Deleted all instead of reprocessing`
|
|
);
|
|
return [];
|
|
}
|
|
|
|
const items = await textsecure.storage.unprocessed.getAll();
|
|
window.log.info('getAllFromCache loaded', items.length, 'saved envelopes');
|
|
|
|
return Promise.all(
|
|
_.map(items, async item => {
|
|
const attempts = 1 + (item.attempts || 0);
|
|
|
|
try {
|
|
if (attempts >= 3) {
|
|
window.log.warn(
|
|
'getAllFromCache final attempt for envelope',
|
|
item.id
|
|
);
|
|
await textsecure.storage.unprocessed.remove(item.id);
|
|
} else {
|
|
await textsecure.storage.unprocessed.updateAttempts(
|
|
item.id,
|
|
attempts
|
|
);
|
|
}
|
|
} catch (error) {
|
|
window.log.error(
|
|
'getAllFromCache error updating item after load:',
|
|
error && error.stack ? error.stack : error
|
|
);
|
|
}
|
|
|
|
return item;
|
|
})
|
|
);
|
|
},
|
|
async addToCache(envelope, plaintext) {
|
|
const { id } = envelope;
|
|
const data = {
|
|
id,
|
|
version: 2,
|
|
envelope: await MessageReceiver.arrayBufferToStringBase64(plaintext),
|
|
timestamp: Date.now(),
|
|
attempts: 1,
|
|
};
|
|
return textsecure.storage.unprocessed.add(data);
|
|
},
|
|
async updateCache(envelope, plaintext) {
|
|
const { id } = envelope;
|
|
const item = await textsecure.storage.unprocessed.get(id);
|
|
if (!item) {
|
|
window.log.error(
|
|
`updateCache: Didn't find item ${id} in cache to update`
|
|
);
|
|
return null;
|
|
}
|
|
|
|
item.source = envelope.source;
|
|
item.sourceDevice = envelope.sourceDevice;
|
|
item.serverTimestamp = envelope.serverTimestamp;
|
|
|
|
if (item.version === 2) {
|
|
item.decrypted = await MessageReceiver.arrayBufferToStringBase64(
|
|
plaintext
|
|
);
|
|
} else {
|
|
item.decrypted = await MessageReceiver.arrayBufferToString(plaintext);
|
|
}
|
|
|
|
return textsecure.storage.unprocessed.addDecryptedData(item.id, item);
|
|
},
|
|
removeFromCache(envelope) {
|
|
const { id } = envelope;
|
|
return textsecure.storage.unprocessed.remove(id);
|
|
},
|
|
queueDecryptedEnvelope(envelope, plaintext) {
|
|
const id = this.getEnvelopeId(envelope);
|
|
window.log.info('queueing decrypted envelope', id);
|
|
|
|
const task = this.handleDecryptedEnvelope.bind(this, envelope, plaintext);
|
|
const taskWithTimeout = textsecure.createTaskWithTimeout(
|
|
task,
|
|
`queueEncryptedEnvelope ${id}`
|
|
);
|
|
const promise = this.addToQueue(taskWithTimeout);
|
|
|
|
return promise.catch(error => {
|
|
window.log.error(
|
|
`queueDecryptedEnvelope error handling envelope ${id}:`,
|
|
error && error.stack ? error.stack : error
|
|
);
|
|
});
|
|
},
|
|
queueEnvelope(envelope, onSuccess = null, onFailure = null) {
|
|
const id = this.getEnvelopeId(envelope);
|
|
window.log.info('queueing envelope', id);
|
|
|
|
const task = this.handleEnvelope.bind(this, envelope);
|
|
const taskWithTimeout = textsecure.createTaskWithTimeout(
|
|
task,
|
|
`queueEnvelope ${id}`
|
|
);
|
|
const promise = this.addToQueue(taskWithTimeout);
|
|
promise.then(() => {
|
|
if (onSuccess) {
|
|
onSuccess();
|
|
}
|
|
});
|
|
|
|
return promise.catch(error => {
|
|
window.log.error(
|
|
'queueEnvelope error handling envelope',
|
|
id,
|
|
':',
|
|
error && error.stack ? error.stack : error
|
|
);
|
|
if (onFailure) {
|
|
onFailure();
|
|
}
|
|
});
|
|
},
|
|
// Same as handleEnvelope, just without the decryption step. Necessary for handling
|
|
// messages which were successfully decrypted, but application logic didn't finish
|
|
// processing.
|
|
handleDecryptedEnvelope(envelope, plaintext) {
|
|
if (this.stoppingProcessing) {
|
|
return Promise.resolve();
|
|
}
|
|
// No decryption is required for delivery receipts, so the decrypted field of
|
|
// the Unprocessed model will never be set
|
|
|
|
if (envelope.content) {
|
|
return this.innerHandleContentMessage(envelope, plaintext);
|
|
} else if (envelope.legacyMessage) {
|
|
return this.innerHandleLegacyMessage(envelope, plaintext);
|
|
}
|
|
this.removeFromCache(envelope);
|
|
throw new Error('Received message with no content and no legacyMessage');
|
|
},
|
|
handleEnvelope(envelope) {
|
|
if (this.stoppingProcessing) {
|
|
return Promise.resolve();
|
|
}
|
|
|
|
if (envelope.type === textsecure.protobuf.Envelope.Type.RECEIPT) {
|
|
return this.onDeliveryReceipt(envelope);
|
|
}
|
|
|
|
if (envelope.content) {
|
|
return this.handleContentMessage(envelope);
|
|
}
|
|
if (envelope.legacyMessage) {
|
|
return this.handleLegacyMessage(envelope);
|
|
}
|
|
this.removeFromCache(envelope);
|
|
throw new Error('Received message with no content and no legacyMessage');
|
|
},
|
|
getStatus() {
|
|
if (this.httpPollingResource) {
|
|
return this.httpPollingResource.isConnected()
|
|
? WebSocket.OPEN
|
|
: WebSocket.CLOSED;
|
|
}
|
|
if (this.socket) {
|
|
return this.socket.readyState;
|
|
} else if (this.hasConnected) {
|
|
return WebSocket.CLOSED;
|
|
}
|
|
return -1;
|
|
},
|
|
onDeliveryReceipt(envelope) {
|
|
return new Promise((resolve, reject) => {
|
|
const ev = new Event('delivery');
|
|
ev.confirm = this.removeFromCache.bind(this, envelope);
|
|
ev.deliveryReceipt = {
|
|
timestamp: envelope.timestamp.toNumber(),
|
|
source: envelope.source,
|
|
sourceDevice: envelope.sourceDevice,
|
|
};
|
|
this.dispatchAndWait(ev).then(resolve, reject);
|
|
});
|
|
},
|
|
unpad(paddedData) {
|
|
const paddedPlaintext = new Uint8Array(paddedData);
|
|
let plaintext;
|
|
|
|
for (let i = paddedPlaintext.length - 1; i >= 0; i -= 1) {
|
|
if (paddedPlaintext[i] === 0x80) {
|
|
plaintext = new Uint8Array(i);
|
|
plaintext.set(paddedPlaintext.subarray(0, i));
|
|
plaintext = plaintext.buffer;
|
|
break;
|
|
} else if (paddedPlaintext[i] !== 0x00) {
|
|
throw new Error('Invalid padding');
|
|
}
|
|
}
|
|
|
|
return plaintext;
|
|
},
|
|
async decrypt(envelope, ciphertext) {
|
|
let promise;
|
|
|
|
const ourNumber = textsecure.storage.user.getNumber();
|
|
const me = {
|
|
number: ourNumber,
|
|
deviceId: parseInt(textsecure.storage.user.getDeviceId(), 10),
|
|
};
|
|
|
|
// Envelope.source will be null on UNIDENTIFIED_SENDER
|
|
// Don't use it there!
|
|
const address = new libsignal.SignalProtocolAddress(
|
|
envelope.source,
|
|
envelope.sourceDevice
|
|
);
|
|
|
|
const lokiSessionCipher = new libloki.crypto.LokiSessionCipher(
|
|
textsecure.storage.protocol,
|
|
address
|
|
);
|
|
|
|
switch (envelope.type) {
|
|
case textsecure.protobuf.Envelope.Type.CIPHERTEXT:
|
|
window.log.info('message from', this.getEnvelopeId(envelope));
|
|
promise = lokiSessionCipher
|
|
.decryptWhisperMessage(ciphertext)
|
|
.then(this.unpad);
|
|
break;
|
|
case textsecure.protobuf.Envelope.Type.FRIEND_REQUEST: {
|
|
window.log.info('friend-request message from ', envelope.source);
|
|
|
|
const fallBackSessionCipher = new libloki.crypto.FallBackSessionCipher(
|
|
address
|
|
);
|
|
|
|
promise = fallBackSessionCipher
|
|
.decrypt(ciphertext.toArrayBuffer())
|
|
.then(this.unpad);
|
|
break;
|
|
}
|
|
case textsecure.protobuf.Envelope.Type.PREKEY_BUNDLE:
|
|
window.log.info('prekey message from', this.getEnvelopeId(envelope));
|
|
promise = this.decryptPreKeyWhisperMessage(
|
|
ciphertext,
|
|
lokiSessionCipher,
|
|
address
|
|
);
|
|
break;
|
|
case textsecure.protobuf.Envelope.Type.UNIDENTIFIED_SENDER: {
|
|
window.log.info('received unidentified sender message');
|
|
|
|
const secretSessionCipher = new window.Signal.Metadata.SecretSessionCipher(
|
|
textsecure.storage.protocol
|
|
);
|
|
|
|
promise = secretSessionCipher
|
|
.decrypt(ciphertext.toArrayBuffer(), me)
|
|
.then(
|
|
result => {
|
|
const { isMe, sender, content, type } = result;
|
|
|
|
// We need to drop incoming messages from ourself since server can't
|
|
// do it for us
|
|
if (isMe) {
|
|
return { isMe: true };
|
|
}
|
|
|
|
// We might have substituted the type based on decrypted content
|
|
if (type === textsecure.protobuf.Envelope.Type.FRIEND_REQUEST) {
|
|
// eslint-disable-next-line no-param-reassign
|
|
envelope.type =
|
|
textsecure.protobuf.Envelope.Type.FRIEND_REQUEST;
|
|
}
|
|
|
|
if (this.isBlocked(sender.getName())) {
|
|
window.log.info(
|
|
'Dropping blocked message after sealed sender decryption'
|
|
);
|
|
return { isBlocked: true };
|
|
}
|
|
|
|
// Here we take this sender information and attach it back to the envelope
|
|
// to make the rest of the app work properly.
|
|
|
|
const originalSource = envelope.source;
|
|
|
|
// eslint-disable-next-line no-param-reassign
|
|
envelope.source = sender.getName();
|
|
// eslint-disable-next-line no-param-reassign
|
|
envelope.sourceDevice = sender.getDeviceId();
|
|
// eslint-disable-next-line no-param-reassign
|
|
envelope.unidentifiedDeliveryReceived = !originalSource;
|
|
|
|
// Return just the content because that matches the signature of the other
|
|
// decrypt methods used above.
|
|
return this.unpad(content);
|
|
},
|
|
error => {
|
|
const { sender } = error || {};
|
|
|
|
if (sender) {
|
|
const originalSource = envelope.source;
|
|
|
|
if (this.isBlocked(sender.getName())) {
|
|
window.log.info(
|
|
'Dropping blocked message with error after sealed sender decryption'
|
|
);
|
|
return { isBlocked: true };
|
|
}
|
|
|
|
// eslint-disable-next-line no-param-reassign
|
|
envelope.source = sender.getName();
|
|
// eslint-disable-next-line no-param-reassign
|
|
envelope.sourceDevice = sender.getDeviceId();
|
|
// eslint-disable-next-line no-param-reassign
|
|
envelope.unidentifiedDeliveryReceived = !originalSource;
|
|
|
|
throw error;
|
|
}
|
|
|
|
return this.removeFromCache(envelope).then(() => {
|
|
throw error;
|
|
});
|
|
}
|
|
);
|
|
break;
|
|
}
|
|
default:
|
|
promise = Promise.reject(new Error('Unknown message type'));
|
|
}
|
|
|
|
return promise
|
|
.then(async plaintext => {
|
|
const { isMe, isBlocked } = plaintext || {};
|
|
if (isMe || isBlocked) {
|
|
this.removeFromCache(envelope);
|
|
return null;
|
|
}
|
|
|
|
let conversation;
|
|
try {
|
|
conversation = await window.ConversationController.getOrCreateAndWait(
|
|
envelope.source,
|
|
'private'
|
|
);
|
|
} catch (e) {
|
|
window.log.info('Error getting conversation: ', envelope.source);
|
|
}
|
|
|
|
// Type here can actually be UNIDENTIFIED_SENDER even if
|
|
// the underlying message is FRIEND_REQUEST
|
|
if (
|
|
envelope.type !== textsecure.protobuf.Envelope.Type.FRIEND_REQUEST
|
|
) {
|
|
// If we got here there is a valid session, which meants friend request
|
|
// is complete (if it wasn't already)
|
|
if (conversation) {
|
|
const isFriendRequestAccept = await conversation.onFriendRequestAccepted();
|
|
if (isFriendRequestAccept) {
|
|
await conversation.notifyFriendRequest(
|
|
envelope.source,
|
|
'accepted'
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
this.updateCache(envelope, plaintext).catch(error => {
|
|
window.log.error(
|
|
'decrypt failed to save decrypted message contents to cache:',
|
|
error && error.stack ? error.stack : error
|
|
);
|
|
});
|
|
|
|
return plaintext;
|
|
})
|
|
.catch(error => {
|
|
let errorToThrow = error;
|
|
|
|
const noSession =
|
|
error &&
|
|
(error.message.indexOf('No record for device') === 0 ||
|
|
error.message.indexOf('decryptWithSessionList: list is empty') ===
|
|
0);
|
|
|
|
if (error && error.message === 'Unknown identity key') {
|
|
// create an error that the UI will pick up and ask the
|
|
// user if they want to re-negotiate
|
|
const buffer = dcodeIO.ByteBuffer.wrap(ciphertext);
|
|
errorToThrow = new textsecure.IncomingIdentityKeyError(
|
|
address.toString(),
|
|
buffer.toArrayBuffer(),
|
|
error.identityKey
|
|
);
|
|
} else if (!noSession) {
|
|
// We want to handle "no-session" error, not re-throw it
|
|
throw error;
|
|
}
|
|
const ev = new Event('error');
|
|
ev.error = errorToThrow;
|
|
ev.proto = envelope;
|
|
ev.confirm = this.removeFromCache.bind(this, envelope);
|
|
|
|
const returnError = () => Promise.reject(errorToThrow);
|
|
return this.dispatchAndWait(ev).then(returnError, returnError);
|
|
});
|
|
},
|
|
async decryptPreKeyWhisperMessage(ciphertext, sessionCipher, address) {
|
|
const padded = await sessionCipher.decryptPreKeyWhisperMessage(ciphertext);
|
|
|
|
try {
|
|
return this.unpad(padded);
|
|
} catch (e) {
|
|
if (e.message === 'Unknown identity key') {
|
|
// create an error that the UI will pick up and ask the
|
|
// user if they want to re-negotiate
|
|
const buffer = dcodeIO.ByteBuffer.wrap(ciphertext);
|
|
throw new textsecure.IncomingIdentityKeyError(
|
|
address.toString(),
|
|
buffer.toArrayBuffer(),
|
|
e.identityKey
|
|
);
|
|
}
|
|
throw e;
|
|
}
|
|
},
|
|
// handle a SYNC message for a message
|
|
// sent by another device
|
|
handleSentMessage(envelope, sentContainer, msg) {
|
|
const {
|
|
destination,
|
|
timestamp,
|
|
expirationStartTimestamp,
|
|
unidentifiedStatus,
|
|
} = sentContainer;
|
|
|
|
let p = Promise.resolve();
|
|
// eslint-disable-next-line no-bitwise
|
|
if (msg.flags & textsecure.protobuf.DataMessage.Flags.END_SESSION) {
|
|
p = this.handleEndSession(destination);
|
|
}
|
|
return p.then(() =>
|
|
this.processDecrypted(envelope, msg).then(message => {
|
|
const groupId = message.group && message.group.id;
|
|
const isBlocked = this.isGroupBlocked(groupId);
|
|
const primaryDevicePubKey = window.storage.get('primaryDevicePubKey');
|
|
const isMe =
|
|
envelope.source === textsecure.storage.user.getNumber() ||
|
|
envelope.source === primaryDevicePubKey;
|
|
const isLeavingGroup = Boolean(
|
|
message.group &&
|
|
message.group.type === textsecure.protobuf.GroupContext.Type.QUIT
|
|
);
|
|
|
|
if (groupId && isBlocked && !(isMe && isLeavingGroup)) {
|
|
window.log.warn(
|
|
`Message ${this.getEnvelopeId(
|
|
envelope
|
|
)} ignored; destined for blocked group`
|
|
);
|
|
return this.removeFromCache(envelope);
|
|
}
|
|
|
|
// handle profileKey and avatar updates
|
|
if (envelope.source === primaryDevicePubKey) {
|
|
const { profileKey, profile } = message;
|
|
const primaryConversation = ConversationController.get(
|
|
primaryDevicePubKey
|
|
);
|
|
if (profile) {
|
|
this.updateProfile(primaryConversation, profile, profileKey);
|
|
}
|
|
}
|
|
|
|
const ev = new Event('sent');
|
|
ev.confirm = this.removeFromCache.bind(this, envelope);
|
|
ev.data = {
|
|
destination,
|
|
timestamp: timestamp.toNumber(),
|
|
device: envelope.sourceDevice,
|
|
unidentifiedStatus,
|
|
message,
|
|
};
|
|
if (expirationStartTimestamp) {
|
|
ev.data.expirationStartTimestamp = expirationStartTimestamp.toNumber();
|
|
}
|
|
return this.dispatchAndWait(ev);
|
|
})
|
|
);
|
|
},
|
|
async handleLokiAddressMessage(envelope) {
|
|
window.log.warn('Ignoring a Loki address message');
|
|
return this.removeFromCache(envelope);
|
|
},
|
|
async handlePairingRequest(envelope, pairingRequest) {
|
|
const valid = await libloki.crypto.validateAuthorisation(pairingRequest);
|
|
if (valid) {
|
|
// Pairing dialog is open and is listening
|
|
if (Whisper.events.isListenedTo('devicePairingRequestReceived')) {
|
|
await window.libloki.storage.savePairingAuthorisation(pairingRequest);
|
|
Whisper.events.trigger(
|
|
'devicePairingRequestReceived',
|
|
pairingRequest.secondaryDevicePubKey
|
|
);
|
|
}
|
|
// Ignore requests if the dialog is closed
|
|
}
|
|
return this.removeFromCache(envelope);
|
|
},
|
|
async handleAuthorisationForSelf(
|
|
envelope,
|
|
pairingAuthorisation,
|
|
{ dataMessage, syncMessage }
|
|
) {
|
|
const valid = await libloki.crypto.validateAuthorisation(
|
|
pairingAuthorisation
|
|
);
|
|
const alreadySecondaryDevice = !!window.storage.get('isSecondaryDevice');
|
|
let removedFromCache = false;
|
|
if (alreadySecondaryDevice) {
|
|
window.log.warn(
|
|
'Received an unexpected pairing authorisation (device is already paired as secondary device). Ignoring.'
|
|
);
|
|
} else if (!valid) {
|
|
window.log.warn(
|
|
'Received invalid pairing authorisation for self. Could not verify signature. Ignoring.'
|
|
);
|
|
} else {
|
|
const { primaryDevicePubKey, grantSignature } = pairingAuthorisation;
|
|
if (grantSignature) {
|
|
// Authorisation received to become a secondary device
|
|
window.log.info(
|
|
`Received pairing authorisation from ${primaryDevicePubKey}`
|
|
);
|
|
// Set current device as secondary.
|
|
// This will ensure the authorisation is sent
|
|
// along with each friend request.
|
|
window.storage.remove('secondaryDeviceStatus');
|
|
window.storage.put('isSecondaryDevice', true);
|
|
window.storage.put('primaryDevicePubKey', primaryDevicePubKey);
|
|
await libloki.storage.savePairingAuthorisation(pairingAuthorisation);
|
|
const primaryConversation = await ConversationController.getOrCreateAndWait(
|
|
primaryDevicePubKey,
|
|
'private'
|
|
);
|
|
primaryConversation.trigger('change');
|
|
Whisper.events.trigger('secondaryDeviceRegistration');
|
|
// Update profile
|
|
if (dataMessage) {
|
|
const { profile, profileKey } = dataMessage;
|
|
const ourNumber = window.storage.get('primaryDevicePubKey');
|
|
const me = window.ConversationController.get(ourNumber);
|
|
if (me) {
|
|
this.updateProfile(me, profile, profileKey);
|
|
}
|
|
}
|
|
// Update contact list
|
|
if (syncMessage && syncMessage.contacts) {
|
|
// This call already removes the envelope from the cache
|
|
await this.handleContacts(envelope, syncMessage.contacts);
|
|
removedFromCache = true;
|
|
}
|
|
} else {
|
|
window.log.warn('Unimplemented pairing authorisation message type');
|
|
}
|
|
}
|
|
if (!removedFromCache) {
|
|
await this.removeFromCache(envelope);
|
|
}
|
|
},
|
|
async sendFriendRequestsToSyncContacts(contacts) {
|
|
const attachmentPointer = await this.handleAttachment(contacts);
|
|
const contactBuffer = new ContactBuffer(attachmentPointer.data);
|
|
let contactDetails = contactBuffer.next();
|
|
// Extract just the pubkeys
|
|
const friendPubKeys = [];
|
|
while (contactDetails !== undefined) {
|
|
friendPubKeys.push(contactDetails.number);
|
|
contactDetails = contactBuffer.next();
|
|
}
|
|
return Promise.all(
|
|
friendPubKeys.map(async pubKey => {
|
|
const c = await window.ConversationController.getOrCreateAndWait(
|
|
pubKey,
|
|
'private'
|
|
);
|
|
if (!c) {
|
|
return null;
|
|
}
|
|
const attachments = [];
|
|
const quote = null;
|
|
const linkPreview = null;
|
|
// Send an empty message, the underlying logic will know
|
|
// it should send a friend request
|
|
return c.sendMessage('', attachments, quote, linkPreview);
|
|
})
|
|
);
|
|
},
|
|
async handleAuthorisationForContact(envelope) {
|
|
window.log.error(
|
|
'Unexpected pairing request/authorisation received, ignoring.'
|
|
);
|
|
return this.removeFromCache(envelope);
|
|
},
|
|
async handlePairingAuthorisationMessage(envelope, content) {
|
|
const { pairingAuthorisation } = content;
|
|
const { secondaryDevicePubKey, grantSignature } = pairingAuthorisation;
|
|
const isGrant =
|
|
grantSignature &&
|
|
secondaryDevicePubKey === textsecure.storage.user.getNumber();
|
|
if (isGrant) {
|
|
return this.handleAuthorisationForSelf(
|
|
envelope,
|
|
pairingAuthorisation,
|
|
content
|
|
);
|
|
}
|
|
return this.handlePairingRequest(envelope, pairingAuthorisation);
|
|
},
|
|
|
|
async handleSecondaryDeviceFriendRequest(pubKey, deviceMapping) {
|
|
if (!deviceMapping) {
|
|
return false;
|
|
}
|
|
// Only handle secondary pubkeys
|
|
if (deviceMapping.isPrimary === '1' || !deviceMapping.authorisations) {
|
|
return false;
|
|
}
|
|
const { authorisations } = deviceMapping;
|
|
// Secondary devices should only have 1 authorisation from a primary device
|
|
if (authorisations.length !== 1) {
|
|
return false;
|
|
}
|
|
const authorisation = authorisations[0];
|
|
if (!authorisation) {
|
|
return false;
|
|
}
|
|
if (!authorisation.grantSignature) {
|
|
return false;
|
|
}
|
|
const isValid = await libloki.crypto.validateAuthorisation(authorisation);
|
|
if (!isValid) {
|
|
return false;
|
|
}
|
|
const correctSender = pubKey === authorisation.secondaryDevicePubKey;
|
|
if (!correctSender) {
|
|
return false;
|
|
}
|
|
const { primaryDevicePubKey } = authorisation;
|
|
// ensure the primary device is a friend
|
|
const c = window.ConversationController.get(primaryDevicePubKey);
|
|
if (!c || !c.isFriendWithAnyDevice()) {
|
|
return false;
|
|
}
|
|
await libloki.storage.savePairingAuthorisation(authorisation);
|
|
|
|
return true;
|
|
},
|
|
|
|
async updateProfile(conversation, profile, profileKey) {
|
|
// Retain old values unless changed:
|
|
const newProfile = conversation.get('profile') || {};
|
|
|
|
newProfile.displayName = profile.displayName;
|
|
|
|
// TODO: may need to allow users to reset their avatars to null
|
|
if (profile.avatar) {
|
|
const prevPointer = conversation.get('avatarPointer');
|
|
const needsUpdate =
|
|
!prevPointer || !_.isEqual(prevPointer, profile.avatar);
|
|
|
|
if (needsUpdate) {
|
|
conversation.set('avatarPointer', profile.avatar);
|
|
conversation.set('profileKey', profileKey);
|
|
|
|
const downloaded = await this.downloadAttachment({
|
|
url: profile.avatar,
|
|
isRaw: true,
|
|
});
|
|
|
|
// null => use jazzicon
|
|
let path = null;
|
|
if (profileKey) {
|
|
// Convert profileKey to ArrayBuffer, if needed
|
|
const encoding = typeof profileKey === 'string' ? 'base64' : null;
|
|
try {
|
|
const profileKeyArrayBuffer = dcodeIO.ByteBuffer.wrap(
|
|
profileKey,
|
|
encoding
|
|
).toArrayBuffer();
|
|
const decryptedData = await textsecure.crypto.decryptProfile(
|
|
downloaded.data,
|
|
profileKeyArrayBuffer
|
|
);
|
|
const upgraded = await Signal.Migrations.processNewAttachment({
|
|
...downloaded,
|
|
data: decryptedData,
|
|
});
|
|
({ path } = upgraded);
|
|
} catch (e) {
|
|
window.log.error(`Could not decrypt profile image: ${e}`);
|
|
}
|
|
}
|
|
newProfile.avatar = path;
|
|
}
|
|
} else {
|
|
newProfile.avatar = null;
|
|
}
|
|
|
|
await conversation.setLokiProfile(newProfile);
|
|
},
|
|
handleDataMessage(envelope, msg) {
|
|
window.log.info('data message from', this.getEnvelopeId(envelope));
|
|
let p = Promise.resolve();
|
|
// eslint-disable-next-line no-bitwise
|
|
if (msg.flags & textsecure.protobuf.DataMessage.Flags.END_SESSION) {
|
|
p = this.handleEndSession(envelope.source);
|
|
}
|
|
return p.then(() =>
|
|
this.processDecrypted(envelope, msg).then(async message => {
|
|
const groupId = message.group && message.group.id;
|
|
const isBlocked = this.isGroupBlocked(groupId);
|
|
const ourPubKey = textsecure.storage.user.getNumber();
|
|
const isMe = envelope.source === ourPubKey;
|
|
const conversation = window.ConversationController.get(envelope.source);
|
|
const isLeavingGroup = Boolean(
|
|
message.group &&
|
|
message.group.type === textsecure.protobuf.GroupContext.Type.QUIT
|
|
);
|
|
const friendRequest =
|
|
envelope.type === textsecure.protobuf.Envelope.Type.FRIEND_REQUEST;
|
|
const { UNPAIRING_REQUEST } = textsecure.protobuf.DataMessage.Flags;
|
|
// eslint-disable-next-line no-bitwise
|
|
const isUnpairingRequest = Boolean(message.flags & UNPAIRING_REQUEST);
|
|
|
|
if (!friendRequest && isUnpairingRequest) {
|
|
// TODO: move high-level pairing logic to libloki.multidevice.xx
|
|
|
|
const unpairingRequestIsLegit = async () => {
|
|
const isSecondary = textsecure.storage.get('isSecondaryDevice');
|
|
if (!isSecondary) {
|
|
return false;
|
|
}
|
|
const primaryPubKey = window.storage.get('primaryDevicePubKey');
|
|
// TODO: allow unpairing from any paired device?
|
|
if (envelope.source !== primaryPubKey) {
|
|
return false;
|
|
}
|
|
|
|
const primaryMapping = await lokiFileServerAPI.getUserDeviceMapping(
|
|
primaryPubKey
|
|
);
|
|
|
|
// If we don't have a mapping on the primary then we have been unlinked
|
|
if (!primaryMapping) {
|
|
return true;
|
|
}
|
|
|
|
// We expect the primary device to have updated its mapping
|
|
// before sending the unpairing request
|
|
const found = primaryMapping.authorisations.find(
|
|
authorisation => authorisation.secondaryDevicePubKey === ourPubKey
|
|
);
|
|
|
|
// our pubkey should NOT be in the primary device mapping
|
|
return !found;
|
|
};
|
|
|
|
const legit = await unpairingRequestIsLegit();
|
|
|
|
this.removeFromCache(envelope);
|
|
|
|
if (legit) {
|
|
// remove our device mapping annotations from file server
|
|
await lokiFileServerAPI.clearOurDeviceMappingAnnotations();
|
|
// Delete the account and restart
|
|
try {
|
|
await window.Signal.Logs.deleteAll();
|
|
await window.Signal.Data.removeAll();
|
|
await window.Signal.Data.close();
|
|
await window.Signal.Data.removeDB();
|
|
await window.Signal.Data.removeOtherData();
|
|
// TODO generate an empty db with a flag
|
|
// to display a message about the unpairing
|
|
// after the app restarts
|
|
} catch (error) {
|
|
window.log.error(
|
|
'Something went wrong deleting all data:',
|
|
error && error.stack ? error.stack : error
|
|
);
|
|
}
|
|
window.restart();
|
|
}
|
|
}
|
|
|
|
// Check if we need to update any profile names
|
|
if (!isMe && conversation) {
|
|
if (message.profile) {
|
|
await this.updateProfile(
|
|
conversation,
|
|
message.profile,
|
|
message.profileKey
|
|
);
|
|
}
|
|
}
|
|
|
|
// If we got a friend request message or
|
|
// if we're not friends with the current user that sent this private message
|
|
// Check to see if we need to auto accept their friend request
|
|
const isGroupMessage = !!groupId;
|
|
if (friendRequest || (!isGroupMessage && !conversation.isFriend())) {
|
|
if (isMe) {
|
|
window.log.info('refusing to add a friend request to ourselves');
|
|
throw new Error('Cannot add a friend request for ourselves!');
|
|
} else {
|
|
const senderPubKey = envelope.source;
|
|
// fetch the device mapping from the server
|
|
const deviceMapping = await lokiFileServerAPI.getUserDeviceMapping(
|
|
senderPubKey
|
|
);
|
|
// auto-accept friend request if the device is paired to one of our friend
|
|
const autoAccepted = await this.handleSecondaryDeviceFriendRequest(
|
|
senderPubKey,
|
|
deviceMapping
|
|
);
|
|
if (autoAccepted) {
|
|
// sending a message back = accepting friend request
|
|
// Directly setting friend request status to skip the pending state
|
|
await conversation.setFriendRequestStatus(
|
|
window.friends.friendRequestStatusEnum.friends
|
|
);
|
|
window.libloki.api.sendBackgroundMessage(envelope.source);
|
|
return this.removeFromCache(envelope);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (groupId && isBlocked && !(isMe && isLeavingGroup)) {
|
|
window.log.warn(
|
|
`Message ${this.getEnvelopeId(
|
|
envelope
|
|
)} ignored; destined for blocked group`
|
|
);
|
|
return this.removeFromCache(envelope);
|
|
}
|
|
|
|
const ev = new Event('message');
|
|
ev.confirm = this.removeFromCache.bind(this, envelope);
|
|
ev.data = {
|
|
friendRequest,
|
|
source: envelope.source,
|
|
sourceDevice: envelope.sourceDevice,
|
|
timestamp: envelope.timestamp.toNumber(),
|
|
receivedAt: envelope.receivedAt,
|
|
unidentifiedDeliveryReceived: envelope.unidentifiedDeliveryReceived,
|
|
message,
|
|
};
|
|
return this.dispatchAndWait(ev);
|
|
})
|
|
);
|
|
},
|
|
handleLegacyMessage(envelope) {
|
|
return this.decrypt(envelope, envelope.legacyMessage).then(plaintext => {
|
|
if (!plaintext) {
|
|
window.log.warn('handleLegacyMessage: plaintext was falsey');
|
|
return null;
|
|
}
|
|
return this.innerHandleLegacyMessage(envelope, plaintext);
|
|
});
|
|
},
|
|
innerHandleLegacyMessage(envelope, plaintext) {
|
|
const message = textsecure.protobuf.DataMessage.decode(plaintext);
|
|
return this.handleDataMessage(envelope, message);
|
|
},
|
|
handleContentMessage(envelope) {
|
|
return this.decrypt(envelope, envelope.content).then(plaintext => {
|
|
if (!plaintext) {
|
|
window.log.warn('handleContentMessage: plaintext was falsey');
|
|
return null;
|
|
} else if (
|
|
plaintext instanceof ArrayBuffer &&
|
|
plaintext.byteLength === 0
|
|
) {
|
|
return null;
|
|
}
|
|
return this.innerHandleContentMessage(envelope, plaintext);
|
|
});
|
|
},
|
|
async innerHandleContentMessage(envelope, plaintext) {
|
|
const content = textsecure.protobuf.Content.decode(plaintext);
|
|
|
|
if (content.preKeyBundleMessage) {
|
|
await this.savePreKeyBundleMessage(
|
|
envelope.source,
|
|
content.preKeyBundleMessage
|
|
);
|
|
}
|
|
|
|
if (content.lokiAddressMessage) {
|
|
return this.handleLokiAddressMessage(
|
|
envelope,
|
|
content.lokiAddressMessage
|
|
);
|
|
}
|
|
if (content.pairingAuthorisation) {
|
|
return this.handlePairingAuthorisationMessage(envelope, content);
|
|
}
|
|
if (content.syncMessage) {
|
|
return this.handleSyncMessage(envelope, content.syncMessage);
|
|
}
|
|
if (content.dataMessage) {
|
|
return this.handleDataMessage(envelope, content.dataMessage);
|
|
}
|
|
if (content.nullMessage) {
|
|
return this.handleNullMessage(envelope, content.nullMessage);
|
|
}
|
|
if (content.callMessage) {
|
|
return this.handleCallMessage(envelope, content.callMessage);
|
|
}
|
|
if (content.receiptMessage) {
|
|
return this.handleReceiptMessage(envelope, content.receiptMessage);
|
|
}
|
|
if (content.typingMessage) {
|
|
return this.handleTypingMessage(envelope, content.typingMessage);
|
|
}
|
|
|
|
return null;
|
|
},
|
|
handleCallMessage(envelope) {
|
|
window.log.info('call message from', this.getEnvelopeId(envelope));
|
|
this.removeFromCache(envelope);
|
|
},
|
|
handleReceiptMessage(envelope, receiptMessage) {
|
|
const results = [];
|
|
if (
|
|
receiptMessage.type === textsecure.protobuf.ReceiptMessage.Type.DELIVERY
|
|
) {
|
|
for (let i = 0; i < receiptMessage.timestamp.length; i += 1) {
|
|
const ev = new Event('delivery');
|
|
ev.confirm = this.removeFromCache.bind(this, envelope);
|
|
ev.deliveryReceipt = {
|
|
timestamp: receiptMessage.timestamp[i].toNumber(),
|
|
source: envelope.source,
|
|
sourceDevice: envelope.sourceDevice,
|
|
};
|
|
results.push(this.dispatchAndWait(ev));
|
|
}
|
|
} else if (
|
|
receiptMessage.type === textsecure.protobuf.ReceiptMessage.Type.READ
|
|
) {
|
|
for (let i = 0; i < receiptMessage.timestamp.length; i += 1) {
|
|
const ev = new Event('read');
|
|
ev.confirm = this.removeFromCache.bind(this, envelope);
|
|
ev.timestamp = envelope.timestamp.toNumber();
|
|
ev.read = {
|
|
timestamp: receiptMessage.timestamp[i].toNumber(),
|
|
reader: envelope.source,
|
|
};
|
|
results.push(this.dispatchAndWait(ev));
|
|
}
|
|
}
|
|
return Promise.all(results);
|
|
},
|
|
handleTypingMessage(envelope, typingMessage) {
|
|
const ev = new Event('typing');
|
|
|
|
this.removeFromCache(envelope);
|
|
|
|
if (envelope.timestamp && typingMessage.timestamp) {
|
|
const envelopeTimestamp = envelope.timestamp.toNumber();
|
|
const typingTimestamp = typingMessage.timestamp.toNumber();
|
|
|
|
if (typingTimestamp !== envelopeTimestamp) {
|
|
window.log.warn(
|
|
`Typing message envelope timestamp (${envelopeTimestamp}) did not match typing timestamp (${typingTimestamp})`
|
|
);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
ev.sender = envelope.source;
|
|
ev.senderDevice = envelope.sourceDevice;
|
|
ev.typing = {
|
|
typingMessage,
|
|
timestamp: typingMessage.timestamp
|
|
? typingMessage.timestamp.toNumber()
|
|
: Date.now(),
|
|
groupId: typingMessage.groupId
|
|
? typingMessage.groupId.toString('binary')
|
|
: null,
|
|
started:
|
|
typingMessage.action ===
|
|
textsecure.protobuf.TypingMessage.Action.STARTED,
|
|
stopped:
|
|
typingMessage.action ===
|
|
textsecure.protobuf.TypingMessage.Action.STOPPED,
|
|
};
|
|
|
|
return this.dispatchEvent(ev);
|
|
},
|
|
handleNullMessage(envelope) {
|
|
window.log.info('null message from', this.getEnvelopeId(envelope));
|
|
this.removeFromCache(envelope);
|
|
},
|
|
async handleSyncMessage(envelope, syncMessage) {
|
|
// We should only accept sync messages from our devices
|
|
const ourNumber = textsecure.storage.user.getNumber();
|
|
const ourPrimaryNumber = window.storage.get('primaryDevicePubKey');
|
|
const ourOtherDevices = await libloki.storage.getAllDevicePubKeysForPrimaryPubKey(
|
|
window.storage.get('primaryDevicePubKey')
|
|
);
|
|
const ourDevices = new Set([
|
|
ourNumber,
|
|
ourPrimaryNumber,
|
|
...ourOtherDevices,
|
|
]);
|
|
const validSyncSender = ourDevices.has(envelope.source);
|
|
if (!validSyncSender) {
|
|
throw new Error(
|
|
"Received sync message from a device we aren't paired with"
|
|
);
|
|
}
|
|
|
|
if (syncMessage.sent) {
|
|
const sentMessage = syncMessage.sent;
|
|
const to = sentMessage.message.group
|
|
? `group(${sentMessage.message.group.id.toBinary()})`
|
|
: sentMessage.destination;
|
|
|
|
window.log.info(
|
|
'sent message to',
|
|
to,
|
|
sentMessage.timestamp.toNumber(),
|
|
'from',
|
|
this.getEnvelopeId(envelope)
|
|
);
|
|
return this.handleSentMessage(envelope, sentMessage, sentMessage.message);
|
|
} else if (syncMessage.contacts) {
|
|
return this.handleContacts(envelope, syncMessage.contacts);
|
|
} else if (syncMessage.groups) {
|
|
return this.handleGroups(envelope, syncMessage.groups);
|
|
} else if (syncMessage.openGroups) {
|
|
return this.handleOpenGroups(envelope, syncMessage.openGroups);
|
|
} else if (syncMessage.blocked) {
|
|
return this.handleBlocked(envelope, syncMessage.blocked);
|
|
} else if (syncMessage.request) {
|
|
window.log.info('Got SyncMessage Request');
|
|
return this.removeFromCache(envelope);
|
|
} else if (syncMessage.read && syncMessage.read.length) {
|
|
window.log.info('read messages from', this.getEnvelopeId(envelope));
|
|
return this.handleRead(envelope, syncMessage.read);
|
|
} else if (syncMessage.verified) {
|
|
return this.handleVerified(envelope, syncMessage.verified);
|
|
} else if (syncMessage.configuration) {
|
|
return this.handleConfiguration(envelope, syncMessage.configuration);
|
|
}
|
|
throw new Error('Got empty SyncMessage');
|
|
},
|
|
handleConfiguration(envelope, configuration) {
|
|
window.log.info('got configuration sync message');
|
|
const ev = new Event('configuration');
|
|
ev.confirm = this.removeFromCache.bind(this, envelope);
|
|
ev.configuration = configuration;
|
|
return this.dispatchAndWait(ev);
|
|
},
|
|
handleVerified(envelope, verified) {
|
|
const ev = new Event('verified');
|
|
ev.confirm = this.removeFromCache.bind(this, envelope);
|
|
ev.verified = {
|
|
state: verified.state,
|
|
destination: verified.destination,
|
|
identityKey: verified.identityKey.toArrayBuffer(),
|
|
};
|
|
return this.dispatchAndWait(ev);
|
|
},
|
|
handleRead(envelope, read) {
|
|
const results = [];
|
|
for (let i = 0; i < read.length; i += 1) {
|
|
const ev = new Event('readSync');
|
|
ev.confirm = this.removeFromCache.bind(this, envelope);
|
|
ev.timestamp = envelope.timestamp.toNumber();
|
|
ev.read = {
|
|
timestamp: read[i].timestamp.toNumber(),
|
|
sender: read[i].sender,
|
|
};
|
|
results.push(this.dispatchAndWait(ev));
|
|
}
|
|
return Promise.all(results);
|
|
},
|
|
handleContacts(envelope, contacts) {
|
|
window.log.info('contact sync');
|
|
// const { blob } = contacts;
|
|
|
|
// Note: we do not return here because we don't want to block the next message on
|
|
// this attachment download and a lot of processing of that attachment.
|
|
this.handleAttachment(contacts).then(attachmentPointer => {
|
|
const results = [];
|
|
const contactBuffer = new ContactBuffer(attachmentPointer.data);
|
|
let contactDetails = contactBuffer.next();
|
|
while (contactDetails !== undefined) {
|
|
const ev = new Event('contact');
|
|
ev.contactDetails = contactDetails;
|
|
results.push(this.dispatchAndWait(ev));
|
|
|
|
contactDetails = contactBuffer.next();
|
|
}
|
|
|
|
const ev = new Event('contactsync');
|
|
results.push(this.dispatchAndWait(ev));
|
|
|
|
return Promise.all(results).then(() => {
|
|
window.log.info('handleContacts: finished');
|
|
return this.removeFromCache(envelope);
|
|
});
|
|
});
|
|
},
|
|
handleGroups(envelope, groups) {
|
|
window.log.info('group sync');
|
|
|
|
// Note: we do not return here because we don't want to block the next message on
|
|
// this attachment download and a lot of processing of that attachment.
|
|
this.handleAttachment(groups).then(attachmentPointer => {
|
|
const groupBuffer = new GroupBuffer(attachmentPointer.data);
|
|
let groupDetails = groupBuffer.next();
|
|
const promises = [];
|
|
while (groupDetails !== undefined) {
|
|
groupDetails.id = groupDetails.id.toBinary();
|
|
const ev = new Event('group');
|
|
ev.confirm = this.removeFromCache.bind(this, envelope);
|
|
ev.groupDetails = groupDetails;
|
|
const promise = this.dispatchAndWait(ev).catch(e => {
|
|
window.log.error('error processing group', e);
|
|
});
|
|
groupDetails = groupBuffer.next();
|
|
promises.push(promise);
|
|
}
|
|
|
|
Promise.all(promises).then(() => {
|
|
const ev = new Event('groupsync');
|
|
ev.confirm = this.removeFromCache.bind(this, envelope);
|
|
return this.dispatchAndWait(ev);
|
|
});
|
|
});
|
|
},
|
|
handleOpenGroups(envelope, openGroups) {
|
|
openGroups.forEach(({ url, channelId }) => {
|
|
window.attemptConnection(url, channelId);
|
|
});
|
|
return this.removeFromCache(envelope);
|
|
},
|
|
handleBlocked(envelope, blocked) {
|
|
window.log.info('Setting these numbers as blocked:', blocked.numbers);
|
|
textsecure.storage.put('blocked', blocked.numbers);
|
|
|
|
const groupIds = _.map(blocked.groupIds, groupId => groupId.toBinary());
|
|
window.log.info(
|
|
'Setting these groups as blocked:',
|
|
groupIds.map(groupId => `group(${groupId})`)
|
|
);
|
|
textsecure.storage.put('blocked-groups', groupIds);
|
|
|
|
return this.removeFromCache(envelope);
|
|
},
|
|
async savePreKeyBundleMessage(pubKey, preKeyBundleMessage) {
|
|
const [identityKey, preKey, signedKey, signature] = [
|
|
preKeyBundleMessage.identityKey,
|
|
preKeyBundleMessage.preKey,
|
|
preKeyBundleMessage.signedKey,
|
|
preKeyBundleMessage.signature,
|
|
].map(k => dcodeIO.ByteBuffer.wrap(k).toArrayBuffer());
|
|
|
|
const { preKeyId, signedKeyId } = preKeyBundleMessage;
|
|
|
|
if (pubKey !== StringView.arrayBufferToHex(identityKey)) {
|
|
throw new Error(
|
|
'Error in savePreKeyBundleMessage: envelope pubkey does not match pubkey in prekey bundle'
|
|
);
|
|
}
|
|
|
|
await libloki.storage.saveContactPreKeyBundle({
|
|
pubKey,
|
|
preKeyId,
|
|
signedKeyId,
|
|
preKey,
|
|
signedKey,
|
|
signature,
|
|
});
|
|
},
|
|
isBlocked(number) {
|
|
return textsecure.storage.get('blocked', []).indexOf(number) >= 0;
|
|
},
|
|
isGroupBlocked(groupId) {
|
|
return textsecure.storage.get('blocked-groups', []).indexOf(groupId) >= 0;
|
|
},
|
|
cleanAttachment(attachment) {
|
|
return {
|
|
..._.omit(attachment, 'thumbnail'),
|
|
id: attachment.id.toString(),
|
|
key: attachment.key ? attachment.key.toString('base64') : null,
|
|
digest: attachment.digest ? attachment.digest.toString('base64') : null,
|
|
};
|
|
},
|
|
async downloadAttachment(attachment) {
|
|
// The attachment id is actually just the absolute url of the attachment
|
|
let data = await this.server.getAttachment(attachment.url);
|
|
if (!attachment.isRaw) {
|
|
const { key, digest, size } = attachment;
|
|
|
|
data = await textsecure.crypto.decryptAttachment(
|
|
data,
|
|
window.Signal.Crypto.base64ToArrayBuffer(key),
|
|
window.Signal.Crypto.base64ToArrayBuffer(digest)
|
|
);
|
|
|
|
if (!size || size !== data.byteLength) {
|
|
throw new Error(
|
|
`downloadAttachment: Size ${size} did not match downloaded attachment size ${
|
|
data.byteLength
|
|
}`
|
|
);
|
|
}
|
|
}
|
|
|
|
return {
|
|
..._.omit(attachment, 'digest', 'key'),
|
|
data,
|
|
};
|
|
},
|
|
handleAttachment(attachment) {
|
|
// window.log.info('Not handling attachments.');
|
|
return Promise.resolve({
|
|
...attachment,
|
|
data: dcodeIO.ByteBuffer.wrap(attachment.data).toArrayBuffer(), // ByteBuffer to ArrayBuffer
|
|
});
|
|
|
|
const cleaned = this.cleanAttachment(attachment);
|
|
return this.downloadAttachment(cleaned);
|
|
},
|
|
async handleEndSession(number) {
|
|
window.log.info('got end session');
|
|
const deviceIds = await textsecure.storage.protocol.getDeviceIds(number);
|
|
const identityKey = StringView.hexToArrayBuffer(number);
|
|
let conversation;
|
|
try {
|
|
conversation = window.ConversationController.get(number);
|
|
} catch (e) {
|
|
window.log.error('Error getting conversation: ', number);
|
|
}
|
|
|
|
await Promise.all(
|
|
deviceIds.map(async deviceId => {
|
|
const address = new libsignal.SignalProtocolAddress(number, deviceId);
|
|
// Instead of deleting the sessions now,
|
|
// we process the new prekeys and initiate a new session.
|
|
// The old sessions will get deleted once the correspondant
|
|
// has switch the the new session.
|
|
const [preKey, signedPreKey] = await Promise.all([
|
|
textsecure.storage.protocol.loadContactPreKey(number),
|
|
textsecure.storage.protocol.loadContactSignedPreKey(number),
|
|
]);
|
|
if (preKey === undefined || signedPreKey === undefined) {
|
|
return;
|
|
}
|
|
const device = {
|
|
identityKey,
|
|
deviceId,
|
|
preKey,
|
|
signedPreKey,
|
|
registrationId: 0,
|
|
};
|
|
const builder = new libsignal.SessionBuilder(
|
|
textsecure.storage.protocol,
|
|
address
|
|
);
|
|
await builder.processPreKey(device);
|
|
})
|
|
);
|
|
await conversation.onSessionResetReceived();
|
|
},
|
|
processDecrypted(envelope, decrypted) {
|
|
/* eslint-disable no-bitwise, no-param-reassign */
|
|
const FLAGS = textsecure.protobuf.DataMessage.Flags;
|
|
|
|
// Now that its decrypted, validate the message and clean it up for consumer
|
|
// processing
|
|
// Note that messages may (generally) only perform one action and we ignore remaining
|
|
// fields after the first action.
|
|
|
|
if (decrypted.flags == null) {
|
|
decrypted.flags = 0;
|
|
}
|
|
if (decrypted.expireTimer == null) {
|
|
decrypted.expireTimer = 0;
|
|
}
|
|
|
|
if (decrypted.flags & FLAGS.END_SESSION) {
|
|
decrypted.body = null;
|
|
decrypted.attachments = [];
|
|
decrypted.group = null;
|
|
return Promise.resolve(decrypted);
|
|
} else if (decrypted.flags & FLAGS.EXPIRATION_TIMER_UPDATE) {
|
|
decrypted.body = null;
|
|
decrypted.attachments = [];
|
|
} else if (decrypted.flags & FLAGS.PROFILE_KEY_UPDATE) {
|
|
decrypted.body = null;
|
|
decrypted.attachments = [];
|
|
} else if (decrypted.flags & FLAGS.SESSION_REQUEST) {
|
|
// do nothing
|
|
} else if (decrypted.flags & FLAGS.SESSION_RESTORE) {
|
|
// do nothing
|
|
} else if (decrypted.flags & FLAGS.UNPAIRING_REQUEST) {
|
|
// do nothing
|
|
} else if (decrypted.flags !== 0) {
|
|
throw new Error('Unknown flags in message');
|
|
}
|
|
|
|
const promises = [];
|
|
|
|
if (decrypted.group !== null) {
|
|
decrypted.group.id = decrypted.group.id.toBinary();
|
|
|
|
switch (decrypted.group.type) {
|
|
case textsecure.protobuf.GroupContext.Type.UPDATE:
|
|
decrypted.body = null;
|
|
decrypted.attachments = [];
|
|
break;
|
|
case textsecure.protobuf.GroupContext.Type.QUIT:
|
|
decrypted.body = null;
|
|
decrypted.attachments = [];
|
|
break;
|
|
case textsecure.protobuf.GroupContext.Type.DELIVER:
|
|
decrypted.group.name = null;
|
|
decrypted.group.members = [];
|
|
decrypted.group.avatar = null;
|
|
break;
|
|
case textsecure.protobuf.GroupContext.Type.REQUEST_INFO:
|
|
decrypted.body = null;
|
|
decrypted.attachments = [];
|
|
break;
|
|
default:
|
|
this.removeFromCache(envelope);
|
|
throw new Error('Unknown group message type');
|
|
}
|
|
}
|
|
|
|
const attachmentCount = decrypted.attachments.length;
|
|
const ATTACHMENT_MAX = 32;
|
|
if (attachmentCount > ATTACHMENT_MAX) {
|
|
throw new Error(
|
|
`Too many attachments: ${attachmentCount} included in one message, max is ${ATTACHMENT_MAX}`
|
|
);
|
|
}
|
|
|
|
// Here we go from binary to string/base64 in all AttachmentPointer digest/key fields
|
|
|
|
if (
|
|
decrypted.group &&
|
|
decrypted.group.type === textsecure.protobuf.GroupContext.Type.UPDATE
|
|
) {
|
|
if (decrypted.group.avatar !== null) {
|
|
decrypted.group.avatar = this.cleanAttachment(decrypted.group.avatar);
|
|
}
|
|
}
|
|
|
|
decrypted.attachments = (decrypted.attachments || []).map(
|
|
this.cleanAttachment.bind(this)
|
|
);
|
|
decrypted.preview = (decrypted.preview || []).map(item => {
|
|
const { image } = item;
|
|
|
|
if (!image) {
|
|
return item;
|
|
}
|
|
|
|
return {
|
|
...item,
|
|
image: this.cleanAttachment(image),
|
|
};
|
|
});
|
|
decrypted.contact = (decrypted.contact || []).map(item => {
|
|
const { avatar } = item;
|
|
|
|
if (!avatar || !avatar.avatar) {
|
|
return item;
|
|
}
|
|
|
|
return {
|
|
...item,
|
|
avatar: {
|
|
...item.avatar,
|
|
avatar: this.cleanAttachment(item.avatar.avatar),
|
|
},
|
|
};
|
|
});
|
|
|
|
if (decrypted.quote && decrypted.quote.id) {
|
|
decrypted.quote.id = decrypted.quote.id.toNumber();
|
|
}
|
|
|
|
if (decrypted.quote) {
|
|
decrypted.quote.attachments = (decrypted.quote.attachments || []).map(
|
|
item => {
|
|
const { thumbnail } = item;
|
|
|
|
if (!thumbnail) {
|
|
return item;
|
|
}
|
|
|
|
return {
|
|
...item,
|
|
thumbnail: this.cleanAttachment(item.thumbnail),
|
|
};
|
|
}
|
|
);
|
|
}
|
|
|
|
return Promise.all(promises).then(() => decrypted);
|
|
/* eslint-enable no-bitwise, no-param-reassign */
|
|
},
|
|
});
|
|
|
|
window.textsecure = window.textsecure || {};
|
|
|
|
textsecure.MessageReceiver = function MessageReceiverWrapper(
|
|
username,
|
|
password,
|
|
signalingKey,
|
|
options
|
|
) {
|
|
const messageReceiver = new MessageReceiver(
|
|
username,
|
|
password,
|
|
signalingKey,
|
|
options
|
|
);
|
|
this.addEventListener = messageReceiver.addEventListener.bind(
|
|
messageReceiver
|
|
);
|
|
this.removeEventListener = messageReceiver.removeEventListener.bind(
|
|
messageReceiver
|
|
);
|
|
this.getStatus = messageReceiver.getStatus.bind(messageReceiver);
|
|
this.handleEndSession = messageReceiver.handleEndSession.bind(
|
|
messageReceiver
|
|
);
|
|
this.close = messageReceiver.close.bind(messageReceiver);
|
|
this.savePreKeyBundleMessage = messageReceiver.savePreKeyBundleMessage.bind(
|
|
messageReceiver
|
|
);
|
|
|
|
this.downloadAttachment = messageReceiver.downloadAttachment.bind(
|
|
messageReceiver
|
|
);
|
|
this.stopProcessing = messageReceiver.stopProcessing.bind(messageReceiver);
|
|
|
|
messageReceiver.connect();
|
|
};
|
|
|
|
textsecure.MessageReceiver.prototype = {
|
|
constructor: textsecure.MessageReceiver,
|
|
};
|
|
|
|
textsecure.MessageReceiver.stringToArrayBuffer =
|
|
MessageReceiver.stringToArrayBuffer;
|
|
textsecure.MessageReceiver.arrayBufferToString =
|
|
MessageReceiver.arrayBufferToString;
|
|
textsecure.MessageReceiver.stringToArrayBufferBase64 =
|
|
MessageReceiver.stringToArrayBufferBase64;
|
|
textsecure.MessageReceiver.arrayBufferToStringBase64 =
|
|
MessageReceiver.arrayBufferToStringBase64;
|