contrib/llvm/tools/lld/tools/lld/lld.cpp

   1 //===- tools/lld/lld.cpp - Linker Driver Dispatcher -----------------------===//
   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 // This is the entry point to the lld driver. This is a thin wrapper which
  11 // dispatches to the given platform specific driver.
  12 //
  13 // If there is -flavor option, it is dispatched according to the arguments.
  14 // If the flavor parameter is not present, then it is dispatched according
  15 // to argv[0].
  16 //
  17 //===----------------------------------------------------------------------===//
  18
  19 #include "lld/Common/Driver.h"
  20 #include "llvm/ADT/STLExtras.h"
  21 #include "llvm/ADT/StringSwitch.h"
  22 #include "llvm/ADT/Twine.h"
  23 #include "llvm/Support/ManagedStatic.h"
  24 #include "llvm/Support/Path.h"
  25 #include "llvm/Support/PrettyStackTrace.h"
  26 #include "llvm/Support/Signals.h"
  27
  28 using namespace lld;
  29 using namespace llvm;
  30 using namespace llvm::sys;
  31
  32 enum Flavor {
  33   Invalid,
  34   Gnu,     // -flavor gnu
  35   WinLink, // -flavor link
  36   Darwin,  // -flavor darwin
  37   Wasm,    // -flavor wasm
  38 };
  39
  40 LLVM_ATTRIBUTE_NORETURN static void die(const Twine &S) {
  41   errs() << S << "\n";
  42   exit(1);
  43 }
  44
  45 static Flavor getFlavor(StringRef S) {
  46   return StringSwitch<Flavor>(S)
  47       .CasesLower("ld", "ld.lld", "gnu", Gnu)
  48       .CasesLower("wasm", "ld-wasm", Wasm)
  49       .CaseLower("link", WinLink)
  50       .CasesLower("ld64", "ld64.lld", "darwin", Darwin)
  51       .Default(Invalid);
  52 }
  53
  54 static bool isPETarget(const std::vector<const char *> &V) {
  55   for (auto It = V.begin(); It + 1 != V.end(); ++It) {
  56     if (StringRef(*It) != "-m")
  57       continue;
  58     StringRef S = *(It + 1);
  59     return S == "i386pe" || S == "i386pep" || S == "thumb2pe" || S == "arm64pe";
  60   }
  61   return false;
  62 }
  63
  64 static Flavor parseProgname(StringRef Progname) {
  65 #if __APPLE__
  66   // Use Darwin driver for "ld" on Darwin.
  67   if (Progname == "ld")
  68     return Darwin;
  69 #endif
  70
  71 #if LLVM_ON_UNIX
  72   // Use GNU driver for "ld" on other Unix-like system.
  73   if (Progname == "ld")
  74     return Gnu;
  75 #endif
  76
  77   // Progname may be something like "lld-gnu". Parse it.
  78   SmallVector<StringRef, 3> V;
  79   Progname.split(V, "-");
  80   for (StringRef S : V)
  81     if (Flavor F = getFlavor(S))
  82       return F;
  83   return Invalid;
  84 }
  85
  86 static Flavor parseFlavor(std::vector<const char *> &V) {
  87   // Parse -flavor option.
  88   if (V.size() > 1 && V[1] == StringRef("-flavor")) {
  89     if (V.size() <= 2)
  90       die("missing arg value for '-flavor'");
  91     Flavor F = getFlavor(V[2]);
  92     if (F == Invalid)
  93       die("Unknown flavor: " + StringRef(V[2]));
  94     V.erase(V.begin() + 1, V.begin() + 3);
  95     return F;
  96   }
  97
  98   // Deduct the flavor from argv[0].
  99   StringRef Arg0 = path::filename(V[0]);
 100   if (Arg0.endswith_lower(".exe"))
 101     Arg0 = Arg0.drop_back(4);
 102   return parseProgname(Arg0);
 103 }
 104
 105 /// Universal linker main(). This linker emulates the gnu, darwin, or
 106 /// windows linker based on the argv[0] or -flavor option.
 107 int main(int Argc, const char **Argv) {
 108   // Standard set up, so program fails gracefully.
 109   sys::PrintStackTraceOnErrorSignal(Argv[0]);
 110   PrettyStackTraceProgram StackPrinter(Argc, Argv);
 111   llvm_shutdown_obj Shutdown;
 112
 113   std::vector<const char *> Args(Argv, Argv + Argc);
 114 #ifdef __FreeBSD__
 115   return !elf::link(Args, true);
 116 #else
 117   switch (parseFlavor(Args)) {
 118   case Gnu:
 119     if (isPETarget(Args))
 120       return !mingw::link(Args);
 121     return !elf::link(Args, true);
 122   case WinLink:
 123     return !coff::link(Args, true);
 124   case Darwin:
 125     return !mach_o::link(Args);
 126   case Wasm:
 127     return !wasm::link(Args, true);
 128   default:
 129     die("lld is a generic driver.\n"
 130         "Invoke ld.lld (Unix), ld (macOS) or lld-link (Windows) instead.");
 131   }
 132 #endif
 133 }