xref: /llvm-project/clang/lib/Basic/Targets/BPF.cpp (revision ca79ff07d8ae7a0c2531bfdb1cb623e25e5bd486)
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.inc"
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 
39   Builder.defineMacro("__BPF_FEATURE_ADDR_SPACE_CAST");
40   Builder.defineMacro("__BPF_FEATURE_MAY_GOTO");
41   Builder.defineMacro("__BPF_FEATURE_ATOMIC_MEM_ORDERING");
42 
43   if (CPU.empty())
44     CPU = "v3";
45 
46   if (CPU == "generic" || CPU == "v1") {
47     Builder.defineMacro("__BPF_CPU_VERSION__", "1");
48     return;
49   }
50 
51   std::string CpuVerNumStr = CPU.substr(1);
52   Builder.defineMacro("__BPF_CPU_VERSION__", CpuVerNumStr);
53 
54   int CpuVerNum = std::stoi(CpuVerNumStr);
55   if (CpuVerNum >= 2)
56     Builder.defineMacro("__BPF_FEATURE_JMP_EXT");
57 
58   if (CpuVerNum >= 3) {
59     Builder.defineMacro("__BPF_FEATURE_JMP32");
60     Builder.defineMacro("__BPF_FEATURE_ALU32");
61   }
62 
63   if (CpuVerNum >= 4) {
64     Builder.defineMacro("__BPF_FEATURE_LDSX");
65     Builder.defineMacro("__BPF_FEATURE_MOVSX");
66     Builder.defineMacro("__BPF_FEATURE_BSWAP");
67     Builder.defineMacro("__BPF_FEATURE_SDIV_SMOD");
68     Builder.defineMacro("__BPF_FEATURE_GOTOL");
69     Builder.defineMacro("__BPF_FEATURE_ST");
70   }
71 }
72 
73 static constexpr llvm::StringLiteral ValidCPUNames[] = {"generic", "v1", "v2",
74                                                         "v3", "v4", "probe"};
75 
76 bool BPFTargetInfo::isValidCPUName(StringRef Name) const {
77   return llvm::is_contained(ValidCPUNames, Name);
78 }
79 
80 void BPFTargetInfo::fillValidCPUList(SmallVectorImpl<StringRef> &Values) const {
81   Values.append(std::begin(ValidCPUNames), std::end(ValidCPUNames));
82 }
83 
84 ArrayRef<Builtin::Info> BPFTargetInfo::getTargetBuiltins() const {
85   return llvm::ArrayRef(BuiltinInfo,
86                         clang::BPF::LastTSBuiltin - Builtin::FirstTSBuiltin);
87 }
88 
89 bool BPFTargetInfo::handleTargetFeatures(std::vector<std::string> &Features,
90                                          DiagnosticsEngine &Diags) {
91   for (const auto &Feature : Features) {
92     if (Feature == "+alu32") {
93       HasAlu32 = true;
94     }
95   }
96 
97   return true;
98 }
99