bininfo.go 24.3 KB
Newer Older
1 2 3
package proc

import (
4
	"bytes"
5 6 7
	"debug/dwarf"
	"debug/elf"
	"debug/macho"
8
	"debug/pe"
9
	"encoding/binary"
10
	"encoding/hex"
11 12 13 14
	"errors"
	"fmt"
	"io"
	"os"
15 16
	"sort"
	"strings"
17 18 19 20
	"sync"
	"time"

	"github.com/derekparker/delve/pkg/dwarf/frame"
21
	"github.com/derekparker/delve/pkg/dwarf/godwarf"
22
	"github.com/derekparker/delve/pkg/dwarf/line"
23
	"github.com/derekparker/delve/pkg/dwarf/op"
24
	"github.com/derekparker/delve/pkg/dwarf/reader"
25
	"github.com/derekparker/delve/pkg/goversion"
26 27
)

28
// BinaryInfo holds information on the binary being executed.
29 30 31
type BinaryInfo struct {
	lastModified time.Time // Time the executable of this process was last modified

32 33 34
	GOOS           string
	closer         io.Closer
	sepDebugCloser io.Closer
35

36 37
	staticBase uint64

38 39 40
	// Maps package names to package paths, needed to lookup types inside DWARF info
	packageMap map[string]string

41 42 43
	Arch          Arch
	dwarf         *dwarf.Data
	frameEntries  frame.FrameDescriptionEntries
44
	loclist       loclistReader
45
	compileUnits  []*compileUnit
46
	types         map[string]dwarf.Offset
47
	packageVars   []packageVar // packageVars is a list of all global/package variables in debug_info, sorted by address
48
	gStructOffset uint64
49

50
	// Functions is a list of all DW_TAG_subprogram entries in debug_info, sorted by entry point
51 52 53 54 55 56
	Functions []Function
	// Sources is a list of all source files found in debug_line.
	Sources []string
	// LookupFunc maps function names to a description of the function.
	LookupFunc map[string]*Function

57 58
	typeCache map[dwarf.Offset]godwarf.Type

59 60 61
	loadModuleDataOnce sync.Once
	moduleData         []moduleData
	nameOfRuntimeType  map[uintptr]nameOfRuntimeTypeEntry
62

63 64 65 66 67 68
	// runtimeTypeToDIE maps between the offset of a runtime._type in
	// runtime.moduledata.types and the offset of the DIE in debug_info. This
	// map is filled by using the extended attribute godwarf.AttrGoRuntimeType
	// which was added in go 1.11.
	runtimeTypeToDIE map[uint64]runtimeTypeDIE

69 70 71
	// consts[off] lists all the constants with the type defined at offset off.
	consts constantsMap

72 73
	loadErrMu sync.Mutex
	loadErr   error
74 75

	dwarfReader *dwarf.Reader
76 77
}

78 79 80 81 82 83 84 85
// ErrUnsupportedLinuxArch is returned when attempting to debug a binary compiled for an unsupported architecture.
var ErrUnsupportedLinuxArch = errors.New("unsupported architecture - only linux/amd64 is supported")

// ErrUnsupportedWindowsArch is returned when attempting to debug a binary compiled for an unsupported architecture.
var ErrUnsupportedWindowsArch = errors.New("unsupported architecture of windows/386 - only windows/amd64 is supported")

// ErrUnsupportedDarwinArch is returned when attempting to debug a binary compiled for an unsupported architecture.
var ErrUnsupportedDarwinArch = errors.New("unsupported architecture - only darwin/amd64 is supported")
86

87 88
var ErrCouldNotDetermineRelocation = errors.New("could not determine the base address of a PIE")

89 90 91
const dwarfGoLanguage = 22 // DW_LANG_Go (from DWARF v5, section 7.12, page 231)

type compileUnit struct {
92 93 94
	Name   string // univocal name for non-go compile units
	LowPC  uint64
	Ranges [][2]uint64
95 96 97 98 99 100 101

	entry              *dwarf.Entry        // debug_info entry describing this compile unit
	isgo               bool                // true if this is the go compile unit
	lineInfo           *line.DebugLineInfo // debug_line segment associated with this compile unit
	concreteInlinedFns []inlinedFn         // list of concrete inlined functions within this compile unit
	optimized          bool                // this compile unit is optimized
	producer           string              // producer attribute
102 103

	startOffset, endOffset dwarf.Offset // interval of offsets contained in this compile unit
104 105
}

106 107 108 109 110 111 112
type partialUnitConstant struct {
	name  string
	typ   dwarf.Offset
	value int64
}

type partialUnit struct {
113 114 115 116 117
	entry     *dwarf.Entry
	types     map[string]dwarf.Offset
	variables []packageVar
	constants []partialUnitConstant
	functions []Function
118 119
}

120 121 122 123 124 125 126 127 128 129
// inlinedFn represents a concrete inlined function, e.g.
// an entry for the generated code of an inlined function.
type inlinedFn struct {
	Name          string    // Name of the function that was inlined
	LowPC, HighPC uint64    // Address range of the generated inlined instructions
	CallFile      string    // File of the call site of the inlined function
	CallLine      int64     // Line of the call site of the inlined function
	Parent        *Function // The function that contains this inlined function
}

130 131 132 133 134 135 136 137 138 139 140 141
// Function describes a function in the target program.
type Function struct {
	Name       string
	Entry, End uint64 // same as DW_AT_lowpc and DW_AT_highpc
	offset     dwarf.Offset
	cu         *compileUnit
}

// PackageName returns the package part of the symbol name,
// or the empty string if there is none.
// Borrowed from $GOROOT/debug/gosym/symtab.go
func (fn *Function) PackageName() string {
142 143 144 145 146
	return packageName(fn.Name)
}

func packageName(name string) string {
	pathend := strings.LastIndex(name, "/")
147 148 149 150
	if pathend < 0 {
		pathend = 0
	}

151 152
	if i := strings.Index(name[pathend:], "."); i != -1 {
		return name[:pathend+i]
153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181
	}
	return ""
}

// ReceiverName returns the receiver type name of this symbol,
// or the empty string if there is none.
// Borrowed from $GOROOT/debug/gosym/symtab.go
func (fn *Function) ReceiverName() string {
	pathend := strings.LastIndex(fn.Name, "/")
	if pathend < 0 {
		pathend = 0
	}
	l := strings.Index(fn.Name[pathend:], ".")
	r := strings.LastIndex(fn.Name[pathend:], ".")
	if l == -1 || r == -1 || l == r {
		return ""
	}
	return fn.Name[pathend+l+1 : pathend+r]
}

// BaseName returns the symbol name without the package or receiver name.
// Borrowed from $GOROOT/debug/gosym/symtab.go
func (fn *Function) BaseName() string {
	if i := strings.LastIndex(fn.Name, "."); i != -1 {
		return fn.Name[i+1:]
	}
	return fn.Name
}

182 183 184 185 186
// Optimized returns true if the function was optimized by the compiler.
func (fn *Function) Optimized() bool {
	return fn.cu.optimized
}

187 188 189 190 191 192 193 194 195 196 197 198 199 200
type constantsMap map[dwarf.Offset]*constantType

type constantType struct {
	initialized bool
	values      []constantValue
}

type constantValue struct {
	name      string
	fullName  string
	value     int64
	singleBit bool
}

201 202 203 204 205 206 207 208 209
// packageVar represents a package-level variable (or a C global variable).
// If a global variable does not have an address (for example it's stored in
// a register, or non-contiguously) addr will be 0.
type packageVar struct {
	name   string
	offset dwarf.Offset
	addr   uint64
}

210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264
type loclistReader struct {
	data  []byte
	cur   int
	ptrSz int
}

func (rdr *loclistReader) Seek(off int) {
	rdr.cur = off
}

func (rdr *loclistReader) read(sz int) []byte {
	r := rdr.data[rdr.cur : rdr.cur+sz]
	rdr.cur += sz
	return r
}

func (rdr *loclistReader) oneAddr() uint64 {
	switch rdr.ptrSz {
	case 4:
		addr := binary.LittleEndian.Uint32(rdr.read(rdr.ptrSz))
		if addr == ^uint32(0) {
			return ^uint64(0)
		}
		return uint64(addr)
	case 8:
		addr := uint64(binary.LittleEndian.Uint64(rdr.read(rdr.ptrSz)))
		return addr
	default:
		panic("bad address size")
	}
}

func (rdr *loclistReader) Next(e *loclistEntry) bool {
	e.lowpc = rdr.oneAddr()
	e.highpc = rdr.oneAddr()

	if e.lowpc == 0 && e.highpc == 0 {
		return false
	}

	if e.BaseAddressSelection() {
		e.instr = nil
		return true
	}

	instrlen := binary.LittleEndian.Uint16(rdr.read(2))
	e.instr = rdr.read(int(instrlen))
	return true
}

type loclistEntry struct {
	lowpc, highpc uint64
	instr         []byte
}

265 266 267 268 269
type runtimeTypeDIE struct {
	offset dwarf.Offset
	kind   int64
}

270 271 272 273
func (e *loclistEntry) BaseAddressSelection() bool {
	return e.lowpc == ^uint64(0)
}

274
type buildIDHeader struct {
275 276 277 278 279
	Namesz uint32
	Descsz uint32
	Type   uint32
}

280 281 282
// NewBinaryInfo returns an initialized but unloaded BinaryInfo struct.
func NewBinaryInfo(goos, goarch string) *BinaryInfo {
	r := &BinaryInfo{GOOS: goos, nameOfRuntimeType: make(map[uintptr]nameOfRuntimeTypeEntry), typeCache: make(map[dwarf.Offset]godwarf.Type)}
283

284
	// TODO: find better way to determine proc arch (perhaps use executable file info).
285 286
	switch goarch {
	case "amd64":
287
		r.Arch = AMD64Arch(goos)
288 289 290 291 292
	}

	return r
}

293 294 295
// LoadBinaryInfo will load and store the information from the binary at 'path'.
// It is expected this will be called in parallel with other initialization steps
// so a sync.WaitGroup must be provided.
296
func (bi *BinaryInfo) LoadBinaryInfo(path string, entryPoint uint64, wg *sync.WaitGroup) error {
297 298
	fi, err := os.Stat(path)
	if err == nil {
299
		bi.lastModified = fi.ModTime()
300 301
	}

302
	switch bi.GOOS {
303
	case "linux":
304
		return bi.LoadBinaryInfoElf(path, entryPoint, wg)
305
	case "windows":
306
		return bi.LoadBinaryInfoPE(path, entryPoint, wg)
307
	case "darwin":
308
		return bi.LoadBinaryInfoMacho(path, entryPoint, wg)
309 310
	}
	return errors.New("unsupported operating system")
311
	return nil
312 313
}

314 315 316 317 318 319
// GStructOffset returns the offset of the G
// struct in thread local storage.
func (bi *BinaryInfo) GStructOffset() uint64 {
	return bi.gStructOffset
}

320
// LastModified returns the last modified time of the binary.
321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339
func (bi *BinaryInfo) LastModified() time.Time {
	return bi.lastModified
}

// DwarfReader returns a reader for the dwarf data
func (bi *BinaryInfo) DwarfReader() *reader.Reader {
	return reader.New(bi.dwarf)
}

// Types returns list of types present in the debugged program.
func (bi *BinaryInfo) Types() ([]string, error) {
	types := make([]string, 0, len(bi.types))
	for k := range bi.types {
		types = append(types, k)
	}
	return types, nil
}

// PCToLine converts an instruction address to a file/line/function.
340 341 342 343 344
func (bi *BinaryInfo) PCToLine(pc uint64) (string, int, *Function) {
	fn := bi.PCToFunc(pc)
	if fn == nil {
		return "", 0, nil
	}
A
aarzilli 已提交
345
	f, ln := fn.cu.lineInfo.PCToLine(fn.Entry, pc)
346
	return f, ln, fn
347 348
}

349
// LineToPC converts a file:line into a memory address.
350 351 352 353
func (bi *BinaryInfo) LineToPC(filename string, lineno int) (pc uint64, fn *Function, err error) {
	for _, cu := range bi.compileUnits {
		if cu.lineInfo.Lookup[filename] != nil {
			pc = cu.lineInfo.LineToPC(filename, lineno)
354 355 356 357 358 359 360 361 362 363 364
			if pc == 0 {
				// Check to see if this file:line belongs to the call site
				// of an inlined function.
				for _, ifn := range cu.concreteInlinedFns {
					if strings.Contains(ifn.CallFile, filename) && ifn.CallLine == int64(lineno) {
						pc = ifn.LowPC
						fn = ifn.Parent
						return
					}
				}
			}
365
			fn = bi.PCToFunc(pc)
366 367
			if fn != nil {
				return
368 369 370 371 372
			}
		}
	}
	err = fmt.Errorf("could not find %s:%d", filename, lineno)
	return
373 374
}

A
aarzilli 已提交
375 376 377 378 379 380 381 382 383 384 385
// AllPCsForFileLine returns all PC addresses for the given filename:lineno.
func (bi *BinaryInfo) AllPCsForFileLine(filename string, lineno int) []uint64 {
	r := make([]uint64, 0, 1)
	for _, cu := range bi.compileUnits {
		if cu.lineInfo.Lookup[filename] != nil {
			r = append(r, cu.lineInfo.AllPCsForFileLine(filename, lineno)...)
		}
	}
	return r
}

386
// PCToFunc returns the function containing the given PC address
387 388 389 390 391 392 393 394 395 396 397 398
func (bi *BinaryInfo) PCToFunc(pc uint64) *Function {
	i := sort.Search(len(bi.Functions), func(i int) bool {
		fn := bi.Functions[i]
		return pc <= fn.Entry || (fn.Entry <= pc && pc < fn.End)
	})
	if i != len(bi.Functions) {
		fn := &bi.Functions[i]
		if fn.Entry <= pc && pc < fn.End {
			return fn
		}
	}
	return nil
399 400
}

401
// Close closes all internal readers.
402
func (bi *BinaryInfo) Close() error {
403 404 405
	if bi.sepDebugCloser != nil {
		bi.sepDebugCloser.Close()
	}
406 407 408
	return bi.closer.Close()
}

409 410 411 412 413 414
func (bi *BinaryInfo) setLoadError(fmtstr string, args ...interface{}) {
	bi.loadErrMu.Lock()
	bi.loadErr = fmt.Errorf(fmtstr, args...)
	bi.loadErrMu.Unlock()
}

415
// LoadError returns any internal load error.
416 417 418 419
func (bi *BinaryInfo) LoadError() error {
	return bi.loadErr
}

420 421 422 423
type nilCloser struct{}

func (c *nilCloser) Close() error { return nil }

424 425 426
// LoadFromData creates a new BinaryInfo object using the specified data.
// This is used for debugging BinaryInfo, you should use LoadBinary instead.
func (bi *BinaryInfo) LoadFromData(dwdata *dwarf.Data, debugFrameBytes, debugLineBytes, debugLocBytes []byte) {
427
	bi.closer = (*nilCloser)(nil)
428
	bi.sepDebugCloser = (*nilCloser)(nil)
429 430 431
	bi.dwarf = dwdata

	if debugFrameBytes != nil {
432
		bi.frameEntries = frame.Parse(debugFrameBytes, frame.DwarfEndian(debugFrameBytes), bi.staticBase)
433 434
	}

435 436
	bi.loclistInit(debugLocBytes)

437
	bi.loadDebugInfoMaps(debugLineBytes, nil, nil)
438 439 440 441 442 443 444
}

func (bi *BinaryInfo) loclistInit(data []byte) {
	bi.loclist.data = data
	bi.loclist.ptrSz = bi.Arch.PtrSize()
}

445
func (bi *BinaryInfo) locationExpr(entry reader.Entry, attr dwarf.Attr, pc uint64) ([]byte, string, error) {
446 447
	a := entry.Val(attr)
	if a == nil {
448
		return nil, "", fmt.Errorf("no location attribute %s", attr)
449 450
	}
	if instr, ok := a.([]byte); ok {
451 452 453
		var descr bytes.Buffer
		fmt.Fprintf(&descr, "[block] ")
		op.PrettyPrint(&descr, instr)
454
		return instr, descr.String(), nil
455 456 457
	}
	off, ok := a.(int64)
	if !ok {
458
		return nil, "", fmt.Errorf("could not interpret location attribute %s", attr)
459 460
	}
	if bi.loclist.data == nil {
461
		return nil, "", fmt.Errorf("could not find loclist entry at %#x for address %#x (no debug_loc section found)", off, pc)
462 463 464
	}
	instr := bi.loclistEntry(off, pc)
	if instr == nil {
465
		return nil, "", fmt.Errorf("could not find loclist entry at %#x for address %#x", off, pc)
466
	}
467 468 469
	var descr bytes.Buffer
	fmt.Fprintf(&descr, "[%#x:%#x] ", off, pc)
	op.PrettyPrint(&descr, instr)
470 471 472 473 474 475 476 477 478 479 480 481
	return instr, descr.String(), nil
}

// Location returns the location described by attribute attr of entry.
// This will either be an int64 address or a slice of Pieces for locations
// that don't correspond to a single memory address (registers, composite
// locations).
func (bi *BinaryInfo) Location(entry reader.Entry, attr dwarf.Attr, pc uint64, regs op.DwarfRegisters) (int64, []op.Piece, string, error) {
	instr, descr, err := bi.locationExpr(entry, attr, pc)
	if err != nil {
		return 0, nil, "", err
	}
482
	addr, pieces, err := op.ExecuteStackProgram(regs, instr)
483
	return addr, pieces, descr, err
484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511
}

// loclistEntry returns the loclist entry in the loclist starting at off,
// for address pc.
func (bi *BinaryInfo) loclistEntry(off int64, pc uint64) []byte {
	var base uint64
	if cu := bi.findCompileUnit(pc); cu != nil {
		base = cu.LowPC
	}

	bi.loclist.Seek(int(off))
	var e loclistEntry
	for bi.loclist.Next(&e) {
		if e.BaseAddressSelection() {
			base = e.highpc
			continue
		}
		if pc >= e.lowpc+base && pc < e.highpc+base {
			return e.instr
		}
	}

	return nil
}

// findCompileUnit returns the compile unit containing address pc.
func (bi *BinaryInfo) findCompileUnit(pc uint64) *compileUnit {
	for _, cu := range bi.compileUnits {
512 513 514 515
		for _, rng := range cu.Ranges {
			if pc >= rng[0] && pc < rng[1] {
				return cu
			}
516 517 518 519 520 521 522 523 524
		}
	}
	return nil
}

func (bi *BinaryInfo) findCompileUnitForOffset(off dwarf.Offset) *compileUnit {
	for _, cu := range bi.compileUnits {
		if off >= cu.startOffset && off < cu.endOffset {
			return cu
525 526 527
		}
	}
	return nil
528 529
}

530
// Producer returns the value of DW_AT_producer.
531 532 533 534 535 536 537 538 539
func (bi *BinaryInfo) Producer() string {
	for _, cu := range bi.compileUnits {
		if cu.isgo && cu.producer != "" {
			return cu.producer
		}
	}
	return ""
}

540 541 542 543 544
// Type returns the Dwarf type entry at `offset`.
func (bi *BinaryInfo) Type(offset dwarf.Offset) (godwarf.Type, error) {
	return godwarf.ReadType(bi.dwarf, offset, bi.typeCache)
}

545 546
// ELF ///////////////////////////////////////////////////////////////

547
// ErrNoBuildIDNote is used in openSeparateDebugInfo to signal there's no
548 549
// build-id note on the binary, so LoadBinaryInfoElf will return
// the error message coming from elfFile.DWARF() instead.
550
type ErrNoBuildIDNote struct{}
551

552
func (e *ErrNoBuildIDNote) Error() string {
553 554 555 556 557 558 559 560 561 562 563
	return "can't find build-id note on binary"
}

// openSeparateDebugInfo searches for a file containing the separate
// debug info for the binary using the "build ID" method as described
// in GDB's documentation [1], and if found returns two handles, one
// for the bare file, and another for its corresponding elf.File.
// [1] https://sourceware.org/gdb/onlinedocs/gdb/Separate-Debug-Files.html
func (bi *BinaryInfo) openSeparateDebugInfo(exe *elf.File) (*os.File, *elf.File, error) {
	buildid := exe.Section(".note.gnu.build-id")
	if buildid == nil {
564
		return nil, nil, &ErrNoBuildIDNote{}
565 566 567
	}

	br := buildid.Open()
568
	bh := new(buildIDHeader)
569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596
	if err := binary.Read(br, binary.LittleEndian, bh); err != nil {
		return nil, nil, errors.New("can't read build-id header: " + err.Error())
	}

	name := make([]byte, bh.Namesz)
	if err := binary.Read(br, binary.LittleEndian, name); err != nil {
		return nil, nil, errors.New("can't read build-id name: " + err.Error())
	}

	if strings.TrimSpace(string(name)) != "GNU\x00" {
		return nil, nil, errors.New("invalid build-id signature")
	}

	descBinary := make([]byte, bh.Descsz)
	if err := binary.Read(br, binary.LittleEndian, descBinary); err != nil {
		return nil, nil, errors.New("can't read build-id desc: " + err.Error())
	}
	desc := hex.EncodeToString(descBinary)

	debugPath := fmt.Sprintf("/usr/lib/debug/.build-id/%s/%s.debug", desc[:2], desc[2:])
	sepFile, err := os.OpenFile(debugPath, 0, os.ModePerm)
	if err != nil {
		return nil, nil, errors.New("can't open separate debug file: " + err.Error())
	}

	elfFile, err := elf.NewFile(sepFile)
	if err != nil {
		sepFile.Close()
597
		return nil, nil, fmt.Errorf("can't open separate debug file %q: %v", debugPath, err.Error())
598 599 600 601
	}

	if elfFile.Machine != elf.EM_X86_64 {
		sepFile.Close()
602
		return nil, nil, fmt.Errorf("can't open separate debug file %q: %v", debugPath, ErrUnsupportedLinuxArch.Error())
603 604 605 606 607
	}

	return sepFile, elfFile, nil
}

608
// LoadBinaryInfoElf specifically loads information from an ELF binary.
609
func (bi *BinaryInfo) LoadBinaryInfoElf(path string, entryPoint uint64, wg *sync.WaitGroup) error {
610 611 612 613 614 615 616 617 618 619
	exe, err := os.OpenFile(path, 0, os.ModePerm)
	if err != nil {
		return err
	}
	bi.closer = exe
	elfFile, err := elf.NewFile(exe)
	if err != nil {
		return err
	}
	if elfFile.Machine != elf.EM_X86_64 {
620
		return ErrUnsupportedLinuxArch
621
	}
622 623 624 625 626 627 628 629 630

	if entryPoint != 0 {
		bi.staticBase = entryPoint - elfFile.Entry
	} else {
		if elfFile.Type == elf.ET_DYN {
			return ErrCouldNotDetermineRelocation
		}
	}

631
	dwarfFile := elfFile
632

633 634
	bi.dwarf, err = elfFile.DWARF()
	if err != nil {
635 636 637 638
		var sepFile *os.File
		var serr error
		sepFile, dwarfFile, serr = bi.openSeparateDebugInfo(elfFile)
		if serr != nil {
639
			if _, ok := serr.(*ErrNoBuildIDNote); ok {
640 641 642 643 644 645 646 647 648
				return err
			}
			return serr
		}
		bi.sepDebugCloser = sepFile
		bi.dwarf, err = dwarfFile.DWARF()
		if err != nil {
			return err
		}
649 650
	}

651 652
	bi.dwarfReader = bi.dwarf.Reader()

653
	debugLineBytes, err := godwarf.GetDebugSectionElf(dwarfFile, "line")
654 655 656
	if err != nil {
		return err
	}
657 658
	debugLocBytes, _ := godwarf.GetDebugSectionElf(dwarfFile, "loc")
	bi.loclistInit(debugLocBytes)
659 660

	wg.Add(3)
661
	go bi.parseDebugFrameElf(dwarfFile, wg)
662
	go bi.loadDebugInfoMaps(debugLineBytes, wg, nil)
663
	go bi.setGStructOffsetElf(dwarfFile, wg)
664 665 666 667 668 669
	return nil
}

func (bi *BinaryInfo) parseDebugFrameElf(exe *elf.File, wg *sync.WaitGroup) {
	defer wg.Done()

670 671 672
	debugFrameData, err := godwarf.GetDebugSectionElf(exe, "frame")
	if err != nil {
		bi.setLoadError("could not get .debug_frame section: %v", err)
673
		return
674
	}
675 676 677 678
	debugInfoData, err := godwarf.GetDebugSectionElf(exe, "info")
	if err != nil {
		bi.setLoadError("could not get .debug_info section: %v", err)
		return
679 680
	}

681
	bi.frameEntries = frame.Parse(debugFrameData, frame.DwarfEndian(debugInfoData), bi.staticBase)
682 683
}

684 685 686 687 688 689 690 691 692 693 694
func (bi *BinaryInfo) setGStructOffsetElf(exe *elf.File, wg *sync.WaitGroup) {
	defer wg.Done()

	// This is a bit arcane. Essentially:
	// - If the program is pure Go, it can do whatever it wants, and puts the G
	//   pointer at %fs-8.
	// - Otherwise, Go asks the external linker to place the G pointer by
	//   emitting runtime.tlsg, a TLS symbol, which is relocated to the chosen
	//   offset in libc's TLS block.
	symbols, err := exe.Symbols()
	if err != nil {
695 696
		bi.setLoadError("could not parse ELF symbols: %v", err)
		return
697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721
	}
	var tlsg *elf.Symbol
	for _, symbol := range symbols {
		if symbol.Name == "runtime.tlsg" {
			s := symbol
			tlsg = &s
			break
		}
	}
	if tlsg == nil {
		bi.gStructOffset = ^uint64(8) + 1 // -8
		return
	}
	var tls *elf.Prog
	for _, prog := range exe.Progs {
		if prog.Type == elf.PT_TLS {
			tls = prog
			break
		}
	}
	// The TLS register points to the end of the TLS block, which is
	// tls.Memsz long. runtime.tlsg is an offset from the beginning of that block.
	bi.gStructOffset = ^(tls.Memsz) + 1 + tlsg.Value // -tls.Memsz + tlsg.Value
}

722 723
// PE ////////////////////////////////////////////////////////////////

724 725
const _IMAGE_DLLCHARACTERISTICS_DYNAMIC_BASE = 0x0040

726
// LoadBinaryInfoPE specifically loads information from a PE binary.
727
func (bi *BinaryInfo) LoadBinaryInfoPE(path string, entryPoint uint64, wg *sync.WaitGroup) error {
728 729 730 731 732 733
	peFile, closer, err := openExecutablePathPE(path)
	if err != nil {
		return err
	}
	bi.closer = closer
	if peFile.Machine != pe.IMAGE_FILE_MACHINE_AMD64 {
734
		return ErrUnsupportedWindowsArch
735
	}
736
	bi.dwarf, err = peFile.DWARF()
737 738 739 740
	if err != nil {
		return err
	}

741 742 743 744 745 746 747 748 749 750
	//TODO(aarzilli): actually test this when Go supports PIE buildmode on Windows.
	opth := peFile.OptionalHeader.(*pe.OptionalHeader64)
	if entryPoint != 0 {
		bi.staticBase = entryPoint - opth.ImageBase
	} else {
		if opth.DllCharacteristics&_IMAGE_DLLCHARACTERISTICS_DYNAMIC_BASE != 0 {
			return ErrCouldNotDetermineRelocation
		}
	}

751 752
	bi.dwarfReader = bi.dwarf.Reader()

753
	debugLineBytes, err := godwarf.GetDebugSectionPE(peFile, "line")
754 755 756
	if err != nil {
		return err
	}
757 758
	debugLocBytes, _ := godwarf.GetDebugSectionPE(peFile, "loc")
	bi.loclistInit(debugLocBytes)
759 760

	wg.Add(2)
761
	go bi.parseDebugFramePE(peFile, wg)
762
	go bi.loadDebugInfoMaps(debugLineBytes, wg, nil)
763 764 765 766 767 768

	// Use ArbitraryUserPointer (0x28) as pointer to pointer
	// to G struct per:
	// https://golang.org/src/runtime/cgo/gcc_windows_amd64.c

	bi.gStructOffset = 0x28
769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787
	return nil
}

func openExecutablePathPE(path string) (*pe.File, io.Closer, error) {
	f, err := os.OpenFile(path, 0, os.ModePerm)
	if err != nil {
		return nil, nil, err
	}
	peFile, err := pe.NewFile(f)
	if err != nil {
		f.Close()
		return nil, nil, err
	}
	return peFile, f, nil
}

func (bi *BinaryInfo) parseDebugFramePE(exe *pe.File, wg *sync.WaitGroup) {
	defer wg.Done()

788 789 790
	debugFrameBytes, err := godwarf.GetDebugSectionPE(exe, "frame")
	if err != nil {
		bi.setLoadError("could not get .debug_frame section: %v", err)
791
		return
792
	}
793 794 795 796 797 798
	debugInfoBytes, err := godwarf.GetDebugSectionPE(exe, "info")
	if err != nil {
		bi.setLoadError("could not get .debug_info section: %v", err)
		return
	}

799
	bi.frameEntries = frame.Parse(debugFrameBytes, frame.DwarfEndian(debugInfoBytes), bi.staticBase)
800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820
}

// Borrowed from https://golang.org/src/cmd/internal/objfile/pe.go
func findPESymbol(f *pe.File, name string) (*pe.Symbol, error) {
	for _, s := range f.Symbols {
		if s.Name != name {
			continue
		}
		if s.SectionNumber <= 0 {
			return nil, fmt.Errorf("symbol %s: invalid section number %d", name, s.SectionNumber)
		}
		if len(f.Sections) < int(s.SectionNumber) {
			return nil, fmt.Errorf("symbol %s: section number %d is larger than max %d", name, s.SectionNumber, len(f.Sections))
		}
		return s, nil
	}
	return nil, fmt.Errorf("no %s symbol found", name)
}

// MACH-O ////////////////////////////////////////////////////////////

821
// LoadBinaryInfoMacho specifically loads information from a Mach-O binary.
822
func (bi *BinaryInfo) LoadBinaryInfoMacho(path string, entryPoint uint64, wg *sync.WaitGroup) error {
823 824 825 826 827 828
	exe, err := macho.Open(path)
	if err != nil {
		return err
	}
	bi.closer = exe
	if exe.Cpu != macho.CpuAmd64 {
829
		return ErrUnsupportedDarwinArch
830 831 832 833 834 835
	}
	bi.dwarf, err = exe.DWARF()
	if err != nil {
		return err
	}

836 837
	bi.dwarfReader = bi.dwarf.Reader()

838
	debugLineBytes, err := godwarf.GetDebugSectionMacho(exe, "line")
839 840 841
	if err != nil {
		return err
	}
842 843
	debugLocBytes, _ := godwarf.GetDebugSectionMacho(exe, "loc")
	bi.loclistInit(debugLocBytes)
844 845

	wg.Add(2)
846
	go bi.parseDebugFrameMacho(exe, wg)
847
	go bi.loadDebugInfoMaps(debugLineBytes, wg, bi.setGStructOffsetMacho)
848 849 850
	return nil
}

851 852 853 854 855 856 857 858 859 860 861 862
func (bi *BinaryInfo) setGStructOffsetMacho() {
	// In go1.11 it's 0x30, before 0x8a0, see:
	// https://github.com/golang/go/issues/23617
	// and go commit b3a854c733257c5249c3435ffcee194f8439676a
	producer := bi.Producer()
	if producer != "" && goversion.ProducerAfterOrEqual(producer, 1, 11) {
		bi.gStructOffset = 0x30
		return
	}
	bi.gStructOffset = 0x8a0
}

863 864 865
func (bi *BinaryInfo) parseDebugFrameMacho(exe *macho.File, wg *sync.WaitGroup) {
	defer wg.Done()

866 867 868
	debugFrameBytes, err := godwarf.GetDebugSectionMacho(exe, "frame")
	if err != nil {
		bi.setLoadError("could not get __debug_frame section: %v", err)
869
		return
870
	}
871 872 873 874
	debugInfoBytes, err := godwarf.GetDebugSectionMacho(exe, "info")
	if err != nil {
		bi.setLoadError("could not get .debug_info section: %v", err)
		return
875
	}
876

877
	bi.frameEntries = frame.Parse(debugFrameBytes, frame.DwarfEndian(debugInfoBytes), bi.staticBase)
878
}