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