torbaileys 8.0.0

package/lib/Utils/noise-handler.js

@@ -0,0 +1,144 @@
+//=======================================================//
+import { aesDecryptGCM, aesEncryptGCM, Curve, hkdf, sha256 } from "./crypto.js";
+import { NOISE_MODE, WA_CERT_DETAILS } from "../Defaults/index.js";
+import { decodeBinaryNode } from "../WABinary/index.js";
+import { proto } from "../../WAProto/index.js";
+import { Boom } from "@hapi/boom";
+//=======================================================//
+const generateIV = (counter) => {
+  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, logger, routingInfo }) => {
+  logger = logger.child({ class: "ns" });
+  const authenticate = (data) => {
+    if (!isFinished) {
+      hash = sha256(Buffer.concat([hash, data]));
+    }
+  };
+  const encrypt = (plaintext) => {
+    const result = aesEncryptGCM(plaintext, encKey, generateIV(writeCounter), hash);
+    writeCounter += 1;
+    authenticate(result);
+    return result;
+  };
+  const decrypt = (ciphertext) => {
+    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 = async (data) => {
+    const key = await hkdf(Buffer.from(data), 64, { salt, info: "" });
+    return [key.slice(0, 32), key.slice(32)];
+  };
+  const mixIntoKey = async (data) => {
+    const [write, read] = await localHKDF(data);
+    salt = write;
+    encKey = read;
+    decKey = read;
+    readCounter = 0;
+    writeCounter = 0;
+  };
+  const finishInit = async () => {
+    const [write, read] = await 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 = 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: async ({ serverHello }, noiseKey) => {
+      authenticate(serverHello.ephemeral);
+      await mixIntoKey(Curve.sharedKey(privateKey, serverHello.ephemeral));
+      const decStaticContent = decrypt(serverHello.static);
+      await mixIntoKey(Curve.sharedKey(privateKey, decStaticContent));
+      const certDecoded = decrypt(serverHello.payload);
+      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);
+      await mixIntoKey(Curve.sharedKey(noiseKey.private, serverHello.ephemeral));
+      return keyEnc;
+    },
+    encodeFrame: (data) => {
+      if (isFinished) {
+        data = encrypt(data);
+      }
+      let header;
+      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: async (newData, onFrame) => {
+      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 = inBytes.slice(3, size + 3);
+        inBytes = inBytes.slice(size + 3);
+        if (isFinished) {
+          const result = decrypt(frame);
+          frame = await decodeBinaryNode(result);
+        }
+        logger.trace({ msg: frame?.attrs?.id }, "recv frame");
+        onFrame(frame);
+        size = getBytesSize();
+      }
+    }
+  };
+};
+//=======================================================//

package/lib/Utils/pre-key-manager.js

@@ -0,0 +1,85 @@
+//=======================================================//
+import PQueue from "p-queue";
+//=======================================================//
+export class PreKeyManager {
+  constructor(store, logger) {
+    this.store = store;
+    this.logger = logger;
+    this.queues = new Map();
+  }
+  getQueue(keyType) {
+    if (!this.queues.has(keyType)) {
+      this.queues.set(keyType, new PQueue({ concurrency: 1 }));
+    }
+    return this.queues.get(keyType);
+  }
+  async processOperations(data, keyType, transactionCache, mutations, isInTransaction) {
+    const keyData = data[keyType];
+    if (!keyData)
+      return;
+    return this.getQueue(keyType).add(async () => {
+      transactionCache[keyType] = transactionCache[keyType] || {};
+      mutations[keyType] = mutations[keyType] || {};
+      const deletions = [];
+      const updates = {};
+      for (const keyId in keyData) {
+        if (keyData[keyId] === null) {
+          deletions.push(keyId);
+        }
+        else {
+          updates[keyId] = keyData[keyId];
+        }
+      }
+      if (Object.keys(updates).length > 0) {
+        Object.assign(transactionCache[keyType], updates);
+        Object.assign(mutations[keyType], updates);
+      }
+      if (deletions.length > 0) {
+        await this.processDeletions(keyType, deletions, transactionCache, mutations, isInTransaction);
+      }
+    });
+  }
+  async processDeletions(keyType, ids, transactionCache, mutations, isInTransaction) {
+    if (isInTransaction) {
+      for (const keyId of ids) {
+        if (transactionCache[keyType]?.[keyId]) {
+          transactionCache[keyType][keyId] = null;
+          mutations[keyType][keyId] = null;
+        }
+        else {
+          this.logger.warn(`Skipping deletion of non-existent ${keyType} in transaction: ${keyId}`);
+        }
+      }
+    }
+    else {
+      const existingKeys = await this.store.get(keyType, ids);
+      for (const keyId of ids) {
+        if (existingKeys[keyId]) {
+          transactionCache[keyType][keyId] = null;
+          mutations[keyType][keyId] = null;
+        }
+        else {
+          this.logger.warn(`Skipping deletion of non-existent ${keyType}: ${keyId}`);
+        }
+      }
+    }
+  }
+  async validateDeletions(data, keyType) {
+    const keyData = data[keyType];
+    if (!keyData)
+      return;
+    return this.getQueue(keyType).add(async () => {
+      const deletionIds = Object.keys(keyData).filter(id => keyData[id] === null);
+      if (deletionIds.length === 0)
+        return;
+      const existingKeys = await this.store.get(keyType, deletionIds);
+      for (const keyId of deletionIds) {
+        if (!existingKeys[keyId]) {
+          this.logger.warn(`Skipping deletion of non-existent ${keyType}: ${keyId}`);
+          delete data[keyType][keyId];
+        }
+      }
+    });
+  }
+}
+//=======================================================//

package/lib/Utils/process-message.js

@@ -0,0 +1,341 @@
+//=======================================================//
+import { areJidsSameUser, isHostedLidUser, isHostedPnUser, isJidBroadcast, isJidStatusBroadcast, jidDecode, jidEncode, jidNormalizedUser } from "../WABinary/index.js";
+import { getContentType, normalizeMessageContent } from "../Utils/messages.js";
+import { downloadAndProcessHistorySyncNotification } from "./history.js";
+import { WAMessageStubType } from "../Types/index.js";
+import { aesDecryptGCM, hmacSign } from "./crypto.js";
+import { proto } from "../../WAProto/index.js";
+import { toNumber } from "./generics.js";
+//=======================================================//
+const REAL_MSG_STUB_TYPES = new Set([
+  WAMessageStubType.CALL_MISSED_GROUP_VIDEO,
+  WAMessageStubType.CALL_MISSED_GROUP_VOICE,
+  WAMessageStubType.CALL_MISSED_VIDEO,
+  WAMessageStubType.CALL_MISSED_VOICE
+]);
+//=======================================================//
+const REAL_MSG_REQ_ME_STUB_TYPES = new Set([WAMessageStubType.GROUP_PARTICIPANT_ADD]);
+//=======================================================//
+export const cleanMessage = (message, meId, meLid) => {
+  if (isHostedPnUser(message.key.remoteJid) || isHostedLidUser(message.key.remoteJid)) {
+    message.key.remoteJid = jidEncode(jidDecode(message.key?.remoteJid)?.user, isHostedPnUser(message.key.remoteJid) ? "s.whatsapp.net" : "lid");
+  }
+  else {
+    message.key.remoteJid = jidNormalizedUser(message.key.remoteJid);
+  }
+  if (isHostedPnUser(message.key.participant) || isHostedLidUser(message.key.participant)) {
+    message.key.participant = jidEncode(jidDecode(message.key.participant)?.user, isHostedPnUser(message.key.participant) ? "s.whatsapp.net" : "lid");
+  }
+  else {
+    message.key.participant = jidNormalizedUser(message.key.participant);
+  }
+  const content = normalizeMessageContent(message.message);
+  if (content?.reactionMessage) {
+    normaliseKey(content.reactionMessage.key);
+  }
+  if (content?.pollUpdateMessage) {
+    normaliseKey(content.pollUpdateMessage.pollCreationMessageKey);
+  }
+  function normaliseKey(msgKey) {
+    if (!message.key.fromMe) {
+      msgKey.fromMe = !msgKey.fromMe
+        ? areJidsSameUser(msgKey.participant || msgKey.remoteJid, meId) ||
+          areJidsSameUser(msgKey.participant || msgKey.remoteJid, meLid)
+        :
+          false;
+      msgKey.remoteJid = message.key.remoteJid;
+      msgKey.participant = msgKey.participant || message.key.participant;
+    }
+  }
+};
+//=======================================================//
+export const isRealMessage = (message) => {
+  const normalizedContent = normalizeMessageContent(message.message);
+  const hasSomeContent = !!getContentType(normalizedContent);
+  return ((!!normalizedContent ||
+    REAL_MSG_STUB_TYPES.has(message.messageStubType) ||
+    REAL_MSG_REQ_ME_STUB_TYPES.has(message.messageStubType)) &&
+    hasSomeContent &&
+    !normalizedContent?.protocolMessage &&
+    !normalizedContent?.reactionMessage &&
+    !normalizedContent?.pollUpdateMessage);
+};
+//=======================================================//
+export const shouldIncrementChatUnread = (message) => !message.key.fromMe && !message.messageStubType;
+//=======================================================//
+export const getChatId = ({ remoteJid, participant, fromMe }) => {
+  if (isJidBroadcast(remoteJid) && !isJidStatusBroadcast(remoteJid) && !fromMe) {
+    return participant;
+  }
+  return remoteJid;
+};
+//=======================================================//
+export function decryptPollVote({ encPayload, encIv }, { pollCreatorJid, pollMsgId, pollEncKey, voterJid }) {
+  const sign = Buffer.concat([
+    toBinary(pollMsgId),
+    toBinary(pollCreatorJid),
+    toBinary(voterJid),
+    toBinary("Poll Vote"),
+    new Uint8Array([1])
+  ]);
+  const key0 = hmacSign(pollEncKey, new Uint8Array(32), "sha256");
+  const decKey = hmacSign(sign, key0, "sha256");
+  const aad = toBinary(`${pollMsgId}\u0000${voterJid}`);
+  const decrypted = aesDecryptGCM(encPayload, decKey, encIv, aad);
+  return proto.Message.PollVoteMessage.decode(decrypted);
+  function toBinary(txt) {
+    return Buffer.from(txt);
+  }
+}
+const processMessage = async (message, { shouldProcessHistoryMsg, placeholderResendCache, ev, creds, signalRepository, keyStore, logger, options }) => {
+  const meId = creds.me.id;
+  const { accountSettings } = creds;
+  const chat = { id: jidNormalizedUser(getChatId(message.key)) };
+  const isRealMsg = isRealMessage(message);
+  if (isRealMsg) {
+    chat.messages = [{ message }];
+    chat.conversationTimestamp = toNumber(message.messageTimestamp);
+    if (shouldIncrementChatUnread(message)) {
+      chat.unreadCount = (chat.unreadCount || 0) + 1;
+    }
+  }
+  const content = normalizeMessageContent(message.message);
+  if ((isRealMsg || content?.reactionMessage?.key?.fromMe) && accountSettings?.unarchiveChats) {
+    chat.archived = false;
+    chat.readOnly = false;
+  }
+  const protocolMsg = content?.protocolMessage;
+  if (protocolMsg) {
+    switch (protocolMsg.type) {
+      case proto.Message.ProtocolMessage.Type.HISTORY_SYNC_NOTIFICATION:
+        const histNotification = protocolMsg.historySyncNotification;
+        const process = shouldProcessHistoryMsg;
+        const isLatest = !creds.processedHistoryMessages?.length;
+        logger?.info({
+          histNotification,
+          process,
+          id: message.key.id,
+          isLatest
+        }, "got history notification");
+        if (process) {
+          if (histNotification.syncType !== proto.HistorySync.HistorySyncType.ON_DEMAND) {
+            ev.emit("creds.update", {
+              processedHistoryMessages: [
+                ...(creds.processedHistoryMessages || []),
+                { key: message.key, messageTimestamp: message.messageTimestamp }
+              ]
+            });
+          }
+          const data = await downloadAndProcessHistorySyncNotification(histNotification, options);
+          ev.emit("messaging-history.set", {
+            ...data,
+            isLatest: histNotification.syncType !== proto.HistorySync.HistorySyncType.ON_DEMAND ? isLatest : undefined,
+            peerDataRequestSessionId: histNotification.peerDataRequestSessionId
+          });
+        }
+        break;
+      case proto.Message.ProtocolMessage.Type.APP_STATE_SYNC_KEY_SHARE:
+        const keys = protocolMsg.appStateSyncKeyShare.keys;
+        if (keys?.length) {
+          let newAppStateSyncKeyId = "";
+          await keyStore.transaction(async () => {
+            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 } });
+              newAppStateSyncKeyId = strKeyId;
+            }
+            logger?.info({ newAppStateSyncKeyId, newKeys }, "injecting new app state sync keys");
+          }, meId);
+          ev.emit("creds.update", { myAppStateKeyId: newAppStateSyncKeyId });
+        }
+        else {
+          logger?.info({ protocolMsg }, "recv app state sync with 0 keys");
+        }
+        break;
+      case proto.Message.ProtocolMessage.Type.REVOKE:
+        ev.emit("messages.update", [
+          {
+            key: {
+              ...message.key,
+              id: protocolMsg.key.id
+            },
+            update: { message: null, messageStubType: WAMessageStubType.REVOKE, key: message.key }
+          }
+        ]);
+        break;
+      case proto.Message.ProtocolMessage.Type.EPHEMERAL_SETTING:
+        Object.assign(chat, {
+          ephemeralSettingTimestamp: toNumber(message.messageTimestamp),
+          ephemeralExpiration: protocolMsg.ephemeralExpiration || null
+        });
+        break;
+      case proto.Message.ProtocolMessage.Type.PEER_DATA_OPERATION_REQUEST_RESPONSE_MESSAGE:
+        const response = protocolMsg.peerDataOperationRequestResponseMessage;
+        if (response) {
+          await placeholderResendCache?.del(response.stanzaId);
+          const { peerDataOperationResult } = response;
+          for (const result of peerDataOperationResult) {
+            const { placeholderMessageResendResponse: retryResponse } = result;
+            if (retryResponse) {
+              const webMessageInfo = proto.WebMessageInfo.decode(retryResponse.webMessageInfoBytes);
+              setTimeout(() => {
+                ev.emit("messages.upsert", {
+                  messages: [webMessageInfo],
+                  type: "notify",
+                  requestId: response.stanzaId
+                });
+              }, 500);
+            }
+          }
+        }
+        break;
+      case proto.Message.ProtocolMessage.Type.MESSAGE_EDIT:
+        ev.emit("messages.update", [
+          {
+            key: { ...message.key, id: protocolMsg.key?.id },
+            update: {
+              message: {
+                editedMessage: {
+                  message: protocolMsg.editedMessage
+                }
+              },
+              messageTimestamp: protocolMsg.timestampMs
+                ? Math.floor(toNumber(protocolMsg.timestampMs) / 1000)
+                : message.messageTimestamp
+            }
+          }
+        ]);
+        break;
+      case proto.Message.ProtocolMessage.Type.LID_MIGRATION_MAPPING_SYNC:
+        const encodedPayload = protocolMsg.lidMigrationMappingSyncMessage?.encodedMappingPayload;
+        const { pnToLidMappings, chatDbMigrationTimestamp } = proto.LIDMigrationMappingSyncPayload.decode(encodedPayload);
+        logger?.debug({ pnToLidMappings, chatDbMigrationTimestamp }, "got lid mappings and chat db migration timestamp");
+        const pairs = [];
+        for (const { pn, latestLid, assignedLid } of pnToLidMappings) {
+          const lid = latestLid || assignedLid;
+          pairs.push({ lid: `${lid}@lid`, pn: `${pn}@s.whatsapp.net` });
+        }
+        await signalRepository.lidMapping.storeLIDPNMappings(pairs);
+        if (pairs.length) {
+          for (const { pn, lid } of pairs) {
+            await signalRepository.migrateSession(pn, lid);
+          }
+        }
+    }
+  }
+  else if (content?.reactionMessage) {
+    const reaction = {
+      ...content.reactionMessage,
+      key: message.key
+    };
+    ev.emit("messages.reaction", [
+      {
+        reaction,
+        key: content.reactionMessage?.key
+      }
+    ]);
+  }
+  else if (message.messageStubType) {
+    const jid = message.key?.remoteJid;
+    let participants;
+    const emitParticipantsUpdate = (action) => ev.emit("group-participants.update", {
+      id: jid,
+      author: message.key.participant,
+      authorPn: message.key.participantAlt,
+      participants,
+      action
+    });
+    const emitGroupUpdate = (update) => {
+      ev.emit("groups.update", [
+        { id: jid, ...update, author: message.key.participant ?? undefined, authorPn: message.key.participantAlt }
+      ]);
+    };
+    const emitGroupRequestJoin = (participant, action, method) => {
+      ev.emit("group.join-request", {
+        id: jid,
+        author: message.key.participant,
+        authorPn: message.key.participantAlt,
+        participant: participant.lid,
+        participantPn: participant.pn,
+        action,
+        method: method
+      });
+    };
+    const participantsIncludesMe = () => participants.find(jid => areJidsSameUser(meId, jid.phoneNumber));
+    switch (message.messageStubType) {
+      case WAMessageStubType.GROUP_PARTICIPANT_CHANGE_NUMBER:
+        participants = message.messageStubParameters.map((a) => JSON.parse(a)) || [];
+        emitParticipantsUpdate("modify");
+        break;
+      case WAMessageStubType.GROUP_PARTICIPANT_LEAVE:
+      case WAMessageStubType.GROUP_PARTICIPANT_REMOVE:
+        participants = message.messageStubParameters.map((a) => JSON.parse(a)) || [];
+        emitParticipantsUpdate("remove");
+        if (participantsIncludesMe()) {
+          chat.readOnly = true;
+        }
+        break;
+      case WAMessageStubType.GROUP_PARTICIPANT_ADD:
+      case WAMessageStubType.GROUP_PARTICIPANT_INVITE:
+      case WAMessageStubType.GROUP_PARTICIPANT_ADD_REQUEST_JOIN:
+        participants = message.messageStubParameters.map((a) => JSON.parse(a)) || [];
+        if (participantsIncludesMe()) {
+          chat.readOnly = false;
+        }
+        emitParticipantsUpdate("add");
+        break;
+      case WAMessageStubType.GROUP_PARTICIPANT_DEMOTE:
+        participants = message.messageStubParameters.map((a) => JSON.parse(a)) || [];
+        emitParticipantsUpdate("demote");
+        break;
+      case WAMessageStubType.GROUP_PARTICIPANT_PROMOTE:
+        participants = message.messageStubParameters.map((a) => JSON.parse(a)) || [];
+        emitParticipantsUpdate("promote");
+        break;
+      case WAMessageStubType.GROUP_CHANGE_ANNOUNCE:
+        const announceValue = message.messageStubParameters?.[0];
+        emitGroupUpdate({ announce: announceValue === "true" || announceValue === "on" });
+        break;
+      case WAMessageStubType.GROUP_CHANGE_RESTRICT:
+        const restrictValue = message.messageStubParameters?.[0];
+        emitGroupUpdate({ restrict: restrictValue === "true" || restrictValue === "on" });
+        break;
+      case WAMessageStubType.GROUP_CHANGE_SUBJECT:
+        const name = message.messageStubParameters?.[0];
+        chat.name = name;
+        emitGroupUpdate({ subject: name });
+        break;
+      case WAMessageStubType.GROUP_CHANGE_DESCRIPTION:
+        const description = message.messageStubParameters?.[0];
+        chat.description = description;
+        emitGroupUpdate({ desc: description });
+        break;
+      case WAMessageStubType.GROUP_CHANGE_INVITE_LINK:
+        const code = message.messageStubParameters?.[0];
+        emitGroupUpdate({ inviteCode: code });
+        break;
+      case WAMessageStubType.GROUP_MEMBER_ADD_MODE:
+        const memberAddValue = message.messageStubParameters?.[0];
+        emitGroupUpdate({ memberAddMode: memberAddValue === "all_member_add" });
+        break;
+      case WAMessageStubType.GROUP_MEMBERSHIP_JOIN_APPROVAL_MODE:
+        const approvalMode = message.messageStubParameters?.[0];
+        emitGroupUpdate({ joinApprovalMode: approvalMode === "on" });
+        break;
+      case WAMessageStubType.GROUP_MEMBERSHIP_JOIN_APPROVAL_REQUEST_NON_ADMIN_ADD:
+        const participant = JSON.parse(message.messageStubParameters?.[0]);
+        const action = message.messageStubParameters?.[1];
+        const method = message.messageStubParameters?.[2];
+        emitGroupRequestJoin(participant, action, method);
+        break;
+    }
+  }
+  if (Object.keys(chat).length > 1) {
+    ev.emit("chats.update", [chat]);
+  }
+};
+//=======================================================//
+export default processMessage;
+//=======================================================//