MemoryManager.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696
  1. /*
  2. * Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
  3. * All rights reserved.
  4. *
  5. * Redistribution and use in source and binary forms, with or without
  6. * modification, are permitted provided that the following conditions are met:
  7. *
  8. * 1. Redistributions of source code must retain the above copyright notice, this
  9. * list of conditions and the following disclaimer.
  10. *
  11. * 2. Redistributions in binary form must reproduce the above copyright notice,
  12. * this list of conditions and the following disclaimer in the documentation
  13. * and/or other materials provided with the distribution.
  14. *
  15. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  16. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  18. * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
  19. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  21. * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  22. * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  23. * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  24. * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25. */
  26. #include "CMOS.h"
  27. #include "Process.h"
  28. #include "StdLib.h"
  29. #include <AK/Assertions.h>
  30. #include <AK/kstdio.h>
  31. #include <Kernel/Arch/i386/CPU.h>
  32. #include <Kernel/FileSystem/Inode.h>
  33. #include <Kernel/Multiboot.h>
  34. #include <Kernel/VM/AnonymousVMObject.h>
  35. #include <Kernel/VM/InodeVMObject.h>
  36. #include <Kernel/VM/MemoryManager.h>
  37. #include <Kernel/VM/PurgeableVMObject.h>
  38. //#define MM_DEBUG
  39. //#define PAGE_FAULT_DEBUG
  40. static MemoryManager* s_the;
  41. MemoryManager& MM
  42. {
  43. return *s_the;
  44. }
  45. MemoryManager::MemoryManager()
  46. {
  47. m_kernel_page_directory = PageDirectory::create_kernel_page_directory();
  48. parse_memory_map();
  49. asm volatile("movl %%eax, %%cr3" ::"a"(kernel_page_directory().cr3()));
  50. setup_low_1mb();
  51. protect_kernel_image();
  52. }
  53. MemoryManager::~MemoryManager()
  54. {
  55. }
  56. void MemoryManager::protect_kernel_image()
  57. {
  58. // Disable writing to the kernel text and rodata segments.
  59. extern uintptr_t start_of_kernel_text;
  60. extern uintptr_t start_of_kernel_data;
  61. for (size_t i = (uintptr_t)&start_of_kernel_text; i < (uintptr_t)&start_of_kernel_data; i += PAGE_SIZE) {
  62. auto& pte = ensure_pte(kernel_page_directory(), VirtualAddress(i));
  63. pte.set_writable(false);
  64. }
  65. if (g_cpu_supports_nx) {
  66. // Disable execution of the kernel data and bss segments.
  67. extern uintptr_t end_of_kernel_bss;
  68. for (size_t i = (uintptr_t)&start_of_kernel_data; i < (uintptr_t)&end_of_kernel_bss; i += PAGE_SIZE) {
  69. auto& pte = ensure_pte(kernel_page_directory(), VirtualAddress(i));
  70. pte.set_execute_disabled(true);
  71. }
  72. }
  73. }
  74. void MemoryManager::setup_low_1mb()
  75. {
  76. m_low_page_table = allocate_user_physical_page(ShouldZeroFill::Yes);
  77. auto* pd_zero = quickmap_pd(kernel_page_directory(), 0);
  78. pd_zero[1].set_present(false);
  79. pd_zero[2].set_present(false);
  80. pd_zero[3].set_present(false);
  81. auto& pde_zero = pd_zero[0];
  82. pde_zero.set_page_table_base(m_low_page_table->paddr().get());
  83. pde_zero.set_present(true);
  84. pde_zero.set_huge(false);
  85. pde_zero.set_writable(true);
  86. pde_zero.set_user_allowed(false);
  87. if (g_cpu_supports_nx)
  88. pde_zero.set_execute_disabled(true);
  89. for (uintptr_t offset = 0; offset < (2 * MB); offset += PAGE_SIZE) {
  90. auto& page_table_page = m_low_page_table;
  91. auto& pte = quickmap_pt(page_table_page->paddr())[offset / PAGE_SIZE];
  92. pte.set_physical_page_base(offset);
  93. pte.set_user_allowed(false);
  94. pte.set_present(offset != 0);
  95. pte.set_writable(offset < (1 * MB));
  96. }
  97. }
  98. void MemoryManager::parse_memory_map()
  99. {
  100. RefPtr<PhysicalRegion> region;
  101. bool region_is_super = false;
  102. auto* mmap = (multiboot_memory_map_t*)(low_physical_to_virtual(multiboot_info_ptr->mmap_addr));
  103. for (; (unsigned long)mmap < (low_physical_to_virtual(multiboot_info_ptr->mmap_addr)) + (multiboot_info_ptr->mmap_length); mmap = (multiboot_memory_map_t*)((unsigned long)mmap + mmap->size + sizeof(mmap->size))) {
  104. kprintf("MM: Multiboot mmap: base_addr = 0x%x%08x, length = 0x%x%08x, type = 0x%x\n",
  105. (uintptr_t)(mmap->addr >> 32),
  106. (uintptr_t)(mmap->addr & 0xffffffff),
  107. (uintptr_t)(mmap->len >> 32),
  108. (uintptr_t)(mmap->len & 0xffffffff),
  109. (uintptr_t)mmap->type);
  110. if (mmap->type != MULTIBOOT_MEMORY_AVAILABLE)
  111. continue;
  112. // FIXME: Maybe make use of stuff below the 1MB mark?
  113. if (mmap->addr < (1 * MB))
  114. continue;
  115. if ((mmap->addr + mmap->len) > 0xffffffff)
  116. continue;
  117. auto diff = (uintptr_t)mmap->addr % PAGE_SIZE;
  118. if (diff != 0) {
  119. kprintf("MM: got an unaligned region base from the bootloader; correcting %p by %d bytes\n", mmap->addr, diff);
  120. diff = PAGE_SIZE - diff;
  121. mmap->addr += diff;
  122. mmap->len -= diff;
  123. }
  124. if ((mmap->len % PAGE_SIZE) != 0) {
  125. kprintf("MM: got an unaligned region length from the bootloader; correcting %d by %d bytes\n", mmap->len, mmap->len % PAGE_SIZE);
  126. mmap->len -= mmap->len % PAGE_SIZE;
  127. }
  128. if (mmap->len < PAGE_SIZE) {
  129. kprintf("MM: memory region from bootloader is too small; we want >= %d bytes, but got %d bytes\n", PAGE_SIZE, mmap->len);
  130. continue;
  131. }
  132. #ifdef MM_DEBUG
  133. kprintf("MM: considering memory at %p - %p\n",
  134. (uintptr_t)mmap->addr, (uintptr_t)(mmap->addr + mmap->len));
  135. #endif
  136. for (size_t page_base = mmap->addr; page_base < (mmap->addr + mmap->len); page_base += PAGE_SIZE) {
  137. auto addr = PhysicalAddress(page_base);
  138. if (page_base < 7 * MB) {
  139. // nothing
  140. } else if (page_base >= 7 * MB && page_base < 8 * MB) {
  141. if (region.is_null() || !region_is_super || region->upper().offset(PAGE_SIZE) != addr) {
  142. m_super_physical_regions.append(PhysicalRegion::create(addr, addr));
  143. region = m_super_physical_regions.last();
  144. region_is_super = true;
  145. } else {
  146. region->expand(region->lower(), addr);
  147. }
  148. } else {
  149. if (region.is_null() || region_is_super || region->upper().offset(PAGE_SIZE) != addr) {
  150. m_user_physical_regions.append(PhysicalRegion::create(addr, addr));
  151. region = m_user_physical_regions.last();
  152. region_is_super = false;
  153. } else {
  154. region->expand(region->lower(), addr);
  155. }
  156. }
  157. }
  158. }
  159. for (auto& region : m_super_physical_regions)
  160. m_super_physical_pages += region.finalize_capacity();
  161. for (auto& region : m_user_physical_regions)
  162. m_user_physical_pages += region.finalize_capacity();
  163. }
  164. PageTableEntry& MemoryManager::ensure_pte(PageDirectory& page_directory, VirtualAddress vaddr)
  165. {
  166. ASSERT_INTERRUPTS_DISABLED();
  167. u32 page_directory_table_index = (vaddr.get() >> 30) & 0x3;
  168. u32 page_directory_index = (vaddr.get() >> 21) & 0x1ff;
  169. u32 page_table_index = (vaddr.get() >> 12) & 0x1ff;
  170. auto* pd = quickmap_pd(page_directory, page_directory_table_index);
  171. PageDirectoryEntry& pde = pd[page_directory_index];
  172. if (!pde.is_present()) {
  173. #ifdef MM_DEBUG
  174. dbgprintf("MM: PDE %u not present (requested for V%p), allocating\n", page_directory_index, vaddr.get());
  175. #endif
  176. auto page_table = allocate_user_physical_page(ShouldZeroFill::Yes);
  177. #ifdef MM_DEBUG
  178. dbgprintf("MM: PD K%p (%s) at P%p allocated page table #%u (for V%p) at P%p\n",
  179. &page_directory,
  180. &page_directory == m_kernel_page_directory ? "Kernel" : "User",
  181. page_directory.cr3(),
  182. page_directory_index,
  183. vaddr.get(),
  184. page_table->paddr().get());
  185. #endif
  186. pde.set_page_table_base(page_table->paddr().get());
  187. pde.set_user_allowed(true);
  188. pde.set_present(true);
  189. pde.set_writable(true);
  190. pde.set_global(&page_directory == m_kernel_page_directory.ptr());
  191. page_directory.m_physical_pages.set(page_directory_index, move(page_table));
  192. }
  193. return quickmap_pt(PhysicalAddress((uintptr_t)pde.page_table_base()))[page_table_index];
  194. }
  195. void MemoryManager::initialize()
  196. {
  197. s_the = new MemoryManager;
  198. }
  199. Region* MemoryManager::kernel_region_from_vaddr(VirtualAddress vaddr)
  200. {
  201. if (vaddr.get() < 0xc0000000)
  202. return nullptr;
  203. for (auto& region : MM.m_kernel_regions) {
  204. if (region.contains(vaddr))
  205. return &region;
  206. }
  207. return nullptr;
  208. }
  209. Region* MemoryManager::user_region_from_vaddr(Process& process, VirtualAddress vaddr)
  210. {
  211. // FIXME: Use a binary search tree (maybe red/black?) or some other more appropriate data structure!
  212. for (auto& region : process.m_regions) {
  213. if (region.contains(vaddr))
  214. return &region;
  215. }
  216. dbg() << process << " Couldn't find user region for " << vaddr;
  217. return nullptr;
  218. }
  219. Region* MemoryManager::region_from_vaddr(Process& process, VirtualAddress vaddr)
  220. {
  221. if (auto* region = kernel_region_from_vaddr(vaddr))
  222. return region;
  223. return user_region_from_vaddr(process, vaddr);
  224. }
  225. const Region* MemoryManager::region_from_vaddr(const Process& process, VirtualAddress vaddr)
  226. {
  227. if (auto* region = kernel_region_from_vaddr(vaddr))
  228. return region;
  229. return user_region_from_vaddr(const_cast<Process&>(process), vaddr);
  230. }
  231. Region* MemoryManager::region_from_vaddr(VirtualAddress vaddr)
  232. {
  233. if (auto* region = kernel_region_from_vaddr(vaddr))
  234. return region;
  235. auto page_directory = PageDirectory::find_by_cr3(cpu_cr3());
  236. if (!page_directory)
  237. return nullptr;
  238. ASSERT(page_directory->process());
  239. return user_region_from_vaddr(*page_directory->process(), vaddr);
  240. }
  241. PageFaultResponse MemoryManager::handle_page_fault(const PageFault& fault)
  242. {
  243. ASSERT_INTERRUPTS_DISABLED();
  244. ASSERT(current);
  245. #ifdef PAGE_FAULT_DEBUG
  246. dbgprintf("MM: handle_page_fault(%w) at V%p\n", fault.code(), fault.vaddr().get());
  247. #endif
  248. auto* region = region_from_vaddr(fault.vaddr());
  249. if (!region) {
  250. kprintf("NP(error) fault at invalid address V%p\n", fault.vaddr().get());
  251. return PageFaultResponse::ShouldCrash;
  252. }
  253. return region->handle_fault(fault);
  254. }
  255. OwnPtr<Region> MemoryManager::allocate_kernel_region(size_t size, const StringView& name, u8 access, bool user_accessible, bool should_commit, bool cacheable)
  256. {
  257. InterruptDisabler disabler;
  258. ASSERT(!(size % PAGE_SIZE));
  259. auto range = kernel_page_directory().range_allocator().allocate_anywhere(size);
  260. ASSERT(range.is_valid());
  261. OwnPtr<Region> region;
  262. if (user_accessible)
  263. region = Region::create_user_accessible(range, name, access, cacheable);
  264. else
  265. region = Region::create_kernel_only(range, name, access, cacheable);
  266. region->set_page_directory(kernel_page_directory());
  267. // FIXME: It would be cool if these could zero-fill on demand instead.
  268. if (should_commit)
  269. region->commit();
  270. return region;
  271. }
  272. OwnPtr<Region> MemoryManager::allocate_kernel_region(PhysicalAddress paddr, size_t size, const StringView& name, u8 access, bool user_accessible, bool cacheable)
  273. {
  274. InterruptDisabler disabler;
  275. ASSERT(!(size % PAGE_SIZE));
  276. auto range = kernel_page_directory().range_allocator().allocate_anywhere(size);
  277. ASSERT(range.is_valid());
  278. OwnPtr<Region> region;
  279. if (user_accessible)
  280. region = Region::create_user_accessible(range, AnonymousVMObject::create_for_physical_range(paddr, size), 0, name, access, cacheable);
  281. else
  282. region = Region::create_kernel_only(range, AnonymousVMObject::create_for_physical_range(paddr, size), 0, name, access, cacheable);
  283. region->map(kernel_page_directory());
  284. return region;
  285. }
  286. OwnPtr<Region> MemoryManager::allocate_user_accessible_kernel_region(size_t size, const StringView& name, u8 access, bool cacheable)
  287. {
  288. return allocate_kernel_region(size, name, access, true, true, cacheable);
  289. }
  290. OwnPtr<Region> MemoryManager::allocate_kernel_region_with_vmobject(VMObject& vmobject, size_t size, const StringView& name, u8 access, bool user_accessible, bool cacheable)
  291. {
  292. InterruptDisabler disabler;
  293. ASSERT(!(size % PAGE_SIZE));
  294. auto range = kernel_page_directory().range_allocator().allocate_anywhere(size);
  295. ASSERT(range.is_valid());
  296. OwnPtr<Region> region;
  297. if (user_accessible)
  298. region = Region::create_user_accessible(range, vmobject, 0, name, access, cacheable);
  299. else
  300. region = Region::create_kernel_only(range, vmobject, 0, name, access, cacheable);
  301. region->map(kernel_page_directory());
  302. return region;
  303. }
  304. void MemoryManager::deallocate_user_physical_page(PhysicalPage&& page)
  305. {
  306. for (auto& region : m_user_physical_regions) {
  307. if (!region.contains(page)) {
  308. kprintf(
  309. "MM: deallocate_user_physical_page: %p not in %p -> %p\n",
  310. page.paddr().get(), region.lower().get(), region.upper().get());
  311. continue;
  312. }
  313. region.return_page(move(page));
  314. --m_user_physical_pages_used;
  315. return;
  316. }
  317. kprintf("MM: deallocate_user_physical_page couldn't figure out region for user page @ %p\n", page.paddr().get());
  318. ASSERT_NOT_REACHED();
  319. }
  320. RefPtr<PhysicalPage> MemoryManager::find_free_user_physical_page()
  321. {
  322. RefPtr<PhysicalPage> page;
  323. for (auto& region : m_user_physical_regions) {
  324. page = region.take_free_page(false);
  325. if (!page.is_null())
  326. break;
  327. }
  328. return page;
  329. }
  330. RefPtr<PhysicalPage> MemoryManager::allocate_user_physical_page(ShouldZeroFill should_zero_fill)
  331. {
  332. InterruptDisabler disabler;
  333. RefPtr<PhysicalPage> page = find_free_user_physical_page();
  334. if (!page) {
  335. if (m_user_physical_regions.is_empty()) {
  336. kprintf("MM: no user physical regions available (?)\n");
  337. }
  338. for_each_vmobject([&](auto& vmobject) {
  339. if (vmobject.is_purgeable()) {
  340. auto& purgeable_vmobject = static_cast<PurgeableVMObject&>(vmobject);
  341. int purged_page_count = purgeable_vmobject.purge_with_interrupts_disabled({});
  342. if (purged_page_count) {
  343. kprintf("MM: Purge saved the day! Purged %d pages from PurgeableVMObject{%p}\n", purged_page_count, &purgeable_vmobject);
  344. page = find_free_user_physical_page();
  345. ASSERT(page);
  346. return IterationDecision::Break;
  347. }
  348. }
  349. return IterationDecision::Continue;
  350. });
  351. if (!page) {
  352. kprintf("MM: no user physical pages available\n");
  353. ASSERT_NOT_REACHED();
  354. return {};
  355. }
  356. }
  357. #ifdef MM_DEBUG
  358. dbgprintf("MM: allocate_user_physical_page vending P%p\n", page->paddr().get());
  359. #endif
  360. if (should_zero_fill == ShouldZeroFill::Yes) {
  361. auto* ptr = quickmap_page(*page);
  362. memset(ptr, 0, PAGE_SIZE);
  363. unquickmap_page();
  364. }
  365. ++m_user_physical_pages_used;
  366. return page;
  367. }
  368. void MemoryManager::deallocate_supervisor_physical_page(PhysicalPage&& page)
  369. {
  370. for (auto& region : m_super_physical_regions) {
  371. if (!region.contains(page)) {
  372. kprintf(
  373. "MM: deallocate_supervisor_physical_page: %p not in %p -> %p\n",
  374. page.paddr().get(), region.lower().get(), region.upper().get());
  375. continue;
  376. }
  377. region.return_page(move(page));
  378. --m_super_physical_pages_used;
  379. return;
  380. }
  381. kprintf("MM: deallocate_supervisor_physical_page couldn't figure out region for super page @ %p\n", page.paddr().get());
  382. ASSERT_NOT_REACHED();
  383. }
  384. RefPtr<PhysicalPage> MemoryManager::allocate_supervisor_physical_page()
  385. {
  386. InterruptDisabler disabler;
  387. RefPtr<PhysicalPage> page;
  388. for (auto& region : m_super_physical_regions) {
  389. page = region.take_free_page(true);
  390. if (page.is_null())
  391. continue;
  392. }
  393. if (!page) {
  394. if (m_super_physical_regions.is_empty()) {
  395. kprintf("MM: no super physical regions available (?)\n");
  396. }
  397. kprintf("MM: no super physical pages available\n");
  398. ASSERT_NOT_REACHED();
  399. return {};
  400. }
  401. #ifdef MM_DEBUG
  402. dbgprintf("MM: allocate_supervisor_physical_page vending P%p\n", page->paddr().get());
  403. #endif
  404. fast_u32_fill((u32*)page->paddr().offset(0xc0000000).as_ptr(), 0, PAGE_SIZE / sizeof(u32));
  405. ++m_super_physical_pages_used;
  406. return page;
  407. }
  408. void MemoryManager::enter_process_paging_scope(Process& process)
  409. {
  410. ASSERT(current);
  411. InterruptDisabler disabler;
  412. current->tss().cr3 = process.page_directory().cr3();
  413. asm volatile("movl %%eax, %%cr3" ::"a"(process.page_directory().cr3())
  414. : "memory");
  415. }
  416. void MemoryManager::flush_entire_tlb()
  417. {
  418. asm volatile(
  419. "mov %%cr3, %%eax\n"
  420. "mov %%eax, %%cr3\n" ::
  421. : "%eax", "memory");
  422. }
  423. void MemoryManager::flush_tlb(VirtualAddress vaddr)
  424. {
  425. #ifdef MM_DEBUG
  426. dbgprintf("MM: Flush page V%p\n", vaddr.get());
  427. #endif
  428. asm volatile("invlpg %0"
  429. :
  430. : "m"(*(char*)vaddr.get())
  431. : "memory");
  432. }
  433. extern "C" PageTableEntry boot_pd3_pde1023_pt[1024];
  434. PageDirectoryEntry* MemoryManager::quickmap_pd(PageDirectory& directory, size_t pdpt_index)
  435. {
  436. auto& pte = boot_pd3_pde1023_pt[4];
  437. auto pd_paddr = directory.m_directory_pages[pdpt_index]->paddr();
  438. if (pte.physical_page_base() != pd_paddr.as_ptr()) {
  439. #ifdef MM_DEBUG
  440. dbgprintf("quickmap_pd: Mapping P%p at 0xffe04000 in pte @ %p\n", directory.m_directory_pages[pdpt_index]->paddr().as_ptr(), &pte);
  441. #endif
  442. pte.set_physical_page_base(pd_paddr.get());
  443. pte.set_present(true);
  444. pte.set_writable(true);
  445. pte.set_user_allowed(false);
  446. flush_tlb(VirtualAddress(0xffe04000));
  447. }
  448. return (PageDirectoryEntry*)0xffe04000;
  449. }
  450. PageTableEntry* MemoryManager::quickmap_pt(PhysicalAddress pt_paddr)
  451. {
  452. auto& pte = boot_pd3_pde1023_pt[8];
  453. if (pte.physical_page_base() != pt_paddr.as_ptr()) {
  454. #ifdef MM_DEBUG
  455. dbgprintf("quickmap_pt: Mapping P%p at 0xffe08000 in pte @ %p\n", pt_paddr.as_ptr(), &pte);
  456. #endif
  457. pte.set_physical_page_base(pt_paddr.get());
  458. pte.set_present(true);
  459. pte.set_writable(true);
  460. pte.set_user_allowed(false);
  461. flush_tlb(VirtualAddress(0xffe08000));
  462. }
  463. return (PageTableEntry*)0xffe08000;
  464. }
  465. void MemoryManager::map_for_kernel(VirtualAddress vaddr, PhysicalAddress paddr, bool cache_disabled)
  466. {
  467. auto& pte = ensure_pte(kernel_page_directory(), vaddr);
  468. pte.set_physical_page_base(paddr.get());
  469. pte.set_present(true);
  470. pte.set_writable(true);
  471. pte.set_user_allowed(false);
  472. pte.set_cache_disabled(cache_disabled);
  473. flush_tlb(vaddr);
  474. }
  475. u8* MemoryManager::quickmap_page(PhysicalPage& physical_page)
  476. {
  477. ASSERT_INTERRUPTS_DISABLED();
  478. ASSERT(!m_quickmap_in_use);
  479. m_quickmap_in_use = true;
  480. auto& pte = boot_pd3_pde1023_pt[0];
  481. if (pte.physical_page_base() != physical_page.paddr().as_ptr()) {
  482. #ifdef MM_DEBUG
  483. dbgprintf("quickmap_page: Mapping P%p at 0xffe00000 in pte @ %p\n", physical_page.paddr().as_ptr(), &pte);
  484. #endif
  485. pte.set_physical_page_base(physical_page.paddr().get());
  486. pte.set_present(true);
  487. pte.set_writable(true);
  488. pte.set_user_allowed(false);
  489. flush_tlb(VirtualAddress(0xffe00000));
  490. }
  491. return (u8*)0xffe00000;
  492. }
  493. void MemoryManager::unquickmap_page()
  494. {
  495. ASSERT_INTERRUPTS_DISABLED();
  496. ASSERT(m_quickmap_in_use);
  497. auto& pte = boot_pd3_pde1023_pt[0];
  498. pte.set_physical_page_base(0);
  499. pte.set_present(false);
  500. flush_tlb(VirtualAddress(0xffe00000));
  501. m_quickmap_in_use = false;
  502. }
  503. template<MemoryManager::AccessSpace space, MemoryManager::AccessType access_type>
  504. bool MemoryManager::validate_range(const Process& process, VirtualAddress base_vaddr, size_t size) const
  505. {
  506. ASSERT(size);
  507. VirtualAddress vaddr = base_vaddr.page_base();
  508. VirtualAddress end_vaddr = base_vaddr.offset(size - 1).page_base();
  509. if (end_vaddr < vaddr) {
  510. dbg() << *current << " Shenanigans! Asked to validate " << base_vaddr << " size=" << size;
  511. return false;
  512. }
  513. const Region* region = nullptr;
  514. while (vaddr <= end_vaddr) {
  515. if (!region || !region->contains(vaddr)) {
  516. if (space == AccessSpace::Kernel)
  517. region = kernel_region_from_vaddr(vaddr);
  518. if (!region || !region->contains(vaddr))
  519. region = user_region_from_vaddr(const_cast<Process&>(process), vaddr);
  520. if (!region
  521. || (space == AccessSpace::User && !region->is_user_accessible())
  522. || (access_type == AccessType::Read && !region->is_readable())
  523. || (access_type == AccessType::Write && !region->is_writable())) {
  524. return false;
  525. }
  526. }
  527. vaddr = vaddr.offset(PAGE_SIZE);
  528. }
  529. return true;
  530. }
  531. bool MemoryManager::validate_user_stack(const Process& process, VirtualAddress vaddr) const
  532. {
  533. if (!is_user_address(vaddr))
  534. return false;
  535. auto* region = user_region_from_vaddr(const_cast<Process&>(process), vaddr);
  536. return region && region->is_user_accessible() && region->is_stack();
  537. }
  538. bool MemoryManager::validate_kernel_read(const Process& process, VirtualAddress vaddr, size_t size) const
  539. {
  540. return validate_range<AccessSpace::Kernel, AccessType::Read>(process, vaddr, size);
  541. }
  542. bool MemoryManager::validate_user_read(const Process& process, VirtualAddress vaddr, size_t size) const
  543. {
  544. if (!is_user_address(vaddr))
  545. return false;
  546. return validate_range<AccessSpace::User, AccessType::Read>(process, vaddr, size);
  547. }
  548. bool MemoryManager::validate_user_write(const Process& process, VirtualAddress vaddr, size_t size) const
  549. {
  550. if (!is_user_address(vaddr))
  551. return false;
  552. return validate_range<AccessSpace::User, AccessType::Write>(process, vaddr, size);
  553. }
  554. void MemoryManager::register_vmobject(VMObject& vmobject)
  555. {
  556. InterruptDisabler disabler;
  557. m_vmobjects.append(&vmobject);
  558. }
  559. void MemoryManager::unregister_vmobject(VMObject& vmobject)
  560. {
  561. InterruptDisabler disabler;
  562. m_vmobjects.remove(&vmobject);
  563. }
  564. void MemoryManager::register_region(Region& region)
  565. {
  566. InterruptDisabler disabler;
  567. if (region.vaddr().get() >= 0xc0000000)
  568. m_kernel_regions.append(&region);
  569. else
  570. m_user_regions.append(&region);
  571. }
  572. void MemoryManager::unregister_region(Region& region)
  573. {
  574. InterruptDisabler disabler;
  575. if (region.vaddr().get() >= 0xc0000000)
  576. m_kernel_regions.remove(&region);
  577. else
  578. m_user_regions.remove(&region);
  579. }
  580. void MemoryManager::dump_kernel_regions()
  581. {
  582. kprintf("Kernel regions:\n");
  583. kprintf("BEGIN END SIZE ACCESS NAME\n");
  584. for (auto& region : MM.m_kernel_regions) {
  585. kprintf("%08x -- %08x %08x %c%c%c%c%c%c %s\n",
  586. region.vaddr().get(),
  587. region.vaddr().offset(region.size() - 1).get(),
  588. region.size(),
  589. region.is_readable() ? 'R' : ' ',
  590. region.is_writable() ? 'W' : ' ',
  591. region.is_executable() ? 'X' : ' ',
  592. region.is_shared() ? 'S' : ' ',
  593. region.is_stack() ? 'T' : ' ',
  594. region.vmobject().is_purgeable() ? 'P' : ' ',
  595. region.name().characters());
  596. }
  597. }
  598. ProcessPagingScope::ProcessPagingScope(Process& process)
  599. {
  600. ASSERT(current);
  601. asm("movl %%cr3, %%eax"
  602. : "=a"(m_previous_cr3));
  603. MM.enter_process_paging_scope(process);
  604. }
  605. ProcessPagingScope::~ProcessPagingScope()
  606. {
  607. InterruptDisabler disabler;
  608. current->tss().cr3 = m_previous_cr3;
  609. asm volatile("movl %%eax, %%cr3" ::"a"(m_previous_cr3)
  610. : "memory");
  611. }