contrib/llvm/tools/lld/Common/Args.cpp

   1 //===- Args.cpp -----------------------------------------------------------===//
   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 #include "lld/Common/Args.h"
  11 #include "lld/Common/ErrorHandler.h"
  12 #include "llvm/ADT/SmallVector.h"
  13 #include "llvm/ADT/StringExtras.h"
  14 #include "llvm/ADT/StringRef.h"
  15 #include "llvm/Option/ArgList.h"
  16 #include "llvm/Support/Path.h"
  17
  18 using namespace llvm;
  19 using namespace lld;
  20
  21 int lld::args::getInteger(opt::InputArgList &Args, unsigned Key, int Default) {
  22   auto *A = Args.getLastArg(Key);
  23   if (!A)
  24     return Default;
  25
  26   int V;
  27   if (to_integer(A->getValue(), V, 10))
  28     return V;
  29
  30   StringRef Spelling = Args.getArgString(A->getIndex());
  31   error(Spelling + ": number expected, but got '" + A->getValue() + "'");
  32   return 0;
  33 }
  34
  35 std::vector<StringRef> lld::args::getStrings(opt::InputArgList &Args, int Id) {
  36   std::vector<StringRef> V;
  37   for (auto *Arg : Args.filtered(Id))
  38     V.push_back(Arg->getValue());
  39   return V;
  40 }
  41
  42 uint64_t lld::args::getZOptionValue(opt::InputArgList &Args, int Id,
  43                                     StringRef Key, uint64_t Default) {
  44   for (auto *Arg : Args.filtered_reverse(Id)) {
  45     std::pair<StringRef, StringRef> KV = StringRef(Arg->getValue()).split('=');
  46     if (KV.first == Key) {
  47       uint64_t Result = Default;
  48       if (!to_integer(KV.second, Result))
  49         error("invalid " + Key + ": " + KV.second);
  50       return Result;
  51     }
  52   }
  53   return Default;
  54 }
  55
  56 std::vector<StringRef> lld::args::getLines(MemoryBufferRef MB) {
  57   SmallVector<StringRef, 0> Arr;
  58   MB.getBuffer().split(Arr, '\n');
  59
  60   std::vector<StringRef> Ret;
  61   for (StringRef S : Arr) {
  62     S = S.trim();
  63     if (!S.empty() && S[0] != '#')
  64       Ret.push_back(S);
  65   }
  66   return Ret;
  67 }
  68
  69 StringRef lld::args::getFilenameWithoutExe(StringRef Path) {
  70   if (Path.endswith_lower(".exe"))
  71     return sys::path::stem(Path);
  72   return sys::path::filename(Path);
  73 }