Continuously send messages to create load (#118)

* save

* Continuously send messages to create load

* add metrics

Author: JoeHowarth

ethereum/contracts/mock/MockRelayerIntegration.sol

@@ -10,6 +10,13 @@ import "../interfaces/IWormholeReceiver.sol";
 
 import "forge-std/console.sol";
 
+interface Structs {
+    struct XAddress {
+        uint16 chainId;
+        bytes32 addr;
+    }
+}
+
 contract MockRelayerIntegration is IWormholeReceiver {
     using BytesLib for bytes;
 
@@ -28,7 +35,8 @@ contract MockRelayerIntegration is IWormholeReceiver {
     // mapping of other MockRelayerIntegration contracts
     mapping(uint16 => bytes32) registeredContracts;
 
-    bytes[] messages;
+    // bytes[] messages;
+    bytes[][] messageHistory;
 
     struct FurtherInstructions {
         bool keepSending;
@@ -139,14 +147,17 @@ contract MockRelayerIntegration is IWormholeReceiver {
     function receiveWormholeMessages(bytes[] memory wormholeObservations, bytes[] memory) public payable override {
         // loop through the array of wormhole observations from the batch and store each payload
         uint256 numObservations = wormholeObservations.length;
-        messages = new bytes[](wormholeObservations.length - 2);
+        bytes[] memory messages = new bytes[](wormholeObservations.length - 2);
+        uint16 emitterChainId;
         for (uint256 i = 0; i < numObservations - 2; i++) {
             (IWormhole.VM memory parsed, bool valid, string memory reason) =
                 wormhole.parseAndVerifyVM(wormholeObservations[i]);
             require(valid, reason);
             require(registeredContracts[parsed.emitterChainId] == parsed.emitterAddress);
+            emitterChainId = parsed.emitterChainId;
             messages[i] = parsed.payload;
         }
+        messageHistory.push(messages);
 
         (IWormhole.VM memory parsed, bool valid, string memory reason) =
             wormhole.parseAndVerifyVM(wormholeObservations[wormholeObservations.length - 2]);
@@ -183,14 +194,21 @@ contract MockRelayerIntegration is IWormholeReceiver {
     }
 
     function getMessage() public view returns (bytes memory) {
-        if (messages.length == 0) {
+        if (messageHistory.length == 0 || messageHistory[messageHistory.length - 1].length == 0) {
             return new bytes(0);
         }
-        return messages[0];
+        return messageHistory[messageHistory.length - 1][0];
     }
 
     function getMessages() public view returns (bytes[] memory) {
-        return messages;
+        if (messageHistory.length == 0 || messageHistory[messageHistory.length - 1].length == 0) {
+            return new bytes[](0);
+        }
+        return messageHistory[messageHistory.length - 1];
+    }
+
+    function getMessageHistory() public view returns (bytes[][] memory) {
+        return messageHistory;
     }
 
     function clearPayload(bytes32 hash) public {
@@ -210,6 +228,13 @@ contract MockRelayerIntegration is IWormholeReceiver {
         registeredContracts[chainId] = emitterAddress;
     }
 
+    function registerEmitters(Structs.XAddress[] calldata emitters) public {
+        require(msg.sender == owner);
+        for (uint256 i = 0; i < emitters.length; i++) {
+            registeredContracts[emitters[i].chainId] = emitters[i].addr;
+        }
+    }
+
     function encodeFurtherInstructions(FurtherInstructions memory furtherInstructions)
         public
         view

ethereum/forge-test/CoreRelayer.t.sol

@@ -23,7 +23,7 @@ import {IWormhole} from "../contracts/interfaces/IWormhole.sol";
 import {WormholeSimulator, FakeWormholeSimulator} from "./WormholeSimulator.sol";
 import {IWormholeReceiver} from "../contracts/interfaces/IWormholeReceiver.sol";
 import {AttackForwardIntegration} from "../contracts/mock/AttackForwardIntegration.sol";
-import {MockRelayerIntegration} from "../contracts/mock/MockRelayerIntegration.sol";
+import {MockRelayerIntegration, Structs} from "../contracts/mock/MockRelayerIntegration.sol";
 import "../contracts/libraries/external/BytesLib.sol";
 
 import "forge-std/Test.sol";
@@ -255,6 +255,9 @@ contract TestCoreRelayer is Test {
                 );
                 map[i].relayProvider.updateMaximumBudget(j, maxBudget);
                 map[i].integration.registerEmitter(j, bytes32(uint256(uint160(address(map[j].integration)))));
+                Structs.XAddress[] memory addresses = new Structs.XAddress[](1);
+                addresses[0] = Structs.XAddress(j, bytes32(uint256(uint160(address(map[j].integration)))));
+                map[i].integration.registerEmitters(addresses);
             }
         }
     }