git.distorted.org.uk Git - u/mdw/putty/blob - misc.c

   1 #include <windows.h>
   2 #ifndef AUTO_WINSOCK
   3 #ifdef WINSOCK_TWO
   4 #include <winsock2.h>
   5 #else
   6 #include <winsock.h>
   7 #endif
   8 #endif
   9 #include <stdio.h>
  10 #include <stdlib.h>
  11 #include "putty.h"
  12
  13 /* My own versions of malloc, realloc and free. Because I want malloc and
  14  * realloc to bomb out and exit the program if they run out of memory,
  15  * realloc to reliably call malloc if passed a NULL pointer, and free
  16  * to reliably do nothing if passed a NULL pointer. Of course we can also
  17  * put trace printouts in, if we need to. */
  18
  19 #ifdef MALLOC_LOG
  20 static FILE *fp = NULL;
  21
  22 void mlog(char *file, int line) {
  23     if (!fp) {
  24         fp = fopen("putty_mem.log", "w");
  25         setvbuf(fp, NULL, _IONBF, BUFSIZ);
  26     }
  27     if (fp)
  28         fprintf (fp, "%s:%d: ", file, line);
  29 }
  30 #endif
  31
  32 void *safemalloc(size_t size) {
  33     void *p = malloc (size);
  34     if (!p) {
  35         MessageBox(NULL, "Out of memory!", "PuTTY Fatal Error",
  36                    MB_SYSTEMMODAL | MB_ICONERROR | MB_OK);
  37         exit(1);
  38     }
  39 #ifdef MALLOC_LOG
  40     if (fp)
  41         fprintf(fp, "malloc(%d) returns %p\n", size, p);
  42 #endif
  43     return p;
  44 }
  45
  46 void *saferealloc(void *ptr, size_t size) {
  47     void *p;
  48     if (!ptr)
  49         p = malloc (size);
  50     else
  51         p = realloc (ptr, size);
  52     if (!p) {
  53         MessageBox(NULL, "Out of memory!", "PuTTY Fatal Error",
  54                    MB_SYSTEMMODAL | MB_ICONERROR | MB_OK);
  55         exit(1);
  56     }
  57 #ifdef MALLOC_LOG
  58     if (fp)
  59         fprintf(fp, "realloc(%p,%d) returns %p\n", ptr, size, p);
  60 #endif
  61     return p;
  62 }
  63
  64 void safefree(void *ptr) {
  65     if (ptr) {
  66 #ifdef MALLOC_LOG
  67         if (fp)
  68             fprintf(fp, "free(%p)\n", ptr);
  69 #endif
  70         free (ptr);
  71     }
  72 #ifdef MALLOC_LOG
  73     else if (fp)
  74         fprintf(fp, "freeing null pointer - no action taken\n");
  75 #endif
  76 }