forked from OSchip/llvm-project
886 lines
30 KiB
C++
886 lines
30 KiB
C++
//===- Driver.cpp ---------------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "Driver.h"
|
|
#include "Config.h"
|
|
#include "InputFiles.h"
|
|
#include "LTO.h"
|
|
#include "ObjC.h"
|
|
#include "OutputSection.h"
|
|
#include "OutputSegment.h"
|
|
#include "SymbolTable.h"
|
|
#include "Symbols.h"
|
|
#include "SyntheticSections.h"
|
|
#include "Target.h"
|
|
#include "Writer.h"
|
|
|
|
#include "lld/Common/Args.h"
|
|
#include "lld/Common/Driver.h"
|
|
#include "lld/Common/ErrorHandler.h"
|
|
#include "lld/Common/LLVM.h"
|
|
#include "lld/Common/Memory.h"
|
|
#include "lld/Common/Reproduce.h"
|
|
#include "lld/Common/Version.h"
|
|
#include "llvm/ADT/DenseSet.h"
|
|
#include "llvm/ADT/StringExtras.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/BinaryFormat/MachO.h"
|
|
#include "llvm/BinaryFormat/Magic.h"
|
|
#include "llvm/LTO/LTO.h"
|
|
#include "llvm/Object/Archive.h"
|
|
#include "llvm/Option/ArgList.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/Host.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/Path.h"
|
|
#include "llvm/Support/TarWriter.h"
|
|
#include "llvm/Support/TargetSelect.h"
|
|
#include "llvm/TextAPI/MachO/PackedVersion.h"
|
|
|
|
#include <algorithm>
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::MachO;
|
|
using namespace llvm::object;
|
|
using namespace llvm::opt;
|
|
using namespace llvm::sys;
|
|
using namespace lld;
|
|
using namespace lld::macho;
|
|
|
|
Configuration *lld::macho::config;
|
|
|
|
static HeaderFileType getOutputType(const opt::InputArgList &args) {
|
|
// TODO: -r, -dylinker, -preload...
|
|
opt::Arg *outputArg = args.getLastArg(OPT_bundle, OPT_dylib, OPT_execute);
|
|
if (outputArg == nullptr)
|
|
return MH_EXECUTE;
|
|
|
|
switch (outputArg->getOption().getID()) {
|
|
case OPT_bundle:
|
|
return MH_BUNDLE;
|
|
case OPT_dylib:
|
|
return MH_DYLIB;
|
|
case OPT_execute:
|
|
return MH_EXECUTE;
|
|
default:
|
|
llvm_unreachable("internal error");
|
|
}
|
|
}
|
|
|
|
static Optional<std::string>
|
|
findAlongPathsWithExtensions(StringRef name, ArrayRef<StringRef> extensions) {
|
|
SmallString<261> base;
|
|
for (StringRef dir : config->librarySearchPaths) {
|
|
base = dir;
|
|
path::append(base, Twine("lib") + name);
|
|
for (StringRef ext : extensions) {
|
|
Twine location = base + ext;
|
|
if (fs::exists(location))
|
|
return location.str();
|
|
}
|
|
}
|
|
return {};
|
|
}
|
|
|
|
static Optional<std::string> findLibrary(StringRef name) {
|
|
if (config->searchDylibsFirst) {
|
|
if (Optional<std::string> path =
|
|
findAlongPathsWithExtensions(name, {".tbd", ".dylib"}))
|
|
return path;
|
|
return findAlongPathsWithExtensions(name, {".a"});
|
|
}
|
|
return findAlongPathsWithExtensions(name, {".tbd", ".dylib", ".a"});
|
|
}
|
|
|
|
static Optional<std::string> findFramework(StringRef name) {
|
|
SmallString<260> symlink;
|
|
StringRef suffix;
|
|
std::tie(name, suffix) = name.split(",");
|
|
for (StringRef dir : config->frameworkSearchPaths) {
|
|
symlink = dir;
|
|
path::append(symlink, name + ".framework", name);
|
|
|
|
if (!suffix.empty()) {
|
|
// NOTE: we must resolve the symlink before trying the suffixes, because
|
|
// there are no symlinks for the suffixed paths.
|
|
SmallString<260> location;
|
|
if (!fs::real_path(symlink, location)) {
|
|
// only append suffix if realpath() succeeds
|
|
Twine suffixed = location + suffix;
|
|
if (fs::exists(suffixed))
|
|
return suffixed.str();
|
|
}
|
|
// Suffix lookup failed, fall through to the no-suffix case.
|
|
}
|
|
|
|
if (Optional<std::string> path = resolveDylibPath(symlink))
|
|
return path;
|
|
}
|
|
return {};
|
|
}
|
|
|
|
static TargetInfo *createTargetInfo(opt::InputArgList &args) {
|
|
StringRef arch = args.getLastArgValue(OPT_arch, "x86_64");
|
|
config->arch = MachO::getArchitectureFromName(
|
|
args.getLastArgValue(OPT_arch, arch));
|
|
switch (config->arch) {
|
|
case MachO::AK_x86_64:
|
|
case MachO::AK_x86_64h:
|
|
return createX86_64TargetInfo();
|
|
default:
|
|
fatal("missing or unsupported -arch " + arch);
|
|
}
|
|
}
|
|
|
|
static bool warnIfNotDirectory(StringRef option, StringRef path) {
|
|
if (!fs::exists(path)) {
|
|
warn("directory not found for option -" + option + path);
|
|
return false;
|
|
} else if (!fs::is_directory(path)) {
|
|
warn("option -" + option + path + " references a non-directory path");
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static std::vector<StringRef>
|
|
getSearchPaths(unsigned optionCode, opt::InputArgList &args,
|
|
const std::vector<StringRef> &roots,
|
|
const SmallVector<StringRef, 2> &systemPaths) {
|
|
std::vector<StringRef> paths;
|
|
StringRef optionLetter{optionCode == OPT_F ? "F" : "L"};
|
|
for (StringRef path : args::getStrings(args, optionCode)) {
|
|
// NOTE: only absolute paths are re-rooted to syslibroot(s)
|
|
bool found = false;
|
|
if (path::is_absolute(path, path::Style::posix)) {
|
|
for (StringRef root : roots) {
|
|
SmallString<261> buffer(root);
|
|
path::append(buffer, path);
|
|
// Do not warn about paths that are computed via the syslib roots
|
|
if (fs::is_directory(buffer)) {
|
|
paths.push_back(saver.save(buffer.str()));
|
|
found = true;
|
|
}
|
|
}
|
|
}
|
|
if (!found && warnIfNotDirectory(optionLetter, path))
|
|
paths.push_back(path);
|
|
}
|
|
|
|
// `-Z` suppresses the standard "system" search paths.
|
|
if (args.hasArg(OPT_Z))
|
|
return paths;
|
|
|
|
for (auto const &path : systemPaths) {
|
|
for (auto root : roots) {
|
|
SmallString<261> buffer(root);
|
|
path::append(buffer, path);
|
|
if (fs::is_directory(buffer))
|
|
paths.push_back(saver.save(buffer.str()));
|
|
}
|
|
}
|
|
return paths;
|
|
}
|
|
|
|
static std::vector<StringRef> getSystemLibraryRoots(opt::InputArgList &args) {
|
|
std::vector<StringRef> roots;
|
|
for (const Arg *arg : args.filtered(OPT_syslibroot))
|
|
roots.push_back(arg->getValue());
|
|
// NOTE: the final `-syslibroot` being `/` will ignore all roots
|
|
if (roots.size() && roots.back() == "/")
|
|
roots.clear();
|
|
// NOTE: roots can never be empty - add an empty root to simplify the library
|
|
// and framework search path computation.
|
|
if (roots.empty())
|
|
roots.emplace_back("");
|
|
return roots;
|
|
}
|
|
|
|
static std::vector<StringRef>
|
|
getLibrarySearchPaths(opt::InputArgList &args,
|
|
const std::vector<StringRef> &roots) {
|
|
return getSearchPaths(OPT_L, args, roots, {"/usr/lib", "/usr/local/lib"});
|
|
}
|
|
|
|
static std::vector<StringRef>
|
|
getFrameworkSearchPaths(opt::InputArgList &args,
|
|
const std::vector<StringRef> &roots) {
|
|
return getSearchPaths(OPT_F, args, roots,
|
|
{"/Library/Frameworks", "/System/Library/Frameworks"});
|
|
}
|
|
|
|
namespace {
|
|
struct ArchiveMember {
|
|
MemoryBufferRef mbref;
|
|
uint32_t modTime;
|
|
};
|
|
} // namespace
|
|
|
|
// Returns slices of MB by parsing MB as an archive file.
|
|
// Each slice consists of a member file in the archive.
|
|
static std::vector<ArchiveMember> getArchiveMembers(MemoryBufferRef mb) {
|
|
std::unique_ptr<Archive> file =
|
|
CHECK(Archive::create(mb),
|
|
mb.getBufferIdentifier() + ": failed to parse archive");
|
|
Archive *archive = file.get();
|
|
make<std::unique_ptr<Archive>>(std::move(file)); // take ownership
|
|
|
|
std::vector<ArchiveMember> v;
|
|
Error err = Error::success();
|
|
|
|
// Thin archives refer to .o files, so --reproduces needs the .o files too.
|
|
bool addToTar = archive->isThin() && tar;
|
|
|
|
for (const Archive::Child &c : archive->children(err)) {
|
|
MemoryBufferRef mbref =
|
|
CHECK(c.getMemoryBufferRef(),
|
|
mb.getBufferIdentifier() +
|
|
": could not get the buffer for a child of the archive");
|
|
if (addToTar)
|
|
tar->append(relativeToRoot(check(c.getFullName())), mbref.getBuffer());
|
|
uint32_t modTime = toTimeT(
|
|
CHECK(c.getLastModified(), mb.getBufferIdentifier() +
|
|
": could not get the modification "
|
|
"time for a child of the archive"));
|
|
v.push_back({mbref, modTime});
|
|
}
|
|
if (err)
|
|
fatal(mb.getBufferIdentifier() +
|
|
": Archive::children failed: " + toString(std::move(err)));
|
|
|
|
return v;
|
|
}
|
|
|
|
static InputFile *addFile(StringRef path, bool forceLoadArchive) {
|
|
Optional<MemoryBufferRef> buffer = readFile(path);
|
|
if (!buffer)
|
|
return nullptr;
|
|
MemoryBufferRef mbref = *buffer;
|
|
InputFile *newFile = nullptr;
|
|
|
|
auto magic = identify_magic(mbref.getBuffer());
|
|
switch (magic) {
|
|
case file_magic::archive: {
|
|
std::unique_ptr<object::Archive> file = CHECK(
|
|
object::Archive::create(mbref), path + ": failed to parse archive");
|
|
|
|
if (!file->isEmpty() && !file->hasSymbolTable())
|
|
error(path + ": archive has no index; run ranlib to add one");
|
|
|
|
if (config->allLoad || forceLoadArchive) {
|
|
if (Optional<MemoryBufferRef> buffer = readFile(path)) {
|
|
for (const ArchiveMember &member : getArchiveMembers(*buffer)) {
|
|
inputFiles.insert(make<ObjFile>(member.mbref, member.modTime, path));
|
|
printArchiveMemberLoad(
|
|
(forceLoadArchive ? "-force_load" : "-all_load"),
|
|
inputFiles.back());
|
|
}
|
|
}
|
|
} else if (config->forceLoadObjC) {
|
|
for (const object::Archive::Symbol &sym : file->symbols())
|
|
if (sym.getName().startswith(objc::klass))
|
|
symtab->addUndefined(sym.getName(), /*isWeakRef=*/false);
|
|
|
|
// TODO: no need to look for ObjC sections for a given archive member if
|
|
// we already found that it contains an ObjC symbol. We should also
|
|
// consider creating a LazyObjFile class in order to avoid double-loading
|
|
// these files here and below (as part of the ArchiveFile).
|
|
if (Optional<MemoryBufferRef> buffer = readFile(path)) {
|
|
for (const ArchiveMember &member : getArchiveMembers(*buffer)) {
|
|
if (hasObjCSection(member.mbref)) {
|
|
inputFiles.insert(
|
|
make<ObjFile>(member.mbref, member.modTime, path));
|
|
printArchiveMemberLoad("-ObjC", inputFiles.back());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
newFile = make<ArchiveFile>(std::move(file));
|
|
break;
|
|
}
|
|
case file_magic::macho_object:
|
|
newFile = make<ObjFile>(mbref, getModTime(path), "");
|
|
break;
|
|
case file_magic::macho_dynamically_linked_shared_lib:
|
|
case file_magic::macho_dynamically_linked_shared_lib_stub:
|
|
case file_magic::tapi_file: {
|
|
if (Optional<DylibFile *> dylibFile = loadDylib(mbref))
|
|
newFile = *dylibFile;
|
|
break;
|
|
}
|
|
case file_magic::bitcode:
|
|
newFile = make<BitcodeFile>(mbref);
|
|
break;
|
|
default:
|
|
error(path + ": unhandled file type");
|
|
}
|
|
if (newFile) {
|
|
// printArchiveMemberLoad() prints both .a and .o names, so no need to
|
|
// print the .a name here.
|
|
if (config->printEachFile && magic != file_magic::archive)
|
|
message(toString(newFile));
|
|
inputFiles.insert(newFile);
|
|
}
|
|
return newFile;
|
|
}
|
|
|
|
static void addLibrary(StringRef name, bool isWeak) {
|
|
if (Optional<std::string> path = findLibrary(name)) {
|
|
auto *dylibFile = dyn_cast_or_null<DylibFile>(addFile(*path, false));
|
|
if (isWeak && dylibFile)
|
|
dylibFile->forceWeakImport = true;
|
|
return;
|
|
}
|
|
error("library not found for -l" + name);
|
|
}
|
|
|
|
static void addFramework(StringRef name, bool isWeak) {
|
|
if (Optional<std::string> path = findFramework(name)) {
|
|
auto *dylibFile = dyn_cast_or_null<DylibFile>(addFile(*path, false));
|
|
if (isWeak && dylibFile)
|
|
dylibFile->forceWeakImport = true;
|
|
return;
|
|
}
|
|
error("framework not found for -framework " + name);
|
|
}
|
|
|
|
// Parses LC_LINKER_OPTION contents, which can add additional command line flags.
|
|
void macho::parseLCLinkerOption(InputFile* f, unsigned argc, StringRef data) {
|
|
SmallVector<const char *, 4> argv;
|
|
size_t offset = 0;
|
|
for (unsigned i = 0; i < argc && offset < data.size(); ++i) {
|
|
argv.push_back(data.data() + offset);
|
|
offset += strlen(data.data() + offset) + 1;
|
|
}
|
|
if (argv.size() != argc || offset > data.size())
|
|
fatal(toString(f) + ": invalid LC_LINKER_OPTION");
|
|
|
|
MachOOptTable table;
|
|
unsigned missingIndex, missingCount;
|
|
opt::InputArgList args = table.ParseArgs(argv, missingIndex, missingCount);
|
|
if (missingCount)
|
|
fatal(Twine(args.getArgString(missingIndex)) + ": missing argument");
|
|
for (auto *arg : args.filtered(OPT_UNKNOWN))
|
|
error("unknown argument: " + arg->getAsString(args));
|
|
|
|
for (auto *arg : args) {
|
|
switch (arg->getOption().getID()) {
|
|
case OPT_l:
|
|
addLibrary(arg->getValue(), false);
|
|
break;
|
|
case OPT_framework:
|
|
addFramework(arg->getValue(), false);
|
|
break;
|
|
default:
|
|
error(arg->getSpelling() + " is not allowed in LC_LINKER_OPTION");
|
|
}
|
|
}
|
|
}
|
|
|
|
static void addFileList(StringRef path) {
|
|
Optional<MemoryBufferRef> buffer = readFile(path);
|
|
if (!buffer)
|
|
return;
|
|
MemoryBufferRef mbref = *buffer;
|
|
for (StringRef path : args::getLines(mbref))
|
|
addFile(path, false);
|
|
}
|
|
|
|
// An order file has one entry per line, in the following format:
|
|
//
|
|
// <cpu>:<object file>:<symbol name>
|
|
//
|
|
// <cpu> and <object file> are optional. If not specified, then that entry
|
|
// matches any symbol of that name. Parsing this format is not quite
|
|
// straightforward because the symbol name itself can contain colons, so when
|
|
// encountering a colon, we consider the preceding characters to decide if it
|
|
// can be a valid CPU type or file path.
|
|
//
|
|
// If a symbol is matched by multiple entries, then it takes the lowest-ordered
|
|
// entry (the one nearest to the front of the list.)
|
|
//
|
|
// The file can also have line comments that start with '#'.
|
|
static void parseOrderFile(StringRef path) {
|
|
Optional<MemoryBufferRef> buffer = readFile(path);
|
|
if (!buffer) {
|
|
error("Could not read order file at " + path);
|
|
return;
|
|
}
|
|
|
|
MemoryBufferRef mbref = *buffer;
|
|
size_t priority = std::numeric_limits<size_t>::max();
|
|
for (StringRef line : args::getLines(mbref)) {
|
|
StringRef objectFile, symbol;
|
|
line = line.take_until([](char c) { return c == '#'; }); // ignore comments
|
|
line = line.ltrim();
|
|
|
|
CPUType cpuType = StringSwitch<CPUType>(line)
|
|
.StartsWith("i386:", CPU_TYPE_I386)
|
|
.StartsWith("x86_64:", CPU_TYPE_X86_64)
|
|
.StartsWith("arm:", CPU_TYPE_ARM)
|
|
.StartsWith("arm64:", CPU_TYPE_ARM64)
|
|
.StartsWith("ppc:", CPU_TYPE_POWERPC)
|
|
.StartsWith("ppc64:", CPU_TYPE_POWERPC64)
|
|
.Default(CPU_TYPE_ANY);
|
|
// Drop the CPU type as well as the colon
|
|
if (cpuType != CPU_TYPE_ANY)
|
|
line = line.drop_until([](char c) { return c == ':'; }).drop_front();
|
|
// TODO: Update when we extend support for other CPUs
|
|
if (cpuType != CPU_TYPE_ANY && cpuType != CPU_TYPE_X86_64)
|
|
continue;
|
|
|
|
constexpr std::array<StringRef, 2> fileEnds = {".o:", ".o):"};
|
|
for (StringRef fileEnd : fileEnds) {
|
|
size_t pos = line.find(fileEnd);
|
|
if (pos != StringRef::npos) {
|
|
// Split the string around the colon
|
|
objectFile = line.take_front(pos + fileEnd.size() - 1);
|
|
line = line.drop_front(pos + fileEnd.size());
|
|
break;
|
|
}
|
|
}
|
|
symbol = line.trim();
|
|
|
|
if (!symbol.empty()) {
|
|
SymbolPriorityEntry &entry = config->priorities[symbol];
|
|
if (!objectFile.empty())
|
|
entry.objectFiles.insert(std::make_pair(objectFile, priority));
|
|
else
|
|
entry.anyObjectFile = std::max(entry.anyObjectFile, priority);
|
|
}
|
|
|
|
--priority;
|
|
}
|
|
}
|
|
|
|
// We expect sub-library names of the form "libfoo", which will match a dylib
|
|
// with a path of .*/libfoo.{dylib, tbd}.
|
|
// XXX ld64 seems to ignore the extension entirely when matching sub-libraries;
|
|
// I'm not sure what the use case for that is.
|
|
static bool markReexport(StringRef searchName, ArrayRef<StringRef> extensions) {
|
|
for (InputFile *file : inputFiles) {
|
|
if (auto *dylibFile = dyn_cast<DylibFile>(file)) {
|
|
StringRef filename = path::filename(dylibFile->getName());
|
|
if (filename.consume_front(searchName) &&
|
|
(filename.empty() ||
|
|
find(extensions, filename) != extensions.end())) {
|
|
dylibFile->reexport = true;
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
// This function is called on startup. We need this for LTO since
|
|
// LTO calls LLVM functions to compile bitcode files to native code.
|
|
// Technically this can be delayed until we read bitcode files, but
|
|
// we don't bother to do lazily because the initialization is fast.
|
|
static void initLLVM() {
|
|
InitializeAllTargets();
|
|
InitializeAllTargetMCs();
|
|
InitializeAllAsmPrinters();
|
|
InitializeAllAsmParsers();
|
|
}
|
|
|
|
static void compileBitcodeFiles() {
|
|
auto lto = make<BitcodeCompiler>();
|
|
for (InputFile *file : inputFiles)
|
|
if (auto *bitcodeFile = dyn_cast<BitcodeFile>(file))
|
|
lto->add(*bitcodeFile);
|
|
|
|
for (ObjFile *file : lto->compile())
|
|
inputFiles.insert(file);
|
|
}
|
|
|
|
// Replaces common symbols with defined symbols residing in __common sections.
|
|
// This function must be called after all symbol names are resolved (i.e. after
|
|
// all InputFiles have been loaded.) As a result, later operations won't see
|
|
// any CommonSymbols.
|
|
static void replaceCommonSymbols() {
|
|
for (macho::Symbol *sym : symtab->getSymbols()) {
|
|
auto *common = dyn_cast<CommonSymbol>(sym);
|
|
if (common == nullptr)
|
|
continue;
|
|
|
|
auto *isec = make<InputSection>();
|
|
isec->file = common->file;
|
|
isec->name = section_names::common;
|
|
isec->segname = segment_names::data;
|
|
isec->align = common->align;
|
|
// Casting to size_t will truncate large values on 32-bit architectures,
|
|
// but it's not really worth supporting the linking of 64-bit programs on
|
|
// 32-bit archs.
|
|
isec->data = {nullptr, static_cast<size_t>(common->size)};
|
|
isec->flags = S_ZEROFILL;
|
|
inputSections.push_back(isec);
|
|
|
|
replaceSymbol<Defined>(sym, sym->getName(), isec, /*value=*/0,
|
|
/*isWeakDef=*/false,
|
|
/*isExternal=*/true, common->privateExtern);
|
|
}
|
|
}
|
|
|
|
static inline char toLowerDash(char x) {
|
|
if (x >= 'A' && x <= 'Z')
|
|
return x - 'A' + 'a';
|
|
else if (x == ' ')
|
|
return '-';
|
|
return x;
|
|
}
|
|
|
|
static std::string lowerDash(StringRef s) {
|
|
return std::string(map_iterator(s.begin(), toLowerDash),
|
|
map_iterator(s.end(), toLowerDash));
|
|
}
|
|
|
|
static void handlePlatformVersion(const opt::Arg *arg) {
|
|
StringRef platformStr = arg->getValue(0);
|
|
StringRef minVersionStr = arg->getValue(1);
|
|
StringRef sdkVersionStr = arg->getValue(2);
|
|
|
|
// TODO(compnerd) see if we can generate this case list via XMACROS
|
|
config->platform.kind =
|
|
StringSwitch<PlatformKind>(lowerDash(platformStr))
|
|
.Cases("macos", "1", PlatformKind::macOS)
|
|
.Cases("ios", "2", PlatformKind::iOS)
|
|
.Cases("tvos", "3", PlatformKind::tvOS)
|
|
.Cases("watchos", "4", PlatformKind::watchOS)
|
|
.Cases("bridgeos", "5", PlatformKind::bridgeOS)
|
|
.Cases("mac-catalyst", "6", PlatformKind::macCatalyst)
|
|
.Cases("ios-simulator", "7", PlatformKind::iOSSimulator)
|
|
.Cases("tvos-simulator", "8", PlatformKind::tvOSSimulator)
|
|
.Cases("watchos-simulator", "9", PlatformKind::watchOSSimulator)
|
|
.Cases("driverkit", "10", PlatformKind::driverKit)
|
|
.Default(PlatformKind::unknown);
|
|
if (config->platform.kind == PlatformKind::unknown)
|
|
error(Twine("malformed platform: ") + platformStr);
|
|
// TODO: check validity of version strings, which varies by platform
|
|
// NOTE: ld64 accepts version strings with 5 components
|
|
// llvm::VersionTuple accepts no more than 4 components
|
|
// Has Apple ever published version strings with 5 components?
|
|
if (config->platform.minimum.tryParse(minVersionStr))
|
|
error(Twine("malformed minimum version: ") + minVersionStr);
|
|
if (config->platform.sdk.tryParse(sdkVersionStr))
|
|
error(Twine("malformed sdk version: ") + sdkVersionStr);
|
|
}
|
|
|
|
static void handleUndefined(const opt::Arg *arg) {
|
|
StringRef treatmentStr = arg->getValue(0);
|
|
config->undefinedSymbolTreatment =
|
|
StringSwitch<UndefinedSymbolTreatment>(treatmentStr)
|
|
.Case("error", UndefinedSymbolTreatment::error)
|
|
.Case("warning", UndefinedSymbolTreatment::warning)
|
|
.Case("suppress", UndefinedSymbolTreatment::suppress)
|
|
.Case("dynamic_lookup", UndefinedSymbolTreatment::dynamic_lookup)
|
|
.Default(UndefinedSymbolTreatment::unknown);
|
|
if (config->undefinedSymbolTreatment == UndefinedSymbolTreatment::unknown) {
|
|
warn(Twine("unknown -undefined TREATMENT '") + treatmentStr +
|
|
"', defaulting to 'error'");
|
|
config->undefinedSymbolTreatment = UndefinedSymbolTreatment::error;
|
|
}
|
|
}
|
|
|
|
static void warnIfDeprecatedOption(const opt::Option &opt) {
|
|
if (!opt.getGroup().isValid())
|
|
return;
|
|
if (opt.getGroup().getID() == OPT_grp_deprecated) {
|
|
warn("Option `" + opt.getPrefixedName() + "' is deprecated in ld64:");
|
|
warn(opt.getHelpText());
|
|
}
|
|
}
|
|
|
|
static void warnIfUnimplementedOption(const opt::Option &opt) {
|
|
if (!opt.getGroup().isValid() || !opt.hasFlag(DriverFlag::HelpHidden))
|
|
return;
|
|
switch (opt.getGroup().getID()) {
|
|
case OPT_grp_deprecated:
|
|
// warn about deprecated options elsewhere
|
|
break;
|
|
case OPT_grp_undocumented:
|
|
warn("Option `" + opt.getPrefixedName() +
|
|
"' is undocumented. Should lld implement it?");
|
|
break;
|
|
case OPT_grp_obsolete:
|
|
warn("Option `" + opt.getPrefixedName() +
|
|
"' is obsolete. Please modernize your usage.");
|
|
break;
|
|
case OPT_grp_ignored:
|
|
warn("Option `" + opt.getPrefixedName() + "' is ignored.");
|
|
break;
|
|
default:
|
|
warn("Option `" + opt.getPrefixedName() +
|
|
"' is not yet implemented. Stay tuned...");
|
|
break;
|
|
}
|
|
}
|
|
|
|
static const char *getReproduceOption(opt::InputArgList &args) {
|
|
if (auto *arg = args.getLastArg(OPT_reproduce))
|
|
return arg->getValue();
|
|
return getenv("LLD_REPRODUCE");
|
|
}
|
|
|
|
static bool isPie(opt::InputArgList &args) {
|
|
if (config->outputType != MH_EXECUTE || args.hasArg(OPT_no_pie))
|
|
return false;
|
|
|
|
// TODO: add logic here as we support more archs. E.g. i386 should default
|
|
// to PIE from 10.7, arm64 should always be PIE, etc
|
|
assert(config->arch == AK_x86_64 || config->arch == AK_x86_64h);
|
|
|
|
PlatformKind kind = config->platform.kind;
|
|
if (kind == PlatformKind::macOS &&
|
|
config->platform.minimum >= VersionTuple(10, 6))
|
|
return true;
|
|
|
|
if (kind == PlatformKind::iOSSimulator || kind == PlatformKind::driverKit)
|
|
return true;
|
|
|
|
return args.hasArg(OPT_pie);
|
|
}
|
|
|
|
static void parseClangOption(StringRef opt, const Twine &msg) {
|
|
std::string err;
|
|
raw_string_ostream os(err);
|
|
|
|
const char *argv[] = {"lld", opt.data()};
|
|
if (cl::ParseCommandLineOptions(2, argv, "", &os))
|
|
return;
|
|
os.flush();
|
|
error(msg + ": " + StringRef(err).trim());
|
|
}
|
|
|
|
static uint32_t parseDylibVersion(const opt::ArgList& args, unsigned id) {
|
|
const opt::Arg *arg = args.getLastArg(id);
|
|
if (!arg)
|
|
return 0;
|
|
|
|
if (config->outputType != MH_DYLIB) {
|
|
error(arg->getAsString(args) + ": only valid with -dylib");
|
|
return 0;
|
|
}
|
|
|
|
PackedVersion version;
|
|
if (!version.parse32(arg->getValue())) {
|
|
error(arg->getAsString(args) + ": malformed version");
|
|
return 0;
|
|
}
|
|
|
|
return version.rawValue();
|
|
}
|
|
|
|
bool macho::link(ArrayRef<const char *> argsArr, bool canExitEarly,
|
|
raw_ostream &stdoutOS, raw_ostream &stderrOS) {
|
|
lld::stdoutOS = &stdoutOS;
|
|
lld::stderrOS = &stderrOS;
|
|
|
|
stderrOS.enable_colors(stderrOS.has_colors());
|
|
// TODO: Set up error handler properly, e.g. the errorLimitExceededMsg
|
|
|
|
errorHandler().cleanupCallback = []() { freeArena(); };
|
|
|
|
MachOOptTable parser;
|
|
opt::InputArgList args = parser.parse(argsArr.slice(1));
|
|
|
|
if (args.hasArg(OPT_help_hidden)) {
|
|
parser.printHelp(argsArr[0], /*showHidden=*/true);
|
|
return true;
|
|
}
|
|
if (args.hasArg(OPT_help)) {
|
|
parser.printHelp(argsArr[0], /*showHidden=*/false);
|
|
return true;
|
|
}
|
|
if (args.hasArg(OPT_version)) {
|
|
message(getLLDVersion());
|
|
return true;
|
|
}
|
|
|
|
if (const char *path = getReproduceOption(args)) {
|
|
// Note that --reproduce is a debug option so you can ignore it
|
|
// if you are trying to understand the whole picture of the code.
|
|
Expected<std::unique_ptr<TarWriter>> errOrWriter =
|
|
TarWriter::create(path, path::stem(path));
|
|
if (errOrWriter) {
|
|
tar = std::move(*errOrWriter);
|
|
tar->append("response.txt", createResponseFile(args));
|
|
tar->append("version.txt", getLLDVersion() + "\n");
|
|
} else {
|
|
error("--reproduce: " + toString(errOrWriter.takeError()));
|
|
}
|
|
}
|
|
|
|
config = make<Configuration>();
|
|
symtab = make<SymbolTable>();
|
|
target = createTargetInfo(args);
|
|
|
|
config->entry = symtab->addUndefined(args.getLastArgValue(OPT_e, "_main"),
|
|
/*isWeakRef=*/false);
|
|
config->outputFile = args.getLastArgValue(OPT_o, "a.out");
|
|
config->installName =
|
|
args.getLastArgValue(OPT_install_name, config->outputFile);
|
|
config->headerPad = args::getHex(args, OPT_headerpad, /*Default=*/32);
|
|
config->headerPadMaxInstallNames =
|
|
args.hasArg(OPT_headerpad_max_install_names);
|
|
config->printEachFile = args.hasArg(OPT_t);
|
|
config->printWhyLoad = args.hasArg(OPT_why_load);
|
|
config->outputType = getOutputType(args);
|
|
config->ltoObjPath = args.getLastArgValue(OPT_object_path_lto);
|
|
config->runtimePaths = args::getStrings(args, OPT_rpath);
|
|
config->allLoad = args.hasArg(OPT_all_load);
|
|
config->forceLoadObjC = args.hasArg(OPT_ObjC);
|
|
config->demangle = args.hasArg(OPT_demangle);
|
|
config->implicitDylibs = !args.hasArg(OPT_no_implicit_dylibs);
|
|
|
|
if (const opt::Arg *arg = args.getLastArg(OPT_static, OPT_dynamic))
|
|
config->staticLink = (arg->getOption().getID() == OPT_static);
|
|
|
|
config->systemLibraryRoots = getSystemLibraryRoots(args);
|
|
config->librarySearchPaths =
|
|
getLibrarySearchPaths(args, config->systemLibraryRoots);
|
|
config->frameworkSearchPaths =
|
|
getFrameworkSearchPaths(args, config->systemLibraryRoots);
|
|
if (const opt::Arg *arg =
|
|
args.getLastArg(OPT_search_paths_first, OPT_search_dylibs_first))
|
|
config->searchDylibsFirst =
|
|
arg->getOption().getID() == OPT_search_dylibs_first;
|
|
|
|
config->dylibCompatibilityVersion =
|
|
parseDylibVersion(args, OPT_compatibility_version);
|
|
config->dylibCurrentVersion = parseDylibVersion(args, OPT_current_version);
|
|
|
|
config->saveTemps = args.hasArg(OPT_save_temps);
|
|
|
|
if (args.hasArg(OPT_v)) {
|
|
message(getLLDVersion());
|
|
message(StringRef("Library search paths:") +
|
|
(config->librarySearchPaths.size()
|
|
? "\n\t" + join(config->librarySearchPaths, "\n\t")
|
|
: ""));
|
|
message(StringRef("Framework search paths:") +
|
|
(config->frameworkSearchPaths.size()
|
|
? "\n\t" + join(config->frameworkSearchPaths, "\n\t")
|
|
: ""));
|
|
freeArena();
|
|
return !errorCount();
|
|
}
|
|
|
|
initLLVM(); // must be run before any call to addFile()
|
|
|
|
for (const auto &arg : args) {
|
|
const auto &opt = arg->getOption();
|
|
warnIfDeprecatedOption(opt);
|
|
warnIfUnimplementedOption(opt);
|
|
// TODO: are any of these better handled via filtered() or getLastArg()?
|
|
switch (opt.getID()) {
|
|
case OPT_INPUT:
|
|
addFile(arg->getValue(), false);
|
|
break;
|
|
case OPT_weak_library: {
|
|
auto *dylibFile =
|
|
dyn_cast_or_null<DylibFile>(addFile(arg->getValue(), false));
|
|
if (dylibFile)
|
|
dylibFile->forceWeakImport = true;
|
|
break;
|
|
}
|
|
case OPT_filelist:
|
|
addFileList(arg->getValue());
|
|
break;
|
|
case OPT_force_load:
|
|
addFile(arg->getValue(), true);
|
|
break;
|
|
case OPT_l:
|
|
case OPT_weak_l:
|
|
addLibrary(arg->getValue(), opt.getID() == OPT_weak_l);
|
|
break;
|
|
case OPT_framework:
|
|
case OPT_weak_framework:
|
|
addFramework(arg->getValue(), opt.getID() == OPT_weak_framework);
|
|
break;
|
|
case OPT_platform_version:
|
|
handlePlatformVersion(arg);
|
|
break;
|
|
case OPT_undefined:
|
|
handleUndefined(arg);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
config->isPic = config->outputType == MH_DYLIB ||
|
|
config->outputType == MH_BUNDLE || isPie(args);
|
|
|
|
// Now that all dylibs have been loaded, search for those that should be
|
|
// re-exported.
|
|
for (opt::Arg *arg : args.filtered(OPT_sub_library, OPT_sub_umbrella)) {
|
|
config->hasReexports = true;
|
|
StringRef searchName = arg->getValue();
|
|
std::vector<StringRef> extensions;
|
|
if (arg->getOption().getID() == OPT_sub_library)
|
|
extensions = {".dylib", ".tbd"};
|
|
else
|
|
extensions = {".tbd"};
|
|
if (!markReexport(searchName, extensions))
|
|
error(arg->getSpelling() + " " + searchName +
|
|
" does not match a supplied dylib");
|
|
}
|
|
|
|
// Parse LTO options.
|
|
if (auto *arg = args.getLastArg(OPT_mcpu))
|
|
parseClangOption(saver.save("-mcpu=" + StringRef(arg->getValue())),
|
|
arg->getSpelling());
|
|
|
|
for (auto *arg : args.filtered(OPT_mllvm))
|
|
parseClangOption(arg->getValue(), arg->getSpelling());
|
|
|
|
compileBitcodeFiles();
|
|
replaceCommonSymbols();
|
|
|
|
StringRef orderFile = args.getLastArgValue(OPT_order_file);
|
|
if (!orderFile.empty())
|
|
parseOrderFile(orderFile);
|
|
|
|
if (config->outputType == MH_EXECUTE && isa<Undefined>(config->entry)) {
|
|
error("undefined symbol: " + toString(*config->entry));
|
|
return false;
|
|
}
|
|
|
|
createSyntheticSections();
|
|
symtab->addDSOHandle(in.header);
|
|
|
|
for (opt::Arg *arg : args.filtered(OPT_sectcreate)) {
|
|
StringRef segName = arg->getValue(0);
|
|
StringRef sectName = arg->getValue(1);
|
|
StringRef fileName = arg->getValue(2);
|
|
Optional<MemoryBufferRef> buffer = readFile(fileName);
|
|
if (buffer)
|
|
inputFiles.insert(make<OpaqueFile>(*buffer, segName, sectName));
|
|
}
|
|
|
|
// Initialize InputSections.
|
|
for (InputFile *file : inputFiles) {
|
|
for (SubsectionMap &map : file->subsections) {
|
|
for (auto &p : map) {
|
|
InputSection *isec = p.second;
|
|
inputSections.push_back(isec);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Write to an output file.
|
|
writeResult();
|
|
|
|
if (canExitEarly)
|
|
exitLld(errorCount() ? 1 : 0);
|
|
|
|
return !errorCount();
|
|
}
|