src/cmd/link/internal/ld/data_test.go

   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
   5 package ld
   6
   7 import (
   8         "cmd/internal/objabi"
   9         "cmd/internal/sys"
  10         "cmd/link/internal/loader"
  11         "testing"
  12 )
  13
  14 func setUpContext(arch *sys.Arch, iself bool, ht objabi.HeadType, bm, lm string) *Link {
  15         ctxt := linknew(arch)
  16         edummy := func(str string, off int) {}
  17         ctxt.HeadType = ht
  18         er := loader.ErrorReporter{}
  19         ctxt.loader = loader.NewLoader(0, edummy, &er)
  20         ctxt.BuildMode.Set(bm)
  21         ctxt.LinkMode.Set(lm)
  22         ctxt.IsELF = iself
  23         ctxt.mustSetHeadType()
  24         ctxt.setArchSyms()
  25         return ctxt
  26 }
  27
  28 // Make sure the addgotsym properly increases the symbols.
  29 func TestAddGotSym(t *testing.T) {
  30         tests := []struct {
  31                 arch    *sys.Arch
  32                 ht      objabi.HeadType
  33                 bm, lm  string
  34                 rel     string
  35                 relsize int
  36                 gotsize int
  37         }{
  38                 {
  39                         arch:    sys.Arch386,
  40                         ht:      objabi.Hlinux,
  41                         bm:      "pie",
  42                         lm:      "internal",
  43                         rel:     ".rel",
  44                         relsize: 2 * sys.Arch386.PtrSize,
  45                         gotsize: sys.Arch386.PtrSize,
  46                 },
  47                 {
  48                         arch:    sys.ArchAMD64,
  49                         ht:      objabi.Hlinux,
  50                         bm:      "pie",
  51                         lm:      "internal",
  52                         rel:     ".rela",
  53                         relsize: 3 * sys.ArchAMD64.PtrSize,
  54                         gotsize: sys.ArchAMD64.PtrSize,
  55                 },
  56                 {
  57                         arch:    sys.ArchAMD64,
  58                         ht:      objabi.Hdarwin,
  59                         bm:      "pie",
  60                         lm:      "external",
  61                         gotsize: sys.ArchAMD64.PtrSize,
  62                 },
  63         }
  64
  65         // Save the architecture as we're going to set it on each test run.
  66         origArch := objabi.GOARCH
  67         defer func() {
  68                 objabi.GOARCH = origArch
  69         }()
  70
  71         for i, test := range tests {
  72                 iself := len(test.rel) != 0
  73                 objabi.GOARCH = test.arch.Name
  74                 ctxt := setUpContext(test.arch, iself, test.ht, test.bm, test.lm)
  75                 foo := ctxt.loader.CreateSymForUpdate("foo", 0)
  76                 ctxt.loader.CreateExtSym("bar", 0)
  77                 AddGotSym(&ctxt.Target, ctxt.loader, &ctxt.ArchSyms, foo.Sym(), 0)
  78
  79                 if iself {
  80                         rel := ctxt.loader.Lookup(test.rel, 0)
  81                         if rel == 0 {
  82                                 t.Fatalf("[%d] could not find symbol: %q", i, test.rel)
  83                         }
  84                         if s := ctxt.loader.SymSize(rel); s != int64(test.relsize) {
  85                                 t.Fatalf("[%d] expected ldr.Size(%q) == %v, got %v", i, test.rel, test.relsize, s)
  86                         }
  87                 }
  88                 if s := ctxt.loader.SymSize(ctxt.loader.Lookup(".got", 0)); s != int64(test.gotsize) {
  89                         t.Fatalf(`[%d] expected ldr.Size(".got") == %v, got %v`, i, test.gotsize, s)
  90                 }
  91         }
  92 }