chore: backport tss updater

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

@@ -83,6 +83,19 @@ contract GatewayEVM is
         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);
 }