keejef
/
session-desktop
mirror of https://github.com/oxen-io/session-desktop
2
0
Fork 0
Code Issues 8 Projects Releases Wiki Activity

Remove preKeys on friend request decline.

Browse Source
pull/54/head
Mikunj 7 years ago
parent 0dabce9e28
commit 9fd822a7e3
6 changed files with 66 additions and 45 deletions
Show Stats Download Patch File Download Diff File

16
app/sql.js
Unescape Escape View File

@ -49,7 +49,7 @@ module.exports = {
getContactPreKeys,
getAllContactPreKeys,
bulkAddContactPreKeys,
removeContactPreKeyById,
removeContactPreKeyByIdentityKey,
removeAllContactPreKeys,
createOrUpdateContactSignedPreKey,
@ -57,7 +57,7 @@ module.exports = {
getContactSignedPreKeyByIdentityKey,
getContactSignedPreKeys,
bulkAddContactSignedPreKeys,
removeContactSignedPreKeyById,
removeContactSignedPreKeyByIdentityKey,
removeAllContactSignedPreKeys,
createOrUpdateItem,
@ -712,8 +712,10 @@ async function getContactPreKeys(keyId, identityKeyString) {
async function bulkAddContactPreKeys(array) {
return bulkAdd(CONTACT_PRE_KEYS_TABLE, array);
}
async function removeContactPreKeyById(id) {
return removeById(CONTACT_PRE_KEYS_TABLE, id);
async function removeContactPreKeyByIdentityKey(key) {
await db.run(`DELETE FROM ${CONTACT_PRE_KEYS_TABLE} WHERE identityKeyString = $identityKeyString;`, {
$identityKeyString: key,
});
}
async function removeAllContactPreKeys() {
return removeAllFromTable(CONTACT_PRE_KEYS_TABLE);
@ -765,8 +767,10 @@ async function getContactSignedPreKeys(keyId, identityKeyString) {
async function bulkAddContactSignedPreKeys(array) {
return bulkAdd(CONTACT_SIGNED_PRE_KEYS_TABLE, array);
}
async function removeContactSignedPreKeyById(id) {
return removeById(CONTACT_SIGNED_PRE_KEYS_TABLE, id);
async function removeContactSignedPreKeyByIdentityKey(key) {
await db.run(`DELETE FROM ${CONTACT_SIGNED_PRE_KEYS_TABLE} WHERE identityKeyString = $identityKeyString;`, {
$identityKeyString: key,
});
}
async function removeAllContactSignedPreKeys() {
return removeAllFromTable(CONTACT_SIGNED_PRE_KEYS_TABLE);

42
js/modules/data.js
Unescape Escape View File

@ -80,7 +80,7 @@ module.exports = {
getContactPreKeys,
getAllContactPreKeys,
bulkAddContactPreKeys,
removeContactPreKeyById,
removeContactPreKeyByIdentityKey,
removeAllContactPreKeys,
createOrUpdateContactSignedPreKey,
@ -88,7 +88,7 @@ module.exports = {
getContactSignedPreKeyByIdentityKey,
getContactSignedPreKeys,
bulkAddContactSignedPreKeys,
removeContactSignedPreKeyById,
removeContactSignedPreKeyByIdentityKey,
removeAllContactSignedPreKeys,
createOrUpdateItem,
@ -528,8 +528,8 @@ async function bulkAddContactPreKeys(array) {
const updated = map(array, data => keysFromArrayBuffer(PRE_KEY_KEYS, data));
await channels.bulkAddContactPreKeys(updated);
}
async function removeContactPreKeyById(id) {
await channels.removePreContactKeyById(id);
async function removeContactPreKeyByIdentityKey(id) {
await channels.removeContactPreKeyByIdentityKey(id);
}
async function removeAllContactPreKeys() {
await channels.removeAllContactPreKeys();
@ -556,8 +556,8 @@ async function bulkAddContactSignedPreKeys(array) {
const updated = map(array, data => keysFromArrayBuffer(PRE_KEY_KEYS, data));
await channels.bulkAddContactSignedPreKeys(updated);
}
async function removeContactSignedPreKeyById(id) {
await channels.removePreContactSignedKeyById(id);
async function removeContactSignedPreKeyByIdentityKey(id) {
await channels.removeContactSignedPreKeyByIdentityKey(id);
}
async function removeAllContactSignedPreKeys() {
await channels.removeAllContactSignedPreKeys();
@ -812,22 +812,18 @@ async function getMessageById(id, { Message }) {
return null;
}
const encoded = keysToArrayBuffer(MESSAGE_PRE_KEYS, message);
return new Message(encoded);
return new Message(message);
}
// For testing only
async function getAllMessages({ MessageCollection }) {
const messages = await channels.getAllMessages();
const encoded = messages.map(m => keysToArrayBuffer(MESSAGE_PRE_KEYS, m));
return new MessageCollection(encoded);
return new MessageCollection(messages);
}
async function getAllUnsentMessages({ MessageCollection }) {
const messages = await channels.getAllUnsentMessages();
const encoded = messages.map(m => keysToArrayBuffer(MESSAGE_PRE_KEYS, m));
return new MessageCollection(encoded);
return new MessageCollection(messages);
}
async function getAllMessageIds() {
@ -849,9 +845,7 @@ async function getMessageBySender(
return null;
}
const encoded = keysToArrayBuffer(MESSAGE_PRE_KEYS, messages[0]);
return new Message(encoded);
return new Message(messages[0]);
}
async function getUnreadByConversation(conversationId, { MessageCollection }) {
@ -869,9 +863,7 @@ async function getMessagesByConversation(
type,
});
const encoded = messages.map(m => keysToArrayBuffer(MESSAGE_PRE_KEYS, m));
return new MessageCollection(encoded);
return new MessageCollection(messages);
}
async function getSeenMessagesByHashList(
@ -913,26 +905,22 @@ async function removeAllMessagesInConversation(
async function getMessagesBySentAt(sentAt, { MessageCollection }) {
const messages = await channels.getMessagesBySentAt(sentAt);
const encoded = messages.map(m => keysToArrayBuffer(MESSAGE_PRE_KEYS, m));
return new MessageCollection(encoded);
return new MessageCollection(messages);
}
async function getExpiredMessages({ MessageCollection }) {
const messages = await channels.getExpiredMessages();
const encoded = messages.map(m => keysToArrayBuffer(MESSAGE_PRE_KEYS, m));
return new MessageCollection(encoded);
return new MessageCollection(messages);
}
async function getOutgoingWithoutExpiresAt({ MessageCollection }) {
const messages = await channels.getOutgoingWithoutExpiresAt();
const encoded = messages.map(m => keysToArrayBuffer(MESSAGE_PRE_KEYS, m));
return new MessageCollection(encoded);
return new MessageCollection(messages);
}
async function getNextExpiringMessage({ MessageCollection }) {
const messages = await channels.getNextExpiringMessage();
const encoded = messages.map(m => keysToArrayBuffer(MESSAGE_PRE_KEYS, m));
return new MessageCollection(encoded);
return new MessageCollection(messages);
}
// Unprocessed

18
js/signal_protocol_store.js
Unescape Escape View File

@ -162,7 +162,7 @@
},
async getLocalRegistrationId() {
return 1;
// const item = await window.Signal.Data.getItemById('registrationId');
// if (item) {
// return item.value;
@ -197,6 +197,8 @@
recipient: key.recipient,
};
}
return undefined;
},
async loadContactPreKey(pubKey) {
const preKey = await window.Signal.Data.getContactPreKeyByIdentityKey(pubKey);
@ -266,6 +268,13 @@
await window.Signal.Data.removeAllPreKeys();
},
async removeContactPreKey(pubKey) {
await window.Signal.Data.removeContactPreKeyByIdentityKey(pubKey);
},
async clearContactPreKeysStore() {
await window.Signal.Data.removeAllContactPreKeys();
},
/* Returns a signed keypair object or undefined */
async loadSignedPreKey(keyId) {
const key = await window.Signal.Data.getSignedPreKeyById(keyId);
@ -366,6 +375,13 @@
await window.Signal.Data.removeAllSignedPreKeys();
},
async removeContactSignedPreKey(pubKey) {
await window.Signal.Data.removeContactSignedPreKeyByIdentityKey(pubKey);
},
async clearContactSignedPreKeysStore() {
await window.Signal.Data.removeAllContactSignedPreKeys();
},
async loadSession(encodedNumber) {
if (encodedNumber === null || encodedNumber === undefined) {
throw new Error('Tried to get session for undefined/null number');

8
libloki/libloki-protocol.js
Unescape Escape View File

@ -122,6 +122,13 @@
await Promise.all([signedKeyPromise, preKeyPromise]);
}
async function removePreKeyBundleForNumber(pubKey) {
await Promise.all([
textsecure.storage.protocol.removeContactPreKey(pubKey),
textsecure.storage.protocol.removeContactSignedPreKey(pubKey),
]);
}
async function sendFriendRequestAccepted(pubKey) {
// empty content message
const content = new textsecure.protobuf.Content();
@ -154,5 +161,6 @@
window.libloki.getPreKeyBundleForNumber = getPreKeyBundleForNumber;
window.libloki.FallBackDecryptionError = FallBackDecryptionError;
window.libloki.savePreKeyBundleForNumber = savePreKeyBundleForNumber;
window.libloki.removePreKeyBundleForNumber = removePreKeyBundleForNumber;
window.libloki.sendFriendRequestAccepted = sendFriendRequestAccepted;
})();

4
libtextsecure/account_manager.js
Unescape Escape View File

@ -328,7 +328,7 @@
textsecure.storage.remove('userAgent'),
textsecure.storage.remove('read-receipts-setting'),
]);
// update our own identity key, which may have changed
// if we're relinking after a reinstall on the master device
await textsecure.storage.protocol.saveIdentityWithAttributes(pubKeyString, {
@ -362,7 +362,9 @@
window.log.info('clearing all sessions, prekeys, and signed prekeys');
return Promise.all([
store.clearPreKeyStore(),
store.clearContactPreKeysStore(),
store.clearSignedPreKeysStore(),
store.clearContactSignedPreKeysStore(),
store.clearSessionStore(),
]);
},

23
libtextsecure/message_receiver.js
Unescape Escape View File

@ -1005,20 +1005,23 @@ MessageReceiver.prototype.extend({
await conversation.updateTextInputState();
}
// If we accepted an incoming friend request then update our state
if (message.direction === 'incoming' && message.friendStatus === 'accepted') {
// Check if we changed the state of the incoming friend request
if (message.direction === 'incoming') {
// If we accepted an incoming friend request then update our state
if (message.friendStatus === 'accepted') {
// Accept the friend request
if (conversation) {
await conversation.onFriendRequestAccepted();
}
// Accept the friend request
if (conversation) {
await conversation.onFriendRequestAccepted();
// Send a reply back
libloki.sendFriendRequestAccepted(pubKey);
} else if (message.friendStatus === 'declined') {
// Delete the preKeys
await libloki.removePreKeyBundleForNumber(pubKey);
}
// Send a reply back
libloki.sendFriendRequestAccepted(pubKey);
}
// TODO: If we decline a friend request then delete preKeys from our db
window.log.info(`Friend request for ${pubKey} was ${message.friendStatus}`, message);
},
async innerHandleContentMessage(envelope, plaintext) {

Write Preview
Loading…
Cancel
Save