Sinipelto
/
banan-os
forked from Bananymous/banan-os
1
0
Fork 0
Code Pull Requests Activity

Kernel: FixedWidthAllocator operates on MMU 

Instead of Process* we use MMU& in FixedWidthAllocator since it is
everything it actually needs :)
This commit is contained in:
Bananymous 2023-05-08 00:06:56 +03:00
parent f32d594879
commit 2d0a5a9e15
3 changed files with 21 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
kernel/include/kernel/Memory/FixedWidthAllocator.h
View File

@ -1,19 +1,18 @@
#pragma once #pragma once
#include <kernel/Memory/Heap.h> #include <kernel/Memory/Heap.h>
#include <kernel/Memory/MMU.h>
namespace Kernel namespace Kernel
{ {
class Process;
class FixedWidthAllocator class FixedWidthAllocator
{ {
BAN_NON_COPYABLE(FixedWidthAllocator); BAN_NON_COPYABLE(FixedWidthAllocator);
BAN_NON_MOVABLE(FixedWidthAllocator);
public: public:
FixedWidthAllocator(Process*, uint32_t); FixedWidthAllocator(MMU&, uint32_t);
FixedWidthAllocator(FixedWidthAllocator&&);
~FixedWidthAllocator(); ~FixedWidthAllocator();
vaddr_t allocate(); vaddr_t allocate();
@ -38,7 +37,7 @@ namespace Kernel
private: private:
static constexpr uint32_t m_min_allocation_size = 16; static constexpr uint32_t m_min_allocation_size = 16;
Process* m_process; MMU& m_mmu;
const uint32_t m_allocation_size; const uint32_t m_allocation_size;
vaddr_t m_nodes_page { 0 }; vaddr_t m_nodes_page { 0 };

47
kernel/kernel/Memory/FixedWidthAllocator.cpp
View File

@ -6,23 +6,23 @@
namespace Kernel namespace Kernel
{ {
FixedWidthAllocator::FixedWidthAllocator(Process* process, uint32_t allocation_size) FixedWidthAllocator::FixedWidthAllocator(MMU& mmu, uint32_t allocation_size)
: m_process(process) : m_mmu(mmu)
, m_allocation_size(BAN::Math::max(allocation_size, m_min_allocation_size)) , m_allocation_size(BAN::Math::max(allocation_size, m_min_allocation_size))
{ {
ASSERT(BAN::Math::is_power_of_two(allocation_size)); ASSERT(BAN::Math::is_power_of_two(allocation_size));
paddr_t nodes_paddr = Heap::get().take_free_page(); paddr_t nodes_paddr = Heap::get().take_free_page();
m_nodes_page = m_process->mmu().get_free_page(); m_nodes_page = m_mmu.get_free_page();
m_process->mmu().map_page_at(nodes_paddr, m_nodes_page, MMU::Flags::ReadWrite | MMU::Flags::Present); m_mmu.map_page_at(nodes_paddr, m_nodes_page, MMU::Flags::ReadWrite | MMU::Flags::Present);
paddr_t allocated_pages_paddr = Heap::get().take_free_page(); paddr_t allocated_pages_paddr = Heap::get().take_free_page();
m_allocated_pages = m_process->mmu().get_free_page(); m_allocated_pages = m_mmu.get_free_page();
m_process->mmu().map_page_at(allocated_pages_paddr, m_allocated_pages, MMU::Flags::ReadWrite | MMU::Flags::Present); m_mmu.map_page_at(allocated_pages_paddr, m_allocated_pages, MMU::Flags::ReadWrite | MMU::Flags::Present);
CriticalScope _; CriticalScope _;
m_process->mmu().load(); m_mmu.load();
memset((void*)m_nodes_page, 0, PAGE_SIZE); memset((void*)m_nodes_page, 0, PAGE_SIZE);
memset((void*)m_allocated_pages, 0, PAGE_SIZE); memset((void*)m_allocated_pages, 0, PAGE_SIZE);
@ -42,25 +42,10 @@ namespace Kernel
Process::current().mmu().load(); Process::current().mmu().load();
} }
FixedWidthAllocator::FixedWidthAllocator(FixedWidthAllocator&& other)
: m_process(other.m_process)
, m_allocation_size(other.m_allocation_size)
, m_nodes_page(other.m_nodes_page)
, m_allocated_pages(other.m_allocated_pages)
, m_free_list(other.m_free_list)
, m_used_list(other.m_used_list)
, m_allocations(other.m_allocations)
{
other.m_process = nullptr;
}
FixedWidthAllocator::~FixedWidthAllocator() FixedWidthAllocator::~FixedWidthAllocator()
{ {
if (m_process == nullptr) Heap::get().release_page(m_mmu.physical_address_of(m_nodes_page));
return; m_mmu.unmap_page(m_nodes_page);
Heap::get().release_page(m_process->mmu().physical_address_of(m_nodes_page));
m_process->mmu().unmap_page(m_nodes_page);
for (uint32_t page_index = 0; page_index < PAGE_SIZE / sizeof(vaddr_t); page_index++) for (uint32_t page_index = 0; page_index < PAGE_SIZE / sizeof(vaddr_t); page_index++)
{ {
@ -68,15 +53,15 @@ namespace Kernel
if (page_vaddr == 0) if (page_vaddr == 0)
continue; continue;
ASSERT(!m_process->mmu().is_page_free(page_vaddr)); ASSERT(!m_mmu.is_page_free(page_vaddr));
paddr_t page_paddr = m_process->mmu().physical_address_of(page_vaddr); paddr_t page_paddr = m_mmu.physical_address_of(page_vaddr);
Heap::get().release_page(page_paddr); Heap::get().release_page(page_paddr);
m_process->mmu().unmap_page(page_vaddr); m_mmu.unmap_page(page_vaddr);
} }
Heap::get().release_page(m_process->mmu().physical_address_of(m_allocated_pages)); Heap::get().release_page(m_mmu.physical_address_of(m_allocated_pages));
m_process->mmu().unmap_page(m_allocated_pages); m_mmu.unmap_page(m_allocated_pages);
} }
paddr_t FixedWidthAllocator::allocate() paddr_t FixedWidthAllocator::allocate()
@ -202,8 +187,8 @@ namespace Kernel
paddr_t page_paddr = Heap::get().take_free_page(); paddr_t page_paddr = Heap::get().take_free_page();
ASSERT(page_paddr); ASSERT(page_paddr);
page_vaddr = m_process->mmu().get_free_page(); page_vaddr = m_mmu.get_free_page();
m_process->mmu().map_page_at(page_paddr, page_vaddr, MMU::Flags::UserSupervisor | MMU::Flags::ReadWrite | MMU::Flags::Present); m_mmu.map_page_at(page_paddr, page_vaddr, MMU::Flags::UserSupervisor | MMU::Flags::ReadWrite | MMU::Flags::Present);
} }
} }

2
kernel/kernel/Process.cpp
View File

@ -397,7 +397,7 @@ namespace Kernel
if (allocator.allocation_size() == allocation_size && allocator.allocations() < allocator.max_allocations()) if (allocator.allocation_size() == allocation_size && allocator.allocations() < allocator.max_allocations())
return (void*)allocator.allocate(); return (void*)allocator.allocate();
MUST(m_fixed_width_allocators.emplace_back(this, allocation_size)); MUST(m_fixed_width_allocators.emplace_back(mmu(), allocation_size));
return (void*)m_fixed_width_allocators.back().allocate(); return (void*)m_fixed_width_allocators.back().allocate();
} }
else else