mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-29 09:27:36 +00:00
330 lines
9.7 KiB
Go
330 lines
9.7 KiB
Go
package core
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/config/netmode"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/block"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/dao"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/interop"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/interop/crypto"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/interop/enumerator"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/interop/iterator"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/state"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/storage"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/transaction"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/hash"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/manifest"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/trigger"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/opcode"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
/* Missing tests:
|
|
* TestTxGetWitnesses
|
|
* TestBcGetAccount
|
|
* TestAccountGetBalance
|
|
* TestAccountIsStandard
|
|
* TestCreateContractStateFromVM
|
|
* TestContractCreate
|
|
* TestContractMigrate
|
|
* TestRuntimeSerialize
|
|
* TestRuntimeDeserialize
|
|
*/
|
|
|
|
func TestGetTrigger(t *testing.T) {
|
|
v, _, context, chain := createVMAndPushBlock(t)
|
|
defer chain.Close()
|
|
require.NoError(t, runtimeGetTrigger(context, v))
|
|
}
|
|
|
|
func TestStorageFind(t *testing.T) {
|
|
v, contractState, context, chain := createVMAndContractState(t)
|
|
defer chain.Close()
|
|
|
|
skeys := [][]byte{{0x01, 0x02}, {0x02, 0x01}, {0x01, 0x01}}
|
|
items := []*state.StorageItem{
|
|
{
|
|
Value: []byte{0x01, 0x02, 0x03, 0x04},
|
|
},
|
|
{
|
|
Value: []byte{0x04, 0x03, 0x02, 0x01},
|
|
},
|
|
{
|
|
Value: []byte{0x03, 0x04, 0x05, 0x06},
|
|
},
|
|
}
|
|
|
|
require.NoError(t, context.DAO.PutContractState(contractState))
|
|
|
|
id := contractState.ID
|
|
|
|
for i := range skeys {
|
|
err := context.DAO.PutStorageItem(id, skeys[i], items[i])
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
t.Run("normal invocation", func(t *testing.T) {
|
|
v.Estack().PushVal([]byte{0x01})
|
|
v.Estack().PushVal(stackitem.NewInterop(&StorageContext{ID: id}))
|
|
|
|
err := storageFind(context, v)
|
|
require.NoError(t, err)
|
|
|
|
var iter *stackitem.Interop
|
|
require.NotPanics(t, func() { iter = v.Estack().Top().Interop() })
|
|
|
|
require.NoError(t, enumerator.Next(context, v))
|
|
require.True(t, v.Estack().Pop().Bool())
|
|
|
|
v.Estack().PushVal(iter)
|
|
require.NoError(t, iterator.Key(context, v))
|
|
require.Equal(t, []byte{0x01, 0x01}, v.Estack().Pop().Bytes())
|
|
|
|
v.Estack().PushVal(iter)
|
|
require.NoError(t, enumerator.Value(context, v))
|
|
require.Equal(t, []byte{0x03, 0x04, 0x05, 0x06}, v.Estack().Pop().Bytes())
|
|
|
|
v.Estack().PushVal(iter)
|
|
require.NoError(t, enumerator.Next(context, v))
|
|
require.True(t, v.Estack().Pop().Bool())
|
|
|
|
v.Estack().PushVal(iter)
|
|
require.NoError(t, iterator.Key(context, v))
|
|
require.Equal(t, []byte{0x01, 0x02}, v.Estack().Pop().Bytes())
|
|
|
|
v.Estack().PushVal(iter)
|
|
require.NoError(t, enumerator.Value(context, v))
|
|
require.Equal(t, []byte{0x01, 0x02, 0x03, 0x04}, v.Estack().Pop().Bytes())
|
|
|
|
v.Estack().PushVal(iter)
|
|
require.NoError(t, enumerator.Next(context, v))
|
|
require.False(t, v.Estack().Pop().Bool())
|
|
})
|
|
|
|
t.Run("normal invocation, empty result", func(t *testing.T) {
|
|
v.Estack().PushVal([]byte{0x03})
|
|
v.Estack().PushVal(stackitem.NewInterop(&StorageContext{ID: id}))
|
|
|
|
err := storageFind(context, v)
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, enumerator.Next(context, v))
|
|
require.False(t, v.Estack().Pop().Bool())
|
|
})
|
|
|
|
t.Run("invalid type for StorageContext", func(t *testing.T) {
|
|
v.Estack().PushVal([]byte{0x01})
|
|
v.Estack().PushVal(stackitem.NewInterop(nil))
|
|
|
|
require.Error(t, storageFind(context, v))
|
|
})
|
|
|
|
t.Run("invalid id", func(t *testing.T) {
|
|
invalidID := id + 1
|
|
|
|
v.Estack().PushVal([]byte{0x01})
|
|
v.Estack().PushVal(stackitem.NewInterop(&StorageContext{ID: invalidID}))
|
|
|
|
require.NoError(t, storageFind(context, v))
|
|
require.NoError(t, enumerator.Next(context, v))
|
|
require.False(t, v.Estack().Pop().Bool())
|
|
})
|
|
}
|
|
|
|
func TestECDSAVerify(t *testing.T) {
|
|
priv, err := keys.NewPrivateKey()
|
|
require.NoError(t, err)
|
|
|
|
chain := newTestChain(t)
|
|
defer chain.Close()
|
|
|
|
ic := chain.newInteropContext(trigger.Application, dao.NewSimple(storage.NewMemoryStore(), netmode.UnitTestNet), nil, nil)
|
|
runCase := func(t *testing.T, isErr bool, result interface{}, args ...interface{}) {
|
|
v := vm.New()
|
|
for i := range args {
|
|
v.Estack().PushVal(args[i])
|
|
}
|
|
|
|
var err error
|
|
func() {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
err = fmt.Errorf("panic: %v", r)
|
|
}
|
|
}()
|
|
err = crypto.ECDSASecp256r1Verify(ic, v)
|
|
}()
|
|
|
|
if isErr {
|
|
require.Error(t, err)
|
|
return
|
|
}
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, v.Estack().Len())
|
|
require.Equal(t, result, v.Estack().Pop().Value().(bool))
|
|
}
|
|
|
|
msg := []byte("test message")
|
|
|
|
t.Run("success", func(t *testing.T) {
|
|
sign := priv.Sign(msg)
|
|
runCase(t, false, true, sign, priv.PublicKey().Bytes(), msg)
|
|
})
|
|
|
|
t.Run("signed interop item", func(t *testing.T) {
|
|
tx := transaction.New(netmode.UnitTestNet, []byte{0, 1, 2}, 1)
|
|
msg := tx.GetSignedPart()
|
|
sign := priv.Sign(msg)
|
|
runCase(t, false, true, sign, priv.PublicKey().Bytes(), stackitem.NewInterop(tx))
|
|
})
|
|
|
|
t.Run("signed script container", func(t *testing.T) {
|
|
tx := transaction.New(netmode.UnitTestNet, []byte{0, 1, 2}, 1)
|
|
msg := tx.GetSignedPart()
|
|
sign := priv.Sign(msg)
|
|
ic.Container = tx
|
|
runCase(t, false, true, sign, priv.PublicKey().Bytes(), stackitem.Null{})
|
|
})
|
|
|
|
t.Run("missing arguments", func(t *testing.T) {
|
|
runCase(t, true, false)
|
|
sign := priv.Sign(msg)
|
|
runCase(t, true, false, sign)
|
|
runCase(t, true, false, sign, priv.PublicKey().Bytes())
|
|
})
|
|
|
|
t.Run("invalid signature", func(t *testing.T) {
|
|
sign := priv.Sign(msg)
|
|
sign[0] = ^sign[0]
|
|
runCase(t, false, false, sign, priv.PublicKey().Bytes(), msg)
|
|
})
|
|
|
|
t.Run("invalid public key", func(t *testing.T) {
|
|
sign := priv.Sign(msg)
|
|
pub := priv.PublicKey().Bytes()
|
|
pub[0] = 0xFF // invalid prefix
|
|
runCase(t, true, false, sign, pub, msg)
|
|
})
|
|
|
|
t.Run("invalid message", func(t *testing.T) {
|
|
sign := priv.Sign(msg)
|
|
runCase(t, false, false, sign, priv.PublicKey().Bytes(),
|
|
stackitem.NewArray([]stackitem.Item{stackitem.NewByteArray(msg)}))
|
|
})
|
|
}
|
|
|
|
func TestRuntimeEncode(t *testing.T) {
|
|
str := []byte("my pretty string")
|
|
v, ic, bc := createVM(t)
|
|
defer bc.Close()
|
|
|
|
v.Estack().PushVal(str)
|
|
require.NoError(t, runtimeEncode(ic, v))
|
|
|
|
expected := []byte(base64.StdEncoding.EncodeToString(str))
|
|
actual := v.Estack().Pop().Bytes()
|
|
require.Equal(t, expected, actual)
|
|
}
|
|
|
|
func TestRuntimeDecode(t *testing.T) {
|
|
expected := []byte("my pretty string")
|
|
str := base64.StdEncoding.EncodeToString(expected)
|
|
v, ic, bc := createVM(t)
|
|
defer bc.Close()
|
|
|
|
t.Run("positive", func(t *testing.T) {
|
|
v.Estack().PushVal(str)
|
|
require.NoError(t, runtimeDecode(ic, v))
|
|
|
|
actual := v.Estack().Pop().Bytes()
|
|
require.Equal(t, expected, actual)
|
|
})
|
|
|
|
t.Run("error", func(t *testing.T) {
|
|
v.Estack().PushVal(str + "%")
|
|
require.Error(t, runtimeDecode(ic, v))
|
|
})
|
|
}
|
|
|
|
// Helper functions to create VM, InteropContext, TX, Account, Contract.
|
|
|
|
func createVM(t *testing.T) (*vm.VM, *interop.Context, *Blockchain) {
|
|
v := vm.New()
|
|
chain := newTestChain(t)
|
|
context := chain.newInteropContext(trigger.Application,
|
|
dao.NewSimple(storage.NewMemoryStore(), netmode.UnitTestNet), nil, nil)
|
|
return v, context, chain
|
|
}
|
|
|
|
func createVMAndPushBlock(t *testing.T) (*vm.VM, *block.Block, *interop.Context, *Blockchain) {
|
|
v, block, context, chain := createVMAndBlock(t)
|
|
v.Estack().PushVal(stackitem.NewInterop(block))
|
|
return v, block, context, chain
|
|
}
|
|
|
|
func createVMAndBlock(t *testing.T) (*vm.VM, *block.Block, *interop.Context, *Blockchain) {
|
|
v := vm.New()
|
|
block := newDumbBlock()
|
|
chain := newTestChain(t)
|
|
context := chain.newInteropContext(trigger.Application, dao.NewSimple(storage.NewMemoryStore(), netmode.UnitTestNet), block, nil)
|
|
return v, block, context, chain
|
|
}
|
|
|
|
func createVMAndPushTX(t *testing.T) (*vm.VM, *transaction.Transaction, *interop.Context, *Blockchain) {
|
|
v, tx, context, chain := createVMAndTX(t)
|
|
v.Estack().PushVal(stackitem.NewInterop(tx))
|
|
return v, tx, context, chain
|
|
}
|
|
|
|
func createVMAndContractState(t *testing.T) (*vm.VM, *state.Contract, *interop.Context, *Blockchain) {
|
|
v := vm.New()
|
|
script := []byte("testscript")
|
|
m := manifest.NewManifest(hash.Hash160(script))
|
|
m.Features = smartcontract.HasStorage
|
|
contractState := &state.Contract{
|
|
Script: script,
|
|
Manifest: *m,
|
|
ID: 123,
|
|
}
|
|
|
|
chain := newTestChain(t)
|
|
context := chain.newInteropContext(trigger.Application, dao.NewSimple(storage.NewMemoryStore(), netmode.UnitTestNet), nil, nil)
|
|
return v, contractState, context, chain
|
|
}
|
|
|
|
func createVMAndAccState(t *testing.T) (*vm.VM, *state.Account, *interop.Context, *Blockchain) {
|
|
v := vm.New()
|
|
rawHash := "4d3b96ae1bcc5a585e075e3b81920210dec16302"
|
|
hash, err := util.Uint160DecodeStringBE(rawHash)
|
|
accountState := state.NewAccount(hash)
|
|
|
|
require.NoError(t, err)
|
|
chain := newTestChain(t)
|
|
context := chain.newInteropContext(trigger.Application, dao.NewSimple(storage.NewMemoryStore(), netmode.UnitTestNet), nil, nil)
|
|
return v, accountState, context, chain
|
|
}
|
|
|
|
func createVMAndTX(t *testing.T) (*vm.VM, *transaction.Transaction, *interop.Context, *Blockchain) {
|
|
v := vm.New()
|
|
script := []byte{byte(opcode.PUSH1), byte(opcode.RET)}
|
|
tx := transaction.New(netmode.UnitTestNet, script, 0)
|
|
|
|
bytes := make([]byte, 1)
|
|
attributes := append(tx.Attributes, transaction.Attribute{
|
|
Usage: transaction.DescriptionURL,
|
|
Data: bytes,
|
|
})
|
|
|
|
tx.Attributes = attributes
|
|
chain := newTestChain(t)
|
|
context := chain.newInteropContext(trigger.Application, dao.NewSimple(storage.NewMemoryStore(), netmode.UnitTestNet), nil, tx)
|
|
return v, tx, context, chain
|
|
}
|