CGExprConstant.cpp revision 1de4d4e8cb2e9c88809fea8092bc6e835a5473d2
1//===--- CGExprConstant.cpp - Emit LLVM Code from Constant Expressions ----===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This contains code to emit Constant Expr nodes as LLVM code.
11//
12//===----------------------------------------------------------------------===//
13
14#include "CodeGenFunction.h"
15#include "CodeGenModule.h"
16#include "CGCXXABI.h"
17#include "CGObjCRuntime.h"
18#include "CGRecordLayout.h"
19#include "clang/AST/APValue.h"
20#include "clang/AST/ASTContext.h"
21#include "clang/AST/RecordLayout.h"
22#include "clang/AST/StmtVisitor.h"
23#include "clang/Basic/Builtins.h"
24#include "llvm/Constants.h"
25#include "llvm/Function.h"
26#include "llvm/GlobalVariable.h"
27#include "llvm/Target/TargetData.h"
28using namespace clang;
29using namespace CodeGen;
30
31//===----------------------------------------------------------------------===//
32//                            ConstStructBuilder
33//===----------------------------------------------------------------------===//
34
35namespace {
36class ConstStructBuilder {
37  CodeGenModule &CGM;
38  CodeGenFunction *CGF;
39
40  bool Packed;
41  CharUnits NextFieldOffsetInChars;
42  CharUnits LLVMStructAlignment;
43  std::vector<llvm::Constant *> Elements;
44public:
45  static llvm::Constant *BuildStruct(CodeGenModule &CGM, CodeGenFunction *CGF,
46                                     InitListExpr *ILE);
47
48private:
49  ConstStructBuilder(CodeGenModule &CGM, CodeGenFunction *CGF)
50    : CGM(CGM), CGF(CGF), Packed(false),
51    NextFieldOffsetInChars(CharUnits::Zero()),
52    LLVMStructAlignment(CharUnits::One()) { }
53
54  bool AppendField(const FieldDecl *Field, uint64_t FieldOffset,
55                   llvm::Constant *InitExpr);
56
57  void AppendBitField(const FieldDecl *Field, uint64_t FieldOffset,
58                      llvm::ConstantInt *InitExpr);
59
60  void AppendPadding(CharUnits PadSize);
61
62  void AppendTailPadding(CharUnits RecordSize);
63
64  void ConvertStructToPacked();
65
66  bool Build(InitListExpr *ILE);
67
68  CharUnits getAlignment(const llvm::Constant *C) const {
69    if (Packed)  return CharUnits::One();
70    return CharUnits::fromQuantity(
71        CGM.getTargetData().getABITypeAlignment(C->getType()));
72  }
73
74  CharUnits getSizeInChars(const llvm::Constant *C) const {
75    return CharUnits::fromQuantity(
76        CGM.getTargetData().getTypeAllocSize(C->getType()));
77  }
78};
79
80bool ConstStructBuilder::
81AppendField(const FieldDecl *Field, uint64_t FieldOffset,
82            llvm::Constant *InitCst) {
83
84  const ASTContext &Context = CGM.getContext();
85
86  CharUnits FieldOffsetInChars = Context.toCharUnitsFromBits(FieldOffset);
87
88  assert(NextFieldOffsetInChars <= FieldOffsetInChars
89         && "Field offset mismatch!");
90
91  CharUnits FieldAlignment = getAlignment(InitCst);
92
93  // Round up the field offset to the alignment of the field type.
94  CharUnits AlignedNextFieldOffsetInChars =
95    NextFieldOffsetInChars.RoundUpToAlignment(FieldAlignment);
96
97  if (AlignedNextFieldOffsetInChars > FieldOffsetInChars) {
98    assert(!Packed && "Alignment is wrong even with a packed struct!");
99
100    // Convert the struct to a packed struct.
101    ConvertStructToPacked();
102
103    AlignedNextFieldOffsetInChars = NextFieldOffsetInChars;
104  }
105
106  if (AlignedNextFieldOffsetInChars < FieldOffsetInChars) {
107    // We need to append padding.
108    AppendPadding(
109        FieldOffsetInChars - NextFieldOffsetInChars);
110
111    assert(NextFieldOffsetInChars == FieldOffsetInChars &&
112           "Did not add enough padding!");
113
114    AlignedNextFieldOffsetInChars = NextFieldOffsetInChars;
115  }
116
117  // Add the field.
118  Elements.push_back(InitCst);
119  NextFieldOffsetInChars = AlignedNextFieldOffsetInChars +
120                           getSizeInChars(InitCst);
121
122  if (Packed)
123    assert(LLVMStructAlignment == CharUnits::One() &&
124           "Packed struct not byte-aligned!");
125  else
126    LLVMStructAlignment = std::max(LLVMStructAlignment, FieldAlignment);
127
128  return true;
129}
130
131void ConstStructBuilder::AppendBitField(const FieldDecl *Field,
132                                        uint64_t FieldOffset,
133                                        llvm::ConstantInt *CI) {
134  const ASTContext &Context = CGM.getContext();
135  const uint64_t CharWidth = Context.getCharWidth();
136  uint64_t NextFieldOffsetInBits = Context.toBits(NextFieldOffsetInChars);
137  if (FieldOffset > NextFieldOffsetInBits) {
138    // We need to add padding.
139    CharUnits PadSize = Context.toCharUnitsFromBits(
140      llvm::RoundUpToAlignment(FieldOffset - NextFieldOffsetInBits,
141                               Context.Target.getCharAlign()));
142
143    AppendPadding(PadSize);
144  }
145
146  uint64_t FieldSize =
147    Field->getBitWidth()->EvaluateAsInt(Context).getZExtValue();
148
149  llvm::APInt FieldValue = CI->getValue();
150
151  // Promote the size of FieldValue if necessary
152  // FIXME: This should never occur, but currently it can because initializer
153  // constants are cast to bool, and because clang is not enforcing bitfield
154  // width limits.
155  if (FieldSize > FieldValue.getBitWidth())
156    FieldValue = FieldValue.zext(FieldSize);
157
158  // Truncate the size of FieldValue to the bit field size.
159  if (FieldSize < FieldValue.getBitWidth())
160    FieldValue = FieldValue.trunc(FieldSize);
161
162  NextFieldOffsetInBits = Context.toBits(NextFieldOffsetInChars);
163  if (FieldOffset < NextFieldOffsetInBits) {
164    // Either part of the field or the entire field can go into the previous
165    // byte.
166    assert(!Elements.empty() && "Elements can't be empty!");
167
168    unsigned BitsInPreviousByte = NextFieldOffsetInBits - FieldOffset;
169
170    bool FitsCompletelyInPreviousByte =
171      BitsInPreviousByte >= FieldValue.getBitWidth();
172
173    llvm::APInt Tmp = FieldValue;
174
175    if (!FitsCompletelyInPreviousByte) {
176      unsigned NewFieldWidth = FieldSize - BitsInPreviousByte;
177
178      if (CGM.getTargetData().isBigEndian()) {
179        Tmp = Tmp.lshr(NewFieldWidth);
180        Tmp = Tmp.trunc(BitsInPreviousByte);
181
182        // We want the remaining high bits.
183        FieldValue = FieldValue.trunc(NewFieldWidth);
184      } else {
185        Tmp = Tmp.trunc(BitsInPreviousByte);
186
187        // We want the remaining low bits.
188        FieldValue = FieldValue.lshr(BitsInPreviousByte);
189        FieldValue = FieldValue.trunc(NewFieldWidth);
190      }
191    }
192
193    Tmp = Tmp.zext(CharWidth);
194    if (CGM.getTargetData().isBigEndian()) {
195      if (FitsCompletelyInPreviousByte)
196        Tmp = Tmp.shl(BitsInPreviousByte - FieldValue.getBitWidth());
197    } else {
198      Tmp = Tmp.shl(CharWidth - BitsInPreviousByte);
199    }
200
201    // 'or' in the bits that go into the previous byte.
202    llvm::Value *LastElt = Elements.back();
203    if (llvm::ConstantInt *Val = dyn_cast<llvm::ConstantInt>(LastElt))
204      Tmp |= Val->getValue();
205    else {
206      assert(isa<llvm::UndefValue>(LastElt));
207      // If there is an undef field that we're adding to, it can either be a
208      // scalar undef (in which case, we just replace it with our field) or it
209      // is an array.  If it is an array, we have to pull one byte off the
210      // array so that the other undef bytes stay around.
211      if (!isa<llvm::IntegerType>(LastElt->getType())) {
212        // The undef padding will be a multibyte array, create a new smaller
213        // padding and then an hole for our i8 to get plopped into.
214        assert(isa<llvm::ArrayType>(LastElt->getType()) &&
215               "Expected array padding of undefs");
216        const llvm::ArrayType *AT = cast<llvm::ArrayType>(LastElt->getType());
217        assert(AT->getElementType()->isIntegerTy(CharWidth) &&
218               AT->getNumElements() != 0 &&
219               "Expected non-empty array padding of undefs");
220
221        // Remove the padding array.
222        NextFieldOffsetInChars -= CharUnits::fromQuantity(AT->getNumElements());
223        Elements.pop_back();
224
225        // Add the padding back in two chunks.
226        AppendPadding(CharUnits::fromQuantity(AT->getNumElements()-1));
227        AppendPadding(CharUnits::One());
228        assert(isa<llvm::UndefValue>(Elements.back()) &&
229               Elements.back()->getType()->isIntegerTy(CharWidth) &&
230               "Padding addition didn't work right");
231      }
232    }
233
234    Elements.back() = llvm::ConstantInt::get(CGM.getLLVMContext(), Tmp);
235
236    if (FitsCompletelyInPreviousByte)
237      return;
238  }
239
240  while (FieldValue.getBitWidth() > CharWidth) {
241    llvm::APInt Tmp;
242
243    if (CGM.getTargetData().isBigEndian()) {
244      // We want the high bits.
245      Tmp =
246        FieldValue.lshr(FieldValue.getBitWidth() - CharWidth).trunc(CharWidth);
247    } else {
248      // We want the low bits.
249      Tmp = FieldValue.trunc(CharWidth);
250
251      FieldValue = FieldValue.lshr(CharWidth);
252    }
253
254    Elements.push_back(llvm::ConstantInt::get(CGM.getLLVMContext(), Tmp));
255    ++NextFieldOffsetInChars;
256
257    FieldValue = FieldValue.trunc(FieldValue.getBitWidth() - CharWidth);
258  }
259
260  assert(FieldValue.getBitWidth() > 0 &&
261         "Should have at least one bit left!");
262  assert(FieldValue.getBitWidth() <= CharWidth &&
263         "Should not have more than a byte left!");
264
265  if (FieldValue.getBitWidth() < CharWidth) {
266    if (CGM.getTargetData().isBigEndian()) {
267      unsigned BitWidth = FieldValue.getBitWidth();
268
269      FieldValue = FieldValue.zext(CharWidth) << (CharWidth - BitWidth);
270    } else
271      FieldValue = FieldValue.zext(CharWidth);
272  }
273
274  // Append the last element.
275  Elements.push_back(llvm::ConstantInt::get(CGM.getLLVMContext(),
276                                            FieldValue));
277  ++NextFieldOffsetInChars;
278}
279
280void ConstStructBuilder::AppendPadding(CharUnits PadSize) {
281  if (PadSize.isZero())
282    return;
283
284  const llvm::Type *Ty = llvm::Type::getInt8Ty(CGM.getLLVMContext());
285  if (PadSize > CharUnits::One())
286    Ty = llvm::ArrayType::get(Ty, PadSize.getQuantity());
287
288  llvm::Constant *C = llvm::UndefValue::get(Ty);
289  Elements.push_back(C);
290  assert(getAlignment(C) == CharUnits::One() &&
291         "Padding must have 1 byte alignment!");
292
293  NextFieldOffsetInChars += getSizeInChars(C);
294}
295
296void ConstStructBuilder::AppendTailPadding(CharUnits RecordSize) {
297  assert(NextFieldOffsetInChars <= RecordSize &&
298         "Size mismatch!");
299
300  AppendPadding(RecordSize - NextFieldOffsetInChars);
301}
302
303void ConstStructBuilder::ConvertStructToPacked() {
304  std::vector<llvm::Constant *> PackedElements;
305  CharUnits ElementOffsetInChars = CharUnits::Zero();
306
307  for (unsigned i = 0, e = Elements.size(); i != e; ++i) {
308    llvm::Constant *C = Elements[i];
309
310    CharUnits ElementAlign = CharUnits::fromQuantity(
311      CGM.getTargetData().getABITypeAlignment(C->getType()));
312    CharUnits AlignedElementOffsetInChars =
313      ElementOffsetInChars.RoundUpToAlignment(ElementAlign);
314
315    if (AlignedElementOffsetInChars > ElementOffsetInChars) {
316      // We need some padding.
317      CharUnits NumChars =
318        AlignedElementOffsetInChars - ElementOffsetInChars;
319
320      const llvm::Type *Ty = llvm::Type::getInt8Ty(CGM.getLLVMContext());
321      if (NumChars > CharUnits::One())
322        Ty = llvm::ArrayType::get(Ty, NumChars.getQuantity());
323
324      llvm::Constant *Padding = llvm::UndefValue::get(Ty);
325      PackedElements.push_back(Padding);
326      ElementOffsetInChars += getSizeInChars(Padding);
327    }
328
329    PackedElements.push_back(C);
330    ElementOffsetInChars += getSizeInChars(C);
331  }
332
333  assert(ElementOffsetInChars == NextFieldOffsetInChars &&
334         "Packing the struct changed its size!");
335
336  Elements = PackedElements;
337  LLVMStructAlignment = CharUnits::One();
338  Packed = true;
339}
340
341bool ConstStructBuilder::Build(InitListExpr *ILE) {
342  RecordDecl *RD = ILE->getType()->getAs<RecordType>()->getDecl();
343  const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
344
345  unsigned FieldNo = 0;
346  unsigned ElementNo = 0;
347  for (RecordDecl::field_iterator Field = RD->field_begin(),
348       FieldEnd = RD->field_end(); Field != FieldEnd; ++Field, ++FieldNo) {
349
350    // If this is a union, skip all the fields that aren't being initialized.
351    if (RD->isUnion() && ILE->getInitializedFieldInUnion() != *Field)
352      continue;
353
354    // Don't emit anonymous bitfields, they just affect layout.
355    if (Field->isBitField() && !Field->getIdentifier())
356      continue;
357
358    // Get the initializer.  A struct can include fields without initializers,
359    // we just use explicit null values for them.
360    llvm::Constant *EltInit;
361    if (ElementNo < ILE->getNumInits())
362      EltInit = CGM.EmitConstantExpr(ILE->getInit(ElementNo++),
363                                     Field->getType(), CGF);
364    else
365      EltInit = CGM.EmitNullConstant(Field->getType());
366
367    if (!EltInit)
368      return false;
369
370    if (!Field->isBitField()) {
371      // Handle non-bitfield members.
372      if (!AppendField(*Field, Layout.getFieldOffset(FieldNo), EltInit))
373        return false;
374    } else {
375      // Otherwise we have a bitfield.
376      AppendBitField(*Field, Layout.getFieldOffset(FieldNo),
377                     cast<llvm::ConstantInt>(EltInit));
378    }
379  }
380
381  CharUnits LayoutSizeInChars = Layout.getSize();
382
383  if (NextFieldOffsetInChars > LayoutSizeInChars) {
384    // If the struct is bigger than the size of the record type,
385    // we must have a flexible array member at the end.
386    assert(RD->hasFlexibleArrayMember() &&
387           "Must have flexible array member if struct is bigger than type!");
388
389    // No tail padding is necessary.
390    return true;
391  }
392
393  CharUnits LLVMSizeInChars =
394    NextFieldOffsetInChars.RoundUpToAlignment(LLVMStructAlignment);
395
396  // Check if we need to convert the struct to a packed struct.
397  if (NextFieldOffsetInChars <= LayoutSizeInChars &&
398      LLVMSizeInChars > LayoutSizeInChars) {
399    assert(!Packed && "Size mismatch!");
400
401    ConvertStructToPacked();
402    assert(NextFieldOffsetInChars <= LayoutSizeInChars &&
403           "Converting to packed did not help!");
404  }
405
406  // Append tail padding if necessary.
407  AppendTailPadding(LayoutSizeInChars);
408
409  assert(LayoutSizeInChars == NextFieldOffsetInChars &&
410         "Tail padding mismatch!");
411
412  return true;
413}
414
415llvm::Constant *ConstStructBuilder::
416  BuildStruct(CodeGenModule &CGM, CodeGenFunction *CGF, InitListExpr *ILE) {
417  ConstStructBuilder Builder(CGM, CGF);
418
419  if (!Builder.Build(ILE))
420    return 0;
421
422  llvm::Constant *Result =
423  llvm::ConstantStruct::get(CGM.getLLVMContext(),
424                            Builder.Elements, Builder.Packed);
425
426  assert(Builder.NextFieldOffsetInChars.RoundUpToAlignment(
427           Builder.getAlignment(Result)) ==
428         Builder.getSizeInChars(Result) && "Size mismatch!");
429
430  return Result;
431}
432
433
434//===----------------------------------------------------------------------===//
435//                             ConstExprEmitter
436//===----------------------------------------------------------------------===//
437
438class ConstExprEmitter :
439  public StmtVisitor<ConstExprEmitter, llvm::Constant*> {
440  CodeGenModule &CGM;
441  CodeGenFunction *CGF;
442  llvm::LLVMContext &VMContext;
443public:
444  ConstExprEmitter(CodeGenModule &cgm, CodeGenFunction *cgf)
445    : CGM(cgm), CGF(cgf), VMContext(cgm.getLLVMContext()) {
446  }
447
448  //===--------------------------------------------------------------------===//
449  //                            Visitor Methods
450  //===--------------------------------------------------------------------===//
451
452  llvm::Constant *VisitStmt(Stmt *S) {
453    return 0;
454  }
455
456  llvm::Constant *VisitParenExpr(ParenExpr *PE) {
457    return Visit(PE->getSubExpr());
458  }
459
460  llvm::Constant *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
461    return Visit(E->getInitializer());
462  }
463
464  llvm::Constant *VisitUnaryAddrOf(UnaryOperator *E) {
465    if (E->getType()->isMemberPointerType())
466      return CGM.getMemberPointerConstant(E);
467
468    return 0;
469  }
470
471  llvm::Constant *VisitBinSub(BinaryOperator *E) {
472    // This must be a pointer/pointer subtraction.  This only happens for
473    // address of label.
474    if (!isa<AddrLabelExpr>(E->getLHS()->IgnoreParenNoopCasts(CGM.getContext())) ||
475       !isa<AddrLabelExpr>(E->getRHS()->IgnoreParenNoopCasts(CGM.getContext())))
476      return 0;
477
478    llvm::Constant *LHS = CGM.EmitConstantExpr(E->getLHS(),
479                                               E->getLHS()->getType(), CGF);
480    llvm::Constant *RHS = CGM.EmitConstantExpr(E->getRHS(),
481                                               E->getRHS()->getType(), CGF);
482
483    const llvm::Type *ResultType = ConvertType(E->getType());
484    LHS = llvm::ConstantExpr::getPtrToInt(LHS, ResultType);
485    RHS = llvm::ConstantExpr::getPtrToInt(RHS, ResultType);
486
487    // No need to divide by element size, since addr of label is always void*,
488    // which has size 1 in GNUish.
489    return llvm::ConstantExpr::getSub(LHS, RHS);
490  }
491
492  llvm::Constant *VisitCastExpr(CastExpr* E) {
493    Expr *subExpr = E->getSubExpr();
494    llvm::Constant *C = CGM.EmitConstantExpr(subExpr, subExpr->getType(), CGF);
495    if (!C) return 0;
496
497    const llvm::Type *destType = ConvertType(E->getType());
498
499    switch (E->getCastKind()) {
500    case CK_ToUnion: {
501      // GCC cast to union extension
502      assert(E->getType()->isUnionType() &&
503             "Destination type is not union type!");
504
505      // Build a struct with the union sub-element as the first member,
506      // and padded to the appropriate size
507      std::vector<llvm::Constant*> Elts;
508      std::vector<const llvm::Type*> Types;
509      Elts.push_back(C);
510      Types.push_back(C->getType());
511      unsigned CurSize = CGM.getTargetData().getTypeAllocSize(C->getType());
512      unsigned TotalSize = CGM.getTargetData().getTypeAllocSize(destType);
513
514      assert(CurSize <= TotalSize && "Union size mismatch!");
515      if (unsigned NumPadBytes = TotalSize - CurSize) {
516        const llvm::Type *Ty = llvm::Type::getInt8Ty(VMContext);
517        if (NumPadBytes > 1)
518          Ty = llvm::ArrayType::get(Ty, NumPadBytes);
519
520        Elts.push_back(llvm::UndefValue::get(Ty));
521        Types.push_back(Ty);
522      }
523
524      llvm::StructType* STy =
525        llvm::StructType::get(C->getType()->getContext(), Types, false);
526      return llvm::ConstantStruct::get(STy, Elts);
527    }
528    case CK_NullToMemberPointer: {
529      const MemberPointerType *MPT = E->getType()->getAs<MemberPointerType>();
530      return CGM.getCXXABI().EmitNullMemberPointer(MPT);
531    }
532
533    case CK_DerivedToBaseMemberPointer:
534    case CK_BaseToDerivedMemberPointer:
535      return CGM.getCXXABI().EmitMemberPointerConversion(C, E);
536
537    case CK_LValueToRValue:
538    case CK_NoOp:
539      return C;
540
541    case CK_AnyPointerToObjCPointerCast:
542    case CK_AnyPointerToBlockPointerCast:
543    case CK_LValueBitCast:
544    case CK_BitCast:
545      if (C->getType() == destType) return C;
546      return llvm::ConstantExpr::getBitCast(C, destType);
547
548    case CK_Dependent: llvm_unreachable("saw dependent cast!");
549
550    // These will never be supported.
551    case CK_ObjCObjectLValueCast:
552    case CK_GetObjCProperty:
553    case CK_ToVoid:
554    case CK_Dynamic:
555    case CK_ResolveUnknownAnyType:
556      return 0;
557
558    // These might need to be supported for constexpr.
559    case CK_UserDefinedConversion:
560    case CK_ConstructorConversion:
561      return 0;
562
563    // These should eventually be supported.
564    case CK_ArrayToPointerDecay:
565    case CK_FunctionToPointerDecay:
566    case CK_BaseToDerived:
567    case CK_DerivedToBase:
568    case CK_UncheckedDerivedToBase:
569    case CK_MemberPointerToBoolean:
570    case CK_VectorSplat:
571    case CK_FloatingRealToComplex:
572    case CK_FloatingComplexToReal:
573    case CK_FloatingComplexToBoolean:
574    case CK_FloatingComplexCast:
575    case CK_FloatingComplexToIntegralComplex:
576    case CK_IntegralRealToComplex:
577    case CK_IntegralComplexToReal:
578    case CK_IntegralComplexToBoolean:
579    case CK_IntegralComplexCast:
580    case CK_IntegralComplexToFloatingComplex:
581      return 0;
582
583    case CK_PointerToIntegral:
584      if (!E->getType()->isBooleanType())
585        return llvm::ConstantExpr::getPtrToInt(C, destType);
586      // fallthrough
587
588    case CK_PointerToBoolean:
589      return llvm::ConstantExpr::getICmp(llvm::CmpInst::ICMP_EQ, C,
590        llvm::ConstantPointerNull::get(cast<llvm::PointerType>(C->getType())));
591
592    case CK_NullToPointer:
593      return llvm::ConstantPointerNull::get(cast<llvm::PointerType>(destType));
594
595    case CK_IntegralCast: {
596      bool isSigned = subExpr->getType()->isSignedIntegerType();
597      return llvm::ConstantExpr::getIntegerCast(C, destType, isSigned);
598    }
599
600    case CK_IntegralToPointer: {
601      bool isSigned = subExpr->getType()->isSignedIntegerType();
602      C = llvm::ConstantExpr::getIntegerCast(C, CGM.IntPtrTy, isSigned);
603      return llvm::ConstantExpr::getIntToPtr(C, destType);
604    }
605
606    case CK_IntegralToBoolean:
607      return llvm::ConstantExpr::getICmp(llvm::CmpInst::ICMP_EQ, C,
608                             llvm::Constant::getNullValue(C->getType()));
609
610    case CK_IntegralToFloating:
611      if (subExpr->getType()->isSignedIntegerType())
612        return llvm::ConstantExpr::getSIToFP(C, destType);
613      else
614        return llvm::ConstantExpr::getUIToFP(C, destType);
615
616    case CK_FloatingToIntegral:
617      if (E->getType()->isSignedIntegerType())
618        return llvm::ConstantExpr::getFPToSI(C, destType);
619      else
620        return llvm::ConstantExpr::getFPToUI(C, destType);
621
622    case CK_FloatingToBoolean:
623      return llvm::ConstantExpr::getFCmp(llvm::CmpInst::FCMP_UNE, C,
624                             llvm::Constant::getNullValue(C->getType()));
625
626    case CK_FloatingCast:
627      return llvm::ConstantExpr::getFPCast(C, destType);
628    }
629    llvm_unreachable("Invalid CastKind");
630  }
631
632  llvm::Constant *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
633    return Visit(DAE->getExpr());
634  }
635
636  llvm::Constant *EmitArrayInitialization(InitListExpr *ILE) {
637    unsigned NumInitElements = ILE->getNumInits();
638    if (NumInitElements == 1 && ILE->getType() == ILE->getInit(0)->getType() &&
639        (isa<StringLiteral>(ILE->getInit(0)) ||
640         isa<ObjCEncodeExpr>(ILE->getInit(0))))
641      return Visit(ILE->getInit(0));
642
643    std::vector<llvm::Constant*> Elts;
644    const llvm::ArrayType *AType =
645        cast<llvm::ArrayType>(ConvertType(ILE->getType()));
646    const llvm::Type *ElemTy = AType->getElementType();
647    unsigned NumElements = AType->getNumElements();
648
649    // Initialising an array requires us to automatically
650    // initialise any elements that have not been initialised explicitly
651    unsigned NumInitableElts = std::min(NumInitElements, NumElements);
652
653    // Copy initializer elements.
654    unsigned i = 0;
655    bool RewriteType = false;
656    for (; i < NumInitableElts; ++i) {
657      Expr *Init = ILE->getInit(i);
658      llvm::Constant *C = CGM.EmitConstantExpr(Init, Init->getType(), CGF);
659      if (!C)
660        return 0;
661      RewriteType |= (C->getType() != ElemTy);
662      Elts.push_back(C);
663    }
664
665    // Initialize remaining array elements.
666    // FIXME: This doesn't handle member pointers correctly!
667    for (; i < NumElements; ++i)
668      Elts.push_back(llvm::Constant::getNullValue(ElemTy));
669
670    if (RewriteType) {
671      // FIXME: Try to avoid packing the array
672      std::vector<const llvm::Type*> Types;
673      for (unsigned i = 0; i < Elts.size(); ++i)
674        Types.push_back(Elts[i]->getType());
675      const llvm::StructType *SType = llvm::StructType::get(AType->getContext(),
676                                                            Types, true);
677      return llvm::ConstantStruct::get(SType, Elts);
678    }
679
680    return llvm::ConstantArray::get(AType, Elts);
681  }
682
683  llvm::Constant *EmitStructInitialization(InitListExpr *ILE) {
684    return ConstStructBuilder::BuildStruct(CGM, CGF, ILE);
685  }
686
687  llvm::Constant *EmitUnionInitialization(InitListExpr *ILE) {
688    return ConstStructBuilder::BuildStruct(CGM, CGF, ILE);
689  }
690
691  llvm::Constant *VisitImplicitValueInitExpr(ImplicitValueInitExpr* E) {
692    return CGM.EmitNullConstant(E->getType());
693  }
694
695  llvm::Constant *VisitInitListExpr(InitListExpr *ILE) {
696    if (ILE->getType()->isScalarType()) {
697      // We have a scalar in braces. Just use the first element.
698      if (ILE->getNumInits() > 0) {
699        Expr *Init = ILE->getInit(0);
700        return CGM.EmitConstantExpr(Init, Init->getType(), CGF);
701      }
702      return CGM.EmitNullConstant(ILE->getType());
703    }
704
705    if (ILE->getType()->isArrayType())
706      return EmitArrayInitialization(ILE);
707
708    if (ILE->getType()->isRecordType())
709      return EmitStructInitialization(ILE);
710
711    if (ILE->getType()->isUnionType())
712      return EmitUnionInitialization(ILE);
713
714    // If ILE was a constant vector, we would have handled it already.
715    if (ILE->getType()->isVectorType())
716      return 0;
717
718    assert(0 && "Unable to handle InitListExpr");
719    // Get rid of control reaches end of void function warning.
720    // Not reached.
721    return 0;
722  }
723
724  llvm::Constant *VisitCXXConstructExpr(CXXConstructExpr *E) {
725    if (!E->getConstructor()->isTrivial())
726      return 0;
727
728    QualType Ty = E->getType();
729
730    // FIXME: We should not have to call getBaseElementType here.
731    const RecordType *RT =
732      CGM.getContext().getBaseElementType(Ty)->getAs<RecordType>();
733    const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
734
735    // If the class doesn't have a trivial destructor, we can't emit it as a
736    // constant expr.
737    if (!RD->hasTrivialDestructor())
738      return 0;
739
740    // Only copy and default constructors can be trivial.
741
742
743    if (E->getNumArgs()) {
744      assert(E->getNumArgs() == 1 && "trivial ctor with > 1 argument");
745      assert(E->getConstructor()->isCopyConstructor() &&
746             "trivial ctor has argument but isn't a copy ctor");
747
748      Expr *Arg = E->getArg(0);
749      assert(CGM.getContext().hasSameUnqualifiedType(Ty, Arg->getType()) &&
750             "argument to copy ctor is of wrong type");
751
752      return Visit(Arg);
753    }
754
755    return CGM.EmitNullConstant(Ty);
756  }
757
758  llvm::Constant *VisitStringLiteral(StringLiteral *E) {
759    assert(!E->getType()->isPointerType() && "Strings are always arrays");
760
761    // This must be a string initializing an array in a static initializer.
762    // Don't emit it as the address of the string, emit the string data itself
763    // as an inline array.
764    return llvm::ConstantArray::get(VMContext,
765                                    CGM.GetStringForStringLiteral(E), false);
766  }
767
768  llvm::Constant *VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
769    // This must be an @encode initializing an array in a static initializer.
770    // Don't emit it as the address of the string, emit the string data itself
771    // as an inline array.
772    std::string Str;
773    CGM.getContext().getObjCEncodingForType(E->getEncodedType(), Str);
774    const ConstantArrayType *CAT = cast<ConstantArrayType>(E->getType());
775
776    // Resize the string to the right size, adding zeros at the end, or
777    // truncating as needed.
778    Str.resize(CAT->getSize().getZExtValue(), '\0');
779    return llvm::ConstantArray::get(VMContext, Str, false);
780  }
781
782  llvm::Constant *VisitUnaryExtension(const UnaryOperator *E) {
783    return Visit(E->getSubExpr());
784  }
785
786  // Utility methods
787  const llvm::Type *ConvertType(QualType T) {
788    return CGM.getTypes().ConvertType(T);
789  }
790
791public:
792  llvm::Constant *EmitLValue(Expr *E) {
793    switch (E->getStmtClass()) {
794    default: break;
795    case Expr::CompoundLiteralExprClass: {
796      // Note that due to the nature of compound literals, this is guaranteed
797      // to be the only use of the variable, so we just generate it here.
798      CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
799      llvm::Constant* C = Visit(CLE->getInitializer());
800      // FIXME: "Leaked" on failure.
801      if (C)
802        C = new llvm::GlobalVariable(CGM.getModule(), C->getType(),
803                                     E->getType().isConstant(CGM.getContext()),
804                                     llvm::GlobalValue::InternalLinkage,
805                                     C, ".compoundliteral", 0, false,
806                          CGM.getContext().getTargetAddressSpace(E->getType()));
807      return C;
808    }
809    case Expr::DeclRefExprClass: {
810      ValueDecl *Decl = cast<DeclRefExpr>(E)->getDecl();
811      if (Decl->hasAttr<WeakRefAttr>())
812        return CGM.GetWeakRefReference(Decl);
813      if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Decl))
814        return CGM.GetAddrOfFunction(FD);
815      if (const VarDecl* VD = dyn_cast<VarDecl>(Decl)) {
816        // We can never refer to a variable with local storage.
817        if (!VD->hasLocalStorage()) {
818          if (VD->isFileVarDecl() || VD->hasExternalStorage())
819            return CGM.GetAddrOfGlobalVar(VD);
820          else if (VD->isLocalVarDecl()) {
821            assert(CGF && "Can't access static local vars without CGF");
822            return CGF->GetAddrOfStaticLocalVar(VD);
823          }
824        }
825      }
826      break;
827    }
828    case Expr::StringLiteralClass:
829      return CGM.GetAddrOfConstantStringFromLiteral(cast<StringLiteral>(E));
830    case Expr::ObjCEncodeExprClass:
831      return CGM.GetAddrOfConstantStringFromObjCEncode(cast<ObjCEncodeExpr>(E));
832    case Expr::ObjCStringLiteralClass: {
833      ObjCStringLiteral* SL = cast<ObjCStringLiteral>(E);
834      llvm::Constant *C =
835          CGM.getObjCRuntime().GenerateConstantString(SL->getString());
836      return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
837    }
838    case Expr::PredefinedExprClass: {
839      unsigned Type = cast<PredefinedExpr>(E)->getIdentType();
840      if (CGF) {
841        LValue Res = CGF->EmitPredefinedLValue(cast<PredefinedExpr>(E));
842        return cast<llvm::Constant>(Res.getAddress());
843      } else if (Type == PredefinedExpr::PrettyFunction) {
844        return CGM.GetAddrOfConstantCString("top level", ".tmp");
845      }
846
847      return CGM.GetAddrOfConstantCString("", ".tmp");
848    }
849    case Expr::AddrLabelExprClass: {
850      assert(CGF && "Invalid address of label expression outside function.");
851      llvm::Constant *Ptr =
852        CGF->GetAddrOfLabel(cast<AddrLabelExpr>(E)->getLabel());
853      return llvm::ConstantExpr::getBitCast(Ptr, ConvertType(E->getType()));
854    }
855    case Expr::CallExprClass: {
856      CallExpr* CE = cast<CallExpr>(E);
857      unsigned builtin = CE->isBuiltinCall(CGM.getContext());
858      if (builtin !=
859            Builtin::BI__builtin___CFStringMakeConstantString &&
860          builtin !=
861            Builtin::BI__builtin___NSStringMakeConstantString)
862        break;
863      const Expr *Arg = CE->getArg(0)->IgnoreParenCasts();
864      const StringLiteral *Literal = cast<StringLiteral>(Arg);
865      if (builtin ==
866            Builtin::BI__builtin___NSStringMakeConstantString) {
867        return CGM.getObjCRuntime().GenerateConstantString(Literal);
868      }
869      // FIXME: need to deal with UCN conversion issues.
870      return CGM.GetAddrOfConstantCFString(Literal);
871    }
872    case Expr::BlockExprClass: {
873      std::string FunctionName;
874      if (CGF)
875        FunctionName = CGF->CurFn->getName();
876      else
877        FunctionName = "global";
878
879      return CGM.GetAddrOfGlobalBlock(cast<BlockExpr>(E), FunctionName.c_str());
880    }
881    }
882
883    return 0;
884  }
885};
886
887}  // end anonymous namespace.
888
889llvm::Constant *CodeGenModule::EmitConstantExpr(const Expr *E,
890                                                QualType DestType,
891                                                CodeGenFunction *CGF) {
892  Expr::EvalResult Result;
893
894  bool Success = false;
895
896  if (DestType->isReferenceType())
897    Success = E->EvaluateAsLValue(Result, Context);
898  else
899    Success = E->Evaluate(Result, Context);
900
901  if (Success && !Result.HasSideEffects) {
902    switch (Result.Val.getKind()) {
903    case APValue::Uninitialized:
904      assert(0 && "Constant expressions should be initialized.");
905      return 0;
906    case APValue::LValue: {
907      const llvm::Type *DestTy = getTypes().ConvertTypeForMem(DestType);
908      llvm::Constant *Offset =
909        llvm::ConstantInt::get(llvm::Type::getInt64Ty(VMContext),
910                               Result.Val.getLValueOffset().getQuantity());
911
912      llvm::Constant *C;
913      if (const Expr *LVBase = Result.Val.getLValueBase()) {
914        C = ConstExprEmitter(*this, CGF).EmitLValue(const_cast<Expr*>(LVBase));
915
916        // Apply offset if necessary.
917        if (!Offset->isNullValue()) {
918          const llvm::Type *Type = llvm::Type::getInt8PtrTy(VMContext);
919          llvm::Constant *Casted = llvm::ConstantExpr::getBitCast(C, Type);
920          Casted = llvm::ConstantExpr::getGetElementPtr(Casted, &Offset, 1);
921          C = llvm::ConstantExpr::getBitCast(Casted, C->getType());
922        }
923
924        // Convert to the appropriate type; this could be an lvalue for
925        // an integer.
926        if (isa<llvm::PointerType>(DestTy))
927          return llvm::ConstantExpr::getBitCast(C, DestTy);
928
929        return llvm::ConstantExpr::getPtrToInt(C, DestTy);
930      } else {
931        C = Offset;
932
933        // Convert to the appropriate type; this could be an lvalue for
934        // an integer.
935        if (isa<llvm::PointerType>(DestTy))
936          return llvm::ConstantExpr::getIntToPtr(C, DestTy);
937
938        // If the types don't match this should only be a truncate.
939        if (C->getType() != DestTy)
940          return llvm::ConstantExpr::getTrunc(C, DestTy);
941
942        return C;
943      }
944    }
945    case APValue::Int: {
946      llvm::Constant *C = llvm::ConstantInt::get(VMContext,
947                                                 Result.Val.getInt());
948
949      if (C->getType()->isIntegerTy(1)) {
950        const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
951        C = llvm::ConstantExpr::getZExt(C, BoolTy);
952      }
953      return C;
954    }
955    case APValue::ComplexInt: {
956      llvm::Constant *Complex[2];
957
958      Complex[0] = llvm::ConstantInt::get(VMContext,
959                                          Result.Val.getComplexIntReal());
960      Complex[1] = llvm::ConstantInt::get(VMContext,
961                                          Result.Val.getComplexIntImag());
962
963      // FIXME: the target may want to specify that this is packed.
964      return llvm::ConstantStruct::get(VMContext, Complex, 2, false);
965    }
966    case APValue::Float:
967      return llvm::ConstantFP::get(VMContext, Result.Val.getFloat());
968    case APValue::ComplexFloat: {
969      llvm::Constant *Complex[2];
970
971      Complex[0] = llvm::ConstantFP::get(VMContext,
972                                         Result.Val.getComplexFloatReal());
973      Complex[1] = llvm::ConstantFP::get(VMContext,
974                                         Result.Val.getComplexFloatImag());
975
976      // FIXME: the target may want to specify that this is packed.
977      return llvm::ConstantStruct::get(VMContext, Complex, 2, false);
978    }
979    case APValue::Vector: {
980      llvm::SmallVector<llvm::Constant *, 4> Inits;
981      unsigned NumElts = Result.Val.getVectorLength();
982
983      if (Context.getLangOptions().AltiVec &&
984          isa<CastExpr>(E) &&
985          cast<CastExpr>(E)->getCastKind() == CK_VectorSplat) {
986        // AltiVec vector initialization with a single literal
987        APValue &Elt = Result.Val.getVectorElt(0);
988
989        llvm::Constant* InitValue = Elt.isInt()
990          ? cast<llvm::Constant>
991              (llvm::ConstantInt::get(VMContext, Elt.getInt()))
992          : cast<llvm::Constant>
993              (llvm::ConstantFP::get(VMContext, Elt.getFloat()));
994
995        for (unsigned i = 0; i != NumElts; ++i)
996          Inits.push_back(InitValue);
997
998      } else {
999        for (unsigned i = 0; i != NumElts; ++i) {
1000          APValue &Elt = Result.Val.getVectorElt(i);
1001          if (Elt.isInt())
1002            Inits.push_back(llvm::ConstantInt::get(VMContext, Elt.getInt()));
1003          else
1004            Inits.push_back(llvm::ConstantFP::get(VMContext, Elt.getFloat()));
1005        }
1006      }
1007      return llvm::ConstantVector::get(Inits);
1008    }
1009    }
1010  }
1011
1012  llvm::Constant* C = ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
1013  if (C && C->getType()->isIntegerTy(1)) {
1014    const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
1015    C = llvm::ConstantExpr::getZExt(C, BoolTy);
1016  }
1017  return C;
1018}
1019
1020static uint64_t getFieldOffset(ASTContext &C, const FieldDecl *field) {
1021  const ASTRecordLayout &layout = C.getASTRecordLayout(field->getParent());
1022  return layout.getFieldOffset(field->getFieldIndex());
1023}
1024
1025llvm::Constant *
1026CodeGenModule::getMemberPointerConstant(const UnaryOperator *uo) {
1027  // Member pointer constants always have a very particular form.
1028  const MemberPointerType *type = cast<MemberPointerType>(uo->getType());
1029  const ValueDecl *decl = cast<DeclRefExpr>(uo->getSubExpr())->getDecl();
1030
1031  // A member function pointer.
1032  if (const CXXMethodDecl *method = dyn_cast<CXXMethodDecl>(decl))
1033    return getCXXABI().EmitMemberPointer(method);
1034
1035  // Otherwise, a member data pointer.
1036  uint64_t fieldOffset;
1037  if (const FieldDecl *field = dyn_cast<FieldDecl>(decl))
1038    fieldOffset = getFieldOffset(getContext(), field);
1039  else {
1040    const IndirectFieldDecl *ifield = cast<IndirectFieldDecl>(decl);
1041
1042    fieldOffset = 0;
1043    for (IndirectFieldDecl::chain_iterator ci = ifield->chain_begin(),
1044           ce = ifield->chain_end(); ci != ce; ++ci)
1045      fieldOffset += getFieldOffset(getContext(), cast<FieldDecl>(*ci));
1046  }
1047
1048  CharUnits chars = getContext().toCharUnitsFromBits((int64_t) fieldOffset);
1049  return getCXXABI().EmitMemberDataPointer(type, chars);
1050}
1051
1052static void
1053FillInNullDataMemberPointers(CodeGenModule &CGM, QualType T,
1054                             std::vector<llvm::Constant *> &Elements,
1055                             uint64_t StartOffset) {
1056  assert(StartOffset % CGM.getContext().getCharWidth() == 0 &&
1057         "StartOffset not byte aligned!");
1058
1059  if (CGM.getTypes().isZeroInitializable(T))
1060    return;
1061
1062  if (const ConstantArrayType *CAT =
1063        CGM.getContext().getAsConstantArrayType(T)) {
1064    QualType ElementTy = CAT->getElementType();
1065    uint64_t ElementSize = CGM.getContext().getTypeSize(ElementTy);
1066
1067    for (uint64_t I = 0, E = CAT->getSize().getZExtValue(); I != E; ++I) {
1068      FillInNullDataMemberPointers(CGM, ElementTy, Elements,
1069                                   StartOffset + I * ElementSize);
1070    }
1071  } else if (const RecordType *RT = T->getAs<RecordType>()) {
1072    const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
1073    const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
1074
1075    // Go through all bases and fill in any null pointer to data members.
1076    for (CXXRecordDecl::base_class_const_iterator I = RD->bases_begin(),
1077         E = RD->bases_end(); I != E; ++I) {
1078      if (I->isVirtual()) {
1079        // Ignore virtual bases.
1080        continue;
1081      }
1082
1083      const CXXRecordDecl *BaseDecl =
1084      cast<CXXRecordDecl>(I->getType()->getAs<RecordType>()->getDecl());
1085
1086      // Ignore empty bases.
1087      if (BaseDecl->isEmpty())
1088        continue;
1089
1090      // Ignore bases that don't have any pointer to data members.
1091      if (CGM.getTypes().isZeroInitializable(BaseDecl))
1092        continue;
1093
1094      uint64_t BaseOffset = Layout.getBaseClassOffsetInBits(BaseDecl);
1095      FillInNullDataMemberPointers(CGM, I->getType(),
1096                                   Elements, StartOffset + BaseOffset);
1097    }
1098
1099    // Visit all fields.
1100    unsigned FieldNo = 0;
1101    for (RecordDecl::field_iterator I = RD->field_begin(),
1102         E = RD->field_end(); I != E; ++I, ++FieldNo) {
1103      QualType FieldType = I->getType();
1104
1105      if (CGM.getTypes().isZeroInitializable(FieldType))
1106        continue;
1107
1108      uint64_t FieldOffset = StartOffset + Layout.getFieldOffset(FieldNo);
1109      FillInNullDataMemberPointers(CGM, FieldType, Elements, FieldOffset);
1110    }
1111  } else {
1112    assert(T->isMemberPointerType() && "Should only see member pointers here!");
1113    assert(!T->getAs<MemberPointerType>()->getPointeeType()->isFunctionType() &&
1114           "Should only see pointers to data members here!");
1115
1116    CharUnits StartIndex = CGM.getContext().toCharUnitsFromBits(StartOffset);
1117    CharUnits EndIndex = StartIndex + CGM.getContext().getTypeSizeInChars(T);
1118
1119    // FIXME: hardcodes Itanium member pointer representation!
1120    llvm::Constant *NegativeOne =
1121      llvm::ConstantInt::get(llvm::Type::getInt8Ty(CGM.getLLVMContext()),
1122                             -1ULL, /*isSigned*/true);
1123
1124    // Fill in the null data member pointer.
1125    for (CharUnits I = StartIndex; I != EndIndex; ++I)
1126      Elements[I.getQuantity()] = NegativeOne;
1127  }
1128}
1129
1130static llvm::Constant *EmitNullConstantForBase(CodeGenModule &CGM,
1131                                               const llvm::Type *baseType,
1132                                               const CXXRecordDecl *base);
1133
1134static llvm::Constant *EmitNullConstant(CodeGenModule &CGM,
1135                                        const CXXRecordDecl *record,
1136                                        bool asCompleteObject) {
1137  const CGRecordLayout &layout = CGM.getTypes().getCGRecordLayout(record);
1138  const llvm::StructType *structure =
1139    (asCompleteObject ? layout.getLLVMType()
1140                      : layout.getBaseSubobjectLLVMType());
1141
1142  unsigned numElements = structure->getNumElements();
1143  std::vector<llvm::Constant *> elements(numElements);
1144
1145  // Fill in all the bases.
1146  for (CXXRecordDecl::base_class_const_iterator
1147         I = record->bases_begin(), E = record->bases_end(); I != E; ++I) {
1148    if (I->isVirtual()) {
1149      // Ignore virtual bases; if we're laying out for a complete
1150      // object, we'll lay these out later.
1151      continue;
1152    }
1153
1154    const CXXRecordDecl *base =
1155      cast<CXXRecordDecl>(I->getType()->castAs<RecordType>()->getDecl());
1156
1157    // Ignore empty bases.
1158    if (base->isEmpty())
1159      continue;
1160
1161    unsigned fieldIndex = layout.getNonVirtualBaseLLVMFieldNo(base);
1162    const llvm::Type *baseType = structure->getElementType(fieldIndex);
1163    elements[fieldIndex] = EmitNullConstantForBase(CGM, baseType, base);
1164  }
1165
1166  // Fill in all the fields.
1167  for (RecordDecl::field_iterator I = record->field_begin(),
1168         E = record->field_end(); I != E; ++I) {
1169    const FieldDecl *field = *I;
1170
1171    // Ignore bit fields.
1172    if (field->isBitField())
1173      continue;
1174
1175    unsigned fieldIndex = layout.getLLVMFieldNo(field);
1176    elements[fieldIndex] = CGM.EmitNullConstant(field->getType());
1177  }
1178
1179  // Fill in the virtual bases, if we're working with the complete object.
1180  if (asCompleteObject) {
1181    for (CXXRecordDecl::base_class_const_iterator
1182           I = record->vbases_begin(), E = record->vbases_end(); I != E; ++I) {
1183      const CXXRecordDecl *base =
1184        cast<CXXRecordDecl>(I->getType()->castAs<RecordType>()->getDecl());
1185
1186      // Ignore empty bases.
1187      if (base->isEmpty())
1188        continue;
1189
1190      unsigned fieldIndex = layout.getVirtualBaseIndex(base);
1191
1192      // We might have already laid this field out.
1193      if (elements[fieldIndex]) continue;
1194
1195      const llvm::Type *baseType = structure->getElementType(fieldIndex);
1196      elements[fieldIndex] = EmitNullConstantForBase(CGM, baseType, base);
1197    }
1198  }
1199
1200  // Now go through all other fields and zero them out.
1201  for (unsigned i = 0; i != numElements; ++i) {
1202    if (!elements[i])
1203      elements[i] = llvm::Constant::getNullValue(structure->getElementType(i));
1204  }
1205
1206  return llvm::ConstantStruct::get(structure, elements);
1207}
1208
1209/// Emit the null constant for a base subobject.
1210static llvm::Constant *EmitNullConstantForBase(CodeGenModule &CGM,
1211                                               const llvm::Type *baseType,
1212                                               const CXXRecordDecl *base) {
1213  const CGRecordLayout &baseLayout = CGM.getTypes().getCGRecordLayout(base);
1214
1215  // Just zero out bases that don't have any pointer to data members.
1216  if (baseLayout.isZeroInitializableAsBase())
1217    return llvm::Constant::getNullValue(baseType);
1218
1219  // If the base type is a struct, we can just use its null constant.
1220  if (isa<llvm::StructType>(baseType)) {
1221    return EmitNullConstant(CGM, base, /*complete*/ false);
1222  }
1223
1224  // Otherwise, some bases are represented as arrays of i8 if the size
1225  // of the base is smaller than its corresponding LLVM type.  Figure
1226  // out how many elements this base array has.
1227  const llvm::ArrayType *baseArrayType = cast<llvm::ArrayType>(baseType);
1228  unsigned numBaseElements = baseArrayType->getNumElements();
1229
1230  // Fill in null data member pointers.
1231  std::vector<llvm::Constant *> baseElements(numBaseElements);
1232  FillInNullDataMemberPointers(CGM, CGM.getContext().getTypeDeclType(base),
1233                               baseElements, 0);
1234
1235  // Now go through all other elements and zero them out.
1236  if (numBaseElements) {
1237    const llvm::Type *i8 = llvm::Type::getInt8Ty(CGM.getLLVMContext());
1238    llvm::Constant *i8_zero = llvm::Constant::getNullValue(i8);
1239    for (unsigned i = 0; i != numBaseElements; ++i) {
1240      if (!baseElements[i])
1241        baseElements[i] = i8_zero;
1242    }
1243  }
1244
1245  return llvm::ConstantArray::get(baseArrayType, baseElements);
1246}
1247
1248llvm::Constant *CodeGenModule::EmitNullConstant(QualType T) {
1249  if (getTypes().isZeroInitializable(T))
1250    return llvm::Constant::getNullValue(getTypes().ConvertTypeForMem(T));
1251
1252  if (const ConstantArrayType *CAT = Context.getAsConstantArrayType(T)) {
1253
1254    QualType ElementTy = CAT->getElementType();
1255
1256    llvm::Constant *Element = EmitNullConstant(ElementTy);
1257    unsigned NumElements = CAT->getSize().getZExtValue();
1258    std::vector<llvm::Constant *> Array(NumElements);
1259    for (unsigned i = 0; i != NumElements; ++i)
1260      Array[i] = Element;
1261
1262    const llvm::ArrayType *ATy =
1263      cast<llvm::ArrayType>(getTypes().ConvertTypeForMem(T));
1264    return llvm::ConstantArray::get(ATy, Array);
1265  }
1266
1267  if (const RecordType *RT = T->getAs<RecordType>()) {
1268    const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
1269    return ::EmitNullConstant(*this, RD, /*complete object*/ true);
1270  }
1271
1272  assert(T->isMemberPointerType() && "Should only see member pointers here!");
1273  assert(!T->getAs<MemberPointerType>()->getPointeeType()->isFunctionType() &&
1274         "Should only see pointers to data members here!");
1275
1276  // Itanium C++ ABI 2.3:
1277  //   A NULL pointer is represented as -1.
1278  return getCXXABI().EmitNullMemberPointer(T->castAs<MemberPointerType>());
1279}
1280