xmalloc.c
上传用户:aoeyumen
上传日期:2007-01-06
资源大小:3329k
文件大小:3k
源码类别:

DVD

开发平台:

Unix_Linux

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