Merge pull request #80511 from slavapestov/fix-issue-79367

Sema: Fix regression with 'Failure' type witness inference hack

Author: slavapestov

lib/Sema/AssociatedTypeInference.cpp

@@ -2496,7 +2496,11 @@ AssociatedTypeInference::computeFailureTypeWitness(
     if (!isAsyncIteratorProtocolNext(witness.first))
       continue;
 
-    if (!witness.second || witness.second->getDeclContext() != dc)
+    // Different extensions of the same nominal are OK, but if the witness is in
+    // a protocol extension or a superclass or something, give up.
+    if (!witness.second ||
+        witness.second->getDeclContext()->getSelfNominalTypeDecl()
+            != dc->getSelfNominalTypeDecl())
       continue;
 
     if (auto witnessFunc = dyn_cast<AbstractFunctionDecl>(witness.second)) {

test/AssociatedTypeInference/issue-79367.swift

@@ -0,0 +1,15 @@
+// RUN: %target-typecheck-verify-swift
+
+// https://github.com/swiftlang/swift/issues/79367
+
+// 'Failure' type witness inference should still take place when
+// the 'next()' witness is in a different extension than the
+// conformance.
+
+struct AsyncIteratorImpl<Element>: AsyncIteratorProtocol {}
+
+extension AsyncIteratorImpl {
+  func next() async throws -> Element? {
+    fatalError()
+  }
+}