src/mem.c

   1 /*
   2     Copyright (C) 2006-2018 Ben Kibbey <bjk@luxsci.net>
   3
   4     This file is part of pwmd.
   5
   6     Pwmd 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 of the License, or
   9     (at your option) any later version.
  10
  11     Pwmd 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 Pwmd.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19 #ifdef HAVE_CONFIG_H
  20 #include <config.h>
  21 #endif
  22
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <unistd.h>
  26 #include <string.h>
  27 #include <stddef.h>
  28
  29 #include "mem.h"
  30
  31 void *
  32 xrealloc_gpgrt (void *p, size_t n)
  33 {
  34   if (n)
  35     {
  36       if (p)
  37         return xrealloc (p, n);
  38
  39       return xmalloc (n);
  40     }
  41
  42   xfree (p);
  43   return NULL;
  44 }
  45
  46 #ifndef MEM_DEBUG
  47 struct memchunk_s
  48 {
  49   size_t size;
  50   char data[1];
  51 } __attribute ((packed));
  52
  53 void
  54 xfree (void *ptr)
  55 {
  56   struct memchunk_s *m;
  57   void *p;
  58
  59   if (!ptr)
  60     return;
  61
  62   m = (struct memchunk_s *)((char *)ptr-(offsetof (struct memchunk_s, data)));
  63   p = (void *)((char *)m+(offsetof (struct memchunk_s, data)));
  64   wipememory (p, 0, m->size);
  65   free (m);
  66 }
  67
  68 void *
  69 xmalloc (size_t size)
  70 {
  71   struct memchunk_s *m;
  72
  73   if (!size)
  74     return NULL;
  75
  76   m = malloc (sizeof (struct memchunk_s)+size);
  77   if (!m)
  78     return NULL;
  79
  80   m->size = size;
  81   return (void *)((char *)m+(offsetof (struct memchunk_s, data)));
  82 }
  83
  84 void *
  85 xcalloc (size_t nmemb, size_t size)
  86 {
  87   void *p;
  88   struct memchunk_s *m;
  89
  90   m = malloc (sizeof (struct memchunk_s)+(nmemb*size));
  91   if (!m)
  92     return NULL;
  93
  94   m->size = nmemb*size;
  95   p = (void *)((char *)m+(offsetof (struct memchunk_s, data)));
  96   memset (p, 0, m->size);
  97   return p;
  98 }
  99
 100 void *
 101 xrealloc (void *ptr, size_t size)
 102 {
 103   void *p, *np;
 104   struct memchunk_s *m, *mp;
 105   size_t n;
 106
 107   if (!size && ptr)
 108     {
 109       m = (struct memchunk_s *)((char *)ptr-(offsetof (struct memchunk_s, data)));
 110       p = (void *)((char *)m+(offsetof (struct memchunk_s, data)));
 111       wipememory (p, 0, m->size);
 112       free (m);
 113       return NULL;
 114     }
 115   else if (!ptr)
 116     return xmalloc (size);
 117
 118   m = malloc (sizeof (struct memchunk_s)+size);
 119   if (!m)
 120     return NULL;
 121
 122   m->size = size;
 123   np = (void *)((char *)m+(offsetof (struct memchunk_s, data)));
 124
 125   mp = (struct memchunk_s *)((char *)ptr-(offsetof (struct memchunk_s, data)));
 126   p = (void *)((char *)mp+(offsetof (struct memchunk_s, data)));
 127
 128   n = size > mp->size ? mp->size : size;
 129   memcpy (np, p, n);
 130   wipememory (p, 0, mp->size);
 131
 132   free (mp);
 133   return (void *)((char *)m+(offsetof (struct memchunk_s, data)));
 134 }
 135
 136 #endif