include/ircd_alloc.h

   1 /*
   2  * IRC - Internet Relay Chat, include/ircd_alloc.h
   3  * Copyright (C) 1999 Thomas Helvey <tomh@inxpress.net>
   4  *
   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, or (at your option)
   9  * 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  * Commentary by Bleep (Thomas Helvey)
  21  *
  22  * $Id$
  23  */
  24 #ifndef INCLUDED_ircd_alloc_h
  25 #define INCLUDED_ircd_alloc_h
  26
  27 /*
  28  * memory resource allocation and test functions
  29  */
  30 typedef void (*OutOfMemoryHandler)(void);
  31 extern void set_nomem_handler(OutOfMemoryHandler handler);
  32
  33 /* The mappings for the My* functions... */
  34 #define MyMalloc(size) \
  35   DoMalloc(size, "malloc", __FILE__, __LINE__)
  36
  37 #define MyCalloc(nelem, size) \
  38   DoMallocZero(size * nelem, "calloc", __FILE__, __LINE__)
  39
  40 #define MyFree(p) \
  41   if (p) \
  42     DoFree(p, __FILE__, __LINE__)
  43
  44 /* No realloc because it is not currently used, and it is not really the
  45  * nicest function to be using anyway(i.e. its evil if you want it
  46  * go ahead and write it).
  47  */
  48
  49 extern void *malloc_tmp;
  50
  51 /* First version: fast non-debugging macros... */
  52 #ifndef MDEBUG
  53 #ifndef INCLUDED_stdlib_h
  54 #include <stdlib.h> /* free */
  55 #define INCLUDED_stdlib_h
  56 #endif
  57
  58 extern OutOfMemoryHandler noMemHandler;
  59
  60 #define DoFree(x, file, line) do { free((x)); (x) = 0; } while(0)
  61 #define DoMalloc(size, type, file, line) \
  62   (\
  63      (malloc_tmp = malloc(size), \
  64      (malloc_tmp == NULL) ? noMemHandler() : 0), \
  65   malloc_tmp)
  66
  67 #define DoMallocZero(size, type, file, line) \
  68   (\
  69     (DoMalloc(size, type, file, line), \
  70     memset(malloc_tmp, 0, size)), \
  71   malloc_tmp)
  72
  73 /* Second version: slower debugging versions... */
  74 #else /* defined(MDEBUG) */
  75 #include <sys/types.h>
  76 #include "memdebug.h"
  77
  78 #define DoMalloc(size, type, file, line) \
  79   dbg_malloc(size, type, file, line)
  80 #define DoMallocZero(size, type, file, line) \
  81   dbg_malloc_zero(size, type, file, line)
  82 #define DoFree(p, file, line) \
  83   dbg_free(p, file, line)
  84 #endif /* defined(MDEBUG) */
  85
  86 #endif /* INCLUDED_ircd_alloc_h */