Commit | Line | Data |
---|---|---|
252b5132 | 1 | /* memory allocation routines with error checking. |
c1687039 | 2 | Copyright 1989, 90, 91, 92, 93, 94 Free Software Foundation, Inc. |
252b5132 RH |
3 | |
4 | This file is part of the libiberty library. | |
5 | Libiberty is free software; you can redistribute it and/or | |
6 | modify it under the terms of the GNU Library General Public | |
7 | License as published by the Free Software Foundation; either | |
8 | version 2 of the License, or (at your option) any later version. | |
9 | ||
10 | Libiberty is distributed in the hope that it will be useful, | |
11 | but WITHOUT ANY WARRANTY; without even the implied warranty of | |
12 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
13 | Library General Public License for more details. | |
14 | ||
15 | You should have received a copy of the GNU Library General Public | |
16 | License along with libiberty; see the file COPYING.LIB. If | |
17 | not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, | |
18 | Boston, MA 02111-1307, USA. */ | |
19 | ||
39423523 DD |
20 | /* |
21 | ||
22 | @deftypefn Replacement void* xmalloc (size_t) | |
23 | ||
24 | Allocate memory without fail. If @code{malloc} fails, this will print | |
fa9f0e33 DD |
25 | a message to @code{stderr} (using the name set by |
26 | @code{xmalloc_set_program_name}, | |
39423523 DD |
27 | if any) and then call @code{xexit}. Note that it is therefore safe for |
28 | a program to contain @code{#define malloc xmalloc} in its source. | |
29 | ||
30 | @end deftypefn | |
31 | ||
99b58139 | 32 | @deftypefn Replacement void* xrealloc (void *@var{ptr}, size_t @var{size}) |
39423523 DD |
33 | Reallocate memory without fail. This routine functions like @code{realloc}, |
34 | but will behave the same as @code{xmalloc} if memory cannot be found. | |
35 | ||
36 | @end deftypefn | |
37 | ||
99b58139 | 38 | @deftypefn Replacement void* xcalloc (size_t @var{nelem}, size_t @var{elsize}) |
39423523 DD |
39 | |
40 | Allocate memory without fail, and set it to zero. This routine functions | |
41 | like @code{calloc}, but will behave the same as @code{xmalloc} if memory | |
42 | cannot be found. | |
43 | ||
44 | @end deftypefn | |
45 | ||
46 | @deftypefn Replacement void xmalloc_set_program_name (const char *@var{name}) | |
47 | ||
48 | You can use this to set the name of the program used by | |
49 | @code{xmalloc_failed} when printing a failure message. | |
50 | ||
51 | @end deftypefn | |
52 | ||
53 | @deftypefn Replacement void xmalloc_failed (size_t) | |
54 | ||
55 | This function is not meant to be called by client code, and is listed | |
56 | here for completeness only. If any of the allocation routines fail, this | |
57 | function will be called to print an error message and terminate execution. | |
58 | ||
59 | @end deftypefn | |
60 | ||
61 | */ | |
62 | ||
eb383413 L |
63 | #ifdef HAVE_CONFIG_H |
64 | #include "config.h" | |
65 | #endif | |
252b5132 RH |
66 | #include "ansidecl.h" |
67 | #include "libiberty.h" | |
68 | ||
69 | #include <stdio.h> | |
70 | ||
71 | #ifdef __STDC__ | |
72 | #include <stddef.h> | |
73 | #else | |
74 | #define size_t unsigned long | |
75 | #define ptrdiff_t long | |
76 | #endif | |
77 | ||
78 | #if VMS | |
79 | #include <stdlib.h> | |
80 | #include <unixlib.h> | |
81 | #else | |
82 | /* For systems with larger pointers than ints, these must be declared. */ | |
83 | PTR malloc PARAMS ((size_t)); | |
84 | PTR realloc PARAMS ((PTR, size_t)); | |
85 | PTR calloc PARAMS ((size_t, size_t)); | |
86 | PTR sbrk PARAMS ((ptrdiff_t)); | |
87 | #endif | |
88 | ||
89 | /* The program name if set. */ | |
90 | static const char *name = ""; | |
91 | ||
c1687039 | 92 | #ifdef HAVE_SBRK |
252b5132 RH |
93 | /* The initial sbrk, set when the program name is set. Not used for win32 |
94 | ports other than cygwin32. */ | |
95 | static char *first_break = NULL; | |
c1687039 | 96 | #endif /* HAVE_SBRK */ |
252b5132 RH |
97 | |
98 | void | |
99 | xmalloc_set_program_name (s) | |
100 | const char *s; | |
101 | { | |
102 | name = s; | |
c1687039 ILT |
103 | #ifdef HAVE_SBRK |
104 | /* Win32 ports other than cygwin32 don't have brk() */ | |
252b5132 RH |
105 | if (first_break == NULL) |
106 | first_break = (char *) sbrk (0); | |
c1687039 | 107 | #endif /* HAVE_SBRK */ |
252b5132 RH |
108 | } |
109 | ||
2ea7befd DD |
110 | void |
111 | xmalloc_failed (size) | |
112 | size_t size; | |
113 | { | |
114 | #ifdef HAVE_SBRK | |
115 | extern char **environ; | |
116 | size_t allocated; | |
117 | ||
118 | if (first_break != NULL) | |
119 | allocated = (char *) sbrk (0) - first_break; | |
120 | else | |
121 | allocated = (char *) sbrk (0) - (char *) &environ; | |
122 | fprintf (stderr, | |
123 | "\n%s%sCannot allocate %lu bytes after allocating %lu bytes\n", | |
124 | name, *name ? ": " : "", | |
125 | (unsigned long) size, (unsigned long) allocated); | |
126 | #else /* HAVE_SBRK */ | |
127 | fprintf (stderr, | |
128 | "\n%s%sCannot allocate %lu bytes\n", | |
129 | name, *name ? ": " : "", | |
130 | (unsigned long) size); | |
131 | #endif /* HAVE_SBRK */ | |
132 | xexit (1); | |
133 | } | |
134 | ||
252b5132 RH |
135 | PTR |
136 | xmalloc (size) | |
137 | size_t size; | |
138 | { | |
139 | PTR newmem; | |
140 | ||
141 | if (size == 0) | |
142 | size = 1; | |
143 | newmem = malloc (size); | |
144 | if (!newmem) | |
2ea7befd DD |
145 | xmalloc_failed (size); |
146 | ||
252b5132 RH |
147 | return (newmem); |
148 | } | |
149 | ||
150 | PTR | |
151 | xcalloc (nelem, elsize) | |
152 | size_t nelem, elsize; | |
153 | { | |
154 | PTR newmem; | |
155 | ||
156 | if (nelem == 0 || elsize == 0) | |
157 | nelem = elsize = 1; | |
158 | ||
159 | newmem = calloc (nelem, elsize); | |
160 | if (!newmem) | |
2ea7befd DD |
161 | xmalloc_failed (nelem * elsize); |
162 | ||
252b5132 RH |
163 | return (newmem); |
164 | } | |
165 | ||
166 | PTR | |
167 | xrealloc (oldmem, size) | |
168 | PTR oldmem; | |
169 | size_t size; | |
170 | { | |
171 | PTR newmem; | |
172 | ||
173 | if (size == 0) | |
174 | size = 1; | |
175 | if (!oldmem) | |
176 | newmem = malloc (size); | |
177 | else | |
178 | newmem = realloc (oldmem, size); | |
179 | if (!newmem) | |
2ea7befd DD |
180 | xmalloc_failed (size); |
181 | ||
252b5132 RH |
182 | return (newmem); |
183 | } |