Add task_yield_maybe

3 files changed, 21 insertions, 1 deletions

diff --git a/projects/hsm/hsm.c b/projects/hsm/hsm.c
index 800edcc..b6b8820 100644
--- a/projects/hsm/hsm.c
+++ b/projects/hsm/hsm.c
@@ -393,6 +393,11 @@ void hal_task_yield(void)
     task_yield();
 }
 
+void hal_task_yield_maybe(void)
+{
+    task_yield_maybe();
+}
+
 /* A mutex to arbitrate concurrent access to the keystore.
  */
 task_mutex_t ks_mutex = { 0 };
diff --git a/task.c b/task.c
index e156940..d8af217 100644
--- a/task.c
+++ b/task.c
@@ -83,12 +83,16 @@ static tcb_t *cur_task = NULL;
 
 #ifdef TASK_METRICS
 static uint32_t tick_start = 0;
-static uint32_t tick_prev  = 0;
 static uint32_t tick_idle  = 0;
 static uint32_t tick_max   = 0;
 static uint32_t nyield     = 0;
 #endif
 
+static uint32_t tick_prev  = 0;
+#ifndef TASK_YIELD_THRESHOLD
+#define TASK_YIELD_THRESHOLD 100
+#endif
+
 /* Add a task.
  */
 tcb_t *task_add(char *name, funcp_t func, void *cookie, void *stack, size_t stack_len)
@@ -221,6 +225,8 @@ void task_yield(void)
     }
     tick_prev = tick;
     ++nyield;
+#else
+    tick_prev = HAL_GetTick();
 #endif
 
     /* If there are no other runnable tasks (and cur_task is runnable),
@@ -256,6 +262,14 @@ void task_yield(void)
     }
 }
 
+/* Yield if it's been "too long" since the last yield.
+ */
+void task_yield_maybe(void)
+{
+    if (HAL_GetTick() - tick_prev >= TASK_YIELD_THRESHOLD)
+        task_yield();
+}
+
 /* Put the current task to sleep (make it non-runnable).
  */
 void task_sleep(void)
diff --git a/task.h b/task.h
index 24f87ce..73ff33f 100644
--- a/task.h
+++ b/task.h
@@ -55,6 +55,7 @@ extern tcb_t *task_add(char *name, funcp_t func, void *cookie, void *stack, size
 extern void task_set_idle_hook(funcp_t func);
 
 extern void task_yield(void);
+extern void task_yield_maybe(void);
 extern void task_sleep(void);
 extern void task_wake(tcb_t *t);