blob: 01280b68f4d769910a630773d47b8910e9c58050 [file] [log] [blame]
Rui Ueyama411c63602015-05-28 19:09:30 +00001//===- Driver.cpp ---------------------------------------------------------===//
2//
3// The LLVM Linker
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10#include "Config.h"
11#include "Driver.h"
Rui Ueyama562daa82015-06-18 21:50:38 +000012#include "Error.h"
Rui Ueyama411c63602015-05-28 19:09:30 +000013#include "InputFiles.h"
Rui Ueyama411c63602015-05-28 19:09:30 +000014#include "SymbolTable.h"
15#include "Writer.h"
Rui Ueyama411c63602015-05-28 19:09:30 +000016#include "llvm/ADT/Optional.h"
17#include "llvm/ADT/STLExtras.h"
Rui Ueyama3ee0fe42015-05-31 03:55:46 +000018#include "llvm/ADT/StringSwitch.h"
Peter Collingbournebd1cb792015-06-09 21:52:48 +000019#include "llvm/LibDriver/LibDriver.h"
Rui Ueyama411c63602015-05-28 19:09:30 +000020#include "llvm/Option/Arg.h"
21#include "llvm/Option/ArgList.h"
22#include "llvm/Option/Option.h"
23#include "llvm/Support/CommandLine.h"
24#include "llvm/Support/Debug.h"
Rui Ueyama411c63602015-05-28 19:09:30 +000025#include "llvm/Support/Path.h"
Rui Ueyama54b71da2015-05-31 19:17:12 +000026#include "llvm/Support/Process.h"
Peter Collingbourne60c16162015-06-01 20:10:10 +000027#include "llvm/Support/TargetSelect.h"
Rui Ueyama411c63602015-05-28 19:09:30 +000028#include "llvm/Support/raw_ostream.h"
Rui Ueyama2bf6a122015-06-14 21:50:50 +000029#include <algorithm>
Rui Ueyama411c63602015-05-28 19:09:30 +000030#include <memory>
31
32using namespace llvm;
Rui Ueyama84936e02015-07-07 23:39:18 +000033using namespace llvm::COFF;
Rui Ueyama54b71da2015-05-31 19:17:12 +000034using llvm::sys::Process;
Peter Collingbournebaf5f872015-06-26 19:20:09 +000035using llvm::sys::fs::OpenFlags;
Rui Ueyama711cd2d2015-05-31 21:17:10 +000036using llvm::sys::fs::file_magic;
37using llvm::sys::fs::identify_magic;
Rui Ueyama411c63602015-05-28 19:09:30 +000038
Rui Ueyama3500f662015-05-28 20:30:06 +000039namespace lld {
40namespace coff {
Rui Ueyama411c63602015-05-28 19:09:30 +000041
Rui Ueyama3500f662015-05-28 20:30:06 +000042Configuration *Config;
Rui Ueyamaa9cbbf82015-05-31 19:17:09 +000043LinkerDriver *Driver;
44
David Blaikie00818192015-06-22 22:06:48 +000045bool link(llvm::ArrayRef<const char *> Args) {
Rui Ueyamaa9cbbf82015-05-31 19:17:09 +000046 auto C = make_unique<Configuration>();
47 Config = C.get();
48 auto D = make_unique<LinkerDriver>();
49 Driver = D.get();
David Blaikieb2b1c7c2015-06-21 06:32:10 +000050 return Driver->link(Args);
Rui Ueyamaa9cbbf82015-05-31 19:17:09 +000051}
Rui Ueyama411c63602015-05-28 19:09:30 +000052
Rui Ueyamaad660982015-06-07 00:20:32 +000053// Drop directory components and replace extension with ".exe".
54static std::string getOutputPath(StringRef Path) {
55 auto P = Path.find_last_of("\\/");
56 StringRef S = (P == StringRef::npos) ? Path : Path.substr(P + 1);
57 return (S.substr(0, S.rfind('.')) + ".exe").str();
Rui Ueyama411c63602015-05-28 19:09:30 +000058}
59
Rui Ueyamad7c2f582015-05-31 21:04:56 +000060// Opens a file. Path has to be resolved already.
61// Newly created memory buffers are owned by this driver.
Rui Ueyama2bf6a122015-06-14 21:50:50 +000062ErrorOr<MemoryBufferRef> LinkerDriver::openFile(StringRef Path) {
Rui Ueyamad7c2f582015-05-31 21:04:56 +000063 auto MBOrErr = MemoryBuffer::getFile(Path);
64 if (auto EC = MBOrErr.getError())
65 return EC;
66 std::unique_ptr<MemoryBuffer> MB = std::move(MBOrErr.get());
67 MemoryBufferRef MBRef = MB->getMemBufferRef();
68 OwningMBs.push_back(std::move(MB)); // take ownership
Rui Ueyama2bf6a122015-06-14 21:50:50 +000069 return MBRef;
70}
Rui Ueyama711cd2d2015-05-31 21:17:10 +000071
Rui Ueyama2bf6a122015-06-14 21:50:50 +000072static std::unique_ptr<InputFile> createFile(MemoryBufferRef MB) {
Rui Ueyama711cd2d2015-05-31 21:17:10 +000073 // File type is detected by contents, not by file extension.
Rui Ueyama2bf6a122015-06-14 21:50:50 +000074 file_magic Magic = identify_magic(MB.getBuffer());
Rui Ueyama711cd2d2015-05-31 21:17:10 +000075 if (Magic == file_magic::archive)
Rui Ueyama2bf6a122015-06-14 21:50:50 +000076 return std::unique_ptr<InputFile>(new ArchiveFile(MB));
Peter Collingbourne60c16162015-06-01 20:10:10 +000077 if (Magic == file_magic::bitcode)
Rui Ueyama2bf6a122015-06-14 21:50:50 +000078 return std::unique_ptr<InputFile>(new BitcodeFile(MB));
Rui Ueyamaad660982015-06-07 00:20:32 +000079 if (Config->OutputFile == "")
Rui Ueyama2bf6a122015-06-14 21:50:50 +000080 Config->OutputFile = getOutputPath(MB.getBufferIdentifier());
81 return std::unique_ptr<InputFile>(new ObjectFile(MB));
Rui Ueyama411c63602015-05-28 19:09:30 +000082}
83
Rui Ueyama411c63602015-05-28 19:09:30 +000084// Parses .drectve section contents and returns a list of files
85// specified by /defaultlib.
Rui Ueyamaa9cbbf82015-05-31 19:17:09 +000086std::error_code
Rui Ueyama0d2e9992015-06-23 23:56:39 +000087LinkerDriver::parseDirectives(StringRef S) {
Rui Ueyama115d7c12015-06-07 02:55:19 +000088 auto ArgsOrErr = Parser.parse(S);
Rui Ueyama411c63602015-05-28 19:09:30 +000089 if (auto EC = ArgsOrErr.getError())
90 return EC;
David Blaikie6521ed92015-06-22 22:06:52 +000091 llvm::opt::InputArgList Args = std::move(ArgsOrErr.get());
Rui Ueyama411c63602015-05-28 19:09:30 +000092
David Blaikie6521ed92015-06-22 22:06:52 +000093 for (auto *Arg : Args) {
Rui Ueyama562daa82015-06-18 21:50:38 +000094 switch (Arg->getOption().getID()) {
95 case OPT_alternatename:
96 if (auto EC = parseAlternateName(Arg->getValue()))
Rui Ueyamad7c2f582015-05-31 21:04:56 +000097 return EC;
Rui Ueyama562daa82015-06-18 21:50:38 +000098 break;
99 case OPT_defaultlib:
100 if (Optional<StringRef> Path = findLib(Arg->getValue())) {
101 ErrorOr<MemoryBufferRef> MBOrErr = openFile(*Path);
102 if (auto EC = MBOrErr.getError())
103 return EC;
Rui Ueyama0d2e9992015-06-23 23:56:39 +0000104 Symtab.addFile(createFile(MBOrErr.get()));
Rui Ueyama562daa82015-06-18 21:50:38 +0000105 }
106 break;
107 case OPT_export: {
108 ErrorOr<Export> E = parseExport(Arg->getValue());
109 if (auto EC = E.getError())
110 return EC;
111 Config->Exports.push_back(E.get());
112 break;
113 }
114 case OPT_failifmismatch:
115 if (auto EC = checkFailIfMismatch(Arg->getValue()))
116 return EC;
117 break;
Rui Ueyama08d5e182015-06-18 23:20:11 +0000118 case OPT_incl:
Rui Ueyama32f8e1c2015-06-26 03:44:00 +0000119 addUndefined(Arg->getValue());
Rui Ueyama08d5e182015-06-18 23:20:11 +0000120 break;
Rui Ueyamace86c992015-06-18 23:22:39 +0000121 case OPT_merge:
Rui Ueyama6600eb12015-07-04 23:37:32 +0000122 if (auto EC = parseMerge(Arg->getValue()))
123 return EC;
Rui Ueyamace86c992015-06-18 23:22:39 +0000124 break;
125 case OPT_nodefaultlib:
126 Config->NoDefaultLibs.insert(doFindLib(Arg->getValue()));
127 break;
Rui Ueyama562daa82015-06-18 21:50:38 +0000128 default:
129 llvm::errs() << Arg->getSpelling() << " is not allowed in .drectve\n";
130 return make_error_code(LLDError::InvalidOption);
Rui Ueyamad7c2f582015-05-31 21:04:56 +0000131 }
132 }
Rui Ueyama411c63602015-05-28 19:09:30 +0000133 return std::error_code();
134}
135
Rui Ueyama54b71da2015-05-31 19:17:12 +0000136// Find file from search paths. You can omit ".obj", this function takes
137// care of that. Note that the returned path is not guaranteed to exist.
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000138StringRef LinkerDriver::doFindFile(StringRef Filename) {
Rui Ueyama54b71da2015-05-31 19:17:12 +0000139 bool hasPathSep = (Filename.find_first_of("/\\") != StringRef::npos);
140 if (hasPathSep)
141 return Filename;
142 bool hasExt = (Filename.find('.') != StringRef::npos);
143 for (StringRef Dir : SearchPaths) {
144 SmallString<128> Path = Dir;
145 llvm::sys::path::append(Path, Filename);
146 if (llvm::sys::fs::exists(Path.str()))
147 return Alloc.save(Path.str());
148 if (!hasExt) {
149 Path.append(".obj");
150 if (llvm::sys::fs::exists(Path.str()))
151 return Alloc.save(Path.str());
152 }
153 }
154 return Filename;
155}
156
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000157// Resolves a file path. This never returns the same path
158// (in that case, it returns None).
159Optional<StringRef> LinkerDriver::findFile(StringRef Filename) {
160 StringRef Path = doFindFile(Filename);
161 bool Seen = !VisitedFiles.insert(Path.lower()).second;
162 if (Seen)
163 return None;
164 return Path;
Rui Ueyama54b71da2015-05-31 19:17:12 +0000165}
166
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000167// Find library file from search path.
168StringRef LinkerDriver::doFindLib(StringRef Filename) {
169 // Add ".lib" to Filename if that has no file extension.
Rui Ueyama54b71da2015-05-31 19:17:12 +0000170 bool hasExt = (Filename.find('.') != StringRef::npos);
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000171 if (!hasExt)
172 Filename = Alloc.save(Filename + ".lib");
173 return doFindFile(Filename);
174}
175
176// Resolves a library path. /nodefaultlib options are taken into
177// consideration. This never returns the same path (in that case,
178// it returns None).
179Optional<StringRef> LinkerDriver::findLib(StringRef Filename) {
180 if (Config->NoDefaultLibAll)
181 return None;
182 StringRef Path = doFindLib(Filename);
183 if (Config->NoDefaultLibs.count(Path))
184 return None;
185 bool Seen = !VisitedFiles.insert(Path.lower()).second;
186 if (Seen)
187 return None;
188 return Path;
Rui Ueyama54b71da2015-05-31 19:17:12 +0000189}
190
191// Parses LIB environment which contains a list of search paths.
Rui Ueyamaf00df0a2015-06-19 22:39:48 +0000192void LinkerDriver::addLibSearchPaths() {
Rui Ueyama54b71da2015-05-31 19:17:12 +0000193 Optional<std::string> EnvOpt = Process::GetEnv("LIB");
194 if (!EnvOpt.hasValue())
Rui Ueyamaf00df0a2015-06-19 22:39:48 +0000195 return;
Rui Ueyama54b71da2015-05-31 19:17:12 +0000196 StringRef Env = Alloc.save(*EnvOpt);
197 while (!Env.empty()) {
198 StringRef Path;
199 std::tie(Path, Env) = Env.split(';');
Rui Ueyamaf00df0a2015-06-19 22:39:48 +0000200 SearchPaths.push_back(Path);
Rui Ueyama54b71da2015-05-31 19:17:12 +0000201 }
Rui Ueyama54b71da2015-05-31 19:17:12 +0000202}
203
Rui Ueyama6bf638e2015-07-02 00:04:14 +0000204Undefined *LinkerDriver::addUndefined(StringRef Name) {
205 Undefined *U = Symtab.addUndefined(Name);
Rui Ueyama18f8d2c2015-07-02 00:21:08 +0000206 Config->GCRoot.insert(U);
Rui Ueyama6bf638e2015-07-02 00:04:14 +0000207 return U;
Rui Ueyama32f8e1c2015-06-26 03:44:00 +0000208}
209
Rui Ueyama7c3e23f2015-07-09 01:25:49 +0000210// Symbol names are mangled by appending "_" prefix on x86.
211StringRef LinkerDriver::mangle(StringRef Sym) {
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000212 assert(Config->MachineType != IMAGE_FILE_MACHINE_UNKNOWN);
Rui Ueyama7c3e23f2015-07-09 01:25:49 +0000213 if (Config->MachineType == IMAGE_FILE_MACHINE_I386)
214 return Alloc.save("_" + Sym);
215 return Sym;
216}
217
Rui Ueyama45044f42015-06-29 01:03:53 +0000218// Windows specific -- find default entry point name.
219StringRef LinkerDriver::findDefaultEntry() {
220 // User-defined main functions and their corresponding entry points.
221 static const char *Entries[][2] = {
222 {"main", "mainCRTStartup"},
223 {"wmain", "wmainCRTStartup"},
224 {"WinMain", "WinMainCRTStartup"},
225 {"wWinMain", "wWinMainCRTStartup"},
226 };
227 for (auto E : Entries) {
Rui Ueyama7c3e23f2015-07-09 01:25:49 +0000228 Symbol *Sym = Symtab.find(mangle(E[0]));
Rui Ueyama183f53f2015-07-06 17:45:22 +0000229 if (Sym && !isa<Undefined>(Sym->Body))
Rui Ueyama7c3e23f2015-07-09 01:25:49 +0000230 return mangle(E[1]);
Rui Ueyama45044f42015-06-29 01:03:53 +0000231 }
232 return "";
233}
234
235WindowsSubsystem LinkerDriver::inferSubsystem() {
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000236 if (Config->DLL)
237 return IMAGE_SUBSYSTEM_WINDOWS_GUI;
Rui Ueyama7c3e23f2015-07-09 01:25:49 +0000238 if (Symtab.find(mangle("main")) || Symtab.find(mangle("wmain")))
Rui Ueyama45044f42015-06-29 01:03:53 +0000239 return IMAGE_SUBSYSTEM_WINDOWS_CUI;
Rui Ueyama7c3e23f2015-07-09 01:25:49 +0000240 if (Symtab.find(mangle("WinMain")) || Symtab.find(mangle("wWinMain")))
Rui Ueyama45044f42015-06-29 01:03:53 +0000241 return IMAGE_SUBSYSTEM_WINDOWS_GUI;
242 return IMAGE_SUBSYSTEM_UNKNOWN;
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000243}
244
David Blaikie00818192015-06-22 22:06:48 +0000245bool LinkerDriver::link(llvm::ArrayRef<const char *> ArgsArr) {
Peter Collingbourne60c16162015-06-01 20:10:10 +0000246 // Needed for LTO.
247 llvm::InitializeAllTargetInfos();
248 llvm::InitializeAllTargets();
249 llvm::InitializeAllTargetMCs();
250 llvm::InitializeAllAsmParsers();
251 llvm::InitializeAllAsmPrinters();
252 llvm::InitializeAllDisassemblers();
253
Peter Collingbournebd1cb792015-06-09 21:52:48 +0000254 // If the first command line argument is "/lib", link.exe acts like lib.exe.
255 // We call our own implementation of lib.exe that understands bitcode files.
David Blaikieb2b1c7c2015-06-21 06:32:10 +0000256 if (ArgsArr.size() > 1 && StringRef(ArgsArr[1]).equals_lower("/lib"))
257 return llvm::libDriverMain(ArgsArr.slice(1)) == 0;
Peter Collingbournebd1cb792015-06-09 21:52:48 +0000258
Rui Ueyama411c63602015-05-28 19:09:30 +0000259 // Parse command line options.
Rui Ueyama9d72f092015-06-28 03:05:38 +0000260 auto ArgsOrErr = Parser.parseLINK(ArgsArr.slice(1));
Rui Ueyama411c63602015-05-28 19:09:30 +0000261 if (auto EC = ArgsOrErr.getError()) {
262 llvm::errs() << EC.message() << "\n";
263 return false;
264 }
David Blaikie6521ed92015-06-22 22:06:52 +0000265 llvm::opt::InputArgList Args = std::move(ArgsOrErr.get());
Rui Ueyama411c63602015-05-28 19:09:30 +0000266
Rui Ueyama5c726432015-05-29 16:11:52 +0000267 // Handle /help
David Blaikie6521ed92015-06-22 22:06:52 +0000268 if (Args.hasArg(OPT_help)) {
David Blaikieb2b1c7c2015-06-21 06:32:10 +0000269 printHelp(ArgsArr[0]);
Rui Ueyama5c726432015-05-29 16:11:52 +0000270 return true;
271 }
272
David Blaikie6521ed92015-06-22 22:06:52 +0000273 if (Args.filtered_begin(OPT_INPUT) == Args.filtered_end()) {
Rui Ueyama411c63602015-05-28 19:09:30 +0000274 llvm::errs() << "no input files.\n";
275 return false;
276 }
Rui Ueyama3d3e6fb2015-05-29 16:06:00 +0000277
Rui Ueyamaf00df0a2015-06-19 22:39:48 +0000278 // Construct search path list.
279 SearchPaths.push_back("");
David Blaikie6521ed92015-06-22 22:06:52 +0000280 for (auto *Arg : Args.filtered(OPT_libpath))
Rui Ueyamaf00df0a2015-06-19 22:39:48 +0000281 SearchPaths.push_back(Arg->getValue());
282 addLibSearchPaths();
283
Rui Ueyamaad660982015-06-07 00:20:32 +0000284 // Handle /out
David Blaikie6521ed92015-06-22 22:06:52 +0000285 if (auto *Arg = Args.getLastArg(OPT_out))
Rui Ueyamaad660982015-06-07 00:20:32 +0000286 Config->OutputFile = Arg->getValue();
287
Rui Ueyama3d3e6fb2015-05-29 16:06:00 +0000288 // Handle /verbose
David Blaikie6521ed92015-06-22 22:06:52 +0000289 if (Args.hasArg(OPT_verbose))
Rui Ueyama411c63602015-05-28 19:09:30 +0000290 Config->Verbose = true;
Rui Ueyama3d3e6fb2015-05-29 16:06:00 +0000291
Rui Ueyama95925fd2015-06-28 19:35:15 +0000292 // Handle /force or /force:unresolved
293 if (Args.hasArg(OPT_force) || Args.hasArg(OPT_force_unresolved))
294 Config->Force = true;
295
Rui Ueyama32f8e1c2015-06-26 03:44:00 +0000296 // Handle /entry
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000297 StringRef Entry;
Rui Ueyama6bf638e2015-07-02 00:04:14 +0000298 if (auto *Arg = Args.getLastArg(OPT_entry))
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000299 Entry = Arg->getValue();
Rui Ueyama32f8e1c2015-06-26 03:44:00 +0000300
Rui Ueyama6600eb12015-07-04 23:37:32 +0000301 // Handle /debug
302 if (Args.hasArg(OPT_debug))
303 Config->Debug = true;
304
Rui Ueyamaa8b60452015-06-28 19:56:30 +0000305 // Handle /noentry
306 if (Args.hasArg(OPT_noentry)) {
307 if (!Args.hasArg(OPT_dll)) {
308 llvm::errs() << "/noentry must be specified with /dll\n";
309 return false;
310 }
311 Config->NoEntry = true;
312 }
313
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000314 // Handle /dll
David Blaikie6521ed92015-06-22 22:06:52 +0000315 if (Args.hasArg(OPT_dll)) {
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000316 Config->DLL = true;
Rui Ueyama49d6cd32015-07-03 00:02:19 +0000317 Config->ImageBase = 0x180000000U;
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000318 Config->ManifestID = 2;
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000319 if (Entry.empty() && !Config->NoEntry)
320 Entry = "_DllMainCRTStartup";
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000321 }
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000322
Rui Ueyama588e8322015-06-15 01:23:58 +0000323 // Handle /fixed
David Blaikie6521ed92015-06-22 22:06:52 +0000324 if (Args.hasArg(OPT_fixed)) {
325 if (Args.hasArg(OPT_dynamicbase)) {
Rui Ueyama6592ff82015-06-16 23:13:00 +0000326 llvm::errs() << "/fixed must not be specified with /dynamicbase\n";
327 return false;
328 }
Rui Ueyama588e8322015-06-15 01:23:58 +0000329 Config->Relocatable = false;
Rui Ueyama6592ff82015-06-16 23:13:00 +0000330 Config->DynamicBase = false;
331 }
Rui Ueyama588e8322015-06-15 01:23:58 +0000332
Rui Ueyama3d3e6fb2015-05-29 16:06:00 +0000333 // Handle /machine
Rui Ueyamae16a75d52015-07-08 18:14:51 +0000334 if (auto *Arg = Args.getLastArg(OPT_machine)) {
335 ErrorOr<MachineTypes> MTOrErr = getMachineType(Arg->getValue());
336 if (MTOrErr.getError())
337 return false;
338 Config->MachineType = MTOrErr.get();
Rui Ueyama3d3e6fb2015-05-29 16:06:00 +0000339 }
Rui Ueyama3d3e6fb2015-05-29 16:06:00 +0000340
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000341 // Handle /nodefaultlib:<filename>
David Blaikie6521ed92015-06-22 22:06:52 +0000342 for (auto *Arg : Args.filtered(OPT_nodefaultlib))
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000343 Config->NoDefaultLibs.insert(doFindLib(Arg->getValue()));
344
345 // Handle /nodefaultlib
David Blaikie6521ed92015-06-22 22:06:52 +0000346 if (Args.hasArg(OPT_nodefaultlib_all))
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000347 Config->NoDefaultLibAll = true;
348
Rui Ueyama804a8b62015-05-29 16:18:15 +0000349 // Handle /base
David Blaikie6521ed92015-06-22 22:06:52 +0000350 if (auto *Arg = Args.getLastArg(OPT_base)) {
Rui Ueyama804a8b62015-05-29 16:18:15 +0000351 if (auto EC = parseNumbers(Arg->getValue(), &Config->ImageBase)) {
Rui Ueyamab41b7e52015-05-29 16:21:11 +0000352 llvm::errs() << "/base: " << EC.message() << "\n";
353 return false;
354 }
355 }
356
357 // Handle /stack
David Blaikie6521ed92015-06-22 22:06:52 +0000358 if (auto *Arg = Args.getLastArg(OPT_stack)) {
Rui Ueyamab41b7e52015-05-29 16:21:11 +0000359 if (auto EC = parseNumbers(Arg->getValue(), &Config->StackReserve,
360 &Config->StackCommit)) {
361 llvm::errs() << "/stack: " << EC.message() << "\n";
Rui Ueyama804a8b62015-05-29 16:18:15 +0000362 return false;
363 }
364 }
365
Rui Ueyamac377e9a2015-05-29 16:23:40 +0000366 // Handle /heap
David Blaikie6521ed92015-06-22 22:06:52 +0000367 if (auto *Arg = Args.getLastArg(OPT_heap)) {
Rui Ueyamac377e9a2015-05-29 16:23:40 +0000368 if (auto EC = parseNumbers(Arg->getValue(), &Config->HeapReserve,
369 &Config->HeapCommit)) {
370 llvm::errs() << "/heap: " << EC.message() << "\n";
371 return false;
372 }
373 }
374
Rui Ueyamab9dcdb52015-05-29 16:28:29 +0000375 // Handle /version
David Blaikie6521ed92015-06-22 22:06:52 +0000376 if (auto *Arg = Args.getLastArg(OPT_version)) {
Rui Ueyamab9dcdb52015-05-29 16:28:29 +0000377 if (auto EC = parseVersion(Arg->getValue(), &Config->MajorImageVersion,
378 &Config->MinorImageVersion)) {
379 llvm::errs() << "/version: " << EC.message() << "\n";
380 return false;
381 }
382 }
383
Rui Ueyama15cc47e2015-05-29 16:34:31 +0000384 // Handle /subsystem
David Blaikie6521ed92015-06-22 22:06:52 +0000385 if (auto *Arg = Args.getLastArg(OPT_subsystem)) {
Rui Ueyama15cc47e2015-05-29 16:34:31 +0000386 if (auto EC = parseSubsystem(Arg->getValue(), &Config->Subsystem,
387 &Config->MajorOSVersion,
388 &Config->MinorOSVersion)) {
389 llvm::errs() << "/subsystem: " << EC.message() << "\n";
390 return false;
391 }
392 }
393
Rui Ueyama2edb35a2015-06-18 19:09:30 +0000394 // Handle /alternatename
David Blaikie6521ed92015-06-22 22:06:52 +0000395 for (auto *Arg : Args.filtered(OPT_alternatename))
Rui Ueyama2edb35a2015-06-18 19:09:30 +0000396 if (parseAlternateName(Arg->getValue()))
397 return false;
398
Rui Ueyama08d5e182015-06-18 23:20:11 +0000399 // Handle /include
David Blaikie6521ed92015-06-22 22:06:52 +0000400 for (auto *Arg : Args.filtered(OPT_incl))
Rui Ueyama32f8e1c2015-06-26 03:44:00 +0000401 addUndefined(Arg->getValue());
Rui Ueyama08d5e182015-06-18 23:20:11 +0000402
Rui Ueyamab95188c2015-06-18 20:27:09 +0000403 // Handle /implib
David Blaikie6521ed92015-06-22 22:06:52 +0000404 if (auto *Arg = Args.getLastArg(OPT_implib))
Rui Ueyamab95188c2015-06-18 20:27:09 +0000405 Config->Implib = Arg->getValue();
406
Rui Ueyamae2cbfea2015-06-07 03:17:42 +0000407 // Handle /opt
David Blaikie6521ed92015-06-22 22:06:52 +0000408 for (auto *Arg : Args.filtered(OPT_opt)) {
Rui Ueyamae2cbfea2015-06-07 03:17:42 +0000409 std::string S = StringRef(Arg->getValue()).lower();
410 if (S == "noref") {
411 Config->DoGC = false;
412 continue;
413 }
Rui Ueyamaf799ede2015-06-25 23:26:58 +0000414 if (S == "lldicf") {
Rui Ueyamaddf71fc2015-06-24 04:36:52 +0000415 Config->ICF = true;
416 continue;
417 }
Rui Ueyamae2cbfea2015-06-07 03:17:42 +0000418 if (S != "ref" && S != "icf" && S != "noicf" &&
419 S != "lbr" && S != "nolbr" &&
420 !StringRef(S).startswith("icf=")) {
421 llvm::errs() << "/opt: unknown option: " << S << "\n";
422 return false;
423 }
424 }
425
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000426 // Handle /export
David Blaikie6521ed92015-06-22 22:06:52 +0000427 for (auto *Arg : Args.filtered(OPT_export)) {
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000428 ErrorOr<Export> E = parseExport(Arg->getValue());
429 if (E.getError())
430 return false;
431 Config->Exports.push_back(E.get());
432 }
433
Rui Ueyamaa77336b2015-06-21 22:31:52 +0000434 // Handle /delayload
David Blaikie6521ed92015-06-22 22:06:52 +0000435 for (auto *Arg : Args.filtered(OPT_delayload)) {
Rui Ueyama7a247ee2015-07-03 01:40:14 +0000436 Config->DelayLoads.insert(StringRef(Arg->getValue()).lower());
Rui Ueyama32f8e1c2015-06-26 03:44:00 +0000437 addUndefined("__delayLoadHelper2");
Rui Ueyamaa77336b2015-06-21 22:31:52 +0000438 }
439
Rui Ueyama8854d8a2015-06-04 19:21:24 +0000440 // Handle /failifmismatch
David Blaikie6521ed92015-06-22 22:06:52 +0000441 for (auto *Arg : Args.filtered(OPT_failifmismatch))
Rui Ueyama75b098b2015-06-18 21:23:34 +0000442 if (checkFailIfMismatch(Arg->getValue()))
443 return false;
Rui Ueyama8854d8a2015-06-04 19:21:24 +0000444
Rui Ueyama1f373702015-06-17 19:19:25 +0000445 // Handle /def
David Blaikie6521ed92015-06-22 22:06:52 +0000446 if (auto *Arg = Args.getLastArg(OPT_deffile)) {
Rui Ueyama1f373702015-06-17 19:19:25 +0000447 ErrorOr<MemoryBufferRef> MBOrErr = openFile(Arg->getValue());
448 if (auto EC = MBOrErr.getError()) {
449 llvm::errs() << "/def: " << EC.message() << "\n";
450 return false;
451 }
452 // parseModuleDefs mutates Config object.
453 if (parseModuleDefs(MBOrErr.get()))
454 return false;
455 }
456
Rui Ueyama6600eb12015-07-04 23:37:32 +0000457 // Handle /merge
458 for (auto *Arg : Args.filtered(OPT_merge))
459 if (parseMerge(Arg->getValue()))
460 return false;
461
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000462 // Handle /manifest
David Blaikie6521ed92015-06-22 22:06:52 +0000463 if (auto *Arg = Args.getLastArg(OPT_manifest_colon)) {
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000464 if (auto EC = parseManifest(Arg->getValue())) {
465 llvm::errs() << "/manifest: " << EC.message() << "\n";
466 return false;
467 }
468 }
469
470 // Handle /manifestuac
David Blaikie6521ed92015-06-22 22:06:52 +0000471 if (auto *Arg = Args.getLastArg(OPT_manifestuac)) {
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000472 if (auto EC = parseManifestUAC(Arg->getValue())) {
473 llvm::errs() << "/manifestuac: " << EC.message() << "\n";
474 return false;
475 }
476 }
477
478 // Handle /manifestdependency
David Blaikie6521ed92015-06-22 22:06:52 +0000479 if (auto *Arg = Args.getLastArg(OPT_manifestdependency))
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000480 Config->ManifestDependency = Arg->getValue();
481
482 // Handle /manifestfile
David Blaikie6521ed92015-06-22 22:06:52 +0000483 if (auto *Arg = Args.getLastArg(OPT_manifestfile))
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000484 Config->ManifestFile = Arg->getValue();
485
Rui Ueyama6592ff82015-06-16 23:13:00 +0000486 // Handle miscellaneous boolean flags.
David Blaikie6521ed92015-06-22 22:06:52 +0000487 if (Args.hasArg(OPT_allowbind_no))
488 Config->AllowBind = false;
489 if (Args.hasArg(OPT_allowisolation_no))
490 Config->AllowIsolation = false;
491 if (Args.hasArg(OPT_dynamicbase_no))
492 Config->DynamicBase = false;
493 if (Args.hasArg(OPT_highentropyva_no))
494 Config->HighEntropyVA = false;
495 if (Args.hasArg(OPT_nxcompat_no))
496 Config->NxCompat = false;
497 if (Args.hasArg(OPT_tsaware_no))
498 Config->TerminalServerAware = false;
Rui Ueyama6592ff82015-06-16 23:13:00 +0000499
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000500 // Create a list of input files. Files can be given as arguments
501 // for /defaultlib option.
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000502 std::vector<StringRef> Paths;
503 std::vector<MemoryBufferRef> MBs;
David Blaikie6521ed92015-06-22 22:06:52 +0000504 for (auto *Arg : Args.filtered(OPT_INPUT))
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000505 if (Optional<StringRef> Path = findFile(Arg->getValue()))
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000506 Paths.push_back(*Path);
David Blaikie6521ed92015-06-22 22:06:52 +0000507 for (auto *Arg : Args.filtered(OPT_defaultlib))
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000508 if (Optional<StringRef> Path = findLib(Arg->getValue()))
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000509 Paths.push_back(*Path);
510 for (StringRef Path : Paths) {
Rui Ueyama2bf6a122015-06-14 21:50:50 +0000511 ErrorOr<MemoryBufferRef> MBOrErr = openFile(Path);
512 if (auto EC = MBOrErr.getError()) {
513 llvm::errs() << "cannot open " << Path << ": " << EC.message() << "\n";
514 return false;
515 }
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000516 MBs.push_back(MBOrErr.get());
Rui Ueyama2bf6a122015-06-14 21:50:50 +0000517 }
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000518
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000519 // Windows specific -- Create a resource file containing a manifest file.
520 if (Config->Manifest == Configuration::Embed) {
521 auto MBOrErr = createManifestRes();
522 if (MBOrErr.getError())
523 return false;
524 std::unique_ptr<MemoryBuffer> MB = std::move(MBOrErr.get());
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000525 MBs.push_back(MB->getMemBufferRef());
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000526 OwningMBs.push_back(std::move(MB)); // take ownership
527 }
528
Rui Ueyama2bf6a122015-06-14 21:50:50 +0000529 // Windows specific -- Input files can be Windows resource files (.res files).
530 // We invoke cvtres.exe to convert resource files to a regular COFF file
531 // then link the result file normally.
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000532 std::vector<MemoryBufferRef> Resources;
Rui Ueyama77731b42015-06-26 23:59:13 +0000533 auto NotResource = [](MemoryBufferRef MB) {
534 return identify_magic(MB.getBuffer()) != file_magic::windows_resource;
Rui Ueyama2bf6a122015-06-14 21:50:50 +0000535 };
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000536 auto It = std::stable_partition(MBs.begin(), MBs.end(), NotResource);
537 if (It != MBs.end()) {
538 Resources.insert(Resources.end(), It, MBs.end());
539 MBs.erase(It, MBs.end());
Rui Ueyama2bf6a122015-06-14 21:50:50 +0000540 }
541
Rui Ueyama85225b02015-07-02 03:15:15 +0000542 // Read all input files given via the command line. Note that step()
543 // doesn't read files that are specified by directive sections.
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000544 for (MemoryBufferRef MB : MBs)
Rui Ueyama0d2e9992015-06-23 23:56:39 +0000545 Symtab.addFile(createFile(MB));
Rui Ueyama85225b02015-07-02 03:15:15 +0000546 if (auto EC = Symtab.step()) {
Rui Ueyama0d2e9992015-06-23 23:56:39 +0000547 llvm::errs() << EC.message() << "\n";
548 return false;
Rui Ueyama411c63602015-05-28 19:09:30 +0000549 }
Rui Ueyama5cff6852015-05-31 03:34:08 +0000550
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000551 // Determine machine type and check if all object files are
552 // for the same CPU type. Note that this needs to be done before
553 // any call to mangle().
554 for (std::unique_ptr<InputFile> &File : Symtab.getFiles()) {
555 MachineTypes MT = File->getMachineType();
556 if (MT == IMAGE_FILE_MACHINE_UNKNOWN)
557 continue;
558 if (Config->MachineType == IMAGE_FILE_MACHINE_UNKNOWN) {
559 Config->MachineType = MT;
560 continue;
561 }
562 if (Config->MachineType != MT) {
563 llvm::errs() << File->getShortName() << ": machine type "
564 << machineTypeToStr(MT) << " conflicts with "
565 << machineTypeToStr(Config->MachineType) << "\n";
566 return false;
567 }
568 }
569 if (Config->MachineType == IMAGE_FILE_MACHINE_UNKNOWN) {
570 llvm::errs() << "warning: /machine is not specified. x64 is assumed.\n";
571 Config->MachineType = IMAGE_FILE_MACHINE_AMD64;
572 }
573
574 // Windows specific -- Convert Windows resource files to a COFF file.
575 if (!Resources.empty()) {
576 auto MBOrErr = convertResToCOFF(Resources);
577 if (MBOrErr.getError())
578 return false;
579 std::unique_ptr<MemoryBuffer> MB = std::move(MBOrErr.get());
580 Symtab.addFile(createFile(MB->getMemBufferRef()));
581 OwningMBs.push_back(std::move(MB)); // take ownership
582 }
583
584 if (!Entry.empty())
585 Config->Entry = addUndefined(mangle(Entry));
586 Symtab.addAbsolute(mangle("__ImageBase"), Config->ImageBase);
587
Rui Ueyama45044f42015-06-29 01:03:53 +0000588 // Windows specific -- If entry point name is not given, we need to
589 // infer that from user-defined entry name.
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000590 if (Entry.empty() && !Config->NoEntry) {
Rui Ueyama45044f42015-06-29 01:03:53 +0000591 StringRef S = findDefaultEntry();
592 if (S.empty()) {
593 llvm::errs() << "entry point must be defined\n";
594 return false;
595 }
Rui Ueyama6bf638e2015-07-02 00:04:14 +0000596 Config->Entry = addUndefined(S);
Rui Ueyama85225b02015-07-02 03:15:15 +0000597 if (Config->Verbose)
598 llvm::outs() << "Entry name inferred: " << S << "\n";
Rui Ueyama45044f42015-06-29 01:03:53 +0000599 }
600
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000601 // Read as much files as we can from directives sections.
Rui Ueyama85225b02015-07-02 03:15:15 +0000602 if (auto EC = Symtab.run()) {
603 llvm::errs() << EC.message() << "\n";
604 return false;
605 }
606
607 // Resolve auxiliary symbols until we get a convergence.
Rui Ueyama573bf7d2015-06-19 21:12:48 +0000608 // (Trying to resolve a symbol may trigger a Lazy symbol to load a new file.
609 // A new file may contain a directive section to add new command line options.
610 // That's why we have to repeat until converge.)
611 for (;;) {
Rui Ueyama6bf638e2015-07-02 00:04:14 +0000612 // Windows specific -- if entry point is not found,
613 // search for its mangled names.
614 if (Config->Entry)
615 Symtab.mangleMaybe(Config->Entry);
616
Rui Ueyama573bf7d2015-06-19 21:12:48 +0000617 // Windows specific -- Make sure we resolve all dllexported symbols.
Rui Ueyama6bf638e2015-07-02 00:04:14 +0000618 for (Export &E : Config->Exports) {
619 E.Sym = addUndefined(E.Name);
620 Symtab.mangleMaybe(E.Sym);
621 }
Rui Ueyama573bf7d2015-06-19 21:12:48 +0000622
623 // Add weak aliases. Weak aliases is a mechanism to give remaining
624 // undefined symbols final chance to be resolved successfully.
Rui Ueyama3d4c69c2015-07-02 02:38:59 +0000625 for (auto Pair : Config->AlternateNames) {
626 StringRef From = Pair.first;
627 StringRef To = Pair.second;
Rui Ueyama458d7442015-07-02 03:59:04 +0000628 Symbol *Sym = Symtab.find(From);
Rui Ueyama3d4c69c2015-07-02 02:38:59 +0000629 if (!Sym)
630 continue;
Rui Ueyama183f53f2015-07-06 17:45:22 +0000631 if (auto *U = dyn_cast<Undefined>(Sym->Body))
Rui Ueyama3d4c69c2015-07-02 02:38:59 +0000632 if (!U->WeakAlias)
633 U->WeakAlias = Symtab.addUndefined(To);
Rui Ueyama2edb35a2015-06-18 19:09:30 +0000634 }
Rui Ueyama573bf7d2015-06-19 21:12:48 +0000635
Rui Ueyama3d4c69c2015-07-02 02:38:59 +0000636 if (Symtab.queueEmpty())
637 break;
Rui Ueyama0d2e9992015-06-23 23:56:39 +0000638 if (auto EC = Symtab.run()) {
639 llvm::errs() << EC.message() << "\n";
640 return false;
641 }
Rui Ueyama2edb35a2015-06-18 19:09:30 +0000642 }
643
Rui Ueyamaeb262ce2015-06-04 02:12:16 +0000644 // Do LTO by compiling bitcode input files to a native COFF file
645 // then link that file.
Peter Collingbourne60c16162015-06-01 20:10:10 +0000646 if (auto EC = Symtab.addCombinedLTOObject()) {
647 llvm::errs() << EC.message() << "\n";
648 return false;
649 }
650
Peter Collingbourne2612a322015-07-04 05:28:41 +0000651 // Make sure we have resolved all symbols.
652 if (Symtab.reportRemainingUndefines(/*Resolve=*/true))
653 return false;
654
Rui Ueyama3ee0fe42015-05-31 03:55:46 +0000655 // Windows specific -- if no /subsystem is given, we need to infer
656 // that from entry point name.
657 if (Config->Subsystem == IMAGE_SUBSYSTEM_UNKNOWN) {
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000658 Config->Subsystem = inferSubsystem();
Rui Ueyama3ee0fe42015-05-31 03:55:46 +0000659 if (Config->Subsystem == IMAGE_SUBSYSTEM_UNKNOWN) {
660 llvm::errs() << "subsystem must be defined\n";
661 return false;
662 }
663 }
664
Rui Ueyama151d8622015-06-17 20:40:43 +0000665 // Windows specific -- when we are creating a .dll file, we also
666 // need to create a .lib file.
667 if (!Config->Exports.empty())
668 writeImportLibrary();
669
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000670 // Windows specific -- fix up dllexported symbols.
Rui Ueyamaf5313b32015-06-28 22:16:41 +0000671 if (!Config->Exports.empty())
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000672 if (fixupExports())
673 return false;
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000674
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000675 // Windows specific -- Create a side-by-side manifest file.
676 if (Config->Manifest == Configuration::SideBySide)
677 if (createSideBySideManifest())
678 return false;
679
Rui Ueyama0fc26d22015-06-29 14:27:12 +0000680 // Create a dummy PDB file to satisfy build sytem rules.
681 if (auto *Arg = Args.getLastArg(OPT_pdb))
682 touchFile(Arg->getValue());
683
Rui Ueyama411c63602015-05-28 19:09:30 +0000684 // Write the result.
685 Writer Out(&Symtab);
Rui Ueyamaad660982015-06-07 00:20:32 +0000686 if (auto EC = Out.write(Config->OutputFile)) {
Rui Ueyama411c63602015-05-28 19:09:30 +0000687 llvm::errs() << EC.message() << "\n";
688 return false;
689 }
Peter Collingbournebe549552015-06-26 18:58:24 +0000690
Rui Ueyama016414f2015-06-28 20:07:08 +0000691 // Create a symbol map file containing symbol VAs and their names
692 // to help debugging.
Peter Collingbournebe549552015-06-26 18:58:24 +0000693 if (auto *Arg = Args.getLastArg(OPT_lldmap)) {
694 std::error_code EC;
Peter Collingbournebaf5f872015-06-26 19:20:09 +0000695 llvm::raw_fd_ostream Out(Arg->getValue(), EC, OpenFlags::F_Text);
Peter Collingbournebe549552015-06-26 18:58:24 +0000696 if (EC) {
697 llvm::errs() << EC.message() << "\n";
698 return false;
699 }
700 Symtab.printMap(Out);
701 }
Rui Ueyamaa51ce712015-07-03 05:31:35 +0000702 // Call exit to avoid calling destructors.
703 exit(0);
Rui Ueyama411c63602015-05-28 19:09:30 +0000704}
705
Rui Ueyama411c63602015-05-28 19:09:30 +0000706} // namespace coff
707} // namespace lld