xref: /plan9/sys/src/ape/cmd/diff/xmalloc.c (revision 0b459c2cb92b7c9d88818e9a2f72e678e5bc4553)
1*0b459c2cSDavid du Colombier /* xmalloc.c -- malloc with out of memory checking
2*0b459c2cSDavid du Colombier    Copyright (C) 1990, 1991, 1993 Free Software Foundation, Inc.
3*0b459c2cSDavid du Colombier 
4*0b459c2cSDavid du Colombier    This program is free software; you can redistribute it and/or modify
5*0b459c2cSDavid du Colombier    it under the terms of the GNU General Public License as published by
6*0b459c2cSDavid du Colombier    the Free Software Foundation; either version 2, or (at your option)
7*0b459c2cSDavid du Colombier    any later version.
8*0b459c2cSDavid du Colombier 
9*0b459c2cSDavid du Colombier    This program is distributed in the hope that it will be useful,
10*0b459c2cSDavid du Colombier    but WITHOUT ANY WARRANTY; without even the implied warranty of
11*0b459c2cSDavid du Colombier    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12*0b459c2cSDavid du Colombier    GNU General Public License for more details.
13*0b459c2cSDavid du Colombier 
14*0b459c2cSDavid du Colombier    You should have received a copy of the GNU General Public License
15*0b459c2cSDavid du Colombier    along with this program; if not, write to the Free Software
16*0b459c2cSDavid du Colombier    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
17*0b459c2cSDavid du Colombier 
18*0b459c2cSDavid du Colombier #ifdef HAVE_CONFIG_H
19*0b459c2cSDavid du Colombier #include <config.h>
20*0b459c2cSDavid du Colombier #endif
21*0b459c2cSDavid du Colombier 
22*0b459c2cSDavid du Colombier #if __STDC__
23*0b459c2cSDavid du Colombier #define VOID void
24*0b459c2cSDavid du Colombier #else
25*0b459c2cSDavid du Colombier #define VOID char
26*0b459c2cSDavid du Colombier #endif
27*0b459c2cSDavid du Colombier 
28*0b459c2cSDavid du Colombier #include <sys/types.h>
29*0b459c2cSDavid du Colombier 
30*0b459c2cSDavid du Colombier #if STDC_HEADERS
31*0b459c2cSDavid du Colombier #include <stdlib.h>
32*0b459c2cSDavid du Colombier #else
33*0b459c2cSDavid du Colombier VOID *malloc ();
34*0b459c2cSDavid du Colombier VOID *realloc ();
35*0b459c2cSDavid du Colombier void free ();
36*0b459c2cSDavid du Colombier #endif
37*0b459c2cSDavid du Colombier 
38*0b459c2cSDavid du Colombier #if __STDC__ && defined (HAVE_VPRINTF)
39*0b459c2cSDavid du Colombier void error (int, int, char const *, ...);
40*0b459c2cSDavid du Colombier #else
41*0b459c2cSDavid du Colombier void error ();
42*0b459c2cSDavid du Colombier #endif
43*0b459c2cSDavid du Colombier 
44*0b459c2cSDavid du Colombier /* Allocate N bytes of memory dynamically, with error checking.  */
45*0b459c2cSDavid du Colombier 
46*0b459c2cSDavid du Colombier VOID *
xmalloc(n)47*0b459c2cSDavid du Colombier xmalloc (n)
48*0b459c2cSDavid du Colombier      size_t n;
49*0b459c2cSDavid du Colombier {
50*0b459c2cSDavid du Colombier   VOID *p;
51*0b459c2cSDavid du Colombier 
52*0b459c2cSDavid du Colombier   p = malloc (n);
53*0b459c2cSDavid du Colombier   if (p == 0)
54*0b459c2cSDavid du Colombier     /* Must exit with 2 for `cmp'.  */
55*0b459c2cSDavid du Colombier     error (2, 0, "memory exhausted");
56*0b459c2cSDavid du Colombier   return p;
57*0b459c2cSDavid du Colombier }
58*0b459c2cSDavid du Colombier 
59*0b459c2cSDavid du Colombier /* Change the size of an allocated block of memory P to N bytes,
60*0b459c2cSDavid du Colombier    with error checking.
61*0b459c2cSDavid du Colombier    If P is NULL, run xmalloc.
62*0b459c2cSDavid du Colombier    If N is 0, run free and return NULL.  */
63*0b459c2cSDavid du Colombier 
64*0b459c2cSDavid du Colombier VOID *
xrealloc(p,n)65*0b459c2cSDavid du Colombier xrealloc (p, n)
66*0b459c2cSDavid du Colombier      VOID *p;
67*0b459c2cSDavid du Colombier      size_t n;
68*0b459c2cSDavid du Colombier {
69*0b459c2cSDavid du Colombier   if (p == 0)
70*0b459c2cSDavid du Colombier     return xmalloc (n);
71*0b459c2cSDavid du Colombier   if (n == 0)
72*0b459c2cSDavid du Colombier     {
73*0b459c2cSDavid du Colombier       free (p);
74*0b459c2cSDavid du Colombier       return 0;
75*0b459c2cSDavid du Colombier     }
76*0b459c2cSDavid du Colombier   p = realloc (p, n);
77*0b459c2cSDavid du Colombier   if (p == 0)
78*0b459c2cSDavid du Colombier     /* Must exit with 2 for `cmp'.  */
79*0b459c2cSDavid du Colombier     error (2, 0, "memory exhausted");
80*0b459c2cSDavid du Colombier   return p;
81*0b459c2cSDavid du Colombier }
82