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 1, 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, 675 Mass Ave, Cambridge, MA 02139, 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 #if !defined (PTR_T)
  42 #  if defined (__STDC__)
  43 #    define PTR_T void *
  44 #  else
  45 #    define PTR_T char *
  46 #  endif /* !__STDC__ */
  47 #endif /* !PTR_T */
  48
  49 #if defined (HAVE_SBRK) && !defined (SBRK_DECLARED)
  50 extern char *sbrk();
  51 #endif
  52
  53 static PTR_T lbreak;
  54 static int brkfound;
  55 static size_t allocated;
  56
  57 /* **************************************************************** */
  58 /*                                                                  */
  59 /*                 Memory Allocation and Deallocation.              */
  60 /*                                                                  */
  61 /* **************************************************************** */
  62
  63 #if defined (HAVE_SBRK)
  64 static size_t
  65 findbrk ()
  66 {
  67   if (brkfound == 0)
  68     {
  69       lbreak = (PTR_T)sbrk (0);
  70       brkfound++;
  71     }
  72   return (char *)sbrk (0) - (char *)lbreak;
  73 }
  74 #endif
  75
  76 /* Return a pointer to free()able block of memory large enough
  77    to hold BYTES number of bytes.  If the memory cannot be allocated,
  78    print an error message and abort. */
  79 char *
  80 xmalloc (bytes)
  81      size_t bytes;
  82 {
  83   char *temp;
  84
  85   temp = (char *)malloc (bytes);
  86
  87   if (temp == 0)
  88     {
  89 #if defined (HAVE_SBRK)
  90       allocated = findbrk ();
  91       fatal_error ("xmalloc: cannot allocate %lu bytes (%lu bytes allocated)", (unsigned long)bytes, (unsigned long)allocated);
  92 #else
  93       fatal_error ("xmalloc: cannot allocate %lu bytes", (unsigned long)bytes);
  94 #endif /* !HAVE_SBRK */
  95     }
  96
  97   return (temp);
  98 }
  99
 100 char *
 101 xrealloc (pointer, bytes)
 102      PTR_T pointer;
 103      size_t bytes;
 104 {
 105   char *temp;
 106
 107   temp = pointer ? (char *)realloc (pointer, bytes) : (char *)malloc (bytes);
 108
 109   if (temp == 0)
 110     {
 111 #if defined (HAVE_SBRK)
 112       allocated = findbrk ();
 113       fatal_error ("xrealloc: cannot reallocate %lu bytes (%lu bytes allocated)", (unsigned long)bytes, (unsigned long)allocated);
 114 #else
 115       fatal_error ("xmalloc: cannot allocate %lu bytes", (unsigned long)bytes);
 116 #endif /* !HAVE_SBRK */
 117     }
 118
 119   return (temp);
 120 }
 121
 122 /* Use this as the function to call when adding unwind protects so we
 123    don't need to know what free() returns. */
 124 void
 125 xfree (string)
 126      char *string;
 127 {
 128   if (string)
 129     free (string);
 130 }