20e652d627
Add SharedLibraryFile and ArchiveLibraryFile subclasses of File. Add command line options to lld-core to set various ResolverOptions settings and added lots of test cases to verify the options work. llvm-svn: 155183
107 lines
3.2 KiB
C++
107 lines
3.2 KiB
C++
//===- Core/SymbolTable.h - Main Symbol Table -----------------------------===//
|
|
//
|
|
// The LLVM Linker
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLD_CORE_SYMBOL_TABLE_H_
|
|
#define LLD_CORE_SYMBOL_TABLE_H_
|
|
|
|
#include "lld/Core/LLVM.h"
|
|
|
|
#include "llvm/ADT/DenseSet.h"
|
|
#include "llvm/ADT/StringExtras.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
|
|
#include <cstring>
|
|
#include <map>
|
|
#include <vector>
|
|
|
|
namespace lld {
|
|
|
|
class Atom;
|
|
class DefinedAtom;
|
|
class UndefinedAtom;
|
|
class SharedLibraryAtom;
|
|
class AbsoluteAtom;
|
|
class ResolverOptions;
|
|
|
|
/// The SymbolTable class is responsible for coalescing atoms.
|
|
///
|
|
/// All atoms coalescable by-name or by-content should be added.
|
|
/// The method replacement() can be used to find the replacement atom
|
|
/// if an atom has been coalesced away.
|
|
class SymbolTable {
|
|
public:
|
|
SymbolTable(ResolverOptions&);
|
|
|
|
/// @brief add atom to symbol table
|
|
void add(const DefinedAtom &);
|
|
|
|
/// @brief add atom to symbol table
|
|
void add(const UndefinedAtom &);
|
|
|
|
/// @brief add atom to symbol table
|
|
void add(const SharedLibraryAtom &);
|
|
|
|
/// @brief add atom to symbol table
|
|
void add(const AbsoluteAtom &);
|
|
|
|
/// @brief checks if name is in symbol table and if so atom is not
|
|
/// UndefinedAtom
|
|
bool isDefined(StringRef sym);
|
|
|
|
/// @brief returns atom in symbol table for specified name (or nullptr)
|
|
const Atom *findByName(StringRef sym);
|
|
|
|
/// @brief returns vector of remaining UndefinedAtoms
|
|
void undefines(std::vector<const Atom *>&);
|
|
|
|
/// returns vector of tentative definitions
|
|
void tentativeDefinitions(std::vector<StringRef> &);
|
|
|
|
/// @brief count of by-name entries in symbol table
|
|
unsigned int size();
|
|
|
|
/// @brief if atom has been coalesced away, return replacement, else return atom
|
|
const Atom *replacement(const Atom *);
|
|
|
|
private:
|
|
typedef llvm::DenseMap<const Atom *, const Atom *> AtomToAtom;
|
|
|
|
struct StringRefMappingInfo {
|
|
static StringRef getEmptyKey() { return StringRef(); }
|
|
static StringRef getTombstoneKey() { return StringRef(" ", 0); }
|
|
static unsigned getHashValue(StringRef const val) {
|
|
return llvm::HashString(val); }
|
|
static bool isEqual(StringRef const lhs,
|
|
StringRef const rhs) { return lhs.equals(rhs); }
|
|
};
|
|
typedef llvm::DenseMap<StringRef, const Atom *,
|
|
StringRefMappingInfo> NameToAtom;
|
|
|
|
struct AtomMappingInfo {
|
|
static const DefinedAtom * getEmptyKey() { return nullptr; }
|
|
static const DefinedAtom * getTombstoneKey() { return (DefinedAtom*)(-1); }
|
|
static unsigned getHashValue(const DefinedAtom * const Val);
|
|
static bool isEqual(const DefinedAtom * const LHS,
|
|
const DefinedAtom * const RHS);
|
|
};
|
|
typedef llvm::DenseSet<const DefinedAtom*, AtomMappingInfo> AtomContentSet;
|
|
|
|
void addByName(const Atom &);
|
|
void addByContent(const DefinedAtom &);
|
|
|
|
ResolverOptions &_options;
|
|
AtomToAtom _replacedAtoms;
|
|
NameToAtom _nameTable;
|
|
AtomContentSet _contentTable;
|
|
};
|
|
|
|
} // namespace lld
|
|
|
|
#endif // LLD_CORE_SYMBOL_TABLE_H_
|