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 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156
|
package x448
import (
"crypto/rand"
"encoding/hex"
"encoding/json"
"flag"
"fmt"
"io"
"io/ioutil"
"os"
"testing"
"github.com/cloudflare/circl/internal/test"
)
func hexStr2Key(k *Key, s string) {
b, err := hex.DecodeString(s)
if err != nil {
panic("Can't convert string to key")
}
copy(k[:], b)
}
// Indicates whether long tests should be run
var runLongTest = flag.Bool("long", false, "runs longer tests")
type katVector struct {
Public string `json:"input"`
Shared string `json:"output"`
Private string `json:"scalar"`
}
func TestRFC7748Kat(t *testing.T) {
const nameFile = "testdata/rfc7748_kat_test.json"
var kat []katVector
jsonFile, err := os.Open(nameFile)
if err != nil {
t.Fatalf("File %v can not be opened. Error: %v", nameFile, err)
}
defer jsonFile.Close()
input, _ := ioutil.ReadAll(jsonFile)
err = json.Unmarshal(input, &kat)
if err != nil {
t.Fatalf("File %v can not be loaded. Error: %v", nameFile, err)
}
var priv, pub, got, want Key
for _, v := range kat {
hexStr2Key(&pub, v.Public)
hexStr2Key(&priv, v.Private)
Shared(&got, &priv, &pub)
hexStr2Key(&want, v.Shared)
if got != want {
test.ReportError(t, got, want, v)
}
}
}
type katTimes struct {
Times uint32 `json:"times"`
Key string `json:"key"`
}
func TestRFC7748Times(t *testing.T) {
const nameFile = "testdata/rfc7748_times_test.json"
jsonFile, err := os.Open(nameFile)
if err != nil {
t.Fatalf("File %v can not be opened. Error: %v", nameFile, err)
}
defer jsonFile.Close()
input, _ := ioutil.ReadAll(jsonFile)
var kat []katTimes
err = json.Unmarshal(input, &kat)
if err != nil {
t.Fatalf("File %v can not be loaded. Error: %v", nameFile, err)
}
var got, want Key
for _, v := range kat {
if !*runLongTest && v.Times == uint32(1000000) {
t.Log("Skipped one long test, add -long flag to run longer tests")
continue
}
u := Key{5}
k := u
r := u
for i := uint32(0); i < v.Times; i++ {
Shared(&r, &k, &u)
u = k
k = r
}
got = k
hexStr2Key(&want, v.Key)
if got != want {
test.ReportError(t, got, want, v.Times)
}
}
}
func TestBase(t *testing.T) {
testTimes := 1 << 10
var got, want, secret Key
gen := Key{5}
for i := 0; i < testTimes; i++ {
_, _ = io.ReadFull(rand.Reader, secret[:])
KeyGen(&got, &secret)
Shared(&want, &secret, &gen)
if got != want {
test.ReportError(t, got, want, secret)
}
}
}
func BenchmarkX448(b *testing.B) {
var x, y, z Key
_, _ = io.ReadFull(rand.Reader, x[:])
_, _ = io.ReadFull(rand.Reader, y[:])
_, _ = io.ReadFull(rand.Reader, z[:])
b.Run("KeyGen", func(b *testing.B) {
for i := 0; i < b.N; i++ {
KeyGen(&x, &y)
}
})
b.Run("Shared", func(b *testing.B) {
for i := 0; i < b.N; i++ {
Shared(&z, &x, &y)
}
})
}
func Example_x448() {
var AliceSecret, BobSecret,
AlicePublic, BobPublic,
AliceShared, BobShared Key
// Generating Alice's secret and public keys
_, _ = io.ReadFull(rand.Reader, AliceSecret[:])
KeyGen(&AlicePublic, &AliceSecret)
// Generating Bob's secret and public keys
_, _ = io.ReadFull(rand.Reader, BobSecret[:])
KeyGen(&BobPublic, &BobSecret)
// Deriving Alice's shared key
okA := Shared(&AliceShared, &AliceSecret, &BobPublic)
// Deriving Bob's shared key
okB := Shared(&BobShared, &BobSecret, &AlicePublic)
fmt.Println(AliceShared == BobShared && okA && okB)
// Output: true
}
|