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