2013-06-27 08:43:07 +02:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2013 Joris Vink <joris@coders.se>
|
|
|
|
*
|
|
|
|
* Permission to use, copy, modify, and distribute this software for any
|
|
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
|
|
* copyright notice and this permission notice appear in all copies.
|
|
|
|
*
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
|
|
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
|
|
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
*/
|
|
|
|
|
2013-07-22 22:44:42 +02:00
|
|
|
#include <sys/param.h>
|
|
|
|
|
2013-07-13 21:47:04 +02:00
|
|
|
#include <stdlib.h>
|
|
|
|
|
2013-06-27 08:43:07 +02:00
|
|
|
#include "kore.h"
|
|
|
|
|
|
|
|
#define KORE_MEM_MAGIC 0xd0d0
|
2013-07-13 22:19:50 +02:00
|
|
|
#define KORE_MEMSIZE(x) \
|
|
|
|
(*(u_int32_t *)((u_int8_t *)x - sizeof(u_int32_t)))
|
2013-06-27 08:43:07 +02:00
|
|
|
#define KORE_MEMINFO(x) \
|
2013-07-13 22:19:50 +02:00
|
|
|
(struct meminfo *)((u_int8_t *)x + KORE_MEMSIZE(x))
|
2013-06-27 08:43:07 +02:00
|
|
|
|
|
|
|
struct meminfo {
|
2013-06-27 12:27:17 +02:00
|
|
|
u_int16_t magic;
|
2014-04-18 17:41:56 +02:00
|
|
|
};
|
2013-06-27 08:43:07 +02:00
|
|
|
|
|
|
|
void
|
|
|
|
kore_mem_init(void)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
void *
|
|
|
|
kore_malloc(size_t len)
|
|
|
|
{
|
|
|
|
size_t mlen;
|
2013-07-13 22:19:50 +02:00
|
|
|
void *ptr;
|
2013-06-27 08:43:07 +02:00
|
|
|
struct meminfo *mem;
|
2013-07-13 22:19:50 +02:00
|
|
|
u_int8_t *addr;
|
|
|
|
u_int32_t *plen;
|
2013-06-27 08:43:07 +02:00
|
|
|
|
2013-07-13 22:19:50 +02:00
|
|
|
mlen = sizeof(u_int32_t) + len + sizeof(struct meminfo);
|
|
|
|
if ((ptr = malloc(mlen)) == NULL)
|
2013-06-27 08:43:07 +02:00
|
|
|
fatal("kore_malloc(%d): %d", len, errno);
|
|
|
|
|
2013-07-13 22:19:50 +02:00
|
|
|
plen = (u_int32_t *)ptr;
|
|
|
|
*plen = len;
|
|
|
|
addr = (u_int8_t *)ptr + sizeof(u_int32_t);
|
|
|
|
|
|
|
|
mem = KORE_MEMINFO(addr);
|
2013-06-27 08:43:07 +02:00
|
|
|
mem->magic = KORE_MEM_MAGIC;
|
|
|
|
|
2014-04-22 12:46:23 +02:00
|
|
|
#if defined(KORE_PEDANTIC_MALLOC)
|
|
|
|
explicit_bzero(addr, len);
|
|
|
|
#endif
|
|
|
|
|
2013-07-13 22:19:50 +02:00
|
|
|
return (addr);
|
2013-06-27 08:43:07 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void *
|
|
|
|
kore_realloc(void *ptr, size_t len)
|
|
|
|
{
|
|
|
|
struct meminfo *mem;
|
|
|
|
void *nptr;
|
|
|
|
|
2013-06-27 09:20:48 +02:00
|
|
|
if (ptr == NULL) {
|
|
|
|
nptr = kore_malloc(len);
|
|
|
|
} else {
|
|
|
|
mem = KORE_MEMINFO(ptr);
|
|
|
|
if (mem->magic != KORE_MEM_MAGIC)
|
|
|
|
fatal("kore_realloc(): magic boundary not found");
|
|
|
|
|
|
|
|
nptr = kore_malloc(len);
|
2013-07-22 22:44:42 +02:00
|
|
|
memcpy(nptr, ptr, MIN(len, KORE_MEMSIZE(ptr)));
|
2013-06-27 09:20:48 +02:00
|
|
|
kore_mem_free(ptr);
|
|
|
|
}
|
|
|
|
|
|
|
|
return (nptr);
|
2013-06-27 08:43:07 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void *
|
|
|
|
kore_calloc(size_t memb, size_t len)
|
|
|
|
{
|
|
|
|
return (kore_malloc(memb * len));
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
kore_mem_free(void *ptr)
|
|
|
|
{
|
2013-07-13 22:19:50 +02:00
|
|
|
u_int8_t *addr;
|
2013-06-27 08:43:07 +02:00
|
|
|
struct meminfo *mem;
|
|
|
|
|
|
|
|
mem = KORE_MEMINFO(ptr);
|
|
|
|
if (mem->magic != KORE_MEM_MAGIC)
|
|
|
|
fatal("kore_mem_free(): magic boundary not found");
|
|
|
|
|
2014-04-22 12:46:23 +02:00
|
|
|
#if defined(KORE_PEDANTIC_MALLOC)
|
|
|
|
explicit_bzero(ptr, KORE_MEMSIZE(ptr));
|
|
|
|
#endif
|
|
|
|
|
2013-07-13 22:19:50 +02:00
|
|
|
addr = (u_int8_t *)ptr - sizeof(u_int32_t);
|
|
|
|
free(addr);
|
2013-06-27 08:43:07 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
char *
|
|
|
|
kore_strdup(const char *str)
|
|
|
|
{
|
|
|
|
size_t len;
|
|
|
|
char *nstr;
|
|
|
|
|
|
|
|
len = strlen(str) + 1;
|
2013-07-13 21:08:55 +02:00
|
|
|
nstr = kore_malloc(len);
|
2013-06-27 08:43:07 +02:00
|
|
|
kore_strlcpy(nstr, str, len);
|
|
|
|
|
|
|
|
return (nstr);
|
|
|
|
}
|
2014-04-22 12:46:23 +02:00
|
|
|
|
|
|
|
void
|
|
|
|
explicit_bzero(void *addr, size_t len)
|
|
|
|
{
|
|
|
|
bzero(addr, len);
|
|
|
|
}
|