xmalloc.c revision 33965 
1/* memory allocation routines with error checking.
2   Copyright 1989, 90, 91, 92, 93, 94 Free Software Foundation, Inc.
3
4This file is part of the libiberty library.
5Libiberty is free software; you can redistribute it and/or
6modify it under the terms of the GNU Library General Public
7License as published by the Free Software Foundation; either
8version 2 of the License, or (at your option) any later version.
9
10Libiberty is distributed in the hope that it will be useful,
11but WITHOUT ANY WARRANTY; without even the implied warranty of
12MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13Library General Public License for more details.
14
15You should have received a copy of the GNU Library General Public
16License along with libiberty; see the file COPYING.LIB.  If
17not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18Boston, MA 02111-1307, USA.  */
19
20#include "ansidecl.h"
21#include "libiberty.h"
22
23#include <stdio.h>
24
25#ifdef __STDC__
26#include <stddef.h>
27#else
28#define size_t unsigned long
29#define ptrdiff_t long
30#endif
31
32#if VMS
33#include <stdlib.h>
34#include <unixlib.h>
35#else
36/* For systems with larger pointers than ints, these must be declared.  */
37PTR malloc PARAMS ((size_t));
38PTR realloc PARAMS ((PTR, size_t));
39PTR sbrk PARAMS ((ptrdiff_t));
40#endif
41
42/* The program name if set.  */
43static const char *name = "";
44
45/* The initial sbrk, set when the program name is set.  */
46static char *first_break = NULL;
47
48void
49xmalloc_set_program_name (s)
50     const char *s;
51{
52  name = s;
53  if (first_break == NULL)
54    first_break = (char *) sbrk (0);
55}
56
57PTR
58xmalloc (size)
59    size_t size;
60{
61  PTR newmem;
62
63  if (size == 0)
64    size = 1;
65  newmem = malloc (size);
66  if (!newmem)
67    {
68      extern char **environ;
69      size_t allocated;
70
71      if (first_break != NULL)
72	allocated = (char *) sbrk (0) - first_break;
73      else
74	allocated = (char *) sbrk (0) - (char *) &environ;
75      fprintf (stderr,
76	       "\n%s%sCan not allocate %lu bytes after allocating %lu bytes\n",
77	       name, *name ? ": " : "",
78	       (unsigned long) size, (unsigned long) allocated);
79      xexit (1);
80    }
81  return (newmem);
82}
83
84PTR
85xrealloc (oldmem, size)
86    PTR oldmem;
87    size_t size;
88{
89  PTR newmem;
90
91  if (size == 0)
92    size = 1;
93  if (!oldmem)
94    newmem = malloc (size);
95  else
96    newmem = realloc (oldmem, size);
97  if (!newmem)
98    {
99      extern char **environ;
100      size_t allocated;
101
102      if (first_break != NULL)
103	allocated = (char *) sbrk (0) - first_break;
104      else
105	allocated = (char *) sbrk (0) - (char *) &environ;
106      fprintf (stderr,
107	       "\n%s%sCan not reallocate %lu bytes after allocating %lu bytes\n",
108	       name, *name ? ": " : "",
109	       (unsigned long) size, (unsigned long) allocated);
110      xexit (1);
111    }
112  return (newmem);
113}
114