WINGs/memory.c

   1 /*
   2  *  Window Maker miscelaneous function library
   3  *
   4  *  Copyright (c) 1997 Alfredo K. Kojima
   5  *
   6  *  This program 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  *  You should have received a copy of the GNU General Public License
  17  *  along with this program; if not, write to the Free Software
  18  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  19  */
  20
  21
  22 #include "../src/config.h"
  23
  24 #include "WUtil.h"
  25
  26 #include <stdlib.h>
  27 #include <sys/types.h>
  28 #include <unistd.h>
  29 #include <stdio.h>
  30 #include <string.h>
  31 #include <assert.h>
  32
  33 #ifndef False
  34 # define False 0
  35 #endif
  36 #ifndef True
  37 # define True 1
  38 #endif
  39
  40 extern void wAbort(int);
  41
  42 static int Aborting=0; /* if we're in the middle of an emergency exit */
  43
  44
  45 static WMHashTable *table = NULL;
  46
  47
  48 void *wmalloc(size_t size)
  49 {
  50     void *tmp;
  51
  52     tmp = malloc(size);
  53     if (tmp == NULL) {
  54         wwarning("malloc() failed. Retrying after 2s.");
  55         sleep(2);
  56         tmp = malloc(size);
  57         if (tmp == NULL) {
  58             if (Aborting) {
  59                 puts("Real Bad Error: recursive malloc() failure.");
  60                 exit(-1);
  61             } else {
  62                 wfatal("virtual memory exhausted");
  63                 Aborting=1;
  64                 wAbort(False);
  65             }
  66         }
  67     }
  68     return tmp;
  69 }
  70
  71
  72 void *wrealloc(void *ptr, size_t newsize)
  73 {
  74     void *nptr;
  75
  76     if (!ptr) {
  77         nptr = malloc(newsize);
  78     } else {
  79         nptr=realloc(ptr, newsize);
  80     }
  81     if (nptr==NULL) {
  82         printf("Could not do realloc");
  83         return NULL;
  84     }
  85     return nptr;
  86 }
  87
  88
  89 void*
  90 wretain(void *ptr)
  91 {
  92     int *refcount;
  93
  94     if (!table) {
  95         table = WMCreateHashTable(WMIntHashCallbacks);
  96     }
  97
  98     refcount = WMHashGet(table, ptr);
  99     if (!refcount) {
 100         refcount = wmalloc(sizeof(int));
 101         *refcount = 1;
 102         WMHashInsert(table, ptr, refcount);
 103 #ifdef VERBOSE
 104         printf("== %i (%p)\n", *refcount, ptr);
 105 #endif
 106     } else {
 107         (*refcount)++;
 108 #ifdef VERBOSE
 109         printf("+ %i (%p)\n", *refcount, ptr);
 110 #endif
 111     }
 112
 113     return ptr;
 114 }
 115
 116
 117 void
 118 wrelease(void *ptr)
 119 {
 120     int *refcount;
 121
 122     refcount = WMHashGet(table, ptr);
 123     if (!refcount) {
 124         wwarning("trying to release unexisting data %p", ptr);
 125     } else {
 126         (*refcount)--;
 127         if (*refcount < 1) {
 128 #ifdef VERBOSE
 129             printf("RELEASING %p\n", ptr);
 130 #endif
 131             WMHashRemove(table, ptr);
 132             free(refcount);
 133             free(ptr);
 134         }
 135 #ifdef VERBOSE
 136         else {
 137             printf("- %i (%p)\n", *refcount, ptr);
 138         }
 139 #endif
 140     }
 141 }
 142
 143
 144 char*
 145 wstrdup(char *str)
 146 {
 147     assert(str!=NULL);
 148
 149     return strcpy(wmalloc(strlen(str)+1), str);
 150 }
 151
 152
 153 char*
 154 wstrappend(char *dst, char *src)
 155 {
 156     char *str;
 157
 158     if (!dst)
 159         return wstrdup(src);
 160     else if (!src)
 161         return wstrdup(dst);
 162
 163     str = wmalloc(strlen(dst)+strlen(src)+1);
 164     strcpy(str, dst);
 165     strcat(str, src);
 166
 167     return str;
 168 }
 169