sm3: kdf, reduce memory allocation times #220

emmansun · May 15, 2024 · 105331f · 105331f
1 parent 4fe8d28
commit 105331f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sm3/sm3.go b/sm3/sm3.go
@@ -220,7 +220,7 @@ func Kdf(z []byte, keyLen int) []byte {
 	}
 	var countBytes [4]byte
 	var ct uint32 = 1
-	var k []byte
+	k := make([]byte, keyLen)
 	baseMD := new(digest)
 	baseMD.Reset()
 	baseMD.Write(z)
@@ -229,8 +229,8 @@ func Kdf(z []byte, keyLen int) []byte {
 		md := *baseMD
 		md.Write(countBytes[:])
 		h := md.checkSum()
-		k = append(k, h[:]...)
+		copy(k[i*Size:], h[:])
 		ct++
 	}
-	return k[:keyLen]
+	return k
 }