使用Rust开发操作系统(异常处理)

在上一篇文章中我们完成了对GDT,IDT,TSS以及PIC8259A的初始化以及加载工作,现在我们需要为操作系添加一些异常处理功能,并且使用8259A编写一个简易的键盘驱动

注册异常处理函数

在实现IDT的过程中我们提到了x86-interrut调用约定(详情请查看使用Rust开发操作系统(中断描述符表–IDT)关于调用约定章节)注册处理函数的诀窍是 首先编写一个 extern x86-interrupt中断处理函数,函数的参数是根据你要处理异常的类别去传递的,主要分为2种,带错误码的和不带错误码的,然后将编写的函数注册到对应的异常中去
例如:

// 注:库及结构已经更改过,请以github代码为准(https://github.com/VenmoTools/OperatingSystem)
use system::ia_32e::descriptor::{InterruptDescriptorTable, InterruptStackFrame};
use system::bits::PageFaultErrorCode;

// 不带错误码
extern "x86-interrupt" fn example_interrupt(_stackframe: &mut InterruptStackFrame) {
}
// 带错误码(错误码需要解析)
extern "x86-interrupt" fn exmple_with_err_code(stackframe: &mut InterruptStackFrame, err: u64) {
 
}
// 页异常需要单独的错误码
extern "x86-interrupt" fn page_fault(stackframe: &mut InterruptStackFrame, code: PageFaultErrorCode) {
}

在使用之前一定在kernel/src/main.rs或者kernel/src/lib.rs文件的头部添加一下内容

#![feature(abi_x86_interrupt)]

注册异常处理函数例子

extern "x86-interrupt" fn page_fault(stackframe: &mut InterruptStackFrame, code: PageFaultErrorCode) {
	......
}
....
let mut idt = InterruptDescriptorTable::new();
idt.page_fault.set_handler_fn(page_fault);
....

需要注意的是,中断和异常都需要用到IDT,注册中断处理函数和异常处理函数的方式有很大区别在下文中我们详细阐述注册中断处理函数

定义异常处理函数

好了,掌握异常处理函数的注册技巧后我们开始注册20个异常处理函数(木有办法。。就是这么多),我们暂时只打印异常信息不做具体处理(为了完整性,列出了所有异常)

// in kernel/src/descriptor/idt.rs
// 调试中断
extern "x86-interrupt" fn breakpoint(stackframe: &mut InterruptStackFrame) {
    println!("EXCEPTION:BREAKPOINT\n{:#?}", stackframe);
}
// 缺页错误
extern "x86-interrupt" fn page_fault(stackframe: &mut InterruptStackFrame, code: PageFaultErrorCode) {
    println!("EXCEPTION:PageFault");
    println!("Error Code:{:?}", code);
    println!("{:#?}", stackframe);
    loop_hlt();
}
// 除零错误
extern "x86-interrupt" fn divide_by_zero(stackframe: &mut InterruptStackFrame) {
    println!("EXCEPTION:divide_by_zero\n{:#?}", stackframe);
}
// NMI不可屏蔽中断
extern "x86-interrupt" fn non_maskable_interrupt(stackframe: &mut InterruptStackFrame) {
    println!("EXCEPTION:non_maskable_interrupt\n{:#?}", stackframe);
}
// 调试
extern "x86-interrupt" fn debug(stackframe: &mut InterruptStackFrame) {
    println!("EXCEPTION:debug\n{:#?}", stackframe);
}
/// 溢出异常
extern "x86-interrupt" fn overflow(stackframe: &mut InterruptStackFrame) {
    println!("EXCEPTION:overflow\n{:#?}", stackframe);
}
/// 超出范围
extern "x86-interrupt" fn bound_range_exceeded(stackframe: &mut InterruptStackFrame) {
    println!("EXCEPTION:bound_range_exceeded\n{:#?}", stackframe);
}
/// 无效的操作码
extern "x86-interrupt" fn invalid_opcode(stackframe: &mut InterruptStackFrame) {
    println!("EXCEPTION:invalid_opcode\n{:#?}", stackframe);
}
/// 设备不可用
extern "x86-interrupt" fn device_not_available(stackframe: &mut InterruptStackFrame) {
    println!("EXCEPTION:device_not_available\n{:#?}", stackframe);
}
/// x87浮点异常
extern "x86-interrupt" fn x87_floating_point(stackframe: &mut InterruptStackFrame) {
    println!("EXCEPTION:x87_floating_point\n{:#?}", stackframe);
}
/// 机器检查
extern "x86-interrupt" fn machine_check(stackframe: &mut InterruptStackFrame) {
    println!("EXCEPTION:machine_check\n{:#?}", stackframe);
}
/// SIMD浮点异常
extern "x86-interrupt" fn simd_floating_point(stackframe: &mut InterruptStackFrame) {
    println!("EXCEPTION:simd_floating_point\n{:#?}", stackframe);
}
/// 虚拟化异常
extern "x86-interrupt" fn virtualization(stackframe: &mut InterruptStackFrame) {
    println!("EXCEPTION:virtualization\n{:#?}", stackframe);
}
/// 无效TSS
extern "x86-interrupt" fn invalid_tss(stackframe: &mut InterruptStackFrame, _err: u64) {
    println!("EXCEPTION:invalid_tss\n{:#?}\nCode:{:#?}\n", stackframe, _err);
}
/// 段不存在
extern "x86-interrupt" fn segment_not_present(stackframe: &mut InterruptStackFrame, _err: u64) {
    println!("EXCEPTION:segment_not_present\n{:#?}\nCode:{:#?}\n", stackframe, _err);
}
/// 堆栈段错误
extern "x86-interrupt" fn stack_segment_fault(stackframe: &mut InterruptStackFrame, _err: u64) {
    println!("EXCEPTION:stack_segment_fault\n{:#?}\nCode:{:#?}\n", stackframe, _err);
}
/// 常规保护异常
extern "x86-interrupt" fn general_protection_fault(stackframe: &mut InterruptStackFrame, _err: u64) {
    println!("EXCEPTION:general_protection_fault\n{:#?}\nCode:{:#?}\n", stackframe, _err);
}
/// 对齐检查异常
extern "x86-interrupt" fn alignment_check(stackframe: &mut InterruptStackFrame, _err: u64) {
    println!("EXCEPTION:alignment_check\n{:#?}\nCode:{:#?}\n", stackframe, _err);
}
/// 安全检查异常
extern "x86-interrupt" fn security_exception(stackframe: &mut InterruptStackFrame, _err: u64) {
    println!("EXCEPTION:security_exception\n{:#?}\nCode:{:#?}\n", stackframe, _err);
}

注册异常处理函数

好了,我们完成了19个异常处理函数,还有一个双重异常处理函数我们在下文编写

然后我们开始注册

// in kernel/src/descriptor/idt.rs
use system::ia_32e::descriptor::{InterruptDescriptorTable, InterruptStackFrame};
use system::bits::PageFaultErrorCode;

lazy_static! {
    static ref IDT: InterruptDescriptorTable = {
        let mut idt = InterruptDescriptorTable::new();
        idt.breakpoint.set_handler_fn(breakpoint);
        idt.page_fault.set_handler_fn(page_fault);
        idt.divide_by_zero.set_handler_fn(divide_by_zero);
        idt.invalid_tss.set_handler_fn(invalid_tss);
        idt.security_exception.set_handler_fn(security_exception);
        idt.segment_not_present.set_handler_fn(segment_not_present);
        idt.alignment_check.set_handler_fn(alignment_check);
        idt.bound_range_exceeded.set_handler_fn(bound_range_exceeded);
        idt.device_not_available.set_handler_fn(device_not_available);
        idt.general_protection_fault.set_handler_fn(general_protection_fault);
        idt.invalid_opcode.set_handler_fn(invalid_opcode);
        idt.machine_check.set_handler_fn(machine_check);
        idt.non_maskable_interrupt.set_handler_fn(non_maskable_interrupt);
        idt.virtualization.set_handler_fn(virtualization);
        idt.x87_floating_point.set_handler_fn(x87_floating_point);
        idt.stack_segment_fault.set_handler_fn(stack_segment_fault);
        idt.simd_floating_point.set_handler_fn(simd_floating_point);
        idt.overflow.set_handler_fn(overflow);
        idt.debug.set_handler_fn(debug);
        idt
    };
}

好了,这样子我们就注册成功了,当发生异常的时候我们就可以显示异常的信息了

双重异常Double Faults

当系统正在处理一个异常时,如果又检测到一个异常,处理器试图向系统通知一个双重故障

双重故障的行为类似于普通异常,但是提供双重故障处理程序确实很重要,因为如果未处理双重故障,则会发生致命的三次故障。无法捕获三重故障,大多数硬件都会对复位系统。

我们为其编写一个双重异常处理函数

extern "x86-interrupt" fn double_fault(stackframe: &mut InterruptStackFrame, _err: u64) {
    println!("EXCEPTION:DOUBLE FAULT\n{:#?}", stackframe);
}

我们编写的这个双重异常处理函数似乎跟其他的异常处理函数没有什么区别,但是我们能捕获一个基本的双重异常,防止硬件对系统进行复位,但是双重异常处理没有这么简单,在开头我们只说了双重错误是一种特殊的异常,它发生在CPU无法调用异常处理程序时,那么发生异常时转到对应的异常处理函数时如何恢复?当处理一个异常时又发生一个异常又该怎么处理?

在AMD64手册具有确切的定义(在第8.2.9节中)

在处理先前(第一个)异常处理程序期间发生第二个异常时,可能会发生双重错误异常”。这个 “可能”很重要:只有异常的非常特殊的组合才会导致双重错误。这些组合是

+--------------------------+------------------------+
| First Exception          |Second Exception        |
+--------------------------+------------------------+
|Divide-by-zero,           |Invalid TSS,            |
|Invalid TSS,              |Segment Not Present,    |
|Segment Not Present,      |Stack-Segment Fault,    |
|Stack-Segment Fault,      |General Protection Fault|
|General Protection Fault  |                        |
+--------------------------+------------------------+
|                          |Page Fault,             |
|                          |Invalid TSS,            |
|Page Fault                |Segment Not Present,    |
|                          |Stack-Segment Fault,    |
|                          |General Protection Fault|
+--------------------------+------------------------+

例如:

当正处理一个Divide-by-zero异常时,有可能又产生一个Page Fault异常。在这种情况下,通知给系统的是一个Page Fault异常而不是Double Fault异常。但是,如果正处理一个Invalid TSS时,又产生了一个General Protection Fault|;或者如果正处理一个Page Fault时,又产生了Page Fault,那么就引起双重故障。

实际上,即使是没有注册对应处理程序的情况也遵循此方案,使用没有注册的异常处理函数会导致Segment Not Present的异常。假如没有定义一个Segment Not Present的处理程序,因此发生Segment Not Present的异常。根据表,这样会导致双重异常

堆栈切换

在发生中断或者系统调用时会发生堆栈切换,在切换的过程中会涉及到特权级的改变,幸运的是在IA-32e架构下提供了自动切换栈帧的机制(中断堆栈表的堆栈切换机制 IST)

注:只64非兼容模式下,64位兼容模式依旧按照32位堆栈切换方式

在IA-32e模式中提供了7个栈指针,共IDT使用,IST位就是用于为中断/异常处理程序提供IST栈表索引,当确定目标以后会强制将SS段寄存器赋值为NULL选择子,并将中断栈地址加载到RSP寄存器中,最后将原SS,RSP,RFLAGS,CS和RIP寄存器压入新栈中

为双重处理添加一个堆栈

知道如何切换堆栈后我们为双重异常注册一个栈,tss使用的诀窍是需要向对应的ISP索引处添加一个堆栈方便进行上线文切换,记住,栈的生长方向是从高地址到低地址,因此我们需要把高地址注册到TSS中

// in kernel/descriptor/gdt.rs

// IST索引值为0,其他IST索引也可以
pub const DOUBLE_FAULT_LIST_INDEX: u16 = 0;

lazy_static! {
    // 现在的双重故障堆栈没有保护页面以防止堆栈溢出(还没有做内存管理),因此不能在双重故障处理程序中执行任何占用大量堆栈的操作
    static ref TSS: TaskStateSegment = {
        let mut tss = TaskStateSegment::new();
        let stack_len =
        {
            // 这里STACK_SIZE需要显示写出是usize的,因为我们的VirtAddr只完成了对usize的加法操作
            // 按照4kb对齐刚好是一个页面
            const STACK_SIZE:usize = 4096;
            // 指定使用的栈大小位4096
            static mut STACK:[u8;STACK_SIZE] = [0;STACK_SIZE];
            // 获取STACK的虚拟地址
            let stack_start = VirtAddr::from_pointer(unsafe{&STACK});
            // 栈的生长方向是高地址向低地址我们添加栈的高地址
            let stack_end = stack_start + STACK_SIZE;
            stack_end
        };
        tss.interrupt_stack_table[DOUBLE_FAULT_LIST_INDEX as usize] = stack_len;
        tss
    };
}

注册双重异常处理函数

然后我们可以注册我们的双重异常处理函数了

lazy_static! {
    static ref IDT: InterruptDescriptorTable = {
      	....
        idt.simd_floating_point.set_handler_fn(simd_floating_point);
        idt.overflow.set_handler_fn(overflow);
        idt.debug.set_handler_fn(debug);
        // 在IDT中为双重错误处理程序设置堆栈索引
        unsafe {
            idt.double_fault.set_handler_fn(double_fault).set_stack_index(gdt::DOUBLE_FAULT_LIST_INDEX);
        }
        idt
    };
}

我们将堆栈索引的地址放到IST位(具体请查看使用Rust开发操作系统(中断描述符表–IDT)拆分中断门/陷进门的结构一节)

这样我们的双重处理函数就注册完毕

下一步要做什么?

在下一篇文章中我们将要学习如何编写中断处理函数,以及提供基本的时钟中断

发布了16 篇原创文章 · 获赞 0 · 访问量 564
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 数字20 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览