lld/MachO/SymbolTable.h

   1 //===- SymbolTable.h --------------------------------------------*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #ifndef LLD_MACHO_SYMBOL_TABLE_H
  10 #define LLD_MACHO_SYMBOL_TABLE_H
  11
  12 #include "Symbols.h"
  13
  14 #include "lld/Common/LLVM.h"
  15 #include "llvm/ADT/CachedHashString.h"
  16 #include "llvm/ADT/DenseMap.h"
  17 #include "llvm/Object/Archive.h"
  18
  19 namespace lld {
  20 namespace macho {
  21
  22 class ArchiveFile;
  23 class DylibFile;
  24 class InputFile;
  25 class ObjFile;
  26 class InputSection;
  27 class MachHeaderSection;
  28 class Symbol;
  29 class Defined;
  30 class Undefined;
  31
  32 /*
  33  * Note that the SymbolTable handles name collisions by calling
  34  * replaceSymbol(), which does an in-place update of the Symbol via `placement
  35  * new`. Therefore, there is no need to update any relocations that hold
  36  * pointers the "old" Symbol -- they will automatically point to the new one.
  37  */
  38 class SymbolTable {
  39 public:
  40   Defined *addDefined(StringRef name, InputFile *, InputSection *,
  41                       uint64_t value, uint64_t size, bool isWeakDef,
  42                       bool isPrivateExtern, bool isThumb,
  43                       bool isReferencedDynamically, bool noDeadStrip,
  44                       bool isWeakDefCanBeHidden);
  45
  46   Symbol *addUndefined(StringRef name, InputFile *, bool isWeakRef);
  47
  48   Symbol *addCommon(StringRef name, InputFile *, uint64_t size, uint32_t align,
  49                     bool isPrivateExtern);
  50
  51   Symbol *addDylib(StringRef name, DylibFile *file, bool isWeakDef, bool isTlv);
  52   Symbol *addDynamicLookup(StringRef name);
  53
  54   Symbol *addLazyArchive(StringRef name, ArchiveFile *file,
  55                          const llvm::object::Archive::Symbol &sym);
  56   Symbol *addLazyObject(StringRef name, InputFile &file);
  57
  58   Defined *addSynthetic(StringRef name, InputSection *, uint64_t value,
  59                         bool isPrivateExtern, bool includeInSymtab,
  60                         bool referencedDynamically);
  61
  62   ArrayRef<Symbol *> getSymbols() const { return symVector; }
  63   Symbol *find(llvm::CachedHashStringRef name);
  64   Symbol *find(StringRef name) { return find(llvm::CachedHashStringRef(name)); }
  65
  66 private:
  67   std::pair<Symbol *, bool> insert(StringRef name, const InputFile *);
  68   llvm::DenseMap<llvm::CachedHashStringRef, int> symMap;
  69   std::vector<Symbol *> symVector;
  70 };
  71
  72 void reportPendingUndefinedSymbols();
  73
  74 // Call reportPendingUndefinedSymbols() to emit diagnostics.
  75 void treatUndefinedSymbol(const Undefined &, StringRef source);
  76 void treatUndefinedSymbol(const Undefined &, const InputSection *,
  77                           uint64_t offset);
  78
  79 extern std::unique_ptr<SymbolTable> symtab;
  80
  81 } // namespace macho
  82 } // namespace lld
  83
  84 #endif