mirror of
https://github.com/ethereum/solidity
synced 2023-10-03 13:03:40 +00:00
46 lines
987 B
Python
46 lines
987 B
Python
import sys
|
|
|
|
from z3 import sat, Solver, unknown, unsat
|
|
|
|
class Rule:
|
|
def __init__(self):
|
|
self.requirements = []
|
|
self.constraints = []
|
|
self.solver = Solver()
|
|
self.setTimeout(60000)
|
|
|
|
def setTimeout(self, _t):
|
|
self.solver.set("timeout", _t)
|
|
|
|
def __lshift__(self, _c):
|
|
self.constraints.append(_c)
|
|
|
|
def require(self, _r):
|
|
self.requirements.append(_r)
|
|
|
|
def check(self, _nonopt, _opt):
|
|
self.solver.add(self.requirements)
|
|
result = self.solver.check()
|
|
|
|
if result == unknown:
|
|
self.error('Unable to satisfy requirements.')
|
|
elif result == unsat:
|
|
self.error('Requirements are unsatisfiable.')
|
|
|
|
self.solver.push()
|
|
self.solver.add(self.constraints)
|
|
self.solver.add(_nonopt != _opt)
|
|
|
|
result = self.solver.check()
|
|
if result == unknown:
|
|
self.error('Unable to prove rule.')
|
|
elif result == sat:
|
|
m = self.solver.model()
|
|
self.error('Rule is incorrect.\nModel: ' + str(m))
|
|
self.solver.pop()
|
|
|
|
@classmethod
|
|
def error(cls, msg):
|
|
print(msg)
|
|
sys.exit(1)
|