Index: lib/CodeGen/ImplicitNullChecks.cpp =================================================================== --- lib/CodeGen/ImplicitNullChecks.cpp +++ lib/CodeGen/ImplicitNullChecks.cpp @@ -158,11 +158,17 @@ MachineBasicBlock *HandlerMBB); void rewriteNullChecks(ArrayRef NullCheckList); + enum SuitableResult { + SR_OK, + SR_Continue, + SR_Impossible + }; + /// Is \p MI a memory operation that can be used to implicitly null check the /// value in \p PointerReg? \p PrevInsts is the set of instruction seen since /// the explicit null check on \p PointerReg. - bool isSuitableMemoryOp(MachineInstr &MI, unsigned PointerReg, - ArrayRef PrevInsts); + SuitableResult isSuitableMemoryOp(MachineInstr &MI, unsigned PointerReg, + ArrayRef PrevInsts); /// Return true if \p FaultingMI can be hoisted from after the the /// instructions in \p InstsSeenSoFar to before them. Set \p Dependence to a @@ -283,19 +289,19 @@ return false; } -bool ImplicitNullChecks::isSuitableMemoryOp( +ImplicitNullChecks::SuitableResult ImplicitNullChecks::isSuitableMemoryOp( MachineInstr &MI, unsigned PointerReg, ArrayRef PrevInsts) { int64_t Offset; unsigned BaseReg; if (!TII->getMemOpBaseRegImmOfs(MI, BaseReg, Offset, TRI) || BaseReg != PointerReg) - return false; + return SR_Continue; // We want the load to be issued at a sane offset from PointerReg, so that // if PointerReg is null then the load reliably page faults. if (!(MI.mayLoad() && !MI.isPredicable() && Offset < PageSize)) - return false; + return SR_Continue; // Finally, we need to make sure that the load instruction actually is // loading from PointerReg, and there isn't some re-definition of PointerReg @@ -304,9 +310,9 @@ for (auto &PrevMO : PrevMI->operands()) if (PrevMO.isReg() && PrevMO.getReg() && TRI->regsOverlap(PrevMO.getReg(), PointerReg)) - return false; + return SR_Impossible; - return true; + return SR_OK; } bool ImplicitNullChecks::canHoistLoadInst( @@ -481,7 +487,10 @@ return false; MachineInstr *Dependence; - if (isSuitableMemoryOp(MI, PointerReg, InstsSeenSoFar) && + SuitableResult SRResult = isSuitableMemoryOp(MI, PointerReg, InstsSeenSoFar); + if (SRResult == SR_Impossible) + return false; + if ((SRResult == SR_OK) && canHoistLoadInst(&MI, PointerReg, InstsSeenSoFar, NullSucc, Dependence)) { NullCheckList.emplace_back(&MI, MBP.ConditionDef, &MBB, NotNullSucc,