CGExprConstant.cpp 46 KB

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