xmalloc.c

   1 /*
   2  * netsniff-ng - the packet sniffing beast
   3  * Copyright 2009, 2010, 2011, 2012 Daniel Borkmann.
   4  * Subject to the GPL, version 2.
   5  */
   6
   7 #define _GNU_SOURCE
   8 #include <stdarg.h>
   9 #include <stdio.h>
  10 #include <stdlib.h>
  11 #include <string.h>
  12 #include <unistd.h>
  13 #include <signal.h>
  14 #include <limits.h>
  15 #include <sys/types.h>
  16
  17 #include "xmalloc.h"
  18 #include "built_in.h"
  19 #include "die.h"
  20 #include "str.h"
  21
  22 void *xmalloc(size_t size)
  23 {
  24         void *ptr;
  25
  26         if (unlikely(size == 0))
  27                 panic("xmalloc: zero size\n");
  28
  29         ptr = malloc(size);
  30         if (unlikely(ptr == NULL))
  31                 panic("xmalloc: out of memory (allocating %zu bytes)\n",
  32                       size);
  33
  34         return ptr;
  35 }
  36
  37 void *xzmalloc(size_t size)
  38 {
  39         void *ptr = xmalloc(size);
  40         memset(ptr, 0, size);
  41         return ptr;
  42 }
  43
  44 void *xmalloc_aligned(size_t size, size_t alignment)
  45 {
  46         int ret;
  47         void *ptr;
  48
  49         if (unlikely(size == 0))
  50                 panic("xmalloc_aligned: zero size\n");
  51
  52         ret = posix_memalign(&ptr, alignment, size);
  53         if (unlikely(ret != 0))
  54                 panic("xmalloc_aligned: out of memory (allocating %zu "
  55                       "bytes)\n", size);
  56
  57         return ptr;
  58 }
  59
  60 void *xzmalloc_aligned(size_t size, size_t alignment)
  61 {
  62         void *ptr = xmalloc_aligned(size, alignment);
  63         memset(ptr, 0, size);
  64         return ptr;
  65 }
  66
  67 void *xmallocz(size_t size)
  68 {
  69         void *ptr;
  70
  71         if (unlikely(size + 1 < size))
  72                 panic("xmallocz: data too large to fit into virtual "
  73                       "memory space\n");
  74
  75         ptr = xmalloc(size + 1);
  76         ((char*) ptr)[size] = 0;
  77
  78         return ptr;
  79 }
  80
  81 void *xmemdupz(const void *data, size_t len)
  82 {
  83         return memcpy(xmallocz(len), data, len);
  84 }
  85
  86 void *xrealloc(void *ptr, size_t nmemb, size_t size)
  87 {
  88         void *new_ptr;
  89         size_t new_size = nmemb * size;
  90
  91         if (unlikely(new_size == 0))
  92                 panic("xrealloc: zero size\n");
  93         if (unlikely(((size_t) ~0) / nmemb < size))
  94                 panic("xrealloc: nmemb * size > SIZE_T_MAX\n");
  95
  96         if (ptr == NULL)
  97                 new_ptr = malloc(new_size);
  98         else
  99                 new_ptr = realloc(ptr, new_size);
 100
 101         if (unlikely(new_ptr == NULL))
 102                 panic("xrealloc: out of memory (new_size %zu bytes)\n",
 103                       new_size);
 104
 105         return new_ptr;
 106 }
 107
 108 void xfree_func(void *ptr)
 109 {
 110         if (unlikely(ptr == NULL))
 111                 panic("xfree: NULL pointer given as argument\n");
 112
 113         free(ptr);
 114 }
 115
 116 char *xstrdup(const char *str)
 117 {
 118         size_t len;
 119         char *cp;
 120
 121         len = strlen(str) + 1;
 122         cp = xmalloc(len);
 123
 124         strlcpy(cp, str, len);
 125
 126         return cp;
 127 }
 128
 129 char *xstrndup(const char *str, size_t size)
 130 {
 131         size_t len;
 132         char *cp;
 133
 134         len = strlen(str) + 1;
 135         if (size < len)
 136                 len = size;
 137
 138         cp = xmalloc(len);
 139
 140         strlcpy(cp, str, len);
 141
 142         return cp;
 143 }