dfa/imalloc.c

   1 /* This file is part of the Zebra server.
   2    Copyright (C) 1994-2011 Index Data
   3
   4 Zebra is free software; you can redistribute it and/or modify it under
   5 the terms of the GNU General Public License as published by the Free
   6 Software Foundation; either version 2, or (at your option) any later
   7 version.
   8
   9 Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
  10 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12 for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with this program; if not, write to the Free Software
  16 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  17
  18 */
  19
  20
  21 #include <stdio.h>
  22 #include <assert.h>
  23 #include <stdlib.h>
  24
  25 #include <idzebra/util.h>
  26 #include <yaz/xmalloc.h>
  27 #include "imalloc.h"
  28
  29 #if MEMDEBUG
  30 #define MAG1 0x8fe1
  31 #define MAG2 0x91
  32 #define MAG3 0xee
  33
  34 long alloc       = 0L;
  35 long max_alloc   = 0L;
  36 int  alloc_calls = 0;
  37 int  free_calls  = 0;
  38 #endif
  39
  40 void *imalloc (size_t size)
  41 {
  42 #if MEMDEBUG
  43     size_t words = (4*sizeof(unsigned) -1 + size)/sizeof(unsigned);
  44     char *p = (char *)xmalloc( words*sizeof(unsigned) );
  45     if( !p )
  46         yaz_log (YLOG_FATAL, "No memory: imalloc(%u); c/f %d/%d; %ld/%ld",
  47            size, alloc_calls, free_calls, alloc, max_alloc );
  48     *((unsigned *)p) = size;
  49     ((unsigned *)p)[1] = MAG1;
  50     p += sizeof(unsigned)*2;
  51     size[(unsigned char *) p] = MAG2;
  52     size[(unsigned char *) p+1] = MAG3;
  53     if( (alloc+=size) > max_alloc )
  54         max_alloc = alloc;
  55     ++alloc_calls;
  56     return (void *) p;
  57 #else
  58     void *p = (void *)xmalloc( size );
  59     if( !p )
  60         yaz_log (YLOG_FATAL, "Out of memory (imalloc)" );
  61     return p;
  62 #endif
  63 }
  64
  65 void *icalloc (size_t size)
  66 {
  67 #if MEMDEBUG
  68     unsigned words = (4*sizeof(unsigned) -1 + size)/sizeof(unsigned);
  69     char *p = (char *) xcalloc( words*sizeof(unsigned), 1 );
  70     if( !p )
  71         yaz_log (YLOG_FATAL, "No memory: icalloc(%u); c/f %d/%d; %ld/%ld",
  72            size, alloc_calls, free_calls, alloc, max_alloc );
  73     ((unsigned *)p)[0] = size;
  74     ((unsigned *)p)[1] = MAG1;
  75     p += sizeof(unsigned)*2;
  76     size[(unsigned char *) p] = MAG2;
  77     size[(unsigned char *) p+1] = MAG3;
  78     if( (alloc+=size) > max_alloc )
  79         max_alloc = alloc;
  80     ++alloc_calls;
  81     return (void *)p;
  82 #else
  83     void *p = (void *) xcalloc( size, 1 );
  84     if( !p )
  85         yaz_log (YLOG_FATAL, "Out of memory (icalloc)" );
  86     return p;
  87 #endif
  88 }
  89
  90 void ifree (void *p)
  91 {
  92 #if MEMDEBUG
  93     size_t size;
  94     if( !p )
  95         return;
  96     ++free_calls;
  97     size = (-2)[(unsigned *) p];
  98     if( (-1)[(unsigned *) p] != MAG1 )
  99         yaz_log (YLOG_FATAL,"Internal: ifree(%u) magic 1 corrupted", size );
 100     if( size[(unsigned char *) p] != MAG2 )
 101         yaz_log (YLOG_FATAL,"Internal: ifree(%u) magic 2 corrupted", size );
 102     if( (size+1)[(unsigned char *) p] != MAG3 )
 103         yaz_log (YLOG_FATAL,"Internal: ifree(%u) magic 3 corrupted", size );
 104     alloc -= size;
 105     if( alloc < 0L )
 106         yaz_log (YLOG_FATAL,"Internal: ifree(%u) negative alloc.", size );
 107     xfree( (unsigned *) p-2 );
 108 #else
 109     xfree (p);
 110 #endif
 111 }
 112
 113 #if MEMDEBUG
 114 void imemstat (void)
 115 {
 116     fprintf( stdout, "imalloc: calls malloc/free %d/%d, ",
 117                                      alloc_calls, free_calls );
 118     if( alloc )
 119         fprintf( stdout, "memory cur/max %ld/%ld : unreleased",
 120                                                           alloc, max_alloc );
 121     else
 122         fprintf( stdout, "memory max %ld", max_alloc );
 123     fputc( '\n', stdout );
 124 }
 125 #endif
 126 /*
 127  * Local variables:
 128  * c-basic-offset: 4
 129  * c-file-style: "Stroustrup"
 130  * indent-tabs-mode: nil
 131  * End:
 132  * vim: shiftwidth=4 tabstop=8 expandtab
 133  */
 134