From 27bdb315b5fdf95fd0aa5eae52d7ded5584382e3 Mon Sep 17 00:00:00 2001
From: kimroniny <1301862177@qq.com>
Date: Mon, 23 May 2022 17:47:06 +0800
Subject: [PATCH] Counter:  SimpleCounter; MultiSigCounter

---
 .../common_tools/counter/MultiSigCounter.sol  | 164 ++++++++++++++++++
 .../common_tools/counter/SimpleCounter.sol    |  49 ++++++
 2 files changed, 213 insertions(+)
 create mode 100644 contracts/common_tools/counter/MultiSigCounter.sol
 create mode 100644 contracts/common_tools/counter/SimpleCounter.sol

diff --git a/contracts/common_tools/counter/MultiSigCounter.sol b/contracts/common_tools/counter/MultiSigCounter.sol
new file mode 100644
index 00000000..538f8b4e
--- /dev/null
+++ b/contracts/common_tools/counter/MultiSigCounter.sol
@@ -0,0 +1,164 @@
+pragma solidity 0.4.25;
+
+// Contract: 
+//      MultiSigCounter
+// Description:
+//      inc/dec happens only when enough voters grant
+// Core Functions:
+//      constructor(address[] memory _voters, uint _limit): constructor of this MultiSigCounter
+//              args:
+//                  _voters: all granted voters for this counter
+//                  _limit: the minimum of voters
+//      new_event(uint8 _opr_type, uint _opr_x) -> (uint event_id): start a new event, 
+//              args:
+//                  _opr_type: 0(inc), 1(dec);
+//                  _opr_x: inc/dec counter by x
+//              return:
+//                  event_id: id of new event
+//      vote(uint event_id) -> (uint8 code, uint val): vote for an event,    
+//              args:
+//                  event_id: id of the event
+//              return:
+//                  code: 0: event expired
+//                      1: voting
+//                      2: voted and exec
+//                      3: voted but executed
+//                      4: exec but failed
+//                      5: exec and succeed
+//                  val: value of counter
+//      view_event(uint event_id) -> (bool flag, uint cnt): view an event
+//              args:
+//                  event_id: id of the event
+//              return:
+//                  flag: true if the event exists, false if not
+//                  cnt: number of voted voters if the event exists, 0 if not
+//       
+
+contract MultiSigCounter {
+    uint256 counter = 0;
+    uint256 MAXVAL = 1<<256-1;
+    address[] public voterset;
+    mapping(address => uint) votersidx;
+    uint limit;
+
+    struct Operation {
+        uint8 opr_status; // 0: no opr, 1: voting, 2: voted, 3: voted and executed
+        uint8 opr_type; // 0: inc, 1: dec
+        uint opr_x; // inc/dec counter by x;
+        uint timestamp; // timeout when 10 blocks
+        mapping(address => bool) voters;
+
+    }
+    Operation[] public operations;
+    
+
+    constructor(address[] memory _voters, uint _limit) public {
+        counter = 0;
+        voterset = _voters;
+        limit = _limit;
+        for (uint i = 0; i < voterset.length; i++) {
+            votersidx[voterset[i]] = i;
+        }
+    }
+
+    function _valid_opr_type(uint8 _type) internal pure returns(bool) {
+        return _type == 1 || _type == 2;
+    }
+
+    function new_event(uint8 _opr_type, uint _opr_x) public returns(uint) {
+        require(_valid_opr_type(_opr_type), "oprration type invalid!");
+        operations.push(Operation(
+            1,
+            _opr_type,
+            _opr_x,
+            block.number
+        ));
+        operations[operations.length-1].voters[msg.sender] = true;
+        return operations.length-1;
+    }
+
+    function view_event(uint event_id) public view returns(bool, uint) {
+        if (event_id >= operations.length) {return (false, 0);}
+        Operation storage opr = operations[event_id];
+        uint cnt = 0;
+        for (uint i = 0; i < voterset.length; i++) {
+            if (opr.voters[voterset[i]]) {
+                cnt += 1;
+            }
+        }
+        return (true, cnt);
+    }
+
+    function can_exec(uint event_id) internal view returns(uint8) {
+        require(event_id < operations.length, "event id invalid");
+        Operation storage opr = operations[event_id];
+        if (block.number - opr.timestamp > 100) {
+            return 0;
+        }
+        uint cnt = 0;
+        for (uint i = 0; i < voterset.length; i++) {
+            if (opr.voters[voterset[i]]) {
+                cnt += 1;
+            }
+        }
+        if (cnt < limit) {return 1;} // voting
+        if (cnt == limit) {return 2;} // voted and can execute
+        if (cnt > limit) {return 3;} // voted and executed
+    }
+
+    function vote(uint event_id) public returns(uint8, uint) {
+        /*
+            (code, val):
+            0: event expired
+            1: voting
+            2: voted and exec
+            3: voted but executed
+            4: exec but failed
+            5: exec and succeed
+        */
+        require(event_id < operations.length, "event id invalid");
+        operations[event_id].voters[msg.sender] = true;
+        uint8 status = can_exec(event_id);
+        if (status == 2) {
+            Operation memory opr = operations[event_id];
+            return exec(opr.opr_type, opr.opr_x);
+        }
+        return (status, get());
+    }
+
+    function exec(uint8 _type, uint arg) internal returns(uint8, uint) {
+         bool flag = false; 
+        uint val = get();
+        if (_type == 0) {
+            (flag, val) = inc_val(arg);
+        }else if (_type == 1) {
+            (flag, val) = dec_val(arg);
+        }
+        if (!flag) {
+            return (4, val);
+        }else{
+            return (5, val);
+        }
+    }
+
+    function inc_val(uint x) internal returns(bool, uint) {
+        if (MAXVAL - counter >= x) {
+            counter += x;
+            return (true, counter);
+        }
+        return (false, counter);
+    }
+
+    function dec_val(uint x) internal returns(bool, uint) {
+        if (counter - x >= 0) {
+            counter -= x;
+            return (true, counter);
+        }
+        return (false, counter);
+    }
+
+    function get() public view returns(uint) {
+        return counter;
+    }
+
+}
\ No newline at end of file
diff --git a/contracts/common_tools/counter/SimpleCounter.sol b/contracts/common_tools/counter/SimpleCounter.sol
new file mode 100644
index 00000000..deb1c9fb
--- /dev/null
+++ b/contracts/common_tools/counter/SimpleCounter.sol
@@ -0,0 +1,49 @@
+pragma solidity 0.4.25;
+
+// Contract: SimpleCounter
+// Description:
+//      inc/dec counter by val
+//      safe inc and dec with maximum uint256 and minimum 0
+// Core Functions:
+//      inc_val(uint x) -> (bool, uint): inc counter by x
+//      dec_val(uint x) -> (bool, uint): dec counter by x
+//      inc() -> (bool, uint): inc counter by 1
+//      dec() -> (bool, uint): dec counter by 1
+
+contract SimpleCounter {
+    uint256 counter = 0;
+    uint256 MAXVAL = 1<<256-1;
+
+    constructor() public {
+        counter = 0;
+    }
+
+    function inc_val(uint x) public returns(bool, uint) {
+        if (MAXVAL - counter >= x) {
+            counter += x;
+            return (true, counter);
+        }
+        return (false, counter);
+    }
+
+    function dec_val(uint x) public returns(bool, uint) {
+        if (counter - x >= 0) {
+            counter -= x;
+            return (true, counter);
+        }
+        return (false, counter);
+    }
+
+    function inc() public returns(bool, uint) {
+        return inc_val(1);
+    }
+
+    function dec() public returns(bool, uint) {
+        return dec_val(1);
+    }
+
+    function get() public view returns(uint) {
+        return counter;
+    }
+
+}
\ No newline at end of file