linted
Vincent
5 years ago
parent
bbfa6499e0
commit
afcbf7049e
@ -1,251 +0,0 @@
|
|||||||
import { SessionResetMessage } from '../messages/outgoing';
|
|
||||||
// import { MessageSender } from '../sending';
|
|
||||||
|
|
||||||
interface StringToNumberMap {
|
|
||||||
[key: string]: number;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* This map olds the sent session timestamps, i.e. session requests message effectively sent to the recipient.
|
|
||||||
* It is backed by a database entry so it's loaded from db on startup.
|
|
||||||
* This map should not be used directly, but instead through
|
|
||||||
* `_updateSendSessionTimestamp()`, `_getSendSessionRequest()` or `_hasSendSessionRequest()`
|
|
||||||
*/
|
|
||||||
let sentSessionsTimestamp: StringToNumberMap;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* This map olds the processed session timestamps, i.e. when we received a session request and handled it.
|
|
||||||
* It is backed by a database entry so it's loaded from db on startup.
|
|
||||||
* This map should not be used directly, but instead through
|
|
||||||
* `_updateProcessedSessionTimestamp()`, `_getProcessedSessionRequest()` or `_hasProcessedSessionRequest()`
|
|
||||||
*/
|
|
||||||
let processedSessionsTimestamp: StringToNumberMap;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* This map olds the timestamp on which a sent session reset is triggered for a specific device.
|
|
||||||
* Once the message is sent or failed to sent, this device is removed from here.
|
|
||||||
* This is a memory only map. Which means that on app restart it's starts empty.
|
|
||||||
*/
|
|
||||||
const pendingSendSessionsTimestamp: Set<string> = new Set();
|
|
||||||
|
|
||||||
/** ======= exported functions ======= */
|
|
||||||
|
|
||||||
/** Returns true if we already have a session with that device */
|
|
||||||
export async function hasSession(device: string): Promise<boolean> {
|
|
||||||
// Session does not use the concept of a deviceId, thus it's always 1
|
|
||||||
const address = new window.libsignal.SignalProtocolAddress(device, 1);
|
|
||||||
const sessionCipher = new window.libsignal.SessionCipher(
|
|
||||||
window.textsecure.storage.protocol,
|
|
||||||
address
|
|
||||||
);
|
|
||||||
|
|
||||||
return sessionCipher.hasOpenSession();
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns true if we sent a session request to that device already OR
|
|
||||||
* if a session request to that device is right now being sent.
|
|
||||||
*/
|
|
||||||
export async function hasSentSessionRequest(device: string): Promise<boolean> {
|
|
||||||
const pendingSend = pendingSendSessionsTimestamp.has(device);
|
|
||||||
const hasSent = await _hasSentSessionRequest(device);
|
|
||||||
|
|
||||||
return pendingSend || hasSent;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Triggers a SessionResetMessage to be sent if:
|
|
||||||
* - we do not already have a session and
|
|
||||||
* - we did not sent a session request already to that device and
|
|
||||||
* - we do not have a session request currently being send to that device
|
|
||||||
*/
|
|
||||||
export async function sendSessionRequestIfNeeded(
|
|
||||||
device: string
|
|
||||||
): Promise<void> {
|
|
||||||
if (hasSession(device) || hasSentSessionRequest(device)) {
|
|
||||||
return Promise.resolve();
|
|
||||||
}
|
|
||||||
|
|
||||||
const preKeyBundle = await window.libloki.storage.getPreKeyBundleForContact(
|
|
||||||
device
|
|
||||||
);
|
|
||||||
const sessionReset = new SessionResetMessage({
|
|
||||||
preKeyBundle,
|
|
||||||
timestamp: Date.now(),
|
|
||||||
});
|
|
||||||
|
|
||||||
return sendSessionRequests(sessionReset, device);
|
|
||||||
}
|
|
||||||
|
|
||||||
/** */
|
|
||||||
export async function sendSessionRequests(
|
|
||||||
message: SessionResetMessage,
|
|
||||||
device: string
|
|
||||||
): Promise<void> {
|
|
||||||
const timestamp = Date.now();
|
|
||||||
|
|
||||||
// mark the session as being pending send with current timestamp
|
|
||||||
// so we know we already triggered a new session with that device
|
|
||||||
pendingSendSessionsTimestamp.add(device);
|
|
||||||
// const rawMessage = toRawMessage(message);
|
|
||||||
// // TODO: Send out the request via MessageSender
|
|
||||||
|
|
||||||
// return MessageSender.send(rawMessage)
|
|
||||||
// .then(async () => {
|
|
||||||
// await _updateSentSessionTimestamp(device, timestamp);
|
|
||||||
// pendingSendSessionsTimestamp.delete(device);
|
|
||||||
// })
|
|
||||||
// .catch(() => {
|
|
||||||
// pendingSendSessionsTimestamp.delete(device);
|
|
||||||
// });
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Called when a session is establish so we store on database this info.
|
|
||||||
*/
|
|
||||||
export async function onSessionEstablished(device: string) {
|
|
||||||
// remove our existing sent timestamp for that device
|
|
||||||
return _updateSentSessionTimestamp(device, undefined);
|
|
||||||
}
|
|
||||||
|
|
||||||
export async function shouldProcessSessionRequest(
|
|
||||||
device: string,
|
|
||||||
messageTimestamp: number
|
|
||||||
): Promise<boolean> {
|
|
||||||
const existingSentTimestamp = (await _getSentSessionRequest(device)) || 0;
|
|
||||||
const existingProcessedTimestamp =
|
|
||||||
(await _getProcessedSessionRequest(device)) || 0;
|
|
||||||
|
|
||||||
return (
|
|
||||||
messageTimestamp > existingSentTimestamp &&
|
|
||||||
messageTimestamp > existingProcessedTimestamp
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
export async function onSessionRequestProcessed(device: string) {
|
|
||||||
return _updateProcessedSessionTimestamp(device, Date.now());
|
|
||||||
}
|
|
||||||
|
|
||||||
/** ======= local / utility functions ======= */
|
|
||||||
|
|
||||||
/**
|
|
||||||
* We only need to fetch once from the database, because we are the only one writing to it
|
|
||||||
*/
|
|
||||||
async function _fetchFromDBIfNeeded(): Promise<void> {
|
|
||||||
if (!sentSessionsTimestamp) {
|
|
||||||
const sentItem = await window.Signal.Data.getItemById(
|
|
||||||
'sentSessionsTimestamp'
|
|
||||||
);
|
|
||||||
if (sentItem) {
|
|
||||||
sentSessionsTimestamp = sentItem.value;
|
|
||||||
} else {
|
|
||||||
sentSessionsTimestamp = {};
|
|
||||||
}
|
|
||||||
|
|
||||||
const processedItem = await window.Signal.Data.getItemById(
|
|
||||||
'processedSessionsTimestamp'
|
|
||||||
);
|
|
||||||
if (processedItem) {
|
|
||||||
processedSessionsTimestamp = processedItem.value;
|
|
||||||
} else {
|
|
||||||
processedSessionsTimestamp = {};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function _writeToDBSentSessions(): Promise<void> {
|
|
||||||
const data = {
|
|
||||||
id: 'sentSessionsTimestamp',
|
|
||||||
value: JSON.stringify(sentSessionsTimestamp),
|
|
||||||
};
|
|
||||||
|
|
||||||
await window.Signal.Data.createOrUpdateItem(data);
|
|
||||||
}
|
|
||||||
|
|
||||||
async function _writeToDBProcessedSessions(): Promise<void> {
|
|
||||||
const data = {
|
|
||||||
id: 'processedSessionsTimestamp',
|
|
||||||
value: JSON.stringify(processedSessionsTimestamp),
|
|
||||||
};
|
|
||||||
|
|
||||||
await window.Signal.Data.createOrUpdateItem(data);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* This is a utility function to avoid duplicated code of _updateSentSessionTimestamp and _updateProcessedSessionTimestamp
|
|
||||||
*/
|
|
||||||
async function _updateSessionTimestamp(
|
|
||||||
device: string,
|
|
||||||
timestamp: number | undefined,
|
|
||||||
map: StringToNumberMap
|
|
||||||
): Promise<boolean> {
|
|
||||||
await _fetchFromDBIfNeeded();
|
|
||||||
if (!timestamp) {
|
|
||||||
if (!!map[device]) {
|
|
||||||
delete map.device;
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
map[device] = timestamp;
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
*
|
|
||||||
* @param device the device id
|
|
||||||
* @param timestamp undefined to remove the key/value pair, otherwise updates the sent timestamp and write to DB
|
|
||||||
*/
|
|
||||||
async function _updateSentSessionTimestamp(
|
|
||||||
device: string,
|
|
||||||
timestamp: number | undefined
|
|
||||||
): Promise<void> {
|
|
||||||
if (_updateSessionTimestamp(device, timestamp, sentSessionsTimestamp)) {
|
|
||||||
await _writeToDBSentSessions();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* timestamp undefined to remove the key/value pair, otherwise updates the processed timestamp and writes to DB
|
|
||||||
*/
|
|
||||||
async function _updateProcessedSessionTimestamp(
|
|
||||||
device: string,
|
|
||||||
timestamp: number | undefined
|
|
||||||
): Promise<void> {
|
|
||||||
if (_updateSessionTimestamp(device, timestamp, processedSessionsTimestamp)) {
|
|
||||||
await _writeToDBProcessedSessions();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* This is a utility function to avoid duplicate code between `_getProcessedSessionRequest()` and `_getSentSessionRequest()`
|
|
||||||
*/
|
|
||||||
async function _getSessionRequest(
|
|
||||||
device: string,
|
|
||||||
map: StringToNumberMap
|
|
||||||
): Promise<number | undefined> {
|
|
||||||
await _fetchFromDBIfNeeded();
|
|
||||||
|
|
||||||
return map[device];
|
|
||||||
}
|
|
||||||
|
|
||||||
async function _getSentSessionRequest(
|
|
||||||
device: string
|
|
||||||
): Promise<number | undefined> {
|
|
||||||
return _getSessionRequest(device, sentSessionsTimestamp);
|
|
||||||
}
|
|
||||||
|
|
||||||
async function _getProcessedSessionRequest(
|
|
||||||
device: string
|
|
||||||
): Promise<number | undefined> {
|
|
||||||
return _getSessionRequest(device, processedSessionsTimestamp);
|
|
||||||
}
|
|
||||||
|
|
||||||
async function _hasSentSessionRequest(device: string): Promise<boolean> {
|
|
||||||
await _fetchFromDBIfNeeded();
|
|
||||||
|
|
||||||
return !!sentSessionsTimestamp[device];
|
|
||||||
}
|
|
||||||
@ -1,9 +0,0 @@
|
|||||||
// import { expect } from 'chai';
|
|
||||||
// import { SessionProtocol } from '../../../session/protocols';
|
|
||||||
|
|
||||||
// describe('SessionProtocol', () => {
|
|
||||||
// it('has ', () => {
|
|
||||||
|
|
||||||
// });
|
|
||||||
|
|
||||||
// });
|
|
||||||
Loading…
Reference in New Issue