X-Git-Url: https://git.gag.com/?p=fw%2Faltos;a=blobdiff_plain;f=ao_task.c;h=24bd319ac563b69d33dd77999a0a532763eaa592;hp=0e976d7c4247e5e341c40684c2d62a8e2c59018c;hb=fb63262699d3a6fbf347d24efda8b01f75b7d0d3;hpb=823f4f92de0c1f8dd7a644a8e56ffe9822bee6e2

diff --git a/ao_task.c b/ao_task.c
index 0e976d7c..24bd319a 100644
--- a/ao_task.c
+++ b/ao_task.c
@@ -29,7 +29,7 @@ ao_add_task(__xdata struct ao_task * task, void (*start)(void))
 {
 	uint8_t	__xdata *stack;
 	if (ao_num_tasks == AO_NUM_TASKS)
-		ao_panic(AO_ERROR_NO_TASK);
+		ao_panic(AO_PANIC_NO_TASK);
 	ao_tasks[ao_num_tasks++] = task;
 	/*
 	 * Construct a stack frame so that it will 'return'
@@ -162,7 +162,9 @@ ao_yield(void) _naked
 int
 ao_sleep(__xdata void *wchan)
 {
+	ao_interrupt_disable();
 	ao_cur_task->wchan = wchan;
+	ao_interrupt_enable();
 	ao_yield();
 }
 
@@ -179,8 +181,13 @@ ao_wakeup(__xdata void *wchan)
 void
 ao_start_scheduler(void)
 {
+	ao_timer_init();
+	ao_adc_init();
+	ao_beep_init();
+	ao_led_init();
+	ao_usb_init();
+
 	ao_cur_task_id = AO_NO_TASK;
 	ao_cur_task = NULL;
-	ao_timer_init();
 	ao_yield();
 }