enable and disable channel through governance

bnb-chain · Jun 24, 2020 · 470257e · 470257e
1 parent 6f548e1
commit 470257e
Show file tree

Hide file tree

Showing 4 changed files with 50 additions and 1 deletion.
diff --git a/contracts/CrossChain.sol b/contracts/CrossChain.sol
@@ -41,6 +41,7 @@ contract CrossChain is System, ICrossChain, IParamSubscriber{
   event unexpectedRevertInPackageHandler(address indexed contractAddr, string reason);
   event unexpectedFailureAssertionInPackageHandler(address indexed contractAddr, bytes lowLevelData);
   event paramChange(string key, bytes value);
+  event enableOrDisableChannel(uint8 indexed channelId, bool isEnable);
   event addChannel(uint8 indexed channelId, address indexed contractAddr);
 
   modifier sequenceInOrder(uint64 _sequence, uint8 _channelID) {
@@ -287,6 +288,23 @@ function encodePayload(uint8 packageType, uint256 relayFee, bytes memory msgByte
       registeredContractMap[handlerContract] = true;
       isRelayRewardFromSystemReward[channelId] = (isRewardFromSystem == 0x0);
       emit addChannel(channelId, handlerContract);
+    } else if (Memory.compareStrings(key, "enableOrDisableChannel")) {
+      bytes memory valueLocal = value;
+      require(valueLocal.length == 2, "length of value for enableOrDisableChannel should be 2, channelId:isEnable");
+
+      uint8 channelId;
+      assembly {
+        channelId := mload(add(valueLocal, 1))
+      }
+      uint8 status;
+      assembly {
+        status := mload(add(valueLocal, 2))
+      }
+      bool isEnable = (status == 1);
+
+      address handlerContract = channelHandlerContractMap[channelId];
+      registeredContractMap[handlerContract] = isEnable;
+      emit enableOrDisableChannel(channelId, isEnable);
     } else {
       require(false, "unknown param");
     }

diff --git a/contracts/CrossChain.template b/contracts/CrossChain.template
@@ -42,6 +42,7 @@ contract CrossChain is System, ICrossChain, IParamSubscriber{
   event unexpectedRevertInPackageHandler(address indexed contractAddr, string reason);
   event unexpectedFailureAssertionInPackageHandler(address indexed contractAddr, bytes lowLevelData);
   event paramChange(string key, bytes value);
+  event enableOrDisableChannel(uint8 indexed channelId, bool isEnable);
   event addChannel(uint8 indexed channelId, address indexed contractAddr);
 
   modifier sequenceInOrder(uint64 _sequence, uint8 _channelID) {
@@ -288,6 +289,23 @@ function encodePayload(uint8 packageType, uint256 relayFee, bytes memory msgByte
       registeredContractMap[handlerContract] = true;
       isRelayRewardFromSystemReward[channelId] = (isRewardFromSystem == 0x0);
       emit addChannel(channelId, handlerContract);
+    } else if (Memory.compareStrings(key, "enableOrDisableChannel")) {
+      bytes memory valueLocal = value;
+      require(valueLocal.length == 2, "length of value for enableOrDisableChannel should be 2, channelId:isEnable");
+
+      uint8 channelId;
+      assembly {
+        channelId := mload(add(valueLocal, 1))
+      }
+      uint8 status;
+      assembly {
+        status := mload(add(valueLocal, 2))
+      }
+      bool isEnable = (status == 1);
+
+      address handlerContract = channelHandlerContractMap[channelId];
+      registeredContractMap[handlerContract] = isEnable;
+      emit enableOrDisableChannel(channelId, isEnable);
     } else {
       require(false, "unknown param");
     }