Home | History | Annotate | Download | only in lib
      1 /* xmalloc.c -- malloc with out of memory checking
      2 
      3    Copyright (C) 1990-2000, 2002-2006, 2008-2012 Free Software Foundation, Inc.
      4 
      5    This program is free software: you can redistribute it and/or modify
      6    it under the terms of the GNU General Public License as published by
      7    the Free Software Foundation; either version 3 of the License, or
      8    (at your option) any later version.
      9 
     10    This program is distributed in the hope that it will be useful,
     11    but WITHOUT ANY WARRANTY; without even the implied warranty of
     12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     13    GNU General Public License for more details.
     14 
     15    You should have received a copy of the GNU General Public License
     16    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
     17 
     18 #include <config.h>
     19 
     20 #define XALLOC_INLINE _GL_EXTERN_INLINE
     21 
     22 #include "xalloc.h"
     23 
     24 #include <stdlib.h>
     25 #include <string.h>
     26 
     27 /* 1 if calloc is known to be compatible with GNU calloc.  This
     28    matters if we are not also using the calloc module, which defines
     29    HAVE_CALLOC_GNU and supports the GNU API even on non-GNU platforms.  */
     30 #if defined HAVE_CALLOC_GNU || (defined __GLIBC__ && !defined __UCLIBC__)
     31 enum { HAVE_GNU_CALLOC = 1 };
     32 #else
     33 enum { HAVE_GNU_CALLOC = 0 };
     34 #endif
     35 
     36 /* Allocate N bytes of memory dynamically, with error checking.  */
     37 
     38 void *
     39 xmalloc (size_t n)
     40 {
     41   void *p = malloc (n);
     42   if (!p && n != 0)
     43     xalloc_die ();
     44   return p;
     45 }
     46 
     47 /* Change the size of an allocated block of memory P to N bytes,
     48    with error checking.  */
     49 
     50 void *
     51 xrealloc (void *p, size_t n)
     52 {
     53   if (!n && p)
     54     {
     55       /* The GNU and C99 realloc behaviors disagree here.  Act like
     56          GNU, even if the underlying realloc is C99.  */
     57       free (p);
     58       return NULL;
     59     }
     60 
     61   p = realloc (p, n);
     62   if (!p && n)
     63     xalloc_die ();
     64   return p;
     65 }
     66 
     67 /* If P is null, allocate a block of at least *PN bytes; otherwise,
     68    reallocate P so that it contains more than *PN bytes.  *PN must be
     69    nonzero unless P is null.  Set *PN to the new block's size, and
     70    return the pointer to the new block.  *PN is never set to zero, and
     71    the returned pointer is never null.  */
     72 
     73 void *
     74 x2realloc (void *p, size_t *pn)
     75 {
     76   return x2nrealloc (p, pn, 1);
     77 }
     78 
     79 /* Allocate S bytes of zeroed memory dynamically, with error checking.
     80    There's no need for xnzalloc (N, S), since it would be equivalent
     81    to xcalloc (N, S).  */
     82 
     83 void *
     84 xzalloc (size_t s)
     85 {
     86   return memset (xmalloc (s), 0, s);
     87 }
     88 
     89 /* Allocate zeroed memory for N elements of S bytes, with error
     90    checking.  S must be nonzero.  */
     91 
     92 void *
     93 xcalloc (size_t n, size_t s)
     94 {
     95   void *p;
     96   /* Test for overflow, since some calloc implementations don't have
     97      proper overflow checks.  But omit overflow and size-zero tests if
     98      HAVE_GNU_CALLOC, since GNU calloc catches overflow and never
     99      returns NULL if successful.  */
    100   if ((! HAVE_GNU_CALLOC && xalloc_oversized (n, s))
    101       || (! (p = calloc (n, s)) && (HAVE_GNU_CALLOC || n != 0)))
    102     xalloc_die ();
    103   return p;
    104 }
    105 
    106 /* Clone an object P of size S, with error checking.  There's no need
    107    for xnmemdup (P, N, S), since xmemdup (P, N * S) works without any
    108    need for an arithmetic overflow check.  */
    109 
    110 void *
    111 xmemdup (void const *p, size_t s)
    112 {
    113   return memcpy (xmalloc (s), p, s);
    114 }
    115 
    116 /* Clone STRING.  */
    117 
    118 char *
    119 xstrdup (char const *string)
    120 {
    121   return xmemdup (string, strlen (string) + 1);
    122 }
    123