140 lines
3.8 KiB
C++
140 lines
3.8 KiB
C++
/****************************************************************************
|
|
* Copyright (C) 2015 Dimok
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that 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 "logger.h"
|
|
#include <cerrno>
|
|
#include <coreinit/cache.h>
|
|
#include <coreinit/memdefaultheap.h>
|
|
#include <coreinit/memexpheap.h>
|
|
#include <coreinit/memorymap.h>
|
|
#include <cstring>
|
|
#include <malloc.h>
|
|
|
|
extern MEMHeapHandle gHeapHandle;
|
|
|
|
void *MEMAllocSafe(uint32_t size, uint32_t align) {
|
|
void *res = nullptr;
|
|
MEMHeapHandle heapHandle = gHeapHandle;
|
|
res = MEMAllocFromExpHeapEx(heapHandle, size, align);
|
|
return res;
|
|
}
|
|
|
|
void *MemoryAlloc(uint32_t size) {
|
|
void *res = MEMAllocSafe(size, 4);
|
|
if (res == nullptr) {
|
|
OSFatal("Failed to MemoryAlloc");
|
|
}
|
|
return res;
|
|
}
|
|
|
|
void *MemoryAllocEx(uint32_t size, uint32_t align) {
|
|
void *res = MEMAllocSafe(size, align);
|
|
if (res == nullptr) {
|
|
OSFatal("Failed to MemoryAllocEX");
|
|
}
|
|
return res;
|
|
}
|
|
|
|
void MemoryFree(void *ptr) {
|
|
if (ptr) {
|
|
MEMFreeToExpHeap(gHeapHandle, ptr);
|
|
} else {
|
|
OSFatal("Failed to free");
|
|
}
|
|
}
|
|
|
|
uint32_t MEMAlloc __attribute__((__section__(".data"))) = (uint32_t) MemoryAlloc;
|
|
uint32_t MEMAllocEx __attribute__((__section__(".data"))) = (uint32_t) MemoryAllocEx;
|
|
uint32_t MEMFree __attribute__((__section__(".data"))) = (uint32_t) MemoryFree;
|
|
|
|
//!-------------------------------------------------------------------------------------------
|
|
//! reent versions
|
|
//!-------------------------------------------------------------------------------------------
|
|
void *_malloc_r(struct _reent *r, size_t size) {
|
|
void *ptr = MemoryAllocEx(size, 4);
|
|
if (!ptr) {
|
|
r->_errno = ENOMEM;
|
|
}
|
|
return ptr;
|
|
}
|
|
|
|
void *_calloc_r(struct _reent *r, size_t num, size_t size) {
|
|
void *ptr = MemoryAllocEx(num * size, 4);
|
|
if (ptr) {
|
|
memset(ptr, 0, num * size);
|
|
} else {
|
|
r->_errno = ENOMEM;
|
|
}
|
|
|
|
return ptr;
|
|
}
|
|
|
|
void *_memalign_r(struct _reent *r, size_t align, size_t size) {
|
|
return MemoryAllocEx(size, align);
|
|
}
|
|
|
|
void _free_r(struct _reent *r, void *ptr) {
|
|
if (ptr) {
|
|
MemoryFree(ptr);
|
|
}
|
|
}
|
|
|
|
void *_realloc_r(struct _reent *r, void *p, size_t size) {
|
|
void *new_ptr = MemoryAllocEx(size, 4);
|
|
if (!new_ptr) {
|
|
r->_errno = ENOMEM;
|
|
return new_ptr;
|
|
}
|
|
|
|
if (p) {
|
|
size_t old_size = MEMGetSizeForMBlockExpHeap(p);
|
|
memcpy(new_ptr, p, old_size <= size ? old_size : size);
|
|
MemoryFree(p);
|
|
}
|
|
return new_ptr;
|
|
}
|
|
|
|
struct mallinfo _mallinfo_r(struct _reent *r) {
|
|
struct mallinfo info = {0};
|
|
return info;
|
|
}
|
|
|
|
void _malloc_stats_r(struct _reent *r) {
|
|
}
|
|
|
|
int _mallopt_r(struct _reent *r, int param, int value) {
|
|
return 0;
|
|
}
|
|
|
|
size_t
|
|
_malloc_usable_size_r(struct _reent *r, void *ptr) {
|
|
return MEMGetSizeForMBlockExpHeap(ptr);
|
|
}
|
|
|
|
void *
|
|
_valloc_r(struct _reent *r, size_t size) {
|
|
return MemoryAllocEx(size, OS_PAGE_SIZE);
|
|
}
|
|
|
|
void *
|
|
_pvalloc_r(struct _reent *r, size_t size) {
|
|
return MemoryAllocEx((size + (OS_PAGE_SIZE - 1)) & ~(OS_PAGE_SIZE - 1), OS_PAGE_SIZE);
|
|
}
|
|
|
|
int _malloc_trim_r(struct _reent *r, size_t pad) {
|
|
return 0;
|
|
}
|