xref: /DragonOS/kernel/src/driver/firmware/efi/mod.rs (revision 2eab6dd743e94a86a685f1f3c01e599adf86610a)
1 use log::{error, warn};
2 use system_error::SystemError;
3 
4 use crate::{
5     libs::rwlock::{RwLock, RwLockReadGuard, RwLockWriteGuard},
6     mm::PhysAddr,
7 };
8 
9 use self::{guid::DragonStubPayloadEFI, memmap::EFIMemoryMapInfo};
10 
11 pub mod esrt;
12 mod fdt;
13 pub mod guid;
14 pub mod init;
15 pub mod memmap;
16 pub mod tables;
17 
18 static EFI_MANAGER: EFIManager = EFIManager::new();
19 
20 /// EFI管理器
21 ///
22 /// 数据成员可参考: https://code.dragonos.org.cn/xref/linux-6.1.9/include/linux/efi.h#620
23 #[derive(Debug)]
24 pub struct EFIManager {
25     inner: RwLock<InnerEFIManager>,
26 }
27 
28 #[inline(always)]
efi_manager() -> &'static EFIManager29 pub fn efi_manager() -> &'static EFIManager {
30     &EFI_MANAGER
31 }
32 
33 #[derive(Debug)]
34 struct InnerEFIManager {
35     pub mmap: EFIMemoryMapInfo,
36     /// EFI模块启动时状态的标识
37     pub init_flags: EFIInitFlags,
38     /// runtime services的物理地址
39     pub runtime_paddr: Option<PhysAddr>,
40     /// runtime services的版本号
41     pub runtime_service_version: Option<uefi_raw::table::Revision>,
42     pub dragonstub_load_info: Option<DragonStubPayloadEFI>,
43     /// uefi 内存属性表的物理地址
44     pub memory_attribute_table_paddr: Option<PhysAddr>,
45     /// uefi 内存保留表的物理地址
46     pub memreserve_table_paddr: Option<PhysAddr>,
47     /// uefi esrt表的物理地址
48     pub esrt_table_paddr: Option<PhysAddr>,
49 }
50 
51 impl EFIManager {
new() -> Self52     const fn new() -> Self {
53         EFIManager {
54             inner: RwLock::new(InnerEFIManager {
55                 mmap: EFIMemoryMapInfo::DEFAULT,
56                 init_flags: EFIInitFlags::empty(),
57                 runtime_paddr: None,
58                 runtime_service_version: None,
59                 dragonstub_load_info: None,
60                 memory_attribute_table_paddr: None,
61                 memreserve_table_paddr: None,
62                 esrt_table_paddr: None,
63             }),
64         }
65     }
66 
desc_version(&self) -> usize67     pub fn desc_version(&self) -> usize {
68         return self.inner.read().mmap.desc_version;
69     }
70 
71     /// 内核加载的地址、大小的信息
72     #[allow(dead_code)]
kernel_load_info(&self) -> Option<DragonStubPayloadEFI>73     pub fn kernel_load_info(&self) -> Option<DragonStubPayloadEFI> {
74         return self.inner.read().dragonstub_load_info;
75     }
76 
77     /// 检查是否为有效的system table表头
78     ///
79     /// ## 参数
80     ///
81     /// - header: system table表头
82     /// - min_major: 最小的major版本号。如果不满足,则会输出Warning,并返回Ok
83     ///
84     /// ## 返回
85     ///
86     /// - Ok(()): 检查通过
87     /// - Err(SystemError::EINVAL): header无效
check_system_table_header( &self, header: &uefi_raw::table::Header, min_major: u16, ) -> Result<(), SystemError>88     pub fn check_system_table_header(
89         &self,
90         header: &uefi_raw::table::Header,
91         min_major: u16,
92     ) -> Result<(), SystemError> {
93         if header.signature != uefi_raw::table::system::SystemTable::SIGNATURE {
94             error!("System table signature mismatch!");
95             return Err(SystemError::EINVAL);
96         }
97 
98         if header.revision.major() < min_major {
99             warn!(
100                 "System table version: {:?}, expected {}.00 or greater!",
101                 header.revision, min_major
102             );
103         }
104 
105         return Ok(());
106     }
107 
inner_read(&self) -> RwLockReadGuard<InnerEFIManager>108     fn inner_read(&self) -> RwLockReadGuard<InnerEFIManager> {
109         self.inner.read()
110     }
111 
inner_write(&self) -> RwLockWriteGuard<InnerEFIManager>112     fn inner_write(&self) -> RwLockWriteGuard<InnerEFIManager> {
113         self.inner.write()
114     }
115 
116     /// 是否存在ESRT表
esrt_table_exists(&self) -> bool117     fn esrt_table_exists(&self) -> bool {
118         self.inner_read().esrt_table_paddr.is_some()
119     }
120 }
121 
122 // 在Rust中,我们使用枚举和bitflags来表示这些宏
123 bitflags! {
124     pub struct EFIInitFlags: u32 {
125         /// 当前使用EFI启动
126         const BOOT = 1 << 0;
127         /// 是否可以使用EFI配置表
128         const CONFIG_TABLES = 1 << 1;
129         /// 是否可以使用运行时服务
130         const RUNTIME_SERVICES = 1 << 2;
131         /// 是否可以使用EFI内存映射
132         const MEMMAP = 1 << 3;
133         /// 固件是否为64位
134         const EFI_64BIT = 1 << 4;
135         /// 访问是否通过虚拟化接口
136         const PARAVIRT = 1 << 5;
137         /// 第一架构特定位
138         const ARCH_1 = 1 << 6;
139         /// 打印附加运行时调试信息
140         const DBG = 1 << 7;
141         /// 是否可以在运行时数据区域映射非可执行
142         const NX_PE_DATA = 1 << 8;
143         /// 固件是否发布了一个EFI_MEMORY_ATTRIBUTES表
144         const MEM_ATTR = 1 << 9;
145         /// 内核是否配置为忽略软保留
146         const MEM_NO_SOFT_RESERVE = 1 << 10;
147         /// 是否可以使用EFI引导服务内存段
148         const PRESERVE_BS_REGIONS = 1 << 11;
149 
150     }
151 }
152