mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-12-24 05:11:14 +00:00
73 lines
2.4 KiB
C++
73 lines
2.4 KiB
C++
/*
|
|
* Copyright (c) Atmosphère-NX
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms and conditions of the GNU General Public License,
|
|
* version 2, as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#pragma once
|
|
#include <stratosphere/os.hpp>
|
|
#include <stratosphere/mem/impl/mem_impl_declarations.hpp>
|
|
|
|
namespace ams::mem {
|
|
|
|
class StandardAllocator {
|
|
NON_COPYABLE(StandardAllocator);
|
|
NON_MOVEABLE(StandardAllocator);
|
|
public:
|
|
using WalkCallback = int (*)(void *ptr, size_t size, void *user_data);
|
|
|
|
struct AllocatorHash {
|
|
size_t allocated_count;
|
|
size_t allocated_size;
|
|
size_t hash;
|
|
};
|
|
private:
|
|
bool m_initialized;
|
|
bool m_enable_thread_cache;
|
|
uintptr_t m_unused;
|
|
os::TlsSlot m_tls_slot;
|
|
impl::InternalCentralHeapStorage m_central_heap_storage;
|
|
public:
|
|
StandardAllocator();
|
|
StandardAllocator(void *mem, size_t size);
|
|
StandardAllocator(void *mem, size_t size, bool enable_cache);
|
|
|
|
~StandardAllocator() {
|
|
if (m_initialized) {
|
|
this->Finalize();
|
|
}
|
|
}
|
|
|
|
void Initialize(void *mem, size_t size);
|
|
void Initialize(void *mem, size_t size, bool enable_cache);
|
|
void Finalize();
|
|
|
|
void *Allocate(size_t size);
|
|
void *Allocate(size_t size, size_t alignment);
|
|
void Free(void *ptr);
|
|
void *Reallocate(void *ptr, size_t new_size);
|
|
size_t Shrink(void *ptr, size_t new_size);
|
|
|
|
void ClearThreadCache() const;
|
|
void CleanUpManagementArea() const;
|
|
|
|
size_t GetSizeOf(const void *ptr) const;
|
|
size_t GetTotalFreeSize() const;
|
|
size_t GetAllocatableSize() const;
|
|
|
|
void WalkAllocatedBlocks(WalkCallback callback, void *user_data) const;
|
|
|
|
void Dump() const;
|
|
AllocatorHash Hash() const;
|
|
};
|
|
|
|
}
|