wmaker/WINGs/memory.c

/*
 *  Window Maker miscelaneous function library
 *
 *  Copyright (c) 1997-2003 Alfredo K. Kojima
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */


#include "wconfig.h"
#include "WUtil.h"

#include <stdlib.h>
#include <sys/types.h>
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <assert.h>
#include <signal.h>

#ifdef TEST_WITH_GC
#include <gc/gc.h>
#endif

#ifndef False
# define False 0
#endif
#ifndef True
# define True 1
#endif


static void
defaultHandler(int bla)
{
    if (bla)
        kill(getpid(), SIGABRT);
    else
        exit(1);
}


static waborthandler *aborthandler = (waborthandler*)defaultHandler;

#define wAbort(a) (*aborthandler)(a)


waborthandler*
wsetabort(waborthandler *handler)
{
    waborthandler *old = aborthandler;

    aborthandler = handler;

    return old;
}



static int Aborting=0; /* if we're in the middle of an emergency exit */


static WMHashTable *table = NULL;


void*
wmalloc(size_t size)
{
    void *tmp;

    assert(size > 0);

#ifdef TEST_WITH_GC
    tmp = GC_malloc(size);
#else
    tmp = malloc(size);
#endif
    if (tmp == NULL) {
        wwarning("malloc() failed. Retrying after 2s.");
        sleep(2);
#ifdef TEST_WITH_GC
        tmp = GC_malloc(size);
#else
        tmp = malloc(size);
#endif
        if (tmp == NULL) {
            if (Aborting) {
                fputs("Really Bad Error: recursive malloc() failure.", stderr);
                exit(-1);
            } else {
                wfatal("virtual memory exhausted");
                Aborting=1;
                wAbort(False);
            }
        }
    }
    return tmp;
}


void*
wmalloc0(size_t size)
{
    void *ptr= wmalloc(size);
    if (!ptr)
      return NULL;

    memset(ptr, 0, size);

    return ptr;
}


void*
wrealloc(void *ptr, size_t newsize)
{
    void *nptr;

    if (!ptr) {
        nptr = wmalloc(newsize);
    } else if (newsize==0) {
        wfree(ptr);
        nptr = NULL;
    } else {
#ifdef TEST_WITH_GC
        nptr = GC_realloc(ptr, newsize);
#else
        nptr = realloc(ptr, newsize);
#endif
        if (nptr==NULL) {
            wwarning("realloc() failed. Retrying after 2s.");
            sleep(2);
#ifdef TEST_WITH_GC
            nptr = GC_realloc(ptr, newsize);
#else
            nptr = realloc(ptr, newsize);
#endif
            if (nptr == NULL) {
                if (Aborting) {
                    fputs("Really Bad Error: recursive realloc() failure.",
                          stderr);
                    exit(-1);
                } else {
                    wfatal("virtual memory exhausted");
                    Aborting=1;
                    wAbort(False);
                }
            }
        }
    }
    return nptr;
}


void*
wretain(void *ptr)
{
    int *refcount;

    if (!table) {
        table = WMCreateHashTable(WMIntHashCallbacks);
    }

    refcount = WMHashGet(table, ptr);
    if (!refcount) {
        refcount = wmalloc(sizeof(int));
        *refcount = 1;
        WMHashInsert(table, ptr, refcount);
#ifdef VERBOSE
        printf("== %i (%p)\n", *refcount, ptr);
#endif
    } else {
        (*refcount)++;
#ifdef VERBOSE
        printf("+ %i (%p)\n", *refcount, ptr);
#endif
    }

    return ptr;
}



void
wfree(void *ptr)
{
#ifdef TEST_WITH_GC
    GC_free(ptr);
#else
    free(ptr);
#endif
}



void
wrelease(void *ptr)
{
    int *refcount;

    refcount = WMHashGet(table, ptr);
    if (!refcount) {
        wwarning("trying to release unexisting data %p", ptr);
    } else {
        (*refcount)--;
        if (*refcount < 1) {
#ifdef VERBOSE
            printf("RELEASING %p\n", ptr);
#endif
            WMHashRemove(table, ptr);
            wfree(refcount);
            wfree(ptr);
        }
#ifdef VERBOSE
        else {
            printf("- %i (%p)\n", *refcount, ptr);
        }
#endif
    }
}