Fix memory dump console printing alignment and also trace expressions for allocations.

This commit is contained in:
Dale Weiler 2013-10-17 04:21:25 -04:00
parent 05ac126d6f
commit d6f020fd6a
2 changed files with 40 additions and 28 deletions

10
gmqcc.h
View file

@ -256,14 +256,14 @@ GMQCC_IND_STRING(GMQCC_VERSION_PATCH) \
/* stat.c */ /* stat.c */
void stat_info (void); void stat_info (void);
char *stat_mem_strdup (const char *, size_t, const char *, bool); char *stat_mem_strdup (const char *, size_t, const char *, bool);
void *stat_mem_reallocate(void *, size_t, size_t, const char *); void *stat_mem_reallocate(void *, size_t, size_t, const char *, const char *);
void stat_mem_deallocate(void *); void stat_mem_deallocate(void *);
void *stat_mem_allocate (size_t, size_t, const char *); void *stat_mem_allocate (size_t, size_t, const char *, const char *);
#define mem_a(SIZE) stat_mem_allocate ((SIZE), __LINE__, __FILE__) #define mem_a(SIZE) stat_mem_allocate ((SIZE), __LINE__, __FILE__, #SIZE)
#define mem_d(PTRN) stat_mem_deallocate((void*)(PTRN)) #define mem_d(PTRN) stat_mem_deallocate((void*)(PTRN))
#define mem_r(PTRN, SIZE) stat_mem_reallocate((void*)(PTRN), (SIZE), __LINE__, __FILE__) #define mem_r(PTRN, SIZE) stat_mem_reallocate((void*)(PTRN), (SIZE), __LINE__, __FILE__, #SIZE)
#define mem_af(SIZE, FILE, LINE) stat_mem_allocate ((SIZE), (LINE), (FILE)) #define mem_af(SIZE, FILE, LINE) stat_mem_allocate ((SIZE), (LINE), (FILE), #SIZE)
/* TODO: rename to mem variations */ /* TODO: rename to mem variations */
#define util_strdup(SRC) stat_mem_strdup((char*)(SRC), __LINE__, __FILE__, false) #define util_strdup(SRC) stat_mem_strdup((char*)(SRC), __LINE__, __FILE__, false)

58
stat.c
View file

@ -56,6 +56,7 @@ typedef struct stat_mem_block_s {
const char *file; const char *file;
size_t line; size_t line;
size_t size; size_t size;
const char *expr;
struct stat_mem_block_s *next; struct stat_mem_block_s *next;
struct stat_mem_block_s *prev; struct stat_mem_block_s *prev;
} stat_mem_block_t; } stat_mem_block_t;
@ -119,7 +120,7 @@ static void stat_size_put(stat_size_table_t table, size_t key, size_t value) {
* information as a header, returns the memory + 1 past it, can be * information as a header, returns the memory + 1 past it, can be
* retrieved again with - 1. Where type is stat_mem_block_t*. * retrieved again with - 1. Where type is stat_mem_block_t*.
*/ */
void *stat_mem_allocate(size_t size, size_t line, const char *file) { void *stat_mem_allocate(size_t size, size_t line, const char *file, const char *expr) {
stat_mem_block_t *info = (stat_mem_block_t*)malloc(sizeof(stat_mem_block_t) + size); stat_mem_block_t *info = (stat_mem_block_t*)malloc(sizeof(stat_mem_block_t) + size);
void *data = (void*)(info + 1); void *data = (void*)(info + 1);
@ -129,6 +130,7 @@ void *stat_mem_allocate(size_t size, size_t line, const char *file) {
info->line = line; info->line = line;
info->size = size; info->size = size;
info->file = file; info->file = file;
info->expr = expr;
info->prev = NULL; info->prev = NULL;
info->next = stat_mem_block_root; info->next = stat_mem_block_root;
@ -193,12 +195,12 @@ void stat_mem_deallocate(void *ptr) {
VALGRIND_FREELIKE_BLOCK(ptr, sizeof(stat_mem_block_t)); VALGRIND_FREELIKE_BLOCK(ptr, sizeof(stat_mem_block_t));
} }
void *stat_mem_reallocate(void *ptr, size_t size, size_t line, const char *file) { void *stat_mem_reallocate(void *ptr, size_t size, size_t line, const char *file, const char *expr) {
stat_mem_block_t *oldinfo = NULL; stat_mem_block_t *oldinfo = NULL;
stat_mem_block_t *newinfo; stat_mem_block_t *newinfo;
if (GMQCC_UNLIKELY(!ptr)) if (GMQCC_UNLIKELY(!ptr))
return stat_mem_allocate(size, line, file); return stat_mem_allocate(size, line, file, expr);
/* stay consistent with glibc */ /* stay consistent with glibc */
if (GMQCC_UNLIKELY(!size)) { if (GMQCC_UNLIKELY(!size)) {
@ -247,6 +249,7 @@ void *stat_mem_reallocate(void *ptr, size_t size, size_t line, const char *file)
newinfo->line = line; newinfo->line = line;
newinfo->size = size; newinfo->size = size;
newinfo->file = file; newinfo->file = file;
newinfo->expr = expr;
newinfo->prev = NULL; newinfo->prev = NULL;
newinfo->next = stat_mem_block_root; newinfo->next = stat_mem_block_root;
@ -296,7 +299,7 @@ char *stat_mem_strdup(const char *src, size_t line, const char *file, bool empty
return NULL; return NULL;
len = strlen(src); len = strlen(src);
if (((!empty) ? len : true) && (ptr = (char*)stat_mem_allocate(len + 1, line, file))) { if (((!empty) ? len : true) && (ptr = (char*)stat_mem_allocate(len + 1, line, file, NULL))) {
memcpy(ptr, src, len); memcpy(ptr, src, len);
ptr[len] = '\0'; ptr[len] = '\0';
} }
@ -681,26 +684,34 @@ void util_htdel(hash_table_t *ht) {
* The following functions below implement printing / dumping of statistical * The following functions below implement printing / dumping of statistical
* information. * information.
*/ */
static void stat_dump_mem_contents(stat_mem_block_t *memory, uint16_t cols) { static void stat_dump_mem_contents(stat_mem_block_t *block, uint16_t cols) {
uint32_t i, j; unsigned char *buffer = mem_a(cols);
for (i = 0; i < memory->size + ((memory->size % cols) ? (cols - memory->size % cols) : 0); i++) { unsigned char *memory = (unsigned char *)(block + 1);
if (i % cols == 0) con_out(" 0x%06X: ", i); size_t i;
if (i < memory->size) con_out("%02X " , 0xFF & ((unsigned char*)(memory + 1))[i]);
else con_out(" ");
if ((uint16_t)(i % cols) == (cols - 1)) { for (i = 0; i < block->size; i++) {
for (j = i - (cols - 1); j <= i; j++) { if (!(i % 16)) {
con_out("%c", if (i != 0)
(j >= memory->size) con_out(" %s\n", buffer);
? ' ' con_out(" 0x%08X: ", i);
: (util_isprint(((unsigned char*)(memory + 1))[j]))
? 0xFF & ((unsigned char*)(memory + 1)) [j]
: '.'
);
}
con_out("\n");
} }
con_out(" %02X", memory[i]);
buffer[i % cols] = ((memory[i] < 0x20) || (memory[i] > 0x7E))
? '.'
: memory[i];
buffer[(i % cols) + 1] = '\0';
} }
while ((i % cols) != 0) {
con_out(" ");
i++;
}
con_out(" %s\n", buffer);
mem_d(buffer);
} }
static void stat_dump_mem_leaks(void) { static void stat_dump_mem_leaks(void) {
@ -710,10 +721,11 @@ static void stat_dump_mem_leaks(void) {
for (info = stat_mem_block_root; info; info = info->next) { for (info = stat_mem_block_root; info; info = info->next) {
/* we need access to the block */ /* we need access to the block */
VALGRIND_MAKE_MEM_DEFINED(info, sizeof(stat_mem_block_t)); VALGRIND_MAKE_MEM_DEFINED(info, sizeof(stat_mem_block_t));
con_out("lost: %u (bytes) at %s:%u\n", con_out("lost: %u (bytes) at %s:%u %s\n",
info->size, info->size,
info->file, info->file,
info->line info->line,
info->expr
); );
stat_dump_mem_contents(info, OPTS_OPTION_U16(OPTION_MEMDUMPCOLS)); stat_dump_mem_contents(info, OPTS_OPTION_U16(OPTION_MEMDUMPCOLS));