1// Copyright 2021 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
5package abi_test
6
7import (
8	"internal/abi"
9	"internal/testenv"
10	"path/filepath"
11	"strings"
12	"testing"
13)
14
15func TestFuncPC(t *testing.T) {
16	// Test that FuncPC* can get correct function PC.
17	pcFromAsm := abi.FuncPCTestFnAddr
18
19	// Test FuncPC for locally defined function
20	pcFromGo := abi.FuncPCTest()
21	if pcFromGo != pcFromAsm {
22		t.Errorf("FuncPC returns wrong PC, want %x, got %x", pcFromAsm, pcFromGo)
23	}
24
25	// Test FuncPC for imported function
26	pcFromGo = abi.FuncPCABI0(abi.FuncPCTestFn)
27	if pcFromGo != pcFromAsm {
28		t.Errorf("FuncPC returns wrong PC, want %x, got %x", pcFromAsm, pcFromGo)
29	}
30}
31
32func TestFuncPCCompileError(t *testing.T) {
33	// Test that FuncPC* on a function of a mismatched ABI is rejected.
34	testenv.MustHaveGoBuild(t)
35
36	// We want to test internal package, which we cannot normally import.
37	// Run the assembler and compiler manually.
38	tmpdir := t.TempDir()
39	asmSrc := filepath.Join("testdata", "x.s")
40	goSrc := filepath.Join("testdata", "x.go")
41	symabi := filepath.Join(tmpdir, "symabi")
42	obj := filepath.Join(tmpdir, "x.o")
43
44	// Write an importcfg file for the dependencies of the package.
45	importcfgfile := filepath.Join(tmpdir, "hello.importcfg")
46	testenv.WriteImportcfg(t, importcfgfile, nil, "internal/abi")
47
48	// parse assembly code for symabi.
49	cmd := testenv.Command(t, testenv.GoToolPath(t), "tool", "asm", "-p=p", "-gensymabis", "-o", symabi, asmSrc)
50	out, err := cmd.CombinedOutput()
51	if err != nil {
52		t.Fatalf("go tool asm -gensymabis failed: %v\n%s", err, out)
53	}
54
55	// compile go code.
56	cmd = testenv.Command(t, testenv.GoToolPath(t), "tool", "compile", "-importcfg="+importcfgfile, "-p=p", "-symabis", symabi, "-o", obj, goSrc)
57	out, err = cmd.CombinedOutput()
58	if err == nil {
59		t.Fatalf("go tool compile did not fail")
60	}
61
62	// Expect errors in line 17, 18, 20, no errors on other lines.
63	want := []string{"x.go:17", "x.go:18", "x.go:20"}
64	got := strings.Split(string(out), "\n")
65	if got[len(got)-1] == "" {
66		got = got[:len(got)-1] // remove last empty line
67	}
68	for i, s := range got {
69		if !strings.Contains(s, want[i]) {
70			t.Errorf("did not error on line %s", want[i])
71		}
72	}
73	if len(got) != len(want) {
74		t.Errorf("unexpected number of errors, want %d, got %d", len(want), len(got))
75	}
76	if t.Failed() {
77		t.Logf("output:\n%s", string(out))
78	}
79}
80