update from main archive 961008
[kopensolaris-gnu/glibc.git] / locale / programs / xmalloc.c
1 /* xmalloc.c -- malloc with out of memory checking
2    Copyright (C) 1990, 91, 92, 93, 94, 95, 96 Free Software Foundation, Inc.
3
4    This program is free software; you can redistribute it and/or modify
5    it under the terms of the GNU General Public License as published by
6    the Free Software Foundation; either version 2, or (at your option)
7    any later version.
8
9    This program is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12    GNU General Public License for more details.
13
14    You should have received a copy of the GNU General Public License
15    along with this program; if not, write to the Free Software
16    Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */
17
18 #ifdef HAVE_CONFIG_H
19 #include <config.h>
20 #endif
21
22 #if __STDC__
23 #define VOID void
24 #else
25 #define VOID char
26 #endif
27
28 #include <sys/types.h>
29
30 #if STDC_HEADERS || _LIBC
31 #include <stdlib.h>
32 static VOID *fixup_null_alloc __P ((size_t n));
33 #else
34 VOID *calloc ();
35 VOID *malloc ();
36 VOID *realloc ();
37 void free ();
38 #endif
39
40 #include <libintl.h>
41 #include "error.h"
42
43 #ifndef _
44 # define _(str) gettext (str)
45 #endif
46
47 #ifndef EXIT_FAILURE
48 #define EXIT_FAILURE 4
49 #endif
50
51 /* Exit value when the requested amount of memory is not available.
52    The caller may set it to some other value.  */
53 int xmalloc_exit_failure = EXIT_FAILURE;
54
55 static VOID *
56 fixup_null_alloc (n)
57      size_t n;
58 {
59   VOID *p;
60
61   p = 0;
62   if (n == 0)
63     p = malloc ((size_t) 1);
64   if (p == 0)
65     error (xmalloc_exit_failure, 0, _("memory exhausted"));
66   return p;
67 }
68
69 /* Allocate N bytes of memory dynamically, with error checking.  */
70
71 VOID *
72 xmalloc (n)
73      size_t n;
74 {
75   VOID *p;
76
77   p = malloc (n);
78   if (p == 0)
79     p = fixup_null_alloc (n);
80   return p;
81 }
82
83 /* Allocate memory for N elements of S bytes, with error checking.  */
84
85 VOID *
86 xcalloc (n, s)
87      size_t n, s;
88 {
89   VOID *p;
90
91   p = calloc (n, s);
92   if (p == 0)
93     p = fixup_null_alloc (n);
94   return p;
95 }
96
97 /* Change the size of an allocated block of memory P to N bytes,
98    with error checking.
99    If P is NULL, run xmalloc.  */
100
101 VOID *
102 xrealloc (p, n)
103      VOID *p;
104      size_t n;
105 {
106   if (p == 0)
107     return xmalloc (n);
108   p = realloc (p, n);
109   if (p == 0)
110     p = fixup_null_alloc (n);
111   return p;
112 }