线程的结束

优质
小牛编辑
126浏览
2023-12-01

线程的结束

现有问题

当内核线程终止时,会发生什么?如果就按目前的实现,我们会发现线程所执行的函数末尾会触发 Exception::InstructionPageFault 而终止,其中访问的的地址 stval = 0

这是因为内核线程在执行完 entry_point 所指向的函数后会返回到 ra 指向的地址,而我们没有为其赋初值(初值为 0)。此时,程序就会尝试跳转到 0x0 地址,而显然它是不存在的。

解决办法

很自然的,我们希望能够让内核线程在结束时触发一个友善的中断(而不是一个看上去像是错误的缺页异常),然后被操作系统释放。我们可能会想到系统调用,但很可惜我们无法使用它,因为系统调用的本质是一个环境调用 ecall,而在内核线程(内核态)中进行的环境调用是用来与 M 态通信的。我们之前实现的 SBI 调用就是使用的 S 态 ecall

因此,我们设计一个折衷的解决办法:内核线程将自己标记为“已结束”,同时触发一个普通的异常 ebreak。此时操作系统观察到线程的标记,便将其终止。

os/src/main.rs

/// 内核线程需要调用这个函数来退出
fn kernel_thread_exit() {
    // 当前线程标记为结束
    PROCESSOR.lock().current_thread().as_ref().inner().dead = true;
    // 制造一个中断来交给操作系统处理
    unsafe { llvm_asm!("ebreak" :::: "volatile") };
}

然后,我们将这个函数作为内核线程的 ra,使得它执行的函数完成后便执行 kernel_thread_exit()

os/src/main.rs

/// 创建一个内核进程
pub fn create_kernel_thread(
    process: Arc<Process>,
    entry_point: usize,
    arguments: Option<&[usize]>,
) -> Arc<Thread> {
    // 创建线程
    let thread = Thread::new(process, entry_point, arguments).unwrap();
    // 设置线程的返回地址为 kernel_thread_exit
    thread.as_ref().inner().context.as_mut().unwrap()
        .set_ra(kernel_thread_exit as usize);
    thread
}