Ensure that ptrauth subtarget feature is checked for ptrauth-returns

Author: kovdan01

llvm/lib/Target/AArch64/AArch64FrameLowering.cpp

@@ -1511,7 +1511,9 @@ void AArch64FrameLowering::emitPrologue(MachineFunction &MF,
   // If we're saving LR, sign it first.
   if (shouldAuthenticateLR(MF)) {
     if (LLVM_UNLIKELY(!Subtarget.hasPAuth()))
-      report_fatal_error("arm64e LR authentication requires ptrauth");
+      report_fatal_error(
+          StringRef(Subtarget.isTargetMachO() ? "arm64e" : "aarch64") +
+          " LR authentication requires ptrauth");
     for (const CalleeSavedInfo &Info : MFI.getCalleeSavedInfo()) {
       if (Info.getReg() != AArch64::LR)
         continue;
@@ -2051,7 +2053,9 @@ void AArch64FrameLowering::emitEpilogue(MachineFunction &MF,
   auto InsertAuthLROnExit = make_scope_exit([&]() {
     if (shouldAuthenticateLR(MF)) {
       if (LLVM_UNLIKELY(!Subtarget.hasPAuth()))
-        report_fatal_error("arm64e LR authentication requires ptrauth");
+        report_fatal_error(
+            StringRef(Subtarget.isTargetMachO() ? "arm64e" : "aarch64") +
+            " LR authentication requires ptrauth");
       for (const CalleeSavedInfo &Info : MFI.getCalleeSavedInfo()) {
         if (Info.getReg() != AArch64::LR)
           continue;

llvm/test/CodeGen/AArch64/GlobalISel/ptrauth-lowering-err.ll

@@ -52,3 +52,21 @@ define i8* @foo() #0 {
 @g_weak.ptrauth = private constant { i8*, i32, i64, i64 } { i8* bitcast (i32* @g_weak to i8*), i32 0, i64 0, i64 0 }, section "llvm.ptrauth"
 
 attributes #0 = { "ptrauth-calls" "target-cpu"="generic" }
+
+;--- ptrauth-returns.ll
+
+; RUN: not --crash llc -mtriple aarch64-elf ptrauth-returns.ll 2>&1 | \
+; RUN:   FileCheck ptrauth-returns.ll
+
+; CHECK: LLVM ERROR: aarch64 LR authentication requires ptrauth
+
+define i32 @bar() #0 {
+  ret i32 42
+}
+
+define i32 @foo() {
+  %tmp = call i32 @bar()
+  ret i32 %tmp
+}
+
+attributes #0 = { "ptrauth-returns" "target-cpu"="generic" }