Annotation of early-roguelike/urogue/memory.c, Revision 1.1
1.1 ! rubenllo 1: /*
! 2: memory.c
! 3:
! 4: UltraRogue: The Ultimate Adventure in the Dungeons of Doom
! 5: Copyright (C) 1995 Herb Chong
! 6: All rights reserved.
! 7:
! 8: See the file LICENSE.TXT for full copyright and licensing information.
! 9: */
! 10:
! 11: #include <stdio.h>
! 12: #include <stdlib.h>
! 13:
! 14: #include "dict.h"
! 15: #include "memory.h"
! 16: #include "rogue.h"
! 17:
! 18: #ifdef HAVE_CONFIG_H
! 19: #include "config.h"
! 20: #endif
! 21:
! 22: static char sccsid[] = "%W%\t%G%";
! 23:
! 24: /* Debugging memory allocation code that tries to trap common memory problems
! 25: like overwriting storage and stepping on memory pointer chains. If code
! 26: doesn't use malloc, free, and realloc a lot, these routines can be left in
! 27: as added protection against undetected storage bugs.
! 28: */
! 29:
! 30: /* FENCE_SIZE should be a multiple of sizeof(size_t) to prevent alignment problems.
! 31: The code assumes that malloc and realloc return pointers aligned at least on size_t
! 32: sized boundaries and that a pointer needs alignment no more strict than that of an
! 33: object needed to hold a size_t.
! 34: */
! 35:
! 36: #define FENCE_SIZE (sizeof(size_t) * 1024)
! 37:
! 38: static int memdebug_level = 0;
! 39: static DICTIONARY *allocations = NULL;
! 40: static FILE *trace_file = NULL;
! 41:
! 42: /* set the debug level */
! 43: void mem_debug(const int level)
! 44: {
! 45: memdebug_level = level;
! 46:
! 47: #ifdef MEM_DEBUG
! 48: if (trace_file == NULL)
! 49: trace_file = fopen("trace", "w");
! 50:
! 51: /* all except 0, 1, and unknown fall through */
! 52: switch(memdebug_level) {
! 53: case 2:
! 54: fprintf(trace_file, "+++ Memory tracking possible, ");
! 55: case 1:
! 56: fprintf(trace_file, "+++ Memory debugging enabled, ");
! 57: break;
! 58: case 0:
! 59: fprintf(trace_file, "+++ Memory debugging disabled, ");
! 60: break;
! 61: default:
! 62: fprintf(trace_file, "!!! Unknown memory debug level set, enabling level 1, ");
! 63: memdebug_level = 1;
! 64: break;
! 65: }
! 66: fprintf(trace_file, "fence size = %d\n", FENCE_SIZE);
! 67: #endif
! 68: }
! 69:
! 70: /* set memory tracking on or off */
! 71: /* turning it off deletes all tracking data */
! 72: void mem_tracking(int flag)
! 73: {
! 74: #ifdef MEM_DEBUG
! 75: /* do nothing if debuglevel is too low */
! 76: if (memdebug_level < 2)
! 77: return;
! 78:
! 79: /* turn on tracking */
! 80: if (flag > 0) {
! 81: if (allocations != NULL) {
! 82: dict_destroy(allocations);
! 83: allocations = NULL;
! 84: }
! 85: allocations = dict_create(8, 100, 4, 20);
! 86: if (allocations == NULL) {
! 87: fprintf(trace_file, "!!! Unable to allocate tracking table!\n");
! 88: abort();
! 89: }
! 90: }
! 91: /* turn off tracking */
! 92: else if (allocations != NULL) {
! 93: dict_destroy(allocations);
! 94: allocations = NULL;
! 95: }
! 96: #endif
! 97: }
! 98:
! 99: /* go through all pointers and see if they are OK, aborting if not */
! 100: /* always returns 1 if not aborting so that it can be included in */
! 101: /* if statement boolean expressions */
! 102: int mem_check(char *fname, int linenum)
! 103: {
! 104: #ifdef MEM_DEBUG
! 105: STRING_ENTRY *se;
! 106:
! 107: /* scan of a NULL dictionary always succeeds */
! 108: if (allocations == NULL)
! 109: return TRUE;
! 110:
! 111: if (!dict_scan_begin(allocations)) {
! 112: fprintf(trace_file, "!!! Dictionary scan initialization failed!\n");
! 113: abort();
! 114: }
! 115:
! 116: fprintf(trace_file, "\n+++ --- Starting pointer scan\n");
! 117: fprintf(trace_file, "+++ --- At %s, %d\n", fname, linenum);
! 118:
! 119: /* mem_validate aborts if there is a problem */
! 120: while((se = dict_scan_next(allocations)) != NULL)
! 121: mem_validate(se->any_ptr);
! 122:
! 123: fprintf(trace_file, "+++ --- Done pointer scan\n\n");
! 124:
! 125: #endif
! 126: /* always return a good value if execution arrives here */
! 127: return 1;
! 128: }
! 129:
! 130: /* allocate some memory and initialize header and trailer */
! 131: void *mem_malloc(const size_t bytes)
! 132: {
! 133: #ifdef MEM_DEBUG
! 134: char *mem_temp;
! 135: size_t real_size = bytes + (FENCE_SIZE << 1);
! 136:
! 137: /* allocate including guard bytes to detect some ways of overwriting of memory areas */
! 138: mem_temp = (void *)malloc(real_size);
! 139: if (memdebug_level > 0) {
! 140: fprintf(trace_file, "+++ Requested size of %ld bytes\n", bytes);
! 141: fprintf(trace_file, "+++ Actual malloc of %ld bytes located at %p\n", real_size, mem_temp);
! 142: }
! 143:
! 144: /* if allocation succeeded, set management data */
! 145: if (mem_temp != NULL) {
! 146: size_t i;
! 147: char *end;
! 148:
! 149: /* do beginning marker bytes */
! 150: for (i = 0; i < FENCE_SIZE - sizeof(size_t); i++)
! 151: *mem_temp++ = 145;
! 152:
! 153: /* save size in header too */
! 154: if (memdebug_level > 0)
! 155: fprintf(trace_file, "*** Requested memory size stored at %p\n", mem_temp);
! 156: *(size_t *)mem_temp = bytes;
! 157:
! 158: /* finally, point to storage we are going to hand out */
! 159: mem_temp += sizeof(size_t);
! 160:
! 161: /* now, point to trailer bytes and do them */
! 162: end = mem_temp + bytes;
! 163: for (i = 0; i < FENCE_SIZE; i++)
! 164: *end++ = 145;
! 165:
! 166: /* now zap contents to zero */
! 167: for (i = 0; i < bytes; i++)
! 168: mem_temp[i] = 0;
! 169: }
! 170:
! 171: /* track pointer if needed */
! 172: if (memdebug_level > 1 && allocations != NULL) {
! 173: char key[16];
! 174: long temp;
! 175:
! 176: sprintf(key, "%p", mem_temp);
! 177: if (dict_insert(allocations, key, 1, (const unsigned long) bytes, mem_temp, &temp) == NULL) {
! 178: fprintf(trace_file, "!!! Insert of pointer tracking info failed\n");
! 179: abort();
! 180: }
! 181: }
! 182:
! 183: /* allow caller to do error handling */
! 184: if (memdebug_level > 0) {
! 185: fprintf(trace_file, "--- Returning pointer of %p\n", mem_temp);
! 186: fflush(trace_file);
! 187: }
! 188: return (void *)mem_temp;
! 189: #else
! 190: return malloc(bytes);
! 191: #endif
! 192: }
! 193:
! 194: /* release some memory, making sure that it was properly allocated */
! 195: void mem_free(const void *ptr)
! 196: {
! 197: #ifdef MEM_DEBUG
! 198: char *mem_temp;
! 199: size_t mem_size;
! 200: size_t i;
! 201:
! 202: if (memdebug_level > 0)
! 203: fprintf(trace_file, "+++ Free of memory located at %p\n", ptr);
! 204: if (ptr == NULL) {
! 205: if (memdebug_level > 0) {
! 206: fprintf(trace_file, "!!! Freeing NULL pointer\n");
! 207: fflush(trace_file);
! 208: }
! 209: abort();
! 210: }
! 211:
! 212: mem_validate(ptr); /* doesn't return on error */
! 213:
! 214: /* get location of size of area */
! 215: mem_temp = (char *)ptr - sizeof(size_t);
! 216:
! 217: /* get and calculate real size */
! 218: mem_size = *(size_t *)mem_temp + (FENCE_SIZE << 1);
! 219:
! 220: /* if doing memory tracking */
! 221: if (memdebug_level > 1 && allocations != NULL) {
! 222: char key[16];
! 223: STRING_ENTRY *se;
! 224: long temp;
! 225:
! 226: sprintf(key, "%p", ptr);
! 227:
! 228: if ((se = dict_search(allocations, key, &temp)) == NULL) {
! 229: fprintf(trace_file, "!!! Deleting pointer not found in tracking info\n");
! 230: abort();
! 231: }
! 232:
! 233: if (se->count == 0) {
! 234: fprintf(trace_file, "!!! Freeing a pointer that has already been freed!\n");
! 235: abort();
! 236: }
! 237: else if (se->flags != mem_size - (FENCE_SIZE << 1)) {
! 238: fprintf(trace_file, "!!! Stored size different from tracking size!\n");
! 239: abort();
! 240: }
! 241:
! 242: /* remember deleted stuff by zeroing the allocation count */
! 243: se->count = 0;
! 244: se->flags = 0;
! 245: }
! 246:
! 247: /* zap bytes being freed */
! 248: for (i = 0, mem_temp = (char *)ptr - FENCE_SIZE; i < mem_size; i++, mem_temp++)
! 249: *mem_temp = 243;
! 250:
! 251: if (memdebug_level > 0)
! 252: fflush(trace_file);
! 253:
! 254: mem_temp = (char *)ptr - FENCE_SIZE;
! 255: free((void *)mem_temp);
! 256: #else
! 257: free((void *) ptr);
! 258: #endif
! 259: }
! 260:
! 261: /* reallocate some memory, making sure that it was properly allocated */
! 262: void *mem_realloc(const void *ptr, const size_t new_size)
! 263: {
! 264: #ifdef MEM_DEBUG
! 265: char *mem_temp = (char *)ptr;
! 266: size_t real_size = new_size + (FENCE_SIZE << 1);
! 267: size_t mem_size;
! 268: long i;
! 269:
! 270: if (memdebug_level > 0) {
! 271: fprintf(trace_file, "+++ Requested size of %ld bytes\n", new_size);
! 272: fprintf(trace_file, "+++ Actual realloc of %ld bytes located at %p\n", real_size, mem_temp);
! 273: }
! 274: if (ptr == NULL) {
! 275: if (memdebug_level > 0) {
! 276: fprintf(trace_file, "!!! Reallocating NULL pointer\n");
! 277: fflush(trace_file);
! 278: }
! 279: abort();
! 280: }
! 281:
! 282: mem_validate(ptr); /* doesn't return on error */
! 283:
! 284: /* if doing memory tracking */
! 285: if (memdebug_level > 1 && allocations != NULL) {
! 286: char key[16];
! 287: STRING_ENTRY *se;
! 288: long temp;
! 289:
! 290: sprintf(key, "%p", ptr);
! 291:
! 292: if ((se = dict_search(allocations, key, &temp)) == NULL) {
! 293: fprintf(trace_file, "!!! Deleting a pointer not found in tracking info!\n");
! 294: abort();
! 295: }
! 296:
! 297: /* point to size bytes */
! 298: mem_temp = (char *)ptr - sizeof(size_t);
! 299:
! 300: /* get user size */
! 301: mem_size = *(size_t *)mem_temp;
! 302:
! 303: if (se->count == 0) {
! 304: fprintf(trace_file, "!!! Freeing a pointer that has already been freed!\n");
! 305: abort();
! 306: }
! 307: else if (se->flags != mem_size) {
! 308: fprintf(trace_file, "!!! Stored size different from tracking size!\n");
! 309: abort();
! 310: }
! 311:
! 312: /* remember deleted stuff by zeroing the allocation count */
! 313: se->count = 0;
! 314: se->flags = 0;
! 315: }
! 316:
! 317:
! 318: /* header marker bytes will be copied by the realloc */
! 319: mem_temp = (char *)ptr - FENCE_SIZE;
! 320: mem_temp = realloc((void *)mem_temp, real_size);
! 321:
! 322: if (mem_temp != NULL) {
! 323: char *end;
! 324:
! 325: /* save size in header too */
! 326: mem_temp += FENCE_SIZE - sizeof(size_t);
! 327: if (memdebug_level > 0)
! 328: fprintf(trace_file, "*** Requested memory size stored at %p\n", mem_temp);
! 329: *(size_t *)mem_temp = new_size;
! 330:
! 331: /* finally, point to storage we are going to hand out */
! 332: mem_temp += sizeof(size_t);
! 333:
! 334: /* now, point to trailer bytes and do them */
! 335: end = mem_temp + new_size;
! 336: for (i = 0; i < FENCE_SIZE; i++)
! 337: *end++ = 145;
! 338: }
! 339:
! 340: if (memdebug_level > 1 && allocations != NULL) {
! 341: char key[16];
! 342: long temp;
! 343:
! 344: sprintf(key, "%p", mem_temp);
! 345: if (dict_insert(allocations, key, 1, (const unsigned long)new_size, mem_temp, &temp) == NULL) {
! 346: fprintf(trace_file, "!!! Insert of pointer tracking info failed\n");
! 347: abort();
! 348: }
! 349: }
! 350:
! 351: if (memdebug_level > 0) {
! 352: fprintf(trace_file, "--- Returning pointer of %p\n", mem_temp);
! 353: fflush(trace_file);
! 354: }
! 355: return (void *)mem_temp;
! 356: #else
! 357: return realloc((void *) ptr, new_size);
! 358: #endif
! 359: }
! 360:
! 361: /* check a pointer to be sure all check bytes are OK. abort if not */
! 362: /* always returns 1 if not aborting so that it can be included in */
! 363: /* if statement boolean expressions */
! 364: int mem_validate(const void *ptr)
! 365: {
! 366: #ifdef MEM_DEBUG
! 367: unsigned char *mem_temp = (unsigned char *)ptr;
! 368: size_t mem_size;
! 369: size_t i;
! 370:
! 371: /* NULL pointers are always valid */
! 372: if (ptr == NULL)
! 373: return 1;
! 374:
! 375: if (memdebug_level > 0)
! 376: fprintf(trace_file, "+++ Checking %p as pointer\n", ptr);
! 377:
! 378:
! 379: if (memdebug_level > 1 && allocations != NULL) {
! 380: char key[16];
! 381: STRING_ENTRY *se;
! 382: long temp;
! 383:
! 384: sprintf(key, "%p", ptr);
! 385:
! 386: if ((se = dict_search(allocations, key, &temp)) == NULL) {
! 387: fprintf(trace_file, "!!! Pointer not found in tracking info!\n");
! 388: abort();
! 389: }
! 390:
! 391: /* point to size bytes */
! 392: mem_temp = (unsigned char *)ptr - sizeof(size_t);
! 393:
! 394: /* get user size */
! 395: mem_size = *(size_t *)mem_temp;
! 396:
! 397: if (se->count == 0) {
! 398: fprintf(trace_file, "!!! Checking pointer has been freed!\n");
! 399: abort();
! 400: }
! 401: else if (se->flags != mem_size) {
! 402: fprintf(trace_file, "!!! Stored size different from tracking size!\n");
! 403: abort();
! 404: }
! 405: }
! 406:
! 407: /* check the header bytes */
! 408: mem_temp = (unsigned char *) ptr - FENCE_SIZE;
! 409: if (memdebug_level > 0)
! 410: fprintf(trace_file, "+++ Real pointer at %p\n", mem_temp);
! 411:
! 412:
! 413: for (i = 0; i < FENCE_SIZE - sizeof(size_t); i++)
! 414: if (*mem_temp++ != 145) {
! 415: if (memdebug_level > 0) {
! 416: fprintf(trace_file, "!!! The user pointer at %p has been overwritten\n", ptr);
! 417: fprintf(trace_file, "!!! Header offset %ld has been changed\n", i - 1);
! 418: fflush(trace_file);
! 419: }
! 420: abort();
! 421: }
! 422:
! 423: /* check size */
! 424: i = *(size_t *)mem_temp;
! 425: if (memdebug_level > 0)
! 426: fprintf(trace_file, "*** Stored memory size of %ld bytes in header\n", i);
! 427:
! 428:
! 429: /* now point to where trailer should be */
! 430: mem_temp = (unsigned char *)ptr + i;
! 431: for (i = 0; i < FENCE_SIZE; i++)
! 432: if (*mem_temp++ != 145) {
! 433: if (memdebug_level > 0) {
! 434: fprintf(trace_file, "!!! The user pointer at %p has been overwritten\n", ptr);
! 435: fprintf(trace_file, "!!! Trailer offset %ld has been changed\n", i - 1);
! 436: fflush(trace_file);
! 437: }
! 438: abort();
! 439: }
! 440: if (memdebug_level > 0)
! 441: fflush(trace_file);
! 442: #endif
! 443: return 1;
! 444: }
CVSweb