granicus.if.org Git - strace/blob - xmalloc.c

   1 /*
   2  * Copyright (c) 2015 Dmitry V. Levin <ldv@altlinux.org>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. The name of the author may not be used to endorse or promote products
  14  *    derived from this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #ifdef HAVE_CONFIG_H
  29 # include "config.h"
  30 #endif
  31
  32 #include <stdlib.h>
  33 #include <string.h>
  34
  35 #include "error_prints.h"
  36 #include "xmalloc.h"
  37
  38 static void
  39 die_out_of_memory(void)
  40 {
  41         static int recursed;
  42
  43         if (recursed)
  44                 exit(1);
  45         recursed = 1;
  46
  47         error_msg_and_die("Out of memory");
  48 }
  49
  50 void *
  51 xmalloc(size_t size)
  52 {
  53         void *p = malloc(size);
  54
  55         if (!p)
  56                 die_out_of_memory();
  57
  58         return p;
  59 }
  60
  61 void *
  62 xcalloc(size_t nmemb, size_t size)
  63 {
  64         void *p = calloc(nmemb, size);
  65
  66         if (!p)
  67                 die_out_of_memory();
  68
  69         return p;
  70 }
  71
  72 #define HALF_SIZE_T     (((size_t) 1) << (sizeof(size_t) * 4))
  73
  74 void *
  75 xreallocarray(void *ptr, size_t nmemb, size_t size)
  76 {
  77         size_t bytes = nmemb * size;
  78
  79         if ((nmemb | size) >= HALF_SIZE_T &&
  80             size && bytes / size != nmemb)
  81                 die_out_of_memory();
  82
  83         void *p = realloc(ptr, bytes);
  84
  85         if (!p)
  86                 die_out_of_memory();
  87
  88         return p;
  89 }
  90
  91 char *
  92 xstrdup(const char *str)
  93 {
  94         if (!str)
  95                 return NULL;
  96
  97         char *p = strdup(str);
  98
  99         if (!p)
 100                 die_out_of_memory();
 101
 102         return p;
 103 }
 104
 105 char *
 106 xstrndup(const char *str, size_t n)
 107 {
 108         char *p;
 109
 110         if (!str)
 111                 return NULL;
 112
 113 #ifdef HAVE_STRNDUP
 114         p = strndup(str, n);
 115 #else
 116         p = xmalloc(n + 1);
 117 #endif
 118
 119         if (!p)
 120                 die_out_of_memory();
 121
 122 #ifndef HAVE_STRNDUP
 123         strncpy(p, str, n);
 124         p[n] = '\0';
 125 #endif
 126
 127         return p;
 128 }