Index: clang/lib/CodeGen/CGExpr.cpp =================================================================== --- clang/lib/CodeGen/CGExpr.cpp +++ clang/lib/CodeGen/CGExpr.cpp @@ -2821,6 +2821,7 @@ CGM.getModule(), Descriptor->getType(), /*isConstant=*/true, llvm::GlobalVariable::PrivateLinkage, Descriptor); GV->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global); + CGM.setPragmaSectionAttributes(CurFuncDecl, GV); CGM.getSanitizerMetadata()->disableSanitizerForGlobal(GV); // Remember the descriptor for this type. @@ -2900,6 +2901,7 @@ } auto FilenameGV = CGM.GetAddrOfConstantCString(FilenameString, ".src"); + CGM.setPragmaSectionAttributes(CurFuncDecl, cast(FilenameGV.getPointer())); CGM.getSanitizerMetadata()->disableSanitizerForGlobal( cast(FilenameGV.getPointer())); Filename = FilenameGV.getPointer(); @@ -3073,6 +3075,7 @@ new llvm::GlobalVariable(CGM.getModule(), Info->getType(), false, llvm::GlobalVariable::PrivateLinkage, Info); InfoPtr->setUnnamedAddr(llvm::GlobalValue::UnnamedAddr::Global); + CGM.setPragmaSectionAttributes(CurFuncDecl, InfoPtr); CGM.getSanitizerMetadata()->disableSanitizerForGlobal(InfoPtr); Args.push_back(Builder.CreateBitCast(InfoPtr, Int8PtrTy)); ArgTypes.push_back(Int8PtrTy); Index: clang/lib/CodeGen/CodeGenModule.h =================================================================== --- clang/lib/CodeGen/CodeGenModule.h +++ clang/lib/CodeGen/CodeGenModule.h @@ -1342,6 +1342,11 @@ /// \param QT is the clang QualType of the null pointer. llvm::Constant *getNullPointer(llvm::PointerType *T, QualType QT); + /// Set section attributes requested by "#pragma clang section" + /// \param D is the declaration to read semantic attributes from. + /// \param GO is the global object to set section attributes. + void setPragmaSectionAttributes(const Decl *D, llvm::GlobalObject *GO); + private: llvm::Constant *GetOrCreateLLVMFunction( StringRef MangledName, llvm::Type *Ty, GlobalDecl D, bool ForVTable, Index: clang/lib/CodeGen/CodeGenModule.cpp =================================================================== --- clang/lib/CodeGen/CodeGenModule.cpp +++ clang/lib/CodeGen/CodeGenModule.cpp @@ -1704,11 +1704,8 @@ return AddedAttr; } -void CodeGenModule::setNonAliasAttributes(GlobalDecl GD, - llvm::GlobalObject *GO) { - const Decl *D = GD.getDecl(); - SetCommonAttributes(GD, GO); - +void CodeGenModule::setPragmaSectionAttributes(const Decl *D, + llvm::GlobalObject *GO) { if (D) { if (auto *GV = dyn_cast(GO)) { if (auto *SA = D->getAttr()) @@ -1724,6 +1721,24 @@ if (!D->getAttr()) F->addFnAttr("implicit-section-name", SA->getName()); + if (auto *SA = D->getAttr()) + F->addFnAttr("bss-section", SA->getName()); + if (auto *SA = D->getAttr()) + F->addFnAttr("data-section", SA->getName()); + if (auto *SA = D->getAttr()) + F->addFnAttr("rodata-section", SA->getName()); + } + } +} + +void CodeGenModule::setNonAliasAttributes(GlobalDecl GD, + llvm::GlobalObject *GO) { + const Decl *D = GD.getDecl(); + SetCommonAttributes(GD, GO); + setPragmaSectionAttributes(D, GO); + + if (D) { + if (auto *F = dyn_cast(GO)) { llvm::AttrBuilder Attrs; if (GetCPUAndFeaturesAttributes(GD, Attrs)) { // We know that GetCPUAndFeaturesAttributes will always have the Index: clang/lib/Sema/SemaDecl.cpp =================================================================== --- clang/lib/Sema/SemaDecl.cpp +++ clang/lib/Sema/SemaDecl.cpp @@ -9004,6 +9004,21 @@ Context, PragmaClangTextSection.SectionName, PragmaClangTextSection.PragmaLocation, AttributeCommonInfo::AS_Pragma)); + if (D.isFunctionDefinition()) { + if (PragmaClangBSSSection.Valid) + NewFD->addAttr(PragmaClangBSSSectionAttr::CreateImplicit( + Context, PragmaClangBSSSection.SectionName, + PragmaClangBSSSection.PragmaLocation)); + if (PragmaClangDataSection.Valid) + NewFD->addAttr(PragmaClangDataSectionAttr::CreateImplicit( + Context, PragmaClangDataSection.SectionName, + PragmaClangDataSection.PragmaLocation)); + if (PragmaClangRodataSection.Valid) + NewFD->addAttr(PragmaClangRodataSectionAttr::CreateImplicit( + Context, PragmaClangRodataSection.SectionName, + PragmaClangRodataSection.PragmaLocation)); + } + // Apply an implicit SectionAttr if #pragma code_seg is active. if (CodeSegStack.CurrentValue && D.isFunctionDefinition() && !NewFD->hasAttr()) { Index: clang/test/CodeGen/cfi-pragma-section.c =================================================================== --- /dev/null +++ clang/test/CodeGen/cfi-pragma-section.c @@ -0,0 +1,32 @@ +// Check that CFI-generated data structures are tagged with +// "#pragma clang section" attributes + +// RUN: %clang_cc1 -triple x86_64-unknown-linux -fsanitize=cfi-icall \ +// RUN: -fno-sanitize-trap=cfi-icall -emit-llvm -o - %s | FileCheck %s + +// CHECK-DAG: attributes [[ATTR:#[0-9]+]]{{.*}}bss-section{{.*}}data-section{{.*}}rodata-section +// CHECK-DAG: @.src = private unnamed_addr constant{{.*}}cfi-pragma-section.c{{.*}}[[ATTR]] +// CHECK-DAG: @{{[0-9]+}} = private unnamed_addr constant{{.*}}int (int){{.*}}[[ATTR]] +// CHECK-DAG: @{{[0-9]+}} = private unnamed_addr global{{.*}}@.src{{.*}}[[ATTR]] + +typedef int (*int_arg_fn)(int); + +static int int_arg1(int arg) { + return 0; +} + +static int int_arg2(int arg) { + return 1; +} + +int_arg_fn int_funcs[2] = {int_arg1, int_arg2}; + +#pragma clang section bss = ".bss.mycfi" +#pragma clang section data = ".data.mycfi" +#pragma clang section rodata = ".rodata.mycfi" + +int main(int argc, const char *argv[]) { + + int idx = argv[1][0] - '0'; + return int_funcs[argc](idx); +}