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
19using namespace clang;
20using namespace clang::targets;
21
22static 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
28void 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
65static constexpr llvm::StringLiteral ValidCPUNames[] = {"generic", "v1", "v2",
66                                                        "v3", "v4", "probe"};
67
68bool BPFTargetInfo::isValidCPUName(StringRef Name) const {
69  return llvm::is_contained(ValidCPUNames, Name);
70}
71
72void BPFTargetInfo::fillValidCPUList(SmallVectorImpl<StringRef> &Values) const {
73  Values.append(std::begin(ValidCPUNames), std::end(ValidCPUNames));
74}
75
76ArrayRef<Builtin::Info> BPFTargetInfo::getTargetBuiltins() const {
77  return llvm::ArrayRef(BuiltinInfo,
78                        clang::BPF::LastTSBuiltin - Builtin::FirstTSBuiltin);
79}
80
81bool 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}
91