contrib/llvm/tools/clang/lib/CodeGen/CodeGenModule.cpp

   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 "CGBlocks.h"
  16 #include "CGCUDARuntime.h"
  17 #include "CGCXXABI.h"
  18 #include "CGCall.h"
  19 #include "CGDebugInfo.h"
  20 #include "CGObjCRuntime.h"
  21 #include "CGOpenCLRuntime.h"
  22 #include "CGOpenMPRuntime.h"
  23 #include "CGOpenMPRuntimeNVPTX.h"
  24 #include "CodeGenFunction.h"
  25 #include "CodeGenPGO.h"
  26 #include "CodeGenTBAA.h"
  27 #include "ConstantBuilder.h"
  28 #include "CoverageMappingGen.h"
  29 #include "TargetInfo.h"
  30 #include "clang/AST/ASTContext.h"
  31 #include "clang/AST/CharUnits.h"
  32 #include "clang/AST/DeclCXX.h"
  33 #include "clang/AST/DeclObjC.h"
  34 #include "clang/AST/DeclTemplate.h"
  35 #include "clang/AST/Mangle.h"
  36 #include "clang/AST/RecordLayout.h"
  37 #include "clang/AST/RecursiveASTVisitor.h"
  38 #include "clang/Basic/Builtins.h"
  39 #include "clang/Basic/CharInfo.h"
  40 #include "clang/Basic/Diagnostic.h"
  41 #include "clang/Basic/Module.h"
  42 #include "clang/Basic/SourceManager.h"
  43 #include "clang/Basic/TargetInfo.h"
  44 #include "clang/Basic/Version.h"
  45 #include "clang/Frontend/CodeGenOptions.h"
  46 #include "clang/Sema/SemaDiagnostic.h"
  47 #include "llvm/ADT/Triple.h"
  48 #include "llvm/IR/CallSite.h"
  49 #include "llvm/IR/CallingConv.h"
  50 #include "llvm/IR/DataLayout.h"
  51 #include "llvm/IR/Intrinsics.h"
  52 #include "llvm/IR/LLVMContext.h"
  53 #include "llvm/IR/Module.h"
  54 #include "llvm/ProfileData/InstrProfReader.h"
  55 #include "llvm/Support/ConvertUTF.h"
  56 #include "llvm/Support/ErrorHandling.h"
  57 #include "llvm/Support/MD5.h"
  58
  59 using namespace clang;
  60 using namespace CodeGen;
  61
  62 static const char AnnotationSection[] = "llvm.metadata";
  63
  64 static CGCXXABI *createCXXABI(CodeGenModule &CGM) {
  65   switch (CGM.getTarget().getCXXABI().getKind()) {
  66   case TargetCXXABI::GenericAArch64:
  67   case TargetCXXABI::GenericARM:
  68   case TargetCXXABI::iOS:
  69   case TargetCXXABI::iOS64:
  70   case TargetCXXABI::WatchOS:
  71   case TargetCXXABI::GenericMIPS:
  72   case TargetCXXABI::GenericItanium:
  73   case TargetCXXABI::WebAssembly:
  74     return CreateItaniumCXXABI(CGM);
  75   case TargetCXXABI::Microsoft:
  76     return CreateMicrosoftCXXABI(CGM);
  77   }
  78
  79   llvm_unreachable("invalid C++ ABI kind");
  80 }
  81
  82 CodeGenModule::CodeGenModule(ASTContext &C, const HeaderSearchOptions &HSO,
  83                              const PreprocessorOptions &PPO,
  84                              const CodeGenOptions &CGO, llvm::Module &M,
  85                              DiagnosticsEngine &diags,
  86                              CoverageSourceInfo *CoverageInfo)
  87     : Context(C), LangOpts(C.getLangOpts()), HeaderSearchOpts(HSO),
  88       PreprocessorOpts(PPO), CodeGenOpts(CGO), TheModule(M), Diags(diags),
  89       Target(C.getTargetInfo()), ABI(createCXXABI(*this)),
  90       VMContext(M.getContext()), Types(*this), VTables(*this),
  91       SanitizerMD(new SanitizerMetadata(*this)) {
  92
  93   // Initialize the type cache.
  94   llvm::LLVMContext &LLVMContext = M.getContext();
  95   VoidTy = llvm::Type::getVoidTy(LLVMContext);
  96   Int8Ty = llvm::Type::getInt8Ty(LLVMContext);
  97   Int16Ty = llvm::Type::getInt16Ty(LLVMContext);
  98   Int32Ty = llvm::Type::getInt32Ty(LLVMContext);
  99   Int64Ty = llvm::Type::getInt64Ty(LLVMContext);
 100   FloatTy = llvm::Type::getFloatTy(LLVMContext);
 101   DoubleTy = llvm::Type::getDoubleTy(LLVMContext);
 102   PointerWidthInBits = C.getTargetInfo().getPointerWidth(0);
 103   PointerAlignInBytes =
 104     C.toCharUnitsFromBits(C.getTargetInfo().getPointerAlign(0)).getQuantity();
 105   SizeSizeInBytes =
 106     C.toCharUnitsFromBits(C.getTargetInfo().getMaxPointerWidth()).getQuantity();
 107   IntAlignInBytes =
 108     C.toCharUnitsFromBits(C.getTargetInfo().getIntAlign()).getQuantity();
 109   IntTy = llvm::IntegerType::get(LLVMContext, C.getTargetInfo().getIntWidth());
 110   IntPtrTy = llvm::IntegerType::get(LLVMContext,
 111     C.getTargetInfo().getMaxPointerWidth());
 112   Int8PtrTy = Int8Ty->getPointerTo(0);
 113   Int8PtrPtrTy = Int8PtrTy->getPointerTo(0);
 114
 115   RuntimeCC = getTargetCodeGenInfo().getABIInfo().getRuntimeCC();
 116   BuiltinCC = getTargetCodeGenInfo().getABIInfo().getBuiltinCC();
 117
 118   if (LangOpts.ObjC1)
 119     createObjCRuntime();
 120   if (LangOpts.OpenCL)
 121     createOpenCLRuntime();
 122   if (LangOpts.OpenMP)
 123     createOpenMPRuntime();
 124   if (LangOpts.CUDA)
 125     createCUDARuntime();
 126
 127   // Enable TBAA unless it's suppressed. ThreadSanitizer needs TBAA even at O0.
 128   if (LangOpts.Sanitize.has(SanitizerKind::Thread) ||
 129       (!CodeGenOpts.RelaxedAliasing && CodeGenOpts.OptimizationLevel > 0))
 130     TBAA.reset(new CodeGenTBAA(Context, VMContext, CodeGenOpts, getLangOpts(),
 131                                getCXXABI().getMangleContext()));
 132
 133   // If debug info or coverage generation is enabled, create the CGDebugInfo
 134   // object.
 135   if (CodeGenOpts.getDebugInfo() != codegenoptions::NoDebugInfo ||
 136       CodeGenOpts.EmitGcovArcs || CodeGenOpts.EmitGcovNotes)
 137     DebugInfo.reset(new CGDebugInfo(*this));
 138
 139   Block.GlobalUniqueCount = 0;
 140
 141   if (C.getLangOpts().ObjC1)
 142     ObjCData.reset(new ObjCEntrypoints());
 143
 144   if (CodeGenOpts.hasProfileClangUse()) {
 145     auto ReaderOrErr = llvm::IndexedInstrProfReader::create(
 146         CodeGenOpts.ProfileInstrumentUsePath);
 147     if (auto E = ReaderOrErr.takeError()) {
 148       unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
 149                                               "Could not read profile %0: %1");
 150       llvm::handleAllErrors(std::move(E), [&](const llvm::ErrorInfoBase &EI) {
 151         getDiags().Report(DiagID) << CodeGenOpts.ProfileInstrumentUsePath
 152                                   << EI.message();
 153       });
 154     } else
 155       PGOReader = std::move(ReaderOrErr.get());
 156   }
 157
 158   // If coverage mapping generation is enabled, create the
 159   // CoverageMappingModuleGen object.
 160   if (CodeGenOpts.CoverageMapping)
 161     CoverageMapping.reset(new CoverageMappingModuleGen(*this, *CoverageInfo));
 162 }
 163
 164 CodeGenModule::~CodeGenModule() {}
 165
 166 void CodeGenModule::createObjCRuntime() {
 167   // This is just isGNUFamily(), but we want to force implementors of
 168   // new ABIs to decide how best to do this.
 169   switch (LangOpts.ObjCRuntime.getKind()) {
 170   case ObjCRuntime::GNUstep:
 171   case ObjCRuntime::GCC:
 172   case ObjCRuntime::ObjFW:
 173     ObjCRuntime.reset(CreateGNUObjCRuntime(*this));
 174     return;
 175
 176   case ObjCRuntime::FragileMacOSX:
 177   case ObjCRuntime::MacOSX:
 178   case ObjCRuntime::iOS:
 179   case ObjCRuntime::WatchOS:
 180     ObjCRuntime.reset(CreateMacObjCRuntime(*this));
 181     return;
 182   }
 183   llvm_unreachable("bad runtime kind");
 184 }
 185
 186 void CodeGenModule::createOpenCLRuntime() {
 187   OpenCLRuntime.reset(new CGOpenCLRuntime(*this));
 188 }
 189
 190 void CodeGenModule::createOpenMPRuntime() {
 191   // Select a specialized code generation class based on the target, if any.
 192   // If it does not exist use the default implementation.
 193   switch (getTriple().getArch()) {
 194   case llvm::Triple::nvptx:
 195   case llvm::Triple::nvptx64:
 196     assert(getLangOpts().OpenMPIsDevice &&
 197            "OpenMP NVPTX is only prepared to deal with device code.");
 198     OpenMPRuntime.reset(new CGOpenMPRuntimeNVPTX(*this));
 199     break;
 200   default:
 201     OpenMPRuntime.reset(new CGOpenMPRuntime(*this));
 202     break;
 203   }
 204 }
 205
 206 void CodeGenModule::createCUDARuntime() {
 207   CUDARuntime.reset(CreateNVCUDARuntime(*this));
 208 }
 209
 210 void CodeGenModule::addReplacement(StringRef Name, llvm::Constant *C) {
 211   Replacements[Name] = C;
 212 }
 213
 214 void CodeGenModule::applyReplacements() {
 215   for (auto &I : Replacements) {
 216     StringRef MangledName = I.first();
 217     llvm::Constant *Replacement = I.second;
 218     llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
 219     if (!Entry)
 220       continue;
 221     auto *OldF = cast<llvm::Function>(Entry);
 222     auto *NewF = dyn_cast<llvm::Function>(Replacement);
 223     if (!NewF) {
 224       if (auto *Alias = dyn_cast<llvm::GlobalAlias>(Replacement)) {
 225         NewF = dyn_cast<llvm::Function>(Alias->getAliasee());
 226       } else {
 227         auto *CE = cast<llvm::ConstantExpr>(Replacement);
 228         assert(CE->getOpcode() == llvm::Instruction::BitCast ||
 229                CE->getOpcode() == llvm::Instruction::GetElementPtr);
 230         NewF = dyn_cast<llvm::Function>(CE->getOperand(0));
 231       }
 232     }
 233
 234     // Replace old with new, but keep the old order.
 235     OldF->replaceAllUsesWith(Replacement);
 236     if (NewF) {
 237       NewF->removeFromParent();
 238       OldF->getParent()->getFunctionList().insertAfter(OldF->getIterator(),
 239                                                        NewF);
 240     }
 241     OldF->eraseFromParent();
 242   }
 243 }
 244
 245 void CodeGenModule::addGlobalValReplacement(llvm::GlobalValue *GV, llvm::Constant *C) {
 246   GlobalValReplacements.push_back(std::make_pair(GV, C));
 247 }
 248
 249 void CodeGenModule::applyGlobalValReplacements() {
 250   for (auto &I : GlobalValReplacements) {
 251     llvm::GlobalValue *GV = I.first;
 252     llvm::Constant *C = I.second;
 253
 254     GV->replaceAllUsesWith(C);
 255     GV->eraseFromParent();
 256   }
 257 }
 258
 259 // This is only used in aliases that we created and we know they have a
 260 // linear structure.
 261 static const llvm::GlobalObject *getAliasedGlobal(
 262     const llvm::GlobalIndirectSymbol &GIS) {
 263   llvm::SmallPtrSet<const llvm::GlobalIndirectSymbol*, 4> Visited;
 264   const llvm::Constant *C = &GIS;
 265   for (;;) {
 266     C = C->stripPointerCasts();
 267     if (auto *GO = dyn_cast<llvm::GlobalObject>(C))
 268       return GO;
 269     // stripPointerCasts will not walk over weak aliases.
 270     auto *GIS2 = dyn_cast<llvm::GlobalIndirectSymbol>(C);
 271     if (!GIS2)
 272       return nullptr;
 273     if (!Visited.insert(GIS2).second)
 274       return nullptr;
 275     C = GIS2->getIndirectSymbol();
 276   }
 277 }
 278
 279 void CodeGenModule::checkAliases() {
 280   // Check if the constructed aliases are well formed. It is really unfortunate
 281   // that we have to do this in CodeGen, but we only construct mangled names
 282   // and aliases during codegen.
 283   bool Error = false;
 284   DiagnosticsEngine &Diags = getDiags();
 285   for (const GlobalDecl &GD : Aliases) {
 286     const auto *D = cast<ValueDecl>(GD.getDecl());
 287     SourceLocation Location;
 288     bool IsIFunc = D->hasAttr<IFuncAttr>();
 289     if (const Attr *A = D->getDefiningAttr())
 290       Location = A->getLocation();
 291     else
 292       llvm_unreachable("Not an alias or ifunc?");
 293     StringRef MangledName = getMangledName(GD);
 294     llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
 295     auto *Alias  = cast<llvm::GlobalIndirectSymbol>(Entry);
 296     const llvm::GlobalValue *GV = getAliasedGlobal(*Alias);
 297     if (!GV) {
 298       Error = true;
 299       Diags.Report(Location, diag::err_cyclic_alias) << IsIFunc;
 300     } else if (GV->isDeclaration()) {
 301       Error = true;
 302       Diags.Report(Location, diag::err_alias_to_undefined)
 303           << IsIFunc << IsIFunc;
 304     } else if (IsIFunc) {
 305       // Check resolver function type.
 306       llvm::FunctionType *FTy = dyn_cast<llvm::FunctionType>(
 307           GV->getType()->getPointerElementType());
 308       assert(FTy);
 309       if (!FTy->getReturnType()->isPointerTy())
 310         Diags.Report(Location, diag::err_ifunc_resolver_return);
 311       if (FTy->getNumParams())
 312         Diags.Report(Location, diag::err_ifunc_resolver_params);
 313     }
 314
 315     llvm::Constant *Aliasee = Alias->getIndirectSymbol();
 316     llvm::GlobalValue *AliaseeGV;
 317     if (auto CE = dyn_cast<llvm::ConstantExpr>(Aliasee))
 318       AliaseeGV = cast<llvm::GlobalValue>(CE->getOperand(0));
 319     else
 320       AliaseeGV = cast<llvm::GlobalValue>(Aliasee);
 321
 322     if (const SectionAttr *SA = D->getAttr<SectionAttr>()) {
 323       StringRef AliasSection = SA->getName();
 324       if (AliasSection != AliaseeGV->getSection())
 325         Diags.Report(SA->getLocation(), diag::warn_alias_with_section)
 326             << AliasSection << IsIFunc << IsIFunc;
 327     }
 328
 329     // We have to handle alias to weak aliases in here. LLVM itself disallows
 330     // this since the object semantics would not match the IL one. For
 331     // compatibility with gcc we implement it by just pointing the alias
 332     // to its aliasee's aliasee. We also warn, since the user is probably
 333     // expecting the link to be weak.
 334     if (auto GA = dyn_cast<llvm::GlobalIndirectSymbol>(AliaseeGV)) {
 335       if (GA->isInterposable()) {
 336         Diags.Report(Location, diag::warn_alias_to_weak_alias)
 337             << GV->getName() << GA->getName() << IsIFunc;
 338         Aliasee = llvm::ConstantExpr::getPointerBitCastOrAddrSpaceCast(
 339             GA->getIndirectSymbol(), Alias->getType());
 340         Alias->setIndirectSymbol(Aliasee);
 341       }
 342     }
 343   }
 344   if (!Error)
 345     return;
 346
 347   for (const GlobalDecl &GD : Aliases) {
 348     StringRef MangledName = getMangledName(GD);
 349     llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
 350     auto *Alias = dyn_cast<llvm::GlobalIndirectSymbol>(Entry);
 351     Alias->replaceAllUsesWith(llvm::UndefValue::get(Alias->getType()));
 352     Alias->eraseFromParent();
 353   }
 354 }
 355
 356 void CodeGenModule::clear() {
 357   DeferredDeclsToEmit.clear();
 358   if (OpenMPRuntime)
 359     OpenMPRuntime->clear();
 360 }
 361
 362 void InstrProfStats::reportDiagnostics(DiagnosticsEngine &Diags,
 363                                        StringRef MainFile) {
 364   if (!hasDiagnostics())
 365     return;
 366   if (VisitedInMainFile > 0 && VisitedInMainFile == MissingInMainFile) {
 367     if (MainFile.empty())
 368       MainFile = "<stdin>";
 369     Diags.Report(diag::warn_profile_data_unprofiled) << MainFile;
 370   } else
 371     Diags.Report(diag::warn_profile_data_out_of_date) << Visited << Missing
 372                                                       << Mismatched;
 373 }
 374
 375 void CodeGenModule::Release() {
 376   EmitDeferred();
 377   applyGlobalValReplacements();
 378   applyReplacements();
 379   checkAliases();
 380   EmitCXXGlobalInitFunc();
 381   EmitCXXGlobalDtorFunc();
 382   EmitCXXThreadLocalInitFunc();
 383   if (ObjCRuntime)
 384     if (llvm::Function *ObjCInitFunction = ObjCRuntime->ModuleInitFunction())
 385       AddGlobalCtor(ObjCInitFunction);
 386   if (Context.getLangOpts().CUDA && !Context.getLangOpts().CUDAIsDevice &&
 387       CUDARuntime) {
 388     if (llvm::Function *CudaCtorFunction = CUDARuntime->makeModuleCtorFunction())
 389       AddGlobalCtor(CudaCtorFunction);
 390     if (llvm::Function *CudaDtorFunction = CUDARuntime->makeModuleDtorFunction())
 391       AddGlobalDtor(CudaDtorFunction);
 392   }
 393   if (OpenMPRuntime)
 394     if (llvm::Function *OpenMPRegistrationFunction =
 395             OpenMPRuntime->emitRegistrationFunction())
 396       AddGlobalCtor(OpenMPRegistrationFunction, 0);
 397   if (PGOReader) {
 398     getModule().setProfileSummary(PGOReader->getSummary().getMD(VMContext));
 399     if (PGOStats.hasDiagnostics())
 400       PGOStats.reportDiagnostics(getDiags(), getCodeGenOpts().MainFileName);
 401   }
 402   EmitCtorList(GlobalCtors, "llvm.global_ctors");
 403   EmitCtorList(GlobalDtors, "llvm.global_dtors");
 404   EmitGlobalAnnotations();
 405   EmitStaticExternCAliases();
 406   EmitDeferredUnusedCoverageMappings();
 407   if (CoverageMapping)
 408     CoverageMapping->emit();
 409   if (CodeGenOpts.SanitizeCfiCrossDso)
 410     CodeGenFunction(*this).EmitCfiCheckFail();
 411   emitLLVMUsed();
 412   if (SanStats)
 413     SanStats->finish();
 414
 415   if (CodeGenOpts.Autolink &&
 416       (Context.getLangOpts().Modules || !LinkerOptionsMetadata.empty())) {
 417     EmitModuleLinkOptions();
 418   }
 419   if (CodeGenOpts.DwarfVersion) {
 420     // We actually want the latest version when there are conflicts.
 421     // We can change from Warning to Latest if such mode is supported.
 422     getModule().addModuleFlag(llvm::Module::Warning, "Dwarf Version",
 423                               CodeGenOpts.DwarfVersion);
 424   }
 425   if (CodeGenOpts.EmitCodeView) {
 426     // Indicate that we want CodeView in the metadata.
 427     getModule().addModuleFlag(llvm::Module::Warning, "CodeView", 1);
 428   }
 429   if (CodeGenOpts.OptimizationLevel > 0 && CodeGenOpts.StrictVTablePointers) {
 430     // We don't support LTO with 2 with different StrictVTablePointers
 431     // FIXME: we could support it by stripping all the information introduced
 432     // by StrictVTablePointers.
 433
 434     getModule().addModuleFlag(llvm::Module::Error, "StrictVTablePointers",1);
 435
 436     llvm::Metadata *Ops[2] = {
 437               llvm::MDString::get(VMContext, "StrictVTablePointers"),
 438               llvm::ConstantAsMetadata::get(llvm::ConstantInt::get(
 439                   llvm::Type::getInt32Ty(VMContext), 1))};
 440
 441     getModule().addModuleFlag(llvm::Module::Require,
 442                               "StrictVTablePointersRequirement",
 443                               llvm::MDNode::get(VMContext, Ops));
 444   }
 445   if (DebugInfo)
 446     // We support a single version in the linked module. The LLVM
 447     // parser will drop debug info with a different version number
 448     // (and warn about it, too).
 449     getModule().addModuleFlag(llvm::Module::Warning, "Debug Info Version",
 450                               llvm::DEBUG_METADATA_VERSION);
 451
 452   // We need to record the widths of enums and wchar_t, so that we can generate
 453   // the correct build attributes in the ARM backend.
 454   llvm::Triple::ArchType Arch = Context.getTargetInfo().getTriple().getArch();
 455   if (   Arch == llvm::Triple::arm
 456       || Arch == llvm::Triple::armeb
 457       || Arch == llvm::Triple::thumb
 458       || Arch == llvm::Triple::thumbeb) {
 459     // Width of wchar_t in bytes
 460     uint64_t WCharWidth =
 461         Context.getTypeSizeInChars(Context.getWideCharType()).getQuantity();
 462     getModule().addModuleFlag(llvm::Module::Error, "wchar_size", WCharWidth);
 463
 464     // The minimum width of an enum in bytes
 465     uint64_t EnumWidth = Context.getLangOpts().ShortEnums ? 1 : 4;
 466     getModule().addModuleFlag(llvm::Module::Error, "min_enum_size", EnumWidth);
 467   }
 468
 469   if (CodeGenOpts.SanitizeCfiCrossDso) {
 470     // Indicate that we want cross-DSO control flow integrity checks.
 471     getModule().addModuleFlag(llvm::Module::Override, "Cross-DSO CFI", 1);
 472   }
 473
 474   if (LangOpts.CUDAIsDevice && getTriple().isNVPTX()) {
 475     // Indicate whether __nvvm_reflect should be configured to flush denormal
 476     // floating point values to 0.  (This corresponds to its "__CUDA_FTZ"
 477     // property.)
 478     getModule().addModuleFlag(llvm::Module::Override, "nvvm-reflect-ftz",
 479                               LangOpts.CUDADeviceFlushDenormalsToZero ? 1 : 0);
 480   }
 481
 482   if (uint32_t PLevel = Context.getLangOpts().PICLevel) {
 483     assert(PLevel < 3 && "Invalid PIC Level");
 484     getModule().setPICLevel(static_cast<llvm::PICLevel::Level>(PLevel));
 485     if (Context.getLangOpts().PIE)
 486       getModule().setPIELevel(static_cast<llvm::PIELevel::Level>(PLevel));
 487   }
 488
 489   SimplifyPersonality();
 490
 491   if (getCodeGenOpts().EmitDeclMetadata)
 492     EmitDeclMetadata();
 493
 494   if (getCodeGenOpts().EmitGcovArcs || getCodeGenOpts().EmitGcovNotes)
 495     EmitCoverageFile();
 496
 497   if (DebugInfo)
 498     DebugInfo->finalize();
 499
 500   EmitVersionIdentMetadata();
 501
 502   EmitTargetMetadata();
 503 }
 504
 505 void CodeGenModule::UpdateCompletedType(const TagDecl *TD) {
 506   // Make sure that this type is translated.
 507   Types.UpdateCompletedType(TD);
 508 }
 509
 510 void CodeGenModule::RefreshTypeCacheForClass(const CXXRecordDecl *RD) {
 511   // Make sure that this type is translated.
 512   Types.RefreshTypeCacheForClass(RD);
 513 }
 514
 515 llvm::MDNode *CodeGenModule::getTBAAInfo(QualType QTy) {
 516   if (!TBAA)
 517     return nullptr;
 518   return TBAA->getTBAAInfo(QTy);
 519 }
 520
 521 llvm::MDNode *CodeGenModule::getTBAAInfoForVTablePtr() {
 522   if (!TBAA)
 523     return nullptr;
 524   return TBAA->getTBAAInfoForVTablePtr();
 525 }
 526
 527 llvm::MDNode *CodeGenModule::getTBAAStructInfo(QualType QTy) {
 528   if (!TBAA)
 529     return nullptr;
 530   return TBAA->getTBAAStructInfo(QTy);
 531 }
 532
 533 llvm::MDNode *CodeGenModule::getTBAAStructTagInfo(QualType BaseTy,
 534                                                   llvm::MDNode *AccessN,
 535                                                   uint64_t O) {
 536   if (!TBAA)
 537     return nullptr;
 538   return TBAA->getTBAAStructTagInfo(BaseTy, AccessN, O);
 539 }
 540
 541 /// Decorate the instruction with a TBAA tag. For both scalar TBAA
 542 /// and struct-path aware TBAA, the tag has the same format:
 543 /// base type, access type and offset.
 544 /// When ConvertTypeToTag is true, we create a tag based on the scalar type.
 545 void CodeGenModule::DecorateInstructionWithTBAA(llvm::Instruction *Inst,
 546                                                 llvm::MDNode *TBAAInfo,
 547                                                 bool ConvertTypeToTag) {
 548   if (ConvertTypeToTag && TBAA)
 549     Inst->setMetadata(llvm::LLVMContext::MD_tbaa,
 550                       TBAA->getTBAAScalarTagInfo(TBAAInfo));
 551   else
 552     Inst->setMetadata(llvm::LLVMContext::MD_tbaa, TBAAInfo);
 553 }
 554
 555 void CodeGenModule::DecorateInstructionWithInvariantGroup(
 556     llvm::Instruction *I, const CXXRecordDecl *RD) {
 557   llvm::Metadata *MD = CreateMetadataIdentifierForType(QualType(RD->getTypeForDecl(), 0));
 558   auto *MetaDataNode = dyn_cast<llvm::MDNode>(MD);
 559   // Check if we have to wrap MDString in MDNode.
 560   if (!MetaDataNode)
 561     MetaDataNode = llvm::MDNode::get(getLLVMContext(), MD);
 562   I->setMetadata(llvm::LLVMContext::MD_invariant_group, MetaDataNode);
 563 }
 564
 565 void CodeGenModule::Error(SourceLocation loc, StringRef message) {
 566   unsigned diagID = getDiags().getCustomDiagID(DiagnosticsEngine::Error, "%0");
 567   getDiags().Report(Context.getFullLoc(loc), diagID) << message;
 568 }
 569
 570 /// ErrorUnsupported - Print out an error that codegen doesn't support the
 571 /// specified stmt yet.
 572 void CodeGenModule::ErrorUnsupported(const Stmt *S, const char *Type) {
 573   unsigned DiagID = getDiags().getCustomDiagID(DiagnosticsEngine::Error,
 574                                                "cannot compile this %0 yet");
 575   std::string Msg = Type;
 576   getDiags().Report(Context.getFullLoc(S->getLocStart()), DiagID)
 577     << Msg << S->getSourceRange();
 578 }
 579
 580 /// ErrorUnsupported - Print out an error that codegen doesn't support the
 581 /// specified decl yet.
 582 void CodeGenModule::ErrorUnsupported(const Decl *D, const char *Type) {
 583   unsigned DiagID = getDiags().getCustomDiagID(DiagnosticsEngine::Error,
 584                                                "cannot compile this %0 yet");
 585   std::string Msg = Type;
 586   getDiags().Report(Context.getFullLoc(D->getLocation()), DiagID) << Msg;
 587 }
 588
 589 llvm::ConstantInt *CodeGenModule::getSize(CharUnits size) {
 590   return llvm::ConstantInt::get(SizeTy, size.getQuantity());
 591 }
 592
 593 void CodeGenModule::setGlobalVisibility(llvm::GlobalValue *GV,
 594                                         const NamedDecl *D) const {
 595   // Internal definitions always have default visibility.
 596   if (GV->hasLocalLinkage()) {
 597     GV->setVisibility(llvm::GlobalValue::DefaultVisibility);
 598     return;
 599   }
 600
 601   // Set visibility for definitions.
 602   LinkageInfo LV = D->getLinkageAndVisibility();
 603   if (LV.isVisibilityExplicit() || !GV->hasAvailableExternallyLinkage())
 604     GV->setVisibility(GetLLVMVisibility(LV.getVisibility()));
 605 }
 606
 607 static llvm::GlobalVariable::ThreadLocalMode GetLLVMTLSModel(StringRef S) {
 608   return llvm::StringSwitch<llvm::GlobalVariable::ThreadLocalMode>(S)
 609       .Case("global-dynamic", llvm::GlobalVariable::GeneralDynamicTLSModel)
 610       .Case("local-dynamic", llvm::GlobalVariable::LocalDynamicTLSModel)
 611       .Case("initial-exec", llvm::GlobalVariable::InitialExecTLSModel)
 612       .Case("local-exec", llvm::GlobalVariable::LocalExecTLSModel);
 613 }
 614
 615 static llvm::GlobalVariable::ThreadLocalMode GetLLVMTLSModel(
 616     CodeGenOptions::TLSModel M) {
 617   switch (M) {
 618   case CodeGenOptions::GeneralDynamicTLSModel:
 619     return llvm::GlobalVariable::GeneralDynamicTLSModel;
 620   case CodeGenOptions::LocalDynamicTLSModel:
 621     return llvm::GlobalVariable::LocalDynamicTLSModel;
 622   case CodeGenOptions::InitialExecTLSModel:
 623     return llvm::GlobalVariable::InitialExecTLSModel;
 624   case CodeGenOptions::LocalExecTLSModel:
 625     return llvm::GlobalVariable::LocalExecTLSModel;
 626   }
 627   llvm_unreachable("Invalid TLS model!");
 628 }
 629
 630 void CodeGenModule::setTLSMode(llvm::GlobalValue *GV, const VarDecl &D) const {
 631   assert(D.getTLSKind() && "setting TLS mode on non-TLS var!");
 632
 633   llvm::GlobalValue::ThreadLocalMode TLM;
 634   TLM = GetLLVMTLSModel(CodeGenOpts.getDefaultTLSModel());
 635
 636   // Override the TLS model if it is explicitly specified.
 637   if (const TLSModelAttr *Attr = D.getAttr<TLSModelAttr>()) {
 638     TLM = GetLLVMTLSModel(Attr->getModel());
 639   }
 640
 641   GV->setThreadLocalMode(TLM);
 642 }
 643
 644 StringRef CodeGenModule::getMangledName(GlobalDecl GD) {
 645   GlobalDecl CanonicalGD = GD.getCanonicalDecl();
 646
 647   // Some ABIs don't have constructor variants.  Make sure that base and
 648   // complete constructors get mangled the same.
 649   if (const auto *CD = dyn_cast<CXXConstructorDecl>(CanonicalGD.getDecl())) {
 650     if (!getTarget().getCXXABI().hasConstructorVariants()) {
 651       CXXCtorType OrigCtorType = GD.getCtorType();
 652       assert(OrigCtorType == Ctor_Base || OrigCtorType == Ctor_Complete);
 653       if (OrigCtorType == Ctor_Base)
 654         CanonicalGD = GlobalDecl(CD, Ctor_Complete);
 655     }
 656   }
 657
 658   StringRef &FoundStr = MangledDeclNames[CanonicalGD];
 659   if (!FoundStr.empty())
 660     return FoundStr;
 661
 662   const auto *ND = cast<NamedDecl>(GD.getDecl());
 663   SmallString<256> Buffer;
 664   StringRef Str;
 665   if (getCXXABI().getMangleContext().shouldMangleDeclName(ND)) {
 666     llvm::raw_svector_ostream Out(Buffer);
 667     if (const auto *D = dyn_cast<CXXConstructorDecl>(ND))
 668       getCXXABI().getMangleContext().mangleCXXCtor(D, GD.getCtorType(), Out);
 669     else if (const auto *D = dyn_cast<CXXDestructorDecl>(ND))
 670       getCXXABI().getMangleContext().mangleCXXDtor(D, GD.getDtorType(), Out);
 671     else
 672       getCXXABI().getMangleContext().mangleName(ND, Out);
 673     Str = Out.str();
 674   } else {
 675     IdentifierInfo *II = ND->getIdentifier();
 676     assert(II && "Attempt to mangle unnamed decl.");
 677     const auto *FD = dyn_cast<FunctionDecl>(ND);
 678
 679     if (FD &&
 680         FD->getType()->castAs<FunctionType>()->getCallConv() == CC_X86RegCall) {
 681       llvm::raw_svector_ostream Out(Buffer);
 682       Out << "__regcall3__" << II->getName();
 683       Str = Out.str();
 684     } else {
 685       Str = II->getName();
 686     }
 687   }
 688
 689   // Keep the first result in the case of a mangling collision.
 690   auto Result = Manglings.insert(std::make_pair(Str, GD));
 691   return FoundStr = Result.first->first();
 692 }
 693
 694 StringRef CodeGenModule::getBlockMangledName(GlobalDecl GD,
 695                                              const BlockDecl *BD) {
 696   MangleContext &MangleCtx = getCXXABI().getMangleContext();
 697   const Decl *D = GD.getDecl();
 698
 699   SmallString<256> Buffer;
 700   llvm::raw_svector_ostream Out(Buffer);
 701   if (!D)
 702     MangleCtx.mangleGlobalBlock(BD,
 703       dyn_cast_or_null<VarDecl>(initializedGlobalDecl.getDecl()), Out);
 704   else if (const auto *CD = dyn_cast<CXXConstructorDecl>(D))
 705     MangleCtx.mangleCtorBlock(CD, GD.getCtorType(), BD, Out);
 706   else if (const auto *DD = dyn_cast<CXXDestructorDecl>(D))
 707     MangleCtx.mangleDtorBlock(DD, GD.getDtorType(), BD, Out);
 708   else
 709     MangleCtx.mangleBlock(cast<DeclContext>(D), BD, Out);
 710
 711   auto Result = Manglings.insert(std::make_pair(Out.str(), BD));
 712   return Result.first->first();
 713 }
 714
 715 llvm::GlobalValue *CodeGenModule::GetGlobalValue(StringRef Name) {
 716   return getModule().getNamedValue(Name);
 717 }
 718
 719 /// AddGlobalCtor - Add a function to the list that will be called before
 720 /// main() runs.
 721 void CodeGenModule::AddGlobalCtor(llvm::Function *Ctor, int Priority,
 722                                   llvm::Constant *AssociatedData) {
 723   // FIXME: Type coercion of void()* types.
 724   GlobalCtors.push_back(Structor(Priority, Ctor, AssociatedData));
 725 }
 726
 727 /// AddGlobalDtor - Add a function to the list that will be called
 728 /// when the module is unloaded.
 729 void CodeGenModule::AddGlobalDtor(llvm::Function *Dtor, int Priority) {
 730   // FIXME: Type coercion of void()* types.
 731   GlobalDtors.push_back(Structor(Priority, Dtor, nullptr));
 732 }
 733
 734 void CodeGenModule::EmitCtorList(CtorList &Fns, const char *GlobalName) {
 735   if (Fns.empty()) return;
 736
 737   // Ctor function type is void()*.
 738   llvm::FunctionType* CtorFTy = llvm::FunctionType::get(VoidTy, false);
 739   llvm::Type *CtorPFTy = llvm::PointerType::getUnqual(CtorFTy);
 740
 741   // Get the type of a ctor entry, { i32, void ()*, i8* }.
 742   llvm::StructType *CtorStructTy = llvm::StructType::get(
 743       Int32Ty, llvm::PointerType::getUnqual(CtorFTy), VoidPtrTy, nullptr);
 744
 745   // Construct the constructor and destructor arrays.
 746   ConstantInitBuilder builder(*this);
 747   auto ctors = builder.beginArray(CtorStructTy);
 748   for (const auto &I : Fns) {
 749     auto ctor = ctors.beginStruct(CtorStructTy);
 750     ctor.addInt(Int32Ty, I.Priority);
 751     ctor.add(llvm::ConstantExpr::getBitCast(I.Initializer, CtorPFTy));
 752     if (I.AssociatedData)
 753       ctor.add(llvm::ConstantExpr::getBitCast(I.AssociatedData, VoidPtrTy));
 754     else
 755       ctor.addNullPointer(VoidPtrTy);
 756     ctor.finishAndAddTo(ctors);
 757   }
 758
 759   auto list =
 760     ctors.finishAndCreateGlobal(GlobalName, getPointerAlign(),
 761                                 /*constant*/ false,
 762                                 llvm::GlobalValue::AppendingLinkage);
 763
 764   // The LTO linker doesn't seem to like it when we set an alignment
 765   // on appending variables.  Take it off as a workaround.
 766   list->setAlignment(0);
 767
 768   Fns.clear();
 769 }
 770
 771 llvm::GlobalValue::LinkageTypes
 772 CodeGenModule::getFunctionLinkage(GlobalDecl GD) {
 773   const auto *D = cast<FunctionDecl>(GD.getDecl());
 774
 775   GVALinkage Linkage = getContext().GetGVALinkageForFunction(D);
 776
 777   if (isa<CXXDestructorDecl>(D) &&
 778       getCXXABI().useThunkForDtorVariant(cast<CXXDestructorDecl>(D),
 779                                          GD.getDtorType())) {
 780     // Destructor variants in the Microsoft C++ ABI are always internal or
 781     // linkonce_odr thunks emitted on an as-needed basis.
 782     return Linkage == GVA_Internal ? llvm::GlobalValue::InternalLinkage
 783                                    : llvm::GlobalValue::LinkOnceODRLinkage;
 784   }
 785
 786   if (isa<CXXConstructorDecl>(D) &&
 787       cast<CXXConstructorDecl>(D)->isInheritingConstructor() &&
 788       Context.getTargetInfo().getCXXABI().isMicrosoft()) {
 789     // Our approach to inheriting constructors is fundamentally different from
 790     // that used by the MS ABI, so keep our inheriting constructor thunks
 791     // internal rather than trying to pick an unambiguous mangling for them.
 792     return llvm::GlobalValue::InternalLinkage;
 793   }
 794
 795   return getLLVMLinkageForDeclarator(D, Linkage, /*isConstantVariable=*/false);
 796 }
 797
 798 void CodeGenModule::setFunctionDLLStorageClass(GlobalDecl GD, llvm::Function *F) {
 799   const auto *FD = cast<FunctionDecl>(GD.getDecl());
 800
 801   if (const auto *Dtor = dyn_cast_or_null<CXXDestructorDecl>(FD)) {
 802     if (getCXXABI().useThunkForDtorVariant(Dtor, GD.getDtorType())) {
 803       // Don't dllexport/import destructor thunks.
 804       F->setDLLStorageClass(llvm::GlobalValue::DefaultStorageClass);
 805       return;
 806     }
 807   }
 808
 809   if (FD->hasAttr<DLLImportAttr>())
 810     F->setDLLStorageClass(llvm::GlobalVariable::DLLImportStorageClass);
 811   else if (FD->hasAttr<DLLExportAttr>())
 812     F->setDLLStorageClass(llvm::GlobalVariable::DLLExportStorageClass);
 813   else
 814     F->setDLLStorageClass(llvm::GlobalVariable::DefaultStorageClass);
 815 }
 816
 817 llvm::ConstantInt *CodeGenModule::CreateCrossDsoCfiTypeId(llvm::Metadata *MD) {
 818   llvm::MDString *MDS = dyn_cast<llvm::MDString>(MD);
 819   if (!MDS) return nullptr;
 820
 821   return llvm::ConstantInt::get(Int64Ty, llvm::MD5Hash(MDS->getString()));
 822 }
 823
 824 void CodeGenModule::setFunctionDefinitionAttributes(const FunctionDecl *D,
 825                                                     llvm::Function *F) {
 826   setNonAliasAttributes(D, F);
 827 }
 828
 829 void CodeGenModule::SetLLVMFunctionAttributes(const Decl *D,
 830                                               const CGFunctionInfo &Info,
 831                                               llvm::Function *F) {
 832   unsigned CallingConv;
 833   AttributeListType AttributeList;
 834   ConstructAttributeList(F->getName(), Info, D, AttributeList, CallingConv,
 835                          false);
 836   F->setAttributes(llvm::AttributeSet::get(getLLVMContext(), AttributeList));
 837   F->setCallingConv(static_cast<llvm::CallingConv::ID>(CallingConv));
 838 }
 839
 840 /// Determines whether the language options require us to model
 841 /// unwind exceptions.  We treat -fexceptions as mandating this
 842 /// except under the fragile ObjC ABI with only ObjC exceptions
 843 /// enabled.  This means, for example, that C with -fexceptions
 844 /// enables this.
 845 static bool hasUnwindExceptions(const LangOptions &LangOpts) {
 846   // If exceptions are completely disabled, obviously this is false.
 847   if (!LangOpts.Exceptions) return false;
 848
 849   // If C++ exceptions are enabled, this is true.
 850   if (LangOpts.CXXExceptions) return true;
 851
 852   // If ObjC exceptions are enabled, this depends on the ABI.
 853   if (LangOpts.ObjCExceptions) {
 854     return LangOpts.ObjCRuntime.hasUnwindExceptions();
 855   }
 856
 857   return true;
 858 }
 859
 860 void CodeGenModule::SetLLVMFunctionAttributesForDefinition(const Decl *D,
 861                                                            llvm::Function *F) {
 862   llvm::AttrBuilder B;
 863
 864   if (CodeGenOpts.UnwindTables)
 865     B.addAttribute(llvm::Attribute::UWTable);
 866
 867   if (!hasUnwindExceptions(LangOpts))
 868     B.addAttribute(llvm::Attribute::NoUnwind);
 869
 870   if (LangOpts.getStackProtector() == LangOptions::SSPOn)
 871     B.addAttribute(llvm::Attribute::StackProtect);
 872   else if (LangOpts.getStackProtector() == LangOptions::SSPStrong)
 873     B.addAttribute(llvm::Attribute::StackProtectStrong);
 874   else if (LangOpts.getStackProtector() == LangOptions::SSPReq)
 875     B.addAttribute(llvm::Attribute::StackProtectReq);
 876
 877   if (!D) {
 878     // If we don't have a declaration to control inlining, the function isn't
 879     // explicitly marked as alwaysinline for semantic reasons, and inlining is
 880     // disabled, mark the function as noinline.
 881     if (!F->hasFnAttribute(llvm::Attribute::AlwaysInline) &&
 882         CodeGenOpts.getInlining() == CodeGenOptions::OnlyAlwaysInlining)
 883       B.addAttribute(llvm::Attribute::NoInline);
 884
 885     F->addAttributes(llvm::AttributeSet::FunctionIndex,
 886                      llvm::AttributeSet::get(
 887                          F->getContext(),
 888                          llvm::AttributeSet::FunctionIndex, B));
 889     return;
 890   }
 891
 892   if (D->hasAttr<OptimizeNoneAttr>()) {
 893     B.addAttribute(llvm::Attribute::OptimizeNone);
 894
 895     // OptimizeNone implies noinline; we should not be inlining such functions.
 896     B.addAttribute(llvm::Attribute::NoInline);
 897     assert(!F->hasFnAttribute(llvm::Attribute::AlwaysInline) &&
 898            "OptimizeNone and AlwaysInline on same function!");
 899
 900     // We still need to handle naked functions even though optnone subsumes
 901     // much of their semantics.
 902     if (D->hasAttr<NakedAttr>())
 903       B.addAttribute(llvm::Attribute::Naked);
 904
 905     // OptimizeNone wins over OptimizeForSize and MinSize.
 906     F->removeFnAttr(llvm::Attribute::OptimizeForSize);
 907     F->removeFnAttr(llvm::Attribute::MinSize);
 908   } else if (D->hasAttr<NakedAttr>()) {
 909     // Naked implies noinline: we should not be inlining such functions.
 910     B.addAttribute(llvm::Attribute::Naked);
 911     B.addAttribute(llvm::Attribute::NoInline);
 912   } else if (D->hasAttr<NoDuplicateAttr>()) {
 913     B.addAttribute(llvm::Attribute::NoDuplicate);
 914   } else if (D->hasAttr<NoInlineAttr>()) {
 915     B.addAttribute(llvm::Attribute::NoInline);
 916   } else if (D->hasAttr<AlwaysInlineAttr>() &&
 917              !F->hasFnAttribute(llvm::Attribute::NoInline)) {
 918     // (noinline wins over always_inline, and we can't specify both in IR)
 919     B.addAttribute(llvm::Attribute::AlwaysInline);
 920   } else if (CodeGenOpts.getInlining() == CodeGenOptions::OnlyAlwaysInlining) {
 921     // If we're not inlining, then force everything that isn't always_inline to
 922     // carry an explicit noinline attribute.
 923     if (!F->hasFnAttribute(llvm::Attribute::AlwaysInline))
 924       B.addAttribute(llvm::Attribute::NoInline);
 925   } else {
 926     // Otherwise, propagate the inline hint attribute and potentially use its
 927     // absence to mark things as noinline.
 928     if (auto *FD = dyn_cast<FunctionDecl>(D)) {
 929       if (any_of(FD->redecls(), [&](const FunctionDecl *Redecl) {
 930             return Redecl->isInlineSpecified();
 931           })) {
 932         B.addAttribute(llvm::Attribute::InlineHint);
 933       } else if (CodeGenOpts.getInlining() ==
 934                      CodeGenOptions::OnlyHintInlining &&
 935                  !FD->isInlined() &&
 936                  !F->hasFnAttribute(llvm::Attribute::AlwaysInline)) {
 937         B.addAttribute(llvm::Attribute::NoInline);
 938       }
 939     }
 940   }
 941
 942   // Add other optimization related attributes if we are optimizing this
 943   // function.
 944   if (!D->hasAttr<OptimizeNoneAttr>()) {
 945     if (D->hasAttr<ColdAttr>()) {
 946       B.addAttribute(llvm::Attribute::OptimizeForSize);
 947       B.addAttribute(llvm::Attribute::Cold);
 948     }
 949
 950     if (D->hasAttr<MinSizeAttr>())
 951       B.addAttribute(llvm::Attribute::MinSize);
 952   }
 953
 954   F->addAttributes(llvm::AttributeSet::FunctionIndex,
 955                    llvm::AttributeSet::get(
 956                        F->getContext(), llvm::AttributeSet::FunctionIndex, B));
 957
 958   unsigned alignment = D->getMaxAlignment() / Context.getCharWidth();
 959   if (alignment)
 960     F->setAlignment(alignment);
 961
 962   // Some C++ ABIs require 2-byte alignment for member functions, in order to
 963   // reserve a bit for differentiating between virtual and non-virtual member
 964   // functions. If the current target's C++ ABI requires this and this is a
 965   // member function, set its alignment accordingly.
 966   if (getTarget().getCXXABI().areMemberFunctionsAligned()) {
 967     if (F->getAlignment() < 2 && isa<CXXMethodDecl>(D))
 968       F->setAlignment(2);
 969   }
 970
 971   // In the cross-dso CFI mode, we want !type attributes on definitions only.
 972   if (CodeGenOpts.SanitizeCfiCrossDso)
 973     if (auto *FD = dyn_cast<FunctionDecl>(D))
 974       CreateFunctionTypeMetadata(FD, F);
 975 }
 976
 977 void CodeGenModule::SetCommonAttributes(const Decl *D,
 978                                         llvm::GlobalValue *GV) {
 979   if (const auto *ND = dyn_cast_or_null<NamedDecl>(D))
 980     setGlobalVisibility(GV, ND);
 981   else
 982     GV->setVisibility(llvm::GlobalValue::DefaultVisibility);
 983
 984   if (D && D->hasAttr<UsedAttr>())
 985     addUsedGlobal(GV);
 986 }
 987
 988 void CodeGenModule::setAliasAttributes(const Decl *D,
 989                                        llvm::GlobalValue *GV) {
 990   SetCommonAttributes(D, GV);
 991
 992   // Process the dllexport attribute based on whether the original definition
 993   // (not necessarily the aliasee) was exported.
 994   if (D->hasAttr<DLLExportAttr>())
 995     GV->setDLLStorageClass(llvm::GlobalValue::DLLExportStorageClass);
 996 }
 997
 998 void CodeGenModule::setNonAliasAttributes(const Decl *D,
 999                                           llvm::GlobalObject *GO) {
1000   SetCommonAttributes(D, GO);
1001
1002   if (D)
1003     if (const SectionAttr *SA = D->getAttr<SectionAttr>())
1004       GO->setSection(SA->getName());
1005
1006   getTargetCodeGenInfo().setTargetAttributes(D, GO, *this);
1007 }
1008
1009 void CodeGenModule::SetInternalFunctionAttributes(const Decl *D,
1010                                                   llvm::Function *F,
1011                                                   const CGFunctionInfo &FI) {
1012   SetLLVMFunctionAttributes(D, FI, F);
1013   SetLLVMFunctionAttributesForDefinition(D, F);
1014
1015   F->setLinkage(llvm::Function::InternalLinkage);
1016
1017   setNonAliasAttributes(D, F);
1018 }
1019
1020 static void setLinkageAndVisibilityForGV(llvm::GlobalValue *GV,
1021                                          const NamedDecl *ND) {
1022   // Set linkage and visibility in case we never see a definition.
1023   LinkageInfo LV = ND->getLinkageAndVisibility();
1024   if (LV.getLinkage() != ExternalLinkage) {
1025     // Don't set internal linkage on declarations.
1026   } else {
1027     if (ND->hasAttr<DLLImportAttr>()) {
1028       GV->setLinkage(llvm::GlobalValue::ExternalLinkage);
1029       GV->setDLLStorageClass(llvm::GlobalValue::DLLImportStorageClass);
1030     } else if (ND->hasAttr<DLLExportAttr>()) {
1031       GV->setLinkage(llvm::GlobalValue::ExternalLinkage);
1032       GV->setDLLStorageClass(llvm::GlobalValue::DLLExportStorageClass);
1033     } else if (ND->hasAttr<WeakAttr>() || ND->isWeakImported()) {
1034       // "extern_weak" is overloaded in LLVM; we probably should have
1035       // separate linkage types for this.
1036       GV->setLinkage(llvm::GlobalValue::ExternalWeakLinkage);
1037     }
1038
1039     // Set visibility on a declaration only if it's explicit.
1040     if (LV.isVisibilityExplicit())
1041       GV->setVisibility(CodeGenModule::GetLLVMVisibility(LV.getVisibility()));
1042   }
1043 }
1044
1045 void CodeGenModule::CreateFunctionTypeMetadata(const FunctionDecl *FD,
1046                                                llvm::Function *F) {
1047   // Only if we are checking indirect calls.
1048   if (!LangOpts.Sanitize.has(SanitizerKind::CFIICall))
1049     return;
1050
1051   // Non-static class methods are handled via vtable pointer checks elsewhere.
1052   if (isa<CXXMethodDecl>(FD) && !cast<CXXMethodDecl>(FD)->isStatic())
1053     return;
1054
1055   // Additionally, if building with cross-DSO support...
1056   if (CodeGenOpts.SanitizeCfiCrossDso) {
1057     // Skip available_externally functions. They won't be codegen'ed in the
1058     // current module anyway.
1059     if (getContext().GetGVALinkageForFunction(FD) == GVA_AvailableExternally)
1060       return;
1061   }
1062
1063   llvm::Metadata *MD = CreateMetadataIdentifierForType(FD->getType());
1064   F->addTypeMetadata(0, MD);
1065
1066   // Emit a hash-based bit set entry for cross-DSO calls.
1067   if (CodeGenOpts.SanitizeCfiCrossDso)
1068     if (auto CrossDsoTypeId = CreateCrossDsoCfiTypeId(MD))
1069       F->addTypeMetadata(0, llvm::ConstantAsMetadata::get(CrossDsoTypeId));
1070 }
1071
1072 void CodeGenModule::SetFunctionAttributes(GlobalDecl GD, llvm::Function *F,
1073                                           bool IsIncompleteFunction,
1074                                           bool IsThunk) {
1075   if (llvm::Intrinsic::ID IID = F->getIntrinsicID()) {
1076     // If this is an intrinsic function, set the function's attributes
1077     // to the intrinsic's attributes.
1078     F->setAttributes(llvm::Intrinsic::getAttributes(getLLVMContext(), IID));
1079     return;
1080   }
1081
1082   const auto *FD = cast<FunctionDecl>(GD.getDecl());
1083
1084   if (!IsIncompleteFunction)
1085     SetLLVMFunctionAttributes(FD, getTypes().arrangeGlobalDeclaration(GD), F);
1086
1087   // Add the Returned attribute for "this", except for iOS 5 and earlier
1088   // where substantial code, including the libstdc++ dylib, was compiled with
1089   // GCC and does not actually return "this".
1090   if (!IsThunk && getCXXABI().HasThisReturn(GD) &&
1091       !(getTriple().isiOS() && getTriple().isOSVersionLT(6))) {
1092     assert(!F->arg_empty() &&
1093            F->arg_begin()->getType()
1094              ->canLosslesslyBitCastTo(F->getReturnType()) &&
1095            "unexpected this return");
1096     F->addAttribute(1, llvm::Attribute::Returned);
1097   }
1098
1099   // Only a few attributes are set on declarations; these may later be
1100   // overridden by a definition.
1101
1102   setLinkageAndVisibilityForGV(F, FD);
1103
1104   if (const SectionAttr *SA = FD->getAttr<SectionAttr>())
1105     F->setSection(SA->getName());
1106
1107   if (FD->isReplaceableGlobalAllocationFunction()) {
1108     // A replaceable global allocation function does not act like a builtin by
1109     // default, only if it is invoked by a new-expression or delete-expression.
1110     F->addAttribute(llvm::AttributeSet::FunctionIndex,
1111                     llvm::Attribute::NoBuiltin);
1112
1113     // A sane operator new returns a non-aliasing pointer.
1114     // FIXME: Also add NonNull attribute to the return value
1115     // for the non-nothrow forms?
1116     auto Kind = FD->getDeclName().getCXXOverloadedOperator();
1117     if (getCodeGenOpts().AssumeSaneOperatorNew &&
1118         (Kind == OO_New || Kind == OO_Array_New))
1119       F->addAttribute(llvm::AttributeSet::ReturnIndex,
1120                       llvm::Attribute::NoAlias);
1121   }
1122
1123   if (isa<CXXConstructorDecl>(FD) || isa<CXXDestructorDecl>(FD))
1124     F->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
1125   else if (const auto *MD = dyn_cast<CXXMethodDecl>(FD))
1126     if (MD->isVirtual())
1127       F->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
1128
1129   // Don't emit entries for function declarations in the cross-DSO mode. This
1130   // is handled with better precision by the receiving DSO.
1131   if (!CodeGenOpts.SanitizeCfiCrossDso)
1132     CreateFunctionTypeMetadata(FD, F);
1133 }
1134
1135 void CodeGenModule::addUsedGlobal(llvm::GlobalValue *GV) {
1136   assert(!GV->isDeclaration() &&
1137          "Only globals with definition can force usage.");
1138   LLVMUsed.emplace_back(GV);
1139 }
1140
1141 void CodeGenModule::addCompilerUsedGlobal(llvm::GlobalValue *GV) {
1142   assert(!GV->isDeclaration() &&
1143          "Only globals with definition can force usage.");
1144   LLVMCompilerUsed.emplace_back(GV);
1145 }
1146
1147 static void emitUsed(CodeGenModule &CGM, StringRef Name,
1148                      std::vector<llvm::WeakVH> &List) {
1149   // Don't create llvm.used if there is no need.
1150   if (List.empty())
1151     return;
1152
1153   // Convert List to what ConstantArray needs.
1154   SmallVector<llvm::Constant*, 8> UsedArray;
1155   UsedArray.resize(List.size());
1156   for (unsigned i = 0, e = List.size(); i != e; ++i) {
1157     UsedArray[i] =
1158         llvm::ConstantExpr::getPointerBitCastOrAddrSpaceCast(
1159             cast<llvm::Constant>(&*List[i]), CGM.Int8PtrTy);
1160   }
1161
1162   if (UsedArray.empty())
1163     return;
1164   llvm::ArrayType *ATy = llvm::ArrayType::get(CGM.Int8PtrTy, UsedArray.size());
1165
1166   auto *GV = new llvm::GlobalVariable(
1167       CGM.getModule(), ATy, false, llvm::GlobalValue::AppendingLinkage,
1168       llvm::ConstantArray::get(ATy, UsedArray), Name);
1169
1170   GV->setSection("llvm.metadata");
1171 }
1172
1173 void CodeGenModule::emitLLVMUsed() {
1174   emitUsed(*this, "llvm.used", LLVMUsed);
1175   emitUsed(*this, "llvm.compiler.used", LLVMCompilerUsed);
1176 }
1177
1178 void CodeGenModule::AppendLinkerOptions(StringRef Opts) {
1179   auto *MDOpts = llvm::MDString::get(getLLVMContext(), Opts);
1180   LinkerOptionsMetadata.push_back(llvm::MDNode::get(getLLVMContext(), MDOpts));
1181 }
1182
1183 void CodeGenModule::AddDetectMismatch(StringRef Name, StringRef Value) {
1184   llvm::SmallString<32> Opt;
1185   getTargetCodeGenInfo().getDetectMismatchOption(Name, Value, Opt);
1186   auto *MDOpts = llvm::MDString::get(getLLVMContext(), Opt);
1187   LinkerOptionsMetadata.push_back(llvm::MDNode::get(getLLVMContext(), MDOpts));
1188 }
1189
1190 void CodeGenModule::AddDependentLib(StringRef Lib) {
1191   llvm::SmallString<24> Opt;
1192   getTargetCodeGenInfo().getDependentLibraryOption(Lib, Opt);
1193   auto *MDOpts = llvm::MDString::get(getLLVMContext(), Opt);
1194   LinkerOptionsMetadata.push_back(llvm::MDNode::get(getLLVMContext(), MDOpts));
1195 }
1196
1197 /// \brief Add link options implied by the given module, including modules
1198 /// it depends on, using a postorder walk.
1199 static void addLinkOptionsPostorder(CodeGenModule &CGM, Module *Mod,
1200                                     SmallVectorImpl<llvm::Metadata *> &Metadata,
1201                                     llvm::SmallPtrSet<Module *, 16> &Visited) {
1202   // Import this module's parent.
1203   if (Mod->Parent && Visited.insert(Mod->Parent).second) {
1204     addLinkOptionsPostorder(CGM, Mod->Parent, Metadata, Visited);
1205   }
1206
1207   // Import this module's dependencies.
1208   for (unsigned I = Mod->Imports.size(); I > 0; --I) {
1209     if (Visited.insert(Mod->Imports[I - 1]).second)
1210       addLinkOptionsPostorder(CGM, Mod->Imports[I-1], Metadata, Visited);
1211   }
1212
1213   // Add linker options to link against the libraries/frameworks
1214   // described by this module.
1215   llvm::LLVMContext &Context = CGM.getLLVMContext();
1216   for (unsigned I = Mod->LinkLibraries.size(); I > 0; --I) {
1217     // Link against a framework.  Frameworks are currently Darwin only, so we
1218     // don't to ask TargetCodeGenInfo for the spelling of the linker option.
1219     if (Mod->LinkLibraries[I-1].IsFramework) {
1220       llvm::Metadata *Args[2] = {
1221           llvm::MDString::get(Context, "-framework"),
1222           llvm::MDString::get(Context, Mod->LinkLibraries[I - 1].Library)};
1223
1224       Metadata.push_back(llvm::MDNode::get(Context, Args));
1225       continue;
1226     }
1227
1228     // Link against a library.
1229     llvm::SmallString<24> Opt;
1230     CGM.getTargetCodeGenInfo().getDependentLibraryOption(
1231       Mod->LinkLibraries[I-1].Library, Opt);
1232     auto *OptString = llvm::MDString::get(Context, Opt);
1233     Metadata.push_back(llvm::MDNode::get(Context, OptString));
1234   }
1235 }
1236
1237 void CodeGenModule::EmitModuleLinkOptions() {
1238   // Collect the set of all of the modules we want to visit to emit link
1239   // options, which is essentially the imported modules and all of their
1240   // non-explicit child modules.
1241   llvm::SetVector<clang::Module *> LinkModules;
1242   llvm::SmallPtrSet<clang::Module *, 16> Visited;
1243   SmallVector<clang::Module *, 16> Stack;
1244
1245   // Seed the stack with imported modules.
1246   for (Module *M : ImportedModules) {
1247     // Do not add any link flags when an implementation TU of a module imports
1248     // a header of that same module.
1249     if (M->getTopLevelModuleName() == getLangOpts().CurrentModule &&
1250         !getLangOpts().isCompilingModule())
1251       continue;
1252     if (Visited.insert(M).second)
1253       Stack.push_back(M);
1254   }
1255
1256   // Find all of the modules to import, making a little effort to prune
1257   // non-leaf modules.
1258   while (!Stack.empty()) {
1259     clang::Module *Mod = Stack.pop_back_val();
1260
1261     bool AnyChildren = false;
1262
1263     // Visit the submodules of this module.
1264     for (clang::Module::submodule_iterator Sub = Mod->submodule_begin(),
1265                                         SubEnd = Mod->submodule_end();
1266          Sub != SubEnd; ++Sub) {
1267       // Skip explicit children; they need to be explicitly imported to be
1268       // linked against.
1269       if ((*Sub)->IsExplicit)
1270         continue;
1271
1272       if (Visited.insert(*Sub).second) {
1273         Stack.push_back(*Sub);
1274         AnyChildren = true;
1275       }
1276     }
1277
1278     // We didn't find any children, so add this module to the list of
1279     // modules to link against.
1280     if (!AnyChildren) {
1281       LinkModules.insert(Mod);
1282     }
1283   }
1284
1285   // Add link options for all of the imported modules in reverse topological
1286   // order.  We don't do anything to try to order import link flags with respect
1287   // to linker options inserted by things like #pragma comment().
1288   SmallVector<llvm::Metadata *, 16> MetadataArgs;
1289   Visited.clear();
1290   for (Module *M : LinkModules)
1291     if (Visited.insert(M).second)
1292       addLinkOptionsPostorder(*this, M, MetadataArgs, Visited);
1293   std::reverse(MetadataArgs.begin(), MetadataArgs.end());
1294   LinkerOptionsMetadata.append(MetadataArgs.begin(), MetadataArgs.end());
1295
1296   // Add the linker options metadata flag.
1297   getModule().addModuleFlag(llvm::Module::AppendUnique, "Linker Options",
1298                             llvm::MDNode::get(getLLVMContext(),
1299                                               LinkerOptionsMetadata));
1300 }
1301
1302 void CodeGenModule::EmitDeferred() {
1303   // Emit code for any potentially referenced deferred decls.  Since a
1304   // previously unused static decl may become used during the generation of code
1305   // for a static function, iterate until no changes are made.
1306
1307   if (!DeferredVTables.empty()) {
1308     EmitDeferredVTables();
1309
1310     // Emitting a vtable doesn't directly cause more vtables to
1311     // become deferred, although it can cause functions to be
1312     // emitted that then need those vtables.
1313     assert(DeferredVTables.empty());
1314   }
1315
1316   // Stop if we're out of both deferred vtables and deferred declarations.
1317   if (DeferredDeclsToEmit.empty())
1318     return;
1319
1320   // Grab the list of decls to emit. If EmitGlobalDefinition schedules more
1321   // work, it will not interfere with this.
1322   std::vector<DeferredGlobal> CurDeclsToEmit;
1323   CurDeclsToEmit.swap(DeferredDeclsToEmit);
1324
1325   for (DeferredGlobal &G : CurDeclsToEmit) {
1326     GlobalDecl D = G.GD;
1327     G.GV = nullptr;
1328
1329     // We should call GetAddrOfGlobal with IsForDefinition set to true in order
1330     // to get GlobalValue with exactly the type we need, not something that
1331     // might had been created for another decl with the same mangled name but
1332     // different type.
1333     llvm::GlobalValue *GV = dyn_cast<llvm::GlobalValue>(
1334         GetAddrOfGlobal(D, ForDefinition));
1335
1336     // In case of different address spaces, we may still get a cast, even with
1337     // IsForDefinition equal to true. Query mangled names table to get
1338     // GlobalValue.
1339     if (!GV)
1340       GV = GetGlobalValue(getMangledName(D));
1341
1342     // Make sure GetGlobalValue returned non-null.
1343     assert(GV);
1344
1345     // Check to see if we've already emitted this.  This is necessary
1346     // for a couple of reasons: first, decls can end up in the
1347     // deferred-decls queue multiple times, and second, decls can end
1348     // up with definitions in unusual ways (e.g. by an extern inline
1349     // function acquiring a strong function redefinition).  Just
1350     // ignore these cases.
1351     if (!GV->isDeclaration())
1352       continue;
1353
1354     // Otherwise, emit the definition and move on to the next one.
1355     EmitGlobalDefinition(D, GV);
1356
1357     // If we found out that we need to emit more decls, do that recursively.
1358     // This has the advantage that the decls are emitted in a DFS and related
1359     // ones are close together, which is convenient for testing.
1360     if (!DeferredVTables.empty() || !DeferredDeclsToEmit.empty()) {
1361       EmitDeferred();
1362       assert(DeferredVTables.empty() && DeferredDeclsToEmit.empty());
1363     }
1364   }
1365 }
1366
1367 void CodeGenModule::EmitGlobalAnnotations() {
1368   if (Annotations.empty())
1369     return;
1370
1371   // Create a new global variable for the ConstantStruct in the Module.
1372   llvm::Constant *Array = llvm::ConstantArray::get(llvm::ArrayType::get(
1373     Annotations[0]->getType(), Annotations.size()), Annotations);
1374   auto *gv = new llvm::GlobalVariable(getModule(), Array->getType(), false,
1375                                       llvm::GlobalValue::AppendingLinkage,
1376                                       Array, "llvm.global.annotations");
1377   gv->setSection(AnnotationSection);
1378 }
1379
1380 llvm::Constant *CodeGenModule::EmitAnnotationString(StringRef Str) {
1381   llvm::Constant *&AStr = AnnotationStrings[Str];
1382   if (AStr)
1383     return AStr;
1384
1385   // Not found yet, create a new global.
1386   llvm::Constant *s = llvm::ConstantDataArray::getString(getLLVMContext(), Str);
1387   auto *gv =
1388       new llvm::GlobalVariable(getModule(), s->getType(), true,
1389                                llvm::GlobalValue::PrivateLinkage, s, ".str");
1390   gv->setSection(AnnotationSection);
1391   gv->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
1392   AStr = gv;
1393   return gv;
1394 }
1395
1396 llvm::Constant *CodeGenModule::EmitAnnotationUnit(SourceLocation Loc) {
1397   SourceManager &SM = getContext().getSourceManager();
1398   PresumedLoc PLoc = SM.getPresumedLoc(Loc);
1399   if (PLoc.isValid())
1400     return EmitAnnotationString(PLoc.getFilename());
1401   return EmitAnnotationString(SM.getBufferName(Loc));
1402 }
1403
1404 llvm::Constant *CodeGenModule::EmitAnnotationLineNo(SourceLocation L) {
1405   SourceManager &SM = getContext().getSourceManager();
1406   PresumedLoc PLoc = SM.getPresumedLoc(L);
1407   unsigned LineNo = PLoc.isValid() ? PLoc.getLine() :
1408     SM.getExpansionLineNumber(L);
1409   return llvm::ConstantInt::get(Int32Ty, LineNo);
1410 }
1411
1412 llvm::Constant *CodeGenModule::EmitAnnotateAttr(llvm::GlobalValue *GV,
1413                                                 const AnnotateAttr *AA,
1414                                                 SourceLocation L) {
1415   // Get the globals for file name, annotation, and the line number.
1416   llvm::Constant *AnnoGV = EmitAnnotationString(AA->getAnnotation()),
1417                  *UnitGV = EmitAnnotationUnit(L),
1418                  *LineNoCst = EmitAnnotationLineNo(L);
1419
1420   // Create the ConstantStruct for the global annotation.
1421   llvm::Constant *Fields[4] = {
1422     llvm::ConstantExpr::getBitCast(GV, Int8PtrTy),
1423     llvm::ConstantExpr::getBitCast(AnnoGV, Int8PtrTy),
1424     llvm::ConstantExpr::getBitCast(UnitGV, Int8PtrTy),
1425     LineNoCst
1426   };
1427   return llvm::ConstantStruct::getAnon(Fields);
1428 }
1429
1430 void CodeGenModule::AddGlobalAnnotations(const ValueDecl *D,
1431                                          llvm::GlobalValue *GV) {
1432   assert(D->hasAttr<AnnotateAttr>() && "no annotate attribute");
1433   // Get the struct elements for these annotations.
1434   for (const auto *I : D->specific_attrs<AnnotateAttr>())
1435     Annotations.push_back(EmitAnnotateAttr(GV, I, D->getLocation()));
1436 }
1437
1438 bool CodeGenModule::isInSanitizerBlacklist(llvm::Function *Fn,
1439                                            SourceLocation Loc) const {
1440   const auto &SanitizerBL = getContext().getSanitizerBlacklist();
1441   // Blacklist by function name.
1442   if (SanitizerBL.isBlacklistedFunction(Fn->getName()))
1443     return true;
1444   // Blacklist by location.
1445   if (Loc.isValid())
1446     return SanitizerBL.isBlacklistedLocation(Loc);
1447   // If location is unknown, this may be a compiler-generated function. Assume
1448   // it's located in the main file.
1449   auto &SM = Context.getSourceManager();
1450   if (const auto *MainFile = SM.getFileEntryForID(SM.getMainFileID())) {
1451     return SanitizerBL.isBlacklistedFile(MainFile->getName());
1452   }
1453   return false;
1454 }
1455
1456 bool CodeGenModule::isInSanitizerBlacklist(llvm::GlobalVariable *GV,
1457                                            SourceLocation Loc, QualType Ty,
1458                                            StringRef Category) const {
1459   // For now globals can be blacklisted only in ASan and KASan.
1460   if (!LangOpts.Sanitize.hasOneOf(
1461           SanitizerKind::Address | SanitizerKind::KernelAddress))
1462     return false;
1463   const auto &SanitizerBL = getContext().getSanitizerBlacklist();
1464   if (SanitizerBL.isBlacklistedGlobal(GV->getName(), Category))
1465     return true;
1466   if (SanitizerBL.isBlacklistedLocation(Loc, Category))
1467     return true;
1468   // Check global type.
1469   if (!Ty.isNull()) {
1470     // Drill down the array types: if global variable of a fixed type is
1471     // blacklisted, we also don't instrument arrays of them.
1472     while (auto AT = dyn_cast<ArrayType>(Ty.getTypePtr()))
1473       Ty = AT->getElementType();
1474     Ty = Ty.getCanonicalType().getUnqualifiedType();
1475     // We allow to blacklist only record types (classes, structs etc.)
1476     if (Ty->isRecordType()) {
1477       std::string TypeStr = Ty.getAsString(getContext().getPrintingPolicy());
1478       if (SanitizerBL.isBlacklistedType(TypeStr, Category))
1479         return true;
1480     }
1481   }
1482   return false;
1483 }
1484
1485 bool CodeGenModule::MustBeEmitted(const ValueDecl *Global) {
1486   // Never defer when EmitAllDecls is specified.
1487   if (LangOpts.EmitAllDecls)
1488     return true;
1489
1490   return getContext().DeclMustBeEmitted(Global);
1491 }
1492
1493 bool CodeGenModule::MayBeEmittedEagerly(const ValueDecl *Global) {
1494   if (const auto *FD = dyn_cast<FunctionDecl>(Global))
1495     if (FD->getTemplateSpecializationKind() == TSK_ImplicitInstantiation)
1496       // Implicit template instantiations may change linkage if they are later
1497       // explicitly instantiated, so they should not be emitted eagerly.
1498       return false;
1499   if (const auto *VD = dyn_cast<VarDecl>(Global))
1500     if (Context.getInlineVariableDefinitionKind(VD) ==
1501         ASTContext::InlineVariableDefinitionKind::WeakUnknown)
1502       // A definition of an inline constexpr static data member may change
1503       // linkage later if it's redeclared outside the class.
1504       return false;
1505   // If OpenMP is enabled and threadprivates must be generated like TLS, delay
1506   // codegen for global variables, because they may be marked as threadprivate.
1507   if (LangOpts.OpenMP && LangOpts.OpenMPUseTLS &&
1508       getContext().getTargetInfo().isTLSSupported() && isa<VarDecl>(Global))
1509     return false;
1510
1511   return true;
1512 }
1513
1514 ConstantAddress CodeGenModule::GetAddrOfUuidDescriptor(
1515     const CXXUuidofExpr* E) {
1516   // Sema has verified that IIDSource has a __declspec(uuid()), and that its
1517   // well-formed.
1518   StringRef Uuid = E->getUuidStr();
1519   std::string Name = "_GUID_" + Uuid.lower();
1520   std::replace(Name.begin(), Name.end(), '-', '_');
1521
1522   // The UUID descriptor should be pointer aligned.
1523   CharUnits Alignment = CharUnits::fromQuantity(PointerAlignInBytes);
1524
1525   // Look for an existing global.
1526   if (llvm::GlobalVariable *GV = getModule().getNamedGlobal(Name))
1527     return ConstantAddress(GV, Alignment);
1528
1529   llvm::Constant *Init = EmitUuidofInitializer(Uuid);
1530   assert(Init && "failed to initialize as constant");
1531
1532   auto *GV = new llvm::GlobalVariable(
1533       getModule(), Init->getType(),
1534       /*isConstant=*/true, llvm::GlobalValue::LinkOnceODRLinkage, Init, Name);
1535   if (supportsCOMDAT())
1536     GV->setComdat(TheModule.getOrInsertComdat(GV->getName()));
1537   return ConstantAddress(GV, Alignment);
1538 }
1539
1540 ConstantAddress CodeGenModule::GetWeakRefReference(const ValueDecl *VD) {
1541   const AliasAttr *AA = VD->getAttr<AliasAttr>();
1542   assert(AA && "No alias?");
1543
1544   CharUnits Alignment = getContext().getDeclAlign(VD);
1545   llvm::Type *DeclTy = getTypes().ConvertTypeForMem(VD->getType());
1546
1547   // See if there is already something with the target's name in the module.
1548   llvm::GlobalValue *Entry = GetGlobalValue(AA->getAliasee());
1549   if (Entry) {
1550     unsigned AS = getContext().getTargetAddressSpace(VD->getType());
1551     auto Ptr = llvm::ConstantExpr::getBitCast(Entry, DeclTy->getPointerTo(AS));
1552     return ConstantAddress(Ptr, Alignment);
1553   }
1554
1555   llvm::Constant *Aliasee;
1556   if (isa<llvm::FunctionType>(DeclTy))
1557     Aliasee = GetOrCreateLLVMFunction(AA->getAliasee(), DeclTy,
1558                                       GlobalDecl(cast<FunctionDecl>(VD)),
1559                                       /*ForVTable=*/false);
1560   else
1561     Aliasee = GetOrCreateLLVMGlobal(AA->getAliasee(),
1562                                     llvm::PointerType::getUnqual(DeclTy),
1563                                     nullptr);
1564
1565   auto *F = cast<llvm::GlobalValue>(Aliasee);
1566   F->setLinkage(llvm::Function::ExternalWeakLinkage);
1567   WeakRefReferences.insert(F);
1568
1569   return ConstantAddress(Aliasee, Alignment);
1570 }
1571
1572 void CodeGenModule::EmitGlobal(GlobalDecl GD) {
1573   const auto *Global = cast<ValueDecl>(GD.getDecl());
1574
1575   // Weak references don't produce any output by themselves.
1576   if (Global->hasAttr<WeakRefAttr>())
1577     return;
1578
1579   // If this is an alias definition (which otherwise looks like a declaration)
1580   // emit it now.
1581   if (Global->hasAttr<AliasAttr>())
1582     return EmitAliasDefinition(GD);
1583
1584   // IFunc like an alias whose value is resolved at runtime by calling resolver.
1585   if (Global->hasAttr<IFuncAttr>())
1586     return emitIFuncDefinition(GD);
1587
1588   // If this is CUDA, be selective about which declarations we emit.
1589   if (LangOpts.CUDA) {
1590     if (LangOpts.CUDAIsDevice) {
1591       if (!Global->hasAttr<CUDADeviceAttr>() &&
1592           !Global->hasAttr<CUDAGlobalAttr>() &&
1593           !Global->hasAttr<CUDAConstantAttr>() &&
1594           !Global->hasAttr<CUDASharedAttr>())
1595         return;
1596     } else {
1597       // We need to emit host-side 'shadows' for all global
1598       // device-side variables because the CUDA runtime needs their
1599       // size and host-side address in order to provide access to
1600       // their device-side incarnations.
1601
1602       // So device-only functions are the only things we skip.
1603       if (isa<FunctionDecl>(Global) && !Global->hasAttr<CUDAHostAttr>() &&
1604           Global->hasAttr<CUDADeviceAttr>())
1605         return;
1606
1607       assert((isa<FunctionDecl>(Global) || isa<VarDecl>(Global)) &&
1608              "Expected Variable or Function");
1609     }
1610   }
1611
1612   if (LangOpts.OpenMP) {
1613     // If this is OpenMP device, check if it is legal to emit this global
1614     // normally.
1615     if (OpenMPRuntime && OpenMPRuntime->emitTargetGlobal(GD))
1616       return;
1617     if (auto *DRD = dyn_cast<OMPDeclareReductionDecl>(Global)) {
1618       if (MustBeEmitted(Global))
1619         EmitOMPDeclareReduction(DRD);
1620       return;
1621     }
1622   }
1623
1624   // Ignore declarations, they will be emitted on their first use.
1625   if (const auto *FD = dyn_cast<FunctionDecl>(Global)) {
1626     // Forward declarations are emitted lazily on first use.
1627     if (!FD->doesThisDeclarationHaveABody()) {
1628       if (!FD->doesDeclarationForceExternallyVisibleDefinition())
1629         return;
1630
1631       StringRef MangledName = getMangledName(GD);
1632
1633       // Compute the function info and LLVM type.
1634       const CGFunctionInfo &FI = getTypes().arrangeGlobalDeclaration(GD);
1635       llvm::Type *Ty = getTypes().GetFunctionType(FI);
1636
1637       GetOrCreateLLVMFunction(MangledName, Ty, GD, /*ForVTable=*/false,
1638                               /*DontDefer=*/false);
1639       return;
1640     }
1641   } else {
1642     const auto *VD = cast<VarDecl>(Global);
1643     assert(VD->isFileVarDecl() && "Cannot emit local var decl as global.");
1644     // We need to emit device-side global CUDA variables even if a
1645     // variable does not have a definition -- we still need to define
1646     // host-side shadow for it.
1647     bool MustEmitForCuda = LangOpts.CUDA && !LangOpts.CUDAIsDevice &&
1648                            !VD->hasDefinition() &&
1649                            (VD->hasAttr<CUDAConstantAttr>() ||
1650                             VD->hasAttr<CUDADeviceAttr>());
1651     if (!MustEmitForCuda &&
1652         VD->isThisDeclarationADefinition() != VarDecl::Definition &&
1653         !Context.isMSStaticDataMemberInlineDefinition(VD)) {
1654       // If this declaration may have caused an inline variable definition to
1655       // change linkage, make sure that it's emitted.
1656       if (Context.getInlineVariableDefinitionKind(VD) ==
1657           ASTContext::InlineVariableDefinitionKind::Strong)
1658         GetAddrOfGlobalVar(VD);
1659       return;
1660     }
1661   }
1662
1663   // Defer code generation to first use when possible, e.g. if this is an inline
1664   // function. If the global must always be emitted, do it eagerly if possible
1665   // to benefit from cache locality.
1666   if (MustBeEmitted(Global) && MayBeEmittedEagerly(Global)) {
1667     // Emit the definition if it can't be deferred.
1668     EmitGlobalDefinition(GD);
1669     return;
1670   }
1671
1672   // If we're deferring emission of a C++ variable with an
1673   // initializer, remember the order in which it appeared in the file.
1674   if (getLangOpts().CPlusPlus && isa<VarDecl>(Global) &&
1675       cast<VarDecl>(Global)->hasInit()) {
1676     DelayedCXXInitPosition[Global] = CXXGlobalInits.size();
1677     CXXGlobalInits.push_back(nullptr);
1678   }
1679
1680   StringRef MangledName = getMangledName(GD);
1681   if (llvm::GlobalValue *GV = GetGlobalValue(MangledName)) {
1682     // The value has already been used and should therefore be emitted.
1683     addDeferredDeclToEmit(GV, GD);
1684   } else if (MustBeEmitted(Global)) {
1685     // The value must be emitted, but cannot be emitted eagerly.
1686     assert(!MayBeEmittedEagerly(Global));
1687     addDeferredDeclToEmit(/*GV=*/nullptr, GD);
1688   } else {
1689     // Otherwise, remember that we saw a deferred decl with this name.  The
1690     // first use of the mangled name will cause it to move into
1691     // DeferredDeclsToEmit.
1692     DeferredDecls[MangledName] = GD;
1693   }
1694 }
1695
1696 namespace {
1697   struct FunctionIsDirectlyRecursive :
1698     public RecursiveASTVisitor<FunctionIsDirectlyRecursive> {
1699     const StringRef Name;
1700     const Builtin::Context &BI;
1701     bool Result;
1702     FunctionIsDirectlyRecursive(StringRef N, const Builtin::Context &C) :
1703       Name(N), BI(C), Result(false) {
1704     }
1705     typedef RecursiveASTVisitor<FunctionIsDirectlyRecursive> Base;
1706
1707     bool TraverseCallExpr(CallExpr *E) {
1708       const FunctionDecl *FD = E->getDirectCallee();
1709       if (!FD)
1710         return true;
1711       AsmLabelAttr *Attr = FD->getAttr<AsmLabelAttr>();
1712       if (Attr && Name == Attr->getLabel()) {
1713         Result = true;
1714         return false;
1715       }
1716       unsigned BuiltinID = FD->getBuiltinID();
1717       if (!BuiltinID || !BI.isLibFunction(BuiltinID))
1718         return true;
1719       StringRef BuiltinName = BI.getName(BuiltinID);
1720       if (BuiltinName.startswith("__builtin_") &&
1721           Name == BuiltinName.slice(strlen("__builtin_"), StringRef::npos)) {
1722         Result = true;
1723         return false;
1724       }
1725       return true;
1726     }
1727   };
1728
1729   struct DLLImportFunctionVisitor
1730       : public RecursiveASTVisitor<DLLImportFunctionVisitor> {
1731     bool SafeToInline = true;
1732
1733     bool shouldVisitImplicitCode() const { return true; }
1734
1735     bool VisitVarDecl(VarDecl *VD) {
1736       // A thread-local variable cannot be imported.
1737       SafeToInline = !VD->getTLSKind();
1738       return SafeToInline;
1739     }
1740
1741     // Make sure we're not referencing non-imported vars or functions.
1742     bool VisitDeclRefExpr(DeclRefExpr *E) {
1743       ValueDecl *VD = E->getDecl();
1744       if (isa<FunctionDecl>(VD))
1745         SafeToInline = VD->hasAttr<DLLImportAttr>();
1746       else if (VarDecl *V = dyn_cast<VarDecl>(VD))
1747         SafeToInline = !V->hasGlobalStorage() || V->hasAttr<DLLImportAttr>();
1748       return SafeToInline;
1749     }
1750     bool VisitCXXConstructExpr(CXXConstructExpr *E) {
1751       SafeToInline = E->getConstructor()->hasAttr<DLLImportAttr>();
1752       return SafeToInline;
1753     }
1754     bool VisitCXXDeleteExpr(CXXDeleteExpr *E) {
1755       SafeToInline = E->getOperatorDelete()->hasAttr<DLLImportAttr>();
1756       return SafeToInline;
1757     }
1758     bool VisitCXXNewExpr(CXXNewExpr *E) {
1759       SafeToInline = E->getOperatorNew()->hasAttr<DLLImportAttr>();
1760       return SafeToInline;
1761     }
1762   };
1763 }
1764
1765 // isTriviallyRecursive - Check if this function calls another
1766 // decl that, because of the asm attribute or the other decl being a builtin,
1767 // ends up pointing to itself.
1768 bool
1769 CodeGenModule::isTriviallyRecursive(const FunctionDecl *FD) {
1770   StringRef Name;
1771   if (getCXXABI().getMangleContext().shouldMangleDeclName(FD)) {
1772     // asm labels are a special kind of mangling we have to support.
1773     AsmLabelAttr *Attr = FD->getAttr<AsmLabelAttr>();
1774     if (!Attr)
1775       return false;
1776     Name = Attr->getLabel();
1777   } else {
1778     Name = FD->getName();
1779   }
1780
1781   FunctionIsDirectlyRecursive Walker(Name, Context.BuiltinInfo);
1782   Walker.TraverseFunctionDecl(const_cast<FunctionDecl*>(FD));
1783   return Walker.Result;
1784 }
1785
1786 // Check if T is a class type with a destructor that's not dllimport.
1787 static bool HasNonDllImportDtor(QualType T) {
1788   if (const RecordType *RT = dyn_cast<RecordType>(T))
1789     if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl()))
1790       if (RD->getDestructor() && !RD->getDestructor()->hasAttr<DLLImportAttr>())
1791         return true;
1792
1793   return false;
1794 }
1795
1796 bool CodeGenModule::shouldEmitFunction(GlobalDecl GD) {
1797   if (getFunctionLinkage(GD) != llvm::Function::AvailableExternallyLinkage)
1798     return true;
1799   const auto *F = cast<FunctionDecl>(GD.getDecl());
1800   if (CodeGenOpts.OptimizationLevel == 0 && !F->hasAttr<AlwaysInlineAttr>())
1801     return false;
1802
1803   if (F->hasAttr<DLLImportAttr>()) {
1804     // Check whether it would be safe to inline this dllimport function.
1805     DLLImportFunctionVisitor Visitor;
1806     Visitor.TraverseFunctionDecl(const_cast<FunctionDecl*>(F));
1807     if (!Visitor.SafeToInline)
1808       return false;
1809
1810     if (const CXXDestructorDecl *Dtor = dyn_cast<CXXDestructorDecl>(F)) {
1811       // Implicit destructor invocations aren't captured in the AST, so the
1812       // check above can't see them. Check for them manually here.
1813       for (const Decl *Member : Dtor->getParent()->decls())
1814         if (isa<FieldDecl>(Member))
1815           if (HasNonDllImportDtor(cast<FieldDecl>(Member)->getType()))
1816             return false;
1817       for (const CXXBaseSpecifier &B : Dtor->getParent()->bases())
1818         if (HasNonDllImportDtor(B.getType()))
1819           return false;
1820     }
1821   }
1822
1823   // PR9614. Avoid cases where the source code is lying to us. An available
1824   // externally function should have an equivalent function somewhere else,
1825   // but a function that calls itself is clearly not equivalent to the real
1826   // implementation.
1827   // This happens in glibc's btowc and in some configure checks.
1828   return !isTriviallyRecursive(F);
1829 }
1830
1831 /// If the type for the method's class was generated by
1832 /// CGDebugInfo::createContextChain(), the cache contains only a
1833 /// limited DIType without any declarations. Since EmitFunctionStart()
1834 /// needs to find the canonical declaration for each method, we need
1835 /// to construct the complete type prior to emitting the method.
1836 void CodeGenModule::CompleteDIClassType(const CXXMethodDecl* D) {
1837   if (!D->isInstance())
1838     return;
1839
1840   if (CGDebugInfo *DI = getModuleDebugInfo())
1841     if (getCodeGenOpts().getDebugInfo() >= codegenoptions::LimitedDebugInfo) {
1842       const auto *ThisPtr = cast<PointerType>(D->getThisType(getContext()));
1843       DI->getOrCreateRecordType(ThisPtr->getPointeeType(), D->getLocation());
1844     }
1845 }
1846
1847 void CodeGenModule::EmitGlobalDefinition(GlobalDecl GD, llvm::GlobalValue *GV) {
1848   const auto *D = cast<ValueDecl>(GD.getDecl());
1849
1850   PrettyStackTraceDecl CrashInfo(const_cast<ValueDecl *>(D), D->getLocation(),
1851                                  Context.getSourceManager(),
1852                                  "Generating code for declaration");
1853
1854   if (isa<FunctionDecl>(D)) {
1855     // At -O0, don't generate IR for functions with available_externally
1856     // linkage.
1857     if (!shouldEmitFunction(GD))
1858       return;
1859
1860     if (const auto *Method = dyn_cast<CXXMethodDecl>(D)) {
1861       CompleteDIClassType(Method);
1862       // Make sure to emit the definition(s) before we emit the thunks.
1863       // This is necessary for the generation of certain thunks.
1864       if (const auto *CD = dyn_cast<CXXConstructorDecl>(Method))
1865         ABI->emitCXXStructor(CD, getFromCtorType(GD.getCtorType()));
1866       else if (const auto *DD = dyn_cast<CXXDestructorDecl>(Method))
1867         ABI->emitCXXStructor(DD, getFromDtorType(GD.getDtorType()));
1868       else
1869         EmitGlobalFunctionDefinition(GD, GV);
1870
1871       if (Method->isVirtual())
1872         getVTables().EmitThunks(GD);
1873
1874       return;
1875     }
1876
1877     return EmitGlobalFunctionDefinition(GD, GV);
1878   }
1879
1880   if (const auto *VD = dyn_cast<VarDecl>(D))
1881     return EmitGlobalVarDefinition(VD, !VD->hasDefinition());
1882
1883   llvm_unreachable("Invalid argument to EmitGlobalDefinition()");
1884 }
1885
1886 static void ReplaceUsesOfNonProtoTypeWithRealFunction(llvm::GlobalValue *Old,
1887                                                       llvm::Function *NewFn);
1888
1889 /// GetOrCreateLLVMFunction - If the specified mangled name is not in the
1890 /// module, create and return an llvm Function with the specified type. If there
1891 /// is something in the module with the specified name, return it potentially
1892 /// bitcasted to the right type.
1893 ///
1894 /// If D is non-null, it specifies a decl that correspond to this.  This is used
1895 /// to set the attributes on the function when it is first created.
1896 llvm::Constant *
1897 CodeGenModule::GetOrCreateLLVMFunction(StringRef MangledName,
1898                                        llvm::Type *Ty,
1899                                        GlobalDecl GD, bool ForVTable,
1900                                        bool DontDefer, bool IsThunk,
1901                                        llvm::AttributeSet ExtraAttrs,
1902                                        ForDefinition_t IsForDefinition) {
1903   const Decl *D = GD.getDecl();
1904
1905   // Lookup the entry, lazily creating it if necessary.
1906   llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
1907   if (Entry) {
1908     if (WeakRefReferences.erase(Entry)) {
1909       const FunctionDecl *FD = cast_or_null<FunctionDecl>(D);
1910       if (FD && !FD->hasAttr<WeakAttr>())
1911         Entry->setLinkage(llvm::Function::ExternalLinkage);
1912     }
1913
1914     // Handle dropped DLL attributes.
1915     if (D && !D->hasAttr<DLLImportAttr>() && !D->hasAttr<DLLExportAttr>())
1916       Entry->setDLLStorageClass(llvm::GlobalValue::DefaultStorageClass);
1917
1918     // If there are two attempts to define the same mangled name, issue an
1919     // error.
1920     if (IsForDefinition && !Entry->isDeclaration()) {
1921       GlobalDecl OtherGD;
1922       // Check that GD is not yet in DiagnosedConflictingDefinitions is required
1923       // to make sure that we issue an error only once.
1924       if (lookupRepresentativeDecl(MangledName, OtherGD) &&
1925           (GD.getCanonicalDecl().getDecl() !=
1926            OtherGD.getCanonicalDecl().getDecl()) &&
1927           DiagnosedConflictingDefinitions.insert(GD).second) {
1928         getDiags().Report(D->getLocation(),
1929                           diag::err_duplicate_mangled_name);
1930         getDiags().Report(OtherGD.getDecl()->getLocation(),
1931                           diag::note_previous_definition);
1932       }
1933     }
1934
1935     if ((isa<llvm::Function>(Entry) || isa<llvm::GlobalAlias>(Entry)) &&
1936         (Entry->getType()->getElementType() == Ty)) {
1937       return Entry;
1938     }
1939
1940     // Make sure the result is of the correct type.
1941     // (If function is requested for a definition, we always need to create a new
1942     // function, not just return a bitcast.)
1943     if (!IsForDefinition)
1944       return llvm::ConstantExpr::getBitCast(Entry, Ty->getPointerTo());
1945   }
1946
1947   // This function doesn't have a complete type (for example, the return
1948   // type is an incomplete struct). Use a fake type instead, and make
1949   // sure not to try to set attributes.
1950   bool IsIncompleteFunction = false;
1951
1952   llvm::FunctionType *FTy;
1953   if (isa<llvm::FunctionType>(Ty)) {
1954     FTy = cast<llvm::FunctionType>(Ty);
1955   } else {
1956     FTy = llvm::FunctionType::get(VoidTy, false);
1957     IsIncompleteFunction = true;
1958   }
1959
1960   llvm::Function *F =
1961       llvm::Function::Create(FTy, llvm::Function::ExternalLinkage,
1962                              Entry ? StringRef() : MangledName, &getModule());
1963
1964   // If we already created a function with the same mangled name (but different
1965   // type) before, take its name and add it to the list of functions to be
1966   // replaced with F at the end of CodeGen.
1967   //
1968   // This happens if there is a prototype for a function (e.g. "int f()") and
1969   // then a definition of a different type (e.g. "int f(int x)").
1970   if (Entry) {
1971     F->takeName(Entry);
1972
1973     // This might be an implementation of a function without a prototype, in
1974     // which case, try to do special replacement of calls which match the new
1975     // prototype.  The really key thing here is that we also potentially drop
1976     // arguments from the call site so as to make a direct call, which makes the
1977     // inliner happier and suppresses a number of optimizer warnings (!) about
1978     // dropping arguments.
1979     if (!Entry->use_empty()) {
1980       ReplaceUsesOfNonProtoTypeWithRealFunction(Entry, F);
1981       Entry->removeDeadConstantUsers();
1982     }
1983
1984     llvm::Constant *BC = llvm::ConstantExpr::getBitCast(
1985         F, Entry->getType()->getElementType()->getPointerTo());
1986     addGlobalValReplacement(Entry, BC);
1987   }
1988
1989   assert(F->getName() == MangledName && "name was uniqued!");
1990   if (D)
1991     SetFunctionAttributes(GD, F, IsIncompleteFunction, IsThunk);
1992   if (ExtraAttrs.hasAttributes(llvm::AttributeSet::FunctionIndex)) {
1993     llvm::AttrBuilder B(ExtraAttrs, llvm::AttributeSet::FunctionIndex);
1994     F->addAttributes(llvm::AttributeSet::FunctionIndex,
1995                      llvm::AttributeSet::get(VMContext,
1996                                              llvm::AttributeSet::FunctionIndex,
1997                                              B));
1998   }
1999
2000   if (!DontDefer) {
2001     // All MSVC dtors other than the base dtor are linkonce_odr and delegate to
2002     // each other bottoming out with the base dtor.  Therefore we emit non-base
2003     // dtors on usage, even if there is no dtor definition in the TU.
2004     if (D && isa<CXXDestructorDecl>(D) &&
2005         getCXXABI().useThunkForDtorVariant(cast<CXXDestructorDecl>(D),
2006                                            GD.getDtorType()))
2007       addDeferredDeclToEmit(F, GD);
2008
2009     // This is the first use or definition of a mangled name.  If there is a
2010     // deferred decl with this name, remember that we need to emit it at the end
2011     // of the file.
2012     auto DDI = DeferredDecls.find(MangledName);
2013     if (DDI != DeferredDecls.end()) {
2014       // Move the potentially referenced deferred decl to the
2015       // DeferredDeclsToEmit list, and remove it from DeferredDecls (since we
2016       // don't need it anymore).
2017       addDeferredDeclToEmit(F, DDI->second);
2018       DeferredDecls.erase(DDI);
2019
2020       // Otherwise, there are cases we have to worry about where we're
2021       // using a declaration for which we must emit a definition but where
2022       // we might not find a top-level definition:
2023       //   - member functions defined inline in their classes
2024       //   - friend functions defined inline in some class
2025       //   - special member functions with implicit definitions
2026       // If we ever change our AST traversal to walk into class methods,
2027       // this will be unnecessary.
2028       //
2029       // We also don't emit a definition for a function if it's going to be an
2030       // entry in a vtable, unless it's already marked as used.
2031     } else if (getLangOpts().CPlusPlus && D) {
2032       // Look for a declaration that's lexically in a record.
2033       for (const auto *FD = cast<FunctionDecl>(D)->getMostRecentDecl(); FD;
2034            FD = FD->getPreviousDecl()) {
2035         if (isa<CXXRecordDecl>(FD->getLexicalDeclContext())) {
2036           if (FD->doesThisDeclarationHaveABody()) {
2037             addDeferredDeclToEmit(F, GD.getWithDecl(FD));
2038             break;
2039           }
2040         }
2041       }
2042     }
2043   }
2044
2045   // Make sure the result is of the requested type.
2046   if (!IsIncompleteFunction) {
2047     assert(F->getType()->getElementType() == Ty);
2048     return F;
2049   }
2050
2051   llvm::Type *PTy = llvm::PointerType::getUnqual(Ty);
2052   return llvm::ConstantExpr::getBitCast(F, PTy);
2053 }
2054
2055 /// GetAddrOfFunction - Return the address of the given function.  If Ty is
2056 /// non-null, then this function will use the specified type if it has to
2057 /// create it (this occurs when we see a definition of the function).
2058 llvm::Constant *CodeGenModule::GetAddrOfFunction(GlobalDecl GD,
2059                                                  llvm::Type *Ty,
2060                                                  bool ForVTable,
2061                                                  bool DontDefer,
2062                                               ForDefinition_t IsForDefinition) {
2063   // If there was no specific requested type, just convert it now.
2064   if (!Ty) {
2065     const auto *FD = cast<FunctionDecl>(GD.getDecl());
2066     auto CanonTy = Context.getCanonicalType(FD->getType());
2067     Ty = getTypes().ConvertFunctionType(CanonTy, FD);
2068   }
2069
2070   StringRef MangledName = getMangledName(GD);
2071   return GetOrCreateLLVMFunction(MangledName, Ty, GD, ForVTable, DontDefer,
2072                                  /*IsThunk=*/false, llvm::AttributeSet(),
2073                                  IsForDefinition);
2074 }
2075
2076 static const FunctionDecl *
2077 GetRuntimeFunctionDecl(ASTContext &C, StringRef Name) {
2078   TranslationUnitDecl *TUDecl = C.getTranslationUnitDecl();
2079   DeclContext *DC = TranslationUnitDecl::castToDeclContext(TUDecl);
2080
2081   IdentifierInfo &CII = C.Idents.get(Name);
2082   for (const auto &Result : DC->lookup(&CII))
2083     if (const auto FD = dyn_cast<FunctionDecl>(Result))
2084       return FD;
2085
2086   if (!C.getLangOpts().CPlusPlus)
2087     return nullptr;
2088
2089   // Demangle the premangled name from getTerminateFn()
2090   IdentifierInfo &CXXII =
2091       (Name == "_ZSt9terminatev" || Name == "\01?terminate@@YAXXZ")
2092           ? C.Idents.get("terminate")
2093           : C.Idents.get(Name);
2094
2095   for (const auto &N : {"__cxxabiv1", "std"}) {
2096     IdentifierInfo &NS = C.Idents.get(N);
2097     for (const auto &Result : DC->lookup(&NS)) {
2098       NamespaceDecl *ND = dyn_cast<NamespaceDecl>(Result);
2099       if (auto LSD = dyn_cast<LinkageSpecDecl>(Result))
2100         for (const auto &Result : LSD->lookup(&NS))
2101           if ((ND = dyn_cast<NamespaceDecl>(Result)))
2102             break;
2103
2104       if (ND)
2105         for (const auto &Result : ND->lookup(&CXXII))
2106           if (const auto *FD = dyn_cast<FunctionDecl>(Result))
2107             return FD;
2108     }
2109   }
2110
2111   return nullptr;
2112 }
2113
2114 /// CreateRuntimeFunction - Create a new runtime function with the specified
2115 /// type and name.
2116 llvm::Constant *
2117 CodeGenModule::CreateRuntimeFunction(llvm::FunctionType *FTy, StringRef Name,
2118                                      llvm::AttributeSet ExtraAttrs,
2119                                      bool Local) {
2120   llvm::Constant *C =
2121       GetOrCreateLLVMFunction(Name, FTy, GlobalDecl(), /*ForVTable=*/false,
2122                               /*DontDefer=*/false, /*IsThunk=*/false,
2123                               ExtraAttrs);
2124
2125   if (auto *F = dyn_cast<llvm::Function>(C)) {
2126     if (F->empty()) {
2127       F->setCallingConv(getRuntimeCC());
2128
2129       if (!Local && getTriple().isOSBinFormatCOFF() &&
2130           !getCodeGenOpts().LTOVisibilityPublicStd) {
2131         const FunctionDecl *FD = GetRuntimeFunctionDecl(Context, Name);
2132         if (!FD || FD->hasAttr<DLLImportAttr>()) {
2133           F->setDLLStorageClass(llvm::GlobalValue::DLLImportStorageClass);
2134           F->setLinkage(llvm::GlobalValue::ExternalLinkage);
2135         }
2136       }
2137     }
2138   }
2139
2140   return C;
2141 }
2142
2143 /// CreateBuiltinFunction - Create a new builtin function with the specified
2144 /// type and name.
2145 llvm::Constant *
2146 CodeGenModule::CreateBuiltinFunction(llvm::FunctionType *FTy,
2147                                      StringRef Name,
2148                                      llvm::AttributeSet ExtraAttrs) {
2149   llvm::Constant *C =
2150       GetOrCreateLLVMFunction(Name, FTy, GlobalDecl(), /*ForVTable=*/false,
2151                               /*DontDefer=*/false, /*IsThunk=*/false, ExtraAttrs);
2152   if (auto *F = dyn_cast<llvm::Function>(C))
2153     if (F->empty())
2154       F->setCallingConv(getBuiltinCC());
2155   return C;
2156 }
2157
2158 /// isTypeConstant - Determine whether an object of this type can be emitted
2159 /// as a constant.
2160 ///
2161 /// If ExcludeCtor is true, the duration when the object's constructor runs
2162 /// will not be considered. The caller will need to verify that the object is
2163 /// not written to during its construction.
2164 bool CodeGenModule::isTypeConstant(QualType Ty, bool ExcludeCtor) {
2165   if (!Ty.isConstant(Context) && !Ty->isReferenceType())
2166     return false;
2167
2168   if (Context.getLangOpts().CPlusPlus) {
2169     if (const CXXRecordDecl *Record
2170           = Context.getBaseElementType(Ty)->getAsCXXRecordDecl())
2171       return ExcludeCtor && !Record->hasMutableFields() &&
2172              Record->hasTrivialDestructor();
2173   }
2174
2175   return true;
2176 }
2177
2178 /// GetOrCreateLLVMGlobal - If the specified mangled name is not in the module,
2179 /// create and return an llvm GlobalVariable with the specified type.  If there
2180 /// is something in the module with the specified name, return it potentially
2181 /// bitcasted to the right type.
2182 ///
2183 /// If D is non-null, it specifies a decl that correspond to this.  This is used
2184 /// to set the attributes on the global when it is first created.
2185 ///
2186 /// If IsForDefinition is true, it is guranteed that an actual global with
2187 /// type Ty will be returned, not conversion of a variable with the same
2188 /// mangled name but some other type.
2189 llvm::Constant *
2190 CodeGenModule::GetOrCreateLLVMGlobal(StringRef MangledName,
2191                                      llvm::PointerType *Ty,
2192                                      const VarDecl *D,
2193                                      ForDefinition_t IsForDefinition) {
2194   // Lookup the entry, lazily creating it if necessary.
2195   llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
2196   if (Entry) {
2197     if (WeakRefReferences.erase(Entry)) {
2198       if (D && !D->hasAttr<WeakAttr>())
2199         Entry->setLinkage(llvm::Function::ExternalLinkage);
2200     }
2201
2202     // Handle dropped DLL attributes.
2203     if (D && !D->hasAttr<DLLImportAttr>() && !D->hasAttr<DLLExportAttr>())
2204       Entry->setDLLStorageClass(llvm::GlobalValue::DefaultStorageClass);
2205
2206     if (Entry->getType() == Ty)
2207       return Entry;
2208
2209     // If there are two attempts to define the same mangled name, issue an
2210     // error.
2211     if (IsForDefinition && !Entry->isDeclaration()) {
2212       GlobalDecl OtherGD;
2213       const VarDecl *OtherD;
2214
2215       // Check that D is not yet in DiagnosedConflictingDefinitions is required
2216       // to make sure that we issue an error only once.
2217       if (D && lookupRepresentativeDecl(MangledName, OtherGD) &&
2218           (D->getCanonicalDecl() != OtherGD.getCanonicalDecl().getDecl()) &&
2219           (OtherD = dyn_cast<VarDecl>(OtherGD.getDecl())) &&
2220           OtherD->hasInit() &&
2221           DiagnosedConflictingDefinitions.insert(D).second) {
2222         getDiags().Report(D->getLocation(),
2223                           diag::err_duplicate_mangled_name);
2224         getDiags().Report(OtherGD.getDecl()->getLocation(),
2225                           diag::note_previous_definition);
2226       }
2227     }
2228
2229     // Make sure the result is of the correct type.
2230     if (Entry->getType()->getAddressSpace() != Ty->getAddressSpace())
2231       return llvm::ConstantExpr::getAddrSpaceCast(Entry, Ty);
2232
2233     // (If global is requested for a definition, we always need to create a new
2234     // global, not just return a bitcast.)
2235     if (!IsForDefinition)
2236       return llvm::ConstantExpr::getBitCast(Entry, Ty);
2237   }
2238
2239   unsigned AddrSpace = GetGlobalVarAddressSpace(D, Ty->getAddressSpace());
2240   auto *GV = new llvm::GlobalVariable(
2241       getModule(), Ty->getElementType(), false,
2242       llvm::GlobalValue::ExternalLinkage, nullptr, MangledName, nullptr,
2243       llvm::GlobalVariable::NotThreadLocal, AddrSpace);
2244
2245   // If we already created a global with the same mangled name (but different
2246   // type) before, take its name and remove it from its parent.
2247   if (Entry) {
2248     GV->takeName(Entry);
2249
2250     if (!Entry->use_empty()) {
2251       llvm::Constant *NewPtrForOldDecl =
2252           llvm::ConstantExpr::getBitCast(GV, Entry->getType());
2253       Entry->replaceAllUsesWith(NewPtrForOldDecl);
2254     }
2255
2256     Entry->eraseFromParent();
2257   }
2258
2259   // This is the first use or definition of a mangled name.  If there is a
2260   // deferred decl with this name, remember that we need to emit it at the end
2261   // of the file.
2262   auto DDI = DeferredDecls.find(MangledName);
2263   if (DDI != DeferredDecls.end()) {
2264     // Move the potentially referenced deferred decl to the DeferredDeclsToEmit
2265     // list, and remove it from DeferredDecls (since we don't need it anymore).
2266     addDeferredDeclToEmit(GV, DDI->second);
2267     DeferredDecls.erase(DDI);
2268   }
2269
2270   // Handle things which are present even on external declarations.
2271   if (D) {
2272     // FIXME: This code is overly simple and should be merged with other global
2273     // handling.
2274     GV->setConstant(isTypeConstant(D->getType(), false));
2275
2276     GV->setAlignment(getContext().getDeclAlign(D).getQuantity());
2277
2278     setLinkageAndVisibilityForGV(GV, D);
2279
2280     if (D->getTLSKind()) {
2281       if (D->getTLSKind() == VarDecl::TLS_Dynamic)
2282         CXXThreadLocals.push_back(D);
2283       setTLSMode(GV, *D);
2284     }
2285
2286     // If required by the ABI, treat declarations of static data members with
2287     // inline initializers as definitions.
2288     if (getContext().isMSStaticDataMemberInlineDefinition(D)) {
2289       EmitGlobalVarDefinition(D);
2290     }
2291
2292     // Handle XCore specific ABI requirements.
2293     if (getTriple().getArch() == llvm::Triple::xcore &&
2294         D->getLanguageLinkage() == CLanguageLinkage &&
2295         D->getType().isConstant(Context) &&
2296         isExternallyVisible(D->getLinkageAndVisibility().getLinkage()))
2297       GV->setSection(".cp.rodata");
2298   }
2299
2300   if (AddrSpace != Ty->getAddressSpace())
2301     return llvm::ConstantExpr::getAddrSpaceCast(GV, Ty);
2302
2303   return GV;
2304 }
2305
2306 llvm::Constant *
2307 CodeGenModule::GetAddrOfGlobal(GlobalDecl GD,
2308                                ForDefinition_t IsForDefinition) {
2309   const Decl *D = GD.getDecl();
2310   if (isa<CXXConstructorDecl>(D))
2311     return getAddrOfCXXStructor(cast<CXXConstructorDecl>(D),
2312                                 getFromCtorType(GD.getCtorType()),
2313                                 /*FnInfo=*/nullptr, /*FnType=*/nullptr,
2314                                 /*DontDefer=*/false, IsForDefinition);
2315   else if (isa<CXXDestructorDecl>(D))
2316     return getAddrOfCXXStructor(cast<CXXDestructorDecl>(D),
2317                                 getFromDtorType(GD.getDtorType()),
2318                                 /*FnInfo=*/nullptr, /*FnType=*/nullptr,
2319                                 /*DontDefer=*/false, IsForDefinition);
2320   else if (isa<CXXMethodDecl>(D)) {
2321     auto FInfo = &getTypes().arrangeCXXMethodDeclaration(
2322         cast<CXXMethodDecl>(D));
2323     auto Ty = getTypes().GetFunctionType(*FInfo);
2324     return GetAddrOfFunction(GD, Ty, /*ForVTable=*/false, /*DontDefer=*/false,
2325                              IsForDefinition);
2326   } else if (isa<FunctionDecl>(D)) {
2327     const CGFunctionInfo &FI = getTypes().arrangeGlobalDeclaration(GD);
2328     llvm::FunctionType *Ty = getTypes().GetFunctionType(FI);
2329     return GetAddrOfFunction(GD, Ty, /*ForVTable=*/false, /*DontDefer=*/false,
2330                              IsForDefinition);
2331   } else
2332     return GetAddrOfGlobalVar(cast<VarDecl>(D), /*Ty=*/nullptr,
2333                               IsForDefinition);
2334 }
2335
2336 llvm::GlobalVariable *
2337 CodeGenModule::CreateOrReplaceCXXRuntimeVariable(StringRef Name,
2338                                       llvm::Type *Ty,
2339                                       llvm::GlobalValue::LinkageTypes Linkage) {
2340   llvm::GlobalVariable *GV = getModule().getNamedGlobal(Name);
2341   llvm::GlobalVariable *OldGV = nullptr;
2342
2343   if (GV) {
2344     // Check if the variable has the right type.
2345     if (GV->getType()->getElementType() == Ty)
2346       return GV;
2347
2348     // Because C++ name mangling, the only way we can end up with an already
2349     // existing global with the same name is if it has been declared extern "C".
2350     assert(GV->isDeclaration() && "Declaration has wrong type!");
2351     OldGV = GV;
2352   }
2353
2354   // Create a new variable.
2355   GV = new llvm::GlobalVariable(getModule(), Ty, /*isConstant=*/true,
2356                                 Linkage, nullptr, Name);
2357
2358   if (OldGV) {
2359     // Replace occurrences of the old variable if needed.
2360     GV->takeName(OldGV);
2361
2362     if (!OldGV->use_empty()) {
2363       llvm::Constant *NewPtrForOldDecl =
2364       llvm::ConstantExpr::getBitCast(GV, OldGV->getType());
2365       OldGV->replaceAllUsesWith(NewPtrForOldDecl);
2366     }
2367
2368     OldGV->eraseFromParent();
2369   }
2370
2371   if (supportsCOMDAT() && GV->isWeakForLinker() &&
2372       !GV->hasAvailableExternallyLinkage())
2373     GV->setComdat(TheModule.getOrInsertComdat(GV->getName()));
2374
2375   return GV;
2376 }
2377
2378 /// GetAddrOfGlobalVar - Return the llvm::Constant for the address of the
2379 /// given global variable.  If Ty is non-null and if the global doesn't exist,
2380 /// then it will be created with the specified type instead of whatever the
2381 /// normal requested type would be. If IsForDefinition is true, it is guranteed
2382 /// that an actual global with type Ty will be returned, not conversion of a
2383 /// variable with the same mangled name but some other type.
2384 llvm::Constant *CodeGenModule::GetAddrOfGlobalVar(const VarDecl *D,
2385                                                   llvm::Type *Ty,
2386                                            ForDefinition_t IsForDefinition) {
2387   assert(D->hasGlobalStorage() && "Not a global variable");
2388   QualType ASTTy = D->getType();
2389   if (!Ty)
2390     Ty = getTypes().ConvertTypeForMem(ASTTy);
2391
2392   llvm::PointerType *PTy =
2393     llvm::PointerType::get(Ty, getContext().getTargetAddressSpace(ASTTy));
2394
2395   StringRef MangledName = getMangledName(D);
2396   return GetOrCreateLLVMGlobal(MangledName, PTy, D, IsForDefinition);
2397 }
2398
2399 /// CreateRuntimeVariable - Create a new runtime global variable with the
2400 /// specified type and name.
2401 llvm::Constant *
2402 CodeGenModule::CreateRuntimeVariable(llvm::Type *Ty,
2403                                      StringRef Name) {
2404   return GetOrCreateLLVMGlobal(Name, llvm::PointerType::getUnqual(Ty), nullptr);
2405 }
2406
2407 void CodeGenModule::EmitTentativeDefinition(const VarDecl *D) {
2408   assert(!D->getInit() && "Cannot emit definite definitions here!");
2409
2410   StringRef MangledName = getMangledName(D);
2411   llvm::GlobalValue *GV = GetGlobalValue(MangledName);
2412
2413   // We already have a definition, not declaration, with the same mangled name.
2414   // Emitting of declaration is not required (and actually overwrites emitted
2415   // definition).
2416   if (GV && !GV->isDeclaration())
2417     return;
2418
2419   // If we have not seen a reference to this variable yet, place it into the
2420   // deferred declarations table to be emitted if needed later.
2421   if (!MustBeEmitted(D) && !GV) {
2422       DeferredDecls[MangledName] = D;
2423       return;
2424   }
2425
2426   // The tentative definition is the only definition.
2427   EmitGlobalVarDefinition(D);
2428 }
2429
2430 CharUnits CodeGenModule::GetTargetTypeStoreSize(llvm::Type *Ty) const {
2431   return Context.toCharUnitsFromBits(
2432       getDataLayout().getTypeStoreSizeInBits(Ty));
2433 }
2434
2435 unsigned CodeGenModule::GetGlobalVarAddressSpace(const VarDecl *D,
2436                                                  unsigned AddrSpace) {
2437   if (D && LangOpts.CUDA && LangOpts.CUDAIsDevice) {
2438     if (D->hasAttr<CUDAConstantAttr>())
2439       AddrSpace = getContext().getTargetAddressSpace(LangAS::cuda_constant);
2440     else if (D->hasAttr<CUDASharedAttr>())
2441       AddrSpace = getContext().getTargetAddressSpace(LangAS::cuda_shared);
2442     else
2443       AddrSpace = getContext().getTargetAddressSpace(LangAS::cuda_device);
2444   }
2445
2446   return AddrSpace;
2447 }
2448
2449 template<typename SomeDecl>
2450 void CodeGenModule::MaybeHandleStaticInExternC(const SomeDecl *D,
2451                                                llvm::GlobalValue *GV) {
2452   if (!getLangOpts().CPlusPlus)
2453     return;
2454
2455   // Must have 'used' attribute, or else inline assembly can't rely on
2456   // the name existing.
2457   if (!D->template hasAttr<UsedAttr>())
2458     return;
2459
2460   // Must have internal linkage and an ordinary name.
2461   if (!D->getIdentifier() || D->getFormalLinkage() != InternalLinkage)
2462     return;
2463
2464   // Must be in an extern "C" context. Entities declared directly within
2465   // a record are not extern "C" even if the record is in such a context.
2466   const SomeDecl *First = D->getFirstDecl();
2467   if (First->getDeclContext()->isRecord() || !First->isInExternCContext())
2468     return;
2469
2470   // OK, this is an internal linkage entity inside an extern "C" linkage
2471   // specification. Make a note of that so we can give it the "expected"
2472   // mangled name if nothing else is using that name.
2473   std::pair<StaticExternCMap::iterator, bool> R =
2474       StaticExternCValues.insert(std::make_pair(D->getIdentifier(), GV));
2475
2476   // If we have multiple internal linkage entities with the same name
2477   // in extern "C" regions, none of them gets that name.
2478   if (!R.second)
2479     R.first->second = nullptr;
2480 }
2481
2482 static bool shouldBeInCOMDAT(CodeGenModule &CGM, const Decl &D) {
2483   if (!CGM.supportsCOMDAT())
2484     return false;
2485
2486   if (D.hasAttr<SelectAnyAttr>())
2487     return true;
2488
2489   GVALinkage Linkage;
2490   if (auto *VD = dyn_cast<VarDecl>(&D))
2491     Linkage = CGM.getContext().GetGVALinkageForVariable(VD);
2492   else
2493     Linkage = CGM.getContext().GetGVALinkageForFunction(cast<FunctionDecl>(&D));
2494
2495   switch (Linkage) {
2496   case GVA_Internal:
2497   case GVA_AvailableExternally:
2498   case GVA_StrongExternal:
2499     return false;
2500   case GVA_DiscardableODR:
2501   case GVA_StrongODR:
2502     return true;
2503   }
2504   llvm_unreachable("No such linkage");
2505 }
2506
2507 void CodeGenModule::maybeSetTrivialComdat(const Decl &D,
2508                                           llvm::GlobalObject &GO) {
2509   if (!shouldBeInCOMDAT(*this, D))
2510     return;
2511   GO.setComdat(TheModule.getOrInsertComdat(GO.getName()));
2512 }
2513
2514 /// Pass IsTentative as true if you want to create a tentative definition.
2515 void CodeGenModule::EmitGlobalVarDefinition(const VarDecl *D,
2516                                             bool IsTentative) {
2517   // OpenCL global variables of sampler type are translated to function calls,
2518   // therefore no need to be translated.
2519   QualType ASTTy = D->getType();
2520   if (getLangOpts().OpenCL && ASTTy->isSamplerT())
2521     return;
2522
2523   llvm::Constant *Init = nullptr;
2524   CXXRecordDecl *RD = ASTTy->getBaseElementTypeUnsafe()->getAsCXXRecordDecl();
2525   bool NeedsGlobalCtor = false;
2526   bool NeedsGlobalDtor = RD && !RD->hasTrivialDestructor();
2527
2528   const VarDecl *InitDecl;
2529   const Expr *InitExpr = D->getAnyInitializer(InitDecl);
2530
2531   // CUDA E.2.4.1 "__shared__ variables cannot have an initialization
2532   // as part of their declaration."  Sema has already checked for
2533   // error cases, so we just need to set Init to UndefValue.
2534   if (getLangOpts().CUDA && getLangOpts().CUDAIsDevice &&
2535       D->hasAttr<CUDASharedAttr>())
2536     Init = llvm::UndefValue::get(getTypes().ConvertType(ASTTy));
2537   else if (!InitExpr) {
2538     // This is a tentative definition; tentative definitions are
2539     // implicitly initialized with { 0 }.
2540     //
2541     // Note that tentative definitions are only emitted at the end of
2542     // a translation unit, so they should never have incomplete
2543     // type. In addition, EmitTentativeDefinition makes sure that we
2544     // never attempt to emit a tentative definition if a real one
2545     // exists. A use may still exists, however, so we still may need
2546     // to do a RAUW.
2547     assert(!ASTTy->isIncompleteType() && "Unexpected incomplete type");
2548     Init = EmitNullConstant(D->getType());
2549   } else {
2550     initializedGlobalDecl = GlobalDecl(D);
2551     Init = EmitConstantInit(*InitDecl);
2552
2553     if (!Init) {
2554       QualType T = InitExpr->getType();
2555       if (D->getType()->isReferenceType())
2556         T = D->getType();
2557
2558       if (getLangOpts().CPlusPlus) {
2559         Init = EmitNullConstant(T);
2560         NeedsGlobalCtor = true;
2561       } else {
2562         ErrorUnsupported(D, "static initializer");
2563         Init = llvm::UndefValue::get(getTypes().ConvertType(T));
2564       }
2565     } else {
2566       // We don't need an initializer, so remove the entry for the delayed
2567       // initializer position (just in case this entry was delayed) if we
2568       // also don't need to register a destructor.
2569       if (getLangOpts().CPlusPlus && !NeedsGlobalDtor)
2570         DelayedCXXInitPosition.erase(D);
2571     }
2572   }
2573
2574   llvm::Type* InitType = Init->getType();
2575   llvm::Constant *Entry =
2576       GetAddrOfGlobalVar(D, InitType, ForDefinition_t(!IsTentative));
2577
2578   // Strip off a bitcast if we got one back.
2579   if (auto *CE = dyn_cast<llvm::ConstantExpr>(Entry)) {
2580     assert(CE->getOpcode() == llvm::Instruction::BitCast ||
2581            CE->getOpcode() == llvm::Instruction::AddrSpaceCast ||
2582            // All zero index gep.
2583            CE->getOpcode() == llvm::Instruction::GetElementPtr);
2584     Entry = CE->getOperand(0);
2585   }
2586
2587   // Entry is now either a Function or GlobalVariable.
2588   auto *GV = dyn_cast<llvm::GlobalVariable>(Entry);
2589
2590   // We have a definition after a declaration with the wrong type.
2591   // We must make a new GlobalVariable* and update everything that used OldGV
2592   // (a declaration or tentative definition) with the new GlobalVariable*
2593   // (which will be a definition).
2594   //
2595   // This happens if there is a prototype for a global (e.g.
2596   // "extern int x[];") and then a definition of a different type (e.g.
2597   // "int x[10];"). This also happens when an initializer has a different type
2598   // from the type of the global (this happens with unions).
2599   if (!GV ||
2600       GV->getType()->getElementType() != InitType ||
2601       GV->getType()->getAddressSpace() !=
2602        GetGlobalVarAddressSpace(D, getContext().getTargetAddressSpace(ASTTy))) {
2603
2604     // Move the old entry aside so that we'll create a new one.
2605     Entry->setName(StringRef());
2606
2607     // Make a new global with the correct type, this is now guaranteed to work.
2608     GV = cast<llvm::GlobalVariable>(
2609         GetAddrOfGlobalVar(D, InitType, ForDefinition_t(!IsTentative)));
2610
2611     // Replace all uses of the old global with the new global
2612     llvm::Constant *NewPtrForOldDecl =
2613         llvm::ConstantExpr::getBitCast(GV, Entry->getType());
2614     Entry->replaceAllUsesWith(NewPtrForOldDecl);
2615
2616     // Erase the old global, since it is no longer used.
2617     cast<llvm::GlobalValue>(Entry)->eraseFromParent();
2618   }
2619
2620   MaybeHandleStaticInExternC(D, GV);
2621
2622   if (D->hasAttr<AnnotateAttr>())
2623     AddGlobalAnnotations(D, GV);
2624
2625   // Set the llvm linkage type as appropriate.
2626   llvm::GlobalValue::LinkageTypes Linkage =
2627       getLLVMLinkageVarDefinition(D, GV->isConstant());
2628
2629   // CUDA B.2.1 "The __device__ qualifier declares a variable that resides on
2630   // the device. [...]"
2631   // CUDA B.2.2 "The __constant__ qualifier, optionally used together with
2632   // __device__, declares a variable that: [...]
2633   // Is accessible from all the threads within the grid and from the host
2634   // through the runtime library (cudaGetSymbolAddress() / cudaGetSymbolSize()
2635   // / cudaMemcpyToSymbol() / cudaMemcpyFromSymbol())."
2636   if (GV && LangOpts.CUDA) {
2637     if (LangOpts.CUDAIsDevice) {
2638       if (D->hasAttr<CUDADeviceAttr>() || D->hasAttr<CUDAConstantAttr>())
2639         GV->setExternallyInitialized(true);
2640     } else {
2641       // Host-side shadows of external declarations of device-side
2642       // global variables become internal definitions. These have to
2643       // be internal in order to prevent name conflicts with global
2644       // host variables with the same name in a different TUs.
2645       if (D->hasAttr<CUDADeviceAttr>() || D->hasAttr<CUDAConstantAttr>()) {
2646         Linkage = llvm::GlobalValue::InternalLinkage;
2647
2648         // Shadow variables and their properties must be registered
2649         // with CUDA runtime.
2650         unsigned Flags = 0;
2651         if (!D->hasDefinition())
2652           Flags |= CGCUDARuntime::ExternDeviceVar;
2653         if (D->hasAttr<CUDAConstantAttr>())
2654           Flags |= CGCUDARuntime::ConstantDeviceVar;
2655         getCUDARuntime().registerDeviceVar(*GV, Flags);
2656       } else if (D->hasAttr<CUDASharedAttr>())
2657         // __shared__ variables are odd. Shadows do get created, but
2658         // they are not registered with the CUDA runtime, so they
2659         // can't really be used to access their device-side
2660         // counterparts. It's not clear yet whether it's nvcc's bug or
2661         // a feature, but we've got to do the same for compatibility.
2662         Linkage = llvm::GlobalValue::InternalLinkage;
2663     }
2664   }
2665   GV->setInitializer(Init);
2666
2667   // If it is safe to mark the global 'constant', do so now.
2668   GV->setConstant(!NeedsGlobalCtor && !NeedsGlobalDtor &&
2669                   isTypeConstant(D->getType(), true));
2670
2671   // If it is in a read-only section, mark it 'constant'.
2672   if (const SectionAttr *SA = D->getAttr<SectionAttr>()) {
2673     const ASTContext::SectionInfo &SI = Context.SectionInfos[SA->getName()];
2674     if ((SI.SectionFlags & ASTContext::PSF_Write) == 0)
2675       GV->setConstant(true);
2676   }
2677
2678   GV->setAlignment(getContext().getDeclAlign(D).getQuantity());
2679
2680
2681   // On Darwin, if the normal linkage of a C++ thread_local variable is
2682   // LinkOnce or Weak, we keep the normal linkage to prevent multiple
2683   // copies within a linkage unit; otherwise, the backing variable has
2684   // internal linkage and all accesses should just be calls to the
2685   // Itanium-specified entry point, which has the normal linkage of the
2686   // variable. This is to preserve the ability to change the implementation
2687   // behind the scenes.
2688   if (!D->isStaticLocal() && D->getTLSKind() == VarDecl::TLS_Dynamic &&
2689       Context.getTargetInfo().getTriple().isOSDarwin() &&
2690       !llvm::GlobalVariable::isLinkOnceLinkage(Linkage) &&
2691       !llvm::GlobalVariable::isWeakLinkage(Linkage))
2692     Linkage = llvm::GlobalValue::InternalLinkage;
2693
2694   GV->setLinkage(Linkage);
2695   if (D->hasAttr<DLLImportAttr>())
2696     GV->setDLLStorageClass(llvm::GlobalVariable::DLLImportStorageClass);
2697   else if (D->hasAttr<DLLExportAttr>())
2698     GV->setDLLStorageClass(llvm::GlobalVariable::DLLExportStorageClass);
2699   else
2700     GV->setDLLStorageClass(llvm::GlobalVariable::DefaultStorageClass);
2701
2702   if (Linkage == llvm::GlobalVariable::CommonLinkage) {
2703     // common vars aren't constant even if declared const.
2704     GV->setConstant(false);
2705     // Tentative definition of global variables may be initialized with
2706     // non-zero null pointers. In this case they should have weak linkage
2707     // since common linkage must have zero initializer and must not have
2708     // explicit section therefore cannot have non-zero initial value.
2709     if (!GV->getInitializer()->isNullValue())
2710       GV->setLinkage(llvm::GlobalVariable::WeakAnyLinkage);
2711   }
2712
2713   setNonAliasAttributes(D, GV);
2714
2715   if (D->getTLSKind() && !GV->isThreadLocal()) {
2716     if (D->getTLSKind() == VarDecl::TLS_Dynamic)
2717       CXXThreadLocals.push_back(D);
2718     setTLSMode(GV, *D);
2719   }
2720
2721   maybeSetTrivialComdat(*D, *GV);
2722
2723   // Emit the initializer function if necessary.
2724   if (NeedsGlobalCtor || NeedsGlobalDtor)
2725     EmitCXXGlobalVarDeclInitFunc(D, GV, NeedsGlobalCtor);
2726
2727   SanitizerMD->reportGlobalToASan(GV, *D, NeedsGlobalCtor);
2728
2729   // Emit global variable debug information.
2730   if (CGDebugInfo *DI = getModuleDebugInfo())
2731     if (getCodeGenOpts().getDebugInfo() >= codegenoptions::LimitedDebugInfo)
2732       DI->EmitGlobalVariable(GV, D);
2733 }
2734
2735 static bool isVarDeclStrongDefinition(const ASTContext &Context,
2736                                       CodeGenModule &CGM, const VarDecl *D,
2737                                       bool NoCommon) {
2738   // Don't give variables common linkage if -fno-common was specified unless it
2739   // was overridden by a NoCommon attribute.
2740   if ((NoCommon || D->hasAttr<NoCommonAttr>()) && !D->hasAttr<CommonAttr>())
2741     return true;
2742
2743   // C11 6.9.2/2:
2744   //   A declaration of an identifier for an object that has file scope without
2745   //   an initializer, and without a storage-class specifier or with the
2746   //   storage-class specifier static, constitutes a tentative definition.
2747   if (D->getInit() || D->hasExternalStorage())
2748     return true;
2749
2750   // A variable cannot be both common and exist in a section.
2751   if (D->hasAttr<SectionAttr>())
2752     return true;
2753
2754   // Thread local vars aren't considered common linkage.
2755   if (D->getTLSKind())
2756     return true;
2757
2758   // Tentative definitions marked with WeakImportAttr are true definitions.
2759   if (D->hasAttr<WeakImportAttr>())
2760     return true;
2761
2762   // A variable cannot be both common and exist in a comdat.
2763   if (shouldBeInCOMDAT(CGM, *D))
2764     return true;
2765
2766   // Declarations with a required alignment do not have common linkage in MSVC
2767   // mode.
2768   if (Context.getTargetInfo().getCXXABI().isMicrosoft()) {
2769     if (D->hasAttr<AlignedAttr>())
2770       return true;
2771     QualType VarType = D->getType();
2772     if (Context.isAlignmentRequired(VarType))
2773       return true;
2774
2775     if (const auto *RT = VarType->getAs<RecordType>()) {
2776       const RecordDecl *RD = RT->getDecl();
2777       for (const FieldDecl *FD : RD->fields()) {
2778         if (FD->isBitField())
2779           continue;
2780         if (FD->hasAttr<AlignedAttr>())
2781           return true;
2782         if (Context.isAlignmentRequired(FD->getType()))
2783           return true;
2784       }
2785     }
2786   }
2787
2788   return false;
2789 }
2790
2791 llvm::GlobalValue::LinkageTypes CodeGenModule::getLLVMLinkageForDeclarator(
2792     const DeclaratorDecl *D, GVALinkage Linkage, bool IsConstantVariable) {
2793   if (Linkage == GVA_Internal)
2794     return llvm::Function::InternalLinkage;
2795
2796   if (D->hasAttr<WeakAttr>()) {
2797     if (IsConstantVariable)
2798       return llvm::GlobalVariable::WeakODRLinkage;
2799     else
2800       return llvm::GlobalVariable::WeakAnyLinkage;
2801   }
2802
2803   // We are guaranteed to have a strong definition somewhere else,
2804   // so we can use available_externally linkage.
2805   if (Linkage == GVA_AvailableExternally)
2806     return llvm::Function::AvailableExternallyLinkage;
2807
2808   // Note that Apple's kernel linker doesn't support symbol
2809   // coalescing, so we need to avoid linkonce and weak linkages there.
2810   // Normally, this means we just map to internal, but for explicit
2811   // instantiations we'll map to external.
2812
2813   // In C++, the compiler has to emit a definition in every translation unit
2814   // that references the function.  We should use linkonce_odr because
2815   // a) if all references in this translation unit are optimized away, we
2816   // don't need to codegen it.  b) if the function persists, it needs to be
2817   // merged with other definitions. c) C++ has the ODR, so we know the
2818   // definition is dependable.
2819   if (Linkage == GVA_DiscardableODR)
2820     return !Context.getLangOpts().AppleKext ? llvm::Function::LinkOnceODRLinkage
2821                                             : llvm::Function::InternalLinkage;
2822
2823   // An explicit instantiation of a template has weak linkage, since
2824   // explicit instantiations can occur in multiple translation units
2825   // and must all be equivalent. However, we are not allowed to
2826   // throw away these explicit instantiations.
2827   //
2828   // We don't currently support CUDA device code spread out across multiple TUs,
2829   // so say that CUDA templates are either external (for kernels) or internal.
2830   // This lets llvm perform aggressive inter-procedural optimizations.
2831   if (Linkage == GVA_StrongODR) {
2832     if (Context.getLangOpts().AppleKext)
2833       return llvm::Function::ExternalLinkage;
2834     if (Context.getLangOpts().CUDA && Context.getLangOpts().CUDAIsDevice)
2835       return D->hasAttr<CUDAGlobalAttr>() ? llvm::Function::ExternalLinkage
2836                                           : llvm::Function::InternalLinkage;
2837     return llvm::Function::WeakODRLinkage;
2838   }
2839
2840   // C++ doesn't have tentative definitions and thus cannot have common
2841   // linkage.
2842   if (!getLangOpts().CPlusPlus && isa<VarDecl>(D) &&
2843       !isVarDeclStrongDefinition(Context, *this, cast<VarDecl>(D),
2844                                  CodeGenOpts.NoCommon))
2845     return llvm::GlobalVariable::CommonLinkage;
2846
2847   // selectany symbols are externally visible, so use weak instead of
2848   // linkonce.  MSVC optimizes away references to const selectany globals, so
2849   // all definitions should be the same and ODR linkage should be used.
2850   // http://msdn.microsoft.com/en-us/library/5tkz6s71.aspx
2851   if (D->hasAttr<SelectAnyAttr>())
2852     return llvm::GlobalVariable::WeakODRLinkage;
2853
2854   // Otherwise, we have strong external linkage.
2855   assert(Linkage == GVA_StrongExternal);
2856   return llvm::GlobalVariable::ExternalLinkage;
2857 }
2858
2859 llvm::GlobalValue::LinkageTypes CodeGenModule::getLLVMLinkageVarDefinition(
2860     const VarDecl *VD, bool IsConstant) {
2861   GVALinkage Linkage = getContext().GetGVALinkageForVariable(VD);
2862   return getLLVMLinkageForDeclarator(VD, Linkage, IsConstant);
2863 }
2864
2865 /// Replace the uses of a function that was declared with a non-proto type.
2866 /// We want to silently drop extra arguments from call sites
2867 static void replaceUsesOfNonProtoConstant(llvm::Constant *old,
2868                                           llvm::Function *newFn) {
2869   // Fast path.
2870   if (old->use_empty()) return;
2871
2872   llvm::Type *newRetTy = newFn->getReturnType();
2873   SmallVector<llvm::Value*, 4> newArgs;
2874   SmallVector<llvm::OperandBundleDef, 1> newBundles;
2875
2876   for (llvm::Value::use_iterator ui = old->use_begin(), ue = old->use_end();
2877          ui != ue; ) {
2878     llvm::Value::use_iterator use = ui++; // Increment before the use is erased.
2879     llvm::User *user = use->getUser();
2880
2881     // Recognize and replace uses of bitcasts.  Most calls to
2882     // unprototyped functions will use bitcasts.
2883     if (auto *bitcast = dyn_cast<llvm::ConstantExpr>(user)) {
2884       if (bitcast->getOpcode() == llvm::Instruction::BitCast)
2885         replaceUsesOfNonProtoConstant(bitcast, newFn);
2886       continue;
2887     }
2888
2889     // Recognize calls to the function.
2890     llvm::CallSite callSite(user);
2891     if (!callSite) continue;
2892     if (!callSite.isCallee(&*use)) continue;
2893
2894     // If the return types don't match exactly, then we can't
2895     // transform this call unless it's dead.
2896     if (callSite->getType() != newRetTy && !callSite->use_empty())
2897       continue;
2898
2899     // Get the call site's attribute list.
2900     SmallVector<llvm::AttributeSet, 8> newAttrs;
2901     llvm::AttributeSet oldAttrs = callSite.getAttributes();
2902
2903     // Collect any return attributes from the call.
2904     if (oldAttrs.hasAttributes(llvm::AttributeSet::ReturnIndex))
2905       newAttrs.push_back(
2906         llvm::AttributeSet::get(newFn->getContext(),
2907                                 oldAttrs.getRetAttributes()));
2908
2909     // If the function was passed too few arguments, don't transform.
2910     unsigned newNumArgs = newFn->arg_size();
2911     if (callSite.arg_size() < newNumArgs) continue;
2912
2913     // If extra arguments were passed, we silently drop them.
2914     // If any of the types mismatch, we don't transform.
2915     unsigned argNo = 0;
2916     bool dontTransform = false;
2917     for (llvm::Function::arg_iterator ai = newFn->arg_begin(),
2918            ae = newFn->arg_end(); ai != ae; ++ai, ++argNo) {
2919       if (callSite.getArgument(argNo)->getType() != ai->getType()) {
2920         dontTransform = true;
2921         break;
2922       }
2923
2924       // Add any parameter attributes.
2925       if (oldAttrs.hasAttributes(argNo + 1))
2926         newAttrs.
2927           push_back(llvm::
2928                     AttributeSet::get(newFn->getContext(),
2929                                       oldAttrs.getParamAttributes(argNo + 1)));
2930     }
2931     if (dontTransform)
2932       continue;
2933
2934     if (oldAttrs.hasAttributes(llvm::AttributeSet::FunctionIndex))
2935       newAttrs.push_back(llvm::AttributeSet::get(newFn->getContext(),
2936                                                  oldAttrs.getFnAttributes()));
2937
2938     // Okay, we can transform this.  Create the new call instruction and copy
2939     // over the required information.
2940     newArgs.append(callSite.arg_begin(), callSite.arg_begin() + argNo);
2941
2942     // Copy over any operand bundles.
2943     callSite.getOperandBundlesAsDefs(newBundles);
2944
2945     llvm::CallSite newCall;
2946     if (callSite.isCall()) {
2947       newCall = llvm::CallInst::Create(newFn, newArgs, newBundles, "",
2948                                        callSite.getInstruction());
2949     } else {
2950       auto *oldInvoke = cast<llvm::InvokeInst>(callSite.getInstruction());
2951       newCall = llvm::InvokeInst::Create(newFn,
2952                                          oldInvoke->getNormalDest(),
2953                                          oldInvoke->getUnwindDest(),
2954                                          newArgs, newBundles, "",
2955                                          callSite.getInstruction());
2956     }
2957     newArgs.clear(); // for the next iteration
2958
2959     if (!newCall->getType()->isVoidTy())
2960       newCall->takeName(callSite.getInstruction());
2961     newCall.setAttributes(
2962                      llvm::AttributeSet::get(newFn->getContext(), newAttrs));
2963     newCall.setCallingConv(callSite.getCallingConv());
2964
2965     // Finally, remove the old call, replacing any uses with the new one.
2966     if (!callSite->use_empty())
2967       callSite->replaceAllUsesWith(newCall.getInstruction());
2968
2969     // Copy debug location attached to CI.
2970     if (callSite->getDebugLoc())
2971       newCall->setDebugLoc(callSite->getDebugLoc());
2972
2973     callSite->eraseFromParent();
2974   }
2975 }
2976
2977 /// ReplaceUsesOfNonProtoTypeWithRealFunction - This function is called when we
2978 /// implement a function with no prototype, e.g. "int foo() {}".  If there are
2979 /// existing call uses of the old function in the module, this adjusts them to
2980 /// call the new function directly.
2981 ///
2982 /// This is not just a cleanup: the always_inline pass requires direct calls to
2983 /// functions to be able to inline them.  If there is a bitcast in the way, it
2984 /// won't inline them.  Instcombine normally deletes these calls, but it isn't
2985 /// run at -O0.
2986 static void ReplaceUsesOfNonProtoTypeWithRealFunction(llvm::GlobalValue *Old,
2987                                                       llvm::Function *NewFn) {
2988   // If we're redefining a global as a function, don't transform it.
2989   if (!isa<llvm::Function>(Old)) return;
2990
2991   replaceUsesOfNonProtoConstant(Old, NewFn);
2992 }
2993
2994 void CodeGenModule::HandleCXXStaticMemberVarInstantiation(VarDecl *VD) {
2995   auto DK = VD->isThisDeclarationADefinition();
2996   if (DK == VarDecl::Definition && VD->hasAttr<DLLImportAttr>())
2997     return;
2998
2999   TemplateSpecializationKind TSK = VD->getTemplateSpecializationKind();
3000   // If we have a definition, this might be a deferred decl. If the
3001   // instantiation is explicit, make sure we emit it at the end.
3002   if (VD->getDefinition() && TSK == TSK_ExplicitInstantiationDefinition)
3003     GetAddrOfGlobalVar(VD);
3004
3005   EmitTopLevelDecl(VD);
3006 }
3007
3008 void CodeGenModule::EmitGlobalFunctionDefinition(GlobalDecl GD,
3009                                                  llvm::GlobalValue *GV) {
3010   const auto *D = cast<FunctionDecl>(GD.getDecl());
3011
3012   // Compute the function info and LLVM type.
3013   const CGFunctionInfo &FI = getTypes().arrangeGlobalDeclaration(GD);
3014   llvm::FunctionType *Ty = getTypes().GetFunctionType(FI);
3015
3016   // Get or create the prototype for the function.
3017   if (!GV || (GV->getType()->getElementType() != Ty))
3018     GV = cast<llvm::GlobalValue>(GetAddrOfFunction(GD, Ty, /*ForVTable=*/false,
3019                                                    /*DontDefer=*/true,
3020                                                    ForDefinition));
3021
3022   // Already emitted.
3023   if (!GV->isDeclaration())
3024     return;
3025
3026   // We need to set linkage and visibility on the function before
3027   // generating code for it because various parts of IR generation
3028   // want to propagate this information down (e.g. to local static
3029   // declarations).
3030   auto *Fn = cast<llvm::Function>(GV);
3031   setFunctionLinkage(GD, Fn);
3032   setFunctionDLLStorageClass(GD, Fn);
3033
3034   // FIXME: this is redundant with part of setFunctionDefinitionAttributes
3035   setGlobalVisibility(Fn, D);
3036
3037   MaybeHandleStaticInExternC(D, Fn);
3038
3039   maybeSetTrivialComdat(*D, *Fn);
3040
3041   CodeGenFunction(*this).GenerateCode(D, Fn, FI);
3042
3043   setFunctionDefinitionAttributes(D, Fn);
3044   SetLLVMFunctionAttributesForDefinition(D, Fn);
3045
3046   if (const ConstructorAttr *CA = D->getAttr<ConstructorAttr>())
3047     AddGlobalCtor(Fn, CA->getPriority());
3048   if (const DestructorAttr *DA = D->getAttr<DestructorAttr>())
3049     AddGlobalDtor(Fn, DA->getPriority());
3050   if (D->hasAttr<AnnotateAttr>())
3051     AddGlobalAnnotations(D, Fn);
3052 }
3053
3054 void CodeGenModule::EmitAliasDefinition(GlobalDecl GD) {
3055   const auto *D = cast<ValueDecl>(GD.getDecl());
3056   const AliasAttr *AA = D->getAttr<AliasAttr>();
3057   assert(AA && "Not an alias?");
3058
3059   StringRef MangledName = getMangledName(GD);
3060
3061   if (AA->getAliasee() == MangledName) {
3062     Diags.Report(AA->getLocation(), diag::err_cyclic_alias) << 0;
3063     return;
3064   }
3065
3066   // If there is a definition in the module, then it wins over the alias.
3067   // This is dubious, but allow it to be safe.  Just ignore the alias.
3068   llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
3069   if (Entry && !Entry->isDeclaration())
3070     return;
3071
3072   Aliases.push_back(GD);
3073
3074   llvm::Type *DeclTy = getTypes().ConvertTypeForMem(D->getType());
3075
3076   // Create a reference to the named value.  This ensures that it is emitted
3077   // if a deferred decl.
3078   llvm::Constant *Aliasee;
3079   if (isa<llvm::FunctionType>(DeclTy))
3080     Aliasee = GetOrCreateLLVMFunction(AA->getAliasee(), DeclTy, GD,
3081                                       /*ForVTable=*/false);
3082   else
3083     Aliasee = GetOrCreateLLVMGlobal(AA->getAliasee(),
3084                                     llvm::PointerType::getUnqual(DeclTy),
3085                                     /*D=*/nullptr);
3086
3087   // Create the new alias itself, but don't set a name yet.
3088   auto *GA = llvm::GlobalAlias::create(
3089       DeclTy, 0, llvm::Function::ExternalLinkage, "", Aliasee, &getModule());
3090
3091   if (Entry) {
3092     if (GA->getAliasee() == Entry) {
3093       Diags.Report(AA->getLocation(), diag::err_cyclic_alias) << 0;
3094       return;
3095     }
3096
3097     assert(Entry->isDeclaration());
3098
3099     // If there is a declaration in the module, then we had an extern followed
3100     // by the alias, as in:
3101     //   extern int test6();
3102     //   ...
3103     //   int test6() __attribute__((alias("test7")));
3104     //
3105     // Remove it and replace uses of it with the alias.
3106     GA->takeName(Entry);
3107
3108     Entry->replaceAllUsesWith(llvm::ConstantExpr::getBitCast(GA,
3109                                                           Entry->getType()));
3110     Entry->eraseFromParent();
3111   } else {
3112     GA->setName(MangledName);
3113   }
3114
3115   // Set attributes which are particular to an alias; this is a
3116   // specialization of the attributes which may be set on a global
3117   // variable/function.
3118   if (D->hasAttr<WeakAttr>() || D->hasAttr<WeakRefAttr>() ||
3119       D->isWeakImported()) {
3120     GA->setLinkage(llvm::Function::WeakAnyLinkage);
3121   }
3122
3123   if (const auto *VD = dyn_cast<VarDecl>(D))
3124     if (VD->getTLSKind())
3125       setTLSMode(GA, *VD);
3126
3127   setAliasAttributes(D, GA);
3128 }
3129
3130 void CodeGenModule::emitIFuncDefinition(GlobalDecl GD) {
3131   const auto *D = cast<ValueDecl>(GD.getDecl());
3132   const IFuncAttr *IFA = D->getAttr<IFuncAttr>();
3133   assert(IFA && "Not an ifunc?");
3134
3135   StringRef MangledName = getMangledName(GD);
3136
3137   if (IFA->getResolver() == MangledName) {
3138     Diags.Report(IFA->getLocation(), diag::err_cyclic_alias) << 1;
3139     return;
3140   }
3141
3142   // Report an error if some definition overrides ifunc.
3143   llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
3144   if (Entry && !Entry->isDeclaration()) {
3145     GlobalDecl OtherGD;
3146     if (lookupRepresentativeDecl(MangledName, OtherGD) &&
3147         DiagnosedConflictingDefinitions.insert(GD).second) {
3148       Diags.Report(D->getLocation(), diag::err_duplicate_mangled_name);
3149       Diags.Report(OtherGD.getDecl()->getLocation(),
3150                    diag::note_previous_definition);
3151     }
3152     return;
3153   }
3154
3155   Aliases.push_back(GD);
3156
3157   llvm::Type *DeclTy = getTypes().ConvertTypeForMem(D->getType());
3158   llvm::Constant *Resolver =
3159       GetOrCreateLLVMFunction(IFA->getResolver(), DeclTy, GD,
3160                               /*ForVTable=*/false);
3161   llvm::GlobalIFunc *GIF =
3162       llvm::GlobalIFunc::create(DeclTy, 0, llvm::Function::ExternalLinkage,
3163                                 "", Resolver, &getModule());
3164   if (Entry) {
3165     if (GIF->getResolver() == Entry) {
3166       Diags.Report(IFA->getLocation(), diag::err_cyclic_alias) << 1;
3167       return;
3168     }
3169     assert(Entry->isDeclaration());
3170
3171     // If there is a declaration in the module, then we had an extern followed
3172     // by the ifunc, as in:
3173     //   extern int test();
3174     //   ...
3175     //   int test() __attribute__((ifunc("resolver")));
3176     //
3177     // Remove it and replace uses of it with the ifunc.
3178     GIF->takeName(Entry);
3179
3180     Entry->replaceAllUsesWith(llvm::ConstantExpr::getBitCast(GIF,
3181                                                           Entry->getType()));
3182     Entry->eraseFromParent();
3183   } else
3184     GIF->setName(MangledName);
3185
3186   SetCommonAttributes(D, GIF);
3187 }
3188
3189 llvm::Function *CodeGenModule::getIntrinsic(unsigned IID,
3190                                             ArrayRef<llvm::Type*> Tys) {
3191   return llvm::Intrinsic::getDeclaration(&getModule(), (llvm::Intrinsic::ID)IID,
3192                                          Tys);
3193 }
3194
3195 static llvm::StringMapEntry<llvm::GlobalVariable *> &
3196 GetConstantCFStringEntry(llvm::StringMap<llvm::GlobalVariable *> &Map,
3197                          const StringLiteral *Literal, bool TargetIsLSB,
3198                          bool &IsUTF16, unsigned &StringLength) {
3199   StringRef String = Literal->getString();
3200   unsigned NumBytes = String.size();
3201
3202   // Check for simple case.
3203   if (!Literal->containsNonAsciiOrNull()) {
3204     StringLength = NumBytes;
3205     return *Map.insert(std::make_pair(String, nullptr)).first;
3206   }
3207
3208   // Otherwise, convert the UTF8 literals into a string of shorts.
3209   IsUTF16 = true;
3210
3211   SmallVector<llvm::UTF16, 128> ToBuf(NumBytes + 1); // +1 for ending nulls.
3212   const llvm::UTF8 *FromPtr = (const llvm::UTF8 *)String.data();
3213   llvm::UTF16 *ToPtr = &ToBuf[0];
3214
3215   (void)llvm::ConvertUTF8toUTF16(&FromPtr, FromPtr + NumBytes, &ToPtr,
3216                                  ToPtr + NumBytes, llvm::strictConversion);
3217
3218   // ConvertUTF8toUTF16 returns the length in ToPtr.
3219   StringLength = ToPtr - &ToBuf[0];
3220
3221   // Add an explicit null.
3222   *ToPtr = 0;
3223   return *Map.insert(std::make_pair(
3224                          StringRef(reinterpret_cast<const char *>(ToBuf.data()),
3225                                    (StringLength + 1) * 2),
3226                          nullptr)).first;
3227 }
3228
3229 ConstantAddress
3230 CodeGenModule::GetAddrOfConstantCFString(const StringLiteral *Literal) {
3231   unsigned StringLength = 0;
3232   bool isUTF16 = false;
3233   llvm::StringMapEntry<llvm::GlobalVariable *> &Entry =
3234       GetConstantCFStringEntry(CFConstantStringMap, Literal,
3235                                getDataLayout().isLittleEndian(), isUTF16,
3236                                StringLength);
3237
3238   if (auto *C = Entry.second)
3239     return ConstantAddress(C, CharUnits::fromQuantity(C->getAlignment()));
3240
3241   llvm::Constant *Zero = llvm::Constant::getNullValue(Int32Ty);
3242   llvm::Constant *Zeros[] = { Zero, Zero };
3243
3244   // If we don't already have it, get __CFConstantStringClassReference.
3245   if (!CFConstantStringClassRef) {
3246     llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
3247     Ty = llvm::ArrayType::get(Ty, 0);
3248     llvm::Constant *GV =
3249         CreateRuntimeVariable(Ty, "__CFConstantStringClassReference");
3250
3251     if (getTriple().isOSBinFormatCOFF()) {
3252       IdentifierInfo &II = getContext().Idents.get(GV->getName());
3253       TranslationUnitDecl *TUDecl = getContext().getTranslationUnitDecl();
3254       DeclContext *DC = TranslationUnitDecl::castToDeclContext(TUDecl);
3255       llvm::GlobalValue *CGV = cast<llvm::GlobalValue>(GV);
3256
3257       const VarDecl *VD = nullptr;
3258       for (const auto &Result : DC->lookup(&II))
3259         if ((VD = dyn_cast<VarDecl>(Result)))
3260           break;
3261
3262       if (!VD || !VD->hasAttr<DLLExportAttr>()) {
3263         CGV->setDLLStorageClass(llvm::GlobalValue::DLLImportStorageClass);
3264         CGV->setLinkage(llvm::GlobalValue::ExternalLinkage);
3265       } else {
3266         CGV->setDLLStorageClass(llvm::GlobalValue::DLLExportStorageClass);
3267         CGV->setLinkage(llvm::GlobalValue::ExternalLinkage);
3268       }
3269     }
3270
3271     // Decay array -> ptr
3272     CFConstantStringClassRef =
3273         llvm::ConstantExpr::getGetElementPtr(Ty, GV, Zeros);
3274   }
3275
3276   QualType CFTy = getContext().getCFConstantStringType();
3277
3278   auto *STy = cast<llvm::StructType>(getTypes().ConvertType(CFTy));
3279
3280   ConstantInitBuilder Builder(*this);
3281   auto Fields = Builder.beginStruct(STy);
3282
3283   // Class pointer.
3284   Fields.add(cast<llvm::ConstantExpr>(CFConstantStringClassRef));
3285
3286   // Flags.
3287   Fields.addInt(IntTy, isUTF16 ? 0x07d0 : 0x07C8);
3288
3289   // String pointer.
3290   llvm::Constant *C = nullptr;
3291   if (isUTF16) {
3292     auto Arr = llvm::makeArrayRef(
3293         reinterpret_cast<uint16_t *>(const_cast<char *>(Entry.first().data())),
3294         Entry.first().size() / 2);
3295     C = llvm::ConstantDataArray::get(VMContext, Arr);
3296   } else {
3297     C = llvm::ConstantDataArray::getString(VMContext, Entry.first());
3298   }
3299
3300   // Note: -fwritable-strings doesn't make the backing store strings of
3301   // CFStrings writable. (See <rdar://problem/10657500>)
3302   auto *GV =
3303       new llvm::GlobalVariable(getModule(), C->getType(), /*isConstant=*/true,
3304                                llvm::GlobalValue::PrivateLinkage, C, ".str");
3305   GV->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
3306   // Don't enforce the target's minimum global alignment, since the only use
3307   // of the string is via this class initializer.
3308   CharUnits Align = isUTF16
3309                         ? getContext().getTypeAlignInChars(getContext().ShortTy)
3310                         : getContext().getTypeAlignInChars(getContext().CharTy);
3311   GV->setAlignment(Align.getQuantity());
3312
3313   // FIXME: We set the section explicitly to avoid a bug in ld64 224.1.
3314   // Without it LLVM can merge the string with a non unnamed_addr one during
3315   // LTO.  Doing that changes the section it ends in, which surprises ld64.
3316   if (getTriple().isOSBinFormatMachO())
3317     GV->setSection(isUTF16 ? "__TEXT,__ustring"
3318                            : "__TEXT,__cstring,cstring_literals");
3319
3320   // String.
3321   llvm::Constant *Str =
3322       llvm::ConstantExpr::getGetElementPtr(GV->getValueType(), GV, Zeros);
3323
3324   if (isUTF16)
3325     // Cast the UTF16 string to the correct type.
3326     Str = llvm::ConstantExpr::getBitCast(Str, Int8PtrTy);
3327   Fields.add(Str);
3328
3329   // String length.
3330   auto Ty = getTypes().ConvertType(getContext().LongTy);
3331   Fields.addInt(cast<llvm::IntegerType>(Ty), StringLength);
3332
3333   CharUnits Alignment = getPointerAlign();
3334
3335   // The struct.
3336   GV = Fields.finishAndCreateGlobal("_unnamed_cfstring_", Alignment,
3337                                     /*isConstant=*/false,
3338                                     llvm::GlobalVariable::PrivateLinkage);
3339   switch (getTriple().getObjectFormat()) {
3340   case llvm::Triple::UnknownObjectFormat:
3341     llvm_unreachable("unknown file format");
3342   case llvm::Triple::COFF:
3343   case llvm::Triple::ELF:
3344     GV->setSection("cfstring");
3345     break;
3346   case llvm::Triple::MachO:
3347     GV->setSection("__DATA,__cfstring");
3348     break;
3349   }
3350   Entry.second = GV;
3351
3352   return ConstantAddress(GV, Alignment);
3353 }
3354
3355 QualType CodeGenModule::getObjCFastEnumerationStateType() {
3356   if (ObjCFastEnumerationStateType.isNull()) {
3357     RecordDecl *D = Context.buildImplicitRecord("__objcFastEnumerationState");
3358     D->startDefinition();
3359
3360     QualType FieldTypes[] = {
3361       Context.UnsignedLongTy,
3362       Context.getPointerType(Context.getObjCIdType()),
3363       Context.getPointerType(Context.UnsignedLongTy),
3364       Context.getConstantArrayType(Context.UnsignedLongTy,
3365                            llvm::APInt(32, 5), ArrayType::Normal, 0)
3366     };
3367
3368     for (size_t i = 0; i < 4; ++i) {
3369       FieldDecl *Field = FieldDecl::Create(Context,
3370                                            D,
3371                                            SourceLocation(),
3372                                            SourceLocation(), nullptr,
3373                                            FieldTypes[i], /*TInfo=*/nullptr,
3374                                            /*BitWidth=*/nullptr,
3375                                            /*Mutable=*/false,
3376                                            ICIS_NoInit);
3377       Field->setAccess(AS_public);
3378       D->addDecl(Field);
3379     }
3380
3381     D->completeDefinition();
3382     ObjCFastEnumerationStateType = Context.getTagDeclType(D);
3383   }
3384
3385   return ObjCFastEnumerationStateType;
3386 }
3387
3388 llvm::Constant *
3389 CodeGenModule::GetConstantArrayFromStringLiteral(const StringLiteral *E) {
3390   assert(!E->getType()->isPointerType() && "Strings are always arrays");
3391
3392   // Don't emit it as the address of the string, emit the string data itself
3393   // as an inline array.
3394   if (E->getCharByteWidth() == 1) {
3395     SmallString<64> Str(E->getString());
3396
3397     // Resize the string to the right size, which is indicated by its type.
3398     const ConstantArrayType *CAT = Context.getAsConstantArrayType(E->getType());
3399     Str.resize(CAT->getSize().getZExtValue());
3400     return llvm::ConstantDataArray::getString(VMContext, Str, false);
3401   }
3402
3403   auto *AType = cast<llvm::ArrayType>(getTypes().ConvertType(E->getType()));
3404   llvm::Type *ElemTy = AType->getElementType();
3405   unsigned NumElements = AType->getNumElements();
3406
3407   // Wide strings have either 2-byte or 4-byte elements.
3408   if (ElemTy->getPrimitiveSizeInBits() == 16) {
3409     SmallVector<uint16_t, 32> Elements;
3410     Elements.reserve(NumElements);
3411
3412     for(unsigned i = 0, e = E->getLength(); i != e; ++i)
3413       Elements.push_back(E->getCodeUnit(i));
3414     Elements.resize(NumElements);
3415     return llvm::ConstantDataArray::get(VMContext, Elements);
3416   }
3417
3418   assert(ElemTy->getPrimitiveSizeInBits() == 32);
3419   SmallVector<uint32_t, 32> Elements;
3420   Elements.reserve(NumElements);
3421
3422   for(unsigned i = 0, e = E->getLength(); i != e; ++i)
3423     Elements.push_back(E->getCodeUnit(i));
3424   Elements.resize(NumElements);
3425   return llvm::ConstantDataArray::get(VMContext, Elements);
3426 }
3427
3428 static llvm::GlobalVariable *
3429 GenerateStringLiteral(llvm::Constant *C, llvm::GlobalValue::LinkageTypes LT,
3430                       CodeGenModule &CGM, StringRef GlobalName,
3431                       CharUnits Alignment) {
3432   // OpenCL v1.2 s6.5.3: a string literal is in the constant address space.
3433   unsigned AddrSpace = 0;
3434   if (CGM.getLangOpts().OpenCL)
3435     AddrSpace = CGM.getContext().getTargetAddressSpace(LangAS::opencl_constant);
3436
3437   llvm::Module &M = CGM.getModule();
3438   // Create a global variable for this string
3439   auto *GV = new llvm::GlobalVariable(
3440       M, C->getType(), !CGM.getLangOpts().WritableStrings, LT, C, GlobalName,
3441       nullptr, llvm::GlobalVariable::NotThreadLocal, AddrSpace);
3442   GV->setAlignment(Alignment.getQuantity());
3443   GV->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
3444   if (GV->isWeakForLinker()) {
3445     assert(CGM.supportsCOMDAT() && "Only COFF uses weak string literals");
3446     GV->setComdat(M.getOrInsertComdat(GV->getName()));
3447   }
3448
3449   return GV;
3450 }
3451
3452 /// GetAddrOfConstantStringFromLiteral - Return a pointer to a
3453 /// constant array for the given string literal.
3454 ConstantAddress
3455 CodeGenModule::GetAddrOfConstantStringFromLiteral(const StringLiteral *S,
3456                                                   StringRef Name) {
3457   CharUnits Alignment = getContext().getAlignOfGlobalVarInChars(S->getType());
3458
3459   llvm::Constant *C = GetConstantArrayFromStringLiteral(S);
3460   llvm::GlobalVariable **Entry = nullptr;
3461   if (!LangOpts.WritableStrings) {
3462     Entry = &ConstantStringMap[C];
3463     if (auto GV = *Entry) {
3464       if (Alignment.getQuantity() > GV->getAlignment())
3465         GV->setAlignment(Alignment.getQuantity());
3466       return ConstantAddress(GV, Alignment);
3467     }
3468   }
3469
3470   SmallString<256> MangledNameBuffer;
3471   StringRef GlobalVariableName;
3472   llvm::GlobalValue::LinkageTypes LT;
3473
3474   // Mangle the string literal if the ABI allows for it.  However, we cannot
3475   // do this if  we are compiling with ASan or -fwritable-strings because they
3476   // rely on strings having normal linkage.
3477   if (!LangOpts.WritableStrings &&
3478       !LangOpts.Sanitize.has(SanitizerKind::Address) &&
3479       getCXXABI().getMangleContext().shouldMangleStringLiteral(S)) {
3480     llvm::raw_svector_ostream Out(MangledNameBuffer);
3481     getCXXABI().getMangleContext().mangleStringLiteral(S, Out);
3482
3483     LT = llvm::GlobalValue::LinkOnceODRLinkage;
3484     GlobalVariableName = MangledNameBuffer;
3485   } else {
3486     LT = llvm::GlobalValue::PrivateLinkage;
3487     GlobalVariableName = Name;
3488   }
3489
3490   auto GV = GenerateStringLiteral(C, LT, *this, GlobalVariableName, Alignment);
3491   if (Entry)
3492     *Entry = GV;
3493
3494   SanitizerMD->reportGlobalToASan(GV, S->getStrTokenLoc(0), "<string literal>",
3495                                   QualType());
3496   return ConstantAddress(GV, Alignment);
3497 }
3498
3499 /// GetAddrOfConstantStringFromObjCEncode - Return a pointer to a constant
3500 /// array for the given ObjCEncodeExpr node.
3501 ConstantAddress
3502 CodeGenModule::GetAddrOfConstantStringFromObjCEncode(const ObjCEncodeExpr *E) {
3503   std::string Str;
3504   getContext().getObjCEncodingForType(E->getEncodedType(), Str);
3505
3506   return GetAddrOfConstantCString(Str);
3507 }
3508
3509 /// GetAddrOfConstantCString - Returns a pointer to a character array containing
3510 /// the literal and a terminating '\0' character.
3511 /// The result has pointer to array type.
3512 ConstantAddress CodeGenModule::GetAddrOfConstantCString(
3513     const std::string &Str, const char *GlobalName) {
3514   StringRef StrWithNull(Str.c_str(), Str.size() + 1);
3515   CharUnits Alignment =
3516     getContext().getAlignOfGlobalVarInChars(getContext().CharTy);
3517
3518   llvm::Constant *C =
3519       llvm::ConstantDataArray::getString(getLLVMContext(), StrWithNull, false);
3520
3521   // Don't share any string literals if strings aren't constant.
3522   llvm::GlobalVariable **Entry = nullptr;
3523   if (!LangOpts.WritableStrings) {
3524     Entry = &ConstantStringMap[C];
3525     if (auto GV = *Entry) {
3526       if (Alignment.getQuantity() > GV->getAlignment())
3527         GV->setAlignment(Alignment.getQuantity());
3528       return ConstantAddress(GV, Alignment);
3529     }
3530   }
3531
3532   // Get the default prefix if a name wasn't specified.
3533   if (!GlobalName)
3534     GlobalName = ".str";
3535   // Create a global variable for this.
3536   auto GV = GenerateStringLiteral(C, llvm::GlobalValue::PrivateLinkage, *this,
3537                                   GlobalName, Alignment);
3538   if (Entry)
3539     *Entry = GV;
3540   return ConstantAddress(GV, Alignment);
3541 }
3542
3543 ConstantAddress CodeGenModule::GetAddrOfGlobalTemporary(
3544     const MaterializeTemporaryExpr *E, const Expr *Init) {
3545   assert((E->getStorageDuration() == SD_Static ||
3546           E->getStorageDuration() == SD_Thread) && "not a global temporary");
3547   const auto *VD = cast<VarDecl>(E->getExtendingDecl());
3548
3549   // If we're not materializing a subobject of the temporary, keep the
3550   // cv-qualifiers from the type of the MaterializeTemporaryExpr.
3551   QualType MaterializedType = Init->getType();
3552   if (Init == E->GetTemporaryExpr())
3553     MaterializedType = E->getType();
3554
3555   CharUnits Align = getContext().getTypeAlignInChars(MaterializedType);
3556
3557   if (llvm::Constant *Slot = MaterializedGlobalTemporaryMap[E])
3558     return ConstantAddress(Slot, Align);
3559
3560   // FIXME: If an externally-visible declaration extends multiple temporaries,
3561   // we need to give each temporary the same name in every translation unit (and
3562   // we also need to make the temporaries externally-visible).
3563   SmallString<256> Name;
3564   llvm::raw_svector_ostream Out(Name);
3565   getCXXABI().getMangleContext().mangleReferenceTemporary(
3566       VD, E->getManglingNumber(), Out);
3567
3568   APValue *Value = nullptr;
3569   if (E->getStorageDuration() == SD_Static) {
3570     // We might have a cached constant initializer for this temporary. Note
3571     // that this might have a different value from the value computed by
3572     // evaluating the initializer if the surrounding constant expression
3573     // modifies the temporary.
3574     Value = getContext().getMaterializedTemporaryValue(E, false);
3575     if (Value && Value->isUninit())
3576       Value = nullptr;
3577   }
3578
3579   // Try evaluating it now, it might have a constant initializer.
3580   Expr::EvalResult EvalResult;
3581   if (!Value && Init->EvaluateAsRValue(EvalResult, getContext()) &&
3582       !EvalResult.hasSideEffects())
3583     Value = &EvalResult.Val;
3584
3585   llvm::Constant *InitialValue = nullptr;
3586   bool Constant = false;
3587   llvm::Type *Type;
3588   if (Value) {
3589     // The temporary has a constant initializer, use it.
3590     InitialValue = EmitConstantValue(*Value, MaterializedType, nullptr);
3591     Constant = isTypeConstant(MaterializedType, /*ExcludeCtor*/Value);
3592     Type = InitialValue->getType();
3593   } else {
3594     // No initializer, the initialization will be provided when we
3595     // initialize the declaration which performed lifetime extension.
3596     Type = getTypes().ConvertTypeForMem(MaterializedType);
3597   }
3598
3599   // Create a global variable for this lifetime-extended temporary.
3600   llvm::GlobalValue::LinkageTypes Linkage =
3601       getLLVMLinkageVarDefinition(VD, Constant);
3602   if (Linkage == llvm::GlobalVariable::ExternalLinkage) {
3603     const VarDecl *InitVD;
3604     if (VD->isStaticDataMember() && VD->getAnyInitializer(InitVD) &&
3605         isa<CXXRecordDecl>(InitVD->getLexicalDeclContext())) {
3606       // Temporaries defined inside a class get linkonce_odr linkage because the
3607       // class can be defined in multipe translation units.
3608       Linkage = llvm::GlobalVariable::LinkOnceODRLinkage;
3609     } else {
3610       // There is no need for this temporary to have external linkage if the
3611       // VarDecl has external linkage.
3612       Linkage = llvm::GlobalVariable::InternalLinkage;
3613     }
3614   }
3615   unsigned AddrSpace = GetGlobalVarAddressSpace(
3616       VD, getContext().getTargetAddressSpace(MaterializedType));
3617   auto *GV = new llvm::GlobalVariable(
3618       getModule(), Type, Constant, Linkage, InitialValue, Name.c_str(),
3619       /*InsertBefore=*/nullptr, llvm::GlobalVariable::NotThreadLocal,
3620       AddrSpace);
3621   setGlobalVisibility(GV, VD);
3622   GV->setAlignment(Align.getQuantity());
3623   if (supportsCOMDAT() && GV->isWeakForLinker())
3624     GV->setComdat(TheModule.getOrInsertComdat(GV->getName()));
3625   if (VD->getTLSKind())
3626     setTLSMode(GV, *VD);
3627   MaterializedGlobalTemporaryMap[E] = GV;
3628   return ConstantAddress(GV, Align);
3629 }
3630
3631 /// EmitObjCPropertyImplementations - Emit information for synthesized
3632 /// properties for an implementation.
3633 void CodeGenModule::EmitObjCPropertyImplementations(const
3634                                                     ObjCImplementationDecl *D) {
3635   for (const auto *PID : D->property_impls()) {
3636     // Dynamic is just for type-checking.
3637     if (PID->getPropertyImplementation() == ObjCPropertyImplDecl::Synthesize) {
3638       ObjCPropertyDecl *PD = PID->getPropertyDecl();
3639
3640       // Determine which methods need to be implemented, some may have
3641       // been overridden. Note that ::isPropertyAccessor is not the method
3642       // we want, that just indicates if the decl came from a
3643       // property. What we want to know is if the method is defined in
3644       // this implementation.
3645       if (!D->getInstanceMethod(PD->getGetterName()))
3646         CodeGenFunction(*this).GenerateObjCGetter(
3647                                  const_cast<ObjCImplementationDecl *>(D), PID);
3648       if (!PD->isReadOnly() &&
3649           !D->getInstanceMethod(PD->getSetterName()))
3650         CodeGenFunction(*this).GenerateObjCSetter(
3651                                  const_cast<ObjCImplementationDecl *>(D), PID);
3652     }
3653   }
3654 }
3655
3656 static bool needsDestructMethod(ObjCImplementationDecl *impl) {
3657   const ObjCInterfaceDecl *iface = impl->getClassInterface();
3658   for (const ObjCIvarDecl *ivar = iface->all_declared_ivar_begin();
3659        ivar; ivar = ivar->getNextIvar())
3660     if (ivar->getType().isDestructedType())
3661       return true;
3662
3663   return false;
3664 }
3665
3666 static bool AllTrivialInitializers(CodeGenModule &CGM,
3667                                    ObjCImplementationDecl *D) {
3668   CodeGenFunction CGF(CGM);
3669   for (ObjCImplementationDecl::init_iterator B = D->init_begin(),
3670        E = D->init_end(); B != E; ++B) {
3671     CXXCtorInitializer *CtorInitExp = *B;
3672     Expr *Init = CtorInitExp->getInit();
3673     if (!CGF.isTrivialInitializer(Init))
3674       return false;
3675   }
3676   return true;
3677 }
3678
3679 /// EmitObjCIvarInitializations - Emit information for ivar initialization
3680 /// for an implementation.
3681 void CodeGenModule::EmitObjCIvarInitializations(ObjCImplementationDecl *D) {
3682   // We might need a .cxx_destruct even if we don't have any ivar initializers.
3683   if (needsDestructMethod(D)) {
3684     IdentifierInfo *II = &getContext().Idents.get(".cxx_destruct");
3685     Selector cxxSelector = getContext().Selectors.getSelector(0, &II);
3686     ObjCMethodDecl *DTORMethod =
3687       ObjCMethodDecl::Create(getContext(), D->getLocation(), D->getLocation(),
3688                              cxxSelector, getContext().VoidTy, nullptr, D,
3689                              /*isInstance=*/true, /*isVariadic=*/false,
3690                           /*isPropertyAccessor=*/true, /*isImplicitlyDeclared=*/true,
3691                              /*isDefined=*/false, ObjCMethodDecl::Required);
3692     D->addInstanceMethod(DTORMethod);
3693     CodeGenFunction(*this).GenerateObjCCtorDtorMethod(D, DTORMethod, false);
3694     D->setHasDestructors(true);
3695   }
3696
3697   // If the implementation doesn't have any ivar initializers, we don't need
3698   // a .cxx_construct.
3699   if (D->getNumIvarInitializers() == 0 ||
3700       AllTrivialInitializers(*this, D))
3701     return;
3702
3703   IdentifierInfo *II = &getContext().Idents.get(".cxx_construct");
3704   Selector cxxSelector = getContext().Selectors.getSelector(0, &II);
3705   // The constructor returns 'self'.
3706   ObjCMethodDecl *CTORMethod = ObjCMethodDecl::Create(getContext(),
3707                                                 D->getLocation(),
3708                                                 D->getLocation(),
3709                                                 cxxSelector,
3710                                                 getContext().getObjCIdType(),
3711                                                 nullptr, D, /*isInstance=*/true,
3712                                                 /*isVariadic=*/false,
3713                                                 /*isPropertyAccessor=*/true,
3714                                                 /*isImplicitlyDeclared=*/true,
3715                                                 /*isDefined=*/false,
3716                                                 ObjCMethodDecl::Required);
3717   D->addInstanceMethod(CTORMethod);
3718   CodeGenFunction(*this).GenerateObjCCtorDtorMethod(D, CTORMethod, true);
3719   D->setHasNonZeroConstructors(true);
3720 }
3721
3722 // EmitLinkageSpec - Emit all declarations in a linkage spec.
3723 void CodeGenModule::EmitLinkageSpec(const LinkageSpecDecl *LSD) {
3724   if (LSD->getLanguage() != LinkageSpecDecl::lang_c &&
3725       LSD->getLanguage() != LinkageSpecDecl::lang_cxx) {
3726     ErrorUnsupported(LSD, "linkage spec");
3727     return;
3728   }
3729
3730   EmitDeclContext(LSD);
3731 }
3732
3733 void CodeGenModule::EmitDeclContext(const DeclContext *DC) {
3734   for (auto *I : DC->decls()) {
3735     // Unlike other DeclContexts, the contents of an ObjCImplDecl at TU scope
3736     // are themselves considered "top-level", so EmitTopLevelDecl on an
3737     // ObjCImplDecl does not recursively visit them. We need to do that in
3738     // case they're nested inside another construct (LinkageSpecDecl /
3739     // ExportDecl) that does stop them from being considered "top-level".
3740     if (auto *OID = dyn_cast<ObjCImplDecl>(I)) {
3741       for (auto *M : OID->methods())
3742         EmitTopLevelDecl(M);
3743     }
3744
3745     EmitTopLevelDecl(I);
3746   }
3747 }
3748
3749 /// EmitTopLevelDecl - Emit code for a single top level declaration.
3750 void CodeGenModule::EmitTopLevelDecl(Decl *D) {
3751   // Ignore dependent declarations.
3752   if (D->getDeclContext() && D->getDeclContext()->isDependentContext())
3753     return;
3754
3755   switch (D->getKind()) {
3756   case Decl::CXXConversion:
3757   case Decl::CXXMethod:
3758   case Decl::Function:
3759     // Skip function templates
3760     if (cast<FunctionDecl>(D)->getDescribedFunctionTemplate() ||
3761         cast<FunctionDecl>(D)->isLateTemplateParsed())
3762       return;
3763
3764     EmitGlobal(cast<FunctionDecl>(D));
3765     // Always provide some coverage mapping
3766     // even for the functions that aren't emitted.
3767     AddDeferredUnusedCoverageMapping(D);
3768     break;
3769
3770   case Decl::Var:
3771   case Decl::Decomposition:
3772     // Skip variable templates
3773     if (cast<VarDecl>(D)->getDescribedVarTemplate())
3774       return;
3775   case Decl::VarTemplateSpecialization:
3776     EmitGlobal(cast<VarDecl>(D));
3777     if (auto *DD = dyn_cast<DecompositionDecl>(D))
3778       for (auto *B : DD->bindings())
3779         if (auto *HD = B->getHoldingVar())
3780           EmitGlobal(HD);
3781     break;
3782
3783   // Indirect fields from global anonymous structs and unions can be
3784   // ignored; only the actual variable requires IR gen support.
3785   case Decl::IndirectField:
3786     break;
3787
3788   // C++ Decls
3789   case Decl::Namespace:
3790     EmitDeclContext(cast<NamespaceDecl>(D));
3791     break;
3792   case Decl::CXXRecord:
3793     // Emit any static data members, they may be definitions.
3794     for (auto *I : cast<CXXRecordDecl>(D)->decls())
3795       if (isa<VarDecl>(I) || isa<CXXRecordDecl>(I))
3796         EmitTopLevelDecl(I);
3797     break;
3798     // No code generation needed.
3799   case Decl::UsingShadow:
3800   case Decl::ClassTemplate:
3801   case Decl::VarTemplate:
3802   case Decl::VarTemplatePartialSpecialization:
3803   case Decl::FunctionTemplate:
3804   case Decl::TypeAliasTemplate:
3805   case Decl::Block:
3806   case Decl::Empty:
3807     break;
3808   case Decl::Using:          // using X; [C++]
3809     if (CGDebugInfo *DI = getModuleDebugInfo())
3810         DI->EmitUsingDecl(cast<UsingDecl>(*D));
3811     return;
3812   case Decl::NamespaceAlias:
3813     if (CGDebugInfo *DI = getModuleDebugInfo())
3814         DI->EmitNamespaceAlias(cast<NamespaceAliasDecl>(*D));
3815     return;
3816   case Decl::UsingDirective: // using namespace X; [C++]
3817     if (CGDebugInfo *DI = getModuleDebugInfo())
3818       DI->EmitUsingDirective(cast<UsingDirectiveDecl>(*D));
3819     return;
3820   case Decl::CXXConstructor:
3821     // Skip function templates
3822     if (cast<FunctionDecl>(D)->getDescribedFunctionTemplate() ||
3823         cast<FunctionDecl>(D)->isLateTemplateParsed())
3824       return;
3825
3826     getCXXABI().EmitCXXConstructors(cast<CXXConstructorDecl>(D));
3827     break;
3828   case Decl::CXXDestructor:
3829     if (cast<FunctionDecl>(D)->isLateTemplateParsed())
3830       return;
3831     getCXXABI().EmitCXXDestructors(cast<CXXDestructorDecl>(D));
3832     break;
3833
3834   case Decl::StaticAssert:
3835     // Nothing to do.
3836     break;
3837
3838   // Objective-C Decls
3839
3840   // Forward declarations, no (immediate) code generation.
3841   case Decl::ObjCInterface:
3842   case Decl::ObjCCategory:
3843     break;
3844
3845   case Decl::ObjCProtocol: {
3846     auto *Proto = cast<ObjCProtocolDecl>(D);
3847     if (Proto->isThisDeclarationADefinition())
3848       ObjCRuntime->GenerateProtocol(Proto);
3849     break;
3850   }
3851
3852   case Decl::ObjCCategoryImpl:
3853     // Categories have properties but don't support synthesize so we
3854     // can ignore them here.
3855     ObjCRuntime->GenerateCategory(cast<ObjCCategoryImplDecl>(D));
3856     break;
3857
3858   case Decl::ObjCImplementation: {
3859     auto *OMD = cast<ObjCImplementationDecl>(D);
3860     EmitObjCPropertyImplementations(OMD);
3861     EmitObjCIvarInitializations(OMD);
3862     ObjCRuntime->GenerateClass(OMD);
3863     // Emit global variable debug information.
3864     if (CGDebugInfo *DI = getModuleDebugInfo())
3865       if (getCodeGenOpts().getDebugInfo() >= codegenoptions::LimitedDebugInfo)
3866         DI->getOrCreateInterfaceType(getContext().getObjCInterfaceType(
3867             OMD->getClassInterface()), OMD->getLocation());
3868     break;
3869   }
3870   case Decl::ObjCMethod: {
3871     auto *OMD = cast<ObjCMethodDecl>(D);
3872     // If this is not a prototype, emit the body.
3873     if (OMD->getBody())
3874       CodeGenFunction(*this).GenerateObjCMethod(OMD);
3875     break;
3876   }
3877   case Decl::ObjCCompatibleAlias:
3878     ObjCRuntime->RegisterAlias(cast<ObjCCompatibleAliasDecl>(D));
3879     break;
3880
3881   case Decl::PragmaComment: {
3882     const auto *PCD = cast<PragmaCommentDecl>(D);
3883     switch (PCD->getCommentKind()) {
3884     case PCK_Unknown:
3885       llvm_unreachable("unexpected pragma comment kind");
3886     case PCK_Linker:
3887       AppendLinkerOptions(PCD->getArg());
3888       break;
3889     case PCK_Lib:
3890       AddDependentLib(PCD->getArg());
3891       break;
3892     case PCK_Compiler:
3893     case PCK_ExeStr:
3894     case PCK_User:
3895       break; // We ignore all of these.
3896     }
3897     break;
3898   }
3899
3900   case Decl::PragmaDetectMismatch: {
3901     const auto *PDMD = cast<PragmaDetectMismatchDecl>(D);
3902     AddDetectMismatch(PDMD->getName(), PDMD->getValue());
3903     break;
3904   }
3905
3906   case Decl::LinkageSpec:
3907     EmitLinkageSpec(cast<LinkageSpecDecl>(D));
3908     break;
3909
3910   case Decl::FileScopeAsm: {
3911     // File-scope asm is ignored during device-side CUDA compilation.
3912     if (LangOpts.CUDA && LangOpts.CUDAIsDevice)
3913       break;
3914     // File-scope asm is ignored during device-side OpenMP compilation.
3915     if (LangOpts.OpenMPIsDevice)
3916       break;
3917     auto *AD = cast<FileScopeAsmDecl>(D);
3918     getModule().appendModuleInlineAsm(AD->getAsmString()->getString());
3919     break;
3920   }
3921
3922   case Decl::Import: {
3923     auto *Import = cast<ImportDecl>(D);
3924
3925     // If we've already imported this module, we're done.
3926     if (!ImportedModules.insert(Import->getImportedModule()))
3927       break;
3928
3929     // Emit debug information for direct imports.
3930     if (!Import->getImportedOwningModule()) {
3931       if (CGDebugInfo *DI = getModuleDebugInfo())
3932         DI->EmitImportDecl(*Import);
3933     }
3934
3935     // Find all of the submodules and emit the module initializers.
3936     llvm::SmallPtrSet<clang::Module *, 16> Visited;
3937     SmallVector<clang::Module *, 16> Stack;
3938     Visited.insert(Import->getImportedModule());
3939     Stack.push_back(Import->getImportedModule());
3940
3941     while (!Stack.empty()) {
3942       clang::Module *Mod = Stack.pop_back_val();
3943       if (!EmittedModuleInitializers.insert(Mod).second)
3944         continue;
3945
3946       for (auto *D : Context.getModuleInitializers(Mod))
3947         EmitTopLevelDecl(D);
3948
3949       // Visit the submodules of this module.
3950       for (clang::Module::submodule_iterator Sub = Mod->submodule_begin(),
3951                                              SubEnd = Mod->submodule_end();
3952            Sub != SubEnd; ++Sub) {
3953         // Skip explicit children; they need to be explicitly imported to emit
3954         // the initializers.
3955         if ((*Sub)->IsExplicit)
3956           continue;
3957
3958         if (Visited.insert(*Sub).second)
3959           Stack.push_back(*Sub);
3960       }
3961     }
3962     break;
3963   }
3964
3965   case Decl::Export:
3966     EmitDeclContext(cast<ExportDecl>(D));
3967     break;
3968
3969   case Decl::OMPThreadPrivate:
3970     EmitOMPThreadPrivateDecl(cast<OMPThreadPrivateDecl>(D));
3971     break;
3972
3973   case Decl::ClassTemplateSpecialization: {
3974     const auto *Spec = cast<ClassTemplateSpecializationDecl>(D);
3975     if (DebugInfo &&
3976         Spec->getSpecializationKind() == TSK_ExplicitInstantiationDefinition &&
3977         Spec->hasDefinition())
3978       DebugInfo->completeTemplateDefinition(*Spec);
3979     break;
3980   }
3981
3982   case Decl::OMPDeclareReduction:
3983     EmitOMPDeclareReduction(cast<OMPDeclareReductionDecl>(D));
3984     break;
3985
3986   default:
3987     // Make sure we handled everything we should, every other kind is a
3988     // non-top-level decl.  FIXME: Would be nice to have an isTopLevelDeclKind
3989     // function. Need to recode Decl::Kind to do that easily.
3990     assert(isa<TypeDecl>(D) && "Unsupported decl kind");
3991     break;
3992   }
3993 }
3994
3995 void CodeGenModule::AddDeferredUnusedCoverageMapping(Decl *D) {
3996   // Do we need to generate coverage mapping?
3997   if (!CodeGenOpts.CoverageMapping)
3998     return;
3999   switch (D->getKind()) {
4000   case Decl::CXXConversion:
4001   case Decl::CXXMethod:
4002   case Decl::Function:
4003   case Decl::ObjCMethod:
4004   case Decl::CXXConstructor:
4005   case Decl::CXXDestructor: {
4006     if (!cast<FunctionDecl>(D)->doesThisDeclarationHaveABody())
4007       return;
4008     auto I = DeferredEmptyCoverageMappingDecls.find(D);
4009     if (I == DeferredEmptyCoverageMappingDecls.end())
4010       DeferredEmptyCoverageMappingDecls[D] = true;
4011     break;
4012   }
4013   default:
4014     break;
4015   };
4016 }
4017
4018 void CodeGenModule::ClearUnusedCoverageMapping(const Decl *D) {
4019   // Do we need to generate coverage mapping?
4020   if (!CodeGenOpts.CoverageMapping)
4021     return;
4022   if (const auto *Fn = dyn_cast<FunctionDecl>(D)) {
4023     if (Fn->isTemplateInstantiation())
4024       ClearUnusedCoverageMapping(Fn->getTemplateInstantiationPattern());
4025   }
4026   auto I = DeferredEmptyCoverageMappingDecls.find(D);
4027   if (I == DeferredEmptyCoverageMappingDecls.end())
4028     DeferredEmptyCoverageMappingDecls[D] = false;
4029   else
4030     I->second = false;
4031 }
4032
4033 void CodeGenModule::EmitDeferredUnusedCoverageMappings() {
4034   std::vector<const Decl *> DeferredDecls;
4035   for (const auto &I : DeferredEmptyCoverageMappingDecls) {
4036     if (!I.second)
4037       continue;
4038     DeferredDecls.push_back(I.first);
4039   }
4040   // Sort the declarations by their location to make sure that the tests get a
4041   // predictable order for the coverage mapping for the unused declarations.
4042   if (CodeGenOpts.DumpCoverageMapping)
4043     std::sort(DeferredDecls.begin(), DeferredDecls.end(),
4044               [] (const Decl *LHS, const Decl *RHS) {
4045       return LHS->getLocStart() < RHS->getLocStart();
4046     });
4047   for (const auto *D : DeferredDecls) {
4048     switch (D->getKind()) {
4049     case Decl::CXXConversion:
4050     case Decl::CXXMethod:
4051     case Decl::Function:
4052     case Decl::ObjCMethod: {
4053       CodeGenPGO PGO(*this);
4054       GlobalDecl GD(cast<FunctionDecl>(D));
4055       PGO.emitEmptyCounterMapping(D, getMangledName(GD),
4056                                   getFunctionLinkage(GD));
4057       break;
4058     }
4059     case Decl::CXXConstructor: {
4060       CodeGenPGO PGO(*this);
4061       GlobalDecl GD(cast<CXXConstructorDecl>(D), Ctor_Base);
4062       PGO.emitEmptyCounterMapping(D, getMangledName(GD),
4063                                   getFunctionLinkage(GD));
4064       break;
4065     }
4066     case Decl::CXXDestructor: {
4067       CodeGenPGO PGO(*this);
4068       GlobalDecl GD(cast<CXXDestructorDecl>(D), Dtor_Base);
4069       PGO.emitEmptyCounterMapping(D, getMangledName(GD),
4070                                   getFunctionLinkage(GD));
4071       break;
4072     }
4073     default:
4074       break;
4075     };
4076   }
4077 }
4078
4079 /// Turns the given pointer into a constant.
4080 static llvm::Constant *GetPointerConstant(llvm::LLVMContext &Context,
4081                                           const void *Ptr) {
4082   uintptr_t PtrInt = reinterpret_cast<uintptr_t>(Ptr);
4083   llvm::Type *i64 = llvm::Type::getInt64Ty(Context);
4084   return llvm::ConstantInt::get(i64, PtrInt);
4085 }
4086
4087 static void EmitGlobalDeclMetadata(CodeGenModule &CGM,
4088                                    llvm::NamedMDNode *&GlobalMetadata,
4089                                    GlobalDecl D,
4090                                    llvm::GlobalValue *Addr) {
4091   if (!GlobalMetadata)
4092     GlobalMetadata =
4093       CGM.getModule().getOrInsertNamedMetadata("clang.global.decl.ptrs");
4094
4095   // TODO: should we report variant information for ctors/dtors?
4096   llvm::Metadata *Ops[] = {llvm::ConstantAsMetadata::get(Addr),
4097                            llvm::ConstantAsMetadata::get(GetPointerConstant(
4098                                CGM.getLLVMContext(), D.getDecl()))};
4099   GlobalMetadata->addOperand(llvm::MDNode::get(CGM.getLLVMContext(), Ops));
4100 }
4101
4102 /// For each function which is declared within an extern "C" region and marked
4103 /// as 'used', but has internal linkage, create an alias from the unmangled
4104 /// name to the mangled name if possible. People expect to be able to refer
4105 /// to such functions with an unmangled name from inline assembly within the
4106 /// same translation unit.
4107 void CodeGenModule::EmitStaticExternCAliases() {
4108   // Don't do anything if we're generating CUDA device code -- the NVPTX
4109   // assembly target doesn't support aliases.
4110   if (Context.getTargetInfo().getTriple().isNVPTX())
4111     return;
4112   for (auto &I : StaticExternCValues) {
4113     IdentifierInfo *Name = I.first;
4114     llvm::GlobalValue *Val = I.second;
4115     if (Val && !getModule().getNamedValue(Name->getName()))
4116       addUsedGlobal(llvm::GlobalAlias::create(Name->getName(), Val));
4117   }
4118 }
4119
4120 bool CodeGenModule::lookupRepresentativeDecl(StringRef MangledName,
4121                                              GlobalDecl &Result) const {
4122   auto Res = Manglings.find(MangledName);
4123   if (Res == Manglings.end())
4124     return false;
4125   Result = Res->getValue();
4126   return true;
4127 }
4128
4129 /// Emits metadata nodes associating all the global values in the
4130 /// current module with the Decls they came from.  This is useful for
4131 /// projects using IR gen as a subroutine.
4132 ///
4133 /// Since there's currently no way to associate an MDNode directly
4134 /// with an llvm::GlobalValue, we create a global named metadata
4135 /// with the name 'clang.global.decl.ptrs'.
4136 void CodeGenModule::EmitDeclMetadata() {
4137   llvm::NamedMDNode *GlobalMetadata = nullptr;
4138
4139   for (auto &I : MangledDeclNames) {
4140     llvm::GlobalValue *Addr = getModule().getNamedValue(I.second);
4141     // Some mangled names don't necessarily have an associated GlobalValue
4142     // in this module, e.g. if we mangled it for DebugInfo.
4143     if (Addr)
4144       EmitGlobalDeclMetadata(*this, GlobalMetadata, I.first, Addr);
4145   }
4146 }
4147
4148 /// Emits metadata nodes for all the local variables in the current
4149 /// function.
4150 void CodeGenFunction::EmitDeclMetadata() {
4151   if (LocalDeclMap.empty()) return;
4152
4153   llvm::LLVMContext &Context = getLLVMContext();
4154
4155   // Find the unique metadata ID for this name.
4156   unsigned DeclPtrKind = Context.getMDKindID("clang.decl.ptr");
4157
4158   llvm::NamedMDNode *GlobalMetadata = nullptr;
4159
4160   for (auto &I : LocalDeclMap) {
4161     const Decl *D = I.first;
4162     llvm::Value *Addr = I.second.getPointer();
4163     if (auto *Alloca = dyn_cast<llvm::AllocaInst>(Addr)) {
4164       llvm::Value *DAddr = GetPointerConstant(getLLVMContext(), D);
4165       Alloca->setMetadata(
4166           DeclPtrKind, llvm::MDNode::get(
4167                            Context, llvm::ValueAsMetadata::getConstant(DAddr)));
4168     } else if (auto *GV = dyn_cast<llvm::GlobalValue>(Addr)) {
4169       GlobalDecl GD = GlobalDecl(cast<VarDecl>(D));
4170       EmitGlobalDeclMetadata(CGM, GlobalMetadata, GD, GV);
4171     }
4172   }
4173 }
4174
4175 void CodeGenModule::EmitVersionIdentMetadata() {
4176   llvm::NamedMDNode *IdentMetadata =
4177     TheModule.getOrInsertNamedMetadata("llvm.ident");
4178   std::string Version = getClangFullVersion();
4179   llvm::LLVMContext &Ctx = TheModule.getContext();
4180
4181   llvm::Metadata *IdentNode[] = {llvm::MDString::get(Ctx, Version)};
4182   IdentMetadata->addOperand(llvm::MDNode::get(Ctx, IdentNode));
4183 }
4184
4185 void CodeGenModule::EmitTargetMetadata() {
4186   // Warning, new MangledDeclNames may be appended within this loop.
4187   // We rely on MapVector insertions adding new elements to the end
4188   // of the container.
4189   // FIXME: Move this loop into the one target that needs it, and only
4190   // loop over those declarations for which we couldn't emit the target
4191   // metadata when we emitted the declaration.
4192   for (unsigned I = 0; I != MangledDeclNames.size(); ++I) {
4193     auto Val = *(MangledDeclNames.begin() + I);
4194     const Decl *D = Val.first.getDecl()->getMostRecentDecl();
4195     llvm::GlobalValue *GV = GetGlobalValue(Val.second);
4196     getTargetCodeGenInfo().emitTargetMD(D, GV, *this);
4197   }
4198 }
4199
4200 void CodeGenModule::EmitCoverageFile() {
4201   if (getCodeGenOpts().CoverageDataFile.empty() &&
4202       getCodeGenOpts().CoverageNotesFile.empty())
4203     return;
4204
4205   llvm::NamedMDNode *CUNode = TheModule.getNamedMetadata("llvm.dbg.cu");
4206   if (!CUNode)
4207     return;
4208
4209   llvm::NamedMDNode *GCov = TheModule.getOrInsertNamedMetadata("llvm.gcov");
4210   llvm::LLVMContext &Ctx = TheModule.getContext();
4211   auto *CoverageDataFile =
4212       llvm::MDString::get(Ctx, getCodeGenOpts().CoverageDataFile);
4213   auto *CoverageNotesFile =
4214       llvm::MDString::get(Ctx, getCodeGenOpts().CoverageNotesFile);
4215   for (int i = 0, e = CUNode->getNumOperands(); i != e; ++i) {
4216     llvm::MDNode *CU = CUNode->getOperand(i);
4217     llvm::Metadata *Elts[] = {CoverageNotesFile, CoverageDataFile, CU};
4218     GCov->addOperand(llvm::MDNode::get(Ctx, Elts));
4219   }
4220 }
4221
4222 llvm::Constant *CodeGenModule::EmitUuidofInitializer(StringRef Uuid) {
4223   // Sema has checked that all uuid strings are of the form
4224   // "12345678-1234-1234-1234-1234567890ab".
4225   assert(Uuid.size() == 36);
4226   for (unsigned i = 0; i < 36; ++i) {
4227     if (i == 8 || i == 13 || i == 18 || i == 23) assert(Uuid[i] == '-');
4228     else                                         assert(isHexDigit(Uuid[i]));
4229   }
4230
4231   // The starts of all bytes of Field3 in Uuid. Field 3 is "1234-1234567890ab".
4232   const unsigned Field3ValueOffsets[8] = { 19, 21, 24, 26, 28, 30, 32, 34 };
4233
4234   llvm::Constant *Field3[8];
4235   for (unsigned Idx = 0; Idx < 8; ++Idx)
4236     Field3[Idx] = llvm::ConstantInt::get(
4237         Int8Ty, Uuid.substr(Field3ValueOffsets[Idx], 2), 16);
4238
4239   llvm::Constant *Fields[4] = {
4240     llvm::ConstantInt::get(Int32Ty, Uuid.substr(0,  8), 16),
4241     llvm::ConstantInt::get(Int16Ty, Uuid.substr(9,  4), 16),
4242     llvm::ConstantInt::get(Int16Ty, Uuid.substr(14, 4), 16),
4243     llvm::ConstantArray::get(llvm::ArrayType::get(Int8Ty, 8), Field3)
4244   };
4245
4246   return llvm::ConstantStruct::getAnon(Fields);
4247 }
4248
4249 llvm::Constant *CodeGenModule::GetAddrOfRTTIDescriptor(QualType Ty,
4250                                                        bool ForEH) {
4251   // Return a bogus pointer if RTTI is disabled, unless it's for EH.
4252   // FIXME: should we even be calling this method if RTTI is disabled
4253   // and it's not for EH?
4254   if (!ForEH && !getLangOpts().RTTI)
4255     return llvm::Constant::getNullValue(Int8PtrTy);
4256
4257   if (ForEH && Ty->isObjCObjectPointerType() &&
4258       LangOpts.ObjCRuntime.isGNUFamily())
4259     return ObjCRuntime->GetEHType(Ty);
4260
4261   return getCXXABI().getAddrOfRTTIDescriptor(Ty);
4262 }
4263
4264 void CodeGenModule::EmitOMPThreadPrivateDecl(const OMPThreadPrivateDecl *D) {
4265   for (auto RefExpr : D->varlists()) {
4266     auto *VD = cast<VarDecl>(cast<DeclRefExpr>(RefExpr)->getDecl());
4267     bool PerformInit =
4268         VD->getAnyInitializer() &&
4269         !VD->getAnyInitializer()->isConstantInitializer(getContext(),
4270                                                         /*ForRef=*/false);
4271
4272     Address Addr(GetAddrOfGlobalVar(VD), getContext().getDeclAlign(VD));
4273     if (auto InitFunction = getOpenMPRuntime().emitThreadPrivateVarDefinition(
4274             VD, Addr, RefExpr->getLocStart(), PerformInit))
4275       CXXGlobalInits.push_back(InitFunction);
4276   }
4277 }
4278
4279 llvm::Metadata *CodeGenModule::CreateMetadataIdentifierForType(QualType T) {
4280   llvm::Metadata *&InternalId = MetadataIdMap[T.getCanonicalType()];
4281   if (InternalId)
4282     return InternalId;
4283
4284   if (isExternallyVisible(T->getLinkage())) {
4285     std::string OutName;
4286     llvm::raw_string_ostream Out(OutName);
4287     getCXXABI().getMangleContext().mangleTypeName(T, Out);
4288
4289     InternalId = llvm::MDString::get(getLLVMContext(), Out.str());
4290   } else {
4291     InternalId = llvm::MDNode::getDistinct(getLLVMContext(),
4292                                            llvm::ArrayRef<llvm::Metadata *>());
4293   }
4294
4295   return InternalId;
4296 }
4297
4298 /// Returns whether this module needs the "all-vtables" type identifier.
4299 bool CodeGenModule::NeedAllVtablesTypeId() const {
4300   // Returns true if at least one of vtable-based CFI checkers is enabled and
4301   // is not in the trapping mode.
4302   return ((LangOpts.Sanitize.has(SanitizerKind::CFIVCall) &&
4303            !CodeGenOpts.SanitizeTrap.has(SanitizerKind::CFIVCall)) ||
4304           (LangOpts.Sanitize.has(SanitizerKind::CFINVCall) &&
4305            !CodeGenOpts.SanitizeTrap.has(SanitizerKind::CFINVCall)) ||
4306           (LangOpts.Sanitize.has(SanitizerKind::CFIDerivedCast) &&
4307            !CodeGenOpts.SanitizeTrap.has(SanitizerKind::CFIDerivedCast)) ||
4308           (LangOpts.Sanitize.has(SanitizerKind::CFIUnrelatedCast) &&
4309            !CodeGenOpts.SanitizeTrap.has(SanitizerKind::CFIUnrelatedCast)));
4310 }
4311
4312 void CodeGenModule::AddVTableTypeMetadata(llvm::GlobalVariable *VTable,
4313                                           CharUnits Offset,
4314                                           const CXXRecordDecl *RD) {
4315   llvm::Metadata *MD =
4316       CreateMetadataIdentifierForType(QualType(RD->getTypeForDecl(), 0));
4317   VTable->addTypeMetadata(Offset.getQuantity(), MD);
4318
4319   if (CodeGenOpts.SanitizeCfiCrossDso)
4320     if (auto CrossDsoTypeId = CreateCrossDsoCfiTypeId(MD))
4321       VTable->addTypeMetadata(Offset.getQuantity(),
4322                               llvm::ConstantAsMetadata::get(CrossDsoTypeId));
4323
4324   if (NeedAllVtablesTypeId()) {
4325     llvm::Metadata *MD = llvm::MDString::get(getLLVMContext(), "all-vtables");
4326     VTable->addTypeMetadata(Offset.getQuantity(), MD);
4327   }
4328 }
4329
4330 // Fills in the supplied string map with the set of target features for the
4331 // passed in function.
4332 void CodeGenModule::getFunctionFeatureMap(llvm::StringMap<bool> &FeatureMap,
4333                                           const FunctionDecl *FD) {
4334   StringRef TargetCPU = Target.getTargetOpts().CPU;
4335   if (const auto *TD = FD->getAttr<TargetAttr>()) {
4336     // If we have a TargetAttr build up the feature map based on that.
4337     TargetAttr::ParsedTargetAttr ParsedAttr = TD->parse();
4338
4339     // Make a copy of the features as passed on the command line into the
4340     // beginning of the additional features from the function to override.
4341     ParsedAttr.first.insert(ParsedAttr.first.begin(),
4342                             Target.getTargetOpts().FeaturesAsWritten.begin(),
4343                             Target.getTargetOpts().FeaturesAsWritten.end());
4344
4345     if (ParsedAttr.second != "")
4346       TargetCPU = ParsedAttr.second;
4347
4348     // Now populate the feature map, first with the TargetCPU which is either
4349     // the default or a new one from the target attribute string. Then we'll use
4350     // the passed in features (FeaturesAsWritten) along with the new ones from
4351     // the attribute.
4352     Target.initFeatureMap(FeatureMap, getDiags(), TargetCPU, ParsedAttr.first);
4353   } else {
4354     Target.initFeatureMap(FeatureMap, getDiags(), TargetCPU,
4355                           Target.getTargetOpts().Features);
4356   }
4357 }
4358
4359 llvm::SanitizerStatReport &CodeGenModule::getSanStats() {
4360   if (!SanStats)
4361     SanStats = llvm::make_unique<llvm::SanitizerStatReport>(&getModule());
4362
4363   return *SanStats;
4364 }
4365 llvm::Value *
4366 CodeGenModule::createOpenCLIntToSamplerConversion(const Expr *E,
4367                                                   CodeGenFunction &CGF) {
4368   llvm::Constant *C = EmitConstantExpr(E, E->getType(), &CGF);
4369   auto SamplerT = getOpenCLRuntime().getSamplerType();
4370   auto FTy = llvm::FunctionType::get(SamplerT, {C->getType()}, false);
4371   return CGF.Builder.CreateCall(CreateRuntimeFunction(FTy,
4372                                 "__translate_sampler_initializer"),
4373                                 {C});
4374 }