lld/Common/Args.cpp

   1 //===- Args.cpp -----------------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "lld/Common/Args.h"
  10 #include "lld/Common/ErrorHandler.h"
  11 #include "llvm/ADT/SmallVector.h"
  12 #include "llvm/ADT/StringExtras.h"
  13 #include "llvm/ADT/StringRef.h"
  14 #include "llvm/Option/ArgList.h"
  15 #include "llvm/Support/Path.h"
  16
  17 using namespace llvm;
  18 using namespace lld;
  19
  20 // TODO(sbc): Remove this once CGOptLevel can be set completely based on bitcode
  21 // function metadata.
  22 int lld::args::getCGOptLevel(int optLevelLTO) {
  23   return std::clamp(optLevelLTO, 2, 3);
  24 }
  25
  26 static int64_t getInteger(opt::InputArgList &args, unsigned key,
  27                           int64_t Default, unsigned base) {
  28   auto *a = args.getLastArg(key);
  29   if (!a)
  30     return Default;
  31
  32   int64_t v;
  33   StringRef s = a->getValue();
  34   if (base == 16 && (s.startswith("0x") || s.startswith("0X")))
  35     s = s.drop_front(2);
  36   if (to_integer(s, v, base))
  37     return v;
  38
  39   StringRef spelling = args.getArgString(a->getIndex());
  40   error(spelling + ": number expected, but got '" + a->getValue() + "'");
  41   return 0;
  42 }
  43
  44 int64_t lld::args::getInteger(opt::InputArgList &args, unsigned key,
  45                               int64_t Default) {
  46   return ::getInteger(args, key, Default, 10);
  47 }
  48
  49 int64_t lld::args::getHex(opt::InputArgList &args, unsigned key,
  50                           int64_t Default) {
  51   return ::getInteger(args, key, Default, 16);
  52 }
  53
  54 SmallVector<StringRef, 0> lld::args::getStrings(opt::InputArgList &args,
  55                                                 int id) {
  56   SmallVector<StringRef, 0> v;
  57   for (auto *arg : args.filtered(id))
  58     v.push_back(arg->getValue());
  59   return v;
  60 }
  61
  62 uint64_t lld::args::getZOptionValue(opt::InputArgList &args, int id,
  63                                     StringRef key, uint64_t Default) {
  64   for (auto *arg : args.filtered_reverse(id)) {
  65     std::pair<StringRef, StringRef> kv = StringRef(arg->getValue()).split('=');
  66     if (kv.first == key) {
  67       uint64_t result = Default;
  68       if (!to_integer(kv.second, result))
  69         error("invalid " + key + ": " + kv.second);
  70       return result;
  71     }
  72   }
  73   return Default;
  74 }
  75
  76 std::vector<StringRef> lld::args::getLines(MemoryBufferRef mb) {
  77   SmallVector<StringRef, 0> arr;
  78   mb.getBuffer().split(arr, '\n');
  79
  80   std::vector<StringRef> ret;
  81   for (StringRef s : arr) {
  82     s = s.trim();
  83     if (!s.empty() && s[0] != '#')
  84       ret.push_back(s);
  85   }
  86   return ret;
  87 }
  88
  89 StringRef lld::args::getFilenameWithoutExe(StringRef path) {
  90   if (path.endswith_insensitive(".exe"))
  91     return sys::path::stem(path);
  92   return sys::path::filename(path);
  93 }