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