2 * Copyright (c) 2004 Teodor Sigaev <teodor@sigaev.ru>
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. Neither the name of the author nor the names of any co-contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
17 * THIS SOFTWARE IS PROVIDED BY CONTRIBUTORS ``AS IS'' AND ANY EXPRESS
18 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL CONTRIBUTORS BE LIABLE FOR ANY
21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
23 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
38 tmalloc(size_t size) {
39 void *ptr = malloc(size);
41 tlog(TL_CRIT|TL_EXIT, "Can't allocate %d bytes", size);
46 t0malloc(size_t size) {
47 void *ptr = tmalloc(size);
53 trealloc(void * ptr, size_t size) {
55 ptr = realloc(ptr,size);
57 tlog(TL_CRIT|TL_EXIT, "Can't reallocate to %d bytes", size);
70 char * dest = strdup(src);
72 tlog(TL_CRIT|TL_EXIT, "Can't strdup %d bytes", strlen(src)+1);
77 tnstrdup(char *src, int len) {
78 char *dest=(char*)tmalloc(len+1);
79 memcpy(dest, src, len);
85 strlower(char * src) {
89 *ptr = tolower(*(unsigned char *) ptr);
96 strupper(char * src) {
100 *ptr = toupper(*(unsigned char *) ptr);
107 clrspace(char *buf) {
108 char *ptr=buf, *ptrc=buf;
110 if (!isspace(*ptr)) {
122 allocMemoryContext(MemoryContext* parent, int flags) {
125 res = (MemoryContext*)tmalloc( sizeof(MemoryContext) );
127 res->parent = parent;
129 res->child = parent->child;
134 res->chunk = (MemoryChunk*)tmalloc( MEMCHNKHDRSZ + CNTXCHUNK );
135 res->chunk->size = res->chunk->freesize = CNTXCHUNK;
136 res->chunk->next = NULL;
142 freeMemoryContext(MemoryContext *cntx) {
143 MemoryContext *ptr=cntx;
144 MemoryChunk *chunk, *chunkptr;
147 cntx->parent->child=NULL;
154 chunk=chunkptr->next;
165 resetMemoryContext(MemoryContext *cntx) {
166 MemoryChunk *chunk, *chunkptr;
169 chunkptr = cntx->chunk;
170 chunkptr->freesize = chunkptr->size;
171 chunkptr = chunkptr->next;
172 cntx->chunk->next = NULL;
175 chunk=chunkptr->next;
184 mcalloc(MemoryContext *cntx, size_t size) {
185 MemoryChunk *chunk = cntx->chunk;
186 MCAllocatedSpace *alloc;
188 size = PTRALIGN(size) + MCASHDRSZ;
189 if ( cntx->flags & MC_DEBUG )
190 size += sizeof(u_int32_t);
192 if ( chunk->freesize < size ) {
193 MemoryChunk *newchunk;
195 if ( size >= CNTXCHUNK ) {
196 newchunk = (MemoryChunk*)tmalloc(MEMCHNKHDRSZ + size);
197 newchunk->size = newchunk->freesize = size;
198 newchunk->next = chunk->next;
199 chunk->next = newchunk;
201 newchunk = (MemoryChunk*)tmalloc(MEMCHNKHDRSZ + CNTXCHUNK);
202 newchunk->size = newchunk->freesize = CNTXCHUNK;
203 newchunk->next = chunk;
204 cntx->chunk = newchunk;
210 alloc = (MCAllocatedSpace*)( chunk->data + chunk->size - chunk->freesize );
213 chunk->freesize -= size;
215 if ( cntx->flags & MC_DEBUG ) {
216 *(u_int32_t*)((char*)alloc + alloc->size - sizeof(u_int32_t) ) = MCMAGICKNUMBER;
217 memset( alloc->data, 0xc3, size - MCASHDRSZ - sizeof(u_int32_t) );
220 return (void*)(alloc->data);
224 mc0alloc(MemoryContext *cntx, size_t size) {
225 void *res = mcalloc(cntx, size);
226 memset( res, 0, size );
231 mcrealloc(void * ptr, size_t size) {
232 MCAllocatedSpace *alloc = (MCAllocatedSpace*)( (char*)ptr - MCASHDRSZ );
236 tlog(TL_CRIT|TL_EXIT, "mcrealloc: realloc null pointer");
238 if ( alloc->cntx->flags & MC_DEBUG ) {
239 tassert( *(u_int32_t*)( (char*)alloc + alloc->size - sizeof(u_int32_t) ) == MCMAGICKNUMBER );
240 realsize = alloc->size - MCASHDRSZ - sizeof(u_int32_t);
242 realsize = alloc->size - MCASHDRSZ;
244 if ( size > realsize ) {
245 MemoryChunk *chunk = alloc->cntx->chunk;
246 if ( (char*)alloc == chunk->data + chunk->size - chunk->freesize - alloc->size &&
247 PTRALIGN(size)-realsize <= alloc->cntx->chunk->freesize ) {
249 alloc->cntx->chunk->freesize -= PTRALIGN(size)-realsize;
250 alloc->size+=PTRALIGN(size)-realsize;
251 if ( alloc->cntx->flags & MC_DEBUG ) {
252 memset( (char*)(alloc->data) + realsize, 0xc3, PTRALIGN(size)-realsize );
253 *(u_int32_t*)((char*)alloc + alloc->size - sizeof(u_int32_t) ) = MCMAGICKNUMBER;
256 void *newptr = mcalloc(alloc->cntx, size);
257 memcpy( newptr, ptr, realsize );
259 if ( alloc->cntx->flags & MC_DEBUG )
260 memset( (char*)ptr + realsize, 0xc3, PTRALIGN(size)-realsize );
270 MCAllocatedSpace *alloc = (MCAllocatedSpace*)( (char*)ptr - MCASHDRSZ );
274 tlog(TL_CRIT|TL_EXIT, "mcfree: free null pointer");
276 if ( alloc->cntx->flags & MC_DEBUG )
277 tassert( *(u_int32_t*)((char*)alloc + alloc->size - sizeof(u_int32_t) ) == MCMAGICKNUMBER );
279 chunk = alloc->cntx->chunk;
280 if ( (char*)alloc == chunk->data + chunk->size - chunk->freesize - alloc->size ) /* last allocated value */
281 alloc->cntx->chunk->freesize+=alloc->size;
287 mcstrdup(MemoryContext *cntx, char * src) {
288 return mcnstrdup(cntx, src, strlen(src));
292 mcnstrdup(MemoryContext *cntx, char *src, int len) {
293 char *dest=(char*)mcalloc(cntx, len+1);
294 memcpy(dest, src, len);