Reduce allocations for Reset()

dedis · Mar 28, 2024 · 009ea8c · 009ea8c
1 parent cb72df8
commit 009ea8c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/xof/blake2xb/blake.go b/xof/blake2xb/blake.go
@@ -66,8 +66,10 @@ func (x *xof) Reseed() {
 }
 
 func (x *xof) Reset() {
-	y := New(x.seed)
-	x.impl = y.(*xof).impl
+	x.impl.Reset()
+	if len(x.seed) > blake2b.Size {
+		x.impl.Write(x.seed[blake2b.Size:])
+	}
 }
 
 func (x *xof) XORKeyStream(dst, src []byte) {

diff --git a/xof/blake2xs/blake.go b/xof/blake2xs/blake.go
@@ -66,8 +66,10 @@ func (x *xof) Reseed() {
 }
 
 func (x *xof) Reset() {
-	y := New(x.seed)
-	x.impl = y.(*xof).impl
+	x.impl.Reset()
+	if len(x.seed) > blake2s.Size {
+		x.impl.Write(x.seed[blake2s.Size:])
+	}
 }
 
 func (x *xof) XORKeyStream(dst, src []byte) {