// Copyright 2014 The Go Authors. All rights reserved.
|
// Use of this source code is governed by a BSD-style
|
// license that can be found in the LICENSE file.
|
|
package main
|
|
import (
|
"flag"
|
"fmt"
|
"go/build"
|
"internal/testenv"
|
"io/ioutil"
|
"os"
|
"os/exec"
|
"path/filepath"
|
"runtime"
|
"strings"
|
"testing"
|
)
|
|
var tmp, exe string // populated by buildObjdump
|
|
func TestMain(m *testing.M) {
|
if !testenv.HasGoBuild() {
|
return
|
}
|
var exitcode int
|
if err := buildObjdump(); err == nil {
|
exitcode = m.Run()
|
} else {
|
fmt.Println(err)
|
exitcode = 1
|
}
|
os.RemoveAll(tmp)
|
os.Exit(exitcode)
|
}
|
|
func buildObjdump() error {
|
var err error
|
tmp, err = ioutil.TempDir("", "TestObjDump")
|
if err != nil {
|
return fmt.Errorf("TempDir failed: %v", err)
|
}
|
|
exe = filepath.Join(tmp, "testobjdump.exe")
|
gotool, err := testenv.GoTool()
|
if err != nil {
|
return err
|
}
|
out, err := exec.Command(gotool, "build", "-o", exe, "cmd/objdump").CombinedOutput()
|
if err != nil {
|
os.RemoveAll(tmp)
|
return fmt.Errorf("go build -o %v cmd/objdump: %v\n%s", exe, err, string(out))
|
}
|
|
return nil
|
}
|
|
var x86Need = []string{
|
"JMP main.main(SB)",
|
"CALL main.Println(SB)",
|
"RET",
|
}
|
|
var armNeed = []string{
|
"B main.main(SB)",
|
"BL main.Println(SB)",
|
"RET",
|
}
|
|
var ppcNeed = []string{
|
"BR main.main(SB)",
|
"CALL main.Println(SB)",
|
"RET",
|
}
|
|
var target = flag.String("target", "", "test disassembly of `goos/goarch` binary")
|
|
// objdump is fully cross platform: it can handle binaries
|
// from any known operating system and architecture.
|
// We could in principle add binaries to testdata and check
|
// all the supported systems during this test. However, the
|
// binaries would be about 1 MB each, and we don't want to
|
// add that much junk to the hg repository. Instead, build a
|
// binary for the current system (only) and test that objdump
|
// can handle that one.
|
|
func testDisasm(t *testing.T, printCode bool, flags ...string) {
|
goarch := runtime.GOARCH
|
if *target != "" {
|
f := strings.Split(*target, "/")
|
if len(f) != 2 {
|
t.Fatalf("-target argument must be goos/goarch")
|
}
|
defer os.Setenv("GOOS", os.Getenv("GOOS"))
|
defer os.Setenv("GOARCH", os.Getenv("GOARCH"))
|
os.Setenv("GOOS", f[0])
|
os.Setenv("GOARCH", f[1])
|
goarch = f[1]
|
}
|
|
hello := filepath.Join(tmp, "hello.exe")
|
args := []string{"build", "-o", hello}
|
args = append(args, flags...)
|
args = append(args, "testdata/fmthello.go")
|
out, err := exec.Command(testenv.GoToolPath(t), args...).CombinedOutput()
|
if err != nil {
|
t.Fatalf("go build fmthello.go: %v\n%s", err, out)
|
}
|
need := []string{
|
"TEXT main.main(SB)",
|
}
|
|
if printCode {
|
need = append(need, ` Println("hello, world")`)
|
} else {
|
need = append(need, "fmthello.go:6")
|
}
|
|
switch goarch {
|
case "amd64", "386":
|
need = append(need, x86Need...)
|
case "arm":
|
need = append(need, armNeed...)
|
case "ppc64", "ppc64le":
|
need = append(need, ppcNeed...)
|
}
|
|
args = []string{
|
"-s", "main.main",
|
hello,
|
}
|
|
if printCode {
|
args = append([]string{"-S"}, args...)
|
}
|
|
out, err = exec.Command(exe, args...).CombinedOutput()
|
if err != nil {
|
t.Fatalf("objdump fmthello.exe: %v\n%s", err, out)
|
}
|
|
text := string(out)
|
ok := true
|
for _, s := range need {
|
if !strings.Contains(text, s) {
|
t.Errorf("disassembly missing '%s'", s)
|
ok = false
|
}
|
}
|
if goarch == "386" {
|
if strings.Contains(text, "(IP)") {
|
t.Errorf("disassembly contains PC-Relative addressing on 386")
|
ok = false
|
}
|
}
|
|
if !ok {
|
t.Logf("full disassembly:\n%s", text)
|
}
|
}
|
|
func TestDisasm(t *testing.T) {
|
switch runtime.GOARCH {
|
case "mips", "mipsle", "mips64", "mips64le":
|
t.Skipf("skipping on %s, issue 12559", runtime.GOARCH)
|
case "s390x":
|
t.Skipf("skipping on %s, issue 15255", runtime.GOARCH)
|
}
|
testDisasm(t, false)
|
}
|
|
func TestDisasmCode(t *testing.T) {
|
switch runtime.GOARCH {
|
case "mips", "mipsle", "mips64", "mips64le":
|
t.Skipf("skipping on %s, issue 12559", runtime.GOARCH)
|
case "s390x":
|
t.Skipf("skipping on %s, issue 15255", runtime.GOARCH)
|
}
|
testDisasm(t, true)
|
}
|
|
func TestDisasmExtld(t *testing.T) {
|
switch runtime.GOOS {
|
case "plan9", "windows":
|
t.Skipf("skipping on %s", runtime.GOOS)
|
}
|
switch runtime.GOARCH {
|
case "ppc64":
|
t.Skipf("skipping on %s, no support for external linking, issue 9038", runtime.GOARCH)
|
case "mips64", "mips64le", "mips", "mipsle":
|
t.Skipf("skipping on %s, issue 12559 and 12560", runtime.GOARCH)
|
case "s390x":
|
t.Skipf("skipping on %s, issue 15255", runtime.GOARCH)
|
}
|
// TODO(jsing): Reenable once openbsd/arm has external linking support.
|
if runtime.GOOS == "openbsd" && runtime.GOARCH == "arm" {
|
t.Skip("skipping on openbsd/arm, no support for external linking, issue 10619")
|
}
|
if !build.Default.CgoEnabled {
|
t.Skip("skipping because cgo is not enabled")
|
}
|
testDisasm(t, false, "-ldflags=-linkmode=external")
|
}
|
|
func TestDisasmGoobj(t *testing.T) {
|
switch runtime.GOARCH {
|
case "mips", "mipsle", "mips64", "mips64le":
|
t.Skipf("skipping on %s, issue 12559", runtime.GOARCH)
|
case "s390x":
|
t.Skipf("skipping on %s, issue 15255", runtime.GOARCH)
|
}
|
|
hello := filepath.Join(tmp, "hello.o")
|
args := []string{"tool", "compile", "-o", hello}
|
args = append(args, "testdata/fmthello.go")
|
out, err := exec.Command(testenv.GoToolPath(t), args...).CombinedOutput()
|
if err != nil {
|
t.Fatalf("go tool compile fmthello.go: %v\n%s", err, out)
|
}
|
need := []string{
|
"main(SB)",
|
"fmthello.go:6",
|
}
|
|
args = []string{
|
"-s", "main",
|
hello,
|
}
|
|
out, err = exec.Command(exe, args...).CombinedOutput()
|
if err != nil {
|
t.Fatalf("objdump fmthello.o: %v\n%s", err, out)
|
}
|
|
text := string(out)
|
ok := true
|
for _, s := range need {
|
if !strings.Contains(text, s) {
|
t.Errorf("disassembly missing '%s'", s)
|
ok = false
|
}
|
}
|
if runtime.GOARCH == "386" {
|
if strings.Contains(text, "(IP)") {
|
t.Errorf("disassembly contains PC-Relative addressing on 386")
|
ok = false
|
}
|
}
|
if !ok {
|
t.Logf("full disassembly:\n%s", text)
|
}
|
}
|
