| //===- 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 "OutputSegment.h" |
| #include "SymbolTable.h" |
| #include "Symbols.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/Version.h" |
| #include "llvm/ADT/StringRef.h" |
| #include "llvm/BinaryFormat/MachO.h" |
| #include "llvm/BinaryFormat/Magic.h" |
| #include "llvm/Option/ArgList.h" |
| #include "llvm/Option/Option.h" |
| #include "llvm/Support/MemoryBuffer.h" |
| |
| using namespace llvm; |
| using namespace llvm::MachO; |
| using namespace llvm::sys; |
| using namespace lld; |
| using namespace lld::macho; |
| |
| Configuration *lld::macho::config; |
| |
| // Create prefix string literals used in Options.td |
| #define PREFIX(NAME, VALUE) const char *NAME[] = VALUE; |
| #include "Options.inc" |
| #undef PREFIX |
| |
| // Create table mapping all options defined in Options.td |
| static const opt::OptTable::Info optInfo[] = { |
| #define OPTION(X1, X2, ID, KIND, GROUP, ALIAS, X7, X8, X9, X10, X11, X12) \ |
| {X1, X2, X10, X11, OPT_##ID, opt::Option::KIND##Class, \ |
| X9, X8, OPT_##GROUP, OPT_##ALIAS, X7, X12}, |
| #include "Options.inc" |
| #undef OPTION |
| }; |
| |
| MachOOptTable::MachOOptTable() : OptTable(optInfo) {} |
| |
| opt::InputArgList MachOOptTable::parse(ArrayRef<const char *> argv) { |
| // Make InputArgList from string vectors. |
| unsigned missingIndex; |
| unsigned missingCount; |
| SmallVector<const char *, 256> vec(argv.data(), argv.data() + argv.size()); |
| |
| opt::InputArgList args = ParseArgs(vec, missingIndex, missingCount); |
| |
| if (missingCount) |
| error(Twine(args.getArgString(missingIndex)) + ": missing argument"); |
| |
| for (opt::Arg *arg : args.filtered(OPT_UNKNOWN)) |
| error("unknown argument: " + arg->getSpelling()); |
| return args; |
| } |
| |
| static TargetInfo *createTargetInfo(opt::InputArgList &args) { |
| StringRef s = args.getLastArgValue(OPT_arch, "x86_64"); |
| if (s != "x86_64") |
| error("missing or unsupported -arch " + s); |
| return createX86_64TargetInfo(); |
| } |
| |
| static void addFile(StringRef path) { |
| Optional<MemoryBufferRef> buffer = readFile(path); |
| if (!buffer) |
| return; |
| MemoryBufferRef mbref = *buffer; |
| |
| switch (identify_magic(mbref.getBuffer())) { |
| case file_magic::macho_object: |
| inputFiles.push_back(make<ObjFile>(mbref)); |
| break; |
| default: |
| error(path + ": unhandled file type"); |
| } |
| } |
| |
| bool macho::link(llvm::ArrayRef<const char *> argsArr, bool canExitEarly, |
| raw_ostream &stdoutOS, raw_ostream &stderrOS) { |
| lld::stdoutOS = &stdoutOS; |
| lld::stderrOS = &stderrOS; |
| |
| MachOOptTable parser; |
| opt::InputArgList args = parser.parse(argsArr.slice(1)); |
| |
| if (args.hasArg(OPT_v)) { |
| message(getLLDVersion()); |
| freeArena(); |
| return !errorCount(); |
| } |
| |
| config = make<Configuration>(); |
| symtab = make<SymbolTable>(); |
| target = createTargetInfo(args); |
| |
| config->entry = symtab->addUndefined(args.getLastArgValue(OPT_e, "_main")); |
| config->outputFile = args.getLastArgValue(OPT_o, "a.out"); |
| |
| getOrCreateOutputSegment("__TEXT", VM_PROT_READ | VM_PROT_EXECUTE); |
| getOrCreateOutputSegment("__DATA", VM_PROT_READ | VM_PROT_WRITE); |
| |
| for (opt::Arg *arg : args) { |
| switch (arg->getOption().getID()) { |
| case OPT_INPUT: |
| addFile(arg->getValue()); |
| break; |
| } |
| } |
| |
| if (!isa<Defined>(config->entry)) { |
| error("undefined symbol: " + config->entry->getName()); |
| return false; |
| } |
| |
| // Initialize InputSections. |
| for (InputFile *file : inputFiles) |
| for (InputSection *sec : file->sections) |
| inputSections.push_back(sec); |
| |
| // Add input sections to output segments. |
| for (InputSection *isec : inputSections) { |
| OutputSegment *os = |
| getOrCreateOutputSegment(isec->segname, VM_PROT_READ | VM_PROT_WRITE); |
| os->sections[isec->name].push_back(isec); |
| } |
| |
| // Write to an output file. |
| writeResult(); |
| |
| if (canExitEarly) |
| exitLld(errorCount() ? 1 : 0); |
| |
| freeArena(); |
| return !errorCount(); |
| } |