You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
suricata/src/util-pool-thread.c

459 lines
11 KiB
C

/* Copyright (C) 2013 Open Information Security Foundation
*
* You can copy, redistribute or modify this Program under the terms of
* the GNU General Public License version 2 as published by the Free
* Software Foundation.
*
* This program 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
* version 2 along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
* 02110-1301, USA.
*/
/**
* \defgroup utilpool Pool
*
* @{
*/
/**
* \file
*
* \author Victor Julien <victor@inliniac.net>
*
* Pool utility functions
*/
#include "suricata-common.h"
#include "util-pool.h"
#include "util-pool-thread.h"
#include "util-unittest.h"
#include "util-debug.h"
PoolThread *PoolThreadInit(int threads, uint32_t size, uint32_t prealloc_size, uint32_t elt_size, void *(*Alloc)(void), int (*Init)(void *, void *), void *InitData, void (*Cleanup)(void *), void (*Free)(void *))
{
PoolThread *pt = NULL;
int i;
if (threads <= 0) {
SCLogDebug("error");
goto error;
}
pt = SCMalloc(sizeof(*pt));
if (unlikely(pt == NULL)) {
SCLogDebug("memory alloc error");
goto error;
}
SCLogDebug("size %d", threads);
pt->array = SCMalloc(threads * sizeof(PoolThreadElement));
if (pt->array == NULL) {
SCLogDebug("memory alloc error");
goto error;
}
pt->size = threads;
for (i = 0; i < threads; i++) {
PoolThreadElement *e = &pt->array[i];
SCMutexInit(&e->lock, NULL);
SCMutexLock(&e->lock);
// SCLogDebug("size %u prealloc_size %u elt_size %u Alloc %p Init %p InitData %p Cleanup %p Free %p",
// size, prealloc_size, elt_size,
// Alloc, Init, InitData, Cleanup, Free);
e->pool = PoolInit(size, prealloc_size, elt_size, Alloc, Init, InitData, Cleanup, Free);
SCMutexUnlock(&e->lock);
if (e->pool == NULL) {
SCLogDebug("error");
goto error;
}
}
return pt;
error:
if (pt != NULL)
PoolThreadFree(pt);
return NULL;
}
/**
*
*/
int PoolThreadGrow(PoolThread *pt, uint32_t size, uint32_t prealloc_size, uint32_t elt_size, void *(*Alloc)(void), int (*Init)(void *, void *), void *InitData, void (*Cleanup)(void *), void (*Free)(void *)) {
void *ptmp;
size_t newsize;
PoolThreadElement *e = NULL;
if (pt == NULL || pt->array == NULL) {
SCLogError(SC_ERR_POOL_INIT, "pool grow failed");
return -1;
}
newsize = pt->size + 1;
SCLogDebug("newsize %"PRIuMAX, (uintmax_t)newsize);
ptmp = SCRealloc(pt->array, (newsize * sizeof(PoolThreadElement)));
if (ptmp == NULL) {
SCFree(pt->array);
pt->array = NULL;
SCLogError(SC_ERR_POOL_INIT, "pool grow failed");
return -1;
}
pt->array = ptmp;
pt->size = newsize;
e = &pt->array[newsize - 1];
memset(e, 0x00, sizeof(*e));
SCMutexInit(&e->lock, NULL);
SCMutexLock(&e->lock);
e->pool = PoolInit(size, prealloc_size, elt_size, Alloc, Init, InitData, Cleanup, Free);
SCMutexUnlock(&e->lock);
if (e->pool == NULL) {
SCLogError(SC_ERR_POOL_INIT, "pool grow failed");
return -1;
}
return (int)(newsize - 1);
}
int PoolThreadSize(PoolThread *pt)
{
if (pt == NULL)
return -1;
return (int)pt->size;
}
void PoolThreadFree(PoolThread *pt)
{
int i;
if (pt == NULL)
return;
if (pt->array != NULL) {
for (i = 0; i < (int)pt->size; i++) {
PoolThreadElement *e = &pt->array[i];
SCMutexLock(&e->lock);
PoolFree(e->pool);
SCMutexUnlock(&e->lock);
SCMutexDestroy(&e->lock);
}
SCFree(pt->array);
}
SCFree(pt);
}
void *PoolThreadGetById(PoolThread *pt, uint16_t id)
{
void *data = NULL;
if (pt == NULL || id >= pt->size)
return NULL;
PoolThreadElement *e = &pt->array[id];
SCMutexLock(&e->lock);
data = PoolGet(e->pool);
SCMutexUnlock(&e->lock);
if (data) {
PoolThreadReserved *did = data;
*did = id;
}
return data;
}
void PoolThreadReturn(PoolThread *pt, void *data)
{
PoolThreadReserved *id = data;
if (pt == NULL || *id >= pt->size)
return;
SCLogDebug("returning to id %u", *id);
PoolThreadElement *e = &pt->array[*id];
SCMutexLock(&e->lock);
PoolReturn(e->pool, data);
SCMutexUnlock(&e->lock);
}
#ifdef UNITTESTS
struct PoolThreadTestData {
PoolThreadReserved res;
int abc;
};
static void *PoolThreadTestAlloc(void)
{
void *data = SCMalloc(sizeof(struct PoolThreadTestData));
return data;
}
static
int PoolThreadTestInit(void *data, void *allocdata)
{
if (!data)
return 0;
memset(data,0x00,sizeof(allocdata));
struct PoolThreadTestData *pdata = data;
pdata->abc = *(int *)allocdata;
return 1;
}
static
void PoolThreadTestFree(void *data)
{
}
static int PoolThreadTestInit01(void)
{
PoolThread *pt = PoolThreadInit(4, /* threads */
10, 5, 10, PoolThreadTestAlloc, NULL, NULL, NULL, NULL);
if (pt == NULL)
return 0;
PoolThreadFree(pt);
return 1;
}
static int PoolThreadTestInit02(void)
{
int i = 123;
PoolThread *pt = PoolThreadInit(4, /* threads */
10, 5, 10, PoolThreadTestAlloc, PoolThreadTestInit, &i, PoolThreadTestFree, NULL);
if (pt == NULL)
return 0;
PoolThreadFree(pt);
return 1;
}
static int PoolThreadTestGet01(void)
{
int result = 0;
PoolThread *pt = PoolThreadInit(4, /* threads */
10, 5, 10, PoolThreadTestAlloc, NULL, NULL, NULL, NULL);
if (pt == NULL)
return 0;
void *data = PoolThreadGetById(pt, 3);
if (data == NULL) {
printf("data == NULL: ");
goto end;
}
struct PoolThreadTestData *pdata = data;
if (pdata->res != 3) {
printf("res != 3, but %d: ", pdata->res);
goto end;
}
result = 1;
end:
PoolThreadFree(pt);
return result;
}
static int PoolThreadTestGet02(void)
{
int i = 123;
int result = 0;
PoolThread *pt = PoolThreadInit(4, /* threads */
10, 5, 10, PoolThreadTestAlloc, PoolThreadTestInit, &i, PoolThreadTestFree, NULL);
if (pt == NULL)
return 0;
void *data = PoolThreadGetById(pt, 3);
if (data == NULL) {
printf("data == NULL: ");
goto end;
}
struct PoolThreadTestData *pdata = data;
if (pdata->res != 3) {
printf("res != 3, but %d: ", pdata->res);
goto end;
}
if (pdata->abc != 123) {
printf("abc != 123, but %d: ", pdata->abc);
goto end;
}
result = 1;
end:
PoolThreadFree(pt);
return result;
}
static int PoolThreadTestReturn01(void)
{
int i = 123;
int result = 0;
PoolThread *pt = PoolThreadInit(4, /* threads */
10, 5, 10, PoolThreadTestAlloc, PoolThreadTestInit, &i, PoolThreadTestFree, NULL);
if (pt == NULL)
return 0;
void *data = PoolThreadGetById(pt, 3);
if (data == NULL) {
printf("data == NULL: ");
goto end;
}
struct PoolThreadTestData *pdata = data;
if (pdata->res != 3) {
printf("res != 3, but %d: ", pdata->res);
goto end;
}
if (pdata->abc != 123) {
printf("abc != 123, but %d: ", pdata->abc);
goto end;
}
if (pt->array[3].pool->outstanding != 1) {
printf("pool outstanding count wrong %u: ",
pt->array[3].pool->outstanding);
goto end;
}
PoolThreadReturn(pt, data);
if (pt->array[3].pool->outstanding != 0) {
printf("pool outstanding count wrong %u: ",
pt->array[3].pool->outstanding);
goto end;
}
result = 1;
end:
PoolThreadFree(pt);
return result;
}
static int PoolThreadTestGrow01(void)
{
PoolThread *pt = PoolThreadInit(4, /* threads */
10, 5, 10, PoolThreadTestAlloc, NULL, NULL, NULL, NULL);
if (pt == NULL)
return 0;
if (PoolThreadGrow(pt,
10, 5, 10, PoolThreadTestAlloc, NULL, NULL, NULL, NULL) < 0) {
PoolThreadFree(pt);
return 0;
}
PoolThreadFree(pt);
return 1;
}
static int PoolThreadTestGrow02(void)
{
int i = 123;
PoolThread *pt = PoolThreadInit(4, /* threads */
10, 5, 10, PoolThreadTestAlloc, PoolThreadTestInit, &i, PoolThreadTestFree, NULL);
if (pt == NULL)
return 0;
if (PoolThreadGrow(pt,
10, 5, 10, PoolThreadTestAlloc, PoolThreadTestInit, &i, PoolThreadTestFree, NULL) < 0) {
PoolThreadFree(pt);
return 0;
}
PoolThreadFree(pt);
return 1;
}
static int PoolThreadTestGrow03(void)
{
int i = 123;
int result = 0;
PoolThread *pt = PoolThreadInit(4, /* threads */
10, 5, 10, PoolThreadTestAlloc, PoolThreadTestInit, &i, PoolThreadTestFree, NULL);
if (pt == NULL)
return 0;
if (PoolThreadGrow(pt,
10, 5, 10, PoolThreadTestAlloc, PoolThreadTestInit, &i, PoolThreadTestFree, NULL) < 0) {
PoolThreadFree(pt);
return 0;
}
void *data = PoolThreadGetById(pt, 4);
if (data == NULL) {
printf("data == NULL: ");
goto end;
}
struct PoolThreadTestData *pdata = data;
if (pdata->res != 4) {
printf("res != 5, but %d: ", pdata->res);
goto end;
}
if (pdata->abc != 123) {
printf("abc != 123, but %d: ", pdata->abc);
goto end;
}
if (pt->array[4].pool->outstanding != 1) {
printf("pool outstanding count wrong %u: ",
pt->array[4].pool->outstanding);
goto end;
}
PoolThreadReturn(pt, data);
if (pt->array[4].pool->outstanding != 0) {
printf("pool outstanding count wrong %u: ",
pt->array[4].pool->outstanding);
goto end;
}
result = 1;
end:
PoolThreadFree(pt);
return result;
}
#endif
void PoolThreadRegisterTests(void)
{
#ifdef UNITTESTS
UtRegisterTest("PoolThreadTestInit01", PoolThreadTestInit01);
UtRegisterTest("PoolThreadTestInit02", PoolThreadTestInit02);
UtRegisterTest("PoolThreadTestGet01", PoolThreadTestGet01);
UtRegisterTest("PoolThreadTestGet02", PoolThreadTestGet02);
UtRegisterTest("PoolThreadTestReturn01", PoolThreadTestReturn01);
UtRegisterTest("PoolThreadTestGrow01", PoolThreadTestGrow01);
UtRegisterTest("PoolThreadTestGrow02", PoolThreadTestGrow02);
UtRegisterTest("PoolThreadTestGrow03", PoolThreadTestGrow03);
#endif
}
/**
* @}
*/