1*479ab7f0SSascha Wildner /*
2*479ab7f0SSascha Wildner * This module derived from code donated to the FreeBSD Project by
3*479ab7f0SSascha Wildner * Matthew Dillon <dillon@backplane.com>
4*479ab7f0SSascha Wildner *
5*479ab7f0SSascha Wildner * Copyright (c) 1998 The FreeBSD Project
6*479ab7f0SSascha Wildner * All rights reserved.
7*479ab7f0SSascha Wildner *
8*479ab7f0SSascha Wildner * Redistribution and use in source and binary forms, with or without
9*479ab7f0SSascha Wildner * modification, are permitted provided that the following conditions
10*479ab7f0SSascha Wildner * are met:
11*479ab7f0SSascha Wildner * 1. Redistributions of source code must retain the above copyright
12*479ab7f0SSascha Wildner * notice, this list of conditions and the following disclaimer.
13*479ab7f0SSascha Wildner * 2. Redistributions in binary form must reproduce the above copyright
14*479ab7f0SSascha Wildner * notice, this list of conditions and the following disclaimer in the
15*479ab7f0SSascha Wildner * documentation and/or other materials provided with the distribution.
16*479ab7f0SSascha Wildner *
17*479ab7f0SSascha Wildner * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18*479ab7f0SSascha Wildner * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19*479ab7f0SSascha Wildner * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20*479ab7f0SSascha Wildner * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21*479ab7f0SSascha Wildner * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22*479ab7f0SSascha Wildner * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23*479ab7f0SSascha Wildner * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24*479ab7f0SSascha Wildner * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25*479ab7f0SSascha Wildner * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26*479ab7f0SSascha Wildner * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27*479ab7f0SSascha Wildner * SUCH DAMAGE.
28*479ab7f0SSascha Wildner *
29*479ab7f0SSascha Wildner * $FreeBSD: src/lib/libstand/zalloc_malloc.c,v 1.5 1999/08/28 00:05:35 peter Exp $
30*479ab7f0SSascha Wildner * $DragonFly: src/lib/libstand/zalloc_malloc.c,v 1.3 2005/03/13 15:10:03 swildner Exp $
31*479ab7f0SSascha Wildner */
32*479ab7f0SSascha Wildner
33*479ab7f0SSascha Wildner /*
34*479ab7f0SSascha Wildner * MALLOC.C - malloc equivalent, runs on top of zalloc and uses sbrk
35*479ab7f0SSascha Wildner */
36*479ab7f0SSascha Wildner
37*479ab7f0SSascha Wildner #include "zalloc_defs.h"
38*479ab7f0SSascha Wildner
39*479ab7f0SSascha Wildner static MemPool MallocPool;
40*479ab7f0SSascha Wildner
41*479ab7f0SSascha Wildner #ifdef DMALLOCDEBUG
42*479ab7f0SSascha Wildner static int MallocMax;
43*479ab7f0SSascha Wildner static int MallocCount;
44*479ab7f0SSascha Wildner
45*479ab7f0SSascha Wildner void mallocstats(void);
46*479ab7f0SSascha Wildner #endif
47*479ab7f0SSascha Wildner
48*479ab7f0SSascha Wildner #ifdef malloc
49*479ab7f0SSascha Wildner #undef malloc
50*479ab7f0SSascha Wildner #undef free
51*479ab7f0SSascha Wildner #endif
52*479ab7f0SSascha Wildner
53*479ab7f0SSascha Wildner void *
malloc(size_t bytes)54*479ab7f0SSascha Wildner malloc(size_t bytes)
55*479ab7f0SSascha Wildner {
56*479ab7f0SSascha Wildner Guard *res;
57*479ab7f0SSascha Wildner
58*479ab7f0SSascha Wildner #ifdef USEENDGUARD
59*479ab7f0SSascha Wildner bytes += MALLOCALIGN + 1;
60*479ab7f0SSascha Wildner #else
61*479ab7f0SSascha Wildner bytes += MALLOCALIGN;
62*479ab7f0SSascha Wildner #endif
63*479ab7f0SSascha Wildner
64*479ab7f0SSascha Wildner while ((res = znalloc(&MallocPool, bytes)) == NULL) {
65*479ab7f0SSascha Wildner int incr = (bytes + BLKEXTENDMASK) & ~BLKEXTENDMASK;
66*479ab7f0SSascha Wildner char *base;
67*479ab7f0SSascha Wildner
68*479ab7f0SSascha Wildner if ((base = sbrk(incr)) == (char *)-1)
69*479ab7f0SSascha Wildner return(NULL);
70*479ab7f0SSascha Wildner zextendPool(&MallocPool, base, incr);
71*479ab7f0SSascha Wildner zfree(&MallocPool, base, incr);
72*479ab7f0SSascha Wildner }
73*479ab7f0SSascha Wildner #ifdef DMALLOCDEBUG
74*479ab7f0SSascha Wildner if (++MallocCount > MallocMax)
75*479ab7f0SSascha Wildner MallocMax = MallocCount;
76*479ab7f0SSascha Wildner #endif
77*479ab7f0SSascha Wildner #ifdef USEGUARD
78*479ab7f0SSascha Wildner res->ga_Magic = GAMAGIC;
79*479ab7f0SSascha Wildner #endif
80*479ab7f0SSascha Wildner res->ga_Bytes = bytes;
81*479ab7f0SSascha Wildner #ifdef USEENDGUARD
82*479ab7f0SSascha Wildner *((char *)res + bytes - 1) = -2;
83*479ab7f0SSascha Wildner #endif
84*479ab7f0SSascha Wildner return((char *)res + MALLOCALIGN);
85*479ab7f0SSascha Wildner }
86*479ab7f0SSascha Wildner
87*479ab7f0SSascha Wildner void
free(void * ptr)88*479ab7f0SSascha Wildner free(void *ptr)
89*479ab7f0SSascha Wildner {
90*479ab7f0SSascha Wildner size_t bytes;
91*479ab7f0SSascha Wildner
92*479ab7f0SSascha Wildner if (ptr != NULL) {
93*479ab7f0SSascha Wildner Guard *res = (void *)((char *)ptr - MALLOCALIGN);
94*479ab7f0SSascha Wildner
95*479ab7f0SSascha Wildner #ifdef USEGUARD
96*479ab7f0SSascha Wildner if (res->ga_Magic != GAMAGIC)
97*479ab7f0SSascha Wildner panic("free: guard1x fail @ %p", ptr);
98*479ab7f0SSascha Wildner res->ga_Magic = -1;
99*479ab7f0SSascha Wildner #endif
100*479ab7f0SSascha Wildner #ifdef USEENDGUARD
101*479ab7f0SSascha Wildner if (*((char *)res + res->ga_Bytes - 1) != -2) {
102*479ab7f0SSascha Wildner panic("free: guard2 fail @ %p + %zu %d/-2",
103*479ab7f0SSascha Wildner ptr,
104*479ab7f0SSascha Wildner res->ga_Bytes - MALLOCALIGN - 1,
105*479ab7f0SSascha Wildner *((char *)res + res->ga_Bytes - 1));
106*479ab7f0SSascha Wildner }
107*479ab7f0SSascha Wildner *((char *)res + res->ga_Bytes - 1) = -1;
108*479ab7f0SSascha Wildner #endif
109*479ab7f0SSascha Wildner
110*479ab7f0SSascha Wildner bytes = res->ga_Bytes;
111*479ab7f0SSascha Wildner zfree(&MallocPool, res, bytes);
112*479ab7f0SSascha Wildner #ifdef DMALLOCDEBUG
113*479ab7f0SSascha Wildner --MallocCount;
114*479ab7f0SSascha Wildner #endif
115*479ab7f0SSascha Wildner }
116*479ab7f0SSascha Wildner }
117*479ab7f0SSascha Wildner
118*479ab7f0SSascha Wildner
119*479ab7f0SSascha Wildner void *
calloc(size_t n1,size_t n2)120*479ab7f0SSascha Wildner calloc(size_t n1, size_t n2)
121*479ab7f0SSascha Wildner {
122*479ab7f0SSascha Wildner uintptr_t bytes = (uintptr_t)n1 * (uintptr_t)n2;
123*479ab7f0SSascha Wildner void *res;
124*479ab7f0SSascha Wildner
125*479ab7f0SSascha Wildner if ((res = malloc(bytes)) != NULL) {
126*479ab7f0SSascha Wildner bzero(res, bytes);
127*479ab7f0SSascha Wildner }
128*479ab7f0SSascha Wildner return(res);
129*479ab7f0SSascha Wildner }
130*479ab7f0SSascha Wildner
131*479ab7f0SSascha Wildner /*
132*479ab7f0SSascha Wildner * realloc() - I could be fancier here and free the old buffer before
133*479ab7f0SSascha Wildner * allocating the new one (saving potential fragmentation
134*479ab7f0SSascha Wildner * and potential buffer copies). But I don't bother.
135*479ab7f0SSascha Wildner */
136*479ab7f0SSascha Wildner
137*479ab7f0SSascha Wildner void *
realloc(void * ptr,size_t size)138*479ab7f0SSascha Wildner realloc(void *ptr, size_t size)
139*479ab7f0SSascha Wildner {
140*479ab7f0SSascha Wildner void *res;
141*479ab7f0SSascha Wildner size_t old;
142*479ab7f0SSascha Wildner
143*479ab7f0SSascha Wildner if ((res = malloc(size)) != NULL) {
144*479ab7f0SSascha Wildner if (ptr) {
145*479ab7f0SSascha Wildner old = *(size_t *)((char *)ptr - MALLOCALIGN) - MALLOCALIGN;
146*479ab7f0SSascha Wildner #ifdef USEENDGUARD
147*479ab7f0SSascha Wildner --old;
148*479ab7f0SSascha Wildner #endif
149*479ab7f0SSascha Wildner if (old < size)
150*479ab7f0SSascha Wildner bcopy(ptr, res, old);
151*479ab7f0SSascha Wildner else
152*479ab7f0SSascha Wildner bcopy(ptr, res, size);
153*479ab7f0SSascha Wildner free(ptr);
154*479ab7f0SSascha Wildner } else {
155*479ab7f0SSascha Wildner #ifdef DMALLOCDEBUG
156*479ab7f0SSascha Wildner #ifdef EXITSTATS
157*479ab7f0SSascha Wildner if (DidAtExit == 0) {
158*479ab7f0SSascha Wildner DidAtExit = 1;
159*479ab7f0SSascha Wildner atexit(mallocstats);
160*479ab7f0SSascha Wildner }
161*479ab7f0SSascha Wildner #endif
162*479ab7f0SSascha Wildner #endif
163*479ab7f0SSascha Wildner }
164*479ab7f0SSascha Wildner }
165*479ab7f0SSascha Wildner return(res);
166*479ab7f0SSascha Wildner }
167*479ab7f0SSascha Wildner
168*479ab7f0SSascha Wildner void *
reallocf(void * ptr,size_t size)169*479ab7f0SSascha Wildner reallocf(void *ptr, size_t size)
170*479ab7f0SSascha Wildner {
171*479ab7f0SSascha Wildner void *res;
172*479ab7f0SSascha Wildner
173*479ab7f0SSascha Wildner if ((res = realloc(ptr, size)) == NULL)
174*479ab7f0SSascha Wildner free(ptr);
175*479ab7f0SSascha Wildner return(res);
176*479ab7f0SSascha Wildner }
177*479ab7f0SSascha Wildner
178*479ab7f0SSascha Wildner #ifdef DMALLOCDEBUG
179*479ab7f0SSascha Wildner
180*479ab7f0SSascha Wildner void
mallocstats(void)181*479ab7f0SSascha Wildner mallocstats(void)
182*479ab7f0SSascha Wildner {
183*479ab7f0SSascha Wildner printf("Active Allocations: %d/%d\n", MallocCount, MallocMax);
184*479ab7f0SSascha Wildner #ifdef ZALLOCDEBUG
185*479ab7f0SSascha Wildner zallocstats(&MallocPool);
186*479ab7f0SSascha Wildner #endif
187*479ab7f0SSascha Wildner }
188*479ab7f0SSascha Wildner
189*479ab7f0SSascha Wildner #endif
190*479ab7f0SSascha Wildner
191