TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-12-25 03:56:34 +00:00
Code Issues Projects Releases Packages Wiki Activity

vm: fail if DROP has no argument

Browse source
This commit is contained in:
Evgenii Stratonikov 2019-09-12 11:05:10 +03:00
parent 729b7a0b24
commit 6e4014547d
2 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
pkg/vm/vm.go
View file

@ -439,6 +439,9 @@ func (v *VM) execute(ctx *Context, op Instruction) {
} }
case DROP: case DROP:
if v.estack.Len() < 1 {
panic("stack is too small")
}
v.estack.Pop() v.estack.Pop()
case EQUAL: case EQUAL:

16
pkg/vm/vm_test.go
View file

@ -803,6 +803,22 @@ func TestOVERgood(t *testing.T) {
assert.Equal(t, 3, vm.estack.Len()) assert.Equal(t, 3, vm.estack.Len())
} }
func TestDROPBadNoItem(t *testing.T) {
prog := makeProgram(DROP)
vm := load(prog)
vm.Run()
assert.Equal(t, true, vm.state.HasFlag(faultState))
}
func TestDROPGood(t *testing.T) {
prog := makeProgram(DROP)
vm := load(prog)
vm.estack.PushVal(1)
vm.Run()
assert.Equal(t, false, vm.state.HasFlag(faultState))
assert.Equal(t, 0, vm.estack.Len())
}
func TestXDROPbadNoitem(t *testing.T) { func TestXDROPbadNoitem(t *testing.T) {
prog := makeProgram(XDROP) prog := makeProgram(XDROP)
vm := load(prog) vm := load(prog)