2018-09-24 18:07:34 +00:00
|
|
|
package crypto
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/hex"
|
2018-09-27 13:07:32 +00:00
|
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
|
|
"github.com/stretchr/testify/require"
|
2018-09-24 18:07:34 +00:00
|
|
|
"testing"
|
|
|
|
)
|
|
|
|
|
2018-10-04 16:53:48 +00:00
|
|
|
func TestExtractSignatures(t *testing.T) {
|
|
|
|
const content1 = "045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b04ca82dd41fa592bf46ecf7e2eddae61013fc95a565b59c49f37f06b1b591ed3bd24e143495f2d1e241e151ab3572ac108d577be349d4b88d3d5a50c481ab35441"
|
|
|
|
const content2 = "045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b04ca82dd41fa592bf46ecf7e2eddae61013fc95a565b59c49f37f06b1b591ed3bd24e143495f2d1e241e151ab3572ac108d577be349d4b88d3d5a50c481ab35440"
|
|
|
|
|
|
|
|
key1, err := crypto.GenerateKey()
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
key2, err := crypto.GenerateKey()
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
signature1, err := Sign(content1, key1)
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
signature2, err := Sign(content2, key2)
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
key1String := hex.EncodeToString(crypto.FromECDSAPub(&key1.PublicKey))
|
|
|
|
key2String := hex.EncodeToString(crypto.FromECDSAPub(&key2.PublicKey))
|
|
|
|
|
|
|
|
pair1 := [2]string{content1, signature1}
|
|
|
|
pair2 := [2]string{content2, signature2}
|
|
|
|
|
|
|
|
signaturePairs := [][2]string{pair1, pair2}
|
|
|
|
|
|
|
|
extractedSignatures, err := ExtractSignatures(signaturePairs)
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, []string{key1String, key2String}, extractedSignatures)
|
|
|
|
|
|
|
|
// Test wrong content
|
|
|
|
pair3 := [2]string{content1, signature2}
|
|
|
|
|
|
|
|
signaturePairs = [][2]string{pair1, pair2, pair3}
|
|
|
|
|
|
|
|
extractedSignatures, err = ExtractSignatures(signaturePairs)
|
|
|
|
require.NoError(t, err)
|
|
|
|
// The public key is neither the one which generated the content, nor the one generated the signature
|
|
|
|
require.NotEqual(t, []string{key1String, key2String, key1String}, extractedSignatures)
|
|
|
|
require.NotEqual(t, []string{key1String, key2String, key2String}, extractedSignatures)
|
|
|
|
}
|
|
|
|
|
2018-09-27 13:07:32 +00:00
|
|
|
func TestVerifySignature(t *testing.T) {
|
|
|
|
const content1 = "045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b04ca82dd41fa592bf46ecf7e2eddae61013fc95a565b59c49f37f06b1b591ed3bd24e143495f2d1e241e151ab3572ac108d577be349d4b88d3d5a50c481ab35441"
|
|
|
|
const content2 = "045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b04ca82dd41fa592bf46ecf7e2eddae61013fc95a565b59c49f37f06b1b591ed3bd24e143495f2d1e241e151ab3572ac108d577be349d4b88d3d5a50c481ab35440"
|
|
|
|
|
|
|
|
key1, err := crypto.GenerateKey()
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
key2, err := crypto.GenerateKey()
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
signature1, err := Sign(content1, key1)
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
signature2, err := Sign(content2, key2)
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
key1String := hex.EncodeToString(crypto.FromECDSAPub(&key1.PublicKey))
|
|
|
|
key2String := hex.EncodeToString(crypto.FromECDSAPub(&key2.PublicKey))
|
|
|
|
|
|
|
|
pair1 := [3]string{content1, signature1, key1String}
|
|
|
|
pair2 := [3]string{content2, signature2, key2String}
|
|
|
|
|
|
|
|
signaturePairs := [][3]string{pair1, pair2}
|
|
|
|
|
|
|
|
err = VerifySignatures(signaturePairs)
|
|
|
|
require.NoError(t, err)
|
2018-09-24 18:07:34 +00:00
|
|
|
|
2018-09-27 13:07:32 +00:00
|
|
|
// Test wrong content
|
|
|
|
pair3 := [3]string{content1, signature2, key2String}
|
|
|
|
|
|
|
|
signaturePairs = [][3]string{pair1, pair2, pair3}
|
|
|
|
|
|
|
|
err = VerifySignatures(signaturePairs)
|
|
|
|
require.Error(t, err)
|
|
|
|
|
|
|
|
// Test wrong signature
|
|
|
|
pair3 = [3]string{content1, signature2, key1String}
|
|
|
|
|
|
|
|
signaturePairs = [][3]string{pair1, pair2, pair3}
|
|
|
|
|
|
|
|
err = VerifySignatures(signaturePairs)
|
|
|
|
require.Error(t, err)
|
|
|
|
|
|
|
|
// Test wrong pubkey
|
|
|
|
pair3 = [3]string{content1, signature1, key2String}
|
|
|
|
|
|
|
|
signaturePairs = [][3]string{pair1, pair2, pair3}
|
|
|
|
|
|
|
|
err = VerifySignatures(signaturePairs)
|
|
|
|
require.Error(t, err)
|
|
|
|
}
|
2018-09-24 18:07:34 +00:00
|
|
|
|
|
|
|
func TestSymmetricEncryption(t *testing.T) {
|
2018-09-27 13:07:32 +00:00
|
|
|
const rawKey = "0000000000000000000000000000000000000000000000000000000000000000"
|
|
|
|
expectedPlaintext := []byte("test")
|
|
|
|
key, err := hex.DecodeString(rawKey)
|
|
|
|
require.Nil(t, err, "Key should be generated without errors")
|
2018-09-24 18:07:34 +00:00
|
|
|
|
|
|
|
cyphertext1, err := EncryptSymmetric(key, expectedPlaintext)
|
2018-09-27 13:07:32 +00:00
|
|
|
require.Nil(t, err, "Cyphertext should be generated without errors")
|
2018-09-24 18:07:34 +00:00
|
|
|
|
|
|
|
cyphertext2, err := EncryptSymmetric(key, expectedPlaintext)
|
2018-09-27 13:07:32 +00:00
|
|
|
require.Nil(t, err, "Cyphertext should be generated without errors")
|
2018-09-24 18:07:34 +00:00
|
|
|
|
2018-09-27 13:07:32 +00:00
|
|
|
require.Equalf(
|
2018-09-24 18:07:34 +00:00
|
|
|
t,
|
|
|
|
32,
|
|
|
|
len(cyphertext1),
|
|
|
|
"Cyphertext with the correct lenght should be generated")
|
|
|
|
|
2018-09-27 13:07:32 +00:00
|
|
|
require.NotEqualf(
|
2018-09-24 18:07:34 +00:00
|
|
|
t,
|
|
|
|
cyphertext1,
|
|
|
|
cyphertext2,
|
|
|
|
"Same plaintext should not be encrypted in the same way")
|
|
|
|
|
|
|
|
plaintext, err := DecryptSymmetric(key, cyphertext1)
|
2018-09-27 13:07:32 +00:00
|
|
|
require.Nil(t, err, "Cyphertext should be decrypted without errors")
|
2018-09-24 18:07:34 +00:00
|
|
|
|
2018-09-27 13:07:32 +00:00
|
|
|
require.Equalf(
|
2018-09-24 18:07:34 +00:00
|
|
|
t,
|
|
|
|
expectedPlaintext,
|
|
|
|
plaintext,
|
|
|
|
"Cypther text should be decrypted successfully")
|
|
|
|
}
|