Keccak256 host

op-program/client/io_wasm.go

@@ -75,7 +75,7 @@ func (o wasmHostIO) Get(key preimage.Key) []byte {
 	// Integrity check
 	// TODO: can use customized circuit to optimize
 	if !_isPublic {
-		//hash := crypto.Keccak256Hash(buf)
+		// hash := crypto.Keccak256Hash(buf)
 		hash := Keccak256Hash(buf)
 		hash[0] = _key[0]
 		require_bool(hash == _key)

op-program/client/keccak256_wasm.go

@@ -55,15 +55,15 @@ func (kh *KeccakHasher) UpdateByte(v byte) {
 func (kh *KeccakHasher) Finalize() [4]uint64 {
 	bytesToPad := 136 - kh.byteIdx - kh.bufSize*8
 	if bytesToPad == 1 {
-		var result uint64 = 0x86 << 56
+		var result uint64 = 0x81 << 56
 		keccak_push(kh.data + result)
 	} else {
 		kh.UpdateByte(1)
 		for i := 0; i < int(bytesToPad)-2; i++ {
 			kh.UpdateByte(0)
 		}
 		var result uint64 = 0x80 << 56
-		keccak_push(kh.data + result)
+		keccak_push(kh.data ^ result)
 	}
 
 	return [4]uint64{
@@ -88,6 +88,20 @@ func Keccak256Hash(data ...[]byte) (output [32]byte) {
 	return output
 }
 
+func Keccak256HashSimple(data ...[]byte) (output [32]byte) {
+	hasher := NewKeccakHasher()
+	for _, value := range data {
+		for _, byteValue := range value {
+			hasher.UpdateByte(byteValue)
+		}
+	}
+	result := hasher.Finalize()
+	for i, val := range result {
+		binary.LittleEndian.PutUint64(output[i*8:], val)
+	}
+	return output
+}
+
 /*
 // for test