186d7f5d3SJohn Marino /* $NetBSD: pool.c,v 1.1.1.2 2009/12/02 00:26:09 haad Exp $ */
286d7f5d3SJohn Marino
386d7f5d3SJohn Marino /*
486d7f5d3SJohn Marino * Copyright (C) 2001-2004 Sistina Software, Inc. All rights reserved.
586d7f5d3SJohn Marino * Copyright (C) 2004-2005 Red Hat, Inc. All rights reserved.
686d7f5d3SJohn Marino *
786d7f5d3SJohn Marino * This file is part of the device-mapper userspace tools.
886d7f5d3SJohn Marino *
986d7f5d3SJohn Marino * This copyrighted material is made available to anyone wishing to use,
1086d7f5d3SJohn Marino * modify, copy, or redistribute it subject to the terms and conditions
1186d7f5d3SJohn Marino * of the GNU Lesser General Public License v.2.1.
1286d7f5d3SJohn Marino *
1386d7f5d3SJohn Marino * You should have received a copy of the GNU Lesser General Public License
1486d7f5d3SJohn Marino * along with this program; if not, write to the Free Software Foundation,
1586d7f5d3SJohn Marino * Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
1686d7f5d3SJohn Marino */
1786d7f5d3SJohn Marino
1886d7f5d3SJohn Marino #include "dmlib.h"
1986d7f5d3SJohn Marino
2086d7f5d3SJohn Marino /* FIXME: thread unsafe */
2186d7f5d3SJohn Marino static DM_LIST_INIT(_dm_pools);
2286d7f5d3SJohn Marino void dm_pools_check_leaks(void);
2386d7f5d3SJohn Marino
2486d7f5d3SJohn Marino #ifdef DEBUG_POOL
2586d7f5d3SJohn Marino #include "pool-debug.c"
2686d7f5d3SJohn Marino #else
2786d7f5d3SJohn Marino #include "pool-fast.c"
2886d7f5d3SJohn Marino #endif
2986d7f5d3SJohn Marino
dm_pool_strdup(struct dm_pool * p,const char * str)3086d7f5d3SJohn Marino char *dm_pool_strdup(struct dm_pool *p, const char *str)
3186d7f5d3SJohn Marino {
3286d7f5d3SJohn Marino char *ret = dm_pool_alloc(p, strlen(str) + 1);
3386d7f5d3SJohn Marino
3486d7f5d3SJohn Marino if (ret)
3586d7f5d3SJohn Marino strcpy(ret, str);
3686d7f5d3SJohn Marino
3786d7f5d3SJohn Marino return ret;
3886d7f5d3SJohn Marino }
3986d7f5d3SJohn Marino
dm_pool_strndup(struct dm_pool * p,const char * str,size_t n)4086d7f5d3SJohn Marino char *dm_pool_strndup(struct dm_pool *p, const char *str, size_t n)
4186d7f5d3SJohn Marino {
4286d7f5d3SJohn Marino char *ret = dm_pool_alloc(p, n + 1);
4386d7f5d3SJohn Marino
4486d7f5d3SJohn Marino if (ret) {
4586d7f5d3SJohn Marino strncpy(ret, str, n);
4686d7f5d3SJohn Marino ret[n] = '\0';
4786d7f5d3SJohn Marino }
4886d7f5d3SJohn Marino
4986d7f5d3SJohn Marino return ret;
5086d7f5d3SJohn Marino }
5186d7f5d3SJohn Marino
dm_pool_zalloc(struct dm_pool * p,size_t s)5286d7f5d3SJohn Marino void *dm_pool_zalloc(struct dm_pool *p, size_t s)
5386d7f5d3SJohn Marino {
5486d7f5d3SJohn Marino void *ptr = dm_pool_alloc(p, s);
5586d7f5d3SJohn Marino
5686d7f5d3SJohn Marino if (ptr)
5786d7f5d3SJohn Marino memset(ptr, 0, s);
5886d7f5d3SJohn Marino
5986d7f5d3SJohn Marino return ptr;
6086d7f5d3SJohn Marino }
6186d7f5d3SJohn Marino
dm_pools_check_leaks(void)6286d7f5d3SJohn Marino void dm_pools_check_leaks(void)
6386d7f5d3SJohn Marino {
6486d7f5d3SJohn Marino struct dm_pool *p;
6586d7f5d3SJohn Marino
6686d7f5d3SJohn Marino if (dm_list_empty(&_dm_pools))
6786d7f5d3SJohn Marino return;
6886d7f5d3SJohn Marino
6986d7f5d3SJohn Marino log_error("You have a memory leak (not released memory pool):");
7086d7f5d3SJohn Marino dm_list_iterate_items(p, &_dm_pools) {
7186d7f5d3SJohn Marino #ifdef DEBUG_POOL
7286d7f5d3SJohn Marino log_error(" [%p] %s (%u bytes)",
7386d7f5d3SJohn Marino p->orig_pool,
7486d7f5d3SJohn Marino p->name, p->stats.bytes);
7586d7f5d3SJohn Marino #else
7686d7f5d3SJohn Marino log_error(" [%p]", p);
7786d7f5d3SJohn Marino #endif
7886d7f5d3SJohn Marino }
7986d7f5d3SJohn Marino }
80