TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

core: check call flags in LoadToken

Browse source 
Follow neo-project/neo#2292.
This commit is contained in:
Evgeniy Stratonikov 2021-02-25 18:14:46 +03:00
parent 4db5ef05f8
commit 663afbe4df
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
pkg/core/interop/contract/call.go
View file

@ -19,6 +19,9 @@ import (
func LoadToken(ic *interop.Context) func(id int32) error { func LoadToken(ic *interop.Context) func(id int32) error {
return func(id int32) error { return func(id int32) error {
ctx := ic.VM.Context() ctx := ic.VM.Context()
if !ctx.GetCallFlags().Has(callflag.ReadStates | callflag.AllowCall) {
return errors.New("invalid call flags")
}
tok := ctx.NEF.Tokens[id] tok := ctx.NEF.Tokens[id]
if int(tok.ParamCount) > ctx.Estack().Len() { if int(tok.ParamCount) > ctx.Estack().Len() {
return errors.New("stack is too small") return errors.New("stack is too small")