Support coro_result in Task C impl

Author: itamaro

Include/internal/pycore_global_objects_fini_generated.h

@@ -337,6 +337,7 @@ struct _Py_global_strings {
         STRUCT_FOR_ID(copy)
         STRUCT_FOR_ID(copyreg)
         STRUCT_FOR_ID(coro)
+        STRUCT_FOR_ID(coro_result)
         STRUCT_FOR_ID(count)
         STRUCT_FOR_ID(cwd)
         STRUCT_FOR_ID(d)

Include/internal/pycore_global_strings.h

@@ -156,6 +156,9 @@ class _asyncio.Future "FutureObj *" "&Future_Type"
 /* Get FutureIter from Future */
 static PyObject * future_new_iter(PyObject *);
 
+static PyObject *
+task_step2_impl(asyncio_state *state, TaskObj *task, PyObject *result);
+
 
 static int
 _is_coroutine(asyncio_state *state, PyObject *coro)
@@ -2033,15 +2036,16 @@ _asyncio.Task.__init__
     loop: object = None
     name: object = None
     context: object = None
+    coro_result: object = NULL
 
 A coroutine wrapped in a Future.
 [clinic start generated code]*/
 
 static int
 _asyncio_Task___init___impl(TaskObj *self, PyObject *coro, PyObject *loop,
-                            PyObject *name, PyObject *context)
-/*[clinic end generated code: output=49ac96fe33d0e5c7 input=924522490c8ce825]*/
-
+                            PyObject *name, PyObject *context,
+                            PyObject *coro_result)
+/*[clinic end generated code: output=e241855787412a77 input=3fcd7fb1c00d3f87]*/
 {
     if (future_init((FutureObj*)self, loop)) {
         return -1;
@@ -2089,8 +2093,14 @@ _asyncio_Task___init___impl(TaskObj *self, PyObject *coro, PyObject *loop,
         return -1;
     }
 
-    if (task_call_step_soon(state, self, NULL)) {
-        return -1;
+    if (coro_result == NULL) {
+        if (task_call_step_soon(state, self, NULL)) {
+            return -1;
+        }
+    }
+    else {
+        Py_INCREF(coro_result);
+        task_step2_impl(state, self, coro_result);
     }
     return register_task(state, (PyObject*)self);
 }
@@ -2844,6 +2854,20 @@ task_step_impl(asyncio_state *state, TaskObj *task, PyObject *exc)
         Py_RETURN_NONE;
     }
 
+    return task_step2_impl(state, task, result);
+
+fail:
+    Py_XDECREF(result);
+    return NULL;
+}
+
+
+static PyObject *
+task_step2_impl(asyncio_state *state, TaskObj *task, PyObject *result)
+{
+    int res;
+    PyObject *o;
+
     if (result == (PyObject*)task) {
         /* We have a task that wants to await on itself */
         goto self_await;

Include/internal/pycore_runtime_init_generated.h

@@ -151,7 +151,7 @@ def counting_task_constructor(coro, *, loop=None, name=None, context=None, coro_
                 # only count calls that will actually result a task scheduled to the event loop
                 # (if coro_result is non-None, it will return synchronously)
                 self.task_count += 1
-            return asyncio.tasks._PyTask(coro, loop=loop, name=name, context=context, coro_result=coro_result)
+            return asyncio.Task(coro, loop=loop, name=name, context=context, coro_result=coro_result)
 
         def counting_task_factory(loop, coro, *, name=None, context=None, coro_result=None):
             return counting_task_constructor(coro, loop=loop, name=name, context=context, coro_result=coro_result)