socketon 0.31.0 → 1.51.16

package/lib/Utils/decode-wa-message.js

@@ -1,198 +1,267 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.decryptMessageNode = exports.decodeMessageNode = void 0;
-const boom_1 = require("@hapi/boom");
-const WAProto_1 = require("../../WAProto");
-const WABinary_1 = require("../WABinary");
-const generics_1 = require("./generics");
-const NO_MESSAGE_FOUND_ERROR_TEXT = 'Message absent from node';
-/**
- * Decode the received node as a message.
- * @note this will only parse the message, not decrypt it
- */
-function decodeMessageNode(stanza, meId, meLid) {
-    var _a, _b;
-    let msgType;
-    let chatId;
-    let author;
-    const msgId = stanza.attrs.id;
-    const from = stanza.attrs.from;
-    const participant = stanza.attrs.participant;
-    const recipient = stanza.attrs.recipient;
-    const isMe = (jid) => (0, WABinary_1.areJidsSameUser)(jid, meId);
-    const isMeLid = (jid) => (0, WABinary_1.areJidsSameUser)(jid, meLid);
-    if ((0, WABinary_1.isJidUser)(from)) {
-        if (recipient) {
-            if (!isMe(from)) {
-                throw new boom_1.Boom('receipient present, but msg not from me', { data: stanza });
-            }
-            chatId = recipient;
-        }
-        else {
-            chatId = from;
-        }
-        msgType = 'chat';
-        author = from;
+//=======================================================//
+import { areJidsSameUser, isHostedLidUser, isHostedPnUser, isJidBroadcast, isJidGroup, isJidMetaAI, isJidNewsletter, isJidStatusBroadcast, isLidUser, isPnUser } from "../WABinary/index.js";
+import { unpadRandomMax16 } from "./generics.js";
+import { proto } from "../../WAProto/index.js";
+import { Boom } from "@hapi/boom";
+//=======================================================//
+export const getDecryptionJid = async (sender, repository) => {
+  if (isLidUser(sender) || isHostedLidUser(sender)) {
+    return sender;
+  }
+  const mapped = await repository.lidMapping.getLIDForPN(sender);
+  return mapped || sender;
+};
+//=======================================================//
+const storeMappingFromEnvelope = async (stanza, sender, repository, decryptionJid, logger) => {
+  const { senderAlt } = extractAddressingContext(stanza);
+  if (senderAlt && isLidUser(senderAlt) && isPnUser(sender) && decryptionJid === sender) {
+    try {
+      await repository.lidMapping.storeLIDPNMappings([{ lid: senderAlt, pn: sender }]);
+      await repository.migrateSession(sender, senderAlt);
+      logger.debug({ sender, senderAlt }, "Stored LID mapping from envelope");
     }
-    else if ((0, WABinary_1.isLidUser)(from)) {
-        if (recipient) {
-            if (!isMeLid(from)) {
-                throw new boom_1.Boom('receipient present, but msg not from me', { data: stanza });
-            }
-            chatId = recipient;
-        }
-        else {
-            chatId = from;
-        }
-        msgType = 'chat';
-        author = from;
+    catch (error) {
+      logger.warn({ sender, senderAlt, error }, "Failed to store LID mapping");
     }
-    else if ((0, WABinary_1.isJidGroup)(from)) {
-        if (!participant) {
-            throw new boom_1.Boom('No participant in group message');
-        }
-        msgType = 'group';
-        author = participant;
-        chatId = from;
+  }
+};
+//=======================================================//
+export const NO_MESSAGE_FOUND_ERROR_TEXT = "Message absent from node";
+export const MISSING_KEYS_ERROR_TEXT = "Key used already or never filled";
+//=======================================================//
+export const DECRYPTION_RETRY_CONFIG = {
+  maxRetries: 3,
+  baseDelayMs: 100,
+  sessionRecordErrors: ["No session record", "SessionError: No session record"]
+};
+//=======================================================//
+export const NACK_REASONS = {
+  ParsingError: 487,
+  UnrecognizedStanza: 488,
+  UnrecognizedStanzaClass: 489,
+  UnrecognizedStanzaType: 490,
+  InvalidProtobuf: 491,
+  InvalidHostedCompanionStanza: 493,
+  MissingMessageSecret: 495,
+  SignalErrorOldCounter: 496,
+  MessageDeletedOnPeer: 499,
+  UnhandledError: 500,
+  UnsupportedAdminRevoke: 550,
+  UnsupportedLIDGroup: 551,
+  DBOperationFailed: 552
+};
+//=======================================================//
+export const extractAddressingContext = (stanza) => {
+  let senderAlt;
+  let recipientAlt;
+  const sender = stanza.attrs.participant || stanza.attrs.from;
+  const addressingMode = stanza.attrs.addressing_mode || (sender?.endsWith("lid") ? "lid" : "pn");
+  if (addressingMode === "lid") {
+    senderAlt = stanza.attrs.participant_pn || stanza.attrs.sender_pn || stanza.attrs.peer_recipient_pn;
+    recipientAlt = stanza.attrs.recipient_pn;
+  }
+  else {
+    senderAlt = stanza.attrs.participant_lid || stanza.attrs.sender_lid || stanza.attrs.peer_recipient_lid;
+    recipientAlt = stanza.attrs.recipient_lid;
+  }
+  return {
+    addressingMode,
+    senderAlt,
+    recipientAlt
+  };
+};
+//=======================================================//
+export function decodeMessageNode(stanza, meId, meLid) {
+  let msgType;
+  let chatId;
+  let author;
+  let fromMe = false;
+  const msgId = stanza.attrs.id;
+  const from = stanza.attrs.from;
+  const participant = stanza.attrs.participant;
+  const recipient = stanza.attrs.recipient;
+  const addressingContext = extractAddressingContext(stanza);
+  const isMe = (jid) => areJidsSameUser(jid, meId);
+  const isMeLid = (jid) => areJidsSameUser(jid, meLid);
+  if (isPnUser(from) || isLidUser(from) || isHostedLidUser(from) || isHostedPnUser(from)) {
+    if (recipient && !isJidMetaAI(recipient)) {
+      if (!isMe(from) && !isMeLid(from)) {
+        throw new Boom("receipient present, but msg not from me", { data: stanza });
+      }
+      if (isMe(from) || isMeLid(from)) {
+        fromMe = true;
+      }
+      chatId = recipient;
     }
-    else if ((0, WABinary_1.isJidBroadcast)(from)) {
-        if (!participant) {
-            throw new boom_1.Boom('No participant in group message');
-        }
-        const isParticipantMe = isMe(participant);
-        if ((0, WABinary_1.isJidStatusBroadcast)(from)) {
-            msgType = isParticipantMe ? 'direct_peer_status' : 'other_status';
-        }
-        else {
-            msgType = isParticipantMe ? 'peer_broadcast' : 'other_broadcast';
-        }
-        chatId = from;
-        author = participant;
+    else {
+      chatId = from;
     }
-    else if ((0, WABinary_1.isJidNewsLetter)(from)) {
-        msgType = 'newsletter';
-        author = from;
-        chatId = from;
+    msgType = "chat";
+    author = from;
+  }
+  else if (isJidGroup(from)) {
+    if (!participant) {
+      throw new Boom("No participant in group message");
     }
-    else {
-        throw new boom_1.Boom('Unknown message type', { data: stanza });
+    if (isMe(participant) || isMeLid(participant)) {
+      fromMe = true;
     }
-    const fromMe = (0, WABinary_1.isJidNewsLetter)(from) ? !!((_a = stanza.attrs) === null || _a === void 0 ? void 0 : _a.is_sender) : ((0, WABinary_1.isLidUser)(from) ? isMeLid : isMe)(stanza.attrs.participant || stanza.attrs.from);
-    const pushname = stanza.attrs.notify;
-    const key = {
-        remoteJid: chatId,
-        fromMe,
-        id: msgId,
-        participant
-    };
-    const fullMessage = {
-        key,
-        messageTimestamp: +stanza.attrs.t,
-        pushName: pushname,
-        broadcast: (0, WABinary_1.isJidBroadcast)(from)
-    };
-    if (msgType === 'newsletter') {
-        fullMessage.newsletterServerId = +((_b = stanza.attrs) === null || _b === void 0 ? void 0 : _b.server_id);
+    msgType = "group";
+    author = participant;
+    chatId = from;
+  }
+  else if (isJidBroadcast(from)) {
+    if (!participant) {
+      throw new Boom("No participant in group message");
     }
-    if (key.fromMe) {
-        fullMessage.status = WAProto_1.proto.WebMessageInfo.Status.SERVER_ACK;
+    const isParticipantMe = isMe(participant);
+    if (isJidStatusBroadcast(from)) {
+      msgType = isParticipantMe ? "direct_peer_status" : "other_status";
     }
-    return {
-        fullMessage,
-        author,
-        sender: msgType === 'chat' ? author : chatId
-    };
+    else {
+      msgType = isParticipantMe ? "peer_broadcast" : "other_broadcast";
+    }
+    fromMe = isParticipantMe;
+    chatId = from;
+    author = participant;
+  }
+  else if (isJidNewsletter(from)) {
+    msgType = "newsletter";
+    chatId = from;
+    author = from;
+    if (isMe(from) || isMeLid(from)) {
+      fromMe = true;
+    }
+  }
+  else {
+    throw new Boom("Unknown message type", { data: stanza });
+  }
+  const pushname = stanza?.attrs?.notify;
+  const key = {
+    remoteJid: chatId,
+    remoteJidAlt: !isJidGroup(chatId) ? addressingContext.senderAlt : undefined,
+    fromMe,
+    id: msgId,
+    participant,
+    participantAlt: isJidGroup(chatId) ? addressingContext.senderAlt : undefined,
+    addressingMode: addressingContext.addressingMode,
+    ...(msgType === "newsletter" && stanza.attrs.server_id ? { server_id: stanza.attrs.server_id } : {})
+  };
+  const fullMessage = {
+    key,
+    messageTimestamp: +stanza.attrs.t,
+    pushName: pushname,
+    broadcast: isJidBroadcast(from)
+  };
+  if (key.fromMe) {
+    fullMessage.status = proto.WebMessageInfo.Status.SERVER_ACK;
+  }
+  return {
+    fullMessage,
+    author,
+    sender: msgType === "chat" ? author : chatId
+  };
 }
-exports.decodeMessageNode = decodeMessageNode;
-const decryptMessageNode = (stanza, meId, meLid, repository, logger) => {
-    const { fullMessage, author, sender } = decodeMessageNode(stanza, meId, meLid);
-    return {
-        fullMessage,
-        category: stanza.attrs.category,
-        author,
-        async decrypt() {
-            var _a;
-            let decryptables = 0;
-            async function processSenderKeyDistribution(msg) {
-                if (msg.senderKeyDistributionMessage) {
-                    try {
-                        await repository.processSenderKeyDistributionMessage({
-                            authorJid: author,
-                            item: msg.senderKeyDistributionMessage
-                        });
-                    }
-                    catch (err) {
-                        logger.error({ key: fullMessage.key, err }, 'failed to process senderKeyDistribution');
-                    }
-                }
+//=======================================================//
+export const decryptMessageNode = (stanza, meId, meLid, repository, logger) => {
+  const { fullMessage, author, sender } = decodeMessageNode(stanza, meId, meLid);
+  return {
+    fullMessage,
+    category: stanza.attrs.category,
+    author,
+    async decrypt() {
+      let decryptables = 0;
+      if (Array.isArray(stanza.content)) {
+        for (const { tag, attrs, content } of stanza.content) {
+          if (tag === "verified_name" && content instanceof Uint8Array) {
+            const cert = proto.VerifiedNameCertificate.decode(content);
+            const details = proto.VerifiedNameCertificate.Details.decode(cert.details);
+            fullMessage.verifiedBizName = details.verifiedName;
+          }
+          if (tag === "unavailable" && attrs.type === "view_once") {
+            fullMessage.key.isViewOnce = true;
+          }
+          if (tag !== "enc" && tag !== "plaintext") {
+            continue;
+          }
+          if (!(content instanceof Uint8Array)) {
+            continue;
+          }
+          decryptables += 1;
+          let msgBuffer;
+          const decryptionJid = await getDecryptionJid(author, repository);
+          if (tag !== "plaintext") {
+            await storeMappingFromEnvelope(stanza, author, repository, decryptionJid, logger);
+          }
+          try {
+            const e2eType = tag === "plaintext" ? "plaintext" : attrs.type;
+            switch (e2eType) {
+              case "skmsg":
+                msgBuffer = await repository.decryptGroupMessage({
+                  group: sender,
+                  authorJid: author,
+                  msg: content
+                });
+                break;
+              case "pkmsg":
+              case "msg":
+                msgBuffer = await repository.decryptMessage({
+                  jid: decryptionJid,
+                  type: e2eType,
+                  ciphertext: content
+                });
+                break;
+              case "plaintext":
+                msgBuffer = content;
+                break;
+              default:
+                throw new Error(`Unknown e2e type: ${e2eType}`);
             }
-            if ((0, WABinary_1.isJidNewsLetter)(fullMessage.key.remoteJid)) {
-                const node = (0, WABinary_1.getBinaryNodeChild)(stanza, 'plaintext');
-                const msg = WAProto_1.proto.Message.decode(node === null || node === void 0 ? void 0 : node.content);
-                await processSenderKeyDistribution(msg);
-                fullMessage.message = msg;
-                decryptables += 1;
+            let msg = proto.Message.decode(e2eType !== "plaintext" ? unpadRandomMax16(msgBuffer) : msgBuffer);
+            msg = msg.deviceSentMessage?.message || msg;
+            if (msg.senderKeyDistributionMessage) {
+              try {
+                await repository.processSenderKeyDistributionMessage({
+                  authorJid: author,
+                  item: msg.senderKeyDistributionMessage
+                });
+              }
+              catch (err) {
+                logger.error({ key: fullMessage.key, err }, "failed to process sender key distribution message");
+              }
             }
-            else if (Array.isArray(stanza.content)) {
-                for (const { tag, attrs, content } of stanza.content) {
-                    if (tag === 'verified_name' && content instanceof Uint8Array) {
-                        const cert = WAProto_1.proto.VerifiedNameCertificate.decode(content);
-                        const details = WAProto_1.proto.VerifiedNameCertificate.Details.decode(cert.details);
-                        fullMessage.verifiedBizName = details.verifiedName;
-                    }
-                    if (tag !== 'enc') {
-                        continue;
-                    }
-                    if (!(content instanceof Uint8Array)) {
-                        continue;
-                    }
-                    decryptables += 1;
-                    let msgBuffer;
-                    try {
-                        const e2eType = attrs.type;
-                        switch (e2eType) {
-                            case 'skmsg':
-                                msgBuffer = await repository.decryptGroupMessage({
-                                    group: sender,
-                                    authorJid: author,
-                                    msg: content
-                                });
-                                break;
-                            case 'pkmsg':
-                            case 'msg':
-                                const user = (0, WABinary_1.isJidUser)(sender) ? sender : author;
-                                msgBuffer = await repository.decryptMessage({
-                                    jid: user,
-                                    type: e2eType,
-                                    ciphertext: content
-                                });
-                                break;
-                            default:
-                                throw new Error(`Unknown e2e type: ${e2eType}`);
-                        }
-                        let msg = WAProto_1.proto.Message.decode((0, generics_1.unpadRandomMax16)(msgBuffer));
-                        msg = ((_a = msg.deviceSentMessage) === null || _a === void 0 ? void 0 : _a.message) || msg;
-                        await processSenderKeyDistribution(msg);
-                        if (fullMessage.message) {
-                            Object.assign(fullMessage.message, msg);
-                        }
-                        else {
-                            fullMessage.message = msg;
-                        }
-                    }
-                    catch (err) {
-                        logger.error({ key: fullMessage.key, err }, 'failed to decrypt message');
-                        fullMessage.messageStubType = WAProto_1.proto.WebMessageInfo.StubType.CIPHERTEXT;
-                        fullMessage.messageStubParameters = [err.message];
-                    }
-                }
+            if (fullMessage.message) {
+              Object.assign(fullMessage.message, msg);
             }
-            // if nothing was found to decrypt
-            if (!decryptables) {
-                fullMessage.messageStubType = WAProto_1.proto.WebMessageInfo.StubType.CIPHERTEXT;
-                fullMessage.messageStubParameters = [NO_MESSAGE_FOUND_ERROR_TEXT, JSON.stringify(stanza, generics_1.BufferJSON.replacer)];
+            else {
+              fullMessage.message = msg;
             }
+          }
+          catch (err) {
+            const errorContext = {
+              key: fullMessage.key,
+              err,
+              messageType: tag === "plaintext" ? "plaintext" : attrs.type,
+              sender,
+              author,
+              isSessionRecordError: isSessionRecordError(err)
+            };
+            logger.error(errorContext, "failed to decrypt message");
+            fullMessage.messageStubType = proto.WebMessageInfo.StubType.CIPHERTEXT;
+            fullMessage.messageStubParameters = [err.message.toString()];
+          }
         }
-    };
+      }
+      if (!decryptables) {
+        fullMessage.messageStubType = proto.WebMessageInfo.StubType.CIPHERTEXT;
+        fullMessage.messageStubParameters = [NO_MESSAGE_FOUND_ERROR_TEXT];
+      }
+    }
+  };
 };
-exports.decryptMessageNode = decryptMessageNode;
+//=======================================================//
+function isSessionRecordError(error) {
+  const errorMessage = error?.message || error?.toString() || "";
+  return DECRYPTION_RETRY_CONFIG.sessionRecordErrors.some(errorPattern => errorMessage.includes(errorPattern));
+}
+//=======================================================//