17a51313dSChris Lattner //===--- CGExprAgg.cpp - Emit LLVM Code from Aggregate Expressions --------===//
27a51313dSChris Lattner //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
67a51313dSChris Lattner //
77a51313dSChris Lattner //===----------------------------------------------------------------------===//
87a51313dSChris Lattner //
97a51313dSChris Lattner // This contains code to emit Aggregate Expr nodes as LLVM code.
107a51313dSChris Lattner //
117a51313dSChris Lattner //===----------------------------------------------------------------------===//
127a51313dSChris Lattner 
130683c0e6SEric Fiselier #include "CGCXXABI.h"
145f21d2f6SFariborz Jahanian #include "CGObjCRuntime.h"
159803178aSReid Kleckner #include "CodeGenFunction.h"
163a02247dSChandler Carruth #include "CodeGenModule.h"
17e0ef348cSIvan A. Kosarev #include "ConstantEmitter.h"
185be9b8cbSMichael Liao #include "TargetInfo.h"
19ad319a73SDaniel Dunbar #include "clang/AST/ASTContext.h"
209803178aSReid Kleckner #include "clang/AST/Attr.h"
21b7f8f594SAnders Carlsson #include "clang/AST/DeclCXX.h"
22c83ed824SSebastian Redl #include "clang/AST/DeclTemplate.h"
23ad319a73SDaniel Dunbar #include "clang/AST/StmtVisitor.h"
24ffd5551bSChandler Carruth #include "llvm/IR/Constants.h"
25ffd5551bSChandler Carruth #include "llvm/IR/Function.h"
26ffd5551bSChandler Carruth #include "llvm/IR/GlobalVariable.h"
274deb75d2SGeorge Burgess IV #include "llvm/IR/IntrinsicInst.h"
289803178aSReid Kleckner #include "llvm/IR/Intrinsics.h"
297a51313dSChris Lattner using namespace clang;
307a51313dSChris Lattner using namespace CodeGen;
317a51313dSChris Lattner 
327a51313dSChris Lattner //===----------------------------------------------------------------------===//
337a51313dSChris Lattner //                        Aggregate Expression Emitter
347a51313dSChris Lattner //===----------------------------------------------------------------------===//
357a51313dSChris Lattner 
367a51313dSChris Lattner namespace  {
37337e3a5fSBenjamin Kramer class AggExprEmitter : public StmtVisitor<AggExprEmitter> {
387a51313dSChris Lattner   CodeGenFunction &CGF;
39cb463859SDaniel Dunbar   CGBuilderTy &Builder;
407a626f63SJohn McCall   AggValueSlot Dest;
416aab1117SLeny Kholodov   bool IsResultUnused;
4278a15113SJohn McCall 
437a626f63SJohn McCall   AggValueSlot EnsureSlot(QualType T) {
447a626f63SJohn McCall     if (!Dest.isIgnored()) return Dest;
457a626f63SJohn McCall     return CGF.CreateAggTemp(T, "agg.tmp.ensured");
4678a15113SJohn McCall   }
474e8ca4faSJohn McCall   void EnsureDest(QualType T) {
484e8ca4faSJohn McCall     if (!Dest.isIgnored()) return;
494e8ca4faSJohn McCall     Dest = CGF.CreateAggTemp(T, "agg.tmp.ensured");
504e8ca4faSJohn McCall   }
51cc04e9f6SJohn McCall 
5256e5a2e1SGeorge Burgess IV   // Calls `Fn` with a valid return value slot, potentially creating a temporary
5356e5a2e1SGeorge Burgess IV   // to do so. If a temporary is created, an appropriate copy into `Dest` will
544deb75d2SGeorge Burgess IV   // be emitted, as will lifetime markers.
5556e5a2e1SGeorge Burgess IV   //
5656e5a2e1SGeorge Burgess IV   // The given function should take a ReturnValueSlot, and return an RValue that
5756e5a2e1SGeorge Burgess IV   // points to said slot.
5856e5a2e1SGeorge Burgess IV   void withReturnValueSlot(const Expr *E,
5956e5a2e1SGeorge Burgess IV                            llvm::function_ref<RValue(ReturnValueSlot)> Fn);
6056e5a2e1SGeorge Burgess IV 
617a51313dSChris Lattner public:
626aab1117SLeny Kholodov   AggExprEmitter(CodeGenFunction &cgf, AggValueSlot Dest, bool IsResultUnused)
636aab1117SLeny Kholodov     : CGF(cgf), Builder(CGF.Builder), Dest(Dest),
646aab1117SLeny Kholodov     IsResultUnused(IsResultUnused) { }
657a51313dSChris Lattner 
667a51313dSChris Lattner   //===--------------------------------------------------------------------===//
677a51313dSChris Lattner   //                               Utilities
687a51313dSChris Lattner   //===--------------------------------------------------------------------===//
697a51313dSChris Lattner 
707a51313dSChris Lattner   /// EmitAggLoadOfLValue - Given an expression with aggregate type that
717a51313dSChris Lattner   /// represents a value lvalue, this method emits the address of the lvalue,
727a51313dSChris Lattner   /// then loads the result into DestPtr.
737a51313dSChris Lattner   void EmitAggLoadOfLValue(const Expr *E);
747a51313dSChris Lattner 
757275da0fSAkira Hatanaka   enum ExprValueKind {
767275da0fSAkira Hatanaka     EVK_RValue,
777275da0fSAkira Hatanaka     EVK_NonRValue
787275da0fSAkira Hatanaka   };
797275da0fSAkira Hatanaka 
80ca9fc09cSMike Stump   /// EmitFinalDestCopy - Perform the final copy to DestPtr, if desired.
817275da0fSAkira Hatanaka   /// SrcIsRValue is true if source comes from an RValue.
827275da0fSAkira Hatanaka   void EmitFinalDestCopy(QualType type, const LValue &src,
837275da0fSAkira Hatanaka                          ExprValueKind SrcValueKind = EVK_NonRValue);
847f416cc4SJohn McCall   void EmitFinalDestCopy(QualType type, RValue src);
854e8ca4faSJohn McCall   void EmitCopy(QualType type, const AggValueSlot &dest,
864e8ca4faSJohn McCall                 const AggValueSlot &src);
87ca9fc09cSMike Stump 
88a5efa738SJohn McCall   void EmitMoveFromReturnSlot(const Expr *E, RValue Src);
89cc04e9f6SJohn McCall 
907f416cc4SJohn McCall   void EmitArrayInit(Address DestPtr, llvm::ArrayType *AType,
91e0ef348cSIvan A. Kosarev                      QualType ArrayQTy, InitListExpr *E);
92c83ed824SSebastian Redl 
938d6fc958SJohn McCall   AggValueSlot::NeedsGCBarriers_t needsGC(QualType T) {
94bbafb8a7SDavid Blaikie     if (CGF.getLangOpts().getGC() && TypeRequiresGCollection(T))
958d6fc958SJohn McCall       return AggValueSlot::NeedsGCBarriers;
968d6fc958SJohn McCall     return AggValueSlot::DoesNotNeedGCBarriers;
978d6fc958SJohn McCall   }
988d6fc958SJohn McCall 
99cc04e9f6SJohn McCall   bool TypeRequiresGCollection(QualType T);
100cc04e9f6SJohn McCall 
1017a51313dSChris Lattner   //===--------------------------------------------------------------------===//
1027a51313dSChris Lattner   //                            Visitor Methods
1037a51313dSChris Lattner   //===--------------------------------------------------------------------===//
1047a51313dSChris Lattner 
10501fb5fb1SDavid Blaikie   void Visit(Expr *E) {
1069b479666SDavid Blaikie     ApplyDebugLocation DL(CGF, E);
10701fb5fb1SDavid Blaikie     StmtVisitor<AggExprEmitter>::Visit(E);
10801fb5fb1SDavid Blaikie   }
10901fb5fb1SDavid Blaikie 
1107a51313dSChris Lattner   void VisitStmt(Stmt *S) {
111a7c8cf62SDaniel Dunbar     CGF.ErrorUnsupported(S, "aggregate expression");
1127a51313dSChris Lattner   }
1137a51313dSChris Lattner   void VisitParenExpr(ParenExpr *PE) { Visit(PE->getSubExpr()); }
11491147596SPeter Collingbourne   void VisitGenericSelectionExpr(GenericSelectionExpr *GE) {
11591147596SPeter Collingbourne     Visit(GE->getResultExpr());
11691147596SPeter Collingbourne   }
1175eb58583SGor Nishanov   void VisitCoawaitExpr(CoawaitExpr *E) {
1185eb58583SGor Nishanov     CGF.EmitCoawaitExpr(*E, Dest, IsResultUnused);
1195eb58583SGor Nishanov   }
1205eb58583SGor Nishanov   void VisitCoyieldExpr(CoyieldExpr *E) {
1215eb58583SGor Nishanov     CGF.EmitCoyieldExpr(*E, Dest, IsResultUnused);
1225eb58583SGor Nishanov   }
1235eb58583SGor Nishanov   void VisitUnaryCoawait(UnaryOperator *E) { Visit(E->getSubExpr()); }
1243f66b84cSEli Friedman   void VisitUnaryExtension(UnaryOperator *E) { Visit(E->getSubExpr()); }
1257c454bb8SJohn McCall   void VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *E) {
1267c454bb8SJohn McCall     return Visit(E->getReplacement());
1277c454bb8SJohn McCall   }
1287a51313dSChris Lattner 
1298003edc9SBill Wendling   void VisitConstantExpr(ConstantExpr *E) {
13051e4aa87STyker     if (llvm::Value *Result = ConstantEmitter(CGF).tryEmitConstantExpr(E)) {
13151e4aa87STyker       CGF.EmitAggregateStore(Result, Dest.getAddress(),
13251e4aa87STyker                              E->getType().isVolatileQualified());
13351e4aa87STyker       return;
13451e4aa87STyker     }
1358003edc9SBill Wendling     return Visit(E->getSubExpr());
1368003edc9SBill Wendling   }
1378003edc9SBill Wendling 
1387a51313dSChris Lattner   // l-values.
1396cc8317cSAlex Lorenz   void VisitDeclRefExpr(DeclRefExpr *E) { EmitAggLoadOfLValue(E); }
1407a51313dSChris Lattner   void VisitMemberExpr(MemberExpr *ME) { EmitAggLoadOfLValue(ME); }
1417a51313dSChris Lattner   void VisitUnaryDeref(UnaryOperator *E) { EmitAggLoadOfLValue(E); }
142d443c0a0SDaniel Dunbar   void VisitStringLiteral(StringLiteral *E) { EmitAggLoadOfLValue(E); }
1439b71f0cfSDouglas Gregor   void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
1447a51313dSChris Lattner   void VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
1457a51313dSChris Lattner     EmitAggLoadOfLValue(E);
1467a51313dSChris Lattner   }
1472f343dd5SChris Lattner   void VisitPredefinedExpr(const PredefinedExpr *E) {
1482f343dd5SChris Lattner     EmitAggLoadOfLValue(E);
1492f343dd5SChris Lattner   }
150bc7d67ceSMike Stump 
1517a51313dSChris Lattner   // Operators.
152ec143777SAnders Carlsson   void VisitCastExpr(CastExpr *E);
1537a51313dSChris Lattner   void VisitCallExpr(const CallExpr *E);
1547a51313dSChris Lattner   void VisitStmtExpr(const StmtExpr *E);
1557a51313dSChris Lattner   void VisitBinaryOperator(const BinaryOperator *BO);
156ffba662dSFariborz Jahanian   void VisitPointerToDataMemberBinaryOperator(const BinaryOperator *BO);
1577a51313dSChris Lattner   void VisitBinAssign(const BinaryOperator *E);
1584b0e2a30SEli Friedman   void VisitBinComma(const BinaryOperator *E);
1590683c0e6SEric Fiselier   void VisitBinCmp(const BinaryOperator *E);
160778dc0f1SRichard Smith   void VisitCXXRewrittenBinaryOperator(CXXRewrittenBinaryOperator *E) {
161778dc0f1SRichard Smith     Visit(E->getSemanticForm());
162778dc0f1SRichard Smith   }
1637a51313dSChris Lattner 
164b1d329daSChris Lattner   void VisitObjCMessageExpr(ObjCMessageExpr *E);
165c8317a44SDaniel Dunbar   void VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
166c8317a44SDaniel Dunbar     EmitAggLoadOfLValue(E);
167c8317a44SDaniel Dunbar   }
1687a51313dSChris Lattner 
169cb77930dSYunzhong Gao   void VisitDesignatedInitUpdateExpr(DesignatedInitUpdateExpr *E);
170c07a0c7eSJohn McCall   void VisitAbstractConditionalOperator(const AbstractConditionalOperator *CO);
1715b2095ceSAnders Carlsson   void VisitChooseExpr(const ChooseExpr *CE);
1727a51313dSChris Lattner   void VisitInitListExpr(InitListExpr *E);
173939b6880SRichard Smith   void VisitArrayInitLoopExpr(const ArrayInitLoopExpr *E,
174939b6880SRichard Smith                               llvm::Value *outerBegin = nullptr);
17518ada985SAnders Carlsson   void VisitImplicitValueInitExpr(ImplicitValueInitExpr *E);
176cb77930dSYunzhong Gao   void VisitNoInitExpr(NoInitExpr *E) { } // Do nothing.
177aa9c7aedSChris Lattner   void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
178708afb56SEric Fiselier     CodeGenFunction::CXXDefaultArgExprScope Scope(CGF, DAE);
179aa9c7aedSChris Lattner     Visit(DAE->getExpr());
180aa9c7aedSChris Lattner   }
181852c9db7SRichard Smith   void VisitCXXDefaultInitExpr(CXXDefaultInitExpr *DIE) {
182708afb56SEric Fiselier     CodeGenFunction::CXXDefaultInitExprScope Scope(CGF, DIE);
183852c9db7SRichard Smith     Visit(DIE->getExpr());
184852c9db7SRichard Smith   }
1853be22e27SAnders Carlsson   void VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E);
1861619a504SAnders Carlsson   void VisitCXXConstructExpr(const CXXConstructExpr *E);
1875179eb78SRichard Smith   void VisitCXXInheritedCtorInitExpr(const CXXInheritedCtorInitExpr *E);
188c370a7eeSEli Friedman   void VisitLambdaExpr(LambdaExpr *E);
189cc1b96d3SRichard Smith   void VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr *E);
1905d413781SJohn McCall   void VisitExprWithCleanups(ExprWithCleanups *E);
191747eb784SDouglas Gregor   void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
1925bbbb137SMike Stump   void VisitCXXTypeidExpr(CXXTypeidExpr *E) { EmitAggLoadOfLValue(E); }
193fe31481fSDouglas Gregor   void VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E);
1941bf5846aSJohn McCall   void VisitOpaqueValueExpr(OpaqueValueExpr *E);
1951bf5846aSJohn McCall 
196fe96e0b6SJohn McCall   void VisitPseudoObjectExpr(PseudoObjectExpr *E) {
197fe96e0b6SJohn McCall     if (E->isGLValue()) {
198fe96e0b6SJohn McCall       LValue LV = CGF.EmitPseudoObjectLValue(E);
1994e8ca4faSJohn McCall       return EmitFinalDestCopy(E->getType(), LV);
200fe96e0b6SJohn McCall     }
201fe96e0b6SJohn McCall 
202fe96e0b6SJohn McCall     CGF.EmitPseudoObjectRValue(E, EnsureSlot(E->getType()));
203fe96e0b6SJohn McCall   }
204fe96e0b6SJohn McCall 
20521911e89SEli Friedman   void VisitVAArgExpr(VAArgExpr *E);
206579a05d7SChris Lattner 
207615ed1a3SChad Rosier   void EmitInitializationToLValue(Expr *E, LValue Address);
2081553b190SJohn McCall   void EmitNullInitializationToLValue(LValue Address);
2097a51313dSChris Lattner   //  case Expr::ChooseExprClass:
210f16b8c30SMike Stump   void VisitCXXThrowExpr(const CXXThrowExpr *E) { CGF.EmitCXXThrowExpr(E); }
211df14b3a8SEli Friedman   void VisitAtomicExpr(AtomicExpr *E) {
212cc2a6e06STim Northover     RValue Res = CGF.EmitAtomicExpr(E);
213cc2a6e06STim Northover     EmitFinalDestCopy(E->getType(), Res);
214df14b3a8SEli Friedman   }
2157a51313dSChris Lattner };
2167a51313dSChris Lattner }  // end anonymous namespace.
2177a51313dSChris Lattner 
2187a51313dSChris Lattner //===----------------------------------------------------------------------===//
2197a51313dSChris Lattner //                                Utilities
2207a51313dSChris Lattner //===----------------------------------------------------------------------===//
2217a51313dSChris Lattner 
2227a51313dSChris Lattner /// EmitAggLoadOfLValue - Given an expression with aggregate type that
2237a51313dSChris Lattner /// represents a value lvalue, this method emits the address of the lvalue,
2247a51313dSChris Lattner /// then loads the result into DestPtr.
2257a51313dSChris Lattner void AggExprEmitter::EmitAggLoadOfLValue(const Expr *E) {
2267a51313dSChris Lattner   LValue LV = CGF.EmitLValue(E);
227a8ec7eb9SJohn McCall 
228a8ec7eb9SJohn McCall   // If the type of the l-value is atomic, then do an atomic load.
229a5b195a1SDavid Majnemer   if (LV.getType()->isAtomicType() || CGF.LValueIsSuitableForInlineAtomic(LV)) {
2302d84e842SNick Lewycky     CGF.EmitAtomicLoad(LV, E->getExprLoc(), Dest);
231a8ec7eb9SJohn McCall     return;
232a8ec7eb9SJohn McCall   }
233a8ec7eb9SJohn McCall 
2344e8ca4faSJohn McCall   EmitFinalDestCopy(E->getType(), LV);
235ca9fc09cSMike Stump }
236ca9fc09cSMike Stump 
2379fc8faf9SAdrian Prantl /// True if the given aggregate type requires special GC API calls.
238cc04e9f6SJohn McCall bool AggExprEmitter::TypeRequiresGCollection(QualType T) {
239cc04e9f6SJohn McCall   // Only record types have members that might require garbage collection.
240cc04e9f6SJohn McCall   const RecordType *RecordTy = T->getAs<RecordType>();
241cc04e9f6SJohn McCall   if (!RecordTy) return false;
242cc04e9f6SJohn McCall 
243cc04e9f6SJohn McCall   // Don't mess with non-trivial C++ types.
244cc04e9f6SJohn McCall   RecordDecl *Record = RecordTy->getDecl();
245cc04e9f6SJohn McCall   if (isa<CXXRecordDecl>(Record) &&
24616488472SRichard Smith       (cast<CXXRecordDecl>(Record)->hasNonTrivialCopyConstructor() ||
247cc04e9f6SJohn McCall        !cast<CXXRecordDecl>(Record)->hasTrivialDestructor()))
248cc04e9f6SJohn McCall     return false;
249cc04e9f6SJohn McCall 
250cc04e9f6SJohn McCall   // Check whether the type has an object member.
251cc04e9f6SJohn McCall   return Record->hasObjectMember();
252cc04e9f6SJohn McCall }
253cc04e9f6SJohn McCall 
25456e5a2e1SGeorge Burgess IV void AggExprEmitter::withReturnValueSlot(
25556e5a2e1SGeorge Burgess IV     const Expr *E, llvm::function_ref<RValue(ReturnValueSlot)> EmitCall) {
25656e5a2e1SGeorge Burgess IV   QualType RetTy = E->getType();
25756e5a2e1SGeorge Burgess IV   bool RequiresDestruction =
258d35a4541SAkira Hatanaka       !Dest.isExternallyDestructed() &&
25956e5a2e1SGeorge Burgess IV       RetTy.isDestructedType() == QualType::DK_nontrivial_c_struct;
2607275da0fSAkira Hatanaka 
26156e5a2e1SGeorge Burgess IV   // If it makes no observable difference, save a memcpy + temporary.
26256e5a2e1SGeorge Burgess IV   //
26356e5a2e1SGeorge Burgess IV   // We need to always provide our own temporary if destruction is required.
26456e5a2e1SGeorge Burgess IV   // Otherwise, EmitCall will emit its own, notice that it's "unused", and end
26556e5a2e1SGeorge Burgess IV   // its lifetime before we have the chance to emit a proper destructor call.
26656e5a2e1SGeorge Burgess IV   bool UseTemp = Dest.isPotentiallyAliased() || Dest.requiresGCollection() ||
26756e5a2e1SGeorge Burgess IV                  (RequiresDestruction && !Dest.getAddress().isValid());
26856e5a2e1SGeorge Burgess IV 
26956e5a2e1SGeorge Burgess IV   Address RetAddr = Address::invalid();
270a2a9cfabSYaxun Liu   Address RetAllocaAddr = Address::invalid();
2714deb75d2SGeorge Burgess IV 
2724deb75d2SGeorge Burgess IV   EHScopeStack::stable_iterator LifetimeEndBlock;
2734deb75d2SGeorge Burgess IV   llvm::Value *LifetimeSizePtr = nullptr;
2744deb75d2SGeorge Burgess IV   llvm::IntrinsicInst *LifetimeStartInst = nullptr;
27556e5a2e1SGeorge Burgess IV   if (!UseTemp) {
27656e5a2e1SGeorge Burgess IV     RetAddr = Dest.getAddress();
27756e5a2e1SGeorge Burgess IV   } else {
278a2a9cfabSYaxun Liu     RetAddr = CGF.CreateMemTemp(RetTy, "tmp", &RetAllocaAddr);
27956e5a2e1SGeorge Burgess IV     uint64_t Size =
28056e5a2e1SGeorge Burgess IV         CGF.CGM.getDataLayout().getTypeAllocSize(CGF.ConvertTypeForMem(RetTy));
281a2a9cfabSYaxun Liu     LifetimeSizePtr = CGF.EmitLifetimeStart(Size, RetAllocaAddr.getPointer());
2824deb75d2SGeorge Burgess IV     if (LifetimeSizePtr) {
2834deb75d2SGeorge Burgess IV       LifetimeStartInst =
2844deb75d2SGeorge Burgess IV           cast<llvm::IntrinsicInst>(std::prev(Builder.GetInsertPoint()));
2854deb75d2SGeorge Burgess IV       assert(LifetimeStartInst->getIntrinsicID() ==
2864deb75d2SGeorge Burgess IV                  llvm::Intrinsic::lifetime_start &&
2874deb75d2SGeorge Burgess IV              "Last insertion wasn't a lifetime.start?");
2884deb75d2SGeorge Burgess IV 
28956e5a2e1SGeorge Burgess IV       CGF.pushFullExprCleanup<CodeGenFunction::CallLifetimeEnd>(
290a2a9cfabSYaxun Liu           NormalEHLifetimeMarker, RetAllocaAddr, LifetimeSizePtr);
2914deb75d2SGeorge Burgess IV       LifetimeEndBlock = CGF.EHStack.stable_begin();
2924deb75d2SGeorge Burgess IV     }
293021510e9SFariborz Jahanian   }
294a5efa738SJohn McCall 
29556e5a2e1SGeorge Burgess IV   RValue Src =
296d35a4541SAkira Hatanaka       EmitCall(ReturnValueSlot(RetAddr, Dest.isVolatile(), IsResultUnused,
297d35a4541SAkira Hatanaka                                Dest.isExternallyDestructed()));
29856e5a2e1SGeorge Burgess IV 
2994deb75d2SGeorge Burgess IV   if (!UseTemp)
3004deb75d2SGeorge Burgess IV     return;
3014deb75d2SGeorge Burgess IV 
30256e5a2e1SGeorge Burgess IV   assert(Dest.getPointer() != Src.getAggregatePointer());
30356e5a2e1SGeorge Burgess IV   EmitFinalDestCopy(E->getType(), Src);
3044deb75d2SGeorge Burgess IV 
3054deb75d2SGeorge Burgess IV   if (!RequiresDestruction && LifetimeStartInst) {
3064deb75d2SGeorge Burgess IV     // If there's no dtor to run, the copy was the last use of our temporary.
3074deb75d2SGeorge Burgess IV     // Since we're not guaranteed to be in an ExprWithCleanups, clean up
3084deb75d2SGeorge Burgess IV     // eagerly.
3094deb75d2SGeorge Burgess IV     CGF.DeactivateCleanupBlock(LifetimeEndBlock, LifetimeStartInst);
310a2a9cfabSYaxun Liu     CGF.EmitLifetimeEnd(LifetimeSizePtr, RetAllocaAddr.getPointer());
31156e5a2e1SGeorge Burgess IV   }
312cc04e9f6SJohn McCall }
313cc04e9f6SJohn McCall 
314ca9fc09cSMike Stump /// EmitFinalDestCopy - Perform the final copy to DestPtr, if desired.
3157f416cc4SJohn McCall void AggExprEmitter::EmitFinalDestCopy(QualType type, RValue src) {
3164e8ca4faSJohn McCall   assert(src.isAggregate() && "value must be aggregate value!");
3177f416cc4SJohn McCall   LValue srcLV = CGF.MakeAddrLValue(src.getAggregateAddress(), type);
3187275da0fSAkira Hatanaka   EmitFinalDestCopy(type, srcLV, EVK_RValue);
3194e8ca4faSJohn McCall }
3207a51313dSChris Lattner 
3214e8ca4faSJohn McCall /// EmitFinalDestCopy - Perform the final copy to DestPtr, if desired.
3227275da0fSAkira Hatanaka void AggExprEmitter::EmitFinalDestCopy(QualType type, const LValue &src,
3237275da0fSAkira Hatanaka                                        ExprValueKind SrcValueKind) {
3247a626f63SJohn McCall   // If Dest is ignored, then we're evaluating an aggregate expression
3254e8ca4faSJohn McCall   // in a context that doesn't care about the result.  Note that loads
3264e8ca4faSJohn McCall   // from volatile l-values force the existence of a non-ignored
3274e8ca4faSJohn McCall   // destination.
3284e8ca4faSJohn McCall   if (Dest.isIgnored())
329ec3cbfe8SMike Stump     return;
330c123623dSFariborz Jahanian 
3317275da0fSAkira Hatanaka   // Copy non-trivial C structs here.
3327275da0fSAkira Hatanaka   LValue DstLV = CGF.MakeAddrLValue(
3337275da0fSAkira Hatanaka       Dest.getAddress(), Dest.isVolatile() ? type.withVolatile() : type);
3347275da0fSAkira Hatanaka 
3357275da0fSAkira Hatanaka   if (SrcValueKind == EVK_RValue) {
3367275da0fSAkira Hatanaka     if (type.isNonTrivialToPrimitiveDestructiveMove() == QualType::PCK_Struct) {
3377275da0fSAkira Hatanaka       if (Dest.isPotentiallyAliased())
3387275da0fSAkira Hatanaka         CGF.callCStructMoveAssignmentOperator(DstLV, src);
3397275da0fSAkira Hatanaka       else
3407275da0fSAkira Hatanaka         CGF.callCStructMoveConstructor(DstLV, src);
3417275da0fSAkira Hatanaka       return;
3427275da0fSAkira Hatanaka     }
3437275da0fSAkira Hatanaka   } else {
3447275da0fSAkira Hatanaka     if (type.isNonTrivialToPrimitiveCopy() == QualType::PCK_Struct) {
3457275da0fSAkira Hatanaka       if (Dest.isPotentiallyAliased())
3467275da0fSAkira Hatanaka         CGF.callCStructCopyAssignmentOperator(DstLV, src);
3477275da0fSAkira Hatanaka       else
3487275da0fSAkira Hatanaka         CGF.callCStructCopyConstructor(DstLV, src);
3497275da0fSAkira Hatanaka       return;
3507275da0fSAkira Hatanaka     }
3517275da0fSAkira Hatanaka   }
3527275da0fSAkira Hatanaka 
353f139ae3dSAkira Hatanaka   AggValueSlot srcAgg = AggValueSlot::forLValue(
354f139ae3dSAkira Hatanaka       src, CGF, AggValueSlot::IsDestructed, needsGC(type),
355f139ae3dSAkira Hatanaka       AggValueSlot::IsAliased, AggValueSlot::MayOverlap);
3564e8ca4faSJohn McCall   EmitCopy(type, Dest, srcAgg);
357332ec2ceSMike Stump }
3587a51313dSChris Lattner 
3594e8ca4faSJohn McCall /// Perform a copy from the source into the destination.
3604e8ca4faSJohn McCall ///
3614e8ca4faSJohn McCall /// \param type - the type of the aggregate being copied; qualifiers are
3624e8ca4faSJohn McCall ///   ignored
3634e8ca4faSJohn McCall void AggExprEmitter::EmitCopy(QualType type, const AggValueSlot &dest,
3644e8ca4faSJohn McCall                               const AggValueSlot &src) {
3654e8ca4faSJohn McCall   if (dest.requiresGCollection()) {
366e78fac51SRichard Smith     CharUnits sz = dest.getPreferredSize(CGF.getContext(), type);
3674e8ca4faSJohn McCall     llvm::Value *size = llvm::ConstantInt::get(CGF.SizeTy, sz.getQuantity());
368879d7266SFariborz Jahanian     CGF.CGM.getObjCRuntime().EmitGCMemmoveCollectable(CGF,
3697f416cc4SJohn McCall                                                       dest.getAddress(),
3707f416cc4SJohn McCall                                                       src.getAddress(),
3714e8ca4faSJohn McCall                                                       size);
372879d7266SFariborz Jahanian     return;
373879d7266SFariborz Jahanian   }
3744e8ca4faSJohn McCall 
375ca9fc09cSMike Stump   // If the result of the assignment is used, copy the LHS there also.
3764e8ca4faSJohn McCall   // It's volatile if either side is.  Use the minimum alignment of
3774e8ca4faSJohn McCall   // the two sides.
3781860b520SIvan A. Kosarev   LValue DestLV = CGF.MakeAddrLValue(dest.getAddress(), type);
3791860b520SIvan A. Kosarev   LValue SrcLV = CGF.MakeAddrLValue(src.getAddress(), type);
380e78fac51SRichard Smith   CGF.EmitAggregateCopy(DestLV, SrcLV, type, dest.mayOverlap(),
3817f416cc4SJohn McCall                         dest.isVolatile() || src.isVolatile());
3827a51313dSChris Lattner }
3837a51313dSChris Lattner 
3849fc8faf9SAdrian Prantl /// Emit the initializer for a std::initializer_list initialized with a
385c83ed824SSebastian Redl /// real initializer list.
386cc1b96d3SRichard Smith void
387cc1b96d3SRichard Smith AggExprEmitter::VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr *E) {
388cc1b96d3SRichard Smith   // Emit an array containing the elements.  The array is externally destructed
389cc1b96d3SRichard Smith   // if the std::initializer_list object is.
390cc1b96d3SRichard Smith   ASTContext &Ctx = CGF.getContext();
391cc1b96d3SRichard Smith   LValue Array = CGF.EmitLValue(E->getSubExpr());
392cc1b96d3SRichard Smith   assert(Array.isSimple() && "initializer_list array not a simple lvalue");
393f139ae3dSAkira Hatanaka   Address ArrayPtr = Array.getAddress(CGF);
394c83ed824SSebastian Redl 
395cc1b96d3SRichard Smith   const ConstantArrayType *ArrayType =
396cc1b96d3SRichard Smith       Ctx.getAsConstantArrayType(E->getSubExpr()->getType());
397cc1b96d3SRichard Smith   assert(ArrayType && "std::initializer_list constructed from non-array");
398c83ed824SSebastian Redl 
399cc1b96d3SRichard Smith   // FIXME: Perform the checks on the field types in SemaInit.
400cc1b96d3SRichard Smith   RecordDecl *Record = E->getType()->castAs<RecordType>()->getDecl();
401cc1b96d3SRichard Smith   RecordDecl::field_iterator Field = Record->field_begin();
402cc1b96d3SRichard Smith   if (Field == Record->field_end()) {
403cc1b96d3SRichard Smith     CGF.ErrorUnsupported(E, "weird std::initializer_list");
404f2e0a307SSebastian Redl     return;
405c83ed824SSebastian Redl   }
406c83ed824SSebastian Redl 
407c83ed824SSebastian Redl   // Start pointer.
408cc1b96d3SRichard Smith   if (!Field->getType()->isPointerType() ||
409cc1b96d3SRichard Smith       !Ctx.hasSameType(Field->getType()->getPointeeType(),
410cc1b96d3SRichard Smith                        ArrayType->getElementType())) {
411cc1b96d3SRichard Smith     CGF.ErrorUnsupported(E, "weird std::initializer_list");
412f2e0a307SSebastian Redl     return;
413c83ed824SSebastian Redl   }
414c83ed824SSebastian Redl 
415cc1b96d3SRichard Smith   AggValueSlot Dest = EnsureSlot(E->getType());
4167f416cc4SJohn McCall   LValue DestLV = CGF.MakeAddrLValue(Dest.getAddress(), E->getType());
417cc1b96d3SRichard Smith   LValue Start = CGF.EmitLValueForFieldInitialization(DestLV, *Field);
418cc1b96d3SRichard Smith   llvm::Value *Zero = llvm::ConstantInt::get(CGF.PtrDiffTy, 0);
419cc1b96d3SRichard Smith   llvm::Value *IdxStart[] = { Zero, Zero };
420cc1b96d3SRichard Smith   llvm::Value *ArrayStart =
4217f416cc4SJohn McCall       Builder.CreateInBoundsGEP(ArrayPtr.getPointer(), IdxStart, "arraystart");
422cc1b96d3SRichard Smith   CGF.EmitStoreThroughLValue(RValue::get(ArrayStart), Start);
423cc1b96d3SRichard Smith   ++Field;
424cc1b96d3SRichard Smith 
425cc1b96d3SRichard Smith   if (Field == Record->field_end()) {
426cc1b96d3SRichard Smith     CGF.ErrorUnsupported(E, "weird std::initializer_list");
427f2e0a307SSebastian Redl     return;
428c83ed824SSebastian Redl   }
429cc1b96d3SRichard Smith 
430cc1b96d3SRichard Smith   llvm::Value *Size = Builder.getInt(ArrayType->getSize());
431cc1b96d3SRichard Smith   LValue EndOrLength = CGF.EmitLValueForFieldInitialization(DestLV, *Field);
432cc1b96d3SRichard Smith   if (Field->getType()->isPointerType() &&
433cc1b96d3SRichard Smith       Ctx.hasSameType(Field->getType()->getPointeeType(),
434cc1b96d3SRichard Smith                       ArrayType->getElementType())) {
435c83ed824SSebastian Redl     // End pointer.
436cc1b96d3SRichard Smith     llvm::Value *IdxEnd[] = { Zero, Size };
437cc1b96d3SRichard Smith     llvm::Value *ArrayEnd =
4387f416cc4SJohn McCall         Builder.CreateInBoundsGEP(ArrayPtr.getPointer(), IdxEnd, "arrayend");
439cc1b96d3SRichard Smith     CGF.EmitStoreThroughLValue(RValue::get(ArrayEnd), EndOrLength);
440cc1b96d3SRichard Smith   } else if (Ctx.hasSameType(Field->getType(), Ctx.getSizeType())) {
441c83ed824SSebastian Redl     // Length.
442cc1b96d3SRichard Smith     CGF.EmitStoreThroughLValue(RValue::get(Size), EndOrLength);
443c83ed824SSebastian Redl   } else {
444cc1b96d3SRichard Smith     CGF.ErrorUnsupported(E, "weird std::initializer_list");
445f2e0a307SSebastian Redl     return;
446c83ed824SSebastian Redl   }
447c83ed824SSebastian Redl }
448c83ed824SSebastian Redl 
4499fc8faf9SAdrian Prantl /// Determine if E is a trivial array filler, that is, one that is
4508edda962SRichard Smith /// equivalent to zero-initialization.
4518edda962SRichard Smith static bool isTrivialFiller(Expr *E) {
4528edda962SRichard Smith   if (!E)
4538edda962SRichard Smith     return true;
4548edda962SRichard Smith 
4558edda962SRichard Smith   if (isa<ImplicitValueInitExpr>(E))
4568edda962SRichard Smith     return true;
4578edda962SRichard Smith 
4588edda962SRichard Smith   if (auto *ILE = dyn_cast<InitListExpr>(E)) {
4598edda962SRichard Smith     if (ILE->getNumInits())
4608edda962SRichard Smith       return false;
4618edda962SRichard Smith     return isTrivialFiller(ILE->getArrayFiller());
4628edda962SRichard Smith   }
4638edda962SRichard Smith 
4648edda962SRichard Smith   if (auto *Cons = dyn_cast_or_null<CXXConstructExpr>(E))
4658edda962SRichard Smith     return Cons->getConstructor()->isDefaultConstructor() &&
4668edda962SRichard Smith            Cons->getConstructor()->isTrivial();
4678edda962SRichard Smith 
4688edda962SRichard Smith   // FIXME: Are there other cases where we can avoid emitting an initializer?
4698edda962SRichard Smith   return false;
4708edda962SRichard Smith }
4718edda962SRichard Smith 
4729fc8faf9SAdrian Prantl /// Emit initialization of an array from an initializer list.
4737f416cc4SJohn McCall void AggExprEmitter::EmitArrayInit(Address DestPtr, llvm::ArrayType *AType,
474e0ef348cSIvan A. Kosarev                                    QualType ArrayQTy, InitListExpr *E) {
475c83ed824SSebastian Redl   uint64_t NumInitElements = E->getNumInits();
476c83ed824SSebastian Redl 
477c83ed824SSebastian Redl   uint64_t NumArrayElements = AType->getNumElements();
478c83ed824SSebastian Redl   assert(NumInitElements <= NumArrayElements);
479c83ed824SSebastian Redl 
480e0ef348cSIvan A. Kosarev   QualType elementType =
481e0ef348cSIvan A. Kosarev       CGF.getContext().getAsArrayType(ArrayQTy)->getElementType();
482e0ef348cSIvan A. Kosarev 
483c83ed824SSebastian Redl   // DestPtr is an array*.  Construct an elementType* by drilling
484c83ed824SSebastian Redl   // down a level.
485c83ed824SSebastian Redl   llvm::Value *zero = llvm::ConstantInt::get(CGF.SizeTy, 0);
486c83ed824SSebastian Redl   llvm::Value *indices[] = { zero, zero };
487c83ed824SSebastian Redl   llvm::Value *begin =
4887f416cc4SJohn McCall     Builder.CreateInBoundsGEP(DestPtr.getPointer(), indices, "arrayinit.begin");
4897f416cc4SJohn McCall 
4907f416cc4SJohn McCall   CharUnits elementSize = CGF.getContext().getTypeSizeInChars(elementType);
4917f416cc4SJohn McCall   CharUnits elementAlign =
4927f416cc4SJohn McCall     DestPtr.getAlignment().alignmentOfArrayElement(elementSize);
493c83ed824SSebastian Redl 
494e0ef348cSIvan A. Kosarev   // Consider initializing the array by copying from a global. For this to be
495e0ef348cSIvan A. Kosarev   // more efficient than per-element initialization, the size of the elements
496e0ef348cSIvan A. Kosarev   // with explicit initializers should be large enough.
497e0ef348cSIvan A. Kosarev   if (NumInitElements * elementSize.getQuantity() > 16 &&
498e0ef348cSIvan A. Kosarev       elementType.isTriviallyCopyableType(CGF.getContext())) {
499e0ef348cSIvan A. Kosarev     CodeGen::CodeGenModule &CGM = CGF.CGM;
5001ac700cdSJohannes Altmanninger     ConstantEmitter Emitter(CGF);
501e0ef348cSIvan A. Kosarev     LangAS AS = ArrayQTy.getAddressSpace();
502e0ef348cSIvan A. Kosarev     if (llvm::Constant *C = Emitter.tryEmitForInitializer(E, AS, ArrayQTy)) {
503e0ef348cSIvan A. Kosarev       auto GV = new llvm::GlobalVariable(
504e0ef348cSIvan A. Kosarev           CGM.getModule(), C->getType(),
505e0ef348cSIvan A. Kosarev           CGM.isTypeConstant(ArrayQTy, /* ExcludeCtorDtor= */ true),
506e0ef348cSIvan A. Kosarev           llvm::GlobalValue::PrivateLinkage, C, "constinit",
507e0ef348cSIvan A. Kosarev           /* InsertBefore= */ nullptr, llvm::GlobalVariable::NotThreadLocal,
508e0ef348cSIvan A. Kosarev           CGM.getContext().getTargetAddressSpace(AS));
509e0ef348cSIvan A. Kosarev       Emitter.finalize(GV);
510e0ef348cSIvan A. Kosarev       CharUnits Align = CGM.getContext().getTypeAlignInChars(ArrayQTy);
511c79099e0SGuillaume Chatelet       GV->setAlignment(Align.getAsAlign());
512e0ef348cSIvan A. Kosarev       EmitFinalDestCopy(ArrayQTy, CGF.MakeAddrLValue(GV, ArrayQTy, Align));
513e0ef348cSIvan A. Kosarev       return;
514e0ef348cSIvan A. Kosarev     }
515e0ef348cSIvan A. Kosarev   }
516e0ef348cSIvan A. Kosarev 
517c83ed824SSebastian Redl   // Exception safety requires us to destroy all the
518c83ed824SSebastian Redl   // already-constructed members if an initializer throws.
519c83ed824SSebastian Redl   // For that, we'll need an EH cleanup.
520c83ed824SSebastian Redl   QualType::DestructionKind dtorKind = elementType.isDestructedType();
5217f416cc4SJohn McCall   Address endOfInit = Address::invalid();
522c83ed824SSebastian Redl   EHScopeStack::stable_iterator cleanup;
5238a13c418SCraig Topper   llvm::Instruction *cleanupDominator = nullptr;
524c83ed824SSebastian Redl   if (CGF.needsEHCleanup(dtorKind)) {
525c83ed824SSebastian Redl     // In principle we could tell the cleanup where we are more
526c83ed824SSebastian Redl     // directly, but the control flow can get so varied here that it
527c83ed824SSebastian Redl     // would actually be quite complex.  Therefore we go through an
528c83ed824SSebastian Redl     // alloca.
5297f416cc4SJohn McCall     endOfInit = CGF.CreateTempAlloca(begin->getType(), CGF.getPointerAlign(),
530c83ed824SSebastian Redl                                      "arrayinit.endOfInit");
531c83ed824SSebastian Redl     cleanupDominator = Builder.CreateStore(begin, endOfInit);
532c83ed824SSebastian Redl     CGF.pushIrregularPartialArrayCleanup(begin, endOfInit, elementType,
5337f416cc4SJohn McCall                                          elementAlign,
534c83ed824SSebastian Redl                                          CGF.getDestroyer(dtorKind));
535c83ed824SSebastian Redl     cleanup = CGF.EHStack.stable_begin();
536c83ed824SSebastian Redl 
537c83ed824SSebastian Redl   // Otherwise, remember that we didn't need a cleanup.
538c83ed824SSebastian Redl   } else {
539c83ed824SSebastian Redl     dtorKind = QualType::DK_none;
540c83ed824SSebastian Redl   }
541c83ed824SSebastian Redl 
542c83ed824SSebastian Redl   llvm::Value *one = llvm::ConstantInt::get(CGF.SizeTy, 1);
543c83ed824SSebastian Redl 
544c83ed824SSebastian Redl   // The 'current element to initialize'.  The invariants on this
545c83ed824SSebastian Redl   // variable are complicated.  Essentially, after each iteration of
546c83ed824SSebastian Redl   // the loop, it points to the last initialized element, except
547c83ed824SSebastian Redl   // that it points to the beginning of the array before any
548c83ed824SSebastian Redl   // elements have been initialized.
549c83ed824SSebastian Redl   llvm::Value *element = begin;
550c83ed824SSebastian Redl 
551c83ed824SSebastian Redl   // Emit the explicit initializers.
552c83ed824SSebastian Redl   for (uint64_t i = 0; i != NumInitElements; ++i) {
553c83ed824SSebastian Redl     // Advance to the next element.
554c83ed824SSebastian Redl     if (i > 0) {
555c83ed824SSebastian Redl       element = Builder.CreateInBoundsGEP(element, one, "arrayinit.element");
556c83ed824SSebastian Redl 
557c83ed824SSebastian Redl       // Tell the cleanup that it needs to destroy up to this
558c83ed824SSebastian Redl       // element.  TODO: some of these stores can be trivially
559c83ed824SSebastian Redl       // observed to be unnecessary.
5607f416cc4SJohn McCall       if (endOfInit.isValid()) Builder.CreateStore(element, endOfInit);
561c83ed824SSebastian Redl     }
562c83ed824SSebastian Redl 
5637f416cc4SJohn McCall     LValue elementLV =
5647f416cc4SJohn McCall       CGF.MakeAddrLValue(Address(element, elementAlign), elementType);
565615ed1a3SChad Rosier     EmitInitializationToLValue(E->getInit(i), elementLV);
566c83ed824SSebastian Redl   }
567c83ed824SSebastian Redl 
568c83ed824SSebastian Redl   // Check whether there's a non-trivial array-fill expression.
569c83ed824SSebastian Redl   Expr *filler = E->getArrayFiller();
5708edda962SRichard Smith   bool hasTrivialFiller = isTrivialFiller(filler);
571c83ed824SSebastian Redl 
572c83ed824SSebastian Redl   // Any remaining elements need to be zero-initialized, possibly
573c83ed824SSebastian Redl   // using the filler expression.  We can skip this if the we're
574c83ed824SSebastian Redl   // emitting to zeroed memory.
575c83ed824SSebastian Redl   if (NumInitElements != NumArrayElements &&
576c83ed824SSebastian Redl       !(Dest.isZeroed() && hasTrivialFiller &&
577c83ed824SSebastian Redl         CGF.getTypes().isZeroInitializable(elementType))) {
578c83ed824SSebastian Redl 
579c83ed824SSebastian Redl     // Use an actual loop.  This is basically
580c83ed824SSebastian Redl     //   do { *array++ = filler; } while (array != end);
581c83ed824SSebastian Redl 
582c83ed824SSebastian Redl     // Advance to the start of the rest of the array.
583c83ed824SSebastian Redl     if (NumInitElements) {
584c83ed824SSebastian Redl       element = Builder.CreateInBoundsGEP(element, one, "arrayinit.start");
5857f416cc4SJohn McCall       if (endOfInit.isValid()) Builder.CreateStore(element, endOfInit);
586c83ed824SSebastian Redl     }
587c83ed824SSebastian Redl 
588c83ed824SSebastian Redl     // Compute the end of the array.
589c83ed824SSebastian Redl     llvm::Value *end = Builder.CreateInBoundsGEP(begin,
590c83ed824SSebastian Redl                       llvm::ConstantInt::get(CGF.SizeTy, NumArrayElements),
591c83ed824SSebastian Redl                                                  "arrayinit.end");
592c83ed824SSebastian Redl 
593c83ed824SSebastian Redl     llvm::BasicBlock *entryBB = Builder.GetInsertBlock();
594c83ed824SSebastian Redl     llvm::BasicBlock *bodyBB = CGF.createBasicBlock("arrayinit.body");
595c83ed824SSebastian Redl 
596c83ed824SSebastian Redl     // Jump into the body.
597c83ed824SSebastian Redl     CGF.EmitBlock(bodyBB);
598c83ed824SSebastian Redl     llvm::PHINode *currentElement =
599c83ed824SSebastian Redl       Builder.CreatePHI(element->getType(), 2, "arrayinit.cur");
600c83ed824SSebastian Redl     currentElement->addIncoming(element, entryBB);
601c83ed824SSebastian Redl 
602c83ed824SSebastian Redl     // Emit the actual filler expression.
60372236372SRichard Smith     {
60472236372SRichard Smith       // C++1z [class.temporary]p5:
60572236372SRichard Smith       //   when a default constructor is called to initialize an element of
60672236372SRichard Smith       //   an array with no corresponding initializer [...] the destruction of
60772236372SRichard Smith       //   every temporary created in a default argument is sequenced before
60872236372SRichard Smith       //   the construction of the next array element, if any
60972236372SRichard Smith       CodeGenFunction::RunCleanupsScope CleanupsScope(CGF);
6107f416cc4SJohn McCall       LValue elementLV =
6117f416cc4SJohn McCall         CGF.MakeAddrLValue(Address(currentElement, elementAlign), elementType);
612c83ed824SSebastian Redl       if (filler)
613615ed1a3SChad Rosier         EmitInitializationToLValue(filler, elementLV);
614c83ed824SSebastian Redl       else
615c83ed824SSebastian Redl         EmitNullInitializationToLValue(elementLV);
61672236372SRichard Smith     }
617c83ed824SSebastian Redl 
618c83ed824SSebastian Redl     // Move on to the next element.
619c83ed824SSebastian Redl     llvm::Value *nextElement =
620c83ed824SSebastian Redl       Builder.CreateInBoundsGEP(currentElement, one, "arrayinit.next");
621c83ed824SSebastian Redl 
622c83ed824SSebastian Redl     // Tell the EH cleanup that we finished with the last element.
6237f416cc4SJohn McCall     if (endOfInit.isValid()) Builder.CreateStore(nextElement, endOfInit);
624c83ed824SSebastian Redl 
625c83ed824SSebastian Redl     // Leave the loop if we're done.
626c83ed824SSebastian Redl     llvm::Value *done = Builder.CreateICmpEQ(nextElement, end,
627c83ed824SSebastian Redl                                              "arrayinit.done");
628c83ed824SSebastian Redl     llvm::BasicBlock *endBB = CGF.createBasicBlock("arrayinit.end");
629c83ed824SSebastian Redl     Builder.CreateCondBr(done, endBB, bodyBB);
630c83ed824SSebastian Redl     currentElement->addIncoming(nextElement, Builder.GetInsertBlock());
631c83ed824SSebastian Redl 
632c83ed824SSebastian Redl     CGF.EmitBlock(endBB);
633c83ed824SSebastian Redl   }
634c83ed824SSebastian Redl 
635c83ed824SSebastian Redl   // Leave the partial-array cleanup if we entered one.
636c83ed824SSebastian Redl   if (dtorKind) CGF.DeactivateCleanupBlock(cleanup, cleanupDominator);
637c83ed824SSebastian Redl }
638c83ed824SSebastian Redl 
6397a51313dSChris Lattner //===----------------------------------------------------------------------===//
6407a51313dSChris Lattner //                            Visitor Methods
6417a51313dSChris Lattner //===----------------------------------------------------------------------===//
6427a51313dSChris Lattner 
643fe31481fSDouglas Gregor void AggExprEmitter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E){
644b0561b33STyker   Visit(E->getSubExpr());
645fe31481fSDouglas Gregor }
646fe31481fSDouglas Gregor 
6471bf5846aSJohn McCall void AggExprEmitter::VisitOpaqueValueExpr(OpaqueValueExpr *e) {
648797afe3aSAkira Hatanaka   // If this is a unique OVE, just visit its source expression.
649797afe3aSAkira Hatanaka   if (e->isUnique())
650797afe3aSAkira Hatanaka     Visit(e->getSourceExpr());
651797afe3aSAkira Hatanaka   else
652797afe3aSAkira Hatanaka     EmitFinalDestCopy(e->getType(), CGF.getOrCreateOpaqueLValueMapping(e));
6531bf5846aSJohn McCall }
6541bf5846aSJohn McCall 
6559b71f0cfSDouglas Gregor void
6569b71f0cfSDouglas Gregor AggExprEmitter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
657bea4c3d8SJohn McCall   if (Dest.isPotentiallyAliased() &&
658bea4c3d8SJohn McCall       E->getType().isPODType(CGF.getContext())) {
6596c9d31ebSDouglas Gregor     // For a POD type, just emit a load of the lvalue + a copy, because our
6606c9d31ebSDouglas Gregor     // compound literal might alias the destination.
6616c9d31ebSDouglas Gregor     EmitAggLoadOfLValue(E);
6626c9d31ebSDouglas Gregor     return;
6636c9d31ebSDouglas Gregor   }
6646c9d31ebSDouglas Gregor 
6659b71f0cfSDouglas Gregor   AggValueSlot Slot = EnsureSlot(E->getType());
66640568fecSAkira Hatanaka 
66740568fecSAkira Hatanaka   // Block-scope compound literals are destroyed at the end of the enclosing
66840568fecSAkira Hatanaka   // scope in C.
66940568fecSAkira Hatanaka   bool Destruct =
67040568fecSAkira Hatanaka       !CGF.getLangOpts().CPlusPlus && !Slot.isExternallyDestructed();
67140568fecSAkira Hatanaka   if (Destruct)
67240568fecSAkira Hatanaka     Slot.setExternallyDestructed();
67340568fecSAkira Hatanaka 
6749b71f0cfSDouglas Gregor   CGF.EmitAggExpr(E->getInitializer(), Slot);
67540568fecSAkira Hatanaka 
67640568fecSAkira Hatanaka   if (Destruct)
67740568fecSAkira Hatanaka     if (QualType::DestructionKind DtorKind = E->getType().isDestructedType())
67840568fecSAkira Hatanaka       CGF.pushLifetimeExtendedDestroy(
67940568fecSAkira Hatanaka           CGF.getCleanupKind(DtorKind), Slot.getAddress(), E->getType(),
68040568fecSAkira Hatanaka           CGF.getDestroyer(DtorKind), DtorKind & EHCleanup);
6819b71f0cfSDouglas Gregor }
6829b71f0cfSDouglas Gregor 
683a8ec7eb9SJohn McCall /// Attempt to look through various unimportant expressions to find a
684a8ec7eb9SJohn McCall /// cast of the given kind.
68503a9526fSEhud Katz static Expr *findPeephole(Expr *op, CastKind kind, const ASTContext &ctx) {
68603a9526fSEhud Katz   op = op->IgnoreParenNoopCasts(ctx);
68703a9526fSEhud Katz   if (auto castE = dyn_cast<CastExpr>(op)) {
688a8ec7eb9SJohn McCall     if (castE->getCastKind() == kind)
689a8ec7eb9SJohn McCall       return castE->getSubExpr();
690a8ec7eb9SJohn McCall   }
6918a13c418SCraig Topper   return nullptr;
692a8ec7eb9SJohn McCall }
6939b71f0cfSDouglas Gregor 
694ec143777SAnders Carlsson void AggExprEmitter::VisitCastExpr(CastExpr *E) {
6952bf9b4c0SAlexey Bataev   if (const auto *ECE = dyn_cast<ExplicitCastExpr>(E))
6962bf9b4c0SAlexey Bataev     CGF.CGM.EmitExplicitCastExprType(ECE, &CGF);
6971fb7ae9eSAnders Carlsson   switch (E->getCastKind()) {
6988a01a751SAnders Carlsson   case CK_Dynamic: {
69969d0d262SRichard Smith     // FIXME: Can this actually happen? We have no test coverage for it.
7001c073f47SDouglas Gregor     assert(isa<CXXDynamicCastExpr>(E) && "CK_Dynamic without a dynamic_cast?");
70169d0d262SRichard Smith     LValue LV = CGF.EmitCheckedLValue(E->getSubExpr(),
7024d1458edSRichard Smith                                       CodeGenFunction::TCK_Load);
7031c073f47SDouglas Gregor     // FIXME: Do we also need to handle property references here?
7041c073f47SDouglas Gregor     if (LV.isSimple())
705f139ae3dSAkira Hatanaka       CGF.EmitDynamicCast(LV.getAddress(CGF), cast<CXXDynamicCastExpr>(E));
7061c073f47SDouglas Gregor     else
7071c073f47SDouglas Gregor       CGF.CGM.ErrorUnsupported(E, "non-simple lvalue dynamic_cast");
7081c073f47SDouglas Gregor 
7097a626f63SJohn McCall     if (!Dest.isIgnored())
7101c073f47SDouglas Gregor       CGF.CGM.ErrorUnsupported(E, "lvalue dynamic_cast with a destination");
7111c073f47SDouglas Gregor     break;
7121c073f47SDouglas Gregor   }
7131c073f47SDouglas Gregor 
714e302792bSJohn McCall   case CK_ToUnion: {
715892bb0caSReid Kleckner     // Evaluate even if the destination is ignored.
716892bb0caSReid Kleckner     if (Dest.isIgnored()) {
717892bb0caSReid Kleckner       CGF.EmitAnyExpr(E->getSubExpr(), AggValueSlot::ignored(),
718892bb0caSReid Kleckner                       /*ignoreResult=*/true);
719892bb0caSReid Kleckner       break;
720892bb0caSReid Kleckner     }
72158989b71SJohn McCall 
7227ffcf93bSNuno Lopes     // GCC union extension
7232e442a00SDaniel Dunbar     QualType Ty = E->getSubExpr()->getType();
7247f416cc4SJohn McCall     Address CastPtr =
7257f416cc4SJohn McCall       Builder.CreateElementBitCast(Dest.getAddress(), CGF.ConvertType(Ty));
7261553b190SJohn McCall     EmitInitializationToLValue(E->getSubExpr(),
727615ed1a3SChad Rosier                                CGF.MakeAddrLValue(CastPtr, Ty));
7281fb7ae9eSAnders Carlsson     break;
7297ffcf93bSNuno Lopes   }
7307ffcf93bSNuno Lopes 
731eee944e7SErik Pilkington   case CK_LValueToRValueBitCast: {
732eee944e7SErik Pilkington     if (Dest.isIgnored()) {
733eee944e7SErik Pilkington       CGF.EmitAnyExpr(E->getSubExpr(), AggValueSlot::ignored(),
734eee944e7SErik Pilkington                       /*ignoreResult=*/true);
735eee944e7SErik Pilkington       break;
736eee944e7SErik Pilkington     }
737eee944e7SErik Pilkington 
738eee944e7SErik Pilkington     LValue SourceLV = CGF.EmitLValue(E->getSubExpr());
739eee944e7SErik Pilkington     Address SourceAddress =
740f139ae3dSAkira Hatanaka         Builder.CreateElementBitCast(SourceLV.getAddress(CGF), CGF.Int8Ty);
741eee944e7SErik Pilkington     Address DestAddress =
742eee944e7SErik Pilkington         Builder.CreateElementBitCast(Dest.getAddress(), CGF.Int8Ty);
743eee944e7SErik Pilkington     llvm::Value *SizeVal = llvm::ConstantInt::get(
744eee944e7SErik Pilkington         CGF.SizeTy,
745eee944e7SErik Pilkington         CGF.getContext().getTypeSizeInChars(E->getType()).getQuantity());
746eee944e7SErik Pilkington     Builder.CreateMemCpy(DestAddress, SourceAddress, SizeVal);
747eee944e7SErik Pilkington     break;
748eee944e7SErik Pilkington   }
749eee944e7SErik Pilkington 
750e302792bSJohn McCall   case CK_DerivedToBase:
751e302792bSJohn McCall   case CK_BaseToDerived:
752e302792bSJohn McCall   case CK_UncheckedDerivedToBase: {
75383d382b1SDavid Blaikie     llvm_unreachable("cannot perform hierarchy conversion in EmitAggExpr: "
754aae38d66SDouglas Gregor                 "should have been unpacked before we got here");
755aae38d66SDouglas Gregor   }
756aae38d66SDouglas Gregor 
757a8ec7eb9SJohn McCall   case CK_NonAtomicToAtomic:
758a8ec7eb9SJohn McCall   case CK_AtomicToNonAtomic: {
759a8ec7eb9SJohn McCall     bool isToAtomic = (E->getCastKind() == CK_NonAtomicToAtomic);
760a8ec7eb9SJohn McCall 
761a8ec7eb9SJohn McCall     // Determine the atomic and value types.
762a8ec7eb9SJohn McCall     QualType atomicType = E->getSubExpr()->getType();
763a8ec7eb9SJohn McCall     QualType valueType = E->getType();
764a8ec7eb9SJohn McCall     if (isToAtomic) std::swap(atomicType, valueType);
765a8ec7eb9SJohn McCall 
766a8ec7eb9SJohn McCall     assert(atomicType->isAtomicType());
767a8ec7eb9SJohn McCall     assert(CGF.getContext().hasSameUnqualifiedType(valueType,
768a8ec7eb9SJohn McCall                           atomicType->castAs<AtomicType>()->getValueType()));
769a8ec7eb9SJohn McCall 
770a8ec7eb9SJohn McCall     // Just recurse normally if we're ignoring the result or the
771a8ec7eb9SJohn McCall     // atomic type doesn't change representation.
772a8ec7eb9SJohn McCall     if (Dest.isIgnored() || !CGF.CGM.isPaddedAtomicType(atomicType)) {
773a8ec7eb9SJohn McCall       return Visit(E->getSubExpr());
774a8ec7eb9SJohn McCall     }
775a8ec7eb9SJohn McCall 
776a8ec7eb9SJohn McCall     CastKind peepholeTarget =
777a8ec7eb9SJohn McCall       (isToAtomic ? CK_AtomicToNonAtomic : CK_NonAtomicToAtomic);
778a8ec7eb9SJohn McCall 
779a8ec7eb9SJohn McCall     // These two cases are reverses of each other; try to peephole them.
78003a9526fSEhud Katz     if (Expr *op =
78103a9526fSEhud Katz             findPeephole(E->getSubExpr(), peepholeTarget, CGF.getContext())) {
782a8ec7eb9SJohn McCall       assert(CGF.getContext().hasSameUnqualifiedType(op->getType(),
783a8ec7eb9SJohn McCall                                                      E->getType()) &&
784a8ec7eb9SJohn McCall            "peephole significantly changed types?");
785a8ec7eb9SJohn McCall       return Visit(op);
786a8ec7eb9SJohn McCall     }
787a8ec7eb9SJohn McCall 
788a8ec7eb9SJohn McCall     // If we're converting an r-value of non-atomic type to an r-value
789be4504dfSEli Friedman     // of atomic type, just emit directly into the relevant sub-object.
790a8ec7eb9SJohn McCall     if (isToAtomic) {
791be4504dfSEli Friedman       AggValueSlot valueDest = Dest;
792be4504dfSEli Friedman       if (!valueDest.isIgnored() && CGF.CGM.isPaddedAtomicType(atomicType)) {
7932a8c18d9SAlexander Kornienko         // Zero-initialize.  (Strictly speaking, we only need to initialize
794be4504dfSEli Friedman         // the padding at the end, but this is simpler.)
795be4504dfSEli Friedman         if (!Dest.isZeroed())
7967f416cc4SJohn McCall           CGF.EmitNullInitialization(Dest.getAddress(), atomicType);
797be4504dfSEli Friedman 
798be4504dfSEli Friedman         // Build a GEP to refer to the subobject.
7997f416cc4SJohn McCall         Address valueAddr =
800751fe286SJames Y Knight             CGF.Builder.CreateStructGEP(valueDest.getAddress(), 0);
801be4504dfSEli Friedman         valueDest = AggValueSlot::forAddr(valueAddr,
802be4504dfSEli Friedman                                           valueDest.getQualifiers(),
803be4504dfSEli Friedman                                           valueDest.isExternallyDestructed(),
804be4504dfSEli Friedman                                           valueDest.requiresGCollection(),
805be4504dfSEli Friedman                                           valueDest.isPotentiallyAliased(),
806e78fac51SRichard Smith                                           AggValueSlot::DoesNotOverlap,
807be4504dfSEli Friedman                                           AggValueSlot::IsZeroed);
808be4504dfSEli Friedman       }
809be4504dfSEli Friedman 
810035b39e3SEli Friedman       CGF.EmitAggExpr(E->getSubExpr(), valueDest);
811a8ec7eb9SJohn McCall       return;
812a8ec7eb9SJohn McCall     }
813a8ec7eb9SJohn McCall 
814a8ec7eb9SJohn McCall     // Otherwise, we're converting an atomic type to a non-atomic type.
815be4504dfSEli Friedman     // Make an atomic temporary, emit into that, and then copy the value out.
816a8ec7eb9SJohn McCall     AggValueSlot atomicSlot =
817a8ec7eb9SJohn McCall       CGF.CreateAggTemp(atomicType, "atomic-to-nonatomic.temp");
818a8ec7eb9SJohn McCall     CGF.EmitAggExpr(E->getSubExpr(), atomicSlot);
819a8ec7eb9SJohn McCall 
820751fe286SJames Y Knight     Address valueAddr = Builder.CreateStructGEP(atomicSlot.getAddress(), 0);
821a8ec7eb9SJohn McCall     RValue rvalue = RValue::getAggregate(valueAddr, atomicSlot.isVolatile());
822a8ec7eb9SJohn McCall     return EmitFinalDestCopy(valueType, rvalue);
823a8ec7eb9SJohn McCall   }
824094c7266SAnastasia Stulova   case CK_AddressSpaceConversion:
825094c7266SAnastasia Stulova      return Visit(E->getSubExpr());
826a8ec7eb9SJohn McCall 
8274e8ca4faSJohn McCall   case CK_LValueToRValue:
8284e8ca4faSJohn McCall     // If we're loading from a volatile type, force the destination
8294e8ca4faSJohn McCall     // into existence.
8304e8ca4faSJohn McCall     if (E->getSubExpr()->getType().isVolatileQualified()) {
831d35a4541SAkira Hatanaka       bool Destruct =
832d35a4541SAkira Hatanaka           !Dest.isExternallyDestructed() &&
833d35a4541SAkira Hatanaka           E->getType().isDestructedType() == QualType::DK_nontrivial_c_struct;
834d35a4541SAkira Hatanaka       if (Destruct)
835d35a4541SAkira Hatanaka         Dest.setExternallyDestructed();
8364e8ca4faSJohn McCall       EnsureDest(E->getType());
837d35a4541SAkira Hatanaka       Visit(E->getSubExpr());
838d35a4541SAkira Hatanaka 
839d35a4541SAkira Hatanaka       if (Destruct)
840d35a4541SAkira Hatanaka         CGF.pushDestroy(QualType::DK_nontrivial_c_struct, Dest.getAddress(),
841d35a4541SAkira Hatanaka                         E->getType());
842d35a4541SAkira Hatanaka 
843d35a4541SAkira Hatanaka       return;
8444e8ca4faSJohn McCall     }
845a8ec7eb9SJohn McCall 
846f3b3ccdaSAdrian Prantl     LLVM_FALLTHROUGH;
8474e8ca4faSJohn McCall 
848094c7266SAnastasia Stulova 
849e302792bSJohn McCall   case CK_NoOp:
850e302792bSJohn McCall   case CK_UserDefinedConversion:
851e302792bSJohn McCall   case CK_ConstructorConversion:
8522a69547fSEli Friedman     assert(CGF.getContext().hasSameUnqualifiedType(E->getSubExpr()->getType(),
8532a69547fSEli Friedman                                                    E->getType()) &&
8540f398c44SChris Lattner            "Implicit cast types must be compatible");
8557a51313dSChris Lattner     Visit(E->getSubExpr());
8561fb7ae9eSAnders Carlsson     break;
857b05a3e55SAnders Carlsson 
858e302792bSJohn McCall   case CK_LValueBitCast:
859f3735e01SJohn McCall     llvm_unreachable("should not be emitting lvalue bitcast as rvalue");
86031996343SJohn McCall 
861f3735e01SJohn McCall   case CK_Dependent:
862f3735e01SJohn McCall   case CK_BitCast:
863f3735e01SJohn McCall   case CK_ArrayToPointerDecay:
864f3735e01SJohn McCall   case CK_FunctionToPointerDecay:
865f3735e01SJohn McCall   case CK_NullToPointer:
866f3735e01SJohn McCall   case CK_NullToMemberPointer:
867f3735e01SJohn McCall   case CK_BaseToDerivedMemberPointer:
868f3735e01SJohn McCall   case CK_DerivedToBaseMemberPointer:
869f3735e01SJohn McCall   case CK_MemberPointerToBoolean:
870c62bb391SJohn McCall   case CK_ReinterpretMemberPointer:
871f3735e01SJohn McCall   case CK_IntegralToPointer:
872f3735e01SJohn McCall   case CK_PointerToIntegral:
873f3735e01SJohn McCall   case CK_PointerToBoolean:
874f3735e01SJohn McCall   case CK_ToVoid:
875f3735e01SJohn McCall   case CK_VectorSplat:
876f3735e01SJohn McCall   case CK_IntegralCast:
877df1ed009SGeorge Burgess IV   case CK_BooleanToSignedIntegral:
878f3735e01SJohn McCall   case CK_IntegralToBoolean:
879f3735e01SJohn McCall   case CK_IntegralToFloating:
880f3735e01SJohn McCall   case CK_FloatingToIntegral:
881f3735e01SJohn McCall   case CK_FloatingToBoolean:
882f3735e01SJohn McCall   case CK_FloatingCast:
8839320b87cSJohn McCall   case CK_CPointerToObjCPointerCast:
8849320b87cSJohn McCall   case CK_BlockPointerToObjCPointerCast:
885f3735e01SJohn McCall   case CK_AnyPointerToBlockPointerCast:
886f3735e01SJohn McCall   case CK_ObjCObjectLValueCast:
887f3735e01SJohn McCall   case CK_FloatingRealToComplex:
888f3735e01SJohn McCall   case CK_FloatingComplexToReal:
889f3735e01SJohn McCall   case CK_FloatingComplexToBoolean:
890f3735e01SJohn McCall   case CK_FloatingComplexCast:
891f3735e01SJohn McCall   case CK_FloatingComplexToIntegralComplex:
892f3735e01SJohn McCall   case CK_IntegralRealToComplex:
893f3735e01SJohn McCall   case CK_IntegralComplexToReal:
894f3735e01SJohn McCall   case CK_IntegralComplexToBoolean:
895f3735e01SJohn McCall   case CK_IntegralComplexCast:
896f3735e01SJohn McCall   case CK_IntegralComplexToFloatingComplex:
8972d637d2eSJohn McCall   case CK_ARCProduceObject:
8982d637d2eSJohn McCall   case CK_ARCConsumeObject:
8992d637d2eSJohn McCall   case CK_ARCReclaimReturnedObject:
9002d637d2eSJohn McCall   case CK_ARCExtendBlockObject:
901ed90df38SDouglas Gregor   case CK_CopyAndAutoreleaseBlockObject:
90234866c77SEli Friedman   case CK_BuiltinFnToFnPtr:
903b555b76eSAndrew Savonichev   case CK_ZeroToOCLOpaqueType:
904094c7266SAnastasia Stulova 
9050bc4b2d3SYaxun Liu   case CK_IntToOCLSampler:
9069fa7f484SBevin Hansson   case CK_FloatingToFixedPoint:
9079fa7f484SBevin Hansson   case CK_FixedPointToFloating:
90899bda375SLeonard Chan   case CK_FixedPointCast:
909b4ba467dSLeonard Chan   case CK_FixedPointToBoolean:
9108f7caae0SLeonard Chan   case CK_FixedPointToIntegral:
9118f7caae0SLeonard Chan   case CK_IntegralToFixedPoint:
912f3735e01SJohn McCall     llvm_unreachable("cast kind invalid for aggregate types");
9131fb7ae9eSAnders Carlsson   }
9147a51313dSChris Lattner }
9157a51313dSChris Lattner 
9160f398c44SChris Lattner void AggExprEmitter::VisitCallExpr(const CallExpr *E) {
917ced8bdf7SDavid Majnemer   if (E->getCallReturnType(CGF.getContext())->isReferenceType()) {
918ddcbfe7bSAnders Carlsson     EmitAggLoadOfLValue(E);
919ddcbfe7bSAnders Carlsson     return;
920ddcbfe7bSAnders Carlsson   }
921ddcbfe7bSAnders Carlsson 
92256e5a2e1SGeorge Burgess IV   withReturnValueSlot(E, [&](ReturnValueSlot Slot) {
92356e5a2e1SGeorge Burgess IV     return CGF.EmitCallExpr(E, Slot);
92456e5a2e1SGeorge Burgess IV   });
9257a51313dSChris Lattner }
9260f398c44SChris Lattner 
9270f398c44SChris Lattner void AggExprEmitter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
92856e5a2e1SGeorge Burgess IV   withReturnValueSlot(E, [&](ReturnValueSlot Slot) {
92956e5a2e1SGeorge Burgess IV     return CGF.EmitObjCMessageExpr(E, Slot);
93056e5a2e1SGeorge Burgess IV   });
931b1d329daSChris Lattner }
9327a51313dSChris Lattner 
9330f398c44SChris Lattner void AggExprEmitter::VisitBinComma(const BinaryOperator *E) {
934a2342eb8SJohn McCall   CGF.EmitIgnoredExpr(E->getLHS());
9357a626f63SJohn McCall   Visit(E->getRHS());
9364b0e2a30SEli Friedman }
9374b0e2a30SEli Friedman 
9387a51313dSChris Lattner void AggExprEmitter::VisitStmtExpr(const StmtExpr *E) {
939ce1de617SJohn McCall   CodeGenFunction::StmtExprEvaluation eval(CGF);
9407a626f63SJohn McCall   CGF.EmitCompoundStmt(*E->getSubStmt(), true, Dest);
9417a51313dSChris Lattner }
9427a51313dSChris Lattner 
9430683c0e6SEric Fiselier enum CompareKind {
9440683c0e6SEric Fiselier   CK_Less,
9450683c0e6SEric Fiselier   CK_Greater,
9460683c0e6SEric Fiselier   CK_Equal,
9470683c0e6SEric Fiselier };
9480683c0e6SEric Fiselier 
9490683c0e6SEric Fiselier static llvm::Value *EmitCompare(CGBuilderTy &Builder, CodeGenFunction &CGF,
9500683c0e6SEric Fiselier                                 const BinaryOperator *E, llvm::Value *LHS,
9510683c0e6SEric Fiselier                                 llvm::Value *RHS, CompareKind Kind,
9520683c0e6SEric Fiselier                                 const char *NameSuffix = "") {
9530683c0e6SEric Fiselier   QualType ArgTy = E->getLHS()->getType();
9540683c0e6SEric Fiselier   if (const ComplexType *CT = ArgTy->getAs<ComplexType>())
9550683c0e6SEric Fiselier     ArgTy = CT->getElementType();
9560683c0e6SEric Fiselier 
9570683c0e6SEric Fiselier   if (const auto *MPT = ArgTy->getAs<MemberPointerType>()) {
9580683c0e6SEric Fiselier     assert(Kind == CK_Equal &&
9590683c0e6SEric Fiselier            "member pointers may only be compared for equality");
9600683c0e6SEric Fiselier     return CGF.CGM.getCXXABI().EmitMemberPointerComparison(
9610683c0e6SEric Fiselier         CGF, LHS, RHS, MPT, /*IsInequality*/ false);
9620683c0e6SEric Fiselier   }
9630683c0e6SEric Fiselier 
9640683c0e6SEric Fiselier   // Compute the comparison instructions for the specified comparison kind.
9650683c0e6SEric Fiselier   struct CmpInstInfo {
9660683c0e6SEric Fiselier     const char *Name;
9670683c0e6SEric Fiselier     llvm::CmpInst::Predicate FCmp;
9680683c0e6SEric Fiselier     llvm::CmpInst::Predicate SCmp;
9690683c0e6SEric Fiselier     llvm::CmpInst::Predicate UCmp;
9700683c0e6SEric Fiselier   };
9710683c0e6SEric Fiselier   CmpInstInfo InstInfo = [&]() -> CmpInstInfo {
9720683c0e6SEric Fiselier     using FI = llvm::FCmpInst;
9730683c0e6SEric Fiselier     using II = llvm::ICmpInst;
9740683c0e6SEric Fiselier     switch (Kind) {
9750683c0e6SEric Fiselier     case CK_Less:
9760683c0e6SEric Fiselier       return {"cmp.lt", FI::FCMP_OLT, II::ICMP_SLT, II::ICMP_ULT};
9770683c0e6SEric Fiselier     case CK_Greater:
9780683c0e6SEric Fiselier       return {"cmp.gt", FI::FCMP_OGT, II::ICMP_SGT, II::ICMP_UGT};
9790683c0e6SEric Fiselier     case CK_Equal:
9800683c0e6SEric Fiselier       return {"cmp.eq", FI::FCMP_OEQ, II::ICMP_EQ, II::ICMP_EQ};
9810683c0e6SEric Fiselier     }
9823366dcfeSSimon Pilgrim     llvm_unreachable("Unrecognised CompareKind enum");
9830683c0e6SEric Fiselier   }();
9840683c0e6SEric Fiselier 
9850683c0e6SEric Fiselier   if (ArgTy->hasFloatingRepresentation())
9860683c0e6SEric Fiselier     return Builder.CreateFCmp(InstInfo.FCmp, LHS, RHS,
9870683c0e6SEric Fiselier                               llvm::Twine(InstInfo.Name) + NameSuffix);
9880683c0e6SEric Fiselier   if (ArgTy->isIntegralOrEnumerationType() || ArgTy->isPointerType()) {
9890683c0e6SEric Fiselier     auto Inst =
9900683c0e6SEric Fiselier         ArgTy->hasSignedIntegerRepresentation() ? InstInfo.SCmp : InstInfo.UCmp;
9910683c0e6SEric Fiselier     return Builder.CreateICmp(Inst, LHS, RHS,
9920683c0e6SEric Fiselier                               llvm::Twine(InstInfo.Name) + NameSuffix);
9930683c0e6SEric Fiselier   }
9940683c0e6SEric Fiselier 
9950683c0e6SEric Fiselier   llvm_unreachable("unsupported aggregate binary expression should have "
9960683c0e6SEric Fiselier                    "already been handled");
9970683c0e6SEric Fiselier }
9980683c0e6SEric Fiselier 
9990683c0e6SEric Fiselier void AggExprEmitter::VisitBinCmp(const BinaryOperator *E) {
10000683c0e6SEric Fiselier   using llvm::BasicBlock;
10010683c0e6SEric Fiselier   using llvm::PHINode;
10020683c0e6SEric Fiselier   using llvm::Value;
10030683c0e6SEric Fiselier   assert(CGF.getContext().hasSameType(E->getLHS()->getType(),
10040683c0e6SEric Fiselier                                       E->getRHS()->getType()));
10050683c0e6SEric Fiselier   const ComparisonCategoryInfo &CmpInfo =
10060683c0e6SEric Fiselier       CGF.getContext().CompCategories.getInfoForType(E->getType());
10070683c0e6SEric Fiselier   assert(CmpInfo.Record->isTriviallyCopyable() &&
10080683c0e6SEric Fiselier          "cannot copy non-trivially copyable aggregate");
10090683c0e6SEric Fiselier 
10100683c0e6SEric Fiselier   QualType ArgTy = E->getLHS()->getType();
10110683c0e6SEric Fiselier 
10120683c0e6SEric Fiselier   if (!ArgTy->isIntegralOrEnumerationType() && !ArgTy->isRealFloatingType() &&
10130683c0e6SEric Fiselier       !ArgTy->isNullPtrType() && !ArgTy->isPointerType() &&
10140683c0e6SEric Fiselier       !ArgTy->isMemberPointerType() && !ArgTy->isAnyComplexType()) {
1015c5fb8580SEric Fiselier     return CGF.ErrorUnsupported(E, "aggregate three-way comparison");
10160683c0e6SEric Fiselier   }
10170683c0e6SEric Fiselier   bool IsComplex = ArgTy->isAnyComplexType();
10180683c0e6SEric Fiselier 
10190683c0e6SEric Fiselier   // Evaluate the operands to the expression and extract their values.
10200683c0e6SEric Fiselier   auto EmitOperand = [&](Expr *E) -> std::pair<Value *, Value *> {
10210683c0e6SEric Fiselier     RValue RV = CGF.EmitAnyExpr(E);
10220683c0e6SEric Fiselier     if (RV.isScalar())
10230683c0e6SEric Fiselier       return {RV.getScalarVal(), nullptr};
10240683c0e6SEric Fiselier     if (RV.isAggregate())
10250683c0e6SEric Fiselier       return {RV.getAggregatePointer(), nullptr};
10260683c0e6SEric Fiselier     assert(RV.isComplex());
10270683c0e6SEric Fiselier     return RV.getComplexVal();
10280683c0e6SEric Fiselier   };
10290683c0e6SEric Fiselier   auto LHSValues = EmitOperand(E->getLHS()),
10300683c0e6SEric Fiselier        RHSValues = EmitOperand(E->getRHS());
10310683c0e6SEric Fiselier 
10320683c0e6SEric Fiselier   auto EmitCmp = [&](CompareKind K) {
10330683c0e6SEric Fiselier     Value *Cmp = EmitCompare(Builder, CGF, E, LHSValues.first, RHSValues.first,
10340683c0e6SEric Fiselier                              K, IsComplex ? ".r" : "");
10350683c0e6SEric Fiselier     if (!IsComplex)
10360683c0e6SEric Fiselier       return Cmp;
10370683c0e6SEric Fiselier     assert(K == CompareKind::CK_Equal);
10380683c0e6SEric Fiselier     Value *CmpImag = EmitCompare(Builder, CGF, E, LHSValues.second,
10390683c0e6SEric Fiselier                                  RHSValues.second, K, ".i");
10400683c0e6SEric Fiselier     return Builder.CreateAnd(Cmp, CmpImag, "and.eq");
10410683c0e6SEric Fiselier   };
10420683c0e6SEric Fiselier   auto EmitCmpRes = [&](const ComparisonCategoryInfo::ValueInfo *VInfo) {
10430683c0e6SEric Fiselier     return Builder.getInt(VInfo->getIntValue());
10440683c0e6SEric Fiselier   };
10450683c0e6SEric Fiselier 
10460683c0e6SEric Fiselier   Value *Select;
10470683c0e6SEric Fiselier   if (ArgTy->isNullPtrType()) {
10480683c0e6SEric Fiselier     Select = EmitCmpRes(CmpInfo.getEqualOrEquiv());
10490683c0e6SEric Fiselier   } else if (!CmpInfo.isPartial()) {
10500683c0e6SEric Fiselier     Value *SelectOne =
10510683c0e6SEric Fiselier         Builder.CreateSelect(EmitCmp(CK_Less), EmitCmpRes(CmpInfo.getLess()),
10520683c0e6SEric Fiselier                              EmitCmpRes(CmpInfo.getGreater()), "sel.lt");
10530683c0e6SEric Fiselier     Select = Builder.CreateSelect(EmitCmp(CK_Equal),
10540683c0e6SEric Fiselier                                   EmitCmpRes(CmpInfo.getEqualOrEquiv()),
10550683c0e6SEric Fiselier                                   SelectOne, "sel.eq");
10560683c0e6SEric Fiselier   } else {
10570683c0e6SEric Fiselier     Value *SelectEq = Builder.CreateSelect(
10580683c0e6SEric Fiselier         EmitCmp(CK_Equal), EmitCmpRes(CmpInfo.getEqualOrEquiv()),
10590683c0e6SEric Fiselier         EmitCmpRes(CmpInfo.getUnordered()), "sel.eq");
10600683c0e6SEric Fiselier     Value *SelectGT = Builder.CreateSelect(EmitCmp(CK_Greater),
10610683c0e6SEric Fiselier                                            EmitCmpRes(CmpInfo.getGreater()),
10620683c0e6SEric Fiselier                                            SelectEq, "sel.gt");
10630683c0e6SEric Fiselier     Select = Builder.CreateSelect(
10640683c0e6SEric Fiselier         EmitCmp(CK_Less), EmitCmpRes(CmpInfo.getLess()), SelectGT, "sel.lt");
10650683c0e6SEric Fiselier   }
10660683c0e6SEric Fiselier   // Create the return value in the destination slot.
10670683c0e6SEric Fiselier   EnsureDest(E->getType());
10680683c0e6SEric Fiselier   LValue DestLV = CGF.MakeAddrLValue(Dest.getAddress(), E->getType());
10690683c0e6SEric Fiselier 
10700683c0e6SEric Fiselier   // Emit the address of the first (and only) field in the comparison category
10710683c0e6SEric Fiselier   // type, and initialize it from the constant integer value selected above.
10720683c0e6SEric Fiselier   LValue FieldLV = CGF.EmitLValueForFieldInitialization(
10730683c0e6SEric Fiselier       DestLV, *CmpInfo.Record->field_begin());
10740683c0e6SEric Fiselier   CGF.EmitStoreThroughLValue(RValue::get(Select), FieldLV, /*IsInit*/ true);
10750683c0e6SEric Fiselier 
10760683c0e6SEric Fiselier   // All done! The result is in the Dest slot.
10770683c0e6SEric Fiselier }
10780683c0e6SEric Fiselier 
10797a51313dSChris Lattner void AggExprEmitter::VisitBinaryOperator(const BinaryOperator *E) {
1080e302792bSJohn McCall   if (E->getOpcode() == BO_PtrMemD || E->getOpcode() == BO_PtrMemI)
1081ffba662dSFariborz Jahanian     VisitPointerToDataMemberBinaryOperator(E);
1082ffba662dSFariborz Jahanian   else
1083a7c8cf62SDaniel Dunbar     CGF.ErrorUnsupported(E, "aggregate binary expression");
10847a51313dSChris Lattner }
10857a51313dSChris Lattner 
1086ffba662dSFariborz Jahanian void AggExprEmitter::VisitPointerToDataMemberBinaryOperator(
1087ffba662dSFariborz Jahanian                                                     const BinaryOperator *E) {
1088ffba662dSFariborz Jahanian   LValue LV = CGF.EmitPointerToDataMemberBinaryExpr(E);
10894e8ca4faSJohn McCall   EmitFinalDestCopy(E->getType(), LV);
10904e8ca4faSJohn McCall }
10914e8ca4faSJohn McCall 
10924e8ca4faSJohn McCall /// Is the value of the given expression possibly a reference to or
10934e8ca4faSJohn McCall /// into a __block variable?
10944e8ca4faSJohn McCall static bool isBlockVarRef(const Expr *E) {
10954e8ca4faSJohn McCall   // Make sure we look through parens.
10964e8ca4faSJohn McCall   E = E->IgnoreParens();
10974e8ca4faSJohn McCall 
10984e8ca4faSJohn McCall   // Check for a direct reference to a __block variable.
10994e8ca4faSJohn McCall   if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
11004e8ca4faSJohn McCall     const VarDecl *var = dyn_cast<VarDecl>(DRE->getDecl());
11014e8ca4faSJohn McCall     return (var && var->hasAttr<BlocksAttr>());
11024e8ca4faSJohn McCall   }
11034e8ca4faSJohn McCall 
11044e8ca4faSJohn McCall   // More complicated stuff.
11054e8ca4faSJohn McCall 
11064e8ca4faSJohn McCall   // Binary operators.
11074e8ca4faSJohn McCall   if (const BinaryOperator *op = dyn_cast<BinaryOperator>(E)) {
11084e8ca4faSJohn McCall     // For an assignment or pointer-to-member operation, just care
11094e8ca4faSJohn McCall     // about the LHS.
11104e8ca4faSJohn McCall     if (op->isAssignmentOp() || op->isPtrMemOp())
11114e8ca4faSJohn McCall       return isBlockVarRef(op->getLHS());
11124e8ca4faSJohn McCall 
11134e8ca4faSJohn McCall     // For a comma, just care about the RHS.
11144e8ca4faSJohn McCall     if (op->getOpcode() == BO_Comma)
11154e8ca4faSJohn McCall       return isBlockVarRef(op->getRHS());
11164e8ca4faSJohn McCall 
11174e8ca4faSJohn McCall     // FIXME: pointer arithmetic?
11184e8ca4faSJohn McCall     return false;
11194e8ca4faSJohn McCall 
11204e8ca4faSJohn McCall   // Check both sides of a conditional operator.
11214e8ca4faSJohn McCall   } else if (const AbstractConditionalOperator *op
11224e8ca4faSJohn McCall                = dyn_cast<AbstractConditionalOperator>(E)) {
11234e8ca4faSJohn McCall     return isBlockVarRef(op->getTrueExpr())
11244e8ca4faSJohn McCall         || isBlockVarRef(op->getFalseExpr());
11254e8ca4faSJohn McCall 
11264e8ca4faSJohn McCall   // OVEs are required to support BinaryConditionalOperators.
11274e8ca4faSJohn McCall   } else if (const OpaqueValueExpr *op
11284e8ca4faSJohn McCall                = dyn_cast<OpaqueValueExpr>(E)) {
11294e8ca4faSJohn McCall     if (const Expr *src = op->getSourceExpr())
11304e8ca4faSJohn McCall       return isBlockVarRef(src);
11314e8ca4faSJohn McCall 
11324e8ca4faSJohn McCall   // Casts are necessary to get things like (*(int*)&var) = foo().
11334e8ca4faSJohn McCall   // We don't really care about the kind of cast here, except
11344e8ca4faSJohn McCall   // we don't want to look through l2r casts, because it's okay
11354e8ca4faSJohn McCall   // to get the *value* in a __block variable.
11364e8ca4faSJohn McCall   } else if (const CastExpr *cast = dyn_cast<CastExpr>(E)) {
11374e8ca4faSJohn McCall     if (cast->getCastKind() == CK_LValueToRValue)
11384e8ca4faSJohn McCall       return false;
11394e8ca4faSJohn McCall     return isBlockVarRef(cast->getSubExpr());
11404e8ca4faSJohn McCall 
11414e8ca4faSJohn McCall   // Handle unary operators.  Again, just aggressively look through
11424e8ca4faSJohn McCall   // it, ignoring the operation.
11434e8ca4faSJohn McCall   } else if (const UnaryOperator *uop = dyn_cast<UnaryOperator>(E)) {
11444e8ca4faSJohn McCall     return isBlockVarRef(uop->getSubExpr());
11454e8ca4faSJohn McCall 
11464e8ca4faSJohn McCall   // Look into the base of a field access.
11474e8ca4faSJohn McCall   } else if (const MemberExpr *mem = dyn_cast<MemberExpr>(E)) {
11484e8ca4faSJohn McCall     return isBlockVarRef(mem->getBase());
11494e8ca4faSJohn McCall 
11504e8ca4faSJohn McCall   // Look into the base of a subscript.
11514e8ca4faSJohn McCall   } else if (const ArraySubscriptExpr *sub = dyn_cast<ArraySubscriptExpr>(E)) {
11524e8ca4faSJohn McCall     return isBlockVarRef(sub->getBase());
11534e8ca4faSJohn McCall   }
11544e8ca4faSJohn McCall 
11554e8ca4faSJohn McCall   return false;
1156ffba662dSFariborz Jahanian }
1157ffba662dSFariborz Jahanian 
11587a51313dSChris Lattner void AggExprEmitter::VisitBinAssign(const BinaryOperator *E) {
11597a51313dSChris Lattner   // For an assignment to work, the value on the right has
11607a51313dSChris Lattner   // to be compatible with the value on the left.
11612a69547fSEli Friedman   assert(CGF.getContext().hasSameUnqualifiedType(E->getLHS()->getType(),
11622a69547fSEli Friedman                                                  E->getRHS()->getType())
11637a51313dSChris Lattner          && "Invalid assignment");
1164d0a30016SJohn McCall 
11654e8ca4faSJohn McCall   // If the LHS might be a __block variable, and the RHS can
11664e8ca4faSJohn McCall   // potentially cause a block copy, we need to evaluate the RHS first
11674e8ca4faSJohn McCall   // so that the assignment goes the right place.
11684e8ca4faSJohn McCall   // This is pretty semantically fragile.
11694e8ca4faSJohn McCall   if (isBlockVarRef(E->getLHS()) &&
117099514b91SFariborz Jahanian       E->getRHS()->HasSideEffects(CGF.getContext())) {
11714e8ca4faSJohn McCall     // Ensure that we have a destination, and evaluate the RHS into that.
11724e8ca4faSJohn McCall     EnsureDest(E->getRHS()->getType());
11734e8ca4faSJohn McCall     Visit(E->getRHS());
11744e8ca4faSJohn McCall 
11754e8ca4faSJohn McCall     // Now emit the LHS and copy into it.
1176e30752c9SRichard Smith     LValue LHS = CGF.EmitCheckedLValue(E->getLHS(), CodeGenFunction::TCK_Store);
11774e8ca4faSJohn McCall 
1178a8ec7eb9SJohn McCall     // That copy is an atomic copy if the LHS is atomic.
1179a5b195a1SDavid Majnemer     if (LHS.getType()->isAtomicType() ||
1180a5b195a1SDavid Majnemer         CGF.LValueIsSuitableForInlineAtomic(LHS)) {
1181a8ec7eb9SJohn McCall       CGF.EmitAtomicStore(Dest.asRValue(), LHS, /*isInit*/ false);
1182a8ec7eb9SJohn McCall       return;
1183a8ec7eb9SJohn McCall     }
1184a8ec7eb9SJohn McCall 
11854e8ca4faSJohn McCall     EmitCopy(E->getLHS()->getType(),
1186f139ae3dSAkira Hatanaka              AggValueSlot::forLValue(LHS, CGF, AggValueSlot::IsDestructed,
118746759f4fSJohn McCall                                      needsGC(E->getLHS()->getType()),
1188e78fac51SRichard Smith                                      AggValueSlot::IsAliased,
1189e78fac51SRichard Smith                                      AggValueSlot::MayOverlap),
11904e8ca4faSJohn McCall              Dest);
119199514b91SFariborz Jahanian     return;
119299514b91SFariborz Jahanian   }
119399514b91SFariborz Jahanian 
11947a51313dSChris Lattner   LValue LHS = CGF.EmitLValue(E->getLHS());
11957a51313dSChris Lattner 
1196a8ec7eb9SJohn McCall   // If we have an atomic type, evaluate into the destination and then
1197a8ec7eb9SJohn McCall   // do an atomic copy.
1198a5b195a1SDavid Majnemer   if (LHS.getType()->isAtomicType() ||
1199a5b195a1SDavid Majnemer       CGF.LValueIsSuitableForInlineAtomic(LHS)) {
1200a8ec7eb9SJohn McCall     EnsureDest(E->getRHS()->getType());
1201a8ec7eb9SJohn McCall     Visit(E->getRHS());
1202a8ec7eb9SJohn McCall     CGF.EmitAtomicStore(Dest.asRValue(), LHS, /*isInit*/ false);
1203a8ec7eb9SJohn McCall     return;
1204a8ec7eb9SJohn McCall   }
1205a8ec7eb9SJohn McCall 
12067a51313dSChris Lattner   // Codegen the RHS so that it stores directly into the LHS.
1207f139ae3dSAkira Hatanaka   AggValueSlot LHSSlot = AggValueSlot::forLValue(
1208f139ae3dSAkira Hatanaka       LHS, CGF, AggValueSlot::IsDestructed, needsGC(E->getLHS()->getType()),
1209f139ae3dSAkira Hatanaka       AggValueSlot::IsAliased, AggValueSlot::MayOverlap);
12107865220dSFariborz Jahanian   // A non-volatile aggregate destination might have volatile member.
12117865220dSFariborz Jahanian   if (!LHSSlot.isVolatile() &&
12127865220dSFariborz Jahanian       CGF.hasVolatileMember(E->getLHS()->getType()))
12137865220dSFariborz Jahanian     LHSSlot.setVolatile(true);
12147865220dSFariborz Jahanian 
12154e8ca4faSJohn McCall   CGF.EmitAggExpr(E->getRHS(), LHSSlot);
12164e8ca4faSJohn McCall 
12174e8ca4faSJohn McCall   // Copy into the destination if the assignment isn't ignored.
12184e8ca4faSJohn McCall   EmitFinalDestCopy(E->getType(), LHS);
121971e1a56dSAkira Hatanaka 
122071e1a56dSAkira Hatanaka   if (!Dest.isIgnored() && !Dest.isExternallyDestructed() &&
122171e1a56dSAkira Hatanaka       E->getType().isDestructedType() == QualType::DK_nontrivial_c_struct)
122271e1a56dSAkira Hatanaka     CGF.pushDestroy(QualType::DK_nontrivial_c_struct, Dest.getAddress(),
122371e1a56dSAkira Hatanaka                     E->getType());
12247a51313dSChris Lattner }
12257a51313dSChris Lattner 
1226c07a0c7eSJohn McCall void AggExprEmitter::
1227c07a0c7eSJohn McCall VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) {
1228a612e79bSDaniel Dunbar   llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true");
1229a612e79bSDaniel Dunbar   llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false");
1230a612e79bSDaniel Dunbar   llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end");
12317a51313dSChris Lattner 
1232c07a0c7eSJohn McCall   // Bind the common expression if necessary.
123348fd89adSEli Friedman   CodeGenFunction::OpaqueValueMapping binding(CGF, E);
1234c07a0c7eSJohn McCall 
1235ce1de617SJohn McCall   CodeGenFunction::ConditionalEvaluation eval(CGF);
123666242d6cSJustin Bogner   CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock,
123766242d6cSJustin Bogner                            CGF.getProfileCount(E));
12387a51313dSChris Lattner 
12395b26f65bSJohn McCall   // Save whether the destination's lifetime is externally managed.
1240cac93853SJohn McCall   bool isExternallyDestructed = Dest.isExternallyDestructed();
124171e1a56dSAkira Hatanaka   bool destructNonTrivialCStruct =
124271e1a56dSAkira Hatanaka       !isExternallyDestructed &&
124371e1a56dSAkira Hatanaka       E->getType().isDestructedType() == QualType::DK_nontrivial_c_struct;
124471e1a56dSAkira Hatanaka   isExternallyDestructed |= destructNonTrivialCStruct;
124571e1a56dSAkira Hatanaka   Dest.setExternallyDestructed(isExternallyDestructed);
12467a51313dSChris Lattner 
1247ce1de617SJohn McCall   eval.begin(CGF);
1248ce1de617SJohn McCall   CGF.EmitBlock(LHSBlock);
124966242d6cSJustin Bogner   CGF.incrementProfileCounter(E);
1250c07a0c7eSJohn McCall   Visit(E->getTrueExpr());
1251ce1de617SJohn McCall   eval.end(CGF);
12527a51313dSChris Lattner 
1253ce1de617SJohn McCall   assert(CGF.HaveInsertPoint() && "expression evaluation ended with no IP!");
1254ce1de617SJohn McCall   CGF.Builder.CreateBr(ContBlock);
12557a51313dSChris Lattner 
12565b26f65bSJohn McCall   // If the result of an agg expression is unused, then the emission
12575b26f65bSJohn McCall   // of the LHS might need to create a destination slot.  That's fine
12585b26f65bSJohn McCall   // with us, and we can safely emit the RHS into the same slot, but
1259cac93853SJohn McCall   // we shouldn't claim that it's already being destructed.
1260cac93853SJohn McCall   Dest.setExternallyDestructed(isExternallyDestructed);
12615b26f65bSJohn McCall 
1262ce1de617SJohn McCall   eval.begin(CGF);
1263ce1de617SJohn McCall   CGF.EmitBlock(RHSBlock);
1264c07a0c7eSJohn McCall   Visit(E->getFalseExpr());
1265ce1de617SJohn McCall   eval.end(CGF);
12667a51313dSChris Lattner 
126771e1a56dSAkira Hatanaka   if (destructNonTrivialCStruct)
126871e1a56dSAkira Hatanaka     CGF.pushDestroy(QualType::DK_nontrivial_c_struct, Dest.getAddress(),
126971e1a56dSAkira Hatanaka                     E->getType());
127071e1a56dSAkira Hatanaka 
12717a51313dSChris Lattner   CGF.EmitBlock(ContBlock);
12727a51313dSChris Lattner }
12737a51313dSChris Lattner 
12745b2095ceSAnders Carlsson void AggExprEmitter::VisitChooseExpr(const ChooseExpr *CE) {
127575807f23SEli Friedman   Visit(CE->getChosenSubExpr());
12765b2095ceSAnders Carlsson }
12775b2095ceSAnders Carlsson 
127821911e89SEli Friedman void AggExprEmitter::VisitVAArgExpr(VAArgExpr *VE) {
1279c7d5c94fSCharles Davis   Address ArgValue = Address::invalid();
1280c7d5c94fSCharles Davis   Address ArgPtr = CGF.EmitVAArg(VE, ArgValue);
128113abd7e9SAnders Carlsson 
128229b5f086SJames Y Knight   // If EmitVAArg fails, emit an error.
12837f416cc4SJohn McCall   if (!ArgPtr.isValid()) {
128429b5f086SJames Y Knight     CGF.ErrorUnsupported(VE, "aggregate va_arg expression");
1285020cddcfSSebastian Redl     return;
1286020cddcfSSebastian Redl   }
128713abd7e9SAnders Carlsson 
12884e8ca4faSJohn McCall   EmitFinalDestCopy(VE->getType(), CGF.MakeAddrLValue(ArgPtr, VE->getType()));
128921911e89SEli Friedman }
129021911e89SEli Friedman 
12913be22e27SAnders Carlsson void AggExprEmitter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
12927a626f63SJohn McCall   // Ensure that we have a slot, but if we already do, remember
1293cac93853SJohn McCall   // whether it was externally destructed.
1294cac93853SJohn McCall   bool wasExternallyDestructed = Dest.isExternallyDestructed();
12954e8ca4faSJohn McCall   EnsureDest(E->getType());
1296cac93853SJohn McCall 
1297cac93853SJohn McCall   // We're going to push a destructor if there isn't already one.
1298cac93853SJohn McCall   Dest.setExternallyDestructed();
12993be22e27SAnders Carlsson 
13003be22e27SAnders Carlsson   Visit(E->getSubExpr());
13013be22e27SAnders Carlsson 
1302cac93853SJohn McCall   // Push that destructor we promised.
1303cac93853SJohn McCall   if (!wasExternallyDestructed)
13047f416cc4SJohn McCall     CGF.EmitCXXTemporary(E->getTemporary(), E->getType(), Dest.getAddress());
13053be22e27SAnders Carlsson }
13063be22e27SAnders Carlsson 
1307b7f8f594SAnders Carlsson void
13081619a504SAnders Carlsson AggExprEmitter::VisitCXXConstructExpr(const CXXConstructExpr *E) {
13097a626f63SJohn McCall   AggValueSlot Slot = EnsureSlot(E->getType());
13107a626f63SJohn McCall   CGF.EmitCXXConstructExpr(E, Slot);
1311c82b86dfSAnders Carlsson }
1312c82b86dfSAnders Carlsson 
13135179eb78SRichard Smith void AggExprEmitter::VisitCXXInheritedCtorInitExpr(
13145179eb78SRichard Smith     const CXXInheritedCtorInitExpr *E) {
13155179eb78SRichard Smith   AggValueSlot Slot = EnsureSlot(E->getType());
13165179eb78SRichard Smith   CGF.EmitInheritedCXXConstructorCall(
13175179eb78SRichard Smith       E->getConstructor(), E->constructsVBase(), Slot.getAddress(),
13185179eb78SRichard Smith       E->inheritedFromVBase(), E);
13195179eb78SRichard Smith }
13205179eb78SRichard Smith 
1321c370a7eeSEli Friedman void
1322c370a7eeSEli Friedman AggExprEmitter::VisitLambdaExpr(LambdaExpr *E) {
1323c370a7eeSEli Friedman   AggValueSlot Slot = EnsureSlot(E->getType());
13240444006fSRichard Smith   LValue SlotLV = CGF.MakeAddrLValue(Slot.getAddress(), E->getType());
13250444006fSRichard Smith 
13260444006fSRichard Smith   // We'll need to enter cleanup scopes in case any of the element
13270444006fSRichard Smith   // initializers throws an exception.
13280444006fSRichard Smith   SmallVector<EHScopeStack::stable_iterator, 16> Cleanups;
13290444006fSRichard Smith   llvm::Instruction *CleanupDominator = nullptr;
13300444006fSRichard Smith 
13310444006fSRichard Smith   CXXRecordDecl::field_iterator CurField = E->getLambdaClass()->field_begin();
13320444006fSRichard Smith   for (LambdaExpr::const_capture_init_iterator i = E->capture_init_begin(),
13330444006fSRichard Smith                                                e = E->capture_init_end();
13340444006fSRichard Smith        i != e; ++i, ++CurField) {
13350444006fSRichard Smith     // Emit initialization
13360444006fSRichard Smith     LValue LV = CGF.EmitLValueForFieldInitialization(SlotLV, *CurField);
13370444006fSRichard Smith     if (CurField->hasCapturedVLAType()) {
13380444006fSRichard Smith       CGF.EmitLambdaVLACapture(CurField->getCapturedVLAType(), LV);
13390444006fSRichard Smith       continue;
13400444006fSRichard Smith     }
13410444006fSRichard Smith 
13420444006fSRichard Smith     EmitInitializationToLValue(*i, LV);
13430444006fSRichard Smith 
13440444006fSRichard Smith     // Push a destructor if necessary.
13450444006fSRichard Smith     if (QualType::DestructionKind DtorKind =
13460444006fSRichard Smith             CurField->getType().isDestructedType()) {
13470444006fSRichard Smith       assert(LV.isSimple());
13480444006fSRichard Smith       if (CGF.needsEHCleanup(DtorKind)) {
13490444006fSRichard Smith         if (!CleanupDominator)
13500444006fSRichard Smith           CleanupDominator = CGF.Builder.CreateAlignedLoad(
13510444006fSRichard Smith               CGF.Int8Ty,
13520444006fSRichard Smith               llvm::Constant::getNullValue(CGF.Int8PtrTy),
13530444006fSRichard Smith               CharUnits::One()); // placeholder
13540444006fSRichard Smith 
1355f139ae3dSAkira Hatanaka         CGF.pushDestroy(EHCleanup, LV.getAddress(CGF), CurField->getType(),
13560444006fSRichard Smith                         CGF.getDestroyer(DtorKind), false);
13570444006fSRichard Smith         Cleanups.push_back(CGF.EHStack.stable_begin());
13580444006fSRichard Smith       }
13590444006fSRichard Smith     }
13600444006fSRichard Smith   }
13610444006fSRichard Smith 
13620444006fSRichard Smith   // Deactivate all the partial cleanups in reverse order, which
13630444006fSRichard Smith   // generally means popping them.
13640444006fSRichard Smith   for (unsigned i = Cleanups.size(); i != 0; --i)
13650444006fSRichard Smith     CGF.DeactivateCleanupBlock(Cleanups[i-1], CleanupDominator);
13660444006fSRichard Smith 
13670444006fSRichard Smith   // Destroy the placeholder if we made one.
13680444006fSRichard Smith   if (CleanupDominator)
13690444006fSRichard Smith     CleanupDominator->eraseFromParent();
1370c370a7eeSEli Friedman }
1371c370a7eeSEli Friedman 
13725d413781SJohn McCall void AggExprEmitter::VisitExprWithCleanups(ExprWithCleanups *E) {
137308ef4660SJohn McCall   CodeGenFunction::RunCleanupsScope cleanups(CGF);
137408ef4660SJohn McCall   Visit(E->getSubExpr());
1375b7f8f594SAnders Carlsson }
1376b7f8f594SAnders Carlsson 
1377747eb784SDouglas Gregor void AggExprEmitter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
13787a626f63SJohn McCall   QualType T = E->getType();
13797a626f63SJohn McCall   AggValueSlot Slot = EnsureSlot(T);
13807f416cc4SJohn McCall   EmitNullInitializationToLValue(CGF.MakeAddrLValue(Slot.getAddress(), T));
138118ada985SAnders Carlsson }
138218ada985SAnders Carlsson 
138318ada985SAnders Carlsson void AggExprEmitter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
13847a626f63SJohn McCall   QualType T = E->getType();
13857a626f63SJohn McCall   AggValueSlot Slot = EnsureSlot(T);
13867f416cc4SJohn McCall   EmitNullInitializationToLValue(CGF.MakeAddrLValue(Slot.getAddress(), T));
1387ff3507b9SNuno Lopes }
1388ff3507b9SNuno Lopes 
1389d4aac678SRichard Smith /// Determine whether the given cast kind is known to always convert values
1390d4aac678SRichard Smith /// with all zero bits in their value representation to values with all zero
1391d4aac678SRichard Smith /// bits in their value representation.
1392d4aac678SRichard Smith static bool castPreservesZero(const CastExpr *CE) {
1393d4aac678SRichard Smith   switch (CE->getCastKind()) {
1394d4aac678SRichard Smith     // No-ops.
1395d4aac678SRichard Smith   case CK_NoOp:
1396d4aac678SRichard Smith   case CK_UserDefinedConversion:
1397d4aac678SRichard Smith   case CK_ConstructorConversion:
1398d4aac678SRichard Smith   case CK_BitCast:
1399d4aac678SRichard Smith   case CK_ToUnion:
1400d4aac678SRichard Smith   case CK_ToVoid:
1401d4aac678SRichard Smith     // Conversions between (possibly-complex) integral, (possibly-complex)
1402d4aac678SRichard Smith     // floating-point, and bool.
1403d4aac678SRichard Smith   case CK_BooleanToSignedIntegral:
1404d4aac678SRichard Smith   case CK_FloatingCast:
1405d4aac678SRichard Smith   case CK_FloatingComplexCast:
1406d4aac678SRichard Smith   case CK_FloatingComplexToBoolean:
1407d4aac678SRichard Smith   case CK_FloatingComplexToIntegralComplex:
1408d4aac678SRichard Smith   case CK_FloatingComplexToReal:
1409d4aac678SRichard Smith   case CK_FloatingRealToComplex:
1410d4aac678SRichard Smith   case CK_FloatingToBoolean:
1411d4aac678SRichard Smith   case CK_FloatingToIntegral:
1412d4aac678SRichard Smith   case CK_IntegralCast:
1413d4aac678SRichard Smith   case CK_IntegralComplexCast:
1414d4aac678SRichard Smith   case CK_IntegralComplexToBoolean:
1415d4aac678SRichard Smith   case CK_IntegralComplexToFloatingComplex:
1416d4aac678SRichard Smith   case CK_IntegralComplexToReal:
1417d4aac678SRichard Smith   case CK_IntegralRealToComplex:
1418d4aac678SRichard Smith   case CK_IntegralToBoolean:
1419d4aac678SRichard Smith   case CK_IntegralToFloating:
1420d4aac678SRichard Smith     // Reinterpreting integers as pointers and vice versa.
1421d4aac678SRichard Smith   case CK_IntegralToPointer:
1422d4aac678SRichard Smith   case CK_PointerToIntegral:
1423d4aac678SRichard Smith     // Language extensions.
1424d4aac678SRichard Smith   case CK_VectorSplat:
1425d4aac678SRichard Smith   case CK_NonAtomicToAtomic:
1426d4aac678SRichard Smith   case CK_AtomicToNonAtomic:
1427d4aac678SRichard Smith     return true;
1428d4aac678SRichard Smith 
1429d4aac678SRichard Smith   case CK_BaseToDerivedMemberPointer:
1430d4aac678SRichard Smith   case CK_DerivedToBaseMemberPointer:
1431d4aac678SRichard Smith   case CK_MemberPointerToBoolean:
1432d4aac678SRichard Smith   case CK_NullToMemberPointer:
1433d4aac678SRichard Smith   case CK_ReinterpretMemberPointer:
1434d4aac678SRichard Smith     // FIXME: ABI-dependent.
1435d4aac678SRichard Smith     return false;
1436d4aac678SRichard Smith 
1437d4aac678SRichard Smith   case CK_AnyPointerToBlockPointerCast:
1438d4aac678SRichard Smith   case CK_BlockPointerToObjCPointerCast:
1439d4aac678SRichard Smith   case CK_CPointerToObjCPointerCast:
1440d4aac678SRichard Smith   case CK_ObjCObjectLValueCast:
1441d4aac678SRichard Smith   case CK_IntToOCLSampler:
1442d4aac678SRichard Smith   case CK_ZeroToOCLOpaqueType:
1443d4aac678SRichard Smith     // FIXME: Check these.
1444d4aac678SRichard Smith     return false;
1445d4aac678SRichard Smith 
1446d4aac678SRichard Smith   case CK_FixedPointCast:
1447d4aac678SRichard Smith   case CK_FixedPointToBoolean:
1448d4aac678SRichard Smith   case CK_FixedPointToFloating:
1449d4aac678SRichard Smith   case CK_FixedPointToIntegral:
1450d4aac678SRichard Smith   case CK_FloatingToFixedPoint:
1451d4aac678SRichard Smith   case CK_IntegralToFixedPoint:
1452d4aac678SRichard Smith     // FIXME: Do all fixed-point types represent zero as all 0 bits?
1453d4aac678SRichard Smith     return false;
1454d4aac678SRichard Smith 
1455d4aac678SRichard Smith   case CK_AddressSpaceConversion:
1456d4aac678SRichard Smith   case CK_BaseToDerived:
1457d4aac678SRichard Smith   case CK_DerivedToBase:
1458d4aac678SRichard Smith   case CK_Dynamic:
1459d4aac678SRichard Smith   case CK_NullToPointer:
1460d4aac678SRichard Smith   case CK_PointerToBoolean:
1461d4aac678SRichard Smith     // FIXME: Preserves zeroes only if zero pointers and null pointers have the
1462d4aac678SRichard Smith     // same representation in all involved address spaces.
1463d4aac678SRichard Smith     return false;
1464d4aac678SRichard Smith 
1465d4aac678SRichard Smith   case CK_ARCConsumeObject:
1466d4aac678SRichard Smith   case CK_ARCExtendBlockObject:
1467d4aac678SRichard Smith   case CK_ARCProduceObject:
1468d4aac678SRichard Smith   case CK_ARCReclaimReturnedObject:
1469d4aac678SRichard Smith   case CK_CopyAndAutoreleaseBlockObject:
1470d4aac678SRichard Smith   case CK_ArrayToPointerDecay:
1471d4aac678SRichard Smith   case CK_FunctionToPointerDecay:
1472d4aac678SRichard Smith   case CK_BuiltinFnToFnPtr:
1473d4aac678SRichard Smith   case CK_Dependent:
1474d4aac678SRichard Smith   case CK_LValueBitCast:
1475d4aac678SRichard Smith   case CK_LValueToRValue:
1476d4aac678SRichard Smith   case CK_LValueToRValueBitCast:
1477d4aac678SRichard Smith   case CK_UncheckedDerivedToBase:
1478d4aac678SRichard Smith     return false;
1479d4aac678SRichard Smith   }
14807fe7d9b1SSimon Pilgrim   llvm_unreachable("Unhandled clang::CastKind enum");
1481d4aac678SRichard Smith }
1482d4aac678SRichard Smith 
148327a3631bSChris Lattner /// isSimpleZero - If emitting this value will obviously just cause a store of
148427a3631bSChris Lattner /// zero to memory, return true.  This can return false if uncertain, so it just
148527a3631bSChris Lattner /// handles simple cases.
148627a3631bSChris Lattner static bool isSimpleZero(const Expr *E, CodeGenFunction &CGF) {
148791147596SPeter Collingbourne   E = E->IgnoreParens();
1488d4aac678SRichard Smith   while (auto *CE = dyn_cast<CastExpr>(E)) {
1489d4aac678SRichard Smith     if (!castPreservesZero(CE))
1490d4aac678SRichard Smith       break;
1491d4aac678SRichard Smith     E = CE->getSubExpr()->IgnoreParens();
1492d4aac678SRichard Smith   }
149391147596SPeter Collingbourne 
149427a3631bSChris Lattner   // 0
149527a3631bSChris Lattner   if (const IntegerLiteral *IL = dyn_cast<IntegerLiteral>(E))
149627a3631bSChris Lattner     return IL->getValue() == 0;
149727a3631bSChris Lattner   // +0.0
149827a3631bSChris Lattner   if (const FloatingLiteral *FL = dyn_cast<FloatingLiteral>(E))
149927a3631bSChris Lattner     return FL->getValue().isPosZero();
150027a3631bSChris Lattner   // int()
150127a3631bSChris Lattner   if ((isa<ImplicitValueInitExpr>(E) || isa<CXXScalarValueInitExpr>(E)) &&
150227a3631bSChris Lattner       CGF.getTypes().isZeroInitializable(E->getType()))
150327a3631bSChris Lattner     return true;
150427a3631bSChris Lattner   // (int*)0 - Null pointer expressions.
150527a3631bSChris Lattner   if (const CastExpr *ICE = dyn_cast<CastExpr>(E))
1506402804b6SYaxun Liu     return ICE->getCastKind() == CK_NullToPointer &&
150727252a1fSRichard Smith            CGF.getTypes().isPointerZeroInitializable(E->getType()) &&
150827252a1fSRichard Smith            !E->HasSideEffects(CGF.getContext());
150927a3631bSChris Lattner   // '\0'
151027a3631bSChris Lattner   if (const CharacterLiteral *CL = dyn_cast<CharacterLiteral>(E))
151127a3631bSChris Lattner     return CL->getValue() == 0;
151227a3631bSChris Lattner 
151327a3631bSChris Lattner   // Otherwise, hard case: conservatively return false.
151427a3631bSChris Lattner   return false;
151527a3631bSChris Lattner }
151627a3631bSChris Lattner 
151727a3631bSChris Lattner 
1518b247350eSAnders Carlsson void
1519615ed1a3SChad Rosier AggExprEmitter::EmitInitializationToLValue(Expr *E, LValue LV) {
15201553b190SJohn McCall   QualType type = LV.getType();
1521df0fe27bSMike Stump   // FIXME: Ignore result?
1522579a05d7SChris Lattner   // FIXME: Are initializers affected by volatile?
152327a3631bSChris Lattner   if (Dest.isZeroed() && isSimpleZero(E, CGF)) {
152427a3631bSChris Lattner     // Storing "i32 0" to a zero'd memory location is a noop.
152547fb9508SJohn McCall     return;
1526d82a2ce3SRichard Smith   } else if (isa<ImplicitValueInitExpr>(E) || isa<CXXScalarValueInitExpr>(E)) {
152747fb9508SJohn McCall     return EmitNullInitializationToLValue(LV);
1528cb77930dSYunzhong Gao   } else if (isa<NoInitExpr>(E)) {
1529cb77930dSYunzhong Gao     // Do nothing.
1530cb77930dSYunzhong Gao     return;
15311553b190SJohn McCall   } else if (type->isReferenceType()) {
1532a1c9d4d9SRichard Smith     RValue RV = CGF.EmitReferenceBindingToExpr(E);
153347fb9508SJohn McCall     return CGF.EmitStoreThroughLValue(RV, LV);
153447fb9508SJohn McCall   }
153547fb9508SJohn McCall 
153647fb9508SJohn McCall   switch (CGF.getEvaluationKind(type)) {
153747fb9508SJohn McCall   case TEK_Complex:
153847fb9508SJohn McCall     CGF.EmitComplexExprIntoLValue(E, LV, /*isInit*/ true);
153947fb9508SJohn McCall     return;
154047fb9508SJohn McCall   case TEK_Aggregate:
1541f139ae3dSAkira Hatanaka     CGF.EmitAggExpr(
1542f139ae3dSAkira Hatanaka         E, AggValueSlot::forLValue(LV, CGF, AggValueSlot::IsDestructed,
15438d6fc958SJohn McCall                                    AggValueSlot::DoesNotNeedGCBarriers,
1544a5efa738SJohn McCall                                    AggValueSlot::IsNotAliased,
1545f139ae3dSAkira Hatanaka                                    AggValueSlot::MayOverlap, Dest.isZeroed()));
154647fb9508SJohn McCall     return;
154747fb9508SJohn McCall   case TEK_Scalar:
154847fb9508SJohn McCall     if (LV.isSimple()) {
15498a13c418SCraig Topper       CGF.EmitScalarInit(E, /*D=*/nullptr, LV, /*Captured=*/false);
15506e313210SEli Friedman     } else {
155155e1fbc8SJohn McCall       CGF.EmitStoreThroughLValue(RValue::get(CGF.EmitScalarExpr(E)), LV);
15527a51313dSChris Lattner     }
155347fb9508SJohn McCall     return;
155447fb9508SJohn McCall   }
155547fb9508SJohn McCall   llvm_unreachable("bad evaluation kind");
1556579a05d7SChris Lattner }
1557579a05d7SChris Lattner 
15581553b190SJohn McCall void AggExprEmitter::EmitNullInitializationToLValue(LValue lv) {
15591553b190SJohn McCall   QualType type = lv.getType();
15601553b190SJohn McCall 
156127a3631bSChris Lattner   // If the destination slot is already zeroed out before the aggregate is
156227a3631bSChris Lattner   // copied into it, we don't have to emit any zeros here.
15631553b190SJohn McCall   if (Dest.isZeroed() && CGF.getTypes().isZeroInitializable(type))
156427a3631bSChris Lattner     return;
156527a3631bSChris Lattner 
156647fb9508SJohn McCall   if (CGF.hasScalarEvaluationKind(type)) {
1567d82a2ce3SRichard Smith     // For non-aggregates, we can store the appropriate null constant.
1568d82a2ce3SRichard Smith     llvm::Value *null = CGF.CGM.EmitNullConstant(type);
156991d5bb1eSEli Friedman     // Note that the following is not equivalent to
157091d5bb1eSEli Friedman     // EmitStoreThroughBitfieldLValue for ARC types.
1571cb3785e4SEli Friedman     if (lv.isBitField()) {
157291d5bb1eSEli Friedman       CGF.EmitStoreThroughBitfieldLValue(RValue::get(null), lv);
1573cb3785e4SEli Friedman     } else {
157491d5bb1eSEli Friedman       assert(lv.isSimple());
157591d5bb1eSEli Friedman       CGF.EmitStoreOfScalar(null, lv, /* isInitialization */ true);
1576cb3785e4SEli Friedman     }
1577579a05d7SChris Lattner   } else {
1578579a05d7SChris Lattner     // There's a potential optimization opportunity in combining
1579579a05d7SChris Lattner     // memsets; that would be easy for arrays, but relatively
1580579a05d7SChris Lattner     // difficult for structures with the current code.
1581f139ae3dSAkira Hatanaka     CGF.EmitNullInitialization(lv.getAddress(CGF), lv.getType());
1582579a05d7SChris Lattner   }
1583579a05d7SChris Lattner }
1584579a05d7SChris Lattner 
1585579a05d7SChris Lattner void AggExprEmitter::VisitInitListExpr(InitListExpr *E) {
1586f5d08c9eSEli Friedman #if 0
15876d11ec8cSEli Friedman   // FIXME: Assess perf here?  Figure out what cases are worth optimizing here
15886d11ec8cSEli Friedman   // (Length of globals? Chunks of zeroed-out space?).
1589f5d08c9eSEli Friedman   //
159018bb9284SMike Stump   // If we can, prefer a copy from a global; this is a lot less code for long
159118bb9284SMike Stump   // globals, and it's easier for the current optimizers to analyze.
15926d11ec8cSEli Friedman   if (llvm::Constant* C = CGF.CGM.EmitConstantExpr(E, E->getType(), &CGF)) {
1593c59bb48eSEli Friedman     llvm::GlobalVariable* GV =
15946d11ec8cSEli Friedman     new llvm::GlobalVariable(CGF.CGM.getModule(), C->getType(), true,
15956d11ec8cSEli Friedman                              llvm::GlobalValue::InternalLinkage, C, "");
15964e8ca4faSJohn McCall     EmitFinalDestCopy(E->getType(), CGF.MakeAddrLValue(GV, E->getType()));
1597c59bb48eSEli Friedman     return;
1598c59bb48eSEli Friedman   }
1599f5d08c9eSEli Friedman #endif
1600f53c0968SChris Lattner   if (E->hadArrayRangeDesignator())
1601bf7207a1SDouglas Gregor     CGF.ErrorUnsupported(E, "GNU array range designator extension");
1602bf7207a1SDouglas Gregor 
1603122f88d4SRichard Smith   if (E->isTransparent())
1604122f88d4SRichard Smith     return Visit(E->getInit(0));
1605122f88d4SRichard Smith 
1606be93c00aSRichard Smith   AggValueSlot Dest = EnsureSlot(E->getType());
1607be93c00aSRichard Smith 
16087f416cc4SJohn McCall   LValue DestLV = CGF.MakeAddrLValue(Dest.getAddress(), E->getType());
16097a626f63SJohn McCall 
1610579a05d7SChris Lattner   // Handle initialization of an array.
1611579a05d7SChris Lattner   if (E->getType()->isArrayType()) {
16127f416cc4SJohn McCall     auto AType = cast<llvm::ArrayType>(Dest.getAddress().getElementType());
1613e0ef348cSIvan A. Kosarev     EmitArrayInit(Dest.getAddress(), AType, E->getType(), E);
1614579a05d7SChris Lattner     return;
1615579a05d7SChris Lattner   }
1616579a05d7SChris Lattner 
1617579a05d7SChris Lattner   assert(E->getType()->isRecordType() && "Only support structs/unions here!");
1618579a05d7SChris Lattner 
1619579a05d7SChris Lattner   // Do struct initialization; this code just sets each individual member
1620579a05d7SChris Lattner   // to the approprate value.  This makes bitfield support automatic;
1621579a05d7SChris Lattner   // the disadvantage is that the generated code is more difficult for
1622579a05d7SChris Lattner   // the optimizer, especially with bitfields.
1623579a05d7SChris Lattner   unsigned NumInitElements = E->getNumInits();
16243b935d33SJohn McCall   RecordDecl *record = E->getType()->castAs<RecordType>()->getDecl();
162552bcf963SChris Lattner 
1626872307e2SRichard Smith   // We'll need to enter cleanup scopes in case any of the element
1627872307e2SRichard Smith   // initializers throws an exception.
1628872307e2SRichard Smith   SmallVector<EHScopeStack::stable_iterator, 16> cleanups;
1629872307e2SRichard Smith   llvm::Instruction *cleanupDominator = nullptr;
16303bdb7a90SSaleem Abdulrasool   auto addCleanup = [&](const EHScopeStack::stable_iterator &cleanup) {
16313bdb7a90SSaleem Abdulrasool     cleanups.push_back(cleanup);
16323bdb7a90SSaleem Abdulrasool     if (!cleanupDominator) // create placeholder once needed
16333bdb7a90SSaleem Abdulrasool       cleanupDominator = CGF.Builder.CreateAlignedLoad(
16343bdb7a90SSaleem Abdulrasool           CGF.Int8Ty, llvm::Constant::getNullValue(CGF.Int8PtrTy),
16353bdb7a90SSaleem Abdulrasool           CharUnits::One());
16363bdb7a90SSaleem Abdulrasool   };
1637872307e2SRichard Smith 
1638872307e2SRichard Smith   unsigned curInitIndex = 0;
1639872307e2SRichard Smith 
1640872307e2SRichard Smith   // Emit initialization of base classes.
1641872307e2SRichard Smith   if (auto *CXXRD = dyn_cast<CXXRecordDecl>(record)) {
1642872307e2SRichard Smith     assert(E->getNumInits() >= CXXRD->getNumBases() &&
1643872307e2SRichard Smith            "missing initializer for base class");
1644872307e2SRichard Smith     for (auto &Base : CXXRD->bases()) {
1645872307e2SRichard Smith       assert(!Base.isVirtual() && "should not see vbases here");
1646872307e2SRichard Smith       auto *BaseRD = Base.getType()->getAsCXXRecordDecl();
1647872307e2SRichard Smith       Address V = CGF.GetAddressOfDirectBaseInCompleteClass(
1648872307e2SRichard Smith           Dest.getAddress(), CXXRD, BaseRD,
1649872307e2SRichard Smith           /*isBaseVirtual*/ false);
1650e78fac51SRichard Smith       AggValueSlot AggSlot = AggValueSlot::forAddr(
1651e78fac51SRichard Smith           V, Qualifiers(),
1652872307e2SRichard Smith           AggValueSlot::IsDestructed,
1653872307e2SRichard Smith           AggValueSlot::DoesNotNeedGCBarriers,
1654e78fac51SRichard Smith           AggValueSlot::IsNotAliased,
16558cca3a5aSRichard Smith           CGF.getOverlapForBaseInit(CXXRD, BaseRD, Base.isVirtual()));
1656872307e2SRichard Smith       CGF.EmitAggExpr(E->getInit(curInitIndex++), AggSlot);
1657872307e2SRichard Smith 
1658872307e2SRichard Smith       if (QualType::DestructionKind dtorKind =
1659872307e2SRichard Smith               Base.getType().isDestructedType()) {
1660872307e2SRichard Smith         CGF.pushDestroy(dtorKind, V, Base.getType());
16613bdb7a90SSaleem Abdulrasool         addCleanup(CGF.EHStack.stable_begin());
1662872307e2SRichard Smith       }
1663872307e2SRichard Smith     }
1664872307e2SRichard Smith   }
1665872307e2SRichard Smith 
1666852c9db7SRichard Smith   // Prepare a 'this' for CXXDefaultInitExprs.
16677f416cc4SJohn McCall   CodeGenFunction::FieldConstructionScope FCS(CGF, Dest.getAddress());
1668852c9db7SRichard Smith 
16693b935d33SJohn McCall   if (record->isUnion()) {
16705169570eSDouglas Gregor     // Only initialize one field of a union. The field itself is
16715169570eSDouglas Gregor     // specified by the initializer list.
16725169570eSDouglas Gregor     if (!E->getInitializedFieldInUnion()) {
16735169570eSDouglas Gregor       // Empty union; we have nothing to do.
16745169570eSDouglas Gregor 
16755169570eSDouglas Gregor #ifndef NDEBUG
16765169570eSDouglas Gregor       // Make sure that it's really an empty and not a failure of
16775169570eSDouglas Gregor       // semantic analysis.
1678e8a8baefSAaron Ballman       for (const auto *Field : record->fields())
16795169570eSDouglas Gregor         assert(Field->isUnnamedBitfield() && "Only unnamed bitfields allowed");
16805169570eSDouglas Gregor #endif
16815169570eSDouglas Gregor       return;
16825169570eSDouglas Gregor     }
16835169570eSDouglas Gregor 
16845169570eSDouglas Gregor     // FIXME: volatility
16855169570eSDouglas Gregor     FieldDecl *Field = E->getInitializedFieldInUnion();
16865169570eSDouglas Gregor 
16877f1ff600SEli Friedman     LValue FieldLoc = CGF.EmitLValueForFieldInitialization(DestLV, Field);
16885169570eSDouglas Gregor     if (NumInitElements) {
16895169570eSDouglas Gregor       // Store the initializer into the field
1690615ed1a3SChad Rosier       EmitInitializationToLValue(E->getInit(0), FieldLoc);
16915169570eSDouglas Gregor     } else {
169227a3631bSChris Lattner       // Default-initialize to null.
16931553b190SJohn McCall       EmitNullInitializationToLValue(FieldLoc);
16945169570eSDouglas Gregor     }
16955169570eSDouglas Gregor 
16965169570eSDouglas Gregor     return;
16975169570eSDouglas Gregor   }
1698579a05d7SChris Lattner 
1699579a05d7SChris Lattner   // Here we iterate over the fields; this makes it simpler to both
1700579a05d7SChris Lattner   // default-initialize fields and skip over unnamed fields.
1701e8a8baefSAaron Ballman   for (const auto *field : record->fields()) {
17023b935d33SJohn McCall     // We're done once we hit the flexible array member.
17033b935d33SJohn McCall     if (field->getType()->isIncompleteArrayType())
170491f84216SDouglas Gregor       break;
170591f84216SDouglas Gregor 
17063b935d33SJohn McCall     // Always skip anonymous bitfields.
17073b935d33SJohn McCall     if (field->isUnnamedBitfield())
1708579a05d7SChris Lattner       continue;
170917bd094aSDouglas Gregor 
17103b935d33SJohn McCall     // We're done if we reach the end of the explicit initializers, we
17113b935d33SJohn McCall     // have a zeroed object, and the rest of the fields are
17123b935d33SJohn McCall     // zero-initializable.
17133b935d33SJohn McCall     if (curInitIndex == NumInitElements && Dest.isZeroed() &&
171427a3631bSChris Lattner         CGF.getTypes().isZeroInitializable(E->getType()))
171527a3631bSChris Lattner       break;
171627a3631bSChris Lattner 
17177f1ff600SEli Friedman 
1718e8a8baefSAaron Ballman     LValue LV = CGF.EmitLValueForFieldInitialization(DestLV, field);
17197c1baf46SFariborz Jahanian     // We never generate write-barries for initialized fields.
17203b935d33SJohn McCall     LV.setNonGC(true);
172127a3631bSChris Lattner 
17223b935d33SJohn McCall     if (curInitIndex < NumInitElements) {
1723e18aaf2cSChris Lattner       // Store the initializer into the field.
1724615ed1a3SChad Rosier       EmitInitializationToLValue(E->getInit(curInitIndex++), LV);
1725579a05d7SChris Lattner     } else {
17262c51880aSSimon Pilgrim       // We're out of initializers; default-initialize to null
17273b935d33SJohn McCall       EmitNullInitializationToLValue(LV);
17283b935d33SJohn McCall     }
17293b935d33SJohn McCall 
17303b935d33SJohn McCall     // Push a destructor if necessary.
17313b935d33SJohn McCall     // FIXME: if we have an array of structures, all explicitly
17323b935d33SJohn McCall     // initialized, we can end up pushing a linear number of cleanups.
17333b935d33SJohn McCall     bool pushedCleanup = false;
17343b935d33SJohn McCall     if (QualType::DestructionKind dtorKind
17353b935d33SJohn McCall           = field->getType().isDestructedType()) {
17363b935d33SJohn McCall       assert(LV.isSimple());
17373b935d33SJohn McCall       if (CGF.needsEHCleanup(dtorKind)) {
1738f139ae3dSAkira Hatanaka         CGF.pushDestroy(EHCleanup, LV.getAddress(CGF), field->getType(),
17393b935d33SJohn McCall                         CGF.getDestroyer(dtorKind), false);
17403bdb7a90SSaleem Abdulrasool         addCleanup(CGF.EHStack.stable_begin());
17413b935d33SJohn McCall         pushedCleanup = true;
17423b935d33SJohn McCall       }
1743579a05d7SChris Lattner     }
174427a3631bSChris Lattner 
174527a3631bSChris Lattner     // If the GEP didn't get used because of a dead zero init or something
174627a3631bSChris Lattner     // else, clean it up for -O0 builds and general tidiness.
17473b935d33SJohn McCall     if (!pushedCleanup && LV.isSimple())
174827a3631bSChris Lattner       if (llvm::GetElementPtrInst *GEP =
1749f139ae3dSAkira Hatanaka               dyn_cast<llvm::GetElementPtrInst>(LV.getPointer(CGF)))
175027a3631bSChris Lattner         if (GEP->use_empty())
175127a3631bSChris Lattner           GEP->eraseFromParent();
17527a51313dSChris Lattner   }
17533b935d33SJohn McCall 
17543b935d33SJohn McCall   // Deactivate all the partial cleanups in reverse order, which
17553b935d33SJohn McCall   // generally means popping them.
17563bdb7a90SSaleem Abdulrasool   assert((cleanupDominator || cleanups.empty()) &&
17573bdb7a90SSaleem Abdulrasool          "Missing cleanupDominator before deactivating cleanup blocks");
17583b935d33SJohn McCall   for (unsigned i = cleanups.size(); i != 0; --i)
1759f4beacd0SJohn McCall     CGF.DeactivateCleanupBlock(cleanups[i-1], cleanupDominator);
1760f4beacd0SJohn McCall 
1761f4beacd0SJohn McCall   // Destroy the placeholder if we made one.
1762f4beacd0SJohn McCall   if (cleanupDominator)
1763f4beacd0SJohn McCall     cleanupDominator->eraseFromParent();
17647a51313dSChris Lattner }
17657a51313dSChris Lattner 
1766939b6880SRichard Smith void AggExprEmitter::VisitArrayInitLoopExpr(const ArrayInitLoopExpr *E,
1767939b6880SRichard Smith                                             llvm::Value *outerBegin) {
1768410306bfSRichard Smith   // Emit the common subexpression.
1769410306bfSRichard Smith   CodeGenFunction::OpaqueValueMapping binding(CGF, E->getCommonExpr());
1770410306bfSRichard Smith 
1771410306bfSRichard Smith   Address destPtr = EnsureSlot(E->getType()).getAddress();
1772410306bfSRichard Smith   uint64_t numElements = E->getArraySize().getZExtValue();
1773410306bfSRichard Smith 
1774410306bfSRichard Smith   if (!numElements)
1775410306bfSRichard Smith     return;
1776410306bfSRichard Smith 
1777410306bfSRichard Smith   // destPtr is an array*. Construct an elementType* by drilling down a level.
1778410306bfSRichard Smith   llvm::Value *zero = llvm::ConstantInt::get(CGF.SizeTy, 0);
1779410306bfSRichard Smith   llvm::Value *indices[] = {zero, zero};
1780410306bfSRichard Smith   llvm::Value *begin = Builder.CreateInBoundsGEP(destPtr.getPointer(), indices,
1781410306bfSRichard Smith                                                  "arrayinit.begin");
1782410306bfSRichard Smith 
1783939b6880SRichard Smith   // Prepare to special-case multidimensional array initialization: we avoid
1784939b6880SRichard Smith   // emitting multiple destructor loops in that case.
1785939b6880SRichard Smith   if (!outerBegin)
1786939b6880SRichard Smith     outerBegin = begin;
1787939b6880SRichard Smith   ArrayInitLoopExpr *InnerLoop = dyn_cast<ArrayInitLoopExpr>(E->getSubExpr());
1788939b6880SRichard Smith 
178930e304e2SRichard Smith   QualType elementType =
179030e304e2SRichard Smith       CGF.getContext().getAsArrayType(E->getType())->getElementType();
1791410306bfSRichard Smith   CharUnits elementSize = CGF.getContext().getTypeSizeInChars(elementType);
1792410306bfSRichard Smith   CharUnits elementAlign =
1793410306bfSRichard Smith       destPtr.getAlignment().alignmentOfArrayElement(elementSize);
1794410306bfSRichard Smith 
1795410306bfSRichard Smith   llvm::BasicBlock *entryBB = Builder.GetInsertBlock();
1796410306bfSRichard Smith   llvm::BasicBlock *bodyBB = CGF.createBasicBlock("arrayinit.body");
1797410306bfSRichard Smith 
1798410306bfSRichard Smith   // Jump into the body.
1799410306bfSRichard Smith   CGF.EmitBlock(bodyBB);
1800410306bfSRichard Smith   llvm::PHINode *index =
1801410306bfSRichard Smith       Builder.CreatePHI(zero->getType(), 2, "arrayinit.index");
1802410306bfSRichard Smith   index->addIncoming(zero, entryBB);
1803410306bfSRichard Smith   llvm::Value *element = Builder.CreateInBoundsGEP(begin, index);
1804410306bfSRichard Smith 
180530e304e2SRichard Smith   // Prepare for a cleanup.
180630e304e2SRichard Smith   QualType::DestructionKind dtorKind = elementType.isDestructedType();
180730e304e2SRichard Smith   EHScopeStack::stable_iterator cleanup;
1808939b6880SRichard Smith   if (CGF.needsEHCleanup(dtorKind) && !InnerLoop) {
1809939b6880SRichard Smith     if (outerBegin->getType() != element->getType())
1810939b6880SRichard Smith       outerBegin = Builder.CreateBitCast(outerBegin, element->getType());
1811939b6880SRichard Smith     CGF.pushRegularPartialArrayCleanup(outerBegin, element, elementType,
1812939b6880SRichard Smith                                        elementAlign,
1813939b6880SRichard Smith                                        CGF.getDestroyer(dtorKind));
181430e304e2SRichard Smith     cleanup = CGF.EHStack.stable_begin();
181530e304e2SRichard Smith   } else {
181630e304e2SRichard Smith     dtorKind = QualType::DK_none;
181730e304e2SRichard Smith   }
1818410306bfSRichard Smith 
1819410306bfSRichard Smith   // Emit the actual filler expression.
1820410306bfSRichard Smith   {
182130e304e2SRichard Smith     // Temporaries created in an array initialization loop are destroyed
182230e304e2SRichard Smith     // at the end of each iteration.
182330e304e2SRichard Smith     CodeGenFunction::RunCleanupsScope CleanupsScope(CGF);
1824410306bfSRichard Smith     CodeGenFunction::ArrayInitLoopExprScope Scope(CGF, index);
1825410306bfSRichard Smith     LValue elementLV =
1826410306bfSRichard Smith         CGF.MakeAddrLValue(Address(element, elementAlign), elementType);
1827939b6880SRichard Smith 
1828939b6880SRichard Smith     if (InnerLoop) {
1829939b6880SRichard Smith       // If the subexpression is an ArrayInitLoopExpr, share its cleanup.
1830939b6880SRichard Smith       auto elementSlot = AggValueSlot::forLValue(
1831f139ae3dSAkira Hatanaka           elementLV, CGF, AggValueSlot::IsDestructed,
1832f139ae3dSAkira Hatanaka           AggValueSlot::DoesNotNeedGCBarriers, AggValueSlot::IsNotAliased,
1833e78fac51SRichard Smith           AggValueSlot::DoesNotOverlap);
1834939b6880SRichard Smith       AggExprEmitter(CGF, elementSlot, false)
1835939b6880SRichard Smith           .VisitArrayInitLoopExpr(InnerLoop, outerBegin);
1836939b6880SRichard Smith     } else
1837410306bfSRichard Smith       EmitInitializationToLValue(E->getSubExpr(), elementLV);
1838410306bfSRichard Smith   }
1839410306bfSRichard Smith 
1840410306bfSRichard Smith   // Move on to the next element.
1841410306bfSRichard Smith   llvm::Value *nextIndex = Builder.CreateNUWAdd(
1842410306bfSRichard Smith       index, llvm::ConstantInt::get(CGF.SizeTy, 1), "arrayinit.next");
1843410306bfSRichard Smith   index->addIncoming(nextIndex, Builder.GetInsertBlock());
1844410306bfSRichard Smith 
1845410306bfSRichard Smith   // Leave the loop if we're done.
1846410306bfSRichard Smith   llvm::Value *done = Builder.CreateICmpEQ(
1847410306bfSRichard Smith       nextIndex, llvm::ConstantInt::get(CGF.SizeTy, numElements),
1848410306bfSRichard Smith       "arrayinit.done");
1849410306bfSRichard Smith   llvm::BasicBlock *endBB = CGF.createBasicBlock("arrayinit.end");
1850410306bfSRichard Smith   Builder.CreateCondBr(done, endBB, bodyBB);
1851410306bfSRichard Smith 
1852410306bfSRichard Smith   CGF.EmitBlock(endBB);
1853410306bfSRichard Smith 
1854410306bfSRichard Smith   // Leave the partial-array cleanup if we entered one.
185530e304e2SRichard Smith   if (dtorKind)
185630e304e2SRichard Smith     CGF.DeactivateCleanupBlock(cleanup, index);
1857410306bfSRichard Smith }
1858410306bfSRichard Smith 
1859cb77930dSYunzhong Gao void AggExprEmitter::VisitDesignatedInitUpdateExpr(DesignatedInitUpdateExpr *E) {
1860cb77930dSYunzhong Gao   AggValueSlot Dest = EnsureSlot(E->getType());
1861cb77930dSYunzhong Gao 
18627f416cc4SJohn McCall   LValue DestLV = CGF.MakeAddrLValue(Dest.getAddress(), E->getType());
1863cb77930dSYunzhong Gao   EmitInitializationToLValue(E->getBase(), DestLV);
1864cb77930dSYunzhong Gao   VisitInitListExpr(E->getUpdater());
1865cb77930dSYunzhong Gao }
1866cb77930dSYunzhong Gao 
18677a51313dSChris Lattner //===----------------------------------------------------------------------===//
18687a51313dSChris Lattner //                        Entry Points into this File
18697a51313dSChris Lattner //===----------------------------------------------------------------------===//
18707a51313dSChris Lattner 
187127a3631bSChris Lattner /// GetNumNonZeroBytesInInit - Get an approximate count of the number of
187227a3631bSChris Lattner /// non-zero bytes that will be stored when outputting the initializer for the
187327a3631bSChris Lattner /// specified initializer expression.
1874df94cb7dSKen Dyck static CharUnits GetNumNonZeroBytesInInit(const Expr *E, CodeGenFunction &CGF) {
187548c70c16SRichard Smith   if (auto *MTE = dyn_cast<MaterializeTemporaryExpr>(E))
187648c70c16SRichard Smith     E = MTE->getSubExpr();
187748c70c16SRichard Smith   E = E->IgnoreParenNoopCasts(CGF.getContext());
187827a3631bSChris Lattner 
187927a3631bSChris Lattner   // 0 and 0.0 won't require any non-zero stores!
1880df94cb7dSKen Dyck   if (isSimpleZero(E, CGF)) return CharUnits::Zero();
188127a3631bSChris Lattner 
188227a3631bSChris Lattner   // If this is an initlist expr, sum up the size of sizes of the (present)
188327a3631bSChris Lattner   // elements.  If this is something weird, assume the whole thing is non-zero.
188427a3631bSChris Lattner   const InitListExpr *ILE = dyn_cast<InitListExpr>(E);
18858fa638aeSRichard Smith   while (ILE && ILE->isTransparent())
18868fa638aeSRichard Smith     ILE = dyn_cast<InitListExpr>(ILE->getInit(0));
18878a13c418SCraig Topper   if (!ILE || !CGF.getTypes().isZeroInitializable(ILE->getType()))
1888df94cb7dSKen Dyck     return CGF.getContext().getTypeSizeInChars(E->getType());
188927a3631bSChris Lattner 
1890c5cc2fb9SChris Lattner   // InitListExprs for structs have to be handled carefully.  If there are
1891c5cc2fb9SChris Lattner   // reference members, we need to consider the size of the reference, not the
1892c5cc2fb9SChris Lattner   // referencee.  InitListExprs for unions and arrays can't have references.
18935cd84755SChris Lattner   if (const RecordType *RT = E->getType()->getAs<RecordType>()) {
18945cd84755SChris Lattner     if (!RT->isUnionType()) {
1895f7133b79SSimon Pilgrim       RecordDecl *SD = RT->getDecl();
1896df94cb7dSKen Dyck       CharUnits NumNonZeroBytes = CharUnits::Zero();
1897c5cc2fb9SChris Lattner 
1898c5cc2fb9SChris Lattner       unsigned ILEElement = 0;
1899872307e2SRichard Smith       if (auto *CXXRD = dyn_cast<CXXRecordDecl>(SD))
19006365e464SRichard Smith         while (ILEElement != CXXRD->getNumBases())
1901872307e2SRichard Smith           NumNonZeroBytes +=
1902872307e2SRichard Smith               GetNumNonZeroBytesInInit(ILE->getInit(ILEElement++), CGF);
1903e8a8baefSAaron Ballman       for (const auto *Field : SD->fields()) {
1904c5cc2fb9SChris Lattner         // We're done once we hit the flexible array member or run out of
1905c5cc2fb9SChris Lattner         // InitListExpr elements.
1906c5cc2fb9SChris Lattner         if (Field->getType()->isIncompleteArrayType() ||
1907c5cc2fb9SChris Lattner             ILEElement == ILE->getNumInits())
1908c5cc2fb9SChris Lattner           break;
1909c5cc2fb9SChris Lattner         if (Field->isUnnamedBitfield())
1910c5cc2fb9SChris Lattner           continue;
1911c5cc2fb9SChris Lattner 
1912c5cc2fb9SChris Lattner         const Expr *E = ILE->getInit(ILEElement++);
1913c5cc2fb9SChris Lattner 
1914c5cc2fb9SChris Lattner         // Reference values are always non-null and have the width of a pointer.
19155cd84755SChris Lattner         if (Field->getType()->isReferenceType())
1916df94cb7dSKen Dyck           NumNonZeroBytes += CGF.getContext().toCharUnitsFromBits(
1917c8e01705SJohn McCall               CGF.getTarget().getPointerWidth(0));
19185cd84755SChris Lattner         else
1919c5cc2fb9SChris Lattner           NumNonZeroBytes += GetNumNonZeroBytesInInit(E, CGF);
1920c5cc2fb9SChris Lattner       }
1921c5cc2fb9SChris Lattner 
1922c5cc2fb9SChris Lattner       return NumNonZeroBytes;
1923c5cc2fb9SChris Lattner     }
19245cd84755SChris Lattner   }
1925c5cc2fb9SChris Lattner 
192648c70c16SRichard Smith   // FIXME: This overestimates the number of non-zero bytes for bit-fields.
1927df94cb7dSKen Dyck   CharUnits NumNonZeroBytes = CharUnits::Zero();
192827a3631bSChris Lattner   for (unsigned i = 0, e = ILE->getNumInits(); i != e; ++i)
192927a3631bSChris Lattner     NumNonZeroBytes += GetNumNonZeroBytesInInit(ILE->getInit(i), CGF);
193027a3631bSChris Lattner   return NumNonZeroBytes;
193127a3631bSChris Lattner }
193227a3631bSChris Lattner 
193327a3631bSChris Lattner /// CheckAggExprForMemSetUse - If the initializer is large and has a lot of
193427a3631bSChris Lattner /// zeros in it, emit a memset and avoid storing the individual zeros.
193527a3631bSChris Lattner ///
193627a3631bSChris Lattner static void CheckAggExprForMemSetUse(AggValueSlot &Slot, const Expr *E,
193727a3631bSChris Lattner                                      CodeGenFunction &CGF) {
193827a3631bSChris Lattner   // If the slot is already known to be zeroed, nothing to do.  Don't mess with
193927a3631bSChris Lattner   // volatile stores.
19407f416cc4SJohn McCall   if (Slot.isZeroed() || Slot.isVolatile() || !Slot.getAddress().isValid())
19418a13c418SCraig Topper     return;
194227a3631bSChris Lattner 
194303535265SArgyrios Kyrtzidis   // C++ objects with a user-declared constructor don't need zero'ing.
19449c6890a7SRichard Smith   if (CGF.getLangOpts().CPlusPlus)
194503535265SArgyrios Kyrtzidis     if (const RecordType *RT = CGF.getContext()
194603535265SArgyrios Kyrtzidis                        .getBaseElementType(E->getType())->getAs<RecordType>()) {
194703535265SArgyrios Kyrtzidis       const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
194803535265SArgyrios Kyrtzidis       if (RD->hasUserDeclaredConstructor())
194903535265SArgyrios Kyrtzidis         return;
195003535265SArgyrios Kyrtzidis     }
195103535265SArgyrios Kyrtzidis 
195227a3631bSChris Lattner   // If the type is 16-bytes or smaller, prefer individual stores over memset.
1953e78fac51SRichard Smith   CharUnits Size = Slot.getPreferredSize(CGF.getContext(), E->getType());
19547f416cc4SJohn McCall   if (Size <= CharUnits::fromQuantity(16))
195527a3631bSChris Lattner     return;
195627a3631bSChris Lattner 
195727a3631bSChris Lattner   // Check to see if over 3/4 of the initializer are known to be zero.  If so,
195827a3631bSChris Lattner   // we prefer to emit memset + individual stores for the rest.
1959239a3357SKen Dyck   CharUnits NumNonZeroBytes = GetNumNonZeroBytesInInit(E, CGF);
19607f416cc4SJohn McCall   if (NumNonZeroBytes*4 > Size)
196127a3631bSChris Lattner     return;
196227a3631bSChris Lattner 
196327a3631bSChris Lattner   // Okay, it seems like a good idea to use an initial memset, emit the call.
19647f416cc4SJohn McCall   llvm::Constant *SizeVal = CGF.Builder.getInt64(Size.getQuantity());
196527a3631bSChris Lattner 
19667f416cc4SJohn McCall   Address Loc = Slot.getAddress();
19677f416cc4SJohn McCall   Loc = CGF.Builder.CreateElementBitCast(Loc, CGF.Int8Ty);
19687f416cc4SJohn McCall   CGF.Builder.CreateMemSet(Loc, CGF.Builder.getInt8(0), SizeVal, false);
196927a3631bSChris Lattner 
197027a3631bSChris Lattner   // Tell the AggExprEmitter that the slot is known zero.
197127a3631bSChris Lattner   Slot.setZeroed();
197227a3631bSChris Lattner }
197327a3631bSChris Lattner 
197427a3631bSChris Lattner 
197527a3631bSChris Lattner 
197627a3631bSChris Lattner 
197725306cacSMike Stump /// EmitAggExpr - Emit the computation of the specified expression of aggregate
197825306cacSMike Stump /// type.  The result is computed into DestPtr.  Note that if DestPtr is null,
197925306cacSMike Stump /// the value of the aggregate expression is not needed.  If VolatileDest is
198025306cacSMike Stump /// true, DestPtr cannot be 0.
19814e8ca4faSJohn McCall void CodeGenFunction::EmitAggExpr(const Expr *E, AggValueSlot Slot) {
198247fb9508SJohn McCall   assert(E && hasAggregateEvaluationKind(E->getType()) &&
19837a51313dSChris Lattner          "Invalid aggregate expression to emit");
19847f416cc4SJohn McCall   assert((Slot.getAddress().isValid() || Slot.isIgnored()) &&
198527a3631bSChris Lattner          "slot has bits but no address");
19867a51313dSChris Lattner 
198727a3631bSChris Lattner   // Optimize the slot if possible.
198827a3631bSChris Lattner   CheckAggExprForMemSetUse(Slot, E, *this);
198927a3631bSChris Lattner 
19906aab1117SLeny Kholodov   AggExprEmitter(*this, Slot, Slot.isIgnored()).Visit(const_cast<Expr*>(E));
19917a51313dSChris Lattner }
19920bc8e86dSDaniel Dunbar 
1993d0bc7b9dSDaniel Dunbar LValue CodeGenFunction::EmitAggExprToLValue(const Expr *E) {
199447fb9508SJohn McCall   assert(hasAggregateEvaluationKind(E->getType()) && "Invalid argument!");
19957f416cc4SJohn McCall   Address Temp = CreateMemTemp(E->getType());
19962e442a00SDaniel Dunbar   LValue LV = MakeAddrLValue(Temp, E->getType());
1997f139ae3dSAkira Hatanaka   EmitAggExpr(E, AggValueSlot::forLValue(
1998f139ae3dSAkira Hatanaka                      LV, *this, AggValueSlot::IsNotDestructed,
199946759f4fSJohn McCall                      AggValueSlot::DoesNotNeedGCBarriers,
2000f139ae3dSAkira Hatanaka                      AggValueSlot::IsNotAliased, AggValueSlot::DoesNotOverlap));
20012e442a00SDaniel Dunbar   return LV;
2002d0bc7b9dSDaniel Dunbar }
2003d0bc7b9dSDaniel Dunbar 
200478b239eaSRichard Smith AggValueSlot::Overlap_t
20058cca3a5aSRichard Smith CodeGenFunction::getOverlapForFieldInit(const FieldDecl *FD) {
200678b239eaSRichard Smith   if (!FD->hasAttr<NoUniqueAddressAttr>() || !FD->getType()->isRecordType())
200778b239eaSRichard Smith     return AggValueSlot::DoesNotOverlap;
200878b239eaSRichard Smith 
200978b239eaSRichard Smith   // If the field lies entirely within the enclosing class's nvsize, its tail
201078b239eaSRichard Smith   // padding cannot overlap any already-initialized object. (The only subobjects
201178b239eaSRichard Smith   // with greater addresses that might already be initialized are vbases.)
201278b239eaSRichard Smith   const RecordDecl *ClassRD = FD->getParent();
201378b239eaSRichard Smith   const ASTRecordLayout &Layout = getContext().getASTRecordLayout(ClassRD);
201478b239eaSRichard Smith   if (Layout.getFieldOffset(FD->getFieldIndex()) +
201578b239eaSRichard Smith           getContext().getTypeSize(FD->getType()) <=
201678b239eaSRichard Smith       (uint64_t)getContext().toBits(Layout.getNonVirtualSize()))
201778b239eaSRichard Smith     return AggValueSlot::DoesNotOverlap;
201878b239eaSRichard Smith 
201978b239eaSRichard Smith   // The tail padding may contain values we need to preserve.
202078b239eaSRichard Smith   return AggValueSlot::MayOverlap;
202178b239eaSRichard Smith }
202278b239eaSRichard Smith 
20238cca3a5aSRichard Smith AggValueSlot::Overlap_t CodeGenFunction::getOverlapForBaseInit(
2024e78fac51SRichard Smith     const CXXRecordDecl *RD, const CXXRecordDecl *BaseRD, bool IsVirtual) {
202578b239eaSRichard Smith   // If the most-derived object is a field declared with [[no_unique_address]],
202678b239eaSRichard Smith   // the tail padding of any virtual base could be reused for other subobjects
202778b239eaSRichard Smith   // of that field's class.
2028e78fac51SRichard Smith   if (IsVirtual)
202978b239eaSRichard Smith     return AggValueSlot::MayOverlap;
2030e78fac51SRichard Smith 
2031e78fac51SRichard Smith   // If the base class is laid out entirely within the nvsize of the derived
2032e78fac51SRichard Smith   // class, its tail padding cannot yet be initialized, so we can issue
2033e78fac51SRichard Smith   // stores at the full width of the base class.
2034e78fac51SRichard Smith   const ASTRecordLayout &Layout = getContext().getASTRecordLayout(RD);
2035e78fac51SRichard Smith   if (Layout.getBaseClassOffset(BaseRD) +
2036e78fac51SRichard Smith           getContext().getASTRecordLayout(BaseRD).getSize() <=
2037e78fac51SRichard Smith       Layout.getNonVirtualSize())
2038e78fac51SRichard Smith     return AggValueSlot::DoesNotOverlap;
2039e78fac51SRichard Smith 
2040e78fac51SRichard Smith   // The tail padding may contain values we need to preserve.
2041e78fac51SRichard Smith   return AggValueSlot::MayOverlap;
2042e78fac51SRichard Smith }
2043e78fac51SRichard Smith 
2044e78fac51SRichard Smith void CodeGenFunction::EmitAggregateCopy(LValue Dest, LValue Src, QualType Ty,
2045e78fac51SRichard Smith                                         AggValueSlot::Overlap_t MayOverlap,
2046e78fac51SRichard Smith                                         bool isVolatile) {
2047615ed1a3SChad Rosier   assert(!Ty->isAnyComplexType() && "Shouldn't happen for complex");
20480bc8e86dSDaniel Dunbar 
2049f139ae3dSAkira Hatanaka   Address DestPtr = Dest.getAddress(*this);
2050f139ae3dSAkira Hatanaka   Address SrcPtr = Src.getAddress(*this);
20511860b520SIvan A. Kosarev 
20529c6890a7SRichard Smith   if (getLangOpts().CPlusPlus) {
2053615ed1a3SChad Rosier     if (const RecordType *RT = Ty->getAs<RecordType>()) {
2054615ed1a3SChad Rosier       CXXRecordDecl *Record = cast<CXXRecordDecl>(RT->getDecl());
2055615ed1a3SChad Rosier       assert((Record->hasTrivialCopyConstructor() ||
2056615ed1a3SChad Rosier               Record->hasTrivialCopyAssignment() ||
2057615ed1a3SChad Rosier               Record->hasTrivialMoveConstructor() ||
2058419bd094SRichard Smith               Record->hasTrivialMoveAssignment() ||
2059*c8227f06SArthur Eubanks               Record->hasAttr<TrivialABIAttr>() || Record->isUnion()) &&
206016488472SRichard Smith              "Trying to aggregate-copy a type without a trivial copy/move "
2061f22101a0SDouglas Gregor              "constructor or assignment operator");
2062615ed1a3SChad Rosier       // Ignore empty classes in C++.
2063615ed1a3SChad Rosier       if (Record->isEmpty())
206416e94af6SAnders Carlsson         return;
206516e94af6SAnders Carlsson     }
206616e94af6SAnders Carlsson   }
206716e94af6SAnders Carlsson 
20685be9b8cbSMichael Liao   if (getLangOpts().CUDAIsDevice) {
20695be9b8cbSMichael Liao     if (Ty->isCUDADeviceBuiltinSurfaceType()) {
20705be9b8cbSMichael Liao       if (getTargetHooks().emitCUDADeviceBuiltinSurfaceDeviceCopy(*this, Dest,
20715be9b8cbSMichael Liao                                                                   Src))
20725be9b8cbSMichael Liao         return;
20735be9b8cbSMichael Liao     } else if (Ty->isCUDADeviceBuiltinTextureType()) {
20745be9b8cbSMichael Liao       if (getTargetHooks().emitCUDADeviceBuiltinTextureDeviceCopy(*this, Dest,
20755be9b8cbSMichael Liao                                                                   Src))
20765be9b8cbSMichael Liao         return;
20775be9b8cbSMichael Liao     }
20785be9b8cbSMichael Liao   }
20795be9b8cbSMichael Liao 
2080ca05dfefSChris Lattner   // Aggregate assignment turns into llvm.memcpy.  This is almost valid per
20813ef668c2SChris Lattner   // C99 6.5.16.1p3, which states "If the value being stored in an object is
20823ef668c2SChris Lattner   // read from another object that overlaps in anyway the storage of the first
20833ef668c2SChris Lattner   // object, then the overlap shall be exact and the two objects shall have
20843ef668c2SChris Lattner   // qualified or unqualified versions of a compatible type."
20853ef668c2SChris Lattner   //
2086ca05dfefSChris Lattner   // memcpy is not defined if the source and destination pointers are exactly
20873ef668c2SChris Lattner   // equal, but other compilers do this optimization, and almost every memcpy
20883ef668c2SChris Lattner   // implementation handles this case safely.  If there is a libc that does not
20893ef668c2SChris Lattner   // safely handle this, we can add a target hook.
20900bc8e86dSDaniel Dunbar 
2091e78fac51SRichard Smith   // Get data size info for this aggregate. Don't copy the tail padding if this
2092e78fac51SRichard Smith   // might be a potentially-overlapping subobject, since the tail padding might
2093e78fac51SRichard Smith   // be occupied by a different object. Otherwise, copying it is fine.
2094101309feSBevin Hansson   TypeInfoChars TypeInfo;
2095e78fac51SRichard Smith   if (MayOverlap)
20961ca66919SBenjamin Kramer     TypeInfo = getContext().getTypeInfoDataSizeInChars(Ty);
20971ca66919SBenjamin Kramer   else
20981ca66919SBenjamin Kramer     TypeInfo = getContext().getTypeInfoInChars(Ty);
2099615ed1a3SChad Rosier 
210016dc7b68SAlexey Bataev   llvm::Value *SizeVal = nullptr;
2101101309feSBevin Hansson   if (TypeInfo.Width.isZero()) {
210216dc7b68SAlexey Bataev     // But note that getTypeInfo returns 0 for a VLA.
210316dc7b68SAlexey Bataev     if (auto *VAT = dyn_cast_or_null<VariableArrayType>(
210416dc7b68SAlexey Bataev             getContext().getAsArrayType(Ty))) {
210516dc7b68SAlexey Bataev       QualType BaseEltTy;
210616dc7b68SAlexey Bataev       SizeVal = emitArrayLength(VAT, BaseEltTy, DestPtr);
2107e78fac51SRichard Smith       TypeInfo = getContext().getTypeInfoInChars(BaseEltTy);
2108101309feSBevin Hansson       assert(!TypeInfo.Width.isZero());
210916dc7b68SAlexey Bataev       SizeVal = Builder.CreateNUWMul(
211016dc7b68SAlexey Bataev           SizeVal,
2111101309feSBevin Hansson           llvm::ConstantInt::get(SizeTy, TypeInfo.Width.getQuantity()));
211216dc7b68SAlexey Bataev     }
211316dc7b68SAlexey Bataev   }
211416dc7b68SAlexey Bataev   if (!SizeVal) {
2115101309feSBevin Hansson     SizeVal = llvm::ConstantInt::get(SizeTy, TypeInfo.Width.getQuantity());
211616dc7b68SAlexey Bataev   }
2117615ed1a3SChad Rosier 
2118615ed1a3SChad Rosier   // FIXME: If we have a volatile struct, the optimizer can remove what might
2119615ed1a3SChad Rosier   // appear to be `extra' memory ops:
2120615ed1a3SChad Rosier   //
2121615ed1a3SChad Rosier   // volatile struct { int i; } a, b;
2122615ed1a3SChad Rosier   //
2123615ed1a3SChad Rosier   // int main() {
2124615ed1a3SChad Rosier   //   a = b;
2125615ed1a3SChad Rosier   //   a = b;
2126615ed1a3SChad Rosier   // }
2127615ed1a3SChad Rosier   //
2128615ed1a3SChad Rosier   // we need to use a different call here.  We use isVolatile to indicate when
2129615ed1a3SChad Rosier   // either the source or the destination is volatile.
2130615ed1a3SChad Rosier 
21317f416cc4SJohn McCall   DestPtr = Builder.CreateElementBitCast(DestPtr, Int8Ty);
21327f416cc4SJohn McCall   SrcPtr = Builder.CreateElementBitCast(SrcPtr, Int8Ty);
2133615ed1a3SChad Rosier 
2134615ed1a3SChad Rosier   // Don't do any of the memmove_collectable tests if GC isn't set.
2135615ed1a3SChad Rosier   if (CGM.getLangOpts().getGC() == LangOptions::NonGC) {
2136615ed1a3SChad Rosier     // fall through
2137615ed1a3SChad Rosier   } else if (const RecordType *RecordTy = Ty->getAs<RecordType>()) {
2138615ed1a3SChad Rosier     RecordDecl *Record = RecordTy->getDecl();
2139615ed1a3SChad Rosier     if (Record->hasObjectMember()) {
2140615ed1a3SChad Rosier       CGM.getObjCRuntime().EmitGCMemmoveCollectable(*this, DestPtr, SrcPtr,
2141615ed1a3SChad Rosier                                                     SizeVal);
2142615ed1a3SChad Rosier       return;
2143615ed1a3SChad Rosier     }
2144615ed1a3SChad Rosier   } else if (Ty->isArrayType()) {
2145615ed1a3SChad Rosier     QualType BaseType = getContext().getBaseElementType(Ty);
2146615ed1a3SChad Rosier     if (const RecordType *RecordTy = BaseType->getAs<RecordType>()) {
2147615ed1a3SChad Rosier       if (RecordTy->getDecl()->hasObjectMember()) {
2148615ed1a3SChad Rosier         CGM.getObjCRuntime().EmitGCMemmoveCollectable(*this, DestPtr, SrcPtr,
2149615ed1a3SChad Rosier                                                       SizeVal);
2150615ed1a3SChad Rosier         return;
2151615ed1a3SChad Rosier       }
2152615ed1a3SChad Rosier     }
2153615ed1a3SChad Rosier   }
2154615ed1a3SChad Rosier 
21557f416cc4SJohn McCall   auto Inst = Builder.CreateMemCpy(DestPtr, SrcPtr, SizeVal, isVolatile);
21567f416cc4SJohn McCall 
215722695fceSDan Gohman   // Determine the metadata to describe the position of any padding in this
215822695fceSDan Gohman   // memcpy, as well as the TBAA tags for the members of the struct, in case
215922695fceSDan Gohman   // the optimizer wishes to expand it in to scalar memory operations.
21607f416cc4SJohn McCall   if (llvm::MDNode *TBAAStructTag = CGM.getTBAAStructInfo(Ty))
21617f416cc4SJohn McCall     Inst->setMetadata(llvm::LLVMContext::MD_tbaa_struct, TBAAStructTag);
21621860b520SIvan A. Kosarev 
21631860b520SIvan A. Kosarev   if (CGM.getCodeGenOpts().NewStructPathTBAA) {
21641860b520SIvan A. Kosarev     TBAAAccessInfo TBAAInfo = CGM.mergeTBAAInfoForMemoryTransfer(
21651860b520SIvan A. Kosarev         Dest.getTBAAInfo(), Src.getTBAAInfo());
21661860b520SIvan A. Kosarev     CGM.DecorateInstructionWithTBAA(Inst, TBAAInfo);
21671860b520SIvan A. Kosarev   }
21680bc8e86dSDaniel Dunbar }
2169