forked from TrueCloudLab/neoneo-go
vm: add XDROP implementation
This commit is contained in:
parent
974f5db6bc
commit
b494e094c0
2 changed files with 48 additions and 0 deletions
10
pkg/vm/vm.go
10
pkg/vm/vm.go
|
@ -300,6 +300,16 @@ func (v *VM) execute(ctx *Context, op Instruction) {
|
||||||
}
|
}
|
||||||
v.estack.InsertAt(a, 2)
|
v.estack.InsertAt(a, 2)
|
||||||
|
|
||||||
|
case XDROP:
|
||||||
|
n := int(v.estack.Pop().BigInt().Int64())
|
||||||
|
if n < 0 {
|
||||||
|
panic("invalid length")
|
||||||
|
}
|
||||||
|
e := v.estack.RemoveAt(n)
|
||||||
|
if e == nil {
|
||||||
|
panic("bad index")
|
||||||
|
}
|
||||||
|
|
||||||
case XSWAP:
|
case XSWAP:
|
||||||
n := int(v.estack.Pop().BigInt().Int64())
|
n := int(v.estack.Pop().BigInt().Int64())
|
||||||
if n < 0 {
|
if n < 0 {
|
||||||
|
|
|
@ -408,6 +408,44 @@ func TestOVERgood(t *testing.T) {
|
||||||
assert.Equal(t, 3, vm.estack.Len())
|
assert.Equal(t, 3, vm.estack.Len())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestXDROPbadNoitem(t *testing.T) {
|
||||||
|
prog := makeProgram(XDROP)
|
||||||
|
vm := load(prog)
|
||||||
|
vm.Run()
|
||||||
|
assert.Equal(t, true, vm.state.HasFlag(faultState))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestXDROPbadNoN(t *testing.T) {
|
||||||
|
prog := makeProgram(XDROP)
|
||||||
|
vm := load(prog)
|
||||||
|
vm.estack.PushVal(1)
|
||||||
|
vm.estack.PushVal(2)
|
||||||
|
vm.Run()
|
||||||
|
assert.Equal(t, true, vm.state.HasFlag(faultState))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestXDROPbadNegative(t *testing.T) {
|
||||||
|
prog := makeProgram(XDROP)
|
||||||
|
vm := load(prog)
|
||||||
|
vm.estack.PushVal(1)
|
||||||
|
vm.estack.PushVal(-1)
|
||||||
|
vm.Run()
|
||||||
|
assert.Equal(t, true, vm.state.HasFlag(faultState))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestXDROPgood(t *testing.T) {
|
||||||
|
prog := makeProgram(XDROP)
|
||||||
|
vm := load(prog)
|
||||||
|
vm.estack.PushVal(0)
|
||||||
|
vm.estack.PushVal(1)
|
||||||
|
vm.estack.PushVal(2)
|
||||||
|
vm.estack.PushVal(2)
|
||||||
|
vm.Run()
|
||||||
|
assert.Equal(t, false, vm.state.HasFlag(faultState))
|
||||||
|
assert.Equal(t, 2, vm.estack.Len())
|
||||||
|
assert.Equal(t, int64(2), vm.estack.Peek(0).BigInt().Int64())
|
||||||
|
assert.Equal(t, int64(1), vm.estack.Peek(1).BigInt().Int64())
|
||||||
|
}
|
||||||
|
|
||||||
func makeProgram(opcodes ...Instruction) []byte {
|
func makeProgram(opcodes ...Instruction) []byte {
|
||||||
prog := make([]byte, len(opcodes)+1) // RET
|
prog := make([]byte, len(opcodes)+1) // RET
|
||||||
|
|
Loading…
Reference in a new issue