109 lines
3 KiB
C
109 lines
3 KiB
C
|
/* deos_malloc.c
|
||
|
|
*
|
||
|
|
* Copyright (C) 2006-2020 wolfSSL Inc.
|
||
|
|
*
|
||
|
|
* This file is part of wolfSSL.
|
||
|
|
*
|
||
|
|
* wolfSSL 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 2 of the License, or
|
||
|
|
* (at your option) any later version.
|
||
|
|
*
|
||
|
|
* wolfSSL 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, write to the Free Software
|
||
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA
|
||
|
|
*/
|
||
|
|
#include <wolfssl/wolfcrypt/settings.h>
|
||
|
|
#include <wolfssl/ssl.h>
|
||
|
|
|
||
|
|
#define ROUND_UP(x, align) (((int) (x) + (align - 1)) & ~(align - 1))
|
||
|
|
#define SIZEOF_HEADER sizeof(size_t) /* tracks size of allocated block */
|
||
|
|
|
||
|
|
#define HEAP_SIZE_MAX (1*1024*1024)
|
||
|
|
|
||
|
|
static size_t allocatedMemory = 0;
|
||
|
|
|
||
|
|
size_t getMemAllocatedSize_deos(size_t* size){
|
||
|
|
|
||
|
|
if (size)
|
||
|
|
*size = allocatedMemory;
|
||
|
|
|
||
|
|
return allocatedMemory;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Simply returns without freeing any memory. */
|
||
|
|
|
||
|
|
void free_deos(void *ptr) {
|
||
|
|
//printf("fake free_deos()\n");
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
void *realloc_deos(void *ptr, size_t size) {
|
||
|
|
void *newptr;
|
||
|
|
|
||
|
|
if (size == 0)
|
||
|
|
return ptr;
|
||
|
|
newptr = malloc_deos(size);
|
||
|
|
|
||
|
|
if (ptr != NULL && newptr != NULL) {
|
||
|
|
|
||
|
|
if ( *((char *)ptr - SIZEOF_HEADER) < *((char *)newptr - SIZEOF_HEADER))
|
||
|
|
size = *((char *)ptr - SIZEOF_HEADER);
|
||
|
|
|
||
|
|
XMEMCPY((char *) newptr, (const char *) ptr, size);
|
||
|
|
free_deos(ptr);
|
||
|
|
}
|
||
|
|
|
||
|
|
return newptr;
|
||
|
|
}
|
||
|
|
|
||
|
|
void *malloc_deos(size_t size) {
|
||
|
|
PDEOS_SYSTEM_INFO systemInfoPtr;
|
||
|
|
static VirtualAddressTYP heapAddr = NULL;
|
||
|
|
static VirtualAddressTYP freeAddr = NULL;
|
||
|
|
VirtualAddressTYP retAddr = NULL;
|
||
|
|
DWORD allocationSize = 0;
|
||
|
|
static int initialized = 0;
|
||
|
|
|
||
|
|
if (size <= 0)
|
||
|
|
return NULL;
|
||
|
|
|
||
|
|
if (!initialized) {
|
||
|
|
systemInfoPtr = (PDEOS_SYSTEM_INFO)getSystemInfoDEOS();
|
||
|
|
freeAddr = (VirtualAddressTYP)getNextLibraryStartAddress();
|
||
|
|
allocationSize = (((HEAP_SIZE_MAX - 1) / systemInfoPtr->dwPageSize) + 1) *
|
||
|
|
systemInfoPtr->dwPageSize;
|
||
|
|
|
||
|
|
if (virtualAllocDEOS(freeAddr, allocationSize) != allocSuccess){
|
||
|
|
printf("ERROR: virtualAllocDEOS failed\n");
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
setNextLibraryStartAddress(freeAddr + allocationSize);
|
||
|
|
heapAddr = freeAddr;
|
||
|
|
|
||
|
|
initialized = 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
size = ROUND_UP(size, sizeof(size_t));
|
||
|
|
|
||
|
|
if ((size + SIZEOF_HEADER) > (HEAP_SIZE_MAX - (freeAddr - heapAddr))){
|
||
|
|
printf("ERROR: malloc_deos cannot allocate from heap memory anymore\n");
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
*freeAddr = size;
|
||
|
|
freeAddr += SIZEOF_HEADER;
|
||
|
|
retAddr = freeAddr;
|
||
|
|
XMEMSET(retAddr, 0, size);
|
||
|
|
freeAddr += size;
|
||
|
|
allocatedMemory += size;
|
||
|
|
|
||
|
|
return retAddr;
|
||
|
|
}
|