TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2025-05-03 23:02:27 +00:00
Code Issues Projects Releases Packages Wiki Activity

compiler: calculate stack size more precisely

Browse source 
When calculating number of local variables, only
defining assignments (i.e. via `:=`) must be taken into account.
This commit is contained in:
Evgenii Stratonikov 2020-04-01 17:25:08 +03:00
parent 9b5dab57e8
commit a43c9b9246
2 changed files with 21 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
pkg/compiler/func_scope.go
View file

@ -2,6 +2,7 @@ package compiler
import (
"go/ast"
"go/token"
)
// A funcScope represents the scope within the function context.
@ -76,7 +77,9 @@ func (c *funcScope) stackSize() int64 {
ast.Inspect(c.decl, func(n ast.Node) bool {
switch n := n.(type) {
case *ast.AssignStmt:
size += len(n.Rhs)
if n.Tok == token.DEFINE {
size += len(n.Rhs)
}
case *ast.ReturnStmt, *ast.IfStmt:
size++
// This handles the inline GenDecl like "var x = 2"