blob: 19c3b064d65d8188658637bb51328a83b8fad21e [file] [log] [blame]
Chris Lattnerbed31442007-05-28 01:07:47 +00001//===--- CodeGenModule.cpp - Emit LLVM Code from ASTs for a Module --------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner5b12ab82007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chris Lattnerbed31442007-05-28 01:07:47 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This coordinates the per-module state used while generating code.
11//
12//===----------------------------------------------------------------------===//
13
14#include "CodeGenModule.h"
Chris Lattner984fac52009-03-26 05:00:52 +000015#include "CGDebugInfo.h"
Chris Lattnerbed31442007-05-28 01:07:47 +000016#include "CodeGenFunction.h"
Dan Gohman947c9af2010-10-14 23:06:10 +000017#include "CodeGenTBAA.h"
Daniel Dunbar3d7c90b2008-09-08 21:33:45 +000018#include "CGCall.h"
John McCall5d865c322010-08-31 07:33:07 +000019#include "CGCXXABI.h"
Daniel Dunbar89da6ad2008-08-13 00:59:25 +000020#include "CGObjCRuntime.h"
Anton Korobeynikov55bcea12010-01-10 12:58:08 +000021#include "TargetInfo.h"
Chandler Carruth85098242010-06-15 23:19:56 +000022#include "clang/Frontend/CodeGenOptions.h"
Chris Lattner2ccb73b2007-06-16 00:16:26 +000023#include "clang/AST/ASTContext.h"
Ken Dyck98ca7942010-01-26 13:48:07 +000024#include "clang/AST/CharUnits.h"
Daniel Dunbar6e8aa532008-08-11 05:35:13 +000025#include "clang/AST/DeclObjC.h"
Chris Lattnerb8c18fa2008-11-04 16:51:42 +000026#include "clang/AST/DeclCXX.h"
Douglas Gregor5dd34742010-06-21 18:41:26 +000027#include "clang/AST/DeclTemplate.h"
Peter Collingbourne0ff0b372011-01-13 18:57:25 +000028#include "clang/AST/Mangle.h"
Anders Carlssonb1d3f7c2009-11-30 23:41:22 +000029#include "clang/AST/RecordLayout.h"
Chris Lattnerd45aa2a2007-12-02 07:19:18 +000030#include "clang/Basic/Diagnostic.h"
Nate Begemanfaae0812008-04-19 04:17:09 +000031#include "clang/Basic/SourceManager.h"
Chris Lattner09153c02007-06-22 18:48:09 +000032#include "clang/Basic/TargetInfo.h"
Steve Naroff29cae662009-04-01 15:50:34 +000033#include "clang/Basic/ConvertUTF.h"
Nate Begemanaca747a2008-03-09 03:09:36 +000034#include "llvm/CallingConv.h"
Chris Lattner1eec6602007-08-31 04:31:45 +000035#include "llvm/Module.h"
Chris Lattner09153c02007-06-22 18:48:09 +000036#include "llvm/Intrinsics.h"
Chris Lattner5e124bf2009-12-28 21:44:41 +000037#include "llvm/LLVMContext.h"
John McCallbeec5a02010-03-06 00:35:14 +000038#include "llvm/ADT/Triple.h"
Rafael Espindolaac00f5d2011-02-10 23:59:36 +000039#include "llvm/Target/Mangler.h"
Anton Korobeynikov1200aca2008-06-01 14:13:53 +000040#include "llvm/Target/TargetData.h"
Gabor Greifd0ef1342010-04-10 02:56:12 +000041#include "llvm/Support/CallSite.h"
Chris Lattner15275e52009-11-07 09:22:46 +000042#include "llvm/Support/ErrorHandling.h"
Chris Lattnerbed31442007-05-28 01:07:47 +000043using namespace clang;
44using namespace CodeGen;
45
Julien Lerouge5a6b6982011-09-09 22:41:49 +000046static const char AnnotationSection[] = "llvm.metadata";
47
John McCall614dbdc2010-08-22 21:01:12 +000048static CGCXXABI &createCXXABI(CodeGenModule &CGM) {
Douglas Gregore8bbc122011-09-02 00:18:52 +000049 switch (CGM.getContext().getTargetInfo().getCXXABI()) {
John McCall614dbdc2010-08-22 21:01:12 +000050 case CXXABI_ARM: return *CreateARMCXXABI(CGM);
51 case CXXABI_Itanium: return *CreateItaniumCXXABI(CGM);
52 case CXXABI_Microsoft: return *CreateMicrosoftCXXABI(CGM);
53 }
54
55 llvm_unreachable("invalid C++ ABI kind");
56 return *CreateItaniumCXXABI(CGM);
57}
58
Chris Lattnerbed31442007-05-28 01:07:47 +000059
Chandler Carruthbc55fe22009-11-12 17:24:48 +000060CodeGenModule::CodeGenModule(ASTContext &C, const CodeGenOptions &CGO,
John McCall731be662010-03-04 04:29:44 +000061 llvm::Module &M, const llvm::TargetData &TD,
62 Diagnostic &diags)
John McCallad7c5c12011-02-08 08:22:06 +000063 : Context(C), Features(C.getLangOptions()), CodeGenOpts(CGO), TheModule(M),
John McCall731be662010-03-04 04:29:44 +000064 TheTargetData(TD), TheTargetCodeGenInfo(0), Diags(diags),
John McCall614dbdc2010-08-22 21:01:12 +000065 ABI(createCXXABI(*this)),
Daniel Dunbar0de2faa2011-06-21 18:54:39 +000066 Types(C, M, TD, getTargetCodeGenInfo().getABIInfo(), ABI, CGO),
Dan Gohman947c9af2010-10-14 23:06:10 +000067 TBAA(0),
Peter Collingbournee1d20992011-07-27 20:29:46 +000068 VTables(*this), ObjCRuntime(0), DebugInfo(0), ARCData(0), RRData(0),
Fariborz Jahanian50c925f2010-10-19 17:19:29 +000069 CFConstantStringClassRef(0), ConstantStringClassRef(0),
Douglas Gregorabf4e0d2011-08-09 15:54:21 +000070 NSConstantStringType(0),
Daniel Dunbar900546d2010-07-16 00:00:15 +000071 VMContext(M.getContext()),
72 NSConcreteGlobalBlock(0), NSConcreteStackBlock(0),
John McCallad7c5c12011-02-08 08:22:06 +000073 BlockObjectAssign(0), BlockObjectDispose(0),
74 BlockDescriptorType(0), GenericBlockLiteralType(0) {
David Chisnall067f0ed2011-03-22 21:21:24 +000075 if (Features.ObjC1)
76 createObjCRuntime();
Sanjiv Gupta15cb6692008-05-08 08:54:20 +000077
Dan Gohman947c9af2010-10-14 23:06:10 +000078 // Enable TBAA unless it's suppressed.
79 if (!CodeGenOpts.RelaxedAliasing && CodeGenOpts.OptimizationLevel > 0)
Dan Gohman2e29eb52010-10-15 20:23:12 +000080 TBAA = new CodeGenTBAA(Context, VMContext, getLangOptions(),
81 ABI.getMangleContext());
Dan Gohman947c9af2010-10-14 23:06:10 +000082
Nick Lewycky207bce32011-04-21 23:44:07 +000083 // If debug info or coverage generation is enabled, create the CGDebugInfo
84 // object.
85 if (CodeGenOpts.DebugInfo || CodeGenOpts.EmitGcovArcs ||
86 CodeGenOpts.EmitGcovNotes)
87 DebugInfo = new CGDebugInfo(*this);
John McCallad7c5c12011-02-08 08:22:06 +000088
89 Block.GlobalUniqueCount = 0;
John McCalle3dc1702011-02-15 09:22:45 +000090
John McCall31168b02011-06-15 23:02:42 +000091 if (C.getLangOptions().ObjCAutoRefCount)
92 ARCData = new ARCEntrypoints();
93 RRData = new RREntrypoints();
94
John McCalle3dc1702011-02-15 09:22:45 +000095 // Initialize the type cache.
96 llvm::LLVMContext &LLVMContext = M.getContext();
John McCall9dc0db22011-05-15 01:53:33 +000097 VoidTy = llvm::Type::getVoidTy(LLVMContext);
98 Int8Ty = llvm::Type::getInt8Ty(LLVMContext);
99 Int32Ty = llvm::Type::getInt32Ty(LLVMContext);
100 Int64Ty = llvm::Type::getInt64Ty(LLVMContext);
Douglas Gregore8bbc122011-09-02 00:18:52 +0000101 PointerWidthInBits = C.getTargetInfo().getPointerWidth(0);
John McCallc533cb72011-02-22 06:44:22 +0000102 PointerAlignInBytes =
Douglas Gregore8bbc122011-09-02 00:18:52 +0000103 C.toCharUnitsFromBits(C.getTargetInfo().getPointerAlign(0)).getQuantity();
104 IntTy = llvm::IntegerType::get(LLVMContext, C.getTargetInfo().getIntWidth());
John McCalle3dc1702011-02-15 09:22:45 +0000105 IntPtrTy = llvm::IntegerType::get(LLVMContext, PointerWidthInBits);
106 Int8PtrTy = Int8Ty->getPointerTo(0);
107 Int8PtrPtrTy = Int8PtrTy->getPointerTo(0);
Chris Lattnera087ff92008-03-01 08:45:05 +0000108}
109
110CodeGenModule::~CodeGenModule() {
Peter Collingbournee1d20992011-07-27 20:29:46 +0000111 delete ObjCRuntime;
John McCall614dbdc2010-08-22 21:01:12 +0000112 delete &ABI;
Dan Gohmand19ee8a2010-10-15 18:04:46 +0000113 delete TBAA;
Ted Kremenek2c674f62008-08-05 18:50:11 +0000114 delete DebugInfo;
John McCall31168b02011-06-15 23:02:42 +0000115 delete ARCData;
116 delete RRData;
Ted Kremenek2c674f62008-08-05 18:50:11 +0000117}
118
David Chisnall481e3a82010-01-23 02:40:42 +0000119void CodeGenModule::createObjCRuntime() {
120 if (!Features.NeXTRuntime)
Peter Collingbournee1d20992011-07-27 20:29:46 +0000121 ObjCRuntime = CreateGNUObjCRuntime(*this);
David Chisnall481e3a82010-01-23 02:40:42 +0000122 else
Peter Collingbournee1d20992011-07-27 20:29:46 +0000123 ObjCRuntime = CreateMacObjCRuntime(*this);
David Chisnall481e3a82010-01-23 02:40:42 +0000124}
125
Ted Kremenek2c674f62008-08-05 18:50:11 +0000126void CodeGenModule::Release() {
Chris Lattnera5ae54a2009-03-22 21:21:57 +0000127 EmitDeferred();
Eli Friedman5866fe32010-01-08 00:50:11 +0000128 EmitCXXGlobalInitFunc();
Daniel Dunbarfe06df42010-03-20 04:15:41 +0000129 EmitCXXGlobalDtorFunc();
Peter Collingbournee1d20992011-07-27 20:29:46 +0000130 if (ObjCRuntime)
131 if (llvm::Function *ObjCInitFunction = ObjCRuntime->ModuleInitFunction())
Daniel Dunbar8d480592008-08-11 18:12:00 +0000132 AddGlobalCtor(ObjCInitFunction);
Daniel Dunbar74aa7e12008-08-01 00:01:51 +0000133 EmitCtorList(GlobalCtors, "llvm.global_ctors");
134 EmitCtorList(GlobalDtors, "llvm.global_dtors");
Julien Lerouge5a6b6982011-09-09 22:41:49 +0000135 EmitGlobalAnnotations();
Daniel Dunbar08b26a02009-02-13 20:29:50 +0000136 EmitLLVMUsed();
John McCall09ae0322010-07-06 23:57:41 +0000137
John McCall0bdb1fd2010-09-16 06:16:50 +0000138 SimplifyPersonality();
139
John McCall09ae0322010-07-06 23:57:41 +0000140 if (getCodeGenOpts().EmitDeclMetadata)
141 EmitDeclMetadata();
Nick Lewycky480cb992011-05-04 20:46:58 +0000142
143 if (getCodeGenOpts().EmitGcovArcs || getCodeGenOpts().EmitGcovNotes)
Nick Lewycky85c011d2011-05-05 00:08:20 +0000144 EmitCoverageFile();
Devang Patelffa30ab2011-08-16 20:58:22 +0000145
146 if (DebugInfo)
147 DebugInfo->finalize();
Daniel Dunbar23fd4622008-10-01 00:49:24 +0000148}
149
Devang Patel945b8ae2011-03-23 16:29:39 +0000150void 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
Dan Gohman947c9af2010-10-14 23:06:10 +0000157llvm::MDNode *CodeGenModule::getTBAAInfo(QualType QTy) {
158 if (!TBAA)
159 return 0;
160 return TBAA->getTBAAInfo(QTy);
161}
162
163void CodeGenModule::DecorateInstruction(llvm::Instruction *Inst,
164 llvm::MDNode *TBAAInfo) {
165 Inst->setMetadata(llvm::LLVMContext::MD_tbaa, TBAAInfo);
166}
167
John McCallbeec5a02010-03-06 00:35:14 +0000168bool CodeGenModule::isTargetDarwin() const {
Douglas Gregore8bbc122011-09-02 00:18:52 +0000169 return getContext().getTargetInfo().getTriple().isOSDarwin();
John McCallbeec5a02010-03-06 00:35:14 +0000170}
171
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000172void CodeGenModule::Error(SourceLocation loc, StringRef error) {
John McCall7ef5cb32011-03-18 02:56:14 +0000173 unsigned diagID = getDiags().getCustomDiagID(Diagnostic::Error, error);
174 getDiags().Report(Context.getFullLoc(loc), diagID);
175}
176
Daniel Dunbara7c8cf62008-08-16 00:56:44 +0000177/// ErrorUnsupported - Print out an error that codegen doesn't support the
Chris Lattnerd45aa2a2007-12-02 07:19:18 +0000178/// specified stmt yet.
Daniel Dunbarf2cf6d12008-09-04 03:43:08 +0000179void CodeGenModule::ErrorUnsupported(const Stmt *S, const char *Type,
180 bool OmitOnError) {
181 if (OmitOnError && getDiags().hasErrorOccurred())
182 return;
Mike Stump11289f42009-09-09 15:08:12 +0000183 unsigned DiagID = getDiags().getCustomDiagID(Diagnostic::Error,
Daniel Dunbarfe2fb0a2009-02-06 19:18:03 +0000184 "cannot compile this %0 yet");
Chris Lattnerd45aa2a2007-12-02 07:19:18 +0000185 std::string Msg = Type;
Chris Lattner8488c822008-11-18 07:04:44 +0000186 getDiags().Report(Context.getFullLoc(S->getLocStart()), DiagID)
187 << Msg << S->getSourceRange();
Chris Lattnerd45aa2a2007-12-02 07:19:18 +0000188}
Chris Lattner41af8182007-12-02 06:27:33 +0000189
Daniel Dunbara7c8cf62008-08-16 00:56:44 +0000190/// ErrorUnsupported - Print out an error that codegen doesn't support the
Chris Lattner38376f12008-01-12 07:05:38 +0000191/// specified decl yet.
Daniel Dunbarf2cf6d12008-09-04 03:43:08 +0000192void CodeGenModule::ErrorUnsupported(const Decl *D, const char *Type,
193 bool OmitOnError) {
194 if (OmitOnError && getDiags().hasErrorOccurred())
195 return;
Mike Stump11289f42009-09-09 15:08:12 +0000196 unsigned DiagID = getDiags().getCustomDiagID(Diagnostic::Error,
Daniel Dunbarfe2fb0a2009-02-06 19:18:03 +0000197 "cannot compile this %0 yet");
Chris Lattner38376f12008-01-12 07:05:38 +0000198 std::string Msg = Type;
Chris Lattner8488c822008-11-18 07:04:44 +0000199 getDiags().Report(Context.getFullLoc(D->getLocation()), DiagID) << Msg;
Chris Lattner38376f12008-01-12 07:05:38 +0000200}
201
John McCall23c29fe2011-06-24 21:55:10 +0000202llvm::ConstantInt *CodeGenModule::getSize(CharUnits size) {
203 return llvm::ConstantInt::get(SizeTy, size.getQuantity());
204}
205
Mike Stump11289f42009-09-09 15:08:12 +0000206void CodeGenModule::setGlobalVisibility(llvm::GlobalValue *GV,
Anders Carlssonc6a47892011-01-29 19:39:23 +0000207 const NamedDecl *D) const {
Daniel Dunbarf5f359f2009-04-14 06:00:08 +0000208 // Internal definitions always have default visibility.
Chris Lattner6a0f9072009-04-14 05:27:13 +0000209 if (GV->hasLocalLinkage()) {
Daniel Dunbard272cca2009-04-10 20:26:50 +0000210 GV->setVisibility(llvm::GlobalValue::DefaultVisibility);
Daniel Dunbar15894b72009-04-07 05:48:37 +0000211 return;
Daniel Dunbard272cca2009-04-10 20:26:50 +0000212 }
Daniel Dunbar15894b72009-04-07 05:48:37 +0000213
John McCallc273f242010-10-30 11:50:40 +0000214 // Set visibility for definitions.
215 NamedDecl::LinkageInfo LV = D->getLinkageAndVisibility();
Fariborz Jahanian8feee2d2011-06-16 20:14:50 +0000216 if (LV.visibilityExplicit() || !GV->hasAvailableExternallyLinkage())
217 GV->setVisibility(GetLLVMVisibility(LV.visibility()));
Dan Gohman75d69da2008-05-22 00:50:06 +0000218}
219
John McCalle16adc22010-08-04 08:34:44 +0000220/// Set the symbol visibility of type information (vtable and RTTI)
221/// associated with the given type.
222void CodeGenModule::setTypeVisibility(llvm::GlobalValue *GV,
223 const CXXRecordDecl *RD,
Anders Carlsson265aa7c2011-01-29 20:24:48 +0000224 TypeVisibilityKind TVK) const {
Anders Carlssonc6a47892011-01-29 19:39:23 +0000225 setGlobalVisibility(GV, RD);
John McCalle16adc22010-08-04 08:34:44 +0000226
John McCallb3732bb2010-08-12 23:36:15 +0000227 if (!CodeGenOpts.HiddenWeakVTables)
228 return;
229
Anders Carlsson678632f2011-01-29 20:36:11 +0000230 // We never want to drop the visibility for RTTI names.
231 if (TVK == TVK_ForRTTIName)
232 return;
233
John McCalle16adc22010-08-04 08:34:44 +0000234 // We want to drop the visibility to hidden for weak type symbols.
235 // This isn't possible if there might be unresolved references
236 // elsewhere that rely on this symbol being visible.
237
John McCall5513fce92010-08-05 20:39:18 +0000238 // This should be kept roughly in sync with setThunkVisibility
239 // in CGVTables.cpp.
240
John McCalle16adc22010-08-04 08:34:44 +0000241 // Preconditions.
Anders Carlsson571e2ad2011-01-24 00:46:19 +0000242 if (GV->getLinkage() != llvm::GlobalVariable::LinkOnceODRLinkage ||
John McCalle16adc22010-08-04 08:34:44 +0000243 GV->getVisibility() != llvm::GlobalVariable::DefaultVisibility)
244 return;
245
246 // Don't override an explicit visibility attribute.
Douglas Gregor1baf38f2011-03-26 12:10:19 +0000247 if (RD->getExplicitVisibility())
John McCalle16adc22010-08-04 08:34:44 +0000248 return;
249
250 switch (RD->getTemplateSpecializationKind()) {
251 // We have to disable the optimization if this is an EI definition
252 // because there might be EI declarations in other shared objects.
253 case TSK_ExplicitInstantiationDefinition:
254 case TSK_ExplicitInstantiationDeclaration:
255 return;
256
John McCall5513fce92010-08-05 20:39:18 +0000257 // Every use of a non-template class's type information has to emit it.
John McCalle16adc22010-08-04 08:34:44 +0000258 case TSK_Undeclared:
259 break;
260
John McCall5513fce92010-08-05 20:39:18 +0000261 // In theory, implicit instantiations can ignore the possibility of
262 // an explicit instantiation declaration because there necessarily
263 // must be an EI definition somewhere with default visibility. In
264 // practice, it's possible to have an explicit instantiation for
265 // an arbitrary template class, and linkers aren't necessarily able
266 // to deal with mixed-visibility symbols.
267 case TSK_ExplicitSpecialization:
John McCalle16adc22010-08-04 08:34:44 +0000268 case TSK_ImplicitInstantiation:
John McCallb3732bb2010-08-12 23:36:15 +0000269 if (!CodeGenOpts.HiddenWeakTemplateVTables)
John McCall5513fce92010-08-05 20:39:18 +0000270 return;
John McCalle16adc22010-08-04 08:34:44 +0000271 break;
272 }
273
274 // If there's a key function, there may be translation units
275 // that don't have the key function's definition. But ignore
276 // this if we're emitting RTTI under -fno-rtti.
Anders Carlsson265aa7c2011-01-29 20:24:48 +0000277 if (!(TVK != TVK_ForRTTI) || Features.RTTI) {
John McCalle16adc22010-08-04 08:34:44 +0000278 if (Context.getKeyFunction(RD))
279 return;
Anders Carlsson265aa7c2011-01-29 20:24:48 +0000280 }
John McCalle16adc22010-08-04 08:34:44 +0000281
282 // Otherwise, drop the visibility to hidden.
283 GV->setVisibility(llvm::GlobalValue::HiddenVisibility);
Rafael Espindolab1e879c2011-01-11 21:44:37 +0000284 GV->setUnnamedAddr(true);
John McCalle16adc22010-08-04 08:34:44 +0000285}
286
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000287StringRef CodeGenModule::getMangledName(GlobalDecl GD) {
Anders Carlsson2e2f4d22010-06-22 16:05:32 +0000288 const NamedDecl *ND = cast<NamedDecl>(GD.getDecl());
289
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000290 StringRef &Str = MangledDeclNames[GD.getCanonicalDecl()];
Anders Carlsson2e2f4d22010-06-22 16:05:32 +0000291 if (!Str.empty())
292 return Str;
293
John McCall5d865c322010-08-31 07:33:07 +0000294 if (!getCXXABI().getMangleContext().shouldMangleDeclName(ND)) {
Anders Carlsson2e2f4d22010-06-22 16:05:32 +0000295 IdentifierInfo *II = ND->getIdentifier();
296 assert(II && "Attempt to mangle unnamed decl.");
297
298 Str = II->getName();
299 return Str;
300 }
301
302 llvm::SmallString<256> Buffer;
Rafael Espindolaac00f5d2011-02-10 23:59:36 +0000303 llvm::raw_svector_ostream Out(Buffer);
Anders Carlsson2e2f4d22010-06-22 16:05:32 +0000304 if (const CXXConstructorDecl *D = dyn_cast<CXXConstructorDecl>(ND))
Rafael Espindolaac00f5d2011-02-10 23:59:36 +0000305 getCXXABI().getMangleContext().mangleCXXCtor(D, GD.getCtorType(), Out);
Anders Carlsson2e2f4d22010-06-22 16:05:32 +0000306 else if (const CXXDestructorDecl *D = dyn_cast<CXXDestructorDecl>(ND))
Rafael Espindolaac00f5d2011-02-10 23:59:36 +0000307 getCXXABI().getMangleContext().mangleCXXDtor(D, GD.getDtorType(), Out);
Anders Carlsson2e2f4d22010-06-22 16:05:32 +0000308 else if (const BlockDecl *BD = dyn_cast<BlockDecl>(ND))
Rafael Espindolaac00f5d2011-02-10 23:59:36 +0000309 getCXXABI().getMangleContext().mangleBlock(BD, Out);
Anders Carlsson2e2f4d22010-06-22 16:05:32 +0000310 else
Rafael Espindolaac00f5d2011-02-10 23:59:36 +0000311 getCXXABI().getMangleContext().mangleName(ND, Out);
Anders Carlsson2e2f4d22010-06-22 16:05:32 +0000312
313 // Allocate space for the mangled name.
Rafael Espindolaac00f5d2011-02-10 23:59:36 +0000314 Out.flush();
Anders Carlsson2e2f4d22010-06-22 16:05:32 +0000315 size_t Length = Buffer.size();
316 char *Name = MangledNamesAllocator.Allocate<char>(Length);
317 std::copy(Buffer.begin(), Buffer.end(), Name);
318
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000319 Str = StringRef(Name, Length);
Anders Carlsson2e2f4d22010-06-22 16:05:32 +0000320
321 return Str;
322}
323
Peter Collingbourne0ff0b372011-01-13 18:57:25 +0000324void CodeGenModule::getBlockMangledName(GlobalDecl GD, MangleBuffer &Buffer,
325 const BlockDecl *BD) {
326 MangleContext &MangleCtx = getCXXABI().getMangleContext();
327 const Decl *D = GD.getDecl();
Rafael Espindolaac00f5d2011-02-10 23:59:36 +0000328 llvm::raw_svector_ostream Out(Buffer.getBuffer());
Peter Collingbourne0ff0b372011-01-13 18:57:25 +0000329 if (D == 0)
Rafael Espindolaac00f5d2011-02-10 23:59:36 +0000330 MangleCtx.mangleGlobalBlock(BD, Out);
Peter Collingbourne0ff0b372011-01-13 18:57:25 +0000331 else if (const CXXConstructorDecl *CD = dyn_cast<CXXConstructorDecl>(D))
Rafael Espindolaac00f5d2011-02-10 23:59:36 +0000332 MangleCtx.mangleCtorBlock(CD, GD.getCtorType(), BD, Out);
Peter Collingbourne0ff0b372011-01-13 18:57:25 +0000333 else if (const CXXDestructorDecl *DD = dyn_cast<CXXDestructorDecl>(D))
Rafael Espindolaac00f5d2011-02-10 23:59:36 +0000334 MangleCtx.mangleDtorBlock(DD, GD.getDtorType(), BD, Out);
Peter Collingbourne0ff0b372011-01-13 18:57:25 +0000335 else
Rafael Espindolaac00f5d2011-02-10 23:59:36 +0000336 MangleCtx.mangleBlock(cast<DeclContext>(D), BD, Out);
Anders Carlsson635186a2010-06-09 02:36:32 +0000337}
338
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000339llvm::GlobalValue *CodeGenModule::GetGlobalValue(StringRef Name) {
John McCall7ec50432010-03-19 23:29:14 +0000340 return getModule().getNamedValue(Name);
Douglas Gregor5fec5b02009-02-13 00:10:09 +0000341}
342
Chris Lattner5ff2d5d2008-03-14 17:18:18 +0000343/// AddGlobalCtor - Add a function to the list that will be called before
344/// main() runs.
Daniel Dunbar74aa7e12008-08-01 00:01:51 +0000345void CodeGenModule::AddGlobalCtor(llvm::Function * Ctor, int Priority) {
Daniel Dunbardec798b2009-01-13 02:25:00 +0000346 // FIXME: Type coercion of void()* types.
Daniel Dunbar74aa7e12008-08-01 00:01:51 +0000347 GlobalCtors.push_back(std::make_pair(Ctor, Priority));
Chris Lattner5ff2d5d2008-03-14 17:18:18 +0000348}
349
Daniel Dunbar74aa7e12008-08-01 00:01:51 +0000350/// AddGlobalDtor - Add a function to the list that will be called
351/// when the module is unloaded.
352void CodeGenModule::AddGlobalDtor(llvm::Function * Dtor, int Priority) {
Daniel Dunbardec798b2009-01-13 02:25:00 +0000353 // FIXME: Type coercion of void()* types.
Daniel Dunbar74aa7e12008-08-01 00:01:51 +0000354 GlobalDtors.push_back(std::make_pair(Dtor, Priority));
355}
356
357void CodeGenModule::EmitCtorList(const CtorList &Fns, const char *GlobalName) {
358 // Ctor function type is void()*.
John McCall9dc0db22011-05-15 01:53:33 +0000359 llvm::FunctionType* CtorFTy = llvm::FunctionType::get(VoidTy, false);
Owen Anderson9793f0e2009-07-29 22:16:19 +0000360 llvm::Type *CtorPFTy = llvm::PointerType::getUnqual(CtorFTy);
Daniel Dunbar74aa7e12008-08-01 00:01:51 +0000361
362 // Get the type of a ctor entry, { i32, void ()* }.
Chris Lattnere64d7ba2011-06-20 04:01:35 +0000363 llvm::StructType *CtorStructTy =
Chris Lattner845511f2011-06-18 22:49:11 +0000364 llvm::StructType::get(llvm::Type::getInt32Ty(VMContext),
Owen Anderson9793f0e2009-07-29 22:16:19 +0000365 llvm::PointerType::getUnqual(CtorFTy), NULL);
Chris Lattner5ff2d5d2008-03-14 17:18:18 +0000366
Daniel Dunbar74aa7e12008-08-01 00:01:51 +0000367 // Construct the constructor and destructor arrays.
368 std::vector<llvm::Constant*> Ctors;
369 for (CtorList::const_iterator I = Fns.begin(), E = Fns.end(); I != E; ++I) {
370 std::vector<llvm::Constant*> S;
Mike Stump11289f42009-09-09 15:08:12 +0000371 S.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext),
Owen Anderson41a75022009-08-13 21:57:51 +0000372 I->second, false));
Owen Andersonade90fd2009-07-29 18:54:39 +0000373 S.push_back(llvm::ConstantExpr::getBitCast(I->first, CtorPFTy));
Owen Anderson0e0189d2009-07-27 22:29:56 +0000374 Ctors.push_back(llvm::ConstantStruct::get(CtorStructTy, S));
Chris Lattner5ff2d5d2008-03-14 17:18:18 +0000375 }
Daniel Dunbar74aa7e12008-08-01 00:01:51 +0000376
377 if (!Ctors.empty()) {
Owen Anderson9793f0e2009-07-29 22:16:19 +0000378 llvm::ArrayType *AT = llvm::ArrayType::get(CtorStructTy, Ctors.size());
Owen Andersonc10c8d32009-07-08 19:05:04 +0000379 new llvm::GlobalVariable(TheModule, AT, false,
Daniel Dunbar74aa7e12008-08-01 00:01:51 +0000380 llvm::GlobalValue::AppendingLinkage,
Owen Anderson47034e12009-07-28 18:33:04 +0000381 llvm::ConstantArray::get(AT, Ctors),
Owen Andersonc10c8d32009-07-08 19:05:04 +0000382 GlobalName);
Daniel Dunbar74aa7e12008-08-01 00:01:51 +0000383 }
Chris Lattner5ff2d5d2008-03-14 17:18:18 +0000384}
385
John McCalld4324142010-02-19 01:32:20 +0000386llvm::GlobalValue::LinkageTypes
387CodeGenModule::getFunctionLinkage(const FunctionDecl *D) {
Argyrios Kyrtzidisc81af032010-07-29 18:15:58 +0000388 GVALinkage Linkage = getContext().GetGVALinkageForFunction(D);
Daniel Dunbar38932572009-04-14 08:05:55 +0000389
Chris Lattner749b8ed2010-06-30 16:58:07 +0000390 if (Linkage == GVA_Internal)
John McCalld4324142010-02-19 01:32:20 +0000391 return llvm::Function::InternalLinkage;
Chris Lattner749b8ed2010-06-30 16:58:07 +0000392
393 if (D->hasAttr<DLLExportAttr>())
John McCalld4324142010-02-19 01:32:20 +0000394 return llvm::Function::DLLExportLinkage;
Chris Lattner749b8ed2010-06-30 16:58:07 +0000395
396 if (D->hasAttr<WeakAttr>())
John McCalld4324142010-02-19 01:32:20 +0000397 return llvm::Function::WeakAnyLinkage;
Chris Lattner749b8ed2010-06-30 16:58:07 +0000398
399 // In C99 mode, 'inline' functions are guaranteed to have a strong
400 // definition somewhere else, so we can use available_externally linkage.
401 if (Linkage == GVA_C99Inline)
Fariborz Jahanian7cadb2f2011-02-04 00:08:13 +0000402 return llvm::Function::AvailableExternallyLinkage;
Chris Lattner749b8ed2010-06-30 16:58:07 +0000403
404 // In C++, the compiler has to emit a definition in every translation unit
405 // that references the function. We should use linkonce_odr because
406 // a) if all references in this translation unit are optimized away, we
407 // don't need to codegen it. b) if the function persists, it needs to be
408 // merged with other definitions. c) C++ has the ODR, so we know the
409 // definition is dependable.
410 if (Linkage == GVA_CXXInline || Linkage == GVA_TemplateInstantiation)
Fariborz Jahanianf7f04452011-02-04 00:01:24 +0000411 return !Context.getLangOptions().AppleKext
412 ? llvm::Function::LinkOnceODRLinkage
413 : llvm::Function::InternalLinkage;
Chris Lattner749b8ed2010-06-30 16:58:07 +0000414
415 // An explicit instantiation of a template has weak linkage, since
416 // explicit instantiations can occur in multiple translation units
417 // and must all be equivalent. However, we are not allowed to
418 // throw away these explicit instantiations.
419 if (Linkage == GVA_ExplicitTemplateInstantiation)
Fariborz Jahanianf7f04452011-02-04 00:01:24 +0000420 return !Context.getLangOptions().AppleKext
421 ? llvm::Function::WeakODRLinkage
422 : llvm::Function::InternalLinkage;
Chris Lattner749b8ed2010-06-30 16:58:07 +0000423
424 // Otherwise, we have strong external linkage.
425 assert(Linkage == GVA_StrongExternal);
426 return llvm::Function::ExternalLinkage;
John McCalld4324142010-02-19 01:32:20 +0000427}
Nuno Lopesb6f79532008-06-08 15:45:52 +0000428
John McCalld4324142010-02-19 01:32:20 +0000429
430/// SetFunctionDefinitionAttributes - Set attributes for a global.
431///
432/// FIXME: This is currently only done for aliases and functions, but not for
433/// variables (these details are set in EmitGlobalVarDefinition for variables).
434void CodeGenModule::SetFunctionDefinitionAttributes(const FunctionDecl *D,
435 llvm::GlobalValue *GV) {
Daniel Dunbar38932572009-04-14 08:05:55 +0000436 SetCommonAttributes(D, GV);
Nuno Lopesb6f79532008-06-08 15:45:52 +0000437}
438
Daniel Dunbaraeddffc2009-04-14 07:08:30 +0000439void CodeGenModule::SetLLVMFunctionAttributes(const Decl *D,
Mike Stump11289f42009-09-09 15:08:12 +0000440 const CGFunctionInfo &Info,
Daniel Dunbaraeddffc2009-04-14 07:08:30 +0000441 llvm::Function *F) {
Daniel Dunbar0ef34792009-09-12 00:59:20 +0000442 unsigned CallingConv;
Devang Patel322300d2008-09-25 21:02:23 +0000443 AttributeListType AttributeList;
Daniel Dunbar0ef34792009-09-12 00:59:20 +0000444 ConstructAttributeList(Info, D, AttributeList, CallingConv);
Devang Patel322300d2008-09-25 21:02:23 +0000445 F->setAttributes(llvm::AttrListPtr::get(AttributeList.begin(),
Daniel Dunbar0ef34792009-09-12 00:59:20 +0000446 AttributeList.size()));
447 F->setCallingConv(static_cast<llvm::CallingConv::ID>(CallingConv));
Daniel Dunbar449a3392008-09-04 23:41:35 +0000448}
449
Daniel Dunbar38932572009-04-14 08:05:55 +0000450void CodeGenModule::SetLLVMFunctionAttributesForDefinition(const Decl *D,
451 llvm::Function *F) {
Rafael Espindolac1ee12c2011-05-25 03:44:55 +0000452 if (CodeGenOpts.UnwindTables)
453 F->setHasUWTable();
454
Daniel Dunbar0f3403c2009-03-02 04:58:03 +0000455 if (!Features.Exceptions && !Features.ObjCNonFragileABI)
Mike Stump11289f42009-09-09 15:08:12 +0000456 F->addFnAttr(llvm::Attribute::NoUnwind);
Daniel Dunbar03a38442008-10-28 00:17:57 +0000457
Eli Friedmanc55efe42011-08-22 23:55:33 +0000458 if (D->hasAttr<NakedAttr>()) {
459 // Naked implies noinline: we should not be inlining such functions.
Daniel Dunbar8caf6412010-09-29 18:20:25 +0000460 F->addFnAttr(llvm::Attribute::Naked);
Eli Friedmanc55efe42011-08-22 23:55:33 +0000461 F->addFnAttr(llvm::Attribute::NoInline);
462 }
Daniel Dunbar8caf6412010-09-29 18:20:25 +0000463
Mike Stump3722f582009-08-26 22:31:08 +0000464 if (D->hasAttr<NoInlineAttr>())
Anders Carlssonf96954c2009-02-19 19:22:11 +0000465 F->addFnAttr(llvm::Attribute::NoInline);
Mike Stumpc5e153c2009-10-05 21:58:44 +0000466
Eli Friedmanc55efe42011-08-22 23:55:33 +0000467 // (noinline wins over always_inline, and we can't specify both in IR)
468 if (D->hasAttr<AlwaysInlineAttr>() &&
469 !F->hasFnAttr(llvm::Attribute::NoInline))
470 F->addFnAttr(llvm::Attribute::AlwaysInline);
471
Rafael Espindola0ee986c1f2011-01-11 00:26:26 +0000472 if (isa<CXXConstructorDecl>(D) || isa<CXXDestructorDecl>(D))
473 F->setUnnamedAddr(true);
474
Douglas Gregor79a91412011-09-13 17:21:33 +0000475 if (Features.getStackProtector() == LangOptions::SSPOn)
Anders Carlsson0d82fa62009-11-16 16:56:03 +0000476 F->addFnAttr(llvm::Attribute::StackProtect);
Douglas Gregor79a91412011-09-13 17:21:33 +0000477 else if (Features.getStackProtector() == LangOptions::SSPReq)
Anders Carlsson0d82fa62009-11-16 16:56:03 +0000478 F->addFnAttr(llvm::Attribute::StackProtectReq);
479
Alexis Huntdcfba7b2010-08-18 23:23:40 +0000480 unsigned alignment = D->getMaxAlignment() / Context.getCharWidth();
481 if (alignment)
482 F->setAlignment(alignment);
483
Mike Stump3472ae52009-10-05 22:49:20 +0000484 // C++ ABI requires 2-byte alignment for member functions.
Mike Stump916c0062009-10-05 23:08:21 +0000485 if (F->getAlignment() < 2 && isa<CXXMethodDecl>(D))
486 F->setAlignment(2);
Daniel Dunbar449a3392008-09-04 23:41:35 +0000487}
488
Mike Stump11289f42009-09-09 15:08:12 +0000489void CodeGenModule::SetCommonAttributes(const Decl *D,
Daniel Dunbar38932572009-04-14 08:05:55 +0000490 llvm::GlobalValue *GV) {
Anders Carlsson072ef742011-01-29 19:41:00 +0000491 if (const NamedDecl *ND = dyn_cast<NamedDecl>(D))
492 setGlobalVisibility(GV, ND);
John McCall457a04e2010-10-22 21:05:15 +0000493 else
494 GV->setVisibility(llvm::GlobalValue::DefaultVisibility);
Daniel Dunbar38932572009-04-14 08:05:55 +0000495
Argyrios Kyrtzidisb4b64ca2009-06-30 02:34:44 +0000496 if (D->hasAttr<UsedAttr>())
Daniel Dunbar38932572009-04-14 08:05:55 +0000497 AddUsedGlobal(GV);
498
Argyrios Kyrtzidisb4b64ca2009-06-30 02:34:44 +0000499 if (const SectionAttr *SA = D->getAttr<SectionAttr>())
Daniel Dunbar38932572009-04-14 08:05:55 +0000500 GV->setSection(SA->getName());
Anton Korobeynikov55bcea12010-01-10 12:58:08 +0000501
502 getTargetCodeGenInfo().SetTargetAttributes(D, GV, *this);
Daniel Dunbar38932572009-04-14 08:05:55 +0000503}
504
Daniel Dunbarc3e7cff2009-04-17 00:48:04 +0000505void CodeGenModule::SetInternalFunctionAttributes(const Decl *D,
506 llvm::Function *F,
507 const CGFunctionInfo &FI) {
508 SetLLVMFunctionAttributes(D, FI, F);
509 SetLLVMFunctionAttributesForDefinition(D, F);
Daniel Dunbar38932572009-04-14 08:05:55 +0000510
511 F->setLinkage(llvm::Function::InternalLinkage);
512
Daniel Dunbarc3e7cff2009-04-17 00:48:04 +0000513 SetCommonAttributes(D, F);
Daniel Dunbar449a3392008-09-04 23:41:35 +0000514}
515
Anders Carlsson6710c532010-02-06 02:44:09 +0000516void CodeGenModule::SetFunctionAttributes(GlobalDecl GD,
Eli Friedman895771a2009-05-26 01:22:57 +0000517 llvm::Function *F,
518 bool IsIncompleteFunction) {
Peter Collingbourneeafa4e42011-04-06 12:29:04 +0000519 if (unsigned IID = F->getIntrinsicID()) {
520 // If this is an intrinsic function, set the function's attributes
521 // to the intrinsic's attributes.
522 F->setAttributes(llvm::Intrinsic::getAttributes((llvm::Intrinsic::ID)IID));
523 return;
524 }
525
Anders Carlsson6710c532010-02-06 02:44:09 +0000526 const FunctionDecl *FD = cast<FunctionDecl>(GD.getDecl());
527
Eli Friedman895771a2009-05-26 01:22:57 +0000528 if (!IsIncompleteFunction)
Anders Carlsson6710c532010-02-06 02:44:09 +0000529 SetLLVMFunctionAttributes(FD, getTypes().getFunctionInfo(GD), F);
Mike Stump11289f42009-09-09 15:08:12 +0000530
Daniel Dunbar38932572009-04-14 08:05:55 +0000531 // Only a few attributes are set on declarations; these may later be
532 // overridden by a definition.
Mike Stump11289f42009-09-09 15:08:12 +0000533
Argyrios Kyrtzidisb4b64ca2009-06-30 02:34:44 +0000534 if (FD->hasAttr<DLLImportAttr>()) {
Daniel Dunbar38932572009-04-14 08:05:55 +0000535 F->setLinkage(llvm::Function::DLLImportLinkage);
Mike Stump11289f42009-09-09 15:08:12 +0000536 } else if (FD->hasAttr<WeakAttr>() ||
Douglas Gregor20b2ebd2011-03-23 00:50:03 +0000537 FD->isWeakImported()) {
Daniel Dunbar38932572009-04-14 08:05:55 +0000538 // "extern_weak" is overloaded in LLVM; we probably should have
Mike Stump11289f42009-09-09 15:08:12 +0000539 // separate linkage types for this.
Daniel Dunbar38932572009-04-14 08:05:55 +0000540 F->setLinkage(llvm::Function::ExternalWeakLinkage);
541 } else {
Mike Stump11289f42009-09-09 15:08:12 +0000542 F->setLinkage(llvm::Function::ExternalLinkage);
John McCallc273f242010-10-30 11:50:40 +0000543
544 NamedDecl::LinkageInfo LV = FD->getLinkageAndVisibility();
545 if (LV.linkage() == ExternalLinkage && LV.visibilityExplicit()) {
546 F->setVisibility(GetLLVMVisibility(LV.visibility()));
547 }
Daniel Dunbar38932572009-04-14 08:05:55 +0000548 }
549
Argyrios Kyrtzidisb4b64ca2009-06-30 02:34:44 +0000550 if (const SectionAttr *SA = FD->getAttr<SectionAttr>())
Daniel Dunbar38932572009-04-14 08:05:55 +0000551 F->setSection(SA->getName());
Daniel Dunbar0beedc12008-09-08 23:44:31 +0000552}
553
Daniel Dunbar08b26a02009-02-13 20:29:50 +0000554void CodeGenModule::AddUsedGlobal(llvm::GlobalValue *GV) {
Mike Stump11289f42009-09-09 15:08:12 +0000555 assert(!GV->isDeclaration() &&
Daniel Dunbar08b26a02009-02-13 20:29:50 +0000556 "Only globals with definition can force usage.");
Chris Lattnerf41e87f2009-03-31 22:37:52 +0000557 LLVMUsed.push_back(GV);
Daniel Dunbar08b26a02009-02-13 20:29:50 +0000558}
559
560void CodeGenModule::EmitLLVMUsed() {
561 // Don't create llvm.used if there is no need.
Chris Lattnerf56501c2009-07-17 23:57:13 +0000562 if (LLVMUsed.empty())
Daniel Dunbar08b26a02009-02-13 20:29:50 +0000563 return;
564
Chris Lattner2192fe52011-07-18 04:24:23 +0000565 llvm::Type *i8PTy = llvm::Type::getInt8PtrTy(VMContext);
Mike Stump11289f42009-09-09 15:08:12 +0000566
Chris Lattnerf41e87f2009-03-31 22:37:52 +0000567 // Convert LLVMUsed to what ConstantArray needs.
568 std::vector<llvm::Constant*> UsedArray;
569 UsedArray.resize(LLVMUsed.size());
570 for (unsigned i = 0, e = LLVMUsed.size(); i != e; ++i) {
Mike Stump11289f42009-09-09 15:08:12 +0000571 UsedArray[i] =
572 llvm::ConstantExpr::getBitCast(cast<llvm::Constant>(&*LLVMUsed[i]),
Owen Anderson170229f2009-07-14 23:10:40 +0000573 i8PTy);
Chris Lattnerf41e87f2009-03-31 22:37:52 +0000574 }
Mike Stump11289f42009-09-09 15:08:12 +0000575
Fariborz Jahanian248c7192009-06-23 21:47:46 +0000576 if (UsedArray.empty())
577 return;
Owen Anderson9793f0e2009-07-29 22:16:19 +0000578 llvm::ArrayType *ATy = llvm::ArrayType::get(i8PTy, UsedArray.size());
Mike Stump11289f42009-09-09 15:08:12 +0000579
580 llvm::GlobalVariable *GV =
581 new llvm::GlobalVariable(getModule(), ATy, false,
Daniel Dunbar08b26a02009-02-13 20:29:50 +0000582 llvm::GlobalValue::AppendingLinkage,
Owen Anderson47034e12009-07-28 18:33:04 +0000583 llvm::ConstantArray::get(ATy, UsedArray),
Owen Andersonc10c8d32009-07-08 19:05:04 +0000584 "llvm.used");
Daniel Dunbar08b26a02009-02-13 20:29:50 +0000585
586 GV->setSection("llvm.metadata");
587}
588
589void CodeGenModule::EmitDeferred() {
Chris Lattner45470942009-03-21 09:44:56 +0000590 // Emit code for any potentially referenced deferred decls. Since a
591 // previously unused static decl may become used during the generation of code
Nick Lewycky26da4dd2011-07-18 05:26:13 +0000592 // for a static function, iterate until no changes are made.
Rafael Espindolae7113ca2010-03-10 02:19:29 +0000593
Anders Carlsson11e51402010-04-17 20:15:18 +0000594 while (!DeferredDeclsToEmit.empty() || !DeferredVTables.empty()) {
595 if (!DeferredVTables.empty()) {
596 const CXXRecordDecl *RD = DeferredVTables.back();
597 DeferredVTables.pop_back();
598 getVTables().GenerateClassData(getVTableLinkage(RD), RD);
Rafael Espindolae7113ca2010-03-10 02:19:29 +0000599 continue;
600 }
601
Anders Carlssondae1abc2009-05-05 04:44:02 +0000602 GlobalDecl D = DeferredDeclsToEmit.back();
Chris Lattner45470942009-03-21 09:44:56 +0000603 DeferredDeclsToEmit.pop_back();
604
John McCallc2af9392010-05-27 01:45:30 +0000605 // Check to see if we've already emitted this. This is necessary
606 // for a couple of reasons: first, decls can end up in the
607 // deferred-decls queue multiple times, and second, decls can end
608 // up with definitions in unusual ways (e.g. by an extern inline
609 // function acquiring a strong function redefinition). Just
610 // ignore these cases.
611 //
612 // TODO: That said, looking this up multiple times is very wasteful.
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000613 StringRef Name = getMangledName(D);
John McCall7ec50432010-03-19 23:29:14 +0000614 llvm::GlobalValue *CGRef = GetGlobalValue(Name);
Chris Lattner45470942009-03-21 09:44:56 +0000615 assert(CGRef && "Deferred decl wasn't referenced?");
Mike Stump11289f42009-09-09 15:08:12 +0000616
Chris Lattner45470942009-03-21 09:44:56 +0000617 if (!CGRef->isDeclaration())
618 continue;
Mike Stump11289f42009-09-09 15:08:12 +0000619
John McCallc2af9392010-05-27 01:45:30 +0000620 // GlobalAlias::isDeclaration() defers to the aliasee, but for our
621 // purposes an alias counts as a definition.
622 if (isa<llvm::GlobalAlias>(CGRef))
623 continue;
624
Chris Lattner45470942009-03-21 09:44:56 +0000625 // Otherwise, emit the definition and move on to the next one.
626 EmitGlobalDefinition(D);
627 }
Chris Lattnerbed31442007-05-28 01:07:47 +0000628}
Chris Lattner09153c02007-06-22 18:48:09 +0000629
Julien Lerouge5a6b6982011-09-09 22:41:49 +0000630void CodeGenModule::EmitGlobalAnnotations() {
631 if (Annotations.empty())
632 return;
633
634 // Create a new global variable for the ConstantStruct in the Module.
635 llvm::Constant *Array = llvm::ConstantArray::get(llvm::ArrayType::get(
636 Annotations[0]->getType(), Annotations.size()), Annotations);
637 llvm::GlobalValue *gv = new llvm::GlobalVariable(getModule(),
638 Array->getType(), false, llvm::GlobalValue::AppendingLinkage, Array,
639 "llvm.global.annotations");
640 gv->setSection(AnnotationSection);
641}
642
643llvm::Constant *CodeGenModule::EmitAnnotationString(llvm::StringRef Str) {
644 llvm::StringMap<llvm::Constant*>::iterator i = AnnotationStrings.find(Str);
645 if (i != AnnotationStrings.end())
646 return i->second;
647
648 // Not found yet, create a new global.
649 llvm::Constant *s = llvm::ConstantArray::get(getLLVMContext(), Str, true);
650 llvm::GlobalValue *gv = new llvm::GlobalVariable(getModule(), s->getType(),
651 true, llvm::GlobalValue::PrivateLinkage, s, ".str");
652 gv->setSection(AnnotationSection);
653 gv->setUnnamedAddr(true);
654 AnnotationStrings[Str] = gv;
655 return gv;
656}
657
658llvm::Constant *CodeGenModule::EmitAnnotationUnit(SourceLocation Loc) {
659 SourceManager &SM = getContext().getSourceManager();
660 PresumedLoc PLoc = SM.getPresumedLoc(Loc);
661 if (PLoc.isValid())
662 return EmitAnnotationString(PLoc.getFilename());
663 return EmitAnnotationString(SM.getBufferName(Loc));
664}
665
666llvm::Constant *CodeGenModule::EmitAnnotationLineNo(SourceLocation L) {
667 SourceManager &SM = getContext().getSourceManager();
668 PresumedLoc PLoc = SM.getPresumedLoc(L);
669 unsigned LineNo = PLoc.isValid() ? PLoc.getLine() :
670 SM.getExpansionLineNumber(L);
671 return llvm::ConstantInt::get(Int32Ty, LineNo);
672}
673
Mike Stump11289f42009-09-09 15:08:12 +0000674llvm::Constant *CodeGenModule::EmitAnnotateAttr(llvm::GlobalValue *GV,
Nate Begemanfaae0812008-04-19 04:17:09 +0000675 const AnnotateAttr *AA,
Julien Lerouge5a6b6982011-09-09 22:41:49 +0000676 SourceLocation L) {
677 // Get the globals for file name, annotation, and the line number.
678 llvm::Constant *AnnoGV = EmitAnnotationString(AA->getAnnotation()),
679 *UnitGV = EmitAnnotationUnit(L),
680 *LineNoCst = EmitAnnotationLineNo(L);
Nate Begemanfaae0812008-04-19 04:17:09 +0000681
Daniel Dunbar346892a2009-04-14 22:41:13 +0000682 // Create the ConstantStruct for the global annotation.
Nate Begemanfaae0812008-04-19 04:17:09 +0000683 llvm::Constant *Fields[4] = {
Julien Lerouge5a6b6982011-09-09 22:41:49 +0000684 llvm::ConstantExpr::getBitCast(GV, Int8PtrTy),
685 llvm::ConstantExpr::getBitCast(AnnoGV, Int8PtrTy),
686 llvm::ConstantExpr::getBitCast(UnitGV, Int8PtrTy),
687 LineNoCst
Nate Begemanfaae0812008-04-19 04:17:09 +0000688 };
Chris Lattnere64d7ba2011-06-20 04:01:35 +0000689 return llvm::ConstantStruct::getAnon(Fields);
Nate Begemanfaae0812008-04-19 04:17:09 +0000690}
691
Julien Lerouge5a6b6982011-09-09 22:41:49 +0000692void CodeGenModule::AddGlobalAnnotations(const ValueDecl *D,
693 llvm::GlobalValue *GV) {
694 assert(D->hasAttr<AnnotateAttr>() && "no annotate attribute");
695 // Get the struct elements for these annotations.
696 for (specific_attr_iterator<AnnotateAttr>
697 ai = D->specific_attr_begin<AnnotateAttr>(),
698 ae = D->specific_attr_end<AnnotateAttr>(); ai != ae; ++ai)
699 Annotations.push_back(EmitAnnotateAttr(GV, *ai, D->getLocation()));
700}
701
Argyrios Kyrtzidisc0279a92010-07-27 22:37:14 +0000702bool CodeGenModule::MayDeferGeneration(const ValueDecl *Global) {
Argyrios Kyrtzidisc81af032010-07-29 18:15:58 +0000703 // Never defer when EmitAllDecls is specified.
704 if (Features.EmitAllDecls)
Daniel Dunbar6b8720e2009-02-13 21:18:01 +0000705 return false;
Daniel Dunbar9c426522008-07-29 23:18:29 +0000706
Argyrios Kyrtzidisc9049332010-07-29 20:08:05 +0000707 return !getContext().DeclMustBeEmitted(Global);
Daniel Dunbar6b8720e2009-02-13 21:18:01 +0000708}
709
Rafael Espindola2e42fec2010-03-04 18:17:24 +0000710llvm::Constant *CodeGenModule::GetWeakRefReference(const ValueDecl *VD) {
711 const AliasAttr *AA = VD->getAttr<AliasAttr>();
712 assert(AA && "No alias?");
713
Chris Lattner2192fe52011-07-18 04:24:23 +0000714 llvm::Type *DeclTy = getTypes().ConvertTypeForMem(VD->getType());
Rafael Espindola2e42fec2010-03-04 18:17:24 +0000715
Rafael Espindola2e42fec2010-03-04 18:17:24 +0000716 // See if there is already something with the target's name in the module.
John McCall7ec50432010-03-19 23:29:14 +0000717 llvm::GlobalValue *Entry = GetGlobalValue(AA->getAliasee());
Rafael Espindola2e42fec2010-03-04 18:17:24 +0000718
719 llvm::Constant *Aliasee;
720 if (isa<llvm::FunctionType>(DeclTy))
Anders Carlsson3c239482011-02-05 04:35:53 +0000721 Aliasee = GetOrCreateLLVMFunction(AA->getAliasee(), DeclTy, GlobalDecl(),
722 /*ForVTable=*/false);
Rafael Espindola2e42fec2010-03-04 18:17:24 +0000723 else
John McCall7ec50432010-03-19 23:29:14 +0000724 Aliasee = GetOrCreateLLVMGlobal(AA->getAliasee(),
Rafael Espindola2e42fec2010-03-04 18:17:24 +0000725 llvm::PointerType::getUnqual(DeclTy), 0);
726 if (!Entry) {
727 llvm::GlobalValue* F = cast<llvm::GlobalValue>(Aliasee);
728 F->setLinkage(llvm::Function::ExternalWeakLinkage);
729 WeakRefReferences.insert(F);
730 }
731
732 return Aliasee;
733}
734
Chris Lattnere0be0df2009-05-12 21:21:08 +0000735void CodeGenModule::EmitGlobal(GlobalDecl GD) {
Anders Carlsson38988d72009-09-10 23:38:47 +0000736 const ValueDecl *Global = cast<ValueDecl>(GD.getDecl());
Mike Stump11289f42009-09-09 15:08:12 +0000737
Rafael Espindola2e42fec2010-03-04 18:17:24 +0000738 // Weak references don't produce any output by themselves.
739 if (Global->hasAttr<WeakRefAttr>())
740 return;
741
Chris Lattner54041692009-03-22 21:47:11 +0000742 // If this is an alias definition (which otherwise looks like a declaration)
743 // emit it now.
Argyrios Kyrtzidisb4b64ca2009-06-30 02:34:44 +0000744 if (Global->hasAttr<AliasAttr>())
John McCall7ec50432010-03-19 23:29:14 +0000745 return EmitAliasDefinition(GD);
Daniel Dunbar0beedc12008-09-08 23:44:31 +0000746
Chris Lattner45470942009-03-21 09:44:56 +0000747 // Ignore declarations, they will be emitted on their first use.
Daniel Dunbar4e004ed2009-03-19 08:27:24 +0000748 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Global)) {
Daniel Dunbar6b8720e2009-02-13 21:18:01 +0000749 // Forward declarations are emitted lazily on first use.
Nick Lewycky26da4dd2011-07-18 05:26:13 +0000750 if (!FD->doesThisDeclarationHaveABody()) {
751 if (!FD->doesDeclarationForceExternallyVisibleDefinition())
752 return;
753
754 const FunctionDecl *InlineDefinition = 0;
755 FD->getBody(InlineDefinition);
756
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000757 StringRef MangledName = getMangledName(GD);
Nick Lewycky26da4dd2011-07-18 05:26:13 +0000758 llvm::StringMap<GlobalDecl>::iterator DDI =
759 DeferredDecls.find(MangledName);
760 if (DDI != DeferredDecls.end())
761 DeferredDecls.erase(DDI);
762 EmitGlobalDefinition(InlineDefinition);
Daniel Dunbar6b8720e2009-02-13 21:18:01 +0000763 return;
Nick Lewycky26da4dd2011-07-18 05:26:13 +0000764 }
Daniel Dunbar08b26a02009-02-13 20:29:50 +0000765 } else {
766 const VarDecl *VD = cast<VarDecl>(Global);
Daniel Dunbar9c426522008-07-29 23:18:29 +0000767 assert(VD->isFileVarDecl() && "Cannot emit local var decl as global.");
768
Douglas Gregor61f6db52010-02-06 05:15:45 +0000769 if (VD->isThisDeclarationADefinition() != VarDecl::Definition)
Douglas Gregorbeecd582009-04-21 17:11:58 +0000770 return;
Nate Begeman8e8d4982008-04-20 06:29:50 +0000771 }
772
Chris Lattner45470942009-03-21 09:44:56 +0000773 // Defer code generation when possible if this is a static definition, inline
774 // function etc. These we only want to emit if they are used.
Chris Lattner7a4a29f2010-04-13 17:39:09 +0000775 if (!MayDeferGeneration(Global)) {
776 // Emit the definition if it can't be deferred.
777 EmitGlobalDefinition(GD);
Daniel Dunbar9c426522008-07-29 23:18:29 +0000778 return;
779 }
John McCall70013b62010-07-15 23:40:35 +0000780
781 // If we're deferring emission of a C++ variable with an
782 // initializer, remember the order in which it appeared in the file.
783 if (getLangOptions().CPlusPlus && isa<VarDecl>(Global) &&
784 cast<VarDecl>(Global)->hasInit()) {
785 DelayedCXXInitPosition[Global] = CXXGlobalInits.size();
786 CXXGlobalInits.push_back(0);
787 }
Chris Lattner7a4a29f2010-04-13 17:39:09 +0000788
789 // If the value has already been used, add it directly to the
790 // DeferredDeclsToEmit list.
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000791 StringRef MangledName = getMangledName(GD);
Chris Lattner7a4a29f2010-04-13 17:39:09 +0000792 if (GetGlobalValue(MangledName))
793 DeferredDeclsToEmit.push_back(GD);
794 else {
795 // Otherwise, remember that we saw a deferred decl with this name. The
796 // first use of the mangled name will cause it to move into
797 // DeferredDeclsToEmit.
798 DeferredDecls[MangledName] = GD;
799 }
Nate Begeman8e8d4982008-04-20 06:29:50 +0000800}
801
Chris Lattnere0be0df2009-05-12 21:21:08 +0000802void CodeGenModule::EmitGlobalDefinition(GlobalDecl GD) {
Anders Carlsson38988d72009-09-10 23:38:47 +0000803 const ValueDecl *D = cast<ValueDecl>(GD.getDecl());
Mike Stump11289f42009-09-09 15:08:12 +0000804
Dan Gohman145f3f12010-04-19 16:39:44 +0000805 PrettyStackTraceDecl CrashInfo(const_cast<ValueDecl *>(D), D->getLocation(),
Anders Carlsson29295bf2009-10-27 14:32:27 +0000806 Context.getSourceManager(),
807 "Generating code for declaration");
808
Douglas Gregora700f682010-07-13 06:02:28 +0000809 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
810 // At -O0, don't generate IR for functions with available_externally
811 // linkage.
Douglas Gregor89976902010-07-15 22:58:18 +0000812 if (CodeGenOpts.OptimizationLevel == 0 &&
813 !Function->hasAttr<AlwaysInlineAttr>() &&
Douglas Gregora700f682010-07-13 06:02:28 +0000814 getFunctionLinkage(Function)
815 == llvm::Function::AvailableExternallyLinkage)
816 return;
Anders Carlssonaf82f632010-03-23 04:31:31 +0000817
Douglas Gregora700f682010-07-13 06:02:28 +0000818 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(D)) {
Eli Friedman49a94b12011-05-06 17:27:27 +0000819 // Make sure to emit the definition(s) before we emit the thunks.
820 // This is necessary for the generation of certain thunks.
821 if (const CXXConstructorDecl *CD = dyn_cast<CXXConstructorDecl>(Method))
822 EmitCXXConstructor(CD, GD.getCtorType());
823 else if (const CXXDestructorDecl *DD =dyn_cast<CXXDestructorDecl>(Method))
824 EmitCXXDestructor(DD, GD.getDtorType());
825 else
826 EmitGlobalFunctionDefinition(GD);
827
Douglas Gregora700f682010-07-13 06:02:28 +0000828 if (Method->isVirtual())
829 getVTables().EmitThunks(GD);
830
Eli Friedman49a94b12011-05-06 17:27:27 +0000831 return;
Douglas Gregora700f682010-07-13 06:02:28 +0000832 }
Chris Lattnerd8d760c2010-04-13 17:57:11 +0000833
Chris Lattnerd8d760c2010-04-13 17:57:11 +0000834 return EmitGlobalFunctionDefinition(GD);
Douglas Gregora700f682010-07-13 06:02:28 +0000835 }
Chris Lattnerd8d760c2010-04-13 17:57:11 +0000836
837 if (const VarDecl *VD = dyn_cast<VarDecl>(D))
838 return EmitGlobalVarDefinition(VD);
Chris Lattner7a4a29f2010-04-13 17:39:09 +0000839
840 assert(0 && "Invalid argument to EmitGlobalDefinition()");
Daniel Dunbar9c426522008-07-29 23:18:29 +0000841}
842
Chris Lattnerd4808922009-03-22 21:03:39 +0000843/// GetOrCreateLLVMFunction - If the specified mangled name is not in the
844/// module, create and return an llvm Function with the specified type. If there
845/// is something in the module with the specified name, return it potentially
846/// bitcasted to the right type.
847///
848/// If D is non-null, it specifies a decl that correspond to this. This is used
849/// to set the attributes on the function when it is first created.
John McCall7ec50432010-03-19 23:29:14 +0000850llvm::Constant *
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000851CodeGenModule::GetOrCreateLLVMFunction(StringRef MangledName,
Chris Lattner2192fe52011-07-18 04:24:23 +0000852 llvm::Type *Ty,
John McCall31168b02011-06-15 23:02:42 +0000853 GlobalDecl D, bool ForVTable,
854 llvm::Attributes ExtraAttrs) {
Chris Lattnera85d68e2009-03-21 09:25:43 +0000855 // Lookup the entry, lazily creating it if necessary.
John McCall7ec50432010-03-19 23:29:14 +0000856 llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
Chris Lattnera85d68e2009-03-21 09:25:43 +0000857 if (Entry) {
Rafael Espindola2e42fec2010-03-04 18:17:24 +0000858 if (WeakRefReferences.count(Entry)) {
859 const FunctionDecl *FD = cast_or_null<FunctionDecl>(D.getDecl());
860 if (FD && !FD->hasAttr<WeakAttr>())
Anders Carlssonaf82f632010-03-23 04:31:31 +0000861 Entry->setLinkage(llvm::Function::ExternalLinkage);
Rafael Espindola2e42fec2010-03-04 18:17:24 +0000862
863 WeakRefReferences.erase(Entry);
864 }
865
Chris Lattnera85d68e2009-03-21 09:25:43 +0000866 if (Entry->getType()->getElementType() == Ty)
867 return Entry;
Mike Stump11289f42009-09-09 15:08:12 +0000868
Chris Lattnera85d68e2009-03-21 09:25:43 +0000869 // Make sure the result is of the correct type.
Chris Lattnera5f58b02011-07-09 17:41:47 +0000870 return llvm::ConstantExpr::getBitCast(Entry, Ty->getPointerTo());
Chris Lattnera85d68e2009-03-21 09:25:43 +0000871 }
Mike Stump11289f42009-09-09 15:08:12 +0000872
Eli Friedmancc522d92009-11-09 05:07:37 +0000873 // This function doesn't have a complete type (for example, the return
874 // type is an incomplete struct). Use a fake type instead, and make
875 // sure not to try to set attributes.
876 bool IsIncompleteFunction = false;
John McCalld06fb862010-04-28 00:00:30 +0000877
Chris Lattner2192fe52011-07-18 04:24:23 +0000878 llvm::FunctionType *FTy;
John McCalld06fb862010-04-28 00:00:30 +0000879 if (isa<llvm::FunctionType>(Ty)) {
880 FTy = cast<llvm::FunctionType>(Ty);
881 } else {
John McCall9dc0db22011-05-15 01:53:33 +0000882 FTy = llvm::FunctionType::get(VoidTy, false);
Eli Friedmancc522d92009-11-09 05:07:37 +0000883 IsIncompleteFunction = true;
884 }
Chris Lattner5c740f12010-06-30 19:14:05 +0000885
John McCalld06fb862010-04-28 00:00:30 +0000886 llvm::Function *F = llvm::Function::Create(FTy,
Eli Friedmancc522d92009-11-09 05:07:37 +0000887 llvm::Function::ExternalLinkage,
John McCall7ec50432010-03-19 23:29:14 +0000888 MangledName, &getModule());
889 assert(F->getName() == MangledName && "name was uniqued!");
Eli Friedmancc522d92009-11-09 05:07:37 +0000890 if (D.getDecl())
Anders Carlsson6710c532010-02-06 02:44:09 +0000891 SetFunctionAttributes(D, F, IsIncompleteFunction);
John McCall31168b02011-06-15 23:02:42 +0000892 if (ExtraAttrs != llvm::Attribute::None)
893 F->addFnAttr(ExtraAttrs);
Eli Friedmancc522d92009-11-09 05:07:37 +0000894
Chris Lattner45470942009-03-21 09:44:56 +0000895 // This is the first use or definition of a mangled name. If there is a
896 // deferred decl with this name, remember that we need to emit it at the end
897 // of the file.
John McCall7ec50432010-03-19 23:29:14 +0000898 llvm::StringMap<GlobalDecl>::iterator DDI = DeferredDecls.find(MangledName);
Chris Lattner45470942009-03-21 09:44:56 +0000899 if (DDI != DeferredDecls.end()) {
900 // Move the potentially referenced deferred decl to the DeferredDeclsToEmit
901 // list, and remove it from DeferredDecls (since we don't need it anymore).
902 DeferredDeclsToEmit.push_back(DDI->second);
903 DeferredDecls.erase(DDI);
John McCall357d0f32010-12-15 04:00:32 +0000904
905 // Otherwise, there are cases we have to worry about where we're
906 // using a declaration for which we must emit a definition but where
907 // we might not find a top-level definition:
908 // - member functions defined inline in their classes
909 // - friend functions defined inline in some class
910 // - special member functions with implicit definitions
911 // If we ever change our AST traversal to walk into class methods,
912 // this will be unnecessary.
Anders Carlsson3c239482011-02-05 04:35:53 +0000913 //
914 // We also don't emit a definition for a function if it's going to be an entry
915 // in a vtable, unless it's already marked as used.
Rafael Espindolaaf5b92e2011-02-03 06:30:58 +0000916 } else if (getLangOptions().CPlusPlus && D.getDecl()) {
John McCall357d0f32010-12-15 04:00:32 +0000917 // Look for a declaration that's lexically in a record.
918 const FunctionDecl *FD = cast<FunctionDecl>(D.getDecl());
919 do {
920 if (isa<CXXRecordDecl>(FD->getLexicalDeclContext())) {
Anders Carlsson3c239482011-02-05 04:35:53 +0000921 if (FD->isImplicit() && !ForVTable) {
John McCall357d0f32010-12-15 04:00:32 +0000922 assert(FD->isUsed() && "Sema didn't mark implicit function as used!");
Douglas Gregor25756312011-02-09 02:03:05 +0000923 DeferredDeclsToEmit.push_back(D.getWithDecl(FD));
John McCall357d0f32010-12-15 04:00:32 +0000924 break;
Alexis Hunt4a8ea102011-05-06 20:44:56 +0000925 } else if (FD->doesThisDeclarationHaveABody()) {
Douglas Gregor25756312011-02-09 02:03:05 +0000926 DeferredDeclsToEmit.push_back(D.getWithDecl(FD));
John McCall357d0f32010-12-15 04:00:32 +0000927 break;
928 }
Rafael Espindola70e040d2010-03-02 21:28:26 +0000929 }
John McCall357d0f32010-12-15 04:00:32 +0000930 FD = FD->getPreviousDeclaration();
931 } while (FD);
Chris Lattner45470942009-03-21 09:44:56 +0000932 }
Mike Stump11289f42009-09-09 15:08:12 +0000933
John McCalld06fb862010-04-28 00:00:30 +0000934 // Make sure the result is of the requested type.
935 if (!IsIncompleteFunction) {
936 assert(F->getType()->getElementType() == Ty);
937 return F;
938 }
939
Chris Lattnera5f58b02011-07-09 17:41:47 +0000940 llvm::Type *PTy = llvm::PointerType::getUnqual(Ty);
John McCalld06fb862010-04-28 00:00:30 +0000941 return llvm::ConstantExpr::getBitCast(F, PTy);
Chris Lattnera85d68e2009-03-21 09:25:43 +0000942}
943
Chris Lattnerd4808922009-03-22 21:03:39 +0000944/// GetAddrOfFunction - Return the address of the given function. If Ty is
945/// non-null, then this function will use the specified type if it has to
946/// create it (this occurs when we see a definition of the function).
Chris Lattnere0be0df2009-05-12 21:21:08 +0000947llvm::Constant *CodeGenModule::GetAddrOfFunction(GlobalDecl GD,
Chris Lattner2192fe52011-07-18 04:24:23 +0000948 llvm::Type *Ty,
Anders Carlsson3c239482011-02-05 04:35:53 +0000949 bool ForVTable) {
Chris Lattnerd4808922009-03-22 21:03:39 +0000950 // If there was no specific requested type, just convert it now.
951 if (!Ty)
Anders Carlsson38988d72009-09-10 23:38:47 +0000952 Ty = getTypes().ConvertType(cast<ValueDecl>(GD.getDecl())->getType());
Chris Lattner5c740f12010-06-30 19:14:05 +0000953
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000954 StringRef MangledName = getMangledName(GD);
Anders Carlsson3c239482011-02-05 04:35:53 +0000955 return GetOrCreateLLVMFunction(MangledName, Ty, GD, ForVTable);
Chris Lattnerd4808922009-03-22 21:03:39 +0000956}
Eli Friedmanc18d9d52008-05-30 19:50:47 +0000957
Chris Lattnerd4808922009-03-22 21:03:39 +0000958/// CreateRuntimeFunction - Create a new runtime function with the specified
959/// type and name.
960llvm::Constant *
Chris Lattner2192fe52011-07-18 04:24:23 +0000961CodeGenModule::CreateRuntimeFunction(llvm::FunctionType *FTy,
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000962 StringRef Name,
John McCall31168b02011-06-15 23:02:42 +0000963 llvm::Attributes ExtraAttrs) {
964 return GetOrCreateLLVMFunction(Name, FTy, GlobalDecl(), /*ForVTable=*/false,
965 ExtraAttrs);
Chris Lattnerd4808922009-03-22 21:03:39 +0000966}
Daniel Dunbar9c426522008-07-29 23:18:29 +0000967
Douglas Gregor2d1c0552011-05-07 22:06:45 +0000968static bool DeclIsConstantGlobal(ASTContext &Context, const VarDecl *D,
969 bool ConstantInit) {
John McCall340aafa2010-02-08 21:46:50 +0000970 if (!D->getType().isConstant(Context) && !D->getType()->isReferenceType())
Eli Friedmanb095e152009-12-11 21:23:03 +0000971 return false;
Douglas Gregor2d1c0552011-05-07 22:06:45 +0000972
973 if (Context.getLangOptions().CPlusPlus) {
974 if (const RecordType *Record
975 = Context.getBaseElementType(D->getType())->getAs<RecordType>())
Douglas Gregor61226d32011-05-13 01:05:07 +0000976 return ConstantInit &&
977 cast<CXXRecordDecl>(Record->getDecl())->isPOD() &&
978 !cast<CXXRecordDecl>(Record->getDecl())->hasMutableFields();
Eli Friedmanb095e152009-12-11 21:23:03 +0000979 }
Douglas Gregor2d1c0552011-05-07 22:06:45 +0000980
Eli Friedmanb095e152009-12-11 21:23:03 +0000981 return true;
982}
983
Chris Lattnerd4808922009-03-22 21:03:39 +0000984/// GetOrCreateLLVMGlobal - If the specified mangled name is not in the module,
985/// create and return an llvm GlobalVariable with the specified type. If there
986/// is something in the module with the specified name, return it potentially
987/// bitcasted to the right type.
988///
989/// If D is non-null, it specifies a decl that correspond to this. This is used
990/// to set the attributes on the global when it is first created.
John McCall7ec50432010-03-19 23:29:14 +0000991llvm::Constant *
Chris Lattner0e62c1c2011-07-23 10:55:15 +0000992CodeGenModule::GetOrCreateLLVMGlobal(StringRef MangledName,
Chris Lattner2192fe52011-07-18 04:24:23 +0000993 llvm::PointerType *Ty,
Rafael Espindolad661a852011-01-18 21:07:57 +0000994 const VarDecl *D,
995 bool UnnamedAddr) {
Daniel Dunbar829e9882008-08-05 23:31:02 +0000996 // Lookup the entry, lazily creating it if necessary.
John McCall7ec50432010-03-19 23:29:14 +0000997 llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
Chris Lattner3bfce182009-03-21 08:03:33 +0000998 if (Entry) {
Rafael Espindola2e42fec2010-03-04 18:17:24 +0000999 if (WeakRefReferences.count(Entry)) {
1000 if (D && !D->hasAttr<WeakAttr>())
Anders Carlssonaf82f632010-03-23 04:31:31 +00001001 Entry->setLinkage(llvm::Function::ExternalLinkage);
Rafael Espindola2e42fec2010-03-04 18:17:24 +00001002
1003 WeakRefReferences.erase(Entry);
1004 }
1005
Rafael Espindolad661a852011-01-18 21:07:57 +00001006 if (UnnamedAddr)
1007 Entry->setUnnamedAddr(true);
1008
Chris Lattnerd4808922009-03-22 21:03:39 +00001009 if (Entry->getType() == Ty)
Chris Lattner149927c2009-03-21 09:16:30 +00001010 return Entry;
Mike Stump11289f42009-09-09 15:08:12 +00001011
Chris Lattner3bfce182009-03-21 08:03:33 +00001012 // Make sure the result is of the correct type.
Owen Andersonade90fd2009-07-29 18:54:39 +00001013 return llvm::ConstantExpr::getBitCast(Entry, Ty);
Daniel Dunbardec798b2009-01-13 02:25:00 +00001014 }
Mike Stump11289f42009-09-09 15:08:12 +00001015
Chris Lattner45470942009-03-21 09:44:56 +00001016 // This is the first use or definition of a mangled name. If there is a
1017 // deferred decl with this name, remember that we need to emit it at the end
1018 // of the file.
John McCall7ec50432010-03-19 23:29:14 +00001019 llvm::StringMap<GlobalDecl>::iterator DDI = DeferredDecls.find(MangledName);
Chris Lattner45470942009-03-21 09:44:56 +00001020 if (DDI != DeferredDecls.end()) {
1021 // Move the potentially referenced deferred decl to the DeferredDeclsToEmit
1022 // list, and remove it from DeferredDecls (since we don't need it anymore).
1023 DeferredDeclsToEmit.push_back(DDI->second);
1024 DeferredDecls.erase(DDI);
1025 }
Mike Stump11289f42009-09-09 15:08:12 +00001026
1027 llvm::GlobalVariable *GV =
1028 new llvm::GlobalVariable(getModule(), Ty->getElementType(), false,
Chris Lattner3bfce182009-03-21 08:03:33 +00001029 llvm::GlobalValue::ExternalLinkage,
John McCall7ec50432010-03-19 23:29:14 +00001030 0, MangledName, 0,
Eli Friedman4f856742009-04-19 21:05:03 +00001031 false, Ty->getAddressSpace());
Chris Lattner3bfce182009-03-21 08:03:33 +00001032
1033 // Handle things which are present even on external declarations.
Chris Lattnerd4808922009-03-22 21:03:39 +00001034 if (D) {
Mike Stump18bb9282009-05-16 07:57:57 +00001035 // FIXME: This code is overly simple and should be merged with other global
1036 // handling.
Douglas Gregor2d1c0552011-05-07 22:06:45 +00001037 GV->setConstant(DeclIsConstantGlobal(Context, D, false));
Chris Lattner3bfce182009-03-21 08:03:33 +00001038
John McCall37bb6c92010-10-29 22:22:43 +00001039 // Set linkage and visibility in case we never see a definition.
John McCallc273f242010-10-30 11:50:40 +00001040 NamedDecl::LinkageInfo LV = D->getLinkageAndVisibility();
1041 if (LV.linkage() != ExternalLinkage) {
John McCall83779672011-02-19 02:53:41 +00001042 // Don't set internal linkage on declarations.
John McCall37bb6c92010-10-29 22:22:43 +00001043 } else {
1044 if (D->hasAttr<DLLImportAttr>())
1045 GV->setLinkage(llvm::GlobalValue::DLLImportLinkage);
Douglas Gregor20b2ebd2011-03-23 00:50:03 +00001046 else if (D->hasAttr<WeakAttr>() || D->isWeakImported())
John McCall37bb6c92010-10-29 22:22:43 +00001047 GV->setLinkage(llvm::GlobalValue::ExternalWeakLinkage);
Chris Lattner3bfce182009-03-21 08:03:33 +00001048
John McCallc273f242010-10-30 11:50:40 +00001049 // Set visibility on a declaration only if it's explicit.
1050 if (LV.visibilityExplicit())
1051 GV->setVisibility(GetLLVMVisibility(LV.visibility()));
John McCall37bb6c92010-10-29 22:22:43 +00001052 }
Eli Friedman4f856742009-04-19 21:05:03 +00001053
1054 GV->setThreadLocal(D->isThreadSpecified());
Chris Lattnerd4808922009-03-22 21:03:39 +00001055 }
Mike Stump11289f42009-09-09 15:08:12 +00001056
John McCall7ec50432010-03-19 23:29:14 +00001057 return GV;
Daniel Dunbar9c426522008-07-29 23:18:29 +00001058}
1059
Chris Lattnerd4808922009-03-22 21:03:39 +00001060
Anders Carlssonda80af32011-01-29 18:20:20 +00001061llvm::GlobalVariable *
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001062CodeGenModule::CreateOrReplaceCXXRuntimeVariable(StringRef Name,
Chris Lattner2192fe52011-07-18 04:24:23 +00001063 llvm::Type *Ty,
Anders Carlssonda80af32011-01-29 18:20:20 +00001064 llvm::GlobalValue::LinkageTypes Linkage) {
1065 llvm::GlobalVariable *GV = getModule().getNamedGlobal(Name);
1066 llvm::GlobalVariable *OldGV = 0;
1067
1068
1069 if (GV) {
1070 // Check if the variable has the right type.
1071 if (GV->getType()->getElementType() == Ty)
1072 return GV;
1073
1074 // Because C++ name mangling, the only way we can end up with an already
1075 // existing global with the same name is if it has been declared extern "C".
Anders Carlsson93be9a92011-01-29 18:25:07 +00001076 assert(GV->isDeclaration() && "Declaration has wrong type!");
Anders Carlssonda80af32011-01-29 18:20:20 +00001077 OldGV = GV;
1078 }
1079
1080 // Create a new variable.
1081 GV = new llvm::GlobalVariable(getModule(), Ty, /*isConstant=*/true,
1082 Linkage, 0, Name);
1083
1084 if (OldGV) {
1085 // Replace occurrences of the old variable if needed.
1086 GV->takeName(OldGV);
1087
1088 if (!OldGV->use_empty()) {
1089 llvm::Constant *NewPtrForOldDecl =
1090 llvm::ConstantExpr::getBitCast(GV, OldGV->getType());
1091 OldGV->replaceAllUsesWith(NewPtrForOldDecl);
1092 }
1093
1094 OldGV->eraseFromParent();
1095 }
1096
1097 return GV;
1098}
1099
Chris Lattnerd4808922009-03-22 21:03:39 +00001100/// GetAddrOfGlobalVar - Return the llvm::Constant for the address of the
1101/// given global variable. If Ty is non-null and if the global doesn't exist,
1102/// then it will be greated with the specified type instead of whatever the
1103/// normal requested type would be.
1104llvm::Constant *CodeGenModule::GetAddrOfGlobalVar(const VarDecl *D,
Chris Lattner2192fe52011-07-18 04:24:23 +00001105 llvm::Type *Ty) {
Chris Lattnerd4808922009-03-22 21:03:39 +00001106 assert(D->hasGlobalStorage() && "Not a global variable");
1107 QualType ASTTy = D->getType();
1108 if (Ty == 0)
1109 Ty = getTypes().ConvertTypeForMem(ASTTy);
Mike Stump11289f42009-09-09 15:08:12 +00001110
Chris Lattner2192fe52011-07-18 04:24:23 +00001111 llvm::PointerType *PTy =
Peter Collingbourne599cb8e2011-03-18 22:38:29 +00001112 llvm::PointerType::get(Ty, getContext().getTargetAddressSpace(ASTTy));
John McCall7ec50432010-03-19 23:29:14 +00001113
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001114 StringRef MangledName = getMangledName(D);
John McCall7ec50432010-03-19 23:29:14 +00001115 return GetOrCreateLLVMGlobal(MangledName, PTy, D);
Chris Lattnerd4808922009-03-22 21:03:39 +00001116}
1117
1118/// CreateRuntimeVariable - Create a new runtime global variable with the
1119/// specified type and name.
1120llvm::Constant *
Chris Lattner2192fe52011-07-18 04:24:23 +00001121CodeGenModule::CreateRuntimeVariable(llvm::Type *Ty,
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001122 StringRef Name) {
John McCall31996342011-04-07 08:22:57 +00001123 return GetOrCreateLLVMGlobal(Name, llvm::PointerType::getUnqual(Ty), 0,
Rafael Espindolad661a852011-01-18 21:07:57 +00001124 true);
Chris Lattnerd4808922009-03-22 21:03:39 +00001125}
1126
Daniel Dunbar7dd749e2009-04-15 22:08:45 +00001127void CodeGenModule::EmitTentativeDefinition(const VarDecl *D) {
1128 assert(!D->getInit() && "Cannot emit definite definitions here!");
1129
Douglas Gregorfa9ab532009-04-21 19:28:58 +00001130 if (MayDeferGeneration(D)) {
1131 // If we have not seen a reference to this variable yet, place it
1132 // into the deferred declarations table to be emitted if needed
1133 // later.
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001134 StringRef MangledName = getMangledName(D);
John McCall7ec50432010-03-19 23:29:14 +00001135 if (!GetGlobalValue(MangledName)) {
Anders Carlssonecf9bf02009-09-10 23:43:36 +00001136 DeferredDecls[MangledName] = D;
Daniel Dunbar7dd749e2009-04-15 22:08:45 +00001137 return;
Douglas Gregorfa9ab532009-04-21 19:28:58 +00001138 }
1139 }
1140
1141 // The tentative definition is the only definition.
Daniel Dunbar7dd749e2009-04-15 22:08:45 +00001142 EmitGlobalVarDefinition(D);
1143}
1144
Douglas Gregor88d292c2010-05-13 16:44:06 +00001145void CodeGenModule::EmitVTable(CXXRecordDecl *Class, bool DefinitionRequired) {
1146 if (DefinitionRequired)
1147 getVTables().GenerateClassData(getVTableLinkage(Class), Class);
1148}
1149
Douglas Gregorccecc1b2010-01-06 20:27:16 +00001150llvm::GlobalVariable::LinkageTypes
Anders Carlsson11e51402010-04-17 20:15:18 +00001151CodeGenModule::getVTableLinkage(const CXXRecordDecl *RD) {
Eli Friedman300f55d2011-06-10 21:53:06 +00001152 if (RD->getLinkage() != ExternalLinkage)
Douglas Gregor2a34df32010-01-06 22:00:56 +00001153 return llvm::GlobalVariable::InternalLinkage;
1154
1155 if (const CXXMethodDecl *KeyFunction
1156 = RD->getASTContext().getKeyFunction(RD)) {
1157 // If this class has a key function, use that to determine the linkage of
1158 // the vtable.
Douglas Gregorccecc1b2010-01-06 20:27:16 +00001159 const FunctionDecl *Def = 0;
Argyrios Kyrtzidis36ea3222010-07-07 11:31:19 +00001160 if (KeyFunction->hasBody(Def))
Douglas Gregorccecc1b2010-01-06 20:27:16 +00001161 KeyFunction = cast<CXXMethodDecl>(Def);
Douglas Gregor2a34df32010-01-06 22:00:56 +00001162
Douglas Gregorccecc1b2010-01-06 20:27:16 +00001163 switch (KeyFunction->getTemplateSpecializationKind()) {
1164 case TSK_Undeclared:
1165 case TSK_ExplicitSpecialization:
Anders Carlssona03f3a82011-01-30 20:45:54 +00001166 // When compiling with optimizations turned on, we emit all vtables,
1167 // even if the key function is not defined in the current translation
1168 // unit. If this is the case, use available_externally linkage.
1169 if (!Def && CodeGenOpts.OptimizationLevel)
1170 return llvm::GlobalVariable::AvailableExternallyLinkage;
1171
Douglas Gregorccecc1b2010-01-06 20:27:16 +00001172 if (KeyFunction->isInlined())
Fariborz Jahanianf7f04452011-02-04 00:01:24 +00001173 return !Context.getLangOptions().AppleKext ?
1174 llvm::GlobalVariable::LinkOnceODRLinkage :
1175 llvm::Function::InternalLinkage;
Douglas Gregorccecc1b2010-01-06 20:27:16 +00001176
1177 return llvm::GlobalVariable::ExternalLinkage;
1178
1179 case TSK_ImplicitInstantiation:
Fariborz Jahanianf7f04452011-02-04 00:01:24 +00001180 return !Context.getLangOptions().AppleKext ?
1181 llvm::GlobalVariable::LinkOnceODRLinkage :
1182 llvm::Function::InternalLinkage;
Anders Carlsson571e2ad2011-01-24 00:46:19 +00001183
Douglas Gregorccecc1b2010-01-06 20:27:16 +00001184 case TSK_ExplicitInstantiationDefinition:
Fariborz Jahanianf7f04452011-02-04 00:01:24 +00001185 return !Context.getLangOptions().AppleKext ?
1186 llvm::GlobalVariable::WeakODRLinkage :
1187 llvm::Function::InternalLinkage;
Anders Carlsson571e2ad2011-01-24 00:46:19 +00001188
Douglas Gregorccecc1b2010-01-06 20:27:16 +00001189 case TSK_ExplicitInstantiationDeclaration:
1190 // FIXME: Use available_externally linkage. However, this currently
1191 // breaks LLVM's build due to undefined symbols.
1192 // return llvm::GlobalVariable::AvailableExternallyLinkage;
Fariborz Jahanianf7f04452011-02-04 00:01:24 +00001193 return !Context.getLangOptions().AppleKext ?
1194 llvm::GlobalVariable::LinkOnceODRLinkage :
1195 llvm::Function::InternalLinkage;
Douglas Gregorccecc1b2010-01-06 20:27:16 +00001196 }
Douglas Gregor2a34df32010-01-06 22:00:56 +00001197 }
1198
Fariborz Jahaniane28342c2011-02-04 00:32:39 +00001199 if (Context.getLangOptions().AppleKext)
1200 return llvm::Function::InternalLinkage;
1201
Douglas Gregor2a34df32010-01-06 22:00:56 +00001202 switch (RD->getTemplateSpecializationKind()) {
1203 case TSK_Undeclared:
1204 case TSK_ExplicitSpecialization:
1205 case TSK_ImplicitInstantiation:
Douglas Gregor2a34df32010-01-06 22:00:56 +00001206 // FIXME: Use available_externally linkage. However, this currently
1207 // breaks LLVM's build due to undefined symbols.
1208 // return llvm::GlobalVariable::AvailableExternallyLinkage;
Fariborz Jahanianf7f04452011-02-04 00:01:24 +00001209 case TSK_ExplicitInstantiationDeclaration:
Fariborz Jahaniane28342c2011-02-04 00:32:39 +00001210 return llvm::GlobalVariable::LinkOnceODRLinkage;
Fariborz Jahanianf7f04452011-02-04 00:01:24 +00001211
1212 case TSK_ExplicitInstantiationDefinition:
Fariborz Jahaniane28342c2011-02-04 00:32:39 +00001213 return llvm::GlobalVariable::WeakODRLinkage;
Douglas Gregorccecc1b2010-01-06 20:27:16 +00001214 }
1215
1216 // Silence GCC warning.
Fariborz Jahaniane28342c2011-02-04 00:32:39 +00001217 return llvm::GlobalVariable::LinkOnceODRLinkage;
Douglas Gregorccecc1b2010-01-06 20:27:16 +00001218}
1219
Chris Lattner2192fe52011-07-18 04:24:23 +00001220CharUnits CodeGenModule::GetTargetTypeStoreSize(llvm::Type *Ty) const {
Ken Dyck9a648692011-01-18 02:01:14 +00001221 return Context.toCharUnitsFromBits(
1222 TheTargetData.getTypeStoreSizeInBits(Ty));
Ken Dyck98ca7942010-01-26 13:48:07 +00001223}
1224
Daniel Dunbar9c426522008-07-29 23:18:29 +00001225void CodeGenModule::EmitGlobalVarDefinition(const VarDecl *D) {
Chris Lattner9d3b0e02007-07-14 00:23:28 +00001226 llvm::Constant *Init = 0;
Eli Friedmanc18d9d52008-05-30 19:50:47 +00001227 QualType ASTTy = D->getType();
Eli Friedman5866fe32010-01-08 00:50:11 +00001228 bool NonConstInit = false;
Mike Stump11289f42009-09-09 15:08:12 +00001229
Sebastian Redl5ca79842010-02-01 20:16:42 +00001230 const Expr *InitExpr = D->getAnyInitializer();
Anders Carlssonca4a5452010-01-26 17:43:42 +00001231
1232 if (!InitExpr) {
Eli Friedman6859a1b2008-05-30 20:39:54 +00001233 // This is a tentative definition; tentative definitions are
Daniel Dunbar7dd749e2009-04-15 22:08:45 +00001234 // implicitly initialized with { 0 }.
1235 //
1236 // Note that tentative definitions are only emitted at the end of
1237 // a translation unit, so they should never have incomplete
1238 // type. In addition, EmitTentativeDefinition makes sure that we
1239 // never attempt to emit a tentative definition if a real one
1240 // exists. A use may still exists, however, so we still may need
1241 // to do a RAUW.
1242 assert(!ASTTy->isIncompleteType() && "Unexpected incomplete type");
Anders Carlssonf18318c2009-08-02 21:18:22 +00001243 Init = EmitNullConstant(D->getType());
Eli Friedmanc18d9d52008-05-30 19:50:47 +00001244 } else {
Douglas Gregord450f062010-05-05 20:15:55 +00001245 Init = EmitConstantExpr(InitExpr, D->getType());
Eli Friedman719ed1a2009-02-20 01:18:21 +00001246 if (!Init) {
Anders Carlssonca4a5452010-01-26 17:43:42 +00001247 QualType T = InitExpr->getType();
Douglas Gregord450f062010-05-05 20:15:55 +00001248 if (D->getType()->isReferenceType())
1249 T = D->getType();
1250
Anders Carlssonb8be93f2009-08-08 23:24:23 +00001251 if (getLangOptions().CPlusPlus) {
Anders Carlssonb8be93f2009-08-08 23:24:23 +00001252 Init = EmitNullConstant(T);
Eli Friedman5866fe32010-01-08 00:50:11 +00001253 NonConstInit = true;
Anders Carlssonb8be93f2009-08-08 23:24:23 +00001254 } else {
1255 ErrorUnsupported(D, "static initializer");
1256 Init = llvm::UndefValue::get(getTypes().ConvertType(T));
1257 }
John McCall70013b62010-07-15 23:40:35 +00001258 } else {
1259 // We don't need an initializer, so remove the entry for the delayed
1260 // initializer position (just in case this entry was delayed).
1261 if (getLangOptions().CPlusPlus)
1262 DelayedCXXInitPosition.erase(D);
Eli Friedman719ed1a2009-02-20 01:18:21 +00001263 }
Eli Friedmanc18d9d52008-05-30 19:50:47 +00001264 }
Eli Friedmanc18d9d52008-05-30 19:50:47 +00001265
Chris Lattner2192fe52011-07-18 04:24:23 +00001266 llvm::Type* InitType = Init->getType();
Chris Lattner149927c2009-03-21 09:16:30 +00001267 llvm::Constant *Entry = GetAddrOfGlobalVar(D, InitType);
Mike Stump11289f42009-09-09 15:08:12 +00001268
Chris Lattner149927c2009-03-21 09:16:30 +00001269 // Strip off a bitcast if we got one back.
1270 if (llvm::ConstantExpr *CE = dyn_cast<llvm::ConstantExpr>(Entry)) {
Chris Lattneraa64ca22009-07-16 16:48:25 +00001271 assert(CE->getOpcode() == llvm::Instruction::BitCast ||
1272 // all zero index gep.
1273 CE->getOpcode() == llvm::Instruction::GetElementPtr);
Chris Lattner149927c2009-03-21 09:16:30 +00001274 Entry = CE->getOperand(0);
1275 }
Mike Stump11289f42009-09-09 15:08:12 +00001276
Chris Lattner149927c2009-03-21 09:16:30 +00001277 // Entry is now either a Function or GlobalVariable.
1278 llvm::GlobalVariable *GV = dyn_cast<llvm::GlobalVariable>(Entry);
Mike Stump11289f42009-09-09 15:08:12 +00001279
Chris Lattner149927c2009-03-21 09:16:30 +00001280 // We have a definition after a declaration with the wrong type.
1281 // We must make a new GlobalVariable* and update everything that used OldGV
1282 // (a declaration or tentative definition) with the new GlobalVariable*
1283 // (which will be a definition).
1284 //
1285 // This happens if there is a prototype for a global (e.g.
1286 // "extern int x[];") and then a definition of a different type (e.g.
1287 // "int x[10];"). This also happens when an initializer has a different type
1288 // from the type of the global (this happens with unions).
Chris Lattner149927c2009-03-21 09:16:30 +00001289 if (GV == 0 ||
1290 GV->getType()->getElementType() != InitType ||
Peter Collingbourne599cb8e2011-03-18 22:38:29 +00001291 GV->getType()->getAddressSpace() !=
1292 getContext().getTargetAddressSpace(ASTTy)) {
Mike Stump11289f42009-09-09 15:08:12 +00001293
John McCall7ec50432010-03-19 23:29:14 +00001294 // Move the old entry aside so that we'll create a new one.
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001295 Entry->setName(StringRef());
Daniel Dunbarb2f4cdb2009-02-19 05:36:41 +00001296
Chris Lattner149927c2009-03-21 09:16:30 +00001297 // Make a new global with the correct type, this is now guaranteed to work.
1298 GV = cast<llvm::GlobalVariable>(GetAddrOfGlobalVar(D, InitType));
Chris Lattnera85d68e2009-03-21 09:25:43 +00001299
Eli Friedmanc18d9d52008-05-30 19:50:47 +00001300 // Replace all uses of the old global with the new global
Mike Stump11289f42009-09-09 15:08:12 +00001301 llvm::Constant *NewPtrForOldDecl =
Owen Andersonade90fd2009-07-29 18:54:39 +00001302 llvm::ConstantExpr::getBitCast(GV, Entry->getType());
Chris Lattner149927c2009-03-21 09:16:30 +00001303 Entry->replaceAllUsesWith(NewPtrForOldDecl);
Eli Friedmanc18d9d52008-05-30 19:50:47 +00001304
1305 // Erase the old global, since it is no longer used.
Chris Lattner149927c2009-03-21 09:16:30 +00001306 cast<llvm::GlobalValue>(Entry)->eraseFromParent();
Chris Lattner9d3b0e02007-07-14 00:23:28 +00001307 }
Devang Patel19c2b9a2007-10-26 16:31:40 +00001308
Julien Lerouge5a6b6982011-09-09 22:41:49 +00001309 if (D->hasAttr<AnnotateAttr>())
1310 AddGlobalAnnotations(D, GV);
Nate Begemanfaae0812008-04-19 04:17:09 +00001311
Chris Lattnerd14bfa92007-07-13 05:13:43 +00001312 GV->setInitializer(Init);
Chris Lattnerf49573d2009-08-05 05:20:29 +00001313
1314 // If it is safe to mark the global 'constant', do so now.
1315 GV->setConstant(false);
Douglas Gregor2d1c0552011-05-07 22:06:45 +00001316 if (!NonConstInit && DeclIsConstantGlobal(Context, D, true))
Chris Lattnerf49573d2009-08-05 05:20:29 +00001317 GV->setConstant(true);
Mike Stump11289f42009-09-09 15:08:12 +00001318
Ken Dyck160146e2010-01-27 17:10:57 +00001319 GV->setAlignment(getContext().getDeclAlign(D).getQuantity());
Fariborz Jahanian1518a5ec2010-10-27 16:21:54 +00001320
Chris Lattnerd14bfa92007-07-13 05:13:43 +00001321 // Set the llvm linkage type as appropriate.
Fariborz Jahanian1518a5ec2010-10-27 16:21:54 +00001322 llvm::GlobalValue::LinkageTypes Linkage =
1323 GetLLVMLinkageVarDefinition(D, GV);
1324 GV->setLinkage(Linkage);
1325 if (Linkage == llvm::GlobalVariable::CommonLinkage)
Chris Lattnerf49573d2009-08-05 05:20:29 +00001326 // common vars aren't constant even if declared const.
1327 GV->setConstant(false);
Daniel Dunbard272cca2009-04-10 20:26:50 +00001328
Daniel Dunbar38932572009-04-14 08:05:55 +00001329 SetCommonAttributes(D, GV);
Daniel Dunbarf5f359f2009-04-14 06:00:08 +00001330
John McCallcdf7ef52010-11-06 09:44:32 +00001331 // Emit the initializer function if necessary.
1332 if (NonConstInit)
1333 EmitCXXGlobalVarDeclInitFunc(D, GV);
1334
Sanjiv Gupta158143a2008-06-05 08:59:10 +00001335 // Emit global variable debug information.
Devang Pateld6ffebb2011-03-07 18:45:56 +00001336 if (CGDebugInfo *DI = getModuleDebugInfo()) {
Daniel Dunbarb9fd9022008-10-17 16:15:48 +00001337 DI->setLocation(D->getLocation());
Sanjiv Gupta158143a2008-06-05 08:59:10 +00001338 DI->EmitGlobalVariable(GV, D);
1339 }
Chris Lattnerd14bfa92007-07-13 05:13:43 +00001340}
Chris Lattner09153c02007-06-22 18:48:09 +00001341
Fariborz Jahanian1518a5ec2010-10-27 16:21:54 +00001342llvm::GlobalValue::LinkageTypes
1343CodeGenModule::GetLLVMLinkageVarDefinition(const VarDecl *D,
1344 llvm::GlobalVariable *GV) {
1345 GVALinkage Linkage = getContext().GetGVALinkageForVariable(D);
1346 if (Linkage == GVA_Internal)
1347 return llvm::Function::InternalLinkage;
1348 else if (D->hasAttr<DLLImportAttr>())
1349 return llvm::Function::DLLImportLinkage;
1350 else if (D->hasAttr<DLLExportAttr>())
1351 return llvm::Function::DLLExportLinkage;
1352 else if (D->hasAttr<WeakAttr>()) {
1353 if (GV->isConstant())
1354 return llvm::GlobalVariable::WeakODRLinkage;
1355 else
1356 return llvm::GlobalVariable::WeakAnyLinkage;
1357 } else if (Linkage == GVA_TemplateInstantiation ||
1358 Linkage == GVA_ExplicitTemplateInstantiation)
John McCalla97f3292011-04-12 01:46:54 +00001359 return llvm::GlobalVariable::WeakODRLinkage;
Eric Christopher8a2ee392010-12-02 02:45:55 +00001360 else if (!getLangOptions().CPlusPlus &&
1361 ((!CodeGenOpts.NoCommon && !D->getAttr<NoCommonAttr>()) ||
1362 D->getAttr<CommonAttr>()) &&
Fariborz Jahanian1518a5ec2010-10-27 16:21:54 +00001363 !D->hasExternalStorage() && !D->getInit() &&
Fariborz Jahanianab578bf2011-06-20 17:50:03 +00001364 !D->getAttr<SectionAttr>() && !D->isThreadSpecified() &&
1365 !D->getAttr<WeakImportAttr>()) {
Fariborz Jahanian1518a5ec2010-10-27 16:21:54 +00001366 // Thread local vars aren't considered common linkage.
1367 return llvm::GlobalVariable::CommonLinkage;
1368 }
1369 return llvm::GlobalVariable::ExternalLinkage;
1370}
1371
Chris Lattner36797ab2009-05-05 06:16:31 +00001372/// ReplaceUsesOfNonProtoTypeWithRealFunction - This function is called when we
1373/// implement a function with no prototype, e.g. "int foo() {}". If there are
1374/// existing call uses of the old function in the module, this adjusts them to
1375/// call the new function directly.
1376///
1377/// This is not just a cleanup: the always_inline pass requires direct calls to
1378/// functions to be able to inline them. If there is a bitcast in the way, it
1379/// won't inline them. Instcombine normally deletes these calls, but it isn't
1380/// run at -O0.
1381static void ReplaceUsesOfNonProtoTypeWithRealFunction(llvm::GlobalValue *Old,
1382 llvm::Function *NewFn) {
1383 // If we're redefining a global as a function, don't transform it.
1384 llvm::Function *OldFn = dyn_cast<llvm::Function>(Old);
1385 if (OldFn == 0) return;
Mike Stump11289f42009-09-09 15:08:12 +00001386
Chris Lattner2192fe52011-07-18 04:24:23 +00001387 llvm::Type *NewRetTy = NewFn->getReturnType();
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001388 SmallVector<llvm::Value*, 4> ArgList;
Chris Lattner36797ab2009-05-05 06:16:31 +00001389
1390 for (llvm::Value::use_iterator UI = OldFn->use_begin(), E = OldFn->use_end();
Benjamin Kramer2e8ca0b2010-04-10 11:02:40 +00001391 UI != E; ) {
Chris Lattner36797ab2009-05-05 06:16:31 +00001392 // TODO: Do invokes ever occur in C code? If so, we should handle them too.
Benjamin Kramer2e8ca0b2010-04-10 11:02:40 +00001393 llvm::Value::use_iterator I = UI++; // Increment before the CI is erased.
1394 llvm::CallInst *CI = dyn_cast<llvm::CallInst>(*I);
Gabor Greif79ac9ed2010-07-28 09:19:33 +00001395 if (!CI) continue; // FIXME: when we allow Invoke, just do CallSite CS(*I)
Gabor Greifd394aec2010-04-10 03:45:50 +00001396 llvm::CallSite CS(CI);
Benjamin Kramer2e8ca0b2010-04-10 11:02:40 +00001397 if (!CI || !CS.isCallee(I)) continue;
Mike Stump11289f42009-09-09 15:08:12 +00001398
Chris Lattner36797ab2009-05-05 06:16:31 +00001399 // If the return types don't match exactly, and if the call isn't dead, then
1400 // we can't transform this call.
1401 if (CI->getType() != NewRetTy && !CI->use_empty())
1402 continue;
1403
John McCall625ed882011-08-03 00:43:55 +00001404 // Get the attribute list.
1405 llvm::SmallVector<llvm::AttributeWithIndex, 8> AttrVec;
1406 llvm::AttrListPtr AttrList = CI->getAttributes();
1407
1408 // Get any return attributes.
1409 llvm::Attributes RAttrs = AttrList.getRetAttributes();
1410
1411 // Add the return attributes.
1412 if (RAttrs)
1413 AttrVec.push_back(llvm::AttributeWithIndex::get(0, RAttrs));
1414
Chris Lattner36797ab2009-05-05 06:16:31 +00001415 // If the function was passed too few arguments, don't transform. If extra
1416 // arguments were passed, we silently drop them. If any of the types
1417 // mismatch, we don't transform.
1418 unsigned ArgNo = 0;
1419 bool DontTransform = false;
1420 for (llvm::Function::arg_iterator AI = NewFn->arg_begin(),
1421 E = NewFn->arg_end(); AI != E; ++AI, ++ArgNo) {
Gabor Greifd394aec2010-04-10 03:45:50 +00001422 if (CS.arg_size() == ArgNo ||
1423 CS.getArgument(ArgNo)->getType() != AI->getType()) {
Chris Lattner36797ab2009-05-05 06:16:31 +00001424 DontTransform = true;
1425 break;
1426 }
John McCall625ed882011-08-03 00:43:55 +00001427
1428 // Add any parameter attributes.
1429 if (llvm::Attributes PAttrs = AttrList.getParamAttributes(ArgNo + 1))
1430 AttrVec.push_back(llvm::AttributeWithIndex::get(ArgNo + 1, PAttrs));
Chris Lattner36797ab2009-05-05 06:16:31 +00001431 }
1432 if (DontTransform)
1433 continue;
Mike Stump11289f42009-09-09 15:08:12 +00001434
John McCall625ed882011-08-03 00:43:55 +00001435 if (llvm::Attributes FnAttrs = AttrList.getFnAttributes())
1436 AttrVec.push_back(llvm::AttributeWithIndex::get(~0, FnAttrs));
1437
Chris Lattner36797ab2009-05-05 06:16:31 +00001438 // Okay, we can transform this. Create the new call instruction and copy
1439 // over the required information.
Gabor Greifd0ef1342010-04-10 02:56:12 +00001440 ArgList.append(CS.arg_begin(), CS.arg_begin() + ArgNo);
Jay Foad5bd375a2011-07-15 08:37:34 +00001441 llvm::CallInst *NewCall = llvm::CallInst::Create(NewFn, ArgList, "", CI);
Chris Lattner36797ab2009-05-05 06:16:31 +00001442 ArgList.clear();
Benjamin Kramerdde0fee2009-10-05 13:47:21 +00001443 if (!NewCall->getType()->isVoidTy())
Chris Lattner36797ab2009-05-05 06:16:31 +00001444 NewCall->takeName(CI);
John McCall625ed882011-08-03 00:43:55 +00001445 NewCall->setAttributes(llvm::AttrListPtr::get(AttrVec.begin(),
1446 AttrVec.end()));
Daniel Dunbar0ef34792009-09-12 00:59:20 +00001447 NewCall->setCallingConv(CI->getCallingConv());
Chris Lattner36797ab2009-05-05 06:16:31 +00001448
1449 // Finally, remove the old call, replacing any uses with the new one.
Chris Lattner734351d2009-10-14 05:49:21 +00001450 if (!CI->use_empty())
1451 CI->replaceAllUsesWith(NewCall);
Devang Patel74684892009-10-13 17:02:04 +00001452
Chris Lattnere675d0f2010-04-01 06:31:43 +00001453 // Copy debug location attached to CI.
1454 if (!CI->getDebugLoc().isUnknown())
1455 NewCall->setDebugLoc(CI->getDebugLoc());
Chris Lattner36797ab2009-05-05 06:16:31 +00001456 CI->eraseFromParent();
1457 }
1458}
1459
Daniel Dunbar9c426522008-07-29 23:18:29 +00001460
Chris Lattnere0be0df2009-05-12 21:21:08 +00001461void CodeGenModule::EmitGlobalFunctionDefinition(GlobalDecl GD) {
Chris Lattnere0be0df2009-05-12 21:21:08 +00001462 const FunctionDecl *D = cast<FunctionDecl>(GD.getDecl());
John McCalla738c252011-03-09 04:27:21 +00001463
John McCall46288ef2011-03-09 08:12:35 +00001464 // Compute the function info and LLVM type.
John McCalla738c252011-03-09 04:27:21 +00001465 const CGFunctionInfo &FI = getTypes().getFunctionInfo(GD);
John McCall46288ef2011-03-09 08:12:35 +00001466 bool variadic = false;
1467 if (const FunctionProtoType *fpt = D->getType()->getAs<FunctionProtoType>())
1468 variadic = fpt->isVariadic();
Chris Lattner2192fe52011-07-18 04:24:23 +00001469 llvm::FunctionType *Ty = getTypes().GetFunctionType(FI, variadic);
John McCalla738c252011-03-09 04:27:21 +00001470
Chris Lattnereb7466d2009-05-12 20:58:15 +00001471 // Get or create the prototype for the function.
Chris Lattnere0be0df2009-05-12 21:21:08 +00001472 llvm::Constant *Entry = GetAddrOfFunction(GD, Ty);
Mike Stump11289f42009-09-09 15:08:12 +00001473
Chris Lattnera85d68e2009-03-21 09:25:43 +00001474 // Strip off a bitcast if we got one back.
1475 if (llvm::ConstantExpr *CE = dyn_cast<llvm::ConstantExpr>(Entry)) {
1476 assert(CE->getOpcode() == llvm::Instruction::BitCast);
1477 Entry = CE->getOperand(0);
1478 }
Mike Stump11289f42009-09-09 15:08:12 +00001479
1480
Chris Lattnera85d68e2009-03-21 09:25:43 +00001481 if (cast<llvm::GlobalValue>(Entry)->getType()->getElementType() != Ty) {
Chris Lattner36797ab2009-05-05 06:16:31 +00001482 llvm::GlobalValue *OldFn = cast<llvm::GlobalValue>(Entry);
Mike Stump11289f42009-09-09 15:08:12 +00001483
Daniel Dunbar99d28352009-03-09 23:53:08 +00001484 // If the types mismatch then we have to rewrite the definition.
Chris Lattner36797ab2009-05-05 06:16:31 +00001485 assert(OldFn->isDeclaration() &&
Chris Lattnera85d68e2009-03-21 09:25:43 +00001486 "Shouldn't replace non-declaration");
Chris Lattner832323e2009-03-21 08:53:37 +00001487
Chris Lattner5eaee562009-03-21 08:38:50 +00001488 // F is the Function* for the one with the wrong type, we must make a new
1489 // Function* and update everything that used F (a declaration) with the new
1490 // Function* (which will be a definition).
1491 //
1492 // This happens if there is a prototype for a function
1493 // (e.g. "int f()") and then a definition of a different type
John McCall7ec50432010-03-19 23:29:14 +00001494 // (e.g. "int f(int x)"). Move the old function aside so that it
1495 // doesn't interfere with GetAddrOfFunction.
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001496 OldFn->setName(StringRef());
Chris Lattnere0be0df2009-05-12 21:21:08 +00001497 llvm::Function *NewFn = cast<llvm::Function>(GetAddrOfFunction(GD, Ty));
Mike Stump11289f42009-09-09 15:08:12 +00001498
Chris Lattner36797ab2009-05-05 06:16:31 +00001499 // If this is an implementation of a function without a prototype, try to
1500 // replace any existing uses of the function (which may be calls) with uses
1501 // of the new function
Chris Lattnereb7466d2009-05-12 20:58:15 +00001502 if (D->getType()->isFunctionNoProtoType()) {
Chris Lattner36797ab2009-05-05 06:16:31 +00001503 ReplaceUsesOfNonProtoTypeWithRealFunction(OldFn, NewFn);
Chris Lattnereb7466d2009-05-12 20:58:15 +00001504 OldFn->removeDeadConstantUsers();
1505 }
Mike Stump11289f42009-09-09 15:08:12 +00001506
Chris Lattner5eaee562009-03-21 08:38:50 +00001507 // Replace uses of F with the Function we will endow with a body.
Chris Lattner36797ab2009-05-05 06:16:31 +00001508 if (!Entry->use_empty()) {
Mike Stump11289f42009-09-09 15:08:12 +00001509 llvm::Constant *NewPtrForOldDecl =
Owen Andersonade90fd2009-07-29 18:54:39 +00001510 llvm::ConstantExpr::getBitCast(NewFn, Entry->getType());
Chris Lattner36797ab2009-05-05 06:16:31 +00001511 Entry->replaceAllUsesWith(NewPtrForOldDecl);
1512 }
Mike Stump11289f42009-09-09 15:08:12 +00001513
Chris Lattner5eaee562009-03-21 08:38:50 +00001514 // Ok, delete the old function now, which is dead.
Chris Lattner36797ab2009-05-05 06:16:31 +00001515 OldFn->eraseFromParent();
Mike Stump11289f42009-09-09 15:08:12 +00001516
Chris Lattnera85d68e2009-03-21 09:25:43 +00001517 Entry = NewFn;
Daniel Dunbar9c426522008-07-29 23:18:29 +00001518 }
Mike Stump11289f42009-09-09 15:08:12 +00001519
John McCall8e7cb6d2010-11-02 21:04:24 +00001520 // We need to set linkage and visibility on the function before
1521 // generating code for it because various parts of IR generation
1522 // want to propagate this information down (e.g. to local static
1523 // declarations).
Chris Lattnera85d68e2009-03-21 09:25:43 +00001524 llvm::Function *Fn = cast<llvm::Function>(Entry);
John McCall7cb02202010-05-25 04:30:21 +00001525 setFunctionLinkage(D, Fn);
Daniel Dunbar9c426522008-07-29 23:18:29 +00001526
John McCall8e7cb6d2010-11-02 21:04:24 +00001527 // FIXME: this is redundant with part of SetFunctionDefinitionAttributes
Anders Carlssonc6a47892011-01-29 19:39:23 +00001528 setGlobalVisibility(Fn, D);
John McCall8e7cb6d2010-11-02 21:04:24 +00001529
John McCalla738c252011-03-09 04:27:21 +00001530 CodeGenFunction(*this).GenerateCode(D, Fn, FI);
Daniel Dunbar74aa7e12008-08-01 00:01:51 +00001531
Daniel Dunbar38932572009-04-14 08:05:55 +00001532 SetFunctionDefinitionAttributes(D, Fn);
1533 SetLLVMFunctionAttributesForDefinition(D, Fn);
Mike Stump11289f42009-09-09 15:08:12 +00001534
Argyrios Kyrtzidisb4b64ca2009-06-30 02:34:44 +00001535 if (const ConstructorAttr *CA = D->getAttr<ConstructorAttr>())
Daniel Dunbar0beedc12008-09-08 23:44:31 +00001536 AddGlobalCtor(Fn, CA->getPriority());
Argyrios Kyrtzidisb4b64ca2009-06-30 02:34:44 +00001537 if (const DestructorAttr *DA = D->getAttr<DestructorAttr>())
Daniel Dunbar0beedc12008-09-08 23:44:31 +00001538 AddGlobalDtor(Fn, DA->getPriority());
Julien Lerouge5a6b6982011-09-09 22:41:49 +00001539 if (D->hasAttr<AnnotateAttr>())
1540 AddGlobalAnnotations(D, Fn);
Daniel Dunbar9c426522008-07-29 23:18:29 +00001541}
1542
John McCall7ec50432010-03-19 23:29:14 +00001543void CodeGenModule::EmitAliasDefinition(GlobalDecl GD) {
1544 const ValueDecl *D = cast<ValueDecl>(GD.getDecl());
Argyrios Kyrtzidisb4b64ca2009-06-30 02:34:44 +00001545 const AliasAttr *AA = D->getAttr<AliasAttr>();
Chris Lattner54041692009-03-22 21:47:11 +00001546 assert(AA && "Not an alias?");
1547
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001548 StringRef MangledName = getMangledName(GD);
Mike Stump11289f42009-09-09 15:08:12 +00001549
John McCall7ec50432010-03-19 23:29:14 +00001550 // If there is a definition in the module, then it wins over the alias.
1551 // This is dubious, but allow it to be safe. Just ignore the alias.
1552 llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
1553 if (Entry && !Entry->isDeclaration())
1554 return;
1555
Chris Lattner2192fe52011-07-18 04:24:23 +00001556 llvm::Type *DeclTy = getTypes().ConvertTypeForMem(D->getType());
Chris Lattner54041692009-03-22 21:47:11 +00001557
1558 // Create a reference to the named value. This ensures that it is emitted
1559 // if a deferred decl.
1560 llvm::Constant *Aliasee;
1561 if (isa<llvm::FunctionType>(DeclTy))
Anders Carlsson3c239482011-02-05 04:35:53 +00001562 Aliasee = GetOrCreateLLVMFunction(AA->getAliasee(), DeclTy, GlobalDecl(),
1563 /*ForVTable=*/false);
Chris Lattner54041692009-03-22 21:47:11 +00001564 else
John McCall7ec50432010-03-19 23:29:14 +00001565 Aliasee = GetOrCreateLLVMGlobal(AA->getAliasee(),
Owen Anderson9793f0e2009-07-29 22:16:19 +00001566 llvm::PointerType::getUnqual(DeclTy), 0);
Chris Lattner54041692009-03-22 21:47:11 +00001567
1568 // Create the new alias itself, but don't set a name yet.
Mike Stump11289f42009-09-09 15:08:12 +00001569 llvm::GlobalValue *GA =
Chris Lattner54041692009-03-22 21:47:11 +00001570 new llvm::GlobalAlias(Aliasee->getType(),
1571 llvm::Function::ExternalLinkage,
1572 "", Aliasee, &getModule());
Mike Stump11289f42009-09-09 15:08:12 +00001573
Chris Lattner54041692009-03-22 21:47:11 +00001574 if (Entry) {
John McCall7ec50432010-03-19 23:29:14 +00001575 assert(Entry->isDeclaration());
1576
Chris Lattner54041692009-03-22 21:47:11 +00001577 // If there is a declaration in the module, then we had an extern followed
1578 // by the alias, as in:
1579 // extern int test6();
1580 // ...
1581 // int test6() __attribute__((alias("test7")));
1582 //
1583 // Remove it and replace uses of it with the alias.
John McCall7ec50432010-03-19 23:29:14 +00001584 GA->takeName(Entry);
Mike Stump11289f42009-09-09 15:08:12 +00001585
Owen Andersonade90fd2009-07-29 18:54:39 +00001586 Entry->replaceAllUsesWith(llvm::ConstantExpr::getBitCast(GA,
Chris Lattner54041692009-03-22 21:47:11 +00001587 Entry->getType()));
Chris Lattner54041692009-03-22 21:47:11 +00001588 Entry->eraseFromParent();
John McCall7ec50432010-03-19 23:29:14 +00001589 } else {
Anders Carlssonea836bc2010-06-22 16:16:50 +00001590 GA->setName(MangledName);
Chris Lattner54041692009-03-22 21:47:11 +00001591 }
Mike Stump11289f42009-09-09 15:08:12 +00001592
Daniel Dunbar38932572009-04-14 08:05:55 +00001593 // Set attributes which are particular to an alias; this is a
1594 // specialization of the attributes which may be set on a global
1595 // variable/function.
Argyrios Kyrtzidisb4b64ca2009-06-30 02:34:44 +00001596 if (D->hasAttr<DLLExportAttr>()) {
Daniel Dunbar38932572009-04-14 08:05:55 +00001597 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
1598 // The dllexport attribute is ignored for undefined symbols.
Argyrios Kyrtzidis36ea3222010-07-07 11:31:19 +00001599 if (FD->hasBody())
Daniel Dunbar38932572009-04-14 08:05:55 +00001600 GA->setLinkage(llvm::Function::DLLExportLinkage);
1601 } else {
1602 GA->setLinkage(llvm::Function::DLLExportLinkage);
1603 }
Mike Stump11289f42009-09-09 15:08:12 +00001604 } else if (D->hasAttr<WeakAttr>() ||
Rafael Espindolac18086a2010-02-23 22:00:30 +00001605 D->hasAttr<WeakRefAttr>() ||
Douglas Gregor20b2ebd2011-03-23 00:50:03 +00001606 D->isWeakImported()) {
Daniel Dunbar38932572009-04-14 08:05:55 +00001607 GA->setLinkage(llvm::Function::WeakAnyLinkage);
1608 }
1609
1610 SetCommonAttributes(D, GA);
Chris Lattner54041692009-03-22 21:47:11 +00001611}
1612
Benjamin Kramer8d375ce2011-07-14 17:45:50 +00001613llvm::Function *CodeGenModule::getIntrinsic(unsigned IID,
Chris Lattner54b16772011-07-23 17:14:25 +00001614 ArrayRef<llvm::Type*> Tys) {
Jay Foadb804a2b2011-07-12 14:06:48 +00001615 return llvm::Intrinsic::getDeclaration(&getModule(), (llvm::Intrinsic::ID)IID,
Benjamin Kramer8d375ce2011-07-14 17:45:50 +00001616 Tys);
Chris Lattnerb8be97e2007-12-18 00:25:38 +00001617}
Chris Lattner1eec6602007-08-31 04:31:45 +00001618
Daniel Dunbar64509b22009-07-23 22:52:48 +00001619static llvm::StringMapEntry<llvm::Constant*> &
1620GetConstantCFStringEntry(llvm::StringMap<llvm::Constant*> &Map,
1621 const StringLiteral *Literal,
Daniel Dunbar91ade142009-07-23 23:41:22 +00001622 bool TargetIsLSB,
Daniel Dunbar64509b22009-07-23 22:52:48 +00001623 bool &IsUTF16,
1624 unsigned &StringLength) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001625 StringRef String = Literal->getString();
Benjamin Kramer35b077e2010-08-17 12:54:38 +00001626 unsigned NumBytes = String.size();
Daniel Dunbar64509b22009-07-23 22:52:48 +00001627
Daniel Dunbarb879c3c2009-09-22 10:03:52 +00001628 // Check for simple case.
1629 if (!Literal->containsNonAsciiOrNull()) {
1630 StringLength = NumBytes;
Benjamin Kramer35b077e2010-08-17 12:54:38 +00001631 return Map.GetOrCreateValue(String);
Daniel Dunbarb879c3c2009-09-22 10:03:52 +00001632 }
1633
Daniel Dunbar64509b22009-07-23 22:52:48 +00001634 // Otherwise, convert the UTF8 literals into a byte string.
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001635 SmallVector<UTF16, 128> ToBuf(NumBytes);
Benjamin Kramer35b077e2010-08-17 12:54:38 +00001636 const UTF8 *FromPtr = (UTF8 *)String.data();
Daniel Dunbar64509b22009-07-23 22:52:48 +00001637 UTF16 *ToPtr = &ToBuf[0];
Mike Stump11289f42009-09-09 15:08:12 +00001638
Fariborz Jahanian535618b2010-09-07 19:57:04 +00001639 (void)ConvertUTF8toUTF16(&FromPtr, FromPtr + NumBytes,
1640 &ToPtr, ToPtr + NumBytes,
1641 strictConversion);
Mike Stump11289f42009-09-09 15:08:12 +00001642
Daniel Dunbar91ade142009-07-23 23:41:22 +00001643 // ConvertUTF8toUTF16 returns the length in ToPtr.
Daniel Dunbar64509b22009-07-23 22:52:48 +00001644 StringLength = ToPtr - &ToBuf[0];
Daniel Dunbar91ade142009-07-23 23:41:22 +00001645
1646 // Render the UTF-16 string into a byte array and convert to the target byte
1647 // order.
1648 //
1649 // FIXME: This isn't something we should need to do here.
1650 llvm::SmallString<128> AsBytes;
1651 AsBytes.reserve(StringLength * 2);
1652 for (unsigned i = 0; i != StringLength; ++i) {
1653 unsigned short Val = ToBuf[i];
1654 if (TargetIsLSB) {
1655 AsBytes.push_back(Val & 0xFF);
1656 AsBytes.push_back(Val >> 8);
1657 } else {
1658 AsBytes.push_back(Val >> 8);
1659 AsBytes.push_back(Val & 0xFF);
1660 }
1661 }
Daniel Dunbar4d93a4f2009-08-03 21:47:08 +00001662 // Append one extra null character, the second is automatically added by our
1663 // caller.
1664 AsBytes.push_back(0);
Daniel Dunbar91ade142009-07-23 23:41:22 +00001665
Daniel Dunbar64509b22009-07-23 22:52:48 +00001666 IsUTF16 = true;
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001667 return Map.GetOrCreateValue(StringRef(AsBytes.data(), AsBytes.size()));
Daniel Dunbar64509b22009-07-23 22:52:48 +00001668}
1669
Fariborz Jahaniana52b1f72011-05-13 18:13:10 +00001670static llvm::StringMapEntry<llvm::Constant*> &
1671GetConstantStringEntry(llvm::StringMap<llvm::Constant*> &Map,
1672 const StringLiteral *Literal,
1673 unsigned &StringLength)
1674{
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001675 StringRef String = Literal->getString();
Fariborz Jahaniana52b1f72011-05-13 18:13:10 +00001676 StringLength = String.size();
1677 return Map.GetOrCreateValue(String);
1678}
1679
Daniel Dunbar64509b22009-07-23 22:52:48 +00001680llvm::Constant *
1681CodeGenModule::GetAddrOfConstantCFString(const StringLiteral *Literal) {
1682 unsigned StringLength = 0;
1683 bool isUTF16 = false;
1684 llvm::StringMapEntry<llvm::Constant*> &Entry =
Mike Stump11289f42009-09-09 15:08:12 +00001685 GetConstantCFStringEntry(CFConstantStringMap, Literal,
Daniel Dunbar91ade142009-07-23 23:41:22 +00001686 getTargetData().isLittleEndian(),
1687 isUTF16, StringLength);
Mike Stump11289f42009-09-09 15:08:12 +00001688
Daniel Dunbar64509b22009-07-23 22:52:48 +00001689 if (llvm::Constant *C = Entry.getValue())
1690 return C;
Mike Stump11289f42009-09-09 15:08:12 +00001691
Owen Anderson41a75022009-08-13 21:57:51 +00001692 llvm::Constant *Zero =
1693 llvm::Constant::getNullValue(llvm::Type::getInt32Ty(VMContext));
Daniel Dunbard644ad62008-08-23 18:37:06 +00001694 llvm::Constant *Zeros[] = { Zero, Zero };
Mike Stump11289f42009-09-09 15:08:12 +00001695
Chris Lattneraa64ca22009-07-16 16:48:25 +00001696 // If we don't already have it, get __CFConstantStringClassReference.
Anders Carlssonb04ea612007-08-21 00:21:21 +00001697 if (!CFConstantStringClassRef) {
Chris Lattner2192fe52011-07-18 04:24:23 +00001698 llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
Owen Anderson9793f0e2009-07-29 22:16:19 +00001699 Ty = llvm::ArrayType::get(Ty, 0);
Mike Stump11289f42009-09-09 15:08:12 +00001700 llvm::Constant *GV = CreateRuntimeVariable(Ty,
Chris Lattneraa64ca22009-07-16 16:48:25 +00001701 "__CFConstantStringClassReference");
Daniel Dunbard644ad62008-08-23 18:37:06 +00001702 // Decay array -> ptr
1703 CFConstantStringClassRef =
Jay Foaded8db7d2011-07-21 14:31:17 +00001704 llvm::ConstantExpr::getGetElementPtr(GV, Zeros);
Anders Carlssonb04ea612007-08-21 00:21:21 +00001705 }
Mike Stump11289f42009-09-09 15:08:12 +00001706
Anders Carlssonc2480a52008-11-15 18:54:24 +00001707 QualType CFTy = getContext().getCFConstantStringType();
Daniel Dunbard644ad62008-08-23 18:37:06 +00001708
Chris Lattner2192fe52011-07-18 04:24:23 +00001709 llvm::StructType *STy =
Anders Carlssonc2480a52008-11-15 18:54:24 +00001710 cast<llvm::StructType>(getTypes().ConvertType(CFTy));
1711
Anders Carlsson157c3212009-08-16 05:55:31 +00001712 std::vector<llvm::Constant*> Fields(4);
Douglas Gregor91f84212008-12-11 16:49:14 +00001713
Anders Carlssonb04ea612007-08-21 00:21:21 +00001714 // Class pointer.
Anders Carlsson157c3212009-08-16 05:55:31 +00001715 Fields[0] = CFConstantStringClassRef;
Mike Stump11289f42009-09-09 15:08:12 +00001716
Anders Carlssonb04ea612007-08-21 00:21:21 +00001717 // Flags.
Chris Lattner2192fe52011-07-18 04:24:23 +00001718 llvm::Type *Ty = getTypes().ConvertType(getContext().UnsignedIntTy);
Mike Stump11289f42009-09-09 15:08:12 +00001719 Fields[1] = isUTF16 ? llvm::ConstantInt::get(Ty, 0x07d0) :
Anders Carlsson157c3212009-08-16 05:55:31 +00001720 llvm::ConstantInt::get(Ty, 0x07C8);
1721
Anders Carlssonb04ea612007-08-21 00:21:21 +00001722 // String pointer.
Owen Anderson41a75022009-08-13 21:57:51 +00001723 llvm::Constant *C = llvm::ConstantArray::get(VMContext, Entry.getKey().str());
Daniel Dunbarfd6cfcf2009-04-03 00:57:44 +00001724
Chris Lattner3afa3e12009-07-16 05:03:48 +00001725 llvm::GlobalValue::LinkageTypes Linkage;
Chris Lattner4f8a2e22009-10-14 05:55:45 +00001726 bool isConstant;
Daniel Dunbarfd6cfcf2009-04-03 00:57:44 +00001727 if (isUTF16) {
Chris Lattner4f8a2e22009-10-14 05:55:45 +00001728 // FIXME: why do utf strings get "_" labels instead of "L" labels?
Chris Lattner3afa3e12009-07-16 05:03:48 +00001729 Linkage = llvm::GlobalValue::InternalLinkage;
Chris Lattner4f8a2e22009-10-14 05:55:45 +00001730 // Note: -fwritable-strings doesn't make unicode CFStrings writable, but
1731 // does make plain ascii ones writable.
Daniel Dunbarfd6cfcf2009-04-03 00:57:44 +00001732 isConstant = true;
Chris Lattner4f8a2e22009-10-14 05:55:45 +00001733 } else {
Rafael Espindola7a6cf012011-03-14 17:55:00 +00001734 // FIXME: With OS X ld 123.2 (xcode 4) and LTO we would get a linker error
1735 // when using private linkage. It is not clear if this is a bug in ld
1736 // or a reasonable new restriction.
Rafael Espindola0c1f0982011-03-14 21:08:19 +00001737 Linkage = llvm::GlobalValue::LinkerPrivateLinkage;
Chris Lattner4f8a2e22009-10-14 05:55:45 +00001738 isConstant = !Features.WritableStrings;
Daniel Dunbarfd6cfcf2009-04-03 00:57:44 +00001739 }
Chris Lattner4f8a2e22009-10-14 05:55:45 +00001740
Mike Stump11289f42009-09-09 15:08:12 +00001741 llvm::GlobalVariable *GV =
Chris Lattner4f8a2e22009-10-14 05:55:45 +00001742 new llvm::GlobalVariable(getModule(), C->getType(), isConstant, Linkage, C,
1743 ".str");
Rafael Espindolae79d43da2011-01-17 16:31:00 +00001744 GV->setUnnamedAddr(true);
Daniel Dunbarfd6cfcf2009-04-03 00:57:44 +00001745 if (isUTF16) {
Ken Dycka0f99ff2010-01-26 18:46:23 +00001746 CharUnits Align = getContext().getTypeAlignInChars(getContext().ShortTy);
1747 GV->setAlignment(Align.getQuantity());
Daniel Dunbar9c8cd4c2011-04-12 23:30:52 +00001748 } else {
1749 CharUnits Align = getContext().getTypeAlignInChars(getContext().CharTy);
1750 GV->setAlignment(Align.getQuantity());
Daniel Dunbarfd6cfcf2009-04-03 00:57:44 +00001751 }
Jay Foaded8db7d2011-07-21 14:31:17 +00001752 Fields[2] = llvm::ConstantExpr::getGetElementPtr(GV, Zeros);
Anders Carlsson157c3212009-08-16 05:55:31 +00001753
Anders Carlssonb04ea612007-08-21 00:21:21 +00001754 // String length.
1755 Ty = getTypes().ConvertType(getContext().LongTy);
Anders Carlsson157c3212009-08-16 05:55:31 +00001756 Fields[3] = llvm::ConstantInt::get(Ty, StringLength);
Mike Stump11289f42009-09-09 15:08:12 +00001757
Anders Carlssonb04ea612007-08-21 00:21:21 +00001758 // The struct.
Owen Anderson0e0189d2009-07-27 22:29:56 +00001759 C = llvm::ConstantStruct::get(STy, Fields);
Mike Stump11289f42009-09-09 15:08:12 +00001760 GV = new llvm::GlobalVariable(getModule(), C->getType(), true,
1761 llvm::GlobalVariable::PrivateLinkage, C,
Chris Lattner3afa3e12009-07-16 05:03:48 +00001762 "_unnamed_cfstring_");
Douglas Gregore8bbc122011-09-02 00:18:52 +00001763 if (const char *Sect = getContext().getTargetInfo().getCFStringSection())
Daniel Dunbar08b216a2009-03-31 23:42:16 +00001764 GV->setSection(Sect);
Daniel Dunbar64509b22009-07-23 22:52:48 +00001765 Entry.setValue(GV);
Mike Stump11289f42009-09-09 15:08:12 +00001766
Anders Carlsson41b7c6b2007-11-01 00:41:52 +00001767 return GV;
Anders Carlssonb04ea612007-08-21 00:21:21 +00001768}
Chris Lattnerfb300092007-11-28 05:34:05 +00001769
Douglas Gregorabf4e0d2011-08-09 15:54:21 +00001770static RecordDecl *
1771CreateRecordDecl(const ASTContext &Ctx, RecordDecl::TagKind TK,
1772 DeclContext *DC, IdentifierInfo *Id) {
1773 SourceLocation Loc;
1774 if (Ctx.getLangOptions().CPlusPlus)
1775 return CXXRecordDecl::Create(Ctx, TK, DC, Loc, Loc, Id);
1776 else
1777 return RecordDecl::Create(Ctx, TK, DC, Loc, Loc, Id);
1778}
1779
Fariborz Jahanian63408e82010-04-22 20:26:39 +00001780llvm::Constant *
Fariborz Jahanian50c925f2010-10-19 17:19:29 +00001781CodeGenModule::GetAddrOfConstantString(const StringLiteral *Literal) {
Fariborz Jahaniane804c282010-04-23 17:41:07 +00001782 unsigned StringLength = 0;
Fariborz Jahaniane804c282010-04-23 17:41:07 +00001783 llvm::StringMapEntry<llvm::Constant*> &Entry =
Fariborz Jahaniana52b1f72011-05-13 18:13:10 +00001784 GetConstantStringEntry(CFConstantStringMap, Literal, StringLength);
Fariborz Jahaniane804c282010-04-23 17:41:07 +00001785
1786 if (llvm::Constant *C = Entry.getValue())
1787 return C;
1788
1789 llvm::Constant *Zero =
1790 llvm::Constant::getNullValue(llvm::Type::getInt32Ty(VMContext));
1791 llvm::Constant *Zeros[] = { Zero, Zero };
1792
Fariborz Jahaniand3fa7012010-04-23 22:33:39 +00001793 // If we don't already have it, get _NSConstantStringClassReference.
Fariborz Jahanian50c925f2010-10-19 17:19:29 +00001794 if (!ConstantStringClassRef) {
1795 std::string StringClass(getLangOptions().ObjCConstantStringClass);
Chris Lattner2192fe52011-07-18 04:24:23 +00001796 llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
Fariborz Jahanian50c925f2010-10-19 17:19:29 +00001797 llvm::Constant *GV;
Fariborz Jahanian7bd3d1c2011-05-17 22:21:16 +00001798 if (Features.ObjCNonFragileABI) {
Fariborz Jahanianccdfa392011-05-17 22:46:11 +00001799 std::string str =
1800 StringClass.empty() ? "OBJC_CLASS_$_NSConstantString"
1801 : "OBJC_CLASS_$_" + StringClass;
Fariborz Jahanian7bd3d1c2011-05-17 22:21:16 +00001802 GV = getObjCRuntime().GetClassGlobal(str);
1803 // Make sure the result is of the correct type.
Chris Lattner2192fe52011-07-18 04:24:23 +00001804 llvm::Type *PTy = llvm::PointerType::getUnqual(Ty);
Fariborz Jahanian7bd3d1c2011-05-17 22:21:16 +00001805 ConstantStringClassRef =
1806 llvm::ConstantExpr::getBitCast(GV, PTy);
1807 } else {
Fariborz Jahanianccdfa392011-05-17 22:46:11 +00001808 std::string str =
1809 StringClass.empty() ? "_NSConstantStringClassReference"
1810 : "_" + StringClass + "ClassReference";
Chris Lattner2192fe52011-07-18 04:24:23 +00001811 llvm::Type *PTy = llvm::ArrayType::get(Ty, 0);
Fariborz Jahanianccdfa392011-05-17 22:46:11 +00001812 GV = CreateRuntimeVariable(PTy, str);
Fariborz Jahanian7bd3d1c2011-05-17 22:21:16 +00001813 // Decay array -> ptr
1814 ConstantStringClassRef =
Jay Foaded8db7d2011-07-21 14:31:17 +00001815 llvm::ConstantExpr::getGetElementPtr(GV, Zeros);
Fariborz Jahanian50c925f2010-10-19 17:19:29 +00001816 }
Fariborz Jahaniane804c282010-04-23 17:41:07 +00001817 }
Douglas Gregorabf4e0d2011-08-09 15:54:21 +00001818
1819 if (!NSConstantStringType) {
1820 // Construct the type for a constant NSString.
1821 RecordDecl *D = CreateRecordDecl(Context, TTK_Struct,
1822 Context.getTranslationUnitDecl(),
1823 &Context.Idents.get("__builtin_NSString"));
1824 D->startDefinition();
1825
1826 QualType FieldTypes[3];
1827
1828 // const int *isa;
1829 FieldTypes[0] = Context.getPointerType(Context.IntTy.withConst());
1830 // const char *str;
1831 FieldTypes[1] = Context.getPointerType(Context.CharTy.withConst());
1832 // unsigned int length;
1833 FieldTypes[2] = Context.UnsignedIntTy;
1834
1835 // Create fields
1836 for (unsigned i = 0; i < 3; ++i) {
1837 FieldDecl *Field = FieldDecl::Create(Context, D,
1838 SourceLocation(),
1839 SourceLocation(), 0,
1840 FieldTypes[i], /*TInfo=*/0,
1841 /*BitWidth=*/0,
1842 /*Mutable=*/false,
1843 /*HasInit=*/false);
1844 Field->setAccess(AS_public);
1845 D->addDecl(Field);
1846 }
1847
1848 D->completeDefinition();
1849 QualType NSTy = Context.getTagDeclType(D);
1850 NSConstantStringType = cast<llvm::StructType>(getTypes().ConvertType(NSTy));
1851 }
Fariborz Jahaniane804c282010-04-23 17:41:07 +00001852
1853 std::vector<llvm::Constant*> Fields(3);
1854
1855 // Class pointer.
Fariborz Jahanian50c925f2010-10-19 17:19:29 +00001856 Fields[0] = ConstantStringClassRef;
Fariborz Jahaniane804c282010-04-23 17:41:07 +00001857
1858 // String pointer.
1859 llvm::Constant *C = llvm::ConstantArray::get(VMContext, Entry.getKey().str());
1860
1861 llvm::GlobalValue::LinkageTypes Linkage;
1862 bool isConstant;
Fariborz Jahaniana52b1f72011-05-13 18:13:10 +00001863 Linkage = llvm::GlobalValue::PrivateLinkage;
1864 isConstant = !Features.WritableStrings;
Fariborz Jahaniane804c282010-04-23 17:41:07 +00001865
1866 llvm::GlobalVariable *GV =
1867 new llvm::GlobalVariable(getModule(), C->getType(), isConstant, Linkage, C,
1868 ".str");
Rafael Espindolade089d42011-01-17 22:11:21 +00001869 GV->setUnnamedAddr(true);
Fariborz Jahaniana52b1f72011-05-13 18:13:10 +00001870 CharUnits Align = getContext().getTypeAlignInChars(getContext().CharTy);
1871 GV->setAlignment(Align.getQuantity());
Jay Foaded8db7d2011-07-21 14:31:17 +00001872 Fields[1] = llvm::ConstantExpr::getGetElementPtr(GV, Zeros);
Fariborz Jahaniane804c282010-04-23 17:41:07 +00001873
1874 // String length.
Chris Lattner2192fe52011-07-18 04:24:23 +00001875 llvm::Type *Ty = getTypes().ConvertType(getContext().UnsignedIntTy);
Fariborz Jahaniane804c282010-04-23 17:41:07 +00001876 Fields[2] = llvm::ConstantInt::get(Ty, StringLength);
1877
1878 // The struct.
Douglas Gregorabf4e0d2011-08-09 15:54:21 +00001879 C = llvm::ConstantStruct::get(NSConstantStringType, Fields);
Fariborz Jahaniane804c282010-04-23 17:41:07 +00001880 GV = new llvm::GlobalVariable(getModule(), C->getType(), true,
1881 llvm::GlobalVariable::PrivateLinkage, C,
1882 "_unnamed_nsstring_");
1883 // FIXME. Fix section.
Fariborz Jahaniand3fa7012010-04-23 22:33:39 +00001884 if (const char *Sect =
1885 Features.ObjCNonFragileABI
Douglas Gregore8bbc122011-09-02 00:18:52 +00001886 ? getContext().getTargetInfo().getNSStringNonFragileABISection()
1887 : getContext().getTargetInfo().getNSStringSection())
Fariborz Jahaniane804c282010-04-23 17:41:07 +00001888 GV->setSection(Sect);
1889 Entry.setValue(GV);
1890
1891 return GV;
Fariborz Jahanian63408e82010-04-22 20:26:39 +00001892}
1893
Douglas Gregor636e2002011-08-09 17:23:49 +00001894QualType CodeGenModule::getObjCFastEnumerationStateType() {
1895 if (ObjCFastEnumerationStateType.isNull()) {
1896 RecordDecl *D = CreateRecordDecl(Context, TTK_Struct,
1897 Context.getTranslationUnitDecl(),
1898 &Context.Idents.get("__objcFastEnumerationState"));
1899 D->startDefinition();
1900
1901 QualType FieldTypes[] = {
1902 Context.UnsignedLongTy,
1903 Context.getPointerType(Context.getObjCIdType()),
1904 Context.getPointerType(Context.UnsignedLongTy),
1905 Context.getConstantArrayType(Context.UnsignedLongTy,
1906 llvm::APInt(32, 5), ArrayType::Normal, 0)
1907 };
1908
1909 for (size_t i = 0; i < 4; ++i) {
1910 FieldDecl *Field = FieldDecl::Create(Context,
1911 D,
1912 SourceLocation(),
1913 SourceLocation(), 0,
1914 FieldTypes[i], /*TInfo=*/0,
1915 /*BitWidth=*/0,
1916 /*Mutable=*/false,
1917 /*HasInit=*/false);
1918 Field->setAccess(AS_public);
1919 D->addDecl(Field);
1920 }
1921
1922 D->completeDefinition();
1923 ObjCFastEnumerationStateType = Context.getTagDeclType(D);
1924 }
1925
1926 return ObjCFastEnumerationStateType;
1927}
1928
Daniel Dunbarc4baa062008-08-13 23:20:05 +00001929/// GetStringForStringLiteral - Return the appropriate bytes for a
Daniel Dunbar6dfdf8c2008-08-10 20:25:57 +00001930/// string literal, properly padded to match the literal type.
Daniel Dunbarc4baa062008-08-13 23:20:05 +00001931std::string CodeGenModule::GetStringForStringLiteral(const StringLiteral *E) {
Ken Dycka45a70c2011-01-29 17:53:12 +00001932 const ASTContext &Context = getContext();
Daniel Dunbar6dfdf8c2008-08-10 20:25:57 +00001933 const ConstantArrayType *CAT =
Ken Dycka45a70c2011-01-29 17:53:12 +00001934 Context.getAsConstantArrayType(E->getType());
Daniel Dunbar6dfdf8c2008-08-10 20:25:57 +00001935 assert(CAT && "String isn't pointer or array!");
Mike Stump11289f42009-09-09 15:08:12 +00001936
Chris Lattnerd42c29f2009-02-26 23:01:51 +00001937 // Resize the string to the right size.
Daniel Dunbar6dfdf8c2008-08-10 20:25:57 +00001938 uint64_t RealLen = CAT->getSize().getZExtValue();
Mike Stump11289f42009-09-09 15:08:12 +00001939
Douglas Gregorfb65e592011-07-27 05:40:30 +00001940 switch (E->getKind()) {
1941 case StringLiteral::Ascii:
1942 case StringLiteral::UTF8:
1943 break;
1944 case StringLiteral::Wide:
Douglas Gregore8bbc122011-09-02 00:18:52 +00001945 RealLen *= Context.getTargetInfo().getWCharWidth() / Context.getCharWidth();
Douglas Gregorfb65e592011-07-27 05:40:30 +00001946 break;
1947 case StringLiteral::UTF16:
Douglas Gregore8bbc122011-09-02 00:18:52 +00001948 RealLen *= Context.getTargetInfo().getChar16Width() / Context.getCharWidth();
Douglas Gregorfb65e592011-07-27 05:40:30 +00001949 break;
1950 case StringLiteral::UTF32:
Douglas Gregore8bbc122011-09-02 00:18:52 +00001951 RealLen *= Context.getTargetInfo().getChar32Width() / Context.getCharWidth();
Douglas Gregorfb65e592011-07-27 05:40:30 +00001952 break;
1953 }
Mike Stump11289f42009-09-09 15:08:12 +00001954
Benjamin Kramer35b077e2010-08-17 12:54:38 +00001955 std::string Str = E->getString().str();
Daniel Dunbar6dfdf8c2008-08-10 20:25:57 +00001956 Str.resize(RealLen, '\0');
Mike Stump11289f42009-09-09 15:08:12 +00001957
Daniel Dunbar6dfdf8c2008-08-10 20:25:57 +00001958 return Str;
1959}
1960
Daniel Dunbarc4baa062008-08-13 23:20:05 +00001961/// GetAddrOfConstantStringFromLiteral - Return a pointer to a
1962/// constant array for the given string literal.
1963llvm::Constant *
1964CodeGenModule::GetAddrOfConstantStringFromLiteral(const StringLiteral *S) {
1965 // FIXME: This can be more efficient.
Eli Friedman49ddc5f2009-11-16 05:55:46 +00001966 // FIXME: We shouldn't need to bitcast the constant in the wide string case.
John McCall9b24df42011-08-04 01:03:22 +00001967 CharUnits Align = getContext().getTypeAlignInChars(S->getType());
1968 llvm::Constant *C = GetAddrOfConstantString(GetStringForStringLiteral(S),
1969 /* GlobalName */ 0,
1970 Align.getQuantity());
Douglas Gregorfb65e592011-07-27 05:40:30 +00001971 if (S->isWide() || S->isUTF16() || S->isUTF32()) {
Eli Friedman49ddc5f2009-11-16 05:55:46 +00001972 llvm::Type *DestTy =
1973 llvm::PointerType::getUnqual(getTypes().ConvertType(S->getType()));
1974 C = llvm::ConstantExpr::getBitCast(C, DestTy);
1975 }
1976 return C;
Daniel Dunbarc4baa062008-08-13 23:20:05 +00001977}
1978
Chris Lattnerd7e7b8e2009-02-24 22:18:39 +00001979/// GetAddrOfConstantStringFromObjCEncode - Return a pointer to a constant
1980/// array for the given ObjCEncodeExpr node.
1981llvm::Constant *
1982CodeGenModule::GetAddrOfConstantStringFromObjCEncode(const ObjCEncodeExpr *E) {
1983 std::string Str;
1984 getContext().getObjCEncodingForType(E->getEncodedType(), Str);
Eli Friedman4663a332009-03-07 20:17:55 +00001985
1986 return GetAddrOfConstantCString(Str);
Chris Lattnerd7e7b8e2009-02-24 22:18:39 +00001987}
1988
1989
Chris Lattner36fc8792008-02-11 00:02:17 +00001990/// GenerateWritableString -- Creates storage for a string literal.
John McCall9b24df42011-08-04 01:03:22 +00001991static llvm::GlobalVariable *GenerateStringLiteral(StringRef str,
Chris Lattnerfb300092007-11-28 05:34:05 +00001992 bool constant,
Daniel Dunbardfcf5992008-10-17 21:56:50 +00001993 CodeGenModule &CGM,
John McCall9b24df42011-08-04 01:03:22 +00001994 const char *GlobalName,
1995 unsigned Alignment) {
Daniel Dunbarc4baa062008-08-13 23:20:05 +00001996 // Create Constant for this string literal. Don't add a '\0'.
Owen Anderson41a75022009-08-13 21:57:51 +00001997 llvm::Constant *C =
1998 llvm::ConstantArray::get(CGM.getLLVMContext(), str, false);
Mike Stump11289f42009-09-09 15:08:12 +00001999
Chris Lattnerfb300092007-11-28 05:34:05 +00002000 // Create a global variable for this string
Rafael Espindolab7f60e32011-01-10 22:34:03 +00002001 llvm::GlobalVariable *GV =
2002 new llvm::GlobalVariable(CGM.getModule(), C->getType(), constant,
2003 llvm::GlobalValue::PrivateLinkage,
2004 C, GlobalName);
John McCall9b24df42011-08-04 01:03:22 +00002005 GV->setAlignment(Alignment);
Rafael Espindolab7f60e32011-01-10 22:34:03 +00002006 GV->setUnnamedAddr(true);
2007 return GV;
Chris Lattnerfb300092007-11-28 05:34:05 +00002008}
2009
Daniel Dunbarc4baa062008-08-13 23:20:05 +00002010/// GetAddrOfConstantString - Returns a pointer to a character array
2011/// containing the literal. This contents are exactly that of the
2012/// given string, i.e. it will not be null terminated automatically;
2013/// see GetAddrOfConstantCString. Note that whether the result is
2014/// actually a pointer to an LLVM constant depends on
2015/// Feature.WriteableStrings.
2016///
2017/// The result has pointer to array type.
Chris Lattner0e62c1c2011-07-23 10:55:15 +00002018llvm::Constant *CodeGenModule::GetAddrOfConstantString(StringRef Str,
John McCall9b24df42011-08-04 01:03:22 +00002019 const char *GlobalName,
2020 unsigned Alignment) {
Daniel Dunbar08b216a2009-03-31 23:42:16 +00002021 bool IsConstant = !Features.WritableStrings;
2022
2023 // Get the default prefix if a name wasn't specified.
2024 if (!GlobalName)
Chris Lattner3afa3e12009-07-16 05:03:48 +00002025 GlobalName = ".str";
Daniel Dunbar08b216a2009-03-31 23:42:16 +00002026
2027 // Don't share any string literals if strings aren't constant.
2028 if (!IsConstant)
John McCall9b24df42011-08-04 01:03:22 +00002029 return GenerateStringLiteral(Str, false, *this, GlobalName, Alignment);
Mike Stump11289f42009-09-09 15:08:12 +00002030
John McCall9b24df42011-08-04 01:03:22 +00002031 llvm::StringMapEntry<llvm::GlobalVariable *> &Entry =
Benjamin Kramer0cd19fb2011-03-05 13:45:23 +00002032 ConstantStringMap.GetOrCreateValue(Str);
Chris Lattnerfb300092007-11-28 05:34:05 +00002033
John McCall9b24df42011-08-04 01:03:22 +00002034 if (llvm::GlobalVariable *GV = Entry.getValue()) {
2035 if (Alignment > GV->getAlignment()) {
2036 GV->setAlignment(Alignment);
2037 }
2038 return GV;
2039 }
Chris Lattnerfb300092007-11-28 05:34:05 +00002040
2041 // Create a global variable for this.
John McCall9b24df42011-08-04 01:03:22 +00002042 llvm::GlobalVariable *GV = GenerateStringLiteral(Str, true, *this, GlobalName, Alignment);
2043 Entry.setValue(GV);
2044 return GV;
Chris Lattnerfb300092007-11-28 05:34:05 +00002045}
Daniel Dunbarc4baa062008-08-13 23:20:05 +00002046
2047/// GetAddrOfConstantCString - Returns a pointer to a character
Benjamin Kramer0cd19fb2011-03-05 13:45:23 +00002048/// array containing the literal and a terminating '\0'
Daniel Dunbarc4baa062008-08-13 23:20:05 +00002049/// character. The result has pointer to array type.
Benjamin Kramer0cd19fb2011-03-05 13:45:23 +00002050llvm::Constant *CodeGenModule::GetAddrOfConstantCString(const std::string &Str,
John McCall9b24df42011-08-04 01:03:22 +00002051 const char *GlobalName,
2052 unsigned Alignment) {
Chris Lattner0e62c1c2011-07-23 10:55:15 +00002053 StringRef StrWithNull(Str.c_str(), Str.size() + 1);
John McCall9b24df42011-08-04 01:03:22 +00002054 return GetAddrOfConstantString(StrWithNull, GlobalName, Alignment);
Daniel Dunbarc4baa062008-08-13 23:20:05 +00002055}
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002056
Daniel Dunbar89654ee2008-08-26 08:29:31 +00002057/// EmitObjCPropertyImplementations - Emit information for synthesized
2058/// properties for an implementation.
Mike Stump11289f42009-09-09 15:08:12 +00002059void CodeGenModule::EmitObjCPropertyImplementations(const
Daniel Dunbar89654ee2008-08-26 08:29:31 +00002060 ObjCImplementationDecl *D) {
Mike Stump11289f42009-09-09 15:08:12 +00002061 for (ObjCImplementationDecl::propimpl_iterator
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +00002062 i = D->propimpl_begin(), e = D->propimpl_end(); i != e; ++i) {
Daniel Dunbar89654ee2008-08-26 08:29:31 +00002063 ObjCPropertyImplDecl *PID = *i;
Mike Stump11289f42009-09-09 15:08:12 +00002064
Daniel Dunbar89654ee2008-08-26 08:29:31 +00002065 // Dynamic is just for type-checking.
2066 if (PID->getPropertyImplementation() == ObjCPropertyImplDecl::Synthesize) {
2067 ObjCPropertyDecl *PD = PID->getPropertyDecl();
2068
2069 // Determine which methods need to be implemented, some may have
2070 // been overridden. Note that ::isSynthesized is not the method
2071 // we want, that just indicates if the decl came from a
2072 // property. What we want to know is if the method is defined in
2073 // this implementation.
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +00002074 if (!D->getInstanceMethod(PD->getGetterName()))
Fariborz Jahanian3d8552a2008-12-09 20:23:04 +00002075 CodeGenFunction(*this).GenerateObjCGetter(
2076 const_cast<ObjCImplementationDecl *>(D), PID);
Daniel Dunbar89654ee2008-08-26 08:29:31 +00002077 if (!PD->isReadOnly() &&
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +00002078 !D->getInstanceMethod(PD->getSetterName()))
Fariborz Jahanian3d8552a2008-12-09 20:23:04 +00002079 CodeGenFunction(*this).GenerateObjCSetter(
2080 const_cast<ObjCImplementationDecl *>(D), PID);
Daniel Dunbar89654ee2008-08-26 08:29:31 +00002081 }
2082 }
2083}
2084
John McCall6a4fa522011-03-22 07:05:39 +00002085static bool needsDestructMethod(ObjCImplementationDecl *impl) {
Jordy Rosea91768e2011-07-22 02:08:32 +00002086 const ObjCInterfaceDecl *iface = impl->getClassInterface();
2087 for (const ObjCIvarDecl *ivar = iface->all_declared_ivar_begin();
John McCall6a4fa522011-03-22 07:05:39 +00002088 ivar; ivar = ivar->getNextIvar())
2089 if (ivar->getType().isDestructedType())
2090 return true;
2091
2092 return false;
2093}
2094
Fariborz Jahanian0dec1e02010-04-28 21:28:56 +00002095/// EmitObjCIvarInitializations - Emit information for ivar initialization
2096/// for an implementation.
2097void CodeGenModule::EmitObjCIvarInitializations(ObjCImplementationDecl *D) {
John McCall6a4fa522011-03-22 07:05:39 +00002098 // We might need a .cxx_destruct even if we don't have any ivar initializers.
2099 if (needsDestructMethod(D)) {
2100 IdentifierInfo *II = &getContext().Idents.get(".cxx_destruct");
2101 Selector cxxSelector = getContext().Selectors.getSelector(0, &II);
2102 ObjCMethodDecl *DTORMethod =
2103 ObjCMethodDecl::Create(getContext(), D->getLocation(), D->getLocation(),
Argyrios Kyrtzidis004df6e2011-08-17 19:25:08 +00002104 cxxSelector, getContext().VoidTy, 0, D,
2105 /*isInstance=*/true, /*isVariadic=*/false,
2106 /*isSynthesized=*/true, /*isImplicitlyDeclared=*/true,
2107 /*isDefined=*/false, ObjCMethodDecl::Required);
John McCall6a4fa522011-03-22 07:05:39 +00002108 D->addInstanceMethod(DTORMethod);
2109 CodeGenFunction(*this).GenerateObjCCtorDtorMethod(D, DTORMethod, false);
John McCall31168b02011-06-15 23:02:42 +00002110 D->setHasCXXStructors(true);
John McCall6a4fa522011-03-22 07:05:39 +00002111 }
2112
2113 // If the implementation doesn't have any ivar initializers, we don't need
2114 // a .cxx_construct.
David Chisnalla9e54602011-03-17 14:19:08 +00002115 if (D->getNumIvarInitializers() == 0)
Fariborz Jahanian0dec1e02010-04-28 21:28:56 +00002116 return;
Fariborz Jahanian0dec1e02010-04-28 21:28:56 +00002117
John McCall6a4fa522011-03-22 07:05:39 +00002118 IdentifierInfo *II = &getContext().Idents.get(".cxx_construct");
2119 Selector cxxSelector = getContext().Selectors.getSelector(0, &II);
Fariborz Jahanian0dec1e02010-04-28 21:28:56 +00002120 // The constructor returns 'self'.
2121 ObjCMethodDecl *CTORMethod = ObjCMethodDecl::Create(getContext(),
2122 D->getLocation(),
2123 D->getLocation(), cxxSelector,
2124 getContext().getObjCIdType(), 0,
Argyrios Kyrtzidis004df6e2011-08-17 19:25:08 +00002125 D, /*isInstance=*/true,
2126 /*isVariadic=*/false,
2127 /*isSynthesized=*/true,
2128 /*isImplicitlyDeclared=*/true,
2129 /*isDefined=*/false,
Fariborz Jahanian0dec1e02010-04-28 21:28:56 +00002130 ObjCMethodDecl::Required);
2131 D->addInstanceMethod(CTORMethod);
2132 CodeGenFunction(*this).GenerateObjCCtorDtorMethod(D, CTORMethod, true);
John McCall31168b02011-06-15 23:02:42 +00002133 D->setHasCXXStructors(true);
Fariborz Jahanian0dec1e02010-04-28 21:28:56 +00002134}
2135
Anders Carlssoncbaeb9e2009-04-02 05:55:18 +00002136/// EmitNamespace - Emit all declarations in a namespace.
Anders Carlsson237f3492009-04-01 00:58:25 +00002137void CodeGenModule::EmitNamespace(const NamespaceDecl *ND) {
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +00002138 for (RecordDecl::decl_iterator I = ND->decls_begin(), E = ND->decls_end();
Anders Carlsson237f3492009-04-01 00:58:25 +00002139 I != E; ++I)
2140 EmitTopLevelDecl(*I);
2141}
2142
Anders Carlssoncbaeb9e2009-04-02 05:55:18 +00002143// EmitLinkageSpec - Emit all declarations in a linkage spec.
2144void CodeGenModule::EmitLinkageSpec(const LinkageSpecDecl *LSD) {
Eli Friedmane480ce32009-08-01 20:48:04 +00002145 if (LSD->getLanguage() != LinkageSpecDecl::lang_c &&
2146 LSD->getLanguage() != LinkageSpecDecl::lang_cxx) {
Anders Carlssoncbaeb9e2009-04-02 05:55:18 +00002147 ErrorUnsupported(LSD, "linkage spec");
2148 return;
2149 }
2150
Argyrios Kyrtzidiscfbfe782009-06-30 02:36:12 +00002151 for (RecordDecl::decl_iterator I = LSD->decls_begin(), E = LSD->decls_end();
Anders Carlssoncbaeb9e2009-04-02 05:55:18 +00002152 I != E; ++I)
2153 EmitTopLevelDecl(*I);
2154}
2155
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002156/// EmitTopLevelDecl - Emit code for a single top level declaration.
2157void CodeGenModule::EmitTopLevelDecl(Decl *D) {
2158 // If an error has occurred, stop code generation, but continue
2159 // parsing and semantic analysis (to ensure all warnings and errors
2160 // are emitted).
2161 if (Diags.hasErrorOccurred())
2162 return;
2163
Douglas Gregor70d83e22009-06-29 17:30:29 +00002164 // Ignore dependent declarations.
2165 if (D->getDeclContext() && D->getDeclContext()->isDependentContext())
2166 return;
Mike Stump11289f42009-09-09 15:08:12 +00002167
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002168 switch (D->getKind()) {
Anders Carlsson6c0a6e42009-08-25 13:14:46 +00002169 case Decl::CXXConversion:
Anders Carlsson468fa632009-04-04 20:47:02 +00002170 case Decl::CXXMethod:
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002171 case Decl::Function:
Douglas Gregor70d83e22009-06-29 17:30:29 +00002172 // Skip function templates
Francois Pichet1c229c02011-04-22 22:18:13 +00002173 if (cast<FunctionDecl>(D)->getDescribedFunctionTemplate() ||
2174 cast<FunctionDecl>(D)->isLateTemplateParsed())
Douglas Gregor70d83e22009-06-29 17:30:29 +00002175 return;
Mike Stump11289f42009-09-09 15:08:12 +00002176
Anders Carlssonecf9bf02009-09-10 23:43:36 +00002177 EmitGlobal(cast<FunctionDecl>(D));
2178 break;
2179
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002180 case Decl::Var:
Anders Carlssonecf9bf02009-09-10 23:43:36 +00002181 EmitGlobal(cast<VarDecl>(D));
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002182 break;
2183
John McCall32f44bd2011-04-12 01:01:22 +00002184 // Indirect fields from global anonymous structs and unions can be
2185 // ignored; only the actual variable requires IR gen support.
2186 case Decl::IndirectField:
2187 break;
2188
Anders Carlssonf7475242009-04-15 15:55:24 +00002189 // C++ Decls
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002190 case Decl::Namespace:
Anders Carlsson237f3492009-04-01 00:58:25 +00002191 EmitNamespace(cast<NamespaceDecl>(D));
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002192 break;
Douglas Gregorfec52632009-06-20 00:51:54 +00002193 // No code generation needed.
John McCall1e9de052009-11-17 09:33:40 +00002194 case Decl::UsingShadow:
Douglas Gregorfec52632009-06-20 00:51:54 +00002195 case Decl::Using:
Douglas Gregore5feb512009-09-02 23:49:23 +00002196 case Decl::UsingDirective:
Douglas Gregor8f5d4422009-06-29 20:59:39 +00002197 case Decl::ClassTemplate:
2198 case Decl::FunctionTemplate:
Richard Smith3f1b5d02011-05-05 21:57:07 +00002199 case Decl::TypeAliasTemplate:
Anders Carlsson649a17e2009-09-23 19:19:16 +00002200 case Decl::NamespaceAlias:
Douglas Gregor0aa91e02011-06-05 05:04:23 +00002201 case Decl::Block:
Douglas Gregorfec52632009-06-20 00:51:54 +00002202 break;
Anders Carlssonf7475242009-04-15 15:55:24 +00002203 case Decl::CXXConstructor:
Anders Carlsson0ade9712009-11-24 05:16:24 +00002204 // Skip function templates
Francois Pichet1c229c02011-04-22 22:18:13 +00002205 if (cast<FunctionDecl>(D)->getDescribedFunctionTemplate() ||
2206 cast<FunctionDecl>(D)->isLateTemplateParsed())
Anders Carlsson0ade9712009-11-24 05:16:24 +00002207 return;
2208
Anders Carlssonf7475242009-04-15 15:55:24 +00002209 EmitCXXConstructors(cast<CXXConstructorDecl>(D));
2210 break;
Anders Carlssoneaa28f72009-04-17 01:58:57 +00002211 case Decl::CXXDestructor:
Francois Pichet1c229c02011-04-22 22:18:13 +00002212 if (cast<FunctionDecl>(D)->isLateTemplateParsed())
2213 return;
Anders Carlssoneaa28f72009-04-17 01:58:57 +00002214 EmitCXXDestructors(cast<CXXDestructorDecl>(D));
2215 break;
Anders Carlsson87835432009-06-11 21:22:55 +00002216
2217 case Decl::StaticAssert:
2218 // Nothing to do.
2219 break;
2220
Anders Carlssonf7475242009-04-15 15:55:24 +00002221 // Objective-C Decls
Mike Stump11289f42009-09-09 15:08:12 +00002222
Fariborz Jahanian3654e652009-03-18 22:33:24 +00002223 // Forward declarations, no (immediate) code generation.
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002224 case Decl::ObjCClass:
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002225 case Decl::ObjCForwardProtocol:
Chris Lattnerd18136a2009-04-01 02:36:43 +00002226 case Decl::ObjCInterface:
Chris Lattnerd18136a2009-04-01 02:36:43 +00002227 break;
Fariborz Jahanianbf9294f2010-08-23 18:51:39 +00002228
Chris Lattnere4ec5ab2011-04-09 07:11:53 +00002229 case Decl::ObjCCategory: {
2230 ObjCCategoryDecl *CD = cast<ObjCCategoryDecl>(D);
2231 if (CD->IsClassExtension() && CD->hasSynthBitfield())
2232 Context.ResetObjCLayout(CD->getClassInterface());
2233 break;
2234 }
Chris Lattnerd18136a2009-04-01 02:36:43 +00002235
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002236 case Decl::ObjCProtocol:
Peter Collingbournee1d20992011-07-27 20:29:46 +00002237 ObjCRuntime->GenerateProtocol(cast<ObjCProtocolDecl>(D));
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002238 break;
2239
2240 case Decl::ObjCCategoryImpl:
Daniel Dunbar89654ee2008-08-26 08:29:31 +00002241 // Categories have properties but don't support synthesize so we
2242 // can ignore them here.
Peter Collingbournee1d20992011-07-27 20:29:46 +00002243 ObjCRuntime->GenerateCategory(cast<ObjCCategoryImplDecl>(D));
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002244 break;
2245
Daniel Dunbar89654ee2008-08-26 08:29:31 +00002246 case Decl::ObjCImplementation: {
2247 ObjCImplementationDecl *OMD = cast<ObjCImplementationDecl>(D);
Fariborz Jahanianbf9294f2010-08-23 18:51:39 +00002248 if (Features.ObjCNonFragileABI2 && OMD->hasSynthBitfield())
2249 Context.ResetObjCLayout(OMD->getClassInterface());
Daniel Dunbar89654ee2008-08-26 08:29:31 +00002250 EmitObjCPropertyImplementations(OMD);
Fariborz Jahanian0dec1e02010-04-28 21:28:56 +00002251 EmitObjCIvarInitializations(OMD);
Peter Collingbournee1d20992011-07-27 20:29:46 +00002252 ObjCRuntime->GenerateClass(OMD);
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002253 break;
Mike Stump11289f42009-09-09 15:08:12 +00002254 }
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002255 case Decl::ObjCMethod: {
2256 ObjCMethodDecl *OMD = cast<ObjCMethodDecl>(D);
2257 // If this is not a prototype, emit the body.
Argyrios Kyrtzidisddcd1322009-06-30 02:35:26 +00002258 if (OMD->getBody())
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002259 CodeGenFunction(*this).GenerateObjCMethod(OMD);
2260 break;
2261 }
Mike Stump11289f42009-09-09 15:08:12 +00002262 case Decl::ObjCCompatibleAlias:
Fariborz Jahanian17290c32009-01-08 01:10:55 +00002263 // compatibility-alias is a directive and has no code gen.
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002264 break;
2265
Anders Carlssoncbaeb9e2009-04-02 05:55:18 +00002266 case Decl::LinkageSpec:
2267 EmitLinkageSpec(cast<LinkageSpecDecl>(D));
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002268 break;
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002269
2270 case Decl::FileScopeAsm: {
2271 FileScopeAsmDecl *AD = cast<FileScopeAsmDecl>(D);
Chris Lattner0e62c1c2011-07-23 10:55:15 +00002272 StringRef AsmString = AD->getAsmString()->getString();
Mike Stump11289f42009-09-09 15:08:12 +00002273
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002274 const std::string &S = getModule().getModuleInlineAsm();
2275 if (S.empty())
2276 getModule().setModuleInlineAsm(AsmString);
Chris Lattner84037d32011-07-23 20:04:25 +00002277 else if (*--S.end() == '\n')
2278 getModule().setModuleInlineAsm(S + AsmString.str());
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002279 else
Benjamin Kramerb11118b2009-12-11 13:33:18 +00002280 getModule().setModuleInlineAsm(S + '\n' + AsmString.str());
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002281 break;
2282 }
Mike Stump11289f42009-09-09 15:08:12 +00002283
2284 default:
Mike Stump18bb9282009-05-16 07:57:57 +00002285 // Make sure we handled everything we should, every other kind is a
2286 // non-top-level decl. FIXME: Would be nice to have an isTopLevelDeclKind
2287 // function. Need to recode Decl::Kind to do that easily.
Daniel Dunbarfce4be82008-08-15 23:26:23 +00002288 assert(isa<TypeDecl>(D) && "Unsupported decl kind");
2289 }
2290}
John McCall09ae0322010-07-06 23:57:41 +00002291
2292/// Turns the given pointer into a constant.
2293static llvm::Constant *GetPointerConstant(llvm::LLVMContext &Context,
2294 const void *Ptr) {
2295 uintptr_t PtrInt = reinterpret_cast<uintptr_t>(Ptr);
Chris Lattner2192fe52011-07-18 04:24:23 +00002296 llvm::Type *i64 = llvm::Type::getInt64Ty(Context);
John McCall09ae0322010-07-06 23:57:41 +00002297 return llvm::ConstantInt::get(i64, PtrInt);
2298}
2299
2300static void EmitGlobalDeclMetadata(CodeGenModule &CGM,
2301 llvm::NamedMDNode *&GlobalMetadata,
2302 GlobalDecl D,
2303 llvm::GlobalValue *Addr) {
2304 if (!GlobalMetadata)
2305 GlobalMetadata =
2306 CGM.getModule().getOrInsertNamedMetadata("clang.global.decl.ptrs");
2307
2308 // TODO: should we report variant information for ctors/dtors?
2309 llvm::Value *Ops[] = {
2310 Addr,
2311 GetPointerConstant(CGM.getLLVMContext(), D.getDecl())
2312 };
Jay Foadea324f12011-04-21 19:59:12 +00002313 GlobalMetadata->addOperand(llvm::MDNode::get(CGM.getLLVMContext(), Ops));
John McCall09ae0322010-07-06 23:57:41 +00002314}
2315
2316/// Emits metadata nodes associating all the global values in the
2317/// current module with the Decls they came from. This is useful for
2318/// projects using IR gen as a subroutine.
2319///
2320/// Since there's currently no way to associate an MDNode directly
2321/// with an llvm::GlobalValue, we create a global named metadata
2322/// with the name 'clang.global.decl.ptrs'.
2323void CodeGenModule::EmitDeclMetadata() {
2324 llvm::NamedMDNode *GlobalMetadata = 0;
2325
2326 // StaticLocalDeclMap
Chris Lattner0e62c1c2011-07-23 10:55:15 +00002327 for (llvm::DenseMap<GlobalDecl,StringRef>::iterator
John McCall09ae0322010-07-06 23:57:41 +00002328 I = MangledDeclNames.begin(), E = MangledDeclNames.end();
2329 I != E; ++I) {
2330 llvm::GlobalValue *Addr = getModule().getNamedValue(I->second);
2331 EmitGlobalDeclMetadata(*this, GlobalMetadata, I->first, Addr);
2332 }
2333}
2334
2335/// Emits metadata nodes for all the local variables in the current
2336/// function.
2337void CodeGenFunction::EmitDeclMetadata() {
2338 if (LocalDeclMap.empty()) return;
2339
2340 llvm::LLVMContext &Context = getLLVMContext();
2341
2342 // Find the unique metadata ID for this name.
2343 unsigned DeclPtrKind = Context.getMDKindID("clang.decl.ptr");
2344
2345 llvm::NamedMDNode *GlobalMetadata = 0;
2346
2347 for (llvm::DenseMap<const Decl*, llvm::Value*>::iterator
2348 I = LocalDeclMap.begin(), E = LocalDeclMap.end(); I != E; ++I) {
2349 const Decl *D = I->first;
2350 llvm::Value *Addr = I->second;
2351
2352 if (llvm::AllocaInst *Alloca = dyn_cast<llvm::AllocaInst>(Addr)) {
2353 llvm::Value *DAddr = GetPointerConstant(getLLVMContext(), D);
Jay Foadea324f12011-04-21 19:59:12 +00002354 Alloca->setMetadata(DeclPtrKind, llvm::MDNode::get(Context, DAddr));
John McCall09ae0322010-07-06 23:57:41 +00002355 } else if (llvm::GlobalValue *GV = dyn_cast<llvm::GlobalValue>(Addr)) {
2356 GlobalDecl GD = GlobalDecl(cast<VarDecl>(D));
2357 EmitGlobalDeclMetadata(CGM, GlobalMetadata, GD, GV);
2358 }
2359 }
2360}
Daniel Dunbar900546d2010-07-16 00:00:15 +00002361
Nick Lewycky85c011d2011-05-05 00:08:20 +00002362void CodeGenModule::EmitCoverageFile() {
2363 if (!getCodeGenOpts().CoverageFile.empty()) {
Nick Lewycky480cb992011-05-04 20:46:58 +00002364 if (llvm::NamedMDNode *CUNode = TheModule.getNamedMetadata("llvm.dbg.cu")) {
2365 llvm::NamedMDNode *GCov = TheModule.getOrInsertNamedMetadata("llvm.gcov");
2366 llvm::LLVMContext &Ctx = TheModule.getContext();
Nick Lewycky85c011d2011-05-05 00:08:20 +00002367 llvm::MDString *CoverageFile =
2368 llvm::MDString::get(Ctx, getCodeGenOpts().CoverageFile);
Nick Lewycky480cb992011-05-04 20:46:58 +00002369 for (int i = 0, e = CUNode->getNumOperands(); i != e; ++i) {
2370 llvm::MDNode *CU = CUNode->getOperand(i);
Nick Lewycky85c011d2011-05-05 00:08:20 +00002371 llvm::Value *node[] = { CoverageFile, CU };
Nick Lewycky480cb992011-05-04 20:46:58 +00002372 llvm::MDNode *N = llvm::MDNode::get(Ctx, node);
2373 GCov->addOperand(N);
2374 }
2375 }
2376 }
2377}