File: es256k_test.go

package info (click to toggle)
golang-github-lestrrat-go-jwx 2.1.4-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,872 kB
  • sloc: sh: 222; makefile: 86; perl: 62
file content (68 lines) | stat: -rw-r--r-- 1,548 bytes parent folder | download
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
//go:build jwx_es256k
// +build jwx_es256k

package jwk_test

import (
	"crypto/ecdsa"
	"encoding/base64"
	"encoding/json"
	"math/big"
	"testing"

	"github.com/decred/dcrd/dcrec/secp256k1/v4"
	"github.com/lestrrat-go/jwx/v2/internal/ecutil"
	"github.com/lestrrat-go/jwx/v2/jwa"
	"github.com/lestrrat-go/jwx/v2/jwk"
	"github.com/stretchr/testify/require"
)

func TestES256K(t *testing.T) {
	require.True(t, ecutil.IsAvailable(jwa.Secp256k1), `jwa.Secp256k1 should be available`)
}

func BenchmarkKeyInstantiation(b *testing.B) {
	const xb64 = "YAXIamcY9mIhcTp3BzxBKRzDq7_NA6pJVemytQ2_f5s"
	const yb64 = "ZnLa0NRq3mHjgveYiKc-p4mdlBm-zx1snsIIfBGI-hg"

	x, err := base64.RawURLEncoding.DecodeString(xb64)
	require.NoError(b, err, `DecodeBase64 should succeed`)
	y, err := base64.RawURLEncoding.DecodeString(yb64)
	require.NoError(b, err, `DecodeBase64 should succeed`)

	b.Run("Use json.Marshal/json.Unmarshal", func(b *testing.B) {
		for i := 0; i < b.N; i++ {
			serialized, err := json.Marshal(map[string]interface{}{
				"kty": "EC",
				"crv": "secp256k1",
				"x":   xb64,
				"y":   yb64,
			})
			if err != nil {
				panic(err)
			}

			key, err := jwk.Parse(serialized)
			if err != nil {
				panic(err)
			}
			_ = key
		}
	})
	b.Run("Use jwk.FromRaw", func(b *testing.B) {
		for i := 0; i < b.N; i++ {
			var raw ecdsa.PublicKey
			raw.Curve = secp256k1.S256()
			raw.X = &big.Int{}
			raw.Y = &big.Int{}
			raw.X.SetBytes(x)
			raw.Y.SetBytes(y)

			key, err := jwk.FromRaw(&raw)
			if err != nil {
				panic(err)
			}
			_ = key
		}
	})
}