lab5：深入理解进程切换

在调度器选择下一个进程后，由context_switch完成进程的上下文切换，将CPU当前运行的进程prev切换到另一个进程next

/*
 * context_switch - switch to the new MM and the new thread's register state.
 */
static __always_inline struct rq *
context_switch(struct rq *rq, struct task_struct *prev,
	       struct task_struct *next, struct rq_flags *rf)
{
	prepare_task_switch(rq, prev, next);

	/*
	 * For paravirt, this is coupled with an exit in switch_to to
	 * combine the page table reload and the switch backend into
	 * one hypercall.
	 */
	arch_start_context_switch(prev);

	/*
	 * kernel -> kernel   lazy + transfer active
	 *   user -> kernel   lazy + mmgrab() active
	 *
	 * kernel ->   user   switch + mmdrop() active
	 *   user ->   user   switch
	 */
	if (!next->mm) {                                // to kernel
		enter_lazy_tlb(prev->active_mm, next);

		next->active_mm = prev->active_mm;
		if (prev->mm)                           // from user
			mmgrab(prev->active_mm);
		else
			prev->active_mm = NULL;
	} else {                                        // to user
		membarrier_switch_mm(rq, prev->active_mm, next->mm);
		/*
		 * sys_membarrier() requires an smp_mb() between setting
		 * rq->curr / membarrier_switch_mm() and returning to userspace.
		 *
		 * The below provides this either through switch_mm(), or in
		 * case 'prev->active_mm == next->mm' through
		 * finish_task_switch()'s mmdrop().
		 */
		switch_mm_irqs_off(prev->active_mm, next->mm, next);

		if (!prev->mm) {                        // from kernel
			/* will mmdrop() in finish_task_switch(). */
			rq->prev_mm = prev->active_mm;
			prev->active_mm = NULL;
		}
	}

	rq->clock_update_flags &= ~(RQCF_ACT_SKIP|RQCF_REQ_SKIP);

	prepare_lock_switch(rq, next, rf);

	/* Here we just switch the register state and the stack. */
	switch_to(prev, next, prev);
	barrier();

	return finish_task_switch(prev);
}

函数开始调用“prepare_task_switch()”函数，在实际上下文切换之前执行任何必要的准备工作。然后调用“arch_start_context_switch()”函数，用于执行与上下文切换所需的特定于体系结构的操作。

代码的下一部分确定下一个任务是内核任务还是用户任务：

• 如果下一个任务不是用户任务，则进入惰性 TLB 模式，并设置“next”任务的活动内存管理器为“prev”任务的活动内存管理器。如果“prev”任务的 mm 指针不为 NULL（即从用户任务切换），则调用“mmgrab()”函数来增加“prev”任务的活动内存管理器的引用计数，否则将“prev”任务的活动内存管理器设置为 NULL。
• 如果下一个任务是用户任务，则调用“membarrier_switch_mm()”函数来执行特定于内存栅栏的操作，然后调用“switch_mm_irqs_off()”函数来执行上下文切换，并将“prev”任务的活动内存管理器设置为 NULL（如果从内核任务切换）。

ENTRY(__switch_to_asm)
    pushq    %rbp
    pushq    %rbx
    pushq    %r12
    pushq    %r13
    pushq    %r14
    pushq    %r15
    /* switch stack */
    movq    %rsp, TASK_threadsp(%rdi)
    movq    TASK_threadsp(%rsi), %rsp
 */
    popq    %r15
    popq    %r14
    popq    %r13
    popq    %r12
    popq    %rbx
    popq    %rbp
    jmp    __switch_to
END(__switch_to

调用“switch_to()”函数来切换寄存器状态和堆栈。最后，代码调用“finish_task_switch()”函数来完成上下文切换，使得能够正确的释放锁并返回指向“prev”任务的指针。