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 * 3. 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 * @(#)getproto.c 8.1 (Berkeley) 6/4/93 30 * $FreeBSD: src/lib/libc/net/getproto.c,v 1.7 2007/01/09 00:28:02 imp Exp $ 31 */ 32 33 #include <netdb.h> 34 #include <nsswitch.h> 35 #include "netdb_private.h" 36 #ifdef NS_CACHING 37 #include "nscache.h" 38 #endif 39 #include "nss_tls.h" 40 41 static const ns_src defaultsrc[] = { 42 { NSSRC_FILES, NS_SUCCESS }, 43 { NULL, 0 } 44 }; 45 46 #ifdef NS_CACHING 47 extern int __proto_id_func(char *, size_t *, va_list, void *); 48 extern int __proto_marshal_func(char *, size_t *, void *, va_list, void *); 49 extern int __proto_unmarshal_func(char *, size_t, void *, va_list, void *); 50 #endif 51 52 static int 53 files_getprotobynumber(void *retval, void *mdata __unused, va_list ap) 54 { 55 struct protoent pe; 56 struct protoent_data *ped; 57 int error; 58 59 int number; 60 struct protoent *pptr; 61 char *buffer; 62 size_t buflen; 63 int *errnop; 64 65 number = va_arg(ap, int); 66 pptr = va_arg(ap, struct protoent *); 67 buffer = va_arg(ap, char *); 68 buflen = va_arg(ap, size_t); 69 errnop = va_arg(ap, int *); 70 71 if ((ped = __protoent_data_init()) == NULL) { 72 *errnop = -1; 73 return (NS_NOTFOUND); 74 } 75 76 __setprotoent_p(ped->stayopen, ped); 77 while ((error = __getprotoent_p(&pe, ped)) == 0) 78 if (pe.p_proto == number) 79 break; 80 if (!ped->stayopen) 81 __endprotoent_p(ped); 82 if (error != 0) { 83 *errnop = -1; 84 return (NS_NOTFOUND); 85 } 86 if (__copy_protoent(&pe, pptr, buffer, buflen) != 0) { 87 *errnop = -1; 88 return (NS_NOTFOUND); 89 } 90 91 *((struct protoent **)retval) = pptr; 92 return (NS_SUCCESS); 93 } 94 95 int 96 getprotobynumber_r(int proto, struct protoent *pptr, char *buffer, 97 size_t buflen, struct protoent **result) 98 { 99 #ifdef NS_CACHING 100 static const nss_cache_info cache_info = 101 NS_COMMON_CACHE_INFO_INITIALIZER( 102 protocols, (void *)nss_lt_id, 103 __proto_id_func, __proto_marshal_func, __proto_unmarshal_func); 104 #endif 105 106 static const ns_dtab dtab[] = { 107 { NSSRC_FILES, files_getprotobynumber, NULL }, 108 #ifdef NS_CACHING 109 NS_CACHE_CB(&cache_info) 110 #endif 111 { NULL, NULL, NULL } 112 }; 113 int rv, ret_errno; 114 115 ret_errno = 0; 116 *result = NULL; 117 rv = nsdispatch(result, dtab, NSDB_PROTOCOLS, "getprotobynumber_r", 118 defaultsrc, proto, pptr, buffer, buflen, &ret_errno); 119 120 if (rv == NS_SUCCESS) 121 return (0); 122 else 123 return (ret_errno); 124 } 125 126 struct protoent * 127 getprotobynumber(int proto) 128 { 129 struct protodata *pd; 130 struct protoent *rval; 131 132 if ((pd = __protodata_init()) == NULL) 133 return (NULL); 134 if (getprotobynumber_r(proto, &pd->proto, pd->data, sizeof(pd->data), 135 &rval) != 0) 136 return (NULL); 137 return (rval); 138 } 139