xref: /onnv-gate/usr/src/cmd/fm/eversholt/common/alloc.c (revision 0:68f95e015346)
1*0Sstevel@tonic-gate /*
2*0Sstevel@tonic-gate  * CDDL HEADER START
3*0Sstevel@tonic-gate  *
4*0Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5*0Sstevel@tonic-gate  * Common Development and Distribution License, Version 1.0 only
6*0Sstevel@tonic-gate  * (the "License").  You may not use this file except in compliance
7*0Sstevel@tonic-gate  * with the License.
8*0Sstevel@tonic-gate  *
9*0Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*0Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
11*0Sstevel@tonic-gate  * See the License for the specific language governing permissions
12*0Sstevel@tonic-gate  * and limitations under the License.
13*0Sstevel@tonic-gate  *
14*0Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
15*0Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*0Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
17*0Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
18*0Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
19*0Sstevel@tonic-gate  *
20*0Sstevel@tonic-gate  * CDDL HEADER END
21*0Sstevel@tonic-gate  */
22*0Sstevel@tonic-gate /*
23*0Sstevel@tonic-gate  * Copyright 2004 Sun Microsystems, Inc.  All rights reserved.
24*0Sstevel@tonic-gate  * Use is subject to license terms.
25*0Sstevel@tonic-gate  *
26*0Sstevel@tonic-gate  * alloc.c -- memory allocation wrapper functions, replacable in more
27*0Sstevel@tonic-gate  * constrained environments, such as within a DE.
28*0Sstevel@tonic-gate  */
29*0Sstevel@tonic-gate 
30*0Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
31*0Sstevel@tonic-gate 
32*0Sstevel@tonic-gate #include <stdlib.h>
33*0Sstevel@tonic-gate #include <string.h>
34*0Sstevel@tonic-gate 
35*0Sstevel@tonic-gate #include "alloc.h"
36*0Sstevel@tonic-gate #include "out.h"
37*0Sstevel@tonic-gate #include "stats.h"
38*0Sstevel@tonic-gate 
39*0Sstevel@tonic-gate static struct stats *Malloctotal;
40*0Sstevel@tonic-gate static struct stats *Malloccount;
41*0Sstevel@tonic-gate 
42*0Sstevel@tonic-gate void
43*0Sstevel@tonic-gate alloc_init(void)
44*0Sstevel@tonic-gate {
45*0Sstevel@tonic-gate 	Malloctotal = stats_new_counter("alloc.total", "bytes allocated", 1);
46*0Sstevel@tonic-gate 	Malloccount = stats_new_counter("alloc.calls", "total calls", 1);
47*0Sstevel@tonic-gate }
48*0Sstevel@tonic-gate 
49*0Sstevel@tonic-gate void
50*0Sstevel@tonic-gate alloc_fini(void)
51*0Sstevel@tonic-gate {
52*0Sstevel@tonic-gate 	struct stats *mt, *mc;
53*0Sstevel@tonic-gate 
54*0Sstevel@tonic-gate 	mt = Malloctotal;
55*0Sstevel@tonic-gate 	mc = Malloccount;
56*0Sstevel@tonic-gate 
57*0Sstevel@tonic-gate 	Malloctotal = NULL;
58*0Sstevel@tonic-gate 	Malloccount = NULL;
59*0Sstevel@tonic-gate 
60*0Sstevel@tonic-gate 	stats_delete(mt);
61*0Sstevel@tonic-gate 	stats_delete(mc);
62*0Sstevel@tonic-gate }
63*0Sstevel@tonic-gate 
64*0Sstevel@tonic-gate /*
65*0Sstevel@tonic-gate  * alloc_malloc -- a malloc() with checks
66*0Sstevel@tonic-gate  *
67*0Sstevel@tonic-gate  * this routine is typically called via the MALLOC() macro in alloc.h
68*0Sstevel@tonic-gate  */
69*0Sstevel@tonic-gate 
70*0Sstevel@tonic-gate void *
71*0Sstevel@tonic-gate alloc_malloc(size_t nbytes, const char *fname, int line)
72*0Sstevel@tonic-gate {
73*0Sstevel@tonic-gate 	void *retval = malloc(nbytes);
74*0Sstevel@tonic-gate 
75*0Sstevel@tonic-gate 	if (retval == NULL)
76*0Sstevel@tonic-gate 		outfl(O_DIE, fname, line, "malloc: out of memory");
77*0Sstevel@tonic-gate 
78*0Sstevel@tonic-gate 	if (Malloctotal)
79*0Sstevel@tonic-gate 		stats_counter_add(Malloctotal, nbytes);
80*0Sstevel@tonic-gate 
81*0Sstevel@tonic-gate 	if (Malloccount)
82*0Sstevel@tonic-gate 		stats_counter_bump(Malloccount);
83*0Sstevel@tonic-gate 
84*0Sstevel@tonic-gate 	return (retval);
85*0Sstevel@tonic-gate }
86*0Sstevel@tonic-gate 
87*0Sstevel@tonic-gate /*
88*0Sstevel@tonic-gate  * alloc_realloc -- a realloc() with checks
89*0Sstevel@tonic-gate  *
90*0Sstevel@tonic-gate  * this routine is typically called via the REALLOC() macro in alloc.h
91*0Sstevel@tonic-gate  */
92*0Sstevel@tonic-gate void *
93*0Sstevel@tonic-gate alloc_realloc(void *ptr, size_t nbytes, const char *fname, int line)
94*0Sstevel@tonic-gate {
95*0Sstevel@tonic-gate 	void *retval = realloc(ptr, nbytes);
96*0Sstevel@tonic-gate 
97*0Sstevel@tonic-gate 	if (retval == NULL)
98*0Sstevel@tonic-gate 		out(O_DIE, fname, line, "realloc: out of memory");
99*0Sstevel@tonic-gate 
100*0Sstevel@tonic-gate 	return (retval);
101*0Sstevel@tonic-gate }
102*0Sstevel@tonic-gate 
103*0Sstevel@tonic-gate /*
104*0Sstevel@tonic-gate  * alloc_strdup -- a strdup() with checks
105*0Sstevel@tonic-gate  *
106*0Sstevel@tonic-gate  * this routine is typically called via the STRDUP() macro in alloc.h
107*0Sstevel@tonic-gate  */
108*0Sstevel@tonic-gate char *
109*0Sstevel@tonic-gate alloc_strdup(const char *ptr, const char *fname, int line)
110*0Sstevel@tonic-gate {
111*0Sstevel@tonic-gate 	char *retval = strdup(ptr);
112*0Sstevel@tonic-gate 
113*0Sstevel@tonic-gate 	if (retval == NULL)
114*0Sstevel@tonic-gate 		outfl(O_DIE, fname, line, "strdup: out of memory");
115*0Sstevel@tonic-gate 
116*0Sstevel@tonic-gate 	return (retval);
117*0Sstevel@tonic-gate }
118*0Sstevel@tonic-gate 
119*0Sstevel@tonic-gate /*
120*0Sstevel@tonic-gate  * alloc_free -- a free() with checks
121*0Sstevel@tonic-gate  *
122*0Sstevel@tonic-gate  * this routine is typically called via the FREE() macro in alloc.h
123*0Sstevel@tonic-gate  */
124*0Sstevel@tonic-gate /*ARGSUSED1*/
125*0Sstevel@tonic-gate void
126*0Sstevel@tonic-gate alloc_free(void *ptr, const char *fname, int line)
127*0Sstevel@tonic-gate {
128*0Sstevel@tonic-gate 	/* nothing to check in this version */
129*0Sstevel@tonic-gate 	free(ptr);
130*0Sstevel@tonic-gate }
131