annotate lld/COFF/InputFiles.cpp @ 150:1d019706d866

LLVM10
author anatofuz
date Thu, 13 Feb 2020 15:10:13 +0900
parents
children 0572611fdcc8
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
150
anatofuz
parents:
diff changeset
1 //===- InputFiles.cpp -----------------------------------------------------===//
anatofuz
parents:
diff changeset
2 //
anatofuz
parents:
diff changeset
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
anatofuz
parents:
diff changeset
4 // See https://llvm.org/LICENSE.txt for license information.
anatofuz
parents:
diff changeset
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
anatofuz
parents:
diff changeset
6 //
anatofuz
parents:
diff changeset
7 //===----------------------------------------------------------------------===//
anatofuz
parents:
diff changeset
8
anatofuz
parents:
diff changeset
9 #include "InputFiles.h"
anatofuz
parents:
diff changeset
10 #include "Chunks.h"
anatofuz
parents:
diff changeset
11 #include "Config.h"
anatofuz
parents:
diff changeset
12 #include "DebugTypes.h"
anatofuz
parents:
diff changeset
13 #include "Driver.h"
anatofuz
parents:
diff changeset
14 #include "SymbolTable.h"
anatofuz
parents:
diff changeset
15 #include "Symbols.h"
anatofuz
parents:
diff changeset
16 #include "lld/Common/DWARF.h"
anatofuz
parents:
diff changeset
17 #include "lld/Common/ErrorHandler.h"
anatofuz
parents:
diff changeset
18 #include "lld/Common/Memory.h"
anatofuz
parents:
diff changeset
19 #include "llvm-c/lto.h"
anatofuz
parents:
diff changeset
20 #include "llvm/ADT/SmallVector.h"
anatofuz
parents:
diff changeset
21 #include "llvm/ADT/Triple.h"
anatofuz
parents:
diff changeset
22 #include "llvm/ADT/Twine.h"
anatofuz
parents:
diff changeset
23 #include "llvm/BinaryFormat/COFF.h"
anatofuz
parents:
diff changeset
24 #include "llvm/DebugInfo/CodeView/DebugSubsectionRecord.h"
anatofuz
parents:
diff changeset
25 #include "llvm/DebugInfo/CodeView/SymbolDeserializer.h"
anatofuz
parents:
diff changeset
26 #include "llvm/DebugInfo/CodeView/SymbolRecord.h"
anatofuz
parents:
diff changeset
27 #include "llvm/DebugInfo/CodeView/TypeDeserializer.h"
anatofuz
parents:
diff changeset
28 #include "llvm/LTO/LTO.h"
anatofuz
parents:
diff changeset
29 #include "llvm/Object/Binary.h"
anatofuz
parents:
diff changeset
30 #include "llvm/Object/COFF.h"
anatofuz
parents:
diff changeset
31 #include "llvm/Support/Casting.h"
anatofuz
parents:
diff changeset
32 #include "llvm/Support/Endian.h"
anatofuz
parents:
diff changeset
33 #include "llvm/Support/Error.h"
anatofuz
parents:
diff changeset
34 #include "llvm/Support/ErrorOr.h"
anatofuz
parents:
diff changeset
35 #include "llvm/Support/FileSystem.h"
anatofuz
parents:
diff changeset
36 #include "llvm/Support/Path.h"
anatofuz
parents:
diff changeset
37 #include "llvm/Target/TargetOptions.h"
anatofuz
parents:
diff changeset
38 #include <cstring>
anatofuz
parents:
diff changeset
39 #include <system_error>
anatofuz
parents:
diff changeset
40 #include <utility>
anatofuz
parents:
diff changeset
41
anatofuz
parents:
diff changeset
42 using namespace llvm;
anatofuz
parents:
diff changeset
43 using namespace llvm::COFF;
anatofuz
parents:
diff changeset
44 using namespace llvm::codeview;
anatofuz
parents:
diff changeset
45 using namespace llvm::object;
anatofuz
parents:
diff changeset
46 using namespace llvm::support::endian;
anatofuz
parents:
diff changeset
47
anatofuz
parents:
diff changeset
48 using llvm::Triple;
anatofuz
parents:
diff changeset
49 using llvm::support::ulittle32_t;
anatofuz
parents:
diff changeset
50
anatofuz
parents:
diff changeset
51 namespace lld {
anatofuz
parents:
diff changeset
52
anatofuz
parents:
diff changeset
53 // Returns the last element of a path, which is supposed to be a filename.
anatofuz
parents:
diff changeset
54 static StringRef getBasename(StringRef path) {
anatofuz
parents:
diff changeset
55 return sys::path::filename(path, sys::path::Style::windows);
anatofuz
parents:
diff changeset
56 }
anatofuz
parents:
diff changeset
57
anatofuz
parents:
diff changeset
58 // Returns a string in the format of "foo.obj" or "foo.obj(bar.lib)".
anatofuz
parents:
diff changeset
59 std::string toString(const coff::InputFile *file) {
anatofuz
parents:
diff changeset
60 if (!file)
anatofuz
parents:
diff changeset
61 return "<internal>";
anatofuz
parents:
diff changeset
62 if (file->parentName.empty() || file->kind() == coff::InputFile::ImportKind)
anatofuz
parents:
diff changeset
63 return std::string(file->getName());
anatofuz
parents:
diff changeset
64
anatofuz
parents:
diff changeset
65 return (getBasename(file->parentName) + "(" + getBasename(file->getName()) +
anatofuz
parents:
diff changeset
66 ")")
anatofuz
parents:
diff changeset
67 .str();
anatofuz
parents:
diff changeset
68 }
anatofuz
parents:
diff changeset
69
anatofuz
parents:
diff changeset
70 namespace coff {
anatofuz
parents:
diff changeset
71
anatofuz
parents:
diff changeset
72 std::vector<ObjFile *> ObjFile::instances;
anatofuz
parents:
diff changeset
73 std::vector<ImportFile *> ImportFile::instances;
anatofuz
parents:
diff changeset
74 std::vector<BitcodeFile *> BitcodeFile::instances;
anatofuz
parents:
diff changeset
75
anatofuz
parents:
diff changeset
76 /// Checks that Source is compatible with being a weak alias to Target.
anatofuz
parents:
diff changeset
77 /// If Source is Undefined and has no weak alias set, makes it a weak
anatofuz
parents:
diff changeset
78 /// alias to Target.
anatofuz
parents:
diff changeset
79 static void checkAndSetWeakAlias(SymbolTable *symtab, InputFile *f,
anatofuz
parents:
diff changeset
80 Symbol *source, Symbol *target) {
anatofuz
parents:
diff changeset
81 if (auto *u = dyn_cast<Undefined>(source)) {
anatofuz
parents:
diff changeset
82 if (u->weakAlias && u->weakAlias != target) {
anatofuz
parents:
diff changeset
83 // Weak aliases as produced by GCC are named in the form
anatofuz
parents:
diff changeset
84 // .weak.<weaksymbol>.<othersymbol>, where <othersymbol> is the name
anatofuz
parents:
diff changeset
85 // of another symbol emitted near the weak symbol.
anatofuz
parents:
diff changeset
86 // Just use the definition from the first object file that defined
anatofuz
parents:
diff changeset
87 // this weak symbol.
anatofuz
parents:
diff changeset
88 if (config->mingw)
anatofuz
parents:
diff changeset
89 return;
anatofuz
parents:
diff changeset
90 symtab->reportDuplicate(source, f);
anatofuz
parents:
diff changeset
91 }
anatofuz
parents:
diff changeset
92 u->weakAlias = target;
anatofuz
parents:
diff changeset
93 }
anatofuz
parents:
diff changeset
94 }
anatofuz
parents:
diff changeset
95
anatofuz
parents:
diff changeset
96 static bool ignoredSymbolName(StringRef name) {
anatofuz
parents:
diff changeset
97 return name == "@feat.00" || name == "@comp.id";
anatofuz
parents:
diff changeset
98 }
anatofuz
parents:
diff changeset
99
anatofuz
parents:
diff changeset
100 ArchiveFile::ArchiveFile(MemoryBufferRef m) : InputFile(ArchiveKind, m) {}
anatofuz
parents:
diff changeset
101
anatofuz
parents:
diff changeset
102 void ArchiveFile::parse() {
anatofuz
parents:
diff changeset
103 // Parse a MemoryBufferRef as an archive file.
anatofuz
parents:
diff changeset
104 file = CHECK(Archive::create(mb), this);
anatofuz
parents:
diff changeset
105
anatofuz
parents:
diff changeset
106 // Read the symbol table to construct Lazy objects.
anatofuz
parents:
diff changeset
107 for (const Archive::Symbol &sym : file->symbols())
anatofuz
parents:
diff changeset
108 symtab->addLazyArchive(this, sym);
anatofuz
parents:
diff changeset
109 }
anatofuz
parents:
diff changeset
110
anatofuz
parents:
diff changeset
111 // Returns a buffer pointing to a member file containing a given symbol.
anatofuz
parents:
diff changeset
112 void ArchiveFile::addMember(const Archive::Symbol &sym) {
anatofuz
parents:
diff changeset
113 const Archive::Child &c =
anatofuz
parents:
diff changeset
114 CHECK(sym.getMember(),
anatofuz
parents:
diff changeset
115 "could not get the member for symbol " + toCOFFString(sym));
anatofuz
parents:
diff changeset
116
anatofuz
parents:
diff changeset
117 // Return an empty buffer if we have already returned the same buffer.
anatofuz
parents:
diff changeset
118 if (!seen.insert(c.getChildOffset()).second)
anatofuz
parents:
diff changeset
119 return;
anatofuz
parents:
diff changeset
120
anatofuz
parents:
diff changeset
121 driver->enqueueArchiveMember(c, sym, getName());
anatofuz
parents:
diff changeset
122 }
anatofuz
parents:
diff changeset
123
anatofuz
parents:
diff changeset
124 std::vector<MemoryBufferRef> getArchiveMembers(Archive *file) {
anatofuz
parents:
diff changeset
125 std::vector<MemoryBufferRef> v;
anatofuz
parents:
diff changeset
126 Error err = Error::success();
anatofuz
parents:
diff changeset
127 for (const Archive::Child &c : file->children(err)) {
anatofuz
parents:
diff changeset
128 MemoryBufferRef mbref =
anatofuz
parents:
diff changeset
129 CHECK(c.getMemoryBufferRef(),
anatofuz
parents:
diff changeset
130 file->getFileName() +
anatofuz
parents:
diff changeset
131 ": could not get the buffer for a child of the archive");
anatofuz
parents:
diff changeset
132 v.push_back(mbref);
anatofuz
parents:
diff changeset
133 }
anatofuz
parents:
diff changeset
134 if (err)
anatofuz
parents:
diff changeset
135 fatal(file->getFileName() +
anatofuz
parents:
diff changeset
136 ": Archive::children failed: " + toString(std::move(err)));
anatofuz
parents:
diff changeset
137 return v;
anatofuz
parents:
diff changeset
138 }
anatofuz
parents:
diff changeset
139
anatofuz
parents:
diff changeset
140 void LazyObjFile::fetch() {
anatofuz
parents:
diff changeset
141 if (mb.getBuffer().empty())
anatofuz
parents:
diff changeset
142 return;
anatofuz
parents:
diff changeset
143
anatofuz
parents:
diff changeset
144 InputFile *file;
anatofuz
parents:
diff changeset
145 if (isBitcode(mb))
anatofuz
parents:
diff changeset
146 file = make<BitcodeFile>(mb, "", 0, std::move(symbols));
anatofuz
parents:
diff changeset
147 else
anatofuz
parents:
diff changeset
148 file = make<ObjFile>(mb, std::move(symbols));
anatofuz
parents:
diff changeset
149 mb = {};
anatofuz
parents:
diff changeset
150 symtab->addFile(file);
anatofuz
parents:
diff changeset
151 }
anatofuz
parents:
diff changeset
152
anatofuz
parents:
diff changeset
153 void LazyObjFile::parse() {
anatofuz
parents:
diff changeset
154 if (isBitcode(this->mb)) {
anatofuz
parents:
diff changeset
155 // Bitcode file.
anatofuz
parents:
diff changeset
156 std::unique_ptr<lto::InputFile> obj =
anatofuz
parents:
diff changeset
157 CHECK(lto::InputFile::create(this->mb), this);
anatofuz
parents:
diff changeset
158 for (const lto::InputFile::Symbol &sym : obj->symbols()) {
anatofuz
parents:
diff changeset
159 if (!sym.isUndefined())
anatofuz
parents:
diff changeset
160 symtab->addLazyObject(this, sym.getName());
anatofuz
parents:
diff changeset
161 }
anatofuz
parents:
diff changeset
162 return;
anatofuz
parents:
diff changeset
163 }
anatofuz
parents:
diff changeset
164
anatofuz
parents:
diff changeset
165 // Native object file.
anatofuz
parents:
diff changeset
166 std::unique_ptr<Binary> coffObjPtr = CHECK(createBinary(mb), this);
anatofuz
parents:
diff changeset
167 COFFObjectFile *coffObj = cast<COFFObjectFile>(coffObjPtr.get());
anatofuz
parents:
diff changeset
168 uint32_t numSymbols = coffObj->getNumberOfSymbols();
anatofuz
parents:
diff changeset
169 for (uint32_t i = 0; i < numSymbols; ++i) {
anatofuz
parents:
diff changeset
170 COFFSymbolRef coffSym = check(coffObj->getSymbol(i));
anatofuz
parents:
diff changeset
171 if (coffSym.isUndefined() || !coffSym.isExternal() ||
anatofuz
parents:
diff changeset
172 coffSym.isWeakExternal())
anatofuz
parents:
diff changeset
173 continue;
anatofuz
parents:
diff changeset
174 StringRef name;
anatofuz
parents:
diff changeset
175 coffObj->getSymbolName(coffSym, name);
anatofuz
parents:
diff changeset
176 if (coffSym.isAbsolute() && ignoredSymbolName(name))
anatofuz
parents:
diff changeset
177 continue;
anatofuz
parents:
diff changeset
178 symtab->addLazyObject(this, name);
anatofuz
parents:
diff changeset
179 i += coffSym.getNumberOfAuxSymbols();
anatofuz
parents:
diff changeset
180 }
anatofuz
parents:
diff changeset
181 }
anatofuz
parents:
diff changeset
182
anatofuz
parents:
diff changeset
183 void ObjFile::parse() {
anatofuz
parents:
diff changeset
184 // Parse a memory buffer as a COFF file.
anatofuz
parents:
diff changeset
185 std::unique_ptr<Binary> bin = CHECK(createBinary(mb), this);
anatofuz
parents:
diff changeset
186
anatofuz
parents:
diff changeset
187 if (auto *obj = dyn_cast<COFFObjectFile>(bin.get())) {
anatofuz
parents:
diff changeset
188 bin.release();
anatofuz
parents:
diff changeset
189 coffObj.reset(obj);
anatofuz
parents:
diff changeset
190 } else {
anatofuz
parents:
diff changeset
191 fatal(toString(this) + " is not a COFF file");
anatofuz
parents:
diff changeset
192 }
anatofuz
parents:
diff changeset
193
anatofuz
parents:
diff changeset
194 // Read section and symbol tables.
anatofuz
parents:
diff changeset
195 initializeChunks();
anatofuz
parents:
diff changeset
196 initializeSymbols();
anatofuz
parents:
diff changeset
197 initializeFlags();
anatofuz
parents:
diff changeset
198 initializeDependencies();
anatofuz
parents:
diff changeset
199 }
anatofuz
parents:
diff changeset
200
anatofuz
parents:
diff changeset
201 const coff_section* ObjFile::getSection(uint32_t i) {
anatofuz
parents:
diff changeset
202 const coff_section *sec;
anatofuz
parents:
diff changeset
203 if (auto ec = coffObj->getSection(i, sec))
anatofuz
parents:
diff changeset
204 fatal("getSection failed: #" + Twine(i) + ": " + ec.message());
anatofuz
parents:
diff changeset
205 return sec;
anatofuz
parents:
diff changeset
206 }
anatofuz
parents:
diff changeset
207
anatofuz
parents:
diff changeset
208 // We set SectionChunk pointers in the SparseChunks vector to this value
anatofuz
parents:
diff changeset
209 // temporarily to mark comdat sections as having an unknown resolution. As we
anatofuz
parents:
diff changeset
210 // walk the object file's symbol table, once we visit either a leader symbol or
anatofuz
parents:
diff changeset
211 // an associative section definition together with the parent comdat's leader,
anatofuz
parents:
diff changeset
212 // we set the pointer to either nullptr (to mark the section as discarded) or a
anatofuz
parents:
diff changeset
213 // valid SectionChunk for that section.
anatofuz
parents:
diff changeset
214 static SectionChunk *const pendingComdat = reinterpret_cast<SectionChunk *>(1);
anatofuz
parents:
diff changeset
215
anatofuz
parents:
diff changeset
216 void ObjFile::initializeChunks() {
anatofuz
parents:
diff changeset
217 uint32_t numSections = coffObj->getNumberOfSections();
anatofuz
parents:
diff changeset
218 chunks.reserve(numSections);
anatofuz
parents:
diff changeset
219 sparseChunks.resize(numSections + 1);
anatofuz
parents:
diff changeset
220 for (uint32_t i = 1; i < numSections + 1; ++i) {
anatofuz
parents:
diff changeset
221 const coff_section *sec = getSection(i);
anatofuz
parents:
diff changeset
222 if (sec->Characteristics & IMAGE_SCN_LNK_COMDAT)
anatofuz
parents:
diff changeset
223 sparseChunks[i] = pendingComdat;
anatofuz
parents:
diff changeset
224 else
anatofuz
parents:
diff changeset
225 sparseChunks[i] = readSection(i, nullptr, "");
anatofuz
parents:
diff changeset
226 }
anatofuz
parents:
diff changeset
227 }
anatofuz
parents:
diff changeset
228
anatofuz
parents:
diff changeset
229 SectionChunk *ObjFile::readSection(uint32_t sectionNumber,
anatofuz
parents:
diff changeset
230 const coff_aux_section_definition *def,
anatofuz
parents:
diff changeset
231 StringRef leaderName) {
anatofuz
parents:
diff changeset
232 const coff_section *sec = getSection(sectionNumber);
anatofuz
parents:
diff changeset
233
anatofuz
parents:
diff changeset
234 StringRef name;
anatofuz
parents:
diff changeset
235 if (Expected<StringRef> e = coffObj->getSectionName(sec))
anatofuz
parents:
diff changeset
236 name = *e;
anatofuz
parents:
diff changeset
237 else
anatofuz
parents:
diff changeset
238 fatal("getSectionName failed: #" + Twine(sectionNumber) + ": " +
anatofuz
parents:
diff changeset
239 toString(e.takeError()));
anatofuz
parents:
diff changeset
240
anatofuz
parents:
diff changeset
241 if (name == ".drectve") {
anatofuz
parents:
diff changeset
242 ArrayRef<uint8_t> data;
anatofuz
parents:
diff changeset
243 cantFail(coffObj->getSectionContents(sec, data));
anatofuz
parents:
diff changeset
244 directives = StringRef((const char *)data.data(), data.size());
anatofuz
parents:
diff changeset
245 return nullptr;
anatofuz
parents:
diff changeset
246 }
anatofuz
parents:
diff changeset
247
anatofuz
parents:
diff changeset
248 if (name == ".llvm_addrsig") {
anatofuz
parents:
diff changeset
249 addrsigSec = sec;
anatofuz
parents:
diff changeset
250 return nullptr;
anatofuz
parents:
diff changeset
251 }
anatofuz
parents:
diff changeset
252
anatofuz
parents:
diff changeset
253 // Object files may have DWARF debug info or MS CodeView debug info
anatofuz
parents:
diff changeset
254 // (or both).
anatofuz
parents:
diff changeset
255 //
anatofuz
parents:
diff changeset
256 // DWARF sections don't need any special handling from the perspective
anatofuz
parents:
diff changeset
257 // of the linker; they are just a data section containing relocations.
anatofuz
parents:
diff changeset
258 // We can just link them to complete debug info.
anatofuz
parents:
diff changeset
259 //
anatofuz
parents:
diff changeset
260 // CodeView needs linker support. We need to interpret debug info,
anatofuz
parents:
diff changeset
261 // and then write it to a separate .pdb file.
anatofuz
parents:
diff changeset
262
anatofuz
parents:
diff changeset
263 // Ignore DWARF debug info unless /debug is given.
anatofuz
parents:
diff changeset
264 if (!config->debug && name.startswith(".debug_"))
anatofuz
parents:
diff changeset
265 return nullptr;
anatofuz
parents:
diff changeset
266
anatofuz
parents:
diff changeset
267 if (sec->Characteristics & llvm::COFF::IMAGE_SCN_LNK_REMOVE)
anatofuz
parents:
diff changeset
268 return nullptr;
anatofuz
parents:
diff changeset
269 auto *c = make<SectionChunk>(this, sec);
anatofuz
parents:
diff changeset
270 if (def)
anatofuz
parents:
diff changeset
271 c->checksum = def->CheckSum;
anatofuz
parents:
diff changeset
272
anatofuz
parents:
diff changeset
273 // CodeView sections are stored to a different vector because they are not
anatofuz
parents:
diff changeset
274 // linked in the regular manner.
anatofuz
parents:
diff changeset
275 if (c->isCodeView())
anatofuz
parents:
diff changeset
276 debugChunks.push_back(c);
anatofuz
parents:
diff changeset
277 else if (name == ".gfids$y")
anatofuz
parents:
diff changeset
278 guardFidChunks.push_back(c);
anatofuz
parents:
diff changeset
279 else if (name == ".gljmp$y")
anatofuz
parents:
diff changeset
280 guardLJmpChunks.push_back(c);
anatofuz
parents:
diff changeset
281 else if (name == ".sxdata")
anatofuz
parents:
diff changeset
282 sXDataChunks.push_back(c);
anatofuz
parents:
diff changeset
283 else if (config->tailMerge && sec->NumberOfRelocations == 0 &&
anatofuz
parents:
diff changeset
284 name == ".rdata" && leaderName.startswith("??_C@"))
anatofuz
parents:
diff changeset
285 // COFF sections that look like string literal sections (i.e. no
anatofuz
parents:
diff changeset
286 // relocations, in .rdata, leader symbol name matches the MSVC name mangling
anatofuz
parents:
diff changeset
287 // for string literals) are subject to string tail merging.
anatofuz
parents:
diff changeset
288 MergeChunk::addSection(c);
anatofuz
parents:
diff changeset
289 else if (name == ".rsrc" || name.startswith(".rsrc$"))
anatofuz
parents:
diff changeset
290 resourceChunks.push_back(c);
anatofuz
parents:
diff changeset
291 else
anatofuz
parents:
diff changeset
292 chunks.push_back(c);
anatofuz
parents:
diff changeset
293
anatofuz
parents:
diff changeset
294 return c;
anatofuz
parents:
diff changeset
295 }
anatofuz
parents:
diff changeset
296
anatofuz
parents:
diff changeset
297 void ObjFile::includeResourceChunks() {
anatofuz
parents:
diff changeset
298 chunks.insert(chunks.end(), resourceChunks.begin(), resourceChunks.end());
anatofuz
parents:
diff changeset
299 }
anatofuz
parents:
diff changeset
300
anatofuz
parents:
diff changeset
301 void ObjFile::readAssociativeDefinition(
anatofuz
parents:
diff changeset
302 COFFSymbolRef sym, const coff_aux_section_definition *def) {
anatofuz
parents:
diff changeset
303 readAssociativeDefinition(sym, def, def->getNumber(sym.isBigObj()));
anatofuz
parents:
diff changeset
304 }
anatofuz
parents:
diff changeset
305
anatofuz
parents:
diff changeset
306 void ObjFile::readAssociativeDefinition(COFFSymbolRef sym,
anatofuz
parents:
diff changeset
307 const coff_aux_section_definition *def,
anatofuz
parents:
diff changeset
308 uint32_t parentIndex) {
anatofuz
parents:
diff changeset
309 SectionChunk *parent = sparseChunks[parentIndex];
anatofuz
parents:
diff changeset
310 int32_t sectionNumber = sym.getSectionNumber();
anatofuz
parents:
diff changeset
311
anatofuz
parents:
diff changeset
312 auto diag = [&]() {
anatofuz
parents:
diff changeset
313 StringRef name, parentName;
anatofuz
parents:
diff changeset
314 coffObj->getSymbolName(sym, name);
anatofuz
parents:
diff changeset
315
anatofuz
parents:
diff changeset
316 const coff_section *parentSec = getSection(parentIndex);
anatofuz
parents:
diff changeset
317 if (Expected<StringRef> e = coffObj->getSectionName(parentSec))
anatofuz
parents:
diff changeset
318 parentName = *e;
anatofuz
parents:
diff changeset
319 error(toString(this) + ": associative comdat " + name + " (sec " +
anatofuz
parents:
diff changeset
320 Twine(sectionNumber) + ") has invalid reference to section " +
anatofuz
parents:
diff changeset
321 parentName + " (sec " + Twine(parentIndex) + ")");
anatofuz
parents:
diff changeset
322 };
anatofuz
parents:
diff changeset
323
anatofuz
parents:
diff changeset
324 if (parent == pendingComdat) {
anatofuz
parents:
diff changeset
325 // This can happen if an associative comdat refers to another associative
anatofuz
parents:
diff changeset
326 // comdat that appears after it (invalid per COFF spec) or to a section
anatofuz
parents:
diff changeset
327 // without any symbols.
anatofuz
parents:
diff changeset
328 diag();
anatofuz
parents:
diff changeset
329 return;
anatofuz
parents:
diff changeset
330 }
anatofuz
parents:
diff changeset
331
anatofuz
parents:
diff changeset
332 // Check whether the parent is prevailing. If it is, so are we, and we read
anatofuz
parents:
diff changeset
333 // the section; otherwise mark it as discarded.
anatofuz
parents:
diff changeset
334 if (parent) {
anatofuz
parents:
diff changeset
335 SectionChunk *c = readSection(sectionNumber, def, "");
anatofuz
parents:
diff changeset
336 sparseChunks[sectionNumber] = c;
anatofuz
parents:
diff changeset
337 if (c) {
anatofuz
parents:
diff changeset
338 c->selection = IMAGE_COMDAT_SELECT_ASSOCIATIVE;
anatofuz
parents:
diff changeset
339 parent->addAssociative(c);
anatofuz
parents:
diff changeset
340 }
anatofuz
parents:
diff changeset
341 } else {
anatofuz
parents:
diff changeset
342 sparseChunks[sectionNumber] = nullptr;
anatofuz
parents:
diff changeset
343 }
anatofuz
parents:
diff changeset
344 }
anatofuz
parents:
diff changeset
345
anatofuz
parents:
diff changeset
346 void ObjFile::recordPrevailingSymbolForMingw(
anatofuz
parents:
diff changeset
347 COFFSymbolRef sym, DenseMap<StringRef, uint32_t> &prevailingSectionMap) {
anatofuz
parents:
diff changeset
348 // For comdat symbols in executable sections, where this is the copy
anatofuz
parents:
diff changeset
349 // of the section chunk we actually include instead of discarding it,
anatofuz
parents:
diff changeset
350 // add the symbol to a map to allow using it for implicitly
anatofuz
parents:
diff changeset
351 // associating .[px]data$<func> sections to it.
anatofuz
parents:
diff changeset
352 int32_t sectionNumber = sym.getSectionNumber();
anatofuz
parents:
diff changeset
353 SectionChunk *sc = sparseChunks[sectionNumber];
anatofuz
parents:
diff changeset
354 if (sc && sc->getOutputCharacteristics() & IMAGE_SCN_MEM_EXECUTE) {
anatofuz
parents:
diff changeset
355 StringRef name;
anatofuz
parents:
diff changeset
356 coffObj->getSymbolName(sym, name);
anatofuz
parents:
diff changeset
357 if (getMachineType() == I386)
anatofuz
parents:
diff changeset
358 name.consume_front("_");
anatofuz
parents:
diff changeset
359 prevailingSectionMap[name] = sectionNumber;
anatofuz
parents:
diff changeset
360 }
anatofuz
parents:
diff changeset
361 }
anatofuz
parents:
diff changeset
362
anatofuz
parents:
diff changeset
363 void ObjFile::maybeAssociateSEHForMingw(
anatofuz
parents:
diff changeset
364 COFFSymbolRef sym, const coff_aux_section_definition *def,
anatofuz
parents:
diff changeset
365 const DenseMap<StringRef, uint32_t> &prevailingSectionMap) {
anatofuz
parents:
diff changeset
366 StringRef name;
anatofuz
parents:
diff changeset
367 coffObj->getSymbolName(sym, name);
anatofuz
parents:
diff changeset
368 if (name.consume_front(".pdata$") || name.consume_front(".xdata$") ||
anatofuz
parents:
diff changeset
369 name.consume_front(".eh_frame$")) {
anatofuz
parents:
diff changeset
370 // For MinGW, treat .[px]data$<func> and .eh_frame$<func> as implicitly
anatofuz
parents:
diff changeset
371 // associative to the symbol <func>.
anatofuz
parents:
diff changeset
372 auto parentSym = prevailingSectionMap.find(name);
anatofuz
parents:
diff changeset
373 if (parentSym != prevailingSectionMap.end())
anatofuz
parents:
diff changeset
374 readAssociativeDefinition(sym, def, parentSym->second);
anatofuz
parents:
diff changeset
375 }
anatofuz
parents:
diff changeset
376 }
anatofuz
parents:
diff changeset
377
anatofuz
parents:
diff changeset
378 Symbol *ObjFile::createRegular(COFFSymbolRef sym) {
anatofuz
parents:
diff changeset
379 SectionChunk *sc = sparseChunks[sym.getSectionNumber()];
anatofuz
parents:
diff changeset
380 if (sym.isExternal()) {
anatofuz
parents:
diff changeset
381 StringRef name;
anatofuz
parents:
diff changeset
382 coffObj->getSymbolName(sym, name);
anatofuz
parents:
diff changeset
383 if (sc)
anatofuz
parents:
diff changeset
384 return symtab->addRegular(this, name, sym.getGeneric(), sc,
anatofuz
parents:
diff changeset
385 sym.getValue());
anatofuz
parents:
diff changeset
386 // For MinGW symbols named .weak.* that point to a discarded section,
anatofuz
parents:
diff changeset
387 // don't create an Undefined symbol. If nothing ever refers to the symbol,
anatofuz
parents:
diff changeset
388 // everything should be fine. If something actually refers to the symbol
anatofuz
parents:
diff changeset
389 // (e.g. the undefined weak alias), linking will fail due to undefined
anatofuz
parents:
diff changeset
390 // references at the end.
anatofuz
parents:
diff changeset
391 if (config->mingw && name.startswith(".weak."))
anatofuz
parents:
diff changeset
392 return nullptr;
anatofuz
parents:
diff changeset
393 return symtab->addUndefined(name, this, false);
anatofuz
parents:
diff changeset
394 }
anatofuz
parents:
diff changeset
395 if (sc)
anatofuz
parents:
diff changeset
396 return make<DefinedRegular>(this, /*Name*/ "", /*IsCOMDAT*/ false,
anatofuz
parents:
diff changeset
397 /*IsExternal*/ false, sym.getGeneric(), sc);
anatofuz
parents:
diff changeset
398 return nullptr;
anatofuz
parents:
diff changeset
399 }
anatofuz
parents:
diff changeset
400
anatofuz
parents:
diff changeset
401 void ObjFile::initializeSymbols() {
anatofuz
parents:
diff changeset
402 uint32_t numSymbols = coffObj->getNumberOfSymbols();
anatofuz
parents:
diff changeset
403 symbols.resize(numSymbols);
anatofuz
parents:
diff changeset
404
anatofuz
parents:
diff changeset
405 SmallVector<std::pair<Symbol *, uint32_t>, 8> weakAliases;
anatofuz
parents:
diff changeset
406 std::vector<uint32_t> pendingIndexes;
anatofuz
parents:
diff changeset
407 pendingIndexes.reserve(numSymbols);
anatofuz
parents:
diff changeset
408
anatofuz
parents:
diff changeset
409 DenseMap<StringRef, uint32_t> prevailingSectionMap;
anatofuz
parents:
diff changeset
410 std::vector<const coff_aux_section_definition *> comdatDefs(
anatofuz
parents:
diff changeset
411 coffObj->getNumberOfSections() + 1);
anatofuz
parents:
diff changeset
412
anatofuz
parents:
diff changeset
413 for (uint32_t i = 0; i < numSymbols; ++i) {
anatofuz
parents:
diff changeset
414 COFFSymbolRef coffSym = check(coffObj->getSymbol(i));
anatofuz
parents:
diff changeset
415 bool prevailingComdat;
anatofuz
parents:
diff changeset
416 if (coffSym.isUndefined()) {
anatofuz
parents:
diff changeset
417 symbols[i] = createUndefined(coffSym);
anatofuz
parents:
diff changeset
418 } else if (coffSym.isWeakExternal()) {
anatofuz
parents:
diff changeset
419 symbols[i] = createUndefined(coffSym);
anatofuz
parents:
diff changeset
420 uint32_t tagIndex = coffSym.getAux<coff_aux_weak_external>()->TagIndex;
anatofuz
parents:
diff changeset
421 weakAliases.emplace_back(symbols[i], tagIndex);
anatofuz
parents:
diff changeset
422 } else if (Optional<Symbol *> optSym =
anatofuz
parents:
diff changeset
423 createDefined(coffSym, comdatDefs, prevailingComdat)) {
anatofuz
parents:
diff changeset
424 symbols[i] = *optSym;
anatofuz
parents:
diff changeset
425 if (config->mingw && prevailingComdat)
anatofuz
parents:
diff changeset
426 recordPrevailingSymbolForMingw(coffSym, prevailingSectionMap);
anatofuz
parents:
diff changeset
427 } else {
anatofuz
parents:
diff changeset
428 // createDefined() returns None if a symbol belongs to a section that
anatofuz
parents:
diff changeset
429 // was pending at the point when the symbol was read. This can happen in
anatofuz
parents:
diff changeset
430 // two cases:
anatofuz
parents:
diff changeset
431 // 1) section definition symbol for a comdat leader;
anatofuz
parents:
diff changeset
432 // 2) symbol belongs to a comdat section associated with another section.
anatofuz
parents:
diff changeset
433 // In both of these cases, we can expect the section to be resolved by
anatofuz
parents:
diff changeset
434 // the time we finish visiting the remaining symbols in the symbol
anatofuz
parents:
diff changeset
435 // table. So we postpone the handling of this symbol until that time.
anatofuz
parents:
diff changeset
436 pendingIndexes.push_back(i);
anatofuz
parents:
diff changeset
437 }
anatofuz
parents:
diff changeset
438 i += coffSym.getNumberOfAuxSymbols();
anatofuz
parents:
diff changeset
439 }
anatofuz
parents:
diff changeset
440
anatofuz
parents:
diff changeset
441 for (uint32_t i : pendingIndexes) {
anatofuz
parents:
diff changeset
442 COFFSymbolRef sym = check(coffObj->getSymbol(i));
anatofuz
parents:
diff changeset
443 if (const coff_aux_section_definition *def = sym.getSectionDefinition()) {
anatofuz
parents:
diff changeset
444 if (def->Selection == IMAGE_COMDAT_SELECT_ASSOCIATIVE)
anatofuz
parents:
diff changeset
445 readAssociativeDefinition(sym, def);
anatofuz
parents:
diff changeset
446 else if (config->mingw)
anatofuz
parents:
diff changeset
447 maybeAssociateSEHForMingw(sym, def, prevailingSectionMap);
anatofuz
parents:
diff changeset
448 }
anatofuz
parents:
diff changeset
449 if (sparseChunks[sym.getSectionNumber()] == pendingComdat) {
anatofuz
parents:
diff changeset
450 StringRef name;
anatofuz
parents:
diff changeset
451 coffObj->getSymbolName(sym, name);
anatofuz
parents:
diff changeset
452 log("comdat section " + name +
anatofuz
parents:
diff changeset
453 " without leader and unassociated, discarding");
anatofuz
parents:
diff changeset
454 continue;
anatofuz
parents:
diff changeset
455 }
anatofuz
parents:
diff changeset
456 symbols[i] = createRegular(sym);
anatofuz
parents:
diff changeset
457 }
anatofuz
parents:
diff changeset
458
anatofuz
parents:
diff changeset
459 for (auto &kv : weakAliases) {
anatofuz
parents:
diff changeset
460 Symbol *sym = kv.first;
anatofuz
parents:
diff changeset
461 uint32_t idx = kv.second;
anatofuz
parents:
diff changeset
462 checkAndSetWeakAlias(symtab, this, sym, symbols[idx]);
anatofuz
parents:
diff changeset
463 }
anatofuz
parents:
diff changeset
464 }
anatofuz
parents:
diff changeset
465
anatofuz
parents:
diff changeset
466 Symbol *ObjFile::createUndefined(COFFSymbolRef sym) {
anatofuz
parents:
diff changeset
467 StringRef name;
anatofuz
parents:
diff changeset
468 coffObj->getSymbolName(sym, name);
anatofuz
parents:
diff changeset
469 return symtab->addUndefined(name, this, sym.isWeakExternal());
anatofuz
parents:
diff changeset
470 }
anatofuz
parents:
diff changeset
471
anatofuz
parents:
diff changeset
472 void ObjFile::handleComdatSelection(COFFSymbolRef sym, COMDATType &selection,
anatofuz
parents:
diff changeset
473 bool &prevailing, DefinedRegular *leader) {
anatofuz
parents:
diff changeset
474 if (prevailing)
anatofuz
parents:
diff changeset
475 return;
anatofuz
parents:
diff changeset
476 // There's already an existing comdat for this symbol: `Leader`.
anatofuz
parents:
diff changeset
477 // Use the comdats's selection field to determine if the new
anatofuz
parents:
diff changeset
478 // symbol in `Sym` should be discarded, produce a duplicate symbol
anatofuz
parents:
diff changeset
479 // error, etc.
anatofuz
parents:
diff changeset
480
anatofuz
parents:
diff changeset
481 SectionChunk *leaderChunk = nullptr;
anatofuz
parents:
diff changeset
482 COMDATType leaderSelection = IMAGE_COMDAT_SELECT_ANY;
anatofuz
parents:
diff changeset
483
anatofuz
parents:
diff changeset
484 if (leader->data) {
anatofuz
parents:
diff changeset
485 leaderChunk = leader->getChunk();
anatofuz
parents:
diff changeset
486 leaderSelection = leaderChunk->selection;
anatofuz
parents:
diff changeset
487 } else {
anatofuz
parents:
diff changeset
488 // FIXME: comdats from LTO files don't know their selection; treat them
anatofuz
parents:
diff changeset
489 // as "any".
anatofuz
parents:
diff changeset
490 selection = leaderSelection;
anatofuz
parents:
diff changeset
491 }
anatofuz
parents:
diff changeset
492
anatofuz
parents:
diff changeset
493 if ((selection == IMAGE_COMDAT_SELECT_ANY &&
anatofuz
parents:
diff changeset
494 leaderSelection == IMAGE_COMDAT_SELECT_LARGEST) ||
anatofuz
parents:
diff changeset
495 (selection == IMAGE_COMDAT_SELECT_LARGEST &&
anatofuz
parents:
diff changeset
496 leaderSelection == IMAGE_COMDAT_SELECT_ANY)) {
anatofuz
parents:
diff changeset
497 // cl.exe picks "any" for vftables when building with /GR- and
anatofuz
parents:
diff changeset
498 // "largest" when building with /GR. To be able to link object files
anatofuz
parents:
diff changeset
499 // compiled with each flag, "any" and "largest" are merged as "largest".
anatofuz
parents:
diff changeset
500 leaderSelection = selection = IMAGE_COMDAT_SELECT_LARGEST;
anatofuz
parents:
diff changeset
501 }
anatofuz
parents:
diff changeset
502
anatofuz
parents:
diff changeset
503 // GCCs __declspec(selectany) doesn't actually pick "any" but "same size as".
anatofuz
parents:
diff changeset
504 // Clang on the other hand picks "any". To be able to link two object files
anatofuz
parents:
diff changeset
505 // with a __declspec(selectany) declaration, one compiled with gcc and the
anatofuz
parents:
diff changeset
506 // other with clang, we merge them as proper "same size as"
anatofuz
parents:
diff changeset
507 if (config->mingw && ((selection == IMAGE_COMDAT_SELECT_ANY &&
anatofuz
parents:
diff changeset
508 leaderSelection == IMAGE_COMDAT_SELECT_SAME_SIZE) ||
anatofuz
parents:
diff changeset
509 (selection == IMAGE_COMDAT_SELECT_SAME_SIZE &&
anatofuz
parents:
diff changeset
510 leaderSelection == IMAGE_COMDAT_SELECT_ANY))) {
anatofuz
parents:
diff changeset
511 leaderSelection = selection = IMAGE_COMDAT_SELECT_SAME_SIZE;
anatofuz
parents:
diff changeset
512 }
anatofuz
parents:
diff changeset
513
anatofuz
parents:
diff changeset
514 // Other than that, comdat selections must match. This is a bit more
anatofuz
parents:
diff changeset
515 // strict than link.exe which allows merging "any" and "largest" if "any"
anatofuz
parents:
diff changeset
516 // is the first symbol the linker sees, and it allows merging "largest"
anatofuz
parents:
diff changeset
517 // with everything (!) if "largest" is the first symbol the linker sees.
anatofuz
parents:
diff changeset
518 // Making this symmetric independent of which selection is seen first
anatofuz
parents:
diff changeset
519 // seems better though.
anatofuz
parents:
diff changeset
520 // (This behavior matches ModuleLinker::getComdatResult().)
anatofuz
parents:
diff changeset
521 if (selection != leaderSelection) {
anatofuz
parents:
diff changeset
522 log(("conflicting comdat type for " + toString(*leader) + ": " +
anatofuz
parents:
diff changeset
523 Twine((int)leaderSelection) + " in " + toString(leader->getFile()) +
anatofuz
parents:
diff changeset
524 " and " + Twine((int)selection) + " in " + toString(this))
anatofuz
parents:
diff changeset
525 .str());
anatofuz
parents:
diff changeset
526 symtab->reportDuplicate(leader, this);
anatofuz
parents:
diff changeset
527 return;
anatofuz
parents:
diff changeset
528 }
anatofuz
parents:
diff changeset
529
anatofuz
parents:
diff changeset
530 switch (selection) {
anatofuz
parents:
diff changeset
531 case IMAGE_COMDAT_SELECT_NODUPLICATES:
anatofuz
parents:
diff changeset
532 symtab->reportDuplicate(leader, this);
anatofuz
parents:
diff changeset
533 break;
anatofuz
parents:
diff changeset
534
anatofuz
parents:
diff changeset
535 case IMAGE_COMDAT_SELECT_ANY:
anatofuz
parents:
diff changeset
536 // Nothing to do.
anatofuz
parents:
diff changeset
537 break;
anatofuz
parents:
diff changeset
538
anatofuz
parents:
diff changeset
539 case IMAGE_COMDAT_SELECT_SAME_SIZE:
anatofuz
parents:
diff changeset
540 if (leaderChunk->getSize() != getSection(sym)->SizeOfRawData)
anatofuz
parents:
diff changeset
541 symtab->reportDuplicate(leader, this);
anatofuz
parents:
diff changeset
542 break;
anatofuz
parents:
diff changeset
543
anatofuz
parents:
diff changeset
544 case IMAGE_COMDAT_SELECT_EXACT_MATCH: {
anatofuz
parents:
diff changeset
545 SectionChunk newChunk(this, getSection(sym));
anatofuz
parents:
diff changeset
546 // link.exe only compares section contents here and doesn't complain
anatofuz
parents:
diff changeset
547 // if the two comdat sections have e.g. different alignment.
anatofuz
parents:
diff changeset
548 // Match that.
anatofuz
parents:
diff changeset
549 if (leaderChunk->getContents() != newChunk.getContents())
anatofuz
parents:
diff changeset
550 symtab->reportDuplicate(leader, this, &newChunk, sym.getValue());
anatofuz
parents:
diff changeset
551 break;
anatofuz
parents:
diff changeset
552 }
anatofuz
parents:
diff changeset
553
anatofuz
parents:
diff changeset
554 case IMAGE_COMDAT_SELECT_ASSOCIATIVE:
anatofuz
parents:
diff changeset
555 // createDefined() is never called for IMAGE_COMDAT_SELECT_ASSOCIATIVE.
anatofuz
parents:
diff changeset
556 // (This means lld-link doesn't produce duplicate symbol errors for
anatofuz
parents:
diff changeset
557 // associative comdats while link.exe does, but associate comdats
anatofuz
parents:
diff changeset
558 // are never extern in practice.)
anatofuz
parents:
diff changeset
559 llvm_unreachable("createDefined not called for associative comdats");
anatofuz
parents:
diff changeset
560
anatofuz
parents:
diff changeset
561 case IMAGE_COMDAT_SELECT_LARGEST:
anatofuz
parents:
diff changeset
562 if (leaderChunk->getSize() < getSection(sym)->SizeOfRawData) {
anatofuz
parents:
diff changeset
563 // Replace the existing comdat symbol with the new one.
anatofuz
parents:
diff changeset
564 StringRef name;
anatofuz
parents:
diff changeset
565 coffObj->getSymbolName(sym, name);
anatofuz
parents:
diff changeset
566 // FIXME: This is incorrect: With /opt:noref, the previous sections
anatofuz
parents:
diff changeset
567 // make it into the final executable as well. Correct handling would
anatofuz
parents:
diff changeset
568 // be to undo reading of the whole old section that's being replaced,
anatofuz
parents:
diff changeset
569 // or doing one pass that determines what the final largest comdat
anatofuz
parents:
diff changeset
570 // is for all IMAGE_COMDAT_SELECT_LARGEST comdats and then reading
anatofuz
parents:
diff changeset
571 // only the largest one.
anatofuz
parents:
diff changeset
572 replaceSymbol<DefinedRegular>(leader, this, name, /*IsCOMDAT*/ true,
anatofuz
parents:
diff changeset
573 /*IsExternal*/ true, sym.getGeneric(),
anatofuz
parents:
diff changeset
574 nullptr);
anatofuz
parents:
diff changeset
575 prevailing = true;
anatofuz
parents:
diff changeset
576 }
anatofuz
parents:
diff changeset
577 break;
anatofuz
parents:
diff changeset
578
anatofuz
parents:
diff changeset
579 case IMAGE_COMDAT_SELECT_NEWEST:
anatofuz
parents:
diff changeset
580 llvm_unreachable("should have been rejected earlier");
anatofuz
parents:
diff changeset
581 }
anatofuz
parents:
diff changeset
582 }
anatofuz
parents:
diff changeset
583
anatofuz
parents:
diff changeset
584 Optional<Symbol *> ObjFile::createDefined(
anatofuz
parents:
diff changeset
585 COFFSymbolRef sym,
anatofuz
parents:
diff changeset
586 std::vector<const coff_aux_section_definition *> &comdatDefs,
anatofuz
parents:
diff changeset
587 bool &prevailing) {
anatofuz
parents:
diff changeset
588 prevailing = false;
anatofuz
parents:
diff changeset
589 auto getName = [&]() {
anatofuz
parents:
diff changeset
590 StringRef s;
anatofuz
parents:
diff changeset
591 coffObj->getSymbolName(sym, s);
anatofuz
parents:
diff changeset
592 return s;
anatofuz
parents:
diff changeset
593 };
anatofuz
parents:
diff changeset
594
anatofuz
parents:
diff changeset
595 if (sym.isCommon()) {
anatofuz
parents:
diff changeset
596 auto *c = make<CommonChunk>(sym);
anatofuz
parents:
diff changeset
597 chunks.push_back(c);
anatofuz
parents:
diff changeset
598 return symtab->addCommon(this, getName(), sym.getValue(), sym.getGeneric(),
anatofuz
parents:
diff changeset
599 c);
anatofuz
parents:
diff changeset
600 }
anatofuz
parents:
diff changeset
601
anatofuz
parents:
diff changeset
602 if (sym.isAbsolute()) {
anatofuz
parents:
diff changeset
603 StringRef name = getName();
anatofuz
parents:
diff changeset
604
anatofuz
parents:
diff changeset
605 if (name == "@feat.00")
anatofuz
parents:
diff changeset
606 feat00Flags = sym.getValue();
anatofuz
parents:
diff changeset
607 // Skip special symbols.
anatofuz
parents:
diff changeset
608 if (ignoredSymbolName(name))
anatofuz
parents:
diff changeset
609 return nullptr;
anatofuz
parents:
diff changeset
610
anatofuz
parents:
diff changeset
611 if (sym.isExternal())
anatofuz
parents:
diff changeset
612 return symtab->addAbsolute(name, sym);
anatofuz
parents:
diff changeset
613 return make<DefinedAbsolute>(name, sym);
anatofuz
parents:
diff changeset
614 }
anatofuz
parents:
diff changeset
615
anatofuz
parents:
diff changeset
616 int32_t sectionNumber = sym.getSectionNumber();
anatofuz
parents:
diff changeset
617 if (sectionNumber == llvm::COFF::IMAGE_SYM_DEBUG)
anatofuz
parents:
diff changeset
618 return nullptr;
anatofuz
parents:
diff changeset
619
anatofuz
parents:
diff changeset
620 if (llvm::COFF::isReservedSectionNumber(sectionNumber))
anatofuz
parents:
diff changeset
621 fatal(toString(this) + ": " + getName() +
anatofuz
parents:
diff changeset
622 " should not refer to special section " + Twine(sectionNumber));
anatofuz
parents:
diff changeset
623
anatofuz
parents:
diff changeset
624 if ((uint32_t)sectionNumber >= sparseChunks.size())
anatofuz
parents:
diff changeset
625 fatal(toString(this) + ": " + getName() +
anatofuz
parents:
diff changeset
626 " should not refer to non-existent section " + Twine(sectionNumber));
anatofuz
parents:
diff changeset
627
anatofuz
parents:
diff changeset
628 // Comdat handling.
anatofuz
parents:
diff changeset
629 // A comdat symbol consists of two symbol table entries.
anatofuz
parents:
diff changeset
630 // The first symbol entry has the name of the section (e.g. .text), fixed
anatofuz
parents:
diff changeset
631 // values for the other fields, and one auxiliary record.
anatofuz
parents:
diff changeset
632 // The second symbol entry has the name of the comdat symbol, called the
anatofuz
parents:
diff changeset
633 // "comdat leader".
anatofuz
parents:
diff changeset
634 // When this function is called for the first symbol entry of a comdat,
anatofuz
parents:
diff changeset
635 // it sets comdatDefs and returns None, and when it's called for the second
anatofuz
parents:
diff changeset
636 // symbol entry it reads comdatDefs and then sets it back to nullptr.
anatofuz
parents:
diff changeset
637
anatofuz
parents:
diff changeset
638 // Handle comdat leader.
anatofuz
parents:
diff changeset
639 if (const coff_aux_section_definition *def = comdatDefs[sectionNumber]) {
anatofuz
parents:
diff changeset
640 comdatDefs[sectionNumber] = nullptr;
anatofuz
parents:
diff changeset
641 DefinedRegular *leader;
anatofuz
parents:
diff changeset
642
anatofuz
parents:
diff changeset
643 if (sym.isExternal()) {
anatofuz
parents:
diff changeset
644 std::tie(leader, prevailing) =
anatofuz
parents:
diff changeset
645 symtab->addComdat(this, getName(), sym.getGeneric());
anatofuz
parents:
diff changeset
646 } else {
anatofuz
parents:
diff changeset
647 leader = make<DefinedRegular>(this, /*Name*/ "", /*IsCOMDAT*/ false,
anatofuz
parents:
diff changeset
648 /*IsExternal*/ false, sym.getGeneric());
anatofuz
parents:
diff changeset
649 prevailing = true;
anatofuz
parents:
diff changeset
650 }
anatofuz
parents:
diff changeset
651
anatofuz
parents:
diff changeset
652 if (def->Selection < (int)IMAGE_COMDAT_SELECT_NODUPLICATES ||
anatofuz
parents:
diff changeset
653 // Intentionally ends at IMAGE_COMDAT_SELECT_LARGEST: link.exe
anatofuz
parents:
diff changeset
654 // doesn't understand IMAGE_COMDAT_SELECT_NEWEST either.
anatofuz
parents:
diff changeset
655 def->Selection > (int)IMAGE_COMDAT_SELECT_LARGEST) {
anatofuz
parents:
diff changeset
656 fatal("unknown comdat type " + std::to_string((int)def->Selection) +
anatofuz
parents:
diff changeset
657 " for " + getName() + " in " + toString(this));
anatofuz
parents:
diff changeset
658 }
anatofuz
parents:
diff changeset
659 COMDATType selection = (COMDATType)def->Selection;
anatofuz
parents:
diff changeset
660
anatofuz
parents:
diff changeset
661 if (leader->isCOMDAT)
anatofuz
parents:
diff changeset
662 handleComdatSelection(sym, selection, prevailing, leader);
anatofuz
parents:
diff changeset
663
anatofuz
parents:
diff changeset
664 if (prevailing) {
anatofuz
parents:
diff changeset
665 SectionChunk *c = readSection(sectionNumber, def, getName());
anatofuz
parents:
diff changeset
666 sparseChunks[sectionNumber] = c;
anatofuz
parents:
diff changeset
667 c->sym = cast<DefinedRegular>(leader);
anatofuz
parents:
diff changeset
668 c->selection = selection;
anatofuz
parents:
diff changeset
669 cast<DefinedRegular>(leader)->data = &c->repl;
anatofuz
parents:
diff changeset
670 } else {
anatofuz
parents:
diff changeset
671 sparseChunks[sectionNumber] = nullptr;
anatofuz
parents:
diff changeset
672 }
anatofuz
parents:
diff changeset
673 return leader;
anatofuz
parents:
diff changeset
674 }
anatofuz
parents:
diff changeset
675
anatofuz
parents:
diff changeset
676 // Prepare to handle the comdat leader symbol by setting the section's
anatofuz
parents:
diff changeset
677 // ComdatDefs pointer if we encounter a non-associative comdat.
anatofuz
parents:
diff changeset
678 if (sparseChunks[sectionNumber] == pendingComdat) {
anatofuz
parents:
diff changeset
679 if (const coff_aux_section_definition *def = sym.getSectionDefinition()) {
anatofuz
parents:
diff changeset
680 if (def->Selection != IMAGE_COMDAT_SELECT_ASSOCIATIVE)
anatofuz
parents:
diff changeset
681 comdatDefs[sectionNumber] = def;
anatofuz
parents:
diff changeset
682 }
anatofuz
parents:
diff changeset
683 return None;
anatofuz
parents:
diff changeset
684 }
anatofuz
parents:
diff changeset
685
anatofuz
parents:
diff changeset
686 return createRegular(sym);
anatofuz
parents:
diff changeset
687 }
anatofuz
parents:
diff changeset
688
anatofuz
parents:
diff changeset
689 MachineTypes ObjFile::getMachineType() {
anatofuz
parents:
diff changeset
690 if (coffObj)
anatofuz
parents:
diff changeset
691 return static_cast<MachineTypes>(coffObj->getMachine());
anatofuz
parents:
diff changeset
692 return IMAGE_FILE_MACHINE_UNKNOWN;
anatofuz
parents:
diff changeset
693 }
anatofuz
parents:
diff changeset
694
anatofuz
parents:
diff changeset
695 ArrayRef<uint8_t> ObjFile::getDebugSection(StringRef secName) {
anatofuz
parents:
diff changeset
696 if (SectionChunk *sec = SectionChunk::findByName(debugChunks, secName))
anatofuz
parents:
diff changeset
697 return sec->consumeDebugMagic();
anatofuz
parents:
diff changeset
698 return {};
anatofuz
parents:
diff changeset
699 }
anatofuz
parents:
diff changeset
700
anatofuz
parents:
diff changeset
701 // OBJ files systematically store critical information in a .debug$S stream,
anatofuz
parents:
diff changeset
702 // even if the TU was compiled with no debug info. At least two records are
anatofuz
parents:
diff changeset
703 // always there. S_OBJNAME stores a 32-bit signature, which is loaded into the
anatofuz
parents:
diff changeset
704 // PCHSignature member. S_COMPILE3 stores compile-time cmd-line flags. This is
anatofuz
parents:
diff changeset
705 // currently used to initialize the hotPatchable member.
anatofuz
parents:
diff changeset
706 void ObjFile::initializeFlags() {
anatofuz
parents:
diff changeset
707 ArrayRef<uint8_t> data = getDebugSection(".debug$S");
anatofuz
parents:
diff changeset
708 if (data.empty())
anatofuz
parents:
diff changeset
709 return;
anatofuz
parents:
diff changeset
710
anatofuz
parents:
diff changeset
711 DebugSubsectionArray subsections;
anatofuz
parents:
diff changeset
712
anatofuz
parents:
diff changeset
713 BinaryStreamReader reader(data, support::little);
anatofuz
parents:
diff changeset
714 ExitOnError exitOnErr;
anatofuz
parents:
diff changeset
715 exitOnErr(reader.readArray(subsections, data.size()));
anatofuz
parents:
diff changeset
716
anatofuz
parents:
diff changeset
717 for (const DebugSubsectionRecord &ss : subsections) {
anatofuz
parents:
diff changeset
718 if (ss.kind() != DebugSubsectionKind::Symbols)
anatofuz
parents:
diff changeset
719 continue;
anatofuz
parents:
diff changeset
720
anatofuz
parents:
diff changeset
721 unsigned offset = 0;
anatofuz
parents:
diff changeset
722
anatofuz
parents:
diff changeset
723 // Only parse the first two records. We are only looking for S_OBJNAME
anatofuz
parents:
diff changeset
724 // and S_COMPILE3, and they usually appear at the beginning of the
anatofuz
parents:
diff changeset
725 // stream.
anatofuz
parents:
diff changeset
726 for (unsigned i = 0; i < 2; ++i) {
anatofuz
parents:
diff changeset
727 Expected<CVSymbol> sym = readSymbolFromStream(ss.getRecordData(), offset);
anatofuz
parents:
diff changeset
728 if (!sym) {
anatofuz
parents:
diff changeset
729 consumeError(sym.takeError());
anatofuz
parents:
diff changeset
730 return;
anatofuz
parents:
diff changeset
731 }
anatofuz
parents:
diff changeset
732 if (sym->kind() == SymbolKind::S_COMPILE3) {
anatofuz
parents:
diff changeset
733 auto cs =
anatofuz
parents:
diff changeset
734 cantFail(SymbolDeserializer::deserializeAs<Compile3Sym>(sym.get()));
anatofuz
parents:
diff changeset
735 hotPatchable =
anatofuz
parents:
diff changeset
736 (cs.Flags & CompileSym3Flags::HotPatch) != CompileSym3Flags::None;
anatofuz
parents:
diff changeset
737 }
anatofuz
parents:
diff changeset
738 if (sym->kind() == SymbolKind::S_OBJNAME) {
anatofuz
parents:
diff changeset
739 auto objName = cantFail(SymbolDeserializer::deserializeAs<ObjNameSym>(
anatofuz
parents:
diff changeset
740 sym.get()));
anatofuz
parents:
diff changeset
741 pchSignature = objName.Signature;
anatofuz
parents:
diff changeset
742 }
anatofuz
parents:
diff changeset
743 offset += sym->length();
anatofuz
parents:
diff changeset
744 }
anatofuz
parents:
diff changeset
745 }
anatofuz
parents:
diff changeset
746 }
anatofuz
parents:
diff changeset
747
anatofuz
parents:
diff changeset
748 // Depending on the compilation flags, OBJs can refer to external files,
anatofuz
parents:
diff changeset
749 // necessary to merge this OBJ into the final PDB. We currently support two
anatofuz
parents:
diff changeset
750 // types of external files: Precomp/PCH OBJs, when compiling with /Yc and /Yu.
anatofuz
parents:
diff changeset
751 // And PDB type servers, when compiling with /Zi. This function extracts these
anatofuz
parents:
diff changeset
752 // dependencies and makes them available as a TpiSource interface (see
anatofuz
parents:
diff changeset
753 // DebugTypes.h). Both cases only happen with cl.exe: clang-cl produces regular
anatofuz
parents:
diff changeset
754 // output even with /Yc and /Yu and with /Zi.
anatofuz
parents:
diff changeset
755 void ObjFile::initializeDependencies() {
anatofuz
parents:
diff changeset
756 if (!config->debug)
anatofuz
parents:
diff changeset
757 return;
anatofuz
parents:
diff changeset
758
anatofuz
parents:
diff changeset
759 bool isPCH = false;
anatofuz
parents:
diff changeset
760
anatofuz
parents:
diff changeset
761 ArrayRef<uint8_t> data = getDebugSection(".debug$P");
anatofuz
parents:
diff changeset
762 if (!data.empty())
anatofuz
parents:
diff changeset
763 isPCH = true;
anatofuz
parents:
diff changeset
764 else
anatofuz
parents:
diff changeset
765 data = getDebugSection(".debug$T");
anatofuz
parents:
diff changeset
766
anatofuz
parents:
diff changeset
767 if (data.empty())
anatofuz
parents:
diff changeset
768 return;
anatofuz
parents:
diff changeset
769
anatofuz
parents:
diff changeset
770 CVTypeArray types;
anatofuz
parents:
diff changeset
771 BinaryStreamReader reader(data, support::little);
anatofuz
parents:
diff changeset
772 cantFail(reader.readArray(types, reader.getLength()));
anatofuz
parents:
diff changeset
773
anatofuz
parents:
diff changeset
774 CVTypeArray::Iterator firstType = types.begin();
anatofuz
parents:
diff changeset
775 if (firstType == types.end())
anatofuz
parents:
diff changeset
776 return;
anatofuz
parents:
diff changeset
777
anatofuz
parents:
diff changeset
778 // Remember the .debug$T or .debug$P section.
anatofuz
parents:
diff changeset
779 debugTypes = data;
anatofuz
parents:
diff changeset
780
anatofuz
parents:
diff changeset
781 if (isPCH) {
anatofuz
parents:
diff changeset
782 debugTypesObj = makePrecompSource(this);
anatofuz
parents:
diff changeset
783 return;
anatofuz
parents:
diff changeset
784 }
anatofuz
parents:
diff changeset
785
anatofuz
parents:
diff changeset
786 if (firstType->kind() == LF_TYPESERVER2) {
anatofuz
parents:
diff changeset
787 TypeServer2Record ts = cantFail(
anatofuz
parents:
diff changeset
788 TypeDeserializer::deserializeAs<TypeServer2Record>(firstType->data()));
anatofuz
parents:
diff changeset
789 debugTypesObj = makeUseTypeServerSource(this, &ts);
anatofuz
parents:
diff changeset
790 return;
anatofuz
parents:
diff changeset
791 }
anatofuz
parents:
diff changeset
792
anatofuz
parents:
diff changeset
793 if (firstType->kind() == LF_PRECOMP) {
anatofuz
parents:
diff changeset
794 PrecompRecord precomp = cantFail(
anatofuz
parents:
diff changeset
795 TypeDeserializer::deserializeAs<PrecompRecord>(firstType->data()));
anatofuz
parents:
diff changeset
796 debugTypesObj = makeUsePrecompSource(this, &precomp);
anatofuz
parents:
diff changeset
797 return;
anatofuz
parents:
diff changeset
798 }
anatofuz
parents:
diff changeset
799
anatofuz
parents:
diff changeset
800 debugTypesObj = makeTpiSource(this);
anatofuz
parents:
diff changeset
801 }
anatofuz
parents:
diff changeset
802
anatofuz
parents:
diff changeset
803 // Used only for DWARF debug info, which is not common (except in MinGW
anatofuz
parents:
diff changeset
804 // environments). This returns an optional pair of file name and line
anatofuz
parents:
diff changeset
805 // number for where the variable was defined.
anatofuz
parents:
diff changeset
806 Optional<std::pair<StringRef, uint32_t>>
anatofuz
parents:
diff changeset
807 ObjFile::getVariableLocation(StringRef var) {
anatofuz
parents:
diff changeset
808 if (!dwarf) {
anatofuz
parents:
diff changeset
809 dwarf = make<DWARFCache>(DWARFContext::create(*getCOFFObj()));
anatofuz
parents:
diff changeset
810 if (!dwarf)
anatofuz
parents:
diff changeset
811 return None;
anatofuz
parents:
diff changeset
812 }
anatofuz
parents:
diff changeset
813 if (config->machine == I386)
anatofuz
parents:
diff changeset
814 var.consume_front("_");
anatofuz
parents:
diff changeset
815 Optional<std::pair<std::string, unsigned>> ret = dwarf->getVariableLoc(var);
anatofuz
parents:
diff changeset
816 if (!ret)
anatofuz
parents:
diff changeset
817 return None;
anatofuz
parents:
diff changeset
818 return std::make_pair(saver.save(ret->first), ret->second);
anatofuz
parents:
diff changeset
819 }
anatofuz
parents:
diff changeset
820
anatofuz
parents:
diff changeset
821 // Used only for DWARF debug info, which is not common (except in MinGW
anatofuz
parents:
diff changeset
822 // environments).
anatofuz
parents:
diff changeset
823 Optional<DILineInfo> ObjFile::getDILineInfo(uint32_t offset,
anatofuz
parents:
diff changeset
824 uint32_t sectionIndex) {
anatofuz
parents:
diff changeset
825 if (!dwarf) {
anatofuz
parents:
diff changeset
826 dwarf = make<DWARFCache>(DWARFContext::create(*getCOFFObj()));
anatofuz
parents:
diff changeset
827 if (!dwarf)
anatofuz
parents:
diff changeset
828 return None;
anatofuz
parents:
diff changeset
829 }
anatofuz
parents:
diff changeset
830
anatofuz
parents:
diff changeset
831 return dwarf->getDILineInfo(offset, sectionIndex);
anatofuz
parents:
diff changeset
832 }
anatofuz
parents:
diff changeset
833
anatofuz
parents:
diff changeset
834 StringRef ltrim1(StringRef s, const char *chars) {
anatofuz
parents:
diff changeset
835 if (!s.empty() && strchr(chars, s[0]))
anatofuz
parents:
diff changeset
836 return s.substr(1);
anatofuz
parents:
diff changeset
837 return s;
anatofuz
parents:
diff changeset
838 }
anatofuz
parents:
diff changeset
839
anatofuz
parents:
diff changeset
840 void ImportFile::parse() {
anatofuz
parents:
diff changeset
841 const char *buf = mb.getBufferStart();
anatofuz
parents:
diff changeset
842 const auto *hdr = reinterpret_cast<const coff_import_header *>(buf);
anatofuz
parents:
diff changeset
843
anatofuz
parents:
diff changeset
844 // Check if the total size is valid.
anatofuz
parents:
diff changeset
845 if (mb.getBufferSize() != sizeof(*hdr) + hdr->SizeOfData)
anatofuz
parents:
diff changeset
846 fatal("broken import library");
anatofuz
parents:
diff changeset
847
anatofuz
parents:
diff changeset
848 // Read names and create an __imp_ symbol.
anatofuz
parents:
diff changeset
849 StringRef name = saver.save(StringRef(buf + sizeof(*hdr)));
anatofuz
parents:
diff changeset
850 StringRef impName = saver.save("__imp_" + name);
anatofuz
parents:
diff changeset
851 const char *nameStart = buf + sizeof(coff_import_header) + name.size() + 1;
anatofuz
parents:
diff changeset
852 dllName = std::string(StringRef(nameStart));
anatofuz
parents:
diff changeset
853 StringRef extName;
anatofuz
parents:
diff changeset
854 switch (hdr->getNameType()) {
anatofuz
parents:
diff changeset
855 case IMPORT_ORDINAL:
anatofuz
parents:
diff changeset
856 extName = "";
anatofuz
parents:
diff changeset
857 break;
anatofuz
parents:
diff changeset
858 case IMPORT_NAME:
anatofuz
parents:
diff changeset
859 extName = name;
anatofuz
parents:
diff changeset
860 break;
anatofuz
parents:
diff changeset
861 case IMPORT_NAME_NOPREFIX:
anatofuz
parents:
diff changeset
862 extName = ltrim1(name, "?@_");
anatofuz
parents:
diff changeset
863 break;
anatofuz
parents:
diff changeset
864 case IMPORT_NAME_UNDECORATE:
anatofuz
parents:
diff changeset
865 extName = ltrim1(name, "?@_");
anatofuz
parents:
diff changeset
866 extName = extName.substr(0, extName.find('@'));
anatofuz
parents:
diff changeset
867 break;
anatofuz
parents:
diff changeset
868 }
anatofuz
parents:
diff changeset
869
anatofuz
parents:
diff changeset
870 this->hdr = hdr;
anatofuz
parents:
diff changeset
871 externalName = extName;
anatofuz
parents:
diff changeset
872
anatofuz
parents:
diff changeset
873 impSym = symtab->addImportData(impName, this);
anatofuz
parents:
diff changeset
874 // If this was a duplicate, we logged an error but may continue;
anatofuz
parents:
diff changeset
875 // in this case, impSym is nullptr.
anatofuz
parents:
diff changeset
876 if (!impSym)
anatofuz
parents:
diff changeset
877 return;
anatofuz
parents:
diff changeset
878
anatofuz
parents:
diff changeset
879 if (hdr->getType() == llvm::COFF::IMPORT_CONST)
anatofuz
parents:
diff changeset
880 static_cast<void>(symtab->addImportData(name, this));
anatofuz
parents:
diff changeset
881
anatofuz
parents:
diff changeset
882 // If type is function, we need to create a thunk which jump to an
anatofuz
parents:
diff changeset
883 // address pointed by the __imp_ symbol. (This allows you to call
anatofuz
parents:
diff changeset
884 // DLL functions just like regular non-DLL functions.)
anatofuz
parents:
diff changeset
885 if (hdr->getType() == llvm::COFF::IMPORT_CODE)
anatofuz
parents:
diff changeset
886 thunkSym = symtab->addImportThunk(
anatofuz
parents:
diff changeset
887 name, cast_or_null<DefinedImportData>(impSym), hdr->Machine);
anatofuz
parents:
diff changeset
888 }
anatofuz
parents:
diff changeset
889
anatofuz
parents:
diff changeset
890 BitcodeFile::BitcodeFile(MemoryBufferRef mb, StringRef archiveName,
anatofuz
parents:
diff changeset
891 uint64_t offsetInArchive)
anatofuz
parents:
diff changeset
892 : BitcodeFile(mb, archiveName, offsetInArchive, {}) {}
anatofuz
parents:
diff changeset
893
anatofuz
parents:
diff changeset
894 BitcodeFile::BitcodeFile(MemoryBufferRef mb, StringRef archiveName,
anatofuz
parents:
diff changeset
895 uint64_t offsetInArchive,
anatofuz
parents:
diff changeset
896 std::vector<Symbol *> &&symbols)
anatofuz
parents:
diff changeset
897 : InputFile(BitcodeKind, mb), symbols(std::move(symbols)) {
anatofuz
parents:
diff changeset
898 std::string path = mb.getBufferIdentifier().str();
anatofuz
parents:
diff changeset
899 if (config->thinLTOIndexOnly)
anatofuz
parents:
diff changeset
900 path = replaceThinLTOSuffix(mb.getBufferIdentifier());
anatofuz
parents:
diff changeset
901
anatofuz
parents:
diff changeset
902 // ThinLTO assumes that all MemoryBufferRefs given to it have a unique
anatofuz
parents:
diff changeset
903 // name. If two archives define two members with the same name, this
anatofuz
parents:
diff changeset
904 // causes a collision which result in only one of the objects being taken
anatofuz
parents:
diff changeset
905 // into consideration at LTO time (which very likely causes undefined
anatofuz
parents:
diff changeset
906 // symbols later in the link stage). So we append file offset to make
anatofuz
parents:
diff changeset
907 // filename unique.
anatofuz
parents:
diff changeset
908 MemoryBufferRef mbref(
anatofuz
parents:
diff changeset
909 mb.getBuffer(),
anatofuz
parents:
diff changeset
910 saver.save(archiveName + path +
anatofuz
parents:
diff changeset
911 (archiveName.empty() ? "" : utostr(offsetInArchive))));
anatofuz
parents:
diff changeset
912
anatofuz
parents:
diff changeset
913 obj = check(lto::InputFile::create(mbref));
anatofuz
parents:
diff changeset
914 }
anatofuz
parents:
diff changeset
915
anatofuz
parents:
diff changeset
916 BitcodeFile::~BitcodeFile() = default;
anatofuz
parents:
diff changeset
917
anatofuz
parents:
diff changeset
918 void BitcodeFile::parse() {
anatofuz
parents:
diff changeset
919 std::vector<std::pair<Symbol *, bool>> comdat(obj->getComdatTable().size());
anatofuz
parents:
diff changeset
920 for (size_t i = 0; i != obj->getComdatTable().size(); ++i)
anatofuz
parents:
diff changeset
921 // FIXME: lto::InputFile doesn't keep enough data to do correct comdat
anatofuz
parents:
diff changeset
922 // selection handling.
anatofuz
parents:
diff changeset
923 comdat[i] = symtab->addComdat(this, saver.save(obj->getComdatTable()[i]));
anatofuz
parents:
diff changeset
924 for (const lto::InputFile::Symbol &objSym : obj->symbols()) {
anatofuz
parents:
diff changeset
925 StringRef symName = saver.save(objSym.getName());
anatofuz
parents:
diff changeset
926 int comdatIndex = objSym.getComdatIndex();
anatofuz
parents:
diff changeset
927 Symbol *sym;
anatofuz
parents:
diff changeset
928 if (objSym.isUndefined()) {
anatofuz
parents:
diff changeset
929 sym = symtab->addUndefined(symName, this, false);
anatofuz
parents:
diff changeset
930 } else if (objSym.isCommon()) {
anatofuz
parents:
diff changeset
931 sym = symtab->addCommon(this, symName, objSym.getCommonSize());
anatofuz
parents:
diff changeset
932 } else if (objSym.isWeak() && objSym.isIndirect()) {
anatofuz
parents:
diff changeset
933 // Weak external.
anatofuz
parents:
diff changeset
934 sym = symtab->addUndefined(symName, this, true);
anatofuz
parents:
diff changeset
935 std::string fallback = std::string(objSym.getCOFFWeakExternalFallback());
anatofuz
parents:
diff changeset
936 Symbol *alias = symtab->addUndefined(saver.save(fallback));
anatofuz
parents:
diff changeset
937 checkAndSetWeakAlias(symtab, this, sym, alias);
anatofuz
parents:
diff changeset
938 } else if (comdatIndex != -1) {
anatofuz
parents:
diff changeset
939 if (symName == obj->getComdatTable()[comdatIndex])
anatofuz
parents:
diff changeset
940 sym = comdat[comdatIndex].first;
anatofuz
parents:
diff changeset
941 else if (comdat[comdatIndex].second)
anatofuz
parents:
diff changeset
942 sym = symtab->addRegular(this, symName);
anatofuz
parents:
diff changeset
943 else
anatofuz
parents:
diff changeset
944 sym = symtab->addUndefined(symName, this, false);
anatofuz
parents:
diff changeset
945 } else {
anatofuz
parents:
diff changeset
946 sym = symtab->addRegular(this, symName);
anatofuz
parents:
diff changeset
947 }
anatofuz
parents:
diff changeset
948 symbols.push_back(sym);
anatofuz
parents:
diff changeset
949 if (objSym.isUsed())
anatofuz
parents:
diff changeset
950 config->gcroot.push_back(sym);
anatofuz
parents:
diff changeset
951 }
anatofuz
parents:
diff changeset
952 directives = obj->getCOFFLinkerOpts();
anatofuz
parents:
diff changeset
953 }
anatofuz
parents:
diff changeset
954
anatofuz
parents:
diff changeset
955 MachineTypes BitcodeFile::getMachineType() {
anatofuz
parents:
diff changeset
956 switch (Triple(obj->getTargetTriple()).getArch()) {
anatofuz
parents:
diff changeset
957 case Triple::x86_64:
anatofuz
parents:
diff changeset
958 return AMD64;
anatofuz
parents:
diff changeset
959 case Triple::x86:
anatofuz
parents:
diff changeset
960 return I386;
anatofuz
parents:
diff changeset
961 case Triple::arm:
anatofuz
parents:
diff changeset
962 return ARMNT;
anatofuz
parents:
diff changeset
963 case Triple::aarch64:
anatofuz
parents:
diff changeset
964 return ARM64;
anatofuz
parents:
diff changeset
965 default:
anatofuz
parents:
diff changeset
966 return IMAGE_FILE_MACHINE_UNKNOWN;
anatofuz
parents:
diff changeset
967 }
anatofuz
parents:
diff changeset
968 }
anatofuz
parents:
diff changeset
969
anatofuz
parents:
diff changeset
970 std::string replaceThinLTOSuffix(StringRef path) {
anatofuz
parents:
diff changeset
971 StringRef suffix = config->thinLTOObjectSuffixReplace.first;
anatofuz
parents:
diff changeset
972 StringRef repl = config->thinLTOObjectSuffixReplace.second;
anatofuz
parents:
diff changeset
973
anatofuz
parents:
diff changeset
974 if (path.consume_back(suffix))
anatofuz
parents:
diff changeset
975 return (path + repl).str();
anatofuz
parents:
diff changeset
976 return std::string(path);
anatofuz
parents:
diff changeset
977 }
anatofuz
parents:
diff changeset
978
anatofuz
parents:
diff changeset
979 } // namespace coff
anatofuz
parents:
diff changeset
980 } // namespace lld