github.com/eun/go@v0.0.0-20170811110501-92cfd07a6cfd/src/runtime/pprof/pprof.go (about)

     1  // Copyright 2010 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  // Package pprof writes runtime profiling data in the format expected
     6  // by the pprof visualization tool.
     7  //
     8  // Profiling a Go program
     9  //
    10  // The first step to profiling a Go program is to enable profiling.
    11  // Support for profiling benchmarks built with the standard testing
    12  // package is built into go test. For example, the following command
    13  // runs benchmarks in the current directory and writes the CPU and
    14  // memory profiles to cpu.prof and mem.prof:
    15  //
    16  //     go test -cpuprofile cpu.prof -memprofile mem.prof -bench .
    17  //
    18  // To add equivalent profiling support to a standalone program, add
    19  // code like the following to your main function:
    20  //
    21  //    var cpuprofile = flag.String("cpuprofile", "", "write cpu profile `file`")
    22  //    var memprofile = flag.String("memprofile", "", "write memory profile to `file`")
    23  //
    24  //    func main() {
    25  //        flag.Parse()
    26  //        if *cpuprofile != "" {
    27  //            f, err := os.Create(*cpuprofile)
    28  //            if err != nil {
    29  //                log.Fatal("could not create CPU profile: ", err)
    30  //            }
    31  //            if err := pprof.StartCPUProfile(f); err != nil {
    32  //                log.Fatal("could not start CPU profile: ", err)
    33  //            }
    34  //            defer pprof.StopCPUProfile()
    35  //        }
    36  //
    37  //        // ... rest of the program ...
    38  //
    39  //        if *memprofile != "" {
    40  //            f, err := os.Create(*memprofile)
    41  //            if err != nil {
    42  //                log.Fatal("could not create memory profile: ", err)
    43  //            }
    44  //            runtime.GC() // get up-to-date statistics
    45  //            if err := pprof.WriteHeapProfile(f); err != nil {
    46  //                log.Fatal("could not write memory profile: ", err)
    47  //            }
    48  //            f.Close()
    49  //        }
    50  //    }
    51  //
    52  // There is also a standard HTTP interface to profiling data. Adding
    53  // the following line will install handlers under the /debug/pprof/
    54  // URL to download live profiles:
    55  //
    56  //    import _ "net/http/pprof"
    57  //
    58  // See the net/http/pprof package for more details.
    59  //
    60  // Profiles can then be visualized with the pprof tool:
    61  //
    62  //    go tool pprof cpu.prof
    63  //
    64  // There are many commands available from the pprof command line.
    65  // Commonly used commands include "top", which prints a summary of the
    66  // top program hot-spots, and "web", which opens an interactive graph
    67  // of hot-spots and their call graphs. Use "help" for information on
    68  // all pprof commands.
    69  //
    70  // For more information about pprof, see
    71  // https://github.com/google/pprof/blob/master/doc/pprof.md.
    72  package pprof
    73  
    74  import (
    75  	"bufio"
    76  	"bytes"
    77  	"fmt"
    78  	"io"
    79  	"runtime"
    80  	"sort"
    81  	"strings"
    82  	"sync"
    83  	"text/tabwriter"
    84  	"time"
    85  	"unsafe"
    86  )
    87  
    88  // BUG(rsc): Profiles are only as good as the kernel support used to generate them.
    89  // See https://golang.org/issue/13841 for details about known problems.
    90  
    91  // A Profile is a collection of stack traces showing the call sequences
    92  // that led to instances of a particular event, such as allocation.
    93  // Packages can create and maintain their own profiles; the most common
    94  // use is for tracking resources that must be explicitly closed, such as files
    95  // or network connections.
    96  //
    97  // A Profile's methods can be called from multiple goroutines simultaneously.
    98  //
    99  // Each Profile has a unique name. A few profiles are predefined:
   100  //
   101  //	goroutine    - stack traces of all current goroutines
   102  //	heap         - a sampling of all heap allocations
   103  //	threadcreate - stack traces that led to the creation of new OS threads
   104  //	block        - stack traces that led to blocking on synchronization primitives
   105  //	mutex        - stack traces of holders of contended mutexes
   106  //
   107  // These predefined profiles maintain themselves and panic on an explicit
   108  // Add or Remove method call.
   109  //
   110  // The heap profile reports statistics as of the most recently completed
   111  // garbage collection; it elides more recent allocation to avoid skewing
   112  // the profile away from live data and toward garbage.
   113  // If there has been no garbage collection at all, the heap profile reports
   114  // all known allocations. This exception helps mainly in programs running
   115  // without garbage collection enabled, usually for debugging purposes.
   116  //
   117  // The CPU profile is not available as a Profile. It has a special API,
   118  // the StartCPUProfile and StopCPUProfile functions, because it streams
   119  // output to a writer during profiling.
   120  //
   121  type Profile struct {
   122  	name  string
   123  	mu    sync.Mutex
   124  	m     map[interface{}][]uintptr
   125  	count func() int
   126  	write func(io.Writer, int) error
   127  }
   128  
   129  // profiles records all registered profiles.
   130  var profiles struct {
   131  	mu sync.Mutex
   132  	m  map[string]*Profile
   133  }
   134  
   135  var goroutineProfile = &Profile{
   136  	name:  "goroutine",
   137  	count: countGoroutine,
   138  	write: writeGoroutine,
   139  }
   140  
   141  var threadcreateProfile = &Profile{
   142  	name:  "threadcreate",
   143  	count: countThreadCreate,
   144  	write: writeThreadCreate,
   145  }
   146  
   147  var heapProfile = &Profile{
   148  	name:  "heap",
   149  	count: countHeap,
   150  	write: writeHeap,
   151  }
   152  
   153  var blockProfile = &Profile{
   154  	name:  "block",
   155  	count: countBlock,
   156  	write: writeBlock,
   157  }
   158  
   159  var mutexProfile = &Profile{
   160  	name:  "mutex",
   161  	count: countMutex,
   162  	write: writeMutex,
   163  }
   164  
   165  func lockProfiles() {
   166  	profiles.mu.Lock()
   167  	if profiles.m == nil {
   168  		// Initial built-in profiles.
   169  		profiles.m = map[string]*Profile{
   170  			"goroutine":    goroutineProfile,
   171  			"threadcreate": threadcreateProfile,
   172  			"heap":         heapProfile,
   173  			"block":        blockProfile,
   174  			"mutex":        mutexProfile,
   175  		}
   176  	}
   177  }
   178  
   179  func unlockProfiles() {
   180  	profiles.mu.Unlock()
   181  }
   182  
   183  // NewProfile creates a new profile with the given name.
   184  // If a profile with that name already exists, NewProfile panics.
   185  // The convention is to use a 'import/path.' prefix to create
   186  // separate name spaces for each package.
   187  // For compatibility with various tools that read pprof data,
   188  // profile names should not contain spaces.
   189  func NewProfile(name string) *Profile {
   190  	lockProfiles()
   191  	defer unlockProfiles()
   192  	if name == "" {
   193  		panic("pprof: NewProfile with empty name")
   194  	}
   195  	if profiles.m[name] != nil {
   196  		panic("pprof: NewProfile name already in use: " + name)
   197  	}
   198  	p := &Profile{
   199  		name: name,
   200  		m:    map[interface{}][]uintptr{},
   201  	}
   202  	profiles.m[name] = p
   203  	return p
   204  }
   205  
   206  // Lookup returns the profile with the given name, or nil if no such profile exists.
   207  func Lookup(name string) *Profile {
   208  	lockProfiles()
   209  	defer unlockProfiles()
   210  	return profiles.m[name]
   211  }
   212  
   213  // Profiles returns a slice of all the known profiles, sorted by name.
   214  func Profiles() []*Profile {
   215  	lockProfiles()
   216  	defer unlockProfiles()
   217  
   218  	all := make([]*Profile, 0, len(profiles.m))
   219  	for _, p := range profiles.m {
   220  		all = append(all, p)
   221  	}
   222  
   223  	sort.Slice(all, func(i, j int) bool { return all[i].name < all[j].name })
   224  	return all
   225  }
   226  
   227  // Name returns this profile's name, which can be passed to Lookup to reobtain the profile.
   228  func (p *Profile) Name() string {
   229  	return p.name
   230  }
   231  
   232  // Count returns the number of execution stacks currently in the profile.
   233  func (p *Profile) Count() int {
   234  	p.mu.Lock()
   235  	defer p.mu.Unlock()
   236  	if p.count != nil {
   237  		return p.count()
   238  	}
   239  	return len(p.m)
   240  }
   241  
   242  // Add adds the current execution stack to the profile, associated with value.
   243  // Add stores value in an internal map, so value must be suitable for use as
   244  // a map key and will not be garbage collected until the corresponding
   245  // call to Remove. Add panics if the profile already contains a stack for value.
   246  //
   247  // The skip parameter has the same meaning as runtime.Caller's skip
   248  // and controls where the stack trace begins. Passing skip=0 begins the
   249  // trace in the function calling Add. For example, given this
   250  // execution stack:
   251  //
   252  //	Add
   253  //	called from rpc.NewClient
   254  //	called from mypkg.Run
   255  //	called from main.main
   256  //
   257  // Passing skip=0 begins the stack trace at the call to Add inside rpc.NewClient.
   258  // Passing skip=1 begins the stack trace at the call to NewClient inside mypkg.Run.
   259  //
   260  func (p *Profile) Add(value interface{}, skip int) {
   261  	if p.name == "" {
   262  		panic("pprof: use of uninitialized Profile")
   263  	}
   264  	if p.write != nil {
   265  		panic("pprof: Add called on built-in Profile " + p.name)
   266  	}
   267  
   268  	stk := make([]uintptr, 32)
   269  	n := runtime.Callers(skip+1, stk[:])
   270  	stk = stk[:n]
   271  	if len(stk) == 0 {
   272  		// The value for skip is too large, and there's no stack trace to record.
   273  		stk = []uintptr{funcPC(lostProfileEvent)}
   274  	}
   275  
   276  	p.mu.Lock()
   277  	defer p.mu.Unlock()
   278  	if p.m[value] != nil {
   279  		panic("pprof: Profile.Add of duplicate value")
   280  	}
   281  	p.m[value] = stk
   282  }
   283  
   284  // Remove removes the execution stack associated with value from the profile.
   285  // It is a no-op if the value is not in the profile.
   286  func (p *Profile) Remove(value interface{}) {
   287  	p.mu.Lock()
   288  	defer p.mu.Unlock()
   289  	delete(p.m, value)
   290  }
   291  
   292  // WriteTo writes a pprof-formatted snapshot of the profile to w.
   293  // If a write to w returns an error, WriteTo returns that error.
   294  // Otherwise, WriteTo returns nil.
   295  //
   296  // The debug parameter enables additional output.
   297  // Passing debug=0 prints only the hexadecimal addresses that pprof needs.
   298  // Passing debug=1 adds comments translating addresses to function names
   299  // and line numbers, so that a programmer can read the profile without tools.
   300  //
   301  // The predefined profiles may assign meaning to other debug values;
   302  // for example, when printing the "goroutine" profile, debug=2 means to
   303  // print the goroutine stacks in the same form that a Go program uses
   304  // when dying due to an unrecovered panic.
   305  func (p *Profile) WriteTo(w io.Writer, debug int) error {
   306  	if p.name == "" {
   307  		panic("pprof: use of zero Profile")
   308  	}
   309  	if p.write != nil {
   310  		return p.write(w, debug)
   311  	}
   312  
   313  	// Obtain consistent snapshot under lock; then process without lock.
   314  	p.mu.Lock()
   315  	all := make([][]uintptr, 0, len(p.m))
   316  	for _, stk := range p.m {
   317  		all = append(all, stk)
   318  	}
   319  	p.mu.Unlock()
   320  
   321  	// Map order is non-deterministic; make output deterministic.
   322  	sort.Sort(stackProfile(all))
   323  
   324  	return printCountProfile(w, debug, p.name, stackProfile(all))
   325  }
   326  
   327  type stackProfile [][]uintptr
   328  
   329  func (x stackProfile) Len() int              { return len(x) }
   330  func (x stackProfile) Stack(i int) []uintptr { return x[i] }
   331  func (x stackProfile) Swap(i, j int)         { x[i], x[j] = x[j], x[i] }
   332  func (x stackProfile) Less(i, j int) bool {
   333  	t, u := x[i], x[j]
   334  	for k := 0; k < len(t) && k < len(u); k++ {
   335  		if t[k] != u[k] {
   336  			return t[k] < u[k]
   337  		}
   338  	}
   339  	return len(t) < len(u)
   340  }
   341  
   342  // A countProfile is a set of stack traces to be printed as counts
   343  // grouped by stack trace. There are multiple implementations:
   344  // all that matters is that we can find out how many traces there are
   345  // and obtain each trace in turn.
   346  type countProfile interface {
   347  	Len() int
   348  	Stack(i int) []uintptr
   349  }
   350  
   351  // printCountProfile prints a countProfile at the specified debug level.
   352  // The profile will be in compressed proto format unless debug is nonzero.
   353  func printCountProfile(w io.Writer, debug int, name string, p countProfile) error {
   354  	// Build count of each stack.
   355  	var buf bytes.Buffer
   356  	key := func(stk []uintptr) string {
   357  		buf.Reset()
   358  		fmt.Fprintf(&buf, "@")
   359  		for _, pc := range stk {
   360  			fmt.Fprintf(&buf, " %#x", pc)
   361  		}
   362  		return buf.String()
   363  	}
   364  	count := map[string]int{}
   365  	index := map[string]int{}
   366  	var keys []string
   367  	n := p.Len()
   368  	for i := 0; i < n; i++ {
   369  		k := key(p.Stack(i))
   370  		if count[k] == 0 {
   371  			index[k] = i
   372  			keys = append(keys, k)
   373  		}
   374  		count[k]++
   375  	}
   376  
   377  	sort.Sort(&keysByCount{keys, count})
   378  
   379  	if debug > 0 {
   380  		// Print debug profile in legacy format
   381  		tw := tabwriter.NewWriter(w, 1, 8, 1, '\t', 0)
   382  		fmt.Fprintf(tw, "%s profile: total %d\n", name, p.Len())
   383  		for _, k := range keys {
   384  			fmt.Fprintf(tw, "%d %s\n", count[k], k)
   385  			printStackRecord(tw, p.Stack(index[k]), false)
   386  		}
   387  		return tw.Flush()
   388  	}
   389  
   390  	// Output profile in protobuf form.
   391  	b := newProfileBuilder(w)
   392  	b.pbValueType(tagProfile_PeriodType, name, "count")
   393  	b.pb.int64Opt(tagProfile_Period, 1)
   394  	b.pbValueType(tagProfile_SampleType, name, "count")
   395  
   396  	values := []int64{0}
   397  	var locs []uint64
   398  	for _, k := range keys {
   399  		values[0] = int64(count[k])
   400  		locs = locs[:0]
   401  		for _, addr := range p.Stack(index[k]) {
   402  			// For count profiles, all stack addresses are
   403  			// return PCs, which is what locForPC expects.
   404  			l := b.locForPC(addr)
   405  			if l == 0 { // runtime.goexit
   406  				continue
   407  			}
   408  			locs = append(locs, l)
   409  		}
   410  		b.pbSample(values, locs, nil)
   411  	}
   412  	b.build()
   413  	return nil
   414  }
   415  
   416  // keysByCount sorts keys with higher counts first, breaking ties by key string order.
   417  type keysByCount struct {
   418  	keys  []string
   419  	count map[string]int
   420  }
   421  
   422  func (x *keysByCount) Len() int      { return len(x.keys) }
   423  func (x *keysByCount) Swap(i, j int) { x.keys[i], x.keys[j] = x.keys[j], x.keys[i] }
   424  func (x *keysByCount) Less(i, j int) bool {
   425  	ki, kj := x.keys[i], x.keys[j]
   426  	ci, cj := x.count[ki], x.count[kj]
   427  	if ci != cj {
   428  		return ci > cj
   429  	}
   430  	return ki < kj
   431  }
   432  
   433  // printStackRecord prints the function + source line information
   434  // for a single stack trace.
   435  func printStackRecord(w io.Writer, stk []uintptr, allFrames bool) {
   436  	show := allFrames
   437  	frames := runtime.CallersFrames(stk)
   438  	for {
   439  		frame, more := frames.Next()
   440  		name := frame.Function
   441  		if name == "" {
   442  			show = true
   443  			fmt.Fprintf(w, "#\t%#x\n", frame.PC)
   444  		} else if name != "runtime.goexit" && (show || !strings.HasPrefix(name, "runtime.")) {
   445  			// Hide runtime.goexit and any runtime functions at the beginning.
   446  			// This is useful mainly for allocation traces.
   447  			show = true
   448  			fmt.Fprintf(w, "#\t%#x\t%s+%#x\t%s:%d\n", frame.PC, name, frame.PC-frame.Entry, frame.File, frame.Line)
   449  		}
   450  		if !more {
   451  			break
   452  		}
   453  	}
   454  	if !show {
   455  		// We didn't print anything; do it again,
   456  		// and this time include runtime functions.
   457  		printStackRecord(w, stk, true)
   458  		return
   459  	}
   460  	fmt.Fprintf(w, "\n")
   461  }
   462  
   463  // Interface to system profiles.
   464  
   465  // WriteHeapProfile is shorthand for Lookup("heap").WriteTo(w, 0).
   466  // It is preserved for backwards compatibility.
   467  func WriteHeapProfile(w io.Writer) error {
   468  	return writeHeap(w, 0)
   469  }
   470  
   471  // countHeap returns the number of records in the heap profile.
   472  func countHeap() int {
   473  	n, _ := runtime.MemProfile(nil, true)
   474  	return n
   475  }
   476  
   477  // writeHeap writes the current runtime heap profile to w.
   478  func writeHeap(w io.Writer, debug int) error {
   479  	// Find out how many records there are (MemProfile(nil, true)),
   480  	// allocate that many records, and get the data.
   481  	// There's a race—more records might be added between
   482  	// the two calls—so allocate a few extra records for safety
   483  	// and also try again if we're very unlucky.
   484  	// The loop should only execute one iteration in the common case.
   485  	var p []runtime.MemProfileRecord
   486  	n, ok := runtime.MemProfile(nil, true)
   487  	for {
   488  		// Allocate room for a slightly bigger profile,
   489  		// in case a few more entries have been added
   490  		// since the call to MemProfile.
   491  		p = make([]runtime.MemProfileRecord, n+50)
   492  		n, ok = runtime.MemProfile(p, true)
   493  		if ok {
   494  			p = p[0:n]
   495  			break
   496  		}
   497  		// Profile grew; try again.
   498  	}
   499  
   500  	if debug == 0 {
   501  		return writeHeapProto(w, p, int64(runtime.MemProfileRate))
   502  	}
   503  
   504  	sort.Slice(p, func(i, j int) bool { return p[i].InUseBytes() > p[j].InUseBytes() })
   505  
   506  	b := bufio.NewWriter(w)
   507  	tw := tabwriter.NewWriter(b, 1, 8, 1, '\t', 0)
   508  	w = tw
   509  
   510  	var total runtime.MemProfileRecord
   511  	for i := range p {
   512  		r := &p[i]
   513  		total.AllocBytes += r.AllocBytes
   514  		total.AllocObjects += r.AllocObjects
   515  		total.FreeBytes += r.FreeBytes
   516  		total.FreeObjects += r.FreeObjects
   517  	}
   518  
   519  	// Technically the rate is MemProfileRate not 2*MemProfileRate,
   520  	// but early versions of the C++ heap profiler reported 2*MemProfileRate,
   521  	// so that's what pprof has come to expect.
   522  	fmt.Fprintf(w, "heap profile: %d: %d [%d: %d] @ heap/%d\n",
   523  		total.InUseObjects(), total.InUseBytes(),
   524  		total.AllocObjects, total.AllocBytes,
   525  		2*runtime.MemProfileRate)
   526  
   527  	for i := range p {
   528  		r := &p[i]
   529  		fmt.Fprintf(w, "%d: %d [%d: %d] @",
   530  			r.InUseObjects(), r.InUseBytes(),
   531  			r.AllocObjects, r.AllocBytes)
   532  		for _, pc := range r.Stack() {
   533  			fmt.Fprintf(w, " %#x", pc)
   534  		}
   535  		fmt.Fprintf(w, "\n")
   536  		printStackRecord(w, r.Stack(), false)
   537  	}
   538  
   539  	// Print memstats information too.
   540  	// Pprof will ignore, but useful for people
   541  	s := new(runtime.MemStats)
   542  	runtime.ReadMemStats(s)
   543  	fmt.Fprintf(w, "\n# runtime.MemStats\n")
   544  	fmt.Fprintf(w, "# Alloc = %d\n", s.Alloc)
   545  	fmt.Fprintf(w, "# TotalAlloc = %d\n", s.TotalAlloc)
   546  	fmt.Fprintf(w, "# Sys = %d\n", s.Sys)
   547  	fmt.Fprintf(w, "# Lookups = %d\n", s.Lookups)
   548  	fmt.Fprintf(w, "# Mallocs = %d\n", s.Mallocs)
   549  	fmt.Fprintf(w, "# Frees = %d\n", s.Frees)
   550  
   551  	fmt.Fprintf(w, "# HeapAlloc = %d\n", s.HeapAlloc)
   552  	fmt.Fprintf(w, "# HeapSys = %d\n", s.HeapSys)
   553  	fmt.Fprintf(w, "# HeapIdle = %d\n", s.HeapIdle)
   554  	fmt.Fprintf(w, "# HeapInuse = %d\n", s.HeapInuse)
   555  	fmt.Fprintf(w, "# HeapReleased = %d\n", s.HeapReleased)
   556  	fmt.Fprintf(w, "# HeapObjects = %d\n", s.HeapObjects)
   557  
   558  	fmt.Fprintf(w, "# Stack = %d / %d\n", s.StackInuse, s.StackSys)
   559  	fmt.Fprintf(w, "# MSpan = %d / %d\n", s.MSpanInuse, s.MSpanSys)
   560  	fmt.Fprintf(w, "# MCache = %d / %d\n", s.MCacheInuse, s.MCacheSys)
   561  	fmt.Fprintf(w, "# BuckHashSys = %d\n", s.BuckHashSys)
   562  	fmt.Fprintf(w, "# GCSys = %d\n", s.GCSys)
   563  	fmt.Fprintf(w, "# OtherSys = %d\n", s.OtherSys)
   564  
   565  	fmt.Fprintf(w, "# NextGC = %d\n", s.NextGC)
   566  	fmt.Fprintf(w, "# LastGC = %d\n", s.LastGC)
   567  	fmt.Fprintf(w, "# PauseNs = %d\n", s.PauseNs)
   568  	fmt.Fprintf(w, "# PauseEnd = %d\n", s.PauseEnd)
   569  	fmt.Fprintf(w, "# NumGC = %d\n", s.NumGC)
   570  	fmt.Fprintf(w, "# NumForcedGC = %d\n", s.NumForcedGC)
   571  	fmt.Fprintf(w, "# GCCPUFraction = %v\n", s.GCCPUFraction)
   572  	fmt.Fprintf(w, "# DebugGC = %v\n", s.DebugGC)
   573  
   574  	tw.Flush()
   575  	return b.Flush()
   576  }
   577  
   578  // countThreadCreate returns the size of the current ThreadCreateProfile.
   579  func countThreadCreate() int {
   580  	n, _ := runtime.ThreadCreateProfile(nil)
   581  	return n
   582  }
   583  
   584  // writeThreadCreate writes the current runtime ThreadCreateProfile to w.
   585  func writeThreadCreate(w io.Writer, debug int) error {
   586  	return writeRuntimeProfile(w, debug, "threadcreate", runtime.ThreadCreateProfile)
   587  }
   588  
   589  // countGoroutine returns the number of goroutines.
   590  func countGoroutine() int {
   591  	return runtime.NumGoroutine()
   592  }
   593  
   594  // writeGoroutine writes the current runtime GoroutineProfile to w.
   595  func writeGoroutine(w io.Writer, debug int) error {
   596  	if debug >= 2 {
   597  		return writeGoroutineStacks(w)
   598  	}
   599  	return writeRuntimeProfile(w, debug, "goroutine", runtime.GoroutineProfile)
   600  }
   601  
   602  func writeGoroutineStacks(w io.Writer) error {
   603  	// We don't know how big the buffer needs to be to collect
   604  	// all the goroutines. Start with 1 MB and try a few times, doubling each time.
   605  	// Give up and use a truncated trace if 64 MB is not enough.
   606  	buf := make([]byte, 1<<20)
   607  	for i := 0; ; i++ {
   608  		n := runtime.Stack(buf, true)
   609  		if n < len(buf) {
   610  			buf = buf[:n]
   611  			break
   612  		}
   613  		if len(buf) >= 64<<20 {
   614  			// Filled 64 MB - stop there.
   615  			break
   616  		}
   617  		buf = make([]byte, 2*len(buf))
   618  	}
   619  	_, err := w.Write(buf)
   620  	return err
   621  }
   622  
   623  func writeRuntimeProfile(w io.Writer, debug int, name string, fetch func([]runtime.StackRecord) (int, bool)) error {
   624  	// Find out how many records there are (fetch(nil)),
   625  	// allocate that many records, and get the data.
   626  	// There's a race—more records might be added between
   627  	// the two calls—so allocate a few extra records for safety
   628  	// and also try again if we're very unlucky.
   629  	// The loop should only execute one iteration in the common case.
   630  	var p []runtime.StackRecord
   631  	n, ok := fetch(nil)
   632  	for {
   633  		// Allocate room for a slightly bigger profile,
   634  		// in case a few more entries have been added
   635  		// since the call to ThreadProfile.
   636  		p = make([]runtime.StackRecord, n+10)
   637  		n, ok = fetch(p)
   638  		if ok {
   639  			p = p[0:n]
   640  			break
   641  		}
   642  		// Profile grew; try again.
   643  	}
   644  
   645  	return printCountProfile(w, debug, name, runtimeProfile(p))
   646  }
   647  
   648  type runtimeProfile []runtime.StackRecord
   649  
   650  func (p runtimeProfile) Len() int              { return len(p) }
   651  func (p runtimeProfile) Stack(i int) []uintptr { return p[i].Stack() }
   652  
   653  var cpu struct {
   654  	sync.Mutex
   655  	profiling bool
   656  	done      chan bool
   657  }
   658  
   659  // StartCPUProfile enables CPU profiling for the current process.
   660  // While profiling, the profile will be buffered and written to w.
   661  // StartCPUProfile returns an error if profiling is already enabled.
   662  //
   663  // On Unix-like systems, StartCPUProfile does not work by default for
   664  // Go code built with -buildmode=c-archive or -buildmode=c-shared.
   665  // StartCPUProfile relies on the SIGPROF signal, but that signal will
   666  // be delivered to the main program's SIGPROF signal handler (if any)
   667  // not to the one used by Go. To make it work, call os/signal.Notify
   668  // for syscall.SIGPROF, but note that doing so may break any profiling
   669  // being done by the main program.
   670  func StartCPUProfile(w io.Writer) error {
   671  	// The runtime routines allow a variable profiling rate,
   672  	// but in practice operating systems cannot trigger signals
   673  	// at more than about 500 Hz, and our processing of the
   674  	// signal is not cheap (mostly getting the stack trace).
   675  	// 100 Hz is a reasonable choice: it is frequent enough to
   676  	// produce useful data, rare enough not to bog down the
   677  	// system, and a nice round number to make it easy to
   678  	// convert sample counts to seconds. Instead of requiring
   679  	// each client to specify the frequency, we hard code it.
   680  	const hz = 100
   681  
   682  	cpu.Lock()
   683  	defer cpu.Unlock()
   684  	if cpu.done == nil {
   685  		cpu.done = make(chan bool)
   686  	}
   687  	// Double-check.
   688  	if cpu.profiling {
   689  		return fmt.Errorf("cpu profiling already in use")
   690  	}
   691  	cpu.profiling = true
   692  	runtime.SetCPUProfileRate(hz)
   693  	go profileWriter(w)
   694  	return nil
   695  }
   696  
   697  // readProfile, provided by the runtime, returns the next chunk of
   698  // binary CPU profiling stack trace data, blocking until data is available.
   699  // If profiling is turned off and all the profile data accumulated while it was
   700  // on has been returned, readProfile returns eof=true.
   701  // The caller must save the returned data and tags before calling readProfile again.
   702  func readProfile() (data []uint64, tags []unsafe.Pointer, eof bool)
   703  
   704  func profileWriter(w io.Writer) {
   705  	b := newProfileBuilder(w)
   706  	var err error
   707  	for {
   708  		time.Sleep(100 * time.Millisecond)
   709  		data, tags, eof := readProfile()
   710  		if e := b.addCPUData(data, tags); e != nil && err == nil {
   711  			err = e
   712  		}
   713  		if eof {
   714  			break
   715  		}
   716  	}
   717  	if err != nil {
   718  		// The runtime should never produce an invalid or truncated profile.
   719  		// It drops records that can't fit into its log buffers.
   720  		panic("runtime/pprof: converting profile: " + err.Error())
   721  	}
   722  	b.build()
   723  	cpu.done <- true
   724  }
   725  
   726  // StopCPUProfile stops the current CPU profile, if any.
   727  // StopCPUProfile only returns after all the writes for the
   728  // profile have completed.
   729  func StopCPUProfile() {
   730  	cpu.Lock()
   731  	defer cpu.Unlock()
   732  
   733  	if !cpu.profiling {
   734  		return
   735  	}
   736  	cpu.profiling = false
   737  	runtime.SetCPUProfileRate(0)
   738  	<-cpu.done
   739  }
   740  
   741  // countBlock returns the number of records in the blocking profile.
   742  func countBlock() int {
   743  	n, _ := runtime.BlockProfile(nil)
   744  	return n
   745  }
   746  
   747  // countMutex returns the number of records in the mutex profile.
   748  func countMutex() int {
   749  	n, _ := runtime.MutexProfile(nil)
   750  	return n
   751  }
   752  
   753  // writeBlock writes the current blocking profile to w.
   754  func writeBlock(w io.Writer, debug int) error {
   755  	var p []runtime.BlockProfileRecord
   756  	n, ok := runtime.BlockProfile(nil)
   757  	for {
   758  		p = make([]runtime.BlockProfileRecord, n+50)
   759  		n, ok = runtime.BlockProfile(p)
   760  		if ok {
   761  			p = p[:n]
   762  			break
   763  		}
   764  	}
   765  
   766  	sort.Slice(p, func(i, j int) bool { return p[i].Cycles > p[j].Cycles })
   767  
   768  	b := bufio.NewWriter(w)
   769  	var tw *tabwriter.Writer
   770  	w = b
   771  	if debug > 0 {
   772  		tw = tabwriter.NewWriter(w, 1, 8, 1, '\t', 0)
   773  		w = tw
   774  	}
   775  
   776  	fmt.Fprintf(w, "--- contention:\n")
   777  	fmt.Fprintf(w, "cycles/second=%v\n", runtime_cyclesPerSecond())
   778  	for i := range p {
   779  		r := &p[i]
   780  		fmt.Fprintf(w, "%v %v @", r.Cycles, r.Count)
   781  		for _, pc := range r.Stack() {
   782  			fmt.Fprintf(w, " %#x", pc)
   783  		}
   784  		fmt.Fprint(w, "\n")
   785  		if debug > 0 {
   786  			printStackRecord(w, r.Stack(), true)
   787  		}
   788  	}
   789  
   790  	if tw != nil {
   791  		tw.Flush()
   792  	}
   793  	return b.Flush()
   794  }
   795  
   796  // writeMutex writes the current mutex profile to w.
   797  func writeMutex(w io.Writer, debug int) error {
   798  	// TODO(pjw): too much common code with writeBlock. FIX!
   799  	var p []runtime.BlockProfileRecord
   800  	n, ok := runtime.MutexProfile(nil)
   801  	for {
   802  		p = make([]runtime.BlockProfileRecord, n+50)
   803  		n, ok = runtime.MutexProfile(p)
   804  		if ok {
   805  			p = p[:n]
   806  			break
   807  		}
   808  	}
   809  
   810  	sort.Slice(p, func(i, j int) bool { return p[i].Cycles > p[j].Cycles })
   811  
   812  	b := bufio.NewWriter(w)
   813  	var tw *tabwriter.Writer
   814  	w = b
   815  	if debug > 0 {
   816  		tw = tabwriter.NewWriter(w, 1, 8, 1, '\t', 0)
   817  		w = tw
   818  	}
   819  
   820  	fmt.Fprintf(w, "--- mutex:\n")
   821  	fmt.Fprintf(w, "cycles/second=%v\n", runtime_cyclesPerSecond())
   822  	fmt.Fprintf(w, "sampling period=%d\n", runtime.SetMutexProfileFraction(-1))
   823  	for i := range p {
   824  		r := &p[i]
   825  		fmt.Fprintf(w, "%v %v @", r.Cycles, r.Count)
   826  		for _, pc := range r.Stack() {
   827  			fmt.Fprintf(w, " %#x", pc)
   828  		}
   829  		fmt.Fprint(w, "\n")
   830  		if debug > 0 {
   831  			printStackRecord(w, r.Stack(), true)
   832  		}
   833  	}
   834  
   835  	if tw != nil {
   836  		tw.Flush()
   837  	}
   838  	return b.Flush()
   839  }
   840  
   841  func runtime_cyclesPerSecond() int64