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
22void *
23ereallocz(
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
85void *
86oreallocarrayxz(
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
116char *
117estrdup_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
143void *
144emalloc(size_t newsz)
145{
146	return ereallocz(NULL, newsz, 0, FALSE);
147}
148#endif
149#endif
150
151