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