runtime: fix leak in UnownedTask

Author: Darksonn

tokio/src/runtime/task/mod.rs

@@ -369,10 +369,16 @@ impl<S: Schedule> UnownedTask<S> {
         let raw = self.raw;
         mem::forget(self);
 
-        // Poll the task
+        // Transfer one ref-count to a Task object.
+        let task = Task::<S> {
+            raw,
+            _p: PhantomData,
+        };
+
+        // Use the other ref-count to poll the task.
         raw.poll();
         // Decrement our extra ref-count
-        raw.header().state.ref_dec();
+        drop(task);
     }
 
     pub(crate) fn shutdown(self) {

tokio/src/runtime/tests/task.rs

@@ -111,6 +111,12 @@ fn create_shutdown2() {
     drop(join);
 }
 
+#[test]
+fn unowned_poll() {
+    let (task, _) = unowned(async {}, NoopSchedule);
+    task.run();
+}
+
 #[test]
 fn schedule() {
     with(|rt| {