This diff adds support for weak definitions, though it doesn't handle weak symbols in dylibs quite correctly -- we need to emit binding opcodes for them in the weak binding section rather than the lazy binding section. What *is* covered in this diff: 1. Reading the weak flag from symbol table / export trie, and writing it to the export trie 2. Refining the symbol table's rules for choosing one symbol definition over another. Wrote a few dozen test cases to make sure we were matching ld64's behavior. We can now link basic C++ programs. Reviewed By: #lld-macho, compnerd Differential Revision: https://reviews.llvm.org/D83532
58 lines
1.6 KiB
C++
58 lines
1.6 KiB
C++
//===- SymbolTable.h --------------------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLD_MACHO_SYMBOL_TABLE_H
|
|
#define LLD_MACHO_SYMBOL_TABLE_H
|
|
|
|
#include "lld/Common/LLVM.h"
|
|
#include "llvm/ADT/CachedHashString.h"
|
|
#include "llvm/ADT/DenseMap.h"
|
|
#include "llvm/Object/Archive.h"
|
|
|
|
namespace lld {
|
|
namespace macho {
|
|
|
|
class ArchiveFile;
|
|
class DylibFile;
|
|
class InputSection;
|
|
class Symbol;
|
|
|
|
/*
|
|
* Note that the SymbolTable handles name collisions by calling
|
|
* replaceSymbol(), which does an in-place update of the Symbol via `placement
|
|
* new`. Therefore, there is no need to update any relocations that hold
|
|
* pointers the "old" Symbol -- they will automatically point to the new one.
|
|
*/
|
|
class SymbolTable {
|
|
public:
|
|
Symbol *addDefined(StringRef name, InputSection *isec, uint32_t value,
|
|
bool isWeakDef);
|
|
|
|
Symbol *addUndefined(StringRef name);
|
|
|
|
Symbol *addDylib(StringRef name, DylibFile *file, bool isWeakDef);
|
|
|
|
Symbol *addLazy(StringRef name, ArchiveFile *file,
|
|
const llvm::object::Archive::Symbol &sym);
|
|
|
|
ArrayRef<Symbol *> getSymbols() const { return symVector; }
|
|
Symbol *find(StringRef name);
|
|
|
|
private:
|
|
std::pair<Symbol *, bool> insert(StringRef name);
|
|
llvm::DenseMap<llvm::CachedHashStringRef, int> symMap;
|
|
std::vector<Symbol *> symVector;
|
|
};
|
|
|
|
extern SymbolTable *symtab;
|
|
|
|
} // namespace macho
|
|
} // namespace lld
|
|
|
|
#endif
|