WIP wrapped hash

Consensys · Oct 23, 2023 · 54c2c35 · 54c2c35
1 parent 7a6d821
commit 54c2c35
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/std/recursion/wrapped_hash.go b/std/recursion/wrapped_hash.go
@@ -16,6 +16,8 @@ import (
 type shortNativeHash struct {
 	bitLength int
 	wrapped   hash.Hash
+
+	buf []byte
 }
 
 // NewShort returns a native hash function which reads elements in the current native
@@ -62,11 +64,14 @@ func newShortFromParam(hf hash.Hash, bitLength int) hash.Hash {
 	return &shortNativeHash{
 		bitLength: bitLength,
 		wrapped:   hf,
+
+		buf: make([]byte, hf.BlockSize()),
 	}
 }
 
 func (h *shortNativeHash) Write(p []byte) (n int, err error) {
-	return h.wrapped.Write(p)
+	h.buf = append(h.buf[:0], p...)
+	return h.wrapped.Write(h.buf)
 }
 
 func (h *shortNativeHash) Sum(b []byte) []byte {