Annotation of src/usr.bin/sudo/alloc.c, Revision 1.2.4.1
1.1 millert 1: /*
1.2.4.1 ! millert 2: * Copyright (c) 1999-2002 Todd C. Miller <Todd.Miller@courtesan.com>
1.1 millert 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:
1.2.4.1 ! millert 37: #include <sys/types.h>
! 38: #include <sys/param.h>
1.1 millert 39: #include <stdio.h>
40: #ifdef STDC_HEADERS
1.2.4.1 ! millert 41: # include <stdlib.h>
! 42: # include <stddef.h>
! 43: #else
! 44: # ifdef HAVE_STDLIB_H
! 45: # include <stdlib.h>
! 46: # endif
1.1 millert 47: #endif /* STDC_HEADERS */
48: #ifdef HAVE_STRING_H
1.2.4.1 ! millert 49: # include <string.h>
! 50: #else
! 51: # ifdef HAVE_STRINGS_H
! 52: # include <strings.h>
! 53: # endif
1.1 millert 54: #endif /* HAVE_STRING_H */
55: #if defined(HAVE_MALLOC_H) && !defined(STDC_HEADERS)
1.2.4.1 ! millert 56: # include <malloc.h>
1.1 millert 57: #endif /* HAVE_MALLOC_H && !STDC_HEADERS */
58:
59: #include "sudo.h"
60:
61: #ifndef lint
1.2.4.1 ! millert 62: static const char rcsid[] = "$Sudo: alloc.c,v 1.11 2002/01/09 16:56:04 millert Exp $";
1.1 millert 63: #endif /* lint */
64:
1.2.4.1 ! millert 65: extern char **Argv; /* from sudo.c */
1.1 millert 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:
1.2.4.1 ! millert 77: if ((ptr = (VOID *) malloc(size)) == NULL) {
1.1 millert 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:
1.2.4.1 ! millert 95: ptr = ptr ? (VOID *) realloc(ptr, size) : (VOID *) malloc(size);
! 96: if (ptr == NULL) {
1.1 millert 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: */
1.2 millert 124: int
1.1 millert 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: }
1.2 millert 151: return(len);
1.1 millert 152: }
153:
154: /*
155: * evasprintf() calls vasprintf() and exits with an error if vasprintf()
156: * returns -1 (out of memory).
157: */
1.2 millert 158: int
1.1 millert 159: evasprintf(ret, format, args)
160: char **ret;
161: const char *format;
162: va_list args;
163: {
1.2 millert 164: int len;
1.1 millert 165:
1.2 millert 166: if ((len = vasprintf(ret, format, args)) == -1) {
1.1 millert 167: (void) fprintf(stderr, "%s: cannot allocate memory!\n", Argv[0]);
168: exit(1);
169: }
1.2 millert 170: return(len);
1.1 millert 171: }