RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-26 02:07:35 +00:00
Code Issues Activity Actions

AK: Allow configuring the BumpAllocator chunk size

Browse source
This commit is contained in:
Timothy Flynn 2021-08-02 15:04:15 -04:00 • committed by Andreas Kling
parent 510bbcd8e0
commit dae7674ca9
1 changed files with 10 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

18
AK/BumpAllocator.h
View file

@ -13,15 +13,15 @@
namespace AK { namespace AK {
template<bool use_mmap = false> template<bool use_mmap = false, size_t chunk_size = use_mmap ? 4 * MiB : 4 * KiB>
class BumpAllocator { class BumpAllocator {
public: public:
BumpAllocator() BumpAllocator()
{ {
if constexpr (use_mmap) if constexpr (use_mmap)
m_chunk_size = 4 * MiB; m_chunk_size = chunk_size;
else else
m_chunk_size = kmalloc_good_size(4 * KiB); m_chunk_size = kmalloc_good_size(chunk_size);
} }
~BumpAllocator() ~BumpAllocator()
@ -137,8 +137,10 @@ protected:
// size_t m_allocations_in_previous_chunk { 0 }; // size_t m_allocations_in_previous_chunk { 0 };
}; };
template<typename T, bool use_mmap = false> template<typename T, bool use_mmap = false, size_t chunk_size = use_mmap ? 4 * MiB : 4 * KiB>
class UniformBumpAllocator : protected BumpAllocator<use_mmap> { class UniformBumpAllocator : protected BumpAllocator<use_mmap, chunk_size> {
using Allocator = BumpAllocator<use_mmap, chunk_size>;
public: public:
UniformBumpAllocator() = default; UniformBumpAllocator() = default;
~UniformBumpAllocator() ~UniformBumpAllocator()
@ -148,19 +150,19 @@ public:
T* allocate() T* allocate()
{ {
return BumpAllocator<use_mmap>::template allocate<T>(); return Allocator::template allocate<T>();
} }
void deallocate_all() void deallocate_all()
{ {
destroy_all(); destroy_all();
BumpAllocator<use_mmap>::deallocate_all(); Allocator::deallocate_all();
} }
void destroy_all() void destroy_all()
{ {
this->for_each_chunk([&](auto chunk) { this->for_each_chunk([&](auto chunk) {
auto base_ptr = align_up_to(chunk + sizeof(typename BumpAllocator<use_mmap>::ChunkHeader), alignof(T)); auto base_ptr = align_up_to(chunk + sizeof(typename Allocator::ChunkHeader), alignof(T));
FlatPtr end_offset = this->m_chunk_size; FlatPtr end_offset = this->m_chunk_size;
if (chunk == this->m_current_chunk) if (chunk == this->m_current_chunk)
end_offset = this->m_byte_offset_into_current_chunk; end_offset = this->m_byte_offset_into_current_chunk;