src/alloc-srvx.c

   1 /* alloc-srvx.c - Debug allocation wrapper
   2  * Copyright 2005 srvx Development Team
   3  *
   4  * This file is part of srvx.
   5  *
   6  * srvx is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  */
  16
  17 #include "common.h"
  18
  19 #undef malloc
  20 #undef free
  21
  22 /* cookies for anybody who recognizes these bytes without help :) */
  23 #define ALLOC_MAGIC 0x1acf
  24 #define FREE_MAGIC  0xfc1d
  25 const char redzone[] = { '\x03', '\x47', '\x76', '\xc7' };
  26
  27 struct alloc_header {
  28     unsigned int file_id : 8;
  29     unsigned int size : 24;
  30     unsigned int line : 16;
  31     unsigned int magic : 16;
  32 };
  33
  34 static char file_id_map[256][32];
  35 static unsigned int file_ids_used;
  36 unsigned long alloc_count, alloc_size;
  37
  38 static int
  39 file_id_cmp(const void *a_, const void *b_)
  40 {
  41     return strcmp(a_, b_);
  42 }
  43
  44 static unsigned int
  45 get_file_id(const char *fname)
  46 {
  47     void *entry;
  48
  49     entry = bsearch(fname, file_id_map, file_ids_used, sizeof(file_id_map[0]), file_id_cmp);
  50     if (entry)
  51         return ((char*)entry - file_id_map[0]) / sizeof(file_id_map[0]);
  52     strcpy(file_id_map[file_ids_used++], fname);
  53     qsort(file_id_map, file_ids_used, sizeof(file_id_map[0]), file_id_cmp);
  54     return file_ids_used - 1;
  55 }
  56
  57 void *
  58 srvx_malloc(const char *file, unsigned int line, size_t size)
  59 {
  60     struct alloc_header *block;
  61
  62     block = malloc(sizeof(*block) + size + sizeof(redzone));
  63     assert(block != NULL);
  64     memset(block, 0, sizeof(*block) + size);
  65     memcpy((char*)(block + 1) + size, redzone, sizeof(redzone));
  66     block->file_id = get_file_id(file);
  67     block->line = line;
  68     block->size = size;
  69     block->magic = ALLOC_MAGIC;
  70     alloc_count++;
  71     alloc_size += size;
  72     return block + 1;
  73 }
  74
  75 void *
  76 srvx_realloc(const char *file, unsigned int line, void *ptr, size_t size)
  77 {
  78     struct alloc_header *block = NULL, *newblock;
  79
  80     if (ptr) {
  81         block = (struct alloc_header *)ptr - 1;
  82         assert(block->magic == ALLOC_MAGIC);
  83         assert(0 == memcmp((char*)(block + 1) + block->size, redzone, sizeof(redzone)));
  84         if (block->size >= size)
  85             return block + 1;
  86     }
  87
  88     newblock = malloc(sizeof(*newblock) + size + sizeof(redzone));
  89     assert(newblock != NULL);
  90     memset(newblock, 0, sizeof(*newblock) + size + sizeof(redzone));
  91     memcpy((char*)(newblock + 1) + size, redzone, sizeof(redzone));
  92     newblock->file_id = get_file_id(file);
  93     newblock->line = line;
  94     newblock->size = size;
  95     newblock->magic = ALLOC_MAGIC;
  96     alloc_count++;
  97     alloc_size += size;
  98
  99     if (ptr) {
 100         memcpy(newblock + 1, block + 1, block->size);
 101         size = block->size + sizeof(*block);
 102         memset(block, 0, size);
 103         block->magic = FREE_MAGIC;
 104         free(block);
 105         alloc_count--;
 106         alloc_size -= size - sizeof(*block);
 107     }
 108
 109     return newblock + 1;
 110 }
 111
 112 char *
 113 srvx_strdup(const char *file, unsigned int line, const char *src)
 114 {
 115     char *target;
 116     size_t len;
 117
 118     len = strlen(src) + 1;
 119     target = srvx_malloc(file, line, len);
 120     memcpy(target, src, len);
 121     return target;
 122 }
 123
 124 void
 125 srvx_free(const char *file, unsigned int line, void *ptr)
 126 {
 127     struct alloc_header *block;
 128     size_t size;
 129
 130     if (!ptr)
 131         return;
 132     block = (struct alloc_header *)ptr - 1;
 133     assert(block->magic == ALLOC_MAGIC);
 134     assert(0 == memcmp((char*)(block + 1) + block->size, redzone, sizeof(redzone)));
 135     size = block->size + sizeof(*block);
 136     memset(block, 0, size);
 137     block->magic = FREE_MAGIC;
 138     free(block);
 139     alloc_count--;
 140     alloc_size -= size - sizeof(*block);
 141     (void)file; (void)line;
 142 }