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

BAN: Add allocator/deallocator references for kernel/userspace

Browse Source
This commit is contained in:
Bananymous
2022-12-13 13:40:54 +02:00
parent 52c793bb5c
commit b49a873d54
2 changed files with 25 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
BAN/include/BAN/Memory.h Normal file
View File

@@ -0,0 +1,18 @@
#pragma once
#if defined(__is_kernel)
#include <kernel/kmalloc.h>
#else
#include <stdlib.h>
#endif
namespace BAN
{
#if defined(__is_kernel)
static constexpr auto& allocator = kmalloc;
static constexpr auto& deallocator = kfree;
#else
static constexpr auto& allocator = malloc;
static constexpr auto& deallocator = free;
#endif
}

27
BAN/include/BAN/Queue.h
View File

@@ -1,12 +1,7 @@
#pragma once
#include <BAN/Errors.h>
#if defined(__is_kernel)
#include <kernel/kmalloc.h>
#else
#include <stdlib.h>
#endif
#include <BAN/Memory.h>
#include <assert.h>
#include <stdint.h>
@@ -19,16 +14,6 @@ namespace BAN
template<typename T>
class Queue
{
private:
#if defined(__is_kernel)
static constexpr auto& allocator = kmalloc;
static constexpr auto& deallocator = kfree;
#else
static constexpr auto& allocator = malloc;
static constexpr auto& deallocator = free;
#endif
public:
using size_type = uint32_t;
using value_type = T;
@@ -58,7 +43,9 @@ namespace BAN
template<typename T>
Queue<T>::~Queue()
{
Queue<T>::deallocator(m_data);
for (size_type i = 0; i < m_size; i++)
m_data[i].~T();
BAN::deallocator(m_data);
}
template<typename T>
@@ -110,12 +97,12 @@ namespace BAN
return {};
size_type new_cap = MAX(m_capacity * 1.5f, m_capacity + 1);
void* new_data = Queue<T>::allocator(new_cap * sizeof(T));
void* new_data = BAN::allocator(new_cap * sizeof(T));
if (new_data == nullptr)
return Error::FromString("Queue: out of memory");
return Error::FromString("Queue: Could not allocate memory");
memcpy(new_data, m_data, m_size * sizeof(T));
Queue<T>::deallocator(m_data);
BAN::deallocator(m_data);
m_data = (T*)new_data;
m_capacity = new_cap;