Merge pull request #36842 from aschwaighofer/fix_withCancellationHandler_arm64e_windows

Fix with cancellation handler arm64e windows

Author: aschwaighofer

stdlib/public/Concurrency/Task.cpp

@@ -865,9 +865,9 @@ swift_task_addCancellationHandlerImpl(
     void *context) {
   void *allocation =
       swift_task_alloc(sizeof(CancellationNotificationStatusRecord));
-  auto *record =
-      new (allocation) CancellationNotificationStatusRecord(
-          handler, context);
+  auto unsigned_handler = swift_auth_code(handler, 3848);
+  auto *record = new (allocation)
+      CancellationNotificationStatusRecord(unsigned_handler, context);
 
   swift_task_addStatusRecord(record);
   return record;

test/Concurrency/Runtime/cancellation_handler.swift

@@ -4,6 +4,14 @@
 
 // rdar://76038845
 // UNSUPPORTED: use_os_stdlib
+// UNSUPPORTED: OS=windows-msvc
+
+// for sleep
+#if canImport(Darwin)
+    import Darwin
+#elseif canImport(Glibc)
+    import Glibc
+#endif
 
 class Canary {
   deinit {
@@ -21,10 +29,22 @@ if #available(macOS 9999, iOS 9999, watchOS 9999, tvOS 9999, *) {
     }
   }
   task.cancel()
+  sleep(1)
+  detach {
+    await Task.withCancellationHandler {
+        print("Task was cancelled!")
+    }
+    operation: {
+        print("Running the operation...")
+    }
+  }
+  sleep(10)
 } else {
   // Fake prints to satisfy FileCheck.
   print("Canary")
   print("canary died")
+  print("Running the operation...")
 }
 // CHECK: Canary
 // CHECK-NEXT: canary died
+// CHECK-NEXT: Running the operation...