x86_64: 使用Rust重写内核初始化代码 (#507)

* x86_64: 使用Rust重写内核初始化代码

kernel/src/arch/x86_64/init/mod.rs

@@ -1,6 +1,123 @@
+use core::sync::atomic::{compiler_fence, Ordering};
+
+use system_error::SystemError;
+use x86::dtables::DescriptorTablePointer;
+
+use crate::{
+    arch::process::table::TSSManager,
+    driver::pci::pci::pci_init,
+    include::bindings::bindings::{cpu_init, irq_init},
+    init::init::start_kernel,
+    kdebug,
+    mm::{MemoryManagementArch, PhysAddr},
+};
+
+use super::{
+    driver::{
+        hpet::{hpet_init, hpet_instance},
+        tsc::TSCManager,
+    },
+    MMArch,
+};
+
 #[derive(Debug)]
 pub struct ArchBootParams {}
 
 impl ArchBootParams {
     pub const DEFAULT: Self = ArchBootParams {};
 }
+
+extern "C" {
+    static mut GDT_Table: [usize; 0usize];
+    static mut IDT_Table: [usize; 0usize];
+    fn head_stack_start();
+
+    fn multiboot2_init(mb2_info: u64, mb2_magic: u32) -> bool;
+    fn __init_set_cpu_stack_start(cpu: u32, stack_start: u64);
+    fn sys_vector_init();
+}
+
+#[no_mangle]
+unsafe extern "C" fn kernel_main(
+    mb2_info: u64,
+    mb2_magic: u64,
+    bsp_gdt_size: u64,
+    bsp_idt_size: u64,
+) -> ! {
+    let mut gdtp = DescriptorTablePointer::<usize>::default();
+    let gdt_vaddr =
+        MMArch::phys_2_virt(PhysAddr::new(&GDT_Table as *const usize as usize)).unwrap();
+    let idt_vaddr =
+        MMArch::phys_2_virt(PhysAddr::new(&IDT_Table as *const usize as usize)).unwrap();
+    gdtp.base = gdt_vaddr.data() as *const usize;
+    gdtp.limit = bsp_gdt_size as u16 - 1;
+
+    let mut idtp = DescriptorTablePointer::<usize>::default();
+    idtp.base = idt_vaddr.data() as *const usize;
+    idtp.limit = bsp_idt_size as u16 - 1;
+
+    x86::dtables::lgdt(&gdtp);
+    x86::dtables::lidt(&idtp);
+
+    compiler_fence(Ordering::SeqCst);
+    multiboot2_init(mb2_info, (mb2_magic & 0xFFFF_FFFF) as u32);
+    compiler_fence(Ordering::SeqCst);
+
+    start_kernel();
+}
+
+/// 在内存管理初始化之前的架构相关的早期初始化
+#[inline(never)]
+pub fn early_setup_arch() -> Result<(), SystemError> {
+    let stack_start = unsafe { *(head_stack_start as *const u64) } as usize;
+    kdebug!("head_stack_start={:#x}\n", stack_start);
+    unsafe {
+        let gdt_vaddr =
+            MMArch::phys_2_virt(PhysAddr::new(&GDT_Table as *const usize as usize)).unwrap();
+        let idt_vaddr =
+            MMArch::phys_2_virt(PhysAddr::new(&IDT_Table as *const usize as usize)).unwrap();
+
+        kdebug!("GDT_Table={:?}, IDT_Table={:?}\n", gdt_vaddr, idt_vaddr);
+    }
+
+    set_current_core_tss(stack_start, 0);
+    unsafe { TSSManager::load_tr() };
+    unsafe { __init_set_cpu_stack_start(0, stack_start as u64) };
+    unsafe { sys_vector_init() };
+
+    return Ok(());
+}
+
+/// 架构相关的初始化
+#[inline(never)]
+pub fn setup_arch() -> Result<(), SystemError> {
+    unsafe {
+        irq_init();
+        cpu_init();
+    }
+
+    // todo: 将来pci接入设备驱动模型之后，删掉这里。
+    pci_init();
+    return Ok(());
+}
+
+/// 架构相关的初始化（在IDLE的最后一个阶段）
+#[inline(never)]
+pub fn setup_arch_post() -> Result<(), SystemError> {
+    hpet_init().expect("hpet init failed");
+    hpet_instance().hpet_enable().expect("hpet enable failed");
+    TSCManager::init().expect("tsc init failed");
+
+    return Ok(());
+}
+
+fn set_current_core_tss(stack_start: usize, ist0: usize) {
+    let current_tss = unsafe { TSSManager::current_tss() };
+    kdebug!(
+        "set_current_core_tss: stack_start={:#x}, ist0={:#x}\n",
+        stack_start,
+        ist0
+    );
+    current_tss.set_rsp(x86::Ring::Ring0, stack_start as u64);
+    current_tss.set_ist(0, ist0 as u64);
+}