mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-17 19:47:19 +00:00
8d4dd2d2e1
This allows easier reuse of opcodes and in some cases allows to eliminate dependencies on the whole vm package, like in compiler that only needs opcodes and doesn't care about VM for any other purpose. And yes, they're opcodes because an instruction is a whole thing with operands, that's what context.Next() returns.
228 lines
6.7 KiB
Go
228 lines
6.7 KiB
Go
package vm
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/vm/opcode"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestIsSignatureContractGood(t *testing.T) {
|
|
prog := make([]byte, 35)
|
|
prog[0] = byte(opcode.PUSHBYTES33)
|
|
prog[34] = byte(opcode.CHECKSIG)
|
|
assert.Equal(t, true, IsSignatureContract(prog))
|
|
assert.Equal(t, true, IsStandardContract(prog))
|
|
}
|
|
|
|
func TestIsSignatureContractBadNoCheckSig(t *testing.T) {
|
|
prog := make([]byte, 34)
|
|
prog[0] = byte(opcode.PUSHBYTES33)
|
|
assert.Equal(t, false, IsSignatureContract(prog))
|
|
assert.Equal(t, false, IsStandardContract(prog))
|
|
}
|
|
|
|
func TestIsSignatureContractBadNoCheckSig2(t *testing.T) {
|
|
prog := make([]byte, 35)
|
|
prog[0] = byte(opcode.PUSHBYTES33)
|
|
prog[34] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, false, IsSignatureContract(prog))
|
|
}
|
|
|
|
func TestIsSignatureContractBadWrongPush(t *testing.T) {
|
|
prog := make([]byte, 35)
|
|
prog[0] = byte(opcode.PUSHBYTES32)
|
|
prog[33] = byte(opcode.NOP)
|
|
prog[34] = byte(opcode.CHECKSIG)
|
|
assert.Equal(t, false, IsSignatureContract(prog))
|
|
}
|
|
|
|
func TestIsSignatureContractBadWrongInstr(t *testing.T) {
|
|
prog := make([]byte, 30)
|
|
prog[0] = byte(opcode.PUSHBYTES33)
|
|
assert.Equal(t, false, IsSignatureContract(prog))
|
|
}
|
|
|
|
func TestIsSignatureContractBadExcessiveInstr(t *testing.T) {
|
|
prog := make([]byte, 36)
|
|
prog[0] = byte(opcode.PUSHBYTES33)
|
|
prog[34] = byte(opcode.CHECKSIG)
|
|
prog[35] = byte(opcode.RET)
|
|
assert.Equal(t, false, IsSignatureContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractGood(t *testing.T) {
|
|
prog := make([]byte, 71)
|
|
prog[0] = byte(opcode.PUSH2)
|
|
prog[1] = byte(opcode.PUSHBYTES33)
|
|
prog[35] = byte(opcode.PUSHBYTES33)
|
|
prog[69] = byte(opcode.PUSH2)
|
|
prog[70] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, true, IsMultiSigContract(prog))
|
|
assert.Equal(t, true, IsStandardContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractGoodPushBytes1(t *testing.T) {
|
|
prog := make([]byte, 73)
|
|
prog[0] = byte(opcode.PUSHBYTES1)
|
|
prog[1] = 2
|
|
prog[2] = byte(opcode.PUSHBYTES33)
|
|
prog[36] = byte(opcode.PUSHBYTES33)
|
|
prog[70] = byte(opcode.PUSHBYTES1)
|
|
prog[71] = 2
|
|
prog[72] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, true, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractGoodPushBytes2(t *testing.T) {
|
|
prog := make([]byte, 75)
|
|
prog[0] = byte(opcode.PUSHBYTES2)
|
|
prog[1] = 2
|
|
prog[3] = byte(opcode.PUSHBYTES33)
|
|
prog[37] = byte(opcode.PUSHBYTES33)
|
|
prog[71] = byte(opcode.PUSHBYTES2)
|
|
prog[72] = 2
|
|
prog[74] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, true, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadNSigs1(t *testing.T) {
|
|
prog := make([]byte, 71)
|
|
prog[0] = byte(opcode.PUSH0)
|
|
prog[1] = byte(opcode.PUSHBYTES33)
|
|
prog[35] = byte(opcode.PUSHBYTES33)
|
|
prog[69] = byte(opcode.PUSH2)
|
|
prog[70] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
assert.Equal(t, false, IsStandardContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadNSigs2(t *testing.T) {
|
|
prog := make([]byte, 73)
|
|
prog[0] = byte(opcode.PUSHBYTES2)
|
|
prog[1] = 0xff
|
|
prog[2] = 0xff
|
|
prog[3] = byte(opcode.PUSHBYTES33)
|
|
prog[37] = byte(opcode.PUSHBYTES33)
|
|
prog[71] = byte(opcode.PUSH2)
|
|
prog[72] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadNSigs3(t *testing.T) {
|
|
prog := make([]byte, 71)
|
|
prog[0] = byte(opcode.PUSH5)
|
|
prog[1] = byte(opcode.PUSHBYTES33)
|
|
prog[35] = byte(opcode.PUSHBYTES33)
|
|
prog[69] = byte(opcode.PUSH2)
|
|
prog[70] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadExcessiveNOP1(t *testing.T) {
|
|
prog := make([]byte, 72)
|
|
prog[0] = byte(opcode.PUSH2)
|
|
prog[1] = byte(opcode.NOP)
|
|
prog[2] = byte(opcode.PUSHBYTES33)
|
|
prog[36] = byte(opcode.PUSHBYTES33)
|
|
prog[70] = byte(opcode.PUSH2)
|
|
prog[71] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadExcessiveNOP2(t *testing.T) {
|
|
prog := make([]byte, 72)
|
|
prog[0] = byte(opcode.PUSH2)
|
|
prog[1] = byte(opcode.PUSHBYTES33)
|
|
prog[35] = byte(opcode.NOP)
|
|
prog[36] = byte(opcode.PUSHBYTES33)
|
|
prog[70] = byte(opcode.PUSH2)
|
|
prog[71] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadExcessiveNOP3(t *testing.T) {
|
|
prog := make([]byte, 72)
|
|
prog[0] = byte(opcode.PUSH2)
|
|
prog[1] = byte(opcode.PUSHBYTES33)
|
|
prog[35] = byte(opcode.PUSHBYTES33)
|
|
prog[69] = byte(opcode.NOP)
|
|
prog[70] = byte(opcode.PUSH2)
|
|
prog[71] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadExcessiveNOP4(t *testing.T) {
|
|
prog := make([]byte, 72)
|
|
prog[0] = byte(opcode.PUSH2)
|
|
prog[1] = byte(opcode.PUSHBYTES33)
|
|
prog[35] = byte(opcode.PUSHBYTES33)
|
|
prog[69] = byte(opcode.PUSH2)
|
|
prog[70] = byte(opcode.NOP)
|
|
prog[71] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadExcessiveNOP5(t *testing.T) {
|
|
prog := make([]byte, 72)
|
|
prog[0] = byte(opcode.PUSH2)
|
|
prog[1] = byte(opcode.PUSHBYTES33)
|
|
prog[35] = byte(opcode.PUSHBYTES33)
|
|
prog[69] = byte(opcode.PUSH2)
|
|
prog[70] = byte(opcode.CHECKMULTISIG)
|
|
prog[71] = byte(opcode.NOP)
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadNKeys1(t *testing.T) {
|
|
prog := make([]byte, 71)
|
|
prog[0] = byte(opcode.PUSH2)
|
|
prog[1] = byte(opcode.PUSHBYTES33)
|
|
prog[35] = byte(opcode.PUSHBYTES33)
|
|
prog[69] = byte(opcode.PUSH3)
|
|
prog[70] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadNKeys2(t *testing.T) {
|
|
prog := make([]byte, 1)
|
|
prog[0] = byte(opcode.PUSH10)
|
|
key := make([]byte, 33)
|
|
var asize = uint16(MaxArraySize + 1)
|
|
for i := 0; i < int(asize); i++ {
|
|
prog = append(prog, byte(opcode.PUSHBYTES33))
|
|
prog = append(prog, key...)
|
|
}
|
|
prog = append(prog, byte(opcode.PUSHBYTES2), byte(asize&0xff), byte((asize<<8)&0xff), byte(opcode.CHECKMULTISIG))
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadRead1(t *testing.T) {
|
|
prog := make([]byte, 71)
|
|
prog[0] = byte(opcode.PUSHBYTES75)
|
|
prog[1] = byte(opcode.PUSHBYTES33)
|
|
prog[35] = byte(opcode.PUSHBYTES33)
|
|
prog[69] = byte(opcode.PUSH2)
|
|
prog[70] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadRead2(t *testing.T) {
|
|
prog := make([]byte, 71)
|
|
prog[0] = byte(opcode.PUSH2)
|
|
prog[1] = byte(opcode.PUSHBYTES33)
|
|
prog[35] = byte(opcode.PUSHBYTES75)
|
|
prog[69] = byte(opcode.PUSH2)
|
|
prog[70] = byte(opcode.CHECKMULTISIG)
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
}
|
|
|
|
func TestIsMultiSigContractBadRead3(t *testing.T) {
|
|
prog := make([]byte, 71)
|
|
prog[0] = byte(opcode.PUSH2)
|
|
prog[1] = byte(opcode.PUSHBYTES33)
|
|
prog[35] = byte(opcode.PUSHBYTES33)
|
|
prog[69] = byte(opcode.PUSH2)
|
|
prog[70] = byte(opcode.PUSHBYTES1)
|
|
assert.Equal(t, false, IsMultiSigContract(prog))
|
|
}
|