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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
|
/*
Unix SMB/CIFS implementation.
Samba utility functions
Copyright (C) Andrew Tridgell 1992-1998
Copyright (C) Jeremy Allison 2001-2007
Copyright (C) Simo Sorce 2001
Copyright (C) Jim McDonough <jmcd@us.ibm.com> 2003
Copyright (C) James Peach 2006
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 3 of the License, or
(at your option) any later version.
This program 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "includes.h"
/* Max allowable allococation - 256mb - 0x10000000 */
#define MAX_ALLOC_SIZE (1024*1024*256)
#if defined(PARANOID_MALLOC_CHECKER)
/****************************************************************************
Internal malloc wrapper. Externally visible.
****************************************************************************/
void *malloc_(size_t size)
{
if (size == 0) {
return NULL;
}
#undef malloc
return malloc(size);
#define malloc(s) __ERROR_DONT_USE_MALLOC_DIRECTLY
}
/****************************************************************************
Internal calloc wrapper. Not externally visible.
****************************************************************************/
static void *calloc_(size_t count, size_t size)
{
if (size == 0 || count == 0) {
return NULL;
}
#undef calloc
return calloc(count, size);
#define calloc(n,s) __ERROR_DONT_USE_CALLOC_DIRECTLY
}
/****************************************************************************
Internal realloc wrapper. Not externally visible.
****************************************************************************/
static void *realloc_(void *ptr, size_t size)
{
#undef realloc
return realloc(ptr, size);
#define realloc(p,s) __ERROR_DONT_USE_RELLOC_DIRECTLY
}
#endif /* PARANOID_MALLOC_CHECKER */
/****************************************************************************
Type-safe memalign
****************************************************************************/
void *memalign_array(size_t el_size, size_t align, unsigned int count)
{
if (count >= MAX_ALLOC_SIZE/el_size) {
return NULL;
}
return sys_memalign(align, el_size*count);
}
/****************************************************************************
Type-safe calloc.
****************************************************************************/
void *calloc_array(size_t size, size_t nmemb)
{
if (nmemb >= MAX_ALLOC_SIZE/size) {
return NULL;
}
if (size == 0 || nmemb == 0) {
return NULL;
}
#if defined(PARANOID_MALLOC_CHECKER)
return calloc_(nmemb, size);
#else
return calloc(nmemb, size);
#endif
}
/****************************************************************************
Expand a pointer to be a particular size.
Note that this version of Realloc has an extra parameter that decides
whether to free the passed in storage on allocation failure or if the
new size is zero.
This is designed for use in the typical idiom of :
p = SMB_REALLOC(p, size)
if (!p) {
return error;
}
and not to have to keep track of the old 'p' contents to free later, nor
to worry if the size parameter was zero. In the case where NULL is returned
we guarentee that p has been freed.
If free later semantics are desired, then pass 'free_old_on_error' as False which
guarentees that the old contents are not freed on error, even if size == 0. To use
this idiom use :
tmp = SMB_REALLOC_KEEP_OLD_ON_ERROR(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
Changes were instigated by Coverity error checking. JRA.
****************************************************************************/
void *Realloc(void *p, size_t size, bool free_old_on_error)
{
void *ret=NULL;
if (size == 0) {
if (free_old_on_error) {
SAFE_FREE(p);
}
DEBUG(2,("Realloc asked for 0 bytes\n"));
return NULL;
}
#if defined(PARANOID_MALLOC_CHECKER)
if (!p) {
ret = (void *)malloc_(size);
} else {
ret = (void *)realloc_(p,size);
}
#else
if (!p) {
ret = (void *)malloc(size);
} else {
ret = (void *)realloc(p,size);
}
#endif
if (!ret) {
if (free_old_on_error && p) {
SAFE_FREE(p);
}
DEBUG(0,("Memory allocation error: failed to expand to %d bytes\n",(int)size));
}
return(ret);
}
|
