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