This repository has been archived on 2022-05-21. You can view files and clone it, but cannot push or open issues or pull requests.
LEFTONE/util/memory/memory.c

82 lines
1.7 KiB
C
Raw Permalink Normal View History

#include "./memory.h"
#include <assert.h>
#include <stdarg.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#ifdef __linux__
# include <execinfo.h>
#endif
#define MEMORY_LOG_FILE "memory-trace"
void* memory_new(size_t size) {
if (size == 0) return NULL;
void* ptr = malloc(size);
if (ptr == NULL) {
fprintf(stderr, "failed to allocate %zu bytes memory", size);
abort();
}
memory_log("new 0x%tx (%zu bytes)", (ptrdiff_t) ptr, size);
return ptr;
}
void* memory_resize(void* ptr, size_t size) {
if (ptr == NULL) return memory_new(size);
if (size == 0) return ptr;
void* newptr = realloc(ptr, size);
if (newptr == NULL) {
fprintf(stderr,
"failed to allocate %zu bytes memory for resizing 0x%tx",
size, (ptrdiff_t) ptr);
abort();
}
memory_log("resize 0x%tx -> 0x%tx (%zu bytes)",
(ptrdiff_t) ptr, (ptrdiff_t) newptr, size);
return newptr;
}
void memory_delete(void* ptr) {
if (ptr == NULL) return;
free(ptr);
memory_log("delete 0x%tx", (ptrdiff_t) ptr);
}
#ifndef NDEBUG
void memory_log(const char* fmt, ...) {
static FILE* fp = NULL;
if (fp == NULL) {
fp = fopen(MEMORY_LOG_FILE, "w");
assert(fp != NULL);
}
va_list args;
va_start(args, fmt);
vfprintf(fp, fmt, args);
va_end(args);
fprintf(fp, "\n");
# if __linux__
void* callstack[32];
const size_t frames = backtrace(callstack, sizeof(callstack));
char** symbols = backtrace_symbols(callstack, frames);
for (size_t i = 0; i < frames; ++i) {
fprintf(fp, " %s\n", symbols[i]);
}
free(symbols);
# endif /* __linux__ */
fprintf(fp, "\n");
fflush(fp);
}
#endif /* NDEBUG */