efl/src/modules/eina/mp/one_big/eina_one_big.c

323 lines
7.1 KiB
C

/* EINA - EFL data type library
* Copyright (C) 2010 Cedric BAIL, Vincent Torri
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library;
* if not, see <http://www.gnu.org/licenses/>.
*/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include "eina_config.h"
#include "eina_mempool.h"
#include "eina_trash.h"
#include "eina_inlist.h"
#include "eina_log.h"
#include "eina_lock.h"
#include "eina_thread.h"
#ifndef NVALGRIND
# include <memcheck.h>
#endif
#include "eina_private.h"
#ifdef INF
#undef INF
#endif
#define INF(...) EINA_LOG_DOM_INFO(_eina_mempool_log_dom, __VA_ARGS__)
#ifdef WRN
#undef WRN
#endif
#define WRN(...) EINA_LOG_DOM_WARN(_eina_one_big_mp_log_dom, __VA_ARGS__)
static int _eina_one_big_mp_log_dom = -1;
typedef struct _One_Big One_Big;
struct _One_Big
{
const char *name;
int item_size;
int usage;
int over;
int served;
int max;
unsigned char *base;
Eina_Trash *empty;
Eina_Inlist *over_list;
#ifdef EINA_HAVE_DEBUG_THREADS
Eina_Thread self;
#endif
Eina_Lock mutex;
};
static void *
eina_one_big_malloc(void *data, EINA_UNUSED unsigned int size)
{
One_Big *pool = data;
unsigned char *mem = NULL;
if (!eina_lock_take(&pool->mutex))
{
#ifdef EINA_HAVE_DEBUG_THREADS
assert(eina_thread_equal(pool->self, eina_thread_self()));
#endif
}
if (pool->empty)
{
#ifndef NVALGRIND
VALGRIND_MAKE_MEM_DEFINED(pool->empty, pool->item_size);
#endif
mem = eina_trash_pop(&pool->empty);
pool->usage++;
goto on_exit;
}
if (!pool->base)
{
pool->base = malloc(pool->item_size * pool->max);
if (!pool->base) goto retry_smaller;
#ifndef NVALGRIND
VALGRIND_MAKE_MEM_NOACCESS(pool->base, pool->item_size * pool->max);
#endif
}
if (pool->served < pool->max)
{
mem = pool->base + (pool->served++ *pool->item_size);
pool->usage++;
goto on_exit;
}
retry_smaller:
mem = malloc(sizeof(Eina_Inlist) + pool->item_size);
if (mem)
{
pool->over++;
/* Only need to zero list elements and not the payload here */
memset(mem, 0, sizeof(Eina_Inlist));
pool->over_list = eina_inlist_append(pool->over_list,
(Eina_Inlist *)mem);
mem = ((unsigned char *)mem) + sizeof(Eina_Inlist);
}
#ifndef NVALGRIND
VALGRIND_MAKE_MEM_NOACCESS(mem, pool->item_size);
#endif
on_exit:
eina_lock_release(&pool->mutex);
#ifndef NVALGRIND
VALGRIND_MEMPOOL_ALLOC(pool, mem, pool->item_size);
#endif
return mem;
}
static void
eina_one_big_free(void *data, void *ptr)
{
One_Big *pool = data;
if (!eina_lock_take(&pool->mutex))
{
#ifdef EINA_HAVE_DEBUG_THREADS
assert(eina_thread_equal(pool->self, eina_thread_self()));
#endif
}
if ((void *)pool->base <= ptr
&& ptr < (void *)(pool->base + (pool->max * pool->item_size)))
{
eina_trash_push(&pool->empty, ptr);
pool->usage--;
}
else
{
#ifndef NDEBUG
Eina_Inlist *it;
#endif
Eina_Inlist *il;
il = (Eina_Inlist *)(((unsigned char *)ptr) - sizeof(Eina_Inlist));
#ifndef NDEBUG
for (it = pool->over_list; it != NULL; it = it->next)
if (it == il) break;
assert(it != NULL);
#endif
pool->over_list = eina_inlist_remove(pool->over_list, il);
free(il);
pool->over--;
}
#ifndef NVALGRIND
VALGRIND_MEMPOOL_FREE(pool, ptr);
#endif
eina_lock_release(&pool->mutex);
}
static void *
eina_one_big_realloc(EINA_UNUSED void *data,
EINA_UNUSED void *element,
EINA_UNUSED unsigned int size)
{
return NULL;
}
static void *
eina_one_big_init(const char *context,
EINA_UNUSED const char *option,
va_list args)
{
One_Big *pool;
int item_size;
size_t length;
length = context ? strlen(context) + 1 : 0;
pool = calloc(1, sizeof (One_Big) + length);
if (!pool)
return NULL;
item_size = va_arg(args, int);
pool->item_size = eina_mempool_alignof(item_size);
pool->max = va_arg(args, int);
if (length)
{
pool->name = (const char *)(pool + 1);
memcpy((char *)pool->name, context, length);
}
#ifdef EINA_HAVE_DEBUG_THREADS
pool->self = eina_thread_self();
#endif
eina_lock_new(&pool->mutex);
#ifndef NVALGRIND
VALGRIND_CREATE_MEMPOOL(pool, 0, 1);
#endif
return pool;
}
static void
eina_one_big_shutdown(void *data)
{
One_Big *pool = data;
if (!pool) return;
if (!eina_lock_take(&pool->mutex))
{
#ifdef EINA_HAVE_DEBUG_THREADS
assert(eina_thread_equal(pool->self, eina_thread_self()));
#endif
}
if (pool->over > 0)
{
// FIXME: should we warn here? one_big mempool exceeded its alloc and now
// mempool is cleaning up the mess created. be quiet for now as we were before
// but edje seems to be a big offender at the moment! bad cedric! :)
// WRN(
// "Pool [%s] over by %i. cleaning up for you",
// pool->name, pool->over);
while (pool->over_list)
{
Eina_Inlist *il = pool->over_list;
pool->over_list = eina_inlist_remove(pool->over_list, il);
free(il);
pool->over--;
}
}
if (pool->over > 0)
{
WRN(
"Pool [%s] still over by %i\n",
pool->name, pool->over);
}
#ifndef NVALGRIND
VALGRIND_DESTROY_MEMPOOL(pool);
#endif
if (pool->base) free(pool->base);
eina_lock_release(&pool->mutex);
eina_lock_free(&pool->mutex);
free(pool);
}
static Eina_Mempool_Backend _eina_one_big_mp_backend = {
"one_big",
&eina_one_big_init,
&eina_one_big_free,
&eina_one_big_malloc,
&eina_one_big_realloc,
NULL,
NULL,
&eina_one_big_shutdown,
NULL
};
Eina_Bool one_big_init(void)
{
#ifdef DEBUG
_eina_one_big_mp_log_dom = eina_log_domain_register("eina_one_big_mempool",
EINA_LOG_COLOR_DEFAULT);
if (_eina_one_big_mp_log_dom < 0)
{
EINA_LOG_ERR("Could not register log domain: eina_one_big_mempool");
return EINA_FALSE;
}
#endif
return eina_mempool_register(&_eina_one_big_mp_backend);
}
void one_big_shutdown(void)
{
eina_mempool_unregister(&_eina_one_big_mp_backend);
#ifdef DEBUG
eina_log_domain_unregister(_eina_one_big_mp_log_dom);
_eina_one_big_mp_log_dom = -1;
#endif
}
#ifndef EINA_STATIC_BUILD_ONE_BIG
EINA_MODULE_INIT(one_big_init);
EINA_MODULE_SHUTDOWN(one_big_shutdown);
#endif /* ! EINA_STATIC_BUILD_ONE_BIG */