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