fix: missing tss update in erc20custody, zetaConnector and gatewayEVM (#363)

Author: skosito

v2/contracts/evm/ERC20Custody.sol

@@ -58,6 +58,22 @@ contract ERC20Custody is IERC20Custody, ReentrancyGuard, AccessControl, Pausable
         _unpause();
     }
 
+    /// @notice Update tss address
+    /// @param newTSSAddress new tss address
+    function updateTSSAddress(address newTSSAddress) external onlyRole(DEFAULT_ADMIN_ROLE) {
+        if (newTSSAddress == address(0)) revert ZeroAddress();
+
+        _revokeRole(WITHDRAWER_ROLE, tssAddress);
+        _revokeRole(WHITELISTER_ROLE, tssAddress);
+
+        _grantRole(WITHDRAWER_ROLE, newTSSAddress);
+        _grantRole(WHITELISTER_ROLE, newTSSAddress);
+
+        tssAddress = newTSSAddress;
+
+        emit UpdatedCustodyTSSAddress(newTSSAddress);
+    }
+
     /// @notice Unpause contract.
     function setSupportsLegacy(bool _supportsLegacy) external onlyRole(DEFAULT_ADMIN_ROLE) {
         supportsLegacy = _supportsLegacy;

v2/contracts/evm/GatewayEVM.sol

@@ -71,6 +71,30 @@ contract GatewayEVM is
     /// @param newImplementation Address of the new implementation.
     function _authorizeUpgrade(address newImplementation) internal override onlyRole(DEFAULT_ADMIN_ROLE) { }
 
+    /// @dev Internal function to execute a call to a destination address.
+    /// @param destination Address to call.
+    /// @param data Calldata to pass to the call.
+    /// @return The result of the call.
+    function _execute(address destination, bytes calldata data) internal returns (bytes memory) {
+        (bool success, bytes memory result) = destination.call{ value: msg.value }(data);
+        if (!success) revert ExecutionFailed();
+
+        return result;
+    }
+
+    /// @notice Update tss address
+    /// @param newTSSAddress new tss address
+    function updateTSSAddress(address newTSSAddress) external onlyRole(DEFAULT_ADMIN_ROLE) {
+        if (newTSSAddress == address(0)) revert ZeroAddress();
+
+        _revokeRole(TSS_ROLE, tssAddress);
+        _grantRole(TSS_ROLE, newTSSAddress);
+
+        tssAddress = newTSSAddress;
+
+        emit UpdatedGatewayTSSAddress(newTSSAddress);
+    }
+
     /// @notice Pause contract.
     function pause() external onlyRole(PAUSER_ROLE) {
         _pause();

v2/contracts/evm/ZetaConnectorBase.sol

@@ -24,6 +24,8 @@ abstract contract ZetaConnectorBase is IZetaConnectorEvents, ReentrancyGuard, Pa
     IGatewayEVM public immutable gateway;
     /// @notice The address of the Zeta token.
     address public immutable zetaToken;
+    /// @notice The address of the TSS (Threshold Signature Scheme) contract.
+    address public tssAddress;
 
     /// @notice New role identifier for withdrawer role.
     bytes32 public constant WITHDRAWER_ROLE = keccak256("WITHDRAWER_ROLE");
@@ -40,13 +42,30 @@ abstract contract ZetaConnectorBase is IZetaConnectorEvents, ReentrancyGuard, Pa
         }
         gateway = IGatewayEVM(gateway_);
         zetaToken = zetaToken_;
+        tssAddress = tssAddress_;
 
         _grantRole(DEFAULT_ADMIN_ROLE, admin_);
         _grantRole(WITHDRAWER_ROLE, tssAddress_);
         _grantRole(TSS_ROLE, tssAddress_);
         _grantRole(PAUSER_ROLE, admin_);
     }
 
+    /// @notice Update tss address
+    /// @param newTSSAddress new tss address
+    function updateTSSAddress(address newTSSAddress) external onlyRole(DEFAULT_ADMIN_ROLE) {
+        if (newTSSAddress == address(0)) revert ZeroAddress();
+
+        _revokeRole(WITHDRAWER_ROLE, tssAddress);
+        _revokeRole(TSS_ROLE, tssAddress);
+
+        _grantRole(WITHDRAWER_ROLE, newTSSAddress);
+        _grantRole(TSS_ROLE, newTSSAddress);
+
+        tssAddress = newTSSAddress;
+
+        emit UpdatedZetaConnectorTSSAddress(newTSSAddress);
+    }
+
     /// @notice Pause contract.
     function pause() external onlyRole(PAUSER_ROLE) {
         _pause();

v2/contracts/evm/interfaces/IERC20Custody.sol

@@ -40,6 +40,10 @@ interface IERC20CustodyEvents {
 
     /// @notice Emitted in legacy deposit method.
     event Deposited(bytes recipient, IERC20 indexed asset, uint256 amount, bytes message);
+
+    /// @notice Emitted when tss address is updated
+    /// @param newTSSAddress new tss address
+    event UpdatedCustodyTSSAddress(address newTSSAddress);
 }
 
 /// @title IERC20CustodyErrors

v2/contracts/evm/interfaces/IGatewayEVM.sol

@@ -49,6 +49,10 @@ interface IGatewayEVMEvents {
     /// @param payload The calldata passed to the call.
     /// @param revertOptions Revert options.
     event Called(address indexed sender, address indexed receiver, bytes payload, RevertOptions revertOptions);
+
+    /// @notice Emitted when tss address is updated
+    /// @param newTSSAddress new tss address
+    event UpdatedGatewayTSSAddress(address newTSSAddress);
 }
 
 /// @title IGatewayEVMErrors

v2/contracts/evm/interfaces/IZetaConnector.sol

@@ -23,4 +23,8 @@ interface IZetaConnectorEvents {
     /// @param data The calldata passed to the contract call.
     /// @param revertContext Revert context to pass to onRevert.
     event WithdrawnAndReverted(address indexed to, uint256 amount, bytes data, RevertContext revertContext);
+
+    /// @notice Emitted when tss address is updated
+    /// @param newTSSAddress new tss address
+    event UpdatedZetaConnectorTSSAddress(address newTSSAddress);
 }

v2/docs/src/contracts/Revert.sol/interface.Revertable.md

@@ -1,5 +1,5 @@
 # Revertable
-[Git Source](https://github.com/zeta-chain/protocol-contracts/blob/3a274ce7bad045a879c73669586611d35509cbce/contracts/Revert.sol)
+[Git Source](https://github.com/zeta-chain/protocol-contracts/blob/5f09d7eb47b707c65cea167574b26d208e366094/contracts/Revert.sol)
 
 Interface for contracts that support revertable calls.
 

v2/docs/src/contracts/Revert.sol/struct.RevertContext.md

@@ -1,5 +1,5 @@
 # RevertContext
-[Git Source](https://github.com/zeta-chain/protocol-contracts/blob/3a274ce7bad045a879c73669586611d35509cbce/contracts/Revert.sol)
+[Git Source](https://github.com/zeta-chain/protocol-contracts/blob/5f09d7eb47b707c65cea167574b26d208e366094/contracts/Revert.sol)
 
 Struct containing revert context passed to onRevert.
 

v2/docs/src/contracts/Revert.sol/struct.RevertOptions.md

@@ -1,5 +1,5 @@
 # RevertOptions
-[Git Source](https://github.com/zeta-chain/protocol-contracts/blob/3a274ce7bad045a879c73669586611d35509cbce/contracts/Revert.sol)
+[Git Source](https://github.com/zeta-chain/protocol-contracts/blob/5f09d7eb47b707c65cea167574b26d208e366094/contracts/Revert.sol)
 
 Struct containing revert options
 

v2/docs/src/contracts/evm/ERC20Custody.sol/contract.ERC20Custody.md

@@ -1,5 +1,5 @@
 # ERC20Custody
-[Git Source](https://github.com/zeta-chain/protocol-contracts/blob/3a274ce7bad045a879c73669586611d35509cbce/contracts/evm/ERC20Custody.sol)
+[Git Source](https://github.com/zeta-chain/protocol-contracts/blob/5f09d7eb47b707c65cea167574b26d208e366094/contracts/evm/ERC20Custody.sol)
 
 **Inherits:**
 [IERC20Custody](/contracts/evm/interfaces/IERC20Custody.sol/interface.IERC20Custody.md), ReentrancyGuard, AccessControl, Pausable
@@ -103,6 +103,21 @@ Unpause contract.
 function unpause() external onlyRole(PAUSER_ROLE);
 ```
 
+### updateTSSAddress
+
+Update tss address
+
+
+```solidity
+function updateTSSAddress(address newTSSAddress) external onlyRole(DEFAULT_ADMIN_ROLE);
+```
+**Parameters**
+
+|Name|Type|Description|
+|----|----|-----------|
+|`newTSSAddress`|`address`|new tss address|
+
+
 ### setSupportsLegacy
 
 Unpause contract.