17a51313dSChris Lattner //===--- CGExprAgg.cpp - Emit LLVM Code from Aggregate Expressions --------===//
27a51313dSChris Lattner //
37a51313dSChris Lattner //                     The LLVM Compiler Infrastructure
47a51313dSChris Lattner //
57a51313dSChris Lattner // This file is distributed under the University of Illinois Open Source
67a51313dSChris Lattner // License. See LICENSE.TXT for details.
77a51313dSChris Lattner //
87a51313dSChris Lattner //===----------------------------------------------------------------------===//
97a51313dSChris Lattner //
107a51313dSChris Lattner // This contains code to emit Aggregate Expr nodes as LLVM code.
117a51313dSChris Lattner //
127a51313dSChris Lattner //===----------------------------------------------------------------------===//
137a51313dSChris Lattner 
147a51313dSChris Lattner #include "CodeGenFunction.h"
155f21d2f6SFariborz Jahanian #include "CGObjCRuntime.h"
163a02247dSChandler Carruth #include "CodeGenModule.h"
17e0ef348cSIvan A. Kosarev #include "ConstantEmitter.h"
18ad319a73SDaniel Dunbar #include "clang/AST/ASTContext.h"
19b7f8f594SAnders Carlsson #include "clang/AST/DeclCXX.h"
20c83ed824SSebastian Redl #include "clang/AST/DeclTemplate.h"
21ad319a73SDaniel Dunbar #include "clang/AST/StmtVisitor.h"
22ffd5551bSChandler Carruth #include "llvm/IR/Constants.h"
23ffd5551bSChandler Carruth #include "llvm/IR/Function.h"
24ffd5551bSChandler Carruth #include "llvm/IR/GlobalVariable.h"
25ffd5551bSChandler Carruth #include "llvm/IR/Intrinsics.h"
264deb75d2SGeorge Burgess IV #include "llvm/IR/IntrinsicInst.h"
277a51313dSChris Lattner using namespace clang;
287a51313dSChris Lattner using namespace CodeGen;
297a51313dSChris Lattner 
307a51313dSChris Lattner //===----------------------------------------------------------------------===//
317a51313dSChris Lattner //                        Aggregate Expression Emitter
327a51313dSChris Lattner //===----------------------------------------------------------------------===//
337a51313dSChris Lattner 
347a51313dSChris Lattner namespace  {
35337e3a5fSBenjamin Kramer class AggExprEmitter : public StmtVisitor<AggExprEmitter> {
367a51313dSChris Lattner   CodeGenFunction &CGF;
37cb463859SDaniel Dunbar   CGBuilderTy &Builder;
387a626f63SJohn McCall   AggValueSlot Dest;
396aab1117SLeny Kholodov   bool IsResultUnused;
4078a15113SJohn McCall 
417a626f63SJohn McCall   AggValueSlot EnsureSlot(QualType T) {
427a626f63SJohn McCall     if (!Dest.isIgnored()) return Dest;
437a626f63SJohn McCall     return CGF.CreateAggTemp(T, "agg.tmp.ensured");
4478a15113SJohn McCall   }
454e8ca4faSJohn McCall   void EnsureDest(QualType T) {
464e8ca4faSJohn McCall     if (!Dest.isIgnored()) return;
474e8ca4faSJohn McCall     Dest = CGF.CreateAggTemp(T, "agg.tmp.ensured");
484e8ca4faSJohn McCall   }
49cc04e9f6SJohn McCall 
5056e5a2e1SGeorge Burgess IV   // Calls `Fn` with a valid return value slot, potentially creating a temporary
5156e5a2e1SGeorge Burgess IV   // to do so. If a temporary is created, an appropriate copy into `Dest` will
524deb75d2SGeorge Burgess IV   // be emitted, as will lifetime markers.
5356e5a2e1SGeorge Burgess IV   //
5456e5a2e1SGeorge Burgess IV   // The given function should take a ReturnValueSlot, and return an RValue that
5556e5a2e1SGeorge Burgess IV   // points to said slot.
5656e5a2e1SGeorge Burgess IV   void withReturnValueSlot(const Expr *E,
5756e5a2e1SGeorge Burgess IV                            llvm::function_ref<RValue(ReturnValueSlot)> Fn);
5856e5a2e1SGeorge Burgess IV 
597a51313dSChris Lattner public:
606aab1117SLeny Kholodov   AggExprEmitter(CodeGenFunction &cgf, AggValueSlot Dest, bool IsResultUnused)
616aab1117SLeny Kholodov     : CGF(cgf), Builder(CGF.Builder), Dest(Dest),
626aab1117SLeny Kholodov     IsResultUnused(IsResultUnused) { }
637a51313dSChris Lattner 
647a51313dSChris Lattner   //===--------------------------------------------------------------------===//
657a51313dSChris Lattner   //                               Utilities
667a51313dSChris Lattner   //===--------------------------------------------------------------------===//
677a51313dSChris Lattner 
687a51313dSChris Lattner   /// EmitAggLoadOfLValue - Given an expression with aggregate type that
697a51313dSChris Lattner   /// represents a value lvalue, this method emits the address of the lvalue,
707a51313dSChris Lattner   /// then loads the result into DestPtr.
717a51313dSChris Lattner   void EmitAggLoadOfLValue(const Expr *E);
727a51313dSChris Lattner 
737275da0fSAkira Hatanaka   enum ExprValueKind {
747275da0fSAkira Hatanaka     EVK_RValue,
757275da0fSAkira Hatanaka     EVK_NonRValue
767275da0fSAkira Hatanaka   };
777275da0fSAkira Hatanaka 
78ca9fc09cSMike Stump   /// EmitFinalDestCopy - Perform the final copy to DestPtr, if desired.
797275da0fSAkira Hatanaka   /// SrcIsRValue is true if source comes from an RValue.
807275da0fSAkira Hatanaka   void EmitFinalDestCopy(QualType type, const LValue &src,
817275da0fSAkira Hatanaka                          ExprValueKind SrcValueKind = EVK_NonRValue);
827f416cc4SJohn McCall   void EmitFinalDestCopy(QualType type, RValue src);
834e8ca4faSJohn McCall   void EmitCopy(QualType type, const AggValueSlot &dest,
844e8ca4faSJohn McCall                 const AggValueSlot &src);
85ca9fc09cSMike Stump 
86a5efa738SJohn McCall   void EmitMoveFromReturnSlot(const Expr *E, RValue Src);
87cc04e9f6SJohn McCall 
887f416cc4SJohn McCall   void EmitArrayInit(Address DestPtr, llvm::ArrayType *AType,
89e0ef348cSIvan A. Kosarev                      QualType ArrayQTy, InitListExpr *E);
90c83ed824SSebastian Redl 
918d6fc958SJohn McCall   AggValueSlot::NeedsGCBarriers_t needsGC(QualType T) {
92bbafb8a7SDavid Blaikie     if (CGF.getLangOpts().getGC() && TypeRequiresGCollection(T))
938d6fc958SJohn McCall       return AggValueSlot::NeedsGCBarriers;
948d6fc958SJohn McCall     return AggValueSlot::DoesNotNeedGCBarriers;
958d6fc958SJohn McCall   }
968d6fc958SJohn McCall 
97cc04e9f6SJohn McCall   bool TypeRequiresGCollection(QualType T);
98cc04e9f6SJohn McCall 
997a51313dSChris Lattner   //===--------------------------------------------------------------------===//
1007a51313dSChris Lattner   //                            Visitor Methods
1017a51313dSChris Lattner   //===--------------------------------------------------------------------===//
1027a51313dSChris Lattner 
10301fb5fb1SDavid Blaikie   void Visit(Expr *E) {
1049b479666SDavid Blaikie     ApplyDebugLocation DL(CGF, E);
10501fb5fb1SDavid Blaikie     StmtVisitor<AggExprEmitter>::Visit(E);
10601fb5fb1SDavid Blaikie   }
10701fb5fb1SDavid Blaikie 
1087a51313dSChris Lattner   void VisitStmt(Stmt *S) {
109a7c8cf62SDaniel Dunbar     CGF.ErrorUnsupported(S, "aggregate expression");
1107a51313dSChris Lattner   }
1117a51313dSChris Lattner   void VisitParenExpr(ParenExpr *PE) { Visit(PE->getSubExpr()); }
11291147596SPeter Collingbourne   void VisitGenericSelectionExpr(GenericSelectionExpr *GE) {
11391147596SPeter Collingbourne     Visit(GE->getResultExpr());
11491147596SPeter Collingbourne   }
1155eb58583SGor Nishanov   void VisitCoawaitExpr(CoawaitExpr *E) {
1165eb58583SGor Nishanov     CGF.EmitCoawaitExpr(*E, Dest, IsResultUnused);
1175eb58583SGor Nishanov   }
1185eb58583SGor Nishanov   void VisitCoyieldExpr(CoyieldExpr *E) {
1195eb58583SGor Nishanov     CGF.EmitCoyieldExpr(*E, Dest, IsResultUnused);
1205eb58583SGor Nishanov   }
1215eb58583SGor Nishanov   void VisitUnaryCoawait(UnaryOperator *E) { Visit(E->getSubExpr()); }
1223f66b84cSEli Friedman   void VisitUnaryExtension(UnaryOperator *E) { Visit(E->getSubExpr()); }
1237c454bb8SJohn McCall   void VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *E) {
1247c454bb8SJohn McCall     return Visit(E->getReplacement());
1257c454bb8SJohn McCall   }
1267a51313dSChris Lattner 
1277a51313dSChris Lattner   // l-values.
1286cc8317cSAlex Lorenz   void VisitDeclRefExpr(DeclRefExpr *E) { EmitAggLoadOfLValue(E); }
1297a51313dSChris Lattner   void VisitMemberExpr(MemberExpr *ME) { EmitAggLoadOfLValue(ME); }
1307a51313dSChris Lattner   void VisitUnaryDeref(UnaryOperator *E) { EmitAggLoadOfLValue(E); }
131d443c0a0SDaniel Dunbar   void VisitStringLiteral(StringLiteral *E) { EmitAggLoadOfLValue(E); }
1329b71f0cfSDouglas Gregor   void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
1337a51313dSChris Lattner   void VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
1347a51313dSChris Lattner     EmitAggLoadOfLValue(E);
1357a51313dSChris Lattner   }
1362f343dd5SChris Lattner   void VisitPredefinedExpr(const PredefinedExpr *E) {
1372f343dd5SChris Lattner     EmitAggLoadOfLValue(E);
1382f343dd5SChris Lattner   }
139bc7d67ceSMike Stump 
1407a51313dSChris Lattner   // Operators.
141ec143777SAnders Carlsson   void VisitCastExpr(CastExpr *E);
1427a51313dSChris Lattner   void VisitCallExpr(const CallExpr *E);
1437a51313dSChris Lattner   void VisitStmtExpr(const StmtExpr *E);
1447a51313dSChris Lattner   void VisitBinaryOperator(const BinaryOperator *BO);
145ffba662dSFariborz Jahanian   void VisitPointerToDataMemberBinaryOperator(const BinaryOperator *BO);
1467a51313dSChris Lattner   void VisitBinAssign(const BinaryOperator *E);
1474b0e2a30SEli Friedman   void VisitBinComma(const BinaryOperator *E);
1487a51313dSChris Lattner 
149b1d329daSChris Lattner   void VisitObjCMessageExpr(ObjCMessageExpr *E);
150c8317a44SDaniel Dunbar   void VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
151c8317a44SDaniel Dunbar     EmitAggLoadOfLValue(E);
152c8317a44SDaniel Dunbar   }
1537a51313dSChris Lattner 
154cb77930dSYunzhong Gao   void VisitDesignatedInitUpdateExpr(DesignatedInitUpdateExpr *E);
155c07a0c7eSJohn McCall   void VisitAbstractConditionalOperator(const AbstractConditionalOperator *CO);
1565b2095ceSAnders Carlsson   void VisitChooseExpr(const ChooseExpr *CE);
1577a51313dSChris Lattner   void VisitInitListExpr(InitListExpr *E);
158939b6880SRichard Smith   void VisitArrayInitLoopExpr(const ArrayInitLoopExpr *E,
159939b6880SRichard Smith                               llvm::Value *outerBegin = nullptr);
16018ada985SAnders Carlsson   void VisitImplicitValueInitExpr(ImplicitValueInitExpr *E);
161cb77930dSYunzhong Gao   void VisitNoInitExpr(NoInitExpr *E) { } // Do nothing.
162aa9c7aedSChris Lattner   void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
163aa9c7aedSChris Lattner     Visit(DAE->getExpr());
164aa9c7aedSChris Lattner   }
165852c9db7SRichard Smith   void VisitCXXDefaultInitExpr(CXXDefaultInitExpr *DIE) {
166852c9db7SRichard Smith     CodeGenFunction::CXXDefaultInitExprScope Scope(CGF);
167852c9db7SRichard Smith     Visit(DIE->getExpr());
168852c9db7SRichard Smith   }
1693be22e27SAnders Carlsson   void VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E);
1701619a504SAnders Carlsson   void VisitCXXConstructExpr(const CXXConstructExpr *E);
1715179eb78SRichard Smith   void VisitCXXInheritedCtorInitExpr(const CXXInheritedCtorInitExpr *E);
172c370a7eeSEli Friedman   void VisitLambdaExpr(LambdaExpr *E);
173cc1b96d3SRichard Smith   void VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr *E);
1745d413781SJohn McCall   void VisitExprWithCleanups(ExprWithCleanups *E);
175747eb784SDouglas Gregor   void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
1765bbbb137SMike Stump   void VisitCXXTypeidExpr(CXXTypeidExpr *E) { EmitAggLoadOfLValue(E); }
177fe31481fSDouglas Gregor   void VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E);
1781bf5846aSJohn McCall   void VisitOpaqueValueExpr(OpaqueValueExpr *E);
1791bf5846aSJohn McCall 
180fe96e0b6SJohn McCall   void VisitPseudoObjectExpr(PseudoObjectExpr *E) {
181fe96e0b6SJohn McCall     if (E->isGLValue()) {
182fe96e0b6SJohn McCall       LValue LV = CGF.EmitPseudoObjectLValue(E);
1834e8ca4faSJohn McCall       return EmitFinalDestCopy(E->getType(), LV);
184fe96e0b6SJohn McCall     }
185fe96e0b6SJohn McCall 
186fe96e0b6SJohn McCall     CGF.EmitPseudoObjectRValue(E, EnsureSlot(E->getType()));
187fe96e0b6SJohn McCall   }
188fe96e0b6SJohn McCall 
18921911e89SEli Friedman   void VisitVAArgExpr(VAArgExpr *E);
190579a05d7SChris Lattner 
191615ed1a3SChad Rosier   void EmitInitializationToLValue(Expr *E, LValue Address);
1921553b190SJohn McCall   void EmitNullInitializationToLValue(LValue Address);
1937a51313dSChris Lattner   //  case Expr::ChooseExprClass:
194f16b8c30SMike Stump   void VisitCXXThrowExpr(const CXXThrowExpr *E) { CGF.EmitCXXThrowExpr(E); }
195df14b3a8SEli Friedman   void VisitAtomicExpr(AtomicExpr *E) {
196cc2a6e06STim Northover     RValue Res = CGF.EmitAtomicExpr(E);
197cc2a6e06STim Northover     EmitFinalDestCopy(E->getType(), Res);
198df14b3a8SEli Friedman   }
1997a51313dSChris Lattner };
2007a51313dSChris Lattner }  // end anonymous namespace.
2017a51313dSChris Lattner 
2027a51313dSChris Lattner //===----------------------------------------------------------------------===//
2037a51313dSChris Lattner //                                Utilities
2047a51313dSChris Lattner //===----------------------------------------------------------------------===//
2057a51313dSChris Lattner 
2067a51313dSChris Lattner /// EmitAggLoadOfLValue - Given an expression with aggregate type that
2077a51313dSChris Lattner /// represents a value lvalue, this method emits the address of the lvalue,
2087a51313dSChris Lattner /// then loads the result into DestPtr.
2097a51313dSChris Lattner void AggExprEmitter::EmitAggLoadOfLValue(const Expr *E) {
2107a51313dSChris Lattner   LValue LV = CGF.EmitLValue(E);
211a8ec7eb9SJohn McCall 
212a8ec7eb9SJohn McCall   // If the type of the l-value is atomic, then do an atomic load.
213a5b195a1SDavid Majnemer   if (LV.getType()->isAtomicType() || CGF.LValueIsSuitableForInlineAtomic(LV)) {
2142d84e842SNick Lewycky     CGF.EmitAtomicLoad(LV, E->getExprLoc(), Dest);
215a8ec7eb9SJohn McCall     return;
216a8ec7eb9SJohn McCall   }
217a8ec7eb9SJohn McCall 
2184e8ca4faSJohn McCall   EmitFinalDestCopy(E->getType(), LV);
219ca9fc09cSMike Stump }
220ca9fc09cSMike Stump 
221cc04e9f6SJohn McCall /// \brief True if the given aggregate type requires special GC API calls.
222cc04e9f6SJohn McCall bool AggExprEmitter::TypeRequiresGCollection(QualType T) {
223cc04e9f6SJohn McCall   // Only record types have members that might require garbage collection.
224cc04e9f6SJohn McCall   const RecordType *RecordTy = T->getAs<RecordType>();
225cc04e9f6SJohn McCall   if (!RecordTy) return false;
226cc04e9f6SJohn McCall 
227cc04e9f6SJohn McCall   // Don't mess with non-trivial C++ types.
228cc04e9f6SJohn McCall   RecordDecl *Record = RecordTy->getDecl();
229cc04e9f6SJohn McCall   if (isa<CXXRecordDecl>(Record) &&
23016488472SRichard Smith       (cast<CXXRecordDecl>(Record)->hasNonTrivialCopyConstructor() ||
231cc04e9f6SJohn McCall        !cast<CXXRecordDecl>(Record)->hasTrivialDestructor()))
232cc04e9f6SJohn McCall     return false;
233cc04e9f6SJohn McCall 
234cc04e9f6SJohn McCall   // Check whether the type has an object member.
235cc04e9f6SJohn McCall   return Record->hasObjectMember();
236cc04e9f6SJohn McCall }
237cc04e9f6SJohn McCall 
23856e5a2e1SGeorge Burgess IV void AggExprEmitter::withReturnValueSlot(
23956e5a2e1SGeorge Burgess IV     const Expr *E, llvm::function_ref<RValue(ReturnValueSlot)> EmitCall) {
24056e5a2e1SGeorge Burgess IV   QualType RetTy = E->getType();
24156e5a2e1SGeorge Burgess IV   bool RequiresDestruction =
24256e5a2e1SGeorge Burgess IV       Dest.isIgnored() &&
24356e5a2e1SGeorge Burgess IV       RetTy.isDestructedType() == QualType::DK_nontrivial_c_struct;
2447275da0fSAkira Hatanaka 
24556e5a2e1SGeorge Burgess IV   // If it makes no observable difference, save a memcpy + temporary.
24656e5a2e1SGeorge Burgess IV   //
24756e5a2e1SGeorge Burgess IV   // We need to always provide our own temporary if destruction is required.
24856e5a2e1SGeorge Burgess IV   // Otherwise, EmitCall will emit its own, notice that it's "unused", and end
24956e5a2e1SGeorge Burgess IV   // its lifetime before we have the chance to emit a proper destructor call.
25056e5a2e1SGeorge Burgess IV   bool UseTemp = Dest.isPotentiallyAliased() || Dest.requiresGCollection() ||
25156e5a2e1SGeorge Burgess IV                  (RequiresDestruction && !Dest.getAddress().isValid());
25256e5a2e1SGeorge Burgess IV 
25356e5a2e1SGeorge Burgess IV   Address RetAddr = Address::invalid();
2544deb75d2SGeorge Burgess IV 
2554deb75d2SGeorge Burgess IV   EHScopeStack::stable_iterator LifetimeEndBlock;
2564deb75d2SGeorge Burgess IV   llvm::Value *LifetimeSizePtr = nullptr;
2574deb75d2SGeorge Burgess IV   llvm::IntrinsicInst *LifetimeStartInst = nullptr;
25856e5a2e1SGeorge Burgess IV   if (!UseTemp) {
25956e5a2e1SGeorge Burgess IV     RetAddr = Dest.getAddress();
26056e5a2e1SGeorge Burgess IV   } else {
26156e5a2e1SGeorge Burgess IV     RetAddr = CGF.CreateMemTemp(RetTy);
26256e5a2e1SGeorge Burgess IV     uint64_t Size =
26356e5a2e1SGeorge Burgess IV         CGF.CGM.getDataLayout().getTypeAllocSize(CGF.ConvertTypeForMem(RetTy));
2644deb75d2SGeorge Burgess IV     LifetimeSizePtr = CGF.EmitLifetimeStart(Size, RetAddr.getPointer());
2654deb75d2SGeorge Burgess IV     if (LifetimeSizePtr) {
2664deb75d2SGeorge Burgess IV       LifetimeStartInst =
2674deb75d2SGeorge Burgess IV           cast<llvm::IntrinsicInst>(std::prev(Builder.GetInsertPoint()));
2684deb75d2SGeorge Burgess IV       assert(LifetimeStartInst->getIntrinsicID() ==
2694deb75d2SGeorge Burgess IV                  llvm::Intrinsic::lifetime_start &&
2704deb75d2SGeorge Burgess IV              "Last insertion wasn't a lifetime.start?");
2714deb75d2SGeorge Burgess IV 
27256e5a2e1SGeorge Burgess IV       CGF.pushFullExprCleanup<CodeGenFunction::CallLifetimeEnd>(
27356e5a2e1SGeorge Burgess IV           NormalEHLifetimeMarker, RetAddr, LifetimeSizePtr);
2744deb75d2SGeorge Burgess IV       LifetimeEndBlock = CGF.EHStack.stable_begin();
2754deb75d2SGeorge Burgess IV     }
276021510e9SFariborz Jahanian   }
277a5efa738SJohn McCall 
27856e5a2e1SGeorge Burgess IV   RValue Src =
27956e5a2e1SGeorge Burgess IV       EmitCall(ReturnValueSlot(RetAddr, Dest.isVolatile(), IsResultUnused));
28056e5a2e1SGeorge Burgess IV 
28156e5a2e1SGeorge Burgess IV   if (RequiresDestruction)
28256e5a2e1SGeorge Burgess IV     CGF.pushDestroy(RetTy.isDestructedType(), Src.getAggregateAddress(), RetTy);
28356e5a2e1SGeorge Burgess IV 
2844deb75d2SGeorge Burgess IV   if (!UseTemp)
2854deb75d2SGeorge Burgess IV     return;
2864deb75d2SGeorge Burgess IV 
28756e5a2e1SGeorge Burgess IV   assert(Dest.getPointer() != Src.getAggregatePointer());
28856e5a2e1SGeorge Burgess IV   EmitFinalDestCopy(E->getType(), Src);
2894deb75d2SGeorge Burgess IV 
2904deb75d2SGeorge Burgess IV   if (!RequiresDestruction && LifetimeStartInst) {
2914deb75d2SGeorge Burgess IV     // If there's no dtor to run, the copy was the last use of our temporary.
2924deb75d2SGeorge Burgess IV     // Since we're not guaranteed to be in an ExprWithCleanups, clean up
2934deb75d2SGeorge Burgess IV     // eagerly.
2944deb75d2SGeorge Burgess IV     CGF.DeactivateCleanupBlock(LifetimeEndBlock, LifetimeStartInst);
2954deb75d2SGeorge Burgess IV     CGF.EmitLifetimeEnd(LifetimeSizePtr, RetAddr.getPointer());
29656e5a2e1SGeorge Burgess IV   }
297cc04e9f6SJohn McCall }
298cc04e9f6SJohn McCall 
299ca9fc09cSMike Stump /// EmitFinalDestCopy - Perform the final copy to DestPtr, if desired.
3007f416cc4SJohn McCall void AggExprEmitter::EmitFinalDestCopy(QualType type, RValue src) {
3014e8ca4faSJohn McCall   assert(src.isAggregate() && "value must be aggregate value!");
3027f416cc4SJohn McCall   LValue srcLV = CGF.MakeAddrLValue(src.getAggregateAddress(), type);
3037275da0fSAkira Hatanaka   EmitFinalDestCopy(type, srcLV, EVK_RValue);
3044e8ca4faSJohn McCall }
3057a51313dSChris Lattner 
3064e8ca4faSJohn McCall /// EmitFinalDestCopy - Perform the final copy to DestPtr, if desired.
3077275da0fSAkira Hatanaka void AggExprEmitter::EmitFinalDestCopy(QualType type, const LValue &src,
3087275da0fSAkira Hatanaka                                        ExprValueKind SrcValueKind) {
3097a626f63SJohn McCall   // If Dest is ignored, then we're evaluating an aggregate expression
3104e8ca4faSJohn McCall   // in a context that doesn't care about the result.  Note that loads
3114e8ca4faSJohn McCall   // from volatile l-values force the existence of a non-ignored
3124e8ca4faSJohn McCall   // destination.
3134e8ca4faSJohn McCall   if (Dest.isIgnored())
314ec3cbfe8SMike Stump     return;
315c123623dSFariborz Jahanian 
3167275da0fSAkira Hatanaka   // Copy non-trivial C structs here.
3177275da0fSAkira Hatanaka   LValue DstLV = CGF.MakeAddrLValue(
3187275da0fSAkira Hatanaka       Dest.getAddress(), Dest.isVolatile() ? type.withVolatile() : type);
3197275da0fSAkira Hatanaka 
3207275da0fSAkira Hatanaka   if (SrcValueKind == EVK_RValue) {
3217275da0fSAkira Hatanaka     if (type.isNonTrivialToPrimitiveDestructiveMove() == QualType::PCK_Struct) {
3227275da0fSAkira Hatanaka       if (Dest.isPotentiallyAliased())
3237275da0fSAkira Hatanaka         CGF.callCStructMoveAssignmentOperator(DstLV, src);
3247275da0fSAkira Hatanaka       else
3257275da0fSAkira Hatanaka         CGF.callCStructMoveConstructor(DstLV, src);
3267275da0fSAkira Hatanaka       return;
3277275da0fSAkira Hatanaka     }
3287275da0fSAkira Hatanaka   } else {
3297275da0fSAkira Hatanaka     if (type.isNonTrivialToPrimitiveCopy() == QualType::PCK_Struct) {
3307275da0fSAkira Hatanaka       if (Dest.isPotentiallyAliased())
3317275da0fSAkira Hatanaka         CGF.callCStructCopyAssignmentOperator(DstLV, src);
3327275da0fSAkira Hatanaka       else
3337275da0fSAkira Hatanaka         CGF.callCStructCopyConstructor(DstLV, src);
3347275da0fSAkira Hatanaka       return;
3357275da0fSAkira Hatanaka     }
3367275da0fSAkira Hatanaka   }
3377275da0fSAkira Hatanaka 
3384e8ca4faSJohn McCall   AggValueSlot srcAgg =
3394e8ca4faSJohn McCall     AggValueSlot::forLValue(src, AggValueSlot::IsDestructed,
340*e78fac51SRichard Smith                             needsGC(type), AggValueSlot::IsAliased,
341*e78fac51SRichard Smith                             AggValueSlot::MayOverlap);
3424e8ca4faSJohn McCall   EmitCopy(type, Dest, srcAgg);
343332ec2ceSMike Stump }
3447a51313dSChris Lattner 
3454e8ca4faSJohn McCall /// Perform a copy from the source into the destination.
3464e8ca4faSJohn McCall ///
3474e8ca4faSJohn McCall /// \param type - the type of the aggregate being copied; qualifiers are
3484e8ca4faSJohn McCall ///   ignored
3494e8ca4faSJohn McCall void AggExprEmitter::EmitCopy(QualType type, const AggValueSlot &dest,
3504e8ca4faSJohn McCall                               const AggValueSlot &src) {
3514e8ca4faSJohn McCall   if (dest.requiresGCollection()) {
352*e78fac51SRichard Smith     CharUnits sz = dest.getPreferredSize(CGF.getContext(), type);
3534e8ca4faSJohn McCall     llvm::Value *size = llvm::ConstantInt::get(CGF.SizeTy, sz.getQuantity());
354879d7266SFariborz Jahanian     CGF.CGM.getObjCRuntime().EmitGCMemmoveCollectable(CGF,
3557f416cc4SJohn McCall                                                       dest.getAddress(),
3567f416cc4SJohn McCall                                                       src.getAddress(),
3574e8ca4faSJohn McCall                                                       size);
358879d7266SFariborz Jahanian     return;
359879d7266SFariborz Jahanian   }
3604e8ca4faSJohn McCall 
361ca9fc09cSMike Stump   // If the result of the assignment is used, copy the LHS there also.
3624e8ca4faSJohn McCall   // It's volatile if either side is.  Use the minimum alignment of
3634e8ca4faSJohn McCall   // the two sides.
3641860b520SIvan A. Kosarev   LValue DestLV = CGF.MakeAddrLValue(dest.getAddress(), type);
3651860b520SIvan A. Kosarev   LValue SrcLV = CGF.MakeAddrLValue(src.getAddress(), type);
366*e78fac51SRichard Smith   CGF.EmitAggregateCopy(DestLV, SrcLV, type, dest.mayOverlap(),
3677f416cc4SJohn McCall                         dest.isVolatile() || src.isVolatile());
3687a51313dSChris Lattner }
3697a51313dSChris Lattner 
370c83ed824SSebastian Redl /// \brief Emit the initializer for a std::initializer_list initialized with a
371c83ed824SSebastian Redl /// real initializer list.
372cc1b96d3SRichard Smith void
373cc1b96d3SRichard Smith AggExprEmitter::VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr *E) {
374cc1b96d3SRichard Smith   // Emit an array containing the elements.  The array is externally destructed
375cc1b96d3SRichard Smith   // if the std::initializer_list object is.
376cc1b96d3SRichard Smith   ASTContext &Ctx = CGF.getContext();
377cc1b96d3SRichard Smith   LValue Array = CGF.EmitLValue(E->getSubExpr());
378cc1b96d3SRichard Smith   assert(Array.isSimple() && "initializer_list array not a simple lvalue");
3797f416cc4SJohn McCall   Address ArrayPtr = Array.getAddress();
380c83ed824SSebastian Redl 
381cc1b96d3SRichard Smith   const ConstantArrayType *ArrayType =
382cc1b96d3SRichard Smith       Ctx.getAsConstantArrayType(E->getSubExpr()->getType());
383cc1b96d3SRichard Smith   assert(ArrayType && "std::initializer_list constructed from non-array");
384c83ed824SSebastian Redl 
385cc1b96d3SRichard Smith   // FIXME: Perform the checks on the field types in SemaInit.
386cc1b96d3SRichard Smith   RecordDecl *Record = E->getType()->castAs<RecordType>()->getDecl();
387cc1b96d3SRichard Smith   RecordDecl::field_iterator Field = Record->field_begin();
388cc1b96d3SRichard Smith   if (Field == Record->field_end()) {
389cc1b96d3SRichard Smith     CGF.ErrorUnsupported(E, "weird std::initializer_list");
390f2e0a307SSebastian Redl     return;
391c83ed824SSebastian Redl   }
392c83ed824SSebastian Redl 
393c83ed824SSebastian Redl   // Start pointer.
394cc1b96d3SRichard Smith   if (!Field->getType()->isPointerType() ||
395cc1b96d3SRichard Smith       !Ctx.hasSameType(Field->getType()->getPointeeType(),
396cc1b96d3SRichard Smith                        ArrayType->getElementType())) {
397cc1b96d3SRichard Smith     CGF.ErrorUnsupported(E, "weird std::initializer_list");
398f2e0a307SSebastian Redl     return;
399c83ed824SSebastian Redl   }
400c83ed824SSebastian Redl 
401cc1b96d3SRichard Smith   AggValueSlot Dest = EnsureSlot(E->getType());
4027f416cc4SJohn McCall   LValue DestLV = CGF.MakeAddrLValue(Dest.getAddress(), E->getType());
403cc1b96d3SRichard Smith   LValue Start = CGF.EmitLValueForFieldInitialization(DestLV, *Field);
404cc1b96d3SRichard Smith   llvm::Value *Zero = llvm::ConstantInt::get(CGF.PtrDiffTy, 0);
405cc1b96d3SRichard Smith   llvm::Value *IdxStart[] = { Zero, Zero };
406cc1b96d3SRichard Smith   llvm::Value *ArrayStart =
4077f416cc4SJohn McCall       Builder.CreateInBoundsGEP(ArrayPtr.getPointer(), IdxStart, "arraystart");
408cc1b96d3SRichard Smith   CGF.EmitStoreThroughLValue(RValue::get(ArrayStart), Start);
409cc1b96d3SRichard Smith   ++Field;
410cc1b96d3SRichard Smith 
411cc1b96d3SRichard Smith   if (Field == Record->field_end()) {
412cc1b96d3SRichard Smith     CGF.ErrorUnsupported(E, "weird std::initializer_list");
413f2e0a307SSebastian Redl     return;
414c83ed824SSebastian Redl   }
415cc1b96d3SRichard Smith 
416cc1b96d3SRichard Smith   llvm::Value *Size = Builder.getInt(ArrayType->getSize());
417cc1b96d3SRichard Smith   LValue EndOrLength = CGF.EmitLValueForFieldInitialization(DestLV, *Field);
418cc1b96d3SRichard Smith   if (Field->getType()->isPointerType() &&
419cc1b96d3SRichard Smith       Ctx.hasSameType(Field->getType()->getPointeeType(),
420cc1b96d3SRichard Smith                       ArrayType->getElementType())) {
421c83ed824SSebastian Redl     // End pointer.
422cc1b96d3SRichard Smith     llvm::Value *IdxEnd[] = { Zero, Size };
423cc1b96d3SRichard Smith     llvm::Value *ArrayEnd =
4247f416cc4SJohn McCall         Builder.CreateInBoundsGEP(ArrayPtr.getPointer(), IdxEnd, "arrayend");
425cc1b96d3SRichard Smith     CGF.EmitStoreThroughLValue(RValue::get(ArrayEnd), EndOrLength);
426cc1b96d3SRichard Smith   } else if (Ctx.hasSameType(Field->getType(), Ctx.getSizeType())) {
427c83ed824SSebastian Redl     // Length.
428cc1b96d3SRichard Smith     CGF.EmitStoreThroughLValue(RValue::get(Size), EndOrLength);
429c83ed824SSebastian Redl   } else {
430cc1b96d3SRichard Smith     CGF.ErrorUnsupported(E, "weird std::initializer_list");
431f2e0a307SSebastian Redl     return;
432c83ed824SSebastian Redl   }
433c83ed824SSebastian Redl }
434c83ed824SSebastian Redl 
4358edda962SRichard Smith /// \brief Determine if E is a trivial array filler, that is, one that is
4368edda962SRichard Smith /// equivalent to zero-initialization.
4378edda962SRichard Smith static bool isTrivialFiller(Expr *E) {
4388edda962SRichard Smith   if (!E)
4398edda962SRichard Smith     return true;
4408edda962SRichard Smith 
4418edda962SRichard Smith   if (isa<ImplicitValueInitExpr>(E))
4428edda962SRichard Smith     return true;
4438edda962SRichard Smith 
4448edda962SRichard Smith   if (auto *ILE = dyn_cast<InitListExpr>(E)) {
4458edda962SRichard Smith     if (ILE->getNumInits())
4468edda962SRichard Smith       return false;
4478edda962SRichard Smith     return isTrivialFiller(ILE->getArrayFiller());
4488edda962SRichard Smith   }
4498edda962SRichard Smith 
4508edda962SRichard Smith   if (auto *Cons = dyn_cast_or_null<CXXConstructExpr>(E))
4518edda962SRichard Smith     return Cons->getConstructor()->isDefaultConstructor() &&
4528edda962SRichard Smith            Cons->getConstructor()->isTrivial();
4538edda962SRichard Smith 
4548edda962SRichard Smith   // FIXME: Are there other cases where we can avoid emitting an initializer?
4558edda962SRichard Smith   return false;
4568edda962SRichard Smith }
4578edda962SRichard Smith 
458c83ed824SSebastian Redl /// \brief Emit initialization of an array from an initializer list.
4597f416cc4SJohn McCall void AggExprEmitter::EmitArrayInit(Address DestPtr, llvm::ArrayType *AType,
460e0ef348cSIvan A. Kosarev                                    QualType ArrayQTy, InitListExpr *E) {
461c83ed824SSebastian Redl   uint64_t NumInitElements = E->getNumInits();
462c83ed824SSebastian Redl 
463c83ed824SSebastian Redl   uint64_t NumArrayElements = AType->getNumElements();
464c83ed824SSebastian Redl   assert(NumInitElements <= NumArrayElements);
465c83ed824SSebastian Redl 
466e0ef348cSIvan A. Kosarev   QualType elementType =
467e0ef348cSIvan A. Kosarev       CGF.getContext().getAsArrayType(ArrayQTy)->getElementType();
468e0ef348cSIvan A. Kosarev 
469c83ed824SSebastian Redl   // DestPtr is an array*.  Construct an elementType* by drilling
470c83ed824SSebastian Redl   // down a level.
471c83ed824SSebastian Redl   llvm::Value *zero = llvm::ConstantInt::get(CGF.SizeTy, 0);
472c83ed824SSebastian Redl   llvm::Value *indices[] = { zero, zero };
473c83ed824SSebastian Redl   llvm::Value *begin =
4747f416cc4SJohn McCall     Builder.CreateInBoundsGEP(DestPtr.getPointer(), indices, "arrayinit.begin");
4757f416cc4SJohn McCall 
4767f416cc4SJohn McCall   CharUnits elementSize = CGF.getContext().getTypeSizeInChars(elementType);
4777f416cc4SJohn McCall   CharUnits elementAlign =
4787f416cc4SJohn McCall     DestPtr.getAlignment().alignmentOfArrayElement(elementSize);
479c83ed824SSebastian Redl 
480e0ef348cSIvan A. Kosarev   // Consider initializing the array by copying from a global. For this to be
481e0ef348cSIvan A. Kosarev   // more efficient than per-element initialization, the size of the elements
482e0ef348cSIvan A. Kosarev   // with explicit initializers should be large enough.
483e0ef348cSIvan A. Kosarev   if (NumInitElements * elementSize.getQuantity() > 16 &&
484e0ef348cSIvan A. Kosarev       elementType.isTriviallyCopyableType(CGF.getContext())) {
485e0ef348cSIvan A. Kosarev     CodeGen::CodeGenModule &CGM = CGF.CGM;
486e0ef348cSIvan A. Kosarev     ConstantEmitter Emitter(CGM);
487e0ef348cSIvan A. Kosarev     LangAS AS = ArrayQTy.getAddressSpace();
488e0ef348cSIvan A. Kosarev     if (llvm::Constant *C = Emitter.tryEmitForInitializer(E, AS, ArrayQTy)) {
489e0ef348cSIvan A. Kosarev       auto GV = new llvm::GlobalVariable(
490e0ef348cSIvan A. Kosarev           CGM.getModule(), C->getType(),
491e0ef348cSIvan A. Kosarev           CGM.isTypeConstant(ArrayQTy, /* ExcludeCtorDtor= */ true),
492e0ef348cSIvan A. Kosarev           llvm::GlobalValue::PrivateLinkage, C, "constinit",
493e0ef348cSIvan A. Kosarev           /* InsertBefore= */ nullptr, llvm::GlobalVariable::NotThreadLocal,
494e0ef348cSIvan A. Kosarev           CGM.getContext().getTargetAddressSpace(AS));
495e0ef348cSIvan A. Kosarev       Emitter.finalize(GV);
496e0ef348cSIvan A. Kosarev       CharUnits Align = CGM.getContext().getTypeAlignInChars(ArrayQTy);
497e0ef348cSIvan A. Kosarev       GV->setAlignment(Align.getQuantity());
498e0ef348cSIvan A. Kosarev       EmitFinalDestCopy(ArrayQTy, CGF.MakeAddrLValue(GV, ArrayQTy, Align));
499e0ef348cSIvan A. Kosarev       return;
500e0ef348cSIvan A. Kosarev     }
501e0ef348cSIvan A. Kosarev   }
502e0ef348cSIvan A. Kosarev 
503c83ed824SSebastian Redl   // Exception safety requires us to destroy all the
504c83ed824SSebastian Redl   // already-constructed members if an initializer throws.
505c83ed824SSebastian Redl   // For that, we'll need an EH cleanup.
506c83ed824SSebastian Redl   QualType::DestructionKind dtorKind = elementType.isDestructedType();
5077f416cc4SJohn McCall   Address endOfInit = Address::invalid();
508c83ed824SSebastian Redl   EHScopeStack::stable_iterator cleanup;
5098a13c418SCraig Topper   llvm::Instruction *cleanupDominator = nullptr;
510c83ed824SSebastian Redl   if (CGF.needsEHCleanup(dtorKind)) {
511c83ed824SSebastian Redl     // In principle we could tell the cleanup where we are more
512c83ed824SSebastian Redl     // directly, but the control flow can get so varied here that it
513c83ed824SSebastian Redl     // would actually be quite complex.  Therefore we go through an
514c83ed824SSebastian Redl     // alloca.
5157f416cc4SJohn McCall     endOfInit = CGF.CreateTempAlloca(begin->getType(), CGF.getPointerAlign(),
516c83ed824SSebastian Redl                                      "arrayinit.endOfInit");
517c83ed824SSebastian Redl     cleanupDominator = Builder.CreateStore(begin, endOfInit);
518c83ed824SSebastian Redl     CGF.pushIrregularPartialArrayCleanup(begin, endOfInit, elementType,
5197f416cc4SJohn McCall                                          elementAlign,
520c83ed824SSebastian Redl                                          CGF.getDestroyer(dtorKind));
521c83ed824SSebastian Redl     cleanup = CGF.EHStack.stable_begin();
522c83ed824SSebastian Redl 
523c83ed824SSebastian Redl   // Otherwise, remember that we didn't need a cleanup.
524c83ed824SSebastian Redl   } else {
525c83ed824SSebastian Redl     dtorKind = QualType::DK_none;
526c83ed824SSebastian Redl   }
527c83ed824SSebastian Redl 
528c83ed824SSebastian Redl   llvm::Value *one = llvm::ConstantInt::get(CGF.SizeTy, 1);
529c83ed824SSebastian Redl 
530c83ed824SSebastian Redl   // The 'current element to initialize'.  The invariants on this
531c83ed824SSebastian Redl   // variable are complicated.  Essentially, after each iteration of
532c83ed824SSebastian Redl   // the loop, it points to the last initialized element, except
533c83ed824SSebastian Redl   // that it points to the beginning of the array before any
534c83ed824SSebastian Redl   // elements have been initialized.
535c83ed824SSebastian Redl   llvm::Value *element = begin;
536c83ed824SSebastian Redl 
537c83ed824SSebastian Redl   // Emit the explicit initializers.
538c83ed824SSebastian Redl   for (uint64_t i = 0; i != NumInitElements; ++i) {
539c83ed824SSebastian Redl     // Advance to the next element.
540c83ed824SSebastian Redl     if (i > 0) {
541c83ed824SSebastian Redl       element = Builder.CreateInBoundsGEP(element, one, "arrayinit.element");
542c83ed824SSebastian Redl 
543c83ed824SSebastian Redl       // Tell the cleanup that it needs to destroy up to this
544c83ed824SSebastian Redl       // element.  TODO: some of these stores can be trivially
545c83ed824SSebastian Redl       // observed to be unnecessary.
5467f416cc4SJohn McCall       if (endOfInit.isValid()) Builder.CreateStore(element, endOfInit);
547c83ed824SSebastian Redl     }
548c83ed824SSebastian Redl 
5497f416cc4SJohn McCall     LValue elementLV =
5507f416cc4SJohn McCall       CGF.MakeAddrLValue(Address(element, elementAlign), elementType);
551615ed1a3SChad Rosier     EmitInitializationToLValue(E->getInit(i), elementLV);
552c83ed824SSebastian Redl   }
553c83ed824SSebastian Redl 
554c83ed824SSebastian Redl   // Check whether there's a non-trivial array-fill expression.
555c83ed824SSebastian Redl   Expr *filler = E->getArrayFiller();
5568edda962SRichard Smith   bool hasTrivialFiller = isTrivialFiller(filler);
557c83ed824SSebastian Redl 
558c83ed824SSebastian Redl   // Any remaining elements need to be zero-initialized, possibly
559c83ed824SSebastian Redl   // using the filler expression.  We can skip this if the we're
560c83ed824SSebastian Redl   // emitting to zeroed memory.
561c83ed824SSebastian Redl   if (NumInitElements != NumArrayElements &&
562c83ed824SSebastian Redl       !(Dest.isZeroed() && hasTrivialFiller &&
563c83ed824SSebastian Redl         CGF.getTypes().isZeroInitializable(elementType))) {
564c83ed824SSebastian Redl 
565c83ed824SSebastian Redl     // Use an actual loop.  This is basically
566c83ed824SSebastian Redl     //   do { *array++ = filler; } while (array != end);
567c83ed824SSebastian Redl 
568c83ed824SSebastian Redl     // Advance to the start of the rest of the array.
569c83ed824SSebastian Redl     if (NumInitElements) {
570c83ed824SSebastian Redl       element = Builder.CreateInBoundsGEP(element, one, "arrayinit.start");
5717f416cc4SJohn McCall       if (endOfInit.isValid()) Builder.CreateStore(element, endOfInit);
572c83ed824SSebastian Redl     }
573c83ed824SSebastian Redl 
574c83ed824SSebastian Redl     // Compute the end of the array.
575c83ed824SSebastian Redl     llvm::Value *end = Builder.CreateInBoundsGEP(begin,
576c83ed824SSebastian Redl                       llvm::ConstantInt::get(CGF.SizeTy, NumArrayElements),
577c83ed824SSebastian Redl                                                  "arrayinit.end");
578c83ed824SSebastian Redl 
579c83ed824SSebastian Redl     llvm::BasicBlock *entryBB = Builder.GetInsertBlock();
580c83ed824SSebastian Redl     llvm::BasicBlock *bodyBB = CGF.createBasicBlock("arrayinit.body");
581c83ed824SSebastian Redl 
582c83ed824SSebastian Redl     // Jump into the body.
583c83ed824SSebastian Redl     CGF.EmitBlock(bodyBB);
584c83ed824SSebastian Redl     llvm::PHINode *currentElement =
585c83ed824SSebastian Redl       Builder.CreatePHI(element->getType(), 2, "arrayinit.cur");
586c83ed824SSebastian Redl     currentElement->addIncoming(element, entryBB);
587c83ed824SSebastian Redl 
588c83ed824SSebastian Redl     // Emit the actual filler expression.
58972236372SRichard Smith     {
59072236372SRichard Smith       // C++1z [class.temporary]p5:
59172236372SRichard Smith       //   when a default constructor is called to initialize an element of
59272236372SRichard Smith       //   an array with no corresponding initializer [...] the destruction of
59372236372SRichard Smith       //   every temporary created in a default argument is sequenced before
59472236372SRichard Smith       //   the construction of the next array element, if any
59572236372SRichard Smith       CodeGenFunction::RunCleanupsScope CleanupsScope(CGF);
5967f416cc4SJohn McCall       LValue elementLV =
5977f416cc4SJohn McCall         CGF.MakeAddrLValue(Address(currentElement, elementAlign), elementType);
598c83ed824SSebastian Redl       if (filler)
599615ed1a3SChad Rosier         EmitInitializationToLValue(filler, elementLV);
600c83ed824SSebastian Redl       else
601c83ed824SSebastian Redl         EmitNullInitializationToLValue(elementLV);
60272236372SRichard Smith     }
603c83ed824SSebastian Redl 
604c83ed824SSebastian Redl     // Move on to the next element.
605c83ed824SSebastian Redl     llvm::Value *nextElement =
606c83ed824SSebastian Redl       Builder.CreateInBoundsGEP(currentElement, one, "arrayinit.next");
607c83ed824SSebastian Redl 
608c83ed824SSebastian Redl     // Tell the EH cleanup that we finished with the last element.
6097f416cc4SJohn McCall     if (endOfInit.isValid()) Builder.CreateStore(nextElement, endOfInit);
610c83ed824SSebastian Redl 
611c83ed824SSebastian Redl     // Leave the loop if we're done.
612c83ed824SSebastian Redl     llvm::Value *done = Builder.CreateICmpEQ(nextElement, end,
613c83ed824SSebastian Redl                                              "arrayinit.done");
614c83ed824SSebastian Redl     llvm::BasicBlock *endBB = CGF.createBasicBlock("arrayinit.end");
615c83ed824SSebastian Redl     Builder.CreateCondBr(done, endBB, bodyBB);
616c83ed824SSebastian Redl     currentElement->addIncoming(nextElement, Builder.GetInsertBlock());
617c83ed824SSebastian Redl 
618c83ed824SSebastian Redl     CGF.EmitBlock(endBB);
619c83ed824SSebastian Redl   }
620c83ed824SSebastian Redl 
621c83ed824SSebastian Redl   // Leave the partial-array cleanup if we entered one.
622c83ed824SSebastian Redl   if (dtorKind) CGF.DeactivateCleanupBlock(cleanup, cleanupDominator);
623c83ed824SSebastian Redl }
624c83ed824SSebastian Redl 
6257a51313dSChris Lattner //===----------------------------------------------------------------------===//
6267a51313dSChris Lattner //                            Visitor Methods
6277a51313dSChris Lattner //===----------------------------------------------------------------------===//
6287a51313dSChris Lattner 
629fe31481fSDouglas Gregor void AggExprEmitter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E){
630fe31481fSDouglas Gregor   Visit(E->GetTemporaryExpr());
631fe31481fSDouglas Gregor }
632fe31481fSDouglas Gregor 
6331bf5846aSJohn McCall void AggExprEmitter::VisitOpaqueValueExpr(OpaqueValueExpr *e) {
634797afe3aSAkira Hatanaka   // If this is a unique OVE, just visit its source expression.
635797afe3aSAkira Hatanaka   if (e->isUnique())
636797afe3aSAkira Hatanaka     Visit(e->getSourceExpr());
637797afe3aSAkira Hatanaka   else
638797afe3aSAkira Hatanaka     EmitFinalDestCopy(e->getType(), CGF.getOrCreateOpaqueLValueMapping(e));
6391bf5846aSJohn McCall }
6401bf5846aSJohn McCall 
6419b71f0cfSDouglas Gregor void
6429b71f0cfSDouglas Gregor AggExprEmitter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
643bea4c3d8SJohn McCall   if (Dest.isPotentiallyAliased() &&
644bea4c3d8SJohn McCall       E->getType().isPODType(CGF.getContext())) {
6456c9d31ebSDouglas Gregor     // For a POD type, just emit a load of the lvalue + a copy, because our
6466c9d31ebSDouglas Gregor     // compound literal might alias the destination.
6476c9d31ebSDouglas Gregor     EmitAggLoadOfLValue(E);
6486c9d31ebSDouglas Gregor     return;
6496c9d31ebSDouglas Gregor   }
6506c9d31ebSDouglas Gregor 
6519b71f0cfSDouglas Gregor   AggValueSlot Slot = EnsureSlot(E->getType());
6529b71f0cfSDouglas Gregor   CGF.EmitAggExpr(E->getInitializer(), Slot);
6539b71f0cfSDouglas Gregor }
6549b71f0cfSDouglas Gregor 
655a8ec7eb9SJohn McCall /// Attempt to look through various unimportant expressions to find a
656a8ec7eb9SJohn McCall /// cast of the given kind.
657a8ec7eb9SJohn McCall static Expr *findPeephole(Expr *op, CastKind kind) {
658a8ec7eb9SJohn McCall   while (true) {
659a8ec7eb9SJohn McCall     op = op->IgnoreParens();
660a8ec7eb9SJohn McCall     if (CastExpr *castE = dyn_cast<CastExpr>(op)) {
661a8ec7eb9SJohn McCall       if (castE->getCastKind() == kind)
662a8ec7eb9SJohn McCall         return castE->getSubExpr();
663a8ec7eb9SJohn McCall       if (castE->getCastKind() == CK_NoOp)
664a8ec7eb9SJohn McCall         continue;
665a8ec7eb9SJohn McCall     }
6668a13c418SCraig Topper     return nullptr;
667a8ec7eb9SJohn McCall   }
668a8ec7eb9SJohn McCall }
6699b71f0cfSDouglas Gregor 
670ec143777SAnders Carlsson void AggExprEmitter::VisitCastExpr(CastExpr *E) {
6712bf9b4c0SAlexey Bataev   if (const auto *ECE = dyn_cast<ExplicitCastExpr>(E))
6722bf9b4c0SAlexey Bataev     CGF.CGM.EmitExplicitCastExprType(ECE, &CGF);
6731fb7ae9eSAnders Carlsson   switch (E->getCastKind()) {
6748a01a751SAnders Carlsson   case CK_Dynamic: {
67569d0d262SRichard Smith     // FIXME: Can this actually happen? We have no test coverage for it.
6761c073f47SDouglas Gregor     assert(isa<CXXDynamicCastExpr>(E) && "CK_Dynamic without a dynamic_cast?");
67769d0d262SRichard Smith     LValue LV = CGF.EmitCheckedLValue(E->getSubExpr(),
6784d1458edSRichard Smith                                       CodeGenFunction::TCK_Load);
6791c073f47SDouglas Gregor     // FIXME: Do we also need to handle property references here?
6801c073f47SDouglas Gregor     if (LV.isSimple())
6811c073f47SDouglas Gregor       CGF.EmitDynamicCast(LV.getAddress(), cast<CXXDynamicCastExpr>(E));
6821c073f47SDouglas Gregor     else
6831c073f47SDouglas Gregor       CGF.CGM.ErrorUnsupported(E, "non-simple lvalue dynamic_cast");
6841c073f47SDouglas Gregor 
6857a626f63SJohn McCall     if (!Dest.isIgnored())
6861c073f47SDouglas Gregor       CGF.CGM.ErrorUnsupported(E, "lvalue dynamic_cast with a destination");
6871c073f47SDouglas Gregor     break;
6881c073f47SDouglas Gregor   }
6891c073f47SDouglas Gregor 
690e302792bSJohn McCall   case CK_ToUnion: {
691892bb0caSReid Kleckner     // Evaluate even if the destination is ignored.
692892bb0caSReid Kleckner     if (Dest.isIgnored()) {
693892bb0caSReid Kleckner       CGF.EmitAnyExpr(E->getSubExpr(), AggValueSlot::ignored(),
694892bb0caSReid Kleckner                       /*ignoreResult=*/true);
695892bb0caSReid Kleckner       break;
696892bb0caSReid Kleckner     }
69758989b71SJohn McCall 
6987ffcf93bSNuno Lopes     // GCC union extension
6992e442a00SDaniel Dunbar     QualType Ty = E->getSubExpr()->getType();
7007f416cc4SJohn McCall     Address CastPtr =
7017f416cc4SJohn McCall       Builder.CreateElementBitCast(Dest.getAddress(), CGF.ConvertType(Ty));
7021553b190SJohn McCall     EmitInitializationToLValue(E->getSubExpr(),
703615ed1a3SChad Rosier                                CGF.MakeAddrLValue(CastPtr, Ty));
7041fb7ae9eSAnders Carlsson     break;
7057ffcf93bSNuno Lopes   }
7067ffcf93bSNuno Lopes 
707e302792bSJohn McCall   case CK_DerivedToBase:
708e302792bSJohn McCall   case CK_BaseToDerived:
709e302792bSJohn McCall   case CK_UncheckedDerivedToBase: {
71083d382b1SDavid Blaikie     llvm_unreachable("cannot perform hierarchy conversion in EmitAggExpr: "
711aae38d66SDouglas Gregor                 "should have been unpacked before we got here");
712aae38d66SDouglas Gregor   }
713aae38d66SDouglas Gregor 
714a8ec7eb9SJohn McCall   case CK_NonAtomicToAtomic:
715a8ec7eb9SJohn McCall   case CK_AtomicToNonAtomic: {
716a8ec7eb9SJohn McCall     bool isToAtomic = (E->getCastKind() == CK_NonAtomicToAtomic);
717a8ec7eb9SJohn McCall 
718a8ec7eb9SJohn McCall     // Determine the atomic and value types.
719a8ec7eb9SJohn McCall     QualType atomicType = E->getSubExpr()->getType();
720a8ec7eb9SJohn McCall     QualType valueType = E->getType();
721a8ec7eb9SJohn McCall     if (isToAtomic) std::swap(atomicType, valueType);
722a8ec7eb9SJohn McCall 
723a8ec7eb9SJohn McCall     assert(atomicType->isAtomicType());
724a8ec7eb9SJohn McCall     assert(CGF.getContext().hasSameUnqualifiedType(valueType,
725a8ec7eb9SJohn McCall                           atomicType->castAs<AtomicType>()->getValueType()));
726a8ec7eb9SJohn McCall 
727a8ec7eb9SJohn McCall     // Just recurse normally if we're ignoring the result or the
728a8ec7eb9SJohn McCall     // atomic type doesn't change representation.
729a8ec7eb9SJohn McCall     if (Dest.isIgnored() || !CGF.CGM.isPaddedAtomicType(atomicType)) {
730a8ec7eb9SJohn McCall       return Visit(E->getSubExpr());
731a8ec7eb9SJohn McCall     }
732a8ec7eb9SJohn McCall 
733a8ec7eb9SJohn McCall     CastKind peepholeTarget =
734a8ec7eb9SJohn McCall       (isToAtomic ? CK_AtomicToNonAtomic : CK_NonAtomicToAtomic);
735a8ec7eb9SJohn McCall 
736a8ec7eb9SJohn McCall     // These two cases are reverses of each other; try to peephole them.
737a8ec7eb9SJohn McCall     if (Expr *op = findPeephole(E->getSubExpr(), peepholeTarget)) {
738a8ec7eb9SJohn McCall       assert(CGF.getContext().hasSameUnqualifiedType(op->getType(),
739a8ec7eb9SJohn McCall                                                      E->getType()) &&
740a8ec7eb9SJohn McCall            "peephole significantly changed types?");
741a8ec7eb9SJohn McCall       return Visit(op);
742a8ec7eb9SJohn McCall     }
743a8ec7eb9SJohn McCall 
744a8ec7eb9SJohn McCall     // If we're converting an r-value of non-atomic type to an r-value
745be4504dfSEli Friedman     // of atomic type, just emit directly into the relevant sub-object.
746a8ec7eb9SJohn McCall     if (isToAtomic) {
747be4504dfSEli Friedman       AggValueSlot valueDest = Dest;
748be4504dfSEli Friedman       if (!valueDest.isIgnored() && CGF.CGM.isPaddedAtomicType(atomicType)) {
749be4504dfSEli Friedman         // Zero-initialize.  (Strictly speaking, we only need to intialize
750be4504dfSEli Friedman         // the padding at the end, but this is simpler.)
751be4504dfSEli Friedman         if (!Dest.isZeroed())
7527f416cc4SJohn McCall           CGF.EmitNullInitialization(Dest.getAddress(), atomicType);
753be4504dfSEli Friedman 
754be4504dfSEli Friedman         // Build a GEP to refer to the subobject.
7557f416cc4SJohn McCall         Address valueAddr =
7567f416cc4SJohn McCall             CGF.Builder.CreateStructGEP(valueDest.getAddress(), 0,
7577f416cc4SJohn McCall                                         CharUnits());
758be4504dfSEli Friedman         valueDest = AggValueSlot::forAddr(valueAddr,
759be4504dfSEli Friedman                                           valueDest.getQualifiers(),
760be4504dfSEli Friedman                                           valueDest.isExternallyDestructed(),
761be4504dfSEli Friedman                                           valueDest.requiresGCollection(),
762be4504dfSEli Friedman                                           valueDest.isPotentiallyAliased(),
763*e78fac51SRichard Smith                                           AggValueSlot::DoesNotOverlap,
764be4504dfSEli Friedman                                           AggValueSlot::IsZeroed);
765be4504dfSEli Friedman       }
766be4504dfSEli Friedman 
767035b39e3SEli Friedman       CGF.EmitAggExpr(E->getSubExpr(), valueDest);
768a8ec7eb9SJohn McCall       return;
769a8ec7eb9SJohn McCall     }
770a8ec7eb9SJohn McCall 
771a8ec7eb9SJohn McCall     // Otherwise, we're converting an atomic type to a non-atomic type.
772be4504dfSEli Friedman     // Make an atomic temporary, emit into that, and then copy the value out.
773a8ec7eb9SJohn McCall     AggValueSlot atomicSlot =
774a8ec7eb9SJohn McCall       CGF.CreateAggTemp(atomicType, "atomic-to-nonatomic.temp");
775a8ec7eb9SJohn McCall     CGF.EmitAggExpr(E->getSubExpr(), atomicSlot);
776a8ec7eb9SJohn McCall 
7777f416cc4SJohn McCall     Address valueAddr =
7787f416cc4SJohn McCall       Builder.CreateStructGEP(atomicSlot.getAddress(), 0, CharUnits());
779a8ec7eb9SJohn McCall     RValue rvalue = RValue::getAggregate(valueAddr, atomicSlot.isVolatile());
780a8ec7eb9SJohn McCall     return EmitFinalDestCopy(valueType, rvalue);
781a8ec7eb9SJohn McCall   }
782a8ec7eb9SJohn McCall 
7834e8ca4faSJohn McCall   case CK_LValueToRValue:
7844e8ca4faSJohn McCall     // If we're loading from a volatile type, force the destination
7854e8ca4faSJohn McCall     // into existence.
7864e8ca4faSJohn McCall     if (E->getSubExpr()->getType().isVolatileQualified()) {
7874e8ca4faSJohn McCall       EnsureDest(E->getType());
7884e8ca4faSJohn McCall       return Visit(E->getSubExpr());
7894e8ca4faSJohn McCall     }
790a8ec7eb9SJohn McCall 
791f3b3ccdaSAdrian Prantl     LLVM_FALLTHROUGH;
7924e8ca4faSJohn McCall 
793e302792bSJohn McCall   case CK_NoOp:
794e302792bSJohn McCall   case CK_UserDefinedConversion:
795e302792bSJohn McCall   case CK_ConstructorConversion:
7962a69547fSEli Friedman     assert(CGF.getContext().hasSameUnqualifiedType(E->getSubExpr()->getType(),
7972a69547fSEli Friedman                                                    E->getType()) &&
7980f398c44SChris Lattner            "Implicit cast types must be compatible");
7997a51313dSChris Lattner     Visit(E->getSubExpr());
8001fb7ae9eSAnders Carlsson     break;
801b05a3e55SAnders Carlsson 
802e302792bSJohn McCall   case CK_LValueBitCast:
803f3735e01SJohn McCall     llvm_unreachable("should not be emitting lvalue bitcast as rvalue");
80431996343SJohn McCall 
805f3735e01SJohn McCall   case CK_Dependent:
806f3735e01SJohn McCall   case CK_BitCast:
807f3735e01SJohn McCall   case CK_ArrayToPointerDecay:
808f3735e01SJohn McCall   case CK_FunctionToPointerDecay:
809f3735e01SJohn McCall   case CK_NullToPointer:
810f3735e01SJohn McCall   case CK_NullToMemberPointer:
811f3735e01SJohn McCall   case CK_BaseToDerivedMemberPointer:
812f3735e01SJohn McCall   case CK_DerivedToBaseMemberPointer:
813f3735e01SJohn McCall   case CK_MemberPointerToBoolean:
814c62bb391SJohn McCall   case CK_ReinterpretMemberPointer:
815f3735e01SJohn McCall   case CK_IntegralToPointer:
816f3735e01SJohn McCall   case CK_PointerToIntegral:
817f3735e01SJohn McCall   case CK_PointerToBoolean:
818f3735e01SJohn McCall   case CK_ToVoid:
819f3735e01SJohn McCall   case CK_VectorSplat:
820f3735e01SJohn McCall   case CK_IntegralCast:
821df1ed009SGeorge Burgess IV   case CK_BooleanToSignedIntegral:
822f3735e01SJohn McCall   case CK_IntegralToBoolean:
823f3735e01SJohn McCall   case CK_IntegralToFloating:
824f3735e01SJohn McCall   case CK_FloatingToIntegral:
825f3735e01SJohn McCall   case CK_FloatingToBoolean:
826f3735e01SJohn McCall   case CK_FloatingCast:
8279320b87cSJohn McCall   case CK_CPointerToObjCPointerCast:
8289320b87cSJohn McCall   case CK_BlockPointerToObjCPointerCast:
829f3735e01SJohn McCall   case CK_AnyPointerToBlockPointerCast:
830f3735e01SJohn McCall   case CK_ObjCObjectLValueCast:
831f3735e01SJohn McCall   case CK_FloatingRealToComplex:
832f3735e01SJohn McCall   case CK_FloatingComplexToReal:
833f3735e01SJohn McCall   case CK_FloatingComplexToBoolean:
834f3735e01SJohn McCall   case CK_FloatingComplexCast:
835f3735e01SJohn McCall   case CK_FloatingComplexToIntegralComplex:
836f3735e01SJohn McCall   case CK_IntegralRealToComplex:
837f3735e01SJohn McCall   case CK_IntegralComplexToReal:
838f3735e01SJohn McCall   case CK_IntegralComplexToBoolean:
839f3735e01SJohn McCall   case CK_IntegralComplexCast:
840f3735e01SJohn McCall   case CK_IntegralComplexToFloatingComplex:
8412d637d2eSJohn McCall   case CK_ARCProduceObject:
8422d637d2eSJohn McCall   case CK_ARCConsumeObject:
8432d637d2eSJohn McCall   case CK_ARCReclaimReturnedObject:
8442d637d2eSJohn McCall   case CK_ARCExtendBlockObject:
845ed90df38SDouglas Gregor   case CK_CopyAndAutoreleaseBlockObject:
84634866c77SEli Friedman   case CK_BuiltinFnToFnPtr:
8471b4fb3e0SGuy Benyei   case CK_ZeroToOCLEvent:
84889831421SEgor Churaev   case CK_ZeroToOCLQueue:
849e1468322SDavid Tweed   case CK_AddressSpaceConversion:
8500bc4b2d3SYaxun Liu   case CK_IntToOCLSampler:
851f3735e01SJohn McCall     llvm_unreachable("cast kind invalid for aggregate types");
8521fb7ae9eSAnders Carlsson   }
8537a51313dSChris Lattner }
8547a51313dSChris Lattner 
8550f398c44SChris Lattner void AggExprEmitter::VisitCallExpr(const CallExpr *E) {
856ced8bdf7SDavid Majnemer   if (E->getCallReturnType(CGF.getContext())->isReferenceType()) {
857ddcbfe7bSAnders Carlsson     EmitAggLoadOfLValue(E);
858ddcbfe7bSAnders Carlsson     return;
859ddcbfe7bSAnders Carlsson   }
860ddcbfe7bSAnders Carlsson 
86156e5a2e1SGeorge Burgess IV   withReturnValueSlot(E, [&](ReturnValueSlot Slot) {
86256e5a2e1SGeorge Burgess IV     return CGF.EmitCallExpr(E, Slot);
86356e5a2e1SGeorge Burgess IV   });
8647a51313dSChris Lattner }
8650f398c44SChris Lattner 
8660f398c44SChris Lattner void AggExprEmitter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
86756e5a2e1SGeorge Burgess IV   withReturnValueSlot(E, [&](ReturnValueSlot Slot) {
86856e5a2e1SGeorge Burgess IV     return CGF.EmitObjCMessageExpr(E, Slot);
86956e5a2e1SGeorge Burgess IV   });
870b1d329daSChris Lattner }
8717a51313dSChris Lattner 
8720f398c44SChris Lattner void AggExprEmitter::VisitBinComma(const BinaryOperator *E) {
873a2342eb8SJohn McCall   CGF.EmitIgnoredExpr(E->getLHS());
8747a626f63SJohn McCall   Visit(E->getRHS());
8754b0e2a30SEli Friedman }
8764b0e2a30SEli Friedman 
8777a51313dSChris Lattner void AggExprEmitter::VisitStmtExpr(const StmtExpr *E) {
878ce1de617SJohn McCall   CodeGenFunction::StmtExprEvaluation eval(CGF);
8797a626f63SJohn McCall   CGF.EmitCompoundStmt(*E->getSubStmt(), true, Dest);
8807a51313dSChris Lattner }
8817a51313dSChris Lattner 
8827a51313dSChris Lattner void AggExprEmitter::VisitBinaryOperator(const BinaryOperator *E) {
883e302792bSJohn McCall   if (E->getOpcode() == BO_PtrMemD || E->getOpcode() == BO_PtrMemI)
884ffba662dSFariborz Jahanian     VisitPointerToDataMemberBinaryOperator(E);
885ffba662dSFariborz Jahanian   else
886a7c8cf62SDaniel Dunbar     CGF.ErrorUnsupported(E, "aggregate binary expression");
8877a51313dSChris Lattner }
8887a51313dSChris Lattner 
889ffba662dSFariborz Jahanian void AggExprEmitter::VisitPointerToDataMemberBinaryOperator(
890ffba662dSFariborz Jahanian                                                     const BinaryOperator *E) {
891ffba662dSFariborz Jahanian   LValue LV = CGF.EmitPointerToDataMemberBinaryExpr(E);
8924e8ca4faSJohn McCall   EmitFinalDestCopy(E->getType(), LV);
8934e8ca4faSJohn McCall }
8944e8ca4faSJohn McCall 
8954e8ca4faSJohn McCall /// Is the value of the given expression possibly a reference to or
8964e8ca4faSJohn McCall /// into a __block variable?
8974e8ca4faSJohn McCall static bool isBlockVarRef(const Expr *E) {
8984e8ca4faSJohn McCall   // Make sure we look through parens.
8994e8ca4faSJohn McCall   E = E->IgnoreParens();
9004e8ca4faSJohn McCall 
9014e8ca4faSJohn McCall   // Check for a direct reference to a __block variable.
9024e8ca4faSJohn McCall   if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
9034e8ca4faSJohn McCall     const VarDecl *var = dyn_cast<VarDecl>(DRE->getDecl());
9044e8ca4faSJohn McCall     return (var && var->hasAttr<BlocksAttr>());
9054e8ca4faSJohn McCall   }
9064e8ca4faSJohn McCall 
9074e8ca4faSJohn McCall   // More complicated stuff.
9084e8ca4faSJohn McCall 
9094e8ca4faSJohn McCall   // Binary operators.
9104e8ca4faSJohn McCall   if (const BinaryOperator *op = dyn_cast<BinaryOperator>(E)) {
9114e8ca4faSJohn McCall     // For an assignment or pointer-to-member operation, just care
9124e8ca4faSJohn McCall     // about the LHS.
9134e8ca4faSJohn McCall     if (op->isAssignmentOp() || op->isPtrMemOp())
9144e8ca4faSJohn McCall       return isBlockVarRef(op->getLHS());
9154e8ca4faSJohn McCall 
9164e8ca4faSJohn McCall     // For a comma, just care about the RHS.
9174e8ca4faSJohn McCall     if (op->getOpcode() == BO_Comma)
9184e8ca4faSJohn McCall       return isBlockVarRef(op->getRHS());
9194e8ca4faSJohn McCall 
9204e8ca4faSJohn McCall     // FIXME: pointer arithmetic?
9214e8ca4faSJohn McCall     return false;
9224e8ca4faSJohn McCall 
9234e8ca4faSJohn McCall   // Check both sides of a conditional operator.
9244e8ca4faSJohn McCall   } else if (const AbstractConditionalOperator *op
9254e8ca4faSJohn McCall                = dyn_cast<AbstractConditionalOperator>(E)) {
9264e8ca4faSJohn McCall     return isBlockVarRef(op->getTrueExpr())
9274e8ca4faSJohn McCall         || isBlockVarRef(op->getFalseExpr());
9284e8ca4faSJohn McCall 
9294e8ca4faSJohn McCall   // OVEs are required to support BinaryConditionalOperators.
9304e8ca4faSJohn McCall   } else if (const OpaqueValueExpr *op
9314e8ca4faSJohn McCall                = dyn_cast<OpaqueValueExpr>(E)) {
9324e8ca4faSJohn McCall     if (const Expr *src = op->getSourceExpr())
9334e8ca4faSJohn McCall       return isBlockVarRef(src);
9344e8ca4faSJohn McCall 
9354e8ca4faSJohn McCall   // Casts are necessary to get things like (*(int*)&var) = foo().
9364e8ca4faSJohn McCall   // We don't really care about the kind of cast here, except
9374e8ca4faSJohn McCall   // we don't want to look through l2r casts, because it's okay
9384e8ca4faSJohn McCall   // to get the *value* in a __block variable.
9394e8ca4faSJohn McCall   } else if (const CastExpr *cast = dyn_cast<CastExpr>(E)) {
9404e8ca4faSJohn McCall     if (cast->getCastKind() == CK_LValueToRValue)
9414e8ca4faSJohn McCall       return false;
9424e8ca4faSJohn McCall     return isBlockVarRef(cast->getSubExpr());
9434e8ca4faSJohn McCall 
9444e8ca4faSJohn McCall   // Handle unary operators.  Again, just aggressively look through
9454e8ca4faSJohn McCall   // it, ignoring the operation.
9464e8ca4faSJohn McCall   } else if (const UnaryOperator *uop = dyn_cast<UnaryOperator>(E)) {
9474e8ca4faSJohn McCall     return isBlockVarRef(uop->getSubExpr());
9484e8ca4faSJohn McCall 
9494e8ca4faSJohn McCall   // Look into the base of a field access.
9504e8ca4faSJohn McCall   } else if (const MemberExpr *mem = dyn_cast<MemberExpr>(E)) {
9514e8ca4faSJohn McCall     return isBlockVarRef(mem->getBase());
9524e8ca4faSJohn McCall 
9534e8ca4faSJohn McCall   // Look into the base of a subscript.
9544e8ca4faSJohn McCall   } else if (const ArraySubscriptExpr *sub = dyn_cast<ArraySubscriptExpr>(E)) {
9554e8ca4faSJohn McCall     return isBlockVarRef(sub->getBase());
9564e8ca4faSJohn McCall   }
9574e8ca4faSJohn McCall 
9584e8ca4faSJohn McCall   return false;
959ffba662dSFariborz Jahanian }
960ffba662dSFariborz Jahanian 
9617a51313dSChris Lattner void AggExprEmitter::VisitBinAssign(const BinaryOperator *E) {
9627a51313dSChris Lattner   // For an assignment to work, the value on the right has
9637a51313dSChris Lattner   // to be compatible with the value on the left.
9642a69547fSEli Friedman   assert(CGF.getContext().hasSameUnqualifiedType(E->getLHS()->getType(),
9652a69547fSEli Friedman                                                  E->getRHS()->getType())
9667a51313dSChris Lattner          && "Invalid assignment");
967d0a30016SJohn McCall 
9684e8ca4faSJohn McCall   // If the LHS might be a __block variable, and the RHS can
9694e8ca4faSJohn McCall   // potentially cause a block copy, we need to evaluate the RHS first
9704e8ca4faSJohn McCall   // so that the assignment goes the right place.
9714e8ca4faSJohn McCall   // This is pretty semantically fragile.
9724e8ca4faSJohn McCall   if (isBlockVarRef(E->getLHS()) &&
97399514b91SFariborz Jahanian       E->getRHS()->HasSideEffects(CGF.getContext())) {
9744e8ca4faSJohn McCall     // Ensure that we have a destination, and evaluate the RHS into that.
9754e8ca4faSJohn McCall     EnsureDest(E->getRHS()->getType());
9764e8ca4faSJohn McCall     Visit(E->getRHS());
9774e8ca4faSJohn McCall 
9784e8ca4faSJohn McCall     // Now emit the LHS and copy into it.
979e30752c9SRichard Smith     LValue LHS = CGF.EmitCheckedLValue(E->getLHS(), CodeGenFunction::TCK_Store);
9804e8ca4faSJohn McCall 
981a8ec7eb9SJohn McCall     // That copy is an atomic copy if the LHS is atomic.
982a5b195a1SDavid Majnemer     if (LHS.getType()->isAtomicType() ||
983a5b195a1SDavid Majnemer         CGF.LValueIsSuitableForInlineAtomic(LHS)) {
984a8ec7eb9SJohn McCall       CGF.EmitAtomicStore(Dest.asRValue(), LHS, /*isInit*/ false);
985a8ec7eb9SJohn McCall       return;
986a8ec7eb9SJohn McCall     }
987a8ec7eb9SJohn McCall 
9884e8ca4faSJohn McCall     EmitCopy(E->getLHS()->getType(),
9894e8ca4faSJohn McCall              AggValueSlot::forLValue(LHS, AggValueSlot::IsDestructed,
99046759f4fSJohn McCall                                      needsGC(E->getLHS()->getType()),
991*e78fac51SRichard Smith                                      AggValueSlot::IsAliased,
992*e78fac51SRichard Smith                                      AggValueSlot::MayOverlap),
9934e8ca4faSJohn McCall              Dest);
99499514b91SFariborz Jahanian     return;
99599514b91SFariborz Jahanian   }
99699514b91SFariborz Jahanian 
9977a51313dSChris Lattner   LValue LHS = CGF.EmitLValue(E->getLHS());
9987a51313dSChris Lattner 
999a8ec7eb9SJohn McCall   // If we have an atomic type, evaluate into the destination and then
1000a8ec7eb9SJohn McCall   // do an atomic copy.
1001a5b195a1SDavid Majnemer   if (LHS.getType()->isAtomicType() ||
1002a5b195a1SDavid Majnemer       CGF.LValueIsSuitableForInlineAtomic(LHS)) {
1003a8ec7eb9SJohn McCall     EnsureDest(E->getRHS()->getType());
1004a8ec7eb9SJohn McCall     Visit(E->getRHS());
1005a8ec7eb9SJohn McCall     CGF.EmitAtomicStore(Dest.asRValue(), LHS, /*isInit*/ false);
1006a8ec7eb9SJohn McCall     return;
1007a8ec7eb9SJohn McCall   }
1008a8ec7eb9SJohn McCall 
10097a51313dSChris Lattner   // Codegen the RHS so that it stores directly into the LHS.
10108d6fc958SJohn McCall   AggValueSlot LHSSlot =
10118d6fc958SJohn McCall     AggValueSlot::forLValue(LHS, AggValueSlot::IsDestructed,
101246759f4fSJohn McCall                             needsGC(E->getLHS()->getType()),
1013*e78fac51SRichard Smith                             AggValueSlot::IsAliased,
1014*e78fac51SRichard Smith                             AggValueSlot::MayOverlap);
10157865220dSFariborz Jahanian   // A non-volatile aggregate destination might have volatile member.
10167865220dSFariborz Jahanian   if (!LHSSlot.isVolatile() &&
10177865220dSFariborz Jahanian       CGF.hasVolatileMember(E->getLHS()->getType()))
10187865220dSFariborz Jahanian     LHSSlot.setVolatile(true);
10197865220dSFariborz Jahanian 
10204e8ca4faSJohn McCall   CGF.EmitAggExpr(E->getRHS(), LHSSlot);
10214e8ca4faSJohn McCall 
10224e8ca4faSJohn McCall   // Copy into the destination if the assignment isn't ignored.
10234e8ca4faSJohn McCall   EmitFinalDestCopy(E->getType(), LHS);
10247a51313dSChris Lattner }
10257a51313dSChris Lattner 
1026c07a0c7eSJohn McCall void AggExprEmitter::
1027c07a0c7eSJohn McCall VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) {
1028a612e79bSDaniel Dunbar   llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true");
1029a612e79bSDaniel Dunbar   llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false");
1030a612e79bSDaniel Dunbar   llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end");
10317a51313dSChris Lattner 
1032c07a0c7eSJohn McCall   // Bind the common expression if necessary.
103348fd89adSEli Friedman   CodeGenFunction::OpaqueValueMapping binding(CGF, E);
1034c07a0c7eSJohn McCall 
1035ce1de617SJohn McCall   CodeGenFunction::ConditionalEvaluation eval(CGF);
103666242d6cSJustin Bogner   CGF.EmitBranchOnBoolExpr(E->getCond(), LHSBlock, RHSBlock,
103766242d6cSJustin Bogner                            CGF.getProfileCount(E));
10387a51313dSChris Lattner 
10395b26f65bSJohn McCall   // Save whether the destination's lifetime is externally managed.
1040cac93853SJohn McCall   bool isExternallyDestructed = Dest.isExternallyDestructed();
10417a51313dSChris Lattner 
1042ce1de617SJohn McCall   eval.begin(CGF);
1043ce1de617SJohn McCall   CGF.EmitBlock(LHSBlock);
104466242d6cSJustin Bogner   CGF.incrementProfileCounter(E);
1045c07a0c7eSJohn McCall   Visit(E->getTrueExpr());
1046ce1de617SJohn McCall   eval.end(CGF);
10477a51313dSChris Lattner 
1048ce1de617SJohn McCall   assert(CGF.HaveInsertPoint() && "expression evaluation ended with no IP!");
1049ce1de617SJohn McCall   CGF.Builder.CreateBr(ContBlock);
10507a51313dSChris Lattner 
10515b26f65bSJohn McCall   // If the result of an agg expression is unused, then the emission
10525b26f65bSJohn McCall   // of the LHS might need to create a destination slot.  That's fine
10535b26f65bSJohn McCall   // with us, and we can safely emit the RHS into the same slot, but
1054cac93853SJohn McCall   // we shouldn't claim that it's already being destructed.
1055cac93853SJohn McCall   Dest.setExternallyDestructed(isExternallyDestructed);
10565b26f65bSJohn McCall 
1057ce1de617SJohn McCall   eval.begin(CGF);
1058ce1de617SJohn McCall   CGF.EmitBlock(RHSBlock);
1059c07a0c7eSJohn McCall   Visit(E->getFalseExpr());
1060ce1de617SJohn McCall   eval.end(CGF);
10617a51313dSChris Lattner 
10627a51313dSChris Lattner   CGF.EmitBlock(ContBlock);
10637a51313dSChris Lattner }
10647a51313dSChris Lattner 
10655b2095ceSAnders Carlsson void AggExprEmitter::VisitChooseExpr(const ChooseExpr *CE) {
106675807f23SEli Friedman   Visit(CE->getChosenSubExpr());
10675b2095ceSAnders Carlsson }
10685b2095ceSAnders Carlsson 
106921911e89SEli Friedman void AggExprEmitter::VisitVAArgExpr(VAArgExpr *VE) {
1070c7d5c94fSCharles Davis   Address ArgValue = Address::invalid();
1071c7d5c94fSCharles Davis   Address ArgPtr = CGF.EmitVAArg(VE, ArgValue);
107213abd7e9SAnders Carlsson 
107329b5f086SJames Y Knight   // If EmitVAArg fails, emit an error.
10747f416cc4SJohn McCall   if (!ArgPtr.isValid()) {
107529b5f086SJames Y Knight     CGF.ErrorUnsupported(VE, "aggregate va_arg expression");
1076020cddcfSSebastian Redl     return;
1077020cddcfSSebastian Redl   }
107813abd7e9SAnders Carlsson 
10794e8ca4faSJohn McCall   EmitFinalDestCopy(VE->getType(), CGF.MakeAddrLValue(ArgPtr, VE->getType()));
108021911e89SEli Friedman }
108121911e89SEli Friedman 
10823be22e27SAnders Carlsson void AggExprEmitter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
10837a626f63SJohn McCall   // Ensure that we have a slot, but if we already do, remember
1084cac93853SJohn McCall   // whether it was externally destructed.
1085cac93853SJohn McCall   bool wasExternallyDestructed = Dest.isExternallyDestructed();
10864e8ca4faSJohn McCall   EnsureDest(E->getType());
1087cac93853SJohn McCall 
1088cac93853SJohn McCall   // We're going to push a destructor if there isn't already one.
1089cac93853SJohn McCall   Dest.setExternallyDestructed();
10903be22e27SAnders Carlsson 
10913be22e27SAnders Carlsson   Visit(E->getSubExpr());
10923be22e27SAnders Carlsson 
1093cac93853SJohn McCall   // Push that destructor we promised.
1094cac93853SJohn McCall   if (!wasExternallyDestructed)
10957f416cc4SJohn McCall     CGF.EmitCXXTemporary(E->getTemporary(), E->getType(), Dest.getAddress());
10963be22e27SAnders Carlsson }
10973be22e27SAnders Carlsson 
1098b7f8f594SAnders Carlsson void
10991619a504SAnders Carlsson AggExprEmitter::VisitCXXConstructExpr(const CXXConstructExpr *E) {
11007a626f63SJohn McCall   AggValueSlot Slot = EnsureSlot(E->getType());
11017a626f63SJohn McCall   CGF.EmitCXXConstructExpr(E, Slot);
1102c82b86dfSAnders Carlsson }
1103c82b86dfSAnders Carlsson 
11045179eb78SRichard Smith void AggExprEmitter::VisitCXXInheritedCtorInitExpr(
11055179eb78SRichard Smith     const CXXInheritedCtorInitExpr *E) {
11065179eb78SRichard Smith   AggValueSlot Slot = EnsureSlot(E->getType());
11075179eb78SRichard Smith   CGF.EmitInheritedCXXConstructorCall(
11085179eb78SRichard Smith       E->getConstructor(), E->constructsVBase(), Slot.getAddress(),
11095179eb78SRichard Smith       E->inheritedFromVBase(), E);
11105179eb78SRichard Smith }
11115179eb78SRichard Smith 
1112c370a7eeSEli Friedman void
1113c370a7eeSEli Friedman AggExprEmitter::VisitLambdaExpr(LambdaExpr *E) {
1114c370a7eeSEli Friedman   AggValueSlot Slot = EnsureSlot(E->getType());
1115c370a7eeSEli Friedman   CGF.EmitLambdaExpr(E, Slot);
1116c370a7eeSEli Friedman }
1117c370a7eeSEli Friedman 
11185d413781SJohn McCall void AggExprEmitter::VisitExprWithCleanups(ExprWithCleanups *E) {
111908ef4660SJohn McCall   CGF.enterFullExpression(E);
112008ef4660SJohn McCall   CodeGenFunction::RunCleanupsScope cleanups(CGF);
112108ef4660SJohn McCall   Visit(E->getSubExpr());
1122b7f8f594SAnders Carlsson }
1123b7f8f594SAnders Carlsson 
1124747eb784SDouglas Gregor void AggExprEmitter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
11257a626f63SJohn McCall   QualType T = E->getType();
11267a626f63SJohn McCall   AggValueSlot Slot = EnsureSlot(T);
11277f416cc4SJohn McCall   EmitNullInitializationToLValue(CGF.MakeAddrLValue(Slot.getAddress(), T));
112818ada985SAnders Carlsson }
112918ada985SAnders Carlsson 
113018ada985SAnders Carlsson void AggExprEmitter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
11317a626f63SJohn McCall   QualType T = E->getType();
11327a626f63SJohn McCall   AggValueSlot Slot = EnsureSlot(T);
11337f416cc4SJohn McCall   EmitNullInitializationToLValue(CGF.MakeAddrLValue(Slot.getAddress(), T));
1134ff3507b9SNuno Lopes }
1135ff3507b9SNuno Lopes 
113627a3631bSChris Lattner /// isSimpleZero - If emitting this value will obviously just cause a store of
113727a3631bSChris Lattner /// zero to memory, return true.  This can return false if uncertain, so it just
113827a3631bSChris Lattner /// handles simple cases.
113927a3631bSChris Lattner static bool isSimpleZero(const Expr *E, CodeGenFunction &CGF) {
114091147596SPeter Collingbourne   E = E->IgnoreParens();
114191147596SPeter Collingbourne 
114227a3631bSChris Lattner   // 0
114327a3631bSChris Lattner   if (const IntegerLiteral *IL = dyn_cast<IntegerLiteral>(E))
114427a3631bSChris Lattner     return IL->getValue() == 0;
114527a3631bSChris Lattner   // +0.0
114627a3631bSChris Lattner   if (const FloatingLiteral *FL = dyn_cast<FloatingLiteral>(E))
114727a3631bSChris Lattner     return FL->getValue().isPosZero();
114827a3631bSChris Lattner   // int()
114927a3631bSChris Lattner   if ((isa<ImplicitValueInitExpr>(E) || isa<CXXScalarValueInitExpr>(E)) &&
115027a3631bSChris Lattner       CGF.getTypes().isZeroInitializable(E->getType()))
115127a3631bSChris Lattner     return true;
115227a3631bSChris Lattner   // (int*)0 - Null pointer expressions.
115327a3631bSChris Lattner   if (const CastExpr *ICE = dyn_cast<CastExpr>(E))
1154402804b6SYaxun Liu     return ICE->getCastKind() == CK_NullToPointer &&
1155402804b6SYaxun Liu         CGF.getTypes().isPointerZeroInitializable(E->getType());
115627a3631bSChris Lattner   // '\0'
115727a3631bSChris Lattner   if (const CharacterLiteral *CL = dyn_cast<CharacterLiteral>(E))
115827a3631bSChris Lattner     return CL->getValue() == 0;
115927a3631bSChris Lattner 
116027a3631bSChris Lattner   // Otherwise, hard case: conservatively return false.
116127a3631bSChris Lattner   return false;
116227a3631bSChris Lattner }
116327a3631bSChris Lattner 
116427a3631bSChris Lattner 
1165b247350eSAnders Carlsson void
1166615ed1a3SChad Rosier AggExprEmitter::EmitInitializationToLValue(Expr *E, LValue LV) {
11671553b190SJohn McCall   QualType type = LV.getType();
1168df0fe27bSMike Stump   // FIXME: Ignore result?
1169579a05d7SChris Lattner   // FIXME: Are initializers affected by volatile?
117027a3631bSChris Lattner   if (Dest.isZeroed() && isSimpleZero(E, CGF)) {
117127a3631bSChris Lattner     // Storing "i32 0" to a zero'd memory location is a noop.
117247fb9508SJohn McCall     return;
1173d82a2ce3SRichard Smith   } else if (isa<ImplicitValueInitExpr>(E) || isa<CXXScalarValueInitExpr>(E)) {
117447fb9508SJohn McCall     return EmitNullInitializationToLValue(LV);
1175cb77930dSYunzhong Gao   } else if (isa<NoInitExpr>(E)) {
1176cb77930dSYunzhong Gao     // Do nothing.
1177cb77930dSYunzhong Gao     return;
11781553b190SJohn McCall   } else if (type->isReferenceType()) {
1179a1c9d4d9SRichard Smith     RValue RV = CGF.EmitReferenceBindingToExpr(E);
118047fb9508SJohn McCall     return CGF.EmitStoreThroughLValue(RV, LV);
118147fb9508SJohn McCall   }
118247fb9508SJohn McCall 
118347fb9508SJohn McCall   switch (CGF.getEvaluationKind(type)) {
118447fb9508SJohn McCall   case TEK_Complex:
118547fb9508SJohn McCall     CGF.EmitComplexExprIntoLValue(E, LV, /*isInit*/ true);
118647fb9508SJohn McCall     return;
118747fb9508SJohn McCall   case TEK_Aggregate:
11888d6fc958SJohn McCall     CGF.EmitAggExpr(E, AggValueSlot::forLValue(LV,
11898d6fc958SJohn McCall                                                AggValueSlot::IsDestructed,
11908d6fc958SJohn McCall                                       AggValueSlot::DoesNotNeedGCBarriers,
1191a5efa738SJohn McCall                                                AggValueSlot::IsNotAliased,
1192*e78fac51SRichard Smith                                                AggValueSlot::MayOverlap,
11931553b190SJohn McCall                                                Dest.isZeroed()));
119447fb9508SJohn McCall     return;
119547fb9508SJohn McCall   case TEK_Scalar:
119647fb9508SJohn McCall     if (LV.isSimple()) {
11978a13c418SCraig Topper       CGF.EmitScalarInit(E, /*D=*/nullptr, LV, /*Captured=*/false);
11986e313210SEli Friedman     } else {
119955e1fbc8SJohn McCall       CGF.EmitStoreThroughLValue(RValue::get(CGF.EmitScalarExpr(E)), LV);
12007a51313dSChris Lattner     }
120147fb9508SJohn McCall     return;
120247fb9508SJohn McCall   }
120347fb9508SJohn McCall   llvm_unreachable("bad evaluation kind");
1204579a05d7SChris Lattner }
1205579a05d7SChris Lattner 
12061553b190SJohn McCall void AggExprEmitter::EmitNullInitializationToLValue(LValue lv) {
12071553b190SJohn McCall   QualType type = lv.getType();
12081553b190SJohn McCall 
120927a3631bSChris Lattner   // If the destination slot is already zeroed out before the aggregate is
121027a3631bSChris Lattner   // copied into it, we don't have to emit any zeros here.
12111553b190SJohn McCall   if (Dest.isZeroed() && CGF.getTypes().isZeroInitializable(type))
121227a3631bSChris Lattner     return;
121327a3631bSChris Lattner 
121447fb9508SJohn McCall   if (CGF.hasScalarEvaluationKind(type)) {
1215d82a2ce3SRichard Smith     // For non-aggregates, we can store the appropriate null constant.
1216d82a2ce3SRichard Smith     llvm::Value *null = CGF.CGM.EmitNullConstant(type);
121791d5bb1eSEli Friedman     // Note that the following is not equivalent to
121891d5bb1eSEli Friedman     // EmitStoreThroughBitfieldLValue for ARC types.
1219cb3785e4SEli Friedman     if (lv.isBitField()) {
122091d5bb1eSEli Friedman       CGF.EmitStoreThroughBitfieldLValue(RValue::get(null), lv);
1221cb3785e4SEli Friedman     } else {
122291d5bb1eSEli Friedman       assert(lv.isSimple());
122391d5bb1eSEli Friedman       CGF.EmitStoreOfScalar(null, lv, /* isInitialization */ true);
1224cb3785e4SEli Friedman     }
1225579a05d7SChris Lattner   } else {
1226579a05d7SChris Lattner     // There's a potential optimization opportunity in combining
1227579a05d7SChris Lattner     // memsets; that would be easy for arrays, but relatively
1228579a05d7SChris Lattner     // difficult for structures with the current code.
12291553b190SJohn McCall     CGF.EmitNullInitialization(lv.getAddress(), lv.getType());
1230579a05d7SChris Lattner   }
1231579a05d7SChris Lattner }
1232579a05d7SChris Lattner 
1233579a05d7SChris Lattner void AggExprEmitter::VisitInitListExpr(InitListExpr *E) {
1234f5d08c9eSEli Friedman #if 0
12356d11ec8cSEli Friedman   // FIXME: Assess perf here?  Figure out what cases are worth optimizing here
12366d11ec8cSEli Friedman   // (Length of globals? Chunks of zeroed-out space?).
1237f5d08c9eSEli Friedman   //
123818bb9284SMike Stump   // If we can, prefer a copy from a global; this is a lot less code for long
123918bb9284SMike Stump   // globals, and it's easier for the current optimizers to analyze.
12406d11ec8cSEli Friedman   if (llvm::Constant* C = CGF.CGM.EmitConstantExpr(E, E->getType(), &CGF)) {
1241c59bb48eSEli Friedman     llvm::GlobalVariable* GV =
12426d11ec8cSEli Friedman     new llvm::GlobalVariable(CGF.CGM.getModule(), C->getType(), true,
12436d11ec8cSEli Friedman                              llvm::GlobalValue::InternalLinkage, C, "");
12444e8ca4faSJohn McCall     EmitFinalDestCopy(E->getType(), CGF.MakeAddrLValue(GV, E->getType()));
1245c59bb48eSEli Friedman     return;
1246c59bb48eSEli Friedman   }
1247f5d08c9eSEli Friedman #endif
1248f53c0968SChris Lattner   if (E->hadArrayRangeDesignator())
1249bf7207a1SDouglas Gregor     CGF.ErrorUnsupported(E, "GNU array range designator extension");
1250bf7207a1SDouglas Gregor 
1251122f88d4SRichard Smith   if (E->isTransparent())
1252122f88d4SRichard Smith     return Visit(E->getInit(0));
1253122f88d4SRichard Smith 
1254be93c00aSRichard Smith   AggValueSlot Dest = EnsureSlot(E->getType());
1255be93c00aSRichard Smith 
12567f416cc4SJohn McCall   LValue DestLV = CGF.MakeAddrLValue(Dest.getAddress(), E->getType());
12577a626f63SJohn McCall 
1258579a05d7SChris Lattner   // Handle initialization of an array.
1259579a05d7SChris Lattner   if (E->getType()->isArrayType()) {
12607f416cc4SJohn McCall     auto AType = cast<llvm::ArrayType>(Dest.getAddress().getElementType());
1261e0ef348cSIvan A. Kosarev     EmitArrayInit(Dest.getAddress(), AType, E->getType(), E);
1262579a05d7SChris Lattner     return;
1263579a05d7SChris Lattner   }
1264579a05d7SChris Lattner 
1265579a05d7SChris Lattner   assert(E->getType()->isRecordType() && "Only support structs/unions here!");
1266579a05d7SChris Lattner 
1267579a05d7SChris Lattner   // Do struct initialization; this code just sets each individual member
1268579a05d7SChris Lattner   // to the approprate value.  This makes bitfield support automatic;
1269579a05d7SChris Lattner   // the disadvantage is that the generated code is more difficult for
1270579a05d7SChris Lattner   // the optimizer, especially with bitfields.
1271579a05d7SChris Lattner   unsigned NumInitElements = E->getNumInits();
12723b935d33SJohn McCall   RecordDecl *record = E->getType()->castAs<RecordType>()->getDecl();
127352bcf963SChris Lattner 
1274872307e2SRichard Smith   // We'll need to enter cleanup scopes in case any of the element
1275872307e2SRichard Smith   // initializers throws an exception.
1276872307e2SRichard Smith   SmallVector<EHScopeStack::stable_iterator, 16> cleanups;
1277872307e2SRichard Smith   llvm::Instruction *cleanupDominator = nullptr;
1278872307e2SRichard Smith 
1279872307e2SRichard Smith   unsigned curInitIndex = 0;
1280872307e2SRichard Smith 
1281872307e2SRichard Smith   // Emit initialization of base classes.
1282872307e2SRichard Smith   if (auto *CXXRD = dyn_cast<CXXRecordDecl>(record)) {
1283872307e2SRichard Smith     assert(E->getNumInits() >= CXXRD->getNumBases() &&
1284872307e2SRichard Smith            "missing initializer for base class");
1285872307e2SRichard Smith     for (auto &Base : CXXRD->bases()) {
1286872307e2SRichard Smith       assert(!Base.isVirtual() && "should not see vbases here");
1287872307e2SRichard Smith       auto *BaseRD = Base.getType()->getAsCXXRecordDecl();
1288872307e2SRichard Smith       Address V = CGF.GetAddressOfDirectBaseInCompleteClass(
1289872307e2SRichard Smith           Dest.getAddress(), CXXRD, BaseRD,
1290872307e2SRichard Smith           /*isBaseVirtual*/ false);
1291*e78fac51SRichard Smith       AggValueSlot AggSlot = AggValueSlot::forAddr(
1292*e78fac51SRichard Smith           V, Qualifiers(),
1293872307e2SRichard Smith           AggValueSlot::IsDestructed,
1294872307e2SRichard Smith           AggValueSlot::DoesNotNeedGCBarriers,
1295*e78fac51SRichard Smith           AggValueSlot::IsNotAliased,
1296*e78fac51SRichard Smith           CGF.overlapForBaseInit(CXXRD, BaseRD, Base.isVirtual()));
1297872307e2SRichard Smith       CGF.EmitAggExpr(E->getInit(curInitIndex++), AggSlot);
1298872307e2SRichard Smith 
1299872307e2SRichard Smith       if (QualType::DestructionKind dtorKind =
1300872307e2SRichard Smith               Base.getType().isDestructedType()) {
1301872307e2SRichard Smith         CGF.pushDestroy(dtorKind, V, Base.getType());
1302872307e2SRichard Smith         cleanups.push_back(CGF.EHStack.stable_begin());
1303872307e2SRichard Smith       }
1304872307e2SRichard Smith     }
1305872307e2SRichard Smith   }
1306872307e2SRichard Smith 
1307852c9db7SRichard Smith   // Prepare a 'this' for CXXDefaultInitExprs.
13087f416cc4SJohn McCall   CodeGenFunction::FieldConstructionScope FCS(CGF, Dest.getAddress());
1309852c9db7SRichard Smith 
13103b935d33SJohn McCall   if (record->isUnion()) {
13115169570eSDouglas Gregor     // Only initialize one field of a union. The field itself is
13125169570eSDouglas Gregor     // specified by the initializer list.
13135169570eSDouglas Gregor     if (!E->getInitializedFieldInUnion()) {
13145169570eSDouglas Gregor       // Empty union; we have nothing to do.
13155169570eSDouglas Gregor 
13165169570eSDouglas Gregor #ifndef NDEBUG
13175169570eSDouglas Gregor       // Make sure that it's really an empty and not a failure of
13185169570eSDouglas Gregor       // semantic analysis.
1319e8a8baefSAaron Ballman       for (const auto *Field : record->fields())
13205169570eSDouglas Gregor         assert(Field->isUnnamedBitfield() && "Only unnamed bitfields allowed");
13215169570eSDouglas Gregor #endif
13225169570eSDouglas Gregor       return;
13235169570eSDouglas Gregor     }
13245169570eSDouglas Gregor 
13255169570eSDouglas Gregor     // FIXME: volatility
13265169570eSDouglas Gregor     FieldDecl *Field = E->getInitializedFieldInUnion();
13275169570eSDouglas Gregor 
13287f1ff600SEli Friedman     LValue FieldLoc = CGF.EmitLValueForFieldInitialization(DestLV, Field);
13295169570eSDouglas Gregor     if (NumInitElements) {
13305169570eSDouglas Gregor       // Store the initializer into the field
1331615ed1a3SChad Rosier       EmitInitializationToLValue(E->getInit(0), FieldLoc);
13325169570eSDouglas Gregor     } else {
133327a3631bSChris Lattner       // Default-initialize to null.
13341553b190SJohn McCall       EmitNullInitializationToLValue(FieldLoc);
13355169570eSDouglas Gregor     }
13365169570eSDouglas Gregor 
13375169570eSDouglas Gregor     return;
13385169570eSDouglas Gregor   }
1339579a05d7SChris Lattner 
1340579a05d7SChris Lattner   // Here we iterate over the fields; this makes it simpler to both
1341579a05d7SChris Lattner   // default-initialize fields and skip over unnamed fields.
1342e8a8baefSAaron Ballman   for (const auto *field : record->fields()) {
13433b935d33SJohn McCall     // We're done once we hit the flexible array member.
13443b935d33SJohn McCall     if (field->getType()->isIncompleteArrayType())
134591f84216SDouglas Gregor       break;
134691f84216SDouglas Gregor 
13473b935d33SJohn McCall     // Always skip anonymous bitfields.
13483b935d33SJohn McCall     if (field->isUnnamedBitfield())
1349579a05d7SChris Lattner       continue;
135017bd094aSDouglas Gregor 
13513b935d33SJohn McCall     // We're done if we reach the end of the explicit initializers, we
13523b935d33SJohn McCall     // have a zeroed object, and the rest of the fields are
13533b935d33SJohn McCall     // zero-initializable.
13543b935d33SJohn McCall     if (curInitIndex == NumInitElements && Dest.isZeroed() &&
135527a3631bSChris Lattner         CGF.getTypes().isZeroInitializable(E->getType()))
135627a3631bSChris Lattner       break;
135727a3631bSChris Lattner 
13587f1ff600SEli Friedman 
1359e8a8baefSAaron Ballman     LValue LV = CGF.EmitLValueForFieldInitialization(DestLV, field);
13607c1baf46SFariborz Jahanian     // We never generate write-barries for initialized fields.
13613b935d33SJohn McCall     LV.setNonGC(true);
136227a3631bSChris Lattner 
13633b935d33SJohn McCall     if (curInitIndex < NumInitElements) {
1364e18aaf2cSChris Lattner       // Store the initializer into the field.
1365615ed1a3SChad Rosier       EmitInitializationToLValue(E->getInit(curInitIndex++), LV);
1366579a05d7SChris Lattner     } else {
13672c51880aSSimon Pilgrim       // We're out of initializers; default-initialize to null
13683b935d33SJohn McCall       EmitNullInitializationToLValue(LV);
13693b935d33SJohn McCall     }
13703b935d33SJohn McCall 
13713b935d33SJohn McCall     // Push a destructor if necessary.
13723b935d33SJohn McCall     // FIXME: if we have an array of structures, all explicitly
13733b935d33SJohn McCall     // initialized, we can end up pushing a linear number of cleanups.
13743b935d33SJohn McCall     bool pushedCleanup = false;
13753b935d33SJohn McCall     if (QualType::DestructionKind dtorKind
13763b935d33SJohn McCall           = field->getType().isDestructedType()) {
13773b935d33SJohn McCall       assert(LV.isSimple());
13783b935d33SJohn McCall       if (CGF.needsEHCleanup(dtorKind)) {
1379f4beacd0SJohn McCall         if (!cleanupDominator)
13807f416cc4SJohn McCall           cleanupDominator = CGF.Builder.CreateAlignedLoad(
13815ee4b9a1SReid Kleckner               CGF.Int8Ty,
13827f416cc4SJohn McCall               llvm::Constant::getNullValue(CGF.Int8PtrTy),
13837f416cc4SJohn McCall               CharUnits::One()); // placeholder
1384f4beacd0SJohn McCall 
13853b935d33SJohn McCall         CGF.pushDestroy(EHCleanup, LV.getAddress(), field->getType(),
13863b935d33SJohn McCall                         CGF.getDestroyer(dtorKind), false);
13873b935d33SJohn McCall         cleanups.push_back(CGF.EHStack.stable_begin());
13883b935d33SJohn McCall         pushedCleanup = true;
13893b935d33SJohn McCall       }
1390579a05d7SChris Lattner     }
139127a3631bSChris Lattner 
139227a3631bSChris Lattner     // If the GEP didn't get used because of a dead zero init or something
139327a3631bSChris Lattner     // else, clean it up for -O0 builds and general tidiness.
13943b935d33SJohn McCall     if (!pushedCleanup && LV.isSimple())
139527a3631bSChris Lattner       if (llvm::GetElementPtrInst *GEP =
13967f416cc4SJohn McCall             dyn_cast<llvm::GetElementPtrInst>(LV.getPointer()))
139727a3631bSChris Lattner         if (GEP->use_empty())
139827a3631bSChris Lattner           GEP->eraseFromParent();
13997a51313dSChris Lattner   }
14003b935d33SJohn McCall 
14013b935d33SJohn McCall   // Deactivate all the partial cleanups in reverse order, which
14023b935d33SJohn McCall   // generally means popping them.
14033b935d33SJohn McCall   for (unsigned i = cleanups.size(); i != 0; --i)
1404f4beacd0SJohn McCall     CGF.DeactivateCleanupBlock(cleanups[i-1], cleanupDominator);
1405f4beacd0SJohn McCall 
1406f4beacd0SJohn McCall   // Destroy the placeholder if we made one.
1407f4beacd0SJohn McCall   if (cleanupDominator)
1408f4beacd0SJohn McCall     cleanupDominator->eraseFromParent();
14097a51313dSChris Lattner }
14107a51313dSChris Lattner 
1411939b6880SRichard Smith void AggExprEmitter::VisitArrayInitLoopExpr(const ArrayInitLoopExpr *E,
1412939b6880SRichard Smith                                             llvm::Value *outerBegin) {
1413410306bfSRichard Smith   // Emit the common subexpression.
1414410306bfSRichard Smith   CodeGenFunction::OpaqueValueMapping binding(CGF, E->getCommonExpr());
1415410306bfSRichard Smith 
1416410306bfSRichard Smith   Address destPtr = EnsureSlot(E->getType()).getAddress();
1417410306bfSRichard Smith   uint64_t numElements = E->getArraySize().getZExtValue();
1418410306bfSRichard Smith 
1419410306bfSRichard Smith   if (!numElements)
1420410306bfSRichard Smith     return;
1421410306bfSRichard Smith 
1422410306bfSRichard Smith   // destPtr is an array*. Construct an elementType* by drilling down a level.
1423410306bfSRichard Smith   llvm::Value *zero = llvm::ConstantInt::get(CGF.SizeTy, 0);
1424410306bfSRichard Smith   llvm::Value *indices[] = {zero, zero};
1425410306bfSRichard Smith   llvm::Value *begin = Builder.CreateInBoundsGEP(destPtr.getPointer(), indices,
1426410306bfSRichard Smith                                                  "arrayinit.begin");
1427410306bfSRichard Smith 
1428939b6880SRichard Smith   // Prepare to special-case multidimensional array initialization: we avoid
1429939b6880SRichard Smith   // emitting multiple destructor loops in that case.
1430939b6880SRichard Smith   if (!outerBegin)
1431939b6880SRichard Smith     outerBegin = begin;
1432939b6880SRichard Smith   ArrayInitLoopExpr *InnerLoop = dyn_cast<ArrayInitLoopExpr>(E->getSubExpr());
1433939b6880SRichard Smith 
143430e304e2SRichard Smith   QualType elementType =
143530e304e2SRichard Smith       CGF.getContext().getAsArrayType(E->getType())->getElementType();
1436410306bfSRichard Smith   CharUnits elementSize = CGF.getContext().getTypeSizeInChars(elementType);
1437410306bfSRichard Smith   CharUnits elementAlign =
1438410306bfSRichard Smith       destPtr.getAlignment().alignmentOfArrayElement(elementSize);
1439410306bfSRichard Smith 
1440410306bfSRichard Smith   llvm::BasicBlock *entryBB = Builder.GetInsertBlock();
1441410306bfSRichard Smith   llvm::BasicBlock *bodyBB = CGF.createBasicBlock("arrayinit.body");
1442410306bfSRichard Smith 
1443410306bfSRichard Smith   // Jump into the body.
1444410306bfSRichard Smith   CGF.EmitBlock(bodyBB);
1445410306bfSRichard Smith   llvm::PHINode *index =
1446410306bfSRichard Smith       Builder.CreatePHI(zero->getType(), 2, "arrayinit.index");
1447410306bfSRichard Smith   index->addIncoming(zero, entryBB);
1448410306bfSRichard Smith   llvm::Value *element = Builder.CreateInBoundsGEP(begin, index);
1449410306bfSRichard Smith 
145030e304e2SRichard Smith   // Prepare for a cleanup.
145130e304e2SRichard Smith   QualType::DestructionKind dtorKind = elementType.isDestructedType();
145230e304e2SRichard Smith   EHScopeStack::stable_iterator cleanup;
1453939b6880SRichard Smith   if (CGF.needsEHCleanup(dtorKind) && !InnerLoop) {
1454939b6880SRichard Smith     if (outerBegin->getType() != element->getType())
1455939b6880SRichard Smith       outerBegin = Builder.CreateBitCast(outerBegin, element->getType());
1456939b6880SRichard Smith     CGF.pushRegularPartialArrayCleanup(outerBegin, element, elementType,
1457939b6880SRichard Smith                                        elementAlign,
1458939b6880SRichard Smith                                        CGF.getDestroyer(dtorKind));
145930e304e2SRichard Smith     cleanup = CGF.EHStack.stable_begin();
146030e304e2SRichard Smith   } else {
146130e304e2SRichard Smith     dtorKind = QualType::DK_none;
146230e304e2SRichard Smith   }
1463410306bfSRichard Smith 
1464410306bfSRichard Smith   // Emit the actual filler expression.
1465410306bfSRichard Smith   {
146630e304e2SRichard Smith     // Temporaries created in an array initialization loop are destroyed
146730e304e2SRichard Smith     // at the end of each iteration.
146830e304e2SRichard Smith     CodeGenFunction::RunCleanupsScope CleanupsScope(CGF);
1469410306bfSRichard Smith     CodeGenFunction::ArrayInitLoopExprScope Scope(CGF, index);
1470410306bfSRichard Smith     LValue elementLV =
1471410306bfSRichard Smith         CGF.MakeAddrLValue(Address(element, elementAlign), elementType);
1472939b6880SRichard Smith 
1473939b6880SRichard Smith     if (InnerLoop) {
1474939b6880SRichard Smith       // If the subexpression is an ArrayInitLoopExpr, share its cleanup.
1475939b6880SRichard Smith       auto elementSlot = AggValueSlot::forLValue(
1476939b6880SRichard Smith           elementLV, AggValueSlot::IsDestructed,
1477*e78fac51SRichard Smith           AggValueSlot::DoesNotNeedGCBarriers,
1478*e78fac51SRichard Smith           AggValueSlot::IsNotAliased,
1479*e78fac51SRichard Smith           AggValueSlot::DoesNotOverlap);
1480939b6880SRichard Smith       AggExprEmitter(CGF, elementSlot, false)
1481939b6880SRichard Smith           .VisitArrayInitLoopExpr(InnerLoop, outerBegin);
1482939b6880SRichard Smith     } else
1483410306bfSRichard Smith       EmitInitializationToLValue(E->getSubExpr(), elementLV);
1484410306bfSRichard Smith   }
1485410306bfSRichard Smith 
1486410306bfSRichard Smith   // Move on to the next element.
1487410306bfSRichard Smith   llvm::Value *nextIndex = Builder.CreateNUWAdd(
1488410306bfSRichard Smith       index, llvm::ConstantInt::get(CGF.SizeTy, 1), "arrayinit.next");
1489410306bfSRichard Smith   index->addIncoming(nextIndex, Builder.GetInsertBlock());
1490410306bfSRichard Smith 
1491410306bfSRichard Smith   // Leave the loop if we're done.
1492410306bfSRichard Smith   llvm::Value *done = Builder.CreateICmpEQ(
1493410306bfSRichard Smith       nextIndex, llvm::ConstantInt::get(CGF.SizeTy, numElements),
1494410306bfSRichard Smith       "arrayinit.done");
1495410306bfSRichard Smith   llvm::BasicBlock *endBB = CGF.createBasicBlock("arrayinit.end");
1496410306bfSRichard Smith   Builder.CreateCondBr(done, endBB, bodyBB);
1497410306bfSRichard Smith 
1498410306bfSRichard Smith   CGF.EmitBlock(endBB);
1499410306bfSRichard Smith 
1500410306bfSRichard Smith   // Leave the partial-array cleanup if we entered one.
150130e304e2SRichard Smith   if (dtorKind)
150230e304e2SRichard Smith     CGF.DeactivateCleanupBlock(cleanup, index);
1503410306bfSRichard Smith }
1504410306bfSRichard Smith 
1505cb77930dSYunzhong Gao void AggExprEmitter::VisitDesignatedInitUpdateExpr(DesignatedInitUpdateExpr *E) {
1506cb77930dSYunzhong Gao   AggValueSlot Dest = EnsureSlot(E->getType());
1507cb77930dSYunzhong Gao 
15087f416cc4SJohn McCall   LValue DestLV = CGF.MakeAddrLValue(Dest.getAddress(), E->getType());
1509cb77930dSYunzhong Gao   EmitInitializationToLValue(E->getBase(), DestLV);
1510cb77930dSYunzhong Gao   VisitInitListExpr(E->getUpdater());
1511cb77930dSYunzhong Gao }
1512cb77930dSYunzhong Gao 
15137a51313dSChris Lattner //===----------------------------------------------------------------------===//
15147a51313dSChris Lattner //                        Entry Points into this File
15157a51313dSChris Lattner //===----------------------------------------------------------------------===//
15167a51313dSChris Lattner 
151727a3631bSChris Lattner /// GetNumNonZeroBytesInInit - Get an approximate count of the number of
151827a3631bSChris Lattner /// non-zero bytes that will be stored when outputting the initializer for the
151927a3631bSChris Lattner /// specified initializer expression.
1520df94cb7dSKen Dyck static CharUnits GetNumNonZeroBytesInInit(const Expr *E, CodeGenFunction &CGF) {
152191147596SPeter Collingbourne   E = E->IgnoreParens();
152227a3631bSChris Lattner 
152327a3631bSChris Lattner   // 0 and 0.0 won't require any non-zero stores!
1524df94cb7dSKen Dyck   if (isSimpleZero(E, CGF)) return CharUnits::Zero();
152527a3631bSChris Lattner 
152627a3631bSChris Lattner   // If this is an initlist expr, sum up the size of sizes of the (present)
152727a3631bSChris Lattner   // elements.  If this is something weird, assume the whole thing is non-zero.
152827a3631bSChris Lattner   const InitListExpr *ILE = dyn_cast<InitListExpr>(E);
15298a13c418SCraig Topper   if (!ILE || !CGF.getTypes().isZeroInitializable(ILE->getType()))
1530df94cb7dSKen Dyck     return CGF.getContext().getTypeSizeInChars(E->getType());
153127a3631bSChris Lattner 
1532c5cc2fb9SChris Lattner   // InitListExprs for structs have to be handled carefully.  If there are
1533c5cc2fb9SChris Lattner   // reference members, we need to consider the size of the reference, not the
1534c5cc2fb9SChris Lattner   // referencee.  InitListExprs for unions and arrays can't have references.
15355cd84755SChris Lattner   if (const RecordType *RT = E->getType()->getAs<RecordType>()) {
15365cd84755SChris Lattner     if (!RT->isUnionType()) {
1537c5cc2fb9SChris Lattner       RecordDecl *SD = E->getType()->getAs<RecordType>()->getDecl();
1538df94cb7dSKen Dyck       CharUnits NumNonZeroBytes = CharUnits::Zero();
1539c5cc2fb9SChris Lattner 
1540c5cc2fb9SChris Lattner       unsigned ILEElement = 0;
1541872307e2SRichard Smith       if (auto *CXXRD = dyn_cast<CXXRecordDecl>(SD))
15426365e464SRichard Smith         while (ILEElement != CXXRD->getNumBases())
1543872307e2SRichard Smith           NumNonZeroBytes +=
1544872307e2SRichard Smith               GetNumNonZeroBytesInInit(ILE->getInit(ILEElement++), CGF);
1545e8a8baefSAaron Ballman       for (const auto *Field : SD->fields()) {
1546c5cc2fb9SChris Lattner         // We're done once we hit the flexible array member or run out of
1547c5cc2fb9SChris Lattner         // InitListExpr elements.
1548c5cc2fb9SChris Lattner         if (Field->getType()->isIncompleteArrayType() ||
1549c5cc2fb9SChris Lattner             ILEElement == ILE->getNumInits())
1550c5cc2fb9SChris Lattner           break;
1551c5cc2fb9SChris Lattner         if (Field->isUnnamedBitfield())
1552c5cc2fb9SChris Lattner           continue;
1553c5cc2fb9SChris Lattner 
1554c5cc2fb9SChris Lattner         const Expr *E = ILE->getInit(ILEElement++);
1555c5cc2fb9SChris Lattner 
1556c5cc2fb9SChris Lattner         // Reference values are always non-null and have the width of a pointer.
15575cd84755SChris Lattner         if (Field->getType()->isReferenceType())
1558df94cb7dSKen Dyck           NumNonZeroBytes += CGF.getContext().toCharUnitsFromBits(
1559c8e01705SJohn McCall               CGF.getTarget().getPointerWidth(0));
15605cd84755SChris Lattner         else
1561c5cc2fb9SChris Lattner           NumNonZeroBytes += GetNumNonZeroBytesInInit(E, CGF);
1562c5cc2fb9SChris Lattner       }
1563c5cc2fb9SChris Lattner 
1564c5cc2fb9SChris Lattner       return NumNonZeroBytes;
1565c5cc2fb9SChris Lattner     }
15665cd84755SChris Lattner   }
1567c5cc2fb9SChris Lattner 
1568c5cc2fb9SChris Lattner 
1569df94cb7dSKen Dyck   CharUnits NumNonZeroBytes = CharUnits::Zero();
157027a3631bSChris Lattner   for (unsigned i = 0, e = ILE->getNumInits(); i != e; ++i)
157127a3631bSChris Lattner     NumNonZeroBytes += GetNumNonZeroBytesInInit(ILE->getInit(i), CGF);
157227a3631bSChris Lattner   return NumNonZeroBytes;
157327a3631bSChris Lattner }
157427a3631bSChris Lattner 
157527a3631bSChris Lattner /// CheckAggExprForMemSetUse - If the initializer is large and has a lot of
157627a3631bSChris Lattner /// zeros in it, emit a memset and avoid storing the individual zeros.
157727a3631bSChris Lattner ///
157827a3631bSChris Lattner static void CheckAggExprForMemSetUse(AggValueSlot &Slot, const Expr *E,
157927a3631bSChris Lattner                                      CodeGenFunction &CGF) {
158027a3631bSChris Lattner   // If the slot is already known to be zeroed, nothing to do.  Don't mess with
158127a3631bSChris Lattner   // volatile stores.
15827f416cc4SJohn McCall   if (Slot.isZeroed() || Slot.isVolatile() || !Slot.getAddress().isValid())
15838a13c418SCraig Topper     return;
158427a3631bSChris Lattner 
158503535265SArgyrios Kyrtzidis   // C++ objects with a user-declared constructor don't need zero'ing.
15869c6890a7SRichard Smith   if (CGF.getLangOpts().CPlusPlus)
158703535265SArgyrios Kyrtzidis     if (const RecordType *RT = CGF.getContext()
158803535265SArgyrios Kyrtzidis                        .getBaseElementType(E->getType())->getAs<RecordType>()) {
158903535265SArgyrios Kyrtzidis       const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
159003535265SArgyrios Kyrtzidis       if (RD->hasUserDeclaredConstructor())
159103535265SArgyrios Kyrtzidis         return;
159203535265SArgyrios Kyrtzidis     }
159303535265SArgyrios Kyrtzidis 
159427a3631bSChris Lattner   // If the type is 16-bytes or smaller, prefer individual stores over memset.
1595*e78fac51SRichard Smith   CharUnits Size = Slot.getPreferredSize(CGF.getContext(), E->getType());
15967f416cc4SJohn McCall   if (Size <= CharUnits::fromQuantity(16))
159727a3631bSChris Lattner     return;
159827a3631bSChris Lattner 
159927a3631bSChris Lattner   // Check to see if over 3/4 of the initializer are known to be zero.  If so,
160027a3631bSChris Lattner   // we prefer to emit memset + individual stores for the rest.
1601239a3357SKen Dyck   CharUnits NumNonZeroBytes = GetNumNonZeroBytesInInit(E, CGF);
16027f416cc4SJohn McCall   if (NumNonZeroBytes*4 > Size)
160327a3631bSChris Lattner     return;
160427a3631bSChris Lattner 
160527a3631bSChris Lattner   // Okay, it seems like a good idea to use an initial memset, emit the call.
16067f416cc4SJohn McCall   llvm::Constant *SizeVal = CGF.Builder.getInt64(Size.getQuantity());
160727a3631bSChris Lattner 
16087f416cc4SJohn McCall   Address Loc = Slot.getAddress();
16097f416cc4SJohn McCall   Loc = CGF.Builder.CreateElementBitCast(Loc, CGF.Int8Ty);
16107f416cc4SJohn McCall   CGF.Builder.CreateMemSet(Loc, CGF.Builder.getInt8(0), SizeVal, false);
161127a3631bSChris Lattner 
161227a3631bSChris Lattner   // Tell the AggExprEmitter that the slot is known zero.
161327a3631bSChris Lattner   Slot.setZeroed();
161427a3631bSChris Lattner }
161527a3631bSChris Lattner 
161627a3631bSChris Lattner 
161727a3631bSChris Lattner 
161827a3631bSChris Lattner 
161925306cacSMike Stump /// EmitAggExpr - Emit the computation of the specified expression of aggregate
162025306cacSMike Stump /// type.  The result is computed into DestPtr.  Note that if DestPtr is null,
162125306cacSMike Stump /// the value of the aggregate expression is not needed.  If VolatileDest is
162225306cacSMike Stump /// true, DestPtr cannot be 0.
16234e8ca4faSJohn McCall void CodeGenFunction::EmitAggExpr(const Expr *E, AggValueSlot Slot) {
162447fb9508SJohn McCall   assert(E && hasAggregateEvaluationKind(E->getType()) &&
16257a51313dSChris Lattner          "Invalid aggregate expression to emit");
16267f416cc4SJohn McCall   assert((Slot.getAddress().isValid() || Slot.isIgnored()) &&
162727a3631bSChris Lattner          "slot has bits but no address");
16287a51313dSChris Lattner 
162927a3631bSChris Lattner   // Optimize the slot if possible.
163027a3631bSChris Lattner   CheckAggExprForMemSetUse(Slot, E, *this);
163127a3631bSChris Lattner 
16326aab1117SLeny Kholodov   AggExprEmitter(*this, Slot, Slot.isIgnored()).Visit(const_cast<Expr*>(E));
16337a51313dSChris Lattner }
16340bc8e86dSDaniel Dunbar 
1635d0bc7b9dSDaniel Dunbar LValue CodeGenFunction::EmitAggExprToLValue(const Expr *E) {
163647fb9508SJohn McCall   assert(hasAggregateEvaluationKind(E->getType()) && "Invalid argument!");
16377f416cc4SJohn McCall   Address Temp = CreateMemTemp(E->getType());
16382e442a00SDaniel Dunbar   LValue LV = MakeAddrLValue(Temp, E->getType());
16398d6fc958SJohn McCall   EmitAggExpr(E, AggValueSlot::forLValue(LV, AggValueSlot::IsNotDestructed,
164046759f4fSJohn McCall                                          AggValueSlot::DoesNotNeedGCBarriers,
1641*e78fac51SRichard Smith                                          AggValueSlot::IsNotAliased,
1642*e78fac51SRichard Smith                                          AggValueSlot::DoesNotOverlap));
16432e442a00SDaniel Dunbar   return LV;
1644d0bc7b9dSDaniel Dunbar }
1645d0bc7b9dSDaniel Dunbar 
1646*e78fac51SRichard Smith AggValueSlot::Overlap_t CodeGenFunction::overlapForBaseInit(
1647*e78fac51SRichard Smith     const CXXRecordDecl *RD, const CXXRecordDecl *BaseRD, bool IsVirtual) {
1648*e78fac51SRichard Smith   // Virtual bases are initialized first, in address order, so there's never
1649*e78fac51SRichard Smith   // any overlap during their initialization.
1650*e78fac51SRichard Smith   //
1651*e78fac51SRichard Smith   // FIXME: Under P0840, this is no longer true: the tail padding of a vbase
1652*e78fac51SRichard Smith   // of a field could be reused by a vbase of a containing class.
1653*e78fac51SRichard Smith   if (IsVirtual)
1654*e78fac51SRichard Smith     return AggValueSlot::DoesNotOverlap;
1655*e78fac51SRichard Smith 
1656*e78fac51SRichard Smith   // If the base class is laid out entirely within the nvsize of the derived
1657*e78fac51SRichard Smith   // class, its tail padding cannot yet be initialized, so we can issue
1658*e78fac51SRichard Smith   // stores at the full width of the base class.
1659*e78fac51SRichard Smith   const ASTRecordLayout &Layout = getContext().getASTRecordLayout(RD);
1660*e78fac51SRichard Smith   if (Layout.getBaseClassOffset(BaseRD) +
1661*e78fac51SRichard Smith           getContext().getASTRecordLayout(BaseRD).getSize() <=
1662*e78fac51SRichard Smith       Layout.getNonVirtualSize())
1663*e78fac51SRichard Smith     return AggValueSlot::DoesNotOverlap;
1664*e78fac51SRichard Smith 
1665*e78fac51SRichard Smith   // The tail padding may contain values we need to preserve.
1666*e78fac51SRichard Smith   return AggValueSlot::MayOverlap;
1667*e78fac51SRichard Smith }
1668*e78fac51SRichard Smith 
1669*e78fac51SRichard Smith void CodeGenFunction::EmitAggregateCopy(LValue Dest, LValue Src, QualType Ty,
1670*e78fac51SRichard Smith                                         AggValueSlot::Overlap_t MayOverlap,
1671*e78fac51SRichard Smith                                         bool isVolatile) {
1672615ed1a3SChad Rosier   assert(!Ty->isAnyComplexType() && "Shouldn't happen for complex");
16730bc8e86dSDaniel Dunbar 
16741860b520SIvan A. Kosarev   Address DestPtr = Dest.getAddress();
16751860b520SIvan A. Kosarev   Address SrcPtr = Src.getAddress();
16761860b520SIvan A. Kosarev 
16779c6890a7SRichard Smith   if (getLangOpts().CPlusPlus) {
1678615ed1a3SChad Rosier     if (const RecordType *RT = Ty->getAs<RecordType>()) {
1679615ed1a3SChad Rosier       CXXRecordDecl *Record = cast<CXXRecordDecl>(RT->getDecl());
1680615ed1a3SChad Rosier       assert((Record->hasTrivialCopyConstructor() ||
1681615ed1a3SChad Rosier               Record->hasTrivialCopyAssignment() ||
1682615ed1a3SChad Rosier               Record->hasTrivialMoveConstructor() ||
1683419bd094SRichard Smith               Record->hasTrivialMoveAssignment() ||
1684419bd094SRichard Smith               Record->isUnion()) &&
168516488472SRichard Smith              "Trying to aggregate-copy a type without a trivial copy/move "
1686f22101a0SDouglas Gregor              "constructor or assignment operator");
1687615ed1a3SChad Rosier       // Ignore empty classes in C++.
1688615ed1a3SChad Rosier       if (Record->isEmpty())
168916e94af6SAnders Carlsson         return;
169016e94af6SAnders Carlsson     }
169116e94af6SAnders Carlsson   }
169216e94af6SAnders Carlsson 
1693ca05dfefSChris Lattner   // Aggregate assignment turns into llvm.memcpy.  This is almost valid per
16943ef668c2SChris Lattner   // C99 6.5.16.1p3, which states "If the value being stored in an object is
16953ef668c2SChris Lattner   // read from another object that overlaps in anyway the storage of the first
16963ef668c2SChris Lattner   // object, then the overlap shall be exact and the two objects shall have
16973ef668c2SChris Lattner   // qualified or unqualified versions of a compatible type."
16983ef668c2SChris Lattner   //
1699ca05dfefSChris Lattner   // memcpy is not defined if the source and destination pointers are exactly
17003ef668c2SChris Lattner   // equal, but other compilers do this optimization, and almost every memcpy
17013ef668c2SChris Lattner   // implementation handles this case safely.  If there is a libc that does not
17023ef668c2SChris Lattner   // safely handle this, we can add a target hook.
17030bc8e86dSDaniel Dunbar 
1704*e78fac51SRichard Smith   // Get data size info for this aggregate. Don't copy the tail padding if this
1705*e78fac51SRichard Smith   // might be a potentially-overlapping subobject, since the tail padding might
1706*e78fac51SRichard Smith   // be occupied by a different object. Otherwise, copying it is fine.
17071ca66919SBenjamin Kramer   std::pair<CharUnits, CharUnits> TypeInfo;
1708*e78fac51SRichard Smith   if (MayOverlap)
17091ca66919SBenjamin Kramer     TypeInfo = getContext().getTypeInfoDataSizeInChars(Ty);
17101ca66919SBenjamin Kramer   else
17111ca66919SBenjamin Kramer     TypeInfo = getContext().getTypeInfoInChars(Ty);
1712615ed1a3SChad Rosier 
171316dc7b68SAlexey Bataev   llvm::Value *SizeVal = nullptr;
171416dc7b68SAlexey Bataev   if (TypeInfo.first.isZero()) {
171516dc7b68SAlexey Bataev     // But note that getTypeInfo returns 0 for a VLA.
171616dc7b68SAlexey Bataev     if (auto *VAT = dyn_cast_or_null<VariableArrayType>(
171716dc7b68SAlexey Bataev             getContext().getAsArrayType(Ty))) {
171816dc7b68SAlexey Bataev       QualType BaseEltTy;
171916dc7b68SAlexey Bataev       SizeVal = emitArrayLength(VAT, BaseEltTy, DestPtr);
1720*e78fac51SRichard Smith       TypeInfo = getContext().getTypeInfoInChars(BaseEltTy);
172116dc7b68SAlexey Bataev       assert(!TypeInfo.first.isZero());
172216dc7b68SAlexey Bataev       SizeVal = Builder.CreateNUWMul(
172316dc7b68SAlexey Bataev           SizeVal,
172416dc7b68SAlexey Bataev           llvm::ConstantInt::get(SizeTy, TypeInfo.first.getQuantity()));
172516dc7b68SAlexey Bataev     }
172616dc7b68SAlexey Bataev   }
172716dc7b68SAlexey Bataev   if (!SizeVal) {
172816dc7b68SAlexey Bataev     SizeVal = llvm::ConstantInt::get(SizeTy, TypeInfo.first.getQuantity());
172916dc7b68SAlexey Bataev   }
1730615ed1a3SChad Rosier 
1731615ed1a3SChad Rosier   // FIXME: If we have a volatile struct, the optimizer can remove what might
1732615ed1a3SChad Rosier   // appear to be `extra' memory ops:
1733615ed1a3SChad Rosier   //
1734615ed1a3SChad Rosier   // volatile struct { int i; } a, b;
1735615ed1a3SChad Rosier   //
1736615ed1a3SChad Rosier   // int main() {
1737615ed1a3SChad Rosier   //   a = b;
1738615ed1a3SChad Rosier   //   a = b;
1739615ed1a3SChad Rosier   // }
1740615ed1a3SChad Rosier   //
1741615ed1a3SChad Rosier   // we need to use a different call here.  We use isVolatile to indicate when
1742615ed1a3SChad Rosier   // either the source or the destination is volatile.
1743615ed1a3SChad Rosier 
17447f416cc4SJohn McCall   DestPtr = Builder.CreateElementBitCast(DestPtr, Int8Ty);
17457f416cc4SJohn McCall   SrcPtr = Builder.CreateElementBitCast(SrcPtr, Int8Ty);
1746615ed1a3SChad Rosier 
1747615ed1a3SChad Rosier   // Don't do any of the memmove_collectable tests if GC isn't set.
1748615ed1a3SChad Rosier   if (CGM.getLangOpts().getGC() == LangOptions::NonGC) {
1749615ed1a3SChad Rosier     // fall through
1750615ed1a3SChad Rosier   } else if (const RecordType *RecordTy = Ty->getAs<RecordType>()) {
1751615ed1a3SChad Rosier     RecordDecl *Record = RecordTy->getDecl();
1752615ed1a3SChad Rosier     if (Record->hasObjectMember()) {
1753615ed1a3SChad Rosier       CGM.getObjCRuntime().EmitGCMemmoveCollectable(*this, DestPtr, SrcPtr,
1754615ed1a3SChad Rosier                                                     SizeVal);
1755615ed1a3SChad Rosier       return;
1756615ed1a3SChad Rosier     }
1757615ed1a3SChad Rosier   } else if (Ty->isArrayType()) {
1758615ed1a3SChad Rosier     QualType BaseType = getContext().getBaseElementType(Ty);
1759615ed1a3SChad Rosier     if (const RecordType *RecordTy = BaseType->getAs<RecordType>()) {
1760615ed1a3SChad Rosier       if (RecordTy->getDecl()->hasObjectMember()) {
1761615ed1a3SChad Rosier         CGM.getObjCRuntime().EmitGCMemmoveCollectable(*this, DestPtr, SrcPtr,
1762615ed1a3SChad Rosier                                                       SizeVal);
1763615ed1a3SChad Rosier         return;
1764615ed1a3SChad Rosier       }
1765615ed1a3SChad Rosier     }
1766615ed1a3SChad Rosier   }
1767615ed1a3SChad Rosier 
17687f416cc4SJohn McCall   auto Inst = Builder.CreateMemCpy(DestPtr, SrcPtr, SizeVal, isVolatile);
17697f416cc4SJohn McCall 
177022695fceSDan Gohman   // Determine the metadata to describe the position of any padding in this
177122695fceSDan Gohman   // memcpy, as well as the TBAA tags for the members of the struct, in case
177222695fceSDan Gohman   // the optimizer wishes to expand it in to scalar memory operations.
17737f416cc4SJohn McCall   if (llvm::MDNode *TBAAStructTag = CGM.getTBAAStructInfo(Ty))
17747f416cc4SJohn McCall     Inst->setMetadata(llvm::LLVMContext::MD_tbaa_struct, TBAAStructTag);
17751860b520SIvan A. Kosarev 
17761860b520SIvan A. Kosarev   if (CGM.getCodeGenOpts().NewStructPathTBAA) {
17771860b520SIvan A. Kosarev     TBAAAccessInfo TBAAInfo = CGM.mergeTBAAInfoForMemoryTransfer(
17781860b520SIvan A. Kosarev         Dest.getTBAAInfo(), Src.getTBAAInfo());
17791860b520SIvan A. Kosarev     CGM.DecorateInstructionWithTBAA(Inst, TBAAInfo);
17801860b520SIvan A. Kosarev   }
17810bc8e86dSDaniel Dunbar }
1782