Last active
February 2, 2023 19:06
-
-
Save frangio/e2438da9b8bdfa08198a9d57dd9c6611 to your computer and use it in GitHub Desktop.
Lightweight explicit state machines in Solidity.
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
// SPDX-License-Identifier: MIT | |
pragma solidity 0.8.16; | |
import "./StateMachines.sol"; | |
contract Pausable is StateMachines { | |
StateMachine m = initial("unpaused"); | |
event Paused(bool paused); | |
function pause() external transition(m, "unpaused", "paused") { | |
emit Paused(true); | |
} | |
function unpause() external { | |
// More efficient than transition modifier for atomic transitions | |
move(m, "paused", "unpaused"); | |
emit Paused(false); | |
} | |
function isPaused() external view returns (bool) { | |
return isState(m, "paused"); | |
} | |
} |
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
// SPDX-License-Identifier: MIT | |
pragma solidity 0.8.16; | |
import "./ToString.sol"; | |
struct StateMachine { | |
bytes31 _state; | |
bool _transitioning; | |
} | |
function initial(bytes31 state) pure returns (StateMachine memory) { | |
return StateMachine(state, false); | |
} | |
contract StateMachines { | |
modifier transition(StateMachine storage m, bytes31 from, bytes31 to) { | |
depart(m, from); | |
_; | |
arrive(m, to); | |
} | |
} | |
function depart(StateMachine storage m, bytes31 from) { | |
checkState(m, from); | |
m._transitioning = true; | |
} | |
function arrive(StateMachine storage m, bytes31 to) { | |
checkTransitioning(m, true); | |
m._transitioning = false; | |
m._state = to; | |
} | |
function move(StateMachine storage m, bytes31 from, bytes31 to) { | |
checkState(m, from); | |
m._state = to; | |
} | |
error StateViolation(string current, string expected); | |
error AtomicityViolation(string current); | |
function checkTransitioning(StateMachine storage m, bool transitioning) view { | |
if (m._transitioning != transitioning) { | |
revert AtomicityViolation(toString(m._state)); | |
} | |
} | |
function checkState(StateMachine storage m, bytes31 expected) view { | |
if (m._state != expected) { | |
revert StateViolation(toString(m._state), toString(expected)); | |
} | |
checkTransitioning(m, false); | |
} | |
function isState(StateMachine storage m, bytes31 expected) view returns (bool) { | |
return !m._transitioning && m._state == expected; | |
} |
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
// SPDX-License-Identifier: MIT | |
pragma solidity 0.8.16; | |
function toString(bytes32 s) pure returns (string memory) { | |
uint x = uint(s); | |
uint len = 0; | |
for (uint m = 0xff << 248; x & m != 0 && m != 0; m >>= 8) { | |
len += 1; | |
} | |
bytes memory b = new bytes(32); | |
assembly { | |
mstore(b, len) | |
mstore(add(b, 32), s) | |
} | |
return string(b); | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment