contrib/llvm/lib/ToolDrivers/llvm-lib/LibDriver.cpp

   1 //===- LibDriver.cpp - lib.exe-compatible driver --------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // Defines an interface to a lib.exe-compatible driver that also understands
  11 // bitcode files. Used by llvm-lib and lld-link /lib.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/ToolDrivers/llvm-lib/LibDriver.h"
  16 #include "llvm/ADT/STLExtras.h"
  17 #include "llvm/BinaryFormat/Magic.h"
  18 #include "llvm/Object/ArchiveWriter.h"
  19 #include "llvm/Option/Arg.h"
  20 #include "llvm/Option/ArgList.h"
  21 #include "llvm/Option/Option.h"
  22 #include "llvm/Support/CommandLine.h"
  23 #include "llvm/Support/Path.h"
  24 #include "llvm/Support/Process.h"
  25 #include "llvm/Support/StringSaver.h"
  26 #include "llvm/Support/raw_ostream.h"
  27
  28 using namespace llvm;
  29
  30 namespace {
  31
  32 enum {
  33   OPT_INVALID = 0,
  34 #define OPTION(_1, _2, ID, _4, _5, _6, _7, _8, _9, _10, _11, _12) OPT_##ID,
  35 #include "Options.inc"
  36 #undef OPTION
  37 };
  38
  39 #define PREFIX(NAME, VALUE) const char *const NAME[] = VALUE;
  40 #include "Options.inc"
  41 #undef PREFIX
  42
  43 static const opt::OptTable::Info InfoTable[] = {
  44 #define OPTION(X1, X2, ID, KIND, GROUP, ALIAS, X7, X8, X9, X10, X11, X12)      \
  45   {X1, X2, X10,         X11,         OPT_##ID, opt::Option::KIND##Class,       \
  46    X9, X8, OPT_##GROUP, OPT_##ALIAS, X7,       X12},
  47 #include "Options.inc"
  48 #undef OPTION
  49 };
  50
  51 class LibOptTable : public opt::OptTable {
  52 public:
  53   LibOptTable() : OptTable(InfoTable, true) {}
  54 };
  55
  56 }
  57
  58 static std::string getOutputPath(opt::InputArgList *Args,
  59                                  const NewArchiveMember &FirstMember) {
  60   if (auto *Arg = Args->getLastArg(OPT_out))
  61     return Arg->getValue();
  62   SmallString<128> Val = StringRef(FirstMember.Buf->getBufferIdentifier());
  63   sys::path::replace_extension(Val, ".lib");
  64   return Val.str();
  65 }
  66
  67 static std::vector<StringRef> getSearchPaths(opt::InputArgList *Args,
  68                                              StringSaver &Saver) {
  69   std::vector<StringRef> Ret;
  70   // Add current directory as first item of the search path.
  71   Ret.push_back("");
  72
  73   // Add /libpath flags.
  74   for (auto *Arg : Args->filtered(OPT_libpath))
  75     Ret.push_back(Arg->getValue());
  76
  77   // Add $LIB.
  78   Optional<std::string> EnvOpt = sys::Process::GetEnv("LIB");
  79   if (!EnvOpt.hasValue())
  80     return Ret;
  81   StringRef Env = Saver.save(*EnvOpt);
  82   while (!Env.empty()) {
  83     StringRef Path;
  84     std::tie(Path, Env) = Env.split(';');
  85     Ret.push_back(Path);
  86   }
  87   return Ret;
  88 }
  89
  90 static std::string findInputFile(StringRef File, ArrayRef<StringRef> Paths) {
  91   for (StringRef Dir : Paths) {
  92     SmallString<128> Path = Dir;
  93     sys::path::append(Path, File);
  94     if (sys::fs::exists(Path))
  95       return Path.str().str();
  96   }
  97   return "";
  98 }
  99
 100 int llvm::libDriverMain(ArrayRef<const char *> ArgsArr) {
 101   BumpPtrAllocator Alloc;
 102   StringSaver Saver(Alloc);
 103
 104   // Parse command line arguments.
 105   SmallVector<const char *, 20> NewArgs(ArgsArr.begin(), ArgsArr.end());
 106   cl::ExpandResponseFiles(Saver, cl::TokenizeWindowsCommandLine, NewArgs);
 107   ArgsArr = NewArgs;
 108
 109   LibOptTable Table;
 110   unsigned MissingIndex;
 111   unsigned MissingCount;
 112   opt::InputArgList Args =
 113       Table.ParseArgs(ArgsArr.slice(1), MissingIndex, MissingCount);
 114   if (MissingCount) {
 115     llvm::errs() << "missing arg value for \""
 116                  << Args.getArgString(MissingIndex) << "\", expected "
 117                  << MissingCount
 118                  << (MissingCount == 1 ? " argument.\n" : " arguments.\n");
 119     return 1;
 120   }
 121   for (auto *Arg : Args.filtered(OPT_UNKNOWN))
 122     llvm::errs() << "ignoring unknown argument: " << Arg->getSpelling() << "\n";
 123
 124   // Handle /help
 125   if (Args.hasArg(OPT_help)) {
 126     Table.PrintHelp(outs(), ArgsArr[0], "LLVM Lib");
 127     return 0;
 128   }
 129
 130   // If no input files, silently do nothing to match lib.exe.
 131   if (!Args.hasArgNoClaim(OPT_INPUT))
 132     return 0;
 133
 134   std::vector<StringRef> SearchPaths = getSearchPaths(&Args, Saver);
 135
 136   // Create a NewArchiveMember for each input file.
 137   std::vector<NewArchiveMember> Members;
 138   for (auto *Arg : Args.filtered(OPT_INPUT)) {
 139     std::string Path = findInputFile(Arg->getValue(), SearchPaths);
 140     if (Path.empty()) {
 141       llvm::errs() << Arg->getValue() << ": no such file or directory\n";
 142       return 1;
 143     }
 144
 145     Expected<NewArchiveMember> MOrErr =
 146         NewArchiveMember::getFile(Saver.save(Path), /*Deterministic=*/true);
 147     if (!MOrErr) {
 148       handleAllErrors(MOrErr.takeError(), [&](const ErrorInfoBase &EIB) {
 149         llvm::errs() << Arg->getValue() << ": " << EIB.message() << "\n";
 150       });
 151       return 1;
 152     }
 153
 154     file_magic Magic = identify_magic(MOrErr->Buf->getBuffer());
 155     if (Magic != file_magic::coff_object && Magic != file_magic::bitcode &&
 156         Magic != file_magic::windows_resource) {
 157       llvm::errs() << Arg->getValue()
 158                    << ": not a COFF object, bitcode or resource file\n";
 159       return 1;
 160     }
 161     Members.emplace_back(std::move(*MOrErr));
 162   }
 163
 164   // Create an archive file.
 165   std::string OutputPath = getOutputPath(&Args, Members[0]);
 166   if (Error E =
 167           writeArchive(OutputPath, Members,
 168                        /*WriteSymtab=*/true, object::Archive::K_GNU,
 169                        /*Deterministic*/ true, Args.hasArg(OPT_llvmlibthin))) {
 170     handleAllErrors(std::move(E), [&](const ErrorInfoBase &EI) {
 171       llvm::errs() << OutputPath << ": " << EI.message() << "\n";
 172     });
 173     return 1;
 174   }
 175
 176   return 0;
 177 }