blob: 6d9741b8326c08d49627a047beb58a6436e7add1 [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;
Rui Ueyamabbdec4f2015-07-09 22:51:41 +0000111 if (!Config->is64() && E->ExtName.startswith("_"))
112 E->ExtName = E->ExtName.substr(1);
Rui Ueyama562daa82015-06-18 21:50:38 +0000113 Config->Exports.push_back(E.get());
114 break;
115 }
116 case OPT_failifmismatch:
117 if (auto EC = checkFailIfMismatch(Arg->getValue()))
118 return EC;
119 break;
Rui Ueyama08d5e182015-06-18 23:20:11 +0000120 case OPT_incl:
Rui Ueyama32f8e1c2015-06-26 03:44:00 +0000121 addUndefined(Arg->getValue());
Rui Ueyama08d5e182015-06-18 23:20:11 +0000122 break;
Rui Ueyamace86c992015-06-18 23:22:39 +0000123 case OPT_merge:
Rui Ueyama6600eb12015-07-04 23:37:32 +0000124 if (auto EC = parseMerge(Arg->getValue()))
125 return EC;
Rui Ueyamace86c992015-06-18 23:22:39 +0000126 break;
127 case OPT_nodefaultlib:
128 Config->NoDefaultLibs.insert(doFindLib(Arg->getValue()));
129 break;
Rui Ueyama562daa82015-06-18 21:50:38 +0000130 default:
131 llvm::errs() << Arg->getSpelling() << " is not allowed in .drectve\n";
132 return make_error_code(LLDError::InvalidOption);
Rui Ueyamad7c2f582015-05-31 21:04:56 +0000133 }
134 }
Rui Ueyama411c63602015-05-28 19:09:30 +0000135 return std::error_code();
136}
137
Rui Ueyama54b71da2015-05-31 19:17:12 +0000138// Find file from search paths. You can omit ".obj", this function takes
139// care of that. Note that the returned path is not guaranteed to exist.
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000140StringRef LinkerDriver::doFindFile(StringRef Filename) {
Rui Ueyama54b71da2015-05-31 19:17:12 +0000141 bool hasPathSep = (Filename.find_first_of("/\\") != StringRef::npos);
142 if (hasPathSep)
143 return Filename;
144 bool hasExt = (Filename.find('.') != StringRef::npos);
145 for (StringRef Dir : SearchPaths) {
146 SmallString<128> Path = Dir;
147 llvm::sys::path::append(Path, Filename);
148 if (llvm::sys::fs::exists(Path.str()))
149 return Alloc.save(Path.str());
150 if (!hasExt) {
151 Path.append(".obj");
152 if (llvm::sys::fs::exists(Path.str()))
153 return Alloc.save(Path.str());
154 }
155 }
156 return Filename;
157}
158
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000159// Resolves a file path. This never returns the same path
160// (in that case, it returns None).
161Optional<StringRef> LinkerDriver::findFile(StringRef Filename) {
162 StringRef Path = doFindFile(Filename);
163 bool Seen = !VisitedFiles.insert(Path.lower()).second;
164 if (Seen)
165 return None;
166 return Path;
Rui Ueyama54b71da2015-05-31 19:17:12 +0000167}
168
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000169// Find library file from search path.
170StringRef LinkerDriver::doFindLib(StringRef Filename) {
171 // Add ".lib" to Filename if that has no file extension.
Rui Ueyama54b71da2015-05-31 19:17:12 +0000172 bool hasExt = (Filename.find('.') != StringRef::npos);
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000173 if (!hasExt)
174 Filename = Alloc.save(Filename + ".lib");
175 return doFindFile(Filename);
176}
177
178// Resolves a library path. /nodefaultlib options are taken into
179// consideration. This never returns the same path (in that case,
180// it returns None).
181Optional<StringRef> LinkerDriver::findLib(StringRef Filename) {
182 if (Config->NoDefaultLibAll)
183 return None;
184 StringRef Path = doFindLib(Filename);
185 if (Config->NoDefaultLibs.count(Path))
186 return None;
187 bool Seen = !VisitedFiles.insert(Path.lower()).second;
188 if (Seen)
189 return None;
190 return Path;
Rui Ueyama54b71da2015-05-31 19:17:12 +0000191}
192
193// Parses LIB environment which contains a list of search paths.
Rui Ueyamaf00df0a2015-06-19 22:39:48 +0000194void LinkerDriver::addLibSearchPaths() {
Rui Ueyama54b71da2015-05-31 19:17:12 +0000195 Optional<std::string> EnvOpt = Process::GetEnv("LIB");
196 if (!EnvOpt.hasValue())
Rui Ueyamaf00df0a2015-06-19 22:39:48 +0000197 return;
Rui Ueyama54b71da2015-05-31 19:17:12 +0000198 StringRef Env = Alloc.save(*EnvOpt);
199 while (!Env.empty()) {
200 StringRef Path;
201 std::tie(Path, Env) = Env.split(';');
Rui Ueyamaf00df0a2015-06-19 22:39:48 +0000202 SearchPaths.push_back(Path);
Rui Ueyama54b71da2015-05-31 19:17:12 +0000203 }
Rui Ueyama54b71da2015-05-31 19:17:12 +0000204}
205
Rui Ueyama6bf638e2015-07-02 00:04:14 +0000206Undefined *LinkerDriver::addUndefined(StringRef Name) {
207 Undefined *U = Symtab.addUndefined(Name);
Rui Ueyama18f8d2c2015-07-02 00:21:08 +0000208 Config->GCRoot.insert(U);
Rui Ueyama6bf638e2015-07-02 00:04:14 +0000209 return U;
Rui Ueyama32f8e1c2015-06-26 03:44:00 +0000210}
211
Rui Ueyama7c3e23f2015-07-09 01:25:49 +0000212// Symbol names are mangled by appending "_" prefix on x86.
213StringRef LinkerDriver::mangle(StringRef Sym) {
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000214 assert(Config->MachineType != IMAGE_FILE_MACHINE_UNKNOWN);
Rui Ueyama7c3e23f2015-07-09 01:25:49 +0000215 if (Config->MachineType == IMAGE_FILE_MACHINE_I386)
216 return Alloc.save("_" + Sym);
217 return Sym;
218}
219
Rui Ueyama45044f42015-06-29 01:03:53 +0000220// Windows specific -- find default entry point name.
221StringRef LinkerDriver::findDefaultEntry() {
222 // User-defined main functions and their corresponding entry points.
223 static const char *Entries[][2] = {
224 {"main", "mainCRTStartup"},
225 {"wmain", "wmainCRTStartup"},
226 {"WinMain", "WinMainCRTStartup"},
227 {"wWinMain", "wWinMainCRTStartup"},
228 };
229 for (auto E : Entries) {
Rui Ueyama7c3e23f2015-07-09 01:25:49 +0000230 Symbol *Sym = Symtab.find(mangle(E[0]));
Rui Ueyama183f53f2015-07-06 17:45:22 +0000231 if (Sym && !isa<Undefined>(Sym->Body))
Rui Ueyama7c3e23f2015-07-09 01:25:49 +0000232 return mangle(E[1]);
Rui Ueyama45044f42015-06-29 01:03:53 +0000233 }
234 return "";
235}
236
237WindowsSubsystem LinkerDriver::inferSubsystem() {
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000238 if (Config->DLL)
239 return IMAGE_SUBSYSTEM_WINDOWS_GUI;
Rui Ueyama7c3e23f2015-07-09 01:25:49 +0000240 if (Symtab.find(mangle("main")) || Symtab.find(mangle("wmain")))
Rui Ueyama45044f42015-06-29 01:03:53 +0000241 return IMAGE_SUBSYSTEM_WINDOWS_CUI;
Rui Ueyama7c3e23f2015-07-09 01:25:49 +0000242 if (Symtab.find(mangle("WinMain")) || Symtab.find(mangle("wWinMain")))
Rui Ueyama45044f42015-06-29 01:03:53 +0000243 return IMAGE_SUBSYSTEM_WINDOWS_GUI;
244 return IMAGE_SUBSYSTEM_UNKNOWN;
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000245}
246
David Blaikie00818192015-06-22 22:06:48 +0000247bool LinkerDriver::link(llvm::ArrayRef<const char *> ArgsArr) {
Peter Collingbourne60c16162015-06-01 20:10:10 +0000248 // Needed for LTO.
249 llvm::InitializeAllTargetInfos();
250 llvm::InitializeAllTargets();
251 llvm::InitializeAllTargetMCs();
252 llvm::InitializeAllAsmParsers();
253 llvm::InitializeAllAsmPrinters();
254 llvm::InitializeAllDisassemblers();
255
Peter Collingbournebd1cb792015-06-09 21:52:48 +0000256 // If the first command line argument is "/lib", link.exe acts like lib.exe.
257 // We call our own implementation of lib.exe that understands bitcode files.
David Blaikieb2b1c7c2015-06-21 06:32:10 +0000258 if (ArgsArr.size() > 1 && StringRef(ArgsArr[1]).equals_lower("/lib"))
259 return llvm::libDriverMain(ArgsArr.slice(1)) == 0;
Peter Collingbournebd1cb792015-06-09 21:52:48 +0000260
Rui Ueyama411c63602015-05-28 19:09:30 +0000261 // Parse command line options.
Rui Ueyama9d72f092015-06-28 03:05:38 +0000262 auto ArgsOrErr = Parser.parseLINK(ArgsArr.slice(1));
Rui Ueyama411c63602015-05-28 19:09:30 +0000263 if (auto EC = ArgsOrErr.getError()) {
264 llvm::errs() << EC.message() << "\n";
265 return false;
266 }
David Blaikie6521ed92015-06-22 22:06:52 +0000267 llvm::opt::InputArgList Args = std::move(ArgsOrErr.get());
Rui Ueyama411c63602015-05-28 19:09:30 +0000268
Rui Ueyama5c726432015-05-29 16:11:52 +0000269 // Handle /help
David Blaikie6521ed92015-06-22 22:06:52 +0000270 if (Args.hasArg(OPT_help)) {
David Blaikieb2b1c7c2015-06-21 06:32:10 +0000271 printHelp(ArgsArr[0]);
Rui Ueyama5c726432015-05-29 16:11:52 +0000272 return true;
273 }
274
David Blaikie6521ed92015-06-22 22:06:52 +0000275 if (Args.filtered_begin(OPT_INPUT) == Args.filtered_end()) {
Rui Ueyama411c63602015-05-28 19:09:30 +0000276 llvm::errs() << "no input files.\n";
277 return false;
278 }
Rui Ueyama3d3e6fb2015-05-29 16:06:00 +0000279
Rui Ueyamaf00df0a2015-06-19 22:39:48 +0000280 // Construct search path list.
281 SearchPaths.push_back("");
David Blaikie6521ed92015-06-22 22:06:52 +0000282 for (auto *Arg : Args.filtered(OPT_libpath))
Rui Ueyamaf00df0a2015-06-19 22:39:48 +0000283 SearchPaths.push_back(Arg->getValue());
284 addLibSearchPaths();
285
Rui Ueyamaad660982015-06-07 00:20:32 +0000286 // Handle /out
David Blaikie6521ed92015-06-22 22:06:52 +0000287 if (auto *Arg = Args.getLastArg(OPT_out))
Rui Ueyamaad660982015-06-07 00:20:32 +0000288 Config->OutputFile = Arg->getValue();
289
Rui Ueyama3d3e6fb2015-05-29 16:06:00 +0000290 // Handle /verbose
David Blaikie6521ed92015-06-22 22:06:52 +0000291 if (Args.hasArg(OPT_verbose))
Rui Ueyama411c63602015-05-28 19:09:30 +0000292 Config->Verbose = true;
Rui Ueyama3d3e6fb2015-05-29 16:06:00 +0000293
Rui Ueyama95925fd2015-06-28 19:35:15 +0000294 // Handle /force or /force:unresolved
295 if (Args.hasArg(OPT_force) || Args.hasArg(OPT_force_unresolved))
296 Config->Force = true;
297
Rui Ueyama6600eb12015-07-04 23:37:32 +0000298 // Handle /debug
299 if (Args.hasArg(OPT_debug))
300 Config->Debug = true;
301
Rui Ueyamaa8b60452015-06-28 19:56:30 +0000302 // Handle /noentry
303 if (Args.hasArg(OPT_noentry)) {
304 if (!Args.hasArg(OPT_dll)) {
305 llvm::errs() << "/noentry must be specified with /dll\n";
306 return false;
307 }
308 Config->NoEntry = true;
309 }
310
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000311 // Handle /dll
David Blaikie6521ed92015-06-22 22:06:52 +0000312 if (Args.hasArg(OPT_dll)) {
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000313 Config->DLL = true;
Rui Ueyama49d6cd32015-07-03 00:02:19 +0000314 Config->ImageBase = 0x180000000U;
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000315 Config->ManifestID = 2;
316 }
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000317
Rui Ueyama588e8322015-06-15 01:23:58 +0000318 // Handle /fixed
David Blaikie6521ed92015-06-22 22:06:52 +0000319 if (Args.hasArg(OPT_fixed)) {
320 if (Args.hasArg(OPT_dynamicbase)) {
Rui Ueyama6592ff82015-06-16 23:13:00 +0000321 llvm::errs() << "/fixed must not be specified with /dynamicbase\n";
322 return false;
323 }
Rui Ueyama588e8322015-06-15 01:23:58 +0000324 Config->Relocatable = false;
Rui Ueyama6592ff82015-06-16 23:13:00 +0000325 Config->DynamicBase = false;
326 }
Rui Ueyama588e8322015-06-15 01:23:58 +0000327
Rui Ueyama3d3e6fb2015-05-29 16:06:00 +0000328 // Handle /machine
Rui Ueyamae16a75d52015-07-08 18:14:51 +0000329 if (auto *Arg = Args.getLastArg(OPT_machine)) {
330 ErrorOr<MachineTypes> MTOrErr = getMachineType(Arg->getValue());
331 if (MTOrErr.getError())
332 return false;
333 Config->MachineType = MTOrErr.get();
Rui Ueyama3d3e6fb2015-05-29 16:06:00 +0000334 }
Rui Ueyama3d3e6fb2015-05-29 16:06:00 +0000335
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000336 // Handle /nodefaultlib:<filename>
David Blaikie6521ed92015-06-22 22:06:52 +0000337 for (auto *Arg : Args.filtered(OPT_nodefaultlib))
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000338 Config->NoDefaultLibs.insert(doFindLib(Arg->getValue()));
339
340 // Handle /nodefaultlib
David Blaikie6521ed92015-06-22 22:06:52 +0000341 if (Args.hasArg(OPT_nodefaultlib_all))
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000342 Config->NoDefaultLibAll = true;
343
Rui Ueyama804a8b62015-05-29 16:18:15 +0000344 // Handle /base
David Blaikie6521ed92015-06-22 22:06:52 +0000345 if (auto *Arg = Args.getLastArg(OPT_base)) {
Rui Ueyama804a8b62015-05-29 16:18:15 +0000346 if (auto EC = parseNumbers(Arg->getValue(), &Config->ImageBase)) {
Rui Ueyamab41b7e52015-05-29 16:21:11 +0000347 llvm::errs() << "/base: " << EC.message() << "\n";
348 return false;
349 }
350 }
351
352 // Handle /stack
David Blaikie6521ed92015-06-22 22:06:52 +0000353 if (auto *Arg = Args.getLastArg(OPT_stack)) {
Rui Ueyamab41b7e52015-05-29 16:21:11 +0000354 if (auto EC = parseNumbers(Arg->getValue(), &Config->StackReserve,
355 &Config->StackCommit)) {
356 llvm::errs() << "/stack: " << EC.message() << "\n";
Rui Ueyama804a8b62015-05-29 16:18:15 +0000357 return false;
358 }
359 }
360
Rui Ueyamac377e9a2015-05-29 16:23:40 +0000361 // Handle /heap
David Blaikie6521ed92015-06-22 22:06:52 +0000362 if (auto *Arg = Args.getLastArg(OPT_heap)) {
Rui Ueyamac377e9a2015-05-29 16:23:40 +0000363 if (auto EC = parseNumbers(Arg->getValue(), &Config->HeapReserve,
364 &Config->HeapCommit)) {
365 llvm::errs() << "/heap: " << EC.message() << "\n";
366 return false;
367 }
368 }
369
Rui Ueyamab9dcdb52015-05-29 16:28:29 +0000370 // Handle /version
David Blaikie6521ed92015-06-22 22:06:52 +0000371 if (auto *Arg = Args.getLastArg(OPT_version)) {
Rui Ueyamab9dcdb52015-05-29 16:28:29 +0000372 if (auto EC = parseVersion(Arg->getValue(), &Config->MajorImageVersion,
373 &Config->MinorImageVersion)) {
374 llvm::errs() << "/version: " << EC.message() << "\n";
375 return false;
376 }
377 }
378
Rui Ueyama15cc47e2015-05-29 16:34:31 +0000379 // Handle /subsystem
David Blaikie6521ed92015-06-22 22:06:52 +0000380 if (auto *Arg = Args.getLastArg(OPT_subsystem)) {
Rui Ueyama15cc47e2015-05-29 16:34:31 +0000381 if (auto EC = parseSubsystem(Arg->getValue(), &Config->Subsystem,
382 &Config->MajorOSVersion,
383 &Config->MinorOSVersion)) {
384 llvm::errs() << "/subsystem: " << EC.message() << "\n";
385 return false;
386 }
387 }
388
Rui Ueyama2edb35a2015-06-18 19:09:30 +0000389 // Handle /alternatename
David Blaikie6521ed92015-06-22 22:06:52 +0000390 for (auto *Arg : Args.filtered(OPT_alternatename))
Rui Ueyama2edb35a2015-06-18 19:09:30 +0000391 if (parseAlternateName(Arg->getValue()))
392 return false;
393
Rui Ueyama08d5e182015-06-18 23:20:11 +0000394 // Handle /include
David Blaikie6521ed92015-06-22 22:06:52 +0000395 for (auto *Arg : Args.filtered(OPT_incl))
Rui Ueyama32f8e1c2015-06-26 03:44:00 +0000396 addUndefined(Arg->getValue());
Rui Ueyama08d5e182015-06-18 23:20:11 +0000397
Rui Ueyamab95188c2015-06-18 20:27:09 +0000398 // Handle /implib
David Blaikie6521ed92015-06-22 22:06:52 +0000399 if (auto *Arg = Args.getLastArg(OPT_implib))
Rui Ueyamab95188c2015-06-18 20:27:09 +0000400 Config->Implib = Arg->getValue();
401
Rui Ueyamae2cbfea2015-06-07 03:17:42 +0000402 // Handle /opt
David Blaikie6521ed92015-06-22 22:06:52 +0000403 for (auto *Arg : Args.filtered(OPT_opt)) {
Rui Ueyamae2cbfea2015-06-07 03:17:42 +0000404 std::string S = StringRef(Arg->getValue()).lower();
405 if (S == "noref") {
406 Config->DoGC = false;
407 continue;
408 }
Rui Ueyamaf799ede2015-06-25 23:26:58 +0000409 if (S == "lldicf") {
Rui Ueyamaddf71fc2015-06-24 04:36:52 +0000410 Config->ICF = true;
411 continue;
412 }
Rui Ueyamae2cbfea2015-06-07 03:17:42 +0000413 if (S != "ref" && S != "icf" && S != "noicf" &&
414 S != "lbr" && S != "nolbr" &&
415 !StringRef(S).startswith("icf=")) {
416 llvm::errs() << "/opt: unknown option: " << S << "\n";
417 return false;
418 }
419 }
420
Rui Ueyamaa77336b2015-06-21 22:31:52 +0000421 // Handle /delayload
David Blaikie6521ed92015-06-22 22:06:52 +0000422 for (auto *Arg : Args.filtered(OPT_delayload)) {
Rui Ueyama7a247ee2015-07-03 01:40:14 +0000423 Config->DelayLoads.insert(StringRef(Arg->getValue()).lower());
Rui Ueyama32f8e1c2015-06-26 03:44:00 +0000424 addUndefined("__delayLoadHelper2");
Rui Ueyamaa77336b2015-06-21 22:31:52 +0000425 }
426
Rui Ueyama8854d8a2015-06-04 19:21:24 +0000427 // Handle /failifmismatch
David Blaikie6521ed92015-06-22 22:06:52 +0000428 for (auto *Arg : Args.filtered(OPT_failifmismatch))
Rui Ueyama75b098b2015-06-18 21:23:34 +0000429 if (checkFailIfMismatch(Arg->getValue()))
430 return false;
Rui Ueyama8854d8a2015-06-04 19:21:24 +0000431
Rui Ueyama6600eb12015-07-04 23:37:32 +0000432 // Handle /merge
433 for (auto *Arg : Args.filtered(OPT_merge))
434 if (parseMerge(Arg->getValue()))
435 return false;
436
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000437 // Handle /manifest
David Blaikie6521ed92015-06-22 22:06:52 +0000438 if (auto *Arg = Args.getLastArg(OPT_manifest_colon)) {
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000439 if (auto EC = parseManifest(Arg->getValue())) {
440 llvm::errs() << "/manifest: " << EC.message() << "\n";
441 return false;
442 }
443 }
444
445 // Handle /manifestuac
David Blaikie6521ed92015-06-22 22:06:52 +0000446 if (auto *Arg = Args.getLastArg(OPT_manifestuac)) {
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000447 if (auto EC = parseManifestUAC(Arg->getValue())) {
448 llvm::errs() << "/manifestuac: " << EC.message() << "\n";
449 return false;
450 }
451 }
452
453 // Handle /manifestdependency
David Blaikie6521ed92015-06-22 22:06:52 +0000454 if (auto *Arg = Args.getLastArg(OPT_manifestdependency))
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000455 Config->ManifestDependency = Arg->getValue();
456
457 // Handle /manifestfile
David Blaikie6521ed92015-06-22 22:06:52 +0000458 if (auto *Arg = Args.getLastArg(OPT_manifestfile))
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000459 Config->ManifestFile = Arg->getValue();
460
Rui Ueyama6592ff82015-06-16 23:13:00 +0000461 // Handle miscellaneous boolean flags.
David Blaikie6521ed92015-06-22 22:06:52 +0000462 if (Args.hasArg(OPT_allowbind_no))
463 Config->AllowBind = false;
464 if (Args.hasArg(OPT_allowisolation_no))
465 Config->AllowIsolation = false;
466 if (Args.hasArg(OPT_dynamicbase_no))
467 Config->DynamicBase = false;
468 if (Args.hasArg(OPT_highentropyva_no))
469 Config->HighEntropyVA = false;
470 if (Args.hasArg(OPT_nxcompat_no))
471 Config->NxCompat = false;
472 if (Args.hasArg(OPT_tsaware_no))
473 Config->TerminalServerAware = false;
Rui Ueyama6592ff82015-06-16 23:13:00 +0000474
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000475 // Create a list of input files. Files can be given as arguments
476 // for /defaultlib option.
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000477 std::vector<StringRef> Paths;
478 std::vector<MemoryBufferRef> MBs;
David Blaikie6521ed92015-06-22 22:06:52 +0000479 for (auto *Arg : Args.filtered(OPT_INPUT))
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000480 if (Optional<StringRef> Path = findFile(Arg->getValue()))
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000481 Paths.push_back(*Path);
David Blaikie6521ed92015-06-22 22:06:52 +0000482 for (auto *Arg : Args.filtered(OPT_defaultlib))
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000483 if (Optional<StringRef> Path = findLib(Arg->getValue()))
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000484 Paths.push_back(*Path);
485 for (StringRef Path : Paths) {
Rui Ueyama2bf6a122015-06-14 21:50:50 +0000486 ErrorOr<MemoryBufferRef> MBOrErr = openFile(Path);
487 if (auto EC = MBOrErr.getError()) {
488 llvm::errs() << "cannot open " << Path << ": " << EC.message() << "\n";
489 return false;
490 }
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000491 MBs.push_back(MBOrErr.get());
Rui Ueyama2bf6a122015-06-14 21:50:50 +0000492 }
Rui Ueyamad21b00b2015-05-31 19:17:14 +0000493
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000494 // Windows specific -- Create a resource file containing a manifest file.
495 if (Config->Manifest == Configuration::Embed) {
496 auto MBOrErr = createManifestRes();
497 if (MBOrErr.getError())
498 return false;
499 std::unique_ptr<MemoryBuffer> MB = std::move(MBOrErr.get());
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000500 MBs.push_back(MB->getMemBufferRef());
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000501 OwningMBs.push_back(std::move(MB)); // take ownership
502 }
503
Rui Ueyama2bf6a122015-06-14 21:50:50 +0000504 // Windows specific -- Input files can be Windows resource files (.res files).
505 // We invoke cvtres.exe to convert resource files to a regular COFF file
506 // then link the result file normally.
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000507 std::vector<MemoryBufferRef> Resources;
Rui Ueyama77731b42015-06-26 23:59:13 +0000508 auto NotResource = [](MemoryBufferRef MB) {
509 return identify_magic(MB.getBuffer()) != file_magic::windows_resource;
Rui Ueyama2bf6a122015-06-14 21:50:50 +0000510 };
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000511 auto It = std::stable_partition(MBs.begin(), MBs.end(), NotResource);
512 if (It != MBs.end()) {
513 Resources.insert(Resources.end(), It, MBs.end());
514 MBs.erase(It, MBs.end());
Rui Ueyama2bf6a122015-06-14 21:50:50 +0000515 }
516
Rui Ueyama85225b02015-07-02 03:15:15 +0000517 // Read all input files given via the command line. Note that step()
518 // doesn't read files that are specified by directive sections.
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000519 for (MemoryBufferRef MB : MBs)
Rui Ueyama0d2e9992015-06-23 23:56:39 +0000520 Symtab.addFile(createFile(MB));
Rui Ueyama85225b02015-07-02 03:15:15 +0000521 if (auto EC = Symtab.step()) {
Rui Ueyama0d2e9992015-06-23 23:56:39 +0000522 llvm::errs() << EC.message() << "\n";
523 return false;
Rui Ueyama411c63602015-05-28 19:09:30 +0000524 }
Rui Ueyama5cff6852015-05-31 03:34:08 +0000525
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000526 // Determine machine type and check if all object files are
527 // for the same CPU type. Note that this needs to be done before
528 // any call to mangle().
529 for (std::unique_ptr<InputFile> &File : Symtab.getFiles()) {
530 MachineTypes MT = File->getMachineType();
531 if (MT == IMAGE_FILE_MACHINE_UNKNOWN)
532 continue;
533 if (Config->MachineType == IMAGE_FILE_MACHINE_UNKNOWN) {
534 Config->MachineType = MT;
535 continue;
536 }
537 if (Config->MachineType != MT) {
538 llvm::errs() << File->getShortName() << ": machine type "
539 << machineTypeToStr(MT) << " conflicts with "
540 << machineTypeToStr(Config->MachineType) << "\n";
541 return false;
542 }
543 }
544 if (Config->MachineType == IMAGE_FILE_MACHINE_UNKNOWN) {
545 llvm::errs() << "warning: /machine is not specified. x64 is assumed.\n";
546 Config->MachineType = IMAGE_FILE_MACHINE_AMD64;
547 }
548
549 // Windows specific -- Convert Windows resource files to a COFF file.
550 if (!Resources.empty()) {
551 auto MBOrErr = convertResToCOFF(Resources);
552 if (MBOrErr.getError())
553 return false;
554 std::unique_ptr<MemoryBuffer> MB = std::move(MBOrErr.get());
555 Symtab.addFile(createFile(MB->getMemBufferRef()));
556 OwningMBs.push_back(std::move(MB)); // take ownership
557 }
558
Rui Ueyamabbdec4f2015-07-09 22:51:41 +0000559 // Handle /entry and /dll
560 if (auto *Arg = Args.getLastArg(OPT_entry)) {
561 Config->Entry = addUndefined(mangle(Arg->getValue()));
562 } else if (Args.hasArg(OPT_dll) && !Config->NoEntry) {
563 StringRef S =
564 Config->is64() ? "_DllMainCRTStartup" : "__DllMainCRTStartup@12";
565 Config->Entry = addUndefined(S);
566 } else if (!Config->NoEntry) {
567 // Windows specific -- If entry point name is not given, we need to
568 // infer that from user-defined entry name.
Rui Ueyama45044f42015-06-29 01:03:53 +0000569 StringRef S = findDefaultEntry();
570 if (S.empty()) {
571 llvm::errs() << "entry point must be defined\n";
572 return false;
573 }
Rui Ueyama6bf638e2015-07-02 00:04:14 +0000574 Config->Entry = addUndefined(S);
Rui Ueyama85225b02015-07-02 03:15:15 +0000575 if (Config->Verbose)
576 llvm::outs() << "Entry name inferred: " << S << "\n";
Rui Ueyama45044f42015-06-29 01:03:53 +0000577 }
578
Rui Ueyamabbdec4f2015-07-09 22:51:41 +0000579 // Handle /export
580 for (auto *Arg : Args.filtered(OPT_export)) {
581 ErrorOr<Export> E = parseExport(Arg->getValue());
582 if (E.getError())
583 return false;
584 if (!Config->is64() && !E->Name.startswith("_@?"))
585 E->Name = mangle(E->Name);
586 Config->Exports.push_back(E.get());
587 }
588
589 // Handle /def
590 if (auto *Arg = Args.getLastArg(OPT_deffile)) {
591 ErrorOr<MemoryBufferRef> MBOrErr = openFile(Arg->getValue());
592 if (auto EC = MBOrErr.getError()) {
593 llvm::errs() << "/def: " << EC.message() << "\n";
594 return false;
595 }
596 // parseModuleDefs mutates Config object.
597 if (parseModuleDefs(MBOrErr.get(), &Alloc))
598 return false;
599 }
600
601 Symtab.addAbsolute(mangle("__ImageBase"), Config->ImageBase);
602
Rui Ueyamaea533cd2015-07-09 19:54:13 +0000603 // Read as much files as we can from directives sections.
Rui Ueyama85225b02015-07-02 03:15:15 +0000604 if (auto EC = Symtab.run()) {
605 llvm::errs() << EC.message() << "\n";
606 return false;
607 }
608
609 // Resolve auxiliary symbols until we get a convergence.
Rui Ueyama573bf7d2015-06-19 21:12:48 +0000610 // (Trying to resolve a symbol may trigger a Lazy symbol to load a new file.
611 // A new file may contain a directive section to add new command line options.
612 // That's why we have to repeat until converge.)
613 for (;;) {
Rui Ueyama6bf638e2015-07-02 00:04:14 +0000614 // Windows specific -- if entry point is not found,
615 // search for its mangled names.
616 if (Config->Entry)
617 Symtab.mangleMaybe(Config->Entry);
618
Rui Ueyama573bf7d2015-06-19 21:12:48 +0000619 // Windows specific -- Make sure we resolve all dllexported symbols.
Rui Ueyama6bf638e2015-07-02 00:04:14 +0000620 for (Export &E : Config->Exports) {
621 E.Sym = addUndefined(E.Name);
622 Symtab.mangleMaybe(E.Sym);
623 }
Rui Ueyama573bf7d2015-06-19 21:12:48 +0000624
625 // Add weak aliases. Weak aliases is a mechanism to give remaining
626 // undefined symbols final chance to be resolved successfully.
Rui Ueyama3d4c69c2015-07-02 02:38:59 +0000627 for (auto Pair : Config->AlternateNames) {
628 StringRef From = Pair.first;
629 StringRef To = Pair.second;
Rui Ueyama458d7442015-07-02 03:59:04 +0000630 Symbol *Sym = Symtab.find(From);
Rui Ueyama3d4c69c2015-07-02 02:38:59 +0000631 if (!Sym)
632 continue;
Rui Ueyama183f53f2015-07-06 17:45:22 +0000633 if (auto *U = dyn_cast<Undefined>(Sym->Body))
Rui Ueyama3d4c69c2015-07-02 02:38:59 +0000634 if (!U->WeakAlias)
635 U->WeakAlias = Symtab.addUndefined(To);
Rui Ueyama2edb35a2015-06-18 19:09:30 +0000636 }
Rui Ueyama573bf7d2015-06-19 21:12:48 +0000637
Rui Ueyama3d4c69c2015-07-02 02:38:59 +0000638 if (Symtab.queueEmpty())
639 break;
Rui Ueyama0d2e9992015-06-23 23:56:39 +0000640 if (auto EC = Symtab.run()) {
641 llvm::errs() << EC.message() << "\n";
642 return false;
643 }
Rui Ueyama2edb35a2015-06-18 19:09:30 +0000644 }
645
Rui Ueyamaeb262ce2015-06-04 02:12:16 +0000646 // Do LTO by compiling bitcode input files to a native COFF file
647 // then link that file.
Peter Collingbourne60c16162015-06-01 20:10:10 +0000648 if (auto EC = Symtab.addCombinedLTOObject()) {
649 llvm::errs() << EC.message() << "\n";
650 return false;
651 }
652
Peter Collingbourne2612a322015-07-04 05:28:41 +0000653 // Make sure we have resolved all symbols.
654 if (Symtab.reportRemainingUndefines(/*Resolve=*/true))
655 return false;
656
Rui Ueyama3ee0fe42015-05-31 03:55:46 +0000657 // Windows specific -- if no /subsystem is given, we need to infer
658 // that from entry point name.
659 if (Config->Subsystem == IMAGE_SUBSYSTEM_UNKNOWN) {
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000660 Config->Subsystem = inferSubsystem();
Rui Ueyama3ee0fe42015-05-31 03:55:46 +0000661 if (Config->Subsystem == IMAGE_SUBSYSTEM_UNKNOWN) {
662 llvm::errs() << "subsystem must be defined\n";
663 return false;
664 }
665 }
666
Rui Ueyama151d8622015-06-17 20:40:43 +0000667 // Windows specific -- when we are creating a .dll file, we also
668 // need to create a .lib file.
669 if (!Config->Exports.empty())
670 writeImportLibrary();
671
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000672 // Windows specific -- fix up dllexported symbols.
Rui Ueyamaf5313b32015-06-28 22:16:41 +0000673 if (!Config->Exports.empty())
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000674 if (fixupExports())
675 return false;
Rui Ueyama97dff9e2015-06-17 00:16:33 +0000676
Rui Ueyama24c5fd02015-06-18 00:12:42 +0000677 // Windows specific -- Create a side-by-side manifest file.
678 if (Config->Manifest == Configuration::SideBySide)
679 if (createSideBySideManifest())
680 return false;
681
Rui Ueyama0fc26d22015-06-29 14:27:12 +0000682 // Create a dummy PDB file to satisfy build sytem rules.
683 if (auto *Arg = Args.getLastArg(OPT_pdb))
684 touchFile(Arg->getValue());
685
Rui Ueyama411c63602015-05-28 19:09:30 +0000686 // Write the result.
687 Writer Out(&Symtab);
Rui Ueyamaad660982015-06-07 00:20:32 +0000688 if (auto EC = Out.write(Config->OutputFile)) {
Rui Ueyama411c63602015-05-28 19:09:30 +0000689 llvm::errs() << EC.message() << "\n";
690 return false;
691 }
Peter Collingbournebe549552015-06-26 18:58:24 +0000692
Rui Ueyama016414f2015-06-28 20:07:08 +0000693 // Create a symbol map file containing symbol VAs and their names
694 // to help debugging.
Peter Collingbournebe549552015-06-26 18:58:24 +0000695 if (auto *Arg = Args.getLastArg(OPT_lldmap)) {
696 std::error_code EC;
Peter Collingbournebaf5f872015-06-26 19:20:09 +0000697 llvm::raw_fd_ostream Out(Arg->getValue(), EC, OpenFlags::F_Text);
Peter Collingbournebe549552015-06-26 18:58:24 +0000698 if (EC) {
699 llvm::errs() << EC.message() << "\n";
700 return false;
701 }
702 Symtab.printMap(Out);
703 }
Rui Ueyamaa51ce712015-07-03 05:31:35 +0000704 // Call exit to avoid calling destructors.
705 exit(0);
Rui Ueyama411c63602015-05-28 19:09:30 +0000706}
707
Rui Ueyama411c63602015-05-28 19:09:30 +0000708} // namespace coff
709} // namespace lld