solidity/test/formal/mod_add_to_addmod.py
2022-06-22 09:26:09 +02:00

32 lines
458 B
Python

from opcodes import MOD, ADD, ADDMOD
from rule import Rule
from z3 import BitVec
"""
Rule:
MOD(ADD(X, Y), A) -> ADDMOD(X, Y, A)
given
A > 0
A & (A - 1) == 0
"""
rule = Rule()
n_bits = 32
# Input vars
X = BitVec('X', n_bits)
Y = BitVec('Y', n_bits)
A = BitVec('A', n_bits)
# Non optimized result
nonopt = MOD(ADD(X, Y), A)
# Optimized result
opt = ADDMOD(X, Y, A)
rule.require(A > 0)
rule.require(((A & (A - 1)) == 0))
rule.check(nonopt, opt)