diff options
| -rw-r--r-- | hkdf/hkdf.go | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/hkdf/hkdf.go b/hkdf/hkdf.go index dda3f14..f4ded5f 100644 --- a/hkdf/hkdf.go +++ b/hkdf/hkdf.go @@ -56,7 +56,9 @@ func (f *hkdf) Read(p []byte) (int, error) { // Fill the rest of the buffer for len(p) > 0 { - f.expander.Reset() + if f.counter > 1 { + f.expander.Reset() + } f.expander.Write(f.prev) f.expander.Write(f.info) f.expander.Write([]byte{f.counter}) |