|
11 | 11 | // |
12 | 12 | //===----------------------------------------------------------------------===// |
13 | 13 |
|
| 14 | +#include "Opts.inc" |
14 | 15 | #include "llvm/Object/Binary.h" |
| 16 | +#include "llvm/Option/Arg.h" |
| 17 | +#include "llvm/Option/ArgList.h" |
| 18 | +#include "llvm/Option/Option.h" |
15 | 19 | #include "llvm/Support/CommandLine.h" |
16 | 20 | #include "llvm/Support/Error.h" |
17 | 21 | #include "llvm/Support/Format.h" |
18 | 22 | #include "llvm/Support/InitLLVM.h" |
19 | 23 | #include "llvm/Support/MemoryBuffer.h" |
20 | 24 | #include "llvm/Support/Program.h" |
| 25 | +#include "llvm/Support/WithColor.h" |
21 | 26 | #include <cctype> |
22 | 27 | #include <string> |
23 | 28 |
|
24 | 29 | using namespace llvm; |
25 | 30 | using namespace llvm::object; |
26 | 31 |
|
| 32 | +namespace { |
| 33 | +enum ID { |
| 34 | + OPT_INVALID = 0, // This is not an option ID. |
| 35 | +#define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM, \ |
| 36 | + HELPTEXT, METAVAR, VALUES) \ |
| 37 | + OPT_##ID, |
| 38 | +#include "Opts.inc" |
| 39 | +#undef OPTION |
| 40 | +}; |
| 41 | + |
| 42 | +#define PREFIX(NAME, VALUE) const char *const NAME[] = VALUE; |
| 43 | +#include "Opts.inc" |
| 44 | +#undef PREFIX |
| 45 | + |
| 46 | +static const opt::OptTable::Info InfoTable[] = { |
| 47 | +#define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM, \ |
| 48 | + HELPTEXT, METAVAR, VALUES) \ |
| 49 | + { \ |
| 50 | + PREFIX, NAME, HELPTEXT, \ |
| 51 | + METAVAR, OPT_##ID, opt::Option::KIND##Class, \ |
| 52 | + PARAM, FLAGS, OPT_##GROUP, \ |
| 53 | + OPT_##ALIAS, ALIASARGS, VALUES}, |
| 54 | +#include "Opts.inc" |
| 55 | +#undef OPTION |
| 56 | +}; |
| 57 | + |
| 58 | +class StringsOptTable : public opt::OptTable { |
| 59 | +public: |
| 60 | + StringsOptTable() : OptTable(InfoTable) { setGroupedShortOptions(true); } |
| 61 | +}; |
| 62 | +} // namespace |
| 63 | + |
| 64 | +const char ToolName[] = "llvm-strings"; |
| 65 | + |
27 | 66 | static cl::list<std::string> InputFileNames(cl::Positional, |
28 | 67 | cl::desc("<input object files>"), |
29 | 68 | cl::ZeroOrMore); |
30 | 69 |
|
31 | | -static cl::opt<bool> |
32 | | - PrintFileName("print-file-name", |
33 | | - cl::desc("Print the name of the file before each string")); |
34 | | -static cl::alias PrintFileNameShort("f", cl::desc(""), |
35 | | - cl::aliasopt(PrintFileName)); |
36 | | - |
37 | | -static cl::opt<int> |
38 | | - MinLength("bytes", cl::desc("Print sequences of the specified length"), |
39 | | - cl::init(4)); |
40 | | -static cl::alias MinLengthShort("n", cl::desc(""), cl::aliasopt(MinLength)); |
41 | | - |
42 | | -static cl::opt<bool> |
43 | | - AllSections("all", |
44 | | - cl::desc("Check all sections, not just the data section")); |
45 | | -static cl::alias AllSectionsShort("a", cl::desc(""), |
46 | | - cl::aliasopt(AllSections)); |
| 70 | +static int MinLength = 4; |
| 71 | +static bool PrintFileName; |
47 | 72 |
|
48 | 73 | enum radix { none, octal, hexadecimal, decimal }; |
49 | | -static cl::opt<radix> |
50 | | - Radix("radix", cl::desc("print the offset within the file"), |
51 | | - cl::values(clEnumValN(octal, "o", "octal"), |
52 | | - clEnumValN(hexadecimal, "x", "hexadecimal"), |
53 | | - clEnumValN(decimal, "d", "decimal")), |
54 | | - cl::init(none)); |
55 | | -static cl::alias RadixShort("t", cl::desc(""), cl::aliasopt(Radix)); |
| 74 | +static radix Radix; |
56 | 75 |
|
57 | | -static cl::extrahelp |
58 | | - HelpResponse("\nPass @FILE as argument to read options from FILE.\n"); |
| 76 | +LLVM_ATTRIBUTE_NORETURN static void reportCmdLineError(const Twine &Message) { |
| 77 | + WithColor::error(errs(), ToolName) << Message << "\n"; |
| 78 | + exit(1); |
| 79 | +} |
| 80 | + |
| 81 | +template <typename T> |
| 82 | +static void parseIntArg(const opt::InputArgList &Args, int ID, T &Value) { |
| 83 | + if (const opt::Arg *A = Args.getLastArg(ID)) { |
| 84 | + StringRef V(A->getValue()); |
| 85 | + if (!llvm::to_integer(V, Value, 0) || Value <= 0) |
| 86 | + reportCmdLineError("expected a positive integer, but got '" + V + "'"); |
| 87 | + } |
| 88 | +} |
59 | 89 |
|
60 | 90 | static void strings(raw_ostream &OS, StringRef FileName, StringRef Contents) { |
61 | 91 | auto print = [&OS, FileName](unsigned Offset, StringRef L) { |
@@ -96,13 +126,48 @@ static void strings(raw_ostream &OS, StringRef FileName, StringRef Contents) { |
96 | 126 |
|
97 | 127 | int main(int argc, char **argv) { |
98 | 128 | InitLLVM X(argc, argv); |
| 129 | + BumpPtrAllocator A; |
| 130 | + StringSaver Saver(A); |
| 131 | + StringsOptTable Tbl; |
| 132 | + opt::InputArgList Args = |
| 133 | + Tbl.parseArgs(argc, argv, OPT_UNKNOWN, Saver, |
| 134 | + [&](StringRef Msg) { reportCmdLineError(Msg); }); |
| 135 | + if (Args.hasArg(OPT_help)) { |
| 136 | + Tbl.printHelp( |
| 137 | + outs(), |
| 138 | + (Twine(ToolName) + " [options] <input object files>").str().c_str(), |
| 139 | + "llvm string dumper"); |
| 140 | + // TODO Replace this with OptTable API once it adds extrahelp support. |
| 141 | + outs() << "\nPass @FILE as argument to read options from FILE.\n"; |
| 142 | + return 0; |
| 143 | + } |
| 144 | + if (Args.hasArg(OPT_version)) { |
| 145 | + outs() << ToolName << '\n'; |
| 146 | + cl::PrintVersionMessage(); |
| 147 | + return 0; |
| 148 | + } |
| 149 | + |
| 150 | + parseIntArg(Args, OPT_bytes_EQ, MinLength); |
| 151 | + PrintFileName = Args.hasArg(OPT_print_file_name); |
| 152 | + StringRef R = Args.getLastArgValue(OPT_radix_EQ); |
| 153 | + if (R.empty()) |
| 154 | + Radix = none; |
| 155 | + else if (R == "o") |
| 156 | + Radix = octal; |
| 157 | + else if (R == "d") |
| 158 | + Radix = decimal; |
| 159 | + else if (R == "x") |
| 160 | + Radix = hexadecimal; |
| 161 | + else |
| 162 | + reportCmdLineError("--radix value should be one of: '' (no offset), 'o' " |
| 163 | + "(octal), 'd' (decimal), 'x' (hexadecimal)"); |
99 | 164 |
|
100 | | - cl::ParseCommandLineOptions(argc, argv, "llvm string dumper\n"); |
101 | 165 | if (MinLength == 0) { |
102 | 166 | errs() << "invalid minimum string length 0\n"; |
103 | 167 | return EXIT_FAILURE; |
104 | 168 | } |
105 | 169 |
|
| 170 | + std::vector<std::string> InputFileNames = Args.getAllArgValues(OPT_INPUT); |
106 | 171 | if (InputFileNames.empty()) |
107 | 172 | InputFileNames.push_back("-"); |
108 | 173 |
|
|
0 commit comments