1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/* ialloc.h -- malloc with idx_t rather than size_t
Copyright 2021-2023 Free Software Foundation, Inc.
This file is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as
published by the Free Software Foundation; either version 2.1 of the
License, or (at your option) any later version.
This file is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>. */
#ifndef IALLOC_H_
#define IALLOC_H_
/* This file uses _GL_INLINE_HEADER_BEGIN, _GL_INLINE, _GL_ATTRIBUTE_COLD,
_GL_ATTRIBUTE_MALLOC. */
#if !_GL_CONFIG_H_INCLUDED
#error "Please include config.h first."
#endif
#include "idx.h"
#include <errno.h>
#include <stdint.h>
#include <stdlib.h>
_GL_INLINE_HEADER_BEGIN
#ifndef IALLOC_INLINE
# define IALLOC_INLINE _GL_INLINE
#endif
#ifdef __cplusplus
extern "C" {
#endif
IALLOC_INLINE void * _GL_ATTRIBUTE_COLD
_gl_alloc_nomem (void)
{
errno = ENOMEM;
return NULL;
}
/* imalloc (size) is like malloc (size).
It returns a non-NULL pointer to size bytes of memory.
Upon failure, it returns NULL with errno set. */
IALLOC_INLINE
_GL_ATTRIBUTE_MALLOC /*_GL_ATTRIBUTE_DEALLOC_FREE*/
void *
imalloc (idx_t s)
{
return s <= SIZE_MAX ? malloc (s) : _gl_alloc_nomem ();
}
/* irealloc (ptr, size) is like realloc (ptr, size).
It returns a non-NULL pointer to size bytes of memory.
Upon failure, it returns NULL with errno set. */
IALLOC_INLINE
/*_GL_ATTRIBUTE_DEALLOC_FREE*/
void *
irealloc (void *p, idx_t s)
{
/* Work around GNU realloc glitch by treating a zero size as if it
were 1, so that returning NULL is equivalent to failing. */
return s <= SIZE_MAX ? realloc (p, s | !s) : _gl_alloc_nomem ();
}
/* icalloc (num, size) is like calloc (num, size).
It returns a non-NULL pointer to num * size bytes of memory.
Upon failure, it returns NULL with errno set. */
IALLOC_INLINE
_GL_ATTRIBUTE_MALLOC /*_GL_ATTRIBUTE_DEALLOC_FREE*/
void *
icalloc (idx_t n, idx_t s)
{
if (SIZE_MAX < n)
{
if (s != 0)
return _gl_alloc_nomem ();
n = 0;
}
if (SIZE_MAX < s)
{
if (n != 0)
return _gl_alloc_nomem ();
s = 0;
}
return calloc (n, s);
}
/* ireallocarray (ptr, num, size) is like reallocarray (ptr, num, size).
It returns a non-NULL pointer to num * size bytes of memory.
Upon failure, it returns NULL with errno set. */
IALLOC_INLINE void *
ireallocarray (void *p, idx_t n, idx_t s)
{
/* Work around GNU reallocarray glitch by treating a zero size as if
it were 1, so that returning NULL is equivalent to failing. */
if (n == 0 || s == 0)
n = s = 1;
return (n <= SIZE_MAX && s <= SIZE_MAX
? reallocarray (p, n, s)
: _gl_alloc_nomem ());
}
#ifdef __cplusplus
}
#endif
_GL_INLINE_HEADER_END
#endif
|
