symbolbuilder.go

     1  // Copyright 2019 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 loader
     6  
     7  import (
     8  	"cmd/internal/goobj"
     9  	"cmd/internal/objabi"
    10  	"cmd/internal/sys"
    11  	"cmd/link/internal/sym"
    12  	"sort"
    13  )
    14  
    15  // SymbolBuilder is a helper designed to help with the construction
    16  // of new symbol contents.
    17  type SymbolBuilder struct {
    18  	*extSymPayload         // points to payload being updated
    19  	symIdx         Sym     // index of symbol being updated/constructed
    20  	l              *Loader // loader
    21  }
    22  
    23  // MakeSymbolBuilder creates a symbol builder for use in constructing
    24  // an entirely new symbol.
    25  func (l *Loader) MakeSymbolBuilder(name string) *SymbolBuilder {
    26  	// for now assume that any new sym is intended to be static
    27  	symIdx := l.CreateStaticSym(name)
    28  	sb := &SymbolBuilder{l: l, symIdx: symIdx}
    29  	sb.extSymPayload = l.getPayload(symIdx)
    30  	return sb
    31  }
    32  
    33  // MakeSymbolUpdater creates a symbol builder helper for an existing
    34  // symbol 'symIdx'. If 'symIdx' is not an external symbol, then create
    35  // a clone of it (copy name, properties, etc) fix things up so that
    36  // the lookup tables and caches point to the new version, not the old
    37  // version.
    38  func (l *Loader) MakeSymbolUpdater(symIdx Sym) *SymbolBuilder {
    39  	if symIdx == 0 {
    40  		panic("can't update the null symbol")
    41  	}
    42  	if !l.IsExternal(symIdx) {
    43  		// Create a clone with the same name/version/kind etc.
    44  		l.cloneToExternal(symIdx)
    45  	}
    46  
    47  	// Construct updater and return.
    48  	sb := &SymbolBuilder{l: l, symIdx: symIdx}
    49  	sb.extSymPayload = l.getPayload(symIdx)
    50  	return sb
    51  }
    52  
    53  // CreateSymForUpdate creates a symbol with given name and version,
    54  // returns a CreateSymForUpdate for update. If the symbol already
    55  // exists, it will update in-place.
    56  func (l *Loader) CreateSymForUpdate(name string, version int) *SymbolBuilder {
    57  	s := l.LookupOrCreateSym(name, version)
    58  	l.SetAttrReachable(s, true)
    59  	return l.MakeSymbolUpdater(s)
    60  }
    61  
    62  // Getters for properties of the symbol we're working on.
    63  
    64  func (sb *SymbolBuilder) Sym() Sym               { return sb.symIdx }
    65  func (sb *SymbolBuilder) Name() string           { return sb.name }
    66  func (sb *SymbolBuilder) Version() int           { return sb.ver }
    67  func (sb *SymbolBuilder) Type() sym.SymKind      { return sb.kind }
    68  func (sb *SymbolBuilder) Size() int64            { return sb.size }
    69  func (sb *SymbolBuilder) Data() []byte           { return sb.data }
    70  func (sb *SymbolBuilder) Value() int64           { return sb.l.SymValue(sb.symIdx) }
    71  func (sb *SymbolBuilder) Align() int32           { return sb.l.SymAlign(sb.symIdx) }
    72  func (sb *SymbolBuilder) Localentry() uint8      { return sb.l.SymLocalentry(sb.symIdx) }
    73  func (sb *SymbolBuilder) OnList() bool           { return sb.l.AttrOnList(sb.symIdx) }
    74  func (sb *SymbolBuilder) External() bool         { return sb.l.AttrExternal(sb.symIdx) }
    75  func (sb *SymbolBuilder) Extname() string        { return sb.l.SymExtname(sb.symIdx) }
    76  func (sb *SymbolBuilder) CgoExportDynamic() bool { return sb.l.AttrCgoExportDynamic(sb.symIdx) }
    77  func (sb *SymbolBuilder) Dynimplib() string      { return sb.l.SymDynimplib(sb.symIdx) }
    78  func (sb *SymbolBuilder) Dynimpvers() string     { return sb.l.SymDynimpvers(sb.symIdx) }
    79  func (sb *SymbolBuilder) SubSym() Sym            { return sb.l.SubSym(sb.symIdx) }
    80  func (sb *SymbolBuilder) GoType() Sym            { return sb.l.SymGoType(sb.symIdx) }
    81  func (sb *SymbolBuilder) VisibilityHidden() bool { return sb.l.AttrVisibilityHidden(sb.symIdx) }
    82  func (sb *SymbolBuilder) Sect() *sym.Section     { return sb.l.SymSect(sb.symIdx) }
    83  
    84  // Setters for symbol properties.
    85  
    86  func (sb *SymbolBuilder) SetType(kind sym.SymKind)   { sb.kind = kind }
    87  func (sb *SymbolBuilder) SetSize(size int64)         { sb.size = size }
    88  func (sb *SymbolBuilder) SetData(data []byte)        { sb.data = data }
    89  func (sb *SymbolBuilder) SetOnList(v bool)           { sb.l.SetAttrOnList(sb.symIdx, v) }
    90  func (sb *SymbolBuilder) SetExternal(v bool)         { sb.l.SetAttrExternal(sb.symIdx, v) }
    91  func (sb *SymbolBuilder) SetValue(v int64)           { sb.l.SetSymValue(sb.symIdx, v) }
    92  func (sb *SymbolBuilder) SetAlign(align int32)       { sb.l.SetSymAlign(sb.symIdx, align) }
    93  func (sb *SymbolBuilder) SetLocalentry(value uint8)  { sb.l.SetSymLocalentry(sb.symIdx, value) }
    94  func (sb *SymbolBuilder) SetExtname(value string)    { sb.l.SetSymExtname(sb.symIdx, value) }
    95  func (sb *SymbolBuilder) SetDynimplib(value string)  { sb.l.SetSymDynimplib(sb.symIdx, value) }
    96  func (sb *SymbolBuilder) SetDynimpvers(value string) { sb.l.SetSymDynimpvers(sb.symIdx, value) }
    97  func (sb *SymbolBuilder) SetPlt(value int32)         { sb.l.SetPlt(sb.symIdx, value) }
    98  func (sb *SymbolBuilder) SetGot(value int32)         { sb.l.SetGot(sb.symIdx, value) }
    99  func (sb *SymbolBuilder) SetSpecial(value bool)      { sb.l.SetAttrSpecial(sb.symIdx, value) }
   100  func (sb *SymbolBuilder) SetLocal(value bool)        { sb.l.SetAttrLocal(sb.symIdx, value) }
   101  func (sb *SymbolBuilder) SetVisibilityHidden(value bool) {
   102  	sb.l.SetAttrVisibilityHidden(sb.symIdx, value)
   103  }
   104  func (sb *SymbolBuilder) SetNotInSymbolTable(value bool) {
   105  	sb.l.SetAttrNotInSymbolTable(sb.symIdx, value)
   106  }
   107  func (sb *SymbolBuilder) SetSect(sect *sym.Section) { sb.l.SetSymSect(sb.symIdx, sect) }
   108  
   109  func (sb *SymbolBuilder) AddBytes(data []byte) {
   110  	if sb.kind == 0 {
   111  		sb.kind = sym.SDATA
   112  	}
   113  	sb.data = append(sb.data, data...)
   114  	sb.size = int64(len(sb.data))
   115  }
   116  
   117  func (sb *SymbolBuilder) Relocs() Relocs {
   118  	return sb.l.Relocs(sb.symIdx)
   119  }
   120  
   121  // ResetRelocs removes all relocations on this symbol.
   122  func (sb *SymbolBuilder) ResetRelocs() {
   123  	sb.relocs = sb.relocs[:0]
   124  }
   125  
   126  // SetRelocType sets the type of the 'i'-th relocation on this sym to 't'
   127  func (sb *SymbolBuilder) SetRelocType(i int, t objabi.RelocType) {
   128  	sb.relocs[i].SetType(uint16(t))
   129  }
   130  
   131  // SetRelocSym sets the target sym of the 'i'-th relocation on this sym to 's'
   132  func (sb *SymbolBuilder) SetRelocSym(i int, tgt Sym) {
   133  	sb.relocs[i].SetSym(goobj.SymRef{PkgIdx: 0, SymIdx: uint32(tgt)})
   134  }
   135  
   136  // SetRelocAdd sets the addend of the 'i'-th relocation on this sym to 'a'
   137  func (sb *SymbolBuilder) SetRelocAdd(i int, a int64) {
   138  	sb.relocs[i].SetAdd(a)
   139  }
   140  
   141  // Add n relocations, return a handle to the relocations.
   142  func (sb *SymbolBuilder) AddRelocs(n int) Relocs {
   143  	sb.relocs = append(sb.relocs, make([]goobj.Reloc, n)...)
   144  	return sb.l.Relocs(sb.symIdx)
   145  }
   146  
   147  // Add a relocation with given type, return its handle and index
   148  // (to set other fields).
   149  func (sb *SymbolBuilder) AddRel(typ objabi.RelocType) (Reloc, int) {
   150  	j := len(sb.relocs)
   151  	sb.relocs = append(sb.relocs, goobj.Reloc{})
   152  	sb.relocs[j].SetType(uint16(typ))
   153  	relocs := sb.Relocs()
   154  	return relocs.At(j), j
   155  }
   156  
   157  // Sort relocations by offset.
   158  func (sb *SymbolBuilder) SortRelocs() {
   159  	sort.Sort((*relocsByOff)(sb.extSymPayload))
   160  }
   161  
   162  // Implement sort.Interface
   163  type relocsByOff extSymPayload
   164  
   165  func (p *relocsByOff) Len() int           { return len(p.relocs) }
   166  func (p *relocsByOff) Less(i, j int) bool { return p.relocs[i].Off() < p.relocs[j].Off() }
   167  func (p *relocsByOff) Swap(i, j int) {
   168  	p.relocs[i], p.relocs[j] = p.relocs[j], p.relocs[i]
   169  }
   170  
   171  func (sb *SymbolBuilder) Reachable() bool {
   172  	return sb.l.AttrReachable(sb.symIdx)
   173  }
   174  
   175  func (sb *SymbolBuilder) SetReachable(v bool) {
   176  	sb.l.SetAttrReachable(sb.symIdx, v)
   177  }
   178  
   179  func (sb *SymbolBuilder) ReadOnly() bool {
   180  	return sb.l.AttrReadOnly(sb.symIdx)
   181  }
   182  
   183  func (sb *SymbolBuilder) SetReadOnly(v bool) {
   184  	sb.l.SetAttrReadOnly(sb.symIdx, v)
   185  }
   186  
   187  func (sb *SymbolBuilder) DuplicateOK() bool {
   188  	return sb.l.AttrDuplicateOK(sb.symIdx)
   189  }
   190  
   191  func (sb *SymbolBuilder) SetDuplicateOK(v bool) {
   192  	sb.l.SetAttrDuplicateOK(sb.symIdx, v)
   193  }
   194  
   195  func (sb *SymbolBuilder) Outer() Sym {
   196  	return sb.l.OuterSym(sb.symIdx)
   197  }
   198  
   199  func (sb *SymbolBuilder) Sub() Sym {
   200  	return sb.l.SubSym(sb.symIdx)
   201  }
   202  
   203  func (sb *SymbolBuilder) SortSub() {
   204  	sb.l.SortSub(sb.symIdx)
   205  }
   206  
   207  func (sb *SymbolBuilder) AddInteriorSym(sub Sym) {
   208  	sb.l.AddInteriorSym(sb.symIdx, sub)
   209  }
   210  
   211  func (sb *SymbolBuilder) AddUint8(v uint8) int64 {
   212  	off := sb.size
   213  	if sb.kind == 0 {
   214  		sb.kind = sym.SDATA
   215  	}
   216  	sb.size++
   217  	sb.data = append(sb.data, v)
   218  	return off
   219  }
   220  
   221  func (sb *SymbolBuilder) AddUintXX(arch *sys.Arch, v uint64, wid int) int64 {
   222  	off := sb.size
   223  	sb.setUintXX(arch, off, v, int64(wid))
   224  	return off
   225  }
   226  
   227  func (sb *SymbolBuilder) setUintXX(arch *sys.Arch, off int64, v uint64, wid int64) int64 {
   228  	if sb.kind == 0 {
   229  		sb.kind = sym.SDATA
   230  	}
   231  	if sb.size < off+wid {
   232  		sb.size = off + wid
   233  		sb.Grow(sb.size)
   234  	}
   235  
   236  	switch wid {
   237  	case 1:
   238  		sb.data[off] = uint8(v)
   239  	case 2:
   240  		arch.ByteOrder.PutUint16(sb.data[off:], uint16(v))
   241  	case 4:
   242  		arch.ByteOrder.PutUint32(sb.data[off:], uint32(v))
   243  	case 8:
   244  		arch.ByteOrder.PutUint64(sb.data[off:], v)
   245  	}
   246  
   247  	return off + wid
   248  }
   249  
   250  func (sb *SymbolBuilder) AddUint16(arch *sys.Arch, v uint16) int64 {
   251  	return sb.AddUintXX(arch, uint64(v), 2)
   252  }
   253  
   254  func (sb *SymbolBuilder) AddUint32(arch *sys.Arch, v uint32) int64 {
   255  	return sb.AddUintXX(arch, uint64(v), 4)
   256  }
   257  
   258  func (sb *SymbolBuilder) AddUint64(arch *sys.Arch, v uint64) int64 {
   259  	return sb.AddUintXX(arch, v, 8)
   260  }
   261  
   262  func (sb *SymbolBuilder) AddUint(arch *sys.Arch, v uint64) int64 {
   263  	return sb.AddUintXX(arch, v, arch.PtrSize)
   264  }
   265  
   266  func (sb *SymbolBuilder) SetUint8(arch *sys.Arch, r int64, v uint8) int64 {
   267  	return sb.setUintXX(arch, r, uint64(v), 1)
   268  }
   269  
   270  func (sb *SymbolBuilder) SetUint16(arch *sys.Arch, r int64, v uint16) int64 {
   271  	return sb.setUintXX(arch, r, uint64(v), 2)
   272  }
   273  
   274  func (sb *SymbolBuilder) SetUint32(arch *sys.Arch, r int64, v uint32) int64 {
   275  	return sb.setUintXX(arch, r, uint64(v), 4)
   276  }
   277  
   278  func (sb *SymbolBuilder) SetUint(arch *sys.Arch, r int64, v uint64) int64 {
   279  	return sb.setUintXX(arch, r, v, int64(arch.PtrSize))
   280  }
   281  
   282  func (sb *SymbolBuilder) SetUintptr(arch *sys.Arch, r int64, v uintptr) int64 {
   283  	return sb.setUintXX(arch, r, uint64(v), int64(arch.PtrSize))
   284  }
   285  
   286  func (sb *SymbolBuilder) SetAddrPlus(arch *sys.Arch, off int64, tgt Sym, add int64) int64 {
   287  	if sb.Type() == 0 {
   288  		sb.SetType(sym.SDATA)
   289  	}
   290  	if off+int64(arch.PtrSize) > sb.size {
   291  		sb.size = off + int64(arch.PtrSize)
   292  		sb.Grow(sb.size)
   293  	}
   294  	r, _ := sb.AddRel(objabi.R_ADDR)
   295  	r.SetSym(tgt)
   296  	r.SetOff(int32(off))
   297  	r.SetSiz(uint8(arch.PtrSize))
   298  	r.SetAdd(add)
   299  	return off + int64(r.Siz())
   300  }
   301  
   302  func (sb *SymbolBuilder) SetAddr(arch *sys.Arch, off int64, tgt Sym) int64 {
   303  	return sb.SetAddrPlus(arch, off, tgt, 0)
   304  }
   305  
   306  func (sb *SymbolBuilder) AddStringAt(off int64, str string) int64 {
   307  	strLen := int64(len(str))
   308  	if off+strLen > int64(len(sb.data)) {
   309  		panic("attempt to write past end of buffer")
   310  	}
   311  	copy(sb.data[off:off+strLen], str)
   312  	return off + strLen
   313  }
   314  
   315  // AddCStringAt adds str plus a null terminating byte.
   316  func (sb *SymbolBuilder) AddCStringAt(off int64, str string) int64 {
   317  	strLen := int64(len(str))
   318  	if off+strLen+1 > int64(len(sb.data)) {
   319  		panic("attempt to write past end of buffer")
   320  	}
   321  	copy(sb.data[off:off+strLen], str)
   322  	sb.data[off+strLen] = 0
   323  	return off + strLen + 1
   324  }
   325  
   326  func (sb *SymbolBuilder) Addstring(str string) int64 {
   327  	if sb.kind == 0 {
   328  		sb.kind = sym.SNOPTRDATA
   329  	}
   330  	r := sb.size
   331  	sb.data = append(sb.data, str...)
   332  	sb.data = append(sb.data, 0)
   333  	sb.size = int64(len(sb.data))
   334  	return r
   335  }
   336  
   337  func (sb *SymbolBuilder) SetBytesAt(off int64, b []byte) int64 {
   338  	datLen := int64(len(b))
   339  	if off+datLen > int64(len(sb.data)) {
   340  		panic("attempt to write past end of buffer")
   341  	}
   342  	copy(sb.data[off:off+datLen], b)
   343  	return off + datLen
   344  }
   345  
   346  func (sb *SymbolBuilder) addSymRef(tgt Sym, add int64, typ objabi.RelocType, rsize int) int64 {
   347  	if sb.kind == 0 {
   348  		sb.kind = sym.SDATA
   349  	}
   350  	i := sb.size
   351  
   352  	sb.size += int64(rsize)
   353  	sb.Grow(sb.size)
   354  
   355  	r, _ := sb.AddRel(typ)
   356  	r.SetSym(tgt)
   357  	r.SetOff(int32(i))
   358  	r.SetSiz(uint8(rsize))
   359  	r.SetAdd(add)
   360  
   361  	return i + int64(rsize)
   362  }
   363  
   364  // Add a symbol reference (relocation) with given type, addend, and size
   365  // (the most generic form).
   366  func (sb *SymbolBuilder) AddSymRef(arch *sys.Arch, tgt Sym, add int64, typ objabi.RelocType, rsize int) int64 {
   367  	return sb.addSymRef(tgt, add, typ, rsize)
   368  }
   369  
   370  func (sb *SymbolBuilder) AddAddrPlus(arch *sys.Arch, tgt Sym, add int64) int64 {
   371  	return sb.addSymRef(tgt, add, objabi.R_ADDR, arch.PtrSize)
   372  }
   373  
   374  func (sb *SymbolBuilder) AddAddrPlus4(arch *sys.Arch, tgt Sym, add int64) int64 {
   375  	return sb.addSymRef(tgt, add, objabi.R_ADDR, 4)
   376  }
   377  
   378  func (sb *SymbolBuilder) AddAddr(arch *sys.Arch, tgt Sym) int64 {
   379  	return sb.AddAddrPlus(arch, tgt, 0)
   380  }
   381  
   382  func (sb *SymbolBuilder) AddPEImageRelativeAddrPlus(arch *sys.Arch, tgt Sym, add int64) int64 {
   383  	return sb.addSymRef(tgt, add, objabi.R_PEIMAGEOFF, 4)
   384  }
   385  
   386  func (sb *SymbolBuilder) AddPCRelPlus(arch *sys.Arch, tgt Sym, add int64) int64 {
   387  	return sb.addSymRef(tgt, add, objabi.R_PCREL, 4)
   388  }
   389  
   390  func (sb *SymbolBuilder) AddCURelativeAddrPlus(arch *sys.Arch, tgt Sym, add int64) int64 {
   391  	return sb.addSymRef(tgt, add, objabi.R_ADDRCUOFF, arch.PtrSize)
   392  }
   393  
   394  func (sb *SymbolBuilder) AddSize(arch *sys.Arch, tgt Sym) int64 {
   395  	return sb.addSymRef(tgt, 0, objabi.R_SIZE, arch.PtrSize)
   396  }
   397  
   398  // GenAddAddrPlusFunc returns a function to be called when capturing
   399  // a function symbol's address. In later stages of the link (when
   400  // address assignment is done) when doing internal linking and
   401  // targeting an executable, we can just emit the address of a function
   402  // directly instead of generating a relocation. Clients can call
   403  // this function (setting 'internalExec' based on build mode and target)
   404  // and then invoke the returned function in roughly the same way that
   405  // loader.*SymbolBuilder.AddAddrPlus would be used.
   406  func GenAddAddrPlusFunc(internalExec bool) func(s *SymbolBuilder, arch *sys.Arch, tgt Sym, add int64) int64 {
   407  	if internalExec {
   408  		return func(s *SymbolBuilder, arch *sys.Arch, tgt Sym, add int64) int64 {
   409  			if v := s.l.SymValue(tgt); v != 0 {
   410  				return s.AddUint(arch, uint64(v+add))
   411  			}
   412  			return s.AddAddrPlus(arch, tgt, add)
   413  		}
   414  	} else {
   415  		return (*SymbolBuilder).AddAddrPlus
   416  	}
   417  }
   418  
   419  func (sb *SymbolBuilder) MakeWritable() {
   420  	if sb.ReadOnly() {
   421  		sb.data = append([]byte(nil), sb.data...)
   422  		sb.l.SetAttrReadOnly(sb.symIdx, false)
   423  	}
   424  }
   425  
   426  func (sb *SymbolBuilder) AddUleb(v uint64) {
   427  	if v < 128 { // common case: 1 byte
   428  		sb.AddUint8(uint8(v))
   429  		return
   430  	}
   431  	for {
   432  		c := uint8(v & 0x7f)
   433  		v >>= 7
   434  		if v != 0 {
   435  			c |= 0x80
   436  		}
   437  		sb.AddUint8(c)
   438  		if c&0x80 == 0 {
   439  			break
   440  		}
   441  	}
   442  }
   443
View as plain text