199767f8919635c4928607450d9e0abb932109ceToomas Soome/*
199767f8919635c4928607450d9e0abb932109ceToomas Soome * This module derived from code donated to the FreeBSD Project by
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Matthew Dillon <dillon@backplane.com>
199767f8919635c4928607450d9e0abb932109ceToomas Soome *
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Copyright (c) 1998 The FreeBSD Project
199767f8919635c4928607450d9e0abb932109ceToomas Soome * All rights reserved.
199767f8919635c4928607450d9e0abb932109ceToomas Soome *
199767f8919635c4928607450d9e0abb932109ceToomas Soome * Redistribution and use in source and binary forms, with or without
199767f8919635c4928607450d9e0abb932109ceToomas Soome * modification, are permitted provided that the following conditions
199767f8919635c4928607450d9e0abb932109ceToomas Soome * are met:
199767f8919635c4928607450d9e0abb932109ceToomas Soome * 1. Redistributions of source code must retain the above copyright
199767f8919635c4928607450d9e0abb932109ceToomas Soome * notice, this list of conditions and the following disclaimer.
199767f8919635c4928607450d9e0abb932109ceToomas Soome * 2. Redistributions in binary form must reproduce the above copyright
199767f8919635c4928607450d9e0abb932109ceToomas Soome * notice, this list of conditions and the following disclaimer in the
199767f8919635c4928607450d9e0abb932109ceToomas Soome * documentation and/or other materials provided with the distribution.
199767f8919635c4928607450d9e0abb932109ceToomas Soome *
199767f8919635c4928607450d9e0abb932109ceToomas Soome * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
199767f8919635c4928607450d9e0abb932109ceToomas Soome * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
199767f8919635c4928607450d9e0abb932109ceToomas Soome * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
199767f8919635c4928607450d9e0abb932109ceToomas Soome * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
199767f8919635c4928607450d9e0abb932109ceToomas Soome * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
199767f8919635c4928607450d9e0abb932109ceToomas Soome * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
199767f8919635c4928607450d9e0abb932109ceToomas Soome * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
199767f8919635c4928607450d9e0abb932109ceToomas Soome * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
199767f8919635c4928607450d9e0abb932109ceToomas Soome * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
199767f8919635c4928607450d9e0abb932109ceToomas Soome * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
199767f8919635c4928607450d9e0abb932109ceToomas Soome * SUCH DAMAGE.
199767f8919635c4928607450d9e0abb932109ceToomas Soome */
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <sys/cdefs.h>
199767f8919635c4928607450d9e0abb932109ceToomas Soome__FBSDID("$FreeBSD$");
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome/*
199767f8919635c4928607450d9e0abb932109ceToomas Soome * MALLOC.C - malloc equivalent, runs on top of zalloc and uses sbrk
199767f8919635c4928607450d9e0abb932109ceToomas Soome */
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#include "zalloc_defs.h"
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic MemPool MallocPool;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef DMALLOCDEBUG
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int MallocMax;
199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int MallocCount;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomevoid mallocstats(void);
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef malloc
199767f8919635c4928607450d9e0abb932109ceToomas Soome#undef malloc
199767f8919635c4928607450d9e0abb932109ceToomas Soome#undef free
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomevoid *
199767f8919635c4928607450d9e0abb932109ceToomas SoomeMalloc(size_t bytes, const char *file, int line)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome Guard *res;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef USEENDGUARD
199767f8919635c4928607450d9e0abb932109ceToomas Soome bytes += MALLOCALIGN + 1;
199767f8919635c4928607450d9e0abb932109ceToomas Soome#else
199767f8919635c4928607450d9e0abb932109ceToomas Soome bytes += MALLOCALIGN;
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome while ((res = znalloc(&MallocPool, bytes)) == NULL) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome int incr = (bytes + BLKEXTENDMASK) & ~BLKEXTENDMASK;
199767f8919635c4928607450d9e0abb932109ceToomas Soome char *base;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome if ((base = sbrk(incr)) == (char *)-1)
199767f8919635c4928607450d9e0abb932109ceToomas Soome return(NULL);
199767f8919635c4928607450d9e0abb932109ceToomas Soome zextendPool(&MallocPool, base, incr);
199767f8919635c4928607450d9e0abb932109ceToomas Soome zfree(&MallocPool, base, incr);
199767f8919635c4928607450d9e0abb932109ceToomas Soome }
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef DMALLOCDEBUG
199767f8919635c4928607450d9e0abb932109ceToomas Soome if (++MallocCount > MallocMax)
199767f8919635c4928607450d9e0abb932109ceToomas Soome MallocMax = MallocCount;
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef USEGUARD
199767f8919635c4928607450d9e0abb932109ceToomas Soome res->ga_Magic = GAMAGIC;
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome res->ga_Bytes = bytes;
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef USEENDGUARD
199767f8919635c4928607450d9e0abb932109ceToomas Soome *((signed char *)res + bytes - 1) = -2;
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome return((char *)res + MALLOCALIGN);
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomevoid
199767f8919635c4928607450d9e0abb932109ceToomas SoomeFree(void *ptr, const char *file, int line)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome size_t bytes;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome if (ptr != NULL) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome Guard *res = (void *)((char *)ptr - MALLOCALIGN);
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome if (file == NULL)
199767f8919635c4928607450d9e0abb932109ceToomas Soome file = "unknown";
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef USEGUARD
199767f8919635c4928607450d9e0abb932109ceToomas Soome if (res->ga_Magic == GAFREE) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome printf("free: duplicate free @ %p from %s:%d\n", ptr, file, line);
199767f8919635c4928607450d9e0abb932109ceToomas Soome return;
199767f8919635c4928607450d9e0abb932109ceToomas Soome }
199767f8919635c4928607450d9e0abb932109ceToomas Soome if (res->ga_Magic != GAMAGIC)
199767f8919635c4928607450d9e0abb932109ceToomas Soome panic("free: guard1 fail @ %p from %s:%d", ptr, file, line);
199767f8919635c4928607450d9e0abb932109ceToomas Soome res->ga_Magic = GAFREE;
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef USEENDGUARD
199767f8919635c4928607450d9e0abb932109ceToomas Soome if (*((signed char *)res + res->ga_Bytes - 1) == -1) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome printf("free: duplicate2 free @ %p from %s:%d\n", ptr, file, line);
199767f8919635c4928607450d9e0abb932109ceToomas Soome return;
199767f8919635c4928607450d9e0abb932109ceToomas Soome }
199767f8919635c4928607450d9e0abb932109ceToomas Soome if (*((signed char *)res + res->ga_Bytes - 1) != -2)
199767f8919635c4928607450d9e0abb932109ceToomas Soome panic("free: guard2 fail @ %p + %zu from %s:%d", ptr, res->ga_Bytes - MALLOCALIGN, file, line);
199767f8919635c4928607450d9e0abb932109ceToomas Soome *((signed char *)res + res->ga_Bytes - 1) = -1;
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome bytes = res->ga_Bytes;
199767f8919635c4928607450d9e0abb932109ceToomas Soome zfree(&MallocPool, res, bytes);
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef DMALLOCDEBUG
199767f8919635c4928607450d9e0abb932109ceToomas Soome --MallocCount;
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome }
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomevoid *
199767f8919635c4928607450d9e0abb932109ceToomas SoomeCalloc(size_t n1, size_t n2, const char *file, int line)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome uintptr_t bytes = (uintptr_t)n1 * (uintptr_t)n2;
199767f8919635c4928607450d9e0abb932109ceToomas Soome void *res;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome if ((res = Malloc(bytes, file, line)) != NULL) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome bzero(res, bytes);
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef DMALLOCDEBUG
199767f8919635c4928607450d9e0abb932109ceToomas Soome if (++MallocCount > MallocMax)
199767f8919635c4928607450d9e0abb932109ceToomas Soome MallocMax = MallocCount;
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome }
199767f8919635c4928607450d9e0abb932109ceToomas Soome return(res);
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome/*
199767f8919635c4928607450d9e0abb932109ceToomas Soome * realloc() - I could be fancier here and free the old buffer before
199767f8919635c4928607450d9e0abb932109ceToomas Soome * allocating the new one (saving potential fragmentation
199767f8919635c4928607450d9e0abb932109ceToomas Soome * and potential buffer copies). But I don't bother.
199767f8919635c4928607450d9e0abb932109ceToomas Soome */
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomevoid *
199767f8919635c4928607450d9e0abb932109ceToomas SoomeRealloc(void *ptr, size_t size, const char *file, int line)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome void *res;
199767f8919635c4928607450d9e0abb932109ceToomas Soome size_t old;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome if ((res = Malloc(size, file, line)) != NULL) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome if (ptr) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome old = *(size_t *)((char *)ptr - MALLOCALIGN) - MALLOCALIGN;
199767f8919635c4928607450d9e0abb932109ceToomas Soome if (old < size)
199767f8919635c4928607450d9e0abb932109ceToomas Soome bcopy(ptr, res, old);
199767f8919635c4928607450d9e0abb932109ceToomas Soome else
199767f8919635c4928607450d9e0abb932109ceToomas Soome bcopy(ptr, res, size);
199767f8919635c4928607450d9e0abb932109ceToomas Soome Free(ptr, file, line);
199767f8919635c4928607450d9e0abb932109ceToomas Soome } else {
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef DMALLOCDEBUG
199767f8919635c4928607450d9e0abb932109ceToomas Soome if (++MallocCount > MallocMax)
199767f8919635c4928607450d9e0abb932109ceToomas Soome MallocMax = MallocCount;
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef EXITSTATS
199767f8919635c4928607450d9e0abb932109ceToomas Soome if (DidAtExit == 0) {
199767f8919635c4928607450d9e0abb932109ceToomas Soome DidAtExit = 1;
199767f8919635c4928607450d9e0abb932109ceToomas Soome atexit(mallocstats);
199767f8919635c4928607450d9e0abb932109ceToomas Soome }
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome }
199767f8919635c4928607450d9e0abb932109ceToomas Soome }
199767f8919635c4928607450d9e0abb932109ceToomas Soome return(res);
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomevoid *
199767f8919635c4928607450d9e0abb932109ceToomas SoomeReallocf(void *ptr, size_t size, const char *file, int line)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome void *res;
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome if ((res = Realloc(ptr, size, file, line)) == NULL)
199767f8919635c4928607450d9e0abb932109ceToomas Soome Free(ptr, file, line);
199767f8919635c4928607450d9e0abb932109ceToomas Soome return(res);
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef DMALLOCDEBUG
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soomevoid
199767f8919635c4928607450d9e0abb932109ceToomas Soomemallocstats(void)
199767f8919635c4928607450d9e0abb932109ceToomas Soome{
199767f8919635c4928607450d9e0abb932109ceToomas Soome printf("Active Allocations: %d/%d\n", MallocCount, MallocMax);
199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifdef ZALLOCDEBUG
199767f8919635c4928607450d9e0abb932109ceToomas Soome zallocstats(&MallocPool);
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome}
199767f8919635c4928607450d9e0abb932109ceToomas Soome
199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif
199767f8919635c4928607450d9e0abb932109ceToomas Soome