src/mem.c

   1 /*
   2     Copyright (C) 2016
   3     Ben Kibbey <bjk@luxsci.net>
   4
   5     This file is part of libpwmd.
   6
   7     Libpwmd is free software: you can redistribute it and/or modify
   8     it under the terms of the GNU General Public License as published by
   9     the Free Software Foundation, either version 2 of the License, or
  10     (at your option) any later version.
  11
  12     Libpwmd is distributed in the hope that it will be useful,
  13     but WITHOUT ANY WARRANTY; without even the implied warranty of
  14     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15     GNU General Public License for more details.
  16
  17     You should have received a copy of the GNU General Public License
  18     along with libpwmd.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20 #ifdef HAVE_CONFIG_H
  21 #include <config.h>
  22 #endif
  23
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26 #include <unistd.h>
  27 #include <string.h>
  28 #include <stddef.h>
  29
  30 #include "mem.h"
  31
  32 void *
  33 _xrealloc_gpgrt (void *p, size_t n)
  34 {
  35   if (!n)
  36     {
  37       _xfree (p);
  38       return NULL;
  39     }
  40
  41   if (!p)
  42     return _xmalloc (n);
  43
  44   return _xrealloc (p, n);
  45 }
  46
  47 #ifndef MEM_DEBUG
  48
  49 struct memchunk_s
  50 {
  51   size_t size;
  52   char data[1];
  53 };
  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 *)((void *)ptr-(offsetof (struct memchunk_s, data)));
  65   p = (void *)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 *)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 *)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 *)((void *)ptr-(offsetof (struct memchunk_s, data)));
 112       p = (void *)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 *)m+(offsetof (struct memchunk_s, data));
 126
 127   mp = (struct memchunk_s *)((void *)ptr-(offsetof (struct memchunk_s, data)));
 128   p = (void *)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 *)m+(offsetof (struct memchunk_s, data));
 136 }
 137 #endif