1 /*
2 * emalloc - return new memory obtained from the system. Belch if none.
3 */
4 #include <config.h>
5 #include "ntp_types.h"
6 #include "ntp_malloc.h"
7 #include "ntp_syslog.h"
8 #include "ntp_stdlib.h"
9
10
11 /*
12 * When using the debug MS CRT allocator, each allocation stores the
13 * callsite __FILE__ and __LINE__, which is then displayed at process
14 * termination, to track down leaks. We don't want all of our
15 * allocations to show up as coming from emalloc.c, so we preserve the
16 * original callsite's source file and line using macros which pass
17 * __FILE__ and __LINE__ as parameters to these routines.
18 * Other debug malloc implementations can be used by defining
19 * EREALLOC_IMPL() as ports/winnt/include/config.h does.
20 */
21
22 void *
ereallocz(void * ptr,size_t newsz,size_t priorsz,int zero_init,const char * file,int line)23 ereallocz(
24 void * ptr,
25 size_t newsz,
26 size_t priorsz,
27 int zero_init
28 #ifdef EREALLOC_CALLSITE /* ntp_malloc.h */
29 ,
30 const char * file,
31 int line
32 #endif
33 )
34 {
35 char * mem;
36 size_t allocsz;
37
38 if (0 == newsz)
39 allocsz = 1;
40 else
41 allocsz = newsz;
42
43 mem = EREALLOC_IMPL(ptr, allocsz, file, line);
44 if (NULL == mem) {
45 msyslog_term = TRUE;
46 #ifndef EREALLOC_CALLSITE
47 msyslog(LOG_ERR, "fatal out of memory (%lu bytes)",
48 (u_long)newsz);
49 #else
50 msyslog(LOG_ERR,
51 "fatal out of memory %s line %d (%lu bytes)",
52 file, line, (u_long)newsz);
53 #endif
54 exit(1);
55 }
56
57 if (zero_init && newsz > priorsz)
58 zero_mem(mem + priorsz, newsz - priorsz);
59
60 return mem;
61 }
62
63 /* oreallocarray.c is licensed under the following:
64 * Copyright (c) 2008 Otto Moerbeek <otto@drijf.net>
65 *
66 * Permission to use, copy, modify, and distribute this software for any
67 * purpose with or without fee is hereby granted, provided that the above
68 * copyright notice and this permission notice appear in all copies.
69 *
70 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
71 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
72 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
73 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
74 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
75 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
76 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
77 */
78
79 /*
80 * This is sqrt(SIZE_MAX+1), as s1*s2 <= SIZE_MAX
81 * if both s1 < MUL_NO_OVERFLOW and s2 < MUL_NO_OVERFLOW
82 */
83 #define MUL_NO_OVERFLOW ((size_t)1 << (sizeof(size_t) * 4))
84
85 void *
oreallocarrayxz(void * optr,size_t nmemb,size_t size,size_t extra,const char * file,int line)86 oreallocarrayxz(
87 void *optr,
88 size_t nmemb,
89 size_t size,
90 size_t extra
91 #ifdef EREALLOC_CALLSITE /* ntp_malloc.h */
92 ,
93 const char * file,
94 int line
95 #endif
96 )
97 {
98 if ((nmemb >= MUL_NO_OVERFLOW || size >= MUL_NO_OVERFLOW) &&
99 nmemb > 0 && SIZE_MAX / nmemb < size) {
100 #ifndef EREALLOC_CALLSITE
101 msyslog(LOG_ERR, "fatal allocation size overflow");
102 #else
103 msyslog(LOG_ERR,
104 "fatal allocation size overflow %s line %d",
105 file, line);
106 #endif
107 exit(1);
108 }
109 #ifndef EREALLOC_CALLSITE
110 return ereallocz(optr, extra + (size * nmemb), 0, TRUE);
111 #else
112 return ereallocz(optr, extra + (size * nmemb), 0, TRUE, file, line);
113 #endif
114 }
115
116 char *
estrdup_impl(const char * str,const char * file,int line)117 estrdup_impl(
118 const char * str
119 #ifdef EREALLOC_CALLSITE
120 ,
121 const char * file,
122 int line
123 #endif
124 )
125 {
126 char * copy;
127 size_t bytes;
128
129 bytes = strlen(str) + 1;
130 copy = ereallocz(NULL, bytes, 0, FALSE
131 #ifdef EREALLOC_CALLSITE
132 , file, line
133 #endif
134 );
135 memcpy(copy, str, bytes);
136
137 return copy;
138 }
139
140
141 #if 0
142 #ifndef EREALLOC_CALLSITE
143 void *
144 emalloc(size_t newsz)
145 {
146 return ereallocz(NULL, newsz, 0, FALSE);
147 }
148 #endif
149 #endif
150
151