1 //===- MipsCallLowering.cpp -------------------------------------*- C++ -*-===//
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 /// \file
10 /// This file implements the lowering of LLVM calls to machine code calls for
11 /// GlobalISel.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #include "MipsCallLowering.h"
16 #include "MipsCCState.h"
17 #include "MipsMachineFunction.h"
18 #include "MipsTargetMachine.h"
19 #include "llvm/CodeGen/Analysis.h"
20 #include "llvm/CodeGen/GlobalISel/MachineIRBuilder.h"
21
22 using namespace llvm;
23
MipsCallLowering(const MipsTargetLowering & TLI)24 MipsCallLowering::MipsCallLowering(const MipsTargetLowering &TLI)
25 : CallLowering(&TLI) {}
26
assign(Register VReg,const CCValAssign & VA,const EVT & VT)27 bool MipsCallLowering::MipsHandler::assign(Register VReg, const CCValAssign &VA,
28 const EVT &VT) {
29 if (VA.isRegLoc()) {
30 assignValueToReg(VReg, VA, VT);
31 } else if (VA.isMemLoc()) {
32 assignValueToAddress(VReg, VA);
33 } else {
34 return false;
35 }
36 return true;
37 }
38
assignVRegs(ArrayRef<Register> VRegs,ArrayRef<CCValAssign> ArgLocs,unsigned ArgLocsStartIndex,const EVT & VT)39 bool MipsCallLowering::MipsHandler::assignVRegs(ArrayRef<Register> VRegs,
40 ArrayRef<CCValAssign> ArgLocs,
41 unsigned ArgLocsStartIndex,
42 const EVT &VT) {
43 for (unsigned i = 0; i < VRegs.size(); ++i)
44 if (!assign(VRegs[i], ArgLocs[ArgLocsStartIndex + i], VT))
45 return false;
46 return true;
47 }
48
setLeastSignificantFirst(SmallVectorImpl<Register> & VRegs)49 void MipsCallLowering::MipsHandler::setLeastSignificantFirst(
50 SmallVectorImpl<Register> &VRegs) {
51 if (!MIRBuilder.getMF().getDataLayout().isLittleEndian())
52 std::reverse(VRegs.begin(), VRegs.end());
53 }
54
handle(ArrayRef<CCValAssign> ArgLocs,ArrayRef<CallLowering::ArgInfo> Args)55 bool MipsCallLowering::MipsHandler::handle(
56 ArrayRef<CCValAssign> ArgLocs, ArrayRef<CallLowering::ArgInfo> Args) {
57 SmallVector<Register, 4> VRegs;
58 unsigned SplitLength;
59 const Function &F = MIRBuilder.getMF().getFunction();
60 const DataLayout &DL = F.getParent()->getDataLayout();
61 const MipsTargetLowering &TLI = *static_cast<const MipsTargetLowering *>(
62 MIRBuilder.getMF().getSubtarget().getTargetLowering());
63
64 for (unsigned ArgsIndex = 0, ArgLocsIndex = 0; ArgsIndex < Args.size();
65 ++ArgsIndex, ArgLocsIndex += SplitLength) {
66 EVT VT = TLI.getValueType(DL, Args[ArgsIndex].Ty);
67 SplitLength = TLI.getNumRegistersForCallingConv(F.getContext(),
68 F.getCallingConv(), VT);
69 assert(Args[ArgsIndex].Regs.size() == 1 && "Can't handle multple regs yet");
70
71 if (SplitLength > 1) {
72 VRegs.clear();
73 MVT RegisterVT = TLI.getRegisterTypeForCallingConv(
74 F.getContext(), F.getCallingConv(), VT);
75 for (unsigned i = 0; i < SplitLength; ++i)
76 VRegs.push_back(MRI.createGenericVirtualRegister(LLT{RegisterVT}));
77
78 if (!handleSplit(VRegs, ArgLocs, ArgLocsIndex, Args[ArgsIndex].Regs[0],
79 VT))
80 return false;
81 } else {
82 if (!assign(Args[ArgsIndex].Regs[0], ArgLocs[ArgLocsIndex], VT))
83 return false;
84 }
85 }
86 return true;
87 }
88
89 namespace {
90 class IncomingValueHandler : public MipsCallLowering::MipsHandler {
91 public:
IncomingValueHandler(MachineIRBuilder & MIRBuilder,MachineRegisterInfo & MRI)92 IncomingValueHandler(MachineIRBuilder &MIRBuilder, MachineRegisterInfo &MRI)
93 : MipsHandler(MIRBuilder, MRI) {}
94
95 private:
96 void assignValueToReg(Register ValVReg, const CCValAssign &VA,
97 const EVT &VT) override;
98
99 Register getStackAddress(const CCValAssign &VA,
100 MachineMemOperand *&MMO) override;
101
102 void assignValueToAddress(Register ValVReg, const CCValAssign &VA) override;
103
104 bool handleSplit(SmallVectorImpl<Register> &VRegs,
105 ArrayRef<CCValAssign> ArgLocs, unsigned ArgLocsStartIndex,
106 Register ArgsReg, const EVT &VT) override;
107
markPhysRegUsed(unsigned PhysReg)108 virtual void markPhysRegUsed(unsigned PhysReg) {
109 MIRBuilder.getMRI()->addLiveIn(PhysReg);
110 MIRBuilder.getMBB().addLiveIn(PhysReg);
111 }
112
buildLoad(Register Val,const CCValAssign & VA)113 void buildLoad(Register Val, const CCValAssign &VA) {
114 MachineMemOperand *MMO;
115 Register Addr = getStackAddress(VA, MMO);
116 MIRBuilder.buildLoad(Val, Addr, *MMO);
117 }
118 };
119
120 class CallReturnHandler : public IncomingValueHandler {
121 public:
CallReturnHandler(MachineIRBuilder & MIRBuilder,MachineRegisterInfo & MRI,MachineInstrBuilder & MIB)122 CallReturnHandler(MachineIRBuilder &MIRBuilder, MachineRegisterInfo &MRI,
123 MachineInstrBuilder &MIB)
124 : IncomingValueHandler(MIRBuilder, MRI), MIB(MIB) {}
125
126 private:
markPhysRegUsed(unsigned PhysReg)127 void markPhysRegUsed(unsigned PhysReg) override {
128 MIB.addDef(PhysReg, RegState::Implicit);
129 }
130
131 MachineInstrBuilder &MIB;
132 };
133
134 } // end anonymous namespace
135
assignValueToReg(Register ValVReg,const CCValAssign & VA,const EVT & VT)136 void IncomingValueHandler::assignValueToReg(Register ValVReg,
137 const CCValAssign &VA,
138 const EVT &VT) {
139 const MipsSubtarget &STI =
140 static_cast<const MipsSubtarget &>(MIRBuilder.getMF().getSubtarget());
141 Register PhysReg = VA.getLocReg();
142 if (VT == MVT::f64 && PhysReg >= Mips::A0 && PhysReg <= Mips::A3) {
143 const MipsSubtarget &STI =
144 static_cast<const MipsSubtarget &>(MIRBuilder.getMF().getSubtarget());
145
146 MIRBuilder
147 .buildInstr(STI.isFP64bit() ? Mips::BuildPairF64_64
148 : Mips::BuildPairF64)
149 .addDef(ValVReg)
150 .addUse(PhysReg + (STI.isLittle() ? 0 : 1))
151 .addUse(PhysReg + (STI.isLittle() ? 1 : 0))
152 .constrainAllUses(MIRBuilder.getTII(), *STI.getRegisterInfo(),
153 *STI.getRegBankInfo());
154 markPhysRegUsed(PhysReg);
155 markPhysRegUsed(PhysReg + 1);
156 } else if (VT == MVT::f32 && PhysReg >= Mips::A0 && PhysReg <= Mips::A3) {
157 MIRBuilder.buildInstr(Mips::MTC1)
158 .addDef(ValVReg)
159 .addUse(PhysReg)
160 .constrainAllUses(MIRBuilder.getTII(), *STI.getRegisterInfo(),
161 *STI.getRegBankInfo());
162 markPhysRegUsed(PhysReg);
163 } else {
164 switch (VA.getLocInfo()) {
165 case CCValAssign::LocInfo::SExt:
166 case CCValAssign::LocInfo::ZExt:
167 case CCValAssign::LocInfo::AExt: {
168 auto Copy = MIRBuilder.buildCopy(LLT{VA.getLocVT()}, PhysReg);
169 MIRBuilder.buildTrunc(ValVReg, Copy);
170 break;
171 }
172 default:
173 MIRBuilder.buildCopy(ValVReg, PhysReg);
174 break;
175 }
176 markPhysRegUsed(PhysReg);
177 }
178 }
179
getStackAddress(const CCValAssign & VA,MachineMemOperand * & MMO)180 Register IncomingValueHandler::getStackAddress(const CCValAssign &VA,
181 MachineMemOperand *&MMO) {
182 MachineFunction &MF = MIRBuilder.getMF();
183 unsigned Size = alignTo(VA.getValVT().getSizeInBits(), 8) / 8;
184 unsigned Offset = VA.getLocMemOffset();
185 MachineFrameInfo &MFI = MF.getFrameInfo();
186
187 int FI = MFI.CreateFixedObject(Size, Offset, true);
188 MachinePointerInfo MPO =
189 MachinePointerInfo::getFixedStack(MIRBuilder.getMF(), FI);
190
191 const TargetFrameLowering *TFL = MF.getSubtarget().getFrameLowering();
192 unsigned Align = MinAlign(TFL->getStackAlignment(), Offset);
193 MMO = MF.getMachineMemOperand(MPO, MachineMemOperand::MOLoad, Size, Align);
194
195 Register AddrReg = MRI.createGenericVirtualRegister(LLT::pointer(0, 32));
196 MIRBuilder.buildFrameIndex(AddrReg, FI);
197
198 return AddrReg;
199 }
200
assignValueToAddress(Register ValVReg,const CCValAssign & VA)201 void IncomingValueHandler::assignValueToAddress(Register ValVReg,
202 const CCValAssign &VA) {
203 if (VA.getLocInfo() == CCValAssign::SExt ||
204 VA.getLocInfo() == CCValAssign::ZExt ||
205 VA.getLocInfo() == CCValAssign::AExt) {
206 Register LoadReg = MRI.createGenericVirtualRegister(LLT::scalar(32));
207 buildLoad(LoadReg, VA);
208 MIRBuilder.buildTrunc(ValVReg, LoadReg);
209 } else
210 buildLoad(ValVReg, VA);
211 }
212
handleSplit(SmallVectorImpl<Register> & VRegs,ArrayRef<CCValAssign> ArgLocs,unsigned ArgLocsStartIndex,Register ArgsReg,const EVT & VT)213 bool IncomingValueHandler::handleSplit(SmallVectorImpl<Register> &VRegs,
214 ArrayRef<CCValAssign> ArgLocs,
215 unsigned ArgLocsStartIndex,
216 Register ArgsReg, const EVT &VT) {
217 if (!assignVRegs(VRegs, ArgLocs, ArgLocsStartIndex, VT))
218 return false;
219 setLeastSignificantFirst(VRegs);
220 MIRBuilder.buildMerge(ArgsReg, VRegs);
221 return true;
222 }
223
224 namespace {
225 class OutgoingValueHandler : public MipsCallLowering::MipsHandler {
226 public:
OutgoingValueHandler(MachineIRBuilder & MIRBuilder,MachineRegisterInfo & MRI,MachineInstrBuilder & MIB)227 OutgoingValueHandler(MachineIRBuilder &MIRBuilder, MachineRegisterInfo &MRI,
228 MachineInstrBuilder &MIB)
229 : MipsHandler(MIRBuilder, MRI), MIB(MIB) {}
230
231 private:
232 void assignValueToReg(Register ValVReg, const CCValAssign &VA,
233 const EVT &VT) override;
234
235 Register getStackAddress(const CCValAssign &VA,
236 MachineMemOperand *&MMO) override;
237
238 void assignValueToAddress(Register ValVReg, const CCValAssign &VA) override;
239
240 bool handleSplit(SmallVectorImpl<Register> &VRegs,
241 ArrayRef<CCValAssign> ArgLocs, unsigned ArgLocsStartIndex,
242 Register ArgsReg, const EVT &VT) override;
243
244 Register extendRegister(Register ValReg, const CCValAssign &VA);
245
246 MachineInstrBuilder &MIB;
247 };
248 } // end anonymous namespace
249
assignValueToReg(Register ValVReg,const CCValAssign & VA,const EVT & VT)250 void OutgoingValueHandler::assignValueToReg(Register ValVReg,
251 const CCValAssign &VA,
252 const EVT &VT) {
253 Register PhysReg = VA.getLocReg();
254 const MipsSubtarget &STI =
255 static_cast<const MipsSubtarget &>(MIRBuilder.getMF().getSubtarget());
256
257 if (VT == MVT::f64 && PhysReg >= Mips::A0 && PhysReg <= Mips::A3) {
258 MIRBuilder
259 .buildInstr(STI.isFP64bit() ? Mips::ExtractElementF64_64
260 : Mips::ExtractElementF64)
261 .addDef(PhysReg + (STI.isLittle() ? 1 : 0))
262 .addUse(ValVReg)
263 .addImm(1)
264 .constrainAllUses(MIRBuilder.getTII(), *STI.getRegisterInfo(),
265 *STI.getRegBankInfo());
266 MIRBuilder
267 .buildInstr(STI.isFP64bit() ? Mips::ExtractElementF64_64
268 : Mips::ExtractElementF64)
269 .addDef(PhysReg + (STI.isLittle() ? 0 : 1))
270 .addUse(ValVReg)
271 .addImm(0)
272 .constrainAllUses(MIRBuilder.getTII(), *STI.getRegisterInfo(),
273 *STI.getRegBankInfo());
274 } else if (VT == MVT::f32 && PhysReg >= Mips::A0 && PhysReg <= Mips::A3) {
275 MIRBuilder.buildInstr(Mips::MFC1)
276 .addDef(PhysReg)
277 .addUse(ValVReg)
278 .constrainAllUses(MIRBuilder.getTII(), *STI.getRegisterInfo(),
279 *STI.getRegBankInfo());
280 } else {
281 Register ExtReg = extendRegister(ValVReg, VA);
282 MIRBuilder.buildCopy(PhysReg, ExtReg);
283 MIB.addUse(PhysReg, RegState::Implicit);
284 }
285 }
286
getStackAddress(const CCValAssign & VA,MachineMemOperand * & MMO)287 Register OutgoingValueHandler::getStackAddress(const CCValAssign &VA,
288 MachineMemOperand *&MMO) {
289 MachineFunction &MF = MIRBuilder.getMF();
290 const TargetFrameLowering *TFL = MF.getSubtarget().getFrameLowering();
291
292 LLT p0 = LLT::pointer(0, 32);
293 LLT s32 = LLT::scalar(32);
294 Register SPReg = MRI.createGenericVirtualRegister(p0);
295 MIRBuilder.buildCopy(SPReg, Register(Mips::SP));
296
297 Register OffsetReg = MRI.createGenericVirtualRegister(s32);
298 unsigned Offset = VA.getLocMemOffset();
299 MIRBuilder.buildConstant(OffsetReg, Offset);
300
301 Register AddrReg = MRI.createGenericVirtualRegister(p0);
302 MIRBuilder.buildPtrAdd(AddrReg, SPReg, OffsetReg);
303
304 MachinePointerInfo MPO =
305 MachinePointerInfo::getStack(MIRBuilder.getMF(), Offset);
306 unsigned Size = alignTo(VA.getValVT().getSizeInBits(), 8) / 8;
307 unsigned Align = MinAlign(TFL->getStackAlignment(), Offset);
308 MMO = MF.getMachineMemOperand(MPO, MachineMemOperand::MOStore, Size, Align);
309
310 return AddrReg;
311 }
312
assignValueToAddress(Register ValVReg,const CCValAssign & VA)313 void OutgoingValueHandler::assignValueToAddress(Register ValVReg,
314 const CCValAssign &VA) {
315 MachineMemOperand *MMO;
316 Register Addr = getStackAddress(VA, MMO);
317 Register ExtReg = extendRegister(ValVReg, VA);
318 MIRBuilder.buildStore(ExtReg, Addr, *MMO);
319 }
320
extendRegister(Register ValReg,const CCValAssign & VA)321 Register OutgoingValueHandler::extendRegister(Register ValReg,
322 const CCValAssign &VA) {
323 LLT LocTy{VA.getLocVT()};
324 switch (VA.getLocInfo()) {
325 case CCValAssign::SExt: {
326 Register ExtReg = MRI.createGenericVirtualRegister(LocTy);
327 MIRBuilder.buildSExt(ExtReg, ValReg);
328 return ExtReg;
329 }
330 case CCValAssign::ZExt: {
331 Register ExtReg = MRI.createGenericVirtualRegister(LocTy);
332 MIRBuilder.buildZExt(ExtReg, ValReg);
333 return ExtReg;
334 }
335 case CCValAssign::AExt: {
336 Register ExtReg = MRI.createGenericVirtualRegister(LocTy);
337 MIRBuilder.buildAnyExt(ExtReg, ValReg);
338 return ExtReg;
339 }
340 // TODO : handle upper extends
341 case CCValAssign::Full:
342 return ValReg;
343 default:
344 break;
345 }
346 llvm_unreachable("unable to extend register");
347 }
348
handleSplit(SmallVectorImpl<Register> & VRegs,ArrayRef<CCValAssign> ArgLocs,unsigned ArgLocsStartIndex,Register ArgsReg,const EVT & VT)349 bool OutgoingValueHandler::handleSplit(SmallVectorImpl<Register> &VRegs,
350 ArrayRef<CCValAssign> ArgLocs,
351 unsigned ArgLocsStartIndex,
352 Register ArgsReg, const EVT &VT) {
353 MIRBuilder.buildUnmerge(VRegs, ArgsReg);
354 setLeastSignificantFirst(VRegs);
355 if (!assignVRegs(VRegs, ArgLocs, ArgLocsStartIndex, VT))
356 return false;
357
358 return true;
359 }
360
isSupportedArgumentType(Type * T)361 static bool isSupportedArgumentType(Type *T) {
362 if (T->isIntegerTy())
363 return true;
364 if (T->isPointerTy())
365 return true;
366 if (T->isFloatingPointTy())
367 return true;
368 return false;
369 }
370
isSupportedReturnType(Type * T)371 static bool isSupportedReturnType(Type *T) {
372 if (T->isIntegerTy())
373 return true;
374 if (T->isPointerTy())
375 return true;
376 if (T->isFloatingPointTy())
377 return true;
378 if (T->isAggregateType())
379 return true;
380 return false;
381 }
382
determineLocInfo(const MVT RegisterVT,const EVT VT,const ISD::ArgFlagsTy & Flags)383 static CCValAssign::LocInfo determineLocInfo(const MVT RegisterVT, const EVT VT,
384 const ISD::ArgFlagsTy &Flags) {
385 // > does not mean loss of information as type RegisterVT can't hold type VT,
386 // it means that type VT is split into multiple registers of type RegisterVT
387 if (VT.getSizeInBits() >= RegisterVT.getSizeInBits())
388 return CCValAssign::LocInfo::Full;
389 if (Flags.isSExt())
390 return CCValAssign::LocInfo::SExt;
391 if (Flags.isZExt())
392 return CCValAssign::LocInfo::ZExt;
393 return CCValAssign::LocInfo::AExt;
394 }
395
396 template <typename T>
setLocInfo(SmallVectorImpl<CCValAssign> & ArgLocs,const SmallVectorImpl<T> & Arguments)397 static void setLocInfo(SmallVectorImpl<CCValAssign> &ArgLocs,
398 const SmallVectorImpl<T> &Arguments) {
399 for (unsigned i = 0; i < ArgLocs.size(); ++i) {
400 const CCValAssign &VA = ArgLocs[i];
401 CCValAssign::LocInfo LocInfo = determineLocInfo(
402 Arguments[i].VT, Arguments[i].ArgVT, Arguments[i].Flags);
403 if (VA.isMemLoc())
404 ArgLocs[i] =
405 CCValAssign::getMem(VA.getValNo(), VA.getValVT(),
406 VA.getLocMemOffset(), VA.getLocVT(), LocInfo);
407 else
408 ArgLocs[i] = CCValAssign::getReg(VA.getValNo(), VA.getValVT(),
409 VA.getLocReg(), VA.getLocVT(), LocInfo);
410 }
411 }
412
lowerReturn(MachineIRBuilder & MIRBuilder,const Value * Val,ArrayRef<Register> VRegs) const413 bool MipsCallLowering::lowerReturn(MachineIRBuilder &MIRBuilder,
414 const Value *Val,
415 ArrayRef<Register> VRegs) const {
416
417 MachineInstrBuilder Ret = MIRBuilder.buildInstrNoInsert(Mips::RetRA);
418
419 if (Val != nullptr && !isSupportedReturnType(Val->getType()))
420 return false;
421
422 if (!VRegs.empty()) {
423 MachineFunction &MF = MIRBuilder.getMF();
424 const Function &F = MF.getFunction();
425 const DataLayout &DL = MF.getDataLayout();
426 const MipsTargetLowering &TLI = *getTLI<MipsTargetLowering>();
427
428 SmallVector<ArgInfo, 8> RetInfos;
429 SmallVector<unsigned, 8> OrigArgIndices;
430
431 ArgInfo ArgRetInfo(VRegs, Val->getType());
432 setArgFlags(ArgRetInfo, AttributeList::ReturnIndex, DL, F);
433 splitToValueTypes(DL, ArgRetInfo, 0, RetInfos, OrigArgIndices);
434
435 SmallVector<ISD::OutputArg, 8> Outs;
436 subTargetRegTypeForCallingConv(F, RetInfos, OrigArgIndices, Outs);
437
438 SmallVector<CCValAssign, 16> ArgLocs;
439 MipsCCState CCInfo(F.getCallingConv(), F.isVarArg(), MF, ArgLocs,
440 F.getContext());
441 CCInfo.AnalyzeReturn(Outs, TLI.CCAssignFnForReturn());
442 setLocInfo(ArgLocs, Outs);
443
444 OutgoingValueHandler RetHandler(MIRBuilder, MF.getRegInfo(), Ret);
445 if (!RetHandler.handle(ArgLocs, RetInfos)) {
446 return false;
447 }
448 }
449 MIRBuilder.insertInstr(Ret);
450 return true;
451 }
452
lowerFormalArguments(MachineIRBuilder & MIRBuilder,const Function & F,ArrayRef<ArrayRef<Register>> VRegs) const453 bool MipsCallLowering::lowerFormalArguments(
454 MachineIRBuilder &MIRBuilder, const Function &F,
455 ArrayRef<ArrayRef<Register>> VRegs) const {
456
457 // Quick exit if there aren't any args.
458 if (F.arg_empty())
459 return true;
460
461 for (auto &Arg : F.args()) {
462 if (!isSupportedArgumentType(Arg.getType()))
463 return false;
464 }
465
466 MachineFunction &MF = MIRBuilder.getMF();
467 const DataLayout &DL = MF.getDataLayout();
468 const MipsTargetLowering &TLI = *getTLI<MipsTargetLowering>();
469
470 SmallVector<ArgInfo, 8> ArgInfos;
471 SmallVector<unsigned, 8> OrigArgIndices;
472 unsigned i = 0;
473 for (auto &Arg : F.args()) {
474 ArgInfo AInfo(VRegs[i], Arg.getType());
475 setArgFlags(AInfo, i + AttributeList::FirstArgIndex, DL, F);
476 ArgInfos.push_back(AInfo);
477 OrigArgIndices.push_back(i);
478 ++i;
479 }
480
481 SmallVector<ISD::InputArg, 8> Ins;
482 subTargetRegTypeForCallingConv(F, ArgInfos, OrigArgIndices, Ins);
483
484 SmallVector<CCValAssign, 16> ArgLocs;
485 MipsCCState CCInfo(F.getCallingConv(), F.isVarArg(), MF, ArgLocs,
486 F.getContext());
487
488 const MipsTargetMachine &TM =
489 static_cast<const MipsTargetMachine &>(MF.getTarget());
490 const MipsABIInfo &ABI = TM.getABI();
491 CCInfo.AllocateStack(ABI.GetCalleeAllocdArgSizeInBytes(F.getCallingConv()),
492 1);
493 CCInfo.AnalyzeFormalArguments(Ins, TLI.CCAssignFnForCall());
494 setLocInfo(ArgLocs, Ins);
495
496 IncomingValueHandler Handler(MIRBuilder, MF.getRegInfo());
497 if (!Handler.handle(ArgLocs, ArgInfos))
498 return false;
499
500 if (F.isVarArg()) {
501 ArrayRef<MCPhysReg> ArgRegs = ABI.GetVarArgRegs();
502 unsigned Idx = CCInfo.getFirstUnallocated(ArgRegs);
503
504 int VaArgOffset;
505 unsigned RegSize = 4;
506 if (ArgRegs.size() == Idx)
507 VaArgOffset = alignTo(CCInfo.getNextStackOffset(), RegSize);
508 else {
509 VaArgOffset =
510 (int)ABI.GetCalleeAllocdArgSizeInBytes(CCInfo.getCallingConv()) -
511 (int)(RegSize * (ArgRegs.size() - Idx));
512 }
513
514 MachineFrameInfo &MFI = MF.getFrameInfo();
515 int FI = MFI.CreateFixedObject(RegSize, VaArgOffset, true);
516 MF.getInfo<MipsFunctionInfo>()->setVarArgsFrameIndex(FI);
517
518 for (unsigned I = Idx; I < ArgRegs.size(); ++I, VaArgOffset += RegSize) {
519 MIRBuilder.getMBB().addLiveIn(ArgRegs[I]);
520
521 MachineInstrBuilder Copy =
522 MIRBuilder.buildCopy(LLT::scalar(RegSize * 8), Register(ArgRegs[I]));
523 FI = MFI.CreateFixedObject(RegSize, VaArgOffset, true);
524 MachinePointerInfo MPO = MachinePointerInfo::getFixedStack(MF, FI);
525 MachineInstrBuilder FrameIndex =
526 MIRBuilder.buildFrameIndex(LLT::pointer(MPO.getAddrSpace(), 32), FI);
527 MachineMemOperand *MMO =
528 MF.getMachineMemOperand(MPO, MachineMemOperand::MOStore, RegSize,
529 /* Alignment */ RegSize);
530 MIRBuilder.buildStore(Copy, FrameIndex, *MMO);
531 }
532 }
533
534 return true;
535 }
536
lowerCall(MachineIRBuilder & MIRBuilder,CallLoweringInfo & Info) const537 bool MipsCallLowering::lowerCall(MachineIRBuilder &MIRBuilder,
538 CallLoweringInfo &Info) const {
539
540 if (Info.CallConv != CallingConv::C)
541 return false;
542
543 for (auto &Arg : Info.OrigArgs) {
544 if (!isSupportedArgumentType(Arg.Ty))
545 return false;
546 if (Arg.Flags[0].isByVal())
547 return false;
548 if (Arg.Flags[0].isSRet() && !Arg.Ty->isPointerTy())
549 return false;
550 }
551
552 if (!Info.OrigRet.Ty->isVoidTy() && !isSupportedReturnType(Info.OrigRet.Ty))
553 return false;
554
555 MachineFunction &MF = MIRBuilder.getMF();
556 const Function &F = MF.getFunction();
557 const DataLayout &DL = MF.getDataLayout();
558 const MipsTargetLowering &TLI = *getTLI<MipsTargetLowering>();
559 const MipsTargetMachine &TM =
560 static_cast<const MipsTargetMachine &>(MF.getTarget());
561 const MipsABIInfo &ABI = TM.getABI();
562
563 MachineInstrBuilder CallSeqStart =
564 MIRBuilder.buildInstr(Mips::ADJCALLSTACKDOWN);
565
566 const bool IsCalleeGlobalPIC =
567 Info.Callee.isGlobal() && TM.isPositionIndependent();
568
569 MachineInstrBuilder MIB = MIRBuilder.buildInstrNoInsert(
570 Info.Callee.isReg() || IsCalleeGlobalPIC ? Mips::JALRPseudo : Mips::JAL);
571 MIB.addDef(Mips::SP, RegState::Implicit);
572 if (IsCalleeGlobalPIC) {
573 Register CalleeReg =
574 MF.getRegInfo().createGenericVirtualRegister(LLT::pointer(0, 32));
575 MachineInstr *CalleeGlobalValue =
576 MIRBuilder.buildGlobalValue(CalleeReg, Info.Callee.getGlobal());
577 if (!Info.Callee.getGlobal()->hasLocalLinkage())
578 CalleeGlobalValue->getOperand(1).setTargetFlags(MipsII::MO_GOT_CALL);
579 MIB.addUse(CalleeReg);
580 } else
581 MIB.add(Info.Callee);
582 const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo();
583 MIB.addRegMask(TRI->getCallPreservedMask(MF, F.getCallingConv()));
584
585 TargetLowering::ArgListTy FuncOrigArgs;
586 FuncOrigArgs.reserve(Info.OrigArgs.size());
587
588 SmallVector<ArgInfo, 8> ArgInfos;
589 SmallVector<unsigned, 8> OrigArgIndices;
590 unsigned i = 0;
591 for (auto &Arg : Info.OrigArgs) {
592
593 TargetLowering::ArgListEntry Entry;
594 Entry.Ty = Arg.Ty;
595 FuncOrigArgs.push_back(Entry);
596
597 ArgInfos.push_back(Arg);
598 OrigArgIndices.push_back(i);
599 ++i;
600 }
601
602 SmallVector<ISD::OutputArg, 8> Outs;
603 subTargetRegTypeForCallingConv(F, ArgInfos, OrigArgIndices, Outs);
604
605 SmallVector<CCValAssign, 8> ArgLocs;
606 bool IsCalleeVarArg = false;
607 if (Info.Callee.isGlobal()) {
608 const Function *CF = static_cast<const Function *>(Info.Callee.getGlobal());
609 IsCalleeVarArg = CF->isVarArg();
610 }
611 MipsCCState CCInfo(F.getCallingConv(), IsCalleeVarArg, MF, ArgLocs,
612 F.getContext());
613
614 CCInfo.AllocateStack(ABI.GetCalleeAllocdArgSizeInBytes(Info.CallConv), 1);
615 const char *Call =
616 Info.Callee.isSymbol() ? Info.Callee.getSymbolName() : nullptr;
617 CCInfo.AnalyzeCallOperands(Outs, TLI.CCAssignFnForCall(), FuncOrigArgs, Call);
618 setLocInfo(ArgLocs, Outs);
619
620 OutgoingValueHandler RetHandler(MIRBuilder, MF.getRegInfo(), MIB);
621 if (!RetHandler.handle(ArgLocs, ArgInfos)) {
622 return false;
623 }
624
625 unsigned NextStackOffset = CCInfo.getNextStackOffset();
626 const TargetFrameLowering *TFL = MF.getSubtarget().getFrameLowering();
627 unsigned StackAlignment = TFL->getStackAlignment();
628 NextStackOffset = alignTo(NextStackOffset, StackAlignment);
629 CallSeqStart.addImm(NextStackOffset).addImm(0);
630
631 if (IsCalleeGlobalPIC) {
632 MIRBuilder.buildCopy(
633 Register(Mips::GP),
634 MF.getInfo<MipsFunctionInfo>()->getGlobalBaseRegForGlobalISel());
635 MIB.addDef(Mips::GP, RegState::Implicit);
636 }
637 MIRBuilder.insertInstr(MIB);
638 if (MIB->getOpcode() == Mips::JALRPseudo) {
639 const MipsSubtarget &STI =
640 static_cast<const MipsSubtarget &>(MIRBuilder.getMF().getSubtarget());
641 MIB.constrainAllUses(MIRBuilder.getTII(), *STI.getRegisterInfo(),
642 *STI.getRegBankInfo());
643 }
644
645 if (!Info.OrigRet.Ty->isVoidTy()) {
646 ArgInfos.clear();
647 SmallVector<unsigned, 8> OrigRetIndices;
648
649 splitToValueTypes(DL, Info.OrigRet, 0, ArgInfos, OrigRetIndices);
650
651 SmallVector<ISD::InputArg, 8> Ins;
652 subTargetRegTypeForCallingConv(F, ArgInfos, OrigRetIndices, Ins);
653
654 SmallVector<CCValAssign, 8> ArgLocs;
655 MipsCCState CCInfo(F.getCallingConv(), F.isVarArg(), MF, ArgLocs,
656 F.getContext());
657
658 CCInfo.AnalyzeCallResult(Ins, TLI.CCAssignFnForReturn(), Info.OrigRet.Ty,
659 Call);
660 setLocInfo(ArgLocs, Ins);
661
662 CallReturnHandler Handler(MIRBuilder, MF.getRegInfo(), MIB);
663 if (!Handler.handle(ArgLocs, ArgInfos))
664 return false;
665 }
666
667 MIRBuilder.buildInstr(Mips::ADJCALLSTACKUP).addImm(NextStackOffset).addImm(0);
668
669 return true;
670 }
671
672 template <typename T>
subTargetRegTypeForCallingConv(const Function & F,ArrayRef<ArgInfo> Args,ArrayRef<unsigned> OrigArgIndices,SmallVectorImpl<T> & ISDArgs) const673 void MipsCallLowering::subTargetRegTypeForCallingConv(
674 const Function &F, ArrayRef<ArgInfo> Args,
675 ArrayRef<unsigned> OrigArgIndices, SmallVectorImpl<T> &ISDArgs) const {
676 const DataLayout &DL = F.getParent()->getDataLayout();
677 const MipsTargetLowering &TLI = *getTLI<MipsTargetLowering>();
678
679 unsigned ArgNo = 0;
680 for (auto &Arg : Args) {
681
682 EVT VT = TLI.getValueType(DL, Arg.Ty);
683 MVT RegisterVT = TLI.getRegisterTypeForCallingConv(F.getContext(),
684 F.getCallingConv(), VT);
685 unsigned NumRegs = TLI.getNumRegistersForCallingConv(
686 F.getContext(), F.getCallingConv(), VT);
687
688 for (unsigned i = 0; i < NumRegs; ++i) {
689 ISD::ArgFlagsTy Flags = Arg.Flags[0];
690
691 if (i == 0)
692 Flags.setOrigAlign(TLI.getABIAlignmentForCallingConv(Arg.Ty, DL));
693 else
694 Flags.setOrigAlign(Align::None());
695
696 ISDArgs.emplace_back(Flags, RegisterVT, VT, true, OrigArgIndices[ArgNo],
697 0);
698 }
699 ++ArgNo;
700 }
701 }
702
splitToValueTypes(const DataLayout & DL,const ArgInfo & OrigArg,unsigned OriginalIndex,SmallVectorImpl<ArgInfo> & SplitArgs,SmallVectorImpl<unsigned> & SplitArgsOrigIndices) const703 void MipsCallLowering::splitToValueTypes(
704 const DataLayout &DL, const ArgInfo &OrigArg, unsigned OriginalIndex,
705 SmallVectorImpl<ArgInfo> &SplitArgs,
706 SmallVectorImpl<unsigned> &SplitArgsOrigIndices) const {
707
708 SmallVector<EVT, 4> SplitEVTs;
709 SmallVector<Register, 4> SplitVRegs;
710 const MipsTargetLowering &TLI = *getTLI<MipsTargetLowering>();
711 LLVMContext &Ctx = OrigArg.Ty->getContext();
712
713 ComputeValueVTs(TLI, DL, OrigArg.Ty, SplitEVTs);
714
715 for (unsigned i = 0; i < SplitEVTs.size(); ++i) {
716 ArgInfo Info = ArgInfo{OrigArg.Regs[i], SplitEVTs[i].getTypeForEVT(Ctx)};
717 Info.Flags = OrigArg.Flags;
718 SplitArgs.push_back(Info);
719 SplitArgsOrigIndices.push_back(OriginalIndex);
720 }
721 }
722