From c838637132f93fb8b9d436633b3a82d55912946c Mon Sep 17 00:00:00 2001 From: Ada Christine Date: Tue, 30 Mar 2021 19:03:59 +0000 Subject: better page table entry masks --- loader/loader_efi.c | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'loader') diff --git a/loader/loader_efi.c b/loader/loader_efi.c index 63bbaba..6823563 100644 --- a/loader/loader_efi.c +++ b/loader/loader_efi.c @@ -34,9 +34,9 @@ static void *get_acpi_rsdp(void); enum page_type { INVALID_PAGE_TYPE = 0x0, - CODE_PAGE_TYPE = 0x1, - RODATA_PAGE_TYPE = 0x1, - DATA_PAGE_TYPE = 0x3 + CODE_PAGE_TYPE = PAGE_PR, + RODATA_PAGE_TYPE = PAGE_PR|PAGE_NX, + DATA_PAGE_TYPE = PAGE_PR|PAGE_WR|PAGE_NX }; static uint64_t *new_page_table(void) @@ -64,7 +64,7 @@ static uint64_t *get_page_map(void) { uint64_t map; __asm__("mov %%cr3, %0" : "=r"(map)); - return (uint64_t *)(map & PAGE_TABLE_ADDRESS_MASK); + return (uint64_t *)(map & PAGE_ADDRESS_MASK); } void set_page_map(uint64_t *map) @@ -92,7 +92,7 @@ static uint64_t *get_page_table(uint64_t *map, uint64_t virt, int level) else { uint64_t next_map_entry = map[page_table_index(virt, i)]; - next_map_entry &= PAGE_TABLE_ADDRESS_MASK; + next_map_entry &= PAGE_ADDRESS_MASK; map = (uint64_t *)next_map_entry; } } -- cgit v1.3.1-1-g115d