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 bool isCold = false);
44
45 Defined *aliasDefined(Defined *src, StringRef target, InputFile *newFile,
46 bool makePrivateExtern = false);
47
48 Symbol *addUndefined(StringRef name, InputFile *, bool isWeakRef);
49
50 Symbol *addCommon(StringRef name, InputFile *, uint64_t size, uint32_t align,
51 bool isPrivateExtern);
52
53 Symbol *addDylib(StringRef name, DylibFile *file, bool isWeakDef, bool isTlv);
54 Symbol *addDynamicLookup(StringRef name);
55
56 Symbol *addLazyArchive(StringRef name, ArchiveFile *file,
57 const llvm::object::Archive::Symbol &sym);
58 Symbol *addLazyObject(StringRef name, InputFile &file);
59
60 Defined *addSynthetic(StringRef name, InputSection *, uint64_t value,
61 bool isPrivateExtern, bool includeInSymtab,
62 bool referencedDynamically);
63
64 ArrayRef<Symbol *> getSymbols() const { return symVector; }
65 Symbol *find(llvm::CachedHashStringRef name);
66 Symbol *find(StringRef name) { return find(name: llvm::CachedHashStringRef(name)); }
67
68private:
69 std::pair<Symbol *, bool> insert(StringRef name, const InputFile *);
70 llvm::DenseMap<llvm::CachedHashStringRef, int> symMap;
71 std::vector<Symbol *> symVector;
72};
73
74void reportPendingUndefinedSymbols();
75void reportPendingDuplicateSymbols();
76
77// Call reportPendingUndefinedSymbols() to emit diagnostics.
78void treatUndefinedSymbol(const Undefined &, StringRef source);
79void treatUndefinedSymbol(const Undefined &, const InputSection *,
80 uint64_t offset);
81
82extern std::unique_ptr<SymbolTable> symtab;
83
84} // namespace lld::macho
85
86#endif
87