鸿蒙内核源码之线程环境下的任务切换

在鸿蒙的内核线程就是任务,系列篇中说的任务和线程当一个东西去理解.
一般二种场景下需要切换任务上下文:
在线程环境下,从当前线程切换到目标线程,这种方式也称为软切换,能由软件控制的自主式切换.哪些情况下会出现软切换呢?
运行的线程申请某种资源(比如各种锁,读/写消息队列)失败时,需要主动释放cpu的控制权,将自己挂入等待队列,调度算法重新调度新任务运行.
每隔10ms就执行一次的ostickhandler节拍处理函数,检测到任务的时间片用完了,就发起任务的重新调度,切换到新任务运行.
不管是内核态的任务还是用户态的任务,于切换而言是统一处理,一视同仁的,因为切换是需要换栈运行,寄存器有限,需要频繁的复用,这就需要将当前寄存器值先保存到任务自己的栈中,以便别人用完了轮到自己再用时恢复寄存器当时的值,确保老任务还能继续跑下去. 而保存寄存器顺序的结构体叫:任务上下文(taskcontext).
在中断环境下,从当前线程切换到目标线程,这种方式也称为硬切换.不由软件控制的被动式切换.哪些情况下会出现硬切换呢?
由硬件产生的中断,比如 鼠标,键盘外部设备每次点击和敲打,屏幕的触摸,usb的插拔等等这些都是硬中断.同样的需要切换栈运行,需要复用寄存器,但与软切换不一样的是,硬切换会切换工作模式(中断模式).所以会更复杂点,但道理还是一样要保存和恢复切换现场寄存器的值, 而保存寄存器顺序的结构体叫:任务中断上下文(taskirqcontext).
本篇说清楚在线程环境下切换(软切换)的实现过程.中断切换(硬切换)实现过程将在鸿蒙内核源码分析(总目录)中断切换篇中详细说明.
本篇具体说清楚以下几个问题:
任务上下文(taskcontext)怎么保存的?
代码的实现细节是怎样的?
如何保证切换不会发生错误,指令不会丢失?
在 鸿蒙内核源码分析(总目录) 系列篇中已经说清楚了调度机制,线程概念,寄存器,cpu,工作模式,这些是读懂本篇的基础,建议先前往翻看,不然理解本篇会费劲.本篇代码量较多,涉及c和汇编代码,代码都添加了注释,试图把任务的整个切换过程逐行逐行说清楚.
前置条件
一个任务要跑起来,需要两个必不可少的硬性条件:
1.从代码段哪个位置取指令? 也就是入口地址,main函数是应用程序的入口地址, run()是new一个线程执行的入口地址.高级语言是这么叫,但到了汇编层的叫法就是pc寄存器.给pc寄存器喂妹值,指令就从哪里取.
2.运行的场地(栈空间)在哪里? arm有7种工作模式,到了进程层面只需要考虑内核模式和用户模式两种,对应到任务会有内核态栈空间和用户态栈空间.内核模式的任务只有内核态的栈空间,用户模式任务二者都有.栈空间是在初始化一个任务时就分配指定好的.以下是两种栈空间的初始化过程.为了精练省去了部分代码,留下了核心部分.
//任务控制块中对两个栈空间的描述typedef struct { void *stackpointer; /**< task stack pointer */ //内核态栈指针,sp位置,切换任务时先保存上下文并指向taskcontext位置. uint32 stacksize; /**< task stack size */ //内核态栈大小 uintptr topofstack; /**< task stack top */ //内核态栈顶 bottom = top + size // .... uintptr userarea; //使用区域,由运行时划定,根据运行态不同而不同 uintptr usermapbase; //用户态下的栈底位置 uint32 usermapsize; /**pc = (uintptr)ostaskentry;//程序计数器,cpu首次执行task时跑的第一条指令位置 taskcontext->lr = (uintptr)ostaskexit; /* lr should be kept, to distinguish it's thumb or arm instruction */ taskcontext->resved = 0x0; taskcontext->r[0] = taskid; /* r0 */ taskcontext->r[index++] = 0x01010101; /* r1, 0x01010101 : reg initialed magic word */ //0x55 for (; index r[index] = taskcontext->r[index - 1] + taskcontext->r[1]; /* r2 - r12 */ } taskcontext->regpsr = psr_mode_svc_arm; /* cpsr (enable irq and fiq interrupts, arm-mode) */ return (void *)taskcontext;}//用户态运行栈初始化lite_os_sec_text_init void osusertaskstackinit(taskcontext *context, tsk_entry_func taskentry, uintptr stack){ context->regpsr = psr_mode_usr_arm;//工作模式:用户模式 + 工作状态:arm context->r[0] = stack;//栈指针给r0寄存器 context->sp = truncate(stack, loscfg_stack_point_align_size);//给sp寄存器值使用 context->lr = 0;//保存子程序返回地址 例如 a call b ,在b中保存 a地址 context->pc = (uintptr)taskentry;//入口函数} 您一定注意到了taskcontext,说的全是它,这就是任务上下文结构体,理解它是理解任务切换的钥匙.它不仅在c语言层面出现,而且还在汇编层出现,taskcontext是连接或者说打通 c->汇编->c 实现任务切换的最关键概念.本篇全是围绕着它来展开.先看看它张啥样,look!
taskcontext 任务上下文
typedef struct {#if !defined(loscfg_arch_fpu_disable) uint64 d[fp_regs_num]; /* d0-d31 */ uint32 regfpscr; /* fpscr */ uint32 regfpexc; /* fpexc */#endif uint32 resved; /* it's stack 8 aligned */ uint32 regpsr; uint32 r[gen_regs_num]; /* r0-r12 */ uint32 sp; /* r13 */ uint32 lr; /* r14 */ uint32 pc; /* r15 */} taskcontext; 结构很简单,目的更简单,就是用来保存寄存器现场的值的. 鸿蒙内核源码分析(总目录) 系列寄存器篇中已经说过了,到了汇编层就是寄存器在玩,当cpu工作在用户和系统模式下时寄存器是复用的,玩的是17个寄存器和内存地址,访问内存地址也是通过寄存器来玩.
哪17个? r0~r15和cpsr. 当调度(主动式)或者中断(被动式)发生时.将这17个寄存器压入任务的内核栈的过程叫保护案发现场.从任务栈中弹出依次填入寄存器的过程叫恢复案发现场.
从栈空间的具体哪个位置开始恢复呢? 答案是:stackpointer,任务控制块(lostaskcb)的首个变量.对应到汇编层的就是sp寄存器.
而taskcontext(任务上下文)就是一定的顺序来保存和恢复这17个寄存器的.任务上下文在任务还没有开始执行的时候就已经保存在内核栈中了,只不过是一些默认的值,ostaskstackinit干的就是这个默认的事. 而osusertaskstackinit是对用户栈的初始化,改变的是(cpsr)工作模式和sp寄存器.
新任务的运行栈指针(stackpointer)给sp寄存器意味着切换了运行栈,这是本篇最重要的一句话.
以下通过汇编代码逐行分析如何保存和恢复taskcontext(任务上下文)
osschedresched 调度算法
//调度算法的实现void osschedresched(void){ // ...此处省去 ... /* do the task context switch */ ostaskschedule(newtask, runtask);//切换任务上下文,注意ostaskschedule是一个汇编函数 见于 los_dispatch.s} 在鸿蒙内核源码分析(总目录)之调度机制篇中,留了一个问题,ostaskschedule不是一个c函数,而是个汇编函数,就没有往下分析了,本篇要完成整个分析过程.ostaskschedule实现了任务的上下文切换,汇编代码见于los_dispatch.s中
ostaskschedule的参数指向的是新老两个任务,这两个参数分别保存在r0,r1寄存器中.
ostaskschedule 汇编实现
读这段汇编代码一定要对照上面的taskcontext,不然很难看懂,容易懵圈,但对照着看就秒懂.
/* * r0: new task * r1: run task */ostaskschedule: /*任务调度,ostaskschedule的目的是将寄存器值按taskcontext的格式保存起来*/ mrs r2, cpsr /*mrs 指令用于将特殊寄存器(如 cpsr 和 spsr)中的数据传递给通用寄存器,要读取特殊寄存器的数据只能使用 mrs 指令*/ stmfd sp!, {lr} /*返回地址入栈,lr = pc-4 ,对应taskcontext->pc(r15寄存器)*/ stmfd sp!, {lr} /*再次入栈对应,对应taskcontext->lr(r14寄存器)*/ /* jump sp */ sub sp, sp, #4 /* 跳的目的是为了,对应taskcontext->sp(r13寄存器)*/ /* push r0-r12*/ stmfd sp!, {r0-r12} @对应taskcontext->r[gen_regs_num](r0~r12寄存器)。 stmfd sp!, {r2} /*r2 入栈 对应taskcontext->regpsr*/ /* 8 bytes stack align */ sub sp, sp, #4 @栈对齐,对应taskcontext->resved /* save fpu registers */ push_fpu_regs r2 /*保存fpu寄存器*/ /* store sp on running task */ str sp, [r1] @在运行的任务栈中保存sp,即runtask->stackpointer = spostaskcontextload: @加载上下文 /* clear the flag of ldrex */ @ldrex 可从内存加载数据,如果物理地址有共享tlb属性,则ldrex会将该物理地址标记为由当前处理器独占访问,并且会清除该处理器对其他任何物理地址的任何独占访问标记。 clrex @清除ldrex指令的标记 /* switch to new task's sp */ ldr sp, [r0] @ 即:sp = task->stackpointer /* restore fpu registers */ pop_fpu_regs r2 @恢复fpu寄存器,这里用了汇编宏r2是宏的参数 /* 8 bytes stack align */ add sp, sp, #4 @栈对齐 ldmfd sp!, {r0} @此时sp!位置保存的是cpsr的内容,弹出到r0 mov r4, r0 @r4=r0,将cpsr保存在r4, 将在oskerneltaskload中保存到spsr and r0, r0, #cpsr_mask_mode @r0 =r0&cpsr_mask_mode ,目的是清除高16位 cmp r0, #cpsr_user_mode @r0 和 用户模式比较 bne oskerneltaskload @非用户模式则跳转到oskerneltaskload执行,跳出 /*此处省去 loscfg_kernel_smp 部分*/ mvn r3, #cpsr_int_disable @按位取反 r3 = 0x3f and r4, r4, r3 @使能中断 msr spsr_cxsf, r4 @修改spsr值 /* restore r0-r12, lr */ ldmfd sp!, {r0-r12} @恢复寄存器值 ldmfd sp, {r13, r14}^ @恢复sp和lr的值,注意此时sp值已经变了,cpu换地方上班了. add sp, sp, #(2 * 4)@sp = sp + 8 ldmfd sp!, {pc}^ @恢复pc寄存器值,如此一来 sp和pc都有了新值,完成了上下文切换.完美!oskerneltaskload: @内核任务的加载 msr spsr_cxsf, r4 @将r4整个写入到程序状态保存寄存器 /* restore r0-r12, lr */ ldmfd sp!, {r0-r12} @出栈,依次保存到 r0-r12,其实就是恢复现场 add sp, sp, #4 @sp=sp+4 ldmfd sp!, {lr, pc}^ @返回地址赋给pc指针,直接跳出. 解读
汇编分成了三段 ostaskschedule, ostaskcontextload, oskerneltaskload.
第一段ostaskschedule其实就是在保存现场.代码都有注释,对照着taskcontext来的,它就干了一件事把17个寄存器的值按taskcontext的格式入栈,因为鸿蒙用栈方式采用的是满栈递减的方式,所以存放顺序是从最后一个往前依次入栈.
连着来两句stmfd sp!, {lr}之前让笔者懵圈了很久, 看了taskcontext才恍然大悟,因为三级流水线的原因,lr和pc寄存器之间是差了一条指令的,lr指向了处于译码阶段指令,而pc指向了取指阶段的指令,所以此处做了两次lr入栈,其实是保存了未执行的译码指令地址,确保执行不会丢失一条指令.
r1是正在运行的任务栈, ostaskschedule总的理解是在任务r1的运行栈中插入一个taskcontext结构块.而str sp, [r1],是改变了lostaskcb->stackpointer的值,这个值只能在汇编层进行精准的改变,而在整个鸿蒙内核c代码层面都没有看到对它有任何修改的地方.这个改变意义极为重要.因为新的任务被调度后的第一件事情就是恢复现场!!!
在ostaskschedule执行完成后,因为pc寄存器并没有发生跳转,所以紧接着往下执行ostaskcontextload
ostaskcontextload的任务就是恢复现场,谁的现场?当然是r0: new task的,所以第一条指令就是clrex,清除干净后立马执行ldr sp, [r0],所指向的就是lostaskcb->stackpointer,这个位置存的是新任务的taskcontext结构块,是上一次r0任务被打断时保存下来当时这17个寄存器的值啊,依次出栈就是恢复这17个寄存器的值.
ostaskcontextload在开始之前会判断下工作模式,即判断下是内核栈还是用户栈,两种处理方式稍有不同.但都是在恢复现场.
bne oskerneltaskload是查询cpsr后判断此时为内核栈的现场恢复过程,代码很简单就是恢复17个寄存器. 如此一来,任务执行的两个条件,第一个sp的在ldr sp, [r0]时就有了.第二个条件:pc寄存器的值也在最后一条汇编ldmfd sp!, {lr, pc}^ 也已经有了.改变了pc和lr有了新值,下一条指令位置一样是上次任务被中断时还没被执行的处于译码阶段的指令地址.
如果是用户态区别是需要恢复中断.因为用户模式的优先级是最低的,必须允许响应中断,也是依次恢复各寄存器的值,最后一句ldmfd sp!, {pc}^结束本次旅行,下一条指令位置一样是上次任务被中断时还没被执行的处于译码阶段的指令地址.
如此,说清楚了任务上下文切换的整个过程,初看可能不太容易理解,建议多看几篇,用笔画下栈的运行过程,脑海中会很清晰的浮现出整个切换过程的运行图.


iPhone 8这些新功能已确定 想想都激动!
创维发布两款全新产品,电视功能转型成为智慧家电
GE 荣获两项海上风电研发奖项 空客实现2021年商用飞机交付目标
华为P30系列国外售价曝光,新U麒麟985或在研发
viss透明屏,在商业广告中,广告展示效果有什么不同?
鸿蒙内核源码之线程环境下的任务切换
从硬件和软件入手的UPS抗干扰技术
并联有源电力滤波器交流侧滤波电感的优化设计
功率MOSFET零电压软开关ZVS的基础认识
ARM Cortex-M3 处理器的主要特性分析
电气原理图中电器元件的绘制规则
I2C存储器的应用及结构原理解析
独立按键和矩阵键盘驱动原理是什么
合路器和功分器在无线对讲系统中有何区别?
小米8青春版:顶级拍照硬件,不负青春韶华
步入式高低温试验箱:模拟极端环境的关键设备
联想真正的苦日子还没到 花旗称零部件涨价将伤害后续盈利
苹果为何不给iPhone 12标配充电器?
行业 | 日韩冲突升级,存储器产能紧缺?库存成关键
西门子常用PLC指令表