135 lines
2.6 KiB
Go
135 lines
2.6 KiB
Go
package vaa
|
|
|
|
import (
|
|
"crypto/ecdsa"
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/stretchr/testify/require"
|
|
"math/big"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestSerializeDeserialize(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
vaa *VAA
|
|
}{
|
|
{
|
|
name: "BodyTransfer",
|
|
vaa: &VAA{
|
|
Version: 1,
|
|
GuardianSetIndex: 9,
|
|
Signatures: []*Signature{
|
|
{
|
|
Index: 1,
|
|
Signature: [65]byte{},
|
|
},
|
|
},
|
|
Timestamp: time.Unix(2837, 0),
|
|
Payload: &BodyTransfer{
|
|
Nonce: 38,
|
|
SourceChain: 2,
|
|
TargetChain: 1,
|
|
SourceAddress: Address{2, 1, 4},
|
|
TargetAddress: Address{2, 1, 3},
|
|
Asset: &AssetMeta{
|
|
Chain: 9,
|
|
Address: Address{9, 2, 4},
|
|
},
|
|
Amount: big.NewInt(29),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "GuardianSetUpdate",
|
|
vaa: &VAA{
|
|
Version: 1,
|
|
GuardianSetIndex: 9,
|
|
Signatures: []*Signature{
|
|
{
|
|
Index: 1,
|
|
Signature: [65]byte{},
|
|
},
|
|
},
|
|
Timestamp: time.Unix(2837, 0),
|
|
Payload: &BodyGuardianSetUpdate{
|
|
Keys: []common.Address{{}, {}},
|
|
NewIndex: 2,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "ContractUpgrade",
|
|
vaa: &VAA{
|
|
Version: 1,
|
|
GuardianSetIndex: 9,
|
|
Signatures: []*Signature{
|
|
{
|
|
Index: 1,
|
|
Signature: [65]byte{},
|
|
},
|
|
},
|
|
Timestamp: time.Unix(2837, 0),
|
|
Payload: &BodyContractUpgrade{
|
|
ChainID: ChainIDEthereum,
|
|
NewContract: Address{1, 3, 4, 5, 2, 3},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
|
|
vaaData, err := test.vaa.Marshal()
|
|
require.NoError(t, err)
|
|
|
|
println(hex.EncodeToString(vaaData))
|
|
vaaParsed, err := Unmarshal(vaaData)
|
|
require.NoError(t, err)
|
|
|
|
require.EqualValues(t, test.vaa, vaaParsed)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestVerifySignature(t *testing.T) {
|
|
v := &VAA{
|
|
Version: 8,
|
|
GuardianSetIndex: 9,
|
|
Timestamp: time.Unix(2837, 0),
|
|
Payload: &BodyTransfer{
|
|
SourceChain: 2,
|
|
TargetChain: 1,
|
|
TargetAddress: Address{2, 1, 3},
|
|
Asset: &AssetMeta{
|
|
Chain: 9,
|
|
Address: Address{9, 2, 4},
|
|
},
|
|
Amount: big.NewInt(29),
|
|
},
|
|
}
|
|
|
|
data, err := v.SigningMsg()
|
|
require.NoError(t, err)
|
|
|
|
key, err := ecdsa.GenerateKey(crypto.S256(), rand.Reader)
|
|
require.NoError(t, err)
|
|
|
|
sig, err := crypto.Sign(data.Bytes(), key)
|
|
require.NoError(t, err)
|
|
sigData := [65]byte{}
|
|
copy(sigData[:], sig)
|
|
|
|
v.Signatures = append(v.Signatures, &Signature{
|
|
Index: 0,
|
|
Signature: sigData,
|
|
})
|
|
addr := crypto.PubkeyToAddress(key.PublicKey)
|
|
require.True(t, v.VerifySignatures([]common.Address{
|
|
addr,
|
|
}))
|
|
}
|