TrueCloudLab/neo-go
6
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-10-16 15:43:25 +00:00
Code Issues Projects Releases Packages Wiki Activity

vm: panic in cloneIfStruct

Browse source 
It's internal, so we can deduplicate code a bit.
This commit is contained in:
Roman Khimov 2021-07-12 11:51:49 +03:00
parent 4000dd692c
commit fe3c68b92d
1 changed files with 10 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
pkg/vm/vm.go
View file

@ -1056,10 +1056,7 @@ func (v *VM) execute(ctx *Context, op opcode.Opcode, parameter []byte) (err erro
itemElem := v.estack.Pop()
arrElem := v.estack.Pop()
val, err := cloneIfStruct(itemElem.value, MaxStackSize-v.refs.size)
if err != nil {
panic(err)
}
val := cloneIfStruct(itemElem.value, MaxStackSize-v.refs.size)
switch t := arrElem.value.(type) {
case *stackitem.Array:
@ -1373,19 +1370,12 @@ func (v *VM) execute(ctx *Context, op opcode.Opcode, parameter []byte) (err erro
src := t.Value().([]stackitem.Item)
arr = make([]stackitem.Item, len(src))
for i := range src {
arr[i], err = cloneIfStruct(src[i], MaxStackSize-v.refs.size)
if err != nil {
panic(err)
}
arr[i] = cloneIfStruct(src[i], MaxStackSize-v.refs.size)
}
case *stackitem.Map:
arr = make([]stackitem.Item, 0, t.Len())
for k := range t.Value().([]stackitem.MapElement) {
elem, err := cloneIfStruct(t.Value().([]stackitem.MapElement)[k].Value, MaxStackSize-v.refs.size)
if err != nil {
panic(err)
}
arr = append(arr, elem)
arr = append(arr, cloneIfStruct(t.Value().([]stackitem.MapElement)[k].Value, MaxStackSize-v.refs.size))
}
default:
panic("not a Map, Array or Struct")
@ -1751,12 +1741,16 @@ func checkMultisig1(v *VM, curve elliptic.Curve, h []byte, pkeys [][]byte, sig [
return false
}
func cloneIfStruct(item stackitem.Item, limit int) (stackitem.Item, error) {
func cloneIfStruct(item stackitem.Item, limit int) stackitem.Item {
switch it := item.(type) {
case *stackitem.Struct:
return it.Clone(limit)
ret, err := it.Clone(limit)
if err != nil {
panic(err)
}
return ret
default:
return it, nil
return it
}
}