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   DoFree(p, __FILE__, __LINE__)
  42
  43 /* No realloc because it is not currently used, and it is not really the
  44  * nicest function to be using anyway(i.e. its evil if you want it
  45  * go ahead and write it).
  46  */
  47
  48 extern void *malloc_tmp;
  49
  50 /* First version: fast non-debugging macros... */
  51 #ifndef MDEBUG
  52 #ifndef INCLUDED_stdlib_h
  53 #include <stdlib.h> /* free */
  54 #define INCLUDED_stdlib_h
  55 #endif
  56
  57 extern OutOfMemoryHandler noMemHandler;
  58
  59 #define DoFree(x, file, line) do { free((x)); (x) = 0; } while(0)
  60 #define DoMalloc(size, type, file, line) \
  61   (\
  62      (malloc_tmp = malloc(size), \
  63      (malloc_tmp == NULL) ? noMemHandler() : 0), \
  64   malloc_tmp)
  65
  66 #define DoMallocZero(size, type, file, line) \
  67   (\
  68     (DoMalloc(size, type, file, line), \
  69     memset(malloc_tmp, 0, size)), \
  70   malloc_tmp)
  71
  72 /* Second version: slower debugging versions... */
  73 #else /* defined(MDEBUG) */
  74 #include <sys/types.h>
  75 #include "memdebug.h"
  76
  77 #define DoMalloc(size, type, file, line) \
  78   dbg_malloc(size, type, file, line)
  79 #define DoMallocZero(size, type, file, line) \
  80   dbg_malloc_zero(size, type, file, line)
  81 #define DoFree(p, file, line) \
  82   dbg_free(p, file, line)
  83 #endif /* defined(MDEBUG) */
  84
  85 #endif /* INCLUDED_ircd_alloc_h */