src/xalloc.h

   1 #ifndef TINC_XALLOC_H
   2 #define TINC_XALLOC_H
   3
   4 /*
   5    xalloc.h -- malloc and related fuctions with out of memory checking
   6    Copyright (C) 1990, 91, 92, 93, 94, 95, 96, 97 Free Software Foundation, Inc.
   7    Copyright (C) 2011-2013 Guus Sliepen <guus@tinc-vpn.org>
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 2, or (at your option)
  12    any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License along
  20    with this program; if not, write to the Free Software Foundation, Inc., Foundation,
  21    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  22 */
  23
  24 static inline void *xmalloc(size_t n) __attribute__((__malloc__));
  25 static inline void *xmalloc(size_t n) {
  26         void *p = malloc(n);
  27
  28         if(!p) {
  29                 abort();
  30         }
  31
  32         return p;
  33 }
  34
  35 static inline void *xzalloc(size_t n) __attribute__((__malloc__));
  36 static inline void *xzalloc(size_t n) {
  37         void *p = calloc(1, n);
  38
  39         if(!p) {
  40                 abort();
  41         }
  42
  43         return p;
  44 }
  45
  46 static inline void *xrealloc(void *p, size_t n) {
  47         p = realloc(p, n);
  48
  49         if(!p) {
  50                 abort();
  51         }
  52
  53         return p;
  54 }
  55
  56 static inline char *xstrdup(const char *s) __attribute__((__malloc__));
  57 static inline char *xstrdup(const char *s) {
  58         char *p = strdup(s);
  59
  60         if(!p) {
  61                 abort();
  62         }
  63
  64         return p;
  65 }
  66
  67 static inline int xvasprintf(char **strp, const char *fmt, va_list ap) {
  68 #ifdef HAVE_MINGW
  69         char buf[1024];
  70         int result = vsnprintf(buf, sizeof(buf), fmt, ap);
  71
  72         if(result < 0) {
  73                 abort();
  74         }
  75
  76         *strp = xstrdup(buf);
  77 #else
  78         int result = vasprintf(strp, fmt, ap);
  79
  80         if(result < 0) {
  81                 abort();
  82         }
  83
  84 #endif
  85         return result;
  86 }
  87
  88 static inline int xasprintf(char **strp, const char *fmt, ...) __attribute__((__format__(printf, 2, 3)));
  89 static inline int xasprintf(char **strp, const char *fmt, ...) {
  90         va_list ap;
  91         va_start(ap, fmt);
  92         int result = xvasprintf(strp, fmt, ap);
  93         va_end(ap);
  94         return result;
  95 }
  96
  97 #endif