From e015cce9761661ac8af9b65a73c3b0f167ad574b Mon Sep 17 00:00:00 2001
From: Soner Sayakci <s.sayakci@shopware.com>
Date: Wed, 11 Oct 2023 09:10:04 +0200
Subject: [PATCH] fix: memory leak in wasm creation

---
 internal/phplint/lint.go | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/internal/phplint/lint.go b/internal/phplint/lint.go
index 1dbbc77d..a94ca422 100644
--- a/internal/phplint/lint.go
+++ b/internal/phplint/lint.go
@@ -5,6 +5,7 @@ import (
 	"io/fs"
 	"os"
 	"path/filepath"
+	"runtime"
 	"strings"
 
 	"github.com/tetratelabs/wazero"
@@ -24,12 +25,14 @@ func LintFolder(ctx context.Context, phpVersion, folder string) (LintErrors, err
 		return nil, err
 	}
 
-	runtime, err := getWazeroRuntime(ctx)
+	wasmRuntime, err := getWazeroRuntime(ctx)
 	if err != nil {
 		return nil, err
 	}
 
-	wasmCompiled, _ := runtime.CompileModule(ctx, wasmFile)
+	defer wasmRuntime.Close(ctx)
+
+	wasmCompiled, _ := wasmRuntime.CompileModule(ctx, wasmFile)
 
 	dirFs := os.DirFS(folder)
 
@@ -47,6 +50,8 @@ func LintFolder(ctx context.Context, phpVersion, folder string) (LintErrors, err
 
 	errorsChain := make(chan *LintError, len(paths))
 
+	runtime.GOMAXPROCS(2)
+
 	for _, file := range paths {
 		go func(file string) {
 			file, _ = filepath.Rel(folder, file)
@@ -58,7 +63,7 @@ func LintFolder(ctx context.Context, phpVersion, folder string) (LintErrors, err
 				WithArgs("php", "-l", file).
 				WithFS(dirFs)
 
-			if _, err := runtime.InstantiateModule(ctx, wasmCompiled, config); err != nil {
+			if wasmModule, err := wasmRuntime.InstantiateModule(ctx, wasmCompiled, config); err != nil {
 				if exitErr, ok := err.(*sys.ExitError); ok && exitErr.ExitCode() != 0 {
 					errorsChain <- &LintError{
 						File:    file,
@@ -72,7 +77,12 @@ func LintFolder(ctx context.Context, phpVersion, folder string) (LintErrors, err
 				} else {
 					errorsChain <- nil
 				}
+
+				if wasmModule != nil {
+					wasmModule.Close(ctx)
+				}
 			} else {
+				wasmModule.Close(ctx)
 				errorsChain <- nil
 			}
 		}(file)