Annotation of embedaddon/sudo/common/alloc.c, revision 1.1.1.4
1.1 misho 1: /*
1.1.1.4 ! misho 2: * Copyright (c) 1999-2005, 2007, 2010-2013
1.1 misho 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 <stdio.h>
26: #ifdef STDC_HEADERS
27: # include <stdlib.h>
28: # include <stddef.h>
29: #else
30: # ifdef HAVE_STDLIB_H
31: # include <stdlib.h>
32: # endif
33: #endif /* STDC_HEADERS */
34: #ifdef HAVE_STRING_H
35: # if defined(HAVE_MEMORY_H) && !defined(STDC_HEADERS)
36: # include <memory.h>
37: # endif
38: # include <string.h>
39: #endif /* HAVE_STRING_H */
40: #ifdef HAVE_STRINGS_H
41: # include <strings.h>
42: #endif /* HAVE_STRING_H */
43: #if defined(HAVE_MALLOC_H) && !defined(STDC_HEADERS)
44: # include <malloc.h>
45: #endif /* HAVE_MALLOC_H && !STDC_HEADERS */
46: #ifdef HAVE_INTTYPES_H
47: # include <inttypes.h>
48: #endif
49:
50: #include "missing.h"
51: #include "alloc.h"
52: #include "error.h"
1.1.1.4 ! misho 53: #include "errno.h"
1.1 misho 54:
55: #define DEFAULT_TEXT_DOMAIN "sudo"
56: #include "gettext.h"
57:
58: /*
59: * If there is no SIZE_MAX or SIZE_T_MAX we have to assume that size_t
60: * could be signed (as it is on SunOS 4.x). This just means that
61: * emalloc2() and erealloc3() cannot allocate huge amounts on such a
62: * platform but that is OK since sudo doesn't need to do so anyway.
63: */
64: #ifndef SIZE_MAX
65: # ifdef SIZE_T_MAX
66: # define SIZE_MAX SIZE_T_MAX
67: # else
68: # define SIZE_MAX INT_MAX
69: # endif /* SIZE_T_MAX */
70: #endif /* SIZE_MAX */
71:
72: /*
73: * emalloc() calls the system malloc(3) and exits with an error if
74: * malloc(3) fails.
75: */
76: void *
77: emalloc(size_t size)
78: {
79: void *ptr;
80:
81: if (size == 0)
1.1.1.4 ! misho 82: fatalx_nodebug(_("internal error, tried to emalloc(0)"));
1.1 misho 83:
84: if ((ptr = malloc(size)) == NULL)
1.1.1.4 ! misho 85: fatalx_nodebug(NULL);
1.1 misho 86: return ptr;
87: }
88:
89: /*
90: * emalloc2() allocates nmemb * size bytes and exits with an error
91: * if overflow would occur or if the system malloc(3) fails.
92: */
93: void *
94: emalloc2(size_t nmemb, size_t size)
95: {
96: void *ptr;
97:
98: if (nmemb == 0 || size == 0)
1.1.1.4 ! misho 99: fatalx_nodebug(_("internal error, tried to emalloc2(0)"));
1.1 misho 100: if (nmemb > SIZE_MAX / size)
1.1.1.4 ! misho 101: fatalx_nodebug(_("internal error, %s overflow"), "emalloc2()");
1.1 misho 102:
103: size *= nmemb;
104: if ((ptr = malloc(size)) == NULL)
1.1.1.4 ! misho 105: fatalx_nodebug(NULL);
1.1.1.2 misho 106: return ptr;
107: }
108:
109: /*
110: * ecalloc() allocates nmemb * size bytes and exits with an error
111: * if overflow would occur or if the system malloc(3) fails.
112: * On success, the allocated space is zero-filled.
113: */
114: void *
115: ecalloc(size_t nmemb, size_t size)
116: {
117: void *ptr;
118:
119: if (nmemb == 0 || size == 0)
1.1.1.4 ! misho 120: fatalx_nodebug(_("internal error, tried to ecalloc(0)"));
1.1.1.2 misho 121: if (nmemb != 1) {
122: if (nmemb > SIZE_MAX / size)
1.1.1.4 ! misho 123: fatalx_nodebug(_("internal error, %s overflow"), "ecalloc()");
1.1.1.2 misho 124: size *= nmemb;
125: }
126: if ((ptr = malloc(size)) == NULL)
1.1.1.4 ! misho 127: fatalx_nodebug(NULL);
1.1.1.2 misho 128: memset(ptr, 0, size);
1.1 misho 129: return ptr;
130: }
131:
132: /*
133: * erealloc() calls the system realloc(3) and exits with an error if
134: * realloc(3) fails. You can call erealloc() with a NULL pointer even
135: * if the system realloc(3) does not support this.
136: */
137: void *
138: erealloc(void *ptr, size_t size)
139: {
140:
141: if (size == 0)
1.1.1.4 ! misho 142: fatalx_nodebug(_("internal error, tried to erealloc(0)"));
1.1 misho 143:
144: ptr = ptr ? realloc(ptr, size) : malloc(size);
145: if (ptr == NULL)
1.1.1.4 ! misho 146: fatalx_nodebug(NULL);
1.1 misho 147: return ptr;
148: }
149:
150: /*
151: * erealloc3() realloc(3)s nmemb * size bytes and exits with an error
152: * if overflow would occur or if the system malloc(3)/realloc(3) fails.
153: * You can call erealloc() with a NULL pointer even if the system realloc(3)
154: * does not support this.
155: */
156: void *
157: erealloc3(void *ptr, size_t nmemb, size_t size)
158: {
159:
160: if (nmemb == 0 || size == 0)
1.1.1.4 ! misho 161: fatalx_nodebug(_("internal error, tried to erealloc3(0)"));
1.1 misho 162: if (nmemb > SIZE_MAX / size)
1.1.1.4 ! misho 163: fatalx_nodebug(_("internal error, %s overflow"), "erealloc3()");
1.1 misho 164:
165: size *= nmemb;
166: ptr = ptr ? realloc(ptr, size) : malloc(size);
167: if (ptr == NULL)
1.1.1.4 ! misho 168: fatalx_nodebug(NULL);
1.1.1.2 misho 169: return ptr;
170: }
171:
172: #ifdef notyet
173: /*
174: * erecalloc() realloc(3)s nmemb * msize bytes and exits with an error
175: * if overflow would occur or if the system malloc(3)/realloc(3) fails.
176: * On success, the new space is zero-filled. You can call ereallocz()
177: * with a NULL pointer even if the system realloc(3) does not support this.
178: */
179: void *
180: erecalloc(void *ptr, size_t onmemb, size_t nmemb, size_t msize)
181: {
182: size_t size;
183:
184: if (nmemb == 0 || msize == 0)
1.1.1.4 ! misho 185: fatalx_nodebug(_("internal error, tried to erecalloc(0)"));
1.1.1.2 misho 186: if (nmemb > SIZE_MAX / msize)
1.1.1.4 ! misho 187: fatalx_nodebug(_("internal error, %s overflow"), "erecalloc()");
1.1.1.2 misho 188:
189: size = nmemb * msize;
190: ptr = ptr ? realloc(ptr, size) : malloc(size);
191: if (ptr == NULL)
1.1.1.4 ! misho 192: fatalx_nodebug(NULL);
1.1.1.2 misho 193: if (nmemb > onmemb) {
194: size = (nmemb - onmemb) * msize;
195: memset((char *)ptr + (onmemb * msize), 0, size);
196: }
1.1 misho 197: return ptr;
198: }
1.1.1.2 misho 199: #endif
1.1 misho 200:
201: /*
202: * estrdup() is like strdup(3) except that it exits with an error if
203: * malloc(3) fails. NOTE: unlike strdup(3), estrdup(NULL) is legal.
204: */
205: char *
206: estrdup(const char *src)
207: {
208: char *dst = NULL;
209: size_t len;
210:
211: if (src != NULL) {
212: len = strlen(src);
213: dst = (char *) emalloc(len + 1);
214: (void) memcpy(dst, src, len);
215: dst[len] = '\0';
216: }
217: return dst;
218: }
219:
220: /*
221: * estrdup() is like strndup(3) except that it exits with an error if
222: * malloc(3) fails. NOTE: unlike strdup(3), estrdup(NULL) is legal.
223: */
224: char *
225: estrndup(const char *src, size_t maxlen)
226: {
227: char *dst = NULL;
1.1.1.2 misho 228: size_t len = 0;
1.1 misho 229:
230: if (src != NULL) {
1.1.1.2 misho 231: while (maxlen != 0 && src[len] != '\0') {
232: len++;
233: maxlen--;
234: }
1.1 misho 235: dst = (char *) emalloc(len + 1);
236: (void) memcpy(dst, src, len);
237: dst[len] = '\0';
238: }
239: return dst;
240: }
241:
242: /*
243: * easprintf() calls vasprintf() and exits with an error if vasprintf()
244: * returns -1 (out of memory).
245: */
246: int
247: easprintf(char **ret, const char *fmt, ...)
248: {
249: int len;
250: va_list ap;
1.1.1.4 ! misho 251:
1.1 misho 252: va_start(ap, fmt);
253: len = vasprintf(ret, fmt, ap);
254: va_end(ap);
255:
256: if (len == -1)
1.1.1.4 ! misho 257: fatalx_nodebug(NULL);
1.1 misho 258: return len;
259: }
260:
261: /*
262: * evasprintf() calls vasprintf() and exits with an error if vasprintf()
263: * returns -1 (out of memory).
264: */
265: int
266: evasprintf(char **ret, const char *format, va_list args)
267: {
268: int len;
269:
270: if ((len = vasprintf(ret, format, args)) == -1)
1.1.1.4 ! misho 271: fatalx_nodebug(NULL);
1.1 misho 272: return len;
273: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>