llvm/lib/ExecutionEngine/JITLink/COFFDirectiveParser.cpp

   1 //===-- COFFDirectiveParser.cpp - JITLink coff directive parser --*- C++ -*===//
   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 // MSVC COFF directive parser
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #include "COFFDirectiveParser.h"
  14
  15 #include <array>
  16
  17 using namespace llvm;
  18 using namespace jitlink;
  19
  20 #define DEBUG_TYPE "jitlink"
  21
  22 // Create prefix string literals used in Options.td
  23 #define PREFIX(NAME, VALUE)                                                    \
  24   static constexpr StringLiteral NAME##_init[] = VALUE;                        \
  25   static constexpr ArrayRef<StringLiteral> NAME(NAME##_init,                   \
  26                                                 std::size(NAME##_init) - 1);
  27 #include "COFFOptions.inc"
  28 #undef PREFIX
  29
  30 static constexpr const StringLiteral PrefixTable_init[] =
  31 #define PREFIX_UNION(VALUES) VALUES
  32 #include "COFFOptions.inc"
  33 #undef PREFIX_UNION
  34     ;
  35 static constexpr const ArrayRef<StringLiteral>
  36     PrefixTable(PrefixTable_init, std::size(PrefixTable_init) - 1);
  37
  38 // Create table mapping all options defined in COFFOptions.td
  39 using namespace llvm::opt;
  40 static constexpr opt::OptTable::Info infoTable[] = {
  41 #define OPTION(...)                                                            \
  42   LLVM_CONSTRUCT_OPT_INFO_WITH_ID_PREFIX(COFF_OPT_, __VA_ARGS__),
  43 #include "COFFOptions.inc"
  44 #undef OPTION
  45 };
  46
  47 class COFFOptTable : public opt::PrecomputedOptTable {
  48 public:
  49   COFFOptTable() : PrecomputedOptTable(infoTable, PrefixTable, true) {}
  50 };
  51
  52 static COFFOptTable optTable;
  53
  54 Expected<opt::InputArgList> COFFDirectiveParser::parse(StringRef Str) {
  55   SmallVector<StringRef, 16> Tokens;
  56   SmallVector<const char *, 16> Buffer;
  57   cl::TokenizeWindowsCommandLineNoCopy(Str, saver, Tokens);
  58   for (StringRef Tok : Tokens) {
  59     bool HasNul = Tok.end() != Str.end() && Tok.data()[Tok.size()] == '\0';
  60     Buffer.push_back(HasNul ? Tok.data() : saver.save(Tok).data());
  61   }
  62
  63   unsigned missingIndex;
  64   unsigned missingCount;
  65
  66   auto Result = optTable.ParseArgs(Buffer, missingIndex, missingCount);
  67
  68   if (missingCount)
  69     return make_error<JITLinkError>(Twine("COFF directive parsing failed: ") +
  70                                     Result.getArgString(missingIndex) +
  71                                     " missing argument");
  72   LLVM_DEBUG({
  73     for (auto *arg : Result.filtered(COFF_OPT_UNKNOWN))
  74       dbgs() << "Unknown coff option argument: " << arg->getAsString(Result)
  75              << "\n";
  76   });
  77   return std::move(Result);
  78 }