src/mem.c

   1 /*
   2     Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013, 2014, 2015,
   3     2016, 2017
   4     Ben Kibbey <bjk@luxsci.net>
   5
   6     This file is part of pwmd.
   7
   8     Pwmd is free software: you can redistribute it and/or modify
   9     it under the terms of the GNU General Public License as published by
  10     the Free Software Foundation, either version 2 of the License, or
  11     (at your option) any later version.
  12
  13     Pwmd is distributed in the hope that it will be useful,
  14     but WITHOUT ANY WARRANTY; without even the implied warranty of
  15     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16     GNU General Public License for more details.
  17
  18     You should have received a copy of the GNU General Public License
  19     along with Pwmd.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21 #ifdef HAVE_CONFIG_H
  22 #include <config.h>
  23 #endif
  24
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <unistd.h>
  28 #include <string.h>
  29 #include <stddef.h>
  30
  31 #include "mem.h"
  32
  33 void *
  34 xrealloc_gpgrt (void *p, size_t n)
  35 {
  36   if (!n)
  37     {
  38       xfree (p);
  39       return NULL;
  40     }
  41
  42   if (!p)
  43     return xmalloc (n);
  44
  45   return xrealloc (p, n);
  46 }
  47
  48 #ifndef MEM_DEBUG
  49 struct memchunk_s
  50 {
  51   size_t size;
  52   char data[1];
  53 } __attribute ((packed));
  54
  55 void
  56 xfree (void *ptr)
  57 {
  58   struct memchunk_s *m;
  59   void *p;
  60
  61   if (!ptr)
  62     return;
  63
  64   m = (struct memchunk_s *)((char *)ptr-(offsetof (struct memchunk_s, data)));
  65   p = (void *)((char *)m+(offsetof (struct memchunk_s, data)));
  66   wipememory (p, 0, m->size);
  67   free (m);
  68 }
  69
  70 void *
  71 xmalloc (size_t size)
  72 {
  73   struct memchunk_s *m;
  74
  75   if (!size)
  76     return NULL;
  77
  78   m = malloc (sizeof (struct memchunk_s)+size);
  79   if (!m)
  80     return NULL;
  81
  82   m->size = size;
  83   return (void *)((char *)m+(offsetof (struct memchunk_s, data)));
  84 }
  85
  86 void *
  87 xcalloc (size_t nmemb, size_t size)
  88 {
  89   void *p;
  90   struct memchunk_s *m;
  91
  92   m = malloc (sizeof (struct memchunk_s)+(nmemb*size));
  93   if (!m)
  94     return NULL;
  95
  96   m->size = nmemb*size;
  97   p = (void *)((char *)m+(offsetof (struct memchunk_s, data)));
  98   memset (p, 0, m->size);
  99   return p;
 100 }
 101
 102 void *
 103 xrealloc (void *ptr, size_t size)
 104 {
 105   void *p, *np;
 106   struct memchunk_s *m, *mp;
 107   size_t n;
 108
 109   if (!size && ptr)
 110     {
 111       m = (struct memchunk_s *)((char *)ptr-(offsetof (struct memchunk_s, data)));
 112       p = (void *)((char *)m+(offsetof (struct memchunk_s, data)));
 113       wipememory (p, 0, m->size);
 114       free (m);
 115       return NULL;
 116     }
 117   else if (!ptr)
 118     return xmalloc (size);
 119
 120   m = malloc (sizeof (struct memchunk_s)+size);
 121   if (!m)
 122     return NULL;
 123
 124   m->size = size;
 125   np = (void *)((char *)m+(offsetof (struct memchunk_s, data)));
 126
 127   mp = (struct memchunk_s *)((char *)ptr-(offsetof (struct memchunk_s, data)));
 128   p = (void *)((char *)mp+(offsetof (struct memchunk_s, data)));
 129
 130   n = size > mp->size ? mp->size : size;
 131   memcpy (np, p, n);
 132   wipememory (p, 0, mp->size);
 133
 134   free (mp);
 135   return (void *)((char *)m+(offsetof (struct memchunk_s, data)));
 136 }
 137
 138 #endif