def test_vec0(self):
vm = VM()
sha3.ctx_new(vm)
sha3.ctx_finish(vm)
real_hash = int.from_bytes(hashlib.sha3_256().digest(),
byteorder="big")
self.assertEqual(vm.stack[0], real_hash)
def test_random_keccak256(self):
data = bytearray(random.getrandbits(8) for _ in range(64))
vm = VM()
sha3.ctx_new(vm)
for v in data:
vm.push(v)
vm.swap1()
sha3.ctx_pushbyte(vm)
real_hash = int.from_bytes(eth_utils.crypto.keccak(data),
byteorder="big")
sha3.keccak_ctx_finish(vm)
self.assertEqual(real_hash, vm.stack[0])
def test_vec1(self):
vm = VM()
sha3.ctx_new(vm)
hasher = hashlib.sha3_256()
for i in range(200):
vm.push(0xa3)
vm.swap1()
sha3.ctx_pushbyte(vm)
hasher.update(bytes([0xa3]))
real_hash = int.from_bytes(hasher.digest(), byteorder="big")
sha3.ctx_finish(vm)
self.assertEqual(vm.stack[0], real_hash)
def test_random_sha3(self):
data = bytearray(random.getrandbits(8) for _ in range(64))
vm = VM()
sha3.ctx_new(vm)
hasher = hashlib.sha3_256()
hasher.update(data)
for v in data:
vm.push(v)
vm.swap1()
sha3.ctx_pushbyte(vm)
real_hash = int.from_bytes(hasher.digest(), byteorder="big")
sha3.ctx_finish(vm)
self.assertEqual(vm.stack[0], real_hash)
def test_pushblock(self):
data = bytearray(random.getrandbits(8) for _ in range(200))
vm1 = VM()
sha3.ctx_new(vm1)
for v in range(136):
vm1.push(data[v])
vm1.swap1()
sha3.ctx_pushbyte(vm1)
vm2 = VM()
vm2.push(0)
vm2.push(byterange.frombytes(data))
sha3.byterange_get136(vm2)
sha3.ctx_new(vm2)
sha3.ctx_pushblock(vm2)
self.assertEqual(vm1.stack[0], vm2.stack[0])