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

   1 /*
   2  * Copyright (c) 1999-2002 Todd C. Miller <Todd.Miller@courtesan.com>
   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  *
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  *
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * 3. The name of the author may not be used to endorse or promote products
  17  *    derived from this software without specific prior written permission.
  18  *
  19  * 4. Products derived from this software may not be called "Sudo" nor
  20  *    may "Sudo" appear in their names without specific prior written
  21  *    permission from the author.
  22  *
  23  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
  24  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
  25  * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
  26  * THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  27  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  28  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
  29  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  30  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  31  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  32  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  33  */
  34
  35 #include "config.h"
  36
  37 #include <sys/types.h>
  38 #include <sys/param.h>
  39 #include <stdio.h>
  40 #ifdef STDC_HEADERS
  41 # include <stdlib.h>
  42 # include <stddef.h>
  43 #else
  44 # ifdef HAVE_STDLIB_H
  45 #  include <stdlib.h>
  46 # endif
  47 #endif /* STDC_HEADERS */
  48 #ifdef HAVE_STRING_H
  49 # include <string.h>
  50 #else
  51 # ifdef HAVE_STRINGS_H
  52 #  include <strings.h>
  53 # endif
  54 #endif /* HAVE_STRING_H */
  55 #if defined(HAVE_MALLOC_H) && !defined(STDC_HEADERS)
  56 # include <malloc.h>
  57 #endif /* HAVE_MALLOC_H && !STDC_HEADERS */
  58
  59 #include "sudo.h"
  60
  61 #ifndef lint
  62 static const char rcsid[] = "$Sudo: alloc.c,v 1.11 2002/01/09 16:56:04 millert Exp $";
  63 #endif /* lint */
  64
  65 extern char **Argv;             /* from sudo.c */
  66
  67 /*
  68  * emalloc() calls the system malloc(3) and exits with an error if
  69  * malloc(3) fails.
  70  */
  71 VOID *
  72 emalloc(size)
  73     size_t size;
  74 {
  75     VOID *ptr;
  76
  77     if ((ptr = (VOID *) malloc(size)) == NULL) {
  78         (void) fprintf(stderr, "%s: cannot allocate memory!\n", Argv[0]);
  79         exit(1);
  80     }
  81     return(ptr);
  82 }
  83
  84 /*
  85  * erealloc() calls the system realloc(3) and exits with an error if
  86  * realloc(3) fails.  You can call erealloc() with a NULL pointer even
  87  * if the system realloc(3) does not support this.
  88  */
  89 VOID *
  90 erealloc(ptr, size)
  91     VOID *ptr;
  92     size_t size;
  93 {
  94
  95     ptr = ptr ? (VOID *) realloc(ptr, size) : (VOID *) malloc(size);
  96     if (ptr == NULL) {
  97         (void) fprintf(stderr, "%s: cannot allocate memory!\n", Argv[0]);
  98         exit(1);
  99     }
 100     return(ptr);
 101 }
 102
 103 /*
 104  * estrdup() is like strdup(3) except that it exits with an error if
 105  * malloc(3) fails.  NOTE: unlike strdup(3), estrdup(NULL) is legal.
 106  */
 107 char *
 108 estrdup(src)
 109     const char *src;
 110 {
 111     char *dst = NULL;
 112
 113     if (src != NULL) {
 114         dst = (char *) emalloc(strlen(src) + 1);
 115         (void) strcpy(dst, src);
 116     }
 117     return(dst);
 118 }
 119
 120 /*
 121  * easprintf() calls vasprintf() and exits with an error if vasprintf()
 122  * returns -1 (out of memory).
 123  */
 124 int
 125 #ifdef __STDC__
 126 easprintf(char **ret, const char *fmt, ...)
 127 #else
 128 easprintf(va_alist)
 129     va_dcl
 130 #endif
 131 {
 132     int len;
 133     va_list ap;
 134 #ifdef __STDC__
 135     va_start(ap, fmt);
 136 #else
 137     char **ret;
 138     const char *fmt;
 139
 140     va_start(ap);
 141     ret = va_arg(ap, char **);
 142     fmt = va_arg(ap, const char *);
 143 #endif
 144     len = vasprintf(ret, fmt, ap);
 145     va_end(ap);
 146
 147     if (len == -1) {
 148         (void) fprintf(stderr, "%s: cannot allocate memory!\n", Argv[0]);
 149         exit(1);
 150     }
 151     return(len);
 152 }
 153
 154 /*
 155  * evasprintf() calls vasprintf() and exits with an error if vasprintf()
 156  * returns -1 (out of memory).
 157  */
 158 int
 159 evasprintf(ret, format, args)
 160     char **ret;
 161     const char *format;
 162     va_list args;
 163 {
 164     int len;
 165
 166     if ((len = vasprintf(ret, format, args)) == -1) {
 167         (void) fprintf(stderr, "%s: cannot allocate memory!\n", Argv[0]);
 168         exit(1);
 169     }
 170     return(len);
 171 }