lib/Target/BPF/BPFISelLowering.cpp

   1 //===-- BPFISelLowering.cpp - BPF DAG Lowering Implementation  ------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file defines the interfaces that BPF uses to lower LLVM code into a
  11 // selection DAG.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "BPFISelLowering.h"
  16 #include "BPF.h"
  17 #include "BPFTargetMachine.h"
  18 #include "BPFSubtarget.h"
  19 #include "llvm/CodeGen/CallingConvLower.h"
  20 #include "llvm/CodeGen/MachineFrameInfo.h"
  21 #include "llvm/CodeGen/MachineFunction.h"
  22 #include "llvm/CodeGen/MachineInstrBuilder.h"
  23 #include "llvm/CodeGen/MachineRegisterInfo.h"
  24 #include "llvm/CodeGen/SelectionDAGISel.h"
  25 #include "llvm/CodeGen/TargetLoweringObjectFileImpl.h"
  26 #include "llvm/CodeGen/ValueTypes.h"
  27 #include "llvm/Support/CommandLine.h"
  28 #include "llvm/Support/Debug.h"
  29 #include "llvm/Support/ErrorHandling.h"
  30 #include "llvm/Support/raw_ostream.h"
  31 #include "llvm/IR/DiagnosticInfo.h"
  32 #include "llvm/IR/DiagnosticPrinter.h"
  33 using namespace llvm;
  34
  35 #define DEBUG_TYPE "bpf-lower"
  36
  37 namespace {
  38
  39 // Diagnostic information for unimplemented or unsupported feature reporting.
  40 class DiagnosticInfoUnsupported : public DiagnosticInfo {
  41 private:
  42   // Debug location where this diagnostic is triggered.
  43   DebugLoc DLoc;
  44   const Twine &Description;
  45   const Function &Fn;
  46   SDValue Value;
  47
  48   static int KindID;
  49
  50   static int getKindID() {
  51     if (KindID == 0)
  52       KindID = llvm::getNextAvailablePluginDiagnosticKind();
  53     return KindID;
  54   }
  55
  56 public:
  57   DiagnosticInfoUnsupported(SDLoc DLoc, const Function &Fn, const Twine &Desc,
  58                             SDValue Value)
  59       : DiagnosticInfo(getKindID(), DS_Error), DLoc(DLoc.getDebugLoc()),
  60         Description(Desc), Fn(Fn), Value(Value) {}
  61
  62   void print(DiagnosticPrinter &DP) const override {
  63     std::string Str;
  64     raw_string_ostream OS(Str);
  65
  66     if (DLoc) {
  67       auto DIL = DLoc.get();
  68       StringRef Filename = DIL->getFilename();
  69       unsigned Line = DIL->getLine();
  70       unsigned Column = DIL->getColumn();
  71       OS << Filename << ':' << Line << ':' << Column << ' ';
  72     }
  73
  74     OS << "in function " << Fn.getName() << ' ' << *Fn.getFunctionType() << '\n'
  75        << Description;
  76     if (Value)
  77       Value->print(OS);
  78     OS << '\n';
  79     OS.flush();
  80     DP << Str;
  81   }
  82
  83   static bool classof(const DiagnosticInfo *DI) {
  84     return DI->getKind() == getKindID();
  85   }
  86 };
  87
  88 int DiagnosticInfoUnsupported::KindID = 0;
  89 }
  90
  91 BPFTargetLowering::BPFTargetLowering(const TargetMachine &TM,
  92                                      const BPFSubtarget &STI)
  93     : TargetLowering(TM) {
  94
  95   // Set up the register classes.
  96   addRegisterClass(MVT::i64, &BPF::GPRRegClass);
  97
  98   // Compute derived properties from the register classes
  99   computeRegisterProperties(STI.getRegisterInfo());
 100
 101   setStackPointerRegisterToSaveRestore(BPF::R11);
 102
 103   setOperationAction(ISD::BR_CC, MVT::i64, Custom);
 104   setOperationAction(ISD::BR_JT, MVT::Other, Expand);
 105   setOperationAction(ISD::BRCOND, MVT::Other, Expand);
 106   setOperationAction(ISD::SETCC, MVT::i64, Expand);
 107   setOperationAction(ISD::SELECT, MVT::i64, Expand);
 108   setOperationAction(ISD::SELECT_CC, MVT::i64, Custom);
 109
 110   setOperationAction(ISD::GlobalAddress, MVT::i64, Custom);
 111
 112   setOperationAction(ISD::DYNAMIC_STACKALLOC, MVT::i64, Custom);
 113   setOperationAction(ISD::STACKSAVE, MVT::Other, Expand);
 114   setOperationAction(ISD::STACKRESTORE, MVT::Other, Expand);
 115
 116   setOperationAction(ISD::SDIVREM, MVT::i64, Expand);
 117   setOperationAction(ISD::UDIVREM, MVT::i64, Expand);
 118   setOperationAction(ISD::SREM, MVT::i64, Expand);
 119   setOperationAction(ISD::UREM, MVT::i64, Expand);
 120
 121   setOperationAction(ISD::MULHU, MVT::i64, Expand);
 122   setOperationAction(ISD::MULHS, MVT::i64, Expand);
 123   setOperationAction(ISD::UMUL_LOHI, MVT::i64, Expand);
 124   setOperationAction(ISD::SMUL_LOHI, MVT::i64, Expand);
 125
 126   setOperationAction(ISD::ADDC, MVT::i64, Expand);
 127   setOperationAction(ISD::ADDE, MVT::i64, Expand);
 128   setOperationAction(ISD::SUBC, MVT::i64, Expand);
 129   setOperationAction(ISD::SUBE, MVT::i64, Expand);
 130
 131   // no UNDEF allowed
 132   setOperationAction(ISD::UNDEF, MVT::i64, Expand);
 133
 134   setOperationAction(ISD::ROTR, MVT::i64, Expand);
 135   setOperationAction(ISD::ROTL, MVT::i64, Expand);
 136   setOperationAction(ISD::SHL_PARTS, MVT::i64, Expand);
 137   setOperationAction(ISD::SRL_PARTS, MVT::i64, Expand);
 138   setOperationAction(ISD::SRA_PARTS, MVT::i64, Expand);
 139
 140   setOperationAction(ISD::CTTZ, MVT::i64, Custom);
 141   setOperationAction(ISD::CTLZ, MVT::i64, Custom);
 142   setOperationAction(ISD::CTTZ_ZERO_UNDEF, MVT::i64, Custom);
 143   setOperationAction(ISD::CTLZ_ZERO_UNDEF, MVT::i64, Custom);
 144   setOperationAction(ISD::CTPOP, MVT::i64, Expand);
 145
 146   setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i1, Expand);
 147   setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i8, Expand);
 148   setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i16, Expand);
 149   setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i32, Expand);
 150
 151   // Extended load operations for i1 types must be promoted
 152   for (MVT VT : MVT::integer_valuetypes()) {
 153     setLoadExtAction(ISD::EXTLOAD, VT, MVT::i1, Promote);
 154     setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::i1, Promote);
 155     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i1, Promote);
 156
 157     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i8, Expand);
 158     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i16, Expand);
 159     setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i32, Expand);
 160   }
 161
 162   setBooleanContents(ZeroOrOneBooleanContent);
 163
 164   // Function alignments (log2)
 165   setMinFunctionAlignment(3);
 166   setPrefFunctionAlignment(3);
 167
 168   // inline memcpy() for kernel to see explicit copy
 169   MaxStoresPerMemset = MaxStoresPerMemsetOptSize = 128;
 170   MaxStoresPerMemcpy = MaxStoresPerMemcpyOptSize = 128;
 171   MaxStoresPerMemmove = MaxStoresPerMemmoveOptSize = 128;
 172 }
 173
 174 SDValue BPFTargetLowering::LowerOperation(SDValue Op, SelectionDAG &DAG) const {
 175   switch (Op.getOpcode()) {
 176   case ISD::BR_CC:
 177     return LowerBR_CC(Op, DAG);
 178   case ISD::GlobalAddress:
 179     return LowerGlobalAddress(Op, DAG);
 180   case ISD::SELECT_CC:
 181     return LowerSELECT_CC(Op, DAG);
 182   default:
 183     llvm_unreachable("unimplemented operand");
 184   }
 185 }
 186
 187 // Calling Convention Implementation
 188 #include "BPFGenCallingConv.inc"
 189
 190 SDValue BPFTargetLowering::LowerFormalArguments(
 191     SDValue Chain, CallingConv::ID CallConv, bool IsVarArg,
 192     const SmallVectorImpl<ISD::InputArg> &Ins, SDLoc DL, SelectionDAG &DAG,
 193     SmallVectorImpl<SDValue> &InVals) const {
 194   switch (CallConv) {
 195   default:
 196     llvm_unreachable("Unsupported calling convention");
 197   case CallingConv::C:
 198   case CallingConv::Fast:
 199     break;
 200   }
 201
 202   MachineFunction &MF = DAG.getMachineFunction();
 203   MachineRegisterInfo &RegInfo = MF.getRegInfo();
 204
 205   // Assign locations to all of the incoming arguments.
 206   SmallVector<CCValAssign, 16> ArgLocs;
 207   CCState CCInfo(CallConv, IsVarArg, MF, ArgLocs, *DAG.getContext());
 208   CCInfo.AnalyzeFormalArguments(Ins, CC_BPF64);
 209
 210   for (auto &VA : ArgLocs) {
 211     if (VA.isRegLoc()) {
 212       // Arguments passed in registers
 213       EVT RegVT = VA.getLocVT();
 214       switch (RegVT.getSimpleVT().SimpleTy) {
 215       default: {
 216         errs() << "LowerFormalArguments Unhandled argument type: "
 217                << RegVT.getSimpleVT().SimpleTy << '\n';
 218         llvm_unreachable(0);
 219       }
 220       case MVT::i64:
 221         unsigned VReg = RegInfo.createVirtualRegister(&BPF::GPRRegClass);
 222         RegInfo.addLiveIn(VA.getLocReg(), VReg);
 223         SDValue ArgValue = DAG.getCopyFromReg(Chain, DL, VReg, RegVT);
 224
 225         // If this is an 8/16/32-bit value, it is really passed promoted to 64
 226         // bits. Insert an assert[sz]ext to capture this, then truncate to the
 227         // right size.
 228         if (VA.getLocInfo() == CCValAssign::SExt)
 229           ArgValue = DAG.getNode(ISD::AssertSext, DL, RegVT, ArgValue,
 230                                  DAG.getValueType(VA.getValVT()));
 231         else if (VA.getLocInfo() == CCValAssign::ZExt)
 232           ArgValue = DAG.getNode(ISD::AssertZext, DL, RegVT, ArgValue,
 233                                  DAG.getValueType(VA.getValVT()));
 234
 235         if (VA.getLocInfo() != CCValAssign::Full)
 236           ArgValue = DAG.getNode(ISD::TRUNCATE, DL, VA.getValVT(), ArgValue);
 237
 238         InVals.push_back(ArgValue);
 239       }
 240     } else {
 241       DiagnosticInfoUnsupported Err(DL, *MF.getFunction(),
 242                                     "defined with too many args", SDValue());
 243       DAG.getContext()->diagnose(Err);
 244     }
 245   }
 246
 247   if (IsVarArg || MF.getFunction()->hasStructRetAttr()) {
 248     DiagnosticInfoUnsupported Err(
 249         DL, *MF.getFunction(),
 250         "functions with VarArgs or StructRet are not supported", SDValue());
 251     DAG.getContext()->diagnose(Err);
 252   }
 253
 254   return Chain;
 255 }
 256
 257 SDValue BPFTargetLowering::LowerCall(TargetLowering::CallLoweringInfo &CLI,
 258                                      SmallVectorImpl<SDValue> &InVals) const {
 259   SelectionDAG &DAG = CLI.DAG;
 260   auto &Outs = CLI.Outs;
 261   auto &OutVals = CLI.OutVals;
 262   auto &Ins = CLI.Ins;
 263   SDValue Chain = CLI.Chain;
 264   SDValue Callee = CLI.Callee;
 265   bool &IsTailCall = CLI.IsTailCall;
 266   CallingConv::ID CallConv = CLI.CallConv;
 267   bool IsVarArg = CLI.IsVarArg;
 268   MachineFunction &MF = DAG.getMachineFunction();
 269
 270   // BPF target does not support tail call optimization.
 271   IsTailCall = false;
 272
 273   switch (CallConv) {
 274   default:
 275     report_fatal_error("Unsupported calling convention");
 276   case CallingConv::Fast:
 277   case CallingConv::C:
 278     break;
 279   }
 280
 281   // Analyze operands of the call, assigning locations to each operand.
 282   SmallVector<CCValAssign, 16> ArgLocs;
 283   CCState CCInfo(CallConv, IsVarArg, MF, ArgLocs, *DAG.getContext());
 284
 285   CCInfo.AnalyzeCallOperands(Outs, CC_BPF64);
 286
 287   unsigned NumBytes = CCInfo.getNextStackOffset();
 288
 289   if (Outs.size() >= 6) {
 290     DiagnosticInfoUnsupported Err(CLI.DL, *MF.getFunction(),
 291                                   "too many args to ", Callee);
 292     DAG.getContext()->diagnose(Err);
 293   }
 294
 295   for (auto &Arg : Outs) {
 296     ISD::ArgFlagsTy Flags = Arg.Flags;
 297     if (!Flags.isByVal())
 298       continue;
 299
 300     DiagnosticInfoUnsupported Err(CLI.DL, *MF.getFunction(),
 301                                   "pass by value not supported ", Callee);
 302     DAG.getContext()->diagnose(Err);
 303   }
 304
 305   auto PtrVT = getPointerTy(MF.getDataLayout());
 306   Chain = DAG.getCALLSEQ_START(
 307       Chain, DAG.getConstant(NumBytes, CLI.DL, PtrVT, true), CLI.DL);
 308
 309   SmallVector<std::pair<unsigned, SDValue>, 5> RegsToPass;
 310
 311   // Walk arg assignments
 312   for (unsigned i = 0, e = ArgLocs.size(); i != e; ++i) {
 313     CCValAssign &VA = ArgLocs[i];
 314     SDValue Arg = OutVals[i];
 315
 316     // Promote the value if needed.
 317     switch (VA.getLocInfo()) {
 318     default:
 319       llvm_unreachable("Unknown loc info");
 320     case CCValAssign::Full:
 321       break;
 322     case CCValAssign::SExt:
 323       Arg = DAG.getNode(ISD::SIGN_EXTEND, CLI.DL, VA.getLocVT(), Arg);
 324       break;
 325     case CCValAssign::ZExt:
 326       Arg = DAG.getNode(ISD::ZERO_EXTEND, CLI.DL, VA.getLocVT(), Arg);
 327       break;
 328     case CCValAssign::AExt:
 329       Arg = DAG.getNode(ISD::ANY_EXTEND, CLI.DL, VA.getLocVT(), Arg);
 330       break;
 331     }
 332
 333     // Push arguments into RegsToPass vector
 334     if (VA.isRegLoc())
 335       RegsToPass.push_back(std::make_pair(VA.getLocReg(), Arg));
 336     else
 337       llvm_unreachable("call arg pass bug");
 338   }
 339
 340   SDValue InFlag;
 341
 342   // Build a sequence of copy-to-reg nodes chained together with token chain and
 343   // flag operands which copy the outgoing args into registers.  The InFlag in
 344   // necessary since all emitted instructions must be stuck together.
 345   for (auto &Reg : RegsToPass) {
 346     Chain = DAG.getCopyToReg(Chain, CLI.DL, Reg.first, Reg.second, InFlag);
 347     InFlag = Chain.getValue(1);
 348   }
 349
 350   // If the callee is a GlobalAddress node (quite common, every direct call is)
 351   // turn it into a TargetGlobalAddress node so that legalize doesn't hack it.
 352   // Likewise ExternalSymbol -> TargetExternalSymbol.
 353   if (GlobalAddressSDNode *G = dyn_cast<GlobalAddressSDNode>(Callee))
 354     Callee = DAG.getTargetGlobalAddress(G->getGlobal(), CLI.DL, PtrVT,
 355                                         G->getOffset(), 0);
 356   else if (ExternalSymbolSDNode *E = dyn_cast<ExternalSymbolSDNode>(Callee))
 357     Callee = DAG.getTargetExternalSymbol(E->getSymbol(), PtrVT, 0);
 358
 359   // Returns a chain & a flag for retval copy to use.
 360   SDVTList NodeTys = DAG.getVTList(MVT::Other, MVT::Glue);
 361   SmallVector<SDValue, 8> Ops;
 362   Ops.push_back(Chain);
 363   Ops.push_back(Callee);
 364
 365   // Add argument registers to the end of the list so that they are
 366   // known live into the call.
 367   for (auto &Reg : RegsToPass)
 368     Ops.push_back(DAG.getRegister(Reg.first, Reg.second.getValueType()));
 369
 370   if (InFlag.getNode())
 371     Ops.push_back(InFlag);
 372
 373   Chain = DAG.getNode(BPFISD::CALL, CLI.DL, NodeTys, Ops);
 374   InFlag = Chain.getValue(1);
 375
 376   // Create the CALLSEQ_END node.
 377   Chain = DAG.getCALLSEQ_END(
 378       Chain, DAG.getConstant(NumBytes, CLI.DL, PtrVT, true),
 379       DAG.getConstant(0, CLI.DL, PtrVT, true), InFlag, CLI.DL);
 380   InFlag = Chain.getValue(1);
 381
 382   // Handle result values, copying them out of physregs into vregs that we
 383   // return.
 384   return LowerCallResult(Chain, InFlag, CallConv, IsVarArg, Ins, CLI.DL, DAG,
 385                          InVals);
 386 }
 387
 388 SDValue
 389 BPFTargetLowering::LowerReturn(SDValue Chain, CallingConv::ID CallConv,
 390                                bool IsVarArg,
 391                                const SmallVectorImpl<ISD::OutputArg> &Outs,
 392                                const SmallVectorImpl<SDValue> &OutVals,
 393                                SDLoc DL, SelectionDAG &DAG) const {
 394
 395   // CCValAssign - represent the assignment of the return value to a location
 396   SmallVector<CCValAssign, 16> RVLocs;
 397   MachineFunction &MF = DAG.getMachineFunction();
 398
 399   // CCState - Info about the registers and stack slot.
 400   CCState CCInfo(CallConv, IsVarArg, MF, RVLocs, *DAG.getContext());
 401
 402   if (MF.getFunction()->getReturnType()->isAggregateType()) {
 403     DiagnosticInfoUnsupported Err(DL, *MF.getFunction(),
 404                                   "only integer returns supported", SDValue());
 405     DAG.getContext()->diagnose(Err);
 406   }
 407
 408   // Analize return values.
 409   CCInfo.AnalyzeReturn(Outs, RetCC_BPF64);
 410
 411   SDValue Flag;
 412   SmallVector<SDValue, 4> RetOps(1, Chain);
 413
 414   // Copy the result values into the output registers.
 415   for (unsigned i = 0; i != RVLocs.size(); ++i) {
 416     CCValAssign &VA = RVLocs[i];
 417     assert(VA.isRegLoc() && "Can only return in registers!");
 418
 419     Chain = DAG.getCopyToReg(Chain, DL, VA.getLocReg(), OutVals[i], Flag);
 420
 421     // Guarantee that all emitted copies are stuck together,
 422     // avoiding something bad.
 423     Flag = Chain.getValue(1);
 424     RetOps.push_back(DAG.getRegister(VA.getLocReg(), VA.getLocVT()));
 425   }
 426
 427   unsigned Opc = BPFISD::RET_FLAG;
 428   RetOps[0] = Chain; // Update chain.
 429
 430   // Add the flag if we have it.
 431   if (Flag.getNode())
 432     RetOps.push_back(Flag);
 433
 434   return DAG.getNode(Opc, DL, MVT::Other, RetOps);
 435 }
 436
 437 SDValue BPFTargetLowering::LowerCallResult(
 438     SDValue Chain, SDValue InFlag, CallingConv::ID CallConv, bool IsVarArg,
 439     const SmallVectorImpl<ISD::InputArg> &Ins, SDLoc DL, SelectionDAG &DAG,
 440     SmallVectorImpl<SDValue> &InVals) const {
 441
 442   MachineFunction &MF = DAG.getMachineFunction();
 443   // Assign locations to each value returned by this call.
 444   SmallVector<CCValAssign, 16> RVLocs;
 445   CCState CCInfo(CallConv, IsVarArg, MF, RVLocs, *DAG.getContext());
 446
 447   if (Ins.size() >= 2) {
 448     DiagnosticInfoUnsupported Err(DL, *MF.getFunction(),
 449                                   "only small returns supported", SDValue());
 450     DAG.getContext()->diagnose(Err);
 451   }
 452
 453   CCInfo.AnalyzeCallResult(Ins, RetCC_BPF64);
 454
 455   // Copy all of the result registers out of their specified physreg.
 456   for (auto &Val : RVLocs) {
 457     Chain = DAG.getCopyFromReg(Chain, DL, Val.getLocReg(),
 458                                Val.getValVT(), InFlag).getValue(1);
 459     InFlag = Chain.getValue(2);
 460     InVals.push_back(Chain.getValue(0));
 461   }
 462
 463   return Chain;
 464 }
 465
 466 static void NegateCC(SDValue &LHS, SDValue &RHS, ISD::CondCode &CC) {
 467   switch (CC) {
 468   default:
 469     break;
 470   case ISD::SETULT:
 471   case ISD::SETULE:
 472   case ISD::SETLT:
 473   case ISD::SETLE:
 474     CC = ISD::getSetCCSwappedOperands(CC);
 475     std::swap(LHS, RHS);
 476     break;
 477   }
 478 }
 479
 480 SDValue BPFTargetLowering::LowerBR_CC(SDValue Op, SelectionDAG &DAG) const {
 481   SDValue Chain = Op.getOperand(0);
 482   ISD::CondCode CC = cast<CondCodeSDNode>(Op.getOperand(1))->get();
 483   SDValue LHS = Op.getOperand(2);
 484   SDValue RHS = Op.getOperand(3);
 485   SDValue Dest = Op.getOperand(4);
 486   SDLoc DL(Op);
 487
 488   NegateCC(LHS, RHS, CC);
 489
 490   return DAG.getNode(BPFISD::BR_CC, DL, Op.getValueType(), Chain, LHS, RHS,
 491                      DAG.getConstant(CC, DL, MVT::i64), Dest);
 492 }
 493
 494 SDValue BPFTargetLowering::LowerSELECT_CC(SDValue Op, SelectionDAG &DAG) const {
 495   SDValue LHS = Op.getOperand(0);
 496   SDValue RHS = Op.getOperand(1);
 497   SDValue TrueV = Op.getOperand(2);
 498   SDValue FalseV = Op.getOperand(3);
 499   ISD::CondCode CC = cast<CondCodeSDNode>(Op.getOperand(4))->get();
 500   SDLoc DL(Op);
 501
 502   NegateCC(LHS, RHS, CC);
 503
 504   SDValue TargetCC = DAG.getConstant(CC, DL, MVT::i64);
 505
 506   SDVTList VTs = DAG.getVTList(Op.getValueType(), MVT::Glue);
 507   SDValue Ops[] = {LHS, RHS, TargetCC, TrueV, FalseV};
 508
 509   return DAG.getNode(BPFISD::SELECT_CC, DL, VTs, Ops);
 510 }
 511
 512 const char *BPFTargetLowering::getTargetNodeName(unsigned Opcode) const {
 513   switch ((BPFISD::NodeType)Opcode) {
 514   case BPFISD::FIRST_NUMBER:
 515     break;
 516   case BPFISD::RET_FLAG:
 517     return "BPFISD::RET_FLAG";
 518   case BPFISD::CALL:
 519     return "BPFISD::CALL";
 520   case BPFISD::SELECT_CC:
 521     return "BPFISD::SELECT_CC";
 522   case BPFISD::BR_CC:
 523     return "BPFISD::BR_CC";
 524   case BPFISD::Wrapper:
 525     return "BPFISD::Wrapper";
 526   }
 527   return nullptr;
 528 }
 529
 530 SDValue BPFTargetLowering::LowerGlobalAddress(SDValue Op,
 531                                               SelectionDAG &DAG) const {
 532   SDLoc DL(Op);
 533   const GlobalValue *GV = cast<GlobalAddressSDNode>(Op)->getGlobal();
 534   SDValue GA = DAG.getTargetGlobalAddress(GV, DL, MVT::i64);
 535
 536   return DAG.getNode(BPFISD::Wrapper, DL, MVT::i64, GA);
 537 }
 538
 539 MachineBasicBlock *
 540 BPFTargetLowering::EmitInstrWithCustomInserter(MachineInstr *MI,
 541                                                MachineBasicBlock *BB) const {
 542   const TargetInstrInfo &TII = *BB->getParent()->getSubtarget().getInstrInfo();
 543   DebugLoc DL = MI->getDebugLoc();
 544
 545   assert(MI->getOpcode() == BPF::Select && "Unexpected instr type to insert");
 546
 547   // To "insert" a SELECT instruction, we actually have to insert the diamond
 548   // control-flow pattern.  The incoming instruction knows the destination vreg
 549   // to set, the condition code register to branch on, the true/false values to
 550   // select between, and a branch opcode to use.
 551   const BasicBlock *LLVM_BB = BB->getBasicBlock();
 552   MachineFunction::iterator I = BB;
 553   ++I;
 554
 555   // ThisMBB:
 556   // ...
 557   //  TrueVal = ...
 558   //  jmp_XX r1, r2 goto Copy1MBB
 559   //  fallthrough --> Copy0MBB
 560   MachineBasicBlock *ThisMBB = BB;
 561   MachineFunction *F = BB->getParent();
 562   MachineBasicBlock *Copy0MBB = F->CreateMachineBasicBlock(LLVM_BB);
 563   MachineBasicBlock *Copy1MBB = F->CreateMachineBasicBlock(LLVM_BB);
 564
 565   F->insert(I, Copy0MBB);
 566   F->insert(I, Copy1MBB);
 567   // Update machine-CFG edges by transferring all successors of the current
 568   // block to the new block which will contain the Phi node for the select.
 569   Copy1MBB->splice(Copy1MBB->begin(), BB,
 570                    std::next(MachineBasicBlock::iterator(MI)), BB->end());
 571   Copy1MBB->transferSuccessorsAndUpdatePHIs(BB);
 572   // Next, add the true and fallthrough blocks as its successors.
 573   BB->addSuccessor(Copy0MBB);
 574   BB->addSuccessor(Copy1MBB);
 575
 576   // Insert Branch if Flag
 577   unsigned LHS = MI->getOperand(1).getReg();
 578   unsigned RHS = MI->getOperand(2).getReg();
 579   int CC = MI->getOperand(3).getImm();
 580   switch (CC) {
 581   case ISD::SETGT:
 582     BuildMI(BB, DL, TII.get(BPF::JSGT_rr))
 583         .addReg(LHS)
 584         .addReg(RHS)
 585         .addMBB(Copy1MBB);
 586     break;
 587   case ISD::SETUGT:
 588     BuildMI(BB, DL, TII.get(BPF::JUGT_rr))
 589         .addReg(LHS)
 590         .addReg(RHS)
 591         .addMBB(Copy1MBB);
 592     break;
 593   case ISD::SETGE:
 594     BuildMI(BB, DL, TII.get(BPF::JSGE_rr))
 595         .addReg(LHS)
 596         .addReg(RHS)
 597         .addMBB(Copy1MBB);
 598     break;
 599   case ISD::SETUGE:
 600     BuildMI(BB, DL, TII.get(BPF::JUGE_rr))
 601         .addReg(LHS)
 602         .addReg(RHS)
 603         .addMBB(Copy1MBB);
 604     break;
 605   case ISD::SETEQ:
 606     BuildMI(BB, DL, TII.get(BPF::JEQ_rr))
 607         .addReg(LHS)
 608         .addReg(RHS)
 609         .addMBB(Copy1MBB);
 610     break;
 611   case ISD::SETNE:
 612     BuildMI(BB, DL, TII.get(BPF::JNE_rr))
 613         .addReg(LHS)
 614         .addReg(RHS)
 615         .addMBB(Copy1MBB);
 616     break;
 617   default:
 618     report_fatal_error("unimplemented select CondCode " + Twine(CC));
 619   }
 620
 621   // Copy0MBB:
 622   //  %FalseValue = ...
 623   //  # fallthrough to Copy1MBB
 624   BB = Copy0MBB;
 625
 626   // Update machine-CFG edges
 627   BB->addSuccessor(Copy1MBB);
 628
 629   // Copy1MBB:
 630   //  %Result = phi [ %FalseValue, Copy0MBB ], [ %TrueValue, ThisMBB ]
 631   // ...
 632   BB = Copy1MBB;
 633   BuildMI(*BB, BB->begin(), DL, TII.get(BPF::PHI), MI->getOperand(0).getReg())
 634       .addReg(MI->getOperand(5).getReg())
 635       .addMBB(Copy0MBB)
 636       .addReg(MI->getOperand(4).getReg())
 637       .addMBB(ThisMBB);
 638
 639   MI->eraseFromParent(); // The pseudo instruction is gone now.
 640   return BB;
 641 }