/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License (the "License").
* You may not use this file except in compliance with the License.
*
* You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
* or http://www.opensolaris.org/os/licensing.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at usr/src/OPENSOLARIS.LICENSE.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information: Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*/
/*
* Copyright 2007 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
*
* alloc.c -- memory allocation wrapper functions, for eft.so FMD module
*
*/
#pragma ident "%Z%%M% %I% %E% SMI"
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <fm/fmd_api.h>
#include "alloc.h"
#include "out.h"
#include "stats.h"
extern fmd_hdl_t *Hdl; /* handle from eft.c */
/* room to store size, possibly more to maintain alignment for long longs */
#define HDRSIZ sizeof (long long)
static struct stats *Malloctotal;
static struct stats *Freetotal;
static struct stats *Malloccount;
static struct stats *Freecount;
static int totalcount;
void
alloc_init(void)
{
Malloctotal = stats_new_counter("alloc.total", "bytes allocated", 1);
Freetotal = stats_new_counter("free.total", "bytes freed", 1);
Malloccount = stats_new_counter("alloc.calls", "alloc calls", 1);
Freecount = stats_new_counter("free.calls", "free calls", 1);
}
void
alloc_fini(void)
{
struct stats *mt, *ft, *mc, *fc;
mt = Malloctotal;
ft = Freetotal;
mc = Malloccount;
fc = Freecount;
Malloctotal = NULL;
Freetotal = NULL;
Malloccount = NULL;
Freecount = NULL;
stats_delete(mt);
stats_delete(ft);
stats_delete(mc);
stats_delete(fc);
}
/*
* alloc_malloc -- a malloc() with checks
*
* this routine is typically called via the MALLOC() macro in alloc.h
*/
/*ARGSUSED*/
void *
alloc_malloc(size_t nbytes, const char *fname, int line)
{
char *retval;
ASSERT(nbytes > 0);
retval = fmd_hdl_alloc(Hdl, nbytes + HDRSIZ, FMD_SLEEP);
/* retval can't be NULL since fmd_hdl_alloc() sleeps for memory */
bcopy((void *)&nbytes, (void *)retval, sizeof (nbytes));
retval += HDRSIZ;
if (Malloctotal)
stats_counter_add(Malloctotal, nbytes);
if (Malloccount)
stats_counter_bump(Malloccount);
totalcount += nbytes + HDRSIZ;
return ((void *)retval);
}
/*
* alloc_realloc -- a realloc() with checks
*
* this routine is typically called via the REALLOC() macro in alloc.h
*/
void *
alloc_realloc(void *ptr, size_t nbytes, const char *fname, int line)
{
void *retval = alloc_malloc(nbytes, fname, line);
if (ptr != NULL) {
size_t osize;
bcopy((void *)((char *)ptr - HDRSIZ), (void *)&osize,
sizeof (osize));
/* now we have the new memory, copy in the old contents */
bcopy(ptr, retval, (osize < nbytes) ? osize : nbytes);
/* don't need the old memory anymore */
alloc_free((char *)ptr, fname, line);
}
return (retval);
}
/*
* alloc_strdup -- a strdup() with checks
*
* this routine is typically called via the STRDUP() macro in alloc.h
*/
char *
alloc_strdup(const char *ptr, const char *fname, int line)
{
char *retval = alloc_malloc(strlen(ptr) + 1, fname, line);
(void) strcpy(retval, ptr);
return (retval);
}
/*
* alloc_free -- a free() with checks
*
* this routine is typically called via the FREE() macro in alloc.h
*/
/*ARGSUSED*/
void
alloc_free(void *ptr, const char *fname, int line)
{
size_t osize;
ASSERT(ptr != NULL);
bcopy((void *)((char *)ptr - HDRSIZ), (void *)&osize, sizeof (osize));
/* nothing to check in this version */
fmd_hdl_free(Hdl, (char *)ptr - HDRSIZ, osize + HDRSIZ);
if (Freetotal)
stats_counter_add(Freetotal, osize);
if (Freecount)
stats_counter_bump(Freecount);
totalcount -= osize + HDRSIZ;
}
int
alloc_total()
{
return (totalcount);
}
/*
* variants that don't maintain size in header - saves space
*/
void *
alloc_xmalloc(size_t nbytes)
{
char *retval;
ASSERT(nbytes > 0);
retval = fmd_hdl_alloc(Hdl, nbytes, FMD_SLEEP);
if (Malloctotal)
stats_counter_add(Malloctotal, nbytes);
if (Malloccount)
stats_counter_bump(Malloccount);
totalcount += nbytes;
return ((void *)retval);
}
void
alloc_xfree(void *ptr, size_t size)
{
ASSERT(ptr != NULL);
fmd_hdl_free(Hdl, (char *)ptr, size);
if (Freetotal)
stats_counter_add(Freetotal, size);
if (Freecount)
stats_counter_bump(Freecount);
totalcount -= size;
}