1: /*
2: * Copyright (C) 2004, 2005, 2007 Internet Systems Consortium, Inc. ("ISC")
3: * Copyright (C) 1999-2001 Internet Software Consortium.
4: *
5: * Permission to use, copy, modify, and/or distribute this software for any
6: * purpose with or without fee is hereby granted, provided that the above
7: * copyright notice and this permission notice appear in all copies.
8: *
9: * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
10: * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
11: * AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
12: * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
13: * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
14: * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
15: * PERFORMANCE OF THIS SOFTWARE.
16: */
17:
18: /* $Id: taskpool.c,v 1.1.1.1 2012/05/29 12:08:38 misho Exp $ */
19:
20: /*! \file */
21:
22: #include <config.h>
23:
24: #include <isc/mem.h>
25: #include <isc/taskpool.h>
26: #include <isc/util.h>
27:
28: /***
29: *** Types.
30: ***/
31:
32: struct isc_taskpool {
33: isc_mem_t * mctx;
34: unsigned int ntasks;
35: isc_task_t ** tasks;
36: };
37: /***
38: *** Functions.
39: ***/
40:
41: isc_result_t
42: isc_taskpool_create(isc_taskmgr_t *tmgr, isc_mem_t *mctx,
43: unsigned int ntasks, unsigned int quantum,
44: isc_taskpool_t **poolp)
45: {
46: unsigned int i;
47: isc_taskpool_t *pool;
48: isc_result_t result;
49:
50: INSIST(ntasks > 0);
51: pool = isc_mem_get(mctx, sizeof(*pool));
52: if (pool == NULL)
53: return (ISC_R_NOMEMORY);
54: pool->mctx = mctx;
55: pool->ntasks = ntasks;
56: pool->tasks = isc_mem_get(mctx, ntasks * sizeof(isc_task_t *));
57: if (pool->tasks == NULL) {
58: isc_mem_put(mctx, pool, sizeof(*pool));
59: return (ISC_R_NOMEMORY);
60: }
61: for (i = 0; i < ntasks; i++)
62: pool->tasks[i] = NULL;
63: for (i = 0; i < ntasks; i++) {
64: result = isc_task_create(tmgr, quantum, &pool->tasks[i]);
65: if (result != ISC_R_SUCCESS) {
66: isc_taskpool_destroy(&pool);
67: return (result);
68: }
69: isc_task_setname(pool->tasks[i], "taskpool", NULL);
70: }
71: *poolp = pool;
72: return (ISC_R_SUCCESS);
73: }
74:
75: void isc_taskpool_gettask(isc_taskpool_t *pool, unsigned int hash,
76: isc_task_t **targetp)
77: {
78: isc_task_attach(pool->tasks[hash % pool->ntasks], targetp);
79: }
80:
81: void
82: isc_taskpool_destroy(isc_taskpool_t **poolp) {
83: unsigned int i;
84: isc_taskpool_t *pool = *poolp;
85: for (i = 0; i < pool->ntasks; i++) {
86: if (pool->tasks[i] != NULL) {
87: isc_task_detach(&pool->tasks[i]);
88: }
89: }
90: isc_mem_put(pool->mctx, pool->tasks,
91: pool->ntasks * sizeof(isc_task_t *));
92: isc_mem_put(pool->mctx, pool, sizeof(*pool));
93: *poolp = NULL;
94: }
95:
96:
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>