Home | History | Annotate | Line # | Download | only in libpthread
res_state.c revision 1.3 
      1 /*	$NetBSD: res_state.c,v 1.3 2004/05/24 01:20:17 christos Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 2004 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Christos Zoulas.
      9  *
     10  * Redistribution and use in source and binary forms, with or without
     11  * modification, are permitted provided that the following conditions
     12  * are met:
     13  * 1. Redistributions of source code must retain the above copyright
     14  *    notice, this list of conditions and the following disclaimer.
     15  * 2. Redistributions in binary form must reproduce the above copyright
     16  *    notice, this list of conditions and the following disclaimer in the
     17  *    documentation and/or other materials provided with the distribution.
     18  * 3. All advertising materials mentioning features or use of this software
     19  *    must display the following acknowledgement:
     20  *        This product includes software developed by the NetBSD
     21  *        Foundation, Inc. and its contributors.
     22  * 4. Neither the name of The NetBSD Foundation nor the names of its
     23  *    contributors may be used to endorse or promote products derived
     24  *    from this software without specific prior written permission.
     25  *
     26  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     27  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     28  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     29  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     30  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     31  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     32  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     33  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     34  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     35  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     36  * POSSIBILITY OF SUCH DAMAGE.
     37  */
     38 
     39 #include <sys/cdefs.h>
     40 #if defined(LIBC_SCCS) && !defined(lint)
     41 __RCSID("$NetBSD: res_state.c,v 1.3 2004/05/24 01:20:17 christos Exp $");
     42 #endif
     43 
     44 #include <sys/types.h>
     45 #include <sys/queue.h>
     46 #include <arpa/inet.h>
     47 #include <arpa/nameser.h>
     48 #include <stdlib.h>
     49 #include <unistd.h>
     50 #include <resolv.h>
     51 #include <netdb.h>
     52 
     53 #include "pthread.h"
     54 #include "pthread_int.h"
     55 
     56 #define	MY_LIST_HEAD(name, type)					\
     57 struct name {								\
     58 	union type *lh_first;	/* first element */			\
     59 }
     60 
     61 #define	MY_LIST_ENTRY(type)						\
     62 struct {								\
     63 	union type *le_next;	/* next element */			\
     64 	union type **le_prev;	/* address of previous next element */	\
     65 }
     66 
     67 static MY_LIST_HEAD(, _res_st) res_list = LIST_HEAD_INITIALIZER(&res_list);
     68 
     69 union _res_st {
     70 	MY_LIST_ENTRY(_res_st)	st_list;
     71 	struct __res_state	st_res;
     72 };
     73 
     74 static pthread_mutex_t res_mtx = PTHREAD_MUTEX_INITIALIZER;
     75 
     76 res_state __res_state(void);
     77 res_state __res_get_state(void);
     78 void __res_put_state(res_state);
     79 
     80 #ifdef RES_STATE_DEBUG
     81 static void
     82 res_state_debug(const char *msg, void *p)
     83 {
     84 	char buf[512];
     85 	pthread_t self = pthread_self();
     86 	int len = snprintf(buf, sizeof(buf), "%p: %s %p\n", self, msg, p);
     87 
     88 	(void)write(STDOUT_FILENO, buf, (size_t)len);
     89 }
     90 #else
     91 #define res_state_debug(a, b)
     92 #endif
     93 
     94 
     95 res_state
     96 __res_get_state(void)
     97 {
     98 	res_state res;
     99 	union _res_st *st;
    100 	pthread_mutex_lock(&res_mtx);
    101 	st = LIST_FIRST(&res_list);
    102 	if (st != NULL) {
    103 		res_state_debug("checkout from list", st);
    104 		LIST_REMOVE(st, st_list);
    105 		/* XXX: because we trash it with pointers from the list */
    106 		/* we should probably put the pointers elsewhere */
    107 		res = &st->st_res;
    108 		res->options = 0;
    109 	} else {
    110 		res = (res_state)malloc(sizeof(union _res_st));
    111 		if (res == NULL) {
    112 			pthread_mutex_unlock(&res_mtx);
    113 			h_errno = NETDB_INTERNAL;
    114 			return NULL;
    115 		}
    116 		res->options = 0;
    117 		res_state_debug("alloc new", res);
    118 	}
    119 	pthread_mutex_unlock(&res_mtx);
    120 	if ((res->options & RES_INIT) == 0 && res_ninit(res) == -1) {
    121 		h_errno = NETDB_INTERNAL;
    122 		__res_put_state(res);
    123 		return NULL;
    124 	}
    125 	return res;
    126 }
    127 
    128 void
    129 /*ARGSUSED*/
    130 __res_put_state(res_state res)
    131 {
    132 	pthread_mutex_lock(&res_mtx);
    133 	res_state_debug("free", res);
    134 	LIST_INSERT_HEAD(&res_list, (union _res_st *)(void *)res, st_list);
    135 	pthread_mutex_unlock(&res_mtx);
    136 }
    137 
    138 /*
    139  * This is aliased via a macro to _res; don't allow multi-threaded programs
    140  * to use it.
    141  */
    142 res_state
    143 __res_state(void)
    144 {
    145 	static const char res[] = "_res is not supported for multi-threaded"
    146 	    " programs.\n";
    147 	(void)write(STDERR_FILENO, res, sizeof(res) - 1);
    148 	abort();
    149 	return NULL;
    150 }
    151