blob: cdc8bdb97013c70d13f5509421cc8750a6b6fa38 [file] [log] [blame]
Anders Carlssond76cead2008-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"
Daniel Dunbar84bb85f2008-08-13 00:59:25 +000016#include "CGObjCRuntime.h"
Chris Lattner8ce618c2008-10-06 05:59:01 +000017#include "clang/AST/APValue.h"
Daniel Dunbareee5cd12008-08-11 05:00:27 +000018#include "clang/AST/ASTContext.h"
Anders Carlssondbce63a2009-07-24 15:20:52 +000019#include "clang/AST/RecordLayout.h"
Daniel Dunbareee5cd12008-08-11 05:00:27 +000020#include "clang/AST/StmtVisitor.h"
Chris Lattnerc46fcdd2009-06-14 01:54:56 +000021#include "clang/Basic/Builtins.h"
Anders Carlssond76cead2008-01-26 01:36:00 +000022#include "llvm/Constants.h"
23#include "llvm/Function.h"
24#include "llvm/GlobalVariable.h"
25#include "llvm/Support/Compiler.h"
Eli Friedman7529fd62008-05-30 10:24:46 +000026#include "llvm/Target/TargetData.h"
Anders Carlssond76cead2008-01-26 01:36:00 +000027using namespace clang;
28using namespace CodeGen;
29
30namespace {
Anders Carlssondbce63a2009-07-24 15:20:52 +000031
32class VISIBILITY_HIDDEN ConstStructBuilder {
33 CodeGenModule &CGM;
34 CodeGenFunction *CGF;
35
36 bool Packed;
Anders Carlssonc48f0232009-07-25 00:24:56 +000037
Anders Carlssondbce63a2009-07-24 15:20:52 +000038 unsigned NextFieldOffsetInBytes;
39
40 std::vector<llvm::Constant *> Elements;
41
42 ConstStructBuilder(CodeGenModule &CGM, CodeGenFunction *CGF)
43 : CGM(CGM), CGF(CGF), Packed(false), NextFieldOffsetInBytes(0) { }
44
45 bool AppendField(const FieldDecl *Field, uint64_t FieldOffset,
46 const Expr *InitExpr) {
47 uint64_t FieldOffsetInBytes = FieldOffset / 8;
48
49 assert(NextFieldOffsetInBytes <= FieldOffsetInBytes
50 && "Field offset mismatch!");
51
52 // Emit the field.
53 llvm::Constant *C = CGM.EmitConstantExpr(InitExpr, Field->getType(), CGF);
54 if (!C)
55 return false;
56
57 unsigned FieldAlignment = getAlignment(C);
58
59 // Round up the field offset to the alignment of the field type.
60 uint64_t AlignedNextFieldOffsetInBytes =
61 llvm::RoundUpToAlignment(NextFieldOffsetInBytes, FieldAlignment);
62
63 if (AlignedNextFieldOffsetInBytes > FieldOffsetInBytes) {
Anders Carlssonaf3b9372009-07-25 01:40:29 +000064 std::vector<llvm::Constant *> PackedElements;
65
66 assert(!Packed && "Alignment is wrong even with a packed struct!");
67
68 // Convert the struct to a packed struct.
69 uint64_t ElementOffsetInBytes = 0;
70
71 for (unsigned i = 0, e = Elements.size(); i != e; ++i) {
72 llvm::Constant *C = Elements[i];
73
74 unsigned ElementAlign =
75 CGM.getTargetData().getABITypeAlignment(C->getType());
76 uint64_t AlignedElementOffsetInBytes =
77 llvm::RoundUpToAlignment(ElementOffsetInBytes, ElementAlign);
78
79 if (AlignedElementOffsetInBytes > ElementOffsetInBytes) {
80 // We need some padding.
81 uint64_t NumBytes =
82 AlignedElementOffsetInBytes - ElementOffsetInBytes;
83
84 const llvm::Type *Ty = llvm::Type::Int8Ty;
85 if (NumBytes > 1)
Owen Anderson7ec2d8f2009-07-29 22:16:19 +000086 Ty = llvm::ArrayType::get(Ty, NumBytes);
Anders Carlssonaf3b9372009-07-25 01:40:29 +000087
Owen Andersonf37b84b2009-07-31 20:28:54 +000088 llvm::Constant *Padding = llvm::Constant::getNullValue(Ty);
Anders Carlssonaf3b9372009-07-25 01:40:29 +000089 PackedElements.push_back(Padding);
90 ElementOffsetInBytes += getSizeInBytes(Padding);
91 }
92
93 PackedElements.push_back(C);
94 ElementOffsetInBytes += getSizeInBytes(C);
95 }
96
97 assert(ElementOffsetInBytes == NextFieldOffsetInBytes &&
98 "Packing the struct changed its size!");
99
100 Elements = PackedElements;
101 Packed = true;
102 AlignedNextFieldOffsetInBytes = NextFieldOffsetInBytes;
Anders Carlssondbce63a2009-07-24 15:20:52 +0000103 }
104
105 if (AlignedNextFieldOffsetInBytes < FieldOffsetInBytes) {
106 // We need to append padding.
Anders Carlssonaf3b9372009-07-25 01:40:29 +0000107 AppendPadding(FieldOffsetInBytes - NextFieldOffsetInBytes);
Anders Carlssondbce63a2009-07-24 15:20:52 +0000108
109 assert(NextFieldOffsetInBytes == FieldOffsetInBytes &&
110 "Did not add enough padding!");
111
112 AlignedNextFieldOffsetInBytes = NextFieldOffsetInBytes;
113 }
114
115 // Add the field.
116 Elements.push_back(C);
117 NextFieldOffsetInBytes = AlignedNextFieldOffsetInBytes + getSizeInBytes(C);
118
119 return true;
120 }
121
Anders Carlssona02da3f2009-07-27 01:23:51 +0000122 bool AppendBitField(const FieldDecl *Field, uint64_t FieldOffset,
123 const Expr *InitExpr) {
124 llvm::ConstantInt *CI =
125 cast_or_null<llvm::ConstantInt>(CGM.EmitConstantExpr(InitExpr,
126 Field->getType(),
127 CGF));
128 // FIXME: Can this ever happen?
129 if (!CI)
130 return false;
131
132 if (FieldOffset > NextFieldOffsetInBytes * 8) {
Anders Carlssona9715d12009-07-27 02:56:37 +0000133 // We need to add padding.
134 uint64_t NumBytes =
135 llvm::RoundUpToAlignment(FieldOffset -
136 NextFieldOffsetInBytes * 8, 8) / 8;
137
138 AppendPadding(NumBytes);
Anders Carlssona02da3f2009-07-27 01:23:51 +0000139 }
140
141 uint64_t FieldSize =
142 Field->getBitWidth()->EvaluateAsInt(CGM.getContext()).getZExtValue();
143
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())
151 FieldValue.zext(FieldSize);
152
153 // Truncate the size of FieldValue to the bit field size.
154 if (FieldSize < FieldValue.getBitWidth())
155 FieldValue.trunc(FieldSize);
156
157 if (FieldOffset < NextFieldOffsetInBytes * 8) {
Anders Carlssona9715d12009-07-27 02:56:37 +0000158 // 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);
175 Tmp.trunc(BitsInPreviousByte);
176
177 // We want the remaining high bits.
178 FieldValue.trunc(NewFieldWidth);
179 } else {
180 Tmp.trunc(BitsInPreviousByte);
181
182 // We want the remaining low bits.
183 FieldValue = FieldValue.lshr(BitsInPreviousByte);
184 FieldValue.trunc(NewFieldWidth);
185 }
186 }
187
188 Tmp.zext(8);
189 if (CGM.getTargetData().isBigEndian()) {
190 if (FitsCompletelyInPreviousByte)
191 Tmp = Tmp.shl(BitsInPreviousByte - FieldValue.getBitWidth());
192 } else {
193 Tmp = Tmp.shl(8 - BitsInPreviousByte);
194 }
195
196 // Or in the bits that go into the previous byte.
197 Tmp |= cast<llvm::ConstantInt>(Elements.back())->getValue();
198 Elements.back() = llvm::ConstantInt::get(CGM.getLLVMContext(), Tmp);
199
200 if (FitsCompletelyInPreviousByte)
201 return true;
Anders Carlssona02da3f2009-07-27 01:23:51 +0000202 }
203
204 while (FieldValue.getBitWidth() > 8) {
205 llvm::APInt Tmp;
206
207 if (CGM.getTargetData().isBigEndian()) {
Anders Carlssona02da3f2009-07-27 01:23:51 +0000208 // We want the high bits.
Anders Carlssona9715d12009-07-27 02:56:37 +0000209 Tmp = FieldValue;
210 Tmp = Tmp.lshr(Tmp.getBitWidth() - 8);
211 Tmp.trunc(8);
212 } else {
213 // We want the low bits.
214 Tmp = FieldValue;
215 Tmp.trunc(8);
216
217 FieldValue = FieldValue.lshr(8);
Anders Carlssona02da3f2009-07-27 01:23:51 +0000218 }
219
Anders Carlssona02da3f2009-07-27 01:23:51 +0000220 Elements.push_back(llvm::ConstantInt::get(CGM.getLLVMContext(), Tmp));
221 NextFieldOffsetInBytes++;
222
223 FieldValue.trunc(FieldValue.getBitWidth() - 8);
224 }
225
226 assert(FieldValue.getBitWidth() > 0 &&
227 "Should have at least one bit left!");
228 assert(FieldValue.getBitWidth() <= 8 &&
229 "Should not have more than a byte left!");
230
231 if (FieldValue.getBitWidth() < 8) {
232 if (CGM.getTargetData().isBigEndian()) {
233 unsigned BitWidth = FieldValue.getBitWidth();
234
235 FieldValue.zext(8);
236 FieldValue = FieldValue << (8 - BitWidth);
237 } else
238 FieldValue.zext(8);
239 }
240
241 // Append the last element.
242 Elements.push_back(llvm::ConstantInt::get(CGM.getLLVMContext(),
243 FieldValue));
244 NextFieldOffsetInBytes++;
245 return true;
246 }
247
Anders Carlssondbce63a2009-07-24 15:20:52 +0000248 void AppendPadding(uint64_t NumBytes) {
249 if (!NumBytes)
250 return;
251
252 const llvm::Type *Ty = llvm::Type::Int8Ty;
253 if (NumBytes > 1)
Owen Anderson7ec2d8f2009-07-29 22:16:19 +0000254 Ty = llvm::ArrayType::get(Ty, NumBytes);
Anders Carlssondbce63a2009-07-24 15:20:52 +0000255
Owen Andersonf37b84b2009-07-31 20:28:54 +0000256 llvm::Constant *C = llvm::Constant::getNullValue(Ty);
Anders Carlssondbce63a2009-07-24 15:20:52 +0000257 Elements.push_back(C);
258 assert(getAlignment(C) == 1 && "Padding must have 1 byte alignment!");
259
260 NextFieldOffsetInBytes += getSizeInBytes(C);
261 }
262
263 void AppendTailPadding(uint64_t RecordSize) {
264 assert(RecordSize % 8 == 0 && "Invalid record size!");
265
266 uint64_t RecordSizeInBytes = RecordSize / 8;
267 assert(NextFieldOffsetInBytes <= RecordSizeInBytes && "Size mismatch!");
268
269 unsigned NumPadBytes = RecordSizeInBytes - NextFieldOffsetInBytes;
270 AppendPadding(NumPadBytes);
271 }
272
273 bool Build(const InitListExpr *ILE) {
Ted Kremenekd00cd9e2009-07-29 21:53:49 +0000274 RecordDecl *RD = ILE->getType()->getAs<RecordType>()->getDecl();
Anders Carlssondbce63a2009-07-24 15:20:52 +0000275 const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
276
277 unsigned FieldNo = 0;
278 unsigned ElementNo = 0;
279 for (RecordDecl::field_iterator Field = RD->field_begin(),
280 FieldEnd = RD->field_end();
281 ElementNo < ILE->getNumInits() && Field != FieldEnd;
282 ++Field, ++FieldNo) {
283 if (Field->isBitField()) {
284 if (!Field->getIdentifier())
285 continue;
286
Anders Carlssona02da3f2009-07-27 01:23:51 +0000287 if (!AppendBitField(*Field, Layout.getFieldOffset(FieldNo),
288 ILE->getInit(ElementNo)))
289 return false;
Anders Carlssondbce63a2009-07-24 15:20:52 +0000290 } else {
291 if (!AppendField(*Field, Layout.getFieldOffset(FieldNo),
292 ILE->getInit(ElementNo)))
293 return false;
294 }
295
296 ElementNo++;
297 }
298
Anders Carlssonc48f0232009-07-25 00:24:56 +0000299 uint64_t LayoutSizeInBytes = Layout.getSize() / 8;
300
301 if (NextFieldOffsetInBytes > LayoutSizeInBytes) {
302 // If the struct is bigger than the size of the record type,
303 // we must have a flexible array member at the end.
304 assert(RD->hasFlexibleArrayMember() &&
305 "Must have flexible array member if struct is bigger than type!");
306
307 // No tail padding is necessary.
308 return true;
309 }
310
Anders Carlssondbce63a2009-07-24 15:20:52 +0000311 // Append tail padding if necessary.
312 AppendTailPadding(Layout.getSize());
Anders Carlssonc48f0232009-07-25 00:24:56 +0000313
Anders Carlssondbce63a2009-07-24 15:20:52 +0000314 assert(Layout.getSize() / 8 == NextFieldOffsetInBytes &&
315 "Tail padding mismatch!");
Anders Carlssonc48f0232009-07-25 00:24:56 +0000316
Anders Carlssondbce63a2009-07-24 15:20:52 +0000317 return true;
318 }
319
320 unsigned getAlignment(const llvm::Constant *C) const {
321 if (Packed)
322 return 1;
323
324 return CGM.getTargetData().getABITypeAlignment(C->getType());
325 }
326
327 uint64_t getSizeInBytes(const llvm::Constant *C) const {
328 return CGM.getTargetData().getTypeAllocSize(C->getType());
329 }
330
331public:
332 static llvm::Constant *BuildStruct(CodeGenModule &CGM, CodeGenFunction *CGF,
333 const InitListExpr *ILE) {
334 ConstStructBuilder Builder(CGM, CGF);
335
Anders Carlssondbce63a2009-07-24 15:20:52 +0000336 if (!Builder.Build(ILE))
337 return 0;
338
339 llvm::Constant *Result =
Owen Anderson5ccd63c2009-07-27 22:29:56 +0000340 llvm::ConstantStruct::get(Builder.Elements, Builder.Packed);
Anders Carlssondbce63a2009-07-24 15:20:52 +0000341
Anders Carlssonc48f0232009-07-25 00:24:56 +0000342 assert(llvm::RoundUpToAlignment(Builder.NextFieldOffsetInBytes,
343 Builder.getAlignment(Result)) ==
344 Builder.getSizeInBytes(Result) && "Size mismatch!");
Anders Carlssondbce63a2009-07-24 15:20:52 +0000345
Anders Carlsson78b54302009-07-24 18:20:38 +0000346 return Result;
Anders Carlssondbce63a2009-07-24 15:20:52 +0000347 }
348};
349
Anders Carlssonde598292008-01-26 04:30:23 +0000350class VISIBILITY_HIDDEN ConstExprEmitter :
351 public StmtVisitor<ConstExprEmitter, llvm::Constant*> {
Anders Carlssond76cead2008-01-26 01:36:00 +0000352 CodeGenModule &CGM;
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +0000353 CodeGenFunction *CGF;
Owen Anderson73e7f802009-07-14 23:10:40 +0000354 llvm::LLVMContext &VMContext;
Anders Carlssond76cead2008-01-26 01:36:00 +0000355public:
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +0000356 ConstExprEmitter(CodeGenModule &cgm, CodeGenFunction *cgf)
Owen Anderson73e7f802009-07-14 23:10:40 +0000357 : CGM(cgm), CGF(cgf), VMContext(cgm.getLLVMContext()) {
Anders Carlssond76cead2008-01-26 01:36:00 +0000358 }
359
360 //===--------------------------------------------------------------------===//
361 // Visitor Methods
362 //===--------------------------------------------------------------------===//
363
364 llvm::Constant *VisitStmt(Stmt *S) {
Anders Carlsson452fcbc2009-03-03 16:43:34 +0000365 return 0;
Anders Carlssond76cead2008-01-26 01:36:00 +0000366 }
367
368 llvm::Constant *VisitParenExpr(ParenExpr *PE) {
369 return Visit(PE->getSubExpr());
370 }
Anders Carlsson2502a382008-11-22 02:34:39 +0000371
Anders Carlssond76cead2008-01-26 01:36:00 +0000372 llvm::Constant *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
373 return Visit(E->getInitializer());
374 }
375
Argiris Kirtzidisc45e2fb2008-08-18 23:01:59 +0000376 llvm::Constant *VisitCastExpr(CastExpr* E) {
Nuno Lopes1a835322009-01-17 00:48:48 +0000377 // GCC cast to union extension
378 if (E->getType()->isUnionType()) {
379 const llvm::Type *Ty = ConvertType(E->getType());
Anders Carlssone7d97c22009-04-08 04:48:15 +0000380 Expr *SubExpr = E->getSubExpr();
Anders Carlsson9c919ef2009-07-31 21:38:39 +0000381
382 llvm::Constant *C =
383 CGM.EmitConstantExpr(SubExpr, SubExpr->getType(), CGF);
384 if (!C)
385 return 0;
386
387 // Build a struct with the union sub-element as the first member,
388 // and padded to the appropriate size
389 std::vector<llvm::Constant*> Elts;
390 std::vector<const llvm::Type*> Types;
391 Elts.push_back(C);
392 Types.push_back(C->getType());
393 unsigned CurSize = CGM.getTargetData().getTypeAllocSize(C->getType());
394 unsigned TotalSize = CGM.getTargetData().getTypeAllocSize(Ty);
395
396 assert(CurSize <= TotalSize && "Union size mismatch!");
397 if (unsigned NumPadBytes = TotalSize - CurSize) {
398 const llvm::Type *Ty = llvm::Type::Int8Ty;
399 if (NumPadBytes > 1)
400 Ty = llvm::ArrayType::get(Ty, NumPadBytes);
401
402 Elts.push_back(llvm::Constant::getNullValue(Ty));
403 Types.push_back(Ty);
404 }
405
406 llvm::StructType* STy = llvm::StructType::get(Types, false);
407 return llvm::ConstantStruct::get(STy, Elts);
Nuno Lopes1a835322009-01-17 00:48:48 +0000408 }
Anders Carlsson9c919ef2009-07-31 21:38:39 +0000409
Eli Friedman0737c0c2009-04-30 07:03:22 +0000410 // Explicit and implicit no-op casts
411 QualType Ty = E->getType(), SubTy = E->getSubExpr()->getType();
412 if (CGM.getContext().hasSameUnqualifiedType(Ty, SubTy)) {
Eli Friedmane31e02d2009-02-22 07:29:04 +0000413 return Visit(E->getSubExpr());
414 }
415 return 0;
Anders Carlssond76cead2008-01-26 01:36:00 +0000416 }
Devang Patel04520012008-02-05 02:39:50 +0000417
Chris Lattner3e254fb2008-04-08 04:40:51 +0000418 llvm::Constant *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
419 return Visit(DAE->getExpr());
420 }
421
Eli Friedman7529fd62008-05-30 10:24:46 +0000422 llvm::Constant *EmitArrayInitialization(InitListExpr *ILE) {
Eli Friedman8ac5b3a2008-02-21 17:45:41 +0000423 std::vector<llvm::Constant*> Elts;
Eli Friedman7529fd62008-05-30 10:24:46 +0000424 const llvm::ArrayType *AType =
425 cast<llvm::ArrayType>(ConvertType(ILE->getType()));
Eli Friedman8ac5b3a2008-02-21 17:45:41 +0000426 unsigned NumInitElements = ILE->getNumInits();
427 // FIXME: Check for wide strings
Chris Lattnerc5d32632009-02-24 22:18:39 +0000428 // FIXME: Check for NumInitElements exactly equal to 1??
429 if (NumInitElements > 0 &&
430 (isa<StringLiteral>(ILE->getInit(0)) ||
431 isa<ObjCEncodeExpr>(ILE->getInit(0))) &&
Chris Lattnera1923f62008-08-04 07:31:14 +0000432 ILE->getType()->getArrayElementTypeNoTypeQual()->isCharType())
Eli Friedman8ac5b3a2008-02-21 17:45:41 +0000433 return Visit(ILE->getInit(0));
Devang Patel04520012008-02-05 02:39:50 +0000434 const llvm::Type *ElemTy = AType->getElementType();
435 unsigned NumElements = AType->getNumElements();
436
437 // Initialising an array requires us to automatically
438 // initialise any elements that have not been initialised explicitly
439 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
440
441 // Copy initializer elements.
442 unsigned i = 0;
Eli Friedman42fab002008-05-30 19:58:50 +0000443 bool RewriteType = false;
Devang Patel04520012008-02-05 02:39:50 +0000444 for (; i < NumInitableElts; ++i) {
Anders Carlssone7d97c22009-04-08 04:48:15 +0000445 Expr *Init = ILE->getInit(i);
446 llvm::Constant *C = CGM.EmitConstantExpr(Init, Init->getType(), CGF);
Daniel Dunbarf69d19e2009-02-17 18:43:32 +0000447 if (!C)
448 return 0;
Eli Friedman42fab002008-05-30 19:58:50 +0000449 RewriteType |= (C->getType() != ElemTy);
Devang Patel04520012008-02-05 02:39:50 +0000450 Elts.push_back(C);
451 }
Eli Friedman42fab002008-05-30 19:58:50 +0000452
Devang Patel04520012008-02-05 02:39:50 +0000453 // Initialize remaining array elements.
Eli Friedmanb836cd32009-04-13 21:47:26 +0000454 // FIXME: This doesn't handle member pointers correctly!
Devang Patel04520012008-02-05 02:39:50 +0000455 for (; i < NumElements; ++i)
Owen Andersonf37b84b2009-07-31 20:28:54 +0000456 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
Devang Patel04520012008-02-05 02:39:50 +0000457
Eli Friedman42fab002008-05-30 19:58:50 +0000458 if (RewriteType) {
459 // FIXME: Try to avoid packing the array
460 std::vector<const llvm::Type*> Types;
461 for (unsigned i = 0; i < Elts.size(); ++i)
462 Types.push_back(Elts[i]->getType());
Owen Anderson7ec2d8f2009-07-29 22:16:19 +0000463 const llvm::StructType *SType = llvm::StructType::get(Types, true);
Owen Anderson5ccd63c2009-07-27 22:29:56 +0000464 return llvm::ConstantStruct::get(SType, Elts);
Eli Friedman42fab002008-05-30 19:58:50 +0000465 }
466
Owen Anderson8472b492009-07-28 18:33:04 +0000467 return llvm::ConstantArray::get(AType, Elts);
Devang Patel04520012008-02-05 02:39:50 +0000468 }
469
Eli Friedman7529fd62008-05-30 10:24:46 +0000470 llvm::Constant *EmitStructInitialization(InitListExpr *ILE) {
Anders Carlssonfaab0302009-07-27 05:54:15 +0000471 return ConstStructBuilder::BuildStruct(CGM, CGF, ILE);
Devang Patel04520012008-02-05 02:39:50 +0000472 }
Anders Carlsson69896042009-07-31 21:48:56 +0000473
Eli Friedman7529fd62008-05-30 10:24:46 +0000474 llvm::Constant *EmitUnionInitialization(InitListExpr *ILE) {
Anders Carlssonaacac602009-07-31 21:34:04 +0000475 return ConstStructBuilder::BuildStruct(CGM, CGF, ILE);
Eli Friedman7529fd62008-05-30 10:24:46 +0000476 }
477
478 llvm::Constant *EmitVectorInitialization(InitListExpr *ILE) {
479 const llvm::VectorType *VType =
480 cast<llvm::VectorType>(ConvertType(ILE->getType()));
Eli Friedman6c221712008-05-30 12:02:14 +0000481 const llvm::Type *ElemTy = VType->getElementType();
482 std::vector<llvm::Constant*> Elts;
Devang Patel04520012008-02-05 02:39:50 +0000483 unsigned NumElements = VType->getNumElements();
Eli Friedman6c221712008-05-30 12:02:14 +0000484 unsigned NumInitElements = ILE->getNumInits();
Devang Patel04520012008-02-05 02:39:50 +0000485
Eli Friedman6c221712008-05-30 12:02:14 +0000486 unsigned NumInitableElts = std::min(NumInitElements, NumElements);
487
Devang Patel04520012008-02-05 02:39:50 +0000488 // Copy initializer elements.
489 unsigned i = 0;
Eli Friedman6c221712008-05-30 12:02:14 +0000490 for (; i < NumInitableElts; ++i) {
Anders Carlssone7d97c22009-04-08 04:48:15 +0000491 Expr *Init = ILE->getInit(i);
492 llvm::Constant *C = CGM.EmitConstantExpr(Init, Init->getType(), CGF);
Daniel Dunbarf69d19e2009-02-17 18:43:32 +0000493 if (!C)
494 return 0;
Devang Patel04520012008-02-05 02:39:50 +0000495 Elts.push_back(C);
496 }
497
Eli Friedman6c221712008-05-30 12:02:14 +0000498 for (; i < NumElements; ++i)
Owen Andersonf37b84b2009-07-31 20:28:54 +0000499 Elts.push_back(llvm::Constant::getNullValue(ElemTy));
Eli Friedman6c221712008-05-30 12:02:14 +0000500
Owen Anderson17971fa2009-07-28 21:22:35 +0000501 return llvm::ConstantVector::get(VType, Elts);
Devang Patel04520012008-02-05 02:39:50 +0000502 }
Anders Carlsson1ddb85f2009-01-30 06:13:25 +0000503
504 llvm::Constant *VisitImplicitValueInitExpr(ImplicitValueInitExpr* E) {
Eli Friedmanb836cd32009-04-13 21:47:26 +0000505 return CGM.EmitNullConstant(E->getType());
Anders Carlsson1ddb85f2009-01-30 06:13:25 +0000506 }
507
Anders Carlssond76cead2008-01-26 01:36:00 +0000508 llvm::Constant *VisitInitListExpr(InitListExpr *ILE) {
Eli Friedman7529fd62008-05-30 10:24:46 +0000509 if (ILE->getType()->isScalarType()) {
510 // We have a scalar in braces. Just use the first element.
Anders Carlssone7d97c22009-04-08 04:48:15 +0000511 if (ILE->getNumInits() > 0) {
512 Expr *Init = ILE->getInit(0);
513 return CGM.EmitConstantExpr(Init, Init->getType(), CGF);
514 }
Eli Friedmanb836cd32009-04-13 21:47:26 +0000515 return CGM.EmitNullConstant(ILE->getType());
Anders Carlssond76cead2008-01-26 01:36:00 +0000516 }
Chris Lattnera9177982008-10-26 23:53:12 +0000517
Eli Friedman7529fd62008-05-30 10:24:46 +0000518 if (ILE->getType()->isArrayType())
519 return EmitArrayInitialization(ILE);
Devang Pateld72d1012008-01-29 23:23:18 +0000520
Eli Friedman7529fd62008-05-30 10:24:46 +0000521 if (ILE->getType()->isStructureType())
522 return EmitStructInitialization(ILE);
523
524 if (ILE->getType()->isUnionType())
525 return EmitUnionInitialization(ILE);
526
527 if (ILE->getType()->isVectorType())
528 return EmitVectorInitialization(ILE);
529
Devang Patel04520012008-02-05 02:39:50 +0000530 assert(0 && "Unable to handle InitListExpr");
Chris Lattnercbfb5512008-03-01 08:45:05 +0000531 // Get rid of control reaches end of void function warning.
532 // Not reached.
533 return 0;
Anders Carlssond76cead2008-01-26 01:36:00 +0000534 }
Eli Friedman1693a3f2008-02-21 17:57:49 +0000535
Anders Carlssond76cead2008-01-26 01:36:00 +0000536 llvm::Constant *VisitStringLiteral(StringLiteral *E) {
Chris Lattnera6dcce32008-02-11 00:02:17 +0000537 assert(!E->getType()->isPointerType() && "Strings are always arrays");
Anders Carlssond76cead2008-01-26 01:36:00 +0000538
Chris Lattnerc5d32632009-02-24 22:18:39 +0000539 // This must be a string initializing an array in a static initializer.
540 // Don't emit it as the address of the string, emit the string data itself
541 // as an inline array.
Owen Anderson8472b492009-07-28 18:33:04 +0000542 return llvm::ConstantArray::get(CGM.GetStringForStringLiteral(E), false);
Anders Carlssond76cead2008-01-26 01:36:00 +0000543 }
544
Chris Lattnerc5d32632009-02-24 22:18:39 +0000545 llvm::Constant *VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
546 // This must be an @encode initializing an array in a static initializer.
547 // Don't emit it as the address of the string, emit the string data itself
548 // as an inline array.
549 std::string Str;
550 CGM.getContext().getObjCEncodingForType(E->getEncodedType(), Str);
551 const ConstantArrayType *CAT = cast<ConstantArrayType>(E->getType());
552
553 // Resize the string to the right size, adding zeros at the end, or
554 // truncating as needed.
555 Str.resize(CAT->getSize().getZExtValue(), '\0');
Owen Anderson8472b492009-07-28 18:33:04 +0000556 return llvm::ConstantArray::get(Str, false);
Chris Lattnerc5d32632009-02-24 22:18:39 +0000557 }
558
Eli Friedmana36799c2008-05-29 11:22:45 +0000559 llvm::Constant *VisitUnaryExtension(const UnaryOperator *E) {
560 return Visit(E->getSubExpr());
561 }
Mike Stumpae93d652009-02-19 22:01:56 +0000562
Anders Carlssond76cead2008-01-26 01:36:00 +0000563 // Utility methods
564 const llvm::Type *ConvertType(QualType T) {
565 return CGM.getTypes().ConvertType(T);
566 }
Anders Carlssonde598292008-01-26 04:30:23 +0000567
Anders Carlsson85b43812008-11-16 06:23:45 +0000568public:
Anders Carlssonde598292008-01-26 04:30:23 +0000569 llvm::Constant *EmitLValue(Expr *E) {
Anders Carlsson00b837c2008-01-26 02:08:50 +0000570 switch (E->getStmtClass()) {
Eli Friedmane493ea92008-02-11 00:23:10 +0000571 default: break;
Anders Carlsson00b837c2008-01-26 02:08:50 +0000572 case Expr::CompoundLiteralExprClass: {
573 // Note that due to the nature of compound literals, this is guaranteed
574 // to be the only use of the variable, so we just generate it here.
Anders Carlssonde598292008-01-26 04:30:23 +0000575 CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
576 llvm::Constant* C = Visit(CLE->getInitializer());
Daniel Dunbarf69d19e2009-02-17 18:43:32 +0000577 // FIXME: "Leaked" on failure.
578 if (C)
Owen Anderson94148482009-07-08 19:05:04 +0000579 C = new llvm::GlobalVariable(CGM.getModule(), C->getType(),
Daniel Dunbarf69d19e2009-02-17 18:43:32 +0000580 E->getType().isConstQualified(),
581 llvm::GlobalValue::InternalLinkage,
Owen Anderson94148482009-07-08 19:05:04 +0000582 C, ".compoundliteral");
Anders Carlsson00b837c2008-01-26 02:08:50 +0000583 return C;
Anders Carlssonde598292008-01-26 04:30:23 +0000584 }
Douglas Gregor566782a2009-01-06 05:10:23 +0000585 case Expr::DeclRefExprClass:
586 case Expr::QualifiedDeclRefExprClass: {
Douglas Gregord2baafd2008-10-21 16:13:35 +0000587 NamedDecl *Decl = cast<DeclRefExpr>(E)->getDecl();
Anders Carlsson00b837c2008-01-26 02:08:50 +0000588 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Decl))
Chris Lattner80f39cc2009-05-12 21:21:08 +0000589 return CGM.GetAddrOfFunction(GlobalDecl(FD));
Steve Naroff72a6ebc2008-04-15 22:42:06 +0000590 if (const VarDecl* VD = dyn_cast<VarDecl>(Decl)) {
Daniel Dunbarc473a702009-02-24 18:41:57 +0000591 // We can never refer to a variable with local storage.
592 if (!VD->hasLocalStorage()) {
593 if (VD->isFileVarDecl() || VD->hasExternalStorage())
594 return CGM.GetAddrOfGlobalVar(VD);
595 else if (VD->isBlockVarDecl()) {
596 assert(CGF && "Can't access static local vars without CGF");
597 return CGF->GetAddrOfStaticLocalVar(VD);
598 }
Steve Naroff72a6ebc2008-04-15 22:42:06 +0000599 }
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +0000600 }
Eli Friedmane493ea92008-02-11 00:23:10 +0000601 break;
Anders Carlsson00b837c2008-01-26 02:08:50 +0000602 }
Daniel Dunbar31fe9c32008-08-13 23:20:05 +0000603 case Expr::StringLiteralClass:
604 return CGM.GetAddrOfConstantStringFromLiteral(cast<StringLiteral>(E));
Chris Lattnerc5d32632009-02-24 22:18:39 +0000605 case Expr::ObjCEncodeExprClass:
606 return CGM.GetAddrOfConstantStringFromObjCEncode(cast<ObjCEncodeExpr>(E));
Anders Carlsson85b43812008-11-16 06:23:45 +0000607 case Expr::ObjCStringLiteralClass: {
608 ObjCStringLiteral* SL = cast<ObjCStringLiteral>(E);
Steve Naroff2c8a08e2009-03-31 16:53:37 +0000609 llvm::Constant *C = CGM.getObjCRuntime().GenerateConstantString(SL);
Owen Anderson557515b2009-07-29 18:54:39 +0000610 return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
Anders Carlsson85b43812008-11-16 06:23:45 +0000611 }
Chris Lattnerfc370182008-12-12 05:18:02 +0000612 case Expr::PredefinedExprClass: {
613 // __func__/__FUNCTION__ -> "". __PRETTY_FUNCTION__ -> "top level".
614 std::string Str;
615 if (cast<PredefinedExpr>(E)->getIdentType() ==
616 PredefinedExpr::PrettyFunction)
617 Str = "top level";
618
619 return CGM.GetAddrOfConstantCString(Str, ".tmp");
620 }
Eli Friedmanf1941132009-01-25 01:21:06 +0000621 case Expr::AddrLabelExprClass: {
622 assert(CGF && "Invalid address of label expression outside function.");
623 unsigned id = CGF->GetIDForAddrOfLabel(cast<AddrLabelExpr>(E)->getLabel());
Owen Andersonb17ec712009-07-24 23:12:58 +0000624 llvm::Constant *C = llvm::ConstantInt::get(llvm::Type::Int32Ty, id);
Owen Anderson557515b2009-07-29 18:54:39 +0000625 return llvm::ConstantExpr::getIntToPtr(C, ConvertType(E->getType()));
Eli Friedmanf1941132009-01-25 01:21:06 +0000626 }
Eli Friedman67f4ac52009-01-25 01:54:01 +0000627 case Expr::CallExprClass: {
628 CallExpr* CE = cast<CallExpr>(E);
Douglas Gregorb5af7382009-02-14 18:57:46 +0000629 if (CE->isBuiltinCall(CGM.getContext()) !=
630 Builtin::BI__builtin___CFStringMakeConstantString)
Eli Friedman67f4ac52009-01-25 01:54:01 +0000631 break;
632 const Expr *Arg = CE->getArg(0)->IgnoreParenCasts();
633 const StringLiteral *Literal = cast<StringLiteral>(Arg);
Steve Naroff2c8a08e2009-03-31 16:53:37 +0000634 // FIXME: need to deal with UCN conversion issues.
Steve Naroff9a744e52009-04-01 13:55:36 +0000635 return CGM.GetAddrOfConstantCFString(Literal);
Eli Friedman67f4ac52009-01-25 01:54:01 +0000636 }
Mike Stumpd55240e2009-02-19 01:01:04 +0000637 case Expr::BlockExprClass: {
Anders Carlsson6cf64be2009-03-01 01:09:12 +0000638 std::string FunctionName;
639 if (CGF)
640 FunctionName = CGF->CurFn->getName();
641 else
642 FunctionName = "global";
643
644 return CGM.GetAddrOfGlobalBlock(cast<BlockExpr>(E), FunctionName.c_str());
Mike Stumpd55240e2009-02-19 01:01:04 +0000645 }
Eli Friedmane493ea92008-02-11 00:23:10 +0000646 }
Daniel Dunbarf69d19e2009-02-17 18:43:32 +0000647
648 return 0;
Anders Carlsson00b837c2008-01-26 02:08:50 +0000649 }
Anders Carlsson00b837c2008-01-26 02:08:50 +0000650};
651
Anders Carlssond76cead2008-01-26 01:36:00 +0000652} // end anonymous namespace.
653
Lauro Ramos Venancio934fb022008-02-26 21:41:45 +0000654llvm::Constant *CodeGenModule::EmitConstantExpr(const Expr *E,
Anders Carlssone7d97c22009-04-08 04:48:15 +0000655 QualType DestType,
Chris Lattnerc154ac12008-07-26 22:37:01 +0000656 CodeGenFunction *CGF) {
Anders Carlssond6e823b2008-12-01 02:42:14 +0000657 Expr::EvalResult Result;
658
Anders Carlsson5dfe3072009-04-11 01:08:03 +0000659 bool Success = false;
660
Eli Friedmanc0a35bf2009-05-27 06:04:58 +0000661 if (DestType->isReferenceType())
662 Success = E->EvaluateAsLValue(Result, Context);
663 else
Anders Carlsson5dfe3072009-04-11 01:08:03 +0000664 Success = E->Evaluate(Result, Context);
665
666 if (Success) {
Anders Carlssond6e823b2008-12-01 02:42:14 +0000667 assert(!Result.HasSideEffects &&
668 "Constant expr should not have any side effects!");
669 switch (Result.Val.getKind()) {
Daniel Dunbaraf781bb2009-01-28 22:24:07 +0000670 case APValue::Uninitialized:
Daniel Dunbarf69d19e2009-02-17 18:43:32 +0000671 assert(0 && "Constant expressions should be initialized.");
672 return 0;
Anders Carlsson54483902008-11-15 20:45:50 +0000673 case APValue::LValue: {
Anders Carlsson5dfe3072009-04-11 01:08:03 +0000674 const llvm::Type *DestTy = getTypes().ConvertTypeForMem(DestType);
Anders Carlssond6e823b2008-12-01 02:42:14 +0000675 llvm::Constant *Offset =
Owen Andersonb17ec712009-07-24 23:12:58 +0000676 llvm::ConstantInt::get(llvm::Type::Int64Ty,
Anders Carlssond6e823b2008-12-01 02:42:14 +0000677 Result.Val.getLValueOffset());
Anders Carlsson54483902008-11-15 20:45:50 +0000678
Daniel Dunbar786b1462009-02-19 21:44:24 +0000679 llvm::Constant *C;
Anders Carlssond6e823b2008-12-01 02:42:14 +0000680 if (const Expr *LVBase = Result.Val.getLValueBase()) {
Daniel Dunbar786b1462009-02-19 21:44:24 +0000681 C = ConstExprEmitter(*this, CGF).EmitLValue(const_cast<Expr*>(LVBase));
Anders Carlsson85b43812008-11-16 06:23:45 +0000682
Daniel Dunbar786b1462009-02-19 21:44:24 +0000683 // Apply offset if necessary.
684 if (!Offset->isNullValue()) {
685 const llvm::Type *Type =
Owen Anderson7ec2d8f2009-07-29 22:16:19 +0000686 llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
Owen Anderson557515b2009-07-29 18:54:39 +0000687 llvm::Constant *Casted = llvm::ConstantExpr::getBitCast(C, Type);
688 Casted = llvm::ConstantExpr::getGetElementPtr(Casted, &Offset, 1);
689 C = llvm::ConstantExpr::getBitCast(Casted, C->getType());
Daniel Dunbar786b1462009-02-19 21:44:24 +0000690 }
691
692 // Convert to the appropriate type; this could be an lvalue for
693 // an integer.
Anders Carlssona828a722009-04-10 04:59:59 +0000694 if (isa<llvm::PointerType>(DestTy))
Owen Anderson557515b2009-07-29 18:54:39 +0000695 return llvm::ConstantExpr::getBitCast(C, DestTy);
Daniel Dunbar786b1462009-02-19 21:44:24 +0000696
Owen Anderson557515b2009-07-29 18:54:39 +0000697 return llvm::ConstantExpr::getPtrToInt(C, DestTy);
Daniel Dunbar786b1462009-02-19 21:44:24 +0000698 } else {
699 C = Offset;
700
701 // Convert to the appropriate type; this could be an lvalue for
702 // an integer.
Anders Carlssona828a722009-04-10 04:59:59 +0000703 if (isa<llvm::PointerType>(DestTy))
Owen Anderson557515b2009-07-29 18:54:39 +0000704 return llvm::ConstantExpr::getIntToPtr(C, DestTy);
Daniel Dunbar786b1462009-02-19 21:44:24 +0000705
706 // If the types don't match this should only be a truncate.
Anders Carlssona828a722009-04-10 04:59:59 +0000707 if (C->getType() != DestTy)
Owen Anderson557515b2009-07-29 18:54:39 +0000708 return llvm::ConstantExpr::getTrunc(C, DestTy);
Daniel Dunbar786b1462009-02-19 21:44:24 +0000709
710 return C;
Anders Carlsson85b43812008-11-16 06:23:45 +0000711 }
Anders Carlsson54483902008-11-15 20:45:50 +0000712 }
Eli Friedman62301c02008-11-17 03:57:28 +0000713 case APValue::Int: {
Owen Andersonb17ec712009-07-24 23:12:58 +0000714 llvm::Constant *C = llvm::ConstantInt::get(VMContext,
715 Result.Val.getInt());
Anders Carlsson54483902008-11-15 20:45:50 +0000716
717 if (C->getType() == llvm::Type::Int1Ty) {
718 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
Owen Anderson557515b2009-07-29 18:54:39 +0000719 C = llvm::ConstantExpr::getZExt(C, BoolTy);
Anders Carlsson54483902008-11-15 20:45:50 +0000720 }
721 return C;
Eli Friedman62301c02008-11-17 03:57:28 +0000722 }
Daniel Dunbaraf781bb2009-01-28 22:24:07 +0000723 case APValue::ComplexInt: {
724 llvm::Constant *Complex[2];
725
Owen Andersonb17ec712009-07-24 23:12:58 +0000726 Complex[0] = llvm::ConstantInt::get(VMContext,
727 Result.Val.getComplexIntReal());
728 Complex[1] = llvm::ConstantInt::get(VMContext,
729 Result.Val.getComplexIntImag());
Daniel Dunbaraf781bb2009-01-28 22:24:07 +0000730
Owen Anderson5ccd63c2009-07-27 22:29:56 +0000731 return llvm::ConstantStruct::get(Complex, 2);
Daniel Dunbaraf781bb2009-01-28 22:24:07 +0000732 }
Anders Carlsson54483902008-11-15 20:45:50 +0000733 case APValue::Float:
Owen Andersonb56fd0b2009-07-27 21:00:51 +0000734 return llvm::ConstantFP::get(VMContext, Result.Val.getFloat());
Anders Carlsson806e9932008-11-17 01:58:55 +0000735 case APValue::ComplexFloat: {
736 llvm::Constant *Complex[2];
737
Owen Andersonb56fd0b2009-07-27 21:00:51 +0000738 Complex[0] = llvm::ConstantFP::get(VMContext,
739 Result.Val.getComplexFloatReal());
740 Complex[1] = llvm::ConstantFP::get(VMContext,
741 Result.Val.getComplexFloatImag());
Anders Carlsson806e9932008-11-17 01:58:55 +0000742
Owen Anderson5ccd63c2009-07-27 22:29:56 +0000743 return llvm::ConstantStruct::get(Complex, 2);
Anders Carlsson806e9932008-11-17 01:58:55 +0000744 }
Nate Begemanc45d4e82009-01-18 01:01:34 +0000745 case APValue::Vector: {
746 llvm::SmallVector<llvm::Constant *, 4> Inits;
747 unsigned NumElts = Result.Val.getVectorLength();
748
749 for (unsigned i = 0; i != NumElts; ++i) {
750 APValue &Elt = Result.Val.getVectorElt(i);
751 if (Elt.isInt())
Owen Andersonb17ec712009-07-24 23:12:58 +0000752 Inits.push_back(llvm::ConstantInt::get(VMContext, Elt.getInt()));
Nate Begemanc45d4e82009-01-18 01:01:34 +0000753 else
Owen Andersonb56fd0b2009-07-27 21:00:51 +0000754 Inits.push_back(llvm::ConstantFP::get(VMContext, Elt.getFloat()));
Nate Begemanc45d4e82009-01-18 01:01:34 +0000755 }
Owen Anderson17971fa2009-07-28 21:22:35 +0000756 return llvm::ConstantVector::get(&Inits[0], Inits.size());
Nate Begemanc45d4e82009-01-18 01:01:34 +0000757 }
Anders Carlsson54483902008-11-15 20:45:50 +0000758 }
759 }
Eli Friedmane2636632008-06-01 15:31:44 +0000760
761 llvm::Constant* C = ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
Daniel Dunbarf69d19e2009-02-17 18:43:32 +0000762 if (C && C->getType() == llvm::Type::Int1Ty) {
Eli Friedmane2636632008-06-01 15:31:44 +0000763 const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
Owen Anderson557515b2009-07-29 18:54:39 +0000764 C = llvm::ConstantExpr::getZExt(C, BoolTy);
Eli Friedmane2636632008-06-01 15:31:44 +0000765 }
766 return C;
Anders Carlssond76cead2008-01-26 01:36:00 +0000767}
Eli Friedmanb836cd32009-04-13 21:47:26 +0000768
769llvm::Constant *CodeGenModule::EmitNullConstant(QualType T) {
770 // Always return an LLVM null constant for now; this will change when we
771 // get support for IRGen of member pointers.
Owen Andersonf37b84b2009-07-31 20:28:54 +0000772 return llvm::Constant::getNullValue(getTypes().ConvertType(T));
Eli Friedmanb836cd32009-04-13 21:47:26 +0000773}