sparreal_kernel/
kernel.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
use core::ptr::NonNull;

use driver_interface::Register;
use log::*;

use page_table_generic::{AccessSetting, CacheSetting};
use spin_on::spin_on;

use crate::{
    driver::{self, device_tree::set_dtb_addr},
    logger::KLogger,
    mem::{self, *},
    platform::{self, app_main},
    println,
    stdout::{self, EarlyDebugWrite},
};

/// 初始化日志和内存
///
/// # Safety
///
/// 1. BSS 应当清零。
/// 2. 若有MMU，应当已开启，且虚拟地址与代码段映射一致。
pub unsafe fn init_log_and_memory(kconfig: &KernelConfig) {
    set_dtb_addr(kconfig.dtb_addr);
    let _ = log::set_logger(&KLogger);
    log::set_max_level(LevelFilter::Trace);
    stdout::set_stdout(EarlyDebugWrite {});
    info!("Logger initialized.");

    mem::init(kconfig);

    let version = env!("CARGO_PKG_VERSION");
    println!("Welcome to sparreal\nVersion: {version}");
    platform::print_system_info();
}

/// 注册驱动
pub fn driver_register_append(registers: impl IntoIterator<Item = Register>) {
    driver::register_append(registers);
}

/// 运行内核主逻辑
///
/// # Safety
///
/// 需在 [init_log_and_memory] 之后执行，[run] 之前可用 [driver_register_append] 注册驱动。
pub unsafe fn run() -> ! {
    spin_on(async {
        driver::init().await;
    });
    platform::irqs_enable();
    app_main();
    println!("Waiting for interrupt...");
    loop {
        platform::wait_for_interrupt();
    }
}

#[repr(C)]
#[derive(Debug, Clone, Copy)]
pub struct MemoryRange {
    pub start: Phys<u8>,
    pub size: usize,
}

impl MemoryRange {
    pub const fn new() -> Self {
        Self {
            start: Phys::new(),
            size: 0,
        }
    }
}

#[repr(C)]
/// 内核配置
#[derive(Clone)]
pub struct KernelConfig {
    /// 启动配置
    pub boot_info: BootConfig,
    /// 栈顶
    pub stack_top: Phys<u8>,
    /// 设备树地址
    pub dtb_addr: Option<NonNull<u8>>,
}

#[derive(Clone)]
pub struct BootConfig {
    /// Kernel 所在的内存
    pub main_memory: MemoryRange,
    /// 已使用的内存
    pub main_memory_heap_offset: usize,
    /// 每核 Kernel sp 大小
    pub hart_stack_size: usize,
    /// 需要提前map的内存
    pub reserved_memory: [Option<MemoryReservedRange>; 24],
    /// 物理内存和虚拟内存的偏移
    pub va_offset: usize,
}

#[repr(C)]
#[derive(Clone, Copy)]
pub struct MemoryReservedRange {
    pub name: &'static str,
    pub start: Phys<u8>,
    pub size: usize,
    pub access: AccessSetting,
    pub cache: CacheSetting,
}