def print_memfunc(f, in_size, out_size, per_reg=256, initialize=False):
"""Wraps a function that operates on registers in .data and .text sections,
and makes it operate on memory fragments instead."""
in_data = [MemoryFragment(per_reg, '{}(%rsi)'.format(per_reg*i // 8))
for i in range(in_size)]
out_data = [MemoryFragment(per_reg, '{}(%rdi)'.format(per_reg*i // 8))
for i in range(in_size)]
if initialize:
utils.sequence_to_values(in_data, range(0, 509), padding=ZERO)
instructions.reset()
data.reset()
f(out_data, in_data)
print(".data")
print(".p2align 5")
for mask in data.DATASECTION:
print(mask.data())
print(".text")
print(".global {}".format(f.__name__))
print(".global _{}".format(f.__name__))
print("{}:".format(f.__name__))
print("_{}:".format(f.__name__))
for ins in instructions.INSTRUCTIONS:
print(ins)
print("ret")
def test_square_821_patience_callee_save(self):
src = [MemoryFragment(64, '{}(%rsi)'.format(i * 8)) for i in range(16)]
dst = [MemoryFragment(64, '{}(%rdi)'.format(i * 8)) for i in range(16)]
seq = sqGF2N.gen_sequence(5, 821)
sequence_to_values(src, range(0, 821), padding=ZERO)
sqGF2N.square_821_patience(dst, src, 5, 5)
result = sqGF2N.registers_to_sequence(dst)
self.assertEqual(result, seq)
def test_square_350_701(self):
src = [Register(256) for _ in range(3)]
dst = [Register(256) for _ in range(3)]
seq = sqGF2N.gen_sequence(350, 701)
sequence_to_values(src, range(0, 701), padding=ZERO)
sqGF2N.square_350_701(dst, src)
result = sqGF2N.registers_to_sequence(dst)
self.assertEqual(result, seq)
def test_square_509_patience(self):
src = [MemoryFragment(64, '{}(%rsi)'.format(i * 8)) for i in range(8)]
dst = [MemoryFragment(64, '{}(%rdi)'.format(i * 8)) for i in range(8)]
for n in range(10):
seq = sqGF2N.gen_sequence(n, 509)
sequence_to_values(src, range(0, 509), padding=ZERO)
sqGF2N.square_509_patience(dst, src, n)
result = sqGF2N.registers_to_sequence(dst)
if self.assertEqual(result, seq):
break
def test_square_1_701(self):
src = [
MemoryFragment(256, '{}(%rsi)'.format(i * 32)) for i in range(3)
]
dst = [
MemoryFragment(256, '{}(%rdi)'.format(i * 32)) for i in range(3)
]
seq = sqGF2N.gen_sequence(1, 701)
sequence_to_values(src, range(0, 701), padding=ZERO)
sqGF2N.square_1_701(dst, src)
result = sqGF2N.registers_to_sequence(dst)
self.assertEqual(result, seq)
def test_square_821_shufbytes(self):
src = [
MemoryFragment(256, '{}(%rsi)'.format(i * 32)) for i in range(4)
]
dst = [
MemoryFragment(256, '{}(%rdi)'.format(i * 32)) for i in range(4)
]
for n in [1, 3, 6, 12, 24, 48, 51, 102, 204, 408]:
seq = sqGF2N.gen_sequence(n, 821)
sequence_to_values(src, range(0, 821), padding=ZERO)
sqGF2N.square_821_shufbytes(dst, src, n)
result = sqGF2N.registers_to_sequence(dst)
if self.assertEqual(result, seq):
break
def test_square_509_shufbytes(self):
src = [
MemoryFragment(256, '{}(%rsi)'.format(i * 32)) for i in range(2)
]
dst = [
MemoryFragment(256, '{}(%rdi)'.format(i * 32)) for i in range(2)
]
for n in [1, 3, 6, 15, 30, 63, 126, 252]:
seq = sqGF2N.gen_sequence(n, 509)
sequence_to_values(src, range(0, 509), padding=ZERO)
sqGF2N.square_509_shufbytes(dst, src, n)
result = sqGF2N.registers_to_sequence(dst)
if self.assertEqual(result, seq):
break
def test_square_701_shufbytes(self):
src = [
MemoryFragment(256, '{}(%rsi)'.format(i * 32)) for i in range(3)
]
dst = [
MemoryFragment(256, '{}(%rdi)'.format(i * 32)) for i in range(3)
]
for n in [1, 3, 6, 12, 15, 27, 42, 84, 168, 336]:
seq = sqGF2N.gen_sequence(n, 701)
sequence_to_values(src, range(0, 701), padding=ZERO)
sqGF2N.square_701_shufbytes(dst, src, n)
result = sqGF2N.registers_to_sequence(dst)
if self.assertEqual(result, seq):
break
