手动设置特定PTE的NX位

手动设置特定PTE的NX位

在内核为 4.16.7 的 Ubuntu 上,我正在编写一个自定义系统调用,并且我想设置特定页表条目的 NX 位。到目前为止,我有这段代码,我正在执行页表遍历以获得我想要的 PTE,然后尝试设置其 NX 位:

pgd = pgd_offset(mm, addr);
if (pgd_none(*pgd) || pgd_bad(*pgd)){
    printk("Invalid pgd");
    return -1;
}

p4d = p4d_offset(pgd, addr);
if (p4d_none(*p4d) || p4d_bad(*p4d)){
    printk("Invalid p4d");
    return -1;
}

pud = pud_offset(p4d, addr);
if (pud_none(*pud) || pud_bad(*pud)){
    printk("Invalid pud");
    return -1;
}

pmd = pmd_offset(pud, addr);
if (pmd_none(*pmd) || pmd_bad(*pmd)){
    printk("Invalid pmd");
    return -1;
}

ptep = pte_offset_map(pmd, addr);
if (!ptep){
    printk("Invalid ptep");
    return -1;
}
pte = *ptep;

if (pte_present(pte)){
    printk("pte_set_flags");
    printk("NX bit before:  %d", pte_exec(pte));
    // pte_set_flags(pte, _PAGE_NX);
    // printk("NX bit after :  %d", pte_exec(pte));
    printk("pte_clear_flags");
    // pte_clear_flags(pte, _PAGE_NX); // Same as pte_mkexec()
    pte_mkexec(pte);
    printk("NX bit after :  %d", pte_exec(pte));
    page = pte_page(pte);
    if (page){
        printk("Page frame struct is @ %p", page);
    }

    pte_unmap(ptep);
}

但它不起作用。所有printk命令显示相同的结果。有什么见解吗?

答案1

于是我就自己找到了。代码如下:

struct vm_area_struct *vma;
unsigned long oldflags, newflags, pfn;

vma = find_extend_vma(mm, addr);
oldflags = vma->vm_flags;
newflags = oldflags &= ~VM_EXEC;

//...
//...
//...

if (pte_present(pte)){
    printk("NX bit before: %d", pte_exec(pte));

    pte = pte_modify(pte, vm_get_page_prot(newflags));
    printk("NX bit after:  %d", pte_exec(pte));

    pfn = pte_pfn(pte);

    flush_cache_page(vma, addr, pfn);
    set_pte(ptep, pte);
    flush_tlb_page(vma, addr);
    update_mmu_cache(vma, addr, ptep);

    pte_unmap(ptep);
}

从而改变特定PTE的NX位。

相关内容