diff options
Diffstat (limited to 'llvm/lib/Target/BPF/BPFISelLowering.cpp')
| -rw-r--r-- | llvm/lib/Target/BPF/BPFISelLowering.cpp | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/llvm/lib/Target/BPF/BPFISelLowering.cpp b/llvm/lib/Target/BPF/BPFISelLowering.cpp index 066dc5e31e1..b2e4783b3ef 100644 --- a/llvm/lib/Target/BPF/BPFISelLowering.cpp +++ b/llvm/lib/Target/BPF/BPFISelLowering.cpp @@ -209,6 +209,8 @@ SDValue BPFTargetLowering::LowerFormalArguments( return Chain; } +const unsigned BPFTargetLowering::MaxArgs = 5; + SDValue BPFTargetLowering::LowerCall(TargetLowering::CallLoweringInfo &CLI, SmallVectorImpl<SDValue> &InVals) const { SelectionDAG &DAG = CLI.DAG; @@ -241,9 +243,8 @@ SDValue BPFTargetLowering::LowerCall(TargetLowering::CallLoweringInfo &CLI, unsigned NumBytes = CCInfo.getNextStackOffset(); - if (Outs.size() >= 6) { + if (Outs.size() > MaxArgs) fail(CLI.DL, DAG, "too many args to ", Callee); - } for (auto &Arg : Outs) { ISD::ArgFlagsTy Flags = Arg.Flags; @@ -257,10 +258,12 @@ SDValue BPFTargetLowering::LowerCall(TargetLowering::CallLoweringInfo &CLI, Chain = DAG.getCALLSEQ_START( Chain, DAG.getConstant(NumBytes, CLI.DL, PtrVT, true), CLI.DL); - SmallVector<std::pair<unsigned, SDValue>, 5> RegsToPass; + SmallVector<std::pair<unsigned, SDValue>, MaxArgs> RegsToPass; // Walk arg assignments - for (unsigned i = 0, e = ArgLocs.size(); i != e; ++i) { + for (unsigned i = 0, + e = std::min(static_cast<unsigned>(ArgLocs.size()), MaxArgs); + i != e; ++i) { CCValAssign &VA = ArgLocs[i]; SDValue Arg = OutVals[i]; |
