Improve MultiMessageSender (#76)

- Require the caller to specify fees to each adapter. Removing on-chain fee estimation which may not work with some adapters. - Maintain a senderAdapters array sorted in ascending order by address.
MultiMessageAggregation · Sep 22, 2023 · 12f460e · 12f460e
1 parent d7a877c
commit 12f460e
Show file tree

Hide file tree

Showing 17 changed files with 510 additions and 247 deletions.
diff --git a/src/MultiBridgeMessageSender.sol b/src/MultiBridgeMessageSender.sol
diff --git a/src/adapters/axelar/AxelarSenderAdapter.sol b/src/adapters/axelar/AxelarSenderAdapter.sol
@@ -82,17 +82,6 @@ contract AxelarSenderAdapter is BaseSenderAdapter {
         }
     }
 
-    /*/////////////////////////////////////////////////////////////////
-                            EXTERNAL VIEW FUNCTIONS
-    ////////////////////////////////////////////////////////////////*/
-
-    /// @inheritdoc IMessageSenderAdapter
-    function getMessageFee(uint256, address, bytes calldata) external view override returns (uint256) {
-        /// FIXME: axelar has no on-chain message fee estimate. should have to overpay and get refund
-        return 1 ether;
-        // return axelarChainRegistry.getFee(_toChainId, uint32(gac.getGlobalMsgDeliveryGasLimit()));
-    }
-
     /*/////////////////////////////////////////////////////////////////
                             HELPER FUNCTIONS
     ////////////////////////////////////////////////////////////////*/

diff --git a/src/adapters/wormhole/WormholeSenderAdapter.sol b/src/adapters/wormhole/WormholeSenderAdapter.sol
@@ -84,14 +84,4 @@ contract WormholeSenderAdapter is BaseSenderAdapter {
             }
         }
     }
-
-    /*/////////////////////////////////////////////////////////////////
-                            EXTERNAL VIEW FUNCTIONS
-    ////////////////////////////////////////////////////////////////*/
-
-    /// @inheritdoc IMessageSenderAdapter
-    function getMessageFee(uint256 _toChainId, address, bytes calldata) external view override returns (uint256 fee) {
-        /// note: 50000 GAS is commonly used across the MMA; move to some global contract
-        (fee,) = relayer.quoteEVMDeliveryPrice(chainIdMap[_toChainId], 0, senderGAC.getGlobalMsgDeliveryGasLimit());
-    }
 }
diff --git a/src/interfaces/adapters/IMessageSenderAdapter.sol b/src/interfaces/adapters/IMessageSenderAdapter.sol
@@ -25,12 +25,6 @@ interface IMessageSenderAdapter is SingleMessageDispatcher {
     /// @notice returns name of the message bridge wrapped by the adapter
     function name() external view returns (string memory);
 
-    /// @notice return the fee (in native token wei) that would be charged by the bridge for the provided remote call
-    /// @param _toChainId is the destination chain id
-    /// @param _to is the destination address on the destination chain
-    /// @param _data is the data to be sent to the destination chain
-    function getMessageFee(uint256 _toChainId, address _to, bytes calldata _data) external view returns (uint256);
-
     /// @notice returns the bridge receiver adapter address for a given destination chain id
     /// @param _chainId is the destination chain whose receiver adapter address is to be returned
     function getReceiverAdapter(uint256 _chainId) external view returns (address);

diff --git a/src/libraries/Error.sol b/src/libraries/Error.sol
@@ -38,6 +38,12 @@ library Error {
     /// @dev is thrown if sender adapter array has duplicates
     error DUPLICATE_SENDER_ADAPTER();
 
+    /// @dev is thrown if the sender adapter does not exist
+    error SENDER_ADAPTER_NOT_EXIST();
+
+    /// @dev is thrown if sender adapter array is not in ascending order or has duplicates
+    error INVALID_SENDER_ADAPTER_ORDER();
+
     /// @dev is thrown if deadline is lapsed
     error MSG_EXECUTION_PASSED_DEADLINE();
 
@@ -72,6 +78,9 @@ library Error {
     /// @dev is thrown if sender adapter is not allowed on receiver adapter
     error INVALID_SENDER_ADAPTER();
 
+    /// @dev is thrown if the sender adapter fee array is invalid
+    error INVALID_SENDER_ADAPTER_FEES();
+
     /// @dev is thrown if final destination is not mma receiver on receiver adapter
     error INVALID_FINAL_DESTINATION();
 

diff --git a/test/contracts-mock/FailingSenderAdapter.sol b/test/contracts-mock/FailingSenderAdapter.sol
@@ -6,8 +6,4 @@ contract FailingSenderAdapter {
     function dispatchMessage(uint256, address, bytes calldata) external payable returns (bytes32) {
         revert();
     }
-
-    function getMessageFee(uint256, address, bytes calldata) external pure returns (uint256) {
-        return 0;
-    }
 }
diff --git a/test/integration-tests/GracePeriodExpiry.t.sol b/test/integration-tests/GracePeriodExpiry.t.sol
@@ -3,6 +3,7 @@ pragma solidity >=0.8.9;
 
 /// library imports
 import {Vm} from "forge-std/Test.sol";
+import "wormhole-solidity-sdk/interfaces/IWormholeRelayer.sol";
 
 /// local imports
 import "test/Setup.t.sol";
@@ -31,13 +32,19 @@ contract GracePeriodExpiryTest is Setup {
 
         /// send cross-chain message using MMA infra
         vm.recordLogs();
+        uint256[] memory fees = new uint256[](2);
+        (uint256 wormholeFee,) =
+            IWormholeRelayer(POLYGON_RELAYER).quoteEVMDeliveryPrice(_wormholeChainId(DST_CHAIN_ID), 0, 0);
+        fees[0] = wormholeFee;
+        fees[1] = 0.01 ether;
         MultiBridgeMessageSender(contractAddress[SRC_CHAIN_ID][bytes("MMA_SENDER")]).remoteCall{value: 2 ether}(
             DST_CHAIN_ID,
             address(target),
             abi.encode(MockUniswapReceiver.setValue.selector, ""),
             0,
             EXPIRATION_CONSTANT,
-            refundAddress
+            refundAddress,
+            fees
         );
 
         Vm.Log[] memory logs = vm.getRecordedLogs();

diff --git a/test/integration-tests/MultiMessageAggregation.t.sol b/test/integration-tests/MultiMessageAggregation.t.sol
@@ -3,6 +3,7 @@ pragma solidity >=0.8.9;
 
 /// library imports
 import {Vm} from "forge-std/Test.sol";
+import "wormhole-solidity-sdk/interfaces/IWormholeRelayer.sol";
 
 /// local imports
 import "test/Setup.t.sol";
@@ -31,13 +32,19 @@ contract MultiBridgeMessageAggregationTest is Setup {
 
         /// send cross-chain message using MMA infra
         vm.recordLogs();
+        uint256[] memory fees = new uint256[](2);
+        (uint256 wormholeFee,) =
+            IWormholeRelayer(POLYGON_RELAYER).quoteEVMDeliveryPrice(_wormholeChainId(DST_CHAIN_ID), 0, 0);
+        fees[0] = wormholeFee;
+        fees[1] = 0.01 ether;
         MultiBridgeMessageSender(contractAddress[SRC_CHAIN_ID][bytes("MMA_SENDER")]).remoteCall{value: 2 ether}(
             DST_CHAIN_ID,
             address(target),
             abi.encode(MockUniswapReceiver.setValue.selector, ""),
             0,
             EXPIRATION_CONSTANT,
-            refundAddress
+            refundAddress,
+            fees
         );
 
         Vm.Log[] memory logs = vm.getRecordedLogs();

diff --git a/test/integration-tests/RemoteAdapterAdd.t.sol b/test/integration-tests/RemoteAdapterAdd.t.sol
@@ -3,6 +3,7 @@ pragma solidity >=0.8.9;
 
 /// library imports
 import {Vm} from "forge-std/Test.sol";
+import "wormhole-solidity-sdk/interfaces/IWormholeRelayer.sol";
 
 /// local imports
 import "test/Setup.t.sol";
@@ -56,13 +57,19 @@ contract RemoteAdapterAdd is Setup {
 
         /// send cross-chain message using MMA infra
         vm.recordLogs();
+        uint256[] memory fees = new uint256[](2);
+        (uint256 wormholeFee,) =
+            IWormholeRelayer(POLYGON_RELAYER).quoteEVMDeliveryPrice(_wormholeChainId(DST_CHAIN_ID), 0, 0);
+        fees[0] = wormholeFee;
+        fees[1] = 0.01 ether;
         MultiBridgeMessageSender(contractAddress[SRC_CHAIN_ID][bytes("MMA_SENDER")]).remoteCall{value: 2 ether}(
             DST_CHAIN_ID,
             address(contractAddress[DST_CHAIN_ID][bytes("MMA_RECEIVER")]),
             abi.encodeWithSelector(MultiBridgeMessageReceiver.updateReceiverAdapters.selector, adaptersToAdd, operation),
             0,
             EXPIRATION_CONSTANT,
-            refundAddress
+            refundAddress,
+            fees
         );
 
         Vm.Log[] memory logs = vm.getRecordedLogs();

diff --git a/test/integration-tests/RemoteAdapterRemove.t.sol b/test/integration-tests/RemoteAdapterRemove.t.sol
@@ -3,6 +3,7 @@ pragma solidity >=0.8.9;
 
 /// library imports
 import {Vm} from "forge-std/Test.sol";
+import "wormhole-solidity-sdk/interfaces/IWormholeRelayer.sol";
 
 /// local imports
 import "test/Setup.t.sol";
@@ -61,6 +62,11 @@ contract RemoteAdapterRemove is Setup {
 
         /// send cross-chain message using MMA infra
         vm.recordLogs();
+        uint256[] memory fees = new uint256[](2);
+        (uint256 wormholeFee,) =
+            IWormholeRelayer(POLYGON_RELAYER).quoteEVMDeliveryPrice(_wormholeChainId(DST_CHAIN_ID), 0, 0);
+        fees[0] = wormholeFee;
+        fees[1] = 0.01 ether;
         MultiBridgeMessageSender(contractAddress[SRC_CHAIN_ID][bytes("MMA_SENDER")]).remoteCall{value: 2 ether}(
             DST_CHAIN_ID,
             address(contractAddress[DST_CHAIN_ID][bytes("MMA_RECEIVER")]),
@@ -72,7 +78,8 @@ contract RemoteAdapterRemove is Setup {
             ),
             0,
             EXPIRATION_CONSTANT,
-            refundAddress
+            refundAddress,
+            fees
         );
 
         Vm.Log[] memory logs = vm.getRecordedLogs();

diff --git a/test/integration-tests/RemoteSetQuorum.t.sol b/test/integration-tests/RemoteSetQuorum.t.sol
@@ -3,6 +3,7 @@ pragma solidity >=0.8.9;
 
 /// library imports
 import {Vm} from "forge-std/Test.sol";
+import "wormhole-solidity-sdk/interfaces/IWormholeRelayer.sol";
 
 /// local imports
 import "test/Setup.t.sol";
@@ -28,13 +29,19 @@ contract RemoteQuorumUpdate is Setup {
 
         /// send cross-chain message using MMA infra
         vm.recordLogs();
+        uint256[] memory fees = new uint256[](2);
+        (uint256 wormholeFee,) =
+            IWormholeRelayer(POLYGON_RELAYER).quoteEVMDeliveryPrice(_wormholeChainId(DST_CHAIN_ID), 0, 0);
+        fees[0] = wormholeFee;
+        fees[1] = 0.01 ether;
         MultiBridgeMessageSender(contractAddress[SRC_CHAIN_ID][bytes("MMA_SENDER")]).remoteCall{value: 2 ether}(
             DST_CHAIN_ID,
             address(contractAddress[DST_CHAIN_ID][bytes("MMA_RECEIVER")]),
             abi.encodeWithSelector(MultiBridgeMessageReceiver.updateQuorum.selector, newQuorum),
             0,
             EXPIRATION_CONSTANT,
-            refundAddress
+            refundAddress,
+            fees
         );
 
         Vm.Log[] memory logs = vm.getRecordedLogs();

diff --git a/test/integration-tests/RemoteTimelockUpdate.t.sol b/test/integration-tests/RemoteTimelockUpdate.t.sol
@@ -3,6 +3,7 @@ pragma solidity >=0.8.9;
 
 /// library imports
 import {Vm} from "forge-std/Test.sol";
+import "wormhole-solidity-sdk/interfaces/IWormholeRelayer.sol";
 
 /// local imports
 import "test/Setup.t.sol";
@@ -28,13 +29,19 @@ contract RemoteTimelockUpdate is Setup {
 
         /// send cross-chain message using MMA infra
         vm.recordLogs();
+        uint256[] memory fees = new uint256[](2);
+        (uint256 wormholeFee,) =
+            IWormholeRelayer(POLYGON_RELAYER).quoteEVMDeliveryPrice(_wormholeChainId(DST_CHAIN_ID), 0, 0);
+        fees[0] = wormholeFee;
+        fees[1] = 0.01 ether;
         MultiBridgeMessageSender(contractAddress[SRC_CHAIN_ID][bytes("MMA_SENDER")]).remoteCall{value: 2 ether}(
             POLYGON_CHAIN_ID,
             address(contractAddress[POLYGON_CHAIN_ID][bytes("TIMELOCK")]),
             abi.encodeWithSelector(GovernanceTimelock.setDelay.selector, newDelay),
             0,
             EXPIRATION_CONSTANT,
-            refundAddress
+            refundAddress,
+            fees
         );
 
         Vm.Log[] memory logs = vm.getRecordedLogs();

diff --git a/test/integration-tests/TimelockCheck.t.sol b/test/integration-tests/TimelockCheck.t.sol
@@ -3,6 +3,7 @@ pragma solidity >=0.8.9;
 
 /// library imports
 import {Vm} from "forge-std/Test.sol";
+import "wormhole-solidity-sdk/interfaces/IWormholeRelayer.sol";
 
 /// local imports
 import "test/Setup.t.sol";
@@ -33,13 +34,19 @@ contract TimelockCheckTest is Setup {
 
         /// send cross-chain message using MMA infra
         vm.recordLogs();
+        uint256[] memory fees = new uint256[](2);
+        (uint256 wormholeFee,) =
+            IWormholeRelayer(POLYGON_RELAYER).quoteEVMDeliveryPrice(_wormholeChainId(DST_CHAIN_ID), 0, 0);
+        fees[0] = wormholeFee;
+        fees[1] = 0.01 ether;
         MultiBridgeMessageSender(contractAddress[SRC_CHAIN_ID][bytes("MMA_SENDER")]).remoteCall{value: 2 ether}(
             DST_CHAIN_ID,
             address(target),
             abi.encode(MockUniswapReceiver.setValue.selector, ""),
             0,
             EXPIRATION_CONSTANT,
-            refundAddress
+            refundAddress,
+            fees
         );
 
         Vm.Log[] memory logs = vm.getRecordedLogs();

diff --git a/test/unit-tests/MultiBridgeMessageReceiver.t.sol b/test/unit-tests/MultiBridgeMessageReceiver.t.sol
@@ -22,8 +22,8 @@ contract MultiBridgeMessageReceiverTest is Setup {
     );
 
     MultiBridgeMessageReceiver receiver;
-    address wormholeAdapterAddr;
     address axelarAdapterAddr;
+    address wormholeAdapterAddr;
     address timelockAddr;
 
     /// @dev initializes the setup
@@ -32,16 +32,16 @@ contract MultiBridgeMessageReceiverTest is Setup {
 
         vm.selectFork(fork[DST_CHAIN_ID]);
         receiver = MultiBridgeMessageReceiver(contractAddress[DST_CHAIN_ID][bytes("MMA_RECEIVER")]);
-        wormholeAdapterAddr = contractAddress[DST_CHAIN_ID]["WORMHOLE_RECEIVER_ADAPTER"];
         axelarAdapterAddr = contractAddress[DST_CHAIN_ID]["AXELAR_RECEIVER_ADAPTER"];
+        wormholeAdapterAddr = contractAddress[DST_CHAIN_ID]["WORMHOLE_RECEIVER_ADAPTER"];
         timelockAddr = contractAddress[DST_CHAIN_ID]["TIMELOCK"];
     }
 
     /// @dev initializer
     function test_initialize() public {
         address[] memory adapters = new address[](2);
-        adapters[0] = wormholeAdapterAddr;
-        adapters[1] = axelarAdapterAddr;
+        adapters[0] = axelarAdapterAddr;
+        adapters[1] = wormholeAdapterAddr;
 
         bool[] memory operation = new bool[](2);
         operation[0] = true;
@@ -51,8 +51,8 @@ contract MultiBridgeMessageReceiverTest is Setup {
         dummyReceiver.initialize(ETHEREUM_CHAIN_ID, adapters, operation, 2, timelockAddr);
 
         assertEq(dummyReceiver.quorum(), 2);
-        assertTrue(dummyReceiver.isTrustedExecutor(wormholeAdapterAddr));
         assertTrue(dummyReceiver.isTrustedExecutor(axelarAdapterAddr));
+        assertTrue(dummyReceiver.isTrustedExecutor(wormholeAdapterAddr));
     }
 
     /// @dev initializer cannot be called twice