[kernel] add UP scheduler critical switch flag.

This commit is contained in:
Pillar
2025-08-06 15:08:35 +08:00
committed by R b b666
parent 95ea2fc726
commit b084503b6d
2 changed files with 20 additions and 2 deletions

View File

@@ -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.
*

View File

@@ -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__
@@ -51,6 +52,11 @@ extern volatile rt_atomic_t rt_interrupt_nest;
static rt_int16_t rt_scheduler_lock_nest;
rt_uint8_t rt_current_priority;
static rt_int8_t rt_scheduler_critical_switch_flag;
#define IS_CRITICAL_SWITCH_PEND() (rt_scheduler_critical_switch_flag == 1)
#define SET_CRITICAL_SWITCH_FLAG() (rt_scheduler_critical_switch_flag = 1)
#define CLR_CRITICAL_SWITCH_FLAG() (rt_scheduler_critical_switch_flag = 0)
#if defined(RT_USING_HOOK) && defined(RT_HOOK_USING_FUNC_PTR)
static void (*rt_scheduler_hook)(struct rt_thread *from, struct rt_thread *to);
static void (*rt_scheduler_switch_hook)(struct rt_thread *tid);
@@ -236,6 +242,9 @@ void rt_system_scheduler_start(void)
rt_cpu_self()->current_thread = to_thread;
/* flush critical switch flag */
CLR_CRITICAL_SWITCH_FLAG();
rt_sched_remove_thread(to_thread);
RT_SCHED_CTX(to_thread).stat = RT_THREAD_RUNNING;
@@ -387,6 +396,10 @@ void rt_schedule(void)
}
}
}
else
{
SET_CRITICAL_SWITCH_FLAG();
}
/* enable interrupt */
rt_hw_interrupt_enable(level);
@@ -604,6 +617,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 +671,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 +700,10 @@ void rt_exit_critical(void)
/* enable interrupt */
rt_hw_interrupt_enable(level);
if (rt_current_thread)
if (IS_CRITICAL_SWITCH_PEND())
{
/* if scheduler is started, do a schedule */
CLR_CRITICAL_SWITCH_FLAG();
/* if scheduler is started and needs to be scheduled, do a schedule */
rt_schedule();
}
}