Make linked task failure work again

1 files changed, 12 insertions, 3 deletions

diff --git a/src/rt/rust_task.cpp b/src/rt/rust_task.cpp
index b861c60b521..e99ccc1d04b 100644
--- a/src/rt/rust_task.cpp
+++ b/src/rt/rust_task.cpp
@@ -74,6 +74,7 @@ rust_task::rust_task(rust_scheduler *sched, rust_task_list *state,
     local_region(&sched->srv->local_region),
     _on_wakeup(NULL),
     failed(false),
+    killed(false),
     propagate_failure(true),
     dynastack(this)
 {
@@ -253,10 +254,19 @@ rust_task::yield(size_t time_in_us) {
     LOG(this, task, "task %s @0x%" PRIxPTR " yielding for %d us",
         name, this, time_in_us);
 
+    if (killed) {
+        killed = false;
+        fail();
+    }
     yield_timer.reset_us(time_in_us);
 
     // Return to the scheduler.
     ctx.next->swap(ctx);
+
+    if (killed) {
+        killed = false;
+        fail();
+    }
 }
 
 void
@@ -270,12 +280,11 @@ rust_task::kill() {
     // from task A and want to force-fail task B, you do B->kill().
     // If you want to fail yourself you do self->fail().
     LOG(this, task, "killing task %s @0x%" PRIxPTR, name, this);
+    // When the task next goes to yield or resume it will fail
+    killed = true;
     // Unblock the task so it can unwind.
     unblock();
 
-    if (NULL == supervisor && propagate_failure)
-        sched->fail();
-
     sched->lock.signal();
 
     LOG(this, task, "preparing to unwind task: 0x%" PRIxPTR, this);