mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-12-23 04:41:12 +00:00
69 lines
2.1 KiB
C++
69 lines
2.1 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/>.
|
|
*/
|
|
#include <haze.hpp>
|
|
|
|
namespace haze {
|
|
|
|
namespace {
|
|
|
|
/* Allow 30MiB for use by libnx. */
|
|
static constexpr size_t LibnxReservedMemorySize = 30_MB;
|
|
|
|
}
|
|
|
|
void PtpObjectHeap::Initialize() {
|
|
/* If we're already initialized, skip re-initialization. */
|
|
if (m_heap_block_size != 0) {
|
|
return;
|
|
}
|
|
|
|
/* Estimate how much memory we can reserve. */
|
|
size_t mem_used = 0;
|
|
HAZE_R_ABORT_UNLESS(svcGetInfo(std::addressof(mem_used), InfoType_UsedMemorySize, svc::CurrentProcess, 0));
|
|
HAZE_ASSERT(mem_used > LibnxReservedMemorySize);
|
|
mem_used -= LibnxReservedMemorySize;
|
|
|
|
/* Calculate size of blocks. */
|
|
m_heap_block_size = mem_used / NumHeapBlocks;
|
|
HAZE_ASSERT(m_heap_block_size > 0);
|
|
|
|
/* Allocate the memory. */
|
|
for (size_t i = 0; i < NumHeapBlocks; i++) {
|
|
m_heap_blocks[i] = std::malloc(m_heap_block_size);
|
|
HAZE_ASSERT(m_heap_blocks[i] != nullptr);
|
|
}
|
|
|
|
/* Set the address to allocate from. */
|
|
m_next_address = m_heap_blocks[0];
|
|
}
|
|
|
|
void PtpObjectHeap::Finalize() {
|
|
if (m_heap_block_size == 0) {
|
|
return;
|
|
}
|
|
|
|
/* Tear down the heap, allowing a subsequent call to Initialize() if desired. */
|
|
for (size_t i = 0; i < NumHeapBlocks; i++) {
|
|
std::free(m_heap_blocks[i]);
|
|
m_heap_blocks[i] = nullptr;
|
|
}
|
|
|
|
m_next_address = nullptr;
|
|
m_heap_block_size = 0;
|
|
m_current_heap_block = 0;
|
|
}
|
|
|
|
}
|