11573Srgrimes/* 21573Srgrimes * Copyright (c) 1983, 1993 31573Srgrimes * The Regents of the University of California. All rights reserved. 41573Srgrimes * 51573Srgrimes * Redistribution and use in source and binary forms, with or without 61573Srgrimes * modification, are permitted provided that the following conditions 71573Srgrimes * are met: 81573Srgrimes * 1. Redistributions of source code must retain the above copyright 91573Srgrimes * notice, this list of conditions and the following disclaimer. 101573Srgrimes * 2. Redistributions in binary form must reproduce the above copyright 111573Srgrimes * notice, this list of conditions and the following disclaimer in the 121573Srgrimes * documentation and/or other materials provided with the distribution. 131573Srgrimes * 4. Neither the name of the University nor the names of its contributors 141573Srgrimes * may be used to endorse or promote products derived from this software 151573Srgrimes * without specific prior written permission. 161573Srgrimes * 171573Srgrimes * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 181573Srgrimes * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 191573Srgrimes * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 201573Srgrimes * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 211573Srgrimes * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 221573Srgrimes * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 231573Srgrimes * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 241573Srgrimes * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 251573Srgrimes * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 261573Srgrimes * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 271573Srgrimes * SUCH DAMAGE. 281573Srgrimes */ 291573Srgrimes 301573Srgrimes#if defined(LIBC_SCCS) && !defined(lint) 311573Srgrimesstatic char sccsid[] = "@(#)getprotoent.c 8.1 (Berkeley) 6/4/93"; 321573Srgrimes#endif /* LIBC_SCCS and not lint */ 3392889Sobrien#include <sys/cdefs.h> 3492889Sobrien__FBSDID("$FreeBSD: releng/11.0/lib/libc/net/getprotoent.c 298226 2016-04-18 21:05:15Z avos $"); 351573Srgrimes 36157779Sume#include <sys/param.h> 371573Srgrimes#include <sys/socket.h> 38157779Sume#include <errno.h> 39149313Sstefanf#include <limits.h> 401573Srgrimes#include <netdb.h> 41158115Sume#include <nsswitch.h> 421573Srgrimes#include <stdio.h> 431573Srgrimes#include <stdlib.h> 441573Srgrimes#include <string.h> 45145279Sume#include "namespace.h" 46145279Sume#include "reentrant.h" 47145279Sume#include "un-namespace.h" 48145279Sume#include "netdb_private.h" 49158115Sume#ifdef NS_CACHING 50158115Sume#include "nscache.h" 51158115Sume#endif 52158115Sume#include "nss_tls.h" 531573Srgrimes 54158115Sumestatic const ns_src defaultsrc[] = { 55158115Sume { NSSRC_FILES, NS_SUCCESS }, 56158115Sume { NULL, 0 } 57158115Sume}; 58158115Sume 59157779SumeNETDB_THREAD_ALLOC(protoent_data) 60157779SumeNETDB_THREAD_ALLOC(protodata) 611573Srgrimes 62145279Sumestatic void 63145279Sumeprotoent_data_clear(struct protoent_data *ped) 64145279Sume{ 65145279Sume if (ped->fp) { 66145279Sume fclose(ped->fp); 67145279Sume ped->fp = NULL; 68145279Sume } 69145279Sume} 701573Srgrimes 71145279Sumestatic void 72157779Sumeprotoent_data_free(void *ptr) 73145279Sume{ 74157779Sume struct protoent_data *ped = ptr; 75145279Sume 76157779Sume protoent_data_clear(ped); 77157779Sume free(ped); 78145279Sume} 79145279Sume 80145279Sumestatic void 81157779Sumeprotodata_free(void *ptr) 82145279Sume{ 83157779Sume free(ptr); 84145279Sume} 85145279Sume 86158115Sume#ifdef NS_CACHING 87157779Sumeint 88158115Sume__proto_id_func(char *buffer, size_t *buffer_size, va_list ap, 89158115Sume void *cache_mdata) 90158115Sume{ 91158115Sume char *name; 92158115Sume int proto; 93158115Sume 94158115Sume size_t desired_size, size; 95158115Sume enum nss_lookup_type lookup_type; 96158115Sume int res = NS_UNAVAIL; 97158115Sume 98158115Sume lookup_type = (enum nss_lookup_type)cache_mdata; 99158115Sume switch (lookup_type) { 100158115Sume case nss_lt_name: 101158115Sume name = va_arg(ap, char *); 102158115Sume 103158115Sume size = strlen(name); 104158115Sume desired_size = sizeof(enum nss_lookup_type) + size + 1; 105158115Sume if (desired_size > *buffer_size) { 106158115Sume res = NS_RETURN; 107158115Sume goto fin; 108158115Sume } 109158115Sume 110158115Sume memcpy(buffer, &lookup_type, sizeof(enum nss_lookup_type)); 111158115Sume memcpy(buffer + sizeof(enum nss_lookup_type), name, size + 1); 112158115Sume 113158115Sume res = NS_SUCCESS; 114158115Sume break; 115158115Sume case nss_lt_id: 116158115Sume proto = va_arg(ap, int); 117158115Sume 118158115Sume desired_size = sizeof(enum nss_lookup_type) + sizeof(int); 119158115Sume if (desired_size > *buffer_size) { 120158115Sume res = NS_RETURN; 121158115Sume goto fin; 122158115Sume } 123158115Sume 124158115Sume memcpy(buffer, &lookup_type, sizeof(enum nss_lookup_type)); 125158115Sume memcpy(buffer + sizeof(enum nss_lookup_type), &proto, 126158115Sume sizeof(int)); 127158115Sume 128158115Sume res = NS_SUCCESS; 129158115Sume break; 130158115Sume default: 131158115Sume /* should be unreachable */ 132158115Sume return (NS_UNAVAIL); 133158115Sume } 134158115Sume 135158115Sumefin: 136158115Sume *buffer_size = desired_size; 137158115Sume return (res); 138158115Sume} 139158115Sume 140158115Sume 141158115Sumeint 142158115Sume__proto_marshal_func(char *buffer, size_t *buffer_size, void *retval, 143158115Sume va_list ap, void *cache_mdata) 144158115Sume{ 145158115Sume char *name; 146158115Sume int num; 147158115Sume struct protoent *proto; 148158115Sume char *orig_buf; 149158115Sume size_t orig_buf_size; 150158115Sume 151158115Sume struct protoent new_proto; 152158115Sume size_t desired_size, size, aliases_size; 153158115Sume char *p; 154158115Sume char **alias; 155158115Sume 156158115Sume switch ((enum nss_lookup_type)cache_mdata) { 157158115Sume case nss_lt_name: 158158115Sume name = va_arg(ap, char *); 159158115Sume break; 160158115Sume case nss_lt_id: 161158115Sume num = va_arg(ap, int); 162158115Sume break; 163158115Sume case nss_lt_all: 164158115Sume break; 165158115Sume default: 166158115Sume /* should be unreachable */ 167158115Sume return (NS_UNAVAIL); 168158115Sume } 169158115Sume 170158115Sume proto = va_arg(ap, struct protoent *); 171158115Sume orig_buf = va_arg(ap, char *); 172158115Sume orig_buf_size = va_arg(ap, size_t); 173158115Sume 174158115Sume desired_size = _ALIGNBYTES + sizeof(struct protoent) + sizeof(char *); 175158115Sume if (proto->p_name != NULL) 176158115Sume desired_size += strlen(proto->p_name) + 1; 177158115Sume 178158115Sume if (proto->p_aliases != NULL) { 179158115Sume aliases_size = 0; 180158115Sume for (alias = proto->p_aliases; *alias; ++alias) { 181158115Sume desired_size += strlen(*alias) + 1; 182158115Sume ++aliases_size; 183158115Sume } 184158115Sume 185158115Sume desired_size += _ALIGNBYTES + (aliases_size + 1) * 186158115Sume sizeof(char *); 187158115Sume } 188158115Sume 189158115Sume if (*buffer_size < desired_size) { 190158115Sume /* this assignment is here for future use */ 191158115Sume *buffer_size = desired_size; 192158115Sume return (NS_RETURN); 193158115Sume } 194158115Sume 195158115Sume memcpy(&new_proto, proto, sizeof(struct protoent)); 196158115Sume 197158115Sume *buffer_size = desired_size; 198158115Sume memset(buffer, 0, desired_size); 199158115Sume p = buffer + sizeof(struct protoent) + sizeof(char *); 200158115Sume memcpy(buffer + sizeof(struct protoent), &p, sizeof(char *)); 201158115Sume p = (char *)_ALIGN(p); 202158115Sume 203158115Sume if (new_proto.p_name != NULL) { 204158115Sume size = strlen(new_proto.p_name); 205158115Sume memcpy(p, new_proto.p_name, size); 206158115Sume new_proto.p_name = p; 207158115Sume p += size + 1; 208158115Sume } 209158115Sume 210158115Sume if (new_proto.p_aliases != NULL) { 211158115Sume p = (char *)_ALIGN(p); 212158115Sume memcpy(p, new_proto.p_aliases, sizeof(char *) * aliases_size); 213158115Sume new_proto.p_aliases = (char **)p; 214158115Sume p += sizeof(char *) * (aliases_size + 1); 215158115Sume 216158115Sume for (alias = new_proto.p_aliases; *alias; ++alias) { 217158115Sume size = strlen(*alias); 218158115Sume memcpy(p, *alias, size); 219158115Sume *alias = p; 220158115Sume p += size + 1; 221158115Sume } 222158115Sume } 223158115Sume 224158115Sume memcpy(buffer, &new_proto, sizeof(struct protoent)); 225158115Sume return (NS_SUCCESS); 226158115Sume} 227158115Sume 228158115Sumeint 229158115Sume__proto_unmarshal_func(char *buffer, size_t buffer_size, void *retval, 230158115Sume va_list ap, void *cache_mdata) 231158115Sume{ 232158115Sume char *name; 233158115Sume int num; 234158115Sume struct protoent *proto; 235158115Sume char *orig_buf; 236158115Sume size_t orig_buf_size; 237158115Sume int *ret_errno; 238158115Sume 239158115Sume char *p; 240158115Sume char **alias; 241158115Sume 242158115Sume switch ((enum nss_lookup_type)cache_mdata) { 243158115Sume case nss_lt_name: 244158115Sume name = va_arg(ap, char *); 245158115Sume break; 246158115Sume case nss_lt_id: 247158115Sume num = va_arg(ap, int); 248158115Sume break; 249158115Sume case nss_lt_all: 250158115Sume break; 251158115Sume default: 252158115Sume /* should be unreachable */ 253158115Sume return (NS_UNAVAIL); 254158115Sume } 255158115Sume 256158115Sume proto = va_arg(ap, struct protoent *); 257158115Sume orig_buf = va_arg(ap, char *); 258158115Sume orig_buf_size = va_arg(ap, size_t); 259158115Sume ret_errno = va_arg(ap, int *); 260158115Sume 261158115Sume if (orig_buf_size < 262158115Sume buffer_size - sizeof(struct protoent) - sizeof(char *)) { 263158115Sume *ret_errno = ERANGE; 264158115Sume return (NS_RETURN); 265158115Sume } 266158115Sume 267158115Sume memcpy(proto, buffer, sizeof(struct protoent)); 268158115Sume memcpy(&p, buffer + sizeof(struct protoent), sizeof(char *)); 269158115Sume 270158115Sume orig_buf = (char *)_ALIGN(orig_buf); 271158115Sume memcpy(orig_buf, buffer + sizeof(struct protoent) + sizeof(char *) + 272158115Sume _ALIGN(p) - (size_t)p, 273158115Sume buffer_size - sizeof(struct protoent) - sizeof(char *) - 274158115Sume _ALIGN(p) + (size_t)p); 275158115Sume p = (char *)_ALIGN(p); 276158115Sume 277158115Sume NS_APPLY_OFFSET(proto->p_name, orig_buf, p, char *); 278158115Sume if (proto->p_aliases != NULL) { 279158115Sume NS_APPLY_OFFSET(proto->p_aliases, orig_buf, p, char **); 280158115Sume 281158115Sume for (alias = proto->p_aliases; *alias; ++alias) 282158115Sume NS_APPLY_OFFSET(*alias, orig_buf, p, char *); 283158115Sume } 284158115Sume 285158115Sume if (retval != NULL) 286158115Sume *((struct protoent **)retval) = proto; 287158115Sume 288158115Sume return (NS_SUCCESS); 289158115Sume} 290158115Sume 291158115SumeNSS_MP_CACHE_HANDLING(protocols); 292158115Sume#endif /* NS_CACHING */ 293158115Sume 294158115Sumeint 295157779Sume__copy_protoent(struct protoent *pe, struct protoent *pptr, char *buf, 296157779Sume size_t buflen) 297145279Sume{ 298157779Sume char *cp; 299157779Sume int i, n; 300157779Sume int numptr, len; 301145279Sume 302157779Sume /* Find out the amount of space required to store the answer. */ 303157779Sume numptr = 1; /* NULL ptr */ 304157779Sume len = (char *)ALIGN(buf) - buf; 305157779Sume for (i = 0; pe->p_aliases[i]; i++, numptr++) { 306157779Sume len += strlen(pe->p_aliases[i]) + 1; 307157779Sume } 308157779Sume len += strlen(pe->p_name) + 1; 309157779Sume len += numptr * sizeof(char*); 310157779Sume 311157779Sume if (len > (int)buflen) { 312157779Sume errno = ERANGE; 313157779Sume return (-1); 314157779Sume } 315157779Sume 316157779Sume /* copy protocol value*/ 317157779Sume pptr->p_proto = pe->p_proto; 318157779Sume 319157779Sume cp = (char *)ALIGN(buf) + numptr * sizeof(char *); 320157779Sume 321157779Sume /* copy official name */ 322157779Sume n = strlen(pe->p_name) + 1; 323157779Sume strcpy(cp, pe->p_name); 324157779Sume pptr->p_name = cp; 325157779Sume cp += n; 326157779Sume 327157779Sume /* copy aliases */ 328157779Sume pptr->p_aliases = (char **)ALIGN(buf); 329157779Sume for (i = 0 ; pe->p_aliases[i]; i++) { 330157779Sume n = strlen(pe->p_aliases[i]) + 1; 331157779Sume strcpy(cp, pe->p_aliases[i]); 332157779Sume pptr->p_aliases[i] = cp; 333157779Sume cp += n; 334157779Sume } 335157779Sume pptr->p_aliases[i] = NULL; 336157779Sume 337157779Sume return (0); 338145279Sume} 339145279Sume 3401573Srgrimesvoid 341157779Sume__setprotoent_p(int f, struct protoent_data *ped) 3421573Srgrimes{ 343145279Sume if (ped->fp == NULL) 344254700Sjilles ped->fp = fopen(_PATH_PROTOCOLS, "re"); 3451573Srgrimes else 346145279Sume rewind(ped->fp); 347145279Sume ped->stayopen |= f; 3481573Srgrimes} 3491573Srgrimes 3501573Srgrimesvoid 351157779Sume__endprotoent_p(struct protoent_data *ped) 3521573Srgrimes{ 353145279Sume if (ped->fp) { 354145279Sume fclose(ped->fp); 355145279Sume ped->fp = NULL; 3561573Srgrimes } 357145279Sume ped->stayopen = 0; 3581573Srgrimes} 3591573Srgrimes 360145279Sumeint 361157779Sume__getprotoent_p(struct protoent *pe, struct protoent_data *ped) 3621573Srgrimes{ 3631573Srgrimes char *p; 364145279Sume char *cp, **q, *endp; 365145279Sume long l; 3661573Srgrimes 367254700Sjilles if (ped->fp == NULL && (ped->fp = fopen(_PATH_PROTOCOLS, "re")) == NULL) 368145279Sume return (-1); 3691573Srgrimesagain: 370145626Sume if ((p = fgets(ped->line, sizeof ped->line, ped->fp)) == NULL) 371145279Sume return (-1); 3721573Srgrimes if (*p == '#') 3731573Srgrimes goto again; 3741573Srgrimes cp = strpbrk(p, "#\n"); 375139612Ssobomax if (cp != NULL) 376139612Ssobomax *cp = '\0'; 377145279Sume pe->p_name = p; 3781573Srgrimes cp = strpbrk(p, " \t"); 3791573Srgrimes if (cp == NULL) 3801573Srgrimes goto again; 3811573Srgrimes *cp++ = '\0'; 3821573Srgrimes while (*cp == ' ' || *cp == '\t') 3831573Srgrimes cp++; 3841573Srgrimes p = strpbrk(cp, " \t"); 3851573Srgrimes if (p != NULL) 3861573Srgrimes *p++ = '\0'; 387145279Sume l = strtol(cp, &endp, 10); 388145279Sume if (endp == cp || *endp != '\0' || l < 0 || l > USHRT_MAX) 389145279Sume goto again; 390145279Sume pe->p_proto = l; 391145279Sume q = pe->p_aliases = ped->aliases; 3921573Srgrimes if (p != NULL) { 3931573Srgrimes cp = p; 3941573Srgrimes while (cp && *cp) { 3951573Srgrimes if (*cp == ' ' || *cp == '\t') { 3961573Srgrimes cp++; 3971573Srgrimes continue; 3981573Srgrimes } 399145626Sume if (q < &ped->aliases[_MAXALIASES - 1]) 4001573Srgrimes *q++ = cp; 4011573Srgrimes cp = strpbrk(cp, " \t"); 4021573Srgrimes if (cp != NULL) 4031573Srgrimes *cp++ = '\0'; 4041573Srgrimes } 4051573Srgrimes } 4061573Srgrimes *q = NULL; 407145279Sume return (0); 4081573Srgrimes} 409145279Sume 410158115Sumestatic int 411158115Sumefiles_getprotoent_r(void *retval, void *mdata, va_list ap) 412157779Sume{ 413157779Sume struct protoent pe; 414157779Sume struct protoent_data *ped; 415157779Sume 416158115Sume struct protoent *pptr; 417158115Sume char *buffer; 418158115Sume size_t buflen; 419158115Sume int *errnop; 420158115Sume 421158115Sume pptr = va_arg(ap, struct protoent *); 422158115Sume buffer = va_arg(ap, char *); 423158115Sume buflen = va_arg(ap, size_t); 424158115Sume errnop = va_arg(ap, int *); 425158115Sume 426211276Sume if ((ped = __protoent_data_init()) == NULL) { 427211276Sume *errnop = errno; 428211276Sume return (NS_NOTFOUND); 429211276Sume } 430157779Sume 431158115Sume if (__getprotoent_p(&pe, ped) != 0) { 432158115Sume *errnop = errno; 433158115Sume return (NS_NOTFOUND); 434158115Sume } 435158115Sume 436158115Sume if (__copy_protoent(&pe, pptr, buffer, buflen) != 0) { 437158115Sume *errnop = errno; 438211276Sume return (NS_RETURN); 439158115Sume } 440158115Sume 441158115Sume *((struct protoent **)retval) = pptr; 442158115Sume return (NS_SUCCESS); 443157779Sume} 444157779Sume 445158115Sumestatic int 446158115Sumefiles_setprotoent(void *retval, void *mdata, va_list ap) 447145279Sume{ 448157779Sume struct protoent_data *ped; 449158115Sume int f; 450145279Sume 451158115Sume f = va_arg(ap, int); 452157779Sume if ((ped = __protoent_data_init()) == NULL) 453158115Sume return (NS_UNAVAIL); 454158115Sume 455157779Sume __setprotoent_p(f, ped); 456158115Sume return (NS_UNAVAIL); 457145279Sume} 458145279Sume 459158115Sumestatic int 460158115Sumefiles_endprotoent(void *retval, void *mdata, va_list ap) 461145279Sume{ 462157779Sume struct protoent_data *ped; 463145279Sume 464157779Sume if ((ped = __protoent_data_init()) == NULL) 465158115Sume return (NS_UNAVAIL); 466158115Sume 467157779Sume __endprotoent_p(ped); 468158115Sume return (NS_UNAVAIL); 469145279Sume} 470145279Sume 471158115Sumeint 472158115Sumegetprotoent_r(struct protoent *pptr, char *buffer, size_t buflen, 473158115Sume struct protoent **result) 474158115Sume{ 475158115Sume#ifdef NS_CACHING 476158115Sume static const nss_cache_info cache_info = NS_MP_CACHE_INFO_INITIALIZER( 477158115Sume protocols, (void *)nss_lt_all, 478158115Sume __proto_marshal_func, __proto_unmarshal_func); 479158115Sume#endif 480158115Sume static const ns_dtab dtab[] = { 481158115Sume { NSSRC_FILES, files_getprotoent_r, (void *)nss_lt_all }, 482158115Sume#ifdef NS_CACHING 483158115Sume NS_CACHE_CB(&cache_info) 484158115Sume#endif 485158115Sume { NULL, NULL, NULL } 486158115Sume }; 487158115Sume int rv, ret_errno; 488158115Sume 489158115Sume ret_errno = 0; 490158115Sume *result = NULL; 491158115Sume rv = nsdispatch(result, dtab, NSDB_PROTOCOLS, "getprotoent_r", 492158115Sume defaultsrc, pptr, buffer, buflen, &ret_errno); 493158115Sume 494211276Sume if (rv != NS_SUCCESS) { 495211276Sume errno = ret_errno; 496213453Sume return (ret_errno); 497211276Sume } 498211276Sume return (0); 499158115Sume} 500158115Sume 501158115Sumevoid 502158115Sumesetprotoent(int stayopen) 503158115Sume{ 504158115Sume#ifdef NS_CACHING 505158115Sume static const nss_cache_info cache_info = NS_MP_CACHE_INFO_INITIALIZER( 506158115Sume protocols, (void *)nss_lt_all, 507158115Sume NULL, NULL); 508158115Sume#endif 509158115Sume 510158115Sume static const ns_dtab dtab[] = { 511158115Sume { NSSRC_FILES, files_setprotoent, NULL }, 512158115Sume#ifdef NS_CACHING 513158115Sume NS_CACHE_CB(&cache_info) 514158115Sume#endif 515158115Sume { NULL, NULL, NULL } 516158115Sume }; 517158115Sume 518158115Sume (void)nsdispatch(NULL, dtab, NSDB_PROTOCOLS, "setprotoent", defaultsrc, 519158115Sume stayopen); 520158115Sume} 521158115Sume 522158115Sumevoid 523158115Sumeendprotoent(void) 524158115Sume{ 525158115Sume#ifdef NS_CACHING 526158115Sume static const nss_cache_info cache_info = NS_MP_CACHE_INFO_INITIALIZER( 527158115Sume protocols, (void *)nss_lt_all, 528158115Sume NULL, NULL); 529158115Sume#endif 530158115Sume 531158115Sume static const ns_dtab dtab[] = { 532158115Sume { NSSRC_FILES, files_endprotoent, NULL }, 533158115Sume#ifdef NS_CACHING 534158115Sume NS_CACHE_CB(&cache_info) 535158115Sume#endif 536158115Sume { NULL, NULL, NULL } 537158115Sume }; 538158115Sume 539158115Sume (void)nsdispatch(NULL, dtab, NSDB_PROTOCOLS, "endprotoent", defaultsrc); 540158115Sume} 541158115Sume 542145279Sumestruct protoent * 543145279Sumegetprotoent(void) 544145279Sume{ 545145279Sume struct protodata *pd; 546157779Sume struct protoent *rval; 547145279Sume 548145279Sume if ((pd = __protodata_init()) == NULL) 549145279Sume return (NULL); 550157779Sume if (getprotoent_r(&pd->proto, pd->data, sizeof(pd->data), &rval) != 0) 551145279Sume return (NULL); 552157779Sume return (rval); 553145279Sume} 554