[cfi] Support explicit sections for functions in cfi-icall.

Allow explicit section for indirectly called functions in cfi-icall.
Jumptables for functions in the same type class must be contiguous, so they
always go to the default text section.

Fixes PR25079.

llvm-svn: 266486

2 files changed, 29 insertions, 2 deletions

diff --git a/llvm/lib/Transforms/IPO/LowerBitSets.cpp b/llvm/lib/Transforms/IPO/LowerBitSets.cpp
index 1798d83..db5bab0 100644
--- a/llvm/lib/Transforms/IPO/LowerBitSets.cpp
+++ b/llvm/lib/Transforms/IPO/LowerBitSets.cpp
@@ -639,8 +639,9 @@ void LowerBitSets::verifyBitSetMDNode(MDNode *Op) {
 
   if (OpGlobal->isThreadLocal())
     report_fatal_error("Bit set element may not be thread-local");
-  if (OpGlobal->hasSection())
-    report_fatal_error("Bit set element may not have an explicit section");
+  if (isa<GlobalVariable>(OpGlobal) && OpGlobal->hasSection())
+    report_fatal_error(
+        "Bit set global var element may not have an explicit section");
 
   if (isa<GlobalVariable>(OpGlobal) && OpGlobal->isDeclarationForLinker())
     report_fatal_error("Bit set global var element must be a definition");
diff --git a/llvm/test/Transforms/LowerBitSets/section.ll b/llvm/test/Transforms/LowerBitSets/section.ll
new file mode 100644
index 0000000..d5f9414
--- /dev/null
+++ b/llvm/test/Transforms/LowerBitSets/section.ll
@@ -0,0 +1,26 @@
+; Test that functions with "section" attribute are accepted, and jumptables are
+; emitted in ".text".
+
+; RUN: opt -S -lowerbitsets < %s | FileCheck %s
+
+target triple = "x86_64-unknown-linux-gnu"
+
+; CHECK: @[[A:.*]] = private constant {{.*}} section ".text"
+; CHECK: @f = alias void (), bitcast ({{.*}}* @[[A]] to void ()*)
+; CHECK: define private void {{.*}} section "xxx"
+
+define void @f() section "xxx" {
+entry:
+  ret void
+}
+
+define i1 @g() {
+entry:
+  %0 = call i1 @llvm.bitset.test(i8* bitcast (void ()* @f to i8*), metadata !"_ZTSFvE")
+  ret i1 %0
+}
+
+declare i1 @llvm.bitset.test(i8*, metadata) nounwind readnone
+
+!llvm.bitsets = !{!0}
+!0 = !{!"_ZTSFvE", void ()* @f, i64 0}