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 19namespace lld::macho { 20 21class ArchiveFile; 22class DylibFile; 23class InputFile; 24class ObjFile; 25class InputSection; 26class MachHeaderSection; 27class Symbol; 28class Defined; 29class Undefined; 30 31/* 32 * Note that the SymbolTable handles name collisions by calling 33 * replaceSymbol(), which does an in-place update of the Symbol via `placement 34 * new`. Therefore, there is no need to update any relocations that hold 35 * pointers the "old" Symbol -- they will automatically point to the new one. 36 */ 37class SymbolTable { 38public: 39 Defined *addDefined(StringRef name, InputFile *, InputSection *, 40 uint64_t value, uint64_t size, bool isWeakDef, 41 bool isPrivateExtern, bool isReferencedDynamically, 42 bool noDeadStrip, bool isWeakDefCanBeHidden); 43 44 Defined *aliasDefined(Defined *src, StringRef target, InputFile *newFile, 45 bool makePrivateExtern = false); 46 47 Symbol *addUndefined(StringRef name, InputFile *, bool isWeakRef); 48 49 Symbol *addCommon(StringRef name, InputFile *, uint64_t size, uint32_t align, 50 bool isPrivateExtern); 51 52 Symbol *addDylib(StringRef name, DylibFile *file, bool isWeakDef, bool isTlv); 53 Symbol *addDynamicLookup(StringRef name); 54 55 Symbol *addLazyArchive(StringRef name, ArchiveFile *file, 56 const llvm::object::Archive::Symbol &sym); 57 Symbol *addLazyObject(StringRef name, InputFile &file); 58 59 Defined *addSynthetic(StringRef name, InputSection *, uint64_t value, 60 bool isPrivateExtern, bool includeInSymtab, 61 bool referencedDynamically); 62 63 ArrayRef<Symbol *> getSymbols() const { return symVector; } 64 Symbol *find(llvm::CachedHashStringRef name); 65 Symbol *find(StringRef name) { return find(llvm::CachedHashStringRef(name)); } 66 67private: 68 std::pair<Symbol *, bool> insert(StringRef name, const InputFile *); 69 llvm::DenseMap<llvm::CachedHashStringRef, int> symMap; 70 std::vector<Symbol *> symVector; 71}; 72 73void reportPendingUndefinedSymbols(); 74void reportPendingDuplicateSymbols(); 75 76// Call reportPendingUndefinedSymbols() to emit diagnostics. 77void treatUndefinedSymbol(const Undefined &, StringRef source); 78void treatUndefinedSymbol(const Undefined &, const InputSection *, 79 uint64_t offset); 80 81extern std::unique_ptr<SymbolTable> symtab; 82 83} // namespace lld::macho 84 85#endif 86