2015-10-25 13:45:13 -07:00
|
|
|
package crypto
|
2015-10-25 13:42:49 -07:00
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
|
2017-02-22 23:43:26 +01:00
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/stretchr/testify/assert"
|
2015-10-25 13:42:49 -07:00
|
|
|
)
|
|
|
|
|
2016-04-19 01:02:31 -07:00
|
|
|
func TestSignAndValidateEd25519(t *testing.T) {
|
2015-10-25 13:42:49 -07:00
|
|
|
|
|
|
|
privKey := GenPrivKeyEd25519()
|
2018-06-10 10:01:41 +02:00
|
|
|
pubKey, err := privKey.PubKey()
|
|
|
|
require.Nil(t, err)
|
2015-10-25 13:42:49 -07:00
|
|
|
|
|
|
|
msg := CRandBytes(128)
|
2018-06-10 10:01:41 +02:00
|
|
|
sig, err := privKey.Sign(msg)
|
|
|
|
require.Nil(t, err)
|
2015-10-25 13:42:49 -07:00
|
|
|
|
|
|
|
// Test the signature
|
2017-02-23 13:20:30 +01:00
|
|
|
assert.True(t, pubKey.VerifyBytes(msg, sig))
|
2015-10-25 13:42:49 -07:00
|
|
|
|
|
|
|
// Mutate the signature, just one bit.
|
2018-01-14 00:31:39 -08:00
|
|
|
sigEd := sig.(SignatureEd25519)
|
2017-03-21 21:44:24 +01:00
|
|
|
sigEd[7] ^= byte(0x01)
|
2018-01-14 00:31:39 -08:00
|
|
|
sig = sigEd
|
2015-10-25 13:42:49 -07:00
|
|
|
|
2017-02-23 13:20:30 +01:00
|
|
|
assert.False(t, pubKey.VerifyBytes(msg, sig))
|
2015-10-25 13:42:49 -07:00
|
|
|
}
|
|
|
|
|
2016-04-19 01:02:31 -07:00
|
|
|
func TestSignAndValidateSecp256k1(t *testing.T) {
|
|
|
|
privKey := GenPrivKeySecp256k1()
|
2018-06-10 10:01:41 +02:00
|
|
|
pubKey, err := privKey.PubKey()
|
|
|
|
require.Nil(t, err)
|
2016-04-19 01:02:31 -07:00
|
|
|
|
|
|
|
msg := CRandBytes(128)
|
2018-06-10 10:01:41 +02:00
|
|
|
sig, err := privKey.Sign(msg)
|
|
|
|
require.Nil(t, err)
|
2016-04-19 01:02:31 -07:00
|
|
|
|
2017-02-23 13:20:30 +01:00
|
|
|
assert.True(t, pubKey.VerifyBytes(msg, sig))
|
2016-04-19 01:02:31 -07:00
|
|
|
|
|
|
|
// Mutate the signature, just one bit.
|
2018-01-14 00:31:39 -08:00
|
|
|
sigEd := sig.(SignatureSecp256k1)
|
2017-03-21 21:44:24 +01:00
|
|
|
sigEd[3] ^= byte(0x01)
|
2018-01-14 00:31:39 -08:00
|
|
|
sig = sigEd
|
2016-04-19 01:02:31 -07:00
|
|
|
|
2017-02-23 13:20:30 +01:00
|
|
|
assert.False(t, pubKey.VerifyBytes(msg, sig))
|
2016-04-19 01:02:31 -07:00
|
|
|
}
|