81 lines
1.7 KiB
Go
81 lines
1.7 KiB
Go
|
package blake3_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/hex"
|
||
|
|
"encoding/json"
|
||
|
|
"io/ioutil"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"lukechampine.com/blake3"
|
||
|
|
)
|
||
|
|
|
||
|
|
func toHex(data []byte) string {
|
||
|
|
return hex.EncodeToString(data)
|
||
|
|
}
|
||
|
|
|
||
|
|
func fromHex(s string) []byte {
|
||
|
|
data, err := hex.DecodeString(s)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
return data
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestVectors(t *testing.T) {
|
||
|
|
data, err := ioutil.ReadFile("testdata/vectors.json")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
var vectors struct {
|
||
|
|
Key string
|
||
|
|
Cases []struct {
|
||
|
|
InputLen int `json:"input_len"`
|
||
|
|
Hash string `json:"hash"`
|
||
|
|
KeyedHash string `json:"keyed_hash"`
|
||
|
|
DeriveKey string `json:"derive_key"`
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if err := json.Unmarshal(data, &vectors); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
input := make([]byte, 1<<15)
|
||
|
|
for i := range input {
|
||
|
|
input[i] = byte(i % 251)
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, vec := range vectors.Cases {
|
||
|
|
// regular
|
||
|
|
h := blake3.New(len(vec.Hash)/2, nil)
|
||
|
|
h.Write(input[:vec.InputLen])
|
||
|
|
if out := toHex(h.Sum(nil)); out != vec.Hash {
|
||
|
|
t.Errorf("output did not match test vector:\n\texpected: %v...\n\t got: %v...", vec.Hash[:10], out[:10])
|
||
|
|
}
|
||
|
|
// keyed
|
||
|
|
h = blake3.New(len(vec.KeyedHash)/2, []byte(vectors.Key))
|
||
|
|
h.Write(input[:vec.InputLen])
|
||
|
|
if out := toHex(h.Sum(nil)); out != vec.KeyedHash {
|
||
|
|
t.Errorf("output did not match test vector:\n\texpected: %v...\n\t got: %v...", vec.KeyedHash[:10], out[:10])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func BenchmarkWrite(b *testing.B) {
|
||
|
|
h := blake3.New(32, nil)
|
||
|
|
buf := make([]byte, 1<<15)
|
||
|
|
b.SetBytes(int64(len(buf)))
|
||
|
|
for i := 0; i < b.N; i++ {
|
||
|
|
h.Write(buf)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func BenchmarkBlock(b *testing.B) {
|
||
|
|
h := blake3.New(32, nil)
|
||
|
|
buf := make([]byte, h.BlockSize())
|
||
|
|
out := make([]byte, 32)
|
||
|
|
for i := 0; i < b.N; i++ {
|
||
|
|
h.Write(buf)
|
||
|
|
h.Sum(out)
|
||
|
|
}
|
||
|
|
}
|