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 (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 /* Return a pointer to free()able block of memory large enough
  64    to hold BYTES number of bytes.  If the memory cannot be allocated,
  65    print an error message and abort. */
  66 char *
  67 xmalloc (bytes)
  68      size_t bytes;
  69 {
  70   char *temp;
  71
  72   temp = (char *)malloc (bytes);
  73
  74   if (temp == 0)
  75     {
  76       if (brkfound == 0)
  77         {
  78           lbreak = (PTR_T)sbrk (0);
  79           brkfound++;
  80         }
  81       allocated = (char *)sbrk (0) - (char *)lbreak;
  82       fatal_error ("xmalloc: cannot allocate %lu bytes (%lu bytes allocated)", (unsigned long)bytes, (unsigned long)allocated);
  83     }
  84
  85   return (temp);
  86 }
  87
  88 char *
  89 xrealloc (pointer, bytes)
  90      PTR_T pointer;
  91      size_t bytes;
  92 {
  93   char *temp;
  94
  95   temp = pointer ? (char *)realloc (pointer, bytes) : (char *)malloc (bytes);
  96
  97   if (temp == 0)
  98     {
  99       if (brkfound == 0)
 100         {
 101           lbreak = (PTR_T)sbrk (0);
 102           brkfound++;
 103         }
 104       allocated = (char *)sbrk (0) - (char *)lbreak;
 105       fatal_error ("xrealloc: cannot reallocate %lu bytes (%lu bytes allocated)", (unsigned long)bytes, (unsigned long)allocated);
 106     }
 107
 108   return (temp);
 109 }
 110
 111 /* Use this as the function to call when adding unwind protects so we
 112    don't need to know what free() returns. */
 113 void
 114 xfree (string)
 115      char *string;
 116 {
 117   if (string)
 118     free (string);
 119 }