mirror of https://github.com/status-im/go-waku.git
159 lines
3.5 KiB
Go
159 lines
3.5 KiB
Go
package payload
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/waku-org/go-waku/waku/v2/protocol/pb"
|
|
)
|
|
|
|
func createTestMsg(version uint32) *pb.WakuMessage {
|
|
message := new(pb.WakuMessage)
|
|
message.Payload = []byte{0, 1, 2}
|
|
message.Version = version
|
|
message.Timestamp = 123456
|
|
return message
|
|
}
|
|
|
|
func TestEncodeDecodePayload(t *testing.T) {
|
|
data := []byte{0, 1, 2}
|
|
version := uint32(0)
|
|
|
|
keyInfo := new(KeyInfo)
|
|
keyInfo.Kind = None
|
|
|
|
p := Payload{
|
|
Data: data,
|
|
Key: keyInfo,
|
|
}
|
|
|
|
encodedPayload, err := p.Encode(version)
|
|
require.NoError(t, err)
|
|
require.Equal(t, data, encodedPayload)
|
|
|
|
message := new(pb.WakuMessage)
|
|
message.Payload = encodedPayload
|
|
message.Version = version
|
|
message.Timestamp = 123456
|
|
|
|
decodedPayload, err := DecodePayload(message, keyInfo)
|
|
require.NoError(t, err)
|
|
require.Equal(t, data, decodedPayload.Data)
|
|
}
|
|
|
|
func TestEncodeDecodeVersion0(t *testing.T) {
|
|
message := createTestMsg(0)
|
|
|
|
keyInfo := new(KeyInfo)
|
|
keyInfo.Kind = None
|
|
|
|
err := EncodeWakuMessage(message, keyInfo)
|
|
require.NoError(t, err)
|
|
|
|
err = DecodeWakuMessage(message, keyInfo)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func generateSymKey() ([]byte, error) {
|
|
key, err := generateSecureRandomData(aesKeyLength)
|
|
if err != nil {
|
|
return nil, err
|
|
} else if !validateDataIntegrity(key, aesKeyLength) {
|
|
return nil, fmt.Errorf("error in generating symkey: crypto/rand failed to generate random data")
|
|
}
|
|
|
|
return key, nil
|
|
}
|
|
|
|
func TestEncodeDecodeVersion1Symmetric(t *testing.T) {
|
|
message := createTestMsg(1)
|
|
data := message.Payload
|
|
|
|
keyInfo := new(KeyInfo)
|
|
keyInfo.Kind = Symmetric
|
|
|
|
var err error
|
|
keyInfo.SymKey, err = generateSymKey()
|
|
require.NoError(t, err)
|
|
|
|
err = EncodeWakuMessage(message, keyInfo)
|
|
require.NoError(t, err)
|
|
require.NotEqual(t, data, message.Payload)
|
|
require.NotNil(t, message.Payload)
|
|
|
|
err = DecodeWakuMessage(message, keyInfo)
|
|
require.NoError(t, err)
|
|
require.Equal(t, data, message.Payload)
|
|
}
|
|
|
|
func TestEncodeDecodeVersion1Asymmetric(t *testing.T) {
|
|
message := createTestMsg(1)
|
|
data := message.Payload
|
|
|
|
privKey, err := crypto.GenerateKey()
|
|
require.NoError(t, err)
|
|
|
|
keyInfo := new(KeyInfo)
|
|
keyInfo.Kind = Asymmetric
|
|
keyInfo.PubKey = privKey.PublicKey
|
|
|
|
err = EncodeWakuMessage(message, keyInfo)
|
|
require.NoError(t, err)
|
|
require.NotEqual(t, data, message.Payload)
|
|
require.NotNil(t, message.Payload)
|
|
|
|
keyInfo.PrivKey = privKey
|
|
err = DecodeWakuMessage(message, keyInfo)
|
|
require.NoError(t, err)
|
|
require.Equal(t, data, message.Payload)
|
|
}
|
|
|
|
func TestEncodeDecodeIncorrectKey(t *testing.T) {
|
|
message := createTestMsg(1)
|
|
|
|
privKey, err := crypto.GenerateKey()
|
|
require.NoError(t, err)
|
|
|
|
symKey, err := generateSymKey()
|
|
require.NoError(t, err)
|
|
|
|
keyInfo := new(KeyInfo)
|
|
keyInfo.Kind = Asymmetric
|
|
keyInfo.SymKey = symKey
|
|
|
|
err = EncodeWakuMessage(message, keyInfo)
|
|
require.Error(t, err)
|
|
|
|
keyInfo.SymKey = nil
|
|
keyInfo.PrivKey = privKey
|
|
|
|
err = EncodeWakuMessage(message, keyInfo)
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestEncodeUnsupportedVersion(t *testing.T) {
|
|
message := createTestMsg(99)
|
|
|
|
keyInfo := new(KeyInfo)
|
|
keyInfo.Kind = None
|
|
|
|
err := EncodeWakuMessage(message, keyInfo)
|
|
require.Error(t, err)
|
|
require.EqualError(t, err, "unsupported wakumessage version")
|
|
}
|
|
|
|
func TestDecodeUnsupportedVersion(t *testing.T) {
|
|
message := createTestMsg(99)
|
|
|
|
keyInfo := new(KeyInfo)
|
|
keyInfo.Kind = None
|
|
|
|
decodedPayload, err := DecodePayload(message, keyInfo)
|
|
|
|
require.Nil(t, decodedPayload)
|
|
require.Error(t, err)
|
|
require.EqualError(t, err, "unsupported wakumessage version")
|
|
}
|