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