Linux操作系统分析Lab5：深入理解进程切换

一、实验简介

Linux 是一个多任务操作系统，它能支持远大于 CPU 数量的任务同时运行。但实际上同一时刻只会有 CPU 数量的进程在运行，等 CPU 时间片到了之后，进程调度器就会把 CPU 资源分配给其他进程。在这个过程中就会涉及到进程之间的切换，这时候就需要将当前进程的上下文信息保存下来，随后加载被调度进程的上下文信息，这就是上下文切换。这里所说的上下文信息，既包括虚拟内存、栈、全局变量等用户态的资源，也包括内核堆栈、寄存器等内核态的资源。

本文以arm64和x86为例，分析Linux内核中进程切换的具体操作。

二、实验内容

1. 进程切换的时机

一般来说，进程切换有以下几个时机：

• 某个进程时间片耗尽，会被系统挂起，切换到其他等待 CPU 的进程。
• 进程所需系统资源不足，需要等到资源满足时才可运行，此时会被挂起，其他进程会被调度。
• 进程通过 sleep 方法主动挂起，其他进程就有机会被调度。
• 有更高优先级的进程，当前进程会被挂起，高优先级进程会被调度。
• 硬件中断时，CPU 上的进程会被中断挂起，转而执行内核中的中断服务程序。

当发生如上几种情况的时候，就会发生进程调度，进而发生进程上下文切换。

2. 执行进程切换

在中断处理程序中，会直接调用schedule()函数实现进程调度：

1. 首先，schedule()函数会调用next = pick_next_task(rq,prev);所做的工作就是根据调度算法策略，选取要执行的下一个进程。
2. 其次，调用context_switch(rq, prev,next);，完成进程上下文切换。其中，最关键的switch_to(prev,next, last)：切换堆栈和寄存器的状态。

2.1 context_switch进程上下文切换

context_switch其实是一个分配器，他会调用所需的特定体系结构的方法：

• 调用switch_mm()，把虚拟内存从一个进程映射切换到新进程中。switch_mm更换通过task_struct->mm描述的内存管理上下文，该工作的细节取决于处理器，主要包括加载页表，刷出地址转换缓存，向内存管理单元(MMU)提供新的信息。
• 调用switch_to()，从上一个进程的处理器状态切换到新进程的处理器状态。这包括保存、恢复栈信息和寄存器信息。

switch_to切换处理器寄存器的内容和内核栈(虚拟地址空间的用户部分已经通过switch_mm变更，其中也包括了用户状态下的栈，因此switch_to不需要变更用户栈，只需变更内核栈)，此段代码严重依赖于体系结构，且代码通常都是用汇编语言编写。
context_switch函数建立next进程的地址空间。进程描述符的active_mm字段指向进程所使用的内存描述符，而mm字段指向进程所拥有的内存描述符。对于一般的进程，这两个字段有相同的地址，但是，内核线程没有它自己的地址空间所以它的 mm字段总是被设置为 NULL。

context_switch( )函数保证：如果next是一个内核线程, 它使用prev所使用的地址空间。

由于不同架构下地址映射的机制有所区别，而寄存器等信息也是依赖于架构的，因此switch_mm和switch_to两个函数均是体系结构相关的。

context_switch定义在kernel/sched/core.c中，如下所示：

/*
 * 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);
}

context_switch函数是用来完成任务上下文切换的,切换cr3,栈顶寄存器。下面是该函数主要完成的工作：

• 通过 prepare_task_switch() 进行准备，设置 prev 和 next 的一些状态，例如 prev 的状态设置为
  TASK_RUNNING，同时更新 rq 的 curr 和 next 任务的时间戳等。
• 在切换前，调用 arch_start_context_switch(prev)，切换前的准备工作，包括更新上下文寄存器、设置栈等。
• 根据 prev 和 next 任务所属的内存空间类型进行处理： 如果 next 是内核空间任务，调用 enter_lazy_tlb()
  进行懒惰处理，同时将 next 的 active_mm 设置为 prev 的 active_mm，并增加 prev 的 active_mm
  的引用计数。
• 如果 next 是用户空间任务，则需要进行内存空间切换。如果 prev 任务属于内核空间，将 prev 的 active_mm 设置为
  NULL，同时记录 prev 的 active_mm 以便在 finish_task_switch() 中释放，而对于 prev
  属于用户空间的任务，则需要调用 mmgrab() 来增加 prev 的 active_mm 的引用计数。
• 调用 switch_mm_irqs_off() 进行内存空间切换，更新 prev 和 next
  的内存空间等信息。在arm64架构中，实际上调用的是switch_mm函数
• 如果 prev 是内核空间任务，则记录其 active_mm 以便在 finish_task_switch() 中释放。
• 调用 prepare_lock_switch() 对 prev 和 next 进行加锁和解锁。
• 调用 switch_to() 进行寄存器和栈切换，将控制权转移到 next 任务中。
• 调用 finish_task_switch() 完成任务切换的一些收尾工作，例如释放 prev 的内存空间引用计数等，并返回 rq。

2.2 X86架构中的switch_to函数分析

进程关键上下文的切换swtich_to，见arch/x86/include/asm/switch_to.h。其中的__switch_to_asm是一段汇编代码，见arch/x86/entry/entry_64.S：

ENTRY(__switch_to_asm)
  UNWIND_HINT_FUNC
  /*
   * Save callee-saved registers
   * This must match the order in inactive_task_frame
   */
  pushq  %rbp
  pushq  %rbx
  pushq  %r12
  pushq  %r13
  pushq  %r14
  pushq  %r15

  /* switch stack */
  movq  %rsp, TASK_threadsp(%rdi) // 保存旧进程的栈顶
  movq  TASK_threadsp(%rsi), %rsp // 恢复新进程的栈顶

  /* restore callee-saved registers */
  popq  %r15
  popq  %r14
  popq  %r13
  popq  %r12
  popq  %rbx
  popq  %rbp

  jmp  __switch_to
END(__switch_to_asm)

这段汇编代码有内核堆栈栈顶指针RSP寄存器的切换，但是并没有显式的完成RIP寄存器的切换。
实际上__switch_to_asm是在C代码中调用的，也就是使用call指令，而这段汇编的结尾是jmp __switch_to，__switch_to函数是C代码最后有个return，也就是ret指令。将__switch_to_asm和__switch_to结合起来，正好是call指令和ret指令的配对出现。call指令压栈RIP寄存器到进程切换前的prev进程内核堆栈；而ret指令出栈存入RIP寄存器的是进程切换之后的next进程的内核堆栈栈顶数据。