status-im/status-go

package crypto

import (
    "encoding/hex"
    "testing"

    "github.com/stretchr/testify/require"
)

func TestExtractSignatures(t *testing.T) {
    const content1 = "045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b04ca82dd41fa592bf46ecf7e2eddae61013fc95a565b59c49f37f06b1b591ed3bd24e143495f2d1e241e151ab3572ac108d577be349d4b88d3d5a50c481ab35441"
    const content2 = "045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b04ca82dd41fa592bf46ecf7e2eddae61013fc95a565b59c49f37f06b1b591ed3bd24e143495f2d1e241e151ab3572ac108d577be349d4b88d3d5a50c481ab35440"

    key1, err := GenerateKey()
    require.NoError(t, err)

    key2, err := GenerateKey()
    require.NoError(t, err)

    signature1, err := SignStringAsHex(content1, key1)
    require.NoError(t, err)

    signature2, err := SignStringAsHex(content2, key2)
    require.NoError(t, err)

    key1String := hex.EncodeToString(FromECDSAPub(&key1.PublicKey))
    key2String := hex.EncodeToString(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)
}

func TestVerifySignature(t *testing.T) {
    const content1 = "045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b04ca82dd41fa592bf46ecf7e2eddae61013fc95a565b59c49f37f06b1b591ed3bd24e143495f2d1e241e151ab3572ac108d577be349d4b88d3d5a50c481ab35441"
    const content2 = "045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b045a8cae84d8d139e887bb927d2b98cee481afae3770e0ee45f2dc19c6545e45921bc6a55ea92b705e45dfbbe47182c7b1d64a080a220d2781577163923d7cbb4b04ca82dd41fa592bf46ecf7e2eddae61013fc95a565b59c49f37f06b1b591ed3bd24e143495f2d1e241e151ab3572ac108d577be349d4b88d3d5a50c481ab35440"

    key1, err := GenerateKey()
    require.NoError(t, err)

    key2, err := GenerateKey()
    require.NoError(t, err)

    signature1, err := SignStringAsHex(content1, key1)
    require.NoError(t, err)

    signature2, err := SignStringAsHex(content2, key2)
    require.NoError(t, err)

    key1String := hex.EncodeToString(FromECDSAPub(&key1.PublicKey))
    key2String := hex.EncodeToString(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)

    // 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)
}

func TestSymmetricEncryption(t *testing.T) {
    const rawKey = "0000000000000000000000000000000000000000000000000000000000000000"
    expectedPlaintext := []byte("test")
    key, err := hex.DecodeString(rawKey)
    require.Nil(t, err, "Key should be generated without errors")

    cyphertext1, err := EncryptSymmetric(key, expectedPlaintext)
    require.Nil(t, err, "Cyphertext should be generated without errors")

    cyphertext2, err := EncryptSymmetric(key, expectedPlaintext)
    require.Nil(t, err, "Cyphertext should be generated without errors")

    require.Equalf(
        t,
        32,
        len(cyphertext1),
        "Cyphertext with the correct length should be generated")

    require.NotEqualf(
        t,
        cyphertext1,
        cyphertext2,
        "Same plaintext should not be encrypted in the same way")

    plaintext, err := DecryptSymmetric(key, cyphertext1)
    require.Nil(t, err, "Cyphertext should be decrypted without errors")

    require.Equalf(
        t,
        expectedPlaintext,
        plaintext,
        "Cypther text should be decrypted successfully")
}