NewWizardHarness

This commit is contained in:
Aleksander Kryukov
2021-11-17 11:56:24 +02:00
parent eb27bdd282
commit 44113d58f5
7 changed files with 183 additions and 12 deletions

View File

@ -0,0 +1,5 @@
import "../../contracts/token/ERC20/extensions/ERC20Votes.sol";
contract ERC20VotesHarness is ERC20Votes {
constructor(string memory name) ERC20Permit(name) {}
}

View File

@ -33,16 +33,19 @@ contract GovernorBasicHarness is Governor, GovernorCountingSimple, GovernorVotes
uint256 _votingDelay; uint256 _votingDelay;
function votingDelay() public view override virtual returns (uint256) { function votingDelay() public view override virtual returns (uint256) { // HARNESS: pure -> view
return _votingDelay; return _votingDelay;
} }
uint256 _votingPeriod; uint256 _votingPeriod;
function votingPeriod() public view override virtual returns (uint256) { function votingPeriod() public view override virtual returns (uint256) { // HARNESS: pure -> view
return _votingPeriod; return _votingPeriod;
} }
function snapshot(uint256 proposalId) public view returns (uint64) {
return _proposals[proposalId].voteStart._deadline;
}
mapping(uint256 => uint256) public ghost_sum_vote_power_by_id; mapping(uint256 => uint256) public ghost_sum_vote_power_by_id;
@ -65,13 +68,15 @@ contract GovernorBasicHarness is Governor, GovernorCountingSimple, GovernorVotes
return super.propose(targets, values, calldatas, ""); return super.propose(targets, values, calldatas, "");
} }
/* // Harness of castVoteWithReason to be able to impose requirement on the proposal ID.
mapping (address => mapping (uint256 => uint256)) _getVotes; uint256 public _pId_Harness;
function castVoteWithReason(uint256 proposalId, uint8 support, string calldata reason)
function getVotesHarnness(address account, uint256 blockNumber) public { public
_getVotes[account][blockNumber] = getVotes(account, blockNumber); override(IGovernor, Governor)
returns (uint256) {
require(proposalId == _pId_Harness);
return super.castVoteWithReason(proposalId, support, reason);
} }
*/
// The following functions are overrides required by Solidity. // The following functions are overrides required by Solidity.

View File

@ -0,0 +1,152 @@
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.2;
import "../../contracts/governance/Governor.sol";
import "../../contracts/governance/extensions/GovernorCountingSimple.sol";
import "../../contracts/governance/extensions/GovernorVotes.sol";
import "../../contracts/governance/extensions/GovernorVotesQuorumFraction.sol";
import "../../contracts/governance/extensions/GovernorTimelockControl.sol";
import "../../contracts/governance/extensions/GovernorProposalThreshold.sol";
/*
Wizard options:
ProposalThreshhold = 10
ERC20Votes
TimelockCOntroller
*/
contract WizardHarness1 is Governor, GovernorProposalThreshold, GovernorCountingSimple, GovernorVotes, GovernorVotesQuorumFraction, GovernorTimelockControl {
constructor(ERC20Votes _token, TimelockController _timelock, string memory name, uint256 quorumFraction)
Governor(name)
GovernorVotes(_token)
GovernorVotesQuorumFraction(quorumFraction)
GovernorTimelockControl(_timelock)
{}
//HARNESS
function isExecuted(uint256 proposalId) public view returns (bool) {
return _proposals[proposalId].executed;
}
function isCanceled(uint256 proposalId) public view returns (bool) {
return _proposals[proposalId].canceled;
}
uint256 _votingDelay;
uint256 _votingPeriod;
uint256 _proposalThreshold;
mapping(uint256 => uint256) public ghost_sum_vote_power_by_id;
function _castVote(
uint256 proposalId,
address account,
uint8 support,
string memory reason
) internal override virtual returns (uint256) {
uint256 deltaWeight = super._castVote(proposalId, account, support, reason); //HARNESS
ghost_sum_vote_power_by_id[proposalId] += deltaWeight;
return deltaWeight;
}
function callPropose(address[] memory targets,
uint256[] memory values,
bytes[] memory calldatas) public virtual returns (uint256) {
return super.propose(targets, values, calldatas, "");
}
function snapshot(uint256 proposalId) public view returns (uint64) {
return _proposals[proposalId].voteStart._deadline;
}
// original code
function votingDelay() public view override returns (uint256) { // HARNESS: pure -> view
return _votingDelay; // HARNESS: parametric
}
function votingPeriod() public view override returns (uint256) { // HARNESS: pure -> view
return _votingPeriod; // HARNESS: parametric
}
function proposalThreshold() public view override returns (uint256) { // HARNESS: pure -> view
return _proposalThreshold; // HARNESS: parametric
}
// The following functions are overrides required by Solidity.
function quorum(uint256 blockNumber)
public
view
override(IGovernor, GovernorVotesQuorumFraction)
returns (uint256)
{
return super.quorum(blockNumber);
}
function getVotes(address account, uint256 blockNumber)
public
view
override(IGovernor, GovernorVotes)
returns (uint256)
{
return super.getVotes(account, blockNumber);
}
function state(uint256 proposalId)
public
view
override(Governor, GovernorTimelockControl)
returns (ProposalState)
{
return super.state(proposalId);
}
function propose(address[] memory targets, uint256[] memory values, bytes[] memory calldatas, string memory description)
public
override(Governor, GovernorProposalThreshold, IGovernor)
returns (uint256)
{
return super.propose(targets, values, calldatas, description);
}
function _execute(uint256 proposalId, address[] memory targets, uint256[] memory values, bytes[] memory calldatas, bytes32 descriptionHash)
internal
override(Governor, GovernorTimelockControl)
{
super._execute(proposalId, targets, values, calldatas, descriptionHash);
}
function _cancel(address[] memory targets, uint256[] memory values, bytes[] memory calldatas, bytes32 descriptionHash)
internal
override(Governor, GovernorTimelockControl)
returns (uint256)
{
return super._cancel(targets, values, calldatas, descriptionHash);
}
function _executor()
internal
view
override(Governor, GovernorTimelockControl)
returns (address)
{
return super._executor();
}
function supportsInterface(bytes4 interfaceId)
public
view
override(Governor, GovernorTimelockControl)
returns (bool)
{
return super.supportsInterface(interfaceId);
}
}

View File

@ -1,8 +1,8 @@
certoraRun certora/harnesses/GovernorBasicHarness.sol \ certoraRun certora/harnesses/GovernorBasicHarness.sol \
--verify GovernorBasicHarness:certora/specs/GovernorBase.spec \ --verify GovernorBasicHarness:certora/specs/GovernorBase.spec \
--solc solc8.0 \ --solc solc8.2 \
---staging shelly/stringCVL \ --staging shelly/forSasha \
--optimistic_loop \ --optimistic_loop \
--settings -copyLoopUnroll=4 \ --settings -copyLoopUnroll=4 \
--rule unaffectedThreshhold \ --rule allFunctionsRevertIfCanceled \
--msg "$1" --msg "$1"

View File

@ -1,7 +1,7 @@
certoraRun certora/harnesses/GovernorBasicHarness.sol \ certoraRun certora/harnesses/GovernorBasicHarness.sol \
--verify GovernorBasicHarness:certora/specs/GovernorCountingSimple.spec \ --verify GovernorBasicHarness:certora/specs/GovernorCountingSimple.spec \
--solc solc8.2 \ --solc solc8.2 \
--staging alex/external-timeout-for-solvers \ --staging shelly/forSasha \
--optimistic_loop \ --optimistic_loop \
--settings -copyLoopUnroll=4 \ --settings -copyLoopUnroll=4 \
--rule hasVotedCorrelation \ --rule hasVotedCorrelation \

View File

@ -0,0 +1,8 @@
certoraRun certora/harnesses/WizardHarness1.sol \
--verify WizardHarness1:certora/specs/GovernorBase.spec \
--solc solc8.2 \
--staging shelly/forSasha \
--optimistic_loop \
--settings -copyLoopUnroll=4 \
--rule allFunctionsRevertIfCanceled \
--msg "$1"

View File

@ -274,6 +274,7 @@ rule hasVotedCorrelation(uint256 pId, method f, env e, uint256 bn) filtered {f -
// how to check executor()? // how to check executor()?
// to make it public instead of internal is not best idea, I think. // to make it public instead of internal is not best idea, I think.
// currentContract gives a violation in // currentContract gives a violation in
// maybe need harness implementation for one of the contracts
rule privilegedOnly(method f){ rule privilegedOnly(method f){
env e; env e;
calldataarg arg; calldataarg arg;