gateway revert

zeta-chain · Nov 20, 2024 · 786d629 · 786d629
1 parent c89c6da
commit 786d629
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 4 deletions.
diff --git a/examples/call/contracts/Universal.sol b/examples/call/contracts/Universal.sol
@@ -11,10 +11,12 @@ contract Universal is UniversalContract {
 
     event HelloEvent(string, string);
     event RevertEvent(string, RevertContext);
+
     error TransferFailed();
+    error Unauthorized();
 
     modifier onlyGateway() {
-        require(msg.sender == address(gateway), "Caller is not the gateway");
+        if (msg.sender != address(gateway)) revert Unauthorized();
         _;
     }
 

diff --git a/examples/hello/contracts/Universal.sol b/examples/hello/contracts/Universal.sol
@@ -7,9 +7,10 @@ contract Universal is UniversalContract {
     GatewayZEVM public immutable gateway;
 
     event HelloEvent(string, string);
+    error Unauthorized();
 
     modifier onlyGateway() {
-        require(msg.sender == address(gateway), "Caller is not the gateway");
+        if (msg.sender != address(gateway)) revert Unauthorized();
         _;
     }
 

diff --git a/examples/swap/contracts/Swap.sol b/examples/swap/contracts/Swap.sol
@@ -17,9 +17,10 @@ contract Swap is UniversalContract {
     uint256 constant BITCOIN = 18332;
 
     error InvalidAddress();
+    error Unauthorized();
 
     modifier onlyGateway() {
-        require(msg.sender == address(gateway), "Caller is not the gateway");
+        if (msg.sender != address(gateway)) revert Unauthorized();
         _;
     }
 

diff --git a/examples/swap/contracts/SwapToAnyToken.sol b/examples/swap/contracts/SwapToAnyToken.sol
@@ -16,10 +16,12 @@ contract SwapToAnyToken is UniversalContract {
     address public immutable uniswapRouter;
     GatewayZEVM public gateway;
     uint256 constant BITCOIN = 18332;
+
     error InvalidAddress();
+    error Unauthorized();
 
     modifier onlyGateway() {
-        require(msg.sender == address(gateway), "Caller is not the gateway");
+        if (msg.sender != address(gateway)) revert Unauthorized();
         _;
     }