1/*-
2 * Copyright (c) 1994, Garrett Wollman
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 *    notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 *    notice, this list of conditions and the following disclaimer in the
11 *    documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY THE CONTRIBUTORS ``AS IS'' AND
14 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
16 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
17 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
18 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
19 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
20 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
21 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
22 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
23 * SUCH DAMAGE.
24 */
25
26#include <sys/cdefs.h>
27__FBSDID("$FreeBSD$");
28
29#include "namespace.h"
30#include "reentrant.h"
31#include <sys/param.h>
32#include <sys/socket.h>
33#include <netinet/in.h>
34#include <arpa/inet.h>
35#include <netdb.h>
36#include <stdio.h>
37#include <ctype.h>
38#include <errno.h>
39#include <stdlib.h>
40#include <string.h>
41#include <stdarg.h>
42#include <nsswitch.h>
43#include "un-namespace.h"
44#include "netdb_private.h"
45#ifdef NS_CACHING
46#include "nscache.h"
47#endif
48
49extern int _ht_getnetbyname(void *, void *, va_list);
50extern int _dns_getnetbyname(void *, void *, va_list);
51extern int _nis_getnetbyname(void *, void *, va_list);
52extern int _ht_getnetbyaddr(void *, void *, va_list);
53extern int _dns_getnetbyaddr(void *, void *, va_list);
54extern int _nis_getnetbyaddr(void *, void *, va_list);
55
56/* Network lookup order if nsswitch.conf is broken or nonexistant */
57static const ns_src default_src[] = {
58	{ NSSRC_FILES, NS_SUCCESS },
59	{ NSSRC_DNS, NS_SUCCESS },
60	{ 0 }
61};
62
63NETDB_THREAD_ALLOC(netent_data)
64NETDB_THREAD_ALLOC(netdata)
65
66#ifdef NS_CACHING
67static int
68net_id_func(char *buffer, size_t *buffer_size, va_list ap, void *cache_mdata)
69{
70	char *name;
71	uint32_t net;
72	int type;
73
74	size_t desired_size, size;
75	enum nss_lookup_type lookup_type;
76	int res = NS_UNAVAIL;
77
78	lookup_type = (enum nss_lookup_type)cache_mdata;
79	switch (lookup_type) {
80	case nss_lt_name:
81		name = va_arg(ap, char *);
82
83		size = strlen(name);
84		desired_size = sizeof(enum nss_lookup_type) + size + 1;
85		if (desired_size > *buffer_size) {
86			res = NS_RETURN;
87			goto fin;
88		}
89
90		memcpy(buffer, &lookup_type, sizeof(enum nss_lookup_type));
91		memcpy(buffer + sizeof(enum nss_lookup_type), name, size + 1);
92
93		res = NS_SUCCESS;
94		break;
95	case nss_lt_id:
96		net = va_arg(ap, uint32_t);
97		type = va_arg(ap, int);
98
99		desired_size = sizeof(enum nss_lookup_type) +
100		    sizeof(uint32_t) + sizeof(int);
101		if (desired_size > *buffer_size) {
102			res = NS_RETURN;
103			goto fin;
104		}
105
106		memcpy(buffer, &lookup_type, sizeof(enum nss_lookup_type));
107		memcpy(buffer + sizeof(enum nss_lookup_type), &net,
108		    sizeof(uint32_t));
109		memcpy(buffer + sizeof(enum nss_lookup_type) + sizeof(uint32_t),
110		    &type, sizeof(int));
111
112		res = NS_SUCCESS;
113		break;
114	default:
115		/* should be unreachable */
116		return (NS_UNAVAIL);
117	}
118
119fin:
120	*buffer_size = desired_size;
121	return (res);
122}
123
124
125static int
126net_marshal_func(char *buffer, size_t *buffer_size, void *retval, va_list ap,
127    void *cache_mdata)
128{
129	char *name;
130	uint32_t net;
131	int type;
132	struct netent *ne;
133	char *orig_buf;
134	size_t orig_buf_size;
135
136	struct netent new_ne;
137	size_t desired_size, size, aliases_size;
138	char *p;
139	char **alias;
140
141	switch ((enum nss_lookup_type)cache_mdata) {
142	case nss_lt_name:
143		name = va_arg(ap, char *);
144		break;
145	case nss_lt_id:
146		net = va_arg(ap, uint32_t);
147		type = va_arg(ap, int);
148	break;
149	case nss_lt_all:
150		break;
151	default:
152		/* should be unreachable */
153		return (NS_UNAVAIL);
154	}
155
156	ne = va_arg(ap, struct netent *);
157	orig_buf = va_arg(ap, char *);
158	orig_buf_size = va_arg(ap, size_t);
159
160	desired_size = _ALIGNBYTES + sizeof(struct netent) + sizeof(char *);
161	if (ne->n_name != NULL)
162		desired_size += strlen(ne->n_name) + 1;
163
164	if (ne->n_aliases != NULL) {
165		aliases_size = 0;
166		for (alias = ne->n_aliases; *alias; ++alias) {
167			desired_size += strlen(*alias) + 1;
168			++aliases_size;
169		}
170
171		desired_size += _ALIGNBYTES +
172		    (aliases_size + 1) * sizeof(char *);
173	}
174
175	if (*buffer_size < desired_size) {
176		/* this assignment is here for future use */
177		*buffer_size = desired_size;
178		return (NS_RETURN);
179	}
180
181	memcpy(&new_ne, ne, sizeof(struct netent));
182
183	*buffer_size = desired_size;
184	memset(buffer, 0, desired_size);
185	p = buffer + sizeof(struct netent) + sizeof(char *);
186	memcpy(buffer + sizeof(struct netent), &p, sizeof(char *));
187	p = (char *)_ALIGN(p);
188
189	if (new_ne.n_name != NULL) {
190		size = strlen(new_ne.n_name);
191		memcpy(p, new_ne.n_name, size);
192		new_ne.n_name = p;
193		p += size + 1;
194	}
195
196	if (new_ne.n_aliases != NULL) {
197		p = (char *)_ALIGN(p);
198		memcpy(p, new_ne.n_aliases, sizeof(char *) * aliases_size);
199		new_ne.n_aliases = (char **)p;
200		p += sizeof(char *) * (aliases_size + 1);
201
202		for (alias = new_ne.n_aliases; *alias; ++alias) {
203			size = strlen(*alias);
204			memcpy(p, *alias, size);
205			*alias = p;
206			p += size + 1;
207		}
208	}
209
210	memcpy(buffer, &new_ne, sizeof(struct netent));
211	return (NS_SUCCESS);
212}
213
214static int
215net_unmarshal_func(char *buffer, size_t buffer_size, void *retval, va_list ap,
216    void *cache_mdata)
217{
218	char *name;
219	uint32_t net;
220	int type;
221	struct netent *ne;
222	char *orig_buf;
223	size_t orig_buf_size;
224	int *ret_errno;
225
226	char *p;
227	char **alias;
228
229	switch ((enum nss_lookup_type)cache_mdata) {
230	case nss_lt_name:
231		name = va_arg(ap, char *);
232		break;
233	case nss_lt_id:
234		net = va_arg(ap, uint32_t);
235		type = va_arg(ap, int);
236		break;
237	case nss_lt_all:
238		break;
239	default:
240		/* should be unreachable */
241		return (NS_UNAVAIL);
242	}
243
244	ne = va_arg(ap, struct netent *);
245	orig_buf = va_arg(ap, char *);
246	orig_buf_size = va_arg(ap, size_t);
247	ret_errno = va_arg(ap, int *);
248
249	if (orig_buf_size <
250	    buffer_size - sizeof(struct netent) - sizeof(char *)) {
251		*ret_errno = ERANGE;
252		return (NS_RETURN);
253	}
254
255	memcpy(ne, buffer, sizeof(struct netent));
256	memcpy(&p, buffer + sizeof(struct netent), sizeof(char *));
257
258	orig_buf = (char *)_ALIGN(orig_buf);
259	memcpy(orig_buf, buffer + sizeof(struct netent) + sizeof(char *) +
260	    _ALIGN(p) - (size_t)p,
261	    buffer_size - sizeof(struct netent) - sizeof(char *) -
262	    _ALIGN(p) + (size_t)p);
263	p = (char *)_ALIGN(p);
264
265	NS_APPLY_OFFSET(ne->n_name, orig_buf, p, char *);
266	if (ne->n_aliases != NULL) {
267		NS_APPLY_OFFSET(ne->n_aliases, orig_buf, p, char **);
268
269		for (alias = ne->n_aliases; *alias; ++alias)
270			NS_APPLY_OFFSET(*alias, orig_buf, p, char *);
271	}
272
273	if (retval != NULL)
274		*((struct netent **)retval) = ne;
275
276	return (NS_SUCCESS);
277}
278#endif /* NS_CACHING */
279
280static void
281netent_data_free(void *ptr)
282{
283	struct netent_data *ned = ptr;
284
285	if (ned == NULL)
286		return;
287	ned->stayopen = 0;
288	_endnethtent(ned);
289	free(ned);
290}
291
292static void
293netdata_free(void *ptr)
294{
295	free(ptr);
296}
297
298int
299__copy_netent(struct netent *ne, struct netent *nptr, char *buf, size_t buflen)
300{
301	char *cp;
302	int i, n;
303	int numptr, len;
304
305	/* Find out the amount of space required to store the answer. */
306	numptr = 1; /* NULL ptr */
307	len = (char *)ALIGN(buf) - buf;
308	for (i = 0; ne->n_aliases[i]; i++, numptr++) {
309		len += strlen(ne->n_aliases[i]) + 1;
310	}
311	len += strlen(ne->n_name) + 1;
312	len += numptr * sizeof(char*);
313
314	if (len > (int)buflen) {
315		errno = ERANGE;
316		return (-1);
317	}
318
319	/* copy net value and type */
320	nptr->n_addrtype = ne->n_addrtype;
321	nptr->n_net = ne->n_net;
322
323	cp = (char *)ALIGN(buf) + numptr * sizeof(char *);
324
325	/* copy official name */
326	n = strlen(ne->n_name) + 1;
327	strcpy(cp, ne->n_name);
328	nptr->n_name = cp;
329	cp += n;
330
331	/* copy aliases */
332	nptr->n_aliases = (char **)ALIGN(buf);
333	for (i = 0 ; ne->n_aliases[i]; i++) {
334		n = strlen(ne->n_aliases[i]) + 1;
335		strcpy(cp, ne->n_aliases[i]);
336		nptr->n_aliases[i] = cp;
337		cp += n;
338	}
339	nptr->n_aliases[i] = NULL;
340
341	return (0);
342}
343
344int
345getnetbyname_r(const char *name, struct netent *ne, char *buffer,
346    size_t buflen, struct netent **result, int *h_errorp)
347{
348#ifdef NS_CACHING
349	static const nss_cache_info cache_info =
350    		NS_COMMON_CACHE_INFO_INITIALIZER(
351		networks, (void *)nss_lt_name,
352		net_id_func, net_marshal_func, net_unmarshal_func);
353#endif
354	static const ns_dtab dtab[] = {
355		NS_FILES_CB(_ht_getnetbyname, NULL)
356		{ NSSRC_DNS, _dns_getnetbyname, NULL },
357		NS_NIS_CB(_nis_getnetbyname, NULL) /* force -DHESIOD */
358#ifdef NS_CACHING
359		NS_CACHE_CB(&cache_info)
360#endif
361		{ 0 }
362	};
363	int rval, ret_errno = 0;
364
365	rval = _nsdispatch((void *)result, dtab, NSDB_NETWORKS,
366	    "getnetbyname_r", default_src, name, ne, buffer, buflen,
367	    &ret_errno, h_errorp);
368
369	if (rval != NS_SUCCESS) {
370		errno = ret_errno;
371		return ((ret_errno != 0) ? ret_errno : -1);
372	}
373	return (0);
374}
375
376int
377getnetbyaddr_r(uint32_t addr, int af, struct netent *ne, char *buffer,
378    size_t buflen, struct netent **result, int *h_errorp)
379{
380#ifdef NS_CACHING
381	static const nss_cache_info cache_info =
382    		NS_COMMON_CACHE_INFO_INITIALIZER(
383		networks, (void *)nss_lt_id,
384		net_id_func, net_marshal_func, net_unmarshal_func);
385#endif
386	static const ns_dtab dtab[] = {
387		NS_FILES_CB(_ht_getnetbyaddr, NULL)
388		{ NSSRC_DNS, _dns_getnetbyaddr, NULL },
389		NS_NIS_CB(_nis_getnetbyaddr, NULL) /* force -DHESIOD */
390#ifdef NS_CACHING
391		NS_CACHE_CB(&cache_info)
392#endif
393		{ 0 }
394	};
395	int rval, ret_errno = 0;
396
397	rval = _nsdispatch((void *)result, dtab, NSDB_NETWORKS,
398	    "getnetbyaddr_r", default_src, addr, af, ne, buffer, buflen,
399	    &ret_errno, h_errorp);
400
401	if (rval != NS_SUCCESS) {
402		errno = ret_errno;
403		return ((ret_errno != 0) ? ret_errno : -1);
404	}
405	return (0);
406}
407
408struct netent *
409getnetbyname(const char *name)
410{
411	struct netdata *nd;
412	struct netent *rval;
413	int ret_h_errno;
414
415	if ((nd = __netdata_init()) == NULL)
416		return (NULL);
417	if (getnetbyname_r(name, &nd->net, nd->data, sizeof(nd->data), &rval,
418	    &ret_h_errno) != 0)
419		return (NULL);
420	return (rval);
421}
422
423struct netent *
424getnetbyaddr(uint32_t addr, int af)
425{
426	struct netdata *nd;
427	struct netent *rval;
428	int ret_h_errno;
429
430	if ((nd = __netdata_init()) == NULL)
431		return (NULL);
432	if (getnetbyaddr_r(addr, af, &nd->net, nd->data, sizeof(nd->data),
433	    &rval, &ret_h_errno) != 0)
434		return (NULL);
435	return (rval);
436}
437
438void
439setnetent(int stayopen)
440{
441	struct netent_data *ned;
442
443	if ((ned = __netent_data_init()) == NULL)
444		return;
445	_setnethtent(stayopen, ned);
446	_setnetdnsent(stayopen);
447}
448
449void
450endnetent(void)
451{
452	struct netent_data *ned;
453
454	if ((ned = __netent_data_init()) == NULL)
455		return;
456	_endnethtent(ned);
457	_endnetdnsent();
458}
459