git.gag.com Git - debian/sudo/blob - common/alloc.c

   1 /*
   2  * Copyright (c) 1999-2005, 2007, 2010-2011
   3  *      Todd C. Miller <Todd.Miller@courtesan.com>
   4  *
   5  * Permission to use, copy, modify, and distribute this software for any
   6  * purpose with or without fee is hereby granted, provided that the above
   7  * copyright notice and this permission notice appear in all copies.
   8  *
   9  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  10  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  11  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  12  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  13  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  14  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  15  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  16  *
  17  * Sponsored in part by the Defense Advanced Research Projects
  18  * Agency (DARPA) and Air Force Research Laboratory, Air Force
  19  * Materiel Command, USAF, under agreement number F39502-99-1-0512.
  20  */
  21
  22 #include <config.h>
  23
  24 #include <sys/types.h>
  25 #include <sys/param.h>
  26 #include <stdio.h>
  27 #ifdef STDC_HEADERS
  28 # include <stdlib.h>
  29 # include <stddef.h>
  30 #else
  31 # ifdef HAVE_STDLIB_H
  32 #  include <stdlib.h>
  33 # endif
  34 #endif /* STDC_HEADERS */
  35 #ifdef HAVE_STRING_H
  36 # if defined(HAVE_MEMORY_H) && !defined(STDC_HEADERS)
  37 #  include <memory.h>
  38 # endif
  39 # include <string.h>
  40 #endif /* HAVE_STRING_H */
  41 #ifdef HAVE_STRINGS_H
  42 # include <strings.h>
  43 #endif /* HAVE_STRING_H */
  44 #if defined(HAVE_MALLOC_H) && !defined(STDC_HEADERS)
  45 # include <malloc.h>
  46 #endif /* HAVE_MALLOC_H && !STDC_HEADERS */
  47 #ifdef HAVE_INTTYPES_H
  48 # include <inttypes.h>
  49 #endif
  50
  51 #include "missing.h"
  52 #include "alloc.h"
  53 #include "error.h"
  54
  55 /*
  56  * If there is no SIZE_MAX or SIZE_T_MAX we have to assume that size_t
  57  * could be signed (as it is on SunOS 4.x).  This just means that
  58  * emalloc2() and erealloc3() cannot allocate huge amounts on such a
  59  * platform but that is OK since sudo doesn't need to do so anyway.
  60  */
  61 #ifndef SIZE_MAX
  62 # ifdef SIZE_T_MAX
  63 #  define SIZE_MAX      SIZE_T_MAX
  64 # else
  65 #  define SIZE_MAX      INT_MAX
  66 # endif /* SIZE_T_MAX */
  67 #endif /* SIZE_MAX */
  68
  69 /*
  70  * emalloc() calls the system malloc(3) and exits with an error if
  71  * malloc(3) fails.
  72  */
  73 void *
  74 emalloc(size_t size)
  75 {
  76     void *ptr;
  77
  78     if (size == 0)
  79         errorx(1, "internal error, tried to emalloc(0)");
  80
  81     if ((ptr = malloc(size)) == NULL)
  82         errorx(1, "unable to allocate memory");
  83     return ptr;
  84 }
  85
  86 /*
  87  * emalloc2() allocates nmemb * size bytes and exits with an error
  88  * if overflow would occur or if the system malloc(3) fails.
  89  */
  90 void *
  91 emalloc2(size_t nmemb, size_t size)
  92 {
  93     void *ptr;
  94
  95     if (nmemb == 0 || size == 0)
  96         errorx(1, "internal error, tried to emalloc2(0)");
  97     if (nmemb > SIZE_MAX / size)
  98         errorx(1, "internal error, emalloc2() overflow");
  99
 100     size *= nmemb;
 101     if ((ptr = malloc(size)) == NULL)
 102         errorx(1, "unable to allocate memory");
 103     return ptr;
 104 }
 105
 106 /*
 107  * erealloc() calls the system realloc(3) and exits with an error if
 108  * realloc(3) fails.  You can call erealloc() with a NULL pointer even
 109  * if the system realloc(3) does not support this.
 110  */
 111 void *
 112 erealloc(void *ptr, size_t size)
 113 {
 114
 115     if (size == 0)
 116         errorx(1, "internal error, tried to erealloc(0)");
 117
 118     ptr = ptr ? realloc(ptr, size) : malloc(size);
 119     if (ptr == NULL)
 120         errorx(1, "unable to allocate memory");
 121     return ptr;
 122 }
 123
 124 /*
 125  * erealloc3() realloc(3)s nmemb * size bytes and exits with an error
 126  * if overflow would occur or if the system malloc(3)/realloc(3) fails.
 127  * You can call erealloc() with a NULL pointer even if the system realloc(3)
 128  * does not support this.
 129  */
 130 void *
 131 erealloc3(void *ptr, size_t nmemb, size_t size)
 132 {
 133
 134     if (nmemb == 0 || size == 0)
 135         errorx(1, "internal error, tried to erealloc3(0)");
 136     if (nmemb > SIZE_MAX / size)
 137         errorx(1, "internal error, erealloc3() overflow");
 138
 139     size *= nmemb;
 140     ptr = ptr ? realloc(ptr, size) : malloc(size);
 141     if (ptr == NULL)
 142         errorx(1, "unable to allocate memory");
 143     return ptr;
 144 }
 145
 146 /*
 147  * estrdup() is like strdup(3) except that it exits with an error if
 148  * malloc(3) fails.  NOTE: unlike strdup(3), estrdup(NULL) is legal.
 149  */
 150 char *
 151 estrdup(const char *src)
 152 {
 153     char *dst = NULL;
 154     size_t len;
 155
 156     if (src != NULL) {
 157         len = strlen(src);
 158         dst = (char *) emalloc(len + 1);
 159         (void) memcpy(dst, src, len);
 160         dst[len] = '\0';
 161     }
 162     return dst;
 163 }
 164
 165 /*
 166  * estrdup() is like strndup(3) except that it exits with an error if
 167  * malloc(3) fails.  NOTE: unlike strdup(3), estrdup(NULL) is legal.
 168  */
 169 char *
 170 estrndup(const char *src, size_t maxlen)
 171 {
 172     char *dst = NULL;
 173     size_t len;
 174
 175     if (src != NULL) {
 176         len = strlen(src);
 177         if (len > maxlen)
 178             len = maxlen;
 179         dst = (char *) emalloc(len + 1);
 180         (void) memcpy(dst, src, len);
 181         dst[len] = '\0';
 182     }
 183     return dst;
 184 }
 185
 186 /*
 187  * easprintf() calls vasprintf() and exits with an error if vasprintf()
 188  * returns -1 (out of memory).
 189  */
 190 int
 191 easprintf(char **ret, const char *fmt, ...)
 192 {
 193     int len;
 194     va_list ap;
 195     va_start(ap, fmt);
 196     len = vasprintf(ret, fmt, ap);
 197     va_end(ap);
 198
 199     if (len == -1)
 200         errorx(1, "unable to allocate memory");
 201     return len;
 202 }
 203
 204 /*
 205  * evasprintf() calls vasprintf() and exits with an error if vasprintf()
 206  * returns -1 (out of memory).
 207  */
 208 int
 209 evasprintf(char **ret, const char *format, va_list args)
 210 {
 211     int len;
 212
 213     if ((len = vasprintf(ret, format, args)) == -1)
 214         errorx(1, "unable to allocate memory");
 215     return len;
 216 }
 217
 218 /*
 219  * Wrapper for free(3) so we can depend on C89 semantics.
 220  */
 221 void
 222 efree(void *ptr)
 223 {
 224     if (ptr != NULL)
 225         free(ptr);
 226 }