io.c

   1 #include "io.h"
   2 #include <stdlib.h>
   3 #ifndef _MSC_VER
   4 #include <unistd.h>
   5 #endif
   6
   7
   8 /* FIXME: make allocation dynamic */
   9
  10
  11 #ifdef IMAGER_DEBUG_MALLOC
  12
  13 #define MAXMAL 102400
  14 #define MAXDESC 65
  15
  16 #define UNDRRNVAL 10
  17 #define OVERRNVAL 10
  18
  19 #define PADBYTE 0xaa
  20
  21
  22 static int malloc_need_init = 1;
  23
  24 typedef struct {
  25   void* ptr;
  26   size_t size;
  27   char comm[MAXDESC];
  28 } malloc_entry;
  29
  30 malloc_entry malloc_pointers[MAXMAL];
  31
  32
  33
  34
  35 /* Utility functions */
  36
  37
  38 static
  39 void
  40 malloc_init(void) {
  41   int i;
  42   for(i=0; i<MAXMAL; i++) malloc_pointers[i].ptr = NULL;
  43   malloc_need_init = 0;
  44   atexit(malloc_state);
  45 }
  46
  47
  48 static
  49 int
  50 find_ptr(void *p) {
  51   int i;
  52   for(i=0;i<MAXMAL;i++)
  53     if (malloc_pointers[i].ptr == p)
  54       return i;
  55   return -1;
  56 }
  57
  58
  59 /* Takes a pointer to real start of array,
  60  * sets the entries in the table, returns
  61  * the offset corrected pointer */
  62
  63 static
  64 void *
  65 set_entry(int i, char *buf, size_t size, char *file, int line) {
  66   memset( buf, PADBYTE, UNDRRNVAL );
  67   memset( &buf[UNDRRNVAL+size], PADBYTE, OVERRNVAL );
  68   buf += UNDRRNVAL;
  69   malloc_pointers[i].ptr  = buf;
  70   malloc_pointers[i].size = size;
  71   sprintf(malloc_pointers[i].comm,"%s (%d)", file, line);
  72   return buf;
  73 }
  74
  75
  76
  77
  78 void
  79 malloc_state(void) {
  80   int i, total = 0;
  81
  82   i_clear_error();
  83   mm_log((0,"malloc_state()\n"));
  84   bndcheck_all();
  85   for(i=0; i<MAXMAL; i++) if (malloc_pointers[i].ptr != NULL) {
  86     mm_log((0,"%d: %d (0x%x) : %s\n", i, malloc_pointers[i].size, malloc_pointers[i].ptr, malloc_pointers[i].comm));
  87     total += malloc_pointers[i].size;
  88   }
  89   if (total == 0) mm_log((0,"No memory currently used!\n"))
  90                     else mm_log((0,"total: %d\n",total));
  91 }
  92
  93
  94
  95 void*
  96 mymalloc_file_line(size_t size, char* file, int line) {
  97   char *buf;
  98   int i;
  99   if (malloc_need_init) malloc_init();
 100
 101   /* bndcheck_all(); Uncomment for LOTS OF THRASHING */
 102
 103   if ( (i = find_ptr(NULL)) < 0 ) {
 104     mm_log((0,"more than %d segments allocated at %s (%d)\n", MAXMAL, file, line));
 105     exit(3);
 106   }
 107
 108   if ( (buf = malloc(size+UNDRRNVAL+OVERRNVAL)) == NULL ) {
 109     mm_log((1,"Unable to allocate %i for %s (%i)\n", size, file, line));
 110     exit(3);
 111   }
 112
 113   buf = set_entry(i, buf, size, file, line);
 114   mm_log((1,"mymalloc_file_line: slot <%d> %d bytes allocated at %p for %s (%d)\n", i, size, buf, file, line));
 115   return buf;
 116 }
 117
 118
 119
 120
 121
 122
 123
 124 void*
 125 myrealloc_file_line(void *ptr, size_t newsize, char* file, int line) {
 126   char *buf;
 127   int i;
 128
 129   if (malloc_need_init) malloc_init();
 130   /* bndcheck_all(); ACTIVATE FOR LOTS OF THRASHING */
 131
 132   if (!ptr) {
 133     mm_log((1, "realloc called with ptr = NULL, sending request to malloc\n"));
 134     return mymalloc_file_line(newsize, file, line);
 135   }
 136
 137   if (!newsize) {
 138     mm_log((1, "newsize = 0, sending request to free\n"));
 139     myfree_file_line(ptr, file, line);
 140     return NULL;
 141   }
 142
 143   if ( (i = find_ptr(ptr)) == -1) {
 144     mm_log((0, "Unable to find %p in realloc for %s (%i)\n", ptr, file, line));
 145     exit(3);
 146   }
 147
 148   if ( (buf = realloc(((char *)ptr)-UNDRRNVAL, UNDRRNVAL+OVERRNVAL+newsize)) == NULL ) {
 149     mm_log((1,"Unable to reallocate %i bytes at %p for %s (%i)\n", newsize, ptr, file, line));
 150     exit(3);
 151   }
 152
 153   buf = set_entry(i, buf, newsize, file, line);
 154   mm_log((1,"realloc_file_line: slot <%d> %d bytes allocated at %p for %s (%d)\n", i, newsize, buf, file, line));
 155   return buf;
 156 }
 157
 158
 159
 160
 161 static
 162 void
 163 bndcheck(int idx) {
 164   int i;
 165   size_t s = malloc_pointers[idx].size;
 166   unsigned char *pp = malloc_pointers[idx].ptr;
 167   if (!pp) {
 168     mm_log((1, "bndcheck: No pointer in slot %d\n", idx));
 169     return;
 170   }
 171
 172   for(i=0;i<UNDRRNVAL;i++)
 173      if (pp[-(1+i)] != PADBYTE)
 174      mm_log((1,"bndcheck: UNDERRUN OF %d bytes detected: slot = %d, point = %p, size = %d\n", i+1, idx, pp, s ));
 175
 176      for(i=0;i<OVERRNVAL;i++)
 177     if (pp[s+i] != PADBYTE)
 178       mm_log((1,"bndcheck: OVERRUN OF %d bytes detected: slot = %d, point = %p, size = %d\n", i+1, idx, pp, s ));
 179 }
 180
 181 void
 182 bndcheck_all() {
 183   int idx;
 184   mm_log((1, "bndcheck_all()\n"));
 185   for(idx=0; idx<MAXMAL; idx++)
 186     if (malloc_pointers[idx].ptr)
 187       bndcheck(idx);
 188 }
 189
 190
 191
 192
 193
 194 void
 195 myfree_file_line(void *p, char *file, int line) {
 196   char  *pp = p;
 197   int match = 0;
 198   int i;
 199
 200   for(i=0; i<MAXMAL; i++) if (malloc_pointers[i].ptr == p) {
 201     mm_log((1,"myfree_file_line: pointer %i (%s) freed at %s (%i)\n", i, malloc_pointers[i].comm, file, line));
 202     bndcheck(i);
 203     malloc_pointers[i].ptr = NULL;
 204     match++;
 205   }
 206
 207   mm_log((1, "myfree_file_line: freeing address %p (real %p)\n", pp, pp-UNDRRNVAL));
 208
 209   if (match != 1) {
 210     mm_log((1, "myfree_file_line: INCONSISTENT REFCOUNT %d at %s (%i)\n", match, file, line));
 211     fprintf(stderr, "myfree_file_line: INCONSISTENT REFCOUNT %d at %s (%i)\n", match, file, line);
 212                 exit(255);
 213   }
 214
 215
 216   free(pp-UNDRRNVAL);
 217 }
 218
 219 #else
 220
 221 #define malloc_comm(a,b) (mymalloc(a))
 222
 223 void
 224 malloc_state() {
 225   printf("malloc_state: not in debug mode\n");
 226 }
 227
 228 void*
 229 mymalloc(int size) {
 230   void *buf;
 231
 232   if ( (buf = malloc(size)) == NULL ) {
 233     mm_log((1, "mymalloc: unable to malloc %d\n", size));
 234     fprintf(stderr,"Unable to malloc %d.\n", size); exit(3);
 235   }
 236   mm_log((1, "mymalloc(size %d) -> %p\n", size, buf));
 237   return buf;
 238 }
 239
 240 void
 241 myfree(void *p) {
 242   mm_log((1, "myfree(p %p)\n", p));
 243   free(p);
 244 }
 245
 246 void *
 247 myrealloc(void *block, size_t size) {
 248   void *result;
 249
 250   mm_log((1, "myrealloc(block %p, size %u)\n", block, size));
 251   if ((result = realloc(block, size)) == NULL) {
 252     mm_log((1, "myrealloc: out of memory\n"));
 253     fprintf(stderr, "Out of memory.\n");
 254     exit(3);
 255   }
 256   return result;
 257 }
 258
 259 #endif /* IMAGER_MALLOC_DEBUG */
 260
 261
 262
 263
 264 /* memory pool implementation */
 265
 266 void
 267 i_mempool_init(i_mempool *mp) {
 268   mp->alloc = 10;
 269   mp->used  = 0;
 270   mp->p = mymalloc(sizeof(void*)*mp->alloc);
 271 }
 272
 273 void
 274 i_mempool_extend(i_mempool *mp) {
 275   mp->p = myrealloc(mp->p, mp->alloc * 2);
 276   mp->alloc *=2;
 277 }
 278
 279 void *
 280 i_mempool_alloc(i_mempool *mp, size_t size) {
 281   if (mp->used == mp->alloc) i_mempool_extend(mp);
 282   mp->p[mp->used] = mymalloc(size);
 283   mp->used++;
 284   return mp->p[mp->used-1];
 285 }
 286
 287
 288 void
 289 i_mempool_destroy(i_mempool *mp) {
 290   unsigned int i;
 291   for(i=0; i<mp->used; i++) myfree(mp->p[i]);
 292   myfree(mp->p);
 293 }
 294
 295
 296
 297 /* Should these really be here? */
 298
 299 #undef min
 300 #undef max
 301
 302 int
 303 min(int a,int b) {
 304   if (a<b) return a; else return b;
 305 }
 306
 307 int
 308 max(int a,int b) {
 309   if (a>b) return a; else return b;
 310 }
 311