cmd/go: fix data race in TestScript

Fixes golang#54423
cuiweixie · Aug 14, 2022 · 13faa5c · 13faa5c
1 parent 2f6783c
commit 13faa5c
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/cmd/go/internal/base/base.go b/src/cmd/go/internal/base/base.go
@@ -104,16 +104,23 @@ func (c *Command) Runnable() bool {
 	return c.Run != nil
 }
 
-var atExitFuncs []func()
+var (
+	atExitFuncsLock sync.Mutex
+	atExitFuncs     []func()
+)
 
 func AtExit(f func()) {
+	atExitFuncsLock.Lock()
+	defer atExitFuncsLock.Unlock()
 	atExitFuncs = append(atExitFuncs, f)
 }
 
 func Exit() {
+	atExitFuncsLock.Lock()
 	for _, f := range atExitFuncs {
 		f()
 	}
+	atExitFuncsLock.Unlock()
 	os.Exit(exitStatus)
 }