clang/lib/Basic/Targets/BPF.cpp

   1 //===--- BPF.cpp - Implement BPF target feature support -------------------===//
   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 // This file implements BPF TargetInfo objects.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #include "BPF.h"
  14 #include "Targets.h"
  15 #include "clang/Basic/MacroBuilder.h"
  16 #include "clang/Basic/TargetBuiltins.h"
  17 #include "llvm/ADT/StringRef.h"
  18
  19 using namespace clang;
  20 using namespace clang::targets;
  21
  22 static constexpr Builtin::Info BuiltinInfo[] = {
  23 #define BUILTIN(ID, TYPE, ATTRS)                                               \
  24   {#ID, TYPE, ATTRS, nullptr, HeaderDesc::NO_HEADER, ALL_LANGUAGES},
  25 #include "clang/Basic/BuiltinsBPF.def"
  26 };
  27
  28 void BPFTargetInfo::getTargetDefines(const LangOptions &Opts,
  29                                      MacroBuilder &Builder) const {
  30   Builder.defineMacro("__bpf__");
  31   Builder.defineMacro("__BPF__");
  32
  33   std::string CPU = getTargetOpts().CPU;
  34   if (CPU == "probe") {
  35     Builder.defineMacro("__BPF_CPU_VERSION__", "0");
  36     return;
  37   }
  38   if (CPU.empty() || CPU == "generic" || CPU == "v1") {
  39     Builder.defineMacro("__BPF_CPU_VERSION__", "1");
  40     return;
  41   }
  42
  43   std::string CpuVerNumStr = CPU.substr(1);
  44   Builder.defineMacro("__BPF_CPU_VERSION__", CpuVerNumStr);
  45
  46   int CpuVerNum = std::stoi(CpuVerNumStr);
  47   if (CpuVerNum >= 2)
  48     Builder.defineMacro("__BPF_FEATURE_JMP_EXT");
  49
  50   if (CpuVerNum >= 3) {
  51     Builder.defineMacro("__BPF_FEATURE_JMP32");
  52     Builder.defineMacro("__BPF_FEATURE_ALU32");
  53   }
  54
  55   if (CpuVerNum >= 4) {
  56     Builder.defineMacro("__BPF_FEATURE_LDSX");
  57     Builder.defineMacro("__BPF_FEATURE_MOVSX");
  58     Builder.defineMacro("__BPF_FEATURE_BSWAP");
  59     Builder.defineMacro("__BPF_FEATURE_SDIV_SMOD");
  60     Builder.defineMacro("__BPF_FEATURE_GOTOL");
  61     Builder.defineMacro("__BPF_FEATURE_ST");
  62   }
  63 }
  64
  65 static constexpr llvm::StringLiteral ValidCPUNames[] = {"generic", "v1", "v2",
  66                                                         "v3", "v4", "probe"};
  67
  68 bool BPFTargetInfo::isValidCPUName(StringRef Name) const {
  69   return llvm::is_contained(ValidCPUNames, Name);
  70 }
  71
  72 void BPFTargetInfo::fillValidCPUList(SmallVectorImpl<StringRef> &Values) const {
  73   Values.append(std::begin(ValidCPUNames), std::end(ValidCPUNames));
  74 }
  75
  76 ArrayRef<Builtin::Info> BPFTargetInfo::getTargetBuiltins() const {
  77   return llvm::ArrayRef(BuiltinInfo,
  78                         clang::BPF::LastTSBuiltin - Builtin::FirstTSBuiltin);
  79 }
  80
  81 bool BPFTargetInfo::handleTargetFeatures(std::vector<std::string> &Features,
  82                                          DiagnosticsEngine &Diags) {
  83   for (const auto &Feature : Features) {
  84     if (Feature == "+alu32") {
  85       HasAlu32 = true;
  86     }
  87   }
  88
  89   return true;
  90 }