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 "clang/Basic/MacroBuilder.h"
15#include "clang/Basic/TargetBuiltins.h"
16#include "llvm/ADT/StringRef.h"
17
18using namespace clang;
19using namespace clang::targets;
20
21static constexpr int NumBuiltins =
22 clang::BPF::LastTSBuiltin - Builtin::FirstTSBuiltin;
23
24#define GET_BUILTIN_STR_TABLE
25#include "clang/Basic/BuiltinsBPF.inc"
26#undef GET_BUILTIN_STR_TABLE
27
28static constexpr Builtin::Info BuiltinInfos[] = {
29#define GET_BUILTIN_INFOS
30#include "clang/Basic/BuiltinsBPF.inc"
31#undef GET_BUILTIN_INFOS
32};
33static_assert(std::size(BuiltinInfos) == NumBuiltins);
34
35void BPFTargetInfo::getTargetDefines(const LangOptions &Opts,
36 MacroBuilder &Builder) const {
37 Builder.defineMacro(Name: "__bpf__");
38 Builder.defineMacro(Name: "__BPF__");
39
40 std::string CPU = getTargetOpts().CPU;
41 if (CPU == "probe") {
42 Builder.defineMacro(Name: "__BPF_CPU_VERSION__", Value: "0");
43 return;
44 }
45
46 Builder.defineMacro(Name: "__BPF_FEATURE_ADDR_SPACE_CAST");
47 Builder.defineMacro(Name: "__BPF_FEATURE_MAY_GOTO");
48 Builder.defineMacro(Name: "__BPF_FEATURE_ATOMIC_MEM_ORDERING");
49 Builder.defineMacro(Name: "__BPF_FEATURE_STACK_ARGUMENT");
50
51 if (CPU.empty())
52 CPU = "v3";
53
54 if (CPU == "generic" || CPU == "v1") {
55 Builder.defineMacro(Name: "__BPF_CPU_VERSION__", Value: "1");
56 return;
57 }
58
59 std::string CpuVerNumStr = CPU.substr(pos: 1);
60 Builder.defineMacro(Name: "__BPF_CPU_VERSION__", Value: CpuVerNumStr);
61
62 int CpuVerNum = std::stoi(str: CpuVerNumStr);
63 if (CpuVerNum >= 2)
64 Builder.defineMacro(Name: "__BPF_FEATURE_JMP_EXT");
65
66 if (CpuVerNum >= 3) {
67 Builder.defineMacro(Name: "__BPF_FEATURE_JMP32");
68 Builder.defineMacro(Name: "__BPF_FEATURE_ALU32");
69 }
70
71 if (CpuVerNum >= 4) {
72 Builder.defineMacro(Name: "__BPF_FEATURE_LDSX");
73 Builder.defineMacro(Name: "__BPF_FEATURE_MOVSX");
74 Builder.defineMacro(Name: "__BPF_FEATURE_BSWAP");
75 Builder.defineMacro(Name: "__BPF_FEATURE_SDIV_SMOD");
76 Builder.defineMacro(Name: "__BPF_FEATURE_GOTOL");
77 Builder.defineMacro(Name: "__BPF_FEATURE_ST");
78 Builder.defineMacro(Name: "__BPF_FEATURE_LOAD_ACQ_STORE_REL");
79 Builder.defineMacro(Name: "__BPF_FEATURE_GOTOX");
80 }
81}
82
83static constexpr llvm::StringLiteral ValidCPUNames[] = {"generic", "v1", "v2",
84 "v3", "v4", "probe"};
85
86bool BPFTargetInfo::isValidCPUName(StringRef Name) const {
87 return llvm::is_contained(Range: ValidCPUNames, Element: Name);
88}
89
90void BPFTargetInfo::fillValidCPUList(SmallVectorImpl<StringRef> &Values) const {
91 Values.append(in_start: std::begin(arr: ValidCPUNames), in_end: std::end(arr: ValidCPUNames));
92}
93
94llvm::SmallVector<Builtin::InfosShard>
95BPFTargetInfo::getTargetBuiltins() const {
96 return {{.Strings: &BuiltinStrings, .Infos: BuiltinInfos}};
97}
98
99bool BPFTargetInfo::handleTargetFeatures(std::vector<std::string> &Features,
100 DiagnosticsEngine &Diags) {
101 for (const auto &Feature : Features) {
102 if (Feature == "+alu32") {
103 HasAlu32 = true;
104 }
105 }
106
107 return true;
108}
109