xref: /openbsd-src/gnu/llvm/clang/lib/CodeGen/CGExpr.cpp (revision 12c855180aad702bbcca06e0398d774beeafb155)
1e5dd7070Spatrick //===--- CGExpr.cpp - Emit LLVM Code from Expressions ---------------------===//
2e5dd7070Spatrick //
3e5dd7070Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4e5dd7070Spatrick // See https://llvm.org/LICENSE.txt for license information.
5e5dd7070Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6e5dd7070Spatrick //
7e5dd7070Spatrick //===----------------------------------------------------------------------===//
8e5dd7070Spatrick //
9e5dd7070Spatrick // This contains code to emit Expr nodes as LLVM code.
10e5dd7070Spatrick //
11e5dd7070Spatrick //===----------------------------------------------------------------------===//
12e5dd7070Spatrick 
13a9ac8606Spatrick #include "CGCUDARuntime.h"
14e5dd7070Spatrick #include "CGCXXABI.h"
15e5dd7070Spatrick #include "CGCall.h"
16e5dd7070Spatrick #include "CGCleanup.h"
17e5dd7070Spatrick #include "CGDebugInfo.h"
18e5dd7070Spatrick #include "CGObjCRuntime.h"
19e5dd7070Spatrick #include "CGOpenMPRuntime.h"
20e5dd7070Spatrick #include "CGRecordLayout.h"
21e5dd7070Spatrick #include "CodeGenFunction.h"
22e5dd7070Spatrick #include "CodeGenModule.h"
23e5dd7070Spatrick #include "ConstantEmitter.h"
24e5dd7070Spatrick #include "TargetInfo.h"
25e5dd7070Spatrick #include "clang/AST/ASTContext.h"
26e5dd7070Spatrick #include "clang/AST/Attr.h"
27e5dd7070Spatrick #include "clang/AST/DeclObjC.h"
28e5dd7070Spatrick #include "clang/AST/NSAPI.h"
29e5dd7070Spatrick #include "clang/Basic/Builtins.h"
30e5dd7070Spatrick #include "clang/Basic/CodeGenOptions.h"
31ec727ea7Spatrick #include "clang/Basic/SourceManager.h"
32e5dd7070Spatrick #include "llvm/ADT/Hashing.h"
33e5dd7070Spatrick #include "llvm/ADT/StringExtras.h"
34e5dd7070Spatrick #include "llvm/IR/DataLayout.h"
35e5dd7070Spatrick #include "llvm/IR/Intrinsics.h"
36e5dd7070Spatrick #include "llvm/IR/LLVMContext.h"
37e5dd7070Spatrick #include "llvm/IR/MDBuilder.h"
38*12c85518Srobert #include "llvm/IR/MatrixBuilder.h"
39e5dd7070Spatrick #include "llvm/Support/ConvertUTF.h"
40e5dd7070Spatrick #include "llvm/Support/MathExtras.h"
41e5dd7070Spatrick #include "llvm/Support/Path.h"
42a9ac8606Spatrick #include "llvm/Support/SaveAndRestore.h"
43e5dd7070Spatrick #include "llvm/Transforms/Utils/SanitizerStats.h"
44e5dd7070Spatrick 
45*12c85518Srobert #include <optional>
46e5dd7070Spatrick #include <string>
47e5dd7070Spatrick 
48e5dd7070Spatrick using namespace clang;
49e5dd7070Spatrick using namespace CodeGen;
50e5dd7070Spatrick 
51e5dd7070Spatrick //===--------------------------------------------------------------------===//
52e5dd7070Spatrick //                        Miscellaneous Helper Methods
53e5dd7070Spatrick //===--------------------------------------------------------------------===//
54e5dd7070Spatrick 
EmitCastToVoidPtr(llvm::Value * value)55e5dd7070Spatrick llvm::Value *CodeGenFunction::EmitCastToVoidPtr(llvm::Value *value) {
56e5dd7070Spatrick   unsigned addressSpace =
57e5dd7070Spatrick       cast<llvm::PointerType>(value->getType())->getAddressSpace();
58e5dd7070Spatrick 
59e5dd7070Spatrick   llvm::PointerType *destType = Int8PtrTy;
60e5dd7070Spatrick   if (addressSpace)
61e5dd7070Spatrick     destType = llvm::Type::getInt8PtrTy(getLLVMContext(), addressSpace);
62e5dd7070Spatrick 
63e5dd7070Spatrick   if (value->getType() == destType) return value;
64e5dd7070Spatrick   return Builder.CreateBitCast(value, destType);
65e5dd7070Spatrick }
66e5dd7070Spatrick 
67e5dd7070Spatrick /// CreateTempAlloca - This creates a alloca and inserts it into the entry
68e5dd7070Spatrick /// block.
CreateTempAllocaWithoutCast(llvm::Type * Ty,CharUnits Align,const Twine & Name,llvm::Value * ArraySize)69e5dd7070Spatrick Address CodeGenFunction::CreateTempAllocaWithoutCast(llvm::Type *Ty,
70e5dd7070Spatrick                                                      CharUnits Align,
71e5dd7070Spatrick                                                      const Twine &Name,
72e5dd7070Spatrick                                                      llvm::Value *ArraySize) {
73e5dd7070Spatrick   auto Alloca = CreateTempAlloca(Ty, Name, ArraySize);
74e5dd7070Spatrick   Alloca->setAlignment(Align.getAsAlign());
75*12c85518Srobert   return Address(Alloca, Ty, Align);
76e5dd7070Spatrick }
77e5dd7070Spatrick 
78e5dd7070Spatrick /// CreateTempAlloca - This creates a alloca and inserts it into the entry
79e5dd7070Spatrick /// block. The alloca is casted to default address space if necessary.
CreateTempAlloca(llvm::Type * Ty,CharUnits Align,const Twine & Name,llvm::Value * ArraySize,Address * AllocaAddr)80e5dd7070Spatrick Address CodeGenFunction::CreateTempAlloca(llvm::Type *Ty, CharUnits Align,
81e5dd7070Spatrick                                           const Twine &Name,
82e5dd7070Spatrick                                           llvm::Value *ArraySize,
83e5dd7070Spatrick                                           Address *AllocaAddr) {
84e5dd7070Spatrick   auto Alloca = CreateTempAllocaWithoutCast(Ty, Align, Name, ArraySize);
85e5dd7070Spatrick   if (AllocaAddr)
86e5dd7070Spatrick     *AllocaAddr = Alloca;
87e5dd7070Spatrick   llvm::Value *V = Alloca.getPointer();
88e5dd7070Spatrick   // Alloca always returns a pointer in alloca address space, which may
89e5dd7070Spatrick   // be different from the type defined by the language. For example,
90e5dd7070Spatrick   // in C++ the auto variables are in the default address space. Therefore
91e5dd7070Spatrick   // cast alloca to the default address space when necessary.
92e5dd7070Spatrick   if (getASTAllocaAddressSpace() != LangAS::Default) {
93e5dd7070Spatrick     auto DestAddrSpace = getContext().getTargetAddressSpace(LangAS::Default);
94e5dd7070Spatrick     llvm::IRBuilderBase::InsertPointGuard IPG(Builder);
95e5dd7070Spatrick     // When ArraySize is nullptr, alloca is inserted at AllocaInsertPt,
96e5dd7070Spatrick     // otherwise alloca is inserted at the current insertion point of the
97e5dd7070Spatrick     // builder.
98e5dd7070Spatrick     if (!ArraySize)
99*12c85518Srobert       Builder.SetInsertPoint(getPostAllocaInsertPoint());
100e5dd7070Spatrick     V = getTargetHooks().performAddrSpaceCast(
101e5dd7070Spatrick         *this, V, getASTAllocaAddressSpace(), LangAS::Default,
102e5dd7070Spatrick         Ty->getPointerTo(DestAddrSpace), /*non-null*/ true);
103e5dd7070Spatrick   }
104e5dd7070Spatrick 
105*12c85518Srobert   return Address(V, Ty, Align);
106e5dd7070Spatrick }
107e5dd7070Spatrick 
108e5dd7070Spatrick /// CreateTempAlloca - This creates an alloca and inserts it into the entry
109e5dd7070Spatrick /// block if \p ArraySize is nullptr, otherwise inserts it at the current
110e5dd7070Spatrick /// insertion point of the builder.
CreateTempAlloca(llvm::Type * Ty,const Twine & Name,llvm::Value * ArraySize)111e5dd7070Spatrick llvm::AllocaInst *CodeGenFunction::CreateTempAlloca(llvm::Type *Ty,
112e5dd7070Spatrick                                                     const Twine &Name,
113e5dd7070Spatrick                                                     llvm::Value *ArraySize) {
114e5dd7070Spatrick   if (ArraySize)
115e5dd7070Spatrick     return Builder.CreateAlloca(Ty, ArraySize, Name);
116e5dd7070Spatrick   return new llvm::AllocaInst(Ty, CGM.getDataLayout().getAllocaAddrSpace(),
117e5dd7070Spatrick                               ArraySize, Name, AllocaInsertPt);
118e5dd7070Spatrick }
119e5dd7070Spatrick 
120e5dd7070Spatrick /// CreateDefaultAlignTempAlloca - This creates an alloca with the
121e5dd7070Spatrick /// default alignment of the corresponding LLVM type, which is *not*
122e5dd7070Spatrick /// guaranteed to be related in any way to the expected alignment of
123e5dd7070Spatrick /// an AST type that might have been lowered to Ty.
CreateDefaultAlignTempAlloca(llvm::Type * Ty,const Twine & Name)124e5dd7070Spatrick Address CodeGenFunction::CreateDefaultAlignTempAlloca(llvm::Type *Ty,
125e5dd7070Spatrick                                                       const Twine &Name) {
126e5dd7070Spatrick   CharUnits Align =
127*12c85518Srobert       CharUnits::fromQuantity(CGM.getDataLayout().getPrefTypeAlign(Ty));
128e5dd7070Spatrick   return CreateTempAlloca(Ty, Align, Name);
129e5dd7070Spatrick }
130e5dd7070Spatrick 
CreateIRTemp(QualType Ty,const Twine & Name)131e5dd7070Spatrick Address CodeGenFunction::CreateIRTemp(QualType Ty, const Twine &Name) {
132e5dd7070Spatrick   CharUnits Align = getContext().getTypeAlignInChars(Ty);
133e5dd7070Spatrick   return CreateTempAlloca(ConvertType(Ty), Align, Name);
134e5dd7070Spatrick }
135e5dd7070Spatrick 
CreateMemTemp(QualType Ty,const Twine & Name,Address * Alloca)136e5dd7070Spatrick Address CodeGenFunction::CreateMemTemp(QualType Ty, const Twine &Name,
137e5dd7070Spatrick                                        Address *Alloca) {
138e5dd7070Spatrick   // FIXME: Should we prefer the preferred type alignment here?
139e5dd7070Spatrick   return CreateMemTemp(Ty, getContext().getTypeAlignInChars(Ty), Name, Alloca);
140e5dd7070Spatrick }
141e5dd7070Spatrick 
CreateMemTemp(QualType Ty,CharUnits Align,const Twine & Name,Address * Alloca)142e5dd7070Spatrick Address CodeGenFunction::CreateMemTemp(QualType Ty, CharUnits Align,
143e5dd7070Spatrick                                        const Twine &Name, Address *Alloca) {
144ec727ea7Spatrick   Address Result = CreateTempAlloca(ConvertTypeForMem(Ty), Align, Name,
145e5dd7070Spatrick                                     /*ArraySize=*/nullptr, Alloca);
146ec727ea7Spatrick 
147ec727ea7Spatrick   if (Ty->isConstantMatrixType()) {
148*12c85518Srobert     auto *ArrayTy = cast<llvm::ArrayType>(Result.getElementType());
149ec727ea7Spatrick     auto *VectorTy = llvm::FixedVectorType::get(ArrayTy->getElementType(),
150ec727ea7Spatrick                                                 ArrayTy->getNumElements());
151ec727ea7Spatrick 
152ec727ea7Spatrick     Result = Address(
153ec727ea7Spatrick         Builder.CreateBitCast(Result.getPointer(), VectorTy->getPointerTo()),
154*12c85518Srobert         VectorTy, Result.getAlignment());
155ec727ea7Spatrick   }
156ec727ea7Spatrick   return Result;
157e5dd7070Spatrick }
158e5dd7070Spatrick 
CreateMemTempWithoutCast(QualType Ty,CharUnits Align,const Twine & Name)159e5dd7070Spatrick Address CodeGenFunction::CreateMemTempWithoutCast(QualType Ty, CharUnits Align,
160e5dd7070Spatrick                                                   const Twine &Name) {
161e5dd7070Spatrick   return CreateTempAllocaWithoutCast(ConvertTypeForMem(Ty), Align, Name);
162e5dd7070Spatrick }
163e5dd7070Spatrick 
CreateMemTempWithoutCast(QualType Ty,const Twine & Name)164e5dd7070Spatrick Address CodeGenFunction::CreateMemTempWithoutCast(QualType Ty,
165e5dd7070Spatrick                                                   const Twine &Name) {
166e5dd7070Spatrick   return CreateMemTempWithoutCast(Ty, getContext().getTypeAlignInChars(Ty),
167e5dd7070Spatrick                                   Name);
168e5dd7070Spatrick }
169e5dd7070Spatrick 
170e5dd7070Spatrick /// EvaluateExprAsBool - Perform the usual unary conversions on the specified
171e5dd7070Spatrick /// expression and compare the result against zero, returning an Int1Ty value.
EvaluateExprAsBool(const Expr * E)172e5dd7070Spatrick llvm::Value *CodeGenFunction::EvaluateExprAsBool(const Expr *E) {
173e5dd7070Spatrick   PGO.setCurrentStmt(E);
174e5dd7070Spatrick   if (const MemberPointerType *MPT = E->getType()->getAs<MemberPointerType>()) {
175e5dd7070Spatrick     llvm::Value *MemPtr = EmitScalarExpr(E);
176e5dd7070Spatrick     return CGM.getCXXABI().EmitMemberPointerIsNotNull(*this, MemPtr, MPT);
177e5dd7070Spatrick   }
178e5dd7070Spatrick 
179e5dd7070Spatrick   QualType BoolTy = getContext().BoolTy;
180e5dd7070Spatrick   SourceLocation Loc = E->getExprLoc();
181a9ac8606Spatrick   CGFPOptionsRAII FPOptsRAII(*this, E);
182e5dd7070Spatrick   if (!E->getType()->isAnyComplexType())
183e5dd7070Spatrick     return EmitScalarConversion(EmitScalarExpr(E), E->getType(), BoolTy, Loc);
184e5dd7070Spatrick 
185e5dd7070Spatrick   return EmitComplexToScalarConversion(EmitComplexExpr(E), E->getType(), BoolTy,
186e5dd7070Spatrick                                        Loc);
187e5dd7070Spatrick }
188e5dd7070Spatrick 
189e5dd7070Spatrick /// EmitIgnoredExpr - Emit code to compute the specified expression,
190e5dd7070Spatrick /// ignoring the result.
EmitIgnoredExpr(const Expr * E)191e5dd7070Spatrick void CodeGenFunction::EmitIgnoredExpr(const Expr *E) {
192a9ac8606Spatrick   if (E->isPRValue())
193e5dd7070Spatrick     return (void)EmitAnyExpr(E, AggValueSlot::ignored(), true);
194e5dd7070Spatrick 
195*12c85518Srobert   // if this is a bitfield-resulting conditional operator, we can special case
196*12c85518Srobert   // emit this. The normal 'EmitLValue' version of this is particularly
197*12c85518Srobert   // difficult to codegen for, since creating a single "LValue" for two
198*12c85518Srobert   // different sized arguments here is not particularly doable.
199*12c85518Srobert   if (const auto *CondOp = dyn_cast<AbstractConditionalOperator>(
200*12c85518Srobert           E->IgnoreParenNoopCasts(getContext()))) {
201*12c85518Srobert     if (CondOp->getObjectKind() == OK_BitField)
202*12c85518Srobert       return EmitIgnoredConditionalOperator(CondOp);
203*12c85518Srobert   }
204*12c85518Srobert 
205e5dd7070Spatrick   // Just emit it as an l-value and drop the result.
206e5dd7070Spatrick   EmitLValue(E);
207e5dd7070Spatrick }
208e5dd7070Spatrick 
209e5dd7070Spatrick /// EmitAnyExpr - Emit code to compute the specified expression which
210e5dd7070Spatrick /// can have any type.  The result is returned as an RValue struct.
211e5dd7070Spatrick /// If this is an aggregate expression, AggSlot indicates where the
212e5dd7070Spatrick /// result should be returned.
EmitAnyExpr(const Expr * E,AggValueSlot aggSlot,bool ignoreResult)213e5dd7070Spatrick RValue CodeGenFunction::EmitAnyExpr(const Expr *E,
214e5dd7070Spatrick                                     AggValueSlot aggSlot,
215e5dd7070Spatrick                                     bool ignoreResult) {
216e5dd7070Spatrick   switch (getEvaluationKind(E->getType())) {
217e5dd7070Spatrick   case TEK_Scalar:
218e5dd7070Spatrick     return RValue::get(EmitScalarExpr(E, ignoreResult));
219e5dd7070Spatrick   case TEK_Complex:
220e5dd7070Spatrick     return RValue::getComplex(EmitComplexExpr(E, ignoreResult, ignoreResult));
221e5dd7070Spatrick   case TEK_Aggregate:
222e5dd7070Spatrick     if (!ignoreResult && aggSlot.isIgnored())
223e5dd7070Spatrick       aggSlot = CreateAggTemp(E->getType(), "agg-temp");
224e5dd7070Spatrick     EmitAggExpr(E, aggSlot);
225e5dd7070Spatrick     return aggSlot.asRValue();
226e5dd7070Spatrick   }
227e5dd7070Spatrick   llvm_unreachable("bad evaluation kind");
228e5dd7070Spatrick }
229e5dd7070Spatrick 
230e5dd7070Spatrick /// EmitAnyExprToTemp - Similar to EmitAnyExpr(), however, the result will
231e5dd7070Spatrick /// always be accessible even if no aggregate location is provided.
EmitAnyExprToTemp(const Expr * E)232e5dd7070Spatrick RValue CodeGenFunction::EmitAnyExprToTemp(const Expr *E) {
233e5dd7070Spatrick   AggValueSlot AggSlot = AggValueSlot::ignored();
234e5dd7070Spatrick 
235e5dd7070Spatrick   if (hasAggregateEvaluationKind(E->getType()))
236e5dd7070Spatrick     AggSlot = CreateAggTemp(E->getType(), "agg.tmp");
237e5dd7070Spatrick   return EmitAnyExpr(E, AggSlot);
238e5dd7070Spatrick }
239e5dd7070Spatrick 
240e5dd7070Spatrick /// EmitAnyExprToMem - Evaluate an expression into a given memory
241e5dd7070Spatrick /// location.
EmitAnyExprToMem(const Expr * E,Address Location,Qualifiers Quals,bool IsInit)242e5dd7070Spatrick void CodeGenFunction::EmitAnyExprToMem(const Expr *E,
243e5dd7070Spatrick                                        Address Location,
244e5dd7070Spatrick                                        Qualifiers Quals,
245e5dd7070Spatrick                                        bool IsInit) {
246e5dd7070Spatrick   // FIXME: This function should take an LValue as an argument.
247e5dd7070Spatrick   switch (getEvaluationKind(E->getType())) {
248e5dd7070Spatrick   case TEK_Complex:
249e5dd7070Spatrick     EmitComplexExprIntoLValue(E, MakeAddrLValue(Location, E->getType()),
250e5dd7070Spatrick                               /*isInit*/ false);
251e5dd7070Spatrick     return;
252e5dd7070Spatrick 
253e5dd7070Spatrick   case TEK_Aggregate: {
254e5dd7070Spatrick     EmitAggExpr(E, AggValueSlot::forAddr(Location, Quals,
255e5dd7070Spatrick                                          AggValueSlot::IsDestructed_t(IsInit),
256e5dd7070Spatrick                                          AggValueSlot::DoesNotNeedGCBarriers,
257e5dd7070Spatrick                                          AggValueSlot::IsAliased_t(!IsInit),
258e5dd7070Spatrick                                          AggValueSlot::MayOverlap));
259e5dd7070Spatrick     return;
260e5dd7070Spatrick   }
261e5dd7070Spatrick 
262e5dd7070Spatrick   case TEK_Scalar: {
263e5dd7070Spatrick     RValue RV = RValue::get(EmitScalarExpr(E, /*Ignore*/ false));
264e5dd7070Spatrick     LValue LV = MakeAddrLValue(Location, E->getType());
265e5dd7070Spatrick     EmitStoreThroughLValue(RV, LV);
266e5dd7070Spatrick     return;
267e5dd7070Spatrick   }
268e5dd7070Spatrick   }
269e5dd7070Spatrick   llvm_unreachable("bad evaluation kind");
270e5dd7070Spatrick }
271e5dd7070Spatrick 
272e5dd7070Spatrick static void
pushTemporaryCleanup(CodeGenFunction & CGF,const MaterializeTemporaryExpr * M,const Expr * E,Address ReferenceTemporary)273e5dd7070Spatrick pushTemporaryCleanup(CodeGenFunction &CGF, const MaterializeTemporaryExpr *M,
274e5dd7070Spatrick                      const Expr *E, Address ReferenceTemporary) {
275e5dd7070Spatrick   // Objective-C++ ARC:
276e5dd7070Spatrick   //   If we are binding a reference to a temporary that has ownership, we
277e5dd7070Spatrick   //   need to perform retain/release operations on the temporary.
278e5dd7070Spatrick   //
279e5dd7070Spatrick   // FIXME: This should be looking at E, not M.
280e5dd7070Spatrick   if (auto Lifetime = M->getType().getObjCLifetime()) {
281e5dd7070Spatrick     switch (Lifetime) {
282e5dd7070Spatrick     case Qualifiers::OCL_None:
283e5dd7070Spatrick     case Qualifiers::OCL_ExplicitNone:
284e5dd7070Spatrick       // Carry on to normal cleanup handling.
285e5dd7070Spatrick       break;
286e5dd7070Spatrick 
287e5dd7070Spatrick     case Qualifiers::OCL_Autoreleasing:
288e5dd7070Spatrick       // Nothing to do; cleaned up by an autorelease pool.
289e5dd7070Spatrick       return;
290e5dd7070Spatrick 
291e5dd7070Spatrick     case Qualifiers::OCL_Strong:
292e5dd7070Spatrick     case Qualifiers::OCL_Weak:
293e5dd7070Spatrick       switch (StorageDuration Duration = M->getStorageDuration()) {
294e5dd7070Spatrick       case SD_Static:
295e5dd7070Spatrick         // Note: we intentionally do not register a cleanup to release
296e5dd7070Spatrick         // the object on program termination.
297e5dd7070Spatrick         return;
298e5dd7070Spatrick 
299e5dd7070Spatrick       case SD_Thread:
300e5dd7070Spatrick         // FIXME: We should probably register a cleanup in this case.
301e5dd7070Spatrick         return;
302e5dd7070Spatrick 
303e5dd7070Spatrick       case SD_Automatic:
304e5dd7070Spatrick       case SD_FullExpression:
305e5dd7070Spatrick         CodeGenFunction::Destroyer *Destroy;
306e5dd7070Spatrick         CleanupKind CleanupKind;
307e5dd7070Spatrick         if (Lifetime == Qualifiers::OCL_Strong) {
308e5dd7070Spatrick           const ValueDecl *VD = M->getExtendingDecl();
309e5dd7070Spatrick           bool Precise =
310e5dd7070Spatrick               VD && isa<VarDecl>(VD) && VD->hasAttr<ObjCPreciseLifetimeAttr>();
311e5dd7070Spatrick           CleanupKind = CGF.getARCCleanupKind();
312e5dd7070Spatrick           Destroy = Precise ? &CodeGenFunction::destroyARCStrongPrecise
313e5dd7070Spatrick                             : &CodeGenFunction::destroyARCStrongImprecise;
314e5dd7070Spatrick         } else {
315e5dd7070Spatrick           // __weak objects always get EH cleanups; otherwise, exceptions
316e5dd7070Spatrick           // could cause really nasty crashes instead of mere leaks.
317e5dd7070Spatrick           CleanupKind = NormalAndEHCleanup;
318e5dd7070Spatrick           Destroy = &CodeGenFunction::destroyARCWeak;
319e5dd7070Spatrick         }
320e5dd7070Spatrick         if (Duration == SD_FullExpression)
321e5dd7070Spatrick           CGF.pushDestroy(CleanupKind, ReferenceTemporary,
322e5dd7070Spatrick                           M->getType(), *Destroy,
323e5dd7070Spatrick                           CleanupKind & EHCleanup);
324e5dd7070Spatrick         else
325e5dd7070Spatrick           CGF.pushLifetimeExtendedDestroy(CleanupKind, ReferenceTemporary,
326e5dd7070Spatrick                                           M->getType(),
327e5dd7070Spatrick                                           *Destroy, CleanupKind & EHCleanup);
328e5dd7070Spatrick         return;
329e5dd7070Spatrick 
330e5dd7070Spatrick       case SD_Dynamic:
331e5dd7070Spatrick         llvm_unreachable("temporary cannot have dynamic storage duration");
332e5dd7070Spatrick       }
333e5dd7070Spatrick       llvm_unreachable("unknown storage duration");
334e5dd7070Spatrick     }
335e5dd7070Spatrick   }
336e5dd7070Spatrick 
337e5dd7070Spatrick   CXXDestructorDecl *ReferenceTemporaryDtor = nullptr;
338e5dd7070Spatrick   if (const RecordType *RT =
339e5dd7070Spatrick           E->getType()->getBaseElementTypeUnsafe()->getAs<RecordType>()) {
340e5dd7070Spatrick     // Get the destructor for the reference temporary.
341e5dd7070Spatrick     auto *ClassDecl = cast<CXXRecordDecl>(RT->getDecl());
342e5dd7070Spatrick     if (!ClassDecl->hasTrivialDestructor())
343e5dd7070Spatrick       ReferenceTemporaryDtor = ClassDecl->getDestructor();
344e5dd7070Spatrick   }
345e5dd7070Spatrick 
346e5dd7070Spatrick   if (!ReferenceTemporaryDtor)
347e5dd7070Spatrick     return;
348e5dd7070Spatrick 
349e5dd7070Spatrick   // Call the destructor for the temporary.
350e5dd7070Spatrick   switch (M->getStorageDuration()) {
351e5dd7070Spatrick   case SD_Static:
352e5dd7070Spatrick   case SD_Thread: {
353e5dd7070Spatrick     llvm::FunctionCallee CleanupFn;
354e5dd7070Spatrick     llvm::Constant *CleanupArg;
355e5dd7070Spatrick     if (E->getType()->isArrayType()) {
356e5dd7070Spatrick       CleanupFn = CodeGenFunction(CGF.CGM).generateDestroyHelper(
357e5dd7070Spatrick           ReferenceTemporary, E->getType(),
358e5dd7070Spatrick           CodeGenFunction::destroyCXXObject, CGF.getLangOpts().Exceptions,
359e5dd7070Spatrick           dyn_cast_or_null<VarDecl>(M->getExtendingDecl()));
360e5dd7070Spatrick       CleanupArg = llvm::Constant::getNullValue(CGF.Int8PtrTy);
361e5dd7070Spatrick     } else {
362e5dd7070Spatrick       CleanupFn = CGF.CGM.getAddrAndTypeOfCXXStructor(
363e5dd7070Spatrick           GlobalDecl(ReferenceTemporaryDtor, Dtor_Complete));
364e5dd7070Spatrick       CleanupArg = cast<llvm::Constant>(ReferenceTemporary.getPointer());
365e5dd7070Spatrick     }
366e5dd7070Spatrick     CGF.CGM.getCXXABI().registerGlobalDtor(
367e5dd7070Spatrick         CGF, *cast<VarDecl>(M->getExtendingDecl()), CleanupFn, CleanupArg);
368e5dd7070Spatrick     break;
369e5dd7070Spatrick   }
370e5dd7070Spatrick 
371e5dd7070Spatrick   case SD_FullExpression:
372e5dd7070Spatrick     CGF.pushDestroy(NormalAndEHCleanup, ReferenceTemporary, E->getType(),
373e5dd7070Spatrick                     CodeGenFunction::destroyCXXObject,
374e5dd7070Spatrick                     CGF.getLangOpts().Exceptions);
375e5dd7070Spatrick     break;
376e5dd7070Spatrick 
377e5dd7070Spatrick   case SD_Automatic:
378e5dd7070Spatrick     CGF.pushLifetimeExtendedDestroy(NormalAndEHCleanup,
379e5dd7070Spatrick                                     ReferenceTemporary, E->getType(),
380e5dd7070Spatrick                                     CodeGenFunction::destroyCXXObject,
381e5dd7070Spatrick                                     CGF.getLangOpts().Exceptions);
382e5dd7070Spatrick     break;
383e5dd7070Spatrick 
384e5dd7070Spatrick   case SD_Dynamic:
385e5dd7070Spatrick     llvm_unreachable("temporary cannot have dynamic storage duration");
386e5dd7070Spatrick   }
387e5dd7070Spatrick }
388e5dd7070Spatrick 
createReferenceTemporary(CodeGenFunction & CGF,const MaterializeTemporaryExpr * M,const Expr * Inner,Address * Alloca=nullptr)389e5dd7070Spatrick static Address createReferenceTemporary(CodeGenFunction &CGF,
390e5dd7070Spatrick                                         const MaterializeTemporaryExpr *M,
391e5dd7070Spatrick                                         const Expr *Inner,
392e5dd7070Spatrick                                         Address *Alloca = nullptr) {
393e5dd7070Spatrick   auto &TCG = CGF.getTargetHooks();
394e5dd7070Spatrick   switch (M->getStorageDuration()) {
395e5dd7070Spatrick   case SD_FullExpression:
396e5dd7070Spatrick   case SD_Automatic: {
397e5dd7070Spatrick     // If we have a constant temporary array or record try to promote it into a
398e5dd7070Spatrick     // constant global under the same rules a normal constant would've been
399e5dd7070Spatrick     // promoted. This is easier on the optimizer and generally emits fewer
400e5dd7070Spatrick     // instructions.
401e5dd7070Spatrick     QualType Ty = Inner->getType();
402e5dd7070Spatrick     if (CGF.CGM.getCodeGenOpts().MergeAllConstants &&
403e5dd7070Spatrick         (Ty->isArrayType() || Ty->isRecordType()) &&
404e5dd7070Spatrick         CGF.CGM.isTypeConstant(Ty, true))
405e5dd7070Spatrick       if (auto Init = ConstantEmitter(CGF).tryEmitAbstract(Inner, Ty)) {
406a9ac8606Spatrick         auto AS = CGF.CGM.GetGlobalConstantAddressSpace();
407e5dd7070Spatrick         auto *GV = new llvm::GlobalVariable(
408e5dd7070Spatrick             CGF.CGM.getModule(), Init->getType(), /*isConstant=*/true,
409e5dd7070Spatrick             llvm::GlobalValue::PrivateLinkage, Init, ".ref.tmp", nullptr,
410e5dd7070Spatrick             llvm::GlobalValue::NotThreadLocal,
411e5dd7070Spatrick             CGF.getContext().getTargetAddressSpace(AS));
412e5dd7070Spatrick         CharUnits alignment = CGF.getContext().getTypeAlignInChars(Ty);
413e5dd7070Spatrick         GV->setAlignment(alignment.getAsAlign());
414e5dd7070Spatrick         llvm::Constant *C = GV;
415e5dd7070Spatrick         if (AS != LangAS::Default)
416e5dd7070Spatrick           C = TCG.performAddrSpaceCast(
417e5dd7070Spatrick               CGF.CGM, GV, AS, LangAS::Default,
418e5dd7070Spatrick               GV->getValueType()->getPointerTo(
419e5dd7070Spatrick                   CGF.getContext().getTargetAddressSpace(LangAS::Default)));
420e5dd7070Spatrick         // FIXME: Should we put the new global into a COMDAT?
421*12c85518Srobert         return Address(C, GV->getValueType(), alignment);
422e5dd7070Spatrick       }
423e5dd7070Spatrick     return CGF.CreateMemTemp(Ty, "ref.tmp", Alloca);
424e5dd7070Spatrick   }
425e5dd7070Spatrick   case SD_Thread:
426e5dd7070Spatrick   case SD_Static:
427e5dd7070Spatrick     return CGF.CGM.GetAddrOfGlobalTemporary(M, Inner);
428e5dd7070Spatrick 
429e5dd7070Spatrick   case SD_Dynamic:
430e5dd7070Spatrick     llvm_unreachable("temporary can't have dynamic storage duration");
431e5dd7070Spatrick   }
432e5dd7070Spatrick   llvm_unreachable("unknown storage duration");
433e5dd7070Spatrick }
434e5dd7070Spatrick 
435ec727ea7Spatrick /// Helper method to check if the underlying ABI is AAPCS
isAAPCS(const TargetInfo & TargetInfo)436ec727ea7Spatrick static bool isAAPCS(const TargetInfo &TargetInfo) {
437ec727ea7Spatrick   return TargetInfo.getABI().startswith("aapcs");
438ec727ea7Spatrick }
439ec727ea7Spatrick 
440e5dd7070Spatrick LValue CodeGenFunction::
EmitMaterializeTemporaryExpr(const MaterializeTemporaryExpr * M)441e5dd7070Spatrick EmitMaterializeTemporaryExpr(const MaterializeTemporaryExpr *M) {
442e5dd7070Spatrick   const Expr *E = M->getSubExpr();
443e5dd7070Spatrick 
444e5dd7070Spatrick   assert((!M->getExtendingDecl() || !isa<VarDecl>(M->getExtendingDecl()) ||
445e5dd7070Spatrick           !cast<VarDecl>(M->getExtendingDecl())->isARCPseudoStrong()) &&
446e5dd7070Spatrick          "Reference should never be pseudo-strong!");
447e5dd7070Spatrick 
448e5dd7070Spatrick   // FIXME: ideally this would use EmitAnyExprToMem, however, we cannot do so
449e5dd7070Spatrick   // as that will cause the lifetime adjustment to be lost for ARC
450e5dd7070Spatrick   auto ownership = M->getType().getObjCLifetime();
451e5dd7070Spatrick   if (ownership != Qualifiers::OCL_None &&
452e5dd7070Spatrick       ownership != Qualifiers::OCL_ExplicitNone) {
453e5dd7070Spatrick     Address Object = createReferenceTemporary(*this, M, E);
454e5dd7070Spatrick     if (auto *Var = dyn_cast<llvm::GlobalVariable>(Object.getPointer())) {
455*12c85518Srobert       llvm::Type *Ty = ConvertTypeForMem(E->getType());
456*12c85518Srobert       Object = Address(llvm::ConstantExpr::getBitCast(
457*12c85518Srobert                            Var, Ty->getPointerTo(Object.getAddressSpace())),
458*12c85518Srobert                        Ty, Object.getAlignment());
459e5dd7070Spatrick 
460e5dd7070Spatrick       // createReferenceTemporary will promote the temporary to a global with a
461e5dd7070Spatrick       // constant initializer if it can.  It can only do this to a value of
462e5dd7070Spatrick       // ARC-manageable type if the value is global and therefore "immune" to
463e5dd7070Spatrick       // ref-counting operations.  Therefore we have no need to emit either a
464e5dd7070Spatrick       // dynamic initialization or a cleanup and we can just return the address
465e5dd7070Spatrick       // of the temporary.
466e5dd7070Spatrick       if (Var->hasInitializer())
467e5dd7070Spatrick         return MakeAddrLValue(Object, M->getType(), AlignmentSource::Decl);
468e5dd7070Spatrick 
469e5dd7070Spatrick       Var->setInitializer(CGM.EmitNullConstant(E->getType()));
470e5dd7070Spatrick     }
471e5dd7070Spatrick     LValue RefTempDst = MakeAddrLValue(Object, M->getType(),
472e5dd7070Spatrick                                        AlignmentSource::Decl);
473e5dd7070Spatrick 
474e5dd7070Spatrick     switch (getEvaluationKind(E->getType())) {
475e5dd7070Spatrick     default: llvm_unreachable("expected scalar or aggregate expression");
476e5dd7070Spatrick     case TEK_Scalar:
477e5dd7070Spatrick       EmitScalarInit(E, M->getExtendingDecl(), RefTempDst, false);
478e5dd7070Spatrick       break;
479e5dd7070Spatrick     case TEK_Aggregate: {
480e5dd7070Spatrick       EmitAggExpr(E, AggValueSlot::forAddr(Object,
481e5dd7070Spatrick                                            E->getType().getQualifiers(),
482e5dd7070Spatrick                                            AggValueSlot::IsDestructed,
483e5dd7070Spatrick                                            AggValueSlot::DoesNotNeedGCBarriers,
484e5dd7070Spatrick                                            AggValueSlot::IsNotAliased,
485e5dd7070Spatrick                                            AggValueSlot::DoesNotOverlap));
486e5dd7070Spatrick       break;
487e5dd7070Spatrick     }
488e5dd7070Spatrick     }
489e5dd7070Spatrick 
490e5dd7070Spatrick     pushTemporaryCleanup(*this, M, E, Object);
491e5dd7070Spatrick     return RefTempDst;
492e5dd7070Spatrick   }
493e5dd7070Spatrick 
494e5dd7070Spatrick   SmallVector<const Expr *, 2> CommaLHSs;
495e5dd7070Spatrick   SmallVector<SubobjectAdjustment, 2> Adjustments;
496e5dd7070Spatrick   E = E->skipRValueSubobjectAdjustments(CommaLHSs, Adjustments);
497e5dd7070Spatrick 
498e5dd7070Spatrick   for (const auto &Ignored : CommaLHSs)
499e5dd7070Spatrick     EmitIgnoredExpr(Ignored);
500e5dd7070Spatrick 
501e5dd7070Spatrick   if (const auto *opaque = dyn_cast<OpaqueValueExpr>(E)) {
502e5dd7070Spatrick     if (opaque->getType()->isRecordType()) {
503e5dd7070Spatrick       assert(Adjustments.empty());
504e5dd7070Spatrick       return EmitOpaqueValueLValue(opaque);
505e5dd7070Spatrick     }
506e5dd7070Spatrick   }
507e5dd7070Spatrick 
508e5dd7070Spatrick   // Create and initialize the reference temporary.
509e5dd7070Spatrick   Address Alloca = Address::invalid();
510e5dd7070Spatrick   Address Object = createReferenceTemporary(*this, M, E, &Alloca);
511e5dd7070Spatrick   if (auto *Var = dyn_cast<llvm::GlobalVariable>(
512e5dd7070Spatrick           Object.getPointer()->stripPointerCasts())) {
513*12c85518Srobert     llvm::Type *TemporaryType = ConvertTypeForMem(E->getType());
514e5dd7070Spatrick     Object = Address(llvm::ConstantExpr::getBitCast(
515e5dd7070Spatrick                          cast<llvm::Constant>(Object.getPointer()),
516*12c85518Srobert                          TemporaryType->getPointerTo()),
517*12c85518Srobert                      TemporaryType,
518e5dd7070Spatrick                      Object.getAlignment());
519e5dd7070Spatrick     // If the temporary is a global and has a constant initializer or is a
520e5dd7070Spatrick     // constant temporary that we promoted to a global, we may have already
521e5dd7070Spatrick     // initialized it.
522e5dd7070Spatrick     if (!Var->hasInitializer()) {
523e5dd7070Spatrick       Var->setInitializer(CGM.EmitNullConstant(E->getType()));
524e5dd7070Spatrick       EmitAnyExprToMem(E, Object, Qualifiers(), /*IsInit*/true);
525e5dd7070Spatrick     }
526e5dd7070Spatrick   } else {
527e5dd7070Spatrick     switch (M->getStorageDuration()) {
528e5dd7070Spatrick     case SD_Automatic:
529e5dd7070Spatrick       if (auto *Size = EmitLifetimeStart(
530e5dd7070Spatrick               CGM.getDataLayout().getTypeAllocSize(Alloca.getElementType()),
531e5dd7070Spatrick               Alloca.getPointer())) {
532e5dd7070Spatrick         pushCleanupAfterFullExpr<CallLifetimeEnd>(NormalEHLifetimeMarker,
533e5dd7070Spatrick                                                   Alloca, Size);
534e5dd7070Spatrick       }
535e5dd7070Spatrick       break;
536e5dd7070Spatrick 
537e5dd7070Spatrick     case SD_FullExpression: {
538e5dd7070Spatrick       if (!ShouldEmitLifetimeMarkers)
539e5dd7070Spatrick         break;
540e5dd7070Spatrick 
541e5dd7070Spatrick       // Avoid creating a conditional cleanup just to hold an llvm.lifetime.end
542e5dd7070Spatrick       // marker. Instead, start the lifetime of a conditional temporary earlier
543e5dd7070Spatrick       // so that it's unconditional. Don't do this with sanitizers which need
544e5dd7070Spatrick       // more precise lifetime marks.
545e5dd7070Spatrick       ConditionalEvaluation *OldConditional = nullptr;
546e5dd7070Spatrick       CGBuilderTy::InsertPoint OldIP;
547e5dd7070Spatrick       if (isInConditionalBranch() && !E->getType().isDestructedType() &&
548e5dd7070Spatrick           !SanOpts.has(SanitizerKind::HWAddress) &&
549e5dd7070Spatrick           !SanOpts.has(SanitizerKind::Memory) &&
550e5dd7070Spatrick           !CGM.getCodeGenOpts().SanitizeAddressUseAfterScope) {
551e5dd7070Spatrick         OldConditional = OutermostConditional;
552e5dd7070Spatrick         OutermostConditional = nullptr;
553e5dd7070Spatrick 
554e5dd7070Spatrick         OldIP = Builder.saveIP();
555e5dd7070Spatrick         llvm::BasicBlock *Block = OldConditional->getStartingBlock();
556e5dd7070Spatrick         Builder.restoreIP(CGBuilderTy::InsertPoint(
557e5dd7070Spatrick             Block, llvm::BasicBlock::iterator(Block->back())));
558e5dd7070Spatrick       }
559e5dd7070Spatrick 
560e5dd7070Spatrick       if (auto *Size = EmitLifetimeStart(
561e5dd7070Spatrick               CGM.getDataLayout().getTypeAllocSize(Alloca.getElementType()),
562e5dd7070Spatrick               Alloca.getPointer())) {
563e5dd7070Spatrick         pushFullExprCleanup<CallLifetimeEnd>(NormalEHLifetimeMarker, Alloca,
564e5dd7070Spatrick                                              Size);
565e5dd7070Spatrick       }
566e5dd7070Spatrick 
567e5dd7070Spatrick       if (OldConditional) {
568e5dd7070Spatrick         OutermostConditional = OldConditional;
569e5dd7070Spatrick         Builder.restoreIP(OldIP);
570e5dd7070Spatrick       }
571e5dd7070Spatrick       break;
572e5dd7070Spatrick     }
573e5dd7070Spatrick 
574e5dd7070Spatrick     default:
575e5dd7070Spatrick       break;
576e5dd7070Spatrick     }
577e5dd7070Spatrick     EmitAnyExprToMem(E, Object, Qualifiers(), /*IsInit*/true);
578e5dd7070Spatrick   }
579e5dd7070Spatrick   pushTemporaryCleanup(*this, M, E, Object);
580e5dd7070Spatrick 
581e5dd7070Spatrick   // Perform derived-to-base casts and/or field accesses, to get from the
582e5dd7070Spatrick   // temporary object we created (and, potentially, for which we extended
583e5dd7070Spatrick   // the lifetime) to the subobject we're binding the reference to.
584*12c85518Srobert   for (SubobjectAdjustment &Adjustment : llvm::reverse(Adjustments)) {
585e5dd7070Spatrick     switch (Adjustment.Kind) {
586e5dd7070Spatrick     case SubobjectAdjustment::DerivedToBaseAdjustment:
587e5dd7070Spatrick       Object =
588e5dd7070Spatrick           GetAddressOfBaseClass(Object, Adjustment.DerivedToBase.DerivedClass,
589e5dd7070Spatrick                                 Adjustment.DerivedToBase.BasePath->path_begin(),
590e5dd7070Spatrick                                 Adjustment.DerivedToBase.BasePath->path_end(),
591e5dd7070Spatrick                                 /*NullCheckValue=*/ false, E->getExprLoc());
592e5dd7070Spatrick       break;
593e5dd7070Spatrick 
594e5dd7070Spatrick     case SubobjectAdjustment::FieldAdjustment: {
595e5dd7070Spatrick       LValue LV = MakeAddrLValue(Object, E->getType(), AlignmentSource::Decl);
596e5dd7070Spatrick       LV = EmitLValueForField(LV, Adjustment.Field);
597e5dd7070Spatrick       assert(LV.isSimple() &&
598e5dd7070Spatrick              "materialized temporary field is not a simple lvalue");
599e5dd7070Spatrick       Object = LV.getAddress(*this);
600e5dd7070Spatrick       break;
601e5dd7070Spatrick     }
602e5dd7070Spatrick 
603e5dd7070Spatrick     case SubobjectAdjustment::MemberPointerAdjustment: {
604e5dd7070Spatrick       llvm::Value *Ptr = EmitScalarExpr(Adjustment.Ptr.RHS);
605e5dd7070Spatrick       Object = EmitCXXMemberDataPointerAddress(E, Object, Ptr,
606e5dd7070Spatrick                                                Adjustment.Ptr.MPT);
607e5dd7070Spatrick       break;
608e5dd7070Spatrick     }
609e5dd7070Spatrick     }
610e5dd7070Spatrick   }
611e5dd7070Spatrick 
612e5dd7070Spatrick   return MakeAddrLValue(Object, M->getType(), AlignmentSource::Decl);
613e5dd7070Spatrick }
614e5dd7070Spatrick 
615e5dd7070Spatrick RValue
EmitReferenceBindingToExpr(const Expr * E)616e5dd7070Spatrick CodeGenFunction::EmitReferenceBindingToExpr(const Expr *E) {
617e5dd7070Spatrick   // Emit the expression as an lvalue.
618e5dd7070Spatrick   LValue LV = EmitLValue(E);
619e5dd7070Spatrick   assert(LV.isSimple());
620e5dd7070Spatrick   llvm::Value *Value = LV.getPointer(*this);
621e5dd7070Spatrick 
622e5dd7070Spatrick   if (sanitizePerformTypeCheck() && !E->getType()->isFunctionType()) {
623e5dd7070Spatrick     // C++11 [dcl.ref]p5 (as amended by core issue 453):
624e5dd7070Spatrick     //   If a glvalue to which a reference is directly bound designates neither
625e5dd7070Spatrick     //   an existing object or function of an appropriate type nor a region of
626e5dd7070Spatrick     //   storage of suitable size and alignment to contain an object of the
627e5dd7070Spatrick     //   reference's type, the behavior is undefined.
628e5dd7070Spatrick     QualType Ty = E->getType();
629e5dd7070Spatrick     EmitTypeCheck(TCK_ReferenceBinding, E->getExprLoc(), Value, Ty);
630e5dd7070Spatrick   }
631e5dd7070Spatrick 
632e5dd7070Spatrick   return RValue::get(Value);
633e5dd7070Spatrick }
634e5dd7070Spatrick 
635e5dd7070Spatrick 
636e5dd7070Spatrick /// getAccessedFieldNo - Given an encoded value and a result number, return the
637e5dd7070Spatrick /// input field number being accessed.
getAccessedFieldNo(unsigned Idx,const llvm::Constant * Elts)638e5dd7070Spatrick unsigned CodeGenFunction::getAccessedFieldNo(unsigned Idx,
639e5dd7070Spatrick                                              const llvm::Constant *Elts) {
640e5dd7070Spatrick   return cast<llvm::ConstantInt>(Elts->getAggregateElement(Idx))
641e5dd7070Spatrick       ->getZExtValue();
642e5dd7070Spatrick }
643e5dd7070Spatrick 
644e5dd7070Spatrick /// Emit the hash_16_bytes function from include/llvm/ADT/Hashing.h.
emitHash16Bytes(CGBuilderTy & Builder,llvm::Value * Low,llvm::Value * High)645e5dd7070Spatrick static llvm::Value *emitHash16Bytes(CGBuilderTy &Builder, llvm::Value *Low,
646e5dd7070Spatrick                                     llvm::Value *High) {
647e5dd7070Spatrick   llvm::Value *KMul = Builder.getInt64(0x9ddfea08eb382d69ULL);
648e5dd7070Spatrick   llvm::Value *K47 = Builder.getInt64(47);
649e5dd7070Spatrick   llvm::Value *A0 = Builder.CreateMul(Builder.CreateXor(Low, High), KMul);
650e5dd7070Spatrick   llvm::Value *A1 = Builder.CreateXor(Builder.CreateLShr(A0, K47), A0);
651e5dd7070Spatrick   llvm::Value *B0 = Builder.CreateMul(Builder.CreateXor(High, A1), KMul);
652e5dd7070Spatrick   llvm::Value *B1 = Builder.CreateXor(Builder.CreateLShr(B0, K47), B0);
653e5dd7070Spatrick   return Builder.CreateMul(B1, KMul);
654e5dd7070Spatrick }
655e5dd7070Spatrick 
isNullPointerAllowed(TypeCheckKind TCK)656e5dd7070Spatrick bool CodeGenFunction::isNullPointerAllowed(TypeCheckKind TCK) {
657e5dd7070Spatrick   return TCK == TCK_DowncastPointer || TCK == TCK_Upcast ||
658e5dd7070Spatrick          TCK == TCK_UpcastToVirtualBase || TCK == TCK_DynamicOperation;
659e5dd7070Spatrick }
660e5dd7070Spatrick 
isVptrCheckRequired(TypeCheckKind TCK,QualType Ty)661e5dd7070Spatrick bool CodeGenFunction::isVptrCheckRequired(TypeCheckKind TCK, QualType Ty) {
662e5dd7070Spatrick   CXXRecordDecl *RD = Ty->getAsCXXRecordDecl();
663e5dd7070Spatrick   return (RD && RD->hasDefinition() && RD->isDynamicClass()) &&
664e5dd7070Spatrick          (TCK == TCK_MemberAccess || TCK == TCK_MemberCall ||
665e5dd7070Spatrick           TCK == TCK_DowncastPointer || TCK == TCK_DowncastReference ||
666e5dd7070Spatrick           TCK == TCK_UpcastToVirtualBase || TCK == TCK_DynamicOperation);
667e5dd7070Spatrick }
668e5dd7070Spatrick 
sanitizePerformTypeCheck() const669e5dd7070Spatrick bool CodeGenFunction::sanitizePerformTypeCheck() const {
670*12c85518Srobert   return SanOpts.has(SanitizerKind::Null) ||
671*12c85518Srobert          SanOpts.has(SanitizerKind::Alignment) ||
672*12c85518Srobert          SanOpts.has(SanitizerKind::ObjectSize) ||
673e5dd7070Spatrick          SanOpts.has(SanitizerKind::Vptr);
674e5dd7070Spatrick }
675e5dd7070Spatrick 
EmitTypeCheck(TypeCheckKind TCK,SourceLocation Loc,llvm::Value * Ptr,QualType Ty,CharUnits Alignment,SanitizerSet SkippedChecks,llvm::Value * ArraySize)676e5dd7070Spatrick void CodeGenFunction::EmitTypeCheck(TypeCheckKind TCK, SourceLocation Loc,
677e5dd7070Spatrick                                     llvm::Value *Ptr, QualType Ty,
678e5dd7070Spatrick                                     CharUnits Alignment,
679e5dd7070Spatrick                                     SanitizerSet SkippedChecks,
680e5dd7070Spatrick                                     llvm::Value *ArraySize) {
681e5dd7070Spatrick   if (!sanitizePerformTypeCheck())
682e5dd7070Spatrick     return;
683e5dd7070Spatrick 
684e5dd7070Spatrick   // Don't check pointers outside the default address space. The null check
685e5dd7070Spatrick   // isn't correct, the object-size check isn't supported by LLVM, and we can't
686e5dd7070Spatrick   // communicate the addresses to the runtime handler for the vptr check.
687e5dd7070Spatrick   if (Ptr->getType()->getPointerAddressSpace())
688e5dd7070Spatrick     return;
689e5dd7070Spatrick 
690e5dd7070Spatrick   // Don't check pointers to volatile data. The behavior here is implementation-
691e5dd7070Spatrick   // defined.
692e5dd7070Spatrick   if (Ty.isVolatileQualified())
693e5dd7070Spatrick     return;
694e5dd7070Spatrick 
695e5dd7070Spatrick   SanitizerScope SanScope(this);
696e5dd7070Spatrick 
697e5dd7070Spatrick   SmallVector<std::pair<llvm::Value *, SanitizerMask>, 3> Checks;
698e5dd7070Spatrick   llvm::BasicBlock *Done = nullptr;
699e5dd7070Spatrick 
700e5dd7070Spatrick   // Quickly determine whether we have a pointer to an alloca. It's possible
701e5dd7070Spatrick   // to skip null checks, and some alignment checks, for these pointers. This
702e5dd7070Spatrick   // can reduce compile-time significantly.
703e5dd7070Spatrick   auto PtrToAlloca = dyn_cast<llvm::AllocaInst>(Ptr->stripPointerCasts());
704e5dd7070Spatrick 
705e5dd7070Spatrick   llvm::Value *True = llvm::ConstantInt::getTrue(getLLVMContext());
706e5dd7070Spatrick   llvm::Value *IsNonNull = nullptr;
707e5dd7070Spatrick   bool IsGuaranteedNonNull =
708e5dd7070Spatrick       SkippedChecks.has(SanitizerKind::Null) || PtrToAlloca;
709e5dd7070Spatrick   bool AllowNullPointers = isNullPointerAllowed(TCK);
710e5dd7070Spatrick   if ((SanOpts.has(SanitizerKind::Null) || AllowNullPointers) &&
711e5dd7070Spatrick       !IsGuaranteedNonNull) {
712e5dd7070Spatrick     // The glvalue must not be an empty glvalue.
713e5dd7070Spatrick     IsNonNull = Builder.CreateIsNotNull(Ptr);
714e5dd7070Spatrick 
715e5dd7070Spatrick     // The IR builder can constant-fold the null check if the pointer points to
716e5dd7070Spatrick     // a constant.
717e5dd7070Spatrick     IsGuaranteedNonNull = IsNonNull == True;
718e5dd7070Spatrick 
719e5dd7070Spatrick     // Skip the null check if the pointer is known to be non-null.
720e5dd7070Spatrick     if (!IsGuaranteedNonNull) {
721e5dd7070Spatrick       if (AllowNullPointers) {
722e5dd7070Spatrick         // When performing pointer casts, it's OK if the value is null.
723e5dd7070Spatrick         // Skip the remaining checks in that case.
724e5dd7070Spatrick         Done = createBasicBlock("null");
725e5dd7070Spatrick         llvm::BasicBlock *Rest = createBasicBlock("not.null");
726e5dd7070Spatrick         Builder.CreateCondBr(IsNonNull, Rest, Done);
727e5dd7070Spatrick         EmitBlock(Rest);
728e5dd7070Spatrick       } else {
729e5dd7070Spatrick         Checks.push_back(std::make_pair(IsNonNull, SanitizerKind::Null));
730e5dd7070Spatrick       }
731e5dd7070Spatrick     }
732e5dd7070Spatrick   }
733e5dd7070Spatrick 
734e5dd7070Spatrick   if (SanOpts.has(SanitizerKind::ObjectSize) &&
735e5dd7070Spatrick       !SkippedChecks.has(SanitizerKind::ObjectSize) &&
736e5dd7070Spatrick       !Ty->isIncompleteType()) {
737ec727ea7Spatrick     uint64_t TySize = CGM.getMinimumObjectSize(Ty).getQuantity();
738e5dd7070Spatrick     llvm::Value *Size = llvm::ConstantInt::get(IntPtrTy, TySize);
739e5dd7070Spatrick     if (ArraySize)
740e5dd7070Spatrick       Size = Builder.CreateMul(Size, ArraySize);
741e5dd7070Spatrick 
742e5dd7070Spatrick     // Degenerate case: new X[0] does not need an objectsize check.
743e5dd7070Spatrick     llvm::Constant *ConstantSize = dyn_cast<llvm::Constant>(Size);
744e5dd7070Spatrick     if (!ConstantSize || !ConstantSize->isNullValue()) {
745e5dd7070Spatrick       // The glvalue must refer to a large enough storage region.
746e5dd7070Spatrick       // FIXME: If Address Sanitizer is enabled, insert dynamic instrumentation
747e5dd7070Spatrick       //        to check this.
748e5dd7070Spatrick       // FIXME: Get object address space
749e5dd7070Spatrick       llvm::Type *Tys[2] = { IntPtrTy, Int8PtrTy };
750e5dd7070Spatrick       llvm::Function *F = CGM.getIntrinsic(llvm::Intrinsic::objectsize, Tys);
751e5dd7070Spatrick       llvm::Value *Min = Builder.getFalse();
752e5dd7070Spatrick       llvm::Value *NullIsUnknown = Builder.getFalse();
753e5dd7070Spatrick       llvm::Value *Dynamic = Builder.getFalse();
754e5dd7070Spatrick       llvm::Value *CastAddr = Builder.CreateBitCast(Ptr, Int8PtrTy);
755e5dd7070Spatrick       llvm::Value *LargeEnough = Builder.CreateICmpUGE(
756e5dd7070Spatrick           Builder.CreateCall(F, {CastAddr, Min, NullIsUnknown, Dynamic}), Size);
757e5dd7070Spatrick       Checks.push_back(std::make_pair(LargeEnough, SanitizerKind::ObjectSize));
758e5dd7070Spatrick     }
759e5dd7070Spatrick   }
760e5dd7070Spatrick 
761*12c85518Srobert   llvm::MaybeAlign AlignVal;
762e5dd7070Spatrick   llvm::Value *PtrAsInt = nullptr;
763e5dd7070Spatrick 
764e5dd7070Spatrick   if (SanOpts.has(SanitizerKind::Alignment) &&
765e5dd7070Spatrick       !SkippedChecks.has(SanitizerKind::Alignment)) {
766*12c85518Srobert     AlignVal = Alignment.getAsMaybeAlign();
767e5dd7070Spatrick     if (!Ty->isIncompleteType() && !AlignVal)
768ec727ea7Spatrick       AlignVal = CGM.getNaturalTypeAlignment(Ty, nullptr, nullptr,
769ec727ea7Spatrick                                              /*ForPointeeType=*/true)
770*12c85518Srobert                      .getAsMaybeAlign();
771e5dd7070Spatrick 
772e5dd7070Spatrick     // The glvalue must be suitably aligned.
773*12c85518Srobert     if (AlignVal && *AlignVal > llvm::Align(1) &&
774*12c85518Srobert         (!PtrToAlloca || PtrToAlloca->getAlign() < *AlignVal)) {
775e5dd7070Spatrick       PtrAsInt = Builder.CreatePtrToInt(Ptr, IntPtrTy);
776e5dd7070Spatrick       llvm::Value *Align = Builder.CreateAnd(
777*12c85518Srobert           PtrAsInt, llvm::ConstantInt::get(IntPtrTy, AlignVal->value() - 1));
778e5dd7070Spatrick       llvm::Value *Aligned =
779e5dd7070Spatrick           Builder.CreateICmpEQ(Align, llvm::ConstantInt::get(IntPtrTy, 0));
780e5dd7070Spatrick       if (Aligned != True)
781e5dd7070Spatrick         Checks.push_back(std::make_pair(Aligned, SanitizerKind::Alignment));
782e5dd7070Spatrick     }
783e5dd7070Spatrick   }
784e5dd7070Spatrick 
785e5dd7070Spatrick   if (Checks.size() > 0) {
786e5dd7070Spatrick     llvm::Constant *StaticData[] = {
787e5dd7070Spatrick         EmitCheckSourceLocation(Loc), EmitCheckTypeDescriptor(Ty),
788*12c85518Srobert         llvm::ConstantInt::get(Int8Ty, AlignVal ? llvm::Log2(*AlignVal) : 1),
789e5dd7070Spatrick         llvm::ConstantInt::get(Int8Ty, TCK)};
790e5dd7070Spatrick     EmitCheck(Checks, SanitizerHandler::TypeMismatch, StaticData,
791e5dd7070Spatrick               PtrAsInt ? PtrAsInt : Ptr);
792e5dd7070Spatrick   }
793e5dd7070Spatrick 
794e5dd7070Spatrick   // If possible, check that the vptr indicates that there is a subobject of
795e5dd7070Spatrick   // type Ty at offset zero within this object.
796e5dd7070Spatrick   //
797e5dd7070Spatrick   // C++11 [basic.life]p5,6:
798e5dd7070Spatrick   //   [For storage which does not refer to an object within its lifetime]
799e5dd7070Spatrick   //   The program has undefined behavior if:
800e5dd7070Spatrick   //    -- the [pointer or glvalue] is used to access a non-static data member
801e5dd7070Spatrick   //       or call a non-static member function
802e5dd7070Spatrick   if (SanOpts.has(SanitizerKind::Vptr) &&
803e5dd7070Spatrick       !SkippedChecks.has(SanitizerKind::Vptr) && isVptrCheckRequired(TCK, Ty)) {
804e5dd7070Spatrick     // Ensure that the pointer is non-null before loading it. If there is no
805e5dd7070Spatrick     // compile-time guarantee, reuse the run-time null check or emit a new one.
806e5dd7070Spatrick     if (!IsGuaranteedNonNull) {
807e5dd7070Spatrick       if (!IsNonNull)
808e5dd7070Spatrick         IsNonNull = Builder.CreateIsNotNull(Ptr);
809e5dd7070Spatrick       if (!Done)
810e5dd7070Spatrick         Done = createBasicBlock("vptr.null");
811e5dd7070Spatrick       llvm::BasicBlock *VptrNotNull = createBasicBlock("vptr.not.null");
812e5dd7070Spatrick       Builder.CreateCondBr(IsNonNull, VptrNotNull, Done);
813e5dd7070Spatrick       EmitBlock(VptrNotNull);
814e5dd7070Spatrick     }
815e5dd7070Spatrick 
816e5dd7070Spatrick     // Compute a hash of the mangled name of the type.
817e5dd7070Spatrick     //
818e5dd7070Spatrick     // FIXME: This is not guaranteed to be deterministic! Move to a
819e5dd7070Spatrick     //        fingerprinting mechanism once LLVM provides one. For the time
820e5dd7070Spatrick     //        being the implementation happens to be deterministic.
821e5dd7070Spatrick     SmallString<64> MangledName;
822e5dd7070Spatrick     llvm::raw_svector_ostream Out(MangledName);
823e5dd7070Spatrick     CGM.getCXXABI().getMangleContext().mangleCXXRTTI(Ty.getUnqualifiedType(),
824e5dd7070Spatrick                                                      Out);
825e5dd7070Spatrick 
826a9ac8606Spatrick     // Contained in NoSanitizeList based on the mangled type.
827a9ac8606Spatrick     if (!CGM.getContext().getNoSanitizeList().containsType(SanitizerKind::Vptr,
828a9ac8606Spatrick                                                            Out.str())) {
829e5dd7070Spatrick       llvm::hash_code TypeHash = hash_value(Out.str());
830e5dd7070Spatrick 
831e5dd7070Spatrick       // Load the vptr, and compute hash_16_bytes(TypeHash, vptr).
832e5dd7070Spatrick       llvm::Value *Low = llvm::ConstantInt::get(Int64Ty, TypeHash);
833e5dd7070Spatrick       llvm::Type *VPtrTy = llvm::PointerType::get(IntPtrTy, 0);
834*12c85518Srobert       Address VPtrAddr(Builder.CreateBitCast(Ptr, VPtrTy), IntPtrTy,
835*12c85518Srobert                        getPointerAlign());
836e5dd7070Spatrick       llvm::Value *VPtrVal = Builder.CreateLoad(VPtrAddr);
837e5dd7070Spatrick       llvm::Value *High = Builder.CreateZExt(VPtrVal, Int64Ty);
838e5dd7070Spatrick 
839e5dd7070Spatrick       llvm::Value *Hash = emitHash16Bytes(Builder, Low, High);
840e5dd7070Spatrick       Hash = Builder.CreateTrunc(Hash, IntPtrTy);
841e5dd7070Spatrick 
842e5dd7070Spatrick       // Look the hash up in our cache.
843e5dd7070Spatrick       const int CacheSize = 128;
844e5dd7070Spatrick       llvm::Type *HashTable = llvm::ArrayType::get(IntPtrTy, CacheSize);
845e5dd7070Spatrick       llvm::Value *Cache = CGM.CreateRuntimeVariable(HashTable,
846e5dd7070Spatrick                                                      "__ubsan_vptr_type_cache");
847e5dd7070Spatrick       llvm::Value *Slot = Builder.CreateAnd(Hash,
848e5dd7070Spatrick                                             llvm::ConstantInt::get(IntPtrTy,
849e5dd7070Spatrick                                                                    CacheSize-1));
850e5dd7070Spatrick       llvm::Value *Indices[] = { Builder.getInt32(0), Slot };
851a9ac8606Spatrick       llvm::Value *CacheVal = Builder.CreateAlignedLoad(
852a9ac8606Spatrick           IntPtrTy, Builder.CreateInBoundsGEP(HashTable, Cache, Indices),
853e5dd7070Spatrick           getPointerAlign());
854e5dd7070Spatrick 
855e5dd7070Spatrick       // If the hash isn't in the cache, call a runtime handler to perform the
856e5dd7070Spatrick       // hard work of checking whether the vptr is for an object of the right
857e5dd7070Spatrick       // type. This will either fill in the cache and return, or produce a
858e5dd7070Spatrick       // diagnostic.
859e5dd7070Spatrick       llvm::Value *EqualHash = Builder.CreateICmpEQ(CacheVal, Hash);
860e5dd7070Spatrick       llvm::Constant *StaticData[] = {
861e5dd7070Spatrick         EmitCheckSourceLocation(Loc),
862e5dd7070Spatrick         EmitCheckTypeDescriptor(Ty),
863e5dd7070Spatrick         CGM.GetAddrOfRTTIDescriptor(Ty.getUnqualifiedType()),
864e5dd7070Spatrick         llvm::ConstantInt::get(Int8Ty, TCK)
865e5dd7070Spatrick       };
866e5dd7070Spatrick       llvm::Value *DynamicData[] = { Ptr, Hash };
867e5dd7070Spatrick       EmitCheck(std::make_pair(EqualHash, SanitizerKind::Vptr),
868e5dd7070Spatrick                 SanitizerHandler::DynamicTypeCacheMiss, StaticData,
869e5dd7070Spatrick                 DynamicData);
870e5dd7070Spatrick     }
871e5dd7070Spatrick   }
872e5dd7070Spatrick 
873e5dd7070Spatrick   if (Done) {
874e5dd7070Spatrick     Builder.CreateBr(Done);
875e5dd7070Spatrick     EmitBlock(Done);
876e5dd7070Spatrick   }
877e5dd7070Spatrick }
878e5dd7070Spatrick 
LoadPassedObjectSize(const Expr * E,QualType EltTy)879e5dd7070Spatrick llvm::Value *CodeGenFunction::LoadPassedObjectSize(const Expr *E,
880e5dd7070Spatrick                                                    QualType EltTy) {
881e5dd7070Spatrick   ASTContext &C = getContext();
882e5dd7070Spatrick   uint64_t EltSize = C.getTypeSizeInChars(EltTy).getQuantity();
883e5dd7070Spatrick   if (!EltSize)
884e5dd7070Spatrick     return nullptr;
885e5dd7070Spatrick 
886e5dd7070Spatrick   auto *ArrayDeclRef = dyn_cast<DeclRefExpr>(E->IgnoreParenImpCasts());
887e5dd7070Spatrick   if (!ArrayDeclRef)
888e5dd7070Spatrick     return nullptr;
889e5dd7070Spatrick 
890e5dd7070Spatrick   auto *ParamDecl = dyn_cast<ParmVarDecl>(ArrayDeclRef->getDecl());
891e5dd7070Spatrick   if (!ParamDecl)
892e5dd7070Spatrick     return nullptr;
893e5dd7070Spatrick 
894e5dd7070Spatrick   auto *POSAttr = ParamDecl->getAttr<PassObjectSizeAttr>();
895e5dd7070Spatrick   if (!POSAttr)
896e5dd7070Spatrick     return nullptr;
897e5dd7070Spatrick 
898e5dd7070Spatrick   // Don't load the size if it's a lower bound.
899e5dd7070Spatrick   int POSType = POSAttr->getType();
900e5dd7070Spatrick   if (POSType != 0 && POSType != 1)
901e5dd7070Spatrick     return nullptr;
902e5dd7070Spatrick 
903e5dd7070Spatrick   // Find the implicit size parameter.
904e5dd7070Spatrick   auto PassedSizeIt = SizeArguments.find(ParamDecl);
905e5dd7070Spatrick   if (PassedSizeIt == SizeArguments.end())
906e5dd7070Spatrick     return nullptr;
907e5dd7070Spatrick 
908e5dd7070Spatrick   const ImplicitParamDecl *PassedSizeDecl = PassedSizeIt->second;
909e5dd7070Spatrick   assert(LocalDeclMap.count(PassedSizeDecl) && "Passed size not loadable");
910e5dd7070Spatrick   Address AddrOfSize = LocalDeclMap.find(PassedSizeDecl)->second;
911e5dd7070Spatrick   llvm::Value *SizeInBytes = EmitLoadOfScalar(AddrOfSize, /*Volatile=*/false,
912e5dd7070Spatrick                                               C.getSizeType(), E->getExprLoc());
913e5dd7070Spatrick   llvm::Value *SizeOfElement =
914e5dd7070Spatrick       llvm::ConstantInt::get(SizeInBytes->getType(), EltSize);
915e5dd7070Spatrick   return Builder.CreateUDiv(SizeInBytes, SizeOfElement);
916e5dd7070Spatrick }
917e5dd7070Spatrick 
918e5dd7070Spatrick /// If Base is known to point to the start of an array, return the length of
919e5dd7070Spatrick /// that array. Return 0 if the length cannot be determined.
getArrayIndexingBound(CodeGenFunction & CGF,const Expr * Base,QualType & IndexedType,LangOptions::StrictFlexArraysLevelKind StrictFlexArraysLevel)920*12c85518Srobert static llvm::Value *getArrayIndexingBound(CodeGenFunction &CGF,
921*12c85518Srobert                                           const Expr *Base,
922*12c85518Srobert                                           QualType &IndexedType,
923*12c85518Srobert                                           LangOptions::StrictFlexArraysLevelKind
924*12c85518Srobert                                           StrictFlexArraysLevel) {
925e5dd7070Spatrick   // For the vector indexing extension, the bound is the number of elements.
926e5dd7070Spatrick   if (const VectorType *VT = Base->getType()->getAs<VectorType>()) {
927e5dd7070Spatrick     IndexedType = Base->getType();
928e5dd7070Spatrick     return CGF.Builder.getInt32(VT->getNumElements());
929e5dd7070Spatrick   }
930e5dd7070Spatrick 
931e5dd7070Spatrick   Base = Base->IgnoreParens();
932e5dd7070Spatrick 
933e5dd7070Spatrick   if (const auto *CE = dyn_cast<CastExpr>(Base)) {
934e5dd7070Spatrick     if (CE->getCastKind() == CK_ArrayToPointerDecay &&
935*12c85518Srobert         !CE->getSubExpr()->isFlexibleArrayMemberLike(CGF.getContext(),
936*12c85518Srobert                                                      StrictFlexArraysLevel)) {
937e5dd7070Spatrick       IndexedType = CE->getSubExpr()->getType();
938e5dd7070Spatrick       const ArrayType *AT = IndexedType->castAsArrayTypeUnsafe();
939e5dd7070Spatrick       if (const auto *CAT = dyn_cast<ConstantArrayType>(AT))
940e5dd7070Spatrick         return CGF.Builder.getInt(CAT->getSize());
941e5dd7070Spatrick       else if (const auto *VAT = dyn_cast<VariableArrayType>(AT))
942e5dd7070Spatrick         return CGF.getVLASize(VAT).NumElts;
943e5dd7070Spatrick       // Ignore pass_object_size here. It's not applicable on decayed pointers.
944e5dd7070Spatrick     }
945e5dd7070Spatrick   }
946e5dd7070Spatrick 
947e5dd7070Spatrick   QualType EltTy{Base->getType()->getPointeeOrArrayElementType(), 0};
948e5dd7070Spatrick   if (llvm::Value *POS = CGF.LoadPassedObjectSize(Base, EltTy)) {
949e5dd7070Spatrick     IndexedType = Base->getType();
950e5dd7070Spatrick     return POS;
951e5dd7070Spatrick   }
952e5dd7070Spatrick 
953e5dd7070Spatrick   return nullptr;
954e5dd7070Spatrick }
955e5dd7070Spatrick 
EmitBoundsCheck(const Expr * E,const Expr * Base,llvm::Value * Index,QualType IndexType,bool Accessed)956e5dd7070Spatrick void CodeGenFunction::EmitBoundsCheck(const Expr *E, const Expr *Base,
957e5dd7070Spatrick                                       llvm::Value *Index, QualType IndexType,
958e5dd7070Spatrick                                       bool Accessed) {
959e5dd7070Spatrick   assert(SanOpts.has(SanitizerKind::ArrayBounds) &&
960e5dd7070Spatrick          "should not be called unless adding bounds checks");
961e5dd7070Spatrick   SanitizerScope SanScope(this);
962e5dd7070Spatrick 
963*12c85518Srobert   const LangOptions::StrictFlexArraysLevelKind StrictFlexArraysLevel =
964*12c85518Srobert     getLangOpts().getStrictFlexArraysLevel();
965*12c85518Srobert 
966e5dd7070Spatrick   QualType IndexedType;
967*12c85518Srobert   llvm::Value *Bound =
968*12c85518Srobert       getArrayIndexingBound(*this, Base, IndexedType, StrictFlexArraysLevel);
969e5dd7070Spatrick   if (!Bound)
970e5dd7070Spatrick     return;
971e5dd7070Spatrick 
972e5dd7070Spatrick   bool IndexSigned = IndexType->isSignedIntegerOrEnumerationType();
973e5dd7070Spatrick   llvm::Value *IndexVal = Builder.CreateIntCast(Index, SizeTy, IndexSigned);
974e5dd7070Spatrick   llvm::Value *BoundVal = Builder.CreateIntCast(Bound, SizeTy, false);
975e5dd7070Spatrick 
976e5dd7070Spatrick   llvm::Constant *StaticData[] = {
977e5dd7070Spatrick     EmitCheckSourceLocation(E->getExprLoc()),
978e5dd7070Spatrick     EmitCheckTypeDescriptor(IndexedType),
979e5dd7070Spatrick     EmitCheckTypeDescriptor(IndexType)
980e5dd7070Spatrick   };
981e5dd7070Spatrick   llvm::Value *Check = Accessed ? Builder.CreateICmpULT(IndexVal, BoundVal)
982e5dd7070Spatrick                                 : Builder.CreateICmpULE(IndexVal, BoundVal);
983e5dd7070Spatrick   EmitCheck(std::make_pair(Check, SanitizerKind::ArrayBounds),
984e5dd7070Spatrick             SanitizerHandler::OutOfBounds, StaticData, Index);
985e5dd7070Spatrick }
986e5dd7070Spatrick 
987e5dd7070Spatrick 
988e5dd7070Spatrick CodeGenFunction::ComplexPairTy CodeGenFunction::
EmitComplexPrePostIncDec(const UnaryOperator * E,LValue LV,bool isInc,bool isPre)989e5dd7070Spatrick EmitComplexPrePostIncDec(const UnaryOperator *E, LValue LV,
990e5dd7070Spatrick                          bool isInc, bool isPre) {
991e5dd7070Spatrick   ComplexPairTy InVal = EmitLoadOfComplex(LV, E->getExprLoc());
992e5dd7070Spatrick 
993e5dd7070Spatrick   llvm::Value *NextVal;
994e5dd7070Spatrick   if (isa<llvm::IntegerType>(InVal.first->getType())) {
995e5dd7070Spatrick     uint64_t AmountVal = isInc ? 1 : -1;
996e5dd7070Spatrick     NextVal = llvm::ConstantInt::get(InVal.first->getType(), AmountVal, true);
997e5dd7070Spatrick 
998e5dd7070Spatrick     // Add the inc/dec to the real part.
999e5dd7070Spatrick     NextVal = Builder.CreateAdd(InVal.first, NextVal, isInc ? "inc" : "dec");
1000e5dd7070Spatrick   } else {
1001e5dd7070Spatrick     QualType ElemTy = E->getType()->castAs<ComplexType>()->getElementType();
1002e5dd7070Spatrick     llvm::APFloat FVal(getContext().getFloatTypeSemantics(ElemTy), 1);
1003e5dd7070Spatrick     if (!isInc)
1004e5dd7070Spatrick       FVal.changeSign();
1005e5dd7070Spatrick     NextVal = llvm::ConstantFP::get(getLLVMContext(), FVal);
1006e5dd7070Spatrick 
1007e5dd7070Spatrick     // Add the inc/dec to the real part.
1008e5dd7070Spatrick     NextVal = Builder.CreateFAdd(InVal.first, NextVal, isInc ? "inc" : "dec");
1009e5dd7070Spatrick   }
1010e5dd7070Spatrick 
1011e5dd7070Spatrick   ComplexPairTy IncVal(NextVal, InVal.second);
1012e5dd7070Spatrick 
1013e5dd7070Spatrick   // Store the updated result through the lvalue.
1014e5dd7070Spatrick   EmitStoreOfComplex(IncVal, LV, /*init*/ false);
1015e5dd7070Spatrick   if (getLangOpts().OpenMP)
1016e5dd7070Spatrick     CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(*this,
1017e5dd7070Spatrick                                                               E->getSubExpr());
1018e5dd7070Spatrick 
1019e5dd7070Spatrick   // If this is a postinc, return the value read from memory, otherwise use the
1020e5dd7070Spatrick   // updated value.
1021e5dd7070Spatrick   return isPre ? IncVal : InVal;
1022e5dd7070Spatrick }
1023e5dd7070Spatrick 
EmitExplicitCastExprType(const ExplicitCastExpr * E,CodeGenFunction * CGF)1024e5dd7070Spatrick void CodeGenModule::EmitExplicitCastExprType(const ExplicitCastExpr *E,
1025e5dd7070Spatrick                                              CodeGenFunction *CGF) {
1026e5dd7070Spatrick   // Bind VLAs in the cast type.
1027e5dd7070Spatrick   if (CGF && E->getType()->isVariablyModifiedType())
1028e5dd7070Spatrick     CGF->EmitVariablyModifiedType(E->getType());
1029e5dd7070Spatrick 
1030e5dd7070Spatrick   if (CGDebugInfo *DI = getModuleDebugInfo())
1031e5dd7070Spatrick     DI->EmitExplicitCastType(E->getType());
1032e5dd7070Spatrick }
1033e5dd7070Spatrick 
1034e5dd7070Spatrick //===----------------------------------------------------------------------===//
1035e5dd7070Spatrick //                         LValue Expression Emission
1036e5dd7070Spatrick //===----------------------------------------------------------------------===//
1037e5dd7070Spatrick 
1038e5dd7070Spatrick /// EmitPointerWithAlignment - Given an expression of pointer type, try to
1039e5dd7070Spatrick /// derive a more accurate bound on the alignment of the pointer.
EmitPointerWithAlignment(const Expr * E,LValueBaseInfo * BaseInfo,TBAAAccessInfo * TBAAInfo)1040e5dd7070Spatrick Address CodeGenFunction::EmitPointerWithAlignment(const Expr *E,
1041e5dd7070Spatrick                                                   LValueBaseInfo *BaseInfo,
1042e5dd7070Spatrick                                                   TBAAAccessInfo *TBAAInfo) {
1043e5dd7070Spatrick   // We allow this with ObjC object pointers because of fragile ABIs.
1044e5dd7070Spatrick   assert(E->getType()->isPointerType() ||
1045e5dd7070Spatrick          E->getType()->isObjCObjectPointerType());
1046e5dd7070Spatrick   E = E->IgnoreParens();
1047e5dd7070Spatrick 
1048e5dd7070Spatrick   // Casts:
1049e5dd7070Spatrick   if (const CastExpr *CE = dyn_cast<CastExpr>(E)) {
1050e5dd7070Spatrick     if (const auto *ECE = dyn_cast<ExplicitCastExpr>(CE))
1051e5dd7070Spatrick       CGM.EmitExplicitCastExprType(ECE, this);
1052e5dd7070Spatrick 
1053e5dd7070Spatrick     switch (CE->getCastKind()) {
1054e5dd7070Spatrick     // Non-converting casts (but not C's implicit conversion from void*).
1055e5dd7070Spatrick     case CK_BitCast:
1056e5dd7070Spatrick     case CK_NoOp:
1057e5dd7070Spatrick     case CK_AddressSpaceConversion:
1058e5dd7070Spatrick       if (auto PtrTy = CE->getSubExpr()->getType()->getAs<PointerType>()) {
1059e5dd7070Spatrick         if (PtrTy->getPointeeType()->isVoidType())
1060e5dd7070Spatrick           break;
1061e5dd7070Spatrick 
1062e5dd7070Spatrick         LValueBaseInfo InnerBaseInfo;
1063e5dd7070Spatrick         TBAAAccessInfo InnerTBAAInfo;
1064e5dd7070Spatrick         Address Addr = EmitPointerWithAlignment(CE->getSubExpr(),
1065e5dd7070Spatrick                                                 &InnerBaseInfo,
1066e5dd7070Spatrick                                                 &InnerTBAAInfo);
1067e5dd7070Spatrick         if (BaseInfo) *BaseInfo = InnerBaseInfo;
1068e5dd7070Spatrick         if (TBAAInfo) *TBAAInfo = InnerTBAAInfo;
1069e5dd7070Spatrick 
1070e5dd7070Spatrick         if (isa<ExplicitCastExpr>(CE)) {
1071e5dd7070Spatrick           LValueBaseInfo TargetTypeBaseInfo;
1072e5dd7070Spatrick           TBAAAccessInfo TargetTypeTBAAInfo;
1073ec727ea7Spatrick           CharUnits Align = CGM.getNaturalPointeeTypeAlignment(
1074ec727ea7Spatrick               E->getType(), &TargetTypeBaseInfo, &TargetTypeTBAAInfo);
1075e5dd7070Spatrick           if (TBAAInfo)
1076e5dd7070Spatrick             *TBAAInfo = CGM.mergeTBAAInfoForCast(*TBAAInfo,
1077e5dd7070Spatrick                                                  TargetTypeTBAAInfo);
1078e5dd7070Spatrick           // If the source l-value is opaque, honor the alignment of the
1079e5dd7070Spatrick           // casted-to type.
1080e5dd7070Spatrick           if (InnerBaseInfo.getAlignmentSource() != AlignmentSource::Decl) {
1081e5dd7070Spatrick             if (BaseInfo)
1082e5dd7070Spatrick               BaseInfo->mergeForCast(TargetTypeBaseInfo);
1083*12c85518Srobert             Addr = Address(Addr.getPointer(), Addr.getElementType(), Align);
1084e5dd7070Spatrick           }
1085e5dd7070Spatrick         }
1086e5dd7070Spatrick 
1087e5dd7070Spatrick         if (SanOpts.has(SanitizerKind::CFIUnrelatedCast) &&
1088e5dd7070Spatrick             CE->getCastKind() == CK_BitCast) {
1089e5dd7070Spatrick           if (auto PT = E->getType()->getAs<PointerType>())
1090*12c85518Srobert             EmitVTablePtrCheckForCast(PT->getPointeeType(), Addr,
1091e5dd7070Spatrick                                       /*MayBeNull=*/true,
1092e5dd7070Spatrick                                       CodeGenFunction::CFITCK_UnrelatedCast,
1093e5dd7070Spatrick                                       CE->getBeginLoc());
1094e5dd7070Spatrick         }
1095*12c85518Srobert 
1096*12c85518Srobert         llvm::Type *ElemTy = ConvertTypeForMem(E->getType()->getPointeeType());
1097*12c85518Srobert         Addr = Builder.CreateElementBitCast(Addr, ElemTy);
1098*12c85518Srobert         if (CE->getCastKind() == CK_AddressSpaceConversion)
1099*12c85518Srobert           Addr = Builder.CreateAddrSpaceCast(Addr, ConvertType(E->getType()));
1100*12c85518Srobert         return Addr;
1101e5dd7070Spatrick       }
1102e5dd7070Spatrick       break;
1103e5dd7070Spatrick 
1104e5dd7070Spatrick     // Array-to-pointer decay.
1105e5dd7070Spatrick     case CK_ArrayToPointerDecay:
1106e5dd7070Spatrick       return EmitArrayToPointerDecay(CE->getSubExpr(), BaseInfo, TBAAInfo);
1107e5dd7070Spatrick 
1108e5dd7070Spatrick     // Derived-to-base conversions.
1109e5dd7070Spatrick     case CK_UncheckedDerivedToBase:
1110e5dd7070Spatrick     case CK_DerivedToBase: {
1111e5dd7070Spatrick       // TODO: Support accesses to members of base classes in TBAA. For now, we
1112e5dd7070Spatrick       // conservatively pretend that the complete object is of the base class
1113e5dd7070Spatrick       // type.
1114e5dd7070Spatrick       if (TBAAInfo)
1115e5dd7070Spatrick         *TBAAInfo = CGM.getTBAAAccessInfo(E->getType());
1116e5dd7070Spatrick       Address Addr = EmitPointerWithAlignment(CE->getSubExpr(), BaseInfo);
1117e5dd7070Spatrick       auto Derived = CE->getSubExpr()->getType()->getPointeeCXXRecordDecl();
1118e5dd7070Spatrick       return GetAddressOfBaseClass(Addr, Derived,
1119e5dd7070Spatrick                                    CE->path_begin(), CE->path_end(),
1120e5dd7070Spatrick                                    ShouldNullCheckClassCastValue(CE),
1121e5dd7070Spatrick                                    CE->getExprLoc());
1122e5dd7070Spatrick     }
1123e5dd7070Spatrick 
1124e5dd7070Spatrick     // TODO: Is there any reason to treat base-to-derived conversions
1125e5dd7070Spatrick     // specially?
1126e5dd7070Spatrick     default:
1127e5dd7070Spatrick       break;
1128e5dd7070Spatrick     }
1129e5dd7070Spatrick   }
1130e5dd7070Spatrick 
1131e5dd7070Spatrick   // Unary &.
1132e5dd7070Spatrick   if (const UnaryOperator *UO = dyn_cast<UnaryOperator>(E)) {
1133e5dd7070Spatrick     if (UO->getOpcode() == UO_AddrOf) {
1134e5dd7070Spatrick       LValue LV = EmitLValue(UO->getSubExpr());
1135e5dd7070Spatrick       if (BaseInfo) *BaseInfo = LV.getBaseInfo();
1136e5dd7070Spatrick       if (TBAAInfo) *TBAAInfo = LV.getTBAAInfo();
1137e5dd7070Spatrick       return LV.getAddress(*this);
1138e5dd7070Spatrick     }
1139e5dd7070Spatrick   }
1140e5dd7070Spatrick 
1141*12c85518Srobert   // std::addressof and variants.
1142*12c85518Srobert   if (auto *Call = dyn_cast<CallExpr>(E)) {
1143*12c85518Srobert     switch (Call->getBuiltinCallee()) {
1144*12c85518Srobert     default:
1145*12c85518Srobert       break;
1146*12c85518Srobert     case Builtin::BIaddressof:
1147*12c85518Srobert     case Builtin::BI__addressof:
1148*12c85518Srobert     case Builtin::BI__builtin_addressof: {
1149*12c85518Srobert       LValue LV = EmitLValue(Call->getArg(0));
1150*12c85518Srobert       if (BaseInfo) *BaseInfo = LV.getBaseInfo();
1151*12c85518Srobert       if (TBAAInfo) *TBAAInfo = LV.getTBAAInfo();
1152*12c85518Srobert       return LV.getAddress(*this);
1153*12c85518Srobert     }
1154*12c85518Srobert     }
1155*12c85518Srobert   }
1156*12c85518Srobert 
1157e5dd7070Spatrick   // TODO: conditional operators, comma.
1158e5dd7070Spatrick 
1159e5dd7070Spatrick   // Otherwise, use the alignment of the type.
1160ec727ea7Spatrick   CharUnits Align =
1161ec727ea7Spatrick       CGM.getNaturalPointeeTypeAlignment(E->getType(), BaseInfo, TBAAInfo);
1162*12c85518Srobert   llvm::Type *ElemTy = ConvertTypeForMem(E->getType()->getPointeeType());
1163*12c85518Srobert   return Address(EmitScalarExpr(E), ElemTy, Align);
1164e5dd7070Spatrick }
1165e5dd7070Spatrick 
EmitNonNullRValueCheck(RValue RV,QualType T)1166a9ac8606Spatrick llvm::Value *CodeGenFunction::EmitNonNullRValueCheck(RValue RV, QualType T) {
1167a9ac8606Spatrick   llvm::Value *V = RV.getScalarVal();
1168a9ac8606Spatrick   if (auto MPT = T->getAs<MemberPointerType>())
1169a9ac8606Spatrick     return CGM.getCXXABI().EmitMemberPointerIsNotNull(*this, V, MPT);
1170a9ac8606Spatrick   return Builder.CreateICmpNE(V, llvm::Constant::getNullValue(V->getType()));
1171a9ac8606Spatrick }
1172a9ac8606Spatrick 
GetUndefRValue(QualType Ty)1173e5dd7070Spatrick RValue CodeGenFunction::GetUndefRValue(QualType Ty) {
1174e5dd7070Spatrick   if (Ty->isVoidType())
1175e5dd7070Spatrick     return RValue::get(nullptr);
1176e5dd7070Spatrick 
1177e5dd7070Spatrick   switch (getEvaluationKind(Ty)) {
1178e5dd7070Spatrick   case TEK_Complex: {
1179e5dd7070Spatrick     llvm::Type *EltTy =
1180e5dd7070Spatrick       ConvertType(Ty->castAs<ComplexType>()->getElementType());
1181e5dd7070Spatrick     llvm::Value *U = llvm::UndefValue::get(EltTy);
1182e5dd7070Spatrick     return RValue::getComplex(std::make_pair(U, U));
1183e5dd7070Spatrick   }
1184e5dd7070Spatrick 
1185e5dd7070Spatrick   // If this is a use of an undefined aggregate type, the aggregate must have an
1186e5dd7070Spatrick   // identifiable address.  Just because the contents of the value are undefined
1187e5dd7070Spatrick   // doesn't mean that the address can't be taken and compared.
1188e5dd7070Spatrick   case TEK_Aggregate: {
1189e5dd7070Spatrick     Address DestPtr = CreateMemTemp(Ty, "undef.agg.tmp");
1190e5dd7070Spatrick     return RValue::getAggregate(DestPtr);
1191e5dd7070Spatrick   }
1192e5dd7070Spatrick 
1193e5dd7070Spatrick   case TEK_Scalar:
1194e5dd7070Spatrick     return RValue::get(llvm::UndefValue::get(ConvertType(Ty)));
1195e5dd7070Spatrick   }
1196e5dd7070Spatrick   llvm_unreachable("bad evaluation kind");
1197e5dd7070Spatrick }
1198e5dd7070Spatrick 
EmitUnsupportedRValue(const Expr * E,const char * Name)1199e5dd7070Spatrick RValue CodeGenFunction::EmitUnsupportedRValue(const Expr *E,
1200e5dd7070Spatrick                                               const char *Name) {
1201e5dd7070Spatrick   ErrorUnsupported(E, Name);
1202e5dd7070Spatrick   return GetUndefRValue(E->getType());
1203e5dd7070Spatrick }
1204e5dd7070Spatrick 
EmitUnsupportedLValue(const Expr * E,const char * Name)1205e5dd7070Spatrick LValue CodeGenFunction::EmitUnsupportedLValue(const Expr *E,
1206e5dd7070Spatrick                                               const char *Name) {
1207e5dd7070Spatrick   ErrorUnsupported(E, Name);
1208*12c85518Srobert   llvm::Type *ElTy = ConvertType(E->getType());
1209*12c85518Srobert   llvm::Type *Ty = llvm::PointerType::getUnqual(ElTy);
1210*12c85518Srobert   return MakeAddrLValue(
1211*12c85518Srobert       Address(llvm::UndefValue::get(Ty), ElTy, CharUnits::One()), E->getType());
1212e5dd7070Spatrick }
1213e5dd7070Spatrick 
IsWrappedCXXThis(const Expr * Obj)1214e5dd7070Spatrick bool CodeGenFunction::IsWrappedCXXThis(const Expr *Obj) {
1215e5dd7070Spatrick   const Expr *Base = Obj;
1216e5dd7070Spatrick   while (!isa<CXXThisExpr>(Base)) {
1217e5dd7070Spatrick     // The result of a dynamic_cast can be null.
1218e5dd7070Spatrick     if (isa<CXXDynamicCastExpr>(Base))
1219e5dd7070Spatrick       return false;
1220e5dd7070Spatrick 
1221e5dd7070Spatrick     if (const auto *CE = dyn_cast<CastExpr>(Base)) {
1222e5dd7070Spatrick       Base = CE->getSubExpr();
1223e5dd7070Spatrick     } else if (const auto *PE = dyn_cast<ParenExpr>(Base)) {
1224e5dd7070Spatrick       Base = PE->getSubExpr();
1225e5dd7070Spatrick     } else if (const auto *UO = dyn_cast<UnaryOperator>(Base)) {
1226e5dd7070Spatrick       if (UO->getOpcode() == UO_Extension)
1227e5dd7070Spatrick         Base = UO->getSubExpr();
1228e5dd7070Spatrick       else
1229e5dd7070Spatrick         return false;
1230e5dd7070Spatrick     } else {
1231e5dd7070Spatrick       return false;
1232e5dd7070Spatrick     }
1233e5dd7070Spatrick   }
1234e5dd7070Spatrick   return true;
1235e5dd7070Spatrick }
1236e5dd7070Spatrick 
EmitCheckedLValue(const Expr * E,TypeCheckKind TCK)1237e5dd7070Spatrick LValue CodeGenFunction::EmitCheckedLValue(const Expr *E, TypeCheckKind TCK) {
1238e5dd7070Spatrick   LValue LV;
1239e5dd7070Spatrick   if (SanOpts.has(SanitizerKind::ArrayBounds) && isa<ArraySubscriptExpr>(E))
1240e5dd7070Spatrick     LV = EmitArraySubscriptExpr(cast<ArraySubscriptExpr>(E), /*Accessed*/true);
1241e5dd7070Spatrick   else
1242e5dd7070Spatrick     LV = EmitLValue(E);
1243e5dd7070Spatrick   if (!isa<DeclRefExpr>(E) && !LV.isBitField() && LV.isSimple()) {
1244e5dd7070Spatrick     SanitizerSet SkippedChecks;
1245e5dd7070Spatrick     if (const auto *ME = dyn_cast<MemberExpr>(E)) {
1246e5dd7070Spatrick       bool IsBaseCXXThis = IsWrappedCXXThis(ME->getBase());
1247e5dd7070Spatrick       if (IsBaseCXXThis)
1248e5dd7070Spatrick         SkippedChecks.set(SanitizerKind::Alignment, true);
1249e5dd7070Spatrick       if (IsBaseCXXThis || isa<DeclRefExpr>(ME->getBase()))
1250e5dd7070Spatrick         SkippedChecks.set(SanitizerKind::Null, true);
1251e5dd7070Spatrick     }
1252e5dd7070Spatrick     EmitTypeCheck(TCK, E->getExprLoc(), LV.getPointer(*this), E->getType(),
1253e5dd7070Spatrick                   LV.getAlignment(), SkippedChecks);
1254e5dd7070Spatrick   }
1255e5dd7070Spatrick   return LV;
1256e5dd7070Spatrick }
1257e5dd7070Spatrick 
1258e5dd7070Spatrick /// EmitLValue - Emit code to compute a designator that specifies the location
1259e5dd7070Spatrick /// of the expression.
1260e5dd7070Spatrick ///
1261e5dd7070Spatrick /// This can return one of two things: a simple address or a bitfield reference.
1262e5dd7070Spatrick /// In either case, the LLVM Value* in the LValue structure is guaranteed to be
1263e5dd7070Spatrick /// an LLVM pointer type.
1264e5dd7070Spatrick ///
1265e5dd7070Spatrick /// If this returns a bitfield reference, nothing about the pointee type of the
1266e5dd7070Spatrick /// LLVM value is known: For example, it may not be a pointer to an integer.
1267e5dd7070Spatrick ///
1268e5dd7070Spatrick /// If this returns a normal address, and if the lvalue's C type is fixed size,
1269e5dd7070Spatrick /// this method guarantees that the returned pointer type will point to an LLVM
1270e5dd7070Spatrick /// type of the same size of the lvalue's type.  If the lvalue has a variable
1271e5dd7070Spatrick /// length type, this is not possible.
1272e5dd7070Spatrick ///
EmitLValue(const Expr * E)1273e5dd7070Spatrick LValue CodeGenFunction::EmitLValue(const Expr *E) {
1274e5dd7070Spatrick   ApplyDebugLocation DL(*this, E);
1275e5dd7070Spatrick   switch (E->getStmtClass()) {
1276e5dd7070Spatrick   default: return EmitUnsupportedLValue(E, "l-value expression");
1277e5dd7070Spatrick 
1278e5dd7070Spatrick   case Expr::ObjCPropertyRefExprClass:
1279e5dd7070Spatrick     llvm_unreachable("cannot emit a property reference directly");
1280e5dd7070Spatrick 
1281e5dd7070Spatrick   case Expr::ObjCSelectorExprClass:
1282e5dd7070Spatrick     return EmitObjCSelectorLValue(cast<ObjCSelectorExpr>(E));
1283e5dd7070Spatrick   case Expr::ObjCIsaExprClass:
1284e5dd7070Spatrick     return EmitObjCIsaExpr(cast<ObjCIsaExpr>(E));
1285e5dd7070Spatrick   case Expr::BinaryOperatorClass:
1286e5dd7070Spatrick     return EmitBinaryOperatorLValue(cast<BinaryOperator>(E));
1287e5dd7070Spatrick   case Expr::CompoundAssignOperatorClass: {
1288e5dd7070Spatrick     QualType Ty = E->getType();
1289e5dd7070Spatrick     if (const AtomicType *AT = Ty->getAs<AtomicType>())
1290e5dd7070Spatrick       Ty = AT->getValueType();
1291e5dd7070Spatrick     if (!Ty->isAnyComplexType())
1292e5dd7070Spatrick       return EmitCompoundAssignmentLValue(cast<CompoundAssignOperator>(E));
1293e5dd7070Spatrick     return EmitComplexCompoundAssignmentLValue(cast<CompoundAssignOperator>(E));
1294e5dd7070Spatrick   }
1295e5dd7070Spatrick   case Expr::CallExprClass:
1296e5dd7070Spatrick   case Expr::CXXMemberCallExprClass:
1297e5dd7070Spatrick   case Expr::CXXOperatorCallExprClass:
1298e5dd7070Spatrick   case Expr::UserDefinedLiteralClass:
1299e5dd7070Spatrick     return EmitCallExprLValue(cast<CallExpr>(E));
1300e5dd7070Spatrick   case Expr::CXXRewrittenBinaryOperatorClass:
1301e5dd7070Spatrick     return EmitLValue(cast<CXXRewrittenBinaryOperator>(E)->getSemanticForm());
1302e5dd7070Spatrick   case Expr::VAArgExprClass:
1303e5dd7070Spatrick     return EmitVAArgExprLValue(cast<VAArgExpr>(E));
1304e5dd7070Spatrick   case Expr::DeclRefExprClass:
1305e5dd7070Spatrick     return EmitDeclRefLValue(cast<DeclRefExpr>(E));
1306ec727ea7Spatrick   case Expr::ConstantExprClass: {
1307ec727ea7Spatrick     const ConstantExpr *CE = cast<ConstantExpr>(E);
1308ec727ea7Spatrick     if (llvm::Value *Result = ConstantEmitter(*this).tryEmitConstantExpr(CE)) {
1309ec727ea7Spatrick       QualType RetType = cast<CallExpr>(CE->getSubExpr()->IgnoreImplicit())
1310*12c85518Srobert                              ->getCallReturnType(getContext())
1311*12c85518Srobert                              ->getPointeeType();
1312ec727ea7Spatrick       return MakeNaturalAlignAddrLValue(Result, RetType);
1313ec727ea7Spatrick     }
1314e5dd7070Spatrick     return EmitLValue(cast<ConstantExpr>(E)->getSubExpr());
1315ec727ea7Spatrick   }
1316e5dd7070Spatrick   case Expr::ParenExprClass:
1317e5dd7070Spatrick     return EmitLValue(cast<ParenExpr>(E)->getSubExpr());
1318e5dd7070Spatrick   case Expr::GenericSelectionExprClass:
1319e5dd7070Spatrick     return EmitLValue(cast<GenericSelectionExpr>(E)->getResultExpr());
1320e5dd7070Spatrick   case Expr::PredefinedExprClass:
1321e5dd7070Spatrick     return EmitPredefinedLValue(cast<PredefinedExpr>(E));
1322e5dd7070Spatrick   case Expr::StringLiteralClass:
1323e5dd7070Spatrick     return EmitStringLiteralLValue(cast<StringLiteral>(E));
1324e5dd7070Spatrick   case Expr::ObjCEncodeExprClass:
1325e5dd7070Spatrick     return EmitObjCEncodeExprLValue(cast<ObjCEncodeExpr>(E));
1326e5dd7070Spatrick   case Expr::PseudoObjectExprClass:
1327e5dd7070Spatrick     return EmitPseudoObjectLValue(cast<PseudoObjectExpr>(E));
1328e5dd7070Spatrick   case Expr::InitListExprClass:
1329e5dd7070Spatrick     return EmitInitListLValue(cast<InitListExpr>(E));
1330e5dd7070Spatrick   case Expr::CXXTemporaryObjectExprClass:
1331e5dd7070Spatrick   case Expr::CXXConstructExprClass:
1332e5dd7070Spatrick     return EmitCXXConstructLValue(cast<CXXConstructExpr>(E));
1333e5dd7070Spatrick   case Expr::CXXBindTemporaryExprClass:
1334e5dd7070Spatrick     return EmitCXXBindTemporaryLValue(cast<CXXBindTemporaryExpr>(E));
1335e5dd7070Spatrick   case Expr::CXXUuidofExprClass:
1336e5dd7070Spatrick     return EmitCXXUuidofLValue(cast<CXXUuidofExpr>(E));
1337e5dd7070Spatrick   case Expr::LambdaExprClass:
1338e5dd7070Spatrick     return EmitAggExprToLValue(E);
1339e5dd7070Spatrick 
1340e5dd7070Spatrick   case Expr::ExprWithCleanupsClass: {
1341e5dd7070Spatrick     const auto *cleanups = cast<ExprWithCleanups>(E);
1342e5dd7070Spatrick     RunCleanupsScope Scope(*this);
1343e5dd7070Spatrick     LValue LV = EmitLValue(cleanups->getSubExpr());
1344e5dd7070Spatrick     if (LV.isSimple()) {
1345e5dd7070Spatrick       // Defend against branches out of gnu statement expressions surrounded by
1346e5dd7070Spatrick       // cleanups.
1347*12c85518Srobert       Address Addr = LV.getAddress(*this);
1348*12c85518Srobert       llvm::Value *V = Addr.getPointer();
1349e5dd7070Spatrick       Scope.ForceCleanup({&V});
1350*12c85518Srobert       return LValue::MakeAddr(Addr.withPointer(V), LV.getType(), getContext(),
1351*12c85518Srobert                               LV.getBaseInfo(), LV.getTBAAInfo());
1352e5dd7070Spatrick     }
1353e5dd7070Spatrick     // FIXME: Is it possible to create an ExprWithCleanups that produces a
1354e5dd7070Spatrick     // bitfield lvalue or some other non-simple lvalue?
1355e5dd7070Spatrick     return LV;
1356e5dd7070Spatrick   }
1357e5dd7070Spatrick 
1358e5dd7070Spatrick   case Expr::CXXDefaultArgExprClass: {
1359e5dd7070Spatrick     auto *DAE = cast<CXXDefaultArgExpr>(E);
1360e5dd7070Spatrick     CXXDefaultArgExprScope Scope(*this, DAE);
1361e5dd7070Spatrick     return EmitLValue(DAE->getExpr());
1362e5dd7070Spatrick   }
1363e5dd7070Spatrick   case Expr::CXXDefaultInitExprClass: {
1364e5dd7070Spatrick     auto *DIE = cast<CXXDefaultInitExpr>(E);
1365e5dd7070Spatrick     CXXDefaultInitExprScope Scope(*this, DIE);
1366e5dd7070Spatrick     return EmitLValue(DIE->getExpr());
1367e5dd7070Spatrick   }
1368e5dd7070Spatrick   case Expr::CXXTypeidExprClass:
1369e5dd7070Spatrick     return EmitCXXTypeidLValue(cast<CXXTypeidExpr>(E));
1370e5dd7070Spatrick 
1371e5dd7070Spatrick   case Expr::ObjCMessageExprClass:
1372e5dd7070Spatrick     return EmitObjCMessageExprLValue(cast<ObjCMessageExpr>(E));
1373e5dd7070Spatrick   case Expr::ObjCIvarRefExprClass:
1374e5dd7070Spatrick     return EmitObjCIvarRefLValue(cast<ObjCIvarRefExpr>(E));
1375e5dd7070Spatrick   case Expr::StmtExprClass:
1376e5dd7070Spatrick     return EmitStmtExprLValue(cast<StmtExpr>(E));
1377e5dd7070Spatrick   case Expr::UnaryOperatorClass:
1378e5dd7070Spatrick     return EmitUnaryOpLValue(cast<UnaryOperator>(E));
1379e5dd7070Spatrick   case Expr::ArraySubscriptExprClass:
1380e5dd7070Spatrick     return EmitArraySubscriptExpr(cast<ArraySubscriptExpr>(E));
1381ec727ea7Spatrick   case Expr::MatrixSubscriptExprClass:
1382ec727ea7Spatrick     return EmitMatrixSubscriptExpr(cast<MatrixSubscriptExpr>(E));
1383e5dd7070Spatrick   case Expr::OMPArraySectionExprClass:
1384e5dd7070Spatrick     return EmitOMPArraySectionExpr(cast<OMPArraySectionExpr>(E));
1385e5dd7070Spatrick   case Expr::ExtVectorElementExprClass:
1386e5dd7070Spatrick     return EmitExtVectorElementExpr(cast<ExtVectorElementExpr>(E));
1387*12c85518Srobert   case Expr::CXXThisExprClass:
1388*12c85518Srobert     return MakeAddrLValue(LoadCXXThisAddress(), E->getType());
1389e5dd7070Spatrick   case Expr::MemberExprClass:
1390e5dd7070Spatrick     return EmitMemberExpr(cast<MemberExpr>(E));
1391e5dd7070Spatrick   case Expr::CompoundLiteralExprClass:
1392e5dd7070Spatrick     return EmitCompoundLiteralLValue(cast<CompoundLiteralExpr>(E));
1393e5dd7070Spatrick   case Expr::ConditionalOperatorClass:
1394e5dd7070Spatrick     return EmitConditionalOperatorLValue(cast<ConditionalOperator>(E));
1395e5dd7070Spatrick   case Expr::BinaryConditionalOperatorClass:
1396e5dd7070Spatrick     return EmitConditionalOperatorLValue(cast<BinaryConditionalOperator>(E));
1397e5dd7070Spatrick   case Expr::ChooseExprClass:
1398e5dd7070Spatrick     return EmitLValue(cast<ChooseExpr>(E)->getChosenSubExpr());
1399e5dd7070Spatrick   case Expr::OpaqueValueExprClass:
1400e5dd7070Spatrick     return EmitOpaqueValueLValue(cast<OpaqueValueExpr>(E));
1401e5dd7070Spatrick   case Expr::SubstNonTypeTemplateParmExprClass:
1402e5dd7070Spatrick     return EmitLValue(cast<SubstNonTypeTemplateParmExpr>(E)->getReplacement());
1403e5dd7070Spatrick   case Expr::ImplicitCastExprClass:
1404e5dd7070Spatrick   case Expr::CStyleCastExprClass:
1405e5dd7070Spatrick   case Expr::CXXFunctionalCastExprClass:
1406e5dd7070Spatrick   case Expr::CXXStaticCastExprClass:
1407e5dd7070Spatrick   case Expr::CXXDynamicCastExprClass:
1408e5dd7070Spatrick   case Expr::CXXReinterpretCastExprClass:
1409e5dd7070Spatrick   case Expr::CXXConstCastExprClass:
1410ec727ea7Spatrick   case Expr::CXXAddrspaceCastExprClass:
1411e5dd7070Spatrick   case Expr::ObjCBridgedCastExprClass:
1412e5dd7070Spatrick     return EmitCastLValue(cast<CastExpr>(E));
1413e5dd7070Spatrick 
1414e5dd7070Spatrick   case Expr::MaterializeTemporaryExprClass:
1415e5dd7070Spatrick     return EmitMaterializeTemporaryExpr(cast<MaterializeTemporaryExpr>(E));
1416e5dd7070Spatrick 
1417e5dd7070Spatrick   case Expr::CoawaitExprClass:
1418e5dd7070Spatrick     return EmitCoawaitLValue(cast<CoawaitExpr>(E));
1419e5dd7070Spatrick   case Expr::CoyieldExprClass:
1420e5dd7070Spatrick     return EmitCoyieldLValue(cast<CoyieldExpr>(E));
1421e5dd7070Spatrick   }
1422e5dd7070Spatrick }
1423e5dd7070Spatrick 
1424e5dd7070Spatrick /// Given an object of the given canonical type, can we safely copy a
1425e5dd7070Spatrick /// value out of it based on its initializer?
isConstantEmittableObjectType(QualType type)1426e5dd7070Spatrick static bool isConstantEmittableObjectType(QualType type) {
1427e5dd7070Spatrick   assert(type.isCanonical());
1428e5dd7070Spatrick   assert(!type->isReferenceType());
1429e5dd7070Spatrick 
1430e5dd7070Spatrick   // Must be const-qualified but non-volatile.
1431e5dd7070Spatrick   Qualifiers qs = type.getLocalQualifiers();
1432e5dd7070Spatrick   if (!qs.hasConst() || qs.hasVolatile()) return false;
1433e5dd7070Spatrick 
1434e5dd7070Spatrick   // Otherwise, all object types satisfy this except C++ classes with
1435e5dd7070Spatrick   // mutable subobjects or non-trivial copy/destroy behavior.
1436e5dd7070Spatrick   if (const auto *RT = dyn_cast<RecordType>(type))
1437e5dd7070Spatrick     if (const auto *RD = dyn_cast<CXXRecordDecl>(RT->getDecl()))
1438e5dd7070Spatrick       if (RD->hasMutableFields() || !RD->isTrivial())
1439e5dd7070Spatrick         return false;
1440e5dd7070Spatrick 
1441e5dd7070Spatrick   return true;
1442e5dd7070Spatrick }
1443e5dd7070Spatrick 
1444e5dd7070Spatrick /// Can we constant-emit a load of a reference to a variable of the
1445e5dd7070Spatrick /// given type?  This is different from predicates like
1446e5dd7070Spatrick /// Decl::mightBeUsableInConstantExpressions because we do want it to apply
1447e5dd7070Spatrick /// in situations that don't necessarily satisfy the language's rules
1448e5dd7070Spatrick /// for this (e.g. C++'s ODR-use rules).  For example, we want to able
1449e5dd7070Spatrick /// to do this with const float variables even if those variables
1450e5dd7070Spatrick /// aren't marked 'constexpr'.
1451e5dd7070Spatrick enum ConstantEmissionKind {
1452e5dd7070Spatrick   CEK_None,
1453e5dd7070Spatrick   CEK_AsReferenceOnly,
1454e5dd7070Spatrick   CEK_AsValueOrReference,
1455e5dd7070Spatrick   CEK_AsValueOnly
1456e5dd7070Spatrick };
checkVarTypeForConstantEmission(QualType type)1457e5dd7070Spatrick static ConstantEmissionKind checkVarTypeForConstantEmission(QualType type) {
1458e5dd7070Spatrick   type = type.getCanonicalType();
1459e5dd7070Spatrick   if (const auto *ref = dyn_cast<ReferenceType>(type)) {
1460e5dd7070Spatrick     if (isConstantEmittableObjectType(ref->getPointeeType()))
1461e5dd7070Spatrick       return CEK_AsValueOrReference;
1462e5dd7070Spatrick     return CEK_AsReferenceOnly;
1463e5dd7070Spatrick   }
1464e5dd7070Spatrick   if (isConstantEmittableObjectType(type))
1465e5dd7070Spatrick     return CEK_AsValueOnly;
1466e5dd7070Spatrick   return CEK_None;
1467e5dd7070Spatrick }
1468e5dd7070Spatrick 
1469e5dd7070Spatrick /// Try to emit a reference to the given value without producing it as
1470e5dd7070Spatrick /// an l-value.  This is just an optimization, but it avoids us needing
1471e5dd7070Spatrick /// to emit global copies of variables if they're named without triggering
1472e5dd7070Spatrick /// a formal use in a context where we can't emit a direct reference to them,
1473e5dd7070Spatrick /// for instance if a block or lambda or a member of a local class uses a
1474e5dd7070Spatrick /// const int variable or constexpr variable from an enclosing function.
1475e5dd7070Spatrick CodeGenFunction::ConstantEmission
tryEmitAsConstant(DeclRefExpr * refExpr)1476e5dd7070Spatrick CodeGenFunction::tryEmitAsConstant(DeclRefExpr *refExpr) {
1477e5dd7070Spatrick   ValueDecl *value = refExpr->getDecl();
1478e5dd7070Spatrick 
1479e5dd7070Spatrick   // The value needs to be an enum constant or a constant variable.
1480e5dd7070Spatrick   ConstantEmissionKind CEK;
1481e5dd7070Spatrick   if (isa<ParmVarDecl>(value)) {
1482e5dd7070Spatrick     CEK = CEK_None;
1483e5dd7070Spatrick   } else if (auto *var = dyn_cast<VarDecl>(value)) {
1484e5dd7070Spatrick     CEK = checkVarTypeForConstantEmission(var->getType());
1485e5dd7070Spatrick   } else if (isa<EnumConstantDecl>(value)) {
1486e5dd7070Spatrick     CEK = CEK_AsValueOnly;
1487e5dd7070Spatrick   } else {
1488e5dd7070Spatrick     CEK = CEK_None;
1489e5dd7070Spatrick   }
1490e5dd7070Spatrick   if (CEK == CEK_None) return ConstantEmission();
1491e5dd7070Spatrick 
1492e5dd7070Spatrick   Expr::EvalResult result;
1493e5dd7070Spatrick   bool resultIsReference;
1494e5dd7070Spatrick   QualType resultType;
1495e5dd7070Spatrick 
1496e5dd7070Spatrick   // It's best to evaluate all the way as an r-value if that's permitted.
1497e5dd7070Spatrick   if (CEK != CEK_AsReferenceOnly &&
1498e5dd7070Spatrick       refExpr->EvaluateAsRValue(result, getContext())) {
1499e5dd7070Spatrick     resultIsReference = false;
1500e5dd7070Spatrick     resultType = refExpr->getType();
1501e5dd7070Spatrick 
1502e5dd7070Spatrick   // Otherwise, try to evaluate as an l-value.
1503e5dd7070Spatrick   } else if (CEK != CEK_AsValueOnly &&
1504e5dd7070Spatrick              refExpr->EvaluateAsLValue(result, getContext())) {
1505e5dd7070Spatrick     resultIsReference = true;
1506e5dd7070Spatrick     resultType = value->getType();
1507e5dd7070Spatrick 
1508e5dd7070Spatrick   // Failure.
1509e5dd7070Spatrick   } else {
1510e5dd7070Spatrick     return ConstantEmission();
1511e5dd7070Spatrick   }
1512e5dd7070Spatrick 
1513e5dd7070Spatrick   // In any case, if the initializer has side-effects, abandon ship.
1514e5dd7070Spatrick   if (result.HasSideEffects)
1515e5dd7070Spatrick     return ConstantEmission();
1516e5dd7070Spatrick 
1517a9ac8606Spatrick   // In CUDA/HIP device compilation, a lambda may capture a reference variable
1518a9ac8606Spatrick   // referencing a global host variable by copy. In this case the lambda should
1519a9ac8606Spatrick   // make a copy of the value of the global host variable. The DRE of the
1520a9ac8606Spatrick   // captured reference variable cannot be emitted as load from the host
1521a9ac8606Spatrick   // global variable as compile time constant, since the host variable is not
1522a9ac8606Spatrick   // accessible on device. The DRE of the captured reference variable has to be
1523a9ac8606Spatrick   // loaded from captures.
1524a9ac8606Spatrick   if (CGM.getLangOpts().CUDAIsDevice && result.Val.isLValue() &&
1525a9ac8606Spatrick       refExpr->refersToEnclosingVariableOrCapture()) {
1526a9ac8606Spatrick     auto *MD = dyn_cast_or_null<CXXMethodDecl>(CurCodeDecl);
1527a9ac8606Spatrick     if (MD && MD->getParent()->isLambda() &&
1528a9ac8606Spatrick         MD->getOverloadedOperator() == OO_Call) {
1529a9ac8606Spatrick       const APValue::LValueBase &base = result.Val.getLValueBase();
1530a9ac8606Spatrick       if (const ValueDecl *D = base.dyn_cast<const ValueDecl *>()) {
1531a9ac8606Spatrick         if (const VarDecl *VD = dyn_cast<const VarDecl>(D)) {
1532a9ac8606Spatrick           if (!VD->hasAttr<CUDADeviceAttr>()) {
1533a9ac8606Spatrick             return ConstantEmission();
1534a9ac8606Spatrick           }
1535a9ac8606Spatrick         }
1536a9ac8606Spatrick       }
1537a9ac8606Spatrick     }
1538a9ac8606Spatrick   }
1539a9ac8606Spatrick 
1540e5dd7070Spatrick   // Emit as a constant.
1541e5dd7070Spatrick   auto C = ConstantEmitter(*this).emitAbstract(refExpr->getLocation(),
1542e5dd7070Spatrick                                                result.Val, resultType);
1543e5dd7070Spatrick 
1544e5dd7070Spatrick   // Make sure we emit a debug reference to the global variable.
1545e5dd7070Spatrick   // This should probably fire even for
1546e5dd7070Spatrick   if (isa<VarDecl>(value)) {
1547e5dd7070Spatrick     if (!getContext().DeclMustBeEmitted(cast<VarDecl>(value)))
1548e5dd7070Spatrick       EmitDeclRefExprDbgValue(refExpr, result.Val);
1549e5dd7070Spatrick   } else {
1550e5dd7070Spatrick     assert(isa<EnumConstantDecl>(value));
1551e5dd7070Spatrick     EmitDeclRefExprDbgValue(refExpr, result.Val);
1552e5dd7070Spatrick   }
1553e5dd7070Spatrick 
1554e5dd7070Spatrick   // If we emitted a reference constant, we need to dereference that.
1555e5dd7070Spatrick   if (resultIsReference)
1556e5dd7070Spatrick     return ConstantEmission::forReference(C);
1557e5dd7070Spatrick 
1558e5dd7070Spatrick   return ConstantEmission::forValue(C);
1559e5dd7070Spatrick }
1560e5dd7070Spatrick 
tryToConvertMemberExprToDeclRefExpr(CodeGenFunction & CGF,const MemberExpr * ME)1561e5dd7070Spatrick static DeclRefExpr *tryToConvertMemberExprToDeclRefExpr(CodeGenFunction &CGF,
1562e5dd7070Spatrick                                                         const MemberExpr *ME) {
1563e5dd7070Spatrick   if (auto *VD = dyn_cast<VarDecl>(ME->getMemberDecl())) {
1564e5dd7070Spatrick     // Try to emit static variable member expressions as DREs.
1565e5dd7070Spatrick     return DeclRefExpr::Create(
1566e5dd7070Spatrick         CGF.getContext(), NestedNameSpecifierLoc(), SourceLocation(), VD,
1567e5dd7070Spatrick         /*RefersToEnclosingVariableOrCapture=*/false, ME->getExprLoc(),
1568e5dd7070Spatrick         ME->getType(), ME->getValueKind(), nullptr, nullptr, ME->isNonOdrUse());
1569e5dd7070Spatrick   }
1570e5dd7070Spatrick   return nullptr;
1571e5dd7070Spatrick }
1572e5dd7070Spatrick 
1573e5dd7070Spatrick CodeGenFunction::ConstantEmission
tryEmitAsConstant(const MemberExpr * ME)1574e5dd7070Spatrick CodeGenFunction::tryEmitAsConstant(const MemberExpr *ME) {
1575e5dd7070Spatrick   if (DeclRefExpr *DRE = tryToConvertMemberExprToDeclRefExpr(*this, ME))
1576e5dd7070Spatrick     return tryEmitAsConstant(DRE);
1577e5dd7070Spatrick   return ConstantEmission();
1578e5dd7070Spatrick }
1579e5dd7070Spatrick 
emitScalarConstant(const CodeGenFunction::ConstantEmission & Constant,Expr * E)1580e5dd7070Spatrick llvm::Value *CodeGenFunction::emitScalarConstant(
1581e5dd7070Spatrick     const CodeGenFunction::ConstantEmission &Constant, Expr *E) {
1582e5dd7070Spatrick   assert(Constant && "not a constant");
1583e5dd7070Spatrick   if (Constant.isReference())
1584e5dd7070Spatrick     return EmitLoadOfLValue(Constant.getReferenceLValue(*this, E),
1585e5dd7070Spatrick                             E->getExprLoc())
1586e5dd7070Spatrick         .getScalarVal();
1587e5dd7070Spatrick   return Constant.getValue();
1588e5dd7070Spatrick }
1589e5dd7070Spatrick 
EmitLoadOfScalar(LValue lvalue,SourceLocation Loc)1590e5dd7070Spatrick llvm::Value *CodeGenFunction::EmitLoadOfScalar(LValue lvalue,
1591e5dd7070Spatrick                                                SourceLocation Loc) {
1592e5dd7070Spatrick   return EmitLoadOfScalar(lvalue.getAddress(*this), lvalue.isVolatile(),
1593e5dd7070Spatrick                           lvalue.getType(), Loc, lvalue.getBaseInfo(),
1594e5dd7070Spatrick                           lvalue.getTBAAInfo(), lvalue.isNontemporal());
1595e5dd7070Spatrick }
1596e5dd7070Spatrick 
hasBooleanRepresentation(QualType Ty)1597e5dd7070Spatrick static bool hasBooleanRepresentation(QualType Ty) {
1598e5dd7070Spatrick   if (Ty->isBooleanType())
1599e5dd7070Spatrick     return true;
1600e5dd7070Spatrick 
1601e5dd7070Spatrick   if (const EnumType *ET = Ty->getAs<EnumType>())
1602e5dd7070Spatrick     return ET->getDecl()->getIntegerType()->isBooleanType();
1603e5dd7070Spatrick 
1604e5dd7070Spatrick   if (const AtomicType *AT = Ty->getAs<AtomicType>())
1605e5dd7070Spatrick     return hasBooleanRepresentation(AT->getValueType());
1606e5dd7070Spatrick 
1607e5dd7070Spatrick   return false;
1608e5dd7070Spatrick }
1609e5dd7070Spatrick 
getRangeForType(CodeGenFunction & CGF,QualType Ty,llvm::APInt & Min,llvm::APInt & End,bool StrictEnums,bool IsBool)1610e5dd7070Spatrick static bool getRangeForType(CodeGenFunction &CGF, QualType Ty,
1611e5dd7070Spatrick                             llvm::APInt &Min, llvm::APInt &End,
1612e5dd7070Spatrick                             bool StrictEnums, bool IsBool) {
1613e5dd7070Spatrick   const EnumType *ET = Ty->getAs<EnumType>();
1614e5dd7070Spatrick   bool IsRegularCPlusPlusEnum = CGF.getLangOpts().CPlusPlus && StrictEnums &&
1615e5dd7070Spatrick                                 ET && !ET->getDecl()->isFixed();
1616e5dd7070Spatrick   if (!IsBool && !IsRegularCPlusPlusEnum)
1617e5dd7070Spatrick     return false;
1618e5dd7070Spatrick 
1619e5dd7070Spatrick   if (IsBool) {
1620e5dd7070Spatrick     Min = llvm::APInt(CGF.getContext().getTypeSize(Ty), 0);
1621e5dd7070Spatrick     End = llvm::APInt(CGF.getContext().getTypeSize(Ty), 2);
1622e5dd7070Spatrick   } else {
1623e5dd7070Spatrick     const EnumDecl *ED = ET->getDecl();
1624*12c85518Srobert     ED->getValueRange(End, Min);
1625e5dd7070Spatrick   }
1626e5dd7070Spatrick   return true;
1627e5dd7070Spatrick }
1628e5dd7070Spatrick 
getRangeForLoadFromType(QualType Ty)1629e5dd7070Spatrick llvm::MDNode *CodeGenFunction::getRangeForLoadFromType(QualType Ty) {
1630e5dd7070Spatrick   llvm::APInt Min, End;
1631e5dd7070Spatrick   if (!getRangeForType(*this, Ty, Min, End, CGM.getCodeGenOpts().StrictEnums,
1632e5dd7070Spatrick                        hasBooleanRepresentation(Ty)))
1633e5dd7070Spatrick     return nullptr;
1634e5dd7070Spatrick 
1635e5dd7070Spatrick   llvm::MDBuilder MDHelper(getLLVMContext());
1636e5dd7070Spatrick   return MDHelper.createRange(Min, End);
1637e5dd7070Spatrick }
1638e5dd7070Spatrick 
EmitScalarRangeCheck(llvm::Value * Value,QualType Ty,SourceLocation Loc)1639e5dd7070Spatrick bool CodeGenFunction::EmitScalarRangeCheck(llvm::Value *Value, QualType Ty,
1640e5dd7070Spatrick                                            SourceLocation Loc) {
1641e5dd7070Spatrick   bool HasBoolCheck = SanOpts.has(SanitizerKind::Bool);
1642e5dd7070Spatrick   bool HasEnumCheck = SanOpts.has(SanitizerKind::Enum);
1643e5dd7070Spatrick   if (!HasBoolCheck && !HasEnumCheck)
1644e5dd7070Spatrick     return false;
1645e5dd7070Spatrick 
1646e5dd7070Spatrick   bool IsBool = hasBooleanRepresentation(Ty) ||
1647e5dd7070Spatrick                 NSAPI(CGM.getContext()).isObjCBOOLType(Ty);
1648e5dd7070Spatrick   bool NeedsBoolCheck = HasBoolCheck && IsBool;
1649e5dd7070Spatrick   bool NeedsEnumCheck = HasEnumCheck && Ty->getAs<EnumType>();
1650e5dd7070Spatrick   if (!NeedsBoolCheck && !NeedsEnumCheck)
1651e5dd7070Spatrick     return false;
1652e5dd7070Spatrick 
1653e5dd7070Spatrick   // Single-bit booleans don't need to be checked. Special-case this to avoid
1654e5dd7070Spatrick   // a bit width mismatch when handling bitfield values. This is handled by
1655e5dd7070Spatrick   // EmitFromMemory for the non-bitfield case.
1656e5dd7070Spatrick   if (IsBool &&
1657e5dd7070Spatrick       cast<llvm::IntegerType>(Value->getType())->getBitWidth() == 1)
1658e5dd7070Spatrick     return false;
1659e5dd7070Spatrick 
1660e5dd7070Spatrick   llvm::APInt Min, End;
1661e5dd7070Spatrick   if (!getRangeForType(*this, Ty, Min, End, /*StrictEnums=*/true, IsBool))
1662e5dd7070Spatrick     return true;
1663e5dd7070Spatrick 
1664e5dd7070Spatrick   auto &Ctx = getLLVMContext();
1665e5dd7070Spatrick   SanitizerScope SanScope(this);
1666e5dd7070Spatrick   llvm::Value *Check;
1667e5dd7070Spatrick   --End;
1668e5dd7070Spatrick   if (!Min) {
1669e5dd7070Spatrick     Check = Builder.CreateICmpULE(Value, llvm::ConstantInt::get(Ctx, End));
1670e5dd7070Spatrick   } else {
1671e5dd7070Spatrick     llvm::Value *Upper =
1672e5dd7070Spatrick         Builder.CreateICmpSLE(Value, llvm::ConstantInt::get(Ctx, End));
1673e5dd7070Spatrick     llvm::Value *Lower =
1674e5dd7070Spatrick         Builder.CreateICmpSGE(Value, llvm::ConstantInt::get(Ctx, Min));
1675e5dd7070Spatrick     Check = Builder.CreateAnd(Upper, Lower);
1676e5dd7070Spatrick   }
1677e5dd7070Spatrick   llvm::Constant *StaticArgs[] = {EmitCheckSourceLocation(Loc),
1678e5dd7070Spatrick                                   EmitCheckTypeDescriptor(Ty)};
1679e5dd7070Spatrick   SanitizerMask Kind =
1680e5dd7070Spatrick       NeedsEnumCheck ? SanitizerKind::Enum : SanitizerKind::Bool;
1681e5dd7070Spatrick   EmitCheck(std::make_pair(Check, Kind), SanitizerHandler::LoadInvalidValue,
1682e5dd7070Spatrick             StaticArgs, EmitCheckValue(Value));
1683e5dd7070Spatrick   return true;
1684e5dd7070Spatrick }
1685e5dd7070Spatrick 
EmitLoadOfScalar(Address Addr,bool Volatile,QualType Ty,SourceLocation Loc,LValueBaseInfo BaseInfo,TBAAAccessInfo TBAAInfo,bool isNontemporal)1686e5dd7070Spatrick llvm::Value *CodeGenFunction::EmitLoadOfScalar(Address Addr, bool Volatile,
1687e5dd7070Spatrick                                                QualType Ty,
1688e5dd7070Spatrick                                                SourceLocation Loc,
1689e5dd7070Spatrick                                                LValueBaseInfo BaseInfo,
1690e5dd7070Spatrick                                                TBAAAccessInfo TBAAInfo,
1691e5dd7070Spatrick                                                bool isNontemporal) {
1692*12c85518Srobert   if (auto *GV = dyn_cast<llvm::GlobalValue>(Addr.getPointer()))
1693*12c85518Srobert     if (GV->isThreadLocal())
1694*12c85518Srobert       Addr = Addr.withPointer(Builder.CreateThreadLocalAddress(GV));
1695e5dd7070Spatrick 
1696*12c85518Srobert   if (const auto *ClangVecTy = Ty->getAs<VectorType>()) {
1697*12c85518Srobert     // Boolean vectors use `iN` as storage type.
1698*12c85518Srobert     if (ClangVecTy->isExtVectorBoolType()) {
1699*12c85518Srobert       llvm::Type *ValTy = ConvertType(Ty);
1700*12c85518Srobert       unsigned ValNumElems =
1701*12c85518Srobert           cast<llvm::FixedVectorType>(ValTy)->getNumElements();
1702*12c85518Srobert       // Load the `iP` storage object (P is the padded vector size).
1703*12c85518Srobert       auto *RawIntV = Builder.CreateLoad(Addr, Volatile, "load_bits");
1704*12c85518Srobert       const auto *RawIntTy = RawIntV->getType();
1705*12c85518Srobert       assert(RawIntTy->isIntegerTy() && "compressed iN storage for bitvectors");
1706*12c85518Srobert       // Bitcast iP --> <P x i1>.
1707*12c85518Srobert       auto *PaddedVecTy = llvm::FixedVectorType::get(
1708*12c85518Srobert           Builder.getInt1Ty(), RawIntTy->getPrimitiveSizeInBits());
1709*12c85518Srobert       llvm::Value *V = Builder.CreateBitCast(RawIntV, PaddedVecTy);
1710*12c85518Srobert       // Shuffle <P x i1> --> <N x i1> (N is the actual bit size).
1711*12c85518Srobert       V = emitBoolVecConversion(V, ValNumElems, "extractvec");
1712*12c85518Srobert 
1713*12c85518Srobert       return EmitFromMemory(V, Ty);
1714*12c85518Srobert     }
1715e5dd7070Spatrick 
1716e5dd7070Spatrick     // Handle vectors of size 3 like size 4 for better performance.
1717*12c85518Srobert     const llvm::Type *EltTy = Addr.getElementType();
1718*12c85518Srobert     const auto *VTy = cast<llvm::FixedVectorType>(EltTy);
1719*12c85518Srobert 
1720*12c85518Srobert     if (!CGM.getCodeGenOpts().PreserveVec3Type && VTy->getNumElements() == 3) {
1721e5dd7070Spatrick 
1722e5dd7070Spatrick       // Bitcast to vec4 type.
1723*12c85518Srobert       llvm::VectorType *vec4Ty =
1724*12c85518Srobert           llvm::FixedVectorType::get(VTy->getElementType(), 4);
1725e5dd7070Spatrick       Address Cast = Builder.CreateElementBitCast(Addr, vec4Ty, "castToVec4");
1726e5dd7070Spatrick       // Now load value.
1727e5dd7070Spatrick       llvm::Value *V = Builder.CreateLoad(Cast, Volatile, "loadVec4");
1728e5dd7070Spatrick 
1729e5dd7070Spatrick       // Shuffle vector to get vec3.
1730*12c85518Srobert       V = Builder.CreateShuffleVector(V, ArrayRef<int>{0, 1, 2}, "extractVec");
1731e5dd7070Spatrick       return EmitFromMemory(V, Ty);
1732e5dd7070Spatrick     }
1733e5dd7070Spatrick   }
1734e5dd7070Spatrick 
1735e5dd7070Spatrick   // Atomic operations have to be done on integral types.
1736e5dd7070Spatrick   LValue AtomicLValue =
1737e5dd7070Spatrick       LValue::MakeAddr(Addr, Ty, getContext(), BaseInfo, TBAAInfo);
1738e5dd7070Spatrick   if (Ty->isAtomicType() || LValueIsSuitableForInlineAtomic(AtomicLValue)) {
1739e5dd7070Spatrick     return EmitAtomicLoad(AtomicLValue, Loc).getScalarVal();
1740e5dd7070Spatrick   }
1741e5dd7070Spatrick 
1742e5dd7070Spatrick   llvm::LoadInst *Load = Builder.CreateLoad(Addr, Volatile);
1743e5dd7070Spatrick   if (isNontemporal) {
1744e5dd7070Spatrick     llvm::MDNode *Node = llvm::MDNode::get(
1745e5dd7070Spatrick         Load->getContext(), llvm::ConstantAsMetadata::get(Builder.getInt32(1)));
1746e5dd7070Spatrick     Load->setMetadata(CGM.getModule().getMDKindID("nontemporal"), Node);
1747e5dd7070Spatrick   }
1748e5dd7070Spatrick 
1749e5dd7070Spatrick   CGM.DecorateInstructionWithTBAA(Load, TBAAInfo);
1750e5dd7070Spatrick 
1751e5dd7070Spatrick   if (EmitScalarRangeCheck(Load, Ty, Loc)) {
1752e5dd7070Spatrick     // In order to prevent the optimizer from throwing away the check, don't
1753e5dd7070Spatrick     // attach range metadata to the load.
1754e5dd7070Spatrick   } else if (CGM.getCodeGenOpts().OptimizationLevel > 0)
1755*12c85518Srobert     if (llvm::MDNode *RangeInfo = getRangeForLoadFromType(Ty)) {
1756e5dd7070Spatrick       Load->setMetadata(llvm::LLVMContext::MD_range, RangeInfo);
1757*12c85518Srobert       Load->setMetadata(llvm::LLVMContext::MD_noundef,
1758*12c85518Srobert                         llvm::MDNode::get(getLLVMContext(), std::nullopt));
1759*12c85518Srobert     }
1760e5dd7070Spatrick 
1761e5dd7070Spatrick   return EmitFromMemory(Load, Ty);
1762e5dd7070Spatrick }
1763e5dd7070Spatrick 
EmitToMemory(llvm::Value * Value,QualType Ty)1764e5dd7070Spatrick llvm::Value *CodeGenFunction::EmitToMemory(llvm::Value *Value, QualType Ty) {
1765e5dd7070Spatrick   // Bool has a different representation in memory than in registers.
1766e5dd7070Spatrick   if (hasBooleanRepresentation(Ty)) {
1767e5dd7070Spatrick     // This should really always be an i1, but sometimes it's already
1768e5dd7070Spatrick     // an i8, and it's awkward to track those cases down.
1769e5dd7070Spatrick     if (Value->getType()->isIntegerTy(1))
1770e5dd7070Spatrick       return Builder.CreateZExt(Value, ConvertTypeForMem(Ty), "frombool");
1771e5dd7070Spatrick     assert(Value->getType()->isIntegerTy(getContext().getTypeSize(Ty)) &&
1772e5dd7070Spatrick            "wrong value rep of bool");
1773e5dd7070Spatrick   }
1774e5dd7070Spatrick 
1775e5dd7070Spatrick   return Value;
1776e5dd7070Spatrick }
1777e5dd7070Spatrick 
EmitFromMemory(llvm::Value * Value,QualType Ty)1778e5dd7070Spatrick llvm::Value *CodeGenFunction::EmitFromMemory(llvm::Value *Value, QualType Ty) {
1779e5dd7070Spatrick   // Bool has a different representation in memory than in registers.
1780e5dd7070Spatrick   if (hasBooleanRepresentation(Ty)) {
1781e5dd7070Spatrick     assert(Value->getType()->isIntegerTy(getContext().getTypeSize(Ty)) &&
1782e5dd7070Spatrick            "wrong value rep of bool");
1783e5dd7070Spatrick     return Builder.CreateTrunc(Value, Builder.getInt1Ty(), "tobool");
1784e5dd7070Spatrick   }
1785*12c85518Srobert   if (Ty->isExtVectorBoolType()) {
1786*12c85518Srobert     const auto *RawIntTy = Value->getType();
1787*12c85518Srobert     // Bitcast iP --> <P x i1>.
1788*12c85518Srobert     auto *PaddedVecTy = llvm::FixedVectorType::get(
1789*12c85518Srobert         Builder.getInt1Ty(), RawIntTy->getPrimitiveSizeInBits());
1790*12c85518Srobert     auto *V = Builder.CreateBitCast(Value, PaddedVecTy);
1791*12c85518Srobert     // Shuffle <P x i1> --> <N x i1> (N is the actual bit size).
1792*12c85518Srobert     llvm::Type *ValTy = ConvertType(Ty);
1793*12c85518Srobert     unsigned ValNumElems = cast<llvm::FixedVectorType>(ValTy)->getNumElements();
1794*12c85518Srobert     return emitBoolVecConversion(V, ValNumElems, "extractvec");
1795*12c85518Srobert   }
1796e5dd7070Spatrick 
1797e5dd7070Spatrick   return Value;
1798e5dd7070Spatrick }
1799e5dd7070Spatrick 
1800ec727ea7Spatrick // Convert the pointer of \p Addr to a pointer to a vector (the value type of
1801ec727ea7Spatrick // MatrixType), if it points to a array (the memory type of MatrixType).
MaybeConvertMatrixAddress(Address Addr,CodeGenFunction & CGF,bool IsVector=true)1802ec727ea7Spatrick static Address MaybeConvertMatrixAddress(Address Addr, CodeGenFunction &CGF,
1803ec727ea7Spatrick                                          bool IsVector = true) {
1804*12c85518Srobert   auto *ArrayTy = dyn_cast<llvm::ArrayType>(Addr.getElementType());
1805ec727ea7Spatrick   if (ArrayTy && IsVector) {
1806ec727ea7Spatrick     auto *VectorTy = llvm::FixedVectorType::get(ArrayTy->getElementType(),
1807ec727ea7Spatrick                                                 ArrayTy->getNumElements());
1808ec727ea7Spatrick 
1809ec727ea7Spatrick     return Address(CGF.Builder.CreateElementBitCast(Addr, VectorTy));
1810ec727ea7Spatrick   }
1811*12c85518Srobert   auto *VectorTy = dyn_cast<llvm::VectorType>(Addr.getElementType());
1812ec727ea7Spatrick   if (VectorTy && !IsVector) {
1813a9ac8606Spatrick     auto *ArrayTy = llvm::ArrayType::get(
1814a9ac8606Spatrick         VectorTy->getElementType(),
1815a9ac8606Spatrick         cast<llvm::FixedVectorType>(VectorTy)->getNumElements());
1816ec727ea7Spatrick 
1817ec727ea7Spatrick     return Address(CGF.Builder.CreateElementBitCast(Addr, ArrayTy));
1818ec727ea7Spatrick   }
1819ec727ea7Spatrick 
1820ec727ea7Spatrick   return Addr;
1821ec727ea7Spatrick }
1822ec727ea7Spatrick 
1823ec727ea7Spatrick // Emit a store of a matrix LValue. This may require casting the original
1824ec727ea7Spatrick // pointer to memory address (ArrayType) to a pointer to the value type
1825ec727ea7Spatrick // (VectorType).
EmitStoreOfMatrixScalar(llvm::Value * value,LValue lvalue,bool isInit,CodeGenFunction & CGF)1826ec727ea7Spatrick static void EmitStoreOfMatrixScalar(llvm::Value *value, LValue lvalue,
1827ec727ea7Spatrick                                     bool isInit, CodeGenFunction &CGF) {
1828ec727ea7Spatrick   Address Addr = MaybeConvertMatrixAddress(lvalue.getAddress(CGF), CGF,
1829ec727ea7Spatrick                                            value->getType()->isVectorTy());
1830ec727ea7Spatrick   CGF.EmitStoreOfScalar(value, Addr, lvalue.isVolatile(), lvalue.getType(),
1831ec727ea7Spatrick                         lvalue.getBaseInfo(), lvalue.getTBAAInfo(), isInit,
1832ec727ea7Spatrick                         lvalue.isNontemporal());
1833ec727ea7Spatrick }
1834ec727ea7Spatrick 
EmitStoreOfScalar(llvm::Value * Value,Address Addr,bool Volatile,QualType Ty,LValueBaseInfo BaseInfo,TBAAAccessInfo TBAAInfo,bool isInit,bool isNontemporal)1835e5dd7070Spatrick void CodeGenFunction::EmitStoreOfScalar(llvm::Value *Value, Address Addr,
1836e5dd7070Spatrick                                         bool Volatile, QualType Ty,
1837e5dd7070Spatrick                                         LValueBaseInfo BaseInfo,
1838e5dd7070Spatrick                                         TBAAAccessInfo TBAAInfo,
1839e5dd7070Spatrick                                         bool isInit, bool isNontemporal) {
1840*12c85518Srobert   if (auto *GV = dyn_cast<llvm::GlobalValue>(Addr.getPointer()))
1841*12c85518Srobert     if (GV->isThreadLocal())
1842*12c85518Srobert       Addr = Addr.withPointer(Builder.CreateThreadLocalAddress(GV));
1843*12c85518Srobert 
1844e5dd7070Spatrick   llvm::Type *SrcTy = Value->getType();
1845*12c85518Srobert   if (const auto *ClangVecTy = Ty->getAs<VectorType>()) {
1846*12c85518Srobert     auto *VecTy = dyn_cast<llvm::FixedVectorType>(SrcTy);
1847*12c85518Srobert     if (VecTy && ClangVecTy->isExtVectorBoolType()) {
1848*12c85518Srobert       auto *MemIntTy = cast<llvm::IntegerType>(Addr.getElementType());
1849*12c85518Srobert       // Expand to the memory bit width.
1850*12c85518Srobert       unsigned MemNumElems = MemIntTy->getPrimitiveSizeInBits();
1851*12c85518Srobert       // <N x i1> --> <P x i1>.
1852*12c85518Srobert       Value = emitBoolVecConversion(Value, MemNumElems, "insertvec");
1853*12c85518Srobert       // <P x i1> --> iP.
1854*12c85518Srobert       Value = Builder.CreateBitCast(Value, MemIntTy);
1855*12c85518Srobert     } else if (!CGM.getCodeGenOpts().PreserveVec3Type) {
1856e5dd7070Spatrick       // Handle vec3 special.
1857a9ac8606Spatrick       if (VecTy && cast<llvm::FixedVectorType>(VecTy)->getNumElements() == 3) {
1858e5dd7070Spatrick         // Our source is a vec3, do a shuffle vector to make it a vec4.
1859a9ac8606Spatrick         Value = Builder.CreateShuffleVector(Value, ArrayRef<int>{0, 1, 2, -1},
1860ec727ea7Spatrick                                             "extractVec");
1861ec727ea7Spatrick         SrcTy = llvm::FixedVectorType::get(VecTy->getElementType(), 4);
1862e5dd7070Spatrick       }
1863e5dd7070Spatrick       if (Addr.getElementType() != SrcTy) {
1864e5dd7070Spatrick         Addr = Builder.CreateElementBitCast(Addr, SrcTy, "storetmp");
1865e5dd7070Spatrick       }
1866e5dd7070Spatrick     }
1867e5dd7070Spatrick   }
1868e5dd7070Spatrick 
1869e5dd7070Spatrick   Value = EmitToMemory(Value, Ty);
1870e5dd7070Spatrick 
1871e5dd7070Spatrick   LValue AtomicLValue =
1872e5dd7070Spatrick       LValue::MakeAddr(Addr, Ty, getContext(), BaseInfo, TBAAInfo);
1873e5dd7070Spatrick   if (Ty->isAtomicType() ||
1874e5dd7070Spatrick       (!isInit && LValueIsSuitableForInlineAtomic(AtomicLValue))) {
1875e5dd7070Spatrick     EmitAtomicStore(RValue::get(Value), AtomicLValue, isInit);
1876e5dd7070Spatrick     return;
1877e5dd7070Spatrick   }
1878e5dd7070Spatrick 
1879e5dd7070Spatrick   llvm::StoreInst *Store = Builder.CreateStore(Value, Addr, Volatile);
1880e5dd7070Spatrick   if (isNontemporal) {
1881e5dd7070Spatrick     llvm::MDNode *Node =
1882e5dd7070Spatrick         llvm::MDNode::get(Store->getContext(),
1883e5dd7070Spatrick                           llvm::ConstantAsMetadata::get(Builder.getInt32(1)));
1884e5dd7070Spatrick     Store->setMetadata(CGM.getModule().getMDKindID("nontemporal"), Node);
1885e5dd7070Spatrick   }
1886e5dd7070Spatrick 
1887e5dd7070Spatrick   CGM.DecorateInstructionWithTBAA(Store, TBAAInfo);
1888e5dd7070Spatrick }
1889e5dd7070Spatrick 
EmitStoreOfScalar(llvm::Value * value,LValue lvalue,bool isInit)1890e5dd7070Spatrick void CodeGenFunction::EmitStoreOfScalar(llvm::Value *value, LValue lvalue,
1891e5dd7070Spatrick                                         bool isInit) {
1892ec727ea7Spatrick   if (lvalue.getType()->isConstantMatrixType()) {
1893ec727ea7Spatrick     EmitStoreOfMatrixScalar(value, lvalue, isInit, *this);
1894ec727ea7Spatrick     return;
1895ec727ea7Spatrick   }
1896ec727ea7Spatrick 
1897e5dd7070Spatrick   EmitStoreOfScalar(value, lvalue.getAddress(*this), lvalue.isVolatile(),
1898e5dd7070Spatrick                     lvalue.getType(), lvalue.getBaseInfo(),
1899e5dd7070Spatrick                     lvalue.getTBAAInfo(), isInit, lvalue.isNontemporal());
1900e5dd7070Spatrick }
1901e5dd7070Spatrick 
1902ec727ea7Spatrick // Emit a load of a LValue of matrix type. This may require casting the pointer
1903ec727ea7Spatrick // to memory address (ArrayType) to a pointer to the value type (VectorType).
EmitLoadOfMatrixLValue(LValue LV,SourceLocation Loc,CodeGenFunction & CGF)1904ec727ea7Spatrick static RValue EmitLoadOfMatrixLValue(LValue LV, SourceLocation Loc,
1905ec727ea7Spatrick                                      CodeGenFunction &CGF) {
1906ec727ea7Spatrick   assert(LV.getType()->isConstantMatrixType());
1907ec727ea7Spatrick   Address Addr = MaybeConvertMatrixAddress(LV.getAddress(CGF), CGF);
1908ec727ea7Spatrick   LV.setAddress(Addr);
1909ec727ea7Spatrick   return RValue::get(CGF.EmitLoadOfScalar(LV, Loc));
1910ec727ea7Spatrick }
1911ec727ea7Spatrick 
1912e5dd7070Spatrick /// EmitLoadOfLValue - Given an expression that represents a value lvalue, this
1913e5dd7070Spatrick /// method emits the address of the lvalue, then loads the result as an rvalue,
1914e5dd7070Spatrick /// returning the rvalue.
EmitLoadOfLValue(LValue LV,SourceLocation Loc)1915e5dd7070Spatrick RValue CodeGenFunction::EmitLoadOfLValue(LValue LV, SourceLocation Loc) {
1916e5dd7070Spatrick   if (LV.isObjCWeak()) {
1917e5dd7070Spatrick     // load of a __weak object.
1918e5dd7070Spatrick     Address AddrWeakObj = LV.getAddress(*this);
1919e5dd7070Spatrick     return RValue::get(CGM.getObjCRuntime().EmitObjCWeakRead(*this,
1920e5dd7070Spatrick                                                              AddrWeakObj));
1921e5dd7070Spatrick   }
1922e5dd7070Spatrick   if (LV.getQuals().getObjCLifetime() == Qualifiers::OCL_Weak) {
1923e5dd7070Spatrick     // In MRC mode, we do a load+autorelease.
1924e5dd7070Spatrick     if (!getLangOpts().ObjCAutoRefCount) {
1925e5dd7070Spatrick       return RValue::get(EmitARCLoadWeak(LV.getAddress(*this)));
1926e5dd7070Spatrick     }
1927e5dd7070Spatrick 
1928e5dd7070Spatrick     // In ARC mode, we load retained and then consume the value.
1929e5dd7070Spatrick     llvm::Value *Object = EmitARCLoadWeakRetained(LV.getAddress(*this));
1930e5dd7070Spatrick     Object = EmitObjCConsumeObject(LV.getType(), Object);
1931e5dd7070Spatrick     return RValue::get(Object);
1932e5dd7070Spatrick   }
1933e5dd7070Spatrick 
1934e5dd7070Spatrick   if (LV.isSimple()) {
1935e5dd7070Spatrick     assert(!LV.getType()->isFunctionType());
1936e5dd7070Spatrick 
1937ec727ea7Spatrick     if (LV.getType()->isConstantMatrixType())
1938ec727ea7Spatrick       return EmitLoadOfMatrixLValue(LV, Loc, *this);
1939ec727ea7Spatrick 
1940e5dd7070Spatrick     // Everything needs a load.
1941e5dd7070Spatrick     return RValue::get(EmitLoadOfScalar(LV, Loc));
1942e5dd7070Spatrick   }
1943e5dd7070Spatrick 
1944e5dd7070Spatrick   if (LV.isVectorElt()) {
1945e5dd7070Spatrick     llvm::LoadInst *Load = Builder.CreateLoad(LV.getVectorAddress(),
1946e5dd7070Spatrick                                               LV.isVolatileQualified());
1947e5dd7070Spatrick     return RValue::get(Builder.CreateExtractElement(Load, LV.getVectorIdx(),
1948e5dd7070Spatrick                                                     "vecext"));
1949e5dd7070Spatrick   }
1950e5dd7070Spatrick 
1951e5dd7070Spatrick   // If this is a reference to a subset of the elements of a vector, either
1952e5dd7070Spatrick   // shuffle the input or extract/insert them as appropriate.
1953ec727ea7Spatrick   if (LV.isExtVectorElt()) {
1954e5dd7070Spatrick     return EmitLoadOfExtVectorElementLValue(LV);
1955ec727ea7Spatrick   }
1956e5dd7070Spatrick 
1957e5dd7070Spatrick   // Global Register variables always invoke intrinsics
1958e5dd7070Spatrick   if (LV.isGlobalReg())
1959e5dd7070Spatrick     return EmitLoadOfGlobalRegLValue(LV);
1960e5dd7070Spatrick 
1961ec727ea7Spatrick   if (LV.isMatrixElt()) {
1962*12c85518Srobert     llvm::Value *Idx = LV.getMatrixIdx();
1963*12c85518Srobert     if (CGM.getCodeGenOpts().OptimizationLevel > 0) {
1964*12c85518Srobert       const auto *const MatTy = LV.getType()->castAs<ConstantMatrixType>();
1965*12c85518Srobert       llvm::MatrixBuilder MB(Builder);
1966*12c85518Srobert       MB.CreateIndexAssumption(Idx, MatTy->getNumElementsFlattened());
1967*12c85518Srobert     }
1968ec727ea7Spatrick     llvm::LoadInst *Load =
1969ec727ea7Spatrick         Builder.CreateLoad(LV.getMatrixAddress(), LV.isVolatileQualified());
1970*12c85518Srobert     return RValue::get(Builder.CreateExtractElement(Load, Idx, "matrixext"));
1971ec727ea7Spatrick   }
1972ec727ea7Spatrick 
1973e5dd7070Spatrick   assert(LV.isBitField() && "Unknown LValue type!");
1974e5dd7070Spatrick   return EmitLoadOfBitfieldLValue(LV, Loc);
1975e5dd7070Spatrick }
1976e5dd7070Spatrick 
EmitLoadOfBitfieldLValue(LValue LV,SourceLocation Loc)1977e5dd7070Spatrick RValue CodeGenFunction::EmitLoadOfBitfieldLValue(LValue LV,
1978e5dd7070Spatrick                                                  SourceLocation Loc) {
1979e5dd7070Spatrick   const CGBitFieldInfo &Info = LV.getBitFieldInfo();
1980e5dd7070Spatrick 
1981e5dd7070Spatrick   // Get the output type.
1982e5dd7070Spatrick   llvm::Type *ResLTy = ConvertType(LV.getType());
1983e5dd7070Spatrick 
1984e5dd7070Spatrick   Address Ptr = LV.getBitFieldAddress();
1985a9ac8606Spatrick   llvm::Value *Val =
1986a9ac8606Spatrick       Builder.CreateLoad(Ptr, LV.isVolatileQualified(), "bf.load");
1987e5dd7070Spatrick 
1988a9ac8606Spatrick   bool UseVolatile = LV.isVolatileQualified() &&
1989a9ac8606Spatrick                      Info.VolatileStorageSize != 0 && isAAPCS(CGM.getTarget());
1990a9ac8606Spatrick   const unsigned Offset = UseVolatile ? Info.VolatileOffset : Info.Offset;
1991a9ac8606Spatrick   const unsigned StorageSize =
1992a9ac8606Spatrick       UseVolatile ? Info.VolatileStorageSize : Info.StorageSize;
1993e5dd7070Spatrick   if (Info.IsSigned) {
1994a9ac8606Spatrick     assert(static_cast<unsigned>(Offset + Info.Size) <= StorageSize);
1995a9ac8606Spatrick     unsigned HighBits = StorageSize - Offset - Info.Size;
1996e5dd7070Spatrick     if (HighBits)
1997e5dd7070Spatrick       Val = Builder.CreateShl(Val, HighBits, "bf.shl");
1998a9ac8606Spatrick     if (Offset + HighBits)
1999a9ac8606Spatrick       Val = Builder.CreateAShr(Val, Offset + HighBits, "bf.ashr");
2000e5dd7070Spatrick   } else {
2001a9ac8606Spatrick     if (Offset)
2002a9ac8606Spatrick       Val = Builder.CreateLShr(Val, Offset, "bf.lshr");
2003a9ac8606Spatrick     if (static_cast<unsigned>(Offset) + Info.Size < StorageSize)
2004a9ac8606Spatrick       Val = Builder.CreateAnd(
2005a9ac8606Spatrick           Val, llvm::APInt::getLowBitsSet(StorageSize, Info.Size), "bf.clear");
2006e5dd7070Spatrick   }
2007e5dd7070Spatrick   Val = Builder.CreateIntCast(Val, ResLTy, Info.IsSigned, "bf.cast");
2008e5dd7070Spatrick   EmitScalarRangeCheck(Val, LV.getType(), Loc);
2009e5dd7070Spatrick   return RValue::get(Val);
2010e5dd7070Spatrick }
2011e5dd7070Spatrick 
2012e5dd7070Spatrick // If this is a reference to a subset of the elements of a vector, create an
2013e5dd7070Spatrick // appropriate shufflevector.
EmitLoadOfExtVectorElementLValue(LValue LV)2014e5dd7070Spatrick RValue CodeGenFunction::EmitLoadOfExtVectorElementLValue(LValue LV) {
2015e5dd7070Spatrick   llvm::Value *Vec = Builder.CreateLoad(LV.getExtVectorAddress(),
2016e5dd7070Spatrick                                         LV.isVolatileQualified());
2017e5dd7070Spatrick 
2018e5dd7070Spatrick   const llvm::Constant *Elts = LV.getExtVectorElts();
2019e5dd7070Spatrick 
2020e5dd7070Spatrick   // If the result of the expression is a non-vector type, we must be extracting
2021e5dd7070Spatrick   // a single element.  Just codegen as an extractelement.
2022e5dd7070Spatrick   const VectorType *ExprVT = LV.getType()->getAs<VectorType>();
2023e5dd7070Spatrick   if (!ExprVT) {
2024e5dd7070Spatrick     unsigned InIdx = getAccessedFieldNo(0, Elts);
2025e5dd7070Spatrick     llvm::Value *Elt = llvm::ConstantInt::get(SizeTy, InIdx);
2026e5dd7070Spatrick     return RValue::get(Builder.CreateExtractElement(Vec, Elt));
2027e5dd7070Spatrick   }
2028e5dd7070Spatrick 
2029e5dd7070Spatrick   // Always use shuffle vector to try to retain the original program structure
2030e5dd7070Spatrick   unsigned NumResultElts = ExprVT->getNumElements();
2031e5dd7070Spatrick 
2032ec727ea7Spatrick   SmallVector<int, 4> Mask;
2033e5dd7070Spatrick   for (unsigned i = 0; i != NumResultElts; ++i)
2034ec727ea7Spatrick     Mask.push_back(getAccessedFieldNo(i, Elts));
2035e5dd7070Spatrick 
2036a9ac8606Spatrick   Vec = Builder.CreateShuffleVector(Vec, Mask);
2037e5dd7070Spatrick   return RValue::get(Vec);
2038e5dd7070Spatrick }
2039e5dd7070Spatrick 
2040e5dd7070Spatrick /// Generates lvalue for partial ext_vector access.
EmitExtVectorElementLValue(LValue LV)2041e5dd7070Spatrick Address CodeGenFunction::EmitExtVectorElementLValue(LValue LV) {
2042e5dd7070Spatrick   Address VectorAddress = LV.getExtVectorAddress();
2043e5dd7070Spatrick   QualType EQT = LV.getType()->castAs<VectorType>()->getElementType();
2044e5dd7070Spatrick   llvm::Type *VectorElementTy = CGM.getTypes().ConvertType(EQT);
2045e5dd7070Spatrick 
2046e5dd7070Spatrick   Address CastToPointerElement =
2047e5dd7070Spatrick     Builder.CreateElementBitCast(VectorAddress, VectorElementTy,
2048e5dd7070Spatrick                                  "conv.ptr.element");
2049e5dd7070Spatrick 
2050e5dd7070Spatrick   const llvm::Constant *Elts = LV.getExtVectorElts();
2051e5dd7070Spatrick   unsigned ix = getAccessedFieldNo(0, Elts);
2052e5dd7070Spatrick 
2053e5dd7070Spatrick   Address VectorBasePtrPlusIx =
2054e5dd7070Spatrick     Builder.CreateConstInBoundsGEP(CastToPointerElement, ix,
2055e5dd7070Spatrick                                    "vector.elt");
2056e5dd7070Spatrick 
2057e5dd7070Spatrick   return VectorBasePtrPlusIx;
2058e5dd7070Spatrick }
2059e5dd7070Spatrick 
2060e5dd7070Spatrick /// Load of global gamed gegisters are always calls to intrinsics.
EmitLoadOfGlobalRegLValue(LValue LV)2061e5dd7070Spatrick RValue CodeGenFunction::EmitLoadOfGlobalRegLValue(LValue LV) {
2062e5dd7070Spatrick   assert((LV.getType()->isIntegerType() || LV.getType()->isPointerType()) &&
2063e5dd7070Spatrick          "Bad type for register variable");
2064e5dd7070Spatrick   llvm::MDNode *RegName = cast<llvm::MDNode>(
2065e5dd7070Spatrick       cast<llvm::MetadataAsValue>(LV.getGlobalReg())->getMetadata());
2066e5dd7070Spatrick 
2067e5dd7070Spatrick   // We accept integer and pointer types only
2068e5dd7070Spatrick   llvm::Type *OrigTy = CGM.getTypes().ConvertType(LV.getType());
2069e5dd7070Spatrick   llvm::Type *Ty = OrigTy;
2070e5dd7070Spatrick   if (OrigTy->isPointerTy())
2071e5dd7070Spatrick     Ty = CGM.getTypes().getDataLayout().getIntPtrType(OrigTy);
2072e5dd7070Spatrick   llvm::Type *Types[] = { Ty };
2073e5dd7070Spatrick 
2074e5dd7070Spatrick   llvm::Function *F = CGM.getIntrinsic(llvm::Intrinsic::read_register, Types);
2075e5dd7070Spatrick   llvm::Value *Call = Builder.CreateCall(
2076e5dd7070Spatrick       F, llvm::MetadataAsValue::get(Ty->getContext(), RegName));
2077e5dd7070Spatrick   if (OrigTy->isPointerTy())
2078e5dd7070Spatrick     Call = Builder.CreateIntToPtr(Call, OrigTy);
2079e5dd7070Spatrick   return RValue::get(Call);
2080e5dd7070Spatrick }
2081e5dd7070Spatrick 
2082e5dd7070Spatrick /// EmitStoreThroughLValue - Store the specified rvalue into the specified
2083e5dd7070Spatrick /// lvalue, where both are guaranteed to the have the same type, and that type
2084e5dd7070Spatrick /// is 'Ty'.
EmitStoreThroughLValue(RValue Src,LValue Dst,bool isInit)2085e5dd7070Spatrick void CodeGenFunction::EmitStoreThroughLValue(RValue Src, LValue Dst,
2086e5dd7070Spatrick                                              bool isInit) {
2087e5dd7070Spatrick   if (!Dst.isSimple()) {
2088e5dd7070Spatrick     if (Dst.isVectorElt()) {
2089e5dd7070Spatrick       // Read/modify/write the vector, inserting the new element.
2090e5dd7070Spatrick       llvm::Value *Vec = Builder.CreateLoad(Dst.getVectorAddress(),
2091e5dd7070Spatrick                                             Dst.isVolatileQualified());
2092*12c85518Srobert       auto *IRStoreTy = dyn_cast<llvm::IntegerType>(Vec->getType());
2093*12c85518Srobert       if (IRStoreTy) {
2094*12c85518Srobert         auto *IRVecTy = llvm::FixedVectorType::get(
2095*12c85518Srobert             Builder.getInt1Ty(), IRStoreTy->getPrimitiveSizeInBits());
2096*12c85518Srobert         Vec = Builder.CreateBitCast(Vec, IRVecTy);
2097*12c85518Srobert         // iN --> <N x i1>.
2098*12c85518Srobert       }
2099e5dd7070Spatrick       Vec = Builder.CreateInsertElement(Vec, Src.getScalarVal(),
2100e5dd7070Spatrick                                         Dst.getVectorIdx(), "vecins");
2101*12c85518Srobert       if (IRStoreTy) {
2102*12c85518Srobert         // <N x i1> --> <iN>.
2103*12c85518Srobert         Vec = Builder.CreateBitCast(Vec, IRStoreTy);
2104*12c85518Srobert       }
2105e5dd7070Spatrick       Builder.CreateStore(Vec, Dst.getVectorAddress(),
2106e5dd7070Spatrick                           Dst.isVolatileQualified());
2107e5dd7070Spatrick       return;
2108e5dd7070Spatrick     }
2109e5dd7070Spatrick 
2110e5dd7070Spatrick     // If this is an update of extended vector elements, insert them as
2111e5dd7070Spatrick     // appropriate.
2112e5dd7070Spatrick     if (Dst.isExtVectorElt())
2113e5dd7070Spatrick       return EmitStoreThroughExtVectorComponentLValue(Src, Dst);
2114e5dd7070Spatrick 
2115e5dd7070Spatrick     if (Dst.isGlobalReg())
2116e5dd7070Spatrick       return EmitStoreThroughGlobalRegLValue(Src, Dst);
2117e5dd7070Spatrick 
2118ec727ea7Spatrick     if (Dst.isMatrixElt()) {
2119*12c85518Srobert       llvm::Value *Idx = Dst.getMatrixIdx();
2120*12c85518Srobert       if (CGM.getCodeGenOpts().OptimizationLevel > 0) {
2121*12c85518Srobert         const auto *const MatTy = Dst.getType()->castAs<ConstantMatrixType>();
2122*12c85518Srobert         llvm::MatrixBuilder MB(Builder);
2123*12c85518Srobert         MB.CreateIndexAssumption(Idx, MatTy->getNumElementsFlattened());
2124*12c85518Srobert       }
2125*12c85518Srobert       llvm::Instruction *Load = Builder.CreateLoad(Dst.getMatrixAddress());
2126*12c85518Srobert       llvm::Value *Vec =
2127*12c85518Srobert           Builder.CreateInsertElement(Load, Src.getScalarVal(), Idx, "matins");
2128ec727ea7Spatrick       Builder.CreateStore(Vec, Dst.getMatrixAddress(),
2129ec727ea7Spatrick                           Dst.isVolatileQualified());
2130ec727ea7Spatrick       return;
2131ec727ea7Spatrick     }
2132ec727ea7Spatrick 
2133e5dd7070Spatrick     assert(Dst.isBitField() && "Unknown LValue type");
2134e5dd7070Spatrick     return EmitStoreThroughBitfieldLValue(Src, Dst);
2135e5dd7070Spatrick   }
2136e5dd7070Spatrick 
2137e5dd7070Spatrick   // There's special magic for assigning into an ARC-qualified l-value.
2138e5dd7070Spatrick   if (Qualifiers::ObjCLifetime Lifetime = Dst.getQuals().getObjCLifetime()) {
2139e5dd7070Spatrick     switch (Lifetime) {
2140e5dd7070Spatrick     case Qualifiers::OCL_None:
2141e5dd7070Spatrick       llvm_unreachable("present but none");
2142e5dd7070Spatrick 
2143e5dd7070Spatrick     case Qualifiers::OCL_ExplicitNone:
2144e5dd7070Spatrick       // nothing special
2145e5dd7070Spatrick       break;
2146e5dd7070Spatrick 
2147e5dd7070Spatrick     case Qualifiers::OCL_Strong:
2148e5dd7070Spatrick       if (isInit) {
2149e5dd7070Spatrick         Src = RValue::get(EmitARCRetain(Dst.getType(), Src.getScalarVal()));
2150e5dd7070Spatrick         break;
2151e5dd7070Spatrick       }
2152e5dd7070Spatrick       EmitARCStoreStrong(Dst, Src.getScalarVal(), /*ignore*/ true);
2153e5dd7070Spatrick       return;
2154e5dd7070Spatrick 
2155e5dd7070Spatrick     case Qualifiers::OCL_Weak:
2156e5dd7070Spatrick       if (isInit)
2157e5dd7070Spatrick         // Initialize and then skip the primitive store.
2158e5dd7070Spatrick         EmitARCInitWeak(Dst.getAddress(*this), Src.getScalarVal());
2159e5dd7070Spatrick       else
2160e5dd7070Spatrick         EmitARCStoreWeak(Dst.getAddress(*this), Src.getScalarVal(),
2161e5dd7070Spatrick                          /*ignore*/ true);
2162e5dd7070Spatrick       return;
2163e5dd7070Spatrick 
2164e5dd7070Spatrick     case Qualifiers::OCL_Autoreleasing:
2165e5dd7070Spatrick       Src = RValue::get(EmitObjCExtendObjectLifetime(Dst.getType(),
2166e5dd7070Spatrick                                                      Src.getScalarVal()));
2167e5dd7070Spatrick       // fall into the normal path
2168e5dd7070Spatrick       break;
2169e5dd7070Spatrick     }
2170e5dd7070Spatrick   }
2171e5dd7070Spatrick 
2172e5dd7070Spatrick   if (Dst.isObjCWeak() && !Dst.isNonGC()) {
2173e5dd7070Spatrick     // load of a __weak object.
2174e5dd7070Spatrick     Address LvalueDst = Dst.getAddress(*this);
2175e5dd7070Spatrick     llvm::Value *src = Src.getScalarVal();
2176e5dd7070Spatrick      CGM.getObjCRuntime().EmitObjCWeakAssign(*this, src, LvalueDst);
2177e5dd7070Spatrick     return;
2178e5dd7070Spatrick   }
2179e5dd7070Spatrick 
2180e5dd7070Spatrick   if (Dst.isObjCStrong() && !Dst.isNonGC()) {
2181e5dd7070Spatrick     // load of a __strong object.
2182e5dd7070Spatrick     Address LvalueDst = Dst.getAddress(*this);
2183e5dd7070Spatrick     llvm::Value *src = Src.getScalarVal();
2184e5dd7070Spatrick     if (Dst.isObjCIvar()) {
2185e5dd7070Spatrick       assert(Dst.getBaseIvarExp() && "BaseIvarExp is NULL");
2186e5dd7070Spatrick       llvm::Type *ResultType = IntPtrTy;
2187e5dd7070Spatrick       Address dst = EmitPointerWithAlignment(Dst.getBaseIvarExp());
2188e5dd7070Spatrick       llvm::Value *RHS = dst.getPointer();
2189e5dd7070Spatrick       RHS = Builder.CreatePtrToInt(RHS, ResultType, "sub.ptr.rhs.cast");
2190e5dd7070Spatrick       llvm::Value *LHS =
2191e5dd7070Spatrick         Builder.CreatePtrToInt(LvalueDst.getPointer(), ResultType,
2192e5dd7070Spatrick                                "sub.ptr.lhs.cast");
2193e5dd7070Spatrick       llvm::Value *BytesBetween = Builder.CreateSub(LHS, RHS, "ivar.offset");
2194e5dd7070Spatrick       CGM.getObjCRuntime().EmitObjCIvarAssign(*this, src, dst,
2195e5dd7070Spatrick                                               BytesBetween);
2196e5dd7070Spatrick     } else if (Dst.isGlobalObjCRef()) {
2197e5dd7070Spatrick       CGM.getObjCRuntime().EmitObjCGlobalAssign(*this, src, LvalueDst,
2198e5dd7070Spatrick                                                 Dst.isThreadLocalRef());
2199e5dd7070Spatrick     }
2200e5dd7070Spatrick     else
2201e5dd7070Spatrick       CGM.getObjCRuntime().EmitObjCStrongCastAssign(*this, src, LvalueDst);
2202e5dd7070Spatrick     return;
2203e5dd7070Spatrick   }
2204e5dd7070Spatrick 
2205e5dd7070Spatrick   assert(Src.isScalar() && "Can't emit an agg store with this method");
2206e5dd7070Spatrick   EmitStoreOfScalar(Src.getScalarVal(), Dst, isInit);
2207e5dd7070Spatrick }
2208e5dd7070Spatrick 
EmitStoreThroughBitfieldLValue(RValue Src,LValue Dst,llvm::Value ** Result)2209e5dd7070Spatrick void CodeGenFunction::EmitStoreThroughBitfieldLValue(RValue Src, LValue Dst,
2210e5dd7070Spatrick                                                      llvm::Value **Result) {
2211e5dd7070Spatrick   const CGBitFieldInfo &Info = Dst.getBitFieldInfo();
2212e5dd7070Spatrick   llvm::Type *ResLTy = ConvertTypeForMem(Dst.getType());
2213e5dd7070Spatrick   Address Ptr = Dst.getBitFieldAddress();
2214e5dd7070Spatrick 
2215e5dd7070Spatrick   // Get the source value, truncated to the width of the bit-field.
2216e5dd7070Spatrick   llvm::Value *SrcVal = Src.getScalarVal();
2217e5dd7070Spatrick 
2218e5dd7070Spatrick   // Cast the source to the storage type and shift it into place.
2219e5dd7070Spatrick   SrcVal = Builder.CreateIntCast(SrcVal, Ptr.getElementType(),
2220e5dd7070Spatrick                                  /*isSigned=*/false);
2221e5dd7070Spatrick   llvm::Value *MaskedVal = SrcVal;
2222e5dd7070Spatrick 
2223a9ac8606Spatrick   const bool UseVolatile =
2224a9ac8606Spatrick       CGM.getCodeGenOpts().AAPCSBitfieldWidth && Dst.isVolatileQualified() &&
2225a9ac8606Spatrick       Info.VolatileStorageSize != 0 && isAAPCS(CGM.getTarget());
2226a9ac8606Spatrick   const unsigned StorageSize =
2227a9ac8606Spatrick       UseVolatile ? Info.VolatileStorageSize : Info.StorageSize;
2228a9ac8606Spatrick   const unsigned Offset = UseVolatile ? Info.VolatileOffset : Info.Offset;
2229e5dd7070Spatrick   // See if there are other bits in the bitfield's storage we'll need to load
2230e5dd7070Spatrick   // and mask together with source before storing.
2231a9ac8606Spatrick   if (StorageSize != Info.Size) {
2232a9ac8606Spatrick     assert(StorageSize > Info.Size && "Invalid bitfield size.");
2233e5dd7070Spatrick     llvm::Value *Val =
2234e5dd7070Spatrick         Builder.CreateLoad(Ptr, Dst.isVolatileQualified(), "bf.load");
2235e5dd7070Spatrick 
2236e5dd7070Spatrick     // Mask the source value as needed.
2237e5dd7070Spatrick     if (!hasBooleanRepresentation(Dst.getType()))
2238a9ac8606Spatrick       SrcVal = Builder.CreateAnd(
2239a9ac8606Spatrick           SrcVal, llvm::APInt::getLowBitsSet(StorageSize, Info.Size),
2240e5dd7070Spatrick           "bf.value");
2241e5dd7070Spatrick     MaskedVal = SrcVal;
2242a9ac8606Spatrick     if (Offset)
2243a9ac8606Spatrick       SrcVal = Builder.CreateShl(SrcVal, Offset, "bf.shl");
2244e5dd7070Spatrick 
2245e5dd7070Spatrick     // Mask out the original value.
2246a9ac8606Spatrick     Val = Builder.CreateAnd(
2247a9ac8606Spatrick         Val, ~llvm::APInt::getBitsSet(StorageSize, Offset, Offset + Info.Size),
2248e5dd7070Spatrick         "bf.clear");
2249e5dd7070Spatrick 
2250e5dd7070Spatrick     // Or together the unchanged values and the source value.
2251e5dd7070Spatrick     SrcVal = Builder.CreateOr(Val, SrcVal, "bf.set");
2252e5dd7070Spatrick   } else {
2253a9ac8606Spatrick     assert(Offset == 0);
2254ec727ea7Spatrick     // According to the AACPS:
2255ec727ea7Spatrick     // When a volatile bit-field is written, and its container does not overlap
2256a9ac8606Spatrick     // with any non-bit-field member, its container must be read exactly once
2257a9ac8606Spatrick     // and written exactly once using the access width appropriate to the type
2258a9ac8606Spatrick     // of the container. The two accesses are not atomic.
2259ec727ea7Spatrick     if (Dst.isVolatileQualified() && isAAPCS(CGM.getTarget()) &&
2260ec727ea7Spatrick         CGM.getCodeGenOpts().ForceAAPCSBitfieldLoad)
2261ec727ea7Spatrick       Builder.CreateLoad(Ptr, true, "bf.load");
2262e5dd7070Spatrick   }
2263e5dd7070Spatrick 
2264e5dd7070Spatrick   // Write the new value back out.
2265e5dd7070Spatrick   Builder.CreateStore(SrcVal, Ptr, Dst.isVolatileQualified());
2266e5dd7070Spatrick 
2267e5dd7070Spatrick   // Return the new value of the bit-field, if requested.
2268e5dd7070Spatrick   if (Result) {
2269e5dd7070Spatrick     llvm::Value *ResultVal = MaskedVal;
2270e5dd7070Spatrick 
2271e5dd7070Spatrick     // Sign extend the value if needed.
2272e5dd7070Spatrick     if (Info.IsSigned) {
2273a9ac8606Spatrick       assert(Info.Size <= StorageSize);
2274a9ac8606Spatrick       unsigned HighBits = StorageSize - Info.Size;
2275e5dd7070Spatrick       if (HighBits) {
2276e5dd7070Spatrick         ResultVal = Builder.CreateShl(ResultVal, HighBits, "bf.result.shl");
2277e5dd7070Spatrick         ResultVal = Builder.CreateAShr(ResultVal, HighBits, "bf.result.ashr");
2278e5dd7070Spatrick       }
2279e5dd7070Spatrick     }
2280e5dd7070Spatrick 
2281e5dd7070Spatrick     ResultVal = Builder.CreateIntCast(ResultVal, ResLTy, Info.IsSigned,
2282e5dd7070Spatrick                                       "bf.result.cast");
2283e5dd7070Spatrick     *Result = EmitFromMemory(ResultVal, Dst.getType());
2284e5dd7070Spatrick   }
2285e5dd7070Spatrick }
2286e5dd7070Spatrick 
EmitStoreThroughExtVectorComponentLValue(RValue Src,LValue Dst)2287e5dd7070Spatrick void CodeGenFunction::EmitStoreThroughExtVectorComponentLValue(RValue Src,
2288e5dd7070Spatrick                                                                LValue Dst) {
2289e5dd7070Spatrick   // This access turns into a read/modify/write of the vector.  Load the input
2290e5dd7070Spatrick   // value now.
2291e5dd7070Spatrick   llvm::Value *Vec = Builder.CreateLoad(Dst.getExtVectorAddress(),
2292e5dd7070Spatrick                                         Dst.isVolatileQualified());
2293e5dd7070Spatrick   const llvm::Constant *Elts = Dst.getExtVectorElts();
2294e5dd7070Spatrick 
2295e5dd7070Spatrick   llvm::Value *SrcVal = Src.getScalarVal();
2296e5dd7070Spatrick 
2297e5dd7070Spatrick   if (const VectorType *VTy = Dst.getType()->getAs<VectorType>()) {
2298e5dd7070Spatrick     unsigned NumSrcElts = VTy->getNumElements();
2299ec727ea7Spatrick     unsigned NumDstElts =
2300a9ac8606Spatrick         cast<llvm::FixedVectorType>(Vec->getType())->getNumElements();
2301e5dd7070Spatrick     if (NumDstElts == NumSrcElts) {
2302e5dd7070Spatrick       // Use shuffle vector is the src and destination are the same number of
2303e5dd7070Spatrick       // elements and restore the vector mask since it is on the side it will be
2304e5dd7070Spatrick       // stored.
2305ec727ea7Spatrick       SmallVector<int, 4> Mask(NumDstElts);
2306e5dd7070Spatrick       for (unsigned i = 0; i != NumSrcElts; ++i)
2307ec727ea7Spatrick         Mask[getAccessedFieldNo(i, Elts)] = i;
2308e5dd7070Spatrick 
2309a9ac8606Spatrick       Vec = Builder.CreateShuffleVector(SrcVal, Mask);
2310e5dd7070Spatrick     } else if (NumDstElts > NumSrcElts) {
2311e5dd7070Spatrick       // Extended the source vector to the same length and then shuffle it
2312e5dd7070Spatrick       // into the destination.
2313e5dd7070Spatrick       // FIXME: since we're shuffling with undef, can we just use the indices
2314e5dd7070Spatrick       //        into that?  This could be simpler.
2315ec727ea7Spatrick       SmallVector<int, 4> ExtMask;
2316e5dd7070Spatrick       for (unsigned i = 0; i != NumSrcElts; ++i)
2317ec727ea7Spatrick         ExtMask.push_back(i);
2318ec727ea7Spatrick       ExtMask.resize(NumDstElts, -1);
2319a9ac8606Spatrick       llvm::Value *ExtSrcVal = Builder.CreateShuffleVector(SrcVal, ExtMask);
2320e5dd7070Spatrick       // build identity
2321ec727ea7Spatrick       SmallVector<int, 4> Mask;
2322e5dd7070Spatrick       for (unsigned i = 0; i != NumDstElts; ++i)
2323ec727ea7Spatrick         Mask.push_back(i);
2324e5dd7070Spatrick 
2325e5dd7070Spatrick       // When the vector size is odd and .odd or .hi is used, the last element
2326e5dd7070Spatrick       // of the Elts constant array will be one past the size of the vector.
2327e5dd7070Spatrick       // Ignore the last element here, if it is greater than the mask size.
2328e5dd7070Spatrick       if (getAccessedFieldNo(NumSrcElts - 1, Elts) == Mask.size())
2329e5dd7070Spatrick         NumSrcElts--;
2330e5dd7070Spatrick 
2331e5dd7070Spatrick       // modify when what gets shuffled in
2332e5dd7070Spatrick       for (unsigned i = 0; i != NumSrcElts; ++i)
2333ec727ea7Spatrick         Mask[getAccessedFieldNo(i, Elts)] = i + NumDstElts;
2334ec727ea7Spatrick       Vec = Builder.CreateShuffleVector(Vec, ExtSrcVal, Mask);
2335e5dd7070Spatrick     } else {
2336e5dd7070Spatrick       // We should never shorten the vector
2337e5dd7070Spatrick       llvm_unreachable("unexpected shorten vector length");
2338e5dd7070Spatrick     }
2339e5dd7070Spatrick   } else {
2340e5dd7070Spatrick     // If the Src is a scalar (not a vector) it must be updating one element.
2341e5dd7070Spatrick     unsigned InIdx = getAccessedFieldNo(0, Elts);
2342e5dd7070Spatrick     llvm::Value *Elt = llvm::ConstantInt::get(SizeTy, InIdx);
2343e5dd7070Spatrick     Vec = Builder.CreateInsertElement(Vec, SrcVal, Elt);
2344e5dd7070Spatrick   }
2345e5dd7070Spatrick 
2346e5dd7070Spatrick   Builder.CreateStore(Vec, Dst.getExtVectorAddress(),
2347e5dd7070Spatrick                       Dst.isVolatileQualified());
2348e5dd7070Spatrick }
2349e5dd7070Spatrick 
2350e5dd7070Spatrick /// Store of global named registers are always calls to intrinsics.
EmitStoreThroughGlobalRegLValue(RValue Src,LValue Dst)2351e5dd7070Spatrick void CodeGenFunction::EmitStoreThroughGlobalRegLValue(RValue Src, LValue Dst) {
2352e5dd7070Spatrick   assert((Dst.getType()->isIntegerType() || Dst.getType()->isPointerType()) &&
2353e5dd7070Spatrick          "Bad type for register variable");
2354e5dd7070Spatrick   llvm::MDNode *RegName = cast<llvm::MDNode>(
2355e5dd7070Spatrick       cast<llvm::MetadataAsValue>(Dst.getGlobalReg())->getMetadata());
2356e5dd7070Spatrick   assert(RegName && "Register LValue is not metadata");
2357e5dd7070Spatrick 
2358e5dd7070Spatrick   // We accept integer and pointer types only
2359e5dd7070Spatrick   llvm::Type *OrigTy = CGM.getTypes().ConvertType(Dst.getType());
2360e5dd7070Spatrick   llvm::Type *Ty = OrigTy;
2361e5dd7070Spatrick   if (OrigTy->isPointerTy())
2362e5dd7070Spatrick     Ty = CGM.getTypes().getDataLayout().getIntPtrType(OrigTy);
2363e5dd7070Spatrick   llvm::Type *Types[] = { Ty };
2364e5dd7070Spatrick 
2365e5dd7070Spatrick   llvm::Function *F = CGM.getIntrinsic(llvm::Intrinsic::write_register, Types);
2366e5dd7070Spatrick   llvm::Value *Value = Src.getScalarVal();
2367e5dd7070Spatrick   if (OrigTy->isPointerTy())
2368e5dd7070Spatrick     Value = Builder.CreatePtrToInt(Value, Ty);
2369e5dd7070Spatrick   Builder.CreateCall(
2370e5dd7070Spatrick       F, {llvm::MetadataAsValue::get(Ty->getContext(), RegName), Value});
2371e5dd7070Spatrick }
2372e5dd7070Spatrick 
2373e5dd7070Spatrick // setObjCGCLValueClass - sets class of the lvalue for the purpose of
2374e5dd7070Spatrick // generating write-barries API. It is currently a global, ivar,
2375e5dd7070Spatrick // or neither.
setObjCGCLValueClass(const ASTContext & Ctx,const Expr * E,LValue & LV,bool IsMemberAccess=false)2376e5dd7070Spatrick static void setObjCGCLValueClass(const ASTContext &Ctx, const Expr *E,
2377e5dd7070Spatrick                                  LValue &LV,
2378e5dd7070Spatrick                                  bool IsMemberAccess=false) {
2379e5dd7070Spatrick   if (Ctx.getLangOpts().getGC() == LangOptions::NonGC)
2380e5dd7070Spatrick     return;
2381e5dd7070Spatrick 
2382e5dd7070Spatrick   if (isa<ObjCIvarRefExpr>(E)) {
2383e5dd7070Spatrick     QualType ExpTy = E->getType();
2384e5dd7070Spatrick     if (IsMemberAccess && ExpTy->isPointerType()) {
2385e5dd7070Spatrick       // If ivar is a structure pointer, assigning to field of
2386e5dd7070Spatrick       // this struct follows gcc's behavior and makes it a non-ivar
2387e5dd7070Spatrick       // writer-barrier conservatively.
2388e5dd7070Spatrick       ExpTy = ExpTy->castAs<PointerType>()->getPointeeType();
2389e5dd7070Spatrick       if (ExpTy->isRecordType()) {
2390e5dd7070Spatrick         LV.setObjCIvar(false);
2391e5dd7070Spatrick         return;
2392e5dd7070Spatrick       }
2393e5dd7070Spatrick     }
2394e5dd7070Spatrick     LV.setObjCIvar(true);
2395e5dd7070Spatrick     auto *Exp = cast<ObjCIvarRefExpr>(const_cast<Expr *>(E));
2396e5dd7070Spatrick     LV.setBaseIvarExp(Exp->getBase());
2397e5dd7070Spatrick     LV.setObjCArray(E->getType()->isArrayType());
2398e5dd7070Spatrick     return;
2399e5dd7070Spatrick   }
2400e5dd7070Spatrick 
2401e5dd7070Spatrick   if (const auto *Exp = dyn_cast<DeclRefExpr>(E)) {
2402e5dd7070Spatrick     if (const auto *VD = dyn_cast<VarDecl>(Exp->getDecl())) {
2403e5dd7070Spatrick       if (VD->hasGlobalStorage()) {
2404e5dd7070Spatrick         LV.setGlobalObjCRef(true);
2405e5dd7070Spatrick         LV.setThreadLocalRef(VD->getTLSKind() != VarDecl::TLS_None);
2406e5dd7070Spatrick       }
2407e5dd7070Spatrick     }
2408e5dd7070Spatrick     LV.setObjCArray(E->getType()->isArrayType());
2409e5dd7070Spatrick     return;
2410e5dd7070Spatrick   }
2411e5dd7070Spatrick 
2412e5dd7070Spatrick   if (const auto *Exp = dyn_cast<UnaryOperator>(E)) {
2413e5dd7070Spatrick     setObjCGCLValueClass(Ctx, Exp->getSubExpr(), LV, IsMemberAccess);
2414e5dd7070Spatrick     return;
2415e5dd7070Spatrick   }
2416e5dd7070Spatrick 
2417e5dd7070Spatrick   if (const auto *Exp = dyn_cast<ParenExpr>(E)) {
2418e5dd7070Spatrick     setObjCGCLValueClass(Ctx, Exp->getSubExpr(), LV, IsMemberAccess);
2419e5dd7070Spatrick     if (LV.isObjCIvar()) {
2420e5dd7070Spatrick       // If cast is to a structure pointer, follow gcc's behavior and make it
2421e5dd7070Spatrick       // a non-ivar write-barrier.
2422e5dd7070Spatrick       QualType ExpTy = E->getType();
2423e5dd7070Spatrick       if (ExpTy->isPointerType())
2424e5dd7070Spatrick         ExpTy = ExpTy->castAs<PointerType>()->getPointeeType();
2425e5dd7070Spatrick       if (ExpTy->isRecordType())
2426e5dd7070Spatrick         LV.setObjCIvar(false);
2427e5dd7070Spatrick     }
2428e5dd7070Spatrick     return;
2429e5dd7070Spatrick   }
2430e5dd7070Spatrick 
2431e5dd7070Spatrick   if (const auto *Exp = dyn_cast<GenericSelectionExpr>(E)) {
2432e5dd7070Spatrick     setObjCGCLValueClass(Ctx, Exp->getResultExpr(), LV);
2433e5dd7070Spatrick     return;
2434e5dd7070Spatrick   }
2435e5dd7070Spatrick 
2436e5dd7070Spatrick   if (const auto *Exp = dyn_cast<ImplicitCastExpr>(E)) {
2437e5dd7070Spatrick     setObjCGCLValueClass(Ctx, Exp->getSubExpr(), LV, IsMemberAccess);
2438e5dd7070Spatrick     return;
2439e5dd7070Spatrick   }
2440e5dd7070Spatrick 
2441e5dd7070Spatrick   if (const auto *Exp = dyn_cast<CStyleCastExpr>(E)) {
2442e5dd7070Spatrick     setObjCGCLValueClass(Ctx, Exp->getSubExpr(), LV, IsMemberAccess);
2443e5dd7070Spatrick     return;
2444e5dd7070Spatrick   }
2445e5dd7070Spatrick 
2446e5dd7070Spatrick   if (const auto *Exp = dyn_cast<ObjCBridgedCastExpr>(E)) {
2447e5dd7070Spatrick     setObjCGCLValueClass(Ctx, Exp->getSubExpr(), LV, IsMemberAccess);
2448e5dd7070Spatrick     return;
2449e5dd7070Spatrick   }
2450e5dd7070Spatrick 
2451e5dd7070Spatrick   if (const auto *Exp = dyn_cast<ArraySubscriptExpr>(E)) {
2452e5dd7070Spatrick     setObjCGCLValueClass(Ctx, Exp->getBase(), LV);
2453e5dd7070Spatrick     if (LV.isObjCIvar() && !LV.isObjCArray())
2454e5dd7070Spatrick       // Using array syntax to assigning to what an ivar points to is not
2455e5dd7070Spatrick       // same as assigning to the ivar itself. {id *Names;} Names[i] = 0;
2456e5dd7070Spatrick       LV.setObjCIvar(false);
2457e5dd7070Spatrick     else if (LV.isGlobalObjCRef() && !LV.isObjCArray())
2458e5dd7070Spatrick       // Using array syntax to assigning to what global points to is not
2459e5dd7070Spatrick       // same as assigning to the global itself. {id *G;} G[i] = 0;
2460e5dd7070Spatrick       LV.setGlobalObjCRef(false);
2461e5dd7070Spatrick     return;
2462e5dd7070Spatrick   }
2463e5dd7070Spatrick 
2464e5dd7070Spatrick   if (const auto *Exp = dyn_cast<MemberExpr>(E)) {
2465e5dd7070Spatrick     setObjCGCLValueClass(Ctx, Exp->getBase(), LV, true);
2466e5dd7070Spatrick     // We don't know if member is an 'ivar', but this flag is looked at
2467e5dd7070Spatrick     // only in the context of LV.isObjCIvar().
2468e5dd7070Spatrick     LV.setObjCArray(E->getType()->isArrayType());
2469e5dd7070Spatrick     return;
2470e5dd7070Spatrick   }
2471e5dd7070Spatrick }
2472e5dd7070Spatrick 
2473e5dd7070Spatrick static llvm::Value *
EmitBitCastOfLValueToProperType(CodeGenFunction & CGF,llvm::Value * V,llvm::Type * IRType,StringRef Name=StringRef ())2474e5dd7070Spatrick EmitBitCastOfLValueToProperType(CodeGenFunction &CGF,
2475e5dd7070Spatrick                                 llvm::Value *V, llvm::Type *IRType,
2476e5dd7070Spatrick                                 StringRef Name = StringRef()) {
2477e5dd7070Spatrick   unsigned AS = cast<llvm::PointerType>(V->getType())->getAddressSpace();
2478e5dd7070Spatrick   return CGF.Builder.CreateBitCast(V, IRType->getPointerTo(AS), Name);
2479e5dd7070Spatrick }
2480e5dd7070Spatrick 
EmitThreadPrivateVarDeclLValue(CodeGenFunction & CGF,const VarDecl * VD,QualType T,Address Addr,llvm::Type * RealVarTy,SourceLocation Loc)2481e5dd7070Spatrick static LValue EmitThreadPrivateVarDeclLValue(
2482e5dd7070Spatrick     CodeGenFunction &CGF, const VarDecl *VD, QualType T, Address Addr,
2483e5dd7070Spatrick     llvm::Type *RealVarTy, SourceLocation Loc) {
2484ec727ea7Spatrick   if (CGF.CGM.getLangOpts().OpenMPIRBuilder)
2485ec727ea7Spatrick     Addr = CodeGenFunction::OMPBuilderCBHelpers::getAddrOfThreadPrivate(
2486ec727ea7Spatrick         CGF, VD, Addr, Loc);
2487ec727ea7Spatrick   else
2488ec727ea7Spatrick     Addr =
2489ec727ea7Spatrick         CGF.CGM.getOpenMPRuntime().getAddrOfThreadPrivate(CGF, VD, Addr, Loc);
2490ec727ea7Spatrick 
2491e5dd7070Spatrick   Addr = CGF.Builder.CreateElementBitCast(Addr, RealVarTy);
2492e5dd7070Spatrick   return CGF.MakeAddrLValue(Addr, T, AlignmentSource::Decl);
2493e5dd7070Spatrick }
2494e5dd7070Spatrick 
emitDeclTargetVarDeclLValue(CodeGenFunction & CGF,const VarDecl * VD,QualType T)2495e5dd7070Spatrick static Address emitDeclTargetVarDeclLValue(CodeGenFunction &CGF,
2496e5dd7070Spatrick                                            const VarDecl *VD, QualType T) {
2497*12c85518Srobert   std::optional<OMPDeclareTargetDeclAttr::MapTypeTy> Res =
2498e5dd7070Spatrick       OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(VD);
2499*12c85518Srobert   // Return an invalid address if variable is MT_To (or MT_Enter starting with
2500*12c85518Srobert   // OpenMP 5.2) and unified memory is not enabled. For all other cases: MT_Link
2501*12c85518Srobert   // and MT_To (or MT_Enter) with unified memory, return a valid address.
2502*12c85518Srobert   if (!Res || ((*Res == OMPDeclareTargetDeclAttr::MT_To ||
2503*12c85518Srobert                 *Res == OMPDeclareTargetDeclAttr::MT_Enter) &&
2504e5dd7070Spatrick                !CGF.CGM.getOpenMPRuntime().hasRequiresUnifiedSharedMemory()))
2505e5dd7070Spatrick     return Address::invalid();
2506e5dd7070Spatrick   assert(((*Res == OMPDeclareTargetDeclAttr::MT_Link) ||
2507*12c85518Srobert           ((*Res == OMPDeclareTargetDeclAttr::MT_To ||
2508*12c85518Srobert             *Res == OMPDeclareTargetDeclAttr::MT_Enter) &&
2509e5dd7070Spatrick            CGF.CGM.getOpenMPRuntime().hasRequiresUnifiedSharedMemory())) &&
2510e5dd7070Spatrick          "Expected link clause OR to clause with unified memory enabled.");
2511e5dd7070Spatrick   QualType PtrTy = CGF.getContext().getPointerType(VD->getType());
2512e5dd7070Spatrick   Address Addr = CGF.CGM.getOpenMPRuntime().getAddrOfDeclareTargetVar(VD);
2513e5dd7070Spatrick   return CGF.EmitLoadOfPointer(Addr, PtrTy->castAs<PointerType>());
2514e5dd7070Spatrick }
2515e5dd7070Spatrick 
2516e5dd7070Spatrick Address
EmitLoadOfReference(LValue RefLVal,LValueBaseInfo * PointeeBaseInfo,TBAAAccessInfo * PointeeTBAAInfo)2517e5dd7070Spatrick CodeGenFunction::EmitLoadOfReference(LValue RefLVal,
2518e5dd7070Spatrick                                      LValueBaseInfo *PointeeBaseInfo,
2519e5dd7070Spatrick                                      TBAAAccessInfo *PointeeTBAAInfo) {
2520e5dd7070Spatrick   llvm::LoadInst *Load =
2521e5dd7070Spatrick       Builder.CreateLoad(RefLVal.getAddress(*this), RefLVal.isVolatile());
2522e5dd7070Spatrick   CGM.DecorateInstructionWithTBAA(Load, RefLVal.getTBAAInfo());
2523e5dd7070Spatrick 
2524*12c85518Srobert   QualType PointeeType = RefLVal.getType()->getPointeeType();
2525ec727ea7Spatrick   CharUnits Align = CGM.getNaturalTypeAlignment(
2526*12c85518Srobert       PointeeType, PointeeBaseInfo, PointeeTBAAInfo,
2527e5dd7070Spatrick       /* forPointeeType= */ true);
2528*12c85518Srobert   return Address(Load, ConvertTypeForMem(PointeeType), Align);
2529e5dd7070Spatrick }
2530e5dd7070Spatrick 
EmitLoadOfReferenceLValue(LValue RefLVal)2531e5dd7070Spatrick LValue CodeGenFunction::EmitLoadOfReferenceLValue(LValue RefLVal) {
2532e5dd7070Spatrick   LValueBaseInfo PointeeBaseInfo;
2533e5dd7070Spatrick   TBAAAccessInfo PointeeTBAAInfo;
2534e5dd7070Spatrick   Address PointeeAddr = EmitLoadOfReference(RefLVal, &PointeeBaseInfo,
2535e5dd7070Spatrick                                             &PointeeTBAAInfo);
2536e5dd7070Spatrick   return MakeAddrLValue(PointeeAddr, RefLVal.getType()->getPointeeType(),
2537e5dd7070Spatrick                         PointeeBaseInfo, PointeeTBAAInfo);
2538e5dd7070Spatrick }
2539e5dd7070Spatrick 
EmitLoadOfPointer(Address Ptr,const PointerType * PtrTy,LValueBaseInfo * BaseInfo,TBAAAccessInfo * TBAAInfo)2540e5dd7070Spatrick Address CodeGenFunction::EmitLoadOfPointer(Address Ptr,
2541e5dd7070Spatrick                                            const PointerType *PtrTy,
2542e5dd7070Spatrick                                            LValueBaseInfo *BaseInfo,
2543e5dd7070Spatrick                                            TBAAAccessInfo *TBAAInfo) {
2544e5dd7070Spatrick   llvm::Value *Addr = Builder.CreateLoad(Ptr);
2545*12c85518Srobert   return Address(Addr, ConvertTypeForMem(PtrTy->getPointeeType()),
2546*12c85518Srobert                  CGM.getNaturalTypeAlignment(PtrTy->getPointeeType(), BaseInfo,
2547*12c85518Srobert                                              TBAAInfo,
2548e5dd7070Spatrick                                              /*forPointeeType=*/true));
2549e5dd7070Spatrick }
2550e5dd7070Spatrick 
EmitLoadOfPointerLValue(Address PtrAddr,const PointerType * PtrTy)2551e5dd7070Spatrick LValue CodeGenFunction::EmitLoadOfPointerLValue(Address PtrAddr,
2552e5dd7070Spatrick                                                 const PointerType *PtrTy) {
2553e5dd7070Spatrick   LValueBaseInfo BaseInfo;
2554e5dd7070Spatrick   TBAAAccessInfo TBAAInfo;
2555e5dd7070Spatrick   Address Addr = EmitLoadOfPointer(PtrAddr, PtrTy, &BaseInfo, &TBAAInfo);
2556e5dd7070Spatrick   return MakeAddrLValue(Addr, PtrTy->getPointeeType(), BaseInfo, TBAAInfo);
2557e5dd7070Spatrick }
2558e5dd7070Spatrick 
EmitGlobalVarDeclLValue(CodeGenFunction & CGF,const Expr * E,const VarDecl * VD)2559e5dd7070Spatrick static LValue EmitGlobalVarDeclLValue(CodeGenFunction &CGF,
2560e5dd7070Spatrick                                       const Expr *E, const VarDecl *VD) {
2561e5dd7070Spatrick   QualType T = E->getType();
2562e5dd7070Spatrick 
2563e5dd7070Spatrick   // If it's thread_local, emit a call to its wrapper function instead.
2564e5dd7070Spatrick   if (VD->getTLSKind() == VarDecl::TLS_Dynamic &&
2565e5dd7070Spatrick       CGF.CGM.getCXXABI().usesThreadWrapperFunction(VD))
2566e5dd7070Spatrick     return CGF.CGM.getCXXABI().EmitThreadLocalVarDeclLValue(CGF, VD, T);
2567e5dd7070Spatrick   // Check if the variable is marked as declare target with link clause in
2568e5dd7070Spatrick   // device codegen.
2569e5dd7070Spatrick   if (CGF.getLangOpts().OpenMPIsDevice) {
2570e5dd7070Spatrick     Address Addr = emitDeclTargetVarDeclLValue(CGF, VD, T);
2571e5dd7070Spatrick     if (Addr.isValid())
2572e5dd7070Spatrick       return CGF.MakeAddrLValue(Addr, T, AlignmentSource::Decl);
2573e5dd7070Spatrick   }
2574e5dd7070Spatrick 
2575e5dd7070Spatrick   llvm::Value *V = CGF.CGM.GetAddrOfGlobalVar(VD);
2576*12c85518Srobert 
2577*12c85518Srobert   if (VD->getTLSKind() != VarDecl::TLS_None)
2578*12c85518Srobert     V = CGF.Builder.CreateThreadLocalAddress(V);
2579*12c85518Srobert 
2580e5dd7070Spatrick   llvm::Type *RealVarTy = CGF.getTypes().ConvertTypeForMem(VD->getType());
2581e5dd7070Spatrick   V = EmitBitCastOfLValueToProperType(CGF, V, RealVarTy);
2582e5dd7070Spatrick   CharUnits Alignment = CGF.getContext().getDeclAlign(VD);
2583*12c85518Srobert   Address Addr(V, RealVarTy, Alignment);
2584e5dd7070Spatrick   // Emit reference to the private copy of the variable if it is an OpenMP
2585e5dd7070Spatrick   // threadprivate variable.
2586e5dd7070Spatrick   if (CGF.getLangOpts().OpenMP && !CGF.getLangOpts().OpenMPSimd &&
2587e5dd7070Spatrick       VD->hasAttr<OMPThreadPrivateDeclAttr>()) {
2588e5dd7070Spatrick     return EmitThreadPrivateVarDeclLValue(CGF, VD, T, Addr, RealVarTy,
2589e5dd7070Spatrick                                           E->getExprLoc());
2590e5dd7070Spatrick   }
2591e5dd7070Spatrick   LValue LV = VD->getType()->isReferenceType() ?
2592e5dd7070Spatrick       CGF.EmitLoadOfReferenceLValue(Addr, VD->getType(),
2593e5dd7070Spatrick                                     AlignmentSource::Decl) :
2594e5dd7070Spatrick       CGF.MakeAddrLValue(Addr, T, AlignmentSource::Decl);
2595e5dd7070Spatrick   setObjCGCLValueClass(CGF.getContext(), E, LV);
2596e5dd7070Spatrick   return LV;
2597e5dd7070Spatrick }
2598e5dd7070Spatrick 
EmitFunctionDeclPointer(CodeGenModule & CGM,GlobalDecl GD)2599e5dd7070Spatrick static llvm::Constant *EmitFunctionDeclPointer(CodeGenModule &CGM,
2600ec727ea7Spatrick                                                GlobalDecl GD) {
2601ec727ea7Spatrick   const FunctionDecl *FD = cast<FunctionDecl>(GD.getDecl());
2602e5dd7070Spatrick   if (FD->hasAttr<WeakRefAttr>()) {
2603e5dd7070Spatrick     ConstantAddress aliasee = CGM.GetWeakRefReference(FD);
2604e5dd7070Spatrick     return aliasee.getPointer();
2605e5dd7070Spatrick   }
2606e5dd7070Spatrick 
2607ec727ea7Spatrick   llvm::Constant *V = CGM.GetAddrOfFunction(GD);
2608e5dd7070Spatrick   if (!FD->hasPrototype()) {
2609e5dd7070Spatrick     if (const FunctionProtoType *Proto =
2610e5dd7070Spatrick             FD->getType()->getAs<FunctionProtoType>()) {
2611e5dd7070Spatrick       // Ugly case: for a K&R-style definition, the type of the definition
2612e5dd7070Spatrick       // isn't the same as the type of a use.  Correct for this with a
2613e5dd7070Spatrick       // bitcast.
2614e5dd7070Spatrick       QualType NoProtoType =
2615e5dd7070Spatrick           CGM.getContext().getFunctionNoProtoType(Proto->getReturnType());
2616e5dd7070Spatrick       NoProtoType = CGM.getContext().getPointerType(NoProtoType);
2617e5dd7070Spatrick       V = llvm::ConstantExpr::getBitCast(V,
2618e5dd7070Spatrick                                       CGM.getTypes().ConvertType(NoProtoType));
2619e5dd7070Spatrick     }
2620e5dd7070Spatrick   }
2621e5dd7070Spatrick   return V;
2622e5dd7070Spatrick }
2623e5dd7070Spatrick 
EmitFunctionDeclLValue(CodeGenFunction & CGF,const Expr * E,GlobalDecl GD)2624ec727ea7Spatrick static LValue EmitFunctionDeclLValue(CodeGenFunction &CGF, const Expr *E,
2625ec727ea7Spatrick                                      GlobalDecl GD) {
2626ec727ea7Spatrick   const FunctionDecl *FD = cast<FunctionDecl>(GD.getDecl());
2627ec727ea7Spatrick   llvm::Value *V = EmitFunctionDeclPointer(CGF.CGM, GD);
2628e5dd7070Spatrick   CharUnits Alignment = CGF.getContext().getDeclAlign(FD);
2629e5dd7070Spatrick   return CGF.MakeAddrLValue(V, E->getType(), Alignment,
2630e5dd7070Spatrick                             AlignmentSource::Decl);
2631e5dd7070Spatrick }
2632e5dd7070Spatrick 
EmitCapturedFieldLValue(CodeGenFunction & CGF,const FieldDecl * FD,llvm::Value * ThisValue)2633e5dd7070Spatrick static LValue EmitCapturedFieldLValue(CodeGenFunction &CGF, const FieldDecl *FD,
2634e5dd7070Spatrick                                       llvm::Value *ThisValue) {
2635e5dd7070Spatrick   QualType TagType = CGF.getContext().getTagDeclType(FD->getParent());
2636e5dd7070Spatrick   LValue LV = CGF.MakeNaturalAlignAddrLValue(ThisValue, TagType);
2637e5dd7070Spatrick   return CGF.EmitLValueForField(LV, FD);
2638e5dd7070Spatrick }
2639e5dd7070Spatrick 
2640e5dd7070Spatrick /// Named Registers are named metadata pointing to the register name
2641e5dd7070Spatrick /// which will be read from/written to as an argument to the intrinsic
2642e5dd7070Spatrick /// @llvm.read/write_register.
2643e5dd7070Spatrick /// So far, only the name is being passed down, but other options such as
2644e5dd7070Spatrick /// register type, allocation type or even optimization options could be
2645e5dd7070Spatrick /// passed down via the metadata node.
EmitGlobalNamedRegister(const VarDecl * VD,CodeGenModule & CGM)2646e5dd7070Spatrick static LValue EmitGlobalNamedRegister(const VarDecl *VD, CodeGenModule &CGM) {
2647e5dd7070Spatrick   SmallString<64> Name("llvm.named.register.");
2648e5dd7070Spatrick   AsmLabelAttr *Asm = VD->getAttr<AsmLabelAttr>();
2649e5dd7070Spatrick   assert(Asm->getLabel().size() < 64-Name.size() &&
2650e5dd7070Spatrick       "Register name too big");
2651e5dd7070Spatrick   Name.append(Asm->getLabel());
2652e5dd7070Spatrick   llvm::NamedMDNode *M =
2653e5dd7070Spatrick     CGM.getModule().getOrInsertNamedMetadata(Name);
2654e5dd7070Spatrick   if (M->getNumOperands() == 0) {
2655e5dd7070Spatrick     llvm::MDString *Str = llvm::MDString::get(CGM.getLLVMContext(),
2656e5dd7070Spatrick                                               Asm->getLabel());
2657e5dd7070Spatrick     llvm::Metadata *Ops[] = {Str};
2658e5dd7070Spatrick     M->addOperand(llvm::MDNode::get(CGM.getLLVMContext(), Ops));
2659e5dd7070Spatrick   }
2660e5dd7070Spatrick 
2661e5dd7070Spatrick   CharUnits Alignment = CGM.getContext().getDeclAlign(VD);
2662e5dd7070Spatrick 
2663e5dd7070Spatrick   llvm::Value *Ptr =
2664e5dd7070Spatrick     llvm::MetadataAsValue::get(CGM.getLLVMContext(), M->getOperand(0));
2665*12c85518Srobert   return LValue::MakeGlobalReg(Ptr, Alignment, VD->getType());
2666e5dd7070Spatrick }
2667e5dd7070Spatrick 
2668e5dd7070Spatrick /// Determine whether we can emit a reference to \p VD from the current
2669e5dd7070Spatrick /// context, despite not necessarily having seen an odr-use of the variable in
2670e5dd7070Spatrick /// this context.
canEmitSpuriousReferenceToVariable(CodeGenFunction & CGF,const DeclRefExpr * E,const VarDecl * VD,bool IsConstant)2671e5dd7070Spatrick static bool canEmitSpuriousReferenceToVariable(CodeGenFunction &CGF,
2672e5dd7070Spatrick                                                const DeclRefExpr *E,
2673e5dd7070Spatrick                                                const VarDecl *VD,
2674e5dd7070Spatrick                                                bool IsConstant) {
2675e5dd7070Spatrick   // For a variable declared in an enclosing scope, do not emit a spurious
2676e5dd7070Spatrick   // reference even if we have a capture, as that will emit an unwarranted
2677e5dd7070Spatrick   // reference to our capture state, and will likely generate worse code than
2678e5dd7070Spatrick   // emitting a local copy.
2679e5dd7070Spatrick   if (E->refersToEnclosingVariableOrCapture())
2680e5dd7070Spatrick     return false;
2681e5dd7070Spatrick 
2682e5dd7070Spatrick   // For a local declaration declared in this function, we can always reference
2683e5dd7070Spatrick   // it even if we don't have an odr-use.
2684e5dd7070Spatrick   if (VD->hasLocalStorage()) {
2685e5dd7070Spatrick     return VD->getDeclContext() ==
2686e5dd7070Spatrick            dyn_cast_or_null<DeclContext>(CGF.CurCodeDecl);
2687e5dd7070Spatrick   }
2688e5dd7070Spatrick 
2689e5dd7070Spatrick   // For a global declaration, we can emit a reference to it if we know
2690e5dd7070Spatrick   // for sure that we are able to emit a definition of it.
2691e5dd7070Spatrick   VD = VD->getDefinition(CGF.getContext());
2692e5dd7070Spatrick   if (!VD)
2693e5dd7070Spatrick     return false;
2694e5dd7070Spatrick 
2695e5dd7070Spatrick   // Don't emit a spurious reference if it might be to a variable that only
2696e5dd7070Spatrick   // exists on a different device / target.
2697e5dd7070Spatrick   // FIXME: This is unnecessarily broad. Check whether this would actually be a
2698e5dd7070Spatrick   // cross-target reference.
2699e5dd7070Spatrick   if (CGF.getLangOpts().OpenMP || CGF.getLangOpts().CUDA ||
2700e5dd7070Spatrick       CGF.getLangOpts().OpenCL) {
2701e5dd7070Spatrick     return false;
2702e5dd7070Spatrick   }
2703e5dd7070Spatrick 
2704e5dd7070Spatrick   // We can emit a spurious reference only if the linkage implies that we'll
2705e5dd7070Spatrick   // be emitting a non-interposable symbol that will be retained until link
2706e5dd7070Spatrick   // time.
2707e5dd7070Spatrick   switch (CGF.CGM.getLLVMLinkageVarDefinition(VD, IsConstant)) {
2708e5dd7070Spatrick   case llvm::GlobalValue::ExternalLinkage:
2709e5dd7070Spatrick   case llvm::GlobalValue::LinkOnceODRLinkage:
2710e5dd7070Spatrick   case llvm::GlobalValue::WeakODRLinkage:
2711e5dd7070Spatrick   case llvm::GlobalValue::InternalLinkage:
2712e5dd7070Spatrick   case llvm::GlobalValue::PrivateLinkage:
2713e5dd7070Spatrick     return true;
2714e5dd7070Spatrick   default:
2715e5dd7070Spatrick     return false;
2716e5dd7070Spatrick   }
2717e5dd7070Spatrick }
2718e5dd7070Spatrick 
EmitDeclRefLValue(const DeclRefExpr * E)2719e5dd7070Spatrick LValue CodeGenFunction::EmitDeclRefLValue(const DeclRefExpr *E) {
2720e5dd7070Spatrick   const NamedDecl *ND = E->getDecl();
2721e5dd7070Spatrick   QualType T = E->getType();
2722e5dd7070Spatrick 
2723e5dd7070Spatrick   assert(E->isNonOdrUse() != NOUR_Unevaluated &&
2724e5dd7070Spatrick          "should not emit an unevaluated operand");
2725e5dd7070Spatrick 
2726e5dd7070Spatrick   if (const auto *VD = dyn_cast<VarDecl>(ND)) {
2727e5dd7070Spatrick     // Global Named registers access via intrinsics only
2728e5dd7070Spatrick     if (VD->getStorageClass() == SC_Register &&
2729e5dd7070Spatrick         VD->hasAttr<AsmLabelAttr>() && !VD->isLocalVarDecl())
2730e5dd7070Spatrick       return EmitGlobalNamedRegister(VD, CGM);
2731e5dd7070Spatrick 
2732e5dd7070Spatrick     // If this DeclRefExpr does not constitute an odr-use of the variable,
2733e5dd7070Spatrick     // we're not permitted to emit a reference to it in general, and it might
2734e5dd7070Spatrick     // not be captured if capture would be necessary for a use. Emit the
2735e5dd7070Spatrick     // constant value directly instead.
2736e5dd7070Spatrick     if (E->isNonOdrUse() == NOUR_Constant &&
2737e5dd7070Spatrick         (VD->getType()->isReferenceType() ||
2738e5dd7070Spatrick          !canEmitSpuriousReferenceToVariable(*this, E, VD, true))) {
2739e5dd7070Spatrick       VD->getAnyInitializer(VD);
2740e5dd7070Spatrick       llvm::Constant *Val = ConstantEmitter(*this).emitAbstract(
2741e5dd7070Spatrick           E->getLocation(), *VD->evaluateValue(), VD->getType());
2742e5dd7070Spatrick       assert(Val && "failed to emit constant expression");
2743e5dd7070Spatrick 
2744e5dd7070Spatrick       Address Addr = Address::invalid();
2745e5dd7070Spatrick       if (!VD->getType()->isReferenceType()) {
2746e5dd7070Spatrick         // Spill the constant value to a global.
2747e5dd7070Spatrick         Addr = CGM.createUnnamedGlobalFrom(*VD, Val,
2748e5dd7070Spatrick                                            getContext().getDeclAlign(VD));
2749e5dd7070Spatrick         llvm::Type *VarTy = getTypes().ConvertTypeForMem(VD->getType());
2750e5dd7070Spatrick         auto *PTy = llvm::PointerType::get(
2751*12c85518Srobert             VarTy, getTypes().getTargetAddressSpace(VD->getType()));
2752*12c85518Srobert         Addr = Builder.CreatePointerBitCastOrAddrSpaceCast(Addr, PTy, VarTy);
2753e5dd7070Spatrick       } else {
2754e5dd7070Spatrick         // Should we be using the alignment of the constant pointer we emitted?
2755e5dd7070Spatrick         CharUnits Alignment =
2756ec727ea7Spatrick             CGM.getNaturalTypeAlignment(E->getType(),
2757e5dd7070Spatrick                                         /* BaseInfo= */ nullptr,
2758e5dd7070Spatrick                                         /* TBAAInfo= */ nullptr,
2759e5dd7070Spatrick                                         /* forPointeeType= */ true);
2760*12c85518Srobert         Addr = Address(Val, ConvertTypeForMem(E->getType()), Alignment);
2761e5dd7070Spatrick       }
2762e5dd7070Spatrick       return MakeAddrLValue(Addr, T, AlignmentSource::Decl);
2763e5dd7070Spatrick     }
2764e5dd7070Spatrick 
2765e5dd7070Spatrick     // FIXME: Handle other kinds of non-odr-use DeclRefExprs.
2766e5dd7070Spatrick 
2767e5dd7070Spatrick     // Check for captured variables.
2768e5dd7070Spatrick     if (E->refersToEnclosingVariableOrCapture()) {
2769e5dd7070Spatrick       VD = VD->getCanonicalDecl();
2770e5dd7070Spatrick       if (auto *FD = LambdaCaptureFields.lookup(VD))
2771e5dd7070Spatrick         return EmitCapturedFieldLValue(*this, FD, CXXABIThisValue);
2772e5dd7070Spatrick       if (CapturedStmtInfo) {
2773e5dd7070Spatrick         auto I = LocalDeclMap.find(VD);
2774e5dd7070Spatrick         if (I != LocalDeclMap.end()) {
2775e5dd7070Spatrick           LValue CapLVal;
2776e5dd7070Spatrick           if (VD->getType()->isReferenceType())
2777e5dd7070Spatrick             CapLVal = EmitLoadOfReferenceLValue(I->second, VD->getType(),
2778e5dd7070Spatrick                                                 AlignmentSource::Decl);
2779e5dd7070Spatrick           else
2780e5dd7070Spatrick             CapLVal = MakeAddrLValue(I->second, T);
2781e5dd7070Spatrick           // Mark lvalue as nontemporal if the variable is marked as nontemporal
2782e5dd7070Spatrick           // in simd context.
2783e5dd7070Spatrick           if (getLangOpts().OpenMP &&
2784e5dd7070Spatrick               CGM.getOpenMPRuntime().isNontemporalDecl(VD))
2785e5dd7070Spatrick             CapLVal.setNontemporal(/*Value=*/true);
2786e5dd7070Spatrick           return CapLVal;
2787e5dd7070Spatrick         }
2788e5dd7070Spatrick         LValue CapLVal =
2789e5dd7070Spatrick             EmitCapturedFieldLValue(*this, CapturedStmtInfo->lookup(VD),
2790e5dd7070Spatrick                                     CapturedStmtInfo->getContextValue());
2791*12c85518Srobert         Address LValueAddress = CapLVal.getAddress(*this);
2792e5dd7070Spatrick         CapLVal = MakeAddrLValue(
2793*12c85518Srobert             Address(LValueAddress.getPointer(), LValueAddress.getElementType(),
2794*12c85518Srobert                     getContext().getDeclAlign(VD)),
2795e5dd7070Spatrick             CapLVal.getType(), LValueBaseInfo(AlignmentSource::Decl),
2796e5dd7070Spatrick             CapLVal.getTBAAInfo());
2797e5dd7070Spatrick         // Mark lvalue as nontemporal if the variable is marked as nontemporal
2798e5dd7070Spatrick         // in simd context.
2799e5dd7070Spatrick         if (getLangOpts().OpenMP &&
2800e5dd7070Spatrick             CGM.getOpenMPRuntime().isNontemporalDecl(VD))
2801e5dd7070Spatrick           CapLVal.setNontemporal(/*Value=*/true);
2802e5dd7070Spatrick         return CapLVal;
2803e5dd7070Spatrick       }
2804e5dd7070Spatrick 
2805e5dd7070Spatrick       assert(isa<BlockDecl>(CurCodeDecl));
2806e5dd7070Spatrick       Address addr = GetAddrOfBlockDecl(VD);
2807e5dd7070Spatrick       return MakeAddrLValue(addr, T, AlignmentSource::Decl);
2808e5dd7070Spatrick     }
2809e5dd7070Spatrick   }
2810e5dd7070Spatrick 
2811e5dd7070Spatrick   // FIXME: We should be able to assert this for FunctionDecls as well!
2812e5dd7070Spatrick   // FIXME: We should be able to assert this for all DeclRefExprs, not just
2813e5dd7070Spatrick   // those with a valid source location.
2814e5dd7070Spatrick   assert((ND->isUsed(false) || !isa<VarDecl>(ND) || E->isNonOdrUse() ||
2815e5dd7070Spatrick           !E->getLocation().isValid()) &&
2816e5dd7070Spatrick          "Should not use decl without marking it used!");
2817e5dd7070Spatrick 
2818e5dd7070Spatrick   if (ND->hasAttr<WeakRefAttr>()) {
2819e5dd7070Spatrick     const auto *VD = cast<ValueDecl>(ND);
2820e5dd7070Spatrick     ConstantAddress Aliasee = CGM.GetWeakRefReference(VD);
2821e5dd7070Spatrick     return MakeAddrLValue(Aliasee, T, AlignmentSource::Decl);
2822e5dd7070Spatrick   }
2823e5dd7070Spatrick 
2824e5dd7070Spatrick   if (const auto *VD = dyn_cast<VarDecl>(ND)) {
2825e5dd7070Spatrick     // Check if this is a global variable.
2826e5dd7070Spatrick     if (VD->hasLinkage() || VD->isStaticDataMember())
2827e5dd7070Spatrick       return EmitGlobalVarDeclLValue(*this, E, VD);
2828e5dd7070Spatrick 
2829e5dd7070Spatrick     Address addr = Address::invalid();
2830e5dd7070Spatrick 
2831e5dd7070Spatrick     // The variable should generally be present in the local decl map.
2832e5dd7070Spatrick     auto iter = LocalDeclMap.find(VD);
2833e5dd7070Spatrick     if (iter != LocalDeclMap.end()) {
2834e5dd7070Spatrick       addr = iter->second;
2835e5dd7070Spatrick 
2836e5dd7070Spatrick     // Otherwise, it might be static local we haven't emitted yet for
2837e5dd7070Spatrick     // some reason; most likely, because it's in an outer function.
2838e5dd7070Spatrick     } else if (VD->isStaticLocal()) {
2839*12c85518Srobert       llvm::Constant *var = CGM.getOrCreateStaticVarDecl(
2840*12c85518Srobert           *VD, CGM.getLLVMLinkageVarDefinition(VD, /*IsConstant=*/false));
2841*12c85518Srobert       addr = Address(
2842*12c85518Srobert           var, ConvertTypeForMem(VD->getType()), getContext().getDeclAlign(VD));
2843e5dd7070Spatrick 
2844e5dd7070Spatrick     // No other cases for now.
2845e5dd7070Spatrick     } else {
2846e5dd7070Spatrick       llvm_unreachable("DeclRefExpr for Decl not entered in LocalDeclMap?");
2847e5dd7070Spatrick     }
2848e5dd7070Spatrick 
2849*12c85518Srobert     // Handle threadlocal function locals.
2850*12c85518Srobert     if (VD->getTLSKind() != VarDecl::TLS_None)
2851*12c85518Srobert       addr =
2852*12c85518Srobert           addr.withPointer(Builder.CreateThreadLocalAddress(addr.getPointer()));
2853e5dd7070Spatrick 
2854e5dd7070Spatrick     // Check for OpenMP threadprivate variables.
2855e5dd7070Spatrick     if (getLangOpts().OpenMP && !getLangOpts().OpenMPSimd &&
2856e5dd7070Spatrick         VD->hasAttr<OMPThreadPrivateDeclAttr>()) {
2857e5dd7070Spatrick       return EmitThreadPrivateVarDeclLValue(
2858e5dd7070Spatrick           *this, VD, T, addr, getTypes().ConvertTypeForMem(VD->getType()),
2859e5dd7070Spatrick           E->getExprLoc());
2860e5dd7070Spatrick     }
2861e5dd7070Spatrick 
2862e5dd7070Spatrick     // Drill into block byref variables.
2863e5dd7070Spatrick     bool isBlockByref = VD->isEscapingByref();
2864e5dd7070Spatrick     if (isBlockByref) {
2865e5dd7070Spatrick       addr = emitBlockByrefAddress(addr, VD);
2866e5dd7070Spatrick     }
2867e5dd7070Spatrick 
2868e5dd7070Spatrick     // Drill into reference types.
2869e5dd7070Spatrick     LValue LV = VD->getType()->isReferenceType() ?
2870e5dd7070Spatrick         EmitLoadOfReferenceLValue(addr, VD->getType(), AlignmentSource::Decl) :
2871e5dd7070Spatrick         MakeAddrLValue(addr, T, AlignmentSource::Decl);
2872e5dd7070Spatrick 
2873e5dd7070Spatrick     bool isLocalStorage = VD->hasLocalStorage();
2874e5dd7070Spatrick 
2875e5dd7070Spatrick     bool NonGCable = isLocalStorage &&
2876e5dd7070Spatrick                      !VD->getType()->isReferenceType() &&
2877e5dd7070Spatrick                      !isBlockByref;
2878e5dd7070Spatrick     if (NonGCable) {
2879e5dd7070Spatrick       LV.getQuals().removeObjCGCAttr();
2880e5dd7070Spatrick       LV.setNonGC(true);
2881e5dd7070Spatrick     }
2882e5dd7070Spatrick 
2883e5dd7070Spatrick     bool isImpreciseLifetime =
2884e5dd7070Spatrick       (isLocalStorage && !VD->hasAttr<ObjCPreciseLifetimeAttr>());
2885e5dd7070Spatrick     if (isImpreciseLifetime)
2886e5dd7070Spatrick       LV.setARCPreciseLifetime(ARCImpreciseLifetime);
2887e5dd7070Spatrick     setObjCGCLValueClass(getContext(), E, LV);
2888e5dd7070Spatrick     return LV;
2889e5dd7070Spatrick   }
2890e5dd7070Spatrick 
2891a9ac8606Spatrick   if (const auto *FD = dyn_cast<FunctionDecl>(ND)) {
2892a9ac8606Spatrick     LValue LV = EmitFunctionDeclLValue(*this, E, FD);
2893a9ac8606Spatrick 
2894a9ac8606Spatrick     // Emit debuginfo for the function declaration if the target wants to.
2895a9ac8606Spatrick     if (getContext().getTargetInfo().allowDebugInfoForExternalRef()) {
2896a9ac8606Spatrick       if (CGDebugInfo *DI = CGM.getModuleDebugInfo()) {
2897a9ac8606Spatrick         auto *Fn =
2898a9ac8606Spatrick             cast<llvm::Function>(LV.getPointer(*this)->stripPointerCasts());
2899a9ac8606Spatrick         if (!Fn->getSubprogram())
2900a9ac8606Spatrick           DI->EmitFunctionDecl(FD, FD->getLocation(), T, Fn);
2901a9ac8606Spatrick       }
2902a9ac8606Spatrick     }
2903a9ac8606Spatrick 
2904a9ac8606Spatrick     return LV;
2905a9ac8606Spatrick   }
2906e5dd7070Spatrick 
2907e5dd7070Spatrick   // FIXME: While we're emitting a binding from an enclosing scope, all other
2908e5dd7070Spatrick   // DeclRefExprs we see should be implicitly treated as if they also refer to
2909e5dd7070Spatrick   // an enclosing scope.
2910*12c85518Srobert   if (const auto *BD = dyn_cast<BindingDecl>(ND)) {
2911*12c85518Srobert     if (E->refersToEnclosingVariableOrCapture()) {
2912*12c85518Srobert       auto *FD = LambdaCaptureFields.lookup(BD);
2913*12c85518Srobert       return EmitCapturedFieldLValue(*this, FD, CXXABIThisValue);
2914*12c85518Srobert     }
2915e5dd7070Spatrick     return EmitLValue(BD->getBinding());
2916*12c85518Srobert   }
2917e5dd7070Spatrick 
2918ec727ea7Spatrick   // We can form DeclRefExprs naming GUID declarations when reconstituting
2919ec727ea7Spatrick   // non-type template parameters into expressions.
2920ec727ea7Spatrick   if (const auto *GD = dyn_cast<MSGuidDecl>(ND))
2921ec727ea7Spatrick     return MakeAddrLValue(CGM.GetAddrOfMSGuidDecl(GD), T,
2922ec727ea7Spatrick                           AlignmentSource::Decl);
2923ec727ea7Spatrick 
2924a9ac8606Spatrick   if (const auto *TPO = dyn_cast<TemplateParamObjectDecl>(ND))
2925a9ac8606Spatrick     return MakeAddrLValue(CGM.GetAddrOfTemplateParamObject(TPO), T,
2926a9ac8606Spatrick                           AlignmentSource::Decl);
2927a9ac8606Spatrick 
2928e5dd7070Spatrick   llvm_unreachable("Unhandled DeclRefExpr");
2929e5dd7070Spatrick }
2930e5dd7070Spatrick 
EmitUnaryOpLValue(const UnaryOperator * E)2931e5dd7070Spatrick LValue CodeGenFunction::EmitUnaryOpLValue(const UnaryOperator *E) {
2932e5dd7070Spatrick   // __extension__ doesn't affect lvalue-ness.
2933e5dd7070Spatrick   if (E->getOpcode() == UO_Extension)
2934e5dd7070Spatrick     return EmitLValue(E->getSubExpr());
2935e5dd7070Spatrick 
2936e5dd7070Spatrick   QualType ExprTy = getContext().getCanonicalType(E->getSubExpr()->getType());
2937e5dd7070Spatrick   switch (E->getOpcode()) {
2938e5dd7070Spatrick   default: llvm_unreachable("Unknown unary operator lvalue!");
2939e5dd7070Spatrick   case UO_Deref: {
2940e5dd7070Spatrick     QualType T = E->getSubExpr()->getType()->getPointeeType();
2941e5dd7070Spatrick     assert(!T.isNull() && "CodeGenFunction::EmitUnaryOpLValue: Illegal type");
2942e5dd7070Spatrick 
2943e5dd7070Spatrick     LValueBaseInfo BaseInfo;
2944e5dd7070Spatrick     TBAAAccessInfo TBAAInfo;
2945e5dd7070Spatrick     Address Addr = EmitPointerWithAlignment(E->getSubExpr(), &BaseInfo,
2946e5dd7070Spatrick                                             &TBAAInfo);
2947e5dd7070Spatrick     LValue LV = MakeAddrLValue(Addr, T, BaseInfo, TBAAInfo);
2948e5dd7070Spatrick     LV.getQuals().setAddressSpace(ExprTy.getAddressSpace());
2949e5dd7070Spatrick 
2950e5dd7070Spatrick     // We should not generate __weak write barrier on indirect reference
2951e5dd7070Spatrick     // of a pointer to object; as in void foo (__weak id *param); *param = 0;
2952e5dd7070Spatrick     // But, we continue to generate __strong write barrier on indirect write
2953e5dd7070Spatrick     // into a pointer to object.
2954e5dd7070Spatrick     if (getLangOpts().ObjC &&
2955e5dd7070Spatrick         getLangOpts().getGC() != LangOptions::NonGC &&
2956e5dd7070Spatrick         LV.isObjCWeak())
2957e5dd7070Spatrick       LV.setNonGC(!E->isOBJCGCCandidate(getContext()));
2958e5dd7070Spatrick     return LV;
2959e5dd7070Spatrick   }
2960e5dd7070Spatrick   case UO_Real:
2961e5dd7070Spatrick   case UO_Imag: {
2962e5dd7070Spatrick     LValue LV = EmitLValue(E->getSubExpr());
2963e5dd7070Spatrick     assert(LV.isSimple() && "real/imag on non-ordinary l-value");
2964e5dd7070Spatrick 
2965e5dd7070Spatrick     // __real is valid on scalars.  This is a faster way of testing that.
2966e5dd7070Spatrick     // __imag can only produce an rvalue on scalars.
2967e5dd7070Spatrick     if (E->getOpcode() == UO_Real &&
2968e5dd7070Spatrick         !LV.getAddress(*this).getElementType()->isStructTy()) {
2969e5dd7070Spatrick       assert(E->getSubExpr()->getType()->isArithmeticType());
2970e5dd7070Spatrick       return LV;
2971e5dd7070Spatrick     }
2972e5dd7070Spatrick 
2973e5dd7070Spatrick     QualType T = ExprTy->castAs<ComplexType>()->getElementType();
2974e5dd7070Spatrick 
2975e5dd7070Spatrick     Address Component =
2976e5dd7070Spatrick         (E->getOpcode() == UO_Real
2977e5dd7070Spatrick              ? emitAddrOfRealComponent(LV.getAddress(*this), LV.getType())
2978e5dd7070Spatrick              : emitAddrOfImagComponent(LV.getAddress(*this), LV.getType()));
2979e5dd7070Spatrick     LValue ElemLV = MakeAddrLValue(Component, T, LV.getBaseInfo(),
2980e5dd7070Spatrick                                    CGM.getTBAAInfoForSubobject(LV, T));
2981e5dd7070Spatrick     ElemLV.getQuals().addQualifiers(LV.getQuals());
2982e5dd7070Spatrick     return ElemLV;
2983e5dd7070Spatrick   }
2984e5dd7070Spatrick   case UO_PreInc:
2985e5dd7070Spatrick   case UO_PreDec: {
2986e5dd7070Spatrick     LValue LV = EmitLValue(E->getSubExpr());
2987e5dd7070Spatrick     bool isInc = E->getOpcode() == UO_PreInc;
2988e5dd7070Spatrick 
2989e5dd7070Spatrick     if (E->getType()->isAnyComplexType())
2990e5dd7070Spatrick       EmitComplexPrePostIncDec(E, LV, isInc, true/*isPre*/);
2991e5dd7070Spatrick     else
2992e5dd7070Spatrick       EmitScalarPrePostIncDec(E, LV, isInc, true/*isPre*/);
2993e5dd7070Spatrick     return LV;
2994e5dd7070Spatrick   }
2995e5dd7070Spatrick   }
2996e5dd7070Spatrick }
2997e5dd7070Spatrick 
EmitStringLiteralLValue(const StringLiteral * E)2998e5dd7070Spatrick LValue CodeGenFunction::EmitStringLiteralLValue(const StringLiteral *E) {
2999e5dd7070Spatrick   return MakeAddrLValue(CGM.GetAddrOfConstantStringFromLiteral(E),
3000e5dd7070Spatrick                         E->getType(), AlignmentSource::Decl);
3001e5dd7070Spatrick }
3002e5dd7070Spatrick 
EmitObjCEncodeExprLValue(const ObjCEncodeExpr * E)3003e5dd7070Spatrick LValue CodeGenFunction::EmitObjCEncodeExprLValue(const ObjCEncodeExpr *E) {
3004e5dd7070Spatrick   return MakeAddrLValue(CGM.GetAddrOfConstantStringFromObjCEncode(E),
3005e5dd7070Spatrick                         E->getType(), AlignmentSource::Decl);
3006e5dd7070Spatrick }
3007e5dd7070Spatrick 
EmitPredefinedLValue(const PredefinedExpr * E)3008e5dd7070Spatrick LValue CodeGenFunction::EmitPredefinedLValue(const PredefinedExpr *E) {
3009e5dd7070Spatrick   auto SL = E->getFunctionName();
3010e5dd7070Spatrick   assert(SL != nullptr && "No StringLiteral name in PredefinedExpr");
3011e5dd7070Spatrick   StringRef FnName = CurFn->getName();
3012e5dd7070Spatrick   if (FnName.startswith("\01"))
3013e5dd7070Spatrick     FnName = FnName.substr(1);
3014e5dd7070Spatrick   StringRef NameItems[] = {
3015e5dd7070Spatrick       PredefinedExpr::getIdentKindName(E->getIdentKind()), FnName};
3016e5dd7070Spatrick   std::string GVName = llvm::join(NameItems, NameItems + 2, ".");
3017e5dd7070Spatrick   if (auto *BD = dyn_cast_or_null<BlockDecl>(CurCodeDecl)) {
3018ec727ea7Spatrick     std::string Name = std::string(SL->getString());
3019e5dd7070Spatrick     if (!Name.empty()) {
3020e5dd7070Spatrick       unsigned Discriminator =
3021e5dd7070Spatrick           CGM.getCXXABI().getMangleContext().getBlockId(BD, true);
3022e5dd7070Spatrick       if (Discriminator)
3023e5dd7070Spatrick         Name += "_" + Twine(Discriminator + 1).str();
3024e5dd7070Spatrick       auto C = CGM.GetAddrOfConstantCString(Name, GVName.c_str());
3025e5dd7070Spatrick       return MakeAddrLValue(C, E->getType(), AlignmentSource::Decl);
3026e5dd7070Spatrick     } else {
3027ec727ea7Spatrick       auto C =
3028ec727ea7Spatrick           CGM.GetAddrOfConstantCString(std::string(FnName), GVName.c_str());
3029e5dd7070Spatrick       return MakeAddrLValue(C, E->getType(), AlignmentSource::Decl);
3030e5dd7070Spatrick     }
3031e5dd7070Spatrick   }
3032e5dd7070Spatrick   auto C = CGM.GetAddrOfConstantStringFromLiteral(SL, GVName);
3033e5dd7070Spatrick   return MakeAddrLValue(C, E->getType(), AlignmentSource::Decl);
3034e5dd7070Spatrick }
3035e5dd7070Spatrick 
3036e5dd7070Spatrick /// Emit a type description suitable for use by a runtime sanitizer library. The
3037e5dd7070Spatrick /// format of a type descriptor is
3038e5dd7070Spatrick ///
3039e5dd7070Spatrick /// \code
3040e5dd7070Spatrick ///   { i16 TypeKind, i16 TypeInfo }
3041e5dd7070Spatrick /// \endcode
3042e5dd7070Spatrick ///
3043e5dd7070Spatrick /// followed by an array of i8 containing the type name. TypeKind is 0 for an
3044e5dd7070Spatrick /// integer, 1 for a floating point value, and -1 for anything else.
EmitCheckTypeDescriptor(QualType T)3045e5dd7070Spatrick llvm::Constant *CodeGenFunction::EmitCheckTypeDescriptor(QualType T) {
3046e5dd7070Spatrick   // Only emit each type's descriptor once.
3047e5dd7070Spatrick   if (llvm::Constant *C = CGM.getTypeDescriptorFromMap(T))
3048e5dd7070Spatrick     return C;
3049e5dd7070Spatrick 
3050e5dd7070Spatrick   uint16_t TypeKind = -1;
3051e5dd7070Spatrick   uint16_t TypeInfo = 0;
3052e5dd7070Spatrick 
3053e5dd7070Spatrick   if (T->isIntegerType()) {
3054e5dd7070Spatrick     TypeKind = 0;
3055e5dd7070Spatrick     TypeInfo = (llvm::Log2_32(getContext().getTypeSize(T)) << 1) |
3056e5dd7070Spatrick                (T->isSignedIntegerType() ? 1 : 0);
3057e5dd7070Spatrick   } else if (T->isFloatingType()) {
3058e5dd7070Spatrick     TypeKind = 1;
3059e5dd7070Spatrick     TypeInfo = getContext().getTypeSize(T);
3060e5dd7070Spatrick   }
3061e5dd7070Spatrick 
3062e5dd7070Spatrick   // Format the type name as if for a diagnostic, including quotes and
3063e5dd7070Spatrick   // optionally an 'aka'.
3064e5dd7070Spatrick   SmallString<32> Buffer;
3065*12c85518Srobert   CGM.getDiags().ConvertArgToString(
3066*12c85518Srobert       DiagnosticsEngine::ak_qualtype, (intptr_t)T.getAsOpaquePtr(), StringRef(),
3067*12c85518Srobert       StringRef(), std::nullopt, Buffer, std::nullopt);
3068e5dd7070Spatrick 
3069e5dd7070Spatrick   llvm::Constant *Components[] = {
3070e5dd7070Spatrick     Builder.getInt16(TypeKind), Builder.getInt16(TypeInfo),
3071e5dd7070Spatrick     llvm::ConstantDataArray::getString(getLLVMContext(), Buffer)
3072e5dd7070Spatrick   };
3073e5dd7070Spatrick   llvm::Constant *Descriptor = llvm::ConstantStruct::getAnon(Components);
3074e5dd7070Spatrick 
3075e5dd7070Spatrick   auto *GV = new llvm::GlobalVariable(
3076e5dd7070Spatrick       CGM.getModule(), Descriptor->getType(),
3077e5dd7070Spatrick       /*isConstant=*/true, llvm::GlobalVariable::PrivateLinkage, Descriptor);
3078e5dd7070Spatrick   GV->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
3079e5dd7070Spatrick   CGM.getSanitizerMetadata()->disableSanitizerForGlobal(GV);
3080e5dd7070Spatrick 
3081e5dd7070Spatrick   // Remember the descriptor for this type.
3082e5dd7070Spatrick   CGM.setTypeDescriptorInMap(T, GV);
3083e5dd7070Spatrick 
3084e5dd7070Spatrick   return GV;
3085e5dd7070Spatrick }
3086e5dd7070Spatrick 
EmitCheckValue(llvm::Value * V)3087e5dd7070Spatrick llvm::Value *CodeGenFunction::EmitCheckValue(llvm::Value *V) {
3088e5dd7070Spatrick   llvm::Type *TargetTy = IntPtrTy;
3089e5dd7070Spatrick 
3090e5dd7070Spatrick   if (V->getType() == TargetTy)
3091e5dd7070Spatrick     return V;
3092e5dd7070Spatrick 
3093e5dd7070Spatrick   // Floating-point types which fit into intptr_t are bitcast to integers
3094e5dd7070Spatrick   // and then passed directly (after zero-extension, if necessary).
3095e5dd7070Spatrick   if (V->getType()->isFloatingPointTy()) {
3096*12c85518Srobert     unsigned Bits = V->getType()->getPrimitiveSizeInBits().getFixedValue();
3097e5dd7070Spatrick     if (Bits <= TargetTy->getIntegerBitWidth())
3098e5dd7070Spatrick       V = Builder.CreateBitCast(V, llvm::Type::getIntNTy(getLLVMContext(),
3099e5dd7070Spatrick                                                          Bits));
3100e5dd7070Spatrick   }
3101e5dd7070Spatrick 
3102e5dd7070Spatrick   // Integers which fit in intptr_t are zero-extended and passed directly.
3103e5dd7070Spatrick   if (V->getType()->isIntegerTy() &&
3104e5dd7070Spatrick       V->getType()->getIntegerBitWidth() <= TargetTy->getIntegerBitWidth())
3105e5dd7070Spatrick     return Builder.CreateZExt(V, TargetTy);
3106e5dd7070Spatrick 
3107e5dd7070Spatrick   // Pointers are passed directly, everything else is passed by address.
3108e5dd7070Spatrick   if (!V->getType()->isPointerTy()) {
3109e5dd7070Spatrick     Address Ptr = CreateDefaultAlignTempAlloca(V->getType());
3110e5dd7070Spatrick     Builder.CreateStore(V, Ptr);
3111e5dd7070Spatrick     V = Ptr.getPointer();
3112e5dd7070Spatrick   }
3113e5dd7070Spatrick   return Builder.CreatePtrToInt(V, TargetTy);
3114e5dd7070Spatrick }
3115e5dd7070Spatrick 
3116e5dd7070Spatrick /// Emit a representation of a SourceLocation for passing to a handler
3117e5dd7070Spatrick /// in a sanitizer runtime library. The format for this data is:
3118e5dd7070Spatrick /// \code
3119e5dd7070Spatrick ///   struct SourceLocation {
3120e5dd7070Spatrick ///     const char *Filename;
3121e5dd7070Spatrick ///     int32_t Line, Column;
3122e5dd7070Spatrick ///   };
3123e5dd7070Spatrick /// \endcode
3124e5dd7070Spatrick /// For an invalid SourceLocation, the Filename pointer is null.
EmitCheckSourceLocation(SourceLocation Loc)3125e5dd7070Spatrick llvm::Constant *CodeGenFunction::EmitCheckSourceLocation(SourceLocation Loc) {
3126e5dd7070Spatrick   llvm::Constant *Filename;
3127e5dd7070Spatrick   int Line, Column;
3128e5dd7070Spatrick 
3129e5dd7070Spatrick   PresumedLoc PLoc = getContext().getSourceManager().getPresumedLoc(Loc);
3130e5dd7070Spatrick   if (PLoc.isValid()) {
3131e5dd7070Spatrick     StringRef FilenameString = PLoc.getFilename();
3132e5dd7070Spatrick 
3133e5dd7070Spatrick     int PathComponentsToStrip =
3134e5dd7070Spatrick         CGM.getCodeGenOpts().EmitCheckPathComponentsToStrip;
3135e5dd7070Spatrick     if (PathComponentsToStrip < 0) {
3136e5dd7070Spatrick       assert(PathComponentsToStrip != INT_MIN);
3137e5dd7070Spatrick       int PathComponentsToKeep = -PathComponentsToStrip;
3138e5dd7070Spatrick       auto I = llvm::sys::path::rbegin(FilenameString);
3139e5dd7070Spatrick       auto E = llvm::sys::path::rend(FilenameString);
3140e5dd7070Spatrick       while (I != E && --PathComponentsToKeep)
3141e5dd7070Spatrick         ++I;
3142e5dd7070Spatrick 
3143e5dd7070Spatrick       FilenameString = FilenameString.substr(I - E);
3144e5dd7070Spatrick     } else if (PathComponentsToStrip > 0) {
3145e5dd7070Spatrick       auto I = llvm::sys::path::begin(FilenameString);
3146e5dd7070Spatrick       auto E = llvm::sys::path::end(FilenameString);
3147e5dd7070Spatrick       while (I != E && PathComponentsToStrip--)
3148e5dd7070Spatrick         ++I;
3149e5dd7070Spatrick 
3150e5dd7070Spatrick       if (I != E)
3151e5dd7070Spatrick         FilenameString =
3152e5dd7070Spatrick             FilenameString.substr(I - llvm::sys::path::begin(FilenameString));
3153e5dd7070Spatrick       else
3154e5dd7070Spatrick         FilenameString = llvm::sys::path::filename(FilenameString);
3155e5dd7070Spatrick     }
3156e5dd7070Spatrick 
3157ec727ea7Spatrick     auto FilenameGV =
3158ec727ea7Spatrick         CGM.GetAddrOfConstantCString(std::string(FilenameString), ".src");
3159e5dd7070Spatrick     CGM.getSanitizerMetadata()->disableSanitizerForGlobal(
3160*12c85518Srobert         cast<llvm::GlobalVariable>(
3161*12c85518Srobert             FilenameGV.getPointer()->stripPointerCasts()));
3162e5dd7070Spatrick     Filename = FilenameGV.getPointer();
3163e5dd7070Spatrick     Line = PLoc.getLine();
3164e5dd7070Spatrick     Column = PLoc.getColumn();
3165e5dd7070Spatrick   } else {
3166e5dd7070Spatrick     Filename = llvm::Constant::getNullValue(Int8PtrTy);
3167e5dd7070Spatrick     Line = Column = 0;
3168e5dd7070Spatrick   }
3169e5dd7070Spatrick 
3170e5dd7070Spatrick   llvm::Constant *Data[] = {Filename, Builder.getInt32(Line),
3171e5dd7070Spatrick                             Builder.getInt32(Column)};
3172e5dd7070Spatrick 
3173e5dd7070Spatrick   return llvm::ConstantStruct::getAnon(Data);
3174e5dd7070Spatrick }
3175e5dd7070Spatrick 
3176e5dd7070Spatrick namespace {
3177e5dd7070Spatrick /// Specify under what conditions this check can be recovered
3178e5dd7070Spatrick enum class CheckRecoverableKind {
3179e5dd7070Spatrick   /// Always terminate program execution if this check fails.
3180e5dd7070Spatrick   Unrecoverable,
3181e5dd7070Spatrick   /// Check supports recovering, runtime has both fatal (noreturn) and
3182e5dd7070Spatrick   /// non-fatal handlers for this check.
3183e5dd7070Spatrick   Recoverable,
3184e5dd7070Spatrick   /// Runtime conditionally aborts, always need to support recovery.
3185e5dd7070Spatrick   AlwaysRecoverable
3186e5dd7070Spatrick };
3187e5dd7070Spatrick }
3188e5dd7070Spatrick 
getRecoverableKind(SanitizerMask Kind)3189e5dd7070Spatrick static CheckRecoverableKind getRecoverableKind(SanitizerMask Kind) {
3190e5dd7070Spatrick   assert(Kind.countPopulation() == 1);
3191e5dd7070Spatrick   if (Kind == SanitizerKind::Function || Kind == SanitizerKind::Vptr)
3192e5dd7070Spatrick     return CheckRecoverableKind::AlwaysRecoverable;
3193e5dd7070Spatrick   else if (Kind == SanitizerKind::Return || Kind == SanitizerKind::Unreachable)
3194e5dd7070Spatrick     return CheckRecoverableKind::Unrecoverable;
3195e5dd7070Spatrick   else
3196e5dd7070Spatrick     return CheckRecoverableKind::Recoverable;
3197e5dd7070Spatrick }
3198e5dd7070Spatrick 
3199e5dd7070Spatrick namespace {
3200e5dd7070Spatrick struct SanitizerHandlerInfo {
3201e5dd7070Spatrick   char const *const Name;
3202e5dd7070Spatrick   unsigned Version;
3203e5dd7070Spatrick };
3204e5dd7070Spatrick }
3205e5dd7070Spatrick 
3206e5dd7070Spatrick const SanitizerHandlerInfo SanitizerHandlers[] = {
3207e5dd7070Spatrick #define SANITIZER_CHECK(Enum, Name, Version) {#Name, Version},
3208e5dd7070Spatrick     LIST_SANITIZER_CHECKS
3209e5dd7070Spatrick #undef SANITIZER_CHECK
3210e5dd7070Spatrick };
3211e5dd7070Spatrick 
emitCheckHandlerCall(CodeGenFunction & CGF,llvm::FunctionType * FnType,ArrayRef<llvm::Value * > FnArgs,SanitizerHandler CheckHandler,CheckRecoverableKind RecoverKind,bool IsFatal,llvm::BasicBlock * ContBB)3212e5dd7070Spatrick static void emitCheckHandlerCall(CodeGenFunction &CGF,
3213e5dd7070Spatrick                                  llvm::FunctionType *FnType,
3214e5dd7070Spatrick                                  ArrayRef<llvm::Value *> FnArgs,
3215e5dd7070Spatrick                                  SanitizerHandler CheckHandler,
3216e5dd7070Spatrick                                  CheckRecoverableKind RecoverKind, bool IsFatal,
3217e5dd7070Spatrick                                  llvm::BasicBlock *ContBB) {
3218e5dd7070Spatrick   assert(IsFatal || RecoverKind != CheckRecoverableKind::Unrecoverable);
3219*12c85518Srobert   std::optional<ApplyDebugLocation> DL;
3220e5dd7070Spatrick   if (!CGF.Builder.getCurrentDebugLocation()) {
3221e5dd7070Spatrick     // Ensure that the call has at least an artificial debug location.
3222e5dd7070Spatrick     DL.emplace(CGF, SourceLocation());
3223e5dd7070Spatrick   }
3224e5dd7070Spatrick   bool NeedsAbortSuffix =
3225e5dd7070Spatrick       IsFatal && RecoverKind != CheckRecoverableKind::Unrecoverable;
3226e5dd7070Spatrick   bool MinimalRuntime = CGF.CGM.getCodeGenOpts().SanitizeMinimalRuntime;
3227e5dd7070Spatrick   const SanitizerHandlerInfo &CheckInfo = SanitizerHandlers[CheckHandler];
3228e5dd7070Spatrick   const StringRef CheckName = CheckInfo.Name;
3229e5dd7070Spatrick   std::string FnName = "__ubsan_handle_" + CheckName.str();
3230e5dd7070Spatrick   if (CheckInfo.Version && !MinimalRuntime)
3231e5dd7070Spatrick     FnName += "_v" + llvm::utostr(CheckInfo.Version);
3232e5dd7070Spatrick   if (MinimalRuntime)
3233e5dd7070Spatrick     FnName += "_minimal";
3234e5dd7070Spatrick   if (NeedsAbortSuffix)
3235e5dd7070Spatrick     FnName += "_abort";
3236e5dd7070Spatrick   bool MayReturn =
3237e5dd7070Spatrick       !IsFatal || RecoverKind == CheckRecoverableKind::AlwaysRecoverable;
3238e5dd7070Spatrick 
3239*12c85518Srobert   llvm::AttrBuilder B(CGF.getLLVMContext());
3240e5dd7070Spatrick   if (!MayReturn) {
3241e5dd7070Spatrick     B.addAttribute(llvm::Attribute::NoReturn)
3242e5dd7070Spatrick         .addAttribute(llvm::Attribute::NoUnwind);
3243e5dd7070Spatrick   }
3244*12c85518Srobert   B.addUWTableAttr(llvm::UWTableKind::Default);
3245e5dd7070Spatrick 
3246e5dd7070Spatrick   llvm::FunctionCallee Fn = CGF.CGM.CreateRuntimeFunction(
3247e5dd7070Spatrick       FnType, FnName,
3248e5dd7070Spatrick       llvm::AttributeList::get(CGF.getLLVMContext(),
3249e5dd7070Spatrick                                llvm::AttributeList::FunctionIndex, B),
3250e5dd7070Spatrick       /*Local=*/true);
3251e5dd7070Spatrick   llvm::CallInst *HandlerCall = CGF.EmitNounwindRuntimeCall(Fn, FnArgs);
3252e5dd7070Spatrick   if (!MayReturn) {
3253e5dd7070Spatrick     HandlerCall->setDoesNotReturn();
3254e5dd7070Spatrick     CGF.Builder.CreateUnreachable();
3255e5dd7070Spatrick   } else {
3256e5dd7070Spatrick     CGF.Builder.CreateBr(ContBB);
3257e5dd7070Spatrick   }
3258e5dd7070Spatrick }
3259e5dd7070Spatrick 
EmitCheck(ArrayRef<std::pair<llvm::Value *,SanitizerMask>> Checked,SanitizerHandler CheckHandler,ArrayRef<llvm::Constant * > StaticArgs,ArrayRef<llvm::Value * > DynamicArgs)3260e5dd7070Spatrick void CodeGenFunction::EmitCheck(
3261e5dd7070Spatrick     ArrayRef<std::pair<llvm::Value *, SanitizerMask>> Checked,
3262e5dd7070Spatrick     SanitizerHandler CheckHandler, ArrayRef<llvm::Constant *> StaticArgs,
3263e5dd7070Spatrick     ArrayRef<llvm::Value *> DynamicArgs) {
3264e5dd7070Spatrick   assert(IsSanitizerScope);
3265e5dd7070Spatrick   assert(Checked.size() > 0);
3266e5dd7070Spatrick   assert(CheckHandler >= 0 &&
3267*12c85518Srobert          size_t(CheckHandler) < std::size(SanitizerHandlers));
3268e5dd7070Spatrick   const StringRef CheckName = SanitizerHandlers[CheckHandler].Name;
3269e5dd7070Spatrick 
3270e5dd7070Spatrick   llvm::Value *FatalCond = nullptr;
3271e5dd7070Spatrick   llvm::Value *RecoverableCond = nullptr;
3272e5dd7070Spatrick   llvm::Value *TrapCond = nullptr;
3273e5dd7070Spatrick   for (int i = 0, n = Checked.size(); i < n; ++i) {
3274e5dd7070Spatrick     llvm::Value *Check = Checked[i].first;
3275e5dd7070Spatrick     // -fsanitize-trap= overrides -fsanitize-recover=.
3276e5dd7070Spatrick     llvm::Value *&Cond =
3277e5dd7070Spatrick         CGM.getCodeGenOpts().SanitizeTrap.has(Checked[i].second)
3278e5dd7070Spatrick             ? TrapCond
3279e5dd7070Spatrick             : CGM.getCodeGenOpts().SanitizeRecover.has(Checked[i].second)
3280e5dd7070Spatrick                   ? RecoverableCond
3281e5dd7070Spatrick                   : FatalCond;
3282e5dd7070Spatrick     Cond = Cond ? Builder.CreateAnd(Cond, Check) : Check;
3283e5dd7070Spatrick   }
3284e5dd7070Spatrick 
3285e5dd7070Spatrick   if (TrapCond)
3286a9ac8606Spatrick     EmitTrapCheck(TrapCond, CheckHandler);
3287e5dd7070Spatrick   if (!FatalCond && !RecoverableCond)
3288e5dd7070Spatrick     return;
3289e5dd7070Spatrick 
3290e5dd7070Spatrick   llvm::Value *JointCond;
3291e5dd7070Spatrick   if (FatalCond && RecoverableCond)
3292e5dd7070Spatrick     JointCond = Builder.CreateAnd(FatalCond, RecoverableCond);
3293e5dd7070Spatrick   else
3294e5dd7070Spatrick     JointCond = FatalCond ? FatalCond : RecoverableCond;
3295e5dd7070Spatrick   assert(JointCond);
3296e5dd7070Spatrick 
3297e5dd7070Spatrick   CheckRecoverableKind RecoverKind = getRecoverableKind(Checked[0].second);
3298e5dd7070Spatrick   assert(SanOpts.has(Checked[0].second));
3299e5dd7070Spatrick #ifndef NDEBUG
3300e5dd7070Spatrick   for (int i = 1, n = Checked.size(); i < n; ++i) {
3301e5dd7070Spatrick     assert(RecoverKind == getRecoverableKind(Checked[i].second) &&
3302e5dd7070Spatrick            "All recoverable kinds in a single check must be same!");
3303e5dd7070Spatrick     assert(SanOpts.has(Checked[i].second));
3304e5dd7070Spatrick   }
3305e5dd7070Spatrick #endif
3306e5dd7070Spatrick 
3307e5dd7070Spatrick   llvm::BasicBlock *Cont = createBasicBlock("cont");
3308e5dd7070Spatrick   llvm::BasicBlock *Handlers = createBasicBlock("handler." + CheckName);
3309e5dd7070Spatrick   llvm::Instruction *Branch = Builder.CreateCondBr(JointCond, Cont, Handlers);
3310e5dd7070Spatrick   // Give hint that we very much don't expect to execute the handler
3311e5dd7070Spatrick   // Value chosen to match UR_NONTAKEN_WEIGHT, see BranchProbabilityInfo.cpp
3312e5dd7070Spatrick   llvm::MDBuilder MDHelper(getLLVMContext());
3313e5dd7070Spatrick   llvm::MDNode *Node = MDHelper.createBranchWeights((1U << 20) - 1, 1);
3314e5dd7070Spatrick   Branch->setMetadata(llvm::LLVMContext::MD_prof, Node);
3315e5dd7070Spatrick   EmitBlock(Handlers);
3316e5dd7070Spatrick 
3317e5dd7070Spatrick   // Handler functions take an i8* pointing to the (handler-specific) static
3318e5dd7070Spatrick   // information block, followed by a sequence of intptr_t arguments
3319e5dd7070Spatrick   // representing operand values.
3320e5dd7070Spatrick   SmallVector<llvm::Value *, 4> Args;
3321e5dd7070Spatrick   SmallVector<llvm::Type *, 4> ArgTypes;
3322e5dd7070Spatrick   if (!CGM.getCodeGenOpts().SanitizeMinimalRuntime) {
3323e5dd7070Spatrick     Args.reserve(DynamicArgs.size() + 1);
3324e5dd7070Spatrick     ArgTypes.reserve(DynamicArgs.size() + 1);
3325e5dd7070Spatrick 
3326e5dd7070Spatrick     // Emit handler arguments and create handler function type.
3327e5dd7070Spatrick     if (!StaticArgs.empty()) {
3328e5dd7070Spatrick       llvm::Constant *Info = llvm::ConstantStruct::getAnon(StaticArgs);
3329*12c85518Srobert       auto *InfoPtr = new llvm::GlobalVariable(
3330*12c85518Srobert           CGM.getModule(), Info->getType(), false,
3331*12c85518Srobert           llvm::GlobalVariable::PrivateLinkage, Info, "", nullptr,
3332*12c85518Srobert           llvm::GlobalVariable::NotThreadLocal,
3333*12c85518Srobert           CGM.getDataLayout().getDefaultGlobalsAddressSpace());
3334e5dd7070Spatrick       InfoPtr->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
3335e5dd7070Spatrick       CGM.getSanitizerMetadata()->disableSanitizerForGlobal(InfoPtr);
3336*12c85518Srobert       Args.push_back(EmitCastToVoidPtr(InfoPtr));
3337*12c85518Srobert       ArgTypes.push_back(Args.back()->getType());
3338e5dd7070Spatrick     }
3339e5dd7070Spatrick 
3340e5dd7070Spatrick     for (size_t i = 0, n = DynamicArgs.size(); i != n; ++i) {
3341e5dd7070Spatrick       Args.push_back(EmitCheckValue(DynamicArgs[i]));
3342e5dd7070Spatrick       ArgTypes.push_back(IntPtrTy);
3343e5dd7070Spatrick     }
3344e5dd7070Spatrick   }
3345e5dd7070Spatrick 
3346e5dd7070Spatrick   llvm::FunctionType *FnType =
3347e5dd7070Spatrick     llvm::FunctionType::get(CGM.VoidTy, ArgTypes, false);
3348e5dd7070Spatrick 
3349e5dd7070Spatrick   if (!FatalCond || !RecoverableCond) {
3350e5dd7070Spatrick     // Simple case: we need to generate a single handler call, either
3351e5dd7070Spatrick     // fatal, or non-fatal.
3352e5dd7070Spatrick     emitCheckHandlerCall(*this, FnType, Args, CheckHandler, RecoverKind,
3353e5dd7070Spatrick                          (FatalCond != nullptr), Cont);
3354e5dd7070Spatrick   } else {
3355e5dd7070Spatrick     // Emit two handler calls: first one for set of unrecoverable checks,
3356e5dd7070Spatrick     // another one for recoverable.
3357e5dd7070Spatrick     llvm::BasicBlock *NonFatalHandlerBB =
3358e5dd7070Spatrick         createBasicBlock("non_fatal." + CheckName);
3359e5dd7070Spatrick     llvm::BasicBlock *FatalHandlerBB = createBasicBlock("fatal." + CheckName);
3360e5dd7070Spatrick     Builder.CreateCondBr(FatalCond, NonFatalHandlerBB, FatalHandlerBB);
3361e5dd7070Spatrick     EmitBlock(FatalHandlerBB);
3362e5dd7070Spatrick     emitCheckHandlerCall(*this, FnType, Args, CheckHandler, RecoverKind, true,
3363e5dd7070Spatrick                          NonFatalHandlerBB);
3364e5dd7070Spatrick     EmitBlock(NonFatalHandlerBB);
3365e5dd7070Spatrick     emitCheckHandlerCall(*this, FnType, Args, CheckHandler, RecoverKind, false,
3366e5dd7070Spatrick                          Cont);
3367e5dd7070Spatrick   }
3368e5dd7070Spatrick 
3369e5dd7070Spatrick   EmitBlock(Cont);
3370e5dd7070Spatrick }
3371e5dd7070Spatrick 
EmitCfiSlowPathCheck(SanitizerMask Kind,llvm::Value * Cond,llvm::ConstantInt * TypeId,llvm::Value * Ptr,ArrayRef<llvm::Constant * > StaticArgs)3372e5dd7070Spatrick void CodeGenFunction::EmitCfiSlowPathCheck(
3373e5dd7070Spatrick     SanitizerMask Kind, llvm::Value *Cond, llvm::ConstantInt *TypeId,
3374e5dd7070Spatrick     llvm::Value *Ptr, ArrayRef<llvm::Constant *> StaticArgs) {
3375e5dd7070Spatrick   llvm::BasicBlock *Cont = createBasicBlock("cfi.cont");
3376e5dd7070Spatrick 
3377e5dd7070Spatrick   llvm::BasicBlock *CheckBB = createBasicBlock("cfi.slowpath");
3378e5dd7070Spatrick   llvm::BranchInst *BI = Builder.CreateCondBr(Cond, Cont, CheckBB);
3379e5dd7070Spatrick 
3380e5dd7070Spatrick   llvm::MDBuilder MDHelper(getLLVMContext());
3381e5dd7070Spatrick   llvm::MDNode *Node = MDHelper.createBranchWeights((1U << 20) - 1, 1);
3382e5dd7070Spatrick   BI->setMetadata(llvm::LLVMContext::MD_prof, Node);
3383e5dd7070Spatrick 
3384e5dd7070Spatrick   EmitBlock(CheckBB);
3385e5dd7070Spatrick 
3386e5dd7070Spatrick   bool WithDiag = !CGM.getCodeGenOpts().SanitizeTrap.has(Kind);
3387e5dd7070Spatrick 
3388e5dd7070Spatrick   llvm::CallInst *CheckCall;
3389e5dd7070Spatrick   llvm::FunctionCallee SlowPathFn;
3390e5dd7070Spatrick   if (WithDiag) {
3391e5dd7070Spatrick     llvm::Constant *Info = llvm::ConstantStruct::getAnon(StaticArgs);
3392e5dd7070Spatrick     auto *InfoPtr =
3393e5dd7070Spatrick         new llvm::GlobalVariable(CGM.getModule(), Info->getType(), false,
3394e5dd7070Spatrick                                  llvm::GlobalVariable::PrivateLinkage, Info);
3395e5dd7070Spatrick     InfoPtr->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
3396e5dd7070Spatrick     CGM.getSanitizerMetadata()->disableSanitizerForGlobal(InfoPtr);
3397e5dd7070Spatrick 
3398e5dd7070Spatrick     SlowPathFn = CGM.getModule().getOrInsertFunction(
3399e5dd7070Spatrick         "__cfi_slowpath_diag",
3400e5dd7070Spatrick         llvm::FunctionType::get(VoidTy, {Int64Ty, Int8PtrTy, Int8PtrTy},
3401e5dd7070Spatrick                                 false));
3402e5dd7070Spatrick     CheckCall = Builder.CreateCall(
3403e5dd7070Spatrick         SlowPathFn, {TypeId, Ptr, Builder.CreateBitCast(InfoPtr, Int8PtrTy)});
3404e5dd7070Spatrick   } else {
3405e5dd7070Spatrick     SlowPathFn = CGM.getModule().getOrInsertFunction(
3406e5dd7070Spatrick         "__cfi_slowpath",
3407e5dd7070Spatrick         llvm::FunctionType::get(VoidTy, {Int64Ty, Int8PtrTy}, false));
3408e5dd7070Spatrick     CheckCall = Builder.CreateCall(SlowPathFn, {TypeId, Ptr});
3409e5dd7070Spatrick   }
3410e5dd7070Spatrick 
3411e5dd7070Spatrick   CGM.setDSOLocal(
3412e5dd7070Spatrick       cast<llvm::GlobalValue>(SlowPathFn.getCallee()->stripPointerCasts()));
3413e5dd7070Spatrick   CheckCall->setDoesNotThrow();
3414e5dd7070Spatrick 
3415e5dd7070Spatrick   EmitBlock(Cont);
3416e5dd7070Spatrick }
3417e5dd7070Spatrick 
3418e5dd7070Spatrick // Emit a stub for __cfi_check function so that the linker knows about this
3419e5dd7070Spatrick // symbol in LTO mode.
EmitCfiCheckStub()3420e5dd7070Spatrick void CodeGenFunction::EmitCfiCheckStub() {
3421e5dd7070Spatrick   llvm::Module *M = &CGM.getModule();
3422e5dd7070Spatrick   auto &Ctx = M->getContext();
3423e5dd7070Spatrick   llvm::Function *F = llvm::Function::Create(
3424e5dd7070Spatrick       llvm::FunctionType::get(VoidTy, {Int64Ty, Int8PtrTy, Int8PtrTy}, false),
3425e5dd7070Spatrick       llvm::GlobalValue::WeakAnyLinkage, "__cfi_check", M);
3426e5dd7070Spatrick   CGM.setDSOLocal(F);
3427e5dd7070Spatrick   llvm::BasicBlock *BB = llvm::BasicBlock::Create(Ctx, "entry", F);
3428e5dd7070Spatrick   // FIXME: consider emitting an intrinsic call like
3429e5dd7070Spatrick   // call void @llvm.cfi_check(i64 %0, i8* %1, i8* %2)
3430e5dd7070Spatrick   // which can be lowered in CrossDSOCFI pass to the actual contents of
3431e5dd7070Spatrick   // __cfi_check. This would allow inlining of __cfi_check calls.
3432e5dd7070Spatrick   llvm::CallInst::Create(
3433e5dd7070Spatrick       llvm::Intrinsic::getDeclaration(M, llvm::Intrinsic::trap), "", BB);
3434e5dd7070Spatrick   llvm::ReturnInst::Create(Ctx, nullptr, BB);
3435e5dd7070Spatrick }
3436e5dd7070Spatrick 
3437e5dd7070Spatrick // This function is basically a switch over the CFI failure kind, which is
3438e5dd7070Spatrick // extracted from CFICheckFailData (1st function argument). Each case is either
3439e5dd7070Spatrick // llvm.trap or a call to one of the two runtime handlers, based on
3440e5dd7070Spatrick // -fsanitize-trap and -fsanitize-recover settings.  Default case (invalid
3441e5dd7070Spatrick // failure kind) traps, but this should really never happen.  CFICheckFailData
3442e5dd7070Spatrick // can be nullptr if the calling module has -fsanitize-trap behavior for this
3443e5dd7070Spatrick // check kind; in this case __cfi_check_fail traps as well.
EmitCfiCheckFail()3444e5dd7070Spatrick void CodeGenFunction::EmitCfiCheckFail() {
3445e5dd7070Spatrick   SanitizerScope SanScope(this);
3446e5dd7070Spatrick   FunctionArgList Args;
3447e5dd7070Spatrick   ImplicitParamDecl ArgData(getContext(), getContext().VoidPtrTy,
3448e5dd7070Spatrick                             ImplicitParamDecl::Other);
3449e5dd7070Spatrick   ImplicitParamDecl ArgAddr(getContext(), getContext().VoidPtrTy,
3450e5dd7070Spatrick                             ImplicitParamDecl::Other);
3451e5dd7070Spatrick   Args.push_back(&ArgData);
3452e5dd7070Spatrick   Args.push_back(&ArgAddr);
3453e5dd7070Spatrick 
3454e5dd7070Spatrick   const CGFunctionInfo &FI =
3455e5dd7070Spatrick     CGM.getTypes().arrangeBuiltinFunctionDeclaration(getContext().VoidTy, Args);
3456e5dd7070Spatrick 
3457e5dd7070Spatrick   llvm::Function *F = llvm::Function::Create(
3458e5dd7070Spatrick       llvm::FunctionType::get(VoidTy, {VoidPtrTy, VoidPtrTy}, false),
3459e5dd7070Spatrick       llvm::GlobalValue::WeakODRLinkage, "__cfi_check_fail", &CGM.getModule());
3460e5dd7070Spatrick 
3461a9ac8606Spatrick   CGM.SetLLVMFunctionAttributes(GlobalDecl(), FI, F, /*IsThunk=*/false);
3462e5dd7070Spatrick   CGM.SetLLVMFunctionAttributesForDefinition(nullptr, F);
3463e5dd7070Spatrick   F->setVisibility(llvm::GlobalValue::HiddenVisibility);
3464e5dd7070Spatrick 
3465e5dd7070Spatrick   StartFunction(GlobalDecl(), CGM.getContext().VoidTy, F, FI, Args,
3466e5dd7070Spatrick                 SourceLocation());
3467e5dd7070Spatrick 
3468a9ac8606Spatrick   // This function is not affected by NoSanitizeList. This function does
3469e5dd7070Spatrick   // not have a source location, but "src:*" would still apply. Revert any
3470e5dd7070Spatrick   // changes to SanOpts made in StartFunction.
3471e5dd7070Spatrick   SanOpts = CGM.getLangOpts().Sanitize;
3472e5dd7070Spatrick 
3473e5dd7070Spatrick   llvm::Value *Data =
3474e5dd7070Spatrick       EmitLoadOfScalar(GetAddrOfLocalVar(&ArgData), /*Volatile=*/false,
3475e5dd7070Spatrick                        CGM.getContext().VoidPtrTy, ArgData.getLocation());
3476e5dd7070Spatrick   llvm::Value *Addr =
3477e5dd7070Spatrick       EmitLoadOfScalar(GetAddrOfLocalVar(&ArgAddr), /*Volatile=*/false,
3478e5dd7070Spatrick                        CGM.getContext().VoidPtrTy, ArgAddr.getLocation());
3479e5dd7070Spatrick 
3480e5dd7070Spatrick   // Data == nullptr means the calling module has trap behaviour for this check.
3481e5dd7070Spatrick   llvm::Value *DataIsNotNullPtr =
3482e5dd7070Spatrick       Builder.CreateICmpNE(Data, llvm::ConstantPointerNull::get(Int8PtrTy));
3483a9ac8606Spatrick   EmitTrapCheck(DataIsNotNullPtr, SanitizerHandler::CFICheckFail);
3484e5dd7070Spatrick 
3485e5dd7070Spatrick   llvm::StructType *SourceLocationTy =
3486e5dd7070Spatrick       llvm::StructType::get(VoidPtrTy, Int32Ty, Int32Ty);
3487e5dd7070Spatrick   llvm::StructType *CfiCheckFailDataTy =
3488e5dd7070Spatrick       llvm::StructType::get(Int8Ty, SourceLocationTy, VoidPtrTy);
3489e5dd7070Spatrick 
3490e5dd7070Spatrick   llvm::Value *V = Builder.CreateConstGEP2_32(
3491e5dd7070Spatrick       CfiCheckFailDataTy,
3492e5dd7070Spatrick       Builder.CreatePointerCast(Data, CfiCheckFailDataTy->getPointerTo(0)), 0,
3493e5dd7070Spatrick       0);
3494*12c85518Srobert 
3495*12c85518Srobert   Address CheckKindAddr(V, Int8Ty, getIntAlign());
3496e5dd7070Spatrick   llvm::Value *CheckKind = Builder.CreateLoad(CheckKindAddr);
3497e5dd7070Spatrick 
3498e5dd7070Spatrick   llvm::Value *AllVtables = llvm::MetadataAsValue::get(
3499e5dd7070Spatrick       CGM.getLLVMContext(),
3500e5dd7070Spatrick       llvm::MDString::get(CGM.getLLVMContext(), "all-vtables"));
3501e5dd7070Spatrick   llvm::Value *ValidVtable = Builder.CreateZExt(
3502e5dd7070Spatrick       Builder.CreateCall(CGM.getIntrinsic(llvm::Intrinsic::type_test),
3503e5dd7070Spatrick                          {Addr, AllVtables}),
3504e5dd7070Spatrick       IntPtrTy);
3505e5dd7070Spatrick 
3506e5dd7070Spatrick   const std::pair<int, SanitizerMask> CheckKinds[] = {
3507e5dd7070Spatrick       {CFITCK_VCall, SanitizerKind::CFIVCall},
3508e5dd7070Spatrick       {CFITCK_NVCall, SanitizerKind::CFINVCall},
3509e5dd7070Spatrick       {CFITCK_DerivedCast, SanitizerKind::CFIDerivedCast},
3510e5dd7070Spatrick       {CFITCK_UnrelatedCast, SanitizerKind::CFIUnrelatedCast},
3511e5dd7070Spatrick       {CFITCK_ICall, SanitizerKind::CFIICall}};
3512e5dd7070Spatrick 
3513e5dd7070Spatrick   SmallVector<std::pair<llvm::Value *, SanitizerMask>, 5> Checks;
3514e5dd7070Spatrick   for (auto CheckKindMaskPair : CheckKinds) {
3515e5dd7070Spatrick     int Kind = CheckKindMaskPair.first;
3516e5dd7070Spatrick     SanitizerMask Mask = CheckKindMaskPair.second;
3517e5dd7070Spatrick     llvm::Value *Cond =
3518e5dd7070Spatrick         Builder.CreateICmpNE(CheckKind, llvm::ConstantInt::get(Int8Ty, Kind));
3519e5dd7070Spatrick     if (CGM.getLangOpts().Sanitize.has(Mask))
3520e5dd7070Spatrick       EmitCheck(std::make_pair(Cond, Mask), SanitizerHandler::CFICheckFail, {},
3521e5dd7070Spatrick                 {Data, Addr, ValidVtable});
3522e5dd7070Spatrick     else
3523a9ac8606Spatrick       EmitTrapCheck(Cond, SanitizerHandler::CFICheckFail);
3524e5dd7070Spatrick   }
3525e5dd7070Spatrick 
3526e5dd7070Spatrick   FinishFunction();
3527e5dd7070Spatrick   // The only reference to this function will be created during LTO link.
3528e5dd7070Spatrick   // Make sure it survives until then.
3529e5dd7070Spatrick   CGM.addUsedGlobal(F);
3530e5dd7070Spatrick }
3531e5dd7070Spatrick 
EmitUnreachable(SourceLocation Loc)3532e5dd7070Spatrick void CodeGenFunction::EmitUnreachable(SourceLocation Loc) {
3533e5dd7070Spatrick   if (SanOpts.has(SanitizerKind::Unreachable)) {
3534e5dd7070Spatrick     SanitizerScope SanScope(this);
3535e5dd7070Spatrick     EmitCheck(std::make_pair(static_cast<llvm::Value *>(Builder.getFalse()),
3536e5dd7070Spatrick                              SanitizerKind::Unreachable),
3537e5dd7070Spatrick               SanitizerHandler::BuiltinUnreachable,
3538*12c85518Srobert               EmitCheckSourceLocation(Loc), std::nullopt);
3539e5dd7070Spatrick   }
3540e5dd7070Spatrick   Builder.CreateUnreachable();
3541e5dd7070Spatrick }
3542e5dd7070Spatrick 
EmitTrapCheck(llvm::Value * Checked,SanitizerHandler CheckHandlerID)3543a9ac8606Spatrick void CodeGenFunction::EmitTrapCheck(llvm::Value *Checked,
3544a9ac8606Spatrick                                     SanitizerHandler CheckHandlerID) {
3545e5dd7070Spatrick   llvm::BasicBlock *Cont = createBasicBlock("cont");
3546e5dd7070Spatrick 
3547e5dd7070Spatrick   // If we're optimizing, collapse all calls to trap down to just one per
3548a9ac8606Spatrick   // check-type per function to save on code size.
3549a9ac8606Spatrick   if (TrapBBs.size() <= CheckHandlerID)
3550a9ac8606Spatrick     TrapBBs.resize(CheckHandlerID + 1);
3551a9ac8606Spatrick   llvm::BasicBlock *&TrapBB = TrapBBs[CheckHandlerID];
3552a9ac8606Spatrick 
3553*12c85518Srobert   if (!CGM.getCodeGenOpts().OptimizationLevel || !TrapBB ||
3554*12c85518Srobert       (CurCodeDecl && CurCodeDecl->hasAttr<OptimizeNoneAttr>())) {
3555e5dd7070Spatrick     TrapBB = createBasicBlock("trap");
3556e5dd7070Spatrick     Builder.CreateCondBr(Checked, Cont, TrapBB);
3557e5dd7070Spatrick     EmitBlock(TrapBB);
3558a9ac8606Spatrick 
3559a9ac8606Spatrick     llvm::CallInst *TrapCall =
3560a9ac8606Spatrick         Builder.CreateCall(CGM.getIntrinsic(llvm::Intrinsic::ubsantrap),
3561a9ac8606Spatrick                            llvm::ConstantInt::get(CGM.Int8Ty, CheckHandlerID));
3562a9ac8606Spatrick 
3563a9ac8606Spatrick     if (!CGM.getCodeGenOpts().TrapFuncName.empty()) {
3564a9ac8606Spatrick       auto A = llvm::Attribute::get(getLLVMContext(), "trap-func-name",
3565a9ac8606Spatrick                                     CGM.getCodeGenOpts().TrapFuncName);
3566*12c85518Srobert       TrapCall->addFnAttr(A);
3567a9ac8606Spatrick     }
3568e5dd7070Spatrick     TrapCall->setDoesNotReturn();
3569e5dd7070Spatrick     TrapCall->setDoesNotThrow();
3570e5dd7070Spatrick     Builder.CreateUnreachable();
3571e5dd7070Spatrick   } else {
3572a9ac8606Spatrick     auto Call = TrapBB->begin();
3573a9ac8606Spatrick     assert(isa<llvm::CallInst>(Call) && "Expected call in trap BB");
3574a9ac8606Spatrick 
3575a9ac8606Spatrick     Call->applyMergedLocation(Call->getDebugLoc(),
3576a9ac8606Spatrick                               Builder.getCurrentDebugLocation());
3577e5dd7070Spatrick     Builder.CreateCondBr(Checked, Cont, TrapBB);
3578e5dd7070Spatrick   }
3579e5dd7070Spatrick 
3580e5dd7070Spatrick   EmitBlock(Cont);
3581e5dd7070Spatrick }
3582e5dd7070Spatrick 
EmitTrapCall(llvm::Intrinsic::ID IntrID)3583e5dd7070Spatrick llvm::CallInst *CodeGenFunction::EmitTrapCall(llvm::Intrinsic::ID IntrID) {
3584a9ac8606Spatrick   llvm::CallInst *TrapCall =
3585a9ac8606Spatrick       Builder.CreateCall(CGM.getIntrinsic(IntrID));
3586e5dd7070Spatrick 
3587e5dd7070Spatrick   if (!CGM.getCodeGenOpts().TrapFuncName.empty()) {
3588e5dd7070Spatrick     auto A = llvm::Attribute::get(getLLVMContext(), "trap-func-name",
3589e5dd7070Spatrick                                   CGM.getCodeGenOpts().TrapFuncName);
3590*12c85518Srobert     TrapCall->addFnAttr(A);
3591e5dd7070Spatrick   }
3592e5dd7070Spatrick 
3593e5dd7070Spatrick   return TrapCall;
3594e5dd7070Spatrick }
3595e5dd7070Spatrick 
EmitArrayToPointerDecay(const Expr * E,LValueBaseInfo * BaseInfo,TBAAAccessInfo * TBAAInfo)3596e5dd7070Spatrick Address CodeGenFunction::EmitArrayToPointerDecay(const Expr *E,
3597e5dd7070Spatrick                                                  LValueBaseInfo *BaseInfo,
3598e5dd7070Spatrick                                                  TBAAAccessInfo *TBAAInfo) {
3599e5dd7070Spatrick   assert(E->getType()->isArrayType() &&
3600e5dd7070Spatrick          "Array to pointer decay must have array source type!");
3601e5dd7070Spatrick 
3602e5dd7070Spatrick   // Expressions of array type can't be bitfields or vector elements.
3603e5dd7070Spatrick   LValue LV = EmitLValue(E);
3604e5dd7070Spatrick   Address Addr = LV.getAddress(*this);
3605e5dd7070Spatrick 
3606e5dd7070Spatrick   // If the array type was an incomplete type, we need to make sure
3607e5dd7070Spatrick   // the decay ends up being the right type.
3608e5dd7070Spatrick   llvm::Type *NewTy = ConvertType(E->getType());
3609e5dd7070Spatrick   Addr = Builder.CreateElementBitCast(Addr, NewTy);
3610e5dd7070Spatrick 
3611e5dd7070Spatrick   // Note that VLA pointers are always decayed, so we don't need to do
3612e5dd7070Spatrick   // anything here.
3613e5dd7070Spatrick   if (!E->getType()->isVariableArrayType()) {
3614e5dd7070Spatrick     assert(isa<llvm::ArrayType>(Addr.getElementType()) &&
3615e5dd7070Spatrick            "Expected pointer to array");
3616e5dd7070Spatrick     Addr = Builder.CreateConstArrayGEP(Addr, 0, "arraydecay");
3617e5dd7070Spatrick   }
3618e5dd7070Spatrick 
3619e5dd7070Spatrick   // The result of this decay conversion points to an array element within the
3620e5dd7070Spatrick   // base lvalue. However, since TBAA currently does not support representing
3621e5dd7070Spatrick   // accesses to elements of member arrays, we conservatively represent accesses
3622e5dd7070Spatrick   // to the pointee object as if it had no any base lvalue specified.
3623e5dd7070Spatrick   // TODO: Support TBAA for member arrays.
3624e5dd7070Spatrick   QualType EltType = E->getType()->castAsArrayTypeUnsafe()->getElementType();
3625e5dd7070Spatrick   if (BaseInfo) *BaseInfo = LV.getBaseInfo();
3626e5dd7070Spatrick   if (TBAAInfo) *TBAAInfo = CGM.getTBAAAccessInfo(EltType);
3627e5dd7070Spatrick 
3628e5dd7070Spatrick   return Builder.CreateElementBitCast(Addr, ConvertTypeForMem(EltType));
3629e5dd7070Spatrick }
3630e5dd7070Spatrick 
3631e5dd7070Spatrick /// isSimpleArrayDecayOperand - If the specified expr is a simple decay from an
3632e5dd7070Spatrick /// array to pointer, return the array subexpression.
isSimpleArrayDecayOperand(const Expr * E)3633e5dd7070Spatrick static const Expr *isSimpleArrayDecayOperand(const Expr *E) {
3634e5dd7070Spatrick   // If this isn't just an array->pointer decay, bail out.
3635e5dd7070Spatrick   const auto *CE = dyn_cast<CastExpr>(E);
3636e5dd7070Spatrick   if (!CE || CE->getCastKind() != CK_ArrayToPointerDecay)
3637e5dd7070Spatrick     return nullptr;
3638e5dd7070Spatrick 
3639e5dd7070Spatrick   // If this is a decay from variable width array, bail out.
3640e5dd7070Spatrick   const Expr *SubExpr = CE->getSubExpr();
3641e5dd7070Spatrick   if (SubExpr->getType()->isVariableArrayType())
3642e5dd7070Spatrick     return nullptr;
3643e5dd7070Spatrick 
3644e5dd7070Spatrick   return SubExpr;
3645e5dd7070Spatrick }
3646e5dd7070Spatrick 
emitArraySubscriptGEP(CodeGenFunction & CGF,llvm::Type * elemType,llvm::Value * ptr,ArrayRef<llvm::Value * > indices,bool inbounds,bool signedIndices,SourceLocation loc,const llvm::Twine & name="arrayidx")3647e5dd7070Spatrick static llvm::Value *emitArraySubscriptGEP(CodeGenFunction &CGF,
3648a9ac8606Spatrick                                           llvm::Type *elemType,
3649e5dd7070Spatrick                                           llvm::Value *ptr,
3650e5dd7070Spatrick                                           ArrayRef<llvm::Value*> indices,
3651e5dd7070Spatrick                                           bool inbounds,
3652e5dd7070Spatrick                                           bool signedIndices,
3653e5dd7070Spatrick                                           SourceLocation loc,
3654e5dd7070Spatrick                                     const llvm::Twine &name = "arrayidx") {
3655e5dd7070Spatrick   if (inbounds) {
3656*12c85518Srobert     return CGF.EmitCheckedInBoundsGEP(elemType, ptr, indices, signedIndices,
3657e5dd7070Spatrick                                       CodeGenFunction::NotSubtraction, loc,
3658e5dd7070Spatrick                                       name);
3659e5dd7070Spatrick   } else {
3660a9ac8606Spatrick     return CGF.Builder.CreateGEP(elemType, ptr, indices, name);
3661e5dd7070Spatrick   }
3662e5dd7070Spatrick }
3663e5dd7070Spatrick 
getArrayElementAlign(CharUnits arrayAlign,llvm::Value * idx,CharUnits eltSize)3664e5dd7070Spatrick static CharUnits getArrayElementAlign(CharUnits arrayAlign,
3665e5dd7070Spatrick                                       llvm::Value *idx,
3666e5dd7070Spatrick                                       CharUnits eltSize) {
3667e5dd7070Spatrick   // If we have a constant index, we can use the exact offset of the
3668e5dd7070Spatrick   // element we're accessing.
3669e5dd7070Spatrick   if (auto constantIdx = dyn_cast<llvm::ConstantInt>(idx)) {
3670e5dd7070Spatrick     CharUnits offset = constantIdx->getZExtValue() * eltSize;
3671e5dd7070Spatrick     return arrayAlign.alignmentAtOffset(offset);
3672e5dd7070Spatrick 
3673e5dd7070Spatrick   // Otherwise, use the worst-case alignment for any element.
3674e5dd7070Spatrick   } else {
3675e5dd7070Spatrick     return arrayAlign.alignmentOfArrayElement(eltSize);
3676e5dd7070Spatrick   }
3677e5dd7070Spatrick }
3678e5dd7070Spatrick 
getFixedSizeElementType(const ASTContext & ctx,const VariableArrayType * vla)3679e5dd7070Spatrick static QualType getFixedSizeElementType(const ASTContext &ctx,
3680e5dd7070Spatrick                                         const VariableArrayType *vla) {
3681e5dd7070Spatrick   QualType eltType;
3682e5dd7070Spatrick   do {
3683e5dd7070Spatrick     eltType = vla->getElementType();
3684e5dd7070Spatrick   } while ((vla = ctx.getAsVariableArrayType(eltType)));
3685e5dd7070Spatrick   return eltType;
3686e5dd7070Spatrick }
3687e5dd7070Spatrick 
3688e5dd7070Spatrick /// Given an array base, check whether its member access belongs to a record
3689e5dd7070Spatrick /// with preserve_access_index attribute or not.
IsPreserveAIArrayBase(CodeGenFunction & CGF,const Expr * ArrayBase)3690e5dd7070Spatrick static bool IsPreserveAIArrayBase(CodeGenFunction &CGF, const Expr *ArrayBase) {
3691e5dd7070Spatrick   if (!ArrayBase || !CGF.getDebugInfo())
3692e5dd7070Spatrick     return false;
3693e5dd7070Spatrick 
3694e5dd7070Spatrick   // Only support base as either a MemberExpr or DeclRefExpr.
3695e5dd7070Spatrick   // DeclRefExpr to cover cases like:
3696e5dd7070Spatrick   //    struct s { int a; int b[10]; };
3697e5dd7070Spatrick   //    struct s *p;
3698e5dd7070Spatrick   //    p[1].a
3699e5dd7070Spatrick   // p[1] will generate a DeclRefExpr and p[1].a is a MemberExpr.
3700e5dd7070Spatrick   // p->b[5] is a MemberExpr example.
3701e5dd7070Spatrick   const Expr *E = ArrayBase->IgnoreImpCasts();
3702e5dd7070Spatrick   if (const auto *ME = dyn_cast<MemberExpr>(E))
3703e5dd7070Spatrick     return ME->getMemberDecl()->hasAttr<BPFPreserveAccessIndexAttr>();
3704e5dd7070Spatrick 
3705e5dd7070Spatrick   if (const auto *DRE = dyn_cast<DeclRefExpr>(E)) {
3706e5dd7070Spatrick     const auto *VarDef = dyn_cast<VarDecl>(DRE->getDecl());
3707e5dd7070Spatrick     if (!VarDef)
3708e5dd7070Spatrick       return false;
3709e5dd7070Spatrick 
3710e5dd7070Spatrick     const auto *PtrT = VarDef->getType()->getAs<PointerType>();
3711e5dd7070Spatrick     if (!PtrT)
3712e5dd7070Spatrick       return false;
3713e5dd7070Spatrick 
3714e5dd7070Spatrick     const auto *PointeeT = PtrT->getPointeeType()
3715e5dd7070Spatrick                              ->getUnqualifiedDesugaredType();
3716e5dd7070Spatrick     if (const auto *RecT = dyn_cast<RecordType>(PointeeT))
3717e5dd7070Spatrick       return RecT->getDecl()->hasAttr<BPFPreserveAccessIndexAttr>();
3718e5dd7070Spatrick     return false;
3719e5dd7070Spatrick   }
3720e5dd7070Spatrick 
3721e5dd7070Spatrick   return false;
3722e5dd7070Spatrick }
3723e5dd7070Spatrick 
emitArraySubscriptGEP(CodeGenFunction & CGF,Address addr,ArrayRef<llvm::Value * > indices,QualType eltType,bool inbounds,bool signedIndices,SourceLocation loc,QualType * arrayType=nullptr,const Expr * Base=nullptr,const llvm::Twine & name="arrayidx")3724e5dd7070Spatrick static Address emitArraySubscriptGEP(CodeGenFunction &CGF, Address addr,
3725e5dd7070Spatrick                                      ArrayRef<llvm::Value *> indices,
3726e5dd7070Spatrick                                      QualType eltType, bool inbounds,
3727e5dd7070Spatrick                                      bool signedIndices, SourceLocation loc,
3728e5dd7070Spatrick                                      QualType *arrayType = nullptr,
3729e5dd7070Spatrick                                      const Expr *Base = nullptr,
3730e5dd7070Spatrick                                      const llvm::Twine &name = "arrayidx") {
3731e5dd7070Spatrick   // All the indices except that last must be zero.
3732e5dd7070Spatrick #ifndef NDEBUG
3733*12c85518Srobert   for (auto *idx : indices.drop_back())
3734e5dd7070Spatrick     assert(isa<llvm::ConstantInt>(idx) &&
3735e5dd7070Spatrick            cast<llvm::ConstantInt>(idx)->isZero());
3736e5dd7070Spatrick #endif
3737e5dd7070Spatrick 
3738e5dd7070Spatrick   // Determine the element size of the statically-sized base.  This is
3739e5dd7070Spatrick   // the thing that the indices are expressed in terms of.
3740e5dd7070Spatrick   if (auto vla = CGF.getContext().getAsVariableArrayType(eltType)) {
3741e5dd7070Spatrick     eltType = getFixedSizeElementType(CGF.getContext(), vla);
3742e5dd7070Spatrick   }
3743e5dd7070Spatrick 
3744e5dd7070Spatrick   // We can use that to compute the best alignment of the element.
3745e5dd7070Spatrick   CharUnits eltSize = CGF.getContext().getTypeSizeInChars(eltType);
3746e5dd7070Spatrick   CharUnits eltAlign =
3747e5dd7070Spatrick     getArrayElementAlign(addr.getAlignment(), indices.back(), eltSize);
3748e5dd7070Spatrick 
3749e5dd7070Spatrick   llvm::Value *eltPtr;
3750e5dd7070Spatrick   auto LastIndex = dyn_cast<llvm::ConstantInt>(indices.back());
3751e5dd7070Spatrick   if (!LastIndex ||
3752e5dd7070Spatrick       (!CGF.IsInPreservedAIRegion && !IsPreserveAIArrayBase(CGF, Base))) {
3753e5dd7070Spatrick     eltPtr = emitArraySubscriptGEP(
3754a9ac8606Spatrick         CGF, addr.getElementType(), addr.getPointer(), indices, inbounds,
3755a9ac8606Spatrick         signedIndices, loc, name);
3756e5dd7070Spatrick   } else {
3757e5dd7070Spatrick     // Remember the original array subscript for bpf target
3758e5dd7070Spatrick     unsigned idx = LastIndex->getZExtValue();
3759e5dd7070Spatrick     llvm::DIType *DbgInfo = nullptr;
3760e5dd7070Spatrick     if (arrayType)
3761e5dd7070Spatrick       DbgInfo = CGF.getDebugInfo()->getOrCreateStandaloneType(*arrayType, loc);
3762e5dd7070Spatrick     eltPtr = CGF.Builder.CreatePreserveArrayAccessIndex(addr.getElementType(),
3763e5dd7070Spatrick                                                         addr.getPointer(),
3764e5dd7070Spatrick                                                         indices.size() - 1,
3765e5dd7070Spatrick                                                         idx, DbgInfo);
3766e5dd7070Spatrick   }
3767e5dd7070Spatrick 
3768*12c85518Srobert   return Address(eltPtr, CGF.ConvertTypeForMem(eltType), eltAlign);
3769e5dd7070Spatrick }
3770e5dd7070Spatrick 
EmitArraySubscriptExpr(const ArraySubscriptExpr * E,bool Accessed)3771e5dd7070Spatrick LValue CodeGenFunction::EmitArraySubscriptExpr(const ArraySubscriptExpr *E,
3772e5dd7070Spatrick                                                bool Accessed) {
3773e5dd7070Spatrick   // The index must always be an integer, which is not an aggregate.  Emit it
3774e5dd7070Spatrick   // in lexical order (this complexity is, sadly, required by C++17).
3775e5dd7070Spatrick   llvm::Value *IdxPre =
3776e5dd7070Spatrick       (E->getLHS() == E->getIdx()) ? EmitScalarExpr(E->getIdx()) : nullptr;
3777e5dd7070Spatrick   bool SignedIndices = false;
3778e5dd7070Spatrick   auto EmitIdxAfterBase = [&, IdxPre](bool Promote) -> llvm::Value * {
3779e5dd7070Spatrick     auto *Idx = IdxPre;
3780e5dd7070Spatrick     if (E->getLHS() != E->getIdx()) {
3781e5dd7070Spatrick       assert(E->getRHS() == E->getIdx() && "index was neither LHS nor RHS");
3782e5dd7070Spatrick       Idx = EmitScalarExpr(E->getIdx());
3783e5dd7070Spatrick     }
3784e5dd7070Spatrick 
3785e5dd7070Spatrick     QualType IdxTy = E->getIdx()->getType();
3786e5dd7070Spatrick     bool IdxSigned = IdxTy->isSignedIntegerOrEnumerationType();
3787e5dd7070Spatrick     SignedIndices |= IdxSigned;
3788e5dd7070Spatrick 
3789e5dd7070Spatrick     if (SanOpts.has(SanitizerKind::ArrayBounds))
3790e5dd7070Spatrick       EmitBoundsCheck(E, E->getBase(), Idx, IdxTy, Accessed);
3791e5dd7070Spatrick 
3792e5dd7070Spatrick     // Extend or truncate the index type to 32 or 64-bits.
3793e5dd7070Spatrick     if (Promote && Idx->getType() != IntPtrTy)
3794e5dd7070Spatrick       Idx = Builder.CreateIntCast(Idx, IntPtrTy, IdxSigned, "idxprom");
3795e5dd7070Spatrick 
3796e5dd7070Spatrick     return Idx;
3797e5dd7070Spatrick   };
3798e5dd7070Spatrick   IdxPre = nullptr;
3799e5dd7070Spatrick 
3800e5dd7070Spatrick   // If the base is a vector type, then we are forming a vector element lvalue
3801e5dd7070Spatrick   // with this subscript.
3802e5dd7070Spatrick   if (E->getBase()->getType()->isVectorType() &&
3803e5dd7070Spatrick       !isa<ExtVectorElementExpr>(E->getBase())) {
3804e5dd7070Spatrick     // Emit the vector as an lvalue to get its address.
3805e5dd7070Spatrick     LValue LHS = EmitLValue(E->getBase());
3806e5dd7070Spatrick     auto *Idx = EmitIdxAfterBase(/*Promote*/false);
3807e5dd7070Spatrick     assert(LHS.isSimple() && "Can only subscript lvalue vectors here!");
3808e5dd7070Spatrick     return LValue::MakeVectorElt(LHS.getAddress(*this), Idx,
3809e5dd7070Spatrick                                  E->getBase()->getType(), LHS.getBaseInfo(),
3810e5dd7070Spatrick                                  TBAAAccessInfo());
3811e5dd7070Spatrick   }
3812e5dd7070Spatrick 
3813e5dd7070Spatrick   // All the other cases basically behave like simple offsetting.
3814e5dd7070Spatrick 
3815e5dd7070Spatrick   // Handle the extvector case we ignored above.
3816e5dd7070Spatrick   if (isa<ExtVectorElementExpr>(E->getBase())) {
3817e5dd7070Spatrick     LValue LV = EmitLValue(E->getBase());
3818e5dd7070Spatrick     auto *Idx = EmitIdxAfterBase(/*Promote*/true);
3819e5dd7070Spatrick     Address Addr = EmitExtVectorElementLValue(LV);
3820e5dd7070Spatrick 
3821e5dd7070Spatrick     QualType EltType = LV.getType()->castAs<VectorType>()->getElementType();
3822e5dd7070Spatrick     Addr = emitArraySubscriptGEP(*this, Addr, Idx, EltType, /*inbounds*/ true,
3823e5dd7070Spatrick                                  SignedIndices, E->getExprLoc());
3824e5dd7070Spatrick     return MakeAddrLValue(Addr, EltType, LV.getBaseInfo(),
3825e5dd7070Spatrick                           CGM.getTBAAInfoForSubobject(LV, EltType));
3826e5dd7070Spatrick   }
3827e5dd7070Spatrick 
3828e5dd7070Spatrick   LValueBaseInfo EltBaseInfo;
3829e5dd7070Spatrick   TBAAAccessInfo EltTBAAInfo;
3830e5dd7070Spatrick   Address Addr = Address::invalid();
3831e5dd7070Spatrick   if (const VariableArrayType *vla =
3832e5dd7070Spatrick            getContext().getAsVariableArrayType(E->getType())) {
3833e5dd7070Spatrick     // The base must be a pointer, which is not an aggregate.  Emit
3834e5dd7070Spatrick     // it.  It needs to be emitted first in case it's what captures
3835e5dd7070Spatrick     // the VLA bounds.
3836e5dd7070Spatrick     Addr = EmitPointerWithAlignment(E->getBase(), &EltBaseInfo, &EltTBAAInfo);
3837e5dd7070Spatrick     auto *Idx = EmitIdxAfterBase(/*Promote*/true);
3838e5dd7070Spatrick 
3839e5dd7070Spatrick     // The element count here is the total number of non-VLA elements.
3840e5dd7070Spatrick     llvm::Value *numElements = getVLASize(vla).NumElts;
3841e5dd7070Spatrick 
3842e5dd7070Spatrick     // Effectively, the multiply by the VLA size is part of the GEP.
3843e5dd7070Spatrick     // GEP indexes are signed, and scaling an index isn't permitted to
3844e5dd7070Spatrick     // signed-overflow, so we use the same semantics for our explicit
3845e5dd7070Spatrick     // multiply.  We suppress this if overflow is not undefined behavior.
3846e5dd7070Spatrick     if (getLangOpts().isSignedOverflowDefined()) {
3847e5dd7070Spatrick       Idx = Builder.CreateMul(Idx, numElements);
3848e5dd7070Spatrick     } else {
3849e5dd7070Spatrick       Idx = Builder.CreateNSWMul(Idx, numElements);
3850e5dd7070Spatrick     }
3851e5dd7070Spatrick 
3852e5dd7070Spatrick     Addr = emitArraySubscriptGEP(*this, Addr, Idx, vla->getElementType(),
3853e5dd7070Spatrick                                  !getLangOpts().isSignedOverflowDefined(),
3854e5dd7070Spatrick                                  SignedIndices, E->getExprLoc());
3855e5dd7070Spatrick 
3856e5dd7070Spatrick   } else if (const ObjCObjectType *OIT = E->getType()->getAs<ObjCObjectType>()){
3857e5dd7070Spatrick     // Indexing over an interface, as in "NSString *P; P[4];"
3858e5dd7070Spatrick 
3859e5dd7070Spatrick     // Emit the base pointer.
3860e5dd7070Spatrick     Addr = EmitPointerWithAlignment(E->getBase(), &EltBaseInfo, &EltTBAAInfo);
3861e5dd7070Spatrick     auto *Idx = EmitIdxAfterBase(/*Promote*/true);
3862e5dd7070Spatrick 
3863e5dd7070Spatrick     CharUnits InterfaceSize = getContext().getTypeSizeInChars(OIT);
3864e5dd7070Spatrick     llvm::Value *InterfaceSizeVal =
3865e5dd7070Spatrick         llvm::ConstantInt::get(Idx->getType(), InterfaceSize.getQuantity());
3866e5dd7070Spatrick 
3867e5dd7070Spatrick     llvm::Value *ScaledIdx = Builder.CreateMul(Idx, InterfaceSizeVal);
3868e5dd7070Spatrick 
3869e5dd7070Spatrick     // We don't necessarily build correct LLVM struct types for ObjC
3870e5dd7070Spatrick     // interfaces, so we can't rely on GEP to do this scaling
3871e5dd7070Spatrick     // correctly, so we need to cast to i8*.  FIXME: is this actually
3872e5dd7070Spatrick     // true?  A lot of other things in the fragile ABI would break...
3873*12c85518Srobert     llvm::Type *OrigBaseElemTy = Addr.getElementType();
3874e5dd7070Spatrick     Addr = Builder.CreateElementBitCast(Addr, Int8Ty);
3875e5dd7070Spatrick 
3876e5dd7070Spatrick     // Do the GEP.
3877e5dd7070Spatrick     CharUnits EltAlign =
3878e5dd7070Spatrick       getArrayElementAlign(Addr.getAlignment(), Idx, InterfaceSize);
3879e5dd7070Spatrick     llvm::Value *EltPtr =
3880a9ac8606Spatrick         emitArraySubscriptGEP(*this, Addr.getElementType(), Addr.getPointer(),
3881a9ac8606Spatrick                               ScaledIdx, false, SignedIndices, E->getExprLoc());
3882*12c85518Srobert     Addr = Address(EltPtr, Addr.getElementType(), EltAlign);
3883e5dd7070Spatrick 
3884e5dd7070Spatrick     // Cast back.
3885*12c85518Srobert     Addr = Builder.CreateElementBitCast(Addr, OrigBaseElemTy);
3886e5dd7070Spatrick   } else if (const Expr *Array = isSimpleArrayDecayOperand(E->getBase())) {
3887e5dd7070Spatrick     // If this is A[i] where A is an array, the frontend will have decayed the
3888e5dd7070Spatrick     // base to be a ArrayToPointerDecay implicit cast.  While correct, it is
3889e5dd7070Spatrick     // inefficient at -O0 to emit a "gep A, 0, 0" when codegen'ing it, then a
3890e5dd7070Spatrick     // "gep x, i" here.  Emit one "gep A, 0, i".
3891e5dd7070Spatrick     assert(Array->getType()->isArrayType() &&
3892e5dd7070Spatrick            "Array to pointer decay must have array source type!");
3893e5dd7070Spatrick     LValue ArrayLV;
3894e5dd7070Spatrick     // For simple multidimensional array indexing, set the 'accessed' flag for
3895e5dd7070Spatrick     // better bounds-checking of the base expression.
3896e5dd7070Spatrick     if (const auto *ASE = dyn_cast<ArraySubscriptExpr>(Array))
3897e5dd7070Spatrick       ArrayLV = EmitArraySubscriptExpr(ASE, /*Accessed*/ true);
3898e5dd7070Spatrick     else
3899e5dd7070Spatrick       ArrayLV = EmitLValue(Array);
3900e5dd7070Spatrick     auto *Idx = EmitIdxAfterBase(/*Promote*/true);
3901e5dd7070Spatrick 
3902e5dd7070Spatrick     // Propagate the alignment from the array itself to the result.
3903e5dd7070Spatrick     QualType arrayType = Array->getType();
3904e5dd7070Spatrick     Addr = emitArraySubscriptGEP(
3905e5dd7070Spatrick         *this, ArrayLV.getAddress(*this), {CGM.getSize(CharUnits::Zero()), Idx},
3906e5dd7070Spatrick         E->getType(), !getLangOpts().isSignedOverflowDefined(), SignedIndices,
3907e5dd7070Spatrick         E->getExprLoc(), &arrayType, E->getBase());
3908e5dd7070Spatrick     EltBaseInfo = ArrayLV.getBaseInfo();
3909e5dd7070Spatrick     EltTBAAInfo = CGM.getTBAAInfoForSubobject(ArrayLV, E->getType());
3910e5dd7070Spatrick   } else {
3911e5dd7070Spatrick     // The base must be a pointer; emit it with an estimate of its alignment.
3912e5dd7070Spatrick     Addr = EmitPointerWithAlignment(E->getBase(), &EltBaseInfo, &EltTBAAInfo);
3913e5dd7070Spatrick     auto *Idx = EmitIdxAfterBase(/*Promote*/true);
3914e5dd7070Spatrick     QualType ptrType = E->getBase()->getType();
3915e5dd7070Spatrick     Addr = emitArraySubscriptGEP(*this, Addr, Idx, E->getType(),
3916e5dd7070Spatrick                                  !getLangOpts().isSignedOverflowDefined(),
3917e5dd7070Spatrick                                  SignedIndices, E->getExprLoc(), &ptrType,
3918e5dd7070Spatrick                                  E->getBase());
3919e5dd7070Spatrick   }
3920e5dd7070Spatrick 
3921e5dd7070Spatrick   LValue LV = MakeAddrLValue(Addr, E->getType(), EltBaseInfo, EltTBAAInfo);
3922e5dd7070Spatrick 
3923e5dd7070Spatrick   if (getLangOpts().ObjC &&
3924e5dd7070Spatrick       getLangOpts().getGC() != LangOptions::NonGC) {
3925e5dd7070Spatrick     LV.setNonGC(!E->isOBJCGCCandidate(getContext()));
3926e5dd7070Spatrick     setObjCGCLValueClass(getContext(), E, LV);
3927e5dd7070Spatrick   }
3928e5dd7070Spatrick   return LV;
3929e5dd7070Spatrick }
3930e5dd7070Spatrick 
EmitMatrixSubscriptExpr(const MatrixSubscriptExpr * E)3931ec727ea7Spatrick LValue CodeGenFunction::EmitMatrixSubscriptExpr(const MatrixSubscriptExpr *E) {
3932ec727ea7Spatrick   assert(
3933ec727ea7Spatrick       !E->isIncomplete() &&
3934ec727ea7Spatrick       "incomplete matrix subscript expressions should be rejected during Sema");
3935ec727ea7Spatrick   LValue Base = EmitLValue(E->getBase());
3936ec727ea7Spatrick   llvm::Value *RowIdx = EmitScalarExpr(E->getRowIdx());
3937ec727ea7Spatrick   llvm::Value *ColIdx = EmitScalarExpr(E->getColumnIdx());
3938ec727ea7Spatrick   llvm::Value *NumRows = Builder.getIntN(
3939ec727ea7Spatrick       RowIdx->getType()->getScalarSizeInBits(),
3940a9ac8606Spatrick       E->getBase()->getType()->castAs<ConstantMatrixType>()->getNumRows());
3941ec727ea7Spatrick   llvm::Value *FinalIdx =
3942ec727ea7Spatrick       Builder.CreateAdd(Builder.CreateMul(ColIdx, NumRows), RowIdx);
3943ec727ea7Spatrick   return LValue::MakeMatrixElt(
3944ec727ea7Spatrick       MaybeConvertMatrixAddress(Base.getAddress(*this), *this), FinalIdx,
3945ec727ea7Spatrick       E->getBase()->getType(), Base.getBaseInfo(), TBAAAccessInfo());
3946ec727ea7Spatrick }
3947ec727ea7Spatrick 
emitOMPArraySectionBase(CodeGenFunction & CGF,const Expr * Base,LValueBaseInfo & BaseInfo,TBAAAccessInfo & TBAAInfo,QualType BaseTy,QualType ElTy,bool IsLowerBound)3948e5dd7070Spatrick static Address emitOMPArraySectionBase(CodeGenFunction &CGF, const Expr *Base,
3949e5dd7070Spatrick                                        LValueBaseInfo &BaseInfo,
3950e5dd7070Spatrick                                        TBAAAccessInfo &TBAAInfo,
3951e5dd7070Spatrick                                        QualType BaseTy, QualType ElTy,
3952e5dd7070Spatrick                                        bool IsLowerBound) {
3953e5dd7070Spatrick   LValue BaseLVal;
3954e5dd7070Spatrick   if (auto *ASE = dyn_cast<OMPArraySectionExpr>(Base->IgnoreParenImpCasts())) {
3955e5dd7070Spatrick     BaseLVal = CGF.EmitOMPArraySectionExpr(ASE, IsLowerBound);
3956e5dd7070Spatrick     if (BaseTy->isArrayType()) {
3957e5dd7070Spatrick       Address Addr = BaseLVal.getAddress(CGF);
3958e5dd7070Spatrick       BaseInfo = BaseLVal.getBaseInfo();
3959e5dd7070Spatrick 
3960e5dd7070Spatrick       // If the array type was an incomplete type, we need to make sure
3961e5dd7070Spatrick       // the decay ends up being the right type.
3962e5dd7070Spatrick       llvm::Type *NewTy = CGF.ConvertType(BaseTy);
3963e5dd7070Spatrick       Addr = CGF.Builder.CreateElementBitCast(Addr, NewTy);
3964e5dd7070Spatrick 
3965e5dd7070Spatrick       // Note that VLA pointers are always decayed, so we don't need to do
3966e5dd7070Spatrick       // anything here.
3967e5dd7070Spatrick       if (!BaseTy->isVariableArrayType()) {
3968e5dd7070Spatrick         assert(isa<llvm::ArrayType>(Addr.getElementType()) &&
3969e5dd7070Spatrick                "Expected pointer to array");
3970e5dd7070Spatrick         Addr = CGF.Builder.CreateConstArrayGEP(Addr, 0, "arraydecay");
3971e5dd7070Spatrick       }
3972e5dd7070Spatrick 
3973e5dd7070Spatrick       return CGF.Builder.CreateElementBitCast(Addr,
3974e5dd7070Spatrick                                               CGF.ConvertTypeForMem(ElTy));
3975e5dd7070Spatrick     }
3976e5dd7070Spatrick     LValueBaseInfo TypeBaseInfo;
3977e5dd7070Spatrick     TBAAAccessInfo TypeTBAAInfo;
3978ec727ea7Spatrick     CharUnits Align =
3979ec727ea7Spatrick         CGF.CGM.getNaturalTypeAlignment(ElTy, &TypeBaseInfo, &TypeTBAAInfo);
3980e5dd7070Spatrick     BaseInfo.mergeForCast(TypeBaseInfo);
3981e5dd7070Spatrick     TBAAInfo = CGF.CGM.mergeTBAAInfoForCast(TBAAInfo, TypeTBAAInfo);
3982*12c85518Srobert     return Address(CGF.Builder.CreateLoad(BaseLVal.getAddress(CGF)),
3983*12c85518Srobert                    CGF.ConvertTypeForMem(ElTy), Align);
3984e5dd7070Spatrick   }
3985e5dd7070Spatrick   return CGF.EmitPointerWithAlignment(Base, &BaseInfo, &TBAAInfo);
3986e5dd7070Spatrick }
3987e5dd7070Spatrick 
EmitOMPArraySectionExpr(const OMPArraySectionExpr * E,bool IsLowerBound)3988e5dd7070Spatrick LValue CodeGenFunction::EmitOMPArraySectionExpr(const OMPArraySectionExpr *E,
3989e5dd7070Spatrick                                                 bool IsLowerBound) {
3990e5dd7070Spatrick   QualType BaseTy = OMPArraySectionExpr::getBaseOriginalType(E->getBase());
3991e5dd7070Spatrick   QualType ResultExprTy;
3992e5dd7070Spatrick   if (auto *AT = getContext().getAsArrayType(BaseTy))
3993e5dd7070Spatrick     ResultExprTy = AT->getElementType();
3994e5dd7070Spatrick   else
3995e5dd7070Spatrick     ResultExprTy = BaseTy->getPointeeType();
3996e5dd7070Spatrick   llvm::Value *Idx = nullptr;
3997ec727ea7Spatrick   if (IsLowerBound || E->getColonLocFirst().isInvalid()) {
3998e5dd7070Spatrick     // Requesting lower bound or upper bound, but without provided length and
3999e5dd7070Spatrick     // without ':' symbol for the default length -> length = 1.
4000e5dd7070Spatrick     // Idx = LowerBound ?: 0;
4001e5dd7070Spatrick     if (auto *LowerBound = E->getLowerBound()) {
4002e5dd7070Spatrick       Idx = Builder.CreateIntCast(
4003e5dd7070Spatrick           EmitScalarExpr(LowerBound), IntPtrTy,
4004e5dd7070Spatrick           LowerBound->getType()->hasSignedIntegerRepresentation());
4005e5dd7070Spatrick     } else
4006e5dd7070Spatrick       Idx = llvm::ConstantInt::getNullValue(IntPtrTy);
4007e5dd7070Spatrick   } else {
4008e5dd7070Spatrick     // Try to emit length or lower bound as constant. If this is possible, 1
4009e5dd7070Spatrick     // is subtracted from constant length or lower bound. Otherwise, emit LLVM
4010e5dd7070Spatrick     // IR (LB + Len) - 1.
4011e5dd7070Spatrick     auto &C = CGM.getContext();
4012e5dd7070Spatrick     auto *Length = E->getLength();
4013e5dd7070Spatrick     llvm::APSInt ConstLength;
4014e5dd7070Spatrick     if (Length) {
4015e5dd7070Spatrick       // Idx = LowerBound + Length - 1;
4016*12c85518Srobert       if (std::optional<llvm::APSInt> CL = Length->getIntegerConstantExpr(C)) {
4017a9ac8606Spatrick         ConstLength = CL->zextOrTrunc(PointerWidthInBits);
4018e5dd7070Spatrick         Length = nullptr;
4019e5dd7070Spatrick       }
4020e5dd7070Spatrick       auto *LowerBound = E->getLowerBound();
4021e5dd7070Spatrick       llvm::APSInt ConstLowerBound(PointerWidthInBits, /*isUnsigned=*/false);
4022a9ac8606Spatrick       if (LowerBound) {
4023*12c85518Srobert         if (std::optional<llvm::APSInt> LB =
4024*12c85518Srobert                 LowerBound->getIntegerConstantExpr(C)) {
4025a9ac8606Spatrick           ConstLowerBound = LB->zextOrTrunc(PointerWidthInBits);
4026e5dd7070Spatrick           LowerBound = nullptr;
4027e5dd7070Spatrick         }
4028a9ac8606Spatrick       }
4029e5dd7070Spatrick       if (!Length)
4030e5dd7070Spatrick         --ConstLength;
4031e5dd7070Spatrick       else if (!LowerBound)
4032e5dd7070Spatrick         --ConstLowerBound;
4033e5dd7070Spatrick 
4034e5dd7070Spatrick       if (Length || LowerBound) {
4035e5dd7070Spatrick         auto *LowerBoundVal =
4036e5dd7070Spatrick             LowerBound
4037e5dd7070Spatrick                 ? Builder.CreateIntCast(
4038e5dd7070Spatrick                       EmitScalarExpr(LowerBound), IntPtrTy,
4039e5dd7070Spatrick                       LowerBound->getType()->hasSignedIntegerRepresentation())
4040e5dd7070Spatrick                 : llvm::ConstantInt::get(IntPtrTy, ConstLowerBound);
4041e5dd7070Spatrick         auto *LengthVal =
4042e5dd7070Spatrick             Length
4043e5dd7070Spatrick                 ? Builder.CreateIntCast(
4044e5dd7070Spatrick                       EmitScalarExpr(Length), IntPtrTy,
4045e5dd7070Spatrick                       Length->getType()->hasSignedIntegerRepresentation())
4046e5dd7070Spatrick                 : llvm::ConstantInt::get(IntPtrTy, ConstLength);
4047e5dd7070Spatrick         Idx = Builder.CreateAdd(LowerBoundVal, LengthVal, "lb_add_len",
4048e5dd7070Spatrick                                 /*HasNUW=*/false,
4049e5dd7070Spatrick                                 !getLangOpts().isSignedOverflowDefined());
4050e5dd7070Spatrick         if (Length && LowerBound) {
4051e5dd7070Spatrick           Idx = Builder.CreateSub(
4052e5dd7070Spatrick               Idx, llvm::ConstantInt::get(IntPtrTy, /*V=*/1), "idx_sub_1",
4053e5dd7070Spatrick               /*HasNUW=*/false, !getLangOpts().isSignedOverflowDefined());
4054e5dd7070Spatrick         }
4055e5dd7070Spatrick       } else
4056e5dd7070Spatrick         Idx = llvm::ConstantInt::get(IntPtrTy, ConstLength + ConstLowerBound);
4057e5dd7070Spatrick     } else {
4058e5dd7070Spatrick       // Idx = ArraySize - 1;
4059e5dd7070Spatrick       QualType ArrayTy = BaseTy->isPointerType()
4060e5dd7070Spatrick                              ? E->getBase()->IgnoreParenImpCasts()->getType()
4061e5dd7070Spatrick                              : BaseTy;
4062e5dd7070Spatrick       if (auto *VAT = C.getAsVariableArrayType(ArrayTy)) {
4063e5dd7070Spatrick         Length = VAT->getSizeExpr();
4064*12c85518Srobert         if (std::optional<llvm::APSInt> L = Length->getIntegerConstantExpr(C)) {
4065a9ac8606Spatrick           ConstLength = *L;
4066e5dd7070Spatrick           Length = nullptr;
4067a9ac8606Spatrick         }
4068e5dd7070Spatrick       } else {
4069e5dd7070Spatrick         auto *CAT = C.getAsConstantArrayType(ArrayTy);
4070e5dd7070Spatrick         ConstLength = CAT->getSize();
4071e5dd7070Spatrick       }
4072e5dd7070Spatrick       if (Length) {
4073e5dd7070Spatrick         auto *LengthVal = Builder.CreateIntCast(
4074e5dd7070Spatrick             EmitScalarExpr(Length), IntPtrTy,
4075e5dd7070Spatrick             Length->getType()->hasSignedIntegerRepresentation());
4076e5dd7070Spatrick         Idx = Builder.CreateSub(
4077e5dd7070Spatrick             LengthVal, llvm::ConstantInt::get(IntPtrTy, /*V=*/1), "len_sub_1",
4078e5dd7070Spatrick             /*HasNUW=*/false, !getLangOpts().isSignedOverflowDefined());
4079e5dd7070Spatrick       } else {
4080e5dd7070Spatrick         ConstLength = ConstLength.zextOrTrunc(PointerWidthInBits);
4081e5dd7070Spatrick         --ConstLength;
4082e5dd7070Spatrick         Idx = llvm::ConstantInt::get(IntPtrTy, ConstLength);
4083e5dd7070Spatrick       }
4084e5dd7070Spatrick     }
4085e5dd7070Spatrick   }
4086e5dd7070Spatrick   assert(Idx);
4087e5dd7070Spatrick 
4088e5dd7070Spatrick   Address EltPtr = Address::invalid();
4089e5dd7070Spatrick   LValueBaseInfo BaseInfo;
4090e5dd7070Spatrick   TBAAAccessInfo TBAAInfo;
4091e5dd7070Spatrick   if (auto *VLA = getContext().getAsVariableArrayType(ResultExprTy)) {
4092e5dd7070Spatrick     // The base must be a pointer, which is not an aggregate.  Emit
4093e5dd7070Spatrick     // it.  It needs to be emitted first in case it's what captures
4094e5dd7070Spatrick     // the VLA bounds.
4095e5dd7070Spatrick     Address Base =
4096e5dd7070Spatrick         emitOMPArraySectionBase(*this, E->getBase(), BaseInfo, TBAAInfo,
4097e5dd7070Spatrick                                 BaseTy, VLA->getElementType(), IsLowerBound);
4098e5dd7070Spatrick     // The element count here is the total number of non-VLA elements.
4099e5dd7070Spatrick     llvm::Value *NumElements = getVLASize(VLA).NumElts;
4100e5dd7070Spatrick 
4101e5dd7070Spatrick     // Effectively, the multiply by the VLA size is part of the GEP.
4102e5dd7070Spatrick     // GEP indexes are signed, and scaling an index isn't permitted to
4103e5dd7070Spatrick     // signed-overflow, so we use the same semantics for our explicit
4104e5dd7070Spatrick     // multiply.  We suppress this if overflow is not undefined behavior.
4105e5dd7070Spatrick     if (getLangOpts().isSignedOverflowDefined())
4106e5dd7070Spatrick       Idx = Builder.CreateMul(Idx, NumElements);
4107e5dd7070Spatrick     else
4108e5dd7070Spatrick       Idx = Builder.CreateNSWMul(Idx, NumElements);
4109e5dd7070Spatrick     EltPtr = emitArraySubscriptGEP(*this, Base, Idx, VLA->getElementType(),
4110e5dd7070Spatrick                                    !getLangOpts().isSignedOverflowDefined(),
4111e5dd7070Spatrick                                    /*signedIndices=*/false, E->getExprLoc());
4112e5dd7070Spatrick   } else if (const Expr *Array = isSimpleArrayDecayOperand(E->getBase())) {
4113e5dd7070Spatrick     // If this is A[i] where A is an array, the frontend will have decayed the
4114e5dd7070Spatrick     // base to be a ArrayToPointerDecay implicit cast.  While correct, it is
4115e5dd7070Spatrick     // inefficient at -O0 to emit a "gep A, 0, 0" when codegen'ing it, then a
4116e5dd7070Spatrick     // "gep x, i" here.  Emit one "gep A, 0, i".
4117e5dd7070Spatrick     assert(Array->getType()->isArrayType() &&
4118e5dd7070Spatrick            "Array to pointer decay must have array source type!");
4119e5dd7070Spatrick     LValue ArrayLV;
4120e5dd7070Spatrick     // For simple multidimensional array indexing, set the 'accessed' flag for
4121e5dd7070Spatrick     // better bounds-checking of the base expression.
4122e5dd7070Spatrick     if (const auto *ASE = dyn_cast<ArraySubscriptExpr>(Array))
4123e5dd7070Spatrick       ArrayLV = EmitArraySubscriptExpr(ASE, /*Accessed*/ true);
4124e5dd7070Spatrick     else
4125e5dd7070Spatrick       ArrayLV = EmitLValue(Array);
4126e5dd7070Spatrick 
4127e5dd7070Spatrick     // Propagate the alignment from the array itself to the result.
4128e5dd7070Spatrick     EltPtr = emitArraySubscriptGEP(
4129e5dd7070Spatrick         *this, ArrayLV.getAddress(*this), {CGM.getSize(CharUnits::Zero()), Idx},
4130e5dd7070Spatrick         ResultExprTy, !getLangOpts().isSignedOverflowDefined(),
4131e5dd7070Spatrick         /*signedIndices=*/false, E->getExprLoc());
4132e5dd7070Spatrick     BaseInfo = ArrayLV.getBaseInfo();
4133e5dd7070Spatrick     TBAAInfo = CGM.getTBAAInfoForSubobject(ArrayLV, ResultExprTy);
4134e5dd7070Spatrick   } else {
4135e5dd7070Spatrick     Address Base = emitOMPArraySectionBase(*this, E->getBase(), BaseInfo,
4136e5dd7070Spatrick                                            TBAAInfo, BaseTy, ResultExprTy,
4137e5dd7070Spatrick                                            IsLowerBound);
4138e5dd7070Spatrick     EltPtr = emitArraySubscriptGEP(*this, Base, Idx, ResultExprTy,
4139e5dd7070Spatrick                                    !getLangOpts().isSignedOverflowDefined(),
4140e5dd7070Spatrick                                    /*signedIndices=*/false, E->getExprLoc());
4141e5dd7070Spatrick   }
4142e5dd7070Spatrick 
4143e5dd7070Spatrick   return MakeAddrLValue(EltPtr, ResultExprTy, BaseInfo, TBAAInfo);
4144e5dd7070Spatrick }
4145e5dd7070Spatrick 
4146e5dd7070Spatrick LValue CodeGenFunction::
EmitExtVectorElementExpr(const ExtVectorElementExpr * E)4147e5dd7070Spatrick EmitExtVectorElementExpr(const ExtVectorElementExpr *E) {
4148e5dd7070Spatrick   // Emit the base vector as an l-value.
4149e5dd7070Spatrick   LValue Base;
4150e5dd7070Spatrick 
4151e5dd7070Spatrick   // ExtVectorElementExpr's base can either be a vector or pointer to vector.
4152e5dd7070Spatrick   if (E->isArrow()) {
4153e5dd7070Spatrick     // If it is a pointer to a vector, emit the address and form an lvalue with
4154e5dd7070Spatrick     // it.
4155e5dd7070Spatrick     LValueBaseInfo BaseInfo;
4156e5dd7070Spatrick     TBAAAccessInfo TBAAInfo;
4157e5dd7070Spatrick     Address Ptr = EmitPointerWithAlignment(E->getBase(), &BaseInfo, &TBAAInfo);
4158e5dd7070Spatrick     const auto *PT = E->getBase()->getType()->castAs<PointerType>();
4159e5dd7070Spatrick     Base = MakeAddrLValue(Ptr, PT->getPointeeType(), BaseInfo, TBAAInfo);
4160e5dd7070Spatrick     Base.getQuals().removeObjCGCAttr();
4161e5dd7070Spatrick   } else if (E->getBase()->isGLValue()) {
4162e5dd7070Spatrick     // Otherwise, if the base is an lvalue ( as in the case of foo.x.x),
4163e5dd7070Spatrick     // emit the base as an lvalue.
4164e5dd7070Spatrick     assert(E->getBase()->getType()->isVectorType());
4165e5dd7070Spatrick     Base = EmitLValue(E->getBase());
4166e5dd7070Spatrick   } else {
4167e5dd7070Spatrick     // Otherwise, the base is a normal rvalue (as in (V+V).x), emit it as such.
4168e5dd7070Spatrick     assert(E->getBase()->getType()->isVectorType() &&
4169e5dd7070Spatrick            "Result must be a vector");
4170e5dd7070Spatrick     llvm::Value *Vec = EmitScalarExpr(E->getBase());
4171e5dd7070Spatrick 
4172e5dd7070Spatrick     // Store the vector to memory (because LValue wants an address).
4173e5dd7070Spatrick     Address VecMem = CreateMemTemp(E->getBase()->getType());
4174e5dd7070Spatrick     Builder.CreateStore(Vec, VecMem);
4175e5dd7070Spatrick     Base = MakeAddrLValue(VecMem, E->getBase()->getType(),
4176e5dd7070Spatrick                           AlignmentSource::Decl);
4177e5dd7070Spatrick   }
4178e5dd7070Spatrick 
4179e5dd7070Spatrick   QualType type =
4180e5dd7070Spatrick     E->getType().withCVRQualifiers(Base.getQuals().getCVRQualifiers());
4181e5dd7070Spatrick 
4182e5dd7070Spatrick   // Encode the element access list into a vector of unsigned indices.
4183e5dd7070Spatrick   SmallVector<uint32_t, 4> Indices;
4184e5dd7070Spatrick   E->getEncodedElementAccess(Indices);
4185e5dd7070Spatrick 
4186e5dd7070Spatrick   if (Base.isSimple()) {
4187e5dd7070Spatrick     llvm::Constant *CV =
4188e5dd7070Spatrick         llvm::ConstantDataVector::get(getLLVMContext(), Indices);
4189e5dd7070Spatrick     return LValue::MakeExtVectorElt(Base.getAddress(*this), CV, type,
4190e5dd7070Spatrick                                     Base.getBaseInfo(), TBAAAccessInfo());
4191e5dd7070Spatrick   }
4192e5dd7070Spatrick   assert(Base.isExtVectorElt() && "Can only subscript lvalue vec elts here!");
4193e5dd7070Spatrick 
4194e5dd7070Spatrick   llvm::Constant *BaseElts = Base.getExtVectorElts();
4195e5dd7070Spatrick   SmallVector<llvm::Constant *, 4> CElts;
4196e5dd7070Spatrick 
4197e5dd7070Spatrick   for (unsigned i = 0, e = Indices.size(); i != e; ++i)
4198e5dd7070Spatrick     CElts.push_back(BaseElts->getAggregateElement(Indices[i]));
4199e5dd7070Spatrick   llvm::Constant *CV = llvm::ConstantVector::get(CElts);
4200e5dd7070Spatrick   return LValue::MakeExtVectorElt(Base.getExtVectorAddress(), CV, type,
4201e5dd7070Spatrick                                   Base.getBaseInfo(), TBAAAccessInfo());
4202e5dd7070Spatrick }
4203e5dd7070Spatrick 
EmitMemberExpr(const MemberExpr * E)4204e5dd7070Spatrick LValue CodeGenFunction::EmitMemberExpr(const MemberExpr *E) {
4205e5dd7070Spatrick   if (DeclRefExpr *DRE = tryToConvertMemberExprToDeclRefExpr(*this, E)) {
4206e5dd7070Spatrick     EmitIgnoredExpr(E->getBase());
4207e5dd7070Spatrick     return EmitDeclRefLValue(DRE);
4208e5dd7070Spatrick   }
4209e5dd7070Spatrick 
4210e5dd7070Spatrick   Expr *BaseExpr = E->getBase();
4211e5dd7070Spatrick   // If this is s.x, emit s as an lvalue.  If it is s->x, emit s as a scalar.
4212e5dd7070Spatrick   LValue BaseLV;
4213e5dd7070Spatrick   if (E->isArrow()) {
4214e5dd7070Spatrick     LValueBaseInfo BaseInfo;
4215e5dd7070Spatrick     TBAAAccessInfo TBAAInfo;
4216e5dd7070Spatrick     Address Addr = EmitPointerWithAlignment(BaseExpr, &BaseInfo, &TBAAInfo);
4217e5dd7070Spatrick     QualType PtrTy = BaseExpr->getType()->getPointeeType();
4218e5dd7070Spatrick     SanitizerSet SkippedChecks;
4219e5dd7070Spatrick     bool IsBaseCXXThis = IsWrappedCXXThis(BaseExpr);
4220e5dd7070Spatrick     if (IsBaseCXXThis)
4221e5dd7070Spatrick       SkippedChecks.set(SanitizerKind::Alignment, true);
4222e5dd7070Spatrick     if (IsBaseCXXThis || isa<DeclRefExpr>(BaseExpr))
4223e5dd7070Spatrick       SkippedChecks.set(SanitizerKind::Null, true);
4224e5dd7070Spatrick     EmitTypeCheck(TCK_MemberAccess, E->getExprLoc(), Addr.getPointer(), PtrTy,
4225e5dd7070Spatrick                   /*Alignment=*/CharUnits::Zero(), SkippedChecks);
4226e5dd7070Spatrick     BaseLV = MakeAddrLValue(Addr, PtrTy, BaseInfo, TBAAInfo);
4227e5dd7070Spatrick   } else
4228e5dd7070Spatrick     BaseLV = EmitCheckedLValue(BaseExpr, TCK_MemberAccess);
4229e5dd7070Spatrick 
4230e5dd7070Spatrick   NamedDecl *ND = E->getMemberDecl();
4231e5dd7070Spatrick   if (auto *Field = dyn_cast<FieldDecl>(ND)) {
4232e5dd7070Spatrick     LValue LV = EmitLValueForField(BaseLV, Field);
4233e5dd7070Spatrick     setObjCGCLValueClass(getContext(), E, LV);
4234e5dd7070Spatrick     if (getLangOpts().OpenMP) {
4235e5dd7070Spatrick       // If the member was explicitly marked as nontemporal, mark it as
4236e5dd7070Spatrick       // nontemporal. If the base lvalue is marked as nontemporal, mark access
4237e5dd7070Spatrick       // to children as nontemporal too.
4238e5dd7070Spatrick       if ((IsWrappedCXXThis(BaseExpr) &&
4239e5dd7070Spatrick            CGM.getOpenMPRuntime().isNontemporalDecl(Field)) ||
4240e5dd7070Spatrick           BaseLV.isNontemporal())
4241e5dd7070Spatrick         LV.setNontemporal(/*Value=*/true);
4242e5dd7070Spatrick     }
4243e5dd7070Spatrick     return LV;
4244e5dd7070Spatrick   }
4245e5dd7070Spatrick 
4246e5dd7070Spatrick   if (const auto *FD = dyn_cast<FunctionDecl>(ND))
4247e5dd7070Spatrick     return EmitFunctionDeclLValue(*this, E, FD);
4248e5dd7070Spatrick 
4249e5dd7070Spatrick   llvm_unreachable("Unhandled member declaration!");
4250e5dd7070Spatrick }
4251e5dd7070Spatrick 
4252e5dd7070Spatrick /// Given that we are currently emitting a lambda, emit an l-value for
4253e5dd7070Spatrick /// one of its members.
EmitLValueForLambdaField(const FieldDecl * Field)4254e5dd7070Spatrick LValue CodeGenFunction::EmitLValueForLambdaField(const FieldDecl *Field) {
4255a9ac8606Spatrick   if (CurCodeDecl) {
4256e5dd7070Spatrick     assert(cast<CXXMethodDecl>(CurCodeDecl)->getParent()->isLambda());
4257e5dd7070Spatrick     assert(cast<CXXMethodDecl>(CurCodeDecl)->getParent() == Field->getParent());
4258a9ac8606Spatrick   }
4259e5dd7070Spatrick   QualType LambdaTagType =
4260e5dd7070Spatrick     getContext().getTagDeclType(Field->getParent());
4261e5dd7070Spatrick   LValue LambdaLV = MakeNaturalAlignAddrLValue(CXXABIThisValue, LambdaTagType);
4262e5dd7070Spatrick   return EmitLValueForField(LambdaLV, Field);
4263e5dd7070Spatrick }
4264e5dd7070Spatrick 
4265e5dd7070Spatrick /// Get the field index in the debug info. The debug info structure/union
4266e5dd7070Spatrick /// will ignore the unnamed bitfields.
getDebugInfoFIndex(const RecordDecl * Rec,unsigned FieldIndex)4267e5dd7070Spatrick unsigned CodeGenFunction::getDebugInfoFIndex(const RecordDecl *Rec,
4268e5dd7070Spatrick                                              unsigned FieldIndex) {
4269e5dd7070Spatrick   unsigned I = 0, Skipped = 0;
4270e5dd7070Spatrick 
4271*12c85518Srobert   for (auto *F : Rec->getDefinition()->fields()) {
4272e5dd7070Spatrick     if (I == FieldIndex)
4273e5dd7070Spatrick       break;
4274e5dd7070Spatrick     if (F->isUnnamedBitfield())
4275e5dd7070Spatrick       Skipped++;
4276e5dd7070Spatrick     I++;
4277e5dd7070Spatrick   }
4278e5dd7070Spatrick 
4279e5dd7070Spatrick   return FieldIndex - Skipped;
4280e5dd7070Spatrick }
4281e5dd7070Spatrick 
4282e5dd7070Spatrick /// Get the address of a zero-sized field within a record. The resulting
4283e5dd7070Spatrick /// address doesn't necessarily have the right type.
emitAddrOfZeroSizeField(CodeGenFunction & CGF,Address Base,const FieldDecl * Field)4284e5dd7070Spatrick static Address emitAddrOfZeroSizeField(CodeGenFunction &CGF, Address Base,
4285e5dd7070Spatrick                                        const FieldDecl *Field) {
4286e5dd7070Spatrick   CharUnits Offset = CGF.getContext().toCharUnitsFromBits(
4287e5dd7070Spatrick       CGF.getContext().getFieldOffset(Field));
4288e5dd7070Spatrick   if (Offset.isZero())
4289e5dd7070Spatrick     return Base;
4290e5dd7070Spatrick   Base = CGF.Builder.CreateElementBitCast(Base, CGF.Int8Ty);
4291e5dd7070Spatrick   return CGF.Builder.CreateConstInBoundsByteGEP(Base, Offset);
4292e5dd7070Spatrick }
4293e5dd7070Spatrick 
4294e5dd7070Spatrick /// Drill down to the storage of a field without walking into
4295e5dd7070Spatrick /// reference types.
4296e5dd7070Spatrick ///
4297e5dd7070Spatrick /// The resulting address doesn't necessarily have the right type.
emitAddrOfFieldStorage(CodeGenFunction & CGF,Address base,const FieldDecl * field)4298e5dd7070Spatrick static Address emitAddrOfFieldStorage(CodeGenFunction &CGF, Address base,
4299e5dd7070Spatrick                                       const FieldDecl *field) {
4300e5dd7070Spatrick   if (field->isZeroSize(CGF.getContext()))
4301e5dd7070Spatrick     return emitAddrOfZeroSizeField(CGF, base, field);
4302e5dd7070Spatrick 
4303e5dd7070Spatrick   const RecordDecl *rec = field->getParent();
4304e5dd7070Spatrick 
4305e5dd7070Spatrick   unsigned idx =
4306e5dd7070Spatrick     CGF.CGM.getTypes().getCGRecordLayout(rec).getLLVMFieldNo(field);
4307e5dd7070Spatrick 
4308e5dd7070Spatrick   return CGF.Builder.CreateStructGEP(base, idx, field->getName());
4309e5dd7070Spatrick }
4310e5dd7070Spatrick 
emitPreserveStructAccess(CodeGenFunction & CGF,LValue base,Address addr,const FieldDecl * field)4311ec727ea7Spatrick static Address emitPreserveStructAccess(CodeGenFunction &CGF, LValue base,
4312ec727ea7Spatrick                                         Address addr, const FieldDecl *field) {
4313e5dd7070Spatrick   const RecordDecl *rec = field->getParent();
4314ec727ea7Spatrick   llvm::DIType *DbgInfo = CGF.getDebugInfo()->getOrCreateStandaloneType(
4315ec727ea7Spatrick       base.getType(), rec->getLocation());
4316e5dd7070Spatrick 
4317e5dd7070Spatrick   unsigned idx =
4318e5dd7070Spatrick       CGF.CGM.getTypes().getCGRecordLayout(rec).getLLVMFieldNo(field);
4319e5dd7070Spatrick 
4320e5dd7070Spatrick   return CGF.Builder.CreatePreserveStructAccessIndex(
4321ec727ea7Spatrick       addr, idx, CGF.getDebugInfoFIndex(rec, field->getFieldIndex()), DbgInfo);
4322e5dd7070Spatrick }
4323e5dd7070Spatrick 
hasAnyVptr(const QualType Type,const ASTContext & Context)4324e5dd7070Spatrick static bool hasAnyVptr(const QualType Type, const ASTContext &Context) {
4325e5dd7070Spatrick   const auto *RD = Type.getTypePtr()->getAsCXXRecordDecl();
4326e5dd7070Spatrick   if (!RD)
4327e5dd7070Spatrick     return false;
4328e5dd7070Spatrick 
4329e5dd7070Spatrick   if (RD->isDynamicClass())
4330e5dd7070Spatrick     return true;
4331e5dd7070Spatrick 
4332e5dd7070Spatrick   for (const auto &Base : RD->bases())
4333e5dd7070Spatrick     if (hasAnyVptr(Base.getType(), Context))
4334e5dd7070Spatrick       return true;
4335e5dd7070Spatrick 
4336e5dd7070Spatrick   for (const FieldDecl *Field : RD->fields())
4337e5dd7070Spatrick     if (hasAnyVptr(Field->getType(), Context))
4338e5dd7070Spatrick       return true;
4339e5dd7070Spatrick 
4340e5dd7070Spatrick   return false;
4341e5dd7070Spatrick }
4342e5dd7070Spatrick 
EmitLValueForField(LValue base,const FieldDecl * field)4343e5dd7070Spatrick LValue CodeGenFunction::EmitLValueForField(LValue base,
4344e5dd7070Spatrick                                            const FieldDecl *field) {
4345e5dd7070Spatrick   LValueBaseInfo BaseInfo = base.getBaseInfo();
4346e5dd7070Spatrick 
4347e5dd7070Spatrick   if (field->isBitField()) {
4348e5dd7070Spatrick     const CGRecordLayout &RL =
4349e5dd7070Spatrick         CGM.getTypes().getCGRecordLayout(field->getParent());
4350e5dd7070Spatrick     const CGBitFieldInfo &Info = RL.getBitFieldInfo(field);
4351a9ac8606Spatrick     const bool UseVolatile = isAAPCS(CGM.getTarget()) &&
4352a9ac8606Spatrick                              CGM.getCodeGenOpts().AAPCSBitfieldWidth &&
4353a9ac8606Spatrick                              Info.VolatileStorageSize != 0 &&
4354a9ac8606Spatrick                              field->getType()
4355a9ac8606Spatrick                                  .withCVRQualifiers(base.getVRQualifiers())
4356a9ac8606Spatrick                                  .isVolatileQualified();
4357e5dd7070Spatrick     Address Addr = base.getAddress(*this);
4358e5dd7070Spatrick     unsigned Idx = RL.getLLVMFieldNo(field);
4359e5dd7070Spatrick     const RecordDecl *rec = field->getParent();
4360a9ac8606Spatrick     if (!UseVolatile) {
4361e5dd7070Spatrick       if (!IsInPreservedAIRegion &&
4362e5dd7070Spatrick           (!getDebugInfo() || !rec->hasAttr<BPFPreserveAccessIndexAttr>())) {
4363e5dd7070Spatrick         if (Idx != 0)
4364e5dd7070Spatrick           // For structs, we GEP to the field that the record layout suggests.
4365e5dd7070Spatrick           Addr = Builder.CreateStructGEP(Addr, Idx, field->getName());
4366e5dd7070Spatrick       } else {
4367e5dd7070Spatrick         llvm::DIType *DbgInfo = getDebugInfo()->getOrCreateRecordType(
4368e5dd7070Spatrick             getContext().getRecordType(rec), rec->getLocation());
4369a9ac8606Spatrick         Addr = Builder.CreatePreserveStructAccessIndex(
4370a9ac8606Spatrick             Addr, Idx, getDebugInfoFIndex(rec, field->getFieldIndex()),
4371e5dd7070Spatrick             DbgInfo);
4372e5dd7070Spatrick       }
4373a9ac8606Spatrick     }
4374a9ac8606Spatrick     const unsigned SS =
4375a9ac8606Spatrick         UseVolatile ? Info.VolatileStorageSize : Info.StorageSize;
4376e5dd7070Spatrick     // Get the access type.
4377a9ac8606Spatrick     llvm::Type *FieldIntTy = llvm::Type::getIntNTy(getLLVMContext(), SS);
4378e5dd7070Spatrick     if (Addr.getElementType() != FieldIntTy)
4379e5dd7070Spatrick       Addr = Builder.CreateElementBitCast(Addr, FieldIntTy);
4380a9ac8606Spatrick     if (UseVolatile) {
4381a9ac8606Spatrick       const unsigned VolatileOffset = Info.VolatileStorageOffset.getQuantity();
4382a9ac8606Spatrick       if (VolatileOffset)
4383a9ac8606Spatrick         Addr = Builder.CreateConstInBoundsGEP(Addr, VolatileOffset);
4384a9ac8606Spatrick     }
4385e5dd7070Spatrick 
4386e5dd7070Spatrick     QualType fieldType =
4387e5dd7070Spatrick         field->getType().withCVRQualifiers(base.getVRQualifiers());
4388e5dd7070Spatrick     // TODO: Support TBAA for bit fields.
4389e5dd7070Spatrick     LValueBaseInfo FieldBaseInfo(BaseInfo.getAlignmentSource());
4390e5dd7070Spatrick     return LValue::MakeBitfield(Addr, Info, fieldType, FieldBaseInfo,
4391e5dd7070Spatrick                                 TBAAAccessInfo());
4392e5dd7070Spatrick   }
4393e5dd7070Spatrick 
4394e5dd7070Spatrick   // Fields of may-alias structures are may-alias themselves.
4395e5dd7070Spatrick   // FIXME: this should get propagated down through anonymous structs
4396e5dd7070Spatrick   // and unions.
4397e5dd7070Spatrick   QualType FieldType = field->getType();
4398e5dd7070Spatrick   const RecordDecl *rec = field->getParent();
4399e5dd7070Spatrick   AlignmentSource BaseAlignSource = BaseInfo.getAlignmentSource();
4400e5dd7070Spatrick   LValueBaseInfo FieldBaseInfo(getFieldAlignmentSource(BaseAlignSource));
4401e5dd7070Spatrick   TBAAAccessInfo FieldTBAAInfo;
4402e5dd7070Spatrick   if (base.getTBAAInfo().isMayAlias() ||
4403e5dd7070Spatrick           rec->hasAttr<MayAliasAttr>() || FieldType->isVectorType()) {
4404e5dd7070Spatrick     FieldTBAAInfo = TBAAAccessInfo::getMayAliasInfo();
4405e5dd7070Spatrick   } else if (rec->isUnion()) {
4406e5dd7070Spatrick     // TODO: Support TBAA for unions.
4407e5dd7070Spatrick     FieldTBAAInfo = TBAAAccessInfo::getMayAliasInfo();
4408e5dd7070Spatrick   } else {
4409e5dd7070Spatrick     // If no base type been assigned for the base access, then try to generate
4410e5dd7070Spatrick     // one for this base lvalue.
4411e5dd7070Spatrick     FieldTBAAInfo = base.getTBAAInfo();
4412e5dd7070Spatrick     if (!FieldTBAAInfo.BaseType) {
4413e5dd7070Spatrick         FieldTBAAInfo.BaseType = CGM.getTBAABaseTypeInfo(base.getType());
4414e5dd7070Spatrick         assert(!FieldTBAAInfo.Offset &&
4415e5dd7070Spatrick                "Nonzero offset for an access with no base type!");
4416e5dd7070Spatrick     }
4417e5dd7070Spatrick 
4418e5dd7070Spatrick     // Adjust offset to be relative to the base type.
4419e5dd7070Spatrick     const ASTRecordLayout &Layout =
4420e5dd7070Spatrick         getContext().getASTRecordLayout(field->getParent());
4421e5dd7070Spatrick     unsigned CharWidth = getContext().getCharWidth();
4422e5dd7070Spatrick     if (FieldTBAAInfo.BaseType)
4423e5dd7070Spatrick       FieldTBAAInfo.Offset +=
4424e5dd7070Spatrick           Layout.getFieldOffset(field->getFieldIndex()) / CharWidth;
4425e5dd7070Spatrick 
4426e5dd7070Spatrick     // Update the final access type and size.
4427e5dd7070Spatrick     FieldTBAAInfo.AccessType = CGM.getTBAATypeInfo(FieldType);
4428e5dd7070Spatrick     FieldTBAAInfo.Size =
4429e5dd7070Spatrick         getContext().getTypeSizeInChars(FieldType).getQuantity();
4430e5dd7070Spatrick   }
4431e5dd7070Spatrick 
4432e5dd7070Spatrick   Address addr = base.getAddress(*this);
4433e5dd7070Spatrick   if (auto *ClassDef = dyn_cast<CXXRecordDecl>(rec)) {
4434e5dd7070Spatrick     if (CGM.getCodeGenOpts().StrictVTablePointers &&
4435e5dd7070Spatrick         ClassDef->isDynamicClass()) {
4436e5dd7070Spatrick       // Getting to any field of dynamic object requires stripping dynamic
4437e5dd7070Spatrick       // information provided by invariant.group.  This is because accessing
4438e5dd7070Spatrick       // fields may leak the real address of dynamic object, which could result
4439e5dd7070Spatrick       // in miscompilation when leaked pointer would be compared.
4440e5dd7070Spatrick       auto *stripped = Builder.CreateStripInvariantGroup(addr.getPointer());
4441*12c85518Srobert       addr = Address(stripped, addr.getElementType(), addr.getAlignment());
4442e5dd7070Spatrick     }
4443e5dd7070Spatrick   }
4444e5dd7070Spatrick 
4445e5dd7070Spatrick   unsigned RecordCVR = base.getVRQualifiers();
4446e5dd7070Spatrick   if (rec->isUnion()) {
4447e5dd7070Spatrick     // For unions, there is no pointer adjustment.
4448e5dd7070Spatrick     if (CGM.getCodeGenOpts().StrictVTablePointers &&
4449e5dd7070Spatrick         hasAnyVptr(FieldType, getContext()))
4450e5dd7070Spatrick       // Because unions can easily skip invariant.barriers, we need to add
4451e5dd7070Spatrick       // a barrier every time CXXRecord field with vptr is referenced.
4452*12c85518Srobert       addr = Builder.CreateLaunderInvariantGroup(addr);
4453e5dd7070Spatrick 
4454e5dd7070Spatrick     if (IsInPreservedAIRegion ||
4455e5dd7070Spatrick         (getDebugInfo() && rec->hasAttr<BPFPreserveAccessIndexAttr>())) {
4456e5dd7070Spatrick       // Remember the original union field index
4457ec727ea7Spatrick       llvm::DIType *DbgInfo = getDebugInfo()->getOrCreateStandaloneType(base.getType(),
4458ec727ea7Spatrick           rec->getLocation());
4459e5dd7070Spatrick       addr = Address(
4460e5dd7070Spatrick           Builder.CreatePreserveUnionAccessIndex(
4461e5dd7070Spatrick               addr.getPointer(), getDebugInfoFIndex(rec, field->getFieldIndex()), DbgInfo),
4462*12c85518Srobert           addr.getElementType(), addr.getAlignment());
4463e5dd7070Spatrick     }
4464e5dd7070Spatrick 
4465e5dd7070Spatrick     if (FieldType->isReferenceType())
4466e5dd7070Spatrick       addr = Builder.CreateElementBitCast(
4467e5dd7070Spatrick           addr, CGM.getTypes().ConvertTypeForMem(FieldType), field->getName());
4468e5dd7070Spatrick   } else {
4469e5dd7070Spatrick     if (!IsInPreservedAIRegion &&
4470e5dd7070Spatrick         (!getDebugInfo() || !rec->hasAttr<BPFPreserveAccessIndexAttr>()))
4471e5dd7070Spatrick       // For structs, we GEP to the field that the record layout suggests.
4472e5dd7070Spatrick       addr = emitAddrOfFieldStorage(*this, addr, field);
4473e5dd7070Spatrick     else
4474e5dd7070Spatrick       // Remember the original struct field index
4475ec727ea7Spatrick       addr = emitPreserveStructAccess(*this, base, addr, field);
4476e5dd7070Spatrick   }
4477e5dd7070Spatrick 
4478e5dd7070Spatrick   // If this is a reference field, load the reference right now.
4479e5dd7070Spatrick   if (FieldType->isReferenceType()) {
4480e5dd7070Spatrick     LValue RefLVal =
4481e5dd7070Spatrick         MakeAddrLValue(addr, FieldType, FieldBaseInfo, FieldTBAAInfo);
4482e5dd7070Spatrick     if (RecordCVR & Qualifiers::Volatile)
4483e5dd7070Spatrick       RefLVal.getQuals().addVolatile();
4484e5dd7070Spatrick     addr = EmitLoadOfReference(RefLVal, &FieldBaseInfo, &FieldTBAAInfo);
4485e5dd7070Spatrick 
4486e5dd7070Spatrick     // Qualifiers on the struct don't apply to the referencee.
4487e5dd7070Spatrick     RecordCVR = 0;
4488e5dd7070Spatrick     FieldType = FieldType->getPointeeType();
4489e5dd7070Spatrick   }
4490e5dd7070Spatrick 
4491e5dd7070Spatrick   // Make sure that the address is pointing to the right type.  This is critical
4492e5dd7070Spatrick   // for both unions and structs.  A union needs a bitcast, a struct element
4493e5dd7070Spatrick   // will need a bitcast if the LLVM type laid out doesn't match the desired
4494e5dd7070Spatrick   // type.
4495e5dd7070Spatrick   addr = Builder.CreateElementBitCast(
4496e5dd7070Spatrick       addr, CGM.getTypes().ConvertTypeForMem(FieldType), field->getName());
4497e5dd7070Spatrick 
4498e5dd7070Spatrick   if (field->hasAttr<AnnotateAttr>())
4499e5dd7070Spatrick     addr = EmitFieldAnnotations(field, addr);
4500e5dd7070Spatrick 
4501e5dd7070Spatrick   LValue LV = MakeAddrLValue(addr, FieldType, FieldBaseInfo, FieldTBAAInfo);
4502e5dd7070Spatrick   LV.getQuals().addCVRQualifiers(RecordCVR);
4503e5dd7070Spatrick 
4504e5dd7070Spatrick   // __weak attribute on a field is ignored.
4505e5dd7070Spatrick   if (LV.getQuals().getObjCGCAttr() == Qualifiers::Weak)
4506e5dd7070Spatrick     LV.getQuals().removeObjCGCAttr();
4507e5dd7070Spatrick 
4508e5dd7070Spatrick   return LV;
4509e5dd7070Spatrick }
4510e5dd7070Spatrick 
4511e5dd7070Spatrick LValue
EmitLValueForFieldInitialization(LValue Base,const FieldDecl * Field)4512e5dd7070Spatrick CodeGenFunction::EmitLValueForFieldInitialization(LValue Base,
4513e5dd7070Spatrick                                                   const FieldDecl *Field) {
4514e5dd7070Spatrick   QualType FieldType = Field->getType();
4515e5dd7070Spatrick 
4516e5dd7070Spatrick   if (!FieldType->isReferenceType())
4517e5dd7070Spatrick     return EmitLValueForField(Base, Field);
4518e5dd7070Spatrick 
4519e5dd7070Spatrick   Address V = emitAddrOfFieldStorage(*this, Base.getAddress(*this), Field);
4520e5dd7070Spatrick 
4521e5dd7070Spatrick   // Make sure that the address is pointing to the right type.
4522e5dd7070Spatrick   llvm::Type *llvmType = ConvertTypeForMem(FieldType);
4523e5dd7070Spatrick   V = Builder.CreateElementBitCast(V, llvmType, Field->getName());
4524e5dd7070Spatrick 
4525e5dd7070Spatrick   // TODO: Generate TBAA information that describes this access as a structure
4526e5dd7070Spatrick   // member access and not just an access to an object of the field's type. This
4527e5dd7070Spatrick   // should be similar to what we do in EmitLValueForField().
4528e5dd7070Spatrick   LValueBaseInfo BaseInfo = Base.getBaseInfo();
4529e5dd7070Spatrick   AlignmentSource FieldAlignSource = BaseInfo.getAlignmentSource();
4530e5dd7070Spatrick   LValueBaseInfo FieldBaseInfo(getFieldAlignmentSource(FieldAlignSource));
4531e5dd7070Spatrick   return MakeAddrLValue(V, FieldType, FieldBaseInfo,
4532e5dd7070Spatrick                         CGM.getTBAAInfoForSubobject(Base, FieldType));
4533e5dd7070Spatrick }
4534e5dd7070Spatrick 
EmitCompoundLiteralLValue(const CompoundLiteralExpr * E)4535e5dd7070Spatrick LValue CodeGenFunction::EmitCompoundLiteralLValue(const CompoundLiteralExpr *E){
4536e5dd7070Spatrick   if (E->isFileScope()) {
4537e5dd7070Spatrick     ConstantAddress GlobalPtr = CGM.GetAddrOfConstantCompoundLiteral(E);
4538e5dd7070Spatrick     return MakeAddrLValue(GlobalPtr, E->getType(), AlignmentSource::Decl);
4539e5dd7070Spatrick   }
4540e5dd7070Spatrick   if (E->getType()->isVariablyModifiedType())
4541e5dd7070Spatrick     // make sure to emit the VLA size.
4542e5dd7070Spatrick     EmitVariablyModifiedType(E->getType());
4543e5dd7070Spatrick 
4544e5dd7070Spatrick   Address DeclPtr = CreateMemTemp(E->getType(), ".compoundliteral");
4545e5dd7070Spatrick   const Expr *InitExpr = E->getInitializer();
4546e5dd7070Spatrick   LValue Result = MakeAddrLValue(DeclPtr, E->getType(), AlignmentSource::Decl);
4547e5dd7070Spatrick 
4548e5dd7070Spatrick   EmitAnyExprToMem(InitExpr, DeclPtr, E->getType().getQualifiers(),
4549e5dd7070Spatrick                    /*Init*/ true);
4550e5dd7070Spatrick 
4551ec727ea7Spatrick   // Block-scope compound literals are destroyed at the end of the enclosing
4552ec727ea7Spatrick   // scope in C.
4553ec727ea7Spatrick   if (!getLangOpts().CPlusPlus)
4554ec727ea7Spatrick     if (QualType::DestructionKind DtorKind = E->getType().isDestructedType())
4555ec727ea7Spatrick       pushLifetimeExtendedDestroy(getCleanupKind(DtorKind), DeclPtr,
4556ec727ea7Spatrick                                   E->getType(), getDestroyer(DtorKind),
4557ec727ea7Spatrick                                   DtorKind & EHCleanup);
4558ec727ea7Spatrick 
4559e5dd7070Spatrick   return Result;
4560e5dd7070Spatrick }
4561e5dd7070Spatrick 
EmitInitListLValue(const InitListExpr * E)4562e5dd7070Spatrick LValue CodeGenFunction::EmitInitListLValue(const InitListExpr *E) {
4563e5dd7070Spatrick   if (!E->isGLValue())
4564e5dd7070Spatrick     // Initializing an aggregate temporary in C++11: T{...}.
4565e5dd7070Spatrick     return EmitAggExprToLValue(E);
4566e5dd7070Spatrick 
4567e5dd7070Spatrick   // An lvalue initializer list must be initializing a reference.
4568e5dd7070Spatrick   assert(E->isTransparent() && "non-transparent glvalue init list");
4569e5dd7070Spatrick   return EmitLValue(E->getInit(0));
4570e5dd7070Spatrick }
4571e5dd7070Spatrick 
4572e5dd7070Spatrick /// Emit the operand of a glvalue conditional operator. This is either a glvalue
4573e5dd7070Spatrick /// or a (possibly-parenthesized) throw-expression. If this is a throw, no
4574e5dd7070Spatrick /// LValue is returned and the current block has been terminated.
EmitLValueOrThrowExpression(CodeGenFunction & CGF,const Expr * Operand)4575*12c85518Srobert static std::optional<LValue> EmitLValueOrThrowExpression(CodeGenFunction &CGF,
4576e5dd7070Spatrick                                                          const Expr *Operand) {
4577e5dd7070Spatrick   if (auto *ThrowExpr = dyn_cast<CXXThrowExpr>(Operand->IgnoreParens())) {
4578e5dd7070Spatrick     CGF.EmitCXXThrowExpr(ThrowExpr, /*KeepInsertionPoint*/false);
4579*12c85518Srobert     return std::nullopt;
4580e5dd7070Spatrick   }
4581e5dd7070Spatrick 
4582e5dd7070Spatrick   return CGF.EmitLValue(Operand);
4583e5dd7070Spatrick }
4584e5dd7070Spatrick 
4585*12c85518Srobert namespace {
4586*12c85518Srobert // Handle the case where the condition is a constant evaluatable simple integer,
4587*12c85518Srobert // which means we don't have to separately handle the true/false blocks.
HandleConditionalOperatorLValueSimpleCase(CodeGenFunction & CGF,const AbstractConditionalOperator * E)4588*12c85518Srobert std::optional<LValue> HandleConditionalOperatorLValueSimpleCase(
4589*12c85518Srobert     CodeGenFunction &CGF, const AbstractConditionalOperator *E) {
4590*12c85518Srobert   const Expr *condExpr = E->getCond();
4591*12c85518Srobert   bool CondExprBool;
4592*12c85518Srobert   if (CGF.ConstantFoldsToSimpleInteger(condExpr, CondExprBool)) {
4593*12c85518Srobert     const Expr *Live = E->getTrueExpr(), *Dead = E->getFalseExpr();
4594*12c85518Srobert     if (!CondExprBool)
4595*12c85518Srobert       std::swap(Live, Dead);
4596*12c85518Srobert 
4597*12c85518Srobert     if (!CGF.ContainsLabel(Dead)) {
4598*12c85518Srobert       // If the true case is live, we need to track its region.
4599*12c85518Srobert       if (CondExprBool)
4600*12c85518Srobert         CGF.incrementProfileCounter(E);
4601*12c85518Srobert       // If a throw expression we emit it and return an undefined lvalue
4602*12c85518Srobert       // because it can't be used.
4603*12c85518Srobert       if (auto *ThrowExpr = dyn_cast<CXXThrowExpr>(Live->IgnoreParens())) {
4604*12c85518Srobert         CGF.EmitCXXThrowExpr(ThrowExpr);
4605*12c85518Srobert         llvm::Type *ElemTy = CGF.ConvertType(Dead->getType());
4606*12c85518Srobert         llvm::Type *Ty = llvm::PointerType::getUnqual(ElemTy);
4607*12c85518Srobert         return CGF.MakeAddrLValue(
4608*12c85518Srobert             Address(llvm::UndefValue::get(Ty), ElemTy, CharUnits::One()),
4609*12c85518Srobert             Dead->getType());
4610*12c85518Srobert       }
4611*12c85518Srobert       return CGF.EmitLValue(Live);
4612*12c85518Srobert     }
4613*12c85518Srobert   }
4614*12c85518Srobert   return std::nullopt;
4615*12c85518Srobert }
4616*12c85518Srobert struct ConditionalInfo {
4617*12c85518Srobert   llvm::BasicBlock *lhsBlock, *rhsBlock;
4618*12c85518Srobert   std::optional<LValue> LHS, RHS;
4619*12c85518Srobert };
4620*12c85518Srobert 
4621*12c85518Srobert // Create and generate the 3 blocks for a conditional operator.
4622*12c85518Srobert // Leaves the 'current block' in the continuation basic block.
4623*12c85518Srobert template<typename FuncTy>
EmitConditionalBlocks(CodeGenFunction & CGF,const AbstractConditionalOperator * E,const FuncTy & BranchGenFunc)4624*12c85518Srobert ConditionalInfo EmitConditionalBlocks(CodeGenFunction &CGF,
4625*12c85518Srobert                                       const AbstractConditionalOperator *E,
4626*12c85518Srobert                                       const FuncTy &BranchGenFunc) {
4627*12c85518Srobert   ConditionalInfo Info{CGF.createBasicBlock("cond.true"),
4628*12c85518Srobert                        CGF.createBasicBlock("cond.false"), std::nullopt,
4629*12c85518Srobert                        std::nullopt};
4630*12c85518Srobert   llvm::BasicBlock *endBlock = CGF.createBasicBlock("cond.end");
4631*12c85518Srobert 
4632*12c85518Srobert   CodeGenFunction::ConditionalEvaluation eval(CGF);
4633*12c85518Srobert   CGF.EmitBranchOnBoolExpr(E->getCond(), Info.lhsBlock, Info.rhsBlock,
4634*12c85518Srobert                            CGF.getProfileCount(E));
4635*12c85518Srobert 
4636*12c85518Srobert   // Any temporaries created here are conditional.
4637*12c85518Srobert   CGF.EmitBlock(Info.lhsBlock);
4638*12c85518Srobert   CGF.incrementProfileCounter(E);
4639*12c85518Srobert   eval.begin(CGF);
4640*12c85518Srobert   Info.LHS = BranchGenFunc(CGF, E->getTrueExpr());
4641*12c85518Srobert   eval.end(CGF);
4642*12c85518Srobert   Info.lhsBlock = CGF.Builder.GetInsertBlock();
4643*12c85518Srobert 
4644*12c85518Srobert   if (Info.LHS)
4645*12c85518Srobert     CGF.Builder.CreateBr(endBlock);
4646*12c85518Srobert 
4647*12c85518Srobert   // Any temporaries created here are conditional.
4648*12c85518Srobert   CGF.EmitBlock(Info.rhsBlock);
4649*12c85518Srobert   eval.begin(CGF);
4650*12c85518Srobert   Info.RHS = BranchGenFunc(CGF, E->getFalseExpr());
4651*12c85518Srobert   eval.end(CGF);
4652*12c85518Srobert   Info.rhsBlock = CGF.Builder.GetInsertBlock();
4653*12c85518Srobert   CGF.EmitBlock(endBlock);
4654*12c85518Srobert 
4655*12c85518Srobert   return Info;
4656*12c85518Srobert }
4657*12c85518Srobert } // namespace
4658*12c85518Srobert 
EmitIgnoredConditionalOperator(const AbstractConditionalOperator * E)4659*12c85518Srobert void CodeGenFunction::EmitIgnoredConditionalOperator(
4660*12c85518Srobert     const AbstractConditionalOperator *E) {
4661*12c85518Srobert   if (!E->isGLValue()) {
4662*12c85518Srobert     // ?: here should be an aggregate.
4663*12c85518Srobert     assert(hasAggregateEvaluationKind(E->getType()) &&
4664*12c85518Srobert            "Unexpected conditional operator!");
4665*12c85518Srobert     return (void)EmitAggExprToLValue(E);
4666*12c85518Srobert   }
4667*12c85518Srobert 
4668*12c85518Srobert   OpaqueValueMapping binding(*this, E);
4669*12c85518Srobert   if (HandleConditionalOperatorLValueSimpleCase(*this, E))
4670*12c85518Srobert     return;
4671*12c85518Srobert 
4672*12c85518Srobert   EmitConditionalBlocks(*this, E, [](CodeGenFunction &CGF, const Expr *E) {
4673*12c85518Srobert     CGF.EmitIgnoredExpr(E);
4674*12c85518Srobert     return LValue{};
4675*12c85518Srobert   });
4676*12c85518Srobert }
EmitConditionalOperatorLValue(const AbstractConditionalOperator * expr)4677*12c85518Srobert LValue CodeGenFunction::EmitConditionalOperatorLValue(
4678*12c85518Srobert     const AbstractConditionalOperator *expr) {
4679e5dd7070Spatrick   if (!expr->isGLValue()) {
4680e5dd7070Spatrick     // ?: here should be an aggregate.
4681e5dd7070Spatrick     assert(hasAggregateEvaluationKind(expr->getType()) &&
4682e5dd7070Spatrick            "Unexpected conditional operator!");
4683e5dd7070Spatrick     return EmitAggExprToLValue(expr);
4684e5dd7070Spatrick   }
4685e5dd7070Spatrick 
4686e5dd7070Spatrick   OpaqueValueMapping binding(*this, expr);
4687*12c85518Srobert   if (std::optional<LValue> Res =
4688*12c85518Srobert           HandleConditionalOperatorLValueSimpleCase(*this, expr))
4689*12c85518Srobert     return *Res;
4690e5dd7070Spatrick 
4691*12c85518Srobert   ConditionalInfo Info = EmitConditionalBlocks(
4692*12c85518Srobert       *this, expr, [](CodeGenFunction &CGF, const Expr *E) {
4693*12c85518Srobert         return EmitLValueOrThrowExpression(CGF, E);
4694*12c85518Srobert       });
4695e5dd7070Spatrick 
4696*12c85518Srobert   if ((Info.LHS && !Info.LHS->isSimple()) ||
4697*12c85518Srobert       (Info.RHS && !Info.RHS->isSimple()))
4698e5dd7070Spatrick     return EmitUnsupportedLValue(expr, "conditional operator");
4699e5dd7070Spatrick 
4700*12c85518Srobert   if (Info.LHS && Info.RHS) {
4701*12c85518Srobert     Address lhsAddr = Info.LHS->getAddress(*this);
4702*12c85518Srobert     Address rhsAddr = Info.RHS->getAddress(*this);
4703*12c85518Srobert     llvm::PHINode *phi = Builder.CreatePHI(lhsAddr.getType(), 2, "cond-lvalue");
4704*12c85518Srobert     phi->addIncoming(lhsAddr.getPointer(), Info.lhsBlock);
4705*12c85518Srobert     phi->addIncoming(rhsAddr.getPointer(), Info.rhsBlock);
4706*12c85518Srobert     Address result(phi, lhsAddr.getElementType(),
4707*12c85518Srobert                    std::min(lhsAddr.getAlignment(), rhsAddr.getAlignment()));
4708e5dd7070Spatrick     AlignmentSource alignSource =
4709*12c85518Srobert         std::max(Info.LHS->getBaseInfo().getAlignmentSource(),
4710*12c85518Srobert                  Info.RHS->getBaseInfo().getAlignmentSource());
4711e5dd7070Spatrick     TBAAAccessInfo TBAAInfo = CGM.mergeTBAAInfoForConditionalOperator(
4712*12c85518Srobert         Info.LHS->getTBAAInfo(), Info.RHS->getTBAAInfo());
4713e5dd7070Spatrick     return MakeAddrLValue(result, expr->getType(), LValueBaseInfo(alignSource),
4714e5dd7070Spatrick                           TBAAInfo);
4715e5dd7070Spatrick   } else {
4716*12c85518Srobert     assert((Info.LHS || Info.RHS) &&
4717e5dd7070Spatrick            "both operands of glvalue conditional are throw-expressions?");
4718*12c85518Srobert     return Info.LHS ? *Info.LHS : *Info.RHS;
4719e5dd7070Spatrick   }
4720e5dd7070Spatrick }
4721e5dd7070Spatrick 
4722e5dd7070Spatrick /// EmitCastLValue - Casts are never lvalues unless that cast is to a reference
4723e5dd7070Spatrick /// type. If the cast is to a reference, we can have the usual lvalue result,
4724e5dd7070Spatrick /// otherwise if a cast is needed by the code generator in an lvalue context,
4725e5dd7070Spatrick /// then it must mean that we need the address of an aggregate in order to
4726e5dd7070Spatrick /// access one of its members.  This can happen for all the reasons that casts
4727e5dd7070Spatrick /// are permitted with aggregate result, including noop aggregate casts, and
4728e5dd7070Spatrick /// cast from scalar to union.
EmitCastLValue(const CastExpr * E)4729e5dd7070Spatrick LValue CodeGenFunction::EmitCastLValue(const CastExpr *E) {
4730e5dd7070Spatrick   switch (E->getCastKind()) {
4731e5dd7070Spatrick   case CK_ToVoid:
4732e5dd7070Spatrick   case CK_BitCast:
4733e5dd7070Spatrick   case CK_LValueToRValueBitCast:
4734e5dd7070Spatrick   case CK_ArrayToPointerDecay:
4735e5dd7070Spatrick   case CK_FunctionToPointerDecay:
4736e5dd7070Spatrick   case CK_NullToMemberPointer:
4737e5dd7070Spatrick   case CK_NullToPointer:
4738e5dd7070Spatrick   case CK_IntegralToPointer:
4739e5dd7070Spatrick   case CK_PointerToIntegral:
4740e5dd7070Spatrick   case CK_PointerToBoolean:
4741e5dd7070Spatrick   case CK_VectorSplat:
4742e5dd7070Spatrick   case CK_IntegralCast:
4743e5dd7070Spatrick   case CK_BooleanToSignedIntegral:
4744e5dd7070Spatrick   case CK_IntegralToBoolean:
4745e5dd7070Spatrick   case CK_IntegralToFloating:
4746e5dd7070Spatrick   case CK_FloatingToIntegral:
4747e5dd7070Spatrick   case CK_FloatingToBoolean:
4748e5dd7070Spatrick   case CK_FloatingCast:
4749e5dd7070Spatrick   case CK_FloatingRealToComplex:
4750e5dd7070Spatrick   case CK_FloatingComplexToReal:
4751e5dd7070Spatrick   case CK_FloatingComplexToBoolean:
4752e5dd7070Spatrick   case CK_FloatingComplexCast:
4753e5dd7070Spatrick   case CK_FloatingComplexToIntegralComplex:
4754e5dd7070Spatrick   case CK_IntegralRealToComplex:
4755e5dd7070Spatrick   case CK_IntegralComplexToReal:
4756e5dd7070Spatrick   case CK_IntegralComplexToBoolean:
4757e5dd7070Spatrick   case CK_IntegralComplexCast:
4758e5dd7070Spatrick   case CK_IntegralComplexToFloatingComplex:
4759e5dd7070Spatrick   case CK_DerivedToBaseMemberPointer:
4760e5dd7070Spatrick   case CK_BaseToDerivedMemberPointer:
4761e5dd7070Spatrick   case CK_MemberPointerToBoolean:
4762e5dd7070Spatrick   case CK_ReinterpretMemberPointer:
4763e5dd7070Spatrick   case CK_AnyPointerToBlockPointerCast:
4764e5dd7070Spatrick   case CK_ARCProduceObject:
4765e5dd7070Spatrick   case CK_ARCConsumeObject:
4766e5dd7070Spatrick   case CK_ARCReclaimReturnedObject:
4767e5dd7070Spatrick   case CK_ARCExtendBlockObject:
4768e5dd7070Spatrick   case CK_CopyAndAutoreleaseBlockObject:
4769e5dd7070Spatrick   case CK_IntToOCLSampler:
4770a9ac8606Spatrick   case CK_FloatingToFixedPoint:
4771a9ac8606Spatrick   case CK_FixedPointToFloating:
4772e5dd7070Spatrick   case CK_FixedPointCast:
4773e5dd7070Spatrick   case CK_FixedPointToBoolean:
4774e5dd7070Spatrick   case CK_FixedPointToIntegral:
4775e5dd7070Spatrick   case CK_IntegralToFixedPoint:
4776a9ac8606Spatrick   case CK_MatrixCast:
4777e5dd7070Spatrick     return EmitUnsupportedLValue(E, "unexpected cast lvalue");
4778e5dd7070Spatrick 
4779e5dd7070Spatrick   case CK_Dependent:
4780e5dd7070Spatrick     llvm_unreachable("dependent cast kind in IR gen!");
4781e5dd7070Spatrick 
4782e5dd7070Spatrick   case CK_BuiltinFnToFnPtr:
4783e5dd7070Spatrick     llvm_unreachable("builtin functions are handled elsewhere");
4784e5dd7070Spatrick 
4785e5dd7070Spatrick   // These are never l-values; just use the aggregate emission code.
4786e5dd7070Spatrick   case CK_NonAtomicToAtomic:
4787e5dd7070Spatrick   case CK_AtomicToNonAtomic:
4788e5dd7070Spatrick     return EmitAggExprToLValue(E);
4789e5dd7070Spatrick 
4790e5dd7070Spatrick   case CK_Dynamic: {
4791e5dd7070Spatrick     LValue LV = EmitLValue(E->getSubExpr());
4792e5dd7070Spatrick     Address V = LV.getAddress(*this);
4793e5dd7070Spatrick     const auto *DCE = cast<CXXDynamicCastExpr>(E);
4794e5dd7070Spatrick     return MakeNaturalAlignAddrLValue(EmitDynamicCast(V, DCE), E->getType());
4795e5dd7070Spatrick   }
4796e5dd7070Spatrick 
4797e5dd7070Spatrick   case CK_ConstructorConversion:
4798e5dd7070Spatrick   case CK_UserDefinedConversion:
4799e5dd7070Spatrick   case CK_CPointerToObjCPointerCast:
4800e5dd7070Spatrick   case CK_BlockPointerToObjCPointerCast:
4801e5dd7070Spatrick   case CK_LValueToRValue:
4802e5dd7070Spatrick     return EmitLValue(E->getSubExpr());
4803e5dd7070Spatrick 
4804*12c85518Srobert   case CK_NoOp: {
4805*12c85518Srobert     // CK_NoOp can model a qualification conversion, which can remove an array
4806*12c85518Srobert     // bound and change the IR type.
4807*12c85518Srobert     // FIXME: Once pointee types are removed from IR, remove this.
4808*12c85518Srobert     LValue LV = EmitLValue(E->getSubExpr());
4809*12c85518Srobert     if (LV.isSimple()) {
4810*12c85518Srobert       Address V = LV.getAddress(*this);
4811*12c85518Srobert       if (V.isValid()) {
4812*12c85518Srobert         llvm::Type *T = ConvertTypeForMem(E->getType());
4813*12c85518Srobert         if (V.getElementType() != T)
4814*12c85518Srobert           LV.setAddress(Builder.CreateElementBitCast(V, T));
4815*12c85518Srobert       }
4816*12c85518Srobert     }
4817*12c85518Srobert     return LV;
4818*12c85518Srobert   }
4819*12c85518Srobert 
4820e5dd7070Spatrick   case CK_UncheckedDerivedToBase:
4821e5dd7070Spatrick   case CK_DerivedToBase: {
4822e5dd7070Spatrick     const auto *DerivedClassTy =
4823e5dd7070Spatrick         E->getSubExpr()->getType()->castAs<RecordType>();
4824e5dd7070Spatrick     auto *DerivedClassDecl = cast<CXXRecordDecl>(DerivedClassTy->getDecl());
4825e5dd7070Spatrick 
4826e5dd7070Spatrick     LValue LV = EmitLValue(E->getSubExpr());
4827e5dd7070Spatrick     Address This = LV.getAddress(*this);
4828e5dd7070Spatrick 
4829e5dd7070Spatrick     // Perform the derived-to-base conversion
4830e5dd7070Spatrick     Address Base = GetAddressOfBaseClass(
4831e5dd7070Spatrick         This, DerivedClassDecl, E->path_begin(), E->path_end(),
4832e5dd7070Spatrick         /*NullCheckValue=*/false, E->getExprLoc());
4833e5dd7070Spatrick 
4834e5dd7070Spatrick     // TODO: Support accesses to members of base classes in TBAA. For now, we
4835e5dd7070Spatrick     // conservatively pretend that the complete object is of the base class
4836e5dd7070Spatrick     // type.
4837e5dd7070Spatrick     return MakeAddrLValue(Base, E->getType(), LV.getBaseInfo(),
4838e5dd7070Spatrick                           CGM.getTBAAInfoForSubobject(LV, E->getType()));
4839e5dd7070Spatrick   }
4840e5dd7070Spatrick   case CK_ToUnion:
4841e5dd7070Spatrick     return EmitAggExprToLValue(E);
4842e5dd7070Spatrick   case CK_BaseToDerived: {
4843e5dd7070Spatrick     const auto *DerivedClassTy = E->getType()->castAs<RecordType>();
4844e5dd7070Spatrick     auto *DerivedClassDecl = cast<CXXRecordDecl>(DerivedClassTy->getDecl());
4845e5dd7070Spatrick 
4846e5dd7070Spatrick     LValue LV = EmitLValue(E->getSubExpr());
4847e5dd7070Spatrick 
4848e5dd7070Spatrick     // Perform the base-to-derived conversion
4849e5dd7070Spatrick     Address Derived = GetAddressOfDerivedClass(
4850e5dd7070Spatrick         LV.getAddress(*this), DerivedClassDecl, E->path_begin(), E->path_end(),
4851e5dd7070Spatrick         /*NullCheckValue=*/false);
4852e5dd7070Spatrick 
4853e5dd7070Spatrick     // C++11 [expr.static.cast]p2: Behavior is undefined if a downcast is
4854e5dd7070Spatrick     // performed and the object is not of the derived type.
4855e5dd7070Spatrick     if (sanitizePerformTypeCheck())
4856e5dd7070Spatrick       EmitTypeCheck(TCK_DowncastReference, E->getExprLoc(),
4857e5dd7070Spatrick                     Derived.getPointer(), E->getType());
4858e5dd7070Spatrick 
4859e5dd7070Spatrick     if (SanOpts.has(SanitizerKind::CFIDerivedCast))
4860*12c85518Srobert       EmitVTablePtrCheckForCast(E->getType(), Derived,
4861e5dd7070Spatrick                                 /*MayBeNull=*/false, CFITCK_DerivedCast,
4862e5dd7070Spatrick                                 E->getBeginLoc());
4863e5dd7070Spatrick 
4864e5dd7070Spatrick     return MakeAddrLValue(Derived, E->getType(), LV.getBaseInfo(),
4865e5dd7070Spatrick                           CGM.getTBAAInfoForSubobject(LV, E->getType()));
4866e5dd7070Spatrick   }
4867e5dd7070Spatrick   case CK_LValueBitCast: {
4868e5dd7070Spatrick     // This must be a reinterpret_cast (or c-style equivalent).
4869e5dd7070Spatrick     const auto *CE = cast<ExplicitCastExpr>(E);
4870e5dd7070Spatrick 
4871e5dd7070Spatrick     CGM.EmitExplicitCastExprType(CE, this);
4872e5dd7070Spatrick     LValue LV = EmitLValue(E->getSubExpr());
4873*12c85518Srobert     Address V = Builder.CreateElementBitCast(
4874*12c85518Srobert         LV.getAddress(*this),
4875*12c85518Srobert         ConvertTypeForMem(CE->getTypeAsWritten()->getPointeeType()));
4876e5dd7070Spatrick 
4877e5dd7070Spatrick     if (SanOpts.has(SanitizerKind::CFIUnrelatedCast))
4878*12c85518Srobert       EmitVTablePtrCheckForCast(E->getType(), V,
4879e5dd7070Spatrick                                 /*MayBeNull=*/false, CFITCK_UnrelatedCast,
4880e5dd7070Spatrick                                 E->getBeginLoc());
4881e5dd7070Spatrick 
4882e5dd7070Spatrick     return MakeAddrLValue(V, E->getType(), LV.getBaseInfo(),
4883e5dd7070Spatrick                           CGM.getTBAAInfoForSubobject(LV, E->getType()));
4884e5dd7070Spatrick   }
4885e5dd7070Spatrick   case CK_AddressSpaceConversion: {
4886e5dd7070Spatrick     LValue LV = EmitLValue(E->getSubExpr());
4887e5dd7070Spatrick     QualType DestTy = getContext().getPointerType(E->getType());
4888e5dd7070Spatrick     llvm::Value *V = getTargetHooks().performAddrSpaceCast(
4889e5dd7070Spatrick         *this, LV.getPointer(*this),
4890e5dd7070Spatrick         E->getSubExpr()->getType().getAddressSpace(),
4891e5dd7070Spatrick         E->getType().getAddressSpace(), ConvertType(DestTy));
4892*12c85518Srobert     return MakeAddrLValue(Address(V, ConvertTypeForMem(E->getType()),
4893*12c85518Srobert                                   LV.getAddress(*this).getAlignment()),
4894e5dd7070Spatrick                           E->getType(), LV.getBaseInfo(), LV.getTBAAInfo());
4895e5dd7070Spatrick   }
4896e5dd7070Spatrick   case CK_ObjCObjectLValueCast: {
4897e5dd7070Spatrick     LValue LV = EmitLValue(E->getSubExpr());
4898e5dd7070Spatrick     Address V = Builder.CreateElementBitCast(LV.getAddress(*this),
4899e5dd7070Spatrick                                              ConvertType(E->getType()));
4900e5dd7070Spatrick     return MakeAddrLValue(V, E->getType(), LV.getBaseInfo(),
4901e5dd7070Spatrick                           CGM.getTBAAInfoForSubobject(LV, E->getType()));
4902e5dd7070Spatrick   }
4903e5dd7070Spatrick   case CK_ZeroToOCLOpaqueType:
4904e5dd7070Spatrick     llvm_unreachable("NULL to OpenCL opaque type lvalue cast is not valid");
4905e5dd7070Spatrick   }
4906e5dd7070Spatrick 
4907e5dd7070Spatrick   llvm_unreachable("Unhandled lvalue cast kind?");
4908e5dd7070Spatrick }
4909e5dd7070Spatrick 
EmitOpaqueValueLValue(const OpaqueValueExpr * e)4910e5dd7070Spatrick LValue CodeGenFunction::EmitOpaqueValueLValue(const OpaqueValueExpr *e) {
4911e5dd7070Spatrick   assert(OpaqueValueMappingData::shouldBindAsLValue(e));
4912e5dd7070Spatrick   return getOrCreateOpaqueLValueMapping(e);
4913e5dd7070Spatrick }
4914e5dd7070Spatrick 
4915e5dd7070Spatrick LValue
getOrCreateOpaqueLValueMapping(const OpaqueValueExpr * e)4916e5dd7070Spatrick CodeGenFunction::getOrCreateOpaqueLValueMapping(const OpaqueValueExpr *e) {
4917e5dd7070Spatrick   assert(OpaqueValueMapping::shouldBindAsLValue(e));
4918e5dd7070Spatrick 
4919e5dd7070Spatrick   llvm::DenseMap<const OpaqueValueExpr*,LValue>::iterator
4920e5dd7070Spatrick       it = OpaqueLValues.find(e);
4921e5dd7070Spatrick 
4922e5dd7070Spatrick   if (it != OpaqueLValues.end())
4923e5dd7070Spatrick     return it->second;
4924e5dd7070Spatrick 
4925e5dd7070Spatrick   assert(e->isUnique() && "LValue for a nonunique OVE hasn't been emitted");
4926e5dd7070Spatrick   return EmitLValue(e->getSourceExpr());
4927e5dd7070Spatrick }
4928e5dd7070Spatrick 
4929e5dd7070Spatrick RValue
getOrCreateOpaqueRValueMapping(const OpaqueValueExpr * e)4930e5dd7070Spatrick CodeGenFunction::getOrCreateOpaqueRValueMapping(const OpaqueValueExpr *e) {
4931e5dd7070Spatrick   assert(!OpaqueValueMapping::shouldBindAsLValue(e));
4932e5dd7070Spatrick 
4933e5dd7070Spatrick   llvm::DenseMap<const OpaqueValueExpr*,RValue>::iterator
4934e5dd7070Spatrick       it = OpaqueRValues.find(e);
4935e5dd7070Spatrick 
4936e5dd7070Spatrick   if (it != OpaqueRValues.end())
4937e5dd7070Spatrick     return it->second;
4938e5dd7070Spatrick 
4939e5dd7070Spatrick   assert(e->isUnique() && "RValue for a nonunique OVE hasn't been emitted");
4940e5dd7070Spatrick   return EmitAnyExpr(e->getSourceExpr());
4941e5dd7070Spatrick }
4942e5dd7070Spatrick 
EmitRValueForField(LValue LV,const FieldDecl * FD,SourceLocation Loc)4943e5dd7070Spatrick RValue CodeGenFunction::EmitRValueForField(LValue LV,
4944e5dd7070Spatrick                                            const FieldDecl *FD,
4945e5dd7070Spatrick                                            SourceLocation Loc) {
4946e5dd7070Spatrick   QualType FT = FD->getType();
4947e5dd7070Spatrick   LValue FieldLV = EmitLValueForField(LV, FD);
4948e5dd7070Spatrick   switch (getEvaluationKind(FT)) {
4949e5dd7070Spatrick   case TEK_Complex:
4950e5dd7070Spatrick     return RValue::getComplex(EmitLoadOfComplex(FieldLV, Loc));
4951e5dd7070Spatrick   case TEK_Aggregate:
4952e5dd7070Spatrick     return FieldLV.asAggregateRValue(*this);
4953e5dd7070Spatrick   case TEK_Scalar:
4954e5dd7070Spatrick     // This routine is used to load fields one-by-one to perform a copy, so
4955e5dd7070Spatrick     // don't load reference fields.
4956e5dd7070Spatrick     if (FD->getType()->isReferenceType())
4957e5dd7070Spatrick       return RValue::get(FieldLV.getPointer(*this));
4958e5dd7070Spatrick     // Call EmitLoadOfScalar except when the lvalue is a bitfield to emit a
4959e5dd7070Spatrick     // primitive load.
4960e5dd7070Spatrick     if (FieldLV.isBitField())
4961e5dd7070Spatrick       return EmitLoadOfLValue(FieldLV, Loc);
4962e5dd7070Spatrick     return RValue::get(EmitLoadOfScalar(FieldLV, Loc));
4963e5dd7070Spatrick   }
4964e5dd7070Spatrick   llvm_unreachable("bad evaluation kind");
4965e5dd7070Spatrick }
4966e5dd7070Spatrick 
4967e5dd7070Spatrick //===--------------------------------------------------------------------===//
4968e5dd7070Spatrick //                             Expression Emission
4969e5dd7070Spatrick //===--------------------------------------------------------------------===//
4970e5dd7070Spatrick 
EmitCallExpr(const CallExpr * E,ReturnValueSlot ReturnValue)4971e5dd7070Spatrick RValue CodeGenFunction::EmitCallExpr(const CallExpr *E,
4972e5dd7070Spatrick                                      ReturnValueSlot ReturnValue) {
4973e5dd7070Spatrick   // Builtins never have block type.
4974e5dd7070Spatrick   if (E->getCallee()->getType()->isBlockPointerType())
4975e5dd7070Spatrick     return EmitBlockCallExpr(E, ReturnValue);
4976e5dd7070Spatrick 
4977e5dd7070Spatrick   if (const auto *CE = dyn_cast<CXXMemberCallExpr>(E))
4978e5dd7070Spatrick     return EmitCXXMemberCallExpr(CE, ReturnValue);
4979e5dd7070Spatrick 
4980e5dd7070Spatrick   if (const auto *CE = dyn_cast<CUDAKernelCallExpr>(E))
4981e5dd7070Spatrick     return EmitCUDAKernelCallExpr(CE, ReturnValue);
4982e5dd7070Spatrick 
4983e5dd7070Spatrick   if (const auto *CE = dyn_cast<CXXOperatorCallExpr>(E))
4984e5dd7070Spatrick     if (const CXXMethodDecl *MD =
4985e5dd7070Spatrick           dyn_cast_or_null<CXXMethodDecl>(CE->getCalleeDecl()))
4986e5dd7070Spatrick       return EmitCXXOperatorMemberCallExpr(CE, MD, ReturnValue);
4987e5dd7070Spatrick 
4988e5dd7070Spatrick   CGCallee callee = EmitCallee(E->getCallee());
4989e5dd7070Spatrick 
4990e5dd7070Spatrick   if (callee.isBuiltin()) {
4991e5dd7070Spatrick     return EmitBuiltinExpr(callee.getBuiltinDecl(), callee.getBuiltinID(),
4992e5dd7070Spatrick                            E, ReturnValue);
4993e5dd7070Spatrick   }
4994e5dd7070Spatrick 
4995e5dd7070Spatrick   if (callee.isPseudoDestructor()) {
4996e5dd7070Spatrick     return EmitCXXPseudoDestructorExpr(callee.getPseudoDestructorExpr());
4997e5dd7070Spatrick   }
4998e5dd7070Spatrick 
4999e5dd7070Spatrick   return EmitCall(E->getCallee()->getType(), callee, E, ReturnValue);
5000e5dd7070Spatrick }
5001e5dd7070Spatrick 
5002e5dd7070Spatrick /// Emit a CallExpr without considering whether it might be a subclass.
EmitSimpleCallExpr(const CallExpr * E,ReturnValueSlot ReturnValue)5003e5dd7070Spatrick RValue CodeGenFunction::EmitSimpleCallExpr(const CallExpr *E,
5004e5dd7070Spatrick                                            ReturnValueSlot ReturnValue) {
5005e5dd7070Spatrick   CGCallee Callee = EmitCallee(E->getCallee());
5006e5dd7070Spatrick   return EmitCall(E->getCallee()->getType(), Callee, E, ReturnValue);
5007e5dd7070Spatrick }
5008e5dd7070Spatrick 
5009*12c85518Srobert // Detect the unusual situation where an inline version is shadowed by a
5010*12c85518Srobert // non-inline version. In that case we should pick the external one
5011*12c85518Srobert // everywhere. That's GCC behavior too.
OnlyHasInlineBuiltinDeclaration(const FunctionDecl * FD)5012*12c85518Srobert static bool OnlyHasInlineBuiltinDeclaration(const FunctionDecl *FD) {
5013*12c85518Srobert   for (const FunctionDecl *PD = FD; PD; PD = PD->getPreviousDecl())
5014*12c85518Srobert     if (!PD->isInlineBuiltinDeclaration())
5015*12c85518Srobert       return false;
5016*12c85518Srobert   return true;
5017*12c85518Srobert }
5018*12c85518Srobert 
EmitDirectCallee(CodeGenFunction & CGF,GlobalDecl GD)5019ec727ea7Spatrick static CGCallee EmitDirectCallee(CodeGenFunction &CGF, GlobalDecl GD) {
5020ec727ea7Spatrick   const FunctionDecl *FD = cast<FunctionDecl>(GD.getDecl());
5021e5dd7070Spatrick 
5022e5dd7070Spatrick   if (auto builtinID = FD->getBuiltinID()) {
5023*12c85518Srobert     std::string NoBuiltinFD = ("no-builtin-" + FD->getName()).str();
5024*12c85518Srobert     std::string NoBuiltins = "no-builtins";
5025*12c85518Srobert 
5026*12c85518Srobert     StringRef Ident = CGF.CGM.getMangledName(GD);
5027*12c85518Srobert     std::string FDInlineName = (Ident + ".inline").str();
5028*12c85518Srobert 
5029*12c85518Srobert     bool IsPredefinedLibFunction =
5030*12c85518Srobert         CGF.getContext().BuiltinInfo.isPredefinedLibFunction(builtinID);
5031*12c85518Srobert     bool HasAttributeNoBuiltin =
5032*12c85518Srobert         CGF.CurFn->getAttributes().hasFnAttr(NoBuiltinFD) ||
5033*12c85518Srobert         CGF.CurFn->getAttributes().hasFnAttr(NoBuiltins);
5034*12c85518Srobert 
5035*12c85518Srobert     // When directing calling an inline builtin, call it through it's mangled
5036*12c85518Srobert     // name to make it clear it's not the actual builtin.
5037*12c85518Srobert     if (CGF.CurFn->getName() != FDInlineName &&
5038*12c85518Srobert         OnlyHasInlineBuiltinDeclaration(FD)) {
5039*12c85518Srobert       llvm::Constant *CalleePtr = EmitFunctionDeclPointer(CGF.CGM, GD);
5040*12c85518Srobert       llvm::Function *Fn = llvm::cast<llvm::Function>(CalleePtr);
5041*12c85518Srobert       llvm::Module *M = Fn->getParent();
5042*12c85518Srobert       llvm::Function *Clone = M->getFunction(FDInlineName);
5043*12c85518Srobert       if (!Clone) {
5044*12c85518Srobert         Clone = llvm::Function::Create(Fn->getFunctionType(),
5045*12c85518Srobert                                        llvm::GlobalValue::InternalLinkage,
5046*12c85518Srobert                                        Fn->getAddressSpace(), FDInlineName, M);
5047*12c85518Srobert         Clone->addFnAttr(llvm::Attribute::AlwaysInline);
5048*12c85518Srobert       }
5049*12c85518Srobert       return CGCallee::forDirect(Clone, GD);
5050*12c85518Srobert     }
5051*12c85518Srobert 
5052*12c85518Srobert     // Replaceable builtins provide their own implementation of a builtin. If we
5053*12c85518Srobert     // are in an inline builtin implementation, avoid trivial infinite
5054*12c85518Srobert     // recursion. Honor __attribute__((no_builtin("foo"))) or
5055*12c85518Srobert     // __attribute__((no_builtin)) on the current function unless foo is
5056*12c85518Srobert     // not a predefined library function which means we must generate the
5057*12c85518Srobert     // builtin no matter what.
5058*12c85518Srobert     else if (!IsPredefinedLibFunction || !HasAttributeNoBuiltin)
5059e5dd7070Spatrick       return CGCallee::forBuiltin(builtinID, FD);
5060e5dd7070Spatrick   }
5061e5dd7070Spatrick 
5062a9ac8606Spatrick   llvm::Constant *CalleePtr = EmitFunctionDeclPointer(CGF.CGM, GD);
5063a9ac8606Spatrick   if (CGF.CGM.getLangOpts().CUDA && !CGF.CGM.getLangOpts().CUDAIsDevice &&
5064a9ac8606Spatrick       FD->hasAttr<CUDAGlobalAttr>())
5065a9ac8606Spatrick     CalleePtr = CGF.CGM.getCUDARuntime().getKernelStub(
5066a9ac8606Spatrick         cast<llvm::GlobalValue>(CalleePtr->stripPointerCasts()));
5067*12c85518Srobert 
5068a9ac8606Spatrick   return CGCallee::forDirect(CalleePtr, GD);
5069e5dd7070Spatrick }
5070e5dd7070Spatrick 
EmitCallee(const Expr * E)5071e5dd7070Spatrick CGCallee CodeGenFunction::EmitCallee(const Expr *E) {
5072e5dd7070Spatrick   E = E->IgnoreParens();
5073e5dd7070Spatrick 
5074e5dd7070Spatrick   // Look through function-to-pointer decay.
5075e5dd7070Spatrick   if (auto ICE = dyn_cast<ImplicitCastExpr>(E)) {
5076e5dd7070Spatrick     if (ICE->getCastKind() == CK_FunctionToPointerDecay ||
5077e5dd7070Spatrick         ICE->getCastKind() == CK_BuiltinFnToFnPtr) {
5078e5dd7070Spatrick       return EmitCallee(ICE->getSubExpr());
5079e5dd7070Spatrick     }
5080e5dd7070Spatrick 
5081e5dd7070Spatrick   // Resolve direct calls.
5082e5dd7070Spatrick   } else if (auto DRE = dyn_cast<DeclRefExpr>(E)) {
5083e5dd7070Spatrick     if (auto FD = dyn_cast<FunctionDecl>(DRE->getDecl())) {
5084e5dd7070Spatrick       return EmitDirectCallee(*this, FD);
5085e5dd7070Spatrick     }
5086e5dd7070Spatrick   } else if (auto ME = dyn_cast<MemberExpr>(E)) {
5087e5dd7070Spatrick     if (auto FD = dyn_cast<FunctionDecl>(ME->getMemberDecl())) {
5088e5dd7070Spatrick       EmitIgnoredExpr(ME->getBase());
5089e5dd7070Spatrick       return EmitDirectCallee(*this, FD);
5090e5dd7070Spatrick     }
5091e5dd7070Spatrick 
5092e5dd7070Spatrick   // Look through template substitutions.
5093e5dd7070Spatrick   } else if (auto NTTP = dyn_cast<SubstNonTypeTemplateParmExpr>(E)) {
5094e5dd7070Spatrick     return EmitCallee(NTTP->getReplacement());
5095e5dd7070Spatrick 
5096e5dd7070Spatrick   // Treat pseudo-destructor calls differently.
5097e5dd7070Spatrick   } else if (auto PDE = dyn_cast<CXXPseudoDestructorExpr>(E)) {
5098e5dd7070Spatrick     return CGCallee::forPseudoDestructor(PDE);
5099e5dd7070Spatrick   }
5100e5dd7070Spatrick 
5101e5dd7070Spatrick   // Otherwise, we have an indirect reference.
5102e5dd7070Spatrick   llvm::Value *calleePtr;
5103e5dd7070Spatrick   QualType functionType;
5104e5dd7070Spatrick   if (auto ptrType = E->getType()->getAs<PointerType>()) {
5105e5dd7070Spatrick     calleePtr = EmitScalarExpr(E);
5106e5dd7070Spatrick     functionType = ptrType->getPointeeType();
5107e5dd7070Spatrick   } else {
5108e5dd7070Spatrick     functionType = E->getType();
5109e5dd7070Spatrick     calleePtr = EmitLValue(E).getPointer(*this);
5110e5dd7070Spatrick   }
5111e5dd7070Spatrick   assert(functionType->isFunctionType());
5112e5dd7070Spatrick 
5113e5dd7070Spatrick   GlobalDecl GD;
5114e5dd7070Spatrick   if (const auto *VD =
5115e5dd7070Spatrick           dyn_cast_or_null<VarDecl>(E->getReferencedDeclOfCallee()))
5116e5dd7070Spatrick     GD = GlobalDecl(VD);
5117e5dd7070Spatrick 
5118e5dd7070Spatrick   CGCalleeInfo calleeInfo(functionType->getAs<FunctionProtoType>(), GD);
5119e5dd7070Spatrick   CGCallee callee(calleeInfo, calleePtr);
5120e5dd7070Spatrick   return callee;
5121e5dd7070Spatrick }
5122e5dd7070Spatrick 
EmitBinaryOperatorLValue(const BinaryOperator * E)5123e5dd7070Spatrick LValue CodeGenFunction::EmitBinaryOperatorLValue(const BinaryOperator *E) {
5124e5dd7070Spatrick   // Comma expressions just emit their LHS then their RHS as an l-value.
5125e5dd7070Spatrick   if (E->getOpcode() == BO_Comma) {
5126e5dd7070Spatrick     EmitIgnoredExpr(E->getLHS());
5127e5dd7070Spatrick     EnsureInsertPoint();
5128e5dd7070Spatrick     return EmitLValue(E->getRHS());
5129e5dd7070Spatrick   }
5130e5dd7070Spatrick 
5131e5dd7070Spatrick   if (E->getOpcode() == BO_PtrMemD ||
5132e5dd7070Spatrick       E->getOpcode() == BO_PtrMemI)
5133e5dd7070Spatrick     return EmitPointerToDataMemberBinaryExpr(E);
5134e5dd7070Spatrick 
5135e5dd7070Spatrick   assert(E->getOpcode() == BO_Assign && "unexpected binary l-value");
5136e5dd7070Spatrick 
5137e5dd7070Spatrick   // Note that in all of these cases, __block variables need the RHS
5138e5dd7070Spatrick   // evaluated first just in case the variable gets moved by the RHS.
5139e5dd7070Spatrick 
5140e5dd7070Spatrick   switch (getEvaluationKind(E->getType())) {
5141e5dd7070Spatrick   case TEK_Scalar: {
5142e5dd7070Spatrick     switch (E->getLHS()->getType().getObjCLifetime()) {
5143e5dd7070Spatrick     case Qualifiers::OCL_Strong:
5144e5dd7070Spatrick       return EmitARCStoreStrong(E, /*ignored*/ false).first;
5145e5dd7070Spatrick 
5146e5dd7070Spatrick     case Qualifiers::OCL_Autoreleasing:
5147e5dd7070Spatrick       return EmitARCStoreAutoreleasing(E).first;
5148e5dd7070Spatrick 
5149e5dd7070Spatrick     // No reason to do any of these differently.
5150e5dd7070Spatrick     case Qualifiers::OCL_None:
5151e5dd7070Spatrick     case Qualifiers::OCL_ExplicitNone:
5152e5dd7070Spatrick     case Qualifiers::OCL_Weak:
5153e5dd7070Spatrick       break;
5154e5dd7070Spatrick     }
5155e5dd7070Spatrick 
5156e5dd7070Spatrick     RValue RV = EmitAnyExpr(E->getRHS());
5157e5dd7070Spatrick     LValue LV = EmitCheckedLValue(E->getLHS(), TCK_Store);
5158e5dd7070Spatrick     if (RV.isScalar())
5159e5dd7070Spatrick       EmitNullabilityCheck(LV, RV.getScalarVal(), E->getExprLoc());
5160e5dd7070Spatrick     EmitStoreThroughLValue(RV, LV);
5161e5dd7070Spatrick     if (getLangOpts().OpenMP)
5162e5dd7070Spatrick       CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(*this,
5163e5dd7070Spatrick                                                                 E->getLHS());
5164e5dd7070Spatrick     return LV;
5165e5dd7070Spatrick   }
5166e5dd7070Spatrick 
5167e5dd7070Spatrick   case TEK_Complex:
5168e5dd7070Spatrick     return EmitComplexAssignmentLValue(E);
5169e5dd7070Spatrick 
5170e5dd7070Spatrick   case TEK_Aggregate:
5171e5dd7070Spatrick     return EmitAggExprToLValue(E);
5172e5dd7070Spatrick   }
5173e5dd7070Spatrick   llvm_unreachable("bad evaluation kind");
5174e5dd7070Spatrick }
5175e5dd7070Spatrick 
EmitCallExprLValue(const CallExpr * E)5176e5dd7070Spatrick LValue CodeGenFunction::EmitCallExprLValue(const CallExpr *E) {
5177e5dd7070Spatrick   RValue RV = EmitCallExpr(E);
5178e5dd7070Spatrick 
5179e5dd7070Spatrick   if (!RV.isScalar())
5180e5dd7070Spatrick     return MakeAddrLValue(RV.getAggregateAddress(), E->getType(),
5181e5dd7070Spatrick                           AlignmentSource::Decl);
5182e5dd7070Spatrick 
5183e5dd7070Spatrick   assert(E->getCallReturnType(getContext())->isReferenceType() &&
5184e5dd7070Spatrick          "Can't have a scalar return unless the return type is a "
5185e5dd7070Spatrick          "reference type!");
5186e5dd7070Spatrick 
5187e5dd7070Spatrick   return MakeNaturalAlignPointeeAddrLValue(RV.getScalarVal(), E->getType());
5188e5dd7070Spatrick }
5189e5dd7070Spatrick 
EmitVAArgExprLValue(const VAArgExpr * E)5190e5dd7070Spatrick LValue CodeGenFunction::EmitVAArgExprLValue(const VAArgExpr *E) {
5191e5dd7070Spatrick   // FIXME: This shouldn't require another copy.
5192e5dd7070Spatrick   return EmitAggExprToLValue(E);
5193e5dd7070Spatrick }
5194e5dd7070Spatrick 
EmitCXXConstructLValue(const CXXConstructExpr * E)5195e5dd7070Spatrick LValue CodeGenFunction::EmitCXXConstructLValue(const CXXConstructExpr *E) {
5196e5dd7070Spatrick   assert(E->getType()->getAsCXXRecordDecl()->hasTrivialDestructor()
5197e5dd7070Spatrick          && "binding l-value to type which needs a temporary");
5198e5dd7070Spatrick   AggValueSlot Slot = CreateAggTemp(E->getType());
5199e5dd7070Spatrick   EmitCXXConstructExpr(E, Slot);
5200e5dd7070Spatrick   return MakeAddrLValue(Slot.getAddress(), E->getType(), AlignmentSource::Decl);
5201e5dd7070Spatrick }
5202e5dd7070Spatrick 
5203e5dd7070Spatrick LValue
EmitCXXTypeidLValue(const CXXTypeidExpr * E)5204e5dd7070Spatrick CodeGenFunction::EmitCXXTypeidLValue(const CXXTypeidExpr *E) {
5205e5dd7070Spatrick   return MakeNaturalAlignAddrLValue(EmitCXXTypeidExpr(E), E->getType());
5206e5dd7070Spatrick }
5207e5dd7070Spatrick 
EmitCXXUuidofExpr(const CXXUuidofExpr * E)5208e5dd7070Spatrick Address CodeGenFunction::EmitCXXUuidofExpr(const CXXUuidofExpr *E) {
5209ec727ea7Spatrick   return Builder.CreateElementBitCast(CGM.GetAddrOfMSGuidDecl(E->getGuidDecl()),
5210e5dd7070Spatrick                                       ConvertType(E->getType()));
5211e5dd7070Spatrick }
5212e5dd7070Spatrick 
EmitCXXUuidofLValue(const CXXUuidofExpr * E)5213e5dd7070Spatrick LValue CodeGenFunction::EmitCXXUuidofLValue(const CXXUuidofExpr *E) {
5214e5dd7070Spatrick   return MakeAddrLValue(EmitCXXUuidofExpr(E), E->getType(),
5215e5dd7070Spatrick                         AlignmentSource::Decl);
5216e5dd7070Spatrick }
5217e5dd7070Spatrick 
5218e5dd7070Spatrick LValue
EmitCXXBindTemporaryLValue(const CXXBindTemporaryExpr * E)5219e5dd7070Spatrick CodeGenFunction::EmitCXXBindTemporaryLValue(const CXXBindTemporaryExpr *E) {
5220e5dd7070Spatrick   AggValueSlot Slot = CreateAggTemp(E->getType(), "temp.lvalue");
5221e5dd7070Spatrick   Slot.setExternallyDestructed();
5222e5dd7070Spatrick   EmitAggExpr(E->getSubExpr(), Slot);
5223e5dd7070Spatrick   EmitCXXTemporary(E->getTemporary(), E->getType(), Slot.getAddress());
5224e5dd7070Spatrick   return MakeAddrLValue(Slot.getAddress(), E->getType(), AlignmentSource::Decl);
5225e5dd7070Spatrick }
5226e5dd7070Spatrick 
EmitObjCMessageExprLValue(const ObjCMessageExpr * E)5227e5dd7070Spatrick LValue CodeGenFunction::EmitObjCMessageExprLValue(const ObjCMessageExpr *E) {
5228e5dd7070Spatrick   RValue RV = EmitObjCMessageExpr(E);
5229e5dd7070Spatrick 
5230e5dd7070Spatrick   if (!RV.isScalar())
5231e5dd7070Spatrick     return MakeAddrLValue(RV.getAggregateAddress(), E->getType(),
5232e5dd7070Spatrick                           AlignmentSource::Decl);
5233e5dd7070Spatrick 
5234e5dd7070Spatrick   assert(E->getMethodDecl()->getReturnType()->isReferenceType() &&
5235e5dd7070Spatrick          "Can't have a scalar return unless the return type is a "
5236e5dd7070Spatrick          "reference type!");
5237e5dd7070Spatrick 
5238e5dd7070Spatrick   return MakeNaturalAlignPointeeAddrLValue(RV.getScalarVal(), E->getType());
5239e5dd7070Spatrick }
5240e5dd7070Spatrick 
EmitObjCSelectorLValue(const ObjCSelectorExpr * E)5241e5dd7070Spatrick LValue CodeGenFunction::EmitObjCSelectorLValue(const ObjCSelectorExpr *E) {
5242e5dd7070Spatrick   Address V =
5243e5dd7070Spatrick     CGM.getObjCRuntime().GetAddrOfSelector(*this, E->getSelector());
5244e5dd7070Spatrick   return MakeAddrLValue(V, E->getType(), AlignmentSource::Decl);
5245e5dd7070Spatrick }
5246e5dd7070Spatrick 
EmitIvarOffset(const ObjCInterfaceDecl * Interface,const ObjCIvarDecl * Ivar)5247e5dd7070Spatrick llvm::Value *CodeGenFunction::EmitIvarOffset(const ObjCInterfaceDecl *Interface,
5248e5dd7070Spatrick                                              const ObjCIvarDecl *Ivar) {
5249e5dd7070Spatrick   return CGM.getObjCRuntime().EmitIvarOffset(*this, Interface, Ivar);
5250e5dd7070Spatrick }
5251e5dd7070Spatrick 
5252*12c85518Srobert llvm::Value *
EmitIvarOffsetAsPointerDiff(const ObjCInterfaceDecl * Interface,const ObjCIvarDecl * Ivar)5253*12c85518Srobert CodeGenFunction::EmitIvarOffsetAsPointerDiff(const ObjCInterfaceDecl *Interface,
5254*12c85518Srobert                                              const ObjCIvarDecl *Ivar) {
5255*12c85518Srobert   llvm::Value *OffsetValue = EmitIvarOffset(Interface, Ivar);
5256*12c85518Srobert   QualType PointerDiffType = getContext().getPointerDiffType();
5257*12c85518Srobert   return Builder.CreateZExtOrTrunc(OffsetValue,
5258*12c85518Srobert                                    getTypes().ConvertType(PointerDiffType));
5259*12c85518Srobert }
5260*12c85518Srobert 
EmitLValueForIvar(QualType ObjectTy,llvm::Value * BaseValue,const ObjCIvarDecl * Ivar,unsigned CVRQualifiers)5261e5dd7070Spatrick LValue CodeGenFunction::EmitLValueForIvar(QualType ObjectTy,
5262e5dd7070Spatrick                                           llvm::Value *BaseValue,
5263e5dd7070Spatrick                                           const ObjCIvarDecl *Ivar,
5264e5dd7070Spatrick                                           unsigned CVRQualifiers) {
5265e5dd7070Spatrick   return CGM.getObjCRuntime().EmitObjCValueForIvar(*this, ObjectTy, BaseValue,
5266e5dd7070Spatrick                                                    Ivar, CVRQualifiers);
5267e5dd7070Spatrick }
5268e5dd7070Spatrick 
EmitObjCIvarRefLValue(const ObjCIvarRefExpr * E)5269e5dd7070Spatrick LValue CodeGenFunction::EmitObjCIvarRefLValue(const ObjCIvarRefExpr *E) {
5270e5dd7070Spatrick   // FIXME: A lot of the code below could be shared with EmitMemberExpr.
5271e5dd7070Spatrick   llvm::Value *BaseValue = nullptr;
5272e5dd7070Spatrick   const Expr *BaseExpr = E->getBase();
5273e5dd7070Spatrick   Qualifiers BaseQuals;
5274e5dd7070Spatrick   QualType ObjectTy;
5275e5dd7070Spatrick   if (E->isArrow()) {
5276e5dd7070Spatrick     BaseValue = EmitScalarExpr(BaseExpr);
5277e5dd7070Spatrick     ObjectTy = BaseExpr->getType()->getPointeeType();
5278e5dd7070Spatrick     BaseQuals = ObjectTy.getQualifiers();
5279e5dd7070Spatrick   } else {
5280e5dd7070Spatrick     LValue BaseLV = EmitLValue(BaseExpr);
5281e5dd7070Spatrick     BaseValue = BaseLV.getPointer(*this);
5282e5dd7070Spatrick     ObjectTy = BaseExpr->getType();
5283e5dd7070Spatrick     BaseQuals = ObjectTy.getQualifiers();
5284e5dd7070Spatrick   }
5285e5dd7070Spatrick 
5286e5dd7070Spatrick   LValue LV =
5287e5dd7070Spatrick     EmitLValueForIvar(ObjectTy, BaseValue, E->getDecl(),
5288e5dd7070Spatrick                       BaseQuals.getCVRQualifiers());
5289e5dd7070Spatrick   setObjCGCLValueClass(getContext(), E, LV);
5290e5dd7070Spatrick   return LV;
5291e5dd7070Spatrick }
5292e5dd7070Spatrick 
EmitStmtExprLValue(const StmtExpr * E)5293e5dd7070Spatrick LValue CodeGenFunction::EmitStmtExprLValue(const StmtExpr *E) {
5294e5dd7070Spatrick   // Can only get l-value for message expression returning aggregate type
5295e5dd7070Spatrick   RValue RV = EmitAnyExprToTemp(E);
5296e5dd7070Spatrick   return MakeAddrLValue(RV.getAggregateAddress(), E->getType(),
5297e5dd7070Spatrick                         AlignmentSource::Decl);
5298e5dd7070Spatrick }
5299e5dd7070Spatrick 
EmitCall(QualType CalleeType,const CGCallee & OrigCallee,const CallExpr * E,ReturnValueSlot ReturnValue,llvm::Value * Chain)5300e5dd7070Spatrick RValue CodeGenFunction::EmitCall(QualType CalleeType, const CGCallee &OrigCallee,
5301e5dd7070Spatrick                                  const CallExpr *E, ReturnValueSlot ReturnValue,
5302e5dd7070Spatrick                                  llvm::Value *Chain) {
5303e5dd7070Spatrick   // Get the actual function type. The callee type will always be a pointer to
5304e5dd7070Spatrick   // function type or a block pointer type.
5305e5dd7070Spatrick   assert(CalleeType->isFunctionPointerType() &&
5306e5dd7070Spatrick          "Call must have function pointer type!");
5307e5dd7070Spatrick 
5308e5dd7070Spatrick   const Decl *TargetDecl =
5309e5dd7070Spatrick       OrigCallee.getAbstractInfo().getCalleeDecl().getDecl();
5310e5dd7070Spatrick 
5311e5dd7070Spatrick   CalleeType = getContext().getCanonicalType(CalleeType);
5312e5dd7070Spatrick 
5313e5dd7070Spatrick   auto PointeeType = cast<PointerType>(CalleeType)->getPointeeType();
5314e5dd7070Spatrick 
5315e5dd7070Spatrick   CGCallee Callee = OrigCallee;
5316e5dd7070Spatrick 
5317e5dd7070Spatrick   if (getLangOpts().CPlusPlus && SanOpts.has(SanitizerKind::Function) &&
5318e5dd7070Spatrick       (!TargetDecl || !isa<FunctionDecl>(TargetDecl))) {
5319e5dd7070Spatrick     if (llvm::Constant *PrefixSig =
5320e5dd7070Spatrick             CGM.getTargetCodeGenInfo().getUBSanFunctionSignature(CGM)) {
5321e5dd7070Spatrick       SanitizerScope SanScope(this);
5322e5dd7070Spatrick       // Remove any (C++17) exception specifications, to allow calling e.g. a
5323e5dd7070Spatrick       // noexcept function through a non-noexcept pointer.
5324e5dd7070Spatrick       auto ProtoTy =
5325e5dd7070Spatrick         getContext().getFunctionTypeWithExceptionSpec(PointeeType, EST_None);
5326e5dd7070Spatrick       llvm::Constant *FTRTTIConst =
5327e5dd7070Spatrick           CGM.GetAddrOfRTTIDescriptor(ProtoTy, /*ForEH=*/true);
5328a9ac8606Spatrick       llvm::Type *PrefixSigType = PrefixSig->getType();
5329e5dd7070Spatrick       llvm::StructType *PrefixStructTy = llvm::StructType::get(
5330a9ac8606Spatrick           CGM.getLLVMContext(), {PrefixSigType, Int32Ty}, /*isPacked=*/true);
5331e5dd7070Spatrick 
5332e5dd7070Spatrick       llvm::Value *CalleePtr = Callee.getFunctionPointer();
5333e5dd7070Spatrick 
5334e5dd7070Spatrick       llvm::Value *CalleePrefixStruct = Builder.CreateBitCast(
5335e5dd7070Spatrick           CalleePtr, llvm::PointerType::getUnqual(PrefixStructTy));
5336e5dd7070Spatrick       llvm::Value *CalleeSigPtr =
5337e5dd7070Spatrick           Builder.CreateConstGEP2_32(PrefixStructTy, CalleePrefixStruct, 0, 0);
5338e5dd7070Spatrick       llvm::Value *CalleeSig =
5339a9ac8606Spatrick           Builder.CreateAlignedLoad(PrefixSigType, CalleeSigPtr, getIntAlign());
5340e5dd7070Spatrick       llvm::Value *CalleeSigMatch = Builder.CreateICmpEQ(CalleeSig, PrefixSig);
5341e5dd7070Spatrick 
5342e5dd7070Spatrick       llvm::BasicBlock *Cont = createBasicBlock("cont");
5343e5dd7070Spatrick       llvm::BasicBlock *TypeCheck = createBasicBlock("typecheck");
5344e5dd7070Spatrick       Builder.CreateCondBr(CalleeSigMatch, TypeCheck, Cont);
5345e5dd7070Spatrick 
5346e5dd7070Spatrick       EmitBlock(TypeCheck);
5347e5dd7070Spatrick       llvm::Value *CalleeRTTIPtr =
5348e5dd7070Spatrick           Builder.CreateConstGEP2_32(PrefixStructTy, CalleePrefixStruct, 0, 1);
5349e5dd7070Spatrick       llvm::Value *CalleeRTTIEncoded =
5350a9ac8606Spatrick           Builder.CreateAlignedLoad(Int32Ty, CalleeRTTIPtr, getPointerAlign());
5351e5dd7070Spatrick       llvm::Value *CalleeRTTI =
5352e5dd7070Spatrick           DecodeAddrUsedInPrologue(CalleePtr, CalleeRTTIEncoded);
5353e5dd7070Spatrick       llvm::Value *CalleeRTTIMatch =
5354e5dd7070Spatrick           Builder.CreateICmpEQ(CalleeRTTI, FTRTTIConst);
5355e5dd7070Spatrick       llvm::Constant *StaticData[] = {EmitCheckSourceLocation(E->getBeginLoc()),
5356e5dd7070Spatrick                                       EmitCheckTypeDescriptor(CalleeType)};
5357e5dd7070Spatrick       EmitCheck(std::make_pair(CalleeRTTIMatch, SanitizerKind::Function),
5358e5dd7070Spatrick                 SanitizerHandler::FunctionTypeMismatch, StaticData,
5359e5dd7070Spatrick                 {CalleePtr, CalleeRTTI, FTRTTIConst});
5360e5dd7070Spatrick 
5361e5dd7070Spatrick       Builder.CreateBr(Cont);
5362e5dd7070Spatrick       EmitBlock(Cont);
5363e5dd7070Spatrick     }
5364e5dd7070Spatrick   }
5365e5dd7070Spatrick 
5366e5dd7070Spatrick   const auto *FnType = cast<FunctionType>(PointeeType);
5367e5dd7070Spatrick 
5368e5dd7070Spatrick   // If we are checking indirect calls and this call is indirect, check that the
5369e5dd7070Spatrick   // function pointer is a member of the bit set for the function type.
5370e5dd7070Spatrick   if (SanOpts.has(SanitizerKind::CFIICall) &&
5371e5dd7070Spatrick       (!TargetDecl || !isa<FunctionDecl>(TargetDecl))) {
5372e5dd7070Spatrick     SanitizerScope SanScope(this);
5373e5dd7070Spatrick     EmitSanitizerStatReport(llvm::SanStat_CFI_ICall);
5374e5dd7070Spatrick 
5375e5dd7070Spatrick     llvm::Metadata *MD;
5376e5dd7070Spatrick     if (CGM.getCodeGenOpts().SanitizeCfiICallGeneralizePointers)
5377e5dd7070Spatrick       MD = CGM.CreateMetadataIdentifierGeneralized(QualType(FnType, 0));
5378e5dd7070Spatrick     else
5379e5dd7070Spatrick       MD = CGM.CreateMetadataIdentifierForType(QualType(FnType, 0));
5380e5dd7070Spatrick 
5381e5dd7070Spatrick     llvm::Value *TypeId = llvm::MetadataAsValue::get(getLLVMContext(), MD);
5382e5dd7070Spatrick 
5383e5dd7070Spatrick     llvm::Value *CalleePtr = Callee.getFunctionPointer();
5384e5dd7070Spatrick     llvm::Value *CastedCallee = Builder.CreateBitCast(CalleePtr, Int8PtrTy);
5385e5dd7070Spatrick     llvm::Value *TypeTest = Builder.CreateCall(
5386e5dd7070Spatrick         CGM.getIntrinsic(llvm::Intrinsic::type_test), {CastedCallee, TypeId});
5387e5dd7070Spatrick 
5388e5dd7070Spatrick     auto CrossDsoTypeId = CGM.CreateCrossDsoCfiTypeId(MD);
5389e5dd7070Spatrick     llvm::Constant *StaticData[] = {
5390e5dd7070Spatrick         llvm::ConstantInt::get(Int8Ty, CFITCK_ICall),
5391e5dd7070Spatrick         EmitCheckSourceLocation(E->getBeginLoc()),
5392e5dd7070Spatrick         EmitCheckTypeDescriptor(QualType(FnType, 0)),
5393e5dd7070Spatrick     };
5394e5dd7070Spatrick     if (CGM.getCodeGenOpts().SanitizeCfiCrossDso && CrossDsoTypeId) {
5395e5dd7070Spatrick       EmitCfiSlowPathCheck(SanitizerKind::CFIICall, TypeTest, CrossDsoTypeId,
5396e5dd7070Spatrick                            CastedCallee, StaticData);
5397e5dd7070Spatrick     } else {
5398e5dd7070Spatrick       EmitCheck(std::make_pair(TypeTest, SanitizerKind::CFIICall),
5399e5dd7070Spatrick                 SanitizerHandler::CFICheckFail, StaticData,
5400e5dd7070Spatrick                 {CastedCallee, llvm::UndefValue::get(IntPtrTy)});
5401e5dd7070Spatrick     }
5402e5dd7070Spatrick   }
5403e5dd7070Spatrick 
5404e5dd7070Spatrick   CallArgList Args;
5405e5dd7070Spatrick   if (Chain)
5406e5dd7070Spatrick     Args.add(RValue::get(Builder.CreateBitCast(Chain, CGM.VoidPtrTy)),
5407e5dd7070Spatrick              CGM.getContext().VoidPtrTy);
5408e5dd7070Spatrick 
5409e5dd7070Spatrick   // C++17 requires that we evaluate arguments to a call using assignment syntax
5410e5dd7070Spatrick   // right-to-left, and that we evaluate arguments to certain other operators
5411e5dd7070Spatrick   // left-to-right. Note that we allow this to override the order dictated by
5412e5dd7070Spatrick   // the calling convention on the MS ABI, which means that parameter
5413e5dd7070Spatrick   // destruction order is not necessarily reverse construction order.
5414e5dd7070Spatrick   // FIXME: Revisit this based on C++ committee response to unimplementability.
5415e5dd7070Spatrick   EvaluationOrder Order = EvaluationOrder::Default;
5416e5dd7070Spatrick   if (auto *OCE = dyn_cast<CXXOperatorCallExpr>(E)) {
5417e5dd7070Spatrick     if (OCE->isAssignmentOp())
5418e5dd7070Spatrick       Order = EvaluationOrder::ForceRightToLeft;
5419e5dd7070Spatrick     else {
5420e5dd7070Spatrick       switch (OCE->getOperator()) {
5421e5dd7070Spatrick       case OO_LessLess:
5422e5dd7070Spatrick       case OO_GreaterGreater:
5423e5dd7070Spatrick       case OO_AmpAmp:
5424e5dd7070Spatrick       case OO_PipePipe:
5425e5dd7070Spatrick       case OO_Comma:
5426e5dd7070Spatrick       case OO_ArrowStar:
5427e5dd7070Spatrick         Order = EvaluationOrder::ForceLeftToRight;
5428e5dd7070Spatrick         break;
5429e5dd7070Spatrick       default:
5430e5dd7070Spatrick         break;
5431e5dd7070Spatrick       }
5432e5dd7070Spatrick     }
5433e5dd7070Spatrick   }
5434e5dd7070Spatrick 
5435e5dd7070Spatrick   EmitCallArgs(Args, dyn_cast<FunctionProtoType>(FnType), E->arguments(),
5436e5dd7070Spatrick                E->getDirectCallee(), /*ParamsToSkip*/ 0, Order);
5437e5dd7070Spatrick 
5438e5dd7070Spatrick   const CGFunctionInfo &FnInfo = CGM.getTypes().arrangeFreeFunctionCall(
5439e5dd7070Spatrick       Args, FnType, /*ChainCall=*/Chain);
5440e5dd7070Spatrick 
5441e5dd7070Spatrick   // C99 6.5.2.2p6:
5442e5dd7070Spatrick   //   If the expression that denotes the called function has a type
5443e5dd7070Spatrick   //   that does not include a prototype, [the default argument
5444e5dd7070Spatrick   //   promotions are performed]. If the number of arguments does not
5445e5dd7070Spatrick   //   equal the number of parameters, the behavior is undefined. If
5446e5dd7070Spatrick   //   the function is defined with a type that includes a prototype,
5447e5dd7070Spatrick   //   and either the prototype ends with an ellipsis (, ...) or the
5448e5dd7070Spatrick   //   types of the arguments after promotion are not compatible with
5449e5dd7070Spatrick   //   the types of the parameters, the behavior is undefined. If the
5450e5dd7070Spatrick   //   function is defined with a type that does not include a
5451e5dd7070Spatrick   //   prototype, and the types of the arguments after promotion are
5452e5dd7070Spatrick   //   not compatible with those of the parameters after promotion,
5453e5dd7070Spatrick   //   the behavior is undefined [except in some trivial cases].
5454e5dd7070Spatrick   // That is, in the general case, we should assume that a call
5455e5dd7070Spatrick   // through an unprototyped function type works like a *non-variadic*
5456e5dd7070Spatrick   // call.  The way we make this work is to cast to the exact type
5457e5dd7070Spatrick   // of the promoted arguments.
5458e5dd7070Spatrick   //
5459e5dd7070Spatrick   // Chain calls use this same code path to add the invisible chain parameter
5460e5dd7070Spatrick   // to the function type.
5461e5dd7070Spatrick   if (isa<FunctionNoProtoType>(FnType) || Chain) {
5462e5dd7070Spatrick     llvm::Type *CalleeTy = getTypes().GetFunctionType(FnInfo);
5463ec727ea7Spatrick     int AS = Callee.getFunctionPointer()->getType()->getPointerAddressSpace();
5464ec727ea7Spatrick     CalleeTy = CalleeTy->getPointerTo(AS);
5465e5dd7070Spatrick 
5466e5dd7070Spatrick     llvm::Value *CalleePtr = Callee.getFunctionPointer();
5467e5dd7070Spatrick     CalleePtr = Builder.CreateBitCast(CalleePtr, CalleeTy, "callee.knr.cast");
5468e5dd7070Spatrick     Callee.setFunctionPointer(CalleePtr);
5469e5dd7070Spatrick   }
5470e5dd7070Spatrick 
5471a9ac8606Spatrick   // HIP function pointer contains kernel handle when it is used in triple
5472a9ac8606Spatrick   // chevron. The kernel stub needs to be loaded from kernel handle and used
5473a9ac8606Spatrick   // as callee.
5474a9ac8606Spatrick   if (CGM.getLangOpts().HIP && !CGM.getLangOpts().CUDAIsDevice &&
5475a9ac8606Spatrick       isa<CUDAKernelCallExpr>(E) &&
5476a9ac8606Spatrick       (!TargetDecl || !isa<FunctionDecl>(TargetDecl))) {
5477a9ac8606Spatrick     llvm::Value *Handle = Callee.getFunctionPointer();
5478a9ac8606Spatrick     auto *Cast =
5479a9ac8606Spatrick         Builder.CreateBitCast(Handle, Handle->getType()->getPointerTo());
5480*12c85518Srobert     auto *Stub = Builder.CreateLoad(
5481*12c85518Srobert         Address(Cast, Handle->getType(), CGM.getPointerAlign()));
5482a9ac8606Spatrick     Callee.setFunctionPointer(Stub);
5483a9ac8606Spatrick   }
5484e5dd7070Spatrick   llvm::CallBase *CallOrInvoke = nullptr;
5485e5dd7070Spatrick   RValue Call = EmitCall(FnInfo, Callee, ReturnValue, Args, &CallOrInvoke,
5486a9ac8606Spatrick                          E == MustTailCall, E->getExprLoc());
5487e5dd7070Spatrick 
5488e5dd7070Spatrick   // Generate function declaration DISuprogram in order to be used
5489e5dd7070Spatrick   // in debug info about call sites.
5490e5dd7070Spatrick   if (CGDebugInfo *DI = getDebugInfo()) {
5491*12c85518Srobert     if (auto *CalleeDecl = dyn_cast_or_null<FunctionDecl>(TargetDecl)) {
5492*12c85518Srobert       FunctionArgList Args;
5493*12c85518Srobert       QualType ResTy = BuildFunctionArgList(CalleeDecl, Args);
5494*12c85518Srobert       DI->EmitFuncDeclForCallSite(CallOrInvoke,
5495*12c85518Srobert                                   DI->getFunctionType(CalleeDecl, ResTy, Args),
5496e5dd7070Spatrick                                   CalleeDecl);
5497e5dd7070Spatrick     }
5498*12c85518Srobert   }
5499e5dd7070Spatrick 
5500e5dd7070Spatrick   return Call;
5501e5dd7070Spatrick }
5502e5dd7070Spatrick 
5503e5dd7070Spatrick LValue CodeGenFunction::
EmitPointerToDataMemberBinaryExpr(const BinaryOperator * E)5504e5dd7070Spatrick EmitPointerToDataMemberBinaryExpr(const BinaryOperator *E) {
5505e5dd7070Spatrick   Address BaseAddr = Address::invalid();
5506e5dd7070Spatrick   if (E->getOpcode() == BO_PtrMemI) {
5507e5dd7070Spatrick     BaseAddr = EmitPointerWithAlignment(E->getLHS());
5508e5dd7070Spatrick   } else {
5509e5dd7070Spatrick     BaseAddr = EmitLValue(E->getLHS()).getAddress(*this);
5510e5dd7070Spatrick   }
5511e5dd7070Spatrick 
5512e5dd7070Spatrick   llvm::Value *OffsetV = EmitScalarExpr(E->getRHS());
5513e5dd7070Spatrick   const auto *MPT = E->getRHS()->getType()->castAs<MemberPointerType>();
5514e5dd7070Spatrick 
5515e5dd7070Spatrick   LValueBaseInfo BaseInfo;
5516e5dd7070Spatrick   TBAAAccessInfo TBAAInfo;
5517e5dd7070Spatrick   Address MemberAddr =
5518e5dd7070Spatrick     EmitCXXMemberDataPointerAddress(E, BaseAddr, OffsetV, MPT, &BaseInfo,
5519e5dd7070Spatrick                                     &TBAAInfo);
5520e5dd7070Spatrick 
5521e5dd7070Spatrick   return MakeAddrLValue(MemberAddr, MPT->getPointeeType(), BaseInfo, TBAAInfo);
5522e5dd7070Spatrick }
5523e5dd7070Spatrick 
5524e5dd7070Spatrick /// Given the address of a temporary variable, produce an r-value of
5525e5dd7070Spatrick /// its type.
convertTempToRValue(Address addr,QualType type,SourceLocation loc)5526e5dd7070Spatrick RValue CodeGenFunction::convertTempToRValue(Address addr,
5527e5dd7070Spatrick                                             QualType type,
5528e5dd7070Spatrick                                             SourceLocation loc) {
5529e5dd7070Spatrick   LValue lvalue = MakeAddrLValue(addr, type, AlignmentSource::Decl);
5530e5dd7070Spatrick   switch (getEvaluationKind(type)) {
5531e5dd7070Spatrick   case TEK_Complex:
5532e5dd7070Spatrick     return RValue::getComplex(EmitLoadOfComplex(lvalue, loc));
5533e5dd7070Spatrick   case TEK_Aggregate:
5534e5dd7070Spatrick     return lvalue.asAggregateRValue(*this);
5535e5dd7070Spatrick   case TEK_Scalar:
5536e5dd7070Spatrick     return RValue::get(EmitLoadOfScalar(lvalue, loc));
5537e5dd7070Spatrick   }
5538e5dd7070Spatrick   llvm_unreachable("bad evaluation kind");
5539e5dd7070Spatrick }
5540e5dd7070Spatrick 
SetFPAccuracy(llvm::Value * Val,float Accuracy)5541e5dd7070Spatrick void CodeGenFunction::SetFPAccuracy(llvm::Value *Val, float Accuracy) {
5542e5dd7070Spatrick   assert(Val->getType()->isFPOrFPVectorTy());
5543e5dd7070Spatrick   if (Accuracy == 0.0 || !isa<llvm::Instruction>(Val))
5544e5dd7070Spatrick     return;
5545e5dd7070Spatrick 
5546e5dd7070Spatrick   llvm::MDBuilder MDHelper(getLLVMContext());
5547e5dd7070Spatrick   llvm::MDNode *Node = MDHelper.createFPMath(Accuracy);
5548e5dd7070Spatrick 
5549e5dd7070Spatrick   cast<llvm::Instruction>(Val)->setMetadata(llvm::LLVMContext::MD_fpmath, Node);
5550e5dd7070Spatrick }
5551e5dd7070Spatrick 
5552e5dd7070Spatrick namespace {
5553e5dd7070Spatrick   struct LValueOrRValue {
5554e5dd7070Spatrick     LValue LV;
5555e5dd7070Spatrick     RValue RV;
5556e5dd7070Spatrick   };
5557e5dd7070Spatrick }
5558e5dd7070Spatrick 
emitPseudoObjectExpr(CodeGenFunction & CGF,const PseudoObjectExpr * E,bool forLValue,AggValueSlot slot)5559e5dd7070Spatrick static LValueOrRValue emitPseudoObjectExpr(CodeGenFunction &CGF,
5560e5dd7070Spatrick                                            const PseudoObjectExpr *E,
5561e5dd7070Spatrick                                            bool forLValue,
5562e5dd7070Spatrick                                            AggValueSlot slot) {
5563e5dd7070Spatrick   SmallVector<CodeGenFunction::OpaqueValueMappingData, 4> opaques;
5564e5dd7070Spatrick 
5565e5dd7070Spatrick   // Find the result expression, if any.
5566e5dd7070Spatrick   const Expr *resultExpr = E->getResultExpr();
5567e5dd7070Spatrick   LValueOrRValue result;
5568e5dd7070Spatrick 
5569e5dd7070Spatrick   for (PseudoObjectExpr::const_semantics_iterator
5570e5dd7070Spatrick          i = E->semantics_begin(), e = E->semantics_end(); i != e; ++i) {
5571e5dd7070Spatrick     const Expr *semantic = *i;
5572e5dd7070Spatrick 
5573e5dd7070Spatrick     // If this semantic expression is an opaque value, bind it
5574e5dd7070Spatrick     // to the result of its source expression.
5575e5dd7070Spatrick     if (const auto *ov = dyn_cast<OpaqueValueExpr>(semantic)) {
5576e5dd7070Spatrick       // Skip unique OVEs.
5577e5dd7070Spatrick       if (ov->isUnique()) {
5578e5dd7070Spatrick         assert(ov != resultExpr &&
5579e5dd7070Spatrick                "A unique OVE cannot be used as the result expression");
5580e5dd7070Spatrick         continue;
5581e5dd7070Spatrick       }
5582e5dd7070Spatrick 
5583e5dd7070Spatrick       // If this is the result expression, we may need to evaluate
5584e5dd7070Spatrick       // directly into the slot.
5585e5dd7070Spatrick       typedef CodeGenFunction::OpaqueValueMappingData OVMA;
5586e5dd7070Spatrick       OVMA opaqueData;
5587a9ac8606Spatrick       if (ov == resultExpr && ov->isPRValue() && !forLValue &&
5588e5dd7070Spatrick           CodeGenFunction::hasAggregateEvaluationKind(ov->getType())) {
5589e5dd7070Spatrick         CGF.EmitAggExpr(ov->getSourceExpr(), slot);
5590e5dd7070Spatrick         LValue LV = CGF.MakeAddrLValue(slot.getAddress(), ov->getType(),
5591e5dd7070Spatrick                                        AlignmentSource::Decl);
5592e5dd7070Spatrick         opaqueData = OVMA::bind(CGF, ov, LV);
5593e5dd7070Spatrick         result.RV = slot.asRValue();
5594e5dd7070Spatrick 
5595e5dd7070Spatrick       // Otherwise, emit as normal.
5596e5dd7070Spatrick       } else {
5597e5dd7070Spatrick         opaqueData = OVMA::bind(CGF, ov, ov->getSourceExpr());
5598e5dd7070Spatrick 
5599e5dd7070Spatrick         // If this is the result, also evaluate the result now.
5600e5dd7070Spatrick         if (ov == resultExpr) {
5601e5dd7070Spatrick           if (forLValue)
5602e5dd7070Spatrick             result.LV = CGF.EmitLValue(ov);
5603e5dd7070Spatrick           else
5604e5dd7070Spatrick             result.RV = CGF.EmitAnyExpr(ov, slot);
5605e5dd7070Spatrick         }
5606e5dd7070Spatrick       }
5607e5dd7070Spatrick 
5608e5dd7070Spatrick       opaques.push_back(opaqueData);
5609e5dd7070Spatrick 
5610e5dd7070Spatrick     // Otherwise, if the expression is the result, evaluate it
5611e5dd7070Spatrick     // and remember the result.
5612e5dd7070Spatrick     } else if (semantic == resultExpr) {
5613e5dd7070Spatrick       if (forLValue)
5614e5dd7070Spatrick         result.LV = CGF.EmitLValue(semantic);
5615e5dd7070Spatrick       else
5616e5dd7070Spatrick         result.RV = CGF.EmitAnyExpr(semantic, slot);
5617e5dd7070Spatrick 
5618e5dd7070Spatrick     // Otherwise, evaluate the expression in an ignored context.
5619e5dd7070Spatrick     } else {
5620e5dd7070Spatrick       CGF.EmitIgnoredExpr(semantic);
5621e5dd7070Spatrick     }
5622e5dd7070Spatrick   }
5623e5dd7070Spatrick 
5624e5dd7070Spatrick   // Unbind all the opaques now.
5625e5dd7070Spatrick   for (unsigned i = 0, e = opaques.size(); i != e; ++i)
5626e5dd7070Spatrick     opaques[i].unbind(CGF);
5627e5dd7070Spatrick 
5628e5dd7070Spatrick   return result;
5629e5dd7070Spatrick }
5630e5dd7070Spatrick 
EmitPseudoObjectRValue(const PseudoObjectExpr * E,AggValueSlot slot)5631e5dd7070Spatrick RValue CodeGenFunction::EmitPseudoObjectRValue(const PseudoObjectExpr *E,
5632e5dd7070Spatrick                                                AggValueSlot slot) {
5633e5dd7070Spatrick   return emitPseudoObjectExpr(*this, E, false, slot).RV;
5634e5dd7070Spatrick }
5635e5dd7070Spatrick 
EmitPseudoObjectLValue(const PseudoObjectExpr * E)5636e5dd7070Spatrick LValue CodeGenFunction::EmitPseudoObjectLValue(const PseudoObjectExpr *E) {
5637e5dd7070Spatrick   return emitPseudoObjectExpr(*this, E, true, AggValueSlot::ignored()).LV;
5638e5dd7070Spatrick }
5639