X-Git-Url: https://git.gag.com/?a=blobdiff_plain;f=src%2Fcore%2Fao_task.c;h=df70b906ccfa6bec6c793c417a25bc4d4e652ff6;hb=4b13d3c659240e5a8347b1ba7ab0bf1d8355eba3;hp=65654731aeaeaccc9f45477e156ffa1419d13c0b;hpb=583458772746317b98fced907ec780edff465888;p=fw%2Faltos

diff --git a/src/core/ao_task.c b/src/core/ao_task.c
index 65654731..df70b906 100644
--- a/src/core/ao_task.c
+++ b/src/core/ao_task.c
@@ -16,6 +16,13 @@
  */
 
 #include <ao.h>
+#include <ao_task.h>
+#if HAS_SAMPLE_PROFILE
+#include <ao_sample_profile.h>
+#endif
+#if HAS_STACK_GUARD
+#include <ao_mpu.h>
+#endif
 
 #define AO_NO_TASK_INDEX	0xff
 
@@ -67,6 +74,8 @@ ao_add_task(__xdata struct ao_task * task, void (*start)(void), __code char *nam
 	ao_arch_init_stack(task, start);
 }
 
+__xdata uint8_t	ao_idle;
+
 /* Task switching function. This must not use any stack variables */
 void
 ao_yield(void) ao_arch_naked_define
@@ -77,6 +86,13 @@ ao_yield(void) ao_arch_naked_define
 		ao_cur_task_index = ao_num_tasks-1;
 	else
 	{
+#if HAS_SAMPLE_PROFILE
+		uint16_t	tick = ao_sample_profile_timer_value();
+		uint16_t	run = tick - ao_cur_task->start;
+		if (run > ao_cur_task->max_run)
+			ao_cur_task->max_run = run;
+		++ao_cur_task->yields;
+#endif
 		ao_arch_save_stack();
 	}
 
@@ -110,7 +126,13 @@ ao_yield(void) ao_arch_naked_define
 			if (ao_cur_task_index == ao_last_task_index)
 				ao_arch_cpu_idle();
 		}
+#if HAS_SAMPLE_PROFILE
+	ao_cur_task->start = ao_sample_profile_timer_value();
+#endif
 	}
+#if HAS_STACK_GUARD
+	ao_mpu_stack_guard(ao_cur_task->stack);
+#endif
 #if AO_CHECK_STACK
 	cli();
 	in_yield = 0;