-
Notifications
You must be signed in to change notification settings - Fork 38
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
4df3ca6
commit fcd30de
Showing
11 changed files
with
363 additions
and
899 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
171 changes: 171 additions & 0 deletions
171
evm/src/NttManager/NttManagerWithPerChainTransceivers.sol
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,171 @@ | ||
| // SPDX-License-Identifier: Apache 2 | ||
| pragma solidity >=0.8.8 <0.9.0; | ||
|
|
||
| import "./NttManager.sol"; | ||
|
|
||
| abstract contract NttManagerWithPerChainTransceivers is NttManager { | ||
| constructor( | ||
| address _token, | ||
| Mode _mode, | ||
| uint16 _chainId, | ||
| uint64 _rateLimitDuration, | ||
| bool _skipRateLimiting | ||
| ) NttManager(_token, _mode, _chainId, _rateLimitDuration, _skipRateLimiting) {} | ||
|
|
||
| bytes32 private constant SEND_TRANSCEIVER_BITMAP_SLOT = | ||
| bytes32(uint256(keccak256("nttpct.sendTransceiverBitmap")) - 1); | ||
|
|
||
| bytes32 private constant RECV_TRANSCEIVER_BITMAP_SLOT = | ||
| bytes32(uint256(keccak256("nttpct.recvTransceiverBitmap")) - 1); | ||
|
|
||
| // ==================== Override / implementation of transceiver stuff ========================= | ||
|
|
||
| /// @inheritdoc IManagerBase | ||
| function enableSendTransceiverForChain( | ||
| address transceiver, | ||
| uint16 forChainId | ||
| ) external override(ManagerBase, IManagerBase) onlyOwner { | ||
| _enableTranceiverForChain(transceiver, forChainId, SEND_TRANSCEIVER_BITMAP_SLOT); | ||
| // if (transceiver == address(0)) { | ||
| // revert InvalidTransceiverZeroAddress(); | ||
| // } | ||
|
|
||
| // mapping(address => TransceiverInfo) storage transceiverInfos = _getTransceiverInfosStorage(); | ||
| // if (!transceiverInfos[transceiver].registered) { | ||
| // revert NonRegisteredTransceiver(transceiver); | ||
| // } | ||
|
|
||
| // uint8 index = _getTransceiverInfosStorage()[transceiver].index; | ||
| // mapping(uint16 => _EnabledTransceiverBitmap) storage _bitmaps = | ||
| // _getPerChainTransceiverBitmapStorage(SEND_TRANSCEIVER_BITMAP_SLOT); | ||
| // _bitmaps[forChainId].bitmap |= uint64(1 << index); | ||
|
|
||
| // emit TransceiverEnabledForChain(transceiver, forChainId); | ||
| } | ||
|
|
||
| /// @inheritdoc IManagerBase | ||
| function enableRecvTransceiverForChain( | ||
| address transceiver, | ||
| uint16 forChainId | ||
| ) external override(ManagerBase, IManagerBase) onlyOwner { | ||
| _enableTranceiverForChain(transceiver, forChainId, RECV_TRANSCEIVER_BITMAP_SLOT); | ||
| // if (transceiver == address(0)) { | ||
| // revert InvalidTransceiverZeroAddress(); | ||
| // } | ||
|
|
||
| // mapping(address => TransceiverInfo) storage transceiverInfos = _getTransceiverInfosStorage(); | ||
| // if (!transceiverInfos[transceiver].registered) { | ||
| // revert NonRegisteredTransceiver(transceiver); | ||
| // } | ||
|
|
||
| // uint8 index = _getTransceiverInfosStorage()[transceiver].index; | ||
| // mapping(uint16 => _EnabledTransceiverBitmap) storage _bitmaps = | ||
| // _getPerChainTransceiverBitmapStorage(RECV_TRANSCEIVER_BITMAP_SLOT); | ||
| // _bitmaps[forChainId].bitmap |= uint64(1 << index); | ||
|
|
||
| // emit TransceiverEnabledForChain(transceiver, forChainId); | ||
| } | ||
|
|
||
| function _enableTranceiverForChain( | ||
| address transceiver, | ||
| uint16 forChainId, | ||
| bytes32 tag | ||
| ) internal onlyOwner { | ||
| if (transceiver == address(0)) { | ||
| revert InvalidTransceiverZeroAddress(); | ||
| } | ||
|
|
||
| mapping(address => TransceiverInfo) storage transceiverInfos = _getTransceiverInfosStorage(); | ||
| if (!transceiverInfos[transceiver].registered) { | ||
| revert NonRegisteredTransceiver(transceiver); | ||
| } | ||
|
|
||
| uint8 index = _getTransceiverInfosStorage()[transceiver].index; | ||
| mapping(uint16 => _EnabledTransceiverBitmap) storage _bitmaps = | ||
| _getPerChainTransceiverBitmapStorage(tag); | ||
| _bitmaps[forChainId].bitmap |= uint64(1 << index); | ||
|
|
||
| emit TransceiverEnabledForChain(transceiver, forChainId); | ||
| } | ||
|
|
||
| function _isSendTransceiverEnabledForChain( | ||
| address transceiver, | ||
| uint16 forChainId | ||
| ) internal view override returns (bool) { | ||
| uint64 bitmap = | ||
| _getPerChainTransceiverBitmapStorage(SEND_TRANSCEIVER_BITMAP_SLOT)[forChainId].bitmap; | ||
| if (bitmap == 0) { | ||
| // NOTE: this makes it backwards compatible -- if the bitmap is not | ||
| // set, it's assumed the corridor uses all transceivers. | ||
| bitmap = type(uint64).max; | ||
| } | ||
| uint8 index = _getTransceiverInfosStorage()[transceiver].index; | ||
| return (bitmap & uint64(1 << index)) != 0; | ||
| } | ||
|
|
||
| function _getEnabledRecvTransceiversForChain( | ||
| uint16 forChainId | ||
| ) internal view override returns (uint64 bitmap) { | ||
| bitmap = | ||
| _getPerChainTransceiverBitmapStorage(RECV_TRANSCEIVER_BITMAP_SLOT)[forChainId].bitmap; | ||
| if (bitmap == 0) { | ||
| // NOTE: this makes it backwards compatible -- if the bitmap is not | ||
| // set, it's assumed the corridor uses all transceivers. | ||
| bitmap = type(uint64).max; | ||
| } | ||
| } | ||
|
|
||
| function _getPerChainTransceiverBitmapStorage( | ||
| bytes32 tag | ||
| ) internal pure returns (mapping(uint16 => _EnabledTransceiverBitmap) storage $) { | ||
| // TODO: this is safe (reusing the storage slot, because the mapping | ||
| // doesn't write into the slot itself) buy maybe we shouldn't? | ||
| uint256 slot = uint256(tag); | ||
| assembly ("memory-safe") { | ||
| $.slot := slot | ||
| } | ||
| } | ||
|
|
||
| // ==================== Override / implementation of threshold stuff ========================= | ||
|
|
||
| /// @inheritdoc IManagerBase | ||
| function setPerChainThreshold( | ||
| uint16 forChainId, | ||
| uint8 threshold | ||
| ) external override(ManagerBase, IManagerBase) onlyOwner { | ||
| if (threshold == 0) { | ||
| revert ZeroThreshold(); | ||
| } | ||
|
|
||
| mapping(uint16 => _Threshold) storage _threshold = _getThresholdStoragePerChain(); | ||
| uint8 oldThreshold = _threshold[forChainId].num; | ||
|
|
||
| _threshold[forChainId].num = threshold; | ||
| _checkThresholdInvariants(_threshold[forChainId].num); | ||
|
|
||
| emit PerChainThresholdChanged(forChainId, oldThreshold, threshold); | ||
| } | ||
|
|
||
| function getPerChainThreshold( | ||
| uint16 forChainId | ||
| ) public view override returns (uint8) { | ||
| uint8 threshold = _getThresholdStoragePerChain()[forChainId].num; | ||
| if (threshold == 0) { | ||
| return _getThresholdStorage().num; | ||
| } | ||
| return threshold; | ||
| } | ||
|
|
||
| function _getThresholdStoragePerChain() | ||
| private | ||
| pure | ||
| returns (mapping(uint16 => _Threshold) storage $) | ||
| { | ||
| // TODO: this is safe (reusing the storage slot, because the mapping | ||
| // doesn't write into the slot itself) buy maybe we shouldn't? | ||
| uint256 slot = uint256(THRESHOLD_SLOT); | ||
| assembly ("memory-safe") { | ||
| $.slot := slot | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.