10b57cec5SDimitry Andric //===--- CodeGenModule.cpp - Emit LLVM Code from ASTs for a Module --------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric //
90b57cec5SDimitry Andric // This coordinates the per-module state used while generating code.
100b57cec5SDimitry Andric //
110b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
120b57cec5SDimitry Andric
130b57cec5SDimitry Andric #include "CodeGenModule.h"
14fcaf7f86SDimitry Andric #include "ABIInfo.h"
150b57cec5SDimitry Andric #include "CGBlocks.h"
160b57cec5SDimitry Andric #include "CGCUDARuntime.h"
170b57cec5SDimitry Andric #include "CGCXXABI.h"
180b57cec5SDimitry Andric #include "CGCall.h"
190b57cec5SDimitry Andric #include "CGDebugInfo.h"
2081ad6265SDimitry Andric #include "CGHLSLRuntime.h"
210b57cec5SDimitry Andric #include "CGObjCRuntime.h"
220b57cec5SDimitry Andric #include "CGOpenCLRuntime.h"
230b57cec5SDimitry Andric #include "CGOpenMPRuntime.h"
24349cc55cSDimitry Andric #include "CGOpenMPRuntimeGPU.h"
250b57cec5SDimitry Andric #include "CodeGenFunction.h"
260b57cec5SDimitry Andric #include "CodeGenPGO.h"
270b57cec5SDimitry Andric #include "ConstantEmitter.h"
280b57cec5SDimitry Andric #include "CoverageMappingGen.h"
290b57cec5SDimitry Andric #include "TargetInfo.h"
300b57cec5SDimitry Andric #include "clang/AST/ASTContext.h"
315f757f3fSDimitry Andric #include "clang/AST/ASTLambda.h"
320b57cec5SDimitry Andric #include "clang/AST/CharUnits.h"
330fca6ea1SDimitry Andric #include "clang/AST/Decl.h"
340b57cec5SDimitry Andric #include "clang/AST/DeclCXX.h"
350b57cec5SDimitry Andric #include "clang/AST/DeclObjC.h"
360b57cec5SDimitry Andric #include "clang/AST/DeclTemplate.h"
370b57cec5SDimitry Andric #include "clang/AST/Mangle.h"
380b57cec5SDimitry Andric #include "clang/AST/RecursiveASTVisitor.h"
390b57cec5SDimitry Andric #include "clang/AST/StmtVisitor.h"
400b57cec5SDimitry Andric #include "clang/Basic/Builtins.h"
410b57cec5SDimitry Andric #include "clang/Basic/CharInfo.h"
420b57cec5SDimitry Andric #include "clang/Basic/CodeGenOptions.h"
430b57cec5SDimitry Andric #include "clang/Basic/Diagnostic.h"
445ffd83dbSDimitry Andric #include "clang/Basic/FileManager.h"
450b57cec5SDimitry Andric #include "clang/Basic/Module.h"
460b57cec5SDimitry Andric #include "clang/Basic/SourceManager.h"
470b57cec5SDimitry Andric #include "clang/Basic/TargetInfo.h"
480b57cec5SDimitry Andric #include "clang/Basic/Version.h"
4981ad6265SDimitry Andric #include "clang/CodeGen/BackendUtil.h"
500b57cec5SDimitry Andric #include "clang/CodeGen/ConstantInitBuilder.h"
510b57cec5SDimitry Andric #include "clang/Frontend/FrontendDiagnostic.h"
52bdd1243dSDimitry Andric #include "llvm/ADT/STLExtras.h"
53bdd1243dSDimitry Andric #include "llvm/ADT/StringExtras.h"
540b57cec5SDimitry Andric #include "llvm/ADT/StringSwitch.h"
550b57cec5SDimitry Andric #include "llvm/Analysis/TargetLibraryInfo.h"
560fca6ea1SDimitry Andric #include "llvm/BinaryFormat/ELF.h"
57480093f4SDimitry Andric #include "llvm/Frontend/OpenMP/OMPIRBuilder.h"
5806c3fb27SDimitry Andric #include "llvm/IR/AttributeMask.h"
590b57cec5SDimitry Andric #include "llvm/IR/CallingConv.h"
600b57cec5SDimitry Andric #include "llvm/IR/DataLayout.h"
610b57cec5SDimitry Andric #include "llvm/IR/Intrinsics.h"
620b57cec5SDimitry Andric #include "llvm/IR/LLVMContext.h"
630b57cec5SDimitry Andric #include "llvm/IR/Module.h"
640b57cec5SDimitry Andric #include "llvm/IR/ProfileSummary.h"
650b57cec5SDimitry Andric #include "llvm/ProfileData/InstrProfReader.h"
66bdd1243dSDimitry Andric #include "llvm/ProfileData/SampleProf.h"
67fcaf7f86SDimitry Andric #include "llvm/Support/CRC.h"
680b57cec5SDimitry Andric #include "llvm/Support/CodeGen.h"
69480093f4SDimitry Andric #include "llvm/Support/CommandLine.h"
700b57cec5SDimitry Andric #include "llvm/Support/ConvertUTF.h"
710b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h"
720b57cec5SDimitry Andric #include "llvm/Support/TimeProfiler.h"
73bdd1243dSDimitry Andric #include "llvm/Support/xxhash.h"
740fca6ea1SDimitry Andric #include "llvm/TargetParser/RISCVISAInfo.h"
7506c3fb27SDimitry Andric #include "llvm/TargetParser/Triple.h"
7606c3fb27SDimitry Andric #include "llvm/TargetParser/X86TargetParser.h"
770fca6ea1SDimitry Andric #include "llvm/Transforms/Utils/BuildLibCalls.h"
78bdd1243dSDimitry Andric #include <optional>
790b57cec5SDimitry Andric
800b57cec5SDimitry Andric using namespace clang;
810b57cec5SDimitry Andric using namespace CodeGen;
820b57cec5SDimitry Andric
830b57cec5SDimitry Andric static llvm::cl::opt<bool> LimitedCoverage(
8481ad6265SDimitry Andric "limited-coverage-experimental", llvm::cl::Hidden,
8581ad6265SDimitry Andric llvm::cl::desc("Emit limited coverage mapping information (experimental)"));
860b57cec5SDimitry Andric
870b57cec5SDimitry Andric static const char AnnotationSection[] = "llvm.metadata";
880b57cec5SDimitry Andric
createCXXABI(CodeGenModule & CGM)890b57cec5SDimitry Andric static CGCXXABI *createCXXABI(CodeGenModule &CGM) {
90fe6060f1SDimitry Andric switch (CGM.getContext().getCXXABIKind()) {
91e8d8bef9SDimitry Andric case TargetCXXABI::AppleARM64:
92480093f4SDimitry Andric case TargetCXXABI::Fuchsia:
930b57cec5SDimitry Andric case TargetCXXABI::GenericAArch64:
940b57cec5SDimitry Andric case TargetCXXABI::GenericARM:
950b57cec5SDimitry Andric case TargetCXXABI::iOS:
960b57cec5SDimitry Andric case TargetCXXABI::WatchOS:
970b57cec5SDimitry Andric case TargetCXXABI::GenericMIPS:
980b57cec5SDimitry Andric case TargetCXXABI::GenericItanium:
990b57cec5SDimitry Andric case TargetCXXABI::WebAssembly:
1005ffd83dbSDimitry Andric case TargetCXXABI::XL:
1010b57cec5SDimitry Andric return CreateItaniumCXXABI(CGM);
1020b57cec5SDimitry Andric case TargetCXXABI::Microsoft:
1030b57cec5SDimitry Andric return CreateMicrosoftCXXABI(CGM);
1040b57cec5SDimitry Andric }
1050b57cec5SDimitry Andric
1060b57cec5SDimitry Andric llvm_unreachable("invalid C++ ABI kind");
1070b57cec5SDimitry Andric }
1080b57cec5SDimitry Andric
10906c3fb27SDimitry Andric static std::unique_ptr<TargetCodeGenInfo>
createTargetCodeGenInfo(CodeGenModule & CGM)11006c3fb27SDimitry Andric createTargetCodeGenInfo(CodeGenModule &CGM) {
11106c3fb27SDimitry Andric const TargetInfo &Target = CGM.getTarget();
11206c3fb27SDimitry Andric const llvm::Triple &Triple = Target.getTriple();
11306c3fb27SDimitry Andric const CodeGenOptions &CodeGenOpts = CGM.getCodeGenOpts();
11406c3fb27SDimitry Andric
11506c3fb27SDimitry Andric switch (Triple.getArch()) {
11606c3fb27SDimitry Andric default:
11706c3fb27SDimitry Andric return createDefaultTargetCodeGenInfo(CGM);
11806c3fb27SDimitry Andric
11906c3fb27SDimitry Andric case llvm::Triple::le32:
12006c3fb27SDimitry Andric return createPNaClTargetCodeGenInfo(CGM);
12106c3fb27SDimitry Andric case llvm::Triple::m68k:
12206c3fb27SDimitry Andric return createM68kTargetCodeGenInfo(CGM);
12306c3fb27SDimitry Andric case llvm::Triple::mips:
12406c3fb27SDimitry Andric case llvm::Triple::mipsel:
12506c3fb27SDimitry Andric if (Triple.getOS() == llvm::Triple::NaCl)
12606c3fb27SDimitry Andric return createPNaClTargetCodeGenInfo(CGM);
12706c3fb27SDimitry Andric return createMIPSTargetCodeGenInfo(CGM, /*IsOS32=*/true);
12806c3fb27SDimitry Andric
12906c3fb27SDimitry Andric case llvm::Triple::mips64:
13006c3fb27SDimitry Andric case llvm::Triple::mips64el:
13106c3fb27SDimitry Andric return createMIPSTargetCodeGenInfo(CGM, /*IsOS32=*/false);
13206c3fb27SDimitry Andric
13306c3fb27SDimitry Andric case llvm::Triple::avr: {
13406c3fb27SDimitry Andric // For passing parameters, R8~R25 are used on avr, and R18~R25 are used
13506c3fb27SDimitry Andric // on avrtiny. For passing return value, R18~R25 are used on avr, and
13606c3fb27SDimitry Andric // R22~R25 are used on avrtiny.
13706c3fb27SDimitry Andric unsigned NPR = Target.getABI() == "avrtiny" ? 6 : 18;
13806c3fb27SDimitry Andric unsigned NRR = Target.getABI() == "avrtiny" ? 4 : 8;
13906c3fb27SDimitry Andric return createAVRTargetCodeGenInfo(CGM, NPR, NRR);
14006c3fb27SDimitry Andric }
14106c3fb27SDimitry Andric
14206c3fb27SDimitry Andric case llvm::Triple::aarch64:
14306c3fb27SDimitry Andric case llvm::Triple::aarch64_32:
14406c3fb27SDimitry Andric case llvm::Triple::aarch64_be: {
14506c3fb27SDimitry Andric AArch64ABIKind Kind = AArch64ABIKind::AAPCS;
14606c3fb27SDimitry Andric if (Target.getABI() == "darwinpcs")
14706c3fb27SDimitry Andric Kind = AArch64ABIKind::DarwinPCS;
14806c3fb27SDimitry Andric else if (Triple.isOSWindows())
14906c3fb27SDimitry Andric return createWindowsAArch64TargetCodeGenInfo(CGM, AArch64ABIKind::Win64);
1500fca6ea1SDimitry Andric else if (Target.getABI() == "aapcs-soft")
1510fca6ea1SDimitry Andric Kind = AArch64ABIKind::AAPCSSoft;
1520fca6ea1SDimitry Andric else if (Target.getABI() == "pauthtest")
1530fca6ea1SDimitry Andric Kind = AArch64ABIKind::PAuthTest;
15406c3fb27SDimitry Andric
15506c3fb27SDimitry Andric return createAArch64TargetCodeGenInfo(CGM, Kind);
15606c3fb27SDimitry Andric }
15706c3fb27SDimitry Andric
15806c3fb27SDimitry Andric case llvm::Triple::wasm32:
15906c3fb27SDimitry Andric case llvm::Triple::wasm64: {
16006c3fb27SDimitry Andric WebAssemblyABIKind Kind = WebAssemblyABIKind::MVP;
16106c3fb27SDimitry Andric if (Target.getABI() == "experimental-mv")
16206c3fb27SDimitry Andric Kind = WebAssemblyABIKind::ExperimentalMV;
16306c3fb27SDimitry Andric return createWebAssemblyTargetCodeGenInfo(CGM, Kind);
16406c3fb27SDimitry Andric }
16506c3fb27SDimitry Andric
16606c3fb27SDimitry Andric case llvm::Triple::arm:
16706c3fb27SDimitry Andric case llvm::Triple::armeb:
16806c3fb27SDimitry Andric case llvm::Triple::thumb:
16906c3fb27SDimitry Andric case llvm::Triple::thumbeb: {
17006c3fb27SDimitry Andric if (Triple.getOS() == llvm::Triple::Win32)
17106c3fb27SDimitry Andric return createWindowsARMTargetCodeGenInfo(CGM, ARMABIKind::AAPCS_VFP);
17206c3fb27SDimitry Andric
17306c3fb27SDimitry Andric ARMABIKind Kind = ARMABIKind::AAPCS;
17406c3fb27SDimitry Andric StringRef ABIStr = Target.getABI();
17506c3fb27SDimitry Andric if (ABIStr == "apcs-gnu")
17606c3fb27SDimitry Andric Kind = ARMABIKind::APCS;
17706c3fb27SDimitry Andric else if (ABIStr == "aapcs16")
17806c3fb27SDimitry Andric Kind = ARMABIKind::AAPCS16_VFP;
17906c3fb27SDimitry Andric else if (CodeGenOpts.FloatABI == "hard" ||
180d686ce93SDimitry Andric (CodeGenOpts.FloatABI != "soft" && Triple.isHardFloatABI()))
18106c3fb27SDimitry Andric Kind = ARMABIKind::AAPCS_VFP;
18206c3fb27SDimitry Andric
18306c3fb27SDimitry Andric return createARMTargetCodeGenInfo(CGM, Kind);
18406c3fb27SDimitry Andric }
18506c3fb27SDimitry Andric
18606c3fb27SDimitry Andric case llvm::Triple::ppc: {
18706c3fb27SDimitry Andric if (Triple.isOSAIX())
18806c3fb27SDimitry Andric return createAIXTargetCodeGenInfo(CGM, /*Is64Bit=*/false);
18906c3fb27SDimitry Andric
19006c3fb27SDimitry Andric bool IsSoftFloat =
19106c3fb27SDimitry Andric CodeGenOpts.FloatABI == "soft" || Target.hasFeature("spe");
19206c3fb27SDimitry Andric return createPPC32TargetCodeGenInfo(CGM, IsSoftFloat);
19306c3fb27SDimitry Andric }
19406c3fb27SDimitry Andric case llvm::Triple::ppcle: {
19506c3fb27SDimitry Andric bool IsSoftFloat = CodeGenOpts.FloatABI == "soft";
19606c3fb27SDimitry Andric return createPPC32TargetCodeGenInfo(CGM, IsSoftFloat);
19706c3fb27SDimitry Andric }
19806c3fb27SDimitry Andric case llvm::Triple::ppc64:
19906c3fb27SDimitry Andric if (Triple.isOSAIX())
20006c3fb27SDimitry Andric return createAIXTargetCodeGenInfo(CGM, /*Is64Bit=*/true);
20106c3fb27SDimitry Andric
20206c3fb27SDimitry Andric if (Triple.isOSBinFormatELF()) {
20306c3fb27SDimitry Andric PPC64_SVR4_ABIKind Kind = PPC64_SVR4_ABIKind::ELFv1;
20406c3fb27SDimitry Andric if (Target.getABI() == "elfv2")
20506c3fb27SDimitry Andric Kind = PPC64_SVR4_ABIKind::ELFv2;
20606c3fb27SDimitry Andric bool IsSoftFloat = CodeGenOpts.FloatABI == "soft";
20706c3fb27SDimitry Andric
20806c3fb27SDimitry Andric return createPPC64_SVR4_TargetCodeGenInfo(CGM, Kind, IsSoftFloat);
20906c3fb27SDimitry Andric }
21006c3fb27SDimitry Andric return createPPC64TargetCodeGenInfo(CGM);
21106c3fb27SDimitry Andric case llvm::Triple::ppc64le: {
21206c3fb27SDimitry Andric assert(Triple.isOSBinFormatELF() && "PPC64 LE non-ELF not supported!");
21306c3fb27SDimitry Andric PPC64_SVR4_ABIKind Kind = PPC64_SVR4_ABIKind::ELFv2;
21406c3fb27SDimitry Andric if (Target.getABI() == "elfv1")
21506c3fb27SDimitry Andric Kind = PPC64_SVR4_ABIKind::ELFv1;
21606c3fb27SDimitry Andric bool IsSoftFloat = CodeGenOpts.FloatABI == "soft";
21706c3fb27SDimitry Andric
21806c3fb27SDimitry Andric return createPPC64_SVR4_TargetCodeGenInfo(CGM, Kind, IsSoftFloat);
21906c3fb27SDimitry Andric }
22006c3fb27SDimitry Andric
22106c3fb27SDimitry Andric case llvm::Triple::nvptx:
22206c3fb27SDimitry Andric case llvm::Triple::nvptx64:
22306c3fb27SDimitry Andric return createNVPTXTargetCodeGenInfo(CGM);
22406c3fb27SDimitry Andric
22506c3fb27SDimitry Andric case llvm::Triple::msp430:
22606c3fb27SDimitry Andric return createMSP430TargetCodeGenInfo(CGM);
22706c3fb27SDimitry Andric
22806c3fb27SDimitry Andric case llvm::Triple::riscv32:
22906c3fb27SDimitry Andric case llvm::Triple::riscv64: {
23006c3fb27SDimitry Andric StringRef ABIStr = Target.getABI();
23106c3fb27SDimitry Andric unsigned XLen = Target.getPointerWidth(LangAS::Default);
23206c3fb27SDimitry Andric unsigned ABIFLen = 0;
2335f757f3fSDimitry Andric if (ABIStr.ends_with("f"))
23406c3fb27SDimitry Andric ABIFLen = 32;
2355f757f3fSDimitry Andric else if (ABIStr.ends_with("d"))
23606c3fb27SDimitry Andric ABIFLen = 64;
2377a6dacacSDimitry Andric bool EABI = ABIStr.ends_with("e");
2387a6dacacSDimitry Andric return createRISCVTargetCodeGenInfo(CGM, XLen, ABIFLen, EABI);
23906c3fb27SDimitry Andric }
24006c3fb27SDimitry Andric
24106c3fb27SDimitry Andric case llvm::Triple::systemz: {
24206c3fb27SDimitry Andric bool SoftFloat = CodeGenOpts.FloatABI == "soft";
24306c3fb27SDimitry Andric bool HasVector = !SoftFloat && Target.getABI() == "vector";
24406c3fb27SDimitry Andric return createSystemZTargetCodeGenInfo(CGM, HasVector, SoftFloat);
24506c3fb27SDimitry Andric }
24606c3fb27SDimitry Andric
24706c3fb27SDimitry Andric case llvm::Triple::tce:
24806c3fb27SDimitry Andric case llvm::Triple::tcele:
24906c3fb27SDimitry Andric return createTCETargetCodeGenInfo(CGM);
25006c3fb27SDimitry Andric
25106c3fb27SDimitry Andric case llvm::Triple::x86: {
25206c3fb27SDimitry Andric bool IsDarwinVectorABI = Triple.isOSDarwin();
25306c3fb27SDimitry Andric bool IsWin32FloatStructABI = Triple.isOSWindows() && !Triple.isOSCygMing();
25406c3fb27SDimitry Andric
25506c3fb27SDimitry Andric if (Triple.getOS() == llvm::Triple::Win32) {
25606c3fb27SDimitry Andric return createWinX86_32TargetCodeGenInfo(
25706c3fb27SDimitry Andric CGM, IsDarwinVectorABI, IsWin32FloatStructABI,
25806c3fb27SDimitry Andric CodeGenOpts.NumRegisterParameters);
25906c3fb27SDimitry Andric }
26006c3fb27SDimitry Andric return createX86_32TargetCodeGenInfo(
26106c3fb27SDimitry Andric CGM, IsDarwinVectorABI, IsWin32FloatStructABI,
26206c3fb27SDimitry Andric CodeGenOpts.NumRegisterParameters, CodeGenOpts.FloatABI == "soft");
26306c3fb27SDimitry Andric }
26406c3fb27SDimitry Andric
26506c3fb27SDimitry Andric case llvm::Triple::x86_64: {
26606c3fb27SDimitry Andric StringRef ABI = Target.getABI();
26706c3fb27SDimitry Andric X86AVXABILevel AVXLevel = (ABI == "avx512" ? X86AVXABILevel::AVX512
26806c3fb27SDimitry Andric : ABI == "avx" ? X86AVXABILevel::AVX
26906c3fb27SDimitry Andric : X86AVXABILevel::None);
27006c3fb27SDimitry Andric
27106c3fb27SDimitry Andric switch (Triple.getOS()) {
27206c3fb27SDimitry Andric case llvm::Triple::Win32:
27306c3fb27SDimitry Andric return createWinX86_64TargetCodeGenInfo(CGM, AVXLevel);
27406c3fb27SDimitry Andric default:
27506c3fb27SDimitry Andric return createX86_64TargetCodeGenInfo(CGM, AVXLevel);
27606c3fb27SDimitry Andric }
27706c3fb27SDimitry Andric }
27806c3fb27SDimitry Andric case llvm::Triple::hexagon:
27906c3fb27SDimitry Andric return createHexagonTargetCodeGenInfo(CGM);
28006c3fb27SDimitry Andric case llvm::Triple::lanai:
28106c3fb27SDimitry Andric return createLanaiTargetCodeGenInfo(CGM);
28206c3fb27SDimitry Andric case llvm::Triple::r600:
28306c3fb27SDimitry Andric return createAMDGPUTargetCodeGenInfo(CGM);
28406c3fb27SDimitry Andric case llvm::Triple::amdgcn:
28506c3fb27SDimitry Andric return createAMDGPUTargetCodeGenInfo(CGM);
28606c3fb27SDimitry Andric case llvm::Triple::sparc:
28706c3fb27SDimitry Andric return createSparcV8TargetCodeGenInfo(CGM);
28806c3fb27SDimitry Andric case llvm::Triple::sparcv9:
28906c3fb27SDimitry Andric return createSparcV9TargetCodeGenInfo(CGM);
29006c3fb27SDimitry Andric case llvm::Triple::xcore:
29106c3fb27SDimitry Andric return createXCoreTargetCodeGenInfo(CGM);
29206c3fb27SDimitry Andric case llvm::Triple::arc:
29306c3fb27SDimitry Andric return createARCTargetCodeGenInfo(CGM);
29406c3fb27SDimitry Andric case llvm::Triple::spir:
29506c3fb27SDimitry Andric case llvm::Triple::spir64:
29606c3fb27SDimitry Andric return createCommonSPIRTargetCodeGenInfo(CGM);
29706c3fb27SDimitry Andric case llvm::Triple::spirv32:
29806c3fb27SDimitry Andric case llvm::Triple::spirv64:
29906c3fb27SDimitry Andric return createSPIRVTargetCodeGenInfo(CGM);
30006c3fb27SDimitry Andric case llvm::Triple::ve:
30106c3fb27SDimitry Andric return createVETargetCodeGenInfo(CGM);
30206c3fb27SDimitry Andric case llvm::Triple::csky: {
30306c3fb27SDimitry Andric bool IsSoftFloat = !Target.hasFeature("hard-float-abi");
30406c3fb27SDimitry Andric bool hasFP64 =
30506c3fb27SDimitry Andric Target.hasFeature("fpuv2_df") || Target.hasFeature("fpuv3_df");
30606c3fb27SDimitry Andric return createCSKYTargetCodeGenInfo(CGM, IsSoftFloat ? 0
30706c3fb27SDimitry Andric : hasFP64 ? 64
30806c3fb27SDimitry Andric : 32);
30906c3fb27SDimitry Andric }
31006c3fb27SDimitry Andric case llvm::Triple::bpfeb:
31106c3fb27SDimitry Andric case llvm::Triple::bpfel:
31206c3fb27SDimitry Andric return createBPFTargetCodeGenInfo(CGM);
31306c3fb27SDimitry Andric case llvm::Triple::loongarch32:
31406c3fb27SDimitry Andric case llvm::Triple::loongarch64: {
31506c3fb27SDimitry Andric StringRef ABIStr = Target.getABI();
31606c3fb27SDimitry Andric unsigned ABIFRLen = 0;
3175f757f3fSDimitry Andric if (ABIStr.ends_with("f"))
31806c3fb27SDimitry Andric ABIFRLen = 32;
3195f757f3fSDimitry Andric else if (ABIStr.ends_with("d"))
32006c3fb27SDimitry Andric ABIFRLen = 64;
32106c3fb27SDimitry Andric return createLoongArchTargetCodeGenInfo(
32206c3fb27SDimitry Andric CGM, Target.getPointerWidth(LangAS::Default), ABIFRLen);
32306c3fb27SDimitry Andric }
32406c3fb27SDimitry Andric }
32506c3fb27SDimitry Andric }
32606c3fb27SDimitry Andric
getTargetCodeGenInfo()32706c3fb27SDimitry Andric const TargetCodeGenInfo &CodeGenModule::getTargetCodeGenInfo() {
32806c3fb27SDimitry Andric if (!TheTargetCodeGenInfo)
32906c3fb27SDimitry Andric TheTargetCodeGenInfo = createTargetCodeGenInfo(*this);
33006c3fb27SDimitry Andric return *TheTargetCodeGenInfo;
33106c3fb27SDimitry Andric }
33206c3fb27SDimitry Andric
CodeGenModule(ASTContext & C,IntrusiveRefCntPtr<llvm::vfs::FileSystem> FS,const HeaderSearchOptions & HSO,const PreprocessorOptions & PPO,const CodeGenOptions & CGO,llvm::Module & M,DiagnosticsEngine & diags,CoverageSourceInfo * CoverageInfo)333972a253aSDimitry Andric CodeGenModule::CodeGenModule(ASTContext &C,
334972a253aSDimitry Andric IntrusiveRefCntPtr<llvm::vfs::FileSystem> FS,
335972a253aSDimitry Andric const HeaderSearchOptions &HSO,
3360b57cec5SDimitry Andric const PreprocessorOptions &PPO,
3370b57cec5SDimitry Andric const CodeGenOptions &CGO, llvm::Module &M,
3380b57cec5SDimitry Andric DiagnosticsEngine &diags,
3390b57cec5SDimitry Andric CoverageSourceInfo *CoverageInfo)
34006c3fb27SDimitry Andric : Context(C), LangOpts(C.getLangOpts()), FS(FS), HeaderSearchOpts(HSO),
34106c3fb27SDimitry Andric PreprocessorOpts(PPO), CodeGenOpts(CGO), TheModule(M), Diags(diags),
34206c3fb27SDimitry Andric Target(C.getTargetInfo()), ABI(createCXXABI(*this)),
3435deeebd8SDimitry Andric VMContext(M.getContext()), VTables(*this),
34406c3fb27SDimitry Andric SanitizerMD(new SanitizerMetadata(*this)) {
3450b57cec5SDimitry Andric
3460b57cec5SDimitry Andric // Initialize the type cache.
3475deeebd8SDimitry Andric Types.reset(new CodeGenTypes(*this));
3480b57cec5SDimitry Andric llvm::LLVMContext &LLVMContext = M.getContext();
3490b57cec5SDimitry Andric VoidTy = llvm::Type::getVoidTy(LLVMContext);
3500b57cec5SDimitry Andric Int8Ty = llvm::Type::getInt8Ty(LLVMContext);
3510b57cec5SDimitry Andric Int16Ty = llvm::Type::getInt16Ty(LLVMContext);
3520b57cec5SDimitry Andric Int32Ty = llvm::Type::getInt32Ty(LLVMContext);
3530b57cec5SDimitry Andric Int64Ty = llvm::Type::getInt64Ty(LLVMContext);
3540b57cec5SDimitry Andric HalfTy = llvm::Type::getHalfTy(LLVMContext);
3555ffd83dbSDimitry Andric BFloatTy = llvm::Type::getBFloatTy(LLVMContext);
3560b57cec5SDimitry Andric FloatTy = llvm::Type::getFloatTy(LLVMContext);
3570b57cec5SDimitry Andric DoubleTy = llvm::Type::getDoubleTy(LLVMContext);
358bdd1243dSDimitry Andric PointerWidthInBits = C.getTargetInfo().getPointerWidth(LangAS::Default);
3590b57cec5SDimitry Andric PointerAlignInBytes =
360bdd1243dSDimitry Andric C.toCharUnitsFromBits(C.getTargetInfo().getPointerAlign(LangAS::Default))
361bdd1243dSDimitry Andric .getQuantity();
3620b57cec5SDimitry Andric SizeSizeInBytes =
3630b57cec5SDimitry Andric C.toCharUnitsFromBits(C.getTargetInfo().getMaxPointerWidth()).getQuantity();
3640b57cec5SDimitry Andric IntAlignInBytes =
3650b57cec5SDimitry Andric C.toCharUnitsFromBits(C.getTargetInfo().getIntAlign()).getQuantity();
366e8d8bef9SDimitry Andric CharTy =
367e8d8bef9SDimitry Andric llvm::IntegerType::get(LLVMContext, C.getTargetInfo().getCharWidth());
3680b57cec5SDimitry Andric IntTy = llvm::IntegerType::get(LLVMContext, C.getTargetInfo().getIntWidth());
3690b57cec5SDimitry Andric IntPtrTy = llvm::IntegerType::get(LLVMContext,
3700b57cec5SDimitry Andric C.getTargetInfo().getMaxPointerWidth());
3710fca6ea1SDimitry Andric Int8PtrTy = llvm::PointerType::get(LLVMContext,
3720fca6ea1SDimitry Andric C.getTargetAddressSpace(LangAS::Default));
373349cc55cSDimitry Andric const llvm::DataLayout &DL = M.getDataLayout();
3745f757f3fSDimitry Andric AllocaInt8PtrTy =
3755f757f3fSDimitry Andric llvm::PointerType::get(LLVMContext, DL.getAllocaAddrSpace());
3765f757f3fSDimitry Andric GlobalsInt8PtrTy =
3775f757f3fSDimitry Andric llvm::PointerType::get(LLVMContext, DL.getDefaultGlobalsAddressSpace());
3785f757f3fSDimitry Andric ConstGlobalsPtrTy = llvm::PointerType::get(
3795f757f3fSDimitry Andric LLVMContext, C.getTargetAddressSpace(GetGlobalConstantAddressSpace()));
3800b57cec5SDimitry Andric ASTAllocaAddressSpace = getTargetCodeGenInfo().getASTAllocaAddressSpace();
3810b57cec5SDimitry Andric
382fcaf7f86SDimitry Andric // Build C++20 Module initializers.
383fcaf7f86SDimitry Andric // TODO: Add Microsoft here once we know the mangling required for the
384fcaf7f86SDimitry Andric // initializers.
385fcaf7f86SDimitry Andric CXX20ModuleInits =
386fcaf7f86SDimitry Andric LangOpts.CPlusPlusModules && getCXXABI().getMangleContext().getKind() ==
387fcaf7f86SDimitry Andric ItaniumMangleContext::MK_Itanium;
388fcaf7f86SDimitry Andric
3890b57cec5SDimitry Andric RuntimeCC = getTargetCodeGenInfo().getABIInfo().getRuntimeCC();
3900b57cec5SDimitry Andric
3910b57cec5SDimitry Andric if (LangOpts.ObjC)
3920b57cec5SDimitry Andric createObjCRuntime();
3930b57cec5SDimitry Andric if (LangOpts.OpenCL)
3940b57cec5SDimitry Andric createOpenCLRuntime();
3950b57cec5SDimitry Andric if (LangOpts.OpenMP)
3960b57cec5SDimitry Andric createOpenMPRuntime();
3970b57cec5SDimitry Andric if (LangOpts.CUDA)
3980b57cec5SDimitry Andric createCUDARuntime();
39981ad6265SDimitry Andric if (LangOpts.HLSL)
40081ad6265SDimitry Andric createHLSLRuntime();
4010b57cec5SDimitry Andric
4020b57cec5SDimitry Andric // Enable TBAA unless it's suppressed. ThreadSanitizer needs TBAA even at O0.
4030b57cec5SDimitry Andric if (LangOpts.Sanitize.has(SanitizerKind::Thread) ||
4040b57cec5SDimitry Andric (!CodeGenOpts.RelaxedAliasing && CodeGenOpts.OptimizationLevel > 0))
4050fca6ea1SDimitry Andric TBAA.reset(new CodeGenTBAA(Context, getTypes(), TheModule, CodeGenOpts,
4065deeebd8SDimitry Andric getLangOpts()));
4070b57cec5SDimitry Andric
4080b57cec5SDimitry Andric // If debug info or coverage generation is enabled, create the CGDebugInfo
4090b57cec5SDimitry Andric // object.
41006c3fb27SDimitry Andric if (CodeGenOpts.getDebugInfo() != llvm::codegenoptions::NoDebugInfo ||
41106c3fb27SDimitry Andric CodeGenOpts.CoverageNotesFile.size() ||
41206c3fb27SDimitry Andric CodeGenOpts.CoverageDataFile.size())
4130b57cec5SDimitry Andric DebugInfo.reset(new CGDebugInfo(*this));
4140b57cec5SDimitry Andric
4150b57cec5SDimitry Andric Block.GlobalUniqueCount = 0;
4160b57cec5SDimitry Andric
4170b57cec5SDimitry Andric if (C.getLangOpts().ObjC)
4180b57cec5SDimitry Andric ObjCData.reset(new ObjCEntrypoints());
4190b57cec5SDimitry Andric
4200b57cec5SDimitry Andric if (CodeGenOpts.hasProfileClangUse()) {
4210b57cec5SDimitry Andric auto ReaderOrErr = llvm::IndexedInstrProfReader::create(
42206c3fb27SDimitry Andric CodeGenOpts.ProfileInstrumentUsePath, *FS,
42306c3fb27SDimitry Andric CodeGenOpts.ProfileRemappingFile);
424bdd1243dSDimitry Andric // We're checking for profile read errors in CompilerInvocation, so if
425bdd1243dSDimitry Andric // there was an error it should've already been caught. If it hasn't been
426bdd1243dSDimitry Andric // somehow, trip an assertion.
427bdd1243dSDimitry Andric assert(ReaderOrErr);
4280b57cec5SDimitry Andric PGOReader = std::move(ReaderOrErr.get());
4290b57cec5SDimitry Andric }
4300b57cec5SDimitry Andric
4310b57cec5SDimitry Andric // If coverage mapping generation is enabled, create the
4320b57cec5SDimitry Andric // CoverageMappingModuleGen object.
4330b57cec5SDimitry Andric if (CodeGenOpts.CoverageMapping)
4340b57cec5SDimitry Andric CoverageMapping.reset(new CoverageMappingModuleGen(*this, *CoverageInfo));
435fe6060f1SDimitry Andric
436fe6060f1SDimitry Andric // Generate the module name hash here if needed.
437fe6060f1SDimitry Andric if (CodeGenOpts.UniqueInternalLinkageNames &&
438fe6060f1SDimitry Andric !getModule().getSourceFileName().empty()) {
439fe6060f1SDimitry Andric std::string Path = getModule().getSourceFileName();
440fe6060f1SDimitry Andric // Check if a path substitution is needed from the MacroPrefixMap.
4416e75b2fbSDimitry Andric for (const auto &Entry : LangOpts.MacroPrefixMap)
442fe6060f1SDimitry Andric if (Path.rfind(Entry.first, 0) != std::string::npos) {
443fe6060f1SDimitry Andric Path = Entry.second + Path.substr(Entry.first.size());
444fe6060f1SDimitry Andric break;
445fe6060f1SDimitry Andric }
446bdd1243dSDimitry Andric ModuleNameHash = llvm::getUniqueInternalLinkagePostfix(Path);
447fe6060f1SDimitry Andric }
4480fca6ea1SDimitry Andric
4490fca6ea1SDimitry Andric // Record mregparm value now so it is visible through all of codegen.
4500fca6ea1SDimitry Andric if (Context.getTargetInfo().getTriple().getArch() == llvm::Triple::x86)
4510fca6ea1SDimitry Andric getModule().addModuleFlag(llvm::Module::Error, "NumRegisterParameters",
4520fca6ea1SDimitry Andric CodeGenOpts.NumRegisterParameters);
4530b57cec5SDimitry Andric }
4540b57cec5SDimitry Andric
~CodeGenModule()4550b57cec5SDimitry Andric CodeGenModule::~CodeGenModule() {}
4560b57cec5SDimitry Andric
createObjCRuntime()4570b57cec5SDimitry Andric void CodeGenModule::createObjCRuntime() {
4580b57cec5SDimitry Andric // This is just isGNUFamily(), but we want to force implementors of
4590b57cec5SDimitry Andric // new ABIs to decide how best to do this.
4600b57cec5SDimitry Andric switch (LangOpts.ObjCRuntime.getKind()) {
4610b57cec5SDimitry Andric case ObjCRuntime::GNUstep:
4620b57cec5SDimitry Andric case ObjCRuntime::GCC:
4630b57cec5SDimitry Andric case ObjCRuntime::ObjFW:
4640b57cec5SDimitry Andric ObjCRuntime.reset(CreateGNUObjCRuntime(*this));
4650b57cec5SDimitry Andric return;
4660b57cec5SDimitry Andric
4670b57cec5SDimitry Andric case ObjCRuntime::FragileMacOSX:
4680b57cec5SDimitry Andric case ObjCRuntime::MacOSX:
4690b57cec5SDimitry Andric case ObjCRuntime::iOS:
4700b57cec5SDimitry Andric case ObjCRuntime::WatchOS:
4710b57cec5SDimitry Andric ObjCRuntime.reset(CreateMacObjCRuntime(*this));
4720b57cec5SDimitry Andric return;
4730b57cec5SDimitry Andric }
4740b57cec5SDimitry Andric llvm_unreachable("bad runtime kind");
4750b57cec5SDimitry Andric }
4760b57cec5SDimitry Andric
createOpenCLRuntime()4770b57cec5SDimitry Andric void CodeGenModule::createOpenCLRuntime() {
4780b57cec5SDimitry Andric OpenCLRuntime.reset(new CGOpenCLRuntime(*this));
4790b57cec5SDimitry Andric }
4800b57cec5SDimitry Andric
createOpenMPRuntime()4810b57cec5SDimitry Andric void CodeGenModule::createOpenMPRuntime() {
4820b57cec5SDimitry Andric // Select a specialized code generation class based on the target, if any.
4830b57cec5SDimitry Andric // If it does not exist use the default implementation.
4840b57cec5SDimitry Andric switch (getTriple().getArch()) {
4850b57cec5SDimitry Andric case llvm::Triple::nvptx:
4860b57cec5SDimitry Andric case llvm::Triple::nvptx64:
487e8d8bef9SDimitry Andric case llvm::Triple::amdgcn:
48806c3fb27SDimitry Andric assert(getLangOpts().OpenMPIsTargetDevice &&
489349cc55cSDimitry Andric "OpenMP AMDGPU/NVPTX is only prepared to deal with device code.");
490349cc55cSDimitry Andric OpenMPRuntime.reset(new CGOpenMPRuntimeGPU(*this));
491e8d8bef9SDimitry Andric break;
4920b57cec5SDimitry Andric default:
4930b57cec5SDimitry Andric if (LangOpts.OpenMPSimd)
4940b57cec5SDimitry Andric OpenMPRuntime.reset(new CGOpenMPSIMDRuntime(*this));
4950b57cec5SDimitry Andric else
4960b57cec5SDimitry Andric OpenMPRuntime.reset(new CGOpenMPRuntime(*this));
4970b57cec5SDimitry Andric break;
4980b57cec5SDimitry Andric }
4990b57cec5SDimitry Andric }
5000b57cec5SDimitry Andric
createCUDARuntime()5010b57cec5SDimitry Andric void CodeGenModule::createCUDARuntime() {
5020b57cec5SDimitry Andric CUDARuntime.reset(CreateNVCUDARuntime(*this));
5030b57cec5SDimitry Andric }
5040b57cec5SDimitry Andric
createHLSLRuntime()50581ad6265SDimitry Andric void CodeGenModule::createHLSLRuntime() {
50681ad6265SDimitry Andric HLSLRuntime.reset(new CGHLSLRuntime(*this));
50781ad6265SDimitry Andric }
50881ad6265SDimitry Andric
addReplacement(StringRef Name,llvm::Constant * C)5090b57cec5SDimitry Andric void CodeGenModule::addReplacement(StringRef Name, llvm::Constant *C) {
5100b57cec5SDimitry Andric Replacements[Name] = C;
5110b57cec5SDimitry Andric }
5120b57cec5SDimitry Andric
applyReplacements()5130b57cec5SDimitry Andric void CodeGenModule::applyReplacements() {
5140b57cec5SDimitry Andric for (auto &I : Replacements) {
51506c3fb27SDimitry Andric StringRef MangledName = I.first;
5160b57cec5SDimitry Andric llvm::Constant *Replacement = I.second;
5170b57cec5SDimitry Andric llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
5180b57cec5SDimitry Andric if (!Entry)
5190b57cec5SDimitry Andric continue;
5200b57cec5SDimitry Andric auto *OldF = cast<llvm::Function>(Entry);
5210b57cec5SDimitry Andric auto *NewF = dyn_cast<llvm::Function>(Replacement);
5220b57cec5SDimitry Andric if (!NewF) {
5230b57cec5SDimitry Andric if (auto *Alias = dyn_cast<llvm::GlobalAlias>(Replacement)) {
5240b57cec5SDimitry Andric NewF = dyn_cast<llvm::Function>(Alias->getAliasee());
5250b57cec5SDimitry Andric } else {
5260b57cec5SDimitry Andric auto *CE = cast<llvm::ConstantExpr>(Replacement);
5270b57cec5SDimitry Andric assert(CE->getOpcode() == llvm::Instruction::BitCast ||
5280b57cec5SDimitry Andric CE->getOpcode() == llvm::Instruction::GetElementPtr);
5290b57cec5SDimitry Andric NewF = dyn_cast<llvm::Function>(CE->getOperand(0));
5300b57cec5SDimitry Andric }
5310b57cec5SDimitry Andric }
5320b57cec5SDimitry Andric
5330b57cec5SDimitry Andric // Replace old with new, but keep the old order.
5340b57cec5SDimitry Andric OldF->replaceAllUsesWith(Replacement);
5350b57cec5SDimitry Andric if (NewF) {
5360b57cec5SDimitry Andric NewF->removeFromParent();
5370b57cec5SDimitry Andric OldF->getParent()->getFunctionList().insertAfter(OldF->getIterator(),
5380b57cec5SDimitry Andric NewF);
5390b57cec5SDimitry Andric }
5400b57cec5SDimitry Andric OldF->eraseFromParent();
5410b57cec5SDimitry Andric }
5420b57cec5SDimitry Andric }
5430b57cec5SDimitry Andric
addGlobalValReplacement(llvm::GlobalValue * GV,llvm::Constant * C)5440b57cec5SDimitry Andric void CodeGenModule::addGlobalValReplacement(llvm::GlobalValue *GV, llvm::Constant *C) {
5450b57cec5SDimitry Andric GlobalValReplacements.push_back(std::make_pair(GV, C));
5460b57cec5SDimitry Andric }
5470b57cec5SDimitry Andric
applyGlobalValReplacements()5480b57cec5SDimitry Andric void CodeGenModule::applyGlobalValReplacements() {
5490b57cec5SDimitry Andric for (auto &I : GlobalValReplacements) {
5500b57cec5SDimitry Andric llvm::GlobalValue *GV = I.first;
5510b57cec5SDimitry Andric llvm::Constant *C = I.second;
5520b57cec5SDimitry Andric
5530b57cec5SDimitry Andric GV->replaceAllUsesWith(C);
5540b57cec5SDimitry Andric GV->eraseFromParent();
5550b57cec5SDimitry Andric }
5560b57cec5SDimitry Andric }
5570b57cec5SDimitry Andric
5580b57cec5SDimitry Andric // This is only used in aliases that we created and we know they have a
5590b57cec5SDimitry Andric // linear structure.
getAliasedGlobal(const llvm::GlobalValue * GV)560349cc55cSDimitry Andric static const llvm::GlobalValue *getAliasedGlobal(const llvm::GlobalValue *GV) {
561349cc55cSDimitry Andric const llvm::Constant *C;
562349cc55cSDimitry Andric if (auto *GA = dyn_cast<llvm::GlobalAlias>(GV))
563349cc55cSDimitry Andric C = GA->getAliasee();
564349cc55cSDimitry Andric else if (auto *GI = dyn_cast<llvm::GlobalIFunc>(GV))
565349cc55cSDimitry Andric C = GI->getResolver();
566349cc55cSDimitry Andric else
567349cc55cSDimitry Andric return GV;
568349cc55cSDimitry Andric
569349cc55cSDimitry Andric const auto *AliaseeGV = dyn_cast<llvm::GlobalValue>(C->stripPointerCasts());
570349cc55cSDimitry Andric if (!AliaseeGV)
5710b57cec5SDimitry Andric return nullptr;
572349cc55cSDimitry Andric
573349cc55cSDimitry Andric const llvm::GlobalValue *FinalGV = AliaseeGV->getAliaseeObject();
574349cc55cSDimitry Andric if (FinalGV == GV)
5750b57cec5SDimitry Andric return nullptr;
576349cc55cSDimitry Andric
577349cc55cSDimitry Andric return FinalGV;
5780b57cec5SDimitry Andric }
579349cc55cSDimitry Andric
checkAliasedGlobal(const ASTContext & Context,DiagnosticsEngine & Diags,SourceLocation Location,bool IsIFunc,const llvm::GlobalValue * Alias,const llvm::GlobalValue * & GV,const llvm::MapVector<GlobalDecl,StringRef> & MangledDeclNames,SourceRange AliasRange)58006c3fb27SDimitry Andric static bool checkAliasedGlobal(
5815f757f3fSDimitry Andric const ASTContext &Context, DiagnosticsEngine &Diags, SourceLocation Location,
5825f757f3fSDimitry Andric bool IsIFunc, const llvm::GlobalValue *Alias, const llvm::GlobalValue *&GV,
58306c3fb27SDimitry Andric const llvm::MapVector<GlobalDecl, StringRef> &MangledDeclNames,
58406c3fb27SDimitry Andric SourceRange AliasRange) {
585349cc55cSDimitry Andric GV = getAliasedGlobal(Alias);
586349cc55cSDimitry Andric if (!GV) {
587349cc55cSDimitry Andric Diags.Report(Location, diag::err_cyclic_alias) << IsIFunc;
588349cc55cSDimitry Andric return false;
589349cc55cSDimitry Andric }
590349cc55cSDimitry Andric
5915f757f3fSDimitry Andric if (GV->hasCommonLinkage()) {
5925f757f3fSDimitry Andric const llvm::Triple &Triple = Context.getTargetInfo().getTriple();
5935f757f3fSDimitry Andric if (Triple.getObjectFormat() == llvm::Triple::XCOFF) {
5945f757f3fSDimitry Andric Diags.Report(Location, diag::err_alias_to_common);
5955f757f3fSDimitry Andric return false;
5965f757f3fSDimitry Andric }
5975f757f3fSDimitry Andric }
5985f757f3fSDimitry Andric
599349cc55cSDimitry Andric if (GV->isDeclaration()) {
600349cc55cSDimitry Andric Diags.Report(Location, diag::err_alias_to_undefined) << IsIFunc << IsIFunc;
60106c3fb27SDimitry Andric Diags.Report(Location, diag::note_alias_requires_mangled_name)
60206c3fb27SDimitry Andric << IsIFunc << IsIFunc;
60306c3fb27SDimitry Andric // Provide a note if the given function is not found and exists as a
60406c3fb27SDimitry Andric // mangled name.
60506c3fb27SDimitry Andric for (const auto &[Decl, Name] : MangledDeclNames) {
60606c3fb27SDimitry Andric if (const auto *ND = dyn_cast<NamedDecl>(Decl.getDecl())) {
60706c3fb27SDimitry Andric if (ND->getName() == GV->getName()) {
60806c3fb27SDimitry Andric Diags.Report(Location, diag::note_alias_mangled_name_alternative)
60906c3fb27SDimitry Andric << Name
61006c3fb27SDimitry Andric << FixItHint::CreateReplacement(
61106c3fb27SDimitry Andric AliasRange,
61206c3fb27SDimitry Andric (Twine(IsIFunc ? "ifunc" : "alias") + "(\"" + Name + "\")")
61306c3fb27SDimitry Andric .str());
61406c3fb27SDimitry Andric }
61506c3fb27SDimitry Andric }
61606c3fb27SDimitry Andric }
617349cc55cSDimitry Andric return false;
618349cc55cSDimitry Andric }
619349cc55cSDimitry Andric
620349cc55cSDimitry Andric if (IsIFunc) {
621349cc55cSDimitry Andric // Check resolver function type.
622349cc55cSDimitry Andric const auto *F = dyn_cast<llvm::Function>(GV);
623349cc55cSDimitry Andric if (!F) {
624349cc55cSDimitry Andric Diags.Report(Location, diag::err_alias_to_undefined)
625349cc55cSDimitry Andric << IsIFunc << IsIFunc;
626349cc55cSDimitry Andric return false;
627349cc55cSDimitry Andric }
628349cc55cSDimitry Andric
629349cc55cSDimitry Andric llvm::FunctionType *FTy = F->getFunctionType();
630349cc55cSDimitry Andric if (!FTy->getReturnType()->isPointerTy()) {
631349cc55cSDimitry Andric Diags.Report(Location, diag::err_ifunc_resolver_return);
632349cc55cSDimitry Andric return false;
633349cc55cSDimitry Andric }
634349cc55cSDimitry Andric }
635349cc55cSDimitry Andric
636349cc55cSDimitry Andric return true;
6370b57cec5SDimitry Andric }
6380b57cec5SDimitry Andric
6390fca6ea1SDimitry Andric // Emit a warning if toc-data attribute is requested for global variables that
6400fca6ea1SDimitry Andric // have aliases and remove the toc-data attribute.
checkAliasForTocData(llvm::GlobalVariable * GVar,const CodeGenOptions & CodeGenOpts,DiagnosticsEngine & Diags,SourceLocation Location)6410fca6ea1SDimitry Andric static void checkAliasForTocData(llvm::GlobalVariable *GVar,
6420fca6ea1SDimitry Andric const CodeGenOptions &CodeGenOpts,
6430fca6ea1SDimitry Andric DiagnosticsEngine &Diags,
6440fca6ea1SDimitry Andric SourceLocation Location) {
6450fca6ea1SDimitry Andric if (GVar->hasAttribute("toc-data")) {
6460fca6ea1SDimitry Andric auto GVId = GVar->getName();
6470fca6ea1SDimitry Andric // Is this a global variable specified by the user as local?
6480fca6ea1SDimitry Andric if ((llvm::binary_search(CodeGenOpts.TocDataVarsUserSpecified, GVId))) {
6490fca6ea1SDimitry Andric Diags.Report(Location, diag::warn_toc_unsupported_type)
6500fca6ea1SDimitry Andric << GVId << "the variable has an alias";
6510fca6ea1SDimitry Andric }
6520fca6ea1SDimitry Andric llvm::AttributeSet CurrAttributes = GVar->getAttributes();
6530fca6ea1SDimitry Andric llvm::AttributeSet NewAttributes =
6540fca6ea1SDimitry Andric CurrAttributes.removeAttribute(GVar->getContext(), "toc-data");
6550fca6ea1SDimitry Andric GVar->setAttributes(NewAttributes);
6560fca6ea1SDimitry Andric }
6570fca6ea1SDimitry Andric }
6580fca6ea1SDimitry Andric
checkAliases()6590b57cec5SDimitry Andric void CodeGenModule::checkAliases() {
6600b57cec5SDimitry Andric // Check if the constructed aliases are well formed. It is really unfortunate
6610b57cec5SDimitry Andric // that we have to do this in CodeGen, but we only construct mangled names
6620b57cec5SDimitry Andric // and aliases during codegen.
6630b57cec5SDimitry Andric bool Error = false;
6640b57cec5SDimitry Andric DiagnosticsEngine &Diags = getDiags();
6650b57cec5SDimitry Andric for (const GlobalDecl &GD : Aliases) {
6660b57cec5SDimitry Andric const auto *D = cast<ValueDecl>(GD.getDecl());
6670b57cec5SDimitry Andric SourceLocation Location;
66806c3fb27SDimitry Andric SourceRange Range;
6690b57cec5SDimitry Andric bool IsIFunc = D->hasAttr<IFuncAttr>();
67006c3fb27SDimitry Andric if (const Attr *A = D->getDefiningAttr()) {
6710b57cec5SDimitry Andric Location = A->getLocation();
67206c3fb27SDimitry Andric Range = A->getRange();
67306c3fb27SDimitry Andric } else
6740b57cec5SDimitry Andric llvm_unreachable("Not an alias or ifunc?");
675349cc55cSDimitry Andric
6760b57cec5SDimitry Andric StringRef MangledName = getMangledName(GD);
677349cc55cSDimitry Andric llvm::GlobalValue *Alias = GetGlobalValue(MangledName);
678349cc55cSDimitry Andric const llvm::GlobalValue *GV = nullptr;
6795f757f3fSDimitry Andric if (!checkAliasedGlobal(getContext(), Diags, Location, IsIFunc, Alias, GV,
68006c3fb27SDimitry Andric MangledDeclNames, Range)) {
6810b57cec5SDimitry Andric Error = true;
682349cc55cSDimitry Andric continue;
6830b57cec5SDimitry Andric }
6840b57cec5SDimitry Andric
6850fca6ea1SDimitry Andric if (getContext().getTargetInfo().getTriple().isOSAIX())
6860fca6ea1SDimitry Andric if (const llvm::GlobalVariable *GVar =
6870fca6ea1SDimitry Andric dyn_cast<const llvm::GlobalVariable>(GV))
6880fca6ea1SDimitry Andric checkAliasForTocData(const_cast<llvm::GlobalVariable *>(GVar),
6890fca6ea1SDimitry Andric getCodeGenOpts(), Diags, Location);
6900fca6ea1SDimitry Andric
691349cc55cSDimitry Andric llvm::Constant *Aliasee =
692349cc55cSDimitry Andric IsIFunc ? cast<llvm::GlobalIFunc>(Alias)->getResolver()
693349cc55cSDimitry Andric : cast<llvm::GlobalAlias>(Alias)->getAliasee();
694349cc55cSDimitry Andric
6950b57cec5SDimitry Andric llvm::GlobalValue *AliaseeGV;
6960b57cec5SDimitry Andric if (auto CE = dyn_cast<llvm::ConstantExpr>(Aliasee))
6970b57cec5SDimitry Andric AliaseeGV = cast<llvm::GlobalValue>(CE->getOperand(0));
6980b57cec5SDimitry Andric else
6990b57cec5SDimitry Andric AliaseeGV = cast<llvm::GlobalValue>(Aliasee);
7000b57cec5SDimitry Andric
7010b57cec5SDimitry Andric if (const SectionAttr *SA = D->getAttr<SectionAttr>()) {
7020b57cec5SDimitry Andric StringRef AliasSection = SA->getName();
7030b57cec5SDimitry Andric if (AliasSection != AliaseeGV->getSection())
7040b57cec5SDimitry Andric Diags.Report(SA->getLocation(), diag::warn_alias_with_section)
7050b57cec5SDimitry Andric << AliasSection << IsIFunc << IsIFunc;
7060b57cec5SDimitry Andric }
7070b57cec5SDimitry Andric
7080b57cec5SDimitry Andric // We have to handle alias to weak aliases in here. LLVM itself disallows
7090b57cec5SDimitry Andric // this since the object semantics would not match the IL one. For
7100b57cec5SDimitry Andric // compatibility with gcc we implement it by just pointing the alias
7110b57cec5SDimitry Andric // to its aliasee's aliasee. We also warn, since the user is probably
7120b57cec5SDimitry Andric // expecting the link to be weak.
713349cc55cSDimitry Andric if (auto *GA = dyn_cast<llvm::GlobalAlias>(AliaseeGV)) {
7140b57cec5SDimitry Andric if (GA->isInterposable()) {
7150b57cec5SDimitry Andric Diags.Report(Location, diag::warn_alias_to_weak_alias)
7160b57cec5SDimitry Andric << GV->getName() << GA->getName() << IsIFunc;
7170b57cec5SDimitry Andric Aliasee = llvm::ConstantExpr::getPointerBitCastOrAddrSpaceCast(
718349cc55cSDimitry Andric GA->getAliasee(), Alias->getType());
719349cc55cSDimitry Andric
720349cc55cSDimitry Andric if (IsIFunc)
721349cc55cSDimitry Andric cast<llvm::GlobalIFunc>(Alias)->setResolver(Aliasee);
722349cc55cSDimitry Andric else
723349cc55cSDimitry Andric cast<llvm::GlobalAlias>(Alias)->setAliasee(Aliasee);
7240b57cec5SDimitry Andric }
7250b57cec5SDimitry Andric }
7260fca6ea1SDimitry Andric // ifunc resolvers are usually implemented to run before sanitizer
7270fca6ea1SDimitry Andric // initialization. Disable instrumentation to prevent the ordering issue.
7280fca6ea1SDimitry Andric if (IsIFunc)
7290fca6ea1SDimitry Andric cast<llvm::Function>(Aliasee)->addFnAttr(
7300fca6ea1SDimitry Andric llvm::Attribute::DisableSanitizerInstrumentation);
7310b57cec5SDimitry Andric }
7320b57cec5SDimitry Andric if (!Error)
7330b57cec5SDimitry Andric return;
7340b57cec5SDimitry Andric
7350b57cec5SDimitry Andric for (const GlobalDecl &GD : Aliases) {
7360b57cec5SDimitry Andric StringRef MangledName = getMangledName(GD);
737349cc55cSDimitry Andric llvm::GlobalValue *Alias = GetGlobalValue(MangledName);
7380b57cec5SDimitry Andric Alias->replaceAllUsesWith(llvm::UndefValue::get(Alias->getType()));
7390b57cec5SDimitry Andric Alias->eraseFromParent();
7400b57cec5SDimitry Andric }
7410b57cec5SDimitry Andric }
7420b57cec5SDimitry Andric
clear()7430b57cec5SDimitry Andric void CodeGenModule::clear() {
7440b57cec5SDimitry Andric DeferredDeclsToEmit.clear();
745753f127fSDimitry Andric EmittedDeferredDecls.clear();
7465f757f3fSDimitry Andric DeferredAnnotations.clear();
7470b57cec5SDimitry Andric if (OpenMPRuntime)
7480b57cec5SDimitry Andric OpenMPRuntime->clear();
7490b57cec5SDimitry Andric }
7500b57cec5SDimitry Andric
reportDiagnostics(DiagnosticsEngine & Diags,StringRef MainFile)7510b57cec5SDimitry Andric void InstrProfStats::reportDiagnostics(DiagnosticsEngine &Diags,
7520b57cec5SDimitry Andric StringRef MainFile) {
7530b57cec5SDimitry Andric if (!hasDiagnostics())
7540b57cec5SDimitry Andric return;
7550b57cec5SDimitry Andric if (VisitedInMainFile > 0 && VisitedInMainFile == MissingInMainFile) {
7560b57cec5SDimitry Andric if (MainFile.empty())
7570b57cec5SDimitry Andric MainFile = "<stdin>";
7580b57cec5SDimitry Andric Diags.Report(diag::warn_profile_data_unprofiled) << MainFile;
7590b57cec5SDimitry Andric } else {
7600b57cec5SDimitry Andric if (Mismatched > 0)
7610b57cec5SDimitry Andric Diags.Report(diag::warn_profile_data_out_of_date) << Visited << Mismatched;
7620b57cec5SDimitry Andric
7630b57cec5SDimitry Andric if (Missing > 0)
7640b57cec5SDimitry Andric Diags.Report(diag::warn_profile_data_missing) << Visited << Missing;
7650b57cec5SDimitry Andric }
7660b57cec5SDimitry Andric }
7670b57cec5SDimitry Andric
7687a6dacacSDimitry Andric static std::optional<llvm::GlobalValue::VisibilityTypes>
getLLVMVisibility(clang::LangOptions::VisibilityFromDLLStorageClassKinds K)7697a6dacacSDimitry Andric getLLVMVisibility(clang::LangOptions::VisibilityFromDLLStorageClassKinds K) {
7707a6dacacSDimitry Andric // Map to LLVM visibility.
7717a6dacacSDimitry Andric switch (K) {
7727a6dacacSDimitry Andric case clang::LangOptions::VisibilityFromDLLStorageClassKinds::Keep:
7737a6dacacSDimitry Andric return std::nullopt;
7747a6dacacSDimitry Andric case clang::LangOptions::VisibilityFromDLLStorageClassKinds::Default:
7757a6dacacSDimitry Andric return llvm::GlobalValue::DefaultVisibility;
7767a6dacacSDimitry Andric case clang::LangOptions::VisibilityFromDLLStorageClassKinds::Hidden:
7777a6dacacSDimitry Andric return llvm::GlobalValue::HiddenVisibility;
7787a6dacacSDimitry Andric case clang::LangOptions::VisibilityFromDLLStorageClassKinds::Protected:
7797a6dacacSDimitry Andric return llvm::GlobalValue::ProtectedVisibility;
7807a6dacacSDimitry Andric }
7817a6dacacSDimitry Andric llvm_unreachable("unknown option value!");
7827a6dacacSDimitry Andric }
7837a6dacacSDimitry Andric
setLLVMVisibility(llvm::GlobalValue & GV,std::optional<llvm::GlobalValue::VisibilityTypes> V)7847a6dacacSDimitry Andric void setLLVMVisibility(llvm::GlobalValue &GV,
7857a6dacacSDimitry Andric std::optional<llvm::GlobalValue::VisibilityTypes> V) {
7867a6dacacSDimitry Andric if (!V)
787e8d8bef9SDimitry Andric return;
788e8d8bef9SDimitry Andric
789e8d8bef9SDimitry Andric // Reset DSO locality before setting the visibility. This removes
790e8d8bef9SDimitry Andric // any effects that visibility options and annotations may have
791e8d8bef9SDimitry Andric // had on the DSO locality. Setting the visibility will implicitly set
792e8d8bef9SDimitry Andric // appropriate globals to DSO Local; however, this will be pessimistic
793e8d8bef9SDimitry Andric // w.r.t. to the normal compiler IRGen.
794e8d8bef9SDimitry Andric GV.setDSOLocal(false);
7957a6dacacSDimitry Andric GV.setVisibility(*V);
7967a6dacacSDimitry Andric }
797e8d8bef9SDimitry Andric
setVisibilityFromDLLStorageClass(const clang::LangOptions & LO,llvm::Module & M)7987a6dacacSDimitry Andric static void setVisibilityFromDLLStorageClass(const clang::LangOptions &LO,
7997a6dacacSDimitry Andric llvm::Module &M) {
8007a6dacacSDimitry Andric if (!LO.VisibilityFromDLLStorageClass)
8017a6dacacSDimitry Andric return;
8027a6dacacSDimitry Andric
8037a6dacacSDimitry Andric std::optional<llvm::GlobalValue::VisibilityTypes> DLLExportVisibility =
8047a6dacacSDimitry Andric getLLVMVisibility(LO.getDLLExportVisibility());
8057a6dacacSDimitry Andric
8067a6dacacSDimitry Andric std::optional<llvm::GlobalValue::VisibilityTypes>
8077a6dacacSDimitry Andric NoDLLStorageClassVisibility =
8087a6dacacSDimitry Andric getLLVMVisibility(LO.getNoDLLStorageClassVisibility());
8097a6dacacSDimitry Andric
8107a6dacacSDimitry Andric std::optional<llvm::GlobalValue::VisibilityTypes>
8117a6dacacSDimitry Andric ExternDeclDLLImportVisibility =
8127a6dacacSDimitry Andric getLLVMVisibility(LO.getExternDeclDLLImportVisibility());
8137a6dacacSDimitry Andric
8147a6dacacSDimitry Andric std::optional<llvm::GlobalValue::VisibilityTypes>
8157a6dacacSDimitry Andric ExternDeclNoDLLStorageClassVisibility =
8167a6dacacSDimitry Andric getLLVMVisibility(LO.getExternDeclNoDLLStorageClassVisibility());
8177a6dacacSDimitry Andric
8187a6dacacSDimitry Andric for (llvm::GlobalValue &GV : M.global_values()) {
8197a6dacacSDimitry Andric if (GV.hasAppendingLinkage() || GV.hasLocalLinkage())
8207a6dacacSDimitry Andric continue;
8217a6dacacSDimitry Andric
8227a6dacacSDimitry Andric if (GV.isDeclarationForLinker())
8237a6dacacSDimitry Andric setLLVMVisibility(GV, GV.getDLLStorageClass() ==
824e8d8bef9SDimitry Andric llvm::GlobalValue::DLLImportStorageClass
825e8d8bef9SDimitry Andric ? ExternDeclDLLImportVisibility
826e8d8bef9SDimitry Andric : ExternDeclNoDLLStorageClassVisibility);
8277a6dacacSDimitry Andric else
8287a6dacacSDimitry Andric setLLVMVisibility(GV, GV.getDLLStorageClass() ==
829e8d8bef9SDimitry Andric llvm::GlobalValue::DLLExportStorageClass
830e8d8bef9SDimitry Andric ? DLLExportVisibility
831e8d8bef9SDimitry Andric : NoDLLStorageClassVisibility);
832e8d8bef9SDimitry Andric
833e8d8bef9SDimitry Andric GV.setDLLStorageClass(llvm::GlobalValue::DefaultStorageClass);
834e8d8bef9SDimitry Andric }
835e8d8bef9SDimitry Andric }
836e8d8bef9SDimitry Andric
isStackProtectorOn(const LangOptions & LangOpts,const llvm::Triple & Triple,clang::LangOptions::StackProtectorMode Mode)8375f757f3fSDimitry Andric static bool isStackProtectorOn(const LangOptions &LangOpts,
8385f757f3fSDimitry Andric const llvm::Triple &Triple,
8395f757f3fSDimitry Andric clang::LangOptions::StackProtectorMode Mode) {
8405f757f3fSDimitry Andric if (Triple.isAMDGPU() || Triple.isNVPTX())
8415f757f3fSDimitry Andric return false;
8425f757f3fSDimitry Andric return LangOpts.getStackProtector() == Mode;
8435f757f3fSDimitry Andric }
8445f757f3fSDimitry Andric
Release()8450b57cec5SDimitry Andric void CodeGenModule::Release() {
84606c3fb27SDimitry Andric Module *Primary = getContext().getCurrentNamedModule();
84761cfbce3SDimitry Andric if (CXX20ModuleInits && Primary && !Primary->isHeaderLikeModule())
848fcaf7f86SDimitry Andric EmitModuleInitializers(Primary);
8490b57cec5SDimitry Andric EmitDeferred();
850753f127fSDimitry Andric DeferredDecls.insert(EmittedDeferredDecls.begin(),
851753f127fSDimitry Andric EmittedDeferredDecls.end());
852753f127fSDimitry Andric EmittedDeferredDecls.clear();
8530b57cec5SDimitry Andric EmitVTablesOpportunistically();
8540b57cec5SDimitry Andric applyGlobalValReplacements();
8550b57cec5SDimitry Andric applyReplacements();
8560b57cec5SDimitry Andric emitMultiVersionFunctions();
857bdd1243dSDimitry Andric
858bdd1243dSDimitry Andric if (Context.getLangOpts().IncrementalExtensions &&
859bdd1243dSDimitry Andric GlobalTopLevelStmtBlockInFlight.first) {
860bdd1243dSDimitry Andric const TopLevelStmtDecl *TLSD = GlobalTopLevelStmtBlockInFlight.second;
861bdd1243dSDimitry Andric GlobalTopLevelStmtBlockInFlight.first->FinishFunction(TLSD->getEndLoc());
862bdd1243dSDimitry Andric GlobalTopLevelStmtBlockInFlight = {nullptr, nullptr};
863bdd1243dSDimitry Andric }
864bdd1243dSDimitry Andric
86506c3fb27SDimitry Andric // Module implementations are initialized the same way as a regular TU that
86606c3fb27SDimitry Andric // imports one or more modules.
867fcaf7f86SDimitry Andric if (CXX20ModuleInits && Primary && Primary->isInterfaceOrPartition())
868fcaf7f86SDimitry Andric EmitCXXModuleInitFunc(Primary);
869fcaf7f86SDimitry Andric else
8700b57cec5SDimitry Andric EmitCXXGlobalInitFunc();
8715ffd83dbSDimitry Andric EmitCXXGlobalCleanUpFunc();
8720b57cec5SDimitry Andric registerGlobalDtorsWithAtExit();
8730b57cec5SDimitry Andric EmitCXXThreadLocalInitFunc();
8740b57cec5SDimitry Andric if (ObjCRuntime)
8750b57cec5SDimitry Andric if (llvm::Function *ObjCInitFunction = ObjCRuntime->ModuleInitFunction())
8760b57cec5SDimitry Andric AddGlobalCtor(ObjCInitFunction);
877fe6060f1SDimitry Andric if (Context.getLangOpts().CUDA && CUDARuntime) {
878fe6060f1SDimitry Andric if (llvm::Function *CudaCtorFunction = CUDARuntime->finalizeModule())
8790b57cec5SDimitry Andric AddGlobalCtor(CudaCtorFunction);
8800b57cec5SDimitry Andric }
8810b57cec5SDimitry Andric if (OpenMPRuntime) {
882a7dea167SDimitry Andric OpenMPRuntime->createOffloadEntriesAndInfoMetadata();
8830b57cec5SDimitry Andric OpenMPRuntime->clear();
8840b57cec5SDimitry Andric }
8850b57cec5SDimitry Andric if (PGOReader) {
8860b57cec5SDimitry Andric getModule().setProfileSummary(
8870b57cec5SDimitry Andric PGOReader->getSummary(/* UseCS */ false).getMD(VMContext),
8880b57cec5SDimitry Andric llvm::ProfileSummary::PSK_Instr);
8890b57cec5SDimitry Andric if (PGOStats.hasDiagnostics())
8900b57cec5SDimitry Andric PGOStats.reportDiagnostics(getDiags(), getCodeGenOpts().MainFileName);
8910b57cec5SDimitry Andric }
892bdd1243dSDimitry Andric llvm::stable_sort(GlobalCtors, [](const Structor &L, const Structor &R) {
893bdd1243dSDimitry Andric return L.LexOrder < R.LexOrder;
894bdd1243dSDimitry Andric });
8950b57cec5SDimitry Andric EmitCtorList(GlobalCtors, "llvm.global_ctors");
8960b57cec5SDimitry Andric EmitCtorList(GlobalDtors, "llvm.global_dtors");
8970b57cec5SDimitry Andric EmitGlobalAnnotations();
8980b57cec5SDimitry Andric EmitStaticExternCAliases();
89981ad6265SDimitry Andric checkAliases();
9000b57cec5SDimitry Andric EmitDeferredUnusedCoverageMappings();
901fe6060f1SDimitry Andric CodeGenPGO(*this).setValueProfilingFlag(getModule());
9020fca6ea1SDimitry Andric CodeGenPGO(*this).setProfileVersion(getModule());
9030b57cec5SDimitry Andric if (CoverageMapping)
9040b57cec5SDimitry Andric CoverageMapping->emit();
9050b57cec5SDimitry Andric if (CodeGenOpts.SanitizeCfiCrossDso) {
9060b57cec5SDimitry Andric CodeGenFunction(*this).EmitCfiCheckFail();
9070b57cec5SDimitry Andric CodeGenFunction(*this).EmitCfiCheckStub();
9080b57cec5SDimitry Andric }
909bdd1243dSDimitry Andric if (LangOpts.Sanitize.has(SanitizerKind::KCFI))
910bdd1243dSDimitry Andric finalizeKCFITypes();
9110b57cec5SDimitry Andric emitAtAvailableLinkGuard();
91281ad6265SDimitry Andric if (Context.getTargetInfo().getTriple().isWasm())
9135ffd83dbSDimitry Andric EmitMainVoidAlias();
914fe6060f1SDimitry Andric
9150fca6ea1SDimitry Andric if (getTriple().isAMDGPU() ||
9160fca6ea1SDimitry Andric (getTriple().isSPIRV() && getTriple().getVendor() == llvm::Triple::AMD)) {
9170fca6ea1SDimitry Andric // Emit amdhsa_code_object_version module flag, which is code object version
91881ad6265SDimitry Andric // times 100.
919bdd1243dSDimitry Andric if (getTarget().getTargetOpts().CodeObjectVersion !=
9205f757f3fSDimitry Andric llvm::CodeObjectVersionKind::COV_None) {
92181ad6265SDimitry Andric getModule().addModuleFlag(llvm::Module::Error,
9220fca6ea1SDimitry Andric "amdhsa_code_object_version",
92381ad6265SDimitry Andric getTarget().getTargetOpts().CodeObjectVersion);
92481ad6265SDimitry Andric }
92506c3fb27SDimitry Andric
92606c3fb27SDimitry Andric // Currently, "-mprintf-kind" option is only supported for HIP
92706c3fb27SDimitry Andric if (LangOpts.HIP) {
92806c3fb27SDimitry Andric auto *MDStr = llvm::MDString::get(
92906c3fb27SDimitry Andric getLLVMContext(), (getTarget().getTargetOpts().AMDGPUPrintfKindVal ==
93006c3fb27SDimitry Andric TargetOptions::AMDGPUPrintfKind::Hostcall)
93106c3fb27SDimitry Andric ? "hostcall"
93206c3fb27SDimitry Andric : "buffered");
93306c3fb27SDimitry Andric getModule().addModuleFlag(llvm::Module::Error, "amdgpu_printf_kind",
93406c3fb27SDimitry Andric MDStr);
93506c3fb27SDimitry Andric }
93681ad6265SDimitry Andric }
93781ad6265SDimitry Andric
93881ad6265SDimitry Andric // Emit a global array containing all external kernels or device variables
93981ad6265SDimitry Andric // used by host functions and mark it as used for CUDA/HIP. This is necessary
94081ad6265SDimitry Andric // to get kernels or device variables in archives linked in even if these
94181ad6265SDimitry Andric // kernels or device variables are only used in host functions.
94281ad6265SDimitry Andric if (!Context.CUDAExternalDeviceDeclODRUsedByHost.empty()) {
94381ad6265SDimitry Andric SmallVector<llvm::Constant *, 8> UsedArray;
94481ad6265SDimitry Andric for (auto D : Context.CUDAExternalDeviceDeclODRUsedByHost) {
94581ad6265SDimitry Andric GlobalDecl GD;
94681ad6265SDimitry Andric if (auto *FD = dyn_cast<FunctionDecl>(D))
94781ad6265SDimitry Andric GD = GlobalDecl(FD, KernelReferenceKind::Kernel);
94881ad6265SDimitry Andric else
94981ad6265SDimitry Andric GD = GlobalDecl(D);
95081ad6265SDimitry Andric UsedArray.push_back(llvm::ConstantExpr::getPointerBitCastOrAddrSpaceCast(
95181ad6265SDimitry Andric GetAddrOfGlobal(GD), Int8PtrTy));
95281ad6265SDimitry Andric }
95381ad6265SDimitry Andric
95481ad6265SDimitry Andric llvm::ArrayType *ATy = llvm::ArrayType::get(Int8PtrTy, UsedArray.size());
95581ad6265SDimitry Andric
95681ad6265SDimitry Andric auto *GV = new llvm::GlobalVariable(
95781ad6265SDimitry Andric getModule(), ATy, false, llvm::GlobalValue::InternalLinkage,
95881ad6265SDimitry Andric llvm::ConstantArray::get(ATy, UsedArray), "__clang_gpu_used_external");
95981ad6265SDimitry Andric addCompilerUsedGlobal(GV);
96004eeddc0SDimitry Andric }
9610fca6ea1SDimitry Andric if (LangOpts.HIP && !getLangOpts().OffloadingNewDriver) {
9620fca6ea1SDimitry Andric // Emit a unique ID so that host and device binaries from the same
9630fca6ea1SDimitry Andric // compilation unit can be associated.
9640fca6ea1SDimitry Andric auto *GV = new llvm::GlobalVariable(
9650fca6ea1SDimitry Andric getModule(), Int8Ty, false, llvm::GlobalValue::ExternalLinkage,
9660fca6ea1SDimitry Andric llvm::Constant::getNullValue(Int8Ty),
9670fca6ea1SDimitry Andric "__hip_cuid_" + getContext().getCUIDHash());
9680fca6ea1SDimitry Andric addCompilerUsedGlobal(GV);
9690fca6ea1SDimitry Andric }
9700b57cec5SDimitry Andric emitLLVMUsed();
9710b57cec5SDimitry Andric if (SanStats)
9720b57cec5SDimitry Andric SanStats->finish();
9730b57cec5SDimitry Andric
9740b57cec5SDimitry Andric if (CodeGenOpts.Autolink &&
9750b57cec5SDimitry Andric (Context.getLangOpts().Modules || !LinkerOptionsMetadata.empty())) {
9760b57cec5SDimitry Andric EmitModuleLinkOptions();
9770b57cec5SDimitry Andric }
9780b57cec5SDimitry Andric
9790b57cec5SDimitry Andric // On ELF we pass the dependent library specifiers directly to the linker
9800b57cec5SDimitry Andric // without manipulating them. This is in contrast to other platforms where
9810b57cec5SDimitry Andric // they are mapped to a specific linker option by the compiler. This
9820b57cec5SDimitry Andric // difference is a result of the greater variety of ELF linkers and the fact
9830b57cec5SDimitry Andric // that ELF linkers tend to handle libraries in a more complicated fashion
9840b57cec5SDimitry Andric // than on other platforms. This forces us to defer handling the dependent
9850b57cec5SDimitry Andric // libs to the linker.
9860b57cec5SDimitry Andric //
9870b57cec5SDimitry Andric // CUDA/HIP device and host libraries are different. Currently there is no
9880b57cec5SDimitry Andric // way to differentiate dependent libraries for host or device. Existing
9890b57cec5SDimitry Andric // usage of #pragma comment(lib, *) is intended for host libraries on
9900b57cec5SDimitry Andric // Windows. Therefore emit llvm.dependent-libraries only for host.
9910b57cec5SDimitry Andric if (!ELFDependentLibraries.empty() && !Context.getLangOpts().CUDAIsDevice) {
9920b57cec5SDimitry Andric auto *NMD = getModule().getOrInsertNamedMetadata("llvm.dependent-libraries");
9930b57cec5SDimitry Andric for (auto *MD : ELFDependentLibraries)
9940b57cec5SDimitry Andric NMD->addOperand(MD);
9950b57cec5SDimitry Andric }
9960b57cec5SDimitry Andric
9970b57cec5SDimitry Andric if (CodeGenOpts.DwarfVersion) {
998480093f4SDimitry Andric getModule().addModuleFlag(llvm::Module::Max, "Dwarf Version",
9990b57cec5SDimitry Andric CodeGenOpts.DwarfVersion);
10000b57cec5SDimitry Andric }
10015ffd83dbSDimitry Andric
1002fe6060f1SDimitry Andric if (CodeGenOpts.Dwarf64)
1003fe6060f1SDimitry Andric getModule().addModuleFlag(llvm::Module::Max, "DWARF64", 1);
1004fe6060f1SDimitry Andric
10055ffd83dbSDimitry Andric if (Context.getLangOpts().SemanticInterposition)
10065ffd83dbSDimitry Andric // Require various optimization to respect semantic interposition.
100704eeddc0SDimitry Andric getModule().setSemanticInterposition(true);
10085ffd83dbSDimitry Andric
10090b57cec5SDimitry Andric if (CodeGenOpts.EmitCodeView) {
10100b57cec5SDimitry Andric // Indicate that we want CodeView in the metadata.
10110b57cec5SDimitry Andric getModule().addModuleFlag(llvm::Module::Warning, "CodeView", 1);
10120b57cec5SDimitry Andric }
10130b57cec5SDimitry Andric if (CodeGenOpts.CodeViewGHash) {
10140b57cec5SDimitry Andric getModule().addModuleFlag(llvm::Module::Warning, "CodeViewGHash", 1);
10150b57cec5SDimitry Andric }
10160b57cec5SDimitry Andric if (CodeGenOpts.ControlFlowGuard) {
1017480093f4SDimitry Andric // Function ID tables and checks for Control Flow Guard (cfguard=2).
1018480093f4SDimitry Andric getModule().addModuleFlag(llvm::Module::Warning, "cfguard", 2);
1019480093f4SDimitry Andric } else if (CodeGenOpts.ControlFlowGuardNoChecks) {
1020480093f4SDimitry Andric // Function ID tables for Control Flow Guard (cfguard=1).
1021480093f4SDimitry Andric getModule().addModuleFlag(llvm::Module::Warning, "cfguard", 1);
10220b57cec5SDimitry Andric }
1023fe6060f1SDimitry Andric if (CodeGenOpts.EHContGuard) {
1024fe6060f1SDimitry Andric // Function ID tables for EH Continuation Guard.
1025fe6060f1SDimitry Andric getModule().addModuleFlag(llvm::Module::Warning, "ehcontguard", 1);
1026fe6060f1SDimitry Andric }
1027bdd1243dSDimitry Andric if (Context.getLangOpts().Kernel) {
1028bdd1243dSDimitry Andric // Note if we are compiling with /kernel.
1029bdd1243dSDimitry Andric getModule().addModuleFlag(llvm::Module::Warning, "ms-kernel", 1);
1030bdd1243dSDimitry Andric }
10310b57cec5SDimitry Andric if (CodeGenOpts.OptimizationLevel > 0 && CodeGenOpts.StrictVTablePointers) {
10320b57cec5SDimitry Andric // We don't support LTO with 2 with different StrictVTablePointers
10330b57cec5SDimitry Andric // FIXME: we could support it by stripping all the information introduced
10340b57cec5SDimitry Andric // by StrictVTablePointers.
10350b57cec5SDimitry Andric
10360b57cec5SDimitry Andric getModule().addModuleFlag(llvm::Module::Error, "StrictVTablePointers",1);
10370b57cec5SDimitry Andric
10380b57cec5SDimitry Andric llvm::Metadata *Ops[2] = {
10390b57cec5SDimitry Andric llvm::MDString::get(VMContext, "StrictVTablePointers"),
10400b57cec5SDimitry Andric llvm::ConstantAsMetadata::get(llvm::ConstantInt::get(
10410b57cec5SDimitry Andric llvm::Type::getInt32Ty(VMContext), 1))};
10420b57cec5SDimitry Andric
10430b57cec5SDimitry Andric getModule().addModuleFlag(llvm::Module::Require,
10440b57cec5SDimitry Andric "StrictVTablePointersRequirement",
10450b57cec5SDimitry Andric llvm::MDNode::get(VMContext, Ops));
10460b57cec5SDimitry Andric }
10475ffd83dbSDimitry Andric if (getModuleDebugInfo())
10480b57cec5SDimitry Andric // We support a single version in the linked module. The LLVM
10490b57cec5SDimitry Andric // parser will drop debug info with a different version number
10500b57cec5SDimitry Andric // (and warn about it, too).
10510b57cec5SDimitry Andric getModule().addModuleFlag(llvm::Module::Warning, "Debug Info Version",
10520b57cec5SDimitry Andric llvm::DEBUG_METADATA_VERSION);
10530b57cec5SDimitry Andric
10540b57cec5SDimitry Andric // We need to record the widths of enums and wchar_t, so that we can generate
10550b57cec5SDimitry Andric // the correct build attributes in the ARM backend. wchar_size is also used by
10560b57cec5SDimitry Andric // TargetLibraryInfo.
10570b57cec5SDimitry Andric uint64_t WCharWidth =
10580b57cec5SDimitry Andric Context.getTypeSizeInChars(Context.getWideCharType()).getQuantity();
10590b57cec5SDimitry Andric getModule().addModuleFlag(llvm::Module::Error, "wchar_size", WCharWidth);
10600b57cec5SDimitry Andric
10615f757f3fSDimitry Andric if (getTriple().isOSzOS()) {
10625f757f3fSDimitry Andric getModule().addModuleFlag(llvm::Module::Warning,
10635f757f3fSDimitry Andric "zos_product_major_version",
10645f757f3fSDimitry Andric uint32_t(CLANG_VERSION_MAJOR));
10655f757f3fSDimitry Andric getModule().addModuleFlag(llvm::Module::Warning,
10665f757f3fSDimitry Andric "zos_product_minor_version",
10675f757f3fSDimitry Andric uint32_t(CLANG_VERSION_MINOR));
10685f757f3fSDimitry Andric getModule().addModuleFlag(llvm::Module::Warning, "zos_product_patchlevel",
10695f757f3fSDimitry Andric uint32_t(CLANG_VERSION_PATCHLEVEL));
1070cb14a3feSDimitry Andric std::string ProductId = getClangVendor() + "clang";
10715f757f3fSDimitry Andric getModule().addModuleFlag(llvm::Module::Error, "zos_product_id",
10725f757f3fSDimitry Andric llvm::MDString::get(VMContext, ProductId));
10735f757f3fSDimitry Andric
10745f757f3fSDimitry Andric // Record the language because we need it for the PPA2.
10755f757f3fSDimitry Andric StringRef lang_str = languageToString(
10765f757f3fSDimitry Andric LangStandard::getLangStandardForKind(LangOpts.LangStd).Language);
10775f757f3fSDimitry Andric getModule().addModuleFlag(llvm::Module::Error, "zos_cu_language",
10785f757f3fSDimitry Andric llvm::MDString::get(VMContext, lang_str));
10795f757f3fSDimitry Andric
10805f757f3fSDimitry Andric time_t TT = PreprocessorOpts.SourceDateEpoch
10815f757f3fSDimitry Andric ? *PreprocessorOpts.SourceDateEpoch
10825f757f3fSDimitry Andric : std::time(nullptr);
10835f757f3fSDimitry Andric getModule().addModuleFlag(llvm::Module::Max, "zos_translation_time",
10845f757f3fSDimitry Andric static_cast<uint64_t>(TT));
10855f757f3fSDimitry Andric
10865f757f3fSDimitry Andric // Multiple modes will be supported here.
10875f757f3fSDimitry Andric getModule().addModuleFlag(llvm::Module::Error, "zos_le_char_mode",
10885f757f3fSDimitry Andric llvm::MDString::get(VMContext, "ascii"));
10895f757f3fSDimitry Andric }
10905f757f3fSDimitry Andric
10910fca6ea1SDimitry Andric llvm::Triple T = Context.getTargetInfo().getTriple();
10920fca6ea1SDimitry Andric if (T.isARM() || T.isThumb()) {
10930b57cec5SDimitry Andric // The minimum width of an enum in bytes
10940b57cec5SDimitry Andric uint64_t EnumWidth = Context.getLangOpts().ShortEnums ? 1 : 4;
10950b57cec5SDimitry Andric getModule().addModuleFlag(llvm::Module::Error, "min_enum_size", EnumWidth);
10960b57cec5SDimitry Andric }
10970b57cec5SDimitry Andric
10980fca6ea1SDimitry Andric if (T.isRISCV()) {
109913138422SDimitry Andric StringRef ABIStr = Target.getABI();
110013138422SDimitry Andric llvm::LLVMContext &Ctx = TheModule.getContext();
110113138422SDimitry Andric getModule().addModuleFlag(llvm::Module::Error, "target-abi",
110213138422SDimitry Andric llvm::MDString::get(Ctx, ABIStr));
11033a079333SDimitry Andric
11043a079333SDimitry Andric // Add the canonical ISA string as metadata so the backend can set the ELF
11053a079333SDimitry Andric // attributes correctly. We use AppendUnique so LTO will keep all of the
11063a079333SDimitry Andric // unique ISA strings that were linked together.
11073a079333SDimitry Andric const std::vector<std::string> &Features =
11083a079333SDimitry Andric getTarget().getTargetOpts().Features;
11090fca6ea1SDimitry Andric auto ParseResult =
11100fca6ea1SDimitry Andric llvm::RISCVISAInfo::parseFeatures(T.isRISCV64() ? 64 : 32, Features);
11113a079333SDimitry Andric if (!errorToBool(ParseResult.takeError()))
11123a079333SDimitry Andric getModule().addModuleFlag(
11133a079333SDimitry Andric llvm::Module::AppendUnique, "riscv-isa",
11143a079333SDimitry Andric llvm::MDNode::get(
11153a079333SDimitry Andric Ctx, llvm::MDString::get(Ctx, (*ParseResult)->toString())));
111613138422SDimitry Andric }
111713138422SDimitry Andric
11180b57cec5SDimitry Andric if (CodeGenOpts.SanitizeCfiCrossDso) {
11190b57cec5SDimitry Andric // Indicate that we want cross-DSO control flow integrity checks.
11200b57cec5SDimitry Andric getModule().addModuleFlag(llvm::Module::Override, "Cross-DSO CFI", 1);
11210b57cec5SDimitry Andric }
11220b57cec5SDimitry Andric
11235ffd83dbSDimitry Andric if (CodeGenOpts.WholeProgramVTables) {
11245ffd83dbSDimitry Andric // Indicate whether VFE was enabled for this module, so that the
11255ffd83dbSDimitry Andric // vcall_visibility metadata added under whole program vtables is handled
11265ffd83dbSDimitry Andric // appropriately in the optimizer.
11275ffd83dbSDimitry Andric getModule().addModuleFlag(llvm::Module::Error, "Virtual Function Elim",
11285ffd83dbSDimitry Andric CodeGenOpts.VirtualFunctionElimination);
11295ffd83dbSDimitry Andric }
11305ffd83dbSDimitry Andric
1131a7dea167SDimitry Andric if (LangOpts.Sanitize.has(SanitizerKind::CFIICall)) {
1132a7dea167SDimitry Andric getModule().addModuleFlag(llvm::Module::Override,
1133a7dea167SDimitry Andric "CFI Canonical Jump Tables",
1134a7dea167SDimitry Andric CodeGenOpts.SanitizeCfiCanonicalJumpTables);
1135a7dea167SDimitry Andric }
1136a7dea167SDimitry Andric
1137d686ce93SDimitry Andric if (CodeGenOpts.SanitizeCfiICallNormalizeIntegers) {
1138d686ce93SDimitry Andric getModule().addModuleFlag(llvm::Module::Override, "cfi-normalize-integers",
1139d686ce93SDimitry Andric 1);
1140d686ce93SDimitry Andric }
1141d686ce93SDimitry Andric
1142bdd1243dSDimitry Andric if (LangOpts.Sanitize.has(SanitizerKind::KCFI)) {
1143bdd1243dSDimitry Andric getModule().addModuleFlag(llvm::Module::Override, "kcfi", 1);
1144bdd1243dSDimitry Andric // KCFI assumes patchable-function-prefix is the same for all indirectly
1145bdd1243dSDimitry Andric // called functions. Store the expected offset for code generation.
1146bdd1243dSDimitry Andric if (CodeGenOpts.PatchableFunctionEntryOffset)
1147bdd1243dSDimitry Andric getModule().addModuleFlag(llvm::Module::Override, "kcfi-offset",
1148bdd1243dSDimitry Andric CodeGenOpts.PatchableFunctionEntryOffset);
1149bdd1243dSDimitry Andric }
1150bdd1243dSDimitry Andric
11510b57cec5SDimitry Andric if (CodeGenOpts.CFProtectionReturn &&
11520b57cec5SDimitry Andric Target.checkCFProtectionReturnSupported(getDiags())) {
11530b57cec5SDimitry Andric // Indicate that we want to instrument return control flow protection.
1154fcaf7f86SDimitry Andric getModule().addModuleFlag(llvm::Module::Min, "cf-protection-return",
11550b57cec5SDimitry Andric 1);
11560b57cec5SDimitry Andric }
11570b57cec5SDimitry Andric
11580b57cec5SDimitry Andric if (CodeGenOpts.CFProtectionBranch &&
11590b57cec5SDimitry Andric Target.checkCFProtectionBranchSupported(getDiags())) {
11600b57cec5SDimitry Andric // Indicate that we want to instrument branch control flow protection.
1161fcaf7f86SDimitry Andric getModule().addModuleFlag(llvm::Module::Min, "cf-protection-branch",
11620b57cec5SDimitry Andric 1);
11630b57cec5SDimitry Andric }
11640b57cec5SDimitry Andric
1165fcaf7f86SDimitry Andric if (CodeGenOpts.FunctionReturnThunks)
1166fcaf7f86SDimitry Andric getModule().addModuleFlag(llvm::Module::Override, "function_return_thunk_extern", 1);
116704eeddc0SDimitry Andric
1168bdd1243dSDimitry Andric if (CodeGenOpts.IndirectBranchCSPrefix)
1169bdd1243dSDimitry Andric getModule().addModuleFlag(llvm::Module::Override, "indirect_branch_cs_prefix", 1);
1170bdd1243dSDimitry Andric
11714824e7fdSDimitry Andric // Add module metadata for return address signing (ignoring
11724824e7fdSDimitry Andric // non-leaf/all) and stack tagging. These are actually turned on by function
11734824e7fdSDimitry Andric // attributes, but we use module metadata to emit build attributes. This is
11744824e7fdSDimitry Andric // needed for LTO, where the function attributes are inside bitcode
11754824e7fdSDimitry Andric // serialised into a global variable by the time build attributes are
117681ad6265SDimitry Andric // emitted, so we can't access them. LTO objects could be compiled with
117781ad6265SDimitry Andric // different flags therefore module flags are set to "Min" behavior to achieve
117881ad6265SDimitry Andric // the same end result of the normal build where e.g BTI is off if any object
117981ad6265SDimitry Andric // doesn't support it.
11804824e7fdSDimitry Andric if (Context.getTargetInfo().hasFeature("ptrauth") &&
11814824e7fdSDimitry Andric LangOpts.getSignReturnAddressScope() !=
11824824e7fdSDimitry Andric LangOptions::SignReturnAddressScopeKind::None)
11834824e7fdSDimitry Andric getModule().addModuleFlag(llvm::Module::Override,
11844824e7fdSDimitry Andric "sign-return-address-buildattr", 1);
118581ad6265SDimitry Andric if (LangOpts.Sanitize.has(SanitizerKind::MemtagStack))
11864824e7fdSDimitry Andric getModule().addModuleFlag(llvm::Module::Override,
11874824e7fdSDimitry Andric "tag-stack-memory-buildattr", 1);
11884824e7fdSDimitry Andric
11890fca6ea1SDimitry Andric if (T.isARM() || T.isThumb() || T.isAArch64()) {
1190972a253aSDimitry Andric if (LangOpts.BranchTargetEnforcement)
119181ad6265SDimitry Andric getModule().addModuleFlag(llvm::Module::Min, "branch-target-enforcement",
1192972a253aSDimitry Andric 1);
1193cb14a3feSDimitry Andric if (LangOpts.BranchProtectionPAuthLR)
1194cb14a3feSDimitry Andric getModule().addModuleFlag(llvm::Module::Min, "branch-protection-pauth-lr",
1195cb14a3feSDimitry Andric 1);
1196297eecfbSDimitry Andric if (LangOpts.GuardedControlStack)
1197297eecfbSDimitry Andric getModule().addModuleFlag(llvm::Module::Min, "guarded-control-stack", 1);
1198972a253aSDimitry Andric if (LangOpts.hasSignReturnAddress())
1199972a253aSDimitry Andric getModule().addModuleFlag(llvm::Module::Min, "sign-return-address", 1);
1200972a253aSDimitry Andric if (LangOpts.isSignReturnAddressScopeAll())
120181ad6265SDimitry Andric getModule().addModuleFlag(llvm::Module::Min, "sign-return-address-all",
1202972a253aSDimitry Andric 1);
1203972a253aSDimitry Andric if (!LangOpts.isSignReturnAddressWithAKey())
120481ad6265SDimitry Andric getModule().addModuleFlag(llvm::Module::Min,
1205972a253aSDimitry Andric "sign-return-address-with-bkey", 1);
12060fca6ea1SDimitry Andric
12070fca6ea1SDimitry Andric if (getTriple().isOSLinux()) {
12080fca6ea1SDimitry Andric assert(getTriple().isOSBinFormatELF());
12090fca6ea1SDimitry Andric using namespace llvm::ELF;
12100fca6ea1SDimitry Andric uint64_t PAuthABIVersion =
12110fca6ea1SDimitry Andric (LangOpts.PointerAuthIntrinsics
12120fca6ea1SDimitry Andric << AARCH64_PAUTH_PLATFORM_LLVM_LINUX_VERSION_INTRINSICS) |
12130fca6ea1SDimitry Andric (LangOpts.PointerAuthCalls
12140fca6ea1SDimitry Andric << AARCH64_PAUTH_PLATFORM_LLVM_LINUX_VERSION_CALLS) |
12150fca6ea1SDimitry Andric (LangOpts.PointerAuthReturns
12160fca6ea1SDimitry Andric << AARCH64_PAUTH_PLATFORM_LLVM_LINUX_VERSION_RETURNS) |
12170fca6ea1SDimitry Andric (LangOpts.PointerAuthAuthTraps
12180fca6ea1SDimitry Andric << AARCH64_PAUTH_PLATFORM_LLVM_LINUX_VERSION_AUTHTRAPS) |
12190fca6ea1SDimitry Andric (LangOpts.PointerAuthVTPtrAddressDiscrimination
12200fca6ea1SDimitry Andric << AARCH64_PAUTH_PLATFORM_LLVM_LINUX_VERSION_VPTRADDRDISCR) |
12210fca6ea1SDimitry Andric (LangOpts.PointerAuthVTPtrTypeDiscrimination
12220fca6ea1SDimitry Andric << AARCH64_PAUTH_PLATFORM_LLVM_LINUX_VERSION_VPTRTYPEDISCR) |
12230fca6ea1SDimitry Andric (LangOpts.PointerAuthInitFini
12240fca6ea1SDimitry Andric << AARCH64_PAUTH_PLATFORM_LLVM_LINUX_VERSION_INITFINI);
12250fca6ea1SDimitry Andric static_assert(AARCH64_PAUTH_PLATFORM_LLVM_LINUX_VERSION_INITFINI ==
12260fca6ea1SDimitry Andric AARCH64_PAUTH_PLATFORM_LLVM_LINUX_VERSION_LAST,
12270fca6ea1SDimitry Andric "Update when new enum items are defined");
12280fca6ea1SDimitry Andric if (PAuthABIVersion != 0) {
12290fca6ea1SDimitry Andric getModule().addModuleFlag(llvm::Module::Error,
12300fca6ea1SDimitry Andric "aarch64-elf-pauthabi-platform",
12310fca6ea1SDimitry Andric AARCH64_PAUTH_PLATFORM_LLVM_LINUX);
12320fca6ea1SDimitry Andric getModule().addModuleFlag(llvm::Module::Error,
12330fca6ea1SDimitry Andric "aarch64-elf-pauthabi-version",
12340fca6ea1SDimitry Andric PAuthABIVersion);
12350fca6ea1SDimitry Andric }
12360fca6ea1SDimitry Andric }
1237e8d8bef9SDimitry Andric }
1238e8d8bef9SDimitry Andric
12395f757f3fSDimitry Andric if (CodeGenOpts.StackClashProtector)
12405f757f3fSDimitry Andric getModule().addModuleFlag(
12415f757f3fSDimitry Andric llvm::Module::Override, "probe-stack",
12425f757f3fSDimitry Andric llvm::MDString::get(TheModule.getContext(), "inline-asm"));
12435f757f3fSDimitry Andric
12445f757f3fSDimitry Andric if (CodeGenOpts.StackProbeSize && CodeGenOpts.StackProbeSize != 4096)
12455f757f3fSDimitry Andric getModule().addModuleFlag(llvm::Module::Min, "stack-probe-size",
12465f757f3fSDimitry Andric CodeGenOpts.StackProbeSize);
12475f757f3fSDimitry Andric
1248e8d8bef9SDimitry Andric if (!CodeGenOpts.MemoryProfileOutput.empty()) {
1249e8d8bef9SDimitry Andric llvm::LLVMContext &Ctx = TheModule.getContext();
1250e8d8bef9SDimitry Andric getModule().addModuleFlag(
1251e8d8bef9SDimitry Andric llvm::Module::Error, "MemProfProfileFilename",
1252e8d8bef9SDimitry Andric llvm::MDString::get(Ctx, CodeGenOpts.MemoryProfileOutput));
1253e8d8bef9SDimitry Andric }
1254e8d8bef9SDimitry Andric
12550b57cec5SDimitry Andric if (LangOpts.CUDAIsDevice && getTriple().isNVPTX()) {
12560b57cec5SDimitry Andric // Indicate whether __nvvm_reflect should be configured to flush denormal
12570b57cec5SDimitry Andric // floating point values to 0. (This corresponds to its "__CUDA_FTZ"
12580b57cec5SDimitry Andric // property.)
12590b57cec5SDimitry Andric getModule().addModuleFlag(llvm::Module::Override, "nvvm-reflect-ftz",
12605ffd83dbSDimitry Andric CodeGenOpts.FP32DenormalMode.Output !=
12615ffd83dbSDimitry Andric llvm::DenormalMode::IEEE);
12620b57cec5SDimitry Andric }
12630b57cec5SDimitry Andric
1264fe6060f1SDimitry Andric if (LangOpts.EHAsynch)
1265fe6060f1SDimitry Andric getModule().addModuleFlag(llvm::Module::Warning, "eh-asynch", 1);
1266fe6060f1SDimitry Andric
1267fe6060f1SDimitry Andric // Indicate whether this Module was compiled with -fopenmp
1268fe6060f1SDimitry Andric if (getLangOpts().OpenMP && !getLangOpts().OpenMPSimd)
1269fe6060f1SDimitry Andric getModule().addModuleFlag(llvm::Module::Max, "openmp", LangOpts.OpenMP);
127006c3fb27SDimitry Andric if (getLangOpts().OpenMPIsTargetDevice)
1271fe6060f1SDimitry Andric getModule().addModuleFlag(llvm::Module::Max, "openmp-device",
1272fe6060f1SDimitry Andric LangOpts.OpenMP);
1273fe6060f1SDimitry Andric
12740b57cec5SDimitry Andric // Emit OpenCL specific module metadata: OpenCL/SPIR version.
127581ad6265SDimitry Andric if (LangOpts.OpenCL || (LangOpts.CUDAIsDevice && getTriple().isSPIRV())) {
12760b57cec5SDimitry Andric EmitOpenCLMetadata();
12770b57cec5SDimitry Andric // Emit SPIR version.
12780b57cec5SDimitry Andric if (getTriple().isSPIR()) {
12790b57cec5SDimitry Andric // SPIR v2.0 s2.12 - The SPIR version used by the module is stored in the
12800b57cec5SDimitry Andric // opencl.spir.version named metadata.
1281349cc55cSDimitry Andric // C++ for OpenCL has a distinct mapping for version compatibility with
1282349cc55cSDimitry Andric // OpenCL.
1283349cc55cSDimitry Andric auto Version = LangOpts.getOpenCLCompatibleVersion();
12840b57cec5SDimitry Andric llvm::Metadata *SPIRVerElts[] = {
12850b57cec5SDimitry Andric llvm::ConstantAsMetadata::get(llvm::ConstantInt::get(
12860b57cec5SDimitry Andric Int32Ty, Version / 100)),
12870b57cec5SDimitry Andric llvm::ConstantAsMetadata::get(llvm::ConstantInt::get(
12880b57cec5SDimitry Andric Int32Ty, (Version / 100 > 1) ? 0 : 2))};
12890b57cec5SDimitry Andric llvm::NamedMDNode *SPIRVerMD =
12900b57cec5SDimitry Andric TheModule.getOrInsertNamedMetadata("opencl.spir.version");
12910b57cec5SDimitry Andric llvm::LLVMContext &Ctx = TheModule.getContext();
12920b57cec5SDimitry Andric SPIRVerMD->addOperand(llvm::MDNode::get(Ctx, SPIRVerElts));
12930b57cec5SDimitry Andric }
12940b57cec5SDimitry Andric }
12950b57cec5SDimitry Andric
129681ad6265SDimitry Andric // HLSL related end of code gen work items.
129781ad6265SDimitry Andric if (LangOpts.HLSL)
129881ad6265SDimitry Andric getHLSLRuntime().finishCodeGen();
129981ad6265SDimitry Andric
13000b57cec5SDimitry Andric if (uint32_t PLevel = Context.getLangOpts().PICLevel) {
13010b57cec5SDimitry Andric assert(PLevel < 3 && "Invalid PIC Level");
13020b57cec5SDimitry Andric getModule().setPICLevel(static_cast<llvm::PICLevel::Level>(PLevel));
13030b57cec5SDimitry Andric if (Context.getLangOpts().PIE)
13040b57cec5SDimitry Andric getModule().setPIELevel(static_cast<llvm::PIELevel::Level>(PLevel));
13050b57cec5SDimitry Andric }
13060b57cec5SDimitry Andric
13070b57cec5SDimitry Andric if (getCodeGenOpts().CodeModel.size() > 0) {
13080b57cec5SDimitry Andric unsigned CM = llvm::StringSwitch<unsigned>(getCodeGenOpts().CodeModel)
13090b57cec5SDimitry Andric .Case("tiny", llvm::CodeModel::Tiny)
13100b57cec5SDimitry Andric .Case("small", llvm::CodeModel::Small)
13110b57cec5SDimitry Andric .Case("kernel", llvm::CodeModel::Kernel)
13120b57cec5SDimitry Andric .Case("medium", llvm::CodeModel::Medium)
13130b57cec5SDimitry Andric .Case("large", llvm::CodeModel::Large)
13140b57cec5SDimitry Andric .Default(~0u);
13150b57cec5SDimitry Andric if (CM != ~0u) {
13160b57cec5SDimitry Andric llvm::CodeModel::Model codeModel = static_cast<llvm::CodeModel::Model>(CM);
13170b57cec5SDimitry Andric getModule().setCodeModel(codeModel);
13185f757f3fSDimitry Andric
13197a6dacacSDimitry Andric if ((CM == llvm::CodeModel::Medium || CM == llvm::CodeModel::Large) &&
13205f757f3fSDimitry Andric Context.getTargetInfo().getTriple().getArch() ==
13215f757f3fSDimitry Andric llvm::Triple::x86_64) {
13225f757f3fSDimitry Andric getModule().setLargeDataThreshold(getCodeGenOpts().LargeDataThreshold);
13235f757f3fSDimitry Andric }
13240b57cec5SDimitry Andric }
13250b57cec5SDimitry Andric }
13260b57cec5SDimitry Andric
13270b57cec5SDimitry Andric if (CodeGenOpts.NoPLT)
13280b57cec5SDimitry Andric getModule().setRtLibUseGOT();
132906c3fb27SDimitry Andric if (getTriple().isOSBinFormatELF() &&
133006c3fb27SDimitry Andric CodeGenOpts.DirectAccessExternalData !=
133106c3fb27SDimitry Andric getModule().getDirectAccessExternalData()) {
133206c3fb27SDimitry Andric getModule().setDirectAccessExternalData(
133306c3fb27SDimitry Andric CodeGenOpts.DirectAccessExternalData);
133406c3fb27SDimitry Andric }
1335fe6060f1SDimitry Andric if (CodeGenOpts.UnwindTables)
133681ad6265SDimitry Andric getModule().setUwtable(llvm::UWTableKind(CodeGenOpts.UnwindTables));
1337fe6060f1SDimitry Andric
1338fe6060f1SDimitry Andric switch (CodeGenOpts.getFramePointer()) {
1339fe6060f1SDimitry Andric case CodeGenOptions::FramePointerKind::None:
1340fe6060f1SDimitry Andric // 0 ("none") is the default.
1341fe6060f1SDimitry Andric break;
13420fca6ea1SDimitry Andric case CodeGenOptions::FramePointerKind::Reserved:
13430fca6ea1SDimitry Andric getModule().setFramePointer(llvm::FramePointerKind::Reserved);
13440fca6ea1SDimitry Andric break;
1345fe6060f1SDimitry Andric case CodeGenOptions::FramePointerKind::NonLeaf:
1346fe6060f1SDimitry Andric getModule().setFramePointer(llvm::FramePointerKind::NonLeaf);
1347fe6060f1SDimitry Andric break;
1348fe6060f1SDimitry Andric case CodeGenOptions::FramePointerKind::All:
1349fe6060f1SDimitry Andric getModule().setFramePointer(llvm::FramePointerKind::All);
1350fe6060f1SDimitry Andric break;
1351fe6060f1SDimitry Andric }
13520b57cec5SDimitry Andric
13530b57cec5SDimitry Andric SimplifyPersonality();
13540b57cec5SDimitry Andric
13550b57cec5SDimitry Andric if (getCodeGenOpts().EmitDeclMetadata)
13560b57cec5SDimitry Andric EmitDeclMetadata();
13570b57cec5SDimitry Andric
135806c3fb27SDimitry Andric if (getCodeGenOpts().CoverageNotesFile.size() ||
135906c3fb27SDimitry Andric getCodeGenOpts().CoverageDataFile.size())
13600b57cec5SDimitry Andric EmitCoverageFile();
13610b57cec5SDimitry Andric
13625ffd83dbSDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
13635ffd83dbSDimitry Andric DI->finalize();
13640b57cec5SDimitry Andric
13650b57cec5SDimitry Andric if (getCodeGenOpts().EmitVersionIdentMetadata)
13660b57cec5SDimitry Andric EmitVersionIdentMetadata();
13670b57cec5SDimitry Andric
13680b57cec5SDimitry Andric if (!getCodeGenOpts().RecordCommandLine.empty())
13690b57cec5SDimitry Andric EmitCommandLineMetadata();
13700b57cec5SDimitry Andric
1371fe6060f1SDimitry Andric if (!getCodeGenOpts().StackProtectorGuard.empty())
1372fe6060f1SDimitry Andric getModule().setStackProtectorGuard(getCodeGenOpts().StackProtectorGuard);
1373fe6060f1SDimitry Andric if (!getCodeGenOpts().StackProtectorGuardReg.empty())
1374fe6060f1SDimitry Andric getModule().setStackProtectorGuardReg(
1375fe6060f1SDimitry Andric getCodeGenOpts().StackProtectorGuardReg);
1376753f127fSDimitry Andric if (!getCodeGenOpts().StackProtectorGuardSymbol.empty())
1377753f127fSDimitry Andric getModule().setStackProtectorGuardSymbol(
1378753f127fSDimitry Andric getCodeGenOpts().StackProtectorGuardSymbol);
1379fe6060f1SDimitry Andric if (getCodeGenOpts().StackProtectorGuardOffset != INT_MAX)
1380fe6060f1SDimitry Andric getModule().setStackProtectorGuardOffset(
1381fe6060f1SDimitry Andric getCodeGenOpts().StackProtectorGuardOffset);
1382fe6060f1SDimitry Andric if (getCodeGenOpts().StackAlignment)
1383fe6060f1SDimitry Andric getModule().setOverrideStackAlignment(getCodeGenOpts().StackAlignment);
1384349cc55cSDimitry Andric if (getCodeGenOpts().SkipRaxSetup)
1385349cc55cSDimitry Andric getModule().addModuleFlag(llvm::Module::Override, "SkipRaxSetup", 1);
13865f757f3fSDimitry Andric if (getLangOpts().RegCall4)
13875f757f3fSDimitry Andric getModule().addModuleFlag(llvm::Module::Override, "RegCallv4", 1);
1388fe6060f1SDimitry Andric
138906c3fb27SDimitry Andric if (getContext().getTargetInfo().getMaxTLSAlign())
139006c3fb27SDimitry Andric getModule().addModuleFlag(llvm::Module::Error, "MaxTLSAlign",
139106c3fb27SDimitry Andric getContext().getTargetInfo().getMaxTLSAlign());
139206c3fb27SDimitry Andric
13935f757f3fSDimitry Andric getTargetCodeGenInfo().emitTargetGlobals(*this);
13945f757f3fSDimitry Andric
13955ffd83dbSDimitry Andric getTargetCodeGenInfo().emitTargetMetadata(*this, MangledDeclNames);
13965ffd83dbSDimitry Andric
13975ffd83dbSDimitry Andric EmitBackendOptionsMetadata(getCodeGenOpts());
1398e8d8bef9SDimitry Andric
139981ad6265SDimitry Andric // If there is device offloading code embed it in the host now.
140081ad6265SDimitry Andric EmbedObject(&getModule(), CodeGenOpts, getDiags());
140181ad6265SDimitry Andric
1402e8d8bef9SDimitry Andric // Set visibility from DLL storage class
1403e8d8bef9SDimitry Andric // We do this at the end of LLVM IR generation; after any operation
1404e8d8bef9SDimitry Andric // that might affect the DLL storage class or the visibility, and
1405e8d8bef9SDimitry Andric // before anything that might act on these.
1406e8d8bef9SDimitry Andric setVisibilityFromDLLStorageClass(LangOpts, getModule());
14070fca6ea1SDimitry Andric
14080fca6ea1SDimitry Andric // Check the tail call symbols are truly undefined.
14090fca6ea1SDimitry Andric if (getTriple().isPPC() && !MustTailCallUndefinedGlobals.empty()) {
14100fca6ea1SDimitry Andric for (auto &I : MustTailCallUndefinedGlobals) {
14110fca6ea1SDimitry Andric if (!I.first->isDefined())
14120fca6ea1SDimitry Andric getDiags().Report(I.second, diag::err_ppc_impossible_musttail) << 2;
14130fca6ea1SDimitry Andric else {
14140fca6ea1SDimitry Andric StringRef MangledName = getMangledName(GlobalDecl(I.first));
14150fca6ea1SDimitry Andric llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
14160fca6ea1SDimitry Andric if (!Entry || Entry->isWeakForLinker() ||
14170fca6ea1SDimitry Andric Entry->isDeclarationForLinker())
14180fca6ea1SDimitry Andric getDiags().Report(I.second, diag::err_ppc_impossible_musttail) << 2;
14190fca6ea1SDimitry Andric }
14200fca6ea1SDimitry Andric }
14210fca6ea1SDimitry Andric }
14220b57cec5SDimitry Andric }
14230b57cec5SDimitry Andric
EmitOpenCLMetadata()14240b57cec5SDimitry Andric void CodeGenModule::EmitOpenCLMetadata() {
14250b57cec5SDimitry Andric // SPIR v2.0 s2.13 - The OpenCL version used by the module is stored in the
14260b57cec5SDimitry Andric // opencl.ocl.version named metadata node.
14270fca6ea1SDimitry Andric // C++ for OpenCL has a distinct mapping for versions compatible with OpenCL.
14280fca6ea1SDimitry Andric auto CLVersion = LangOpts.getOpenCLCompatibleVersion();
14290fca6ea1SDimitry Andric
14300fca6ea1SDimitry Andric auto EmitVersion = [this](StringRef MDName, int Version) {
14310b57cec5SDimitry Andric llvm::Metadata *OCLVerElts[] = {
14320fca6ea1SDimitry Andric llvm::ConstantAsMetadata::get(
14330fca6ea1SDimitry Andric llvm::ConstantInt::get(Int32Ty, Version / 100)),
14340fca6ea1SDimitry Andric llvm::ConstantAsMetadata::get(
14350fca6ea1SDimitry Andric llvm::ConstantInt::get(Int32Ty, (Version % 100) / 10))};
14360fca6ea1SDimitry Andric llvm::NamedMDNode *OCLVerMD = TheModule.getOrInsertNamedMetadata(MDName);
14370b57cec5SDimitry Andric llvm::LLVMContext &Ctx = TheModule.getContext();
14380b57cec5SDimitry Andric OCLVerMD->addOperand(llvm::MDNode::get(Ctx, OCLVerElts));
14390fca6ea1SDimitry Andric };
14400fca6ea1SDimitry Andric
14410fca6ea1SDimitry Andric EmitVersion("opencl.ocl.version", CLVersion);
14420fca6ea1SDimitry Andric if (LangOpts.OpenCLCPlusPlus) {
14430fca6ea1SDimitry Andric // In addition to the OpenCL compatible version, emit the C++ version.
14440fca6ea1SDimitry Andric EmitVersion("opencl.cxx.version", LangOpts.OpenCLCPlusPlusVersion);
14450fca6ea1SDimitry Andric }
14460b57cec5SDimitry Andric }
14470b57cec5SDimitry Andric
EmitBackendOptionsMetadata(const CodeGenOptions & CodeGenOpts)14485ffd83dbSDimitry Andric void CodeGenModule::EmitBackendOptionsMetadata(
144906c3fb27SDimitry Andric const CodeGenOptions &CodeGenOpts) {
1450bdd1243dSDimitry Andric if (getTriple().isRISCV()) {
145106c3fb27SDimitry Andric getModule().addModuleFlag(llvm::Module::Min, "SmallDataLimit",
14525ffd83dbSDimitry Andric CodeGenOpts.SmallDataLimit);
14535ffd83dbSDimitry Andric }
14545ffd83dbSDimitry Andric }
14555ffd83dbSDimitry Andric
UpdateCompletedType(const TagDecl * TD)14560b57cec5SDimitry Andric void CodeGenModule::UpdateCompletedType(const TagDecl *TD) {
14570b57cec5SDimitry Andric // Make sure that this type is translated.
14585deeebd8SDimitry Andric getTypes().UpdateCompletedType(TD);
14590b57cec5SDimitry Andric }
14600b57cec5SDimitry Andric
RefreshTypeCacheForClass(const CXXRecordDecl * RD)14610b57cec5SDimitry Andric void CodeGenModule::RefreshTypeCacheForClass(const CXXRecordDecl *RD) {
14620b57cec5SDimitry Andric // Make sure that this type is translated.
14635deeebd8SDimitry Andric getTypes().RefreshTypeCacheForClass(RD);
14640b57cec5SDimitry Andric }
14650b57cec5SDimitry Andric
getTBAATypeInfo(QualType QTy)14660b57cec5SDimitry Andric llvm::MDNode *CodeGenModule::getTBAATypeInfo(QualType QTy) {
14670b57cec5SDimitry Andric if (!TBAA)
14680b57cec5SDimitry Andric return nullptr;
14690b57cec5SDimitry Andric return TBAA->getTypeInfo(QTy);
14700b57cec5SDimitry Andric }
14710b57cec5SDimitry Andric
getTBAAAccessInfo(QualType AccessType)14720b57cec5SDimitry Andric TBAAAccessInfo CodeGenModule::getTBAAAccessInfo(QualType AccessType) {
14730b57cec5SDimitry Andric if (!TBAA)
14740b57cec5SDimitry Andric return TBAAAccessInfo();
14755ffd83dbSDimitry Andric if (getLangOpts().CUDAIsDevice) {
14765ffd83dbSDimitry Andric // As CUDA builtin surface/texture types are replaced, skip generating TBAA
14775ffd83dbSDimitry Andric // access info.
14785ffd83dbSDimitry Andric if (AccessType->isCUDADeviceBuiltinSurfaceType()) {
14795ffd83dbSDimitry Andric if (getTargetCodeGenInfo().getCUDADeviceBuiltinSurfaceDeviceType() !=
14805ffd83dbSDimitry Andric nullptr)
14815ffd83dbSDimitry Andric return TBAAAccessInfo();
14825ffd83dbSDimitry Andric } else if (AccessType->isCUDADeviceBuiltinTextureType()) {
14835ffd83dbSDimitry Andric if (getTargetCodeGenInfo().getCUDADeviceBuiltinTextureDeviceType() !=
14845ffd83dbSDimitry Andric nullptr)
14855ffd83dbSDimitry Andric return TBAAAccessInfo();
14865ffd83dbSDimitry Andric }
14875ffd83dbSDimitry Andric }
14880b57cec5SDimitry Andric return TBAA->getAccessInfo(AccessType);
14890b57cec5SDimitry Andric }
14900b57cec5SDimitry Andric
14910b57cec5SDimitry Andric TBAAAccessInfo
getTBAAVTablePtrAccessInfo(llvm::Type * VTablePtrType)14920b57cec5SDimitry Andric CodeGenModule::getTBAAVTablePtrAccessInfo(llvm::Type *VTablePtrType) {
14930b57cec5SDimitry Andric if (!TBAA)
14940b57cec5SDimitry Andric return TBAAAccessInfo();
14950b57cec5SDimitry Andric return TBAA->getVTablePtrAccessInfo(VTablePtrType);
14960b57cec5SDimitry Andric }
14970b57cec5SDimitry Andric
getTBAAStructInfo(QualType QTy)14980b57cec5SDimitry Andric llvm::MDNode *CodeGenModule::getTBAAStructInfo(QualType QTy) {
14990b57cec5SDimitry Andric if (!TBAA)
15000b57cec5SDimitry Andric return nullptr;
15010b57cec5SDimitry Andric return TBAA->getTBAAStructInfo(QTy);
15020b57cec5SDimitry Andric }
15030b57cec5SDimitry Andric
getTBAABaseTypeInfo(QualType QTy)15040b57cec5SDimitry Andric llvm::MDNode *CodeGenModule::getTBAABaseTypeInfo(QualType QTy) {
15050b57cec5SDimitry Andric if (!TBAA)
15060b57cec5SDimitry Andric return nullptr;
15070b57cec5SDimitry Andric return TBAA->getBaseTypeInfo(QTy);
15080b57cec5SDimitry Andric }
15090b57cec5SDimitry Andric
getTBAAAccessTagInfo(TBAAAccessInfo Info)15100b57cec5SDimitry Andric llvm::MDNode *CodeGenModule::getTBAAAccessTagInfo(TBAAAccessInfo Info) {
15110b57cec5SDimitry Andric if (!TBAA)
15120b57cec5SDimitry Andric return nullptr;
15130b57cec5SDimitry Andric return TBAA->getAccessTagInfo(Info);
15140b57cec5SDimitry Andric }
15150b57cec5SDimitry Andric
mergeTBAAInfoForCast(TBAAAccessInfo SourceInfo,TBAAAccessInfo TargetInfo)15160b57cec5SDimitry Andric TBAAAccessInfo CodeGenModule::mergeTBAAInfoForCast(TBAAAccessInfo SourceInfo,
15170b57cec5SDimitry Andric TBAAAccessInfo TargetInfo) {
15180b57cec5SDimitry Andric if (!TBAA)
15190b57cec5SDimitry Andric return TBAAAccessInfo();
15200b57cec5SDimitry Andric return TBAA->mergeTBAAInfoForCast(SourceInfo, TargetInfo);
15210b57cec5SDimitry Andric }
15220b57cec5SDimitry Andric
15230b57cec5SDimitry Andric TBAAAccessInfo
mergeTBAAInfoForConditionalOperator(TBAAAccessInfo InfoA,TBAAAccessInfo InfoB)15240b57cec5SDimitry Andric CodeGenModule::mergeTBAAInfoForConditionalOperator(TBAAAccessInfo InfoA,
15250b57cec5SDimitry Andric TBAAAccessInfo InfoB) {
15260b57cec5SDimitry Andric if (!TBAA)
15270b57cec5SDimitry Andric return TBAAAccessInfo();
15280b57cec5SDimitry Andric return TBAA->mergeTBAAInfoForConditionalOperator(InfoA, InfoB);
15290b57cec5SDimitry Andric }
15300b57cec5SDimitry Andric
15310b57cec5SDimitry Andric TBAAAccessInfo
mergeTBAAInfoForMemoryTransfer(TBAAAccessInfo DestInfo,TBAAAccessInfo SrcInfo)15320b57cec5SDimitry Andric CodeGenModule::mergeTBAAInfoForMemoryTransfer(TBAAAccessInfo DestInfo,
15330b57cec5SDimitry Andric TBAAAccessInfo SrcInfo) {
15340b57cec5SDimitry Andric if (!TBAA)
15350b57cec5SDimitry Andric return TBAAAccessInfo();
15360b57cec5SDimitry Andric return TBAA->mergeTBAAInfoForConditionalOperator(DestInfo, SrcInfo);
15370b57cec5SDimitry Andric }
15380b57cec5SDimitry Andric
DecorateInstructionWithTBAA(llvm::Instruction * Inst,TBAAAccessInfo TBAAInfo)15390b57cec5SDimitry Andric void CodeGenModule::DecorateInstructionWithTBAA(llvm::Instruction *Inst,
15400b57cec5SDimitry Andric TBAAAccessInfo TBAAInfo) {
15410b57cec5SDimitry Andric if (llvm::MDNode *Tag = getTBAAAccessTagInfo(TBAAInfo))
15420b57cec5SDimitry Andric Inst->setMetadata(llvm::LLVMContext::MD_tbaa, Tag);
15430b57cec5SDimitry Andric }
15440b57cec5SDimitry Andric
DecorateInstructionWithInvariantGroup(llvm::Instruction * I,const CXXRecordDecl * RD)15450b57cec5SDimitry Andric void CodeGenModule::DecorateInstructionWithInvariantGroup(
15460b57cec5SDimitry Andric llvm::Instruction *I, const CXXRecordDecl *RD) {
15470b57cec5SDimitry Andric I->setMetadata(llvm::LLVMContext::MD_invariant_group,
15480b57cec5SDimitry Andric llvm::MDNode::get(getLLVMContext(), {}));
15490b57cec5SDimitry Andric }
15500b57cec5SDimitry Andric
Error(SourceLocation loc,StringRef message)15510b57cec5SDimitry Andric void CodeGenModule::Error(SourceLocation loc, StringRef message) {
15520b57cec5SDimitry Andric unsigned diagID = getDiags().getCustomDiagID(DiagnosticsEngine::Error, "%0");
15530b57cec5SDimitry Andric getDiags().Report(Context.getFullLoc(loc), diagID) << message;
15540b57cec5SDimitry Andric }
15550b57cec5SDimitry Andric
15560b57cec5SDimitry Andric /// ErrorUnsupported - Print out an error that codegen doesn't support the
15570b57cec5SDimitry Andric /// specified stmt yet.
ErrorUnsupported(const Stmt * S,const char * Type)15580b57cec5SDimitry Andric void CodeGenModule::ErrorUnsupported(const Stmt *S, const char *Type) {
15590b57cec5SDimitry Andric unsigned DiagID = getDiags().getCustomDiagID(DiagnosticsEngine::Error,
15600b57cec5SDimitry Andric "cannot compile this %0 yet");
15610b57cec5SDimitry Andric std::string Msg = Type;
15620b57cec5SDimitry Andric getDiags().Report(Context.getFullLoc(S->getBeginLoc()), DiagID)
15630b57cec5SDimitry Andric << Msg << S->getSourceRange();
15640b57cec5SDimitry Andric }
15650b57cec5SDimitry Andric
15660b57cec5SDimitry Andric /// ErrorUnsupported - Print out an error that codegen doesn't support the
15670b57cec5SDimitry Andric /// specified decl yet.
ErrorUnsupported(const Decl * D,const char * Type)15680b57cec5SDimitry Andric void CodeGenModule::ErrorUnsupported(const Decl *D, const char *Type) {
15690b57cec5SDimitry Andric unsigned DiagID = getDiags().getCustomDiagID(DiagnosticsEngine::Error,
15700b57cec5SDimitry Andric "cannot compile this %0 yet");
15710b57cec5SDimitry Andric std::string Msg = Type;
15720b57cec5SDimitry Andric getDiags().Report(Context.getFullLoc(D->getLocation()), DiagID) << Msg;
15730b57cec5SDimitry Andric }
15740b57cec5SDimitry Andric
getSize(CharUnits size)15750b57cec5SDimitry Andric llvm::ConstantInt *CodeGenModule::getSize(CharUnits size) {
15760b57cec5SDimitry Andric return llvm::ConstantInt::get(SizeTy, size.getQuantity());
15770b57cec5SDimitry Andric }
15780b57cec5SDimitry Andric
setGlobalVisibility(llvm::GlobalValue * GV,const NamedDecl * D) const15790b57cec5SDimitry Andric void CodeGenModule::setGlobalVisibility(llvm::GlobalValue *GV,
15800b57cec5SDimitry Andric const NamedDecl *D) const {
15810b57cec5SDimitry Andric // Internal definitions always have default visibility.
15820b57cec5SDimitry Andric if (GV->hasLocalLinkage()) {
15830b57cec5SDimitry Andric GV->setVisibility(llvm::GlobalValue::DefaultVisibility);
15840b57cec5SDimitry Andric return;
15850b57cec5SDimitry Andric }
15860b57cec5SDimitry Andric if (!D)
15870b57cec5SDimitry Andric return;
15885f757f3fSDimitry Andric
15890b57cec5SDimitry Andric // Set visibility for definitions, and for declarations if requested globally
15900b57cec5SDimitry Andric // or set explicitly.
15910b57cec5SDimitry Andric LinkageInfo LV = D->getLinkageAndVisibility();
15925f757f3fSDimitry Andric
15935f757f3fSDimitry Andric // OpenMP declare target variables must be visible to the host so they can
15945f757f3fSDimitry Andric // be registered. We require protected visibility unless the variable has
15955f757f3fSDimitry Andric // the DT_nohost modifier and does not need to be registered.
15965f757f3fSDimitry Andric if (Context.getLangOpts().OpenMP &&
15975f757f3fSDimitry Andric Context.getLangOpts().OpenMPIsTargetDevice && isa<VarDecl>(D) &&
15985f757f3fSDimitry Andric D->hasAttr<OMPDeclareTargetDeclAttr>() &&
15995f757f3fSDimitry Andric D->getAttr<OMPDeclareTargetDeclAttr>()->getDevType() !=
16005f757f3fSDimitry Andric OMPDeclareTargetDeclAttr::DT_NoHost &&
16015f757f3fSDimitry Andric LV.getVisibility() == HiddenVisibility) {
16025f757f3fSDimitry Andric GV->setVisibility(llvm::GlobalValue::ProtectedVisibility);
16035f757f3fSDimitry Andric return;
16045f757f3fSDimitry Andric }
16055f757f3fSDimitry Andric
1606bdd1243dSDimitry Andric if (GV->hasDLLExportStorageClass() || GV->hasDLLImportStorageClass()) {
1607bdd1243dSDimitry Andric // Reject incompatible dlllstorage and visibility annotations.
1608bdd1243dSDimitry Andric if (!LV.isVisibilityExplicit())
1609bdd1243dSDimitry Andric return;
1610bdd1243dSDimitry Andric if (GV->hasDLLExportStorageClass()) {
1611bdd1243dSDimitry Andric if (LV.getVisibility() == HiddenVisibility)
1612bdd1243dSDimitry Andric getDiags().Report(D->getLocation(),
1613bdd1243dSDimitry Andric diag::err_hidden_visibility_dllexport);
1614bdd1243dSDimitry Andric } else if (LV.getVisibility() != DefaultVisibility) {
1615bdd1243dSDimitry Andric getDiags().Report(D->getLocation(),
1616bdd1243dSDimitry Andric diag::err_non_default_visibility_dllimport);
1617bdd1243dSDimitry Andric }
1618bdd1243dSDimitry Andric return;
1619bdd1243dSDimitry Andric }
1620bdd1243dSDimitry Andric
16210b57cec5SDimitry Andric if (LV.isVisibilityExplicit() || getLangOpts().SetVisibilityForExternDecls ||
16220b57cec5SDimitry Andric !GV->isDeclarationForLinker())
16230b57cec5SDimitry Andric GV->setVisibility(GetLLVMVisibility(LV.getVisibility()));
16240b57cec5SDimitry Andric }
16250b57cec5SDimitry Andric
shouldAssumeDSOLocal(const CodeGenModule & CGM,llvm::GlobalValue * GV)16260b57cec5SDimitry Andric static bool shouldAssumeDSOLocal(const CodeGenModule &CGM,
16270b57cec5SDimitry Andric llvm::GlobalValue *GV) {
16280b57cec5SDimitry Andric if (GV->hasLocalLinkage())
16290b57cec5SDimitry Andric return true;
16300b57cec5SDimitry Andric
16310b57cec5SDimitry Andric if (!GV->hasDefaultVisibility() && !GV->hasExternalWeakLinkage())
16320b57cec5SDimitry Andric return true;
16330b57cec5SDimitry Andric
16340b57cec5SDimitry Andric // DLLImport explicitly marks the GV as external.
16350b57cec5SDimitry Andric if (GV->hasDLLImportStorageClass())
16360b57cec5SDimitry Andric return false;
16370b57cec5SDimitry Andric
16380b57cec5SDimitry Andric const llvm::Triple &TT = CGM.getTriple();
16395f757f3fSDimitry Andric const auto &CGOpts = CGM.getCodeGenOpts();
16400b57cec5SDimitry Andric if (TT.isWindowsGNUEnvironment()) {
16410b57cec5SDimitry Andric // In MinGW, variables without DLLImport can still be automatically
16420b57cec5SDimitry Andric // imported from a DLL by the linker; don't mark variables that
16430b57cec5SDimitry Andric // potentially could come from another DLL as DSO local.
1644fe6060f1SDimitry Andric
1645fe6060f1SDimitry Andric // With EmulatedTLS, TLS variables can be autoimported from other DLLs
1646fe6060f1SDimitry Andric // (and this actually happens in the public interface of libstdc++), so
1647fe6060f1SDimitry Andric // such variables can't be marked as DSO local. (Native TLS variables
1648fe6060f1SDimitry Andric // can't be dllimported at all, though.)
16490b57cec5SDimitry Andric if (GV->isDeclarationForLinker() && isa<llvm::GlobalVariable>(GV) &&
16505f757f3fSDimitry Andric (!GV->isThreadLocal() || CGM.getCodeGenOpts().EmulatedTLS) &&
16515f757f3fSDimitry Andric CGOpts.AutoImport)
16520b57cec5SDimitry Andric return false;
16530b57cec5SDimitry Andric }
16540b57cec5SDimitry Andric
16550b57cec5SDimitry Andric // On COFF, don't mark 'extern_weak' symbols as DSO local. If these symbols
16560b57cec5SDimitry Andric // remain unresolved in the link, they can be resolved to zero, which is
16570b57cec5SDimitry Andric // outside the current DSO.
16580b57cec5SDimitry Andric if (TT.isOSBinFormatCOFF() && GV->hasExternalWeakLinkage())
16590b57cec5SDimitry Andric return false;
16600b57cec5SDimitry Andric
16610b57cec5SDimitry Andric // Every other GV is local on COFF.
16620b57cec5SDimitry Andric // Make an exception for windows OS in the triple: Some firmware builds use
16630b57cec5SDimitry Andric // *-win32-macho triples. This (accidentally?) produced windows relocations
16640b57cec5SDimitry Andric // without GOT tables in older clang versions; Keep this behaviour.
16650b57cec5SDimitry Andric // FIXME: even thread local variables?
16660b57cec5SDimitry Andric if (TT.isOSBinFormatCOFF() || (TT.isOSWindows() && TT.isOSBinFormatMachO()))
16670b57cec5SDimitry Andric return true;
16680b57cec5SDimitry Andric
16690b57cec5SDimitry Andric // Only handle COFF and ELF for now.
16700b57cec5SDimitry Andric if (!TT.isOSBinFormatELF())
16710b57cec5SDimitry Andric return false;
16720b57cec5SDimitry Andric
1673fe6060f1SDimitry Andric // If this is not an executable, don't assume anything is local.
1674fe6060f1SDimitry Andric llvm::Reloc::Model RM = CGOpts.RelocationModel;
1675fe6060f1SDimitry Andric const auto &LOpts = CGM.getLangOpts();
1676e8d8bef9SDimitry Andric if (RM != llvm::Reloc::Static && !LOpts.PIE) {
1677e8d8bef9SDimitry Andric // On ELF, if -fno-semantic-interposition is specified and the target
1678e8d8bef9SDimitry Andric // supports local aliases, there will be neither CC1
1679e8d8bef9SDimitry Andric // -fsemantic-interposition nor -fhalf-no-semantic-interposition. Set
1680fe6060f1SDimitry Andric // dso_local on the function if using a local alias is preferable (can avoid
1681fe6060f1SDimitry Andric // PLT indirection).
1682fe6060f1SDimitry Andric if (!(isa<llvm::Function>(GV) && GV->canBenefitFromLocalAlias()))
16830b57cec5SDimitry Andric return false;
1684e8d8bef9SDimitry Andric return !(CGM.getLangOpts().SemanticInterposition ||
1685e8d8bef9SDimitry Andric CGM.getLangOpts().HalfNoSemanticInterposition);
1686e8d8bef9SDimitry Andric }
16870b57cec5SDimitry Andric
16880b57cec5SDimitry Andric // A definition cannot be preempted from an executable.
16890b57cec5SDimitry Andric if (!GV->isDeclarationForLinker())
16900b57cec5SDimitry Andric return true;
16910b57cec5SDimitry Andric
16920b57cec5SDimitry Andric // Most PIC code sequences that assume that a symbol is local cannot produce a
16930b57cec5SDimitry Andric // 0 if it turns out the symbol is undefined. While this is ABI and relocation
16940b57cec5SDimitry Andric // depended, it seems worth it to handle it here.
16950b57cec5SDimitry Andric if (RM == llvm::Reloc::PIC_ && GV->hasExternalWeakLinkage())
16960b57cec5SDimitry Andric return false;
16970b57cec5SDimitry Andric
1698e8d8bef9SDimitry Andric // PowerPC64 prefers TOC indirection to avoid copy relocations.
1699e8d8bef9SDimitry Andric if (TT.isPPC64())
17000b57cec5SDimitry Andric return false;
17010b57cec5SDimitry Andric
1702e8d8bef9SDimitry Andric if (CGOpts.DirectAccessExternalData) {
1703e8d8bef9SDimitry Andric // If -fdirect-access-external-data (default for -fno-pic), set dso_local
1704e8d8bef9SDimitry Andric // for non-thread-local variables. If the symbol is not defined in the
1705e8d8bef9SDimitry Andric // executable, a copy relocation will be needed at link time. dso_local is
1706e8d8bef9SDimitry Andric // excluded for thread-local variables because they generally don't support
1707e8d8bef9SDimitry Andric // copy relocations.
17080b57cec5SDimitry Andric if (auto *Var = dyn_cast<llvm::GlobalVariable>(GV))
1709e8d8bef9SDimitry Andric if (!Var->isThreadLocal())
17100b57cec5SDimitry Andric return true;
17110b57cec5SDimitry Andric
1712e8d8bef9SDimitry Andric // -fno-pic sets dso_local on a function declaration to allow direct
1713e8d8bef9SDimitry Andric // accesses when taking its address (similar to a data symbol). If the
1714e8d8bef9SDimitry Andric // function is not defined in the executable, a canonical PLT entry will be
1715e8d8bef9SDimitry Andric // needed at link time. -fno-direct-access-external-data can avoid the
1716e8d8bef9SDimitry Andric // canonical PLT entry. We don't generalize this condition to -fpie/-fpic as
1717e8d8bef9SDimitry Andric // it could just cause trouble without providing perceptible benefits.
17180b57cec5SDimitry Andric if (isa<llvm::Function>(GV) && !CGOpts.NoPLT && RM == llvm::Reloc::Static)
17190b57cec5SDimitry Andric return true;
1720e8d8bef9SDimitry Andric }
1721e8d8bef9SDimitry Andric
1722e8d8bef9SDimitry Andric // If we can use copy relocations we can assume it is local.
17230b57cec5SDimitry Andric
17245ffd83dbSDimitry Andric // Otherwise don't assume it is local.
17250b57cec5SDimitry Andric return false;
17260b57cec5SDimitry Andric }
17270b57cec5SDimitry Andric
setDSOLocal(llvm::GlobalValue * GV) const17280b57cec5SDimitry Andric void CodeGenModule::setDSOLocal(llvm::GlobalValue *GV) const {
17290b57cec5SDimitry Andric GV->setDSOLocal(shouldAssumeDSOLocal(*this, GV));
17300b57cec5SDimitry Andric }
17310b57cec5SDimitry Andric
setDLLImportDLLExport(llvm::GlobalValue * GV,GlobalDecl GD) const17320b57cec5SDimitry Andric void CodeGenModule::setDLLImportDLLExport(llvm::GlobalValue *GV,
17330b57cec5SDimitry Andric GlobalDecl GD) const {
17340b57cec5SDimitry Andric const auto *D = dyn_cast<NamedDecl>(GD.getDecl());
17350b57cec5SDimitry Andric // C++ destructors have a few C++ ABI specific special cases.
17360b57cec5SDimitry Andric if (const auto *Dtor = dyn_cast_or_null<CXXDestructorDecl>(D)) {
17370b57cec5SDimitry Andric getCXXABI().setCXXDestructorDLLStorage(GV, Dtor, GD.getDtorType());
17380b57cec5SDimitry Andric return;
17390b57cec5SDimitry Andric }
17400b57cec5SDimitry Andric setDLLImportDLLExport(GV, D);
17410b57cec5SDimitry Andric }
17420b57cec5SDimitry Andric
setDLLImportDLLExport(llvm::GlobalValue * GV,const NamedDecl * D) const17430b57cec5SDimitry Andric void CodeGenModule::setDLLImportDLLExport(llvm::GlobalValue *GV,
17440b57cec5SDimitry Andric const NamedDecl *D) const {
17450b57cec5SDimitry Andric if (D && D->isExternallyVisible()) {
17460b57cec5SDimitry Andric if (D->hasAttr<DLLImportAttr>())
17470b57cec5SDimitry Andric GV->setDLLStorageClass(llvm::GlobalVariable::DLLImportStorageClass);
174881ad6265SDimitry Andric else if ((D->hasAttr<DLLExportAttr>() ||
174981ad6265SDimitry Andric shouldMapVisibilityToDLLExport(D)) &&
175081ad6265SDimitry Andric !GV->isDeclarationForLinker())
17510b57cec5SDimitry Andric GV->setDLLStorageClass(llvm::GlobalVariable::DLLExportStorageClass);
17520b57cec5SDimitry Andric }
17530b57cec5SDimitry Andric }
17540b57cec5SDimitry Andric
setGVProperties(llvm::GlobalValue * GV,GlobalDecl GD) const17550b57cec5SDimitry Andric void CodeGenModule::setGVProperties(llvm::GlobalValue *GV,
17560b57cec5SDimitry Andric GlobalDecl GD) const {
17570b57cec5SDimitry Andric setDLLImportDLLExport(GV, GD);
17580b57cec5SDimitry Andric setGVPropertiesAux(GV, dyn_cast<NamedDecl>(GD.getDecl()));
17590b57cec5SDimitry Andric }
17600b57cec5SDimitry Andric
setGVProperties(llvm::GlobalValue * GV,const NamedDecl * D) const17610b57cec5SDimitry Andric void CodeGenModule::setGVProperties(llvm::GlobalValue *GV,
17620b57cec5SDimitry Andric const NamedDecl *D) const {
17630b57cec5SDimitry Andric setDLLImportDLLExport(GV, D);
17640b57cec5SDimitry Andric setGVPropertiesAux(GV, D);
17650b57cec5SDimitry Andric }
17660b57cec5SDimitry Andric
setGVPropertiesAux(llvm::GlobalValue * GV,const NamedDecl * D) const17670b57cec5SDimitry Andric void CodeGenModule::setGVPropertiesAux(llvm::GlobalValue *GV,
17680b57cec5SDimitry Andric const NamedDecl *D) const {
17690b57cec5SDimitry Andric setGlobalVisibility(GV, D);
17700b57cec5SDimitry Andric setDSOLocal(GV);
17710b57cec5SDimitry Andric GV->setPartition(CodeGenOpts.SymbolPartition);
17720b57cec5SDimitry Andric }
17730b57cec5SDimitry Andric
GetLLVMTLSModel(StringRef S)17740b57cec5SDimitry Andric static llvm::GlobalVariable::ThreadLocalMode GetLLVMTLSModel(StringRef S) {
17750b57cec5SDimitry Andric return llvm::StringSwitch<llvm::GlobalVariable::ThreadLocalMode>(S)
17760b57cec5SDimitry Andric .Case("global-dynamic", llvm::GlobalVariable::GeneralDynamicTLSModel)
17770b57cec5SDimitry Andric .Case("local-dynamic", llvm::GlobalVariable::LocalDynamicTLSModel)
17780b57cec5SDimitry Andric .Case("initial-exec", llvm::GlobalVariable::InitialExecTLSModel)
17790b57cec5SDimitry Andric .Case("local-exec", llvm::GlobalVariable::LocalExecTLSModel);
17800b57cec5SDimitry Andric }
17810b57cec5SDimitry Andric
17825ffd83dbSDimitry Andric llvm::GlobalVariable::ThreadLocalMode
GetDefaultLLVMTLSModel() const17835ffd83dbSDimitry Andric CodeGenModule::GetDefaultLLVMTLSModel() const {
17845ffd83dbSDimitry Andric switch (CodeGenOpts.getDefaultTLSModel()) {
17850b57cec5SDimitry Andric case CodeGenOptions::GeneralDynamicTLSModel:
17860b57cec5SDimitry Andric return llvm::GlobalVariable::GeneralDynamicTLSModel;
17870b57cec5SDimitry Andric case CodeGenOptions::LocalDynamicTLSModel:
17880b57cec5SDimitry Andric return llvm::GlobalVariable::LocalDynamicTLSModel;
17890b57cec5SDimitry Andric case CodeGenOptions::InitialExecTLSModel:
17900b57cec5SDimitry Andric return llvm::GlobalVariable::InitialExecTLSModel;
17910b57cec5SDimitry Andric case CodeGenOptions::LocalExecTLSModel:
17920b57cec5SDimitry Andric return llvm::GlobalVariable::LocalExecTLSModel;
17930b57cec5SDimitry Andric }
17940b57cec5SDimitry Andric llvm_unreachable("Invalid TLS model!");
17950b57cec5SDimitry Andric }
17960b57cec5SDimitry Andric
setTLSMode(llvm::GlobalValue * GV,const VarDecl & D) const17970b57cec5SDimitry Andric void CodeGenModule::setTLSMode(llvm::GlobalValue *GV, const VarDecl &D) const {
17980b57cec5SDimitry Andric assert(D.getTLSKind() && "setting TLS mode on non-TLS var!");
17990b57cec5SDimitry Andric
18000b57cec5SDimitry Andric llvm::GlobalValue::ThreadLocalMode TLM;
18015ffd83dbSDimitry Andric TLM = GetDefaultLLVMTLSModel();
18020b57cec5SDimitry Andric
18030b57cec5SDimitry Andric // Override the TLS model if it is explicitly specified.
18040b57cec5SDimitry Andric if (const TLSModelAttr *Attr = D.getAttr<TLSModelAttr>()) {
18050b57cec5SDimitry Andric TLM = GetLLVMTLSModel(Attr->getModel());
18060b57cec5SDimitry Andric }
18070b57cec5SDimitry Andric
18080b57cec5SDimitry Andric GV->setThreadLocalMode(TLM);
18090b57cec5SDimitry Andric }
18100b57cec5SDimitry Andric
getCPUSpecificMangling(const CodeGenModule & CGM,StringRef Name)18110b57cec5SDimitry Andric static std::string getCPUSpecificMangling(const CodeGenModule &CGM,
18120b57cec5SDimitry Andric StringRef Name) {
18130b57cec5SDimitry Andric const TargetInfo &Target = CGM.getTarget();
18140b57cec5SDimitry Andric return (Twine('.') + Twine(Target.CPUSpecificManglingCharacter(Name))).str();
18150b57cec5SDimitry Andric }
18160b57cec5SDimitry Andric
AppendCPUSpecificCPUDispatchMangling(const CodeGenModule & CGM,const CPUSpecificAttr * Attr,unsigned CPUIndex,raw_ostream & Out)18170b57cec5SDimitry Andric static void AppendCPUSpecificCPUDispatchMangling(const CodeGenModule &CGM,
18180b57cec5SDimitry Andric const CPUSpecificAttr *Attr,
18190b57cec5SDimitry Andric unsigned CPUIndex,
18200b57cec5SDimitry Andric raw_ostream &Out) {
18210b57cec5SDimitry Andric // cpu_specific gets the current name, dispatch gets the resolver if IFunc is
18220b57cec5SDimitry Andric // supported.
18230b57cec5SDimitry Andric if (Attr)
18240b57cec5SDimitry Andric Out << getCPUSpecificMangling(CGM, Attr->getCPUName(CPUIndex)->getName());
18250b57cec5SDimitry Andric else if (CGM.getTarget().supportsIFunc())
18260b57cec5SDimitry Andric Out << ".resolver";
18270b57cec5SDimitry Andric }
18280b57cec5SDimitry Andric
1829fe6060f1SDimitry Andric // Returns true if GD is a function decl with internal linkage and
1830fe6060f1SDimitry Andric // needs a unique suffix after the mangled name.
isUniqueInternalLinkageDecl(GlobalDecl GD,CodeGenModule & CGM)1831fe6060f1SDimitry Andric static bool isUniqueInternalLinkageDecl(GlobalDecl GD,
1832fe6060f1SDimitry Andric CodeGenModule &CGM) {
1833fe6060f1SDimitry Andric const Decl *D = GD.getDecl();
1834fe6060f1SDimitry Andric return !CGM.getModuleNameHash().empty() && isa<FunctionDecl>(D) &&
1835fe6060f1SDimitry Andric (CGM.getFunctionLinkage(GD) == llvm::GlobalValue::InternalLinkage);
1836fe6060f1SDimitry Andric }
1837fe6060f1SDimitry Andric
getMangledNameImpl(CodeGenModule & CGM,GlobalDecl GD,const NamedDecl * ND,bool OmitMultiVersionMangling=false)1838fe6060f1SDimitry Andric static std::string getMangledNameImpl(CodeGenModule &CGM, GlobalDecl GD,
18390b57cec5SDimitry Andric const NamedDecl *ND,
18400b57cec5SDimitry Andric bool OmitMultiVersionMangling = false) {
18410b57cec5SDimitry Andric SmallString<256> Buffer;
18420b57cec5SDimitry Andric llvm::raw_svector_ostream Out(Buffer);
18430b57cec5SDimitry Andric MangleContext &MC = CGM.getCXXABI().getMangleContext();
1844fe6060f1SDimitry Andric if (!CGM.getModuleNameHash().empty())
1845fe6060f1SDimitry Andric MC.needsUniqueInternalLinkageNames();
1846fe6060f1SDimitry Andric bool ShouldMangle = MC.shouldMangleDeclName(ND);
1847fe6060f1SDimitry Andric if (ShouldMangle)
18485ffd83dbSDimitry Andric MC.mangleName(GD.getWithDecl(ND), Out);
18495ffd83dbSDimitry Andric else {
18500b57cec5SDimitry Andric IdentifierInfo *II = ND->getIdentifier();
18510b57cec5SDimitry Andric assert(II && "Attempt to mangle unnamed decl.");
18520b57cec5SDimitry Andric const auto *FD = dyn_cast<FunctionDecl>(ND);
18530b57cec5SDimitry Andric
18540b57cec5SDimitry Andric if (FD &&
18550b57cec5SDimitry Andric FD->getType()->castAs<FunctionType>()->getCallConv() == CC_X86RegCall) {
18565f757f3fSDimitry Andric if (CGM.getLangOpts().RegCall4)
18575f757f3fSDimitry Andric Out << "__regcall4__" << II->getName();
18585f757f3fSDimitry Andric else
18590b57cec5SDimitry Andric Out << "__regcall3__" << II->getName();
18605ffd83dbSDimitry Andric } else if (FD && FD->hasAttr<CUDAGlobalAttr>() &&
18615ffd83dbSDimitry Andric GD.getKernelReferenceKind() == KernelReferenceKind::Stub) {
18625ffd83dbSDimitry Andric Out << "__device_stub__" << II->getName();
18630b57cec5SDimitry Andric } else {
18640b57cec5SDimitry Andric Out << II->getName();
18650b57cec5SDimitry Andric }
18660b57cec5SDimitry Andric }
18670b57cec5SDimitry Andric
1868fe6060f1SDimitry Andric // Check if the module name hash should be appended for internal linkage
1869fe6060f1SDimitry Andric // symbols. This should come before multi-version target suffixes are
1870fe6060f1SDimitry Andric // appended. This is to keep the name and module hash suffix of the
1871fe6060f1SDimitry Andric // internal linkage function together. The unique suffix should only be
1872fe6060f1SDimitry Andric // added when name mangling is done to make sure that the final name can
1873fe6060f1SDimitry Andric // be properly demangled. For example, for C functions without prototypes,
1874fe6060f1SDimitry Andric // name mangling is not done and the unique suffix should not be appeneded
1875fe6060f1SDimitry Andric // then.
1876fe6060f1SDimitry Andric if (ShouldMangle && isUniqueInternalLinkageDecl(GD, CGM)) {
1877fe6060f1SDimitry Andric assert(CGM.getCodeGenOpts().UniqueInternalLinkageNames &&
1878fe6060f1SDimitry Andric "Hash computed when not explicitly requested");
1879fe6060f1SDimitry Andric Out << CGM.getModuleNameHash();
1880fe6060f1SDimitry Andric }
1881fe6060f1SDimitry Andric
18820b57cec5SDimitry Andric if (const auto *FD = dyn_cast<FunctionDecl>(ND))
18830b57cec5SDimitry Andric if (FD->isMultiVersion() && !OmitMultiVersionMangling) {
18840b57cec5SDimitry Andric switch (FD->getMultiVersionKind()) {
18850b57cec5SDimitry Andric case MultiVersionKind::CPUDispatch:
18860b57cec5SDimitry Andric case MultiVersionKind::CPUSpecific:
18870b57cec5SDimitry Andric AppendCPUSpecificCPUDispatchMangling(CGM,
18880b57cec5SDimitry Andric FD->getAttr<CPUSpecificAttr>(),
18890b57cec5SDimitry Andric GD.getMultiVersionIndex(), Out);
18900b57cec5SDimitry Andric break;
18910fca6ea1SDimitry Andric case MultiVersionKind::Target: {
18920fca6ea1SDimitry Andric auto *Attr = FD->getAttr<TargetAttr>();
18930fca6ea1SDimitry Andric assert(Attr && "Expected TargetAttr to be present "
18940fca6ea1SDimitry Andric "for attribute mangling");
18950fca6ea1SDimitry Andric const ABIInfo &Info = CGM.getTargetCodeGenInfo().getABIInfo();
18960fca6ea1SDimitry Andric Info.appendAttributeMangling(Attr, Out);
18970b57cec5SDimitry Andric break;
18980fca6ea1SDimitry Andric }
18990fca6ea1SDimitry Andric case MultiVersionKind::TargetVersion: {
19000fca6ea1SDimitry Andric auto *Attr = FD->getAttr<TargetVersionAttr>();
19010fca6ea1SDimitry Andric assert(Attr && "Expected TargetVersionAttr to be present "
19020fca6ea1SDimitry Andric "for attribute mangling");
19030fca6ea1SDimitry Andric const ABIInfo &Info = CGM.getTargetCodeGenInfo().getABIInfo();
19040fca6ea1SDimitry Andric Info.appendAttributeMangling(Attr, Out);
1905bdd1243dSDimitry Andric break;
19060fca6ea1SDimitry Andric }
19070fca6ea1SDimitry Andric case MultiVersionKind::TargetClones: {
19080fca6ea1SDimitry Andric auto *Attr = FD->getAttr<TargetClonesAttr>();
19090fca6ea1SDimitry Andric assert(Attr && "Expected TargetClonesAttr to be present "
19100fca6ea1SDimitry Andric "for attribute mangling");
19110fca6ea1SDimitry Andric unsigned Index = GD.getMultiVersionIndex();
19120fca6ea1SDimitry Andric const ABIInfo &Info = CGM.getTargetCodeGenInfo().getABIInfo();
19130fca6ea1SDimitry Andric Info.appendAttributeMangling(Attr, Index, Out);
19144824e7fdSDimitry Andric break;
19150fca6ea1SDimitry Andric }
19160b57cec5SDimitry Andric case MultiVersionKind::None:
19170b57cec5SDimitry Andric llvm_unreachable("None multiversion type isn't valid here");
19180b57cec5SDimitry Andric }
19190b57cec5SDimitry Andric }
19200b57cec5SDimitry Andric
1921fe6060f1SDimitry Andric // Make unique name for device side static file-scope variable for HIP.
192281ad6265SDimitry Andric if (CGM.getContext().shouldExternalize(ND) &&
1923fe6060f1SDimitry Andric CGM.getLangOpts().GPURelocatableDeviceCode &&
192481ad6265SDimitry Andric CGM.getLangOpts().CUDAIsDevice)
19252a66634dSDimitry Andric CGM.printPostfixForExternalizedDecl(Out, ND);
192681ad6265SDimitry Andric
19275ffd83dbSDimitry Andric return std::string(Out.str());
19280b57cec5SDimitry Andric }
19290b57cec5SDimitry Andric
UpdateMultiVersionNames(GlobalDecl GD,const FunctionDecl * FD,StringRef & CurName)19300b57cec5SDimitry Andric void CodeGenModule::UpdateMultiVersionNames(GlobalDecl GD,
193104eeddc0SDimitry Andric const FunctionDecl *FD,
193204eeddc0SDimitry Andric StringRef &CurName) {
19330b57cec5SDimitry Andric if (!FD->isMultiVersion())
19340b57cec5SDimitry Andric return;
19350b57cec5SDimitry Andric
19360b57cec5SDimitry Andric // Get the name of what this would be without the 'target' attribute. This
19370b57cec5SDimitry Andric // allows us to lookup the version that was emitted when this wasn't a
19380b57cec5SDimitry Andric // multiversion function.
19390b57cec5SDimitry Andric std::string NonTargetName =
19400b57cec5SDimitry Andric getMangledNameImpl(*this, GD, FD, /*OmitMultiVersionMangling=*/true);
19410b57cec5SDimitry Andric GlobalDecl OtherGD;
19420b57cec5SDimitry Andric if (lookupRepresentativeDecl(NonTargetName, OtherGD)) {
19430b57cec5SDimitry Andric assert(OtherGD.getCanonicalDecl()
19440b57cec5SDimitry Andric .getDecl()
19450b57cec5SDimitry Andric ->getAsFunction()
19460b57cec5SDimitry Andric ->isMultiVersion() &&
19470b57cec5SDimitry Andric "Other GD should now be a multiversioned function");
19480b57cec5SDimitry Andric // OtherFD is the version of this function that was mangled BEFORE
19490b57cec5SDimitry Andric // becoming a MultiVersion function. It potentially needs to be updated.
19500b57cec5SDimitry Andric const FunctionDecl *OtherFD = OtherGD.getCanonicalDecl()
19510b57cec5SDimitry Andric .getDecl()
19520b57cec5SDimitry Andric ->getAsFunction()
19530b57cec5SDimitry Andric ->getMostRecentDecl();
19540b57cec5SDimitry Andric std::string OtherName = getMangledNameImpl(*this, OtherGD, OtherFD);
19550b57cec5SDimitry Andric // This is so that if the initial version was already the 'default'
19560b57cec5SDimitry Andric // version, we don't try to update it.
19570b57cec5SDimitry Andric if (OtherName != NonTargetName) {
19580b57cec5SDimitry Andric // Remove instead of erase, since others may have stored the StringRef
19590b57cec5SDimitry Andric // to this.
19600b57cec5SDimitry Andric const auto ExistingRecord = Manglings.find(NonTargetName);
19610b57cec5SDimitry Andric if (ExistingRecord != std::end(Manglings))
19620b57cec5SDimitry Andric Manglings.remove(&(*ExistingRecord));
19630b57cec5SDimitry Andric auto Result = Manglings.insert(std::make_pair(OtherName, OtherGD));
196404eeddc0SDimitry Andric StringRef OtherNameRef = MangledDeclNames[OtherGD.getCanonicalDecl()] =
196504eeddc0SDimitry Andric Result.first->first();
196604eeddc0SDimitry Andric // If this is the current decl is being created, make sure we update the name.
196704eeddc0SDimitry Andric if (GD.getCanonicalDecl() == OtherGD.getCanonicalDecl())
196804eeddc0SDimitry Andric CurName = OtherNameRef;
19690b57cec5SDimitry Andric if (llvm::GlobalValue *Entry = GetGlobalValue(NonTargetName))
19700b57cec5SDimitry Andric Entry->setName(OtherName);
19710b57cec5SDimitry Andric }
19720b57cec5SDimitry Andric }
19730b57cec5SDimitry Andric }
19740b57cec5SDimitry Andric
getMangledName(GlobalDecl GD)19750b57cec5SDimitry Andric StringRef CodeGenModule::getMangledName(GlobalDecl GD) {
19760b57cec5SDimitry Andric GlobalDecl CanonicalGD = GD.getCanonicalDecl();
19770b57cec5SDimitry Andric
19780b57cec5SDimitry Andric // Some ABIs don't have constructor variants. Make sure that base and
19790b57cec5SDimitry Andric // complete constructors get mangled the same.
19800b57cec5SDimitry Andric if (const auto *CD = dyn_cast<CXXConstructorDecl>(CanonicalGD.getDecl())) {
19810b57cec5SDimitry Andric if (!getTarget().getCXXABI().hasConstructorVariants()) {
19820b57cec5SDimitry Andric CXXCtorType OrigCtorType = GD.getCtorType();
19830b57cec5SDimitry Andric assert(OrigCtorType == Ctor_Base || OrigCtorType == Ctor_Complete);
19840b57cec5SDimitry Andric if (OrigCtorType == Ctor_Base)
19850b57cec5SDimitry Andric CanonicalGD = GlobalDecl(CD, Ctor_Complete);
19860b57cec5SDimitry Andric }
19870b57cec5SDimitry Andric }
19880b57cec5SDimitry Andric
1989fe6060f1SDimitry Andric // In CUDA/HIP device compilation with -fgpu-rdc, the mangled name of a
1990fe6060f1SDimitry Andric // static device variable depends on whether the variable is referenced by
1991fe6060f1SDimitry Andric // a host or device host function. Therefore the mangled name cannot be
1992fe6060f1SDimitry Andric // cached.
199381ad6265SDimitry Andric if (!LangOpts.CUDAIsDevice || !getContext().mayExternalize(GD.getDecl())) {
19940b57cec5SDimitry Andric auto FoundName = MangledDeclNames.find(CanonicalGD);
19950b57cec5SDimitry Andric if (FoundName != MangledDeclNames.end())
19960b57cec5SDimitry Andric return FoundName->second;
1997fe6060f1SDimitry Andric }
19980b57cec5SDimitry Andric
19990b57cec5SDimitry Andric // Keep the first result in the case of a mangling collision.
20000b57cec5SDimitry Andric const auto *ND = cast<NamedDecl>(GD.getDecl());
20010b57cec5SDimitry Andric std::string MangledName = getMangledNameImpl(*this, GD, ND);
20020b57cec5SDimitry Andric
20035ffd83dbSDimitry Andric // Ensure either we have different ABIs between host and device compilations,
20045ffd83dbSDimitry Andric // says host compilation following MSVC ABI but device compilation follows
20055ffd83dbSDimitry Andric // Itanium C++ ABI or, if they follow the same ABI, kernel names after
20065ffd83dbSDimitry Andric // mangling should be the same after name stubbing. The later checking is
20075ffd83dbSDimitry Andric // very important as the device kernel name being mangled in host-compilation
20085ffd83dbSDimitry Andric // is used to resolve the device binaries to be executed. Inconsistent naming
20095ffd83dbSDimitry Andric // result in undefined behavior. Even though we cannot check that naming
20105ffd83dbSDimitry Andric // directly between host- and device-compilations, the host- and
20115ffd83dbSDimitry Andric // device-mangling in host compilation could help catching certain ones.
20125ffd83dbSDimitry Andric assert(!isa<FunctionDecl>(ND) || !ND->hasAttr<CUDAGlobalAttr>() ||
201381ad6265SDimitry Andric getContext().shouldExternalize(ND) || getLangOpts().CUDAIsDevice ||
20145ffd83dbSDimitry Andric (getContext().getAuxTargetInfo() &&
20155ffd83dbSDimitry Andric (getContext().getAuxTargetInfo()->getCXXABI() !=
20165ffd83dbSDimitry Andric getContext().getTargetInfo().getCXXABI())) ||
20175ffd83dbSDimitry Andric getCUDARuntime().getDeviceSideName(ND) ==
20185ffd83dbSDimitry Andric getMangledNameImpl(
20195ffd83dbSDimitry Andric *this,
20205ffd83dbSDimitry Andric GD.getWithKernelReferenceKind(KernelReferenceKind::Kernel),
20215ffd83dbSDimitry Andric ND));
20220b57cec5SDimitry Andric
20230b57cec5SDimitry Andric auto Result = Manglings.insert(std::make_pair(MangledName, GD));
20240b57cec5SDimitry Andric return MangledDeclNames[CanonicalGD] = Result.first->first();
20250b57cec5SDimitry Andric }
20260b57cec5SDimitry Andric
getBlockMangledName(GlobalDecl GD,const BlockDecl * BD)20270b57cec5SDimitry Andric StringRef CodeGenModule::getBlockMangledName(GlobalDecl GD,
20280b57cec5SDimitry Andric const BlockDecl *BD) {
20290b57cec5SDimitry Andric MangleContext &MangleCtx = getCXXABI().getMangleContext();
20300b57cec5SDimitry Andric const Decl *D = GD.getDecl();
20310b57cec5SDimitry Andric
20320b57cec5SDimitry Andric SmallString<256> Buffer;
20330b57cec5SDimitry Andric llvm::raw_svector_ostream Out(Buffer);
20340b57cec5SDimitry Andric if (!D)
20350b57cec5SDimitry Andric MangleCtx.mangleGlobalBlock(BD,
20360b57cec5SDimitry Andric dyn_cast_or_null<VarDecl>(initializedGlobalDecl.getDecl()), Out);
20370b57cec5SDimitry Andric else if (const auto *CD = dyn_cast<CXXConstructorDecl>(D))
20380b57cec5SDimitry Andric MangleCtx.mangleCtorBlock(CD, GD.getCtorType(), BD, Out);
20390b57cec5SDimitry Andric else if (const auto *DD = dyn_cast<CXXDestructorDecl>(D))
20400b57cec5SDimitry Andric MangleCtx.mangleDtorBlock(DD, GD.getDtorType(), BD, Out);
20410b57cec5SDimitry Andric else
20420b57cec5SDimitry Andric MangleCtx.mangleBlock(cast<DeclContext>(D), BD, Out);
20430b57cec5SDimitry Andric
20440b57cec5SDimitry Andric auto Result = Manglings.insert(std::make_pair(Out.str(), BD));
20450b57cec5SDimitry Andric return Result.first->first();
20460b57cec5SDimitry Andric }
20470b57cec5SDimitry Andric
getMangledNameDecl(StringRef Name)204881ad6265SDimitry Andric const GlobalDecl CodeGenModule::getMangledNameDecl(StringRef Name) {
204981ad6265SDimitry Andric auto it = MangledDeclNames.begin();
205081ad6265SDimitry Andric while (it != MangledDeclNames.end()) {
205181ad6265SDimitry Andric if (it->second == Name)
205281ad6265SDimitry Andric return it->first;
205381ad6265SDimitry Andric it++;
205481ad6265SDimitry Andric }
205581ad6265SDimitry Andric return GlobalDecl();
205681ad6265SDimitry Andric }
205781ad6265SDimitry Andric
GetGlobalValue(StringRef Name)20580b57cec5SDimitry Andric llvm::GlobalValue *CodeGenModule::GetGlobalValue(StringRef Name) {
20590b57cec5SDimitry Andric return getModule().getNamedValue(Name);
20600b57cec5SDimitry Andric }
20610b57cec5SDimitry Andric
20620b57cec5SDimitry Andric /// AddGlobalCtor - Add a function to the list that will be called before
20630b57cec5SDimitry Andric /// main() runs.
AddGlobalCtor(llvm::Function * Ctor,int Priority,unsigned LexOrder,llvm::Constant * AssociatedData)20640b57cec5SDimitry Andric void CodeGenModule::AddGlobalCtor(llvm::Function *Ctor, int Priority,
2065bdd1243dSDimitry Andric unsigned LexOrder,
20660b57cec5SDimitry Andric llvm::Constant *AssociatedData) {
20670b57cec5SDimitry Andric // FIXME: Type coercion of void()* types.
2068bdd1243dSDimitry Andric GlobalCtors.push_back(Structor(Priority, LexOrder, Ctor, AssociatedData));
20690b57cec5SDimitry Andric }
20700b57cec5SDimitry Andric
20710b57cec5SDimitry Andric /// AddGlobalDtor - Add a function to the list that will be called
20720b57cec5SDimitry Andric /// when the module is unloaded.
AddGlobalDtor(llvm::Function * Dtor,int Priority,bool IsDtorAttrFunc)2073e8d8bef9SDimitry Andric void CodeGenModule::AddGlobalDtor(llvm::Function *Dtor, int Priority,
2074e8d8bef9SDimitry Andric bool IsDtorAttrFunc) {
2075e8d8bef9SDimitry Andric if (CodeGenOpts.RegisterGlobalDtorsWithAtExit &&
2076e8d8bef9SDimitry Andric (!getContext().getTargetInfo().getTriple().isOSAIX() || IsDtorAttrFunc)) {
20770b57cec5SDimitry Andric DtorsUsingAtExit[Priority].push_back(Dtor);
20780b57cec5SDimitry Andric return;
20790b57cec5SDimitry Andric }
20800b57cec5SDimitry Andric
20810b57cec5SDimitry Andric // FIXME: Type coercion of void()* types.
2082bdd1243dSDimitry Andric GlobalDtors.push_back(Structor(Priority, ~0U, Dtor, nullptr));
20830b57cec5SDimitry Andric }
20840b57cec5SDimitry Andric
EmitCtorList(CtorList & Fns,const char * GlobalName)20850b57cec5SDimitry Andric void CodeGenModule::EmitCtorList(CtorList &Fns, const char *GlobalName) {
20860b57cec5SDimitry Andric if (Fns.empty()) return;
20870b57cec5SDimitry Andric
20880b57cec5SDimitry Andric // Ctor function type is void()*.
20890b57cec5SDimitry Andric llvm::FunctionType* CtorFTy = llvm::FunctionType::get(VoidTy, false);
20900b57cec5SDimitry Andric llvm::Type *CtorPFTy = llvm::PointerType::get(CtorFTy,
20910b57cec5SDimitry Andric TheModule.getDataLayout().getProgramAddressSpace());
20920b57cec5SDimitry Andric
20930b57cec5SDimitry Andric // Get the type of a ctor entry, { i32, void ()*, i8* }.
20940b57cec5SDimitry Andric llvm::StructType *CtorStructTy = llvm::StructType::get(
20950b57cec5SDimitry Andric Int32Ty, CtorPFTy, VoidPtrTy);
20960b57cec5SDimitry Andric
20970b57cec5SDimitry Andric // Construct the constructor and destructor arrays.
20980b57cec5SDimitry Andric ConstantInitBuilder builder(*this);
20990b57cec5SDimitry Andric auto ctors = builder.beginArray(CtorStructTy);
21000b57cec5SDimitry Andric for (const auto &I : Fns) {
21010b57cec5SDimitry Andric auto ctor = ctors.beginStruct(CtorStructTy);
21020b57cec5SDimitry Andric ctor.addInt(Int32Ty, I.Priority);
21035f757f3fSDimitry Andric ctor.add(I.Initializer);
21040b57cec5SDimitry Andric if (I.AssociatedData)
21055f757f3fSDimitry Andric ctor.add(I.AssociatedData);
21060b57cec5SDimitry Andric else
21070b57cec5SDimitry Andric ctor.addNullPointer(VoidPtrTy);
21080b57cec5SDimitry Andric ctor.finishAndAddTo(ctors);
21090b57cec5SDimitry Andric }
21100b57cec5SDimitry Andric
21110b57cec5SDimitry Andric auto list =
21120b57cec5SDimitry Andric ctors.finishAndCreateGlobal(GlobalName, getPointerAlign(),
21130b57cec5SDimitry Andric /*constant*/ false,
21140b57cec5SDimitry Andric llvm::GlobalValue::AppendingLinkage);
21150b57cec5SDimitry Andric
21160b57cec5SDimitry Andric // The LTO linker doesn't seem to like it when we set an alignment
21170b57cec5SDimitry Andric // on appending variables. Take it off as a workaround.
2118bdd1243dSDimitry Andric list->setAlignment(std::nullopt);
21190b57cec5SDimitry Andric
21200b57cec5SDimitry Andric Fns.clear();
21210b57cec5SDimitry Andric }
21220b57cec5SDimitry Andric
21230b57cec5SDimitry Andric llvm::GlobalValue::LinkageTypes
getFunctionLinkage(GlobalDecl GD)21240b57cec5SDimitry Andric CodeGenModule::getFunctionLinkage(GlobalDecl GD) {
21250b57cec5SDimitry Andric const auto *D = cast<FunctionDecl>(GD.getDecl());
21260b57cec5SDimitry Andric
21270b57cec5SDimitry Andric GVALinkage Linkage = getContext().GetGVALinkageForFunction(D);
21280b57cec5SDimitry Andric
21290b57cec5SDimitry Andric if (const auto *Dtor = dyn_cast<CXXDestructorDecl>(D))
21300b57cec5SDimitry Andric return getCXXABI().getCXXDestructorLinkage(Linkage, Dtor, GD.getDtorType());
21310b57cec5SDimitry Andric
21328a4dda33SDimitry Andric return getLLVMLinkageForDeclarator(D, Linkage);
21330b57cec5SDimitry Andric }
21340b57cec5SDimitry Andric
CreateCrossDsoCfiTypeId(llvm::Metadata * MD)21350b57cec5SDimitry Andric llvm::ConstantInt *CodeGenModule::CreateCrossDsoCfiTypeId(llvm::Metadata *MD) {
21360b57cec5SDimitry Andric llvm::MDString *MDS = dyn_cast<llvm::MDString>(MD);
21370b57cec5SDimitry Andric if (!MDS) return nullptr;
21380b57cec5SDimitry Andric
21390b57cec5SDimitry Andric return llvm::ConstantInt::get(Int64Ty, llvm::MD5Hash(MDS->getString()));
21400b57cec5SDimitry Andric }
21410b57cec5SDimitry Andric
CreateKCFITypeId(QualType T)2142bdd1243dSDimitry Andric llvm::ConstantInt *CodeGenModule::CreateKCFITypeId(QualType T) {
2143bdd1243dSDimitry Andric if (auto *FnType = T->getAs<FunctionProtoType>())
2144bdd1243dSDimitry Andric T = getContext().getFunctionType(
2145bdd1243dSDimitry Andric FnType->getReturnType(), FnType->getParamTypes(),
2146bdd1243dSDimitry Andric FnType->getExtProtoInfo().withExceptionSpec(EST_None));
2147bdd1243dSDimitry Andric
2148bdd1243dSDimitry Andric std::string OutName;
2149bdd1243dSDimitry Andric llvm::raw_string_ostream Out(OutName);
21505f757f3fSDimitry Andric getCXXABI().getMangleContext().mangleCanonicalTypeName(
215106c3fb27SDimitry Andric T, Out, getCodeGenOpts().SanitizeCfiICallNormalizeIntegers);
215206c3fb27SDimitry Andric
215306c3fb27SDimitry Andric if (getCodeGenOpts().SanitizeCfiICallNormalizeIntegers)
215406c3fb27SDimitry Andric Out << ".normalized";
2155bdd1243dSDimitry Andric
2156bdd1243dSDimitry Andric return llvm::ConstantInt::get(Int32Ty,
2157bdd1243dSDimitry Andric static_cast<uint32_t>(llvm::xxHash64(OutName)));
2158bdd1243dSDimitry Andric }
2159bdd1243dSDimitry Andric
SetLLVMFunctionAttributes(GlobalDecl GD,const CGFunctionInfo & Info,llvm::Function * F,bool IsThunk)21600b57cec5SDimitry Andric void CodeGenModule::SetLLVMFunctionAttributes(GlobalDecl GD,
21610b57cec5SDimitry Andric const CGFunctionInfo &Info,
2162fe6060f1SDimitry Andric llvm::Function *F, bool IsThunk) {
21630b57cec5SDimitry Andric unsigned CallingConv;
21640b57cec5SDimitry Andric llvm::AttributeList PAL;
2165fe6060f1SDimitry Andric ConstructAttributeList(F->getName(), Info, GD, PAL, CallingConv,
2166fe6060f1SDimitry Andric /*AttrOnCallSite=*/false, IsThunk);
21670fca6ea1SDimitry Andric if (CallingConv == llvm::CallingConv::X86_VectorCall &&
21680fca6ea1SDimitry Andric getTarget().getTriple().isWindowsArm64EC()) {
21690fca6ea1SDimitry Andric SourceLocation Loc;
21700fca6ea1SDimitry Andric if (const Decl *D = GD.getDecl())
21710fca6ea1SDimitry Andric Loc = D->getLocation();
21720fca6ea1SDimitry Andric
21730fca6ea1SDimitry Andric Error(Loc, "__vectorcall calling convention is not currently supported");
21740fca6ea1SDimitry Andric }
21750b57cec5SDimitry Andric F->setAttributes(PAL);
21760b57cec5SDimitry Andric F->setCallingConv(static_cast<llvm::CallingConv::ID>(CallingConv));
21770b57cec5SDimitry Andric }
21780b57cec5SDimitry Andric
removeImageAccessQualifier(std::string & TyName)21790b57cec5SDimitry Andric static void removeImageAccessQualifier(std::string& TyName) {
21800b57cec5SDimitry Andric std::string ReadOnlyQual("__read_only");
21810b57cec5SDimitry Andric std::string::size_type ReadOnlyPos = TyName.find(ReadOnlyQual);
21820b57cec5SDimitry Andric if (ReadOnlyPos != std::string::npos)
21830b57cec5SDimitry Andric // "+ 1" for the space after access qualifier.
21840b57cec5SDimitry Andric TyName.erase(ReadOnlyPos, ReadOnlyQual.size() + 1);
21850b57cec5SDimitry Andric else {
21860b57cec5SDimitry Andric std::string WriteOnlyQual("__write_only");
21870b57cec5SDimitry Andric std::string::size_type WriteOnlyPos = TyName.find(WriteOnlyQual);
21880b57cec5SDimitry Andric if (WriteOnlyPos != std::string::npos)
21890b57cec5SDimitry Andric TyName.erase(WriteOnlyPos, WriteOnlyQual.size() + 1);
21900b57cec5SDimitry Andric else {
21910b57cec5SDimitry Andric std::string ReadWriteQual("__read_write");
21920b57cec5SDimitry Andric std::string::size_type ReadWritePos = TyName.find(ReadWriteQual);
21930b57cec5SDimitry Andric if (ReadWritePos != std::string::npos)
21940b57cec5SDimitry Andric TyName.erase(ReadWritePos, ReadWriteQual.size() + 1);
21950b57cec5SDimitry Andric }
21960b57cec5SDimitry Andric }
21970b57cec5SDimitry Andric }
21980b57cec5SDimitry Andric
21990b57cec5SDimitry Andric // Returns the address space id that should be produced to the
22000b57cec5SDimitry Andric // kernel_arg_addr_space metadata. This is always fixed to the ids
22010b57cec5SDimitry Andric // as specified in the SPIR 2.0 specification in order to differentiate
22020b57cec5SDimitry Andric // for example in clGetKernelArgInfo() implementation between the address
22030b57cec5SDimitry Andric // spaces with targets without unique mapping to the OpenCL address spaces
22040b57cec5SDimitry Andric // (basically all single AS CPUs).
ArgInfoAddressSpace(LangAS AS)22050b57cec5SDimitry Andric static unsigned ArgInfoAddressSpace(LangAS AS) {
22060b57cec5SDimitry Andric switch (AS) {
2207e8d8bef9SDimitry Andric case LangAS::opencl_global:
2208e8d8bef9SDimitry Andric return 1;
2209e8d8bef9SDimitry Andric case LangAS::opencl_constant:
2210e8d8bef9SDimitry Andric return 2;
2211e8d8bef9SDimitry Andric case LangAS::opencl_local:
2212e8d8bef9SDimitry Andric return 3;
2213e8d8bef9SDimitry Andric case LangAS::opencl_generic:
2214e8d8bef9SDimitry Andric return 4; // Not in SPIR 2.0 specs.
2215e8d8bef9SDimitry Andric case LangAS::opencl_global_device:
2216e8d8bef9SDimitry Andric return 5;
2217e8d8bef9SDimitry Andric case LangAS::opencl_global_host:
2218e8d8bef9SDimitry Andric return 6;
22190b57cec5SDimitry Andric default:
22200b57cec5SDimitry Andric return 0; // Assume private.
22210b57cec5SDimitry Andric }
22220b57cec5SDimitry Andric }
22230b57cec5SDimitry Andric
GenKernelArgMetadata(llvm::Function * Fn,const FunctionDecl * FD,CodeGenFunction * CGF)222481ad6265SDimitry Andric void CodeGenModule::GenKernelArgMetadata(llvm::Function *Fn,
22250b57cec5SDimitry Andric const FunctionDecl *FD,
22260b57cec5SDimitry Andric CodeGenFunction *CGF) {
22270b57cec5SDimitry Andric assert(((FD && CGF) || (!FD && !CGF)) &&
22280b57cec5SDimitry Andric "Incorrect use - FD and CGF should either be both null or not!");
22290b57cec5SDimitry Andric // Create MDNodes that represent the kernel arg metadata.
22300b57cec5SDimitry Andric // Each MDNode is a list in the form of "key", N number of values which is
22310b57cec5SDimitry Andric // the same number of values as their are kernel arguments.
22320b57cec5SDimitry Andric
22330b57cec5SDimitry Andric const PrintingPolicy &Policy = Context.getPrintingPolicy();
22340b57cec5SDimitry Andric
22350b57cec5SDimitry Andric // MDNode for the kernel argument address space qualifiers.
22360b57cec5SDimitry Andric SmallVector<llvm::Metadata *, 8> addressQuals;
22370b57cec5SDimitry Andric
22380b57cec5SDimitry Andric // MDNode for the kernel argument access qualifiers (images only).
22390b57cec5SDimitry Andric SmallVector<llvm::Metadata *, 8> accessQuals;
22400b57cec5SDimitry Andric
22410b57cec5SDimitry Andric // MDNode for the kernel argument type names.
22420b57cec5SDimitry Andric SmallVector<llvm::Metadata *, 8> argTypeNames;
22430b57cec5SDimitry Andric
22440b57cec5SDimitry Andric // MDNode for the kernel argument base type names.
22450b57cec5SDimitry Andric SmallVector<llvm::Metadata *, 8> argBaseTypeNames;
22460b57cec5SDimitry Andric
22470b57cec5SDimitry Andric // MDNode for the kernel argument type qualifiers.
22480b57cec5SDimitry Andric SmallVector<llvm::Metadata *, 8> argTypeQuals;
22490b57cec5SDimitry Andric
22500b57cec5SDimitry Andric // MDNode for the kernel argument names.
22510b57cec5SDimitry Andric SmallVector<llvm::Metadata *, 8> argNames;
22520b57cec5SDimitry Andric
22530b57cec5SDimitry Andric if (FD && CGF)
22540b57cec5SDimitry Andric for (unsigned i = 0, e = FD->getNumParams(); i != e; ++i) {
22550b57cec5SDimitry Andric const ParmVarDecl *parm = FD->getParamDecl(i);
225681ad6265SDimitry Andric // Get argument name.
225781ad6265SDimitry Andric argNames.push_back(llvm::MDString::get(VMContext, parm->getName()));
225881ad6265SDimitry Andric
225981ad6265SDimitry Andric if (!getLangOpts().OpenCL)
226081ad6265SDimitry Andric continue;
22610b57cec5SDimitry Andric QualType ty = parm->getType();
22620b57cec5SDimitry Andric std::string typeQuals;
22630b57cec5SDimitry Andric
2264fe6060f1SDimitry Andric // Get image and pipe access qualifier:
2265fe6060f1SDimitry Andric if (ty->isImageType() || ty->isPipeType()) {
2266fe6060f1SDimitry Andric const Decl *PDecl = parm;
2267bdd1243dSDimitry Andric if (const auto *TD = ty->getAs<TypedefType>())
2268fe6060f1SDimitry Andric PDecl = TD->getDecl();
2269fe6060f1SDimitry Andric const OpenCLAccessAttr *A = PDecl->getAttr<OpenCLAccessAttr>();
2270fe6060f1SDimitry Andric if (A && A->isWriteOnly())
2271fe6060f1SDimitry Andric accessQuals.push_back(llvm::MDString::get(VMContext, "write_only"));
2272fe6060f1SDimitry Andric else if (A && A->isReadWrite())
2273fe6060f1SDimitry Andric accessQuals.push_back(llvm::MDString::get(VMContext, "read_write"));
2274fe6060f1SDimitry Andric else
2275fe6060f1SDimitry Andric accessQuals.push_back(llvm::MDString::get(VMContext, "read_only"));
2276fe6060f1SDimitry Andric } else
2277fe6060f1SDimitry Andric accessQuals.push_back(llvm::MDString::get(VMContext, "none"));
2278fe6060f1SDimitry Andric
2279fe6060f1SDimitry Andric auto getTypeSpelling = [&](QualType Ty) {
2280fe6060f1SDimitry Andric auto typeName = Ty.getUnqualifiedType().getAsString(Policy);
2281fe6060f1SDimitry Andric
2282fe6060f1SDimitry Andric if (Ty.isCanonical()) {
2283fe6060f1SDimitry Andric StringRef typeNameRef = typeName;
2284fe6060f1SDimitry Andric // Turn "unsigned type" to "utype"
2285fe6060f1SDimitry Andric if (typeNameRef.consume_front("unsigned "))
2286fe6060f1SDimitry Andric return std::string("u") + typeNameRef.str();
2287fe6060f1SDimitry Andric if (typeNameRef.consume_front("signed "))
2288fe6060f1SDimitry Andric return typeNameRef.str();
2289fe6060f1SDimitry Andric }
2290fe6060f1SDimitry Andric
2291fe6060f1SDimitry Andric return typeName;
2292fe6060f1SDimitry Andric };
2293fe6060f1SDimitry Andric
22940b57cec5SDimitry Andric if (ty->isPointerType()) {
22950b57cec5SDimitry Andric QualType pointeeTy = ty->getPointeeType();
22960b57cec5SDimitry Andric
22970b57cec5SDimitry Andric // Get address qualifier.
22980b57cec5SDimitry Andric addressQuals.push_back(
22990b57cec5SDimitry Andric llvm::ConstantAsMetadata::get(CGF->Builder.getInt32(
23000b57cec5SDimitry Andric ArgInfoAddressSpace(pointeeTy.getAddressSpace()))));
23010b57cec5SDimitry Andric
23020b57cec5SDimitry Andric // Get argument type name.
2303fe6060f1SDimitry Andric std::string typeName = getTypeSpelling(pointeeTy) + "*";
23040b57cec5SDimitry Andric std::string baseTypeName =
2305fe6060f1SDimitry Andric getTypeSpelling(pointeeTy.getCanonicalType()) + "*";
2306fe6060f1SDimitry Andric argTypeNames.push_back(llvm::MDString::get(VMContext, typeName));
23070b57cec5SDimitry Andric argBaseTypeNames.push_back(
23080b57cec5SDimitry Andric llvm::MDString::get(VMContext, baseTypeName));
23090b57cec5SDimitry Andric
23100b57cec5SDimitry Andric // Get argument type qualifiers:
23110b57cec5SDimitry Andric if (ty.isRestrictQualified())
23120b57cec5SDimitry Andric typeQuals = "restrict";
23130b57cec5SDimitry Andric if (pointeeTy.isConstQualified() ||
23140b57cec5SDimitry Andric (pointeeTy.getAddressSpace() == LangAS::opencl_constant))
23150b57cec5SDimitry Andric typeQuals += typeQuals.empty() ? "const" : " const";
23160b57cec5SDimitry Andric if (pointeeTy.isVolatileQualified())
23170b57cec5SDimitry Andric typeQuals += typeQuals.empty() ? "volatile" : " volatile";
23180b57cec5SDimitry Andric } else {
23190b57cec5SDimitry Andric uint32_t AddrSpc = 0;
23200b57cec5SDimitry Andric bool isPipe = ty->isPipeType();
23210b57cec5SDimitry Andric if (ty->isImageType() || isPipe)
23220b57cec5SDimitry Andric AddrSpc = ArgInfoAddressSpace(LangAS::opencl_global);
23230b57cec5SDimitry Andric
23240b57cec5SDimitry Andric addressQuals.push_back(
23250b57cec5SDimitry Andric llvm::ConstantAsMetadata::get(CGF->Builder.getInt32(AddrSpc)));
23260b57cec5SDimitry Andric
23270b57cec5SDimitry Andric // Get argument type name.
2328fe6060f1SDimitry Andric ty = isPipe ? ty->castAs<PipeType>()->getElementType() : ty;
2329fe6060f1SDimitry Andric std::string typeName = getTypeSpelling(ty);
2330fe6060f1SDimitry Andric std::string baseTypeName = getTypeSpelling(ty.getCanonicalType());
23310b57cec5SDimitry Andric
23320b57cec5SDimitry Andric // Remove access qualifiers on images
23330b57cec5SDimitry Andric // (as they are inseparable from type in clang implementation,
23340b57cec5SDimitry Andric // but OpenCL spec provides a special query to get access qualifier
23350b57cec5SDimitry Andric // via clGetKernelArgInfo with CL_KERNEL_ARG_ACCESS_QUALIFIER):
23360b57cec5SDimitry Andric if (ty->isImageType()) {
23370b57cec5SDimitry Andric removeImageAccessQualifier(typeName);
23380b57cec5SDimitry Andric removeImageAccessQualifier(baseTypeName);
23390b57cec5SDimitry Andric }
23400b57cec5SDimitry Andric
23410b57cec5SDimitry Andric argTypeNames.push_back(llvm::MDString::get(VMContext, typeName));
23420b57cec5SDimitry Andric argBaseTypeNames.push_back(
23430b57cec5SDimitry Andric llvm::MDString::get(VMContext, baseTypeName));
23440b57cec5SDimitry Andric
23450b57cec5SDimitry Andric if (isPipe)
23460b57cec5SDimitry Andric typeQuals = "pipe";
23470b57cec5SDimitry Andric }
23480b57cec5SDimitry Andric argTypeQuals.push_back(llvm::MDString::get(VMContext, typeQuals));
23490b57cec5SDimitry Andric }
23500b57cec5SDimitry Andric
235181ad6265SDimitry Andric if (getLangOpts().OpenCL) {
23520b57cec5SDimitry Andric Fn->setMetadata("kernel_arg_addr_space",
23530b57cec5SDimitry Andric llvm::MDNode::get(VMContext, addressQuals));
23540b57cec5SDimitry Andric Fn->setMetadata("kernel_arg_access_qual",
23550b57cec5SDimitry Andric llvm::MDNode::get(VMContext, accessQuals));
23560b57cec5SDimitry Andric Fn->setMetadata("kernel_arg_type",
23570b57cec5SDimitry Andric llvm::MDNode::get(VMContext, argTypeNames));
23580b57cec5SDimitry Andric Fn->setMetadata("kernel_arg_base_type",
23590b57cec5SDimitry Andric llvm::MDNode::get(VMContext, argBaseTypeNames));
23600b57cec5SDimitry Andric Fn->setMetadata("kernel_arg_type_qual",
23610b57cec5SDimitry Andric llvm::MDNode::get(VMContext, argTypeQuals));
236281ad6265SDimitry Andric }
236381ad6265SDimitry Andric if (getCodeGenOpts().EmitOpenCLArgMetadata ||
236481ad6265SDimitry Andric getCodeGenOpts().HIPSaveKernelArgName)
23650b57cec5SDimitry Andric Fn->setMetadata("kernel_arg_name",
23660b57cec5SDimitry Andric llvm::MDNode::get(VMContext, argNames));
23670b57cec5SDimitry Andric }
23680b57cec5SDimitry Andric
23690b57cec5SDimitry Andric /// Determines whether the language options require us to model
23700b57cec5SDimitry Andric /// unwind exceptions. We treat -fexceptions as mandating this
23710b57cec5SDimitry Andric /// except under the fragile ObjC ABI with only ObjC exceptions
23720b57cec5SDimitry Andric /// enabled. This means, for example, that C with -fexceptions
23730b57cec5SDimitry Andric /// enables this.
hasUnwindExceptions(const LangOptions & LangOpts)23740b57cec5SDimitry Andric static bool hasUnwindExceptions(const LangOptions &LangOpts) {
23750b57cec5SDimitry Andric // If exceptions are completely disabled, obviously this is false.
23760b57cec5SDimitry Andric if (!LangOpts.Exceptions) return false;
23770b57cec5SDimitry Andric
23780b57cec5SDimitry Andric // If C++ exceptions are enabled, this is true.
23790b57cec5SDimitry Andric if (LangOpts.CXXExceptions) return true;
23800b57cec5SDimitry Andric
23810b57cec5SDimitry Andric // If ObjC exceptions are enabled, this depends on the ABI.
23820b57cec5SDimitry Andric if (LangOpts.ObjCExceptions) {
23830b57cec5SDimitry Andric return LangOpts.ObjCRuntime.hasUnwindExceptions();
23840b57cec5SDimitry Andric }
23850b57cec5SDimitry Andric
23860b57cec5SDimitry Andric return true;
23870b57cec5SDimitry Andric }
23880b57cec5SDimitry Andric
requiresMemberFunctionPointerTypeMetadata(CodeGenModule & CGM,const CXXMethodDecl * MD)23890b57cec5SDimitry Andric static bool requiresMemberFunctionPointerTypeMetadata(CodeGenModule &CGM,
23900b57cec5SDimitry Andric const CXXMethodDecl *MD) {
23910b57cec5SDimitry Andric // Check that the type metadata can ever actually be used by a call.
23920b57cec5SDimitry Andric if (!CGM.getCodeGenOpts().LTOUnit ||
23930b57cec5SDimitry Andric !CGM.HasHiddenLTOVisibility(MD->getParent()))
23940b57cec5SDimitry Andric return false;
23950b57cec5SDimitry Andric
23960b57cec5SDimitry Andric // Only functions whose address can be taken with a member function pointer
23970b57cec5SDimitry Andric // need this sort of type metadata.
23985f757f3fSDimitry Andric return MD->isImplicitObjectMemberFunction() && !MD->isVirtual() &&
23995f757f3fSDimitry Andric !isa<CXXConstructorDecl, CXXDestructorDecl>(MD);
24000b57cec5SDimitry Andric }
24010b57cec5SDimitry Andric
24025f757f3fSDimitry Andric SmallVector<const CXXRecordDecl *, 0>
getMostBaseClasses(const CXXRecordDecl * RD)24030b57cec5SDimitry Andric CodeGenModule::getMostBaseClasses(const CXXRecordDecl *RD) {
24040b57cec5SDimitry Andric llvm::SetVector<const CXXRecordDecl *> MostBases;
24050b57cec5SDimitry Andric
24060b57cec5SDimitry Andric std::function<void (const CXXRecordDecl *)> CollectMostBases;
24070b57cec5SDimitry Andric CollectMostBases = [&](const CXXRecordDecl *RD) {
24080b57cec5SDimitry Andric if (RD->getNumBases() == 0)
24090b57cec5SDimitry Andric MostBases.insert(RD);
24100b57cec5SDimitry Andric for (const CXXBaseSpecifier &B : RD->bases())
24110b57cec5SDimitry Andric CollectMostBases(B.getType()->getAsCXXRecordDecl());
24120b57cec5SDimitry Andric };
24130b57cec5SDimitry Andric CollectMostBases(RD);
24140b57cec5SDimitry Andric return MostBases.takeVector();
24150b57cec5SDimitry Andric }
24160b57cec5SDimitry Andric
SetLLVMFunctionAttributesForDefinition(const Decl * D,llvm::Function * F)24170b57cec5SDimitry Andric void CodeGenModule::SetLLVMFunctionAttributesForDefinition(const Decl *D,
24180b57cec5SDimitry Andric llvm::Function *F) {
241904eeddc0SDimitry Andric llvm::AttrBuilder B(F->getContext());
24200b57cec5SDimitry Andric
2421bdd1243dSDimitry Andric if ((!D || !D->hasAttr<NoUwtableAttr>()) && CodeGenOpts.UnwindTables)
242281ad6265SDimitry Andric B.addUWTableAttr(llvm::UWTableKind(CodeGenOpts.UnwindTables));
24230b57cec5SDimitry Andric
24245ffd83dbSDimitry Andric if (CodeGenOpts.StackClashProtector)
24255ffd83dbSDimitry Andric B.addAttribute("probe-stack", "inline-asm");
24265ffd83dbSDimitry Andric
24275f757f3fSDimitry Andric if (CodeGenOpts.StackProbeSize && CodeGenOpts.StackProbeSize != 4096)
24285f757f3fSDimitry Andric B.addAttribute("stack-probe-size",
24295f757f3fSDimitry Andric std::to_string(CodeGenOpts.StackProbeSize));
24305f757f3fSDimitry Andric
24310b57cec5SDimitry Andric if (!hasUnwindExceptions(LangOpts))
24320b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::NoUnwind);
24330b57cec5SDimitry Andric
2434bdd1243dSDimitry Andric if (D && D->hasAttr<NoStackProtectorAttr>())
2435bdd1243dSDimitry Andric ; // Do nothing.
2436bdd1243dSDimitry Andric else if (D && D->hasAttr<StrictGuardStackCheckAttr>() &&
24375f757f3fSDimitry Andric isStackProtectorOn(LangOpts, getTriple(), LangOptions::SSPOn))
2438bdd1243dSDimitry Andric B.addAttribute(llvm::Attribute::StackProtectStrong);
24395f757f3fSDimitry Andric else if (isStackProtectorOn(LangOpts, getTriple(), LangOptions::SSPOn))
24400b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::StackProtect);
24415f757f3fSDimitry Andric else if (isStackProtectorOn(LangOpts, getTriple(), LangOptions::SSPStrong))
24420b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::StackProtectStrong);
24435f757f3fSDimitry Andric else if (isStackProtectorOn(LangOpts, getTriple(), LangOptions::SSPReq))
24440b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::StackProtectReq);
24450b57cec5SDimitry Andric
24460b57cec5SDimitry Andric if (!D) {
24470b57cec5SDimitry Andric // If we don't have a declaration to control inlining, the function isn't
24480b57cec5SDimitry Andric // explicitly marked as alwaysinline for semantic reasons, and inlining is
24490b57cec5SDimitry Andric // disabled, mark the function as noinline.
24500b57cec5SDimitry Andric if (!F->hasFnAttribute(llvm::Attribute::AlwaysInline) &&
24510b57cec5SDimitry Andric CodeGenOpts.getInlining() == CodeGenOptions::OnlyAlwaysInlining)
24520b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::NoInline);
24530b57cec5SDimitry Andric
2454349cc55cSDimitry Andric F->addFnAttrs(B);
24550b57cec5SDimitry Andric return;
24560b57cec5SDimitry Andric }
24570b57cec5SDimitry Andric
24585f757f3fSDimitry Andric // Handle SME attributes that apply to function definitions,
24595f757f3fSDimitry Andric // rather than to function prototypes.
24605f757f3fSDimitry Andric if (D->hasAttr<ArmLocallyStreamingAttr>())
24615f757f3fSDimitry Andric B.addAttribute("aarch64_pstate_sm_body");
24625f757f3fSDimitry Andric
24637a6dacacSDimitry Andric if (auto *Attr = D->getAttr<ArmNewAttr>()) {
24647a6dacacSDimitry Andric if (Attr->isNewZA())
24650fca6ea1SDimitry Andric B.addAttribute("aarch64_new_za");
24667a6dacacSDimitry Andric if (Attr->isNewZT0())
24677a6dacacSDimitry Andric B.addAttribute("aarch64_new_zt0");
24687a6dacacSDimitry Andric }
24695f757f3fSDimitry Andric
24700b57cec5SDimitry Andric // Track whether we need to add the optnone LLVM attribute,
24710b57cec5SDimitry Andric // starting with the default for this optimization level.
24720b57cec5SDimitry Andric bool ShouldAddOptNone =
24730b57cec5SDimitry Andric !CodeGenOpts.DisableO0ImplyOptNone && CodeGenOpts.OptimizationLevel == 0;
24740b57cec5SDimitry Andric // We can't add optnone in the following cases, it won't pass the verifier.
24750b57cec5SDimitry Andric ShouldAddOptNone &= !D->hasAttr<MinSizeAttr>();
24760b57cec5SDimitry Andric ShouldAddOptNone &= !D->hasAttr<AlwaysInlineAttr>();
24770b57cec5SDimitry Andric
2478480093f4SDimitry Andric // Add optnone, but do so only if the function isn't always_inline.
2479480093f4SDimitry Andric if ((ShouldAddOptNone || D->hasAttr<OptimizeNoneAttr>()) &&
2480480093f4SDimitry Andric !F->hasFnAttribute(llvm::Attribute::AlwaysInline)) {
24810b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::OptimizeNone);
24820b57cec5SDimitry Andric
24830b57cec5SDimitry Andric // OptimizeNone implies noinline; we should not be inlining such functions.
24840b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::NoInline);
24850b57cec5SDimitry Andric
24860b57cec5SDimitry Andric // We still need to handle naked functions even though optnone subsumes
24870b57cec5SDimitry Andric // much of their semantics.
24880b57cec5SDimitry Andric if (D->hasAttr<NakedAttr>())
24890b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::Naked);
24900b57cec5SDimitry Andric
24910b57cec5SDimitry Andric // OptimizeNone wins over OptimizeForSize and MinSize.
24920b57cec5SDimitry Andric F->removeFnAttr(llvm::Attribute::OptimizeForSize);
24930b57cec5SDimitry Andric F->removeFnAttr(llvm::Attribute::MinSize);
24940b57cec5SDimitry Andric } else if (D->hasAttr<NakedAttr>()) {
24950b57cec5SDimitry Andric // Naked implies noinline: we should not be inlining such functions.
24960b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::Naked);
24970b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::NoInline);
24980b57cec5SDimitry Andric } else if (D->hasAttr<NoDuplicateAttr>()) {
24990b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::NoDuplicate);
2500480093f4SDimitry Andric } else if (D->hasAttr<NoInlineAttr>() && !F->hasFnAttribute(llvm::Attribute::AlwaysInline)) {
2501480093f4SDimitry Andric // Add noinline if the function isn't always_inline.
25020b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::NoInline);
25030b57cec5SDimitry Andric } else if (D->hasAttr<AlwaysInlineAttr>() &&
25040b57cec5SDimitry Andric !F->hasFnAttribute(llvm::Attribute::NoInline)) {
25050b57cec5SDimitry Andric // (noinline wins over always_inline, and we can't specify both in IR)
25060b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::AlwaysInline);
25070b57cec5SDimitry Andric } else if (CodeGenOpts.getInlining() == CodeGenOptions::OnlyAlwaysInlining) {
25080b57cec5SDimitry Andric // If we're not inlining, then force everything that isn't always_inline to
25090b57cec5SDimitry Andric // carry an explicit noinline attribute.
25100b57cec5SDimitry Andric if (!F->hasFnAttribute(llvm::Attribute::AlwaysInline))
25110b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::NoInline);
25120b57cec5SDimitry Andric } else {
25130b57cec5SDimitry Andric // Otherwise, propagate the inline hint attribute and potentially use its
25140b57cec5SDimitry Andric // absence to mark things as noinline.
25150b57cec5SDimitry Andric if (auto *FD = dyn_cast<FunctionDecl>(D)) {
25160b57cec5SDimitry Andric // Search function and template pattern redeclarations for inline.
25170b57cec5SDimitry Andric auto CheckForInline = [](const FunctionDecl *FD) {
25180b57cec5SDimitry Andric auto CheckRedeclForInline = [](const FunctionDecl *Redecl) {
25190b57cec5SDimitry Andric return Redecl->isInlineSpecified();
25200b57cec5SDimitry Andric };
25210b57cec5SDimitry Andric if (any_of(FD->redecls(), CheckRedeclForInline))
25220b57cec5SDimitry Andric return true;
25230b57cec5SDimitry Andric const FunctionDecl *Pattern = FD->getTemplateInstantiationPattern();
25240b57cec5SDimitry Andric if (!Pattern)
25250b57cec5SDimitry Andric return false;
25260b57cec5SDimitry Andric return any_of(Pattern->redecls(), CheckRedeclForInline);
25270b57cec5SDimitry Andric };
25280b57cec5SDimitry Andric if (CheckForInline(FD)) {
25290b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::InlineHint);
25300b57cec5SDimitry Andric } else if (CodeGenOpts.getInlining() ==
25310b57cec5SDimitry Andric CodeGenOptions::OnlyHintInlining &&
25320b57cec5SDimitry Andric !FD->isInlined() &&
25330b57cec5SDimitry Andric !F->hasFnAttribute(llvm::Attribute::AlwaysInline)) {
25340b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::NoInline);
25350b57cec5SDimitry Andric }
25360b57cec5SDimitry Andric }
25370b57cec5SDimitry Andric }
25380b57cec5SDimitry Andric
25390b57cec5SDimitry Andric // Add other optimization related attributes if we are optimizing this
25400b57cec5SDimitry Andric // function.
25410b57cec5SDimitry Andric if (!D->hasAttr<OptimizeNoneAttr>()) {
25420b57cec5SDimitry Andric if (D->hasAttr<ColdAttr>()) {
25430b57cec5SDimitry Andric if (!ShouldAddOptNone)
25440b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::OptimizeForSize);
25450b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::Cold);
25460b57cec5SDimitry Andric }
2547e8d8bef9SDimitry Andric if (D->hasAttr<HotAttr>())
2548e8d8bef9SDimitry Andric B.addAttribute(llvm::Attribute::Hot);
25490b57cec5SDimitry Andric if (D->hasAttr<MinSizeAttr>())
25500b57cec5SDimitry Andric B.addAttribute(llvm::Attribute::MinSize);
25510b57cec5SDimitry Andric }
25520b57cec5SDimitry Andric
2553349cc55cSDimitry Andric F->addFnAttrs(B);
25540b57cec5SDimitry Andric
25550b57cec5SDimitry Andric unsigned alignment = D->getMaxAlignment() / Context.getCharWidth();
25560b57cec5SDimitry Andric if (alignment)
2557a7dea167SDimitry Andric F->setAlignment(llvm::Align(alignment));
25580b57cec5SDimitry Andric
25590b57cec5SDimitry Andric if (!D->hasAttr<AlignedAttr>())
25600b57cec5SDimitry Andric if (LangOpts.FunctionAlignment)
2561a7dea167SDimitry Andric F->setAlignment(llvm::Align(1ull << LangOpts.FunctionAlignment));
25620b57cec5SDimitry Andric
25630b57cec5SDimitry Andric // Some C++ ABIs require 2-byte alignment for member functions, in order to
25640b57cec5SDimitry Andric // reserve a bit for differentiating between virtual and non-virtual member
25650b57cec5SDimitry Andric // functions. If the current target's C++ ABI requires this and this is a
25660b57cec5SDimitry Andric // member function, set its alignment accordingly.
25670b57cec5SDimitry Andric if (getTarget().getCXXABI().areMemberFunctionsAligned()) {
25688a4dda33SDimitry Andric if (isa<CXXMethodDecl>(D) && F->getPointerAlignment(getDataLayout()) < 2)
256906c3fb27SDimitry Andric F->setAlignment(std::max(llvm::Align(2), F->getAlign().valueOrOne()));
25700b57cec5SDimitry Andric }
25710b57cec5SDimitry Andric
2572a7dea167SDimitry Andric // In the cross-dso CFI mode with canonical jump tables, we want !type
2573a7dea167SDimitry Andric // attributes on definitions only.
2574a7dea167SDimitry Andric if (CodeGenOpts.SanitizeCfiCrossDso &&
2575a7dea167SDimitry Andric CodeGenOpts.SanitizeCfiCanonicalJumpTables) {
2576a7dea167SDimitry Andric if (auto *FD = dyn_cast<FunctionDecl>(D)) {
2577a7dea167SDimitry Andric // Skip available_externally functions. They won't be codegen'ed in the
2578a7dea167SDimitry Andric // current module anyway.
2579a7dea167SDimitry Andric if (getContext().GetGVALinkageForFunction(FD) != GVA_AvailableExternally)
25800b57cec5SDimitry Andric CreateFunctionTypeMetadataForIcall(FD, F);
2581a7dea167SDimitry Andric }
2582a7dea167SDimitry Andric }
25830b57cec5SDimitry Andric
25840b57cec5SDimitry Andric // Emit type metadata on member functions for member function pointer checks.
25850b57cec5SDimitry Andric // These are only ever necessary on definitions; we're guaranteed that the
25860b57cec5SDimitry Andric // definition will be present in the LTO unit as a result of LTO visibility.
25870b57cec5SDimitry Andric auto *MD = dyn_cast<CXXMethodDecl>(D);
25880b57cec5SDimitry Andric if (MD && requiresMemberFunctionPointerTypeMetadata(*this, MD)) {
25890b57cec5SDimitry Andric for (const CXXRecordDecl *Base : getMostBaseClasses(MD->getParent())) {
25900b57cec5SDimitry Andric llvm::Metadata *Id =
25910b57cec5SDimitry Andric CreateMetadataIdentifierForType(Context.getMemberPointerType(
25920b57cec5SDimitry Andric MD->getType(), Context.getRecordType(Base).getTypePtr()));
25930b57cec5SDimitry Andric F->addTypeMetadata(0, Id);
25940b57cec5SDimitry Andric }
25950b57cec5SDimitry Andric }
25960b57cec5SDimitry Andric }
25970b57cec5SDimitry Andric
SetCommonAttributes(GlobalDecl GD,llvm::GlobalValue * GV)25980b57cec5SDimitry Andric void CodeGenModule::SetCommonAttributes(GlobalDecl GD, llvm::GlobalValue *GV) {
25990b57cec5SDimitry Andric const Decl *D = GD.getDecl();
2600349cc55cSDimitry Andric if (isa_and_nonnull<NamedDecl>(D))
26010b57cec5SDimitry Andric setGVProperties(GV, GD);
26020b57cec5SDimitry Andric else
26030b57cec5SDimitry Andric GV->setVisibility(llvm::GlobalValue::DefaultVisibility);
26040b57cec5SDimitry Andric
26050b57cec5SDimitry Andric if (D && D->hasAttr<UsedAttr>())
2606fe6060f1SDimitry Andric addUsedOrCompilerUsedGlobal(GV);
26070b57cec5SDimitry Andric
260806c3fb27SDimitry Andric if (const auto *VD = dyn_cast_if_present<VarDecl>(D);
260906c3fb27SDimitry Andric VD &&
261006c3fb27SDimitry Andric ((CodeGenOpts.KeepPersistentStorageVariables &&
261106c3fb27SDimitry Andric (VD->getStorageDuration() == SD_Static ||
261206c3fb27SDimitry Andric VD->getStorageDuration() == SD_Thread)) ||
261306c3fb27SDimitry Andric (CodeGenOpts.KeepStaticConsts && VD->getStorageDuration() == SD_Static &&
261406c3fb27SDimitry Andric VD->getType().isConstQualified())))
2615fe6060f1SDimitry Andric addUsedOrCompilerUsedGlobal(GV);
26160b57cec5SDimitry Andric }
26170b57cec5SDimitry Andric
GetCPUAndFeaturesAttributes(GlobalDecl GD,llvm::AttrBuilder & Attrs,bool SetTargetFeatures)26180b57cec5SDimitry Andric bool CodeGenModule::GetCPUAndFeaturesAttributes(GlobalDecl GD,
261906c3fb27SDimitry Andric llvm::AttrBuilder &Attrs,
262006c3fb27SDimitry Andric bool SetTargetFeatures) {
26210b57cec5SDimitry Andric // Add target-cpu and target-features attributes to functions. If
26220b57cec5SDimitry Andric // we have a decl for the function and it has a target attribute then
26230b57cec5SDimitry Andric // parse that and add it to the feature set.
26240b57cec5SDimitry Andric StringRef TargetCPU = getTarget().getTargetOpts().CPU;
2625e8d8bef9SDimitry Andric StringRef TuneCPU = getTarget().getTargetOpts().TuneCPU;
26260b57cec5SDimitry Andric std::vector<std::string> Features;
26270b57cec5SDimitry Andric const auto *FD = dyn_cast_or_null<FunctionDecl>(GD.getDecl());
26280b57cec5SDimitry Andric FD = FD ? FD->getMostRecentDecl() : FD;
26290b57cec5SDimitry Andric const auto *TD = FD ? FD->getAttr<TargetAttr>() : nullptr;
2630bdd1243dSDimitry Andric const auto *TV = FD ? FD->getAttr<TargetVersionAttr>() : nullptr;
2631bdd1243dSDimitry Andric assert((!TD || !TV) && "both target_version and target specified");
26320b57cec5SDimitry Andric const auto *SD = FD ? FD->getAttr<CPUSpecificAttr>() : nullptr;
26334824e7fdSDimitry Andric const auto *TC = FD ? FD->getAttr<TargetClonesAttr>() : nullptr;
26340b57cec5SDimitry Andric bool AddedAttr = false;
2635bdd1243dSDimitry Andric if (TD || TV || SD || TC) {
26360b57cec5SDimitry Andric llvm::StringMap<bool> FeatureMap;
2637480093f4SDimitry Andric getContext().getFunctionFeatureMap(FeatureMap, GD);
26380b57cec5SDimitry Andric
26390b57cec5SDimitry Andric // Produce the canonical string for this set of features.
26400b57cec5SDimitry Andric for (const llvm::StringMap<bool>::value_type &Entry : FeatureMap)
26410b57cec5SDimitry Andric Features.push_back((Entry.getValue() ? "+" : "-") + Entry.getKey().str());
26420b57cec5SDimitry Andric
26430b57cec5SDimitry Andric // Now add the target-cpu and target-features to the function.
26440b57cec5SDimitry Andric // While we populated the feature map above, we still need to
26450b57cec5SDimitry Andric // get and parse the target attribute so we can get the cpu for
26460b57cec5SDimitry Andric // the function.
26470b57cec5SDimitry Andric if (TD) {
2648bdd1243dSDimitry Andric ParsedTargetAttr ParsedAttr =
2649bdd1243dSDimitry Andric Target.parseTargetAttr(TD->getFeaturesStr());
2650bdd1243dSDimitry Andric if (!ParsedAttr.CPU.empty() &&
2651bdd1243dSDimitry Andric getTarget().isValidCPUName(ParsedAttr.CPU)) {
2652bdd1243dSDimitry Andric TargetCPU = ParsedAttr.CPU;
2653e8d8bef9SDimitry Andric TuneCPU = ""; // Clear the tune CPU.
2654e8d8bef9SDimitry Andric }
2655e8d8bef9SDimitry Andric if (!ParsedAttr.Tune.empty() &&
2656e8d8bef9SDimitry Andric getTarget().isValidCPUName(ParsedAttr.Tune))
2657e8d8bef9SDimitry Andric TuneCPU = ParsedAttr.Tune;
26580b57cec5SDimitry Andric }
265981ad6265SDimitry Andric
266081ad6265SDimitry Andric if (SD) {
266181ad6265SDimitry Andric // Apply the given CPU name as the 'tune-cpu' so that the optimizer can
266281ad6265SDimitry Andric // favor this processor.
266306c3fb27SDimitry Andric TuneCPU = SD->getCPUName(GD.getMultiVersionIndex())->getName();
266481ad6265SDimitry Andric }
26650b57cec5SDimitry Andric } else {
26660b57cec5SDimitry Andric // Otherwise just add the existing target cpu and target features to the
26670b57cec5SDimitry Andric // function.
26680b57cec5SDimitry Andric Features = getTarget().getTargetOpts().Features;
26690b57cec5SDimitry Andric }
26700b57cec5SDimitry Andric
2671e8d8bef9SDimitry Andric if (!TargetCPU.empty()) {
26720b57cec5SDimitry Andric Attrs.addAttribute("target-cpu", TargetCPU);
26730b57cec5SDimitry Andric AddedAttr = true;
26740b57cec5SDimitry Andric }
2675e8d8bef9SDimitry Andric if (!TuneCPU.empty()) {
2676e8d8bef9SDimitry Andric Attrs.addAttribute("tune-cpu", TuneCPU);
2677e8d8bef9SDimitry Andric AddedAttr = true;
2678e8d8bef9SDimitry Andric }
267906c3fb27SDimitry Andric if (!Features.empty() && SetTargetFeatures) {
268006c3fb27SDimitry Andric llvm::erase_if(Features, [&](const std::string& F) {
268106c3fb27SDimitry Andric return getTarget().isReadOnlyFeature(F.substr(1));
268206c3fb27SDimitry Andric });
26830b57cec5SDimitry Andric llvm::sort(Features);
26840b57cec5SDimitry Andric Attrs.addAttribute("target-features", llvm::join(Features, ","));
26850b57cec5SDimitry Andric AddedAttr = true;
26860b57cec5SDimitry Andric }
26870b57cec5SDimitry Andric
26880b57cec5SDimitry Andric return AddedAttr;
26890b57cec5SDimitry Andric }
26900b57cec5SDimitry Andric
setNonAliasAttributes(GlobalDecl GD,llvm::GlobalObject * GO)26910b57cec5SDimitry Andric void CodeGenModule::setNonAliasAttributes(GlobalDecl GD,
26920b57cec5SDimitry Andric llvm::GlobalObject *GO) {
26930b57cec5SDimitry Andric const Decl *D = GD.getDecl();
26940b57cec5SDimitry Andric SetCommonAttributes(GD, GO);
26950b57cec5SDimitry Andric
26960b57cec5SDimitry Andric if (D) {
26970b57cec5SDimitry Andric if (auto *GV = dyn_cast<llvm::GlobalVariable>(GO)) {
2698fe6060f1SDimitry Andric if (D->hasAttr<RetainAttr>())
2699fe6060f1SDimitry Andric addUsedGlobal(GV);
27000b57cec5SDimitry Andric if (auto *SA = D->getAttr<PragmaClangBSSSectionAttr>())
27010b57cec5SDimitry Andric GV->addAttribute("bss-section", SA->getName());
27020b57cec5SDimitry Andric if (auto *SA = D->getAttr<PragmaClangDataSectionAttr>())
27030b57cec5SDimitry Andric GV->addAttribute("data-section", SA->getName());
27040b57cec5SDimitry Andric if (auto *SA = D->getAttr<PragmaClangRodataSectionAttr>())
27050b57cec5SDimitry Andric GV->addAttribute("rodata-section", SA->getName());
2706a7dea167SDimitry Andric if (auto *SA = D->getAttr<PragmaClangRelroSectionAttr>())
2707a7dea167SDimitry Andric GV->addAttribute("relro-section", SA->getName());
27080b57cec5SDimitry Andric }
27090b57cec5SDimitry Andric
27100b57cec5SDimitry Andric if (auto *F = dyn_cast<llvm::Function>(GO)) {
2711fe6060f1SDimitry Andric if (D->hasAttr<RetainAttr>())
2712fe6060f1SDimitry Andric addUsedGlobal(F);
27130b57cec5SDimitry Andric if (auto *SA = D->getAttr<PragmaClangTextSectionAttr>())
27140b57cec5SDimitry Andric if (!D->getAttr<SectionAttr>())
27150fca6ea1SDimitry Andric F->setSection(SA->getName());
27160b57cec5SDimitry Andric
271704eeddc0SDimitry Andric llvm::AttrBuilder Attrs(F->getContext());
27180b57cec5SDimitry Andric if (GetCPUAndFeaturesAttributes(GD, Attrs)) {
27190b57cec5SDimitry Andric // We know that GetCPUAndFeaturesAttributes will always have the
27200b57cec5SDimitry Andric // newest set, since it has the newest possible FunctionDecl, so the
27210b57cec5SDimitry Andric // new ones should replace the old.
272204eeddc0SDimitry Andric llvm::AttributeMask RemoveAttrs;
2723e8d8bef9SDimitry Andric RemoveAttrs.addAttribute("target-cpu");
2724e8d8bef9SDimitry Andric RemoveAttrs.addAttribute("target-features");
2725e8d8bef9SDimitry Andric RemoveAttrs.addAttribute("tune-cpu");
2726349cc55cSDimitry Andric F->removeFnAttrs(RemoveAttrs);
2727349cc55cSDimitry Andric F->addFnAttrs(Attrs);
27280b57cec5SDimitry Andric }
27290b57cec5SDimitry Andric }
27300b57cec5SDimitry Andric
27310b57cec5SDimitry Andric if (const auto *CSA = D->getAttr<CodeSegAttr>())
27320b57cec5SDimitry Andric GO->setSection(CSA->getName());
27330b57cec5SDimitry Andric else if (const auto *SA = D->getAttr<SectionAttr>())
27340b57cec5SDimitry Andric GO->setSection(SA->getName());
27350b57cec5SDimitry Andric }
27360b57cec5SDimitry Andric
27370b57cec5SDimitry Andric getTargetCodeGenInfo().setTargetAttributes(D, GO, *this);
27380b57cec5SDimitry Andric }
27390b57cec5SDimitry Andric
SetInternalFunctionAttributes(GlobalDecl GD,llvm::Function * F,const CGFunctionInfo & FI)27400b57cec5SDimitry Andric void CodeGenModule::SetInternalFunctionAttributes(GlobalDecl GD,
27410b57cec5SDimitry Andric llvm::Function *F,
27420b57cec5SDimitry Andric const CGFunctionInfo &FI) {
27430b57cec5SDimitry Andric const Decl *D = GD.getDecl();
2744fe6060f1SDimitry Andric SetLLVMFunctionAttributes(GD, FI, F, /*IsThunk=*/false);
27450b57cec5SDimitry Andric SetLLVMFunctionAttributesForDefinition(D, F);
27460b57cec5SDimitry Andric
27470b57cec5SDimitry Andric F->setLinkage(llvm::Function::InternalLinkage);
27480b57cec5SDimitry Andric
27490b57cec5SDimitry Andric setNonAliasAttributes(GD, F);
27500b57cec5SDimitry Andric }
27510b57cec5SDimitry Andric
setLinkageForGV(llvm::GlobalValue * GV,const NamedDecl * ND)27520b57cec5SDimitry Andric static void setLinkageForGV(llvm::GlobalValue *GV, const NamedDecl *ND) {
27530b57cec5SDimitry Andric // Set linkage and visibility in case we never see a definition.
27540b57cec5SDimitry Andric LinkageInfo LV = ND->getLinkageAndVisibility();
27550b57cec5SDimitry Andric // Don't set internal linkage on declarations.
27560b57cec5SDimitry Andric // "extern_weak" is overloaded in LLVM; we probably should have
27570b57cec5SDimitry Andric // separate linkage types for this.
27580b57cec5SDimitry Andric if (isExternallyVisible(LV.getLinkage()) &&
27590b57cec5SDimitry Andric (ND->hasAttr<WeakAttr>() || ND->isWeakImported()))
27600b57cec5SDimitry Andric GV->setLinkage(llvm::GlobalValue::ExternalWeakLinkage);
27610b57cec5SDimitry Andric }
27620b57cec5SDimitry Andric
CreateFunctionTypeMetadataForIcall(const FunctionDecl * FD,llvm::Function * F)27630b57cec5SDimitry Andric void CodeGenModule::CreateFunctionTypeMetadataForIcall(const FunctionDecl *FD,
27640b57cec5SDimitry Andric llvm::Function *F) {
27650b57cec5SDimitry Andric // Only if we are checking indirect calls.
27660b57cec5SDimitry Andric if (!LangOpts.Sanitize.has(SanitizerKind::CFIICall))
27670b57cec5SDimitry Andric return;
27680b57cec5SDimitry Andric
27690b57cec5SDimitry Andric // Non-static class methods are handled via vtable or member function pointer
27700b57cec5SDimitry Andric // checks elsewhere.
27710b57cec5SDimitry Andric if (isa<CXXMethodDecl>(FD) && !cast<CXXMethodDecl>(FD)->isStatic())
27720b57cec5SDimitry Andric return;
27730b57cec5SDimitry Andric
27740b57cec5SDimitry Andric llvm::Metadata *MD = CreateMetadataIdentifierForType(FD->getType());
27750b57cec5SDimitry Andric F->addTypeMetadata(0, MD);
27760b57cec5SDimitry Andric F->addTypeMetadata(0, CreateMetadataIdentifierGeneralized(FD->getType()));
27770b57cec5SDimitry Andric
27780b57cec5SDimitry Andric // Emit a hash-based bit set entry for cross-DSO calls.
27790b57cec5SDimitry Andric if (CodeGenOpts.SanitizeCfiCrossDso)
27800b57cec5SDimitry Andric if (auto CrossDsoTypeId = CreateCrossDsoCfiTypeId(MD))
27810b57cec5SDimitry Andric F->addTypeMetadata(0, llvm::ConstantAsMetadata::get(CrossDsoTypeId));
27820b57cec5SDimitry Andric }
27830b57cec5SDimitry Andric
setKCFIType(const FunctionDecl * FD,llvm::Function * F)2784bdd1243dSDimitry Andric void CodeGenModule::setKCFIType(const FunctionDecl *FD, llvm::Function *F) {
2785bdd1243dSDimitry Andric llvm::LLVMContext &Ctx = F->getContext();
2786bdd1243dSDimitry Andric llvm::MDBuilder MDB(Ctx);
2787bdd1243dSDimitry Andric F->setMetadata(llvm::LLVMContext::MD_kcfi_type,
2788bdd1243dSDimitry Andric llvm::MDNode::get(
2789bdd1243dSDimitry Andric Ctx, MDB.createConstant(CreateKCFITypeId(FD->getType()))));
2790bdd1243dSDimitry Andric }
2791bdd1243dSDimitry Andric
allowKCFIIdentifier(StringRef Name)2792bdd1243dSDimitry Andric static bool allowKCFIIdentifier(StringRef Name) {
2793bdd1243dSDimitry Andric // KCFI type identifier constants are only necessary for external assembly
2794bdd1243dSDimitry Andric // functions, which means it's safe to skip unusual names. Subset of
2795bdd1243dSDimitry Andric // MCAsmInfo::isAcceptableChar() and MCAsmInfoXCOFF::isAcceptableChar().
2796bdd1243dSDimitry Andric return llvm::all_of(Name, [](const char &C) {
2797bdd1243dSDimitry Andric return llvm::isAlnum(C) || C == '_' || C == '.';
2798bdd1243dSDimitry Andric });
2799bdd1243dSDimitry Andric }
2800bdd1243dSDimitry Andric
finalizeKCFITypes()2801bdd1243dSDimitry Andric void CodeGenModule::finalizeKCFITypes() {
2802bdd1243dSDimitry Andric llvm::Module &M = getModule();
2803bdd1243dSDimitry Andric for (auto &F : M.functions()) {
2804bdd1243dSDimitry Andric // Remove KCFI type metadata from non-address-taken local functions.
2805bdd1243dSDimitry Andric bool AddressTaken = F.hasAddressTaken();
2806bdd1243dSDimitry Andric if (!AddressTaken && F.hasLocalLinkage())
2807bdd1243dSDimitry Andric F.eraseMetadata(llvm::LLVMContext::MD_kcfi_type);
2808bdd1243dSDimitry Andric
2809bdd1243dSDimitry Andric // Generate a constant with the expected KCFI type identifier for all
2810bdd1243dSDimitry Andric // address-taken function declarations to support annotating indirectly
2811bdd1243dSDimitry Andric // called assembly functions.
2812bdd1243dSDimitry Andric if (!AddressTaken || !F.isDeclaration())
2813bdd1243dSDimitry Andric continue;
2814bdd1243dSDimitry Andric
2815bdd1243dSDimitry Andric const llvm::ConstantInt *Type;
2816bdd1243dSDimitry Andric if (const llvm::MDNode *MD = F.getMetadata(llvm::LLVMContext::MD_kcfi_type))
2817bdd1243dSDimitry Andric Type = llvm::mdconst::extract<llvm::ConstantInt>(MD->getOperand(0));
2818bdd1243dSDimitry Andric else
2819bdd1243dSDimitry Andric continue;
2820bdd1243dSDimitry Andric
2821bdd1243dSDimitry Andric StringRef Name = F.getName();
2822bdd1243dSDimitry Andric if (!allowKCFIIdentifier(Name))
2823bdd1243dSDimitry Andric continue;
2824bdd1243dSDimitry Andric
2825bdd1243dSDimitry Andric std::string Asm = (".weak __kcfi_typeid_" + Name + "\n.set __kcfi_typeid_" +
2826bdd1243dSDimitry Andric Name + ", " + Twine(Type->getZExtValue()) + "\n")
2827bdd1243dSDimitry Andric .str();
2828bdd1243dSDimitry Andric M.appendModuleInlineAsm(Asm);
2829bdd1243dSDimitry Andric }
2830bdd1243dSDimitry Andric }
2831bdd1243dSDimitry Andric
SetFunctionAttributes(GlobalDecl GD,llvm::Function * F,bool IsIncompleteFunction,bool IsThunk)28320b57cec5SDimitry Andric void CodeGenModule::SetFunctionAttributes(GlobalDecl GD, llvm::Function *F,
28330b57cec5SDimitry Andric bool IsIncompleteFunction,
28340b57cec5SDimitry Andric bool IsThunk) {
28350b57cec5SDimitry Andric
28360b57cec5SDimitry Andric if (llvm::Intrinsic::ID IID = F->getIntrinsicID()) {
28370b57cec5SDimitry Andric // If this is an intrinsic function, set the function's attributes
28380b57cec5SDimitry Andric // to the intrinsic's attributes.
28390b57cec5SDimitry Andric F->setAttributes(llvm::Intrinsic::getAttributes(getLLVMContext(), IID));
28400b57cec5SDimitry Andric return;
28410b57cec5SDimitry Andric }
28420b57cec5SDimitry Andric
28430b57cec5SDimitry Andric const auto *FD = cast<FunctionDecl>(GD.getDecl());
28440b57cec5SDimitry Andric
28450b57cec5SDimitry Andric if (!IsIncompleteFunction)
2846fe6060f1SDimitry Andric SetLLVMFunctionAttributes(GD, getTypes().arrangeGlobalDeclaration(GD), F,
2847fe6060f1SDimitry Andric IsThunk);
28480b57cec5SDimitry Andric
28490b57cec5SDimitry Andric // Add the Returned attribute for "this", except for iOS 5 and earlier
28500b57cec5SDimitry Andric // where substantial code, including the libstdc++ dylib, was compiled with
28510b57cec5SDimitry Andric // GCC and does not actually return "this".
28520b57cec5SDimitry Andric if (!IsThunk && getCXXABI().HasThisReturn(GD) &&
28530b57cec5SDimitry Andric !(getTriple().isiOS() && getTriple().isOSVersionLT(6))) {
28540b57cec5SDimitry Andric assert(!F->arg_empty() &&
28550b57cec5SDimitry Andric F->arg_begin()->getType()
28560b57cec5SDimitry Andric ->canLosslesslyBitCastTo(F->getReturnType()) &&
28570b57cec5SDimitry Andric "unexpected this return");
2858349cc55cSDimitry Andric F->addParamAttr(0, llvm::Attribute::Returned);
28590b57cec5SDimitry Andric }
28600b57cec5SDimitry Andric
28610b57cec5SDimitry Andric // Only a few attributes are set on declarations; these may later be
28620b57cec5SDimitry Andric // overridden by a definition.
28630b57cec5SDimitry Andric
28640b57cec5SDimitry Andric setLinkageForGV(F, FD);
28650b57cec5SDimitry Andric setGVProperties(F, FD);
28660b57cec5SDimitry Andric
28670b57cec5SDimitry Andric // Setup target-specific attributes.
28680b57cec5SDimitry Andric if (!IsIncompleteFunction && F->isDeclaration())
28690b57cec5SDimitry Andric getTargetCodeGenInfo().setTargetAttributes(FD, F, *this);
28700b57cec5SDimitry Andric
28710b57cec5SDimitry Andric if (const auto *CSA = FD->getAttr<CodeSegAttr>())
28720b57cec5SDimitry Andric F->setSection(CSA->getName());
28730b57cec5SDimitry Andric else if (const auto *SA = FD->getAttr<SectionAttr>())
28740b57cec5SDimitry Andric F->setSection(SA->getName());
28750b57cec5SDimitry Andric
2876349cc55cSDimitry Andric if (const auto *EA = FD->getAttr<ErrorAttr>()) {
2877349cc55cSDimitry Andric if (EA->isError())
2878349cc55cSDimitry Andric F->addFnAttr("dontcall-error", EA->getUserDiagnostic());
2879349cc55cSDimitry Andric else if (EA->isWarning())
2880349cc55cSDimitry Andric F->addFnAttr("dontcall-warn", EA->getUserDiagnostic());
2881349cc55cSDimitry Andric }
2882349cc55cSDimitry Andric
2883d65cd7a5SDimitry Andric // If we plan on emitting this inline builtin, we can't treat it as a builtin.
2884480093f4SDimitry Andric if (FD->isInlineBuiltinDeclaration()) {
2885d65cd7a5SDimitry Andric const FunctionDecl *FDBody;
2886d65cd7a5SDimitry Andric bool HasBody = FD->hasBody(FDBody);
2887d65cd7a5SDimitry Andric (void)HasBody;
2888d65cd7a5SDimitry Andric assert(HasBody && "Inline builtin declarations should always have an "
2889d65cd7a5SDimitry Andric "available body!");
2890d65cd7a5SDimitry Andric if (shouldEmitFunction(FDBody))
2891349cc55cSDimitry Andric F->addFnAttr(llvm::Attribute::NoBuiltin);
2892480093f4SDimitry Andric }
2893480093f4SDimitry Andric
28940b57cec5SDimitry Andric if (FD->isReplaceableGlobalAllocationFunction()) {
28950b57cec5SDimitry Andric // A replaceable global allocation function does not act like a builtin by
28960b57cec5SDimitry Andric // default, only if it is invoked by a new-expression or delete-expression.
2897349cc55cSDimitry Andric F->addFnAttr(llvm::Attribute::NoBuiltin);
28980b57cec5SDimitry Andric }
28990b57cec5SDimitry Andric
29000b57cec5SDimitry Andric if (isa<CXXConstructorDecl>(FD) || isa<CXXDestructorDecl>(FD))
29010b57cec5SDimitry Andric F->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
29020b57cec5SDimitry Andric else if (const auto *MD = dyn_cast<CXXMethodDecl>(FD))
29030b57cec5SDimitry Andric if (MD->isVirtual())
29040b57cec5SDimitry Andric F->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
29050b57cec5SDimitry Andric
29060b57cec5SDimitry Andric // Don't emit entries for function declarations in the cross-DSO mode. This
2907a7dea167SDimitry Andric // is handled with better precision by the receiving DSO. But if jump tables
2908a7dea167SDimitry Andric // are non-canonical then we need type metadata in order to produce the local
2909a7dea167SDimitry Andric // jump table.
2910a7dea167SDimitry Andric if (!CodeGenOpts.SanitizeCfiCrossDso ||
2911a7dea167SDimitry Andric !CodeGenOpts.SanitizeCfiCanonicalJumpTables)
29120b57cec5SDimitry Andric CreateFunctionTypeMetadataForIcall(FD, F);
29130b57cec5SDimitry Andric
2914bdd1243dSDimitry Andric if (LangOpts.Sanitize.has(SanitizerKind::KCFI))
2915bdd1243dSDimitry Andric setKCFIType(FD, F);
2916bdd1243dSDimitry Andric
29170b57cec5SDimitry Andric if (getLangOpts().OpenMP && FD->hasAttr<OMPDeclareSimdDeclAttr>())
29180b57cec5SDimitry Andric getOpenMPRuntime().emitDeclareSimdFunction(FD, F);
29190b57cec5SDimitry Andric
2920bdd1243dSDimitry Andric if (CodeGenOpts.InlineMaxStackSize != UINT_MAX)
2921bdd1243dSDimitry Andric F->addFnAttr("inline-max-stacksize", llvm::utostr(CodeGenOpts.InlineMaxStackSize));
2922bdd1243dSDimitry Andric
29230b57cec5SDimitry Andric if (const auto *CB = FD->getAttr<CallbackAttr>()) {
29240b57cec5SDimitry Andric // Annotate the callback behavior as metadata:
29250b57cec5SDimitry Andric // - The callback callee (as argument number).
29260b57cec5SDimitry Andric // - The callback payloads (as argument numbers).
29270b57cec5SDimitry Andric llvm::LLVMContext &Ctx = F->getContext();
29280b57cec5SDimitry Andric llvm::MDBuilder MDB(Ctx);
29290b57cec5SDimitry Andric
29300b57cec5SDimitry Andric // The payload indices are all but the first one in the encoding. The first
29310b57cec5SDimitry Andric // identifies the callback callee.
29320b57cec5SDimitry Andric int CalleeIdx = *CB->encoding_begin();
29330b57cec5SDimitry Andric ArrayRef<int> PayloadIndices(CB->encoding_begin() + 1, CB->encoding_end());
29340b57cec5SDimitry Andric F->addMetadata(llvm::LLVMContext::MD_callback,
29350b57cec5SDimitry Andric *llvm::MDNode::get(Ctx, {MDB.createCallbackEncoding(
29360b57cec5SDimitry Andric CalleeIdx, PayloadIndices,
29370b57cec5SDimitry Andric /* VarArgsArePassed */ false)}));
29380b57cec5SDimitry Andric }
29390b57cec5SDimitry Andric }
29400b57cec5SDimitry Andric
addUsedGlobal(llvm::GlobalValue * GV)29410b57cec5SDimitry Andric void CodeGenModule::addUsedGlobal(llvm::GlobalValue *GV) {
2942e8d8bef9SDimitry Andric assert((isa<llvm::Function>(GV) || !GV->isDeclaration()) &&
29430b57cec5SDimitry Andric "Only globals with definition can force usage.");
29440b57cec5SDimitry Andric LLVMUsed.emplace_back(GV);
29450b57cec5SDimitry Andric }
29460b57cec5SDimitry Andric
addCompilerUsedGlobal(llvm::GlobalValue * GV)29470b57cec5SDimitry Andric void CodeGenModule::addCompilerUsedGlobal(llvm::GlobalValue *GV) {
29480b57cec5SDimitry Andric assert(!GV->isDeclaration() &&
29490b57cec5SDimitry Andric "Only globals with definition can force usage.");
29500b57cec5SDimitry Andric LLVMCompilerUsed.emplace_back(GV);
29510b57cec5SDimitry Andric }
29520b57cec5SDimitry Andric
addUsedOrCompilerUsedGlobal(llvm::GlobalValue * GV)2953fe6060f1SDimitry Andric void CodeGenModule::addUsedOrCompilerUsedGlobal(llvm::GlobalValue *GV) {
2954fe6060f1SDimitry Andric assert((isa<llvm::Function>(GV) || !GV->isDeclaration()) &&
2955fe6060f1SDimitry Andric "Only globals with definition can force usage.");
2956fe6060f1SDimitry Andric if (getTriple().isOSBinFormatELF())
2957fe6060f1SDimitry Andric LLVMCompilerUsed.emplace_back(GV);
2958fe6060f1SDimitry Andric else
2959fe6060f1SDimitry Andric LLVMUsed.emplace_back(GV);
2960fe6060f1SDimitry Andric }
2961fe6060f1SDimitry Andric
emitUsed(CodeGenModule & CGM,StringRef Name,std::vector<llvm::WeakTrackingVH> & List)29620b57cec5SDimitry Andric static void emitUsed(CodeGenModule &CGM, StringRef Name,
29630b57cec5SDimitry Andric std::vector<llvm::WeakTrackingVH> &List) {
29640b57cec5SDimitry Andric // Don't create llvm.used if there is no need.
29650b57cec5SDimitry Andric if (List.empty())
29660b57cec5SDimitry Andric return;
29670b57cec5SDimitry Andric
29680b57cec5SDimitry Andric // Convert List to what ConstantArray needs.
29690b57cec5SDimitry Andric SmallVector<llvm::Constant*, 8> UsedArray;
29700b57cec5SDimitry Andric UsedArray.resize(List.size());
29710b57cec5SDimitry Andric for (unsigned i = 0, e = List.size(); i != e; ++i) {
29720b57cec5SDimitry Andric UsedArray[i] =
29730b57cec5SDimitry Andric llvm::ConstantExpr::getPointerBitCastOrAddrSpaceCast(
29740b57cec5SDimitry Andric cast<llvm::Constant>(&*List[i]), CGM.Int8PtrTy);
29750b57cec5SDimitry Andric }
29760b57cec5SDimitry Andric
29770b57cec5SDimitry Andric if (UsedArray.empty())
29780b57cec5SDimitry Andric return;
29790b57cec5SDimitry Andric llvm::ArrayType *ATy = llvm::ArrayType::get(CGM.Int8PtrTy, UsedArray.size());
29800b57cec5SDimitry Andric
29810b57cec5SDimitry Andric auto *GV = new llvm::GlobalVariable(
29820b57cec5SDimitry Andric CGM.getModule(), ATy, false, llvm::GlobalValue::AppendingLinkage,
29830b57cec5SDimitry Andric llvm::ConstantArray::get(ATy, UsedArray), Name);
29840b57cec5SDimitry Andric
29850b57cec5SDimitry Andric GV->setSection("llvm.metadata");
29860b57cec5SDimitry Andric }
29870b57cec5SDimitry Andric
emitLLVMUsed()29880b57cec5SDimitry Andric void CodeGenModule::emitLLVMUsed() {
29890b57cec5SDimitry Andric emitUsed(*this, "llvm.used", LLVMUsed);
29900b57cec5SDimitry Andric emitUsed(*this, "llvm.compiler.used", LLVMCompilerUsed);
29910b57cec5SDimitry Andric }
29920b57cec5SDimitry Andric
AppendLinkerOptions(StringRef Opts)29930b57cec5SDimitry Andric void CodeGenModule::AppendLinkerOptions(StringRef Opts) {
29940b57cec5SDimitry Andric auto *MDOpts = llvm::MDString::get(getLLVMContext(), Opts);
29950b57cec5SDimitry Andric LinkerOptionsMetadata.push_back(llvm::MDNode::get(getLLVMContext(), MDOpts));
29960b57cec5SDimitry Andric }
29970b57cec5SDimitry Andric
AddDetectMismatch(StringRef Name,StringRef Value)29980b57cec5SDimitry Andric void CodeGenModule::AddDetectMismatch(StringRef Name, StringRef Value) {
29990b57cec5SDimitry Andric llvm::SmallString<32> Opt;
30000b57cec5SDimitry Andric getTargetCodeGenInfo().getDetectMismatchOption(Name, Value, Opt);
3001480093f4SDimitry Andric if (Opt.empty())
3002480093f4SDimitry Andric return;
30030b57cec5SDimitry Andric auto *MDOpts = llvm::MDString::get(getLLVMContext(), Opt);
30040b57cec5SDimitry Andric LinkerOptionsMetadata.push_back(llvm::MDNode::get(getLLVMContext(), MDOpts));
30050b57cec5SDimitry Andric }
30060b57cec5SDimitry Andric
AddDependentLib(StringRef Lib)30070b57cec5SDimitry Andric void CodeGenModule::AddDependentLib(StringRef Lib) {
30080b57cec5SDimitry Andric auto &C = getLLVMContext();
30090b57cec5SDimitry Andric if (getTarget().getTriple().isOSBinFormatELF()) {
30100b57cec5SDimitry Andric ELFDependentLibraries.push_back(
30110b57cec5SDimitry Andric llvm::MDNode::get(C, llvm::MDString::get(C, Lib)));
30120b57cec5SDimitry Andric return;
30130b57cec5SDimitry Andric }
30140b57cec5SDimitry Andric
30150b57cec5SDimitry Andric llvm::SmallString<24> Opt;
30160b57cec5SDimitry Andric getTargetCodeGenInfo().getDependentLibraryOption(Lib, Opt);
30170b57cec5SDimitry Andric auto *MDOpts = llvm::MDString::get(getLLVMContext(), Opt);
30180b57cec5SDimitry Andric LinkerOptionsMetadata.push_back(llvm::MDNode::get(C, MDOpts));
30190b57cec5SDimitry Andric }
30200b57cec5SDimitry Andric
30210b57cec5SDimitry Andric /// Add link options implied by the given module, including modules
30220b57cec5SDimitry Andric /// it depends on, using a postorder walk.
addLinkOptionsPostorder(CodeGenModule & CGM,Module * Mod,SmallVectorImpl<llvm::MDNode * > & Metadata,llvm::SmallPtrSet<Module *,16> & Visited)30230b57cec5SDimitry Andric static void addLinkOptionsPostorder(CodeGenModule &CGM, Module *Mod,
30240b57cec5SDimitry Andric SmallVectorImpl<llvm::MDNode *> &Metadata,
30250b57cec5SDimitry Andric llvm::SmallPtrSet<Module *, 16> &Visited) {
30260b57cec5SDimitry Andric // Import this module's parent.
30270b57cec5SDimitry Andric if (Mod->Parent && Visited.insert(Mod->Parent).second) {
30280b57cec5SDimitry Andric addLinkOptionsPostorder(CGM, Mod->Parent, Metadata, Visited);
30290b57cec5SDimitry Andric }
30300b57cec5SDimitry Andric
30310b57cec5SDimitry Andric // Import this module's dependencies.
3032349cc55cSDimitry Andric for (Module *Import : llvm::reverse(Mod->Imports)) {
3033349cc55cSDimitry Andric if (Visited.insert(Import).second)
3034349cc55cSDimitry Andric addLinkOptionsPostorder(CGM, Import, Metadata, Visited);
30350b57cec5SDimitry Andric }
30360b57cec5SDimitry Andric
30370b57cec5SDimitry Andric // Add linker options to link against the libraries/frameworks
30380b57cec5SDimitry Andric // described by this module.
30390b57cec5SDimitry Andric llvm::LLVMContext &Context = CGM.getLLVMContext();
30400b57cec5SDimitry Andric bool IsELF = CGM.getTarget().getTriple().isOSBinFormatELF();
30410b57cec5SDimitry Andric
30420b57cec5SDimitry Andric // For modules that use export_as for linking, use that module
30430b57cec5SDimitry Andric // name instead.
30440b57cec5SDimitry Andric if (Mod->UseExportAsModuleLinkName)
30450b57cec5SDimitry Andric return;
30460b57cec5SDimitry Andric
3047349cc55cSDimitry Andric for (const Module::LinkLibrary &LL : llvm::reverse(Mod->LinkLibraries)) {
30480b57cec5SDimitry Andric // Link against a framework. Frameworks are currently Darwin only, so we
30490b57cec5SDimitry Andric // don't to ask TargetCodeGenInfo for the spelling of the linker option.
3050349cc55cSDimitry Andric if (LL.IsFramework) {
3051349cc55cSDimitry Andric llvm::Metadata *Args[2] = {llvm::MDString::get(Context, "-framework"),
3052349cc55cSDimitry Andric llvm::MDString::get(Context, LL.Library)};
30530b57cec5SDimitry Andric
30540b57cec5SDimitry Andric Metadata.push_back(llvm::MDNode::get(Context, Args));
30550b57cec5SDimitry Andric continue;
30560b57cec5SDimitry Andric }
30570b57cec5SDimitry Andric
30580b57cec5SDimitry Andric // Link against a library.
30590b57cec5SDimitry Andric if (IsELF) {
30600b57cec5SDimitry Andric llvm::Metadata *Args[2] = {
30610b57cec5SDimitry Andric llvm::MDString::get(Context, "lib"),
3062349cc55cSDimitry Andric llvm::MDString::get(Context, LL.Library),
30630b57cec5SDimitry Andric };
30640b57cec5SDimitry Andric Metadata.push_back(llvm::MDNode::get(Context, Args));
30650b57cec5SDimitry Andric } else {
30660b57cec5SDimitry Andric llvm::SmallString<24> Opt;
3067349cc55cSDimitry Andric CGM.getTargetCodeGenInfo().getDependentLibraryOption(LL.Library, Opt);
30680b57cec5SDimitry Andric auto *OptString = llvm::MDString::get(Context, Opt);
30690b57cec5SDimitry Andric Metadata.push_back(llvm::MDNode::get(Context, OptString));
30700b57cec5SDimitry Andric }
30710b57cec5SDimitry Andric }
30720b57cec5SDimitry Andric }
30730b57cec5SDimitry Andric
EmitModuleInitializers(clang::Module * Primary)3074fcaf7f86SDimitry Andric void CodeGenModule::EmitModuleInitializers(clang::Module *Primary) {
30755f757f3fSDimitry Andric assert(Primary->isNamedModuleUnit() &&
30765f757f3fSDimitry Andric "We should only emit module initializers for named modules.");
30775f757f3fSDimitry Andric
3078fcaf7f86SDimitry Andric // Emit the initializers in the order that sub-modules appear in the
3079fcaf7f86SDimitry Andric // source, first Global Module Fragments, if present.
3080fcaf7f86SDimitry Andric if (auto GMF = Primary->getGlobalModuleFragment()) {
3081fcaf7f86SDimitry Andric for (Decl *D : getContext().getModuleInitializers(GMF)) {
308261cfbce3SDimitry Andric if (isa<ImportDecl>(D))
308361cfbce3SDimitry Andric continue;
308461cfbce3SDimitry Andric assert(isa<VarDecl>(D) && "GMF initializer decl is not a var?");
3085fcaf7f86SDimitry Andric EmitTopLevelDecl(D);
3086fcaf7f86SDimitry Andric }
3087fcaf7f86SDimitry Andric }
3088fcaf7f86SDimitry Andric // Second any associated with the module, itself.
3089fcaf7f86SDimitry Andric for (Decl *D : getContext().getModuleInitializers(Primary)) {
3090fcaf7f86SDimitry Andric // Skip import decls, the inits for those are called explicitly.
309161cfbce3SDimitry Andric if (isa<ImportDecl>(D))
3092fcaf7f86SDimitry Andric continue;
3093fcaf7f86SDimitry Andric EmitTopLevelDecl(D);
3094fcaf7f86SDimitry Andric }
3095fcaf7f86SDimitry Andric // Third any associated with the Privat eMOdule Fragment, if present.
3096fcaf7f86SDimitry Andric if (auto PMF = Primary->getPrivateModuleFragment()) {
3097fcaf7f86SDimitry Andric for (Decl *D : getContext().getModuleInitializers(PMF)) {
30985f757f3fSDimitry Andric // Skip import decls, the inits for those are called explicitly.
30995f757f3fSDimitry Andric if (isa<ImportDecl>(D))
31005f757f3fSDimitry Andric continue;
310161cfbce3SDimitry Andric assert(isa<VarDecl>(D) && "PMF initializer decl is not a var?");
3102fcaf7f86SDimitry Andric EmitTopLevelDecl(D);
3103fcaf7f86SDimitry Andric }
3104fcaf7f86SDimitry Andric }
3105fcaf7f86SDimitry Andric }
3106fcaf7f86SDimitry Andric
EmitModuleLinkOptions()31070b57cec5SDimitry Andric void CodeGenModule::EmitModuleLinkOptions() {
31080b57cec5SDimitry Andric // Collect the set of all of the modules we want to visit to emit link
31090b57cec5SDimitry Andric // options, which is essentially the imported modules and all of their
31100b57cec5SDimitry Andric // non-explicit child modules.
31110b57cec5SDimitry Andric llvm::SetVector<clang::Module *> LinkModules;
31120b57cec5SDimitry Andric llvm::SmallPtrSet<clang::Module *, 16> Visited;
31130b57cec5SDimitry Andric SmallVector<clang::Module *, 16> Stack;
31140b57cec5SDimitry Andric
31150b57cec5SDimitry Andric // Seed the stack with imported modules.
31160b57cec5SDimitry Andric for (Module *M : ImportedModules) {
31170b57cec5SDimitry Andric // Do not add any link flags when an implementation TU of a module imports
31180b57cec5SDimitry Andric // a header of that same module.
31190b57cec5SDimitry Andric if (M->getTopLevelModuleName() == getLangOpts().CurrentModule &&
31200b57cec5SDimitry Andric !getLangOpts().isCompilingModule())
31210b57cec5SDimitry Andric continue;
31220b57cec5SDimitry Andric if (Visited.insert(M).second)
31230b57cec5SDimitry Andric Stack.push_back(M);
31240b57cec5SDimitry Andric }
31250b57cec5SDimitry Andric
31260b57cec5SDimitry Andric // Find all of the modules to import, making a little effort to prune
31270b57cec5SDimitry Andric // non-leaf modules.
31280b57cec5SDimitry Andric while (!Stack.empty()) {
31290b57cec5SDimitry Andric clang::Module *Mod = Stack.pop_back_val();
31300b57cec5SDimitry Andric
31310b57cec5SDimitry Andric bool AnyChildren = false;
31320b57cec5SDimitry Andric
31330b57cec5SDimitry Andric // Visit the submodules of this module.
31340b57cec5SDimitry Andric for (const auto &SM : Mod->submodules()) {
31350b57cec5SDimitry Andric // Skip explicit children; they need to be explicitly imported to be
31360b57cec5SDimitry Andric // linked against.
31370b57cec5SDimitry Andric if (SM->IsExplicit)
31380b57cec5SDimitry Andric continue;
31390b57cec5SDimitry Andric
31400b57cec5SDimitry Andric if (Visited.insert(SM).second) {
31410b57cec5SDimitry Andric Stack.push_back(SM);
31420b57cec5SDimitry Andric AnyChildren = true;
31430b57cec5SDimitry Andric }
31440b57cec5SDimitry Andric }
31450b57cec5SDimitry Andric
31460b57cec5SDimitry Andric // We didn't find any children, so add this module to the list of
31470b57cec5SDimitry Andric // modules to link against.
31480b57cec5SDimitry Andric if (!AnyChildren) {
31490b57cec5SDimitry Andric LinkModules.insert(Mod);
31500b57cec5SDimitry Andric }
31510b57cec5SDimitry Andric }
31520b57cec5SDimitry Andric
31530b57cec5SDimitry Andric // Add link options for all of the imported modules in reverse topological
31540b57cec5SDimitry Andric // order. We don't do anything to try to order import link flags with respect
31550b57cec5SDimitry Andric // to linker options inserted by things like #pragma comment().
31560b57cec5SDimitry Andric SmallVector<llvm::MDNode *, 16> MetadataArgs;
31570b57cec5SDimitry Andric Visited.clear();
31580b57cec5SDimitry Andric for (Module *M : LinkModules)
31590b57cec5SDimitry Andric if (Visited.insert(M).second)
31600b57cec5SDimitry Andric addLinkOptionsPostorder(*this, M, MetadataArgs, Visited);
31610b57cec5SDimitry Andric std::reverse(MetadataArgs.begin(), MetadataArgs.end());
31620b57cec5SDimitry Andric LinkerOptionsMetadata.append(MetadataArgs.begin(), MetadataArgs.end());
31630b57cec5SDimitry Andric
31640b57cec5SDimitry Andric // Add the linker options metadata flag.
31650b57cec5SDimitry Andric auto *NMD = getModule().getOrInsertNamedMetadata("llvm.linker.options");
31660b57cec5SDimitry Andric for (auto *MD : LinkerOptionsMetadata)
31670b57cec5SDimitry Andric NMD->addOperand(MD);
31680b57cec5SDimitry Andric }
31690b57cec5SDimitry Andric
EmitDeferred()31700b57cec5SDimitry Andric void CodeGenModule::EmitDeferred() {
31710b57cec5SDimitry Andric // Emit deferred declare target declarations.
31720b57cec5SDimitry Andric if (getLangOpts().OpenMP && !getLangOpts().OpenMPSimd)
31730b57cec5SDimitry Andric getOpenMPRuntime().emitDeferredTargetDecls();
31740b57cec5SDimitry Andric
31750b57cec5SDimitry Andric // Emit code for any potentially referenced deferred decls. Since a
31760b57cec5SDimitry Andric // previously unused static decl may become used during the generation of code
31770b57cec5SDimitry Andric // for a static function, iterate until no changes are made.
31780b57cec5SDimitry Andric
31790b57cec5SDimitry Andric if (!DeferredVTables.empty()) {
31800b57cec5SDimitry Andric EmitDeferredVTables();
31810b57cec5SDimitry Andric
31820b57cec5SDimitry Andric // Emitting a vtable doesn't directly cause more vtables to
31830b57cec5SDimitry Andric // become deferred, although it can cause functions to be
31840b57cec5SDimitry Andric // emitted that then need those vtables.
31850b57cec5SDimitry Andric assert(DeferredVTables.empty());
31860b57cec5SDimitry Andric }
31870b57cec5SDimitry Andric
3188e8d8bef9SDimitry Andric // Emit CUDA/HIP static device variables referenced by host code only.
3189fe6060f1SDimitry Andric // Note we should not clear CUDADeviceVarODRUsedByHost since it is still
3190fe6060f1SDimitry Andric // needed for further handling.
3191fe6060f1SDimitry Andric if (getLangOpts().CUDA && getLangOpts().CUDAIsDevice)
319281ad6265SDimitry Andric llvm::append_range(DeferredDeclsToEmit,
319381ad6265SDimitry Andric getContext().CUDADeviceVarODRUsedByHost);
3194e8d8bef9SDimitry Andric
31950b57cec5SDimitry Andric // Stop if we're out of both deferred vtables and deferred declarations.
31960b57cec5SDimitry Andric if (DeferredDeclsToEmit.empty())
31970b57cec5SDimitry Andric return;
31980b57cec5SDimitry Andric
31990b57cec5SDimitry Andric // Grab the list of decls to emit. If EmitGlobalDefinition schedules more
32000b57cec5SDimitry Andric // work, it will not interfere with this.
32010b57cec5SDimitry Andric std::vector<GlobalDecl> CurDeclsToEmit;
32020b57cec5SDimitry Andric CurDeclsToEmit.swap(DeferredDeclsToEmit);
32030b57cec5SDimitry Andric
32040b57cec5SDimitry Andric for (GlobalDecl &D : CurDeclsToEmit) {
32050b57cec5SDimitry Andric // We should call GetAddrOfGlobal with IsForDefinition set to true in order
32060b57cec5SDimitry Andric // to get GlobalValue with exactly the type we need, not something that
32070b57cec5SDimitry Andric // might had been created for another decl with the same mangled name but
32080b57cec5SDimitry Andric // different type.
32090b57cec5SDimitry Andric llvm::GlobalValue *GV = dyn_cast<llvm::GlobalValue>(
32100b57cec5SDimitry Andric GetAddrOfGlobal(D, ForDefinition));
32110b57cec5SDimitry Andric
32120b57cec5SDimitry Andric // In case of different address spaces, we may still get a cast, even with
32130b57cec5SDimitry Andric // IsForDefinition equal to true. Query mangled names table to get
32140b57cec5SDimitry Andric // GlobalValue.
32150b57cec5SDimitry Andric if (!GV)
32160b57cec5SDimitry Andric GV = GetGlobalValue(getMangledName(D));
32170b57cec5SDimitry Andric
32180b57cec5SDimitry Andric // Make sure GetGlobalValue returned non-null.
32190b57cec5SDimitry Andric assert(GV);
32200b57cec5SDimitry Andric
32210b57cec5SDimitry Andric // Check to see if we've already emitted this. This is necessary
32220b57cec5SDimitry Andric // for a couple of reasons: first, decls can end up in the
32230b57cec5SDimitry Andric // deferred-decls queue multiple times, and second, decls can end
32240b57cec5SDimitry Andric // up with definitions in unusual ways (e.g. by an extern inline
32250b57cec5SDimitry Andric // function acquiring a strong function redefinition). Just
32260b57cec5SDimitry Andric // ignore these cases.
32270b57cec5SDimitry Andric if (!GV->isDeclaration())
32280b57cec5SDimitry Andric continue;
32290b57cec5SDimitry Andric
3230a7dea167SDimitry Andric // If this is OpenMP, check if it is legal to emit this global normally.
3231a7dea167SDimitry Andric if (LangOpts.OpenMP && OpenMPRuntime && OpenMPRuntime->emitTargetGlobal(D))
3232a7dea167SDimitry Andric continue;
3233a7dea167SDimitry Andric
32340b57cec5SDimitry Andric // Otherwise, emit the definition and move on to the next one.
32350b57cec5SDimitry Andric EmitGlobalDefinition(D, GV);
32360b57cec5SDimitry Andric
32370b57cec5SDimitry Andric // If we found out that we need to emit more decls, do that recursively.
32380b57cec5SDimitry Andric // This has the advantage that the decls are emitted in a DFS and related
32390b57cec5SDimitry Andric // ones are close together, which is convenient for testing.
32400b57cec5SDimitry Andric if (!DeferredVTables.empty() || !DeferredDeclsToEmit.empty()) {
32410b57cec5SDimitry Andric EmitDeferred();
32420b57cec5SDimitry Andric assert(DeferredVTables.empty() && DeferredDeclsToEmit.empty());
32430b57cec5SDimitry Andric }
32440b57cec5SDimitry Andric }
32450b57cec5SDimitry Andric }
32460b57cec5SDimitry Andric
EmitVTablesOpportunistically()32470b57cec5SDimitry Andric void CodeGenModule::EmitVTablesOpportunistically() {
32480b57cec5SDimitry Andric // Try to emit external vtables as available_externally if they have emitted
32490b57cec5SDimitry Andric // all inlined virtual functions. It runs after EmitDeferred() and therefore
32500b57cec5SDimitry Andric // is not allowed to create new references to things that need to be emitted
32510b57cec5SDimitry Andric // lazily. Note that it also uses fact that we eagerly emitting RTTI.
32520b57cec5SDimitry Andric
32530b57cec5SDimitry Andric assert((OpportunisticVTables.empty() || shouldOpportunisticallyEmitVTables())
32540b57cec5SDimitry Andric && "Only emit opportunistic vtables with optimizations");
32550b57cec5SDimitry Andric
32560b57cec5SDimitry Andric for (const CXXRecordDecl *RD : OpportunisticVTables) {
32570b57cec5SDimitry Andric assert(getVTables().isVTableExternal(RD) &&
32580b57cec5SDimitry Andric "This queue should only contain external vtables");
32590b57cec5SDimitry Andric if (getCXXABI().canSpeculativelyEmitVTable(RD))
32600b57cec5SDimitry Andric VTables.GenerateClassData(RD);
32610b57cec5SDimitry Andric }
32620b57cec5SDimitry Andric OpportunisticVTables.clear();
32630b57cec5SDimitry Andric }
32640b57cec5SDimitry Andric
EmitGlobalAnnotations()32650b57cec5SDimitry Andric void CodeGenModule::EmitGlobalAnnotations() {
32665f757f3fSDimitry Andric for (const auto& [MangledName, VD] : DeferredAnnotations) {
32675f757f3fSDimitry Andric llvm::GlobalValue *GV = GetGlobalValue(MangledName);
32685f757f3fSDimitry Andric if (GV)
32695f757f3fSDimitry Andric AddGlobalAnnotations(VD, GV);
32705f757f3fSDimitry Andric }
32715f757f3fSDimitry Andric DeferredAnnotations.clear();
32725f757f3fSDimitry Andric
32730b57cec5SDimitry Andric if (Annotations.empty())
32740b57cec5SDimitry Andric return;
32750b57cec5SDimitry Andric
32760b57cec5SDimitry Andric // Create a new global variable for the ConstantStruct in the Module.
32770b57cec5SDimitry Andric llvm::Constant *Array = llvm::ConstantArray::get(llvm::ArrayType::get(
32780b57cec5SDimitry Andric Annotations[0]->getType(), Annotations.size()), Annotations);
32790b57cec5SDimitry Andric auto *gv = new llvm::GlobalVariable(getModule(), Array->getType(), false,
32800b57cec5SDimitry Andric llvm::GlobalValue::AppendingLinkage,
32810b57cec5SDimitry Andric Array, "llvm.global.annotations");
32820b57cec5SDimitry Andric gv->setSection(AnnotationSection);
32830b57cec5SDimitry Andric }
32840b57cec5SDimitry Andric
EmitAnnotationString(StringRef Str)32850b57cec5SDimitry Andric llvm::Constant *CodeGenModule::EmitAnnotationString(StringRef Str) {
32860b57cec5SDimitry Andric llvm::Constant *&AStr = AnnotationStrings[Str];
32870b57cec5SDimitry Andric if (AStr)
32880b57cec5SDimitry Andric return AStr;
32890b57cec5SDimitry Andric
32900b57cec5SDimitry Andric // Not found yet, create a new global.
32910b57cec5SDimitry Andric llvm::Constant *s = llvm::ConstantDataArray::getString(getLLVMContext(), Str);
3292bdd1243dSDimitry Andric auto *gv = new llvm::GlobalVariable(
3293bdd1243dSDimitry Andric getModule(), s->getType(), true, llvm::GlobalValue::PrivateLinkage, s,
3294bdd1243dSDimitry Andric ".str", nullptr, llvm::GlobalValue::NotThreadLocal,
3295bdd1243dSDimitry Andric ConstGlobalsPtrTy->getAddressSpace());
32960b57cec5SDimitry Andric gv->setSection(AnnotationSection);
32970b57cec5SDimitry Andric gv->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
32980b57cec5SDimitry Andric AStr = gv;
32990b57cec5SDimitry Andric return gv;
33000b57cec5SDimitry Andric }
33010b57cec5SDimitry Andric
EmitAnnotationUnit(SourceLocation Loc)33020b57cec5SDimitry Andric llvm::Constant *CodeGenModule::EmitAnnotationUnit(SourceLocation Loc) {
33030b57cec5SDimitry Andric SourceManager &SM = getContext().getSourceManager();
33040b57cec5SDimitry Andric PresumedLoc PLoc = SM.getPresumedLoc(Loc);
33050b57cec5SDimitry Andric if (PLoc.isValid())
33060b57cec5SDimitry Andric return EmitAnnotationString(PLoc.getFilename());
33070b57cec5SDimitry Andric return EmitAnnotationString(SM.getBufferName(Loc));
33080b57cec5SDimitry Andric }
33090b57cec5SDimitry Andric
EmitAnnotationLineNo(SourceLocation L)33100b57cec5SDimitry Andric llvm::Constant *CodeGenModule::EmitAnnotationLineNo(SourceLocation L) {
33110b57cec5SDimitry Andric SourceManager &SM = getContext().getSourceManager();
33120b57cec5SDimitry Andric PresumedLoc PLoc = SM.getPresumedLoc(L);
33130b57cec5SDimitry Andric unsigned LineNo = PLoc.isValid() ? PLoc.getLine() :
33140b57cec5SDimitry Andric SM.getExpansionLineNumber(L);
33150b57cec5SDimitry Andric return llvm::ConstantInt::get(Int32Ty, LineNo);
33160b57cec5SDimitry Andric }
33170b57cec5SDimitry Andric
EmitAnnotationArgs(const AnnotateAttr * Attr)3318e8d8bef9SDimitry Andric llvm::Constant *CodeGenModule::EmitAnnotationArgs(const AnnotateAttr *Attr) {
3319e8d8bef9SDimitry Andric ArrayRef<Expr *> Exprs = {Attr->args_begin(), Attr->args_size()};
3320e8d8bef9SDimitry Andric if (Exprs.empty())
3321bdd1243dSDimitry Andric return llvm::ConstantPointerNull::get(ConstGlobalsPtrTy);
3322e8d8bef9SDimitry Andric
3323e8d8bef9SDimitry Andric llvm::FoldingSetNodeID ID;
3324e8d8bef9SDimitry Andric for (Expr *E : Exprs) {
3325e8d8bef9SDimitry Andric ID.Add(cast<clang::ConstantExpr>(E)->getAPValueResult());
3326e8d8bef9SDimitry Andric }
3327e8d8bef9SDimitry Andric llvm::Constant *&Lookup = AnnotationArgs[ID.ComputeHash()];
3328e8d8bef9SDimitry Andric if (Lookup)
3329e8d8bef9SDimitry Andric return Lookup;
3330e8d8bef9SDimitry Andric
3331e8d8bef9SDimitry Andric llvm::SmallVector<llvm::Constant *, 4> LLVMArgs;
3332e8d8bef9SDimitry Andric LLVMArgs.reserve(Exprs.size());
3333e8d8bef9SDimitry Andric ConstantEmitter ConstEmiter(*this);
3334e8d8bef9SDimitry Andric llvm::transform(Exprs, std::back_inserter(LLVMArgs), [&](const Expr *E) {
3335e8d8bef9SDimitry Andric const auto *CE = cast<clang::ConstantExpr>(E);
3336e8d8bef9SDimitry Andric return ConstEmiter.emitAbstract(CE->getBeginLoc(), CE->getAPValueResult(),
3337e8d8bef9SDimitry Andric CE->getType());
3338e8d8bef9SDimitry Andric });
3339e8d8bef9SDimitry Andric auto *Struct = llvm::ConstantStruct::getAnon(LLVMArgs);
3340e8d8bef9SDimitry Andric auto *GV = new llvm::GlobalVariable(getModule(), Struct->getType(), true,
3341e8d8bef9SDimitry Andric llvm::GlobalValue::PrivateLinkage, Struct,
3342e8d8bef9SDimitry Andric ".args");
3343e8d8bef9SDimitry Andric GV->setSection(AnnotationSection);
3344e8d8bef9SDimitry Andric GV->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
3345e8d8bef9SDimitry Andric
33465f757f3fSDimitry Andric Lookup = GV;
33475f757f3fSDimitry Andric return GV;
3348e8d8bef9SDimitry Andric }
3349e8d8bef9SDimitry Andric
EmitAnnotateAttr(llvm::GlobalValue * GV,const AnnotateAttr * AA,SourceLocation L)33500b57cec5SDimitry Andric llvm::Constant *CodeGenModule::EmitAnnotateAttr(llvm::GlobalValue *GV,
33510b57cec5SDimitry Andric const AnnotateAttr *AA,
33520b57cec5SDimitry Andric SourceLocation L) {
33530b57cec5SDimitry Andric // Get the globals for file name, annotation, and the line number.
33540b57cec5SDimitry Andric llvm::Constant *AnnoGV = EmitAnnotationString(AA->getAnnotation()),
33550b57cec5SDimitry Andric *UnitGV = EmitAnnotationUnit(L),
3356e8d8bef9SDimitry Andric *LineNoCst = EmitAnnotationLineNo(L),
3357e8d8bef9SDimitry Andric *Args = EmitAnnotationArgs(AA);
33580b57cec5SDimitry Andric
3359349cc55cSDimitry Andric llvm::Constant *GVInGlobalsAS = GV;
3360349cc55cSDimitry Andric if (GV->getAddressSpace() !=
3361349cc55cSDimitry Andric getDataLayout().getDefaultGlobalsAddressSpace()) {
3362349cc55cSDimitry Andric GVInGlobalsAS = llvm::ConstantExpr::getAddrSpaceCast(
33635f757f3fSDimitry Andric GV,
33645f757f3fSDimitry Andric llvm::PointerType::get(
33655f757f3fSDimitry Andric GV->getContext(), getDataLayout().getDefaultGlobalsAddressSpace()));
3366480093f4SDimitry Andric }
3367480093f4SDimitry Andric
33680b57cec5SDimitry Andric // Create the ConstantStruct for the global annotation.
3369e8d8bef9SDimitry Andric llvm::Constant *Fields[] = {
33705f757f3fSDimitry Andric GVInGlobalsAS, AnnoGV, UnitGV, LineNoCst, Args,
33710b57cec5SDimitry Andric };
33720b57cec5SDimitry Andric return llvm::ConstantStruct::getAnon(Fields);
33730b57cec5SDimitry Andric }
33740b57cec5SDimitry Andric
AddGlobalAnnotations(const ValueDecl * D,llvm::GlobalValue * GV)33750b57cec5SDimitry Andric void CodeGenModule::AddGlobalAnnotations(const ValueDecl *D,
33760b57cec5SDimitry Andric llvm::GlobalValue *GV) {
33770b57cec5SDimitry Andric assert(D->hasAttr<AnnotateAttr>() && "no annotate attribute");
33780b57cec5SDimitry Andric // Get the struct elements for these annotations.
33790b57cec5SDimitry Andric for (const auto *I : D->specific_attrs<AnnotateAttr>())
33800b57cec5SDimitry Andric Annotations.push_back(EmitAnnotateAttr(GV, I, D->getLocation()));
33810b57cec5SDimitry Andric }
33820b57cec5SDimitry Andric
isInNoSanitizeList(SanitizerMask Kind,llvm::Function * Fn,SourceLocation Loc) const3383fe6060f1SDimitry Andric bool CodeGenModule::isInNoSanitizeList(SanitizerMask Kind, llvm::Function *Fn,
33840b57cec5SDimitry Andric SourceLocation Loc) const {
3385fe6060f1SDimitry Andric const auto &NoSanitizeL = getContext().getNoSanitizeList();
3386fe6060f1SDimitry Andric // NoSanitize by function name.
3387fe6060f1SDimitry Andric if (NoSanitizeL.containsFunction(Kind, Fn->getName()))
33880b57cec5SDimitry Andric return true;
3389fcaf7f86SDimitry Andric // NoSanitize by location. Check "mainfile" prefix.
3390fcaf7f86SDimitry Andric auto &SM = Context.getSourceManager();
33915f757f3fSDimitry Andric FileEntryRef MainFile = *SM.getFileEntryRefForID(SM.getMainFileID());
3392fcaf7f86SDimitry Andric if (NoSanitizeL.containsMainFile(Kind, MainFile.getName()))
3393fcaf7f86SDimitry Andric return true;
3394fcaf7f86SDimitry Andric
3395fcaf7f86SDimitry Andric // Check "src" prefix.
33960b57cec5SDimitry Andric if (Loc.isValid())
3397fe6060f1SDimitry Andric return NoSanitizeL.containsLocation(Kind, Loc);
33980b57cec5SDimitry Andric // If location is unknown, this may be a compiler-generated function. Assume
33990b57cec5SDimitry Andric // it's located in the main file.
3400fcaf7f86SDimitry Andric return NoSanitizeL.containsFile(Kind, MainFile.getName());
34010b57cec5SDimitry Andric }
34020b57cec5SDimitry Andric
isInNoSanitizeList(SanitizerMask Kind,llvm::GlobalVariable * GV,SourceLocation Loc,QualType Ty,StringRef Category) const340381ad6265SDimitry Andric bool CodeGenModule::isInNoSanitizeList(SanitizerMask Kind,
340481ad6265SDimitry Andric llvm::GlobalVariable *GV,
34050b57cec5SDimitry Andric SourceLocation Loc, QualType Ty,
34060b57cec5SDimitry Andric StringRef Category) const {
3407fe6060f1SDimitry Andric const auto &NoSanitizeL = getContext().getNoSanitizeList();
340881ad6265SDimitry Andric if (NoSanitizeL.containsGlobal(Kind, GV->getName(), Category))
34090b57cec5SDimitry Andric return true;
3410fcaf7f86SDimitry Andric auto &SM = Context.getSourceManager();
3411fcaf7f86SDimitry Andric if (NoSanitizeL.containsMainFile(
34125f757f3fSDimitry Andric Kind, SM.getFileEntryRefForID(SM.getMainFileID())->getName(),
34135f757f3fSDimitry Andric Category))
3414fcaf7f86SDimitry Andric return true;
341581ad6265SDimitry Andric if (NoSanitizeL.containsLocation(Kind, Loc, Category))
34160b57cec5SDimitry Andric return true;
3417fcaf7f86SDimitry Andric
34180b57cec5SDimitry Andric // Check global type.
34190b57cec5SDimitry Andric if (!Ty.isNull()) {
34200b57cec5SDimitry Andric // Drill down the array types: if global variable of a fixed type is
3421fe6060f1SDimitry Andric // not sanitized, we also don't instrument arrays of them.
34220b57cec5SDimitry Andric while (auto AT = dyn_cast<ArrayType>(Ty.getTypePtr()))
34230b57cec5SDimitry Andric Ty = AT->getElementType();
34240b57cec5SDimitry Andric Ty = Ty.getCanonicalType().getUnqualifiedType();
3425fe6060f1SDimitry Andric // Only record types (classes, structs etc.) are ignored.
34260b57cec5SDimitry Andric if (Ty->isRecordType()) {
34270b57cec5SDimitry Andric std::string TypeStr = Ty.getAsString(getContext().getPrintingPolicy());
342881ad6265SDimitry Andric if (NoSanitizeL.containsType(Kind, TypeStr, Category))
34290b57cec5SDimitry Andric return true;
34300b57cec5SDimitry Andric }
34310b57cec5SDimitry Andric }
34320b57cec5SDimitry Andric return false;
34330b57cec5SDimitry Andric }
34340b57cec5SDimitry Andric
imbueXRayAttrs(llvm::Function * Fn,SourceLocation Loc,StringRef Category) const34350b57cec5SDimitry Andric bool CodeGenModule::imbueXRayAttrs(llvm::Function *Fn, SourceLocation Loc,
34360b57cec5SDimitry Andric StringRef Category) const {
34370b57cec5SDimitry Andric const auto &XRayFilter = getContext().getXRayFilter();
34380b57cec5SDimitry Andric using ImbueAttr = XRayFunctionFilter::ImbueAttribute;
34390b57cec5SDimitry Andric auto Attr = ImbueAttr::NONE;
34400b57cec5SDimitry Andric if (Loc.isValid())
34410b57cec5SDimitry Andric Attr = XRayFilter.shouldImbueLocation(Loc, Category);
34420b57cec5SDimitry Andric if (Attr == ImbueAttr::NONE)
34430b57cec5SDimitry Andric Attr = XRayFilter.shouldImbueFunction(Fn->getName());
34440b57cec5SDimitry Andric switch (Attr) {
34450b57cec5SDimitry Andric case ImbueAttr::NONE:
34460b57cec5SDimitry Andric return false;
34470b57cec5SDimitry Andric case ImbueAttr::ALWAYS:
34480b57cec5SDimitry Andric Fn->addFnAttr("function-instrument", "xray-always");
34490b57cec5SDimitry Andric break;
34500b57cec5SDimitry Andric case ImbueAttr::ALWAYS_ARG1:
34510b57cec5SDimitry Andric Fn->addFnAttr("function-instrument", "xray-always");
34520b57cec5SDimitry Andric Fn->addFnAttr("xray-log-args", "1");
34530b57cec5SDimitry Andric break;
34540b57cec5SDimitry Andric case ImbueAttr::NEVER:
34550b57cec5SDimitry Andric Fn->addFnAttr("function-instrument", "xray-never");
34560b57cec5SDimitry Andric break;
34570b57cec5SDimitry Andric }
34580b57cec5SDimitry Andric return true;
34590b57cec5SDimitry Andric }
34600b57cec5SDimitry Andric
3461bdd1243dSDimitry Andric ProfileList::ExclusionType
isFunctionBlockedByProfileList(llvm::Function * Fn,SourceLocation Loc) const3462bdd1243dSDimitry Andric CodeGenModule::isFunctionBlockedByProfileList(llvm::Function *Fn,
3463e8d8bef9SDimitry Andric SourceLocation Loc) const {
3464e8d8bef9SDimitry Andric const auto &ProfileList = getContext().getProfileList();
3465e8d8bef9SDimitry Andric // If the profile list is empty, then instrument everything.
3466e8d8bef9SDimitry Andric if (ProfileList.isEmpty())
3467bdd1243dSDimitry Andric return ProfileList::Allow;
3468e8d8bef9SDimitry Andric CodeGenOptions::ProfileInstrKind Kind = getCodeGenOpts().getProfileInstr();
3469e8d8bef9SDimitry Andric // First, check the function name.
3470bdd1243dSDimitry Andric if (auto V = ProfileList.isFunctionExcluded(Fn->getName(), Kind))
3471e8d8bef9SDimitry Andric return *V;
3472e8d8bef9SDimitry Andric // Next, check the source location.
3473bdd1243dSDimitry Andric if (Loc.isValid())
3474bdd1243dSDimitry Andric if (auto V = ProfileList.isLocationExcluded(Loc, Kind))
3475e8d8bef9SDimitry Andric return *V;
3476e8d8bef9SDimitry Andric // If location is unknown, this may be a compiler-generated function. Assume
3477e8d8bef9SDimitry Andric // it's located in the main file.
3478e8d8bef9SDimitry Andric auto &SM = Context.getSourceManager();
34795f757f3fSDimitry Andric if (auto MainFile = SM.getFileEntryRefForID(SM.getMainFileID()))
3480bdd1243dSDimitry Andric if (auto V = ProfileList.isFileExcluded(MainFile->getName(), Kind))
3481e8d8bef9SDimitry Andric return *V;
3482bdd1243dSDimitry Andric return ProfileList.getDefault(Kind);
3483e8d8bef9SDimitry Andric }
3484e8d8bef9SDimitry Andric
3485bdd1243dSDimitry Andric ProfileList::ExclusionType
isFunctionBlockedFromProfileInstr(llvm::Function * Fn,SourceLocation Loc) const3486bdd1243dSDimitry Andric CodeGenModule::isFunctionBlockedFromProfileInstr(llvm::Function *Fn,
3487bdd1243dSDimitry Andric SourceLocation Loc) const {
3488bdd1243dSDimitry Andric auto V = isFunctionBlockedByProfileList(Fn, Loc);
3489bdd1243dSDimitry Andric if (V != ProfileList::Allow)
3490bdd1243dSDimitry Andric return V;
3491fcaf7f86SDimitry Andric
3492fcaf7f86SDimitry Andric auto NumGroups = getCodeGenOpts().ProfileTotalFunctionGroups;
3493fcaf7f86SDimitry Andric if (NumGroups > 1) {
3494fcaf7f86SDimitry Andric auto Group = llvm::crc32(arrayRefFromStringRef(Fn->getName())) % NumGroups;
3495fcaf7f86SDimitry Andric if (Group != getCodeGenOpts().ProfileSelectedFunctionGroup)
3496bdd1243dSDimitry Andric return ProfileList::Skip;
3497fcaf7f86SDimitry Andric }
3498bdd1243dSDimitry Andric return ProfileList::Allow;
3499fcaf7f86SDimitry Andric }
3500fcaf7f86SDimitry Andric
MustBeEmitted(const ValueDecl * Global)35010b57cec5SDimitry Andric bool CodeGenModule::MustBeEmitted(const ValueDecl *Global) {
35020b57cec5SDimitry Andric // Never defer when EmitAllDecls is specified.
35030b57cec5SDimitry Andric if (LangOpts.EmitAllDecls)
35040b57cec5SDimitry Andric return true;
35050b57cec5SDimitry Andric
35060b57cec5SDimitry Andric const auto *VD = dyn_cast<VarDecl>(Global);
350706c3fb27SDimitry Andric if (VD &&
350806c3fb27SDimitry Andric ((CodeGenOpts.KeepPersistentStorageVariables &&
350906c3fb27SDimitry Andric (VD->getStorageDuration() == SD_Static ||
351006c3fb27SDimitry Andric VD->getStorageDuration() == SD_Thread)) ||
351106c3fb27SDimitry Andric (CodeGenOpts.KeepStaticConsts && VD->getStorageDuration() == SD_Static &&
351206c3fb27SDimitry Andric VD->getType().isConstQualified())))
35130b57cec5SDimitry Andric return true;
35140b57cec5SDimitry Andric
35150b57cec5SDimitry Andric return getContext().DeclMustBeEmitted(Global);
35160b57cec5SDimitry Andric }
35170b57cec5SDimitry Andric
MayBeEmittedEagerly(const ValueDecl * Global)35180b57cec5SDimitry Andric bool CodeGenModule::MayBeEmittedEagerly(const ValueDecl *Global) {
3519fe6060f1SDimitry Andric // In OpenMP 5.0 variables and function may be marked as
3520fe6060f1SDimitry Andric // device_type(host/nohost) and we should not emit them eagerly unless we sure
3521fe6060f1SDimitry Andric // that they must be emitted on the host/device. To be sure we need to have
3522fe6060f1SDimitry Andric // seen a declare target with an explicit mentioning of the function, we know
3523fe6060f1SDimitry Andric // we have if the level of the declare target attribute is -1. Note that we
3524fe6060f1SDimitry Andric // check somewhere else if we should emit this at all.
3525fe6060f1SDimitry Andric if (LangOpts.OpenMP >= 50 && !LangOpts.OpenMPSimd) {
3526bdd1243dSDimitry Andric std::optional<OMPDeclareTargetDeclAttr *> ActiveAttr =
3527fe6060f1SDimitry Andric OMPDeclareTargetDeclAttr::getActiveAttr(Global);
3528fe6060f1SDimitry Andric if (!ActiveAttr || (*ActiveAttr)->getLevel() != (unsigned)-1)
3529fe6060f1SDimitry Andric return false;
3530fe6060f1SDimitry Andric }
3531fe6060f1SDimitry Andric
3532a7dea167SDimitry Andric if (const auto *FD = dyn_cast<FunctionDecl>(Global)) {
35330b57cec5SDimitry Andric if (FD->getTemplateSpecializationKind() == TSK_ImplicitInstantiation)
35340b57cec5SDimitry Andric // Implicit template instantiations may change linkage if they are later
35350b57cec5SDimitry Andric // explicitly instantiated, so they should not be emitted eagerly.
35360b57cec5SDimitry Andric return false;
35370fca6ea1SDimitry Andric // Defer until all versions have been semantically checked.
35380fca6ea1SDimitry Andric if (FD->hasAttr<TargetVersionAttr>() && !FD->isMultiVersion())
35390fca6ea1SDimitry Andric return false;
3540a7dea167SDimitry Andric }
3541fcaf7f86SDimitry Andric if (const auto *VD = dyn_cast<VarDecl>(Global)) {
35420b57cec5SDimitry Andric if (Context.getInlineVariableDefinitionKind(VD) ==
35430b57cec5SDimitry Andric ASTContext::InlineVariableDefinitionKind::WeakUnknown)
35440b57cec5SDimitry Andric // A definition of an inline constexpr static data member may change
35450b57cec5SDimitry Andric // linkage later if it's redeclared outside the class.
35460b57cec5SDimitry Andric return false;
3547fcaf7f86SDimitry Andric if (CXX20ModuleInits && VD->getOwningModule() &&
3548fcaf7f86SDimitry Andric !VD->getOwningModule()->isModuleMapModule()) {
3549fcaf7f86SDimitry Andric // For CXX20, module-owned initializers need to be deferred, since it is
3550fcaf7f86SDimitry Andric // not known at this point if they will be run for the current module or
3551fcaf7f86SDimitry Andric // as part of the initializer for an imported one.
3552fcaf7f86SDimitry Andric return false;
3553fcaf7f86SDimitry Andric }
3554fcaf7f86SDimitry Andric }
35550b57cec5SDimitry Andric // If OpenMP is enabled and threadprivates must be generated like TLS, delay
35560b57cec5SDimitry Andric // codegen for global variables, because they may be marked as threadprivate.
35570b57cec5SDimitry Andric if (LangOpts.OpenMP && LangOpts.OpenMPUseTLS &&
35580b57cec5SDimitry Andric getContext().getTargetInfo().isTLSSupported() && isa<VarDecl>(Global) &&
35595f757f3fSDimitry Andric !Global->getType().isConstantStorage(getContext(), false, false) &&
35600b57cec5SDimitry Andric !OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(Global))
35610b57cec5SDimitry Andric return false;
35620b57cec5SDimitry Andric
35630b57cec5SDimitry Andric return true;
35640b57cec5SDimitry Andric }
35650b57cec5SDimitry Andric
GetAddrOfMSGuidDecl(const MSGuidDecl * GD)35665ffd83dbSDimitry Andric ConstantAddress CodeGenModule::GetAddrOfMSGuidDecl(const MSGuidDecl *GD) {
35675ffd83dbSDimitry Andric StringRef Name = getMangledName(GD);
35680b57cec5SDimitry Andric
35690b57cec5SDimitry Andric // The UUID descriptor should be pointer aligned.
35700b57cec5SDimitry Andric CharUnits Alignment = CharUnits::fromQuantity(PointerAlignInBytes);
35710b57cec5SDimitry Andric
35720b57cec5SDimitry Andric // Look for an existing global.
35730b57cec5SDimitry Andric if (llvm::GlobalVariable *GV = getModule().getNamedGlobal(Name))
35740eae32dcSDimitry Andric return ConstantAddress(GV, GV->getValueType(), Alignment);
35750b57cec5SDimitry Andric
35765ffd83dbSDimitry Andric ConstantEmitter Emitter(*this);
35775ffd83dbSDimitry Andric llvm::Constant *Init;
35785ffd83dbSDimitry Andric
35795ffd83dbSDimitry Andric APValue &V = GD->getAsAPValue();
35805ffd83dbSDimitry Andric if (!V.isAbsent()) {
35815ffd83dbSDimitry Andric // If possible, emit the APValue version of the initializer. In particular,
35825ffd83dbSDimitry Andric // this gets the type of the constant right.
35835ffd83dbSDimitry Andric Init = Emitter.emitForInitializer(
35845ffd83dbSDimitry Andric GD->getAsAPValue(), GD->getType().getAddressSpace(), GD->getType());
35855ffd83dbSDimitry Andric } else {
35865ffd83dbSDimitry Andric // As a fallback, directly construct the constant.
35875ffd83dbSDimitry Andric // FIXME: This may get padding wrong under esoteric struct layout rules.
35885ffd83dbSDimitry Andric // MSVC appears to create a complete type 'struct __s_GUID' that it
35895ffd83dbSDimitry Andric // presumably uses to represent these constants.
35905ffd83dbSDimitry Andric MSGuidDecl::Parts Parts = GD->getParts();
35915ffd83dbSDimitry Andric llvm::Constant *Fields[4] = {
35925ffd83dbSDimitry Andric llvm::ConstantInt::get(Int32Ty, Parts.Part1),
35935ffd83dbSDimitry Andric llvm::ConstantInt::get(Int16Ty, Parts.Part2),
35945ffd83dbSDimitry Andric llvm::ConstantInt::get(Int16Ty, Parts.Part3),
35955ffd83dbSDimitry Andric llvm::ConstantDataArray::getRaw(
35965ffd83dbSDimitry Andric StringRef(reinterpret_cast<char *>(Parts.Part4And5), 8), 8,
35975ffd83dbSDimitry Andric Int8Ty)};
35985ffd83dbSDimitry Andric Init = llvm::ConstantStruct::getAnon(Fields);
35995ffd83dbSDimitry Andric }
36000b57cec5SDimitry Andric
36010b57cec5SDimitry Andric auto *GV = new llvm::GlobalVariable(
36020b57cec5SDimitry Andric getModule(), Init->getType(),
36030b57cec5SDimitry Andric /*isConstant=*/true, llvm::GlobalValue::LinkOnceODRLinkage, Init, Name);
36040b57cec5SDimitry Andric if (supportsCOMDAT())
36050b57cec5SDimitry Andric GV->setComdat(TheModule.getOrInsertComdat(GV->getName()));
36060b57cec5SDimitry Andric setDSOLocal(GV);
36075ffd83dbSDimitry Andric
36085ffd83dbSDimitry Andric if (!V.isAbsent()) {
36095ffd83dbSDimitry Andric Emitter.finalize(GV);
36100eae32dcSDimitry Andric return ConstantAddress(GV, GV->getValueType(), Alignment);
36115ffd83dbSDimitry Andric }
36120eae32dcSDimitry Andric
36130eae32dcSDimitry Andric llvm::Type *Ty = getTypes().ConvertTypeForMem(GD->getType());
36145f757f3fSDimitry Andric return ConstantAddress(GV, Ty, Alignment);
36150b57cec5SDimitry Andric }
36160b57cec5SDimitry Andric
GetAddrOfUnnamedGlobalConstantDecl(const UnnamedGlobalConstantDecl * GCD)361781ad6265SDimitry Andric ConstantAddress CodeGenModule::GetAddrOfUnnamedGlobalConstantDecl(
361881ad6265SDimitry Andric const UnnamedGlobalConstantDecl *GCD) {
361981ad6265SDimitry Andric CharUnits Alignment = getContext().getTypeAlignInChars(GCD->getType());
362081ad6265SDimitry Andric
362181ad6265SDimitry Andric llvm::GlobalVariable **Entry = nullptr;
362281ad6265SDimitry Andric Entry = &UnnamedGlobalConstantDeclMap[GCD];
362381ad6265SDimitry Andric if (*Entry)
362481ad6265SDimitry Andric return ConstantAddress(*Entry, (*Entry)->getValueType(), Alignment);
362581ad6265SDimitry Andric
362681ad6265SDimitry Andric ConstantEmitter Emitter(*this);
362781ad6265SDimitry Andric llvm::Constant *Init;
362881ad6265SDimitry Andric
362981ad6265SDimitry Andric const APValue &V = GCD->getValue();
363081ad6265SDimitry Andric
363181ad6265SDimitry Andric assert(!V.isAbsent());
363281ad6265SDimitry Andric Init = Emitter.emitForInitializer(V, GCD->getType().getAddressSpace(),
363381ad6265SDimitry Andric GCD->getType());
363481ad6265SDimitry Andric
363581ad6265SDimitry Andric auto *GV = new llvm::GlobalVariable(getModule(), Init->getType(),
363681ad6265SDimitry Andric /*isConstant=*/true,
363781ad6265SDimitry Andric llvm::GlobalValue::PrivateLinkage, Init,
363881ad6265SDimitry Andric ".constant");
363981ad6265SDimitry Andric GV->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
364081ad6265SDimitry Andric GV->setAlignment(Alignment.getAsAlign());
364181ad6265SDimitry Andric
364281ad6265SDimitry Andric Emitter.finalize(GV);
364381ad6265SDimitry Andric
364481ad6265SDimitry Andric *Entry = GV;
364581ad6265SDimitry Andric return ConstantAddress(GV, GV->getValueType(), Alignment);
364681ad6265SDimitry Andric }
364781ad6265SDimitry Andric
GetAddrOfTemplateParamObject(const TemplateParamObjectDecl * TPO)3648e8d8bef9SDimitry Andric ConstantAddress CodeGenModule::GetAddrOfTemplateParamObject(
3649e8d8bef9SDimitry Andric const TemplateParamObjectDecl *TPO) {
3650e8d8bef9SDimitry Andric StringRef Name = getMangledName(TPO);
3651e8d8bef9SDimitry Andric CharUnits Alignment = getNaturalTypeAlignment(TPO->getType());
3652e8d8bef9SDimitry Andric
3653e8d8bef9SDimitry Andric if (llvm::GlobalVariable *GV = getModule().getNamedGlobal(Name))
36540eae32dcSDimitry Andric return ConstantAddress(GV, GV->getValueType(), Alignment);
3655e8d8bef9SDimitry Andric
3656e8d8bef9SDimitry Andric ConstantEmitter Emitter(*this);
3657e8d8bef9SDimitry Andric llvm::Constant *Init = Emitter.emitForInitializer(
3658e8d8bef9SDimitry Andric TPO->getValue(), TPO->getType().getAddressSpace(), TPO->getType());
3659e8d8bef9SDimitry Andric
3660e8d8bef9SDimitry Andric if (!Init) {
3661e8d8bef9SDimitry Andric ErrorUnsupported(TPO, "template parameter object");
3662e8d8bef9SDimitry Andric return ConstantAddress::invalid();
3663e8d8bef9SDimitry Andric }
3664e8d8bef9SDimitry Andric
366506c3fb27SDimitry Andric llvm::GlobalValue::LinkageTypes Linkage =
366606c3fb27SDimitry Andric isExternallyVisible(TPO->getLinkageAndVisibility().getLinkage())
366706c3fb27SDimitry Andric ? llvm::GlobalValue::LinkOnceODRLinkage
366806c3fb27SDimitry Andric : llvm::GlobalValue::InternalLinkage;
366906c3fb27SDimitry Andric auto *GV = new llvm::GlobalVariable(getModule(), Init->getType(),
367006c3fb27SDimitry Andric /*isConstant=*/true, Linkage, Init, Name);
367106c3fb27SDimitry Andric setGVProperties(GV, TPO);
3672e8d8bef9SDimitry Andric if (supportsCOMDAT())
3673e8d8bef9SDimitry Andric GV->setComdat(TheModule.getOrInsertComdat(GV->getName()));
3674e8d8bef9SDimitry Andric Emitter.finalize(GV);
3675e8d8bef9SDimitry Andric
36760eae32dcSDimitry Andric return ConstantAddress(GV, GV->getValueType(), Alignment);
3677e8d8bef9SDimitry Andric }
3678e8d8bef9SDimitry Andric
GetWeakRefReference(const ValueDecl * VD)36790b57cec5SDimitry Andric ConstantAddress CodeGenModule::GetWeakRefReference(const ValueDecl *VD) {
36800b57cec5SDimitry Andric const AliasAttr *AA = VD->getAttr<AliasAttr>();
36810b57cec5SDimitry Andric assert(AA && "No alias?");
36820b57cec5SDimitry Andric
36830b57cec5SDimitry Andric CharUnits Alignment = getContext().getDeclAlign(VD);
36840b57cec5SDimitry Andric llvm::Type *DeclTy = getTypes().ConvertTypeForMem(VD->getType());
36850b57cec5SDimitry Andric
36860b57cec5SDimitry Andric // See if there is already something with the target's name in the module.
36870b57cec5SDimitry Andric llvm::GlobalValue *Entry = GetGlobalValue(AA->getAliasee());
36885f757f3fSDimitry Andric if (Entry)
36895f757f3fSDimitry Andric return ConstantAddress(Entry, DeclTy, Alignment);
36900b57cec5SDimitry Andric
36910b57cec5SDimitry Andric llvm::Constant *Aliasee;
36920b57cec5SDimitry Andric if (isa<llvm::FunctionType>(DeclTy))
36930b57cec5SDimitry Andric Aliasee = GetOrCreateLLVMFunction(AA->getAliasee(), DeclTy,
36940b57cec5SDimitry Andric GlobalDecl(cast<FunctionDecl>(VD)),
36950b57cec5SDimitry Andric /*ForVTable=*/false);
36960b57cec5SDimitry Andric else
3697349cc55cSDimitry Andric Aliasee = GetOrCreateLLVMGlobal(AA->getAliasee(), DeclTy, LangAS::Default,
3698349cc55cSDimitry Andric nullptr);
36990b57cec5SDimitry Andric
37000b57cec5SDimitry Andric auto *F = cast<llvm::GlobalValue>(Aliasee);
37010b57cec5SDimitry Andric F->setLinkage(llvm::Function::ExternalWeakLinkage);
37020b57cec5SDimitry Andric WeakRefReferences.insert(F);
37030b57cec5SDimitry Andric
37040eae32dcSDimitry Andric return ConstantAddress(Aliasee, DeclTy, Alignment);
37050b57cec5SDimitry Andric }
37060b57cec5SDimitry Andric
hasImplicitAttr(const ValueDecl * D)37075f757f3fSDimitry Andric template <typename AttrT> static bool hasImplicitAttr(const ValueDecl *D) {
37085f757f3fSDimitry Andric if (!D)
37095f757f3fSDimitry Andric return false;
37105f757f3fSDimitry Andric if (auto *A = D->getAttr<AttrT>())
37115f757f3fSDimitry Andric return A->isImplicit();
37125f757f3fSDimitry Andric return D->isImplicit();
37135f757f3fSDimitry Andric }
37145f757f3fSDimitry Andric
shouldEmitCUDAGlobalVar(const VarDecl * Global) const37150fca6ea1SDimitry Andric bool CodeGenModule::shouldEmitCUDAGlobalVar(const VarDecl *Global) const {
37160fca6ea1SDimitry Andric assert(LangOpts.CUDA && "Should not be called by non-CUDA languages");
37170fca6ea1SDimitry Andric // We need to emit host-side 'shadows' for all global
37180fca6ea1SDimitry Andric // device-side variables because the CUDA runtime needs their
37190fca6ea1SDimitry Andric // size and host-side address in order to provide access to
37200fca6ea1SDimitry Andric // their device-side incarnations.
37210fca6ea1SDimitry Andric return !LangOpts.CUDAIsDevice || Global->hasAttr<CUDADeviceAttr>() ||
37220fca6ea1SDimitry Andric Global->hasAttr<CUDAConstantAttr>() ||
37230fca6ea1SDimitry Andric Global->hasAttr<CUDASharedAttr>() ||
37240fca6ea1SDimitry Andric Global->getType()->isCUDADeviceBuiltinSurfaceType() ||
37250fca6ea1SDimitry Andric Global->getType()->isCUDADeviceBuiltinTextureType();
37260fca6ea1SDimitry Andric }
37270fca6ea1SDimitry Andric
EmitGlobal(GlobalDecl GD)37280b57cec5SDimitry Andric void CodeGenModule::EmitGlobal(GlobalDecl GD) {
37290b57cec5SDimitry Andric const auto *Global = cast<ValueDecl>(GD.getDecl());
37300b57cec5SDimitry Andric
37310b57cec5SDimitry Andric // Weak references don't produce any output by themselves.
37320b57cec5SDimitry Andric if (Global->hasAttr<WeakRefAttr>())
37330b57cec5SDimitry Andric return;
37340b57cec5SDimitry Andric
37350b57cec5SDimitry Andric // If this is an alias definition (which otherwise looks like a declaration)
37360b57cec5SDimitry Andric // emit it now.
37370b57cec5SDimitry Andric if (Global->hasAttr<AliasAttr>())
37380b57cec5SDimitry Andric return EmitAliasDefinition(GD);
37390b57cec5SDimitry Andric
37400b57cec5SDimitry Andric // IFunc like an alias whose value is resolved at runtime by calling resolver.
37410b57cec5SDimitry Andric if (Global->hasAttr<IFuncAttr>())
37420b57cec5SDimitry Andric return emitIFuncDefinition(GD);
37430b57cec5SDimitry Andric
37440b57cec5SDimitry Andric // If this is a cpu_dispatch multiversion function, emit the resolver.
37450b57cec5SDimitry Andric if (Global->hasAttr<CPUDispatchAttr>())
37460b57cec5SDimitry Andric return emitCPUDispatchDefinition(GD);
37470b57cec5SDimitry Andric
37480b57cec5SDimitry Andric // If this is CUDA, be selective about which declarations we emit.
37495f757f3fSDimitry Andric // Non-constexpr non-lambda implicit host device functions are not emitted
37505f757f3fSDimitry Andric // unless they are used on device side.
37510b57cec5SDimitry Andric if (LangOpts.CUDA) {
37520fca6ea1SDimitry Andric assert((isa<FunctionDecl>(Global) || isa<VarDecl>(Global)) &&
37530fca6ea1SDimitry Andric "Expected Variable or Function");
37540fca6ea1SDimitry Andric if (const auto *VD = dyn_cast<VarDecl>(Global)) {
37550fca6ea1SDimitry Andric if (!shouldEmitCUDAGlobalVar(VD))
37560fca6ea1SDimitry Andric return;
37570fca6ea1SDimitry Andric } else if (LangOpts.CUDAIsDevice) {
37585f757f3fSDimitry Andric const auto *FD = dyn_cast<FunctionDecl>(Global);
37595f757f3fSDimitry Andric if ((!Global->hasAttr<CUDADeviceAttr>() ||
37600fca6ea1SDimitry Andric (LangOpts.OffloadImplicitHostDeviceTemplates &&
37615f757f3fSDimitry Andric hasImplicitAttr<CUDAHostAttr>(FD) &&
37625f757f3fSDimitry Andric hasImplicitAttr<CUDADeviceAttr>(FD) && !FD->isConstexpr() &&
37635f757f3fSDimitry Andric !isLambdaCallOperator(FD) &&
37645f757f3fSDimitry Andric !getContext().CUDAImplicitHostDeviceFunUsedByDevice.count(FD))) &&
37650b57cec5SDimitry Andric !Global->hasAttr<CUDAGlobalAttr>() &&
37665f757f3fSDimitry Andric !(LangOpts.HIPStdPar && isa<FunctionDecl>(Global) &&
37675f757f3fSDimitry Andric !Global->hasAttr<CUDAHostAttr>()))
37680b57cec5SDimitry Andric return;
37690fca6ea1SDimitry Andric // Device-only functions are the only things we skip.
37700fca6ea1SDimitry Andric } else if (!Global->hasAttr<CUDAHostAttr>() &&
37710b57cec5SDimitry Andric Global->hasAttr<CUDADeviceAttr>())
37720b57cec5SDimitry Andric return;
37730b57cec5SDimitry Andric }
37740b57cec5SDimitry Andric
37750b57cec5SDimitry Andric if (LangOpts.OpenMP) {
3776a7dea167SDimitry Andric // If this is OpenMP, check if it is legal to emit this global normally.
37770b57cec5SDimitry Andric if (OpenMPRuntime && OpenMPRuntime->emitTargetGlobal(GD))
37780b57cec5SDimitry Andric return;
37790b57cec5SDimitry Andric if (auto *DRD = dyn_cast<OMPDeclareReductionDecl>(Global)) {
37800b57cec5SDimitry Andric if (MustBeEmitted(Global))
37810b57cec5SDimitry Andric EmitOMPDeclareReduction(DRD);
37820b57cec5SDimitry Andric return;
378306c3fb27SDimitry Andric }
378406c3fb27SDimitry Andric if (auto *DMD = dyn_cast<OMPDeclareMapperDecl>(Global)) {
37850b57cec5SDimitry Andric if (MustBeEmitted(Global))
37860b57cec5SDimitry Andric EmitOMPDeclareMapper(DMD);
37870b57cec5SDimitry Andric return;
37880b57cec5SDimitry Andric }
37890b57cec5SDimitry Andric }
37900b57cec5SDimitry Andric
37910b57cec5SDimitry Andric // Ignore declarations, they will be emitted on their first use.
37920b57cec5SDimitry Andric if (const auto *FD = dyn_cast<FunctionDecl>(Global)) {
37935f757f3fSDimitry Andric // Update deferred annotations with the latest declaration if the function
37945f757f3fSDimitry Andric // function was already used or defined.
37955f757f3fSDimitry Andric if (FD->hasAttr<AnnotateAttr>()) {
37965f757f3fSDimitry Andric StringRef MangledName = getMangledName(GD);
37975f757f3fSDimitry Andric if (GetGlobalValue(MangledName))
37985f757f3fSDimitry Andric DeferredAnnotations[MangledName] = FD;
37995f757f3fSDimitry Andric }
38005f757f3fSDimitry Andric
38010b57cec5SDimitry Andric // Forward declarations are emitted lazily on first use.
38020b57cec5SDimitry Andric if (!FD->doesThisDeclarationHaveABody()) {
38030fca6ea1SDimitry Andric if (!FD->doesDeclarationForceExternallyVisibleDefinition() &&
38040fca6ea1SDimitry Andric (!FD->isMultiVersion() || !getTarget().getTriple().isAArch64()))
38050b57cec5SDimitry Andric return;
38060b57cec5SDimitry Andric
38070b57cec5SDimitry Andric StringRef MangledName = getMangledName(GD);
38080b57cec5SDimitry Andric
38090b57cec5SDimitry Andric // Compute the function info and LLVM type.
38100b57cec5SDimitry Andric const CGFunctionInfo &FI = getTypes().arrangeGlobalDeclaration(GD);
38110b57cec5SDimitry Andric llvm::Type *Ty = getTypes().GetFunctionType(FI);
38120b57cec5SDimitry Andric
38130b57cec5SDimitry Andric GetOrCreateLLVMFunction(MangledName, Ty, GD, /*ForVTable=*/false,
38140b57cec5SDimitry Andric /*DontDefer=*/false);
38150b57cec5SDimitry Andric return;
38160b57cec5SDimitry Andric }
38170b57cec5SDimitry Andric } else {
38180b57cec5SDimitry Andric const auto *VD = cast<VarDecl>(Global);
38190b57cec5SDimitry Andric assert(VD->isFileVarDecl() && "Cannot emit local var decl as global.");
38200b57cec5SDimitry Andric if (VD->isThisDeclarationADefinition() != VarDecl::Definition &&
38210b57cec5SDimitry Andric !Context.isMSStaticDataMemberInlineDefinition(VD)) {
38220b57cec5SDimitry Andric if (LangOpts.OpenMP) {
38230b57cec5SDimitry Andric // Emit declaration of the must-be-emitted declare target variable.
3824bdd1243dSDimitry Andric if (std::optional<OMPDeclareTargetDeclAttr::MapTypeTy> Res =
38250b57cec5SDimitry Andric OMPDeclareTargetDeclAttr::isDeclareTargetDeclaration(VD)) {
38268a4dda33SDimitry Andric
38278a4dda33SDimitry Andric // If this variable has external storage and doesn't require special
38288a4dda33SDimitry Andric // link handling we defer to its canonical definition.
38298a4dda33SDimitry Andric if (VD->hasExternalStorage() &&
38308a4dda33SDimitry Andric Res != OMPDeclareTargetDeclAttr::MT_Link)
38318a4dda33SDimitry Andric return;
38328a4dda33SDimitry Andric
38330b57cec5SDimitry Andric bool UnifiedMemoryEnabled =
38340b57cec5SDimitry Andric getOpenMPRuntime().hasRequiresUnifiedSharedMemory();
3835bdd1243dSDimitry Andric if ((*Res == OMPDeclareTargetDeclAttr::MT_To ||
3836bdd1243dSDimitry Andric *Res == OMPDeclareTargetDeclAttr::MT_Enter) &&
38370b57cec5SDimitry Andric !UnifiedMemoryEnabled) {
38380b57cec5SDimitry Andric (void)GetAddrOfGlobalVar(VD);
38390b57cec5SDimitry Andric } else {
38400b57cec5SDimitry Andric assert(((*Res == OMPDeclareTargetDeclAttr::MT_Link) ||
3841bdd1243dSDimitry Andric ((*Res == OMPDeclareTargetDeclAttr::MT_To ||
3842bdd1243dSDimitry Andric *Res == OMPDeclareTargetDeclAttr::MT_Enter) &&
38430b57cec5SDimitry Andric UnifiedMemoryEnabled)) &&
38440b57cec5SDimitry Andric "Link clause or to clause with unified memory expected.");
38450b57cec5SDimitry Andric (void)getOpenMPRuntime().getAddrOfDeclareTargetVar(VD);
38460b57cec5SDimitry Andric }
38470b57cec5SDimitry Andric
38480b57cec5SDimitry Andric return;
38490b57cec5SDimitry Andric }
38500b57cec5SDimitry Andric }
38510b57cec5SDimitry Andric // If this declaration may have caused an inline variable definition to
38520b57cec5SDimitry Andric // change linkage, make sure that it's emitted.
38530b57cec5SDimitry Andric if (Context.getInlineVariableDefinitionKind(VD) ==
38540b57cec5SDimitry Andric ASTContext::InlineVariableDefinitionKind::Strong)
38550b57cec5SDimitry Andric GetAddrOfGlobalVar(VD);
38560b57cec5SDimitry Andric return;
38570b57cec5SDimitry Andric }
38580b57cec5SDimitry Andric }
38590b57cec5SDimitry Andric
38600b57cec5SDimitry Andric // Defer code generation to first use when possible, e.g. if this is an inline
38610b57cec5SDimitry Andric // function. If the global must always be emitted, do it eagerly if possible
38620b57cec5SDimitry Andric // to benefit from cache locality.
38630b57cec5SDimitry Andric if (MustBeEmitted(Global) && MayBeEmittedEagerly(Global)) {
38640b57cec5SDimitry Andric // Emit the definition if it can't be deferred.
38650b57cec5SDimitry Andric EmitGlobalDefinition(GD);
38668a4dda33SDimitry Andric addEmittedDeferredDecl(GD);
38670b57cec5SDimitry Andric return;
38680b57cec5SDimitry Andric }
38690b57cec5SDimitry Andric
38700b57cec5SDimitry Andric // If we're deferring emission of a C++ variable with an
38710b57cec5SDimitry Andric // initializer, remember the order in which it appeared in the file.
38720b57cec5SDimitry Andric if (getLangOpts().CPlusPlus && isa<VarDecl>(Global) &&
38730b57cec5SDimitry Andric cast<VarDecl>(Global)->hasInit()) {
38740b57cec5SDimitry Andric DelayedCXXInitPosition[Global] = CXXGlobalInits.size();
38750b57cec5SDimitry Andric CXXGlobalInits.push_back(nullptr);
38760b57cec5SDimitry Andric }
38770b57cec5SDimitry Andric
38780b57cec5SDimitry Andric StringRef MangledName = getMangledName(GD);
38790b57cec5SDimitry Andric if (GetGlobalValue(MangledName) != nullptr) {
38800b57cec5SDimitry Andric // The value has already been used and should therefore be emitted.
38810b57cec5SDimitry Andric addDeferredDeclToEmit(GD);
38820b57cec5SDimitry Andric } else if (MustBeEmitted(Global)) {
38830b57cec5SDimitry Andric // The value must be emitted, but cannot be emitted eagerly.
38840b57cec5SDimitry Andric assert(!MayBeEmittedEagerly(Global));
38850b57cec5SDimitry Andric addDeferredDeclToEmit(GD);
38860b57cec5SDimitry Andric } else {
38870b57cec5SDimitry Andric // Otherwise, remember that we saw a deferred decl with this name. The
38880b57cec5SDimitry Andric // first use of the mangled name will cause it to move into
38890b57cec5SDimitry Andric // DeferredDeclsToEmit.
38900b57cec5SDimitry Andric DeferredDecls[MangledName] = GD;
38910b57cec5SDimitry Andric }
38920b57cec5SDimitry Andric }
38930b57cec5SDimitry Andric
38940b57cec5SDimitry Andric // Check if T is a class type with a destructor that's not dllimport.
HasNonDllImportDtor(QualType T)38950b57cec5SDimitry Andric static bool HasNonDllImportDtor(QualType T) {
38960b57cec5SDimitry Andric if (const auto *RT = T->getBaseElementTypeUnsafe()->getAs<RecordType>())
38970b57cec5SDimitry Andric if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(RT->getDecl()))
38980b57cec5SDimitry Andric if (RD->getDestructor() && !RD->getDestructor()->hasAttr<DLLImportAttr>())
38990b57cec5SDimitry Andric return true;
39000b57cec5SDimitry Andric
39010b57cec5SDimitry Andric return false;
39020b57cec5SDimitry Andric }
39030b57cec5SDimitry Andric
39040b57cec5SDimitry Andric namespace {
39050b57cec5SDimitry Andric struct FunctionIsDirectlyRecursive
39060b57cec5SDimitry Andric : public ConstStmtVisitor<FunctionIsDirectlyRecursive, bool> {
39070b57cec5SDimitry Andric const StringRef Name;
39080b57cec5SDimitry Andric const Builtin::Context &BI;
FunctionIsDirectlyRecursive__anona4b266790a11::FunctionIsDirectlyRecursive39090b57cec5SDimitry Andric FunctionIsDirectlyRecursive(StringRef N, const Builtin::Context &C)
39100b57cec5SDimitry Andric : Name(N), BI(C) {}
39110b57cec5SDimitry Andric
VisitCallExpr__anona4b266790a11::FunctionIsDirectlyRecursive39120b57cec5SDimitry Andric bool VisitCallExpr(const CallExpr *E) {
39130b57cec5SDimitry Andric const FunctionDecl *FD = E->getDirectCallee();
39140b57cec5SDimitry Andric if (!FD)
39150b57cec5SDimitry Andric return false;
39160b57cec5SDimitry Andric AsmLabelAttr *Attr = FD->getAttr<AsmLabelAttr>();
39170b57cec5SDimitry Andric if (Attr && Name == Attr->getLabel())
39180b57cec5SDimitry Andric return true;
39190b57cec5SDimitry Andric unsigned BuiltinID = FD->getBuiltinID();
39200b57cec5SDimitry Andric if (!BuiltinID || !BI.isLibFunction(BuiltinID))
39210b57cec5SDimitry Andric return false;
39220b57cec5SDimitry Andric StringRef BuiltinName = BI.getName(BuiltinID);
39235f757f3fSDimitry Andric if (BuiltinName.starts_with("__builtin_") &&
39240b57cec5SDimitry Andric Name == BuiltinName.slice(strlen("__builtin_"), StringRef::npos)) {
39250b57cec5SDimitry Andric return true;
39260b57cec5SDimitry Andric }
39270b57cec5SDimitry Andric return false;
39280b57cec5SDimitry Andric }
39290b57cec5SDimitry Andric
VisitStmt__anona4b266790a11::FunctionIsDirectlyRecursive39300b57cec5SDimitry Andric bool VisitStmt(const Stmt *S) {
39310b57cec5SDimitry Andric for (const Stmt *Child : S->children())
39320b57cec5SDimitry Andric if (Child && this->Visit(Child))
39330b57cec5SDimitry Andric return true;
39340b57cec5SDimitry Andric return false;
39350b57cec5SDimitry Andric }
39360b57cec5SDimitry Andric };
39370b57cec5SDimitry Andric
39380b57cec5SDimitry Andric // Make sure we're not referencing non-imported vars or functions.
39390b57cec5SDimitry Andric struct DLLImportFunctionVisitor
39400b57cec5SDimitry Andric : public RecursiveASTVisitor<DLLImportFunctionVisitor> {
39410b57cec5SDimitry Andric bool SafeToInline = true;
39420b57cec5SDimitry Andric
shouldVisitImplicitCode__anona4b266790a11::DLLImportFunctionVisitor39430b57cec5SDimitry Andric bool shouldVisitImplicitCode() const { return true; }
39440b57cec5SDimitry Andric
VisitVarDecl__anona4b266790a11::DLLImportFunctionVisitor39450b57cec5SDimitry Andric bool VisitVarDecl(VarDecl *VD) {
39460b57cec5SDimitry Andric if (VD->getTLSKind()) {
39470b57cec5SDimitry Andric // A thread-local variable cannot be imported.
39480b57cec5SDimitry Andric SafeToInline = false;
39490b57cec5SDimitry Andric return SafeToInline;
39500b57cec5SDimitry Andric }
39510b57cec5SDimitry Andric
39520b57cec5SDimitry Andric // A variable definition might imply a destructor call.
39530b57cec5SDimitry Andric if (VD->isThisDeclarationADefinition())
39540b57cec5SDimitry Andric SafeToInline = !HasNonDllImportDtor(VD->getType());
39550b57cec5SDimitry Andric
39560b57cec5SDimitry Andric return SafeToInline;
39570b57cec5SDimitry Andric }
39580b57cec5SDimitry Andric
VisitCXXBindTemporaryExpr__anona4b266790a11::DLLImportFunctionVisitor39590b57cec5SDimitry Andric bool VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
39600b57cec5SDimitry Andric if (const auto *D = E->getTemporary()->getDestructor())
39610b57cec5SDimitry Andric SafeToInline = D->hasAttr<DLLImportAttr>();
39620b57cec5SDimitry Andric return SafeToInline;
39630b57cec5SDimitry Andric }
39640b57cec5SDimitry Andric
VisitDeclRefExpr__anona4b266790a11::DLLImportFunctionVisitor39650b57cec5SDimitry Andric bool VisitDeclRefExpr(DeclRefExpr *E) {
39660b57cec5SDimitry Andric ValueDecl *VD = E->getDecl();
39670b57cec5SDimitry Andric if (isa<FunctionDecl>(VD))
39680b57cec5SDimitry Andric SafeToInline = VD->hasAttr<DLLImportAttr>();
39690b57cec5SDimitry Andric else if (VarDecl *V = dyn_cast<VarDecl>(VD))
39700b57cec5SDimitry Andric SafeToInline = !V->hasGlobalStorage() || V->hasAttr<DLLImportAttr>();
39710b57cec5SDimitry Andric return SafeToInline;
39720b57cec5SDimitry Andric }
39730b57cec5SDimitry Andric
VisitCXXConstructExpr__anona4b266790a11::DLLImportFunctionVisitor39740b57cec5SDimitry Andric bool VisitCXXConstructExpr(CXXConstructExpr *E) {
39750b57cec5SDimitry Andric SafeToInline = E->getConstructor()->hasAttr<DLLImportAttr>();
39760b57cec5SDimitry Andric return SafeToInline;
39770b57cec5SDimitry Andric }
39780b57cec5SDimitry Andric
VisitCXXMemberCallExpr__anona4b266790a11::DLLImportFunctionVisitor39790b57cec5SDimitry Andric bool VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
39800b57cec5SDimitry Andric CXXMethodDecl *M = E->getMethodDecl();
39810b57cec5SDimitry Andric if (!M) {
39820b57cec5SDimitry Andric // Call through a pointer to member function. This is safe to inline.
39830b57cec5SDimitry Andric SafeToInline = true;
39840b57cec5SDimitry Andric } else {
39850b57cec5SDimitry Andric SafeToInline = M->hasAttr<DLLImportAttr>();
39860b57cec5SDimitry Andric }
39870b57cec5SDimitry Andric return SafeToInline;
39880b57cec5SDimitry Andric }
39890b57cec5SDimitry Andric
VisitCXXDeleteExpr__anona4b266790a11::DLLImportFunctionVisitor39900b57cec5SDimitry Andric bool VisitCXXDeleteExpr(CXXDeleteExpr *E) {
39910b57cec5SDimitry Andric SafeToInline = E->getOperatorDelete()->hasAttr<DLLImportAttr>();
39920b57cec5SDimitry Andric return SafeToInline;
39930b57cec5SDimitry Andric }
39940b57cec5SDimitry Andric
VisitCXXNewExpr__anona4b266790a11::DLLImportFunctionVisitor39950b57cec5SDimitry Andric bool VisitCXXNewExpr(CXXNewExpr *E) {
39960b57cec5SDimitry Andric SafeToInline = E->getOperatorNew()->hasAttr<DLLImportAttr>();
39970b57cec5SDimitry Andric return SafeToInline;
39980b57cec5SDimitry Andric }
39990b57cec5SDimitry Andric };
40000b57cec5SDimitry Andric }
40010b57cec5SDimitry Andric
40020b57cec5SDimitry Andric // isTriviallyRecursive - Check if this function calls another
40030b57cec5SDimitry Andric // decl that, because of the asm attribute or the other decl being a builtin,
40040b57cec5SDimitry Andric // ends up pointing to itself.
40050b57cec5SDimitry Andric bool
isTriviallyRecursive(const FunctionDecl * FD)40060b57cec5SDimitry Andric CodeGenModule::isTriviallyRecursive(const FunctionDecl *FD) {
40070b57cec5SDimitry Andric StringRef Name;
40080b57cec5SDimitry Andric if (getCXXABI().getMangleContext().shouldMangleDeclName(FD)) {
40090b57cec5SDimitry Andric // asm labels are a special kind of mangling we have to support.
40100b57cec5SDimitry Andric AsmLabelAttr *Attr = FD->getAttr<AsmLabelAttr>();
40110b57cec5SDimitry Andric if (!Attr)
40120b57cec5SDimitry Andric return false;
40130b57cec5SDimitry Andric Name = Attr->getLabel();
40140b57cec5SDimitry Andric } else {
40150b57cec5SDimitry Andric Name = FD->getName();
40160b57cec5SDimitry Andric }
40170b57cec5SDimitry Andric
40180b57cec5SDimitry Andric FunctionIsDirectlyRecursive Walker(Name, Context.BuiltinInfo);
40190b57cec5SDimitry Andric const Stmt *Body = FD->getBody();
40200b57cec5SDimitry Andric return Body ? Walker.Visit(Body) : false;
40210b57cec5SDimitry Andric }
40220b57cec5SDimitry Andric
shouldEmitFunction(GlobalDecl GD)40230b57cec5SDimitry Andric bool CodeGenModule::shouldEmitFunction(GlobalDecl GD) {
40240b57cec5SDimitry Andric if (getFunctionLinkage(GD) != llvm::Function::AvailableExternallyLinkage)
40250b57cec5SDimitry Andric return true;
40265f757f3fSDimitry Andric
40270b57cec5SDimitry Andric const auto *F = cast<FunctionDecl>(GD.getDecl());
40280b57cec5SDimitry Andric if (CodeGenOpts.OptimizationLevel == 0 && !F->hasAttr<AlwaysInlineAttr>())
40290b57cec5SDimitry Andric return false;
40300b57cec5SDimitry Andric
40315f757f3fSDimitry Andric // We don't import function bodies from other named module units since that
40325f757f3fSDimitry Andric // behavior may break ABI compatibility of the current unit.
40335f757f3fSDimitry Andric if (const Module *M = F->getOwningModule();
40345f757f3fSDimitry Andric M && M->getTopLevelModule()->isNamedModule() &&
40350fca6ea1SDimitry Andric getContext().getCurrentNamedModule() != M->getTopLevelModule()) {
40360fca6ea1SDimitry Andric // There are practices to mark template member function as always-inline
40370fca6ea1SDimitry Andric // and mark the template as extern explicit instantiation but not give
40380fca6ea1SDimitry Andric // the definition for member function. So we have to emit the function
40390fca6ea1SDimitry Andric // from explicitly instantiation with always-inline.
40400fca6ea1SDimitry Andric //
40410fca6ea1SDimitry Andric // See https://github.com/llvm/llvm-project/issues/86893 for details.
40420fca6ea1SDimitry Andric //
40430fca6ea1SDimitry Andric // TODO: Maybe it is better to give it a warning if we call a non-inline
40440fca6ea1SDimitry Andric // function from other module units which is marked as always-inline.
40450fca6ea1SDimitry Andric if (!F->isTemplateInstantiation() || !F->hasAttr<AlwaysInlineAttr>()) {
40465f757f3fSDimitry Andric return false;
40470fca6ea1SDimitry Andric }
40480fca6ea1SDimitry Andric }
40495f757f3fSDimitry Andric
40505f757f3fSDimitry Andric if (F->hasAttr<NoInlineAttr>())
40515f757f3fSDimitry Andric return false;
40525f757f3fSDimitry Andric
4053fe6060f1SDimitry Andric if (F->hasAttr<DLLImportAttr>() && !F->hasAttr<AlwaysInlineAttr>()) {
40540b57cec5SDimitry Andric // Check whether it would be safe to inline this dllimport function.
40550b57cec5SDimitry Andric DLLImportFunctionVisitor Visitor;
40560b57cec5SDimitry Andric Visitor.TraverseFunctionDecl(const_cast<FunctionDecl*>(F));
40570b57cec5SDimitry Andric if (!Visitor.SafeToInline)
40580b57cec5SDimitry Andric return false;
40590b57cec5SDimitry Andric
40600b57cec5SDimitry Andric if (const CXXDestructorDecl *Dtor = dyn_cast<CXXDestructorDecl>(F)) {
40610b57cec5SDimitry Andric // Implicit destructor invocations aren't captured in the AST, so the
40620b57cec5SDimitry Andric // check above can't see them. Check for them manually here.
40630b57cec5SDimitry Andric for (const Decl *Member : Dtor->getParent()->decls())
40640b57cec5SDimitry Andric if (isa<FieldDecl>(Member))
40650b57cec5SDimitry Andric if (HasNonDllImportDtor(cast<FieldDecl>(Member)->getType()))
40660b57cec5SDimitry Andric return false;
40670b57cec5SDimitry Andric for (const CXXBaseSpecifier &B : Dtor->getParent()->bases())
40680b57cec5SDimitry Andric if (HasNonDllImportDtor(B.getType()))
40690b57cec5SDimitry Andric return false;
40700b57cec5SDimitry Andric }
40710b57cec5SDimitry Andric }
40720b57cec5SDimitry Andric
4073349cc55cSDimitry Andric // Inline builtins declaration must be emitted. They often are fortified
4074349cc55cSDimitry Andric // functions.
4075349cc55cSDimitry Andric if (F->isInlineBuiltinDeclaration())
4076349cc55cSDimitry Andric return true;
4077349cc55cSDimitry Andric
40780b57cec5SDimitry Andric // PR9614. Avoid cases where the source code is lying to us. An available
40790b57cec5SDimitry Andric // externally function should have an equivalent function somewhere else,
40805ffd83dbSDimitry Andric // but a function that calls itself through asm label/`__builtin_` trickery is
40815ffd83dbSDimitry Andric // clearly not equivalent to the real implementation.
40820b57cec5SDimitry Andric // This happens in glibc's btowc and in some configure checks.
40830b57cec5SDimitry Andric return !isTriviallyRecursive(F);
40840b57cec5SDimitry Andric }
40850b57cec5SDimitry Andric
shouldOpportunisticallyEmitVTables()40860b57cec5SDimitry Andric bool CodeGenModule::shouldOpportunisticallyEmitVTables() {
40870b57cec5SDimitry Andric return CodeGenOpts.OptimizationLevel > 0;
40880b57cec5SDimitry Andric }
40890b57cec5SDimitry Andric
EmitMultiVersionFunctionDefinition(GlobalDecl GD,llvm::GlobalValue * GV)40900b57cec5SDimitry Andric void CodeGenModule::EmitMultiVersionFunctionDefinition(GlobalDecl GD,
40910b57cec5SDimitry Andric llvm::GlobalValue *GV) {
40920b57cec5SDimitry Andric const auto *FD = cast<FunctionDecl>(GD.getDecl());
40930b57cec5SDimitry Andric
40940b57cec5SDimitry Andric if (FD->isCPUSpecificMultiVersion()) {
40950b57cec5SDimitry Andric auto *Spec = FD->getAttr<CPUSpecificAttr>();
40960b57cec5SDimitry Andric for (unsigned I = 0; I < Spec->cpus_size(); ++I)
40970b57cec5SDimitry Andric EmitGlobalFunctionDefinition(GD.getWithMultiVersionIndex(I), nullptr);
40980fca6ea1SDimitry Andric } else if (auto *TC = FD->getAttr<TargetClonesAttr>()) {
40990fca6ea1SDimitry Andric for (unsigned I = 0; I < TC->featuresStrs_size(); ++I)
41000fca6ea1SDimitry Andric // AArch64 favors the default target version over the clone if any.
41010fca6ea1SDimitry Andric if ((!TC->isDefaultVersion(I) || !getTarget().getTriple().isAArch64()) &&
41020fca6ea1SDimitry Andric TC->isFirstOfVersion(I))
41034824e7fdSDimitry Andric EmitGlobalFunctionDefinition(GD.getWithMultiVersionIndex(I), nullptr);
410481ad6265SDimitry Andric // Ensure that the resolver function is also emitted.
410581ad6265SDimitry Andric GetOrCreateMultiVersionResolver(GD);
41060b57cec5SDimitry Andric } else
41070b57cec5SDimitry Andric EmitGlobalFunctionDefinition(GD, GV);
41080fca6ea1SDimitry Andric
41090fca6ea1SDimitry Andric // Defer the resolver emission until we can reason whether the TU
41100fca6ea1SDimitry Andric // contains a default target version implementation.
41110fca6ea1SDimitry Andric if (FD->isTargetVersionMultiVersion())
41120fca6ea1SDimitry Andric AddDeferredMultiVersionResolverToEmit(GD);
41130b57cec5SDimitry Andric }
41140b57cec5SDimitry Andric
EmitGlobalDefinition(GlobalDecl GD,llvm::GlobalValue * GV)41150b57cec5SDimitry Andric void CodeGenModule::EmitGlobalDefinition(GlobalDecl GD, llvm::GlobalValue *GV) {
41160b57cec5SDimitry Andric const auto *D = cast<ValueDecl>(GD.getDecl());
41170b57cec5SDimitry Andric
41180b57cec5SDimitry Andric PrettyStackTraceDecl CrashInfo(const_cast<ValueDecl *>(D), D->getLocation(),
41190b57cec5SDimitry Andric Context.getSourceManager(),
41200b57cec5SDimitry Andric "Generating code for declaration");
41210b57cec5SDimitry Andric
41220b57cec5SDimitry Andric if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
41230b57cec5SDimitry Andric // At -O0, don't generate IR for functions with available_externally
41240b57cec5SDimitry Andric // linkage.
41250b57cec5SDimitry Andric if (!shouldEmitFunction(GD))
41260b57cec5SDimitry Andric return;
41270b57cec5SDimitry Andric
41280b57cec5SDimitry Andric llvm::TimeTraceScope TimeScope("CodeGen Function", [&]() {
41290b57cec5SDimitry Andric std::string Name;
41300b57cec5SDimitry Andric llvm::raw_string_ostream OS(Name);
41310b57cec5SDimitry Andric FD->getNameForDiagnostic(OS, getContext().getPrintingPolicy(),
41320b57cec5SDimitry Andric /*Qualified=*/true);
41330b57cec5SDimitry Andric return Name;
41340b57cec5SDimitry Andric });
41350b57cec5SDimitry Andric
41360b57cec5SDimitry Andric if (const auto *Method = dyn_cast<CXXMethodDecl>(D)) {
41370b57cec5SDimitry Andric // Make sure to emit the definition(s) before we emit the thunks.
41380b57cec5SDimitry Andric // This is necessary for the generation of certain thunks.
41390b57cec5SDimitry Andric if (isa<CXXConstructorDecl>(Method) || isa<CXXDestructorDecl>(Method))
41400b57cec5SDimitry Andric ABI->emitCXXStructor(GD);
41410b57cec5SDimitry Andric else if (FD->isMultiVersion())
41420b57cec5SDimitry Andric EmitMultiVersionFunctionDefinition(GD, GV);
41430b57cec5SDimitry Andric else
41440b57cec5SDimitry Andric EmitGlobalFunctionDefinition(GD, GV);
41450b57cec5SDimitry Andric
41460b57cec5SDimitry Andric if (Method->isVirtual())
41470b57cec5SDimitry Andric getVTables().EmitThunks(GD);
41480b57cec5SDimitry Andric
41490b57cec5SDimitry Andric return;
41500b57cec5SDimitry Andric }
41510b57cec5SDimitry Andric
41520b57cec5SDimitry Andric if (FD->isMultiVersion())
41530b57cec5SDimitry Andric return EmitMultiVersionFunctionDefinition(GD, GV);
41540b57cec5SDimitry Andric return EmitGlobalFunctionDefinition(GD, GV);
41550b57cec5SDimitry Andric }
41560b57cec5SDimitry Andric
41570b57cec5SDimitry Andric if (const auto *VD = dyn_cast<VarDecl>(D))
41580b57cec5SDimitry Andric return EmitGlobalVarDefinition(VD, !VD->hasDefinition());
41590b57cec5SDimitry Andric
41600b57cec5SDimitry Andric llvm_unreachable("Invalid argument to EmitGlobalDefinition()");
41610b57cec5SDimitry Andric }
41620b57cec5SDimitry Andric
41630b57cec5SDimitry Andric static void ReplaceUsesOfNonProtoTypeWithRealFunction(llvm::GlobalValue *Old,
41640b57cec5SDimitry Andric llvm::Function *NewFn);
41650b57cec5SDimitry Andric
41660b57cec5SDimitry Andric static unsigned
TargetMVPriority(const TargetInfo & TI,const CodeGenFunction::MultiVersionResolverOption & RO)41670b57cec5SDimitry Andric TargetMVPriority(const TargetInfo &TI,
41680b57cec5SDimitry Andric const CodeGenFunction::MultiVersionResolverOption &RO) {
41690b57cec5SDimitry Andric unsigned Priority = 0;
4170bdd1243dSDimitry Andric unsigned NumFeatures = 0;
4171bdd1243dSDimitry Andric for (StringRef Feat : RO.Conditions.Features) {
41720b57cec5SDimitry Andric Priority = std::max(Priority, TI.multiVersionSortPriority(Feat));
4173bdd1243dSDimitry Andric NumFeatures++;
4174bdd1243dSDimitry Andric }
41750b57cec5SDimitry Andric
41760b57cec5SDimitry Andric if (!RO.Conditions.Architecture.empty())
41770b57cec5SDimitry Andric Priority = std::max(
41780b57cec5SDimitry Andric Priority, TI.multiVersionSortPriority(RO.Conditions.Architecture));
4179bdd1243dSDimitry Andric
4180bdd1243dSDimitry Andric Priority += TI.multiVersionFeatureCost() * NumFeatures;
4181bdd1243dSDimitry Andric
41820b57cec5SDimitry Andric return Priority;
41830b57cec5SDimitry Andric }
41840b57cec5SDimitry Andric
4185349cc55cSDimitry Andric // Multiversion functions should be at most 'WeakODRLinkage' so that a different
4186349cc55cSDimitry Andric // TU can forward declare the function without causing problems. Particularly
4187349cc55cSDimitry Andric // in the cases of CPUDispatch, this causes issues. This also makes sure we
4188349cc55cSDimitry Andric // work with internal linkage functions, so that the same function name can be
4189349cc55cSDimitry Andric // used with internal linkage in multiple TUs.
getMultiversionLinkage(CodeGenModule & CGM,GlobalDecl GD)4190349cc55cSDimitry Andric llvm::GlobalValue::LinkageTypes getMultiversionLinkage(CodeGenModule &CGM,
4191349cc55cSDimitry Andric GlobalDecl GD) {
4192349cc55cSDimitry Andric const FunctionDecl *FD = cast<FunctionDecl>(GD.getDecl());
41935f757f3fSDimitry Andric if (FD->getFormalLinkage() == Linkage::Internal)
4194349cc55cSDimitry Andric return llvm::GlobalValue::InternalLinkage;
4195349cc55cSDimitry Andric return llvm::GlobalValue::WeakODRLinkage;
4196349cc55cSDimitry Andric }
4197349cc55cSDimitry Andric
emitMultiVersionFunctions()41980b57cec5SDimitry Andric void CodeGenModule::emitMultiVersionFunctions() {
4199fe6060f1SDimitry Andric std::vector<GlobalDecl> MVFuncsToEmit;
4200fe6060f1SDimitry Andric MultiVersionFuncs.swap(MVFuncsToEmit);
4201fe6060f1SDimitry Andric for (GlobalDecl GD : MVFuncsToEmit) {
420281ad6265SDimitry Andric const auto *FD = cast<FunctionDecl>(GD.getDecl());
420381ad6265SDimitry Andric assert(FD && "Expected a FunctionDecl");
420481ad6265SDimitry Andric
42050fca6ea1SDimitry Andric auto createFunction = [&](const FunctionDecl *Decl, unsigned MVIdx = 0) {
42060fca6ea1SDimitry Andric GlobalDecl CurGD{Decl->isDefined() ? Decl->getDefinition() : Decl, MVIdx};
42070fca6ea1SDimitry Andric StringRef MangledName = getMangledName(CurGD);
42080fca6ea1SDimitry Andric llvm::Constant *Func = GetGlobalValue(MangledName);
42090fca6ea1SDimitry Andric if (!Func) {
42100fca6ea1SDimitry Andric if (Decl->isDefined()) {
42110fca6ea1SDimitry Andric EmitGlobalFunctionDefinition(CurGD, nullptr);
42120fca6ea1SDimitry Andric Func = GetGlobalValue(MangledName);
42130fca6ea1SDimitry Andric } else {
42140fca6ea1SDimitry Andric const CGFunctionInfo &FI = getTypes().arrangeGlobalDeclaration(CurGD);
42150fca6ea1SDimitry Andric llvm::FunctionType *Ty = getTypes().GetFunctionType(FI);
42160fca6ea1SDimitry Andric Func = GetAddrOfFunction(CurGD, Ty, /*ForVTable=*/false,
42170fca6ea1SDimitry Andric /*DontDefer=*/false, ForDefinition);
42180fca6ea1SDimitry Andric }
42190fca6ea1SDimitry Andric assert(Func && "This should have just been created");
42200fca6ea1SDimitry Andric }
42210fca6ea1SDimitry Andric return cast<llvm::Function>(Func);
42220fca6ea1SDimitry Andric };
42230fca6ea1SDimitry Andric
42240fca6ea1SDimitry Andric // For AArch64, a resolver is only emitted if a function marked with
42250fca6ea1SDimitry Andric // target_version("default")) or target_clones() is present and defined
42260fca6ea1SDimitry Andric // in this TU. For other architectures it is always emitted.
42270fca6ea1SDimitry Andric bool ShouldEmitResolver = !getTarget().getTriple().isAArch64();
42280b57cec5SDimitry Andric SmallVector<CodeGenFunction::MultiVersionResolverOption, 10> Options;
42290fca6ea1SDimitry Andric
42300b57cec5SDimitry Andric getContext().forEachMultiversionedFunctionVersion(
42310fca6ea1SDimitry Andric FD, [&](const FunctionDecl *CurFD) {
42320b57cec5SDimitry Andric llvm::SmallVector<StringRef, 8> Feats;
42330fca6ea1SDimitry Andric bool IsDefined = CurFD->doesThisDeclarationHaveABody();
42340fca6ea1SDimitry Andric
42350fca6ea1SDimitry Andric if (const auto *TA = CurFD->getAttr<TargetAttr>()) {
42360b57cec5SDimitry Andric TA->getAddedFeatures(Feats);
42370fca6ea1SDimitry Andric llvm::Function *Func = createFunction(CurFD);
42380fca6ea1SDimitry Andric Options.emplace_back(Func, TA->getArchitecture(), Feats);
42390fca6ea1SDimitry Andric } else if (const auto *TVA = CurFD->getAttr<TargetVersionAttr>()) {
42400fca6ea1SDimitry Andric if (TVA->isDefaultVersion() && IsDefined)
42410fca6ea1SDimitry Andric ShouldEmitResolver = true;
4242bdd1243dSDimitry Andric TVA->getFeatures(Feats);
42430fca6ea1SDimitry Andric llvm::Function *Func = createFunction(CurFD);
42440fca6ea1SDimitry Andric Options.emplace_back(Func, /*Architecture*/ "", Feats);
42450fca6ea1SDimitry Andric } else if (const auto *TC = CurFD->getAttr<TargetClonesAttr>()) {
42460fca6ea1SDimitry Andric if (IsDefined)
42470fca6ea1SDimitry Andric ShouldEmitResolver = true;
42480fca6ea1SDimitry Andric for (unsigned I = 0; I < TC->featuresStrs_size(); ++I) {
42490fca6ea1SDimitry Andric if (!TC->isFirstOfVersion(I))
425081ad6265SDimitry Andric continue;
425181ad6265SDimitry Andric
42520fca6ea1SDimitry Andric llvm::Function *Func = createFunction(CurFD, I);
425381ad6265SDimitry Andric StringRef Architecture;
42540fca6ea1SDimitry Andric Feats.clear();
42550fca6ea1SDimitry Andric if (getTarget().getTriple().isAArch64())
42560fca6ea1SDimitry Andric TC->getFeatures(Feats, I);
42570fca6ea1SDimitry Andric else {
42580fca6ea1SDimitry Andric StringRef Version = TC->getFeatureStr(I);
42595f757f3fSDimitry Andric if (Version.starts_with("arch="))
426081ad6265SDimitry Andric Architecture = Version.drop_front(sizeof("arch=") - 1);
426181ad6265SDimitry Andric else if (Version != "default")
42620fca6ea1SDimitry Andric Feats.push_back(Version);
4263bdd1243dSDimitry Andric }
42640fca6ea1SDimitry Andric Options.emplace_back(Func, Architecture, Feats);
42650fca6ea1SDimitry Andric }
42660fca6ea1SDimitry Andric } else
42670fca6ea1SDimitry Andric llvm_unreachable("unexpected MultiVersionKind");
42680fca6ea1SDimitry Andric });
426981ad6265SDimitry Andric
42700fca6ea1SDimitry Andric if (!ShouldEmitResolver)
427181ad6265SDimitry Andric continue;
427281ad6265SDimitry Andric
427381ad6265SDimitry Andric llvm::Constant *ResolverConstant = GetOrCreateMultiVersionResolver(GD);
42745f757f3fSDimitry Andric if (auto *IFunc = dyn_cast<llvm::GlobalIFunc>(ResolverConstant)) {
427581ad6265SDimitry Andric ResolverConstant = IFunc->getResolver();
42760fca6ea1SDimitry Andric if (FD->isTargetClonesMultiVersion() &&
42770fca6ea1SDimitry Andric !getTarget().getTriple().isAArch64()) {
42785f757f3fSDimitry Andric std::string MangledName = getMangledNameImpl(
42795f757f3fSDimitry Andric *this, GD, FD, /*OmitMultiVersionMangling=*/true);
42800fca6ea1SDimitry Andric if (!GetGlobalValue(MangledName + ".ifunc")) {
42810fca6ea1SDimitry Andric const CGFunctionInfo &FI = getTypes().arrangeGlobalDeclaration(GD);
42820fca6ea1SDimitry Andric llvm::FunctionType *DeclTy = getTypes().GetFunctionType(FI);
42835f757f3fSDimitry Andric // In prior versions of Clang, the mangling for ifuncs incorrectly
42845f757f3fSDimitry Andric // included an .ifunc suffix. This alias is generated for backward
42855f757f3fSDimitry Andric // compatibility. It is deprecated, and may be removed in the future.
42865f757f3fSDimitry Andric auto *Alias = llvm::GlobalAlias::create(
42875f757f3fSDimitry Andric DeclTy, 0, getMultiversionLinkage(*this, GD),
42885f757f3fSDimitry Andric MangledName + ".ifunc", IFunc, &getModule());
42895f757f3fSDimitry Andric SetCommonAttributes(FD, Alias);
42905f757f3fSDimitry Andric }
42915f757f3fSDimitry Andric }
42920fca6ea1SDimitry Andric }
429381ad6265SDimitry Andric llvm::Function *ResolverFunc = cast<llvm::Function>(ResolverConstant);
429481ad6265SDimitry Andric
429581ad6265SDimitry Andric ResolverFunc->setLinkage(getMultiversionLinkage(*this, GD));
42960b57cec5SDimitry Andric
42975f757f3fSDimitry Andric if (!ResolverFunc->hasLocalLinkage() && supportsCOMDAT())
42980b57cec5SDimitry Andric ResolverFunc->setComdat(
42990b57cec5SDimitry Andric getModule().getOrInsertComdat(ResolverFunc->getName()));
43000b57cec5SDimitry Andric
430181ad6265SDimitry Andric const TargetInfo &TI = getTarget();
43020b57cec5SDimitry Andric llvm::stable_sort(
43030b57cec5SDimitry Andric Options, [&TI](const CodeGenFunction::MultiVersionResolverOption &LHS,
43040b57cec5SDimitry Andric const CodeGenFunction::MultiVersionResolverOption &RHS) {
43050b57cec5SDimitry Andric return TargetMVPriority(TI, LHS) > TargetMVPriority(TI, RHS);
43060b57cec5SDimitry Andric });
43070b57cec5SDimitry Andric CodeGenFunction CGF(*this);
43080b57cec5SDimitry Andric CGF.EmitMultiVersionResolver(ResolverFunc, Options);
43090b57cec5SDimitry Andric }
4310fe6060f1SDimitry Andric
4311fe6060f1SDimitry Andric // Ensure that any additions to the deferred decls list caused by emitting a
4312fe6060f1SDimitry Andric // variant are emitted. This can happen when the variant itself is inline and
4313fe6060f1SDimitry Andric // calls a function without linkage.
4314fe6060f1SDimitry Andric if (!MVFuncsToEmit.empty())
4315fe6060f1SDimitry Andric EmitDeferred();
4316fe6060f1SDimitry Andric
4317fe6060f1SDimitry Andric // Ensure that any additions to the multiversion funcs list from either the
4318fe6060f1SDimitry Andric // deferred decls or the multiversion functions themselves are emitted.
4319fe6060f1SDimitry Andric if (!MultiVersionFuncs.empty())
4320fe6060f1SDimitry Andric emitMultiVersionFunctions();
43210b57cec5SDimitry Andric }
43220b57cec5SDimitry Andric
replaceDeclarationWith(llvm::GlobalValue * Old,llvm::Constant * New)43230fca6ea1SDimitry Andric static void replaceDeclarationWith(llvm::GlobalValue *Old,
43240fca6ea1SDimitry Andric llvm::Constant *New) {
43250fca6ea1SDimitry Andric assert(cast<llvm::Function>(Old)->isDeclaration() && "Not a declaration");
43260fca6ea1SDimitry Andric New->takeName(Old);
43270fca6ea1SDimitry Andric Old->replaceAllUsesWith(New);
43280fca6ea1SDimitry Andric Old->eraseFromParent();
43290fca6ea1SDimitry Andric }
43300fca6ea1SDimitry Andric
emitCPUDispatchDefinition(GlobalDecl GD)43310b57cec5SDimitry Andric void CodeGenModule::emitCPUDispatchDefinition(GlobalDecl GD) {
43320b57cec5SDimitry Andric const auto *FD = cast<FunctionDecl>(GD.getDecl());
43330b57cec5SDimitry Andric assert(FD && "Not a FunctionDecl?");
433404eeddc0SDimitry Andric assert(FD->isCPUDispatchMultiVersion() && "Not a multiversion function?");
43350b57cec5SDimitry Andric const auto *DD = FD->getAttr<CPUDispatchAttr>();
43360b57cec5SDimitry Andric assert(DD && "Not a cpu_dispatch Function?");
43370b57cec5SDimitry Andric
433881ad6265SDimitry Andric const CGFunctionInfo &FI = getTypes().arrangeGlobalDeclaration(GD);
433981ad6265SDimitry Andric llvm::FunctionType *DeclTy = getTypes().GetFunctionType(FI);
43400b57cec5SDimitry Andric
43410b57cec5SDimitry Andric StringRef ResolverName = getMangledName(GD);
434204eeddc0SDimitry Andric UpdateMultiVersionNames(GD, FD, ResolverName);
43430b57cec5SDimitry Andric
43440b57cec5SDimitry Andric llvm::Type *ResolverType;
43450b57cec5SDimitry Andric GlobalDecl ResolverGD;
434604eeddc0SDimitry Andric if (getTarget().supportsIFunc()) {
43470b57cec5SDimitry Andric ResolverType = llvm::FunctionType::get(
43480b57cec5SDimitry Andric llvm::PointerType::get(DeclTy,
4349bdd1243dSDimitry Andric getTypes().getTargetAddressSpace(FD->getType())),
43500b57cec5SDimitry Andric false);
435104eeddc0SDimitry Andric }
43520b57cec5SDimitry Andric else {
43530b57cec5SDimitry Andric ResolverType = DeclTy;
43540b57cec5SDimitry Andric ResolverGD = GD;
43550b57cec5SDimitry Andric }
43560b57cec5SDimitry Andric
43570b57cec5SDimitry Andric auto *ResolverFunc = cast<llvm::Function>(GetOrCreateLLVMFunction(
43580b57cec5SDimitry Andric ResolverName, ResolverType, ResolverGD, /*ForVTable=*/false));
4359349cc55cSDimitry Andric ResolverFunc->setLinkage(getMultiversionLinkage(*this, GD));
4360a7dea167SDimitry Andric if (supportsCOMDAT())
4361a7dea167SDimitry Andric ResolverFunc->setComdat(
4362a7dea167SDimitry Andric getModule().getOrInsertComdat(ResolverFunc->getName()));
43630b57cec5SDimitry Andric
43640b57cec5SDimitry Andric SmallVector<CodeGenFunction::MultiVersionResolverOption, 10> Options;
43650b57cec5SDimitry Andric const TargetInfo &Target = getTarget();
43660b57cec5SDimitry Andric unsigned Index = 0;
43670b57cec5SDimitry Andric for (const IdentifierInfo *II : DD->cpus()) {
43680b57cec5SDimitry Andric // Get the name of the target function so we can look it up/create it.
43690b57cec5SDimitry Andric std::string MangledName = getMangledNameImpl(*this, GD, FD, true) +
43700b57cec5SDimitry Andric getCPUSpecificMangling(*this, II->getName());
43710b57cec5SDimitry Andric
43720b57cec5SDimitry Andric llvm::Constant *Func = GetGlobalValue(MangledName);
43730b57cec5SDimitry Andric
43740b57cec5SDimitry Andric if (!Func) {
43750b57cec5SDimitry Andric GlobalDecl ExistingDecl = Manglings.lookup(MangledName);
43760b57cec5SDimitry Andric if (ExistingDecl.getDecl() &&
43770b57cec5SDimitry Andric ExistingDecl.getDecl()->getAsFunction()->isDefined()) {
43780b57cec5SDimitry Andric EmitGlobalFunctionDefinition(ExistingDecl, nullptr);
43790b57cec5SDimitry Andric Func = GetGlobalValue(MangledName);
43800b57cec5SDimitry Andric } else {
43810b57cec5SDimitry Andric if (!ExistingDecl.getDecl())
43820b57cec5SDimitry Andric ExistingDecl = GD.getWithMultiVersionIndex(Index);
43830b57cec5SDimitry Andric
43840b57cec5SDimitry Andric Func = GetOrCreateLLVMFunction(
43850b57cec5SDimitry Andric MangledName, DeclTy, ExistingDecl,
43860b57cec5SDimitry Andric /*ForVTable=*/false, /*DontDefer=*/true,
43870b57cec5SDimitry Andric /*IsThunk=*/false, llvm::AttributeList(), ForDefinition);
43880b57cec5SDimitry Andric }
43890b57cec5SDimitry Andric }
43900b57cec5SDimitry Andric
43910b57cec5SDimitry Andric llvm::SmallVector<StringRef, 32> Features;
43920b57cec5SDimitry Andric Target.getCPUSpecificCPUDispatchFeatures(II->getName(), Features);
43930b57cec5SDimitry Andric llvm::transform(Features, Features.begin(),
43940b57cec5SDimitry Andric [](StringRef Str) { return Str.substr(1); });
4395349cc55cSDimitry Andric llvm::erase_if(Features, [&Target](StringRef Feat) {
43960b57cec5SDimitry Andric return !Target.validateCpuSupports(Feat);
4397349cc55cSDimitry Andric });
43980b57cec5SDimitry Andric Options.emplace_back(cast<llvm::Function>(Func), StringRef{}, Features);
43990b57cec5SDimitry Andric ++Index;
44000b57cec5SDimitry Andric }
44010b57cec5SDimitry Andric
4402fe6060f1SDimitry Andric llvm::stable_sort(
44030b57cec5SDimitry Andric Options, [](const CodeGenFunction::MultiVersionResolverOption &LHS,
44040b57cec5SDimitry Andric const CodeGenFunction::MultiVersionResolverOption &RHS) {
4405349cc55cSDimitry Andric return llvm::X86::getCpuSupportsMask(LHS.Conditions.Features) >
4406349cc55cSDimitry Andric llvm::X86::getCpuSupportsMask(RHS.Conditions.Features);
44070b57cec5SDimitry Andric });
44080b57cec5SDimitry Andric
44090b57cec5SDimitry Andric // If the list contains multiple 'default' versions, such as when it contains
44100b57cec5SDimitry Andric // 'pentium' and 'generic', don't emit the call to the generic one (since we
44110b57cec5SDimitry Andric // always run on at least a 'pentium'). We do this by deleting the 'least
44120b57cec5SDimitry Andric // advanced' (read, lowest mangling letter).
44130b57cec5SDimitry Andric while (Options.size() > 1 &&
44145f757f3fSDimitry Andric llvm::all_of(llvm::X86::getCpuSupportsMask(
44155f757f3fSDimitry Andric (Options.end() - 2)->Conditions.Features),
44165f757f3fSDimitry Andric [](auto X) { return X == 0; })) {
44170b57cec5SDimitry Andric StringRef LHSName = (Options.end() - 2)->Function->getName();
44180b57cec5SDimitry Andric StringRef RHSName = (Options.end() - 1)->Function->getName();
44190b57cec5SDimitry Andric if (LHSName.compare(RHSName) < 0)
44200b57cec5SDimitry Andric Options.erase(Options.end() - 2);
44210b57cec5SDimitry Andric else
44220b57cec5SDimitry Andric Options.erase(Options.end() - 1);
44230b57cec5SDimitry Andric }
44240b57cec5SDimitry Andric
44250b57cec5SDimitry Andric CodeGenFunction CGF(*this);
44260b57cec5SDimitry Andric CGF.EmitMultiVersionResolver(ResolverFunc, Options);
4427a7dea167SDimitry Andric
4428a7dea167SDimitry Andric if (getTarget().supportsIFunc()) {
442981ad6265SDimitry Andric llvm::GlobalValue::LinkageTypes Linkage = getMultiversionLinkage(*this, GD);
443081ad6265SDimitry Andric auto *IFunc = cast<llvm::GlobalValue>(GetOrCreateMultiVersionResolver(GD));
443181ad6265SDimitry Andric
443281ad6265SDimitry Andric // Fix up function declarations that were created for cpu_specific before
443381ad6265SDimitry Andric // cpu_dispatch was known
443481ad6265SDimitry Andric if (!isa<llvm::GlobalIFunc>(IFunc)) {
443581ad6265SDimitry Andric auto *GI = llvm::GlobalIFunc::create(DeclTy, 0, Linkage, "", ResolverFunc,
443681ad6265SDimitry Andric &getModule());
44370fca6ea1SDimitry Andric replaceDeclarationWith(IFunc, GI);
443881ad6265SDimitry Andric IFunc = GI;
443981ad6265SDimitry Andric }
444081ad6265SDimitry Andric
4441a7dea167SDimitry Andric std::string AliasName = getMangledNameImpl(
4442a7dea167SDimitry Andric *this, GD, FD, /*OmitMultiVersionMangling=*/true);
4443a7dea167SDimitry Andric llvm::Constant *AliasFunc = GetGlobalValue(AliasName);
4444a7dea167SDimitry Andric if (!AliasFunc) {
444581ad6265SDimitry Andric auto *GA = llvm::GlobalAlias::create(DeclTy, 0, Linkage, AliasName, IFunc,
444681ad6265SDimitry Andric &getModule());
4447a7dea167SDimitry Andric SetCommonAttributes(GD, GA);
4448a7dea167SDimitry Andric }
4449a7dea167SDimitry Andric }
44500b57cec5SDimitry Andric }
44510b57cec5SDimitry Andric
44520fca6ea1SDimitry Andric /// Adds a declaration to the list of multi version functions if not present.
AddDeferredMultiVersionResolverToEmit(GlobalDecl GD)44530fca6ea1SDimitry Andric void CodeGenModule::AddDeferredMultiVersionResolverToEmit(GlobalDecl GD) {
44540fca6ea1SDimitry Andric const auto *FD = cast<FunctionDecl>(GD.getDecl());
44550fca6ea1SDimitry Andric assert(FD && "Not a FunctionDecl?");
44560fca6ea1SDimitry Andric
44570fca6ea1SDimitry Andric if (FD->isTargetVersionMultiVersion() || FD->isTargetClonesMultiVersion()) {
44580fca6ea1SDimitry Andric std::string MangledName =
44590fca6ea1SDimitry Andric getMangledNameImpl(*this, GD, FD, /*OmitMultiVersionMangling=*/true);
44600fca6ea1SDimitry Andric if (!DeferredResolversToEmit.insert(MangledName).second)
44610fca6ea1SDimitry Andric return;
44620fca6ea1SDimitry Andric }
44630fca6ea1SDimitry Andric MultiVersionFuncs.push_back(GD);
44640fca6ea1SDimitry Andric }
44650fca6ea1SDimitry Andric
44660b57cec5SDimitry Andric /// If a dispatcher for the specified mangled name is not in the module, create
44670fca6ea1SDimitry Andric /// and return it. The dispatcher is either an llvm Function with the specified
44680fca6ea1SDimitry Andric /// type, or a global ifunc.
GetOrCreateMultiVersionResolver(GlobalDecl GD)446981ad6265SDimitry Andric llvm::Constant *CodeGenModule::GetOrCreateMultiVersionResolver(GlobalDecl GD) {
447081ad6265SDimitry Andric const auto *FD = cast<FunctionDecl>(GD.getDecl());
447181ad6265SDimitry Andric assert(FD && "Not a FunctionDecl?");
447281ad6265SDimitry Andric
44730b57cec5SDimitry Andric std::string MangledName =
44740b57cec5SDimitry Andric getMangledNameImpl(*this, GD, FD, /*OmitMultiVersionMangling=*/true);
44750b57cec5SDimitry Andric
44760b57cec5SDimitry Andric // Holds the name of the resolver, in ifunc mode this is the ifunc (which has
44770b57cec5SDimitry Andric // a separate resolver).
44780b57cec5SDimitry Andric std::string ResolverName = MangledName;
44795f757f3fSDimitry Andric if (getTarget().supportsIFunc()) {
44800fca6ea1SDimitry Andric switch (FD->getMultiVersionKind()) {
44810fca6ea1SDimitry Andric case MultiVersionKind::None:
44820fca6ea1SDimitry Andric llvm_unreachable("unexpected MultiVersionKind::None for resolver");
44830fca6ea1SDimitry Andric case MultiVersionKind::Target:
44840fca6ea1SDimitry Andric case MultiVersionKind::CPUSpecific:
44850fca6ea1SDimitry Andric case MultiVersionKind::CPUDispatch:
44860b57cec5SDimitry Andric ResolverName += ".ifunc";
44870fca6ea1SDimitry Andric break;
44880fca6ea1SDimitry Andric case MultiVersionKind::TargetClones:
44890fca6ea1SDimitry Andric case MultiVersionKind::TargetVersion:
44900fca6ea1SDimitry Andric break;
44910fca6ea1SDimitry Andric }
44925f757f3fSDimitry Andric } else if (FD->isTargetMultiVersion()) {
44930b57cec5SDimitry Andric ResolverName += ".resolver";
44945f757f3fSDimitry Andric }
44950b57cec5SDimitry Andric
44960fca6ea1SDimitry Andric // If the resolver has already been created, just return it. This lookup may
44970fca6ea1SDimitry Andric // yield a function declaration instead of a resolver on AArch64. That is
44980fca6ea1SDimitry Andric // because we didn't know whether a resolver will be generated when we first
44990fca6ea1SDimitry Andric // encountered a use of the symbol named after this resolver. Therefore,
45000fca6ea1SDimitry Andric // targets which support ifuncs should not return here unless we actually
45010fca6ea1SDimitry Andric // found an ifunc.
45020fca6ea1SDimitry Andric llvm::GlobalValue *ResolverGV = GetGlobalValue(ResolverName);
45030fca6ea1SDimitry Andric if (ResolverGV &&
45040fca6ea1SDimitry Andric (isa<llvm::GlobalIFunc>(ResolverGV) || !getTarget().supportsIFunc()))
45050b57cec5SDimitry Andric return ResolverGV;
45060b57cec5SDimitry Andric
450781ad6265SDimitry Andric const CGFunctionInfo &FI = getTypes().arrangeGlobalDeclaration(GD);
450881ad6265SDimitry Andric llvm::FunctionType *DeclTy = getTypes().GetFunctionType(FI);
45090b57cec5SDimitry Andric
451081ad6265SDimitry Andric // The resolver needs to be created. For target and target_clones, defer
451181ad6265SDimitry Andric // creation until the end of the TU.
451281ad6265SDimitry Andric if (FD->isTargetMultiVersion() || FD->isTargetClonesMultiVersion())
45130fca6ea1SDimitry Andric AddDeferredMultiVersionResolverToEmit(GD);
451481ad6265SDimitry Andric
451581ad6265SDimitry Andric // For cpu_specific, don't create an ifunc yet because we don't know if the
451681ad6265SDimitry Andric // cpu_dispatch will be emitted in this translation unit.
451781ad6265SDimitry Andric if (getTarget().supportsIFunc() && !FD->isCPUSpecificMultiVersion()) {
45180b57cec5SDimitry Andric llvm::Type *ResolverType = llvm::FunctionType::get(
4519bdd1243dSDimitry Andric llvm::PointerType::get(DeclTy,
4520bdd1243dSDimitry Andric getTypes().getTargetAddressSpace(FD->getType())),
45210b57cec5SDimitry Andric false);
45220b57cec5SDimitry Andric llvm::Constant *Resolver = GetOrCreateLLVMFunction(
45230b57cec5SDimitry Andric MangledName + ".resolver", ResolverType, GlobalDecl{},
45240b57cec5SDimitry Andric /*ForVTable=*/false);
4525349cc55cSDimitry Andric llvm::GlobalIFunc *GIF =
4526349cc55cSDimitry Andric llvm::GlobalIFunc::create(DeclTy, 0, getMultiversionLinkage(*this, GD),
4527349cc55cSDimitry Andric "", Resolver, &getModule());
45280b57cec5SDimitry Andric GIF->setName(ResolverName);
45290b57cec5SDimitry Andric SetCommonAttributes(FD, GIF);
45300fca6ea1SDimitry Andric if (ResolverGV)
45310fca6ea1SDimitry Andric replaceDeclarationWith(ResolverGV, GIF);
45320b57cec5SDimitry Andric return GIF;
45330b57cec5SDimitry Andric }
45340b57cec5SDimitry Andric
45350b57cec5SDimitry Andric llvm::Constant *Resolver = GetOrCreateLLVMFunction(
45360b57cec5SDimitry Andric ResolverName, DeclTy, GlobalDecl{}, /*ForVTable=*/false);
45370b57cec5SDimitry Andric assert(isa<llvm::GlobalValue>(Resolver) &&
45380b57cec5SDimitry Andric "Resolver should be created for the first time");
45390b57cec5SDimitry Andric SetCommonAttributes(FD, cast<llvm::GlobalValue>(Resolver));
45400fca6ea1SDimitry Andric if (ResolverGV)
45410fca6ea1SDimitry Andric replaceDeclarationWith(ResolverGV, Resolver);
45420b57cec5SDimitry Andric return Resolver;
45430b57cec5SDimitry Andric }
45440b57cec5SDimitry Andric
shouldDropDLLAttribute(const Decl * D,const llvm::GlobalValue * GV) const45450fca6ea1SDimitry Andric bool CodeGenModule::shouldDropDLLAttribute(const Decl *D,
45460fca6ea1SDimitry Andric const llvm::GlobalValue *GV) const {
45470fca6ea1SDimitry Andric auto SC = GV->getDLLStorageClass();
45480fca6ea1SDimitry Andric if (SC == llvm::GlobalValue::DefaultStorageClass)
45490fca6ea1SDimitry Andric return false;
45500fca6ea1SDimitry Andric const Decl *MRD = D->getMostRecentDecl();
45510fca6ea1SDimitry Andric return (((SC == llvm::GlobalValue::DLLImportStorageClass &&
45520fca6ea1SDimitry Andric !MRD->hasAttr<DLLImportAttr>()) ||
45530fca6ea1SDimitry Andric (SC == llvm::GlobalValue::DLLExportStorageClass &&
45540fca6ea1SDimitry Andric !MRD->hasAttr<DLLExportAttr>())) &&
45550fca6ea1SDimitry Andric !shouldMapVisibilityToDLLExport(cast<NamedDecl>(MRD)));
45560fca6ea1SDimitry Andric }
45570fca6ea1SDimitry Andric
45580b57cec5SDimitry Andric /// GetOrCreateLLVMFunction - If the specified mangled name is not in the
45590b57cec5SDimitry Andric /// module, create and return an llvm Function with the specified type. If there
45600b57cec5SDimitry Andric /// is something in the module with the specified name, return it potentially
45610b57cec5SDimitry Andric /// bitcasted to the right type.
45620b57cec5SDimitry Andric ///
45630b57cec5SDimitry Andric /// If D is non-null, it specifies a decl that correspond to this. This is used
45640b57cec5SDimitry Andric /// to set the attributes on the function when it is first created.
GetOrCreateLLVMFunction(StringRef MangledName,llvm::Type * Ty,GlobalDecl GD,bool ForVTable,bool DontDefer,bool IsThunk,llvm::AttributeList ExtraAttrs,ForDefinition_t IsForDefinition)45650b57cec5SDimitry Andric llvm::Constant *CodeGenModule::GetOrCreateLLVMFunction(
45660b57cec5SDimitry Andric StringRef MangledName, llvm::Type *Ty, GlobalDecl GD, bool ForVTable,
45670b57cec5SDimitry Andric bool DontDefer, bool IsThunk, llvm::AttributeList ExtraAttrs,
45680b57cec5SDimitry Andric ForDefinition_t IsForDefinition) {
45690b57cec5SDimitry Andric const Decl *D = GD.getDecl();
45700b57cec5SDimitry Andric
45710fca6ea1SDimitry Andric std::string NameWithoutMultiVersionMangling;
45720b57cec5SDimitry Andric // Any attempts to use a MultiVersion function should result in retrieving
45730b57cec5SDimitry Andric // the iFunc instead. Name Mangling will handle the rest of the changes.
45740b57cec5SDimitry Andric if (const FunctionDecl *FD = cast_or_null<FunctionDecl>(D)) {
45750b57cec5SDimitry Andric // For the device mark the function as one that should be emitted.
457606c3fb27SDimitry Andric if (getLangOpts().OpenMPIsTargetDevice && OpenMPRuntime &&
45770b57cec5SDimitry Andric !OpenMPRuntime->markAsGlobalTarget(GD) && FD->isDefined() &&
45780b57cec5SDimitry Andric !DontDefer && !IsForDefinition) {
45790b57cec5SDimitry Andric if (const FunctionDecl *FDDef = FD->getDefinition()) {
45800b57cec5SDimitry Andric GlobalDecl GDDef;
45810b57cec5SDimitry Andric if (const auto *CD = dyn_cast<CXXConstructorDecl>(FDDef))
45820b57cec5SDimitry Andric GDDef = GlobalDecl(CD, GD.getCtorType());
45830b57cec5SDimitry Andric else if (const auto *DD = dyn_cast<CXXDestructorDecl>(FDDef))
45840b57cec5SDimitry Andric GDDef = GlobalDecl(DD, GD.getDtorType());
45850b57cec5SDimitry Andric else
45860b57cec5SDimitry Andric GDDef = GlobalDecl(FDDef);
45870b57cec5SDimitry Andric EmitGlobal(GDDef);
45880b57cec5SDimitry Andric }
45890b57cec5SDimitry Andric }
45900b57cec5SDimitry Andric
45910b57cec5SDimitry Andric if (FD->isMultiVersion()) {
459204eeddc0SDimitry Andric UpdateMultiVersionNames(GD, FD, MangledName);
45930fca6ea1SDimitry Andric if (!IsForDefinition) {
45940fca6ea1SDimitry Andric // On AArch64 we do not immediatelly emit an ifunc resolver when a
45950fca6ea1SDimitry Andric // function is used. Instead we defer the emission until we see a
45960fca6ea1SDimitry Andric // default definition. In the meantime we just reference the symbol
45970fca6ea1SDimitry Andric // without FMV mangling (it may or may not be replaced later).
45980fca6ea1SDimitry Andric if (getTarget().getTriple().isAArch64()) {
45990fca6ea1SDimitry Andric AddDeferredMultiVersionResolverToEmit(GD);
46000fca6ea1SDimitry Andric NameWithoutMultiVersionMangling = getMangledNameImpl(
46010fca6ea1SDimitry Andric *this, GD, FD, /*OmitMultiVersionMangling=*/true);
46020fca6ea1SDimitry Andric } else
460381ad6265SDimitry Andric return GetOrCreateMultiVersionResolver(GD);
46040b57cec5SDimitry Andric }
46050b57cec5SDimitry Andric }
46060fca6ea1SDimitry Andric }
46070fca6ea1SDimitry Andric
46080fca6ea1SDimitry Andric if (!NameWithoutMultiVersionMangling.empty())
46090fca6ea1SDimitry Andric MangledName = NameWithoutMultiVersionMangling;
46100b57cec5SDimitry Andric
46110b57cec5SDimitry Andric // Lookup the entry, lazily creating it if necessary.
46120b57cec5SDimitry Andric llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
46130b57cec5SDimitry Andric if (Entry) {
46140b57cec5SDimitry Andric if (WeakRefReferences.erase(Entry)) {
46150b57cec5SDimitry Andric const FunctionDecl *FD = cast_or_null<FunctionDecl>(D);
46160b57cec5SDimitry Andric if (FD && !FD->hasAttr<WeakAttr>())
46170b57cec5SDimitry Andric Entry->setLinkage(llvm::Function::ExternalLinkage);
46180b57cec5SDimitry Andric }
46190b57cec5SDimitry Andric
46200b57cec5SDimitry Andric // Handle dropped DLL attributes.
46210fca6ea1SDimitry Andric if (D && shouldDropDLLAttribute(D, Entry)) {
46220b57cec5SDimitry Andric Entry->setDLLStorageClass(llvm::GlobalValue::DefaultStorageClass);
46230b57cec5SDimitry Andric setDSOLocal(Entry);
46240b57cec5SDimitry Andric }
46250b57cec5SDimitry Andric
46260b57cec5SDimitry Andric // If there are two attempts to define the same mangled name, issue an
46270b57cec5SDimitry Andric // error.
46280b57cec5SDimitry Andric if (IsForDefinition && !Entry->isDeclaration()) {
46290b57cec5SDimitry Andric GlobalDecl OtherGD;
46300b57cec5SDimitry Andric // Check that GD is not yet in DiagnosedConflictingDefinitions is required
46310b57cec5SDimitry Andric // to make sure that we issue an error only once.
46320b57cec5SDimitry Andric if (lookupRepresentativeDecl(MangledName, OtherGD) &&
46330b57cec5SDimitry Andric (GD.getCanonicalDecl().getDecl() !=
46340b57cec5SDimitry Andric OtherGD.getCanonicalDecl().getDecl()) &&
46350b57cec5SDimitry Andric DiagnosedConflictingDefinitions.insert(GD).second) {
46360b57cec5SDimitry Andric getDiags().Report(D->getLocation(), diag::err_duplicate_mangled_name)
46370b57cec5SDimitry Andric << MangledName;
46380b57cec5SDimitry Andric getDiags().Report(OtherGD.getDecl()->getLocation(),
46390b57cec5SDimitry Andric diag::note_previous_definition);
46400b57cec5SDimitry Andric }
46410b57cec5SDimitry Andric }
46420b57cec5SDimitry Andric
46430b57cec5SDimitry Andric if ((isa<llvm::Function>(Entry) || isa<llvm::GlobalAlias>(Entry)) &&
46445ffd83dbSDimitry Andric (Entry->getValueType() == Ty)) {
46450b57cec5SDimitry Andric return Entry;
46460b57cec5SDimitry Andric }
46470b57cec5SDimitry Andric
46480b57cec5SDimitry Andric // Make sure the result is of the correct type.
46490b57cec5SDimitry Andric // (If function is requested for a definition, we always need to create a new
46500b57cec5SDimitry Andric // function, not just return a bitcast.)
46510b57cec5SDimitry Andric if (!IsForDefinition)
46525f757f3fSDimitry Andric return Entry;
46530b57cec5SDimitry Andric }
46540b57cec5SDimitry Andric
46550b57cec5SDimitry Andric // This function doesn't have a complete type (for example, the return
46560b57cec5SDimitry Andric // type is an incomplete struct). Use a fake type instead, and make
46570b57cec5SDimitry Andric // sure not to try to set attributes.
46580b57cec5SDimitry Andric bool IsIncompleteFunction = false;
46590b57cec5SDimitry Andric
46600b57cec5SDimitry Andric llvm::FunctionType *FTy;
46610b57cec5SDimitry Andric if (isa<llvm::FunctionType>(Ty)) {
46620b57cec5SDimitry Andric FTy = cast<llvm::FunctionType>(Ty);
46630b57cec5SDimitry Andric } else {
46640b57cec5SDimitry Andric FTy = llvm::FunctionType::get(VoidTy, false);
46650b57cec5SDimitry Andric IsIncompleteFunction = true;
46660b57cec5SDimitry Andric }
46670b57cec5SDimitry Andric
46680b57cec5SDimitry Andric llvm::Function *F =
46690b57cec5SDimitry Andric llvm::Function::Create(FTy, llvm::Function::ExternalLinkage,
46700b57cec5SDimitry Andric Entry ? StringRef() : MangledName, &getModule());
46710b57cec5SDimitry Andric
46725f757f3fSDimitry Andric // Store the declaration associated with this function so it is potentially
46735f757f3fSDimitry Andric // updated by further declarations or definitions and emitted at the end.
46745f757f3fSDimitry Andric if (D && D->hasAttr<AnnotateAttr>())
46755f757f3fSDimitry Andric DeferredAnnotations[MangledName] = cast<ValueDecl>(D);
46765f757f3fSDimitry Andric
46770b57cec5SDimitry Andric // If we already created a function with the same mangled name (but different
46780b57cec5SDimitry Andric // type) before, take its name and add it to the list of functions to be
46790b57cec5SDimitry Andric // replaced with F at the end of CodeGen.
46800b57cec5SDimitry Andric //
46810b57cec5SDimitry Andric // This happens if there is a prototype for a function (e.g. "int f()") and
46820b57cec5SDimitry Andric // then a definition of a different type (e.g. "int f(int x)").
46830b57cec5SDimitry Andric if (Entry) {
46840b57cec5SDimitry Andric F->takeName(Entry);
46850b57cec5SDimitry Andric
46860b57cec5SDimitry Andric // This might be an implementation of a function without a prototype, in
46870b57cec5SDimitry Andric // which case, try to do special replacement of calls which match the new
46880b57cec5SDimitry Andric // prototype. The really key thing here is that we also potentially drop
46890b57cec5SDimitry Andric // arguments from the call site so as to make a direct call, which makes the
46900b57cec5SDimitry Andric // inliner happier and suppresses a number of optimizer warnings (!) about
46910b57cec5SDimitry Andric // dropping arguments.
46920b57cec5SDimitry Andric if (!Entry->use_empty()) {
46930b57cec5SDimitry Andric ReplaceUsesOfNonProtoTypeWithRealFunction(Entry, F);
46940b57cec5SDimitry Andric Entry->removeDeadConstantUsers();
46950b57cec5SDimitry Andric }
46960b57cec5SDimitry Andric
46975f757f3fSDimitry Andric addGlobalValReplacement(Entry, F);
46980b57cec5SDimitry Andric }
46990b57cec5SDimitry Andric
47000b57cec5SDimitry Andric assert(F->getName() == MangledName && "name was uniqued!");
47010b57cec5SDimitry Andric if (D)
47020b57cec5SDimitry Andric SetFunctionAttributes(GD, F, IsIncompleteFunction, IsThunk);
4703349cc55cSDimitry Andric if (ExtraAttrs.hasFnAttrs()) {
470404eeddc0SDimitry Andric llvm::AttrBuilder B(F->getContext(), ExtraAttrs.getFnAttrs());
4705349cc55cSDimitry Andric F->addFnAttrs(B);
47060b57cec5SDimitry Andric }
47070b57cec5SDimitry Andric
47080b57cec5SDimitry Andric if (!DontDefer) {
47090b57cec5SDimitry Andric // All MSVC dtors other than the base dtor are linkonce_odr and delegate to
47100b57cec5SDimitry Andric // each other bottoming out with the base dtor. Therefore we emit non-base
47110b57cec5SDimitry Andric // dtors on usage, even if there is no dtor definition in the TU.
4712bdd1243dSDimitry Andric if (isa_and_nonnull<CXXDestructorDecl>(D) &&
47130b57cec5SDimitry Andric getCXXABI().useThunkForDtorVariant(cast<CXXDestructorDecl>(D),
47140b57cec5SDimitry Andric GD.getDtorType()))
47150b57cec5SDimitry Andric addDeferredDeclToEmit(GD);
47160b57cec5SDimitry Andric
47170b57cec5SDimitry Andric // This is the first use or definition of a mangled name. If there is a
47180b57cec5SDimitry Andric // deferred decl with this name, remember that we need to emit it at the end
47190b57cec5SDimitry Andric // of the file.
47200b57cec5SDimitry Andric auto DDI = DeferredDecls.find(MangledName);
47210b57cec5SDimitry Andric if (DDI != DeferredDecls.end()) {
47220b57cec5SDimitry Andric // Move the potentially referenced deferred decl to the
47230b57cec5SDimitry Andric // DeferredDeclsToEmit list, and remove it from DeferredDecls (since we
47240b57cec5SDimitry Andric // don't need it anymore).
47250b57cec5SDimitry Andric addDeferredDeclToEmit(DDI->second);
47260b57cec5SDimitry Andric DeferredDecls.erase(DDI);
47270b57cec5SDimitry Andric
47280b57cec5SDimitry Andric // Otherwise, there are cases we have to worry about where we're
47290b57cec5SDimitry Andric // using a declaration for which we must emit a definition but where
47300b57cec5SDimitry Andric // we might not find a top-level definition:
47310b57cec5SDimitry Andric // - member functions defined inline in their classes
47320b57cec5SDimitry Andric // - friend functions defined inline in some class
47330b57cec5SDimitry Andric // - special member functions with implicit definitions
47340b57cec5SDimitry Andric // If we ever change our AST traversal to walk into class methods,
47350b57cec5SDimitry Andric // this will be unnecessary.
47360b57cec5SDimitry Andric //
47370b57cec5SDimitry Andric // We also don't emit a definition for a function if it's going to be an
47380b57cec5SDimitry Andric // entry in a vtable, unless it's already marked as used.
47390b57cec5SDimitry Andric } else if (getLangOpts().CPlusPlus && D) {
47400b57cec5SDimitry Andric // Look for a declaration that's lexically in a record.
47410b57cec5SDimitry Andric for (const auto *FD = cast<FunctionDecl>(D)->getMostRecentDecl(); FD;
47420b57cec5SDimitry Andric FD = FD->getPreviousDecl()) {
47430b57cec5SDimitry Andric if (isa<CXXRecordDecl>(FD->getLexicalDeclContext())) {
47440b57cec5SDimitry Andric if (FD->doesThisDeclarationHaveABody()) {
47450b57cec5SDimitry Andric addDeferredDeclToEmit(GD.getWithDecl(FD));
47460b57cec5SDimitry Andric break;
47470b57cec5SDimitry Andric }
47480b57cec5SDimitry Andric }
47490b57cec5SDimitry Andric }
47500b57cec5SDimitry Andric }
47510b57cec5SDimitry Andric }
47520b57cec5SDimitry Andric
47530b57cec5SDimitry Andric // Make sure the result is of the requested type.
47540b57cec5SDimitry Andric if (!IsIncompleteFunction) {
47555ffd83dbSDimitry Andric assert(F->getFunctionType() == Ty);
47560b57cec5SDimitry Andric return F;
47570b57cec5SDimitry Andric }
47580b57cec5SDimitry Andric
47595f757f3fSDimitry Andric return F;
47600b57cec5SDimitry Andric }
47610b57cec5SDimitry Andric
47620b57cec5SDimitry Andric /// GetAddrOfFunction - Return the address of the given function. If Ty is
47630b57cec5SDimitry Andric /// non-null, then this function will use the specified type if it has to
47640b57cec5SDimitry Andric /// create it (this occurs when we see a definition of the function).
476506c3fb27SDimitry Andric llvm::Constant *
GetAddrOfFunction(GlobalDecl GD,llvm::Type * Ty,bool ForVTable,bool DontDefer,ForDefinition_t IsForDefinition)476606c3fb27SDimitry Andric CodeGenModule::GetAddrOfFunction(GlobalDecl GD, llvm::Type *Ty, bool ForVTable,
47670b57cec5SDimitry Andric bool DontDefer,
47680b57cec5SDimitry Andric ForDefinition_t IsForDefinition) {
47690b57cec5SDimitry Andric // If there was no specific requested type, just convert it now.
47700b57cec5SDimitry Andric if (!Ty) {
47710b57cec5SDimitry Andric const auto *FD = cast<FunctionDecl>(GD.getDecl());
47720b57cec5SDimitry Andric Ty = getTypes().ConvertType(FD->getType());
47730b57cec5SDimitry Andric }
47740b57cec5SDimitry Andric
47750b57cec5SDimitry Andric // Devirtualized destructor calls may come through here instead of via
47760b57cec5SDimitry Andric // getAddrOfCXXStructor. Make sure we use the MS ABI base destructor instead
47770b57cec5SDimitry Andric // of the complete destructor when necessary.
47780b57cec5SDimitry Andric if (const auto *DD = dyn_cast<CXXDestructorDecl>(GD.getDecl())) {
47790b57cec5SDimitry Andric if (getTarget().getCXXABI().isMicrosoft() &&
47800b57cec5SDimitry Andric GD.getDtorType() == Dtor_Complete &&
47810b57cec5SDimitry Andric DD->getParent()->getNumVBases() == 0)
47820b57cec5SDimitry Andric GD = GlobalDecl(DD, Dtor_Base);
47830b57cec5SDimitry Andric }
47840b57cec5SDimitry Andric
47850b57cec5SDimitry Andric StringRef MangledName = getMangledName(GD);
4786fe6060f1SDimitry Andric auto *F = GetOrCreateLLVMFunction(MangledName, Ty, GD, ForVTable, DontDefer,
47870b57cec5SDimitry Andric /*IsThunk=*/false, llvm::AttributeList(),
47880b57cec5SDimitry Andric IsForDefinition);
4789fe6060f1SDimitry Andric // Returns kernel handle for HIP kernel stub function.
4790fe6060f1SDimitry Andric if (LangOpts.CUDA && !LangOpts.CUDAIsDevice &&
4791fe6060f1SDimitry Andric cast<FunctionDecl>(GD.getDecl())->hasAttr<CUDAGlobalAttr>()) {
4792fe6060f1SDimitry Andric auto *Handle = getCUDARuntime().getKernelHandle(
4793fe6060f1SDimitry Andric cast<llvm::Function>(F->stripPointerCasts()), GD);
4794fe6060f1SDimitry Andric if (IsForDefinition)
4795fe6060f1SDimitry Andric return F;
47965f757f3fSDimitry Andric return Handle;
4797fe6060f1SDimitry Andric }
4798fe6060f1SDimitry Andric return F;
47990b57cec5SDimitry Andric }
48000b57cec5SDimitry Andric
GetFunctionStart(const ValueDecl * Decl)48010eae32dcSDimitry Andric llvm::Constant *CodeGenModule::GetFunctionStart(const ValueDecl *Decl) {
48020eae32dcSDimitry Andric llvm::GlobalValue *F =
48030eae32dcSDimitry Andric cast<llvm::GlobalValue>(GetAddrOfFunction(Decl)->stripPointerCasts());
48040eae32dcSDimitry Andric
48055f757f3fSDimitry Andric return llvm::NoCFIValue::get(F);
48060eae32dcSDimitry Andric }
48070eae32dcSDimitry Andric
48080b57cec5SDimitry Andric static const FunctionDecl *
GetRuntimeFunctionDecl(ASTContext & C,StringRef Name)48090b57cec5SDimitry Andric GetRuntimeFunctionDecl(ASTContext &C, StringRef Name) {
48100b57cec5SDimitry Andric TranslationUnitDecl *TUDecl = C.getTranslationUnitDecl();
48110b57cec5SDimitry Andric DeclContext *DC = TranslationUnitDecl::castToDeclContext(TUDecl);
48120b57cec5SDimitry Andric
48130b57cec5SDimitry Andric IdentifierInfo &CII = C.Idents.get(Name);
4814fe6060f1SDimitry Andric for (const auto *Result : DC->lookup(&CII))
4815fe6060f1SDimitry Andric if (const auto *FD = dyn_cast<FunctionDecl>(Result))
48160b57cec5SDimitry Andric return FD;
48170b57cec5SDimitry Andric
48180b57cec5SDimitry Andric if (!C.getLangOpts().CPlusPlus)
48190b57cec5SDimitry Andric return nullptr;
48200b57cec5SDimitry Andric
48210b57cec5SDimitry Andric // Demangle the premangled name from getTerminateFn()
48220b57cec5SDimitry Andric IdentifierInfo &CXXII =
48230b57cec5SDimitry Andric (Name == "_ZSt9terminatev" || Name == "?terminate@@YAXXZ")
48240b57cec5SDimitry Andric ? C.Idents.get("terminate")
48250b57cec5SDimitry Andric : C.Idents.get(Name);
48260b57cec5SDimitry Andric
48270b57cec5SDimitry Andric for (const auto &N : {"__cxxabiv1", "std"}) {
48280b57cec5SDimitry Andric IdentifierInfo &NS = C.Idents.get(N);
4829fe6060f1SDimitry Andric for (const auto *Result : DC->lookup(&NS)) {
4830fe6060f1SDimitry Andric const NamespaceDecl *ND = dyn_cast<NamespaceDecl>(Result);
4831fe6060f1SDimitry Andric if (auto *LSD = dyn_cast<LinkageSpecDecl>(Result))
4832fe6060f1SDimitry Andric for (const auto *Result : LSD->lookup(&NS))
48330b57cec5SDimitry Andric if ((ND = dyn_cast<NamespaceDecl>(Result)))
48340b57cec5SDimitry Andric break;
48350b57cec5SDimitry Andric
48360b57cec5SDimitry Andric if (ND)
4837fe6060f1SDimitry Andric for (const auto *Result : ND->lookup(&CXXII))
48380b57cec5SDimitry Andric if (const auto *FD = dyn_cast<FunctionDecl>(Result))
48390b57cec5SDimitry Andric return FD;
48400b57cec5SDimitry Andric }
48410b57cec5SDimitry Andric }
48420b57cec5SDimitry Andric
48430b57cec5SDimitry Andric return nullptr;
48440b57cec5SDimitry Andric }
48450b57cec5SDimitry Andric
48460b57cec5SDimitry Andric /// CreateRuntimeFunction - Create a new runtime function with the specified
48470b57cec5SDimitry Andric /// type and name.
48480b57cec5SDimitry Andric llvm::FunctionCallee
CreateRuntimeFunction(llvm::FunctionType * FTy,StringRef Name,llvm::AttributeList ExtraAttrs,bool Local,bool AssumeConvergent)48490b57cec5SDimitry Andric CodeGenModule::CreateRuntimeFunction(llvm::FunctionType *FTy, StringRef Name,
4850480093f4SDimitry Andric llvm::AttributeList ExtraAttrs, bool Local,
4851480093f4SDimitry Andric bool AssumeConvergent) {
4852480093f4SDimitry Andric if (AssumeConvergent) {
4853480093f4SDimitry Andric ExtraAttrs =
4854349cc55cSDimitry Andric ExtraAttrs.addFnAttribute(VMContext, llvm::Attribute::Convergent);
4855480093f4SDimitry Andric }
4856480093f4SDimitry Andric
48570b57cec5SDimitry Andric llvm::Constant *C =
48580b57cec5SDimitry Andric GetOrCreateLLVMFunction(Name, FTy, GlobalDecl(), /*ForVTable=*/false,
48590b57cec5SDimitry Andric /*DontDefer=*/false, /*IsThunk=*/false,
48600b57cec5SDimitry Andric ExtraAttrs);
48610b57cec5SDimitry Andric
48620b57cec5SDimitry Andric if (auto *F = dyn_cast<llvm::Function>(C)) {
48630b57cec5SDimitry Andric if (F->empty()) {
48640b57cec5SDimitry Andric F->setCallingConv(getRuntimeCC());
48650b57cec5SDimitry Andric
48660b57cec5SDimitry Andric // In Windows Itanium environments, try to mark runtime functions
48670b57cec5SDimitry Andric // dllimport. For Mingw and MSVC, don't. We don't really know if the user
48680b57cec5SDimitry Andric // will link their standard library statically or dynamically. Marking
48690b57cec5SDimitry Andric // functions imported when they are not imported can cause linker errors
48700b57cec5SDimitry Andric // and warnings.
48710b57cec5SDimitry Andric if (!Local && getTriple().isWindowsItaniumEnvironment() &&
48720b57cec5SDimitry Andric !getCodeGenOpts().LTOVisibilityPublicStd) {
48730b57cec5SDimitry Andric const FunctionDecl *FD = GetRuntimeFunctionDecl(Context, Name);
48740b57cec5SDimitry Andric if (!FD || FD->hasAttr<DLLImportAttr>()) {
48750b57cec5SDimitry Andric F->setDLLStorageClass(llvm::GlobalValue::DLLImportStorageClass);
48760b57cec5SDimitry Andric F->setLinkage(llvm::GlobalValue::ExternalLinkage);
48770b57cec5SDimitry Andric }
48780b57cec5SDimitry Andric }
48790b57cec5SDimitry Andric setDSOLocal(F);
48800fca6ea1SDimitry Andric // FIXME: We should use CodeGenModule::SetLLVMFunctionAttributes() instead
48810fca6ea1SDimitry Andric // of trying to approximate the attributes using the LLVM function
48820fca6ea1SDimitry Andric // signature. This requires revising the API of CreateRuntimeFunction().
48830fca6ea1SDimitry Andric markRegisterParameterAttributes(F);
48840b57cec5SDimitry Andric }
48850b57cec5SDimitry Andric }
48860b57cec5SDimitry Andric
48870b57cec5SDimitry Andric return {FTy, C};
48880b57cec5SDimitry Andric }
48890b57cec5SDimitry Andric
48900b57cec5SDimitry Andric /// GetOrCreateLLVMGlobal - If the specified mangled name is not in the module,
4891fe6060f1SDimitry Andric /// create and return an llvm GlobalVariable with the specified type and address
4892fe6060f1SDimitry Andric /// space. If there is something in the module with the specified name, return
4893fe6060f1SDimitry Andric /// it potentially bitcasted to the right type.
48940b57cec5SDimitry Andric ///
48950b57cec5SDimitry Andric /// If D is non-null, it specifies a decl that correspond to this. This is used
48960b57cec5SDimitry Andric /// to set the attributes on the global when it is first created.
48970b57cec5SDimitry Andric ///
48980b57cec5SDimitry Andric /// If IsForDefinition is true, it is guaranteed that an actual global with
48990b57cec5SDimitry Andric /// type Ty will be returned, not conversion of a variable with the same
49000b57cec5SDimitry Andric /// mangled name but some other type.
49010b57cec5SDimitry Andric llvm::Constant *
GetOrCreateLLVMGlobal(StringRef MangledName,llvm::Type * Ty,LangAS AddrSpace,const VarDecl * D,ForDefinition_t IsForDefinition)4902fe6060f1SDimitry Andric CodeGenModule::GetOrCreateLLVMGlobal(StringRef MangledName, llvm::Type *Ty,
4903349cc55cSDimitry Andric LangAS AddrSpace, const VarDecl *D,
49040b57cec5SDimitry Andric ForDefinition_t IsForDefinition) {
49050b57cec5SDimitry Andric // Lookup the entry, lazily creating it if necessary.
49060b57cec5SDimitry Andric llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
4907349cc55cSDimitry Andric unsigned TargetAS = getContext().getTargetAddressSpace(AddrSpace);
49080b57cec5SDimitry Andric if (Entry) {
49090b57cec5SDimitry Andric if (WeakRefReferences.erase(Entry)) {
49100b57cec5SDimitry Andric if (D && !D->hasAttr<WeakAttr>())
49110b57cec5SDimitry Andric Entry->setLinkage(llvm::Function::ExternalLinkage);
49120b57cec5SDimitry Andric }
49130b57cec5SDimitry Andric
49140b57cec5SDimitry Andric // Handle dropped DLL attributes.
49150fca6ea1SDimitry Andric if (D && shouldDropDLLAttribute(D, Entry))
49160b57cec5SDimitry Andric Entry->setDLLStorageClass(llvm::GlobalValue::DefaultStorageClass);
49170b57cec5SDimitry Andric
49180b57cec5SDimitry Andric if (LangOpts.OpenMP && !LangOpts.OpenMPSimd && D)
49190b57cec5SDimitry Andric getOpenMPRuntime().registerTargetGlobalVariable(D, Entry);
49200b57cec5SDimitry Andric
4921349cc55cSDimitry Andric if (Entry->getValueType() == Ty && Entry->getAddressSpace() == TargetAS)
49220b57cec5SDimitry Andric return Entry;
49230b57cec5SDimitry Andric
49240b57cec5SDimitry Andric // If there are two attempts to define the same mangled name, issue an
49250b57cec5SDimitry Andric // error.
49260b57cec5SDimitry Andric if (IsForDefinition && !Entry->isDeclaration()) {
49270b57cec5SDimitry Andric GlobalDecl OtherGD;
49280b57cec5SDimitry Andric const VarDecl *OtherD;
49290b57cec5SDimitry Andric
49300b57cec5SDimitry Andric // Check that D is not yet in DiagnosedConflictingDefinitions is required
49310b57cec5SDimitry Andric // to make sure that we issue an error only once.
49320b57cec5SDimitry Andric if (D && lookupRepresentativeDecl(MangledName, OtherGD) &&
49330b57cec5SDimitry Andric (D->getCanonicalDecl() != OtherGD.getCanonicalDecl().getDecl()) &&
49340b57cec5SDimitry Andric (OtherD = dyn_cast<VarDecl>(OtherGD.getDecl())) &&
49350b57cec5SDimitry Andric OtherD->hasInit() &&
49360b57cec5SDimitry Andric DiagnosedConflictingDefinitions.insert(D).second) {
49370b57cec5SDimitry Andric getDiags().Report(D->getLocation(), diag::err_duplicate_mangled_name)
49380b57cec5SDimitry Andric << MangledName;
49390b57cec5SDimitry Andric getDiags().Report(OtherGD.getDecl()->getLocation(),
49400b57cec5SDimitry Andric diag::note_previous_definition);
49410b57cec5SDimitry Andric }
49420b57cec5SDimitry Andric }
49430b57cec5SDimitry Andric
49440b57cec5SDimitry Andric // Make sure the result is of the correct type.
49455f757f3fSDimitry Andric if (Entry->getType()->getAddressSpace() != TargetAS)
49465f757f3fSDimitry Andric return llvm::ConstantExpr::getAddrSpaceCast(
49475f757f3fSDimitry Andric Entry, llvm::PointerType::get(Ty->getContext(), TargetAS));
49480b57cec5SDimitry Andric
49490b57cec5SDimitry Andric // (If global is requested for a definition, we always need to create a new
49500b57cec5SDimitry Andric // global, not just return a bitcast.)
49510b57cec5SDimitry Andric if (!IsForDefinition)
49525f757f3fSDimitry Andric return Entry;
49530b57cec5SDimitry Andric }
49540b57cec5SDimitry Andric
4955fe6060f1SDimitry Andric auto DAddrSpace = GetGlobalVarAddressSpace(D);
49560b57cec5SDimitry Andric
49570b57cec5SDimitry Andric auto *GV = new llvm::GlobalVariable(
4958fe6060f1SDimitry Andric getModule(), Ty, false, llvm::GlobalValue::ExternalLinkage, nullptr,
4959fe6060f1SDimitry Andric MangledName, nullptr, llvm::GlobalVariable::NotThreadLocal,
4960349cc55cSDimitry Andric getContext().getTargetAddressSpace(DAddrSpace));
49610b57cec5SDimitry Andric
49620b57cec5SDimitry Andric // If we already created a global with the same mangled name (but different
49630b57cec5SDimitry Andric // type) before, take its name and remove it from its parent.
49640b57cec5SDimitry Andric if (Entry) {
49650b57cec5SDimitry Andric GV->takeName(Entry);
49660b57cec5SDimitry Andric
49670b57cec5SDimitry Andric if (!Entry->use_empty()) {
49685f757f3fSDimitry Andric Entry->replaceAllUsesWith(GV);
49690b57cec5SDimitry Andric }
49700b57cec5SDimitry Andric
49710b57cec5SDimitry Andric Entry->eraseFromParent();
49720b57cec5SDimitry Andric }
49730b57cec5SDimitry Andric
49740b57cec5SDimitry Andric // This is the first use or definition of a mangled name. If there is a
49750b57cec5SDimitry Andric // deferred decl with this name, remember that we need to emit it at the end
49760b57cec5SDimitry Andric // of the file.
49770b57cec5SDimitry Andric auto DDI = DeferredDecls.find(MangledName);
49780b57cec5SDimitry Andric if (DDI != DeferredDecls.end()) {
49790b57cec5SDimitry Andric // Move the potentially referenced deferred decl to the DeferredDeclsToEmit
49800b57cec5SDimitry Andric // list, and remove it from DeferredDecls (since we don't need it anymore).
49810b57cec5SDimitry Andric addDeferredDeclToEmit(DDI->second);
49820b57cec5SDimitry Andric DeferredDecls.erase(DDI);
49830b57cec5SDimitry Andric }
49840b57cec5SDimitry Andric
49850b57cec5SDimitry Andric // Handle things which are present even on external declarations.
49860b57cec5SDimitry Andric if (D) {
49870b57cec5SDimitry Andric if (LangOpts.OpenMP && !LangOpts.OpenMPSimd)
49880b57cec5SDimitry Andric getOpenMPRuntime().registerTargetGlobalVariable(D, GV);
49890b57cec5SDimitry Andric
49900b57cec5SDimitry Andric // FIXME: This code is overly simple and should be merged with other global
49910b57cec5SDimitry Andric // handling.
49925f757f3fSDimitry Andric GV->setConstant(D->getType().isConstantStorage(getContext(), false, false));
49930b57cec5SDimitry Andric
4994a7dea167SDimitry Andric GV->setAlignment(getContext().getDeclAlign(D).getAsAlign());
49950b57cec5SDimitry Andric
49960b57cec5SDimitry Andric setLinkageForGV(GV, D);
49970b57cec5SDimitry Andric
49980b57cec5SDimitry Andric if (D->getTLSKind()) {
49990b57cec5SDimitry Andric if (D->getTLSKind() == VarDecl::TLS_Dynamic)
50000b57cec5SDimitry Andric CXXThreadLocals.push_back(D);
50010b57cec5SDimitry Andric setTLSMode(GV, *D);
50020b57cec5SDimitry Andric }
50030b57cec5SDimitry Andric
50040b57cec5SDimitry Andric setGVProperties(GV, D);
50050b57cec5SDimitry Andric
50060b57cec5SDimitry Andric // If required by the ABI, treat declarations of static data members with
50070b57cec5SDimitry Andric // inline initializers as definitions.
50080b57cec5SDimitry Andric if (getContext().isMSStaticDataMemberInlineDefinition(D)) {
50090b57cec5SDimitry Andric EmitGlobalVarDefinition(D);
50100b57cec5SDimitry Andric }
50110b57cec5SDimitry Andric
50120b57cec5SDimitry Andric // Emit section information for extern variables.
50130b57cec5SDimitry Andric if (D->hasExternalStorage()) {
50140b57cec5SDimitry Andric if (const SectionAttr *SA = D->getAttr<SectionAttr>())
50150b57cec5SDimitry Andric GV->setSection(SA->getName());
50160b57cec5SDimitry Andric }
50170b57cec5SDimitry Andric
50180b57cec5SDimitry Andric // Handle XCore specific ABI requirements.
50190b57cec5SDimitry Andric if (getTriple().getArch() == llvm::Triple::xcore &&
50200b57cec5SDimitry Andric D->getLanguageLinkage() == CLanguageLinkage &&
50210b57cec5SDimitry Andric D->getType().isConstant(Context) &&
50220b57cec5SDimitry Andric isExternallyVisible(D->getLinkageAndVisibility().getLinkage()))
50230b57cec5SDimitry Andric GV->setSection(".cp.rodata");
50240b57cec5SDimitry Andric
50251db9f3b2SDimitry Andric // Handle code model attribute
50261db9f3b2SDimitry Andric if (const auto *CMA = D->getAttr<CodeModelAttr>())
50271db9f3b2SDimitry Andric GV->setCodeModel(CMA->getModel());
50281db9f3b2SDimitry Andric
50290b57cec5SDimitry Andric // Check if we a have a const declaration with an initializer, we may be
50300b57cec5SDimitry Andric // able to emit it as available_externally to expose it's value to the
50310b57cec5SDimitry Andric // optimizer.
50320b57cec5SDimitry Andric if (Context.getLangOpts().CPlusPlus && GV->hasExternalLinkage() &&
50330b57cec5SDimitry Andric D->getType().isConstQualified() && !GV->hasInitializer() &&
50340b57cec5SDimitry Andric !D->hasDefinition() && D->hasInit() && !D->hasAttr<DLLImportAttr>()) {
50350b57cec5SDimitry Andric const auto *Record =
50360b57cec5SDimitry Andric Context.getBaseElementType(D->getType())->getAsCXXRecordDecl();
50370b57cec5SDimitry Andric bool HasMutableFields = Record && Record->hasMutableFields();
50380b57cec5SDimitry Andric if (!HasMutableFields) {
50390b57cec5SDimitry Andric const VarDecl *InitDecl;
50400b57cec5SDimitry Andric const Expr *InitExpr = D->getAnyInitializer(InitDecl);
50410b57cec5SDimitry Andric if (InitExpr) {
50420b57cec5SDimitry Andric ConstantEmitter emitter(*this);
50430b57cec5SDimitry Andric llvm::Constant *Init = emitter.tryEmitForInitializer(*InitDecl);
50440b57cec5SDimitry Andric if (Init) {
50450b57cec5SDimitry Andric auto *InitType = Init->getType();
50465ffd83dbSDimitry Andric if (GV->getValueType() != InitType) {
50470b57cec5SDimitry Andric // The type of the initializer does not match the definition.
50480b57cec5SDimitry Andric // This happens when an initializer has a different type from
50490b57cec5SDimitry Andric // the type of the global (because of padding at the end of a
50500b57cec5SDimitry Andric // structure for instance).
50510b57cec5SDimitry Andric GV->setName(StringRef());
50520b57cec5SDimitry Andric // Make a new global with the correct type, this is now guaranteed
50530b57cec5SDimitry Andric // to work.
50540b57cec5SDimitry Andric auto *NewGV = cast<llvm::GlobalVariable>(
5055a7dea167SDimitry Andric GetAddrOfGlobalVar(D, InitType, IsForDefinition)
5056a7dea167SDimitry Andric ->stripPointerCasts());
50570b57cec5SDimitry Andric
50580b57cec5SDimitry Andric // Erase the old global, since it is no longer used.
50590b57cec5SDimitry Andric GV->eraseFromParent();
50600b57cec5SDimitry Andric GV = NewGV;
50610b57cec5SDimitry Andric } else {
50620b57cec5SDimitry Andric GV->setInitializer(Init);
50630b57cec5SDimitry Andric GV->setConstant(true);
50640b57cec5SDimitry Andric GV->setLinkage(llvm::GlobalValue::AvailableExternallyLinkage);
50650b57cec5SDimitry Andric }
50660b57cec5SDimitry Andric emitter.finalize(GV);
50670b57cec5SDimitry Andric }
50680b57cec5SDimitry Andric }
50690b57cec5SDimitry Andric }
50700b57cec5SDimitry Andric }
50710b57cec5SDimitry Andric }
50720b57cec5SDimitry Andric
507306c3fb27SDimitry Andric if (D &&
507406c3fb27SDimitry Andric D->isThisDeclarationADefinition(Context) == VarDecl::DeclarationOnly) {
5075480093f4SDimitry Andric getTargetCodeGenInfo().setTargetAttributes(D, GV, *this);
5076fe6060f1SDimitry Andric // External HIP managed variables needed to be recorded for transformation
5077fe6060f1SDimitry Andric // in both device and host compilations.
5078fe6060f1SDimitry Andric if (getLangOpts().CUDA && D && D->hasAttr<HIPManagedAttr>() &&
5079fe6060f1SDimitry Andric D->hasExternalStorage())
5080fe6060f1SDimitry Andric getCUDARuntime().handleVarRegistration(D, *GV);
5081fe6060f1SDimitry Andric }
5082480093f4SDimitry Andric
5083753f127fSDimitry Andric if (D)
5084753f127fSDimitry Andric SanitizerMD->reportGlobal(GV, *D);
5085753f127fSDimitry Andric
50860b57cec5SDimitry Andric LangAS ExpectedAS =
50870b57cec5SDimitry Andric D ? D->getType().getAddressSpace()
50880b57cec5SDimitry Andric : (LangOpts.OpenCL ? LangAS::opencl_global : LangAS::Default);
5089349cc55cSDimitry Andric assert(getContext().getTargetAddressSpace(ExpectedAS) == TargetAS);
5090fe6060f1SDimitry Andric if (DAddrSpace != ExpectedAS) {
5091fe6060f1SDimitry Andric return getTargetCodeGenInfo().performAddrSpaceCast(
50925f757f3fSDimitry Andric *this, GV, DAddrSpace, ExpectedAS,
50935f757f3fSDimitry Andric llvm::PointerType::get(getLLVMContext(), TargetAS));
5094fe6060f1SDimitry Andric }
50950b57cec5SDimitry Andric
50960b57cec5SDimitry Andric return GV;
50970b57cec5SDimitry Andric }
50980b57cec5SDimitry Andric
50990b57cec5SDimitry Andric llvm::Constant *
GetAddrOfGlobal(GlobalDecl GD,ForDefinition_t IsForDefinition)51005ffd83dbSDimitry Andric CodeGenModule::GetAddrOfGlobal(GlobalDecl GD, ForDefinition_t IsForDefinition) {
51010b57cec5SDimitry Andric const Decl *D = GD.getDecl();
51025ffd83dbSDimitry Andric
51030b57cec5SDimitry Andric if (isa<CXXConstructorDecl>(D) || isa<CXXDestructorDecl>(D))
51040b57cec5SDimitry Andric return getAddrOfCXXStructor(GD, /*FnInfo=*/nullptr, /*FnType=*/nullptr,
51050b57cec5SDimitry Andric /*DontDefer=*/false, IsForDefinition);
51065ffd83dbSDimitry Andric
51075ffd83dbSDimitry Andric if (isa<CXXMethodDecl>(D)) {
51085ffd83dbSDimitry Andric auto FInfo =
51095ffd83dbSDimitry Andric &getTypes().arrangeCXXMethodDeclaration(cast<CXXMethodDecl>(D));
51100b57cec5SDimitry Andric auto Ty = getTypes().GetFunctionType(*FInfo);
51110b57cec5SDimitry Andric return GetAddrOfFunction(GD, Ty, /*ForVTable=*/false, /*DontDefer=*/false,
51120b57cec5SDimitry Andric IsForDefinition);
51135ffd83dbSDimitry Andric }
51145ffd83dbSDimitry Andric
51155ffd83dbSDimitry Andric if (isa<FunctionDecl>(D)) {
51160b57cec5SDimitry Andric const CGFunctionInfo &FI = getTypes().arrangeGlobalDeclaration(GD);
51170b57cec5SDimitry Andric llvm::FunctionType *Ty = getTypes().GetFunctionType(FI);
51180b57cec5SDimitry Andric return GetAddrOfFunction(GD, Ty, /*ForVTable=*/false, /*DontDefer=*/false,
51190b57cec5SDimitry Andric IsForDefinition);
51205ffd83dbSDimitry Andric }
51215ffd83dbSDimitry Andric
51225ffd83dbSDimitry Andric return GetAddrOfGlobalVar(cast<VarDecl>(D), /*Ty=*/nullptr, IsForDefinition);
51230b57cec5SDimitry Andric }
51240b57cec5SDimitry Andric
CreateOrReplaceCXXRuntimeVariable(StringRef Name,llvm::Type * Ty,llvm::GlobalValue::LinkageTypes Linkage,llvm::Align Alignment)51250b57cec5SDimitry Andric llvm::GlobalVariable *CodeGenModule::CreateOrReplaceCXXRuntimeVariable(
51260b57cec5SDimitry Andric StringRef Name, llvm::Type *Ty, llvm::GlobalValue::LinkageTypes Linkage,
5127bdd1243dSDimitry Andric llvm::Align Alignment) {
51280b57cec5SDimitry Andric llvm::GlobalVariable *GV = getModule().getNamedGlobal(Name);
51290b57cec5SDimitry Andric llvm::GlobalVariable *OldGV = nullptr;
51300b57cec5SDimitry Andric
51310b57cec5SDimitry Andric if (GV) {
51320b57cec5SDimitry Andric // Check if the variable has the right type.
51335ffd83dbSDimitry Andric if (GV->getValueType() == Ty)
51340b57cec5SDimitry Andric return GV;
51350b57cec5SDimitry Andric
51360b57cec5SDimitry Andric // Because C++ name mangling, the only way we can end up with an already
51370b57cec5SDimitry Andric // existing global with the same name is if it has been declared extern "C".
51380b57cec5SDimitry Andric assert(GV->isDeclaration() && "Declaration has wrong type!");
51390b57cec5SDimitry Andric OldGV = GV;
51400b57cec5SDimitry Andric }
51410b57cec5SDimitry Andric
51420b57cec5SDimitry Andric // Create a new variable.
51430b57cec5SDimitry Andric GV = new llvm::GlobalVariable(getModule(), Ty, /*isConstant=*/true,
51440b57cec5SDimitry Andric Linkage, nullptr, Name);
51450b57cec5SDimitry Andric
51460b57cec5SDimitry Andric if (OldGV) {
51470b57cec5SDimitry Andric // Replace occurrences of the old variable if needed.
51480b57cec5SDimitry Andric GV->takeName(OldGV);
51490b57cec5SDimitry Andric
51500b57cec5SDimitry Andric if (!OldGV->use_empty()) {
51515f757f3fSDimitry Andric OldGV->replaceAllUsesWith(GV);
51520b57cec5SDimitry Andric }
51530b57cec5SDimitry Andric
51540b57cec5SDimitry Andric OldGV->eraseFromParent();
51550b57cec5SDimitry Andric }
51560b57cec5SDimitry Andric
51570b57cec5SDimitry Andric if (supportsCOMDAT() && GV->isWeakForLinker() &&
51580b57cec5SDimitry Andric !GV->hasAvailableExternallyLinkage())
51590b57cec5SDimitry Andric GV->setComdat(TheModule.getOrInsertComdat(GV->getName()));
51600b57cec5SDimitry Andric
5161bdd1243dSDimitry Andric GV->setAlignment(Alignment);
51620b57cec5SDimitry Andric
51630b57cec5SDimitry Andric return GV;
51640b57cec5SDimitry Andric }
51650b57cec5SDimitry Andric
51660b57cec5SDimitry Andric /// GetAddrOfGlobalVar - Return the llvm::Constant for the address of the
51670b57cec5SDimitry Andric /// given global variable. If Ty is non-null and if the global doesn't exist,
51680b57cec5SDimitry Andric /// then it will be created with the specified type instead of whatever the
51690b57cec5SDimitry Andric /// normal requested type would be. If IsForDefinition is true, it is guaranteed
51700b57cec5SDimitry Andric /// that an actual global with type Ty will be returned, not conversion of a
51710b57cec5SDimitry Andric /// variable with the same mangled name but some other type.
GetAddrOfGlobalVar(const VarDecl * D,llvm::Type * Ty,ForDefinition_t IsForDefinition)51720b57cec5SDimitry Andric llvm::Constant *CodeGenModule::GetAddrOfGlobalVar(const VarDecl *D,
51730b57cec5SDimitry Andric llvm::Type *Ty,
51740b57cec5SDimitry Andric ForDefinition_t IsForDefinition) {
51750b57cec5SDimitry Andric assert(D->hasGlobalStorage() && "Not a global variable");
51760b57cec5SDimitry Andric QualType ASTTy = D->getType();
51770b57cec5SDimitry Andric if (!Ty)
51780b57cec5SDimitry Andric Ty = getTypes().ConvertTypeForMem(ASTTy);
51790b57cec5SDimitry Andric
51800b57cec5SDimitry Andric StringRef MangledName = getMangledName(D);
5181349cc55cSDimitry Andric return GetOrCreateLLVMGlobal(MangledName, Ty, ASTTy.getAddressSpace(), D,
5182fe6060f1SDimitry Andric IsForDefinition);
51830b57cec5SDimitry Andric }
51840b57cec5SDimitry Andric
51850b57cec5SDimitry Andric /// CreateRuntimeVariable - Create a new runtime global variable with the
51860b57cec5SDimitry Andric /// specified type and name.
51870b57cec5SDimitry Andric llvm::Constant *
CreateRuntimeVariable(llvm::Type * Ty,StringRef Name)51880b57cec5SDimitry Andric CodeGenModule::CreateRuntimeVariable(llvm::Type *Ty,
51890b57cec5SDimitry Andric StringRef Name) {
5190349cc55cSDimitry Andric LangAS AddrSpace = getContext().getLangOpts().OpenCL ? LangAS::opencl_global
5191349cc55cSDimitry Andric : LangAS::Default;
5192fe6060f1SDimitry Andric auto *Ret = GetOrCreateLLVMGlobal(Name, Ty, AddrSpace, nullptr);
51930b57cec5SDimitry Andric setDSOLocal(cast<llvm::GlobalValue>(Ret->stripPointerCasts()));
51940b57cec5SDimitry Andric return Ret;
51950b57cec5SDimitry Andric }
51960b57cec5SDimitry Andric
EmitTentativeDefinition(const VarDecl * D)51970b57cec5SDimitry Andric void CodeGenModule::EmitTentativeDefinition(const VarDecl *D) {
51980b57cec5SDimitry Andric assert(!D->getInit() && "Cannot emit definite definitions here!");
51990b57cec5SDimitry Andric
52000b57cec5SDimitry Andric StringRef MangledName = getMangledName(D);
52010b57cec5SDimitry Andric llvm::GlobalValue *GV = GetGlobalValue(MangledName);
52020b57cec5SDimitry Andric
52030b57cec5SDimitry Andric // We already have a definition, not declaration, with the same mangled name.
52040b57cec5SDimitry Andric // Emitting of declaration is not required (and actually overwrites emitted
52050b57cec5SDimitry Andric // definition).
52060b57cec5SDimitry Andric if (GV && !GV->isDeclaration())
52070b57cec5SDimitry Andric return;
52080b57cec5SDimitry Andric
52090b57cec5SDimitry Andric // If we have not seen a reference to this variable yet, place it into the
52100b57cec5SDimitry Andric // deferred declarations table to be emitted if needed later.
52110b57cec5SDimitry Andric if (!MustBeEmitted(D) && !GV) {
52120b57cec5SDimitry Andric DeferredDecls[MangledName] = D;
52130b57cec5SDimitry Andric return;
52140b57cec5SDimitry Andric }
52150b57cec5SDimitry Andric
52160b57cec5SDimitry Andric // The tentative definition is the only definition.
52170b57cec5SDimitry Andric EmitGlobalVarDefinition(D);
52180b57cec5SDimitry Andric }
52190b57cec5SDimitry Andric
EmitExternalDeclaration(const DeclaratorDecl * D)52200fca6ea1SDimitry Andric void CodeGenModule::EmitExternalDeclaration(const DeclaratorDecl *D) {
52210fca6ea1SDimitry Andric if (auto const *V = dyn_cast<const VarDecl>(D))
52220fca6ea1SDimitry Andric EmitExternalVarDeclaration(V);
52230fca6ea1SDimitry Andric if (auto const *FD = dyn_cast<const FunctionDecl>(D))
52240fca6ea1SDimitry Andric EmitExternalFunctionDeclaration(FD);
5225480093f4SDimitry Andric }
5226480093f4SDimitry Andric
GetTargetTypeStoreSize(llvm::Type * Ty) const52270b57cec5SDimitry Andric CharUnits CodeGenModule::GetTargetTypeStoreSize(llvm::Type *Ty) const {
52280b57cec5SDimitry Andric return Context.toCharUnitsFromBits(
52290b57cec5SDimitry Andric getDataLayout().getTypeStoreSizeInBits(Ty));
52300b57cec5SDimitry Andric }
52310b57cec5SDimitry Andric
GetGlobalVarAddressSpace(const VarDecl * D)52320b57cec5SDimitry Andric LangAS CodeGenModule::GetGlobalVarAddressSpace(const VarDecl *D) {
52330b57cec5SDimitry Andric if (LangOpts.OpenCL) {
5234349cc55cSDimitry Andric LangAS AS = D ? D->getType().getAddressSpace() : LangAS::opencl_global;
5235349cc55cSDimitry Andric assert(AS == LangAS::opencl_global ||
5236349cc55cSDimitry Andric AS == LangAS::opencl_global_device ||
5237349cc55cSDimitry Andric AS == LangAS::opencl_global_host ||
5238349cc55cSDimitry Andric AS == LangAS::opencl_constant ||
5239349cc55cSDimitry Andric AS == LangAS::opencl_local ||
5240349cc55cSDimitry Andric AS >= LangAS::FirstTargetAddressSpace);
5241349cc55cSDimitry Andric return AS;
52420b57cec5SDimitry Andric }
52430b57cec5SDimitry Andric
5244fe6060f1SDimitry Andric if (LangOpts.SYCLIsDevice &&
5245fe6060f1SDimitry Andric (!D || D->getType().getAddressSpace() == LangAS::Default))
5246fe6060f1SDimitry Andric return LangAS::sycl_global;
5247fe6060f1SDimitry Andric
52480b57cec5SDimitry Andric if (LangOpts.CUDA && LangOpts.CUDAIsDevice) {
524906c3fb27SDimitry Andric if (D) {
525006c3fb27SDimitry Andric if (D->hasAttr<CUDAConstantAttr>())
52510b57cec5SDimitry Andric return LangAS::cuda_constant;
525206c3fb27SDimitry Andric if (D->hasAttr<CUDASharedAttr>())
52530b57cec5SDimitry Andric return LangAS::cuda_shared;
525406c3fb27SDimitry Andric if (D->hasAttr<CUDADeviceAttr>())
52550b57cec5SDimitry Andric return LangAS::cuda_device;
525606c3fb27SDimitry Andric if (D->getType().isConstQualified())
52570b57cec5SDimitry Andric return LangAS::cuda_constant;
525806c3fb27SDimitry Andric }
52590b57cec5SDimitry Andric return LangAS::cuda_device;
52600b57cec5SDimitry Andric }
52610b57cec5SDimitry Andric
52620b57cec5SDimitry Andric if (LangOpts.OpenMP) {
52630b57cec5SDimitry Andric LangAS AS;
52640b57cec5SDimitry Andric if (OpenMPRuntime->hasAllocateAttributeForGlobalVar(D, AS))
52650b57cec5SDimitry Andric return AS;
52660b57cec5SDimitry Andric }
52670b57cec5SDimitry Andric return getTargetCodeGenInfo().getGlobalVarAddressSpace(*this, D);
52680b57cec5SDimitry Andric }
52690b57cec5SDimitry Andric
GetGlobalConstantAddressSpace() const5270fe6060f1SDimitry Andric LangAS CodeGenModule::GetGlobalConstantAddressSpace() const {
52710b57cec5SDimitry Andric // OpenCL v1.2 s6.5.3: a string literal is in the constant address space.
52720b57cec5SDimitry Andric if (LangOpts.OpenCL)
52730b57cec5SDimitry Andric return LangAS::opencl_constant;
5274fe6060f1SDimitry Andric if (LangOpts.SYCLIsDevice)
5275fe6060f1SDimitry Andric return LangAS::sycl_global;
5276d56accc7SDimitry Andric if (LangOpts.HIP && LangOpts.CUDAIsDevice && getTriple().isSPIRV())
5277d56accc7SDimitry Andric // For HIPSPV map literals to cuda_device (maps to CrossWorkGroup in SPIR-V)
5278d56accc7SDimitry Andric // instead of default AS (maps to Generic in SPIR-V). Otherwise, we end up
5279d56accc7SDimitry Andric // with OpVariable instructions with Generic storage class which is not
5280d56accc7SDimitry Andric // allowed (SPIR-V V1.6 s3.42.8). Also, mapping literals to SPIR-V
5281d56accc7SDimitry Andric // UniformConstant storage class is not viable as pointers to it may not be
5282d56accc7SDimitry Andric // casted to Generic pointers which are used to model HIP's "flat" pointers.
5283d56accc7SDimitry Andric return LangAS::cuda_device;
52840b57cec5SDimitry Andric if (auto AS = getTarget().getConstantAddressSpace())
528581ad6265SDimitry Andric return *AS;
52860b57cec5SDimitry Andric return LangAS::Default;
52870b57cec5SDimitry Andric }
52880b57cec5SDimitry Andric
52890b57cec5SDimitry Andric // In address space agnostic languages, string literals are in default address
52900b57cec5SDimitry Andric // space in AST. However, certain targets (e.g. amdgcn) request them to be
52910b57cec5SDimitry Andric // emitted in constant address space in LLVM IR. To be consistent with other
52920b57cec5SDimitry Andric // parts of AST, string literal global variables in constant address space
52930b57cec5SDimitry Andric // need to be casted to default address space before being put into address
52940b57cec5SDimitry Andric // map and referenced by other part of CodeGen.
52950b57cec5SDimitry Andric // In OpenCL, string literals are in constant address space in AST, therefore
52960b57cec5SDimitry Andric // they should not be casted to default address space.
52970b57cec5SDimitry Andric static llvm::Constant *
castStringLiteralToDefaultAddressSpace(CodeGenModule & CGM,llvm::GlobalVariable * GV)52980b57cec5SDimitry Andric castStringLiteralToDefaultAddressSpace(CodeGenModule &CGM,
52990b57cec5SDimitry Andric llvm::GlobalVariable *GV) {
53000b57cec5SDimitry Andric llvm::Constant *Cast = GV;
53010b57cec5SDimitry Andric if (!CGM.getLangOpts().OpenCL) {
5302fe6060f1SDimitry Andric auto AS = CGM.GetGlobalConstantAddressSpace();
53030b57cec5SDimitry Andric if (AS != LangAS::Default)
53040b57cec5SDimitry Andric Cast = CGM.getTargetCodeGenInfo().performAddrSpaceCast(
5305fe6060f1SDimitry Andric CGM, GV, AS, LangAS::Default,
53065f757f3fSDimitry Andric llvm::PointerType::get(
53075f757f3fSDimitry Andric CGM.getLLVMContext(),
53080b57cec5SDimitry Andric CGM.getContext().getTargetAddressSpace(LangAS::Default)));
53090b57cec5SDimitry Andric }
53100b57cec5SDimitry Andric return Cast;
53110b57cec5SDimitry Andric }
53120b57cec5SDimitry Andric
53130b57cec5SDimitry Andric template<typename SomeDecl>
MaybeHandleStaticInExternC(const SomeDecl * D,llvm::GlobalValue * GV)53140b57cec5SDimitry Andric void CodeGenModule::MaybeHandleStaticInExternC(const SomeDecl *D,
53150b57cec5SDimitry Andric llvm::GlobalValue *GV) {
53160b57cec5SDimitry Andric if (!getLangOpts().CPlusPlus)
53170b57cec5SDimitry Andric return;
53180b57cec5SDimitry Andric
53190b57cec5SDimitry Andric // Must have 'used' attribute, or else inline assembly can't rely on
53200b57cec5SDimitry Andric // the name existing.
53210b57cec5SDimitry Andric if (!D->template hasAttr<UsedAttr>())
53220b57cec5SDimitry Andric return;
53230b57cec5SDimitry Andric
53240b57cec5SDimitry Andric // Must have internal linkage and an ordinary name.
53255f757f3fSDimitry Andric if (!D->getIdentifier() || D->getFormalLinkage() != Linkage::Internal)
53260b57cec5SDimitry Andric return;
53270b57cec5SDimitry Andric
53280b57cec5SDimitry Andric // Must be in an extern "C" context. Entities declared directly within
53290b57cec5SDimitry Andric // a record are not extern "C" even if the record is in such a context.
53300b57cec5SDimitry Andric const SomeDecl *First = D->getFirstDecl();
53310b57cec5SDimitry Andric if (First->getDeclContext()->isRecord() || !First->isInExternCContext())
53320b57cec5SDimitry Andric return;
53330b57cec5SDimitry Andric
53340b57cec5SDimitry Andric // OK, this is an internal linkage entity inside an extern "C" linkage
53350b57cec5SDimitry Andric // specification. Make a note of that so we can give it the "expected"
53360b57cec5SDimitry Andric // mangled name if nothing else is using that name.
53370b57cec5SDimitry Andric std::pair<StaticExternCMap::iterator, bool> R =
53380b57cec5SDimitry Andric StaticExternCValues.insert(std::make_pair(D->getIdentifier(), GV));
53390b57cec5SDimitry Andric
53400b57cec5SDimitry Andric // If we have multiple internal linkage entities with the same name
53410b57cec5SDimitry Andric // in extern "C" regions, none of them gets that name.
53420b57cec5SDimitry Andric if (!R.second)
53430b57cec5SDimitry Andric R.first->second = nullptr;
53440b57cec5SDimitry Andric }
53450b57cec5SDimitry Andric
shouldBeInCOMDAT(CodeGenModule & CGM,const Decl & D)53460b57cec5SDimitry Andric static bool shouldBeInCOMDAT(CodeGenModule &CGM, const Decl &D) {
53470b57cec5SDimitry Andric if (!CGM.supportsCOMDAT())
53480b57cec5SDimitry Andric return false;
53490b57cec5SDimitry Andric
53500b57cec5SDimitry Andric if (D.hasAttr<SelectAnyAttr>())
53510b57cec5SDimitry Andric return true;
53520b57cec5SDimitry Andric
53530b57cec5SDimitry Andric GVALinkage Linkage;
53540b57cec5SDimitry Andric if (auto *VD = dyn_cast<VarDecl>(&D))
53550b57cec5SDimitry Andric Linkage = CGM.getContext().GetGVALinkageForVariable(VD);
53560b57cec5SDimitry Andric else
53570b57cec5SDimitry Andric Linkage = CGM.getContext().GetGVALinkageForFunction(cast<FunctionDecl>(&D));
53580b57cec5SDimitry Andric
53590b57cec5SDimitry Andric switch (Linkage) {
53600b57cec5SDimitry Andric case GVA_Internal:
53610b57cec5SDimitry Andric case GVA_AvailableExternally:
53620b57cec5SDimitry Andric case GVA_StrongExternal:
53630b57cec5SDimitry Andric return false;
53640b57cec5SDimitry Andric case GVA_DiscardableODR:
53650b57cec5SDimitry Andric case GVA_StrongODR:
53660b57cec5SDimitry Andric return true;
53670b57cec5SDimitry Andric }
53680b57cec5SDimitry Andric llvm_unreachable("No such linkage");
53690b57cec5SDimitry Andric }
53700b57cec5SDimitry Andric
supportsCOMDAT() const537106c3fb27SDimitry Andric bool CodeGenModule::supportsCOMDAT() const {
537206c3fb27SDimitry Andric return getTriple().supportsCOMDAT();
537306c3fb27SDimitry Andric }
537406c3fb27SDimitry Andric
maybeSetTrivialComdat(const Decl & D,llvm::GlobalObject & GO)53750b57cec5SDimitry Andric void CodeGenModule::maybeSetTrivialComdat(const Decl &D,
53760b57cec5SDimitry Andric llvm::GlobalObject &GO) {
53770b57cec5SDimitry Andric if (!shouldBeInCOMDAT(*this, D))
53780b57cec5SDimitry Andric return;
53790b57cec5SDimitry Andric GO.setComdat(TheModule.getOrInsertComdat(GO.getName()));
53800b57cec5SDimitry Andric }
53810b57cec5SDimitry Andric
getABIInfo()53825deeebd8SDimitry Andric const ABIInfo &CodeGenModule::getABIInfo() {
53835deeebd8SDimitry Andric return getTargetCodeGenInfo().getABIInfo();
53845deeebd8SDimitry Andric }
53855deeebd8SDimitry Andric
53860b57cec5SDimitry Andric /// Pass IsTentative as true if you want to create a tentative definition.
EmitGlobalVarDefinition(const VarDecl * D,bool IsTentative)53870b57cec5SDimitry Andric void CodeGenModule::EmitGlobalVarDefinition(const VarDecl *D,
53880b57cec5SDimitry Andric bool IsTentative) {
53890b57cec5SDimitry Andric // OpenCL global variables of sampler type are translated to function calls,
53900b57cec5SDimitry Andric // therefore no need to be translated.
53910b57cec5SDimitry Andric QualType ASTTy = D->getType();
53920b57cec5SDimitry Andric if (getLangOpts().OpenCL && ASTTy->isSamplerT())
53930b57cec5SDimitry Andric return;
53940b57cec5SDimitry Andric
53950b57cec5SDimitry Andric // If this is OpenMP device, check if it is legal to emit this global
53960b57cec5SDimitry Andric // normally.
539706c3fb27SDimitry Andric if (LangOpts.OpenMPIsTargetDevice && OpenMPRuntime &&
53980b57cec5SDimitry Andric OpenMPRuntime->emitTargetGlobalVariable(D))
53990b57cec5SDimitry Andric return;
54000b57cec5SDimitry Andric
5401fe6060f1SDimitry Andric llvm::TrackingVH<llvm::Constant> Init;
54020b57cec5SDimitry Andric bool NeedsGlobalCtor = false;
5403bdd1243dSDimitry Andric // Whether the definition of the variable is available externally.
5404bdd1243dSDimitry Andric // If yes, we shouldn't emit the GloablCtor and GlobalDtor for the variable
5405bdd1243dSDimitry Andric // since this is the job for its original source.
5406bdd1243dSDimitry Andric bool IsDefinitionAvailableExternally =
5407bdd1243dSDimitry Andric getContext().GetGVALinkageForVariable(D) == GVA_AvailableExternally;
5408a7dea167SDimitry Andric bool NeedsGlobalDtor =
5409bdd1243dSDimitry Andric !IsDefinitionAvailableExternally &&
5410a7dea167SDimitry Andric D->needsDestruction(getContext()) == QualType::DK_cxx_destructor;
54110b57cec5SDimitry Andric
54120fca6ea1SDimitry Andric // It is helpless to emit the definition for an available_externally variable
54130fca6ea1SDimitry Andric // which can't be marked as const.
54140fca6ea1SDimitry Andric // We don't need to check if it needs global ctor or dtor. See the above
54150fca6ea1SDimitry Andric // comment for ideas.
54160fca6ea1SDimitry Andric if (IsDefinitionAvailableExternally &&
54170fca6ea1SDimitry Andric (!D->hasConstantInitialization() ||
54180fca6ea1SDimitry Andric // TODO: Update this when we have interface to check constexpr
54190fca6ea1SDimitry Andric // destructor.
54200fca6ea1SDimitry Andric D->needsDestruction(getContext()) ||
54210fca6ea1SDimitry Andric !D->getType().isConstantStorage(getContext(), true, true)))
54220fca6ea1SDimitry Andric return;
54230fca6ea1SDimitry Andric
54240b57cec5SDimitry Andric const VarDecl *InitDecl;
54250b57cec5SDimitry Andric const Expr *InitExpr = D->getAnyInitializer(InitDecl);
54260b57cec5SDimitry Andric
5427bdd1243dSDimitry Andric std::optional<ConstantEmitter> emitter;
54280b57cec5SDimitry Andric
54290b57cec5SDimitry Andric // CUDA E.2.4.1 "__shared__ variables cannot have an initialization
54300b57cec5SDimitry Andric // as part of their declaration." Sema has already checked for
54310b57cec5SDimitry Andric // error cases, so we just need to set Init to UndefValue.
54320b57cec5SDimitry Andric bool IsCUDASharedVar =
54330b57cec5SDimitry Andric getLangOpts().CUDAIsDevice && D->hasAttr<CUDASharedAttr>();
54340b57cec5SDimitry Andric // Shadows of initialized device-side global variables are also left
54350b57cec5SDimitry Andric // undefined.
5436fe6060f1SDimitry Andric // Managed Variables should be initialized on both host side and device side.
54370b57cec5SDimitry Andric bool IsCUDAShadowVar =
5438e8d8bef9SDimitry Andric !getLangOpts().CUDAIsDevice && !D->hasAttr<HIPManagedAttr>() &&
54390b57cec5SDimitry Andric (D->hasAttr<CUDAConstantAttr>() || D->hasAttr<CUDADeviceAttr>() ||
54400b57cec5SDimitry Andric D->hasAttr<CUDASharedAttr>());
54415ffd83dbSDimitry Andric bool IsCUDADeviceShadowVar =
5442fe6060f1SDimitry Andric getLangOpts().CUDAIsDevice && !D->hasAttr<HIPManagedAttr>() &&
54435ffd83dbSDimitry Andric (D->getType()->isCUDADeviceBuiltinSurfaceType() ||
5444fe6060f1SDimitry Andric D->getType()->isCUDADeviceBuiltinTextureType());
54450b57cec5SDimitry Andric if (getLangOpts().CUDA &&
54465ffd83dbSDimitry Andric (IsCUDASharedVar || IsCUDAShadowVar || IsCUDADeviceShadowVar))
5447fe6060f1SDimitry Andric Init = llvm::UndefValue::get(getTypes().ConvertTypeForMem(ASTTy));
54485ffd83dbSDimitry Andric else if (D->hasAttr<LoaderUninitializedAttr>())
5449fe6060f1SDimitry Andric Init = llvm::UndefValue::get(getTypes().ConvertTypeForMem(ASTTy));
54500b57cec5SDimitry Andric else if (!InitExpr) {
54510b57cec5SDimitry Andric // This is a tentative definition; tentative definitions are
54520b57cec5SDimitry Andric // implicitly initialized with { 0 }.
54530b57cec5SDimitry Andric //
54540b57cec5SDimitry Andric // Note that tentative definitions are only emitted at the end of
54550b57cec5SDimitry Andric // a translation unit, so they should never have incomplete
54560b57cec5SDimitry Andric // type. In addition, EmitTentativeDefinition makes sure that we
54570b57cec5SDimitry Andric // never attempt to emit a tentative definition if a real one
54580b57cec5SDimitry Andric // exists. A use may still exists, however, so we still may need
54590b57cec5SDimitry Andric // to do a RAUW.
54600b57cec5SDimitry Andric assert(!ASTTy->isIncompleteType() && "Unexpected incomplete type");
54610b57cec5SDimitry Andric Init = EmitNullConstant(D->getType());
54620b57cec5SDimitry Andric } else {
54630b57cec5SDimitry Andric initializedGlobalDecl = GlobalDecl(D);
54640b57cec5SDimitry Andric emitter.emplace(*this);
5465fe6060f1SDimitry Andric llvm::Constant *Initializer = emitter->tryEmitForInitializer(*InitDecl);
5466fe6060f1SDimitry Andric if (!Initializer) {
54670b57cec5SDimitry Andric QualType T = InitExpr->getType();
54680b57cec5SDimitry Andric if (D->getType()->isReferenceType())
54690b57cec5SDimitry Andric T = D->getType();
54700b57cec5SDimitry Andric
54710b57cec5SDimitry Andric if (getLangOpts().CPlusPlus) {
547281ad6265SDimitry Andric if (InitDecl->hasFlexibleArrayInit(getContext()))
547381ad6265SDimitry Andric ErrorUnsupported(D, "flexible array initializer");
54740b57cec5SDimitry Andric Init = EmitNullConstant(T);
5475bdd1243dSDimitry Andric
5476bdd1243dSDimitry Andric if (!IsDefinitionAvailableExternally)
54770b57cec5SDimitry Andric NeedsGlobalCtor = true;
54780b57cec5SDimitry Andric } else {
54790b57cec5SDimitry Andric ErrorUnsupported(D, "static initializer");
54800b57cec5SDimitry Andric Init = llvm::UndefValue::get(getTypes().ConvertType(T));
54810b57cec5SDimitry Andric }
54820b57cec5SDimitry Andric } else {
5483fe6060f1SDimitry Andric Init = Initializer;
54840b57cec5SDimitry Andric // We don't need an initializer, so remove the entry for the delayed
54850b57cec5SDimitry Andric // initializer position (just in case this entry was delayed) if we
54860b57cec5SDimitry Andric // also don't need to register a destructor.
54870b57cec5SDimitry Andric if (getLangOpts().CPlusPlus && !NeedsGlobalDtor)
54880b57cec5SDimitry Andric DelayedCXXInitPosition.erase(D);
548981ad6265SDimitry Andric
549081ad6265SDimitry Andric #ifndef NDEBUG
549181ad6265SDimitry Andric CharUnits VarSize = getContext().getTypeSizeInChars(ASTTy) +
549281ad6265SDimitry Andric InitDecl->getFlexibleArrayInitChars(getContext());
549381ad6265SDimitry Andric CharUnits CstSize = CharUnits::fromQuantity(
549481ad6265SDimitry Andric getDataLayout().getTypeAllocSize(Init->getType()));
549581ad6265SDimitry Andric assert(VarSize == CstSize && "Emitted constant has unexpected size");
549681ad6265SDimitry Andric #endif
54970b57cec5SDimitry Andric }
54980b57cec5SDimitry Andric }
54990b57cec5SDimitry Andric
55000b57cec5SDimitry Andric llvm::Type* InitType = Init->getType();
55010b57cec5SDimitry Andric llvm::Constant *Entry =
55020b57cec5SDimitry Andric GetAddrOfGlobalVar(D, InitType, ForDefinition_t(!IsTentative));
55030b57cec5SDimitry Andric
5504a7dea167SDimitry Andric // Strip off pointer casts if we got them.
5505a7dea167SDimitry Andric Entry = Entry->stripPointerCasts();
55060b57cec5SDimitry Andric
55070b57cec5SDimitry Andric // Entry is now either a Function or GlobalVariable.
55080b57cec5SDimitry Andric auto *GV = dyn_cast<llvm::GlobalVariable>(Entry);
55090b57cec5SDimitry Andric
55100b57cec5SDimitry Andric // We have a definition after a declaration with the wrong type.
55110b57cec5SDimitry Andric // We must make a new GlobalVariable* and update everything that used OldGV
55120b57cec5SDimitry Andric // (a declaration or tentative definition) with the new GlobalVariable*
55130b57cec5SDimitry Andric // (which will be a definition).
55140b57cec5SDimitry Andric //
55150b57cec5SDimitry Andric // This happens if there is a prototype for a global (e.g.
55160b57cec5SDimitry Andric // "extern int x[];") and then a definition of a different type (e.g.
55170b57cec5SDimitry Andric // "int x[10];"). This also happens when an initializer has a different type
55180b57cec5SDimitry Andric // from the type of the global (this happens with unions).
55195ffd83dbSDimitry Andric if (!GV || GV->getValueType() != InitType ||
55200b57cec5SDimitry Andric GV->getType()->getAddressSpace() !=
55210b57cec5SDimitry Andric getContext().getTargetAddressSpace(GetGlobalVarAddressSpace(D))) {
55220b57cec5SDimitry Andric
55230b57cec5SDimitry Andric // Move the old entry aside so that we'll create a new one.
55240b57cec5SDimitry Andric Entry->setName(StringRef());
55250b57cec5SDimitry Andric
55260b57cec5SDimitry Andric // Make a new global with the correct type, this is now guaranteed to work.
55270b57cec5SDimitry Andric GV = cast<llvm::GlobalVariable>(
5528a7dea167SDimitry Andric GetAddrOfGlobalVar(D, InitType, ForDefinition_t(!IsTentative))
5529a7dea167SDimitry Andric ->stripPointerCasts());
55300b57cec5SDimitry Andric
55310b57cec5SDimitry Andric // Replace all uses of the old global with the new global
55320b57cec5SDimitry Andric llvm::Constant *NewPtrForOldDecl =
5533fe6060f1SDimitry Andric llvm::ConstantExpr::getPointerBitCastOrAddrSpaceCast(GV,
5534fe6060f1SDimitry Andric Entry->getType());
55350b57cec5SDimitry Andric Entry->replaceAllUsesWith(NewPtrForOldDecl);
55360b57cec5SDimitry Andric
55370b57cec5SDimitry Andric // Erase the old global, since it is no longer used.
55380b57cec5SDimitry Andric cast<llvm::GlobalValue>(Entry)->eraseFromParent();
55390b57cec5SDimitry Andric }
55400b57cec5SDimitry Andric
55410b57cec5SDimitry Andric MaybeHandleStaticInExternC(D, GV);
55420b57cec5SDimitry Andric
55430b57cec5SDimitry Andric if (D->hasAttr<AnnotateAttr>())
55440b57cec5SDimitry Andric AddGlobalAnnotations(D, GV);
55450b57cec5SDimitry Andric
55460b57cec5SDimitry Andric // Set the llvm linkage type as appropriate.
55478a4dda33SDimitry Andric llvm::GlobalValue::LinkageTypes Linkage = getLLVMLinkageVarDefinition(D);
55480b57cec5SDimitry Andric
55490b57cec5SDimitry Andric // CUDA B.2.1 "The __device__ qualifier declares a variable that resides on
55500b57cec5SDimitry Andric // the device. [...]"
55510b57cec5SDimitry Andric // CUDA B.2.2 "The __constant__ qualifier, optionally used together with
55520b57cec5SDimitry Andric // __device__, declares a variable that: [...]
55530b57cec5SDimitry Andric // Is accessible from all the threads within the grid and from the host
55540b57cec5SDimitry Andric // through the runtime library (cudaGetSymbolAddress() / cudaGetSymbolSize()
55550b57cec5SDimitry Andric // / cudaMemcpyToSymbol() / cudaMemcpyFromSymbol())."
555606c3fb27SDimitry Andric if (LangOpts.CUDA) {
55570b57cec5SDimitry Andric if (LangOpts.CUDAIsDevice) {
55580b57cec5SDimitry Andric if (Linkage != llvm::GlobalValue::InternalLinkage &&
5559349cc55cSDimitry Andric (D->hasAttr<CUDADeviceAttr>() || D->hasAttr<CUDAConstantAttr>() ||
5560349cc55cSDimitry Andric D->getType()->isCUDADeviceBuiltinSurfaceType() ||
5561349cc55cSDimitry Andric D->getType()->isCUDADeviceBuiltinTextureType()))
55620b57cec5SDimitry Andric GV->setExternallyInitialized(true);
55630b57cec5SDimitry Andric } else {
5564fe6060f1SDimitry Andric getCUDARuntime().internalizeDeviceSideVar(D, Linkage);
55655ffd83dbSDimitry Andric }
5566fe6060f1SDimitry Andric getCUDARuntime().handleVarRegistration(D, *GV);
55670b57cec5SDimitry Andric }
55680b57cec5SDimitry Andric
55690b57cec5SDimitry Andric GV->setInitializer(Init);
55705ffd83dbSDimitry Andric if (emitter)
55715ffd83dbSDimitry Andric emitter->finalize(GV);
55720b57cec5SDimitry Andric
55730b57cec5SDimitry Andric // If it is safe to mark the global 'constant', do so now.
55740b57cec5SDimitry Andric GV->setConstant(!NeedsGlobalCtor && !NeedsGlobalDtor &&
55755f757f3fSDimitry Andric D->getType().isConstantStorage(getContext(), true, true));
55760b57cec5SDimitry Andric
55770b57cec5SDimitry Andric // If it is in a read-only section, mark it 'constant'.
55780b57cec5SDimitry Andric if (const SectionAttr *SA = D->getAttr<SectionAttr>()) {
55790b57cec5SDimitry Andric const ASTContext::SectionInfo &SI = Context.SectionInfos[SA->getName()];
55800b57cec5SDimitry Andric if ((SI.SectionFlags & ASTContext::PSF_Write) == 0)
55810b57cec5SDimitry Andric GV->setConstant(true);
55820b57cec5SDimitry Andric }
55830b57cec5SDimitry Andric
558481ad6265SDimitry Andric CharUnits AlignVal = getContext().getDeclAlign(D);
558581ad6265SDimitry Andric // Check for alignment specifed in an 'omp allocate' directive.
5586bdd1243dSDimitry Andric if (std::optional<CharUnits> AlignValFromAllocate =
558781ad6265SDimitry Andric getOMPAllocateAlignment(D))
558881ad6265SDimitry Andric AlignVal = *AlignValFromAllocate;
558981ad6265SDimitry Andric GV->setAlignment(AlignVal.getAsAlign());
55900b57cec5SDimitry Andric
55915ffd83dbSDimitry Andric // On Darwin, unlike other Itanium C++ ABI platforms, the thread-wrapper
55925ffd83dbSDimitry Andric // function is only defined alongside the variable, not also alongside
55935ffd83dbSDimitry Andric // callers. Normally, all accesses to a thread_local go through the
55945ffd83dbSDimitry Andric // thread-wrapper in order to ensure initialization has occurred, underlying
55955ffd83dbSDimitry Andric // variable will never be used other than the thread-wrapper, so it can be
55965ffd83dbSDimitry Andric // converted to internal linkage.
55975ffd83dbSDimitry Andric //
55985ffd83dbSDimitry Andric // However, if the variable has the 'constinit' attribute, it _can_ be
55995ffd83dbSDimitry Andric // referenced directly, without calling the thread-wrapper, so the linkage
56005ffd83dbSDimitry Andric // must not be changed.
56015ffd83dbSDimitry Andric //
56025ffd83dbSDimitry Andric // Additionally, if the variable isn't plain external linkage, e.g. if it's
56035ffd83dbSDimitry Andric // weak or linkonce, the de-duplication semantics are important to preserve,
56045ffd83dbSDimitry Andric // so we don't change the linkage.
56055ffd83dbSDimitry Andric if (D->getTLSKind() == VarDecl::TLS_Dynamic &&
56065ffd83dbSDimitry Andric Linkage == llvm::GlobalValue::ExternalLinkage &&
56070b57cec5SDimitry Andric Context.getTargetInfo().getTriple().isOSDarwin() &&
56085ffd83dbSDimitry Andric !D->hasAttr<ConstInitAttr>())
56090b57cec5SDimitry Andric Linkage = llvm::GlobalValue::InternalLinkage;
56100b57cec5SDimitry Andric
56110b57cec5SDimitry Andric GV->setLinkage(Linkage);
56120b57cec5SDimitry Andric if (D->hasAttr<DLLImportAttr>())
56130b57cec5SDimitry Andric GV->setDLLStorageClass(llvm::GlobalVariable::DLLImportStorageClass);
56140b57cec5SDimitry Andric else if (D->hasAttr<DLLExportAttr>())
56150b57cec5SDimitry Andric GV->setDLLStorageClass(llvm::GlobalVariable::DLLExportStorageClass);
56160b57cec5SDimitry Andric else
56170b57cec5SDimitry Andric GV->setDLLStorageClass(llvm::GlobalVariable::DefaultStorageClass);
56180b57cec5SDimitry Andric
56190b57cec5SDimitry Andric if (Linkage == llvm::GlobalVariable::CommonLinkage) {
56200b57cec5SDimitry Andric // common vars aren't constant even if declared const.
56210b57cec5SDimitry Andric GV->setConstant(false);
56220b57cec5SDimitry Andric // Tentative definition of global variables may be initialized with
56230b57cec5SDimitry Andric // non-zero null pointers. In this case they should have weak linkage
56240b57cec5SDimitry Andric // since common linkage must have zero initializer and must not have
56250b57cec5SDimitry Andric // explicit section therefore cannot have non-zero initial value.
56260b57cec5SDimitry Andric if (!GV->getInitializer()->isNullValue())
56270b57cec5SDimitry Andric GV->setLinkage(llvm::GlobalVariable::WeakAnyLinkage);
56280b57cec5SDimitry Andric }
56290b57cec5SDimitry Andric
56300b57cec5SDimitry Andric setNonAliasAttributes(D, GV);
56310b57cec5SDimitry Andric
56320b57cec5SDimitry Andric if (D->getTLSKind() && !GV->isThreadLocal()) {
56330b57cec5SDimitry Andric if (D->getTLSKind() == VarDecl::TLS_Dynamic)
56340b57cec5SDimitry Andric CXXThreadLocals.push_back(D);
56350b57cec5SDimitry Andric setTLSMode(GV, *D);
56360b57cec5SDimitry Andric }
56370b57cec5SDimitry Andric
56380b57cec5SDimitry Andric maybeSetTrivialComdat(*D, *GV);
56390b57cec5SDimitry Andric
56400b57cec5SDimitry Andric // Emit the initializer function if necessary.
56410b57cec5SDimitry Andric if (NeedsGlobalCtor || NeedsGlobalDtor)
56420b57cec5SDimitry Andric EmitCXXGlobalVarDeclInitFunc(D, GV, NeedsGlobalCtor);
56430b57cec5SDimitry Andric
564481ad6265SDimitry Andric SanitizerMD->reportGlobal(GV, *D, NeedsGlobalCtor);
56450b57cec5SDimitry Andric
56460b57cec5SDimitry Andric // Emit global variable debug information.
56470b57cec5SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
5648480093f4SDimitry Andric if (getCodeGenOpts().hasReducedDebugInfo())
56490b57cec5SDimitry Andric DI->EmitGlobalVariable(GV, D);
56500b57cec5SDimitry Andric }
56510b57cec5SDimitry Andric
EmitExternalVarDeclaration(const VarDecl * D)5652480093f4SDimitry Andric void CodeGenModule::EmitExternalVarDeclaration(const VarDecl *D) {
5653480093f4SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
5654480093f4SDimitry Andric if (getCodeGenOpts().hasReducedDebugInfo()) {
5655480093f4SDimitry Andric QualType ASTTy = D->getType();
5656480093f4SDimitry Andric llvm::Type *Ty = getTypes().ConvertTypeForMem(D->getType());
5657349cc55cSDimitry Andric llvm::Constant *GV =
5658349cc55cSDimitry Andric GetOrCreateLLVMGlobal(D->getName(), Ty, ASTTy.getAddressSpace(), D);
5659480093f4SDimitry Andric DI->EmitExternalVariable(
5660480093f4SDimitry Andric cast<llvm::GlobalVariable>(GV->stripPointerCasts()), D);
5661480093f4SDimitry Andric }
5662480093f4SDimitry Andric }
5663480093f4SDimitry Andric
EmitExternalFunctionDeclaration(const FunctionDecl * FD)56640fca6ea1SDimitry Andric void CodeGenModule::EmitExternalFunctionDeclaration(const FunctionDecl *FD) {
56650fca6ea1SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
56660fca6ea1SDimitry Andric if (getCodeGenOpts().hasReducedDebugInfo()) {
56670fca6ea1SDimitry Andric auto *Ty = getTypes().ConvertType(FD->getType());
56680fca6ea1SDimitry Andric StringRef MangledName = getMangledName(FD);
56690fca6ea1SDimitry Andric auto *Fn = dyn_cast<llvm::Function>(
56700fca6ea1SDimitry Andric GetOrCreateLLVMFunction(MangledName, Ty, FD, /* ForVTable */ false));
56710fca6ea1SDimitry Andric if (!Fn->getSubprogram())
56720fca6ea1SDimitry Andric DI->EmitFunctionDecl(FD, FD->getLocation(), FD->getType(), Fn);
56730fca6ea1SDimitry Andric }
56740fca6ea1SDimitry Andric }
56750fca6ea1SDimitry Andric
isVarDeclStrongDefinition(const ASTContext & Context,CodeGenModule & CGM,const VarDecl * D,bool NoCommon)56760b57cec5SDimitry Andric static bool isVarDeclStrongDefinition(const ASTContext &Context,
56770b57cec5SDimitry Andric CodeGenModule &CGM, const VarDecl *D,
56780b57cec5SDimitry Andric bool NoCommon) {
56790b57cec5SDimitry Andric // Don't give variables common linkage if -fno-common was specified unless it
56800b57cec5SDimitry Andric // was overridden by a NoCommon attribute.
56810b57cec5SDimitry Andric if ((NoCommon || D->hasAttr<NoCommonAttr>()) && !D->hasAttr<CommonAttr>())
56820b57cec5SDimitry Andric return true;
56830b57cec5SDimitry Andric
56840b57cec5SDimitry Andric // C11 6.9.2/2:
56850b57cec5SDimitry Andric // A declaration of an identifier for an object that has file scope without
56860b57cec5SDimitry Andric // an initializer, and without a storage-class specifier or with the
56870b57cec5SDimitry Andric // storage-class specifier static, constitutes a tentative definition.
56880b57cec5SDimitry Andric if (D->getInit() || D->hasExternalStorage())
56890b57cec5SDimitry Andric return true;
56900b57cec5SDimitry Andric
56910b57cec5SDimitry Andric // A variable cannot be both common and exist in a section.
56920b57cec5SDimitry Andric if (D->hasAttr<SectionAttr>())
56930b57cec5SDimitry Andric return true;
56940b57cec5SDimitry Andric
56950b57cec5SDimitry Andric // A variable cannot be both common and exist in a section.
56960b57cec5SDimitry Andric // We don't try to determine which is the right section in the front-end.
56970b57cec5SDimitry Andric // If no specialized section name is applicable, it will resort to default.
56980b57cec5SDimitry Andric if (D->hasAttr<PragmaClangBSSSectionAttr>() ||
56990b57cec5SDimitry Andric D->hasAttr<PragmaClangDataSectionAttr>() ||
5700a7dea167SDimitry Andric D->hasAttr<PragmaClangRelroSectionAttr>() ||
57010b57cec5SDimitry Andric D->hasAttr<PragmaClangRodataSectionAttr>())
57020b57cec5SDimitry Andric return true;
57030b57cec5SDimitry Andric
57040b57cec5SDimitry Andric // Thread local vars aren't considered common linkage.
57050b57cec5SDimitry Andric if (D->getTLSKind())
57060b57cec5SDimitry Andric return true;
57070b57cec5SDimitry Andric
57080b57cec5SDimitry Andric // Tentative definitions marked with WeakImportAttr are true definitions.
57090b57cec5SDimitry Andric if (D->hasAttr<WeakImportAttr>())
57100b57cec5SDimitry Andric return true;
57110b57cec5SDimitry Andric
57120b57cec5SDimitry Andric // A variable cannot be both common and exist in a comdat.
57130b57cec5SDimitry Andric if (shouldBeInCOMDAT(CGM, *D))
57140b57cec5SDimitry Andric return true;
57150b57cec5SDimitry Andric
57160b57cec5SDimitry Andric // Declarations with a required alignment do not have common linkage in MSVC
57170b57cec5SDimitry Andric // mode.
57180b57cec5SDimitry Andric if (Context.getTargetInfo().getCXXABI().isMicrosoft()) {
57190b57cec5SDimitry Andric if (D->hasAttr<AlignedAttr>())
57200b57cec5SDimitry Andric return true;
57210b57cec5SDimitry Andric QualType VarType = D->getType();
57220b57cec5SDimitry Andric if (Context.isAlignmentRequired(VarType))
57230b57cec5SDimitry Andric return true;
57240b57cec5SDimitry Andric
57250b57cec5SDimitry Andric if (const auto *RT = VarType->getAs<RecordType>()) {
57260b57cec5SDimitry Andric const RecordDecl *RD = RT->getDecl();
57270b57cec5SDimitry Andric for (const FieldDecl *FD : RD->fields()) {
57280b57cec5SDimitry Andric if (FD->isBitField())
57290b57cec5SDimitry Andric continue;
57300b57cec5SDimitry Andric if (FD->hasAttr<AlignedAttr>())
57310b57cec5SDimitry Andric return true;
57320b57cec5SDimitry Andric if (Context.isAlignmentRequired(FD->getType()))
57330b57cec5SDimitry Andric return true;
57340b57cec5SDimitry Andric }
57350b57cec5SDimitry Andric }
57360b57cec5SDimitry Andric }
57370b57cec5SDimitry Andric
57380b57cec5SDimitry Andric // Microsoft's link.exe doesn't support alignments greater than 32 bytes for
57390b57cec5SDimitry Andric // common symbols, so symbols with greater alignment requirements cannot be
57400b57cec5SDimitry Andric // common.
57410b57cec5SDimitry Andric // Other COFF linkers (ld.bfd and LLD) support arbitrary power-of-two
57420b57cec5SDimitry Andric // alignments for common symbols via the aligncomm directive, so this
57430b57cec5SDimitry Andric // restriction only applies to MSVC environments.
57440b57cec5SDimitry Andric if (Context.getTargetInfo().getTriple().isKnownWindowsMSVCEnvironment() &&
57450b57cec5SDimitry Andric Context.getTypeAlignIfKnown(D->getType()) >
57460b57cec5SDimitry Andric Context.toBits(CharUnits::fromQuantity(32)))
57470b57cec5SDimitry Andric return true;
57480b57cec5SDimitry Andric
57490b57cec5SDimitry Andric return false;
57500b57cec5SDimitry Andric }
57510b57cec5SDimitry Andric
57528a4dda33SDimitry Andric llvm::GlobalValue::LinkageTypes
getLLVMLinkageForDeclarator(const DeclaratorDecl * D,GVALinkage Linkage)57538a4dda33SDimitry Andric CodeGenModule::getLLVMLinkageForDeclarator(const DeclaratorDecl *D,
57548a4dda33SDimitry Andric GVALinkage Linkage) {
57550b57cec5SDimitry Andric if (Linkage == GVA_Internal)
57560b57cec5SDimitry Andric return llvm::Function::InternalLinkage;
57570b57cec5SDimitry Andric
575881ad6265SDimitry Andric if (D->hasAttr<WeakAttr>())
57590b57cec5SDimitry Andric return llvm::GlobalVariable::WeakAnyLinkage;
57600b57cec5SDimitry Andric
57610b57cec5SDimitry Andric if (const auto *FD = D->getAsFunction())
57620b57cec5SDimitry Andric if (FD->isMultiVersion() && Linkage == GVA_AvailableExternally)
57630b57cec5SDimitry Andric return llvm::GlobalVariable::LinkOnceAnyLinkage;
57640b57cec5SDimitry Andric
57650b57cec5SDimitry Andric // We are guaranteed to have a strong definition somewhere else,
57660b57cec5SDimitry Andric // so we can use available_externally linkage.
57670b57cec5SDimitry Andric if (Linkage == GVA_AvailableExternally)
57680b57cec5SDimitry Andric return llvm::GlobalValue::AvailableExternallyLinkage;
57690b57cec5SDimitry Andric
57700b57cec5SDimitry Andric // Note that Apple's kernel linker doesn't support symbol
57710b57cec5SDimitry Andric // coalescing, so we need to avoid linkonce and weak linkages there.
57720b57cec5SDimitry Andric // Normally, this means we just map to internal, but for explicit
57730b57cec5SDimitry Andric // instantiations we'll map to external.
57740b57cec5SDimitry Andric
57750b57cec5SDimitry Andric // In C++, the compiler has to emit a definition in every translation unit
57760b57cec5SDimitry Andric // that references the function. We should use linkonce_odr because
57770b57cec5SDimitry Andric // a) if all references in this translation unit are optimized away, we
57780b57cec5SDimitry Andric // don't need to codegen it. b) if the function persists, it needs to be
57790b57cec5SDimitry Andric // merged with other definitions. c) C++ has the ODR, so we know the
57800b57cec5SDimitry Andric // definition is dependable.
57810b57cec5SDimitry Andric if (Linkage == GVA_DiscardableODR)
57820b57cec5SDimitry Andric return !Context.getLangOpts().AppleKext ? llvm::Function::LinkOnceODRLinkage
57830b57cec5SDimitry Andric : llvm::Function::InternalLinkage;
57840b57cec5SDimitry Andric
57850b57cec5SDimitry Andric // An explicit instantiation of a template has weak linkage, since
57860b57cec5SDimitry Andric // explicit instantiations can occur in multiple translation units
57870b57cec5SDimitry Andric // and must all be equivalent. However, we are not allowed to
57880b57cec5SDimitry Andric // throw away these explicit instantiations.
57890b57cec5SDimitry Andric //
5790e8d8bef9SDimitry Andric // CUDA/HIP: For -fno-gpu-rdc case, device code is limited to one TU,
57910b57cec5SDimitry Andric // so say that CUDA templates are either external (for kernels) or internal.
5792e8d8bef9SDimitry Andric // This lets llvm perform aggressive inter-procedural optimizations. For
5793e8d8bef9SDimitry Andric // -fgpu-rdc case, device function calls across multiple TU's are allowed,
5794e8d8bef9SDimitry Andric // therefore we need to follow the normal linkage paradigm.
57950b57cec5SDimitry Andric if (Linkage == GVA_StrongODR) {
5796e8d8bef9SDimitry Andric if (getLangOpts().AppleKext)
57970b57cec5SDimitry Andric return llvm::Function::ExternalLinkage;
5798e8d8bef9SDimitry Andric if (getLangOpts().CUDA && getLangOpts().CUDAIsDevice &&
5799e8d8bef9SDimitry Andric !getLangOpts().GPURelocatableDeviceCode)
58000b57cec5SDimitry Andric return D->hasAttr<CUDAGlobalAttr>() ? llvm::Function::ExternalLinkage
58010b57cec5SDimitry Andric : llvm::Function::InternalLinkage;
58020b57cec5SDimitry Andric return llvm::Function::WeakODRLinkage;
58030b57cec5SDimitry Andric }
58040b57cec5SDimitry Andric
58050b57cec5SDimitry Andric // C++ doesn't have tentative definitions and thus cannot have common
58060b57cec5SDimitry Andric // linkage.
58070b57cec5SDimitry Andric if (!getLangOpts().CPlusPlus && isa<VarDecl>(D) &&
58080b57cec5SDimitry Andric !isVarDeclStrongDefinition(Context, *this, cast<VarDecl>(D),
58090b57cec5SDimitry Andric CodeGenOpts.NoCommon))
58100b57cec5SDimitry Andric return llvm::GlobalVariable::CommonLinkage;
58110b57cec5SDimitry Andric
58120b57cec5SDimitry Andric // selectany symbols are externally visible, so use weak instead of
58130b57cec5SDimitry Andric // linkonce. MSVC optimizes away references to const selectany globals, so
58140b57cec5SDimitry Andric // all definitions should be the same and ODR linkage should be used.
58150b57cec5SDimitry Andric // http://msdn.microsoft.com/en-us/library/5tkz6s71.aspx
58160b57cec5SDimitry Andric if (D->hasAttr<SelectAnyAttr>())
58170b57cec5SDimitry Andric return llvm::GlobalVariable::WeakODRLinkage;
58180b57cec5SDimitry Andric
58190b57cec5SDimitry Andric // Otherwise, we have strong external linkage.
58200b57cec5SDimitry Andric assert(Linkage == GVA_StrongExternal);
58210b57cec5SDimitry Andric return llvm::GlobalVariable::ExternalLinkage;
58220b57cec5SDimitry Andric }
58230b57cec5SDimitry Andric
58248a4dda33SDimitry Andric llvm::GlobalValue::LinkageTypes
getLLVMLinkageVarDefinition(const VarDecl * VD)58258a4dda33SDimitry Andric CodeGenModule::getLLVMLinkageVarDefinition(const VarDecl *VD) {
58260b57cec5SDimitry Andric GVALinkage Linkage = getContext().GetGVALinkageForVariable(VD);
58278a4dda33SDimitry Andric return getLLVMLinkageForDeclarator(VD, Linkage);
58280b57cec5SDimitry Andric }
58290b57cec5SDimitry Andric
58300b57cec5SDimitry Andric /// Replace the uses of a function that was declared with a non-proto type.
58310b57cec5SDimitry Andric /// We want to silently drop extra arguments from call sites
replaceUsesOfNonProtoConstant(llvm::Constant * old,llvm::Function * newFn)58320b57cec5SDimitry Andric static void replaceUsesOfNonProtoConstant(llvm::Constant *old,
58330b57cec5SDimitry Andric llvm::Function *newFn) {
58340b57cec5SDimitry Andric // Fast path.
58350fca6ea1SDimitry Andric if (old->use_empty())
58360fca6ea1SDimitry Andric return;
58370b57cec5SDimitry Andric
58380b57cec5SDimitry Andric llvm::Type *newRetTy = newFn->getReturnType();
58390b57cec5SDimitry Andric SmallVector<llvm::Value *, 4> newArgs;
58400b57cec5SDimitry Andric
58410fca6ea1SDimitry Andric SmallVector<llvm::CallBase *> callSitesToBeRemovedFromParent;
58420fca6ea1SDimitry Andric
58430b57cec5SDimitry Andric for (llvm::Value::use_iterator ui = old->use_begin(), ue = old->use_end();
58440fca6ea1SDimitry Andric ui != ue; ui++) {
58450fca6ea1SDimitry Andric llvm::User *user = ui->getUser();
58460b57cec5SDimitry Andric
58470b57cec5SDimitry Andric // Recognize and replace uses of bitcasts. Most calls to
58480b57cec5SDimitry Andric // unprototyped functions will use bitcasts.
58490b57cec5SDimitry Andric if (auto *bitcast = dyn_cast<llvm::ConstantExpr>(user)) {
58500b57cec5SDimitry Andric if (bitcast->getOpcode() == llvm::Instruction::BitCast)
58510b57cec5SDimitry Andric replaceUsesOfNonProtoConstant(bitcast, newFn);
58520b57cec5SDimitry Andric continue;
58530b57cec5SDimitry Andric }
58540b57cec5SDimitry Andric
58550b57cec5SDimitry Andric // Recognize calls to the function.
58560b57cec5SDimitry Andric llvm::CallBase *callSite = dyn_cast<llvm::CallBase>(user);
58570fca6ea1SDimitry Andric if (!callSite)
58580fca6ea1SDimitry Andric continue;
58590fca6ea1SDimitry Andric if (!callSite->isCallee(&*ui))
58600b57cec5SDimitry Andric continue;
58610b57cec5SDimitry Andric
58620b57cec5SDimitry Andric // If the return types don't match exactly, then we can't
58630b57cec5SDimitry Andric // transform this call unless it's dead.
58640b57cec5SDimitry Andric if (callSite->getType() != newRetTy && !callSite->use_empty())
58650b57cec5SDimitry Andric continue;
58660b57cec5SDimitry Andric
58670b57cec5SDimitry Andric // Get the call site's attribute list.
58680b57cec5SDimitry Andric SmallVector<llvm::AttributeSet, 8> newArgAttrs;
58690b57cec5SDimitry Andric llvm::AttributeList oldAttrs = callSite->getAttributes();
58700b57cec5SDimitry Andric
58710b57cec5SDimitry Andric // If the function was passed too few arguments, don't transform.
58720b57cec5SDimitry Andric unsigned newNumArgs = newFn->arg_size();
58730b57cec5SDimitry Andric if (callSite->arg_size() < newNumArgs)
58740b57cec5SDimitry Andric continue;
58750b57cec5SDimitry Andric
58760b57cec5SDimitry Andric // If extra arguments were passed, we silently drop them.
58770b57cec5SDimitry Andric // If any of the types mismatch, we don't transform.
58780b57cec5SDimitry Andric unsigned argNo = 0;
58790b57cec5SDimitry Andric bool dontTransform = false;
58800b57cec5SDimitry Andric for (llvm::Argument &A : newFn->args()) {
58810b57cec5SDimitry Andric if (callSite->getArgOperand(argNo)->getType() != A.getType()) {
58820b57cec5SDimitry Andric dontTransform = true;
58830b57cec5SDimitry Andric break;
58840b57cec5SDimitry Andric }
58850b57cec5SDimitry Andric
58860b57cec5SDimitry Andric // Add any parameter attributes.
5887349cc55cSDimitry Andric newArgAttrs.push_back(oldAttrs.getParamAttrs(argNo));
58880b57cec5SDimitry Andric argNo++;
58890b57cec5SDimitry Andric }
58900b57cec5SDimitry Andric if (dontTransform)
58910b57cec5SDimitry Andric continue;
58920b57cec5SDimitry Andric
58930b57cec5SDimitry Andric // Okay, we can transform this. Create the new call instruction and copy
58940b57cec5SDimitry Andric // over the required information.
58950b57cec5SDimitry Andric newArgs.append(callSite->arg_begin(), callSite->arg_begin() + argNo);
58960b57cec5SDimitry Andric
58970b57cec5SDimitry Andric // Copy over any operand bundles.
5898fe6060f1SDimitry Andric SmallVector<llvm::OperandBundleDef, 1> newBundles;
58990b57cec5SDimitry Andric callSite->getOperandBundlesAsDefs(newBundles);
59000b57cec5SDimitry Andric
59010b57cec5SDimitry Andric llvm::CallBase *newCall;
5902349cc55cSDimitry Andric if (isa<llvm::CallInst>(callSite)) {
59030b57cec5SDimitry Andric newCall =
59040b57cec5SDimitry Andric llvm::CallInst::Create(newFn, newArgs, newBundles, "", callSite);
59050b57cec5SDimitry Andric } else {
59060b57cec5SDimitry Andric auto *oldInvoke = cast<llvm::InvokeInst>(callSite);
59070b57cec5SDimitry Andric newCall = llvm::InvokeInst::Create(newFn, oldInvoke->getNormalDest(),
59080b57cec5SDimitry Andric oldInvoke->getUnwindDest(), newArgs,
59090b57cec5SDimitry Andric newBundles, "", callSite);
59100b57cec5SDimitry Andric }
59110b57cec5SDimitry Andric newArgs.clear(); // for the next iteration
59120b57cec5SDimitry Andric
59130b57cec5SDimitry Andric if (!newCall->getType()->isVoidTy())
59140b57cec5SDimitry Andric newCall->takeName(callSite);
5915349cc55cSDimitry Andric newCall->setAttributes(
5916349cc55cSDimitry Andric llvm::AttributeList::get(newFn->getContext(), oldAttrs.getFnAttrs(),
5917349cc55cSDimitry Andric oldAttrs.getRetAttrs(), newArgAttrs));
59180b57cec5SDimitry Andric newCall->setCallingConv(callSite->getCallingConv());
59190b57cec5SDimitry Andric
59200b57cec5SDimitry Andric // Finally, remove the old call, replacing any uses with the new one.
59210b57cec5SDimitry Andric if (!callSite->use_empty())
59220b57cec5SDimitry Andric callSite->replaceAllUsesWith(newCall);
59230b57cec5SDimitry Andric
59240b57cec5SDimitry Andric // Copy debug location attached to CI.
59250b57cec5SDimitry Andric if (callSite->getDebugLoc())
59260b57cec5SDimitry Andric newCall->setDebugLoc(callSite->getDebugLoc());
59270b57cec5SDimitry Andric
59280fca6ea1SDimitry Andric callSitesToBeRemovedFromParent.push_back(callSite);
59290fca6ea1SDimitry Andric }
59300fca6ea1SDimitry Andric
59310fca6ea1SDimitry Andric for (auto *callSite : callSitesToBeRemovedFromParent) {
59320b57cec5SDimitry Andric callSite->eraseFromParent();
59330b57cec5SDimitry Andric }
59340b57cec5SDimitry Andric }
59350b57cec5SDimitry Andric
59360b57cec5SDimitry Andric /// ReplaceUsesOfNonProtoTypeWithRealFunction - This function is called when we
59370b57cec5SDimitry Andric /// implement a function with no prototype, e.g. "int foo() {}". If there are
59380b57cec5SDimitry Andric /// existing call uses of the old function in the module, this adjusts them to
59390b57cec5SDimitry Andric /// call the new function directly.
59400b57cec5SDimitry Andric ///
59410b57cec5SDimitry Andric /// This is not just a cleanup: the always_inline pass requires direct calls to
59420b57cec5SDimitry Andric /// functions to be able to inline them. If there is a bitcast in the way, it
59430b57cec5SDimitry Andric /// won't inline them. Instcombine normally deletes these calls, but it isn't
59440b57cec5SDimitry Andric /// run at -O0.
ReplaceUsesOfNonProtoTypeWithRealFunction(llvm::GlobalValue * Old,llvm::Function * NewFn)59450b57cec5SDimitry Andric static void ReplaceUsesOfNonProtoTypeWithRealFunction(llvm::GlobalValue *Old,
59460b57cec5SDimitry Andric llvm::Function *NewFn) {
59470b57cec5SDimitry Andric // If we're redefining a global as a function, don't transform it.
59480b57cec5SDimitry Andric if (!isa<llvm::Function>(Old)) return;
59490b57cec5SDimitry Andric
59500b57cec5SDimitry Andric replaceUsesOfNonProtoConstant(Old, NewFn);
59510b57cec5SDimitry Andric }
59520b57cec5SDimitry Andric
HandleCXXStaticMemberVarInstantiation(VarDecl * VD)59530b57cec5SDimitry Andric void CodeGenModule::HandleCXXStaticMemberVarInstantiation(VarDecl *VD) {
59540b57cec5SDimitry Andric auto DK = VD->isThisDeclarationADefinition();
59550fca6ea1SDimitry Andric if ((DK == VarDecl::Definition && VD->hasAttr<DLLImportAttr>()) ||
59560fca6ea1SDimitry Andric (LangOpts.CUDA && !shouldEmitCUDAGlobalVar(VD)))
59570b57cec5SDimitry Andric return;
59580b57cec5SDimitry Andric
59590b57cec5SDimitry Andric TemplateSpecializationKind TSK = VD->getTemplateSpecializationKind();
59600b57cec5SDimitry Andric // If we have a definition, this might be a deferred decl. If the
59610b57cec5SDimitry Andric // instantiation is explicit, make sure we emit it at the end.
59620b57cec5SDimitry Andric if (VD->getDefinition() && TSK == TSK_ExplicitInstantiationDefinition)
59630b57cec5SDimitry Andric GetAddrOfGlobalVar(VD);
59640b57cec5SDimitry Andric
59650b57cec5SDimitry Andric EmitTopLevelDecl(VD);
59660b57cec5SDimitry Andric }
59670b57cec5SDimitry Andric
EmitGlobalFunctionDefinition(GlobalDecl GD,llvm::GlobalValue * GV)59680b57cec5SDimitry Andric void CodeGenModule::EmitGlobalFunctionDefinition(GlobalDecl GD,
59690b57cec5SDimitry Andric llvm::GlobalValue *GV) {
59700b57cec5SDimitry Andric const auto *D = cast<FunctionDecl>(GD.getDecl());
59710b57cec5SDimitry Andric
59720b57cec5SDimitry Andric // Compute the function info and LLVM type.
59730b57cec5SDimitry Andric const CGFunctionInfo &FI = getTypes().arrangeGlobalDeclaration(GD);
59740b57cec5SDimitry Andric llvm::FunctionType *Ty = getTypes().GetFunctionType(FI);
59750b57cec5SDimitry Andric
59760b57cec5SDimitry Andric // Get or create the prototype for the function.
59775ffd83dbSDimitry Andric if (!GV || (GV->getValueType() != Ty))
59780b57cec5SDimitry Andric GV = cast<llvm::GlobalValue>(GetAddrOfFunction(GD, Ty, /*ForVTable=*/false,
59790b57cec5SDimitry Andric /*DontDefer=*/true,
59800b57cec5SDimitry Andric ForDefinition));
59810b57cec5SDimitry Andric
59820b57cec5SDimitry Andric // Already emitted.
59830b57cec5SDimitry Andric if (!GV->isDeclaration())
59840b57cec5SDimitry Andric return;
59850b57cec5SDimitry Andric
59860b57cec5SDimitry Andric // We need to set linkage and visibility on the function before
59870b57cec5SDimitry Andric // generating code for it because various parts of IR generation
59880b57cec5SDimitry Andric // want to propagate this information down (e.g. to local static
59890b57cec5SDimitry Andric // declarations).
59900b57cec5SDimitry Andric auto *Fn = cast<llvm::Function>(GV);
59910b57cec5SDimitry Andric setFunctionLinkage(GD, Fn);
59920b57cec5SDimitry Andric
59930b57cec5SDimitry Andric // FIXME: this is redundant with part of setFunctionDefinitionAttributes
59940b57cec5SDimitry Andric setGVProperties(Fn, GD);
59950b57cec5SDimitry Andric
59960b57cec5SDimitry Andric MaybeHandleStaticInExternC(D, Fn);
59970b57cec5SDimitry Andric
59980b57cec5SDimitry Andric maybeSetTrivialComdat(*D, *Fn);
59990b57cec5SDimitry Andric
60005ffd83dbSDimitry Andric CodeGenFunction(*this).GenerateCode(GD, Fn, FI);
60010b57cec5SDimitry Andric
60020b57cec5SDimitry Andric setNonAliasAttributes(GD, Fn);
60030b57cec5SDimitry Andric SetLLVMFunctionAttributesForDefinition(D, Fn);
60040b57cec5SDimitry Andric
60050b57cec5SDimitry Andric if (const ConstructorAttr *CA = D->getAttr<ConstructorAttr>())
60060b57cec5SDimitry Andric AddGlobalCtor(Fn, CA->getPriority());
60070b57cec5SDimitry Andric if (const DestructorAttr *DA = D->getAttr<DestructorAttr>())
6008e8d8bef9SDimitry Andric AddGlobalDtor(Fn, DA->getPriority(), true);
60095f757f3fSDimitry Andric if (getLangOpts().OpenMP && D->hasAttr<OMPDeclareTargetDeclAttr>())
60105f757f3fSDimitry Andric getOpenMPRuntime().emitDeclareTargetFunction(D, GV);
60110b57cec5SDimitry Andric }
60120b57cec5SDimitry Andric
EmitAliasDefinition(GlobalDecl GD)60130b57cec5SDimitry Andric void CodeGenModule::EmitAliasDefinition(GlobalDecl GD) {
60140b57cec5SDimitry Andric const auto *D = cast<ValueDecl>(GD.getDecl());
60150b57cec5SDimitry Andric const AliasAttr *AA = D->getAttr<AliasAttr>();
60160b57cec5SDimitry Andric assert(AA && "Not an alias?");
60170b57cec5SDimitry Andric
60180b57cec5SDimitry Andric StringRef MangledName = getMangledName(GD);
60190b57cec5SDimitry Andric
60200b57cec5SDimitry Andric if (AA->getAliasee() == MangledName) {
60210b57cec5SDimitry Andric Diags.Report(AA->getLocation(), diag::err_cyclic_alias) << 0;
60220b57cec5SDimitry Andric return;
60230b57cec5SDimitry Andric }
60240b57cec5SDimitry Andric
60250b57cec5SDimitry Andric // If there is a definition in the module, then it wins over the alias.
60260b57cec5SDimitry Andric // This is dubious, but allow it to be safe. Just ignore the alias.
60270b57cec5SDimitry Andric llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
60280b57cec5SDimitry Andric if (Entry && !Entry->isDeclaration())
60290b57cec5SDimitry Andric return;
60300b57cec5SDimitry Andric
60310b57cec5SDimitry Andric Aliases.push_back(GD);
60320b57cec5SDimitry Andric
60330b57cec5SDimitry Andric llvm::Type *DeclTy = getTypes().ConvertTypeForMem(D->getType());
60340b57cec5SDimitry Andric
60350b57cec5SDimitry Andric // Create a reference to the named value. This ensures that it is emitted
60360b57cec5SDimitry Andric // if a deferred decl.
60370b57cec5SDimitry Andric llvm::Constant *Aliasee;
60380b57cec5SDimitry Andric llvm::GlobalValue::LinkageTypes LT;
60390b57cec5SDimitry Andric if (isa<llvm::FunctionType>(DeclTy)) {
60400b57cec5SDimitry Andric Aliasee = GetOrCreateLLVMFunction(AA->getAliasee(), DeclTy, GD,
60410b57cec5SDimitry Andric /*ForVTable=*/false);
60420b57cec5SDimitry Andric LT = getFunctionLinkage(GD);
60430b57cec5SDimitry Andric } else {
6044349cc55cSDimitry Andric Aliasee = GetOrCreateLLVMGlobal(AA->getAliasee(), DeclTy, LangAS::Default,
60450b57cec5SDimitry Andric /*D=*/nullptr);
6046e8d8bef9SDimitry Andric if (const auto *VD = dyn_cast<VarDecl>(GD.getDecl()))
60478a4dda33SDimitry Andric LT = getLLVMLinkageVarDefinition(VD);
6048e8d8bef9SDimitry Andric else
6049e8d8bef9SDimitry Andric LT = getFunctionLinkage(GD);
60500b57cec5SDimitry Andric }
60510b57cec5SDimitry Andric
60520b57cec5SDimitry Andric // Create the new alias itself, but don't set a name yet.
60535ffd83dbSDimitry Andric unsigned AS = Aliasee->getType()->getPointerAddressSpace();
60540b57cec5SDimitry Andric auto *GA =
60555ffd83dbSDimitry Andric llvm::GlobalAlias::create(DeclTy, AS, LT, "", Aliasee, &getModule());
60560b57cec5SDimitry Andric
60570b57cec5SDimitry Andric if (Entry) {
60580b57cec5SDimitry Andric if (GA->getAliasee() == Entry) {
60590b57cec5SDimitry Andric Diags.Report(AA->getLocation(), diag::err_cyclic_alias) << 0;
60600b57cec5SDimitry Andric return;
60610b57cec5SDimitry Andric }
60620b57cec5SDimitry Andric
60630b57cec5SDimitry Andric assert(Entry->isDeclaration());
60640b57cec5SDimitry Andric
60650b57cec5SDimitry Andric // If there is a declaration in the module, then we had an extern followed
60660b57cec5SDimitry Andric // by the alias, as in:
60670b57cec5SDimitry Andric // extern int test6();
60680b57cec5SDimitry Andric // ...
60690b57cec5SDimitry Andric // int test6() __attribute__((alias("test7")));
60700b57cec5SDimitry Andric //
60710b57cec5SDimitry Andric // Remove it and replace uses of it with the alias.
60720b57cec5SDimitry Andric GA->takeName(Entry);
60730b57cec5SDimitry Andric
60745f757f3fSDimitry Andric Entry->replaceAllUsesWith(GA);
60750b57cec5SDimitry Andric Entry->eraseFromParent();
60760b57cec5SDimitry Andric } else {
60770b57cec5SDimitry Andric GA->setName(MangledName);
60780b57cec5SDimitry Andric }
60790b57cec5SDimitry Andric
60800b57cec5SDimitry Andric // Set attributes which are particular to an alias; this is a
60810b57cec5SDimitry Andric // specialization of the attributes which may be set on a global
60820b57cec5SDimitry Andric // variable/function.
60830b57cec5SDimitry Andric if (D->hasAttr<WeakAttr>() || D->hasAttr<WeakRefAttr>() ||
60840b57cec5SDimitry Andric D->isWeakImported()) {
60850b57cec5SDimitry Andric GA->setLinkage(llvm::Function::WeakAnyLinkage);
60860b57cec5SDimitry Andric }
60870b57cec5SDimitry Andric
60880b57cec5SDimitry Andric if (const auto *VD = dyn_cast<VarDecl>(D))
60890b57cec5SDimitry Andric if (VD->getTLSKind())
60900b57cec5SDimitry Andric setTLSMode(GA, *VD);
60910b57cec5SDimitry Andric
60920b57cec5SDimitry Andric SetCommonAttributes(GD, GA);
609381ad6265SDimitry Andric
609481ad6265SDimitry Andric // Emit global alias debug information.
609581ad6265SDimitry Andric if (isa<VarDecl>(D))
609681ad6265SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
6097bdd1243dSDimitry Andric DI->EmitGlobalAlias(cast<llvm::GlobalValue>(GA->getAliasee()->stripPointerCasts()), GD);
60980b57cec5SDimitry Andric }
60990b57cec5SDimitry Andric
emitIFuncDefinition(GlobalDecl GD)61000b57cec5SDimitry Andric void CodeGenModule::emitIFuncDefinition(GlobalDecl GD) {
61010b57cec5SDimitry Andric const auto *D = cast<ValueDecl>(GD.getDecl());
61020b57cec5SDimitry Andric const IFuncAttr *IFA = D->getAttr<IFuncAttr>();
61030b57cec5SDimitry Andric assert(IFA && "Not an ifunc?");
61040b57cec5SDimitry Andric
61050b57cec5SDimitry Andric StringRef MangledName = getMangledName(GD);
61060b57cec5SDimitry Andric
61070b57cec5SDimitry Andric if (IFA->getResolver() == MangledName) {
61080b57cec5SDimitry Andric Diags.Report(IFA->getLocation(), diag::err_cyclic_alias) << 1;
61090b57cec5SDimitry Andric return;
61100b57cec5SDimitry Andric }
61110b57cec5SDimitry Andric
61120b57cec5SDimitry Andric // Report an error if some definition overrides ifunc.
61130b57cec5SDimitry Andric llvm::GlobalValue *Entry = GetGlobalValue(MangledName);
61140b57cec5SDimitry Andric if (Entry && !Entry->isDeclaration()) {
61150b57cec5SDimitry Andric GlobalDecl OtherGD;
61160b57cec5SDimitry Andric if (lookupRepresentativeDecl(MangledName, OtherGD) &&
61170b57cec5SDimitry Andric DiagnosedConflictingDefinitions.insert(GD).second) {
61180b57cec5SDimitry Andric Diags.Report(D->getLocation(), diag::err_duplicate_mangled_name)
61190b57cec5SDimitry Andric << MangledName;
61200b57cec5SDimitry Andric Diags.Report(OtherGD.getDecl()->getLocation(),
61210b57cec5SDimitry Andric diag::note_previous_definition);
61220b57cec5SDimitry Andric }
61230b57cec5SDimitry Andric return;
61240b57cec5SDimitry Andric }
61250b57cec5SDimitry Andric
61260b57cec5SDimitry Andric Aliases.push_back(GD);
61270b57cec5SDimitry Andric
61280fca6ea1SDimitry Andric // The resolver might not be visited yet. Specify a dummy non-function type to
61290fca6ea1SDimitry Andric // indicate IsIncompleteFunction. Either the type is ignored (if the resolver
61300fca6ea1SDimitry Andric // was emitted) or the whole function will be replaced (if the resolver has
61310fca6ea1SDimitry Andric // not been emitted).
61320b57cec5SDimitry Andric llvm::Constant *Resolver =
61330fca6ea1SDimitry Andric GetOrCreateLLVMFunction(IFA->getResolver(), VoidTy, {},
61340b57cec5SDimitry Andric /*ForVTable=*/false);
61350fca6ea1SDimitry Andric llvm::Type *DeclTy = getTypes().ConvertTypeForMem(D->getType());
61360b57cec5SDimitry Andric llvm::GlobalIFunc *GIF =
61370b57cec5SDimitry Andric llvm::GlobalIFunc::create(DeclTy, 0, llvm::Function::ExternalLinkage,
61380b57cec5SDimitry Andric "", Resolver, &getModule());
61390b57cec5SDimitry Andric if (Entry) {
61400b57cec5SDimitry Andric if (GIF->getResolver() == Entry) {
61410b57cec5SDimitry Andric Diags.Report(IFA->getLocation(), diag::err_cyclic_alias) << 1;
61420b57cec5SDimitry Andric return;
61430b57cec5SDimitry Andric }
61440b57cec5SDimitry Andric assert(Entry->isDeclaration());
61450b57cec5SDimitry Andric
61460b57cec5SDimitry Andric // If there is a declaration in the module, then we had an extern followed
61470b57cec5SDimitry Andric // by the ifunc, as in:
61480b57cec5SDimitry Andric // extern int test();
61490b57cec5SDimitry Andric // ...
61500b57cec5SDimitry Andric // int test() __attribute__((ifunc("resolver")));
61510b57cec5SDimitry Andric //
61520b57cec5SDimitry Andric // Remove it and replace uses of it with the ifunc.
61530b57cec5SDimitry Andric GIF->takeName(Entry);
61540b57cec5SDimitry Andric
61555f757f3fSDimitry Andric Entry->replaceAllUsesWith(GIF);
61560b57cec5SDimitry Andric Entry->eraseFromParent();
61570b57cec5SDimitry Andric } else
61580b57cec5SDimitry Andric GIF->setName(MangledName);
61590b57cec5SDimitry Andric SetCommonAttributes(GD, GIF);
61600b57cec5SDimitry Andric }
61610b57cec5SDimitry Andric
getIntrinsic(unsigned IID,ArrayRef<llvm::Type * > Tys)61620b57cec5SDimitry Andric llvm::Function *CodeGenModule::getIntrinsic(unsigned IID,
61630b57cec5SDimitry Andric ArrayRef<llvm::Type*> Tys) {
61640b57cec5SDimitry Andric return llvm::Intrinsic::getDeclaration(&getModule(), (llvm::Intrinsic::ID)IID,
61650b57cec5SDimitry Andric Tys);
61660b57cec5SDimitry Andric }
61670b57cec5SDimitry Andric
61680b57cec5SDimitry Andric static llvm::StringMapEntry<llvm::GlobalVariable *> &
GetConstantCFStringEntry(llvm::StringMap<llvm::GlobalVariable * > & Map,const StringLiteral * Literal,bool TargetIsLSB,bool & IsUTF16,unsigned & StringLength)61690b57cec5SDimitry Andric GetConstantCFStringEntry(llvm::StringMap<llvm::GlobalVariable *> &Map,
61700b57cec5SDimitry Andric const StringLiteral *Literal, bool TargetIsLSB,
61710b57cec5SDimitry Andric bool &IsUTF16, unsigned &StringLength) {
61720b57cec5SDimitry Andric StringRef String = Literal->getString();
61730b57cec5SDimitry Andric unsigned NumBytes = String.size();
61740b57cec5SDimitry Andric
61750b57cec5SDimitry Andric // Check for simple case.
61760b57cec5SDimitry Andric if (!Literal->containsNonAsciiOrNull()) {
61770b57cec5SDimitry Andric StringLength = NumBytes;
61780b57cec5SDimitry Andric return *Map.insert(std::make_pair(String, nullptr)).first;
61790b57cec5SDimitry Andric }
61800b57cec5SDimitry Andric
61810b57cec5SDimitry Andric // Otherwise, convert the UTF8 literals into a string of shorts.
61820b57cec5SDimitry Andric IsUTF16 = true;
61830b57cec5SDimitry Andric
61840b57cec5SDimitry Andric SmallVector<llvm::UTF16, 128> ToBuf(NumBytes + 1); // +1 for ending nulls.
61850b57cec5SDimitry Andric const llvm::UTF8 *FromPtr = (const llvm::UTF8 *)String.data();
61860b57cec5SDimitry Andric llvm::UTF16 *ToPtr = &ToBuf[0];
61870b57cec5SDimitry Andric
61880b57cec5SDimitry Andric (void)llvm::ConvertUTF8toUTF16(&FromPtr, FromPtr + NumBytes, &ToPtr,
61890b57cec5SDimitry Andric ToPtr + NumBytes, llvm::strictConversion);
61900b57cec5SDimitry Andric
61910b57cec5SDimitry Andric // ConvertUTF8toUTF16 returns the length in ToPtr.
61920b57cec5SDimitry Andric StringLength = ToPtr - &ToBuf[0];
61930b57cec5SDimitry Andric
61940b57cec5SDimitry Andric // Add an explicit null.
61950b57cec5SDimitry Andric *ToPtr = 0;
61960b57cec5SDimitry Andric return *Map.insert(std::make_pair(
61970b57cec5SDimitry Andric StringRef(reinterpret_cast<const char *>(ToBuf.data()),
61980b57cec5SDimitry Andric (StringLength + 1) * 2),
61990b57cec5SDimitry Andric nullptr)).first;
62000b57cec5SDimitry Andric }
62010b57cec5SDimitry Andric
62020b57cec5SDimitry Andric ConstantAddress
GetAddrOfConstantCFString(const StringLiteral * Literal)62030b57cec5SDimitry Andric CodeGenModule::GetAddrOfConstantCFString(const StringLiteral *Literal) {
62040b57cec5SDimitry Andric unsigned StringLength = 0;
62050b57cec5SDimitry Andric bool isUTF16 = false;
62060b57cec5SDimitry Andric llvm::StringMapEntry<llvm::GlobalVariable *> &Entry =
62070b57cec5SDimitry Andric GetConstantCFStringEntry(CFConstantStringMap, Literal,
62080b57cec5SDimitry Andric getDataLayout().isLittleEndian(), isUTF16,
62090b57cec5SDimitry Andric StringLength);
62100b57cec5SDimitry Andric
62110b57cec5SDimitry Andric if (auto *C = Entry.second)
62120eae32dcSDimitry Andric return ConstantAddress(
62130eae32dcSDimitry Andric C, C->getValueType(), CharUnits::fromQuantity(C->getAlignment()));
62140b57cec5SDimitry Andric
62150b57cec5SDimitry Andric const ASTContext &Context = getContext();
62160b57cec5SDimitry Andric const llvm::Triple &Triple = getTriple();
62170b57cec5SDimitry Andric
62180b57cec5SDimitry Andric const auto CFRuntime = getLangOpts().CFRuntime;
62190b57cec5SDimitry Andric const bool IsSwiftABI =
62200b57cec5SDimitry Andric static_cast<unsigned>(CFRuntime) >=
62210b57cec5SDimitry Andric static_cast<unsigned>(LangOptions::CoreFoundationABI::Swift);
62220b57cec5SDimitry Andric const bool IsSwift4_1 = CFRuntime == LangOptions::CoreFoundationABI::Swift4_1;
62230b57cec5SDimitry Andric
62240b57cec5SDimitry Andric // If we don't already have it, get __CFConstantStringClassReference.
62250b57cec5SDimitry Andric if (!CFConstantStringClassRef) {
62260b57cec5SDimitry Andric const char *CFConstantStringClassName = "__CFConstantStringClassReference";
62270b57cec5SDimitry Andric llvm::Type *Ty = getTypes().ConvertType(getContext().IntTy);
62280b57cec5SDimitry Andric Ty = llvm::ArrayType::get(Ty, 0);
62290b57cec5SDimitry Andric
62300b57cec5SDimitry Andric switch (CFRuntime) {
62310b57cec5SDimitry Andric default: break;
6232bdd1243dSDimitry Andric case LangOptions::CoreFoundationABI::Swift: [[fallthrough]];
62330b57cec5SDimitry Andric case LangOptions::CoreFoundationABI::Swift5_0:
62340b57cec5SDimitry Andric CFConstantStringClassName =
62350b57cec5SDimitry Andric Triple.isOSDarwin() ? "$s15SwiftFoundation19_NSCFConstantStringCN"
62360b57cec5SDimitry Andric : "$s10Foundation19_NSCFConstantStringCN";
62370b57cec5SDimitry Andric Ty = IntPtrTy;
62380b57cec5SDimitry Andric break;
62390b57cec5SDimitry Andric case LangOptions::CoreFoundationABI::Swift4_2:
62400b57cec5SDimitry Andric CFConstantStringClassName =
62410b57cec5SDimitry Andric Triple.isOSDarwin() ? "$S15SwiftFoundation19_NSCFConstantStringCN"
62420b57cec5SDimitry Andric : "$S10Foundation19_NSCFConstantStringCN";
62430b57cec5SDimitry Andric Ty = IntPtrTy;
62440b57cec5SDimitry Andric break;
62450b57cec5SDimitry Andric case LangOptions::CoreFoundationABI::Swift4_1:
62460b57cec5SDimitry Andric CFConstantStringClassName =
62470b57cec5SDimitry Andric Triple.isOSDarwin() ? "__T015SwiftFoundation19_NSCFConstantStringCN"
62480b57cec5SDimitry Andric : "__T010Foundation19_NSCFConstantStringCN";
62490b57cec5SDimitry Andric Ty = IntPtrTy;
62500b57cec5SDimitry Andric break;
62510b57cec5SDimitry Andric }
62520b57cec5SDimitry Andric
62530b57cec5SDimitry Andric llvm::Constant *C = CreateRuntimeVariable(Ty, CFConstantStringClassName);
62540b57cec5SDimitry Andric
62550b57cec5SDimitry Andric if (Triple.isOSBinFormatELF() || Triple.isOSBinFormatCOFF()) {
62560b57cec5SDimitry Andric llvm::GlobalValue *GV = nullptr;
62570b57cec5SDimitry Andric
62580b57cec5SDimitry Andric if ((GV = dyn_cast<llvm::GlobalValue>(C))) {
62590b57cec5SDimitry Andric IdentifierInfo &II = Context.Idents.get(GV->getName());
62600b57cec5SDimitry Andric TranslationUnitDecl *TUDecl = Context.getTranslationUnitDecl();
62610b57cec5SDimitry Andric DeclContext *DC = TranslationUnitDecl::castToDeclContext(TUDecl);
62620b57cec5SDimitry Andric
62630b57cec5SDimitry Andric const VarDecl *VD = nullptr;
6264fe6060f1SDimitry Andric for (const auto *Result : DC->lookup(&II))
62650b57cec5SDimitry Andric if ((VD = dyn_cast<VarDecl>(Result)))
62660b57cec5SDimitry Andric break;
62670b57cec5SDimitry Andric
62680b57cec5SDimitry Andric if (Triple.isOSBinFormatELF()) {
62690b57cec5SDimitry Andric if (!VD)
62700b57cec5SDimitry Andric GV->setLinkage(llvm::GlobalValue::ExternalLinkage);
62710b57cec5SDimitry Andric } else {
62720b57cec5SDimitry Andric GV->setLinkage(llvm::GlobalValue::ExternalLinkage);
62730b57cec5SDimitry Andric if (!VD || !VD->hasAttr<DLLExportAttr>())
62740b57cec5SDimitry Andric GV->setDLLStorageClass(llvm::GlobalValue::DLLImportStorageClass);
62750b57cec5SDimitry Andric else
62760b57cec5SDimitry Andric GV->setDLLStorageClass(llvm::GlobalValue::DLLExportStorageClass);
62770b57cec5SDimitry Andric }
62780b57cec5SDimitry Andric
62790b57cec5SDimitry Andric setDSOLocal(GV);
62800b57cec5SDimitry Andric }
62810b57cec5SDimitry Andric }
62820b57cec5SDimitry Andric
62830b57cec5SDimitry Andric // Decay array -> ptr
62840b57cec5SDimitry Andric CFConstantStringClassRef =
62850fca6ea1SDimitry Andric IsSwiftABI ? llvm::ConstantExpr::getPtrToInt(C, Ty) : C;
62860b57cec5SDimitry Andric }
62870b57cec5SDimitry Andric
62880b57cec5SDimitry Andric QualType CFTy = Context.getCFConstantStringType();
62890b57cec5SDimitry Andric
62900b57cec5SDimitry Andric auto *STy = cast<llvm::StructType>(getTypes().ConvertType(CFTy));
62910b57cec5SDimitry Andric
62920b57cec5SDimitry Andric ConstantInitBuilder Builder(*this);
62930b57cec5SDimitry Andric auto Fields = Builder.beginStruct(STy);
62940b57cec5SDimitry Andric
62950b57cec5SDimitry Andric // Class pointer.
629681ad6265SDimitry Andric Fields.add(cast<llvm::Constant>(CFConstantStringClassRef));
62970b57cec5SDimitry Andric
62980b57cec5SDimitry Andric // Flags.
62990b57cec5SDimitry Andric if (IsSwiftABI) {
63000b57cec5SDimitry Andric Fields.addInt(IntPtrTy, IsSwift4_1 ? 0x05 : 0x01);
63010b57cec5SDimitry Andric Fields.addInt(Int64Ty, isUTF16 ? 0x07d0 : 0x07c8);
63020b57cec5SDimitry Andric } else {
63030b57cec5SDimitry Andric Fields.addInt(IntTy, isUTF16 ? 0x07d0 : 0x07C8);
63040b57cec5SDimitry Andric }
63050b57cec5SDimitry Andric
63060b57cec5SDimitry Andric // String pointer.
63070b57cec5SDimitry Andric llvm::Constant *C = nullptr;
63080b57cec5SDimitry Andric if (isUTF16) {
6309bdd1243dSDimitry Andric auto Arr = llvm::ArrayRef(
63100b57cec5SDimitry Andric reinterpret_cast<uint16_t *>(const_cast<char *>(Entry.first().data())),
63110b57cec5SDimitry Andric Entry.first().size() / 2);
63120b57cec5SDimitry Andric C = llvm::ConstantDataArray::get(VMContext, Arr);
63130b57cec5SDimitry Andric } else {
63140b57cec5SDimitry Andric C = llvm::ConstantDataArray::getString(VMContext, Entry.first());
63150b57cec5SDimitry Andric }
63160b57cec5SDimitry Andric
63170b57cec5SDimitry Andric // Note: -fwritable-strings doesn't make the backing store strings of
63185f757f3fSDimitry Andric // CFStrings writable.
63190b57cec5SDimitry Andric auto *GV =
63200b57cec5SDimitry Andric new llvm::GlobalVariable(getModule(), C->getType(), /*isConstant=*/true,
63210b57cec5SDimitry Andric llvm::GlobalValue::PrivateLinkage, C, ".str");
63220b57cec5SDimitry Andric GV->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
63230b57cec5SDimitry Andric // Don't enforce the target's minimum global alignment, since the only use
63240b57cec5SDimitry Andric // of the string is via this class initializer.
63250b57cec5SDimitry Andric CharUnits Align = isUTF16 ? Context.getTypeAlignInChars(Context.ShortTy)
63260b57cec5SDimitry Andric : Context.getTypeAlignInChars(Context.CharTy);
6327a7dea167SDimitry Andric GV->setAlignment(Align.getAsAlign());
63280b57cec5SDimitry Andric
63290b57cec5SDimitry Andric // FIXME: We set the section explicitly to avoid a bug in ld64 224.1.
63300b57cec5SDimitry Andric // Without it LLVM can merge the string with a non unnamed_addr one during
63310b57cec5SDimitry Andric // LTO. Doing that changes the section it ends in, which surprises ld64.
63320b57cec5SDimitry Andric if (Triple.isOSBinFormatMachO())
63330b57cec5SDimitry Andric GV->setSection(isUTF16 ? "__TEXT,__ustring"
63340b57cec5SDimitry Andric : "__TEXT,__cstring,cstring_literals");
63350b57cec5SDimitry Andric // Make sure the literal ends up in .rodata to allow for safe ICF and for
63360b57cec5SDimitry Andric // the static linker to adjust permissions to read-only later on.
63370b57cec5SDimitry Andric else if (Triple.isOSBinFormatELF())
63380b57cec5SDimitry Andric GV->setSection(".rodata");
63390b57cec5SDimitry Andric
63400b57cec5SDimitry Andric // String.
63410fca6ea1SDimitry Andric Fields.add(GV);
63420b57cec5SDimitry Andric
63430b57cec5SDimitry Andric // String length.
63440b57cec5SDimitry Andric llvm::IntegerType *LengthTy =
63450b57cec5SDimitry Andric llvm::IntegerType::get(getModule().getContext(),
63460b57cec5SDimitry Andric Context.getTargetInfo().getLongWidth());
63470b57cec5SDimitry Andric if (IsSwiftABI) {
63480b57cec5SDimitry Andric if (CFRuntime == LangOptions::CoreFoundationABI::Swift4_1 ||
63490b57cec5SDimitry Andric CFRuntime == LangOptions::CoreFoundationABI::Swift4_2)
63500b57cec5SDimitry Andric LengthTy = Int32Ty;
63510b57cec5SDimitry Andric else
63520b57cec5SDimitry Andric LengthTy = IntPtrTy;
63530b57cec5SDimitry Andric }
63540b57cec5SDimitry Andric Fields.addInt(LengthTy, StringLength);
63550b57cec5SDimitry Andric
6356a7dea167SDimitry Andric // Swift ABI requires 8-byte alignment to ensure that the _Atomic(uint64_t) is
6357a7dea167SDimitry Andric // properly aligned on 32-bit platforms.
6358a7dea167SDimitry Andric CharUnits Alignment =
6359a7dea167SDimitry Andric IsSwiftABI ? Context.toCharUnitsFromBits(64) : getPointerAlign();
63600b57cec5SDimitry Andric
63610b57cec5SDimitry Andric // The struct.
63620b57cec5SDimitry Andric GV = Fields.finishAndCreateGlobal("_unnamed_cfstring_", Alignment,
63630b57cec5SDimitry Andric /*isConstant=*/false,
63640b57cec5SDimitry Andric llvm::GlobalVariable::PrivateLinkage);
63650b57cec5SDimitry Andric GV->addAttribute("objc_arc_inert");
63660b57cec5SDimitry Andric switch (Triple.getObjectFormat()) {
63670b57cec5SDimitry Andric case llvm::Triple::UnknownObjectFormat:
63680b57cec5SDimitry Andric llvm_unreachable("unknown file format");
636981ad6265SDimitry Andric case llvm::Triple::DXContainer:
6370e8d8bef9SDimitry Andric case llvm::Triple::GOFF:
637181ad6265SDimitry Andric case llvm::Triple::SPIRV:
63720b57cec5SDimitry Andric case llvm::Triple::XCOFF:
637381ad6265SDimitry Andric llvm_unreachable("unimplemented");
63740b57cec5SDimitry Andric case llvm::Triple::COFF:
63750b57cec5SDimitry Andric case llvm::Triple::ELF:
63760b57cec5SDimitry Andric case llvm::Triple::Wasm:
63770b57cec5SDimitry Andric GV->setSection("cfstring");
63780b57cec5SDimitry Andric break;
63790b57cec5SDimitry Andric case llvm::Triple::MachO:
63800b57cec5SDimitry Andric GV->setSection("__DATA,__cfstring");
63810b57cec5SDimitry Andric break;
63820b57cec5SDimitry Andric }
63830b57cec5SDimitry Andric Entry.second = GV;
63840b57cec5SDimitry Andric
63850eae32dcSDimitry Andric return ConstantAddress(GV, GV->getValueType(), Alignment);
63860b57cec5SDimitry Andric }
63870b57cec5SDimitry Andric
getExpressionLocationsEnabled() const63880b57cec5SDimitry Andric bool CodeGenModule::getExpressionLocationsEnabled() const {
63890b57cec5SDimitry Andric return !CodeGenOpts.EmitCodeView || CodeGenOpts.DebugColumnInfo;
63900b57cec5SDimitry Andric }
63910b57cec5SDimitry Andric
getObjCFastEnumerationStateType()63920b57cec5SDimitry Andric QualType CodeGenModule::getObjCFastEnumerationStateType() {
63930b57cec5SDimitry Andric if (ObjCFastEnumerationStateType.isNull()) {
63940b57cec5SDimitry Andric RecordDecl *D = Context.buildImplicitRecord("__objcFastEnumerationState");
63950b57cec5SDimitry Andric D->startDefinition();
63960b57cec5SDimitry Andric
63970b57cec5SDimitry Andric QualType FieldTypes[] = {
63985f757f3fSDimitry Andric Context.UnsignedLongTy, Context.getPointerType(Context.getObjCIdType()),
63990b57cec5SDimitry Andric Context.getPointerType(Context.UnsignedLongTy),
64005f757f3fSDimitry Andric Context.getConstantArrayType(Context.UnsignedLongTy, llvm::APInt(32, 5),
64015f757f3fSDimitry Andric nullptr, ArraySizeModifier::Normal, 0)};
64020b57cec5SDimitry Andric
64030b57cec5SDimitry Andric for (size_t i = 0; i < 4; ++i) {
64040b57cec5SDimitry Andric FieldDecl *Field = FieldDecl::Create(Context,
64050b57cec5SDimitry Andric D,
64060b57cec5SDimitry Andric SourceLocation(),
64070b57cec5SDimitry Andric SourceLocation(), nullptr,
64080b57cec5SDimitry Andric FieldTypes[i], /*TInfo=*/nullptr,
64090b57cec5SDimitry Andric /*BitWidth=*/nullptr,
64100b57cec5SDimitry Andric /*Mutable=*/false,
64110b57cec5SDimitry Andric ICIS_NoInit);
64120b57cec5SDimitry Andric Field->setAccess(AS_public);
64130b57cec5SDimitry Andric D->addDecl(Field);
64140b57cec5SDimitry Andric }
64150b57cec5SDimitry Andric
64160b57cec5SDimitry Andric D->completeDefinition();
64170b57cec5SDimitry Andric ObjCFastEnumerationStateType = Context.getTagDeclType(D);
64180b57cec5SDimitry Andric }
64190b57cec5SDimitry Andric
64200b57cec5SDimitry Andric return ObjCFastEnumerationStateType;
64210b57cec5SDimitry Andric }
64220b57cec5SDimitry Andric
64230b57cec5SDimitry Andric llvm::Constant *
GetConstantArrayFromStringLiteral(const StringLiteral * E)64240b57cec5SDimitry Andric CodeGenModule::GetConstantArrayFromStringLiteral(const StringLiteral *E) {
64250b57cec5SDimitry Andric assert(!E->getType()->isPointerType() && "Strings are always arrays");
64260b57cec5SDimitry Andric
64270b57cec5SDimitry Andric // Don't emit it as the address of the string, emit the string data itself
64280b57cec5SDimitry Andric // as an inline array.
64290b57cec5SDimitry Andric if (E->getCharByteWidth() == 1) {
64300b57cec5SDimitry Andric SmallString<64> Str(E->getString());
64310b57cec5SDimitry Andric
64320b57cec5SDimitry Andric // Resize the string to the right size, which is indicated by its type.
64330b57cec5SDimitry Andric const ConstantArrayType *CAT = Context.getAsConstantArrayType(E->getType());
643406c3fb27SDimitry Andric assert(CAT && "String literal not of constant array type!");
64350fca6ea1SDimitry Andric Str.resize(CAT->getZExtSize());
64360b57cec5SDimitry Andric return llvm::ConstantDataArray::getString(VMContext, Str, false);
64370b57cec5SDimitry Andric }
64380b57cec5SDimitry Andric
64390b57cec5SDimitry Andric auto *AType = cast<llvm::ArrayType>(getTypes().ConvertType(E->getType()));
64400b57cec5SDimitry Andric llvm::Type *ElemTy = AType->getElementType();
64410b57cec5SDimitry Andric unsigned NumElements = AType->getNumElements();
64420b57cec5SDimitry Andric
64430b57cec5SDimitry Andric // Wide strings have either 2-byte or 4-byte elements.
64440b57cec5SDimitry Andric if (ElemTy->getPrimitiveSizeInBits() == 16) {
64450b57cec5SDimitry Andric SmallVector<uint16_t, 32> Elements;
64460b57cec5SDimitry Andric Elements.reserve(NumElements);
64470b57cec5SDimitry Andric
64480b57cec5SDimitry Andric for(unsigned i = 0, e = E->getLength(); i != e; ++i)
64490b57cec5SDimitry Andric Elements.push_back(E->getCodeUnit(i));
64500b57cec5SDimitry Andric Elements.resize(NumElements);
64510b57cec5SDimitry Andric return llvm::ConstantDataArray::get(VMContext, Elements);
64520b57cec5SDimitry Andric }
64530b57cec5SDimitry Andric
64540b57cec5SDimitry Andric assert(ElemTy->getPrimitiveSizeInBits() == 32);
64550b57cec5SDimitry Andric SmallVector<uint32_t, 32> Elements;
64560b57cec5SDimitry Andric Elements.reserve(NumElements);
64570b57cec5SDimitry Andric
64580b57cec5SDimitry Andric for(unsigned i = 0, e = E->getLength(); i != e; ++i)
64590b57cec5SDimitry Andric Elements.push_back(E->getCodeUnit(i));
64600b57cec5SDimitry Andric Elements.resize(NumElements);
64610b57cec5SDimitry Andric return llvm::ConstantDataArray::get(VMContext, Elements);
64620b57cec5SDimitry Andric }
64630b57cec5SDimitry Andric
64640b57cec5SDimitry Andric static llvm::GlobalVariable *
GenerateStringLiteral(llvm::Constant * C,llvm::GlobalValue::LinkageTypes LT,CodeGenModule & CGM,StringRef GlobalName,CharUnits Alignment)64650b57cec5SDimitry Andric GenerateStringLiteral(llvm::Constant *C, llvm::GlobalValue::LinkageTypes LT,
64660b57cec5SDimitry Andric CodeGenModule &CGM, StringRef GlobalName,
64670b57cec5SDimitry Andric CharUnits Alignment) {
64680b57cec5SDimitry Andric unsigned AddrSpace = CGM.getContext().getTargetAddressSpace(
6469fe6060f1SDimitry Andric CGM.GetGlobalConstantAddressSpace());
64700b57cec5SDimitry Andric
64710b57cec5SDimitry Andric llvm::Module &M = CGM.getModule();
64720b57cec5SDimitry Andric // Create a global variable for this string
64730b57cec5SDimitry Andric auto *GV = new llvm::GlobalVariable(
64740b57cec5SDimitry Andric M, C->getType(), !CGM.getLangOpts().WritableStrings, LT, C, GlobalName,
64750b57cec5SDimitry Andric nullptr, llvm::GlobalVariable::NotThreadLocal, AddrSpace);
6476a7dea167SDimitry Andric GV->setAlignment(Alignment.getAsAlign());
64770b57cec5SDimitry Andric GV->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global);
64780b57cec5SDimitry Andric if (GV->isWeakForLinker()) {
64790b57cec5SDimitry Andric assert(CGM.supportsCOMDAT() && "Only COFF uses weak string literals");
64800b57cec5SDimitry Andric GV->setComdat(M.getOrInsertComdat(GV->getName()));
64810b57cec5SDimitry Andric }
64820b57cec5SDimitry Andric CGM.setDSOLocal(GV);
64830b57cec5SDimitry Andric
64840b57cec5SDimitry Andric return GV;
64850b57cec5SDimitry Andric }
64860b57cec5SDimitry Andric
64870b57cec5SDimitry Andric /// GetAddrOfConstantStringFromLiteral - Return a pointer to a
64880b57cec5SDimitry Andric /// constant array for the given string literal.
64890b57cec5SDimitry Andric ConstantAddress
GetAddrOfConstantStringFromLiteral(const StringLiteral * S,StringRef Name)64900b57cec5SDimitry Andric CodeGenModule::GetAddrOfConstantStringFromLiteral(const StringLiteral *S,
64910b57cec5SDimitry Andric StringRef Name) {
64920fca6ea1SDimitry Andric CharUnits Alignment =
64930fca6ea1SDimitry Andric getContext().getAlignOfGlobalVarInChars(S->getType(), /*VD=*/nullptr);
64940b57cec5SDimitry Andric
64950b57cec5SDimitry Andric llvm::Constant *C = GetConstantArrayFromStringLiteral(S);
64960b57cec5SDimitry Andric llvm::GlobalVariable **Entry = nullptr;
64970b57cec5SDimitry Andric if (!LangOpts.WritableStrings) {
64980b57cec5SDimitry Andric Entry = &ConstantStringMap[C];
64990b57cec5SDimitry Andric if (auto GV = *Entry) {
6500349cc55cSDimitry Andric if (uint64_t(Alignment.getQuantity()) > GV->getAlignment())
6501a7dea167SDimitry Andric GV->setAlignment(Alignment.getAsAlign());
65020b57cec5SDimitry Andric return ConstantAddress(castStringLiteralToDefaultAddressSpace(*this, GV),
65030eae32dcSDimitry Andric GV->getValueType(), Alignment);
65040b57cec5SDimitry Andric }
65050b57cec5SDimitry Andric }
65060b57cec5SDimitry Andric
65070b57cec5SDimitry Andric SmallString<256> MangledNameBuffer;
65080b57cec5SDimitry Andric StringRef GlobalVariableName;
65090b57cec5SDimitry Andric llvm::GlobalValue::LinkageTypes LT;
65100b57cec5SDimitry Andric
65110b57cec5SDimitry Andric // Mangle the string literal if that's how the ABI merges duplicate strings.
65120b57cec5SDimitry Andric // Don't do it if they are writable, since we don't want writes in one TU to
65130b57cec5SDimitry Andric // affect strings in another.
65140b57cec5SDimitry Andric if (getCXXABI().getMangleContext().shouldMangleStringLiteral(S) &&
65150b57cec5SDimitry Andric !LangOpts.WritableStrings) {
65160b57cec5SDimitry Andric llvm::raw_svector_ostream Out(MangledNameBuffer);
65170b57cec5SDimitry Andric getCXXABI().getMangleContext().mangleStringLiteral(S, Out);
65180b57cec5SDimitry Andric LT = llvm::GlobalValue::LinkOnceODRLinkage;
65190b57cec5SDimitry Andric GlobalVariableName = MangledNameBuffer;
65200b57cec5SDimitry Andric } else {
65210b57cec5SDimitry Andric LT = llvm::GlobalValue::PrivateLinkage;
65220b57cec5SDimitry Andric GlobalVariableName = Name;
65230b57cec5SDimitry Andric }
65240b57cec5SDimitry Andric
65250b57cec5SDimitry Andric auto GV = GenerateStringLiteral(C, LT, *this, GlobalVariableName, Alignment);
652681ad6265SDimitry Andric
652781ad6265SDimitry Andric CGDebugInfo *DI = getModuleDebugInfo();
652881ad6265SDimitry Andric if (DI && getCodeGenOpts().hasReducedDebugInfo())
652981ad6265SDimitry Andric DI->AddStringLiteralDebugInfo(GV, S);
653081ad6265SDimitry Andric
65310b57cec5SDimitry Andric if (Entry)
65320b57cec5SDimitry Andric *Entry = GV;
65330b57cec5SDimitry Andric
653481ad6265SDimitry Andric SanitizerMD->reportGlobal(GV, S->getStrTokenLoc(0), "<string literal>");
65350b57cec5SDimitry Andric
65360b57cec5SDimitry Andric return ConstantAddress(castStringLiteralToDefaultAddressSpace(*this, GV),
65370eae32dcSDimitry Andric GV->getValueType(), Alignment);
65380b57cec5SDimitry Andric }
65390b57cec5SDimitry Andric
65400b57cec5SDimitry Andric /// GetAddrOfConstantStringFromObjCEncode - Return a pointer to a constant
65410b57cec5SDimitry Andric /// array for the given ObjCEncodeExpr node.
65420b57cec5SDimitry Andric ConstantAddress
GetAddrOfConstantStringFromObjCEncode(const ObjCEncodeExpr * E)65430b57cec5SDimitry Andric CodeGenModule::GetAddrOfConstantStringFromObjCEncode(const ObjCEncodeExpr *E) {
65440b57cec5SDimitry Andric std::string Str;
65450b57cec5SDimitry Andric getContext().getObjCEncodingForType(E->getEncodedType(), Str);
65460b57cec5SDimitry Andric
65470b57cec5SDimitry Andric return GetAddrOfConstantCString(Str);
65480b57cec5SDimitry Andric }
65490b57cec5SDimitry Andric
65500b57cec5SDimitry Andric /// GetAddrOfConstantCString - Returns a pointer to a character array containing
65510b57cec5SDimitry Andric /// the literal and a terminating '\0' character.
65520b57cec5SDimitry Andric /// The result has pointer to array type.
GetAddrOfConstantCString(const std::string & Str,const char * GlobalName)65530b57cec5SDimitry Andric ConstantAddress CodeGenModule::GetAddrOfConstantCString(
65540b57cec5SDimitry Andric const std::string &Str, const char *GlobalName) {
65550b57cec5SDimitry Andric StringRef StrWithNull(Str.c_str(), Str.size() + 1);
65560fca6ea1SDimitry Andric CharUnits Alignment = getContext().getAlignOfGlobalVarInChars(
65570fca6ea1SDimitry Andric getContext().CharTy, /*VD=*/nullptr);
65580b57cec5SDimitry Andric
65590b57cec5SDimitry Andric llvm::Constant *C =
65600b57cec5SDimitry Andric llvm::ConstantDataArray::getString(getLLVMContext(), StrWithNull, false);
65610b57cec5SDimitry Andric
65620b57cec5SDimitry Andric // Don't share any string literals if strings aren't constant.
65630b57cec5SDimitry Andric llvm::GlobalVariable **Entry = nullptr;
65640b57cec5SDimitry Andric if (!LangOpts.WritableStrings) {
65650b57cec5SDimitry Andric Entry = &ConstantStringMap[C];
65660b57cec5SDimitry Andric if (auto GV = *Entry) {
6567349cc55cSDimitry Andric if (uint64_t(Alignment.getQuantity()) > GV->getAlignment())
6568a7dea167SDimitry Andric GV->setAlignment(Alignment.getAsAlign());
65690b57cec5SDimitry Andric return ConstantAddress(castStringLiteralToDefaultAddressSpace(*this, GV),
65700eae32dcSDimitry Andric GV->getValueType(), Alignment);
65710b57cec5SDimitry Andric }
65720b57cec5SDimitry Andric }
65730b57cec5SDimitry Andric
65740b57cec5SDimitry Andric // Get the default prefix if a name wasn't specified.
65750b57cec5SDimitry Andric if (!GlobalName)
65760b57cec5SDimitry Andric GlobalName = ".str";
65770b57cec5SDimitry Andric // Create a global variable for this.
65780b57cec5SDimitry Andric auto GV = GenerateStringLiteral(C, llvm::GlobalValue::PrivateLinkage, *this,
65790b57cec5SDimitry Andric GlobalName, Alignment);
65800b57cec5SDimitry Andric if (Entry)
65810b57cec5SDimitry Andric *Entry = GV;
65820b57cec5SDimitry Andric
65830b57cec5SDimitry Andric return ConstantAddress(castStringLiteralToDefaultAddressSpace(*this, GV),
65840eae32dcSDimitry Andric GV->getValueType(), Alignment);
65850b57cec5SDimitry Andric }
65860b57cec5SDimitry Andric
GetAddrOfGlobalTemporary(const MaterializeTemporaryExpr * E,const Expr * Init)65870b57cec5SDimitry Andric ConstantAddress CodeGenModule::GetAddrOfGlobalTemporary(
65880b57cec5SDimitry Andric const MaterializeTemporaryExpr *E, const Expr *Init) {
65890b57cec5SDimitry Andric assert((E->getStorageDuration() == SD_Static ||
65900b57cec5SDimitry Andric E->getStorageDuration() == SD_Thread) && "not a global temporary");
65910b57cec5SDimitry Andric const auto *VD = cast<VarDecl>(E->getExtendingDecl());
65920b57cec5SDimitry Andric
65930b57cec5SDimitry Andric // If we're not materializing a subobject of the temporary, keep the
65940b57cec5SDimitry Andric // cv-qualifiers from the type of the MaterializeTemporaryExpr.
65950b57cec5SDimitry Andric QualType MaterializedType = Init->getType();
6596480093f4SDimitry Andric if (Init == E->getSubExpr())
65970b57cec5SDimitry Andric MaterializedType = E->getType();
65980b57cec5SDimitry Andric
65990b57cec5SDimitry Andric CharUnits Align = getContext().getTypeAlignInChars(MaterializedType);
66000b57cec5SDimitry Andric
6601fe6060f1SDimitry Andric auto InsertResult = MaterializedGlobalTemporaryMap.insert({E, nullptr});
6602fe6060f1SDimitry Andric if (!InsertResult.second) {
6603fe6060f1SDimitry Andric // We've seen this before: either we already created it or we're in the
6604fe6060f1SDimitry Andric // process of doing so.
6605fe6060f1SDimitry Andric if (!InsertResult.first->second) {
6606fe6060f1SDimitry Andric // We recursively re-entered this function, probably during emission of
6607fe6060f1SDimitry Andric // the initializer. Create a placeholder. We'll clean this up in the
6608fe6060f1SDimitry Andric // outer call, at the end of this function.
6609fe6060f1SDimitry Andric llvm::Type *Type = getTypes().ConvertTypeForMem(MaterializedType);
6610fe6060f1SDimitry Andric InsertResult.first->second = new llvm::GlobalVariable(
6611fe6060f1SDimitry Andric getModule(), Type, false, llvm::GlobalVariable::InternalLinkage,
6612fe6060f1SDimitry Andric nullptr);
6613fe6060f1SDimitry Andric }
661481ad6265SDimitry Andric return ConstantAddress(InsertResult.first->second,
661581ad6265SDimitry Andric llvm::cast<llvm::GlobalVariable>(
661681ad6265SDimitry Andric InsertResult.first->second->stripPointerCasts())
661781ad6265SDimitry Andric ->getValueType(),
661881ad6265SDimitry Andric Align);
6619fe6060f1SDimitry Andric }
66200b57cec5SDimitry Andric
66210b57cec5SDimitry Andric // FIXME: If an externally-visible declaration extends multiple temporaries,
66220b57cec5SDimitry Andric // we need to give each temporary the same name in every translation unit (and
66230b57cec5SDimitry Andric // we also need to make the temporaries externally-visible).
66240b57cec5SDimitry Andric SmallString<256> Name;
66250b57cec5SDimitry Andric llvm::raw_svector_ostream Out(Name);
66260b57cec5SDimitry Andric getCXXABI().getMangleContext().mangleReferenceTemporary(
66270b57cec5SDimitry Andric VD, E->getManglingNumber(), Out);
66280b57cec5SDimitry Andric
66290b57cec5SDimitry Andric APValue *Value = nullptr;
66305f757f3fSDimitry Andric if (E->getStorageDuration() == SD_Static && VD->evaluateValue()) {
6631a7dea167SDimitry Andric // If the initializer of the extending declaration is a constant
6632a7dea167SDimitry Andric // initializer, we should have a cached constant initializer for this
6633a7dea167SDimitry Andric // temporary. Note that this might have a different value from the value
6634a7dea167SDimitry Andric // computed by evaluating the initializer if the surrounding constant
6635a7dea167SDimitry Andric // expression modifies the temporary.
6636480093f4SDimitry Andric Value = E->getOrCreateValue(false);
66370b57cec5SDimitry Andric }
66380b57cec5SDimitry Andric
66390b57cec5SDimitry Andric // Try evaluating it now, it might have a constant initializer.
66400b57cec5SDimitry Andric Expr::EvalResult EvalResult;
66410b57cec5SDimitry Andric if (!Value && Init->EvaluateAsRValue(EvalResult, getContext()) &&
66420b57cec5SDimitry Andric !EvalResult.hasSideEffects())
66430b57cec5SDimitry Andric Value = &EvalResult.Val;
66440b57cec5SDimitry Andric
66455f757f3fSDimitry Andric LangAS AddrSpace = GetGlobalVarAddressSpace(VD);
66460b57cec5SDimitry Andric
6647bdd1243dSDimitry Andric std::optional<ConstantEmitter> emitter;
66480b57cec5SDimitry Andric llvm::Constant *InitialValue = nullptr;
66490b57cec5SDimitry Andric bool Constant = false;
66500b57cec5SDimitry Andric llvm::Type *Type;
66510b57cec5SDimitry Andric if (Value) {
66520b57cec5SDimitry Andric // The temporary has a constant initializer, use it.
66530b57cec5SDimitry Andric emitter.emplace(*this);
66540b57cec5SDimitry Andric InitialValue = emitter->emitForInitializer(*Value, AddrSpace,
66550b57cec5SDimitry Andric MaterializedType);
66565f757f3fSDimitry Andric Constant =
66575f757f3fSDimitry Andric MaterializedType.isConstantStorage(getContext(), /*ExcludeCtor*/ Value,
665806c3fb27SDimitry Andric /*ExcludeDtor*/ false);
66590b57cec5SDimitry Andric Type = InitialValue->getType();
66600b57cec5SDimitry Andric } else {
66610b57cec5SDimitry Andric // No initializer, the initialization will be provided when we
66620b57cec5SDimitry Andric // initialize the declaration which performed lifetime extension.
66630b57cec5SDimitry Andric Type = getTypes().ConvertTypeForMem(MaterializedType);
66640b57cec5SDimitry Andric }
66650b57cec5SDimitry Andric
66660b57cec5SDimitry Andric // Create a global variable for this lifetime-extended temporary.
66678a4dda33SDimitry Andric llvm::GlobalValue::LinkageTypes Linkage = getLLVMLinkageVarDefinition(VD);
66680b57cec5SDimitry Andric if (Linkage == llvm::GlobalVariable::ExternalLinkage) {
66690b57cec5SDimitry Andric const VarDecl *InitVD;
66700b57cec5SDimitry Andric if (VD->isStaticDataMember() && VD->getAnyInitializer(InitVD) &&
66710b57cec5SDimitry Andric isa<CXXRecordDecl>(InitVD->getLexicalDeclContext())) {
66720b57cec5SDimitry Andric // Temporaries defined inside a class get linkonce_odr linkage because the
66730b57cec5SDimitry Andric // class can be defined in multiple translation units.
66740b57cec5SDimitry Andric Linkage = llvm::GlobalVariable::LinkOnceODRLinkage;
66750b57cec5SDimitry Andric } else {
66760b57cec5SDimitry Andric // There is no need for this temporary to have external linkage if the
66770b57cec5SDimitry Andric // VarDecl has external linkage.
66780b57cec5SDimitry Andric Linkage = llvm::GlobalVariable::InternalLinkage;
66790b57cec5SDimitry Andric }
66800b57cec5SDimitry Andric }
66810b57cec5SDimitry Andric auto TargetAS = getContext().getTargetAddressSpace(AddrSpace);
66820b57cec5SDimitry Andric auto *GV = new llvm::GlobalVariable(
66830b57cec5SDimitry Andric getModule(), Type, Constant, Linkage, InitialValue, Name.c_str(),
66840b57cec5SDimitry Andric /*InsertBefore=*/nullptr, llvm::GlobalVariable::NotThreadLocal, TargetAS);
66850b57cec5SDimitry Andric if (emitter) emitter->finalize(GV);
6686bdd1243dSDimitry Andric // Don't assign dllimport or dllexport to local linkage globals.
6687bdd1243dSDimitry Andric if (!llvm::GlobalValue::isLocalLinkage(Linkage)) {
66880b57cec5SDimitry Andric setGVProperties(GV, VD);
668981ad6265SDimitry Andric if (GV->getDLLStorageClass() == llvm::GlobalVariable::DLLExportStorageClass)
669081ad6265SDimitry Andric // The reference temporary should never be dllexport.
669181ad6265SDimitry Andric GV->setDLLStorageClass(llvm::GlobalVariable::DefaultStorageClass);
6692bdd1243dSDimitry Andric }
6693a7dea167SDimitry Andric GV->setAlignment(Align.getAsAlign());
66940b57cec5SDimitry Andric if (supportsCOMDAT() && GV->isWeakForLinker())
66950b57cec5SDimitry Andric GV->setComdat(TheModule.getOrInsertComdat(GV->getName()));
66960b57cec5SDimitry Andric if (VD->getTLSKind())
66970b57cec5SDimitry Andric setTLSMode(GV, *VD);
66980b57cec5SDimitry Andric llvm::Constant *CV = GV;
66990b57cec5SDimitry Andric if (AddrSpace != LangAS::Default)
67000b57cec5SDimitry Andric CV = getTargetCodeGenInfo().performAddrSpaceCast(
67010b57cec5SDimitry Andric *this, GV, AddrSpace, LangAS::Default,
67025f757f3fSDimitry Andric llvm::PointerType::get(
67035f757f3fSDimitry Andric getLLVMContext(),
67040b57cec5SDimitry Andric getContext().getTargetAddressSpace(LangAS::Default)));
6705fe6060f1SDimitry Andric
6706fe6060f1SDimitry Andric // Update the map with the new temporary. If we created a placeholder above,
6707fe6060f1SDimitry Andric // replace it with the new global now.
6708fe6060f1SDimitry Andric llvm::Constant *&Entry = MaterializedGlobalTemporaryMap[E];
6709fe6060f1SDimitry Andric if (Entry) {
67105f757f3fSDimitry Andric Entry->replaceAllUsesWith(CV);
6711fe6060f1SDimitry Andric llvm::cast<llvm::GlobalVariable>(Entry)->eraseFromParent();
6712fe6060f1SDimitry Andric }
6713fe6060f1SDimitry Andric Entry = CV;
6714fe6060f1SDimitry Andric
67150eae32dcSDimitry Andric return ConstantAddress(CV, Type, Align);
67160b57cec5SDimitry Andric }
67170b57cec5SDimitry Andric
67180b57cec5SDimitry Andric /// EmitObjCPropertyImplementations - Emit information for synthesized
67190b57cec5SDimitry Andric /// properties for an implementation.
EmitObjCPropertyImplementations(const ObjCImplementationDecl * D)67200b57cec5SDimitry Andric void CodeGenModule::EmitObjCPropertyImplementations(const
67210b57cec5SDimitry Andric ObjCImplementationDecl *D) {
67220b57cec5SDimitry Andric for (const auto *PID : D->property_impls()) {
67230b57cec5SDimitry Andric // Dynamic is just for type-checking.
67240b57cec5SDimitry Andric if (PID->getPropertyImplementation() == ObjCPropertyImplDecl::Synthesize) {
67250b57cec5SDimitry Andric ObjCPropertyDecl *PD = PID->getPropertyDecl();
67260b57cec5SDimitry Andric
67270b57cec5SDimitry Andric // Determine which methods need to be implemented, some may have
67280b57cec5SDimitry Andric // been overridden. Note that ::isPropertyAccessor is not the method
67290b57cec5SDimitry Andric // we want, that just indicates if the decl came from a
67300b57cec5SDimitry Andric // property. What we want to know is if the method is defined in
67310b57cec5SDimitry Andric // this implementation.
6732480093f4SDimitry Andric auto *Getter = PID->getGetterMethodDecl();
6733480093f4SDimitry Andric if (!Getter || Getter->isSynthesizedAccessorStub())
67340b57cec5SDimitry Andric CodeGenFunction(*this).GenerateObjCGetter(
67350b57cec5SDimitry Andric const_cast<ObjCImplementationDecl *>(D), PID);
6736480093f4SDimitry Andric auto *Setter = PID->getSetterMethodDecl();
6737480093f4SDimitry Andric if (!PD->isReadOnly() && (!Setter || Setter->isSynthesizedAccessorStub()))
67380b57cec5SDimitry Andric CodeGenFunction(*this).GenerateObjCSetter(
67390b57cec5SDimitry Andric const_cast<ObjCImplementationDecl *>(D), PID);
67400b57cec5SDimitry Andric }
67410b57cec5SDimitry Andric }
67420b57cec5SDimitry Andric }
67430b57cec5SDimitry Andric
needsDestructMethod(ObjCImplementationDecl * impl)67440b57cec5SDimitry Andric static bool needsDestructMethod(ObjCImplementationDecl *impl) {
67450b57cec5SDimitry Andric const ObjCInterfaceDecl *iface = impl->getClassInterface();
67460b57cec5SDimitry Andric for (const ObjCIvarDecl *ivar = iface->all_declared_ivar_begin();
67470b57cec5SDimitry Andric ivar; ivar = ivar->getNextIvar())
67480b57cec5SDimitry Andric if (ivar->getType().isDestructedType())
67490b57cec5SDimitry Andric return true;
67500b57cec5SDimitry Andric
67510b57cec5SDimitry Andric return false;
67520b57cec5SDimitry Andric }
67530b57cec5SDimitry Andric
AllTrivialInitializers(CodeGenModule & CGM,ObjCImplementationDecl * D)67540b57cec5SDimitry Andric static bool AllTrivialInitializers(CodeGenModule &CGM,
67550b57cec5SDimitry Andric ObjCImplementationDecl *D) {
67560b57cec5SDimitry Andric CodeGenFunction CGF(CGM);
67570b57cec5SDimitry Andric for (ObjCImplementationDecl::init_iterator B = D->init_begin(),
67580b57cec5SDimitry Andric E = D->init_end(); B != E; ++B) {
67590b57cec5SDimitry Andric CXXCtorInitializer *CtorInitExp = *B;
67600b57cec5SDimitry Andric Expr *Init = CtorInitExp->getInit();
67610b57cec5SDimitry Andric if (!CGF.isTrivialInitializer(Init))
67620b57cec5SDimitry Andric return false;
67630b57cec5SDimitry Andric }
67640b57cec5SDimitry Andric return true;
67650b57cec5SDimitry Andric }
67660b57cec5SDimitry Andric
67670b57cec5SDimitry Andric /// EmitObjCIvarInitializations - Emit information for ivar initialization
67680b57cec5SDimitry Andric /// for an implementation.
EmitObjCIvarInitializations(ObjCImplementationDecl * D)67690b57cec5SDimitry Andric void CodeGenModule::EmitObjCIvarInitializations(ObjCImplementationDecl *D) {
67700b57cec5SDimitry Andric // We might need a .cxx_destruct even if we don't have any ivar initializers.
67710b57cec5SDimitry Andric if (needsDestructMethod(D)) {
67720fca6ea1SDimitry Andric const IdentifierInfo *II = &getContext().Idents.get(".cxx_destruct");
67730b57cec5SDimitry Andric Selector cxxSelector = getContext().Selectors.getSelector(0, &II);
6774480093f4SDimitry Andric ObjCMethodDecl *DTORMethod = ObjCMethodDecl::Create(
6775480093f4SDimitry Andric getContext(), D->getLocation(), D->getLocation(), cxxSelector,
6776480093f4SDimitry Andric getContext().VoidTy, nullptr, D,
67770b57cec5SDimitry Andric /*isInstance=*/true, /*isVariadic=*/false,
6778480093f4SDimitry Andric /*isPropertyAccessor=*/true, /*isSynthesizedAccessorStub=*/false,
6779480093f4SDimitry Andric /*isImplicitlyDeclared=*/true,
67805f757f3fSDimitry Andric /*isDefined=*/false, ObjCImplementationControl::Required);
67810b57cec5SDimitry Andric D->addInstanceMethod(DTORMethod);
67820b57cec5SDimitry Andric CodeGenFunction(*this).GenerateObjCCtorDtorMethod(D, DTORMethod, false);
67830b57cec5SDimitry Andric D->setHasDestructors(true);
67840b57cec5SDimitry Andric }
67850b57cec5SDimitry Andric
67860b57cec5SDimitry Andric // If the implementation doesn't have any ivar initializers, we don't need
67870b57cec5SDimitry Andric // a .cxx_construct.
67880b57cec5SDimitry Andric if (D->getNumIvarInitializers() == 0 ||
67890b57cec5SDimitry Andric AllTrivialInitializers(*this, D))
67900b57cec5SDimitry Andric return;
67910b57cec5SDimitry Andric
67920fca6ea1SDimitry Andric const IdentifierInfo *II = &getContext().Idents.get(".cxx_construct");
67930b57cec5SDimitry Andric Selector cxxSelector = getContext().Selectors.getSelector(0, &II);
67940b57cec5SDimitry Andric // The constructor returns 'self'.
6795480093f4SDimitry Andric ObjCMethodDecl *CTORMethod = ObjCMethodDecl::Create(
6796480093f4SDimitry Andric getContext(), D->getLocation(), D->getLocation(), cxxSelector,
6797480093f4SDimitry Andric getContext().getObjCIdType(), nullptr, D, /*isInstance=*/true,
67980b57cec5SDimitry Andric /*isVariadic=*/false,
6799480093f4SDimitry Andric /*isPropertyAccessor=*/true, /*isSynthesizedAccessorStub=*/false,
68000b57cec5SDimitry Andric /*isImplicitlyDeclared=*/true,
68015f757f3fSDimitry Andric /*isDefined=*/false, ObjCImplementationControl::Required);
68020b57cec5SDimitry Andric D->addInstanceMethod(CTORMethod);
68030b57cec5SDimitry Andric CodeGenFunction(*this).GenerateObjCCtorDtorMethod(D, CTORMethod, true);
68040b57cec5SDimitry Andric D->setHasNonZeroConstructors(true);
68050b57cec5SDimitry Andric }
68060b57cec5SDimitry Andric
68070b57cec5SDimitry Andric // EmitLinkageSpec - Emit all declarations in a linkage spec.
EmitLinkageSpec(const LinkageSpecDecl * LSD)68080b57cec5SDimitry Andric void CodeGenModule::EmitLinkageSpec(const LinkageSpecDecl *LSD) {
68095f757f3fSDimitry Andric if (LSD->getLanguage() != LinkageSpecLanguageIDs::C &&
68105f757f3fSDimitry Andric LSD->getLanguage() != LinkageSpecLanguageIDs::CXX) {
68110b57cec5SDimitry Andric ErrorUnsupported(LSD, "linkage spec");
68120b57cec5SDimitry Andric return;
68130b57cec5SDimitry Andric }
68140b57cec5SDimitry Andric
68150b57cec5SDimitry Andric EmitDeclContext(LSD);
68160b57cec5SDimitry Andric }
68170b57cec5SDimitry Andric
EmitTopLevelStmt(const TopLevelStmtDecl * D)6818bdd1243dSDimitry Andric void CodeGenModule::EmitTopLevelStmt(const TopLevelStmtDecl *D) {
681906c3fb27SDimitry Andric // Device code should not be at top level.
682006c3fb27SDimitry Andric if (LangOpts.CUDA && LangOpts.CUDAIsDevice)
682106c3fb27SDimitry Andric return;
682206c3fb27SDimitry Andric
6823bdd1243dSDimitry Andric std::unique_ptr<CodeGenFunction> &CurCGF =
6824bdd1243dSDimitry Andric GlobalTopLevelStmtBlockInFlight.first;
6825bdd1243dSDimitry Andric
6826bdd1243dSDimitry Andric // We emitted a top-level stmt but after it there is initialization.
6827bdd1243dSDimitry Andric // Stop squashing the top-level stmts into a single function.
6828bdd1243dSDimitry Andric if (CurCGF && CXXGlobalInits.back() != CurCGF->CurFn) {
6829bdd1243dSDimitry Andric CurCGF->FinishFunction(D->getEndLoc());
6830bdd1243dSDimitry Andric CurCGF = nullptr;
6831bdd1243dSDimitry Andric }
6832bdd1243dSDimitry Andric
6833bdd1243dSDimitry Andric if (!CurCGF) {
6834bdd1243dSDimitry Andric // void __stmts__N(void)
6835bdd1243dSDimitry Andric // FIXME: Ask the ABI name mangler to pick a name.
6836bdd1243dSDimitry Andric std::string Name = "__stmts__" + llvm::utostr(CXXGlobalInits.size());
6837bdd1243dSDimitry Andric FunctionArgList Args;
6838bdd1243dSDimitry Andric QualType RetTy = getContext().VoidTy;
6839bdd1243dSDimitry Andric const CGFunctionInfo &FnInfo =
6840bdd1243dSDimitry Andric getTypes().arrangeBuiltinFunctionDeclaration(RetTy, Args);
6841bdd1243dSDimitry Andric llvm::FunctionType *FnTy = getTypes().GetFunctionType(FnInfo);
6842bdd1243dSDimitry Andric llvm::Function *Fn = llvm::Function::Create(
6843bdd1243dSDimitry Andric FnTy, llvm::GlobalValue::InternalLinkage, Name, &getModule());
6844bdd1243dSDimitry Andric
6845bdd1243dSDimitry Andric CurCGF.reset(new CodeGenFunction(*this));
6846bdd1243dSDimitry Andric GlobalTopLevelStmtBlockInFlight.second = D;
6847bdd1243dSDimitry Andric CurCGF->StartFunction(GlobalDecl(), RetTy, Fn, FnInfo, Args,
6848bdd1243dSDimitry Andric D->getBeginLoc(), D->getBeginLoc());
6849bdd1243dSDimitry Andric CXXGlobalInits.push_back(Fn);
6850bdd1243dSDimitry Andric }
6851bdd1243dSDimitry Andric
6852bdd1243dSDimitry Andric CurCGF->EmitStmt(D->getStmt());
6853bdd1243dSDimitry Andric }
6854bdd1243dSDimitry Andric
EmitDeclContext(const DeclContext * DC)68550b57cec5SDimitry Andric void CodeGenModule::EmitDeclContext(const DeclContext *DC) {
68560b57cec5SDimitry Andric for (auto *I : DC->decls()) {
68570b57cec5SDimitry Andric // Unlike other DeclContexts, the contents of an ObjCImplDecl at TU scope
68580b57cec5SDimitry Andric // are themselves considered "top-level", so EmitTopLevelDecl on an
68590b57cec5SDimitry Andric // ObjCImplDecl does not recursively visit them. We need to do that in
68600b57cec5SDimitry Andric // case they're nested inside another construct (LinkageSpecDecl /
68610b57cec5SDimitry Andric // ExportDecl) that does stop them from being considered "top-level".
68620b57cec5SDimitry Andric if (auto *OID = dyn_cast<ObjCImplDecl>(I)) {
68630b57cec5SDimitry Andric for (auto *M : OID->methods())
68640b57cec5SDimitry Andric EmitTopLevelDecl(M);
68650b57cec5SDimitry Andric }
68660b57cec5SDimitry Andric
68670b57cec5SDimitry Andric EmitTopLevelDecl(I);
68680b57cec5SDimitry Andric }
68690b57cec5SDimitry Andric }
68700b57cec5SDimitry Andric
68710b57cec5SDimitry Andric /// EmitTopLevelDecl - Emit code for a single top level declaration.
EmitTopLevelDecl(Decl * D)68720b57cec5SDimitry Andric void CodeGenModule::EmitTopLevelDecl(Decl *D) {
68730b57cec5SDimitry Andric // Ignore dependent declarations.
68740b57cec5SDimitry Andric if (D->isTemplated())
68750b57cec5SDimitry Andric return;
68760b57cec5SDimitry Andric
68775ffd83dbSDimitry Andric // Consteval function shouldn't be emitted.
687806c3fb27SDimitry Andric if (auto *FD = dyn_cast<FunctionDecl>(D); FD && FD->isImmediateFunction())
68795ffd83dbSDimitry Andric return;
68805ffd83dbSDimitry Andric
68810b57cec5SDimitry Andric switch (D->getKind()) {
68820b57cec5SDimitry Andric case Decl::CXXConversion:
68830b57cec5SDimitry Andric case Decl::CXXMethod:
68840b57cec5SDimitry Andric case Decl::Function:
68850b57cec5SDimitry Andric EmitGlobal(cast<FunctionDecl>(D));
68860b57cec5SDimitry Andric // Always provide some coverage mapping
68870b57cec5SDimitry Andric // even for the functions that aren't emitted.
68880b57cec5SDimitry Andric AddDeferredUnusedCoverageMapping(D);
68890b57cec5SDimitry Andric break;
68900b57cec5SDimitry Andric
68910b57cec5SDimitry Andric case Decl::CXXDeductionGuide:
68920b57cec5SDimitry Andric // Function-like, but does not result in code emission.
68930b57cec5SDimitry Andric break;
68940b57cec5SDimitry Andric
68950b57cec5SDimitry Andric case Decl::Var:
68960b57cec5SDimitry Andric case Decl::Decomposition:
68970b57cec5SDimitry Andric case Decl::VarTemplateSpecialization:
68980b57cec5SDimitry Andric EmitGlobal(cast<VarDecl>(D));
68990b57cec5SDimitry Andric if (auto *DD = dyn_cast<DecompositionDecl>(D))
69000b57cec5SDimitry Andric for (auto *B : DD->bindings())
69010b57cec5SDimitry Andric if (auto *HD = B->getHoldingVar())
69020b57cec5SDimitry Andric EmitGlobal(HD);
69030b57cec5SDimitry Andric break;
69040b57cec5SDimitry Andric
69050b57cec5SDimitry Andric // Indirect fields from global anonymous structs and unions can be
69060b57cec5SDimitry Andric // ignored; only the actual variable requires IR gen support.
69070b57cec5SDimitry Andric case Decl::IndirectField:
69080b57cec5SDimitry Andric break;
69090b57cec5SDimitry Andric
69100b57cec5SDimitry Andric // C++ Decls
69110b57cec5SDimitry Andric case Decl::Namespace:
69120b57cec5SDimitry Andric EmitDeclContext(cast<NamespaceDecl>(D));
69130b57cec5SDimitry Andric break;
69140b57cec5SDimitry Andric case Decl::ClassTemplateSpecialization: {
69150b57cec5SDimitry Andric const auto *Spec = cast<ClassTemplateSpecializationDecl>(D);
69165ffd83dbSDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
69175ffd83dbSDimitry Andric if (Spec->getSpecializationKind() ==
69185ffd83dbSDimitry Andric TSK_ExplicitInstantiationDefinition &&
69190b57cec5SDimitry Andric Spec->hasDefinition())
69205ffd83dbSDimitry Andric DI->completeTemplateDefinition(*Spec);
6921bdd1243dSDimitry Andric } [[fallthrough]];
6922e8d8bef9SDimitry Andric case Decl::CXXRecord: {
6923e8d8bef9SDimitry Andric CXXRecordDecl *CRD = cast<CXXRecordDecl>(D);
6924e8d8bef9SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo()) {
6925e8d8bef9SDimitry Andric if (CRD->hasDefinition())
6926e8d8bef9SDimitry Andric DI->EmitAndRetainType(getContext().getRecordType(cast<RecordDecl>(D)));
69270b57cec5SDimitry Andric if (auto *ES = D->getASTContext().getExternalSource())
69280b57cec5SDimitry Andric if (ES->hasExternalDefinitions(D) == ExternalASTSource::EK_Never)
6929e8d8bef9SDimitry Andric DI->completeUnusedClass(*CRD);
6930e8d8bef9SDimitry Andric }
69310b57cec5SDimitry Andric // Emit any static data members, they may be definitions.
6932e8d8bef9SDimitry Andric for (auto *I : CRD->decls())
69330b57cec5SDimitry Andric if (isa<VarDecl>(I) || isa<CXXRecordDecl>(I))
69340b57cec5SDimitry Andric EmitTopLevelDecl(I);
69350b57cec5SDimitry Andric break;
6936e8d8bef9SDimitry Andric }
69370b57cec5SDimitry Andric // No code generation needed.
69380b57cec5SDimitry Andric case Decl::UsingShadow:
69390b57cec5SDimitry Andric case Decl::ClassTemplate:
69400b57cec5SDimitry Andric case Decl::VarTemplate:
69410b57cec5SDimitry Andric case Decl::Concept:
69420b57cec5SDimitry Andric case Decl::VarTemplatePartialSpecialization:
69430b57cec5SDimitry Andric case Decl::FunctionTemplate:
69440b57cec5SDimitry Andric case Decl::TypeAliasTemplate:
69450b57cec5SDimitry Andric case Decl::Block:
69460b57cec5SDimitry Andric case Decl::Empty:
69470b57cec5SDimitry Andric case Decl::Binding:
69480b57cec5SDimitry Andric break;
69490b57cec5SDimitry Andric case Decl::Using: // using X; [C++]
69500b57cec5SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
69510b57cec5SDimitry Andric DI->EmitUsingDecl(cast<UsingDecl>(*D));
69525ffd83dbSDimitry Andric break;
6953fe6060f1SDimitry Andric case Decl::UsingEnum: // using enum X; [C++]
6954fe6060f1SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
6955fe6060f1SDimitry Andric DI->EmitUsingEnumDecl(cast<UsingEnumDecl>(*D));
6956fe6060f1SDimitry Andric break;
69570b57cec5SDimitry Andric case Decl::NamespaceAlias:
69580b57cec5SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
69590b57cec5SDimitry Andric DI->EmitNamespaceAlias(cast<NamespaceAliasDecl>(*D));
69605ffd83dbSDimitry Andric break;
69610b57cec5SDimitry Andric case Decl::UsingDirective: // using namespace X; [C++]
69620b57cec5SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
69630b57cec5SDimitry Andric DI->EmitUsingDirective(cast<UsingDirectiveDecl>(*D));
69645ffd83dbSDimitry Andric break;
69650b57cec5SDimitry Andric case Decl::CXXConstructor:
69660b57cec5SDimitry Andric getCXXABI().EmitCXXConstructors(cast<CXXConstructorDecl>(D));
69670b57cec5SDimitry Andric break;
69680b57cec5SDimitry Andric case Decl::CXXDestructor:
69690b57cec5SDimitry Andric getCXXABI().EmitCXXDestructors(cast<CXXDestructorDecl>(D));
69700b57cec5SDimitry Andric break;
69710b57cec5SDimitry Andric
69720b57cec5SDimitry Andric case Decl::StaticAssert:
69730b57cec5SDimitry Andric // Nothing to do.
69740b57cec5SDimitry Andric break;
69750b57cec5SDimitry Andric
69760b57cec5SDimitry Andric // Objective-C Decls
69770b57cec5SDimitry Andric
69780b57cec5SDimitry Andric // Forward declarations, no (immediate) code generation.
69790b57cec5SDimitry Andric case Decl::ObjCInterface:
69800b57cec5SDimitry Andric case Decl::ObjCCategory:
69810b57cec5SDimitry Andric break;
69820b57cec5SDimitry Andric
69830b57cec5SDimitry Andric case Decl::ObjCProtocol: {
69840b57cec5SDimitry Andric auto *Proto = cast<ObjCProtocolDecl>(D);
69850b57cec5SDimitry Andric if (Proto->isThisDeclarationADefinition())
69860b57cec5SDimitry Andric ObjCRuntime->GenerateProtocol(Proto);
69870b57cec5SDimitry Andric break;
69880b57cec5SDimitry Andric }
69890b57cec5SDimitry Andric
69900b57cec5SDimitry Andric case Decl::ObjCCategoryImpl:
69910b57cec5SDimitry Andric // Categories have properties but don't support synthesize so we
69920b57cec5SDimitry Andric // can ignore them here.
69930b57cec5SDimitry Andric ObjCRuntime->GenerateCategory(cast<ObjCCategoryImplDecl>(D));
69940b57cec5SDimitry Andric break;
69950b57cec5SDimitry Andric
69960b57cec5SDimitry Andric case Decl::ObjCImplementation: {
69970b57cec5SDimitry Andric auto *OMD = cast<ObjCImplementationDecl>(D);
69980b57cec5SDimitry Andric EmitObjCPropertyImplementations(OMD);
69990b57cec5SDimitry Andric EmitObjCIvarInitializations(OMD);
70000b57cec5SDimitry Andric ObjCRuntime->GenerateClass(OMD);
70010b57cec5SDimitry Andric // Emit global variable debug information.
70020b57cec5SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
7003480093f4SDimitry Andric if (getCodeGenOpts().hasReducedDebugInfo())
70040b57cec5SDimitry Andric DI->getOrCreateInterfaceType(getContext().getObjCInterfaceType(
70050b57cec5SDimitry Andric OMD->getClassInterface()), OMD->getLocation());
70060b57cec5SDimitry Andric break;
70070b57cec5SDimitry Andric }
70080b57cec5SDimitry Andric case Decl::ObjCMethod: {
70090b57cec5SDimitry Andric auto *OMD = cast<ObjCMethodDecl>(D);
70100b57cec5SDimitry Andric // If this is not a prototype, emit the body.
70110b57cec5SDimitry Andric if (OMD->getBody())
70120b57cec5SDimitry Andric CodeGenFunction(*this).GenerateObjCMethod(OMD);
70130b57cec5SDimitry Andric break;
70140b57cec5SDimitry Andric }
70150b57cec5SDimitry Andric case Decl::ObjCCompatibleAlias:
70160b57cec5SDimitry Andric ObjCRuntime->RegisterAlias(cast<ObjCCompatibleAliasDecl>(D));
70170b57cec5SDimitry Andric break;
70180b57cec5SDimitry Andric
70190b57cec5SDimitry Andric case Decl::PragmaComment: {
70200b57cec5SDimitry Andric const auto *PCD = cast<PragmaCommentDecl>(D);
70210b57cec5SDimitry Andric switch (PCD->getCommentKind()) {
70220b57cec5SDimitry Andric case PCK_Unknown:
70230b57cec5SDimitry Andric llvm_unreachable("unexpected pragma comment kind");
70240b57cec5SDimitry Andric case PCK_Linker:
70250b57cec5SDimitry Andric AppendLinkerOptions(PCD->getArg());
70260b57cec5SDimitry Andric break;
70270b57cec5SDimitry Andric case PCK_Lib:
70280b57cec5SDimitry Andric AddDependentLib(PCD->getArg());
70290b57cec5SDimitry Andric break;
70300b57cec5SDimitry Andric case PCK_Compiler:
70310b57cec5SDimitry Andric case PCK_ExeStr:
70320b57cec5SDimitry Andric case PCK_User:
70330b57cec5SDimitry Andric break; // We ignore all of these.
70340b57cec5SDimitry Andric }
70350b57cec5SDimitry Andric break;
70360b57cec5SDimitry Andric }
70370b57cec5SDimitry Andric
70380b57cec5SDimitry Andric case Decl::PragmaDetectMismatch: {
70390b57cec5SDimitry Andric const auto *PDMD = cast<PragmaDetectMismatchDecl>(D);
70400b57cec5SDimitry Andric AddDetectMismatch(PDMD->getName(), PDMD->getValue());
70410b57cec5SDimitry Andric break;
70420b57cec5SDimitry Andric }
70430b57cec5SDimitry Andric
70440b57cec5SDimitry Andric case Decl::LinkageSpec:
70450b57cec5SDimitry Andric EmitLinkageSpec(cast<LinkageSpecDecl>(D));
70460b57cec5SDimitry Andric break;
70470b57cec5SDimitry Andric
70480b57cec5SDimitry Andric case Decl::FileScopeAsm: {
70490b57cec5SDimitry Andric // File-scope asm is ignored during device-side CUDA compilation.
70500b57cec5SDimitry Andric if (LangOpts.CUDA && LangOpts.CUDAIsDevice)
70510b57cec5SDimitry Andric break;
70520b57cec5SDimitry Andric // File-scope asm is ignored during device-side OpenMP compilation.
705306c3fb27SDimitry Andric if (LangOpts.OpenMPIsTargetDevice)
70540b57cec5SDimitry Andric break;
7055fe6060f1SDimitry Andric // File-scope asm is ignored during device-side SYCL compilation.
7056fe6060f1SDimitry Andric if (LangOpts.SYCLIsDevice)
7057fe6060f1SDimitry Andric break;
70580b57cec5SDimitry Andric auto *AD = cast<FileScopeAsmDecl>(D);
70590b57cec5SDimitry Andric getModule().appendModuleInlineAsm(AD->getAsmString()->getString());
70600b57cec5SDimitry Andric break;
70610b57cec5SDimitry Andric }
70620b57cec5SDimitry Andric
7063bdd1243dSDimitry Andric case Decl::TopLevelStmt:
7064bdd1243dSDimitry Andric EmitTopLevelStmt(cast<TopLevelStmtDecl>(D));
7065bdd1243dSDimitry Andric break;
7066bdd1243dSDimitry Andric
70670b57cec5SDimitry Andric case Decl::Import: {
70680b57cec5SDimitry Andric auto *Import = cast<ImportDecl>(D);
70690b57cec5SDimitry Andric
70700b57cec5SDimitry Andric // If we've already imported this module, we're done.
70710b57cec5SDimitry Andric if (!ImportedModules.insert(Import->getImportedModule()))
70720b57cec5SDimitry Andric break;
70730b57cec5SDimitry Andric
70740b57cec5SDimitry Andric // Emit debug information for direct imports.
70750b57cec5SDimitry Andric if (!Import->getImportedOwningModule()) {
70760b57cec5SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
70770b57cec5SDimitry Andric DI->EmitImportDecl(*Import);
70780b57cec5SDimitry Andric }
70790b57cec5SDimitry Andric
7080fcaf7f86SDimitry Andric // For C++ standard modules we are done - we will call the module
7081fcaf7f86SDimitry Andric // initializer for imported modules, and that will likewise call those for
7082fcaf7f86SDimitry Andric // any imports it has.
7083*415efcecSDimitry Andric if (CXX20ModuleInits && Import->getImportedModule() &&
7084*415efcecSDimitry Andric Import->getImportedModule()->isNamedModule())
7085fcaf7f86SDimitry Andric break;
7086fcaf7f86SDimitry Andric
7087fcaf7f86SDimitry Andric // For clang C++ module map modules the initializers for sub-modules are
7088fcaf7f86SDimitry Andric // emitted here.
7089fcaf7f86SDimitry Andric
70900b57cec5SDimitry Andric // Find all of the submodules and emit the module initializers.
70910b57cec5SDimitry Andric llvm::SmallPtrSet<clang::Module *, 16> Visited;
70920b57cec5SDimitry Andric SmallVector<clang::Module *, 16> Stack;
70930b57cec5SDimitry Andric Visited.insert(Import->getImportedModule());
70940b57cec5SDimitry Andric Stack.push_back(Import->getImportedModule());
70950b57cec5SDimitry Andric
70960b57cec5SDimitry Andric while (!Stack.empty()) {
70970b57cec5SDimitry Andric clang::Module *Mod = Stack.pop_back_val();
70980b57cec5SDimitry Andric if (!EmittedModuleInitializers.insert(Mod).second)
70990b57cec5SDimitry Andric continue;
71000b57cec5SDimitry Andric
71010b57cec5SDimitry Andric for (auto *D : Context.getModuleInitializers(Mod))
71020b57cec5SDimitry Andric EmitTopLevelDecl(D);
71030b57cec5SDimitry Andric
71040b57cec5SDimitry Andric // Visit the submodules of this module.
710506c3fb27SDimitry Andric for (auto *Submodule : Mod->submodules()) {
71060b57cec5SDimitry Andric // Skip explicit children; they need to be explicitly imported to emit
71070b57cec5SDimitry Andric // the initializers.
710806c3fb27SDimitry Andric if (Submodule->IsExplicit)
71090b57cec5SDimitry Andric continue;
71100b57cec5SDimitry Andric
711106c3fb27SDimitry Andric if (Visited.insert(Submodule).second)
711206c3fb27SDimitry Andric Stack.push_back(Submodule);
71130b57cec5SDimitry Andric }
71140b57cec5SDimitry Andric }
71150b57cec5SDimitry Andric break;
71160b57cec5SDimitry Andric }
71170b57cec5SDimitry Andric
71180b57cec5SDimitry Andric case Decl::Export:
71190b57cec5SDimitry Andric EmitDeclContext(cast<ExportDecl>(D));
71200b57cec5SDimitry Andric break;
71210b57cec5SDimitry Andric
71220b57cec5SDimitry Andric case Decl::OMPThreadPrivate:
71230b57cec5SDimitry Andric EmitOMPThreadPrivateDecl(cast<OMPThreadPrivateDecl>(D));
71240b57cec5SDimitry Andric break;
71250b57cec5SDimitry Andric
71260b57cec5SDimitry Andric case Decl::OMPAllocate:
7127fe6060f1SDimitry Andric EmitOMPAllocateDecl(cast<OMPAllocateDecl>(D));
71280b57cec5SDimitry Andric break;
71290b57cec5SDimitry Andric
71300b57cec5SDimitry Andric case Decl::OMPDeclareReduction:
71310b57cec5SDimitry Andric EmitOMPDeclareReduction(cast<OMPDeclareReductionDecl>(D));
71320b57cec5SDimitry Andric break;
71330b57cec5SDimitry Andric
71340b57cec5SDimitry Andric case Decl::OMPDeclareMapper:
71350b57cec5SDimitry Andric EmitOMPDeclareMapper(cast<OMPDeclareMapperDecl>(D));
71360b57cec5SDimitry Andric break;
71370b57cec5SDimitry Andric
71380b57cec5SDimitry Andric case Decl::OMPRequires:
71390b57cec5SDimitry Andric EmitOMPRequiresDecl(cast<OMPRequiresDecl>(D));
71400b57cec5SDimitry Andric break;
71410b57cec5SDimitry Andric
7142e8d8bef9SDimitry Andric case Decl::Typedef:
7143e8d8bef9SDimitry Andric case Decl::TypeAlias: // using foo = bar; [C++11]
7144e8d8bef9SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
7145e8d8bef9SDimitry Andric DI->EmitAndRetainType(
7146e8d8bef9SDimitry Andric getContext().getTypedefType(cast<TypedefNameDecl>(D)));
7147e8d8bef9SDimitry Andric break;
7148e8d8bef9SDimitry Andric
7149e8d8bef9SDimitry Andric case Decl::Record:
7150e8d8bef9SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
7151e8d8bef9SDimitry Andric if (cast<RecordDecl>(D)->getDefinition())
7152e8d8bef9SDimitry Andric DI->EmitAndRetainType(getContext().getRecordType(cast<RecordDecl>(D)));
7153e8d8bef9SDimitry Andric break;
7154e8d8bef9SDimitry Andric
7155e8d8bef9SDimitry Andric case Decl::Enum:
7156e8d8bef9SDimitry Andric if (CGDebugInfo *DI = getModuleDebugInfo())
7157e8d8bef9SDimitry Andric if (cast<EnumDecl>(D)->getDefinition())
7158e8d8bef9SDimitry Andric DI->EmitAndRetainType(getContext().getEnumType(cast<EnumDecl>(D)));
7159e8d8bef9SDimitry Andric break;
7160e8d8bef9SDimitry Andric
7161bdd1243dSDimitry Andric case Decl::HLSLBuffer:
7162bdd1243dSDimitry Andric getHLSLRuntime().addBuffer(cast<HLSLBufferDecl>(D));
7163bdd1243dSDimitry Andric break;
7164bdd1243dSDimitry Andric
71650b57cec5SDimitry Andric default:
71660b57cec5SDimitry Andric // Make sure we handled everything we should, every other kind is a
71670b57cec5SDimitry Andric // non-top-level decl. FIXME: Would be nice to have an isTopLevelDeclKind
71680b57cec5SDimitry Andric // function. Need to recode Decl::Kind to do that easily.
71690b57cec5SDimitry Andric assert(isa<TypeDecl>(D) && "Unsupported decl kind");
71700b57cec5SDimitry Andric break;
71710b57cec5SDimitry Andric }
71720b57cec5SDimitry Andric }
71730b57cec5SDimitry Andric
AddDeferredUnusedCoverageMapping(Decl * D)71740b57cec5SDimitry Andric void CodeGenModule::AddDeferredUnusedCoverageMapping(Decl *D) {
71750b57cec5SDimitry Andric // Do we need to generate coverage mapping?
71760b57cec5SDimitry Andric if (!CodeGenOpts.CoverageMapping)
71770b57cec5SDimitry Andric return;
71780b57cec5SDimitry Andric switch (D->getKind()) {
71790b57cec5SDimitry Andric case Decl::CXXConversion:
71800b57cec5SDimitry Andric case Decl::CXXMethod:
71810b57cec5SDimitry Andric case Decl::Function:
71820b57cec5SDimitry Andric case Decl::ObjCMethod:
71830b57cec5SDimitry Andric case Decl::CXXConstructor:
71840b57cec5SDimitry Andric case Decl::CXXDestructor: {
71850b57cec5SDimitry Andric if (!cast<FunctionDecl>(D)->doesThisDeclarationHaveABody())
71865ffd83dbSDimitry Andric break;
71870b57cec5SDimitry Andric SourceManager &SM = getContext().getSourceManager();
71880b57cec5SDimitry Andric if (LimitedCoverage && SM.getMainFileID() != SM.getFileID(D->getBeginLoc()))
71895ffd83dbSDimitry Andric break;
71900fca6ea1SDimitry Andric if (!llvm::coverage::SystemHeadersCoverage &&
71910fca6ea1SDimitry Andric SM.isInSystemHeader(D->getBeginLoc()))
71920fca6ea1SDimitry Andric break;
71935f757f3fSDimitry Andric DeferredEmptyCoverageMappingDecls.try_emplace(D, true);
71940b57cec5SDimitry Andric break;
71950b57cec5SDimitry Andric }
71960b57cec5SDimitry Andric default:
71970b57cec5SDimitry Andric break;
71980b57cec5SDimitry Andric };
71990b57cec5SDimitry Andric }
72000b57cec5SDimitry Andric
ClearUnusedCoverageMapping(const Decl * D)72010b57cec5SDimitry Andric void CodeGenModule::ClearUnusedCoverageMapping(const Decl *D) {
72020b57cec5SDimitry Andric // Do we need to generate coverage mapping?
72030b57cec5SDimitry Andric if (!CodeGenOpts.CoverageMapping)
72040b57cec5SDimitry Andric return;
72050b57cec5SDimitry Andric if (const auto *Fn = dyn_cast<FunctionDecl>(D)) {
72060b57cec5SDimitry Andric if (Fn->isTemplateInstantiation())
72070b57cec5SDimitry Andric ClearUnusedCoverageMapping(Fn->getTemplateInstantiationPattern());
72080b57cec5SDimitry Andric }
72095f757f3fSDimitry Andric DeferredEmptyCoverageMappingDecls.insert_or_assign(D, false);
72100b57cec5SDimitry Andric }
72110b57cec5SDimitry Andric
EmitDeferredUnusedCoverageMappings()72120b57cec5SDimitry Andric void CodeGenModule::EmitDeferredUnusedCoverageMappings() {
72130b57cec5SDimitry Andric // We call takeVector() here to avoid use-after-free.
72140b57cec5SDimitry Andric // FIXME: DeferredEmptyCoverageMappingDecls is getting mutated because
72150b57cec5SDimitry Andric // we deserialize function bodies to emit coverage info for them, and that
72160b57cec5SDimitry Andric // deserializes more declarations. How should we handle that case?
72170b57cec5SDimitry Andric for (const auto &Entry : DeferredEmptyCoverageMappingDecls.takeVector()) {
72180b57cec5SDimitry Andric if (!Entry.second)
72190b57cec5SDimitry Andric continue;
72200b57cec5SDimitry Andric const Decl *D = Entry.first;
72210b57cec5SDimitry Andric switch (D->getKind()) {
72220b57cec5SDimitry Andric case Decl::CXXConversion:
72230b57cec5SDimitry Andric case Decl::CXXMethod:
72240b57cec5SDimitry Andric case Decl::Function:
72250b57cec5SDimitry Andric case Decl::ObjCMethod: {
72260b57cec5SDimitry Andric CodeGenPGO PGO(*this);
72270b57cec5SDimitry Andric GlobalDecl GD(cast<FunctionDecl>(D));
72280b57cec5SDimitry Andric PGO.emitEmptyCounterMapping(D, getMangledName(GD),
72290b57cec5SDimitry Andric getFunctionLinkage(GD));
72300b57cec5SDimitry Andric break;
72310b57cec5SDimitry Andric }
72320b57cec5SDimitry Andric case Decl::CXXConstructor: {
72330b57cec5SDimitry Andric CodeGenPGO PGO(*this);
72340b57cec5SDimitry Andric GlobalDecl GD(cast<CXXConstructorDecl>(D), Ctor_Base);
72350b57cec5SDimitry Andric PGO.emitEmptyCounterMapping(D, getMangledName(GD),
72360b57cec5SDimitry Andric getFunctionLinkage(GD));
72370b57cec5SDimitry Andric break;
72380b57cec5SDimitry Andric }
72390b57cec5SDimitry Andric case Decl::CXXDestructor: {
72400b57cec5SDimitry Andric CodeGenPGO PGO(*this);
72410b57cec5SDimitry Andric GlobalDecl GD(cast<CXXDestructorDecl>(D), Dtor_Base);
72420b57cec5SDimitry Andric PGO.emitEmptyCounterMapping(D, getMangledName(GD),
72430b57cec5SDimitry Andric getFunctionLinkage(GD));
72440b57cec5SDimitry Andric break;
72450b57cec5SDimitry Andric }
72460b57cec5SDimitry Andric default:
72470b57cec5SDimitry Andric break;
72480b57cec5SDimitry Andric };
72490b57cec5SDimitry Andric }
72500b57cec5SDimitry Andric }
72510b57cec5SDimitry Andric
EmitMainVoidAlias()72525ffd83dbSDimitry Andric void CodeGenModule::EmitMainVoidAlias() {
72535ffd83dbSDimitry Andric // In order to transition away from "__original_main" gracefully, emit an
72545ffd83dbSDimitry Andric // alias for "main" in the no-argument case so that libc can detect when
72555ffd83dbSDimitry Andric // new-style no-argument main is in used.
72565ffd83dbSDimitry Andric if (llvm::Function *F = getModule().getFunction("main")) {
72575ffd83dbSDimitry Andric if (!F->isDeclaration() && F->arg_size() == 0 && !F->isVarArg() &&
725881ad6265SDimitry Andric F->getReturnType()->isIntegerTy(Context.getTargetInfo().getIntWidth())) {
725981ad6265SDimitry Andric auto *GA = llvm::GlobalAlias::create("__main_void", F);
726081ad6265SDimitry Andric GA->setVisibility(llvm::GlobalValue::HiddenVisibility);
726181ad6265SDimitry Andric }
72625ffd83dbSDimitry Andric }
72635ffd83dbSDimitry Andric }
72645ffd83dbSDimitry Andric
72650b57cec5SDimitry Andric /// Turns the given pointer into a constant.
GetPointerConstant(llvm::LLVMContext & Context,const void * Ptr)72660b57cec5SDimitry Andric static llvm::Constant *GetPointerConstant(llvm::LLVMContext &Context,
72670b57cec5SDimitry Andric const void *Ptr) {
72680b57cec5SDimitry Andric uintptr_t PtrInt = reinterpret_cast<uintptr_t>(Ptr);
72690b57cec5SDimitry Andric llvm::Type *i64 = llvm::Type::getInt64Ty(Context);
72700b57cec5SDimitry Andric return llvm::ConstantInt::get(i64, PtrInt);
72710b57cec5SDimitry Andric }
72720b57cec5SDimitry Andric
EmitGlobalDeclMetadata(CodeGenModule & CGM,llvm::NamedMDNode * & GlobalMetadata,GlobalDecl D,llvm::GlobalValue * Addr)72730b57cec5SDimitry Andric static void EmitGlobalDeclMetadata(CodeGenModule &CGM,
72740b57cec5SDimitry Andric llvm::NamedMDNode *&GlobalMetadata,
72750b57cec5SDimitry Andric GlobalDecl D,
72760b57cec5SDimitry Andric llvm::GlobalValue *Addr) {
72770b57cec5SDimitry Andric if (!GlobalMetadata)
72780b57cec5SDimitry Andric GlobalMetadata =
72790b57cec5SDimitry Andric CGM.getModule().getOrInsertNamedMetadata("clang.global.decl.ptrs");
72800b57cec5SDimitry Andric
72810b57cec5SDimitry Andric // TODO: should we report variant information for ctors/dtors?
72820b57cec5SDimitry Andric llvm::Metadata *Ops[] = {llvm::ConstantAsMetadata::get(Addr),
72830b57cec5SDimitry Andric llvm::ConstantAsMetadata::get(GetPointerConstant(
72840b57cec5SDimitry Andric CGM.getLLVMContext(), D.getDecl()))};
72850b57cec5SDimitry Andric GlobalMetadata->addOperand(llvm::MDNode::get(CGM.getLLVMContext(), Ops));
72860b57cec5SDimitry Andric }
72870b57cec5SDimitry Andric
CheckAndReplaceExternCIFuncs(llvm::GlobalValue * Elem,llvm::GlobalValue * CppFunc)728881ad6265SDimitry Andric bool CodeGenModule::CheckAndReplaceExternCIFuncs(llvm::GlobalValue *Elem,
728981ad6265SDimitry Andric llvm::GlobalValue *CppFunc) {
729081ad6265SDimitry Andric // Store the list of ifuncs we need to replace uses in.
729181ad6265SDimitry Andric llvm::SmallVector<llvm::GlobalIFunc *> IFuncs;
729281ad6265SDimitry Andric // List of ConstantExprs that we should be able to delete when we're done
729381ad6265SDimitry Andric // here.
729481ad6265SDimitry Andric llvm::SmallVector<llvm::ConstantExpr *> CEs;
729581ad6265SDimitry Andric
729681ad6265SDimitry Andric // It isn't valid to replace the extern-C ifuncs if all we find is itself!
729781ad6265SDimitry Andric if (Elem == CppFunc)
729881ad6265SDimitry Andric return false;
729981ad6265SDimitry Andric
730081ad6265SDimitry Andric // First make sure that all users of this are ifuncs (or ifuncs via a
730181ad6265SDimitry Andric // bitcast), and collect the list of ifuncs and CEs so we can work on them
730281ad6265SDimitry Andric // later.
730381ad6265SDimitry Andric for (llvm::User *User : Elem->users()) {
730481ad6265SDimitry Andric // Users can either be a bitcast ConstExpr that is used by the ifuncs, OR an
730581ad6265SDimitry Andric // ifunc directly. In any other case, just give up, as we don't know what we
730681ad6265SDimitry Andric // could break by changing those.
730781ad6265SDimitry Andric if (auto *ConstExpr = dyn_cast<llvm::ConstantExpr>(User)) {
730881ad6265SDimitry Andric if (ConstExpr->getOpcode() != llvm::Instruction::BitCast)
730981ad6265SDimitry Andric return false;
731081ad6265SDimitry Andric
731181ad6265SDimitry Andric for (llvm::User *CEUser : ConstExpr->users()) {
731281ad6265SDimitry Andric if (auto *IFunc = dyn_cast<llvm::GlobalIFunc>(CEUser)) {
731381ad6265SDimitry Andric IFuncs.push_back(IFunc);
731481ad6265SDimitry Andric } else {
731581ad6265SDimitry Andric return false;
731681ad6265SDimitry Andric }
731781ad6265SDimitry Andric }
731881ad6265SDimitry Andric CEs.push_back(ConstExpr);
731981ad6265SDimitry Andric } else if (auto *IFunc = dyn_cast<llvm::GlobalIFunc>(User)) {
732081ad6265SDimitry Andric IFuncs.push_back(IFunc);
732181ad6265SDimitry Andric } else {
732281ad6265SDimitry Andric // This user is one we don't know how to handle, so fail redirection. This
732381ad6265SDimitry Andric // will result in an ifunc retaining a resolver name that will ultimately
732481ad6265SDimitry Andric // fail to be resolved to a defined function.
732581ad6265SDimitry Andric return false;
732681ad6265SDimitry Andric }
732781ad6265SDimitry Andric }
732881ad6265SDimitry Andric
732981ad6265SDimitry Andric // Now we know this is a valid case where we can do this alias replacement, we
733081ad6265SDimitry Andric // need to remove all of the references to Elem (and the bitcasts!) so we can
733181ad6265SDimitry Andric // delete it.
733281ad6265SDimitry Andric for (llvm::GlobalIFunc *IFunc : IFuncs)
733381ad6265SDimitry Andric IFunc->setResolver(nullptr);
733481ad6265SDimitry Andric for (llvm::ConstantExpr *ConstExpr : CEs)
733581ad6265SDimitry Andric ConstExpr->destroyConstant();
733681ad6265SDimitry Andric
733781ad6265SDimitry Andric // We should now be out of uses for the 'old' version of this function, so we
733881ad6265SDimitry Andric // can erase it as well.
733981ad6265SDimitry Andric Elem->eraseFromParent();
734081ad6265SDimitry Andric
734181ad6265SDimitry Andric for (llvm::GlobalIFunc *IFunc : IFuncs) {
734281ad6265SDimitry Andric // The type of the resolver is always just a function-type that returns the
734381ad6265SDimitry Andric // type of the IFunc, so create that here. If the type of the actual
734481ad6265SDimitry Andric // resolver doesn't match, it just gets bitcast to the right thing.
734581ad6265SDimitry Andric auto *ResolverTy =
734681ad6265SDimitry Andric llvm::FunctionType::get(IFunc->getType(), /*isVarArg*/ false);
734781ad6265SDimitry Andric llvm::Constant *Resolver = GetOrCreateLLVMFunction(
734881ad6265SDimitry Andric CppFunc->getName(), ResolverTy, {}, /*ForVTable*/ false);
734981ad6265SDimitry Andric IFunc->setResolver(Resolver);
735081ad6265SDimitry Andric }
735181ad6265SDimitry Andric return true;
735281ad6265SDimitry Andric }
735381ad6265SDimitry Andric
73540b57cec5SDimitry Andric /// For each function which is declared within an extern "C" region and marked
73550b57cec5SDimitry Andric /// as 'used', but has internal linkage, create an alias from the unmangled
73560b57cec5SDimitry Andric /// name to the mangled name if possible. People expect to be able to refer
73570b57cec5SDimitry Andric /// to such functions with an unmangled name from inline assembly within the
73580b57cec5SDimitry Andric /// same translation unit.
EmitStaticExternCAliases()73590b57cec5SDimitry Andric void CodeGenModule::EmitStaticExternCAliases() {
73600b57cec5SDimitry Andric if (!getTargetCodeGenInfo().shouldEmitStaticExternCAliases())
73610b57cec5SDimitry Andric return;
73620b57cec5SDimitry Andric for (auto &I : StaticExternCValues) {
73630fca6ea1SDimitry Andric const IdentifierInfo *Name = I.first;
73640b57cec5SDimitry Andric llvm::GlobalValue *Val = I.second;
736581ad6265SDimitry Andric
736681ad6265SDimitry Andric // If Val is null, that implies there were multiple declarations that each
736781ad6265SDimitry Andric // had a claim to the unmangled name. In this case, generation of the alias
736881ad6265SDimitry Andric // is suppressed. See CodeGenModule::MaybeHandleStaticInExternC.
736981ad6265SDimitry Andric if (!Val)
737081ad6265SDimitry Andric break;
737181ad6265SDimitry Andric
737281ad6265SDimitry Andric llvm::GlobalValue *ExistingElem =
737381ad6265SDimitry Andric getModule().getNamedValue(Name->getName());
737481ad6265SDimitry Andric
737581ad6265SDimitry Andric // If there is either not something already by this name, or we were able to
737681ad6265SDimitry Andric // replace all uses from IFuncs, create the alias.
737781ad6265SDimitry Andric if (!ExistingElem || CheckAndReplaceExternCIFuncs(ExistingElem, Val))
7378fe6060f1SDimitry Andric addCompilerUsedGlobal(llvm::GlobalAlias::create(Name->getName(), Val));
73790b57cec5SDimitry Andric }
73800b57cec5SDimitry Andric }
73810b57cec5SDimitry Andric
lookupRepresentativeDecl(StringRef MangledName,GlobalDecl & Result) const73820b57cec5SDimitry Andric bool CodeGenModule::lookupRepresentativeDecl(StringRef MangledName,
73830b57cec5SDimitry Andric GlobalDecl &Result) const {
73840b57cec5SDimitry Andric auto Res = Manglings.find(MangledName);
73850b57cec5SDimitry Andric if (Res == Manglings.end())
73860b57cec5SDimitry Andric return false;
73870b57cec5SDimitry Andric Result = Res->getValue();
73880b57cec5SDimitry Andric return true;
73890b57cec5SDimitry Andric }
73900b57cec5SDimitry Andric
73910b57cec5SDimitry Andric /// Emits metadata nodes associating all the global values in the
73920b57cec5SDimitry Andric /// current module with the Decls they came from. This is useful for
73930b57cec5SDimitry Andric /// projects using IR gen as a subroutine.
73940b57cec5SDimitry Andric ///
73950b57cec5SDimitry Andric /// Since there's currently no way to associate an MDNode directly
73960b57cec5SDimitry Andric /// with an llvm::GlobalValue, we create a global named metadata
73970b57cec5SDimitry Andric /// with the name 'clang.global.decl.ptrs'.
EmitDeclMetadata()73980b57cec5SDimitry Andric void CodeGenModule::EmitDeclMetadata() {
73990b57cec5SDimitry Andric llvm::NamedMDNode *GlobalMetadata = nullptr;
74000b57cec5SDimitry Andric
74010b57cec5SDimitry Andric for (auto &I : MangledDeclNames) {
74020b57cec5SDimitry Andric llvm::GlobalValue *Addr = getModule().getNamedValue(I.second);
74030b57cec5SDimitry Andric // Some mangled names don't necessarily have an associated GlobalValue
74040b57cec5SDimitry Andric // in this module, e.g. if we mangled it for DebugInfo.
74050b57cec5SDimitry Andric if (Addr)
74060b57cec5SDimitry Andric EmitGlobalDeclMetadata(*this, GlobalMetadata, I.first, Addr);
74070b57cec5SDimitry Andric }
74080b57cec5SDimitry Andric }
74090b57cec5SDimitry Andric
74100b57cec5SDimitry Andric /// Emits metadata nodes for all the local variables in the current
74110b57cec5SDimitry Andric /// function.
EmitDeclMetadata()74120b57cec5SDimitry Andric void CodeGenFunction::EmitDeclMetadata() {
74130b57cec5SDimitry Andric if (LocalDeclMap.empty()) return;
74140b57cec5SDimitry Andric
74150b57cec5SDimitry Andric llvm::LLVMContext &Context = getLLVMContext();
74160b57cec5SDimitry Andric
74170b57cec5SDimitry Andric // Find the unique metadata ID for this name.
74180b57cec5SDimitry Andric unsigned DeclPtrKind = Context.getMDKindID("clang.decl.ptr");
74190b57cec5SDimitry Andric
74200b57cec5SDimitry Andric llvm::NamedMDNode *GlobalMetadata = nullptr;
74210b57cec5SDimitry Andric
74220b57cec5SDimitry Andric for (auto &I : LocalDeclMap) {
74230b57cec5SDimitry Andric const Decl *D = I.first;
74240fca6ea1SDimitry Andric llvm::Value *Addr = I.second.emitRawPointer(*this);
74250b57cec5SDimitry Andric if (auto *Alloca = dyn_cast<llvm::AllocaInst>(Addr)) {
74260b57cec5SDimitry Andric llvm::Value *DAddr = GetPointerConstant(getLLVMContext(), D);
74270b57cec5SDimitry Andric Alloca->setMetadata(
74280b57cec5SDimitry Andric DeclPtrKind, llvm::MDNode::get(
74290b57cec5SDimitry Andric Context, llvm::ValueAsMetadata::getConstant(DAddr)));
74300b57cec5SDimitry Andric } else if (auto *GV = dyn_cast<llvm::GlobalValue>(Addr)) {
74310b57cec5SDimitry Andric GlobalDecl GD = GlobalDecl(cast<VarDecl>(D));
74320b57cec5SDimitry Andric EmitGlobalDeclMetadata(CGM, GlobalMetadata, GD, GV);
74330b57cec5SDimitry Andric }
74340b57cec5SDimitry Andric }
74350b57cec5SDimitry Andric }
74360b57cec5SDimitry Andric
EmitVersionIdentMetadata()74370b57cec5SDimitry Andric void CodeGenModule::EmitVersionIdentMetadata() {
74380b57cec5SDimitry Andric llvm::NamedMDNode *IdentMetadata =
74390b57cec5SDimitry Andric TheModule.getOrInsertNamedMetadata("llvm.ident");
74400b57cec5SDimitry Andric std::string Version = getClangFullVersion();
74410b57cec5SDimitry Andric llvm::LLVMContext &Ctx = TheModule.getContext();
74420b57cec5SDimitry Andric
74430b57cec5SDimitry Andric llvm::Metadata *IdentNode[] = {llvm::MDString::get(Ctx, Version)};
74440b57cec5SDimitry Andric IdentMetadata->addOperand(llvm::MDNode::get(Ctx, IdentNode));
74450b57cec5SDimitry Andric }
74460b57cec5SDimitry Andric
EmitCommandLineMetadata()74470b57cec5SDimitry Andric void CodeGenModule::EmitCommandLineMetadata() {
74480b57cec5SDimitry Andric llvm::NamedMDNode *CommandLineMetadata =
74490b57cec5SDimitry Andric TheModule.getOrInsertNamedMetadata("llvm.commandline");
74500b57cec5SDimitry Andric std::string CommandLine = getCodeGenOpts().RecordCommandLine;
74510b57cec5SDimitry Andric llvm::LLVMContext &Ctx = TheModule.getContext();
74520b57cec5SDimitry Andric
74530b57cec5SDimitry Andric llvm::Metadata *CommandLineNode[] = {llvm::MDString::get(Ctx, CommandLine)};
74540b57cec5SDimitry Andric CommandLineMetadata->addOperand(llvm::MDNode::get(Ctx, CommandLineNode));
74550b57cec5SDimitry Andric }
74560b57cec5SDimitry Andric
EmitCoverageFile()74570b57cec5SDimitry Andric void CodeGenModule::EmitCoverageFile() {
74580b57cec5SDimitry Andric llvm::NamedMDNode *CUNode = TheModule.getNamedMetadata("llvm.dbg.cu");
74590b57cec5SDimitry Andric if (!CUNode)
74600b57cec5SDimitry Andric return;
74610b57cec5SDimitry Andric
74620b57cec5SDimitry Andric llvm::NamedMDNode *GCov = TheModule.getOrInsertNamedMetadata("llvm.gcov");
74630b57cec5SDimitry Andric llvm::LLVMContext &Ctx = TheModule.getContext();
74640b57cec5SDimitry Andric auto *CoverageDataFile =
74650b57cec5SDimitry Andric llvm::MDString::get(Ctx, getCodeGenOpts().CoverageDataFile);
74660b57cec5SDimitry Andric auto *CoverageNotesFile =
74670b57cec5SDimitry Andric llvm::MDString::get(Ctx, getCodeGenOpts().CoverageNotesFile);
74680b57cec5SDimitry Andric for (int i = 0, e = CUNode->getNumOperands(); i != e; ++i) {
74690b57cec5SDimitry Andric llvm::MDNode *CU = CUNode->getOperand(i);
74700b57cec5SDimitry Andric llvm::Metadata *Elts[] = {CoverageNotesFile, CoverageDataFile, CU};
74710b57cec5SDimitry Andric GCov->addOperand(llvm::MDNode::get(Ctx, Elts));
74720b57cec5SDimitry Andric }
74730b57cec5SDimitry Andric }
74740b57cec5SDimitry Andric
GetAddrOfRTTIDescriptor(QualType Ty,bool ForEH)74750b57cec5SDimitry Andric llvm::Constant *CodeGenModule::GetAddrOfRTTIDescriptor(QualType Ty,
74760b57cec5SDimitry Andric bool ForEH) {
74770b57cec5SDimitry Andric // Return a bogus pointer if RTTI is disabled, unless it's for EH.
74780b57cec5SDimitry Andric // FIXME: should we even be calling this method if RTTI is disabled
74790b57cec5SDimitry Andric // and it's not for EH?
748006c3fb27SDimitry Andric if (!shouldEmitRTTI(ForEH))
748106c3fb27SDimitry Andric return llvm::Constant::getNullValue(GlobalsInt8PtrTy);
74820b57cec5SDimitry Andric
74830b57cec5SDimitry Andric if (ForEH && Ty->isObjCObjectPointerType() &&
74840b57cec5SDimitry Andric LangOpts.ObjCRuntime.isGNUFamily())
74850b57cec5SDimitry Andric return ObjCRuntime->GetEHType(Ty);
74860b57cec5SDimitry Andric
74870b57cec5SDimitry Andric return getCXXABI().getAddrOfRTTIDescriptor(Ty);
74880b57cec5SDimitry Andric }
74890b57cec5SDimitry Andric
EmitOMPThreadPrivateDecl(const OMPThreadPrivateDecl * D)74900b57cec5SDimitry Andric void CodeGenModule::EmitOMPThreadPrivateDecl(const OMPThreadPrivateDecl *D) {
74910b57cec5SDimitry Andric // Do not emit threadprivates in simd-only mode.
74920b57cec5SDimitry Andric if (LangOpts.OpenMP && LangOpts.OpenMPSimd)
74930b57cec5SDimitry Andric return;
74940b57cec5SDimitry Andric for (auto RefExpr : D->varlists()) {
74950b57cec5SDimitry Andric auto *VD = cast<VarDecl>(cast<DeclRefExpr>(RefExpr)->getDecl());
74960b57cec5SDimitry Andric bool PerformInit =
74970b57cec5SDimitry Andric VD->getAnyInitializer() &&
74980b57cec5SDimitry Andric !VD->getAnyInitializer()->isConstantInitializer(getContext(),
74990b57cec5SDimitry Andric /*ForRef=*/false);
75000b57cec5SDimitry Andric
750181ad6265SDimitry Andric Address Addr(GetAddrOfGlobalVar(VD),
750281ad6265SDimitry Andric getTypes().ConvertTypeForMem(VD->getType()),
750381ad6265SDimitry Andric getContext().getDeclAlign(VD));
75040b57cec5SDimitry Andric if (auto InitFunction = getOpenMPRuntime().emitThreadPrivateVarDefinition(
75050b57cec5SDimitry Andric VD, Addr, RefExpr->getBeginLoc(), PerformInit))
75060b57cec5SDimitry Andric CXXGlobalInits.push_back(InitFunction);
75070b57cec5SDimitry Andric }
75080b57cec5SDimitry Andric }
75090b57cec5SDimitry Andric
75100b57cec5SDimitry Andric llvm::Metadata *
CreateMetadataIdentifierImpl(QualType T,MetadataTypeMap & Map,StringRef Suffix)75110b57cec5SDimitry Andric CodeGenModule::CreateMetadataIdentifierImpl(QualType T, MetadataTypeMap &Map,
75120b57cec5SDimitry Andric StringRef Suffix) {
75130eae32dcSDimitry Andric if (auto *FnType = T->getAs<FunctionProtoType>())
75140eae32dcSDimitry Andric T = getContext().getFunctionType(
75150eae32dcSDimitry Andric FnType->getReturnType(), FnType->getParamTypes(),
75160eae32dcSDimitry Andric FnType->getExtProtoInfo().withExceptionSpec(EST_None));
75170eae32dcSDimitry Andric
75180b57cec5SDimitry Andric llvm::Metadata *&InternalId = Map[T.getCanonicalType()];
75190b57cec5SDimitry Andric if (InternalId)
75200b57cec5SDimitry Andric return InternalId;
75210b57cec5SDimitry Andric
75220b57cec5SDimitry Andric if (isExternallyVisible(T->getLinkage())) {
75230b57cec5SDimitry Andric std::string OutName;
75240b57cec5SDimitry Andric llvm::raw_string_ostream Out(OutName);
75255f757f3fSDimitry Andric getCXXABI().getMangleContext().mangleCanonicalTypeName(
752606c3fb27SDimitry Andric T, Out, getCodeGenOpts().SanitizeCfiICallNormalizeIntegers);
752706c3fb27SDimitry Andric
752806c3fb27SDimitry Andric if (getCodeGenOpts().SanitizeCfiICallNormalizeIntegers)
752906c3fb27SDimitry Andric Out << ".normalized";
753006c3fb27SDimitry Andric
75310b57cec5SDimitry Andric Out << Suffix;
75320b57cec5SDimitry Andric
75330b57cec5SDimitry Andric InternalId = llvm::MDString::get(getLLVMContext(), Out.str());
75340b57cec5SDimitry Andric } else {
75350b57cec5SDimitry Andric InternalId = llvm::MDNode::getDistinct(getLLVMContext(),
75360b57cec5SDimitry Andric llvm::ArrayRef<llvm::Metadata *>());
75370b57cec5SDimitry Andric }
75380b57cec5SDimitry Andric
75390b57cec5SDimitry Andric return InternalId;
75400b57cec5SDimitry Andric }
75410b57cec5SDimitry Andric
CreateMetadataIdentifierForType(QualType T)75420b57cec5SDimitry Andric llvm::Metadata *CodeGenModule::CreateMetadataIdentifierForType(QualType T) {
75430b57cec5SDimitry Andric return CreateMetadataIdentifierImpl(T, MetadataIdMap, "");
75440b57cec5SDimitry Andric }
75450b57cec5SDimitry Andric
75460b57cec5SDimitry Andric llvm::Metadata *
CreateMetadataIdentifierForVirtualMemPtrType(QualType T)75470b57cec5SDimitry Andric CodeGenModule::CreateMetadataIdentifierForVirtualMemPtrType(QualType T) {
75480b57cec5SDimitry Andric return CreateMetadataIdentifierImpl(T, VirtualMetadataIdMap, ".virtual");
75490b57cec5SDimitry Andric }
75500b57cec5SDimitry Andric
75510b57cec5SDimitry Andric // Generalize pointer types to a void pointer with the qualifiers of the
75520b57cec5SDimitry Andric // originally pointed-to type, e.g. 'const char *' and 'char * const *'
75530b57cec5SDimitry Andric // generalize to 'const void *' while 'char *' and 'const char **' generalize to
75540b57cec5SDimitry Andric // 'void *'.
GeneralizeType(ASTContext & Ctx,QualType Ty)75550b57cec5SDimitry Andric static QualType GeneralizeType(ASTContext &Ctx, QualType Ty) {
75560b57cec5SDimitry Andric if (!Ty->isPointerType())
75570b57cec5SDimitry Andric return Ty;
75580b57cec5SDimitry Andric
75590b57cec5SDimitry Andric return Ctx.getPointerType(
75600b57cec5SDimitry Andric QualType(Ctx.VoidTy).withCVRQualifiers(
75610b57cec5SDimitry Andric Ty->getPointeeType().getCVRQualifiers()));
75620b57cec5SDimitry Andric }
75630b57cec5SDimitry Andric
75640b57cec5SDimitry Andric // Apply type generalization to a FunctionType's return and argument types
GeneralizeFunctionType(ASTContext & Ctx,QualType Ty)75650b57cec5SDimitry Andric static QualType GeneralizeFunctionType(ASTContext &Ctx, QualType Ty) {
75660b57cec5SDimitry Andric if (auto *FnType = Ty->getAs<FunctionProtoType>()) {
75670b57cec5SDimitry Andric SmallVector<QualType, 8> GeneralizedParams;
75680b57cec5SDimitry Andric for (auto &Param : FnType->param_types())
75690b57cec5SDimitry Andric GeneralizedParams.push_back(GeneralizeType(Ctx, Param));
75700b57cec5SDimitry Andric
75710b57cec5SDimitry Andric return Ctx.getFunctionType(
75720b57cec5SDimitry Andric GeneralizeType(Ctx, FnType->getReturnType()),
75730b57cec5SDimitry Andric GeneralizedParams, FnType->getExtProtoInfo());
75740b57cec5SDimitry Andric }
75750b57cec5SDimitry Andric
75760b57cec5SDimitry Andric if (auto *FnType = Ty->getAs<FunctionNoProtoType>())
75770b57cec5SDimitry Andric return Ctx.getFunctionNoProtoType(
75780b57cec5SDimitry Andric GeneralizeType(Ctx, FnType->getReturnType()));
75790b57cec5SDimitry Andric
75800b57cec5SDimitry Andric llvm_unreachable("Encountered unknown FunctionType");
75810b57cec5SDimitry Andric }
75820b57cec5SDimitry Andric
CreateMetadataIdentifierGeneralized(QualType T)75830b57cec5SDimitry Andric llvm::Metadata *CodeGenModule::CreateMetadataIdentifierGeneralized(QualType T) {
75840b57cec5SDimitry Andric return CreateMetadataIdentifierImpl(GeneralizeFunctionType(getContext(), T),
75850b57cec5SDimitry Andric GeneralizedMetadataIdMap, ".generalized");
75860b57cec5SDimitry Andric }
75870b57cec5SDimitry Andric
75880b57cec5SDimitry Andric /// Returns whether this module needs the "all-vtables" type identifier.
NeedAllVtablesTypeId() const75890b57cec5SDimitry Andric bool CodeGenModule::NeedAllVtablesTypeId() const {
75900b57cec5SDimitry Andric // Returns true if at least one of vtable-based CFI checkers is enabled and
75910b57cec5SDimitry Andric // is not in the trapping mode.
75920b57cec5SDimitry Andric return ((LangOpts.Sanitize.has(SanitizerKind::CFIVCall) &&
75930b57cec5SDimitry Andric !CodeGenOpts.SanitizeTrap.has(SanitizerKind::CFIVCall)) ||
75940b57cec5SDimitry Andric (LangOpts.Sanitize.has(SanitizerKind::CFINVCall) &&
75950b57cec5SDimitry Andric !CodeGenOpts.SanitizeTrap.has(SanitizerKind::CFINVCall)) ||
75960b57cec5SDimitry Andric (LangOpts.Sanitize.has(SanitizerKind::CFIDerivedCast) &&
75970b57cec5SDimitry Andric !CodeGenOpts.SanitizeTrap.has(SanitizerKind::CFIDerivedCast)) ||
75980b57cec5SDimitry Andric (LangOpts.Sanitize.has(SanitizerKind::CFIUnrelatedCast) &&
75990b57cec5SDimitry Andric !CodeGenOpts.SanitizeTrap.has(SanitizerKind::CFIUnrelatedCast)));
76000b57cec5SDimitry Andric }
76010b57cec5SDimitry Andric
AddVTableTypeMetadata(llvm::GlobalVariable * VTable,CharUnits Offset,const CXXRecordDecl * RD)76020b57cec5SDimitry Andric void CodeGenModule::AddVTableTypeMetadata(llvm::GlobalVariable *VTable,
76030b57cec5SDimitry Andric CharUnits Offset,
76040b57cec5SDimitry Andric const CXXRecordDecl *RD) {
76050b57cec5SDimitry Andric llvm::Metadata *MD =
76060b57cec5SDimitry Andric CreateMetadataIdentifierForType(QualType(RD->getTypeForDecl(), 0));
76070b57cec5SDimitry Andric VTable->addTypeMetadata(Offset.getQuantity(), MD);
76080b57cec5SDimitry Andric
76090b57cec5SDimitry Andric if (CodeGenOpts.SanitizeCfiCrossDso)
76100b57cec5SDimitry Andric if (auto CrossDsoTypeId = CreateCrossDsoCfiTypeId(MD))
76110b57cec5SDimitry Andric VTable->addTypeMetadata(Offset.getQuantity(),
76120b57cec5SDimitry Andric llvm::ConstantAsMetadata::get(CrossDsoTypeId));
76130b57cec5SDimitry Andric
76140b57cec5SDimitry Andric if (NeedAllVtablesTypeId()) {
76150b57cec5SDimitry Andric llvm::Metadata *MD = llvm::MDString::get(getLLVMContext(), "all-vtables");
76160b57cec5SDimitry Andric VTable->addTypeMetadata(Offset.getQuantity(), MD);
76170b57cec5SDimitry Andric }
76180b57cec5SDimitry Andric }
76190b57cec5SDimitry Andric
getSanStats()76200b57cec5SDimitry Andric llvm::SanitizerStatReport &CodeGenModule::getSanStats() {
76210b57cec5SDimitry Andric if (!SanStats)
7622a7dea167SDimitry Andric SanStats = std::make_unique<llvm::SanitizerStatReport>(&getModule());
76230b57cec5SDimitry Andric
76240b57cec5SDimitry Andric return *SanStats;
76250b57cec5SDimitry Andric }
762623408297SDimitry Andric
76270b57cec5SDimitry Andric llvm::Value *
createOpenCLIntToSamplerConversion(const Expr * E,CodeGenFunction & CGF)76280b57cec5SDimitry Andric CodeGenModule::createOpenCLIntToSamplerConversion(const Expr *E,
76290b57cec5SDimitry Andric CodeGenFunction &CGF) {
76300b57cec5SDimitry Andric llvm::Constant *C = ConstantEmitter(CGF).emitAbstract(E, E->getType());
763123408297SDimitry Andric auto *SamplerT = getOpenCLRuntime().getSamplerType(E->getType().getTypePtr());
763223408297SDimitry Andric auto *FTy = llvm::FunctionType::get(SamplerT, {C->getType()}, false);
7633fe6060f1SDimitry Andric auto *Call = CGF.EmitRuntimeCall(
763423408297SDimitry Andric CreateRuntimeFunction(FTy, "__translate_sampler_initializer"), {C});
763523408297SDimitry Andric return Call;
76360b57cec5SDimitry Andric }
76375ffd83dbSDimitry Andric
getNaturalPointeeTypeAlignment(QualType T,LValueBaseInfo * BaseInfo,TBAAAccessInfo * TBAAInfo)76385ffd83dbSDimitry Andric CharUnits CodeGenModule::getNaturalPointeeTypeAlignment(
76395ffd83dbSDimitry Andric QualType T, LValueBaseInfo *BaseInfo, TBAAAccessInfo *TBAAInfo) {
76405ffd83dbSDimitry Andric return getNaturalTypeAlignment(T->getPointeeType(), BaseInfo, TBAAInfo,
76415ffd83dbSDimitry Andric /* forPointeeType= */ true);
76425ffd83dbSDimitry Andric }
76435ffd83dbSDimitry Andric
getNaturalTypeAlignment(QualType T,LValueBaseInfo * BaseInfo,TBAAAccessInfo * TBAAInfo,bool forPointeeType)76445ffd83dbSDimitry Andric CharUnits CodeGenModule::getNaturalTypeAlignment(QualType T,
76455ffd83dbSDimitry Andric LValueBaseInfo *BaseInfo,
76465ffd83dbSDimitry Andric TBAAAccessInfo *TBAAInfo,
76475ffd83dbSDimitry Andric bool forPointeeType) {
76485ffd83dbSDimitry Andric if (TBAAInfo)
76495ffd83dbSDimitry Andric *TBAAInfo = getTBAAAccessInfo(T);
76505ffd83dbSDimitry Andric
76515ffd83dbSDimitry Andric // FIXME: This duplicates logic in ASTContext::getTypeAlignIfKnown. But
76525ffd83dbSDimitry Andric // that doesn't return the information we need to compute BaseInfo.
76535ffd83dbSDimitry Andric
76545ffd83dbSDimitry Andric // Honor alignment typedef attributes even on incomplete types.
76555ffd83dbSDimitry Andric // We also honor them straight for C++ class types, even as pointees;
76565ffd83dbSDimitry Andric // there's an expressivity gap here.
76575ffd83dbSDimitry Andric if (auto TT = T->getAs<TypedefType>()) {
76585ffd83dbSDimitry Andric if (auto Align = TT->getDecl()->getMaxAlignment()) {
76595ffd83dbSDimitry Andric if (BaseInfo)
76605ffd83dbSDimitry Andric *BaseInfo = LValueBaseInfo(AlignmentSource::AttributedType);
76615ffd83dbSDimitry Andric return getContext().toCharUnitsFromBits(Align);
76625ffd83dbSDimitry Andric }
76635ffd83dbSDimitry Andric }
76645ffd83dbSDimitry Andric
76655ffd83dbSDimitry Andric bool AlignForArray = T->isArrayType();
76665ffd83dbSDimitry Andric
76675ffd83dbSDimitry Andric // Analyze the base element type, so we don't get confused by incomplete
76685ffd83dbSDimitry Andric // array types.
76695ffd83dbSDimitry Andric T = getContext().getBaseElementType(T);
76705ffd83dbSDimitry Andric
76715ffd83dbSDimitry Andric if (T->isIncompleteType()) {
76725ffd83dbSDimitry Andric // We could try to replicate the logic from
76735ffd83dbSDimitry Andric // ASTContext::getTypeAlignIfKnown, but nothing uses the alignment if the
76745ffd83dbSDimitry Andric // type is incomplete, so it's impossible to test. We could try to reuse
76755ffd83dbSDimitry Andric // getTypeAlignIfKnown, but that doesn't return the information we need
76765ffd83dbSDimitry Andric // to set BaseInfo. So just ignore the possibility that the alignment is
76775ffd83dbSDimitry Andric // greater than one.
76785ffd83dbSDimitry Andric if (BaseInfo)
76795ffd83dbSDimitry Andric *BaseInfo = LValueBaseInfo(AlignmentSource::Type);
76805ffd83dbSDimitry Andric return CharUnits::One();
76815ffd83dbSDimitry Andric }
76825ffd83dbSDimitry Andric
76835ffd83dbSDimitry Andric if (BaseInfo)
76845ffd83dbSDimitry Andric *BaseInfo = LValueBaseInfo(AlignmentSource::Type);
76855ffd83dbSDimitry Andric
76865ffd83dbSDimitry Andric CharUnits Alignment;
7687e8d8bef9SDimitry Andric const CXXRecordDecl *RD;
7688e8d8bef9SDimitry Andric if (T.getQualifiers().hasUnaligned()) {
7689e8d8bef9SDimitry Andric Alignment = CharUnits::One();
7690e8d8bef9SDimitry Andric } else if (forPointeeType && !AlignForArray &&
7691e8d8bef9SDimitry Andric (RD = T->getAsCXXRecordDecl())) {
76925ffd83dbSDimitry Andric // For C++ class pointees, we don't know whether we're pointing at a
76935ffd83dbSDimitry Andric // base or a complete object, so we generally need to use the
76945ffd83dbSDimitry Andric // non-virtual alignment.
76955ffd83dbSDimitry Andric Alignment = getClassPointerAlignment(RD);
76965ffd83dbSDimitry Andric } else {
76975ffd83dbSDimitry Andric Alignment = getContext().getTypeAlignInChars(T);
76985ffd83dbSDimitry Andric }
76995ffd83dbSDimitry Andric
77005ffd83dbSDimitry Andric // Cap to the global maximum type alignment unless the alignment
77015ffd83dbSDimitry Andric // was somehow explicit on the type.
77025ffd83dbSDimitry Andric if (unsigned MaxAlign = getLangOpts().MaxTypeAlign) {
77035ffd83dbSDimitry Andric if (Alignment.getQuantity() > MaxAlign &&
77045ffd83dbSDimitry Andric !getContext().isAlignmentRequired(T))
77055ffd83dbSDimitry Andric Alignment = CharUnits::fromQuantity(MaxAlign);
77065ffd83dbSDimitry Andric }
77075ffd83dbSDimitry Andric return Alignment;
77085ffd83dbSDimitry Andric }
77095ffd83dbSDimitry Andric
stopAutoInit()77105ffd83dbSDimitry Andric bool CodeGenModule::stopAutoInit() {
77115ffd83dbSDimitry Andric unsigned StopAfter = getContext().getLangOpts().TrivialAutoVarInitStopAfter;
77125ffd83dbSDimitry Andric if (StopAfter) {
77135ffd83dbSDimitry Andric // This number is positive only when -ftrivial-auto-var-init-stop-after=* is
77145ffd83dbSDimitry Andric // used
77155ffd83dbSDimitry Andric if (NumAutoVarInit >= StopAfter) {
77165ffd83dbSDimitry Andric return true;
77175ffd83dbSDimitry Andric }
77185ffd83dbSDimitry Andric if (!NumAutoVarInit) {
77195ffd83dbSDimitry Andric unsigned DiagID = getDiags().getCustomDiagID(
77205ffd83dbSDimitry Andric DiagnosticsEngine::Warning,
77215ffd83dbSDimitry Andric "-ftrivial-auto-var-init-stop-after=%0 has been enabled to limit the "
77225ffd83dbSDimitry Andric "number of times ftrivial-auto-var-init=%1 gets applied.");
77235ffd83dbSDimitry Andric getDiags().Report(DiagID)
77245ffd83dbSDimitry Andric << StopAfter
77255ffd83dbSDimitry Andric << (getContext().getLangOpts().getTrivialAutoVarInit() ==
77265ffd83dbSDimitry Andric LangOptions::TrivialAutoVarInitKind::Zero
77275ffd83dbSDimitry Andric ? "zero"
77285ffd83dbSDimitry Andric : "pattern");
77295ffd83dbSDimitry Andric }
77305ffd83dbSDimitry Andric ++NumAutoVarInit;
77315ffd83dbSDimitry Andric }
77325ffd83dbSDimitry Andric return false;
77335ffd83dbSDimitry Andric }
7734fe6060f1SDimitry Andric
printPostfixForExternalizedDecl(llvm::raw_ostream & OS,const Decl * D) const77352a66634dSDimitry Andric void CodeGenModule::printPostfixForExternalizedDecl(llvm::raw_ostream &OS,
77362a66634dSDimitry Andric const Decl *D) const {
77372a66634dSDimitry Andric // ptxas does not allow '.' in symbol names. On the other hand, HIP prefers
77382a66634dSDimitry Andric // postfix beginning with '.' since the symbol name can be demangled.
77392a66634dSDimitry Andric if (LangOpts.HIP)
774081ad6265SDimitry Andric OS << (isa<VarDecl>(D) ? ".static." : ".intern.");
77412a66634dSDimitry Andric else
774281ad6265SDimitry Andric OS << (isa<VarDecl>(D) ? "__static__" : "__intern__");
774381ad6265SDimitry Andric
774481ad6265SDimitry Andric // If the CUID is not specified we try to generate a unique postfix.
774581ad6265SDimitry Andric if (getLangOpts().CUID.empty()) {
774681ad6265SDimitry Andric SourceManager &SM = getContext().getSourceManager();
774781ad6265SDimitry Andric PresumedLoc PLoc = SM.getPresumedLoc(D->getLocation());
774881ad6265SDimitry Andric assert(PLoc.isValid() && "Source location is expected to be valid.");
774981ad6265SDimitry Andric
775081ad6265SDimitry Andric // Get the hash of the user defined macros.
775181ad6265SDimitry Andric llvm::MD5 Hash;
775281ad6265SDimitry Andric llvm::MD5::MD5Result Result;
775381ad6265SDimitry Andric for (const auto &Arg : PreprocessorOpts.Macros)
775481ad6265SDimitry Andric Hash.update(Arg.first);
775581ad6265SDimitry Andric Hash.final(Result);
775681ad6265SDimitry Andric
775781ad6265SDimitry Andric // Get the UniqueID for the file containing the decl.
775881ad6265SDimitry Andric llvm::sys::fs::UniqueID ID;
77595f757f3fSDimitry Andric if (llvm::sys::fs::getUniqueID(PLoc.getFilename(), ID)) {
776081ad6265SDimitry Andric PLoc = SM.getPresumedLoc(D->getLocation(), /*UseLineDirectives=*/false);
776181ad6265SDimitry Andric assert(PLoc.isValid() && "Source location is expected to be valid.");
776281ad6265SDimitry Andric if (auto EC = llvm::sys::fs::getUniqueID(PLoc.getFilename(), ID))
776381ad6265SDimitry Andric SM.getDiagnostics().Report(diag::err_cannot_open_file)
776481ad6265SDimitry Andric << PLoc.getFilename() << EC.message();
776581ad6265SDimitry Andric }
776681ad6265SDimitry Andric OS << llvm::format("%x", ID.getFile()) << llvm::format("%x", ID.getDevice())
776781ad6265SDimitry Andric << "_" << llvm::utohexstr(Result.low(), /*LowerCase=*/true, /*Width=*/8);
776881ad6265SDimitry Andric } else {
776981ad6265SDimitry Andric OS << getContext().getCUIDHash();
777081ad6265SDimitry Andric }
7771fe6060f1SDimitry Andric }
7772fcaf7f86SDimitry Andric
moveLazyEmissionStates(CodeGenModule * NewBuilder)7773fcaf7f86SDimitry Andric void CodeGenModule::moveLazyEmissionStates(CodeGenModule *NewBuilder) {
7774fcaf7f86SDimitry Andric assert(DeferredDeclsToEmit.empty() &&
7775fcaf7f86SDimitry Andric "Should have emitted all decls deferred to emit.");
7776fcaf7f86SDimitry Andric assert(NewBuilder->DeferredDecls.empty() &&
7777fcaf7f86SDimitry Andric "Newly created module should not have deferred decls");
7778fcaf7f86SDimitry Andric NewBuilder->DeferredDecls = std::move(DeferredDecls);
77795f757f3fSDimitry Andric assert(EmittedDeferredDecls.empty() &&
77805f757f3fSDimitry Andric "Still have (unmerged) EmittedDeferredDecls deferred decls");
7781fcaf7f86SDimitry Andric
7782fcaf7f86SDimitry Andric assert(NewBuilder->DeferredVTables.empty() &&
7783fcaf7f86SDimitry Andric "Newly created module should not have deferred vtables");
7784fcaf7f86SDimitry Andric NewBuilder->DeferredVTables = std::move(DeferredVTables);
7785fcaf7f86SDimitry Andric
7786fcaf7f86SDimitry Andric assert(NewBuilder->MangledDeclNames.empty() &&
7787fcaf7f86SDimitry Andric "Newly created module should not have mangled decl names");
7788fcaf7f86SDimitry Andric assert(NewBuilder->Manglings.empty() &&
7789fcaf7f86SDimitry Andric "Newly created module should not have manglings");
7790fcaf7f86SDimitry Andric NewBuilder->Manglings = std::move(Manglings);
7791fcaf7f86SDimitry Andric
7792fcaf7f86SDimitry Andric NewBuilder->WeakRefReferences = std::move(WeakRefReferences);
7793fcaf7f86SDimitry Andric
7794972a253aSDimitry Andric NewBuilder->ABI->MangleCtx = std::move(ABI->MangleCtx);
7795fcaf7f86SDimitry Andric }
7796