Merge pull request #40046 from etcwilde/ewilde/5.5/handle-missing-getMainExecutor-decl

5.5: Workaround to cope with older SDKs

Author: DougGregor

lib/SILGen/SILGenFunction.cpp

@@ -928,7 +928,6 @@ void SILGenFunction::emitAsyncMainThreadStart(SILDeclRef entryPoint) {
   FuncDecl *builtinDecl = cast<FuncDecl>(getBuiltinValueDecl(
       getASTContext(),
       ctx.getIdentifier(getBuiltinName(BuiltinValueKind::CreateAsyncTask))));
-
   auto subs = SubstitutionMap::get(builtinDecl->getGenericSignature(),
                                    {TupleType::getEmpty(ctx)},
                                    ArrayRef<ProtocolConformanceRef>{});
@@ -953,6 +952,7 @@ void SILGenFunction::emitAsyncMainThreadStart(SILDeclRef entryPoint) {
 
   // Get swiftJobRun
   FuncDecl *swiftJobRunFuncDecl = SGM.getSwiftJobRun();
+  assert(swiftJobRunFuncDecl && "Failed to find swift_job_run function decl");
   SILFunction *swiftJobRunSILFunc =
       SGM.getFunction(SILDeclRef(swiftJobRunFuncDecl, SILDeclRef::Kind::Func),
                       NotForDefinition);
@@ -970,6 +970,28 @@ void SILGenFunction::emitAsyncMainThreadStart(SILDeclRef entryPoint) {
 
   // Get main executor
   FuncDecl *getMainExecutorFuncDecl = SGM.getGetMainExecutor();
+  if (!getMainExecutorFuncDecl) {
+    // If it doesn't exist due to an SDK-compiler mismatch, we can conjure one
+    // up instead of crashing:
+    // @available(SwiftStdlib 5.5, *)
+    // @_silgen_name("swift_task_getMainExecutor")
+    // internal func _getMainExecutor() -> Builtin.Executor
+
+    ParameterList *emptyParams = ParameterList::createEmpty(getASTContext());
+    getMainExecutorFuncDecl = FuncDecl::createImplicit(
+        getASTContext(), StaticSpellingKind::None,
+        DeclName(
+            getASTContext(),
+            DeclBaseName(getASTContext().getIdentifier("_getMainExecutor")),
+            /*Arguments*/ emptyParams),
+        {}, /*async*/ false, /*throws*/ false, {}, emptyParams,
+        getASTContext().TheExecutorType,
+        entryPoint.getDecl()->getModuleContext());
+    getMainExecutorFuncDecl->getAttrs().add(
+        new (getASTContext())
+            SILGenNameAttr("swift_task_getMainExecutor", /*implicit*/ true));
+  }
+
   SILFunction *getMainExeutorSILFunc = SGM.getFunction(
       SILDeclRef(getMainExecutorFuncDecl, SILDeclRef::Kind::Func),
       NotForDefinition);
@@ -983,6 +1005,25 @@ void SILGenFunction::emitAsyncMainThreadStart(SILDeclRef entryPoint) {
 
   // Start Main loop!
   FuncDecl *drainQueueFuncDecl = SGM.getAsyncMainDrainQueue();
+  if (!drainQueueFuncDecl) {
+    // If it doesn't exist, we can conjure one up instead of crashing
+    // @available(SwiftStdlib 5.5, *)
+    // @_silgen_name("swift_task_asyncMainDrainQueue")
+    // internal func _asyncMainDrainQueue() -> Never
+    ParameterList *emptyParams = ParameterList::createEmpty(getASTContext());
+    drainQueueFuncDecl = FuncDecl::createImplicit(
+        getASTContext(), StaticSpellingKind::None,
+        DeclName(
+          getASTContext(),
+          DeclBaseName(getASTContext().getIdentifier("_asyncMainDrainQueue")),
+          /*Arguments*/emptyParams),
+          {}, /*async*/ false, /*throws*/ false, {}, emptyParams,
+          getASTContext().getNeverType(),
+          entryPoint.getDecl()->getModuleContext());
+    drainQueueFuncDecl->getAttrs().add(
+        new (getASTContext()) SILGenNameAttr("swift_task_asyncMainDrainQueue", /*implicit*/ true));
+  }
+
   SILFunction *drainQueueSILFunc = SGM.getFunction(
       SILDeclRef(drainQueueFuncDecl, SILDeclRef::Kind::Func), NotForDefinition);
   SILValue drainQueueFunc =