mirror of
				https://github.com/RGBCube/serenity
				synced 2025-10-30 03:22:33 +00:00 
			
		
		
		
	 bcc6ddfb6b
			
		
	
	
		bcc6ddfb6b
		
	
	
	
	
		
			
			Since we transition to a new PageDirectory on exec(), we need a matching RangeAllocator to go with the new directory. Instead of juggling this in Process and MemoryManager, simply attach the RangeAllocator to the PageDirectory instead. Fixes #61.
		
			
				
	
	
		
			679 lines
		
	
	
	
		
			24 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			679 lines
		
	
	
	
		
			24 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| #include <Kernel/VM/MemoryManager.h>
 | |
| #include <Kernel/FileSystem/Inode.h>
 | |
| #include <AK/Assertions.h>
 | |
| #include <AK/kstdio.h>
 | |
| #include "i386.h"
 | |
| #include "StdLib.h"
 | |
| #include "Process.h"
 | |
| #include "CMOS.h"
 | |
| 
 | |
| //#define MM_DEBUG
 | |
| //#define PAGE_FAULT_DEBUG
 | |
| 
 | |
| static MemoryManager* s_the;
 | |
| unsigned MemoryManager::s_user_physical_pages_in_existence;
 | |
| unsigned MemoryManager::s_super_physical_pages_in_existence;
 | |
| 
 | |
| MemoryManager& MM
 | |
| {
 | |
|     return *s_the;
 | |
| }
 | |
| 
 | |
| MemoryManager::MemoryManager()
 | |
| {
 | |
|     // FIXME: This is not the best way to do memory map detection.
 | |
|     //        Rewrite to use BIOS int 15,e820 once we have VM86 support.
 | |
|     word base_memory = (CMOS::read(0x16) << 8) | CMOS::read(0x15);
 | |
|     word ext_memory = (CMOS::read(0x18) << 8) | CMOS::read(0x17);
 | |
| 
 | |
|     kprintf("%u kB base memory\n", base_memory);
 | |
|     kprintf("%u kB extended memory\n", ext_memory);
 | |
| 
 | |
|     m_ram_size = ext_memory * 1024;
 | |
| 
 | |
|     m_kernel_page_directory = PageDirectory::create_at_fixed_address(PhysicalAddress(0x4000));
 | |
|     m_page_table_zero = (dword*)0x6000;
 | |
| 
 | |
|     initialize_paging();
 | |
| 
 | |
|     kprintf("MM initialized.\n");
 | |
| }
 | |
| 
 | |
| MemoryManager::~MemoryManager()
 | |
| {
 | |
| }
 | |
| 
 | |
| void MemoryManager::populate_page_directory(PageDirectory& page_directory)
 | |
| {
 | |
|     page_directory.m_directory_page = allocate_supervisor_physical_page();
 | |
|     page_directory.entries()[0] = kernel_page_directory().entries()[0];
 | |
|     // Defer to the kernel page tables for 0xC0000000-0xFFFFFFFF
 | |
|     for (int i = 768; i < 1024; ++i)
 | |
|         page_directory.entries()[i] = kernel_page_directory().entries()[i];
 | |
| }
 | |
| 
 | |
| void MemoryManager::initialize_paging()
 | |
| {
 | |
|     static_assert(sizeof(MemoryManager::PageDirectoryEntry) == 4);
 | |
|     static_assert(sizeof(MemoryManager::PageTableEntry) == 4);
 | |
|     memset(m_page_table_zero, 0, PAGE_SIZE);
 | |
| 
 | |
| #ifdef MM_DEBUG
 | |
|     dbgprintf("MM: Kernel page directory @ %p\n", kernel_page_directory().cr3());
 | |
| #endif
 | |
| 
 | |
| #ifdef MM_DEBUG
 | |
|     dbgprintf("MM: Protect against null dereferences\n");
 | |
| #endif
 | |
|     // Make null dereferences crash.
 | |
|     map_protected(LinearAddress(0), PAGE_SIZE);
 | |
| 
 | |
| #ifdef MM_DEBUG
 | |
|     dbgprintf("MM: Identity map bottom 4MB\n");
 | |
| #endif
 | |
|     // The bottom 4 MB (except for the null page) are identity mapped & supervisor only.
 | |
|     // Every process shares these mappings.
 | |
|     create_identity_mapping(kernel_page_directory(), LinearAddress(PAGE_SIZE), (4 * MB) - PAGE_SIZE);
 | |
| 
 | |
|     // Basic memory map:
 | |
|     // 0      -> 512 kB         Kernel code. Root page directory & PDE 0.
 | |
|     // (last page before 1MB)   Used by quickmap_page().
 | |
|     // 1 MB   -> 2 MB           kmalloc_eternal() space.
 | |
|     // 2 MB   -> 3 MB           kmalloc() space.
 | |
|     // 3 MB   -> 4 MB           Supervisor physical pages (available for allocation!)
 | |
|     // 4 MB   -> 0xc0000000     Userspace physical pages (available for allocation!)
 | |
|     // 0xc0000000-0xffffffff    Kernel-only linear address space
 | |
| 
 | |
|     for (size_t i = (2 * MB); i < (4 * MB); i += PAGE_SIZE)
 | |
|         m_free_supervisor_physical_pages.append(PhysicalPage::create_eternal(PhysicalAddress(i), true));
 | |
| 
 | |
|     dbgprintf("MM: 4MB-%uMB available for allocation\n", m_ram_size / 1048576);
 | |
|     for (size_t i = (4 * MB); i < m_ram_size; i += PAGE_SIZE)
 | |
|         m_free_physical_pages.append(PhysicalPage::create_eternal(PhysicalAddress(i), false));
 | |
|     m_quickmap_addr = LinearAddress((1 * MB) - PAGE_SIZE);
 | |
| #ifdef MM_DEBUG
 | |
|     dbgprintf("MM: Quickmap will use P%x\n", m_quickmap_addr.get());
 | |
|     dbgprintf("MM: Installing page directory\n");
 | |
| #endif
 | |
| 
 | |
|     asm volatile("movl %%eax, %%cr3"::"a"(kernel_page_directory().cr3()));
 | |
|     asm volatile(
 | |
|         "movl %%cr0, %%eax\n"
 | |
|         "orl $0x80000001, %%eax\n"
 | |
|         "movl %%eax, %%cr0\n"
 | |
|         :::"%eax", "memory");
 | |
| 
 | |
| #ifdef MM_DEBUG
 | |
|     dbgprintf("MM: Paging initialized.\n");
 | |
| #endif
 | |
| }
 | |
| 
 | |
| RetainPtr<PhysicalPage> MemoryManager::allocate_page_table(PageDirectory& page_directory, unsigned index)
 | |
| {
 | |
|     ASSERT(!page_directory.m_physical_pages.contains(index));
 | |
|     auto physical_page = allocate_supervisor_physical_page();
 | |
|     if (!physical_page)
 | |
|         return nullptr;
 | |
|     page_directory.m_physical_pages.set(index, physical_page.copy_ref());
 | |
|     return physical_page;
 | |
| }
 | |
| 
 | |
| void MemoryManager::remove_identity_mapping(PageDirectory& page_directory, LinearAddress laddr, size_t size)
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
|     // FIXME: ASSERT(laddr is 4KB aligned);
 | |
|     for (dword offset = 0; offset < size; offset += PAGE_SIZE) {
 | |
|         auto pte_address = laddr.offset(offset);
 | |
|         auto pte = ensure_pte(page_directory, pte_address);
 | |
|         pte.set_physical_page_base(0);
 | |
|         pte.set_user_allowed(false);
 | |
|         pte.set_present(true);
 | |
|         pte.set_writable(true);
 | |
|         flush_tlb(pte_address);
 | |
|     }
 | |
| }
 | |
| 
 | |
| auto MemoryManager::ensure_pte(PageDirectory& page_directory, LinearAddress laddr) -> PageTableEntry
 | |
| {
 | |
|     ASSERT_INTERRUPTS_DISABLED();
 | |
|     dword page_directory_index = (laddr.get() >> 22) & 0x3ff;
 | |
|     dword page_table_index = (laddr.get() >> 12) & 0x3ff;
 | |
| 
 | |
|     PageDirectoryEntry pde = PageDirectoryEntry(&page_directory.entries()[page_directory_index]);
 | |
|     if (!pde.is_present()) {
 | |
| #ifdef MM_DEBUG
 | |
|         dbgprintf("MM: PDE %u not present (requested for L%x), allocating\n", page_directory_index, laddr.get());
 | |
| #endif
 | |
|         if (page_directory_index == 0) {
 | |
|             ASSERT(&page_directory == m_kernel_page_directory);
 | |
|             pde.set_page_table_base((dword)m_page_table_zero);
 | |
|             pde.set_user_allowed(false);
 | |
|             pde.set_present(true);
 | |
|             pde.set_writable(true);
 | |
|         } else {
 | |
|             //ASSERT(&page_directory != m_kernel_page_directory.ptr());
 | |
|             auto page_table = allocate_page_table(page_directory, page_directory_index);
 | |
| #ifdef MM_DEBUG
 | |
|             dbgprintf("MM: PD K%x (%s) at P%x allocated page table #%u (for L%x) at P%x\n",
 | |
|                 &page_directory,
 | |
|                 &page_directory == m_kernel_page_directory ? "Kernel" : "User",
 | |
|                 page_directory.cr3(),
 | |
|                 page_directory_index,
 | |
|                 laddr.get(),
 | |
|                 page_table->paddr().get());
 | |
| #endif
 | |
| 
 | |
|             pde.set_page_table_base(page_table->paddr().get());
 | |
|             pde.set_user_allowed(true);
 | |
|             pde.set_present(true);
 | |
|             pde.set_writable(true);
 | |
|             page_directory.m_physical_pages.set(page_directory_index, move(page_table));
 | |
|         }
 | |
|     }
 | |
|     return PageTableEntry(&pde.page_table_base()[page_table_index]);
 | |
| }
 | |
| 
 | |
| void MemoryManager::map_protected(LinearAddress laddr, size_t length)
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
|     // FIXME: ASSERT(linearAddress is 4KB aligned);
 | |
|     for (dword offset = 0; offset < length; offset += PAGE_SIZE) {
 | |
|         auto pte_address = laddr.offset(offset);
 | |
|         auto pte = ensure_pte(kernel_page_directory(), pte_address);
 | |
|         pte.set_physical_page_base(pte_address.get());
 | |
|         pte.set_user_allowed(false);
 | |
|         pte.set_present(false);
 | |
|         pte.set_writable(false);
 | |
|         flush_tlb(pte_address);
 | |
|     }
 | |
| }
 | |
| 
 | |
| void MemoryManager::create_identity_mapping(PageDirectory& page_directory, LinearAddress laddr, size_t size)
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
|     ASSERT((laddr.get() & ~PAGE_MASK) == 0);
 | |
|     for (dword offset = 0; offset < size; offset += PAGE_SIZE) {
 | |
|         auto pte_address = laddr.offset(offset);
 | |
|         auto pte = ensure_pte(page_directory, pte_address);
 | |
|         pte.set_physical_page_base(pte_address.get());
 | |
|         pte.set_user_allowed(false);
 | |
|         pte.set_present(true);
 | |
|         pte.set_writable(true);
 | |
|         page_directory.flush(pte_address);
 | |
|     }
 | |
| }
 | |
| 
 | |
| void MemoryManager::initialize()
 | |
| {
 | |
|     s_the = new MemoryManager;
 | |
| }
 | |
| 
 | |
| Region* MemoryManager::region_from_laddr(Process& process, LinearAddress laddr)
 | |
| {
 | |
|     ASSERT_INTERRUPTS_DISABLED();
 | |
| 
 | |
|     if (laddr.get() >= 0xc0000000) {
 | |
|         for (auto& region : MM.m_kernel_regions) {
 | |
|             if (region->contains(laddr))
 | |
|                 return region;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     // FIXME: Use a binary search tree (maybe red/black?) or some other more appropriate data structure!
 | |
|     for (auto& region : process.m_regions) {
 | |
|         if (region->contains(laddr))
 | |
|             return region.ptr();
 | |
|     }
 | |
|     dbgprintf("%s(%u) Couldn't find region for L%x (CR3=%x)\n", process.name().characters(), process.pid(), laddr.get(), process.page_directory().cr3());
 | |
|     return nullptr;
 | |
| }
 | |
| 
 | |
| const Region* MemoryManager::region_from_laddr(const Process& process, LinearAddress laddr)
 | |
| {
 | |
|     if (laddr.get() >= 0xc0000000) {
 | |
|         for (auto& region : MM.m_kernel_regions) {
 | |
|             if (region->contains(laddr))
 | |
|                 return region;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     // FIXME: Use a binary search tree (maybe red/black?) or some other more appropriate data structure!
 | |
|     for (auto& region : process.m_regions) {
 | |
|         if (region->contains(laddr))
 | |
|             return region.ptr();
 | |
|     }
 | |
|     dbgprintf("%s(%u) Couldn't find region for L%x (CR3=%x)\n", process.name().characters(), process.pid(), laddr.get(), process.page_directory().cr3());
 | |
|     return nullptr;
 | |
| }
 | |
| 
 | |
| bool MemoryManager::zero_page(Region& region, unsigned page_index_in_region)
 | |
| {
 | |
|     ASSERT_INTERRUPTS_DISABLED();
 | |
|     auto& vmo = region.vmo();
 | |
|     auto& vmo_page = vmo.physical_pages()[region.first_page_index() + page_index_in_region];
 | |
|     sti();
 | |
|     LOCKER(vmo.m_paging_lock);
 | |
|     cli();
 | |
|     if (!vmo_page.is_null()) {
 | |
| #ifdef PAGE_FAULT_DEBUG
 | |
|         dbgprintf("MM: zero_page() but page already present. Fine with me!\n");
 | |
| #endif
 | |
|         remap_region_page(region, page_index_in_region, true);
 | |
|         return true;
 | |
|     }
 | |
|     auto physical_page = allocate_physical_page(ShouldZeroFill::Yes);
 | |
| #ifdef PAGE_FAULT_DEBUG
 | |
|     dbgprintf("      >> ZERO P%x\n", physical_page->paddr().get());
 | |
| #endif
 | |
|     region.set_should_cow(page_index_in_region, false);
 | |
|     vmo.physical_pages()[page_index_in_region] = move(physical_page);
 | |
|     remap_region_page(region, page_index_in_region, true);
 | |
|     return true;
 | |
| }
 | |
| 
 | |
| bool MemoryManager::copy_on_write(Region& region, unsigned page_index_in_region)
 | |
| {
 | |
|     ASSERT_INTERRUPTS_DISABLED();
 | |
|     auto& vmo = region.vmo();
 | |
|     if (vmo.physical_pages()[page_index_in_region]->retain_count() == 1) {
 | |
| #ifdef PAGE_FAULT_DEBUG
 | |
|         dbgprintf("    >> It's a COW page but nobody is sharing it anymore. Remap r/w\n");
 | |
| #endif
 | |
|         region.set_should_cow(page_index_in_region, false);
 | |
|         remap_region_page(region, page_index_in_region, true);
 | |
|         return true;
 | |
|     }
 | |
| 
 | |
| #ifdef PAGE_FAULT_DEBUG
 | |
|     dbgprintf("    >> It's a COW page and it's time to COW!\n");
 | |
| #endif
 | |
|     auto physical_page_to_copy = move(vmo.physical_pages()[page_index_in_region]);
 | |
|     auto physical_page = allocate_physical_page(ShouldZeroFill::No);
 | |
|     byte* dest_ptr = quickmap_page(*physical_page);
 | |
|     const byte* src_ptr = region.laddr().offset(page_index_in_region * PAGE_SIZE).as_ptr();
 | |
| #ifdef PAGE_FAULT_DEBUG
 | |
|     dbgprintf("      >> COW P%x <- P%x\n", physical_page->paddr().get(), physical_page_to_copy->paddr().get());
 | |
| #endif
 | |
|     memcpy(dest_ptr, src_ptr, PAGE_SIZE);
 | |
|     vmo.physical_pages()[page_index_in_region] = move(physical_page);
 | |
|     unquickmap_page();
 | |
|     region.set_should_cow(page_index_in_region, false);
 | |
|     remap_region_page(region, page_index_in_region, true);
 | |
|     return true;
 | |
| }
 | |
| 
 | |
| 
 | |
| bool MemoryManager::page_in_from_inode(Region& region, unsigned page_index_in_region)
 | |
| {
 | |
|     ASSERT(region.page_directory());
 | |
|     auto& vmo = region.vmo();
 | |
|     ASSERT(!vmo.is_anonymous());
 | |
|     ASSERT(vmo.inode());
 | |
| 
 | |
|     auto& vmo_page = vmo.physical_pages()[region.first_page_index() + page_index_in_region];
 | |
| 
 | |
|     InterruptFlagSaver saver;
 | |
| 
 | |
|     sti();
 | |
|     LOCKER(vmo.m_paging_lock);
 | |
|     cli();
 | |
| 
 | |
|     if (!vmo_page.is_null()) {
 | |
|         dbgprintf("MM: page_in_from_inode() but page already present. Fine with me!\n");
 | |
|         remap_region_page(region, page_index_in_region, true);
 | |
|         return true;
 | |
|     }
 | |
| 
 | |
| #ifdef MM_DEBUG
 | |
|     dbgprintf("MM: page_in_from_inode ready to read from inode\n");
 | |
| #endif
 | |
|     sti();
 | |
|     byte page_buffer[PAGE_SIZE];
 | |
|     auto& inode = *vmo.inode();
 | |
|     auto nread = inode.read_bytes(vmo.inode_offset() + ((region.first_page_index() + page_index_in_region) * PAGE_SIZE), PAGE_SIZE, page_buffer, nullptr);
 | |
|     if (nread < 0) {
 | |
|         kprintf("MM: page_in_from_inode had error (%d) while reading!\n", nread);
 | |
|         return false;
 | |
|     }
 | |
|     if (nread < PAGE_SIZE) {
 | |
|         // If we read less than a page, zero out the rest to avoid leaking uninitialized data.
 | |
|         memset(page_buffer + nread, 0, PAGE_SIZE - nread);
 | |
|     }
 | |
|     cli();
 | |
|     vmo_page = allocate_physical_page(ShouldZeroFill::No);
 | |
|     if (vmo_page.is_null()) {
 | |
|         kprintf("MM: page_in_from_inode was unable to allocate a physical page\n");
 | |
|         return false;
 | |
|     }
 | |
|     remap_region_page(region, page_index_in_region, true);
 | |
|     byte* dest_ptr = region.laddr().offset(page_index_in_region * PAGE_SIZE).as_ptr();
 | |
|     memcpy(dest_ptr, page_buffer, PAGE_SIZE);
 | |
|     return true;
 | |
| }
 | |
| 
 | |
| PageFaultResponse MemoryManager::handle_page_fault(const PageFault& fault)
 | |
| {
 | |
|     ASSERT_INTERRUPTS_DISABLED();
 | |
|     ASSERT(current);
 | |
| #ifdef PAGE_FAULT_DEBUG
 | |
|     dbgprintf("MM: handle_page_fault(%w) at L%x\n", fault.code(), fault.laddr().get());
 | |
| #endif
 | |
|     ASSERT(fault.laddr() != m_quickmap_addr);
 | |
|     auto* region = region_from_laddr(current->process(), fault.laddr());
 | |
|     if (!region) {
 | |
|         kprintf("NP(error) fault at invalid address L%x\n", fault.laddr().get());
 | |
|         return PageFaultResponse::ShouldCrash;
 | |
|     }
 | |
|     auto page_index_in_region = region->page_index_from_address(fault.laddr());
 | |
|     if (fault.is_not_present()) {
 | |
|         if (region->vmo().inode()) {
 | |
| #ifdef PAGE_FAULT_DEBUG
 | |
|             dbgprintf("NP(inode) fault in Region{%p}[%u]\n", region, page_index_in_region);
 | |
| #endif
 | |
|             page_in_from_inode(*region, page_index_in_region);
 | |
|             return PageFaultResponse::Continue;
 | |
|         } else {
 | |
| #ifdef PAGE_FAULT_DEBUG
 | |
|             dbgprintf("NP(zero) fault in Region{%p}[%u]\n", region, page_index_in_region);
 | |
| #endif
 | |
|             zero_page(*region, page_index_in_region);
 | |
|             return PageFaultResponse::Continue;
 | |
|         }
 | |
|     } else if (fault.is_protection_violation()) {
 | |
|         if (region->should_cow(page_index_in_region)) {
 | |
| #ifdef PAGE_FAULT_DEBUG
 | |
|             dbgprintf("PV(cow) fault in Region{%p}[%u]\n", region, page_index_in_region);
 | |
| #endif
 | |
|             bool success = copy_on_write(*region, page_index_in_region);
 | |
|             ASSERT(success);
 | |
|             return PageFaultResponse::Continue;
 | |
|         }
 | |
|         kprintf("PV(error) fault in Region{%p}[%u] at L%x\n", region, page_index_in_region, fault.laddr().get());
 | |
|     } else {
 | |
|         ASSERT_NOT_REACHED();
 | |
|     }
 | |
| 
 | |
|     return PageFaultResponse::ShouldCrash;
 | |
| }
 | |
| 
 | |
| RetainPtr<Region> MemoryManager::allocate_kernel_region(size_t size, String&& name)
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
| 
 | |
|     ASSERT(!(size % PAGE_SIZE));
 | |
|     auto range = kernel_page_directory().range_allocator().allocate_anywhere(size);
 | |
|     ASSERT(range.is_valid());
 | |
|     auto region = adopt(*new Region(range, move(name), true, true, false));
 | |
|     MM.map_region_at_address(*m_kernel_page_directory, *region, range.base(), false);
 | |
|     // FIXME: It would be cool if these could zero-fill on demand instead.
 | |
|     region->commit();
 | |
|     return region;
 | |
| }
 | |
| 
 | |
| RetainPtr<PhysicalPage> MemoryManager::allocate_physical_page(ShouldZeroFill should_zero_fill)
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
|     if (1 > m_free_physical_pages.size()) {
 | |
|         kprintf("FUCK! No physical pages available.\n");
 | |
|         ASSERT_NOT_REACHED();
 | |
|         return { };
 | |
|     }
 | |
| #ifdef MM_DEBUG
 | |
|     dbgprintf("MM: allocate_physical_page vending P%x (%u remaining)\n", m_free_physical_pages.last()->paddr().get(), m_free_physical_pages.size());
 | |
| #endif
 | |
|     auto physical_page = m_free_physical_pages.take_last();
 | |
|     if (should_zero_fill == ShouldZeroFill::Yes) {
 | |
|         auto* ptr = (dword*)quickmap_page(*physical_page);
 | |
|         fast_dword_fill(ptr, 0, PAGE_SIZE / sizeof(dword));
 | |
|         unquickmap_page();
 | |
|     }
 | |
|     return physical_page;
 | |
| }
 | |
| 
 | |
| RetainPtr<PhysicalPage> MemoryManager::allocate_supervisor_physical_page()
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
|     if (1 > m_free_supervisor_physical_pages.size()) {
 | |
|         kprintf("FUCK! No physical pages available.\n");
 | |
|         ASSERT_NOT_REACHED();
 | |
|         return { };
 | |
|     }
 | |
| #ifdef MM_DEBUG
 | |
|     dbgprintf("MM: allocate_supervisor_physical_page vending P%x (%u remaining)\n", m_free_supervisor_physical_pages.last()->paddr().get(), m_free_supervisor_physical_pages.size());
 | |
| #endif
 | |
|     auto physical_page = m_free_supervisor_physical_pages.take_last();
 | |
|     fast_dword_fill((dword*)physical_page->paddr().as_ptr(), 0, PAGE_SIZE / sizeof(dword));
 | |
|     return physical_page;
 | |
| }
 | |
| 
 | |
| void MemoryManager::enter_process_paging_scope(Process& process)
 | |
| {
 | |
|     ASSERT(current);
 | |
|     InterruptDisabler disabler;
 | |
|     current->tss().cr3 = process.page_directory().cr3();
 | |
|     asm volatile("movl %%eax, %%cr3"::"a"(process.page_directory().cr3()):"memory");
 | |
| }
 | |
| 
 | |
| void MemoryManager::enter_kernel_paging_scope()
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
|     asm volatile("movl %%eax, %%cr3"::"a"(kernel_page_directory().cr3()):"memory");
 | |
| }
 | |
| 
 | |
| void MemoryManager::flush_entire_tlb()
 | |
| {
 | |
|     asm volatile(
 | |
|         "mov %%cr3, %%eax\n"
 | |
|         "mov %%eax, %%cr3\n"
 | |
|         ::: "%eax", "memory"
 | |
|     );
 | |
| }
 | |
| 
 | |
| void MemoryManager::flush_tlb(LinearAddress laddr)
 | |
| {
 | |
|     asm volatile("invlpg %0": :"m" (*(char*)laddr.get()) : "memory");
 | |
| }
 | |
| 
 | |
| void MemoryManager::map_for_kernel(LinearAddress laddr, PhysicalAddress paddr)
 | |
| {
 | |
|     auto pte = ensure_pte(kernel_page_directory(), laddr);
 | |
|     pte.set_physical_page_base(paddr.get());
 | |
|     pte.set_present(true);
 | |
|     pte.set_writable(true);
 | |
|     pte.set_user_allowed(false);
 | |
|     flush_tlb(laddr);
 | |
| }
 | |
| 
 | |
| byte* MemoryManager::quickmap_page(PhysicalPage& physical_page)
 | |
| {
 | |
|     ASSERT_INTERRUPTS_DISABLED();
 | |
|     ASSERT(!m_quickmap_in_use);
 | |
|     m_quickmap_in_use = true;
 | |
|     auto page_laddr = m_quickmap_addr;
 | |
|     auto pte = ensure_pte(kernel_page_directory(), page_laddr);
 | |
|     pte.set_physical_page_base(physical_page.paddr().get());
 | |
|     pte.set_present(true);
 | |
|     pte.set_writable(true);
 | |
|     pte.set_user_allowed(false);
 | |
|     flush_tlb(page_laddr);
 | |
|     ASSERT((dword)pte.physical_page_base() == physical_page.paddr().get());
 | |
| #ifdef MM_DEBUG
 | |
|     dbgprintf("MM: >> quickmap_page L%x => P%x @ PTE=%p\n", page_laddr, physical_page.paddr().get(), pte.ptr());
 | |
| #endif
 | |
|     return page_laddr.as_ptr();
 | |
| }
 | |
| 
 | |
| void MemoryManager::unquickmap_page()
 | |
| {
 | |
|     ASSERT_INTERRUPTS_DISABLED();
 | |
|     ASSERT(m_quickmap_in_use);
 | |
|     auto page_laddr = m_quickmap_addr;
 | |
|     auto pte = ensure_pte(kernel_page_directory(), page_laddr);
 | |
| #ifdef MM_DEBUG
 | |
|     auto old_physical_address = pte.physical_page_base();
 | |
| #endif
 | |
|     pte.set_physical_page_base(0);
 | |
|     pte.set_present(false);
 | |
|     pte.set_writable(false);
 | |
|     flush_tlb(page_laddr);
 | |
| #ifdef MM_DEBUG
 | |
|     dbgprintf("MM: >> unquickmap_page L%x =/> P%x\n", page_laddr, old_physical_address);
 | |
| #endif
 | |
|     m_quickmap_in_use = false;
 | |
| }
 | |
| 
 | |
| void MemoryManager::remap_region_page(Region& region, unsigned page_index_in_region, bool user_allowed)
 | |
| {
 | |
|     ASSERT(region.page_directory());
 | |
|     InterruptDisabler disabler;
 | |
|     auto page_laddr = region.laddr().offset(page_index_in_region * PAGE_SIZE);
 | |
|     auto pte = ensure_pte(*region.page_directory(), page_laddr);
 | |
|     auto& physical_page = region.vmo().physical_pages()[page_index_in_region];
 | |
|     ASSERT(physical_page);
 | |
|     pte.set_physical_page_base(physical_page->paddr().get());
 | |
|     pte.set_present(true); // FIXME: Maybe we should use the is_readable flag here?
 | |
|     if (region.should_cow(page_index_in_region))
 | |
|         pte.set_writable(false);
 | |
|     else
 | |
|         pte.set_writable(region.is_writable());
 | |
|     pte.set_cache_disabled(!region.vmo().m_allow_cpu_caching);
 | |
|     pte.set_write_through(!region.vmo().m_allow_cpu_caching);
 | |
|     pte.set_user_allowed(user_allowed);
 | |
|     region.page_directory()->flush(page_laddr);
 | |
| #ifdef MM_DEBUG
 | |
|     dbgprintf("MM: >> remap_region_page (PD=%x, PTE=P%x) '%s' L%x => P%x (@%p)\n", region.page_directory()->cr3(), pte.ptr(), region.name().characters(), page_laddr.get(), physical_page->paddr().get(), physical_page.ptr());
 | |
| #endif
 | |
| }
 | |
| 
 | |
| void MemoryManager::remap_region(PageDirectory& page_directory, Region& region)
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
|     ASSERT(region.page_directory() == &page_directory);
 | |
|     map_region_at_address(page_directory, region, region.laddr(), true);
 | |
| }
 | |
| 
 | |
| void MemoryManager::map_region_at_address(PageDirectory& page_directory, Region& region, LinearAddress laddr, bool user_allowed)
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
|     region.set_page_directory(page_directory);
 | |
|     auto& vmo = region.vmo();
 | |
| #ifdef MM_DEBUG
 | |
|     dbgprintf("MM: map_region_at_address will map VMO pages %u - %u (VMO page count: %u)\n", region.first_page_index(), region.last_page_index(), vmo.page_count());
 | |
| #endif
 | |
|     for (size_t i = 0; i < region.page_count(); ++i) {
 | |
|         auto page_laddr = laddr.offset(i * PAGE_SIZE);
 | |
|         auto pte = ensure_pte(page_directory, page_laddr);
 | |
|         auto& physical_page = vmo.physical_pages()[region.first_page_index() + i];
 | |
|         if (physical_page) {
 | |
|             pte.set_physical_page_base(physical_page->paddr().get());
 | |
|             pte.set_present(true); // FIXME: Maybe we should use the is_readable flag here?
 | |
|             // FIXME: It seems wrong that the *region* cow map is essentially using *VMO* relative indices.
 | |
|             if (region.should_cow(region.first_page_index() + i))
 | |
|                 pte.set_writable(false);
 | |
|             else
 | |
|                 pte.set_writable(region.is_writable());
 | |
|             pte.set_cache_disabled(!region.vmo().m_allow_cpu_caching);
 | |
|             pte.set_write_through(!region.vmo().m_allow_cpu_caching);
 | |
|         } else {
 | |
|             pte.set_physical_page_base(0);
 | |
|             pte.set_present(false);
 | |
|             pte.set_writable(region.is_writable());
 | |
|         }
 | |
|         pte.set_user_allowed(user_allowed);
 | |
|         page_directory.flush(page_laddr);
 | |
| #ifdef MM_DEBUG
 | |
|         dbgprintf("MM: >> map_region_at_address (PD=%x) '%s' L%x => P%x (@%p)\n", &page_directory, region.name().characters(), page_laddr, physical_page ? physical_page->paddr().get() : 0, physical_page.ptr());
 | |
| #endif
 | |
|     }
 | |
| }
 | |
| 
 | |
| bool MemoryManager::unmap_region(Region& region)
 | |
| {
 | |
|     ASSERT(region.page_directory());
 | |
|     InterruptDisabler disabler;
 | |
|     for (size_t i = 0; i < region.page_count(); ++i) {
 | |
|         auto laddr = region.laddr().offset(i * PAGE_SIZE);
 | |
|         auto pte = ensure_pte(*region.page_directory(), laddr);
 | |
|         pte.set_physical_page_base(0);
 | |
|         pte.set_present(false);
 | |
|         pte.set_writable(false);
 | |
|         pte.set_user_allowed(false);
 | |
|         region.page_directory()->flush(laddr);
 | |
| #ifdef MM_DEBUG
 | |
|         auto& physical_page = region.vmo().physical_pages()[region.first_page_index() + i];
 | |
|         dbgprintf("MM: >> Unmapped L%x => P%x <<\n", laddr, physical_page ? physical_page->paddr().get() : 0);
 | |
| #endif
 | |
|     }
 | |
|     region.release_page_directory();
 | |
|     return true;
 | |
| }
 | |
| 
 | |
| bool MemoryManager::map_region(Process& process, Region& region)
 | |
| {
 | |
|     map_region_at_address(process.page_directory(), region, region.laddr(), true);
 | |
|     return true;
 | |
| }
 | |
| 
 | |
| bool MemoryManager::validate_user_read(const Process& process, LinearAddress laddr) const
 | |
| {
 | |
|     auto* region = region_from_laddr(process, laddr);
 | |
|     return region && region->is_readable();
 | |
| }
 | |
| 
 | |
| bool MemoryManager::validate_user_write(const Process& process, LinearAddress laddr) const
 | |
| {
 | |
|     auto* region = region_from_laddr(process, laddr);
 | |
|     return region && region->is_writable();
 | |
| }
 | |
| 
 | |
| void MemoryManager::register_vmo(VMObject& vmo)
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
|     m_vmos.set(&vmo);
 | |
| }
 | |
| 
 | |
| void MemoryManager::unregister_vmo(VMObject& vmo)
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
|     m_vmos.remove(&vmo);
 | |
| }
 | |
| 
 | |
| void MemoryManager::register_region(Region& region)
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
|     if (region.laddr().get() >= 0xc0000000)
 | |
|         m_kernel_regions.set(®ion);
 | |
|     else
 | |
|         m_user_regions.set(®ion);
 | |
| }
 | |
| 
 | |
| void MemoryManager::unregister_region(Region& region)
 | |
| {
 | |
|     InterruptDisabler disabler;
 | |
|     if (region.laddr().get() >= 0xc0000000)
 | |
|         m_kernel_regions.remove(®ion);
 | |
|     else
 | |
|         m_user_regions.remove(®ion);
 | |
| }
 | |
| 
 | |
| ProcessPagingScope::ProcessPagingScope(Process& process)
 | |
| {
 | |
|     ASSERT(current);
 | |
|     MM.enter_process_paging_scope(process);
 | |
| }
 | |
| 
 | |
| ProcessPagingScope::~ProcessPagingScope()
 | |
| {
 | |
|     MM.enter_process_paging_scope(current->process());
 | |
| }
 | |
| 
 | |
| KernelPagingScope::KernelPagingScope()
 | |
| {
 | |
|     ASSERT(current);
 | |
|     MM.enter_kernel_paging_scope();
 | |
| }
 | |
| 
 | |
| KernelPagingScope::~KernelPagingScope()
 | |
| {
 | |
|     MM.enter_process_paging_scope(current->process());
 | |
| }
 |