github.com/golang/gofrontend@v0.0.0-20240429183944-60f985a78526/libgo/misc/cgo/testplugin/testdata/checkdwarf/main.go (about)

     1  // Copyright 2018 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  // Usage:
     6  //
     7  //  checkdwarf <exe> <suffix>
     8  //
     9  // Opens <exe>, which must be an executable or a library and checks that
    10  // there is an entry in .debug_info whose name ends in <suffix>
    11  
    12  package main
    13  
    14  import (
    15  	"debug/dwarf"
    16  	"debug/elf"
    17  	"debug/macho"
    18  	"debug/pe"
    19  	"fmt"
    20  	"os"
    21  	"strings"
    22  )
    23  
    24  func usage() {
    25  	fmt.Fprintf(os.Stderr, "checkdwarf executable-or-library DIE-suffix\n")
    26  }
    27  
    28  type dwarfer interface {
    29  	DWARF() (*dwarf.Data, error)
    30  }
    31  
    32  func openElf(path string) dwarfer {
    33  	exe, err := elf.Open(path)
    34  	if err != nil {
    35  		return nil
    36  	}
    37  	return exe
    38  }
    39  
    40  func openMacho(path string) dwarfer {
    41  	exe, err := macho.Open(path)
    42  	if err != nil {
    43  		return nil
    44  	}
    45  	return exe
    46  }
    47  
    48  func openPE(path string) dwarfer {
    49  	exe, err := pe.Open(path)
    50  	if err != nil {
    51  		return nil
    52  	}
    53  	return exe
    54  }
    55  
    56  func main() {
    57  	if len(os.Args) != 3 {
    58  		usage()
    59  	}
    60  
    61  	exePath := os.Args[1]
    62  	dieSuffix := os.Args[2]
    63  
    64  	var exe dwarfer
    65  
    66  	for _, openfn := range []func(string) dwarfer{openMacho, openPE, openElf} {
    67  		exe = openfn(exePath)
    68  		if exe != nil {
    69  			break
    70  		}
    71  	}
    72  
    73  	if exe == nil {
    74  		fmt.Fprintf(os.Stderr, "could not open %s\n", exePath)
    75  		os.Exit(1)
    76  	}
    77  
    78  	data, err := exe.DWARF()
    79  	if err != nil {
    80  		fmt.Fprintf(os.Stderr, "%s: error opening DWARF: %v\n", exePath, err)
    81  		os.Exit(1)
    82  	}
    83  
    84  	rdr := data.Reader()
    85  	for {
    86  		e, err := rdr.Next()
    87  		if err != nil {
    88  			fmt.Fprintf(os.Stderr, "%s: error reading DWARF: %v\n", exePath, err)
    89  			os.Exit(1)
    90  		}
    91  		if e == nil {
    92  			break
    93  		}
    94  		name, hasname := e.Val(dwarf.AttrName).(string)
    95  		if !hasname {
    96  			continue
    97  		}
    98  		if strings.HasSuffix(name, dieSuffix) {
    99  			// found
   100  			os.Exit(0)
   101  		}
   102  	}
   103  
   104  	fmt.Fprintf(os.Stderr, "%s: no entry with a name ending in %q was found\n", exePath, dieSuffix)
   105  	os.Exit(1)
   106  }