1 #include "internal.h"
2 
3 extern uint64_t mm_total_2M_pages;
4 
5 /**
6  * @brief 获取指定虚拟地址处映射的物理地址
7  *
8  * @param mm 内存空间分布结构体
9  * @param vaddr 虚拟地址
10  * @return uint64_t 已映射的物理地址
11  */
__mm_get_paddr(struct mm_struct * mm,uint64_t vaddr)12 uint64_t __mm_get_paddr(struct mm_struct *mm, uint64_t vaddr)
13 {
14     ul *tmp;
15 
16     tmp = phys_2_virt((ul *)(((ul)mm->pgd) & (~0xfffUL)) + ((vaddr >> PAGE_GDT_SHIFT) & 0x1ff));
17 
18     // pml4页表项为0
19     if (*tmp == 0)
20         return 0;
21 
22     tmp = phys_2_virt((ul *)(*tmp & (~0xfffUL)) + ((vaddr >> PAGE_1G_SHIFT) & 0x1ff));
23 
24     // pdpt页表项为0
25     if (*tmp == 0)
26         return 0;
27 
28     // 读取pdt页表项
29     tmp = phys_2_virt(((ul *)(*tmp & (~0xfffUL)) + (((ul)(vaddr) >> PAGE_2M_SHIFT) & 0x1ff)));
30 
31     // pde页表项为0
32     if (*tmp == 0)
33         return 0;
34 
35     if (*tmp & (1 << 7))
36     {
37         // 当前为2M物理页
38         return (*tmp) & (~0x1fffUL);
39     }
40     else
41     {
42         // 存在4级页表
43         tmp = phys_2_virt(((ul *)(*tmp & (~0xfffUL)) + (((ul)(vaddr) >> PAGE_4K_SHIFT) & 0x1ff)));
44 
45         return (*tmp) & (~0x1ffUL);
46     }
47 }
48 
49 /**
50  * @brief 检测指定地址是否已经被映射
51  *
52  * @param page_table_phys_addr 页表的物理地址
53  * @param virt_addr 要检测的地址
54  * @return true 已经被映射
55  * @return false
56  */
mm_check_mapped(ul page_table_phys_addr,uint64_t virt_addr)57 bool mm_check_mapped(ul page_table_phys_addr, uint64_t virt_addr)
58 {
59     ul *tmp;
60 
61     tmp = phys_2_virt((ul *)((ul)page_table_phys_addr & (~0xfffUL)) + ((virt_addr >> PAGE_GDT_SHIFT) & 0x1ff));
62 
63     // pml4页表项为0
64     if (*tmp == 0)
65         return 0;
66 
67     tmp = phys_2_virt((ul *)(*tmp & (~0xfffUL)) + ((virt_addr >> PAGE_1G_SHIFT) & 0x1ff));
68 
69     // pdpt页表项为0
70     if (*tmp == 0)
71         return 0;
72 
73     // 读取pdt页表项
74     tmp = phys_2_virt(((ul *)(*tmp & (~0xfffUL)) + (((ul)(virt_addr) >> PAGE_2M_SHIFT) & 0x1ff)));
75 
76     // pde页表项为0
77     if (*tmp == 0)
78         return 0;
79 
80     if (*tmp & (1 << 7))
81     {
82         // 当前为2M物理页
83         return true;
84     }
85     else
86     {
87         // 存在4级页表
88         tmp = phys_2_virt(((ul *)(*tmp & (~0xfffUL)) + (((ul)(virt_addr) >> PAGE_4K_SHIFT) & 0x1ff)));
89         if (*tmp != 0)
90             return true;
91         else
92             return false;
93     }
94 }
95 
96 /**
97  * @brief 检测是否为有效的2M页(物理内存页)
98  *
99  * @param paddr 物理地址
100  * @return int8_t 是 -> 1
101  *                 否 -> 0
102  */
mm_is_2M_page(uint64_t paddr)103 int8_t mm_is_2M_page(uint64_t paddr)
104 {
105     if (likely((paddr >> PAGE_2M_SHIFT) < mm_total_2M_pages))
106         return 1;
107     else
108         return 0;
109 }
110