altos: Add ao_delay_until to ao_notask

[fw/altos] / src / kernel / ao_notask.c

diff --git a/src/kernel/ao_notask.c b/src/kernel/ao_notask.c
index e973c7ae1723e8fa8811543e092590694622bf3f..8c5743fdecd0a0a1f7ae62c528e32d27c98a138b 100644 (file)
--- a/src/kernel/ao_notask.c
+++ b/src/kernel/ao_notask.c
@@ -40,18 +40,21 @@ ao_sleep(void *wchan)
 }
 
 #if HAS_AO_DELAY
+void
+ao_delay_until(AO_TICK_TYPE target)
+{
+       ao_arch_block_interrupts();
+       while ((int16_t) (target - ao_tick_count) > 0)
+               ao_sleep((void *) &ao_tick_count);
+       ao_arch_release_interrupts();
+}
+
 void
 ao_delay(AO_TICK_TYPE ticks)
 {
-       AO_TICK_TYPE    target;
-
-       if (!ticks)
-               ticks = 1;
-       target = ao_tick_count + ticks;
-       do {
-               ao_sleep(&ao_time);
-       } while ((int16_t) (target - ao_tick_count) > 0);
+       ao_delay_until(ao_time() + ticks);
 }
+
 #endif
 
 void