xmalloc.c

   1 /* xmalloc.c -- safe versions of malloc and realloc */
   2
   3 /* Copyright (C) 1991 Free Software Foundation, Inc.
   4
   5    This file is part of GNU Readline, a library for reading lines
   6    of text with interactive input and history editing.
   7
   8    Readline is free software; you can redistribute it and/or modify it
   9    under the terms of the GNU General Public License as published by the
  10    Free Software Foundation; either version 2, or (at your option) any
  11    later version.
  12
  13    Readline is distributed in the hope that it will be useful, but
  14    WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16    General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with Readline; see the file COPYING.  If not, write to the Free
  20    Software Foundation, 59 Temple Place, Suite 330, Boston, MA 02111 USA. */
  21
  22 #if defined (HAVE_CONFIG_H)
  23 #include <config.h>
  24 #endif
  25
  26 #include "bashtypes.h"
  27 #include <stdio.h>
  28
  29 #if defined (HAVE_UNISTD_H)
  30 #  include <unistd.h>
  31 #endif
  32
  33 #if defined (HAVE_STDLIB_H)
  34 #  include <stdlib.h>
  35 #else
  36 #  include "ansi_stdlib.h"
  37 #endif /* HAVE_STDLIB_H */
  38
  39 #include "error.h"
  40
  41 #include "bashintl.h"
  42
  43 #if !defined (PTR_T)
  44 #  if defined (__STDC__)
  45 #    define PTR_T void *
  46 #  else
  47 #    define PTR_T char *
  48 #  endif /* !__STDC__ */
  49 #endif /* !PTR_T */
  50
  51 #if defined (HAVE_SBRK) && !HAVE_DECL_SBRK
  52 extern char *sbrk();
  53 #endif
  54
  55 static PTR_T lbreak;
  56 static int brkfound;
  57 static size_t allocated;
  58
  59 /* **************************************************************** */
  60 /*                                                                  */
  61 /*                 Memory Allocation and Deallocation.              */
  62 /*                                                                  */
  63 /* **************************************************************** */
  64
  65 #if defined (HAVE_SBRK)
  66 static size_t
  67 findbrk ()
  68 {
  69   if (brkfound == 0)
  70     {
  71       lbreak = (PTR_T)sbrk (0);
  72       brkfound++;
  73     }
  74   return (char *)sbrk (0) - (char *)lbreak;
  75 }
  76 #endif
  77
  78 /* Return a pointer to free()able block of memory large enough
  79    to hold BYTES number of bytes.  If the memory cannot be allocated,
  80    print an error message and abort. */
  81 PTR_T
  82 xmalloc (bytes)
  83      size_t bytes;
  84 {
  85   PTR_T temp;
  86
  87   temp = malloc (bytes);
  88
  89   if (temp == 0)
  90     {
  91 #if defined (HAVE_SBRK)
  92       allocated = findbrk ();
  93       fatal_error (_("xmalloc: cannot allocate %lu bytes (%lu bytes allocated)"), (unsigned long)bytes, (unsigned long)allocated);
  94 #else
  95       fatal_error (_("xmalloc: cannot allocate %lu bytes"), (unsigned long)bytes);
  96 #endif /* !HAVE_SBRK */
  97     }
  98
  99   return (temp);
 100 }
 101
 102 PTR_T
 103 xrealloc (pointer, bytes)
 104      PTR_T pointer;
 105      size_t bytes;
 106 {
 107   PTR_T temp;
 108
 109   temp = pointer ? realloc (pointer, bytes) : malloc (bytes);
 110
 111   if (temp == 0)
 112     {
 113 #if defined (HAVE_SBRK)
 114       allocated = findbrk ();
 115       fatal_error (_("xrealloc: cannot reallocate %lu bytes (%lu bytes allocated)"), (unsigned long)bytes, (unsigned long)allocated);
 116 #else
 117       fatal_error (_("xrealloc: cannot allocate %lu bytes"), (unsigned long)bytes);
 118 #endif /* !HAVE_SBRK */
 119     }
 120
 121   return (temp);
 122 }
 123
 124 /* Use this as the function to call when adding unwind protects so we
 125    don't need to know what free() returns. */
 126 void
 127 xfree (string)
 128      PTR_T string;
 129 {
 130   if (string)
 131     free (string);
 132 }
 133
 134 #ifdef USING_BASH_MALLOC
 135 #include <malloc/shmalloc.h>
 136
 137 PTR_T
 138 sh_xmalloc (bytes, file, line)
 139      size_t bytes;
 140      char *file;
 141      int line;
 142 {
 143   PTR_T temp;
 144
 145   temp = sh_malloc (bytes, file, line);
 146
 147   if (temp == 0)
 148     {
 149 #if defined (HAVE_SBRK)
 150       allocated = findbrk ();
 151       fatal_error (_("xmalloc: %s:%d: cannot allocate %lu bytes (%lu bytes allocated)"), file, line, (unsigned long)bytes, (unsigned long)allocated);
 152 #else
 153       fatal_error (_("xmalloc: %s:%d: cannot allocate %lu bytes"), file, line, (unsigned long)bytes);
 154 #endif /* !HAVE_SBRK */
 155     }
 156
 157   return (temp);
 158 }
 159
 160 PTR_T
 161 sh_xrealloc (pointer, bytes, file, line)
 162      PTR_T pointer;
 163      size_t bytes;
 164      char *file;
 165      int line;
 166 {
 167   PTR_T temp;
 168
 169   temp = pointer ? sh_realloc (pointer, bytes, file, line) : sh_malloc (bytes, file, line);
 170
 171   if (temp == 0)
 172     {
 173 #if defined (HAVE_SBRK)
 174       allocated = findbrk ();
 175       fatal_error (_("xrealloc: %s:%d: cannot reallocate %lu bytes (%lu bytes allocated)"), file, line, (unsigned long)bytes, (unsigned long)allocated);
 176 #else
 177       fatal_error (_("xrealloc: %s:%d: cannot allocate %lu bytes"), file, line, (unsigned long)bytes);
 178 #endif /* !HAVE_SBRK */
 179     }
 180
 181   return (temp);
 182 }
 183
 184 void
 185 sh_xfree (string, file, line)
 186      PTR_T string;
 187      char *file;
 188      int line;
 189 {
 190   if (string)
 191     sh_free (string, file, line);
 192 }
 193 #endif