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
getInteger(opt::InputArgList & Args,unsigned Key,int Default)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
getStrings(opt::InputArgList & Args,int Id)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
getZOptionValue(opt::InputArgList & Args,int Id,StringRef Key,uint64_t Default)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
getLines(MemoryBufferRef MB)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
getFilenameWithoutExe(StringRef Path)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 }
74