X-Git-Url: https://codewiz.org/gitweb?a=blobdiff_plain;f=bertos%2Fkern%2Fpreempt.c;h=4b1ad9ee2d5dea8e213570eeb1c59fc400290fa6;hb=e62ca0b357f09804d7d894949df44224c9d74bb7;hp=c3d212d446b30c9816d86cbc858515e0c304e517;hpb=d3860ea45a63b0443e4e75a16b402ada83078d12;p=bertos.git

diff --git a/bertos/kern/preempt.c b/bertos/kern/preempt.c
index c3d212d4..4b1ad9ee 100644
--- a/bertos/kern/preempt.c
+++ b/bertos/kern/preempt.c
@@ -65,7 +65,8 @@ CONFIG_DEPEND(CONFIG_KERN_PREEMPT,    CONFIG_KERN_SCHED && CONFIG_TIMER_EVENTS &
 
 MOD_DEFINE(preempt)
 
-int preempt_forbid_cnt;
+/// Global preemption disabling nesting counter
+cpu_atomic_t _preempt_forbid_cnt;
 
 static Timer preempt_timer;
 
@@ -74,7 +75,7 @@ void proc_schedule(void)
 {
 	IRQ_DISABLE;
 
-	ASSERT(preempt_forbid_cnt == 0);
+	ASSERT(proc_allowed());
 	LIST_ASSERT_VALID(&ProcReadyList);
 	CurrentProcess = (struct Process *)list_remHead(&ProcReadyList);
 	ASSERT2(CurrentProcess, "no idle proc?");
@@ -86,7 +87,7 @@ void proc_schedule(void)
 
 void proc_preempt(UNUSED_ARG(void *, param))
 {
-	if (!preempt_forbid_cnt)
+	if (proc_allowed())
 	{
 		IRQ_DISABLE;
 
@@ -122,7 +123,7 @@ void proc_switch(void)
 
 	/* Sleeping with IRQs disabled or preemption forbidden is illegal */
 	IRQ_ASSERT_ENABLED();
-	ASSERT(preempt_forbid_cnt == 0);
+	ASSERT(proc_allowed());
 
 	// Will invoke proc_switch() in interrupt context
 	kill(0, SIGUSR1);