lib/ialloc.h

   1 /* ialloc.h -- malloc with idx_t rather than size_t
   2
   3    Copyright 2021 Free Software Foundation, Inc.
   4
   5    This file is free software: you can redistribute it and/or modify
   6    it under the terms of the GNU Lesser General Public License as
   7    published by the Free Software Foundation; either version 3 of the
   8    License, or (at your option) any later version.
   9
  10    This file 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 Lesser General Public License for more details.
  14
  15    You should have received a copy of the GNU Lesser General Public License
  16    along with this program.  If not, see <https://www.gnu.org/licenses/>.  */
  17
  18 #ifndef IALLOC_H_
  19 #define IALLOC_H_
  20
  21 #include "idx.h"
  22
  23 #include <errno.h>
  24 #include <stdint.h>
  25 #include <stdlib.h>
  26
  27 #ifndef _GL_INLINE_HEADER_BEGIN
  28  #error "Please include config.h first."
  29 #endif
  30 _GL_INLINE_HEADER_BEGIN
  31 #ifndef IALLOC_INLINE
  32 # define IALLOC_INLINE _GL_INLINE
  33 #endif
  34
  35 #ifdef __cplusplus
  36 extern "C" {
  37 #endif
  38
  39 IALLOC_INLINE void * _GL_ATTRIBUTE_COLD
  40 _gl_alloc_nomem (void)
  41 {
  42   errno = ENOMEM;
  43   return NULL;
  44 }
  45
  46 IALLOC_INLINE void *
  47 imalloc (idx_t s)
  48 {
  49   return s <= SIZE_MAX ? malloc (s) : _gl_alloc_nomem ();
  50 }
  51
  52 IALLOC_INLINE void *
  53 irealloc (void *p, idx_t s)
  54 {
  55   /* Work around GNU realloc glitch by treating a zero size as if it
  56      were 1, so that returning NULL is equivalent to failing.  */
  57   return s <= SIZE_MAX ? realloc (p, s | !s) : _gl_alloc_nomem ();
  58 }
  59
  60 IALLOC_INLINE void *
  61 icalloc (idx_t n, idx_t s)
  62 {
  63   if (SIZE_MAX < n)
  64     {
  65       if (s != 0)
  66         return _gl_alloc_nomem ();
  67       n = 0;
  68     }
  69   if (SIZE_MAX < s)
  70     {
  71       if (n != 0)
  72         return _gl_alloc_nomem ();
  73       s = 0;
  74     }
  75   return calloc (n, s);
  76 }
  77
  78 IALLOC_INLINE void *
  79 ireallocarray (void *p, idx_t n, idx_t s)
  80 {
  81   /* Work around GNU reallocarray glitch by treating a zero size as if
  82      it were 1, so that returning NULL is equivalent to failing.  */
  83   if (n == 0 || s == 0)
  84     n = s = 1;
  85   return (n <= SIZE_MAX && s <= SIZE_MAX
  86           ? reallocarray (p, n, s)
  87           : _gl_alloc_nomem ());
  88 }
  89
  90 #ifdef __cplusplus
  91 }
  92 #endif
  93
  94 #endif