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

Merge pull request #2206 from nspcc-dev/compiler-safe-methods

Browse source 
compiler: check that safe methods exist
This commit is contained in:
Roman Khimov 2021-10-07 14:53:37 +03:00 committed by GitHub
commit fb31a81fd2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
pkg/compiler/compiler.go
View file

@ -270,6 +270,11 @@ func CreateManifest(di *DebugInfo, o *Options) (*manifest.Manifest, error) {
if err != nil { if err != nil {
return m, fmt.Errorf("failed to convert debug info to manifest: %w", err) return m, fmt.Errorf("failed to convert debug info to manifest: %w", err)
} }
for _, name := range o.SafeMethods {
if m.ABI.GetMethod(name, -1) == nil {
return m, fmt.Errorf("method %s is marked as safe but missing from manifest", name)
}
}
if !o.NoStandardCheck { if !o.NoStandardCheck {
if err := standard.CheckABI(m, o.ContractSupportedStandards...); err != nil { if err := standard.CheckABI(m, o.ContractSupportedStandards...); err != nil {
return m, err return m, err

14
pkg/compiler/compiler_test.go
View file

@ -125,6 +125,20 @@ func TestOnPayableChecks(t *testing.T) {
}) })
} }
func TestSafeMethodWarnings(t *testing.T) {
src := `package payable
func Main() int { return 1 }`
_, di, err := compiler.CompileWithDebugInfo("eventTest", strings.NewReader(src))
require.NoError(t, err)
_, err = compiler.CreateManifest(di, &compiler.Options{SafeMethods: []string{"main"}})
require.NoError(t, err)
_, err = compiler.CreateManifest(di, &compiler.Options{SafeMethods: []string{"main", "mississippi"}})
require.Error(t, err)
}
func TestEventWarnings(t *testing.T) { func TestEventWarnings(t *testing.T) {
src := `package payable src := `package payable
import "github.com/nspcc-dev/neo-go/pkg/interop/runtime" import "github.com/nspcc-dev/neo-go/pkg/interop/runtime"