test: cover CCTP replay rejection in cross-chain mock

Author: buy doge

contracts/contracts/mocks/crosschain/CCTPMessageTransmitterMock.sol

@@ -44,6 +44,8 @@ contract CCTPMessageTransmitterMock is ICCTPMessageTransmitter {
     Message[] public messages;
     // map of encoded messages to the corresponding message structs
     mapping(bytes32 => Message) public encodedMessages;
+    // tracks relayed messages to mirror CCTP replay protection in tests
+    mapping(bytes32 => bool) public processedMessages;
 
     constructor(address _usdc) {
         usdc = IERC20(_usdc);
@@ -120,7 +122,11 @@ contract CCTPMessageTransmitterMock is ICCTPMessageTransmitter {
         override
         returns (bool)
     {
-        Message memory storedMsg = encodedMessages[keccak256(message)];
+        bytes32 messageHash = keccak256(message);
+        require(!processedMessages[messageHash], "Message already processed");
+        processedMessages[messageHash] = true;
+
+        Message memory storedMsg = encodedMessages[messageHash];
         AbstractCCTPIntegrator recipient = AbstractCCTPIntegrator(
             address(uint160(uint256(storedMsg.recipient)))
         );

contracts/test/strategies/crosschain/cross-chain-strategy.js

@@ -7,7 +7,10 @@ const {
 const { setERC20TokenBalance } = require("../../_fund");
 const { units, usdcUnits } = require("../../helpers");
 const { impersonateAndFund } = require("../../../utils/signers");
-const { encodeBalanceCheckMessageBody } = require("./_crosschain-helpers");
+const {
+  encodeBalanceCheckMessageBody,
+  encodeCCTPMessage,
+} = require("./_crosschain-helpers");
 
 const loadFixture = createFixtureLoader(crossChainFixtureUnit);
 const DAY_IN_SECONDS = 86400;
@@ -456,6 +459,39 @@ describe("ForkTest: CrossChainRemoteStrategy", function () {
     await assertVaultTotalValue("1000");
   });
 
+  it("Should reject direct replay of an already-relayed balance update message", async function () {
+    const { messageTransmitter } = fixture;
+    await sendBalanceUpdateToMaster();
+
+    const nonce = await crossChainMasterStrategy.lastTransferNonce();
+    const balance = await crossChainRemoteStrategy.checkBalance(usdc.address);
+    const latestBlock = await ethers.provider.getBlock("latest");
+    const body = encodeBalanceCheckMessageBody(
+      nonce,
+      balance,
+      false,
+      latestBlock.timestamp
+    );
+    const encodedMessage = encodeCCTPMessage(
+      await crossChainMasterStrategy.peerDomainID(),
+      crossChainRemoteStrategy.address,
+      crossChainMasterStrategy.address,
+      body
+    );
+
+    await expect(messageTransmitter.processFront())
+      .to.emit(crossChainMasterStrategy, "RemoteStrategyBalanceUpdated")
+      .withArgs(balance);
+    await expect(await messageTransmitter.messagesInQueue()).to.eq(0);
+
+    const transmitterSigner = await impersonateAndFund(messageTransmitter.address);
+    await expect(
+      crossChainMasterStrategy
+        .connect(transmitterSigner)
+        .relay(encodedMessage, "0x")
+    ).to.be.revertedWith("Message already processed");
+  });
+
   it("Should emit a BalanceCheckIgnored event if balance update message is too old", async function () {
     const { messageTransmitter, crossChainMasterStrategy } = fixture;
     await sendBalanceUpdateToMaster();