irs_data.c revision 3822:831cb9e53b03 
1/*
2 * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
3 * Use is subject to license terms.
4 */
5
6/*
7 * Copyright (c) 1996,1999 by Internet Software Consortium.
8 *
9 * Permission to use, copy, modify, and distribute this software for any
10 * purpose with or without fee is hereby granted, provided that the above
11 * copyright notice and this permission notice appear in all copies.
12 *
13 * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS
14 * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
15 * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE
16 * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
17 * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
18 * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
19 * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
20 * SOFTWARE.
21 */
22
23#pragma ident	"%Z%%M%	%I%	%E% SMI"
24
25#if !defined(LINT) && !defined(CODECENTER)
26static const char rcsid[] = "$Id: irs_data.c,v 1.22 2003/06/20 07:09:33 marka Exp $";
27#endif
28
29#include "port_before.h"
30
31#ifndef __BIND_NOSTATIC
32
33#include <sys/types.h>
34
35#include <netinet/in.h>
36#include <arpa/nameser.h>
37
38#include <resolv.h>
39#include <stdio.h>
40#include <string.h>
41#include <isc/memcluster.h>
42#include <stdlib.h>
43
44#ifdef DO_PTHREADS
45#include <pthread.h>
46#endif
47
48#include <irs.h>
49
50#include "port_after.h"
51
52#include "irs_data.h"
53#undef _res
54#undef h_errno
55
56extern struct __res_state _res;
57extern int h_errno;
58
59#ifdef	DO_PTHREADS
60static pthread_key_t	key;
61static int		once = 0;
62#else
63static struct net_data	*net_data;
64#endif
65
66void
67irs_destroy(void) {
68#ifndef DO_PTHREADS
69	if (net_data != NULL)
70		net_data_destroy(net_data);
71	net_data = NULL;
72#endif
73}
74
75void
76net_data_destroy(void *p) {
77	struct net_data *net_data = p;
78
79	res_ndestroy(net_data->res);
80	if (net_data->gr != NULL) {
81		(*net_data->gr->close)(net_data->gr);
82		net_data->gr = NULL;
83	}
84	if (net_data->pw != NULL) {
85		(*net_data->pw->close)(net_data->pw);
86		net_data->pw = NULL;
87	}
88	if (net_data->sv != NULL) {
89		(*net_data->sv->close)(net_data->sv);
90		net_data->sv = NULL;
91	}
92	if (net_data->pr != NULL) {
93		(*net_data->pr->close)(net_data->pr);
94		net_data->pr = NULL;
95	}
96	if (net_data->ho != NULL) {
97		(*net_data->ho->close)(net_data->ho);
98		net_data->ho = NULL;
99	}
100	if (net_data->nw != NULL) {
101		(*net_data->nw->close)(net_data->nw);
102		net_data->nw = NULL;
103	}
104	if (net_data->ng != NULL) {
105		(*net_data->ng->close)(net_data->ng);
106		net_data->ng = NULL;
107	}
108	if (net_data->ho_data != NULL) {
109		free(net_data->ho_data);
110		net_data->ho_data = NULL;
111	}
112	if (net_data->nw_data != NULL) {
113		free(net_data->nw_data);
114		net_data->nw_data = NULL;
115	}
116
117	(*net_data->irs->close)(net_data->irs);
118	memput(net_data, sizeof *net_data);
119}
120
121/* applications that need a specific config file other than
122 * _PATH_IRS_CONF should call net_data_init directly rather than letting
123 *   the various wrapper functions make the first call. - brister
124 */
125
126struct net_data *
127net_data_init(const char *conf_file) {
128#ifdef	DO_PTHREADS
129	static pthread_mutex_t keylock = PTHREAD_MUTEX_INITIALIZER;
130	struct net_data *net_data;
131
132	if (!once) {
133		if (pthread_mutex_lock(&keylock) != 0)
134			return (NULL);
135		if (!once) {
136			if (pthread_key_create(&key, net_data_destroy) != 0) {
137				pthread_mutex_unlock(&keylock);
138				return (NULL);
139			}
140			once = 1;
141		}
142		if (pthread_mutex_unlock(&keylock) != 0)
143			return (NULL);
144	}
145	net_data = pthread_getspecific(key);
146#endif
147
148	if (net_data == NULL) {
149		net_data = net_data_create(conf_file);
150		if (net_data == NULL)
151			return (NULL);
152#ifdef	DO_PTHREADS
153		if (pthread_setspecific(key, net_data) != 0) {
154			net_data_destroy(net_data);
155			return (NULL);
156		}
157#endif
158	}
159
160	return (net_data);
161}
162
163struct net_data *
164net_data_create(const char *conf_file) {
165	struct net_data *net_data;
166
167	net_data = memget(sizeof (struct net_data));
168	if (net_data == NULL)
169		return (NULL);
170	memset(net_data, 0, sizeof (struct net_data));
171
172	if ((net_data->irs = irs_gen_acc("", conf_file)) == NULL) {
173		memput(net_data, sizeof (struct net_data));
174		return (NULL);
175	}
176#ifndef DO_PTHREADS
177	(*net_data->irs->res_set)(net_data->irs, &_res, NULL);
178#endif
179
180	net_data->res = (*net_data->irs->res_get)(net_data->irs);
181	if (net_data->res == NULL) {
182		(*net_data->irs->close)(net_data->irs);
183		memput(net_data, sizeof (struct net_data));
184		return (NULL);
185	}
186
187	if ((net_data->res->options & RES_INIT) == 0 &&
188	    res_ninit(net_data->res) == -1) {
189		(*net_data->irs->close)(net_data->irs);
190		memput(net_data, sizeof (struct net_data));
191		return (NULL);
192	}
193
194	return (net_data);
195}
196
197
198
199void
200net_data_minimize(struct net_data *net_data) {
201	res_nclose(net_data->res);
202}
203
204#ifdef _REENTRANT
205struct __res_state *
206__res_state(void) {
207	/* NULL param here means use the default config file. */
208	struct net_data *net_data = net_data_init(NULL);
209	if (net_data && net_data->res)
210		return (net_data->res);
211
212	return (&_res);
213}
214#endif
215
216int *
217__h_errno(void) {
218	/* NULL param here means use the default config file. */
219	struct net_data *net_data = net_data_init(NULL);
220	if (net_data && net_data->res)
221		return (&net_data->res->res_h_errno);
222	return (&h_errno);
223}
224
225void
226__h_errno_set(struct __res_state *res, int err) {
227
228	h_errno = res->res_h_errno = err;
229}
230
231#endif /*__BIND_NOSTATIC*/
232