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() {
  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() {
  80   int i, total = 0;
  81
  82   mm_log((0,"malloc_state()\n"));
  83   bndcheck_all();
  84   for(i=0; i<MAXMAL; i++) if (malloc_pointers[i].ptr != NULL) {
  85     mm_log((0,"%d: %d (0x%x) : %s\n", i, malloc_pointers[i].size, malloc_pointers[i].ptr, malloc_pointers[i].comm));
  86     total += malloc_pointers[i].size;
  87   }
  88   if (total == 0) mm_log((0,"No memory currently used!\n"))
  89                     else mm_log((0,"total: %d\n",total));
  90 }
  91
  92
  93
  94 void*
  95 mymalloc_file_line(size_t size, char* file, int line) {
  96   char *buf;
  97   int i;
  98   if (malloc_need_init) malloc_init();
  99
 100   /* bndcheck_all(); Uncomment for LOTS OF THRASHING */
 101
 102   if ( (i = find_ptr(NULL)) < 0 ) {
 103     mm_log((0,"more than %d segments allocated at %s (%d)\n", MAXMAL, file, line));
 104     exit(3);
 105   }
 106
 107   if ( (buf = malloc(size+UNDRRNVAL+OVERRNVAL)) == NULL ) {
 108     mm_log((1,"Unable to allocate %i for %s (%i)\n", size, file, line));
 109     exit(3);
 110   }
 111
 112   buf = set_entry(i, buf, size, file, line);
 113   mm_log((1,"mymalloc_file_line: slot <%d> %d bytes allocated at %p for %s (%d)\n", i, size, buf, file, line));
 114   return buf;
 115 }
 116
 117
 118
 119
 120
 121
 122
 123 void*
 124 myrealloc_file_line(void *ptr, size_t newsize, char* file, int line) {
 125   char *buf;
 126   int i;
 127
 128   if (malloc_need_init) malloc_init();
 129   /* bndcheck_all(); ACTIVATE FOR LOTS OF THRASHING */
 130
 131   if (!ptr) {
 132     mm_log((1, "realloc called with ptr = NULL, sending request to malloc\n"));
 133     return mymalloc_file_line(newsize, file, line);
 134   }
 135
 136   if (!newsize) {
 137     mm_log((1, "newsize = 0, sending request to free\n"));
 138     myfree_file_line(ptr, file, line);
 139     return NULL;
 140   }
 141
 142   if ( (i = find_ptr(ptr)) == -1) {
 143     mm_log((0, "Unable to find %p in realloc for %s (%i)\n", ptr, file, line));
 144     exit(3);
 145   }
 146
 147   if ( (buf = realloc(ptr-UNDRRNVAL, UNDRRNVAL+OVERRNVAL+newsize)) == NULL ) {
 148     mm_log((1,"Unable to reallocate %i bytes at %p for %s (%i)\n", newsize, ptr, file, line));
 149     exit(3);
 150   }
 151
 152   buf = set_entry(i, buf, newsize, file, line);
 153   mm_log((1,"realloc_file_line: slot <%d> %d bytes allocated at %p for %s (%d)\n", i, newsize, buf, file, line));
 154   return buf;
 155 }
 156
 157
 158
 159
 160 static
 161 void
 162 bndcheck(int idx) {
 163   int i;
 164   size_t s = malloc_pointers[idx].size;
 165   unsigned char *pp = malloc_pointers[idx].ptr;
 166   if (!pp) {
 167     mm_log((1, "bndcheck: No pointer in slot %d\n", idx));
 168     return;
 169   }
 170
 171   for(i=0;i<UNDRRNVAL;i++)
 172      if (pp[-(1+i)] != PADBYTE)
 173      mm_log((1,"bndcheck: UNDERRUN OF %d bytes detected: slot = %d, point = %p, size = %d\n", i+1, idx, pp, s ));
 174
 175      for(i=0;i<OVERRNVAL;i++)
 176     if (pp[s+i] != PADBYTE)
 177       mm_log((1,"bndcheck: OVERRUN OF %d bytes detected: slot = %d, point = %p, size = %d\n", i+1, idx, pp, s ));
 178 }
 179
 180 void
 181 bndcheck_all() {
 182   int idx;
 183   mm_log((1, "bndcheck_all()\n"));
 184   for(idx=0; idx<MAXMAL; idx++)
 185     if (malloc_pointers[idx].ptr)
 186       bndcheck(idx);
 187 }
 188
 189
 190
 191
 192
 193 void
 194 myfree_file_line(void *p, char *file, int line) {
 195   char  *pp = p;
 196   int match = 0;
 197   int i;
 198
 199   for(i=0; i<MAXMAL; i++) if (malloc_pointers[i].ptr == p) {
 200     mm_log((1,"myfree_file_line: pointer %i (%s) freed at %s (%i)\n", i, malloc_pointers[i].comm, file, line));
 201     bndcheck(i);
 202     malloc_pointers[i].ptr = NULL;
 203     match++;
 204   }
 205
 206   if (match != 1) {
 207     mm_log((1, "myfree_file_line: INCONSISTENT REFCOUNT %d at %s (%i)\n", match, file, line));
 208   }
 209
 210   mm_log((1, "myfree_file_line: freeing address %p\n", pp-UNDRRNVAL));
 211
 212   free(pp-UNDRRNVAL);
 213 }
 214
 215 #else
 216
 217 #define malloc_comm(a,b) (mymalloc(a))
 218
 219 void
 220 malloc_state() {
 221   printf("malloc_state: not in debug mode\n");
 222 }
 223
 224 void*
 225 mymalloc(int size) {
 226   void *buf;
 227
 228   if ( (buf = malloc(size)) == NULL ) {
 229     mm_log((1, "mymalloc: unable to malloc %d\n", size));
 230     fprintf(stderr,"Unable to malloc %d.\n", size); exit(3);
 231   }
 232   mm_log((1, "mymalloc(size %d) -> %p\n", size, buf));
 233   return buf;
 234 }
 235
 236 void
 237 myfree(void *p) {
 238   mm_log((1, "myfree(p %p)\n", p));
 239   free(p);
 240 }
 241
 242 void *
 243 myrealloc(void *block, size_t size) {
 244   void *result;
 245
 246   mm_log((1, "myrealloc(block %p, size %u)\n", block, size));
 247   if ((result = realloc(block, size)) == NULL) {
 248     mm_log((1, "myrealloc: out of memory\n"));
 249     fprintf(stderr, "Out of memory.\n");
 250     exit(3);
 251   }
 252   return result;
 253 }
 254
 255 #endif /* IMAGER_MALLOC_DEBUG */
 256
 257
 258
 259
 260
 261
 262 /* Should these really be here? */
 263
 264 #undef min
 265 #undef max
 266
 267 int
 268 min(int a,int b) {
 269   if (a<b) return a; else return b;
 270 }
 271
 272 int
 273 max(int a,int b) {
 274   if (a>b) return a; else return b;
 275 }
 276