From 64d438c91ace51bfda7bff50d381b62f5785f72a Mon Sep 17 00:00:00 2001 From: Mel Date: Tue, 16 Aug 2022 19:49:59 +0000 Subject: Ensure correct argument stack order in all cases --- pkg/lang/compiler/compiler.go | 8 +++----- pkg/lang/compiler/compiler_test.go | 7 +++---- 2 files changed, 6 insertions(+), 9 deletions(-) (limited to 'pkg/lang/compiler') diff --git a/pkg/lang/compiler/compiler.go b/pkg/lang/compiler/compiler.go index d647aff..631dfec 100644 --- a/pkg/lang/compiler/compiler.go +++ b/pkg/lang/compiler/compiler.go @@ -795,8 +795,8 @@ func (comp *Compiler) compileCallExpr(t *code.Builder, expr ast.ExprCall) error return err } - for i := 0; i < len(expr.Args); i++ { - if err := comp.compileExpr(t, expr.Args[i]); err != nil { + for _, arg := range expr.Args { + if err := comp.compileExpr(t, arg); err != nil { return err } } @@ -965,9 +965,7 @@ func (comp *Compiler) compileBlockNode(t *code.Builder, block ast.BlockNode) err } func (comp *Compiler) compileFn(t *code.Builder, block ast.BlockNode, args []ast.IdentNode, addMissingReturn bool) error { - // Arguments are declared in reverse - for i := len(args) - 1; i >= 0; i-- { - arg := args[i] + for _, arg := range args { if _, ok := comp.scopes.Declare(arg.Value); !ok { return fmt.Errorf("variable %s already declared", arg.Value) } diff --git a/pkg/lang/compiler/compiler_test.go b/pkg/lang/compiler/compiler_test.go index 725620c..9e05230 100644 --- a/pkg/lang/compiler/compiler_test.go +++ b/pkg/lang/compiler/compiler_test.go @@ -457,7 +457,6 @@ func TestSimpleFunction(t *testing.T) { } func TestFunctionArgs(t *testing.T) { - // TODO: Are arguments in the correct order? src := ` fn add(a, b) { return a + b @@ -478,8 +477,8 @@ func TestFunctionArgs(t *testing.T) { halt @add: - get_local 1 get_local 0 + get_local 1 add ret ` @@ -581,11 +580,11 @@ func TestType(t *testing.T) { anchor_type get_local 2 - get_local 1 + get_local 0 set_member "name" get_local 2 - get_local 0 + get_local 1 set_member "age" get_local 2 -- cgit 1.4.1