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