COFF: Do not create SectionChunks for discarded comdat sections.

With this change, instead of creating a SectionChunk for each section
in the object file, we only create them when we encounter a prevailing
comdat section.

Also change how symbol resolution occurs between comdat symbols. Now
only the comdat leader participates in comdat resolution, and not any
other external associated symbols. This is more in line with how COFF
semantics are defined, and should allow for a more straightforward
implementation of non-ANY comdat types.

On my machine, this change reduces our runtime linking a release
build of chrome_child.dll with /nopdb from 5.65s to 4.54s (median of
50 runs).

Differential Revision: https://reviews.llvm.org/D40238

llvm-svn: 319090
This commit is contained in:
Peter Collingbourne 2017-11-27 20:42:34 +00:00
parent 1f34379965
commit 3f2921f5ec
10 changed files with 239 additions and 165 deletions

View File

@ -38,9 +38,6 @@ SectionChunk::SectionChunk(ObjFile *F, const coff_section *H)
Alignment = Header->getAlignment(); Alignment = Header->getAlignment();
// Chunks may be discarded during comdat merging.
Discarded = false;
// If linker GC is disabled, every chunk starts out alive. If linker GC is // If linker GC is disabled, every chunk starts out alive. If linker GC is
// enabled, treat non-comdat sections as roots. Generally optimized object // enabled, treat non-comdat sections as roots. Generally optimized object
// files will be built with -ffunction-sections or /Gy, so most things worth // files will be built with -ffunction-sections or /Gy, so most things worth
@ -362,12 +359,8 @@ bool SectionChunk::isCOMDAT() const {
void SectionChunk::printDiscardedMessage() const { void SectionChunk::printDiscardedMessage() const {
// Removed by dead-stripping. If it's removed by ICF, ICF already // Removed by dead-stripping. If it's removed by ICF, ICF already
// printed out the name, so don't repeat that here. // printed out the name, so don't repeat that here.
if (Sym && this == Repl) { if (Sym && this == Repl)
if (Discarded) message("Discarded " + Sym->getName());
message("Discarded comdat symbol " + Sym->getName());
else if (!Live)
message("Discarded " + Sym->getName());
}
} }
StringRef SectionChunk::getDebugName() { StringRef SectionChunk::getDebugName() {

View File

@ -159,10 +159,9 @@ public:
void addAssociative(SectionChunk *Child); void addAssociative(SectionChunk *Child);
StringRef getDebugName() override; StringRef getDebugName() override;
void setSymbol(DefinedRegular *S) { if (!Sym) Sym = S; }
// Returns true if the chunk was not dropped by GC or COMDAT deduplication. // Returns true if the chunk was not dropped by GC.
bool isLive() { return Live && !Discarded; } bool isLive() { return Live; }
// Used by the garbage collector. // Used by the garbage collector.
void markLive() { void markLive() {
@ -171,13 +170,6 @@ public:
Live = true; Live = true;
} }
// Returns true if this chunk was dropped by COMDAT deduplication.
bool isDiscarded() const { return Discarded; }
// Used by the SymbolTable when discarding unused comdat sections. This is
// redundant when GC is enabled, as all comdat sections will start out dead.
void markDiscarded() { Discarded = true; }
// True if this is a codeview debug info chunk. These will not be laid out in // True if this is a codeview debug info chunk. These will not be laid out in
// the image. Instead they will end up in the PDB, if one is requested. // the image. Instead they will end up in the PDB, if one is requested.
bool isCodeView() const { bool isCodeView() const {
@ -213,24 +205,21 @@ public:
// The file that this chunk was created from. // The file that this chunk was created from.
ObjFile *File; ObjFile *File;
// The COMDAT leader symbol if this is a COMDAT chunk.
DefinedRegular *Sym = nullptr;
private: private:
StringRef SectionName; StringRef SectionName;
std::vector<SectionChunk *> AssocChildren; std::vector<SectionChunk *> AssocChildren;
llvm::iterator_range<const coff_relocation *> Relocs; llvm::iterator_range<const coff_relocation *> Relocs;
size_t NumRelocs; size_t NumRelocs;
// True if this chunk was discarded because it was a duplicate comdat section.
bool Discarded;
// Used by the garbage collector. // Used by the garbage collector.
bool Live; bool Live;
// Used for ICF (Identical COMDAT Folding) // Used for ICF (Identical COMDAT Folding)
void replace(SectionChunk *Other); void replace(SectionChunk *Other);
uint32_t Class[2] = {0, 0}; uint32_t Class[2] = {0, 0};
// Sym points to a section symbol if this is a COMDAT chunk.
DefinedRegular *Sym = nullptr;
}; };
// A chunk for common symbols. Common chunks don't have actual data. // A chunk for common symbols. Common chunks don't have actual data.

View File

@ -119,86 +119,158 @@ void ObjFile::parse() {
initializeSEH(); initializeSEH();
} }
// We set SectionChunk pointers in the SparseChunks vector to this value
// temporarily to mark comdat sections as having an unknown resolution. As we
// walk the object file's symbol table, once we visit either a leader symbol or
// an associative section definition together with the parent comdat's leader,
// we set the pointer to either nullptr (to mark the section as discarded) or a
// valid SectionChunk for that section.
static SectionChunk *const PendingComdat = reinterpret_cast<SectionChunk *>(1);
void ObjFile::initializeChunks() { void ObjFile::initializeChunks() {
uint32_t NumSections = COFFObj->getNumberOfSections(); uint32_t NumSections = COFFObj->getNumberOfSections();
Chunks.reserve(NumSections); Chunks.reserve(NumSections);
SparseChunks.resize(NumSections + 1); SparseChunks.resize(NumSections + 1);
for (uint32_t I = 1; I < NumSections + 1; ++I) { for (uint32_t I = 1; I < NumSections + 1; ++I) {
const coff_section *Sec; const coff_section *Sec;
StringRef Name;
if (auto EC = COFFObj->getSection(I, Sec)) if (auto EC = COFFObj->getSection(I, Sec))
fatal("getSection failed: #" + Twine(I) + ": " + EC.message()); fatal("getSection failed: #" + Twine(I) + ": " + EC.message());
if (auto EC = COFFObj->getSectionName(Sec, Name))
fatal("getSectionName failed: #" + Twine(I) + ": " + EC.message());
if (Name == ".sxdata") {
SXData = Sec;
continue;
}
if (Name == ".drectve") {
ArrayRef<uint8_t> Data;
COFFObj->getSectionContents(Sec, Data);
Directives = std::string((const char *)Data.data(), Data.size());
continue;
}
// Object files may have DWARF debug info or MS CodeView debug info if (Sec->Characteristics & IMAGE_SCN_LNK_COMDAT)
// (or both). SparseChunks[I] = PendingComdat;
//
// DWARF sections don't need any special handling from the perspective
// of the linker; they are just a data section containing relocations.
// We can just link them to complete debug info.
//
// CodeView needs a linker support. We need to interpret and debug
// info, and then write it to a separate .pdb file.
// Ignore debug info unless /debug is given.
if (!Config->Debug && Name.startswith(".debug"))
continue;
if (Sec->Characteristics & llvm::COFF::IMAGE_SCN_LNK_REMOVE)
continue;
auto *C = make<SectionChunk>(this, Sec);
// CodeView sections are stored to a different vector because they are not
// linked in the regular manner.
if (C->isCodeView())
DebugChunks.push_back(C);
else else
Chunks.push_back(C); SparseChunks[I] = readSection(I, nullptr);
SparseChunks[I] = C;
} }
} }
SectionChunk *ObjFile::readSection(uint32_t SectionNumber,
const coff_aux_section_definition *Def) {
const coff_section *Sec;
StringRef Name;
if (auto EC = COFFObj->getSection(SectionNumber, Sec))
fatal("getSection failed: #" + Twine(SectionNumber) + ": " + EC.message());
if (auto EC = COFFObj->getSectionName(Sec, Name))
fatal("getSectionName failed: #" + Twine(SectionNumber) + ": " +
EC.message());
if (Name == ".sxdata") {
SXData = Sec;
return nullptr;
}
if (Name == ".drectve") {
ArrayRef<uint8_t> Data;
COFFObj->getSectionContents(Sec, Data);
Directives = std::string((const char *)Data.data(), Data.size());
return nullptr;
}
// Object files may have DWARF debug info or MS CodeView debug info
// (or both).
//
// DWARF sections don't need any special handling from the perspective
// of the linker; they are just a data section containing relocations.
// We can just link them to complete debug info.
//
// CodeView needs a linker support. We need to interpret and debug
// info, and then write it to a separate .pdb file.
// Ignore debug info unless /debug is given.
if (!Config->Debug && Name.startswith(".debug"))
return nullptr;
if (Sec->Characteristics & llvm::COFF::IMAGE_SCN_LNK_REMOVE)
return nullptr;
auto *C = make<SectionChunk>(this, Sec);
if (Def)
C->Checksum = Def->CheckSum;
// CodeView sections are stored to a different vector because they are not
// linked in the regular manner.
if (C->isCodeView())
DebugChunks.push_back(C);
else
Chunks.push_back(C);
return C;
}
void ObjFile::readAssociativeDefinition(
COFFSymbolRef Sym, const coff_aux_section_definition *Def) {
SectionChunk *Parent = SparseChunks[Def->getNumber(Sym.isBigObj())];
// If the parent is pending, it probably means that its section definition
// appears after us in the symbol table. Leave the associated section as
// pending; we will handle it during the second pass in initializeSymbols().
if (Parent == PendingComdat)
return;
// Check whether the parent is prevailing. If it is, so are we, and we read
// the section; otherwise mark it as discarded.
int32_t SectionNumber = Sym.getSectionNumber();
if (Parent) {
SparseChunks[SectionNumber] = readSection(SectionNumber, Def);
Parent->addAssociative(SparseChunks[SectionNumber]);
} else {
SparseChunks[SectionNumber] = nullptr;
}
}
Symbol *ObjFile::createRegular(COFFSymbolRef Sym) {
SectionChunk *SC = SparseChunks[Sym.getSectionNumber()];
if (Sym.isExternal()) {
StringRef Name;
COFFObj->getSymbolName(Sym, Name);
if (SC)
return Symtab->addRegular(this, Name, Sym.getGeneric(), SC);
return Symtab->addUndefined(Name, this, false);
}
if (SC)
return make<DefinedRegular>(this, /*Name*/ "", false,
/*IsExternal*/ false, Sym.getGeneric(), SC);
return nullptr;
}
void ObjFile::initializeSymbols() { void ObjFile::initializeSymbols() {
uint32_t NumSymbols = COFFObj->getNumberOfSymbols(); uint32_t NumSymbols = COFFObj->getNumberOfSymbols();
Symbols.resize(NumSymbols); Symbols.resize(NumSymbols);
SmallVector<std::pair<Symbol *, uint32_t>, 8> WeakAliases; SmallVector<std::pair<Symbol *, uint32_t>, 8> WeakAliases;
int32_t LastSectionNumber = 0; std::vector<uint32_t> PendingIndexes;
PendingIndexes.reserve(NumSymbols);
std::vector<const coff_aux_section_definition *> ComdatDefs(
COFFObj->getNumberOfSections() + 1);
for (uint32_t I = 0; I < NumSymbols; ++I) { for (uint32_t I = 0; I < NumSymbols; ++I) {
COFFSymbolRef COFFSym = check(COFFObj->getSymbol(I)); COFFSymbolRef COFFSym = check(COFFObj->getSymbol(I));
const void *AuxP = nullptr;
if (COFFSym.getNumberOfAuxSymbols())
AuxP = check(COFFObj->getSymbol(I + 1)).getRawPtr();
bool IsFirst = (LastSectionNumber != COFFSym.getSectionNumber());
Symbol *Sym = nullptr;
if (COFFSym.isUndefined()) { if (COFFSym.isUndefined()) {
Sym = createUndefined(COFFSym); Symbols[I] = createUndefined(COFFSym);
} else if (COFFSym.isWeakExternal()) { } else if (COFFSym.isWeakExternal()) {
Sym = createUndefined(COFFSym); Symbols[I] = createUndefined(COFFSym);
uint32_t TagIndex = uint32_t TagIndex = COFFSym.getAux<coff_aux_weak_external>()->TagIndex;
static_cast<const coff_aux_weak_external *>(AuxP)->TagIndex; WeakAliases.emplace_back(Symbols[I], TagIndex);
WeakAliases.emplace_back(Sym, TagIndex); } else if (Optional<Symbol *> OptSym = createDefined(COFFSym, ComdatDefs)) {
Symbols[I] = *OptSym;
} else { } else {
Sym = createDefined(COFFSym, AuxP, IsFirst); // createDefined() returns None if a symbol belongs to a section that
// was pending at the point when the symbol was read. This can happen in
// two cases:
// 1) section definition symbol for a comdat leader;
// 2) symbol belongs to a comdat section associated with a section whose
// section definition symbol appears later in the symbol table.
// In both of these cases, we can expect the section to be resolved by
// the time we finish visiting the remaining symbols in the symbol
// table. So we postpone the handling of this symbol until that time.
PendingIndexes.push_back(I);
} }
Symbols[I] = Sym;
I += COFFSym.getNumberOfAuxSymbols(); I += COFFSym.getNumberOfAuxSymbols();
LastSectionNumber = COFFSym.getSectionNumber(); }
for (uint32_t I : PendingIndexes) {
COFFSymbolRef Sym = check(COFFObj->getSymbol(I));
if (auto *Def = Sym.getSectionDefinition())
if (Def->Selection == IMAGE_COMDAT_SELECT_ASSOCIATIVE)
readAssociativeDefinition(Sym, Def);
Symbols[I] = createRegular(Sym);
} }
for (auto &KV : WeakAliases) { for (auto &KV : WeakAliases) {
@ -214,8 +286,9 @@ Symbol *ObjFile::createUndefined(COFFSymbolRef Sym) {
return Symtab->addUndefined(Name, this, Sym.isWeakExternal()); return Symtab->addUndefined(Name, this, Sym.isWeakExternal());
} }
Symbol *ObjFile::createDefined(COFFSymbolRef Sym, const void *AuxP, Optional<Symbol *> ObjFile::createDefined(
bool IsFirst) { COFFSymbolRef Sym,
std::vector<const coff_aux_section_definition *> &ComdatDefs) {
StringRef Name; StringRef Name;
if (Sym.isCommon()) { if (Sym.isCommon()) {
auto *C = make<CommonChunk>(Sym); auto *C = make<CommonChunk>(Sym);
@ -254,37 +327,46 @@ Symbol *ObjFile::createDefined(COFFSymbolRef Sym, const void *AuxP,
if ((uint32_t)SectionNumber >= SparseChunks.size()) if ((uint32_t)SectionNumber >= SparseChunks.size())
fatal("broken object file: " + toString(this)); fatal("broken object file: " + toString(this));
// Nothing else to do without a section chunk. // Handle comdat leader symbols.
auto *SC = SparseChunks[SectionNumber]; if (const coff_aux_section_definition *Def = ComdatDefs[SectionNumber]) {
if (!SC) ComdatDefs[SectionNumber] = nullptr;
return nullptr; Symbol *Leader;
bool Prevailing;
// Handle section definitions if (Sym.isExternal()) {
if (IsFirst && AuxP) { COFFObj->getSymbolName(Sym, Name);
auto *Aux = reinterpret_cast<const coff_aux_section_definition *>(AuxP); std::tie(Leader, Prevailing) =
if (Aux->Selection == IMAGE_COMDAT_SELECT_ASSOCIATIVE) Symtab->addComdat(this, Name, Sym.getGeneric());
if (auto *ParentSC = SparseChunks[Aux->getNumber(Sym.isBigObj())]) { } else {
ParentSC->addAssociative(SC); Leader = make<DefinedRegular>(this, /*Name*/ "", false,
// If we already discarded the parent, discard the child. /*IsExternal*/ false, Sym.getGeneric());
if (ParentSC->isDiscarded()) Prevailing = true;
SC->markDiscarded(); }
} if (Prevailing) {
SC->Checksum = Aux->CheckSum; SectionChunk *C = readSection(SectionNumber, Def);
SparseChunks[SectionNumber] = C;
C->Sym = cast<DefinedRegular>(Leader);
cast<DefinedRegular>(Leader)->Data = &C->Repl;
} else {
SparseChunks[SectionNumber] = nullptr;
}
return Leader;
} }
DefinedRegular *B; // Read associative section definitions and prepare to handle the comdat
if (Sym.isExternal()) { // leader symbol by setting the section's ComdatDefs pointer if we encounter a
COFFObj->getSymbolName(Sym, Name); // non-associative comdat.
Symbol *S = if (SparseChunks[SectionNumber] == PendingComdat) {
Symtab->addRegular(this, Name, SC->isCOMDAT(), Sym.getGeneric(), SC); if (auto *Def = Sym.getSectionDefinition()) {
B = cast<DefinedRegular>(S); if (Def->Selection == IMAGE_COMDAT_SELECT_ASSOCIATIVE)
} else readAssociativeDefinition(Sym, Def);
B = make<DefinedRegular>(this, /*Name*/ "", SC->isCOMDAT(), else
/*IsExternal*/ false, Sym.getGeneric(), SC); ComdatDefs[SectionNumber] = Def;
if (SC->isCOMDAT() && Sym.getValue() == 0 && !AuxP) }
SC->setSymbol(B); }
return B; if (SparseChunks[SectionNumber] == PendingComdat)
return None;
return createRegular(Sym);
} }
void ObjFile::initializeSEH() { void ObjFile::initializeSEH() {
@ -361,8 +443,12 @@ void ImportFile::parse() {
void BitcodeFile::parse() { void BitcodeFile::parse() {
Obj = check(lto::InputFile::create(MemoryBufferRef( Obj = check(lto::InputFile::create(MemoryBufferRef(
MB.getBuffer(), Saver.save(ParentName + MB.getBufferIdentifier())))); MB.getBuffer(), Saver.save(ParentName + MB.getBufferIdentifier()))));
std::vector<std::pair<Symbol *, bool>> Comdat(Obj->getComdatTable().size());
for (size_t I = 0; I != Obj->getComdatTable().size(); ++I)
Comdat[I] = Symtab->addComdat(this, Saver.save(Obj->getComdatTable()[I]));
for (const lto::InputFile::Symbol &ObjSym : Obj->symbols()) { for (const lto::InputFile::Symbol &ObjSym : Obj->symbols()) {
StringRef SymName = Saver.save(ObjSym.getName()); StringRef SymName = Saver.save(ObjSym.getName());
int ComdatIndex = ObjSym.getComdatIndex();
Symbol *Sym; Symbol *Sym;
if (ObjSym.isUndefined()) { if (ObjSym.isUndefined()) {
Sym = Symtab->addUndefined(SymName, this, false); Sym = Symtab->addUndefined(SymName, this, false);
@ -374,9 +460,15 @@ void BitcodeFile::parse() {
std::string Fallback = ObjSym.getCOFFWeakExternalFallback(); std::string Fallback = ObjSym.getCOFFWeakExternalFallback();
Symbol *Alias = Symtab->addUndefined(Saver.save(Fallback)); Symbol *Alias = Symtab->addUndefined(Saver.save(Fallback));
checkAndSetWeakAlias(Symtab, this, Sym, Alias); checkAndSetWeakAlias(Symtab, this, Sym, Alias);
} else if (ComdatIndex != -1) {
if (SymName == Obj->getComdatTable()[ComdatIndex])
Sym = Comdat[ComdatIndex].first;
else if (Comdat[ComdatIndex].second)
Sym = Symtab->addRegular(this, SymName);
else
Sym = Symtab->addUndefined(SymName, this, false);
} else { } else {
bool IsCOMDAT = ObjSym.getComdatIndex() != -1; Sym = Symtab->addRegular(this, SymName);
Sym = Symtab->addRegular(this, SymName, IsCOMDAT);
} }
SymbolBodies.push_back(Sym); SymbolBodies.push_back(Sym);
} }

View File

@ -142,7 +142,19 @@ private:
void initializeSymbols(); void initializeSymbols();
void initializeSEH(); void initializeSEH();
Symbol *createDefined(COFFSymbolRef Sym, const void *Aux, bool IsFirst); SectionChunk *
readSection(uint32_t SectionNumber,
const llvm::object::coff_aux_section_definition *Def);
void readAssociativeDefinition(
COFFSymbolRef COFFSym,
const llvm::object::coff_aux_section_definition *Def);
llvm::Optional<Symbol *>
createDefined(COFFSymbolRef Sym,
std::vector<const llvm::object::coff_aux_section_definition *>
&ComdatDefs);
Symbol *createRegular(COFFSymbolRef Sym);
Symbol *createUndefined(COFFSymbolRef Sym); Symbol *createUndefined(COFFSymbolRef Sym);
std::unique_ptr<COFFObjectFile> COFFObj; std::unique_ptr<COFFObjectFile> COFFObj;

View File

@ -52,13 +52,6 @@ void markLive(const std::vector<Chunk *> &Chunks) {
while (!Worklist.empty()) { while (!Worklist.empty()) {
SectionChunk *SC = Worklist.pop_back_val(); SectionChunk *SC = Worklist.pop_back_val();
// If this section was discarded, there are relocations referring to
// discarded sections. Ignore these sections to avoid crashing. They will be
// diagnosed during relocation processing.
if (SC->isDiscarded())
continue;
assert(SC->isLive() && "We mark as live when pushing onto the worklist!"); assert(SC->isLive() && "We mark as live when pushing onto the worklist!");
// Mark all symbols listed in the relocation table for this section. // Mark all symbols listed in the relocation table for this section.

View File

@ -24,36 +24,6 @@ using namespace llvm;
namespace lld { namespace lld {
namespace coff { namespace coff {
enum SymbolPreference {
SP_EXISTING = -1,
SP_CONFLICT = 0,
SP_NEW = 1,
};
/// Checks if an existing symbol S should be kept or replaced by a new symbol.
/// Returns SP_EXISTING when S should be kept, SP_NEW when the new symbol
/// should be kept, and SP_CONFLICT if no valid resolution exists.
static SymbolPreference compareDefined(Symbol *S, bool WasInserted,
bool NewIsCOMDAT) {
// If the symbol wasn't previously known, the new symbol wins by default.
if (WasInserted || !isa<Defined>(S))
return SP_NEW;
// If the existing symbol is a DefinedRegular, both it and the new symbol
// must be comdats. In that case, we have no reason to prefer one symbol
// over the other, and we keep the existing one. If one of the symbols
// is not a comdat, we report a conflict.
if (auto *R = dyn_cast<DefinedRegular>(S)) {
if (NewIsCOMDAT && R->isCOMDAT())
return SP_EXISTING;
else
return SP_CONFLICT;
}
// Existing symbol is not a DefinedRegular; new symbol wins.
return SP_NEW;
}
SymbolTable *Symtab; SymbolTable *Symtab;
void SymbolTable::addFile(InputFile *File) { void SymbolTable::addFile(InputFile *File) {
@ -240,7 +210,7 @@ Symbol *SymbolTable::addSynthetic(StringRef N, Chunk *C) {
return S; return S;
} }
Symbol *SymbolTable::addRegular(InputFile *F, StringRef N, bool IsCOMDAT, Symbol *SymbolTable::addRegular(InputFile *F, StringRef N,
const coff_symbol_generic *Sym, const coff_symbol_generic *Sym,
SectionChunk *C) { SectionChunk *C) {
Symbol *S; Symbol *S;
@ -248,22 +218,32 @@ Symbol *SymbolTable::addRegular(InputFile *F, StringRef N, bool IsCOMDAT,
std::tie(S, WasInserted) = insert(N); std::tie(S, WasInserted) = insert(N);
if (!isa<BitcodeFile>(F)) if (!isa<BitcodeFile>(F))
S->IsUsedInRegularObj = true; S->IsUsedInRegularObj = true;
SymbolPreference SP = compareDefined(S, WasInserted, IsCOMDAT); if (WasInserted || !isa<DefinedRegular>(S))
if (SP == SP_CONFLICT) { replaceSymbol<DefinedRegular>(S, F, N, /*IsCOMDAT*/ false,
/*IsExternal*/ true, Sym, C);
else
reportDuplicate(S, F); reportDuplicate(S, F);
} else if (SP == SP_NEW) {
replaceSymbol<DefinedRegular>(S, F, N, IsCOMDAT, /*IsExternal*/ true, Sym,
C);
} else if (SP == SP_EXISTING && IsCOMDAT && C) {
C->markDiscarded();
// Discard associative chunks that we've parsed so far. No need to recurse
// because an associative section cannot have children.
for (SectionChunk *Child : C->children())
Child->markDiscarded();
}
return S; return S;
} }
std::pair<Symbol *, bool>
SymbolTable::addComdat(InputFile *F, StringRef N,
const coff_symbol_generic *Sym) {
Symbol *S;
bool WasInserted;
std::tie(S, WasInserted) = insert(N);
if (!isa<BitcodeFile>(F))
S->IsUsedInRegularObj = true;
if (WasInserted || !isa<DefinedRegular>(S)) {
replaceSymbol<DefinedRegular>(S, F, N, /*IsCOMDAT*/ true,
/*IsExternal*/ true, Sym, nullptr);
return {S, true};
}
if (!cast<DefinedRegular>(S)->isCOMDAT())
reportDuplicate(S, F);
return {S, false};
}
Symbol *SymbolTable::addCommon(InputFile *F, StringRef N, uint64_t Size, Symbol *SymbolTable::addCommon(InputFile *F, StringRef N, uint64_t Size,
const coff_symbol_generic *Sym, CommonChunk *C) { const coff_symbol_generic *Sym, CommonChunk *C) {
Symbol *S; Symbol *S;

View File

@ -83,9 +83,12 @@ public:
Symbol *addUndefined(StringRef Name, InputFile *F, bool IsWeakAlias); Symbol *addUndefined(StringRef Name, InputFile *F, bool IsWeakAlias);
void addLazy(ArchiveFile *F, const Archive::Symbol Sym); void addLazy(ArchiveFile *F, const Archive::Symbol Sym);
Symbol *addAbsolute(StringRef N, COFFSymbolRef S); Symbol *addAbsolute(StringRef N, COFFSymbolRef S);
Symbol *addRegular(InputFile *F, StringRef N, bool IsCOMDAT, Symbol *addRegular(InputFile *F, StringRef N,
const llvm::object::coff_symbol_generic *S = nullptr, const llvm::object::coff_symbol_generic *S = nullptr,
SectionChunk *C = nullptr); SectionChunk *C = nullptr);
std::pair<Symbol *, bool>
addComdat(InputFile *F, StringRef N,
const llvm::object::coff_symbol_generic *S = nullptr);
Symbol *addCommon(InputFile *F, StringRef N, uint64_t Size, Symbol *addCommon(InputFile *F, StringRef N, uint64_t Size,
const llvm::object::coff_symbol_generic *S = nullptr, const llvm::object::coff_symbol_generic *S = nullptr,
CommonChunk *C = nullptr); CommonChunk *C = nullptr);

View File

@ -169,7 +169,6 @@ public:
SectionChunk *getChunk() const { return *Data; } SectionChunk *getChunk() const { return *Data; }
uint32_t getValue() const { return Sym->Value; } uint32_t getValue() const { return Sym->Value; }
private:
SectionChunk **Data; SectionChunk **Data;
}; };

View File

@ -18,7 +18,6 @@ main_global:
.section .CRT$XCU,"dr",associative,main_global .section .CRT$XCU,"dr",associative,main_global
.p2align 3 .p2align 3
.globl assoc_global
assoc_global: assoc_global:
.quad main_global .quad main_global

View File

@ -275,6 +275,8 @@ struct coff_symbol_generic {
support::ulittle32_t Value; support::ulittle32_t Value;
}; };
struct coff_aux_section_definition;
class COFFSymbolRef { class COFFSymbolRef {
public: public:
COFFSymbolRef() = default; COFFSymbolRef() = default;
@ -346,6 +348,18 @@ public:
return (getType() & 0xF0) >> COFF::SCT_COMPLEX_TYPE_SHIFT; return (getType() & 0xF0) >> COFF::SCT_COMPLEX_TYPE_SHIFT;
} }
template <typename T> const T *getAux() const {
return CS16 ? reinterpret_cast<const T *>(CS16 + 1)
: reinterpret_cast<const T *>(CS32 + 1);
}
const coff_aux_section_definition *getSectionDefinition() const {
if (!getNumberOfAuxSymbols() ||
getStorageClass() != COFF::IMAGE_SYM_CLASS_STATIC)
return nullptr;
return getAux<coff_aux_section_definition>();
}
bool isAbsolute() const { bool isAbsolute() const {
return getSectionNumber() == -1; return getSectionNumber() == -1;
} }