sm3: kdf, reduce memory allocation times #220

This commit is contained in:
Sun Yimin 2024-05-15 09:36:38 +08:00 committed by GitHub
parent 4fe8d28d31
commit 105331f164
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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
}