def knuth():
dig = DIG('../invgen/Traces/NLA/knuth.tcs')
rs = dig.getInvs(inv_typ='eqt', seed=0, vs=dig.vs, deg=3)
assert list_str(
rs
) == '-k^2*t + k*t^2 == 0, 1/8*d^2*q + 1/2*d*k - 1/4*d*q - 1/2*d*rvu - nvu + rvu == 0, -k^2*t + t^3 == 0', rs
def paper_multidim():
"""
TODO: sort the result so that it has the same output
"""
dig = DIG('../invgen/Traces/AES/Simple/paper_multidim.tc')
rs = dig.getInvs(inv_typ='simple', seed=0)
#first result
mrs = rs[0].inv #IExp
expect_rs0a = 'lambda A, B, A0: (-A[A0]) + (7*B[2*A0]) + (3*A0) == 0'
expect_rs0b = 'lambda A, B, A0: (A[A0]) + (-7*B[2*A0]) + (-3*A0) == 0'
expect_rs1 = [{'A0': 0}, {'A0': 2}, {'A0': 1}]
assert mrs[0] in [expect_rs0a, expect_rs0b], rs[0]
mrs1 = [tuple(d.items()) for d in mrs[1]]
expect_rs1 = [tuple(d.items()) for d in expect_rs1]
assert set(mrs1) == set(expect_rs1), mrs1
#second result
mrs = rs[1].inv #IExp
expect_rs0a = 'lambda B, A, B0: (7*B[B0]) + (-A[1/2*B0]) + (3/2*B0) == 0'
expect_rs0b = 'lambda B, A, B0: (-7*B[B0]) + (A[1/2*B0]) + (-3/2*B0) == 0'
expect_rs1 = [{'B0': 0}, {'B0': 4}, {'B0': 2}]
assert mrs[0] in [expect_rs0a, expect_rs0b], rs[0]
mrs1 = [tuple(d.items()) for d in mrs[1]]
expect_rs1 = [tuple(d.items()) for d in expect_rs1]
assert set(mrs1) == set(expect_rs1), mrs1
def cohendiv():
dig = DIG('../invgen/Traces/NLA/cohendiv.tcs')
rs = dig.getInvs(inv_typ='eqt', seed=0, vs=dig.vs, deg=2)
q,y,rvu,x,a,b = var('q,y,rvu,x,a,b')
expected_rs = [-q*y - rvu + x == 0, -a*y + b == 0, -q*y - 2*b + x >= 0, -2*a*y + rvu >= 0]
assert set(expected_rs) == set([r.inv for r in rs]), rs
def paper_multidim():
"""
TODO: sort the result so that it has the same output
"""
dig = DIG('../invgen/Traces/AES/Simple/paper_multidim.tc')
rs = dig.getInvs(inv_typ='simple',seed=0)
#first result
mrs = rs[0].inv #IExp
expect_rs0a = 'lambda A, B, A0: (-A[A0]) + (7*B[2*A0]) + (3*A0) == 0'
expect_rs0b = 'lambda A, B, A0: (A[A0]) + (-7*B[2*A0]) + (-3*A0) == 0'
expect_rs1 = [{'A0': 0}, {'A0': 2}, {'A0': 1}]
assert mrs[0] in [expect_rs0a, expect_rs0b], rs[0]
mrs1 = [tuple(d.items()) for d in mrs[1]]
expect_rs1 = [tuple(d.items()) for d in expect_rs1]
assert set(mrs1) == set(expect_rs1), mrs1
#second result
mrs = rs[1].inv #IExp
expect_rs0a = 'lambda B, A, B0: (7*B[B0]) + (-A[1/2*B0]) + (3/2*B0) == 0'
expect_rs0b = 'lambda B, A, B0: (-7*B[B0]) + (A[1/2*B0]) + (-3/2*B0) == 0'
expect_rs1 = [{'B0': 0}, {'B0': 4}, {'B0': 2}]
assert mrs[0] in [expect_rs0a, expect_rs0b], rs[0]
mrs1 = [tuple(d.items()) for d in mrs[1]]
expect_rs1 = [tuple(d.items()) for d in expect_rs1]
assert set(mrs1) == set(expect_rs1), mrs1
def cohendiv():
dig = DIG('../invgen/Traces/NLA/cohendiv.tcs')
rs = dig.getInvs(inv_typ='eqt', seed=0, vs=dig.vs, deg=2)
q, y, rvu, x, a, b = var('q,y,rvu,x,a,b')
expected_rs = [
-q * y - rvu + x == 0, -a * y + b == 0, -q * y - 2 * b + x >= 0,
-2 * a * y + rvu >= 0
]
assert set(expected_rs) == set([r.inv for r in rs]), rs
def init_hw(self):
''' initialize hardware interfaces. should never be called
except on initialization or by the reset function, else competing
for hardware resources
returns log string and status for awg and digitizer as tuples
'''
awgModules, awg_log, awg_status = self.open_modules(
self.awg_slots, 'awg')
digModules, dig_log, dig_status = self.open_modules(
self.dig_slots, 'dig')
if self.__initialized == False:
self.hvi = pyhvi.KtHvi("KtHvi")
self.hvi.platform.chassis.add_auto_detect()
index = 0
for awgModule in awgModules:
awg = AWG(self.hvi, awgModule, index)
self.awgs.append(awg)
index += 1
for digModule in digModules:
dig = DIG(self.hvi, digModule, index)
self.digs.append(dig)
index += 1
return (awg_log, awg_status), (dig_log, dig_status)
def aes_KeySetupEnc6():
dig = DIG('../invgen/Traces/AES/Simple/aes_KeySetupEnc6.tc')
rs = dig.getInvs(inv_typ='simple',seed=0)
rs = rs[0].inv #IExp
expect_rs0a = 'lambda rk, cipherKey, rk1, rk0: (-rk[rk0][rk1]) + (cipherKey[4*rk0 + rk1]) == 0'
expect_rs0b = 'lambda rk, cipherKey, rk1, rk0: (rk[rk0][rk1]) + (-cipherKey[4*rk0 + rk1]) == 0'
expect_rs0c = 'lambda rk, cipherKey, rk0, rk1: (-rk[rk0][rk1]) + (cipherKey[4*rk0 + rk1]) == 0'
expect_rs0d = 'lambda rk, cipherKey, rk0, rk1: (rk[rk0][rk1]) + (-cipherKey[4*rk0 + rk1]) == 0'
expect_rs1 = [{'rk0': 4, 'rk1': 1},
{'rk0': 0, 'rk1': 1},
{'rk0': 3, 'rk1': 0},
{'rk0': 1, 'rk1': 1},
{'rk0': 2, 'rk1': 3},
{'rk0': 1, 'rk1': 0},
{'rk0': 5, 'rk1': 2},
{'rk0': 3, 'rk1': 3},
{'rk0': 4, 'rk1': 3},
{'rk0': 5, 'rk1': 0},
{'rk0': 0, 'rk1': 3},
{'rk0': 5, 'rk1': 3},
{'rk0': 1, 'rk1': 3},
{'rk0': 3, 'rk1': 1},
{'rk0': 2, 'rk1': 1},
{'rk0': 0, 'rk1': 2},
{'rk0': 2, 'rk1': 0},
{'rk0': 2, 'rk1': 2},
{'rk0': 0, 'rk1': 0},
{'rk0': 5, 'rk1': 1},
{'rk0': 3, 'rk1': 2},
{'rk0': 1, 'rk1': 2},
{'rk0': 4, 'rk1': 0},
{'rk0': 4, 'rk1': 2}]
assert rs[0] in [expect_rs0a, expect_rs0b, expect_rs0c, expect_rs0d], rs[0]
rs1 = [tuple(d.items()) for d in rs[1]]
expect_rs1 = [tuple(d.items()) for d in expect_rs1]
assert set(rs1) == set(expect_rs1), rs1
def aes_KeySetupEnc6():
dig = DIG('../invgen/Traces/AES/Simple/aes_KeySetupEnc6.tc')
rs = dig.getInvs(inv_typ='simple', seed=0)
rs = rs[0].inv #IExp
expect_rs0a = 'lambda rk, cipherKey, rk1, rk0: (-rk[rk0][rk1]) + (cipherKey[4*rk0 + rk1]) == 0'
expect_rs0b = 'lambda rk, cipherKey, rk1, rk0: (rk[rk0][rk1]) + (-cipherKey[4*rk0 + rk1]) == 0'
expect_rs0c = 'lambda rk, cipherKey, rk0, rk1: (-rk[rk0][rk1]) + (cipherKey[4*rk0 + rk1]) == 0'
expect_rs0d = 'lambda rk, cipherKey, rk0, rk1: (rk[rk0][rk1]) + (-cipherKey[4*rk0 + rk1]) == 0'
expect_rs1 = [{
'rk0': 4,
'rk1': 1
}, {
'rk0': 0,
'rk1': 1
}, {
'rk0': 3,
'rk1': 0
}, {
'rk0': 1,
'rk1': 1
}, {
'rk0': 2,
'rk1': 3
}, {
'rk0': 1,
'rk1': 0
}, {
'rk0': 5,
'rk1': 2
}, {
'rk0': 3,
'rk1': 3
}, {
'rk0': 4,
'rk1': 3
}, {
'rk0': 5,
'rk1': 0
}, {
'rk0': 0,
'rk1': 3
}, {
'rk0': 5,
'rk1': 3
}, {
'rk0': 1,
'rk1': 3
}, {
'rk0': 3,
'rk1': 1
}, {
'rk0': 2,
'rk1': 1
}, {
'rk0': 0,
'rk1': 2
}, {
'rk0': 2,
'rk1': 0
}, {
'rk0': 2,
'rk1': 2
}, {
'rk0': 0,
'rk1': 0
}, {
'rk0': 5,
'rk1': 1
}, {
'rk0': 3,
'rk1': 2
}, {
'rk0': 1,
'rk1': 2
}, {
'rk0': 4,
'rk1': 0
}, {
'rk0': 4,
'rk1': 2
}]
assert rs[0] in [expect_rs0a, expect_rs0b, expect_rs0c, expect_rs0d], rs[0]
rs1 = [tuple(d.items()) for d in rs[1]]
expect_rs1 = [tuple(d.items()) for d in expect_rs1]
assert set(rs1) == set(expect_rs1), rs1
def sqrt1():
dig = DIG('../invgen/Traces/NLA/sqrt1.tcs')
rs = dig.getInvs(inv_typ='eqt', seed=0, vs=dig.vs, deg=2)
assert list_str(
rs
) == '-1/4*t^2 - a + s - 3/4 == 0, -2*a + t - 1 == 0, 1/4*t^2 + a - nvu + 3/4 <= 0', rs
def aes_addroundkey_vn():
dig = DIG('../invgen/Traces/AES/Nested/aes_addroundkey_vn.tc')
rs = dig.getInvs(inv_typ='nested', seed=0)
assert rs[
0].inv == 'lambda r_,xor,st,rk,i1,i2: r_[i1][i2] == xor(st[i1][i2],rk[i1][i2])', rs[
0].inv
def paper_nested():
dig = DIG('../invgen/Traces/AES/Nested/paper_nested.tc')
rs = dig.getInvs(inv_typ='nested', seed=1)
assert rs[0].inv == 'lambda A,B,C,i1: A[i1] == B[C[2*i1 + 1]]', rs[0].inv
assert rs[1].inv == 'lambda A,B,i1: A[i1] == B[-2*i1 + 5]', rs[1].inv
def benchmark_file(filename, runs, use_specific_deg, read_only, do_parallel):
def get_deg_auto(nss, nts=200, max_deg=7):
if __debug__:
assert nss >= 1, nss
assert nts >= nss, (nts, nss)
assert max_deg >= 1, max_deg
for d in range(1, max_deg + 1):
deg = d
if binomial(nss + deg, deg) - 1 > nts:
deg = d - 1
break
return deg
def get_deg_specific(filename):
deg3 = ['dijkstra', 'cohencu', 'prod4br', 'knuth', 'geo3', 'ps3']
deg4 = ['ps4']
deg5 = ['ps5']
deg6 = ['ps6']
deg7 = ['ps7']
if any(x in filename for x in deg7):
return 7
elif any(x in filename for x in deg6):
return 6
elif any(x in filename for x in deg5):
return 5
elif any(x in filename for x in deg4):
return 4
elif any(x in filename for x in deg3):
return 3
else:
return 2
def get_invs(ig, seed, deg):
start_time = time.time()
ig.set_seed(seed)
#For MPP
#rs = ig.get_ieqs_max_min_fixed()
# if len(ig.ss_num) <= 3:
# print 'also getting general form, {} vars'.format(len(ig.ss_num))
# rs_ = ig.get_ieqs_max_min_gen()
# rs = rs_ + rs
# rs = vset(rs)
# print len(rs)
#DO NOT USE fixed_3, will generate
#lots of formulas that screw up Z3
#stick with fixed_2
#For NLA
#rs = ig.get_eqts_ieqs_max_min_fixed_2(deg=deg)
#eqts
rs = ig.get_invs(deg=deg)
return rs, (time.time() - start_time)
sols_filename = []
start_time = time.time()
ig = DIG(filename)
print "Read time {}".format(time.time() - start_time)
if read_only:
return '', [], []
if len(ig.ss_num) <= 1:
deg = 1
else:
if any(x in filename
for x in ['AesKeySetupEnc_w_LRA', 'AesKeySetupDec_w_LRA']):
deg = 1
elif use_specific_deg:
deg = get_deg_specific(filename)
print 'Use *specific* max degree {}'.format(deg)
else:
if 'l_init' in filename:
deg = 2
print 'Use default max deg {} for l_init'.format(deg)
else:
deg = get_deg_auto(len(ig.ss_num))
print 'Use *auto* max deg {}'.format(deg)
sols_run = []
for nr in range(runs):
print '\n--- Run {}/{} ({}) ---\n'.format(nr + 1, runs, filename)
rs, etime = get_invs(ig, seed=nr, deg=deg)
print etime
sols_filename = sols_filename + [(nr, rs, deg, etime)]
sols_run = sols_run + rs
sols_run = vset(sols_run)
expect = ig.xinfo['Expect']
print_summary(filename, expect, sols_filename, sols_run)
return expect, sols_filename, sols_run
def paper_nested():
dig = DIG('../invgen/Traces/AES/Nested/paper_nested.tc')
rs = dig.getInvs(inv_typ='nested',seed=1)
assert rs[0].inv == 'lambda A,B,C,i1: A[i1] == B[C[2*i1 + 1]]', rs[0].inv
assert rs[1].inv == 'lambda A,B,i1: A[i1] == B[-2*i1 + 5]', rs[1].inv
def knuth():
dig = DIG('../invgen/Traces/NLA/knuth.tcs')
rs = dig.getInvs(inv_typ='eqt', seed=0, vs=dig.vs, deg=3)
assert list_str(rs) =='-k^2*t + k*t^2 == 0, 1/8*d^2*q + 1/2*d*k - 1/4*d*q - 1/2*d*rvu - nvu + rvu == 0, -k^2*t + t^3 == 0', rs
def sqrt1():
dig = DIG('../invgen/Traces/NLA/sqrt1.tcs')
rs = dig.getInvs(inv_typ='eqt', seed=0, vs=dig.vs, deg=2)
assert list_str(rs) == '-1/4*t^2 - a + s - 3/4 == 0, -2*a + t - 1 == 0, 1/4*t^2 + a - nvu + 3/4 <= 0', rs
def aes_addroundkey_vn():
dig = DIG('../invgen/Traces/AES/Nested/aes_addroundkey_vn.tc')
rs = dig.getInvs(inv_typ='nested',seed=0)
assert rs[0].inv == 'lambda r_,xor,st,rk,i1,i2: r_[i1][i2] == xor(st[i1][i2],rk[i1][i2])', rs[0].inv
