mirror of
https://github.com/oxen-io/session-desktop.git
synced 2023-12-14 02:12:57 +01:00
a2c7ac0df9
Add a pendingMessages object to MessageSender. This object holds one promise per recipient number. We init this promise with Promise.resolve(), and chain on promises for message sending, replacing the previous promise with the newly chained promise each time. If the current promise resolves and finds that it is still the last promise in the chain, it removes itself.
470 lines
21 KiB
JavaScript
470 lines
21 KiB
JavaScript
/*
|
|
* vim: ts=4:sw=4:expandtab
|
|
*/
|
|
function MessageSender(url, username, password) {
|
|
this.server = new TextSecureServer(url, username, password);
|
|
this.pendingMessages = {};
|
|
}
|
|
|
|
MessageSender.prototype = {
|
|
constructor: MessageSender,
|
|
// message == DataMessage or ContentMessage proto
|
|
encryptToDevices: function(timestamp, number, deviceObjectList, message) {
|
|
var legacy = (message instanceof textsecure.protobuf.DataMessage);
|
|
var relay = deviceObjectList[0].relay;
|
|
for (var i=1; i < deviceObjectList.length; ++i) {
|
|
if (deviceObjectList[i].relay !== relay) {
|
|
throw new Error("Mismatched relays for number " + number);
|
|
}
|
|
}
|
|
return Promise.all(deviceObjectList.map(function(device) {
|
|
return textsecure.protocol_wrapper.encryptMessageFor(device, message).then(function(encryptedMsg) {
|
|
return textsecure.protocol_wrapper.getRegistrationId(device.encodedNumber).then(function(registrationId) {
|
|
return textsecure.storage.devices.removeTempKeysFromDevice(device.encodedNumber).then(function() {
|
|
var json = {
|
|
type: encryptedMsg.type,
|
|
destinationDeviceId: textsecure.utils.unencodeNumber(device.encodedNumber)[1],
|
|
destinationRegistrationId: registrationId,
|
|
timestamp: timestamp
|
|
};
|
|
|
|
if (device.relay !== undefined) {
|
|
json.relay = device.relay;
|
|
}
|
|
|
|
var content = btoa(encryptedMsg.body);
|
|
if (legacy) {
|
|
json.body = content;
|
|
} else {
|
|
json.content = content;
|
|
}
|
|
|
|
return json;
|
|
});
|
|
});
|
|
});
|
|
}));
|
|
},
|
|
|
|
transmitMessage: function(number, jsonData) {
|
|
return this.server.sendMessages(number, jsonData).catch(function(e) {
|
|
if (e.name === 'HTTPError' && (e.code !== 409 && e.code !== 410)) {
|
|
// 409 and 410 should bubble and be handled by doSendMessage
|
|
// all other network errors can be retried later.
|
|
throw new textsecure.SendMessageNetworkError(number, jsonData, e);
|
|
}
|
|
throw e;
|
|
});
|
|
},
|
|
|
|
makeAttachmentPointer: function(attachment) {
|
|
if (typeof attachment !== 'object' || attachment == null) {
|
|
return Promise.resolve(undefined);
|
|
}
|
|
var proto = new textsecure.protobuf.AttachmentPointer();
|
|
proto.key = textsecure.crypto.getRandomBytes(64);
|
|
|
|
var iv = textsecure.crypto.getRandomBytes(16);
|
|
return textsecure.crypto.encryptAttachment(attachment.data, proto.key, iv).then(function(encryptedBin) {
|
|
return this.server.putAttachment(encryptedBin).then(function(id) {
|
|
proto.id = id;
|
|
proto.contentType = attachment.contentType;
|
|
return proto;
|
|
});
|
|
}.bind(this));
|
|
},
|
|
|
|
tryMessageAgain: function(number, encodedMessage, timestamp) {
|
|
var proto = textsecure.protobuf.DataMessage.decode(encodedMessage);
|
|
return this.sendIndividualProto(number, proto, timestamp);
|
|
},
|
|
|
|
sendMessageProto: function(timestamp, numbers, message, callback) {
|
|
var numbersCompleted = 0;
|
|
var errors = [];
|
|
var successfulNumbers = [];
|
|
|
|
var numberCompleted = function() {
|
|
numbersCompleted++;
|
|
if (numbersCompleted >= numbers.length)
|
|
callback({success: successfulNumbers, failure: errors});
|
|
};
|
|
|
|
var registerError = function(number, reason, error) {
|
|
if (!error || error.name === 'HTTPError') {
|
|
error = new textsecure.OutgoingMessageError(number, message.toArrayBuffer(), timestamp, error);
|
|
}
|
|
|
|
error.number = number;
|
|
error.reason = reason;
|
|
errors[errors.length] = error;
|
|
numberCompleted();
|
|
};
|
|
|
|
var reloadDevicesAndSend = function(number, recurse) {
|
|
return function() {
|
|
return textsecure.storage.devices.getDeviceObjectsForNumber(number).then(function(devicesForNumber) {
|
|
if (devicesForNumber.length == 0)
|
|
return registerError(number, "Got empty device list when loading device keys", null);
|
|
return doSendMessage(number, devicesForNumber, recurse);
|
|
});
|
|
}
|
|
};
|
|
|
|
var getKeysForNumber = function(number, updateDevices) {
|
|
var handleResult = function(response) {
|
|
return Promise.all(response.devices.map(function(device) {
|
|
if (updateDevices === undefined || updateDevices.indexOf(device.deviceId) > -1)
|
|
return textsecure.storage.devices.saveKeysToDeviceObject({
|
|
encodedNumber: number + "." + device.deviceId,
|
|
identityKey: response.identityKey,
|
|
preKey: device.preKey.publicKey,
|
|
preKeyId: device.preKey.keyId,
|
|
signedKey: device.signedPreKey.publicKey,
|
|
signedKeyId: device.signedPreKey.keyId,
|
|
signedKeySignature: device.signedPreKey.signature,
|
|
registrationId: device.registrationId
|
|
}).catch(function(error) {
|
|
if (error.message === "Identity key changed") {
|
|
error = new textsecure.OutgoingIdentityKeyError(number, message.toArrayBuffer(), timestamp, error.identityKey);
|
|
registerError(number, "Identity key changed", error);
|
|
}
|
|
throw error;
|
|
});
|
|
}));
|
|
};
|
|
|
|
if (updateDevices === undefined) {
|
|
return this.server.getKeysForNumber(number).then(handleResult);
|
|
} else {
|
|
var promises = updateDevices.map(function(device) {
|
|
return this.server.getKeysForNumber(number, device).then(handleResult);
|
|
}.bind(this));
|
|
|
|
return Promise.all(promises);
|
|
}
|
|
}.bind(this);
|
|
|
|
var doSendMessage = function(number, devicesForNumber, recurse) {
|
|
return this.encryptToDevices(timestamp, number, devicesForNumber, message).then(function(jsonData) {
|
|
return this.transmitMessage(number, jsonData).then(function() {
|
|
successfulNumbers[successfulNumbers.length] = number;
|
|
numberCompleted();
|
|
}.bind(this));
|
|
}.bind(this)).catch(function(error) {
|
|
if (error instanceof Error && error.name == "HTTPError" && (error.code == 410 || error.code == 409)) {
|
|
if (!recurse)
|
|
return registerError(number, "Hit retry limit attempting to reload device list", error);
|
|
|
|
var p;
|
|
if (error.code == 409) {
|
|
p = textsecure.storage.devices.removeDeviceIdsForNumber(number, error.response.extraDevices);
|
|
} else {
|
|
p = Promise.all(error.response.staleDevices.map(function(deviceId) {
|
|
return textsecure.protocol_wrapper.closeOpenSessionForDevice(number + '.' + deviceId);
|
|
}));
|
|
}
|
|
|
|
return p.then(function() {
|
|
var resetDevices = ((error.code == 410) ? error.response.staleDevices : error.response.missingDevices);
|
|
return getKeysForNumber(number, resetDevices)
|
|
.then(reloadDevicesAndSend(number, (error.code == 409)))
|
|
.catch(function(error) {
|
|
registerError(number, "Failed to reload device keys", error);
|
|
});
|
|
});
|
|
} else {
|
|
registerError(number, "Failed to create or send message", error);
|
|
}
|
|
});
|
|
}.bind(this);
|
|
|
|
function sendToNumber(number) {
|
|
return textsecure.storage.devices.getDeviceObjectsForNumber(number).then(function(devicesForNumber) {
|
|
return Promise.all(devicesForNumber.map(function(device) {
|
|
return textsecure.protocol_wrapper.hasOpenSession(device.encodedNumber).then(function(result) {
|
|
if (!result)
|
|
return getKeysForNumber(number, [parseInt(textsecure.utils.unencodeNumber(device.encodedNumber)[1])]);
|
|
});
|
|
})).then(function() {
|
|
return textsecure.storage.devices.getDeviceObjectsForNumber(number).then(function(devicesForNumber) {
|
|
if (devicesForNumber.length == 0) {
|
|
return getKeysForNumber(number, [1])
|
|
.then(reloadDevicesAndSend(number, true))
|
|
.catch(function(error) {
|
|
registerError(number, "Failed to retreive new device keys for number " + number, error);
|
|
});
|
|
} else
|
|
return doSendMessage(number, devicesForNumber, true);
|
|
});
|
|
});
|
|
});
|
|
}
|
|
|
|
numbers.forEach(function(number) {
|
|
var sendPrevious = this.pendingMessages[number] || Promise.resolve();
|
|
var sendCurrent = this.pendingMessages[number] = sendPrevious.then(function() {
|
|
return sendToNumber(number);
|
|
}).catch(function() {
|
|
return sendToNumber(number);
|
|
});
|
|
sendCurrent.then(function() {
|
|
if (this.pendingMessages[number] === sendCurrent) {
|
|
delete this.pendingMessages[number];
|
|
}
|
|
}.bind(this));
|
|
}.bind(this));
|
|
},
|
|
|
|
sendIndividualProto: function(number, proto, timestamp) {
|
|
return new Promise(function(resolve, reject) {
|
|
this.sendMessageProto(timestamp, [number], proto, function(res) {
|
|
if (res.failure.length > 0)
|
|
reject(res.failure);
|
|
else
|
|
resolve();
|
|
});
|
|
}.bind(this));
|
|
},
|
|
|
|
sendSyncMessage: function(message, timestamp, destination) {
|
|
var myNumber = textsecure.storage.user.getNumber();
|
|
var myDevice = textsecure.storage.user.getDeviceId();
|
|
if (myDevice != 1) {
|
|
var sentMessage = new textsecure.protobuf.SyncMessage.Sent();
|
|
sentMessage.timestamp = timestamp;
|
|
sentMessage.message = message;
|
|
if (destination) {
|
|
sentMessage.destination = destination;
|
|
}
|
|
var syncMessage = new textsecure.protobuf.SyncMessage();
|
|
syncMessage.sent = sentMessage;
|
|
var contentMessage = new textsecure.protobuf.Content();
|
|
contentMessage.syncMessage = syncMessage;
|
|
|
|
return this.sendIndividualProto(myNumber, contentMessage, Date.now());
|
|
}
|
|
},
|
|
|
|
sendRequestGroupSyncMessage: function() {
|
|
var myNumber = textsecure.storage.user.getNumber();
|
|
var myDevice = textsecure.storage.user.getDeviceId();
|
|
if (myDevice != 1) {
|
|
var request = new textsecure.protobuf.SyncMessage.Request();
|
|
request.type = textsecure.protobuf.SyncMessage.Request.Type.GROUPS;
|
|
var syncMessage = new textsecure.protobuf.SyncMessage();
|
|
syncMessage.request = request;
|
|
var contentMessage = new textsecure.protobuf.Content();
|
|
contentMessage.syncMessage = syncMessage;
|
|
|
|
return this.sendIndividualProto(myNumber, contentMessage, Date.now());
|
|
}
|
|
},
|
|
|
|
sendRequestContactSyncMessage: function() {
|
|
var myNumber = textsecure.storage.user.getNumber();
|
|
var myDevice = textsecure.storage.user.getDeviceId();
|
|
if (myDevice != 1) {
|
|
var request = new textsecure.protobuf.SyncMessage.Request();
|
|
request.type = textsecure.protobuf.SyncMessage.Request.Type.CONTACTS;
|
|
var syncMessage = new textsecure.protobuf.SyncMessage();
|
|
syncMessage.request = request;
|
|
var contentMessage = new textsecure.protobuf.Content();
|
|
contentMessage.syncMessage = syncMessage;
|
|
|
|
return this.sendIndividualProto(myNumber, contentMessage, Date.now());
|
|
}
|
|
},
|
|
|
|
sendGroupProto: function(numbers, proto, timestamp) {
|
|
timestamp = timestamp || Date.now();
|
|
var me = textsecure.storage.user.getNumber();
|
|
numbers = numbers.filter(function(number) { return number != me; });
|
|
|
|
return new Promise(function(resolve, reject) {
|
|
this.sendMessageProto(timestamp, numbers, proto, function(res) {
|
|
if (res.failure.length > 0)
|
|
reject(res.failure);
|
|
else
|
|
resolve();
|
|
});
|
|
}.bind(this)).then(this.sendSyncMessage.bind(this, proto, timestamp));
|
|
},
|
|
|
|
sendMessageToNumber: function(number, messageText, attachments, timestamp) {
|
|
var proto = new textsecure.protobuf.DataMessage();
|
|
proto.body = messageText;
|
|
|
|
return Promise.all(attachments.map(this.makeAttachmentPointer.bind(this))).then(function(attachmentsArray) {
|
|
proto.attachments = attachmentsArray;
|
|
return this.sendIndividualProto(number, proto, timestamp).then(function() {
|
|
return this.sendSyncMessage(proto, timestamp, number);
|
|
}.bind(this));
|
|
}.bind(this));
|
|
},
|
|
|
|
closeSession: function(number) {
|
|
var proto = new textsecure.protobuf.DataMessage();
|
|
proto.body = "TERMINATE";
|
|
proto.flags = textsecure.protobuf.DataMessage.Flags.END_SESSION;
|
|
return this.sendIndividualProto(number, proto, Date.now()).then(function(res) {
|
|
return textsecure.storage.devices.getDeviceObjectsForNumber(number).then(function(devices) {
|
|
return Promise.all(devices.map(function(device) {
|
|
return textsecure.protocol_wrapper.closeOpenSessionForDevice(device.encodedNumber);
|
|
})).then(function() {
|
|
return res;
|
|
});
|
|
});
|
|
});
|
|
},
|
|
|
|
sendMessageToGroup: function(groupId, messageText, attachments, timestamp) {
|
|
var proto = new textsecure.protobuf.DataMessage();
|
|
proto.body = messageText;
|
|
proto.group = new textsecure.protobuf.GroupContext();
|
|
proto.group.id = toArrayBuffer(groupId);
|
|
proto.group.type = textsecure.protobuf.GroupContext.Type.DELIVER;
|
|
|
|
return textsecure.storage.groups.getNumbers(groupId).then(function(numbers) {
|
|
if (numbers === undefined)
|
|
return Promise.reject(new Error("Unknown Group"));
|
|
|
|
return Promise.all(attachments.map(this.makeAttachmentPointer.bind(this))).then(function(attachmentsArray) {
|
|
proto.attachments = attachmentsArray;
|
|
return this.sendGroupProto(numbers, proto, timestamp);
|
|
}.bind(this));
|
|
}.bind(this));
|
|
},
|
|
|
|
createGroup: function(numbers, name, avatar) {
|
|
var proto = new textsecure.protobuf.DataMessage();
|
|
proto.group = new textsecure.protobuf.GroupContext();
|
|
|
|
return textsecure.storage.groups.createNewGroup(numbers).then(function(group) {
|
|
proto.group.id = toArrayBuffer(group.id);
|
|
var numbers = group.numbers;
|
|
|
|
proto.group.type = textsecure.protobuf.GroupContext.Type.UPDATE;
|
|
proto.group.members = numbers;
|
|
proto.group.name = name;
|
|
|
|
return this.makeAttachmentPointer(avatar).then(function(attachment) {
|
|
proto.group.avatar = attachment;
|
|
return this.sendGroupProto(numbers, proto).then(function() {
|
|
return proto.group.id;
|
|
});
|
|
}.bind(this));
|
|
}.bind(this));
|
|
},
|
|
|
|
updateGroup: function(groupId, name, avatar, numbers) {
|
|
var proto = new textsecure.protobuf.DataMessage();
|
|
proto.group = new textsecure.protobuf.GroupContext();
|
|
|
|
proto.group.id = toArrayBuffer(groupId);
|
|
proto.group.type = textsecure.protobuf.GroupContext.Type.UPDATE;
|
|
proto.group.name = name;
|
|
|
|
return textsecure.storage.groups.addNumbers(groupId, numbers).then(function(numbers) {
|
|
if (numbers === undefined) {
|
|
return Promise.reject(new Error("Unknown Group"));
|
|
}
|
|
proto.group.members = numbers;
|
|
|
|
return this.makeAttachmentPointer(avatar).then(function(attachment) {
|
|
proto.group.avatar = attachment;
|
|
return this.sendGroupProto(numbers, proto).then(function() {
|
|
return proto.group.id;
|
|
});
|
|
}.bind(this));
|
|
}.bind(this));
|
|
},
|
|
|
|
addNumberToGroup: function(groupId, number) {
|
|
var proto = new textsecure.protobuf.DataMessage();
|
|
proto.group = new textsecure.protobuf.GroupContext();
|
|
proto.group.id = toArrayBuffer(groupId);
|
|
proto.group.type = textsecure.protobuf.GroupContext.Type.UPDATE;
|
|
|
|
return textsecure.storage.groups.addNumbers(groupId, [number]).then(function(numbers) {
|
|
if (numbers === undefined)
|
|
return Promise.reject(new Error("Unknown Group"));
|
|
proto.group.members = numbers;
|
|
|
|
return this.sendGroupProto(numbers, proto);
|
|
}.bind(this));
|
|
},
|
|
|
|
setGroupName: function(groupId, name) {
|
|
var proto = new textsecure.protobuf.DataMessage();
|
|
proto.group = new textsecure.protobuf.GroupContext();
|
|
proto.group.id = toArrayBuffer(groupId);
|
|
proto.group.type = textsecure.protobuf.GroupContext.Type.UPDATE;
|
|
proto.group.name = name;
|
|
|
|
return textsecure.storage.groups.getNumbers(groupId).then(function(numbers) {
|
|
if (numbers === undefined)
|
|
return Promise.reject(new Error("Unknown Group"));
|
|
proto.group.members = numbers;
|
|
|
|
return this.sendGroupProto(numbers, proto);
|
|
}.bind(this));
|
|
},
|
|
|
|
setGroupAvatar: function(groupId, avatar) {
|
|
var proto = new textsecure.protobuf.DataMessage();
|
|
proto.group = new textsecure.protobuf.GroupContext();
|
|
proto.group.id = toArrayBuffer(groupId);
|
|
proto.group.type = textsecure.protobuf.GroupContext.Type.UPDATE;
|
|
|
|
return textsecure.storage.groups.getNumbers(groupId).then(function(numbers) {
|
|
if (numbers === undefined)
|
|
return Promise.reject(new Error("Unknown Group"));
|
|
proto.group.members = numbers;
|
|
|
|
return this.makeAttachmentPointer(avatar).then(function(attachment) {
|
|
proto.group.avatar = attachment;
|
|
return this.sendGroupProto(numbers, proto);
|
|
}.bind(this));
|
|
}.bind(this));
|
|
},
|
|
|
|
leaveGroup: function(groupId) {
|
|
var proto = new textsecure.protobuf.DataMessage();
|
|
proto.group = new textsecure.protobuf.GroupContext();
|
|
proto.group.id = toArrayBuffer(groupId);
|
|
proto.group.type = textsecure.protobuf.GroupContext.Type.QUIT;
|
|
|
|
return textsecure.storage.groups.getNumbers(groupId).then(function(numbers) {
|
|
if (numbers === undefined)
|
|
return Promise.reject(new Error("Unknown Group"));
|
|
return textsecure.storage.groups.deleteGroup(groupId).then(function() {
|
|
return this.sendGroupProto(numbers, proto);
|
|
}.bind(this));
|
|
});
|
|
}
|
|
};
|
|
|
|
window.textsecure = window.textsecure || {};
|
|
|
|
textsecure.MessageSender = function(url, username, password) {
|
|
var sender = new MessageSender(url, username, password);
|
|
textsecure.replay.registerFunction(sender.tryMessageAgain.bind(sender), textsecure.replay.Type.ENCRYPT_MESSAGE);
|
|
textsecure.replay.registerFunction(sender.transmitMessage.bind(sender), textsecure.replay.Type.TRANSMIT_MESSAGE);
|
|
|
|
this.sendRequestGroupSyncMessage = sender.sendRequestGroupSyncMessage .bind(sender);
|
|
this.sendRequestContactSyncMessage = sender.sendRequestContactSyncMessage.bind(sender);
|
|
this.sendMessageToNumber = sender.sendMessageToNumber .bind(sender);
|
|
this.closeSession = sender.closeSession .bind(sender);
|
|
this.sendMessageToGroup = sender.sendMessageToGroup .bind(sender);
|
|
this.createGroup = sender.createGroup .bind(sender);
|
|
this.updateGroup = sender.updateGroup .bind(sender);
|
|
this.addNumberToGroup = sender.addNumberToGroup .bind(sender);
|
|
this.setGroupName = sender.setGroupName .bind(sender);
|
|
this.setGroupAvatar = sender.setGroupAvatar .bind(sender);
|
|
this.leaveGroup = sender.leaveGroup .bind(sender);
|
|
};
|
|
|
|
textsecure.MessageSender.prototype = {
|
|
constructor: textsecure.MessageSender
|
|
};
|