git.meshlink.io Git - meshlink/blob - src/xalloc.h

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