feat: add remote chain check in setRemoteSmartMToken

m0-foundation · Dec 2, 2024 · 9176d84 · 9176d84
1 parent 73d72bb
commit 9176d84
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/Portal.sol b/src/Portal.sol
@@ -80,6 +80,8 @@ abstract contract Portal is NttManagerNoRateLimiting, IPortal {
 
     /// @inheritdoc IPortal
     function setRemoteSmartMToken(uint16 remoteChainId_, bytes32 smartMToken_) external onlyOwner {
+        if (remoteChainId_ == chainId) revert InvalidRemoteChain(remoteChainId_);
+
         remoteSmartMToken[remoteChainId_] = smartMToken_;
         emit RemoteSmartMTokenSet(remoteChainId_, smartMToken_);
     }

diff --git a/src/interfaces/IPortal.sol b/src/interfaces/IPortal.sol
@@ -117,6 +117,10 @@ interface IPortal {
     ///         isn't equal to EVM chainId set in the constructor.
     error InvalidFork(uint256 evmChainId, uint256 blockChainId);
 
+    /// @notice Emitted in `setRemoteSmartMToken` when the remote chain id
+    ///         is equal to the local one.
+    error InvalidRemoteChain(uint16 remoteChainId);
+
     /* ============ View/Pure Functions ============ */
 
     /// @notice The current index of the Portal's earning mechanism.