mirror of
https://github.com/DragonOS-Community/DragonOS.git
synced 2025-06-08 22:36:48 +00:00
520 lines
15 KiB
C
520 lines
15 KiB
C
#pragma once
|
||
|
||
#include <common/glib.h>
|
||
#include <mm/mm-types.h>
|
||
#include <process/process.h>
|
||
|
||
// 每个页表的项数
|
||
// 64位下，每个页表4k，每条页表项8B，故一个页表有512条
|
||
#define PTRS_PER_PGT 512
|
||
|
||
// 内核层的起始地址
|
||
#define PAGE_OFFSET (0xffff800000000000UL)
|
||
#define KERNEL_BASE_LINEAR_ADDR (0xffff800000000000UL)
|
||
#define USER_MAX_LINEAR_ADDR 0x00007fffffffffffUL
|
||
|
||
#define PAGE_4K_SHIFT 12
|
||
#define PAGE_2M_SHIFT 21
|
||
#define PAGE_1G_SHIFT 30
|
||
#define PAGE_GDT_SHIFT 39
|
||
|
||
// 不同大小的页的容量
|
||
#define PAGE_4K_SIZE (1UL << PAGE_4K_SHIFT)
|
||
#define PAGE_2M_SIZE (1UL << PAGE_2M_SHIFT)
|
||
#define PAGE_1G_SIZE (1UL << PAGE_1G_SHIFT)
|
||
|
||
// 屏蔽低于x的数值
|
||
#define PAGE_4K_MASK (~(PAGE_4K_SIZE - 1))
|
||
#define PAGE_2M_MASK (~(PAGE_2M_SIZE - 1))
|
||
|
||
// 将addr按照x的上边界对齐
|
||
#define PAGE_4K_ALIGN(addr) (((unsigned long)(addr) + PAGE_4K_SIZE - 1) & PAGE_4K_MASK)
|
||
#define PAGE_2M_ALIGN(addr) (((unsigned long)(addr) + PAGE_2M_SIZE - 1) & PAGE_2M_MASK)
|
||
|
||
// 虚拟地址与物理地址转换
|
||
#define virt_2_phys(addr) ((unsigned long)(addr)-PAGE_OFFSET)
|
||
#define phys_2_virt(addr) ((unsigned long *)((unsigned long)(addr) + PAGE_OFFSET))
|
||
// 获取对应的页结构体
|
||
#define Virt_To_2M_Page(kaddr) (memory_management_struct.pages_struct + (virt_2_phys(kaddr) >> PAGE_2M_SHIFT))
|
||
#define Phy_to_2M_Page(kaddr) (memory_management_struct.pages_struct + ((unsigned long)(kaddr) >> PAGE_2M_SHIFT))
|
||
|
||
// 在这个地址以上的虚拟空间，用来进行特殊的映射
|
||
#define SPECIAL_MEMOEY_MAPPING_VIRT_ADDR_BASE 0xffffa00000000000UL
|
||
#define FRAME_BUFFER_MAPPING_OFFSET 0x3000000UL
|
||
#define ACPI_RSDT_MAPPING_OFFSET 0x7000000UL
|
||
#define ACPI_XSDT_MAPPING_OFFSET 0x9000000UL
|
||
#define IO_APIC_MAPPING_OFFSET 0xfec00000UL
|
||
#define LOCAL_APIC_MAPPING_OFFSET 0xfee00000UL
|
||
#define AHCI_MAPPING_OFFSET 0xff200000UL // AHCI 映射偏移量,之后使用了4M的地址
|
||
#define XHCI_MAPPING_OFFSET 0x100000000 // XHCI控制器映射偏移量(后方请预留1GB的虚拟空间来映射不同的controller)
|
||
|
||
// ===== 内存区域属性 =====
|
||
// DMA区域
|
||
#define ZONE_DMA (1 << 0)
|
||
// 已在页表中映射的区域
|
||
#define ZONE_NORMAL (1 << 1)
|
||
// 未在页表中映射的区域
|
||
#define ZONE_UNMAPPED_IN_PGT (1 << 2)
|
||
|
||
// ===== 页面属性 =====
|
||
// 页面在页表中已被映射 mapped=1 unmapped=0
|
||
#define PAGE_PGT_MAPPED (1 << 0)
|
||
|
||
// 内核初始化所占用的页 init-code=1 normal-code/data=0
|
||
#define PAGE_KERNEL_INIT (1 << 1)
|
||
|
||
// 1=设备MMIO映射的内存 0=物理内存
|
||
#define PAGE_DEVICE (1 << 2)
|
||
|
||
// 内核层页 kernel=1 memory=0
|
||
#define PAGE_KERNEL (1 << 3)
|
||
|
||
// 共享的页 shared=1 single-use=0
|
||
#define PAGE_SHARED (1 << 4)
|
||
|
||
// =========== 页表项权限 ========
|
||
|
||
// bit 63 Execution Disable:
|
||
#define PAGE_XD (1UL << 63)
|
||
|
||
// bit 12 Page Attribute Table
|
||
#define PAGE_PAT (1UL << 12)
|
||
// 对于PTE而言，第7位是PAT
|
||
#define PAGE_4K_PAT (1UL << 7)
|
||
|
||
// bit 8 Global Page:1,global;0,part
|
||
#define PAGE_GLOBAL (1UL << 8)
|
||
|
||
// bit 7 Page Size:1,big page;0,small page;
|
||
#define PAGE_PS (1UL << 7)
|
||
|
||
// bit 6 Dirty:1,dirty;0,clean;
|
||
#define PAGE_DIRTY (1UL << 6)
|
||
|
||
// bit 5 Accessed:1,visited;0,unvisited;
|
||
#define PAGE_ACCESSED (1UL << 5)
|
||
|
||
// bit 4 Page Level Cache Disable
|
||
#define PAGE_PCD (1UL << 4)
|
||
|
||
// bit 3 Page Level Write Through
|
||
#define PAGE_PWT (1UL << 3)
|
||
|
||
// bit 2 User Supervisor:1,user and supervisor;0,supervisor;
|
||
#define PAGE_U_S (1UL << 2)
|
||
|
||
// bit 1 Read Write:1,read and write;0,read;
|
||
#define PAGE_R_W (1UL << 1)
|
||
|
||
// bit 0 Present:1,present;0,no present;
|
||
#define PAGE_PRESENT (1UL << 0)
|
||
|
||
// 1,0
|
||
#define PAGE_KERNEL_PGT (PAGE_R_W | PAGE_PRESENT)
|
||
|
||
// 1,0
|
||
#define PAGE_KERNEL_DIR (PAGE_R_W | PAGE_PRESENT)
|
||
|
||
// 1,0 (4级页表在3级页表中的页表项的属性)
|
||
#define PAGE_KERNEL_PDE (PAGE_R_W | PAGE_PRESENT)
|
||
|
||
// 7,1,0
|
||
#define PAGE_KERNEL_PAGE (PAGE_PS | PAGE_R_W | PAGE_PRESENT)
|
||
|
||
#define PAGE_KERNEL_4K_PAGE (PAGE_R_W | PAGE_PRESENT)
|
||
|
||
#define PAGE_USER_PGT (PAGE_U_S | PAGE_R_W | PAGE_PRESENT)
|
||
|
||
// 2,1,0
|
||
#define PAGE_USER_DIR (PAGE_U_S | PAGE_R_W | PAGE_PRESENT)
|
||
|
||
// 1,0 (4级页表在3级页表中的页表项的属性)
|
||
#define PAGE_USER_PDE (PAGE_U_S | PAGE_R_W | PAGE_PRESENT)
|
||
// 7,2,1,0
|
||
#define PAGE_USER_PAGE (PAGE_PS | PAGE_U_S | PAGE_R_W | PAGE_PRESENT)
|
||
|
||
#define PAGE_USER_4K_PAGE (PAGE_U_S | PAGE_R_W | PAGE_PRESENT)
|
||
|
||
// ===== 错误码定义 ====
|
||
// 物理页结构体为空
|
||
#define EPAGE_NULL 1
|
||
|
||
/**
|
||
* @brief 刷新TLB的宏定义
|
||
* 由于任何写入cr3的操作都会刷新TLB，因此这个宏定义可以刷新TLB
|
||
*/
|
||
#define flush_tlb() \
|
||
do \
|
||
{ \
|
||
ul tmp; \
|
||
io_mfence(); \
|
||
__asm__ __volatile__( \
|
||
"movq %%cr3, %0\n\t" \
|
||
"movq %0, %%cr3\n\t" \
|
||
: "=r"(tmp)::"memory"); \
|
||
\
|
||
} while (0);
|
||
|
||
/**
|
||
* @brief 系统内存信息结构体（单位：字节）
|
||
*
|
||
*/
|
||
struct mm_stat_t
|
||
{
|
||
uint64_t total; // 计算机的总内存数量大小
|
||
uint64_t used; // 已使用的内存大小
|
||
uint64_t free; // 空闲物理页所占的内存大小
|
||
uint64_t shared; // 共享的内存大小
|
||
uint64_t cache_used; // 位于slab缓冲区中的已使用的内存大小
|
||
uint64_t cache_free; // 位于slab缓冲区中的空闲的内存大小
|
||
uint64_t available; // 系统总空闲内存大小（包括kmalloc缓冲区）
|
||
};
|
||
|
||
/**
|
||
* @brief 虚拟内存区域的操作方法的结构体
|
||
*
|
||
*/
|
||
struct vm_operations_t
|
||
{
|
||
/**
|
||
* @brief vm area 被打开时的回调函数
|
||
*
|
||
*/
|
||
void (*open)(struct vm_area_struct *area);
|
||
/**
|
||
* @brief vm area将要被移除的时候，将会调用该回调函数
|
||
*
|
||
*/
|
||
void (*close)(struct vm_area_struct *area);
|
||
};
|
||
|
||
extern struct memory_desc memory_management_struct;
|
||
|
||
// 导出内核程序的几个段的起止地址
|
||
extern char _text;
|
||
extern char _etext;
|
||
extern char _data;
|
||
extern char _edata;
|
||
extern char _rodata;
|
||
extern char _erodata;
|
||
extern char _bss;
|
||
extern char _ebss;
|
||
extern char _end;
|
||
|
||
// 每个区域的索引
|
||
|
||
int ZONE_DMA_INDEX = 0;
|
||
int ZONE_NORMAL_INDEX = 0;
|
||
int ZONE_UNMAPPED_INDEX = 0;
|
||
|
||
// 初始化内存管理单元
|
||
void mm_init();
|
||
|
||
/**
|
||
* @brief 初始化内存页
|
||
*
|
||
* @param page 内存页结构体
|
||
* @param flags 标志位
|
||
* 本函数只负责初始化内存页，允许对同一页面进行多次初始化
|
||
* 而维护计数器及置位bmp标志位的功能，应当在分配页面的时候手动完成
|
||
* @return unsigned long
|
||
*/
|
||
unsigned long page_init(struct Page *page, ul flags);
|
||
|
||
/**
|
||
* @brief 读取CR3寄存器的值（存储了页目录的基地址）
|
||
*
|
||
* @return unsigned* cr3的值的指针
|
||
*/
|
||
unsigned long *get_CR3()
|
||
{
|
||
ul *tmp;
|
||
__asm__ __volatile__(
|
||
"movq %%cr3, %0\n\t"
|
||
: "=r"(tmp)::"memory");
|
||
return tmp;
|
||
}
|
||
|
||
/**
|
||
* @brief 从已初始化的页结构中搜索符合申请条件的、连续num个struct page
|
||
*
|
||
* @param zone_select 选择内存区域, 可选项：dma, mapped in pgt(normal), unmapped in pgt
|
||
* @param num 需要申请的内存页的数量 num<64
|
||
* @param flags 将页面属性设置成flag
|
||
* @return struct Page*
|
||
*/
|
||
struct Page *alloc_pages(unsigned int zone_select, int num, ul flags);
|
||
|
||
/**
|
||
* @brief 清除页面的引用计数， 计数为0时清空除页表已映射以外的所有属性
|
||
*
|
||
* @param p 物理页结构体
|
||
* @return unsigned long
|
||
*/
|
||
unsigned long page_clean(struct Page *page);
|
||
|
||
/**
|
||
* @brief 释放连续number个内存页
|
||
*
|
||
* @param page 第一个要被释放的页面的结构体
|
||
* @param number 要释放的内存页数量 number<64
|
||
*/
|
||
void free_pages(struct Page *page, int number);
|
||
|
||
/**
|
||
* @brief Get the page's attr
|
||
*
|
||
* @param page 内存页结构体
|
||
* @return ul 属性
|
||
*/
|
||
ul get_page_attr(struct Page *page);
|
||
|
||
/**
|
||
* @brief Set the page's attr
|
||
*
|
||
* @param page 内存页结构体
|
||
* @param flags 属性
|
||
* @return ul 错误码
|
||
*/
|
||
ul set_page_attr(struct Page *page, ul flags);
|
||
|
||
#define mk_pml4t(addr, attr) ((unsigned long)(addr) | (unsigned long)(attr))
|
||
/**
|
||
* @brief 设置pml4页表的页表项
|
||
* @param pml4tptr pml4页表项的地址
|
||
* @param pml4val pml4页表项的值
|
||
*/
|
||
#define set_pml4t(pml4tptr, pml4tval) (*(pml4tptr) = (pml4tval))
|
||
|
||
#define mk_pdpt(addr, attr) ((unsigned long)(addr) | (unsigned long)(attr))
|
||
#define set_pdpt(pdptptr, pdptval) (*(pdptptr) = (pdptval))
|
||
|
||
#define mk_pdt(addr, attr) ((unsigned long)(addr) | (unsigned long)(attr))
|
||
#define set_pdt(pdtptr, pdtval) (*(pdtptr) = (pdtval))
|
||
|
||
#define mk_pt(addr, attr) ((unsigned long)(addr) | (unsigned long)(attr))
|
||
#define set_pt(ptptr, ptval) (*(ptptr) = (ptval))
|
||
|
||
/*
|
||
* vm_area_struct中的vm_flags的可选值
|
||
* 对应的结构体请见mm-types.h
|
||
*/
|
||
#define VM_NONE 0
|
||
#define VM_READ (1 << 0)
|
||
#define VM_WRITE (1 << 1)
|
||
#define VM_EXEC (1 << 2)
|
||
#define VM_SHARED (1 << 3)
|
||
#define VM_IO (1 << 4) // MMIO的内存区域
|
||
#define VM_SOFTDIRTY (1 << 5)
|
||
#define VM_MAYSHARE (1 << 6) // 该vma可被共享
|
||
#define VM_USER (1 << 7) // 该vma可被用户态访问
|
||
#define VM_DONTCOPY (1 << 8) // 当fork的时候不拷贝该虚拟内存区域
|
||
|
||
/* VMA basic access permission flags */
|
||
#define VM_ACCESS_FLAGS (VM_READ | VM_WRITE | VM_EXEC)
|
||
|
||
/**
|
||
* @brief 初始化虚拟内存区域结构体
|
||
*
|
||
* @param vma
|
||
* @param mm
|
||
*/
|
||
static inline void vma_init(struct vm_area_struct *vma, struct mm_struct *mm)
|
||
{
|
||
memset(vma, 0, sizeof(struct vm_area_struct));
|
||
vma->vm_mm = mm;
|
||
vma->vm_prev = vma->vm_next = NULL;
|
||
vma->vm_ops = NULL;
|
||
list_init(&vma->anon_vma_list);
|
||
}
|
||
|
||
/**
|
||
* @brief 判断给定的vma是否为当前进程所属的vma
|
||
*
|
||
* @param vma 给定的vma结构体
|
||
* @return true
|
||
* @return false
|
||
*/
|
||
static inline bool vma_is_foreign(struct vm_area_struct *vma)
|
||
{
|
||
if (current_pcb->mm == NULL)
|
||
return true;
|
||
if (current_pcb->mm != vma->vm_mm)
|
||
return true;
|
||
return false;
|
||
}
|
||
|
||
static inline bool vma_is_accessible(struct vm_area_struct *vma)
|
||
{
|
||
return vma->vm_flags & VM_ACCESS_FLAGS;
|
||
}
|
||
|
||
/**
|
||
* @brief 获取一块新的vma结构体，并将其与指定的mm进行绑定
|
||
*
|
||
* @param mm 与VMA绑定的内存空间分布结构体
|
||
* @return struct vm_area_struct* 新的VMA
|
||
*/
|
||
struct vm_area_struct *vm_area_alloc(struct mm_struct *mm);
|
||
|
||
/**
|
||
* @brief 释放vma结构体
|
||
*
|
||
* @param vma 待释放的vma结构体
|
||
*/
|
||
void vm_area_free(struct vm_area_struct *vma);
|
||
|
||
/**
|
||
* @brief 从链表中删除指定的vma结构体
|
||
*
|
||
* @param vma
|
||
*/
|
||
void vm_area_del(struct vm_area_struct *vma);
|
||
|
||
/**
|
||
* @brief 查找第一个符合“addr < vm_end”条件的vma
|
||
*
|
||
* @param mm 内存空间分布结构体
|
||
* @param addr 虚拟地址
|
||
* @return struct vm_area_struct* 符合条件的vma
|
||
*/
|
||
struct vm_area_struct *vma_find(struct mm_struct *mm, uint64_t addr);
|
||
|
||
/**
|
||
* @brief 插入vma
|
||
*
|
||
* @param mm
|
||
* @param vma
|
||
* @return int
|
||
*/
|
||
int vma_insert(struct mm_struct *mm, struct vm_area_struct *vma);
|
||
|
||
/**
|
||
* @brief 重新初始化页表的函数
|
||
* 将所有物理页映射到线性地址空间
|
||
*/
|
||
void page_table_init();
|
||
|
||
/**
|
||
* @brief 将物理地址映射到页表的函数
|
||
*
|
||
* @param virt_addr_start 要映射到的虚拟地址的起始位置
|
||
* @param phys_addr_start 物理地址的起始位置
|
||
* @param length 要映射的区域的长度（字节）
|
||
* @param flags 标志位
|
||
* @param use4k 是否使用4k页
|
||
*/
|
||
int mm_map_phys_addr(ul virt_addr_start, ul phys_addr_start, ul length, ul flags, bool use4k);
|
||
|
||
/**
|
||
* @brief 将将物理地址填写到进程的页表的函数
|
||
*
|
||
* @param proc_page_table_addr 页表的基地址
|
||
* @param is_phys 页表的基地址是否为物理地址
|
||
* @param virt_addr_start 要映射到的虚拟地址的起始位置
|
||
* @param phys_addr_start 物理地址的起始位置
|
||
* @param length 要映射的区域的长度（字节）
|
||
* @param user 用户态是否可访问
|
||
* @param flush 是否刷新tlb
|
||
* @param use4k 是否使用4k页
|
||
*/
|
||
int mm_map_proc_page_table(ul proc_page_table_addr, bool is_phys, ul virt_addr_start, ul phys_addr_start, ul length, ul flags, bool user, bool flush, bool use4k);
|
||
|
||
int mm_map_phys_addr_user(ul virt_addr_start, ul phys_addr_start, ul length, ul flags);
|
||
|
||
/**
|
||
* @brief 从页表中清除虚拟地址的映射
|
||
*
|
||
* @param proc_page_table_addr 页表的地址
|
||
* @param is_phys 页表地址是否为物理地址
|
||
* @param virt_addr_start 要清除的虚拟地址的起始地址
|
||
* @param length 要清除的区域的长度
|
||
*/
|
||
void mm_unmap_proc_table(ul proc_page_table_addr, bool is_phys, ul virt_addr_start, ul length);
|
||
|
||
/**
|
||
* @brief 取消当前进程的页表中的虚拟地址映射
|
||
*
|
||
* @param virt_addr 虚拟地址
|
||
* @param length 地址长度
|
||
*/
|
||
#define mm_unmap(virt_addr, length) ({ \
|
||
mm_unmap_proc_table((uint64_t)get_CR3(), true, virt_addr, length); \
|
||
})
|
||
|
||
/**
|
||
* @brief 创建VMA
|
||
*
|
||
* @param mm 要绑定的内存空间分布结构体
|
||
* @param vaddr 起始虚拟地址
|
||
* @param length 长度（字节）
|
||
* @param vm_flags vma的标志
|
||
* @param vm_ops vma的操作接口
|
||
* @param res_vma 返回的vma指针
|
||
* @return int 错误码
|
||
*/
|
||
int mm_create_vma(struct mm_struct *mm, uint64_t vaddr, uint64_t length, vm_flags_t vm_flags, struct vm_operations_t *vm_ops, struct vm_area_struct **res_vma);
|
||
|
||
/**
|
||
* @brief 将指定的物理地址映射到指定的vma处
|
||
*
|
||
* @param vma 要进行映射的VMA结构体
|
||
* @param paddr 起始物理地址
|
||
* @return int 错误码
|
||
*/
|
||
int mm_map_vma(struct vm_area_struct *vma, uint64_t paddr);
|
||
|
||
/**
|
||
* @brief 在页表中取消指定的vma的映射
|
||
*
|
||
* @param mm 指定的mm
|
||
* @param vma 待取消映射的vma
|
||
* @param paddr 返回的被取消映射的起始物理地址
|
||
* @return int 返回码
|
||
*/
|
||
int mm_umap_vma(struct mm_struct *mm, struct vm_area_struct *vma, uint64_t *paddr);
|
||
|
||
/**
|
||
* @brief 检测是否为有效的2M页(物理内存页)
|
||
*
|
||
* @param paddr 物理地址
|
||
* @return int8_t 是 -> 1
|
||
* 否 -> 0
|
||
*/
|
||
int8_t mm_is_2M_page(uint64_t paddr);
|
||
|
||
/**
|
||
* @brief 检查页表是否存在不为0的页表项
|
||
*
|
||
* @param ptr 页表基指针
|
||
* @return int8_t 存在 -> 1
|
||
* 不存在 -> 0
|
||
*/
|
||
int8_t mm_check_page_table(uint64_t *ptr);
|
||
|
||
/**
|
||
* @brief 调整堆区域的大小（暂时只能增加堆区域）
|
||
*
|
||
* @todo 缩小堆区域
|
||
* @param old_brk_end_addr 原本的堆内存区域的结束地址
|
||
* @param offset 新的地址相对于原地址的偏移量
|
||
* @return uint64_t
|
||
*/
|
||
uint64_t mm_do_brk(uint64_t old_brk_end_addr, int64_t offset);
|
||
|
||
/**
|
||
* @brief 获取系统当前的内存信息(未上锁，不一定精准)
|
||
*
|
||
* @return struct mm_stat_t 内存信息结构体
|
||
*/
|
||
struct mm_stat_t mm_stat();
|
||
|
||
/**
|
||
* @brief 检测指定地址是否已经被映射
|
||
*
|
||
* @param page_table_phys_addr 页表的物理地址
|
||
* @param virt_addr 要检测的地址
|
||
* @return true 已经被映射
|
||
* @return false
|
||
*/
|
||
bool mm_check_mapped(ul page_table_phys_addr, uint64_t virt_addr); |