emalloc.c revision 290001 
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 *
86oreallocarray(
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
115char *
116estrdup_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
142void *
143emalloc(size_t newsz)
144{
145	return ereallocz(NULL, newsz, 0, FALSE);
146}
147#endif
148#endif
149
150