blob: 0cf8c1ddffa17503b67c08b62f58ac41d7c3f03b [file] [log] [blame]
Scott Baker2c1c4822019-10-16 11:02:41 -07001// Copyright (c) 2016 Uber Technologies, Inc.
2//
3// Permission is hereby granted, free of charge, to any person obtaining a copy
4// of this software and associated documentation files (the "Software"), to deal
5// in the Software without restriction, including without limitation the rights
6// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7// copies of the Software, and to permit persons to whom the Software is
8// furnished to do so, subject to the following conditions:
9//
10// The above copyright notice and this permission notice shall be included in
11// all copies or substantial portions of the Software.
12//
13// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
18// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
19// THE SOFTWARE.
20
21package zap
22
23import (
24 "runtime"
Scott Baker2c1c4822019-10-16 11:02:41 -070025 "sync"
26
27 "go.uber.org/zap/internal/bufferpool"
28)
29
Scott Baker2c1c4822019-10-16 11:02:41 -070030var (
31 _stacktracePool = sync.Pool{
32 New: func() interface{} {
33 return newProgramCounters(64)
34 },
35 }
Scott Baker2c1c4822019-10-16 11:02:41 -070036)
37
khenaidoo26721882021-08-11 17:42:52 -040038func takeStacktrace(skip int) string {
Scott Baker2c1c4822019-10-16 11:02:41 -070039 buffer := bufferpool.Get()
40 defer buffer.Free()
41 programCounters := _stacktracePool.Get().(*programCounters)
42 defer _stacktracePool.Put(programCounters)
43
44 var numFrames int
45 for {
khenaidoo26721882021-08-11 17:42:52 -040046 // Skip the call to runtime.Callers and takeStacktrace so that the
Scott Baker2c1c4822019-10-16 11:02:41 -070047 // program counters start at the caller of takeStacktrace.
khenaidoo26721882021-08-11 17:42:52 -040048 numFrames = runtime.Callers(skip+2, programCounters.pcs)
Scott Baker2c1c4822019-10-16 11:02:41 -070049 if numFrames < len(programCounters.pcs) {
50 break
51 }
52 // Don't put the too-short counter slice back into the pool; this lets
53 // the pool adjust if we consistently take deep stacktraces.
54 programCounters = newProgramCounters(len(programCounters.pcs) * 2)
55 }
56
57 i := 0
Scott Baker2c1c4822019-10-16 11:02:41 -070058 frames := runtime.CallersFrames(programCounters.pcs[:numFrames])
59
60 // Note: On the last iteration, frames.Next() returns false, with a valid
61 // frame, but we ignore this frame. The last frame is a a runtime frame which
62 // adds noise, since it's only either runtime.main or runtime.goexit.
63 for frame, more := frames.Next(); more; frame, more = frames.Next() {
Scott Baker2c1c4822019-10-16 11:02:41 -070064 if i != 0 {
65 buffer.AppendByte('\n')
66 }
67 i++
68 buffer.AppendString(frame.Function)
69 buffer.AppendByte('\n')
70 buffer.AppendByte('\t')
71 buffer.AppendString(frame.File)
72 buffer.AppendByte(':')
73 buffer.AppendInt(int64(frame.Line))
74 }
75
76 return buffer.String()
77}
78
Scott Baker2c1c4822019-10-16 11:02:41 -070079type programCounters struct {
80 pcs []uintptr
81}
82
83func newProgramCounters(size int) *programCounters {
84 return &programCounters{make([]uintptr, size)}
85}