160 lines
3.6 KiB
TypeScript
160 lines
3.6 KiB
TypeScript
// Copyright 2020 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
/* eslint-disable max-classes-per-file */
|
|
|
|
import { Reader } from 'protobufjs';
|
|
|
|
import { SignalService as Proto } from '../protobuf';
|
|
import { normalizeUuid } from '../util/normalizeUuid';
|
|
import { typedArrayToArrayBuffer } from '../Crypto';
|
|
|
|
import Avatar = Proto.ContactDetails.IAvatar;
|
|
|
|
type OptionalAvatar = { avatar?: Avatar | null };
|
|
|
|
type DecoderBase<Message extends OptionalAvatar> = {
|
|
decodeDelimited(reader: Reader): Message | undefined;
|
|
};
|
|
|
|
export type MessageWithAvatar<Message extends OptionalAvatar> = Omit<
|
|
Message,
|
|
'avatar'
|
|
> & {
|
|
avatar?: (Avatar & { data: ArrayBuffer }) | null;
|
|
};
|
|
|
|
export type ModifiedGroupDetails = MessageWithAvatar<Proto.GroupDetails>;
|
|
|
|
export type ModifiedContactDetails = MessageWithAvatar<Proto.ContactDetails>;
|
|
|
|
// TODO: remove once we move away from ArrayBuffers
|
|
const FIXMEU8 = Uint8Array;
|
|
|
|
class ParserBase<
|
|
Message extends OptionalAvatar,
|
|
Decoder extends DecoderBase<Message>
|
|
> {
|
|
protected readonly reader: Reader;
|
|
|
|
constructor(arrayBuffer: ArrayBuffer, private readonly decoder: Decoder) {
|
|
this.reader = new Reader(new FIXMEU8(arrayBuffer));
|
|
}
|
|
|
|
protected decodeDelimited(): MessageWithAvatar<Message> | undefined {
|
|
if (this.reader.pos === this.reader.len) {
|
|
return undefined; // eof
|
|
}
|
|
|
|
try {
|
|
const proto = this.decoder.decodeDelimited(this.reader);
|
|
|
|
if (!proto) {
|
|
return undefined;
|
|
}
|
|
|
|
if (!proto.avatar) {
|
|
return {
|
|
...proto,
|
|
avatar: null,
|
|
};
|
|
}
|
|
|
|
const attachmentLen = proto.avatar.length ?? 0;
|
|
const avatarData = this.reader.buf.slice(
|
|
this.reader.pos,
|
|
this.reader.pos + attachmentLen
|
|
);
|
|
this.reader.skip(attachmentLen);
|
|
|
|
return {
|
|
...proto,
|
|
|
|
avatar: {
|
|
...proto.avatar,
|
|
|
|
data: typedArrayToArrayBuffer(avatarData),
|
|
},
|
|
};
|
|
} catch (error) {
|
|
window.log.error(
|
|
'ProtoParser.next error:',
|
|
error && error.stack ? error.stack : error
|
|
);
|
|
return undefined;
|
|
}
|
|
}
|
|
}
|
|
|
|
export class GroupBuffer extends ParserBase<
|
|
Proto.GroupDetails,
|
|
typeof Proto.GroupDetails
|
|
> {
|
|
constructor(arrayBuffer: ArrayBuffer) {
|
|
super(arrayBuffer, Proto.GroupDetails);
|
|
}
|
|
|
|
public next(): ModifiedGroupDetails | undefined {
|
|
const proto = this.decodeDelimited();
|
|
if (!proto) {
|
|
return undefined;
|
|
}
|
|
|
|
if (!proto.members) {
|
|
return proto;
|
|
}
|
|
return {
|
|
...proto,
|
|
members: proto.members.map((member, i) => {
|
|
if (!member.uuid) {
|
|
return member;
|
|
}
|
|
|
|
return {
|
|
...member,
|
|
uuid: normalizeUuid(member.uuid, `GroupBuffer.member[${i}].uuid`),
|
|
};
|
|
}),
|
|
};
|
|
}
|
|
}
|
|
|
|
export class ContactBuffer extends ParserBase<
|
|
Proto.ContactDetails,
|
|
typeof Proto.ContactDetails
|
|
> {
|
|
constructor(arrayBuffer: ArrayBuffer) {
|
|
super(arrayBuffer, Proto.ContactDetails);
|
|
}
|
|
|
|
public next(): ModifiedContactDetails | undefined {
|
|
const proto = this.decodeDelimited();
|
|
if (!proto) {
|
|
return undefined;
|
|
}
|
|
|
|
if (!proto.uuid) {
|
|
return proto;
|
|
}
|
|
|
|
const { verified } = proto;
|
|
|
|
return {
|
|
...proto,
|
|
|
|
verified:
|
|
verified && verified.destinationUuid
|
|
? {
|
|
...verified,
|
|
|
|
destinationUuid: normalizeUuid(
|
|
verified.destinationUuid,
|
|
'ContactBuffer.verified.destinationUuid'
|
|
),
|
|
}
|
|
: verified,
|
|
|
|
uuid: normalizeUuid(proto.uuid, 'ContactBuffer.uuid'),
|
|
};
|
|
}
|
|
}
|