liyujie
2025-08-28 d9927380ed7c8366f762049be9f3fee225860833
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// Copyright 2013 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
 
// Only works on systems with syscall.Close.
// We need a fast system call to provoke the race,
// and Close(-1) is nearly universally fast.
 
// +build aix darwin dragonfly freebsd linux netbsd openbsd plan9
 
package runtime_test
 
import (
   "runtime"
   "sync"
   "sync/atomic"
   "syscall"
   "testing"
)
 
func TestGoroutineProfile(t *testing.T) {
   // GoroutineProfile used to use the wrong starting sp for
   // goroutines coming out of system calls, causing possible
   // crashes.
   defer runtime.GOMAXPROCS(runtime.GOMAXPROCS(100))
 
   var stop uint32
   defer atomic.StoreUint32(&stop, 1) // in case of panic
 
   var wg sync.WaitGroup
   for i := 0; i < 4; i++ {
       wg.Add(1)
       go func() {
           for atomic.LoadUint32(&stop) == 0 {
               syscall.Close(-1)
           }
           wg.Done()
       }()
   }
 
   max := 10000
   if testing.Short() {
       max = 100
   }
   stk := make([]runtime.StackRecord, 128)
   for n := 0; n < max; n++ {
       _, ok := runtime.GoroutineProfile(stk)
       if !ok {
           t.Fatalf("GoroutineProfile failed")
       }
   }
 
   // If the program didn't crash, we passed.
   atomic.StoreUint32(&stop, 1)
   wg.Wait()
}