blob: ebd374eff8c6f07fd3c98a36314ca90357288656 [file] [log] [blame]
Nick Lewycky5fa40c32013-10-01 21:51:38 +00001//===--- CGCall.cpp - Encapsulate calling convention details --------------===//
Daniel Dunbar3d7c90b2008-09-08 21:33:45 +00002//
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
Daniel Dunbar3d7c90b2008-09-08 21:33:45 +00006//
7//===----------------------------------------------------------------------===//
8//
9// These classes wrap the information about a call or function
10// definition used to handle ABI compliancy.
11//
12//===----------------------------------------------------------------------===//
13
14#include "CGCall.h"
Chris Lattnere70a0072010-06-29 16:40:28 +000015#include "ABIInfo.h"
Akira Hatanaka9d8ac612016-02-17 21:09:50 +000016#include "CGBlocks.h"
Chandler Carruth3a022472012-12-04 09:13:33 +000017#include "CGCXXABI.h"
David Majnemer4e52d6f2015-12-12 05:39:21 +000018#include "CGCleanup.h"
Daniel Dunbar3d7c90b2008-09-08 21:33:45 +000019#include "CodeGenFunction.h"
Daniel Dunbarc68897d2008-09-10 00:41:16 +000020#include "CodeGenModule.h"
John McCalla729c622012-02-17 03:33:10 +000021#include "TargetInfo.h"
Daniel Dunbar3d7c90b2008-09-08 21:33:45 +000022#include "clang/AST/Decl.h"
Anders Carlssonb15b55c2009-04-03 22:48:58 +000023#include "clang/AST/DeclCXX.h"
Daniel Dunbar3d7c90b2008-09-08 21:33:45 +000024#include "clang/AST/DeclObjC.h"
Richard Trieu63688182018-12-11 03:18:39 +000025#include "clang/Basic/CodeGenOptions.h"
Eric Christopher15709992015-10-15 23:47:11 +000026#include "clang/Basic/TargetBuiltins.h"
Chandler Carruth3a022472012-12-04 09:13:33 +000027#include "clang/Basic/TargetInfo.h"
Mark Laceya8e7df32013-10-30 21:53:58 +000028#include "clang/CodeGen/CGFunctionInfo.h"
John McCall12f23522016-04-04 18:33:08 +000029#include "clang/CodeGen/SwiftCallingConv.h"
Bill Wendling706469b2013-02-28 22:49:57 +000030#include "llvm/ADT/StringExtras.h"
David Blaikie181a6132018-06-04 21:23:29 +000031#include "llvm/Transforms/Utils/Local.h"
Nick Lewyckyd9bce502016-09-20 15:49:58 +000032#include "llvm/Analysis/ValueTracking.h"
Chandler Carruthffd55512013-01-02 11:45:17 +000033#include "llvm/IR/Attributes.h"
David Blaikief47ca252018-03-21 22:34:27 +000034#include "llvm/IR/CallingConv.h"
Chandler Carruthffd55512013-01-02 11:45:17 +000035#include "llvm/IR/DataLayout.h"
36#include "llvm/IR/InlineAsm.h"
Saleem Abdulrasool10a49722016-04-08 16:52:00 +000037#include "llvm/IR/IntrinsicInst.h"
David Blaikief47ca252018-03-21 22:34:27 +000038#include "llvm/IR/Intrinsics.h"
Daniel Dunbar3d7c90b2008-09-08 21:33:45 +000039using namespace clang;
40using namespace CodeGen;
41
42/***/
43
Nikolay Haustov8c6538b2016-06-30 09:06:33 +000044unsigned CodeGenTypes::ClangCallConvToLLVMCallConv(CallingConv CC) {
John McCallab26cfa2010-02-05 21:31:56 +000045 switch (CC) {
46 default: return llvm::CallingConv::C;
47 case CC_X86StdCall: return llvm::CallingConv::X86_StdCall;
48 case CC_X86FastCall: return llvm::CallingConv::X86_FastCall;
Erich Keane757d3172016-11-02 18:29:35 +000049 case CC_X86RegCall: return llvm::CallingConv::X86_RegCall;
Douglas Gregora941dca2010-05-18 16:57:00 +000050 case CC_X86ThisCall: return llvm::CallingConv::X86_ThisCall;
Martin Storsjo022e7822017-07-17 20:49:45 +000051 case CC_Win64: return llvm::CallingConv::Win64;
Charles Davisb5a214e2013-08-30 04:39:01 +000052 case CC_X86_64SysV: return llvm::CallingConv::X86_64_SysV;
Anton Korobeynikov231e8752011-04-14 20:06:49 +000053 case CC_AAPCS: return llvm::CallingConv::ARM_AAPCS;
54 case CC_AAPCS_VFP: return llvm::CallingConv::ARM_AAPCS_VFP;
Guy Benyeif0a014b2012-12-25 08:53:55 +000055 case CC_IntelOclBicc: return llvm::CallingConv::Intel_OCL_BI;
Reid Klecknerd7857f02014-10-24 17:42:17 +000056 // TODO: Add support for __pascal to LLVM.
57 case CC_X86Pascal: return llvm::CallingConv::C;
58 // TODO: Add support for __vectorcall to LLVM.
Reid Kleckner80944df2014-10-31 22:00:51 +000059 case CC_X86VectorCall: return llvm::CallingConv::X86_VectorCall;
Sander de Smalen44a22532018-11-26 16:38:37 +000060 case CC_AArch64VectorCall: return llvm::CallingConv::AArch64_VectorCall;
Alexander Kornienko21de0ae2015-01-20 11:20:41 +000061 case CC_SpirFunction: return llvm::CallingConv::SPIR_FUNC;
Nikolay Haustov8c6538b2016-06-30 09:06:33 +000062 case CC_OpenCLKernel: return CGM.getTargetCodeGenInfo().getOpenCLKernelCallingConv();
Roman Levenstein35aa5ce2016-03-16 18:00:46 +000063 case CC_PreserveMost: return llvm::CallingConv::PreserveMost;
64 case CC_PreserveAll: return llvm::CallingConv::PreserveAll;
John McCall12f23522016-04-04 18:33:08 +000065 case CC_Swift: return llvm::CallingConv::Swift;
John McCallab26cfa2010-02-05 21:31:56 +000066 }
67}
68
Mikael Nilsson9d2872d2018-12-13 10:15:27 +000069/// Derives the 'this' type for codegen purposes, i.e. ignoring method CVR
John McCall8ee376f2010-02-24 07:14:12 +000070/// qualification.
Mikael Nilsson9d2872d2018-12-13 10:15:27 +000071static CanQualType GetThisType(ASTContext &Context, const CXXRecordDecl *RD,
72 const CXXMethodDecl *MD) {
John McCall2da83a32010-02-26 00:48:12 +000073 QualType RecTy = Context.getTagDeclType(RD)->getCanonicalTypeInternal();
Mikael Nilsson9d2872d2018-12-13 10:15:27 +000074 if (MD)
Anastasia Stulovac61eaa52019-01-28 11:37:49 +000075 RecTy = Context.getAddrSpaceQualType(RecTy, MD->getMethodQualifiers().getAddressSpace());
John McCall2da83a32010-02-26 00:48:12 +000076 return Context.getPointerType(CanQualType::CreateUnsafe(RecTy));
Daniel Dunbar7a95ca32008-09-10 04:01:49 +000077}
78
John McCall8ee376f2010-02-24 07:14:12 +000079/// Returns the canonical formal type of the given C++ method.
John McCall2da83a32010-02-26 00:48:12 +000080static CanQual<FunctionProtoType> GetFormalType(const CXXMethodDecl *MD) {
81 return MD->getType()->getCanonicalTypeUnqualified()
82 .getAs<FunctionProtoType>();
John McCall8ee376f2010-02-24 07:14:12 +000083}
84
85/// Returns the "extra-canonicalized" return type, which discards
86/// qualifiers on the return type. Codegen doesn't care about them,
87/// and it makes ABI code a little easier to be able to assume that
88/// all parameter and return types are top-level unqualified.
John McCall2da83a32010-02-26 00:48:12 +000089static CanQualType GetReturnType(QualType RetTy) {
90 return RetTy->getCanonicalTypeUnqualified().getUnqualifiedType();
John McCall8ee376f2010-02-24 07:14:12 +000091}
92
John McCall8dda7b22012-07-07 06:41:13 +000093/// Arrange the argument and result information for a value of the given
94/// unprototyped freestanding function type.
John McCall8ee376f2010-02-24 07:14:12 +000095const CGFunctionInfo &
John McCall8dda7b22012-07-07 06:41:13 +000096CodeGenTypes::arrangeFreeFunctionType(CanQual<FunctionNoProtoType> FTNP) {
John McCalla729c622012-02-17 03:33:10 +000097 // When translating an unprototyped function type, always use a
98 // variadic type.
Alp Toker314cc812014-01-25 16:55:45 +000099 return arrangeLLVMFunctionInfo(FTNP->getReturnType().getUnqualifiedType(),
Peter Collingbournef7706832014-12-12 23:41:25 +0000100 /*instanceMethod=*/false,
101 /*chainCall=*/false, None,
John McCallc56a8b32016-03-11 04:30:31 +0000102 FTNP->getExtInfo(), {}, RequiredArgs(0));
John McCall8ee376f2010-02-24 07:14:12 +0000103}
104
George Burgess IVb7760212017-02-24 02:49:47 +0000105static void addExtParameterInfosForCall(
106 llvm::SmallVectorImpl<FunctionProtoType::ExtParameterInfo> &paramInfos,
107 const FunctionProtoType *proto,
108 unsigned prefixArgs,
109 unsigned totalArgs) {
110 assert(proto->hasExtParameterInfos());
111 assert(paramInfos.size() <= prefixArgs);
112 assert(proto->getNumParams() + prefixArgs <= totalArgs);
113
114 paramInfos.reserve(totalArgs);
115
116 // Add default infos for any prefix args that don't already have infos.
117 paramInfos.resize(prefixArgs);
118
119 // Add infos for the prototype.
120 for (const auto &ParamInfo : proto->getExtParameterInfos()) {
121 paramInfos.push_back(ParamInfo);
122 // pass_object_size params have no parameter info.
123 if (ParamInfo.hasPassObjectSize())
124 paramInfos.emplace_back();
125 }
126
127 assert(paramInfos.size() <= totalArgs &&
128 "Did we forget to insert pass_object_size args?");
129 // Add default infos for the variadic and/or suffix arguments.
130 paramInfos.resize(totalArgs);
131}
132
Hiroshi Inouec5e54dd2017-07-03 08:49:44 +0000133/// Adds the formal parameters in FPT to the given prefix. If any parameter in
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000134/// FPT has pass_object_size attrs, then we'll add parameters for those, too.
135static void appendParameterTypes(const CodeGenTypes &CGT,
136 SmallVectorImpl<CanQualType> &prefix,
John McCallc56a8b32016-03-11 04:30:31 +0000137 SmallVectorImpl<FunctionProtoType::ExtParameterInfo> &paramInfos,
George Burgess IVb7760212017-02-24 02:49:47 +0000138 CanQual<FunctionProtoType> FPT) {
139 // Fast path: don't touch param info if we don't need to.
140 if (!FPT->hasExtParameterInfos()) {
141 assert(paramInfos.empty() &&
142 "We have paramInfos, but the prototype doesn't?");
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000143 prefix.append(FPT->param_type_begin(), FPT->param_type_end());
144 return;
145 }
146
George Burgess IVb7760212017-02-24 02:49:47 +0000147 unsigned PrefixSize = prefix.size();
148 // In the vast majority of cases, we'll have precisely FPT->getNumParams()
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000149 // parameters; the only thing that can change this is the presence of
150 // pass_object_size. So, we preallocate for the common case.
151 prefix.reserve(prefix.size() + FPT->getNumParams());
152
George Burgess IVb7760212017-02-24 02:49:47 +0000153 auto ExtInfos = FPT->getExtParameterInfos();
154 assert(ExtInfos.size() == FPT->getNumParams());
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000155 for (unsigned I = 0, E = FPT->getNumParams(); I != E; ++I) {
156 prefix.push_back(FPT->getParamType(I));
George Burgess IVb7760212017-02-24 02:49:47 +0000157 if (ExtInfos[I].hasPassObjectSize())
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000158 prefix.push_back(CGT.getContext().getSizeType());
159 }
George Burgess IVb7760212017-02-24 02:49:47 +0000160
161 addExtParameterInfosForCall(paramInfos, FPT.getTypePtr(), PrefixSize,
162 prefix.size());
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000163}
164
John McCall8dda7b22012-07-07 06:41:13 +0000165/// Arrange the LLVM function layout for a value of the given function
Alexey Samsonove5ef3ca2014-08-13 23:55:54 +0000166/// type, on top of any implicit parameters already stored.
167static const CGFunctionInfo &
Peter Collingbournef7706832014-12-12 23:41:25 +0000168arrangeLLVMFunctionInfo(CodeGenTypes &CGT, bool instanceMethod,
Alexey Samsonove5ef3ca2014-08-13 23:55:54 +0000169 SmallVectorImpl<CanQualType> &prefix,
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000170 CanQual<FunctionProtoType> FTP,
171 const FunctionDecl *FD) {
John McCallc56a8b32016-03-11 04:30:31 +0000172 SmallVector<FunctionProtoType::ExtParameterInfo, 16> paramInfos;
George Burgess IV419996c2016-06-16 23:06:04 +0000173 RequiredArgs Required =
174 RequiredArgs::forPrototypePlus(FTP, prefix.size(), FD);
Daniel Dunbarbf8c24a2009-02-02 23:23:47 +0000175 // FIXME: Kill copy.
George Burgess IVb7760212017-02-24 02:49:47 +0000176 appendParameterTypes(CGT, prefix, paramInfos, FTP);
Alp Toker314cc812014-01-25 16:55:45 +0000177 CanQualType resultType = FTP->getReturnType().getUnqualifiedType();
John McCallc56a8b32016-03-11 04:30:31 +0000178
Peter Collingbournef7706832014-12-12 23:41:25 +0000179 return CGT.arrangeLLVMFunctionInfo(resultType, instanceMethod,
180 /*chainCall=*/false, prefix,
John McCallc56a8b32016-03-11 04:30:31 +0000181 FTP->getExtInfo(), paramInfos,
George Burgess IV419996c2016-06-16 23:06:04 +0000182 Required);
John McCall8ee376f2010-02-24 07:14:12 +0000183}
184
John McCalla729c622012-02-17 03:33:10 +0000185/// Arrange the argument and result information for a value of the
John McCall8dda7b22012-07-07 06:41:13 +0000186/// given freestanding function type.
John McCall8ee376f2010-02-24 07:14:12 +0000187const CGFunctionInfo &
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000188CodeGenTypes::arrangeFreeFunctionType(CanQual<FunctionProtoType> FTP,
189 const FunctionDecl *FD) {
John McCalla729c622012-02-17 03:33:10 +0000190 SmallVector<CanQualType, 16> argTypes;
Peter Collingbournef7706832014-12-12 23:41:25 +0000191 return ::arrangeLLVMFunctionInfo(*this, /*instanceMethod=*/false, argTypes,
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000192 FTP, FD);
Daniel Dunbar7feafc72009-09-11 22:24:53 +0000193}
194
Aaron Ballman0362a6d2013-12-18 16:23:37 +0000195static CallingConv getCallingConventionForDecl(const Decl *D, bool IsWindows) {
Daniel Dunbar7feafc72009-09-11 22:24:53 +0000196 // Set the appropriate calling convention for the Function.
197 if (D->hasAttr<StdCallAttr>())
John McCallab26cfa2010-02-05 21:31:56 +0000198 return CC_X86StdCall;
Daniel Dunbar7feafc72009-09-11 22:24:53 +0000199
200 if (D->hasAttr<FastCallAttr>())
John McCallab26cfa2010-02-05 21:31:56 +0000201 return CC_X86FastCall;
Daniel Dunbar7feafc72009-09-11 22:24:53 +0000202
Erich Keane757d3172016-11-02 18:29:35 +0000203 if (D->hasAttr<RegCallAttr>())
204 return CC_X86RegCall;
205
Douglas Gregora941dca2010-05-18 16:57:00 +0000206 if (D->hasAttr<ThisCallAttr>())
207 return CC_X86ThisCall;
208
Reid Klecknerd7857f02014-10-24 17:42:17 +0000209 if (D->hasAttr<VectorCallAttr>())
210 return CC_X86VectorCall;
211
Dawn Perchik335e16b2010-09-03 01:29:35 +0000212 if (D->hasAttr<PascalAttr>())
213 return CC_X86Pascal;
214
Anton Korobeynikov231e8752011-04-14 20:06:49 +0000215 if (PcsAttr *PCS = D->getAttr<PcsAttr>())
216 return (PCS->getPCS() == PcsAttr::AAPCS ? CC_AAPCS : CC_AAPCS_VFP);
217
Sander de Smalen44a22532018-11-26 16:38:37 +0000218 if (D->hasAttr<AArch64VectorPcsAttr>())
219 return CC_AArch64VectorCall;
220
Guy Benyeif0a014b2012-12-25 08:53:55 +0000221 if (D->hasAttr<IntelOclBiccAttr>())
222 return CC_IntelOclBicc;
223
Aaron Ballman0362a6d2013-12-18 16:23:37 +0000224 if (D->hasAttr<MSABIAttr>())
Martin Storsjo022e7822017-07-17 20:49:45 +0000225 return IsWindows ? CC_C : CC_Win64;
Aaron Ballman0362a6d2013-12-18 16:23:37 +0000226
227 if (D->hasAttr<SysVABIAttr>())
228 return IsWindows ? CC_X86_64SysV : CC_C;
229
Roman Levenstein35aa5ce2016-03-16 18:00:46 +0000230 if (D->hasAttr<PreserveMostAttr>())
231 return CC_PreserveMost;
232
233 if (D->hasAttr<PreserveAllAttr>())
234 return CC_PreserveAll;
235
John McCallab26cfa2010-02-05 21:31:56 +0000236 return CC_C;
Daniel Dunbar7a95ca32008-09-10 04:01:49 +0000237}
238
John McCalla729c622012-02-17 03:33:10 +0000239/// Arrange the argument and result information for a call to an
240/// unknown C++ non-static member function of the given abstract type.
Timur Iskhodzhanov88fd4392013-08-21 06:25:03 +0000241/// (Zero value of RD means we don't have any meaningful "this" argument type,
242/// so fall back to a generic pointer type).
John McCalla729c622012-02-17 03:33:10 +0000243/// The member function must be an ordinary function, i.e. not a
244/// constructor or destructor.
245const CGFunctionInfo &
246CodeGenTypes::arrangeCXXMethodType(const CXXRecordDecl *RD,
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000247 const FunctionProtoType *FTP,
248 const CXXMethodDecl *MD) {
John McCalla729c622012-02-17 03:33:10 +0000249 SmallVector<CanQualType, 16> argTypes;
John McCall8ee376f2010-02-24 07:14:12 +0000250
Anders Carlsson2ee3c012009-10-03 19:43:08 +0000251 // Add the 'this' pointer.
Timur Iskhodzhanov88fd4392013-08-21 06:25:03 +0000252 if (RD)
Mikael Nilsson9d2872d2018-12-13 10:15:27 +0000253 argTypes.push_back(GetThisType(Context, RD, MD));
Timur Iskhodzhanov88fd4392013-08-21 06:25:03 +0000254 else
255 argTypes.push_back(Context.VoidPtrTy);
John McCall8ee376f2010-02-24 07:14:12 +0000256
Alexey Samsonove5ef3ca2014-08-13 23:55:54 +0000257 return ::arrangeLLVMFunctionInfo(
258 *this, true, argTypes,
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000259 FTP->getCanonicalTypeUnqualified().getAs<FunctionProtoType>(), MD);
Anders Carlsson2ee3c012009-10-03 19:43:08 +0000260}
261
Yaxun Liu6c10a662018-06-12 00:16:33 +0000262/// Set calling convention for CUDA/HIP kernel.
263static void setCUDAKernelCallingConvention(CanQualType &FTy, CodeGenModule &CGM,
264 const FunctionDecl *FD) {
265 if (FD->hasAttr<CUDAGlobalAttr>()) {
266 const FunctionType *FT = FTy->getAs<FunctionType>();
267 CGM.getTargetCodeGenInfo().setCUDAKernelCallingConvention(FT);
268 FTy = FT->getCanonicalTypeUnqualified();
269 }
270}
271
John McCalla729c622012-02-17 03:33:10 +0000272/// Arrange the argument and result information for a declaration or
273/// definition of the given C++ non-static member function. The
274/// member function must be an ordinary function, i.e. not a
275/// constructor or destructor.
276const CGFunctionInfo &
277CodeGenTypes::arrangeCXXMethodDeclaration(const CXXMethodDecl *MD) {
Benjamin Kramer60509af2013-09-09 14:48:42 +0000278 assert(!isa<CXXConstructorDecl>(MD) && "wrong method for constructors!");
John McCall0d635f52010-09-03 01:26:39 +0000279 assert(!isa<CXXDestructorDecl>(MD) && "wrong method for destructors!");
280
Yaxun Liu6c10a662018-06-12 00:16:33 +0000281 CanQualType FT = GetFormalType(MD).getAs<Type>();
282 setCUDAKernelCallingConvention(FT, CGM, MD);
283 auto prototype = FT.getAs<FunctionProtoType>();
Mike Stump11289f42009-09-09 15:08:12 +0000284
John McCalla729c622012-02-17 03:33:10 +0000285 if (MD->isInstance()) {
286 // The abstract case is perfectly fine.
Mark Lacey5ea993b2013-10-02 20:35:23 +0000287 const CXXRecordDecl *ThisType = TheCXXABI.getThisArgumentTypeForMethod(MD);
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000288 return arrangeCXXMethodType(ThisType, prototype.getTypePtr(), MD);
John McCalla729c622012-02-17 03:33:10 +0000289 }
290
George Burgess IV3e3bb95b2015-12-02 21:58:08 +0000291 return arrangeFreeFunctionType(prototype, MD);
Anders Carlssonb15b55c2009-04-03 22:48:58 +0000292}
293
Richard Smith5179eb72016-06-28 19:03:57 +0000294bool CodeGenTypes::inheritingCtorHasParams(
295 const InheritedConstructor &Inherited, CXXCtorType Type) {
296 // Parameters are unnecessary if we're constructing a base class subobject
297 // and the inherited constructor lives in a virtual base.
298 return Type == Ctor_Complete ||
299 !Inherited.getShadowDecl()->constructsVirtualBase() ||
300 !Target.getCXXABI().hasConstructorVariants();
301 }
302
John McCalla729c622012-02-17 03:33:10 +0000303const CGFunctionInfo &
Rafael Espindola8d2a19b2014-09-08 16:01:27 +0000304CodeGenTypes::arrangeCXXStructorDeclaration(const CXXMethodDecl *MD,
305 StructorType Type) {
306
John McCalla729c622012-02-17 03:33:10 +0000307 SmallVector<CanQualType, 16> argTypes;
John McCallc56a8b32016-03-11 04:30:31 +0000308 SmallVector<FunctionProtoType::ExtParameterInfo, 16> paramInfos;
Mikael Nilsson9d2872d2018-12-13 10:15:27 +0000309 argTypes.push_back(GetThisType(Context, MD->getParent(), MD));
Stephen Lin9dc6eef2013-06-30 20:40:16 +0000310
Richard Smith5179eb72016-06-28 19:03:57 +0000311 bool PassParams = true;
312
Rafael Espindola8d2a19b2014-09-08 16:01:27 +0000313 GlobalDecl GD;
314 if (auto *CD = dyn_cast<CXXConstructorDecl>(MD)) {
315 GD = GlobalDecl(CD, toCXXCtorType(Type));
Richard Smith5179eb72016-06-28 19:03:57 +0000316
317 // A base class inheriting constructor doesn't get forwarded arguments
318 // needed to construct a virtual base (or base class thereof).
319 if (auto Inherited = CD->getInheritedConstructor())
320 PassParams = inheritingCtorHasParams(Inherited, toCXXCtorType(Type));
Rafael Espindola8d2a19b2014-09-08 16:01:27 +0000321 } else {
322 auto *DD = dyn_cast<CXXDestructorDecl>(MD);
323 GD = GlobalDecl(DD, toCXXDtorType(Type));
324 }
Anders Carlsson82ba57c2009-11-25 03:15:49 +0000325
Rafael Espindola8d2a19b2014-09-08 16:01:27 +0000326 CanQual<FunctionProtoType> FTP = GetFormalType(MD);
John McCall5d865c322010-08-31 07:33:07 +0000327
328 // Add the formal parameters.
Richard Smith5179eb72016-06-28 19:03:57 +0000329 if (PassParams)
George Burgess IVb7760212017-02-24 02:49:47 +0000330 appendParameterTypes(*this, argTypes, paramInfos, FTP);
John McCall5d865c322010-08-31 07:33:07 +0000331
George Burgess IV75b34a92017-02-22 22:38:25 +0000332 CGCXXABI::AddedStructorArgs AddedArgs =
333 TheCXXABI.buildStructorSignature(MD, Type, argTypes);
334 if (!paramInfos.empty()) {
335 // Note: prefix implies after the first param.
336 if (AddedArgs.Prefix)
337 paramInfos.insert(paramInfos.begin() + 1, AddedArgs.Prefix,
338 FunctionProtoType::ExtParameterInfo{});
339 if (AddedArgs.Suffix)
340 paramInfos.append(AddedArgs.Suffix,
341 FunctionProtoType::ExtParameterInfo{});
342 }
Reid Kleckner89077a12013-12-17 19:46:40 +0000343
344 RequiredArgs required =
Richard Smith5179eb72016-06-28 19:03:57 +0000345 (PassParams && MD->isVariadic() ? RequiredArgs(argTypes.size())
346 : RequiredArgs::All);
Reid Kleckner89077a12013-12-17 19:46:40 +0000347
John McCall8dda7b22012-07-07 06:41:13 +0000348 FunctionType::ExtInfo extInfo = FTP->getExtInfo();
David Majnemer0c0b6d92014-10-31 20:09:12 +0000349 CanQualType resultType = TheCXXABI.HasThisReturn(GD)
350 ? argTypes.front()
351 : TheCXXABI.hasMostDerivedReturn(GD)
352 ? CGM.getContext().VoidPtrTy
353 : Context.VoidTy;
Peter Collingbournef7706832014-12-12 23:41:25 +0000354 return arrangeLLVMFunctionInfo(resultType, /*instanceMethod=*/true,
355 /*chainCall=*/false, argTypes, extInfo,
John McCallc56a8b32016-03-11 04:30:31 +0000356 paramInfos, required);
357}
358
359static SmallVector<CanQualType, 16>
360getArgTypesForCall(ASTContext &ctx, const CallArgList &args) {
361 SmallVector<CanQualType, 16> argTypes;
362 for (auto &arg : args)
363 argTypes.push_back(ctx.getCanonicalParamType(arg.Ty));
364 return argTypes;
365}
366
367static SmallVector<CanQualType, 16>
368getArgTypesForDeclaration(ASTContext &ctx, const FunctionArgList &args) {
369 SmallVector<CanQualType, 16> argTypes;
370 for (auto &arg : args)
371 argTypes.push_back(ctx.getCanonicalParamType(arg->getType()));
372 return argTypes;
373}
374
John McCallc56a8b32016-03-11 04:30:31 +0000375static llvm::SmallVector<FunctionProtoType::ExtParameterInfo, 16>
376getExtParameterInfosForCall(const FunctionProtoType *proto,
377 unsigned prefixArgs, unsigned totalArgs) {
378 llvm::SmallVector<FunctionProtoType::ExtParameterInfo, 16> result;
379 if (proto->hasExtParameterInfos()) {
380 addExtParameterInfosForCall(result, proto, prefixArgs, totalArgs);
381 }
382 return result;
Anders Carlsson82ba57c2009-11-25 03:15:49 +0000383}
384
Reid Kleckner314ef7b2014-02-01 00:04:45 +0000385/// Arrange a call to a C++ method, passing the given arguments.
George Burgess IVd0a9e802017-02-23 22:07:35 +0000386///
387/// ExtraPrefixArgs is the number of ABI-specific args passed after the `this`
388/// parameter.
389/// ExtraSuffixArgs is the number of ABI-specific args passed at the end of
390/// args.
391/// PassProtoArgs indicates whether `args` has args for the parameters in the
392/// given CXXConstructorDecl.
Reid Kleckner314ef7b2014-02-01 00:04:45 +0000393const CGFunctionInfo &
394CodeGenTypes::arrangeCXXConstructorCall(const CallArgList &args,
395 const CXXConstructorDecl *D,
396 CXXCtorType CtorKind,
George Burgess IVd0a9e802017-02-23 22:07:35 +0000397 unsigned ExtraPrefixArgs,
398 unsigned ExtraSuffixArgs,
399 bool PassProtoArgs) {
Reid Kleckner314ef7b2014-02-01 00:04:45 +0000400 // FIXME: Kill copy.
401 SmallVector<CanQualType, 16> ArgTypes;
Alexey Samsonov3551e312014-08-13 20:06:24 +0000402 for (const auto &Arg : args)
403 ArgTypes.push_back(Context.getCanonicalParamType(Arg.Ty));
Reid Kleckner314ef7b2014-02-01 00:04:45 +0000404
George Burgess IVd0a9e802017-02-23 22:07:35 +0000405 // +1 for implicit this, which should always be args[0].
406 unsigned TotalPrefixArgs = 1 + ExtraPrefixArgs;
407
Reid Kleckner314ef7b2014-02-01 00:04:45 +0000408 CanQual<FunctionProtoType> FPT = GetFormalType(D);
George Burgess IVd0a9e802017-02-23 22:07:35 +0000409 RequiredArgs Required =
410 RequiredArgs::forPrototypePlus(FPT, TotalPrefixArgs + ExtraSuffixArgs, D);
Reid Kleckner314ef7b2014-02-01 00:04:45 +0000411 GlobalDecl GD(D, CtorKind);
David Majnemer0c0b6d92014-10-31 20:09:12 +0000412 CanQualType ResultType = TheCXXABI.HasThisReturn(GD)
413 ? ArgTypes.front()
414 : TheCXXABI.hasMostDerivedReturn(GD)
415 ? CGM.getContext().VoidPtrTy
416 : Context.VoidTy;
Reid Kleckner314ef7b2014-02-01 00:04:45 +0000417
418 FunctionType::ExtInfo Info = FPT->getExtInfo();
George Burgess IVd0a9e802017-02-23 22:07:35 +0000419 llvm::SmallVector<FunctionProtoType::ExtParameterInfo, 16> ParamInfos;
420 // If the prototype args are elided, we should only have ABI-specific args,
421 // which never have param info.
422 if (PassProtoArgs && FPT->hasExtParameterInfos()) {
423 // ABI-specific suffix arguments are treated the same as variadic arguments.
424 addExtParameterInfosForCall(ParamInfos, FPT.getTypePtr(), TotalPrefixArgs,
425 ArgTypes.size());
426 }
Peter Collingbournef7706832014-12-12 23:41:25 +0000427 return arrangeLLVMFunctionInfo(ResultType, /*instanceMethod=*/true,
428 /*chainCall=*/false, ArgTypes, Info,
John McCallc56a8b32016-03-11 04:30:31 +0000429 ParamInfos, Required);
Reid Kleckner314ef7b2014-02-01 00:04:45 +0000430}
431
John McCalla729c622012-02-17 03:33:10 +0000432/// Arrange the argument and result information for the declaration or
433/// definition of the given function.
434const CGFunctionInfo &
435CodeGenTypes::arrangeFunctionDeclaration(const FunctionDecl *FD) {
Chris Lattnerbea5b622009-05-12 20:27:19 +0000436 if (const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(FD))
Anders Carlssonb15b55c2009-04-03 22:48:58 +0000437 if (MD->isInstance())
John McCalla729c622012-02-17 03:33:10 +0000438 return arrangeCXXMethodDeclaration(MD);
Mike Stump11289f42009-09-09 15:08:12 +0000439
John McCall2da83a32010-02-26 00:48:12 +0000440 CanQualType FTy = FD->getType()->getCanonicalTypeUnqualified();
John McCalla729c622012-02-17 03:33:10 +0000441
John McCall2da83a32010-02-26 00:48:12 +0000442 assert(isa<FunctionType>(FTy));
Yaxun Liu6c10a662018-06-12 00:16:33 +0000443 setCUDAKernelCallingConvention(FTy, CGM, FD);
John McCalla729c622012-02-17 03:33:10 +0000444
445 // When declaring a function without a prototype, always use a
446 // non-variadic type.
George Burgess IV35cfca22017-01-06 19:10:48 +0000447 if (CanQual<FunctionNoProtoType> noProto = FTy.getAs<FunctionNoProtoType>()) {
Peter Collingbournef7706832014-12-12 23:41:25 +0000448 return arrangeLLVMFunctionInfo(
449 noProto->getReturnType(), /*instanceMethod=*/false,
John McCallc56a8b32016-03-11 04:30:31 +0000450 /*chainCall=*/false, None, noProto->getExtInfo(), {},RequiredArgs::All);
John McCalla729c622012-02-17 03:33:10 +0000451 }
452
George Burgess IV35cfca22017-01-06 19:10:48 +0000453 return arrangeFreeFunctionType(FTy.castAs<FunctionProtoType>(), FD);
Daniel Dunbar3d7c90b2008-09-08 21:33:45 +0000454}
455
John McCalla729c622012-02-17 03:33:10 +0000456/// Arrange the argument and result information for the declaration or
457/// definition of an Objective-C method.
458const CGFunctionInfo &
459CodeGenTypes::arrangeObjCMethodDeclaration(const ObjCMethodDecl *MD) {
460 // It happens that this is the same as a call with no optional
461 // arguments, except also using the formal 'self' type.
462 return arrangeObjCMessageSendSignature(MD, MD->getSelfDecl()->getType());
463}
464
465/// Arrange the argument and result information for the function type
466/// through which to perform a send to the given Objective-C method,
467/// using the given receiver type. The receiver type is not always
468/// the 'self' type of the method or even an Objective-C pointer type.
469/// This is *not* the right method for actually performing such a
470/// message send, due to the possibility of optional arguments.
471const CGFunctionInfo &
472CodeGenTypes::arrangeObjCMessageSendSignature(const ObjCMethodDecl *MD,
473 QualType receiverType) {
474 SmallVector<CanQualType, 16> argTys;
Akira Hatanaka98a49332017-09-22 00:41:05 +0000475 SmallVector<FunctionProtoType::ExtParameterInfo, 4> extParamInfos(2);
John McCalla729c622012-02-17 03:33:10 +0000476 argTys.push_back(Context.getCanonicalParamType(receiverType));
477 argTys.push_back(Context.getCanonicalParamType(Context.getObjCSelType()));
Daniel Dunbarbf8c24a2009-02-02 23:23:47 +0000478 // FIXME: Kill copy?
David Majnemer59f77922016-06-24 04:05:48 +0000479 for (const auto *I : MD->parameters()) {
Aaron Ballman43b68be2014-03-07 17:50:17 +0000480 argTys.push_back(Context.getCanonicalParamType(I->getType()));
Akira Hatanaka98a49332017-09-22 00:41:05 +0000481 auto extParamInfo = FunctionProtoType::ExtParameterInfo().withIsNoEscape(
482 I->hasAttr<NoEscapeAttr>());
483 extParamInfos.push_back(extParamInfo);
John McCall8ee376f2010-02-24 07:14:12 +0000484 }
John McCall31168b02011-06-15 23:02:42 +0000485
486 FunctionType::ExtInfo einfo;
Aaron Ballman0362a6d2013-12-18 16:23:37 +0000487 bool IsWindows = getContext().getTargetInfo().getTriple().isOSWindows();
488 einfo = einfo.withCallingConv(getCallingConventionForDecl(MD, IsWindows));
John McCall31168b02011-06-15 23:02:42 +0000489
David Blaikiebbafb8a2012-03-11 07:00:24 +0000490 if (getContext().getLangOpts().ObjCAutoRefCount &&
John McCall31168b02011-06-15 23:02:42 +0000491 MD->hasAttr<NSReturnsRetainedAttr>())
492 einfo = einfo.withProducesResult(true);
493
John McCalla729c622012-02-17 03:33:10 +0000494 RequiredArgs required =
495 (MD->isVariadic() ? RequiredArgs(argTys.size()) : RequiredArgs::All);
496
Peter Collingbournef7706832014-12-12 23:41:25 +0000497 return arrangeLLVMFunctionInfo(
498 GetReturnType(MD->getReturnType()), /*instanceMethod=*/false,
Akira Hatanaka98a49332017-09-22 00:41:05 +0000499 /*chainCall=*/false, argTys, einfo, extParamInfos, required);
John McCallc56a8b32016-03-11 04:30:31 +0000500}
501
502const CGFunctionInfo &
503CodeGenTypes::arrangeUnprototypedObjCMessageSend(QualType returnType,
504 const CallArgList &args) {
505 auto argTypes = getArgTypesForCall(Context, args);
506 FunctionType::ExtInfo einfo;
507
508 return arrangeLLVMFunctionInfo(
509 GetReturnType(returnType), /*instanceMethod=*/false,
510 /*chainCall=*/false, argTypes, einfo, {}, RequiredArgs::All);
Daniel Dunbar3d7c90b2008-09-08 21:33:45 +0000511}
512
John McCalla729c622012-02-17 03:33:10 +0000513const CGFunctionInfo &
514CodeGenTypes::arrangeGlobalDeclaration(GlobalDecl GD) {
Anders Carlsson6710c532010-02-06 02:44:09 +0000515 // FIXME: Do we need to handle ObjCMethodDecl?
516 const FunctionDecl *FD = cast<FunctionDecl>(GD.getDecl());
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +0000517
Anders Carlsson6710c532010-02-06 02:44:09 +0000518 if (const CXXConstructorDecl *CD = dyn_cast<CXXConstructorDecl>(FD))
Rafael Espindola8d2a19b2014-09-08 16:01:27 +0000519 return arrangeCXXStructorDeclaration(CD, getFromCtorType(GD.getCtorType()));
Anders Carlsson6710c532010-02-06 02:44:09 +0000520
521 if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(FD))
Rafael Espindola8d2a19b2014-09-08 16:01:27 +0000522 return arrangeCXXStructorDeclaration(DD, getFromDtorType(GD.getDtorType()));
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +0000523
John McCalla729c622012-02-17 03:33:10 +0000524 return arrangeFunctionDeclaration(FD);
Anders Carlsson6710c532010-02-06 02:44:09 +0000525}
526
Reid Klecknerc3473512014-08-29 21:43:29 +0000527/// Arrange a thunk that takes 'this' as the first parameter followed by
528/// varargs. Return a void pointer, regardless of the actual return type.
529/// The body of the thunk will end in a musttail call to a function of the
530/// correct type, and the caller will bitcast the function to the correct
531/// prototype.
532const CGFunctionInfo &
Reid Kleckner399d96e2018-04-02 20:20:33 +0000533CodeGenTypes::arrangeUnprototypedMustTailThunk(const CXXMethodDecl *MD) {
534 assert(MD->isVirtual() && "only methods have thunks");
Reid Klecknerc3473512014-08-29 21:43:29 +0000535 CanQual<FunctionProtoType> FTP = GetFormalType(MD);
Mikael Nilsson9d2872d2018-12-13 10:15:27 +0000536 CanQualType ArgTys[] = { GetThisType(Context, MD->getParent(), MD) };
Peter Collingbournef7706832014-12-12 23:41:25 +0000537 return arrangeLLVMFunctionInfo(Context.VoidTy, /*instanceMethod=*/false,
538 /*chainCall=*/false, ArgTys,
John McCallc56a8b32016-03-11 04:30:31 +0000539 FTP->getExtInfo(), {}, RequiredArgs(1));
Reid Klecknerc3473512014-08-29 21:43:29 +0000540}
541
David Majnemerdfa6d202015-03-11 18:36:39 +0000542const CGFunctionInfo &
David Majnemer37fd66e2015-03-13 22:36:55 +0000543CodeGenTypes::arrangeMSCtorClosure(const CXXConstructorDecl *CD,
544 CXXCtorType CT) {
545 assert(CT == Ctor_CopyingClosure || CT == Ctor_DefaultClosure);
546
David Majnemerdfa6d202015-03-11 18:36:39 +0000547 CanQual<FunctionProtoType> FTP = GetFormalType(CD);
548 SmallVector<CanQualType, 2> ArgTys;
549 const CXXRecordDecl *RD = CD->getParent();
Mikael Nilsson9d2872d2018-12-13 10:15:27 +0000550 ArgTys.push_back(GetThisType(Context, RD, CD));
David Majnemer37fd66e2015-03-13 22:36:55 +0000551 if (CT == Ctor_CopyingClosure)
552 ArgTys.push_back(*FTP->param_type_begin());
David Majnemerdfa6d202015-03-11 18:36:39 +0000553 if (RD->getNumVBases() > 0)
554 ArgTys.push_back(Context.IntTy);
555 CallingConv CC = Context.getDefaultCallingConvention(
556 /*IsVariadic=*/false, /*IsCXXMethod=*/true);
557 return arrangeLLVMFunctionInfo(Context.VoidTy, /*instanceMethod=*/true,
558 /*chainCall=*/false, ArgTys,
John McCallc56a8b32016-03-11 04:30:31 +0000559 FunctionType::ExtInfo(CC), {},
560 RequiredArgs::All);
David Majnemerdfa6d202015-03-11 18:36:39 +0000561}
562
John McCallc818bbb2012-12-07 07:03:17 +0000563/// Arrange a call as unto a free function, except possibly with an
564/// additional number of formal parameters considered required.
565static const CGFunctionInfo &
566arrangeFreeFunctionLikeCall(CodeGenTypes &CGT,
Mark Lacey23455752013-10-10 20:57:00 +0000567 CodeGenModule &CGM,
John McCallc818bbb2012-12-07 07:03:17 +0000568 const CallArgList &args,
569 const FunctionType *fnType,
Peter Collingbournef7706832014-12-12 23:41:25 +0000570 unsigned numExtraRequiredArgs,
571 bool chainCall) {
John McCallc818bbb2012-12-07 07:03:17 +0000572 assert(args.size() >= numExtraRequiredArgs);
573
John McCallc56a8b32016-03-11 04:30:31 +0000574 llvm::SmallVector<FunctionProtoType::ExtParameterInfo, 16> paramInfos;
575
John McCallc818bbb2012-12-07 07:03:17 +0000576 // In most cases, there are no optional arguments.
577 RequiredArgs required = RequiredArgs::All;
578
579 // If we have a variadic prototype, the required arguments are the
580 // extra prefix plus the arguments in the prototype.
581 if (const FunctionProtoType *proto = dyn_cast<FunctionProtoType>(fnType)) {
582 if (proto->isVariadic())
Alp Toker9cacbab2014-01-20 20:26:09 +0000583 required = RequiredArgs(proto->getNumParams() + numExtraRequiredArgs);
John McCallc818bbb2012-12-07 07:03:17 +0000584
John McCallc56a8b32016-03-11 04:30:31 +0000585 if (proto->hasExtParameterInfos())
586 addExtParameterInfosForCall(paramInfos, proto, numExtraRequiredArgs,
587 args.size());
588
John McCallc818bbb2012-12-07 07:03:17 +0000589 // If we don't have a prototype at all, but we're supposed to
590 // explicitly use the variadic convention for unprototyped calls,
591 // treat all of the arguments as required but preserve the nominal
592 // possibility of variadics.
Mark Lacey23455752013-10-10 20:57:00 +0000593 } else if (CGM.getTargetCodeGenInfo()
594 .isNoProtoCallVariadic(args,
595 cast<FunctionNoProtoType>(fnType))) {
John McCallc818bbb2012-12-07 07:03:17 +0000596 required = RequiredArgs(args.size());
597 }
598
Peter Collingbournef7706832014-12-12 23:41:25 +0000599 // FIXME: Kill copy.
600 SmallVector<CanQualType, 16> argTypes;
601 for (const auto &arg : args)
602 argTypes.push_back(CGT.getContext().getCanonicalParamType(arg.Ty));
603 return CGT.arrangeLLVMFunctionInfo(GetReturnType(fnType->getReturnType()),
604 /*instanceMethod=*/false, chainCall,
John McCallc56a8b32016-03-11 04:30:31 +0000605 argTypes, fnType->getExtInfo(), paramInfos,
606 required);
John McCallc818bbb2012-12-07 07:03:17 +0000607}
608
John McCalla729c622012-02-17 03:33:10 +0000609/// Figure out the rules for calling a function with the given formal
610/// type using the given arguments. The arguments are necessary
611/// because the function might be unprototyped, in which case it's
612/// target-dependent in crazy ways.
613const CGFunctionInfo &
John McCall8dda7b22012-07-07 06:41:13 +0000614CodeGenTypes::arrangeFreeFunctionCall(const CallArgList &args,
Peter Collingbournef7706832014-12-12 23:41:25 +0000615 const FunctionType *fnType,
616 bool chainCall) {
617 return arrangeFreeFunctionLikeCall(*this, CGM, args, fnType,
618 chainCall ? 1 : 0, chainCall);
John McCallc818bbb2012-12-07 07:03:17 +0000619}
John McCalla729c622012-02-17 03:33:10 +0000620
John McCallc56a8b32016-03-11 04:30:31 +0000621/// A block function is essentially a free function with an
John McCallc818bbb2012-12-07 07:03:17 +0000622/// extra implicit argument.
623const CGFunctionInfo &
624CodeGenTypes::arrangeBlockFunctionCall(const CallArgList &args,
625 const FunctionType *fnType) {
Peter Collingbournef7706832014-12-12 23:41:25 +0000626 return arrangeFreeFunctionLikeCall(*this, CGM, args, fnType, 1,
627 /*chainCall=*/false);
John McCalla729c622012-02-17 03:33:10 +0000628}
629
630const CGFunctionInfo &
John McCallc56a8b32016-03-11 04:30:31 +0000631CodeGenTypes::arrangeBlockFunctionDeclaration(const FunctionProtoType *proto,
632 const FunctionArgList &params) {
633 auto paramInfos = getExtParameterInfosForCall(proto, 1, params.size());
634 auto argTypes = getArgTypesForDeclaration(Context, params);
635
George Burgess IV419996c2016-06-16 23:06:04 +0000636 return arrangeLLVMFunctionInfo(
637 GetReturnType(proto->getReturnType()),
638 /*instanceMethod*/ false, /*chainCall*/ false, argTypes,
639 proto->getExtInfo(), paramInfos,
640 RequiredArgs::forPrototypePlus(proto, 1, nullptr));
John McCallc56a8b32016-03-11 04:30:31 +0000641}
642
643const CGFunctionInfo &
644CodeGenTypes::arrangeBuiltinFunctionCall(QualType resultType,
645 const CallArgList &args) {
Daniel Dunbarbf8c24a2009-02-02 23:23:47 +0000646 // FIXME: Kill copy.
John McCalla729c622012-02-17 03:33:10 +0000647 SmallVector<CanQualType, 16> argTypes;
Alexey Samsonov3551e312014-08-13 20:06:24 +0000648 for (const auto &Arg : args)
649 argTypes.push_back(Context.getCanonicalParamType(Arg.Ty));
Peter Collingbournef7706832014-12-12 23:41:25 +0000650 return arrangeLLVMFunctionInfo(
651 GetReturnType(resultType), /*instanceMethod=*/false,
John McCallc56a8b32016-03-11 04:30:31 +0000652 /*chainCall=*/false, argTypes, FunctionType::ExtInfo(),
653 /*paramInfos=*/ {}, RequiredArgs::All);
John McCall8dda7b22012-07-07 06:41:13 +0000654}
655
John McCallc56a8b32016-03-11 04:30:31 +0000656const CGFunctionInfo &
657CodeGenTypes::arrangeBuiltinFunctionDeclaration(QualType resultType,
658 const FunctionArgList &args) {
659 auto argTypes = getArgTypesForDeclaration(Context, args);
660
661 return arrangeLLVMFunctionInfo(
662 GetReturnType(resultType), /*instanceMethod=*/false, /*chainCall=*/false,
663 argTypes, FunctionType::ExtInfo(), {}, RequiredArgs::All);
664}
665
666const CGFunctionInfo &
667CodeGenTypes::arrangeBuiltinFunctionDeclaration(CanQualType resultType,
668 ArrayRef<CanQualType> argTypes) {
669 return arrangeLLVMFunctionInfo(
670 resultType, /*instanceMethod=*/false, /*chainCall=*/false,
671 argTypes, FunctionType::ExtInfo(), {}, RequiredArgs::All);
672}
673
John McCall8dda7b22012-07-07 06:41:13 +0000674/// Arrange a call to a C++ method, passing the given arguments.
George Burgess IVd0a9e802017-02-23 22:07:35 +0000675///
676/// numPrefixArgs is the number of ABI-specific prefix arguments we have. It
677/// does not count `this`.
John McCall8dda7b22012-07-07 06:41:13 +0000678const CGFunctionInfo &
679CodeGenTypes::arrangeCXXMethodCall(const CallArgList &args,
John McCallc56a8b32016-03-11 04:30:31 +0000680 const FunctionProtoType *proto,
George Burgess IVd0a9e802017-02-23 22:07:35 +0000681 RequiredArgs required,
682 unsigned numPrefixArgs) {
683 assert(numPrefixArgs + 1 <= args.size() &&
684 "Emitting a call with less args than the required prefix?");
685 // Add one to account for `this`. It's a bit awkward here, but we don't count
686 // `this` in similar places elsewhere.
John McCallc56a8b32016-03-11 04:30:31 +0000687 auto paramInfos =
George Burgess IVd0a9e802017-02-23 22:07:35 +0000688 getExtParameterInfosForCall(proto, numPrefixArgs + 1, args.size());
John McCallc56a8b32016-03-11 04:30:31 +0000689
John McCall8dda7b22012-07-07 06:41:13 +0000690 // FIXME: Kill copy.
John McCallc56a8b32016-03-11 04:30:31 +0000691 auto argTypes = getArgTypesForCall(Context, args);
John McCall8dda7b22012-07-07 06:41:13 +0000692
John McCallc56a8b32016-03-11 04:30:31 +0000693 FunctionType::ExtInfo info = proto->getExtInfo();
Peter Collingbournef7706832014-12-12 23:41:25 +0000694 return arrangeLLVMFunctionInfo(
John McCallc56a8b32016-03-11 04:30:31 +0000695 GetReturnType(proto->getReturnType()), /*instanceMethod=*/true,
696 /*chainCall=*/false, argTypes, info, paramInfos, required);
Daniel Dunbarbf8c24a2009-02-02 23:23:47 +0000697}
698
John McCalla729c622012-02-17 03:33:10 +0000699const CGFunctionInfo &CodeGenTypes::arrangeNullaryFunction() {
Peter Collingbournef7706832014-12-12 23:41:25 +0000700 return arrangeLLVMFunctionInfo(
701 getContext().VoidTy, /*instanceMethod=*/false, /*chainCall=*/false,
John McCallc56a8b32016-03-11 04:30:31 +0000702 None, FunctionType::ExtInfo(), {}, RequiredArgs::All);
703}
704
705const CGFunctionInfo &
706CodeGenTypes::arrangeCall(const CGFunctionInfo &signature,
707 const CallArgList &args) {
708 assert(signature.arg_size() <= args.size());
709 if (signature.arg_size() == args.size())
710 return signature;
711
712 SmallVector<FunctionProtoType::ExtParameterInfo, 16> paramInfos;
713 auto sigParamInfos = signature.getExtParameterInfos();
714 if (!sigParamInfos.empty()) {
715 paramInfos.append(sigParamInfos.begin(), sigParamInfos.end());
716 paramInfos.resize(args.size());
717 }
718
719 auto argTypes = getArgTypesForCall(Context, args);
720
721 assert(signature.getRequiredArgs().allowsOptionalArgs());
722 return arrangeLLVMFunctionInfo(signature.getReturnType(),
723 signature.isInstanceMethod(),
724 signature.isChainCall(),
725 argTypes,
726 signature.getExtInfo(),
727 paramInfos,
728 signature.getRequiredArgs());
John McCalla738c252011-03-09 04:27:21 +0000729}
730
Pekka Jaaskelainenfc2629a2017-06-01 07:18:49 +0000731namespace clang {
732namespace CodeGen {
733void computeSPIRKernelABIInfo(CodeGenModule &CGM, CGFunctionInfo &FI);
734}
735}
736
John McCalla729c622012-02-17 03:33:10 +0000737/// Arrange the argument and result information for an abstract value
738/// of a given function type. This is the method which all of the
739/// above functions ultimately defer to.
740const CGFunctionInfo &
John McCall8dda7b22012-07-07 06:41:13 +0000741CodeGenTypes::arrangeLLVMFunctionInfo(CanQualType resultType,
Peter Collingbournef7706832014-12-12 23:41:25 +0000742 bool instanceMethod,
743 bool chainCall,
John McCall8dda7b22012-07-07 06:41:13 +0000744 ArrayRef<CanQualType> argTypes,
745 FunctionType::ExtInfo info,
John McCallc56a8b32016-03-11 04:30:31 +0000746 ArrayRef<FunctionProtoType::ExtParameterInfo> paramInfos,
John McCall8dda7b22012-07-07 06:41:13 +0000747 RequiredArgs required) {
Fangrui Song3117b172018-10-20 17:53:42 +0000748 assert(llvm::all_of(argTypes,
749 [](CanQualType T) { return T.isCanonicalAsParam(); }));
John McCall2da83a32010-02-26 00:48:12 +0000750
Daniel Dunbare0be8292009-02-03 00:07:12 +0000751 // Lookup or create unique function info.
752 llvm::FoldingSetNodeID ID;
John McCallc56a8b32016-03-11 04:30:31 +0000753 CGFunctionInfo::Profile(ID, instanceMethod, chainCall, info, paramInfos,
754 required, resultType, argTypes);
Daniel Dunbare0be8292009-02-03 00:07:12 +0000755
Craig Topper8a13c412014-05-21 05:09:00 +0000756 void *insertPos = nullptr;
John McCalla729c622012-02-17 03:33:10 +0000757 CGFunctionInfo *FI = FunctionInfos.FindNodeOrInsertPos(ID, insertPos);
Daniel Dunbare0be8292009-02-03 00:07:12 +0000758 if (FI)
759 return *FI;
760
John McCallc56a8b32016-03-11 04:30:31 +0000761 unsigned CC = ClangCallConvToLLVMCallConv(info.getCC());
762
John McCalla729c622012-02-17 03:33:10 +0000763 // Construct the function info. We co-allocate the ArgInfos.
Peter Collingbournef7706832014-12-12 23:41:25 +0000764 FI = CGFunctionInfo::create(CC, instanceMethod, chainCall, info,
John McCallc56a8b32016-03-11 04:30:31 +0000765 paramInfos, resultType, argTypes, required);
John McCalla729c622012-02-17 03:33:10 +0000766 FunctionInfos.InsertNode(FI, insertPos);
Daniel Dunbar313321e2009-02-03 05:31:23 +0000767
David Blaikie82e95a32014-11-19 07:49:47 +0000768 bool inserted = FunctionsBeingProcessed.insert(FI).second;
769 (void)inserted;
John McCalla729c622012-02-17 03:33:10 +0000770 assert(inserted && "Recursively being processed?");
Pekka Jaaskelainenfc2629a2017-06-01 07:18:49 +0000771
Daniel Dunbar313321e2009-02-03 05:31:23 +0000772 // Compute ABI information.
Pekka Jaaskelainenfc2629a2017-06-01 07:18:49 +0000773 if (CC == llvm::CallingConv::SPIR_KERNEL) {
774 // Force target independent argument handling for the host visible
775 // kernel functions.
776 computeSPIRKernelABIInfo(CGM, *FI);
777 } else if (info.getCC() == CC_Swift) {
John McCall12f23522016-04-04 18:33:08 +0000778 swiftcall::computeABIInfo(CGM, *FI);
Pekka Jaaskelainenfc2629a2017-06-01 07:18:49 +0000779 } else {
780 getABIInfo().computeInfo(*FI);
John McCall12f23522016-04-04 18:33:08 +0000781 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +0000782
Chris Lattnerfe34c1d2010-07-29 06:26:06 +0000783 // Loop over all of the computed argument and return value info. If any of
784 // them are direct or extend without a specified coerce type, specify the
785 // default now.
John McCalla729c622012-02-17 03:33:10 +0000786 ABIArgInfo &retInfo = FI->getReturnInfo();
Craig Topper8a13c412014-05-21 05:09:00 +0000787 if (retInfo.canHaveCoerceToType() && retInfo.getCoerceToType() == nullptr)
John McCalla729c622012-02-17 03:33:10 +0000788 retInfo.setCoerceToType(ConvertType(FI->getReturnType()));
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +0000789
Aaron Ballmanec47bc22014-03-17 18:10:01 +0000790 for (auto &I : FI->arguments())
Craig Topper8a13c412014-05-21 05:09:00 +0000791 if (I.info.canHaveCoerceToType() && I.info.getCoerceToType() == nullptr)
Aaron Ballmanec47bc22014-03-17 18:10:01 +0000792 I.info.setCoerceToType(ConvertType(I.type));
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +0000793
John McCalla729c622012-02-17 03:33:10 +0000794 bool erased = FunctionsBeingProcessed.erase(FI); (void)erased;
795 assert(erased && "Not in set?");
Fangrui Song6907ce22018-07-30 19:24:48 +0000796
Daniel Dunbare0be8292009-02-03 00:07:12 +0000797 return *FI;
Daniel Dunbarbf8c24a2009-02-02 23:23:47 +0000798}
799
John McCalla729c622012-02-17 03:33:10 +0000800CGFunctionInfo *CGFunctionInfo::create(unsigned llvmCC,
Peter Collingbournef7706832014-12-12 23:41:25 +0000801 bool instanceMethod,
802 bool chainCall,
John McCalla729c622012-02-17 03:33:10 +0000803 const FunctionType::ExtInfo &info,
John McCallc56a8b32016-03-11 04:30:31 +0000804 ArrayRef<ExtParameterInfo> paramInfos,
John McCalla729c622012-02-17 03:33:10 +0000805 CanQualType resultType,
806 ArrayRef<CanQualType> argTypes,
807 RequiredArgs required) {
John McCallc56a8b32016-03-11 04:30:31 +0000808 assert(paramInfos.empty() || paramInfos.size() == argTypes.size());
809
810 void *buffer =
811 operator new(totalSizeToAlloc<ArgInfo, ExtParameterInfo>(
812 argTypes.size() + 1, paramInfos.size()));
813
John McCalla729c622012-02-17 03:33:10 +0000814 CGFunctionInfo *FI = new(buffer) CGFunctionInfo();
815 FI->CallingConvention = llvmCC;
816 FI->EffectiveCallingConvention = llvmCC;
817 FI->ASTCallingConvention = info.getCC();
Peter Collingbournef7706832014-12-12 23:41:25 +0000818 FI->InstanceMethod = instanceMethod;
819 FI->ChainCall = chainCall;
John McCalla729c622012-02-17 03:33:10 +0000820 FI->NoReturn = info.getNoReturn();
821 FI->ReturnsRetained = info.getProducesResult();
Oren Ben Simhon318a6ea2017-04-27 12:01:00 +0000822 FI->NoCallerSavedRegs = info.getNoCallerSavedRegs();
Oren Ben Simhon220671a2018-03-17 13:31:35 +0000823 FI->NoCfCheck = info.getNoCfCheck();
John McCalla729c622012-02-17 03:33:10 +0000824 FI->Required = required;
825 FI->HasRegParm = info.getHasRegParm();
826 FI->RegParm = info.getRegParm();
Craig Topper8a13c412014-05-21 05:09:00 +0000827 FI->ArgStruct = nullptr;
John McCall7f416cc2015-09-08 08:05:57 +0000828 FI->ArgStructAlign = 0;
John McCalla729c622012-02-17 03:33:10 +0000829 FI->NumArgs = argTypes.size();
John McCallc56a8b32016-03-11 04:30:31 +0000830 FI->HasExtParameterInfos = !paramInfos.empty();
John McCalla729c622012-02-17 03:33:10 +0000831 FI->getArgsBuffer()[0].type = resultType;
832 for (unsigned i = 0, e = argTypes.size(); i != e; ++i)
833 FI->getArgsBuffer()[i + 1].type = argTypes[i];
John McCallc56a8b32016-03-11 04:30:31 +0000834 for (unsigned i = 0, e = paramInfos.size(); i != e; ++i)
835 FI->getExtParameterInfosBuffer()[i] = paramInfos[i];
John McCalla729c622012-02-17 03:33:10 +0000836 return FI;
Daniel Dunbar313321e2009-02-03 05:31:23 +0000837}
838
839/***/
840
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000841namespace {
842// ABIArgInfo::Expand implementation.
843
844// Specifies the way QualType passed as ABIArgInfo::Expand is expanded.
845struct TypeExpansion {
846 enum TypeExpansionKind {
847 // Elements of constant arrays are expanded recursively.
848 TEK_ConstantArray,
849 // Record fields are expanded recursively (but if record is a union, only
850 // the field with the largest size is expanded).
851 TEK_Record,
852 // For complex types, real and imaginary parts are expanded recursively.
853 TEK_Complex,
854 // All other types are not expandable.
855 TEK_None
856 };
857
858 const TypeExpansionKind Kind;
859
860 TypeExpansion(TypeExpansionKind K) : Kind(K) {}
Angel Garcia Gomez637d1e62015-10-20 13:23:58 +0000861 virtual ~TypeExpansion() {}
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000862};
863
864struct ConstantArrayExpansion : TypeExpansion {
865 QualType EltTy;
866 uint64_t NumElts;
867
868 ConstantArrayExpansion(QualType EltTy, uint64_t NumElts)
869 : TypeExpansion(TEK_ConstantArray), EltTy(EltTy), NumElts(NumElts) {}
870 static bool classof(const TypeExpansion *TE) {
871 return TE->Kind == TEK_ConstantArray;
872 }
873};
874
875struct RecordExpansion : TypeExpansion {
Reid Klecknere9f6a712014-10-31 17:10:41 +0000876 SmallVector<const CXXBaseSpecifier *, 1> Bases;
877
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000878 SmallVector<const FieldDecl *, 1> Fields;
879
Reid Klecknere9f6a712014-10-31 17:10:41 +0000880 RecordExpansion(SmallVector<const CXXBaseSpecifier *, 1> &&Bases,
881 SmallVector<const FieldDecl *, 1> &&Fields)
Benjamin Kramer0bb97742016-02-13 16:00:13 +0000882 : TypeExpansion(TEK_Record), Bases(std::move(Bases)),
883 Fields(std::move(Fields)) {}
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000884 static bool classof(const TypeExpansion *TE) {
885 return TE->Kind == TEK_Record;
886 }
887};
888
889struct ComplexExpansion : TypeExpansion {
890 QualType EltTy;
891
892 ComplexExpansion(QualType EltTy) : TypeExpansion(TEK_Complex), EltTy(EltTy) {}
893 static bool classof(const TypeExpansion *TE) {
894 return TE->Kind == TEK_Complex;
895 }
896};
897
898struct NoExpansion : TypeExpansion {
899 NoExpansion() : TypeExpansion(TEK_None) {}
900 static bool classof(const TypeExpansion *TE) {
901 return TE->Kind == TEK_None;
902 }
903};
904} // namespace
905
906static std::unique_ptr<TypeExpansion>
907getTypeExpansion(QualType Ty, const ASTContext &Context) {
908 if (const ConstantArrayType *AT = Context.getAsConstantArrayType(Ty)) {
909 return llvm::make_unique<ConstantArrayExpansion>(
910 AT->getElementType(), AT->getSize().getZExtValue());
911 }
912 if (const RecordType *RT = Ty->getAs<RecordType>()) {
Reid Klecknere9f6a712014-10-31 17:10:41 +0000913 SmallVector<const CXXBaseSpecifier *, 1> Bases;
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000914 SmallVector<const FieldDecl *, 1> Fields;
Bob Wilsone826a2a2011-08-03 05:58:22 +0000915 const RecordDecl *RD = RT->getDecl();
916 assert(!RD->hasFlexibleArrayMember() &&
917 "Cannot expand structure with flexible array.");
Anton Korobeynikov4215ca72012-04-13 11:22:00 +0000918 if (RD->isUnion()) {
919 // Unions can be here only in degenerative cases - all the fields are same
920 // after flattening. Thus we have to use the "largest" field.
Craig Topper8a13c412014-05-21 05:09:00 +0000921 const FieldDecl *LargestFD = nullptr;
Anton Korobeynikov4215ca72012-04-13 11:22:00 +0000922 CharUnits UnionSize = CharUnits::Zero();
923
Aaron Ballmane8a8bae2014-03-08 20:12:42 +0000924 for (const auto *FD : RD->fields()) {
Richard Smith866dee42018-04-02 18:29:43 +0000925 if (FD->isZeroLengthBitField(Context))
Reid Kleckner80944df2014-10-31 22:00:51 +0000926 continue;
Anton Korobeynikov4215ca72012-04-13 11:22:00 +0000927 assert(!FD->isBitField() &&
928 "Cannot expand structure with bit-field members.");
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000929 CharUnits FieldSize = Context.getTypeSizeInChars(FD->getType());
Anton Korobeynikov4215ca72012-04-13 11:22:00 +0000930 if (UnionSize < FieldSize) {
931 UnionSize = FieldSize;
932 LargestFD = FD;
933 }
934 }
935 if (LargestFD)
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000936 Fields.push_back(LargestFD);
Anton Korobeynikov4215ca72012-04-13 11:22:00 +0000937 } else {
Reid Klecknere9f6a712014-10-31 17:10:41 +0000938 if (const auto *CXXRD = dyn_cast<CXXRecordDecl>(RD)) {
939 assert(!CXXRD->isDynamicClass() &&
940 "cannot expand vtable pointers in dynamic classes");
941 for (const CXXBaseSpecifier &BS : CXXRD->bases())
942 Bases.push_back(&BS);
943 }
944
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000945 for (const auto *FD : RD->fields()) {
Richard Smith866dee42018-04-02 18:29:43 +0000946 if (FD->isZeroLengthBitField(Context))
Reid Kleckner80944df2014-10-31 22:00:51 +0000947 continue;
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000948 assert(!FD->isBitField() &&
Anton Korobeynikov4215ca72012-04-13 11:22:00 +0000949 "Cannot expand structure with bit-field members.");
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000950 Fields.push_back(FD);
Anton Korobeynikov4215ca72012-04-13 11:22:00 +0000951 }
Bob Wilsone826a2a2011-08-03 05:58:22 +0000952 }
Reid Klecknere9f6a712014-10-31 17:10:41 +0000953 return llvm::make_unique<RecordExpansion>(std::move(Bases),
954 std::move(Fields));
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000955 }
956 if (const ComplexType *CT = Ty->getAs<ComplexType>()) {
957 return llvm::make_unique<ComplexExpansion>(CT->getElementType());
958 }
959 return llvm::make_unique<NoExpansion>();
960}
961
Alexey Samsonov52c0f6a2014-09-29 20:30:22 +0000962static int getExpansionSize(QualType Ty, const ASTContext &Context) {
963 auto Exp = getTypeExpansion(Ty, Context);
964 if (auto CAExp = dyn_cast<ConstantArrayExpansion>(Exp.get())) {
965 return CAExp->NumElts * getExpansionSize(CAExp->EltTy, Context);
966 }
967 if (auto RExp = dyn_cast<RecordExpansion>(Exp.get())) {
968 int Res = 0;
Reid Klecknere9f6a712014-10-31 17:10:41 +0000969 for (auto BS : RExp->Bases)
970 Res += getExpansionSize(BS->getType(), Context);
Alexey Samsonov52c0f6a2014-09-29 20:30:22 +0000971 for (auto FD : RExp->Fields)
972 Res += getExpansionSize(FD->getType(), Context);
973 return Res;
974 }
975 if (isa<ComplexExpansion>(Exp.get()))
976 return 2;
977 assert(isa<NoExpansion>(Exp.get()));
978 return 1;
979}
980
Alexey Samsonov153004f2014-09-29 22:08:00 +0000981void
982CodeGenTypes::getExpandedTypes(QualType Ty,
983 SmallVectorImpl<llvm::Type *>::iterator &TI) {
984 auto Exp = getTypeExpansion(Ty, Context);
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000985 if (auto CAExp = dyn_cast<ConstantArrayExpansion>(Exp.get())) {
986 for (int i = 0, n = CAExp->NumElts; i < n; i++) {
Alexey Samsonov153004f2014-09-29 22:08:00 +0000987 getExpandedTypes(CAExp->EltTy, TI);
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000988 }
989 } else if (auto RExp = dyn_cast<RecordExpansion>(Exp.get())) {
Reid Klecknere9f6a712014-10-31 17:10:41 +0000990 for (auto BS : RExp->Bases)
991 getExpandedTypes(BS->getType(), TI);
992 for (auto FD : RExp->Fields)
Alexey Samsonov153004f2014-09-29 22:08:00 +0000993 getExpandedTypes(FD->getType(), TI);
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000994 } else if (auto CExp = dyn_cast<ComplexExpansion>(Exp.get())) {
995 llvm::Type *EltTy = ConvertType(CExp->EltTy);
Alexey Samsonov153004f2014-09-29 22:08:00 +0000996 *TI++ = EltTy;
997 *TI++ = EltTy;
Alexey Samsonov8a0bad02014-09-29 18:41:28 +0000998 } else {
999 assert(isa<NoExpansion>(Exp.get()));
Alexey Samsonov153004f2014-09-29 22:08:00 +00001000 *TI++ = ConvertType(Ty);
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001001 }
Daniel Dunbar8fc81b02008-09-17 00:51:38 +00001002}
1003
John McCall7f416cc2015-09-08 08:05:57 +00001004static void forConstantArrayExpansion(CodeGenFunction &CGF,
1005 ConstantArrayExpansion *CAE,
1006 Address BaseAddr,
1007 llvm::function_ref<void(Address)> Fn) {
1008 CharUnits EltSize = CGF.getContext().getTypeSizeInChars(CAE->EltTy);
1009 CharUnits EltAlign =
1010 BaseAddr.getAlignment().alignmentOfArrayElement(EltSize);
1011
1012 for (int i = 0, n = CAE->NumElts; i < n; i++) {
1013 llvm::Value *EltAddr =
1014 CGF.Builder.CreateConstGEP2_32(nullptr, BaseAddr.getPointer(), 0, i);
1015 Fn(Address(EltAddr, EltAlign));
1016 }
1017}
1018
Alexey Samsonov91cf4552014-08-22 01:06:06 +00001019void CodeGenFunction::ExpandTypeFromArgs(
John McCall12f23522016-04-04 18:33:08 +00001020 QualType Ty, LValue LV, SmallVectorImpl<llvm::Value *>::iterator &AI) {
Mike Stump11289f42009-09-09 15:08:12 +00001021 assert(LV.isSimple() &&
1022 "Unexpected non-simple lvalue during struct expansion.");
Daniel Dunbar8fc81b02008-09-17 00:51:38 +00001023
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001024 auto Exp = getTypeExpansion(Ty, getContext());
1025 if (auto CAExp = dyn_cast<ConstantArrayExpansion>(Exp.get())) {
John McCall7f416cc2015-09-08 08:05:57 +00001026 forConstantArrayExpansion(*this, CAExp, LV.getAddress(),
1027 [&](Address EltAddr) {
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001028 LValue LV = MakeAddrLValue(EltAddr, CAExp->EltTy);
1029 ExpandTypeFromArgs(CAExp->EltTy, LV, AI);
John McCall7f416cc2015-09-08 08:05:57 +00001030 });
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001031 } else if (auto RExp = dyn_cast<RecordExpansion>(Exp.get())) {
John McCall7f416cc2015-09-08 08:05:57 +00001032 Address This = LV.getAddress();
Reid Klecknere9f6a712014-10-31 17:10:41 +00001033 for (const CXXBaseSpecifier *BS : RExp->Bases) {
1034 // Perform a single step derived-to-base conversion.
John McCall7f416cc2015-09-08 08:05:57 +00001035 Address Base =
Reid Klecknere9f6a712014-10-31 17:10:41 +00001036 GetAddressOfBaseClass(This, Ty->getAsCXXRecordDecl(), &BS, &BS + 1,
1037 /*NullCheckValue=*/false, SourceLocation());
1038 LValue SubLV = MakeAddrLValue(Base, BS->getType());
1039
1040 // Recurse onto bases.
1041 ExpandTypeFromArgs(BS->getType(), SubLV, AI);
1042 }
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001043 for (auto FD : RExp->Fields) {
1044 // FIXME: What are the right qualifiers here?
Reid Kleckner9d031092016-05-02 22:42:34 +00001045 LValue SubLV = EmitLValueForFieldInitialization(LV, FD);
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001046 ExpandTypeFromArgs(FD->getType(), SubLV, AI);
Bob Wilsone826a2a2011-08-03 05:58:22 +00001047 }
John McCall7f416cc2015-09-08 08:05:57 +00001048 } else if (isa<ComplexExpansion>(Exp.get())) {
1049 auto realValue = *AI++;
1050 auto imagValue = *AI++;
1051 EmitStoreOfComplex(ComplexPairTy(realValue, imagValue), LV, /*init*/ true);
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001052 } else {
1053 assert(isa<NoExpansion>(Exp.get()));
1054 EmitStoreThroughLValue(RValue::get(*AI++), LV);
Daniel Dunbar8fc81b02008-09-17 00:51:38 +00001055 }
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001056}
1057
1058void CodeGenFunction::ExpandTypeToArgs(
Yaxun Liu5b330e82018-03-15 15:25:19 +00001059 QualType Ty, CallArg Arg, llvm::FunctionType *IRFuncTy,
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001060 SmallVectorImpl<llvm::Value *> &IRCallArgs, unsigned &IRCallArgPos) {
1061 auto Exp = getTypeExpansion(Ty, getContext());
1062 if (auto CAExp = dyn_cast<ConstantArrayExpansion>(Exp.get())) {
Yaxun Liu5b330e82018-03-15 15:25:19 +00001063 Address Addr = Arg.hasLValue() ? Arg.getKnownLValue().getAddress()
1064 : Arg.getKnownRValue().getAggregateAddress();
1065 forConstantArrayExpansion(
1066 *this, CAExp, Addr, [&](Address EltAddr) {
1067 CallArg EltArg = CallArg(
1068 convertTempToRValue(EltAddr, CAExp->EltTy, SourceLocation()),
1069 CAExp->EltTy);
1070 ExpandTypeToArgs(CAExp->EltTy, EltArg, IRFuncTy, IRCallArgs,
1071 IRCallArgPos);
1072 });
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001073 } else if (auto RExp = dyn_cast<RecordExpansion>(Exp.get())) {
Yaxun Liu5b330e82018-03-15 15:25:19 +00001074 Address This = Arg.hasLValue() ? Arg.getKnownLValue().getAddress()
1075 : Arg.getKnownRValue().getAggregateAddress();
Reid Klecknere9f6a712014-10-31 17:10:41 +00001076 for (const CXXBaseSpecifier *BS : RExp->Bases) {
1077 // Perform a single step derived-to-base conversion.
John McCall7f416cc2015-09-08 08:05:57 +00001078 Address Base =
Reid Klecknere9f6a712014-10-31 17:10:41 +00001079 GetAddressOfBaseClass(This, Ty->getAsCXXRecordDecl(), &BS, &BS + 1,
1080 /*NullCheckValue=*/false, SourceLocation());
Yaxun Liu5b330e82018-03-15 15:25:19 +00001081 CallArg BaseArg = CallArg(RValue::getAggregate(Base), BS->getType());
Reid Klecknere9f6a712014-10-31 17:10:41 +00001082
1083 // Recurse onto bases.
Yaxun Liu5b330e82018-03-15 15:25:19 +00001084 ExpandTypeToArgs(BS->getType(), BaseArg, IRFuncTy, IRCallArgs,
Reid Klecknere9f6a712014-10-31 17:10:41 +00001085 IRCallArgPos);
1086 }
1087
1088 LValue LV = MakeAddrLValue(This, Ty);
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001089 for (auto FD : RExp->Fields) {
Yaxun Liu5b330e82018-03-15 15:25:19 +00001090 CallArg FldArg =
1091 CallArg(EmitRValueForField(LV, FD, SourceLocation()), FD->getType());
1092 ExpandTypeToArgs(FD->getType(), FldArg, IRFuncTy, IRCallArgs,
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001093 IRCallArgPos);
1094 }
1095 } else if (isa<ComplexExpansion>(Exp.get())) {
Yaxun Liu5b330e82018-03-15 15:25:19 +00001096 ComplexPairTy CV = Arg.getKnownRValue().getComplexVal();
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001097 IRCallArgs[IRCallArgPos++] = CV.first;
1098 IRCallArgs[IRCallArgPos++] = CV.second;
1099 } else {
1100 assert(isa<NoExpansion>(Exp.get()));
Yaxun Liu5b330e82018-03-15 15:25:19 +00001101 auto RV = Arg.getKnownRValue();
Alexey Samsonov8a0bad02014-09-29 18:41:28 +00001102 assert(RV.isScalar() &&
1103 "Unexpected non-scalar rvalue during struct expansion.");
1104
1105 // Insert a bitcast as needed.
1106 llvm::Value *V = RV.getScalarVal();
1107 if (IRCallArgPos < IRFuncTy->getNumParams() &&
1108 V->getType() != IRFuncTy->getParamType(IRCallArgPos))
1109 V = Builder.CreateBitCast(V, IRFuncTy->getParamType(IRCallArgPos));
1110
1111 IRCallArgs[IRCallArgPos++] = V;
1112 }
Daniel Dunbar8fc81b02008-09-17 00:51:38 +00001113}
1114
John McCall7f416cc2015-09-08 08:05:57 +00001115/// Create a temporary allocation for the purposes of coercion.
1116static Address CreateTempAllocaForCoercion(CodeGenFunction &CGF, llvm::Type *Ty,
1117 CharUnits MinAlign) {
1118 // Don't use an alignment that's worse than what LLVM would prefer.
1119 auto PrefAlign = CGF.CGM.getDataLayout().getPrefTypeAlignment(Ty);
1120 CharUnits Align = std::max(MinAlign, CharUnits::fromQuantity(PrefAlign));
1121
1122 return CGF.CreateTempAlloca(Ty, Align);
1123}
1124
Chris Lattner895c52b2010-06-27 06:04:18 +00001125/// EnterStructPointerForCoercedAccess - Given a struct pointer that we are
Chris Lattner1cd66982010-06-27 05:56:15 +00001126/// accessing some number of bytes out of it, try to gep into the struct to get
1127/// at its inner goodness. Dive as deep as possible without entering an element
1128/// with an in-memory size smaller than DstSize.
John McCall7f416cc2015-09-08 08:05:57 +00001129static Address
1130EnterStructPointerForCoercedAccess(Address SrcPtr,
Chris Lattner2192fe52011-07-18 04:24:23 +00001131 llvm::StructType *SrcSTy,
Chris Lattner895c52b2010-06-27 06:04:18 +00001132 uint64_t DstSize, CodeGenFunction &CGF) {
Chris Lattner1cd66982010-06-27 05:56:15 +00001133 // We can't dive into a zero-element struct.
1134 if (SrcSTy->getNumElements() == 0) return SrcPtr;
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001135
Chris Lattner2192fe52011-07-18 04:24:23 +00001136 llvm::Type *FirstElt = SrcSTy->getElementType(0);
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001137
Chris Lattner1cd66982010-06-27 05:56:15 +00001138 // If the first elt is at least as large as what we're looking for, or if the
James Molloy90d61012014-08-29 10:17:52 +00001139 // first element is the same size as the whole struct, we can enter it. The
1140 // comparison must be made on the store size and not the alloca size. Using
1141 // the alloca size may overstate the size of the load.
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001142 uint64_t FirstEltSize =
James Molloy90d61012014-08-29 10:17:52 +00001143 CGF.CGM.getDataLayout().getTypeStoreSize(FirstElt);
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001144 if (FirstEltSize < DstSize &&
James Molloy90d61012014-08-29 10:17:52 +00001145 FirstEltSize < CGF.CGM.getDataLayout().getTypeStoreSize(SrcSTy))
Chris Lattner1cd66982010-06-27 05:56:15 +00001146 return SrcPtr;
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001147
Chris Lattner1cd66982010-06-27 05:56:15 +00001148 // GEP into the first element.
John McCall7f416cc2015-09-08 08:05:57 +00001149 SrcPtr = CGF.Builder.CreateStructGEP(SrcPtr, 0, CharUnits(), "coerce.dive");
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001150
Chris Lattner1cd66982010-06-27 05:56:15 +00001151 // If the first element is a struct, recurse.
John McCall7f416cc2015-09-08 08:05:57 +00001152 llvm::Type *SrcTy = SrcPtr.getElementType();
Chris Lattner2192fe52011-07-18 04:24:23 +00001153 if (llvm::StructType *SrcSTy = dyn_cast<llvm::StructType>(SrcTy))
Chris Lattner895c52b2010-06-27 06:04:18 +00001154 return EnterStructPointerForCoercedAccess(SrcPtr, SrcSTy, DstSize, CGF);
Chris Lattner1cd66982010-06-27 05:56:15 +00001155
1156 return SrcPtr;
1157}
1158
Chris Lattner055097f2010-06-27 06:26:04 +00001159/// CoerceIntOrPtrToIntOrPtr - Convert a value Val to the specific Ty where both
1160/// are either integers or pointers. This does a truncation of the value if it
1161/// is too large or a zero extension if it is too small.
Jakob Stoklund Olesen36af2522013-06-05 03:00:13 +00001162///
1163/// This behaves as if the value were coerced through memory, so on big-endian
1164/// targets the high bits are preserved in a truncation, while little-endian
1165/// targets preserve the low bits.
Chris Lattner055097f2010-06-27 06:26:04 +00001166static llvm::Value *CoerceIntOrPtrToIntOrPtr(llvm::Value *Val,
Chris Lattner2192fe52011-07-18 04:24:23 +00001167 llvm::Type *Ty,
Chris Lattner055097f2010-06-27 06:26:04 +00001168 CodeGenFunction &CGF) {
1169 if (Val->getType() == Ty)
1170 return Val;
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001171
Chris Lattner055097f2010-06-27 06:26:04 +00001172 if (isa<llvm::PointerType>(Val->getType())) {
1173 // If this is Pointer->Pointer avoid conversion to and from int.
1174 if (isa<llvm::PointerType>(Ty))
1175 return CGF.Builder.CreateBitCast(Val, Ty, "coerce.val");
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001176
Chris Lattner055097f2010-06-27 06:26:04 +00001177 // Convert the pointer to an integer so we can play with its width.
Chris Lattner5e016ae2010-06-27 07:15:29 +00001178 Val = CGF.Builder.CreatePtrToInt(Val, CGF.IntPtrTy, "coerce.val.pi");
Chris Lattner055097f2010-06-27 06:26:04 +00001179 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001180
Chris Lattner2192fe52011-07-18 04:24:23 +00001181 llvm::Type *DestIntTy = Ty;
Chris Lattner055097f2010-06-27 06:26:04 +00001182 if (isa<llvm::PointerType>(DestIntTy))
Chris Lattner5e016ae2010-06-27 07:15:29 +00001183 DestIntTy = CGF.IntPtrTy;
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001184
Jakob Stoklund Olesen36af2522013-06-05 03:00:13 +00001185 if (Val->getType() != DestIntTy) {
1186 const llvm::DataLayout &DL = CGF.CGM.getDataLayout();
1187 if (DL.isBigEndian()) {
1188 // Preserve the high bits on big-endian targets.
1189 // That is what memory coercion does.
James Molloy491cefb2014-05-07 17:41:15 +00001190 uint64_t SrcSize = DL.getTypeSizeInBits(Val->getType());
1191 uint64_t DstSize = DL.getTypeSizeInBits(DestIntTy);
1192
Jakob Stoklund Olesen36af2522013-06-05 03:00:13 +00001193 if (SrcSize > DstSize) {
1194 Val = CGF.Builder.CreateLShr(Val, SrcSize - DstSize, "coerce.highbits");
1195 Val = CGF.Builder.CreateTrunc(Val, DestIntTy, "coerce.val.ii");
1196 } else {
1197 Val = CGF.Builder.CreateZExt(Val, DestIntTy, "coerce.val.ii");
1198 Val = CGF.Builder.CreateShl(Val, DstSize - SrcSize, "coerce.highbits");
1199 }
1200 } else {
1201 // Little-endian targets preserve the low bits. No shifts required.
1202 Val = CGF.Builder.CreateIntCast(Val, DestIntTy, false, "coerce.val.ii");
1203 }
1204 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001205
Chris Lattner055097f2010-06-27 06:26:04 +00001206 if (isa<llvm::PointerType>(Ty))
1207 Val = CGF.Builder.CreateIntToPtr(Val, Ty, "coerce.val.ip");
1208 return Val;
1209}
1210
Chris Lattner1cd66982010-06-27 05:56:15 +00001211
1212
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001213/// CreateCoercedLoad - Create a load from \arg SrcPtr interpreted as
Ulrich Weigand6e2cea62015-07-10 11:31:43 +00001214/// a pointer to an object of type \arg Ty, known to be aligned to
1215/// \arg SrcAlign bytes.
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001216///
1217/// This safely handles the case when the src type is smaller than the
1218/// destination type; in this situation the values of bits which not
1219/// present in the src are undefined.
John McCall7f416cc2015-09-08 08:05:57 +00001220static llvm::Value *CreateCoercedLoad(Address Src, llvm::Type *Ty,
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001221 CodeGenFunction &CGF) {
John McCall7f416cc2015-09-08 08:05:57 +00001222 llvm::Type *SrcTy = Src.getElementType();
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001223
Chris Lattnerd200eda2010-06-28 22:51:39 +00001224 // If SrcTy and Ty are the same, just do a load.
1225 if (SrcTy == Ty)
John McCall7f416cc2015-09-08 08:05:57 +00001226 return CGF.Builder.CreateLoad(Src);
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001227
Micah Villmowdd31ca12012-10-08 16:25:52 +00001228 uint64_t DstSize = CGF.CGM.getDataLayout().getTypeAllocSize(Ty);
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001229
Chris Lattner2192fe52011-07-18 04:24:23 +00001230 if (llvm::StructType *SrcSTy = dyn_cast<llvm::StructType>(SrcTy)) {
John McCall7f416cc2015-09-08 08:05:57 +00001231 Src = EnterStructPointerForCoercedAccess(Src, SrcSTy, DstSize, CGF);
1232 SrcTy = Src.getType()->getElementType();
Chris Lattner1cd66982010-06-27 05:56:15 +00001233 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001234
Micah Villmowdd31ca12012-10-08 16:25:52 +00001235 uint64_t SrcSize = CGF.CGM.getDataLayout().getTypeAllocSize(SrcTy);
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001236
Chris Lattner055097f2010-06-27 06:26:04 +00001237 // If the source and destination are integer or pointer types, just do an
1238 // extension or truncation to the desired type.
1239 if ((isa<llvm::IntegerType>(Ty) || isa<llvm::PointerType>(Ty)) &&
1240 (isa<llvm::IntegerType>(SrcTy) || isa<llvm::PointerType>(SrcTy))) {
John McCall7f416cc2015-09-08 08:05:57 +00001241 llvm::Value *Load = CGF.Builder.CreateLoad(Src);
Chris Lattner055097f2010-06-27 06:26:04 +00001242 return CoerceIntOrPtrToIntOrPtr(Load, Ty, CGF);
1243 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001244
Daniel Dunbarb52d0772009-02-03 05:59:18 +00001245 // If load is legal, just bitcast the src pointer.
Daniel Dunbarffdb8432009-05-13 18:54:26 +00001246 if (SrcSize >= DstSize) {
Mike Stump18bb9282009-05-16 07:57:57 +00001247 // Generally SrcSize is never greater than DstSize, since this means we are
1248 // losing bits. However, this can happen in cases where the structure has
1249 // additional padding, for example due to a user specified alignment.
Daniel Dunbarffdb8432009-05-13 18:54:26 +00001250 //
Mike Stump18bb9282009-05-16 07:57:57 +00001251 // FIXME: Assert that we aren't truncating non-padding bits when have access
1252 // to that information.
Matt Arsenault7a124f32017-08-01 20:36:57 +00001253 Src = CGF.Builder.CreateBitCast(Src,
1254 Ty->getPointerTo(Src.getAddressSpace()));
John McCall7f416cc2015-09-08 08:05:57 +00001255 return CGF.Builder.CreateLoad(Src);
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001256 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001257
John McCall7f416cc2015-09-08 08:05:57 +00001258 // Otherwise do coercion through memory. This is stupid, but simple.
1259 Address Tmp = CreateTempAllocaForCoercion(CGF, Ty, Src.getAlignment());
Yaxun Liu4bbdebc2018-11-08 16:55:46 +00001260 Address Casted = CGF.Builder.CreateElementBitCast(Tmp,CGF.Int8Ty);
1261 Address SrcCasted = CGF.Builder.CreateElementBitCast(Src,CGF.Int8Ty);
Manman Ren84b921f2012-11-28 22:08:52 +00001262 CGF.Builder.CreateMemCpy(Casted, SrcCasted,
1263 llvm::ConstantInt::get(CGF.IntPtrTy, SrcSize),
John McCall7f416cc2015-09-08 08:05:57 +00001264 false);
1265 return CGF.Builder.CreateLoad(Tmp);
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001266}
1267
Eli Friedmanaf9b3252011-05-17 21:08:01 +00001268// Function to store a first-class aggregate into memory. We prefer to
1269// store the elements rather than the aggregate to be more friendly to
1270// fast-isel.
1271// FIXME: Do we need to recurse here?
1272static void BuildAggStore(CodeGenFunction &CGF, llvm::Value *Val,
John McCall7f416cc2015-09-08 08:05:57 +00001273 Address Dest, bool DestIsVolatile) {
Eli Friedmanaf9b3252011-05-17 21:08:01 +00001274 // Prefer scalar stores to first-class aggregate stores.
Chris Lattner2192fe52011-07-18 04:24:23 +00001275 if (llvm::StructType *STy =
Eli Friedmanaf9b3252011-05-17 21:08:01 +00001276 dyn_cast<llvm::StructType>(Val->getType())) {
Ulrich Weigand6e2cea62015-07-10 11:31:43 +00001277 const llvm::StructLayout *Layout =
1278 CGF.CGM.getDataLayout().getStructLayout(STy);
1279
Eli Friedmanaf9b3252011-05-17 21:08:01 +00001280 for (unsigned i = 0, e = STy->getNumElements(); i != e; ++i) {
John McCall7f416cc2015-09-08 08:05:57 +00001281 auto EltOffset = CharUnits::fromQuantity(Layout->getElementOffset(i));
1282 Address EltPtr = CGF.Builder.CreateStructGEP(Dest, i, EltOffset);
Eli Friedmanaf9b3252011-05-17 21:08:01 +00001283 llvm::Value *Elt = CGF.Builder.CreateExtractValue(Val, i);
John McCall7f416cc2015-09-08 08:05:57 +00001284 CGF.Builder.CreateStore(Elt, EltPtr, DestIsVolatile);
Eli Friedmanaf9b3252011-05-17 21:08:01 +00001285 }
1286 } else {
John McCall7f416cc2015-09-08 08:05:57 +00001287 CGF.Builder.CreateStore(Val, Dest, DestIsVolatile);
Eli Friedmanaf9b3252011-05-17 21:08:01 +00001288 }
1289}
1290
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001291/// CreateCoercedStore - Create a store to \arg DstPtr from \arg Src,
Ulrich Weigand6e2cea62015-07-10 11:31:43 +00001292/// where the source and destination may have different types. The
1293/// destination is known to be aligned to \arg DstAlign bytes.
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001294///
1295/// This safely handles the case when the src type is larger than the
1296/// destination type; the upper bits of the src will be lost.
1297static void CreateCoercedStore(llvm::Value *Src,
John McCall7f416cc2015-09-08 08:05:57 +00001298 Address Dst,
Anders Carlsson17490832009-12-24 20:40:36 +00001299 bool DstIsVolatile,
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001300 CodeGenFunction &CGF) {
Chris Lattner2192fe52011-07-18 04:24:23 +00001301 llvm::Type *SrcTy = Src->getType();
John McCall7f416cc2015-09-08 08:05:57 +00001302 llvm::Type *DstTy = Dst.getType()->getElementType();
Chris Lattnerd200eda2010-06-28 22:51:39 +00001303 if (SrcTy == DstTy) {
John McCall7f416cc2015-09-08 08:05:57 +00001304 CGF.Builder.CreateStore(Src, Dst, DstIsVolatile);
Chris Lattnerd200eda2010-06-28 22:51:39 +00001305 return;
1306 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001307
Micah Villmowdd31ca12012-10-08 16:25:52 +00001308 uint64_t SrcSize = CGF.CGM.getDataLayout().getTypeAllocSize(SrcTy);
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001309
Chris Lattner2192fe52011-07-18 04:24:23 +00001310 if (llvm::StructType *DstSTy = dyn_cast<llvm::StructType>(DstTy)) {
John McCall7f416cc2015-09-08 08:05:57 +00001311 Dst = EnterStructPointerForCoercedAccess(Dst, DstSTy, SrcSize, CGF);
1312 DstTy = Dst.getType()->getElementType();
Chris Lattner895c52b2010-06-27 06:04:18 +00001313 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001314
Chris Lattner055097f2010-06-27 06:26:04 +00001315 // If the source and destination are integer or pointer types, just do an
1316 // extension or truncation to the desired type.
1317 if ((isa<llvm::IntegerType>(SrcTy) || isa<llvm::PointerType>(SrcTy)) &&
1318 (isa<llvm::IntegerType>(DstTy) || isa<llvm::PointerType>(DstTy))) {
1319 Src = CoerceIntOrPtrToIntOrPtr(Src, DstTy, CGF);
John McCall7f416cc2015-09-08 08:05:57 +00001320 CGF.Builder.CreateStore(Src, Dst, DstIsVolatile);
Chris Lattner055097f2010-06-27 06:26:04 +00001321 return;
1322 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001323
Micah Villmowdd31ca12012-10-08 16:25:52 +00001324 uint64_t DstSize = CGF.CGM.getDataLayout().getTypeAllocSize(DstTy);
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001325
Daniel Dunbar313321e2009-02-03 05:31:23 +00001326 // If store is legal, just bitcast the src pointer.
Daniel Dunbar4be99ff2009-06-05 07:58:54 +00001327 if (SrcSize <= DstSize) {
Yaxun Liue9e5c4f2017-06-29 18:47:45 +00001328 Dst = CGF.Builder.CreateElementBitCast(Dst, SrcTy);
John McCall7f416cc2015-09-08 08:05:57 +00001329 BuildAggStore(CGF, Src, Dst, DstIsVolatile);
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001330 } else {
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001331 // Otherwise do coercion through memory. This is stupid, but
1332 // simple.
Daniel Dunbar4be99ff2009-06-05 07:58:54 +00001333
1334 // Generally SrcSize is never greater than DstSize, since this means we are
1335 // losing bits. However, this can happen in cases where the structure has
1336 // additional padding, for example due to a user specified alignment.
1337 //
1338 // FIXME: Assert that we aren't truncating non-padding bits when have access
1339 // to that information.
John McCall7f416cc2015-09-08 08:05:57 +00001340 Address Tmp = CreateTempAllocaForCoercion(CGF, SrcTy, Dst.getAlignment());
1341 CGF.Builder.CreateStore(Src, Tmp);
Yaxun Liu4bbdebc2018-11-08 16:55:46 +00001342 Address Casted = CGF.Builder.CreateElementBitCast(Tmp,CGF.Int8Ty);
1343 Address DstCasted = CGF.Builder.CreateElementBitCast(Dst,CGF.Int8Ty);
Manman Ren84b921f2012-11-28 22:08:52 +00001344 CGF.Builder.CreateMemCpy(DstCasted, Casted,
1345 llvm::ConstantInt::get(CGF.IntPtrTy, DstSize),
John McCall7f416cc2015-09-08 08:05:57 +00001346 false);
Daniel Dunbarf5589ac2009-02-02 19:06:38 +00001347 }
1348}
1349
John McCall7f416cc2015-09-08 08:05:57 +00001350static Address emitAddressAtOffset(CodeGenFunction &CGF, Address addr,
Fangrui Song6907ce22018-07-30 19:24:48 +00001351 const ABIArgInfo &info) {
John McCall7f416cc2015-09-08 08:05:57 +00001352 if (unsigned offset = info.getDirectOffset()) {
1353 addr = CGF.Builder.CreateElementBitCast(addr, CGF.Int8Ty);
1354 addr = CGF.Builder.CreateConstInBoundsByteGEP(addr,
1355 CharUnits::fromQuantity(offset));
1356 addr = CGF.Builder.CreateElementBitCast(addr, info.getCoerceToType());
1357 }
1358 return addr;
1359}
1360
Alexey Samsonov153004f2014-09-29 22:08:00 +00001361namespace {
1362
1363/// Encapsulates information about the way function arguments from
1364/// CGFunctionInfo should be passed to actual LLVM IR function.
1365class ClangToLLVMArgMapping {
1366 static const unsigned InvalidIndex = ~0U;
1367 unsigned InallocaArgNo;
1368 unsigned SRetArgNo;
1369 unsigned TotalIRArgs;
1370
1371 /// Arguments of LLVM IR function corresponding to single Clang argument.
1372 struct IRArgs {
1373 unsigned PaddingArgIndex;
1374 // Argument is expanded to IR arguments at positions
1375 // [FirstArgIndex, FirstArgIndex + NumberOfArgs).
1376 unsigned FirstArgIndex;
1377 unsigned NumberOfArgs;
1378
1379 IRArgs()
1380 : PaddingArgIndex(InvalidIndex), FirstArgIndex(InvalidIndex),
1381 NumberOfArgs(0) {}
1382 };
1383
1384 SmallVector<IRArgs, 8> ArgInfo;
1385
1386public:
1387 ClangToLLVMArgMapping(const ASTContext &Context, const CGFunctionInfo &FI,
1388 bool OnlyRequiredArgs = false)
1389 : InallocaArgNo(InvalidIndex), SRetArgNo(InvalidIndex), TotalIRArgs(0),
1390 ArgInfo(OnlyRequiredArgs ? FI.getNumRequiredArgs() : FI.arg_size()) {
1391 construct(Context, FI, OnlyRequiredArgs);
1392 }
1393
1394 bool hasInallocaArg() const { return InallocaArgNo != InvalidIndex; }
1395 unsigned getInallocaArgNo() const {
1396 assert(hasInallocaArg());
1397 return InallocaArgNo;
1398 }
1399
1400 bool hasSRetArg() const { return SRetArgNo != InvalidIndex; }
1401 unsigned getSRetArgNo() const {
1402 assert(hasSRetArg());
1403 return SRetArgNo;
1404 }
1405
1406 unsigned totalIRArgs() const { return TotalIRArgs; }
1407
1408 bool hasPaddingArg(unsigned ArgNo) const {
1409 assert(ArgNo < ArgInfo.size());
1410 return ArgInfo[ArgNo].PaddingArgIndex != InvalidIndex;
1411 }
1412 unsigned getPaddingArgNo(unsigned ArgNo) const {
1413 assert(hasPaddingArg(ArgNo));
1414 return ArgInfo[ArgNo].PaddingArgIndex;
1415 }
1416
1417 /// Returns index of first IR argument corresponding to ArgNo, and their
1418 /// quantity.
1419 std::pair<unsigned, unsigned> getIRArgs(unsigned ArgNo) const {
1420 assert(ArgNo < ArgInfo.size());
1421 return std::make_pair(ArgInfo[ArgNo].FirstArgIndex,
1422 ArgInfo[ArgNo].NumberOfArgs);
1423 }
1424
1425private:
1426 void construct(const ASTContext &Context, const CGFunctionInfo &FI,
1427 bool OnlyRequiredArgs);
1428};
1429
1430void ClangToLLVMArgMapping::construct(const ASTContext &Context,
1431 const CGFunctionInfo &FI,
1432 bool OnlyRequiredArgs) {
1433 unsigned IRArgNo = 0;
1434 bool SwapThisWithSRet = false;
1435 const ABIArgInfo &RetAI = FI.getReturnInfo();
1436
1437 if (RetAI.getKind() == ABIArgInfo::Indirect) {
1438 SwapThisWithSRet = RetAI.isSRetAfterThis();
1439 SRetArgNo = SwapThisWithSRet ? 1 : IRArgNo++;
1440 }
1441
1442 unsigned ArgNo = 0;
1443 unsigned NumArgs = OnlyRequiredArgs ? FI.getNumRequiredArgs() : FI.arg_size();
1444 for (CGFunctionInfo::const_arg_iterator I = FI.arg_begin(); ArgNo < NumArgs;
1445 ++I, ++ArgNo) {
1446 assert(I != FI.arg_end());
1447 QualType ArgType = I->type;
1448 const ABIArgInfo &AI = I->info;
1449 // Collect data about IR arguments corresponding to Clang argument ArgNo.
1450 auto &IRArgs = ArgInfo[ArgNo];
1451
1452 if (AI.getPaddingType())
1453 IRArgs.PaddingArgIndex = IRArgNo++;
1454
1455 switch (AI.getKind()) {
1456 case ABIArgInfo::Extend:
1457 case ABIArgInfo::Direct: {
1458 // FIXME: handle sseregparm someday...
1459 llvm::StructType *STy = dyn_cast<llvm::StructType>(AI.getCoerceToType());
1460 if (AI.isDirect() && AI.getCanBeFlattened() && STy) {
1461 IRArgs.NumberOfArgs = STy->getNumElements();
1462 } else {
1463 IRArgs.NumberOfArgs = 1;
1464 }
1465 break;
1466 }
1467 case ABIArgInfo::Indirect:
1468 IRArgs.NumberOfArgs = 1;
1469 break;
1470 case ABIArgInfo::Ignore:
1471 case ABIArgInfo::InAlloca:
1472 // ignore and inalloca doesn't have matching LLVM parameters.
1473 IRArgs.NumberOfArgs = 0;
1474 break;
John McCallf26e73d2016-03-11 04:30:43 +00001475 case ABIArgInfo::CoerceAndExpand:
1476 IRArgs.NumberOfArgs = AI.getCoerceAndExpandTypeSequence().size();
1477 break;
1478 case ABIArgInfo::Expand:
Alexey Samsonov153004f2014-09-29 22:08:00 +00001479 IRArgs.NumberOfArgs = getExpansionSize(ArgType, Context);
1480 break;
1481 }
Alexey Samsonov153004f2014-09-29 22:08:00 +00001482
1483 if (IRArgs.NumberOfArgs > 0) {
1484 IRArgs.FirstArgIndex = IRArgNo;
1485 IRArgNo += IRArgs.NumberOfArgs;
1486 }
1487
1488 // Skip over the sret parameter when it comes second. We already handled it
1489 // above.
1490 if (IRArgNo == 1 && SwapThisWithSRet)
1491 IRArgNo++;
1492 }
1493 assert(ArgNo == ArgInfo.size());
1494
1495 if (FI.usesInAlloca())
1496 InallocaArgNo = IRArgNo++;
1497
1498 TotalIRArgs = IRArgNo;
1499}
1500} // namespace
1501
Daniel Dunbar8fc81b02008-09-17 00:51:38 +00001502/***/
1503
Daniel Dunbar6f2e8392010-07-14 23:39:36 +00001504bool CodeGenModule::ReturnTypeUsesSRet(const CGFunctionInfo &FI) {
Saleem Abdulrasoolf181f1a2018-02-28 20:16:12 +00001505 const auto &RI = FI.getReturnInfo();
1506 return RI.isIndirect() || (RI.isInAlloca() && RI.getInAllocaSRet());
Daniel Dunbar7633cbf2009-02-02 21:43:58 +00001507}
1508
Tim Northovere77cc392014-03-29 13:28:05 +00001509bool CodeGenModule::ReturnSlotInterferesWithArgs(const CGFunctionInfo &FI) {
1510 return ReturnTypeUsesSRet(FI) &&
1511 getTargetCodeGenInfo().doesReturnSlotInterfereWithArgs();
1512}
1513
Daniel Dunbar6f2e8392010-07-14 23:39:36 +00001514bool CodeGenModule::ReturnTypeUsesFPRet(QualType ResultType) {
1515 if (const BuiltinType *BT = ResultType->getAs<BuiltinType>()) {
1516 switch (BT->getKind()) {
1517 default:
1518 return false;
1519 case BuiltinType::Float:
John McCallc8e01702013-04-16 22:48:15 +00001520 return getTarget().useObjCFPRetForRealType(TargetInfo::Float);
Daniel Dunbar6f2e8392010-07-14 23:39:36 +00001521 case BuiltinType::Double:
John McCallc8e01702013-04-16 22:48:15 +00001522 return getTarget().useObjCFPRetForRealType(TargetInfo::Double);
Daniel Dunbar6f2e8392010-07-14 23:39:36 +00001523 case BuiltinType::LongDouble:
John McCallc8e01702013-04-16 22:48:15 +00001524 return getTarget().useObjCFPRetForRealType(TargetInfo::LongDouble);
Daniel Dunbar6f2e8392010-07-14 23:39:36 +00001525 }
1526 }
1527
1528 return false;
1529}
1530
Anders Carlsson2f1a6c32011-10-31 16:27:11 +00001531bool CodeGenModule::ReturnTypeUsesFP2Ret(QualType ResultType) {
1532 if (const ComplexType *CT = ResultType->getAs<ComplexType>()) {
1533 if (const BuiltinType *BT = CT->getElementType()->getAs<BuiltinType>()) {
1534 if (BT->getKind() == BuiltinType::LongDouble)
John McCallc8e01702013-04-16 22:48:15 +00001535 return getTarget().useObjCFP2RetForComplexLongDouble();
Anders Carlsson2f1a6c32011-10-31 16:27:11 +00001536 }
1537 }
1538
1539 return false;
1540}
1541
Chris Lattnera5f58b02011-07-09 17:41:47 +00001542llvm::FunctionType *CodeGenTypes::GetFunctionType(GlobalDecl GD) {
John McCalla729c622012-02-17 03:33:10 +00001543 const CGFunctionInfo &FI = arrangeGlobalDeclaration(GD);
1544 return GetFunctionType(FI);
John McCallf8ff7b92010-02-23 00:48:20 +00001545}
1546
Chris Lattnera5f58b02011-07-09 17:41:47 +00001547llvm::FunctionType *
John McCalla729c622012-02-17 03:33:10 +00001548CodeGenTypes::GetFunctionType(const CGFunctionInfo &FI) {
Alexey Samsonov153004f2014-09-29 22:08:00 +00001549
David Blaikie82e95a32014-11-19 07:49:47 +00001550 bool Inserted = FunctionsBeingProcessed.insert(&FI).second;
1551 (void)Inserted;
Chris Lattner6fb0ccf2011-07-15 05:16:14 +00001552 assert(Inserted && "Recursively being processed?");
Daniel Dunbar7a95ca32008-09-10 04:01:49 +00001553
Alexey Samsonov153004f2014-09-29 22:08:00 +00001554 llvm::Type *resultType = nullptr;
John McCall85dd2c52011-05-15 02:19:42 +00001555 const ABIArgInfo &retAI = FI.getReturnInfo();
1556 switch (retAI.getKind()) {
Daniel Dunbard3674e62008-09-11 01:48:57 +00001557 case ABIArgInfo::Expand:
John McCall85dd2c52011-05-15 02:19:42 +00001558 llvm_unreachable("Invalid ABI kind for return argument");
Daniel Dunbard3674e62008-09-11 01:48:57 +00001559
Anton Korobeynikov18adbf52009-06-06 09:36:29 +00001560 case ABIArgInfo::Extend:
Daniel Dunbar67dace892009-02-03 06:17:37 +00001561 case ABIArgInfo::Direct:
John McCall85dd2c52011-05-15 02:19:42 +00001562 resultType = retAI.getCoerceToType();
Daniel Dunbar67dace892009-02-03 06:17:37 +00001563 break;
1564
Reid Kleckner314ef7b2014-02-01 00:04:45 +00001565 case ABIArgInfo::InAlloca:
Reid Klecknerfab1e892014-02-25 00:59:14 +00001566 if (retAI.getInAllocaSRet()) {
1567 // sret things on win32 aren't void, they return the sret pointer.
1568 QualType ret = FI.getReturnType();
1569 llvm::Type *ty = ConvertType(ret);
1570 unsigned addressSpace = Context.getTargetAddressSpace(ret);
1571 resultType = llvm::PointerType::get(ty, addressSpace);
1572 } else {
1573 resultType = llvm::Type::getVoidTy(getLLVMContext());
1574 }
Reid Kleckner314ef7b2014-02-01 00:04:45 +00001575 break;
1576
John McCall7f416cc2015-09-08 08:05:57 +00001577 case ABIArgInfo::Indirect:
Daniel Dunbar94a6f252009-01-26 21:26:08 +00001578 case ABIArgInfo::Ignore:
John McCall85dd2c52011-05-15 02:19:42 +00001579 resultType = llvm::Type::getVoidTy(getLLVMContext());
Daniel Dunbar94a6f252009-01-26 21:26:08 +00001580 break;
John McCallf26e73d2016-03-11 04:30:43 +00001581
1582 case ABIArgInfo::CoerceAndExpand:
1583 resultType = retAI.getUnpaddedCoerceAndExpandType();
1584 break;
Daniel Dunbar7a95ca32008-09-10 04:01:49 +00001585 }
Mike Stump11289f42009-09-09 15:08:12 +00001586
Alexey Samsonov153004f2014-09-29 22:08:00 +00001587 ClangToLLVMArgMapping IRFunctionArgs(getContext(), FI, true);
1588 SmallVector<llvm::Type*, 8> ArgTypes(IRFunctionArgs.totalIRArgs());
1589
1590 // Add type for sret argument.
1591 if (IRFunctionArgs.hasSRetArg()) {
1592 QualType Ret = FI.getReturnType();
1593 llvm::Type *Ty = ConvertType(Ret);
1594 unsigned AddressSpace = Context.getTargetAddressSpace(Ret);
1595 ArgTypes[IRFunctionArgs.getSRetArgNo()] =
1596 llvm::PointerType::get(Ty, AddressSpace);
1597 }
1598
1599 // Add type for inalloca argument.
1600 if (IRFunctionArgs.hasInallocaArg()) {
1601 auto ArgStruct = FI.getArgStruct();
1602 assert(ArgStruct);
1603 ArgTypes[IRFunctionArgs.getInallocaArgNo()] = ArgStruct->getPointerTo();
1604 }
1605
John McCallc818bbb2012-12-07 07:03:17 +00001606 // Add in all of the required arguments.
Alexey Samsonov153004f2014-09-29 22:08:00 +00001607 unsigned ArgNo = 0;
Alexey Samsonov34625dd2014-09-29 21:21:48 +00001608 CGFunctionInfo::const_arg_iterator it = FI.arg_begin(),
1609 ie = it + FI.getNumRequiredArgs();
Alexey Samsonov153004f2014-09-29 22:08:00 +00001610 for (; it != ie; ++it, ++ArgNo) {
1611 const ABIArgInfo &ArgInfo = it->info;
Mike Stump11289f42009-09-09 15:08:12 +00001612
Rafael Espindolafad28de2012-10-24 01:59:00 +00001613 // Insert a padding type to ensure proper alignment.
Alexey Samsonov153004f2014-09-29 22:08:00 +00001614 if (IRFunctionArgs.hasPaddingArg(ArgNo))
1615 ArgTypes[IRFunctionArgs.getPaddingArgNo(ArgNo)] =
1616 ArgInfo.getPaddingType();
Rafael Espindolafad28de2012-10-24 01:59:00 +00001617
Alexey Samsonov153004f2014-09-29 22:08:00 +00001618 unsigned FirstIRArg, NumIRArgs;
1619 std::tie(FirstIRArg, NumIRArgs) = IRFunctionArgs.getIRArgs(ArgNo);
1620
1621 switch (ArgInfo.getKind()) {
Daniel Dunbar94a6f252009-01-26 21:26:08 +00001622 case ABIArgInfo::Ignore:
Reid Kleckner314ef7b2014-02-01 00:04:45 +00001623 case ABIArgInfo::InAlloca:
Alexey Samsonov153004f2014-09-29 22:08:00 +00001624 assert(NumIRArgs == 0);
Daniel Dunbar94a6f252009-01-26 21:26:08 +00001625 break;
1626
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00001627 case ABIArgInfo::Indirect: {
Alexey Samsonov153004f2014-09-29 22:08:00 +00001628 assert(NumIRArgs == 1);
Yaxun Liud7523282017-04-17 20:10:44 +00001629 // indirect arguments are always on the stack, which is alloca addr space.
Chris Lattner2192fe52011-07-18 04:24:23 +00001630 llvm::Type *LTy = ConvertTypeForMem(it->type);
Yaxun Liud7523282017-04-17 20:10:44 +00001631 ArgTypes[FirstIRArg] = LTy->getPointerTo(
1632 CGM.getDataLayout().getAllocaAddrSpace());
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00001633 break;
1634 }
1635
1636 case ABIArgInfo::Extend:
Chris Lattner2cdfda42010-07-29 06:44:09 +00001637 case ABIArgInfo::Direct: {
Oliver Stannard2bfdc5b2014-08-27 10:43:15 +00001638 // Fast-isel and the optimizer generally like scalar values better than
1639 // FCAs, so we flatten them if this is safe to do for this argument.
Alexey Samsonov153004f2014-09-29 22:08:00 +00001640 llvm::Type *argType = ArgInfo.getCoerceToType();
James Molloy6f244b62014-05-09 16:21:39 +00001641 llvm::StructType *st = dyn_cast<llvm::StructType>(argType);
Alexey Samsonov153004f2014-09-29 22:08:00 +00001642 if (st && ArgInfo.isDirect() && ArgInfo.getCanBeFlattened()) {
1643 assert(NumIRArgs == st->getNumElements());
John McCall85dd2c52011-05-15 02:19:42 +00001644 for (unsigned i = 0, e = st->getNumElements(); i != e; ++i)
Alexey Samsonov153004f2014-09-29 22:08:00 +00001645 ArgTypes[FirstIRArg + i] = st->getElementType(i);
Chris Lattner3dd716c2010-06-28 23:44:11 +00001646 } else {
Alexey Samsonov153004f2014-09-29 22:08:00 +00001647 assert(NumIRArgs == 1);
1648 ArgTypes[FirstIRArg] = argType;
Chris Lattner3dd716c2010-06-28 23:44:11 +00001649 }
Daniel Dunbar2f219b02009-02-03 19:12:28 +00001650 break;
Chris Lattner2cdfda42010-07-29 06:44:09 +00001651 }
Mike Stump11289f42009-09-09 15:08:12 +00001652
John McCallf26e73d2016-03-11 04:30:43 +00001653 case ABIArgInfo::CoerceAndExpand: {
1654 auto ArgTypesIter = ArgTypes.begin() + FirstIRArg;
1655 for (auto EltTy : ArgInfo.getCoerceAndExpandTypeSequence()) {
1656 *ArgTypesIter++ = EltTy;
1657 }
1658 assert(ArgTypesIter == ArgTypes.begin() + FirstIRArg + NumIRArgs);
1659 break;
1660 }
1661
Daniel Dunbard3674e62008-09-11 01:48:57 +00001662 case ABIArgInfo::Expand:
Alexey Samsonov153004f2014-09-29 22:08:00 +00001663 auto ArgTypesIter = ArgTypes.begin() + FirstIRArg;
1664 getExpandedTypes(it->type, ArgTypesIter);
1665 assert(ArgTypesIter == ArgTypes.begin() + FirstIRArg + NumIRArgs);
Daniel Dunbard3674e62008-09-11 01:48:57 +00001666 break;
1667 }
Daniel Dunbar7a95ca32008-09-10 04:01:49 +00001668 }
1669
Chris Lattner6fb0ccf2011-07-15 05:16:14 +00001670 bool Erased = FunctionsBeingProcessed.erase(&FI); (void)Erased;
1671 assert(Erased && "Not in set?");
Alexey Samsonov153004f2014-09-29 22:08:00 +00001672
1673 return llvm::FunctionType::get(resultType, ArgTypes, FI.isVariadic());
Daniel Dunbar81cf67f2008-09-09 23:48:28 +00001674}
1675
Chris Lattner2192fe52011-07-18 04:24:23 +00001676llvm::Type *CodeGenTypes::GetFunctionTypeForVTable(GlobalDecl GD) {
John McCall5d865c322010-08-31 07:33:07 +00001677 const CXXMethodDecl *MD = cast<CXXMethodDecl>(GD.getDecl());
Anders Carlsson64457732009-11-24 05:08:52 +00001678 const FunctionProtoType *FPT = MD->getType()->getAs<FunctionProtoType>();
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00001679
Chris Lattner8806e322011-07-10 00:18:59 +00001680 if (!isFuncTypeConvertible(FPT))
1681 return llvm::StructType::get(getLLVMContext());
Fangrui Song6907ce22018-07-30 19:24:48 +00001682
Chris Lattner8806e322011-07-10 00:18:59 +00001683 const CGFunctionInfo *Info;
1684 if (isa<CXXDestructorDecl>(MD))
Rafael Espindola8d2a19b2014-09-08 16:01:27 +00001685 Info =
1686 &arrangeCXXStructorDeclaration(MD, getFromDtorType(GD.getDtorType()));
Chris Lattner8806e322011-07-10 00:18:59 +00001687 else
John McCalla729c622012-02-17 03:33:10 +00001688 Info = &arrangeCXXMethodDeclaration(MD);
1689 return GetFunctionType(*Info);
Anders Carlsson64457732009-11-24 05:08:52 +00001690}
1691
Samuel Antao798f11c2015-11-23 22:04:44 +00001692static void AddAttributesFromFunctionProtoType(ASTContext &Ctx,
1693 llvm::AttrBuilder &FuncAttrs,
1694 const FunctionProtoType *FPT) {
1695 if (!FPT)
1696 return;
1697
1698 if (!isUnresolvedExceptionSpec(FPT->getExceptionSpecType()) &&
Richard Smitheaf11ad2018-05-03 03:58:32 +00001699 FPT->isNothrow())
Samuel Antao798f11c2015-11-23 22:04:44 +00001700 FuncAttrs.addAttribute(llvm::Attribute::NoUnwind);
1701}
1702
Justin Lebarb080b632017-01-25 21:29:48 +00001703void CodeGenModule::ConstructDefaultFnAttrList(StringRef Name, bool HasOptnone,
1704 bool AttrOnCallSite,
1705 llvm::AttrBuilder &FuncAttrs) {
1706 // OptimizeNoneAttr takes precedence over -Os or -Oz. No warning needed.
1707 if (!HasOptnone) {
1708 if (CodeGenOpts.OptimizeSize)
1709 FuncAttrs.addAttribute(llvm::Attribute::OptimizeForSize);
1710 if (CodeGenOpts.OptimizeSize == 2)
1711 FuncAttrs.addAttribute(llvm::Attribute::MinSize);
1712 }
1713
1714 if (CodeGenOpts.DisableRedZone)
1715 FuncAttrs.addAttribute(llvm::Attribute::NoRedZone);
Kristina Brooks7f569b72018-10-18 14:07:02 +00001716 if (CodeGenOpts.IndirectTlsSegRefs)
1717 FuncAttrs.addAttribute("indirect-tls-seg-refs");
Justin Lebarb080b632017-01-25 21:29:48 +00001718 if (CodeGenOpts.NoImplicitFloat)
1719 FuncAttrs.addAttribute(llvm::Attribute::NoImplicitFloat);
1720
1721 if (AttrOnCallSite) {
1722 // Attributes that should go on the call site only.
1723 if (!CodeGenOpts.SimplifyLibCalls ||
1724 CodeGenOpts.isNoBuiltinFunc(Name.data()))
1725 FuncAttrs.addAttribute(llvm::Attribute::NoBuiltin);
1726 if (!CodeGenOpts.TrapFuncName.empty())
1727 FuncAttrs.addAttribute("trap-func-name", CodeGenOpts.TrapFuncName);
1728 } else {
1729 // Attributes that should go on the function, but not the call site.
1730 if (!CodeGenOpts.DisableFPElim) {
1731 FuncAttrs.addAttribute("no-frame-pointer-elim", "false");
1732 } else if (CodeGenOpts.OmitLeafFramePointer) {
1733 FuncAttrs.addAttribute("no-frame-pointer-elim", "false");
1734 FuncAttrs.addAttribute("no-frame-pointer-elim-non-leaf");
1735 } else {
1736 FuncAttrs.addAttribute("no-frame-pointer-elim", "true");
1737 FuncAttrs.addAttribute("no-frame-pointer-elim-non-leaf");
1738 }
1739
1740 FuncAttrs.addAttribute("less-precise-fpmad",
1741 llvm::toStringRef(CodeGenOpts.LessPreciseFPMAD));
1742
Manoj Guptada08f6a2018-07-19 00:44:52 +00001743 if (CodeGenOpts.NullPointerIsValid)
1744 FuncAttrs.addAttribute("null-pointer-is-valid", "true");
Justin Lebarb080b632017-01-25 21:29:48 +00001745 if (!CodeGenOpts.FPDenormalMode.empty())
1746 FuncAttrs.addAttribute("denormal-fp-math", CodeGenOpts.FPDenormalMode);
1747
1748 FuncAttrs.addAttribute("no-trapping-math",
1749 llvm::toStringRef(CodeGenOpts.NoTrappingMath));
1750
Sanjay Patelc81450e2018-04-30 18:19:03 +00001751 // Strict (compliant) code is the default, so only add this attribute to
1752 // indicate that we are trying to workaround a problem case.
1753 if (!CodeGenOpts.StrictFloatCastOverflow)
1754 FuncAttrs.addAttribute("strict-float-cast-overflow", "false");
Sanjay Pateld1754762018-04-27 14:22:48 +00001755
Justin Lebarb080b632017-01-25 21:29:48 +00001756 // TODO: Are these all needed?
1757 // unsafe/inf/nan/nsz are handled by instruction-level FastMathFlags.
1758 FuncAttrs.addAttribute("no-infs-fp-math",
1759 llvm::toStringRef(CodeGenOpts.NoInfsFPMath));
1760 FuncAttrs.addAttribute("no-nans-fp-math",
1761 llvm::toStringRef(CodeGenOpts.NoNaNsFPMath));
1762 FuncAttrs.addAttribute("unsafe-fp-math",
1763 llvm::toStringRef(CodeGenOpts.UnsafeFPMath));
1764 FuncAttrs.addAttribute("use-soft-float",
1765 llvm::toStringRef(CodeGenOpts.SoftFloat));
1766 FuncAttrs.addAttribute("stack-protector-buffer-size",
1767 llvm::utostr(CodeGenOpts.SSPBufferSize));
1768 FuncAttrs.addAttribute("no-signed-zeros-fp-math",
1769 llvm::toStringRef(CodeGenOpts.NoSignedZeros));
1770 FuncAttrs.addAttribute(
1771 "correctly-rounded-divide-sqrt-fp-math",
1772 llvm::toStringRef(CodeGenOpts.CorrectlyRoundedDivSqrt));
1773
Alexey Sotkin3858e262018-04-20 08:08:04 +00001774 if (getLangOpts().OpenCL)
1775 FuncAttrs.addAttribute("denorms-are-zero",
1776 llvm::toStringRef(CodeGenOpts.FlushDenorm));
1777
Justin Lebarb080b632017-01-25 21:29:48 +00001778 // TODO: Reciprocal estimate codegen options should apply to instructions?
Craig Topper402b4312017-11-20 17:09:22 +00001779 const std::vector<std::string> &Recips = CodeGenOpts.Reciprocals;
Justin Lebarb080b632017-01-25 21:29:48 +00001780 if (!Recips.empty())
1781 FuncAttrs.addAttribute("reciprocal-estimates",
Erich Keane857ac592017-10-27 18:45:06 +00001782 llvm::join(Recips, ","));
Justin Lebarb080b632017-01-25 21:29:48 +00001783
Craig Topper9a724aa2017-12-11 21:09:19 +00001784 if (!CodeGenOpts.PreferVectorWidth.empty() &&
1785 CodeGenOpts.PreferVectorWidth != "none")
1786 FuncAttrs.addAttribute("prefer-vector-width",
1787 CodeGenOpts.PreferVectorWidth);
1788
Justin Lebarb080b632017-01-25 21:29:48 +00001789 if (CodeGenOpts.StackRealignment)
1790 FuncAttrs.addAttribute("stackrealign");
1791 if (CodeGenOpts.Backchain)
1792 FuncAttrs.addAttribute("backchain");
Chandler Carruth664aa862018-09-04 12:38:00 +00001793
1794 if (CodeGenOpts.SpeculativeLoadHardening)
1795 FuncAttrs.addAttribute(llvm::Attribute::SpeculativeLoadHardening);
Justin Lebarb080b632017-01-25 21:29:48 +00001796 }
1797
Matt Arsenaulta1cf61b2017-10-06 19:34:40 +00001798 if (getLangOpts().assumeFunctionsAreConvergent()) {
1799 // Conservatively, mark all functions and calls in CUDA and OpenCL as
1800 // convergent (meaning, they may call an intrinsically convergent op, such
1801 // as __syncthreads() / barrier(), and so can't have certain optimizations
1802 // applied around them). LLVM will remove this attribute where it safely
1803 // can.
Justin Lebarb080b632017-01-25 21:29:48 +00001804 FuncAttrs.addAttribute(llvm::Attribute::Convergent);
Matt Arsenaulta1cf61b2017-10-06 19:34:40 +00001805 }
Justin Lebarb080b632017-01-25 21:29:48 +00001806
Matt Arsenaulta1cf61b2017-10-06 19:34:40 +00001807 if (getLangOpts().CUDA && getLangOpts().CUDAIsDevice) {
Justin Lebarb080b632017-01-25 21:29:48 +00001808 // Exceptions aren't supported in CUDA device code.
1809 FuncAttrs.addAttribute(llvm::Attribute::NoUnwind);
1810
1811 // Respect -fcuda-flush-denormals-to-zero.
Yaxun Liue1bfbc52018-07-21 02:02:22 +00001812 if (CodeGenOpts.FlushDenorm)
Justin Lebarb080b632017-01-25 21:29:48 +00001813 FuncAttrs.addAttribute("nvptx-f32ftz", "true");
1814 }
Peter Collingbourne87f477b2019-01-04 19:27:04 +00001815
1816 for (StringRef Attr : CodeGenOpts.DefaultFunctionAttrs) {
1817 StringRef Var, Value;
1818 std::tie(Var, Value) = Attr.split('=');
1819 FuncAttrs.addAttribute(Var, Value);
1820 }
Justin Lebarb080b632017-01-25 21:29:48 +00001821}
1822
1823void CodeGenModule::AddDefaultFnAttrs(llvm::Function &F) {
1824 llvm::AttrBuilder FuncAttrs;
1825 ConstructDefaultFnAttrList(F.getName(),
1826 F.hasFnAttribute(llvm::Attribute::OptimizeNone),
1827 /* AttrOnCallsite = */ false, FuncAttrs);
Reid Kleckneree4930b2017-05-02 22:07:37 +00001828 F.addAttributes(llvm::AttributeList::FunctionIndex, FuncAttrs);
Justin Lebarb080b632017-01-25 21:29:48 +00001829}
1830
Chad Rosier7dbc9cf2016-01-06 14:35:46 +00001831void CodeGenModule::ConstructAttributeList(
1832 StringRef Name, const CGFunctionInfo &FI, CGCalleeInfo CalleeInfo,
Reid Klecknercdd26792017-04-18 23:50:03 +00001833 llvm::AttributeList &AttrList, unsigned &CallingConv, bool AttrOnCallSite) {
Bill Wendlinga514ebc2012-10-15 20:36:26 +00001834 llvm::AttrBuilder FuncAttrs;
1835 llvm::AttrBuilder RetAttrs;
Daniel Dunbar76c8eb72008-09-10 00:32:18 +00001836
Daniel Dunbar0ef34792009-09-12 00:59:20 +00001837 CallingConv = FI.getEffectiveCallingConvention();
John McCallab26cfa2010-02-05 21:31:56 +00001838 if (FI.isNoReturn())
Bill Wendling207f0532012-12-20 19:27:06 +00001839 FuncAttrs.addAttribute(llvm::Attribute::NoReturn);
John McCallab26cfa2010-02-05 21:31:56 +00001840
Samuel Antao798f11c2015-11-23 22:04:44 +00001841 // If we have information about the function prototype, we can learn
Craig Topper13d759f2018-04-30 22:02:48 +00001842 // attributes from there.
Samuel Antao798f11c2015-11-23 22:04:44 +00001843 AddAttributesFromFunctionProtoType(getContext(), FuncAttrs,
1844 CalleeInfo.getCalleeFunctionProtoType());
1845
Erich Keanede6480a32018-11-13 15:48:08 +00001846 const Decl *TargetDecl = CalleeInfo.getCalleeDecl().getDecl();
Samuel Antao798f11c2015-11-23 22:04:44 +00001847
Justin Lebarb080b632017-01-25 21:29:48 +00001848 bool HasOptnone = false;
Anton Korobeynikovc8478242009-04-04 00:49:24 +00001849 // FIXME: handle sseregparm someday...
Daniel Dunbar76c8eb72008-09-10 00:32:18 +00001850 if (TargetDecl) {
Rafael Espindola2d21ab02011-10-12 19:51:18 +00001851 if (TargetDecl->hasAttr<ReturnsTwiceAttr>())
Bill Wendling207f0532012-12-20 19:27:06 +00001852 FuncAttrs.addAttribute(llvm::Attribute::ReturnsTwice);
Argyrios Kyrtzidisb4b64ca2009-06-30 02:34:44 +00001853 if (TargetDecl->hasAttr<NoThrowAttr>())
Bill Wendling207f0532012-12-20 19:27:06 +00001854 FuncAttrs.addAttribute(llvm::Attribute::NoUnwind);
Richard Smithdebc59d2013-01-30 05:45:05 +00001855 if (TargetDecl->hasAttr<NoReturnAttr>())
1856 FuncAttrs.addAttribute(llvm::Attribute::NoReturn);
Xinliang David Li4ec36062017-06-13 21:14:07 +00001857 if (TargetDecl->hasAttr<ColdAttr>())
1858 FuncAttrs.addAttribute(llvm::Attribute::Cold);
Aaron Ballman7c19ab12014-02-22 16:59:24 +00001859 if (TargetDecl->hasAttr<NoDuplicateAttr>())
1860 FuncAttrs.addAttribute(llvm::Attribute::NoDuplicate);
Yaxun Liu7d07ae72016-11-01 18:45:32 +00001861 if (TargetDecl->hasAttr<ConvergentAttr>())
1862 FuncAttrs.addAttribute(llvm::Attribute::Convergent);
Richard Smithdebc59d2013-01-30 05:45:05 +00001863
Chandler Carruth45bbe012017-03-24 09:11:57 +00001864 if (const FunctionDecl *Fn = dyn_cast<FunctionDecl>(TargetDecl)) {
Samuel Antao798f11c2015-11-23 22:04:44 +00001865 AddAttributesFromFunctionProtoType(
Chandler Carruth45bbe012017-03-24 09:11:57 +00001866 getContext(), FuncAttrs, Fn->getType()->getAs<FunctionProtoType>());
Richard Smith49af6292013-03-05 08:30:04 +00001867 // Don't use [[noreturn]] or _Noreturn for a call to a virtual function.
1868 // These attributes are not inherited by overloads.
Chandler Carruth45bbe012017-03-24 09:11:57 +00001869 const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(Fn);
1870 if (Fn->isNoReturn() && !(AttrOnCallSite && MD && MD->isVirtual()))
Richard Smithdebc59d2013-01-30 05:45:05 +00001871 FuncAttrs.addAttribute(llvm::Attribute::NoReturn);
John McCallbe349de2010-07-08 06:48:12 +00001872 }
1873
David Majnemer1bf0f8e2015-07-20 22:51:52 +00001874 // 'const', 'pure' and 'noalias' attributed functions are also nounwind.
Eric Christopherbf005ec2011-08-15 22:38:22 +00001875 if (TargetDecl->hasAttr<ConstAttr>()) {
Bill Wendling207f0532012-12-20 19:27:06 +00001876 FuncAttrs.addAttribute(llvm::Attribute::ReadNone);
1877 FuncAttrs.addAttribute(llvm::Attribute::NoUnwind);
Eric Christopherbf005ec2011-08-15 22:38:22 +00001878 } else if (TargetDecl->hasAttr<PureAttr>()) {
Bill Wendling207f0532012-12-20 19:27:06 +00001879 FuncAttrs.addAttribute(llvm::Attribute::ReadOnly);
1880 FuncAttrs.addAttribute(llvm::Attribute::NoUnwind);
David Majnemer1bf0f8e2015-07-20 22:51:52 +00001881 } else if (TargetDecl->hasAttr<NoAliasAttr>()) {
1882 FuncAttrs.addAttribute(llvm::Attribute::ArgMemOnly);
1883 FuncAttrs.addAttribute(llvm::Attribute::NoUnwind);
Eric Christopherbf005ec2011-08-15 22:38:22 +00001884 }
David Majnemer631a90b2015-02-04 07:23:21 +00001885 if (TargetDecl->hasAttr<RestrictAttr>())
Bill Wendling207f0532012-12-20 19:27:06 +00001886 RetAttrs.addAttribute(llvm::Attribute::NoAlias);
Manoj Guptada08f6a2018-07-19 00:44:52 +00001887 if (TargetDecl->hasAttr<ReturnsNonNullAttr>() &&
1888 !CodeGenOpts.NullPointerIsValid)
Hal Finkeld8442b12014-07-12 04:51:04 +00001889 RetAttrs.addAttribute(llvm::Attribute::NonNull);
Oren Ben Simhon318a6ea2017-04-27 12:01:00 +00001890 if (TargetDecl->hasAttr<AnyX86NoCallerSavedRegistersAttr>())
1891 FuncAttrs.addAttribute("no_caller_saved_registers");
Oren Ben Simhon220671a2018-03-17 13:31:35 +00001892 if (TargetDecl->hasAttr<AnyX86NoCfCheckAttr>())
1893 FuncAttrs.addAttribute(llvm::Attribute::NoCfCheck);
Paul Robinson08556952014-12-11 20:14:04 +00001894
1895 HasOptnone = TargetDecl->hasAttr<OptimizeNoneAttr>();
George Burgess IVe3763372016-12-22 02:50:20 +00001896 if (auto *AllocSize = TargetDecl->getAttr<AllocSizeAttr>()) {
1897 Optional<unsigned> NumElemsParam;
Joel E. Denny81508102018-03-13 14:51:22 +00001898 if (AllocSize->getNumElemsParam().isValid())
1899 NumElemsParam = AllocSize->getNumElemsParam().getLLVMIndex();
1900 FuncAttrs.addAllocSizeAttr(AllocSize->getElemSizeParam().getLLVMIndex(),
George Burgess IVe3763372016-12-22 02:50:20 +00001901 NumElemsParam);
1902 }
Daniel Dunbar76c8eb72008-09-10 00:32:18 +00001903 }
1904
Justin Lebarb080b632017-01-25 21:29:48 +00001905 ConstructDefaultFnAttrList(Name, HasOptnone, AttrOnCallSite, FuncAttrs);
Paul Robinson08556952014-12-11 20:14:04 +00001906
Zola Bridges826ef592019-01-18 17:20:46 +00001907 // This must run after constructing the default function attribute list
1908 // to ensure that the speculative load hardening attribute is removed
1909 // in the case where the -mspeculative-load-hardening flag was passed.
1910 if (TargetDecl) {
1911 if (TargetDecl->hasAttr<NoSpeculativeLoadHardeningAttr>())
1912 FuncAttrs.removeAttribute(llvm::Attribute::SpeculativeLoadHardening);
1913 if (TargetDecl->hasAttr<SpeculativeLoadHardeningAttr>())
1914 FuncAttrs.addAttribute(llvm::Attribute::SpeculativeLoadHardening);
1915 }
1916
Peter Collingbourneb4728c12014-05-19 22:14:34 +00001917 if (CodeGenOpts.EnableSegmentedStacks &&
1918 !(TargetDecl && TargetDecl->hasAttr<NoSplitStackAttr>()))
Reid Klecknerfb873af2014-04-10 22:59:13 +00001919 FuncAttrs.addAttribute("split-stack");
Devang Patel6e467b12009-06-04 23:32:02 +00001920
Sriraman Tallam5c651482017-11-07 19:37:51 +00001921 // Add NonLazyBind attribute to function declarations when -fno-plt
1922 // is used.
1923 if (TargetDecl && CodeGenOpts.NoPLT) {
1924 if (auto *Fn = dyn_cast<FunctionDecl>(TargetDecl)) {
1925 if (!Fn->isDefined() && !AttrOnCallSite) {
1926 FuncAttrs.addAttribute(llvm::Attribute::NonLazyBind);
1927 }
1928 }
1929 }
1930
Alexey Sotkin20f65922018-02-22 11:54:14 +00001931 if (TargetDecl && TargetDecl->hasAttr<OpenCLKernelAttr>()) {
1932 if (getLangOpts().OpenCLVersion <= 120) {
1933 // OpenCL v1.2 Work groups are always uniform
1934 FuncAttrs.addAttribute("uniform-work-group-size", "true");
1935 } else {
1936 // OpenCL v2.0 Work groups may be whether uniform or not.
1937 // '-cl-uniform-work-group-size' compile option gets a hint
1938 // to the compiler that the global work-size be a multiple of
1939 // the work-group size specified to clEnqueueNDRangeKernel
1940 // (i.e. work groups are uniform).
1941 FuncAttrs.addAttribute("uniform-work-group-size",
1942 llvm::toStringRef(CodeGenOpts.UniformWGSize));
1943 }
1944 }
1945
Justin Lebarb080b632017-01-25 21:29:48 +00001946 if (!AttrOnCallSite) {
Akira Hatanaka627586b2018-03-02 01:53:15 +00001947 bool DisableTailCalls = false;
1948
1949 if (CodeGenOpts.DisableTailCalls)
1950 DisableTailCalls = true;
1951 else if (TargetDecl) {
1952 if (TargetDecl->hasAttr<DisableTailCallsAttr>() ||
1953 TargetDecl->hasAttr<AnyX86InterruptAttr>())
1954 DisableTailCalls = true;
1955 else if (CodeGenOpts.NoEscapingBlockTailCalls) {
1956 if (const auto *BD = dyn_cast<BlockDecl>(TargetDecl))
1957 if (!BD->doesNotEscape())
1958 DisableTailCalls = true;
1959 }
1960 }
1961
Justin Lebarb080b632017-01-25 21:29:48 +00001962 FuncAttrs.addAttribute("disable-tail-calls",
1963 llvm::toStringRef(DisableTailCalls));
Erich Keanede6480a32018-11-13 15:48:08 +00001964 GetCPUAndFeaturesAttributes(CalleeInfo.getCalleeDecl(), FuncAttrs);
Bill Wendling985d1c52013-02-15 21:30:01 +00001965 }
1966
Alexey Samsonov153004f2014-09-29 22:08:00 +00001967 ClangToLLVMArgMapping IRFunctionArgs(getContext(), FI);
Alexey Samsonov91cf4552014-08-22 01:06:06 +00001968
Daniel Dunbar3668cb22009-02-02 23:43:58 +00001969 QualType RetTy = FI.getReturnType();
Daniel Dunbarb52d0772009-02-03 05:59:18 +00001970 const ABIArgInfo &RetAI = FI.getReturnInfo();
Daniel Dunbar7a95ca32008-09-10 04:01:49 +00001971 switch (RetAI.getKind()) {
Anton Korobeynikov18adbf52009-06-06 09:36:29 +00001972 case ABIArgInfo::Extend:
Alex Bradburye41a5e22018-01-12 20:08:16 +00001973 if (RetAI.isSignExt())
Jakob Stoklund Olesend7bf2932013-05-29 03:57:23 +00001974 RetAttrs.addAttribute(llvm::Attribute::SExt);
Alex Bradburye41a5e22018-01-12 20:08:16 +00001975 else
Jakob Stoklund Olesend7bf2932013-05-29 03:57:23 +00001976 RetAttrs.addAttribute(llvm::Attribute::ZExt);
Adrian Prantlf3b3ccd2017-12-19 22:06:11 +00001977 LLVM_FALLTHROUGH;
Daniel Dunbar67dace892009-02-03 06:17:37 +00001978 case ABIArgInfo::Direct:
Jakob Stoklund Olesena3661142013-06-05 03:00:09 +00001979 if (RetAI.getInReg())
1980 RetAttrs.addAttribute(llvm::Attribute::InReg);
1981 break;
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00001982 case ABIArgInfo::Ignore:
Daniel Dunbara72d4ae2008-09-10 02:41:04 +00001983 break;
1984
Alexey Samsonov91cf4552014-08-22 01:06:06 +00001985 case ABIArgInfo::InAlloca:
Rafael Espindola06b2b4a2012-07-31 02:44:24 +00001986 case ABIArgInfo::Indirect: {
Alexey Samsonov91cf4552014-08-22 01:06:06 +00001987 // inalloca and sret disable readnone and readonly
Bill Wendling207f0532012-12-20 19:27:06 +00001988 FuncAttrs.removeAttribute(llvm::Attribute::ReadOnly)
1989 .removeAttribute(llvm::Attribute::ReadNone);
Daniel Dunbara72d4ae2008-09-10 02:41:04 +00001990 break;
Rafael Espindola06b2b4a2012-07-31 02:44:24 +00001991 }
Daniel Dunbara72d4ae2008-09-10 02:41:04 +00001992
John McCallf26e73d2016-03-11 04:30:43 +00001993 case ABIArgInfo::CoerceAndExpand:
1994 break;
1995
Daniel Dunbard3674e62008-09-11 01:48:57 +00001996 case ABIArgInfo::Expand:
David Blaikie83d382b2011-09-23 05:06:16 +00001997 llvm_unreachable("Invalid ABI kind for return argument");
Daniel Dunbar76c8eb72008-09-10 00:32:18 +00001998 }
Daniel Dunbara72d4ae2008-09-10 02:41:04 +00001999
Hal Finkela2347ba2014-07-18 15:52:10 +00002000 if (const auto *RefTy = RetTy->getAs<ReferenceType>()) {
2001 QualType PTy = RefTy->getPointeeType();
David Majnemer9df56372015-09-10 21:52:00 +00002002 if (!PTy->isIncompleteType() && PTy->isConstantSizeType())
Hal Finkela2347ba2014-07-18 15:52:10 +00002003 RetAttrs.addDereferenceableAttr(getContext().getTypeSizeInChars(PTy)
2004 .getQuantity());
Manoj Guptada08f6a2018-07-19 00:44:52 +00002005 else if (getContext().getTargetAddressSpace(PTy) == 0 &&
2006 !CodeGenOpts.NullPointerIsValid)
Hal Finkela2347ba2014-07-18 15:52:10 +00002007 RetAttrs.addAttribute(llvm::Attribute::NonNull);
2008 }
Nick Lewycky9b46eb82014-05-28 09:56:42 +00002009
John McCall12f23522016-04-04 18:33:08 +00002010 bool hasUsedSRet = false;
Reid Klecknercdd26792017-04-18 23:50:03 +00002011 SmallVector<llvm::AttributeSet, 4> ArgAttrs(IRFunctionArgs.totalIRArgs());
John McCall12f23522016-04-04 18:33:08 +00002012
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002013 // Attach attributes to sret.
2014 if (IRFunctionArgs.hasSRetArg()) {
2015 llvm::AttrBuilder SRETAttrs;
Mandeep Singh Grang2a153102018-07-26 18:07:59 +00002016 if (!RetAI.getSuppressSRet())
2017 SRETAttrs.addAttribute(llvm::Attribute::StructRet);
John McCall12f23522016-04-04 18:33:08 +00002018 hasUsedSRet = true;
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002019 if (RetAI.getInReg())
2020 SRETAttrs.addAttribute(llvm::Attribute::InReg);
Reid Klecknercdd26792017-04-18 23:50:03 +00002021 ArgAttrs[IRFunctionArgs.getSRetArgNo()] =
2022 llvm::AttributeSet::get(getLLVMContext(), SRETAttrs);
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002023 }
2024
2025 // Attach attributes to inalloca argument.
2026 if (IRFunctionArgs.hasInallocaArg()) {
2027 llvm::AttrBuilder Attrs;
2028 Attrs.addAttribute(llvm::Attribute::InAlloca);
Reid Klecknercdd26792017-04-18 23:50:03 +00002029 ArgAttrs[IRFunctionArgs.getInallocaArgNo()] =
2030 llvm::AttributeSet::get(getLLVMContext(), Attrs);
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002031 }
2032
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002033 unsigned ArgNo = 0;
2034 for (CGFunctionInfo::const_arg_iterator I = FI.arg_begin(),
2035 E = FI.arg_end();
2036 I != E; ++I, ++ArgNo) {
2037 QualType ParamType = I->type;
2038 const ABIArgInfo &AI = I->info;
Bill Wendlinga514ebc2012-10-15 20:36:26 +00002039 llvm::AttrBuilder Attrs;
Anton Korobeynikovc8478242009-04-04 00:49:24 +00002040
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002041 // Add attribute for padding argument, if necessary.
2042 if (IRFunctionArgs.hasPaddingArg(ArgNo)) {
Reid Klecknercdd26792017-04-18 23:50:03 +00002043 if (AI.getPaddingInReg()) {
2044 ArgAttrs[IRFunctionArgs.getPaddingArgNo(ArgNo)] =
2045 llvm::AttributeSet::get(
2046 getLLVMContext(),
2047 llvm::AttrBuilder().addAttribute(llvm::Attribute::InReg));
2048 }
Rafael Espindolafad28de2012-10-24 01:59:00 +00002049 }
2050
John McCall39ec71f2010-03-27 00:47:27 +00002051 // 'restrict' -> 'noalias' is done in EmitFunctionProlog when we
2052 // have the corresponding parameter variable. It doesn't make
Daniel Dunbarcb2b3d02011-02-10 18:10:07 +00002053 // sense to do it here because parameters are so messed up.
Daniel Dunbard3674e62008-09-11 01:48:57 +00002054 switch (AI.getKind()) {
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002055 case ABIArgInfo::Extend:
Alex Bradburye41a5e22018-01-12 20:08:16 +00002056 if (AI.isSignExt())
Bill Wendling207f0532012-12-20 19:27:06 +00002057 Attrs.addAttribute(llvm::Attribute::SExt);
Alex Bradburye41a5e22018-01-12 20:08:16 +00002058 else
2059 Attrs.addAttribute(llvm::Attribute::ZExt);
Adrian Prantlf3b3ccd2017-12-19 22:06:11 +00002060 LLVM_FALLTHROUGH;
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002061 case ABIArgInfo::Direct:
Peter Collingbournef7706832014-12-12 23:41:25 +00002062 if (ArgNo == 0 && FI.isChainCall())
2063 Attrs.addAttribute(llvm::Attribute::Nest);
2064 else if (AI.getInReg())
Bill Wendling207f0532012-12-20 19:27:06 +00002065 Attrs.addAttribute(llvm::Attribute::InReg);
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002066 break;
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002067
James Y Knight71608572015-08-21 18:19:06 +00002068 case ABIArgInfo::Indirect: {
Rafael Espindola703c47f2012-10-19 05:04:37 +00002069 if (AI.getInReg())
Bill Wendling207f0532012-12-20 19:27:06 +00002070 Attrs.addAttribute(llvm::Attribute::InReg);
Rafael Espindola703c47f2012-10-19 05:04:37 +00002071
Anders Carlsson20759ad2009-09-16 15:53:40 +00002072 if (AI.getIndirectByVal())
Bill Wendling207f0532012-12-20 19:27:06 +00002073 Attrs.addAttribute(llvm::Attribute::ByVal);
Anders Carlsson20759ad2009-09-16 15:53:40 +00002074
John McCall7f416cc2015-09-08 08:05:57 +00002075 CharUnits Align = AI.getIndirectAlign();
James Y Knight71608572015-08-21 18:19:06 +00002076
2077 // In a byval argument, it is important that the required
2078 // alignment of the type is honored, as LLVM might be creating a
2079 // *new* stack object, and needs to know what alignment to give
2080 // it. (Sometimes it can deduce a sensible alignment on its own,
2081 // but not if clang decides it must emit a packed struct, or the
2082 // user specifies increased alignment requirements.)
2083 //
2084 // This is different from indirect *not* byval, where the object
2085 // exists already, and the align attribute is purely
2086 // informative.
John McCall7f416cc2015-09-08 08:05:57 +00002087 assert(!Align.isZero());
James Y Knight71608572015-08-21 18:19:06 +00002088
John McCall7f416cc2015-09-08 08:05:57 +00002089 // For now, only add this when we have a byval argument.
2090 // TODO: be less lazy about updating test cases.
2091 if (AI.getIndirectByVal())
2092 Attrs.addAlignmentAttr(Align.getQuantity());
Bill Wendlinga7912f82012-10-10 07:36:56 +00002093
Daniel Dunbarc2304432009-03-18 19:51:01 +00002094 // byval disables readnone and readonly.
Bill Wendling207f0532012-12-20 19:27:06 +00002095 FuncAttrs.removeAttribute(llvm::Attribute::ReadOnly)
2096 .removeAttribute(llvm::Attribute::ReadNone);
Daniel Dunbard3674e62008-09-11 01:48:57 +00002097 break;
James Y Knight71608572015-08-21 18:19:06 +00002098 }
Daniel Dunbar94a6f252009-01-26 21:26:08 +00002099 case ABIArgInfo::Ignore:
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002100 case ABIArgInfo::Expand:
John McCallf26e73d2016-03-11 04:30:43 +00002101 case ABIArgInfo::CoerceAndExpand:
2102 break;
Daniel Dunbar94a6f252009-01-26 21:26:08 +00002103
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002104 case ABIArgInfo::InAlloca:
2105 // inalloca disables readnone and readonly.
2106 FuncAttrs.removeAttribute(llvm::Attribute::ReadOnly)
2107 .removeAttribute(llvm::Attribute::ReadNone);
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002108 continue;
Daniel Dunbar76c8eb72008-09-10 00:32:18 +00002109 }
Mike Stump11289f42009-09-09 15:08:12 +00002110
Hal Finkela2347ba2014-07-18 15:52:10 +00002111 if (const auto *RefTy = ParamType->getAs<ReferenceType>()) {
2112 QualType PTy = RefTy->getPointeeType();
David Majnemer9df56372015-09-10 21:52:00 +00002113 if (!PTy->isIncompleteType() && PTy->isConstantSizeType())
Hal Finkela2347ba2014-07-18 15:52:10 +00002114 Attrs.addDereferenceableAttr(getContext().getTypeSizeInChars(PTy)
2115 .getQuantity());
Manoj Guptada08f6a2018-07-19 00:44:52 +00002116 else if (getContext().getTargetAddressSpace(PTy) == 0 &&
2117 !CodeGenOpts.NullPointerIsValid)
Hal Finkela2347ba2014-07-18 15:52:10 +00002118 Attrs.addAttribute(llvm::Attribute::NonNull);
2119 }
Nick Lewycky9b46eb82014-05-28 09:56:42 +00002120
John McCall12f23522016-04-04 18:33:08 +00002121 switch (FI.getExtParameterInfo(ArgNo).getABI()) {
2122 case ParameterABI::Ordinary:
2123 break;
2124
2125 case ParameterABI::SwiftIndirectResult: {
2126 // Add 'sret' if we haven't already used it for something, but
2127 // only if the result is void.
2128 if (!hasUsedSRet && RetTy->isVoidType()) {
2129 Attrs.addAttribute(llvm::Attribute::StructRet);
2130 hasUsedSRet = true;
2131 }
2132
2133 // Add 'noalias' in either case.
2134 Attrs.addAttribute(llvm::Attribute::NoAlias);
2135
2136 // Add 'dereferenceable' and 'alignment'.
2137 auto PTy = ParamType->getPointeeType();
2138 if (!PTy->isIncompleteType() && PTy->isConstantSizeType()) {
2139 auto info = getContext().getTypeInfoInChars(PTy);
2140 Attrs.addDereferenceableAttr(info.first.getQuantity());
2141 Attrs.addAttribute(llvm::Attribute::getWithAlignment(getLLVMContext(),
2142 info.second.getQuantity()));
2143 }
2144 break;
2145 }
2146
2147 case ParameterABI::SwiftErrorResult:
2148 Attrs.addAttribute(llvm::Attribute::SwiftError);
2149 break;
2150
2151 case ParameterABI::SwiftContext:
2152 Attrs.addAttribute(llvm::Attribute::SwiftSelf);
2153 break;
2154 }
2155
Akira Hatanaka98a49332017-09-22 00:41:05 +00002156 if (FI.getExtParameterInfo(ArgNo).isNoEscape())
2157 Attrs.addAttribute(llvm::Attribute::NoCapture);
2158
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002159 if (Attrs.hasAttributes()) {
2160 unsigned FirstIRArg, NumIRArgs;
2161 std::tie(FirstIRArg, NumIRArgs) = IRFunctionArgs.getIRArgs(ArgNo);
2162 for (unsigned i = 0; i < NumIRArgs; i++)
Reid Klecknercdd26792017-04-18 23:50:03 +00002163 ArgAttrs[FirstIRArg + i] =
2164 llvm::AttributeSet::get(getLLVMContext(), Attrs);
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002165 }
Daniel Dunbar76c8eb72008-09-10 00:32:18 +00002166 }
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002167 assert(ArgNo == FI.arg_size());
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002168
Reid Klecknercdd26792017-04-18 23:50:03 +00002169 AttrList = llvm::AttributeList::get(
2170 getLLVMContext(), llvm::AttributeSet::get(getLLVMContext(), FuncAttrs),
2171 llvm::AttributeSet::get(getLLVMContext(), RetAttrs), ArgAttrs);
Daniel Dunbar76c8eb72008-09-10 00:32:18 +00002172}
2173
John McCalla738c252011-03-09 04:27:21 +00002174/// An argument came in as a promoted argument; demote it back to its
2175/// declared type.
2176static llvm::Value *emitArgumentDemotion(CodeGenFunction &CGF,
2177 const VarDecl *var,
2178 llvm::Value *value) {
Chris Lattner2192fe52011-07-18 04:24:23 +00002179 llvm::Type *varType = CGF.ConvertType(var->getType());
John McCalla738c252011-03-09 04:27:21 +00002180
2181 // This can happen with promotions that actually don't change the
2182 // underlying type, like the enum promotions.
2183 if (value->getType() == varType) return value;
2184
2185 assert((varType->isIntegerTy() || varType->isFloatingPointTy())
2186 && "unexpected promotion type");
2187
2188 if (isa<llvm::IntegerType>(varType))
2189 return CGF.Builder.CreateTrunc(value, varType, "arg.unpromote");
2190
2191 return CGF.Builder.CreateFPCast(value, varType, "arg.unpromote");
2192}
2193
Chandler Carruth45bbe012017-03-24 09:11:57 +00002194/// Returns the attribute (either parameter attribute, or function
2195/// attribute), which declares argument ArgNo to be non-null.
2196static const NonNullAttr *getNonNullAttr(const Decl *FD, const ParmVarDecl *PVD,
2197 QualType ArgType, unsigned ArgNo) {
2198 // FIXME: __attribute__((nonnull)) can also be applied to:
2199 // - references to pointers, where the pointee is known to be
2200 // nonnull (apparently a Clang extension)
2201 // - transparent unions containing pointers
2202 // In the former case, LLVM IR cannot represent the constraint. In
2203 // the latter case, we have no guarantee that the transparent union
2204 // is in fact passed as a pointer.
2205 if (!ArgType->isAnyPointerType() && !ArgType->isBlockPointerType())
2206 return nullptr;
2207 // First, check attribute on parameter itself.
2208 if (PVD) {
2209 if (auto ParmNNAttr = PVD->getAttr<NonNullAttr>())
2210 return ParmNNAttr;
2211 }
2212 // Check function attributes.
2213 if (!FD)
2214 return nullptr;
2215 for (const auto *NNAttr : FD->specific_attrs<NonNullAttr>()) {
2216 if (NNAttr->isNonNull(ArgNo))
2217 return NNAttr;
2218 }
2219 return nullptr;
2220}
2221
John McCall12f23522016-04-04 18:33:08 +00002222namespace {
2223 struct CopyBackSwiftError final : EHScopeStack::Cleanup {
2224 Address Temp;
2225 Address Arg;
2226 CopyBackSwiftError(Address temp, Address arg) : Temp(temp), Arg(arg) {}
2227 void Emit(CodeGenFunction &CGF, Flags flags) override {
2228 llvm::Value *errorValue = CGF.Builder.CreateLoad(Temp);
2229 CGF.Builder.CreateStore(errorValue, Arg);
2230 }
2231 };
2232}
2233
Daniel Dunbard931a872009-02-02 22:03:45 +00002234void CodeGenFunction::EmitFunctionProlog(const CGFunctionInfo &FI,
2235 llvm::Function *Fn,
Daniel Dunbar613855c2008-09-09 23:27:19 +00002236 const FunctionArgList &Args) {
Hans Wennborgd71907d2014-09-04 22:16:33 +00002237 if (CurCodeDecl && CurCodeDecl->hasAttr<NakedAttr>())
2238 // Naked functions don't have prologues.
2239 return;
2240
John McCallcaa19452009-07-28 01:00:58 +00002241 // If this is an implicit-return-zero function, go ahead and
2242 // initialize the return value. TODO: it might be nice to have
2243 // a more general mechanism for this that didn't require synthesized
2244 // return statements.
John McCalldec348f72013-05-03 07:33:41 +00002245 if (const FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(CurCodeDecl)) {
John McCallcaa19452009-07-28 01:00:58 +00002246 if (FD->hasImplicitReturnZero()) {
Alp Toker314cc812014-01-25 16:55:45 +00002247 QualType RetTy = FD->getReturnType().getUnqualifiedType();
Chris Lattner2192fe52011-07-18 04:24:23 +00002248 llvm::Type* LLVMTy = CGM.getTypes().ConvertType(RetTy);
Owen Anderson0b75f232009-07-31 20:28:54 +00002249 llvm::Constant* Zero = llvm::Constant::getNullValue(LLVMTy);
John McCallcaa19452009-07-28 01:00:58 +00002250 Builder.CreateStore(Zero, ReturnValue);
2251 }
2252 }
2253
Mike Stump18bb9282009-05-16 07:57:57 +00002254 // FIXME: We no longer need the types from FunctionArgList; lift up and
2255 // simplify.
Daniel Dunbar5a0acdc92009-02-03 06:02:10 +00002256
Alexey Samsonov153004f2014-09-29 22:08:00 +00002257 ClangToLLVMArgMapping IRFunctionArgs(CGM.getContext(), FI);
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002258 // Flattened function arguments.
John McCall12f23522016-04-04 18:33:08 +00002259 SmallVector<llvm::Value *, 16> FnArgs;
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002260 FnArgs.reserve(IRFunctionArgs.totalIRArgs());
2261 for (auto &Arg : Fn->args()) {
2262 FnArgs.push_back(&Arg);
2263 }
2264 assert(FnArgs.size() == IRFunctionArgs.totalIRArgs());
Mike Stump11289f42009-09-09 15:08:12 +00002265
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002266 // If we're using inalloca, all the memory arguments are GEPs off of the last
2267 // parameter, which is a pointer to the complete memory area.
John McCall7f416cc2015-09-08 08:05:57 +00002268 Address ArgStruct = Address::invalid();
2269 const llvm::StructLayout *ArgStructLayout = nullptr;
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002270 if (IRFunctionArgs.hasInallocaArg()) {
John McCall7f416cc2015-09-08 08:05:57 +00002271 ArgStructLayout = CGM.getDataLayout().getStructLayout(FI.getArgStruct());
2272 ArgStruct = Address(FnArgs[IRFunctionArgs.getInallocaArgNo()],
2273 FI.getArgStructAlignment());
2274
2275 assert(ArgStruct.getType() == FI.getArgStruct()->getPointerTo());
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002276 }
2277
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002278 // Name the struct return parameter.
2279 if (IRFunctionArgs.hasSRetArg()) {
John McCall12f23522016-04-04 18:33:08 +00002280 auto AI = cast<llvm::Argument>(FnArgs[IRFunctionArgs.getSRetArgNo()]);
Daniel Dunbar613855c2008-09-09 23:27:19 +00002281 AI->setName("agg.result");
Reid Klecknercdd26792017-04-18 23:50:03 +00002282 AI->addAttr(llvm::Attribute::NoAlias);
Daniel Dunbar613855c2008-09-09 23:27:19 +00002283 }
Mike Stump11289f42009-09-09 15:08:12 +00002284
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002285 // Track if we received the parameter as a pointer (indirect, byval, or
2286 // inalloca). If already have a pointer, EmitParmDecl doesn't need to copy it
2287 // into a local alloca for us.
John McCall7f416cc2015-09-08 08:05:57 +00002288 SmallVector<ParamValue, 16> ArgVals;
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002289 ArgVals.reserve(Args.size());
2290
Reid Kleckner739756c2013-12-04 19:23:12 +00002291 // Create a pointer value for every parameter declaration. This usually
2292 // entails copying one or more LLVM IR arguments into an alloca. Don't push
2293 // any cleanups or do anything that might unwind. We do that separately, so
2294 // we can push the cleanups in the correct order for the ABI.
Daniel Dunbara45bdbb2009-02-04 21:17:21 +00002295 assert(FI.arg_size() == Args.size() &&
2296 "Mismatch between function signature & arguments.");
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002297 unsigned ArgNo = 0;
Daniel Dunbarb52d0772009-02-03 05:59:18 +00002298 CGFunctionInfo::const_arg_iterator info_it = FI.arg_begin();
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002299 for (FunctionArgList::const_iterator i = Args.begin(), e = Args.end();
Devang Patel68a15252011-03-03 20:13:15 +00002300 i != e; ++i, ++info_it, ++ArgNo) {
John McCalla738c252011-03-09 04:27:21 +00002301 const VarDecl *Arg = *i;
Daniel Dunbarb52d0772009-02-03 05:59:18 +00002302 const ABIArgInfo &ArgI = info_it->info;
Daniel Dunbard3674e62008-09-11 01:48:57 +00002303
John McCalla738c252011-03-09 04:27:21 +00002304 bool isPromoted =
2305 isa<ParmVarDecl>(Arg) && cast<ParmVarDecl>(Arg)->isKNRPromoted();
Volodymyr Sapsai17ebdb22018-01-22 22:29:24 +00002306 // We are converting from ABIArgInfo type to VarDecl type directly, unless
2307 // the parameter is promoted. In this case we convert to
2308 // CGFunctionInfo::ArgInfo type with subsequent argument demotion.
2309 QualType Ty = isPromoted ? info_it->type : Arg->getType();
2310 assert(hasScalarEvaluationKind(Ty) ==
2311 hasScalarEvaluationKind(Arg->getType()));
John McCalla738c252011-03-09 04:27:21 +00002312
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002313 unsigned FirstIRArg, NumIRArgs;
2314 std::tie(FirstIRArg, NumIRArgs) = IRFunctionArgs.getIRArgs(ArgNo);
Rafael Espindolafad28de2012-10-24 01:59:00 +00002315
Daniel Dunbard3674e62008-09-11 01:48:57 +00002316 switch (ArgI.getKind()) {
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002317 case ABIArgInfo::InAlloca: {
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002318 assert(NumIRArgs == 0);
John McCall7f416cc2015-09-08 08:05:57 +00002319 auto FieldIndex = ArgI.getInAllocaFieldIndex();
2320 CharUnits FieldOffset =
2321 CharUnits::fromQuantity(ArgStructLayout->getElementOffset(FieldIndex));
2322 Address V = Builder.CreateStructGEP(ArgStruct, FieldIndex, FieldOffset,
2323 Arg->getName());
2324 ArgVals.push_back(ParamValue::forIndirect(V));
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002325 break;
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002326 }
2327
Daniel Dunbar747865a2009-02-05 09:16:39 +00002328 case ABIArgInfo::Indirect: {
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002329 assert(NumIRArgs == 1);
John McCall7f416cc2015-09-08 08:05:57 +00002330 Address ParamAddr = Address(FnArgs[FirstIRArg], ArgI.getIndirectAlign());
Daniel Dunbar7b7c2932010-09-16 20:42:02 +00002331
John McCall47fb9502013-03-07 21:37:08 +00002332 if (!hasScalarEvaluationKind(Ty)) {
Daniel Dunbar7b7c2932010-09-16 20:42:02 +00002333 // Aggregates and complex variables are accessed by reference. All we
John McCall7f416cc2015-09-08 08:05:57 +00002334 // need to do is realign the value, if requested.
2335 Address V = ParamAddr;
Daniel Dunbar7b7c2932010-09-16 20:42:02 +00002336 if (ArgI.getIndirectRealign()) {
John McCall7f416cc2015-09-08 08:05:57 +00002337 Address AlignedTemp = CreateMemTemp(Ty, "coerce");
Daniel Dunbar7b7c2932010-09-16 20:42:02 +00002338
2339 // Copy from the incoming argument pointer to the temporary with the
2340 // appropriate alignment.
2341 //
2342 // FIXME: We should have a common utility for generating an aggregate
2343 // copy.
Ken Dyck705ba072011-01-19 01:58:38 +00002344 CharUnits Size = getContext().getTypeSizeInChars(Ty);
John McCall7f416cc2015-09-08 08:05:57 +00002345 auto SizeVal = llvm::ConstantInt::get(IntPtrTy, Size.getQuantity());
2346 Address Dst = Builder.CreateBitCast(AlignedTemp, Int8PtrTy);
2347 Address Src = Builder.CreateBitCast(ParamAddr, Int8PtrTy);
2348 Builder.CreateMemCpy(Dst, Src, SizeVal, false);
Daniel Dunbar7b7c2932010-09-16 20:42:02 +00002349 V = AlignedTemp;
2350 }
John McCall7f416cc2015-09-08 08:05:57 +00002351 ArgVals.push_back(ParamValue::forIndirect(V));
Daniel Dunbar747865a2009-02-05 09:16:39 +00002352 } else {
2353 // Load scalar value from indirect argument.
John McCall7f416cc2015-09-08 08:05:57 +00002354 llvm::Value *V =
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002355 EmitLoadOfScalar(ParamAddr, false, Ty, Arg->getBeginLoc());
John McCalla738c252011-03-09 04:27:21 +00002356
2357 if (isPromoted)
2358 V = emitArgumentDemotion(*this, Arg, V);
John McCall7f416cc2015-09-08 08:05:57 +00002359 ArgVals.push_back(ParamValue::forDirect(V));
Daniel Dunbar747865a2009-02-05 09:16:39 +00002360 }
Daniel Dunbar747865a2009-02-05 09:16:39 +00002361 break;
2362 }
Anton Korobeynikov18adbf52009-06-06 09:36:29 +00002363
2364 case ABIArgInfo::Extend:
Daniel Dunbar67dace892009-02-03 06:17:37 +00002365 case ABIArgInfo::Direct: {
Akira Hatanaka18334dd2012-01-09 19:08:06 +00002366
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002367 // If we have the trivial case, handle it with no muss and fuss.
2368 if (!isa<llvm::StructType>(ArgI.getCoerceToType()) &&
Volodymyr Sapsai22b00ec2017-12-21 20:52:59 +00002369 ArgI.getCoerceToType() == ConvertType(Ty) &&
Chris Lattner8a2f3c72010-07-30 04:02:24 +00002370 ArgI.getDirectOffset() == 0) {
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002371 assert(NumIRArgs == 1);
John McCall12f23522016-04-04 18:33:08 +00002372 llvm::Value *V = FnArgs[FirstIRArg];
2373 auto AI = cast<llvm::Argument>(V);
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00002374
Hal Finkel48d53e22014-07-19 01:41:07 +00002375 if (const ParmVarDecl *PVD = dyn_cast<ParmVarDecl>(Arg)) {
Alexey Samsonov8e1162c2014-09-08 17:22:45 +00002376 if (getNonNullAttr(CurCodeDecl, PVD, PVD->getType(),
Manoj Guptada08f6a2018-07-19 00:44:52 +00002377 PVD->getFunctionScopeIndex()) &&
2378 !CGM.getCodeGenOpts().NullPointerIsValid)
Reid Klecknercdd26792017-04-18 23:50:03 +00002379 AI->addAttr(llvm::Attribute::NonNull);
Hal Finkel82504f02014-07-11 17:35:21 +00002380
Hal Finkel48d53e22014-07-19 01:41:07 +00002381 QualType OTy = PVD->getOriginalType();
2382 if (const auto *ArrTy =
2383 getContext().getAsConstantArrayType(OTy)) {
2384 // A C99 array parameter declaration with the static keyword also
2385 // indicates dereferenceability, and if the size is constant we can
2386 // use the dereferenceable attribute (which requires the size in
2387 // bytes).
Hal Finkel16e394a2014-07-19 02:13:40 +00002388 if (ArrTy->getSizeModifier() == ArrayType::Static) {
Hal Finkel48d53e22014-07-19 01:41:07 +00002389 QualType ETy = ArrTy->getElementType();
2390 uint64_t ArrSize = ArrTy->getSize().getZExtValue();
2391 if (!ETy->isIncompleteType() && ETy->isConstantSizeType() &&
2392 ArrSize) {
2393 llvm::AttrBuilder Attrs;
2394 Attrs.addDereferenceableAttr(
2395 getContext().getTypeSizeInChars(ETy).getQuantity()*ArrSize);
Reid Kleckner9d16fa02017-04-19 17:28:52 +00002396 AI->addAttrs(Attrs);
Manoj Guptada08f6a2018-07-19 00:44:52 +00002397 } else if (getContext().getTargetAddressSpace(ETy) == 0 &&
2398 !CGM.getCodeGenOpts().NullPointerIsValid) {
Reid Klecknercdd26792017-04-18 23:50:03 +00002399 AI->addAttr(llvm::Attribute::NonNull);
Hal Finkel48d53e22014-07-19 01:41:07 +00002400 }
2401 }
2402 } else if (const auto *ArrTy =
2403 getContext().getAsVariableArrayType(OTy)) {
2404 // For C99 VLAs with the static keyword, we don't know the size so
2405 // we can't use the dereferenceable attribute, but in addrspace(0)
2406 // we know that it must be nonnull.
2407 if (ArrTy->getSizeModifier() == VariableArrayType::Static &&
Manoj Guptada08f6a2018-07-19 00:44:52 +00002408 !getContext().getTargetAddressSpace(ArrTy->getElementType()) &&
2409 !CGM.getCodeGenOpts().NullPointerIsValid)
Reid Klecknercdd26792017-04-18 23:50:03 +00002410 AI->addAttr(llvm::Attribute::NonNull);
Hal Finkel48d53e22014-07-19 01:41:07 +00002411 }
Hal Finkel1b0d24e2014-10-02 21:21:25 +00002412
2413 const auto *AVAttr = PVD->getAttr<AlignValueAttr>();
2414 if (!AVAttr)
2415 if (const auto *TOTy = dyn_cast<TypedefType>(OTy))
2416 AVAttr = TOTy->getDecl()->getAttr<AlignValueAttr>();
Roman Lebedevbd1c0872019-01-15 09:44:25 +00002417 if (AVAttr && !SanOpts.has(SanitizerKind::Alignment)) {
2418 // If alignment-assumption sanitizer is enabled, we do *not* add
2419 // alignment attribute here, but emit normal alignment assumption,
2420 // so the UBSAN check could function.
Hal Finkel1b0d24e2014-10-02 21:21:25 +00002421 llvm::Value *AlignmentValue =
2422 EmitScalarExpr(AVAttr->getAlignment());
2423 llvm::ConstantInt *AlignmentCI =
2424 cast<llvm::ConstantInt>(AlignmentValue);
Reid Kleckner9d16fa02017-04-19 17:28:52 +00002425 unsigned Alignment = std::min((unsigned)AlignmentCI->getZExtValue(),
2426 +llvm::Value::MaximumAlignment);
2427 AI->addAttrs(llvm::AttrBuilder().addAlignmentAttr(Alignment));
Hal Finkel1b0d24e2014-10-02 21:21:25 +00002428 }
Hal Finkel48d53e22014-07-19 01:41:07 +00002429 }
2430
Bill Wendling507c3512012-10-16 05:23:44 +00002431 if (Arg->getType().isRestrictQualified())
Reid Klecknercdd26792017-04-18 23:50:03 +00002432 AI->addAttr(llvm::Attribute::NoAlias);
John McCall39ec71f2010-03-27 00:47:27 +00002433
John McCall12f23522016-04-04 18:33:08 +00002434 // LLVM expects swifterror parameters to be used in very restricted
2435 // ways. Copy the value into a less-restricted temporary.
2436 if (FI.getExtParameterInfo(ArgNo).getABI()
2437 == ParameterABI::SwiftErrorResult) {
2438 QualType pointeeTy = Ty->getPointeeType();
2439 assert(pointeeTy->isPointerType());
2440 Address temp =
2441 CreateMemTemp(pointeeTy, getPointerAlign(), "swifterror.temp");
2442 Address arg = Address(V, getContext().getTypeAlignInChars(pointeeTy));
2443 llvm::Value *incomingErrorValue = Builder.CreateLoad(arg);
2444 Builder.CreateStore(incomingErrorValue, temp);
2445 V = temp.getPointer();
2446
2447 // Push a cleanup to copy the value back at the end of the function.
2448 // The convention does not guarantee that the value will be written
2449 // back if the function exits with an unwind exception.
2450 EHStack.pushCleanup<CopyBackSwiftError>(NormalCleanup, temp, arg);
2451 }
2452
Chris Lattner7369c142011-07-20 06:29:00 +00002453 // Ensure the argument is the correct type.
2454 if (V->getType() != ArgI.getCoerceToType())
2455 V = Builder.CreateBitCast(V, ArgI.getCoerceToType());
2456
John McCalla738c252011-03-09 04:27:21 +00002457 if (isPromoted)
2458 V = emitArgumentDemotion(*this, Arg, V);
Rafael Espindola8778c282012-11-29 16:09:03 +00002459
2460 // Because of merging of function types from multiple decls it is
2461 // possible for the type of an argument to not match the corresponding
2462 // type in the function type. Since we are codegening the callee
2463 // in here, add a cast to the argument type.
2464 llvm::Type *LTy = ConvertType(Arg->getType());
2465 if (V->getType() != LTy)
2466 V = Builder.CreateBitCast(V, LTy);
2467
John McCall7f416cc2015-09-08 08:05:57 +00002468 ArgVals.push_back(ParamValue::forDirect(V));
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002469 break;
Daniel Dunbard5f1f552009-02-10 00:06:49 +00002470 }
Mike Stump11289f42009-09-09 15:08:12 +00002471
Volodymyr Sapsai22b00ec2017-12-21 20:52:59 +00002472 Address Alloca = CreateMemTemp(Ty, getContext().getDeclAlign(Arg),
2473 Arg->getName());
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00002474
John McCall7f416cc2015-09-08 08:05:57 +00002475 // Pointer to store into.
2476 Address Ptr = emitAddressAtOffset(*this, Alloca, ArgI);
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00002477
Oliver Stannard2bfdc5b2014-08-27 10:43:15 +00002478 // Fast-isel and the optimizer generally like scalar values better than
2479 // FCAs, so we flatten them if this is safe to do for this argument.
Evgeniy Stepanov3fae4ae2012-02-10 09:30:15 +00002480 llvm::StructType *STy = dyn_cast<llvm::StructType>(ArgI.getCoerceToType());
Oliver Stannard2bfdc5b2014-08-27 10:43:15 +00002481 if (ArgI.isDirect() && ArgI.getCanBeFlattened() && STy &&
2482 STy->getNumElements() > 1) {
John McCall7f416cc2015-09-08 08:05:57 +00002483 auto SrcLayout = CGM.getDataLayout().getStructLayout(STy);
Micah Villmowdd31ca12012-10-08 16:25:52 +00002484 uint64_t SrcSize = CGM.getDataLayout().getTypeAllocSize(STy);
John McCall7f416cc2015-09-08 08:05:57 +00002485 llvm::Type *DstTy = Ptr.getElementType();
Micah Villmowdd31ca12012-10-08 16:25:52 +00002486 uint64_t DstSize = CGM.getDataLayout().getTypeAllocSize(DstTy);
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00002487
John McCall7f416cc2015-09-08 08:05:57 +00002488 Address AddrToStoreInto = Address::invalid();
Evgeniy Stepanov3fae4ae2012-02-10 09:30:15 +00002489 if (SrcSize <= DstSize) {
Yaxun Liue9e5c4f2017-06-29 18:47:45 +00002490 AddrToStoreInto = Builder.CreateElementBitCast(Ptr, STy);
Evgeniy Stepanov3fae4ae2012-02-10 09:30:15 +00002491 } else {
John McCall7f416cc2015-09-08 08:05:57 +00002492 AddrToStoreInto =
2493 CreateTempAlloca(STy, Alloca.getAlignment(), "coerce");
Chris Lattner15ec3612010-06-29 00:06:42 +00002494 }
John McCall7f416cc2015-09-08 08:05:57 +00002495
2496 assert(STy->getNumElements() == NumIRArgs);
2497 for (unsigned i = 0, e = STy->getNumElements(); i != e; ++i) {
2498 auto AI = FnArgs[FirstIRArg + i];
2499 AI->setName(Arg->getName() + ".coerce" + Twine(i));
2500 auto Offset = CharUnits::fromQuantity(SrcLayout->getElementOffset(i));
2501 Address EltPtr =
2502 Builder.CreateStructGEP(AddrToStoreInto, i, Offset);
2503 Builder.CreateStore(AI, EltPtr);
2504 }
2505
2506 if (SrcSize > DstSize) {
2507 Builder.CreateMemCpy(Ptr, AddrToStoreInto, DstSize);
2508 }
2509
Chris Lattner15ec3612010-06-29 00:06:42 +00002510 } else {
2511 // Simple case, just do a coerced store of the argument into the alloca.
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002512 assert(NumIRArgs == 1);
2513 auto AI = FnArgs[FirstIRArg];
Chris Lattner9e748e92010-06-29 00:14:52 +00002514 AI->setName(Arg->getName() + ".coerce");
John McCall7f416cc2015-09-08 08:05:57 +00002515 CreateCoercedStore(AI, Ptr, /*DestIsVolatile=*/false, *this);
Chris Lattner15ec3612010-06-29 00:06:42 +00002516 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00002517
Daniel Dunbar2f219b02009-02-03 19:12:28 +00002518 // Match to what EmitParmDecl is expecting for this type.
Volodymyr Sapsai22b00ec2017-12-21 20:52:59 +00002519 if (CodeGenFunction::hasScalarEvaluationKind(Ty)) {
John McCall7f416cc2015-09-08 08:05:57 +00002520 llvm::Value *V =
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002521 EmitLoadOfScalar(Alloca, false, Ty, Arg->getBeginLoc());
John McCalla738c252011-03-09 04:27:21 +00002522 if (isPromoted)
2523 V = emitArgumentDemotion(*this, Arg, V);
John McCall7f416cc2015-09-08 08:05:57 +00002524 ArgVals.push_back(ParamValue::forDirect(V));
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002525 } else {
John McCall7f416cc2015-09-08 08:05:57 +00002526 ArgVals.push_back(ParamValue::forIndirect(Alloca));
Daniel Dunbar6e3b7df2009-02-04 07:22:24 +00002527 }
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002528 break;
Daniel Dunbar2f219b02009-02-03 19:12:28 +00002529 }
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002530
John McCallf26e73d2016-03-11 04:30:43 +00002531 case ABIArgInfo::CoerceAndExpand: {
2532 // Reconstruct into a temporary.
2533 Address alloca = CreateMemTemp(Ty, getContext().getDeclAlign(Arg));
2534 ArgVals.push_back(ParamValue::forIndirect(alloca));
2535
2536 auto coercionType = ArgI.getCoerceAndExpandType();
2537 alloca = Builder.CreateElementBitCast(alloca, coercionType);
2538 auto layout = CGM.getDataLayout().getStructLayout(coercionType);
2539
2540 unsigned argIndex = FirstIRArg;
2541 for (unsigned i = 0, e = coercionType->getNumElements(); i != e; ++i) {
2542 llvm::Type *eltType = coercionType->getElementType(i);
2543 if (ABIArgInfo::isPaddingForCoerceAndExpand(eltType))
2544 continue;
2545
2546 auto eltAddr = Builder.CreateStructGEP(alloca, i, layout);
2547 auto elt = FnArgs[argIndex++];
2548 Builder.CreateStore(elt, eltAddr);
2549 }
2550 assert(argIndex == FirstIRArg + NumIRArgs);
2551 break;
2552 }
2553
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002554 case ABIArgInfo::Expand: {
2555 // If this structure was expanded into multiple arguments then
2556 // we need to create a temporary and reconstruct it from the
2557 // arguments.
John McCall7f416cc2015-09-08 08:05:57 +00002558 Address Alloca = CreateMemTemp(Ty, getContext().getDeclAlign(Arg));
2559 LValue LV = MakeAddrLValue(Alloca, Ty);
2560 ArgVals.push_back(ParamValue::forIndirect(Alloca));
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002561
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002562 auto FnArgIter = FnArgs.begin() + FirstIRArg;
2563 ExpandTypeFromArgs(Ty, LV, FnArgIter);
2564 assert(FnArgIter == FnArgs.begin() + FirstIRArg + NumIRArgs);
2565 for (unsigned i = 0, e = NumIRArgs; i != e; ++i) {
2566 auto AI = FnArgs[FirstIRArg + i];
2567 AI->setName(Arg->getName() + "." + Twine(i));
2568 }
2569 break;
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002570 }
2571
2572 case ABIArgInfo::Ignore:
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002573 assert(NumIRArgs == 0);
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002574 // Initialize the local variable appropriately.
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002575 if (!hasScalarEvaluationKind(Ty)) {
John McCall7f416cc2015-09-08 08:05:57 +00002576 ArgVals.push_back(ParamValue::forIndirect(CreateMemTemp(Ty)));
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002577 } else {
2578 llvm::Value *U = llvm::UndefValue::get(ConvertType(Arg->getType()));
John McCall7f416cc2015-09-08 08:05:57 +00002579 ArgVals.push_back(ParamValue::forDirect(U));
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002580 }
Alexey Samsonov91cf4552014-08-22 01:06:06 +00002581 break;
Daniel Dunbard3674e62008-09-11 01:48:57 +00002582 }
Daniel Dunbar613855c2008-09-09 23:27:19 +00002583 }
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002584
Reid Kleckner739756c2013-12-04 19:23:12 +00002585 if (getTarget().getCXXABI().areArgsDestroyedLeftToRightInCallee()) {
2586 for (int I = Args.size() - 1; I >= 0; --I)
John McCall7f416cc2015-09-08 08:05:57 +00002587 EmitParmDecl(*Args[I], ArgVals[I], I + 1);
Reid Kleckner739756c2013-12-04 19:23:12 +00002588 } else {
2589 for (unsigned I = 0, E = Args.size(); I != E; ++I)
John McCall7f416cc2015-09-08 08:05:57 +00002590 EmitParmDecl(*Args[I], ArgVals[I], I + 1);
Reid Kleckner739756c2013-12-04 19:23:12 +00002591 }
Daniel Dunbar613855c2008-09-09 23:27:19 +00002592}
2593
John McCallffa2c1a2012-01-29 07:46:59 +00002594static void eraseUnusedBitCasts(llvm::Instruction *insn) {
2595 while (insn->use_empty()) {
2596 llvm::BitCastInst *bitcast = dyn_cast<llvm::BitCastInst>(insn);
2597 if (!bitcast) return;
2598
2599 // This is "safe" because we would have used a ConstantExpr otherwise.
2600 insn = cast<llvm::Instruction>(bitcast->getOperand(0));
2601 bitcast->eraseFromParent();
2602 }
2603}
2604
John McCall31168b02011-06-15 23:02:42 +00002605/// Try to emit a fused autorelease of a return result.
2606static llvm::Value *tryEmitFusedAutoreleaseOfResult(CodeGenFunction &CGF,
2607 llvm::Value *result) {
2608 // We must be immediately followed the cast.
2609 llvm::BasicBlock *BB = CGF.Builder.GetInsertBlock();
Craig Topper8a13c412014-05-21 05:09:00 +00002610 if (BB->empty()) return nullptr;
2611 if (&BB->back() != result) return nullptr;
John McCall31168b02011-06-15 23:02:42 +00002612
Chris Lattner2192fe52011-07-18 04:24:23 +00002613 llvm::Type *resultType = result->getType();
John McCall31168b02011-06-15 23:02:42 +00002614
2615 // result is in a BasicBlock and is therefore an Instruction.
2616 llvm::Instruction *generator = cast<llvm::Instruction>(result);
2617
Justin Bogner882f8612016-08-18 21:46:54 +00002618 SmallVector<llvm::Instruction *, 4> InstsToKill;
John McCall31168b02011-06-15 23:02:42 +00002619
2620 // Look for:
2621 // %generator = bitcast %type1* %generator2 to %type2*
2622 while (llvm::BitCastInst *bitcast = dyn_cast<llvm::BitCastInst>(generator)) {
2623 // We would have emitted this as a constant if the operand weren't
2624 // an Instruction.
2625 generator = cast<llvm::Instruction>(bitcast->getOperand(0));
2626
2627 // Require the generator to be immediately followed by the cast.
2628 if (generator->getNextNode() != bitcast)
Craig Topper8a13c412014-05-21 05:09:00 +00002629 return nullptr;
John McCall31168b02011-06-15 23:02:42 +00002630
Justin Bogner882f8612016-08-18 21:46:54 +00002631 InstsToKill.push_back(bitcast);
John McCall31168b02011-06-15 23:02:42 +00002632 }
2633
2634 // Look for:
2635 // %generator = call i8* @objc_retain(i8* %originalResult)
2636 // or
2637 // %generator = call i8* @objc_retainAutoreleasedReturnValue(i8* %originalResult)
2638 llvm::CallInst *call = dyn_cast<llvm::CallInst>(generator);
Craig Topper8a13c412014-05-21 05:09:00 +00002639 if (!call) return nullptr;
John McCall31168b02011-06-15 23:02:42 +00002640
2641 bool doRetainAutorelease;
2642
John McCallb04ecb72015-10-21 18:06:43 +00002643 if (call->getCalledValue() == CGF.CGM.getObjCEntrypoints().objc_retain) {
John McCall31168b02011-06-15 23:02:42 +00002644 doRetainAutorelease = true;
John McCallb04ecb72015-10-21 18:06:43 +00002645 } else if (call->getCalledValue() == CGF.CGM.getObjCEntrypoints()
John McCall31168b02011-06-15 23:02:42 +00002646 .objc_retainAutoreleasedReturnValue) {
2647 doRetainAutorelease = false;
2648
John McCallcfa4e9b2012-09-07 23:30:50 +00002649 // If we emitted an assembly marker for this call (and the
2650 // ARCEntrypoints field should have been set if so), go looking
2651 // for that call. If we can't find it, we can't do this
2652 // optimization. But it should always be the immediately previous
2653 // instruction, unless we needed bitcasts around the call.
John McCallb04ecb72015-10-21 18:06:43 +00002654 if (CGF.CGM.getObjCEntrypoints().retainAutoreleasedReturnValueMarker) {
John McCallcfa4e9b2012-09-07 23:30:50 +00002655 llvm::Instruction *prev = call->getPrevNode();
2656 assert(prev);
2657 if (isa<llvm::BitCastInst>(prev)) {
2658 prev = prev->getPrevNode();
2659 assert(prev);
2660 }
2661 assert(isa<llvm::CallInst>(prev));
2662 assert(cast<llvm::CallInst>(prev)->getCalledValue() ==
John McCallb04ecb72015-10-21 18:06:43 +00002663 CGF.CGM.getObjCEntrypoints().retainAutoreleasedReturnValueMarker);
Justin Bogner882f8612016-08-18 21:46:54 +00002664 InstsToKill.push_back(prev);
John McCallcfa4e9b2012-09-07 23:30:50 +00002665 }
John McCall31168b02011-06-15 23:02:42 +00002666 } else {
Craig Topper8a13c412014-05-21 05:09:00 +00002667 return nullptr;
John McCall31168b02011-06-15 23:02:42 +00002668 }
2669
2670 result = call->getArgOperand(0);
Justin Bogner882f8612016-08-18 21:46:54 +00002671 InstsToKill.push_back(call);
John McCall31168b02011-06-15 23:02:42 +00002672
2673 // Keep killing bitcasts, for sanity. Note that we no longer care
2674 // about precise ordering as long as there's exactly one use.
2675 while (llvm::BitCastInst *bitcast = dyn_cast<llvm::BitCastInst>(result)) {
2676 if (!bitcast->hasOneUse()) break;
Justin Bogner882f8612016-08-18 21:46:54 +00002677 InstsToKill.push_back(bitcast);
John McCall31168b02011-06-15 23:02:42 +00002678 result = bitcast->getOperand(0);
2679 }
2680
2681 // Delete all the unnecessary instructions, from latest to earliest.
Justin Bogner882f8612016-08-18 21:46:54 +00002682 for (auto *I : InstsToKill)
Saleem Abdulrasoolbe25c482016-08-18 21:40:06 +00002683 I->eraseFromParent();
John McCall31168b02011-06-15 23:02:42 +00002684
2685 // Do the fused retain/autorelease if we were asked to.
2686 if (doRetainAutorelease)
2687 result = CGF.EmitARCRetainAutoreleaseReturnValue(result);
2688
2689 // Cast back to the result type.
2690 return CGF.Builder.CreateBitCast(result, resultType);
2691}
2692
John McCallffa2c1a2012-01-29 07:46:59 +00002693/// If this is a +1 of the value of an immutable 'self', remove it.
2694static llvm::Value *tryRemoveRetainOfSelf(CodeGenFunction &CGF,
2695 llvm::Value *result) {
2696 // This is only applicable to a method with an immutable 'self'.
John McCallff755cd2012-07-31 00:33:55 +00002697 const ObjCMethodDecl *method =
2698 dyn_cast_or_null<ObjCMethodDecl>(CGF.CurCodeDecl);
Craig Topper8a13c412014-05-21 05:09:00 +00002699 if (!method) return nullptr;
John McCallffa2c1a2012-01-29 07:46:59 +00002700 const VarDecl *self = method->getSelfDecl();
Craig Topper8a13c412014-05-21 05:09:00 +00002701 if (!self->getType().isConstQualified()) return nullptr;
John McCallffa2c1a2012-01-29 07:46:59 +00002702
2703 // Look for a retain call.
2704 llvm::CallInst *retainCall =
2705 dyn_cast<llvm::CallInst>(result->stripPointerCasts());
2706 if (!retainCall ||
John McCallb04ecb72015-10-21 18:06:43 +00002707 retainCall->getCalledValue() != CGF.CGM.getObjCEntrypoints().objc_retain)
Craig Topper8a13c412014-05-21 05:09:00 +00002708 return nullptr;
John McCallffa2c1a2012-01-29 07:46:59 +00002709
2710 // Look for an ordinary load of 'self'.
2711 llvm::Value *retainedValue = retainCall->getArgOperand(0);
2712 llvm::LoadInst *load =
2713 dyn_cast<llvm::LoadInst>(retainedValue->stripPointerCasts());
Fangrui Song6907ce22018-07-30 19:24:48 +00002714 if (!load || load->isAtomic() || load->isVolatile() ||
John McCall7f416cc2015-09-08 08:05:57 +00002715 load->getPointerOperand() != CGF.GetAddrOfLocalVar(self).getPointer())
Craig Topper8a13c412014-05-21 05:09:00 +00002716 return nullptr;
John McCallffa2c1a2012-01-29 07:46:59 +00002717
2718 // Okay! Burn it all down. This relies for correctness on the
2719 // assumption that the retain is emitted as part of the return and
2720 // that thereafter everything is used "linearly".
2721 llvm::Type *resultType = result->getType();
2722 eraseUnusedBitCasts(cast<llvm::Instruction>(result));
2723 assert(retainCall->use_empty());
2724 retainCall->eraseFromParent();
2725 eraseUnusedBitCasts(cast<llvm::Instruction>(retainedValue));
2726
2727 return CGF.Builder.CreateBitCast(load, resultType);
2728}
2729
John McCall31168b02011-06-15 23:02:42 +00002730/// Emit an ARC autorelease of the result of a function.
John McCallffa2c1a2012-01-29 07:46:59 +00002731///
2732/// \return the value to actually return from the function
John McCall31168b02011-06-15 23:02:42 +00002733static llvm::Value *emitAutoreleaseOfResult(CodeGenFunction &CGF,
2734 llvm::Value *result) {
John McCallffa2c1a2012-01-29 07:46:59 +00002735 // If we're returning 'self', kill the initial retain. This is a
2736 // heuristic attempt to "encourage correctness" in the really unfortunate
2737 // case where we have a return of self during a dealloc and we desperately
2738 // need to avoid the possible autorelease.
2739 if (llvm::Value *self = tryRemoveRetainOfSelf(CGF, result))
2740 return self;
2741
John McCall31168b02011-06-15 23:02:42 +00002742 // At -O0, try to emit a fused retain/autorelease.
2743 if (CGF.shouldUseFusedARCCalls())
2744 if (llvm::Value *fused = tryEmitFusedAutoreleaseOfResult(CGF, result))
2745 return fused;
2746
2747 return CGF.EmitARCAutoreleaseReturnValue(result);
2748}
2749
John McCall6e1c0122012-01-29 02:35:02 +00002750/// Heuristically search for a dominating store to the return-value slot.
2751static llvm::StoreInst *findDominatingStoreToReturnValue(CodeGenFunction &CGF) {
Jakub Kuderskif50ab0f2015-09-08 10:36:42 +00002752 // Check if a User is a store which pointerOperand is the ReturnValue.
2753 // We are looking for stores to the ReturnValue, not for stores of the
2754 // ReturnValue to some other location.
2755 auto GetStoreIfValid = [&CGF](llvm::User *U) -> llvm::StoreInst * {
2756 auto *SI = dyn_cast<llvm::StoreInst>(U);
2757 if (!SI || SI->getPointerOperand() != CGF.ReturnValue.getPointer())
2758 return nullptr;
2759 // These aren't actually possible for non-coerced returns, and we
2760 // only care about non-coerced returns on this code path.
2761 assert(!SI->isAtomic() && !SI->isVolatile());
2762 return SI;
2763 };
John McCall6e1c0122012-01-29 02:35:02 +00002764 // If there are multiple uses of the return-value slot, just check
2765 // for something immediately preceding the IP. Sometimes this can
2766 // happen with how we generate implicit-returns; it can also happen
2767 // with noreturn cleanups.
John McCall7f416cc2015-09-08 08:05:57 +00002768 if (!CGF.ReturnValue.getPointer()->hasOneUse()) {
John McCall6e1c0122012-01-29 02:35:02 +00002769 llvm::BasicBlock *IP = CGF.Builder.GetInsertBlock();
Craig Topper8a13c412014-05-21 05:09:00 +00002770 if (IP->empty()) return nullptr;
David Majnemerdc012fa2015-04-22 21:38:15 +00002771 llvm::Instruction *I = &IP->back();
2772
2773 // Skip lifetime markers
2774 for (llvm::BasicBlock::reverse_iterator II = IP->rbegin(),
2775 IE = IP->rend();
2776 II != IE; ++II) {
2777 if (llvm::IntrinsicInst *Intrinsic =
2778 dyn_cast<llvm::IntrinsicInst>(&*II)) {
2779 if (Intrinsic->getIntrinsicID() == llvm::Intrinsic::lifetime_end) {
2780 const llvm::Value *CastAddr = Intrinsic->getArgOperand(1);
2781 ++II;
Alexey Samsonov10544202015-06-12 21:05:32 +00002782 if (II == IE)
2783 break;
2784 if (isa<llvm::BitCastInst>(&*II) && (CastAddr == &*II))
2785 continue;
David Majnemerdc012fa2015-04-22 21:38:15 +00002786 }
2787 }
2788 I = &*II;
2789 break;
2790 }
2791
Jakub Kuderskif50ab0f2015-09-08 10:36:42 +00002792 return GetStoreIfValid(I);
John McCall6e1c0122012-01-29 02:35:02 +00002793 }
2794
2795 llvm::StoreInst *store =
Jakub Kuderskif50ab0f2015-09-08 10:36:42 +00002796 GetStoreIfValid(CGF.ReturnValue.getPointer()->user_back());
Craig Topper8a13c412014-05-21 05:09:00 +00002797 if (!store) return nullptr;
John McCall6e1c0122012-01-29 02:35:02 +00002798
John McCall6e1c0122012-01-29 02:35:02 +00002799 // Now do a first-and-dirty dominance check: just walk up the
2800 // single-predecessors chain from the current insertion point.
2801 llvm::BasicBlock *StoreBB = store->getParent();
2802 llvm::BasicBlock *IP = CGF.Builder.GetInsertBlock();
2803 while (IP != StoreBB) {
2804 if (!(IP = IP->getSinglePredecessor()))
Craig Topper8a13c412014-05-21 05:09:00 +00002805 return nullptr;
John McCall6e1c0122012-01-29 02:35:02 +00002806 }
2807
2808 // Okay, the store's basic block dominates the insertion point; we
2809 // can do our thing.
2810 return store;
2811}
2812
Adrian Prantl3be10542013-05-02 17:30:20 +00002813void CodeGenFunction::EmitFunctionEpilog(const CGFunctionInfo &FI,
Nick Lewycky2d84e842013-10-02 02:29:49 +00002814 bool EmitRetDbgLoc,
2815 SourceLocation EndLoc) {
Vedant Kumar09b5bfd2017-12-21 00:10:25 +00002816 if (FI.isNoReturn()) {
2817 // Noreturn functions don't return.
2818 EmitUnreachable(EndLoc);
2819 return;
2820 }
2821
Hans Wennborgd71907d2014-09-04 22:16:33 +00002822 if (CurCodeDecl && CurCodeDecl->hasAttr<NakedAttr>()) {
2823 // Naked functions don't have epilogues.
2824 Builder.CreateUnreachable();
2825 return;
2826 }
2827
Daniel Dunbara72d4ae2008-09-10 02:41:04 +00002828 // Functions with no result always return void.
John McCall7f416cc2015-09-08 08:05:57 +00002829 if (!ReturnValue.isValid()) {
Daniel Dunbara72d4ae2008-09-10 02:41:04 +00002830 Builder.CreateRetVoid();
Chris Lattner726b3d02010-06-26 23:13:19 +00002831 return;
Daniel Dunbara72d4ae2008-09-10 02:41:04 +00002832 }
Daniel Dunbar6696e222010-06-30 21:27:58 +00002833
Dan Gohman481e40c2010-07-20 20:13:52 +00002834 llvm::DebugLoc RetDbgLoc;
Craig Topper8a13c412014-05-21 05:09:00 +00002835 llvm::Value *RV = nullptr;
Chris Lattner726b3d02010-06-26 23:13:19 +00002836 QualType RetTy = FI.getReturnType();
2837 const ABIArgInfo &RetAI = FI.getReturnInfo();
2838
2839 switch (RetAI.getKind()) {
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002840 case ABIArgInfo::InAlloca:
Reid Klecknerfab1e892014-02-25 00:59:14 +00002841 // Aggregrates get evaluated directly into the destination. Sometimes we
2842 // need to return the sret value in a register, though.
2843 assert(hasAggregateEvaluationKind(RetTy));
2844 if (RetAI.getInAllocaSRet()) {
2845 llvm::Function::arg_iterator EI = CurFn->arg_end();
2846 --EI;
Duncan P. N. Exon Smith9f5260a2015-11-06 23:00:41 +00002847 llvm::Value *ArgStruct = &*EI;
David Blaikie2e804282015-04-05 22:47:07 +00002848 llvm::Value *SRet = Builder.CreateStructGEP(
2849 nullptr, ArgStruct, RetAI.getInAllocaFieldIndex());
John McCall7f416cc2015-09-08 08:05:57 +00002850 RV = Builder.CreateAlignedLoad(SRet, getPointerAlign(), "sret");
Reid Klecknerfab1e892014-02-25 00:59:14 +00002851 }
Reid Kleckner314ef7b2014-02-01 00:04:45 +00002852 break;
2853
Daniel Dunbar03816342010-08-21 02:24:36 +00002854 case ABIArgInfo::Indirect: {
Reid Kleckner37abaca2014-05-09 22:46:15 +00002855 auto AI = CurFn->arg_begin();
2856 if (RetAI.isSRetAfterThis())
2857 ++AI;
John McCall47fb9502013-03-07 21:37:08 +00002858 switch (getEvaluationKind(RetTy)) {
2859 case TEK_Complex: {
2860 ComplexPairTy RT =
John McCall7f416cc2015-09-08 08:05:57 +00002861 EmitLoadOfComplex(MakeAddrLValue(ReturnValue, RetTy), EndLoc);
Duncan P. N. Exon Smith9f5260a2015-11-06 23:00:41 +00002862 EmitStoreOfComplex(RT, MakeNaturalAlignAddrLValue(&*AI, RetTy),
John McCall47fb9502013-03-07 21:37:08 +00002863 /*isInit*/ true);
2864 break;
2865 }
2866 case TEK_Aggregate:
Chris Lattner726b3d02010-06-26 23:13:19 +00002867 // Do nothing; aggregrates get evaluated directly into the destination.
John McCall47fb9502013-03-07 21:37:08 +00002868 break;
2869 case TEK_Scalar:
2870 EmitStoreOfScalar(Builder.CreateLoad(ReturnValue),
Duncan P. N. Exon Smith9f5260a2015-11-06 23:00:41 +00002871 MakeNaturalAlignAddrLValue(&*AI, RetTy),
John McCall47fb9502013-03-07 21:37:08 +00002872 /*isInit*/ true);
2873 break;
Chris Lattner726b3d02010-06-26 23:13:19 +00002874 }
2875 break;
Daniel Dunbar03816342010-08-21 02:24:36 +00002876 }
Chris Lattner726b3d02010-06-26 23:13:19 +00002877
2878 case ABIArgInfo::Extend:
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002879 case ABIArgInfo::Direct:
Chris Lattner8a2f3c72010-07-30 04:02:24 +00002880 if (RetAI.getCoerceToType() == ConvertType(RetTy) &&
2881 RetAI.getDirectOffset() == 0) {
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002882 // The internal return value temp always will have pointer-to-return-type
2883 // type, just do a load.
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00002884
John McCall6e1c0122012-01-29 02:35:02 +00002885 // If there is a dominating store to ReturnValue, we can elide
2886 // the load, zap the store, and usually zap the alloca.
David Majnemerdc012fa2015-04-22 21:38:15 +00002887 if (llvm::StoreInst *SI =
2888 findDominatingStoreToReturnValue(*this)) {
Adrian Prantl4c9a38a2013-05-30 18:12:23 +00002889 // Reuse the debug location from the store unless there is
2890 // cleanup code to be emitted between the store and return
2891 // instruction.
2892 if (EmitRetDbgLoc && !AutoreleaseResult)
Adrian Prantl3be10542013-05-02 17:30:20 +00002893 RetDbgLoc = SI->getDebugLoc();
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002894 // Get the stored value and nuke the now-dead store.
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002895 RV = SI->getValueOperand();
2896 SI->eraseFromParent();
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00002897
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002898 // If that was the only use of the return value, nuke it as well now.
John McCall7f416cc2015-09-08 08:05:57 +00002899 auto returnValueInst = ReturnValue.getPointer();
2900 if (returnValueInst->use_empty()) {
2901 if (auto alloca = dyn_cast<llvm::AllocaInst>(returnValueInst)) {
2902 alloca->eraseFromParent();
2903 ReturnValue = Address::invalid();
2904 }
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002905 }
John McCall6e1c0122012-01-29 02:35:02 +00002906
2907 // Otherwise, we have to do a simple load.
2908 } else {
2909 RV = Builder.CreateLoad(ReturnValue);
Chris Lattner3fcc7902010-06-27 01:06:27 +00002910 }
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002911 } else {
Chris Lattner8a2f3c72010-07-30 04:02:24 +00002912 // If the value is offset in memory, apply the offset now.
John McCall7f416cc2015-09-08 08:05:57 +00002913 Address V = emitAddressAtOffset(*this, ReturnValue, RetAI);
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00002914
John McCall7f416cc2015-09-08 08:05:57 +00002915 RV = CreateCoercedLoad(V, RetAI.getCoerceToType(), *this);
Chris Lattner3fcc7902010-06-27 01:06:27 +00002916 }
John McCall31168b02011-06-15 23:02:42 +00002917
2918 // In ARC, end functions that return a retainable type with a call
2919 // to objc_autoreleaseReturnValue.
2920 if (AutoreleaseResult) {
Akira Hatanaka9d8ac612016-02-17 21:09:50 +00002921#ifndef NDEBUG
2922 // Type::isObjCRetainabletype has to be called on a QualType that hasn't
2923 // been stripped of the typedefs, so we cannot use RetTy here. Get the
2924 // original return type of FunctionDecl, CurCodeDecl, and BlockDecl from
2925 // CurCodeDecl or BlockInfo.
2926 QualType RT;
2927
2928 if (auto *FD = dyn_cast<FunctionDecl>(CurCodeDecl))
2929 RT = FD->getReturnType();
2930 else if (auto *MD = dyn_cast<ObjCMethodDecl>(CurCodeDecl))
2931 RT = MD->getReturnType();
2932 else if (isa<BlockDecl>(CurCodeDecl))
2933 RT = BlockInfo->BlockExpression->getFunctionType()->getReturnType();
2934 else
2935 llvm_unreachable("Unexpected function/method type");
2936
David Blaikiebbafb8a2012-03-11 07:00:24 +00002937 assert(getLangOpts().ObjCAutoRefCount &&
John McCall31168b02011-06-15 23:02:42 +00002938 !FI.isReturnsRetained() &&
Akira Hatanaka9d8ac612016-02-17 21:09:50 +00002939 RT->isObjCRetainableType());
2940#endif
John McCall31168b02011-06-15 23:02:42 +00002941 RV = emitAutoreleaseOfResult(*this, RV);
2942 }
2943
Chris Lattner726b3d02010-06-26 23:13:19 +00002944 break;
Chris Lattner726b3d02010-06-26 23:13:19 +00002945
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00002946 case ABIArgInfo::Ignore:
Chris Lattner726b3d02010-06-26 23:13:19 +00002947 break;
2948
John McCallf26e73d2016-03-11 04:30:43 +00002949 case ABIArgInfo::CoerceAndExpand: {
2950 auto coercionType = RetAI.getCoerceAndExpandType();
2951 auto layout = CGM.getDataLayout().getStructLayout(coercionType);
2952
2953 // Load all of the coerced elements out into results.
2954 llvm::SmallVector<llvm::Value*, 4> results;
2955 Address addr = Builder.CreateElementBitCast(ReturnValue, coercionType);
2956 for (unsigned i = 0, e = coercionType->getNumElements(); i != e; ++i) {
2957 auto coercedEltType = coercionType->getElementType(i);
2958 if (ABIArgInfo::isPaddingForCoerceAndExpand(coercedEltType))
2959 continue;
2960
2961 auto eltAddr = Builder.CreateStructGEP(addr, i, layout);
2962 auto elt = Builder.CreateLoad(eltAddr);
2963 results.push_back(elt);
2964 }
2965
2966 // If we have one result, it's the single direct result type.
2967 if (results.size() == 1) {
2968 RV = results[0];
2969
2970 // Otherwise, we need to make a first-class aggregate.
2971 } else {
2972 // Construct a return type that lacks padding elements.
2973 llvm::Type *returnType = RetAI.getUnpaddedCoerceAndExpandType();
2974
2975 RV = llvm::UndefValue::get(returnType);
2976 for (unsigned i = 0, e = results.size(); i != e; ++i) {
2977 RV = Builder.CreateInsertValue(RV, results[i], i);
2978 }
2979 }
2980 break;
2981 }
2982
Chris Lattner726b3d02010-06-26 23:13:19 +00002983 case ABIArgInfo::Expand:
David Blaikie83d382b2011-09-23 05:06:16 +00002984 llvm_unreachable("Invalid ABI kind for return argument");
Chris Lattner726b3d02010-06-26 23:13:19 +00002985 }
2986
Alexey Samsonovde443c52014-08-13 00:26:40 +00002987 llvm::Instruction *Ret;
2988 if (RV) {
Vedant Kumarc34d3432017-06-23 21:32:38 +00002989 EmitReturnValueCheck(RV);
Alexey Samsonovde443c52014-08-13 00:26:40 +00002990 Ret = Builder.CreateRet(RV);
2991 } else {
2992 Ret = Builder.CreateRetVoid();
2993 }
2994
Duncan P. N. Exon Smith2809cc72015-03-30 20:01:41 +00002995 if (RetDbgLoc)
Benjamin Kramer03278662015-02-07 13:15:54 +00002996 Ret->setDebugLoc(std::move(RetDbgLoc));
Daniel Dunbar613855c2008-09-09 23:27:19 +00002997}
2998
Vedant Kumarc34d3432017-06-23 21:32:38 +00002999void CodeGenFunction::EmitReturnValueCheck(llvm::Value *RV) {
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003000 // A current decl may not be available when emitting vtable thunks.
3001 if (!CurCodeDecl)
3002 return;
3003
3004 ReturnsNonNullAttr *RetNNAttr = nullptr;
3005 if (SanOpts.has(SanitizerKind::ReturnsNonnullAttribute))
3006 RetNNAttr = CurCodeDecl->getAttr<ReturnsNonNullAttr>();
3007
3008 if (!RetNNAttr && !requiresReturnValueNullabilityCheck())
3009 return;
3010
3011 // Prefer the returns_nonnull attribute if it's present.
3012 SourceLocation AttrLoc;
3013 SanitizerMask CheckKind;
Vedant Kumar2b9f48a2017-03-14 16:48:29 +00003014 SanitizerHandler Handler;
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003015 if (RetNNAttr) {
3016 assert(!requiresReturnValueNullabilityCheck() &&
3017 "Cannot check nullability and the nonnull attribute");
3018 AttrLoc = RetNNAttr->getLocation();
3019 CheckKind = SanitizerKind::ReturnsNonnullAttribute;
Vedant Kumar2b9f48a2017-03-14 16:48:29 +00003020 Handler = SanitizerHandler::NonnullReturn;
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003021 } else {
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003022 if (auto *DD = dyn_cast<DeclaratorDecl>(CurCodeDecl))
3023 if (auto *TSI = DD->getTypeSourceInfo())
3024 if (auto FTL = TSI->getTypeLoc().castAs<FunctionTypeLoc>())
3025 AttrLoc = FTL.getReturnLoc().findNullabilityLoc();
3026 CheckKind = SanitizerKind::NullabilityReturn;
Vedant Kumar2b9f48a2017-03-14 16:48:29 +00003027 Handler = SanitizerHandler::NullabilityReturn;
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003028 }
3029
3030 SanitizerScope SanScope(this);
3031
Vedant Kumarc34d3432017-06-23 21:32:38 +00003032 // Make sure the "return" source location is valid. If we're checking a
3033 // nullability annotation, make sure the preconditions for the check are met.
3034 llvm::BasicBlock *Check = createBasicBlock("nullcheck");
3035 llvm::BasicBlock *NoCheck = createBasicBlock("no.nullcheck");
3036 llvm::Value *SLocPtr = Builder.CreateLoad(ReturnLocation, "return.sloc.load");
3037 llvm::Value *CanNullCheck = Builder.CreateIsNotNull(SLocPtr);
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003038 if (requiresReturnValueNullabilityCheck())
Vedant Kumarc34d3432017-06-23 21:32:38 +00003039 CanNullCheck =
3040 Builder.CreateAnd(CanNullCheck, RetValNullabilityPrecondition);
3041 Builder.CreateCondBr(CanNullCheck, Check, NoCheck);
3042 EmitBlock(Check);
3043
3044 // Now do the null check.
3045 llvm::Value *Cond = Builder.CreateIsNotNull(RV);
3046 llvm::Constant *StaticData[] = {EmitCheckSourceLocation(AttrLoc)};
3047 llvm::Value *DynamicData[] = {SLocPtr};
3048 EmitCheck(std::make_pair(Cond, CheckKind), Handler, StaticData, DynamicData);
3049
3050 EmitBlock(NoCheck);
3051
3052#ifndef NDEBUG
3053 // The return location should not be used after the check has been emitted.
3054 ReturnLocation = Address::invalid();
3055#endif
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003056}
3057
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003058static bool isInAllocaArgument(CGCXXABI &ABI, QualType type) {
3059 const CXXRecordDecl *RD = type->getAsCXXRecordDecl();
3060 return RD && ABI.getRecordArgABI(RD) == CGCXXABI::RAA_DirectInMemory;
3061}
3062
John McCall7f416cc2015-09-08 08:05:57 +00003063static AggValueSlot createPlaceholderSlot(CodeGenFunction &CGF,
3064 QualType Ty) {
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003065 // FIXME: Generate IR in one pass, rather than going back and fixing up these
3066 // placeholders.
3067 llvm::Type *IRTy = CGF.ConvertTypeForMem(Ty);
Peter Collingbourneb367c562016-11-28 22:30:21 +00003068 llvm::Type *IRPtrTy = IRTy->getPointerTo();
3069 llvm::Value *Placeholder = llvm::UndefValue::get(IRPtrTy->getPointerTo());
John McCall7f416cc2015-09-08 08:05:57 +00003070
3071 // FIXME: When we generate this IR in one pass, we shouldn't need
3072 // this win32-specific alignment hack.
3073 CharUnits Align = CharUnits::fromQuantity(4);
Peter Collingbourneb367c562016-11-28 22:30:21 +00003074 Placeholder = CGF.Builder.CreateAlignedLoad(IRPtrTy, Placeholder, Align);
John McCall7f416cc2015-09-08 08:05:57 +00003075
3076 return AggValueSlot::forAddr(Address(Placeholder, Align),
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003077 Ty.getQualifiers(),
3078 AggValueSlot::IsNotDestructed,
3079 AggValueSlot::DoesNotNeedGCBarriers,
Richard Smithe78fac52018-04-05 20:52:58 +00003080 AggValueSlot::IsNotAliased,
3081 AggValueSlot::DoesNotOverlap);
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003082}
3083
John McCall32ea9692011-03-11 20:59:21 +00003084void CodeGenFunction::EmitDelegateCallArg(CallArgList &args,
Nick Lewycky2d84e842013-10-02 02:29:49 +00003085 const VarDecl *param,
3086 SourceLocation loc) {
John McCall23f66262010-05-26 22:34:26 +00003087 // StartFunction converted the ABI-lowered parameter(s) into a
3088 // local alloca. We need to turn that into an r-value suitable
3089 // for EmitCall.
John McCall7f416cc2015-09-08 08:05:57 +00003090 Address local = GetAddrOfLocalVar(param);
John McCall23f66262010-05-26 22:34:26 +00003091
John McCall32ea9692011-03-11 20:59:21 +00003092 QualType type = param->getType();
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00003093
Reid Kleckner25b56022018-12-12 23:46:06 +00003094 if (isInAllocaArgument(CGM.getCXXABI(), type)) {
3095 CGM.ErrorUnsupported(param, "forwarded non-trivially copyable parameter");
3096 }
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003097
John McCall811b2912016-11-18 01:08:24 +00003098 // GetAddrOfLocalVar returns a pointer-to-pointer for references,
3099 // but the argument needs to be the original pointer.
3100 if (type->isReferenceType()) {
3101 args.add(RValue::get(Builder.CreateLoad(local)), type);
3102
3103 // In ARC, move out of consumed arguments so that the release cleanup
3104 // entered by StartFunction doesn't cause an over-release. This isn't
3105 // optimal -O0 code generation, but it should get cleaned up when
3106 // optimization is enabled. This also assumes that delegate calls are
3107 // performed exactly once for a set of arguments, but that should be safe.
3108 } else if (getLangOpts().ObjCAutoRefCount &&
3109 param->hasAttr<NSConsumedAttr>() &&
3110 type->isObjCRetainableType()) {
3111 llvm::Value *ptr = Builder.CreateLoad(local);
3112 auto null =
3113 llvm::ConstantPointerNull::get(cast<llvm::PointerType>(ptr->getType()));
3114 Builder.CreateStore(null, local);
3115 args.add(RValue::get(ptr), type);
3116
Richard Smithd62d4982016-06-14 01:13:21 +00003117 // For the most part, we just need to load the alloca, except that
3118 // aggregate r-values are actually pointers to temporaries.
John McCall811b2912016-11-18 01:08:24 +00003119 } else {
Richard Smithd62d4982016-06-14 01:13:21 +00003120 args.add(convertTempToRValue(local, type, loc), type);
John McCall811b2912016-11-18 01:08:24 +00003121 }
Akira Hatanakaccda3d22018-04-27 06:57:00 +00003122
3123 // Deactivate the cleanup for the callee-destructed param that was pushed.
3124 if (hasAggregateEvaluationKind(type) && !CurFuncIsThunk &&
Akira Hatanaka85282972018-05-15 21:00:30 +00003125 type->getAs<RecordType>()->getDecl()->isParamDestroyedInCallee() &&
3126 type.isDestructedType()) {
Akira Hatanakaccda3d22018-04-27 06:57:00 +00003127 EHScopeStack::stable_iterator cleanup =
3128 CalleeDestructedParamCleanups.lookup(cast<ParmVarDecl>(param));
3129 assert(cleanup.isValid() &&
3130 "cleanup for callee-destructed param not recorded");
3131 // This unreachable is a temporary marker which will be removed later.
3132 llvm::Instruction *isActive = Builder.CreateUnreachable();
3133 args.addArgCleanupDeactivation(cleanup, isActive);
3134 }
John McCall23f66262010-05-26 22:34:26 +00003135}
3136
John McCall31168b02011-06-15 23:02:42 +00003137static bool isProvablyNull(llvm::Value *addr) {
3138 return isa<llvm::ConstantPointerNull>(addr);
3139}
3140
John McCall31168b02011-06-15 23:02:42 +00003141/// Emit the actual writing-back of a writeback.
3142static void emitWriteback(CodeGenFunction &CGF,
3143 const CallArgList::Writeback &writeback) {
John McCalleff18842013-03-23 02:35:54 +00003144 const LValue &srcLV = writeback.Source;
John McCall7f416cc2015-09-08 08:05:57 +00003145 Address srcAddr = srcLV.getAddress();
3146 assert(!isProvablyNull(srcAddr.getPointer()) &&
John McCall31168b02011-06-15 23:02:42 +00003147 "shouldn't have writeback for provably null argument");
3148
Craig Topper8a13c412014-05-21 05:09:00 +00003149 llvm::BasicBlock *contBB = nullptr;
John McCall31168b02011-06-15 23:02:42 +00003150
3151 // If the argument wasn't provably non-null, we need to null check
3152 // before doing the store.
Nuno Lopes9211cee2017-09-09 18:25:36 +00003153 bool provablyNonNull = llvm::isKnownNonZero(srcAddr.getPointer(),
3154 CGF.CGM.getDataLayout());
John McCall31168b02011-06-15 23:02:42 +00003155 if (!provablyNonNull) {
3156 llvm::BasicBlock *writebackBB = CGF.createBasicBlock("icr.writeback");
3157 contBB = CGF.createBasicBlock("icr.done");
3158
John McCall7f416cc2015-09-08 08:05:57 +00003159 llvm::Value *isNull =
3160 CGF.Builder.CreateIsNull(srcAddr.getPointer(), "icr.isnull");
John McCall31168b02011-06-15 23:02:42 +00003161 CGF.Builder.CreateCondBr(isNull, contBB, writebackBB);
3162 CGF.EmitBlock(writebackBB);
3163 }
3164
3165 // Load the value to writeback.
3166 llvm::Value *value = CGF.Builder.CreateLoad(writeback.Temporary);
3167
3168 // Cast it back, in case we're writing an id to a Foo* or something.
John McCall7f416cc2015-09-08 08:05:57 +00003169 value = CGF.Builder.CreateBitCast(value, srcAddr.getElementType(),
3170 "icr.writeback-cast");
Fangrui Song6907ce22018-07-30 19:24:48 +00003171
John McCall31168b02011-06-15 23:02:42 +00003172 // Perform the writeback.
John McCalleff18842013-03-23 02:35:54 +00003173
3174 // If we have a "to use" value, it's something we need to emit a use
3175 // of. This has to be carefully threaded in: if it's done after the
3176 // release it's potentially undefined behavior (and the optimizer
3177 // will ignore it), and if it happens before the retain then the
3178 // optimizer could move the release there.
3179 if (writeback.ToUse) {
3180 assert(srcLV.getObjCLifetime() == Qualifiers::OCL_Strong);
3181
3182 // Retain the new value. No need to block-copy here: the block's
3183 // being passed up the stack.
3184 value = CGF.EmitARCRetainNonBlock(value);
3185
3186 // Emit the intrinsic use here.
3187 CGF.EmitARCIntrinsicUse(writeback.ToUse);
3188
3189 // Load the old value (primitively).
Nick Lewycky2d84e842013-10-02 02:29:49 +00003190 llvm::Value *oldValue = CGF.EmitLoadOfScalar(srcLV, SourceLocation());
John McCalleff18842013-03-23 02:35:54 +00003191
3192 // Put the new value in place (primitively).
3193 CGF.EmitStoreOfScalar(value, srcLV, /*init*/ false);
3194
3195 // Release the old value.
3196 CGF.EmitARCRelease(oldValue, srcLV.isARCPreciseLifetime());
3197
3198 // Otherwise, we can just do a normal lvalue store.
3199 } else {
3200 CGF.EmitStoreThroughLValue(RValue::get(value), srcLV);
3201 }
John McCall31168b02011-06-15 23:02:42 +00003202
3203 // Jump to the continuation block.
3204 if (!provablyNonNull)
3205 CGF.EmitBlock(contBB);
3206}
3207
3208static void emitWritebacks(CodeGenFunction &CGF,
3209 const CallArgList &args) {
Aaron Ballman36a7fa82014-03-17 17:22:27 +00003210 for (const auto &I : args.writebacks())
3211 emitWriteback(CGF, I);
John McCall31168b02011-06-15 23:02:42 +00003212}
3213
Reid Kleckner23f4c4b2013-06-21 12:45:15 +00003214static void deactivateArgCleanupsBeforeCall(CodeGenFunction &CGF,
3215 const CallArgList &CallArgs) {
Reid Kleckner23f4c4b2013-06-21 12:45:15 +00003216 ArrayRef<CallArgList::CallArgCleanup> Cleanups =
3217 CallArgs.getCleanupsToDeactivate();
3218 // Iterate in reverse to increase the likelihood of popping the cleanup.
Pete Cooper57d3f142015-07-30 17:22:52 +00003219 for (const auto &I : llvm::reverse(Cleanups)) {
3220 CGF.DeactivateCleanupBlock(I.Cleanup, I.IsActiveIP);
3221 I.IsActiveIP->eraseFromParent();
Reid Kleckner23f4c4b2013-06-21 12:45:15 +00003222 }
3223}
3224
John McCalleff18842013-03-23 02:35:54 +00003225static const Expr *maybeGetUnaryAddrOfOperand(const Expr *E) {
3226 if (const UnaryOperator *uop = dyn_cast<UnaryOperator>(E->IgnoreParens()))
3227 if (uop->getOpcode() == UO_AddrOf)
3228 return uop->getSubExpr();
Craig Topper8a13c412014-05-21 05:09:00 +00003229 return nullptr;
John McCalleff18842013-03-23 02:35:54 +00003230}
3231
John McCall31168b02011-06-15 23:02:42 +00003232/// Emit an argument that's being passed call-by-writeback. That is,
John McCall7f416cc2015-09-08 08:05:57 +00003233/// we are passing the address of an __autoreleased temporary; it
3234/// might be copy-initialized with the current value of the given
3235/// address, but it will definitely be copied out of after the call.
John McCall31168b02011-06-15 23:02:42 +00003236static void emitWritebackArg(CodeGenFunction &CGF, CallArgList &args,
3237 const ObjCIndirectCopyRestoreExpr *CRE) {
John McCalleff18842013-03-23 02:35:54 +00003238 LValue srcLV;
3239
3240 // Make an optimistic effort to emit the address as an l-value.
Eric Christopher2c4555a2015-06-19 01:52:53 +00003241 // This can fail if the argument expression is more complicated.
John McCalleff18842013-03-23 02:35:54 +00003242 if (const Expr *lvExpr = maybeGetUnaryAddrOfOperand(CRE->getSubExpr())) {
3243 srcLV = CGF.EmitLValue(lvExpr);
3244
3245 // Otherwise, just emit it as a scalar.
3246 } else {
John McCall7f416cc2015-09-08 08:05:57 +00003247 Address srcAddr = CGF.EmitPointerWithAlignment(CRE->getSubExpr());
John McCalleff18842013-03-23 02:35:54 +00003248
3249 QualType srcAddrType =
3250 CRE->getSubExpr()->getType()->castAs<PointerType>()->getPointeeType();
John McCall7f416cc2015-09-08 08:05:57 +00003251 srcLV = CGF.MakeAddrLValue(srcAddr, srcAddrType);
John McCalleff18842013-03-23 02:35:54 +00003252 }
John McCall7f416cc2015-09-08 08:05:57 +00003253 Address srcAddr = srcLV.getAddress();
John McCall31168b02011-06-15 23:02:42 +00003254
3255 // The dest and src types don't necessarily match in LLVM terms
3256 // because of the crazy ObjC compatibility rules.
3257
Chris Lattner2192fe52011-07-18 04:24:23 +00003258 llvm::PointerType *destType =
John McCall31168b02011-06-15 23:02:42 +00003259 cast<llvm::PointerType>(CGF.ConvertType(CRE->getType()));
3260
3261 // If the address is a constant null, just pass the appropriate null.
John McCall7f416cc2015-09-08 08:05:57 +00003262 if (isProvablyNull(srcAddr.getPointer())) {
John McCall31168b02011-06-15 23:02:42 +00003263 args.add(RValue::get(llvm::ConstantPointerNull::get(destType)),
3264 CRE->getType());
3265 return;
3266 }
3267
John McCall31168b02011-06-15 23:02:42 +00003268 // Create the temporary.
John McCall7f416cc2015-09-08 08:05:57 +00003269 Address temp = CGF.CreateTempAlloca(destType->getElementType(),
3270 CGF.getPointerAlign(),
3271 "icr.temp");
Fariborz Jahanianfbd19742012-11-27 23:02:53 +00003272 // Loading an l-value can introduce a cleanup if the l-value is __weak,
3273 // and that cleanup will be conditional if we can't prove that the l-value
3274 // isn't null, so we need to register a dominating point so that the cleanups
3275 // system will make valid IR.
3276 CodeGenFunction::ConditionalEvaluation condEval(CGF);
Fangrui Song6907ce22018-07-30 19:24:48 +00003277
John McCall31168b02011-06-15 23:02:42 +00003278 // Zero-initialize it if we're not doing a copy-initialization.
3279 bool shouldCopy = CRE->shouldCopy();
3280 if (!shouldCopy) {
3281 llvm::Value *null =
3282 llvm::ConstantPointerNull::get(
3283 cast<llvm::PointerType>(destType->getElementType()));
3284 CGF.Builder.CreateStore(null, temp);
3285 }
Craig Topper8a13c412014-05-21 05:09:00 +00003286
3287 llvm::BasicBlock *contBB = nullptr;
3288 llvm::BasicBlock *originBB = nullptr;
John McCall31168b02011-06-15 23:02:42 +00003289
3290 // If the address is *not* known to be non-null, we need to switch.
3291 llvm::Value *finalArgument;
3292
Nuno Lopes9211cee2017-09-09 18:25:36 +00003293 bool provablyNonNull = llvm::isKnownNonZero(srcAddr.getPointer(),
3294 CGF.CGM.getDataLayout());
John McCall31168b02011-06-15 23:02:42 +00003295 if (provablyNonNull) {
John McCall7f416cc2015-09-08 08:05:57 +00003296 finalArgument = temp.getPointer();
John McCall31168b02011-06-15 23:02:42 +00003297 } else {
John McCall7f416cc2015-09-08 08:05:57 +00003298 llvm::Value *isNull =
3299 CGF.Builder.CreateIsNull(srcAddr.getPointer(), "icr.isnull");
John McCall31168b02011-06-15 23:02:42 +00003300
Fangrui Song6907ce22018-07-30 19:24:48 +00003301 finalArgument = CGF.Builder.CreateSelect(isNull,
John McCall31168b02011-06-15 23:02:42 +00003302 llvm::ConstantPointerNull::get(destType),
John McCall7f416cc2015-09-08 08:05:57 +00003303 temp.getPointer(), "icr.argument");
John McCall31168b02011-06-15 23:02:42 +00003304
3305 // If we need to copy, then the load has to be conditional, which
3306 // means we need control flow.
3307 if (shouldCopy) {
John McCalleff18842013-03-23 02:35:54 +00003308 originBB = CGF.Builder.GetInsertBlock();
John McCall31168b02011-06-15 23:02:42 +00003309 contBB = CGF.createBasicBlock("icr.cont");
3310 llvm::BasicBlock *copyBB = CGF.createBasicBlock("icr.copy");
3311 CGF.Builder.CreateCondBr(isNull, contBB, copyBB);
3312 CGF.EmitBlock(copyBB);
Fariborz Jahanianfbd19742012-11-27 23:02:53 +00003313 condEval.begin(CGF);
John McCall31168b02011-06-15 23:02:42 +00003314 }
3315 }
3316
Craig Topper8a13c412014-05-21 05:09:00 +00003317 llvm::Value *valueToUse = nullptr;
John McCalleff18842013-03-23 02:35:54 +00003318
John McCall31168b02011-06-15 23:02:42 +00003319 // Perform a copy if necessary.
3320 if (shouldCopy) {
Nick Lewycky2d84e842013-10-02 02:29:49 +00003321 RValue srcRV = CGF.EmitLoadOfLValue(srcLV, SourceLocation());
John McCall31168b02011-06-15 23:02:42 +00003322 assert(srcRV.isScalar());
3323
3324 llvm::Value *src = srcRV.getScalarVal();
3325 src = CGF.Builder.CreateBitCast(src, destType->getElementType(),
3326 "icr.cast");
3327
3328 // Use an ordinary store, not a store-to-lvalue.
3329 CGF.Builder.CreateStore(src, temp);
John McCalleff18842013-03-23 02:35:54 +00003330
3331 // If optimization is enabled, and the value was held in a
3332 // __strong variable, we need to tell the optimizer that this
3333 // value has to stay alive until we're doing the store back.
3334 // This is because the temporary is effectively unretained,
3335 // and so otherwise we can violate the high-level semantics.
3336 if (CGF.CGM.getCodeGenOpts().OptimizationLevel != 0 &&
3337 srcLV.getObjCLifetime() == Qualifiers::OCL_Strong) {
3338 valueToUse = src;
3339 }
John McCall31168b02011-06-15 23:02:42 +00003340 }
Fangrui Song6907ce22018-07-30 19:24:48 +00003341
John McCall31168b02011-06-15 23:02:42 +00003342 // Finish the control flow if we needed it.
Fariborz Jahanianfbd19742012-11-27 23:02:53 +00003343 if (shouldCopy && !provablyNonNull) {
John McCalleff18842013-03-23 02:35:54 +00003344 llvm::BasicBlock *copyBB = CGF.Builder.GetInsertBlock();
John McCall31168b02011-06-15 23:02:42 +00003345 CGF.EmitBlock(contBB);
John McCalleff18842013-03-23 02:35:54 +00003346
3347 // Make a phi for the value to intrinsically use.
3348 if (valueToUse) {
3349 llvm::PHINode *phiToUse = CGF.Builder.CreatePHI(valueToUse->getType(), 2,
3350 "icr.to-use");
3351 phiToUse->addIncoming(valueToUse, copyBB);
3352 phiToUse->addIncoming(llvm::UndefValue::get(valueToUse->getType()),
3353 originBB);
3354 valueToUse = phiToUse;
3355 }
3356
Fariborz Jahanianfbd19742012-11-27 23:02:53 +00003357 condEval.end(CGF);
3358 }
John McCall31168b02011-06-15 23:02:42 +00003359
John McCalleff18842013-03-23 02:35:54 +00003360 args.addWriteback(srcLV, temp, valueToUse);
John McCall31168b02011-06-15 23:02:42 +00003361 args.add(RValue::get(finalArgument), CRE->getType());
3362}
3363
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003364void CallArgList::allocateArgumentMemory(CodeGenFunction &CGF) {
Richard Smith762672a2016-09-28 19:09:10 +00003365 assert(!StackBase);
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003366
3367 // Save the stack.
3368 llvm::Function *F = CGF.CGM.getIntrinsic(llvm::Intrinsic::stacksave);
David Blaikie43f9bb72015-05-18 22:14:03 +00003369 StackBase = CGF.Builder.CreateCall(F, {}, "inalloca.save");
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003370}
3371
Nico Weber8cdb3f92015-08-25 18:43:32 +00003372void CallArgList::freeArgumentMemory(CodeGenFunction &CGF) const {
3373 if (StackBase) {
Reid Kleckner7c2f9e82015-10-08 00:17:45 +00003374 // Restore the stack after the call.
Nico Weber8cdb3f92015-08-25 18:43:32 +00003375 llvm::Value *F = CGF.CGM.getIntrinsic(llvm::Intrinsic::stackrestore);
Nico Weber8cdb3f92015-08-25 18:43:32 +00003376 CGF.Builder.CreateCall(F, StackBase);
3377 }
3378}
3379
Nuno Lopes1ba2d782015-05-30 16:11:40 +00003380void CodeGenFunction::EmitNonNullArgCheck(RValue RV, QualType ArgType,
3381 SourceLocation ArgLoc,
Vedant Kumared00ea02017-03-06 05:28:22 +00003382 AbstractCallee AC,
Nuno Lopes1ba2d782015-05-30 16:11:40 +00003383 unsigned ParmNum) {
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003384 if (!AC.getDecl() || !(SanOpts.has(SanitizerKind::NonnullAttribute) ||
3385 SanOpts.has(SanitizerKind::NullabilityArg)))
Alexey Samsonov8e1162c2014-09-08 17:22:45 +00003386 return;
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003387
3388 // The param decl may be missing in a variadic function.
Vedant Kumared00ea02017-03-06 05:28:22 +00003389 auto PVD = ParmNum < AC.getNumParams() ? AC.getParamDecl(ParmNum) : nullptr;
Alexey Samsonov8e1162c2014-09-08 17:22:45 +00003390 unsigned ArgNo = PVD ? PVD->getFunctionScopeIndex() : ParmNum;
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003391
Fangrui Song6907ce22018-07-30 19:24:48 +00003392 // Prefer the nonnull attribute if it's present.
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003393 const NonNullAttr *NNAttr = nullptr;
3394 if (SanOpts.has(SanitizerKind::NonnullAttribute))
3395 NNAttr = getNonNullAttr(AC.getDecl(), PVD, ArgType, ArgNo);
3396
3397 bool CanCheckNullability = false;
3398 if (SanOpts.has(SanitizerKind::NullabilityArg) && !NNAttr && PVD) {
3399 auto Nullability = PVD->getType()->getNullability(getContext());
3400 CanCheckNullability = Nullability &&
3401 *Nullability == NullabilityKind::NonNull &&
3402 PVD->getTypeSourceInfo();
3403 }
3404
3405 if (!NNAttr && !CanCheckNullability)
Alexey Samsonov8e1162c2014-09-08 17:22:45 +00003406 return;
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003407
3408 SourceLocation AttrLoc;
3409 SanitizerMask CheckKind;
Vedant Kumar2b9f48a2017-03-14 16:48:29 +00003410 SanitizerHandler Handler;
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003411 if (NNAttr) {
3412 AttrLoc = NNAttr->getLocation();
3413 CheckKind = SanitizerKind::NonnullAttribute;
Vedant Kumar2b9f48a2017-03-14 16:48:29 +00003414 Handler = SanitizerHandler::NonnullArg;
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003415 } else {
3416 AttrLoc = PVD->getTypeSourceInfo()->getTypeLoc().findNullabilityLoc();
3417 CheckKind = SanitizerKind::NullabilityArg;
Vedant Kumar2b9f48a2017-03-14 16:48:29 +00003418 Handler = SanitizerHandler::NullabilityArg;
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003419 }
3420
Nuno Lopes1ba2d782015-05-30 16:11:40 +00003421 SanitizerScope SanScope(this);
Alexey Samsonov8e1162c2014-09-08 17:22:45 +00003422 assert(RV.isScalar());
3423 llvm::Value *V = RV.getScalarVal();
3424 llvm::Value *Cond =
Nuno Lopes1ba2d782015-05-30 16:11:40 +00003425 Builder.CreateICmpNE(V, llvm::Constant::getNullValue(V->getType()));
Alexey Samsonov8e1162c2014-09-08 17:22:45 +00003426 llvm::Constant *StaticData[] = {
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003427 EmitCheckSourceLocation(ArgLoc), EmitCheckSourceLocation(AttrLoc),
Nuno Lopes1ba2d782015-05-30 16:11:40 +00003428 llvm::ConstantInt::get(Int32Ty, ArgNo + 1),
Alexey Samsonov8e1162c2014-09-08 17:22:45 +00003429 };
Vedant Kumar2b9f48a2017-03-14 16:48:29 +00003430 EmitCheck(std::make_pair(Cond, CheckKind), Handler, StaticData, None);
Alexey Samsonov8e1162c2014-09-08 17:22:45 +00003431}
3432
David Blaikief05779e2015-07-21 18:37:18 +00003433void CodeGenFunction::EmitCallArgs(
3434 CallArgList &Args, ArrayRef<QualType> ArgTypes,
3435 llvm::iterator_range<CallExpr::const_arg_iterator> ArgRange,
Vedant Kumared00ea02017-03-06 05:28:22 +00003436 AbstractCallee AC, unsigned ParamsToSkip, EvaluationOrder Order) {
David Blaikief05779e2015-07-21 18:37:18 +00003437 assert((int)ArgTypes.size() == (ArgRange.end() - ArgRange.begin()));
George Burgess IV3e3bb95b2015-12-02 21:58:08 +00003438
Reid Kleckner739756c2013-12-04 19:23:12 +00003439 // We *have* to evaluate arguments from right to left in the MS C++ ABI,
Richard Smitha560ccf2016-09-29 21:30:12 +00003440 // because arguments are destroyed left to right in the callee. As a special
3441 // case, there are certain language constructs that require left-to-right
3442 // evaluation, and in those cases we consider the evaluation order requirement
3443 // to trump the "destruction order is reverse construction order" guarantee.
3444 bool LeftToRight =
3445 CGM.getTarget().getCXXABI().areArgsDestroyedLeftToRightInCallee()
3446 ? Order == EvaluationOrder::ForceLeftToRight
3447 : Order != EvaluationOrder::ForceRightToLeft;
3448
George Burgess IV0d6592a2017-02-23 05:59:56 +00003449 auto MaybeEmitImplicitObjectSize = [&](unsigned I, const Expr *Arg,
3450 RValue EmittedArg) {
Vedant Kumared00ea02017-03-06 05:28:22 +00003451 if (!AC.hasFunctionDecl() || I >= AC.getNumParams())
George Burgess IV0d6592a2017-02-23 05:59:56 +00003452 return;
Vedant Kumared00ea02017-03-06 05:28:22 +00003453 auto *PS = AC.getParamDecl(I)->getAttr<PassObjectSizeAttr>();
George Burgess IV0d6592a2017-02-23 05:59:56 +00003454 if (PS == nullptr)
3455 return;
3456
3457 const auto &Context = getContext();
3458 auto SizeTy = Context.getSizeType();
3459 auto T = Builder.getIntNTy(Context.getTypeSize(SizeTy));
3460 assert(EmittedArg.getScalarVal() && "We emitted nothing for the arg?");
3461 llvm::Value *V = evaluateOrEmitBuiltinObjectSize(Arg, PS->getType(), T,
Erik Pilkington9c3b5882019-01-30 20:34:53 +00003462 EmittedArg.getScalarVal(),
3463 /*IsDynamic=*/false);
George Burgess IV0d6592a2017-02-23 05:59:56 +00003464 Args.add(RValue::get(V), SizeTy);
3465 // If we're emitting args in reverse, be sure to do so with
3466 // pass_object_size, as well.
3467 if (!LeftToRight)
3468 std::swap(Args.back(), *(&Args.back() - 1));
3469 };
3470
Richard Smitha560ccf2016-09-29 21:30:12 +00003471 // Insert a stack save if we're going to need any inalloca args.
3472 bool HasInAllocaArgs = false;
3473 if (CGM.getTarget().getCXXABI().isMicrosoft()) {
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003474 for (ArrayRef<QualType>::iterator I = ArgTypes.begin(), E = ArgTypes.end();
3475 I != E && !HasInAllocaArgs; ++I)
3476 HasInAllocaArgs = isInAllocaArgument(CGM.getCXXABI(), *I);
3477 if (HasInAllocaArgs) {
3478 assert(getTarget().getTriple().getArch() == llvm::Triple::x86);
3479 Args.allocateArgumentMemory(*this);
3480 }
Richard Smitha560ccf2016-09-29 21:30:12 +00003481 }
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003482
Richard Smitha560ccf2016-09-29 21:30:12 +00003483 // Evaluate each argument in the appropriate order.
3484 size_t CallArgsStart = Args.size();
3485 for (unsigned I = 0, E = ArgTypes.size(); I != E; ++I) {
3486 unsigned Idx = LeftToRight ? I : E - I - 1;
3487 CallExpr::const_arg_iterator Arg = ArgRange.begin() + Idx;
George Burgess IV0d6592a2017-02-23 05:59:56 +00003488 unsigned InitialArgSize = Args.size();
Akira Hatanaka46dd7dbc2017-06-28 00:42:48 +00003489 // If *Arg is an ObjCIndirectCopyRestoreExpr, check that either the types of
3490 // the argument and parameter match or the objc method is parameterized.
3491 assert((!isa<ObjCIndirectCopyRestoreExpr>(*Arg) ||
3492 getContext().hasSameUnqualifiedType((*Arg)->getType(),
3493 ArgTypes[Idx]) ||
3494 (isa<ObjCMethodDecl>(AC.getDecl()) &&
3495 isObjCMethodWithTypeParams(cast<ObjCMethodDecl>(AC.getDecl())))) &&
3496 "Argument and parameter types don't match");
Richard Smitha560ccf2016-09-29 21:30:12 +00003497 EmitCallArg(Args, *Arg, ArgTypes[Idx]);
George Burgess IV0d6592a2017-02-23 05:59:56 +00003498 // In particular, we depend on it being the last arg in Args, and the
3499 // objectsize bits depend on there only being one arg if !LeftToRight.
3500 assert(InitialArgSize + 1 == Args.size() &&
3501 "The code below depends on only adding one arg per EmitCallArg");
3502 (void)InitialArgSize;
Yaxun Liu5b330e82018-03-15 15:25:19 +00003503 // Since pointer argument are never emitted as LValue, it is safe to emit
3504 // non-null argument check for r-value only.
3505 if (!Args.back().hasLValue()) {
3506 RValue RVArg = Args.back().getKnownRValue();
3507 EmitNonNullArgCheck(RVArg, ArgTypes[Idx], (*Arg)->getExprLoc(), AC,
3508 ParamsToSkip + Idx);
3509 // @llvm.objectsize should never have side-effects and shouldn't need
3510 // destruction/cleanups, so we can safely "emit" it after its arg,
3511 // regardless of right-to-leftness
3512 MaybeEmitImplicitObjectSize(Idx, *Arg, RVArg);
3513 }
Richard Smitha560ccf2016-09-29 21:30:12 +00003514 }
Reid Kleckner739756c2013-12-04 19:23:12 +00003515
Richard Smitha560ccf2016-09-29 21:30:12 +00003516 if (!LeftToRight) {
Reid Kleckner739756c2013-12-04 19:23:12 +00003517 // Un-reverse the arguments we just evaluated so they match up with the LLVM
3518 // IR function.
3519 std::reverse(Args.begin() + CallArgsStart, Args.end());
Reid Kleckner739756c2013-12-04 19:23:12 +00003520 }
3521}
3522
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003523namespace {
3524
David Blaikie7e70d682015-08-18 22:40:54 +00003525struct DestroyUnpassedArg final : EHScopeStack::Cleanup {
John McCall7f416cc2015-09-08 08:05:57 +00003526 DestroyUnpassedArg(Address Addr, QualType Ty)
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003527 : Addr(Addr), Ty(Ty) {}
3528
John McCall7f416cc2015-09-08 08:05:57 +00003529 Address Addr;
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003530 QualType Ty;
3531
Craig Topper4f12f102014-03-12 06:41:41 +00003532 void Emit(CodeGenFunction &CGF, Flags flags) override {
Akira Hatanaka7275da02018-02-28 07:15:55 +00003533 QualType::DestructionKind DtorKind = Ty.isDestructedType();
3534 if (DtorKind == QualType::DK_cxx_destructor) {
3535 const CXXDestructorDecl *Dtor = Ty->getAsCXXRecordDecl()->getDestructor();
3536 assert(!Dtor->isTrivial());
3537 CGF.EmitCXXDestructorCall(Dtor, Dtor_Complete, /*for vbase*/ false,
3538 /*Delegating=*/false, Addr);
3539 } else {
3540 CGF.callCStructDestructor(CGF.MakeAddrLValue(Addr, Ty));
3541 }
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003542 }
3543};
3544
David Blaikie38b25912015-02-09 19:13:51 +00003545struct DisableDebugLocationUpdates {
3546 CodeGenFunction &CGF;
3547 bool disabledDebugInfo;
3548 DisableDebugLocationUpdates(CodeGenFunction &CGF, const Expr *E) : CGF(CGF) {
3549 if ((disabledDebugInfo = isa<CXXDefaultArgExpr>(E) && CGF.getDebugInfo()))
3550 CGF.disableDebugInfo();
3551 }
3552 ~DisableDebugLocationUpdates() {
3553 if (disabledDebugInfo)
3554 CGF.enableDebugInfo();
3555 }
3556};
3557
Benjamin Kramer5b4296a2015-10-28 17:16:26 +00003558} // end anonymous namespace
3559
Yaxun Liu5b330e82018-03-15 15:25:19 +00003560RValue CallArg::getRValue(CodeGenFunction &CGF) const {
3561 if (!HasLV)
3562 return RV;
3563 LValue Copy = CGF.MakeAddrLValue(CGF.CreateMemTemp(Ty), Ty);
Richard Smithe78fac52018-04-05 20:52:58 +00003564 CGF.EmitAggregateCopy(Copy, LV, Ty, AggValueSlot::DoesNotOverlap,
3565 LV.isVolatile());
Yaxun Liu5b330e82018-03-15 15:25:19 +00003566 IsUsed = true;
3567 return RValue::getAggregate(Copy.getAddress());
3568}
3569
3570void CallArg::copyInto(CodeGenFunction &CGF, Address Addr) const {
3571 LValue Dst = CGF.MakeAddrLValue(Addr, Ty);
3572 if (!HasLV && RV.isScalar())
3573 CGF.EmitStoreOfScalar(RV.getScalarVal(), Dst, /*init=*/true);
3574 else if (!HasLV && RV.isComplex())
3575 CGF.EmitStoreOfComplex(RV.getComplexVal(), Dst, /*init=*/true);
3576 else {
3577 auto Addr = HasLV ? LV.getAddress() : RV.getAggregateAddress();
3578 LValue SrcLV = CGF.MakeAddrLValue(Addr, Ty);
Richard Smithe78fac52018-04-05 20:52:58 +00003579 // We assume that call args are never copied into subobjects.
3580 CGF.EmitAggregateCopy(Dst, SrcLV, Ty, AggValueSlot::DoesNotOverlap,
Yaxun Liu5b330e82018-03-15 15:25:19 +00003581 HasLV ? LV.isVolatileQualified()
3582 : RV.isVolatileQualified());
3583 }
3584 IsUsed = true;
3585}
3586
John McCall32ea9692011-03-11 20:59:21 +00003587void CodeGenFunction::EmitCallArg(CallArgList &args, const Expr *E,
3588 QualType type) {
David Blaikie38b25912015-02-09 19:13:51 +00003589 DisableDebugLocationUpdates Dis(*this, E);
John McCall31168b02011-06-15 23:02:42 +00003590 if (const ObjCIndirectCopyRestoreExpr *CRE
3591 = dyn_cast<ObjCIndirectCopyRestoreExpr>(E)) {
Richard Smith9c6890a2012-11-01 22:30:59 +00003592 assert(getLangOpts().ObjCAutoRefCount);
John McCall31168b02011-06-15 23:02:42 +00003593 return emitWritebackArg(*this, args, CRE);
3594 }
3595
John McCall0a76c0c2011-08-26 18:42:59 +00003596 assert(type->isReferenceType() == E->isGLValue() &&
3597 "reference binding to unmaterialized r-value!");
3598
John McCall17054bd62011-08-26 21:08:13 +00003599 if (E->isGLValue()) {
3600 assert(E->getObjectKind() == OK_Ordinary);
Richard Smitha1c9d4d2013-06-12 23:38:09 +00003601 return args.add(EmitReferenceBindingToExpr(E), type);
John McCall17054bd62011-08-26 21:08:13 +00003602 }
Mike Stump11289f42009-09-09 15:08:12 +00003603
Reid Kleckner23f4c4b2013-06-21 12:45:15 +00003604 bool HasAggregateEvalKind = hasAggregateEvaluationKind(type);
3605
3606 // In the Microsoft C++ ABI, aggregate arguments are destructed by the callee.
3607 // However, we still have to push an EH-only cleanup in case we unwind before
3608 // we make it to the call.
Akira Hatanaka85282972018-05-15 21:00:30 +00003609 if (HasAggregateEvalKind &&
3610 type->getAs<RecordType>()->getDecl()->isParamDestroyedInCallee()) {
Reid Klecknerac640602014-05-01 03:07:18 +00003611 // If we're using inalloca, use the argument memory. Otherwise, use a
Reid Klecknere39ee212014-05-03 00:33:28 +00003612 // temporary.
Reid Klecknerac640602014-05-01 03:07:18 +00003613 AggValueSlot Slot;
3614 if (args.isUsingInAlloca())
3615 Slot = createPlaceholderSlot(*this, type);
3616 else
3617 Slot = CreateAggTemp(type, "agg.tmp");
Reid Klecknere39ee212014-05-03 00:33:28 +00003618
Akira Hatanaka4ce0e5a2018-04-18 23:33:15 +00003619 bool DestroyedInCallee = true, NeedsEHCleanup = true;
3620 if (const auto *RD = type->getAsCXXRecordDecl())
3621 DestroyedInCallee = RD->hasNonTrivialDestructor();
3622 else
3623 NeedsEHCleanup = needsEHCleanup(type.isDestructedType());
3624
3625 if (DestroyedInCallee)
3626 Slot.setExternallyDestructed();
Reid Klecknere39ee212014-05-03 00:33:28 +00003627
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003628 EmitAggExpr(E, Slot);
3629 RValue RV = Slot.asRValue();
3630 args.add(RV, type);
Reid Kleckner23f4c4b2013-06-21 12:45:15 +00003631
Akira Hatanaka4ce0e5a2018-04-18 23:33:15 +00003632 if (DestroyedInCallee && NeedsEHCleanup) {
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003633 // Create a no-op GEP between the placeholder and the cleanup so we can
3634 // RAUW it successfully. It also serves as a marker of the first
3635 // instruction where the cleanup is active.
John McCall7f416cc2015-09-08 08:05:57 +00003636 pushFullExprCleanup<DestroyUnpassedArg>(EHCleanup, Slot.getAddress(),
3637 type);
Reid Kleckner23f4c4b2013-06-21 12:45:15 +00003638 // This unreachable is a temporary marker which will be removed later.
3639 llvm::Instruction *IsActive = Builder.CreateUnreachable();
3640 args.addArgCleanupDeactivation(EHStack.getInnermostEHScope(), IsActive);
Reid Kleckner23f4c4b2013-06-21 12:45:15 +00003641 }
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003642 return;
Reid Kleckner23f4c4b2013-06-21 12:45:15 +00003643 }
3644
3645 if (HasAggregateEvalKind && isa<ImplicitCastExpr>(E) &&
Eli Friedmandf968192011-05-26 00:10:27 +00003646 cast<CastExpr>(E)->getCastKind() == CK_LValueToRValue) {
3647 LValue L = EmitLValue(cast<CastExpr>(E)->getSubExpr());
3648 assert(L.isSimple());
Yaxun Liu5b330e82018-03-15 15:25:19 +00003649 args.addUncopiedAggregate(L, type);
Eli Friedmandf968192011-05-26 00:10:27 +00003650 return;
3651 }
3652
John McCall32ea9692011-03-11 20:59:21 +00003653 args.add(EmitAnyExprToTemp(E), type);
Anders Carlsson60ce3fe2009-04-08 20:47:54 +00003654}
3655
Reid Kleckner79b0fd72014-10-10 00:05:45 +00003656QualType CodeGenFunction::getVarArgType(const Expr *Arg) {
3657 // System headers on Windows define NULL to 0 instead of 0LL on Win64. MSVC
3658 // implicitly widens null pointer constants that are arguments to varargs
3659 // functions to pointer-sized ints.
3660 if (!getTarget().getTriple().isOSWindows())
3661 return Arg->getType();
3662
3663 if (Arg->getType()->isIntegerType() &&
3664 getContext().getTypeSize(Arg->getType()) <
3665 getContext().getTargetInfo().getPointerWidth(0) &&
3666 Arg->isNullPointerConstant(getContext(),
3667 Expr::NPC_ValueDependentIsNotNull)) {
3668 return getContext().getIntPtrType();
3669 }
3670
3671 return Arg->getType();
3672}
3673
Dan Gohman515a60d2012-02-16 00:57:37 +00003674// In ObjC ARC mode with no ObjC ARC exception safety, tell the ARC
3675// optimizer it can aggressively ignore unwind edges.
3676void
3677CodeGenFunction::AddObjCARCExceptionMetadata(llvm::Instruction *Inst) {
3678 if (CGM.getCodeGenOpts().OptimizationLevel != 0 &&
3679 !CGM.getCodeGenOpts().ObjCAutoRefCountExceptions)
3680 Inst->setMetadata("clang.arc.no_objc_arc_exceptions",
3681 CGM.getNoObjCARCExceptionsMetadata());
3682}
3683
John McCall882987f2013-02-28 19:01:20 +00003684/// Emits a call to the given no-arguments nounwind runtime function.
3685llvm::CallInst *
3686CodeGenFunction::EmitNounwindRuntimeCall(llvm::Value *callee,
3687 const llvm::Twine &name) {
Craig Topper5fc8fc22014-08-27 06:28:36 +00003688 return EmitNounwindRuntimeCall(callee, None, name);
John McCall882987f2013-02-28 19:01:20 +00003689}
3690
3691/// Emits a call to the given nounwind runtime function.
3692llvm::CallInst *
3693CodeGenFunction::EmitNounwindRuntimeCall(llvm::Value *callee,
3694 ArrayRef<llvm::Value*> args,
3695 const llvm::Twine &name) {
3696 llvm::CallInst *call = EmitRuntimeCall(callee, args, name);
3697 call->setDoesNotThrow();
3698 return call;
3699}
3700
3701/// Emits a simple call (never an invoke) to the given no-arguments
3702/// runtime function.
3703llvm::CallInst *
3704CodeGenFunction::EmitRuntimeCall(llvm::Value *callee,
3705 const llvm::Twine &name) {
Craig Topper5fc8fc22014-08-27 06:28:36 +00003706 return EmitRuntimeCall(callee, None, name);
John McCall882987f2013-02-28 19:01:20 +00003707}
3708
David Majnemer0b17d442015-12-15 21:27:59 +00003709// Calls which may throw must have operand bundles indicating which funclet
3710// they are nested within.
Reid Klecknerb75a3f02018-02-09 00:16:41 +00003711SmallVector<llvm::OperandBundleDef, 1>
3712CodeGenFunction::getBundlesForFunclet(llvm::Value *Callee) {
3713 SmallVector<llvm::OperandBundleDef, 1> BundleList;
Sanjay Patel846b63b2016-01-18 22:15:33 +00003714 // There is no need for a funclet operand bundle if we aren't inside a
3715 // funclet.
David Majnemer0b17d442015-12-15 21:27:59 +00003716 if (!CurrentFuncletPad)
Reid Klecknerb75a3f02018-02-09 00:16:41 +00003717 return BundleList;
David Majnemer0b17d442015-12-15 21:27:59 +00003718
3719 // Skip intrinsics which cannot throw.
3720 auto *CalleeFn = dyn_cast<llvm::Function>(Callee->stripPointerCasts());
3721 if (CalleeFn && CalleeFn->isIntrinsic() && CalleeFn->doesNotThrow())
Reid Klecknerb75a3f02018-02-09 00:16:41 +00003722 return BundleList;
David Majnemer0b17d442015-12-15 21:27:59 +00003723
3724 BundleList.emplace_back("funclet", CurrentFuncletPad);
Reid Klecknerb75a3f02018-02-09 00:16:41 +00003725 return BundleList;
David Majnemer0b17d442015-12-15 21:27:59 +00003726}
3727
David Majnemer971d31b2016-02-24 17:02:45 +00003728/// Emits a simple call (never an invoke) to the given runtime function.
3729llvm::CallInst *
3730CodeGenFunction::EmitRuntimeCall(llvm::Value *callee,
3731 ArrayRef<llvm::Value*> args,
3732 const llvm::Twine &name) {
Reid Klecknerb75a3f02018-02-09 00:16:41 +00003733 llvm::CallInst *call =
3734 Builder.CreateCall(callee, args, getBundlesForFunclet(callee), name);
David Majnemer971d31b2016-02-24 17:02:45 +00003735 call->setCallingConv(getRuntimeCC());
3736 return call;
3737}
3738
John McCall882987f2013-02-28 19:01:20 +00003739/// Emits a call or invoke to the given noreturn runtime function.
3740void CodeGenFunction::EmitNoreturnRuntimeCallOrInvoke(llvm::Value *callee,
3741 ArrayRef<llvm::Value*> args) {
Reid Klecknerb75a3f02018-02-09 00:16:41 +00003742 SmallVector<llvm::OperandBundleDef, 1> BundleList =
3743 getBundlesForFunclet(callee);
David Majnemer0b17d442015-12-15 21:27:59 +00003744
John McCall882987f2013-02-28 19:01:20 +00003745 if (getInvokeDest()) {
Fangrui Song6907ce22018-07-30 19:24:48 +00003746 llvm::InvokeInst *invoke =
John McCall882987f2013-02-28 19:01:20 +00003747 Builder.CreateInvoke(callee,
3748 getUnreachableBlock(),
3749 getInvokeDest(),
David Majnemer0b17d442015-12-15 21:27:59 +00003750 args,
3751 BundleList);
John McCall882987f2013-02-28 19:01:20 +00003752 invoke->setDoesNotReturn();
3753 invoke->setCallingConv(getRuntimeCC());
3754 } else {
David Majnemer0b17d442015-12-15 21:27:59 +00003755 llvm::CallInst *call = Builder.CreateCall(callee, args, BundleList);
John McCall882987f2013-02-28 19:01:20 +00003756 call->setDoesNotReturn();
3757 call->setCallingConv(getRuntimeCC());
3758 Builder.CreateUnreachable();
3759 }
3760}
3761
Sanjay Patel846b63b2016-01-18 22:15:33 +00003762/// Emits a call or invoke instruction to the given nullary runtime function.
James Y Knight3933add2019-01-30 02:54:28 +00003763llvm::CallBase *CodeGenFunction::EmitRuntimeCallOrInvoke(llvm::Value *callee,
3764 const Twine &name) {
Craig Topper5fc8fc22014-08-27 06:28:36 +00003765 return EmitRuntimeCallOrInvoke(callee, None, name);
John McCall882987f2013-02-28 19:01:20 +00003766}
3767
3768/// Emits a call or invoke instruction to the given runtime function.
James Y Knight3933add2019-01-30 02:54:28 +00003769llvm::CallBase *CodeGenFunction::EmitRuntimeCallOrInvoke(
3770 llvm::Value *callee, ArrayRef<llvm::Value *> args, const Twine &name) {
3771 llvm::CallBase *call = EmitCallOrInvoke(callee, args, name);
3772 call->setCallingConv(getRuntimeCC());
3773 return call;
John McCall882987f2013-02-28 19:01:20 +00003774}
3775
John McCallbd309292010-07-06 01:34:17 +00003776/// Emits a call or invoke instruction to the given function, depending
3777/// on the current state of the EH stack.
James Y Knight3933add2019-01-30 02:54:28 +00003778llvm::CallBase *CodeGenFunction::EmitCallOrInvoke(llvm::Value *Callee,
3779 ArrayRef<llvm::Value *> Args,
3780 const Twine &Name) {
John McCallbd309292010-07-06 01:34:17 +00003781 llvm::BasicBlock *InvokeDest = getInvokeDest();
Reid Klecknerb75a3f02018-02-09 00:16:41 +00003782 SmallVector<llvm::OperandBundleDef, 1> BundleList =
3783 getBundlesForFunclet(Callee);
John McCallbd309292010-07-06 01:34:17 +00003784
James Y Knight3933add2019-01-30 02:54:28 +00003785 llvm::CallBase *Inst;
Dan Gohman515a60d2012-02-16 00:57:37 +00003786 if (!InvokeDest)
David Majnemer3df77bc2016-01-26 23:14:47 +00003787 Inst = Builder.CreateCall(Callee, Args, BundleList, Name);
Dan Gohman515a60d2012-02-16 00:57:37 +00003788 else {
3789 llvm::BasicBlock *ContBB = createBasicBlock("invoke.cont");
David Majnemer3df77bc2016-01-26 23:14:47 +00003790 Inst = Builder.CreateInvoke(Callee, ContBB, InvokeDest, Args, BundleList,
3791 Name);
Dan Gohman515a60d2012-02-16 00:57:37 +00003792 EmitBlock(ContBB);
3793 }
3794
3795 // In ObjC ARC mode with no ObjC ARC exception safety, tell the ARC
3796 // optimizer it can aggressively ignore unwind edges.
David Blaikiebbafb8a2012-03-11 07:00:24 +00003797 if (CGM.getLangOpts().ObjCAutoRefCount)
Dan Gohman515a60d2012-02-16 00:57:37 +00003798 AddObjCARCExceptionMetadata(Inst);
3799
James Y Knight3933add2019-01-30 02:54:28 +00003800 return Inst;
John McCallbd309292010-07-06 01:34:17 +00003801}
3802
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003803void CodeGenFunction::deferPlaceholderReplacement(llvm::Instruction *Old,
3804 llvm::Value *New) {
3805 DeferredReplacements.push_back(std::make_pair(Old, New));
3806}
Chris Lattnerd59d8672011-07-12 06:29:11 +00003807
Daniel Dunbard931a872009-02-02 22:03:45 +00003808RValue CodeGenFunction::EmitCall(const CGFunctionInfo &CallInfo,
John McCallb92ab1a2016-10-26 23:46:34 +00003809 const CGCallee &Callee,
Anders Carlsson61a401c2009-12-24 19:25:24 +00003810 ReturnValueSlot ReturnValue,
Daniel Dunbarcdbb5e32009-02-20 18:06:48 +00003811 const CallArgList &CallArgs,
James Y Knight3933add2019-01-30 02:54:28 +00003812 llvm::CallBase **callOrInvoke,
Vedant Kumar09b5bfd2017-12-21 00:10:25 +00003813 SourceLocation Loc) {
Mike Stump18bb9282009-05-16 07:57:57 +00003814 // FIXME: We no longer need the types from CallArgs; lift up and simplify.
Daniel Dunbar613855c2008-09-09 23:27:19 +00003815
Peter Collingbourneea211002018-02-05 23:09:13 +00003816 assert(Callee.isOrdinary() || Callee.isVirtual());
John McCallb92ab1a2016-10-26 23:46:34 +00003817
Daniel Dunbar613855c2008-09-09 23:27:19 +00003818 // Handle struct-return functions by passing a pointer to the
3819 // location that we would like to return into.
Daniel Dunbar7633cbf2009-02-02 21:43:58 +00003820 QualType RetTy = CallInfo.getReturnType();
Daniel Dunbarb52d0772009-02-03 05:59:18 +00003821 const ABIArgInfo &RetAI = CallInfo.getReturnInfo();
Mike Stump11289f42009-09-09 15:08:12 +00003822
John McCallb92ab1a2016-10-26 23:46:34 +00003823 llvm::FunctionType *IRFuncTy = Callee.getFunctionType();
3824
3825 // 1. Set up the arguments.
Mike Stump11289f42009-09-09 15:08:12 +00003826
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003827 // If we're using inalloca, insert the allocation after the stack save.
3828 // FIXME: Do this earlier rather than hacking it in here!
John McCall7f416cc2015-09-08 08:05:57 +00003829 Address ArgMemory = Address::invalid();
3830 const llvm::StructLayout *ArgMemoryLayout = nullptr;
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003831 if (llvm::StructType *ArgStruct = CallInfo.getArgStruct()) {
Matt Arsenault502ad602017-04-10 22:28:02 +00003832 const llvm::DataLayout &DL = CGM.getDataLayout();
3833 ArgMemoryLayout = DL.getStructLayout(ArgStruct);
Reid Kleckner9df1d972014-04-10 01:40:15 +00003834 llvm::Instruction *IP = CallArgs.getStackBase();
3835 llvm::AllocaInst *AI;
3836 if (IP) {
3837 IP = IP->getNextNode();
Matt Arsenault502ad602017-04-10 22:28:02 +00003838 AI = new llvm::AllocaInst(ArgStruct, DL.getAllocaAddrSpace(),
3839 "argmem", IP);
Reid Kleckner9df1d972014-04-10 01:40:15 +00003840 } else {
Reid Kleckner966abe72014-05-15 23:01:46 +00003841 AI = CreateTempAlloca(ArgStruct, "argmem");
Reid Kleckner9df1d972014-04-10 01:40:15 +00003842 }
John McCall7f416cc2015-09-08 08:05:57 +00003843 auto Align = CallInfo.getArgStructAlignment();
3844 AI->setAlignment(Align.getQuantity());
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003845 AI->setUsedWithInAlloca(true);
3846 assert(AI->isUsedWithInAlloca() && !AI->isStaticAlloca());
John McCall7f416cc2015-09-08 08:05:57 +00003847 ArgMemory = Address(AI, Align);
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003848 }
3849
John McCall7f416cc2015-09-08 08:05:57 +00003850 // Helper function to drill into the inalloca allocation.
3851 auto createInAllocaStructGEP = [&](unsigned FieldIndex) -> Address {
3852 auto FieldOffset =
3853 CharUnits::fromQuantity(ArgMemoryLayout->getElementOffset(FieldIndex));
3854 return Builder.CreateStructGEP(ArgMemory, FieldIndex, FieldOffset);
3855 };
3856
Alexey Samsonov153004f2014-09-29 22:08:00 +00003857 ClangToLLVMArgMapping IRFunctionArgs(CGM.getContext(), CallInfo);
Alexey Samsonov91cf4552014-08-22 01:06:06 +00003858 SmallVector<llvm::Value *, 16> IRCallArgs(IRFunctionArgs.totalIRArgs());
3859
Chris Lattner4ca97c32009-06-13 00:26:38 +00003860 // If the call returns a temporary with struct return, create a temporary
Anders Carlsson17490832009-12-24 20:40:36 +00003861 // alloca to hold the result, unless one is given to us.
John McCall7f416cc2015-09-08 08:05:57 +00003862 Address SRetPtr = Address::invalid();
Yaxun Liua2a9cfa2018-05-17 11:16:35 +00003863 Address SRetAlloca = Address::invalid();
George Burgess IV003be7c2018-03-08 05:32:30 +00003864 llvm::Value *UnusedReturnSizePtr = nullptr;
John McCallf26e73d2016-03-11 04:30:43 +00003865 if (RetAI.isIndirect() || RetAI.isInAlloca() || RetAI.isCoerceAndExpand()) {
John McCall7f416cc2015-09-08 08:05:57 +00003866 if (!ReturnValue.isNull()) {
3867 SRetPtr = ReturnValue.getValue();
3868 } else {
Yaxun Liua2a9cfa2018-05-17 11:16:35 +00003869 SRetPtr = CreateMemTemp(RetTy, "tmp", &SRetAlloca);
Leny Kholodov6aab1112015-06-08 10:23:49 +00003870 if (HaveInsertPoint() && ReturnValue.isUnused()) {
3871 uint64_t size =
3872 CGM.getDataLayout().getTypeAllocSize(ConvertTypeForMem(RetTy));
Yaxun Liua2a9cfa2018-05-17 11:16:35 +00003873 UnusedReturnSizePtr = EmitLifetimeStart(size, SRetAlloca.getPointer());
Leny Kholodov6aab1112015-06-08 10:23:49 +00003874 }
3875 }
Alexey Samsonov91cf4552014-08-22 01:06:06 +00003876 if (IRFunctionArgs.hasSRetArg()) {
John McCall7f416cc2015-09-08 08:05:57 +00003877 IRCallArgs[IRFunctionArgs.getSRetArgNo()] = SRetPtr.getPointer();
John McCallf26e73d2016-03-11 04:30:43 +00003878 } else if (RetAI.isInAlloca()) {
John McCall7f416cc2015-09-08 08:05:57 +00003879 Address Addr = createInAllocaStructGEP(RetAI.getInAllocaFieldIndex());
3880 Builder.CreateStore(SRetPtr.getPointer(), Addr);
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003881 }
Anders Carlsson17490832009-12-24 20:40:36 +00003882 }
Mike Stump11289f42009-09-09 15:08:12 +00003883
John McCall12f23522016-04-04 18:33:08 +00003884 Address swiftErrorTemp = Address::invalid();
3885 Address swiftErrorArg = Address::invalid();
3886
John McCallb92ab1a2016-10-26 23:46:34 +00003887 // Translate all of the arguments as necessary to match the IR lowering.
Daniel Dunbara45bdbb2009-02-04 21:17:21 +00003888 assert(CallInfo.arg_size() == CallArgs.size() &&
3889 "Mismatch between function signature & arguments.");
Alexey Samsonov91cf4552014-08-22 01:06:06 +00003890 unsigned ArgNo = 0;
Daniel Dunbarb52d0772009-02-03 05:59:18 +00003891 CGFunctionInfo::const_arg_iterator info_it = CallInfo.arg_begin();
Mike Stump11289f42009-09-09 15:08:12 +00003892 for (CallArgList::const_iterator I = CallArgs.begin(), E = CallArgs.end();
Alexey Samsonov91cf4552014-08-22 01:06:06 +00003893 I != E; ++I, ++info_it, ++ArgNo) {
Daniel Dunbarb52d0772009-02-03 05:59:18 +00003894 const ABIArgInfo &ArgInfo = info_it->info;
Daniel Dunbar8fc81b02008-09-17 00:51:38 +00003895
Rafael Espindolafad28de2012-10-24 01:59:00 +00003896 // Insert a padding argument to ensure proper alignment.
Alexey Samsonov91cf4552014-08-22 01:06:06 +00003897 if (IRFunctionArgs.hasPaddingArg(ArgNo))
3898 IRCallArgs[IRFunctionArgs.getPaddingArgNo(ArgNo)] =
3899 llvm::UndefValue::get(ArgInfo.getPaddingType());
3900
3901 unsigned FirstIRArg, NumIRArgs;
3902 std::tie(FirstIRArg, NumIRArgs) = IRFunctionArgs.getIRArgs(ArgNo);
Rafael Espindolafad28de2012-10-24 01:59:00 +00003903
Daniel Dunbar8fc81b02008-09-17 00:51:38 +00003904 switch (ArgInfo.getKind()) {
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003905 case ABIArgInfo::InAlloca: {
Alexey Samsonov91cf4552014-08-22 01:06:06 +00003906 assert(NumIRArgs == 0);
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003907 assert(getTarget().getTriple().getArch() == llvm::Triple::x86);
Yaxun Liu5b330e82018-03-15 15:25:19 +00003908 if (I->isAggregate()) {
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003909 // Replace the placeholder with the appropriate argument slot GEP.
Yaxun Liu5b330e82018-03-15 15:25:19 +00003910 Address Addr = I->hasLValue()
3911 ? I->getKnownLValue().getAddress()
3912 : I->getKnownRValue().getAggregateAddress();
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003913 llvm::Instruction *Placeholder =
Yaxun Liu5b330e82018-03-15 15:25:19 +00003914 cast<llvm::Instruction>(Addr.getPointer());
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003915 CGBuilderTy::InsertPoint IP = Builder.saveIP();
3916 Builder.SetInsertPoint(Placeholder);
Yaxun Liu5b330e82018-03-15 15:25:19 +00003917 Addr = createInAllocaStructGEP(ArgInfo.getInAllocaFieldIndex());
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003918 Builder.restoreIP(IP);
John McCall7f416cc2015-09-08 08:05:57 +00003919 deferPlaceholderReplacement(Placeholder, Addr.getPointer());
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003920 } else {
3921 // Store the RValue into the argument struct.
John McCall7f416cc2015-09-08 08:05:57 +00003922 Address Addr = createInAllocaStructGEP(ArgInfo.getInAllocaFieldIndex());
3923 unsigned AS = Addr.getType()->getPointerAddressSpace();
David Majnemer32b57b02014-03-31 16:12:47 +00003924 llvm::Type *MemType = ConvertTypeForMem(I->Ty)->getPointerTo(AS);
3925 // There are some cases where a trivial bitcast is not avoidable. The
3926 // definition of a type later in a translation unit may change it's type
3927 // from {}* to (%struct.foo*)*.
John McCall7f416cc2015-09-08 08:05:57 +00003928 if (Addr.getType() != MemType)
David Majnemer32b57b02014-03-31 16:12:47 +00003929 Addr = Builder.CreateBitCast(Addr, MemType);
Yaxun Liu5b330e82018-03-15 15:25:19 +00003930 I->copyInto(*this, Addr);
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003931 }
Alexey Samsonov91cf4552014-08-22 01:06:06 +00003932 break;
Reid Kleckner314ef7b2014-02-01 00:04:45 +00003933 }
3934
Daniel Dunbar03816342010-08-21 02:24:36 +00003935 case ABIArgInfo::Indirect: {
Alexey Samsonov91cf4552014-08-22 01:06:06 +00003936 assert(NumIRArgs == 1);
Yaxun Liu5b330e82018-03-15 15:25:19 +00003937 if (!I->isAggregate()) {
Daniel Dunbar747865a2009-02-05 09:16:39 +00003938 // Make a temporary alloca to pass the argument.
Yaxun Liuaefdb8e2018-06-15 15:33:22 +00003939 Address Addr = CreateMemTempWithoutCast(
3940 I->Ty, ArgInfo.getIndirectAlign(), "indirect-arg-temp");
John McCall7f416cc2015-09-08 08:05:57 +00003941 IRCallArgs[FirstIRArg] = Addr.getPointer();
John McCall47fb9502013-03-07 21:37:08 +00003942
Yaxun Liu5b330e82018-03-15 15:25:19 +00003943 I->copyInto(*this, Addr);
Daniel Dunbar747865a2009-02-05 09:16:39 +00003944 } else {
Eli Friedmaneb7fab62011-06-14 01:37:52 +00003945 // We want to avoid creating an unnecessary temporary+copy here;
Guy Benyei3832bfd2013-03-10 12:59:00 +00003946 // however, we need one in three cases:
Eli Friedmaneb7fab62011-06-14 01:37:52 +00003947 // 1. If the argument is not byval, and we are required to copy the
3948 // source. (This case doesn't occur on any common architecture.)
3949 // 2. If the argument is byval, RV is not sufficiently aligned, and
3950 // we cannot force it to be sufficiently aligned.
Yaxun Liu5b330e82018-03-15 15:25:19 +00003951 // 3. If the argument is byval, but RV is not located in default
3952 // or alloca address space.
3953 Address Addr = I->hasLValue()
3954 ? I->getKnownLValue().getAddress()
3955 : I->getKnownRValue().getAggregateAddress();
3956 llvm::Value *V = Addr.getPointer();
John McCall7f416cc2015-09-08 08:05:57 +00003957 CharUnits Align = ArgInfo.getIndirectAlign();
Micah Villmowdd31ca12012-10-08 16:25:52 +00003958 const llvm::DataLayout *TD = &CGM.getDataLayout();
Yaxun Liu5b330e82018-03-15 15:25:19 +00003959
3960 assert((FirstIRArg >= IRFuncTy->getNumParams() ||
3961 IRFuncTy->getParamType(FirstIRArg)->getPointerAddressSpace() ==
3962 TD->getAllocaAddrSpace()) &&
3963 "indirect argument must be in alloca address space");
3964
3965 bool NeedCopy = false;
3966
3967 if (Addr.getAlignment() < Align &&
3968 llvm::getOrEnforceKnownAlignment(V, Align.getQuantity(), *TD) <
3969 Align.getQuantity()) {
3970 NeedCopy = true;
3971 } else if (I->hasLValue()) {
3972 auto LV = I->getKnownLValue();
3973 auto AS = LV.getAddressSpace();
Andrew Savonichev1bf1a152018-12-10 12:03:00 +00003974
Yaxun Liu5b330e82018-03-15 15:25:19 +00003975 if ((!ArgInfo.getIndirectByVal() &&
3976 (LV.getAlignment() >=
Andrew Savonichev1bf1a152018-12-10 12:03:00 +00003977 getContext().getTypeAlignInChars(I->Ty)))) {
3978 NeedCopy = true;
3979 }
3980 if (!getLangOpts().OpenCL) {
3981 if ((ArgInfo.getIndirectByVal() &&
3982 (AS != LangAS::Default &&
3983 AS != CGM.getASTAllocaAddressSpace()))) {
3984 NeedCopy = true;
3985 }
3986 }
3987 // For OpenCL even if RV is located in default or alloca address space
3988 // we don't want to perform address space cast for it.
3989 else if ((ArgInfo.getIndirectByVal() &&
3990 Addr.getType()->getAddressSpace() != IRFuncTy->
3991 getParamType(FirstIRArg)->getPointerAddressSpace())) {
Yaxun Liu5b330e82018-03-15 15:25:19 +00003992 NeedCopy = true;
3993 }
3994 }
Andrew Savonichev1bf1a152018-12-10 12:03:00 +00003995
Yaxun Liu5b330e82018-03-15 15:25:19 +00003996 if (NeedCopy) {
Eli Friedmaneb7fab62011-06-14 01:37:52 +00003997 // Create an aligned temporary, and copy to it.
Yaxun Liuaefdb8e2018-06-15 15:33:22 +00003998 Address AI = CreateMemTempWithoutCast(
3999 I->Ty, ArgInfo.getIndirectAlign(), "byval-temp");
John McCall7f416cc2015-09-08 08:05:57 +00004000 IRCallArgs[FirstIRArg] = AI.getPointer();
Yaxun Liu5b330e82018-03-15 15:25:19 +00004001 I->copyInto(*this, AI);
Eli Friedmaneb7fab62011-06-14 01:37:52 +00004002 } else {
4003 // Skip the extra memcpy call.
Yaxun Liu5b330e82018-03-15 15:25:19 +00004004 auto *T = V->getType()->getPointerElementType()->getPointerTo(
4005 CGM.getDataLayout().getAllocaAddrSpace());
4006 IRCallArgs[FirstIRArg] = getTargetHooks().performAddrSpaceCast(
4007 *this, V, LangAS::Default, CGM.getASTAllocaAddressSpace(), T,
4008 true);
Eli Friedmaneb7fab62011-06-14 01:37:52 +00004009 }
Daniel Dunbar747865a2009-02-05 09:16:39 +00004010 }
4011 break;
Daniel Dunbar03816342010-08-21 02:24:36 +00004012 }
Daniel Dunbar747865a2009-02-05 09:16:39 +00004013
Daniel Dunbar94a6f252009-01-26 21:26:08 +00004014 case ABIArgInfo::Ignore:
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004015 assert(NumIRArgs == 0);
Daniel Dunbar94a6f252009-01-26 21:26:08 +00004016 break;
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00004017
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00004018 case ABIArgInfo::Extend:
4019 case ABIArgInfo::Direct: {
4020 if (!isa<llvm::StructType>(ArgInfo.getCoerceToType()) &&
Chris Lattner8a2f3c72010-07-30 04:02:24 +00004021 ArgInfo.getCoerceToType() == ConvertType(info_it->type) &&
4022 ArgInfo.getDirectOffset() == 0) {
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004023 assert(NumIRArgs == 1);
Chris Lattnerbb1952c2011-07-12 04:46:18 +00004024 llvm::Value *V;
Yaxun Liu5b330e82018-03-15 15:25:19 +00004025 if (!I->isAggregate())
4026 V = I->getKnownRValue().getScalarVal();
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00004027 else
Yaxun Liu5b330e82018-03-15 15:25:19 +00004028 V = Builder.CreateLoad(
4029 I->hasLValue() ? I->getKnownLValue().getAddress()
4030 : I->getKnownRValue().getAggregateAddress());
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004031
John McCall12f23522016-04-04 18:33:08 +00004032 // Implement swifterror by copying into a new swifterror argument.
4033 // We'll write back in the normal path out of the call.
4034 if (CallInfo.getExtParameterInfo(ArgNo).getABI()
4035 == ParameterABI::SwiftErrorResult) {
4036 assert(!swiftErrorTemp.isValid() && "multiple swifterror args");
4037
4038 QualType pointeeTy = I->Ty->getPointeeType();
4039 swiftErrorArg =
4040 Address(V, getContext().getTypeAlignInChars(pointeeTy));
4041
4042 swiftErrorTemp =
4043 CreateMemTemp(pointeeTy, getPointerAlign(), "swifterror.temp");
4044 V = swiftErrorTemp.getPointer();
4045 cast<llvm::AllocaInst>(V)->setSwiftError(true);
4046
4047 llvm::Value *errorValue = Builder.CreateLoad(swiftErrorArg);
4048 Builder.CreateStore(errorValue, swiftErrorTemp);
4049 }
4050
Reid Kleckner79b0fd72014-10-10 00:05:45 +00004051 // We might have to widen integers, but we should never truncate.
4052 if (ArgInfo.getCoerceToType() != V->getType() &&
4053 V->getType()->isIntegerTy())
4054 V = Builder.CreateZExt(V, ArgInfo.getCoerceToType());
4055
Chris Lattner3ce86682011-07-12 04:53:39 +00004056 // If the argument doesn't match, perform a bitcast to coerce it. This
4057 // can happen due to trivial type mismatches.
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004058 if (FirstIRArg < IRFuncTy->getNumParams() &&
4059 V->getType() != IRFuncTy->getParamType(FirstIRArg))
4060 V = Builder.CreateBitCast(V, IRFuncTy->getParamType(FirstIRArg));
John McCall12f23522016-04-04 18:33:08 +00004061
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004062 IRCallArgs[FirstIRArg] = V;
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00004063 break;
4064 }
Daniel Dunbar94a6f252009-01-26 21:26:08 +00004065
Daniel Dunbar2f219b02009-02-03 19:12:28 +00004066 // FIXME: Avoid the conversion through memory if possible.
John McCall7f416cc2015-09-08 08:05:57 +00004067 Address Src = Address::invalid();
Yaxun Liu5b330e82018-03-15 15:25:19 +00004068 if (!I->isAggregate()) {
John McCall7f416cc2015-09-08 08:05:57 +00004069 Src = CreateMemTemp(I->Ty, "coerce");
Yaxun Liu5b330e82018-03-15 15:25:19 +00004070 I->copyInto(*this, Src);
Ulrich Weigand6e2cea62015-07-10 11:31:43 +00004071 } else {
Yaxun Liu5b330e82018-03-15 15:25:19 +00004072 Src = I->hasLValue() ? I->getKnownLValue().getAddress()
4073 : I->getKnownRValue().getAggregateAddress();
Ulrich Weigand6e2cea62015-07-10 11:31:43 +00004074 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00004075
Chris Lattner8a2f3c72010-07-30 04:02:24 +00004076 // If the value is offset in memory, apply the offset now.
John McCall7f416cc2015-09-08 08:05:57 +00004077 Src = emitAddressAtOffset(*this, Src, ArgInfo);
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00004078
Oliver Stannard2bfdc5b2014-08-27 10:43:15 +00004079 // Fast-isel and the optimizer generally like scalar values better than
4080 // FCAs, so we flatten them if this is safe to do for this argument.
James Molloy6f244b62014-05-09 16:21:39 +00004081 llvm::StructType *STy =
4082 dyn_cast<llvm::StructType>(ArgInfo.getCoerceToType());
Oliver Stannard2bfdc5b2014-08-27 10:43:15 +00004083 if (STy && ArgInfo.isDirect() && ArgInfo.getCanBeFlattened()) {
John McCall7f416cc2015-09-08 08:05:57 +00004084 llvm::Type *SrcTy = Src.getType()->getElementType();
Chandler Carrutha6399a52012-10-10 11:29:08 +00004085 uint64_t SrcSize = CGM.getDataLayout().getTypeAllocSize(SrcTy);
4086 uint64_t DstSize = CGM.getDataLayout().getTypeAllocSize(STy);
4087
4088 // If the source type is smaller than the destination type of the
4089 // coerce-to logic, copy the source value into a temp alloca the size
4090 // of the destination type to allow loading all of it. The bits past
4091 // the source value are left undef.
4092 if (SrcSize < DstSize) {
John McCall7f416cc2015-09-08 08:05:57 +00004093 Address TempAlloca
4094 = CreateTempAlloca(STy, Src.getAlignment(),
4095 Src.getName() + ".coerce");
4096 Builder.CreateMemCpy(TempAlloca, Src, SrcSize);
4097 Src = TempAlloca;
Chandler Carrutha6399a52012-10-10 11:29:08 +00004098 } else {
Matt Arsenault7a124f32017-08-01 20:36:57 +00004099 Src = Builder.CreateBitCast(Src,
4100 STy->getPointerTo(Src.getAddressSpace()));
Chandler Carrutha6399a52012-10-10 11:29:08 +00004101 }
4102
John McCall7f416cc2015-09-08 08:05:57 +00004103 auto SrcLayout = CGM.getDataLayout().getStructLayout(STy);
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004104 assert(NumIRArgs == STy->getNumElements());
Chris Lattnerceddafb2010-07-05 20:41:41 +00004105 for (unsigned i = 0, e = STy->getNumElements(); i != e; ++i) {
John McCall7f416cc2015-09-08 08:05:57 +00004106 auto Offset = CharUnits::fromQuantity(SrcLayout->getElementOffset(i));
4107 Address EltPtr = Builder.CreateStructGEP(Src, i, Offset);
4108 llvm::Value *LI = Builder.CreateLoad(EltPtr);
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004109 IRCallArgs[FirstIRArg + i] = LI;
Chris Lattner15ec3612010-06-29 00:06:42 +00004110 }
Chris Lattner3dd716c2010-06-28 23:44:11 +00004111 } else {
Chris Lattner15ec3612010-06-29 00:06:42 +00004112 // In the simple case, just pass the coerced loaded value.
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004113 assert(NumIRArgs == 1);
4114 IRCallArgs[FirstIRArg] =
John McCall7f416cc2015-09-08 08:05:57 +00004115 CreateCoercedLoad(Src, ArgInfo.getCoerceToType(), *this);
Chris Lattner3dd716c2010-06-28 23:44:11 +00004116 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00004117
Daniel Dunbar2f219b02009-02-03 19:12:28 +00004118 break;
4119 }
4120
John McCallf26e73d2016-03-11 04:30:43 +00004121 case ABIArgInfo::CoerceAndExpand: {
John McCallf26e73d2016-03-11 04:30:43 +00004122 auto coercionType = ArgInfo.getCoerceAndExpandType();
4123 auto layout = CGM.getDataLayout().getStructLayout(coercionType);
4124
John McCall12f23522016-04-04 18:33:08 +00004125 llvm::Value *tempSize = nullptr;
4126 Address addr = Address::invalid();
Yaxun Liua2a9cfa2018-05-17 11:16:35 +00004127 Address AllocaAddr = Address::invalid();
Yaxun Liu5b330e82018-03-15 15:25:19 +00004128 if (I->isAggregate()) {
4129 addr = I->hasLValue() ? I->getKnownLValue().getAddress()
4130 : I->getKnownRValue().getAggregateAddress();
4131
John McCall12f23522016-04-04 18:33:08 +00004132 } else {
Yaxun Liu5b330e82018-03-15 15:25:19 +00004133 RValue RV = I->getKnownRValue();
John McCall12f23522016-04-04 18:33:08 +00004134 assert(RV.isScalar()); // complex should always just be direct
4135
4136 llvm::Type *scalarType = RV.getScalarVal()->getType();
4137 auto scalarSize = CGM.getDataLayout().getTypeAllocSize(scalarType);
4138 auto scalarAlign = CGM.getDataLayout().getPrefTypeAlignment(scalarType);
4139
John McCall12f23522016-04-04 18:33:08 +00004140 // Materialize to a temporary.
4141 addr = CreateTempAlloca(RV.getScalarVal()->getType(),
Yaxun Liua2a9cfa2018-05-17 11:16:35 +00004142 CharUnits::fromQuantity(std::max(
4143 layout->getAlignment(), scalarAlign)),
4144 "tmp",
4145 /*ArraySize=*/nullptr, &AllocaAddr);
4146 tempSize = EmitLifetimeStart(scalarSize, AllocaAddr.getPointer());
John McCall12f23522016-04-04 18:33:08 +00004147
4148 Builder.CreateStore(RV.getScalarVal(), addr);
4149 }
4150
John McCallf26e73d2016-03-11 04:30:43 +00004151 addr = Builder.CreateElementBitCast(addr, coercionType);
4152
4153 unsigned IRArgPos = FirstIRArg;
4154 for (unsigned i = 0, e = coercionType->getNumElements(); i != e; ++i) {
4155 llvm::Type *eltType = coercionType->getElementType(i);
4156 if (ABIArgInfo::isPaddingForCoerceAndExpand(eltType)) continue;
4157 Address eltAddr = Builder.CreateStructGEP(addr, i, layout);
4158 llvm::Value *elt = Builder.CreateLoad(eltAddr);
4159 IRCallArgs[IRArgPos++] = elt;
4160 }
4161 assert(IRArgPos == FirstIRArg + NumIRArgs);
4162
John McCall12f23522016-04-04 18:33:08 +00004163 if (tempSize) {
Yaxun Liua2a9cfa2018-05-17 11:16:35 +00004164 EmitLifetimeEnd(tempSize, AllocaAddr.getPointer());
John McCall12f23522016-04-04 18:33:08 +00004165 }
4166
John McCallf26e73d2016-03-11 04:30:43 +00004167 break;
4168 }
4169
Daniel Dunbar8fc81b02008-09-17 00:51:38 +00004170 case ABIArgInfo::Expand:
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004171 unsigned IRArgPos = FirstIRArg;
Yaxun Liu5b330e82018-03-15 15:25:19 +00004172 ExpandTypeToArgs(I->Ty, *I, IRFuncTy, IRCallArgs, IRArgPos);
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004173 assert(IRArgPos == FirstIRArg + NumIRArgs);
Daniel Dunbar8fc81b02008-09-17 00:51:38 +00004174 break;
Daniel Dunbar613855c2008-09-09 23:27:19 +00004175 }
4176 }
Mike Stump11289f42009-09-09 15:08:12 +00004177
John McCall9831b842018-02-06 18:52:44 +00004178 const CGCallee &ConcreteCallee = Callee.prepareConcreteCallee(*this);
4179 llvm::Value *CalleePtr = ConcreteCallee.getFunctionPointer();
John McCallb92ab1a2016-10-26 23:46:34 +00004180
4181 // If we're using inalloca, set up that argument.
John McCall7f416cc2015-09-08 08:05:57 +00004182 if (ArgMemory.isValid()) {
4183 llvm::Value *Arg = ArgMemory.getPointer();
Reid Klecknerafba553e2014-07-08 02:24:27 +00004184 if (CallInfo.isVariadic()) {
4185 // When passing non-POD arguments by value to variadic functions, we will
4186 // end up with a variadic prototype and an inalloca call site. In such
4187 // cases, we can't do any parameter mismatch checks. Give up and bitcast
4188 // the callee.
John McCallb92ab1a2016-10-26 23:46:34 +00004189 unsigned CalleeAS = CalleePtr->getType()->getPointerAddressSpace();
4190 auto FnTy = getTypes().GetFunctionType(CallInfo)->getPointerTo(CalleeAS);
4191 CalleePtr = Builder.CreateBitCast(CalleePtr, FnTy);
Reid Klecknerafba553e2014-07-08 02:24:27 +00004192 } else {
4193 llvm::Type *LastParamTy =
4194 IRFuncTy->getParamType(IRFuncTy->getNumParams() - 1);
4195 if (Arg->getType() != LastParamTy) {
Reid Kleckner314ef7b2014-02-01 00:04:45 +00004196#ifndef NDEBUG
Reid Klecknerafba553e2014-07-08 02:24:27 +00004197 // Assert that these structs have equivalent element types.
4198 llvm::StructType *FullTy = CallInfo.getArgStruct();
4199 llvm::StructType *DeclaredTy = cast<llvm::StructType>(
4200 cast<llvm::PointerType>(LastParamTy)->getElementType());
4201 assert(DeclaredTy->getNumElements() == FullTy->getNumElements());
4202 for (llvm::StructType::element_iterator DI = DeclaredTy->element_begin(),
4203 DE = DeclaredTy->element_end(),
4204 FI = FullTy->element_begin();
4205 DI != DE; ++DI, ++FI)
4206 assert(*DI == *FI);
Reid Kleckner314ef7b2014-02-01 00:04:45 +00004207#endif
Reid Klecknerafba553e2014-07-08 02:24:27 +00004208 Arg = Builder.CreateBitCast(Arg, LastParamTy);
4209 }
Reid Kleckner314ef7b2014-02-01 00:04:45 +00004210 }
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004211 assert(IRFunctionArgs.hasInallocaArg());
4212 IRCallArgs[IRFunctionArgs.getInallocaArgNo()] = Arg;
Reid Kleckner314ef7b2014-02-01 00:04:45 +00004213 }
4214
John McCallb92ab1a2016-10-26 23:46:34 +00004215 // 2. Prepare the function pointer.
4216
4217 // If the callee is a bitcast of a non-variadic function to have a
4218 // variadic function pointer type, check to see if we can remove the
4219 // bitcast. This comes up with unprototyped functions.
4220 //
4221 // This makes the IR nicer, but more importantly it ensures that we
4222 // can inline the function at -O0 if it is marked always_inline.
4223 auto simplifyVariadicCallee = [](llvm::Value *Ptr) -> llvm::Value* {
4224 llvm::FunctionType *CalleeFT =
4225 cast<llvm::FunctionType>(Ptr->getType()->getPointerElementType());
4226 if (!CalleeFT->isVarArg())
4227 return Ptr;
4228
4229 llvm::ConstantExpr *CE = dyn_cast<llvm::ConstantExpr>(Ptr);
4230 if (!CE || CE->getOpcode() != llvm::Instruction::BitCast)
4231 return Ptr;
4232
4233 llvm::Function *OrigFn = dyn_cast<llvm::Function>(CE->getOperand(0));
4234 if (!OrigFn)
4235 return Ptr;
4236
4237 llvm::FunctionType *OrigFT = OrigFn->getFunctionType();
4238
4239 // If the original type is variadic, or if any of the component types
4240 // disagree, we cannot remove the cast.
4241 if (OrigFT->isVarArg() ||
4242 OrigFT->getNumParams() != CalleeFT->getNumParams() ||
4243 OrigFT->getReturnType() != CalleeFT->getReturnType())
4244 return Ptr;
4245
4246 for (unsigned i = 0, e = OrigFT->getNumParams(); i != e; ++i)
4247 if (OrigFT->getParamType(i) != CalleeFT->getParamType(i))
4248 return Ptr;
4249
4250 return OrigFn;
4251 };
4252 CalleePtr = simplifyVariadicCallee(CalleePtr);
4253
4254 // 3. Perform the actual call.
4255
4256 // Deactivate any cleanups that we're supposed to do immediately before
4257 // the call.
Reid Kleckner23f4c4b2013-06-21 12:45:15 +00004258 if (!CallArgs.getCleanupsToDeactivate().empty())
4259 deactivateArgCleanupsBeforeCall(*this, CallArgs);
4260
John McCallb92ab1a2016-10-26 23:46:34 +00004261 // Assert that the arguments we computed match up. The IR verifier
4262 // will catch this, but this is a common enough source of problems
4263 // during IRGen changes that it's way better for debugging to catch
4264 // it ourselves here.
4265#ifndef NDEBUG
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004266 assert(IRCallArgs.size() == IRFuncTy->getNumParams() || IRFuncTy->isVarArg());
4267 for (unsigned i = 0; i < IRCallArgs.size(); ++i) {
4268 // Inalloca argument can have different type.
4269 if (IRFunctionArgs.hasInallocaArg() &&
4270 i == IRFunctionArgs.getInallocaArgNo())
4271 continue;
4272 if (i < IRFuncTy->getNumParams())
4273 assert(IRCallArgs[i]->getType() == IRFuncTy->getParamType(i));
4274 }
John McCallb92ab1a2016-10-26 23:46:34 +00004275#endif
Alexey Samsonov91cf4552014-08-22 01:06:06 +00004276
Craig Topper3113ec32018-10-24 17:42:17 +00004277 // Update the largest vector width if any arguments have vector types.
4278 for (unsigned i = 0; i < IRCallArgs.size(); ++i) {
4279 if (auto *VT = dyn_cast<llvm::VectorType>(IRCallArgs[i]->getType()))
4280 LargestVectorWidth = std::max(LargestVectorWidth,
4281 VT->getPrimitiveSizeInBits());
4282 }
4283
John McCallb92ab1a2016-10-26 23:46:34 +00004284 // Compute the calling convention and attributes.
Daniel Dunbar0ef34792009-09-12 00:59:20 +00004285 unsigned CallingConv;
Reid Klecknercdd26792017-04-18 23:50:03 +00004286 llvm::AttributeList Attrs;
John McCallb92ab1a2016-10-26 23:46:34 +00004287 CGM.ConstructAttributeList(CalleePtr->getName(), CallInfo,
Reid Klecknercdd26792017-04-18 23:50:03 +00004288 Callee.getAbstractInfo(), Attrs, CallingConv,
Chad Rosier7dbc9cf2016-01-06 14:35:46 +00004289 /*AttrOnCallSite=*/true);
Mike Stump11289f42009-09-09 15:08:12 +00004290
John McCallb92ab1a2016-10-26 23:46:34 +00004291 // Apply some call-site-specific attributes.
4292 // TODO: work this into building the attribute set.
4293
4294 // Apply always_inline to all calls within flatten functions.
4295 // FIXME: should this really take priority over __try, below?
4296 if (CurCodeDecl && CurCodeDecl->hasAttr<FlattenAttr>() &&
Erich Keanede6480a32018-11-13 15:48:08 +00004297 !(Callee.getAbstractInfo().getCalleeDecl().getDecl() &&
4298 Callee.getAbstractInfo()
4299 .getCalleeDecl()
4300 .getDecl()
4301 ->hasAttr<NoInlineAttr>())) {
John McCallb92ab1a2016-10-26 23:46:34 +00004302 Attrs =
Reid Klecknerde864822017-03-21 16:57:30 +00004303 Attrs.addAttribute(getLLVMContext(), llvm::AttributeList::FunctionIndex,
John McCallb92ab1a2016-10-26 23:46:34 +00004304 llvm::Attribute::AlwaysInline);
4305 }
4306
4307 // Disable inlining inside SEH __try blocks.
4308 if (isSEHTryScope()) {
4309 Attrs =
Reid Klecknerde864822017-03-21 16:57:30 +00004310 Attrs.addAttribute(getLLVMContext(), llvm::AttributeList::FunctionIndex,
John McCallb92ab1a2016-10-26 23:46:34 +00004311 llvm::Attribute::NoInline);
4312 }
4313
4314 // Decide whether to use a call or an invoke.
David Majnemer4e52d6f2015-12-12 05:39:21 +00004315 bool CannotThrow;
4316 if (currentFunctionUsesSEHTry()) {
John McCallb92ab1a2016-10-26 23:46:34 +00004317 // SEH cares about asynchronous exceptions, so everything can "throw."
David Majnemer4e52d6f2015-12-12 05:39:21 +00004318 CannotThrow = false;
4319 } else if (isCleanupPadScope() &&
4320 EHPersonality::get(*this).isMSVCXXPersonality()) {
4321 // The MSVC++ personality will implicitly terminate the program if an
John McCallb92ab1a2016-10-26 23:46:34 +00004322 // exception is thrown during a cleanup outside of a try/catch.
4323 // We don't need to model anything in IR to get this behavior.
David Majnemer4e52d6f2015-12-12 05:39:21 +00004324 CannotThrow = true;
4325 } else {
John McCallb92ab1a2016-10-26 23:46:34 +00004326 // Otherwise, nounwind call sites will never throw.
Reid Klecknerde864822017-03-21 16:57:30 +00004327 CannotThrow = Attrs.hasAttribute(llvm::AttributeList::FunctionIndex,
David Majnemer4e52d6f2015-12-12 05:39:21 +00004328 llvm::Attribute::NoUnwind);
4329 }
George Burgess IV003be7c2018-03-08 05:32:30 +00004330
4331 // If we made a temporary, be sure to clean up after ourselves. Note that we
4332 // can't depend on being inside of an ExprWithCleanups, so we need to manually
4333 // pop this cleanup later on. Being eager about this is OK, since this
4334 // temporary is 'invisible' outside of the callee.
4335 if (UnusedReturnSizePtr)
Yaxun Liua2a9cfa2018-05-17 11:16:35 +00004336 pushFullExprCleanup<CallLifetimeEnd>(NormalEHLifetimeMarker, SRetAlloca,
George Burgess IV003be7c2018-03-08 05:32:30 +00004337 UnusedReturnSizePtr);
4338
David Majnemer4e52d6f2015-12-12 05:39:21 +00004339 llvm::BasicBlock *InvokeDest = CannotThrow ? nullptr : getInvokeDest();
John McCallbd309292010-07-06 01:34:17 +00004340
Reid Klecknerb75a3f02018-02-09 00:16:41 +00004341 SmallVector<llvm::OperandBundleDef, 1> BundleList =
4342 getBundlesForFunclet(CalleePtr);
David Majnemer0b17d442015-12-15 21:27:59 +00004343
John McCallb92ab1a2016-10-26 23:46:34 +00004344 // Emit the actual call/invoke instruction.
James Y Knight3933add2019-01-30 02:54:28 +00004345 llvm::CallBase *CI;
John McCallbd309292010-07-06 01:34:17 +00004346 if (!InvokeDest) {
James Y Knight3933add2019-01-30 02:54:28 +00004347 CI = Builder.CreateCall(CalleePtr, IRCallArgs, BundleList);
Daniel Dunbar12347492009-02-23 17:26:39 +00004348 } else {
4349 llvm::BasicBlock *Cont = createBasicBlock("invoke.cont");
James Y Knight3933add2019-01-30 02:54:28 +00004350 CI = Builder.CreateInvoke(CalleePtr, Cont, InvokeDest, IRCallArgs,
David Majnemer0b17d442015-12-15 21:27:59 +00004351 BundleList);
Daniel Dunbar12347492009-02-23 17:26:39 +00004352 EmitBlock(Cont);
Daniel Dunbar5006f4a2009-02-20 18:54:31 +00004353 }
Chris Lattnere70a0072010-06-29 16:40:28 +00004354 if (callOrInvoke)
John McCallb92ab1a2016-10-26 23:46:34 +00004355 *callOrInvoke = CI;
Daniel Dunbar5006f4a2009-02-20 18:54:31 +00004356
John McCallb92ab1a2016-10-26 23:46:34 +00004357 // Apply the attributes and calling convention.
James Y Knight3933add2019-01-30 02:54:28 +00004358 CI->setAttributes(Attrs);
4359 CI->setCallingConv(static_cast<llvm::CallingConv::ID>(CallingConv));
Daniel Dunbarb960b7b2009-03-02 04:32:35 +00004360
John McCallb92ab1a2016-10-26 23:46:34 +00004361 // Apply various metadata.
4362
4363 if (!CI->getType()->isVoidTy())
4364 CI->setName("call");
4365
Craig Topper3113ec32018-10-24 17:42:17 +00004366 // Update largest vector width from the return type.
4367 if (auto *VT = dyn_cast<llvm::VectorType>(CI->getType()))
4368 LargestVectorWidth = std::max(LargestVectorWidth,
4369 VT->getPrimitiveSizeInBits());
4370
Adam Nemet1e217bc2016-03-28 22:18:53 +00004371 // Insert instrumentation or attach profile metadata at indirect call sites.
4372 // For more details, see the comment before the definition of
4373 // IPVK_IndirectCallTarget in InstrProfData.inc.
James Y Knight3933add2019-01-30 02:54:28 +00004374 if (!CI->getCalledFunction())
Betul Buyukkurt518276a2016-01-23 22:50:44 +00004375 PGO.valueProfile(Builder, llvm::IPVK_IndirectCallTarget,
John McCallb92ab1a2016-10-26 23:46:34 +00004376 CI, CalleePtr);
Betul Buyukkurt518276a2016-01-23 22:50:44 +00004377
Dan Gohman515a60d2012-02-16 00:57:37 +00004378 // In ObjC ARC mode with no ObjC ARC exception safety, tell the ARC
4379 // optimizer it can aggressively ignore unwind edges.
David Blaikiebbafb8a2012-03-11 07:00:24 +00004380 if (CGM.getLangOpts().ObjCAutoRefCount)
John McCallb92ab1a2016-10-26 23:46:34 +00004381 AddObjCARCExceptionMetadata(CI);
4382
4383 // Suppress tail calls if requested.
4384 if (llvm::CallInst *Call = dyn_cast<llvm::CallInst>(CI)) {
Erich Keanede6480a32018-11-13 15:48:08 +00004385 const Decl *TargetDecl = Callee.getAbstractInfo().getCalleeDecl().getDecl();
John McCallb92ab1a2016-10-26 23:46:34 +00004386 if (TargetDecl && TargetDecl->hasAttr<NotTailCalledAttr>())
4387 Call->setTailCallKind(llvm::CallInst::TCK_NoTail);
4388 }
4389
4390 // 4. Finish the call.
Dan Gohman515a60d2012-02-16 00:57:37 +00004391
Daniel Dunbarb960b7b2009-03-02 04:32:35 +00004392 // If the call doesn't return, finish the basic block and clear the
John McCallb92ab1a2016-10-26 23:46:34 +00004393 // insertion point; this allows the rest of IRGen to discard
Daniel Dunbarb960b7b2009-03-02 04:32:35 +00004394 // unreachable code.
James Y Knight3933add2019-01-30 02:54:28 +00004395 if (CI->doesNotReturn()) {
George Burgess IV003be7c2018-03-08 05:32:30 +00004396 if (UnusedReturnSizePtr)
4397 PopCleanupBlock();
Leny Kholodov6aab1112015-06-08 10:23:49 +00004398
Julian Lettnerb62e9dc2019-01-24 18:04:21 +00004399 // Strip away the noreturn attribute to better diagnose unreachable UB.
Vedant Kumar09b5bfd2017-12-21 00:10:25 +00004400 if (SanOpts.has(SanitizerKind::Unreachable)) {
Julian Lettnerb6c06dc2019-02-01 02:51:00 +00004401 // Also remove from function since CI->hasFnAttr(..) also checks attributes
4402 // of the called function.
James Y Knight3933add2019-01-30 02:54:28 +00004403 if (auto *F = CI->getCalledFunction())
Vedant Kumar09b5bfd2017-12-21 00:10:25 +00004404 F->removeFnAttr(llvm::Attribute::NoReturn);
James Y Knight3933add2019-01-30 02:54:28 +00004405 CI->removeAttribute(llvm::AttributeList::FunctionIndex,
4406 llvm::Attribute::NoReturn);
Julian Lettnerb6c06dc2019-02-01 02:51:00 +00004407
4408 // Avoid incompatibility with ASan which relies on the `noreturn`
4409 // attribute to insert handler calls.
4410 if (SanOpts.has(SanitizerKind::Address)) {
4411 SanitizerScope SanScope(this);
4412 llvm::IRBuilder<>::InsertPointGuard IPGuard(Builder);
4413 Builder.SetInsertPoint(CI);
4414 auto *FnType = llvm::FunctionType::get(CGM.VoidTy, /*isVarArg=*/false);
4415 auto *Fn = CGM.CreateRuntimeFunction(FnType, "__asan_handle_no_return");
4416 EmitNounwindRuntimeCall(Fn);
4417 }
Vedant Kumar09b5bfd2017-12-21 00:10:25 +00004418 }
4419
4420 EmitUnreachable(Loc);
Daniel Dunbarb960b7b2009-03-02 04:32:35 +00004421 Builder.ClearInsertionPoint();
Mike Stump11289f42009-09-09 15:08:12 +00004422
Mike Stump18bb9282009-05-16 07:57:57 +00004423 // FIXME: For now, emit a dummy basic block because expr emitters in
4424 // generally are not ready to handle emitting expressions at unreachable
4425 // points.
Daniel Dunbarb960b7b2009-03-02 04:32:35 +00004426 EnsureInsertPoint();
Mike Stump11289f42009-09-09 15:08:12 +00004427
Daniel Dunbarb960b7b2009-03-02 04:32:35 +00004428 // Return a reasonable RValue.
4429 return GetUndefRValue(RetTy);
Mike Stump11289f42009-09-09 15:08:12 +00004430 }
Daniel Dunbarb960b7b2009-03-02 04:32:35 +00004431
John McCall12f23522016-04-04 18:33:08 +00004432 // Perform the swifterror writeback.
4433 if (swiftErrorTemp.isValid()) {
4434 llvm::Value *errorResult = Builder.CreateLoad(swiftErrorTemp);
4435 Builder.CreateStore(errorResult, swiftErrorArg);
4436 }
4437
John McCallb92ab1a2016-10-26 23:46:34 +00004438 // Emit any call-associated writebacks immediately. Arguably this
4439 // should happen after any return-value munging.
John McCall31168b02011-06-15 23:02:42 +00004440 if (CallArgs.hasWritebacks())
4441 emitWritebacks(*this, CallArgs);
4442
Nico Weber8cdb3f92015-08-25 18:43:32 +00004443 // The stack cleanup for inalloca arguments has to run out of the normal
4444 // lexical order, so deactivate it and run it manually here.
4445 CallArgs.freeArgumentMemory(*this);
4446
John McCallb92ab1a2016-10-26 23:46:34 +00004447 // Extract the return value.
Hal Finkelee90a222014-09-26 05:04:30 +00004448 RValue Ret = [&] {
4449 switch (RetAI.getKind()) {
John McCallf26e73d2016-03-11 04:30:43 +00004450 case ABIArgInfo::CoerceAndExpand: {
4451 auto coercionType = RetAI.getCoerceAndExpandType();
4452 auto layout = CGM.getDataLayout().getStructLayout(coercionType);
4453
4454 Address addr = SRetPtr;
4455 addr = Builder.CreateElementBitCast(addr, coercionType);
4456
John McCall12f23522016-04-04 18:33:08 +00004457 assert(CI->getType() == RetAI.getUnpaddedCoerceAndExpandType());
4458 bool requiresExtract = isa<llvm::StructType>(CI->getType());
4459
John McCallf26e73d2016-03-11 04:30:43 +00004460 unsigned unpaddedIndex = 0;
4461 for (unsigned i = 0, e = coercionType->getNumElements(); i != e; ++i) {
4462 llvm::Type *eltType = coercionType->getElementType(i);
4463 if (ABIArgInfo::isPaddingForCoerceAndExpand(eltType)) continue;
4464 Address eltAddr = Builder.CreateStructGEP(addr, i, layout);
John McCall12f23522016-04-04 18:33:08 +00004465 llvm::Value *elt = CI;
4466 if (requiresExtract)
4467 elt = Builder.CreateExtractValue(elt, unpaddedIndex++);
4468 else
4469 assert(unpaddedIndex == 0);
John McCallf26e73d2016-03-11 04:30:43 +00004470 Builder.CreateStore(elt, eltAddr);
4471 }
John McCall12f23522016-04-04 18:33:08 +00004472 // FALLTHROUGH
Galina Kistanova0872d6c2017-06-03 06:30:46 +00004473 LLVM_FALLTHROUGH;
John McCall12f23522016-04-04 18:33:08 +00004474 }
4475
4476 case ABIArgInfo::InAlloca:
4477 case ABIArgInfo::Indirect: {
4478 RValue ret = convertTempToRValue(SRetPtr, RetTy, SourceLocation());
George Burgess IV003be7c2018-03-08 05:32:30 +00004479 if (UnusedReturnSizePtr)
4480 PopCleanupBlock();
John McCall12f23522016-04-04 18:33:08 +00004481 return ret;
John McCallf26e73d2016-03-11 04:30:43 +00004482 }
4483
Hal Finkelee90a222014-09-26 05:04:30 +00004484 case ABIArgInfo::Ignore:
4485 // If we are ignoring an argument that had a result, make sure to
4486 // construct the appropriate return value for our caller.
4487 return GetUndefRValue(RetTy);
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00004488
Hal Finkelee90a222014-09-26 05:04:30 +00004489 case ABIArgInfo::Extend:
4490 case ABIArgInfo::Direct: {
4491 llvm::Type *RetIRTy = ConvertType(RetTy);
4492 if (RetAI.getCoerceToType() == RetIRTy && RetAI.getDirectOffset() == 0) {
4493 switch (getEvaluationKind(RetTy)) {
4494 case TEK_Complex: {
4495 llvm::Value *Real = Builder.CreateExtractValue(CI, 0);
4496 llvm::Value *Imag = Builder.CreateExtractValue(CI, 1);
4497 return RValue::getComplex(std::make_pair(Real, Imag));
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00004498 }
Hal Finkelee90a222014-09-26 05:04:30 +00004499 case TEK_Aggregate: {
John McCall7f416cc2015-09-08 08:05:57 +00004500 Address DestPtr = ReturnValue.getValue();
Hal Finkelee90a222014-09-26 05:04:30 +00004501 bool DestIsVolatile = ReturnValue.isVolatile();
4502
John McCall7f416cc2015-09-08 08:05:57 +00004503 if (!DestPtr.isValid()) {
Hal Finkelee90a222014-09-26 05:04:30 +00004504 DestPtr = CreateMemTemp(RetTy, "agg.tmp");
4505 DestIsVolatile = false;
4506 }
John McCall7f416cc2015-09-08 08:05:57 +00004507 BuildAggStore(*this, CI, DestPtr, DestIsVolatile);
Hal Finkelee90a222014-09-26 05:04:30 +00004508 return RValue::getAggregate(DestPtr);
4509 }
4510 case TEK_Scalar: {
4511 // If the argument doesn't match, perform a bitcast to coerce it. This
4512 // can happen due to trivial type mismatches.
4513 llvm::Value *V = CI;
4514 if (V->getType() != RetIRTy)
4515 V = Builder.CreateBitCast(V, RetIRTy);
4516 return RValue::get(V);
4517 }
4518 }
4519 llvm_unreachable("bad evaluation kind");
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00004520 }
Hal Finkelee90a222014-09-26 05:04:30 +00004521
John McCall7f416cc2015-09-08 08:05:57 +00004522 Address DestPtr = ReturnValue.getValue();
Hal Finkelee90a222014-09-26 05:04:30 +00004523 bool DestIsVolatile = ReturnValue.isVolatile();
4524
John McCall7f416cc2015-09-08 08:05:57 +00004525 if (!DestPtr.isValid()) {
Hal Finkelee90a222014-09-26 05:04:30 +00004526 DestPtr = CreateMemTemp(RetTy, "coerce");
4527 DestIsVolatile = false;
John McCall47fb9502013-03-07 21:37:08 +00004528 }
Hal Finkelee90a222014-09-26 05:04:30 +00004529
4530 // If the value is offset in memory, apply the offset now.
John McCall7f416cc2015-09-08 08:05:57 +00004531 Address StorePtr = emitAddressAtOffset(*this, DestPtr, RetAI);
4532 CreateCoercedStore(CI, StorePtr, DestIsVolatile, *this);
Hal Finkelee90a222014-09-26 05:04:30 +00004533
4534 return convertTempToRValue(DestPtr, RetTy, SourceLocation());
Chris Lattnerfe34c1d2010-07-29 06:26:06 +00004535 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00004536
Hal Finkelee90a222014-09-26 05:04:30 +00004537 case ABIArgInfo::Expand:
4538 llvm_unreachable("Invalid ABI kind for return argument");
Anders Carlsson17490832009-12-24 20:40:36 +00004539 }
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00004540
Hal Finkelee90a222014-09-26 05:04:30 +00004541 llvm_unreachable("Unhandled ABIArgInfo::Kind");
4542 } ();
Michael J. Spencerf5a1fbc2010-10-19 06:39:39 +00004543
John McCallb92ab1a2016-10-26 23:46:34 +00004544 // Emit the assume_aligned check on the return value.
Erich Keanede6480a32018-11-13 15:48:08 +00004545 const Decl *TargetDecl = Callee.getAbstractInfo().getCalleeDecl().getDecl();
Hal Finkelee90a222014-09-26 05:04:30 +00004546 if (Ret.isScalar() && TargetDecl) {
4547 if (const auto *AA = TargetDecl->getAttr<AssumeAlignedAttr>()) {
4548 llvm::Value *OffsetValue = nullptr;
4549 if (const auto *Offset = AA->getOffset())
4550 OffsetValue = EmitScalarExpr(Offset);
4551
4552 llvm::Value *Alignment = EmitScalarExpr(AA->getAlignment());
4553 llvm::ConstantInt *AlignmentCI = cast<llvm::ConstantInt>(Alignment);
Roman Lebedevbd1c0872019-01-15 09:44:25 +00004554 EmitAlignmentAssumption(Ret.getScalarVal(), RetTy, Loc, AA->getLocation(),
4555 AlignmentCI->getZExtValue(), OffsetValue);
Erich Keane623efd82017-03-30 21:48:55 +00004556 } else if (const auto *AA = TargetDecl->getAttr<AllocAlignAttr>()) {
Roman Lebedevbd1c0872019-01-15 09:44:25 +00004557 llvm::Value *AlignmentVal = CallArgs[AA->getParamIndex().getLLVMIndex()]
4558 .getRValue(*this)
4559 .getScalarVal();
4560 EmitAlignmentAssumption(Ret.getScalarVal(), RetTy, Loc, AA->getLocation(),
4561 AlignmentVal);
Hal Finkelee90a222014-09-26 05:04:30 +00004562 }
Daniel Dunbar573884e2008-09-10 07:04:09 +00004563 }
Daniel Dunbard3674e62008-09-11 01:48:57 +00004564
Hal Finkelee90a222014-09-26 05:04:30 +00004565 return Ret;
Daniel Dunbar613855c2008-09-09 23:27:19 +00004566}
Daniel Dunbar2d0746f2009-02-10 20:44:09 +00004567
John McCall9831b842018-02-06 18:52:44 +00004568CGCallee CGCallee::prepareConcreteCallee(CodeGenFunction &CGF) const {
4569 if (isVirtual()) {
4570 const CallExpr *CE = getVirtualCallExpr();
4571 return CGF.CGM.getCXXABI().getVirtualFunctionPointer(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004572 CGF, getVirtualMethodDecl(), getThisAddress(), getFunctionType(),
4573 CE ? CE->getBeginLoc() : SourceLocation());
John McCall9831b842018-02-06 18:52:44 +00004574 }
4575
4576 return *this;
4577}
4578
Daniel Dunbar2d0746f2009-02-10 20:44:09 +00004579/* VarArg handling */
4580
Charles Davisc7d5c942015-09-17 20:55:33 +00004581Address CodeGenFunction::EmitVAArg(VAArgExpr *VE, Address &VAListAddr) {
4582 VAListAddr = VE->isMicrosoftABI()
4583 ? EmitMSVAListRef(VE->getSubExpr())
4584 : EmitVAListRef(VE->getSubExpr());
4585 QualType Ty = VE->getType();
4586 if (VE->isMicrosoftABI())
4587 return CGM.getTypes().getABIInfo().EmitMSVAArg(*this, VAListAddr, Ty);
John McCall7f416cc2015-09-08 08:05:57 +00004588 return CGM.getTypes().getABIInfo().EmitVAArg(*this, VAListAddr, Ty);
Daniel Dunbar2d0746f2009-02-10 20:44:09 +00004589}