1 pub mod barrier; 2 pub mod bump; 3 mod c_adapter; 4 5 use alloc::vec::Vec; 6 use hashbrown::HashSet; 7 use x86::time::rdtsc; 8 use x86_64::registers::model_specific::EferFlags; 9 10 use crate::driver::tty::serial::serial8250::send_to_default_serial8250_port; 11 use crate::include::bindings::bindings::{ 12 multiboot2_get_load_base, multiboot2_get_memory, multiboot2_iter, multiboot_mmap_entry_t, 13 multiboot_tag_load_base_addr_t, 14 }; 15 use crate::libs::align::page_align_up; 16 use crate::libs::lib_ui::screen_manager::scm_disable_put_to_window; 17 use crate::libs::printk::PrintkWriter; 18 use crate::libs::spinlock::SpinLock; 19 20 use crate::mm::allocator::page_frame::{FrameAllocator, PageFrameCount, PageFrameUsage}; 21 use crate::mm::memblock::mem_block_manager; 22 use crate::mm::mmio_buddy::mmio_init; 23 use crate::{ 24 arch::MMArch, 25 mm::allocator::{buddy::BuddyAllocator, bump::BumpAllocator}, 26 }; 27 28 use crate::mm::kernel_mapper::KernelMapper; 29 use crate::mm::page::{PageEntry, PageFlags}; 30 use crate::mm::{MemoryManagementArch, PageTableKind, PhysAddr, VirtAddr}; 31 use crate::{kdebug, kinfo, kwarn}; 32 use system_error::SystemError; 33 34 use core::arch::asm; 35 use core::ffi::c_void; 36 use core::fmt::{Debug, Write}; 37 use core::mem::{self}; 38 39 use core::sync::atomic::{compiler_fence, AtomicBool, Ordering}; 40 41 use super::kvm::vmx::vmcs::VmcsFields; 42 use super::kvm::vmx::vmx_asm_wrapper::vmx_vmread; 43 44 pub type PageMapper = 45 crate::mm::page::PageMapper<crate::arch::x86_64::mm::X86_64MMArch, LockedFrameAllocator>; 46 47 /// 初始的CR3寄存器的值,用于内存管理初始化时,创建的第一个内核页表的位置 48 static mut INITIAL_CR3_VALUE: PhysAddr = PhysAddr::new(0); 49 50 /// 内核的第一个页表在pml4中的索引 51 /// 顶级页表的[256, 512)项是内核的页表 52 static KERNEL_PML4E_NO: usize = (X86_64MMArch::PHYS_OFFSET & ((1 << 48) - 1)) >> 39; 53 54 static INNER_ALLOCATOR: SpinLock<Option<BuddyAllocator<MMArch>>> = SpinLock::new(None); 55 56 #[derive(Clone, Copy, Debug)] 57 pub struct X86_64MMBootstrapInfo { 58 kernel_load_base_paddr: usize, 59 kernel_code_start: usize, 60 kernel_code_end: usize, 61 kernel_data_end: usize, 62 kernel_rodata_end: usize, 63 start_brk: usize, 64 } 65 66 pub(super) static mut BOOTSTRAP_MM_INFO: Option<X86_64MMBootstrapInfo> = None; 67 68 /// @brief X86_64的内存管理架构结构体 69 #[derive(Debug, Clone, Copy, Hash)] 70 pub struct X86_64MMArch; 71 72 /// XD标志位是否被保留 73 static XD_RESERVED: AtomicBool = AtomicBool::new(false); 74 75 impl MemoryManagementArch for X86_64MMArch { 76 /// 4K页 77 const PAGE_SHIFT: usize = 12; 78 79 /// 每个页表项占8字节,总共有512个页表项 80 const PAGE_ENTRY_SHIFT: usize = 9; 81 82 /// 四级页表(PML4T、PDPT、PDT、PT) 83 const PAGE_LEVELS: usize = 4; 84 85 /// 页表项的有效位的index。在x86_64中,页表项的第[0, 47]位表示地址和flag, 86 /// 第[48, 51]位表示保留。因此,有效位的index为52。 87 /// 请注意,第63位是XD位,表示是否允许执行。 88 const ENTRY_ADDRESS_SHIFT: usize = 52; 89 90 const ENTRY_FLAG_DEFAULT_PAGE: usize = Self::ENTRY_FLAG_PRESENT; 91 92 const ENTRY_FLAG_DEFAULT_TABLE: usize = Self::ENTRY_FLAG_PRESENT; 93 94 const ENTRY_FLAG_PRESENT: usize = 1 << 0; 95 96 const ENTRY_FLAG_READONLY: usize = 0; 97 98 const ENTRY_FLAG_READWRITE: usize = 1 << 1; 99 100 const ENTRY_FLAG_USER: usize = 1 << 2; 101 102 const ENTRY_FLAG_WRITE_THROUGH: usize = 1 << 3; 103 104 const ENTRY_FLAG_CACHE_DISABLE: usize = 1 << 4; 105 106 const ENTRY_FLAG_NO_EXEC: usize = 1 << 63; 107 /// x86_64不存在EXEC标志位,只有NO_EXEC(XD)标志位 108 const ENTRY_FLAG_EXEC: usize = 0; 109 110 /// 物理地址与虚拟地址的偏移量 111 /// 0xffff_8000_0000_0000 112 const PHYS_OFFSET: usize = Self::PAGE_NEGATIVE_MASK + (Self::PAGE_ADDRESS_SIZE >> 1); 113 114 const USER_END_VADDR: VirtAddr = VirtAddr::new(0x0000_7eff_ffff_ffff); 115 const USER_BRK_START: VirtAddr = VirtAddr::new(0x700000000000); 116 const USER_STACK_START: VirtAddr = VirtAddr::new(0x6ffff0a00000); 117 118 const FIXMAP_START_VADDR: VirtAddr = VirtAddr::new(0xffffb00000000000); 119 /// 设置FIXMAP区域大小为1M 120 const FIXMAP_SIZE: usize = 256 * 4096; 121 122 /// @brief 获取物理内存区域 123 unsafe fn init() { 124 extern "C" { 125 fn _text(); 126 fn _etext(); 127 fn _edata(); 128 fn _erodata(); 129 fn _end(); 130 } 131 132 Self::init_xd_rsvd(); 133 let load_base_paddr = Self::get_load_base_paddr(); 134 135 let bootstrap_info = X86_64MMBootstrapInfo { 136 kernel_load_base_paddr: load_base_paddr.data(), 137 kernel_code_start: _text as usize, 138 kernel_code_end: _etext as usize, 139 kernel_data_end: _edata as usize, 140 kernel_rodata_end: _erodata as usize, 141 start_brk: _end as usize, 142 }; 143 144 unsafe { 145 BOOTSTRAP_MM_INFO = Some(bootstrap_info); 146 } 147 148 // 初始化物理内存区域(从multiboot2中获取) 149 Self::init_memory_area_from_multiboot2().expect("init memory area failed"); 150 151 send_to_default_serial8250_port("x86 64 init end\n\0".as_bytes()); 152 } 153 154 /// @brief 刷新TLB中,关于指定虚拟地址的条目 155 unsafe fn invalidate_page(address: VirtAddr) { 156 compiler_fence(Ordering::SeqCst); 157 asm!("invlpg [{0}]", in(reg) address.data(), options(nostack, preserves_flags)); 158 compiler_fence(Ordering::SeqCst); 159 } 160 161 /// @brief 刷新TLB中,所有的条目 162 unsafe fn invalidate_all() { 163 compiler_fence(Ordering::SeqCst); 164 // 通过设置cr3寄存器,来刷新整个TLB 165 Self::set_table(PageTableKind::User, Self::table(PageTableKind::User)); 166 compiler_fence(Ordering::SeqCst); 167 } 168 169 /// @brief 获取顶级页表的物理地址 170 unsafe fn table(table_kind: PageTableKind) -> PhysAddr { 171 match table_kind { 172 PageTableKind::Kernel | PageTableKind::User => { 173 let paddr: usize; 174 compiler_fence(Ordering::SeqCst); 175 asm!("mov {}, cr3", out(reg) paddr, options(nomem, nostack, preserves_flags)); 176 compiler_fence(Ordering::SeqCst); 177 return PhysAddr::new(paddr); 178 } 179 PageTableKind::EPT => { 180 let eptp = 181 vmx_vmread(VmcsFields::CTRL_EPTP_PTR as u32).expect("Failed to read eptp"); 182 return PhysAddr::new(eptp as usize); 183 } 184 } 185 } 186 187 /// @brief 设置顶级页表的物理地址到处理器中 188 unsafe fn set_table(_table_kind: PageTableKind, table: PhysAddr) { 189 compiler_fence(Ordering::SeqCst); 190 asm!("mov cr3, {}", in(reg) table.data(), options(nostack, preserves_flags)); 191 compiler_fence(Ordering::SeqCst); 192 } 193 194 /// @brief 判断虚拟地址是否合法 195 fn virt_is_valid(virt: VirtAddr) -> bool { 196 return virt.is_canonical(); 197 } 198 199 /// 获取内存管理初始化时,创建的第一个内核页表的地址 200 fn initial_page_table() -> PhysAddr { 201 unsafe { 202 return INITIAL_CR3_VALUE; 203 } 204 } 205 206 /// @brief 创建新的顶层页表 207 /// 208 /// 该函数会创建页表并复制内核的映射到新的页表中 209 /// 210 /// @return 新的页表 211 fn setup_new_usermapper() -> Result<crate::mm::ucontext::UserMapper, SystemError> { 212 let new_umapper: crate::mm::page::PageMapper<X86_64MMArch, LockedFrameAllocator> = unsafe { 213 PageMapper::create(PageTableKind::User, LockedFrameAllocator) 214 .ok_or(SystemError::ENOMEM)? 215 }; 216 217 let current_ktable: KernelMapper = KernelMapper::lock(); 218 let copy_mapping = |pml4_entry_no| unsafe { 219 let entry: PageEntry<X86_64MMArch> = current_ktable 220 .table() 221 .entry(pml4_entry_no) 222 .unwrap_or_else(|| panic!("entry {} not found", pml4_entry_no)); 223 new_umapper.table().set_entry(pml4_entry_no, entry) 224 }; 225 226 // 复制内核的映射 227 for pml4_entry_no in KERNEL_PML4E_NO..512 { 228 copy_mapping(pml4_entry_no); 229 } 230 231 return Ok(crate::mm::ucontext::UserMapper::new(new_umapper)); 232 } 233 234 const PAGE_SIZE: usize = 1 << Self::PAGE_SHIFT; 235 236 const PAGE_OFFSET_MASK: usize = Self::PAGE_SIZE - 1; 237 238 const PAGE_MASK: usize = !(Self::PAGE_OFFSET_MASK); 239 240 const PAGE_ADDRESS_SHIFT: usize = Self::PAGE_LEVELS * Self::PAGE_ENTRY_SHIFT + Self::PAGE_SHIFT; 241 242 const PAGE_ADDRESS_SIZE: usize = 1 << Self::PAGE_ADDRESS_SHIFT; 243 244 const PAGE_ADDRESS_MASK: usize = Self::PAGE_ADDRESS_SIZE - Self::PAGE_SIZE; 245 246 const PAGE_ENTRY_SIZE: usize = 1 << (Self::PAGE_SHIFT - Self::PAGE_ENTRY_SHIFT); 247 248 const PAGE_ENTRY_NUM: usize = 1 << Self::PAGE_ENTRY_SHIFT; 249 250 const PAGE_ENTRY_MASK: usize = Self::PAGE_ENTRY_NUM - 1; 251 252 const PAGE_NEGATIVE_MASK: usize = !((Self::PAGE_ADDRESS_SIZE) - 1); 253 254 const ENTRY_ADDRESS_SIZE: usize = 1 << Self::ENTRY_ADDRESS_SHIFT; 255 256 const ENTRY_ADDRESS_MASK: usize = Self::ENTRY_ADDRESS_SIZE - Self::PAGE_SIZE; 257 258 const ENTRY_FLAGS_MASK: usize = !Self::ENTRY_ADDRESS_MASK; 259 260 unsafe fn read<T>(address: VirtAddr) -> T { 261 return core::ptr::read(address.data() as *const T); 262 } 263 264 unsafe fn write<T>(address: VirtAddr, value: T) { 265 core::ptr::write(address.data() as *mut T, value); 266 } 267 268 unsafe fn write_bytes(address: VirtAddr, value: u8, count: usize) { 269 core::ptr::write_bytes(address.data() as *mut u8, value, count); 270 } 271 272 unsafe fn phys_2_virt(phys: PhysAddr) -> Option<VirtAddr> { 273 if let Some(vaddr) = phys.data().checked_add(Self::PHYS_OFFSET) { 274 return Some(VirtAddr::new(vaddr)); 275 } else { 276 return None; 277 } 278 } 279 280 unsafe fn virt_2_phys(virt: VirtAddr) -> Option<PhysAddr> { 281 if let Some(paddr) = virt.data().checked_sub(Self::PHYS_OFFSET) { 282 return Some(PhysAddr::new(paddr)); 283 } else { 284 return None; 285 } 286 } 287 288 #[inline(always)] 289 fn make_entry(paddr: PhysAddr, page_flags: usize) -> usize { 290 return paddr.data() | page_flags; 291 } 292 } 293 294 impl X86_64MMArch { 295 unsafe fn get_load_base_paddr() -> PhysAddr { 296 let mut mb2_lb_info: [multiboot_tag_load_base_addr_t; 512] = mem::zeroed(); 297 send_to_default_serial8250_port("get_load_base_paddr begin\n\0".as_bytes()); 298 299 let mut mb2_count: u32 = 0; 300 multiboot2_iter( 301 Some(multiboot2_get_load_base), 302 &mut mb2_lb_info as *mut [multiboot_tag_load_base_addr_t; 512] as usize as *mut c_void, 303 &mut mb2_count, 304 ); 305 306 if mb2_count == 0 { 307 send_to_default_serial8250_port( 308 "get_load_base_paddr mb2_count == 0, default to 1MB\n\0".as_bytes(), 309 ); 310 return PhysAddr::new(0x100000); 311 } 312 313 let phys = mb2_lb_info[0].load_base_addr as usize; 314 315 return PhysAddr::new(phys); 316 } 317 unsafe fn init_memory_area_from_multiboot2() -> Result<usize, SystemError> { 318 // 这个数组用来存放内存区域的信息(从C获取) 319 let mut mb2_mem_info: [multiboot_mmap_entry_t; 512] = mem::zeroed(); 320 send_to_default_serial8250_port("init_memory_area_from_multiboot2 begin\n\0".as_bytes()); 321 322 let mut mb2_count: u32 = 0; 323 multiboot2_iter( 324 Some(multiboot2_get_memory), 325 &mut mb2_mem_info as *mut [multiboot_mmap_entry_t; 512] as usize as *mut c_void, 326 &mut mb2_count, 327 ); 328 send_to_default_serial8250_port("init_memory_area_from_multiboot2 2\n\0".as_bytes()); 329 330 let mb2_count = mb2_count as usize; 331 let mut areas_count = 0usize; 332 let mut total_mem_size = 0usize; 333 for i in 0..mb2_count { 334 // Only use the memory area if its type is 1 (RAM) 335 if mb2_mem_info[i].type_ == 1 { 336 // Skip the memory area if its len is 0 337 if mb2_mem_info[i].len == 0 { 338 continue; 339 } 340 341 total_mem_size += mb2_mem_info[i].len as usize; 342 343 mem_block_manager() 344 .add_block( 345 PhysAddr::new(mb2_mem_info[i].addr as usize), 346 mb2_mem_info[i].len as usize, 347 ) 348 .unwrap_or_else(|e| { 349 kwarn!( 350 "Failed to add memory block: base={:#x}, size={:#x}, error={:?}", 351 mb2_mem_info[i].addr, 352 mb2_mem_info[i].len, 353 e 354 ); 355 }); 356 areas_count += 1; 357 } 358 } 359 send_to_default_serial8250_port("init_memory_area_from_multiboot2 end\n\0".as_bytes()); 360 kinfo!("Total memory size: {} MB, total areas from multiboot2: {mb2_count}, valid areas: {areas_count}", total_mem_size / 1024 / 1024); 361 return Ok(areas_count); 362 } 363 364 fn init_xd_rsvd() { 365 // 读取ia32-EFER寄存器的值 366 let efer: EferFlags = x86_64::registers::model_specific::Efer::read(); 367 if !efer.contains(EferFlags::NO_EXECUTE_ENABLE) { 368 // NO_EXECUTE_ENABLE是false,那么就设置xd_reserved为true 369 kdebug!("NO_EXECUTE_ENABLE is false, set XD_RESERVED to true"); 370 XD_RESERVED.store(true, Ordering::Relaxed); 371 } 372 compiler_fence(Ordering::SeqCst); 373 } 374 375 /// 判断XD标志位是否被保留 376 pub fn is_xd_reserved() -> bool { 377 // return XD_RESERVED.load(Ordering::Relaxed); 378 379 // 由于暂时不支持execute disable,因此直接返回true 380 // 不支持的原因是,目前好像没有能正确的设置page-level的xd位,会触发page fault 381 return true; 382 } 383 } 384 385 impl VirtAddr { 386 /// @brief 判断虚拟地址是否合法 387 #[inline(always)] 388 pub fn is_canonical(self) -> bool { 389 let x = self.data() & X86_64MMArch::PHYS_OFFSET; 390 // 如果x为0,说明虚拟地址的高位为0,是合法的用户地址 391 // 如果x为PHYS_OFFSET,说明虚拟地址的高位全为1,是合法的内核地址 392 return x == 0 || x == X86_64MMArch::PHYS_OFFSET; 393 } 394 } 395 396 /// @brief 初始化内存管理模块 397 pub fn mm_init() { 398 send_to_default_serial8250_port("mm_init\n\0".as_bytes()); 399 PrintkWriter 400 .write_fmt(format_args!("mm_init() called\n")) 401 .unwrap(); 402 // printk_color!(GREEN, BLACK, "mm_init() called\n"); 403 static _CALL_ONCE: AtomicBool = AtomicBool::new(false); 404 if _CALL_ONCE 405 .compare_exchange(false, true, Ordering::SeqCst, Ordering::SeqCst) 406 .is_err() 407 { 408 send_to_default_serial8250_port("mm_init err\n\0".as_bytes()); 409 panic!("mm_init() can only be called once"); 410 } 411 412 unsafe { X86_64MMArch::init() }; 413 kdebug!("bootstrap info: {:?}", unsafe { BOOTSTRAP_MM_INFO }); 414 kdebug!("phys[0]=virt[0x{:x}]", unsafe { 415 MMArch::phys_2_virt(PhysAddr::new(0)).unwrap().data() 416 }); 417 418 // 初始化内存管理器 419 unsafe { allocator_init() }; 420 // enable mmio 421 mmio_init(); 422 } 423 424 unsafe fn allocator_init() { 425 let virt_offset = BOOTSTRAP_MM_INFO.unwrap().start_brk; 426 let phy_offset = 427 unsafe { MMArch::virt_2_phys(VirtAddr::new(page_align_up(virt_offset))) }.unwrap(); 428 429 let mut bump_allocator = BumpAllocator::<X86_64MMArch>::new(phy_offset.data()); 430 kdebug!( 431 "BumpAllocator created, offset={:?}", 432 bump_allocator.offset() 433 ); 434 435 // 暂存初始在head.S中指定的页表的地址,后面再考虑是否需要把它加到buddy的可用空间里面! 436 // 现在不加的原因是,我担心会有安全漏洞问题:这些初始的页表,位于内核的数据段。如果归还到buddy, 437 // 可能会产生一定的安全风险(有的代码可能根据虚拟地址来进行安全校验) 438 let _old_page_table = MMArch::table(PageTableKind::Kernel); 439 440 let new_page_table: PhysAddr; 441 // 使用bump分配器,把所有的内存页都映射到页表 442 { 443 // 用bump allocator创建新的页表 444 let mut mapper: crate::mm::page::PageMapper<MMArch, &mut BumpAllocator<MMArch>> = 445 crate::mm::page::PageMapper::<MMArch, _>::create( 446 PageTableKind::Kernel, 447 &mut bump_allocator, 448 ) 449 .expect("Failed to create page mapper"); 450 new_page_table = mapper.table().phys(); 451 kdebug!("PageMapper created"); 452 453 // 取消最开始时候,在head.S中指定的映射(暂时不刷新TLB) 454 { 455 let table = mapper.table(); 456 let empty_entry = PageEntry::<MMArch>::from_usize(0); 457 for i in 0..MMArch::PAGE_ENTRY_NUM { 458 table 459 .set_entry(i, empty_entry) 460 .expect("Failed to empty page table entry"); 461 } 462 } 463 kdebug!("Successfully emptied page table"); 464 465 let total_num = mem_block_manager().total_initial_memory_regions(); 466 for i in 0..total_num { 467 let area = mem_block_manager().get_initial_memory_region(i).unwrap(); 468 // kdebug!("area: base={:?}, size={:#x}, end={:?}", area.base, area.size, area.base + area.size); 469 for i in 0..((area.size + MMArch::PAGE_SIZE - 1) / MMArch::PAGE_SIZE) { 470 let paddr = area.base.add(i * MMArch::PAGE_SIZE); 471 let vaddr = unsafe { MMArch::phys_2_virt(paddr) }.unwrap(); 472 let flags = kernel_page_flags::<MMArch>(vaddr); 473 474 let flusher = mapper 475 .map_phys(vaddr, paddr, flags) 476 .expect("Failed to map frame"); 477 // 暂时不刷新TLB 478 flusher.ignore(); 479 } 480 } 481 482 // 添加低地址的映射(在smp完成初始化之前,需要使用低地址的映射.初始化之后需要取消这一段映射) 483 LowAddressRemapping::remap_at_low_address(&mut mapper); 484 } 485 486 unsafe { 487 INITIAL_CR3_VALUE = new_page_table; 488 } 489 kdebug!( 490 "After mapping all physical memory, DragonOS used: {} KB", 491 bump_allocator.offset() / 1024 492 ); 493 494 // 初始化buddy_allocator 495 let buddy_allocator = unsafe { BuddyAllocator::<X86_64MMArch>::new(bump_allocator).unwrap() }; 496 // 设置全局的页帧分配器 497 unsafe { set_inner_allocator(buddy_allocator) }; 498 kinfo!("Successfully initialized buddy allocator"); 499 // 关闭显示输出 500 scm_disable_put_to_window(); 501 502 // make the new page table current 503 { 504 let mut binding = INNER_ALLOCATOR.lock(); 505 let mut allocator_guard = binding.as_mut().unwrap(); 506 kdebug!("To enable new page table."); 507 compiler_fence(Ordering::SeqCst); 508 let mapper = crate::mm::page::PageMapper::<MMArch, _>::new( 509 PageTableKind::Kernel, 510 new_page_table, 511 &mut allocator_guard, 512 ); 513 compiler_fence(Ordering::SeqCst); 514 mapper.make_current(); 515 compiler_fence(Ordering::SeqCst); 516 kdebug!("New page table enabled"); 517 } 518 kdebug!("Successfully enabled new page table"); 519 } 520 521 #[no_mangle] 522 pub extern "C" fn rs_test_buddy() { 523 test_buddy(); 524 } 525 pub fn test_buddy() { 526 // 申请内存然后写入数据然后free掉 527 // 总共申请200MB内存 528 const TOTAL_SIZE: usize = 200 * 1024 * 1024; 529 530 for i in 0..10 { 531 kdebug!("Test buddy, round: {i}"); 532 // 存放申请的内存块 533 let mut v: Vec<(PhysAddr, PageFrameCount)> = Vec::with_capacity(60 * 1024); 534 // 存放已经申请的内存块的地址(用于检查重复) 535 let mut addr_set: HashSet<PhysAddr> = HashSet::new(); 536 537 let mut allocated = 0usize; 538 539 let mut free_count = 0usize; 540 541 while allocated < TOTAL_SIZE { 542 let mut random_size = 0u64; 543 unsafe { x86::random::rdrand64(&mut random_size) }; 544 // 一次最多申请4M 545 random_size = random_size % (1024 * 4096); 546 if random_size == 0 { 547 continue; 548 } 549 let random_size = 550 core::cmp::min(page_align_up(random_size as usize), TOTAL_SIZE - allocated); 551 let random_size = PageFrameCount::from_bytes(random_size.next_power_of_two()).unwrap(); 552 // 获取帧 553 let (paddr, allocated_frame_count) = 554 unsafe { LockedFrameAllocator.allocate(random_size).unwrap() }; 555 assert!(allocated_frame_count.data().is_power_of_two()); 556 assert!(paddr.data() % MMArch::PAGE_SIZE == 0); 557 unsafe { 558 assert!(MMArch::phys_2_virt(paddr) 559 .as_ref() 560 .unwrap() 561 .check_aligned(allocated_frame_count.data() * MMArch::PAGE_SIZE)); 562 } 563 allocated += allocated_frame_count.data() * MMArch::PAGE_SIZE; 564 v.push((paddr, allocated_frame_count)); 565 assert!(addr_set.insert(paddr), "duplicate address: {:?}", paddr); 566 567 // 写入数据 568 let vaddr = unsafe { MMArch::phys_2_virt(paddr).unwrap() }; 569 let slice = unsafe { 570 core::slice::from_raw_parts_mut( 571 vaddr.data() as *mut u8, 572 allocated_frame_count.data() * MMArch::PAGE_SIZE, 573 ) 574 }; 575 for i in 0..slice.len() { 576 slice[i] = ((i + unsafe { rdtsc() } as usize) % 256) as u8; 577 } 578 579 // 随机释放一个内存块 580 if v.len() > 0 { 581 let mut random_index = 0u64; 582 unsafe { x86::random::rdrand64(&mut random_index) }; 583 // 70%概率释放 584 if random_index % 10 > 7 { 585 continue; 586 } 587 random_index = random_index % v.len() as u64; 588 let random_index = random_index as usize; 589 let (paddr, allocated_frame_count) = v.remove(random_index); 590 assert!(addr_set.remove(&paddr)); 591 unsafe { LockedFrameAllocator.free(paddr, allocated_frame_count) }; 592 free_count += allocated_frame_count.data() * MMArch::PAGE_SIZE; 593 } 594 } 595 596 kdebug!( 597 "Allocated {} MB memory, release: {} MB, no release: {} bytes", 598 allocated / 1024 / 1024, 599 free_count / 1024 / 1024, 600 (allocated - free_count) 601 ); 602 603 kdebug!("Now, to release buddy memory"); 604 // 释放所有的内存 605 for (paddr, allocated_frame_count) in v { 606 unsafe { LockedFrameAllocator.free(paddr, allocated_frame_count) }; 607 assert!(addr_set.remove(&paddr)); 608 free_count += allocated_frame_count.data() * MMArch::PAGE_SIZE; 609 } 610 611 kdebug!("release done!, allocated: {allocated}, free_count: {free_count}"); 612 } 613 } 614 615 /// 全局的页帧分配器 616 #[derive(Debug, Clone, Copy, Hash)] 617 pub struct LockedFrameAllocator; 618 619 impl FrameAllocator for LockedFrameAllocator { 620 unsafe fn allocate(&mut self, count: PageFrameCount) -> Option<(PhysAddr, PageFrameCount)> { 621 if let Some(ref mut allocator) = *INNER_ALLOCATOR.lock_irqsave() { 622 return allocator.allocate(count); 623 } else { 624 return None; 625 } 626 } 627 628 unsafe fn free(&mut self, address: crate::mm::PhysAddr, count: PageFrameCount) { 629 assert!(count.data().is_power_of_two()); 630 if let Some(ref mut allocator) = *INNER_ALLOCATOR.lock_irqsave() { 631 return allocator.free(address, count); 632 } 633 } 634 635 unsafe fn usage(&self) -> PageFrameUsage { 636 if let Some(ref mut allocator) = *INNER_ALLOCATOR.lock_irqsave() { 637 return allocator.usage(); 638 } else { 639 panic!("usage error"); 640 } 641 } 642 } 643 644 /// 获取内核地址默认的页面标志 645 pub unsafe fn kernel_page_flags<A: MemoryManagementArch>(virt: VirtAddr) -> PageFlags<A> { 646 let info: X86_64MMBootstrapInfo = BOOTSTRAP_MM_INFO.clone().unwrap(); 647 648 if virt.data() >= info.kernel_code_start && virt.data() < info.kernel_code_end { 649 // Remap kernel code execute 650 return PageFlags::new().set_execute(true).set_write(true); 651 } else if virt.data() >= info.kernel_data_end && virt.data() < info.kernel_rodata_end { 652 // Remap kernel rodata read only 653 return PageFlags::new().set_execute(true); 654 } else { 655 return PageFlags::new().set_write(true).set_execute(true); 656 } 657 } 658 659 unsafe fn set_inner_allocator(allocator: BuddyAllocator<MMArch>) { 660 static FLAG: AtomicBool = AtomicBool::new(false); 661 if FLAG 662 .compare_exchange(false, true, Ordering::SeqCst, Ordering::SeqCst) 663 .is_err() 664 { 665 panic!("Cannot set inner allocator twice!"); 666 } 667 *INNER_ALLOCATOR.lock() = Some(allocator); 668 } 669 670 /// 低地址重映射的管理器 671 /// 672 /// 低地址重映射的管理器,在smp初始化完成之前,需要使用低地址的映射,因此需要在smp初始化完成之后,取消这一段映射 673 pub struct LowAddressRemapping; 674 675 impl LowAddressRemapping { 676 // 映射32M 677 const REMAP_SIZE: usize = 32 * 1024 * 1024; 678 679 pub unsafe fn remap_at_low_address( 680 mapper: &mut crate::mm::page::PageMapper<MMArch, &mut BumpAllocator<MMArch>>, 681 ) { 682 for i in 0..(Self::REMAP_SIZE / MMArch::PAGE_SIZE) { 683 let paddr = PhysAddr::new(i * MMArch::PAGE_SIZE); 684 let vaddr = VirtAddr::new(i * MMArch::PAGE_SIZE); 685 let flags = kernel_page_flags::<MMArch>(vaddr); 686 687 let flusher = mapper 688 .map_phys(vaddr, paddr, flags) 689 .expect("Failed to map frame"); 690 // 暂时不刷新TLB 691 flusher.ignore(); 692 } 693 } 694 695 /// 取消低地址的映射 696 pub unsafe fn unmap_at_low_address(flush: bool) { 697 let mut mapper = KernelMapper::lock(); 698 assert!(mapper.as_mut().is_some()); 699 for i in 0..(Self::REMAP_SIZE / MMArch::PAGE_SIZE) { 700 let vaddr = VirtAddr::new(i * MMArch::PAGE_SIZE); 701 let (_, _, flusher) = mapper 702 .as_mut() 703 .unwrap() 704 .unmap_phys(vaddr, true) 705 .expect("Failed to unmap frame"); 706 if flush == false { 707 flusher.ignore(); 708 } 709 } 710 } 711 } 712 #[no_mangle] 713 pub extern "C" fn rs_mm_init() { 714 mm_init(); 715 } 716