mirrors/signal-desktop
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages 2 Wiki Activity Actions

MessageReceiver: Extract cipherTextBytes/Type from non-sealed messages

Browse source
This commit is contained in:
Scott Nonnenberg 2022-02-16 09:41:28 -08:00 • committed by GitHub
parent ec3a3bd7b2
commit 0bb0e6e9fc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 27 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

33
ts/textsecure/MessageReceiver.ts
View file

@ -14,6 +14,7 @@ import type {
UnidentifiedSenderMessageContent,
} from '@signalapp/signal-client';
import {
CiphertextMessageType,
DecryptionErrorMessage,
groupDecrypt,
PlaintextContent,
@ -116,7 +117,8 @@ type UnsealedEnvelope = Readonly<
unidentifiedDeliveryReceived?: boolean;
contentHint?: number;
groupId?: string;
usmc?: UnidentifiedSenderMessageContent;
cipherTextBytes?: Uint8Array;
cipherTextType?: number;
certificate?: SenderCertificate;
unsealedContent?: UnidentifiedSenderMessageContent;
}
@ -1128,7 +1130,11 @@ export default class MessageReceiver
}
if (envelope.type !== Proto.Envelope.Type.UNIDENTIFIED_SENDER) {
return envelope;
return {
...envelope,
cipherTextBytes: envelope.content,
cipherTextType: envelopeTypeToCiphertextType(envelope.type),
};
}
if (uuidKind === UUIDKind.PNI) {
@ -1160,6 +1166,9 @@ export default class MessageReceiver
const newEnvelope: UnsealedEnvelope = {
...envelope,
cipherTextBytes: messageContent.contents(),
cipherTextType: messageContent.msgType(),
// Overwrite Envelope fields
source: dropNull(certificate.senderE164()),
sourceUuid: normalizeUuid(
@ -1172,7 +1181,6 @@ export default class MessageReceiver
unidentifiedDeliveryReceived: !(originalSource || originalSourceUuid),
contentHint: messageContent.contentHint(),
groupId: messageContent.groupId()?.toString('base64'),
usmc: messageContent,
certificate,
unsealedContent: messageContent,
};
@ -1660,11 +1668,11 @@ export default class MessageReceiver
}
if (uuid && deviceId) {
const { usmc } = envelope;
const { cipherTextBytes, cipherTextType } = envelope;
const event = new DecryptionErrorEvent(
{
cipherTextBytes: usmc ? usmc.contents() : undefined,
cipherTextType: usmc ? usmc.msgType() : undefined,
cipherTextBytes,
cipherTextType,
contentHint: envelope.contentHint,
groupId: envelope.groupId,
receivedAtCounter: envelope.receivedAtCounter,
@ -2782,3 +2790,16 @@ export default class MessageReceiver
return processDataMessage(decrypted, envelope.timestamp);
}
}
function envelopeTypeToCiphertextType(type: number | undefined): number {
if (type === Proto.Envelope.Type.CIPHERTEXT) {
return CiphertextMessageType.Whisper;
}
if (type === Proto.Envelope.Type.PLAINTEXT_CONTENT) {
return CiphertextMessageType.Plaintext;
}
if (type === Proto.Envelope.Type.PREKEY_BUNDLE) {
return CiphertextMessageType.PreKey;
}
throw new Error(`envelopeTypeToCiphertextType: Unknown type ${type}`);
}