ots/ots_test.go at master · mooorex/ots · GitHub

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
103
104
105
106
107
108
109
110
111
112
package OTS

import (
	"bytes"
	"crypto/rand"
	"testing"
)

func TestSign(t *testing.T) {
	params, err := Setup(rand.Reader)
	if err != nil {
		t.Fatal(err)
	}

	vk, sk, err := params.KeyGen(rand.Reader)
	if err != nil {
		t.Fatal(err)
	}

	msg := []byte("ssssssssss")

	sig, err := sk.Sign(rand.Reader, msg)
	if err != nil {
		t.Fatal(err)
	}
	if !sig.Verify(params, vk, msg) {
		t.Fatalf("emmmm, bug!")
	}
}

func TestVKMarshal(t *testing.T) {
	params, err := Setup(rand.Reader)
	if err != nil {
		t.Fatal(err)
	}
	vk, _, err := params.KeyGen(rand.Reader)
	if err != nil {
		t.Fatal(err)
	}

	vkm := vk.Marshal()
	vk1, err := new(VerKey).Unmarshal(vkm)
	if err != nil || !bytes.Equal(vk1.Marshal(), vkm) {
		t.Log(vk1.Marshal())
		t.Log(vkm)
		t.Fatalf("emmm, bug")
	}
}

func BenchmarkKeyGen(b *testing.B) {
	params, err := Setup(rand.Reader)
	if err != nil {
		b.Fatal(err)
	}

	for i := 0; i < b.N; i++ {
		params.KeyGen(rand.Reader)
	}
}

func BenchmarkSign(b *testing.B) {
	params, err := Setup(rand.Reader)
	if err != nil {
		b.Fatal(err)
	}
	_, sk, err := params.KeyGen(rand.Reader)
	if err != nil {
		b.Fatal(err)
	}

	msg := []byte("ssssssssss")

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		sk.Sign(rand.Reader, msg)
	}
}

func BenchmarkVerify(b *testing.B) {
	params, err := Setup(rand.Reader)
	if err != nil {
		b.Fatal(err)
	}
	vk, sk, err := params.KeyGen(rand.Reader)
	if err != nil {
		b.Fatal(err)
	}

	msg := []byte("ssssssssss")
	sig, err := sk.Sign(rand.Reader, msg)

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		sig.Verify(params, vk, msg)
	}
}

func BenchmarkVKMarshal(b *testing.B) {
	params, err := Setup(rand.Reader)
	if err != nil {
		b.Fatal(err)
	}
	vk, _, err := params.KeyGen(rand.Reader)
	if err != nil {
		b.Fatal(err)
	}

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		vk.Marshal()
	}
}