feat: add skipMessageWhitelistCheck() in MessageBuilder

TICKET: SC-2419

Author: bhavidhingra

modules/abstract-eth/src/lib/messages/eip191/eip191MessageBuilder.ts

@@ -14,6 +14,10 @@ export class Eip191MessageBuilder extends BaseMessageBuilder {
     super(_coinConfig, MessageStandardType.EIP191);
   }
 
+  public skipMessageWhitelistCheck(): boolean {
+    return true;
+  }
+
   /**
    * Builds an EIP-191 message instance with the provided options
    * @param options Options to create the message

modules/account-lib/src/index.ts

@@ -206,8 +206,8 @@ export { Vet };
 import * as CosmosSharedCoin from '@bitgo/sdk-coin-cosmos';
 export { CosmosSharedCoin };
 
-import { validateAgainstMessageTemplates, MIDNIGHT_TNC_HASH } from './utils';
-export { MIDNIGHT_TNC_HASH };
+import { isMessageWhitelisted, MIDNIGHT_TNC_HASH } from './utils';
+export { isMessageWhitelisted, MIDNIGHT_TNC_HASH };
 
 const coinBuilderMap = {
   trx: Trx.WrappedBuilder,
@@ -435,7 +435,7 @@ export async function verifyMessage(
     if (!isValidMessageEncoded) {
       return false;
     }
-    return validateAgainstMessageTemplates(messageRaw);
+    return isMessageWhitelisted(messageRaw);
   } catch (e) {
     console.error(`Error verifying message for coin ${coinName}:`, e);
     return false;

modules/account-lib/src/utils/messages/index.ts

@@ -20,7 +20,7 @@ const whitelistedMessageTemplates = [MIDNIGHT_GLACIER_DROP_CLAIM_MESSAGE_TEMPLAT
  * @param {string} messageRaw - The raw message to validate.
  * @returns {boolean} - Returns true if the message matches any of the whitelisted templates, false otherwise.
  */
-export function validateAgainstMessageTemplates(messageRaw: string): boolean {
+export function isMessageWhitelisted(messageRaw: string): boolean {
   return whitelistedMessageTemplates.some((template) => {
     const regex = new RegExp(`^${template}$`, 's'); // 's' flag to match newlines
     return regex.test(messageRaw);

modules/account-lib/test/unit/utils/messages/index.ts

@@ -1,32 +1,32 @@
 import should from 'should';
-import { MIDNIGHT_TNC_HASH, validateAgainstMessageTemplates } from '../../../../src/utils';
+import { MIDNIGHT_TNC_HASH, isMessageWhitelisted } from '../../../../src/utils';
 
 describe('Message validation', () => {
-  describe('validateAgainstMessageTemplates', () => {
+  describe('isMessageWhitelisted', () => {
     const adaTestnetDestinationAddress = 'addr_test1vz7xs7ceu4xx9n5xn57lfe86vrwddqpp77vjwq5ptlkh49cqy3wur';
     const adaMainnetDestinationAddress =
       'addr1q9k6u7lhf467y2f8skr2dafldx2npsd8fymq0mslnj0t44nd4ealwnt4ug5j0pvx5m6n76v4xrq6wjfkqlhpl8y7httq2m9cmu';
 
     it('should validate testnet message matching the Midnight glacier drop claim template', () => {
       const messageRaw = `STAR 100 to ${adaTestnetDestinationAddress} ${MIDNIGHT_TNC_HASH}`;
 
-      const result = validateAgainstMessageTemplates(messageRaw);
+      const result = isMessageWhitelisted(messageRaw);
 
       should.equal(result, true);
     });
 
     it('should validate mainnet message matching the Midnight glacier drop claim template', () => {
       const messageRaw = `STAR 100 to ${adaMainnetDestinationAddress} ${MIDNIGHT_TNC_HASH}`;
 
-      const result = validateAgainstMessageTemplates(messageRaw);
+      const result = isMessageWhitelisted(messageRaw);
 
       should.equal(result, true);
     });
 
     it('should not validate message with incorrect format', () => {
       const messageRaw = `INCORRECT 100 to ${adaTestnetDestinationAddress} ${MIDNIGHT_TNC_HASH}`;
 
-      const result = validateAgainstMessageTemplates(messageRaw);
+      const result = isMessageWhitelisted(messageRaw);
 
       should.equal(result, false);
     });
@@ -35,7 +35,7 @@ describe('Message validation', () => {
       // Missing "to addr" part
       const messageRaw = `STAR 100 ${MIDNIGHT_TNC_HASH}`;
 
-      const result = validateAgainstMessageTemplates(messageRaw);
+      const result = isMessageWhitelisted(messageRaw);
 
       should.equal(result, false);
     });
@@ -46,21 +46,21 @@ describe('Message validation', () => {
         '5af1adf825baa496729e2eac1e895ebc77973744bce67f44276bf6006f5c21de863ed121e11828d8fc0241773191e26dc1134803a681a9a98ba0ae812553db24';
       const messageRaw = `STAR 100 to ${adaTestnetDestinationAddress} ${incorrectHash}`;
 
-      const result = validateAgainstMessageTemplates(messageRaw);
+      const result = isMessageWhitelisted(messageRaw);
 
       should.equal(result, false);
     });
 
     it('should handle empty message', () => {
-      const result = validateAgainstMessageTemplates('');
+      const result = isMessageWhitelisted('');
 
       should.equal(result, false);
     });
 
     it('should not validate message with special regex characters', () => {
       const messageRaw = `STAR shade.with+special*chars to addr.with[special]chars ${MIDNIGHT_TNC_HASH}`;
 
-      const result = validateAgainstMessageTemplates(messageRaw);
+      const result = isMessageWhitelisted(messageRaw);
 
       should.equal(result, false);
     });

modules/sdk-coin-ada/src/lib/messages/cip8/cip8MessageBuilder.ts

@@ -6,14 +6,6 @@ import { BaseMessageBuilder, IMessage, MessageOptions, MessageStandardType } fro
  * Builder for CIP-8 messages
  */
 export class Cip8MessageBuilder extends BaseMessageBuilder {
-  private static readonly MIDNIGHT_TNC_HASH = '31a6bab50a84b8439adcfb786bb2020f6807e6e8fda629b424110fc7bb1c6b8b';
-
-  /*
-   * matches a message that starts with "STAR ", followed by a number,
-   * then " to addr" or " to addr_test1", followed by a 50+ character alphanumeric address,
-   * and ends with the midnight TnC hash
-   */
-  private static readonly MIDNIGHT_GLACIER_DROP_CLAIM_MESSAGE_TEMPLATE = `STAR \\d+ to addr(?:1|_test1)[a-z0-9]{50,} ${Cip8MessageBuilder.MIDNIGHT_TNC_HASH}`;
   /**
    * Base constructor.
    * @param _coinConfig BaseCoin from statics library
@@ -30,15 +22,4 @@ export class Cip8MessageBuilder extends BaseMessageBuilder {
   async buildMessage(options: MessageOptions): Promise<IMessage> {
     return new Cip8Message(options);
   }
-
-  /**
-   * Validates the signable payload
-   * @param message The message to validate
-   * @returns A boolean indicating whether the signable payload is valid
-   */
-  public validateSignablePayload(message: string | Buffer): boolean {
-    const messageStr = Buffer.isBuffer(message) ? message.toString('utf8') : message;
-    const regex = new RegExp(`^${Cip8MessageBuilder.MIDNIGHT_GLACIER_DROP_CLAIM_MESSAGE_TEMPLATE}$`, 's');
-    return regex.test(messageStr);
-  }
 }

modules/sdk-core/src/account-lib/baseCoin/messages/baseMessageBuilder.ts

@@ -118,13 +118,8 @@ export abstract class BaseMessageBuilder implements IMessageBuilder {
     return this;
   }
 
-  /**
-   * Validates the signable payload
-   * @param message The message to validate
-   * @returns A boolean indicating whether the signable payload is valid
-   */
-  public validateSignablePayload(message: string | Buffer): boolean {
-    return true;
+  public skipMessageWhitelistCheck(): boolean {
+    return false;
   }
 
   /**

modules/sdk-core/src/account-lib/baseCoin/messages/iface.ts

@@ -137,11 +137,9 @@ export interface IMessageBuilder {
   getPayload(): MessagePayload | undefined;
 
   /**
-   * Validates the signable payload
-   * @param message The message to validate
-   * @returns A boolean indicating whether the signable payload is valid
+   * Checks if the message builder should skip the whitelist check for messages
    */
-  validateSignablePayload(message: string | Buffer): boolean;
+  skipMessageWhitelistCheck(): boolean;
 
   /**
    * Gets the current metadata