From a5b5f88fe2a2f4c7968dbcad977895299391282d Mon Sep 17 00:00:00 2001 From: Anna Shaleva Date: Mon, 6 Jun 2022 11:41:28 +0300 Subject: [PATCH] core: doesn't allow to deploy contract with invalid method offset This commit partially reverts 16bf7c1426bb88b371b08a26abb911fd953556ce. --- pkg/core/native/management.go | 2 +- pkg/core/native/native_test/management_test.go | 12 ++---------- 2 files changed, 3 insertions(+), 11 deletions(-) diff --git a/pkg/core/native/management.go b/pkg/core/native/management.go index eee25bef6..3adc703ed 100644 --- a/pkg/core/native/management.go +++ b/pkg/core/native/management.go @@ -614,7 +614,7 @@ func checkScriptAndMethods(script []byte, methods []manifest.Method) error { offsets := bitfield.New(l) for i := range methods { if methods[i].Offset >= l { - continue + return fmt.Errorf("method %s/%d: offset is out of the script range", methods[i].Name, len(methods[i].Parameters)) } offsets.Set(methods[i].Offset) } diff --git a/pkg/core/native/native_test/management_test.go b/pkg/core/native/native_test/management_test.go index 4f2eb3507..ffbc9f37a 100644 --- a/pkg/core/native/native_test/management_test.go +++ b/pkg/core/native/native_test/management_test.go @@ -135,19 +135,11 @@ func TestManagement_ContractDeploy(t *testing.T) { badManifest := cs1.Manifest badManifest.ABI.Methods = make([]manifest.Method, len(cs1.Manifest.ABI.Methods)) copy(badManifest.ABI.Methods, cs1.Manifest.ABI.Methods) - badManifest.ABI.Methods[0].Offset = 100500 // out of bounds, but it's OK, this method will not be checked then. + badManifest.ABI.Methods[0].Offset = 100500 // out of bounds manifB, err := json.Marshal(&badManifest) require.NoError(t, err) - tx := c.PrepareInvokeNoSign(t, "deploy", nefBytes, manifB) - tx.Signers = []transaction.Signer{{}} // Need dummy signer to deploy. - b := c.NewUnsignedBlock(t, tx) - ic := c.Chain.GetTestVM(trigger.Application, tx, b) - t.Cleanup(ic.Finalize) - - ic.VM.LoadWithFlags(tx.Script, callflag.All) - err = ic.VM.Run() - require.NoError(t, err) + managementInvoker.InvokeFail(t, "method add/2: offset is out of the script range", "deploy", nefBytes, manifB) }) t.Run("bad methods in manifest 2", func(t *testing.T) { var badManifest = cs1.Manifest