[IRGen] Sign TaskResumeFunction.

Author: nate-chandler

lib/IRGen/IRGenFunction.cpp

@@ -608,9 +608,15 @@ void IRGenFunction::emitGetAsyncContinuation(SILType resumeTy,
   assert(AsyncCoroutineCurrentResume == nullptr &&
          "Don't support nested get_async_continuation");
   AsyncCoroutineCurrentResume = coroResume;
-  Builder.CreateStore(
-      Builder.CreateBitOrPointerCast(coroResume, IGM.FunctionPtrTy),
-      Address(currTaskResumeTaskAddr, pointerAlignment));
+  llvm::Value *coroResumeValue =
+      Builder.CreateBitOrPointerCast(coroResume, IGM.FunctionPtrTy);
+  if (auto schema = IGM.getOptions().PointerAuth.TaskResumeFunction) {
+    auto authInfo = PointerAuthInfo::emit(*this, schema, currTaskResumeTaskAddr,
+                                          PointerAuthEntity());
+    coroResumeValue = emitPointerAuthSign(*this, coroResumeValue, authInfo);
+  }
+  Builder.CreateStore(coroResumeValue,
+                      Address(currTaskResumeTaskAddr, pointerAlignment));
   // currTask->ResumeContext = &continuation_context;
   auto currTaskResumeCtxtAddr = Builder.CreateStructGEP(currTask, 5);
   llvm::Value *continuationContextValue = Builder.CreateBitOrPointerCast(

test/IRGen/async/get_async_continuation.sil

@@ -37,9 +37,17 @@ bb0:
 // CHECK:  [[exe:%.*]] = load %swift.executor*, %swift.executor** [[exe_addr]]
 // CHECK:  store %swift.executor* [[exe]], %swift.executor** [[exectuor_addr]]
 //   Initialize the async task with the continuation function and async continuation context.
-// CHECK: [[task_continuation_fn_addr:%.*]] = getelementptr inbounds %swift.task, %swift.task* [[tsk]], i32 0, i32 4
+// CHECK: [[continuation_fn_addr:%.*]] = getelementptr inbounds %swift.task, %swift.task* [[tsk]], i32 0, i32 4
 // CHECK: [[continuation_fn:%.*]] = call i8* @llvm.coro.async.resume()
-// CHECK: store i8* [[continuation_fn]], i8** [[task_continuation_fn_addr]]
+
+// CHECK-arm64e: [[continuation_fn_addr_int:%[0-9]+]] = ptrtoint i8** [[continuation_fn_addr]] to i64
+// CHECK-arm64e: [[ptrauth_blend:%[0-9]+]]   = call i64 @llvm.ptrauth.blend.i64(i64 [[continuation_fn_addr_int]], i64 11330)
+// CHECK-arm64e: [[continuation_fn_int:%[0-9]+]]      = ptrtoint i8* [[continuation_fn]] to i64
+// CHECK-arm64e: [[signed_int:%[0-9]+]]      = call i64 @llvm.ptrauth.sign.i64(i64 [[continuation_fn_int]], i32 0, i64 [[ptrauth_blend]])
+// CHECK-arm64e: [[signed_continuation_fn:%[0-9]+]]   = inttoptr i64 [[signed_int]] to i8*
+// CHECK-arm64e: store i8* [[signed_continuation_fn]], i8** [[continuation_fn_addr]]
+// CHECK-x86_64: store i8* [[continuation_fn]], i8** [[continuation_fn_addr]]
+
 // CHECK: [[task_resume_context_addr:%.*]] = getelementptr inbounds %swift.task, %swift.task* [[tsk]], i32 0, i32 5
 // CHECK: [[task_resume_context:%.*]] = bitcast %swift.async_continuation_context* [[cont_context]] to %swift.context*