kamailio/lib/cds/memory.c

/* 
 * Copyright (C) 2005 iptelorg GmbH
 *
 * This file is part of ser, a free SIP server.
 *
 * ser is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version
 *
 * For a license to use the ser software under conditions
 * other than those described here, or to purchase support for this
 * software, please contact iptel.org by e-mail at the following addresses:
 *    [email protected]
 *
 * ser is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */

/* Common Data Structures - functions for memory allocation and deallocation
 * and may be other memory management. */

#include <stdio.h>
#include <stdlib.h>
#include <cds/memory.h>
#include <cds/sync.h>
#include <cds/logger.h>

#ifdef TRACE_CDS_MEMORY

cds_mutex_t *mem_mutex = NULL;
int *allocated_cnt = NULL;
char *debug_file = "/tmp/mem.log";

#define write_debug(s,args...)	if (1) { \
	FILE *f = fopen(debug_file, "a"); \
	if (f) { \
		fprintf(f,s,##args); \
		fclose(f); \
	} \
	TRACE_LOG(s,##args); \
}

void *debug_malloc(int size, const char *file, int line)
{
	void *m = NULL;
	if (allocated_cnt && mem_mutex) {
		cds_mutex_lock(mem_mutex);
		(*allocated_cnt)++;
		cds_mutex_unlock(mem_mutex);
	}
#ifdef SER
	m = shm_malloc(size);
#else
	m = malloc(size);
#endif
	write_debug("ALLOC %p size %u from %s(%d)\n", m, size, file, line);
	/* LOG(L_INFO, "%p\n", m); */
	return m;
}

void debug_free(void *block, const char *file, int line)
{
	if (allocated_cnt && mem_mutex) {
		cds_mutex_lock(mem_mutex);
		(*allocated_cnt)--;
		cds_mutex_unlock(mem_mutex);
	}
#ifdef SER
	shm_free(block);
#else
	free(block);
#endif
	write_debug("FREE %p from %s(%d)\n", block, file, line);
}

void *debug_malloc_ex(unsigned int size)
{
	return debug_malloc(size, "<none>", 0);
}

void debug_free_ex(void *block)
{
	debug_free(block, "<none>", 0);
}

void cds_memory_trace_init()
{
	cds_mutex_init(mem_mutex);
	allocated_cnt = cds_malloc(sizeof(int));
	*allocated_cnt = 0;
}

void cds_memory_trace(char *dst, int dst_len)
{
	if (allocated_cnt && mem_mutex) {
		cds_mutex_lock(mem_mutex);
		snprintf(dst, dst_len, "There are allocated: %d memory blocks\n", *allocated_cnt);
		cds_mutex_unlock(mem_mutex);
	}
}

#else /* ! CDS_TRACE_MEMORY */

#ifdef SER

void* shm_malloc_x(unsigned int size)
{
	return shm_malloc(size);
}

void shm_free_x(void *ptr)
{
	shm_free(ptr);
}

#endif

#endif