mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-28 23:27:36 +00:00
a7457d08a1
This solves two problems: * adds support for shortened SYSCALL form that uses IDs (similar to #434, but for NEO 2.0, supporting both forms), which is important for compatibility with C# node and mainnet chain that uses it from some height * reworks interop plugging to use callbacks rather than appending to the map, these map mangling functions are clearly visible in the VM profiling statistics and we want spawning a VM to be fast, so it makes sense optimizing it. This change moves most of the work to the init() phase making VM setup cheaper. Caveats: * InteropNameToID accepts `[]byte` because that's the thing we have in SYSCALL processing and that's the most often usecase for it, it leads to some conversions in other places but that's acceptable because those are either tests or init() * three getInterop functions are: `getDefaultVMInterop`, `getSystemInterop` and `getNeoInterop` Our 100K (1.4M->1.5M) block import time improves by ~4% with this change.
116 lines
2.7 KiB
Go
116 lines
2.7 KiB
Go
package vm_test
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/compiler"
|
|
"github.com/CityOfZion/neo-go/pkg/vm"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type testCase struct {
|
|
name string
|
|
src string
|
|
result interface{}
|
|
}
|
|
|
|
func runTestCases(t *testing.T, tcases []testCase) {
|
|
for _, tcase := range tcases {
|
|
t.Run(tcase.name, func(t *testing.T) { eval(t, tcase.src, tcase.result) })
|
|
}
|
|
}
|
|
|
|
func eval(t *testing.T, src string, result interface{}) {
|
|
vm := vmAndCompile(t, src)
|
|
err := vm.Run()
|
|
require.NoError(t, err)
|
|
assertResult(t, vm, result)
|
|
}
|
|
|
|
func evalWithArgs(t *testing.T, src string, op []byte, args []vm.StackItem, result interface{}) {
|
|
vm := vmAndCompile(t, src)
|
|
vm.LoadArgs(op, args)
|
|
err := vm.Run()
|
|
require.NoError(t, err)
|
|
assertResult(t, vm, result)
|
|
}
|
|
|
|
func assertResult(t *testing.T, vm *vm.VM, result interface{}) {
|
|
assert.Equal(t, result, vm.PopResult())
|
|
assert.Equal(t, 0, vm.Astack().Len())
|
|
assert.Equal(t, 0, vm.Istack().Len())
|
|
}
|
|
|
|
func vmAndCompile(t *testing.T, src string) *vm.VM {
|
|
vm := vm.New()
|
|
|
|
storePlugin := newStoragePlugin()
|
|
vm.RegisterInteropGetter(storePlugin.getInterop)
|
|
|
|
b, err := compiler.Compile(strings.NewReader(src))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
vm.Load(b)
|
|
return vm
|
|
}
|
|
|
|
type storagePlugin struct {
|
|
mem map[string][]byte
|
|
interops map[uint32]vm.InteropFunc
|
|
}
|
|
|
|
func newStoragePlugin() *storagePlugin {
|
|
s := &storagePlugin{
|
|
mem: make(map[string][]byte),
|
|
interops: make(map[uint32]vm.InteropFunc),
|
|
}
|
|
s.interops[vm.InteropNameToID([]byte("Neo.Storage.Get"))] = s.Get
|
|
s.interops[vm.InteropNameToID([]byte("Neo.Storage.Put"))] = s.Put
|
|
s.interops[vm.InteropNameToID([]byte("Neo.Storage.GetContext"))] = s.GetContext
|
|
return s
|
|
|
|
}
|
|
|
|
func (s *storagePlugin) getInterop(id uint32) *vm.InteropFuncPrice {
|
|
f := s.interops[id]
|
|
if f != nil {
|
|
return &vm.InteropFuncPrice{Func: f, Price: 1}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *storagePlugin) Delete(vm *vm.VM) error {
|
|
vm.Estack().Pop()
|
|
key := vm.Estack().Pop().Bytes()
|
|
delete(s.mem, string(key))
|
|
return nil
|
|
}
|
|
|
|
func (s *storagePlugin) Put(vm *vm.VM) error {
|
|
vm.Estack().Pop()
|
|
key := vm.Estack().Pop().Bytes()
|
|
value := vm.Estack().Pop().Bytes()
|
|
s.mem[string(key)] = value
|
|
return nil
|
|
}
|
|
|
|
func (s *storagePlugin) Get(vm *vm.VM) error {
|
|
vm.Estack().Pop()
|
|
item := vm.Estack().Pop().Bytes()
|
|
if val, ok := s.mem[string(item)]; ok {
|
|
vm.Estack().PushVal(val)
|
|
return nil
|
|
}
|
|
return fmt.Errorf("could not find %+v", item)
|
|
}
|
|
|
|
func (s *storagePlugin) GetContext(vm *vm.VM) error {
|
|
// Pushing anything on the stack here will work. This is just to satisfy
|
|
// the compiler, thinking it has pushed the context ^^.
|
|
vm.Estack().PushVal(10)
|
|
return nil
|
|
}
|