add UP scheduler critical switch flag

Author: PillarShi

src/cpu_up.c

@@ -40,6 +40,7 @@ void rt_spin_lock(struct rt_spinlock *lock)
 
 /**
  * @brief   This function will unlock the spinlock, will unlock the thread scheduler.
+ *          If the scheduling function is called before unlocking, it will be scheduled in this function.
  *
  * @param   lock is a pointer to the spinlock.
  */
@@ -72,6 +73,7 @@ rt_base_t rt_spin_lock_irqsave(struct rt_spinlock *lock)
 
 /**
  * @brief   This function will unlock the spinlock and then restore current cpu interrupt status, will unlock the thread scheduler.
+ *          If the scheduling function is called before unlocking, it will be scheduled in this function.
  *
  * @param   lock is a pointer to the spinlock.
  *

src/scheduler_up.c

@@ -30,6 +30,7 @@
  * 2022-01-07     Gabriel      Moving __on_rt_xxxxx_hook to scheduler.c
  * 2023-03-27     rose_man     Split into scheduler upc and scheduler_mp.c
  * 2023-10-17     ChuShicheng  Modify the timing of clearing RT_THREAD_STAT_YIELD flag bits
+ * 2025-08-04     Pillar       Add rt_scheduler_critical_switch_flag
  */
 
 #define __RT_IPC_SOURCE__
@@ -49,6 +50,7 @@ rt_uint8_t rt_thread_ready_table[32];
 
 extern volatile rt_atomic_t rt_interrupt_nest;
 static rt_int16_t rt_scheduler_lock_nest;
+static rt_int8_t rt_scheduler_critical_switch_flag = -1;
 rt_uint8_t rt_current_priority;
 
 #if defined(RT_USING_HOOK) && defined(RT_HOOK_USING_FUNC_PTR)
@@ -235,6 +237,7 @@ void rt_system_scheduler_start(void)
     to_thread = _scheduler_get_highest_priority_thread(&highest_ready_priority);
 
     rt_cpu_self()->current_thread = to_thread;
+    rt_scheduler_critical_switch_flag = 0;
 
     rt_sched_remove_thread(to_thread);
     RT_SCHED_CTX(to_thread).stat = RT_THREAD_RUNNING;
@@ -387,6 +390,10 @@ void rt_schedule(void)
             }
         }
     }
+    else
+    {
+        rt_scheduler_critical_switch_flag = 1;
+    }
 
     /* enable interrupt */
     rt_hw_interrupt_enable(level);
@@ -604,6 +611,7 @@ void rt_exit_critical_safe(rt_base_t critical_level)
 
 /**
  * @brief Safely exit critical section (non-debug version)
+ *        If the scheduling function is called before exiting, it will be scheduled in this function.
  *
  * @param critical_level The expected critical level (unused in non-debug build)
  *
@@ -657,6 +665,7 @@ RTM_EXPORT(rt_enter_critical);
 
 /**
  * @brief Exit critical section and unlock scheduler
+ *        If the scheduling function is called before exiting, it will be scheduled in this function.
  *
  * @details This function:
  *   - Decrements the scheduler lock nesting count
@@ -685,9 +694,10 @@ void rt_exit_critical(void)
         /* enable interrupt */
         rt_hw_interrupt_enable(level);
 
-        if (rt_current_thread)
+        if (rt_scheduler_critical_switch_flag == 1)
         {
-            /* if scheduler is started, do a schedule */
+            rt_scheduler_critical_switch_flag = 0;
+            /* if scheduler is started and needs to be scheduled, do a schedule */
             rt_schedule();
         }
     }