Source file test/inline_caller.go

     1  // run -gcflags -l=4
     2  
     3  // Copyright 2017 The Go Authors. All rights reserved.
     4  // Use of this source code is governed by a BSD-style
     5  // license that can be found in the LICENSE file.
     6  
     7  package main
     8  
     9  import (
    10  	"fmt"
    11  	"runtime"
    12  )
    13  
    14  type frame struct {
    15  	pc   uintptr
    16  	file string
    17  	line int
    18  	ok   bool
    19  }
    20  
    21  var (
    22  	skip        int
    23  	globalFrame frame
    24  )
    25  
    26  func f() {
    27  	g() // line 27
    28  }
    29  
    30  func g() {
    31  	h() // line 31
    32  }
    33  
    34  func h() {
    35  	x := &globalFrame
    36  	x.pc, x.file, x.line, x.ok = runtime.Caller(skip) // line 36
    37  }
    38  
    39  //go:noinline
    40  func testCaller(skp int) frame {
    41  	skip = skp
    42  	f() // line 42
    43  	frame := globalFrame
    44  	if !frame.ok {
    45  		panic(fmt.Sprintf("skip=%d runtime.Caller failed", skp))
    46  	}
    47  	return frame
    48  }
    49  
    50  type wantFrame struct {
    51  	funcName string
    52  	line     int
    53  }
    54  
    55  // -1 means don't care
    56  var expected = []wantFrame{
    57  	0: {"main.h", 36},
    58  	1: {"main.g", 31},
    59  	2: {"main.f", 27},
    60  	3: {"main.testCaller", 42},
    61  	4: {"main.main", 68},
    62  	5: {"runtime.main", -1},
    63  	6: {"runtime.goexit", -1},
    64  }
    65  
    66  func main() {
    67  	for i := 0; i <= 6; i++ {
    68  		frame := testCaller(i) // line 68
    69  		fn := runtime.FuncForPC(frame.pc)
    70  		if expected[i].line >= 0 && frame.line != expected[i].line {
    71  			panic(fmt.Sprintf("skip=%d expected line %d, got line %d", i, expected[i].line, frame.line))
    72  		}
    73  		if fn.Name() != expected[i].funcName {
    74  			panic(fmt.Sprintf("skip=%d expected function %s, got %s", i, expected[i].funcName, fn.Name()))
    75  		}
    76  	}
    77  }
    78  

View as plain text