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 *
oreallocarray(void * optr,size_t nmemb,size_t size,const char * file,int line)86 oreallocarray(
87 void *optr,
88 size_t nmemb,
89 size_t size
90 #ifdef EREALLOC_CALLSITE /* ntp_malloc.h */
91 ,
92 const char * file,
93 int line
94 #endif
95 )
96 {
97 if ((nmemb >= MUL_NO_OVERFLOW || size >= MUL_NO_OVERFLOW) &&
98 nmemb > 0 && SIZE_MAX / nmemb < size) {
99 #ifndef EREALLOC_CALLSITE
100 msyslog(LOG_ERR, "fatal allocation size overflow");
101 #else
102 msyslog(LOG_ERR,
103 "fatal allocation size overflow %s line %d",
104 file, line);
105 #endif
106 exit(1);
107 }
108 #ifndef EREALLOC_CALLSITE
109 return ereallocz(optr, (size * nmemb), 0, FALSE);
110 #else
111 return ereallocz(optr, (size * nmemb), 0, FALSE, file, line);
112 #endif
113 }
114
115 char *
estrdup_impl(const char * str,const char * file,int line)116 estrdup_impl(
117 const char * str
118 #ifdef EREALLOC_CALLSITE
119 ,
120 const char * file,
121 int line
122 #endif
123 )
124 {
125 char * copy;
126 size_t bytes;
127
128 bytes = strlen(str) + 1;
129 copy = ereallocz(NULL, bytes, 0, FALSE
130 #ifdef EREALLOC_CALLSITE
131 , file, line
132 #endif
133 );
134 memcpy(copy, str, bytes);
135
136 return copy;
137 }
138
139
140 #if 0
141 #ifndef EREALLOC_CALLSITE
142 void *
143 emalloc(size_t newsz)
144 {
145 return ereallocz(NULL, newsz, 0, FALSE);
146 }
147 #endif
148 #endif
149
150