Add TS_GO_DEBUG_STACK_LIMIT to cap goroutine stack size at entry points (#3649)

Copilot · RyanCavanaugh · web-flow · commit 2c251880acd9 · 2026-04-29T20:02:57.000Z
Co-authored-by: copilot-swe-agent[bot] &lt;198982749+Copilot@users.noreply.github.com&gt;
Co-authored-by: RyanCavanaugh &lt;6685088+RyanCavanaugh@users.noreply.github.com&gt;
Co-authored-by: Ryan Cavanaugh &lt;RyanCavanaugh@users.noreply.github.com&gt;
diff --git a/cmd/tsgo/main.go b/cmd/tsgo/main.go
@@ -3,6 +3,7 @@ package main
 import (
 	"os"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/execute"
 )
 
@@ -11,6 +12,7 @@ func main() {
 }
 
 func runMain() int {
+	core.ApplyDebugStackLimit()
 	args := os.Args[1:]
 	if len(args) > 0 {
 		switch args[0] {
diff --git a/internal/api/encoder/testmain_test.go b/internal/api/encoder/testmain_test.go
@@ -3,10 +3,12 @@ package encoder_test
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }
diff --git a/internal/astnav/testmain_test.go b/internal/astnav/testmain_test.go
@@ -3,10 +3,12 @@ package astnav_test
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }
diff --git a/internal/core/core.go b/internal/core/core.go
@@ -4,8 +4,11 @@ import (
 	"iter"
 	"maps"
 	"math"
+	"os"
+	rtdebug "runtime/debug"
 	"slices"
 	"sort"
+	"strconv"
 	"strings"
 	"sync"
 	"unicode"
@@ -18,6 +21,18 @@ import (
 	"github.com/microsoft/typescript-go/internal/tspath"
 )
 
+func ApplyDebugStackLimit() {
+	v := os.Getenv("TS_GO_DEBUG_STACK_LIMIT") //nolint:forbidigo
+	if v == "" {
+		return
+	}
+	n, err := strconv.Atoi(v)
+	if err != nil || n <= 0 {
+		return
+	}
+	rtdebug.SetMaxStack(n)
+}
+
 func Filter[T any](slice []T, f func(T) bool) []T {
 	for i, value := range slice {
 		if !f(value) {
diff --git a/internal/execute/tsctests/testmain_test.go b/internal/execute/tsctests/testmain_test.go
@@ -3,10 +3,12 @@ package tsctests
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }
diff --git a/internal/fourslash/_scripts/convertFourslash.mts b/internal/fourslash/_scripts/convertFourslash.mts
@@ -82,10 +82,12 @@ export async function main() {
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }
diff --git a/internal/fourslash/tests/gen/testmain_test.go b/internal/fourslash/tests/gen/testmain_test.go
@@ -3,10 +3,12 @@ package fourslash_test
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }
diff --git a/internal/fourslash/tests/manual/testmain_test.go b/internal/fourslash/tests/manual/testmain_test.go
@@ -3,10 +3,12 @@ package fourslash_test
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }
diff --git a/internal/fourslash/tests/testmain_test.go b/internal/fourslash/tests/testmain_test.go
@@ -3,10 +3,12 @@ package fourslash_test
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }
diff --git a/internal/ls/autoimport/testmain_test.go b/internal/ls/autoimport/testmain_test.go
@@ -3,10 +3,12 @@ package autoimport_test
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }
diff --git a/internal/lsp/testmain_test.go b/internal/lsp/testmain_test.go
@@ -3,10 +3,12 @@ package lsp
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }
diff --git a/internal/project/ata/testmain_test.go b/internal/project/ata/testmain_test.go
@@ -3,10 +3,12 @@ package ata_test
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }
diff --git a/internal/project/testmain_test.go b/internal/project/testmain_test.go
@@ -3,10 +3,12 @@ package project_test
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }
diff --git a/internal/testrunner/testmain_test.go b/internal/testrunner/testmain_test.go
@@ -3,10 +3,12 @@ package testrunner
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }
diff --git a/internal/tsoptions/testmain_test.go b/internal/tsoptions/testmain_test.go
@@ -3,10 +3,12 @@ package tsoptions_test
 import (
 	"testing"
 
+	"github.com/microsoft/typescript-go/internal/core"
 	"github.com/microsoft/typescript-go/internal/testutil/baseline"
 )
 
 func TestMain(m *testing.M) {
+	core.ApplyDebugStackLimit()
 	defer baseline.Track()()
 	m.Run()
 }

Original file line number	Diff line number	Diff line change
`@@ -3,6 +3,7 @@ package main`
`3`	`3`	`import (`
`4`	`4`	`"os"`
`5`	`5`
	`6`	`+ "github.com/microsoft/typescript-go/internal/core"`
`6`	`7`	`"github.com/microsoft/typescript-go/internal/execute"`
`7`	`8`	`)`
`8`	`9`
`@@ -11,6 +12,7 @@ func main() {`
`11`	`12`	`}`
`12`	`13`
`13`	`14`	`func runMain() int {`
	`15`	`+ core.ApplyDebugStackLimit()`
`14`	`16`	`args := os.Args[1:]`
`15`	`17`	`if len(args) > 0 {`
`16`	`18`	`switch args[0] {`
Original file line number	Diff line number	Diff line change
`@@ -3,10 +3,12 @@ package encoder_test`
`3`	`3`	`import (`
`4`	`4`	`"testing"`
`5`	`5`
	`6`	`+ "github.com/microsoft/typescript-go/internal/core"`
`6`	`7`	`"github.com/microsoft/typescript-go/internal/testutil/baseline"`
`7`	`8`	`)`
`8`	`9`
`9`	`10`	`func TestMain(m *testing.M) {`
	`11`	`+ core.ApplyDebugStackLimit()`
`10`	`12`	`defer baseline.Track()()`
`11`	`13`	`m.Run()`
`12`	`14`	`}`
Original file line number	Diff line number	Diff line change
`@@ -3,10 +3,12 @@ package astnav_test`
`3`	`3`	`import (`
`4`	`4`	`"testing"`
`5`	`5`
	`6`	`+ "github.com/microsoft/typescript-go/internal/core"`
`6`	`7`	`"github.com/microsoft/typescript-go/internal/testutil/baseline"`
`7`	`8`	`)`
`8`	`9`
`9`	`10`	`func TestMain(m *testing.M) {`
	`11`	`+ core.ApplyDebugStackLimit()`
`10`	`12`	`defer baseline.Track()()`
`11`	`13`	`m.Run()`
`12`	`14`	`}`
Original file line number	Diff line number	Diff line change
`@@ -3,10 +3,12 @@ package tsctests`
`3`	`3`	`import (`
`4`	`4`	`"testing"`
`5`	`5`
	`6`	`+ "github.com/microsoft/typescript-go/internal/core"`
`6`	`7`	`"github.com/microsoft/typescript-go/internal/testutil/baseline"`
`7`	`8`	`)`
`8`	`9`
`9`	`10`	`func TestMain(m *testing.M) {`
	`11`	`+ core.ApplyDebugStackLimit()`
`10`	`12`	`defer baseline.Track()()`
`11`	`13`	`m.Run()`
`12`	`14`	`}`
Original file line number	Diff line number	Diff line change
`@@ -82,10 +82,12 @@ export async function main() {`
`82`	`82`	`import (`
`83`	`83`	`"testing"`
`84`	`84`
	`85`	`+ "github.com/microsoft/typescript-go/internal/core"`
`85`	`86`	`"github.com/microsoft/typescript-go/internal/testutil/baseline"`
`86`	`87`	`)`
`87`	`88`
`88`	`89`	`func TestMain(m *testing.M) {`
	`90`	`+ core.ApplyDebugStackLimit()`
`89`	`91`	`defer baseline.Track()()`
`90`	`92`	`m.Run()`
`91`	`93`	`}`
Original file line number	Diff line number	Diff line change
`@@ -3,10 +3,12 @@ package fourslash_test`
`3`	`3`	`import (`
`4`	`4`	`"testing"`
`5`	`5`
	`6`	`+ "github.com/microsoft/typescript-go/internal/core"`
`6`	`7`	`"github.com/microsoft/typescript-go/internal/testutil/baseline"`
`7`	`8`	`)`
`8`	`9`
`9`	`10`	`func TestMain(m *testing.M) {`
	`11`	`+ core.ApplyDebugStackLimit()`
`10`	`12`	`defer baseline.Track()()`
`11`	`13`	`m.Run()`
`12`	`14`	`}`
Original file line number	Diff line number	Diff line change
`@@ -3,10 +3,12 @@ package autoimport_test`
`3`	`3`	`import (`
`4`	`4`	`"testing"`
`5`	`5`
	`6`	`+ "github.com/microsoft/typescript-go/internal/core"`
`6`	`7`	`"github.com/microsoft/typescript-go/internal/testutil/baseline"`
`7`	`8`	`)`
`8`	`9`
`9`	`10`	`func TestMain(m *testing.M) {`
	`11`	`+ core.ApplyDebugStackLimit()`
`10`	`12`	`defer baseline.Track()()`
`11`	`13`	`m.Run()`
`12`	`14`	`}`