|
1 | | -// paging.c |
| 1 | +// [file name]: paging.c |
2 | 2 | #include "paging.h" |
3 | 3 | #include "../kklibc/kklibc.h" |
4 | | -#include "isr.h" |
5 | 4 | #include "../drivers/screen.h" |
6 | 5 |
|
7 | 6 | // Текущий каталог страниц |
8 | | -page_directory_t *current_directory = NULL; |
| 7 | +static page_directory_t *current_directory = NULL; |
| 8 | +// Каталог страниц ядра |
| 9 | +static page_directory_t *kernel_directory = NULL; |
9 | 10 |
|
10 | | -// Внешние функции из assembly |
11 | | -extern void load_page_directory(u32); |
12 | | -extern void enable_paging(); |
| 11 | +// Статистика использования памяти |
| 12 | +static u32 used_pages = 0; |
| 13 | +static u32 total_pages = 0; |
| 14 | + |
| 15 | +// Временная таблица страниц для идентичного отображения |
| 16 | +static page_table_t *identity_table = NULL; |
| 17 | + |
| 18 | +#define PAGING_STRUCTS_START 0x200000 |
| 19 | +#define KERNEL_PAGE_DIRECTORY_ADDR (PAGING_STRUCTS_START) |
| 20 | +#define IDENTITY_PAGE_TABLE_ADDR (PAGING_STRUCTS_START + 0x1000) |
13 | 21 |
|
14 | | -// Инициализация paging |
15 | 22 | void paging_init() { |
16 | 23 | kprint("Initializing paging...\n"); |
17 | 24 |
|
18 | | - // Создаем каталог страниц |
19 | | - current_directory = (page_directory_t*)kmalloc_a(sizeof(page_directory_t)); |
20 | | - memory_set((u8*)current_directory, 0, sizeof(page_directory_t)); |
21 | | - |
22 | | - // Создаем таблицы страниц для идентичного отображения первых 4MB |
23 | | - for (u32 i = 0; i < 1024; i++) { |
24 | | - // Каждая запись каталога указывает на таблицу страниц |
25 | | - page_table_t *table = (page_table_t*)kmalloc_a(sizeof(page_table_t)); |
26 | | - memory_set((u8*)table, 0, sizeof(page_table_t)); |
27 | | - |
28 | | - // Заполняем таблицу страниц |
29 | | - for (u32 j = 0; j < 1024; j++) { |
30 | | - u32 frame = (i * 1024 + j) * PAGE_SIZE; |
31 | | - table->entries[j].present = 1; |
32 | | - table->entries[j].rw = 1; |
33 | | - table->entries[j].frame = frame >> 12; |
34 | | - } |
| 25 | + // Используем фиксированные адреса для структур paging |
| 26 | + kernel_directory = (page_directory_t *)KERNEL_PAGE_DIRECTORY_ADDR; |
| 27 | + memory_set((u8 *)kernel_directory, 0, sizeof(page_directory_t)); |
35 | 28 |
|
36 | | - // Устанавливаем запись в каталоге страниц |
37 | | - current_directory->entries[i].present = 1; |
38 | | - current_directory->entries[i].rw = 1; |
39 | | - current_directory->entries[i].frame = ((u32)table) >> 12; |
| 29 | + // Создаем таблицу для идентичного отображения первых 4MB |
| 30 | + page_table_t *identity_table = (page_table_t *)IDENTITY_PAGE_TABLE_ADDR; |
| 31 | + memory_set((u8 *)identity_table, 0, sizeof(page_table_t)); |
| 32 | + |
| 33 | + // Заполняем таблицу идентичного отображения |
| 34 | + u32 i; |
| 35 | + for (i = 0; i < 1024; i++) { |
| 36 | + identity_table->entries[i].present = 1; |
| 37 | + identity_table->entries[i].rw = 1; |
| 38 | + identity_table->entries[i].frame = i; |
40 | 39 | } |
41 | 40 |
|
42 | | - // Переключаемся на новый каталог страниц |
43 | | - switch_page_directory(current_directory); |
| 41 | + // Добавляем таблицу в каталог |
| 42 | + kernel_directory->entries[0].present = 1; |
| 43 | + kernel_directory->entries[0].rw = 1; |
| 44 | + kernel_directory->entries[0].frame = (u32)identity_table >> 12; |
| 45 | + |
| 46 | + // Устанавливаем каталог страниц |
| 47 | + current_directory = kernel_directory; |
| 48 | + switch_page_directory(kernel_directory); |
44 | 49 |
|
45 | 50 | // Включаем paging |
46 | | - enable_paging(); |
| 51 | + u32 cr0; |
| 52 | + asm volatile("mov %%cr0, %0" : "=r"(cr0)); |
| 53 | + cr0 |= 0x80000000; // Устанавливаем бит PG |
| 54 | + asm volatile("mov %0, %%cr0" :: "r"(cr0)); |
47 | 55 |
|
48 | 56 | kprint("Paging enabled\n"); |
| 57 | + |
| 58 | + // Инициализируем статистику |
| 59 | + total_pages = (HEAP_SIZE + HEAP_START) / PAGE_SIZE; |
| 60 | + used_pages = 1024; // Уже использовано для идентичного отображения |
49 | 61 | } |
50 | 62 |
|
51 | | -// Переключение каталога страниц |
52 | | -void switch_page_directory(page_directory_t *dir) { |
53 | | - current_directory = dir; |
54 | | - load_page_directory((u32)dir); |
| 63 | +void switch_page_directory(page_directory_t *new_dir) { |
| 64 | + current_directory = new_dir; |
| 65 | + asm volatile("mov %0, %%cr3" :: "r"(new_dir)); |
55 | 66 | } |
56 | 67 |
|
57 | | -// Получение текущего каталога страниц |
58 | | -page_directory_t *get_current_page_directory() { |
59 | | - return current_directory; |
| 68 | +page_directory_t *get_kernel_page_directory() { |
| 69 | + return kernel_directory; |
60 | 70 | } |
61 | 71 |
|
62 | | -// Отображение виртуальной страницы на физическую |
63 | 72 | void map_page(void *virtual_addr, void *physical_addr, u32 flags) { |
64 | | - u32 virtual = (u32)virtual_addr; |
65 | | - u32 physical = (u32)physical_addr; |
| 73 | + u32 pd_index = PAGE_DIRECTORY_INDEX((u32)virtual_addr); |
| 74 | + u32 pt_index = PAGE_TABLE_INDEX((u32)virtual_addr); |
66 | 75 |
|
67 | | - u32 pd_index = PAGE_DIRECTORY_INDEX(virtual); |
68 | | - u32 pt_index = PAGE_TABLE_INDEX(virtual); |
| 76 | + // Проверяем, существует ли таблица страниц |
| 77 | + if (!current_directory->entries[pd_index].present) { |
| 78 | + // Создаем новую таблицу страниц |
| 79 | + page_table_t *new_table = (page_table_t *)kmalloc_a(sizeof(page_table_t)); |
| 80 | + memory_set((u8 *)new_table, 0, sizeof(page_table_t)); |
| 81 | + |
| 82 | + // Добавляем таблицу в каталог |
| 83 | + current_directory->entries[pd_index].present = 1; |
| 84 | + current_directory->entries[pd_index].rw = 1; |
| 85 | + current_directory->entries[pd_index].frame = (u32)new_table >> 12; |
| 86 | + } |
69 | 87 |
|
70 | | - // Получаем таблицу страниц |
71 | | - page_table_t *table = (page_table_t*)(current_directory->entries[pd_index].frame << 12); |
| 88 | + // Получаем указатель на таблицу страниц |
| 89 | + page_table_t *table = (page_table_t *)(current_directory->entries[pd_index].frame << 12); |
72 | 90 |
|
73 | | - // Устанавливаем запись в таблице страниц |
| 91 | + // Заполняем запись в таблице страниц |
74 | 92 | table->entries[pt_index].present = (flags & PAGE_PRESENT) ? 1 : 0; |
75 | | - table->entries[pt_index].rw = (flags & PAGE_WRITABLE) ? 1 : 0; |
| 93 | + table->entries[pt_index].rw = (flags & PAGE_WRITE) ? 1 : 0; |
76 | 94 | table->entries[pt_index].user = (flags & PAGE_USER) ? 1 : 0; |
77 | | - table->entries[pt_index].frame = physical >> 12; |
| 95 | + table->entries[pt_index].write_through = (flags & PAGE_WRITE_THROUGH) ? 1 : 0; |
| 96 | + table->entries[pt_index].cache_disable = (flags & PAGE_CACHE_DISABLE) ? 1 : 0; |
| 97 | + table->entries[pt_index].global = (flags & PAGE_GLOBAL) ? 1 : 0; |
| 98 | + table->entries[pt_index].frame = (u32)physical_addr >> 12; |
| 99 | + |
| 100 | + // Обновляем статистику |
| 101 | + if (table->entries[pt_index].present) { |
| 102 | + used_pages++; |
| 103 | + } |
78 | 104 |
|
79 | | - // Инвалидируем TLB |
80 | | - asm volatile("invlpg (%0)" : : "r" (virtual_addr)); |
| 105 | + // Принудительно обновляем TLB |
| 106 | + asm volatile("invlpg (%0)" :: "r"(virtual_addr)); |
81 | 107 | } |
82 | 108 |
|
83 | | -// Удаление отображения страницы |
84 | 109 | void unmap_page(void *virtual_addr) { |
85 | | - u32 virtual = (u32)virtual_addr; |
86 | | - u32 pd_index = PAGE_DIRECTORY_INDEX(virtual); |
87 | | - u32 pt_index = PAGE_TABLE_INDEX(virtual); |
| 110 | + u32 pd_index = PAGE_DIRECTORY_INDEX((u32)virtual_addr); |
| 111 | + u32 pt_index = PAGE_TABLE_INDEX((u32)virtual_addr); |
88 | 112 |
|
89 | | - // Получаем таблицу страниц |
90 | | - page_table_t *table = (page_table_t*)(current_directory->entries[pd_index].frame << 12); |
| 113 | + if (current_directory->entries[pd_index].present) { |
| 114 | + page_table_t *table = (page_table_t *)(current_directory->entries[pd_index].frame << 12); |
| 115 | + if (table->entries[pt_index].present) { |
| 116 | + table->entries[pt_index].present = 0; |
| 117 | + used_pages--; |
91 | 118 |
|
92 | | - // Очищаем запись |
93 | | - table->entries[pt_index].present = 0; |
94 | | - |
95 | | - // Инвалидируем TLB |
96 | | - asm volatile("invlpg (%0)" : : "r" (virtual_addr)); |
| 119 | + // Принудительно обновляем TLB |
| 120 | + asm volatile("invlpg (%0)" :: "r"(virtual_addr)); |
| 121 | + } |
| 122 | + } |
97 | 123 | } |
98 | 124 |
|
99 | | -// Получение физического адреса по виртуальному |
100 | 125 | void *get_physical_address(void *virtual_addr) { |
101 | | - u32 virtual = (u32)virtual_addr; |
102 | | - u32 pd_index = PAGE_DIRECTORY_INDEX(virtual); |
103 | | - u32 pt_index = PAGE_TABLE_INDEX(virtual); |
| 126 | + u32 pd_index = PAGE_DIRECTORY_INDEX((u32)virtual_addr); |
| 127 | + u32 pt_index = PAGE_TABLE_INDEX((u32)virtual_addr); |
104 | 128 |
|
105 | | - // Проверяем наличие страницы |
106 | 129 | if (!current_directory->entries[pd_index].present) { |
107 | 130 | return NULL; |
108 | 131 | } |
109 | 132 |
|
110 | | - // Получаем таблицу страниц |
111 | | - page_table_t *table = (page_table_t*)(current_directory->entries[pd_index].frame << 12); |
112 | | - |
113 | | - // Проверяем наличие записи в таблице |
| 133 | + page_table_t *table = (page_table_t *)(current_directory->entries[pd_index].frame << 12); |
114 | 134 | if (!table->entries[pt_index].present) { |
115 | 135 | return NULL; |
116 | 136 | } |
117 | 137 |
|
118 | | - // Вычисляем физический адрес |
119 | | - u32 physical = (table->entries[pt_index].frame << 12) + (virtual & 0xFFF); |
120 | | - return (void*)physical; |
| 138 | + return (void *)(table->entries[pt_index].frame << 12); |
121 | 139 | } |
122 | 140 |
|
123 | | -// Обработчик page fault |
124 | | -void page_fault_handler(registers_t regs) { |
125 | | - u32 faulting_address; |
126 | | - asm volatile("mov %%cr2, %0" : "=r" (faulting_address)); |
127 | | - |
128 | | - // Проверяем причину page fault |
129 | | - int present = regs.err_code & 0x1; |
130 | | - int rw = regs.err_code & 0x2; |
131 | | - int user = regs.err_code & 0x4; |
132 | | - int reserved = regs.err_code & 0x8; |
133 | | - int id = regs.err_code & 0x10; |
134 | | - |
135 | | - kprintf("Page fault at 0x%x, caused by ", faulting_address); |
136 | | - |
137 | | - if (present) { |
138 | | - kprint("page protection violation"); |
139 | | - } else { |
140 | | - kprint("non-present page"); |
141 | | - } |
142 | | - |
143 | | - if (rw) { |
144 | | - kprint(" write attempt"); |
145 | | - } else { |
146 | | - kprint(" read attempt"); |
147 | | - } |
148 | | - |
149 | | - if (user) { |
150 | | - kprint(" in user mode"); |
151 | | - } else { |
152 | | - kprint(" in kernel mode"); |
153 | | - } |
154 | | - |
155 | | - kprint("\n"); |
| 141 | +u32 get_memory_used_pages() { |
| 142 | + return used_pages; |
| 143 | +} |
156 | 144 |
|
157 | | - // Зависаем систему при page fault в ядре |
158 | | - asm volatile("hlt"); |
| 145 | +u32 get_memory_free_pages() { |
| 146 | + return total_pages - used_pages; |
159 | 147 | } |
0 commit comments