1package blindsecp256k1
  2
  3import (
  4	"encoding/hex"
  5	"encoding/json"
  6	"math/big"
  7	"testing"
  8
  9	"github.com/ethereum/go-ethereum/crypto"
 10	"github.com/stretchr/testify/assert"
 11	"github.com/stretchr/testify/require"
 12)
 13
 14func TestMarshalers(t *testing.T) {
 15	// Point
 16	p := G.Mul(big.NewInt(1234))
 17	b, err := json.Marshal(p)
 18	require.Nil(t, err)
 19	assert.Equal(t,
 20		`{"x":"102884003323827292915668239759940053105992008087520207150474896054185180420338","y":"49384988101491619794462775601349526588349137780292274540231125201115197157452"}`, //nolint:lll
 21		string(b))
 22
 23	var p2 *Point
 24	err = json.Unmarshal(b, &p2)
 25	require.Nil(t, err)
 26	assert.Equal(t, p, p2)
 27
 28	// PublicKey
 29	pk := PublicKey(*p)
 30	b, err = json.Marshal(pk)
 31	require.Nil(t, err)
 32	assert.Equal(t,
 33		`{"x":"102884003323827292915668239759940053105992008087520207150474896054185180420338","y":"49384988101491619794462775601349526588349137780292274540231125201115197157452"}`, //nolint:lll
 34		string(b))
 35
 36	var pk2 PublicKey
 37	err = json.Unmarshal(b, &pk2)
 38	require.Nil(t, err)
 39	assert.Equal(t, pk, pk2)
 40
 41	// Signature
 42	sig := Signature{
 43		S: big.NewInt(9876),
 44		F: p,
 45	}
 46	b, err = json.Marshal(sig)
 47	require.Nil(t, err)
 48	assert.Equal(t,
 49		`{"s":"9876","f":{"x":"102884003323827292915668239759940053105992008087520207150474896054185180420338","y":"49384988101491619794462775601349526588349137780292274540231125201115197157452"}}`, //nolint:lll
 50		string(b))
 51
 52	var sig2 Signature
 53	err = json.Unmarshal(b, &sig2)
 54	require.Nil(t, err)
 55	assert.Equal(t, sig, sig2)
 56}
 57
 58func TestBytes(t *testing.T) {
 59	// Point
 60	p := &Point{
 61		X: big.NewInt(3),
 62		Y: big.NewInt(3),
 63	}
 64	b := p.Bytes()
 65	assert.Equal(t, "03000000000000000000000000000000000000000000000000000000000000000300000000000000000000000000000000000000000000000000000000000000", hex.EncodeToString(b)) //nolint:lll
 66	p2, err := NewPointFromBytes(b)
 67	assert.Nil(t, err)
 68	assert.Equal(t, p, p2)
 69
 70	p = G.Mul(big.NewInt(1234))
 71	b = p.Bytes()
 72	assert.Equal(t, "f258163f65f65865a79a4279e2ebabb5a57b85501dd4b381d1dc605c434876e34c308bd3f18f062d5cc07f34948ced82f9a76f9c3e65ae64f158412da8e92e6d", hex.EncodeToString(b)) //nolint:lll
 73	p2, err = NewPointFromBytes(b)
 74	assert.Nil(t, err)
 75	assert.Equal(t, p, p2)
 76
 77	// PublicKey
 78	pk := PublicKey(*p)
 79	b = pk.Bytes()
 80	assert.Equal(t, "f258163f65f65865a79a4279e2ebabb5a57b85501dd4b381d1dc605c434876e34c308bd3f18f062d5cc07f34948ced82f9a76f9c3e65ae64f158412da8e92e6d", hex.EncodeToString(b)) //nolint:lll
 81	pk2, err := NewPublicKeyFromBytes(b)
 82	assert.Nil(t, err)
 83	assert.Equal(t, &pk, pk2)
 84
 85	// Signature
 86	sig := Signature{
 87		S: big.NewInt(9876),
 88		F: p,
 89	}
 90	b = sig.Bytes()
 91	assert.Equal(t, "9426000000000000000000000000000000000000000000000000000000000000f258163f65f65865a79a4279e2ebabb5a57b85501dd4b381d1dc605c434876e34c308bd3f18f062d5cc07f34948ced82f9a76f9c3e65ae64f158412da8e92e6d", hex.EncodeToString(b)) //nolint:lll
 92	sig2, err := NewSignatureFromBytes(b)
 93	assert.Nil(t, err)
 94	assert.Equal(t, &sig, sig2)
 95
 96	// Signature with bigger values
 97	s, ok := new(big.Int).SetString("43744879514016998261043792362491545206150700367692876136431010903034023684055", 10) //nolint:lll
 98	require.True(t, ok)
 99	x, ok := new(big.Int).SetString("56183217574518331862027285308947626162625485037257226169003339923450551228164", 10) //nolint:lll
100	require.True(t, ok)
101	y, ok := new(big.Int).SetString("62825693913681695979055350889339417157462875026935818721506450621762231021976", 10) //nolint:lll
102	require.True(t, ok)
103	sig = Signature{
104		S: s,
105		F: &Point{X: x, Y: y},
106	}
107	b = sig.Bytes()
108	assert.Equal(t, "d7a75050259cc06415f19bde5460a58325e3050806ba949d9ac9728b71b9b6600457ba001981781ed31acafed3d1e82c2ad53d08e3f293eab2f199ed0193367c98311f1894598c91f10fe415ba4a6d04e1351d07430631c7decdbbdb2615e68a", hex.EncodeToString(b)) //nolint:lll
109	sig2, err = NewSignatureFromBytes(b)
110	assert.Nil(t, err)
111	assert.Equal(t, &sig, sig2)
112}
113
114func TestImportECDSApubKey(t *testing.T) {
115	// Generate an ECDSA key
116	k, err := crypto.GenerateKey()
117	assert.Nil(t, err)
118	// Import the ECDSA Public key bytes into a PublicKey type
119	pk, err := NewPublicKeyFromECDSA(crypto.FromECDSAPub(&k.PublicKey))
120	assert.Nil(t, err)
121	// Set the ECDSA Private key point as a blindsecp256k1 PrivateKey type
122	bk := PrivateKey(*k.D)
123	// Compare both public keys
124	assert.Equal(t, bk.Public().Bytes(), pk.Bytes())
125}