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.starts_with("0x") || s.starts_with("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.ends_with_insensitive(".exe"))
91     return sys::path::stem(path);
92   return sys::path::filename(path);
93 }
94