【TencentOS tiny】深度源碼分析(2)——調度器

舒適提示:本文不描述與浮點相關的寄存器的內容,如需瞭解自行查閱(畢竟我本身也不懂)編程

調度器的基本概念

TencentOS tiny中提供的任務調度器是基於優先級的全搶佔式調度,在系統運行過程當中,當有比當前任務優先級更高的任務就緒時,當前任務將馬上被切出,高優先級任務搶佔處理器運行。函數

TencentOS tiny內核中也容許建立相同優先級的任務。相同優先級的任務採用時間片輪轉方式進行調度(也就是一般說的分時調度器),時間片輪轉調度僅在當前系統中無更高優先級就緒任務的狀況下才有效。ui

爲了保證系統的實時性,系統盡最大可能地保證高優先級的任務得以運行。任務調度的原則是一旦任務狀態發生了改變,而且當前運行的任務優先級小於優先級隊列中任務最高優先級時,馬上進行任務切換(除非當前系統處於中斷處理程序中或禁止任務切換的狀態)。spa

調度器是操做系統的核心,其主要功能就是實現任務的切換,即從就緒列表裏面找到優先級最高的任務,而後去執行該任務。操作系統

啓動調度器

調度器的啓動由cpu_sched_start函數來完成,它會被tos_knl_start函數調用,這個函數中主要作兩件事,首先經過readyqueue_highest_ready_task_get函數獲取當前系統中處於最高優先級的就緒任務,而且將它賦值給指向當前任務控制塊的指針k_curr_task ,而後設置一下系統的狀態爲運行態KNL_STATE_RUNNING.net

固然最重要的是調用匯編代碼寫的函數cpu_sched_start啓動調度器,該函數在源碼的archarmarm-v7m目錄下的port_s.S彙編文件下,TencentOS tiny支持多種內核的芯片,如M3/M4/M7等,不一樣的芯片該函數的實現方式不一樣,port_s.S也是TencentOS tiny做爲軟件與CPU硬件鏈接的橋樑。以M4的cpu_sched_start舉個例子:線程

__API__ k_err_t tos_knl_start(void)
{
    if (tos_knl_is_running()) {
        return K_ERR_KNL_RUNNING;
    }

    k_next_task = readyqueue_highest_ready_task_get();
    k_curr_task = k_next_task;
    k_knl_state = KNL_STATE_RUNNING;
    cpu_sched_start();

    return K_ERR_NONE;
}複製代碼

port_sched_start
    CPSID   I    

    ; set pendsv priority lowest
    ; otherwise trigger pendsv in port_irq_context_switch will cause a context swich in irq
    ; that would be a disaster
    MOV32   R0, NVIC_SYSPRI14
    MOV32   R1, NVIC_PENDSV_PRI
    STRB    R1, [R0]

    LDR     R0, =SCB_VTOR
    LDR     R0, [R0]
    LDR     R0, [R0]
    MSR     MSP, R0

    ; k_curr_task = k_next_task
    MOV32   R0, k_curr_task
    MOV32   R1, k_next_task
    LDR     R2, [R1]
    STR     R2, [R0]

    ; sp = k_next_task->sp
    LDR     R0, [R2]
    ; PSP = sp
    MSR     PSP, R0

    ; using PSP
    MRS     R0, CONTROL
    ORR     R0, R0, #2
    MSR     CONTROL, R0

    ISB

    ; restore r4-11 from new process stack
    LDMFD   SP!, {R4 - R11}

    IF {FPU} != "SoftVFP"
    ; ignore EXC_RETURN the first switch
    LDMFD   SP!, {R0}
    ENDIF

    ; restore r0, r3
    LDMFD    SP!, {R0 - R3}
    ; load R12 and LR
    LDMFD    SP!, {R12, LR}
    ; load PC and discard xPSR
    LDMFD    SP!, {R1, R2}

    CPSIE    I
    BX       R1複製代碼

Cortex-M內核關中斷指令

從上面的彙編代碼,我又想介紹一下Cortex-M內核關中斷指令,唉~感受仍是有點麻煩!爲了快速地開關中斷, Cortex-M內核專門設置了一條 CPS 指令,用於操做PRIMASK寄存器跟FAULTMASK寄存器的,這兩個寄存器是與屏蔽中斷有關的,除此以外Cortex-M內核還存在BASEPRI寄存器也是與中斷有關的,也順帶介紹一下吧。指針

CPSID I     ;PRIMASK=1     ;關中斷
CPSIE I     ;PRIMASK=0     ;開中斷
CPSID F     ;FAULTMASK=1   ;關異常
CPSIE F     ;FAULTMASK=0   ;開異常複製代碼
寄存器 功能
PRIMASK
它被置 1 後,就關掉全部可屏蔽的異常,只剩下 NMI 和HardFault FAULT能夠響應
FAULTMASK
當它置 1 時,只有 NMI 才能響應,全部其它的異常都沒法響應(包括HardFault FAULT)
BASEPRI
這個寄存器最多有 9 位(由表達優先級的位數決定)。它定義了被屏蔽優先級的閾值。當它被設成某個值後,全部優先級號大於等於此值的中斷都被關(優先級號越大,優先級越低)。但若被設成 0,則不關閉任何中斷

更多具體的描述看我之前的文章:RTOS臨界段知識:https://blog.csdn.net/jiejiemcu/article/details/82534974rest

迴歸正題

在啓動內核調度器過程當中須要配置PendSV 的中斷優先級爲最低,就是往NVIC_SYSPRI14(0xE000ED22)地址寫入NVIC_PENDSV_PRI(0xFF)。由於PendSV都會涉及到系統調度,系統調度的優先級要低於系統的其它硬件中斷優先級,即優先響應系統中的外部硬件中斷,因此PendSV的中斷優先級要配置爲最低,否則極可能在中斷上下文中產生任務調度。code

PendSV 異常會自動延遲上下文切換的請求,直到其它的 ISR 都完成了處理後才放行。爲實現這個機制,須要把 PendSV 編程爲最低優先級的異常。若是 OS 檢測到某 ISR 正在活動,它將懸起一個 PendSV 異常,以便緩期執行上下文切換。也就是說,只要將PendSV的優先級設爲最低的,systick即便是打斷了IRQ,它也不會立刻進行上下文切換,而是等到ISR執行完,PendSV 服務例程纔開始執行,而且在裏面執行上下文切換。過程如圖所示:而後獲取MSP主棧指針的地址,在Cortex-M中,0xE000ED08SCB_VTOR寄存器的地址,裏面存放的是向量表的起始地址。

加載k_next_task指向的任務控制塊到 R2,從上一篇文章可知任務控制塊的第一個成員就是棧頂指針,因此此時R2等於棧頂指針。

ps : 在調度器啓動時,k_next_taskk_curr_task是同樣的(k_curr_task = k_next_task

加載R2R0,而後將棧頂指針R0更新到psp,任務執行的時候使用的棧指針是psp

ps:sp指針有兩個,分別爲pspmsp。(能夠簡單理解爲:在任務上下文環境中使用psp,在中斷上下文環境使用msp,也不必定是正確的,這是我我的的理解)

R0爲基地址,將棧中向上增加的8個字的內容加載到CPU寄存器R4~R11,同時R0也會跟着自增

接着須要加載R0 ~ R三、R12以及LR、 PC、xPSR到CPU寄存器組,PC指針指向的是即將要運行的線程,而LR寄存器則指向任務的退出。由於這是第一次啓動任務,要所有手動把任務棧上的寄存器彈到硬件裏,才能進入第一個任務的上下文,由於一開始並無第一個任務運行的上下文環境,而在進入PendSV的時候須要上文保存,因此須要手動創造任務上下文環境(將這些寄存器加載到CPU寄存器組中),第一次的時候此彙編入口函數,sp是指向一個選好的任務的棧頂(k_curr_task)。

看看任務棧的初始化

從上面的瞭解,再來看看任務棧的初始化,可能會有更深一點的印象。主要了解如下幾點便可:

  • 獲取棧頂指針爲stk_base[stk_size]高地址,Cortex-M內核的棧是向下增加的。
  • R0、R一、R二、R三、R十二、R1四、R15和xPSR的位24是會被CPU自動加載與保存的。
  • xPSR的bit24必須置1,即0x01000000。
  • entry是任務的入口地址,即PC
  • R14 (LR)是任務的退出地址,因此任務通常是死循環而不會return
  • R0: arg是任務主體的形參
  • 初始化棧時sp指針會自減
__KERNEL__ k_stack_t *cpu_task_stk_init(void *entry,
                                              void *arg,
                                              void *exit,
                                              k_stack_t *stk_base,
                                              size_t stk_size)
{
    cpu_data_t *sp;

    sp = (cpu_data_t *)&stk_base[stk_size];
    sp = (cpu_data_t *)((cpu_addr_t)(sp) & 0xFFFFFFF8);

    /* auto-saved on exception(pendSV) by hardware */
    *--sp = (cpu_data_t)0x01000000u;    /* xPSR     */
    *--sp = (cpu_data_t)entry;          /* entry    */
    *--sp = (cpu_data_t)exit;           /* R14 (LR) */
    *--sp = (cpu_data_t)0x12121212u;    /* R12      */
    *--sp = (cpu_data_t)0x03030303u;    /* R3       */
    *--sp = (cpu_data_t)0x02020202u;    /* R2       */
    *--sp = (cpu_data_t)0x01010101u;    /* R1       */
    *--sp = (cpu_data_t)arg;            /* R0: arg  */

    /* Remaining registers saved on process stack */
    /* EXC_RETURN = 0xFFFFFFFDL
       Initial state: Thread mode +  non-floating-point state + PSP
       31 - 28 : EXC_RETURN flag, 0xF
       27 -  5 : reserved, 0xFFFFFE
       4       : 1, basic stack frame; 0, extended stack frame
       3       : 1, return to Thread mode; 0, return to Handler mode
       2       : 1, return to PSP; 0, return to MSP
       1       : reserved, 0
       0       : reserved, 1
     */
#if defined (TOS_CFG_CPU_ARM_FPU_EN) && (TOS_CFG_CPU_ARM_FPU_EN == 1U)
    *--sp = (cpu_data_t)0xFFFFFFFDL;
#endif

    *--sp = (cpu_data_t)0x11111111u;    /* R11      */
    *--sp = (cpu_data_t)0x10101010u;    /* R10      */
    *--sp = (cpu_data_t)0x09090909u;    /* R9       */
    *--sp = (cpu_data_t)0x08080808u;    /* R8       */
    *--sp = (cpu_data_t)0x07070707u;    /* R7       */
    *--sp = (cpu_data_t)0x06060606u;    /* R6       */
    *--sp = (cpu_data_t)0x05050505u;    /* R5       */
    *--sp = (cpu_data_t)0x04040404u;    /* R4       */

    return (k_stack_t *)sp;
}複製代碼

查找最高優先級任務

一個操做系統若是隻是具有了高優先級任務可以當即得到處理器並獲得執行的特色,那麼它仍然不算是實時操做系統。由於這個查找最高優先級任務的過程決定了調度時間是否具備肯定性,能夠簡單來講可使用時間複雜度來描述一下吧,若是系統查找最高優先級任務的時間是O(N),那麼這個時間會隨着任務個數的增長而增大,這是不可取的,TencentOS tiny的時間複雜度是O(1),它提供兩種方法查找最高優先級任務,經過TOS_CFG_CPU_LEAD_ZEROS_ASM_PRESENT宏定義決定。

  1. 第一種是使用普通方法,根據就緒列表中k_rdyq.prio_mask[]的變量判斷對應的位是否被置1。
  2. 第二種方法則是特殊方法,利用計算前導零指令CLZ,直接在k_rdyq.prio_mask[]這個32位的變量中直接得出最高優先級所處的位置,這種方法比普通方法更快捷,但受限於平臺(須要硬件前導零指令,在STM32中咱們就可使用這種方法)。

實現過程以下,建議看一看readyqueue_prio_highest_get函數,他的實現仍是很是精妙的~

__STATIC__ k_prio_t readyqueue_prio_highest_get(void)
{
    uint32_t *tbl;
    k_prio_t prio;

    prio    = 0;
    tbl     = &k_rdyq.prio_mask[0];

    while (*tbl == 0) {
        prio += K_PRIO_TBL_SLOT_SIZE;
        ++tbl;
    }
    prio += tos_cpu_clz(*tbl);
    return prio;
}複製代碼
__API__ uint32_t tos_cpu_clz(uint32_t val)
{
#if defined(TOS_CFG_CPU_LEAD_ZEROS_ASM_PRESENT) && (TOS_CFG_CPU_LEAD_ZEROS_ASM_PRESENT == 0u)
    uint32_t nbr_lead_zeros = 0;

    if (!(val & 0XFFFF0000)) {
        val <<= 16;
        nbr_lead_zeros += 16;
    }

    if (!(val & 0XFF000000)) {
        val <<= 8;
        nbr_lead_zeros += 8;
    }

    if (!(val & 0XF0000000)) {
        val <<= 4;
        nbr_lead_zeros += 4;
    }

    if (!(val & 0XC0000000)) {
        val <<= 2;
        nbr_lead_zeros += 2;
    }

    if (!(val & 0X80000000)) {
        nbr_lead_zeros += 1;
    }

    if (!val) {
        nbr_lead_zeros += 1;
    }

    return (nbr_lead_zeros);
#else
    return port_clz(val);
#endif
}複製代碼

任務切換的實現

從前面咱們也知道,任務切換是在PendSV中斷中進行的,這個中斷中實現的內容總結成一句精髓的話就是 上文保存,下文切換,直接看源代碼:

PendSV_Handler
    CPSID   I
    MRS     R0, PSP

_context_save
    ; R0-R3, R12, LR, PC, xPSR is saved automatically here
    IF {FPU} != "SoftVFP"
    ; is it extended frame?
    TST     LR, #0x10
    IT      EQ
    VSTMDBEQ  R0!, {S16 - S31}
    ; S0 - S16, FPSCR saved automatically here

    ; save EXC_RETURN
    STMFD   R0!, {LR}
    ENDIF

    ; save remaining regs r4-11 on process stack
    STMFD   R0!, {R4 - R11}

    ; k_curr_task->sp = PSP
    MOV32   R5, k_curr_task
    LDR     R6, [R5]
    ; R0 is SP of process being switched out
    STR     R0, [R6]

_context_restore
    ; k_curr_task = k_next_task
    MOV32   R1, k_next_task
    LDR     R2, [R1]
    STR     R2, [R5]

    ; R0 = k_next_task->sp
    LDR     R0, [R2]

    ; restore R4 - R11
    LDMFD   R0!, {R4 - R11}

    IF {FPU} != "SoftVFP"
    ; restore EXC_RETURN
    LDMFD   R0!, {LR}
    ; is it extended frame?
    TST     LR, #0x10
    IT      EQ
    VLDMIAEQ    R0!, {S16 - S31}
    ENDIF

    ; Load PSP with new process SP
    MSR     PSP, R0
    CPSIE   I
    ; R0-R3, R12, LR, PC, xPSR restored automatically here
    ; S0 - S16, FPSCR restored automatically here if FPCA = 1
    BX      LR

    ALIGN
    END複製代碼

PSP的值存儲到R0。當進入PendSVC_Handler時,上一個任務運行的環境即: xPSR,PC(任務入口地址),R14,R12,R3,R2,R1,R0這些CPU寄存器的值會自動存儲到任務的棧中,此時psp指針已經被自動更新。而剩下的r4~r11須要手動保存,這也是爲啥要在PendSVC_Handler中保存上文(_context_save)的緣由,主要是加載CPU中不能自動保存的寄存器,將其壓入任務棧中。

接着找到下一個要運行的任務k_next_task,將它的任務棧頂加載到R0,而後手動將新任務棧中的內容(此處是指R4~R11)加載到CPU寄存器組中,這就是下文切換,固然還有一些其餘無法自動保存的內容也是須要手動加載到CPU寄存器組的。手動加載完後,此時R0已經被更新了,更新psp的值,在退出PendSVC_Handler中斷時,會以psp做爲基地址,將任務棧中剩下的內容(xPSR,PC(任務入口地址),R14,R12,R3,R2,R1,R0)自動加載到CPU寄存器。

其實在異常發生時,R14中保存異常返回標誌,包括返回後進入任務模式仍是處理器模式、使用PSP堆棧指針仍是MSP堆棧指針。此時的r14等於0xfffffffd,最表示異常返回後進入任務模式(畢竟PendSVC_Handler優先級是最低的,會返回到任務中),SP以PSP做爲堆棧指針出棧,出棧完畢後PSP指向任務棧的棧頂。當調用 BX R14指令後,系統以PSP做爲SP指針出棧,把接下來要運行的新任務的任務棧中剩下的內容加載到CPU寄存器:R0、R一、R二、R三、R十二、R14(LR)、R15(PC)和xPSR,從而切換到新的任務。

SysTick

SysTick初始化

systick是系統的時基,並且它是內核時鐘,只要是M0/M3/M4/M7內核它都會存在systick時鐘,而且它是能夠被編程配置的,這就對操做系統的移植提供極大的方便。TencentOS tiny會在cpu_init函數中將systick進行初始化,即調用cpu_systick_init函數,這樣子就不須要用戶自行去編寫systick初始化相關的代碼。

__KERNEL__ void cpu_init(void)
{
    k_cpu_cycle_per_tick = TOS_CFG_CPU_CLOCK / k_cpu_tick_per_second;
    cpu_systick_init(k_cpu_cycle_per_tick);

#if (TOS_CFG_CPU_HRTIMER_EN > 0)
    tos_cpu_hrtimer_init();
#endif
}複製代碼

__KERNEL__ void cpu_systick_init(k_cycle_t cycle_per_tick)
{
    port_systick_priority_set(TOS_CFG_CPU_SYSTICK_PRIO);
    port_systick_config(cycle_per_tick);
}複製代碼

SysTick中斷

SysTick中斷服務函數是須要咱們本身編寫的,要在裏面調用一下TencentOS tiny相關的函數,更新系統時基以驅動系統的運行,SysTick_Handler函數的移植以下:

void SysTick_Handler(void)
{
  HAL_IncTick();
  if (tos_knl_is_running())
  {
    tos_knl_irq_enter();
    
    tos_tick_handler();
    
    tos_knl_irq_leave();
  }
}複製代碼

主要是須要調用tos_tick_handler函數將系統時基更新,具體見:

__API__ void tos_tick_handler(void)
{
    if (unlikely(!tos_knl_is_running())) {
        return;
    }

    tick_update((k_tick_t)1u);

#if TOS_CFG_TIMER_EN > 0u && TOS_CFG_TIMER_AS_PROC > 0u
    timer_update();
#endif

#if TOS_CFG_ROUND_ROBIN_EN > 0u
    robin_sched(k_curr_task->prio);
#endif
}複製代碼

不得不說TencentOS tiny源碼的實現很是簡單,我很是喜歡,在tos_tick_handler中,首先判斷一下系統是否已經開始運行,若是沒有運行將直接返回,若是已經運行了,那就調用tick_update函數更新系統時基,若是使能了TOS_CFG_TIMER_EN 宏定義表示使用軟件定時器,則須要更新相應的處理,此處暫且不說起。若是使能了TOS_CFG_ROUND_ROBIN_EN 宏定義,還須要更新時間片相關變量,稍後講解。

__KERNEL__ void tick_update(k_tick_t tick)
{
    TOS_CPU_CPSR_ALLOC();
    k_task_t *first, *task;
    k_list_t *curr, *next;

    TOS_CPU_INT_DISABLE();
    k_tick_count += tick;

    if (tos_list_empty(&k_tick_list)) {
        TOS_CPU_INT_ENABLE();
        return;
    }

    first = TOS_LIST_FIRST_ENTRY(&k_tick_list, k_task_t, tick_list);
    if (first->tick_expires <= tick) {
        first->tick_expires = (k_tick_t)0u;
    } else {
        first->tick_expires -= tick;
        TOS_CPU_INT_ENABLE();
        return;
    }

    TOS_LIST_FOR_EACH_SAFE(curr, next, &k_tick_list) {
        task = TOS_LIST_ENTRY(curr, k_task_t, tick_list);
        if (task->tick_expires > (k_tick_t)0u) {
            break;
        }

        // we are pending on something, but tick's up, no longer waitting
        pend_task_wakeup(task, PEND_STATE_TIMEOUT);
    }

    TOS_CPU_INT_ENABLE();
}複製代碼

tick_update函數的主要功能就是將k_tick_count +1,而且判斷一下時基列表k_tick_list(也能夠成爲延時列表吧)的任務是否超時,若是超時則喚醒該任務,不然就直接退出便可。關於時間片的調度也是很是簡單,將任務的剩餘時間片變量timeslice減一,而後當變量減到0時,將該變量進行重裝載timeslice_reload,而後切換任務knl_sched(),其實現過程以下:

__KERNEL__ void robin_sched(k_prio_t prio)
{
    TOS_CPU_CPSR_ALLOC();
    k_task_t *task;

    if (k_robin_state != TOS_ROBIN_STATE_ENABLED) {
        return;
    }

    TOS_CPU_INT_DISABLE();

    task = readyqueue_first_task_get(prio);
    if (!task || knl_is_idle(task)) {
        TOS_CPU_INT_ENABLE();
        return;
    }

    if (readyqueue_is_prio_onlyone(prio)) {
        TOS_CPU_INT_ENABLE();
        return;
    }

    if (knl_is_sched_locked()) {
        TOS_CPU_INT_ENABLE();
        return;
    }

    if (task->timeslice > (k_timeslice_t)0u) {
        --task->timeslice;
    }

    if (task->timeslice > (k_timeslice_t)0u) {
        TOS_CPU_INT_ENABLE();
        return;
    }

    readyqueue_move_head_to_tail(k_curr_task->prio);

    task = readyqueue_first_task_get(prio);
    if (task->timeslice_reload == (k_timeslice_t)0u) {
        task->timeslice = k_robin_default_timeslice;
    } else {
        task->timeslice = task->timeslice_reload;
    }

    TOS_CPU_INT_ENABLE();
    knl_sched();
}複製代碼

喜歡就關注我吧!

歡迎關注我公衆號

相關代碼能夠在公衆號後臺獲取。歡迎關注「物聯網IoT開發」公衆號

相關文章
相關標籤/搜索