gifted-baileys 1.5.4 → 1.5.5

package/src/Utils/noise-handler.ts

@@ -0,0 +1,197 @@
+import { Boom } from '@hapi/boom'
+import { Logger } from 'pino'
+import { proto } from '../../WAProto'
+import { NOISE_MODE, WA_CERT_DETAILS } from '../Defaults'
+import { KeyPair } from '../Types'
+import { BinaryNode, decodeBinaryNode } from '../WABinary'
+import { aesDecryptGCM, aesEncryptGCM, Curve, hkdf, sha256 } from './crypto'
+
+const generateIV = (counter: number) => {
+	const iv = new ArrayBuffer(12)
+	new DataView(iv).setUint32(8, counter)
+
+	return new Uint8Array(iv)
+}
+
+export const makeNoiseHandler = ({
+	keyPair: { private: privateKey, public: publicKey },
+	NOISE_HEADER,
+	mobile,
+	logger,
+	routingInfo
+}: {
+	keyPair: KeyPair
+	NOISE_HEADER: Uint8Array
+	mobile: boolean
+	logger: Logger
+	routingInfo?: Buffer | undefined
+}) => {
+	logger = logger.child({ class: 'ns' })
+
+	const authenticate = (data: Uint8Array) => {
+		if(!isFinished) {
+			hash = sha256(Buffer.concat([hash, data]))
+		}
+	}
+
+	const encrypt = (plaintext: Uint8Array) => {
+		const result = aesEncryptGCM(plaintext, encKey, generateIV(writeCounter), hash)
+
+		writeCounter += 1
+
+		authenticate(result)
+		return result
+	}
+
+	const decrypt = (ciphertext: Uint8Array) => {
+		// before the handshake is finished, we use the same counter
+		// after handshake, the counters are different
+		const iv = generateIV(isFinished ? readCounter : writeCounter)
+		const result = aesDecryptGCM(ciphertext, decKey, iv, hash)
+
+		if(isFinished) {
+			readCounter += 1
+		} else {
+			writeCounter += 1
+		}
+
+		authenticate(ciphertext)
+		return result
+	}
+
+	const localHKDF = (data: Uint8Array) => {
+		const key = hkdf(Buffer.from(data), 64, { salt, info: '' })
+		return [key.slice(0, 32), key.slice(32)]
+	}
+
+	const mixIntoKey = (data: Uint8Array) => {
+		const [write, read] = localHKDF(data)
+		salt = write
+		encKey = read
+		decKey = read
+		readCounter = 0
+		writeCounter = 0
+	}
+
+	const finishInit = () => {
+		const [write, read] = localHKDF(new Uint8Array(0))
+		encKey = write
+		decKey = read
+		hash = Buffer.from([])
+		readCounter = 0
+		writeCounter = 0
+		isFinished = true
+	}
+
+	const data = Buffer.from(NOISE_MODE)
+	let hash = Buffer.from(data.byteLength === 32 ? data : sha256(data))
+	let salt = hash
+	let encKey = hash
+	let decKey = hash
+	let readCounter = 0
+	let writeCounter = 0
+	let isFinished = false
+	let sentIntro = false
+
+	let inBytes = Buffer.alloc(0)
+
+	authenticate(NOISE_HEADER)
+	authenticate(publicKey)
+
+	return {
+		encrypt,
+		decrypt,
+		authenticate,
+		mixIntoKey,
+		finishInit,
+		processHandshake: ({ serverHello }: proto.HandshakeMessage, noiseKey: KeyPair) => {
+			authenticate(serverHello!.ephemeral!)
+			mixIntoKey(Curve.sharedKey(privateKey, serverHello!.ephemeral!))
+
+			const decStaticContent = decrypt(serverHello!.static!)
+			mixIntoKey(Curve.sharedKey(privateKey, decStaticContent))
+
+			const certDecoded = decrypt(serverHello!.payload!)
+
+			if(mobile) {
+				proto.CertChain.NoiseCertificate.decode(certDecoded)
+			} else {
+				const { intermediate: certIntermediate } = proto.CertChain.decode(certDecoded)
+
+				const { issuerSerial } = proto.CertChain.NoiseCertificate.Details.decode(certIntermediate!.details!)
+
+				if(issuerSerial !== WA_CERT_DETAILS.SERIAL) {
+					throw new Boom('certification match failed', { statusCode: 400 })
+				}
+			}
+
+			const keyEnc = encrypt(noiseKey.public)
+			mixIntoKey(Curve.sharedKey(noiseKey.private, serverHello!.ephemeral!))
+
+			return keyEnc
+		},
+		encodeFrame: (data: Buffer | Uint8Array) => {
+			if(isFinished) {
+				data = encrypt(data)
+			}
+
+			let header: Buffer
+
+			if(routingInfo) {
+				header = Buffer.alloc(7)
+				header.write('ED', 0, 'utf8')
+				header.writeUint8(0, 2)
+				header.writeUint8(1, 3)
+				header.writeUint8(routingInfo.byteLength >> 16, 4)
+				header.writeUint16BE(routingInfo.byteLength & 65535, 5)
+				header = Buffer.concat([header, routingInfo, NOISE_HEADER])
+			} else {
+				header = Buffer.from(NOISE_HEADER)
+			}
+
+			const introSize = sentIntro ? 0 : header.length
+			const frame = Buffer.alloc(introSize + 3 + data.byteLength)
+
+			if(!sentIntro) {
+				frame.set(header)
+				sentIntro = true
+			}
+
+			frame.writeUInt8(data.byteLength >> 16, introSize)
+			frame.writeUInt16BE(65535 & data.byteLength, introSize + 1)
+			frame.set(data, introSize + 3)
+
+			return frame
+		},
+		decodeFrame: (newData: Buffer | Uint8Array, onFrame: (buff: Uint8Array | BinaryNode) => void) => {
+			// the binary protocol uses its own framing mechanism
+			// on top of the WS frames
+			// so we get this data and separate out the frames
+			const getBytesSize = () => {
+				if(inBytes.length >= 3) {
+					return (inBytes.readUInt8() << 16) | inBytes.readUInt16BE(1)
+				}
+			}
+
+			inBytes = Buffer.concat([ inBytes, newData ])
+
+			logger.trace(`recv ${newData.length} bytes, total recv ${inBytes.length} bytes`)
+
+			let size = getBytesSize()
+			while(size && inBytes.length >= size + 3) {
+				let frame: Uint8Array | BinaryNode = inBytes.slice(3, size + 3)
+				inBytes = inBytes.slice(size + 3)
+
+				if(isFinished) {
+					const result = decrypt(frame as Uint8Array)
+					frame = decodeBinaryNode(result)
+				}
+
+				logger.trace({ msg: (frame as any)?.attrs?.id }, 'recv frame')
+
+				onFrame(frame)
+				size = getBytesSize()
+			}
+		}
+	}
+}

package/{lib → src}/Utils/process-message.js

@@ -1,4 +1,13 @@
 "use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getChatId = exports.shouldIncrementChatUnread = exports.isRealMessage = exports.cleanMessage = void 0;
 exports.decryptPollVote = decryptPollVote;
@@ -102,8 +111,8 @@ function decryptPollVote({ encPayload, encIv }, { pollCreatorJid, pollMsgId, pol
         return Buffer.from(txt);
     }
 }
-const processMessage = async (message, { shouldProcessHistoryMsg, ev, creds, keyStore, logger, options, getMessage }) => {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k;
+const processMessage = (message_1, _a) => __awaiter(void 0, [message_1, _a], void 0, function* (message, { shouldProcessHistoryMsg, ev, creds, keyStore, logger, options, getMessage }) {
+    var _b, _c, _d, _e, _f, _g, _h, _j, _k, _l;
     const meId = creds.me.id;
     const { accountSettings } = creds;
     const chat = { id: (0, WABinary_1.jidNormalizedUser)((0, exports.getChatId)(message.key)) };
@@ -118,7 +127,7 @@ const processMessage = async (message, { shouldProcessHistoryMsg, ev, creds, key
     const content = (0, messages_1.normalizeMessageContent)(message.message);
     // unarchive chat if it's a real message, or someone reacted to our message
     // and we've the unarchive chats setting on
-    if ((isRealMsg || ((_b = (_a = content === null || content === void 0 ? void 0 : content.reactionMessage) === null || _a === void 0 ? void 0 : _a.key) === null || _b === void 0 ? void 0 : _b.fromMe))
+    if ((isRealMsg || ((_c = (_b = content === null || content === void 0 ? void 0 : content.reactionMessage) === null || _b === void 0 ? void 0 : _b.key) === null || _c === void 0 ? void 0 : _c.fromMe))
         && (accountSettings === null || accountSettings === void 0 ? void 0 : accountSettings.unarchiveChats)) {
         chat.archived = false;
         chat.readOnly = false;
@@ -129,7 +138,7 @@ const processMessage = async (message, { shouldProcessHistoryMsg, ev, creds, key
             case WAProto_1.proto.Message.ProtocolMessage.Type.HISTORY_SYNC_NOTIFICATION:
                 const histNotification = protocolMsg.historySyncNotification;
                 const process = shouldProcessHistoryMsg;
-                const isLatest = !((_c = creds.processedHistoryMessages) === null || _c === void 0 ? void 0 : _c.length);
+                const isLatest = !((_d = creds.processedHistoryMessages) === null || _d === void 0 ? void 0 : _d.length);
                 logger === null || logger === void 0 ? void 0 : logger.info({
                     histNotification,
                     process,
@@ -143,24 +152,24 @@ const processMessage = async (message, { shouldProcessHistoryMsg, ev, creds, key
                             { key: message.key, messageTimestamp: message.messageTimestamp }
                         ]
                     });
-                    const data = await (0, history_1.downloadAndProcessHistorySyncNotification)(histNotification, options);
-                    ev.emit('messaging-history.set', { ...data, isLatest });
+                    const data = yield (0, history_1.downloadAndProcessHistorySyncNotification)(histNotification, options);
+                    ev.emit('messaging-history.set', Object.assign(Object.assign({}, data), { isLatest }));
                 }
                 break;
             case WAProto_1.proto.Message.ProtocolMessage.Type.APP_STATE_SYNC_KEY_SHARE:
                 const keys = protocolMsg.appStateSyncKeyShare.keys;
                 if (keys === null || keys === void 0 ? void 0 : keys.length) {
                     let newAppStateSyncKeyId = '';
-                    await keyStore.transaction(async () => {
+                    yield keyStore.transaction(() => __awaiter(void 0, void 0, void 0, function* () {
                         const newKeys = [];
                         for (const { keyData, keyId } of keys) {
                             const strKeyId = Buffer.from(keyId.keyId).toString('base64');
                             newKeys.push(strKeyId);
-                            await keyStore.set({ 'app-state-sync-key': { [strKeyId]: keyData } });
+                            yield keyStore.set({ 'app-state-sync-key': { [strKeyId]: keyData } });
                             newAppStateSyncKeyId = strKeyId;
                         }
                         logger === null || logger === void 0 ? void 0 : logger.info({ newAppStateSyncKeyId, newKeys }, 'injecting new app state sync keys');
-                    });
+                    }));
                     ev.emit('creds.update', { myAppStateKeyId: newAppStateSyncKeyId });
                 }
                 else {
@@ -170,10 +179,7 @@ const processMessage = async (message, { shouldProcessHistoryMsg, ev, creds, key
             case WAProto_1.proto.Message.ProtocolMessage.Type.REVOKE:
                 ev.emit('messages.update', [
                     {
-                        key: {
-                            ...message.key,
-                            id: protocolMsg.key.id
-                        },
+                        key: Object.assign(Object.assign({}, message.key), { id: protocolMsg.key.id }),
                         update: { message: null, messageStubType: Types_1.WAMessageStubType.REVOKE, key: message.key }
                     }
                 ]);
@@ -202,10 +208,7 @@ const processMessage = async (message, { shouldProcessHistoryMsg, ev, creds, key
         }
     }
     else if (content === null || content === void 0 ? void 0 : content.reactionMessage) {
-        const reaction = {
-            ...content.reactionMessage,
-            key: message.key,
-        };
+        const reaction = Object.assign(Object.assign({}, content.reactionMessage), { key: message.key });
         ev.emit('messages.reaction', [{
                 reaction,
                 key: content.reactionMessage.key,
@@ -218,7 +221,7 @@ const processMessage = async (message, { shouldProcessHistoryMsg, ev, creds, key
         const emitParticipantsUpdate = (action) => (ev.emit('group-participants.update', { id: jid, author: message.participant, participants, action }));
         const emitGroupUpdate = (update) => {
             var _a;
-            ev.emit('groups.update', [{ id: jid, ...update, author: (_a = message.participant) !== null && _a !== void 0 ? _a : undefined }]);
+            ev.emit('groups.update', [Object.assign(Object.assign({ id: jid }, update), { author: (_a = message.participant) !== null && _a !== void 0 ? _a : undefined })]);
         };
         const participantsIncludesMe = () => participants.find(jid => (0, WABinary_1.areJidsSameUser)(meId, jid));
         switch (message.messageStubType) {
@@ -249,28 +252,28 @@ const processMessage = async (message, { shouldProcessHistoryMsg, ev, creds, key
                 emitParticipantsUpdate('promote');
                 break;
             case Types_1.WAMessageStubType.GROUP_CHANGE_ANNOUNCE:
-                const announceValue = (_d = message.messageStubParameters) === null || _d === void 0 ? void 0 : _d[0];
+                const announceValue = (_e = message.messageStubParameters) === null || _e === void 0 ? void 0 : _e[0];
                 emitGroupUpdate({ announce: announceValue === 'true' || announceValue === 'on' });
                 break;
             case Types_1.WAMessageStubType.GROUP_CHANGE_RESTRICT:
-                const restrictValue = (_e = message.messageStubParameters) === null || _e === void 0 ? void 0 : _e[0];
+                const restrictValue = (_f = message.messageStubParameters) === null || _f === void 0 ? void 0 : _f[0];
                 emitGroupUpdate({ restrict: restrictValue === 'true' || restrictValue === 'on' });
                 break;
             case Types_1.WAMessageStubType.GROUP_CHANGE_SUBJECT:
-                const name = (_f = message.messageStubParameters) === null || _f === void 0 ? void 0 : _f[0];
+                const name = (_g = message.messageStubParameters) === null || _g === void 0 ? void 0 : _g[0];
                 chat.name = name;
                 emitGroupUpdate({ subject: name });
                 break;
             case Types_1.WAMessageStubType.GROUP_CHANGE_INVITE_LINK:
-                const code = (_g = message.messageStubParameters) === null || _g === void 0 ? void 0 : _g[0];
+                const code = (_h = message.messageStubParameters) === null || _h === void 0 ? void 0 : _h[0];
                 emitGroupUpdate({ inviteCode: code });
                 break;
             case Types_1.WAMessageStubType.GROUP_MEMBER_ADD_MODE:
-                const memberAddValue = (_h = message.messageStubParameters) === null || _h === void 0 ? void 0 : _h[0];
+                const memberAddValue = (_j = message.messageStubParameters) === null || _j === void 0 ? void 0 : _j[0];
                 emitGroupUpdate({ memberAddMode: memberAddValue === 'all_member_add' });
                 break;
             case Types_1.WAMessageStubType.GROUP_MEMBERSHIP_JOIN_APPROVAL_MODE:
-                const approvalMode = (_j = message.messageStubParameters) === null || _j === void 0 ? void 0 : _j[0];
+                const approvalMode = (_k = message.messageStubParameters) === null || _k === void 0 ? void 0 : _k[0];
                 emitGroupUpdate({ joinApprovalMode: approvalMode === 'on' });
                 break;
         }
@@ -278,12 +281,12 @@ const processMessage = async (message, { shouldProcessHistoryMsg, ev, creds, key
     else if (content === null || content === void 0 ? void 0 : content.pollUpdateMessage) {
         const creationMsgKey = content.pollUpdateMessage.pollCreationMessageKey;
         // we need to fetch the poll creation message to get the poll enc key
-        const pollMsg = await getMessage(creationMsgKey);
+        const pollMsg = yield getMessage(creationMsgKey);
         if (pollMsg) {
             const meIdNormalised = (0, WABinary_1.jidNormalizedUser)(meId);
             const pollCreatorJid = (0, generics_1.getKeyAuthor)(creationMsgKey, meIdNormalised);
             const voterJid = (0, generics_1.getKeyAuthor)(message.key, meIdNormalised);
-            const pollEncKey = (_k = pollMsg.messageContextInfo) === null || _k === void 0 ? void 0 : _k.messageSecret;
+            const pollEncKey = (_l = pollMsg.messageContextInfo) === null || _l === void 0 ? void 0 : _l.messageSecret;
             try {
                 const voteMsg = decryptPollVote(content.pollUpdateMessage.vote, {
                     pollEncKey,
@@ -317,5 +320,5 @@ const processMessage = async (message, { shouldProcessHistoryMsg, ev, creds, key
     if (Object.keys(chat).length > 1) {
         ev.emit('chats.update', [chat]);
     }
-};
+});
 exports.default = processMessage;