2 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
4 * Copyright 1998-2007 The OpenLDAP Foundation.
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted only as authorized by the OpenLDAP
11 * A copy of this license is available in file LICENSE in the
12 * top-level directory of the distribution or, alternatively, at
13 * <http://www.OpenLDAP.org/license.html>.
20 #include <ac/stdarg.h>
21 #include <ac/stdlib.h>
22 #include <ac/string.h>
27 #include "ldap_pvt_thread.h" /* Get the thread interface */
28 #include "ldap_queue.h"
29 #define LDAP_THREAD_POOL_IMPLEMENTATION
30 #include "ldap_thr_debug.h" /* May rename symbols defined below */
32 #ifndef LDAP_THREAD_HAVE_TPOOL
34 typedef enum ldap_int_thread_pool_state_e {
35 LDAP_INT_THREAD_POOL_RUNNING,
36 LDAP_INT_THREAD_POOL_FINISHING,
37 LDAP_INT_THREAD_POOL_STOPPING
38 } ldap_int_thread_pool_state_t;
40 typedef struct ldap_int_thread_key_s {
43 ldap_pvt_thread_pool_keyfree_t *ltk_free;
44 } ldap_int_thread_key_t;
46 /* Max number of thread-specific keys we store per thread.
47 * We don't expect to use many...
50 #define LDAP_MAXTHR 1024 /* must be a power of 2 */
52 typedef struct ldap_int_thread_userctx_s {
53 ldap_pvt_thread_t ltu_id;
54 ldap_int_thread_key_t ltu_key[MAXKEYS];
55 } ldap_int_thread_userctx_t;
57 static ldap_pvt_thread_t tid_zero;
61 ldap_int_thread_userctx_t *ctx;
62 } thread_keys[LDAP_MAXTHR];
65 typedef struct ldap_int_thread_ctx_s {
67 LDAP_STAILQ_ENTRY(ldap_int_thread_ctx_s) q;
68 LDAP_SLIST_ENTRY(ldap_int_thread_ctx_s) l;
69 LDAP_SLIST_ENTRY(ldap_int_thread_ctx_s) al;
71 ldap_pvt_thread_start_t *ltc_start_routine;
73 } ldap_int_thread_ctx_t;
75 struct ldap_int_thread_pool_s {
76 LDAP_STAILQ_ENTRY(ldap_int_thread_pool_s) ltp_next;
77 ldap_pvt_thread_mutex_t ltp_mutex;
78 ldap_pvt_thread_cond_t ltp_cond;
79 ldap_pvt_thread_cond_t ltp_pcond;
80 LDAP_STAILQ_HEAD(tcq, ldap_int_thread_ctx_s) ltp_pending_list;
81 LDAP_SLIST_HEAD(tcl, ldap_int_thread_ctx_s) ltp_free_list;
82 LDAP_SLIST_HEAD(tclq, ldap_int_thread_ctx_s) ltp_active_list;
83 ldap_int_thread_pool_state_t ltp_state;
87 long ltp_pending_count;
88 long ltp_active_count;
93 static LDAP_STAILQ_HEAD(tpq, ldap_int_thread_pool_s)
94 ldap_int_thread_pool_list =
95 LDAP_STAILQ_HEAD_INITIALIZER(ldap_int_thread_pool_list);
97 static ldap_pvt_thread_mutex_t ldap_pvt_thread_pool_mutex;
99 static void *ldap_int_thread_pool_wrapper( void *pool );
101 static ldap_pvt_thread_t ldap_int_main_tid;
103 static ldap_int_thread_userctx_t ldap_int_main_thrctx;
106 ldap_int_thread_pool_startup ( void )
108 ldap_int_main_tid = ldap_pvt_thread_self();
109 ldap_int_main_thrctx.ltu_id = ldap_int_main_tid;
111 return ldap_pvt_thread_mutex_init(&ldap_pvt_thread_pool_mutex);
115 ldap_int_thread_pool_shutdown ( void )
117 struct ldap_int_thread_pool_s *pool;
119 while ((pool = LDAP_STAILQ_FIRST(&ldap_int_thread_pool_list)) != NULL) {
120 (ldap_pvt_thread_pool_destroy)(&pool, 0); /* ignore thr_debug macro */
122 ldap_pvt_thread_mutex_destroy(&ldap_pvt_thread_pool_mutex);
126 typedef struct ldap_lazy_sem_t {
127 ldap_pvt_thread_mutex_t ls_mutex;
128 ldap_pvt_thread_cond_t ls_cond;
131 * when more than ls_lazy_count number of resources
132 * becmoes available, the thread wating for the resources will
133 * be waken up in order to prevent frequent blocking/waking-up
135 unsigned int ls_lazy_count;
137 * only one thread(listener) will wait on this semaphore
138 * using a flag instead of a list
143 ldap_lazy_sem_t* thread_pool_sem = NULL;
146 ldap_lazy_sem_init( unsigned int value, unsigned int lazyness )
148 thread_pool_sem = (ldap_lazy_sem_t*) LDAP_CALLOC(1,
149 sizeof( ldap_lazy_sem_t ));
151 if( thread_pool_sem == NULL ) return -1;
153 ldap_pvt_thread_mutex_init( &thread_pool_sem->ls_mutex );
154 ldap_pvt_thread_cond_init( &thread_pool_sem->ls_cond );
155 thread_pool_sem->ls_sem_value = value;
156 thread_pool_sem->ls_lazy_count = lazyness;
157 thread_pool_sem->ls_wait = 0;
162 /* FIXME: move to some approprite header */
163 int ldap_lazy_sem_dec( ldap_lazy_sem_t* ls );
164 int ldap_lazy_sem_wait ( ldap_lazy_sem_t* ls );
167 * ldap_lazy_sem_wait is used if a caller is blockable(listener).
168 * Otherwise use ldap_lazy_sem_dec (worker)
171 ldap_lazy_sem_op_submit( ldap_lazy_sem_t* ls )
173 if ( ls == NULL ) return -1;
175 /* only worker thread has its thread ctx */
176 if ( ldap_pvt_thread_pool_context() ) {
178 return ldap_lazy_sem_dec( ls );
181 return ldap_lazy_sem_wait( ls );
186 * test if given semaphore's count is zero.
187 * If 0, the caller is blocked
188 * If not, the count is decremented.
191 ldap_lazy_sem_wait ( ldap_lazy_sem_t* ls )
193 ldap_pvt_thread_mutex_lock( &ls->ls_mutex );
196 if ( ls->ls_sem_value <= 0 ) {
197 /* no more avaliable resources */
199 ldap_pvt_thread_cond_wait( &ls->ls_cond, &ls->ls_mutex );
202 /* avaliable resources */
206 ldap_pvt_thread_mutex_unlock( &ls->ls_mutex );
212 * decrement the count without blocking
213 * even when the count becomes less than or equal to 0
216 ldap_lazy_sem_dec( ldap_lazy_sem_t* ls )
218 ldap_pvt_thread_mutex_lock( &ls->ls_mutex );
222 ldap_pvt_thread_mutex_unlock( &ls->ls_mutex );
228 * Increment the count by one and test if it is greater or
229 * equal to lazyness. If it is, wake up a blocked thread.
232 ldap_lazy_sem_post( ldap_lazy_sem_t* ls )
234 if( ls == NULL ) return (-1);
236 ldap_pvt_thread_mutex_lock( &ls->ls_mutex );
240 if ( ls->ls_sem_value >= ls->ls_lazy_count ) {
242 ldap_pvt_thread_cond_signal( &ls->ls_cond );
246 ldap_pvt_thread_mutex_unlock( &ls->ls_mutex );
252 ldap_pvt_thread_pool_init (
253 ldap_pvt_thread_pool_t *tpool,
257 ldap_pvt_thread_pool_t pool;
261 pool = (ldap_pvt_thread_pool_t) LDAP_CALLOC(1,
262 sizeof(struct ldap_int_thread_pool_s));
264 if (pool == NULL) return(-1);
266 rc = ldap_pvt_thread_mutex_init(&pool->ltp_mutex);
269 rc = ldap_pvt_thread_cond_init(&pool->ltp_cond);
272 rc = ldap_pvt_thread_cond_init(&pool->ltp_pcond);
275 pool->ltp_state = LDAP_INT_THREAD_POOL_RUNNING;
276 pool->ltp_max_count = max_threads;
277 pool->ltp_max_pending = max_pending;
278 LDAP_STAILQ_INIT(&pool->ltp_pending_list);
279 LDAP_SLIST_INIT(&pool->ltp_free_list);
280 LDAP_SLIST_INIT(&pool->ltp_active_list);
281 ldap_pvt_thread_mutex_lock(&ldap_pvt_thread_pool_mutex);
282 LDAP_STAILQ_INSERT_TAIL(&ldap_int_thread_pool_list, pool, ltp_next);
283 ldap_pvt_thread_mutex_unlock(&ldap_pvt_thread_pool_mutex);
286 /* THIS WILL NOT WORK on some systems. If the process
287 * forks after starting a thread, there is no guarantee
288 * that the thread will survive the fork. For example,
289 * slapd forks in order to daemonize, and does so after
290 * calling ldap_pvt_thread_pool_init. On some systems,
291 * this initial thread does not run in the child process,
292 * but ltp_open_count == 1, so two things happen:
293 * 1) the first client connection fails, and 2) when
294 * slapd is kill'ed, it never terminates since it waits
295 * for all worker threads to exit. */
297 /* start up one thread, just so there is one. no need to
298 * lock the mutex right now, since no threads are running.
300 pool->ltp_open_count++;
302 ldap_pvt_thread_t thr;
303 rc = ldap_pvt_thread_create( &thr, 1, ldap_int_thread_pool_wrapper, pool );
306 /* couldn't start one? then don't start any */
307 ldap_pvt_thread_mutex_lock(&ldap_pvt_thread_pool_mutex);
308 LDAP_STAILQ_REMOVE(ldap_int_thread_pool_list, pool,
309 ldap_int_thread_pool_s, ltp_next);
310 ldap_pvt_thread_mutex_unlock(&ldap_pvt_thread_pool_mutex);
311 ldap_pvt_thread_cond_destroy(&pool->ltp_pcond);
312 ldap_pvt_thread_cond_destroy(&pool->ltp_cond);
313 ldap_pvt_thread_mutex_destroy(&pool->ltp_mutex);
323 #define TID_HASH(tid, hash) do { unsigned i; \
324 unsigned char *ptr = (unsigned char *)&(tid); \
325 for (i=0, hash=0; i<sizeof(tid); i++) hash += ptr[i]; } while(0)
328 ldap_pvt_thread_pool_submit (
329 ldap_pvt_thread_pool_t *tpool,
330 ldap_pvt_thread_start_t *start_routine, void *arg )
332 struct ldap_int_thread_pool_s *pool;
333 ldap_int_thread_ctx_t *ctx;
335 ldap_pvt_thread_t thr;
345 ldap_pvt_thread_mutex_lock(&pool->ltp_mutex);
346 if (pool->ltp_state != LDAP_INT_THREAD_POOL_RUNNING
347 || (pool->ltp_max_pending > 0
348 && pool->ltp_pending_count >= pool->ltp_max_pending))
350 ldap_pvt_thread_mutex_unlock(&pool->ltp_mutex);
354 ctx = LDAP_SLIST_FIRST(&pool->ltp_free_list);
356 LDAP_SLIST_REMOVE_HEAD(&pool->ltp_free_list, ltc_next.l);
358 ctx = (ldap_int_thread_ctx_t *) LDAP_MALLOC(
359 sizeof(ldap_int_thread_ctx_t));
361 ldap_pvt_thread_mutex_unlock(&pool->ltp_mutex);
366 ctx->ltc_start_routine = start_routine;
369 pool->ltp_pending_count++;
370 LDAP_STAILQ_INSERT_TAIL(&pool->ltp_pending_list, ctx, ltc_next.q);
371 if (pool->ltp_pause) {
372 ldap_pvt_thread_mutex_unlock(&pool->ltp_mutex);
375 ldap_pvt_thread_cond_signal(&pool->ltp_cond);
376 if (pool->ltp_open_count < pool->ltp_active_count + pool->ltp_pending_count
377 && (pool->ltp_open_count < pool->ltp_max_count ||
378 pool->ltp_max_count <= 0 ))
380 pool->ltp_open_count++;
381 pool->ltp_starting++;
384 ldap_pvt_thread_mutex_unlock(&pool->ltp_mutex);
386 #ifdef LDAP_PVT_THREAD_POOL_SEM_LOAD_CONTROL
387 ldap_lazy_sem_op_submit( thread_pool_sem );
393 ldap_pvt_thread_mutex_lock(&pool->ltp_mutex);
395 rc = ldap_pvt_thread_create( &thr, 1,
396 ldap_int_thread_pool_wrapper, pool );
399 pool->ltp_starting--;
401 /* assign this thread ID to a key slot; start
402 * at the thread ID itself (mod LDAP_MAXTHR) and
403 * look for an empty slot.
406 for (rc = hash & (LDAP_MAXTHR-1);
407 !ldap_pvt_thread_equal(thread_keys[rc].id, tid_zero);
408 rc = (rc+1) & (LDAP_MAXTHR-1));
409 thread_keys[rc].id = thr;
411 /* couldn't create thread. back out of
412 * ltp_open_count and check for even worse things.
414 pool->ltp_open_count--;
415 pool->ltp_starting--;
416 if (pool->ltp_open_count == 0) {
417 /* no open threads at all?!?
419 ldap_int_thread_ctx_t *ptr;
421 /* let pool_destroy know there are no more threads */
422 ldap_pvt_thread_cond_signal(&pool->ltp_cond);
424 LDAP_STAILQ_FOREACH(ptr, &pool->ltp_pending_list, ltc_next.q)
425 if (ptr == ctx) break;
427 /* no open threads, context not handled, so
428 * back out of ltp_pending_count, free the context,
431 LDAP_STAILQ_REMOVE(&pool->ltp_pending_list, ctx,
432 ldap_int_thread_ctx_s, ltc_next.q);
433 pool->ltp_pending_count--;
434 ldap_pvt_thread_mutex_unlock(&pool->ltp_mutex);
439 /* there is another open thread, so this
440 * context will be handled eventually.
441 * continue on and signal that the context
445 ldap_pvt_thread_mutex_unlock(&pool->ltp_mutex);
452 ldap_pvt_thread_pool_maxthreads ( ldap_pvt_thread_pool_t *tpool, int max_threads )
454 struct ldap_int_thread_pool_s *pool;
464 ldap_pvt_thread_mutex_lock(&pool->ltp_mutex);
465 pool->ltp_max_count = max_threads;
466 ldap_pvt_thread_mutex_unlock(&pool->ltp_mutex);
471 ldap_pvt_thread_pool_query ( ldap_pvt_thread_pool_t *tpool, ldap_pvt_thread_pool_param_t param, void *value )
473 struct ldap_int_thread_pool_s *pool;
476 if ( tpool == NULL || value == NULL ) {
482 if ( pool == NULL ) {
486 ldap_pvt_thread_mutex_lock(&pool->ltp_mutex);
488 case LDAP_PVT_THREAD_POOL_PARAM_MAX:
489 count = pool->ltp_max_count;
492 case LDAP_PVT_THREAD_POOL_PARAM_MAX_PENDING:
493 count = pool->ltp_max_pending;
496 case LDAP_PVT_THREAD_POOL_PARAM_OPEN:
497 count = pool->ltp_open_count;
500 case LDAP_PVT_THREAD_POOL_PARAM_STARTING:
501 count = pool->ltp_starting;
504 case LDAP_PVT_THREAD_POOL_PARAM_ACTIVE:
505 count = pool->ltp_active_count;
508 case LDAP_PVT_THREAD_POOL_PARAM_PENDING:
509 count = pool->ltp_pending_count;
512 case LDAP_PVT_THREAD_POOL_PARAM_BACKLOAD:
513 count = pool->ltp_pending_count + pool->ltp_active_count;
516 case LDAP_PVT_THREAD_POOL_PARAM_ACTIVE_MAX:
519 case LDAP_PVT_THREAD_POOL_PARAM_PENDING_MAX:
522 case LDAP_PVT_THREAD_POOL_PARAM_BACKLOAD_MAX:
525 case LDAP_PVT_THREAD_POOL_PARAM_STATE: {
528 ldap_int_thread_pool_state_t state;
530 { "running", LDAP_INT_THREAD_POOL_RUNNING },
531 { "finishing", LDAP_INT_THREAD_POOL_FINISHING },
532 { "stopping", LDAP_INT_THREAD_POOL_STOPPING },
537 if ( pool->ltp_pause ) {
538 *((char **)value) = "pausing";
540 for ( i = 0; str2state[ i ].name != NULL; i++ ) {
541 if ( str2state[ i ].state == pool->ltp_state ) {
545 *((char **)value) = str2state[ i ].name;
547 if ( *((char **)value) != NULL ) {
552 ldap_pvt_thread_mutex_unlock( &pool->ltp_mutex );
555 *((int *)value) = count;
558 return ( count == -1 ? -1 : 0 );
562 * wrapper for ldap_pvt_thread_pool_query(), left around
563 * for backwards compatibility
566 ldap_pvt_thread_pool_backload ( ldap_pvt_thread_pool_t *tpool )
570 rc = ldap_pvt_thread_pool_query( tpool,
571 LDAP_PVT_THREAD_POOL_PARAM_BACKLOAD, (void *)&count );
581 ldap_pvt_thread_pool_destroy ( ldap_pvt_thread_pool_t *tpool, int run_pending )
583 struct ldap_int_thread_pool_s *pool, *pptr;
584 ldap_int_thread_ctx_t *ctx;
591 if (pool == NULL) return(-1);
593 ldap_pvt_thread_mutex_lock(&ldap_pvt_thread_pool_mutex);
594 LDAP_STAILQ_FOREACH(pptr, &ldap_int_thread_pool_list, ltp_next)
595 if (pptr == pool) break;
597 LDAP_STAILQ_REMOVE(&ldap_int_thread_pool_list, pool,
598 ldap_int_thread_pool_s, ltp_next);
599 ldap_pvt_thread_mutex_unlock(&ldap_pvt_thread_pool_mutex);
601 if (pool != pptr) return(-1);
603 ldap_pvt_thread_mutex_lock(&pool->ltp_mutex);
604 pool->ltp_state = run_pending
605 ? LDAP_INT_THREAD_POOL_FINISHING
606 : LDAP_INT_THREAD_POOL_STOPPING;
608 while (pool->ltp_open_count) {
609 if (!pool->ltp_pause)
610 ldap_pvt_thread_cond_broadcast(&pool->ltp_cond);
611 ldap_pvt_thread_cond_wait(&pool->ltp_cond, &pool->ltp_mutex);
613 ldap_pvt_thread_mutex_unlock(&pool->ltp_mutex);
615 while ((ctx = LDAP_STAILQ_FIRST(&pool->ltp_pending_list)) != NULL)
617 LDAP_STAILQ_REMOVE_HEAD(&pool->ltp_pending_list, ltc_next.q);
621 while ((ctx = LDAP_SLIST_FIRST(&pool->ltp_free_list)) != NULL)
623 LDAP_SLIST_REMOVE_HEAD(&pool->ltp_free_list, ltc_next.l);
627 ldap_pvt_thread_cond_destroy(&pool->ltp_pcond);
628 ldap_pvt_thread_cond_destroy(&pool->ltp_cond);
629 ldap_pvt_thread_mutex_destroy(&pool->ltp_mutex);
631 #ifdef LDAP_PVT_THREAD_POOL_SEM_LOAD_CONTROL
632 if ( thread_pool_sem ) {
633 LDAP_FREE( thread_pool_sem );
640 ldap_int_thread_pool_wrapper (
643 struct ldap_int_thread_pool_s *pool = xpool;
644 ldap_int_thread_ctx_t *ctx;
645 ldap_int_thread_userctx_t uctx;
646 int i, keyslot, hash;
651 for ( i=0; i<MAXKEYS; i++ ) {
652 uctx.ltu_key[i].ltk_key = NULL;
655 uctx.ltu_id = ldap_pvt_thread_self();
657 ldap_pvt_thread_mutex_lock(&pool->ltp_mutex);
659 /* store pointer to our keys */
660 TID_HASH(uctx.ltu_id, hash);
661 for (i = hash & (LDAP_MAXTHR-1);
662 !ldap_pvt_thread_equal(thread_keys[i].id, uctx.ltu_id);
663 i = (i+1) & (LDAP_MAXTHR-1));
664 thread_keys[i].ctx = &uctx;
668 while (pool->ltp_pause)
669 ldap_pvt_thread_cond_wait(&pool->ltp_cond, &pool->ltp_mutex);
671 if (pool->ltp_state == LDAP_INT_THREAD_POOL_STOPPING)
674 ctx = LDAP_STAILQ_FIRST(&pool->ltp_pending_list);
676 if (pool->ltp_state == LDAP_INT_THREAD_POOL_FINISHING)
679 if (pool->ltp_max_count > 0
680 && pool->ltp_open_count > pool->ltp_max_count)
682 /* too many threads running (can happen if the
683 * maximum threads value is set during ongoing
684 * operation using ldap_pvt_thread_pool_maxthreads)
685 * so let this thread die.
690 /* we could check an idle timer here, and let the
691 * thread die if it has been inactive for a while.
692 * only die if there are other open threads (i.e.,
693 * always have at least one thread open). the check
694 * should be like this:
695 * if (pool->ltp_open_count > 1 && pool->ltp_starting == 0)
696 * check timer, leave thread (break;)
698 * Just use pthread_cond_timedwait if we want to
702 assert(pool->ltp_state == LDAP_INT_THREAD_POOL_RUNNING);
703 ldap_pvt_thread_cond_wait(&pool->ltp_cond, &pool->ltp_mutex);
707 LDAP_STAILQ_REMOVE_HEAD(&pool->ltp_pending_list, ltc_next.q);
708 pool->ltp_pending_count--;
710 LDAP_SLIST_INSERT_HEAD(&pool->ltp_active_list, ctx, ltc_next.al);
711 pool->ltp_active_count++;
712 ldap_pvt_thread_mutex_unlock(&pool->ltp_mutex);
714 ctx->ltc_start_routine(&uctx, ctx->ltc_arg);
716 #ifdef LDAP_PVT_THREAD_POOL_SEM_LOAD_CONTROL
717 ldap_lazy_sem_post( thread_pool_sem );
719 ldap_pvt_thread_mutex_lock(&pool->ltp_mutex);
720 LDAP_SLIST_REMOVE(&pool->ltp_active_list, ctx,
721 ldap_int_thread_ctx_s, ltc_next.al);
722 LDAP_SLIST_INSERT_HEAD(&pool->ltp_free_list, ctx, ltc_next.l);
723 pool->ltp_active_count--;
725 /* let pool_pause know when it is the sole active thread */
726 if (pool->ltp_active_count < 2)
727 ldap_pvt_thread_cond_signal(&pool->ltp_pcond);
730 ldap_pvt_thread_pool_context_reset(&uctx);
732 thread_keys[keyslot].ctx = NULL;
733 thread_keys[keyslot].id = tid_zero;
735 pool->ltp_open_count--;
737 /* let pool_destroy know we're all done */
738 if (pool->ltp_open_count < 1)
739 ldap_pvt_thread_cond_signal(&pool->ltp_cond);
741 ldap_pvt_thread_mutex_unlock(&pool->ltp_mutex);
743 ldap_pvt_thread_exit(NULL);
748 ldap_pvt_thread_pool_pause (
749 ldap_pvt_thread_pool_t *tpool )
751 struct ldap_int_thread_pool_s *pool;
761 ldap_pvt_thread_mutex_lock(&pool->ltp_mutex);
763 /* If someone else has already requested a pause, we have to wait */
764 while (pool->ltp_pause) {
765 pool->ltp_pending_count++;
766 pool->ltp_active_count--;
767 /* let the other pool_pause() know when it can proceed */
768 if (pool->ltp_active_count < 2)
769 ldap_pvt_thread_cond_signal(&pool->ltp_pcond);
770 ldap_pvt_thread_cond_wait(&pool->ltp_cond, &pool->ltp_mutex);
771 pool->ltp_pending_count--;
772 pool->ltp_active_count++;
775 /* Wait for everyone else to pause or finish */
777 while (pool->ltp_active_count > 1) {
778 ldap_pvt_thread_cond_wait(&pool->ltp_pcond, &pool->ltp_mutex);
781 ldap_pvt_thread_mutex_unlock(&pool->ltp_mutex);
786 ldap_pvt_thread_pool_resume (
787 ldap_pvt_thread_pool_t *tpool )
789 struct ldap_int_thread_pool_s *pool;
799 ldap_pvt_thread_mutex_lock(&pool->ltp_mutex);
801 ldap_pvt_thread_cond_broadcast(&pool->ltp_cond);
802 ldap_pvt_thread_mutex_unlock(&pool->ltp_mutex);
806 int ldap_pvt_thread_pool_getkey(
810 ldap_pvt_thread_pool_keyfree_t **kfree )
812 ldap_int_thread_userctx_t *ctx = xctx;
815 if ( !ctx || !data ) return EINVAL;
817 for ( i=0; i<MAXKEYS && ctx->ltu_key[i].ltk_key; i++ ) {
818 if ( ctx->ltu_key[i].ltk_key == key ) {
819 *data = ctx->ltu_key[i].ltk_data;
820 if ( kfree ) *kfree = ctx->ltu_key[i].ltk_free;
827 int ldap_pvt_thread_pool_setkey(
831 ldap_pvt_thread_pool_keyfree_t *kfree )
833 ldap_int_thread_userctx_t *ctx = xctx;
836 if ( !ctx || !key ) return EINVAL;
838 for ( i=0; i<MAXKEYS; i++ ) {
839 if (( data && !ctx->ltu_key[i].ltk_key ) || ctx->ltu_key[i].ltk_key == key ) {
840 if ( data || kfree ) {
841 ctx->ltu_key[i].ltk_key = key;
842 ctx->ltu_key[i].ltk_data = data;
843 ctx->ltu_key[i].ltk_free = kfree;
846 for ( j=i+1; j<MAXKEYS; j++ )
847 if ( !ctx->ltu_key[j].ltk_key ) break;
850 ctx->ltu_key[i].ltk_key = ctx->ltu_key[j].ltk_key;
851 ctx->ltu_key[i].ltk_data = ctx->ltu_key[j].ltk_data;
852 ctx->ltu_key[i].ltk_free = ctx->ltu_key[j].ltk_free;
854 ctx->ltu_key[j].ltk_key = NULL;
855 ctx->ltu_key[j].ltk_data = NULL;
856 ctx->ltu_key[j].ltk_free = NULL;
864 /* Free all elements with this key, no matter which thread they're in.
865 * May only be called while the pool is paused.
867 void ldap_pvt_thread_pool_purgekey( void *key )
870 ldap_int_thread_userctx_t *ctx;
872 for ( i=0; i<LDAP_MAXTHR; i++ ) {
873 if ( thread_keys[i].ctx ) {
874 ctx = thread_keys[i].ctx;
875 for ( j=0; j<MAXKEYS; j++ ) {
876 if ( ctx->ltu_key[j].ltk_key == key ) {
877 if (ctx->ltu_key[j].ltk_free)
878 ctx->ltu_key[j].ltk_free( ctx->ltu_key[j].ltk_key,
879 ctx->ltu_key[j].ltk_data );
880 ctx->ltu_key[j].ltk_key = NULL;
881 ctx->ltu_key[j].ltk_free = NULL;
890 * This is necessary if the caller does not have access to the
891 * thread context handle (for example, a slapd plugin calling
892 * slapi_search_internal()). No doubt it is more efficient
893 * for the application to keep track of the thread context
896 void *ldap_pvt_thread_pool_context( )
898 ldap_pvt_thread_t tid;
901 tid = ldap_pvt_thread_self();
902 if ( ldap_pvt_thread_equal( tid, ldap_int_main_tid ))
903 return &ldap_int_main_thrctx;
905 TID_HASH( tid, hash );
906 for (i = hash & (LDAP_MAXTHR-1);
907 !ldap_pvt_thread_equal(thread_keys[i].id, tid_zero) &&
908 !ldap_pvt_thread_equal(thread_keys[i].id, tid);
909 i = (i+1) & (LDAP_MAXTHR-1));
911 return thread_keys[i].ctx;
914 void ldap_pvt_thread_pool_context_reset( void *vctx )
916 ldap_int_thread_userctx_t *ctx = vctx;
919 for ( i=MAXKEYS-1; i>=0; i--) {
920 if ( !ctx->ltu_key[i].ltk_key )
922 if ( ctx->ltu_key[i].ltk_free )
923 ctx->ltu_key[i].ltk_free( ctx->ltu_key[i].ltk_key,
924 ctx->ltu_key[i].ltk_data );
925 ctx->ltu_key[i].ltk_key = NULL;
929 ldap_pvt_thread_t ldap_pvt_thread_pool_tid( void *vctx )
931 ldap_int_thread_userctx_t *ctx = vctx;
935 #endif /* LDAP_THREAD_HAVE_TPOOL */