140fe15e0SLoGin use alloc::sync::Arc; 240fe15e0SLoGin 340fe15e0SLoGin use crate::{ 440fe15e0SLoGin arch::MMArch, 540fe15e0SLoGin include::bindings::bindings::{process_control_block, PAGE_OFFSET}, 640fe15e0SLoGin syscall::SystemError, 740fe15e0SLoGin }; 840fe15e0SLoGin 940fe15e0SLoGin use core::{ 1040fe15e0SLoGin cmp, 1140fe15e0SLoGin fmt::Debug, 1240fe15e0SLoGin intrinsics::unlikely, 1340fe15e0SLoGin ops::{Add, AddAssign, Sub, SubAssign}, 1440fe15e0SLoGin ptr, 1540fe15e0SLoGin sync::atomic::{AtomicBool, Ordering}, 1640fe15e0SLoGin }; 1740fe15e0SLoGin 1840fe15e0SLoGin use self::{ 1940fe15e0SLoGin allocator::page_frame::{VirtPageFrame, VirtPageFrameIter}, 2040fe15e0SLoGin page::round_up_to_page_size, 2140fe15e0SLoGin ucontext::{AddressSpace, UserMapper}, 2240fe15e0SLoGin }; 23004e86ffSlogin 2482d2e446Slogin pub mod allocator; 2540fe15e0SLoGin pub mod c_adapter; 266cb769c4Slogin pub mod gfp; 2740fe15e0SLoGin pub mod kernel_mapper; 28c2481452Shoumkh pub mod mmio_buddy; 2940fe15e0SLoGin pub mod no_init; 3040fe15e0SLoGin pub mod page; 31*c3dad001SLoGin pub mod percpu; 32ab5c8ca4Slogin pub mod syscall; 3340fe15e0SLoGin pub mod ucontext; 3440fe15e0SLoGin 3540fe15e0SLoGin /// 内核INIT进程的用户地址空间结构体(仅在process_init中初始化) 3640fe15e0SLoGin static mut __INITIAL_PROCESS_ADDRESS_SPACE: Option<Arc<AddressSpace>> = None; 3740fe15e0SLoGin 3840fe15e0SLoGin /// 获取内核INIT进程的用户地址空间结构体 3940fe15e0SLoGin #[allow(non_snake_case)] 4040fe15e0SLoGin #[inline(always)] 4140fe15e0SLoGin pub fn INITIAL_PROCESS_ADDRESS_SPACE() -> Arc<AddressSpace> { 4240fe15e0SLoGin unsafe { 4340fe15e0SLoGin return __INITIAL_PROCESS_ADDRESS_SPACE 4440fe15e0SLoGin .as_ref() 4540fe15e0SLoGin .expect("INITIAL_PROCESS_ADDRESS_SPACE is null") 4640fe15e0SLoGin .clone(); 4740fe15e0SLoGin } 4840fe15e0SLoGin } 4940fe15e0SLoGin 5040fe15e0SLoGin /// 设置内核INIT进程的用户地址空间结构体全局变量 5140fe15e0SLoGin #[allow(non_snake_case)] 5240fe15e0SLoGin pub unsafe fn set_INITIAL_PROCESS_ADDRESS_SPACE(address_space: Arc<AddressSpace>) { 5340fe15e0SLoGin static INITIALIZED: AtomicBool = AtomicBool::new(false); 5440fe15e0SLoGin if INITIALIZED 5540fe15e0SLoGin .compare_exchange(false, true, Ordering::SeqCst, Ordering::Acquire) 5640fe15e0SLoGin .is_err() 5740fe15e0SLoGin { 5840fe15e0SLoGin panic!("INITIAL_PROCESS_ADDRESS_SPACE is already initialized"); 5940fe15e0SLoGin } 6040fe15e0SLoGin __INITIAL_PROCESS_ADDRESS_SPACE = Some(address_space); 6140fe15e0SLoGin } 62004e86ffSlogin 63004e86ffSlogin /// @brief 将内核空间的虚拟地址转换为物理地址 64004e86ffSlogin #[inline(always)] 65004e86ffSlogin pub fn virt_2_phys(addr: usize) -> usize { 66004e86ffSlogin addr - PAGE_OFFSET as usize 67004e86ffSlogin } 68004e86ffSlogin 69004e86ffSlogin /// @brief 将物理地址转换为内核空间的虚拟地址 70004e86ffSlogin #[inline(always)] 71004e86ffSlogin pub fn phys_2_virt(addr: usize) -> usize { 72004e86ffSlogin addr + PAGE_OFFSET as usize 73004e86ffSlogin } 74bacd691cSlogin 7540fe15e0SLoGin #[derive(Clone, Copy, Debug, Eq, Ord, PartialEq, PartialOrd, Hash)] 7640fe15e0SLoGin pub enum PageTableKind { 7740fe15e0SLoGin /// 用户可访问的页表 7840fe15e0SLoGin User, 7940fe15e0SLoGin /// 内核页表 8040fe15e0SLoGin Kernel, 8140fe15e0SLoGin } 8240fe15e0SLoGin 8340fe15e0SLoGin /// 物理内存地址 8440fe15e0SLoGin #[derive(Clone, Copy, Eq, Ord, PartialEq, PartialOrd, Hash)] 8540fe15e0SLoGin #[repr(transparent)] 8640fe15e0SLoGin pub struct PhysAddr(usize); 8740fe15e0SLoGin 8840fe15e0SLoGin impl PhysAddr { 8940fe15e0SLoGin #[inline(always)] 9040fe15e0SLoGin pub const fn new(address: usize) -> Self { 9140fe15e0SLoGin Self(address) 9240fe15e0SLoGin } 9340fe15e0SLoGin 9440fe15e0SLoGin /// @brief 获取物理地址的值 9540fe15e0SLoGin #[inline(always)] 9640fe15e0SLoGin pub fn data(&self) -> usize { 9740fe15e0SLoGin self.0 9840fe15e0SLoGin } 9940fe15e0SLoGin 10040fe15e0SLoGin /// @brief 将物理地址加上一个偏移量 10140fe15e0SLoGin #[inline(always)] 10240fe15e0SLoGin pub fn add(self, offset: usize) -> Self { 10340fe15e0SLoGin Self(self.0 + offset) 10440fe15e0SLoGin } 10540fe15e0SLoGin 10640fe15e0SLoGin /// @brief 判断物理地址是否按照指定要求对齐 10740fe15e0SLoGin #[inline(always)] 10840fe15e0SLoGin pub fn check_aligned(&self, align: usize) -> bool { 10940fe15e0SLoGin return self.0 & (align - 1) == 0; 11040fe15e0SLoGin } 11140fe15e0SLoGin 11240fe15e0SLoGin #[inline(always)] 11340fe15e0SLoGin pub fn is_null(&self) -> bool { 11440fe15e0SLoGin return self.0 == 0; 11540fe15e0SLoGin } 11640fe15e0SLoGin } 11740fe15e0SLoGin 11840fe15e0SLoGin impl Debug for PhysAddr { 11940fe15e0SLoGin fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result { 12040fe15e0SLoGin write!(f, "PhysAddr({:#x})", self.0) 12140fe15e0SLoGin } 12240fe15e0SLoGin } 12340fe15e0SLoGin 12440fe15e0SLoGin impl core::ops::Add<usize> for PhysAddr { 12540fe15e0SLoGin type Output = Self; 12640fe15e0SLoGin 12740fe15e0SLoGin #[inline(always)] 12840fe15e0SLoGin fn add(self, rhs: usize) -> Self::Output { 12940fe15e0SLoGin return Self(self.0 + rhs); 13040fe15e0SLoGin } 13140fe15e0SLoGin } 13240fe15e0SLoGin 13340fe15e0SLoGin impl core::ops::AddAssign<usize> for PhysAddr { 13440fe15e0SLoGin #[inline(always)] 13540fe15e0SLoGin fn add_assign(&mut self, rhs: usize) { 13640fe15e0SLoGin self.0 += rhs; 13740fe15e0SLoGin } 13840fe15e0SLoGin } 13940fe15e0SLoGin 14040fe15e0SLoGin impl core::ops::Add<PhysAddr> for PhysAddr { 14140fe15e0SLoGin type Output = Self; 14240fe15e0SLoGin 14340fe15e0SLoGin #[inline(always)] 14440fe15e0SLoGin fn add(self, rhs: PhysAddr) -> Self::Output { 14540fe15e0SLoGin return Self(self.0 + rhs.0); 14640fe15e0SLoGin } 14740fe15e0SLoGin } 14840fe15e0SLoGin 14940fe15e0SLoGin impl core::ops::AddAssign<PhysAddr> for PhysAddr { 15040fe15e0SLoGin #[inline(always)] 15140fe15e0SLoGin fn add_assign(&mut self, rhs: PhysAddr) { 15240fe15e0SLoGin self.0 += rhs.0; 15340fe15e0SLoGin } 15440fe15e0SLoGin } 15540fe15e0SLoGin 15640fe15e0SLoGin impl core::ops::Sub<usize> for PhysAddr { 15740fe15e0SLoGin type Output = Self; 15840fe15e0SLoGin 15940fe15e0SLoGin #[inline(always)] 16040fe15e0SLoGin fn sub(self, rhs: usize) -> Self::Output { 16140fe15e0SLoGin return Self(self.0 - rhs); 16240fe15e0SLoGin } 16340fe15e0SLoGin } 16440fe15e0SLoGin 16540fe15e0SLoGin impl core::ops::SubAssign<usize> for PhysAddr { 16640fe15e0SLoGin #[inline(always)] 16740fe15e0SLoGin fn sub_assign(&mut self, rhs: usize) { 16840fe15e0SLoGin self.0 -= rhs; 16940fe15e0SLoGin } 17040fe15e0SLoGin } 17140fe15e0SLoGin 17240fe15e0SLoGin impl core::ops::Sub<PhysAddr> for PhysAddr { 17340fe15e0SLoGin type Output = usize; 17440fe15e0SLoGin 17540fe15e0SLoGin #[inline(always)] 17640fe15e0SLoGin fn sub(self, rhs: PhysAddr) -> Self::Output { 17740fe15e0SLoGin return self.0 - rhs.0; 17840fe15e0SLoGin } 17940fe15e0SLoGin } 18040fe15e0SLoGin 18140fe15e0SLoGin impl core::ops::SubAssign<PhysAddr> for PhysAddr { 18240fe15e0SLoGin #[inline(always)] 18340fe15e0SLoGin fn sub_assign(&mut self, rhs: PhysAddr) { 18440fe15e0SLoGin self.0 -= rhs.0; 18540fe15e0SLoGin } 18640fe15e0SLoGin } 18740fe15e0SLoGin 18840fe15e0SLoGin /// 虚拟内存地址 18940fe15e0SLoGin #[derive(Clone, Copy, Eq, Ord, PartialEq, PartialOrd, Hash)] 19040fe15e0SLoGin #[repr(transparent)] 19140fe15e0SLoGin pub struct VirtAddr(usize); 19240fe15e0SLoGin 19340fe15e0SLoGin impl VirtAddr { 19440fe15e0SLoGin #[inline(always)] 19540fe15e0SLoGin pub const fn new(address: usize) -> Self { 19640fe15e0SLoGin return Self(address); 19740fe15e0SLoGin } 19840fe15e0SLoGin 19940fe15e0SLoGin /// @brief 获取虚拟地址的值 20040fe15e0SLoGin #[inline(always)] 20140fe15e0SLoGin pub fn data(&self) -> usize { 20240fe15e0SLoGin return self.0; 20340fe15e0SLoGin } 20440fe15e0SLoGin 20540fe15e0SLoGin /// @brief 判断虚拟地址的类型 20640fe15e0SLoGin #[inline(always)] 20740fe15e0SLoGin pub fn kind(&self) -> PageTableKind { 20840fe15e0SLoGin if self.check_user() { 20940fe15e0SLoGin return PageTableKind::User; 21040fe15e0SLoGin } else { 21140fe15e0SLoGin return PageTableKind::Kernel; 21240fe15e0SLoGin } 21340fe15e0SLoGin } 21440fe15e0SLoGin 21540fe15e0SLoGin /// @brief 判断虚拟地址是否按照指定要求对齐 21640fe15e0SLoGin #[inline(always)] 21740fe15e0SLoGin pub fn check_aligned(&self, align: usize) -> bool { 21840fe15e0SLoGin return self.0 & (align - 1) == 0; 21940fe15e0SLoGin } 22040fe15e0SLoGin 22140fe15e0SLoGin /// @brief 判断虚拟地址是否在用户空间 22240fe15e0SLoGin #[inline(always)] 22340fe15e0SLoGin pub fn check_user(&self) -> bool { 22440fe15e0SLoGin if self < &MMArch::USER_END_VADDR { 22540fe15e0SLoGin return true; 22640fe15e0SLoGin } else { 22740fe15e0SLoGin return false; 22840fe15e0SLoGin } 22940fe15e0SLoGin } 23040fe15e0SLoGin 23140fe15e0SLoGin #[inline(always)] 23240fe15e0SLoGin pub fn as_ptr<T>(self) -> *mut T { 23340fe15e0SLoGin return self.0 as *mut T; 23440fe15e0SLoGin } 23540fe15e0SLoGin 23640fe15e0SLoGin #[inline(always)] 23740fe15e0SLoGin pub fn is_null(&self) -> bool { 23840fe15e0SLoGin return self.0 == 0; 23940fe15e0SLoGin } 24040fe15e0SLoGin } 24140fe15e0SLoGin 24240fe15e0SLoGin impl Add<VirtAddr> for VirtAddr { 24340fe15e0SLoGin type Output = Self; 24440fe15e0SLoGin 24540fe15e0SLoGin #[inline(always)] 24640fe15e0SLoGin fn add(self, rhs: VirtAddr) -> Self::Output { 24740fe15e0SLoGin return Self(self.0 + rhs.0); 24840fe15e0SLoGin } 24940fe15e0SLoGin } 25040fe15e0SLoGin 25140fe15e0SLoGin impl Add<usize> for VirtAddr { 25240fe15e0SLoGin type Output = Self; 25340fe15e0SLoGin 25440fe15e0SLoGin #[inline(always)] 25540fe15e0SLoGin fn add(self, rhs: usize) -> Self::Output { 25640fe15e0SLoGin return Self(self.0 + rhs); 25740fe15e0SLoGin } 25840fe15e0SLoGin } 25940fe15e0SLoGin 26040fe15e0SLoGin impl Sub<VirtAddr> for VirtAddr { 26140fe15e0SLoGin type Output = usize; 26240fe15e0SLoGin 26340fe15e0SLoGin #[inline(always)] 26440fe15e0SLoGin fn sub(self, rhs: VirtAddr) -> Self::Output { 26540fe15e0SLoGin return self.0 - rhs.0; 26640fe15e0SLoGin } 26740fe15e0SLoGin } 26840fe15e0SLoGin 26940fe15e0SLoGin impl Sub<usize> for VirtAddr { 27040fe15e0SLoGin type Output = Self; 27140fe15e0SLoGin 27240fe15e0SLoGin #[inline(always)] 27340fe15e0SLoGin fn sub(self, rhs: usize) -> Self::Output { 27440fe15e0SLoGin return Self(self.0 - rhs); 27540fe15e0SLoGin } 27640fe15e0SLoGin } 27740fe15e0SLoGin 27840fe15e0SLoGin impl AddAssign<usize> for VirtAddr { 27940fe15e0SLoGin #[inline(always)] 28040fe15e0SLoGin fn add_assign(&mut self, rhs: usize) { 28140fe15e0SLoGin self.0 += rhs; 28240fe15e0SLoGin } 28340fe15e0SLoGin } 28440fe15e0SLoGin 28540fe15e0SLoGin impl AddAssign<VirtAddr> for VirtAddr { 28640fe15e0SLoGin #[inline(always)] 28740fe15e0SLoGin fn add_assign(&mut self, rhs: VirtAddr) { 28840fe15e0SLoGin self.0 += rhs.0; 28940fe15e0SLoGin } 29040fe15e0SLoGin } 29140fe15e0SLoGin 29240fe15e0SLoGin impl SubAssign<usize> for VirtAddr { 29340fe15e0SLoGin #[inline(always)] 29440fe15e0SLoGin fn sub_assign(&mut self, rhs: usize) { 29540fe15e0SLoGin self.0 -= rhs; 29640fe15e0SLoGin } 29740fe15e0SLoGin } 29840fe15e0SLoGin 29940fe15e0SLoGin impl SubAssign<VirtAddr> for VirtAddr { 30040fe15e0SLoGin #[inline(always)] 30140fe15e0SLoGin fn sub_assign(&mut self, rhs: VirtAddr) { 30240fe15e0SLoGin self.0 -= rhs.0; 30340fe15e0SLoGin } 30440fe15e0SLoGin } 30540fe15e0SLoGin 30640fe15e0SLoGin impl Debug for VirtAddr { 30740fe15e0SLoGin fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result { 30840fe15e0SLoGin write!(f, "VirtAddr({:#x})", self.0) 30940fe15e0SLoGin } 31040fe15e0SLoGin } 31140fe15e0SLoGin 31240fe15e0SLoGin /// @brief 物理内存区域 31340fe15e0SLoGin #[derive(Clone, Copy, Debug)] 31440fe15e0SLoGin pub struct PhysMemoryArea { 31540fe15e0SLoGin /// 物理基地址 31640fe15e0SLoGin pub base: PhysAddr, 31740fe15e0SLoGin /// 该区域的物理内存大小 31840fe15e0SLoGin pub size: usize, 31940fe15e0SLoGin } 32040fe15e0SLoGin 32140fe15e0SLoGin pub trait MemoryManagementArch: Clone + Copy + Debug { 32240fe15e0SLoGin /// 页面大小的shift(假如页面4K,那么这个值就是12,因为2^12=4096) 32340fe15e0SLoGin const PAGE_SHIFT: usize; 32440fe15e0SLoGin /// 每个页表的页表项数目。(以2^n次幂来表示)假如有512个页表项,那么这个值就是9 32540fe15e0SLoGin const PAGE_ENTRY_SHIFT: usize; 32640fe15e0SLoGin /// 页表层级数量 32740fe15e0SLoGin const PAGE_LEVELS: usize; 32840fe15e0SLoGin 32940fe15e0SLoGin /// 页表项的有效位的index(假如页表项的第0-51位有效,那么这个值就是52) 33040fe15e0SLoGin const ENTRY_ADDRESS_SHIFT: usize; 33140fe15e0SLoGin /// 页面的页表项的默认值 33240fe15e0SLoGin const ENTRY_FLAG_DEFAULT_PAGE: usize; 33340fe15e0SLoGin /// 页表的页表项的默认值 33440fe15e0SLoGin const ENTRY_FLAG_DEFAULT_TABLE: usize; 33540fe15e0SLoGin /// 页表项的present位被置位之后的值 33640fe15e0SLoGin const ENTRY_FLAG_PRESENT: usize; 33740fe15e0SLoGin /// 页表项为read only时的值 33840fe15e0SLoGin const ENTRY_FLAG_READONLY: usize; 33940fe15e0SLoGin /// 页表项为可读写状态的值 34040fe15e0SLoGin const ENTRY_FLAG_READWRITE: usize; 34140fe15e0SLoGin /// 页面项标记页面为user page的值 34240fe15e0SLoGin const ENTRY_FLAG_USER: usize; 34340fe15e0SLoGin /// 页面项标记页面为write through的值 34440fe15e0SLoGin const ENTRY_FLAG_WRITE_THROUGH: usize; 34540fe15e0SLoGin /// 页面项标记页面为cache disable的值 34640fe15e0SLoGin const ENTRY_FLAG_CACHE_DISABLE: usize; 34740fe15e0SLoGin /// 标记当前页面不可执行的标志位(Execute disable)(也就是说,不能从这段内存里面获取处理器指令) 34840fe15e0SLoGin const ENTRY_FLAG_NO_EXEC: usize; 34940fe15e0SLoGin /// 标记当前页面可执行的标志位(Execute enable) 35040fe15e0SLoGin const ENTRY_FLAG_EXEC: usize; 35140fe15e0SLoGin 35240fe15e0SLoGin /// 虚拟地址与物理地址的偏移量 35340fe15e0SLoGin const PHYS_OFFSET: usize; 35440fe15e0SLoGin 35540fe15e0SLoGin /// 每个页面的大小 35640fe15e0SLoGin const PAGE_SIZE: usize = 1 << Self::PAGE_SHIFT; 35740fe15e0SLoGin /// 通过这个mask,获取地址的页内偏移量 35840fe15e0SLoGin const PAGE_OFFSET_MASK: usize = Self::PAGE_SIZE - 1; 35940fe15e0SLoGin /// 页表项的地址、数据部分的shift。 36040fe15e0SLoGin /// 打个比方,如果这个值为52,那么意味着页表项的[0, 52)位,用于表示地址以及其他的标志位 36140fe15e0SLoGin const PAGE_ADDRESS_SHIFT: usize = Self::PAGE_LEVELS * Self::PAGE_ENTRY_SHIFT + Self::PAGE_SHIFT; 36240fe15e0SLoGin /// 最大的虚拟地址(对于不同的架构,由于上述PAGE_ADDRESS_SHIFT可能包括了reserved bits, 事实上能表示的虚拟地址应该比这个值要小) 36340fe15e0SLoGin const PAGE_ADDRESS_SIZE: usize = 1 << Self::PAGE_ADDRESS_SHIFT; 36440fe15e0SLoGin /// 页表项的值与这个常量进行与运算,得到的结果是所填写的物理地址 36540fe15e0SLoGin const PAGE_ADDRESS_MASK: usize = Self::PAGE_ADDRESS_SIZE - Self::PAGE_SIZE; 36640fe15e0SLoGin /// 每个页表项的大小 36740fe15e0SLoGin const PAGE_ENTRY_SIZE: usize = 1 << (Self::PAGE_SHIFT - Self::PAGE_ENTRY_SHIFT); 36840fe15e0SLoGin /// 每个页表的页表项数目 36940fe15e0SLoGin const PAGE_ENTRY_NUM: usize = 1 << Self::PAGE_ENTRY_SHIFT; 37040fe15e0SLoGin /// 该字段用于根据虚拟地址,获取该虚拟地址在对应的页表中是第几个页表项 37140fe15e0SLoGin const PAGE_ENTRY_MASK: usize = Self::PAGE_ENTRY_NUM - 1; 37240fe15e0SLoGin 37340fe15e0SLoGin const PAGE_NEGATIVE_MASK: usize = !((Self::PAGE_ADDRESS_SIZE) - 1); 37440fe15e0SLoGin 37540fe15e0SLoGin const ENTRY_ADDRESS_SIZE: usize = 1 << Self::ENTRY_ADDRESS_SHIFT; 37640fe15e0SLoGin /// 该mask用于获取页表项中地址字段 37740fe15e0SLoGin const ENTRY_ADDRESS_MASK: usize = Self::ENTRY_ADDRESS_SIZE - Self::PAGE_SIZE; 37840fe15e0SLoGin /// 这个mask用于获取页表项中的flags 37940fe15e0SLoGin const ENTRY_FLAGS_MASK: usize = !Self::ENTRY_ADDRESS_MASK; 38040fe15e0SLoGin 38140fe15e0SLoGin /// 用户空间的最高地址 38240fe15e0SLoGin const USER_END_VADDR: VirtAddr; 38340fe15e0SLoGin /// 用户堆的起始地址 38440fe15e0SLoGin const USER_BRK_START: VirtAddr; 38540fe15e0SLoGin /// 用户栈起始地址(向下生长,不包含该值) 38640fe15e0SLoGin const USER_STACK_START: VirtAddr; 38740fe15e0SLoGin 38840fe15e0SLoGin /// @brief 用于初始化内存管理模块与架构相关的信息。 38940fe15e0SLoGin /// 该函数应调用其他模块的接口,生成内存区域结构体,提供给BumpAllocator使用 39040fe15e0SLoGin unsafe fn init() -> &'static [PhysMemoryArea]; 39140fe15e0SLoGin 39240fe15e0SLoGin /// @brief 读取指定虚拟地址的值,并假设它是类型T的指针 39340fe15e0SLoGin #[inline(always)] 39440fe15e0SLoGin unsafe fn read<T>(address: VirtAddr) -> T { 39540fe15e0SLoGin return ptr::read(address.data() as *const T); 39640fe15e0SLoGin } 39740fe15e0SLoGin 39840fe15e0SLoGin /// @brief 将value写入到指定的虚拟地址 39940fe15e0SLoGin #[inline(always)] 40040fe15e0SLoGin unsafe fn write<T>(address: VirtAddr, value: T) { 40140fe15e0SLoGin ptr::write(address.data() as *mut T, value); 40240fe15e0SLoGin } 40340fe15e0SLoGin 40440fe15e0SLoGin #[inline(always)] 40540fe15e0SLoGin unsafe fn write_bytes(address: VirtAddr, value: u8, count: usize) { 40640fe15e0SLoGin ptr::write_bytes(address.data() as *mut u8, value, count); 40740fe15e0SLoGin } 40840fe15e0SLoGin 40940fe15e0SLoGin /// @brief 刷新TLB中,关于指定虚拟地址的条目 41040fe15e0SLoGin unsafe fn invalidate_page(address: VirtAddr); 41140fe15e0SLoGin 41240fe15e0SLoGin /// @brief 刷新TLB中,所有的条目 41340fe15e0SLoGin unsafe fn invalidate_all(); 41440fe15e0SLoGin 41540fe15e0SLoGin /// @brief 获取顶级页表的物理地址 41640fe15e0SLoGin unsafe fn table(table_kind: PageTableKind) -> PhysAddr; 41740fe15e0SLoGin 41840fe15e0SLoGin /// @brief 设置顶级页表的物理地址到处理器中 41940fe15e0SLoGin unsafe fn set_table(table_kind: PageTableKind, table: PhysAddr); 42040fe15e0SLoGin 42140fe15e0SLoGin /// @brief 将物理地址转换为虚拟地址. 42240fe15e0SLoGin /// 42340fe15e0SLoGin /// @param phys 物理地址 42440fe15e0SLoGin /// 42540fe15e0SLoGin /// @return 转换后的虚拟地址。如果转换失败,返回None 42640fe15e0SLoGin #[inline(always)] 42740fe15e0SLoGin unsafe fn phys_2_virt(phys: PhysAddr) -> Option<VirtAddr> { 42840fe15e0SLoGin if let Some(vaddr) = phys.data().checked_add(Self::PHYS_OFFSET) { 42940fe15e0SLoGin return Some(VirtAddr::new(vaddr)); 43040fe15e0SLoGin } else { 43140fe15e0SLoGin return None; 43240fe15e0SLoGin } 43340fe15e0SLoGin } 43440fe15e0SLoGin 43540fe15e0SLoGin /// 将虚拟地址转换为物理地址 43640fe15e0SLoGin /// 43740fe15e0SLoGin /// ## 参数 43840fe15e0SLoGin /// 43940fe15e0SLoGin /// - `virt` 虚拟地址 44040fe15e0SLoGin /// 44140fe15e0SLoGin /// ## 返回值 44240fe15e0SLoGin /// 44340fe15e0SLoGin /// 转换后的物理地址。如果转换失败,返回None 44440fe15e0SLoGin #[inline(always)] 44540fe15e0SLoGin unsafe fn virt_2_phys(virt: VirtAddr) -> Option<PhysAddr> { 44640fe15e0SLoGin if let Some(paddr) = virt.data().checked_sub(Self::PHYS_OFFSET) { 44740fe15e0SLoGin return Some(PhysAddr::new(paddr)); 44840fe15e0SLoGin } else { 44940fe15e0SLoGin return None; 45040fe15e0SLoGin } 45140fe15e0SLoGin } 45240fe15e0SLoGin 45340fe15e0SLoGin /// @brief 判断指定的虚拟地址是否正确(符合规范) 45440fe15e0SLoGin fn virt_is_valid(virt: VirtAddr) -> bool; 45540fe15e0SLoGin 45640fe15e0SLoGin /// 获取内存管理初始化时,创建的第一个内核页表的地址 45740fe15e0SLoGin fn initial_page_table() -> PhysAddr; 45840fe15e0SLoGin 45940fe15e0SLoGin /// 初始化新的usermapper,为用户进程创建页表 46040fe15e0SLoGin fn setup_new_usermapper() -> Result<UserMapper, SystemError>; 46140fe15e0SLoGin } 46240fe15e0SLoGin 46340fe15e0SLoGin /// @brief 虚拟地址范围 46440fe15e0SLoGin /// 该结构体用于表示一个虚拟地址范围,包括起始地址与大小 46540fe15e0SLoGin /// 46640fe15e0SLoGin /// 请注意与VMA进行区分,该结构体被VMA所包含 46740fe15e0SLoGin #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)] 46840fe15e0SLoGin pub struct VirtRegion { 46940fe15e0SLoGin start: VirtAddr, 47040fe15e0SLoGin size: usize, 47140fe15e0SLoGin } 47240fe15e0SLoGin 47340fe15e0SLoGin #[allow(dead_code)] 47440fe15e0SLoGin impl VirtRegion { 47540fe15e0SLoGin /// # 创建一个新的虚拟地址范围 47640fe15e0SLoGin pub fn new(start: VirtAddr, size: usize) -> Self { 47740fe15e0SLoGin VirtRegion { start, size } 47840fe15e0SLoGin } 47940fe15e0SLoGin 48040fe15e0SLoGin /// 获取虚拟地址范围的起始地址 48140fe15e0SLoGin #[inline(always)] 48240fe15e0SLoGin pub fn start(&self) -> VirtAddr { 48340fe15e0SLoGin self.start 48440fe15e0SLoGin } 48540fe15e0SLoGin 48640fe15e0SLoGin /// 获取虚拟地址范围的截止地址(不包括返回的地址) 48740fe15e0SLoGin #[inline(always)] 48840fe15e0SLoGin pub fn end(&self) -> VirtAddr { 48940fe15e0SLoGin return self.start().add(self.size); 49040fe15e0SLoGin } 49140fe15e0SLoGin 49240fe15e0SLoGin /// # Create a new VirtRegion from a range [start, end) 49340fe15e0SLoGin /// 49440fe15e0SLoGin /// If end <= start, return None 49540fe15e0SLoGin pub fn between(start: VirtAddr, end: VirtAddr) -> Option<Self> { 49640fe15e0SLoGin if unlikely(end.data() <= start.data()) { 49740fe15e0SLoGin return None; 49840fe15e0SLoGin } 49940fe15e0SLoGin let size = end.data() - start.data(); 50040fe15e0SLoGin return Some(VirtRegion::new(start, size)); 50140fe15e0SLoGin } 50240fe15e0SLoGin 50340fe15e0SLoGin /// # 取两个虚拟地址范围的交集 50440fe15e0SLoGin /// 50540fe15e0SLoGin /// 如果两个虚拟地址范围没有交集,返回None 50640fe15e0SLoGin pub fn intersect(&self, other: &VirtRegion) -> Option<VirtRegion> { 50740fe15e0SLoGin let start = self.start.max(other.start); 50840fe15e0SLoGin let end = self.end().min(other.end()); 50940fe15e0SLoGin return VirtRegion::between(start, end); 51040fe15e0SLoGin } 51140fe15e0SLoGin 51240fe15e0SLoGin /// 设置虚拟地址范围的起始地址 51340fe15e0SLoGin #[inline(always)] 51440fe15e0SLoGin pub fn set_start(&mut self, start: VirtAddr) { 51540fe15e0SLoGin self.start = start; 51640fe15e0SLoGin } 51740fe15e0SLoGin 51840fe15e0SLoGin #[inline(always)] 51940fe15e0SLoGin pub fn size(&self) -> usize { 52040fe15e0SLoGin self.size 52140fe15e0SLoGin } 52240fe15e0SLoGin 52340fe15e0SLoGin /// 设置虚拟地址范围的大小 52440fe15e0SLoGin #[inline(always)] 52540fe15e0SLoGin pub fn set_size(&mut self, size: usize) { 52640fe15e0SLoGin self.size = size; 52740fe15e0SLoGin } 52840fe15e0SLoGin 52940fe15e0SLoGin /// 判断虚拟地址范围是否为空 53040fe15e0SLoGin #[inline(always)] 53140fe15e0SLoGin pub fn is_empty(&self) -> bool { 53240fe15e0SLoGin self.size == 0 53340fe15e0SLoGin } 53440fe15e0SLoGin 53540fe15e0SLoGin /// 将虚拟地址区域的大小向上对齐到页大小 53640fe15e0SLoGin #[inline(always)] 53740fe15e0SLoGin pub fn round_up_size_to_page(self) -> Self { 53840fe15e0SLoGin return VirtRegion::new(self.start, round_up_to_page_size(self.size)); 53940fe15e0SLoGin } 54040fe15e0SLoGin 54140fe15e0SLoGin /// 判断两个虚拟地址范围是否由于具有交集而导致冲突 54240fe15e0SLoGin #[inline(always)] 54340fe15e0SLoGin pub fn collide(&self, other: &VirtRegion) -> bool { 54440fe15e0SLoGin return self.intersect(other).is_some(); 54540fe15e0SLoGin } 54640fe15e0SLoGin 54740fe15e0SLoGin pub fn iter_pages(&self) -> VirtPageFrameIter { 54840fe15e0SLoGin return VirtPageFrame::iter_range( 54940fe15e0SLoGin VirtPageFrame::new(self.start), 55040fe15e0SLoGin VirtPageFrame::new(self.end()), 55140fe15e0SLoGin ); 55240fe15e0SLoGin } 55340fe15e0SLoGin 55440fe15e0SLoGin /// 获取[self.start(), region.start())的虚拟地址范围 55540fe15e0SLoGin /// 55640fe15e0SLoGin /// 如果self.start() >= region.start(),返回None 55740fe15e0SLoGin pub fn before(self, region: &VirtRegion) -> Option<Self> { 55840fe15e0SLoGin return Self::between(self.start(), region.start()); 55940fe15e0SLoGin } 56040fe15e0SLoGin 56140fe15e0SLoGin /// 获取[region.end(),self.end())的虚拟地址范围 56240fe15e0SLoGin /// 56340fe15e0SLoGin /// 如果 self.end() >= region.end() ,返回None 56440fe15e0SLoGin pub fn after(self, region: &VirtRegion) -> Option<Self> { 56540fe15e0SLoGin // if self.end() > region.end() none 56640fe15e0SLoGin return Self::between(region.end(), self.end()); 56740fe15e0SLoGin } 56840fe15e0SLoGin 56940fe15e0SLoGin /// 把当前虚拟地址范围内的某个虚拟地址,转换为另一个虚拟地址范围内的虚拟地址 57040fe15e0SLoGin /// 57140fe15e0SLoGin /// 如果vaddr不在当前虚拟地址范围内,返回None 57240fe15e0SLoGin /// 57340fe15e0SLoGin /// 如果vaddr在当前虚拟地址范围内,返回vaddr在new_base中的虚拟地址 57440fe15e0SLoGin pub fn rebase(self, vaddr: VirtAddr, new_base: &VirtRegion) -> Option<VirtAddr> { 57540fe15e0SLoGin if !self.contains(vaddr) { 57640fe15e0SLoGin return None; 57740fe15e0SLoGin } 57840fe15e0SLoGin let offset = vaddr.data() - self.start().data(); 57940fe15e0SLoGin let new_start = new_base.start().data() + offset; 58040fe15e0SLoGin return Some(VirtAddr::new(new_start)); 58140fe15e0SLoGin } 58240fe15e0SLoGin 58340fe15e0SLoGin /// 判断虚拟地址范围是否包含指定的虚拟地址 58440fe15e0SLoGin pub fn contains(&self, addr: VirtAddr) -> bool { 58540fe15e0SLoGin return self.start() <= addr && addr < self.end(); 58640fe15e0SLoGin } 58740fe15e0SLoGin 58840fe15e0SLoGin /// 创建当前虚拟地址范围的页面迭代器 58940fe15e0SLoGin pub fn pages(&self) -> VirtPageFrameIter { 59040fe15e0SLoGin return VirtPageFrame::iter_range( 59140fe15e0SLoGin VirtPageFrame::new(self.start()), 59240fe15e0SLoGin VirtPageFrame::new(self.end()), 59340fe15e0SLoGin ); 59440fe15e0SLoGin } 59540fe15e0SLoGin } 59640fe15e0SLoGin 59740fe15e0SLoGin impl PartialOrd for VirtRegion { 59840fe15e0SLoGin fn partial_cmp(&self, other: &Self) -> Option<cmp::Ordering> { 59940fe15e0SLoGin return self.start.partial_cmp(&other.start); 60040fe15e0SLoGin } 60140fe15e0SLoGin } 60240fe15e0SLoGin 60340fe15e0SLoGin impl Ord for VirtRegion { 60440fe15e0SLoGin fn cmp(&self, other: &Self) -> cmp::Ordering { 60540fe15e0SLoGin return self.start.cmp(&other.start); 60640fe15e0SLoGin } 60740fe15e0SLoGin } 60840fe15e0SLoGin 60940fe15e0SLoGin /// ## 判断虚拟地址是否超出了用户空间 61040fe15e0SLoGin /// 61140fe15e0SLoGin /// 如果虚拟地址超出了用户空间,返回Err(SystemError::EFAULT). 61240fe15e0SLoGin /// 如果end < start,返回Err(SystemError::EOVERFLOW) 61340fe15e0SLoGin /// 61440fe15e0SLoGin /// 否则返回Ok(()) 61540fe15e0SLoGin pub fn verify_area(addr: VirtAddr, size: usize) -> Result<(), SystemError> { 61640fe15e0SLoGin let end = addr.add(size); 61740fe15e0SLoGin if unlikely(end.data() < addr.data()) { 61840fe15e0SLoGin return Err(SystemError::EOVERFLOW); 61940fe15e0SLoGin } 62040fe15e0SLoGin 62140fe15e0SLoGin if !addr.check_user() || !end.check_user() { 62240fe15e0SLoGin return Err(SystemError::EFAULT); 62340fe15e0SLoGin } 62440fe15e0SLoGin 62540fe15e0SLoGin return Ok(()); 62640fe15e0SLoGin } 62740fe15e0SLoGin // ====== 重构内存管理、进程管理后,请删除这几行 BEGIN ====== 628bacd691cSlogin //BUG pcb问题 629bacd691cSlogin unsafe impl Send for process_control_block {} 630bacd691cSlogin unsafe impl Sync for process_control_block {} 631bacd691cSlogin 63240fe15e0SLoGin // ====== 重构内存管理后,请删除这几行 END ======= 633