neoneo-go/pkg/vm/emit/emit_test.go

377 lines
10 KiB
Go
Raw Normal View History

package emit
import (
"encoding/binary"
2020-02-05 07:13:43 +00:00
"errors"
2021-03-05 11:16:34 +00:00
"math"
"math/big"
"testing"
"github.com/nspcc-dev/neo-go/pkg/core/interop/interopnames"
"github.com/nspcc-dev/neo-go/pkg/encoding/bigint"
"github.com/nspcc-dev/neo-go/pkg/io"
"github.com/nspcc-dev/neo-go/pkg/util"
"github.com/nspcc-dev/neo-go/pkg/vm/opcode"
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestEmitInt(t *testing.T) {
2020-02-05 07:29:43 +00:00
t.Run("minis one", func(t *testing.T) {
buf := io.NewBufBinWriter()
Int(buf.BinWriter, -1)
result := buf.Bytes()
assert.Len(t, result, 1)
assert.EqualValues(t, opcode.PUSHM1, result[0])
})
t.Run("zero", func(t *testing.T) {
buf := io.NewBufBinWriter()
Int(buf.BinWriter, 0)
result := buf.Bytes()
assert.Len(t, result, 1)
assert.EqualValues(t, opcode.PUSH0, result[0])
})
t.Run("1-byte int", func(t *testing.T) {
buf := io.NewBufBinWriter()
Int(buf.BinWriter, 10)
result := buf.Bytes()
assert.EqualValues(t, opcode.PUSH10, result[0])
})
2020-04-21 13:45:48 +00:00
t.Run("big 1-byte int", func(t *testing.T) {
buf := io.NewBufBinWriter()
Int(buf.BinWriter, 42)
result := buf.Bytes()
assert.EqualValues(t, opcode.PUSHINT8, result[0])
assert.EqualValues(t, 42, result[1])
})
t.Run("2-byte int", func(t *testing.T) {
buf := io.NewBufBinWriter()
2020-04-21 13:45:48 +00:00
Int(buf.BinWriter, 300)
result := buf.Bytes()
assert.Equal(t, 3, len(result))
assert.EqualValues(t, opcode.PUSHINT16, result[0])
assert.EqualValues(t, 300, bigint.FromBytes(result[1:]).Int64())
2020-04-21 13:45:48 +00:00
})
t.Run("3-byte int", func(t *testing.T) {
buf := io.NewBufBinWriter()
Int(buf.BinWriter, 1<<20)
result := buf.Bytes()
2020-04-21 13:45:48 +00:00
assert.Equal(t, 5, len(result))
assert.EqualValues(t, opcode.PUSHINT32, result[0])
assert.EqualValues(t, 1<<20, bigint.FromBytes(result[1:]).Int64())
})
t.Run("4-byte int", func(t *testing.T) {
buf := io.NewBufBinWriter()
2020-04-21 13:45:48 +00:00
Int(buf.BinWriter, 1<<28)
result := buf.Bytes()
assert.Equal(t, 5, len(result))
assert.EqualValues(t, opcode.PUSHINT32, result[0])
assert.EqualValues(t, 1<<28, bigint.FromBytes(result[1:]).Int64())
2020-04-21 13:45:48 +00:00
})
t.Run("negative 3-byte int with padding", func(t *testing.T) {
const num = -(1 << 23)
buf := io.NewBufBinWriter()
Int(buf.BinWriter, num)
result := buf.Bytes()
2020-04-21 13:45:48 +00:00
assert.Equal(t, 5, len(result))
assert.EqualValues(t, opcode.PUSHINT32, result[0])
assert.EqualValues(t, num, bigint.FromBytes(result[1:]).Int64())
})
}
func TestEmitBigInt(t *testing.T) {
t.Run("biggest positive number", func(t *testing.T) {
buf := io.NewBufBinWriter()
bi := big.NewInt(1)
bi.Lsh(bi, 255)
bi.Sub(bi, big.NewInt(1))
// sanity check
require.NotPanics(t, func() { stackitem.NewBigInteger(bi) })
BigInt(buf.BinWriter, bi)
require.NoError(t, buf.Err)
expected := make([]byte, 33)
expected[0] = byte(opcode.PUSHINT256)
for i := 1; i < 32; i++ {
expected[i] = 0xFF
}
expected[32] = 0x7F
require.Equal(t, expected, buf.Bytes())
})
t.Run("smallest negative number", func(t *testing.T) {
buf := io.NewBufBinWriter()
bi := big.NewInt(-1)
bi.Lsh(bi, 255)
// sanity check
require.NotPanics(t, func() { stackitem.NewBigInteger(bi) })
BigInt(buf.BinWriter, bi)
require.NoError(t, buf.Err)
expected := make([]byte, 33)
expected[0] = byte(opcode.PUSHINT256)
expected[32] = 0x80
require.Equal(t, expected, buf.Bytes())
})
t.Run("biggest positive number plus 1", func(t *testing.T) {
buf := io.NewBufBinWriter()
bi := big.NewInt(1)
bi.Lsh(bi, 255)
// sanity check
require.Panics(t, func() { stackitem.NewBigInteger(bi) })
BigInt(buf.BinWriter, bi)
require.Error(t, buf.Err)
t.Run("do not clear previous error", func(t *testing.T) {
buf.Reset()
expected := errors.New("expected")
buf.Err = expected
BigInt(buf.BinWriter, bi)
require.Equal(t, expected, buf.Err)
})
})
t.Run("smallest negative number minus 1", func(t *testing.T) {
buf := io.NewBufBinWriter()
bi := big.NewInt(-1)
bi.Lsh(bi, 255)
bi.Sub(bi, big.NewInt(1))
// sanity check
require.Panics(t, func() { stackitem.NewBigInteger(bi) })
BigInt(buf.BinWriter, bi)
require.Error(t, buf.Err)
})
}
2020-02-05 07:21:31 +00:00
func getSlice(n int) []byte {
data := make([]byte, n)
for i := range data {
data[i] = byte(i)
}
return data
}
func TestBytes(t *testing.T) {
t.Run("small slice", func(t *testing.T) {
buf := io.NewBufBinWriter()
Bytes(buf.BinWriter, []byte{0, 1, 2, 3})
result := buf.Bytes()
2020-04-21 13:45:48 +00:00
assert.EqualValues(t, opcode.PUSHDATA1, result[0])
assert.EqualValues(t, 4, result[1])
assert.EqualValues(t, []byte{0, 1, 2, 3}, result[2:])
2020-02-05 07:21:31 +00:00
})
t.Run("slice with len <= 255", func(t *testing.T) {
const size = 200
buf := io.NewBufBinWriter()
Bytes(buf.BinWriter, getSlice(size))
result := buf.Bytes()
assert.EqualValues(t, opcode.PUSHDATA1, result[0])
assert.EqualValues(t, size, result[1])
assert.Equal(t, getSlice(size), result[2:])
})
t.Run("slice with len <= 65535", func(t *testing.T) {
const size = 60000
buf := io.NewBufBinWriter()
Bytes(buf.BinWriter, getSlice(size))
result := buf.Bytes()
assert.EqualValues(t, opcode.PUSHDATA2, result[0])
assert.EqualValues(t, size, binary.LittleEndian.Uint16(result[1:3]))
assert.Equal(t, getSlice(size), result[3:])
})
t.Run("slice with len > 65535", func(t *testing.T) {
const size = 100000
buf := io.NewBufBinWriter()
Bytes(buf.BinWriter, getSlice(size))
result := buf.Bytes()
assert.EqualValues(t, opcode.PUSHDATA4, result[0])
assert.EqualValues(t, size, binary.LittleEndian.Uint32(result[1:5]))
assert.Equal(t, getSlice(size), result[5:])
})
}
func TestEmitArray(t *testing.T) {
t.Run("good", func(t *testing.T) {
buf := io.NewBufBinWriter()
var p160 *util.Uint160
var p256 *util.Uint256
u160 := util.Uint160{1, 2, 3}
u256 := util.Uint256{1, 2, 3}
2021-03-05 11:16:34 +00:00
veryBig := new(big.Int).SetUint64(math.MaxUint64)
veryBig.Add(veryBig, big.NewInt(1))
Array(buf.BinWriter, p160, p256, &u160, &u256, u160, u256, big.NewInt(0), veryBig,
[]any{int64(1), int64(2)}, nil, int64(1), "str", false, true, []byte{0xCA, 0xFE})
require.NoError(t, buf.Err)
res := buf.Bytes()
2020-04-21 13:45:48 +00:00
assert.EqualValues(t, opcode.PUSHDATA1, res[0])
assert.EqualValues(t, 2, res[1])
assert.EqualValues(t, []byte{0xCA, 0xFE}, res[2:4])
assert.EqualValues(t, opcode.PUSHT, res[4])
assert.EqualValues(t, opcode.PUSHF, res[5])
assert.EqualValues(t, opcode.PUSHDATA1, res[6])
assert.EqualValues(t, 3, res[7])
assert.EqualValues(t, []byte("str"), res[8:11])
assert.EqualValues(t, opcode.PUSH1, res[11])
assert.EqualValues(t, opcode.PUSHNULL, res[12])
assert.EqualValues(t, opcode.PUSH2, res[13])
assert.EqualValues(t, opcode.PUSH1, res[14])
assert.EqualValues(t, opcode.PUSH2, res[15])
assert.EqualValues(t, opcode.PACK, res[16])
assert.EqualValues(t, opcode.PUSHINT128, res[17])
assert.EqualValues(t, veryBig, bigint.FromBytes(res[18:34]))
assert.EqualValues(t, opcode.PUSH0, res[34])
assert.EqualValues(t, opcode.PUSHDATA1, res[35])
assert.EqualValues(t, 32, res[36])
assert.EqualValues(t, u256.BytesBE(), res[37:69])
assert.EqualValues(t, opcode.PUSHDATA1, res[69])
assert.EqualValues(t, 20, res[70])
assert.EqualValues(t, u160.BytesBE(), res[71:91])
assert.EqualValues(t, opcode.PUSHDATA1, res[91])
assert.EqualValues(t, 32, res[92])
assert.EqualValues(t, u256.BytesBE(), res[93:125])
assert.EqualValues(t, opcode.PUSHDATA1, res[125])
assert.EqualValues(t, 20, res[126])
assert.EqualValues(t, u160.BytesBE(), res[127:147])
assert.EqualValues(t, opcode.PUSHNULL, res[147])
assert.EqualValues(t, opcode.PUSHNULL, res[148])
})
t.Run("empty", func(t *testing.T) {
buf := io.NewBufBinWriter()
Array(buf.BinWriter)
require.NoError(t, buf.Err)
assert.EqualValues(t, []byte{byte(opcode.NEWARRAY0)}, buf.Bytes())
})
t.Run("invalid type", func(t *testing.T) {
buf := io.NewBufBinWriter()
Array(buf.BinWriter, struct{}{})
require.Error(t, buf.Err)
})
}
func TestEmitBool(t *testing.T) {
buf := io.NewBufBinWriter()
Bool(buf.BinWriter, true)
Bool(buf.BinWriter, false)
result := buf.Bytes()
assert.EqualValues(t, opcode.PUSHT, result[0])
assert.EqualValues(t, opcode.PUSHF, result[1])
}
func TestEmitOpcode(t *testing.T) {
w := io.NewBufBinWriter()
Opcodes(w.BinWriter, opcode.PUSH1, opcode.NEWMAP)
result := w.Bytes()
assert.Equal(t, result, []byte{byte(opcode.PUSH1), byte(opcode.NEWMAP)})
}
func TestEmitString(t *testing.T) {
buf := io.NewBufBinWriter()
str := "City Of Zion"
String(buf.BinWriter, str)
2020-04-21 13:45:48 +00:00
assert.Equal(t, buf.Len(), len(str)+2)
assert.Equal(t, buf.Bytes()[2:], []byte(str))
}
func TestEmitSyscall(t *testing.T) {
syscalls := []string{
interopnames.SystemRuntimeLog,
interopnames.SystemRuntimeNotify,
"System.Runtime.Whatever",
}
buf := io.NewBufBinWriter()
for _, syscall := range syscalls {
Syscall(buf.BinWriter, syscall)
result := buf.Bytes()
assert.Equal(t, 5, len(result))
assert.Equal(t, opcode.Opcode(result[0]), opcode.SYSCALL)
assert.Equal(t, binary.LittleEndian.Uint32(result[1:]), interopnames.ToID([]byte(syscall)))
buf.Reset()
}
2020-02-05 07:13:43 +00:00
t.Run("empty syscall", func(t *testing.T) {
buf := io.NewBufBinWriter()
Syscall(buf.BinWriter, "")
assert.Error(t, buf.Err)
})
t.Run("empty syscall after error", func(t *testing.T) {
buf := io.NewBufBinWriter()
err := errors.New("first error")
buf.Err = err
Syscall(buf.BinWriter, "")
assert.Equal(t, err, buf.Err)
})
}
func TestJmp(t *testing.T) {
const label = 0x23
t.Run("correct", func(t *testing.T) {
ops := []opcode.Opcode{opcode.JMP, opcode.JMPIF, opcode.JMPIFNOT, opcode.CALL}
for i := range ops {
t.Run(ops[i].String(), func(t *testing.T) {
buf := io.NewBufBinWriter()
Jmp(buf.BinWriter, ops[i], label)
assert.NoError(t, buf.Err)
result := buf.Bytes()
assert.EqualValues(t, ops[i], result[0])
assert.EqualValues(t, 0x23, binary.LittleEndian.Uint16(result[1:]))
})
}
})
t.Run("not a jump instruction", func(t *testing.T) {
buf := io.NewBufBinWriter()
Jmp(buf.BinWriter, opcode.ABS, label)
assert.Error(t, buf.Err)
})
t.Run("not a jump after error", func(t *testing.T) {
buf := io.NewBufBinWriter()
err := errors.New("first error")
buf.Err = err
Jmp(buf.BinWriter, opcode.ABS, label)
assert.Error(t, buf.Err)
})
}
func TestEmitCall(t *testing.T) {
buf := io.NewBufBinWriter()
Call(buf.BinWriter, opcode.JMP, 100)
result := buf.Bytes()
assert.Equal(t, opcode.Opcode(result[0]), opcode.JMP)
label := binary.LittleEndian.Uint16(result[1:3])
assert.Equal(t, label, uint16(100))
}