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(name: 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