EXER2: 实现寻找虚拟地址对应的页表项*

//get_pte - 通过 la 获取页表项，如果不存在，就创建一个
// parameter:
//  pgdir:  PDT基址，代表二级页表 pde_t表示页目录表项 pte_t表示页表项
//  la:     要映射的地址
//  create: 是否分配一个 Page
// return vaule: 页表项的虚地址
pte_t *
get_pte(pde_t *pgdir, uintptr_t la, bool create) {
    /* LAB2 EXERCISE 2: YOUR CODE
     *
     * Some Useful MACROs and DEFINEs, you can use them in below implementation.
     * MACROs or Functions:
     *   PDX(la) = la 中的页目录表索引，即高 10 位
     *   KADDR(pa) : 物理地址对应的内核虚地址
     *   set_page_ref(page,1) : 设置页被引用一次
     *   page2pa(page): 获取 page 对应的物理地址
     *   struct Page * alloc_page() : 分配一个页
     *   memset(void *s, char c, size_t n) : 设置内存
     *
     * DEFINEs:
     *   PTE_P           0x001                   // 物理页存在
     *   PTE_W           0x002                   // 物理页可写
     *   PTE_U           0x004                   // 用户态可读物理页内容
     */
#if 0
    pde_t *pdep = NULL;   // (1) find page directory entry
    if (0) {              // (2) check if entry is not present
                          // (3) check if creating is needed, then alloc page for page table
                          // CAUTION: this page is used for page table, not for common data page
                          // (4) set page reference
        uintptr_t pa = 0; // (5) get linear address of page
                          // (6) clear page content using memset
                          // (7) set page directory entry's permission
    }
    return NULL;          // (8) return page table entry
#endif
    pde_t *pdep = &pgdir[PDX(la)];                          // 获取页目录项
    if (!(*pdep & PTE_P)) {                                 // 判断页表项是否存在
        struct Page *page;
        if (!create || (page = alloc_page()) == NULL) {     // 不需要分配或分配失败
            return NULL;
        }
        set_page_ref(page, 1);                              // 设置引用次数为 1
        uintptr_t pa = page2pa(page);                       // 得到该页的物理地址
        memset(KADDR(pa), 0, PGSIZE);                       // 初始化，清零
        *pdep = pa & ~0x0FFF | PTE_P | PTE_W | PTE_U;       // 设置页目录表项，可读，可写，存在
    }
    // KADDR(PDE_ADDR(*pdep)) 由页目录项地址得到关联的页表物理地址，再转成虚拟地址
    // PTX(la) 页表项的索引地址，即中 10 位
    // 最后返回的是虚拟地址la对应的页表项入口地址
    return &((pte_t *)KADDR(PDE_ADDR(*pdep)))[PTX(la)];
}