blob: 59317f2a29dbe8ebd4d0dcbbedeea1ba0dc9935f [file] [log] [blame]
Anders Carlsson610ee712008-01-26 01:36:00 +00001//===--- 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"
John McCall5d865c322010-08-31 07:33:07 +000016#include "CGCXXABI.h"
Daniel Dunbar89da6ad2008-08-13 00:59:25 +000017#include "CGObjCRuntime.h"
Daniel Dunbar072d0bb2010-03-30 22:26:10 +000018#include "CGRecordLayout.h"
Chris Lattnere50e9012008-10-06 05:59:01 +000019#include "clang/AST/APValue.h"
Daniel Dunbarad319a72008-08-11 05:00:27 +000020#include "clang/AST/ASTContext.h"
Anders Carlssone1d5ca52009-07-24 15:20:52 +000021#include "clang/AST/RecordLayout.h"
Daniel Dunbarad319a72008-08-11 05:00:27 +000022#include "clang/AST/StmtVisitor.h"
Chris Lattner15ba9492009-06-14 01:54:56 +000023#include "clang/Basic/Builtins.h"
Anders Carlsson610ee712008-01-26 01:36:00 +000024#include "llvm/Constants.h"
25#include "llvm/Function.h"
26#include "llvm/GlobalVariable.h"
Eli Friedmana2eaffc2008-05-30 10:24:46 +000027#include "llvm/Target/TargetData.h"
Anders Carlsson610ee712008-01-26 01:36:00 +000028using namespace clang;
29using namespace CodeGen;
30
Chris Lattnercfa3e7a2010-04-13 17:45:57 +000031//===----------------------------------------------------------------------===//
32// ConstStructBuilder
33//===----------------------------------------------------------------------===//
34
35namespace {
Benjamin Kramer337e3a52009-11-28 19:45:26 +000036class ConstStructBuilder {
Anders Carlssone1d5ca52009-07-24 15:20:52 +000037 CodeGenModule &CGM;
38 CodeGenFunction *CGF;
39
Mike Stump11289f42009-09-09 15:08:12 +000040 bool Packed;
Anders Carlssone1d5ca52009-07-24 15:20:52 +000041 unsigned NextFieldOffsetInBytes;
Anders Carlsson39e3eb12009-10-02 02:15:20 +000042 unsigned LLVMStructAlignment;
Anders Carlssone1d5ca52009-07-24 15:20:52 +000043 std::vector<llvm::Constant *> Elements;
Chris Lattnercfa3e7a2010-04-13 17:45:57 +000044public:
45 static llvm::Constant *BuildStruct(CodeGenModule &CGM, CodeGenFunction *CGF,
46 InitListExpr *ILE);
47
48private:
Anders Carlssone1d5ca52009-07-24 15:20:52 +000049 ConstStructBuilder(CodeGenModule &CGM, CodeGenFunction *CGF)
Anders Carlsson39e3eb12009-10-02 02:15:20 +000050 : CGM(CGM), CGF(CGF), Packed(false), NextFieldOffsetInBytes(0),
51 LLVMStructAlignment(1) { }
Anders Carlssone1d5ca52009-07-24 15:20:52 +000052
Mike Stump11289f42009-09-09 15:08:12 +000053 bool AppendField(const FieldDecl *Field, uint64_t FieldOffset,
Chris Lattnerff0e2a32010-04-13 18:16:19 +000054 llvm::Constant *InitExpr);
Mike Stump11289f42009-09-09 15:08:12 +000055
Chris Lattner9a3459f2010-07-05 17:04:23 +000056 void AppendBitField(const FieldDecl *Field, uint64_t FieldOffset,
57 llvm::ConstantInt *InitExpr);
Mike Stump11289f42009-09-09 15:08:12 +000058
Ken Dyck30a87e32011-03-11 23:42:54 +000059 void AppendPadding(CharUnits PadSize);
Mike Stump11289f42009-09-09 15:08:12 +000060
Ken Dyck327b77a2011-03-11 02:17:05 +000061 void AppendTailPadding(CharUnits RecordSize);
Anders Carlssonba4c6d12009-07-27 01:23:51 +000062
Chris Lattnercfa3e7a2010-04-13 17:45:57 +000063 void ConvertStructToPacked();
Anders Carlsson39e3eb12009-10-02 02:15:20 +000064
Chris Lattnercfa3e7a2010-04-13 17:45:57 +000065 bool Build(InitListExpr *ILE);
Mike Stump11289f42009-09-09 15:08:12 +000066
Anders Carlssone1d5ca52009-07-24 15:20:52 +000067 unsigned getAlignment(const llvm::Constant *C) const {
Chris Lattnercfa3e7a2010-04-13 17:45:57 +000068 if (Packed) return 1;
Anders Carlssone1d5ca52009-07-24 15:20:52 +000069 return CGM.getTargetData().getABITypeAlignment(C->getType());
70 }
Mike Stump11289f42009-09-09 15:08:12 +000071
Anders Carlssone1d5ca52009-07-24 15:20:52 +000072 uint64_t getSizeInBytes(const llvm::Constant *C) const {
73 return CGM.getTargetData().getTypeAllocSize(C->getType());
74 }
Anders Carlssone1d5ca52009-07-24 15:20:52 +000075};
Mike Stump11289f42009-09-09 15:08:12 +000076
Chris Lattnercfa3e7a2010-04-13 17:45:57 +000077bool ConstStructBuilder::
Chris Lattnerff0e2a32010-04-13 18:16:19 +000078AppendField(const FieldDecl *Field, uint64_t FieldOffset,
79 llvm::Constant *InitCst) {
Ken Dyck1c80fd12011-03-15 01:09:02 +000080
81 const ASTContext &Context = CGM.getContext();
82
83 CharUnits FieldOffsetInChars = Context.toCharUnitsFromBits(FieldOffset);
84 uint64_t FieldOffsetInBytes = FieldOffsetInChars.getQuantity();
Chris Lattnercfa3e7a2010-04-13 17:45:57 +000085
86 assert(NextFieldOffsetInBytes <= FieldOffsetInBytes
87 && "Field offset mismatch!");
88
Chris Lattnerff0e2a32010-04-13 18:16:19 +000089 unsigned FieldAlignment = getAlignment(InitCst);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +000090
91 // Round up the field offset to the alignment of the field type.
92 uint64_t AlignedNextFieldOffsetInBytes =
93 llvm::RoundUpToAlignment(NextFieldOffsetInBytes, FieldAlignment);
94
95 if (AlignedNextFieldOffsetInBytes > FieldOffsetInBytes) {
96 assert(!Packed && "Alignment is wrong even with a packed struct!");
97
98 // Convert the struct to a packed struct.
99 ConvertStructToPacked();
100
101 AlignedNextFieldOffsetInBytes = NextFieldOffsetInBytes;
102 }
103
104 if (AlignedNextFieldOffsetInBytes < FieldOffsetInBytes) {
105 // We need to append padding.
Ken Dyck30a87e32011-03-11 23:42:54 +0000106 AppendPadding(
Ken Dyck1c80fd12011-03-15 01:09:02 +0000107 FieldOffsetInChars - CharUnits::fromQuantity(NextFieldOffsetInBytes));
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000108
Ken Dyck1c80fd12011-03-15 01:09:02 +0000109 assert(NextFieldOffsetInBytes == FieldOffsetInChars.getQuantity() &&
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000110 "Did not add enough padding!");
111
112 AlignedNextFieldOffsetInBytes = NextFieldOffsetInBytes;
113 }
114
115 // Add the field.
Chris Lattnerff0e2a32010-04-13 18:16:19 +0000116 Elements.push_back(InitCst);
117 NextFieldOffsetInBytes = AlignedNextFieldOffsetInBytes +
118 getSizeInBytes(InitCst);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000119
120 if (Packed)
121 assert(LLVMStructAlignment == 1 && "Packed struct not byte-aligned!");
122 else
123 LLVMStructAlignment = std::max(LLVMStructAlignment, FieldAlignment);
124
125 return true;
126}
127
Chris Lattner9a3459f2010-07-05 17:04:23 +0000128void ConstStructBuilder::AppendBitField(const FieldDecl *Field,
129 uint64_t FieldOffset,
130 llvm::ConstantInt *CI) {
Ken Dycka862d952011-03-12 12:03:11 +0000131 const ASTContext &Context = CGM.getContext();
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000132 if (FieldOffset > NextFieldOffsetInBytes * 8) {
133 // We need to add padding.
Ken Dycka862d952011-03-12 12:03:11 +0000134 CharUnits PadSize = Context.toCharUnitsFromBits(
135 llvm::RoundUpToAlignment(FieldOffset - NextFieldOffsetInBytes * 8,
136 Context.Target.getCharAlign()));
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000137
Ken Dycka862d952011-03-12 12:03:11 +0000138 AppendPadding(PadSize);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000139 }
140
141 uint64_t FieldSize =
Ken Dycka862d952011-03-12 12:03:11 +0000142 Field->getBitWidth()->EvaluateAsInt(Context).getZExtValue();
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000143
144 llvm::APInt FieldValue = CI->getValue();
145
146 // Promote the size of FieldValue if necessary
147 // FIXME: This should never occur, but currently it can because initializer
148 // constants are cast to bool, and because clang is not enforcing bitfield
149 // width limits.
150 if (FieldSize > FieldValue.getBitWidth())
Jay Foad6d4db0c2010-12-07 08:25:34 +0000151 FieldValue = FieldValue.zext(FieldSize);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000152
153 // Truncate the size of FieldValue to the bit field size.
154 if (FieldSize < FieldValue.getBitWidth())
Jay Foad6d4db0c2010-12-07 08:25:34 +0000155 FieldValue = FieldValue.trunc(FieldSize);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000156
157 if (FieldOffset < NextFieldOffsetInBytes * 8) {
158 // Either part of the field or the entire field can go into the previous
159 // byte.
160 assert(!Elements.empty() && "Elements can't be empty!");
161
162 unsigned BitsInPreviousByte =
163 NextFieldOffsetInBytes * 8 - FieldOffset;
164
165 bool FitsCompletelyInPreviousByte =
166 BitsInPreviousByte >= FieldValue.getBitWidth();
167
168 llvm::APInt Tmp = FieldValue;
169
170 if (!FitsCompletelyInPreviousByte) {
171 unsigned NewFieldWidth = FieldSize - BitsInPreviousByte;
172
173 if (CGM.getTargetData().isBigEndian()) {
174 Tmp = Tmp.lshr(NewFieldWidth);
Jay Foad6d4db0c2010-12-07 08:25:34 +0000175 Tmp = Tmp.trunc(BitsInPreviousByte);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000176
177 // We want the remaining high bits.
Jay Foad6d4db0c2010-12-07 08:25:34 +0000178 FieldValue = FieldValue.trunc(NewFieldWidth);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000179 } else {
Jay Foad6d4db0c2010-12-07 08:25:34 +0000180 Tmp = Tmp.trunc(BitsInPreviousByte);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000181
182 // We want the remaining low bits.
183 FieldValue = FieldValue.lshr(BitsInPreviousByte);
Jay Foad6d4db0c2010-12-07 08:25:34 +0000184 FieldValue = FieldValue.trunc(NewFieldWidth);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000185 }
186 }
187
Jay Foad6d4db0c2010-12-07 08:25:34 +0000188 Tmp = Tmp.zext(8);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000189 if (CGM.getTargetData().isBigEndian()) {
190 if (FitsCompletelyInPreviousByte)
191 Tmp = Tmp.shl(BitsInPreviousByte - FieldValue.getBitWidth());
192 } else {
193 Tmp = Tmp.shl(8 - BitsInPreviousByte);
194 }
195
Chris Lattner53b479f2010-07-05 18:03:30 +0000196 // 'or' in the bits that go into the previous byte.
197 llvm::Value *LastElt = Elements.back();
198 if (llvm::ConstantInt *Val = dyn_cast<llvm::ConstantInt>(LastElt))
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000199 Tmp |= Val->getValue();
Chris Lattner53b479f2010-07-05 18:03:30 +0000200 else {
201 assert(isa<llvm::UndefValue>(LastElt));
202 // If there is an undef field that we're adding to, it can either be a
203 // scalar undef (in which case, we just replace it with our field) or it
204 // is an array. If it is an array, we have to pull one byte off the
205 // array so that the other undef bytes stay around.
206 if (!isa<llvm::IntegerType>(LastElt->getType())) {
207 // The undef padding will be a multibyte array, create a new smaller
208 // padding and then an hole for our i8 to get plopped into.
209 assert(isa<llvm::ArrayType>(LastElt->getType()) &&
210 "Expected array padding of undefs");
211 const llvm::ArrayType *AT = cast<llvm::ArrayType>(LastElt->getType());
212 assert(AT->getElementType()->isIntegerTy(8) &&
213 AT->getNumElements() != 0 &&
214 "Expected non-empty array padding of undefs");
215
216 // Remove the padding array.
217 NextFieldOffsetInBytes -= AT->getNumElements();
218 Elements.pop_back();
219
220 // Add the padding back in two chunks.
Ken Dyck30a87e32011-03-11 23:42:54 +0000221 AppendPadding(CharUnits::fromQuantity(AT->getNumElements()-1));
222 AppendPadding(CharUnits::One());
Chris Lattner53b479f2010-07-05 18:03:30 +0000223 assert(isa<llvm::UndefValue>(Elements.back()) &&
224 Elements.back()->getType()->isIntegerTy(8) &&
225 "Padding addition didn't work right");
226 }
227 }
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000228
229 Elements.back() = llvm::ConstantInt::get(CGM.getLLVMContext(), Tmp);
230
231 if (FitsCompletelyInPreviousByte)
Chris Lattner9a3459f2010-07-05 17:04:23 +0000232 return;
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000233 }
234
235 while (FieldValue.getBitWidth() > 8) {
236 llvm::APInt Tmp;
237
238 if (CGM.getTargetData().isBigEndian()) {
239 // We want the high bits.
Chris Lattnerfb59c7c2011-02-17 22:09:58 +0000240 Tmp = FieldValue.lshr(FieldValue.getBitWidth() - 8).trunc(8);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000241 } else {
242 // We want the low bits.
Jay Foad6d4db0c2010-12-07 08:25:34 +0000243 Tmp = FieldValue.trunc(8);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000244
245 FieldValue = FieldValue.lshr(8);
246 }
247
248 Elements.push_back(llvm::ConstantInt::get(CGM.getLLVMContext(), Tmp));
249 NextFieldOffsetInBytes++;
250
Jay Foad6d4db0c2010-12-07 08:25:34 +0000251 FieldValue = FieldValue.trunc(FieldValue.getBitWidth() - 8);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000252 }
253
254 assert(FieldValue.getBitWidth() > 0 &&
255 "Should have at least one bit left!");
256 assert(FieldValue.getBitWidth() <= 8 &&
257 "Should not have more than a byte left!");
258
259 if (FieldValue.getBitWidth() < 8) {
260 if (CGM.getTargetData().isBigEndian()) {
261 unsigned BitWidth = FieldValue.getBitWidth();
262
Jay Foad6d4db0c2010-12-07 08:25:34 +0000263 FieldValue = FieldValue.zext(8) << (8 - BitWidth);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000264 } else
Jay Foad6d4db0c2010-12-07 08:25:34 +0000265 FieldValue = FieldValue.zext(8);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000266 }
267
268 // Append the last element.
269 Elements.push_back(llvm::ConstantInt::get(CGM.getLLVMContext(),
270 FieldValue));
271 NextFieldOffsetInBytes++;
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000272}
273
Ken Dyck30a87e32011-03-11 23:42:54 +0000274void ConstStructBuilder::AppendPadding(CharUnits PadSize) {
275 if (PadSize.isZero())
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000276 return;
277
278 const llvm::Type *Ty = llvm::Type::getInt8Ty(CGM.getLLVMContext());
Ken Dyck30a87e32011-03-11 23:42:54 +0000279 if (PadSize > CharUnits::One())
280 Ty = llvm::ArrayType::get(Ty, PadSize.getQuantity());
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000281
Nuno Lopes5863c992010-04-16 20:56:35 +0000282 llvm::Constant *C = llvm::UndefValue::get(Ty);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000283 Elements.push_back(C);
284 assert(getAlignment(C) == 1 && "Padding must have 1 byte alignment!");
285
286 NextFieldOffsetInBytes += getSizeInBytes(C);
287}
288
Ken Dyck327b77a2011-03-11 02:17:05 +0000289void ConstStructBuilder::AppendTailPadding(CharUnits RecordSize) {
290 assert(NextFieldOffsetInBytes <= RecordSize.getQuantity() &&
291 "Size mismatch!");
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000292
Ken Dyck30a87e32011-03-11 23:42:54 +0000293 AppendPadding(RecordSize - CharUnits::fromQuantity(NextFieldOffsetInBytes));
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000294}
295
296void ConstStructBuilder::ConvertStructToPacked() {
297 std::vector<llvm::Constant *> PackedElements;
298 uint64_t ElementOffsetInBytes = 0;
299
300 for (unsigned i = 0, e = Elements.size(); i != e; ++i) {
301 llvm::Constant *C = Elements[i];
302
303 unsigned ElementAlign =
304 CGM.getTargetData().getABITypeAlignment(C->getType());
305 uint64_t AlignedElementOffsetInBytes =
306 llvm::RoundUpToAlignment(ElementOffsetInBytes, ElementAlign);
307
308 if (AlignedElementOffsetInBytes > ElementOffsetInBytes) {
309 // We need some padding.
310 uint64_t NumBytes =
311 AlignedElementOffsetInBytes - ElementOffsetInBytes;
312
313 const llvm::Type *Ty = llvm::Type::getInt8Ty(CGM.getLLVMContext());
314 if (NumBytes > 1)
315 Ty = llvm::ArrayType::get(Ty, NumBytes);
316
Nuno Lopes5863c992010-04-16 20:56:35 +0000317 llvm::Constant *Padding = llvm::UndefValue::get(Ty);
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000318 PackedElements.push_back(Padding);
319 ElementOffsetInBytes += getSizeInBytes(Padding);
320 }
321
322 PackedElements.push_back(C);
323 ElementOffsetInBytes += getSizeInBytes(C);
324 }
325
326 assert(ElementOffsetInBytes == NextFieldOffsetInBytes &&
327 "Packing the struct changed its size!");
328
329 Elements = PackedElements;
330 LLVMStructAlignment = 1;
331 Packed = true;
332}
333
334bool ConstStructBuilder::Build(InitListExpr *ILE) {
335 RecordDecl *RD = ILE->getType()->getAs<RecordType>()->getDecl();
336 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
337
338 unsigned FieldNo = 0;
339 unsigned ElementNo = 0;
340 for (RecordDecl::field_iterator Field = RD->field_begin(),
Chris Lattnerff0e2a32010-04-13 18:16:19 +0000341 FieldEnd = RD->field_end(); Field != FieldEnd; ++Field, ++FieldNo) {
342
343 // If this is a union, skip all the fields that aren't being initialized.
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000344 if (RD->isUnion() && ILE->getInitializedFieldInUnion() != *Field)
345 continue;
346
Chris Lattnerff0e2a32010-04-13 18:16:19 +0000347 // Don't emit anonymous bitfields, they just affect layout.
348 if (Field->isBitField() && !Field->getIdentifier())
349 continue;
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000350
Chris Lattnerff0e2a32010-04-13 18:16:19 +0000351 // Get the initializer. A struct can include fields without initializers,
352 // we just use explicit null values for them.
353 llvm::Constant *EltInit;
354 if (ElementNo < ILE->getNumInits())
355 EltInit = CGM.EmitConstantExpr(ILE->getInit(ElementNo++),
356 Field->getType(), CGF);
357 else
358 EltInit = CGM.EmitNullConstant(Field->getType());
Eli Friedman3ee10222010-07-17 23:55:01 +0000359
360 if (!EltInit)
361 return false;
Chris Lattnerff0e2a32010-04-13 18:16:19 +0000362
363 if (!Field->isBitField()) {
364 // Handle non-bitfield members.
365 if (!AppendField(*Field, Layout.getFieldOffset(FieldNo), EltInit))
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000366 return false;
367 } else {
Chris Lattnerff0e2a32010-04-13 18:16:19 +0000368 // Otherwise we have a bitfield.
Chris Lattner9a3459f2010-07-05 17:04:23 +0000369 AppendBitField(*Field, Layout.getFieldOffset(FieldNo),
370 cast<llvm::ConstantInt>(EltInit));
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000371 }
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000372 }
373
Ken Dyckc8ae5502011-02-09 01:59:34 +0000374 uint64_t LayoutSizeInBytes = Layout.getSize().getQuantity();
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000375
376 if (NextFieldOffsetInBytes > LayoutSizeInBytes) {
377 // If the struct is bigger than the size of the record type,
378 // we must have a flexible array member at the end.
379 assert(RD->hasFlexibleArrayMember() &&
380 "Must have flexible array member if struct is bigger than type!");
381
382 // No tail padding is necessary.
383 return true;
384 }
385
386 uint64_t LLVMSizeInBytes = llvm::RoundUpToAlignment(NextFieldOffsetInBytes,
387 LLVMStructAlignment);
388
389 // Check if we need to convert the struct to a packed struct.
390 if (NextFieldOffsetInBytes <= LayoutSizeInBytes &&
391 LLVMSizeInBytes > LayoutSizeInBytes) {
392 assert(!Packed && "Size mismatch!");
393
394 ConvertStructToPacked();
Chris Lattner3cff64a2010-04-16 21:02:32 +0000395 assert(NextFieldOffsetInBytes <= LayoutSizeInBytes &&
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000396 "Converting to packed did not help!");
397 }
398
399 // Append tail padding if necessary.
Ken Dyck327b77a2011-03-11 02:17:05 +0000400 AppendTailPadding(Layout.getSize());
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000401
Ken Dyckc8ae5502011-02-09 01:59:34 +0000402 assert(Layout.getSize().getQuantity() == NextFieldOffsetInBytes &&
Chris Lattnercfa3e7a2010-04-13 17:45:57 +0000403 "Tail padding mismatch!");
404
405 return true;
406}
407
408llvm::Constant *ConstStructBuilder::
409 BuildStruct(CodeGenModule &CGM, CodeGenFunction *CGF, InitListExpr *ILE) {
410 ConstStructBuilder Builder(CGM, CGF);
411
412 if (!Builder.Build(ILE))
413 return 0;
414
415 llvm::Constant *Result =
416 llvm::ConstantStruct::get(CGM.getLLVMContext(),
417 Builder.Elements, Builder.Packed);
418
419 assert(llvm::RoundUpToAlignment(Builder.NextFieldOffsetInBytes,
420 Builder.getAlignment(Result)) ==
421 Builder.getSizeInBytes(Result) && "Size mismatch!");
422
423 return Result;
424}
425
426
427//===----------------------------------------------------------------------===//
428// ConstExprEmitter
429//===----------------------------------------------------------------------===//
430
Benjamin Kramer337e3a52009-11-28 19:45:26 +0000431class ConstExprEmitter :
Anders Carlsson8a88c382008-01-26 04:30:23 +0000432 public StmtVisitor<ConstExprEmitter, llvm::Constant*> {
Anders Carlsson610ee712008-01-26 01:36:00 +0000433 CodeGenModule &CGM;
Lauro Ramos Venancio01a72ff2008-02-26 21:41:45 +0000434 CodeGenFunction *CGF;
Owen Anderson170229f2009-07-14 23:10:40 +0000435 llvm::LLVMContext &VMContext;
Anders Carlsson610ee712008-01-26 01:36:00 +0000436public:
Lauro Ramos Venancio01a72ff2008-02-26 21:41:45 +0000437 ConstExprEmitter(CodeGenModule &cgm, CodeGenFunction *cgf)
Owen Anderson170229f2009-07-14 23:10:40 +0000438 : CGM(cgm), CGF(cgf), VMContext(cgm.getLLVMContext()) {
Anders Carlsson610ee712008-01-26 01:36:00 +0000439 }
Mike Stump11289f42009-09-09 15:08:12 +0000440
Anders Carlsson610ee712008-01-26 01:36:00 +0000441 //===--------------------------------------------------------------------===//
442 // Visitor Methods
443 //===--------------------------------------------------------------------===//
Mike Stump11289f42009-09-09 15:08:12 +0000444
Anders Carlsson610ee712008-01-26 01:36:00 +0000445 llvm::Constant *VisitStmt(Stmt *S) {
Anders Carlsson24b14472009-03-03 16:43:34 +0000446 return 0;
Anders Carlsson610ee712008-01-26 01:36:00 +0000447 }
Mike Stump11289f42009-09-09 15:08:12 +0000448
449 llvm::Constant *VisitParenExpr(ParenExpr *PE) {
450 return Visit(PE->getSubExpr());
Anders Carlsson610ee712008-01-26 01:36:00 +0000451 }
Mike Stump11289f42009-09-09 15:08:12 +0000452
Anders Carlsson610ee712008-01-26 01:36:00 +0000453 llvm::Constant *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
454 return Visit(E->getInitializer());
455 }
John McCallf3a88602011-02-03 08:15:49 +0000456
Anders Carlsson128a5d52009-10-03 15:02:02 +0000457 llvm::Constant *VisitUnaryAddrOf(UnaryOperator *E) {
John McCallf3a88602011-02-03 08:15:49 +0000458 if (E->getType()->isMemberPointerType())
459 return CGM.getMemberPointerConstant(E);
Anders Carlsson259688c2010-02-02 03:37:46 +0000460
Anders Carlsson128a5d52009-10-03 15:02:02 +0000461 return 0;
462 }
463
Chris Lattner3eb172a2009-10-13 07:14:16 +0000464 llvm::Constant *VisitBinSub(BinaryOperator *E) {
465 // This must be a pointer/pointer subtraction. This only happens for
466 // address of label.
467 if (!isa<AddrLabelExpr>(E->getLHS()->IgnoreParenNoopCasts(CGM.getContext())) ||
468 !isa<AddrLabelExpr>(E->getRHS()->IgnoreParenNoopCasts(CGM.getContext())))
469 return 0;
470
471 llvm::Constant *LHS = CGM.EmitConstantExpr(E->getLHS(),
472 E->getLHS()->getType(), CGF);
473 llvm::Constant *RHS = CGM.EmitConstantExpr(E->getRHS(),
474 E->getRHS()->getType(), CGF);
475
476 const llvm::Type *ResultType = ConvertType(E->getType());
477 LHS = llvm::ConstantExpr::getPtrToInt(LHS, ResultType);
478 RHS = llvm::ConstantExpr::getPtrToInt(RHS, ResultType);
479
480 // No need to divide by element size, since addr of label is always void*,
481 // which has size 1 in GNUish.
482 return llvm::ConstantExpr::getSub(LHS, RHS);
483 }
484
Argyrios Kyrtzidis3bab3d22008-08-18 23:01:59 +0000485 llvm::Constant *VisitCastExpr(CastExpr* E) {
Anders Carlsson3b0c5dc2009-08-22 23:54:44 +0000486 switch (E->getCastKind()) {
John McCalle3027922010-08-25 11:45:40 +0000487 case CK_ToUnion: {
Anders Carlsson3b0c5dc2009-08-22 23:54:44 +0000488 // GCC cast to union extension
489 assert(E->getType()->isUnionType() &&
490 "Destination type is not union type!");
Nuno Lopes4d78cf02009-01-17 00:48:48 +0000491 const llvm::Type *Ty = ConvertType(E->getType());
Anders Carlsson80f97ab2009-04-08 04:48:15 +0000492 Expr *SubExpr = E->getSubExpr();
Anders Carlsson3b0c5dc2009-08-22 23:54:44 +0000493
Mike Stump11289f42009-09-09 15:08:12 +0000494 llvm::Constant *C =
Anders Carlssond65ab042009-07-31 21:38:39 +0000495 CGM.EmitConstantExpr(SubExpr, SubExpr->getType(), CGF);
496 if (!C)
497 return 0;
Mike Stump11289f42009-09-09 15:08:12 +0000498
Anders Carlssond65ab042009-07-31 21:38:39 +0000499 // Build a struct with the union sub-element as the first member,
500 // and padded to the appropriate size
501 std::vector<llvm::Constant*> Elts;
502 std::vector<const llvm::Type*> Types;
503 Elts.push_back(C);
504 Types.push_back(C->getType());
505 unsigned CurSize = CGM.getTargetData().getTypeAllocSize(C->getType());
506 unsigned TotalSize = CGM.getTargetData().getTypeAllocSize(Ty);
Mike Stump11289f42009-09-09 15:08:12 +0000507
Anders Carlssond65ab042009-07-31 21:38:39 +0000508 assert(CurSize <= TotalSize && "Union size mismatch!");
509 if (unsigned NumPadBytes = TotalSize - CurSize) {
Owen Anderson41a75022009-08-13 21:57:51 +0000510 const llvm::Type *Ty = llvm::Type::getInt8Ty(VMContext);
Anders Carlssond65ab042009-07-31 21:38:39 +0000511 if (NumPadBytes > 1)
512 Ty = llvm::ArrayType::get(Ty, NumPadBytes);
Anders Carlsson3b0c5dc2009-08-22 23:54:44 +0000513
Nuno Lopes5863c992010-04-16 20:56:35 +0000514 Elts.push_back(llvm::UndefValue::get(Ty));
Anders Carlssond65ab042009-07-31 21:38:39 +0000515 Types.push_back(Ty);
516 }
Mike Stump11289f42009-09-09 15:08:12 +0000517
Anders Carlsson3b0c5dc2009-08-22 23:54:44 +0000518 llvm::StructType* STy =
519 llvm::StructType::get(C->getType()->getContext(), Types, false);
Anders Carlssond65ab042009-07-31 21:38:39 +0000520 return llvm::ConstantStruct::get(STy, Elts);
Nuno Lopes4d78cf02009-01-17 00:48:48 +0000521 }
John McCalle3027922010-08-25 11:45:40 +0000522 case CK_NullToMemberPointer: {
John McCalla1dee5302010-08-22 10:59:02 +0000523 const MemberPointerType *MPT = E->getType()->getAs<MemberPointerType>();
John McCall7a9aac22010-08-23 01:21:21 +0000524 return CGM.getCXXABI().EmitNullMemberPointer(MPT);
John McCalla1dee5302010-08-22 10:59:02 +0000525 }
Anders Carlsson3162e492009-10-03 15:13:22 +0000526
John McCalle3027922010-08-25 11:45:40 +0000527 case CK_BaseToDerivedMemberPointer: {
John McCalla1dee5302010-08-22 10:59:02 +0000528 Expr *SubExpr = E->getSubExpr();
John McCall84fa5102010-08-22 04:16:24 +0000529 llvm::Constant *C =
530 CGM.EmitConstantExpr(SubExpr, SubExpr->getType(), CGF);
531 if (!C) return 0;
Anders Carlsson3162e492009-10-03 15:13:22 +0000532
John McCall7a9aac22010-08-23 01:21:21 +0000533 return CGM.getCXXABI().EmitMemberPointerConversion(C, E);
Anders Carlsson3162e492009-10-03 15:13:22 +0000534 }
Anders Carlsson9500ad12009-10-18 20:31:03 +0000535
John McCalle3027922010-08-25 11:45:40 +0000536 case CK_BitCast:
Anders Carlsson9500ad12009-10-18 20:31:03 +0000537 // This must be a member function pointer cast.
538 return Visit(E->getSubExpr());
539
Anders Carlsson3b0c5dc2009-08-22 23:54:44 +0000540 default: {
541 // FIXME: This should be handled by the CK_NoOp cast kind.
542 // Explicit and implicit no-op casts
543 QualType Ty = E->getType(), SubTy = E->getSubExpr()->getType();
544 if (CGM.getContext().hasSameUnqualifiedType(Ty, SubTy))
Chris Lattnera2f9bd52009-10-13 22:12:09 +0000545 return Visit(E->getSubExpr());
546
547 // Handle integer->integer casts for address-of-label differences.
548 if (Ty->isIntegerType() && SubTy->isIntegerType() &&
549 CGF) {
550 llvm::Value *Src = Visit(E->getSubExpr());
551 if (Src == 0) return 0;
552
553 // Use EmitScalarConversion to perform the conversion.
554 return cast<llvm::Constant>(CGF->EmitScalarConversion(Src, SubTy, Ty));
555 }
556
Anders Carlsson3b0c5dc2009-08-22 23:54:44 +0000557 return 0;
Eli Friedman825fe752009-02-22 07:29:04 +0000558 }
Anders Carlsson3b0c5dc2009-08-22 23:54:44 +0000559 }
Anders Carlsson610ee712008-01-26 01:36:00 +0000560 }
Devang Patela703a672008-02-05 02:39:50 +0000561
Chris Lattneraa9c7ae2008-04-08 04:40:51 +0000562 llvm::Constant *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
563 return Visit(DAE->getExpr());
564 }
565
Eli Friedmana2eaffc2008-05-30 10:24:46 +0000566 llvm::Constant *EmitArrayInitialization(InitListExpr *ILE) {
Nuno Lopes247a1382010-04-18 19:06:43 +0000567 unsigned NumInitElements = ILE->getNumInits();
Benjamin Kramer0f074df2011-03-02 21:27:44 +0000568 if (NumInitElements == 1 && ILE->getType() == ILE->getInit(0)->getType() &&
Nuno Lopes247a1382010-04-18 19:06:43 +0000569 (isa<StringLiteral>(ILE->getInit(0)) ||
570 isa<ObjCEncodeExpr>(ILE->getInit(0))))
571 return Visit(ILE->getInit(0));
572
Nuno Lopes74b59522010-04-16 23:19:41 +0000573 std::vector<llvm::Constant*> Elts;
574 const llvm::ArrayType *AType =
575 cast<llvm::ArrayType>(ConvertType(ILE->getType()));
Devang Patela703a672008-02-05 02:39:50 +0000576 const llvm::Type *ElemTy = AType->getElementType();
577 unsigned NumElements = AType->getNumElements();
578
Mike Stump11289f42009-09-09 15:08:12 +0000579 // Initialising an array requires us to automatically
Devang Patela703a672008-02-05 02:39:50 +0000580 // initialise any elements that have not been initialised explicitly
581 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
582
583 // Copy initializer elements.
584 unsigned i = 0;
Eli Friedman34994cb2008-05-30 19:58:50 +0000585 bool RewriteType = false;
Devang Patela703a672008-02-05 02:39:50 +0000586 for (; i < NumInitableElts; ++i) {
Anders Carlsson80f97ab2009-04-08 04:48:15 +0000587 Expr *Init = ILE->getInit(i);
588 llvm::Constant *C = CGM.EmitConstantExpr(Init, Init->getType(), CGF);
Daniel Dunbar38ad1e62009-02-17 18:43:32 +0000589 if (!C)
590 return 0;
Eli Friedman34994cb2008-05-30 19:58:50 +0000591 RewriteType |= (C->getType() != ElemTy);
Devang Patela703a672008-02-05 02:39:50 +0000592 Elts.push_back(C);
593 }
Eli Friedman34994cb2008-05-30 19:58:50 +0000594
Devang Patela703a672008-02-05 02:39:50 +0000595 // Initialize remaining array elements.
Eli Friedman20bb5e02009-04-13 21:47:26 +0000596 // FIXME: This doesn't handle member pointers correctly!
Devang Patela703a672008-02-05 02:39:50 +0000597 for (; i < NumElements; ++i)
Owen Anderson0b75f232009-07-31 20:28:54 +0000598 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
Devang Patela703a672008-02-05 02:39:50 +0000599
Eli Friedman34994cb2008-05-30 19:58:50 +0000600 if (RewriteType) {
601 // FIXME: Try to avoid packing the array
602 std::vector<const llvm::Type*> Types;
603 for (unsigned i = 0; i < Elts.size(); ++i)
604 Types.push_back(Elts[i]->getType());
Mike Stump11289f42009-09-09 15:08:12 +0000605 const llvm::StructType *SType = llvm::StructType::get(AType->getContext(),
Owen Anderson758428f2009-08-05 23:18:46 +0000606 Types, true);
Owen Anderson0e0189d2009-07-27 22:29:56 +0000607 return llvm::ConstantStruct::get(SType, Elts);
Eli Friedman34994cb2008-05-30 19:58:50 +0000608 }
609
Mike Stump11289f42009-09-09 15:08:12 +0000610 return llvm::ConstantArray::get(AType, Elts);
Devang Patela703a672008-02-05 02:39:50 +0000611 }
612
Eli Friedmana2eaffc2008-05-30 10:24:46 +0000613 llvm::Constant *EmitStructInitialization(InitListExpr *ILE) {
Anders Carlssonb7130ed2009-07-27 05:54:15 +0000614 return ConstStructBuilder::BuildStruct(CGM, CGF, ILE);
Devang Patela703a672008-02-05 02:39:50 +0000615 }
Mike Stump11289f42009-09-09 15:08:12 +0000616
Eli Friedmana2eaffc2008-05-30 10:24:46 +0000617 llvm::Constant *EmitUnionInitialization(InitListExpr *ILE) {
Anders Carlsson0d5f37b2009-07-31 21:34:04 +0000618 return ConstStructBuilder::BuildStruct(CGM, CGF, ILE);
Eli Friedmana2eaffc2008-05-30 10:24:46 +0000619 }
620
Anders Carlsson02714ed2009-01-30 06:13:25 +0000621 llvm::Constant *VisitImplicitValueInitExpr(ImplicitValueInitExpr* E) {
Eli Friedman20bb5e02009-04-13 21:47:26 +0000622 return CGM.EmitNullConstant(E->getType());
Anders Carlsson02714ed2009-01-30 06:13:25 +0000623 }
Mike Stump11289f42009-09-09 15:08:12 +0000624
Anders Carlsson610ee712008-01-26 01:36:00 +0000625 llvm::Constant *VisitInitListExpr(InitListExpr *ILE) {
Eli Friedmana2eaffc2008-05-30 10:24:46 +0000626 if (ILE->getType()->isScalarType()) {
627 // We have a scalar in braces. Just use the first element.
Anders Carlsson80f97ab2009-04-08 04:48:15 +0000628 if (ILE->getNumInits() > 0) {
629 Expr *Init = ILE->getInit(0);
630 return CGM.EmitConstantExpr(Init, Init->getType(), CGF);
631 }
Eli Friedman20bb5e02009-04-13 21:47:26 +0000632 return CGM.EmitNullConstant(ILE->getType());
Anders Carlsson610ee712008-01-26 01:36:00 +0000633 }
Mike Stump11289f42009-09-09 15:08:12 +0000634
Eli Friedmana2eaffc2008-05-30 10:24:46 +0000635 if (ILE->getType()->isArrayType())
636 return EmitArrayInitialization(ILE);
Devang Patel45a65d22008-01-29 23:23:18 +0000637
Anders Carlssone18e29e2009-11-21 23:56:04 +0000638 if (ILE->getType()->isRecordType())
Eli Friedmana2eaffc2008-05-30 10:24:46 +0000639 return EmitStructInitialization(ILE);
640
641 if (ILE->getType()->isUnionType())
642 return EmitUnionInitialization(ILE);
643
Eli Friedmanc52a9362010-01-02 23:43:59 +0000644 // If ILE was a constant vector, we would have handled it already.
Eli Friedmana2eaffc2008-05-30 10:24:46 +0000645 if (ILE->getType()->isVectorType())
Eli Friedmanc52a9362010-01-02 23:43:59 +0000646 return 0;
Eli Friedmana2eaffc2008-05-30 10:24:46 +0000647
Devang Patela703a672008-02-05 02:39:50 +0000648 assert(0 && "Unable to handle InitListExpr");
Chris Lattnera087ff92008-03-01 08:45:05 +0000649 // Get rid of control reaches end of void function warning.
650 // Not reached.
651 return 0;
Anders Carlsson610ee712008-01-26 01:36:00 +0000652 }
Eli Friedmana2433112008-02-21 17:57:49 +0000653
John McCall49786a62010-02-02 08:02:49 +0000654 llvm::Constant *VisitCXXConstructExpr(CXXConstructExpr *E) {
655 if (!E->getConstructor()->isTrivial())
656 return 0;
657
Anders Carlssoncb86e102010-02-05 18:38:45 +0000658 QualType Ty = E->getType();
659
660 // FIXME: We should not have to call getBaseElementType here.
661 const RecordType *RT =
662 CGM.getContext().getBaseElementType(Ty)->getAs<RecordType>();
663 const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
664
665 // If the class doesn't have a trivial destructor, we can't emit it as a
666 // constant expr.
667 if (!RD->hasTrivialDestructor())
668 return 0;
669
John McCall49786a62010-02-02 08:02:49 +0000670 // Only copy and default constructors can be trivial.
671
John McCall49786a62010-02-02 08:02:49 +0000672
673 if (E->getNumArgs()) {
674 assert(E->getNumArgs() == 1 && "trivial ctor with > 1 argument");
675 assert(E->getConstructor()->isCopyConstructor() &&
676 "trivial ctor has argument but isn't a copy ctor");
677
678 Expr *Arg = E->getArg(0);
679 assert(CGM.getContext().hasSameUnqualifiedType(Ty, Arg->getType()) &&
680 "argument to copy ctor is of wrong type");
681
Chandler Carruth173bfe42010-02-02 12:15:55 +0000682 return Visit(Arg);
John McCall49786a62010-02-02 08:02:49 +0000683 }
684
685 return CGM.EmitNullConstant(Ty);
686 }
687
Anders Carlsson610ee712008-01-26 01:36:00 +0000688 llvm::Constant *VisitStringLiteral(StringLiteral *E) {
Chris Lattner36fc8792008-02-11 00:02:17 +0000689 assert(!E->getType()->isPointerType() && "Strings are always arrays");
Mike Stump11289f42009-09-09 15:08:12 +0000690
Chris Lattnerd7e7b8e2009-02-24 22:18:39 +0000691 // This must be a string initializing an array in a static initializer.
692 // Don't emit it as the address of the string, emit the string data itself
693 // as an inline array.
Owen Anderson41a75022009-08-13 21:57:51 +0000694 return llvm::ConstantArray::get(VMContext,
695 CGM.GetStringForStringLiteral(E), false);
Anders Carlsson610ee712008-01-26 01:36:00 +0000696 }
697
Chris Lattnerd7e7b8e2009-02-24 22:18:39 +0000698 llvm::Constant *VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
699 // This must be an @encode initializing an array in a static initializer.
700 // Don't emit it as the address of the string, emit the string data itself
701 // as an inline array.
702 std::string Str;
703 CGM.getContext().getObjCEncodingForType(E->getEncodedType(), Str);
704 const ConstantArrayType *CAT = cast<ConstantArrayType>(E->getType());
Mike Stump11289f42009-09-09 15:08:12 +0000705
Chris Lattnerd7e7b8e2009-02-24 22:18:39 +0000706 // Resize the string to the right size, adding zeros at the end, or
707 // truncating as needed.
708 Str.resize(CAT->getSize().getZExtValue(), '\0');
Owen Anderson41a75022009-08-13 21:57:51 +0000709 return llvm::ConstantArray::get(VMContext, Str, false);
Chris Lattnerd7e7b8e2009-02-24 22:18:39 +0000710 }
Mike Stump11289f42009-09-09 15:08:12 +0000711
Eli Friedman045bf4f2008-05-29 11:22:45 +0000712 llvm::Constant *VisitUnaryExtension(const UnaryOperator *E) {
713 return Visit(E->getSubExpr());
714 }
Mike Stumpa6703322009-02-19 22:01:56 +0000715
Anders Carlsson610ee712008-01-26 01:36:00 +0000716 // Utility methods
717 const llvm::Type *ConvertType(QualType T) {
718 return CGM.getTypes().ConvertType(T);
719 }
Anders Carlsson8a88c382008-01-26 04:30:23 +0000720
Anders Carlsson43088772008-11-16 06:23:45 +0000721public:
Anders Carlsson8a88c382008-01-26 04:30:23 +0000722 llvm::Constant *EmitLValue(Expr *E) {
Anders Carlssona4139112008-01-26 02:08:50 +0000723 switch (E->getStmtClass()) {
Eli Friedmana5b20232008-02-11 00:23:10 +0000724 default: break;
Anders Carlssona4139112008-01-26 02:08:50 +0000725 case Expr::CompoundLiteralExprClass: {
726 // Note that due to the nature of compound literals, this is guaranteed
727 // to be the only use of the variable, so we just generate it here.
Anders Carlsson8a88c382008-01-26 04:30:23 +0000728 CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
729 llvm::Constant* C = Visit(CLE->getInitializer());
Daniel Dunbar38ad1e62009-02-17 18:43:32 +0000730 // FIXME: "Leaked" on failure.
731 if (C)
Owen Andersonc10c8d32009-07-08 19:05:04 +0000732 C = new llvm::GlobalVariable(CGM.getModule(), C->getType(),
Eli Friedman95f89632009-08-26 20:01:39 +0000733 E->getType().isConstant(CGM.getContext()),
Daniel Dunbar38ad1e62009-02-17 18:43:32 +0000734 llvm::GlobalValue::InternalLinkage,
Eli Friedman95f89632009-08-26 20:01:39 +0000735 C, ".compoundliteral", 0, false,
736 E->getType().getAddressSpace());
Anders Carlssona4139112008-01-26 02:08:50 +0000737 return C;
Anders Carlsson8a88c382008-01-26 04:30:23 +0000738 }
Douglas Gregor4bd90e52009-10-23 18:54:35 +0000739 case Expr::DeclRefExprClass: {
Rafael Espindolabef98682010-03-04 21:26:03 +0000740 ValueDecl *Decl = cast<DeclRefExpr>(E)->getDecl();
741 if (Decl->hasAttr<WeakRefAttr>())
Eli Friedman04831922010-08-22 01:00:03 +0000742 return CGM.GetWeakRefReference(Decl);
Anders Carlssona4139112008-01-26 02:08:50 +0000743 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Decl))
Anders Carlssonecf9bf02009-09-10 23:43:36 +0000744 return CGM.GetAddrOfFunction(FD);
Steve Naroff08899ff2008-04-15 22:42:06 +0000745 if (const VarDecl* VD = dyn_cast<VarDecl>(Decl)) {
Daniel Dunbard4ecca12009-02-24 18:41:57 +0000746 // We can never refer to a variable with local storage.
Mike Stump11289f42009-09-09 15:08:12 +0000747 if (!VD->hasLocalStorage()) {
Daniel Dunbard4ecca12009-02-24 18:41:57 +0000748 if (VD->isFileVarDecl() || VD->hasExternalStorage())
749 return CGM.GetAddrOfGlobalVar(VD);
John McCall1c9c3fd2010-10-15 04:57:14 +0000750 else if (VD->isLocalVarDecl()) {
Daniel Dunbard4ecca12009-02-24 18:41:57 +0000751 assert(CGF && "Can't access static local vars without CGF");
752 return CGF->GetAddrOfStaticLocalVar(VD);
753 }
Steve Naroff08899ff2008-04-15 22:42:06 +0000754 }
Lauro Ramos Venancio01a72ff2008-02-26 21:41:45 +0000755 }
Eli Friedmana5b20232008-02-11 00:23:10 +0000756 break;
Anders Carlssona4139112008-01-26 02:08:50 +0000757 }
Daniel Dunbarc4baa062008-08-13 23:20:05 +0000758 case Expr::StringLiteralClass:
759 return CGM.GetAddrOfConstantStringFromLiteral(cast<StringLiteral>(E));
Chris Lattnerd7e7b8e2009-02-24 22:18:39 +0000760 case Expr::ObjCEncodeExprClass:
761 return CGM.GetAddrOfConstantStringFromObjCEncode(cast<ObjCEncodeExpr>(E));
Anders Carlsson43088772008-11-16 06:23:45 +0000762 case Expr::ObjCStringLiteralClass: {
763 ObjCStringLiteral* SL = cast<ObjCStringLiteral>(E);
David Chisnall481e3a82010-01-23 02:40:42 +0000764 llvm::Constant *C =
765 CGM.getObjCRuntime().GenerateConstantString(SL->getString());
Owen Andersonade90fd2009-07-29 18:54:39 +0000766 return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
Anders Carlsson43088772008-11-16 06:23:45 +0000767 }
Chris Lattnerb4396f42008-12-12 05:18:02 +0000768 case Expr::PredefinedExprClass: {
Eli Friedmanb210fc52009-11-14 08:37:13 +0000769 unsigned Type = cast<PredefinedExpr>(E)->getIdentType();
770 if (CGF) {
Daniel Dunbarb1d94a92010-08-21 03:01:12 +0000771 LValue Res = CGF->EmitPredefinedLValue(cast<PredefinedExpr>(E));
Eli Friedmanb210fc52009-11-14 08:37:13 +0000772 return cast<llvm::Constant>(Res.getAddress());
773 } else if (Type == PredefinedExpr::PrettyFunction) {
774 return CGM.GetAddrOfConstantCString("top level", ".tmp");
775 }
Mike Stump11289f42009-09-09 15:08:12 +0000776
Eli Friedmanb210fc52009-11-14 08:37:13 +0000777 return CGM.GetAddrOfConstantCString("", ".tmp");
Chris Lattnerb4396f42008-12-12 05:18:02 +0000778 }
Eli Friedman529a99b2009-01-25 01:21:06 +0000779 case Expr::AddrLabelExprClass: {
780 assert(CGF && "Invalid address of label expression outside function.");
Chris Lattner6c4d2552009-10-28 23:59:40 +0000781 llvm::Constant *Ptr =
782 CGF->GetAddrOfLabel(cast<AddrLabelExpr>(E)->getLabel());
783 return llvm::ConstantExpr::getBitCast(Ptr, ConvertType(E->getType()));
Eli Friedman529a99b2009-01-25 01:21:06 +0000784 }
Eli Friedmanc69d4542009-01-25 01:54:01 +0000785 case Expr::CallExprClass: {
786 CallExpr* CE = cast<CallExpr>(E);
David Chisnall481e3a82010-01-23 02:40:42 +0000787 unsigned builtin = CE->isBuiltinCall(CGM.getContext());
788 if (builtin !=
789 Builtin::BI__builtin___CFStringMakeConstantString &&
790 builtin !=
791 Builtin::BI__builtin___NSStringMakeConstantString)
Eli Friedmanc69d4542009-01-25 01:54:01 +0000792 break;
793 const Expr *Arg = CE->getArg(0)->IgnoreParenCasts();
794 const StringLiteral *Literal = cast<StringLiteral>(Arg);
David Chisnall481e3a82010-01-23 02:40:42 +0000795 if (builtin ==
796 Builtin::BI__builtin___NSStringMakeConstantString) {
797 return CGM.getObjCRuntime().GenerateConstantString(Literal);
798 }
Steve Naroff66afeb52009-03-31 16:53:37 +0000799 // FIXME: need to deal with UCN conversion issues.
Steve Naroffe14b3682009-04-01 13:55:36 +0000800 return CGM.GetAddrOfConstantCFString(Literal);
Eli Friedmanc69d4542009-01-25 01:54:01 +0000801 }
Mike Stump5d2534ad2009-02-19 01:01:04 +0000802 case Expr::BlockExprClass: {
Anders Carlssoned5e69f2009-03-01 01:09:12 +0000803 std::string FunctionName;
804 if (CGF)
805 FunctionName = CGF->CurFn->getName();
806 else
807 FunctionName = "global";
808
809 return CGM.GetAddrOfGlobalBlock(cast<BlockExpr>(E), FunctionName.c_str());
Mike Stump5d2534ad2009-02-19 01:01:04 +0000810 }
Eli Friedmana5b20232008-02-11 00:23:10 +0000811 }
Daniel Dunbar38ad1e62009-02-17 18:43:32 +0000812
813 return 0;
Anders Carlssona4139112008-01-26 02:08:50 +0000814 }
Anders Carlssona4139112008-01-26 02:08:50 +0000815};
Mike Stump11289f42009-09-09 15:08:12 +0000816
Anders Carlsson610ee712008-01-26 01:36:00 +0000817} // end anonymous namespace.
818
Lauro Ramos Venancio01a72ff2008-02-26 21:41:45 +0000819llvm::Constant *CodeGenModule::EmitConstantExpr(const Expr *E,
Anders Carlsson80f97ab2009-04-08 04:48:15 +0000820 QualType DestType,
Chris Lattner0f398c42008-07-26 22:37:01 +0000821 CodeGenFunction *CGF) {
Anders Carlsson38eef1d2008-12-01 02:42:14 +0000822 Expr::EvalResult Result;
Mike Stump11289f42009-09-09 15:08:12 +0000823
Anders Carlssond8e39bb2009-04-11 01:08:03 +0000824 bool Success = false;
Mike Stump11289f42009-09-09 15:08:12 +0000825
Eli Friedman751aa72b72009-05-27 06:04:58 +0000826 if (DestType->isReferenceType())
827 Success = E->EvaluateAsLValue(Result, Context);
Mike Stump11289f42009-09-09 15:08:12 +0000828 else
Anders Carlssond8e39bb2009-04-11 01:08:03 +0000829 Success = E->Evaluate(Result, Context);
Mike Stump11289f42009-09-09 15:08:12 +0000830
Eli Friedman274ab902009-11-14 08:51:33 +0000831 if (Success && !Result.HasSideEffects) {
Anders Carlsson38eef1d2008-12-01 02:42:14 +0000832 switch (Result.Val.getKind()) {
Daniel Dunbarf50e60b2009-01-28 22:24:07 +0000833 case APValue::Uninitialized:
Daniel Dunbar38ad1e62009-02-17 18:43:32 +0000834 assert(0 && "Constant expressions should be initialized.");
835 return 0;
Anders Carlssonee0d90f2008-11-15 20:45:50 +0000836 case APValue::LValue: {
Anders Carlssond8e39bb2009-04-11 01:08:03 +0000837 const llvm::Type *DestTy = getTypes().ConvertTypeForMem(DestType);
Mike Stump11289f42009-09-09 15:08:12 +0000838 llvm::Constant *Offset =
839 llvm::ConstantInt::get(llvm::Type::getInt64Ty(VMContext),
Ken Dyck02990832010-01-15 12:37:54 +0000840 Result.Val.getLValueOffset().getQuantity());
Mike Stump11289f42009-09-09 15:08:12 +0000841
Daniel Dunbar18e14442009-02-19 21:44:24 +0000842 llvm::Constant *C;
Anders Carlsson38eef1d2008-12-01 02:42:14 +0000843 if (const Expr *LVBase = Result.Val.getLValueBase()) {
Daniel Dunbar18e14442009-02-19 21:44:24 +0000844 C = ConstExprEmitter(*this, CGF).EmitLValue(const_cast<Expr*>(LVBase));
Anders Carlsson43088772008-11-16 06:23:45 +0000845
Daniel Dunbar18e14442009-02-19 21:44:24 +0000846 // Apply offset if necessary.
847 if (!Offset->isNullValue()) {
Benjamin Kramerabd5b902009-10-13 10:07:13 +0000848 const llvm::Type *Type = llvm::Type::getInt8PtrTy(VMContext);
Owen Andersonade90fd2009-07-29 18:54:39 +0000849 llvm::Constant *Casted = llvm::ConstantExpr::getBitCast(C, Type);
850 Casted = llvm::ConstantExpr::getGetElementPtr(Casted, &Offset, 1);
851 C = llvm::ConstantExpr::getBitCast(Casted, C->getType());
Daniel Dunbar18e14442009-02-19 21:44:24 +0000852 }
853
854 // Convert to the appropriate type; this could be an lvalue for
855 // an integer.
Anders Carlsson6ef4ea42009-04-10 04:59:59 +0000856 if (isa<llvm::PointerType>(DestTy))
Owen Andersonade90fd2009-07-29 18:54:39 +0000857 return llvm::ConstantExpr::getBitCast(C, DestTy);
Daniel Dunbar18e14442009-02-19 21:44:24 +0000858
Owen Andersonade90fd2009-07-29 18:54:39 +0000859 return llvm::ConstantExpr::getPtrToInt(C, DestTy);
Daniel Dunbar18e14442009-02-19 21:44:24 +0000860 } else {
861 C = Offset;
862
863 // Convert to the appropriate type; this could be an lvalue for
864 // an integer.
Anders Carlsson6ef4ea42009-04-10 04:59:59 +0000865 if (isa<llvm::PointerType>(DestTy))
Owen Andersonade90fd2009-07-29 18:54:39 +0000866 return llvm::ConstantExpr::getIntToPtr(C, DestTy);
Daniel Dunbar18e14442009-02-19 21:44:24 +0000867
868 // If the types don't match this should only be a truncate.
Anders Carlsson6ef4ea42009-04-10 04:59:59 +0000869 if (C->getType() != DestTy)
Owen Andersonade90fd2009-07-29 18:54:39 +0000870 return llvm::ConstantExpr::getTrunc(C, DestTy);
Daniel Dunbar18e14442009-02-19 21:44:24 +0000871
872 return C;
Anders Carlsson43088772008-11-16 06:23:45 +0000873 }
Anders Carlssonee0d90f2008-11-15 20:45:50 +0000874 }
Eli Friedmandcbcbb82008-11-17 03:57:28 +0000875 case APValue::Int: {
Mike Stump11289f42009-09-09 15:08:12 +0000876 llvm::Constant *C = llvm::ConstantInt::get(VMContext,
Owen Andersonb7a2fe62009-07-24 23:12:58 +0000877 Result.Val.getInt());
Mike Stump11289f42009-09-09 15:08:12 +0000878
Chris Lattnerfa20e952010-06-26 21:52:32 +0000879 if (C->getType()->isIntegerTy(1)) {
Anders Carlssonee0d90f2008-11-15 20:45:50 +0000880 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
Owen Andersonade90fd2009-07-29 18:54:39 +0000881 C = llvm::ConstantExpr::getZExt(C, BoolTy);
Anders Carlssonee0d90f2008-11-15 20:45:50 +0000882 }
883 return C;
Eli Friedmandcbcbb82008-11-17 03:57:28 +0000884 }
Daniel Dunbarf50e60b2009-01-28 22:24:07 +0000885 case APValue::ComplexInt: {
886 llvm::Constant *Complex[2];
Mike Stump11289f42009-09-09 15:08:12 +0000887
Owen Andersonb7a2fe62009-07-24 23:12:58 +0000888 Complex[0] = llvm::ConstantInt::get(VMContext,
889 Result.Val.getComplexIntReal());
Mike Stump11289f42009-09-09 15:08:12 +0000890 Complex[1] = llvm::ConstantInt::get(VMContext,
Owen Andersonb7a2fe62009-07-24 23:12:58 +0000891 Result.Val.getComplexIntImag());
Mike Stump11289f42009-09-09 15:08:12 +0000892
Nick Lewycky41eaf0a2009-09-19 20:00:52 +0000893 // FIXME: the target may want to specify that this is packed.
894 return llvm::ConstantStruct::get(VMContext, Complex, 2, false);
Daniel Dunbarf50e60b2009-01-28 22:24:07 +0000895 }
Anders Carlssonee0d90f2008-11-15 20:45:50 +0000896 case APValue::Float:
Owen Andersone05f2ed2009-07-27 21:00:51 +0000897 return llvm::ConstantFP::get(VMContext, Result.Val.getFloat());
Anders Carlsson4425d8e2008-11-17 01:58:55 +0000898 case APValue::ComplexFloat: {
899 llvm::Constant *Complex[2];
Mike Stump11289f42009-09-09 15:08:12 +0000900
901 Complex[0] = llvm::ConstantFP::get(VMContext,
Owen Andersone05f2ed2009-07-27 21:00:51 +0000902 Result.Val.getComplexFloatReal());
903 Complex[1] = llvm::ConstantFP::get(VMContext,
904 Result.Val.getComplexFloatImag());
Mike Stump11289f42009-09-09 15:08:12 +0000905
Nick Lewycky41eaf0a2009-09-19 20:00:52 +0000906 // FIXME: the target may want to specify that this is packed.
907 return llvm::ConstantStruct::get(VMContext, Complex, 2, false);
Anders Carlsson4425d8e2008-11-17 01:58:55 +0000908 }
Nate Begeman1e31b162009-01-18 01:01:34 +0000909 case APValue::Vector: {
910 llvm::SmallVector<llvm::Constant *, 4> Inits;
911 unsigned NumElts = Result.Val.getVectorLength();
Mike Stump11289f42009-09-09 15:08:12 +0000912
Nate Begeman1e31b162009-01-18 01:01:34 +0000913 for (unsigned i = 0; i != NumElts; ++i) {
914 APValue &Elt = Result.Val.getVectorElt(i);
915 if (Elt.isInt())
Owen Andersonb7a2fe62009-07-24 23:12:58 +0000916 Inits.push_back(llvm::ConstantInt::get(VMContext, Elt.getInt()));
Nate Begeman1e31b162009-01-18 01:01:34 +0000917 else
Owen Andersone05f2ed2009-07-27 21:00:51 +0000918 Inits.push_back(llvm::ConstantFP::get(VMContext, Elt.getFloat()));
Nate Begeman1e31b162009-01-18 01:01:34 +0000919 }
Chris Lattner91c08ad2011-02-15 00:14:06 +0000920 return llvm::ConstantVector::get(Inits);
Nate Begeman1e31b162009-01-18 01:01:34 +0000921 }
Anders Carlssonee0d90f2008-11-15 20:45:50 +0000922 }
923 }
Eli Friedman10c24172008-06-01 15:31:44 +0000924
925 llvm::Constant* C = ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
Chris Lattnerfa20e952010-06-26 21:52:32 +0000926 if (C && C->getType()->isIntegerTy(1)) {
Eli Friedman10c24172008-06-01 15:31:44 +0000927 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
Owen Andersonade90fd2009-07-29 18:54:39 +0000928 C = llvm::ConstantExpr::getZExt(C, BoolTy);
Eli Friedman10c24172008-06-01 15:31:44 +0000929 }
930 return C;
Anders Carlsson610ee712008-01-26 01:36:00 +0000931}
Eli Friedman20bb5e02009-04-13 21:47:26 +0000932
John McCallf3a88602011-02-03 08:15:49 +0000933static uint64_t getFieldOffset(ASTContext &C, const FieldDecl *field) {
934 const ASTRecordLayout &layout = C.getASTRecordLayout(field->getParent());
935 return layout.getFieldOffset(field->getFieldIndex());
936}
937
938llvm::Constant *
939CodeGenModule::getMemberPointerConstant(const UnaryOperator *uo) {
940 // Member pointer constants always have a very particular form.
941 const MemberPointerType *type = cast<MemberPointerType>(uo->getType());
942 const ValueDecl *decl = cast<DeclRefExpr>(uo->getSubExpr())->getDecl();
943
944 // A member function pointer.
945 if (const CXXMethodDecl *method = dyn_cast<CXXMethodDecl>(decl))
946 return getCXXABI().EmitMemberPointer(method);
947
948 // Otherwise, a member data pointer.
949 uint64_t fieldOffset;
950 if (const FieldDecl *field = dyn_cast<FieldDecl>(decl))
951 fieldOffset = getFieldOffset(getContext(), field);
952 else {
953 const IndirectFieldDecl *ifield = cast<IndirectFieldDecl>(decl);
954
955 fieldOffset = 0;
956 for (IndirectFieldDecl::chain_iterator ci = ifield->chain_begin(),
957 ce = ifield->chain_end(); ci != ce; ++ci)
958 fieldOffset += getFieldOffset(getContext(), cast<FieldDecl>(*ci));
959 }
960
961 CharUnits chars = getContext().toCharUnitsFromBits((int64_t) fieldOffset);
962 return getCXXABI().EmitMemberDataPointer(type, chars);
963}
964
Anders Carlssonbe48c542010-05-18 16:51:41 +0000965static void
966FillInNullDataMemberPointers(CodeGenModule &CGM, QualType T,
967 std::vector<llvm::Constant *> &Elements,
968 uint64_t StartOffset) {
969 assert(StartOffset % 8 == 0 && "StartOffset not byte aligned!");
970
John McCall614dbdc2010-08-22 21:01:12 +0000971 if (CGM.getTypes().isZeroInitializable(T))
Anders Carlssonbe48c542010-05-18 16:51:41 +0000972 return;
973
974 if (const ConstantArrayType *CAT =
975 CGM.getContext().getAsConstantArrayType(T)) {
976 QualType ElementTy = CAT->getElementType();
977 uint64_t ElementSize = CGM.getContext().getTypeSize(ElementTy);
978
979 for (uint64_t I = 0, E = CAT->getSize().getZExtValue(); I != E; ++I) {
980 FillInNullDataMemberPointers(CGM, ElementTy, Elements,
981 StartOffset + I * ElementSize);
982 }
983 } else if (const RecordType *RT = T->getAs<RecordType>()) {
984 const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
985 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
986
987 // Go through all bases and fill in any null pointer to data members.
988 for (CXXRecordDecl::base_class_const_iterator I = RD->bases_begin(),
989 E = RD->bases_end(); I != E; ++I) {
Anders Carlssonda265b82010-05-27 18:51:01 +0000990 if (I->isVirtual()) {
Anders Carlsson849ea412010-11-22 18:42:14 +0000991 // Ignore virtual bases.
Anders Carlssonda265b82010-05-27 18:51:01 +0000992 continue;
993 }
Anders Carlssonbe48c542010-05-18 16:51:41 +0000994
995 const CXXRecordDecl *BaseDecl =
996 cast<CXXRecordDecl>(I->getType()->getAs<RecordType>()->getDecl());
997
998 // Ignore empty bases.
999 if (BaseDecl->isEmpty())
1000 continue;
1001
1002 // Ignore bases that don't have any pointer to data members.
John McCall614dbdc2010-08-22 21:01:12 +00001003 if (CGM.getTypes().isZeroInitializable(BaseDecl))
Anders Carlssonbe48c542010-05-18 16:51:41 +00001004 continue;
1005
Anders Carlssonfd88a612010-10-31 23:22:37 +00001006 uint64_t BaseOffset = Layout.getBaseClassOffsetInBits(BaseDecl);
Anders Carlssonbe48c542010-05-18 16:51:41 +00001007 FillInNullDataMemberPointers(CGM, I->getType(),
1008 Elements, StartOffset + BaseOffset);
1009 }
1010
1011 // Visit all fields.
1012 unsigned FieldNo = 0;
1013 for (RecordDecl::field_iterator I = RD->field_begin(),
1014 E = RD->field_end(); I != E; ++I, ++FieldNo) {
1015 QualType FieldType = I->getType();
1016
John McCall614dbdc2010-08-22 21:01:12 +00001017 if (CGM.getTypes().isZeroInitializable(FieldType))
Anders Carlssonbe48c542010-05-18 16:51:41 +00001018 continue;
1019
1020 uint64_t FieldOffset = StartOffset + Layout.getFieldOffset(FieldNo);
1021 FillInNullDataMemberPointers(CGM, FieldType, Elements, FieldOffset);
1022 }
1023 } else {
1024 assert(T->isMemberPointerType() && "Should only see member pointers here!");
1025 assert(!T->getAs<MemberPointerType>()->getPointeeType()->isFunctionType() &&
1026 "Should only see pointers to data members here!");
1027
1028 uint64_t StartIndex = StartOffset / 8;
1029 uint64_t EndIndex = StartIndex + CGM.getContext().getTypeSize(T) / 8;
1030
John McCallf3a88602011-02-03 08:15:49 +00001031 // FIXME: hardcodes Itanium member pointer representation!
Anders Carlssonbe48c542010-05-18 16:51:41 +00001032 llvm::Constant *NegativeOne =
1033 llvm::ConstantInt::get(llvm::Type::getInt8Ty(CGM.getLLVMContext()),
John McCallf3a88602011-02-03 08:15:49 +00001034 -1ULL, /*isSigned*/true);
Anders Carlssonbe48c542010-05-18 16:51:41 +00001035
1036 // Fill in the null data member pointer.
1037 for (uint64_t I = StartIndex; I != EndIndex; ++I)
1038 Elements[I] = NegativeOne;
1039 }
1040}
1041
John McCall0217dfc22011-02-15 06:40:56 +00001042static llvm::Constant *EmitNullConstantForBase(CodeGenModule &CGM,
1043 const llvm::Type *baseType,
1044 const CXXRecordDecl *base);
1045
Anders Carlsson849ea412010-11-22 18:42:14 +00001046static llvm::Constant *EmitNullConstant(CodeGenModule &CGM,
John McCall0217dfc22011-02-15 06:40:56 +00001047 const CXXRecordDecl *record,
1048 bool asCompleteObject) {
1049 const CGRecordLayout &layout = CGM.getTypes().getCGRecordLayout(record);
1050 const llvm::StructType *structure =
1051 (asCompleteObject ? layout.getLLVMType()
1052 : layout.getBaseSubobjectLLVMType());
Anders Carlsson849ea412010-11-22 18:42:14 +00001053
John McCall0217dfc22011-02-15 06:40:56 +00001054 unsigned numElements = structure->getNumElements();
1055 std::vector<llvm::Constant *> elements(numElements);
Anders Carlsson849ea412010-11-22 18:42:14 +00001056
John McCall0217dfc22011-02-15 06:40:56 +00001057 // Fill in all the bases.
1058 for (CXXRecordDecl::base_class_const_iterator
1059 I = record->bases_begin(), E = record->bases_end(); I != E; ++I) {
Anders Carlsson849ea412010-11-22 18:42:14 +00001060 if (I->isVirtual()) {
John McCall0217dfc22011-02-15 06:40:56 +00001061 // Ignore virtual bases; if we're laying out for a complete
1062 // object, we'll lay these out later.
Anders Carlsson849ea412010-11-22 18:42:14 +00001063 continue;
1064 }
1065
John McCall0217dfc22011-02-15 06:40:56 +00001066 const CXXRecordDecl *base =
1067 cast<CXXRecordDecl>(I->getType()->castAs<RecordType>()->getDecl());
Anders Carlsson849ea412010-11-22 18:42:14 +00001068
1069 // Ignore empty bases.
John McCall0217dfc22011-02-15 06:40:56 +00001070 if (base->isEmpty())
Anders Carlsson849ea412010-11-22 18:42:14 +00001071 continue;
1072
John McCall0217dfc22011-02-15 06:40:56 +00001073 unsigned fieldIndex = layout.getNonVirtualBaseLLVMFieldNo(base);
1074 const llvm::Type *baseType = structure->getElementType(fieldIndex);
1075 elements[fieldIndex] = EmitNullConstantForBase(CGM, baseType, base);
Anders Carlsson849ea412010-11-22 18:42:14 +00001076 }
1077
John McCall0217dfc22011-02-15 06:40:56 +00001078 // Fill in all the fields.
1079 for (RecordDecl::field_iterator I = record->field_begin(),
1080 E = record->field_end(); I != E; ++I) {
1081 const FieldDecl *field = *I;
Anders Carlsson849ea412010-11-22 18:42:14 +00001082
1083 // Ignore bit fields.
John McCall0217dfc22011-02-15 06:40:56 +00001084 if (field->isBitField())
Anders Carlsson849ea412010-11-22 18:42:14 +00001085 continue;
1086
John McCall0217dfc22011-02-15 06:40:56 +00001087 unsigned fieldIndex = layout.getLLVMFieldNo(field);
1088 elements[fieldIndex] = CGM.EmitNullConstant(field->getType());
1089 }
1090
1091 // Fill in the virtual bases, if we're working with the complete object.
1092 if (asCompleteObject) {
1093 for (CXXRecordDecl::base_class_const_iterator
1094 I = record->vbases_begin(), E = record->vbases_end(); I != E; ++I) {
1095 const CXXRecordDecl *base =
1096 cast<CXXRecordDecl>(I->getType()->castAs<RecordType>()->getDecl());
1097
1098 // Ignore empty bases.
1099 if (base->isEmpty())
1100 continue;
1101
1102 unsigned fieldIndex = layout.getVirtualBaseIndex(base);
1103
1104 // We might have already laid this field out.
1105 if (elements[fieldIndex]) continue;
1106
1107 const llvm::Type *baseType = structure->getElementType(fieldIndex);
1108 elements[fieldIndex] = EmitNullConstantForBase(CGM, baseType, base);
1109 }
Anders Carlsson849ea412010-11-22 18:42:14 +00001110 }
1111
1112 // Now go through all other fields and zero them out.
John McCall0217dfc22011-02-15 06:40:56 +00001113 for (unsigned i = 0; i != numElements; ++i) {
1114 if (!elements[i])
1115 elements[i] = llvm::Constant::getNullValue(structure->getElementType(i));
Anders Carlsson849ea412010-11-22 18:42:14 +00001116 }
1117
John McCall0217dfc22011-02-15 06:40:56 +00001118 return llvm::ConstantStruct::get(structure, elements);
1119}
1120
1121/// Emit the null constant for a base subobject.
1122static llvm::Constant *EmitNullConstantForBase(CodeGenModule &CGM,
1123 const llvm::Type *baseType,
1124 const CXXRecordDecl *base) {
1125 const CGRecordLayout &baseLayout = CGM.getTypes().getCGRecordLayout(base);
1126
1127 // Just zero out bases that don't have any pointer to data members.
1128 if (baseLayout.isZeroInitializableAsBase())
1129 return llvm::Constant::getNullValue(baseType);
1130
1131 // If the base type is a struct, we can just use its null constant.
1132 if (isa<llvm::StructType>(baseType)) {
1133 return EmitNullConstant(CGM, base, /*complete*/ false);
1134 }
1135
1136 // Otherwise, some bases are represented as arrays of i8 if the size
1137 // of the base is smaller than its corresponding LLVM type. Figure
1138 // out how many elements this base array has.
1139 const llvm::ArrayType *baseArrayType = cast<llvm::ArrayType>(baseType);
1140 unsigned numBaseElements = baseArrayType->getNumElements();
1141
1142 // Fill in null data member pointers.
1143 std::vector<llvm::Constant *> baseElements(numBaseElements);
1144 FillInNullDataMemberPointers(CGM, CGM.getContext().getTypeDeclType(base),
1145 baseElements, 0);
1146
1147 // Now go through all other elements and zero them out.
1148 if (numBaseElements) {
1149 const llvm::Type *i8 = llvm::Type::getInt8Ty(CGM.getLLVMContext());
1150 llvm::Constant *i8_zero = llvm::Constant::getNullValue(i8);
1151 for (unsigned i = 0; i != numBaseElements; ++i) {
1152 if (!baseElements[i])
1153 baseElements[i] = i8_zero;
1154 }
1155 }
1156
1157 return llvm::ConstantArray::get(baseArrayType, baseElements);
Anders Carlsson849ea412010-11-22 18:42:14 +00001158}
1159
Eli Friedman20bb5e02009-04-13 21:47:26 +00001160llvm::Constant *CodeGenModule::EmitNullConstant(QualType T) {
John McCall614dbdc2010-08-22 21:01:12 +00001161 if (getTypes().isZeroInitializable(T))
Anders Carlsson867b48f2009-08-24 17:16:23 +00001162 return llvm::Constant::getNullValue(getTypes().ConvertTypeForMem(T));
Anders Carlssone8bfe412010-02-02 05:17:25 +00001163
Anders Carlssonf48123b2009-08-09 18:26:27 +00001164 if (const ConstantArrayType *CAT = Context.getAsConstantArrayType(T)) {
Mike Stump11289f42009-09-09 15:08:12 +00001165
Anders Carlssonf48123b2009-08-09 18:26:27 +00001166 QualType ElementTy = CAT->getElementType();
1167
Anders Carlssone8bfe412010-02-02 05:17:25 +00001168 llvm::Constant *Element = EmitNullConstant(ElementTy);
1169 unsigned NumElements = CAT->getSize().getZExtValue();
1170 std::vector<llvm::Constant *> Array(NumElements);
1171 for (unsigned i = 0; i != NumElements; ++i)
1172 Array[i] = Element;
Mike Stump11289f42009-09-09 15:08:12 +00001173
Anders Carlssone8bfe412010-02-02 05:17:25 +00001174 const llvm::ArrayType *ATy =
1175 cast<llvm::ArrayType>(getTypes().ConvertTypeForMem(T));
1176 return llvm::ConstantArray::get(ATy, Array);
Anders Carlssonf48123b2009-08-09 18:26:27 +00001177 }
Anders Carlssond606de72009-08-23 01:25:01 +00001178
1179 if (const RecordType *RT = T->getAs<RecordType>()) {
Anders Carlssone8bfe412010-02-02 05:17:25 +00001180 const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
John McCall0217dfc22011-02-15 06:40:56 +00001181 return ::EmitNullConstant(*this, RD, /*complete object*/ true);
Anders Carlssond606de72009-08-23 01:25:01 +00001182 }
Mike Stump11289f42009-09-09 15:08:12 +00001183
Anders Carlssonbe48c542010-05-18 16:51:41 +00001184 assert(T->isMemberPointerType() && "Should only see member pointers here!");
Anders Carlssone8bfe412010-02-02 05:17:25 +00001185 assert(!T->getAs<MemberPointerType>()->getPointeeType()->isFunctionType() &&
1186 "Should only see pointers to data members here!");
1187
1188 // Itanium C++ ABI 2.3:
1189 // A NULL pointer is represented as -1.
John McCallf3a88602011-02-03 08:15:49 +00001190 return getCXXABI().EmitNullMemberPointer(T->castAs<MemberPointerType>());
Eli Friedman20bb5e02009-04-13 21:47:26 +00001191}