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
|
package x25519_test
import (
"encoding/hex"
"testing"
"github.com/lestrrat-go/jwx/v2/x25519"
"github.com/stretchr/testify/assert"
)
func TestGenerateKey(t *testing.T) {
t.Run("x25519.GenerateKey(nil)", func(t *testing.T) {
_, _, err := x25519.GenerateKey(nil)
if !assert.NoError(t, err, `x25519.GenerateKey should work even if argument is nil`) {
return
}
})
t.Run("x25519.NewKeyFromSeed(wrongSeedLength)", func(t *testing.T) {
dummy := make([]byte, x25519.SeedSize-1)
_, err := x25519.NewKeyFromSeed(dummy)
if !assert.Error(t, err, `wrong seed size should result in error`) {
return
}
})
}
func TestNewKeyFromSeed(t *testing.T) {
// These test vectors are from RFC7748 Section 6.1
const alicePrivHex = `77076d0a7318a57d3c16c17251b26645df4c2f87ebc0992ab177fba51db92c2a`
const alicePubHex = `8520f0098930a754748b7ddcb43ef75a0dbf3a0d26381af4eba4a98eaa9b4e6a`
const bobPrivHex = `5dab087e624a8a4b79e17f8b83800ee66f3bb1292618b6fd1c2f8b27ff88e0eb`
const bobPubHex = `de9edb7d7b7dc1b4d35b61c2ece435373f8343c85b78674dadfc7e146f882b4f`
alicePrivSeed, err := hex.DecodeString(alicePrivHex)
if !assert.NoError(t, err, `alice seed decoded`) {
return
}
alicePriv, err := x25519.NewKeyFromSeed(alicePrivSeed)
if !assert.NoError(t, err, `alice private key`) {
return
}
alicePub := alicePriv.Public().(x25519.PublicKey)
if !assert.Equal(t, hex.EncodeToString(alicePub), alicePubHex, `alice public key`) {
return
}
bobPrivSeed, err := hex.DecodeString(bobPrivHex)
if !assert.NoError(t, err, `bob seed decoded`) {
return
}
bobPriv, err := x25519.NewKeyFromSeed(bobPrivSeed)
if !assert.NoError(t, err, `bob private key`) {
return
}
bobPub := bobPriv.Public().(x25519.PublicKey)
if !assert.Equal(t, hex.EncodeToString(bobPub), bobPubHex, `bob public key`) {
return
}
if !assert.True(t, bobPriv.Equal(bobPriv), `bobPriv should equal bobPriv`) {
return
}
if !assert.True(t, bobPub.Equal(bobPub), `bobPub should equal bobPub`) {
return
}
if !assert.False(t, bobPriv.Equal(bobPub), `bobPriv should NOT equal bobPub`) {
return
}
if !assert.False(t, bobPub.Equal(bobPriv), `bobPub should NOT equal bobPriv`) {
return
}
}
|
