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
49 /* Network lookup order if nsswitch.conf is broken or nonexistant */
50 static const ns_src default_src[] = {
51 { NSSRC_FILES, NS_SUCCESS },
52 { NSSRC_DNS, NS_SUCCESS },
53 { 0 }
54 };
55
56 NETDB_THREAD_ALLOC(netent_data)
NETDB_THREAD_ALLOC(netdata)57 NETDB_THREAD_ALLOC(netdata)
58
59 #ifdef NS_CACHING
60 static int
61 net_id_func(char *buffer, size_t *buffer_size, va_list ap, void *cache_mdata)
62 {
63 char *name;
64 uint32_t net;
65 int type;
66
67 size_t desired_size, size;
68 enum nss_lookup_type lookup_type;
69 int res = NS_UNAVAIL;
70
71 lookup_type = (enum nss_lookup_type)cache_mdata;
72 switch (lookup_type) {
73 case nss_lt_name:
74 name = va_arg(ap, char *);
75
76 size = strlen(name);
77 desired_size = sizeof(enum nss_lookup_type) + size + 1;
78 if (desired_size > *buffer_size) {
79 res = NS_RETURN;
80 goto fin;
81 }
82
83 memcpy(buffer, &lookup_type, sizeof(enum nss_lookup_type));
84 memcpy(buffer + sizeof(enum nss_lookup_type), name, size + 1);
85
86 res = NS_SUCCESS;
87 break;
88 case nss_lt_id:
89 net = va_arg(ap, uint32_t);
90 type = va_arg(ap, int);
91
92 desired_size = sizeof(enum nss_lookup_type) +
93 sizeof(uint32_t) + sizeof(int);
94 if (desired_size > *buffer_size) {
95 res = NS_RETURN;
96 goto fin;
97 }
98
99 memcpy(buffer, &lookup_type, sizeof(enum nss_lookup_type));
100 memcpy(buffer + sizeof(enum nss_lookup_type), &net,
101 sizeof(uint32_t));
102 memcpy(buffer + sizeof(enum nss_lookup_type) + sizeof(uint32_t),
103 &type, sizeof(int));
104
105 res = NS_SUCCESS;
106 break;
107 default:
108 /* should be unreachable */
109 return (NS_UNAVAIL);
110 }
111
112 fin:
113 *buffer_size = desired_size;
114 return (res);
115 }
116
117
118 static int
net_marshal_func(char * buffer,size_t * buffer_size,void * retval,va_list ap,void * cache_mdata)119 net_marshal_func(char *buffer, size_t *buffer_size, void *retval, va_list ap,
120 void *cache_mdata)
121 {
122 char *name;
123 uint32_t net;
124 int type;
125 struct netent *ne;
126 char *orig_buf;
127 size_t orig_buf_size;
128
129 struct netent new_ne;
130 size_t desired_size, size, aliases_size;
131 char *p;
132 char **alias;
133
134 switch ((enum nss_lookup_type)cache_mdata) {
135 case nss_lt_name:
136 name = va_arg(ap, char *);
137 break;
138 case nss_lt_id:
139 net = va_arg(ap, uint32_t);
140 type = va_arg(ap, int);
141 break;
142 case nss_lt_all:
143 break;
144 default:
145 /* should be unreachable */
146 return (NS_UNAVAIL);
147 }
148
149 ne = va_arg(ap, struct netent *);
150 orig_buf = va_arg(ap, char *);
151 orig_buf_size = va_arg(ap, size_t);
152
153 desired_size = _ALIGNBYTES + sizeof(struct netent) + sizeof(char *);
154 if (ne->n_name != NULL)
155 desired_size += strlen(ne->n_name) + 1;
156
157 if (ne->n_aliases != NULL) {
158 aliases_size = 0;
159 for (alias = ne->n_aliases; *alias; ++alias) {
160 desired_size += strlen(*alias) + 1;
161 ++aliases_size;
162 }
163
164 desired_size += _ALIGNBYTES +
165 (aliases_size + 1) * sizeof(char *);
166 }
167
168 if (*buffer_size < desired_size) {
169 /* this assignment is here for future use */
170 *buffer_size = desired_size;
171 return (NS_RETURN);
172 }
173
174 memcpy(&new_ne, ne, sizeof(struct netent));
175
176 *buffer_size = desired_size;
177 memset(buffer, 0, desired_size);
178 p = buffer + sizeof(struct netent) + sizeof(char *);
179 memcpy(buffer + sizeof(struct netent), &p, sizeof(char *));
180 p = (char *)_ALIGN(p);
181
182 if (new_ne.n_name != NULL) {
183 size = strlen(new_ne.n_name);
184 memcpy(p, new_ne.n_name, size);
185 new_ne.n_name = p;
186 p += size + 1;
187 }
188
189 if (new_ne.n_aliases != NULL) {
190 p = (char *)_ALIGN(p);
191 memcpy(p, new_ne.n_aliases, sizeof(char *) * aliases_size);
192 new_ne.n_aliases = (char **)p;
193 p += sizeof(char *) * (aliases_size + 1);
194
195 for (alias = new_ne.n_aliases; *alias; ++alias) {
196 size = strlen(*alias);
197 memcpy(p, *alias, size);
198 *alias = p;
199 p += size + 1;
200 }
201 }
202
203 memcpy(buffer, &new_ne, sizeof(struct netent));
204 return (NS_SUCCESS);
205 }
206
207 static int
net_unmarshal_func(char * buffer,size_t buffer_size,void * retval,va_list ap,void * cache_mdata)208 net_unmarshal_func(char *buffer, size_t buffer_size, void *retval, va_list ap,
209 void *cache_mdata)
210 {
211 char *name;
212 uint32_t net;
213 int type;
214 struct netent *ne;
215 char *orig_buf;
216 size_t orig_buf_size;
217 int *ret_errno;
218
219 char *p;
220 char **alias;
221
222 switch ((enum nss_lookup_type)cache_mdata) {
223 case nss_lt_name:
224 name = va_arg(ap, char *);
225 break;
226 case nss_lt_id:
227 net = va_arg(ap, uint32_t);
228 type = va_arg(ap, int);
229 break;
230 case nss_lt_all:
231 break;
232 default:
233 /* should be unreachable */
234 return (NS_UNAVAIL);
235 }
236
237 ne = va_arg(ap, struct netent *);
238 orig_buf = va_arg(ap, char *);
239 orig_buf_size = va_arg(ap, size_t);
240 ret_errno = va_arg(ap, int *);
241
242 if (orig_buf_size <
243 buffer_size - sizeof(struct netent) - sizeof(char *)) {
244 *ret_errno = ERANGE;
245 return (NS_RETURN);
246 }
247
248 memcpy(ne, buffer, sizeof(struct netent));
249 memcpy(&p, buffer + sizeof(struct netent), sizeof(char *));
250
251 orig_buf = (char *)_ALIGN(orig_buf);
252 memcpy(orig_buf, buffer + sizeof(struct netent) + sizeof(char *) +
253 _ALIGN(p) - (size_t)p,
254 buffer_size - sizeof(struct netent) - sizeof(char *) -
255 _ALIGN(p) + (size_t)p);
256 p = (char *)_ALIGN(p);
257
258 NS_APPLY_OFFSET(ne->n_name, orig_buf, p, char *);
259 if (ne->n_aliases != NULL) {
260 NS_APPLY_OFFSET(ne->n_aliases, orig_buf, p, char **);
261
262 for (alias = ne->n_aliases; *alias; ++alias)
263 NS_APPLY_OFFSET(*alias, orig_buf, p, char *);
264 }
265
266 if (retval != NULL)
267 *((struct netent **)retval) = ne;
268
269 return (NS_SUCCESS);
270 }
271 #endif /* NS_CACHING */
272
273 static void
netent_data_free(void * ptr)274 netent_data_free(void *ptr)
275 {
276 struct netent_data *ned = ptr;
277
278 if (ned == NULL)
279 return;
280 ned->stayopen = 0;
281 _endnethtent(ned);
282 free(ned);
283 }
284
285 static void
netdata_free(void * ptr)286 netdata_free(void *ptr)
287 {
288 free(ptr);
289 }
290
291 int
__copy_netent(struct netent * ne,struct netent * nptr,char * buf,size_t buflen)292 __copy_netent(struct netent *ne, struct netent *nptr, char *buf, size_t buflen)
293 {
294 char *cp;
295 int i, n;
296 int numptr, len;
297
298 /* Find out the amount of space required to store the answer. */
299 numptr = 1; /* NULL ptr */
300 len = (char *)ALIGN(buf) - buf;
301 for (i = 0; ne->n_aliases[i]; i++, numptr++) {
302 len += strlen(ne->n_aliases[i]) + 1;
303 }
304 len += strlen(ne->n_name) + 1;
305 len += numptr * sizeof(char*);
306
307 if (len > (int)buflen) {
308 errno = ERANGE;
309 return (-1);
310 }
311
312 /* copy net value and type */
313 nptr->n_addrtype = ne->n_addrtype;
314 nptr->n_net = ne->n_net;
315
316 cp = (char *)ALIGN(buf) + numptr * sizeof(char *);
317
318 /* copy official name */
319 n = strlen(ne->n_name) + 1;
320 strcpy(cp, ne->n_name);
321 nptr->n_name = cp;
322 cp += n;
323
324 /* copy aliases */
325 nptr->n_aliases = (char **)ALIGN(buf);
326 for (i = 0 ; ne->n_aliases[i]; i++) {
327 n = strlen(ne->n_aliases[i]) + 1;
328 strcpy(cp, ne->n_aliases[i]);
329 nptr->n_aliases[i] = cp;
330 cp += n;
331 }
332 nptr->n_aliases[i] = NULL;
333
334 return (0);
335 }
336
337 int
getnetbyname_r(const char * name,struct netent * ne,char * buffer,size_t buflen,struct netent ** result,int * h_errorp)338 getnetbyname_r(const char *name, struct netent *ne, char *buffer,
339 size_t buflen, struct netent **result, int *h_errorp)
340 {
341 #ifdef NS_CACHING
342 static const nss_cache_info cache_info =
343 NS_COMMON_CACHE_INFO_INITIALIZER(
344 networks, (void *)nss_lt_name,
345 net_id_func, net_marshal_func, net_unmarshal_func);
346 #endif
347 static const ns_dtab dtab[] = {
348 NS_FILES_CB(_ht_getnetbyname, NULL)
349 { NSSRC_DNS, _dns_getnetbyname, NULL },
350 NS_NIS_CB(_nis_getnetbyname, NULL) /* force -DHESIOD */
351 #ifdef NS_CACHING
352 NS_CACHE_CB(&cache_info)
353 #endif
354 { 0 }
355 };
356 int rval, ret_errno = 0;
357
358 rval = _nsdispatch((void *)result, dtab, NSDB_NETWORKS,
359 "getnetbyname_r", default_src, name, ne, buffer, buflen,
360 &ret_errno, h_errorp);
361
362 if (rval != NS_SUCCESS) {
363 errno = ret_errno;
364 return ((ret_errno != 0) ? ret_errno : -1);
365 }
366 return (0);
367 }
368
369 int
getnetbyaddr_r(uint32_t addr,int af,struct netent * ne,char * buffer,size_t buflen,struct netent ** result,int * h_errorp)370 getnetbyaddr_r(uint32_t addr, int af, struct netent *ne, char *buffer,
371 size_t buflen, struct netent **result, int *h_errorp)
372 {
373 #ifdef NS_CACHING
374 static const nss_cache_info cache_info =
375 NS_COMMON_CACHE_INFO_INITIALIZER(
376 networks, (void *)nss_lt_id,
377 net_id_func, net_marshal_func, net_unmarshal_func);
378 #endif
379 static const ns_dtab dtab[] = {
380 NS_FILES_CB(_ht_getnetbyaddr, NULL)
381 { NSSRC_DNS, _dns_getnetbyaddr, NULL },
382 NS_NIS_CB(_nis_getnetbyaddr, NULL) /* force -DHESIOD */
383 #ifdef NS_CACHING
384 NS_CACHE_CB(&cache_info)
385 #endif
386 { 0 }
387 };
388 int rval, ret_errno = 0;
389
390 rval = _nsdispatch((void *)result, dtab, NSDB_NETWORKS,
391 "getnetbyaddr_r", default_src, addr, af, ne, buffer, buflen,
392 &ret_errno, h_errorp);
393
394 if (rval != NS_SUCCESS) {
395 errno = ret_errno;
396 return ((ret_errno != 0) ? ret_errno : -1);
397 }
398 return (0);
399 }
400
401 struct netent *
getnetbyname(const char * name)402 getnetbyname(const char *name)
403 {
404 struct netdata *nd;
405 struct netent *rval;
406 int ret_h_errno;
407
408 if ((nd = __netdata_init()) == NULL)
409 return (NULL);
410 if (getnetbyname_r(name, &nd->net, nd->data, sizeof(nd->data), &rval,
411 &ret_h_errno) != 0)
412 return (NULL);
413 return (rval);
414 }
415
416 struct netent *
getnetbyaddr(uint32_t addr,int af)417 getnetbyaddr(uint32_t addr, int af)
418 {
419 struct netdata *nd;
420 struct netent *rval;
421 int ret_h_errno;
422
423 if ((nd = __netdata_init()) == NULL)
424 return (NULL);
425 if (getnetbyaddr_r(addr, af, &nd->net, nd->data, sizeof(nd->data),
426 &rval, &ret_h_errno) != 0)
427 return (NULL);
428 return (rval);
429 }
430
431 void
setnetent(int stayopen)432 setnetent(int stayopen)
433 {
434 struct netent_data *ned;
435
436 if ((ned = __netent_data_init()) == NULL)
437 return;
438 _setnethtent(stayopen, ned);
439 _setnetdnsent(stayopen);
440 }
441
442 void
endnetent(void)443 endnetent(void)
444 {
445 struct netent_data *ned;
446
447 if ((ned = __netent_data_init()) == NULL)
448 return;
449 _endnethtent(ned);
450 _endnetdnsent();
451 }
452