1 //===- Symbols.cpp --------------------------------------------------------===//
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
12 #include "InputFiles.h"
15 #include "llvm/ADT/STLExtras.h"
16 #include "llvm/Support/Debug.h"
17 #include "llvm/Support/raw_ostream.h"
20 using namespace llvm::object;
22 // Returns a symbol name for an error message.
23 std::string lld::toString(coff::SymbolBody &B) {
24 if (Optional<std::string> S = coff::demangle(B.getName()))
25 return ("\"" + *S + "\" (" + B.getName() + ")").str();
32 StringRef SymbolBody::getName() {
33 // COFF symbol names are read lazily for a performance reason.
34 // Non-external symbol names are never used by the linker except for logging
35 // or debugging. Their internal references are resolved not by name but by
36 // symbol index. And because they are not external, no one can refer them by
37 // name. Object files contain lots of non-external symbols, and creating
38 // StringRefs for them (which involves lots of strlen() on the string table)
39 // is a waste of time.
41 auto *D = cast<DefinedCOFF>(this);
42 cast<ObjectFile>(D->File)->getCOFFObj()->getSymbolName(D->Sym, Name);
47 InputFile *SymbolBody::getFile() {
48 if (auto *Sym = dyn_cast<DefinedCOFF>(this))
50 if (auto *Sym = dyn_cast<Lazy>(this))
55 COFFSymbolRef DefinedCOFF::getCOFFSymbol() {
57 cast<ObjectFile>(File)->getCOFFObj()->getSymbolTableEntrySize();
58 if (SymSize == sizeof(coff_symbol16))
59 return COFFSymbolRef(reinterpret_cast<const coff_symbol16 *>(Sym));
60 assert(SymSize == sizeof(coff_symbol32));
61 return COFFSymbolRef(reinterpret_cast<const coff_symbol32 *>(Sym));
64 DefinedImportThunk::DefinedImportThunk(StringRef Name, DefinedImportData *S,
66 : Defined(DefinedImportThunkKind, Name) {
68 case AMD64: Data = make<ImportThunkChunkX64>(S); return;
69 case I386: Data = make<ImportThunkChunkX86>(S); return;
70 case ARMNT: Data = make<ImportThunkChunkARM>(S); return;
71 default: llvm_unreachable("unknown machine type");
75 Defined *Undefined::getWeakAlias() {
76 // A weak alias may be a weak alias to another symbol, so check recursively.
77 for (SymbolBody *A = WeakAlias; A; A = cast<Undefined>(A)->WeakAlias)
78 if (auto *D = dyn_cast<Defined>(A))