150 lines
3.7 KiB
Go
150 lines
3.7 KiB
Go
/*-
|
|
* Copyright 2014 Square Inc.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package josecipher
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto"
|
|
"testing"
|
|
)
|
|
|
|
// Taken from: https://tools.ietf.org/id/draft-ietf-jose-json-web-algorithms-38.txt
|
|
func TestVectorConcatKDF(t *testing.T) {
|
|
z := []byte{
|
|
158, 86, 217, 29, 129, 113, 53, 211, 114, 131, 66, 131, 191, 132,
|
|
38, 156, 251, 49, 110, 163, 218, 128, 106, 72, 246, 218, 167, 121,
|
|
140, 254, 144, 196}
|
|
|
|
algID := []byte{0, 0, 0, 7, 65, 49, 50, 56, 71, 67, 77}
|
|
|
|
ptyUInfo := []byte{0, 0, 0, 5, 65, 108, 105, 99, 101}
|
|
ptyVInfo := []byte{0, 0, 0, 3, 66, 111, 98}
|
|
|
|
supPubInfo := []byte{0, 0, 0, 128}
|
|
supPrivInfo := []byte{}
|
|
|
|
expected := []byte{
|
|
86, 170, 141, 234, 248, 35, 109, 32, 92, 34, 40, 205, 113, 167, 16, 26}
|
|
|
|
ckdf := NewConcatKDF(crypto.SHA256, z, algID, ptyUInfo, ptyVInfo, supPubInfo, supPrivInfo)
|
|
|
|
out0 := make([]byte, 9)
|
|
out1 := make([]byte, 7)
|
|
|
|
read0, err := ckdf.Read(out0)
|
|
if err != nil {
|
|
t.Error("error when reading from concat kdf reader", err)
|
|
return
|
|
}
|
|
|
|
read1, err := ckdf.Read(out1)
|
|
if err != nil {
|
|
t.Error("error when reading from concat kdf reader", err)
|
|
return
|
|
}
|
|
|
|
if read0+read1 != len(out0)+len(out1) {
|
|
t.Error("did not receive enough bytes from concat kdf reader")
|
|
return
|
|
}
|
|
|
|
out := []byte{}
|
|
out = append(out, out0...)
|
|
out = append(out, out1...)
|
|
|
|
if bytes.Compare(out, expected) != 0 {
|
|
t.Error("did not receive expected output from concat kdf reader")
|
|
return
|
|
}
|
|
}
|
|
|
|
func TestCache(t *testing.T) {
|
|
z := []byte{
|
|
158, 86, 217, 29, 129, 113, 53, 211, 114, 131, 66, 131, 191, 132,
|
|
38, 156, 251, 49, 110, 163, 218, 128, 106, 72, 246, 218, 167, 121,
|
|
140, 254, 144, 196}
|
|
|
|
algID := []byte{1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4}
|
|
|
|
ptyUInfo := []byte{1, 2, 3, 4}
|
|
ptyVInfo := []byte{4, 3, 2, 1}
|
|
|
|
supPubInfo := []byte{}
|
|
supPrivInfo := []byte{}
|
|
|
|
outputs := [][]byte{}
|
|
|
|
// Read the same amount of data in different chunk sizes
|
|
chunkSizes := []int{1, 2, 4, 8, 16, 32, 64, 128, 256, 512}
|
|
|
|
for _, c := range chunkSizes {
|
|
out := make([]byte, 1024)
|
|
reader := NewConcatKDF(crypto.SHA256, z, algID, ptyUInfo, ptyVInfo, supPubInfo, supPrivInfo)
|
|
|
|
for i := 0; i < 1024; i += c {
|
|
_, _ = reader.Read(out[i : i+c])
|
|
}
|
|
|
|
outputs = append(outputs, out)
|
|
}
|
|
|
|
for i := range outputs {
|
|
if bytes.Compare(outputs[i], outputs[(i+1)%len(outputs)]) != 0 {
|
|
t.Error("not all outputs from KDF matched")
|
|
}
|
|
}
|
|
}
|
|
|
|
func benchmarkKDF(b *testing.B, total int) {
|
|
z := []byte{
|
|
158, 86, 217, 29, 129, 113, 53, 211, 114, 131, 66, 131, 191, 132,
|
|
38, 156, 251, 49, 110, 163, 218, 128, 106, 72, 246, 218, 167, 121,
|
|
140, 254, 144, 196}
|
|
|
|
algID := []byte{1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4}
|
|
|
|
ptyUInfo := []byte{1, 2, 3, 4}
|
|
ptyVInfo := []byte{4, 3, 2, 1}
|
|
|
|
supPubInfo := []byte{}
|
|
supPrivInfo := []byte{}
|
|
|
|
out := make([]byte, total)
|
|
reader := NewConcatKDF(crypto.SHA256, z, algID, ptyUInfo, ptyVInfo, supPubInfo, supPrivInfo)
|
|
|
|
b.ResetTimer()
|
|
b.SetBytes(int64(total))
|
|
for i := 0; i < b.N; i++ {
|
|
_, _ = reader.Read(out)
|
|
}
|
|
}
|
|
|
|
func BenchmarkConcatKDF_1k(b *testing.B) {
|
|
benchmarkKDF(b, 1024)
|
|
}
|
|
|
|
func BenchmarkConcatKDF_64k(b *testing.B) {
|
|
benchmarkKDF(b, 65536)
|
|
}
|
|
|
|
func BenchmarkConcatKDF_1MB(b *testing.B) {
|
|
benchmarkKDF(b, 1048576)
|
|
}
|
|
|
|
func BenchmarkConcatKDF_64MB(b *testing.B) {
|
|
benchmarkKDF(b, 67108864)
|
|
}
|