From c24b883c01fc932dd28add1ac0319459a007d498 Mon Sep 17 00:00:00 2001 From: Mathias Hall-Andersen Date: Fri, 1 Sep 2017 14:31:57 +0200 Subject: Fixed KDF tests --- src/kdf_test.go | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'src/kdf_test.go') diff --git a/src/kdf_test.go b/src/kdf_test.go index 0cce81d..a89dacc 100644 --- a/src/kdf_test.go +++ b/src/kdf_test.go @@ -2,6 +2,7 @@ package main import ( "encoding/hex" + "golang.org/x/crypto/blake2s" "testing" ) @@ -44,10 +45,12 @@ func TestKDF(t *testing.T) { }, } + var t0, t1, t2 [blake2s.Size]byte + for _, test := range tests { key, _ := hex.DecodeString(test.key) input, _ := hex.DecodeString(test.input) - t0, t1, t2 := KDF3(key, input) + KDF3(&t0, &t1, &t2, key, input) t0s := hex.EncodeToString(t0[:]) t1s := hex.EncodeToString(t1[:]) t2s := hex.EncodeToString(t2[:]) @@ -59,7 +62,7 @@ func TestKDF(t *testing.T) { for _, test := range tests { key, _ := hex.DecodeString(test.key) input, _ := hex.DecodeString(test.input) - t0, t1 := KDF2(key, input) + KDF2(&t0, &t1, key, input) t0s := hex.EncodeToString(t0[:]) t1s := hex.EncodeToString(t1[:]) assertEquals(t, t0s, test.t0) @@ -69,7 +72,7 @@ func TestKDF(t *testing.T) { for _, test := range tests { key, _ := hex.DecodeString(test.key) input, _ := hex.DecodeString(test.input) - t0 := KDF1(key, input) + KDF1(&t0, key, input) t0s := hex.EncodeToString(t0[:]) assertEquals(t, t0s, test.t0) } -- cgit v1.2.3-59-g8ed1b