2020-07-23 11:50:23 -05:00
|
|
|
#include "arch_consts.h"
|
2019-04-30 17:11:26 -05:00
|
|
|
#include "paging.h"
|
|
|
|
#include "pmem.h"
|
2020-07-22 19:35:23 -05:00
|
|
|
#include <stdint.h>
|
2019-04-30 17:11:26 -05:00
|
|
|
|
|
|
|
static uint64_t pml4[512] __attribute__((aligned(4096)));
|
|
|
|
static uint64_t pdpt[512] __attribute__((aligned(4096)));
|
|
|
|
static uint64_t page_directory[512] __attribute__((aligned(4096)));
|
|
|
|
uint64_t* curr_structs=(uint64_t*)0xFFFFFF8000000000;
|
|
|
|
|
|
|
|
|
|
|
|
uint64_t get_pml4_entry(int entry) {
|
|
|
|
return ((uint64_t*)(((char*)curr_structs)+0x7FFFFFF000))[entry];
|
|
|
|
}
|
|
|
|
|
|
|
|
uint64_t get_pdpt_entry(int pml4_entry,int entry) {
|
|
|
|
return ((uint64_t*)(((char*)curr_structs)+0x7FFFE00000))[(pml4_entry<<9)+entry];
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
uint64_t get_page_dir_entry(int pml4_entry,int pdpt_entry, int entry) {
|
|
|
|
return ((uint64_t*)(((char*)curr_structs)+0x7FC0000000))[(pml4_entry<<18)+(pdpt_entry<<9)+entry];
|
|
|
|
}
|
|
|
|
|
|
|
|
uint64_t get_page_table_entry(int pml4_entry,int pdpt_entry, int pdir_entry, int entry) {
|
|
|
|
return ((uint64_t*)(((char*)curr_structs)+0x7000000000))[(pml4_entry<<27)+(pdpt_entry<<18)+(pdir_entry<<9)+entry];
|
|
|
|
}
|
|
|
|
|
|
|
|
void set_pml4_entry(int entry,uint64_t mapping) {
|
|
|
|
((uint64_t*)(((char*)curr_structs)+0x7FFFFFF000))[entry]=mapping;
|
|
|
|
}
|
|
|
|
|
|
|
|
void set_pdpt_entry(int pml4_entry,int entry,uint64_t mapping) {
|
|
|
|
((uint64_t*)(((char*)curr_structs)+0x7FFFE00000))[(pml4_entry<<9)+entry]=mapping;
|
|
|
|
}
|
|
|
|
|
|
|
|
void set_page_dir_entry(int pml4_entry,int pdpt_entry, int entry,uint64_t mapping) {
|
|
|
|
((uint64_t*)(((char*)curr_structs)+0x7FC0000000))[(pml4_entry<<18)+(pdpt_entry<<9)+entry]=mapping;
|
|
|
|
}
|
|
|
|
|
|
|
|
void set_page_table_entry(int pml4_entry,int pdpt_entry, int pdir_entry, int entry,uint64_t mapping) {
|
|
|
|
((uint64_t*)(((char*)curr_structs)+0x7000000000))[(pml4_entry<<27)+(pdpt_entry<<18)+(pdir_entry<<9)+entry]=mapping;
|
|
|
|
}
|
|
|
|
|
|
|
|
uint32_t find_free_page() {
|
|
|
|
for (int a=0;a<512;a++) {
|
|
|
|
if (get_pml4_entry(a)&0x1) {
|
|
|
|
for (int b=0;b<512;b++) {
|
|
|
|
if (get_pdpt_entry(a,b)&0x1) {
|
|
|
|
for (int c=0;c<512;c++) {
|
|
|
|
if (get_page_dir_entry(a,b,c)&0x1) {
|
|
|
|
for (int d=0;d<512;d++) {
|
|
|
|
if (!(get_page_table_entry(a,b,c,d)&0x1)) {
|
|
|
|
return (a<<27)+(b<<18)+(c<<9)+d;
|
|
|
|
}
|
|
|
|
} // end for d
|
|
|
|
} else {
|
|
|
|
uint64_t addr=pmem_alloc(1);
|
|
|
|
set_page_dir_entry(a,b,c,addr|0x7);
|
|
|
|
return (a<<27)+(b<<18)+(c<<9);
|
|
|
|
} // end if abc
|
|
|
|
} // end for c
|
|
|
|
} else {
|
|
|
|
uint64_t addr=pmem_alloc(1);
|
|
|
|
set_pdpt_entry(a,b,addr|0x7);
|
|
|
|
addr=pmem_alloc(1);
|
|
|
|
set_page_dir_entry(a,b,0,addr|0x7);
|
|
|
|
return (a<<27)+(b<<18);
|
|
|
|
} // end if ab
|
|
|
|
} // end for b
|
|
|
|
} else {
|
|
|
|
uint64_t addr=pmem_alloc(1);
|
|
|
|
set_pml4_entry(a,addr|0x7);
|
|
|
|
addr=pmem_alloc(1);
|
|
|
|
set_pdpt_entry(a,0,addr|0x7);
|
|
|
|
addr=pmem_alloc(1);
|
|
|
|
set_page_dir_entry(a,0,0,addr|0x7);
|
|
|
|
return a<<27;
|
|
|
|
} //end if a
|
|
|
|
} // end for a
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
void paging_init() {
|
|
|
|
pml4[0]=(((uint64_t)&pdpt)-0xffff800000000000)|0x7;
|
|
|
|
pml4[256]=(((uint64_t)&pdpt)-0xffff800000000000)|0x7;
|
|
|
|
pml4[511]=(((uint64_t)&pml4)-0xffff800000000000)|0x7;
|
|
|
|
pdpt[0]=(((uint64_t)&page_directory)-0xffff800000000000)|0x7;
|
2020-07-23 09:16:41 -05:00
|
|
|
for (int i=0;i<NUM_KERN_FRAMES/1024;i++) {
|
2019-04-30 17:11:26 -05:00
|
|
|
page_directory[i]=(i*0x200000)|0x87;
|
|
|
|
}
|
|
|
|
asm volatile("mov %0,%%cr3"::"r"((uint64_t)pml4-0xffff800000000000));
|
|
|
|
}
|
|
|
|
|
|
|
|
void check_gets() {
|
|
|
|
char str[256];
|
|
|
|
hex_to_ascii(find_free_page(),str);
|
|
|
|
vga_write_string(str);
|
|
|
|
vga_write_string("\n");
|
|
|
|
str[0]='\0';
|
|
|
|
hex_to_ascii(find_free_page(),str);
|
|
|
|
vga_write_string(str);
|
|
|
|
vga_write_string("\n");
|
|
|
|
}
|