MemoryManager.cpp 24 KB

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