1 //===--- CGExprConstant.cpp - Emit LLVM Code from Constant Expressions ----===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This contains code to emit Constant Expr nodes as LLVM code.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "CodeGenFunction.h"
15 #include "CodeGenModule.h"
16 #include "CGObjCRuntime.h"
17 #include "clang/AST/APValue.h"
18 #include "clang/AST/ASTContext.h"
19 #include "clang/AST/StmtVisitor.h"
20 #include "llvm/Constants.h"
21 #include "llvm/Function.h"
22 #include "llvm/GlobalVariable.h"
23 #include "llvm/Support/Compiler.h"
24 #include "llvm/Target/TargetData.h"
25 using namespace clang;
26 using namespace CodeGen;
27 
28 namespace  {
29 class VISIBILITY_HIDDEN ConstExprEmitter :
30   public StmtVisitor<ConstExprEmitter, llvm::Constant*> {
31   CodeGenModule &CGM;
32   CodeGenFunction *CGF;
33 public:
34   ConstExprEmitter(CodeGenModule &cgm, CodeGenFunction *cgf)
35     : CGM(cgm), CGF(cgf) {
36   }
37 
38   //===--------------------------------------------------------------------===//
39   //                            Visitor Methods
40   //===--------------------------------------------------------------------===//
41 
42   llvm::Constant *VisitStmt(Stmt *S) {
43     CGM.ErrorUnsupported(S, "constant expression");
44     QualType T = cast<Expr>(S)->getType();
45     return llvm::UndefValue::get(CGM.getTypes().ConvertType(T));
46   }
47 
48   llvm::Constant *VisitParenExpr(ParenExpr *PE) {
49     return Visit(PE->getSubExpr());
50   }
51 
52   llvm::Constant *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
53     return Visit(E->getInitializer());
54   }
55 
56   llvm::Constant *VisitCastExpr(CastExpr* E) {
57     // GCC cast to union extension
58     if (E->getType()->isUnionType()) {
59       const llvm::Type *Ty = ConvertType(E->getType());
60       return EmitUnion(CGM.EmitConstantExpr(E->getSubExpr(), CGF), Ty);
61     }
62 
63     llvm::Constant *C = Visit(E->getSubExpr());
64 
65     return EmitConversion(C, E->getSubExpr()->getType(), E->getType());
66   }
67 
68   llvm::Constant *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
69     return Visit(DAE->getExpr());
70   }
71 
72   llvm::Constant *EmitArrayInitialization(InitListExpr *ILE) {
73     std::vector<llvm::Constant*> Elts;
74     const llvm::ArrayType *AType =
75         cast<llvm::ArrayType>(ConvertType(ILE->getType()));
76     unsigned NumInitElements = ILE->getNumInits();
77     // FIXME: Check for wide strings
78     if (NumInitElements > 0 && isa<StringLiteral>(ILE->getInit(0)) &&
79         ILE->getType()->getArrayElementTypeNoTypeQual()->isCharType())
80       return Visit(ILE->getInit(0));
81     const llvm::Type *ElemTy = AType->getElementType();
82     unsigned NumElements = AType->getNumElements();
83 
84     // Initialising an array requires us to automatically
85     // initialise any elements that have not been initialised explicitly
86     unsigned NumInitableElts = std::min(NumInitElements, NumElements);
87 
88     // Copy initializer elements.
89     unsigned i = 0;
90     bool RewriteType = false;
91     for (; i < NumInitableElts; ++i) {
92       llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(i), CGF);
93       RewriteType |= (C->getType() != ElemTy);
94       Elts.push_back(C);
95     }
96 
97     // Initialize remaining array elements.
98     for (; i < NumElements; ++i)
99       Elts.push_back(llvm::Constant::getNullValue(ElemTy));
100 
101     if (RewriteType) {
102       // FIXME: Try to avoid packing the array
103       std::vector<const llvm::Type*> Types;
104       for (unsigned i = 0; i < Elts.size(); ++i)
105         Types.push_back(Elts[i]->getType());
106       const llvm::StructType *SType = llvm::StructType::get(Types, true);
107       return llvm::ConstantStruct::get(SType, Elts);
108     }
109 
110     return llvm::ConstantArray::get(AType, Elts);
111   }
112 
113   void InsertBitfieldIntoStruct(std::vector<llvm::Constant*>& Elts,
114                                 FieldDecl* Field, Expr* E) {
115     // Calculate the value to insert
116     llvm::Constant *C = CGM.EmitConstantExpr(E, CGF);
117     llvm::ConstantInt *CI = dyn_cast<llvm::ConstantInt>(C);
118     if (!CI) {
119       CGM.ErrorUnsupported(E, "bitfield initialization");
120       return;
121     }
122     llvm::APInt V = CI->getValue();
123 
124     // Calculate information about the relevant field
125     const llvm::Type* Ty = CI->getType();
126     const llvm::TargetData &TD = CGM.getTypes().getTargetData();
127     unsigned size = TD.getTypeStoreSizeInBits(Ty);
128     unsigned fieldOffset = CGM.getTypes().getLLVMFieldNo(Field) * size;
129     CodeGenTypes::BitFieldInfo bitFieldInfo =
130         CGM.getTypes().getBitFieldInfo(Field);
131     fieldOffset += bitFieldInfo.Begin;
132 
133     // Find where to start the insertion
134     // FIXME: This is O(n^2) in the number of bit-fields!
135     // FIXME: This won't work if the struct isn't completely packed!
136     unsigned offset = 0, i = 0;
137     while (offset < (fieldOffset & -8))
138       offset += TD.getTypeStoreSizeInBits(Elts[i++]->getType());
139 
140     // Advance over 0 sized elements (must terminate in bounds since
141     // the bitfield must have a size).
142     while (TD.getTypeStoreSizeInBits(Elts[i]->getType()) == 0)
143       ++i;
144 
145     // Promote the size of V if necessary
146     // FIXME: This should never occur, but currently it can because
147     // initializer constants are cast to bool, and because clang is
148     // not enforcing bitfield width limits.
149     if (bitFieldInfo.Size > V.getBitWidth())
150       V.zext(bitFieldInfo.Size);
151 
152     // Insert the bits into the struct
153     // FIXME: This algorthm is only correct on X86!
154     // FIXME: THis algorthm assumes bit-fields only have byte-size elements!
155     unsigned bitsToInsert = bitFieldInfo.Size;
156     unsigned curBits = std::min(8 - (fieldOffset & 7), bitsToInsert);
157     unsigned byte = V.getLoBits(curBits).getZExtValue() << (fieldOffset & 7);
158     do {
159       llvm::Constant* byteC = llvm::ConstantInt::get(llvm::Type::Int8Ty, byte);
160       Elts[i] = llvm::ConstantExpr::getOr(Elts[i], byteC);
161       ++i;
162       V = V.lshr(curBits);
163       bitsToInsert -= curBits;
164 
165       if (!bitsToInsert)
166         break;
167 
168       curBits = bitsToInsert > 8 ? 8 : bitsToInsert;
169       byte = V.getLoBits(curBits).getZExtValue();
170     } while (true);
171   }
172 
173   llvm::Constant *EmitStructInitialization(InitListExpr *ILE) {
174     const llvm::StructType *SType =
175         cast<llvm::StructType>(ConvertType(ILE->getType()));
176     RecordDecl *RD = ILE->getType()->getAsRecordType()->getDecl();
177     std::vector<llvm::Constant*> Elts;
178 
179     // Initialize the whole structure to zero.
180     for (unsigned i = 0; i < SType->getNumElements(); ++i) {
181       const llvm::Type *FieldTy = SType->getElementType(i);
182       Elts.push_back(llvm::Constant::getNullValue(FieldTy));
183     }
184 
185     // Copy initializer elements. Skip padding fields.
186     unsigned EltNo = 0;  // Element no in ILE
187     int FieldNo = 0; // Field no in RecordDecl
188     bool RewriteType = false;
189     for (RecordDecl::field_iterator Field = RD->field_begin(),
190                                  FieldEnd = RD->field_end();
191          EltNo < ILE->getNumInits() && Field != FieldEnd; ++Field) {
192       FieldNo++;
193       if (!Field->getIdentifier())
194         continue;
195 
196       if (Field->isBitField()) {
197         InsertBitfieldIntoStruct(Elts, *Field, ILE->getInit(EltNo));
198       } else {
199         unsigned FieldNo = CGM.getTypes().getLLVMFieldNo(*Field);
200         llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(EltNo), CGF);
201         RewriteType |= (C->getType() != Elts[FieldNo]->getType());
202         Elts[FieldNo] = C;
203       }
204       EltNo++;
205     }
206 
207     if (RewriteType) {
208       // FIXME: Make this work for non-packed structs
209       assert(SType->isPacked() && "Cannot recreate unpacked structs");
210       std::vector<const llvm::Type*> Types;
211       for (unsigned i = 0; i < Elts.size(); ++i)
212         Types.push_back(Elts[i]->getType());
213       SType = llvm::StructType::get(Types, true);
214     }
215 
216     return llvm::ConstantStruct::get(SType, Elts);
217   }
218 
219   llvm::Constant *EmitUnion(llvm::Constant *C, const llvm::Type *Ty) {
220     // Build a struct with the union sub-element as the first member,
221     // and padded to the appropriate size
222     std::vector<llvm::Constant*> Elts;
223     std::vector<const llvm::Type*> Types;
224     Elts.push_back(C);
225     Types.push_back(C->getType());
226     unsigned CurSize = CGM.getTargetData().getTypeStoreSize(C->getType());
227     unsigned TotalSize = CGM.getTargetData().getTypeStoreSize(Ty);
228     while (CurSize < TotalSize) {
229       Elts.push_back(llvm::Constant::getNullValue(llvm::Type::Int8Ty));
230       Types.push_back(llvm::Type::Int8Ty);
231       CurSize++;
232     }
233 
234     // This always generates a packed struct
235     // FIXME: Try to generate an unpacked struct when we can
236     llvm::StructType* STy = llvm::StructType::get(Types, true);
237     return llvm::ConstantStruct::get(STy, Elts);
238   }
239 
240   llvm::Constant *EmitUnionInitialization(InitListExpr *ILE) {
241     RecordDecl *RD = ILE->getType()->getAsRecordType()->getDecl();
242     const llvm::Type *Ty = ConvertType(ILE->getType());
243 
244     // Find the field decl we're initializing, if any
245     int FieldNo = 0; // Field no in RecordDecl
246     FieldDecl* curField = 0;
247     for (RecordDecl::field_iterator Field = RD->field_begin(),
248                                  FieldEnd = RD->field_end();
249          Field != FieldEnd; ++Field) {
250       curField = *Field;
251       FieldNo++;
252       if (curField->getIdentifier())
253         break;
254     }
255 
256     if (!curField || !curField->getIdentifier() || ILE->getNumInits() == 0)
257       return llvm::Constant::getNullValue(Ty);
258 
259     if (curField->isBitField()) {
260       // Create a dummy struct for bit-field insertion
261       unsigned NumElts = CGM.getTargetData().getTypePaddedSize(Ty) / 8;
262       llvm::Constant* NV = llvm::Constant::getNullValue(llvm::Type::Int8Ty);
263       std::vector<llvm::Constant*> Elts(NumElts, NV);
264 
265       InsertBitfieldIntoStruct(Elts, curField, ILE->getInit(0));
266       const llvm::ArrayType *RetTy =
267           llvm::ArrayType::get(NV->getType(), NumElts);
268       return llvm::ConstantArray::get(RetTy, Elts);
269     }
270 
271     return EmitUnion(CGM.EmitConstantExpr(ILE->getInit(0), CGF), Ty);
272   }
273 
274   llvm::Constant *EmitVectorInitialization(InitListExpr *ILE) {
275     const llvm::VectorType *VType =
276         cast<llvm::VectorType>(ConvertType(ILE->getType()));
277     const llvm::Type *ElemTy = VType->getElementType();
278     std::vector<llvm::Constant*> Elts;
279     unsigned NumElements = VType->getNumElements();
280     unsigned NumInitElements = ILE->getNumInits();
281 
282     unsigned NumInitableElts = std::min(NumInitElements, NumElements);
283 
284     // Copy initializer elements.
285     unsigned i = 0;
286     for (; i < NumInitableElts; ++i) {
287       llvm::Constant *C = CGM.EmitConstantExpr(ILE->getInit(i), CGF);
288       Elts.push_back(C);
289     }
290 
291     for (; i < NumElements; ++i)
292       Elts.push_back(llvm::Constant::getNullValue(ElemTy));
293 
294     return llvm::ConstantVector::get(VType, Elts);
295   }
296 
297   llvm::Constant *VisitInitListExpr(InitListExpr *ILE) {
298     if (ILE->getType()->isScalarType()) {
299       // We have a scalar in braces. Just use the first element.
300       if (ILE->getNumInits() > 0)
301         return CGM.EmitConstantExpr(ILE->getInit(0), CGF);
302 
303       const llvm::Type* RetTy = CGM.getTypes().ConvertType(ILE->getType());
304       return llvm::Constant::getNullValue(RetTy);
305     }
306 
307     // FIXME: We don't codegen or sema designators yet.
308     if (ILE->hadDesignators()) {
309       CGM.ErrorUnsupported(ILE, "initializer list with designators");
310       return llvm::UndefValue::get(ConvertType(ILE->getType()));
311     }
312 
313     if (ILE->getType()->isArrayType())
314       return EmitArrayInitialization(ILE);
315 
316     if (ILE->getType()->isStructureType())
317       return EmitStructInitialization(ILE);
318 
319     if (ILE->getType()->isUnionType())
320       return EmitUnionInitialization(ILE);
321 
322     if (ILE->getType()->isVectorType())
323       return EmitVectorInitialization(ILE);
324 
325     assert(0 && "Unable to handle InitListExpr");
326     // Get rid of control reaches end of void function warning.
327     // Not reached.
328     return 0;
329   }
330 
331   llvm::Constant *VisitImplicitCastExpr(ImplicitCastExpr *ICExpr) {
332     Expr* SExpr = ICExpr->getSubExpr();
333     QualType SType = SExpr->getType();
334     llvm::Constant *C; // the intermediate expression
335     QualType T;        // the type of the intermediate expression
336     if (SType->isArrayType()) {
337       // Arrays decay to a pointer to the first element
338       // VLAs would require special handling, but they can't occur here
339       C = EmitLValue(SExpr);
340       llvm::Constant *Idx0 = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
341       llvm::Constant *Ops[] = {Idx0, Idx0};
342       C = llvm::ConstantExpr::getGetElementPtr(C, Ops, 2);
343       T = CGM.getContext().getArrayDecayedType(SType);
344     } else if (SType->isFunctionType()) {
345       // Function types decay to a pointer to the function
346       C = EmitLValue(SExpr);
347       T = CGM.getContext().getPointerType(SType);
348     } else {
349       C = Visit(SExpr);
350       T = SType;
351     }
352 
353     // Perform the conversion; note that an implicit cast can both promote
354     // and convert an array/function
355     return EmitConversion(C, T, ICExpr->getType());
356   }
357 
358   llvm::Constant *VisitStringLiteral(StringLiteral *E) {
359     assert(!E->getType()->isPointerType() && "Strings are always arrays");
360 
361     // Otherwise this must be a string initializing an array in a static
362     // initializer.  Don't emit it as the address of the string, emit the string
363     // data itself as an inline array.
364     return llvm::ConstantArray::get(CGM.GetStringForStringLiteral(E), false);
365   }
366 
367   llvm::Constant *VisitUnaryExtension(const UnaryOperator *E) {
368     return Visit(E->getSubExpr());
369   }
370 
371   // Utility methods
372   const llvm::Type *ConvertType(QualType T) {
373     return CGM.getTypes().ConvertType(T);
374   }
375 
376   llvm::Constant *EmitConversionToBool(llvm::Constant *Src, QualType SrcType) {
377     assert(SrcType->isCanonical() && "EmitConversion strips typedefs");
378 
379     if (SrcType->isRealFloatingType()) {
380       // Compare against 0.0 for fp scalars.
381       llvm::Constant *Zero = llvm::Constant::getNullValue(Src->getType());
382       return llvm::ConstantExpr::getFCmp(llvm::FCmpInst::FCMP_UNE, Src, Zero);
383     }
384 
385     assert((SrcType->isIntegerType() || SrcType->isPointerType()) &&
386            "Unknown scalar type to convert");
387 
388     // Compare against an integer or pointer null.
389     llvm::Constant *Zero = llvm::Constant::getNullValue(Src->getType());
390     return llvm::ConstantExpr::getICmp(llvm::ICmpInst::ICMP_NE, Src, Zero);
391   }
392 
393   llvm::Constant *EmitConversion(llvm::Constant *Src, QualType SrcType,
394                                  QualType DstType) {
395     SrcType = CGM.getContext().getCanonicalType(SrcType);
396     DstType = CGM.getContext().getCanonicalType(DstType);
397     if (SrcType == DstType) return Src;
398 
399     // Handle conversions to bool first, they are special: comparisons against 0.
400     if (DstType->isBooleanType())
401       return EmitConversionToBool(Src, SrcType);
402 
403     const llvm::Type *DstTy = ConvertType(DstType);
404 
405     // Ignore conversions like int -> uint.
406     if (Src->getType() == DstTy)
407       return Src;
408 
409     // Handle pointer conversions next: pointers can only be converted to/from
410     // other pointers and integers.
411     if (isa<llvm::PointerType>(DstTy)) {
412       // The source value may be an integer, or a pointer.
413       if (isa<llvm::PointerType>(Src->getType()))
414         return llvm::ConstantExpr::getBitCast(Src, DstTy);
415       assert(SrcType->isIntegerType() &&"Not ptr->ptr or int->ptr conversion?");
416       return llvm::ConstantExpr::getIntToPtr(Src, DstTy);
417     }
418 
419     if (isa<llvm::PointerType>(Src->getType())) {
420       // Must be an ptr to int cast.
421       assert(isa<llvm::IntegerType>(DstTy) && "not ptr->int?");
422       return llvm::ConstantExpr::getPtrToInt(Src, DstTy);
423     }
424 
425     // A scalar source can be splatted to a vector of the same element type
426     if (isa<llvm::VectorType>(DstTy) && !isa<VectorType>(SrcType)) {
427       assert((cast<llvm::VectorType>(DstTy)->getElementType()
428               == Src->getType()) &&
429              "Vector element type must match scalar type to splat.");
430       unsigned NumElements = DstType->getAsVectorType()->getNumElements();
431       llvm::SmallVector<llvm::Constant*, 16> Elements;
432       for (unsigned i = 0; i < NumElements; i++)
433         Elements.push_back(Src);
434 
435       return llvm::ConstantVector::get(&Elements[0], NumElements);
436     }
437 
438     if (isa<llvm::VectorType>(Src->getType()) ||
439         isa<llvm::VectorType>(DstTy)) {
440       return llvm::ConstantExpr::getBitCast(Src, DstTy);
441     }
442 
443     // Finally, we have the arithmetic types: real int/float.
444     if (isa<llvm::IntegerType>(Src->getType())) {
445       bool InputSigned = SrcType->isSignedIntegerType();
446       if (isa<llvm::IntegerType>(DstTy))
447         return llvm::ConstantExpr::getIntegerCast(Src, DstTy, InputSigned);
448       else if (InputSigned)
449         return llvm::ConstantExpr::getSIToFP(Src, DstTy);
450       else
451         return llvm::ConstantExpr::getUIToFP(Src, DstTy);
452     }
453 
454     assert(Src->getType()->isFloatingPoint() && "Unknown real conversion");
455     if (isa<llvm::IntegerType>(DstTy)) {
456       if (DstType->isSignedIntegerType())
457         return llvm::ConstantExpr::getFPToSI(Src, DstTy);
458       else
459         return llvm::ConstantExpr::getFPToUI(Src, DstTy);
460     }
461 
462     assert(DstTy->isFloatingPoint() && "Unknown real conversion");
463     if (DstTy->getTypeID() < Src->getType()->getTypeID())
464       return llvm::ConstantExpr::getFPTrunc(Src, DstTy);
465     else
466       return llvm::ConstantExpr::getFPExtend(Src, DstTy);
467   }
468 
469 public:
470   llvm::Constant *EmitLValue(Expr *E) {
471     switch (E->getStmtClass()) {
472     default: break;
473     case Expr::ParenExprClass:
474       // Elide parenthesis
475       return EmitLValue(cast<ParenExpr>(E)->getSubExpr());
476     case Expr::CompoundLiteralExprClass: {
477       // Note that due to the nature of compound literals, this is guaranteed
478       // to be the only use of the variable, so we just generate it here.
479       CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
480       llvm::Constant* C = Visit(CLE->getInitializer());
481       C = new llvm::GlobalVariable(C->getType(),E->getType().isConstQualified(),
482                                    llvm::GlobalValue::InternalLinkage,
483                                    C, ".compoundliteral", &CGM.getModule());
484       return C;
485     }
486     case Expr::DeclRefExprClass:
487     case Expr::QualifiedDeclRefExprClass: {
488       NamedDecl *Decl = cast<DeclRefExpr>(E)->getDecl();
489       if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Decl))
490         return CGM.GetAddrOfFunction(FD);
491       if (const VarDecl* VD = dyn_cast<VarDecl>(Decl)) {
492         if (VD->isFileVarDecl())
493           return CGM.GetAddrOfGlobalVar(VD);
494         else if (VD->isBlockVarDecl()) {
495           assert(CGF && "Can't access static local vars without CGF");
496           return CGF->GetAddrOfStaticLocalVar(VD);
497         }
498       }
499       break;
500     }
501     case Expr::MemberExprClass: {
502       MemberExpr* ME = cast<MemberExpr>(E);
503       llvm::Constant *Base;
504       if (ME->isArrow())
505         Base = Visit(ME->getBase());
506       else
507         Base = EmitLValue(ME->getBase());
508 
509       FieldDecl *Field = dyn_cast<FieldDecl>(ME->getMemberDecl());
510       // FIXME: Handle other kinds of member expressions.
511       assert(Field && "No code generation for non-field member expressions");
512       unsigned FieldNumber = CGM.getTypes().getLLVMFieldNo(Field);
513       llvm::Constant *Zero = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
514       llvm::Constant *Idx = llvm::ConstantInt::get(llvm::Type::Int32Ty,
515                                                    FieldNumber);
516       llvm::Value *Ops[] = {Zero, Idx};
517       return llvm::ConstantExpr::getGetElementPtr(Base, Ops, 2);
518     }
519     case Expr::ArraySubscriptExprClass: {
520       ArraySubscriptExpr* ASExpr = cast<ArraySubscriptExpr>(E);
521       llvm::Constant *Base = Visit(ASExpr->getBase());
522       llvm::Constant *Index = Visit(ASExpr->getIdx());
523       assert(!ASExpr->getBase()->getType()->isVectorType() &&
524              "Taking the address of a vector component is illegal!");
525       return llvm::ConstantExpr::getGetElementPtr(Base, &Index, 1);
526     }
527     case Expr::StringLiteralClass:
528       return CGM.GetAddrOfConstantStringFromLiteral(cast<StringLiteral>(E));
529     case Expr::ObjCStringLiteralClass: {
530       ObjCStringLiteral* SL = cast<ObjCStringLiteral>(E);
531       std::string S(SL->getString()->getStrData(),
532                     SL->getString()->getByteLength());
533       llvm::Constant *C = CGM.getObjCRuntime().GenerateConstantString(S);
534       return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
535     }
536     case Expr::UnaryOperatorClass: {
537       UnaryOperator *Exp = cast<UnaryOperator>(E);
538       switch (Exp->getOpcode()) {
539       default: break;
540       case UnaryOperator::Extension:
541         // Extension is just a wrapper for expressions
542         return EmitLValue(Exp->getSubExpr());
543       case UnaryOperator::Real:
544       case UnaryOperator::Imag: {
545         // The address of __real or __imag is just a GEP off the address
546         // of the internal expression
547         llvm::Constant* C = EmitLValue(Exp->getSubExpr());
548         llvm::Constant *Zero = llvm::ConstantInt::get(llvm::Type::Int32Ty, 0);
549         llvm::Constant *Idx  = llvm::ConstantInt::get(llvm::Type::Int32Ty,
550                                        Exp->getOpcode() == UnaryOperator::Imag);
551         llvm::Value *Ops[] = {Zero, Idx};
552         return llvm::ConstantExpr::getGetElementPtr(C, Ops, 2);
553       }
554       case UnaryOperator::Deref:
555         // The address of a deref is just the value of the expression
556         return Visit(Exp->getSubExpr());
557       }
558       break;
559     }
560 
561     case Expr::PredefinedExprClass: {
562       // __func__/__FUNCTION__ -> "".  __PRETTY_FUNCTION__ -> "top level".
563       std::string Str;
564       if (cast<PredefinedExpr>(E)->getIdentType() ==
565           PredefinedExpr::PrettyFunction)
566         Str = "top level";
567 
568       return CGM.GetAddrOfConstantCString(Str, ".tmp");
569     }
570     case Expr::AddrLabelExprClass: {
571       assert(CGF && "Invalid address of label expression outside function.");
572       unsigned id = CGF->GetIDForAddrOfLabel(cast<AddrLabelExpr>(E)->getLabel());
573       llvm::Constant *C = llvm::ConstantInt::get(llvm::Type::Int32Ty, id);
574       return llvm::ConstantExpr::getIntToPtr(C, ConvertType(E->getType()));
575     }
576     case Expr::CallExprClass: {
577       CallExpr* CE = cast<CallExpr>(E);
578       if (CE->isBuiltinCall() != Builtin::BI__builtin___CFStringMakeConstantString)
579         break;
580       const Expr *Arg = CE->getArg(0)->IgnoreParenCasts();
581       const StringLiteral *Literal = cast<StringLiteral>(Arg);
582       std::string S(Literal->getStrData(), Literal->getByteLength());
583       return CGM.GetAddrOfConstantCFString(S);
584     }
585     }
586     CGM.ErrorUnsupported(E, "constant l-value expression");
587     llvm::Type *Ty = llvm::PointerType::getUnqual(ConvertType(E->getType()));
588     return llvm::UndefValue::get(Ty);
589   }
590 };
591 
592 }  // end anonymous namespace.
593 
594 llvm::Constant *CodeGenModule::EmitConstantExpr(const Expr *E,
595                                                 CodeGenFunction *CGF) {
596   QualType type = Context.getCanonicalType(E->getType());
597 
598   Expr::EvalResult Result;
599 
600   if (E->Evaluate(Result, Context)) {
601     assert(!Result.HasSideEffects &&
602            "Constant expr should not have any side effects!");
603     switch (Result.Val.getKind()) {
604     default: assert(0 && "unhandled value kind!");
605     case APValue::LValue: {
606       llvm::Constant *Offset =
607         llvm::ConstantInt::get(llvm::Type::Int64Ty,
608                                Result.Val.getLValueOffset());
609 
610       if (const Expr *LVBase = Result.Val.getLValueBase()) {
611         llvm::Constant *C =
612           ConstExprEmitter(*this, CGF).EmitLValue(const_cast<Expr*>(LVBase));
613 
614         const llvm::Type *Type =
615           llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
616         const llvm::Type *DestType = getTypes().ConvertTypeForMem(E->getType());
617 
618         // FIXME: It's a little ugly that we need to cast to a pointer,
619         // apply the GEP and then cast back.
620         C = llvm::ConstantExpr::getBitCast(C, Type);
621         C = llvm::ConstantExpr::getGetElementPtr(C, &Offset, 1);
622 
623         return llvm::ConstantExpr::getBitCast(C, DestType);
624       }
625 
626       return llvm::ConstantExpr::getIntToPtr(Offset,
627                                              getTypes().ConvertType(type));
628     }
629     case APValue::Int: {
630       llvm::Constant *C = llvm::ConstantInt::get(Result.Val.getInt());
631 
632       if (C->getType() == llvm::Type::Int1Ty) {
633         const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
634         C = llvm::ConstantExpr::getZExt(C, BoolTy);
635       }
636       return C;
637     }
638     case APValue::Float:
639       return llvm::ConstantFP::get(Result.Val.getFloat());
640     case APValue::ComplexFloat: {
641       llvm::Constant *Complex[2];
642 
643       Complex[0] = llvm::ConstantFP::get(Result.Val.getComplexFloatReal());
644       Complex[1] = llvm::ConstantFP::get(Result.Val.getComplexFloatImag());
645 
646       return llvm::ConstantStruct::get(Complex, 2);
647     }
648     case APValue::Vector: {
649       llvm::SmallVector<llvm::Constant *, 4> Inits;
650       unsigned NumElts = Result.Val.getVectorLength();
651 
652       for (unsigned i = 0; i != NumElts; ++i) {
653         APValue &Elt = Result.Val.getVectorElt(i);
654         if (Elt.isInt())
655           Inits.push_back(llvm::ConstantInt::get(Elt.getInt()));
656         else
657           Inits.push_back(llvm::ConstantFP::get(Elt.getFloat()));
658       }
659       return llvm::ConstantVector::get(&Inits[0], Inits.size());
660     }
661     }
662   }
663 
664   llvm::Constant* C = ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
665   if (C->getType() == llvm::Type::Int1Ty) {
666     const llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
667     C = llvm::ConstantExpr::getZExt(C, BoolTy);
668   }
669   return C;
670 }
671