lib/libmallocab/libmallocab.c

   1
   2 #include <malloc.h>
   3 #include <stdlib.h>
   4 #include <string.h>
   5 #include <err.h>
   6
   7 /*
   8    This lib implements the "aborting" versions of
   9     - malloc
  10     - realloc
  11     - strdup
  12     - strndup
  13    The program gets abort()ed if they can not allocate enough memory.
  14    Thus the caller don't need to pay attention to returning NULL pointers.
  15    See abort(3).
  16  */
  17
  18 void* mallocab(size_t size)
  19 {
  20         void* ptr;
  21         ptr = malloc(size);
  22         if(size != 0 && ptr == NULL)
  23         {
  24                 warnx("Failed to allocate %d bytes of memory.", size);
  25                 abort();
  26         }
  27         return ptr;
  28 }
  29
  30 void* reallocab(void* ptr0, size_t size)
  31 {
  32         void* ptr;
  33         ptr = realloc(ptr0, size);
  34         if(size != 0 && ptr == NULL)
  35         {
  36                 warnx("Failed to reallocate %d bytes of memory.", size);
  37                 abort();
  38         }
  39         return ptr;
  40 }
  41
  42 char* strdupab(const char* ptr0)
  43 {
  44         char* ptr;
  45         ptr = strdup(ptr0);
  46         if(ptr == NULL)
  47         {
  48                 warnx("Failed to duplicate %p.", ptr0);
  49                 abort();
  50         }
  51         return ptr;
  52 }
  53
  54 char* strndupab(const char* ptr0, size_t size)
  55 {
  56         char* ptr;
  57         ptr = strndup(ptr0, size);
  58         if(ptr == NULL)
  59         {
  60                 warnx("Failed to duplicate %d bytes from %p.", size, ptr0);
  61                 abort();
  62         }
  63         return ptr;
  64 }