1// Copyright 2020 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 obj
6
7import (
8	"bytes"
9	"internal/testenv"
10	"os"
11	"path/filepath"
12	"testing"
13	"unsafe"
14
15	"cmd/internal/goobj"
16	"cmd/internal/sys"
17)
18
19var dummyArch = LinkArch{Arch: sys.ArchAMD64}
20
21func TestContentHash64(t *testing.T) {
22	s1 := &LSym{P: []byte("A")}
23	s2 := &LSym{P: []byte("A\x00\x00\x00")}
24	s1.Set(AttrContentAddressable, true)
25	s2.Set(AttrContentAddressable, true)
26	h1 := contentHash64(s1)
27	h2 := contentHash64(s2)
28	if h1 != h2 {
29		t.Errorf("contentHash64(s1)=%x, contentHash64(s2)=%x, expect equal", h1, h2)
30	}
31
32	ctxt := Linknew(&dummyArch) // little endian
33	s3 := ctxt.Int64Sym(int64('A'))
34	h3 := contentHash64(s3)
35	if h1 != h3 {
36		t.Errorf("contentHash64(s1)=%x, contentHash64(s3)=%x, expect equal", h1, h3)
37	}
38}
39
40func TestContentHash(t *testing.T) {
41	syms := []*LSym{
42		&LSym{P: []byte("TestSymbol")},  // 0
43		&LSym{P: []byte("TestSymbol")},  // 1
44		&LSym{P: []byte("TestSymbol2")}, // 2
45		&LSym{P: []byte("")},            // 3
46		&LSym{P: []byte("")},            // 4
47		&LSym{P: []byte("")},            // 5
48		&LSym{P: []byte("")},            // 6
49	}
50	for _, s := range syms {
51		s.Set(AttrContentAddressable, true)
52		s.PkgIdx = goobj.PkgIdxHashed
53	}
54	// s3 references s0
55	r := Addrel(syms[3])
56	r.Sym = syms[0]
57	// s4 references s0
58	r = Addrel(syms[4])
59	r.Sym = syms[0]
60	// s5 references s1
61	r = Addrel(syms[5])
62	r.Sym = syms[1]
63	// s6 references s2
64	r = Addrel(syms[6])
65	r.Sym = syms[2]
66
67	// compute hashes
68	h := make([]goobj.HashType, len(syms))
69	w := &writer{}
70	for i := range h {
71		h[i] = w.contentHash(syms[i])
72	}
73
74	tests := []struct {
75		a, b  int
76		equal bool
77	}{
78		{0, 1, true},  // same contents, no relocs
79		{0, 2, false}, // different contents
80		{3, 4, true},  // same contents, same relocs
81		{3, 5, true},  // recursively same contents
82		{3, 6, false}, // same contents, different relocs
83	}
84	for _, test := range tests {
85		if (h[test.a] == h[test.b]) != test.equal {
86			eq := "equal"
87			if !test.equal {
88				eq = "not equal"
89			}
90			t.Errorf("h%d=%x, h%d=%x, expect %s", test.a, h[test.a], test.b, h[test.b], eq)
91		}
92	}
93}
94
95func TestSymbolTooLarge(t *testing.T) { // Issue 42054
96	testenv.MustHaveGoBuild(t)
97	if unsafe.Sizeof(uintptr(0)) < 8 {
98		t.Skip("skip on 32-bit architectures")
99	}
100
101	tmpdir, err := os.MkdirTemp("", "TestSymbolTooLarge")
102	if err != nil {
103		t.Fatal(err)
104	}
105	defer os.RemoveAll(tmpdir)
106
107	src := filepath.Join(tmpdir, "p.go")
108	err = os.WriteFile(src, []byte("package p; var x [1<<32]byte"), 0666)
109	if err != nil {
110		t.Fatalf("failed to write source file: %v\n", err)
111	}
112	obj := filepath.Join(tmpdir, "p.o")
113	cmd := testenv.Command(t, testenv.GoToolPath(t), "tool", "compile", "-p=p", "-o", obj, src)
114	out, err := cmd.CombinedOutput()
115	if err == nil {
116		t.Fatalf("did not fail\noutput: %s", out)
117	}
118	const want = "symbol too large"
119	if !bytes.Contains(out, []byte(want)) {
120		t.Errorf("unexpected error message: want: %q, got: %s", want, out)
121	}
122}
123
124func TestNoRefName(t *testing.T) {
125	// Test that the norefname flag works.
126	testenv.MustHaveGoBuild(t)
127
128	tmpdir := t.TempDir()
129
130	src := filepath.Join(tmpdir, "x.go")
131	err := os.WriteFile(src, []byte("package main; import \"fmt\"; func main() { fmt.Println(123) }\n"), 0666)
132	if err != nil {
133		t.Fatalf("failed to write source file: %v\n", err)
134	}
135	exe := filepath.Join(tmpdir, "x.exe")
136
137	// Build the fmt package with norefname. Not rebuilding all packages to save time.
138	// Also testing that norefname and non-norefname packages can link together.
139	cmd := testenv.Command(t, testenv.GoToolPath(t), "build", "-gcflags=fmt=-d=norefname", "-o", exe, src)
140	out, err := cmd.CombinedOutput()
141	if err != nil {
142		t.Fatalf("build failed: %v, output:\n%s", err, out)
143	}
144}
145