CGNonTrivialStruct.cpp 41 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999
  1. //===--- CGNonTrivialStruct.cpp - Emit Special Functions for C Structs ----===//
  2. //
  3. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
  4. // See https://llvm.org/LICENSE.txt for license information.
  5. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  6. //
  7. //===----------------------------------------------------------------------===//
  8. //
  9. // This file defines functions to generate various special functions for C
  10. // structs.
  11. //
  12. //===----------------------------------------------------------------------===//
  13. #include "CodeGenFunction.h"
  14. #include "CodeGenModule.h"
  15. #include "clang/AST/NonTrivialTypeVisitor.h"
  16. #include "clang/CodeGen/CodeGenABITypes.h"
  17. #include "llvm/Support/ScopedPrinter.h"
  18. #include <array>
  19. using namespace clang;
  20. using namespace CodeGen;
  21. // Return the size of a field in number of bits.
  22. static uint64_t getFieldSize(const FieldDecl *FD, QualType FT,
  23. ASTContext &Ctx) {
  24. if (FD && FD->isBitField())
  25. return FD->getBitWidthValue(Ctx);
  26. return Ctx.getTypeSize(FT);
  27. }
  28. namespace {
  29. enum { DstIdx = 0, SrcIdx = 1 };
  30. const char *ValNameStr[2] = {"dst", "src"};
  31. template <class Derived> struct StructVisitor {
  32. StructVisitor(ASTContext &Ctx) : Ctx(Ctx) {}
  33. template <class... Ts>
  34. void visitStructFields(QualType QT, CharUnits CurStructOffset, Ts... Args) {
  35. const RecordDecl *RD = QT->castAs<RecordType>()->getDecl();
  36. // Iterate over the fields of the struct.
  37. for (const FieldDecl *FD : RD->fields()) {
  38. QualType FT = FD->getType();
  39. FT = QT.isVolatileQualified() ? FT.withVolatile() : FT;
  40. asDerived().visit(FT, FD, CurStructOffset, Args...);
  41. }
  42. asDerived().flushTrivialFields(Args...);
  43. }
  44. template <class... Ts> void visitTrivial(Ts... Args) {}
  45. template <class... Ts> void visitCXXDestructor(Ts... Args) {
  46. llvm_unreachable("field of a C++ struct type is not expected");
  47. }
  48. template <class... Ts> void flushTrivialFields(Ts... Args) {}
  49. uint64_t getFieldOffsetInBits(const FieldDecl *FD) {
  50. return FD ? Ctx.getASTRecordLayout(FD->getParent())
  51. .getFieldOffset(FD->getFieldIndex())
  52. : 0;
  53. }
  54. CharUnits getFieldOffset(const FieldDecl *FD) {
  55. return Ctx.toCharUnitsFromBits(getFieldOffsetInBits(FD));
  56. }
  57. Derived &asDerived() { return static_cast<Derived &>(*this); }
  58. ASTContext &getContext() { return Ctx; }
  59. ASTContext &Ctx;
  60. };
  61. template <class Derived, bool IsMove>
  62. struct CopyStructVisitor : StructVisitor<Derived>,
  63. CopiedTypeVisitor<Derived, IsMove> {
  64. using StructVisitor<Derived>::asDerived;
  65. using Super = CopiedTypeVisitor<Derived, IsMove>;
  66. CopyStructVisitor(ASTContext &Ctx) : StructVisitor<Derived>(Ctx) {}
  67. template <class... Ts>
  68. void preVisit(QualType::PrimitiveCopyKind PCK, QualType FT,
  69. const FieldDecl *FD, CharUnits CurStructOffset, Ts &&... Args) {
  70. if (PCK)
  71. asDerived().flushTrivialFields(std::forward<Ts>(Args)...);
  72. }
  73. template <class... Ts>
  74. void visitWithKind(QualType::PrimitiveCopyKind PCK, QualType FT,
  75. const FieldDecl *FD, CharUnits CurStructOffset,
  76. Ts &&... Args) {
  77. if (const auto *AT = asDerived().getContext().getAsArrayType(FT)) {
  78. asDerived().visitArray(PCK, AT, FT.isVolatileQualified(), FD,
  79. CurStructOffset, std::forward<Ts>(Args)...);
  80. return;
  81. }
  82. Super::visitWithKind(PCK, FT, FD, CurStructOffset,
  83. std::forward<Ts>(Args)...);
  84. }
  85. template <class... Ts>
  86. void visitTrivial(QualType FT, const FieldDecl *FD, CharUnits CurStructOffset,
  87. Ts... Args) {
  88. assert(!FT.isVolatileQualified() && "volatile field not expected");
  89. ASTContext &Ctx = asDerived().getContext();
  90. uint64_t FieldSize = getFieldSize(FD, FT, Ctx);
  91. // Ignore zero-sized fields.
  92. if (FieldSize == 0)
  93. return;
  94. uint64_t FStartInBits = asDerived().getFieldOffsetInBits(FD);
  95. uint64_t FEndInBits = FStartInBits + FieldSize;
  96. uint64_t RoundedFEnd = llvm::alignTo(FEndInBits, Ctx.getCharWidth());
  97. // Set Start if this is the first field of a sequence of trivial fields.
  98. if (Start == End)
  99. Start = CurStructOffset + Ctx.toCharUnitsFromBits(FStartInBits);
  100. End = CurStructOffset + Ctx.toCharUnitsFromBits(RoundedFEnd);
  101. }
  102. CharUnits Start = CharUnits::Zero(), End = CharUnits::Zero();
  103. };
  104. // This function creates the mangled name of a special function of a non-trivial
  105. // C struct. Since there is no ODR in C, the function is mangled based on the
  106. // struct contents and not the name. The mangled name has the following
  107. // structure:
  108. //
  109. // <function-name> ::= <prefix> <alignment-info> "_" <struct-field-info>
  110. // <prefix> ::= "__destructor_" | "__default_constructor_" |
  111. // "__copy_constructor_" | "__move_constructor_" |
  112. // "__copy_assignment_" | "__move_assignment_"
  113. // <alignment-info> ::= <dst-alignment> ["_" <src-alignment>]
  114. // <struct-field-info> ::= <field-info>+
  115. // <field-info> ::= <struct-or-scalar-field-info> | <array-field-info>
  116. // <struct-or-scalar-field-info> ::= "_S" <struct-field-info> |
  117. // <strong-field-info> | <trivial-field-info>
  118. // <array-field-info> ::= "_AB" <array-offset> "s" <element-size> "n"
  119. // <num-elements> <innermost-element-info> "_AE"
  120. // <innermost-element-info> ::= <struct-or-scalar-field-info>
  121. // <strong-field-info> ::= "_s" ["b"] ["v"] <field-offset>
  122. // <trivial-field-info> ::= "_t" ["v"] <field-offset> "_" <field-size>
  123. template <class Derived> struct GenFuncNameBase {
  124. std::string getVolatileOffsetStr(bool IsVolatile, CharUnits Offset) {
  125. std::string S;
  126. if (IsVolatile)
  127. S = "v";
  128. S += llvm::to_string(Offset.getQuantity());
  129. return S;
  130. }
  131. void visitARCStrong(QualType FT, const FieldDecl *FD,
  132. CharUnits CurStructOffset) {
  133. appendStr("_s");
  134. if (FT->isBlockPointerType())
  135. appendStr("b");
  136. CharUnits FieldOffset = CurStructOffset + asDerived().getFieldOffset(FD);
  137. appendStr(getVolatileOffsetStr(FT.isVolatileQualified(), FieldOffset));
  138. }
  139. void visitARCWeak(QualType FT, const FieldDecl *FD,
  140. CharUnits CurStructOffset) {
  141. appendStr("_w");
  142. CharUnits FieldOffset = CurStructOffset + asDerived().getFieldOffset(FD);
  143. appendStr(getVolatileOffsetStr(FT.isVolatileQualified(), FieldOffset));
  144. }
  145. void visitStruct(QualType QT, const FieldDecl *FD,
  146. CharUnits CurStructOffset) {
  147. CharUnits FieldOffset = CurStructOffset + asDerived().getFieldOffset(FD);
  148. appendStr("_S");
  149. asDerived().visitStructFields(QT, FieldOffset);
  150. }
  151. template <class FieldKind>
  152. void visitArray(FieldKind FK, const ArrayType *AT, bool IsVolatile,
  153. const FieldDecl *FD, CharUnits CurStructOffset) {
  154. // String for non-volatile trivial fields is emitted when
  155. // flushTrivialFields is called.
  156. if (!FK)
  157. return asDerived().visitTrivial(QualType(AT, 0), FD, CurStructOffset);
  158. asDerived().flushTrivialFields();
  159. CharUnits FieldOffset = CurStructOffset + asDerived().getFieldOffset(FD);
  160. ASTContext &Ctx = asDerived().getContext();
  161. const ConstantArrayType *CAT = cast<ConstantArrayType>(AT);
  162. unsigned NumElts = Ctx.getConstantArrayElementCount(CAT);
  163. QualType EltTy = Ctx.getBaseElementType(CAT);
  164. CharUnits EltSize = Ctx.getTypeSizeInChars(EltTy);
  165. appendStr("_AB" + llvm::to_string(FieldOffset.getQuantity()) + "s" +
  166. llvm::to_string(EltSize.getQuantity()) + "n" +
  167. llvm::to_string(NumElts));
  168. EltTy = IsVolatile ? EltTy.withVolatile() : EltTy;
  169. asDerived().visitWithKind(FK, EltTy, nullptr, FieldOffset);
  170. appendStr("_AE");
  171. }
  172. void appendStr(StringRef Str) { Name += Str; }
  173. std::string getName(QualType QT, bool IsVolatile) {
  174. QT = IsVolatile ? QT.withVolatile() : QT;
  175. asDerived().visitStructFields(QT, CharUnits::Zero());
  176. return Name;
  177. }
  178. Derived &asDerived() { return static_cast<Derived &>(*this); }
  179. std::string Name;
  180. };
  181. template <class Derived>
  182. struct GenUnaryFuncName : StructVisitor<Derived>, GenFuncNameBase<Derived> {
  183. GenUnaryFuncName(StringRef Prefix, CharUnits DstAlignment, ASTContext &Ctx)
  184. : StructVisitor<Derived>(Ctx) {
  185. this->appendStr(Prefix);
  186. this->appendStr(llvm::to_string(DstAlignment.getQuantity()));
  187. }
  188. };
  189. // Helper function to create a null constant.
  190. static llvm::Constant *getNullForVariable(Address Addr) {
  191. llvm::Type *Ty = Addr.getElementType();
  192. return llvm::ConstantPointerNull::get(cast<llvm::PointerType>(Ty));
  193. }
  194. template <bool IsMove>
  195. struct GenBinaryFuncName : CopyStructVisitor<GenBinaryFuncName<IsMove>, IsMove>,
  196. GenFuncNameBase<GenBinaryFuncName<IsMove>> {
  197. GenBinaryFuncName(StringRef Prefix, CharUnits DstAlignment,
  198. CharUnits SrcAlignment, ASTContext &Ctx)
  199. : CopyStructVisitor<GenBinaryFuncName<IsMove>, IsMove>(Ctx) {
  200. this->appendStr(Prefix);
  201. this->appendStr(llvm::to_string(DstAlignment.getQuantity()));
  202. this->appendStr("_" + llvm::to_string(SrcAlignment.getQuantity()));
  203. }
  204. void flushTrivialFields() {
  205. if (this->Start == this->End)
  206. return;
  207. this->appendStr("_t" + llvm::to_string(this->Start.getQuantity()) + "w" +
  208. llvm::to_string((this->End - this->Start).getQuantity()));
  209. this->Start = this->End = CharUnits::Zero();
  210. }
  211. void visitVolatileTrivial(QualType FT, const FieldDecl *FD,
  212. CharUnits CurStructOffset) {
  213. // Because volatile fields can be bit-fields and are individually copied,
  214. // their offset and width are in bits.
  215. uint64_t OffsetInBits =
  216. this->Ctx.toBits(CurStructOffset) + this->getFieldOffsetInBits(FD);
  217. this->appendStr("_tv" + llvm::to_string(OffsetInBits) + "w" +
  218. llvm::to_string(getFieldSize(FD, FT, this->Ctx)));
  219. }
  220. };
  221. struct GenDefaultInitializeFuncName
  222. : GenUnaryFuncName<GenDefaultInitializeFuncName>,
  223. DefaultInitializedTypeVisitor<GenDefaultInitializeFuncName> {
  224. using Super = DefaultInitializedTypeVisitor<GenDefaultInitializeFuncName>;
  225. GenDefaultInitializeFuncName(CharUnits DstAlignment, ASTContext &Ctx)
  226. : GenUnaryFuncName<GenDefaultInitializeFuncName>("__default_constructor_",
  227. DstAlignment, Ctx) {}
  228. void visitWithKind(QualType::PrimitiveDefaultInitializeKind PDIK, QualType FT,
  229. const FieldDecl *FD, CharUnits CurStructOffset) {
  230. if (const auto *AT = getContext().getAsArrayType(FT)) {
  231. visitArray(PDIK, AT, FT.isVolatileQualified(), FD, CurStructOffset);
  232. return;
  233. }
  234. Super::visitWithKind(PDIK, FT, FD, CurStructOffset);
  235. }
  236. };
  237. struct GenDestructorFuncName : GenUnaryFuncName<GenDestructorFuncName>,
  238. DestructedTypeVisitor<GenDestructorFuncName> {
  239. using Super = DestructedTypeVisitor<GenDestructorFuncName>;
  240. GenDestructorFuncName(const char *Prefix, CharUnits DstAlignment,
  241. ASTContext &Ctx)
  242. : GenUnaryFuncName<GenDestructorFuncName>(Prefix, DstAlignment, Ctx) {}
  243. void visitWithKind(QualType::DestructionKind DK, QualType FT,
  244. const FieldDecl *FD, CharUnits CurStructOffset) {
  245. if (const auto *AT = getContext().getAsArrayType(FT)) {
  246. visitArray(DK, AT, FT.isVolatileQualified(), FD, CurStructOffset);
  247. return;
  248. }
  249. Super::visitWithKind(DK, FT, FD, CurStructOffset);
  250. }
  251. };
  252. // Helper function that creates CGFunctionInfo for an N-ary special function.
  253. template <size_t N>
  254. static const CGFunctionInfo &getFunctionInfo(CodeGenModule &CGM,
  255. FunctionArgList &Args) {
  256. ASTContext &Ctx = CGM.getContext();
  257. llvm::SmallVector<ImplicitParamDecl *, N> Params;
  258. QualType ParamTy = Ctx.getPointerType(Ctx.VoidPtrTy);
  259. for (unsigned I = 0; I < N; ++I)
  260. Params.push_back(ImplicitParamDecl::Create(
  261. Ctx, nullptr, SourceLocation(), &Ctx.Idents.get(ValNameStr[I]), ParamTy,
  262. ImplicitParamDecl::Other));
  263. for (auto &P : Params)
  264. Args.push_back(P);
  265. return CGM.getTypes().arrangeBuiltinFunctionDeclaration(Ctx.VoidTy, Args);
  266. }
  267. // Template classes that are used as bases for classes that emit special
  268. // functions.
  269. template <class Derived> struct GenFuncBase {
  270. template <size_t N>
  271. void visitStruct(QualType FT, const FieldDecl *FD, CharUnits CurStructOffset,
  272. std::array<Address, N> Addrs) {
  273. this->asDerived().callSpecialFunction(
  274. FT, CurStructOffset + asDerived().getFieldOffset(FD), Addrs);
  275. }
  276. template <class FieldKind, size_t N>
  277. void visitArray(FieldKind FK, const ArrayType *AT, bool IsVolatile,
  278. const FieldDecl *FD, CharUnits CurStructOffset,
  279. std::array<Address, N> Addrs) {
  280. // Non-volatile trivial fields are copied when flushTrivialFields is called.
  281. if (!FK)
  282. return asDerived().visitTrivial(QualType(AT, 0), FD, CurStructOffset,
  283. Addrs);
  284. asDerived().flushTrivialFields(Addrs);
  285. CodeGenFunction &CGF = *this->CGF;
  286. ASTContext &Ctx = CGF.getContext();
  287. // Compute the end address.
  288. QualType BaseEltQT;
  289. std::array<Address, N> StartAddrs = Addrs;
  290. for (unsigned I = 0; I < N; ++I)
  291. StartAddrs[I] = getAddrWithOffset(Addrs[I], CurStructOffset, FD);
  292. Address DstAddr = StartAddrs[DstIdx];
  293. llvm::Value *NumElts = CGF.emitArrayLength(AT, BaseEltQT, DstAddr);
  294. unsigned BaseEltSize = Ctx.getTypeSizeInChars(BaseEltQT).getQuantity();
  295. llvm::Value *BaseEltSizeVal =
  296. llvm::ConstantInt::get(NumElts->getType(), BaseEltSize);
  297. llvm::Value *SizeInBytes =
  298. CGF.Builder.CreateNUWMul(BaseEltSizeVal, NumElts);
  299. Address BC = CGF.Builder.CreateBitCast(DstAddr, CGF.CGM.Int8PtrTy);
  300. llvm::Value *DstArrayEnd =
  301. CGF.Builder.CreateInBoundsGEP(BC.getPointer(), SizeInBytes);
  302. DstArrayEnd = CGF.Builder.CreateBitCast(DstArrayEnd, CGF.CGM.Int8PtrPtrTy,
  303. "dstarray.end");
  304. llvm::BasicBlock *PreheaderBB = CGF.Builder.GetInsertBlock();
  305. // Create the header block and insert the phi instructions.
  306. llvm::BasicBlock *HeaderBB = CGF.createBasicBlock("loop.header");
  307. CGF.EmitBlock(HeaderBB);
  308. llvm::PHINode *PHIs[N];
  309. for (unsigned I = 0; I < N; ++I) {
  310. PHIs[I] = CGF.Builder.CreatePHI(CGF.CGM.Int8PtrPtrTy, 2, "addr.cur");
  311. PHIs[I]->addIncoming(StartAddrs[I].getPointer(), PreheaderBB);
  312. }
  313. // Create the exit and loop body blocks.
  314. llvm::BasicBlock *ExitBB = CGF.createBasicBlock("loop.exit");
  315. llvm::BasicBlock *LoopBB = CGF.createBasicBlock("loop.body");
  316. // Emit the comparison and conditional branch instruction that jumps to
  317. // either the exit or the loop body.
  318. llvm::Value *Done =
  319. CGF.Builder.CreateICmpEQ(PHIs[DstIdx], DstArrayEnd, "done");
  320. CGF.Builder.CreateCondBr(Done, ExitBB, LoopBB);
  321. // Visit the element of the array in the loop body.
  322. CGF.EmitBlock(LoopBB);
  323. QualType EltQT = AT->getElementType();
  324. CharUnits EltSize = Ctx.getTypeSizeInChars(EltQT);
  325. std::array<Address, N> NewAddrs = Addrs;
  326. for (unsigned I = 0; I < N; ++I)
  327. NewAddrs[I] = Address(
  328. PHIs[I], StartAddrs[I].getAlignment().alignmentAtOffset(EltSize));
  329. EltQT = IsVolatile ? EltQT.withVolatile() : EltQT;
  330. this->asDerived().visitWithKind(FK, EltQT, nullptr, CharUnits::Zero(),
  331. NewAddrs);
  332. LoopBB = CGF.Builder.GetInsertBlock();
  333. for (unsigned I = 0; I < N; ++I) {
  334. // Instrs to update the destination and source addresses.
  335. // Update phi instructions.
  336. NewAddrs[I] = getAddrWithOffset(NewAddrs[I], EltSize);
  337. PHIs[I]->addIncoming(NewAddrs[I].getPointer(), LoopBB);
  338. }
  339. // Insert an unconditional branch to the header block.
  340. CGF.Builder.CreateBr(HeaderBB);
  341. CGF.EmitBlock(ExitBB);
  342. }
  343. /// Return an address with the specified offset from the passed address.
  344. Address getAddrWithOffset(Address Addr, CharUnits Offset) {
  345. assert(Addr.isValid() && "invalid address");
  346. if (Offset.getQuantity() == 0)
  347. return Addr;
  348. Addr = CGF->Builder.CreateBitCast(Addr, CGF->CGM.Int8PtrTy);
  349. Addr = CGF->Builder.CreateConstInBoundsGEP(Addr, Offset.getQuantity());
  350. return CGF->Builder.CreateBitCast(Addr, CGF->CGM.Int8PtrPtrTy);
  351. }
  352. Address getAddrWithOffset(Address Addr, CharUnits StructFieldOffset,
  353. const FieldDecl *FD) {
  354. return getAddrWithOffset(Addr, StructFieldOffset +
  355. asDerived().getFieldOffset(FD));
  356. }
  357. template <size_t N>
  358. llvm::Function *
  359. getFunction(StringRef FuncName, QualType QT, std::array<Address, N> Addrs,
  360. std::array<CharUnits, N> Alignments, CodeGenModule &CGM) {
  361. // If the special function already exists in the module, return it.
  362. if (llvm::Function *F = CGM.getModule().getFunction(FuncName)) {
  363. bool WrongType = false;
  364. if (!F->getReturnType()->isVoidTy())
  365. WrongType = true;
  366. else {
  367. for (const llvm::Argument &Arg : F->args())
  368. if (Arg.getType() != CGM.Int8PtrPtrTy)
  369. WrongType = true;
  370. }
  371. if (WrongType) {
  372. std::string FuncName = F->getName();
  373. SourceLocation Loc = QT->castAs<RecordType>()->getDecl()->getLocation();
  374. CGM.Error(Loc, "special function " + FuncName +
  375. " for non-trivial C struct has incorrect type");
  376. return nullptr;
  377. }
  378. return F;
  379. }
  380. ASTContext &Ctx = CGM.getContext();
  381. FunctionArgList Args;
  382. const CGFunctionInfo &FI = getFunctionInfo<N>(CGM, Args);
  383. llvm::FunctionType *FuncTy = CGM.getTypes().GetFunctionType(FI);
  384. llvm::Function *F =
  385. llvm::Function::Create(FuncTy, llvm::GlobalValue::LinkOnceODRLinkage,
  386. FuncName, &CGM.getModule());
  387. F->setVisibility(llvm::GlobalValue::HiddenVisibility);
  388. CGM.SetLLVMFunctionAttributes(GlobalDecl(), FI, F);
  389. CGM.SetLLVMFunctionAttributesForDefinition(nullptr, F);
  390. IdentifierInfo *II = &Ctx.Idents.get(FuncName);
  391. FunctionDecl *FD = FunctionDecl::Create(
  392. Ctx, Ctx.getTranslationUnitDecl(), SourceLocation(), SourceLocation(),
  393. II, Ctx.getFunctionType(Ctx.VoidTy, llvm::None, {}), nullptr,
  394. SC_PrivateExtern, false, false);
  395. CodeGenFunction NewCGF(CGM);
  396. setCGF(&NewCGF);
  397. CGF->StartFunction(FD, Ctx.VoidTy, F, FI, Args);
  398. for (unsigned I = 0; I < N; ++I) {
  399. llvm::Value *V = CGF->Builder.CreateLoad(CGF->GetAddrOfLocalVar(Args[I]));
  400. Addrs[I] = Address(V, Alignments[I]);
  401. }
  402. asDerived().visitStructFields(QT, CharUnits::Zero(), Addrs);
  403. CGF->FinishFunction();
  404. return F;
  405. }
  406. template <size_t N>
  407. void callFunc(StringRef FuncName, QualType QT, std::array<Address, N> Addrs,
  408. CodeGenFunction &CallerCGF) {
  409. std::array<CharUnits, N> Alignments;
  410. llvm::Value *Ptrs[N];
  411. for (unsigned I = 0; I < N; ++I) {
  412. Alignments[I] = Addrs[I].getAlignment();
  413. Ptrs[I] =
  414. CallerCGF.Builder.CreateBitCast(Addrs[I], CallerCGF.CGM.Int8PtrPtrTy)
  415. .getPointer();
  416. }
  417. if (llvm::Function *F =
  418. getFunction(FuncName, QT, Addrs, Alignments, CallerCGF.CGM))
  419. CallerCGF.EmitNounwindRuntimeCall(F, Ptrs);
  420. }
  421. Derived &asDerived() { return static_cast<Derived &>(*this); }
  422. void setCGF(CodeGenFunction *F) { CGF = F; }
  423. CodeGenFunction *CGF = nullptr;
  424. };
  425. template <class Derived, bool IsMove>
  426. struct GenBinaryFunc : CopyStructVisitor<Derived, IsMove>,
  427. GenFuncBase<Derived> {
  428. GenBinaryFunc(ASTContext &Ctx) : CopyStructVisitor<Derived, IsMove>(Ctx) {}
  429. void flushTrivialFields(std::array<Address, 2> Addrs) {
  430. CharUnits Size = this->End - this->Start;
  431. if (Size.getQuantity() == 0)
  432. return;
  433. Address DstAddr = this->getAddrWithOffset(Addrs[DstIdx], this->Start);
  434. Address SrcAddr = this->getAddrWithOffset(Addrs[SrcIdx], this->Start);
  435. // Emit memcpy.
  436. if (Size.getQuantity() >= 16 || !llvm::isPowerOf2_32(Size.getQuantity())) {
  437. llvm::Value *SizeVal =
  438. llvm::ConstantInt::get(this->CGF->SizeTy, Size.getQuantity());
  439. DstAddr =
  440. this->CGF->Builder.CreateElementBitCast(DstAddr, this->CGF->Int8Ty);
  441. SrcAddr =
  442. this->CGF->Builder.CreateElementBitCast(SrcAddr, this->CGF->Int8Ty);
  443. this->CGF->Builder.CreateMemCpy(DstAddr, SrcAddr, SizeVal, false);
  444. } else {
  445. llvm::Type *Ty = llvm::Type::getIntNTy(
  446. this->CGF->getLLVMContext(),
  447. Size.getQuantity() * this->CGF->getContext().getCharWidth());
  448. DstAddr = this->CGF->Builder.CreateElementBitCast(DstAddr, Ty);
  449. SrcAddr = this->CGF->Builder.CreateElementBitCast(SrcAddr, Ty);
  450. llvm::Value *SrcVal = this->CGF->Builder.CreateLoad(SrcAddr, false);
  451. this->CGF->Builder.CreateStore(SrcVal, DstAddr, false);
  452. }
  453. this->Start = this->End = CharUnits::Zero();
  454. }
  455. template <class... Ts>
  456. void visitVolatileTrivial(QualType FT, const FieldDecl *FD, CharUnits Offset,
  457. std::array<Address, 2> Addrs) {
  458. LValue DstLV, SrcLV;
  459. if (FD) {
  460. QualType RT = QualType(FD->getParent()->getTypeForDecl(), 0);
  461. llvm::PointerType *PtrTy = this->CGF->ConvertType(RT)->getPointerTo();
  462. Address DstAddr = this->getAddrWithOffset(Addrs[DstIdx], Offset);
  463. LValue DstBase = this->CGF->MakeAddrLValue(
  464. this->CGF->Builder.CreateBitCast(DstAddr, PtrTy), FT);
  465. DstLV = this->CGF->EmitLValueForField(DstBase, FD);
  466. Address SrcAddr = this->getAddrWithOffset(Addrs[SrcIdx], Offset);
  467. LValue SrcBase = this->CGF->MakeAddrLValue(
  468. this->CGF->Builder.CreateBitCast(SrcAddr, PtrTy), FT);
  469. SrcLV = this->CGF->EmitLValueForField(SrcBase, FD);
  470. } else {
  471. llvm::PointerType *Ty = this->CGF->ConvertType(FT)->getPointerTo();
  472. Address DstAddr = this->CGF->Builder.CreateBitCast(Addrs[DstIdx], Ty);
  473. Address SrcAddr = this->CGF->Builder.CreateBitCast(Addrs[SrcIdx], Ty);
  474. DstLV = this->CGF->MakeAddrLValue(DstAddr, FT);
  475. SrcLV = this->CGF->MakeAddrLValue(SrcAddr, FT);
  476. }
  477. RValue SrcVal = this->CGF->EmitLoadOfLValue(SrcLV, SourceLocation());
  478. this->CGF->EmitStoreThroughLValue(SrcVal, DstLV);
  479. }
  480. };
  481. // These classes that emit the special functions for a non-trivial struct.
  482. struct GenDestructor : StructVisitor<GenDestructor>,
  483. GenFuncBase<GenDestructor>,
  484. DestructedTypeVisitor<GenDestructor> {
  485. using Super = DestructedTypeVisitor<GenDestructor>;
  486. GenDestructor(ASTContext &Ctx) : StructVisitor<GenDestructor>(Ctx) {}
  487. void visitWithKind(QualType::DestructionKind DK, QualType FT,
  488. const FieldDecl *FD, CharUnits CurStructOffset,
  489. std::array<Address, 1> Addrs) {
  490. if (const auto *AT = getContext().getAsArrayType(FT)) {
  491. visitArray(DK, AT, FT.isVolatileQualified(), FD, CurStructOffset, Addrs);
  492. return;
  493. }
  494. Super::visitWithKind(DK, FT, FD, CurStructOffset, Addrs);
  495. }
  496. void visitARCStrong(QualType QT, const FieldDecl *FD,
  497. CharUnits CurStructOffset, std::array<Address, 1> Addrs) {
  498. CGF->destroyARCStrongImprecise(
  499. *CGF, getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD), QT);
  500. }
  501. void visitARCWeak(QualType QT, const FieldDecl *FD, CharUnits CurStructOffset,
  502. std::array<Address, 1> Addrs) {
  503. CGF->destroyARCWeak(
  504. *CGF, getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD), QT);
  505. }
  506. void callSpecialFunction(QualType FT, CharUnits Offset,
  507. std::array<Address, 1> Addrs) {
  508. CGF->callCStructDestructor(
  509. CGF->MakeAddrLValue(getAddrWithOffset(Addrs[DstIdx], Offset), FT));
  510. }
  511. };
  512. struct GenDefaultInitialize
  513. : StructVisitor<GenDefaultInitialize>,
  514. GenFuncBase<GenDefaultInitialize>,
  515. DefaultInitializedTypeVisitor<GenDefaultInitialize> {
  516. using Super = DefaultInitializedTypeVisitor<GenDefaultInitialize>;
  517. typedef GenFuncBase<GenDefaultInitialize> GenFuncBaseTy;
  518. GenDefaultInitialize(ASTContext &Ctx)
  519. : StructVisitor<GenDefaultInitialize>(Ctx) {}
  520. void visitWithKind(QualType::PrimitiveDefaultInitializeKind PDIK, QualType FT,
  521. const FieldDecl *FD, CharUnits CurStructOffset,
  522. std::array<Address, 1> Addrs) {
  523. if (const auto *AT = getContext().getAsArrayType(FT)) {
  524. visitArray(PDIK, AT, FT.isVolatileQualified(), FD, CurStructOffset,
  525. Addrs);
  526. return;
  527. }
  528. Super::visitWithKind(PDIK, FT, FD, CurStructOffset, Addrs);
  529. }
  530. void visitARCStrong(QualType QT, const FieldDecl *FD,
  531. CharUnits CurStructOffset, std::array<Address, 1> Addrs) {
  532. CGF->EmitNullInitialization(
  533. getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD), QT);
  534. }
  535. void visitARCWeak(QualType QT, const FieldDecl *FD, CharUnits CurStructOffset,
  536. std::array<Address, 1> Addrs) {
  537. CGF->EmitNullInitialization(
  538. getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD), QT);
  539. }
  540. template <class FieldKind, size_t... Is>
  541. void visitArray(FieldKind FK, const ArrayType *AT, bool IsVolatile,
  542. const FieldDecl *FD, CharUnits CurStructOffset,
  543. std::array<Address, 1> Addrs) {
  544. if (!FK)
  545. return visitTrivial(QualType(AT, 0), FD, CurStructOffset, Addrs);
  546. ASTContext &Ctx = getContext();
  547. CharUnits Size = Ctx.getTypeSizeInChars(QualType(AT, 0));
  548. QualType EltTy = Ctx.getBaseElementType(QualType(AT, 0));
  549. if (Size < CharUnits::fromQuantity(16) || EltTy->getAs<RecordType>()) {
  550. GenFuncBaseTy::visitArray(FK, AT, IsVolatile, FD, CurStructOffset, Addrs);
  551. return;
  552. }
  553. llvm::Constant *SizeVal = CGF->Builder.getInt64(Size.getQuantity());
  554. Address DstAddr = getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD);
  555. Address Loc = CGF->Builder.CreateElementBitCast(DstAddr, CGF->Int8Ty);
  556. CGF->Builder.CreateMemSet(Loc, CGF->Builder.getInt8(0), SizeVal,
  557. IsVolatile);
  558. }
  559. void callSpecialFunction(QualType FT, CharUnits Offset,
  560. std::array<Address, 1> Addrs) {
  561. CGF->callCStructDefaultConstructor(
  562. CGF->MakeAddrLValue(getAddrWithOffset(Addrs[DstIdx], Offset), FT));
  563. }
  564. };
  565. struct GenCopyConstructor : GenBinaryFunc<GenCopyConstructor, false> {
  566. GenCopyConstructor(ASTContext &Ctx)
  567. : GenBinaryFunc<GenCopyConstructor, false>(Ctx) {}
  568. void visitARCStrong(QualType QT, const FieldDecl *FD,
  569. CharUnits CurStructOffset, std::array<Address, 2> Addrs) {
  570. Addrs[DstIdx] = getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD);
  571. Addrs[SrcIdx] = getAddrWithOffset(Addrs[SrcIdx], CurStructOffset, FD);
  572. llvm::Value *SrcVal = CGF->EmitLoadOfScalar(
  573. Addrs[SrcIdx], QT.isVolatileQualified(), QT, SourceLocation());
  574. llvm::Value *Val = CGF->EmitARCRetain(QT, SrcVal);
  575. CGF->EmitStoreOfScalar(Val, CGF->MakeAddrLValue(Addrs[DstIdx], QT), true);
  576. }
  577. void visitARCWeak(QualType QT, const FieldDecl *FD, CharUnits CurStructOffset,
  578. std::array<Address, 2> Addrs) {
  579. Addrs[DstIdx] = getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD);
  580. Addrs[SrcIdx] = getAddrWithOffset(Addrs[SrcIdx], CurStructOffset, FD);
  581. CGF->EmitARCCopyWeak(Addrs[DstIdx], Addrs[SrcIdx]);
  582. }
  583. void callSpecialFunction(QualType FT, CharUnits Offset,
  584. std::array<Address, 2> Addrs) {
  585. Addrs[DstIdx] = getAddrWithOffset(Addrs[DstIdx], Offset);
  586. Addrs[SrcIdx] = getAddrWithOffset(Addrs[SrcIdx], Offset);
  587. CGF->callCStructCopyConstructor(CGF->MakeAddrLValue(Addrs[DstIdx], FT),
  588. CGF->MakeAddrLValue(Addrs[SrcIdx], FT));
  589. }
  590. };
  591. struct GenMoveConstructor : GenBinaryFunc<GenMoveConstructor, true> {
  592. GenMoveConstructor(ASTContext &Ctx)
  593. : GenBinaryFunc<GenMoveConstructor, true>(Ctx) {}
  594. void visitARCStrong(QualType QT, const FieldDecl *FD,
  595. CharUnits CurStructOffset, std::array<Address, 2> Addrs) {
  596. Addrs[DstIdx] = getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD);
  597. Addrs[SrcIdx] = getAddrWithOffset(Addrs[SrcIdx], CurStructOffset, FD);
  598. LValue SrcLV = CGF->MakeAddrLValue(Addrs[SrcIdx], QT);
  599. llvm::Value *SrcVal =
  600. CGF->EmitLoadOfLValue(SrcLV, SourceLocation()).getScalarVal();
  601. CGF->EmitStoreOfScalar(getNullForVariable(SrcLV.getAddress()), SrcLV);
  602. CGF->EmitStoreOfScalar(SrcVal, CGF->MakeAddrLValue(Addrs[DstIdx], QT),
  603. /* isInitialization */ true);
  604. }
  605. void visitARCWeak(QualType QT, const FieldDecl *FD, CharUnits CurStructOffset,
  606. std::array<Address, 2> Addrs) {
  607. Addrs[DstIdx] = getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD);
  608. Addrs[SrcIdx] = getAddrWithOffset(Addrs[SrcIdx], CurStructOffset, FD);
  609. CGF->EmitARCMoveWeak(Addrs[DstIdx], Addrs[SrcIdx]);
  610. }
  611. void callSpecialFunction(QualType FT, CharUnits Offset,
  612. std::array<Address, 2> Addrs) {
  613. Addrs[DstIdx] = getAddrWithOffset(Addrs[DstIdx], Offset);
  614. Addrs[SrcIdx] = getAddrWithOffset(Addrs[SrcIdx], Offset);
  615. CGF->callCStructMoveConstructor(CGF->MakeAddrLValue(Addrs[DstIdx], FT),
  616. CGF->MakeAddrLValue(Addrs[SrcIdx], FT));
  617. }
  618. };
  619. struct GenCopyAssignment : GenBinaryFunc<GenCopyAssignment, false> {
  620. GenCopyAssignment(ASTContext &Ctx)
  621. : GenBinaryFunc<GenCopyAssignment, false>(Ctx) {}
  622. void visitARCStrong(QualType QT, const FieldDecl *FD,
  623. CharUnits CurStructOffset, std::array<Address, 2> Addrs) {
  624. Addrs[DstIdx] = getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD);
  625. Addrs[SrcIdx] = getAddrWithOffset(Addrs[SrcIdx], CurStructOffset, FD);
  626. llvm::Value *SrcVal = CGF->EmitLoadOfScalar(
  627. Addrs[SrcIdx], QT.isVolatileQualified(), QT, SourceLocation());
  628. CGF->EmitARCStoreStrong(CGF->MakeAddrLValue(Addrs[DstIdx], QT), SrcVal,
  629. false);
  630. }
  631. void visitARCWeak(QualType QT, const FieldDecl *FD, CharUnits CurStructOffset,
  632. std::array<Address, 2> Addrs) {
  633. Addrs[DstIdx] = getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD);
  634. Addrs[SrcIdx] = getAddrWithOffset(Addrs[SrcIdx], CurStructOffset, FD);
  635. CGF->emitARCCopyAssignWeak(QT, Addrs[DstIdx], Addrs[SrcIdx]);
  636. }
  637. void callSpecialFunction(QualType FT, CharUnits Offset,
  638. std::array<Address, 2> Addrs) {
  639. Addrs[DstIdx] = getAddrWithOffset(Addrs[DstIdx], Offset);
  640. Addrs[SrcIdx] = getAddrWithOffset(Addrs[SrcIdx], Offset);
  641. CGF->callCStructCopyAssignmentOperator(
  642. CGF->MakeAddrLValue(Addrs[DstIdx], FT),
  643. CGF->MakeAddrLValue(Addrs[SrcIdx], FT));
  644. }
  645. };
  646. struct GenMoveAssignment : GenBinaryFunc<GenMoveAssignment, true> {
  647. GenMoveAssignment(ASTContext &Ctx)
  648. : GenBinaryFunc<GenMoveAssignment, true>(Ctx) {}
  649. void visitARCStrong(QualType QT, const FieldDecl *FD,
  650. CharUnits CurStructOffset, std::array<Address, 2> Addrs) {
  651. Addrs[DstIdx] = getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD);
  652. Addrs[SrcIdx] = getAddrWithOffset(Addrs[SrcIdx], CurStructOffset, FD);
  653. LValue SrcLV = CGF->MakeAddrLValue(Addrs[SrcIdx], QT);
  654. llvm::Value *SrcVal =
  655. CGF->EmitLoadOfLValue(SrcLV, SourceLocation()).getScalarVal();
  656. CGF->EmitStoreOfScalar(getNullForVariable(SrcLV.getAddress()), SrcLV);
  657. LValue DstLV = CGF->MakeAddrLValue(Addrs[DstIdx], QT);
  658. llvm::Value *DstVal =
  659. CGF->EmitLoadOfLValue(DstLV, SourceLocation()).getScalarVal();
  660. CGF->EmitStoreOfScalar(SrcVal, DstLV);
  661. CGF->EmitARCRelease(DstVal, ARCImpreciseLifetime);
  662. }
  663. void visitARCWeak(QualType QT, const FieldDecl *FD, CharUnits CurStructOffset,
  664. std::array<Address, 2> Addrs) {
  665. Addrs[DstIdx] = getAddrWithOffset(Addrs[DstIdx], CurStructOffset, FD);
  666. Addrs[SrcIdx] = getAddrWithOffset(Addrs[SrcIdx], CurStructOffset, FD);
  667. CGF->emitARCMoveAssignWeak(QT, Addrs[DstIdx], Addrs[SrcIdx]);
  668. }
  669. void callSpecialFunction(QualType FT, CharUnits Offset,
  670. std::array<Address, 2> Addrs) {
  671. Addrs[DstIdx] = getAddrWithOffset(Addrs[DstIdx], Offset);
  672. Addrs[SrcIdx] = getAddrWithOffset(Addrs[SrcIdx], Offset);
  673. CGF->callCStructMoveAssignmentOperator(
  674. CGF->MakeAddrLValue(Addrs[DstIdx], FT),
  675. CGF->MakeAddrLValue(Addrs[SrcIdx], FT));
  676. }
  677. };
  678. } // namespace
  679. void CodeGenFunction::destroyNonTrivialCStruct(CodeGenFunction &CGF,
  680. Address Addr, QualType Type) {
  681. CGF.callCStructDestructor(CGF.MakeAddrLValue(Addr, Type));
  682. }
  683. // Default-initialize a variable that is a non-trivial struct or an array of
  684. // such structure.
  685. void CodeGenFunction::defaultInitNonTrivialCStructVar(LValue Dst) {
  686. GenDefaultInitialize Gen(getContext());
  687. Address DstPtr = Builder.CreateBitCast(Dst.getAddress(), CGM.Int8PtrPtrTy);
  688. Gen.setCGF(this);
  689. QualType QT = Dst.getType();
  690. QT = Dst.isVolatile() ? QT.withVolatile() : QT;
  691. Gen.visit(QT, nullptr, CharUnits::Zero(), std::array<Address, 1>({{DstPtr}}));
  692. }
  693. template <class G, size_t N>
  694. static void callSpecialFunction(G &&Gen, StringRef FuncName, QualType QT,
  695. bool IsVolatile, CodeGenFunction &CGF,
  696. std::array<Address, N> Addrs) {
  697. for (unsigned I = 0; I < N; ++I)
  698. Addrs[I] = CGF.Builder.CreateBitCast(Addrs[I], CGF.CGM.Int8PtrPtrTy);
  699. QT = IsVolatile ? QT.withVolatile() : QT;
  700. Gen.callFunc(FuncName, QT, Addrs, CGF);
  701. }
  702. template <size_t N> static std::array<Address, N> createNullAddressArray();
  703. template <> std::array<Address, 1> createNullAddressArray() {
  704. return std::array<Address, 1>({{Address(nullptr, CharUnits::Zero())}});
  705. }
  706. template <> std::array<Address, 2> createNullAddressArray() {
  707. return std::array<Address, 2>({{Address(nullptr, CharUnits::Zero()),
  708. Address(nullptr, CharUnits::Zero())}});
  709. }
  710. template <class G, size_t N>
  711. static llvm::Function *
  712. getSpecialFunction(G &&Gen, StringRef FuncName, QualType QT, bool IsVolatile,
  713. std::array<CharUnits, N> Alignments, CodeGenModule &CGM) {
  714. QT = IsVolatile ? QT.withVolatile() : QT;
  715. // The following call requires an array of addresses as arguments, but doesn't
  716. // actually use them (it overwrites them with the addresses of the arguments
  717. // of the created function).
  718. return Gen.getFunction(FuncName, QT, createNullAddressArray<N>(), Alignments,
  719. CGM);
  720. }
  721. // Functions to emit calls to the special functions of a non-trivial C struct.
  722. void CodeGenFunction::callCStructDefaultConstructor(LValue Dst) {
  723. bool IsVolatile = Dst.isVolatile();
  724. Address DstPtr = Dst.getAddress();
  725. QualType QT = Dst.getType();
  726. GenDefaultInitializeFuncName GenName(DstPtr.getAlignment(), getContext());
  727. std::string FuncName = GenName.getName(QT, IsVolatile);
  728. callSpecialFunction(GenDefaultInitialize(getContext()), FuncName, QT,
  729. IsVolatile, *this, std::array<Address, 1>({{DstPtr}}));
  730. }
  731. std::string CodeGenFunction::getNonTrivialCopyConstructorStr(
  732. QualType QT, CharUnits Alignment, bool IsVolatile, ASTContext &Ctx) {
  733. GenBinaryFuncName<false> GenName("", Alignment, Alignment, Ctx);
  734. return GenName.getName(QT, IsVolatile);
  735. }
  736. std::string CodeGenFunction::getNonTrivialDestructorStr(QualType QT,
  737. CharUnits Alignment,
  738. bool IsVolatile,
  739. ASTContext &Ctx) {
  740. GenDestructorFuncName GenName("", Alignment, Ctx);
  741. return GenName.getName(QT, IsVolatile);
  742. }
  743. void CodeGenFunction::callCStructDestructor(LValue Dst) {
  744. bool IsVolatile = Dst.isVolatile();
  745. Address DstPtr = Dst.getAddress();
  746. QualType QT = Dst.getType();
  747. GenDestructorFuncName GenName("__destructor_", DstPtr.getAlignment(),
  748. getContext());
  749. std::string FuncName = GenName.getName(QT, IsVolatile);
  750. callSpecialFunction(GenDestructor(getContext()), FuncName, QT, IsVolatile,
  751. *this, std::array<Address, 1>({{DstPtr}}));
  752. }
  753. void CodeGenFunction::callCStructCopyConstructor(LValue Dst, LValue Src) {
  754. bool IsVolatile = Dst.isVolatile() || Src.isVolatile();
  755. Address DstPtr = Dst.getAddress(), SrcPtr = Src.getAddress();
  756. QualType QT = Dst.getType();
  757. GenBinaryFuncName<false> GenName("__copy_constructor_", DstPtr.getAlignment(),
  758. SrcPtr.getAlignment(), getContext());
  759. std::string FuncName = GenName.getName(QT, IsVolatile);
  760. callSpecialFunction(GenCopyConstructor(getContext()), FuncName, QT,
  761. IsVolatile, *this,
  762. std::array<Address, 2>({{DstPtr, SrcPtr}}));
  763. }
  764. void CodeGenFunction::callCStructCopyAssignmentOperator(LValue Dst, LValue Src
  765. ) {
  766. bool IsVolatile = Dst.isVolatile() || Src.isVolatile();
  767. Address DstPtr = Dst.getAddress(), SrcPtr = Src.getAddress();
  768. QualType QT = Dst.getType();
  769. GenBinaryFuncName<false> GenName("__copy_assignment_", DstPtr.getAlignment(),
  770. SrcPtr.getAlignment(), getContext());
  771. std::string FuncName = GenName.getName(QT, IsVolatile);
  772. callSpecialFunction(GenCopyAssignment(getContext()), FuncName, QT, IsVolatile,
  773. *this, std::array<Address, 2>({{DstPtr, SrcPtr}}));
  774. }
  775. void CodeGenFunction::callCStructMoveConstructor(LValue Dst, LValue Src) {
  776. bool IsVolatile = Dst.isVolatile() || Src.isVolatile();
  777. Address DstPtr = Dst.getAddress(), SrcPtr = Src.getAddress();
  778. QualType QT = Dst.getType();
  779. GenBinaryFuncName<true> GenName("__move_constructor_", DstPtr.getAlignment(),
  780. SrcPtr.getAlignment(), getContext());
  781. std::string FuncName = GenName.getName(QT, IsVolatile);
  782. callSpecialFunction(GenMoveConstructor(getContext()), FuncName, QT,
  783. IsVolatile, *this,
  784. std::array<Address, 2>({{DstPtr, SrcPtr}}));
  785. }
  786. void CodeGenFunction::callCStructMoveAssignmentOperator(LValue Dst, LValue Src
  787. ) {
  788. bool IsVolatile = Dst.isVolatile() || Src.isVolatile();
  789. Address DstPtr = Dst.getAddress(), SrcPtr = Src.getAddress();
  790. QualType QT = Dst.getType();
  791. GenBinaryFuncName<true> GenName("__move_assignment_", DstPtr.getAlignment(),
  792. SrcPtr.getAlignment(), getContext());
  793. std::string FuncName = GenName.getName(QT, IsVolatile);
  794. callSpecialFunction(GenMoveAssignment(getContext()), FuncName, QT, IsVolatile,
  795. *this, std::array<Address, 2>({{DstPtr, SrcPtr}}));
  796. }
  797. llvm::Function *clang::CodeGen::getNonTrivialCStructDefaultConstructor(
  798. CodeGenModule &CGM, CharUnits DstAlignment, bool IsVolatile, QualType QT) {
  799. ASTContext &Ctx = CGM.getContext();
  800. GenDefaultInitializeFuncName GenName(DstAlignment, Ctx);
  801. std::string FuncName = GenName.getName(QT, IsVolatile);
  802. return getSpecialFunction(GenDefaultInitialize(Ctx), FuncName, QT, IsVolatile,
  803. std::array<CharUnits, 1>({{DstAlignment}}), CGM);
  804. }
  805. llvm::Function *clang::CodeGen::getNonTrivialCStructCopyConstructor(
  806. CodeGenModule &CGM, CharUnits DstAlignment, CharUnits SrcAlignment,
  807. bool IsVolatile, QualType QT) {
  808. ASTContext &Ctx = CGM.getContext();
  809. GenBinaryFuncName<false> GenName("__copy_constructor_", DstAlignment,
  810. SrcAlignment, Ctx);
  811. std::string FuncName = GenName.getName(QT, IsVolatile);
  812. return getSpecialFunction(
  813. GenCopyConstructor(Ctx), FuncName, QT, IsVolatile,
  814. std::array<CharUnits, 2>({{DstAlignment, SrcAlignment}}), CGM);
  815. }
  816. llvm::Function *clang::CodeGen::getNonTrivialCStructMoveConstructor(
  817. CodeGenModule &CGM, CharUnits DstAlignment, CharUnits SrcAlignment,
  818. bool IsVolatile, QualType QT) {
  819. ASTContext &Ctx = CGM.getContext();
  820. GenBinaryFuncName<true> GenName("__move_constructor_", DstAlignment,
  821. SrcAlignment, Ctx);
  822. std::string FuncName = GenName.getName(QT, IsVolatile);
  823. return getSpecialFunction(
  824. GenMoveConstructor(Ctx), FuncName, QT, IsVolatile,
  825. std::array<CharUnits, 2>({{DstAlignment, SrcAlignment}}), CGM);
  826. }
  827. llvm::Function *clang::CodeGen::getNonTrivialCStructCopyAssignmentOperator(
  828. CodeGenModule &CGM, CharUnits DstAlignment, CharUnits SrcAlignment,
  829. bool IsVolatile, QualType QT) {
  830. ASTContext &Ctx = CGM.getContext();
  831. GenBinaryFuncName<false> GenName("__copy_assignment_", DstAlignment,
  832. SrcAlignment, Ctx);
  833. std::string FuncName = GenName.getName(QT, IsVolatile);
  834. return getSpecialFunction(
  835. GenCopyAssignment(Ctx), FuncName, QT, IsVolatile,
  836. std::array<CharUnits, 2>({{DstAlignment, SrcAlignment}}), CGM);
  837. }
  838. llvm::Function *clang::CodeGen::getNonTrivialCStructMoveAssignmentOperator(
  839. CodeGenModule &CGM, CharUnits DstAlignment, CharUnits SrcAlignment,
  840. bool IsVolatile, QualType QT) {
  841. ASTContext &Ctx = CGM.getContext();
  842. GenBinaryFuncName<true> GenName("__move_assignment_", DstAlignment,
  843. SrcAlignment, Ctx);
  844. std::string FuncName = GenName.getName(QT, IsVolatile);
  845. return getSpecialFunction(
  846. GenMoveAssignment(Ctx), FuncName, QT, IsVolatile,
  847. std::array<CharUnits, 2>({{DstAlignment, SrcAlignment}}), CGM);
  848. }
  849. llvm::Function *clang::CodeGen::getNonTrivialCStructDestructor(
  850. CodeGenModule &CGM, CharUnits DstAlignment, bool IsVolatile, QualType QT) {
  851. ASTContext &Ctx = CGM.getContext();
  852. GenDestructorFuncName GenName("__destructor_", DstAlignment, Ctx);
  853. std::string FuncName = GenName.getName(QT, IsVolatile);
  854. return getSpecialFunction(GenDestructor(Ctx), FuncName, QT, IsVolatile,
  855. std::array<CharUnits, 1>({{DstAlignment}}), CGM);
  856. }