CodeGenModule.cpp 92 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530
  1. //===--- CodeGenModule.cpp - Emit LLVM Code from ASTs for a Module --------===//
  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 coordinates the per-module state used while generating code.
  11. //
  12. //===----------------------------------------------------------------------===//
  13. #include "CodeGenModule.h"
  14. #include "CGDebugInfo.h"
  15. #include "CodeGenFunction.h"
  16. #include "CodeGenTBAA.h"
  17. #include "CGCall.h"
  18. #include "CGCUDARuntime.h"
  19. #include "CGCXXABI.h"
  20. #include "CGObjCRuntime.h"
  21. #include "CGOpenCLRuntime.h"
  22. #include "TargetInfo.h"
  23. #include "clang/Frontend/CodeGenOptions.h"
  24. #include "clang/AST/ASTContext.h"
  25. #include "clang/AST/CharUnits.h"
  26. #include "clang/AST/DeclObjC.h"
  27. #include "clang/AST/DeclCXX.h"
  28. #include "clang/AST/DeclTemplate.h"
  29. #include "clang/AST/Mangle.h"
  30. #include "clang/AST/RecordLayout.h"
  31. #include "clang/AST/RecursiveASTVisitor.h"
  32. #include "clang/Basic/Builtins.h"
  33. #include "clang/Basic/Diagnostic.h"
  34. #include "clang/Basic/SourceManager.h"
  35. #include "clang/Basic/TargetInfo.h"
  36. #include "clang/Basic/ConvertUTF.h"
  37. #include "llvm/CallingConv.h"
  38. #include "llvm/Module.h"
  39. #include "llvm/Intrinsics.h"
  40. #include "llvm/LLVMContext.h"
  41. #include "llvm/ADT/Triple.h"
  42. #include "llvm/Target/Mangler.h"
  43. #include "llvm/Target/TargetData.h"
  44. #include "llvm/Support/CallSite.h"
  45. #include "llvm/Support/ErrorHandling.h"
  46. using namespace clang;
  47. using namespace CodeGen;
  48. static const char AnnotationSection[] = "llvm.metadata";
  49. static CGCXXABI &createCXXABI(CodeGenModule &CGM) {
  50. switch (CGM.getContext().getTargetInfo().getCXXABI()) {
  51. case CXXABI_ARM: return *CreateARMCXXABI(CGM);
  52. case CXXABI_Itanium: return *CreateItaniumCXXABI(CGM);
  53. case CXXABI_Microsoft: return *CreateMicrosoftCXXABI(CGM);
  54. }
  55. llvm_unreachable("invalid C++ ABI kind");
  56. }
  57. CodeGenModule::CodeGenModule(ASTContext &C, const CodeGenOptions &CGO,
  58. llvm::Module &M, const llvm::TargetData &TD,
  59. DiagnosticsEngine &diags)
  60. : Context(C), Features(C.getLangOptions()), CodeGenOpts(CGO), TheModule(M),
  61. TheTargetData(TD), TheTargetCodeGenInfo(0), Diags(diags),
  62. ABI(createCXXABI(*this)),
  63. Types(C, M, TD, getTargetCodeGenInfo().getABIInfo(), ABI, CGO),
  64. TBAA(0),
  65. VTables(*this), ObjCRuntime(0), OpenCLRuntime(0), CUDARuntime(0),
  66. DebugInfo(0), ARCData(0), RRData(0), CFConstantStringClassRef(0),
  67. ConstantStringClassRef(0), NSConstantStringType(0),
  68. VMContext(M.getContext()),
  69. NSConcreteGlobalBlock(0), NSConcreteStackBlock(0),
  70. BlockObjectAssign(0), BlockObjectDispose(0),
  71. BlockDescriptorType(0), GenericBlockLiteralType(0) {
  72. // Initialize the type cache.
  73. llvm::LLVMContext &LLVMContext = M.getContext();
  74. VoidTy = llvm::Type::getVoidTy(LLVMContext);
  75. Int8Ty = llvm::Type::getInt8Ty(LLVMContext);
  76. Int16Ty = llvm::Type::getInt16Ty(LLVMContext);
  77. Int32Ty = llvm::Type::getInt32Ty(LLVMContext);
  78. Int64Ty = llvm::Type::getInt64Ty(LLVMContext);
  79. FloatTy = llvm::Type::getFloatTy(LLVMContext);
  80. DoubleTy = llvm::Type::getDoubleTy(LLVMContext);
  81. PointerWidthInBits = C.getTargetInfo().getPointerWidth(0);
  82. PointerAlignInBytes =
  83. C.toCharUnitsFromBits(C.getTargetInfo().getPointerAlign(0)).getQuantity();
  84. IntTy = llvm::IntegerType::get(LLVMContext, C.getTargetInfo().getIntWidth());
  85. IntPtrTy = llvm::IntegerType::get(LLVMContext, PointerWidthInBits);
  86. Int8PtrTy = Int8Ty->getPointerTo(0);
  87. Int8PtrPtrTy = Int8PtrTy->getPointerTo(0);
  88. if (Features.ObjC1)
  89. createObjCRuntime();
  90. if (Features.OpenCL)
  91. createOpenCLRuntime();
  92. if (Features.CUDA)
  93. createCUDARuntime();
  94. // Enable TBAA unless it's suppressed.
  95. if (!CodeGenOpts.RelaxedAliasing && CodeGenOpts.OptimizationLevel > 0)
  96. TBAA = new CodeGenTBAA(Context, VMContext, getLangOptions(),
  97. ABI.getMangleContext());
  98. // If debug info or coverage generation is enabled, create the CGDebugInfo
  99. // object.
  100. if (CodeGenOpts.DebugInfo || CodeGenOpts.EmitGcovArcs ||
  101. CodeGenOpts.EmitGcovNotes)
  102. DebugInfo = new CGDebugInfo(*this);
  103. Block.GlobalUniqueCount = 0;
  104. if (C.getLangOptions().ObjCAutoRefCount)
  105. ARCData = new ARCEntrypoints();
  106. RRData = new RREntrypoints();
  107. }
  108. CodeGenModule::~CodeGenModule() {
  109. delete ObjCRuntime;
  110. delete OpenCLRuntime;
  111. delete CUDARuntime;
  112. delete TheTargetCodeGenInfo;
  113. delete &ABI;
  114. delete TBAA;
  115. delete DebugInfo;
  116. delete ARCData;
  117. delete RRData;
  118. }
  119. void CodeGenModule::createObjCRuntime() {
  120. if (!Features.NeXTRuntime)
  121. ObjCRuntime = CreateGNUObjCRuntime(*this);
  122. else
  123. ObjCRuntime = CreateMacObjCRuntime(*this);
  124. }
  125. void CodeGenModule::createOpenCLRuntime() {
  126. OpenCLRuntime = new CGOpenCLRuntime(*this);
  127. }
  128. void CodeGenModule::createCUDARuntime() {
  129. CUDARuntime = CreateNVCUDARuntime(*this);
  130. }
  131. void CodeGenModule::Release() {
  132. EmitDeferred();
  133. EmitCXXGlobalInitFunc();
  134. EmitCXXGlobalDtorFunc();
  135. if (ObjCRuntime)
  136. if (llvm::Function *ObjCInitFunction = ObjCRuntime->ModuleInitFunction())
  137. AddGlobalCtor(ObjCInitFunction);
  138. EmitCtorList(GlobalCtors, "llvm.global_ctors");
  139. EmitCtorList(GlobalDtors, "llvm.global_dtors");
  140. EmitGlobalAnnotations();
  141. EmitLLVMUsed();
  142. SimplifyPersonality();
  143. if (getCodeGenOpts().EmitDeclMetadata)
  144. EmitDeclMetadata();
  145. if (getCodeGenOpts().EmitGcovArcs || getCodeGenOpts().EmitGcovNotes)
  146. EmitCoverageFile();
  147. if (DebugInfo)
  148. DebugInfo->finalize();
  149. }
  150. void CodeGenModule::UpdateCompletedType(const TagDecl *TD) {
  151. // Make sure that this type is translated.
  152. Types.UpdateCompletedType(TD);
  153. if (DebugInfo)
  154. DebugInfo->UpdateCompletedType(TD);
  155. }
  156. llvm::MDNode *CodeGenModule::getTBAAInfo(QualType QTy) {
  157. if (!TBAA)
  158. return 0;
  159. return TBAA->getTBAAInfo(QTy);
  160. }
  161. void CodeGenModule::DecorateInstruction(llvm::Instruction *Inst,
  162. llvm::MDNode *TBAAInfo) {
  163. Inst->setMetadata(llvm::LLVMContext::MD_tbaa, TBAAInfo);
  164. }
  165. bool CodeGenModule::isTargetDarwin() const {
  166. return getContext().getTargetInfo().getTriple().isOSDarwin();
  167. }
  168. void CodeGenModule::Error(SourceLocation loc, StringRef error) {
  169. unsigned diagID = getDiags().getCustomDiagID(DiagnosticsEngine::Error, error);
  170. getDiags().Report(Context.getFullLoc(loc), diagID);
  171. }
  172. /// ErrorUnsupported - Print out an error that codegen doesn't support the
  173. /// specified stmt yet.
  174. void CodeGenModule::ErrorUnsupported(const Stmt *S, const char *Type,
  175. bool OmitOnError) {
  176. if (OmitOnError && getDiags().hasErrorOccurred())
  177. return;
  178. unsigned DiagID = getDiags().getCustomDiagID(DiagnosticsEngine::Error,
  179. "cannot compile this %0 yet");
  180. std::string Msg = Type;
  181. getDiags().Report(Context.getFullLoc(S->getLocStart()), DiagID)
  182. << Msg << S->getSourceRange();
  183. }
  184. /// ErrorUnsupported - Print out an error that codegen doesn't support the
  185. /// specified decl yet.
  186. void CodeGenModule::ErrorUnsupported(const Decl *D, const char *Type,
  187. bool OmitOnError) {
  188. if (OmitOnError && getDiags().hasErrorOccurred())
  189. return;
  190. unsigned DiagID = getDiags().getCustomDiagID(DiagnosticsEngine::Error,
  191. "cannot compile this %0 yet");
  192. std::string Msg = Type;
  193. getDiags().Report(Context.getFullLoc(D->getLocation()), DiagID) << Msg;
  194. }
  195. llvm::ConstantInt *CodeGenModule::getSize(CharUnits size) {
  196. return llvm::ConstantInt::get(SizeTy, size.getQuantity());
  197. }
  198. void CodeGenModule::setGlobalVisibility(llvm::GlobalValue *GV,
  199. const NamedDecl *D) const {
  200. // Internal definitions always have default visibility.
  201. if (GV->hasLocalLinkage()) {
  202. GV->setVisibility(llvm::GlobalValue::DefaultVisibility);
  203. return;
  204. }
  205. // Set visibility for definitions.
  206. NamedDecl::LinkageInfo LV = D->getLinkageAndVisibility();
  207. if (LV.visibilityExplicit() || !GV->hasAvailableExternallyLinkage())
  208. GV->setVisibility(GetLLVMVisibility(LV.visibility()));
  209. }
  210. /// Set the symbol visibility of type information (vtable and RTTI)
  211. /// associated with the given type.
  212. void CodeGenModule::setTypeVisibility(llvm::GlobalValue *GV,
  213. const CXXRecordDecl *RD,
  214. TypeVisibilityKind TVK) const {
  215. setGlobalVisibility(GV, RD);
  216. if (!CodeGenOpts.HiddenWeakVTables)
  217. return;
  218. // We never want to drop the visibility for RTTI names.
  219. if (TVK == TVK_ForRTTIName)
  220. return;
  221. // We want to drop the visibility to hidden for weak type symbols.
  222. // This isn't possible if there might be unresolved references
  223. // elsewhere that rely on this symbol being visible.
  224. // This should be kept roughly in sync with setThunkVisibility
  225. // in CGVTables.cpp.
  226. // Preconditions.
  227. if (GV->getLinkage() != llvm::GlobalVariable::LinkOnceODRLinkage ||
  228. GV->getVisibility() != llvm::GlobalVariable::DefaultVisibility)
  229. return;
  230. // Don't override an explicit visibility attribute.
  231. if (RD->getExplicitVisibility())
  232. return;
  233. switch (RD->getTemplateSpecializationKind()) {
  234. // We have to disable the optimization if this is an EI definition
  235. // because there might be EI declarations in other shared objects.
  236. case TSK_ExplicitInstantiationDefinition:
  237. case TSK_ExplicitInstantiationDeclaration:
  238. return;
  239. // Every use of a non-template class's type information has to emit it.
  240. case TSK_Undeclared:
  241. break;
  242. // In theory, implicit instantiations can ignore the possibility of
  243. // an explicit instantiation declaration because there necessarily
  244. // must be an EI definition somewhere with default visibility. In
  245. // practice, it's possible to have an explicit instantiation for
  246. // an arbitrary template class, and linkers aren't necessarily able
  247. // to deal with mixed-visibility symbols.
  248. case TSK_ExplicitSpecialization:
  249. case TSK_ImplicitInstantiation:
  250. if (!CodeGenOpts.HiddenWeakTemplateVTables)
  251. return;
  252. break;
  253. }
  254. // If there's a key function, there may be translation units
  255. // that don't have the key function's definition. But ignore
  256. // this if we're emitting RTTI under -fno-rtti.
  257. if (!(TVK != TVK_ForRTTI) || Features.RTTI) {
  258. if (Context.getKeyFunction(RD))
  259. return;
  260. }
  261. // Otherwise, drop the visibility to hidden.
  262. GV->setVisibility(llvm::GlobalValue::HiddenVisibility);
  263. GV->setUnnamedAddr(true);
  264. }
  265. StringRef CodeGenModule::getMangledName(GlobalDecl GD) {
  266. const NamedDecl *ND = cast<NamedDecl>(GD.getDecl());
  267. StringRef &Str = MangledDeclNames[GD.getCanonicalDecl()];
  268. if (!Str.empty())
  269. return Str;
  270. if (!getCXXABI().getMangleContext().shouldMangleDeclName(ND)) {
  271. IdentifierInfo *II = ND->getIdentifier();
  272. assert(II && "Attempt to mangle unnamed decl.");
  273. Str = II->getName();
  274. return Str;
  275. }
  276. SmallString<256> Buffer;
  277. llvm::raw_svector_ostream Out(Buffer);
  278. if (const CXXConstructorDecl *D = dyn_cast<CXXConstructorDecl>(ND))
  279. getCXXABI().getMangleContext().mangleCXXCtor(D, GD.getCtorType(), Out);
  280. else if (const CXXDestructorDecl *D = dyn_cast<CXXDestructorDecl>(ND))
  281. getCXXABI().getMangleContext().mangleCXXDtor(D, GD.getDtorType(), Out);
  282. else if (const BlockDecl *BD = dyn_cast<BlockDecl>(ND))
  283. getCXXABI().getMangleContext().mangleBlock(BD, Out);
  284. else
  285. getCXXABI().getMangleContext().mangleName(ND, Out);
  286. // Allocate space for the mangled name.
  287. Out.flush();
  288. size_t Length = Buffer.size();
  289. char *Name = MangledNamesAllocator.Allocate<char>(Length);
  290. std::copy(Buffer.begin(), Buffer.end(), Name);
  291. Str = StringRef(Name, Length);
  292. return Str;
  293. }
  294. void CodeGenModule::getBlockMangledName(GlobalDecl GD, MangleBuffer &Buffer,
  295. const BlockDecl *BD) {
  296. MangleContext &MangleCtx = getCXXABI().getMangleContext();
  297. const Decl *D = GD.getDecl();
  298. llvm::raw_svector_ostream Out(Buffer.getBuffer());
  299. if (D == 0)
  300. MangleCtx.mangleGlobalBlock(BD, Out);
  301. else if (const CXXConstructorDecl *CD = dyn_cast<CXXConstructorDecl>(D))
  302. MangleCtx.mangleCtorBlock(CD, GD.getCtorType(), BD, Out);
  303. else if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(D))
  304. MangleCtx.mangleDtorBlock(DD, GD.getDtorType(), BD, Out);
  305. else
  306. MangleCtx.mangleBlock(cast<DeclContext>(D), BD, Out);
  307. }
  308. llvm::GlobalValue *CodeGenModule::GetGlobalValue(StringRef Name) {
  309. return getModule().getNamedValue(Name);
  310. }
  311. /// AddGlobalCtor - Add a function to the list that will be called before
  312. /// main() runs.
  313. void CodeGenModule::AddGlobalCtor(llvm::Function * Ctor, int Priority) {
  314. // FIXME: Type coercion of void()* types.
  315. GlobalCtors.push_back(std::make_pair(Ctor, Priority));
  316. }
  317. /// AddGlobalDtor - Add a function to the list that will be called
  318. /// when the module is unloaded.
  319. void CodeGenModule::AddGlobalDtor(llvm::Function * Dtor, int Priority) {
  320. // FIXME: Type coercion of void()* types.
  321. GlobalDtors.push_back(std::make_pair(Dtor, Priority));
  322. }
  323. void CodeGenModule::EmitCtorList(const CtorList &Fns, const char *GlobalName) {
  324. // Ctor function type is void()*.
  325. llvm::FunctionType* CtorFTy = llvm::FunctionType::get(VoidTy, false);
  326. llvm::Type *CtorPFTy = llvm::PointerType::getUnqual(CtorFTy);
  327. // Get the type of a ctor entry, { i32, void ()* }.
  328. llvm::StructType *CtorStructTy =
  329. llvm::StructType::get(Int32Ty, llvm::PointerType::getUnqual(CtorFTy), NULL);
  330. // Construct the constructor and destructor arrays.
  331. SmallVector<llvm::Constant*, 8> Ctors;
  332. for (CtorList::const_iterator I = Fns.begin(), E = Fns.end(); I != E; ++I) {
  333. llvm::Constant *S[] = {
  334. llvm::ConstantInt::get(Int32Ty, I->second, false),
  335. llvm::ConstantExpr::getBitCast(I->first, CtorPFTy)
  336. };
  337. Ctors.push_back(llvm::ConstantStruct::get(CtorStructTy, S));
  338. }
  339. if (!Ctors.empty()) {
  340. llvm::ArrayType *AT = llvm::ArrayType::get(CtorStructTy, Ctors.size());
  341. new llvm::GlobalVariable(TheModule, AT, false,
  342. llvm::GlobalValue::AppendingLinkage,
  343. llvm::ConstantArray::get(AT, Ctors),
  344. GlobalName);
  345. }
  346. }
  347. llvm::GlobalValue::LinkageTypes
  348. CodeGenModule::getFunctionLinkage(const FunctionDecl *D) {
  349. GVALinkage Linkage = getContext().GetGVALinkageForFunction(D);
  350. if (Linkage == GVA_Internal)
  351. return llvm::Function::InternalLinkage;
  352. if (D->hasAttr<DLLExportAttr>())
  353. return llvm::Function::DLLExportLinkage;
  354. if (D->hasAttr<WeakAttr>())
  355. return llvm::Function::WeakAnyLinkage;
  356. // In C99 mode, 'inline' functions are guaranteed to have a strong
  357. // definition somewhere else, so we can use available_externally linkage.
  358. if (Linkage == GVA_C99Inline)
  359. return llvm::Function::AvailableExternallyLinkage;
  360. // Note that Apple's kernel linker doesn't support symbol
  361. // coalescing, so we need to avoid linkonce and weak linkages there.
  362. // Normally, this means we just map to internal, but for explicit
  363. // instantiations we'll map to external.
  364. // In C++, the compiler has to emit a definition in every translation unit
  365. // that references the function. We should use linkonce_odr because
  366. // a) if all references in this translation unit are optimized away, we
  367. // don't need to codegen it. b) if the function persists, it needs to be
  368. // merged with other definitions. c) C++ has the ODR, so we know the
  369. // definition is dependable.
  370. if (Linkage == GVA_CXXInline || Linkage == GVA_TemplateInstantiation)
  371. return !Context.getLangOptions().AppleKext
  372. ? llvm::Function::LinkOnceODRLinkage
  373. : llvm::Function::InternalLinkage;
  374. // An explicit instantiation of a template has weak linkage, since
  375. // explicit instantiations can occur in multiple translation units
  376. // and must all be equivalent. However, we are not allowed to
  377. // throw away these explicit instantiations.
  378. if (Linkage == GVA_ExplicitTemplateInstantiation)
  379. return !Context.getLangOptions().AppleKext
  380. ? llvm::Function::WeakODRLinkage
  381. : llvm::Function::ExternalLinkage;
  382. // Otherwise, we have strong external linkage.
  383. assert(Linkage == GVA_StrongExternal);
  384. return llvm::Function::ExternalLinkage;
  385. }
  386. /// SetFunctionDefinitionAttributes - Set attributes for a global.
  387. ///
  388. /// FIXME: This is currently only done for aliases and functions, but not for
  389. /// variables (these details are set in EmitGlobalVarDefinition for variables).
  390. void CodeGenModule::SetFunctionDefinitionAttributes(const FunctionDecl *D,
  391. llvm::GlobalValue *GV) {
  392. SetCommonAttributes(D, GV);
  393. }
  394. void CodeGenModule::SetLLVMFunctionAttributes(const Decl *D,
  395. const CGFunctionInfo &Info,
  396. llvm::Function *F) {
  397. unsigned CallingConv;
  398. AttributeListType AttributeList;
  399. ConstructAttributeList(Info, D, AttributeList, CallingConv);
  400. F->setAttributes(llvm::AttrListPtr::get(AttributeList.begin(),
  401. AttributeList.size()));
  402. F->setCallingConv(static_cast<llvm::CallingConv::ID>(CallingConv));
  403. }
  404. /// Determines whether the language options require us to model
  405. /// unwind exceptions. We treat -fexceptions as mandating this
  406. /// except under the fragile ObjC ABI with only ObjC exceptions
  407. /// enabled. This means, for example, that C with -fexceptions
  408. /// enables this.
  409. static bool hasUnwindExceptions(const LangOptions &Features) {
  410. // If exceptions are completely disabled, obviously this is false.
  411. if (!Features.Exceptions) return false;
  412. // If C++ exceptions are enabled, this is true.
  413. if (Features.CXXExceptions) return true;
  414. // If ObjC exceptions are enabled, this depends on the ABI.
  415. if (Features.ObjCExceptions) {
  416. if (!Features.ObjCNonFragileABI) return false;
  417. }
  418. return true;
  419. }
  420. void CodeGenModule::SetLLVMFunctionAttributesForDefinition(const Decl *D,
  421. llvm::Function *F) {
  422. if (CodeGenOpts.UnwindTables)
  423. F->setHasUWTable();
  424. if (!hasUnwindExceptions(Features))
  425. F->addFnAttr(llvm::Attribute::NoUnwind);
  426. if (D->hasAttr<NakedAttr>()) {
  427. // Naked implies noinline: we should not be inlining such functions.
  428. F->addFnAttr(llvm::Attribute::Naked);
  429. F->addFnAttr(llvm::Attribute::NoInline);
  430. }
  431. if (D->hasAttr<NoInlineAttr>())
  432. F->addFnAttr(llvm::Attribute::NoInline);
  433. // (noinline wins over always_inline, and we can't specify both in IR)
  434. if (D->hasAttr<AlwaysInlineAttr>() &&
  435. !F->hasFnAttr(llvm::Attribute::NoInline))
  436. F->addFnAttr(llvm::Attribute::AlwaysInline);
  437. if (isa<CXXConstructorDecl>(D) || isa<CXXDestructorDecl>(D))
  438. F->setUnnamedAddr(true);
  439. if (Features.getStackProtector() == LangOptions::SSPOn)
  440. F->addFnAttr(llvm::Attribute::StackProtect);
  441. else if (Features.getStackProtector() == LangOptions::SSPReq)
  442. F->addFnAttr(llvm::Attribute::StackProtectReq);
  443. if (Features.AddressSanitizer) {
  444. // When AddressSanitizer is enabled, set AddressSafety attribute
  445. // unless __attribute__((no_address_safety_analysis)) is used.
  446. if (!D->hasAttr<NoAddressSafetyAnalysisAttr>())
  447. F->addFnAttr(llvm::Attribute::AddressSafety);
  448. }
  449. unsigned alignment = D->getMaxAlignment() / Context.getCharWidth();
  450. if (alignment)
  451. F->setAlignment(alignment);
  452. // C++ ABI requires 2-byte alignment for member functions.
  453. if (F->getAlignment() < 2 && isa<CXXMethodDecl>(D))
  454. F->setAlignment(2);
  455. }
  456. void CodeGenModule::SetCommonAttributes(const Decl *D,
  457. llvm::GlobalValue *GV) {
  458. if (const NamedDecl *ND = dyn_cast<NamedDecl>(D))
  459. setGlobalVisibility(GV, ND);
  460. else
  461. GV->setVisibility(llvm::GlobalValue::DefaultVisibility);
  462. if (D->hasAttr<UsedAttr>())
  463. AddUsedGlobal(GV);
  464. if (const SectionAttr *SA = D->getAttr<SectionAttr>())
  465. GV->setSection(SA->getName());
  466. getTargetCodeGenInfo().SetTargetAttributes(D, GV, *this);
  467. }
  468. void CodeGenModule::SetInternalFunctionAttributes(const Decl *D,
  469. llvm::Function *F,
  470. const CGFunctionInfo &FI) {
  471. SetLLVMFunctionAttributes(D, FI, F);
  472. SetLLVMFunctionAttributesForDefinition(D, F);
  473. F->setLinkage(llvm::Function::InternalLinkage);
  474. SetCommonAttributes(D, F);
  475. }
  476. void CodeGenModule::SetFunctionAttributes(GlobalDecl GD,
  477. llvm::Function *F,
  478. bool IsIncompleteFunction) {
  479. if (unsigned IID = F->getIntrinsicID()) {
  480. // If this is an intrinsic function, set the function's attributes
  481. // to the intrinsic's attributes.
  482. F->setAttributes(llvm::Intrinsic::getAttributes((llvm::Intrinsic::ID)IID));
  483. return;
  484. }
  485. const FunctionDecl *FD = cast<FunctionDecl>(GD.getDecl());
  486. if (!IsIncompleteFunction)
  487. SetLLVMFunctionAttributes(FD, getTypes().getFunctionInfo(GD), F);
  488. // Only a few attributes are set on declarations; these may later be
  489. // overridden by a definition.
  490. if (FD->hasAttr<DLLImportAttr>()) {
  491. F->setLinkage(llvm::Function::DLLImportLinkage);
  492. } else if (FD->hasAttr<WeakAttr>() ||
  493. FD->isWeakImported()) {
  494. // "extern_weak" is overloaded in LLVM; we probably should have
  495. // separate linkage types for this.
  496. F->setLinkage(llvm::Function::ExternalWeakLinkage);
  497. } else {
  498. F->setLinkage(llvm::Function::ExternalLinkage);
  499. NamedDecl::LinkageInfo LV = FD->getLinkageAndVisibility();
  500. if (LV.linkage() == ExternalLinkage && LV.visibilityExplicit()) {
  501. F->setVisibility(GetLLVMVisibility(LV.visibility()));
  502. }
  503. }
  504. if (const SectionAttr *SA = FD->getAttr<SectionAttr>())
  505. F->setSection(SA->getName());
  506. }
  507. void CodeGenModule::AddUsedGlobal(llvm::GlobalValue *GV) {
  508. assert(!GV->isDeclaration() &&
  509. "Only globals with definition can force usage.");
  510. LLVMUsed.push_back(GV);
  511. }
  512. void CodeGenModule::EmitLLVMUsed() {
  513. // Don't create llvm.used if there is no need.
  514. if (LLVMUsed.empty())
  515. return;
  516. // Convert LLVMUsed to what ConstantArray needs.
  517. SmallVector<llvm::Constant*, 8> UsedArray;
  518. UsedArray.resize(LLVMUsed.size());
  519. for (unsigned i = 0, e = LLVMUsed.size(); i != e; ++i) {
  520. UsedArray[i] =
  521. llvm::ConstantExpr::getBitCast(cast<llvm::Constant>(&*LLVMUsed[i]),
  522. Int8PtrTy);
  523. }
  524. if (UsedArray.empty())
  525. return;
  526. llvm::ArrayType *ATy = llvm::ArrayType::get(Int8PtrTy, UsedArray.size());
  527. llvm::GlobalVariable *GV =
  528. new llvm::GlobalVariable(getModule(), ATy, false,
  529. llvm::GlobalValue::AppendingLinkage,
  530. llvm::ConstantArray::get(ATy, UsedArray),
  531. "llvm.used");
  532. GV->setSection("llvm.metadata");
  533. }
  534. void CodeGenModule::EmitDeferred() {
  535. // Emit code for any potentially referenced deferred decls. Since a
  536. // previously unused static decl may become used during the generation of code
  537. // for a static function, iterate until no changes are made.
  538. while (!DeferredDeclsToEmit.empty() || !DeferredVTables.empty()) {
  539. if (!DeferredVTables.empty()) {
  540. const CXXRecordDecl *RD = DeferredVTables.back();
  541. DeferredVTables.pop_back();
  542. getVTables().GenerateClassData(getVTableLinkage(RD), RD);
  543. continue;
  544. }
  545. GlobalDecl D = DeferredDeclsToEmit.back();
  546. DeferredDeclsToEmit.pop_back();
  547. // Check to see if we've already emitted this. This is necessary
  548. // for a couple of reasons: first, decls can end up in the
  549. // deferred-decls queue multiple times, and second, decls can end
  550. // up with definitions in unusual ways (e.g. by an extern inline
  551. // function acquiring a strong function redefinition). Just
  552. // ignore these cases.
  553. //
  554. // TODO: That said, looking this up multiple times is very wasteful.
  555. StringRef Name = getMangledName(D);
  556. llvm::GlobalValue *CGRef = GetGlobalValue(Name);
  557. assert(CGRef && "Deferred decl wasn't referenced?");
  558. if (!CGRef->isDeclaration())
  559. continue;
  560. // GlobalAlias::isDeclaration() defers to the aliasee, but for our
  561. // purposes an alias counts as a definition.
  562. if (isa<llvm::GlobalAlias>(CGRef))
  563. continue;
  564. // Otherwise, emit the definition and move on to the next one.
  565. EmitGlobalDefinition(D);
  566. }
  567. }
  568. void CodeGenModule::EmitGlobalAnnotations() {
  569. if (Annotations.empty())
  570. return;
  571. // Create a new global variable for the ConstantStruct in the Module.
  572. llvm::Constant *Array = llvm::ConstantArray::get(llvm::ArrayType::get(
  573. Annotations[0]->getType(), Annotations.size()), Annotations);
  574. llvm::GlobalValue *gv = new llvm::GlobalVariable(getModule(),
  575. Array->getType(), false, llvm::GlobalValue::AppendingLinkage, Array,
  576. "llvm.global.annotations");
  577. gv->setSection(AnnotationSection);
  578. }
  579. llvm::Constant *CodeGenModule::EmitAnnotationString(llvm::StringRef Str) {
  580. llvm::StringMap<llvm::Constant*>::iterator i = AnnotationStrings.find(Str);
  581. if (i != AnnotationStrings.end())
  582. return i->second;
  583. // Not found yet, create a new global.
  584. llvm::Constant *s = llvm::ConstantDataArray::getString(getLLVMContext(), Str);
  585. llvm::GlobalValue *gv = new llvm::GlobalVariable(getModule(), s->getType(),
  586. true, llvm::GlobalValue::PrivateLinkage, s, ".str");
  587. gv->setSection(AnnotationSection);
  588. gv->setUnnamedAddr(true);
  589. AnnotationStrings[Str] = gv;
  590. return gv;
  591. }
  592. llvm::Constant *CodeGenModule::EmitAnnotationUnit(SourceLocation Loc) {
  593. SourceManager &SM = getContext().getSourceManager();
  594. PresumedLoc PLoc = SM.getPresumedLoc(Loc);
  595. if (PLoc.isValid())
  596. return EmitAnnotationString(PLoc.getFilename());
  597. return EmitAnnotationString(SM.getBufferName(Loc));
  598. }
  599. llvm::Constant *CodeGenModule::EmitAnnotationLineNo(SourceLocation L) {
  600. SourceManager &SM = getContext().getSourceManager();
  601. PresumedLoc PLoc = SM.getPresumedLoc(L);
  602. unsigned LineNo = PLoc.isValid() ? PLoc.getLine() :
  603. SM.getExpansionLineNumber(L);
  604. return llvm::ConstantInt::get(Int32Ty, LineNo);
  605. }
  606. llvm::Constant *CodeGenModule::EmitAnnotateAttr(llvm::GlobalValue *GV,
  607. const AnnotateAttr *AA,
  608. SourceLocation L) {
  609. // Get the globals for file name, annotation, and the line number.
  610. llvm::Constant *AnnoGV = EmitAnnotationString(AA->getAnnotation()),
  611. *UnitGV = EmitAnnotationUnit(L),
  612. *LineNoCst = EmitAnnotationLineNo(L);
  613. // Create the ConstantStruct for the global annotation.
  614. llvm::Constant *Fields[4] = {
  615. llvm::ConstantExpr::getBitCast(GV, Int8PtrTy),
  616. llvm::ConstantExpr::getBitCast(AnnoGV, Int8PtrTy),
  617. llvm::ConstantExpr::getBitCast(UnitGV, Int8PtrTy),
  618. LineNoCst
  619. };
  620. return llvm::ConstantStruct::getAnon(Fields);
  621. }
  622. void CodeGenModule::AddGlobalAnnotations(const ValueDecl *D,
  623. llvm::GlobalValue *GV) {
  624. assert(D->hasAttr<AnnotateAttr>() && "no annotate attribute");
  625. // Get the struct elements for these annotations.
  626. for (specific_attr_iterator<AnnotateAttr>
  627. ai = D->specific_attr_begin<AnnotateAttr>(),
  628. ae = D->specific_attr_end<AnnotateAttr>(); ai != ae; ++ai)
  629. Annotations.push_back(EmitAnnotateAttr(GV, *ai, D->getLocation()));
  630. }
  631. bool CodeGenModule::MayDeferGeneration(const ValueDecl *Global) {
  632. // Never defer when EmitAllDecls is specified.
  633. if (Features.EmitAllDecls)
  634. return false;
  635. return !getContext().DeclMustBeEmitted(Global);
  636. }
  637. llvm::Constant *CodeGenModule::GetWeakRefReference(const ValueDecl *VD) {
  638. const AliasAttr *AA = VD->getAttr<AliasAttr>();
  639. assert(AA && "No alias?");
  640. llvm::Type *DeclTy = getTypes().ConvertTypeForMem(VD->getType());
  641. // See if there is already something with the target's name in the module.
  642. llvm::GlobalValue *Entry = GetGlobalValue(AA->getAliasee());
  643. llvm::Constant *Aliasee;
  644. if (isa<llvm::FunctionType>(DeclTy))
  645. Aliasee = GetOrCreateLLVMFunction(AA->getAliasee(), DeclTy, GlobalDecl(),
  646. /*ForVTable=*/false);
  647. else
  648. Aliasee = GetOrCreateLLVMGlobal(AA->getAliasee(),
  649. llvm::PointerType::getUnqual(DeclTy), 0);
  650. if (!Entry) {
  651. llvm::GlobalValue* F = cast<llvm::GlobalValue>(Aliasee);
  652. F->setLinkage(llvm::Function::ExternalWeakLinkage);
  653. WeakRefReferences.insert(F);
  654. }
  655. return Aliasee;
  656. }
  657. void CodeGenModule::EmitGlobal(GlobalDecl GD) {
  658. const ValueDecl *Global = cast<ValueDecl>(GD.getDecl());
  659. // Weak references don't produce any output by themselves.
  660. if (Global->hasAttr<WeakRefAttr>())
  661. return;
  662. // If this is an alias definition (which otherwise looks like a declaration)
  663. // emit it now.
  664. if (Global->hasAttr<AliasAttr>())
  665. return EmitAliasDefinition(GD);
  666. // If this is CUDA, be selective about which declarations we emit.
  667. if (Features.CUDA) {
  668. if (CodeGenOpts.CUDAIsDevice) {
  669. if (!Global->hasAttr<CUDADeviceAttr>() &&
  670. !Global->hasAttr<CUDAGlobalAttr>() &&
  671. !Global->hasAttr<CUDAConstantAttr>() &&
  672. !Global->hasAttr<CUDASharedAttr>())
  673. return;
  674. } else {
  675. if (!Global->hasAttr<CUDAHostAttr>() && (
  676. Global->hasAttr<CUDADeviceAttr>() ||
  677. Global->hasAttr<CUDAConstantAttr>() ||
  678. Global->hasAttr<CUDASharedAttr>()))
  679. return;
  680. }
  681. }
  682. // Ignore declarations, they will be emitted on their first use.
  683. if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Global)) {
  684. // Forward declarations are emitted lazily on first use.
  685. if (!FD->doesThisDeclarationHaveABody()) {
  686. if (!FD->doesDeclarationForceExternallyVisibleDefinition())
  687. return;
  688. const FunctionDecl *InlineDefinition = 0;
  689. FD->getBody(InlineDefinition);
  690. StringRef MangledName = getMangledName(GD);
  691. llvm::StringMap<GlobalDecl>::iterator DDI =
  692. DeferredDecls.find(MangledName);
  693. if (DDI != DeferredDecls.end())
  694. DeferredDecls.erase(DDI);
  695. EmitGlobalDefinition(InlineDefinition);
  696. return;
  697. }
  698. } else {
  699. const VarDecl *VD = cast<VarDecl>(Global);
  700. assert(VD->isFileVarDecl() && "Cannot emit local var decl as global.");
  701. if (VD->isThisDeclarationADefinition() != VarDecl::Definition)
  702. return;
  703. }
  704. // Defer code generation when possible if this is a static definition, inline
  705. // function etc. These we only want to emit if they are used.
  706. if (!MayDeferGeneration(Global)) {
  707. // Emit the definition if it can't be deferred.
  708. EmitGlobalDefinition(GD);
  709. return;
  710. }
  711. // If we're deferring emission of a C++ variable with an
  712. // initializer, remember the order in which it appeared in the file.
  713. if (getLangOptions().CPlusPlus && isa<VarDecl>(Global) &&
  714. cast<VarDecl>(Global)->hasInit()) {
  715. DelayedCXXInitPosition[Global] = CXXGlobalInits.size();
  716. CXXGlobalInits.push_back(0);
  717. }
  718. // If the value has already been used, add it directly to the
  719. // DeferredDeclsToEmit list.
  720. StringRef MangledName = getMangledName(GD);
  721. if (GetGlobalValue(MangledName))
  722. DeferredDeclsToEmit.push_back(GD);
  723. else {
  724. // Otherwise, remember that we saw a deferred decl with this name. The
  725. // first use of the mangled name will cause it to move into
  726. // DeferredDeclsToEmit.
  727. DeferredDecls[MangledName] = GD;
  728. }
  729. }
  730. namespace {
  731. struct FunctionIsDirectlyRecursive :
  732. public RecursiveASTVisitor<FunctionIsDirectlyRecursive> {
  733. const StringRef Name;
  734. const Builtin::Context &BI;
  735. bool Result;
  736. FunctionIsDirectlyRecursive(StringRef N, const Builtin::Context &C) :
  737. Name(N), BI(C), Result(false) {
  738. }
  739. typedef RecursiveASTVisitor<FunctionIsDirectlyRecursive> Base;
  740. bool TraverseCallExpr(CallExpr *E) {
  741. const FunctionDecl *FD = E->getDirectCallee();
  742. if (!FD)
  743. return true;
  744. AsmLabelAttr *Attr = FD->getAttr<AsmLabelAttr>();
  745. if (Attr && Name == Attr->getLabel()) {
  746. Result = true;
  747. return false;
  748. }
  749. unsigned BuiltinID = FD->getBuiltinID();
  750. if (!BuiltinID)
  751. return true;
  752. StringRef BuiltinName = BI.GetName(BuiltinID);
  753. if (BuiltinName.startswith("__builtin_") &&
  754. Name == BuiltinName.slice(strlen("__builtin_"), StringRef::npos)) {
  755. Result = true;
  756. return false;
  757. }
  758. return true;
  759. }
  760. };
  761. }
  762. // isTriviallyRecursive - Check if this function calls another
  763. // decl that, because of the asm attribute or the other decl being a builtin,
  764. // ends up pointing to itself.
  765. bool
  766. CodeGenModule::isTriviallyRecursive(const FunctionDecl *FD) {
  767. StringRef Name;
  768. if (getCXXABI().getMangleContext().shouldMangleDeclName(FD)) {
  769. // asm labels are a special kind of mangling we have to support.
  770. AsmLabelAttr *Attr = FD->getAttr<AsmLabelAttr>();
  771. if (!Attr)
  772. return false;
  773. Name = Attr->getLabel();
  774. } else {
  775. Name = FD->getName();
  776. }
  777. FunctionIsDirectlyRecursive Walker(Name, Context.BuiltinInfo);
  778. Walker.TraverseFunctionDecl(const_cast<FunctionDecl*>(FD));
  779. return Walker.Result;
  780. }
  781. bool
  782. CodeGenModule::shouldEmitFunction(const FunctionDecl *F) {
  783. if (getFunctionLinkage(F) != llvm::Function::AvailableExternallyLinkage)
  784. return true;
  785. if (CodeGenOpts.OptimizationLevel == 0 &&
  786. !F->hasAttr<AlwaysInlineAttr>())
  787. return false;
  788. // PR9614. Avoid cases where the source code is lying to us. An available
  789. // externally function should have an equivalent function somewhere else,
  790. // but a function that calls itself is clearly not equivalent to the real
  791. // implementation.
  792. // This happens in glibc's btowc and in some configure checks.
  793. return !isTriviallyRecursive(F);
  794. }
  795. void CodeGenModule::EmitGlobalDefinition(GlobalDecl GD) {
  796. const ValueDecl *D = cast<ValueDecl>(GD.getDecl());
  797. PrettyStackTraceDecl CrashInfo(const_cast<ValueDecl *>(D), D->getLocation(),
  798. Context.getSourceManager(),
  799. "Generating code for declaration");
  800. if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
  801. // At -O0, don't generate IR for functions with available_externally
  802. // linkage.
  803. if (!shouldEmitFunction(Function))
  804. return;
  805. if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(D)) {
  806. // Make sure to emit the definition(s) before we emit the thunks.
  807. // This is necessary for the generation of certain thunks.
  808. if (const CXXConstructorDecl *CD = dyn_cast<CXXConstructorDecl>(Method))
  809. EmitCXXConstructor(CD, GD.getCtorType());
  810. else if (const CXXDestructorDecl *DD =dyn_cast<CXXDestructorDecl>(Method))
  811. EmitCXXDestructor(DD, GD.getDtorType());
  812. else
  813. EmitGlobalFunctionDefinition(GD);
  814. if (Method->isVirtual())
  815. getVTables().EmitThunks(GD);
  816. return;
  817. }
  818. return EmitGlobalFunctionDefinition(GD);
  819. }
  820. if (const VarDecl *VD = dyn_cast<VarDecl>(D))
  821. return EmitGlobalVarDefinition(VD);
  822. llvm_unreachable("Invalid argument to EmitGlobalDefinition()");
  823. }
  824. /// GetOrCreateLLVMFunction - If the specified mangled name is not in the
  825. /// module, create and return an llvm Function with the specified type. If there
  826. /// is something in the module with the specified name, return it potentially
  827. /// bitcasted to the right type.
  828. ///
  829. /// If D is non-null, it specifies a decl that correspond to this. This is used
  830. /// to set the attributes on the function when it is first created.
  831. llvm::Constant *
  832. CodeGenModule::GetOrCreateLLVMFunction(StringRef MangledName,
  833. llvm::Type *Ty,
  834. GlobalDecl D, bool ForVTable,
  835. llvm::Attributes ExtraAttrs) {
  836. // Lookup the entry, lazily creating it if necessary.
  837. llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
  838. if (Entry) {
  839. if (WeakRefReferences.count(Entry)) {
  840. const FunctionDecl *FD = cast_or_null<FunctionDecl>(D.getDecl());
  841. if (FD && !FD->hasAttr<WeakAttr>())
  842. Entry->setLinkage(llvm::Function::ExternalLinkage);
  843. WeakRefReferences.erase(Entry);
  844. }
  845. if (Entry->getType()->getElementType() == Ty)
  846. return Entry;
  847. // Make sure the result is of the correct type.
  848. return llvm::ConstantExpr::getBitCast(Entry, Ty->getPointerTo());
  849. }
  850. // This function doesn't have a complete type (for example, the return
  851. // type is an incomplete struct). Use a fake type instead, and make
  852. // sure not to try to set attributes.
  853. bool IsIncompleteFunction = false;
  854. llvm::FunctionType *FTy;
  855. if (isa<llvm::FunctionType>(Ty)) {
  856. FTy = cast<llvm::FunctionType>(Ty);
  857. } else {
  858. FTy = llvm::FunctionType::get(VoidTy, false);
  859. IsIncompleteFunction = true;
  860. }
  861. llvm::Function *F = llvm::Function::Create(FTy,
  862. llvm::Function::ExternalLinkage,
  863. MangledName, &getModule());
  864. assert(F->getName() == MangledName && "name was uniqued!");
  865. if (D.getDecl())
  866. SetFunctionAttributes(D, F, IsIncompleteFunction);
  867. if (ExtraAttrs != llvm::Attribute::None)
  868. F->addFnAttr(ExtraAttrs);
  869. // This is the first use or definition of a mangled name. If there is a
  870. // deferred decl with this name, remember that we need to emit it at the end
  871. // of the file.
  872. llvm::StringMap<GlobalDecl>::iterator DDI = DeferredDecls.find(MangledName);
  873. if (DDI != DeferredDecls.end()) {
  874. // Move the potentially referenced deferred decl to the DeferredDeclsToEmit
  875. // list, and remove it from DeferredDecls (since we don't need it anymore).
  876. DeferredDeclsToEmit.push_back(DDI->second);
  877. DeferredDecls.erase(DDI);
  878. // Otherwise, there are cases we have to worry about where we're
  879. // using a declaration for which we must emit a definition but where
  880. // we might not find a top-level definition:
  881. // - member functions defined inline in their classes
  882. // - friend functions defined inline in some class
  883. // - special member functions with implicit definitions
  884. // If we ever change our AST traversal to walk into class methods,
  885. // this will be unnecessary.
  886. //
  887. // We also don't emit a definition for a function if it's going to be an entry
  888. // in a vtable, unless it's already marked as used.
  889. } else if (getLangOptions().CPlusPlus && D.getDecl()) {
  890. // Look for a declaration that's lexically in a record.
  891. const FunctionDecl *FD = cast<FunctionDecl>(D.getDecl());
  892. do {
  893. if (isa<CXXRecordDecl>(FD->getLexicalDeclContext())) {
  894. if (FD->isImplicit() && !ForVTable) {
  895. assert(FD->isUsed() && "Sema didn't mark implicit function as used!");
  896. DeferredDeclsToEmit.push_back(D.getWithDecl(FD));
  897. break;
  898. } else if (FD->doesThisDeclarationHaveABody()) {
  899. DeferredDeclsToEmit.push_back(D.getWithDecl(FD));
  900. break;
  901. }
  902. }
  903. FD = FD->getPreviousDecl();
  904. } while (FD);
  905. }
  906. // Make sure the result is of the requested type.
  907. if (!IsIncompleteFunction) {
  908. assert(F->getType()->getElementType() == Ty);
  909. return F;
  910. }
  911. llvm::Type *PTy = llvm::PointerType::getUnqual(Ty);
  912. return llvm::ConstantExpr::getBitCast(F, PTy);
  913. }
  914. /// GetAddrOfFunction - Return the address of the given function. If Ty is
  915. /// non-null, then this function will use the specified type if it has to
  916. /// create it (this occurs when we see a definition of the function).
  917. llvm::Constant *CodeGenModule::GetAddrOfFunction(GlobalDecl GD,
  918. llvm::Type *Ty,
  919. bool ForVTable) {
  920. // If there was no specific requested type, just convert it now.
  921. if (!Ty)
  922. Ty = getTypes().ConvertType(cast<ValueDecl>(GD.getDecl())->getType());
  923. StringRef MangledName = getMangledName(GD);
  924. return GetOrCreateLLVMFunction(MangledName, Ty, GD, ForVTable);
  925. }
  926. /// CreateRuntimeFunction - Create a new runtime function with the specified
  927. /// type and name.
  928. llvm::Constant *
  929. CodeGenModule::CreateRuntimeFunction(llvm::FunctionType *FTy,
  930. StringRef Name,
  931. llvm::Attributes ExtraAttrs) {
  932. return GetOrCreateLLVMFunction(Name, FTy, GlobalDecl(), /*ForVTable=*/false,
  933. ExtraAttrs);
  934. }
  935. static bool DeclIsConstantGlobal(ASTContext &Context, const VarDecl *D,
  936. bool ConstantInit) {
  937. if (!D->getType().isConstant(Context) && !D->getType()->isReferenceType())
  938. return false;
  939. if (Context.getLangOptions().CPlusPlus) {
  940. if (const RecordType *Record
  941. = Context.getBaseElementType(D->getType())->getAs<RecordType>())
  942. return ConstantInit &&
  943. cast<CXXRecordDecl>(Record->getDecl())->isPOD() &&
  944. !cast<CXXRecordDecl>(Record->getDecl())->hasMutableFields();
  945. }
  946. return true;
  947. }
  948. /// GetOrCreateLLVMGlobal - If the specified mangled name is not in the module,
  949. /// create and return an llvm GlobalVariable with the specified type. If there
  950. /// is something in the module with the specified name, return it potentially
  951. /// bitcasted to the right type.
  952. ///
  953. /// If D is non-null, it specifies a decl that correspond to this. This is used
  954. /// to set the attributes on the global when it is first created.
  955. llvm::Constant *
  956. CodeGenModule::GetOrCreateLLVMGlobal(StringRef MangledName,
  957. llvm::PointerType *Ty,
  958. const VarDecl *D,
  959. bool UnnamedAddr) {
  960. // Lookup the entry, lazily creating it if necessary.
  961. llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
  962. if (Entry) {
  963. if (WeakRefReferences.count(Entry)) {
  964. if (D && !D->hasAttr<WeakAttr>())
  965. Entry->setLinkage(llvm::Function::ExternalLinkage);
  966. WeakRefReferences.erase(Entry);
  967. }
  968. if (UnnamedAddr)
  969. Entry->setUnnamedAddr(true);
  970. if (Entry->getType() == Ty)
  971. return Entry;
  972. // Make sure the result is of the correct type.
  973. return llvm::ConstantExpr::getBitCast(Entry, Ty);
  974. }
  975. // This is the first use or definition of a mangled name. If there is a
  976. // deferred decl with this name, remember that we need to emit it at the end
  977. // of the file.
  978. llvm::StringMap<GlobalDecl>::iterator DDI = DeferredDecls.find(MangledName);
  979. if (DDI != DeferredDecls.end()) {
  980. // Move the potentially referenced deferred decl to the DeferredDeclsToEmit
  981. // list, and remove it from DeferredDecls (since we don't need it anymore).
  982. DeferredDeclsToEmit.push_back(DDI->second);
  983. DeferredDecls.erase(DDI);
  984. }
  985. llvm::GlobalVariable *GV =
  986. new llvm::GlobalVariable(getModule(), Ty->getElementType(), false,
  987. llvm::GlobalValue::ExternalLinkage,
  988. 0, MangledName, 0,
  989. false, Ty->getAddressSpace());
  990. // Handle things which are present even on external declarations.
  991. if (D) {
  992. // FIXME: This code is overly simple and should be merged with other global
  993. // handling.
  994. GV->setConstant(DeclIsConstantGlobal(Context, D, false));
  995. // Set linkage and visibility in case we never see a definition.
  996. NamedDecl::LinkageInfo LV = D->getLinkageAndVisibility();
  997. if (LV.linkage() != ExternalLinkage) {
  998. // Don't set internal linkage on declarations.
  999. } else {
  1000. if (D->hasAttr<DLLImportAttr>())
  1001. GV->setLinkage(llvm::GlobalValue::DLLImportLinkage);
  1002. else if (D->hasAttr<WeakAttr>() || D->isWeakImported())
  1003. GV->setLinkage(llvm::GlobalValue::ExternalWeakLinkage);
  1004. // Set visibility on a declaration only if it's explicit.
  1005. if (LV.visibilityExplicit())
  1006. GV->setVisibility(GetLLVMVisibility(LV.visibility()));
  1007. }
  1008. GV->setThreadLocal(D->isThreadSpecified());
  1009. }
  1010. return GV;
  1011. }
  1012. llvm::GlobalVariable *
  1013. CodeGenModule::CreateOrReplaceCXXRuntimeVariable(StringRef Name,
  1014. llvm::Type *Ty,
  1015. llvm::GlobalValue::LinkageTypes Linkage) {
  1016. llvm::GlobalVariable *GV = getModule().getNamedGlobal(Name);
  1017. llvm::GlobalVariable *OldGV = 0;
  1018. if (GV) {
  1019. // Check if the variable has the right type.
  1020. if (GV->getType()->getElementType() == Ty)
  1021. return GV;
  1022. // Because C++ name mangling, the only way we can end up with an already
  1023. // existing global with the same name is if it has been declared extern "C".
  1024. assert(GV->isDeclaration() && "Declaration has wrong type!");
  1025. OldGV = GV;
  1026. }
  1027. // Create a new variable.
  1028. GV = new llvm::GlobalVariable(getModule(), Ty, /*isConstant=*/true,
  1029. Linkage, 0, Name);
  1030. if (OldGV) {
  1031. // Replace occurrences of the old variable if needed.
  1032. GV->takeName(OldGV);
  1033. if (!OldGV->use_empty()) {
  1034. llvm::Constant *NewPtrForOldDecl =
  1035. llvm::ConstantExpr::getBitCast(GV, OldGV->getType());
  1036. OldGV->replaceAllUsesWith(NewPtrForOldDecl);
  1037. }
  1038. OldGV->eraseFromParent();
  1039. }
  1040. return GV;
  1041. }
  1042. /// GetAddrOfGlobalVar - Return the llvm::Constant for the address of the
  1043. /// given global variable. If Ty is non-null and if the global doesn't exist,
  1044. /// then it will be greated with the specified type instead of whatever the
  1045. /// normal requested type would be.
  1046. llvm::Constant *CodeGenModule::GetAddrOfGlobalVar(const VarDecl *D,
  1047. llvm::Type *Ty) {
  1048. assert(D->hasGlobalStorage() && "Not a global variable");
  1049. QualType ASTTy = D->getType();
  1050. if (Ty == 0)
  1051. Ty = getTypes().ConvertTypeForMem(ASTTy);
  1052. llvm::PointerType *PTy =
  1053. llvm::PointerType::get(Ty, getContext().getTargetAddressSpace(ASTTy));
  1054. StringRef MangledName = getMangledName(D);
  1055. return GetOrCreateLLVMGlobal(MangledName, PTy, D);
  1056. }
  1057. /// CreateRuntimeVariable - Create a new runtime global variable with the
  1058. /// specified type and name.
  1059. llvm::Constant *
  1060. CodeGenModule::CreateRuntimeVariable(llvm::Type *Ty,
  1061. StringRef Name) {
  1062. return GetOrCreateLLVMGlobal(Name, llvm::PointerType::getUnqual(Ty), 0,
  1063. true);
  1064. }
  1065. void CodeGenModule::EmitTentativeDefinition(const VarDecl *D) {
  1066. assert(!D->getInit() && "Cannot emit definite definitions here!");
  1067. if (MayDeferGeneration(D)) {
  1068. // If we have not seen a reference to this variable yet, place it
  1069. // into the deferred declarations table to be emitted if needed
  1070. // later.
  1071. StringRef MangledName = getMangledName(D);
  1072. if (!GetGlobalValue(MangledName)) {
  1073. DeferredDecls[MangledName] = D;
  1074. return;
  1075. }
  1076. }
  1077. // The tentative definition is the only definition.
  1078. EmitGlobalVarDefinition(D);
  1079. }
  1080. void CodeGenModule::EmitVTable(CXXRecordDecl *Class, bool DefinitionRequired) {
  1081. if (DefinitionRequired)
  1082. getVTables().GenerateClassData(getVTableLinkage(Class), Class);
  1083. }
  1084. llvm::GlobalVariable::LinkageTypes
  1085. CodeGenModule::getVTableLinkage(const CXXRecordDecl *RD) {
  1086. if (RD->getLinkage() != ExternalLinkage)
  1087. return llvm::GlobalVariable::InternalLinkage;
  1088. if (const CXXMethodDecl *KeyFunction
  1089. = RD->getASTContext().getKeyFunction(RD)) {
  1090. // If this class has a key function, use that to determine the linkage of
  1091. // the vtable.
  1092. const FunctionDecl *Def = 0;
  1093. if (KeyFunction->hasBody(Def))
  1094. KeyFunction = cast<CXXMethodDecl>(Def);
  1095. switch (KeyFunction->getTemplateSpecializationKind()) {
  1096. case TSK_Undeclared:
  1097. case TSK_ExplicitSpecialization:
  1098. // When compiling with optimizations turned on, we emit all vtables,
  1099. // even if the key function is not defined in the current translation
  1100. // unit. If this is the case, use available_externally linkage.
  1101. if (!Def && CodeGenOpts.OptimizationLevel)
  1102. return llvm::GlobalVariable::AvailableExternallyLinkage;
  1103. if (KeyFunction->isInlined())
  1104. return !Context.getLangOptions().AppleKext ?
  1105. llvm::GlobalVariable::LinkOnceODRLinkage :
  1106. llvm::Function::InternalLinkage;
  1107. return llvm::GlobalVariable::ExternalLinkage;
  1108. case TSK_ImplicitInstantiation:
  1109. return !Context.getLangOptions().AppleKext ?
  1110. llvm::GlobalVariable::LinkOnceODRLinkage :
  1111. llvm::Function::InternalLinkage;
  1112. case TSK_ExplicitInstantiationDefinition:
  1113. return !Context.getLangOptions().AppleKext ?
  1114. llvm::GlobalVariable::WeakODRLinkage :
  1115. llvm::Function::InternalLinkage;
  1116. case TSK_ExplicitInstantiationDeclaration:
  1117. // FIXME: Use available_externally linkage. However, this currently
  1118. // breaks LLVM's build due to undefined symbols.
  1119. // return llvm::GlobalVariable::AvailableExternallyLinkage;
  1120. return !Context.getLangOptions().AppleKext ?
  1121. llvm::GlobalVariable::LinkOnceODRLinkage :
  1122. llvm::Function::InternalLinkage;
  1123. }
  1124. }
  1125. if (Context.getLangOptions().AppleKext)
  1126. return llvm::Function::InternalLinkage;
  1127. switch (RD->getTemplateSpecializationKind()) {
  1128. case TSK_Undeclared:
  1129. case TSK_ExplicitSpecialization:
  1130. case TSK_ImplicitInstantiation:
  1131. // FIXME: Use available_externally linkage. However, this currently
  1132. // breaks LLVM's build due to undefined symbols.
  1133. // return llvm::GlobalVariable::AvailableExternallyLinkage;
  1134. case TSK_ExplicitInstantiationDeclaration:
  1135. return llvm::GlobalVariable::LinkOnceODRLinkage;
  1136. case TSK_ExplicitInstantiationDefinition:
  1137. return llvm::GlobalVariable::WeakODRLinkage;
  1138. }
  1139. llvm_unreachable("Invalid TemplateSpecializationKind!");
  1140. }
  1141. CharUnits CodeGenModule::GetTargetTypeStoreSize(llvm::Type *Ty) const {
  1142. return Context.toCharUnitsFromBits(
  1143. TheTargetData.getTypeStoreSizeInBits(Ty));
  1144. }
  1145. void CodeGenModule::EmitGlobalVarDefinition(const VarDecl *D) {
  1146. llvm::Constant *Init = 0;
  1147. QualType ASTTy = D->getType();
  1148. CXXRecordDecl *RD = ASTTy->getBaseElementTypeUnsafe()->getAsCXXRecordDecl();
  1149. bool NeedsGlobalCtor = false;
  1150. bool NeedsGlobalDtor = RD && !RD->hasTrivialDestructor();
  1151. const VarDecl *InitDecl;
  1152. const Expr *InitExpr = D->getAnyInitializer(InitDecl);
  1153. if (!InitExpr) {
  1154. // This is a tentative definition; tentative definitions are
  1155. // implicitly initialized with { 0 }.
  1156. //
  1157. // Note that tentative definitions are only emitted at the end of
  1158. // a translation unit, so they should never have incomplete
  1159. // type. In addition, EmitTentativeDefinition makes sure that we
  1160. // never attempt to emit a tentative definition if a real one
  1161. // exists. A use may still exists, however, so we still may need
  1162. // to do a RAUW.
  1163. assert(!ASTTy->isIncompleteType() && "Unexpected incomplete type");
  1164. Init = EmitNullConstant(D->getType());
  1165. } else {
  1166. Init = EmitConstantInit(*InitDecl);
  1167. if (!Init) {
  1168. QualType T = InitExpr->getType();
  1169. if (D->getType()->isReferenceType())
  1170. T = D->getType();
  1171. if (getLangOptions().CPlusPlus) {
  1172. Init = EmitNullConstant(T);
  1173. NeedsGlobalCtor = true;
  1174. } else {
  1175. ErrorUnsupported(D, "static initializer");
  1176. Init = llvm::UndefValue::get(getTypes().ConvertType(T));
  1177. }
  1178. } else {
  1179. // We don't need an initializer, so remove the entry for the delayed
  1180. // initializer position (just in case this entry was delayed) if we
  1181. // also don't need to register a destructor.
  1182. if (getLangOptions().CPlusPlus && !NeedsGlobalDtor)
  1183. DelayedCXXInitPosition.erase(D);
  1184. }
  1185. }
  1186. llvm::Type* InitType = Init->getType();
  1187. llvm::Constant *Entry = GetAddrOfGlobalVar(D, InitType);
  1188. // Strip off a bitcast if we got one back.
  1189. if (llvm::ConstantExpr *CE = dyn_cast<llvm::ConstantExpr>(Entry)) {
  1190. assert(CE->getOpcode() == llvm::Instruction::BitCast ||
  1191. // all zero index gep.
  1192. CE->getOpcode() == llvm::Instruction::GetElementPtr);
  1193. Entry = CE->getOperand(0);
  1194. }
  1195. // Entry is now either a Function or GlobalVariable.
  1196. llvm::GlobalVariable *GV = dyn_cast<llvm::GlobalVariable>(Entry);
  1197. // We have a definition after a declaration with the wrong type.
  1198. // We must make a new GlobalVariable* and update everything that used OldGV
  1199. // (a declaration or tentative definition) with the new GlobalVariable*
  1200. // (which will be a definition).
  1201. //
  1202. // This happens if there is a prototype for a global (e.g.
  1203. // "extern int x[];") and then a definition of a different type (e.g.
  1204. // "int x[10];"). This also happens when an initializer has a different type
  1205. // from the type of the global (this happens with unions).
  1206. if (GV == 0 ||
  1207. GV->getType()->getElementType() != InitType ||
  1208. GV->getType()->getAddressSpace() !=
  1209. getContext().getTargetAddressSpace(ASTTy)) {
  1210. // Move the old entry aside so that we'll create a new one.
  1211. Entry->setName(StringRef());
  1212. // Make a new global with the correct type, this is now guaranteed to work.
  1213. GV = cast<llvm::GlobalVariable>(GetAddrOfGlobalVar(D, InitType));
  1214. // Replace all uses of the old global with the new global
  1215. llvm::Constant *NewPtrForOldDecl =
  1216. llvm::ConstantExpr::getBitCast(GV, Entry->getType());
  1217. Entry->replaceAllUsesWith(NewPtrForOldDecl);
  1218. // Erase the old global, since it is no longer used.
  1219. cast<llvm::GlobalValue>(Entry)->eraseFromParent();
  1220. }
  1221. if (D->hasAttr<AnnotateAttr>())
  1222. AddGlobalAnnotations(D, GV);
  1223. GV->setInitializer(Init);
  1224. // If it is safe to mark the global 'constant', do so now.
  1225. GV->setConstant(false);
  1226. if (!NeedsGlobalCtor && !NeedsGlobalDtor &&
  1227. DeclIsConstantGlobal(Context, D, true))
  1228. GV->setConstant(true);
  1229. GV->setAlignment(getContext().getDeclAlign(D).getQuantity());
  1230. // Set the llvm linkage type as appropriate.
  1231. llvm::GlobalValue::LinkageTypes Linkage =
  1232. GetLLVMLinkageVarDefinition(D, GV);
  1233. GV->setLinkage(Linkage);
  1234. if (Linkage == llvm::GlobalVariable::CommonLinkage)
  1235. // common vars aren't constant even if declared const.
  1236. GV->setConstant(false);
  1237. SetCommonAttributes(D, GV);
  1238. // Emit the initializer function if necessary.
  1239. if (NeedsGlobalCtor || NeedsGlobalDtor)
  1240. EmitCXXGlobalVarDeclInitFunc(D, GV, NeedsGlobalCtor);
  1241. // Emit global variable debug information.
  1242. if (CGDebugInfo *DI = getModuleDebugInfo())
  1243. DI->EmitGlobalVariable(GV, D);
  1244. }
  1245. llvm::GlobalValue::LinkageTypes
  1246. CodeGenModule::GetLLVMLinkageVarDefinition(const VarDecl *D,
  1247. llvm::GlobalVariable *GV) {
  1248. GVALinkage Linkage = getContext().GetGVALinkageForVariable(D);
  1249. if (Linkage == GVA_Internal)
  1250. return llvm::Function::InternalLinkage;
  1251. else if (D->hasAttr<DLLImportAttr>())
  1252. return llvm::Function::DLLImportLinkage;
  1253. else if (D->hasAttr<DLLExportAttr>())
  1254. return llvm::Function::DLLExportLinkage;
  1255. else if (D->hasAttr<WeakAttr>()) {
  1256. if (GV->isConstant())
  1257. return llvm::GlobalVariable::WeakODRLinkage;
  1258. else
  1259. return llvm::GlobalVariable::WeakAnyLinkage;
  1260. } else if (Linkage == GVA_TemplateInstantiation ||
  1261. Linkage == GVA_ExplicitTemplateInstantiation)
  1262. return llvm::GlobalVariable::WeakODRLinkage;
  1263. else if (!getLangOptions().CPlusPlus &&
  1264. ((!CodeGenOpts.NoCommon && !D->getAttr<NoCommonAttr>()) ||
  1265. D->getAttr<CommonAttr>()) &&
  1266. !D->hasExternalStorage() && !D->getInit() &&
  1267. !D->getAttr<SectionAttr>() && !D->isThreadSpecified() &&
  1268. !D->getAttr<WeakImportAttr>()) {
  1269. // Thread local vars aren't considered common linkage.
  1270. return llvm::GlobalVariable::CommonLinkage;
  1271. }
  1272. return llvm::GlobalVariable::ExternalLinkage;
  1273. }
  1274. /// ReplaceUsesOfNonProtoTypeWithRealFunction - This function is called when we
  1275. /// implement a function with no prototype, e.g. "int foo() {}". If there are
  1276. /// existing call uses of the old function in the module, this adjusts them to
  1277. /// call the new function directly.
  1278. ///
  1279. /// This is not just a cleanup: the always_inline pass requires direct calls to
  1280. /// functions to be able to inline them. If there is a bitcast in the way, it
  1281. /// won't inline them. Instcombine normally deletes these calls, but it isn't
  1282. /// run at -O0.
  1283. static void ReplaceUsesOfNonProtoTypeWithRealFunction(llvm::GlobalValue *Old,
  1284. llvm::Function *NewFn) {
  1285. // If we're redefining a global as a function, don't transform it.
  1286. llvm::Function *OldFn = dyn_cast<llvm::Function>(Old);
  1287. if (OldFn == 0) return;
  1288. llvm::Type *NewRetTy = NewFn->getReturnType();
  1289. SmallVector<llvm::Value*, 4> ArgList;
  1290. for (llvm::Value::use_iterator UI = OldFn->use_begin(), E = OldFn->use_end();
  1291. UI != E; ) {
  1292. // TODO: Do invokes ever occur in C code? If so, we should handle them too.
  1293. llvm::Value::use_iterator I = UI++; // Increment before the CI is erased.
  1294. llvm::CallInst *CI = dyn_cast<llvm::CallInst>(*I);
  1295. if (!CI) continue; // FIXME: when we allow Invoke, just do CallSite CS(*I)
  1296. llvm::CallSite CS(CI);
  1297. if (!CI || !CS.isCallee(I)) continue;
  1298. // If the return types don't match exactly, and if the call isn't dead, then
  1299. // we can't transform this call.
  1300. if (CI->getType() != NewRetTy && !CI->use_empty())
  1301. continue;
  1302. // Get the attribute list.
  1303. llvm::SmallVector<llvm::AttributeWithIndex, 8> AttrVec;
  1304. llvm::AttrListPtr AttrList = CI->getAttributes();
  1305. // Get any return attributes.
  1306. llvm::Attributes RAttrs = AttrList.getRetAttributes();
  1307. // Add the return attributes.
  1308. if (RAttrs)
  1309. AttrVec.push_back(llvm::AttributeWithIndex::get(0, RAttrs));
  1310. // If the function was passed too few arguments, don't transform. If extra
  1311. // arguments were passed, we silently drop them. If any of the types
  1312. // mismatch, we don't transform.
  1313. unsigned ArgNo = 0;
  1314. bool DontTransform = false;
  1315. for (llvm::Function::arg_iterator AI = NewFn->arg_begin(),
  1316. E = NewFn->arg_end(); AI != E; ++AI, ++ArgNo) {
  1317. if (CS.arg_size() == ArgNo ||
  1318. CS.getArgument(ArgNo)->getType() != AI->getType()) {
  1319. DontTransform = true;
  1320. break;
  1321. }
  1322. // Add any parameter attributes.
  1323. if (llvm::Attributes PAttrs = AttrList.getParamAttributes(ArgNo + 1))
  1324. AttrVec.push_back(llvm::AttributeWithIndex::get(ArgNo + 1, PAttrs));
  1325. }
  1326. if (DontTransform)
  1327. continue;
  1328. if (llvm::Attributes FnAttrs = AttrList.getFnAttributes())
  1329. AttrVec.push_back(llvm::AttributeWithIndex::get(~0, FnAttrs));
  1330. // Okay, we can transform this. Create the new call instruction and copy
  1331. // over the required information.
  1332. ArgList.append(CS.arg_begin(), CS.arg_begin() + ArgNo);
  1333. llvm::CallInst *NewCall = llvm::CallInst::Create(NewFn, ArgList, "", CI);
  1334. ArgList.clear();
  1335. if (!NewCall->getType()->isVoidTy())
  1336. NewCall->takeName(CI);
  1337. NewCall->setAttributes(llvm::AttrListPtr::get(AttrVec.begin(),
  1338. AttrVec.end()));
  1339. NewCall->setCallingConv(CI->getCallingConv());
  1340. // Finally, remove the old call, replacing any uses with the new one.
  1341. if (!CI->use_empty())
  1342. CI->replaceAllUsesWith(NewCall);
  1343. // Copy debug location attached to CI.
  1344. if (!CI->getDebugLoc().isUnknown())
  1345. NewCall->setDebugLoc(CI->getDebugLoc());
  1346. CI->eraseFromParent();
  1347. }
  1348. }
  1349. void CodeGenModule::EmitGlobalFunctionDefinition(GlobalDecl GD) {
  1350. const FunctionDecl *D = cast<FunctionDecl>(GD.getDecl());
  1351. // Compute the function info and LLVM type.
  1352. const CGFunctionInfo &FI = getTypes().getFunctionInfo(GD);
  1353. bool variadic = false;
  1354. if (const FunctionProtoType *fpt = D->getType()->getAs<FunctionProtoType>())
  1355. variadic = fpt->isVariadic();
  1356. llvm::FunctionType *Ty = getTypes().GetFunctionType(FI, variadic);
  1357. // Get or create the prototype for the function.
  1358. llvm::Constant *Entry = GetAddrOfFunction(GD, Ty);
  1359. // Strip off a bitcast if we got one back.
  1360. if (llvm::ConstantExpr *CE = dyn_cast<llvm::ConstantExpr>(Entry)) {
  1361. assert(CE->getOpcode() == llvm::Instruction::BitCast);
  1362. Entry = CE->getOperand(0);
  1363. }
  1364. if (cast<llvm::GlobalValue>(Entry)->getType()->getElementType() != Ty) {
  1365. llvm::GlobalValue *OldFn = cast<llvm::GlobalValue>(Entry);
  1366. // If the types mismatch then we have to rewrite the definition.
  1367. assert(OldFn->isDeclaration() &&
  1368. "Shouldn't replace non-declaration");
  1369. // F is the Function* for the one with the wrong type, we must make a new
  1370. // Function* and update everything that used F (a declaration) with the new
  1371. // Function* (which will be a definition).
  1372. //
  1373. // This happens if there is a prototype for a function
  1374. // (e.g. "int f()") and then a definition of a different type
  1375. // (e.g. "int f(int x)"). Move the old function aside so that it
  1376. // doesn't interfere with GetAddrOfFunction.
  1377. OldFn->setName(StringRef());
  1378. llvm::Function *NewFn = cast<llvm::Function>(GetAddrOfFunction(GD, Ty));
  1379. // If this is an implementation of a function without a prototype, try to
  1380. // replace any existing uses of the function (which may be calls) with uses
  1381. // of the new function
  1382. if (D->getType()->isFunctionNoProtoType()) {
  1383. ReplaceUsesOfNonProtoTypeWithRealFunction(OldFn, NewFn);
  1384. OldFn->removeDeadConstantUsers();
  1385. }
  1386. // Replace uses of F with the Function we will endow with a body.
  1387. if (!Entry->use_empty()) {
  1388. llvm::Constant *NewPtrForOldDecl =
  1389. llvm::ConstantExpr::getBitCast(NewFn, Entry->getType());
  1390. Entry->replaceAllUsesWith(NewPtrForOldDecl);
  1391. }
  1392. // Ok, delete the old function now, which is dead.
  1393. OldFn->eraseFromParent();
  1394. Entry = NewFn;
  1395. }
  1396. // We need to set linkage and visibility on the function before
  1397. // generating code for it because various parts of IR generation
  1398. // want to propagate this information down (e.g. to local static
  1399. // declarations).
  1400. llvm::Function *Fn = cast<llvm::Function>(Entry);
  1401. setFunctionLinkage(D, Fn);
  1402. // FIXME: this is redundant with part of SetFunctionDefinitionAttributes
  1403. setGlobalVisibility(Fn, D);
  1404. CodeGenFunction(*this).GenerateCode(D, Fn, FI);
  1405. SetFunctionDefinitionAttributes(D, Fn);
  1406. SetLLVMFunctionAttributesForDefinition(D, Fn);
  1407. if (const ConstructorAttr *CA = D->getAttr<ConstructorAttr>())
  1408. AddGlobalCtor(Fn, CA->getPriority());
  1409. if (const DestructorAttr *DA = D->getAttr<DestructorAttr>())
  1410. AddGlobalDtor(Fn, DA->getPriority());
  1411. if (D->hasAttr<AnnotateAttr>())
  1412. AddGlobalAnnotations(D, Fn);
  1413. }
  1414. void CodeGenModule::EmitAliasDefinition(GlobalDecl GD) {
  1415. const ValueDecl *D = cast<ValueDecl>(GD.getDecl());
  1416. const AliasAttr *AA = D->getAttr<AliasAttr>();
  1417. assert(AA && "Not an alias?");
  1418. StringRef MangledName = getMangledName(GD);
  1419. // If there is a definition in the module, then it wins over the alias.
  1420. // This is dubious, but allow it to be safe. Just ignore the alias.
  1421. llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
  1422. if (Entry && !Entry->isDeclaration())
  1423. return;
  1424. llvm::Type *DeclTy = getTypes().ConvertTypeForMem(D->getType());
  1425. // Create a reference to the named value. This ensures that it is emitted
  1426. // if a deferred decl.
  1427. llvm::Constant *Aliasee;
  1428. if (isa<llvm::FunctionType>(DeclTy))
  1429. Aliasee = GetOrCreateLLVMFunction(AA->getAliasee(), DeclTy, GlobalDecl(),
  1430. /*ForVTable=*/false);
  1431. else
  1432. Aliasee = GetOrCreateLLVMGlobal(AA->getAliasee(),
  1433. llvm::PointerType::getUnqual(DeclTy), 0);
  1434. // Create the new alias itself, but don't set a name yet.
  1435. llvm::GlobalValue *GA =
  1436. new llvm::GlobalAlias(Aliasee->getType(),
  1437. llvm::Function::ExternalLinkage,
  1438. "", Aliasee, &getModule());
  1439. if (Entry) {
  1440. assert(Entry->isDeclaration());
  1441. // If there is a declaration in the module, then we had an extern followed
  1442. // by the alias, as in:
  1443. // extern int test6();
  1444. // ...
  1445. // int test6() __attribute__((alias("test7")));
  1446. //
  1447. // Remove it and replace uses of it with the alias.
  1448. GA->takeName(Entry);
  1449. Entry->replaceAllUsesWith(llvm::ConstantExpr::getBitCast(GA,
  1450. Entry->getType()));
  1451. Entry->eraseFromParent();
  1452. } else {
  1453. GA->setName(MangledName);
  1454. }
  1455. // Set attributes which are particular to an alias; this is a
  1456. // specialization of the attributes which may be set on a global
  1457. // variable/function.
  1458. if (D->hasAttr<DLLExportAttr>()) {
  1459. if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
  1460. // The dllexport attribute is ignored for undefined symbols.
  1461. if (FD->hasBody())
  1462. GA->setLinkage(llvm::Function::DLLExportLinkage);
  1463. } else {
  1464. GA->setLinkage(llvm::Function::DLLExportLinkage);
  1465. }
  1466. } else if (D->hasAttr<WeakAttr>() ||
  1467. D->hasAttr<WeakRefAttr>() ||
  1468. D->isWeakImported()) {
  1469. GA->setLinkage(llvm::Function::WeakAnyLinkage);
  1470. }
  1471. SetCommonAttributes(D, GA);
  1472. }
  1473. llvm::Function *CodeGenModule::getIntrinsic(unsigned IID,
  1474. ArrayRef<llvm::Type*> Tys) {
  1475. return llvm::Intrinsic::getDeclaration(&getModule(), (llvm::Intrinsic::ID)IID,
  1476. Tys);
  1477. }
  1478. static llvm::StringMapEntry<llvm::Constant*> &
  1479. GetConstantCFStringEntry(llvm::StringMap<llvm::Constant*> &Map,
  1480. const StringLiteral *Literal,
  1481. bool TargetIsLSB,
  1482. bool &IsUTF16,
  1483. unsigned &StringLength) {
  1484. StringRef String = Literal->getString();
  1485. unsigned NumBytes = String.size();
  1486. // Check for simple case.
  1487. if (!Literal->containsNonAsciiOrNull()) {
  1488. StringLength = NumBytes;
  1489. return Map.GetOrCreateValue(String);
  1490. }
  1491. // Otherwise, convert the UTF8 literals into a byte string.
  1492. SmallVector<UTF16, 128> ToBuf(NumBytes);
  1493. const UTF8 *FromPtr = (UTF8 *)String.data();
  1494. UTF16 *ToPtr = &ToBuf[0];
  1495. (void)ConvertUTF8toUTF16(&FromPtr, FromPtr + NumBytes,
  1496. &ToPtr, ToPtr + NumBytes,
  1497. strictConversion);
  1498. // ConvertUTF8toUTF16 returns the length in ToPtr.
  1499. StringLength = ToPtr - &ToBuf[0];
  1500. // Render the UTF-16 string into a byte array and convert to the target byte
  1501. // order.
  1502. //
  1503. // FIXME: This isn't something we should need to do here.
  1504. SmallString<128> AsBytes;
  1505. AsBytes.reserve(StringLength * 2);
  1506. for (unsigned i = 0; i != StringLength; ++i) {
  1507. unsigned short Val = ToBuf[i];
  1508. if (TargetIsLSB) {
  1509. AsBytes.push_back(Val & 0xFF);
  1510. AsBytes.push_back(Val >> 8);
  1511. } else {
  1512. AsBytes.push_back(Val >> 8);
  1513. AsBytes.push_back(Val & 0xFF);
  1514. }
  1515. }
  1516. // Append one extra null character, the second is automatically added by our
  1517. // caller.
  1518. AsBytes.push_back(0);
  1519. IsUTF16 = true;
  1520. return Map.GetOrCreateValue(StringRef(AsBytes.data(), AsBytes.size()));
  1521. }
  1522. static llvm::StringMapEntry<llvm::Constant*> &
  1523. GetConstantStringEntry(llvm::StringMap<llvm::Constant*> &Map,
  1524. const StringLiteral *Literal,
  1525. unsigned &StringLength)
  1526. {
  1527. StringRef String = Literal->getString();
  1528. StringLength = String.size();
  1529. return Map.GetOrCreateValue(String);
  1530. }
  1531. llvm::Constant *
  1532. CodeGenModule::GetAddrOfConstantCFString(const StringLiteral *Literal) {
  1533. unsigned StringLength = 0;
  1534. bool isUTF16 = false;
  1535. llvm::StringMapEntry<llvm::Constant*> &Entry =
  1536. GetConstantCFStringEntry(CFConstantStringMap, Literal,
  1537. getTargetData().isLittleEndian(),
  1538. isUTF16, StringLength);
  1539. if (llvm::Constant *C = Entry.getValue())
  1540. return C;
  1541. llvm::Constant *Zero = llvm::Constant::getNullValue(Int32Ty);
  1542. llvm::Constant *Zeros[] = { Zero, Zero };
  1543. // If we don't already have it, get __CFConstantStringClassReference.
  1544. if (!CFConstantStringClassRef) {
  1545. llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
  1546. Ty = llvm::ArrayType::get(Ty, 0);
  1547. llvm::Constant *GV = CreateRuntimeVariable(Ty,
  1548. "__CFConstantStringClassReference");
  1549. // Decay array -> ptr
  1550. CFConstantStringClassRef =
  1551. llvm::ConstantExpr::getGetElementPtr(GV, Zeros);
  1552. }
  1553. QualType CFTy = getContext().getCFConstantStringType();
  1554. llvm::StructType *STy =
  1555. cast<llvm::StructType>(getTypes().ConvertType(CFTy));
  1556. llvm::Constant *Fields[4];
  1557. // Class pointer.
  1558. Fields[0] = CFConstantStringClassRef;
  1559. // Flags.
  1560. llvm::Type *Ty = getTypes().ConvertType(getContext().UnsignedIntTy);
  1561. Fields[1] = isUTF16 ? llvm::ConstantInt::get(Ty, 0x07d0) :
  1562. llvm::ConstantInt::get(Ty, 0x07C8);
  1563. // String pointer.
  1564. llvm::Constant *C = llvm::ConstantDataArray::getString(VMContext,
  1565. Entry.getKey());
  1566. llvm::GlobalValue::LinkageTypes Linkage;
  1567. if (isUTF16)
  1568. // FIXME: why do utf strings get "_" labels instead of "L" labels?
  1569. Linkage = llvm::GlobalValue::InternalLinkage;
  1570. else
  1571. // FIXME: With OS X ld 123.2 (xcode 4) and LTO we would get a linker error
  1572. // when using private linkage. It is not clear if this is a bug in ld
  1573. // or a reasonable new restriction.
  1574. Linkage = llvm::GlobalValue::LinkerPrivateLinkage;
  1575. // Note: -fwritable-strings doesn't make the backing store strings of
  1576. // CFStrings writable. (See <rdar://problem/10657500>)
  1577. llvm::GlobalVariable *GV =
  1578. new llvm::GlobalVariable(getModule(), C->getType(), /*isConstant=*/true,
  1579. Linkage, C, ".str");
  1580. GV->setUnnamedAddr(true);
  1581. if (isUTF16) {
  1582. CharUnits Align = getContext().getTypeAlignInChars(getContext().ShortTy);
  1583. GV->setAlignment(Align.getQuantity());
  1584. } else {
  1585. CharUnits Align = getContext().getTypeAlignInChars(getContext().CharTy);
  1586. GV->setAlignment(Align.getQuantity());
  1587. }
  1588. Fields[2] = llvm::ConstantExpr::getGetElementPtr(GV, Zeros);
  1589. // String length.
  1590. Ty = getTypes().ConvertType(getContext().LongTy);
  1591. Fields[3] = llvm::ConstantInt::get(Ty, StringLength);
  1592. // The struct.
  1593. C = llvm::ConstantStruct::get(STy, Fields);
  1594. GV = new llvm::GlobalVariable(getModule(), C->getType(), true,
  1595. llvm::GlobalVariable::PrivateLinkage, C,
  1596. "_unnamed_cfstring_");
  1597. if (const char *Sect = getContext().getTargetInfo().getCFStringSection())
  1598. GV->setSection(Sect);
  1599. Entry.setValue(GV);
  1600. return GV;
  1601. }
  1602. static RecordDecl *
  1603. CreateRecordDecl(const ASTContext &Ctx, RecordDecl::TagKind TK,
  1604. DeclContext *DC, IdentifierInfo *Id) {
  1605. SourceLocation Loc;
  1606. if (Ctx.getLangOptions().CPlusPlus)
  1607. return CXXRecordDecl::Create(Ctx, TK, DC, Loc, Loc, Id);
  1608. else
  1609. return RecordDecl::Create(Ctx, TK, DC, Loc, Loc, Id);
  1610. }
  1611. llvm::Constant *
  1612. CodeGenModule::GetAddrOfConstantString(const StringLiteral *Literal) {
  1613. unsigned StringLength = 0;
  1614. llvm::StringMapEntry<llvm::Constant*> &Entry =
  1615. GetConstantStringEntry(CFConstantStringMap, Literal, StringLength);
  1616. if (llvm::Constant *C = Entry.getValue())
  1617. return C;
  1618. llvm::Constant *Zero = llvm::Constant::getNullValue(Int32Ty);
  1619. llvm::Constant *Zeros[] = { Zero, Zero };
  1620. // If we don't already have it, get _NSConstantStringClassReference.
  1621. if (!ConstantStringClassRef) {
  1622. std::string StringClass(getLangOptions().ObjCConstantStringClass);
  1623. llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
  1624. llvm::Constant *GV;
  1625. if (Features.ObjCNonFragileABI) {
  1626. std::string str =
  1627. StringClass.empty() ? "OBJC_CLASS_$_NSConstantString"
  1628. : "OBJC_CLASS_$_" + StringClass;
  1629. GV = getObjCRuntime().GetClassGlobal(str);
  1630. // Make sure the result is of the correct type.
  1631. llvm::Type *PTy = llvm::PointerType::getUnqual(Ty);
  1632. ConstantStringClassRef =
  1633. llvm::ConstantExpr::getBitCast(GV, PTy);
  1634. } else {
  1635. std::string str =
  1636. StringClass.empty() ? "_NSConstantStringClassReference"
  1637. : "_" + StringClass + "ClassReference";
  1638. llvm::Type *PTy = llvm::ArrayType::get(Ty, 0);
  1639. GV = CreateRuntimeVariable(PTy, str);
  1640. // Decay array -> ptr
  1641. ConstantStringClassRef =
  1642. llvm::ConstantExpr::getGetElementPtr(GV, Zeros);
  1643. }
  1644. }
  1645. if (!NSConstantStringType) {
  1646. // Construct the type for a constant NSString.
  1647. RecordDecl *D = CreateRecordDecl(Context, TTK_Struct,
  1648. Context.getTranslationUnitDecl(),
  1649. &Context.Idents.get("__builtin_NSString"));
  1650. D->startDefinition();
  1651. QualType FieldTypes[3];
  1652. // const int *isa;
  1653. FieldTypes[0] = Context.getPointerType(Context.IntTy.withConst());
  1654. // const char *str;
  1655. FieldTypes[1] = Context.getPointerType(Context.CharTy.withConst());
  1656. // unsigned int length;
  1657. FieldTypes[2] = Context.UnsignedIntTy;
  1658. // Create fields
  1659. for (unsigned i = 0; i < 3; ++i) {
  1660. FieldDecl *Field = FieldDecl::Create(Context, D,
  1661. SourceLocation(),
  1662. SourceLocation(), 0,
  1663. FieldTypes[i], /*TInfo=*/0,
  1664. /*BitWidth=*/0,
  1665. /*Mutable=*/false,
  1666. /*HasInit=*/false);
  1667. Field->setAccess(AS_public);
  1668. D->addDecl(Field);
  1669. }
  1670. D->completeDefinition();
  1671. QualType NSTy = Context.getTagDeclType(D);
  1672. NSConstantStringType = cast<llvm::StructType>(getTypes().ConvertType(NSTy));
  1673. }
  1674. llvm::Constant *Fields[3];
  1675. // Class pointer.
  1676. Fields[0] = ConstantStringClassRef;
  1677. // String pointer.
  1678. llvm::Constant *C =
  1679. llvm::ConstantDataArray::getString(VMContext, Entry.getKey());
  1680. llvm::GlobalValue::LinkageTypes Linkage;
  1681. bool isConstant;
  1682. Linkage = llvm::GlobalValue::PrivateLinkage;
  1683. isConstant = !Features.WritableStrings;
  1684. llvm::GlobalVariable *GV =
  1685. new llvm::GlobalVariable(getModule(), C->getType(), isConstant, Linkage, C,
  1686. ".str");
  1687. GV->setUnnamedAddr(true);
  1688. CharUnits Align = getContext().getTypeAlignInChars(getContext().CharTy);
  1689. GV->setAlignment(Align.getQuantity());
  1690. Fields[1] = llvm::ConstantExpr::getGetElementPtr(GV, Zeros);
  1691. // String length.
  1692. llvm::Type *Ty = getTypes().ConvertType(getContext().UnsignedIntTy);
  1693. Fields[2] = llvm::ConstantInt::get(Ty, StringLength);
  1694. // The struct.
  1695. C = llvm::ConstantStruct::get(NSConstantStringType, Fields);
  1696. GV = new llvm::GlobalVariable(getModule(), C->getType(), true,
  1697. llvm::GlobalVariable::PrivateLinkage, C,
  1698. "_unnamed_nsstring_");
  1699. // FIXME. Fix section.
  1700. if (const char *Sect =
  1701. Features.ObjCNonFragileABI
  1702. ? getContext().getTargetInfo().getNSStringNonFragileABISection()
  1703. : getContext().getTargetInfo().getNSStringSection())
  1704. GV->setSection(Sect);
  1705. Entry.setValue(GV);
  1706. return GV;
  1707. }
  1708. QualType CodeGenModule::getObjCFastEnumerationStateType() {
  1709. if (ObjCFastEnumerationStateType.isNull()) {
  1710. RecordDecl *D = CreateRecordDecl(Context, TTK_Struct,
  1711. Context.getTranslationUnitDecl(),
  1712. &Context.Idents.get("__objcFastEnumerationState"));
  1713. D->startDefinition();
  1714. QualType FieldTypes[] = {
  1715. Context.UnsignedLongTy,
  1716. Context.getPointerType(Context.getObjCIdType()),
  1717. Context.getPointerType(Context.UnsignedLongTy),
  1718. Context.getConstantArrayType(Context.UnsignedLongTy,
  1719. llvm::APInt(32, 5), ArrayType::Normal, 0)
  1720. };
  1721. for (size_t i = 0; i < 4; ++i) {
  1722. FieldDecl *Field = FieldDecl::Create(Context,
  1723. D,
  1724. SourceLocation(),
  1725. SourceLocation(), 0,
  1726. FieldTypes[i], /*TInfo=*/0,
  1727. /*BitWidth=*/0,
  1728. /*Mutable=*/false,
  1729. /*HasInit=*/false);
  1730. Field->setAccess(AS_public);
  1731. D->addDecl(Field);
  1732. }
  1733. D->completeDefinition();
  1734. ObjCFastEnumerationStateType = Context.getTagDeclType(D);
  1735. }
  1736. return ObjCFastEnumerationStateType;
  1737. }
  1738. llvm::Constant *
  1739. CodeGenModule::GetConstantArrayFromStringLiteral(const StringLiteral *E) {
  1740. assert(!E->getType()->isPointerType() && "Strings are always arrays");
  1741. // Don't emit it as the address of the string, emit the string data itself
  1742. // as an inline array.
  1743. if (E->getCharByteWidth() == 1) {
  1744. SmallString<64> Str(E->getString());
  1745. // Resize the string to the right size, which is indicated by its type.
  1746. const ConstantArrayType *CAT = Context.getAsConstantArrayType(E->getType());
  1747. Str.resize(CAT->getSize().getZExtValue());
  1748. return llvm::ConstantDataArray::getString(VMContext, Str, false);
  1749. }
  1750. llvm::ArrayType *AType =
  1751. cast<llvm::ArrayType>(getTypes().ConvertType(E->getType()));
  1752. llvm::Type *ElemTy = AType->getElementType();
  1753. unsigned NumElements = AType->getNumElements();
  1754. // Wide strings have either 2-byte or 4-byte elements.
  1755. if (ElemTy->getPrimitiveSizeInBits() == 16) {
  1756. SmallVector<uint16_t, 32> Elements;
  1757. Elements.reserve(NumElements);
  1758. for(unsigned i = 0, e = E->getLength(); i != e; ++i)
  1759. Elements.push_back(E->getCodeUnit(i));
  1760. Elements.resize(NumElements);
  1761. return llvm::ConstantDataArray::get(VMContext, Elements);
  1762. }
  1763. assert(ElemTy->getPrimitiveSizeInBits() == 32);
  1764. SmallVector<uint32_t, 32> Elements;
  1765. Elements.reserve(NumElements);
  1766. for(unsigned i = 0, e = E->getLength(); i != e; ++i)
  1767. Elements.push_back(E->getCodeUnit(i));
  1768. Elements.resize(NumElements);
  1769. return llvm::ConstantDataArray::get(VMContext, Elements);
  1770. }
  1771. /// GetAddrOfConstantStringFromLiteral - Return a pointer to a
  1772. /// constant array for the given string literal.
  1773. llvm::Constant *
  1774. CodeGenModule::GetAddrOfConstantStringFromLiteral(const StringLiteral *S) {
  1775. CharUnits Align = getContext().getTypeAlignInChars(S->getType());
  1776. if (S->isAscii() || S->isUTF8()) {
  1777. SmallString<64> Str(S->getString());
  1778. // Resize the string to the right size, which is indicated by its type.
  1779. const ConstantArrayType *CAT = Context.getAsConstantArrayType(S->getType());
  1780. Str.resize(CAT->getSize().getZExtValue());
  1781. return GetAddrOfConstantString(Str, /*GlobalName*/ 0, Align.getQuantity());
  1782. }
  1783. // FIXME: the following does not memoize wide strings.
  1784. llvm::Constant *C = GetConstantArrayFromStringLiteral(S);
  1785. llvm::GlobalVariable *GV =
  1786. new llvm::GlobalVariable(getModule(),C->getType(),
  1787. !Features.WritableStrings,
  1788. llvm::GlobalValue::PrivateLinkage,
  1789. C,".str");
  1790. GV->setAlignment(Align.getQuantity());
  1791. GV->setUnnamedAddr(true);
  1792. return GV;
  1793. }
  1794. /// GetAddrOfConstantStringFromObjCEncode - Return a pointer to a constant
  1795. /// array for the given ObjCEncodeExpr node.
  1796. llvm::Constant *
  1797. CodeGenModule::GetAddrOfConstantStringFromObjCEncode(const ObjCEncodeExpr *E) {
  1798. std::string Str;
  1799. getContext().getObjCEncodingForType(E->getEncodedType(), Str);
  1800. return GetAddrOfConstantCString(Str);
  1801. }
  1802. /// GenerateWritableString -- Creates storage for a string literal.
  1803. static llvm::GlobalVariable *GenerateStringLiteral(StringRef str,
  1804. bool constant,
  1805. CodeGenModule &CGM,
  1806. const char *GlobalName,
  1807. unsigned Alignment) {
  1808. // Create Constant for this string literal. Don't add a '\0'.
  1809. llvm::Constant *C =
  1810. llvm::ConstantDataArray::getString(CGM.getLLVMContext(), str, false);
  1811. // Create a global variable for this string
  1812. llvm::GlobalVariable *GV =
  1813. new llvm::GlobalVariable(CGM.getModule(), C->getType(), constant,
  1814. llvm::GlobalValue::PrivateLinkage,
  1815. C, GlobalName);
  1816. GV->setAlignment(Alignment);
  1817. GV->setUnnamedAddr(true);
  1818. return GV;
  1819. }
  1820. /// GetAddrOfConstantString - Returns a pointer to a character array
  1821. /// containing the literal. This contents are exactly that of the
  1822. /// given string, i.e. it will not be null terminated automatically;
  1823. /// see GetAddrOfConstantCString. Note that whether the result is
  1824. /// actually a pointer to an LLVM constant depends on
  1825. /// Feature.WriteableStrings.
  1826. ///
  1827. /// The result has pointer to array type.
  1828. llvm::Constant *CodeGenModule::GetAddrOfConstantString(StringRef Str,
  1829. const char *GlobalName,
  1830. unsigned Alignment) {
  1831. // Get the default prefix if a name wasn't specified.
  1832. if (!GlobalName)
  1833. GlobalName = ".str";
  1834. // Don't share any string literals if strings aren't constant.
  1835. if (Features.WritableStrings)
  1836. return GenerateStringLiteral(Str, false, *this, GlobalName, Alignment);
  1837. llvm::StringMapEntry<llvm::GlobalVariable *> &Entry =
  1838. ConstantStringMap.GetOrCreateValue(Str);
  1839. if (llvm::GlobalVariable *GV = Entry.getValue()) {
  1840. if (Alignment > GV->getAlignment()) {
  1841. GV->setAlignment(Alignment);
  1842. }
  1843. return GV;
  1844. }
  1845. // Create a global variable for this.
  1846. llvm::GlobalVariable *GV = GenerateStringLiteral(Str, true, *this, GlobalName,
  1847. Alignment);
  1848. Entry.setValue(GV);
  1849. return GV;
  1850. }
  1851. /// GetAddrOfConstantCString - Returns a pointer to a character
  1852. /// array containing the literal and a terminating '\0'
  1853. /// character. The result has pointer to array type.
  1854. llvm::Constant *CodeGenModule::GetAddrOfConstantCString(const std::string &Str,
  1855. const char *GlobalName,
  1856. unsigned Alignment) {
  1857. StringRef StrWithNull(Str.c_str(), Str.size() + 1);
  1858. return GetAddrOfConstantString(StrWithNull, GlobalName, Alignment);
  1859. }
  1860. /// EmitObjCPropertyImplementations - Emit information for synthesized
  1861. /// properties for an implementation.
  1862. void CodeGenModule::EmitObjCPropertyImplementations(const
  1863. ObjCImplementationDecl *D) {
  1864. for (ObjCImplementationDecl::propimpl_iterator
  1865. i = D->propimpl_begin(), e = D->propimpl_end(); i != e; ++i) {
  1866. ObjCPropertyImplDecl *PID = *i;
  1867. // Dynamic is just for type-checking.
  1868. if (PID->getPropertyImplementation() == ObjCPropertyImplDecl::Synthesize) {
  1869. ObjCPropertyDecl *PD = PID->getPropertyDecl();
  1870. // Determine which methods need to be implemented, some may have
  1871. // been overridden. Note that ::isSynthesized is not the method
  1872. // we want, that just indicates if the decl came from a
  1873. // property. What we want to know is if the method is defined in
  1874. // this implementation.
  1875. if (!D->getInstanceMethod(PD->getGetterName()))
  1876. CodeGenFunction(*this).GenerateObjCGetter(
  1877. const_cast<ObjCImplementationDecl *>(D), PID);
  1878. if (!PD->isReadOnly() &&
  1879. !D->getInstanceMethod(PD->getSetterName()))
  1880. CodeGenFunction(*this).GenerateObjCSetter(
  1881. const_cast<ObjCImplementationDecl *>(D), PID);
  1882. }
  1883. }
  1884. }
  1885. static bool needsDestructMethod(ObjCImplementationDecl *impl) {
  1886. const ObjCInterfaceDecl *iface = impl->getClassInterface();
  1887. for (const ObjCIvarDecl *ivar = iface->all_declared_ivar_begin();
  1888. ivar; ivar = ivar->getNextIvar())
  1889. if (ivar->getType().isDestructedType())
  1890. return true;
  1891. return false;
  1892. }
  1893. /// EmitObjCIvarInitializations - Emit information for ivar initialization
  1894. /// for an implementation.
  1895. void CodeGenModule::EmitObjCIvarInitializations(ObjCImplementationDecl *D) {
  1896. // We might need a .cxx_destruct even if we don't have any ivar initializers.
  1897. if (needsDestructMethod(D)) {
  1898. IdentifierInfo *II = &getContext().Idents.get(".cxx_destruct");
  1899. Selector cxxSelector = getContext().Selectors.getSelector(0, &II);
  1900. ObjCMethodDecl *DTORMethod =
  1901. ObjCMethodDecl::Create(getContext(), D->getLocation(), D->getLocation(),
  1902. cxxSelector, getContext().VoidTy, 0, D,
  1903. /*isInstance=*/true, /*isVariadic=*/false,
  1904. /*isSynthesized=*/true, /*isImplicitlyDeclared=*/true,
  1905. /*isDefined=*/false, ObjCMethodDecl::Required);
  1906. D->addInstanceMethod(DTORMethod);
  1907. CodeGenFunction(*this).GenerateObjCCtorDtorMethod(D, DTORMethod, false);
  1908. D->setHasCXXStructors(true);
  1909. }
  1910. // If the implementation doesn't have any ivar initializers, we don't need
  1911. // a .cxx_construct.
  1912. if (D->getNumIvarInitializers() == 0)
  1913. return;
  1914. IdentifierInfo *II = &getContext().Idents.get(".cxx_construct");
  1915. Selector cxxSelector = getContext().Selectors.getSelector(0, &II);
  1916. // The constructor returns 'self'.
  1917. ObjCMethodDecl *CTORMethod = ObjCMethodDecl::Create(getContext(),
  1918. D->getLocation(),
  1919. D->getLocation(),
  1920. cxxSelector,
  1921. getContext().getObjCIdType(), 0,
  1922. D, /*isInstance=*/true,
  1923. /*isVariadic=*/false,
  1924. /*isSynthesized=*/true,
  1925. /*isImplicitlyDeclared=*/true,
  1926. /*isDefined=*/false,
  1927. ObjCMethodDecl::Required);
  1928. D->addInstanceMethod(CTORMethod);
  1929. CodeGenFunction(*this).GenerateObjCCtorDtorMethod(D, CTORMethod, true);
  1930. D->setHasCXXStructors(true);
  1931. }
  1932. /// EmitNamespace - Emit all declarations in a namespace.
  1933. void CodeGenModule::EmitNamespace(const NamespaceDecl *ND) {
  1934. for (RecordDecl::decl_iterator I = ND->decls_begin(), E = ND->decls_end();
  1935. I != E; ++I)
  1936. EmitTopLevelDecl(*I);
  1937. }
  1938. // EmitLinkageSpec - Emit all declarations in a linkage spec.
  1939. void CodeGenModule::EmitLinkageSpec(const LinkageSpecDecl *LSD) {
  1940. if (LSD->getLanguage() != LinkageSpecDecl::lang_c &&
  1941. LSD->getLanguage() != LinkageSpecDecl::lang_cxx) {
  1942. ErrorUnsupported(LSD, "linkage spec");
  1943. return;
  1944. }
  1945. for (RecordDecl::decl_iterator I = LSD->decls_begin(), E = LSD->decls_end();
  1946. I != E; ++I)
  1947. EmitTopLevelDecl(*I);
  1948. }
  1949. /// EmitTopLevelDecl - Emit code for a single top level declaration.
  1950. void CodeGenModule::EmitTopLevelDecl(Decl *D) {
  1951. // If an error has occurred, stop code generation, but continue
  1952. // parsing and semantic analysis (to ensure all warnings and errors
  1953. // are emitted).
  1954. if (Diags.hasErrorOccurred())
  1955. return;
  1956. // Ignore dependent declarations.
  1957. if (D->getDeclContext() && D->getDeclContext()->isDependentContext())
  1958. return;
  1959. switch (D->getKind()) {
  1960. case Decl::CXXConversion:
  1961. case Decl::CXXMethod:
  1962. case Decl::Function:
  1963. // Skip function templates
  1964. if (cast<FunctionDecl>(D)->getDescribedFunctionTemplate() ||
  1965. cast<FunctionDecl>(D)->isLateTemplateParsed())
  1966. return;
  1967. EmitGlobal(cast<FunctionDecl>(D));
  1968. break;
  1969. case Decl::Var:
  1970. EmitGlobal(cast<VarDecl>(D));
  1971. break;
  1972. // Indirect fields from global anonymous structs and unions can be
  1973. // ignored; only the actual variable requires IR gen support.
  1974. case Decl::IndirectField:
  1975. break;
  1976. // C++ Decls
  1977. case Decl::Namespace:
  1978. EmitNamespace(cast<NamespaceDecl>(D));
  1979. break;
  1980. // No code generation needed.
  1981. case Decl::UsingShadow:
  1982. case Decl::Using:
  1983. case Decl::UsingDirective:
  1984. case Decl::ClassTemplate:
  1985. case Decl::FunctionTemplate:
  1986. case Decl::TypeAliasTemplate:
  1987. case Decl::NamespaceAlias:
  1988. case Decl::Block:
  1989. case Decl::Import:
  1990. break;
  1991. case Decl::CXXConstructor:
  1992. // Skip function templates
  1993. if (cast<FunctionDecl>(D)->getDescribedFunctionTemplate() ||
  1994. cast<FunctionDecl>(D)->isLateTemplateParsed())
  1995. return;
  1996. EmitCXXConstructors(cast<CXXConstructorDecl>(D));
  1997. break;
  1998. case Decl::CXXDestructor:
  1999. if (cast<FunctionDecl>(D)->isLateTemplateParsed())
  2000. return;
  2001. EmitCXXDestructors(cast<CXXDestructorDecl>(D));
  2002. break;
  2003. case Decl::StaticAssert:
  2004. // Nothing to do.
  2005. break;
  2006. // Objective-C Decls
  2007. // Forward declarations, no (immediate) code generation.
  2008. case Decl::ObjCInterface:
  2009. break;
  2010. case Decl::ObjCCategory: {
  2011. ObjCCategoryDecl *CD = cast<ObjCCategoryDecl>(D);
  2012. if (CD->IsClassExtension() && CD->hasSynthBitfield())
  2013. Context.ResetObjCLayout(CD->getClassInterface());
  2014. break;
  2015. }
  2016. case Decl::ObjCProtocol: {
  2017. ObjCProtocolDecl *Proto = cast<ObjCProtocolDecl>(D);
  2018. if (Proto->isThisDeclarationADefinition())
  2019. ObjCRuntime->GenerateProtocol(Proto);
  2020. break;
  2021. }
  2022. case Decl::ObjCCategoryImpl:
  2023. // Categories have properties but don't support synthesize so we
  2024. // can ignore them here.
  2025. ObjCRuntime->GenerateCategory(cast<ObjCCategoryImplDecl>(D));
  2026. break;
  2027. case Decl::ObjCImplementation: {
  2028. ObjCImplementationDecl *OMD = cast<ObjCImplementationDecl>(D);
  2029. if (Features.ObjCNonFragileABI2 && OMD->hasSynthBitfield())
  2030. Context.ResetObjCLayout(OMD->getClassInterface());
  2031. EmitObjCPropertyImplementations(OMD);
  2032. EmitObjCIvarInitializations(OMD);
  2033. ObjCRuntime->GenerateClass(OMD);
  2034. break;
  2035. }
  2036. case Decl::ObjCMethod: {
  2037. ObjCMethodDecl *OMD = cast<ObjCMethodDecl>(D);
  2038. // If this is not a prototype, emit the body.
  2039. if (OMD->getBody())
  2040. CodeGenFunction(*this).GenerateObjCMethod(OMD);
  2041. break;
  2042. }
  2043. case Decl::ObjCCompatibleAlias:
  2044. ObjCRuntime->RegisterAlias(cast<ObjCCompatibleAliasDecl>(D));
  2045. break;
  2046. case Decl::LinkageSpec:
  2047. EmitLinkageSpec(cast<LinkageSpecDecl>(D));
  2048. break;
  2049. case Decl::FileScopeAsm: {
  2050. FileScopeAsmDecl *AD = cast<FileScopeAsmDecl>(D);
  2051. StringRef AsmString = AD->getAsmString()->getString();
  2052. const std::string &S = getModule().getModuleInlineAsm();
  2053. if (S.empty())
  2054. getModule().setModuleInlineAsm(AsmString);
  2055. else if (*--S.end() == '\n')
  2056. getModule().setModuleInlineAsm(S + AsmString.str());
  2057. else
  2058. getModule().setModuleInlineAsm(S + '\n' + AsmString.str());
  2059. break;
  2060. }
  2061. default:
  2062. // Make sure we handled everything we should, every other kind is a
  2063. // non-top-level decl. FIXME: Would be nice to have an isTopLevelDeclKind
  2064. // function. Need to recode Decl::Kind to do that easily.
  2065. assert(isa<TypeDecl>(D) && "Unsupported decl kind");
  2066. }
  2067. }
  2068. /// Turns the given pointer into a constant.
  2069. static llvm::Constant *GetPointerConstant(llvm::LLVMContext &Context,
  2070. const void *Ptr) {
  2071. uintptr_t PtrInt = reinterpret_cast<uintptr_t>(Ptr);
  2072. llvm::Type *i64 = llvm::Type::getInt64Ty(Context);
  2073. return llvm::ConstantInt::get(i64, PtrInt);
  2074. }
  2075. static void EmitGlobalDeclMetadata(CodeGenModule &CGM,
  2076. llvm::NamedMDNode *&GlobalMetadata,
  2077. GlobalDecl D,
  2078. llvm::GlobalValue *Addr) {
  2079. if (!GlobalMetadata)
  2080. GlobalMetadata =
  2081. CGM.getModule().getOrInsertNamedMetadata("clang.global.decl.ptrs");
  2082. // TODO: should we report variant information for ctors/dtors?
  2083. llvm::Value *Ops[] = {
  2084. Addr,
  2085. GetPointerConstant(CGM.getLLVMContext(), D.getDecl())
  2086. };
  2087. GlobalMetadata->addOperand(llvm::MDNode::get(CGM.getLLVMContext(), Ops));
  2088. }
  2089. /// Emits metadata nodes associating all the global values in the
  2090. /// current module with the Decls they came from. This is useful for
  2091. /// projects using IR gen as a subroutine.
  2092. ///
  2093. /// Since there's currently no way to associate an MDNode directly
  2094. /// with an llvm::GlobalValue, we create a global named metadata
  2095. /// with the name 'clang.global.decl.ptrs'.
  2096. void CodeGenModule::EmitDeclMetadata() {
  2097. llvm::NamedMDNode *GlobalMetadata = 0;
  2098. // StaticLocalDeclMap
  2099. for (llvm::DenseMap<GlobalDecl,StringRef>::iterator
  2100. I = MangledDeclNames.begin(), E = MangledDeclNames.end();
  2101. I != E; ++I) {
  2102. llvm::GlobalValue *Addr = getModule().getNamedValue(I->second);
  2103. EmitGlobalDeclMetadata(*this, GlobalMetadata, I->first, Addr);
  2104. }
  2105. }
  2106. /// Emits metadata nodes for all the local variables in the current
  2107. /// function.
  2108. void CodeGenFunction::EmitDeclMetadata() {
  2109. if (LocalDeclMap.empty()) return;
  2110. llvm::LLVMContext &Context = getLLVMContext();
  2111. // Find the unique metadata ID for this name.
  2112. unsigned DeclPtrKind = Context.getMDKindID("clang.decl.ptr");
  2113. llvm::NamedMDNode *GlobalMetadata = 0;
  2114. for (llvm::DenseMap<const Decl*, llvm::Value*>::iterator
  2115. I = LocalDeclMap.begin(), E = LocalDeclMap.end(); I != E; ++I) {
  2116. const Decl *D = I->first;
  2117. llvm::Value *Addr = I->second;
  2118. if (llvm::AllocaInst *Alloca = dyn_cast<llvm::AllocaInst>(Addr)) {
  2119. llvm::Value *DAddr = GetPointerConstant(getLLVMContext(), D);
  2120. Alloca->setMetadata(DeclPtrKind, llvm::MDNode::get(Context, DAddr));
  2121. } else if (llvm::GlobalValue *GV = dyn_cast<llvm::GlobalValue>(Addr)) {
  2122. GlobalDecl GD = GlobalDecl(cast<VarDecl>(D));
  2123. EmitGlobalDeclMetadata(CGM, GlobalMetadata, GD, GV);
  2124. }
  2125. }
  2126. }
  2127. void CodeGenModule::EmitCoverageFile() {
  2128. if (!getCodeGenOpts().CoverageFile.empty()) {
  2129. if (llvm::NamedMDNode *CUNode = TheModule.getNamedMetadata("llvm.dbg.cu")) {
  2130. llvm::NamedMDNode *GCov = TheModule.getOrInsertNamedMetadata("llvm.gcov");
  2131. llvm::LLVMContext &Ctx = TheModule.getContext();
  2132. llvm::MDString *CoverageFile =
  2133. llvm::MDString::get(Ctx, getCodeGenOpts().CoverageFile);
  2134. for (int i = 0, e = CUNode->getNumOperands(); i != e; ++i) {
  2135. llvm::MDNode *CU = CUNode->getOperand(i);
  2136. llvm::Value *node[] = { CoverageFile, CU };
  2137. llvm::MDNode *N = llvm::MDNode::get(Ctx, node);
  2138. GCov->addOperand(N);
  2139. }
  2140. }
  2141. }
  2142. }