blob: 1bd705c5518873dbae2a75a6073751218425ee8a [file] [log] [blame]
Alexei Starovoitove4c8c802015-01-24 17:51:26 +00001//===-- BPFISelDAGToDAG.cpp - A dag to dag inst selector for BPF ----------===//
2//
Chandler Carruth2946cd72019-01-19 08:50:56 +00003// 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
Alexei Starovoitove4c8c802015-01-24 17:51:26 +00006//
7//===----------------------------------------------------------------------===//
8//
9// This file defines a DAG pattern matching instruction selector for BPF,
10// converting from a legalized dag to a BPF dag.
11//
12//===----------------------------------------------------------------------===//
13
14#include "BPF.h"
15#include "BPFRegisterInfo.h"
16#include "BPFSubtarget.h"
17#include "BPFTargetMachine.h"
Yonghong Song5fbe01b2017-06-29 15:18:54 +000018#include "llvm/CodeGen/FunctionLoweringInfo.h"
Alexei Starovoitove4c8c802015-01-24 17:51:26 +000019#include "llvm/CodeGen/MachineConstantPool.h"
Alexei Starovoitove4c8c802015-01-24 17:51:26 +000020#include "llvm/CodeGen/MachineFrameInfo.h"
Benjamin Kramer799003b2015-03-23 19:32:43 +000021#include "llvm/CodeGen/MachineFunction.h"
Alexei Starovoitove4c8c802015-01-24 17:51:26 +000022#include "llvm/CodeGen/MachineInstrBuilder.h"
23#include "llvm/CodeGen/MachineRegisterInfo.h"
24#include "llvm/CodeGen/SelectionDAGISel.h"
Yonghong Songac2e2502017-06-16 15:41:16 +000025#include "llvm/IR/Constants.h"
Benjamin Kramer799003b2015-03-23 19:32:43 +000026#include "llvm/IR/IntrinsicInst.h"
Alexei Starovoitove4c8c802015-01-24 17:51:26 +000027#include "llvm/Support/Debug.h"
Yonghong Songac2e2502017-06-16 15:41:16 +000028#include "llvm/Support/Endian.h"
Alexei Starovoitove4c8c802015-01-24 17:51:26 +000029#include "llvm/Support/ErrorHandling.h"
30#include "llvm/Support/raw_ostream.h"
Benjamin Kramer799003b2015-03-23 19:32:43 +000031#include "llvm/Target/TargetMachine.h"
Yonghong Songac2e2502017-06-16 15:41:16 +000032
Alexei Starovoitove4c8c802015-01-24 17:51:26 +000033using namespace llvm;
34
35#define DEBUG_TYPE "bpf-isel"
36
37// Instruction Selector Implementation
38namespace {
39
40class BPFDAGToDAGISel : public SelectionDAGISel {
Yonghong Songb1a52bd2018-02-23 23:49:28 +000041
42 /// Subtarget - Keep a pointer to the BPFSubtarget around so that we can
43 /// make the right decision when generating code for different subtargets.
44 const BPFSubtarget *Subtarget;
45
Alexei Starovoitove4c8c802015-01-24 17:51:26 +000046public:
Yonghong Songb1a52bd2018-02-23 23:49:28 +000047 explicit BPFDAGToDAGISel(BPFTargetMachine &TM)
48 : SelectionDAGISel(TM), Subtarget(nullptr) {
Yonghong Song9af998e2017-10-24 21:36:33 +000049 curr_func_ = nullptr;
50 }
Alexei Starovoitove4c8c802015-01-24 17:51:26 +000051
Mehdi Amini117296c2016-10-01 02:56:57 +000052 StringRef getPassName() const override {
Alexei Starovoitove4c8c802015-01-24 17:51:26 +000053 return "BPF DAG->DAG Pattern Instruction Selection";
54 }
55
Yonghong Songb1a52bd2018-02-23 23:49:28 +000056 bool runOnMachineFunction(MachineFunction &MF) override {
57 // Reset the subtarget each time through.
58 Subtarget = &MF.getSubtarget<BPFSubtarget>();
59 return SelectionDAGISel::runOnMachineFunction(MF);
60 }
61
Yonghong Songac2e2502017-06-16 15:41:16 +000062 void PreprocessISelDAG() override;
63
Yonghong Song9ef85f02017-09-18 23:29:36 +000064 bool SelectInlineAsmMemoryOperand(const SDValue &Op, unsigned ConstraintCode,
65 std::vector<SDValue> &OutOps) override;
66
67
Alexei Starovoitove4c8c802015-01-24 17:51:26 +000068private:
69// Include the pieces autogenerated from the target description.
70#include "BPFGenDAGISel.inc"
71
Justin Bognerf076e792016-05-12 21:14:47 +000072 void Select(SDNode *N) override;
Alexei Starovoitove4c8c802015-01-24 17:51:26 +000073
74 // Complex Pattern for address selection.
75 bool SelectAddr(SDValue Addr, SDValue &Base, SDValue &Offset);
Alexei Starovoitov4e01a382015-10-06 04:00:53 +000076 bool SelectFIAddr(SDValue Addr, SDValue &Base, SDValue &Offset);
Yonghong Songac2e2502017-06-16 15:41:16 +000077
Yonghong Song5fbe01b2017-06-29 15:18:54 +000078 // Node preprocessing cases
Yonghong Song920df522018-02-15 17:06:45 +000079 void PreprocessLoad(SDNode *Node, SelectionDAG::allnodes_iterator &I);
Yonghong Song5fbe01b2017-06-29 15:18:54 +000080 void PreprocessCopyToReg(SDNode *Node);
Yonghong Song920df522018-02-15 17:06:45 +000081 void PreprocessTrunc(SDNode *Node, SelectionDAG::allnodes_iterator &I);
Yonghong Song5fbe01b2017-06-29 15:18:54 +000082
Yonghong Songac2e2502017-06-16 15:41:16 +000083 // Find constants from a constant structure
84 typedef std::vector<unsigned char> val_vec_type;
85 bool fillGenericConstant(const DataLayout &DL, const Constant *CV,
86 val_vec_type &Vals, uint64_t Offset);
87 bool fillConstantDataArray(const DataLayout &DL, const ConstantDataArray *CDA,
88 val_vec_type &Vals, int Offset);
89 bool fillConstantArray(const DataLayout &DL, const ConstantArray *CA,
90 val_vec_type &Vals, int Offset);
91 bool fillConstantStruct(const DataLayout &DL, const ConstantStruct *CS,
92 val_vec_type &Vals, int Offset);
93 bool getConstantFieldValue(const GlobalAddressSDNode *Node, uint64_t Offset,
94 uint64_t Size, unsigned char *ByteSeq);
Yonghong Song5fbe01b2017-06-29 15:18:54 +000095 bool checkLoadDef(unsigned DefReg, unsigned match_load_op);
Yonghong Songac2e2502017-06-16 15:41:16 +000096
97 // Mapping from ConstantStruct global value to corresponding byte-list values
98 std::map<const void *, val_vec_type> cs_vals_;
Yonghong Song5fbe01b2017-06-29 15:18:54 +000099 // Mapping from vreg to load memory opcode
100 std::map<unsigned, unsigned> load_to_vreg_;
Yonghong Songee68d8e2017-10-24 18:21:10 +0000101 // Current function
102 const Function *curr_func_;
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000103};
Yonghong Songac2e2502017-06-16 15:41:16 +0000104} // namespace
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000105
106// ComplexPattern used on BPF Load/Store instructions
107bool BPFDAGToDAGISel::SelectAddr(SDValue Addr, SDValue &Base, SDValue &Offset) {
108 // if Address is FI, get the TargetFrameIndex.
Alexei Starovoitov659ece92015-04-28 20:38:56 +0000109 SDLoc DL(Addr);
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000110 if (FrameIndexSDNode *FIN = dyn_cast<FrameIndexSDNode>(Addr)) {
Yonghong Songac2e2502017-06-16 15:41:16 +0000111 Base = CurDAG->getTargetFrameIndex(FIN->getIndex(), MVT::i64);
Alexei Starovoitov659ece92015-04-28 20:38:56 +0000112 Offset = CurDAG->getTargetConstant(0, DL, MVT::i64);
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000113 return true;
114 }
115
116 if (Addr.getOpcode() == ISD::TargetExternalSymbol ||
117 Addr.getOpcode() == ISD::TargetGlobalAddress)
118 return false;
119
Alexei Starovoitov4e01a382015-10-06 04:00:53 +0000120 // Addresses of the form Addr+const or Addr|const
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000121 if (CurDAG->isBaseWithConstantOffset(Addr)) {
122 ConstantSDNode *CN = dyn_cast<ConstantSDNode>(Addr.getOperand(1));
Alexei Starovoitov56db1452017-04-13 22:24:13 +0000123 if (isInt<16>(CN->getSExtValue())) {
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000124
125 // If the first operand is a FI, get the TargetFI Node
126 if (FrameIndexSDNode *FIN =
127 dyn_cast<FrameIndexSDNode>(Addr.getOperand(0)))
128 Base = CurDAG->getTargetFrameIndex(FIN->getIndex(), MVT::i64);
129 else
130 Base = Addr.getOperand(0);
131
Alexei Starovoitov659ece92015-04-28 20:38:56 +0000132 Offset = CurDAG->getTargetConstant(CN->getSExtValue(), DL, MVT::i64);
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000133 return true;
134 }
135 }
136
Yonghong Songac2e2502017-06-16 15:41:16 +0000137 Base = Addr;
Alexei Starovoitov659ece92015-04-28 20:38:56 +0000138 Offset = CurDAG->getTargetConstant(0, DL, MVT::i64);
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000139 return true;
140}
141
Alexei Starovoitov4e01a382015-10-06 04:00:53 +0000142// ComplexPattern used on BPF FI instruction
Yonghong Songac2e2502017-06-16 15:41:16 +0000143bool BPFDAGToDAGISel::SelectFIAddr(SDValue Addr, SDValue &Base,
144 SDValue &Offset) {
Alexei Starovoitov4e01a382015-10-06 04:00:53 +0000145 SDLoc DL(Addr);
146
147 if (!CurDAG->isBaseWithConstantOffset(Addr))
148 return false;
149
150 // Addresses of the form Addr+const or Addr|const
151 ConstantSDNode *CN = dyn_cast<ConstantSDNode>(Addr.getOperand(1));
Alexei Starovoitov56db1452017-04-13 22:24:13 +0000152 if (isInt<16>(CN->getSExtValue())) {
Alexei Starovoitov4e01a382015-10-06 04:00:53 +0000153
154 // If the first operand is a FI, get the TargetFI Node
Yonghong Songac2e2502017-06-16 15:41:16 +0000155 if (FrameIndexSDNode *FIN = dyn_cast<FrameIndexSDNode>(Addr.getOperand(0)))
Alexei Starovoitov4e01a382015-10-06 04:00:53 +0000156 Base = CurDAG->getTargetFrameIndex(FIN->getIndex(), MVT::i64);
157 else
158 return false;
159
160 Offset = CurDAG->getTargetConstant(CN->getSExtValue(), DL, MVT::i64);
161 return true;
162 }
163
164 return false;
165}
166
Yonghong Song9ef85f02017-09-18 23:29:36 +0000167bool BPFDAGToDAGISel::SelectInlineAsmMemoryOperand(
168 const SDValue &Op, unsigned ConstraintCode, std::vector<SDValue> &OutOps) {
169 SDValue Op0, Op1;
170 switch (ConstraintCode) {
171 default:
172 return true;
173 case InlineAsm::Constraint_m: // memory
174 if (!SelectAddr(Op, Op0, Op1))
175 return true;
176 break;
177 }
178
179 SDLoc DL(Op);
180 SDValue AluOp = CurDAG->getTargetConstant(ISD::ADD, DL, MVT::i32);;
181 OutOps.push_back(Op0);
182 OutOps.push_back(Op1);
183 OutOps.push_back(AluOp);
184 return false;
185}
186
Justin Bognerf076e792016-05-12 21:14:47 +0000187void BPFDAGToDAGISel::Select(SDNode *Node) {
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000188 unsigned Opcode = Node->getOpcode();
189
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000190 // If we have a custom node, we already have selected!
191 if (Node->isMachineOpcode()) {
Nicola Zaghend34e60c2018-05-14 12:53:11 +0000192 LLVM_DEBUG(dbgs() << "== "; Node->dump(CurDAG); dbgs() << '\n');
Justin Bognerf076e792016-05-12 21:14:47 +0000193 return;
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000194 }
195
196 // tablegen selection should be handled here.
197 switch (Opcode) {
Yonghong Songac2e2502017-06-16 15:41:16 +0000198 default:
199 break;
Alexei Starovoitov7e453bb2016-03-18 22:02:47 +0000200 case ISD::SDIV: {
201 DebugLoc Empty;
202 const DebugLoc &DL = Node->getDebugLoc();
203 if (DL != Empty)
204 errs() << "Error at line " << DL.getLine() << ": ";
205 else
206 errs() << "Error: ";
207 errs() << "Unsupport signed division for DAG: ";
Matthias Braun8c209aa2017-01-28 02:02:38 +0000208 Node->print(errs(), CurDAG);
Alexei Starovoitov7e453bb2016-03-18 22:02:47 +0000209 errs() << "Please convert to unsigned div/mod.\n";
210 break;
211 }
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000212 case ISD::INTRINSIC_W_CHAIN: {
213 unsigned IntNo = cast<ConstantSDNode>(Node->getOperand(1))->getZExtValue();
214 switch (IntNo) {
215 case Intrinsic::bpf_load_byte:
216 case Intrinsic::bpf_load_half:
217 case Intrinsic::bpf_load_word: {
218 SDLoc DL(Node);
219 SDValue Chain = Node->getOperand(0);
220 SDValue N1 = Node->getOperand(1);
221 SDValue Skb = Node->getOperand(2);
222 SDValue N3 = Node->getOperand(3);
223
224 SDValue R6Reg = CurDAG->getRegister(BPF::R6, MVT::i64);
225 Chain = CurDAG->getCopyToReg(Chain, DL, R6Reg, Skb, SDValue());
226 Node = CurDAG->UpdateNodeOperands(Node, Chain, N1, R6Reg, N3);
227 break;
228 }
229 }
230 break;
231 }
232
233 case ISD::FrameIndex: {
Benjamin Kramer619c4e52015-04-10 11:24:51 +0000234 int FI = cast<FrameIndexSDNode>(Node)->getIndex();
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000235 EVT VT = Node->getValueType(0);
236 SDValue TFI = CurDAG->getTargetFrameIndex(FI, VT);
237 unsigned Opc = BPF::MOV_rr;
Justin Bognerf076e792016-05-12 21:14:47 +0000238 if (Node->hasOneUse()) {
239 CurDAG->SelectNodeTo(Node, Opc, VT, TFI);
240 return;
241 }
242 ReplaceNode(Node, CurDAG->getMachineNode(Opc, SDLoc(Node), VT, TFI));
243 return;
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000244 }
245 }
246
247 // Select the default instruction
Justin Bognerf076e792016-05-12 21:14:47 +0000248 SelectCode(Node);
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000249}
250
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000251void BPFDAGToDAGISel::PreprocessLoad(SDNode *Node,
Yonghong Song920df522018-02-15 17:06:45 +0000252 SelectionDAG::allnodes_iterator &I) {
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000253 union {
254 uint8_t c[8];
255 uint16_t s;
256 uint32_t i;
257 uint64_t d;
258 } new_val; // hold up the constant values replacing loads.
259 bool to_replace = false;
260 SDLoc DL(Node);
261 const LoadSDNode *LD = cast<LoadSDNode>(Node);
262 uint64_t size = LD->getMemOperand()->getSize();
263
264 if (!size || size > 8 || (size & (size - 1)))
265 return;
266
267 SDNode *LDAddrNode = LD->getOperand(1).getNode();
268 // Match LDAddr against either global_addr or (global_addr + offset)
269 unsigned opcode = LDAddrNode->getOpcode();
270 if (opcode == ISD::ADD) {
271 SDValue OP1 = LDAddrNode->getOperand(0);
272 SDValue OP2 = LDAddrNode->getOperand(1);
273
274 // We want to find the pattern global_addr + offset
275 SDNode *OP1N = OP1.getNode();
276 if (OP1N->getOpcode() <= ISD::BUILTIN_OP_END || OP1N->getNumOperands() == 0)
277 return;
278
Nicola Zaghend34e60c2018-05-14 12:53:11 +0000279 LLVM_DEBUG(dbgs() << "Check candidate load: "; LD->dump(); dbgs() << '\n');
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000280
281 const GlobalAddressSDNode *GADN =
282 dyn_cast<GlobalAddressSDNode>(OP1N->getOperand(0).getNode());
283 const ConstantSDNode *CDN = dyn_cast<ConstantSDNode>(OP2.getNode());
284 if (GADN && CDN)
285 to_replace =
286 getConstantFieldValue(GADN, CDN->getZExtValue(), size, new_val.c);
287 } else if (LDAddrNode->getOpcode() > ISD::BUILTIN_OP_END &&
288 LDAddrNode->getNumOperands() > 0) {
Nicola Zaghend34e60c2018-05-14 12:53:11 +0000289 LLVM_DEBUG(dbgs() << "Check candidate load: "; LD->dump(); dbgs() << '\n');
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000290
291 SDValue OP1 = LDAddrNode->getOperand(0);
292 if (const GlobalAddressSDNode *GADN =
293 dyn_cast<GlobalAddressSDNode>(OP1.getNode()))
294 to_replace = getConstantFieldValue(GADN, 0, size, new_val.c);
295 }
296
297 if (!to_replace)
298 return;
299
300 // replacing the old with a new value
301 uint64_t val;
302 if (size == 1)
303 val = new_val.c[0];
304 else if (size == 2)
305 val = new_val.s;
306 else if (size == 4)
307 val = new_val.i;
308 else {
309 val = new_val.d;
310 }
311
Nicola Zaghend34e60c2018-05-14 12:53:11 +0000312 LLVM_DEBUG(dbgs() << "Replacing load of size " << size << " with constant "
313 << val << '\n');
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000314 SDValue NVal = CurDAG->getConstant(val, DL, MVT::i64);
315
316 // After replacement, the current node is dead, we need to
317 // go backward one step to make iterator still work
318 I--;
319 SDValue From[] = {SDValue(Node, 0), SDValue(Node, 1)};
320 SDValue To[] = {NVal, NVal};
321 CurDAG->ReplaceAllUsesOfValuesWith(From, To, 2);
322 I++;
323 // It is safe to delete node now
324 CurDAG->DeleteNode(Node);
325}
326
Yonghong Songac2e2502017-06-16 15:41:16 +0000327void BPFDAGToDAGISel::PreprocessISelDAG() {
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000328 // Iterate through all nodes, interested in the following cases:
329 //
330 // . loads from ConstantStruct or ConstantArray of constructs
331 // which can be turns into constant itself, with this we can
332 // avoid reading from read-only section at runtime.
333 //
334 // . reg truncating is often the result of 8/16/32bit->64bit or
335 // 8/16bit->32bit conversion. If the reg value is loaded with
336 // masked byte width, the AND operation can be removed since
337 // BPF LOAD already has zero extension.
338 //
339 // This also solved a correctness issue.
340 // In BPF socket-related program, e.g., __sk_buff->{data, data_end}
341 // are 32-bit registers, but later on, kernel verifier will rewrite
342 // it with 64-bit value. Therefore, truncating the value after the
343 // load will result in incorrect code.
Yonghong Song0f836d52017-10-24 17:29:03 +0000344
345 // clear the load_to_vreg_ map so that we have a clean start
346 // for this function.
Yonghong Songee68d8e2017-10-24 18:21:10 +0000347 if (!curr_func_) {
348 curr_func_ = FuncInfo->Fn;
349 } else if (curr_func_ != FuncInfo->Fn) {
350 load_to_vreg_.clear();
351 curr_func_ = FuncInfo->Fn;
352 }
Yonghong Song0f836d52017-10-24 17:29:03 +0000353
Yonghong Songac2e2502017-06-16 15:41:16 +0000354 for (SelectionDAG::allnodes_iterator I = CurDAG->allnodes_begin(),
355 E = CurDAG->allnodes_end();
356 I != E;) {
357 SDNode *Node = &*I++;
358 unsigned Opcode = Node->getOpcode();
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000359 if (Opcode == ISD::LOAD)
360 PreprocessLoad(Node, I);
361 else if (Opcode == ISD::CopyToReg)
362 PreprocessCopyToReg(Node);
363 else if (Opcode == ISD::AND)
364 PreprocessTrunc(Node, I);
Yonghong Songac2e2502017-06-16 15:41:16 +0000365 }
366}
367
368bool BPFDAGToDAGISel::getConstantFieldValue(const GlobalAddressSDNode *Node,
369 uint64_t Offset, uint64_t Size,
370 unsigned char *ByteSeq) {
371 const GlobalVariable *V = dyn_cast<GlobalVariable>(Node->getGlobal());
372
373 if (!V || !V->hasInitializer())
374 return false;
375
376 const Constant *Init = V->getInitializer();
377 const DataLayout &DL = CurDAG->getDataLayout();
378 val_vec_type TmpVal;
379
380 auto it = cs_vals_.find(static_cast<const void *>(Init));
381 if (it != cs_vals_.end()) {
382 TmpVal = it->second;
383 } else {
384 uint64_t total_size = 0;
385 if (const ConstantStruct *CS = dyn_cast<ConstantStruct>(Init))
386 total_size =
387 DL.getStructLayout(cast<StructType>(CS->getType()))->getSizeInBytes();
388 else if (const ConstantArray *CA = dyn_cast<ConstantArray>(Init))
389 total_size = DL.getTypeAllocSize(CA->getType()->getElementType()) *
390 CA->getNumOperands();
391 else
392 return false;
393
394 val_vec_type Vals(total_size, 0);
395 if (fillGenericConstant(DL, Init, Vals, 0) == false)
396 return false;
397 cs_vals_[static_cast<const void *>(Init)] = Vals;
398 TmpVal = std::move(Vals);
399 }
400
401 // test whether host endianness matches target
Yonghong Songa63178f2017-06-16 23:28:04 +0000402 union {
403 uint8_t c[2];
404 uint16_t s;
405 } test_buf;
Yonghong Songac2e2502017-06-16 15:41:16 +0000406 uint16_t test_val = 0x2345;
407 if (DL.isLittleEndian())
Yonghong Songa63178f2017-06-16 23:28:04 +0000408 support::endian::write16le(test_buf.c, test_val);
Yonghong Songac2e2502017-06-16 15:41:16 +0000409 else
Yonghong Songa63178f2017-06-16 23:28:04 +0000410 support::endian::write16be(test_buf.c, test_val);
Yonghong Songac2e2502017-06-16 15:41:16 +0000411
Yonghong Songa63178f2017-06-16 23:28:04 +0000412 bool endian_match = test_buf.s == test_val;
Yonghong Songac2e2502017-06-16 15:41:16 +0000413 for (uint64_t i = Offset, j = 0; i < Offset + Size; i++, j++)
414 ByteSeq[j] = endian_match ? TmpVal[i] : TmpVal[Offset + Size - 1 - j];
415
416 return true;
417}
418
419bool BPFDAGToDAGISel::fillGenericConstant(const DataLayout &DL,
420 const Constant *CV,
421 val_vec_type &Vals, uint64_t Offset) {
422 uint64_t Size = DL.getTypeAllocSize(CV->getType());
423
424 if (isa<ConstantAggregateZero>(CV) || isa<UndefValue>(CV))
425 return true; // already done
426
427 if (const ConstantInt *CI = dyn_cast<ConstantInt>(CV)) {
428 uint64_t val = CI->getZExtValue();
Nicola Zaghend34e60c2018-05-14 12:53:11 +0000429 LLVM_DEBUG(dbgs() << "Byte array at offset " << Offset << " with value "
430 << val << '\n');
Yonghong Songac2e2502017-06-16 15:41:16 +0000431
432 if (Size > 8 || (Size & (Size - 1)))
433 return false;
434
435 // Store based on target endian
436 for (uint64_t i = 0; i < Size; ++i) {
437 Vals[Offset + i] = DL.isLittleEndian()
438 ? ((val >> (i * 8)) & 0xFF)
439 : ((val >> ((Size - i - 1) * 8)) & 0xFF);
440 }
441 return true;
442 }
443
444 if (const ConstantDataArray *CDA = dyn_cast<ConstantDataArray>(CV))
445 return fillConstantDataArray(DL, CDA, Vals, Offset);
446
447 if (const ConstantArray *CA = dyn_cast<ConstantArray>(CV))
448 return fillConstantArray(DL, CA, Vals, Offset);
449
450 if (const ConstantStruct *CVS = dyn_cast<ConstantStruct>(CV))
451 return fillConstantStruct(DL, CVS, Vals, Offset);
452
453 return false;
454}
455
456bool BPFDAGToDAGISel::fillConstantDataArray(const DataLayout &DL,
457 const ConstantDataArray *CDA,
458 val_vec_type &Vals, int Offset) {
459 for (unsigned i = 0, e = CDA->getNumElements(); i != e; ++i) {
460 if (fillGenericConstant(DL, CDA->getElementAsConstant(i), Vals, Offset) ==
461 false)
462 return false;
463 Offset += DL.getTypeAllocSize(CDA->getElementAsConstant(i)->getType());
464 }
465
466 return true;
467}
468
469bool BPFDAGToDAGISel::fillConstantArray(const DataLayout &DL,
470 const ConstantArray *CA,
471 val_vec_type &Vals, int Offset) {
472 for (unsigned i = 0, e = CA->getNumOperands(); i != e; ++i) {
473 if (fillGenericConstant(DL, CA->getOperand(i), Vals, Offset) == false)
474 return false;
475 Offset += DL.getTypeAllocSize(CA->getOperand(i)->getType());
476 }
477
478 return true;
479}
480
481bool BPFDAGToDAGISel::fillConstantStruct(const DataLayout &DL,
482 const ConstantStruct *CS,
483 val_vec_type &Vals, int Offset) {
484 const StructLayout *Layout = DL.getStructLayout(CS->getType());
485 for (unsigned i = 0, e = CS->getNumOperands(); i != e; ++i) {
486 const Constant *Field = CS->getOperand(i);
487 uint64_t SizeSoFar = Layout->getElementOffset(i);
488 if (fillGenericConstant(DL, Field, Vals, Offset + SizeSoFar) == false)
489 return false;
490 }
491 return true;
492}
493
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000494void BPFDAGToDAGISel::PreprocessCopyToReg(SDNode *Node) {
495 const RegisterSDNode *RegN = dyn_cast<RegisterSDNode>(Node->getOperand(1));
496 if (!RegN || !TargetRegisterInfo::isVirtualRegister(RegN->getReg()))
497 return;
498
499 const LoadSDNode *LD = dyn_cast<LoadSDNode>(Node->getOperand(2));
500 if (!LD)
501 return;
502
503 // Assign a load value to a virtual register. record its load width
504 unsigned mem_load_op = 0;
505 switch (LD->getMemOperand()->getSize()) {
506 default:
507 return;
508 case 4:
509 mem_load_op = BPF::LDW;
510 break;
511 case 2:
512 mem_load_op = BPF::LDH;
513 break;
514 case 1:
515 mem_load_op = BPF::LDB;
516 break;
517 }
518
Nicola Zaghend34e60c2018-05-14 12:53:11 +0000519 LLVM_DEBUG(dbgs() << "Find Load Value to VReg "
520 << TargetRegisterInfo::virtReg2Index(RegN->getReg())
521 << '\n');
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000522 load_to_vreg_[RegN->getReg()] = mem_load_op;
523}
524
525void BPFDAGToDAGISel::PreprocessTrunc(SDNode *Node,
Yonghong Song920df522018-02-15 17:06:45 +0000526 SelectionDAG::allnodes_iterator &I) {
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000527 ConstantSDNode *MaskN = dyn_cast<ConstantSDNode>(Node->getOperand(1));
528 if (!MaskN)
529 return;
530
Yonghong Songb42c7c72018-01-16 07:27:19 +0000531 // The Reg operand should be a virtual register, which is defined
532 // outside the current basic block. DAG combiner has done a pretty
533 // good job in removing truncating inside a single basic block except
534 // when the Reg operand comes from bpf_load_[byte | half | word] for
535 // which the generic optimizer doesn't understand their results are
536 // zero extended.
537 SDValue BaseV = Node->getOperand(0);
538 if (BaseV.getOpcode() == ISD::INTRINSIC_W_CHAIN) {
539 unsigned IntNo = cast<ConstantSDNode>(BaseV->getOperand(1))->getZExtValue();
540 uint64_t MaskV = MaskN->getZExtValue();
541
542 if (!((IntNo == Intrinsic::bpf_load_byte && MaskV == 0xFF) ||
543 (IntNo == Intrinsic::bpf_load_half && MaskV == 0xFFFF) ||
544 (IntNo == Intrinsic::bpf_load_word && MaskV == 0xFFFFFFFF)))
545 return;
546
Nicola Zaghend34e60c2018-05-14 12:53:11 +0000547 LLVM_DEBUG(dbgs() << "Remove the redundant AND operation in: ";
548 Node->dump(); dbgs() << '\n');
Yonghong Songb42c7c72018-01-16 07:27:19 +0000549
550 I--;
551 CurDAG->ReplaceAllUsesWith(SDValue(Node, 0), BaseV);
552 I++;
553 CurDAG->DeleteNode(Node);
554
555 return;
556 }
557
558 // Multiple basic blocks case.
559 if (BaseV.getOpcode() != ISD::CopyFromReg)
560 return;
561
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000562 unsigned match_load_op = 0;
563 switch (MaskN->getZExtValue()) {
564 default:
565 return;
566 case 0xFFFFFFFF:
567 match_load_op = BPF::LDW;
568 break;
569 case 0xFFFF:
570 match_load_op = BPF::LDH;
571 break;
572 case 0xFF:
573 match_load_op = BPF::LDB;
574 break;
575 }
576
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000577 const RegisterSDNode *RegN =
578 dyn_cast<RegisterSDNode>(BaseV.getNode()->getOperand(1));
579 if (!RegN || !TargetRegisterInfo::isVirtualRegister(RegN->getReg()))
580 return;
581 unsigned AndOpReg = RegN->getReg();
Nicola Zaghend34e60c2018-05-14 12:53:11 +0000582 LLVM_DEBUG(dbgs() << "Examine " << printReg(AndOpReg) << '\n');
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000583
584 // Examine the PHI insns in the MachineBasicBlock to found out the
585 // definitions of this virtual register. At this stage (DAG2DAG
586 // transformation), only PHI machine insns are available in the machine basic
587 // block.
588 MachineBasicBlock *MBB = FuncInfo->MBB;
589 MachineInstr *MII = nullptr;
590 for (auto &MI : *MBB) {
591 for (unsigned i = 0; i < MI.getNumOperands(); ++i) {
592 const MachineOperand &MOP = MI.getOperand(i);
593 if (!MOP.isReg() || !MOP.isDef())
594 continue;
595 unsigned Reg = MOP.getReg();
596 if (TargetRegisterInfo::isVirtualRegister(Reg) && Reg == AndOpReg) {
597 MII = &MI;
598 break;
599 }
600 }
601 }
602
603 if (MII == nullptr) {
604 // No phi definition in this block.
605 if (!checkLoadDef(AndOpReg, match_load_op))
606 return;
607 } else {
608 // The PHI node looks like:
Francis Visoiu Mistriha8a83d12017-12-07 10:40:31 +0000609 // %2 = PHI %0, <%bb.1>, %1, <%bb.3>
Francis Visoiu Mistrih25528d62017-12-04 17:18:51 +0000610 // Trace each incoming definition, e.g., (%0, %bb.1) and (%1, %bb.3)
611 // The AND operation can be removed if both %0 in %bb.1 and %1 in
Fangrui Song956ee792018-03-30 22:22:31 +0000612 // %bb.3 are defined with a load matching the MaskN.
Nicola Zaghend34e60c2018-05-14 12:53:11 +0000613 LLVM_DEBUG(dbgs() << "Check PHI Insn: "; MII->dump(); dbgs() << '\n');
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000614 unsigned PrevReg = -1;
615 for (unsigned i = 0; i < MII->getNumOperands(); ++i) {
616 const MachineOperand &MOP = MII->getOperand(i);
617 if (MOP.isReg()) {
618 if (MOP.isDef())
619 continue;
620 PrevReg = MOP.getReg();
621 if (!TargetRegisterInfo::isVirtualRegister(PrevReg))
622 return;
623 if (!checkLoadDef(PrevReg, match_load_op))
624 return;
625 }
626 }
627 }
628
Nicola Zaghend34e60c2018-05-14 12:53:11 +0000629 LLVM_DEBUG(dbgs() << "Remove the redundant AND operation in: "; Node->dump();
630 dbgs() << '\n');
Yonghong Song5fbe01b2017-06-29 15:18:54 +0000631
632 I--;
633 CurDAG->ReplaceAllUsesWith(SDValue(Node, 0), BaseV);
634 I++;
635 CurDAG->DeleteNode(Node);
636}
637
638bool BPFDAGToDAGISel::checkLoadDef(unsigned DefReg, unsigned match_load_op) {
639 auto it = load_to_vreg_.find(DefReg);
640 if (it == load_to_vreg_.end())
641 return false; // The definition of register is not exported yet.
642
643 return it->second == match_load_op;
644}
645
Alexei Starovoitove4c8c802015-01-24 17:51:26 +0000646FunctionPass *llvm::createBPFISelDag(BPFTargetMachine &TM) {
647 return new BPFDAGToDAGISel(TM);
648}