1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
package drbg
import (
"bytes"
"encoding/hex"
"testing"
)
func S2H(s string) []byte {
hex, e := hex.DecodeString(s)
if e != nil {
panic("Can't import private key")
}
return hex
}
func TestNominal(t *testing.T) {
var entropy [16]byte
var data [48]byte
c := NewCtrDrbg()
if !c.Init(entropy[:], nil) {
t.FailNow()
}
c.ReadWithAdditionalData(entropy[0:16], data[:])
exp := S2H("16BA361FA14563FB1E8BCF88932F9FA7")
if !bytes.Equal(exp, entropy[:]) {
t.FailNow()
}
}
// TODO: should parse *.req file from here: https://raw.githubusercontent.com/coruus/nist-testvectors/master/csrc.nist.gov/groups/STM/cavp/documents/drbg/drbgtestvectors/drbgvectors_pr_false/CTR_DRBG.rsp
var vectors = []struct {
EntropyInput []byte
PersonalizationString []byte
EntropyInputReseed []byte
AdditionalInputReseed []byte
AdditionalInput1 []byte
AdditionalInput2 []byte
ReturnedBits []byte
}{
// With Reseeding
{
S2H("99903165903fea49c2db26ed675e44cc14cb2c1f28b836b203240b02771e831146ffc4335373bb344688c5c950670291"),
[]byte{},
S2H("b4ee99fa9e0eddaf4a3612013cd636c4af69177b43eebb3c58a305b9979b68b5cc820504f6c029aad78a5d29c66e84a0"),
S2H("2d8c5c28b05696e74774eb69a10f01c5fabc62691ddf7848a8004bb5eeb4d2c5febe1aa01f4d557b23d7e9a0e4e90655"),
S2H("0dc9cde42ac6e856f01a55f219c614de90c659260948db5053d414bab0ec2e13e995120c3eb5aafc25dc4bdcef8ace24"),
S2H("711be6c035013189f362211889248ca8a3268e63a7eb26836d915810a680ac4a33cd1180811a31a0f44f08db3dd64f91"),
S2H("11c7a0326ea737baa7a993d510fafee5374e7bbe17ef0e3e29f50fa68aac2124b017d449768491cac06d136d691a4e80785739f9aaedf311bba752a3268cc531"),
},
{
S2H("ffad10100025a879672ff50374b286712f457dd01441d76ac1a1cd15c7390dd93179a2f5920d198bf34a1b76fbc21289"),
S2H("1d2be6f25e88fa30c4ef42e4d54efd957dec231fa00143ca47580be666a8c143a916c90b3819a0a7ea914e3c9a2e7a3f"),
S2H("6c1a089cae313363bc76a780139eb4f2f2048b1f6b07896c5c412bff0385440fc43b73facbb79e3a252fa01fe17ab391"),
[]byte{},
[]byte{},
[]byte{},
S2H("e053c7d4bd9099ef6a99f190a5fd80219437d642006672338da6e0fe73ca4d24ffa51151bfbdac78d8a2f6255046edf57a04626e9977139c6933274299f3bdff"),
},
}
func TestVector(t *testing.T) {
for i := range vectors {
result := make([]byte, len(vectors[i].ReturnedBits))
c := NewCtrDrbg()
if !c.Init(vectors[i].EntropyInput[:], vectors[i].PersonalizationString) {
t.Error("Init failed")
}
if len(vectors[i].EntropyInputReseed) > 0 {
c.Reseed(vectors[i].EntropyInputReseed[:], vectors[i].AdditionalInputReseed[:])
}
c.ReadWithAdditionalData(result[:], vectors[i].AdditionalInput1)
c.ReadWithAdditionalData(result[:], vectors[i].AdditionalInput2)
if !bytes.Equal(vectors[i].ReturnedBits[:], result[:]) {
t.Errorf("KAT failed \nexp: %X\ngot: %X\n", vectors[i].ReturnedBits, result[:])
}
}
}
func BenchmarkInit(b *testing.B) {
c := NewCtrDrbg()
for i := 0; i < b.N; i++ {
c.Init(vectors[0].EntropyInput[:], vectors[0].PersonalizationString)
}
}
func BenchmarkRead(b *testing.B) {
var result [16 * 10]byte
c := NewCtrDrbg()
c.Init(vectors[0].EntropyInput[:], vectors[0].PersonalizationString)
for i := 0; i < b.N; i++ {
c.ReadWithAdditionalData(result[:], vectors[0].AdditionalInput1)
}
}
|
