runtime_unix_test.go

Documentation: runtime

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