CGExprConstant.cpp 61 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685
  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 "CGCXXABI.h"
  15. #include "CGObjCRuntime.h"
  16. #include "CGRecordLayout.h"
  17. #include "CodeGenModule.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/IR/Constants.h"
  24. #include "llvm/IR/DataLayout.h"
  25. #include "llvm/IR/Function.h"
  26. #include "llvm/IR/GlobalVariable.h"
  27. #include "CGHLSLRuntime.h" // HLSL Change
  28. using namespace clang;
  29. using namespace CodeGen;
  30. //===----------------------------------------------------------------------===//
  31. // ConstStructBuilder
  32. //===----------------------------------------------------------------------===//
  33. namespace {
  34. class ConstExprEmitter;
  35. class ConstStructBuilder {
  36. CodeGenModule &CGM;
  37. CodeGenFunction *CGF;
  38. bool Packed;
  39. CharUnits NextFieldOffsetInChars;
  40. CharUnits LLVMStructAlignment;
  41. SmallVector<llvm::Constant *, 32> Elements;
  42. public:
  43. static llvm::Constant *BuildStruct(CodeGenModule &CGM, CodeGenFunction *CFG,
  44. ConstExprEmitter *Emitter,
  45. llvm::ConstantStruct *Base,
  46. InitListExpr *Updater);
  47. static llvm::Constant *BuildStruct(CodeGenModule &CGM, CodeGenFunction *CGF,
  48. InitListExpr *ILE);
  49. static llvm::Constant *BuildStruct(CodeGenModule &CGM, CodeGenFunction *CGF,
  50. const APValue &Value, QualType ValTy);
  51. private:
  52. ConstStructBuilder(CodeGenModule &CGM, CodeGenFunction *CGF)
  53. : CGM(CGM), CGF(CGF), Packed(false),
  54. NextFieldOffsetInChars(CharUnits::Zero()),
  55. LLVMStructAlignment(CharUnits::One()) { }
  56. void AppendField(const FieldDecl *Field, uint64_t FieldOffset,
  57. llvm::Constant *InitExpr);
  58. void AppendBytes(CharUnits FieldOffsetInChars, llvm::Constant *InitCst);
  59. void AppendBitField(const FieldDecl *Field, uint64_t FieldOffset,
  60. llvm::ConstantInt *InitExpr);
  61. void AppendPadding(CharUnits PadSize);
  62. void AppendTailPadding(CharUnits RecordSize);
  63. void ConvertStructToPacked();
  64. bool Build(InitListExpr *ILE);
  65. bool Build(ConstExprEmitter *Emitter, llvm::ConstantStruct *Base,
  66. InitListExpr *Updater);
  67. void Build(const APValue &Val, const RecordDecl *RD, bool IsPrimaryBase,
  68. const CXXRecordDecl *VTableClass, CharUnits BaseOffset);
  69. llvm::Constant *Finalize(QualType Ty);
  70. CharUnits getAlignment(const llvm::Constant *C) const {
  71. if (Packed) return CharUnits::One();
  72. return CharUnits::fromQuantity(
  73. CGM.getDataLayout().getABITypeAlignment(C->getType()));
  74. }
  75. CharUnits getSizeInChars(const llvm::Constant *C) const {
  76. return CharUnits::fromQuantity(
  77. CGM.getDataLayout().getTypeAllocSize(C->getType()));
  78. }
  79. };
  80. void ConstStructBuilder::
  81. AppendField(const FieldDecl *Field, uint64_t FieldOffset,
  82. llvm::Constant *InitCst) {
  83. const ASTContext &Context = CGM.getContext();
  84. CharUnits FieldOffsetInChars = Context.toCharUnitsFromBits(FieldOffset);
  85. AppendBytes(FieldOffsetInChars, InitCst);
  86. }
  87. void ConstStructBuilder::
  88. AppendBytes(CharUnits FieldOffsetInChars, llvm::Constant *InitCst) {
  89. assert(NextFieldOffsetInChars <= FieldOffsetInChars
  90. && "Field offset mismatch!");
  91. CharUnits FieldAlignment = getAlignment(InitCst);
  92. // Round up the field offset to the alignment of the field type.
  93. CharUnits AlignedNextFieldOffsetInChars =
  94. NextFieldOffsetInChars.RoundUpToAlignment(FieldAlignment);
  95. if (AlignedNextFieldOffsetInChars < FieldOffsetInChars) {
  96. // We need to append padding.
  97. AppendPadding(FieldOffsetInChars - NextFieldOffsetInChars);
  98. assert(NextFieldOffsetInChars == FieldOffsetInChars &&
  99. "Did not add enough padding!");
  100. AlignedNextFieldOffsetInChars =
  101. NextFieldOffsetInChars.RoundUpToAlignment(FieldAlignment);
  102. }
  103. if (AlignedNextFieldOffsetInChars > FieldOffsetInChars) {
  104. assert(!Packed && "Alignment is wrong even with a packed struct!");
  105. // Convert the struct to a packed struct.
  106. ConvertStructToPacked();
  107. // After we pack the struct, we may need to insert padding.
  108. if (NextFieldOffsetInChars < FieldOffsetInChars) {
  109. // We need to append padding.
  110. AppendPadding(FieldOffsetInChars - NextFieldOffsetInChars);
  111. assert(NextFieldOffsetInChars == FieldOffsetInChars &&
  112. "Did not add enough padding!");
  113. }
  114. AlignedNextFieldOffsetInChars = NextFieldOffsetInChars;
  115. }
  116. // Add the field.
  117. Elements.push_back(InitCst);
  118. NextFieldOffsetInChars = AlignedNextFieldOffsetInChars +
  119. getSizeInChars(InitCst);
  120. if (Packed)
  121. assert(LLVMStructAlignment == CharUnits::One() &&
  122. "Packed struct not byte-aligned!");
  123. else
  124. LLVMStructAlignment = std::max(LLVMStructAlignment, FieldAlignment);
  125. }
  126. void ConstStructBuilder::AppendBitField(const FieldDecl *Field,
  127. uint64_t FieldOffset,
  128. llvm::ConstantInt *CI) {
  129. const ASTContext &Context = CGM.getContext();
  130. const uint64_t CharWidth = Context.getCharWidth();
  131. uint64_t NextFieldOffsetInBits = Context.toBits(NextFieldOffsetInChars);
  132. if (FieldOffset > NextFieldOffsetInBits) {
  133. // We need to add padding.
  134. CharUnits PadSize = Context.toCharUnitsFromBits(
  135. llvm::RoundUpToAlignment(FieldOffset - NextFieldOffsetInBits,
  136. Context.getTargetInfo().getCharAlign()));
  137. AppendPadding(PadSize);
  138. }
  139. uint64_t FieldSize = Field->getBitWidthValue(Context);
  140. llvm::APInt FieldValue = CI->getValue();
  141. // Promote the size of FieldValue if necessary
  142. // FIXME: This should never occur, but currently it can because initializer
  143. // constants are cast to bool, and because clang is not enforcing bitfield
  144. // width limits.
  145. if (FieldSize > FieldValue.getBitWidth())
  146. FieldValue = FieldValue.zext(FieldSize);
  147. // Truncate the size of FieldValue to the bit field size.
  148. if (FieldSize < FieldValue.getBitWidth())
  149. FieldValue = FieldValue.trunc(FieldSize);
  150. NextFieldOffsetInBits = Context.toBits(NextFieldOffsetInChars);
  151. if (FieldOffset < NextFieldOffsetInBits) {
  152. // Either part of the field or the entire field can go into the previous
  153. // byte.
  154. assert(!Elements.empty() && "Elements can't be empty!");
  155. unsigned BitsInPreviousByte = NextFieldOffsetInBits - FieldOffset;
  156. bool FitsCompletelyInPreviousByte =
  157. BitsInPreviousByte >= FieldValue.getBitWidth();
  158. llvm::APInt Tmp = FieldValue;
  159. if (!FitsCompletelyInPreviousByte) {
  160. unsigned NewFieldWidth = FieldSize - BitsInPreviousByte;
  161. if (CGM.getDataLayout().isBigEndian()) {
  162. Tmp = Tmp.lshr(NewFieldWidth);
  163. Tmp = Tmp.trunc(BitsInPreviousByte);
  164. // We want the remaining high bits.
  165. FieldValue = FieldValue.trunc(NewFieldWidth);
  166. } else {
  167. Tmp = Tmp.trunc(BitsInPreviousByte);
  168. // We want the remaining low bits.
  169. FieldValue = FieldValue.lshr(BitsInPreviousByte);
  170. FieldValue = FieldValue.trunc(NewFieldWidth);
  171. }
  172. }
  173. Tmp = Tmp.zext(CharWidth);
  174. if (CGM.getDataLayout().isBigEndian()) {
  175. if (FitsCompletelyInPreviousByte)
  176. Tmp = Tmp.shl(BitsInPreviousByte - FieldValue.getBitWidth());
  177. } else {
  178. Tmp = Tmp.shl(CharWidth - BitsInPreviousByte);
  179. }
  180. // 'or' in the bits that go into the previous byte.
  181. llvm::Value *LastElt = Elements.back();
  182. if (llvm::ConstantInt *Val = dyn_cast<llvm::ConstantInt>(LastElt))
  183. Tmp |= Val->getValue();
  184. else {
  185. assert(isa<llvm::UndefValue>(LastElt));
  186. // If there is an undef field that we're adding to, it can either be a
  187. // scalar undef (in which case, we just replace it with our field) or it
  188. // is an array. If it is an array, we have to pull one byte off the
  189. // array so that the other undef bytes stay around.
  190. if (!isa<llvm::IntegerType>(LastElt->getType())) {
  191. // The undef padding will be a multibyte array, create a new smaller
  192. // padding and then an hole for our i8 to get plopped into.
  193. assert(isa<llvm::ArrayType>(LastElt->getType()) &&
  194. "Expected array padding of undefs");
  195. llvm::ArrayType *AT = cast<llvm::ArrayType>(LastElt->getType());
  196. assert(AT->getElementType()->isIntegerTy(CharWidth) &&
  197. AT->getNumElements() != 0 &&
  198. "Expected non-empty array padding of undefs");
  199. // Remove the padding array.
  200. NextFieldOffsetInChars -= CharUnits::fromQuantity(AT->getNumElements());
  201. Elements.pop_back();
  202. // Add the padding back in two chunks.
  203. AppendPadding(CharUnits::fromQuantity(AT->getNumElements()-1));
  204. AppendPadding(CharUnits::One());
  205. assert(isa<llvm::UndefValue>(Elements.back()) &&
  206. Elements.back()->getType()->isIntegerTy(CharWidth) &&
  207. "Padding addition didn't work right");
  208. }
  209. }
  210. Elements.back() = llvm::ConstantInt::get(CGM.getLLVMContext(), Tmp);
  211. if (FitsCompletelyInPreviousByte)
  212. return;
  213. }
  214. while (FieldValue.getBitWidth() > CharWidth) {
  215. llvm::APInt Tmp;
  216. if (CGM.getDataLayout().isBigEndian()) {
  217. // We want the high bits.
  218. Tmp =
  219. FieldValue.lshr(FieldValue.getBitWidth() - CharWidth).trunc(CharWidth);
  220. } else {
  221. // We want the low bits.
  222. Tmp = FieldValue.trunc(CharWidth);
  223. FieldValue = FieldValue.lshr(CharWidth);
  224. }
  225. Elements.push_back(llvm::ConstantInt::get(CGM.getLLVMContext(), Tmp));
  226. ++NextFieldOffsetInChars;
  227. FieldValue = FieldValue.trunc(FieldValue.getBitWidth() - CharWidth);
  228. }
  229. assert(FieldValue.getBitWidth() > 0 &&
  230. "Should have at least one bit left!");
  231. assert(FieldValue.getBitWidth() <= CharWidth &&
  232. "Should not have more than a byte left!");
  233. if (FieldValue.getBitWidth() < CharWidth) {
  234. if (CGM.getDataLayout().isBigEndian()) {
  235. unsigned BitWidth = FieldValue.getBitWidth();
  236. FieldValue = FieldValue.zext(CharWidth) << (CharWidth - BitWidth);
  237. } else
  238. FieldValue = FieldValue.zext(CharWidth);
  239. }
  240. // Append the last element.
  241. Elements.push_back(llvm::ConstantInt::get(CGM.getLLVMContext(),
  242. FieldValue));
  243. ++NextFieldOffsetInChars;
  244. }
  245. void ConstStructBuilder::AppendPadding(CharUnits PadSize) {
  246. if (PadSize.isZero())
  247. return;
  248. llvm::Type *Ty = CGM.Int8Ty;
  249. if (PadSize > CharUnits::One())
  250. Ty = llvm::ArrayType::get(Ty, PadSize.getQuantity());
  251. llvm::Constant *C = llvm::UndefValue::get(Ty);
  252. Elements.push_back(C);
  253. assert(getAlignment(C) == CharUnits::One() &&
  254. "Padding must have 1 byte alignment!");
  255. NextFieldOffsetInChars += getSizeInChars(C);
  256. }
  257. void ConstStructBuilder::AppendTailPadding(CharUnits RecordSize) {
  258. assert(NextFieldOffsetInChars <= RecordSize &&
  259. "Size mismatch!");
  260. AppendPadding(RecordSize - NextFieldOffsetInChars);
  261. }
  262. void ConstStructBuilder::ConvertStructToPacked() {
  263. SmallVector<llvm::Constant *, 16> PackedElements;
  264. CharUnits ElementOffsetInChars = CharUnits::Zero();
  265. for (unsigned i = 0, e = Elements.size(); i != e; ++i) {
  266. llvm::Constant *C = Elements[i];
  267. CharUnits ElementAlign = CharUnits::fromQuantity(
  268. CGM.getDataLayout().getABITypeAlignment(C->getType()));
  269. CharUnits AlignedElementOffsetInChars =
  270. ElementOffsetInChars.RoundUpToAlignment(ElementAlign);
  271. if (AlignedElementOffsetInChars > ElementOffsetInChars) {
  272. // We need some padding.
  273. CharUnits NumChars =
  274. AlignedElementOffsetInChars - ElementOffsetInChars;
  275. llvm::Type *Ty = CGM.Int8Ty;
  276. if (NumChars > CharUnits::One())
  277. Ty = llvm::ArrayType::get(Ty, NumChars.getQuantity());
  278. llvm::Constant *Padding = llvm::UndefValue::get(Ty);
  279. PackedElements.push_back(Padding);
  280. ElementOffsetInChars += getSizeInChars(Padding);
  281. }
  282. PackedElements.push_back(C);
  283. ElementOffsetInChars += getSizeInChars(C);
  284. }
  285. assert(ElementOffsetInChars == NextFieldOffsetInChars &&
  286. "Packing the struct changed its size!");
  287. Elements.swap(PackedElements);
  288. LLVMStructAlignment = CharUnits::One();
  289. Packed = true;
  290. }
  291. bool ConstStructBuilder::Build(InitListExpr *ILE) {
  292. RecordDecl *RD = ILE->getType()->getAs<RecordType>()->getDecl();
  293. const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
  294. unsigned FieldNo = 0;
  295. unsigned ElementNo = 0;
  296. for (RecordDecl::field_iterator Field = RD->field_begin(),
  297. FieldEnd = RD->field_end(); Field != FieldEnd; ++Field, ++FieldNo) {
  298. // If this is a union, skip all the fields that aren't being initialized.
  299. if (RD->isUnion() && ILE->getInitializedFieldInUnion() != *Field)
  300. continue;
  301. // Don't emit anonymous bitfields, they just affect layout.
  302. if (Field->isUnnamedBitfield())
  303. continue;
  304. // Get the initializer. A struct can include fields without initializers,
  305. // we just use explicit null values for them.
  306. llvm::Constant *EltInit;
  307. if (ElementNo < ILE->getNumInits())
  308. EltInit = CGM.EmitConstantExpr(ILE->getInit(ElementNo++),
  309. Field->getType(), CGF);
  310. else
  311. EltInit = CGM.EmitNullConstant(Field->getType());
  312. if (!EltInit)
  313. return false;
  314. if (!Field->isBitField()) {
  315. // Handle non-bitfield members.
  316. AppendField(*Field, Layout.getFieldOffset(FieldNo), EltInit);
  317. } else {
  318. // Otherwise we have a bitfield.
  319. if (auto *CI = dyn_cast<llvm::ConstantInt>(EltInit)) {
  320. AppendBitField(*Field, Layout.getFieldOffset(FieldNo), CI);
  321. } else {
  322. // We are trying to initialize a bitfield with a non-trivial constant,
  323. // this must require run-time code.
  324. return false;
  325. }
  326. }
  327. }
  328. return true;
  329. }
  330. namespace {
  331. struct BaseInfo {
  332. BaseInfo(const CXXRecordDecl *Decl, CharUnits Offset, unsigned Index)
  333. : Decl(Decl), Offset(Offset), Index(Index) {
  334. }
  335. const CXXRecordDecl *Decl;
  336. CharUnits Offset;
  337. unsigned Index;
  338. bool operator<(const BaseInfo &O) const { return Offset < O.Offset; }
  339. };
  340. }
  341. void ConstStructBuilder::Build(const APValue &Val, const RecordDecl *RD,
  342. bool IsPrimaryBase,
  343. const CXXRecordDecl *VTableClass,
  344. CharUnits Offset) {
  345. const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
  346. if (const CXXRecordDecl *CD = dyn_cast<CXXRecordDecl>(RD)) {
  347. // Add a vtable pointer, if we need one and it hasn't already been added.
  348. if (CD->isDynamicClass() && !IsPrimaryBase) {
  349. llvm::Constant *VTableAddressPoint =
  350. CGM.getCXXABI().getVTableAddressPointForConstExpr(
  351. BaseSubobject(CD, Offset), VTableClass);
  352. AppendBytes(Offset, VTableAddressPoint);
  353. }
  354. // Accumulate and sort bases, in order to visit them in address order, which
  355. // may not be the same as declaration order.
  356. SmallVector<BaseInfo, 8> Bases;
  357. Bases.reserve(CD->getNumBases());
  358. unsigned BaseNo = 0;
  359. for (CXXRecordDecl::base_class_const_iterator Base = CD->bases_begin(),
  360. BaseEnd = CD->bases_end(); Base != BaseEnd; ++Base, ++BaseNo) {
  361. assert(!Base->isVirtual() && "should not have virtual bases here");
  362. const CXXRecordDecl *BD = Base->getType()->getAsCXXRecordDecl();
  363. CharUnits BaseOffset = Layout.getBaseClassOffset(BD);
  364. Bases.push_back(BaseInfo(BD, BaseOffset, BaseNo));
  365. }
  366. std::stable_sort(Bases.begin(), Bases.end());
  367. for (unsigned I = 0, N = Bases.size(); I != N; ++I) {
  368. BaseInfo &Base = Bases[I];
  369. bool IsPrimaryBase = Layout.getPrimaryBase() == Base.Decl;
  370. Build(Val.getStructBase(Base.Index), Base.Decl, IsPrimaryBase,
  371. VTableClass, Offset + Base.Offset);
  372. }
  373. }
  374. unsigned FieldNo = 0;
  375. uint64_t OffsetBits = CGM.getContext().toBits(Offset);
  376. for (RecordDecl::field_iterator Field = RD->field_begin(),
  377. FieldEnd = RD->field_end(); Field != FieldEnd; ++Field, ++FieldNo) {
  378. // If this is a union, skip all the fields that aren't being initialized.
  379. if (RD->isUnion() && Val.getUnionField() != *Field)
  380. continue;
  381. // Don't emit anonymous bitfields, they just affect layout.
  382. if (Field->isUnnamedBitfield())
  383. continue;
  384. // Emit the value of the initializer.
  385. const APValue &FieldValue =
  386. RD->isUnion() ? Val.getUnionValue() : Val.getStructField(FieldNo);
  387. llvm::Constant *EltInit =
  388. CGM.EmitConstantValueForMemory(FieldValue, Field->getType(), CGF);
  389. assert(EltInit && "EmitConstantValue can't fail");
  390. if (!Field->isBitField()) {
  391. // Handle non-bitfield members.
  392. AppendField(*Field, Layout.getFieldOffset(FieldNo) + OffsetBits, EltInit);
  393. } else {
  394. // Otherwise we have a bitfield.
  395. AppendBitField(*Field, Layout.getFieldOffset(FieldNo) + OffsetBits,
  396. cast<llvm::ConstantInt>(EltInit));
  397. }
  398. }
  399. }
  400. llvm::Constant *ConstStructBuilder::Finalize(QualType Ty) {
  401. RecordDecl *RD = Ty->getAs<RecordType>()->getDecl();
  402. const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
  403. CharUnits LayoutSizeInChars = Layout.getSize();
  404. if (NextFieldOffsetInChars > LayoutSizeInChars) {
  405. // If the struct is bigger than the size of the record type,
  406. // we must have a flexible array member at the end.
  407. assert(RD->hasFlexibleArrayMember() &&
  408. "Must have flexible array member if struct is bigger than type!");
  409. // No tail padding is necessary.
  410. } else {
  411. // Append tail padding if necessary.
  412. CharUnits LLVMSizeInChars =
  413. NextFieldOffsetInChars.RoundUpToAlignment(LLVMStructAlignment);
  414. if (LLVMSizeInChars != LayoutSizeInChars)
  415. AppendTailPadding(LayoutSizeInChars);
  416. LLVMSizeInChars =
  417. NextFieldOffsetInChars.RoundUpToAlignment(LLVMStructAlignment);
  418. // Check if we need to convert the struct to a packed struct.
  419. if (NextFieldOffsetInChars <= LayoutSizeInChars &&
  420. LLVMSizeInChars > LayoutSizeInChars) {
  421. assert(!Packed && "Size mismatch!");
  422. ConvertStructToPacked();
  423. assert(NextFieldOffsetInChars <= LayoutSizeInChars &&
  424. "Converting to packed did not help!");
  425. }
  426. LLVMSizeInChars =
  427. NextFieldOffsetInChars.RoundUpToAlignment(LLVMStructAlignment);
  428. assert(LayoutSizeInChars == LLVMSizeInChars &&
  429. "Tail padding mismatch!");
  430. }
  431. // Pick the type to use. If the type is layout identical to the ConvertType
  432. // type then use it, otherwise use whatever the builder produced for us.
  433. llvm::StructType *STy =
  434. llvm::ConstantStruct::getTypeForElements(CGM.getLLVMContext(),
  435. Elements, Packed);
  436. llvm::Type *ValTy = CGM.getTypes().ConvertType(Ty);
  437. if (llvm::StructType *ValSTy = dyn_cast<llvm::StructType>(ValTy)) {
  438. if (ValSTy->isLayoutIdentical(STy))
  439. STy = ValSTy;
  440. }
  441. llvm::Constant *Result = llvm::ConstantStruct::get(STy, Elements);
  442. assert(NextFieldOffsetInChars.RoundUpToAlignment(getAlignment(Result)) ==
  443. getSizeInChars(Result) && "Size mismatch!");
  444. return Result;
  445. }
  446. llvm::Constant *ConstStructBuilder::BuildStruct(CodeGenModule &CGM,
  447. CodeGenFunction *CGF,
  448. ConstExprEmitter *Emitter,
  449. llvm::ConstantStruct *Base,
  450. InitListExpr *Updater) {
  451. ConstStructBuilder Builder(CGM, CGF);
  452. if (!Builder.Build(Emitter, Base, Updater))
  453. return nullptr;
  454. return Builder.Finalize(Updater->getType());
  455. }
  456. llvm::Constant *ConstStructBuilder::BuildStruct(CodeGenModule &CGM,
  457. CodeGenFunction *CGF,
  458. InitListExpr *ILE) {
  459. ConstStructBuilder Builder(CGM, CGF);
  460. if (!Builder.Build(ILE))
  461. return nullptr;
  462. return Builder.Finalize(ILE->getType());
  463. }
  464. llvm::Constant *ConstStructBuilder::BuildStruct(CodeGenModule &CGM,
  465. CodeGenFunction *CGF,
  466. const APValue &Val,
  467. QualType ValTy) {
  468. ConstStructBuilder Builder(CGM, CGF);
  469. const RecordDecl *RD = ValTy->castAs<RecordType>()->getDecl();
  470. const CXXRecordDecl *CD = dyn_cast<CXXRecordDecl>(RD);
  471. Builder.Build(Val, RD, false, CD, CharUnits::Zero());
  472. return Builder.Finalize(ValTy);
  473. }
  474. //===----------------------------------------------------------------------===//
  475. // ConstExprEmitter
  476. //===----------------------------------------------------------------------===//
  477. /// This class only needs to handle two cases:
  478. /// 1) Literals (this is used by APValue emission to emit literals).
  479. /// 2) Arrays, structs and unions (outside C++11 mode, we don't currently
  480. /// constant fold these types).
  481. class ConstExprEmitter :
  482. public StmtVisitor<ConstExprEmitter, llvm::Constant*> {
  483. CodeGenModule &CGM;
  484. CodeGenFunction *CGF;
  485. llvm::LLVMContext &VMContext;
  486. public:
  487. ConstExprEmitter(CodeGenModule &cgm, CodeGenFunction *cgf)
  488. : CGM(cgm), CGF(cgf), VMContext(cgm.getLLVMContext()) {
  489. }
  490. //===--------------------------------------------------------------------===//
  491. // Visitor Methods
  492. //===--------------------------------------------------------------------===//
  493. llvm::Constant *VisitStmt(Stmt *S) {
  494. return nullptr;
  495. }
  496. llvm::Constant *VisitParenExpr(ParenExpr *PE) {
  497. return Visit(PE->getSubExpr());
  498. }
  499. llvm::Constant *
  500. VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *PE) {
  501. return Visit(PE->getReplacement());
  502. }
  503. llvm::Constant *VisitGenericSelectionExpr(GenericSelectionExpr *GE) {
  504. return Visit(GE->getResultExpr());
  505. }
  506. llvm::Constant *VisitChooseExpr(ChooseExpr *CE) {
  507. return Visit(CE->getChosenSubExpr());
  508. }
  509. llvm::Constant *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
  510. return Visit(E->getInitializer());
  511. }
  512. llvm::Constant *VisitCastExpr(CastExpr* E) {
  513. Expr *subExpr = E->getSubExpr();
  514. llvm::Constant *C = CGM.EmitConstantExpr(subExpr, subExpr->getType(), CGF);
  515. if (!C) return nullptr;
  516. llvm::Type *destType = ConvertType(E->getType());
  517. switch (E->getCastKind()) {
  518. case CK_ToUnion: {
  519. // GCC cast to union extension
  520. assert(E->getType()->isUnionType() &&
  521. "Destination type is not union type!");
  522. // Build a struct with the union sub-element as the first member,
  523. // and padded to the appropriate size
  524. SmallVector<llvm::Constant*, 2> Elts;
  525. SmallVector<llvm::Type*, 2> Types;
  526. Elts.push_back(C);
  527. Types.push_back(C->getType());
  528. unsigned CurSize = CGM.getDataLayout().getTypeAllocSize(C->getType());
  529. unsigned TotalSize = CGM.getDataLayout().getTypeAllocSize(destType);
  530. assert(CurSize <= TotalSize && "Union size mismatch!");
  531. if (unsigned NumPadBytes = TotalSize - CurSize) {
  532. llvm::Type *Ty = CGM.Int8Ty;
  533. if (NumPadBytes > 1)
  534. Ty = llvm::ArrayType::get(Ty, NumPadBytes);
  535. Elts.push_back(llvm::UndefValue::get(Ty));
  536. Types.push_back(Ty);
  537. }
  538. llvm::StructType* STy =
  539. llvm::StructType::get(C->getType()->getContext(), Types, false);
  540. return llvm::ConstantStruct::get(STy, Elts);
  541. }
  542. case CK_AddressSpaceConversion:
  543. return llvm::ConstantExpr::getAddrSpaceCast(C, destType);
  544. case CK_LValueToRValue:
  545. case CK_AtomicToNonAtomic:
  546. case CK_NonAtomicToAtomic:
  547. case CK_NoOp:
  548. case CK_ConstructorConversion:
  549. return C;
  550. case CK_Dependent: llvm_unreachable("saw dependent cast!");
  551. case CK_BuiltinFnToFnPtr:
  552. llvm_unreachable("builtin functions are handled elsewhere");
  553. case CK_ReinterpretMemberPointer:
  554. case CK_DerivedToBaseMemberPointer:
  555. case CK_BaseToDerivedMemberPointer:
  556. return CGM.getCXXABI().EmitMemberPointerConversion(E, C);
  557. // These will never be supported.
  558. case CK_ObjCObjectLValueCast:
  559. case CK_ARCProduceObject:
  560. case CK_ARCConsumeObject:
  561. case CK_ARCReclaimReturnedObject:
  562. case CK_ARCExtendBlockObject:
  563. case CK_CopyAndAutoreleaseBlockObject:
  564. return nullptr;
  565. // These don't need to be handled here because Evaluate knows how to
  566. // evaluate them in the cases where they can be folded.
  567. case CK_BitCast:
  568. case CK_ToVoid:
  569. case CK_Dynamic:
  570. case CK_LValueBitCast:
  571. case CK_NullToMemberPointer:
  572. case CK_UserDefinedConversion:
  573. case CK_CPointerToObjCPointerCast:
  574. case CK_BlockPointerToObjCPointerCast:
  575. case CK_AnyPointerToBlockPointerCast:
  576. case CK_ArrayToPointerDecay:
  577. case CK_FunctionToPointerDecay:
  578. case CK_BaseToDerived:
  579. case CK_DerivedToBase:
  580. case CK_UncheckedDerivedToBase:
  581. case CK_MemberPointerToBoolean:
  582. case CK_VectorSplat:
  583. case CK_FloatingRealToComplex:
  584. case CK_FloatingComplexToReal:
  585. case CK_FloatingComplexToBoolean:
  586. case CK_FloatingComplexCast:
  587. case CK_FloatingComplexToIntegralComplex:
  588. case CK_IntegralRealToComplex:
  589. case CK_IntegralComplexToReal:
  590. case CK_IntegralComplexToBoolean:
  591. case CK_IntegralComplexCast:
  592. case CK_IntegralComplexToFloatingComplex:
  593. case CK_PointerToIntegral:
  594. case CK_PointerToBoolean:
  595. case CK_NullToPointer:
  596. case CK_IntegralCast:
  597. case CK_IntegralToPointer:
  598. case CK_IntegralToBoolean:
  599. case CK_IntegralToFloating:
  600. case CK_FloatingToIntegral:
  601. case CK_FloatingToBoolean:
  602. case CK_FloatingCast:
  603. case CK_ZeroToOCLEvent:
  604. return nullptr;
  605. // HLSL Change Begins.
  606. case CK_HLSLCC_FloatingCast:
  607. case CK_HLSLCC_IntegralCast:
  608. case CK_HLSLCC_IntegralToBoolean:
  609. case CK_HLSLCC_IntegralToFloating:
  610. case CK_HLSLCC_FloatingToIntegral:
  611. case CK_HLSLCC_FloatingToBoolean:
  612. // Since these cast kinds have already been handled in ExprConstant.cpp,
  613. // we can reuse the logic there.
  614. return CGM.EmitConstantExpr(E, E->getType(), CGF);
  615. case CK_FlatConversion:
  616. return nullptr;
  617. case CK_HLSLVectorSplat: {
  618. unsigned vecSize = hlsl::GetHLSLVecSize(E->getType());
  619. return llvm::ConstantVector::getSplat(vecSize, C);
  620. }
  621. case CK_HLSLMatrixSplat: {
  622. llvm::StructType *ST =
  623. cast<llvm::StructType>(CGM.getTypes().ConvertType(E->getType()));
  624. unsigned row,col;
  625. hlsl::GetHLSLMatRowColCount(E->getType(), row, col);
  626. llvm::Constant *Row = llvm::ConstantVector::getSplat(col, C);
  627. std::vector<llvm::Constant *> Rows(row, Row);
  628. llvm::Constant *Mat = llvm::ConstantArray::get(
  629. cast<llvm::ArrayType>(ST->getElementType(0)), Rows);
  630. return llvm::ConstantStruct::get(ST, Mat);
  631. }
  632. case CK_HLSLVectorTruncationCast: {
  633. unsigned vecSize = hlsl::GetHLSLVecSize(E->getType());
  634. SmallVector<llvm::Constant*, 4> Elts(vecSize);
  635. if (llvm::ConstantDataVector *CDV = dyn_cast<llvm::ConstantDataVector>(C)) {
  636. for (unsigned i = 0; i < vecSize; i++)
  637. Elts[i] = CDV->getElementAsConstant(i);
  638. } else if (llvm::ConstantVector* CV = dyn_cast<llvm::ConstantVector>(C)) {
  639. for (unsigned i = 0; i < vecSize; i++)
  640. Elts[i] = CV->getOperand(i);
  641. } else {
  642. llvm::ConstantAggregateZero* CAZ = cast<llvm::ConstantAggregateZero>(C);
  643. for (unsigned i = 0; i < vecSize; i++)
  644. Elts[i] = CAZ->getElementValue(i);
  645. }
  646. return llvm::ConstantVector::get(Elts);
  647. }
  648. case CK_HLSLVectorToScalarCast: {
  649. if (llvm::ConstantDataVector* CDV = dyn_cast<llvm::ConstantDataVector>(C)) {
  650. return CDV->getElementAsConstant(0);
  651. }
  652. else if (llvm::ConstantVector* CV = dyn_cast<llvm::ConstantVector>(C)) {
  653. return CV->getOperand(0);
  654. } else {
  655. llvm::ConstantAggregateZero* CAZ = cast<llvm::ConstantAggregateZero>(C);
  656. return CAZ->getElementValue((unsigned)0);
  657. }
  658. }
  659. case CK_HLSLMatrixTruncationCast: {
  660. llvm::StructType *ST =
  661. cast<llvm::StructType>(CGM.getTypes().ConvertType(E->getType()));
  662. unsigned rowCt,colCt;
  663. hlsl::GetHLSLMatRowColCount(E->getType(), rowCt, colCt);
  664. if (llvm::ConstantStruct *CS = dyn_cast<llvm::ConstantStruct>(C)) {
  665. llvm::ConstantArray *CA = dyn_cast<llvm::ConstantArray>(CS->getOperand(0));
  666. SmallVector<llvm::Constant *, 4> Rows(rowCt);
  667. for (unsigned i = 0; i < rowCt; i++) {
  668. SmallVector<llvm::Constant*, 4> Elts(colCt);
  669. if (llvm::ConstantDataVector *CDV = dyn_cast<llvm::ConstantDataVector>(CA->getOperand(i))) {
  670. for (unsigned j = 0; j < colCt; j++)
  671. Elts[j] = CDV->getElementAsConstant(j);
  672. } else {
  673. llvm::ConstantVector *CV = cast<llvm::ConstantVector>(CA->getOperand(i));
  674. for (unsigned j = 0; j < colCt; j++)
  675. Elts[j] = CV->getOperand(j);
  676. }
  677. Rows[i] = llvm::ConstantVector::get(Elts);
  678. }
  679. llvm::Constant *Mat = llvm::ConstantArray::get(
  680. cast<llvm::ArrayType>(ST->getElementType(0)), Rows);
  681. return llvm::ConstantStruct::get(ST, Mat);
  682. }
  683. }
  684. // HLSL Change Ends.
  685. }
  686. llvm_unreachable("Invalid CastKind");
  687. }
  688. llvm::Constant *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
  689. return Visit(DAE->getExpr());
  690. }
  691. llvm::Constant *VisitCXXDefaultInitExpr(CXXDefaultInitExpr *DIE) {
  692. // No need for a DefaultInitExprScope: we don't handle 'this' in a
  693. // constant expression.
  694. return Visit(DIE->getExpr());
  695. }
  696. llvm::Constant *VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E) {
  697. return Visit(E->GetTemporaryExpr());
  698. }
  699. llvm::Constant *EmitArrayInitialization(InitListExpr *ILE) {
  700. if (ILE->isStringLiteralInit())
  701. return Visit(ILE->getInit(0));
  702. if (CGM.getLangOpts().HLSL) return nullptr; // HLSL Change - Not implement yet.
  703. llvm::ArrayType *AType =
  704. cast<llvm::ArrayType>(ConvertType(ILE->getType()));
  705. llvm::Type *ElemTy = AType->getElementType();
  706. unsigned NumInitElements = ILE->getNumInits();
  707. unsigned NumElements = AType->getNumElements();
  708. // Initialising an array requires us to automatically
  709. // initialise any elements that have not been initialised explicitly
  710. unsigned NumInitableElts = std::min(NumInitElements, NumElements);
  711. // Initialize remaining array elements.
  712. // FIXME: This doesn't handle member pointers correctly!
  713. llvm::Constant *fillC;
  714. if (Expr *filler = ILE->getArrayFiller())
  715. fillC = CGM.EmitConstantExpr(filler, filler->getType(), CGF);
  716. else
  717. fillC = llvm::Constant::getNullValue(ElemTy);
  718. if (!fillC)
  719. return nullptr;
  720. // Try to use a ConstantAggregateZero if we can.
  721. if (fillC->isNullValue() && !NumInitableElts)
  722. return llvm::ConstantAggregateZero::get(AType);
  723. // Copy initializer elements.
  724. std::vector<llvm::Constant*> Elts;
  725. Elts.reserve(NumInitableElts + NumElements);
  726. bool RewriteType = false;
  727. for (unsigned i = 0; i < NumInitableElts; ++i) {
  728. Expr *Init = ILE->getInit(i);
  729. llvm::Constant *C = CGM.EmitConstantExpr(Init, Init->getType(), CGF);
  730. if (!C)
  731. return nullptr;
  732. RewriteType |= (C->getType() != ElemTy);
  733. Elts.push_back(C);
  734. }
  735. RewriteType |= (fillC->getType() != ElemTy);
  736. Elts.resize(NumElements, fillC);
  737. if (RewriteType) {
  738. // FIXME: Try to avoid packing the array
  739. std::vector<llvm::Type*> Types;
  740. Types.reserve(NumInitableElts + NumElements);
  741. for (unsigned i = 0, e = Elts.size(); i < e; ++i)
  742. Types.push_back(Elts[i]->getType());
  743. llvm::StructType *SType = llvm::StructType::get(AType->getContext(),
  744. Types, true);
  745. return llvm::ConstantStruct::get(SType, Elts);
  746. }
  747. return llvm::ConstantArray::get(AType, Elts);
  748. }
  749. llvm::Constant *EmitRecordInitialization(InitListExpr *ILE) {
  750. return ConstStructBuilder::BuildStruct(CGM, CGF, ILE);
  751. }
  752. llvm::Constant *VisitImplicitValueInitExpr(ImplicitValueInitExpr* E) {
  753. return CGM.EmitNullConstant(E->getType());
  754. }
  755. llvm::Constant *VisitInitListExpr(InitListExpr *ILE) {
  756. // HLSL Change Begins.
  757. if (CGM.getLangOpts().HLSL)
  758. return CGM.getHLSLRuntime().EmitHLSLConstInitListExpr(CGM, ILE);
  759. // HLSL Change Ends.
  760. if (ILE->getType()->isArrayType())
  761. return EmitArrayInitialization(ILE);
  762. if (ILE->getType()->isRecordType())
  763. return EmitRecordInitialization(ILE);
  764. return nullptr;
  765. }
  766. llvm::Constant *EmitDesignatedInitUpdater(llvm::Constant *Base,
  767. InitListExpr *Updater) {
  768. QualType ExprType = Updater->getType();
  769. if (ExprType->isArrayType()) {
  770. llvm::ArrayType *AType = cast<llvm::ArrayType>(ConvertType(ExprType));
  771. llvm::Type *ElemType = AType->getElementType();
  772. unsigned NumInitElements = Updater->getNumInits();
  773. unsigned NumElements = AType->getNumElements();
  774. std::vector<llvm::Constant *> Elts;
  775. Elts.reserve(NumElements);
  776. if (llvm::ConstantDataArray *DataArray =
  777. dyn_cast<llvm::ConstantDataArray>(Base))
  778. for (unsigned i = 0; i != NumElements; ++i)
  779. Elts.push_back(DataArray->getElementAsConstant(i));
  780. else if (llvm::ConstantArray *Array =
  781. dyn_cast<llvm::ConstantArray>(Base))
  782. for (unsigned i = 0; i != NumElements; ++i)
  783. Elts.push_back(Array->getOperand(i));
  784. else
  785. return nullptr; // FIXME: other array types not implemented
  786. llvm::Constant *fillC = nullptr;
  787. if (Expr *filler = Updater->getArrayFiller())
  788. if (!isa<NoInitExpr>(filler))
  789. fillC = CGM.EmitConstantExpr(filler, filler->getType(), CGF);
  790. bool RewriteType = (fillC && fillC->getType() != ElemType);
  791. for (unsigned i = 0; i != NumElements; ++i) {
  792. Expr *Init = nullptr;
  793. if (i < NumInitElements)
  794. Init = Updater->getInit(i);
  795. if (!Init && fillC)
  796. Elts[i] = fillC;
  797. else if (!Init || isa<NoInitExpr>(Init))
  798. ; // Do nothing.
  799. else if (InitListExpr *ChildILE = dyn_cast<InitListExpr>(Init))
  800. Elts[i] = EmitDesignatedInitUpdater(Elts[i], ChildILE);
  801. else
  802. Elts[i] = CGM.EmitConstantExpr(Init, Init->getType(), CGF);
  803. if (!Elts[i])
  804. return nullptr;
  805. RewriteType |= (Elts[i]->getType() != ElemType);
  806. }
  807. if (RewriteType) {
  808. std::vector<llvm::Type *> Types;
  809. Types.reserve(NumElements);
  810. for (unsigned i = 0; i != NumElements; ++i)
  811. Types.push_back(Elts[i]->getType());
  812. llvm::StructType *SType = llvm::StructType::get(AType->getContext(),
  813. Types, true);
  814. return llvm::ConstantStruct::get(SType, Elts);
  815. }
  816. return llvm::ConstantArray::get(AType, Elts);
  817. }
  818. if (ExprType->isRecordType())
  819. return ConstStructBuilder::BuildStruct(CGM, CGF, this,
  820. dyn_cast<llvm::ConstantStruct>(Base), Updater);
  821. return nullptr;
  822. }
  823. llvm::Constant *VisitDesignatedInitUpdateExpr(DesignatedInitUpdateExpr *E) {
  824. return EmitDesignatedInitUpdater(
  825. CGM.EmitConstantExpr(E->getBase(), E->getType(), CGF),
  826. E->getUpdater());
  827. }
  828. llvm::Constant *VisitCXXConstructExpr(CXXConstructExpr *E) {
  829. if (!E->getConstructor()->isTrivial())
  830. return nullptr;
  831. QualType Ty = E->getType();
  832. // FIXME: We should not have to call getBaseElementType here.
  833. const RecordType *RT =
  834. CGM.getContext().getBaseElementType(Ty)->getAs<RecordType>();
  835. const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
  836. // If the class doesn't have a trivial destructor, we can't emit it as a
  837. // constant expr.
  838. if (!RD->hasTrivialDestructor())
  839. return nullptr;
  840. // Only copy and default constructors can be trivial.
  841. if (E->getNumArgs()) {
  842. assert(E->getNumArgs() == 1 && "trivial ctor with > 1 argument");
  843. assert(E->getConstructor()->isCopyOrMoveConstructor() &&
  844. "trivial ctor has argument but isn't a copy/move ctor");
  845. Expr *Arg = E->getArg(0);
  846. assert(CGM.getContext().hasSameUnqualifiedType(Ty, Arg->getType()) &&
  847. "argument to copy ctor is of wrong type");
  848. return Visit(Arg);
  849. }
  850. return CGM.EmitNullConstant(Ty);
  851. }
  852. llvm::Constant *VisitStringLiteral(StringLiteral *E) {
  853. return CGM.GetConstantArrayFromStringLiteral(E);
  854. }
  855. llvm::Constant *VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
  856. // This must be an @encode initializing an array in a static initializer.
  857. // Don't emit it as the address of the string, emit the string data itself
  858. // as an inline array.
  859. std::string Str;
  860. CGM.getContext().getObjCEncodingForType(E->getEncodedType(), Str);
  861. QualType T = E->getType();
  862. if (T->getTypeClass() == Type::TypeOfExpr)
  863. T = cast<TypeOfExprType>(T)->getUnderlyingExpr()->getType();
  864. const ConstantArrayType *CAT = cast<ConstantArrayType>(T);
  865. // Resize the string to the right size, adding zeros at the end, or
  866. // truncating as needed.
  867. Str.resize(CAT->getSize().getZExtValue(), '\0');
  868. return llvm::ConstantDataArray::getString(VMContext, Str, false);
  869. }
  870. llvm::Constant *VisitUnaryExtension(const UnaryOperator *E) {
  871. return Visit(E->getSubExpr());
  872. }
  873. // Utility methods
  874. llvm::Type *ConvertType(QualType T) {
  875. return CGM.getTypes().ConvertType(T);
  876. }
  877. public:
  878. llvm::Constant *EmitLValue(APValue::LValueBase LVBase) {
  879. if (const ValueDecl *Decl = LVBase.dyn_cast<const ValueDecl*>()) {
  880. if (Decl->hasAttr<WeakRefAttr>())
  881. return CGM.GetWeakRefReference(Decl);
  882. if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Decl))
  883. return CGM.GetAddrOfFunction(FD);
  884. if (const VarDecl* VD = dyn_cast<VarDecl>(Decl)) {
  885. // We can never refer to a variable with local storage.
  886. if (!VD->hasLocalStorage()) {
  887. if (VD->isFileVarDecl() || VD->hasExternalStorage())
  888. return CGM.GetAddrOfGlobalVar(VD);
  889. else if (VD->isLocalVarDecl())
  890. return CGM.getOrCreateStaticVarDecl(
  891. *VD, CGM.getLLVMLinkageVarDefinition(VD, /*isConstant=*/false));
  892. }
  893. }
  894. return nullptr;
  895. }
  896. Expr *E = const_cast<Expr*>(LVBase.get<const Expr*>());
  897. switch (E->getStmtClass()) {
  898. default: break;
  899. case Expr::CompoundLiteralExprClass: {
  900. // Note that due to the nature of compound literals, this is guaranteed
  901. // to be the only use of the variable, so we just generate it here.
  902. CompoundLiteralExpr *CLE = cast<CompoundLiteralExpr>(E);
  903. llvm::Constant* C = CGM.EmitConstantExpr(CLE->getInitializer(),
  904. CLE->getType(), CGF);
  905. // FIXME: "Leaked" on failure.
  906. if (C)
  907. C = new llvm::GlobalVariable(CGM.getModule(), C->getType(),
  908. E->getType().isConstant(CGM.getContext()),
  909. llvm::GlobalValue::InternalLinkage,
  910. C, ".compoundliteral", nullptr,
  911. llvm::GlobalVariable::NotThreadLocal,
  912. CGM.getContext().getTargetAddressSpace(E->getType()));
  913. return C;
  914. }
  915. case Expr::StringLiteralClass:
  916. return CGM.GetAddrOfConstantStringFromLiteral(cast<StringLiteral>(E));
  917. case Expr::ObjCEncodeExprClass:
  918. return CGM.GetAddrOfConstantStringFromObjCEncode(cast<ObjCEncodeExpr>(E));
  919. case Expr::ObjCStringLiteralClass: {
  920. ObjCStringLiteral* SL = cast<ObjCStringLiteral>(E);
  921. llvm::Constant *C =
  922. CGM.getObjCRuntime().GenerateConstantString(SL->getString());
  923. return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
  924. }
  925. case Expr::PredefinedExprClass: {
  926. unsigned Type = cast<PredefinedExpr>(E)->getIdentType();
  927. if (CGF) {
  928. LValue Res = CGF->EmitPredefinedLValue(cast<PredefinedExpr>(E));
  929. return cast<llvm::Constant>(Res.getAddress());
  930. } else if (Type == PredefinedExpr::PrettyFunction) {
  931. return CGM.GetAddrOfConstantCString("top level", ".tmp");
  932. }
  933. return CGM.GetAddrOfConstantCString("", ".tmp");
  934. }
  935. case Expr::AddrLabelExprClass: {
  936. assert(CGF && "Invalid address of label expression outside function.");
  937. llvm::Constant *Ptr =
  938. CGF->GetAddrOfLabel(cast<AddrLabelExpr>(E)->getLabel());
  939. return llvm::ConstantExpr::getBitCast(Ptr, ConvertType(E->getType()));
  940. }
  941. case Expr::CallExprClass: {
  942. CallExpr* CE = cast<CallExpr>(E);
  943. unsigned builtin = CE->getBuiltinCallee();
  944. if (builtin !=
  945. Builtin::BI__builtin___CFStringMakeConstantString &&
  946. builtin !=
  947. Builtin::BI__builtin___NSStringMakeConstantString)
  948. break;
  949. const Expr *Arg = CE->getArg(0)->IgnoreParenCasts();
  950. const StringLiteral *Literal = cast<StringLiteral>(Arg);
  951. if (builtin ==
  952. Builtin::BI__builtin___NSStringMakeConstantString) {
  953. return CGM.getObjCRuntime().GenerateConstantString(Literal);
  954. }
  955. // FIXME: need to deal with UCN conversion issues.
  956. return CGM.GetAddrOfConstantCFString(Literal);
  957. }
  958. case Expr::BlockExprClass: {
  959. std::string FunctionName;
  960. if (CGF)
  961. FunctionName = CGF->CurFn->getName();
  962. else
  963. FunctionName = "global";
  964. return CGM.GetAddrOfGlobalBlock(cast<BlockExpr>(E), FunctionName.c_str());
  965. }
  966. case Expr::CXXTypeidExprClass: {
  967. CXXTypeidExpr *Typeid = cast<CXXTypeidExpr>(E);
  968. QualType T;
  969. if (Typeid->isTypeOperand())
  970. T = Typeid->getTypeOperand(CGM.getContext());
  971. else
  972. T = Typeid->getExprOperand()->getType();
  973. return CGM.GetAddrOfRTTIDescriptor(T);
  974. }
  975. case Expr::CXXUuidofExprClass: {
  976. return CGM.GetAddrOfUuidDescriptor(cast<CXXUuidofExpr>(E));
  977. }
  978. case Expr::MaterializeTemporaryExprClass: {
  979. MaterializeTemporaryExpr *MTE = cast<MaterializeTemporaryExpr>(E);
  980. assert(MTE->getStorageDuration() == SD_Static);
  981. SmallVector<const Expr *, 2> CommaLHSs;
  982. SmallVector<SubobjectAdjustment, 2> Adjustments;
  983. const Expr *Inner = MTE->GetTemporaryExpr()
  984. ->skipRValueSubobjectAdjustments(CommaLHSs, Adjustments);
  985. return CGM.GetAddrOfGlobalTemporary(MTE, Inner);
  986. }
  987. }
  988. return nullptr;
  989. }
  990. };
  991. } // end anonymous namespace.
  992. bool ConstStructBuilder::Build(ConstExprEmitter *Emitter,
  993. llvm::ConstantStruct *Base,
  994. InitListExpr *Updater) {
  995. assert(Base && "base expression should not be empty");
  996. QualType ExprType = Updater->getType();
  997. RecordDecl *RD = ExprType->getAs<RecordType>()->getDecl();
  998. const ASTRecordLayout &Layout = CGM.getContext().getASTRecordLayout(RD);
  999. const llvm::StructLayout *BaseLayout = CGM.getDataLayout().getStructLayout(
  1000. Base->getType());
  1001. unsigned FieldNo = -1;
  1002. unsigned ElementNo = 0;
  1003. for (FieldDecl *Field : RD->fields()) {
  1004. ++FieldNo;
  1005. if (RD->isUnion() && Updater->getInitializedFieldInUnion() != Field)
  1006. continue;
  1007. // Skip anonymous bitfields.
  1008. if (Field->isUnnamedBitfield())
  1009. continue;
  1010. llvm::Constant *EltInit = Base->getOperand(ElementNo);
  1011. // Bail out if the type of the ConstantStruct does not have the same layout
  1012. // as the type of the InitListExpr.
  1013. if (CGM.getTypes().ConvertType(Field->getType()) != EltInit->getType() ||
  1014. Layout.getFieldOffset(ElementNo) !=
  1015. BaseLayout->getElementOffsetInBits(ElementNo))
  1016. return false;
  1017. // Get the initializer. If we encounter an empty field or a NoInitExpr,
  1018. // we use values from the base expression.
  1019. Expr *Init = nullptr;
  1020. if (ElementNo < Updater->getNumInits())
  1021. Init = Updater->getInit(ElementNo);
  1022. if (!Init || isa<NoInitExpr>(Init))
  1023. ; // Do nothing.
  1024. else if (InitListExpr *ChildILE = dyn_cast<InitListExpr>(Init))
  1025. EltInit = Emitter->EmitDesignatedInitUpdater(EltInit, ChildILE);
  1026. else
  1027. EltInit = CGM.EmitConstantExpr(Init, Field->getType(), CGF);
  1028. ++ElementNo;
  1029. if (!EltInit)
  1030. return false;
  1031. if (!Field->isBitField())
  1032. AppendField(Field, Layout.getFieldOffset(FieldNo), EltInit);
  1033. else if (llvm::ConstantInt *CI = dyn_cast<llvm::ConstantInt>(EltInit))
  1034. AppendBitField(Field, Layout.getFieldOffset(FieldNo), CI);
  1035. else
  1036. // Initializing a bitfield with a non-trivial constant?
  1037. return false;
  1038. }
  1039. return true;
  1040. }
  1041. llvm::Constant *CodeGenModule::EmitConstantInit(const VarDecl &D,
  1042. CodeGenFunction *CGF) {
  1043. // Make a quick check if variable can be default NULL initialized
  1044. // and avoid going through rest of code which may do, for c++11,
  1045. // initialization of memory to all NULLs.
  1046. if (!D.hasLocalStorage()) {
  1047. QualType Ty = D.getType();
  1048. if (Ty->isArrayType())
  1049. Ty = Context.getBaseElementType(Ty);
  1050. if (Ty->isRecordType())
  1051. if (const CXXConstructExpr *E =
  1052. dyn_cast_or_null<CXXConstructExpr>(D.getInit())) {
  1053. const CXXConstructorDecl *CD = E->getConstructor();
  1054. if (CD->isTrivial() && CD->isDefaultConstructor())
  1055. return EmitNullConstant(D.getType());
  1056. }
  1057. }
  1058. // HLSL Change Begin - External variable is in cbuffer, cannot use as immediate.
  1059. if (D.hasExternalFormalLinkage() &&
  1060. !isa<EnumConstantDecl>(&D))
  1061. return nullptr;
  1062. // HLSL Change End.
  1063. if (const APValue *Value = D.evaluateValue())
  1064. return EmitConstantValueForMemory(*Value, D.getType(), CGF);
  1065. // FIXME: Implement C++11 [basic.start.init]p2: if the initializer of a
  1066. // reference is a constant expression, and the reference binds to a temporary,
  1067. // then constant initialization is performed. ConstExprEmitter will
  1068. // incorrectly emit a prvalue constant in this case, and the calling code
  1069. // interprets that as the (pointer) value of the reference, rather than the
  1070. // desired value of the referee.
  1071. if (D.getType()->isReferenceType())
  1072. return nullptr;
  1073. const Expr *E = D.getInit();
  1074. assert(E && "No initializer to emit");
  1075. llvm::Constant* C = ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
  1076. if (C && C->getType()->getScalarType()->isIntegerTy(1)) { // HLSL Change
  1077. llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
  1078. C = llvm::ConstantExpr::getZExt(C, BoolTy);
  1079. }
  1080. return C;
  1081. }
  1082. llvm::Constant *CodeGenModule::EmitConstantExpr(const Expr *E,
  1083. QualType DestType,
  1084. CodeGenFunction *CGF) {
  1085. Expr::EvalResult Result;
  1086. bool Success = false;
  1087. if (DestType->isReferenceType())
  1088. Success = E->EvaluateAsLValue(Result, Context);
  1089. else
  1090. Success = E->EvaluateAsRValue(Result, Context);
  1091. llvm::Constant *C = nullptr;
  1092. if (Success && !Result.HasSideEffects)
  1093. C = EmitConstantValue(Result.Val, DestType, CGF);
  1094. else
  1095. C = ConstExprEmitter(*this, CGF).Visit(const_cast<Expr*>(E));
  1096. if (C && C->getType()->getScalarType()->isIntegerTy(1)) { // HLSL Change
  1097. llvm::Type *BoolTy = getTypes().ConvertTypeForMem(E->getType());
  1098. C = llvm::ConstantExpr::getZExt(C, BoolTy);
  1099. }
  1100. return C;
  1101. }
  1102. llvm::Constant *CodeGenModule::EmitConstantValue(const APValue &Value,
  1103. QualType DestType,
  1104. CodeGenFunction *CGF) {
  1105. // For an _Atomic-qualified constant, we may need to add tail padding.
  1106. if (auto *AT = DestType->getAs<AtomicType>()) {
  1107. QualType InnerType = AT->getValueType();
  1108. auto *Inner = EmitConstantValue(Value, InnerType, CGF);
  1109. uint64_t InnerSize = Context.getTypeSize(InnerType);
  1110. uint64_t OuterSize = Context.getTypeSize(DestType);
  1111. if (InnerSize == OuterSize)
  1112. return Inner;
  1113. assert(InnerSize < OuterSize && "emitted over-large constant for atomic");
  1114. llvm::Constant *Elts[] = {
  1115. Inner,
  1116. llvm::ConstantAggregateZero::get(
  1117. llvm::ArrayType::get(Int8Ty, (OuterSize - InnerSize) / 8))
  1118. };
  1119. return llvm::ConstantStruct::getAnon(Elts);
  1120. }
  1121. switch (Value.getKind()) {
  1122. case APValue::Uninitialized:
  1123. llvm_unreachable("Constant expressions should be initialized.");
  1124. case APValue::LValue: {
  1125. llvm::Type *DestTy = getTypes().ConvertTypeForMem(DestType);
  1126. llvm::Constant *Offset =
  1127. llvm::ConstantInt::get(Int64Ty, Value.getLValueOffset().getQuantity());
  1128. llvm::Constant *C;
  1129. if (APValue::LValueBase LVBase = Value.getLValueBase()) {
  1130. // An array can be represented as an lvalue referring to the base.
  1131. if (isa<llvm::ArrayType>(DestTy)) {
  1132. assert(Offset->isNullValue() && "offset on array initializer");
  1133. return ConstExprEmitter(*this, CGF).Visit(
  1134. const_cast<Expr*>(LVBase.get<const Expr*>()));
  1135. }
  1136. C = ConstExprEmitter(*this, CGF).EmitLValue(LVBase);
  1137. // Apply offset if necessary.
  1138. if (!Offset->isNullValue()) {
  1139. unsigned AS = C->getType()->getPointerAddressSpace();
  1140. llvm::Type *CharPtrTy = Int8Ty->getPointerTo(AS);
  1141. llvm::Constant *Casted = llvm::ConstantExpr::getBitCast(C, CharPtrTy);
  1142. Casted = llvm::ConstantExpr::getGetElementPtr(Int8Ty, Casted, Offset);
  1143. C = llvm::ConstantExpr::getPointerCast(Casted, C->getType());
  1144. }
  1145. // Convert to the appropriate type; this could be an lvalue for
  1146. // an integer.
  1147. if (isa<llvm::PointerType>(DestTy))
  1148. return llvm::ConstantExpr::getPointerCast(C, DestTy);
  1149. return llvm::ConstantExpr::getPtrToInt(C, DestTy);
  1150. } else {
  1151. C = Offset;
  1152. // Convert to the appropriate type; this could be an lvalue for
  1153. // an integer.
  1154. if (isa<llvm::PointerType>(DestTy))
  1155. return llvm::ConstantExpr::getIntToPtr(C, DestTy);
  1156. // If the types don't match this should only be a truncate.
  1157. if (C->getType() != DestTy)
  1158. return llvm::ConstantExpr::getTrunc(C, DestTy);
  1159. return C;
  1160. }
  1161. }
  1162. case APValue::Int:
  1163. return llvm::ConstantInt::get(VMContext, Value.getInt());
  1164. case APValue::ComplexInt: {
  1165. llvm::Constant *Complex[2];
  1166. Complex[0] = llvm::ConstantInt::get(VMContext,
  1167. Value.getComplexIntReal());
  1168. Complex[1] = llvm::ConstantInt::get(VMContext,
  1169. Value.getComplexIntImag());
  1170. // FIXME: the target may want to specify that this is packed.
  1171. llvm::StructType *STy = llvm::StructType::get(Complex[0]->getType(),
  1172. Complex[1]->getType(),
  1173. nullptr);
  1174. return llvm::ConstantStruct::get(STy, Complex);
  1175. }
  1176. case APValue::Float: {
  1177. const llvm::APFloat &Init = Value.getFloat();
  1178. // OACR error 6287
  1179. #pragma prefast(disable: __WARNING_REDUNDANTTEST, "language options are constants, by design")
  1180. if (&Init.getSemantics() == &llvm::APFloat::IEEEhalf &&
  1181. !Context.getLangOpts().NativeHalfType &&
  1182. !Context.getLangOpts().HalfArgsAndReturns)
  1183. return llvm::ConstantInt::get(VMContext, Init.bitcastToAPInt());
  1184. else
  1185. return llvm::ConstantFP::get(VMContext, Init);
  1186. }
  1187. case APValue::ComplexFloat: {
  1188. llvm::Constant *Complex[2];
  1189. Complex[0] = llvm::ConstantFP::get(VMContext,
  1190. Value.getComplexFloatReal());
  1191. Complex[1] = llvm::ConstantFP::get(VMContext,
  1192. Value.getComplexFloatImag());
  1193. // FIXME: the target may want to specify that this is packed.
  1194. llvm::StructType *STy = llvm::StructType::get(Complex[0]->getType(),
  1195. Complex[1]->getType(),
  1196. nullptr);
  1197. return llvm::ConstantStruct::get(STy, Complex);
  1198. }
  1199. case APValue::Vector: {
  1200. SmallVector<llvm::Constant *, 4> Inits;
  1201. unsigned NumElts = Value.getVectorLength();
  1202. for (unsigned i = 0; i != NumElts; ++i) {
  1203. const APValue &Elt = Value.getVectorElt(i);
  1204. if (Elt.isInt())
  1205. Inits.push_back(llvm::ConstantInt::get(VMContext, Elt.getInt()));
  1206. else
  1207. Inits.push_back(llvm::ConstantFP::get(VMContext, Elt.getFloat()));
  1208. }
  1209. return llvm::ConstantVector::get(Inits);
  1210. }
  1211. case APValue::AddrLabelDiff: {
  1212. const AddrLabelExpr *LHSExpr = Value.getAddrLabelDiffLHS();
  1213. const AddrLabelExpr *RHSExpr = Value.getAddrLabelDiffRHS();
  1214. llvm::Constant *LHS = EmitConstantExpr(LHSExpr, LHSExpr->getType(), CGF);
  1215. llvm::Constant *RHS = EmitConstantExpr(RHSExpr, RHSExpr->getType(), CGF);
  1216. // Compute difference
  1217. llvm::Type *ResultType = getTypes().ConvertType(DestType);
  1218. LHS = llvm::ConstantExpr::getPtrToInt(LHS, IntPtrTy);
  1219. RHS = llvm::ConstantExpr::getPtrToInt(RHS, IntPtrTy);
  1220. llvm::Constant *AddrLabelDiff = llvm::ConstantExpr::getSub(LHS, RHS);
  1221. // LLVM is a bit sensitive about the exact format of the
  1222. // address-of-label difference; make sure to truncate after
  1223. // the subtraction.
  1224. return llvm::ConstantExpr::getTruncOrBitCast(AddrLabelDiff, ResultType);
  1225. }
  1226. case APValue::Struct:
  1227. case APValue::Union:
  1228. return ConstStructBuilder::BuildStruct(*this, CGF, Value, DestType);
  1229. case APValue::Array: {
  1230. const ArrayType *CAT = Context.getAsArrayType(DestType);
  1231. unsigned NumElements = Value.getArraySize();
  1232. unsigned NumInitElts = Value.getArrayInitializedElts();
  1233. // Emit array filler, if there is one.
  1234. llvm::Constant *Filler = nullptr;
  1235. if (Value.hasArrayFiller())
  1236. Filler = EmitConstantValueForMemory(Value.getArrayFiller(),
  1237. CAT->getElementType(), CGF);
  1238. // Emit initializer elements.
  1239. llvm::Type *CommonElementType =
  1240. getTypes().ConvertType(CAT->getElementType());
  1241. // Try to use a ConstantAggregateZero if we can.
  1242. if (Filler && Filler->isNullValue() && !NumInitElts) {
  1243. llvm::ArrayType *AType =
  1244. llvm::ArrayType::get(CommonElementType, NumElements);
  1245. return llvm::ConstantAggregateZero::get(AType);
  1246. }
  1247. std::vector<llvm::Constant*> Elts;
  1248. Elts.reserve(NumElements);
  1249. for (unsigned I = 0; I < NumElements; ++I) {
  1250. llvm::Constant *C = Filler;
  1251. if (I < NumInitElts)
  1252. C = EmitConstantValueForMemory(Value.getArrayInitializedElt(I),
  1253. CAT->getElementType(), CGF);
  1254. else
  1255. assert(Filler && "Missing filler for implicit elements of initializer");
  1256. if (I == 0)
  1257. CommonElementType = C->getType();
  1258. else if (C->getType() != CommonElementType)
  1259. CommonElementType = nullptr;
  1260. Elts.push_back(C);
  1261. }
  1262. if (!CommonElementType) {
  1263. // FIXME: Try to avoid packing the array
  1264. std::vector<llvm::Type*> Types;
  1265. Types.reserve(NumElements);
  1266. for (unsigned i = 0, e = Elts.size(); i < e; ++i)
  1267. Types.push_back(Elts[i]->getType());
  1268. llvm::StructType *SType = llvm::StructType::get(VMContext, Types, true);
  1269. return llvm::ConstantStruct::get(SType, Elts);
  1270. }
  1271. llvm::ArrayType *AType =
  1272. llvm::ArrayType::get(CommonElementType, NumElements);
  1273. return llvm::ConstantArray::get(AType, Elts);
  1274. }
  1275. case APValue::MemberPointer:
  1276. return getCXXABI().EmitMemberPointer(Value, DestType);
  1277. }
  1278. llvm_unreachable("Unknown APValue kind");
  1279. }
  1280. llvm::Constant *
  1281. CodeGenModule::EmitConstantValueForMemory(const APValue &Value,
  1282. QualType DestType,
  1283. CodeGenFunction *CGF) {
  1284. llvm::Constant *C = EmitConstantValue(Value, DestType, CGF);
  1285. if (C->getType()->getScalarType()->isIntegerTy(1)) { // HLSL Change
  1286. llvm::Type *BoolTy = getTypes().ConvertTypeForMem(DestType);
  1287. C = llvm::ConstantExpr::getZExt(C, BoolTy);
  1288. }
  1289. return C;
  1290. }
  1291. llvm::Constant *
  1292. CodeGenModule::GetAddrOfConstantCompoundLiteral(const CompoundLiteralExpr *E) {
  1293. assert(E->isFileScope() && "not a file-scope compound literal expr");
  1294. return ConstExprEmitter(*this, nullptr).EmitLValue(E);
  1295. }
  1296. llvm::Constant *
  1297. CodeGenModule::getMemberPointerConstant(const UnaryOperator *uo) {
  1298. // Member pointer constants always have a very particular form.
  1299. const MemberPointerType *type = cast<MemberPointerType>(uo->getType());
  1300. const ValueDecl *decl = cast<DeclRefExpr>(uo->getSubExpr())->getDecl();
  1301. // A member function pointer.
  1302. if (const CXXMethodDecl *method = dyn_cast<CXXMethodDecl>(decl))
  1303. return getCXXABI().EmitMemberFunctionPointer(method);
  1304. // Otherwise, a member data pointer.
  1305. uint64_t fieldOffset = getContext().getFieldOffset(decl);
  1306. CharUnits chars = getContext().toCharUnitsFromBits((int64_t) fieldOffset);
  1307. return getCXXABI().EmitMemberDataPointer(type, chars);
  1308. }
  1309. static llvm::Constant *EmitNullConstantForBase(CodeGenModule &CGM,
  1310. llvm::Type *baseType,
  1311. const CXXRecordDecl *base);
  1312. static llvm::Constant *EmitNullConstant(CodeGenModule &CGM,
  1313. const CXXRecordDecl *record,
  1314. bool asCompleteObject) {
  1315. const CGRecordLayout &layout = CGM.getTypes().getCGRecordLayout(record);
  1316. llvm::StructType *structure =
  1317. (asCompleteObject ? layout.getLLVMType()
  1318. : layout.getBaseSubobjectLLVMType());
  1319. unsigned numElements = structure->getNumElements();
  1320. std::vector<llvm::Constant *> elements(numElements);
  1321. // Fill in all the bases.
  1322. for (const auto &I : record->bases()) {
  1323. if (I.isVirtual()) {
  1324. // Ignore virtual bases; if we're laying out for a complete
  1325. // object, we'll lay these out later.
  1326. continue;
  1327. }
  1328. const CXXRecordDecl *base =
  1329. cast<CXXRecordDecl>(I.getType()->castAs<RecordType>()->getDecl());
  1330. // Ignore empty bases.
  1331. if (base->isEmpty())
  1332. continue;
  1333. unsigned fieldIndex = layout.getNonVirtualBaseLLVMFieldNo(base);
  1334. llvm::Type *baseType = structure->getElementType(fieldIndex);
  1335. elements[fieldIndex] = EmitNullConstantForBase(CGM, baseType, base);
  1336. }
  1337. // Fill in all the fields.
  1338. for (const auto *Field : record->fields()) {
  1339. // Fill in non-bitfields. (Bitfields always use a zero pattern, which we
  1340. // will fill in later.)
  1341. if (!Field->isBitField()) {
  1342. unsigned fieldIndex = layout.getLLVMFieldNo(Field);
  1343. elements[fieldIndex] = CGM.EmitNullConstant(Field->getType());
  1344. }
  1345. // For unions, stop after the first named field.
  1346. if (record->isUnion()) {
  1347. if (Field->getIdentifier())
  1348. break;
  1349. if (const auto *FieldRD =
  1350. dyn_cast_or_null<RecordDecl>(Field->getType()->getAsTagDecl()))
  1351. if (FieldRD->findFirstNamedDataMember())
  1352. break;
  1353. }
  1354. }
  1355. // Fill in the virtual bases, if we're working with the complete object.
  1356. if (asCompleteObject) {
  1357. for (const auto &I : record->vbases()) {
  1358. const CXXRecordDecl *base =
  1359. cast<CXXRecordDecl>(I.getType()->castAs<RecordType>()->getDecl());
  1360. // Ignore empty bases.
  1361. if (base->isEmpty())
  1362. continue;
  1363. unsigned fieldIndex = layout.getVirtualBaseIndex(base);
  1364. // We might have already laid this field out.
  1365. if (elements[fieldIndex]) continue;
  1366. llvm::Type *baseType = structure->getElementType(fieldIndex);
  1367. elements[fieldIndex] = EmitNullConstantForBase(CGM, baseType, base);
  1368. }
  1369. }
  1370. // Now go through all other fields and zero them out.
  1371. for (unsigned i = 0; i != numElements; ++i) {
  1372. if (!elements[i])
  1373. elements[i] = llvm::Constant::getNullValue(structure->getElementType(i));
  1374. }
  1375. return llvm::ConstantStruct::get(structure, elements);
  1376. }
  1377. /// Emit the null constant for a base subobject.
  1378. static llvm::Constant *EmitNullConstantForBase(CodeGenModule &CGM,
  1379. llvm::Type *baseType,
  1380. const CXXRecordDecl *base) {
  1381. const CGRecordLayout &baseLayout = CGM.getTypes().getCGRecordLayout(base);
  1382. // Just zero out bases that don't have any pointer to data members.
  1383. if (baseLayout.isZeroInitializableAsBase())
  1384. return llvm::Constant::getNullValue(baseType);
  1385. // Otherwise, we can just use its null constant.
  1386. return EmitNullConstant(CGM, base, /*asCompleteObject=*/false);
  1387. }
  1388. llvm::Constant *CodeGenModule::EmitNullConstant(QualType T) {
  1389. if (getTypes().isZeroInitializable(T))
  1390. return llvm::Constant::getNullValue(getTypes().ConvertTypeForMem(T));
  1391. if (const ConstantArrayType *CAT = Context.getAsConstantArrayType(T)) {
  1392. llvm::ArrayType *ATy =
  1393. cast<llvm::ArrayType>(getTypes().ConvertTypeForMem(T));
  1394. QualType ElementTy = CAT->getElementType();
  1395. llvm::Constant *Element = EmitNullConstant(ElementTy);
  1396. unsigned NumElements = CAT->getSize().getZExtValue();
  1397. SmallVector<llvm::Constant *, 8> Array(NumElements, Element);
  1398. return llvm::ConstantArray::get(ATy, Array);
  1399. }
  1400. if (const RecordType *RT = T->getAs<RecordType>()) {
  1401. const CXXRecordDecl *RD = cast<CXXRecordDecl>(RT->getDecl());
  1402. return ::EmitNullConstant(*this, RD, /*complete object*/ true);
  1403. }
  1404. assert(T->isMemberDataPointerType() &&
  1405. "Should only see pointers to data members here!");
  1406. return getCXXABI().EmitNullMemberPointer(T->castAs<MemberPointerType>());
  1407. }
  1408. llvm::Constant *
  1409. CodeGenModule::EmitNullConstantForBase(const CXXRecordDecl *Record) {
  1410. return ::EmitNullConstant(*this, Record, false);
  1411. }