1b03b48cd14962b0f539291fbc4d0fef3dd12cf7
[dragonfly.git] / lib / libc / net / getnetnamadr.c
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  * $FreeBSD: src/lib/libc/net/getnetnamadr.c,v 1.23 2006/04/28 12:03:35 ume Exp $
26  * $DragonFly: src/lib/libc/net/getnetnamadr.c,v 1.4 2005/11/13 02:04:47 swildner Exp $
27  */
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 extern int _ht_getnetbyname(void *, void *, va_list);
50 extern int _dns_getnetbyname(void *, void *, va_list);
51 extern int _nis_getnetbyname(void *, void *, va_list);
52 extern int _ht_getnetbyaddr(void *, void *, va_list);
53 extern int _dns_getnetbyaddr(void *, void *, va_list);
54 extern int _nis_getnetbyaddr(void *, void *, va_list);
55
56 /* Network lookup order if nsswitch.conf is broken or nonexistant */
57 static const ns_src default_src[] = {
58         { NSSRC_FILES, NS_SUCCESS },
59         { NSSRC_DNS, NS_SUCCESS },
60         { 0 }
61 };
62
63 NETDB_THREAD_ALLOC(netent_data)
64 NETDB_THREAD_ALLOC(netdata)
65
66 #ifdef NS_CACHING
67 static int
68 net_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
119 fin:
120         *buffer_size = desired_size;
121         return (res);
122 }
123
124
125 static int
126 net_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
214 static int
215 net_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
280 static void
281 netent_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
292 static void
293 netdata_free(void *ptr)
294 {
295         free(ptr);
296 }
297
298 int
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
344 int
345 getnetbyname_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;
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         return ((rval == NS_SUCCESS) ? 0 : -1);
370 }
371
372 int
373 getnetbyaddr_r(uint32_t addr, int af, struct netent *ne, char *buffer,
374                size_t buflen, struct netent **result, int *h_errorp)
375 {
376 #ifdef NS_CACHING
377         static const nss_cache_info cache_info =
378                 NS_COMMON_CACHE_INFO_INITIALIZER(
379                 networks, (void *)nss_lt_id,
380                 net_id_func, net_marshal_func, net_unmarshal_func);
381 #endif
382         static const ns_dtab dtab[] = {
383                 NS_FILES_CB(_ht_getnetbyaddr, NULL)
384                 { NSSRC_DNS, _dns_getnetbyaddr, NULL },
385                 NS_NIS_CB(_nis_getnetbyaddr, NULL) /* force -DHESIOD */
386 #ifdef NS_CACHING
387                 NS_CACHE_CB(&cache_info)
388 #endif
389                 { 0 }
390         };
391         int rval, ret_errno;
392
393         rval = _nsdispatch((void *)result, dtab, NSDB_NETWORKS,
394             "getnetbyaddr_r", default_src, addr, af, ne, buffer, buflen,
395             &ret_errno, h_errorp);
396
397         return ((rval == NS_SUCCESS) ? 0 : -1);
398 }
399
400 struct netent *
401 getnetbyname(const char *name)
402 {
403         struct netdata *nd;
404         struct netent *rval;
405         int ret_h_errno;
406
407         if ((nd = __netdata_init()) == NULL)
408                 return (NULL);
409         if (getnetbyname_r(name, &nd->net, nd->data, sizeof(nd->data), &rval,
410             &ret_h_errno) != 0)
411                 return (NULL);
412         return (rval);
413 }
414
415 struct netent *
416 getnetbyaddr(uint32_t addr, int af)
417 {
418         struct netdata *nd;
419         struct netent *rval;
420         int ret_h_errno;
421
422         if ((nd = __netdata_init()) == NULL)
423                 return (NULL);
424         if (getnetbyaddr_r(addr, af, &nd->net, nd->data, sizeof(nd->data),
425             &rval, &ret_h_errno) != 0)
426                 return (NULL);
427         return (rval);
428 }
429
430 void
431 setnetent(int stayopen)
432 {
433         struct netent_data *ned;
434
435         if ((ned = __netent_data_init()) == NULL)
436                 return;
437         _setnethtent(stayopen, ned);
438         _setnetdnsent(stayopen);
439 }
440
441 void
442 endnetent(void)
443 {
444         struct netent_data *ned;
445
446         if ((ned = __netent_data_init()) == NULL)
447                 return;
448         _endnethtent(ned);
449         _endnetdnsent();
450 }