2015-05-29 03:09:30 +08:00
|
|
|
//===- Driver.cpp ---------------------------------------------------------===//
|
|
|
|
//
|
|
|
|
// The LLVM Linker
|
|
|
|
//
|
|
|
|
// This file is distributed under the University of Illinois Open Source
|
|
|
|
// License. See LICENSE.TXT for details.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
|
|
|
#include "Config.h"
|
|
|
|
#include "Driver.h"
|
2015-06-19 05:50:38 +08:00
|
|
|
#include "Error.h"
|
2015-05-29 03:09:30 +08:00
|
|
|
#include "InputFiles.h"
|
|
|
|
#include "SymbolTable.h"
|
2015-08-06 07:43:53 +08:00
|
|
|
#include "Symbols.h"
|
2015-05-29 03:09:30 +08:00
|
|
|
#include "Writer.h"
|
2016-03-03 03:08:05 +08:00
|
|
|
#include "lld/Driver/Driver.h"
|
2015-05-29 03:09:30 +08:00
|
|
|
#include "llvm/ADT/Optional.h"
|
2015-06-10 05:52:48 +08:00
|
|
|
#include "llvm/LibDriver/LibDriver.h"
|
2015-05-29 03:09:30 +08:00
|
|
|
#include "llvm/Option/Arg.h"
|
|
|
|
#include "llvm/Option/ArgList.h"
|
|
|
|
#include "llvm/Option/Option.h"
|
|
|
|
#include "llvm/Support/Debug.h"
|
|
|
|
#include "llvm/Support/Path.h"
|
2015-06-01 03:17:12 +08:00
|
|
|
#include "llvm/Support/Process.h"
|
2015-06-02 04:10:10 +08:00
|
|
|
#include "llvm/Support/TargetSelect.h"
|
2015-05-29 03:09:30 +08:00
|
|
|
#include "llvm/Support/raw_ostream.h"
|
2015-06-15 05:50:50 +08:00
|
|
|
#include <algorithm>
|
2015-05-29 03:09:30 +08:00
|
|
|
#include <memory>
|
|
|
|
|
|
|
|
using namespace llvm;
|
2015-07-08 07:39:18 +08:00
|
|
|
using namespace llvm::COFF;
|
2015-06-01 03:17:12 +08:00
|
|
|
using llvm::sys::Process;
|
2015-06-27 03:20:09 +08:00
|
|
|
using llvm::sys::fs::OpenFlags;
|
2015-06-01 05:17:10 +08:00
|
|
|
using llvm::sys::fs::file_magic;
|
|
|
|
using llvm::sys::fs::identify_magic;
|
2015-05-29 03:09:30 +08:00
|
|
|
|
2015-05-29 04:30:06 +08:00
|
|
|
namespace lld {
|
|
|
|
namespace coff {
|
2015-05-29 03:09:30 +08:00
|
|
|
|
2015-05-29 04:30:06 +08:00
|
|
|
Configuration *Config;
|
2015-06-01 03:17:09 +08:00
|
|
|
LinkerDriver *Driver;
|
|
|
|
|
2016-02-29 03:54:51 +08:00
|
|
|
bool link(llvm::ArrayRef<const char *> Args) {
|
2015-08-18 17:13:25 +08:00
|
|
|
Configuration C;
|
|
|
|
LinkerDriver D;
|
|
|
|
Config = &C;
|
|
|
|
Driver = &D;
|
2016-02-29 03:54:51 +08:00
|
|
|
Driver->link(Args);
|
|
|
|
return true;
|
2015-06-01 03:17:09 +08:00
|
|
|
}
|
2015-05-29 03:09:30 +08:00
|
|
|
|
2015-06-07 08:20:32 +08:00
|
|
|
// Drop directory components and replace extension with ".exe".
|
|
|
|
static std::string getOutputPath(StringRef Path) {
|
|
|
|
auto P = Path.find_last_of("\\/");
|
|
|
|
StringRef S = (P == StringRef::npos) ? Path : Path.substr(P + 1);
|
|
|
|
return (S.substr(0, S.rfind('.')) + ".exe").str();
|
2015-05-29 03:09:30 +08:00
|
|
|
}
|
|
|
|
|
2015-06-01 05:04:56 +08:00
|
|
|
// Opens a file. Path has to be resolved already.
|
|
|
|
// Newly created memory buffers are owned by this driver.
|
2015-08-06 22:58:50 +08:00
|
|
|
MemoryBufferRef LinkerDriver::openFile(StringRef Path) {
|
2015-06-01 05:04:56 +08:00
|
|
|
auto MBOrErr = MemoryBuffer::getFile(Path);
|
2015-08-06 22:58:50 +08:00
|
|
|
error(MBOrErr, Twine("Could not open ") + Path);
|
|
|
|
std::unique_ptr<MemoryBuffer> &MB = *MBOrErr;
|
2015-06-01 05:04:56 +08:00
|
|
|
MemoryBufferRef MBRef = MB->getMemBufferRef();
|
|
|
|
OwningMBs.push_back(std::move(MB)); // take ownership
|
2015-06-15 05:50:50 +08:00
|
|
|
return MBRef;
|
|
|
|
}
|
2015-06-01 05:17:10 +08:00
|
|
|
|
2015-06-15 05:50:50 +08:00
|
|
|
static std::unique_ptr<InputFile> createFile(MemoryBufferRef MB) {
|
2015-06-01 05:17:10 +08:00
|
|
|
// File type is detected by contents, not by file extension.
|
2015-06-15 05:50:50 +08:00
|
|
|
file_magic Magic = identify_magic(MB.getBuffer());
|
2015-06-01 05:17:10 +08:00
|
|
|
if (Magic == file_magic::archive)
|
2015-06-15 05:50:50 +08:00
|
|
|
return std::unique_ptr<InputFile>(new ArchiveFile(MB));
|
2015-06-02 04:10:10 +08:00
|
|
|
if (Magic == file_magic::bitcode)
|
2015-06-15 05:50:50 +08:00
|
|
|
return std::unique_ptr<InputFile>(new BitcodeFile(MB));
|
2015-06-07 08:20:32 +08:00
|
|
|
if (Config->OutputFile == "")
|
2015-06-15 05:50:50 +08:00
|
|
|
Config->OutputFile = getOutputPath(MB.getBufferIdentifier());
|
|
|
|
return std::unique_ptr<InputFile>(new ObjectFile(MB));
|
2015-05-29 03:09:30 +08:00
|
|
|
}
|
|
|
|
|
COFF: Improve dllexported name mangling compatibility.
The rules for dllexported symbols are overly complicated due to
x86 name decoration, fuzzy symbol resolution, and the fact that
one symbol can be resolved by so many different names. The rules
are probably intended to be "intuitive", so that users don't have
to understand the name mangling schemes, but it seems that it can
lead to unintended symbol exports.
To make it clear what I'm trying to do with this patch, let me
write how the export rules are subtle and complicated.
- x86 name decoration: If machine type is i386 and export name
is given by a command line option, like /export:foo, the
real symbol name the linker has to search for is _foo because
all symbols are decorated with "_" prefixes. This doesn't happen
on non-x86 machines. This automatic name decoration happens only
when the name is not C++ mangled.
However, the symbol name exported from DLLs are ones without "_"
on all platforms.
Moreover, if the option is given via .drectve section, no
symbol decoration is done (the reason being that the .drectve
section is created by a compiler and the compiler should always
know the exact name of the symbol, I guess).
- Fuzzy symbol resolution: In addition to x86 name decoration,
the linker has to look for cdecl or C++ mangled symbols
for a given /export. For example, it searches for not only
_foo but also _foo@<number> or ??foo@... for /export:foo.
Previous implementation didn't get it right. I'm trying to make
it as compatible with MSVC linker as possible with this patch
however the rules are. The new code looks a bit messy to me, but
I don't think it can be simpler due to the ad-hoc-ness of the rules.
llvm-svn: 246424
2015-08-31 16:43:21 +08:00
|
|
|
static bool isDecorated(StringRef Sym) {
|
|
|
|
return Sym.startswith("_") || Sym.startswith("@") || Sym.startswith("?");
|
|
|
|
}
|
|
|
|
|
2015-05-29 03:09:30 +08:00
|
|
|
// Parses .drectve section contents and returns a list of files
|
|
|
|
// specified by /defaultlib.
|
2015-08-06 22:58:50 +08:00
|
|
|
void LinkerDriver::parseDirectives(StringRef S) {
|
|
|
|
llvm::opt::InputArgList Args = Parser.parse(S);
|
2015-05-29 03:09:30 +08:00
|
|
|
|
2015-06-23 06:06:52 +08:00
|
|
|
for (auto *Arg : Args) {
|
2015-06-19 05:50:38 +08:00
|
|
|
switch (Arg->getOption().getID()) {
|
|
|
|
case OPT_alternatename:
|
2015-08-06 22:58:50 +08:00
|
|
|
parseAlternateName(Arg->getValue());
|
2015-06-19 05:50:38 +08:00
|
|
|
break;
|
|
|
|
case OPT_defaultlib:
|
|
|
|
if (Optional<StringRef> Path = findLib(Arg->getValue())) {
|
2015-08-06 22:58:50 +08:00
|
|
|
MemoryBufferRef MB = openFile(*Path);
|
|
|
|
Symtab.addFile(createFile(MB));
|
2015-06-19 05:50:38 +08:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
case OPT_export: {
|
2015-08-06 22:58:50 +08:00
|
|
|
Export E = parseExport(Arg->getValue());
|
COFF: Improve dllexported name mangling compatibility.
The rules for dllexported symbols are overly complicated due to
x86 name decoration, fuzzy symbol resolution, and the fact that
one symbol can be resolved by so many different names. The rules
are probably intended to be "intuitive", so that users don't have
to understand the name mangling schemes, but it seems that it can
lead to unintended symbol exports.
To make it clear what I'm trying to do with this patch, let me
write how the export rules are subtle and complicated.
- x86 name decoration: If machine type is i386 and export name
is given by a command line option, like /export:foo, the
real symbol name the linker has to search for is _foo because
all symbols are decorated with "_" prefixes. This doesn't happen
on non-x86 machines. This automatic name decoration happens only
when the name is not C++ mangled.
However, the symbol name exported from DLLs are ones without "_"
on all platforms.
Moreover, if the option is given via .drectve section, no
symbol decoration is done (the reason being that the .drectve
section is created by a compiler and the compiler should always
know the exact name of the symbol, I guess).
- Fuzzy symbol resolution: In addition to x86 name decoration,
the linker has to look for cdecl or C++ mangled symbols
for a given /export. For example, it searches for not only
_foo but also _foo@<number> or ??foo@... for /export:foo.
Previous implementation didn't get it right. I'm trying to make
it as compatible with MSVC linker as possible with this patch
however the rules are. The new code looks a bit messy to me, but
I don't think it can be simpler due to the ad-hoc-ness of the rules.
llvm-svn: 246424
2015-08-31 16:43:21 +08:00
|
|
|
E.Directives = true;
|
2015-08-06 22:58:50 +08:00
|
|
|
Config->Exports.push_back(E);
|
2015-06-19 05:50:38 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
case OPT_failifmismatch:
|
2015-08-06 22:58:50 +08:00
|
|
|
checkFailIfMismatch(Arg->getValue());
|
2015-06-19 05:50:38 +08:00
|
|
|
break;
|
2015-06-19 07:20:11 +08:00
|
|
|
case OPT_incl:
|
2015-06-26 11:44:00 +08:00
|
|
|
addUndefined(Arg->getValue());
|
2015-06-19 07:20:11 +08:00
|
|
|
break;
|
2015-06-19 07:22:39 +08:00
|
|
|
case OPT_merge:
|
2015-08-06 22:58:50 +08:00
|
|
|
parseMerge(Arg->getValue());
|
2015-06-19 07:22:39 +08:00
|
|
|
break;
|
|
|
|
case OPT_nodefaultlib:
|
|
|
|
Config->NoDefaultLibs.insert(doFindLib(Arg->getValue()));
|
|
|
|
break;
|
2015-08-12 00:46:08 +08:00
|
|
|
case OPT_editandcontinue:
|
2016-03-26 02:09:29 +08:00
|
|
|
case OPT_fastfail:
|
2015-09-04 00:20:47 +08:00
|
|
|
case OPT_guardsym:
|
2015-07-30 05:01:15 +08:00
|
|
|
case OPT_throwingnew:
|
2015-07-30 04:29:15 +08:00
|
|
|
break;
|
2015-06-19 05:50:38 +08:00
|
|
|
default:
|
2015-08-06 22:58:50 +08:00
|
|
|
error(Twine(Arg->getSpelling()) + " is not allowed in .drectve");
|
2015-06-01 05:04:56 +08:00
|
|
|
}
|
|
|
|
}
|
2015-05-29 03:09:30 +08:00
|
|
|
}
|
|
|
|
|
2015-06-01 03:17:12 +08:00
|
|
|
// Find file from search paths. You can omit ".obj", this function takes
|
|
|
|
// care of that. Note that the returned path is not guaranteed to exist.
|
2015-06-01 03:17:14 +08:00
|
|
|
StringRef LinkerDriver::doFindFile(StringRef Filename) {
|
2015-06-01 03:17:12 +08:00
|
|
|
bool hasPathSep = (Filename.find_first_of("/\\") != StringRef::npos);
|
|
|
|
if (hasPathSep)
|
|
|
|
return Filename;
|
|
|
|
bool hasExt = (Filename.find('.') != StringRef::npos);
|
|
|
|
for (StringRef Dir : SearchPaths) {
|
|
|
|
SmallString<128> Path = Dir;
|
|
|
|
llvm::sys::path::append(Path, Filename);
|
|
|
|
if (llvm::sys::fs::exists(Path.str()))
|
|
|
|
return Alloc.save(Path.str());
|
|
|
|
if (!hasExt) {
|
|
|
|
Path.append(".obj");
|
|
|
|
if (llvm::sys::fs::exists(Path.str()))
|
|
|
|
return Alloc.save(Path.str());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return Filename;
|
|
|
|
}
|
|
|
|
|
2015-06-01 03:17:14 +08:00
|
|
|
// Resolves a file path. This never returns the same path
|
|
|
|
// (in that case, it returns None).
|
|
|
|
Optional<StringRef> LinkerDriver::findFile(StringRef Filename) {
|
|
|
|
StringRef Path = doFindFile(Filename);
|
|
|
|
bool Seen = !VisitedFiles.insert(Path.lower()).second;
|
|
|
|
if (Seen)
|
|
|
|
return None;
|
|
|
|
return Path;
|
2015-06-01 03:17:12 +08:00
|
|
|
}
|
|
|
|
|
2015-06-01 03:17:14 +08:00
|
|
|
// Find library file from search path.
|
|
|
|
StringRef LinkerDriver::doFindLib(StringRef Filename) {
|
|
|
|
// Add ".lib" to Filename if that has no file extension.
|
2015-06-01 03:17:12 +08:00
|
|
|
bool hasExt = (Filename.find('.') != StringRef::npos);
|
2015-06-01 03:17:14 +08:00
|
|
|
if (!hasExt)
|
|
|
|
Filename = Alloc.save(Filename + ".lib");
|
|
|
|
return doFindFile(Filename);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Resolves a library path. /nodefaultlib options are taken into
|
|
|
|
// consideration. This never returns the same path (in that case,
|
|
|
|
// it returns None).
|
|
|
|
Optional<StringRef> LinkerDriver::findLib(StringRef Filename) {
|
|
|
|
if (Config->NoDefaultLibAll)
|
|
|
|
return None;
|
|
|
|
StringRef Path = doFindLib(Filename);
|
|
|
|
if (Config->NoDefaultLibs.count(Path))
|
|
|
|
return None;
|
|
|
|
bool Seen = !VisitedFiles.insert(Path.lower()).second;
|
|
|
|
if (Seen)
|
|
|
|
return None;
|
|
|
|
return Path;
|
2015-06-01 03:17:12 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Parses LIB environment which contains a list of search paths.
|
2015-06-20 06:39:48 +08:00
|
|
|
void LinkerDriver::addLibSearchPaths() {
|
2015-06-01 03:17:12 +08:00
|
|
|
Optional<std::string> EnvOpt = Process::GetEnv("LIB");
|
|
|
|
if (!EnvOpt.hasValue())
|
2015-06-20 06:39:48 +08:00
|
|
|
return;
|
2015-06-01 03:17:12 +08:00
|
|
|
StringRef Env = Alloc.save(*EnvOpt);
|
|
|
|
while (!Env.empty()) {
|
|
|
|
StringRef Path;
|
|
|
|
std::tie(Path, Env) = Env.split(';');
|
2015-06-20 06:39:48 +08:00
|
|
|
SearchPaths.push_back(Path);
|
2015-06-01 03:17:12 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-07-02 08:04:14 +08:00
|
|
|
Undefined *LinkerDriver::addUndefined(StringRef Name) {
|
|
|
|
Undefined *U = Symtab.addUndefined(Name);
|
2015-07-02 08:21:08 +08:00
|
|
|
Config->GCRoot.insert(U);
|
2015-07-02 08:04:14 +08:00
|
|
|
return U;
|
2015-06-26 11:44:00 +08:00
|
|
|
}
|
|
|
|
|
2015-07-09 09:25:49 +08:00
|
|
|
// Symbol names are mangled by appending "_" prefix on x86.
|
|
|
|
StringRef LinkerDriver::mangle(StringRef Sym) {
|
2015-07-26 05:54:50 +08:00
|
|
|
assert(Config->Machine != IMAGE_FILE_MACHINE_UNKNOWN);
|
|
|
|
if (Config->Machine == I386)
|
2015-07-09 09:25:49 +08:00
|
|
|
return Alloc.save("_" + Sym);
|
|
|
|
return Sym;
|
|
|
|
}
|
|
|
|
|
2015-06-29 09:03:53 +08:00
|
|
|
// Windows specific -- find default entry point name.
|
|
|
|
StringRef LinkerDriver::findDefaultEntry() {
|
|
|
|
// User-defined main functions and their corresponding entry points.
|
|
|
|
static const char *Entries[][2] = {
|
|
|
|
{"main", "mainCRTStartup"},
|
|
|
|
{"wmain", "wmainCRTStartup"},
|
|
|
|
{"WinMain", "WinMainCRTStartup"},
|
|
|
|
{"wWinMain", "wWinMainCRTStartup"},
|
|
|
|
};
|
|
|
|
for (auto E : Entries) {
|
2015-07-14 10:58:13 +08:00
|
|
|
StringRef Entry = Symtab.findMangle(mangle(E[0]));
|
|
|
|
if (!Entry.empty() && !isa<Undefined>(Symtab.find(Entry)->Body))
|
2015-07-09 09:25:49 +08:00
|
|
|
return mangle(E[1]);
|
2015-06-29 09:03:53 +08:00
|
|
|
}
|
|
|
|
return "";
|
|
|
|
}
|
|
|
|
|
|
|
|
WindowsSubsystem LinkerDriver::inferSubsystem() {
|
2015-06-17 08:16:33 +08:00
|
|
|
if (Config->DLL)
|
|
|
|
return IMAGE_SUBSYSTEM_WINDOWS_GUI;
|
2015-08-08 08:23:37 +08:00
|
|
|
if (Symtab.findUnderscore("main") || Symtab.findUnderscore("wmain"))
|
2015-06-29 09:03:53 +08:00
|
|
|
return IMAGE_SUBSYSTEM_WINDOWS_CUI;
|
2015-08-08 08:23:37 +08:00
|
|
|
if (Symtab.findUnderscore("WinMain") || Symtab.findUnderscore("wWinMain"))
|
2015-06-29 09:03:53 +08:00
|
|
|
return IMAGE_SUBSYSTEM_WINDOWS_GUI;
|
|
|
|
return IMAGE_SUBSYSTEM_UNKNOWN;
|
2015-06-17 08:16:33 +08:00
|
|
|
}
|
|
|
|
|
2015-07-26 05:42:33 +08:00
|
|
|
static uint64_t getDefaultImageBase() {
|
|
|
|
if (Config->is64())
|
|
|
|
return Config->DLL ? 0x180000000 : 0x140000000;
|
|
|
|
return Config->DLL ? 0x10000000 : 0x400000;
|
|
|
|
}
|
|
|
|
|
2015-08-06 22:58:50 +08:00
|
|
|
void LinkerDriver::link(llvm::ArrayRef<const char *> ArgsArr) {
|
2015-08-10 04:45:17 +08:00
|
|
|
// If the first command line argument is "/lib", link.exe acts like lib.exe.
|
|
|
|
// We call our own implementation of lib.exe that understands bitcode files.
|
|
|
|
if (ArgsArr.size() > 1 && StringRef(ArgsArr[1]).equals_lower("/lib")) {
|
|
|
|
if (llvm::libDriverMain(ArgsArr.slice(1)) != 0)
|
|
|
|
error("lib failed");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2015-06-02 04:10:10 +08:00
|
|
|
// Needed for LTO.
|
|
|
|
llvm::InitializeAllTargetInfos();
|
|
|
|
llvm::InitializeAllTargets();
|
|
|
|
llvm::InitializeAllTargetMCs();
|
|
|
|
llvm::InitializeAllAsmParsers();
|
|
|
|
llvm::InitializeAllAsmPrinters();
|
|
|
|
llvm::InitializeAllDisassemblers();
|
|
|
|
|
2015-05-29 03:09:30 +08:00
|
|
|
// Parse command line options.
|
2015-08-06 22:58:50 +08:00
|
|
|
llvm::opt::InputArgList Args = Parser.parseLINK(ArgsArr.slice(1));
|
2015-05-29 03:09:30 +08:00
|
|
|
|
2015-05-30 00:11:52 +08:00
|
|
|
// Handle /help
|
2015-06-23 06:06:52 +08:00
|
|
|
if (Args.hasArg(OPT_help)) {
|
2015-06-21 14:32:10 +08:00
|
|
|
printHelp(ArgsArr[0]);
|
2015-08-06 22:58:50 +08:00
|
|
|
return;
|
2015-05-30 00:11:52 +08:00
|
|
|
}
|
|
|
|
|
2015-08-06 22:58:50 +08:00
|
|
|
if (Args.filtered_begin(OPT_INPUT) == Args.filtered_end())
|
|
|
|
error("no input files.");
|
2015-05-30 00:06:00 +08:00
|
|
|
|
2015-06-20 06:39:48 +08:00
|
|
|
// Construct search path list.
|
|
|
|
SearchPaths.push_back("");
|
2015-06-23 06:06:52 +08:00
|
|
|
for (auto *Arg : Args.filtered(OPT_libpath))
|
2015-06-20 06:39:48 +08:00
|
|
|
SearchPaths.push_back(Arg->getValue());
|
|
|
|
addLibSearchPaths();
|
|
|
|
|
2015-06-07 08:20:32 +08:00
|
|
|
// Handle /out
|
2015-06-23 06:06:52 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_out))
|
2015-06-07 08:20:32 +08:00
|
|
|
Config->OutputFile = Arg->getValue();
|
|
|
|
|
2015-05-30 00:06:00 +08:00
|
|
|
// Handle /verbose
|
2015-06-23 06:06:52 +08:00
|
|
|
if (Args.hasArg(OPT_verbose))
|
2015-05-29 03:09:30 +08:00
|
|
|
Config->Verbose = true;
|
2015-05-30 00:06:00 +08:00
|
|
|
|
2015-06-29 03:35:15 +08:00
|
|
|
// Handle /force or /force:unresolved
|
|
|
|
if (Args.hasArg(OPT_force) || Args.hasArg(OPT_force_unresolved))
|
|
|
|
Config->Force = true;
|
|
|
|
|
2015-07-05 07:37:32 +08:00
|
|
|
// Handle /debug
|
|
|
|
if (Args.hasArg(OPT_debug))
|
|
|
|
Config->Debug = true;
|
|
|
|
|
2015-06-29 03:56:30 +08:00
|
|
|
// Handle /noentry
|
|
|
|
if (Args.hasArg(OPT_noentry)) {
|
2015-08-06 22:58:50 +08:00
|
|
|
if (!Args.hasArg(OPT_dll))
|
|
|
|
error("/noentry must be specified with /dll");
|
2015-06-29 03:56:30 +08:00
|
|
|
Config->NoEntry = true;
|
|
|
|
}
|
|
|
|
|
2015-06-17 08:16:33 +08:00
|
|
|
// Handle /dll
|
2015-06-23 06:06:52 +08:00
|
|
|
if (Args.hasArg(OPT_dll)) {
|
2015-06-17 08:16:33 +08:00
|
|
|
Config->DLL = true;
|
2015-06-18 08:12:42 +08:00
|
|
|
Config->ManifestID = 2;
|
|
|
|
}
|
2015-06-17 08:16:33 +08:00
|
|
|
|
2015-06-15 09:23:58 +08:00
|
|
|
// Handle /fixed
|
2015-06-23 06:06:52 +08:00
|
|
|
if (Args.hasArg(OPT_fixed)) {
|
2015-08-06 22:58:50 +08:00
|
|
|
if (Args.hasArg(OPT_dynamicbase))
|
|
|
|
error("/fixed must not be specified with /dynamicbase");
|
2015-06-15 09:23:58 +08:00
|
|
|
Config->Relocatable = false;
|
2015-06-17 07:13:00 +08:00
|
|
|
Config->DynamicBase = false;
|
|
|
|
}
|
2015-06-15 09:23:58 +08:00
|
|
|
|
2015-05-30 00:06:00 +08:00
|
|
|
// Handle /machine
|
2015-08-06 22:58:50 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_machine))
|
|
|
|
Config->Machine = getMachineType(Arg->getValue());
|
2015-05-30 00:06:00 +08:00
|
|
|
|
2015-06-01 03:17:14 +08:00
|
|
|
// Handle /nodefaultlib:<filename>
|
2015-06-23 06:06:52 +08:00
|
|
|
for (auto *Arg : Args.filtered(OPT_nodefaultlib))
|
2015-06-01 03:17:14 +08:00
|
|
|
Config->NoDefaultLibs.insert(doFindLib(Arg->getValue()));
|
|
|
|
|
|
|
|
// Handle /nodefaultlib
|
2015-06-23 06:06:52 +08:00
|
|
|
if (Args.hasArg(OPT_nodefaultlib_all))
|
2015-06-01 03:17:14 +08:00
|
|
|
Config->NoDefaultLibAll = true;
|
|
|
|
|
2015-05-30 00:18:15 +08:00
|
|
|
// Handle /base
|
2015-08-06 22:58:50 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_base))
|
|
|
|
parseNumbers(Arg->getValue(), &Config->ImageBase);
|
2015-05-30 00:21:11 +08:00
|
|
|
|
|
|
|
// Handle /stack
|
2015-08-06 22:58:50 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_stack))
|
|
|
|
parseNumbers(Arg->getValue(), &Config->StackReserve, &Config->StackCommit);
|
2015-05-30 00:18:15 +08:00
|
|
|
|
2015-05-30 00:23:40 +08:00
|
|
|
// Handle /heap
|
2015-08-06 22:58:50 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_heap))
|
|
|
|
parseNumbers(Arg->getValue(), &Config->HeapReserve, &Config->HeapCommit);
|
2015-05-30 00:23:40 +08:00
|
|
|
|
2015-05-30 00:28:29 +08:00
|
|
|
// Handle /version
|
2015-08-06 22:58:50 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_version))
|
|
|
|
parseVersion(Arg->getValue(), &Config->MajorImageVersion,
|
|
|
|
&Config->MinorImageVersion);
|
2015-05-30 00:28:29 +08:00
|
|
|
|
2015-05-30 00:34:31 +08:00
|
|
|
// Handle /subsystem
|
2015-08-06 22:58:50 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_subsystem))
|
|
|
|
parseSubsystem(Arg->getValue(), &Config->Subsystem, &Config->MajorOSVersion,
|
|
|
|
&Config->MinorOSVersion);
|
2015-06-07 11:17:42 +08:00
|
|
|
|
2015-06-19 03:09:30 +08:00
|
|
|
// Handle /alternatename
|
2015-06-23 06:06:52 +08:00
|
|
|
for (auto *Arg : Args.filtered(OPT_alternatename))
|
2015-08-06 22:58:50 +08:00
|
|
|
parseAlternateName(Arg->getValue());
|
2015-06-19 03:09:30 +08:00
|
|
|
|
2015-06-19 07:20:11 +08:00
|
|
|
// Handle /include
|
2015-06-23 06:06:52 +08:00
|
|
|
for (auto *Arg : Args.filtered(OPT_incl))
|
2015-06-26 11:44:00 +08:00
|
|
|
addUndefined(Arg->getValue());
|
2015-06-19 07:20:11 +08:00
|
|
|
|
2015-06-19 04:27:09 +08:00
|
|
|
// Handle /implib
|
2015-06-23 06:06:52 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_implib))
|
2015-06-19 04:27:09 +08:00
|
|
|
Config->Implib = Arg->getValue();
|
|
|
|
|
2015-06-07 11:17:42 +08:00
|
|
|
// Handle /opt
|
2015-06-23 06:06:52 +08:00
|
|
|
for (auto *Arg : Args.filtered(OPT_opt)) {
|
2015-10-20 03:40:43 +08:00
|
|
|
std::string Str = StringRef(Arg->getValue()).lower();
|
|
|
|
SmallVector<StringRef, 1> Vec;
|
|
|
|
StringRef(Str).split(Vec, ',');
|
|
|
|
for (StringRef S : Vec) {
|
|
|
|
if (S == "noref") {
|
|
|
|
Config->DoGC = false;
|
|
|
|
Config->DoICF = false;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if (S == "icf" || StringRef(S).startswith("icf=")) {
|
|
|
|
Config->DoICF = true;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if (S == "noicf") {
|
|
|
|
Config->DoICF = false;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if (StringRef(S).startswith("lldlto=")) {
|
|
|
|
StringRef OptLevel = StringRef(S).substr(7);
|
|
|
|
if (OptLevel.getAsInteger(10, Config->LTOOptLevel) ||
|
|
|
|
Config->LTOOptLevel > 3)
|
|
|
|
error("/opt:lldlto: invalid optimization level: " + OptLevel);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if (StringRef(S).startswith("lldltojobs=")) {
|
|
|
|
StringRef Jobs = StringRef(S).substr(11);
|
|
|
|
if (Jobs.getAsInteger(10, Config->LTOJobs) || Config->LTOJobs == 0)
|
|
|
|
error("/opt:lldltojobs: invalid job count: " + Jobs);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if (S != "ref" && S != "lbr" && S != "nolbr")
|
|
|
|
error(Twine("/opt: unknown option: ") + S);
|
2015-08-14 12:47:07 +08:00
|
|
|
}
|
2015-05-30 00:34:31 +08:00
|
|
|
}
|
|
|
|
|
2015-06-05 03:21:24 +08:00
|
|
|
// Handle /failifmismatch
|
2015-06-23 06:06:52 +08:00
|
|
|
for (auto *Arg : Args.filtered(OPT_failifmismatch))
|
2015-08-06 22:58:50 +08:00
|
|
|
checkFailIfMismatch(Arg->getValue());
|
2015-06-05 03:21:24 +08:00
|
|
|
|
2015-07-05 07:37:32 +08:00
|
|
|
// Handle /merge
|
|
|
|
for (auto *Arg : Args.filtered(OPT_merge))
|
2015-08-06 22:58:50 +08:00
|
|
|
parseMerge(Arg->getValue());
|
2015-07-05 07:37:32 +08:00
|
|
|
|
2015-06-18 08:12:42 +08:00
|
|
|
// Handle /manifest
|
2015-08-06 22:58:50 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_manifest_colon))
|
|
|
|
parseManifest(Arg->getValue());
|
2015-06-18 08:12:42 +08:00
|
|
|
|
|
|
|
// Handle /manifestuac
|
2015-08-06 22:58:50 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_manifestuac))
|
|
|
|
parseManifestUAC(Arg->getValue());
|
2015-06-18 08:12:42 +08:00
|
|
|
|
|
|
|
// Handle /manifestdependency
|
2015-06-23 06:06:52 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_manifestdependency))
|
2015-06-18 08:12:42 +08:00
|
|
|
Config->ManifestDependency = Arg->getValue();
|
|
|
|
|
|
|
|
// Handle /manifestfile
|
2015-06-23 06:06:52 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_manifestfile))
|
2015-06-18 08:12:42 +08:00
|
|
|
Config->ManifestFile = Arg->getValue();
|
|
|
|
|
2015-06-17 07:13:00 +08:00
|
|
|
// Handle miscellaneous boolean flags.
|
2015-06-23 06:06:52 +08:00
|
|
|
if (Args.hasArg(OPT_allowbind_no))
|
|
|
|
Config->AllowBind = false;
|
|
|
|
if (Args.hasArg(OPT_allowisolation_no))
|
|
|
|
Config->AllowIsolation = false;
|
|
|
|
if (Args.hasArg(OPT_dynamicbase_no))
|
|
|
|
Config->DynamicBase = false;
|
|
|
|
if (Args.hasArg(OPT_nxcompat_no))
|
|
|
|
Config->NxCompat = false;
|
|
|
|
if (Args.hasArg(OPT_tsaware_no))
|
|
|
|
Config->TerminalServerAware = false;
|
2015-09-22 07:43:31 +08:00
|
|
|
if (Args.hasArg(OPT_nosymtab))
|
|
|
|
Config->WriteSymtab = false;
|
2015-06-17 07:13:00 +08:00
|
|
|
|
2015-06-01 03:17:14 +08:00
|
|
|
// Create a list of input files. Files can be given as arguments
|
|
|
|
// for /defaultlib option.
|
2015-07-10 03:54:13 +08:00
|
|
|
std::vector<StringRef> Paths;
|
|
|
|
std::vector<MemoryBufferRef> MBs;
|
2015-06-23 06:06:52 +08:00
|
|
|
for (auto *Arg : Args.filtered(OPT_INPUT))
|
2015-06-01 03:17:14 +08:00
|
|
|
if (Optional<StringRef> Path = findFile(Arg->getValue()))
|
2015-07-10 03:54:13 +08:00
|
|
|
Paths.push_back(*Path);
|
2015-06-23 06:06:52 +08:00
|
|
|
for (auto *Arg : Args.filtered(OPT_defaultlib))
|
2015-06-01 03:17:14 +08:00
|
|
|
if (Optional<StringRef> Path = findLib(Arg->getValue()))
|
2015-07-10 03:54:13 +08:00
|
|
|
Paths.push_back(*Path);
|
2015-08-06 22:58:50 +08:00
|
|
|
for (StringRef Path : Paths)
|
|
|
|
MBs.push_back(openFile(Path));
|
2015-06-01 03:17:14 +08:00
|
|
|
|
2015-06-18 08:12:42 +08:00
|
|
|
// Windows specific -- Create a resource file containing a manifest file.
|
|
|
|
if (Config->Manifest == Configuration::Embed) {
|
2015-08-06 22:58:50 +08:00
|
|
|
std::unique_ptr<MemoryBuffer> MB = createManifestRes();
|
2015-07-10 03:54:13 +08:00
|
|
|
MBs.push_back(MB->getMemBufferRef());
|
2015-06-18 08:12:42 +08:00
|
|
|
OwningMBs.push_back(std::move(MB)); // take ownership
|
|
|
|
}
|
|
|
|
|
2015-06-15 05:50:50 +08:00
|
|
|
// Windows specific -- Input files can be Windows resource files (.res files).
|
|
|
|
// We invoke cvtres.exe to convert resource files to a regular COFF file
|
|
|
|
// then link the result file normally.
|
2015-07-10 03:54:13 +08:00
|
|
|
std::vector<MemoryBufferRef> Resources;
|
2015-06-27 07:59:13 +08:00
|
|
|
auto NotResource = [](MemoryBufferRef MB) {
|
|
|
|
return identify_magic(MB.getBuffer()) != file_magic::windows_resource;
|
2015-06-15 05:50:50 +08:00
|
|
|
};
|
2015-07-10 03:54:13 +08:00
|
|
|
auto It = std::stable_partition(MBs.begin(), MBs.end(), NotResource);
|
|
|
|
if (It != MBs.end()) {
|
|
|
|
Resources.insert(Resources.end(), It, MBs.end());
|
|
|
|
MBs.erase(It, MBs.end());
|
2015-06-15 05:50:50 +08:00
|
|
|
}
|
|
|
|
|
COFF: Infer entry point as early as possible, but not too early.
On Windows, we have four different main functions, {w,}{main,WinMain}.
The linker has to choose a corresponding entry point function among
{w,}{main,WinMain}CRTStartup. These entry point functions are defined
in the standard library. The linker resolves one of them by looking at
which main function is defined and adding a corresponding undefined
symbol to the symbol table.
Object files containing entry point functions conflicts each other.
For example, we cannot resolve both mainCRTStartup and WinMainCRTStartup
because other symbols defined in the files conflict.
Previously, we inferred CRT function name at the very end of name
resolution. I found that that is sometimes too late. If the linker
already linked one of these four archive member objects, it's too late
to change the decision.
The right thing to do here is to infer entry point name after adding
all symbols from command line files and before adding any other files
(which are specified by directive sections). This patch does that.
llvm-svn: 241236
2015-07-02 11:15:15 +08:00
|
|
|
// Read all input files given via the command line. Note that step()
|
|
|
|
// doesn't read files that are specified by directive sections.
|
2015-07-10 03:54:13 +08:00
|
|
|
for (MemoryBufferRef MB : MBs)
|
2015-06-24 07:56:39 +08:00
|
|
|
Symtab.addFile(createFile(MB));
|
2015-08-06 22:58:50 +08:00
|
|
|
Symtab.step();
|
2015-05-31 11:34:08 +08:00
|
|
|
|
2015-07-10 03:54:13 +08:00
|
|
|
// Determine machine type and check if all object files are
|
|
|
|
// for the same CPU type. Note that this needs to be done before
|
|
|
|
// any call to mangle().
|
|
|
|
for (std::unique_ptr<InputFile> &File : Symtab.getFiles()) {
|
|
|
|
MachineTypes MT = File->getMachineType();
|
|
|
|
if (MT == IMAGE_FILE_MACHINE_UNKNOWN)
|
|
|
|
continue;
|
2015-07-26 05:54:50 +08:00
|
|
|
if (Config->Machine == IMAGE_FILE_MACHINE_UNKNOWN) {
|
|
|
|
Config->Machine = MT;
|
2015-07-10 03:54:13 +08:00
|
|
|
continue;
|
|
|
|
}
|
2015-08-06 22:58:50 +08:00
|
|
|
if (Config->Machine != MT)
|
|
|
|
error(Twine(File->getShortName()) + ": machine type " + machineToStr(MT) +
|
|
|
|
" conflicts with " + machineToStr(Config->Machine));
|
2015-07-10 03:54:13 +08:00
|
|
|
}
|
2015-07-26 05:54:50 +08:00
|
|
|
if (Config->Machine == IMAGE_FILE_MACHINE_UNKNOWN) {
|
2015-07-10 03:54:13 +08:00
|
|
|
llvm::errs() << "warning: /machine is not specified. x64 is assumed.\n";
|
2015-07-26 05:54:50 +08:00
|
|
|
Config->Machine = AMD64;
|
2015-07-10 03:54:13 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Windows specific -- Convert Windows resource files to a COFF file.
|
|
|
|
if (!Resources.empty()) {
|
2015-08-06 22:58:50 +08:00
|
|
|
std::unique_ptr<MemoryBuffer> MB = convertResToCOFF(Resources);
|
2015-07-10 03:54:13 +08:00
|
|
|
Symtab.addFile(createFile(MB->getMemBufferRef()));
|
|
|
|
OwningMBs.push_back(std::move(MB)); // take ownership
|
|
|
|
}
|
|
|
|
|
2015-07-28 11:12:00 +08:00
|
|
|
// Handle /largeaddressaware
|
|
|
|
if (Config->is64() || Args.hasArg(OPT_largeaddressaware))
|
|
|
|
Config->LargeAddressAware = true;
|
|
|
|
|
2015-07-28 11:15:57 +08:00
|
|
|
// Handle /highentropyva
|
|
|
|
if (Config->is64() && !Args.hasArg(OPT_highentropyva_no))
|
|
|
|
Config->HighEntropyVA = true;
|
|
|
|
|
2015-07-10 06:51:41 +08:00
|
|
|
// Handle /entry and /dll
|
|
|
|
if (auto *Arg = Args.getLastArg(OPT_entry)) {
|
|
|
|
Config->Entry = addUndefined(mangle(Arg->getValue()));
|
|
|
|
} else if (Args.hasArg(OPT_dll) && !Config->NoEntry) {
|
2015-07-26 05:54:50 +08:00
|
|
|
StringRef S = (Config->Machine == I386) ? "__DllMainCRTStartup@12"
|
|
|
|
: "_DllMainCRTStartup";
|
2015-07-10 06:51:41 +08:00
|
|
|
Config->Entry = addUndefined(S);
|
|
|
|
} else if (!Config->NoEntry) {
|
|
|
|
// Windows specific -- If entry point name is not given, we need to
|
|
|
|
// infer that from user-defined entry name.
|
2015-06-29 09:03:53 +08:00
|
|
|
StringRef S = findDefaultEntry();
|
2015-08-06 22:58:50 +08:00
|
|
|
if (S.empty())
|
|
|
|
error("entry point must be defined");
|
2015-07-02 08:04:14 +08:00
|
|
|
Config->Entry = addUndefined(S);
|
COFF: Infer entry point as early as possible, but not too early.
On Windows, we have four different main functions, {w,}{main,WinMain}.
The linker has to choose a corresponding entry point function among
{w,}{main,WinMain}CRTStartup. These entry point functions are defined
in the standard library. The linker resolves one of them by looking at
which main function is defined and adding a corresponding undefined
symbol to the symbol table.
Object files containing entry point functions conflicts each other.
For example, we cannot resolve both mainCRTStartup and WinMainCRTStartup
because other symbols defined in the files conflict.
Previously, we inferred CRT function name at the very end of name
resolution. I found that that is sometimes too late. If the linker
already linked one of these four archive member objects, it's too late
to change the decision.
The right thing to do here is to infer entry point name after adding
all symbols from command line files and before adding any other files
(which are specified by directive sections). This patch does that.
llvm-svn: 241236
2015-07-02 11:15:15 +08:00
|
|
|
if (Config->Verbose)
|
|
|
|
llvm::outs() << "Entry name inferred: " << S << "\n";
|
|
|
|
}
|
|
|
|
|
2015-07-10 06:51:41 +08:00
|
|
|
// Handle /export
|
|
|
|
for (auto *Arg : Args.filtered(OPT_export)) {
|
2015-08-06 22:58:50 +08:00
|
|
|
Export E = parseExport(Arg->getValue());
|
COFF: Improve dllexported name mangling compatibility.
The rules for dllexported symbols are overly complicated due to
x86 name decoration, fuzzy symbol resolution, and the fact that
one symbol can be resolved by so many different names. The rules
are probably intended to be "intuitive", so that users don't have
to understand the name mangling schemes, but it seems that it can
lead to unintended symbol exports.
To make it clear what I'm trying to do with this patch, let me
write how the export rules are subtle and complicated.
- x86 name decoration: If machine type is i386 and export name
is given by a command line option, like /export:foo, the
real symbol name the linker has to search for is _foo because
all symbols are decorated with "_" prefixes. This doesn't happen
on non-x86 machines. This automatic name decoration happens only
when the name is not C++ mangled.
However, the symbol name exported from DLLs are ones without "_"
on all platforms.
Moreover, if the option is given via .drectve section, no
symbol decoration is done (the reason being that the .drectve
section is created by a compiler and the compiler should always
know the exact name of the symbol, I guess).
- Fuzzy symbol resolution: In addition to x86 name decoration,
the linker has to look for cdecl or C++ mangled symbols
for a given /export. For example, it searches for not only
_foo but also _foo@<number> or ??foo@... for /export:foo.
Previous implementation didn't get it right. I'm trying to make
it as compatible with MSVC linker as possible with this patch
however the rules are. The new code looks a bit messy to me, but
I don't think it can be simpler due to the ad-hoc-ness of the rules.
llvm-svn: 246424
2015-08-31 16:43:21 +08:00
|
|
|
if (Config->Machine == I386) {
|
|
|
|
if (!isDecorated(E.Name))
|
|
|
|
E.Name = Alloc.save("_" + E.Name);
|
|
|
|
if (!E.ExtName.empty() && !isDecorated(E.ExtName))
|
|
|
|
E.ExtName = Alloc.save("_" + E.ExtName);
|
|
|
|
}
|
2015-08-06 22:58:50 +08:00
|
|
|
Config->Exports.push_back(E);
|
2015-07-10 06:51:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Handle /def
|
|
|
|
if (auto *Arg = Args.getLastArg(OPT_deffile)) {
|
2015-08-06 22:58:50 +08:00
|
|
|
MemoryBufferRef MB = openFile(Arg->getValue());
|
2015-07-10 06:51:41 +08:00
|
|
|
// parseModuleDefs mutates Config object.
|
2015-08-06 22:58:50 +08:00
|
|
|
parseModuleDefs(MB, &Alloc);
|
2015-07-10 06:51:41 +08:00
|
|
|
}
|
|
|
|
|
2015-07-14 06:31:45 +08:00
|
|
|
// Handle /delayload
|
|
|
|
for (auto *Arg : Args.filtered(OPT_delayload)) {
|
|
|
|
Config->DelayLoads.insert(StringRef(Arg->getValue()).lower());
|
2015-07-26 05:54:50 +08:00
|
|
|
if (Config->Machine == I386) {
|
2015-07-14 06:31:45 +08:00
|
|
|
Config->DelayLoadHelper = addUndefined("___delayLoadHelper2@8");
|
2015-07-25 08:20:06 +08:00
|
|
|
} else {
|
|
|
|
Config->DelayLoadHelper = addUndefined("__delayLoadHelper2");
|
2015-07-14 06:31:45 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-07-26 05:42:33 +08:00
|
|
|
// Set default image base if /base is not given.
|
|
|
|
if (Config->ImageBase == uint64_t(-1))
|
|
|
|
Config->ImageBase = getDefaultImageBase();
|
|
|
|
|
2015-07-25 06:58:44 +08:00
|
|
|
Symtab.addRelative(mangle("__ImageBase"), 0);
|
2015-07-26 05:54:50 +08:00
|
|
|
if (Config->Machine == I386) {
|
2015-07-25 07:51:14 +08:00
|
|
|
Config->SEHTable = Symtab.addRelative("___safe_se_handler_table", 0);
|
|
|
|
Config->SEHCount = Symtab.addAbsolute("___safe_se_handler_count", 0);
|
|
|
|
}
|
2015-07-10 06:51:41 +08:00
|
|
|
|
2015-08-10 05:01:06 +08:00
|
|
|
// We do not support /guard:cf (control flow protection) yet.
|
|
|
|
// Define CFG symbols anyway so that we can link MSVC 2015 CRT.
|
|
|
|
Symtab.addAbsolute(mangle("__guard_fids_table"), 0);
|
|
|
|
Symtab.addAbsolute(mangle("__guard_fids_count"), 0);
|
|
|
|
Symtab.addAbsolute(mangle("__guard_flags"), 0x100);
|
|
|
|
|
2015-07-10 03:54:13 +08:00
|
|
|
// Read as much files as we can from directives sections.
|
2015-08-06 22:58:50 +08:00
|
|
|
Symtab.run();
|
2015-06-29 09:03:53 +08:00
|
|
|
|
COFF: Infer entry point as early as possible, but not too early.
On Windows, we have four different main functions, {w,}{main,WinMain}.
The linker has to choose a corresponding entry point function among
{w,}{main,WinMain}CRTStartup. These entry point functions are defined
in the standard library. The linker resolves one of them by looking at
which main function is defined and adding a corresponding undefined
symbol to the symbol table.
Object files containing entry point functions conflicts each other.
For example, we cannot resolve both mainCRTStartup and WinMainCRTStartup
because other symbols defined in the files conflict.
Previously, we inferred CRT function name at the very end of name
resolution. I found that that is sometimes too late. If the linker
already linked one of these four archive member objects, it's too late
to change the decision.
The right thing to do here is to infer entry point name after adding
all symbols from command line files and before adding any other files
(which are specified by directive sections). This patch does that.
llvm-svn: 241236
2015-07-02 11:15:15 +08:00
|
|
|
// Resolve auxiliary symbols until we get a convergence.
|
2015-06-20 05:12:48 +08:00
|
|
|
// (Trying to resolve a symbol may trigger a Lazy symbol to load a new file.
|
|
|
|
// A new file may contain a directive section to add new command line options.
|
|
|
|
// That's why we have to repeat until converge.)
|
|
|
|
for (;;) {
|
2015-07-02 08:04:14 +08:00
|
|
|
// Windows specific -- if entry point is not found,
|
|
|
|
// search for its mangled names.
|
|
|
|
if (Config->Entry)
|
|
|
|
Symtab.mangleMaybe(Config->Entry);
|
|
|
|
|
2015-06-20 05:12:48 +08:00
|
|
|
// Windows specific -- Make sure we resolve all dllexported symbols.
|
2015-07-02 08:04:14 +08:00
|
|
|
for (Export &E : Config->Exports) {
|
2016-01-09 09:22:00 +08:00
|
|
|
if (!E.ForwardTo.empty())
|
|
|
|
continue;
|
2015-07-02 08:04:14 +08:00
|
|
|
E.Sym = addUndefined(E.Name);
|
COFF: Improve dllexported name mangling compatibility.
The rules for dllexported symbols are overly complicated due to
x86 name decoration, fuzzy symbol resolution, and the fact that
one symbol can be resolved by so many different names. The rules
are probably intended to be "intuitive", so that users don't have
to understand the name mangling schemes, but it seems that it can
lead to unintended symbol exports.
To make it clear what I'm trying to do with this patch, let me
write how the export rules are subtle and complicated.
- x86 name decoration: If machine type is i386 and export name
is given by a command line option, like /export:foo, the
real symbol name the linker has to search for is _foo because
all symbols are decorated with "_" prefixes. This doesn't happen
on non-x86 machines. This automatic name decoration happens only
when the name is not C++ mangled.
However, the symbol name exported from DLLs are ones without "_"
on all platforms.
Moreover, if the option is given via .drectve section, no
symbol decoration is done (the reason being that the .drectve
section is created by a compiler and the compiler should always
know the exact name of the symbol, I guess).
- Fuzzy symbol resolution: In addition to x86 name decoration,
the linker has to look for cdecl or C++ mangled symbols
for a given /export. For example, it searches for not only
_foo but also _foo@<number> or ??foo@... for /export:foo.
Previous implementation didn't get it right. I'm trying to make
it as compatible with MSVC linker as possible with this patch
however the rules are. The new code looks a bit messy to me, but
I don't think it can be simpler due to the ad-hoc-ness of the rules.
llvm-svn: 246424
2015-08-31 16:43:21 +08:00
|
|
|
if (!E.Directives)
|
|
|
|
Symtab.mangleMaybe(E.Sym);
|
2015-07-02 08:04:14 +08:00
|
|
|
}
|
2015-06-20 05:12:48 +08:00
|
|
|
|
|
|
|
// Add weak aliases. Weak aliases is a mechanism to give remaining
|
|
|
|
// undefined symbols final chance to be resolved successfully.
|
2015-07-02 10:38:59 +08:00
|
|
|
for (auto Pair : Config->AlternateNames) {
|
|
|
|
StringRef From = Pair.first;
|
|
|
|
StringRef To = Pair.second;
|
2015-07-02 11:59:04 +08:00
|
|
|
Symbol *Sym = Symtab.find(From);
|
2015-07-02 10:38:59 +08:00
|
|
|
if (!Sym)
|
|
|
|
continue;
|
2015-07-07 01:45:22 +08:00
|
|
|
if (auto *U = dyn_cast<Undefined>(Sym->Body))
|
2015-07-02 10:38:59 +08:00
|
|
|
if (!U->WeakAlias)
|
|
|
|
U->WeakAlias = Symtab.addUndefined(To);
|
2015-05-31 11:34:08 +08:00
|
|
|
}
|
|
|
|
|
2015-07-25 07:51:14 +08:00
|
|
|
// Windows specific -- if __load_config_used can be resolved, resolve it.
|
2015-08-08 06:43:53 +08:00
|
|
|
if (Symtab.findUnderscore("_load_config_used"))
|
|
|
|
addUndefined(mangle("_load_config_used"));
|
2015-07-25 07:51:14 +08:00
|
|
|
|
2015-07-02 10:38:59 +08:00
|
|
|
if (Symtab.queueEmpty())
|
|
|
|
break;
|
2015-08-06 22:58:50 +08:00
|
|
|
Symtab.run();
|
2015-06-19 03:09:30 +08:00
|
|
|
}
|
|
|
|
|
2015-08-29 06:16:09 +08:00
|
|
|
// Do LTO by compiling bitcode input files to a set of native COFF files then
|
|
|
|
// link those files.
|
|
|
|
Symtab.addCombinedLTOObjects();
|
2015-06-02 04:10:10 +08:00
|
|
|
|
2015-07-04 13:28:41 +08:00
|
|
|
// Make sure we have resolved all symbols.
|
2015-08-06 22:58:50 +08:00
|
|
|
Symtab.reportRemainingUndefines(/*Resolve=*/true);
|
2015-07-04 13:28:41 +08:00
|
|
|
|
2015-05-31 11:55:46 +08:00
|
|
|
// Windows specific -- if no /subsystem is given, we need to infer
|
|
|
|
// that from entry point name.
|
|
|
|
if (Config->Subsystem == IMAGE_SUBSYSTEM_UNKNOWN) {
|
2015-06-17 08:16:33 +08:00
|
|
|
Config->Subsystem = inferSubsystem();
|
2015-08-06 22:58:50 +08:00
|
|
|
if (Config->Subsystem == IMAGE_SUBSYSTEM_UNKNOWN)
|
|
|
|
error("subsystem must be defined");
|
2015-05-31 11:55:46 +08:00
|
|
|
}
|
|
|
|
|
2015-07-30 04:25:40 +08:00
|
|
|
// Handle /safeseh.
|
2015-09-15 08:33:11 +08:00
|
|
|
if (Args.hasArg(OPT_safeseh))
|
|
|
|
for (ObjectFile *File : Symtab.ObjectFiles)
|
|
|
|
if (!File->SEHCompat)
|
|
|
|
error("/safeseh: " + File->getName() + " is not compatible with SEH");
|
2015-07-30 04:25:40 +08:00
|
|
|
|
2015-06-18 04:40:43 +08:00
|
|
|
// Windows specific -- when we are creating a .dll file, we also
|
|
|
|
// need to create a .lib file.
|
2015-09-01 17:15:58 +08:00
|
|
|
if (!Config->Exports.empty() || Config->DLL) {
|
2015-08-06 22:58:50 +08:00
|
|
|
fixupExports();
|
|
|
|
writeImportLibrary();
|
2015-07-16 06:21:08 +08:00
|
|
|
assignExportOrdinals();
|
|
|
|
}
|
2015-06-17 08:16:33 +08:00
|
|
|
|
2015-06-18 08:12:42 +08:00
|
|
|
// Windows specific -- Create a side-by-side manifest file.
|
|
|
|
if (Config->Manifest == Configuration::SideBySide)
|
2015-08-06 22:58:50 +08:00
|
|
|
createSideBySideManifest();
|
2015-06-18 08:12:42 +08:00
|
|
|
|
2015-06-29 22:27:12 +08:00
|
|
|
// Create a dummy PDB file to satisfy build sytem rules.
|
|
|
|
if (auto *Arg = Args.getLastArg(OPT_pdb))
|
2015-12-05 07:11:05 +08:00
|
|
|
createPDB(Arg->getValue());
|
2015-06-29 22:27:12 +08:00
|
|
|
|
2015-09-20 05:36:28 +08:00
|
|
|
// Identify unreferenced COMDAT sections.
|
|
|
|
if (Config->DoGC)
|
|
|
|
markLive(Symtab.getChunks());
|
|
|
|
|
|
|
|
// Identify identical COMDAT sections to merge them.
|
|
|
|
if (Config->DoICF)
|
|
|
|
doICF(Symtab.getChunks());
|
|
|
|
|
2015-05-29 03:09:30 +08:00
|
|
|
// Write the result.
|
2015-08-06 22:58:50 +08:00
|
|
|
writeResult(&Symtab);
|
2015-06-27 02:58:24 +08:00
|
|
|
|
2015-06-29 04:07:08 +08:00
|
|
|
// Create a symbol map file containing symbol VAs and their names
|
|
|
|
// to help debugging.
|
2015-06-27 02:58:24 +08:00
|
|
|
if (auto *Arg = Args.getLastArg(OPT_lldmap)) {
|
|
|
|
std::error_code EC;
|
2015-06-27 03:20:09 +08:00
|
|
|
llvm::raw_fd_ostream Out(Arg->getValue(), EC, OpenFlags::F_Text);
|
2015-08-06 22:58:50 +08:00
|
|
|
error(EC, "Could not create the symbol map");
|
2015-06-27 02:58:24 +08:00
|
|
|
Symtab.printMap(Out);
|
|
|
|
}
|
2015-07-03 13:31:35 +08:00
|
|
|
// Call exit to avoid calling destructors.
|
|
|
|
exit(0);
|
2015-05-29 03:09:30 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace coff
|
|
|
|
} // namespace lld
|