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 nonexistent */ 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) 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 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 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 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 286 netdata_free(void *ptr) 287 { 288 free(ptr); 289 } 290 291 int 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 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 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 * 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 * 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 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 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