1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
//===-- COFFDirectiveParser.cpp - JITLink coff directive parser --*- C++ -*===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// MSVC COFF directive parser
//
//===----------------------------------------------------------------------===//
#include "COFFDirectiveParser.h"
#include <array>
using namespace llvm;
using namespace jitlink;
#define DEBUG_TYPE "jitlink"
#define OPTTABLE_STR_TABLE_CODE
#include "COFFOptions.inc"
#undef OPTTABLE_STR_TABLE_CODE
#define OPTTABLE_PREFIXES_TABLE_CODE
#include "COFFOptions.inc"
#undef OPTTABLE_PREFIXES_TABLE_CODE
#define OPTTABLE_PREFIXES_UNION_CODE
#include "COFFOptions.inc"
#undef OPTTABLE_PREFIXES_UNION_CODE
// Create table mapping all options defined in COFFOptions.td
using namespace llvm::opt;
static constexpr opt::OptTable::Info infoTable[] = {
#define OPTION(...) \
LLVM_CONSTRUCT_OPT_INFO_WITH_ID_PREFIX(COFF_OPT_, __VA_ARGS__),
#include "COFFOptions.inc"
#undef OPTION
};
class COFFOptTable : public opt::PrecomputedOptTable {
public:
COFFOptTable()
: PrecomputedOptTable(OptionStrTable, OptionPrefixesTable, infoTable,
OptionPrefixesUnion, true) {}
};
static COFFOptTable optTable;
Expected<opt::InputArgList> COFFDirectiveParser::parse(StringRef Str) {
SmallVector<StringRef, 16> Tokens;
SmallVector<const char *, 16> Buffer;
cl::TokenizeWindowsCommandLineNoCopy(Str, saver, Tokens);
for (StringRef Tok : Tokens) {
bool HasNul = Tok.end() != Str.end() && Tok.data()[Tok.size()] == '\0';
Buffer.push_back(HasNul ? Tok.data() : saver.save(Tok).data());
}
unsigned missingIndex;
unsigned missingCount;
auto Result = optTable.ParseArgs(Buffer, missingIndex, missingCount);
if (missingCount)
return make_error<JITLinkError>(Twine("COFF directive parsing failed: ") +
Result.getArgString(missingIndex) +
" missing argument");
LLVM_DEBUG({
for (auto *arg : Result.filtered(COFF_OPT_UNKNOWN))
dbgs() << "Unknown coff option argument: " << arg->getAsString(Result)
<< "\n";
});
return std::move(Result);
}
|
