1 /* 2 * Copyright (c) 1983, 1993 3 * The Regents of the University of California. All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 4. Neither the name of the University nor the names of its contributors 14 * may be used to endorse or promote products derived from this software 15 * without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27 * SUCH DAMAGE. 28 */ 29 30 #if defined(LIBC_SCCS) && !defined(lint) 31 static char sccsid[] = "@(#)getproto.c 8.1 (Berkeley) 6/4/93"; 32 #endif /* LIBC_SCCS and not lint */ 33 #include <sys/cdefs.h> 34 __FBSDID("$FreeBSD$"); 35 36 #include <errno.h> 37 #include <netdb.h> 38 #include <nsswitch.h> 39 #include "netdb_private.h" 40 #ifdef NS_CACHING 41 #include "nscache.h" 42 #endif 43 #include "nss_tls.h" 44 45 static const ns_src defaultsrc[] = { 46 { NSSRC_FILES, NS_SUCCESS }, 47 { NULL, 0 } 48 }; 49 50 #ifdef NS_CACHING 51 extern int __proto_id_func(char *, size_t *, va_list, void *); 52 extern int __proto_marshal_func(char *, size_t *, void *, va_list, void *); 53 extern int __proto_unmarshal_func(char *, size_t, void *, va_list, void *); 54 #endif 55 56 static int 57 files_getprotobynumber(void *retval, void *mdata, va_list ap) 58 { 59 struct protoent pe; 60 struct protoent_data *ped; 61 int error; 62 63 int number; 64 struct protoent *pptr; 65 char *buffer; 66 size_t buflen; 67 int *errnop; 68 69 number = va_arg(ap, int); 70 pptr = va_arg(ap, struct protoent *); 71 buffer = va_arg(ap, char *); 72 buflen = va_arg(ap, size_t); 73 errnop = va_arg(ap, int *); 74 75 if ((ped = __protoent_data_init()) == NULL) { 76 *errnop = errno; 77 return (NS_NOTFOUND); 78 } 79 80 __setprotoent_p(ped->stayopen, ped); 81 while ((error = __getprotoent_p(&pe, ped)) == 0) 82 if (pe.p_proto == number) 83 break; 84 if (!ped->stayopen) 85 __endprotoent_p(ped); 86 if (error != 0) { 87 *errnop = errno; 88 return (NS_NOTFOUND); 89 } 90 if (__copy_protoent(&pe, pptr, buffer, buflen) != 0) { 91 *errnop = errno; 92 return (NS_RETURN); 93 } 94 95 *((struct protoent **)retval) = pptr; 96 return (NS_SUCCESS); 97 } 98 99 int 100 getprotobynumber_r(int proto, struct protoent *pptr, char *buffer, 101 size_t buflen, struct protoent **result) 102 { 103 #ifdef NS_CACHING 104 static const nss_cache_info cache_info = 105 NS_COMMON_CACHE_INFO_INITIALIZER( 106 protocols, (void *)nss_lt_id, 107 __proto_id_func, __proto_marshal_func, __proto_unmarshal_func); 108 #endif 109 110 static const ns_dtab dtab[] = { 111 { NSSRC_FILES, files_getprotobynumber, NULL }, 112 #ifdef NS_CACHING 113 NS_CACHE_CB(&cache_info) 114 #endif 115 { NULL, NULL, NULL } 116 }; 117 int rv, ret_errno; 118 119 ret_errno = 0; 120 *result = NULL; 121 rv = nsdispatch(result, dtab, NSDB_PROTOCOLS, "getprotobynumber_r", 122 defaultsrc, proto, pptr, buffer, buflen, &ret_errno); 123 124 if (rv != NS_SUCCESS) { 125 errno = ret_errno; 126 return (ret_errno); 127 } 128 return (0); 129 } 130 131 struct protoent * 132 getprotobynumber(int proto) 133 { 134 struct protodata *pd; 135 struct protoent *rval; 136 137 if ((pd = __protodata_init()) == NULL) 138 return (NULL); 139 if (getprotobynumber_r(proto, &pd->proto, pd->data, sizeof(pd->data), 140 &rval) != 0) 141 return (NULL); 142 return (rval); 143 } 144