Merge pull request #37579 from aschwaighofer/async_default_witness

[IRGen] Use a async function pointer for default async protocol witness implemenations

Author: aschwaighofer

lib/IRGen/GenMeta.cpp

@@ -916,6 +916,10 @@ namespace {
         if (!entry.isValid() || entry.getKind() != SILWitnessTable::Method ||
             entry.getMethodWitness().Requirement != func)
           continue;
+        auto silFunc = entry.getMethodWitness().Witness;
+        if (silFunc->isAsync()) {
+          return IGM.getAddrOfAsyncFunctionPointer(silFunc);
+        }
         return IGM.getAddrOfSILFunction(entry.getMethodWitness().Witness,
                                         NotForDefinition);
       }

test/Interpreter/Inputs/resilient_async.swift

@@ -0,0 +1,9 @@
+public protocol Problem : class {
+}
+
+public extension Problem {
+}
+
+public func callGenericWitness<T: Problem> (_ t: T) async -> Int {
+  return 0
+}

test/Interpreter/Inputs/resilient_async2.swift

@@ -0,0 +1,13 @@
+public protocol Problem : class {
+  func theProblem() async -> Int
+}
+
+public extension Problem {
+  func theProblem() async -> Int {
+			return 1
+  }
+}
+
+public func callGenericWitness<T: Problem> (_ t: T) async -> Int {
+  return await t.theProblem()
+}

test/Interpreter/async_resilience.swift

@@ -0,0 +1,29 @@
+// RUN: %empty-directory(%t)
+
+// RUN: %target-build-swift-dylib(%t/%target-library-name(resilient_async)) -enable-library-evolution %S/Inputs/resilient_async.swift -emit-module -emit-module-path %t/resilient_async.swiftmodule -module-name resilient_async
+// RUN: %target-codesign %t/%target-library-name(resilient_async)
+
+// RUN: %target-build-swift -parse-as-library %s -lresilient_async -I %t -L %t -o %t/main %target-rpath(%t)
+// RUN: %target-codesign %t/main
+
+// Introduce a defaulted protocol method.
+// RUN: %target-build-swift-dylib(%t/%target-library-name(resilient_async)) -enable-library-evolution %S/Inputs/resilient_async2.swift -emit-module -emit-module-path %t/resilient_async.swiftmodule -module-name resilient_async
+// RUN: %target-codesign %t/%target-library-name(resilient_async)
+
+// RUN: %target-run %t/main %t/%target-library-name(resilient_async)
+
+// REQUIRES: executable_test
+
+
+import resilient_async
+
+class Impl : Problem {}
+
+@main struct Main {
+  static func main() async {
+      let i = Impl()
+      // This used to crash.
+      let r = await callGenericWitness(i)
+      assert(r == 1)
+  }
+}