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 * $FreeBSD: src/lib/libc/net/getnetbynis.c,v 1.21 2006/04/15 16:20:27 ume Exp $ 26 */ 27 28 #include <sys/param.h> 29 #include <sys/socket.h> 30 #include <netinet/in.h> 31 #include <arpa/inet.h> 32 #include <netdb.h> 33 #include <resolv.h> 34 #include <stdio.h> 35 #include <stdlib.h> 36 #include <ctype.h> 37 #include <errno.h> 38 #include <string.h> 39 #include <stdarg.h> 40 #include <nsswitch.h> 41 #include <arpa/nameser.h> 42 #ifdef YP 43 #include <rpc/rpc.h> 44 #include <rpcsvc/yp_prot.h> 45 #include <rpcsvc/ypclnt.h> 46 #endif 47 #include "netdb_private.h" 48 49 #ifdef YP 50 static int 51 _getnetbynis(const char *name, char *map, int af, struct netent *ne, 52 struct netent_data *ned) 53 { 54 char *p, *bp, *ep; 55 char *cp, **q; 56 char *result; 57 int resultlen, len; 58 char ypbuf[YPMAXRECORD + 2]; 59 60 switch(af) { 61 case AF_INET: 62 break; 63 default: 64 case AF_INET6: 65 errno = EAFNOSUPPORT; 66 return (-1); 67 } 68 69 if (ned->yp_domain == NULL) 70 if (yp_get_default_domain (&ned->yp_domain)) 71 return (-1); 72 73 if (yp_match(ned->yp_domain, map, name, strlen(name), &result, 74 &resultlen)) 75 return (-1); 76 77 bcopy((char *)result, (char *)&ypbuf, resultlen); 78 ypbuf[resultlen] = '\0'; 79 free(result); 80 result = (char *)&ypbuf; 81 82 if ((cp = index(result, '\n'))) 83 *cp = '\0'; 84 85 cp = strpbrk(result, " \t"); 86 *cp++ = '\0'; 87 bp = ned->netbuf; 88 ep = ned->netbuf + sizeof ned->netbuf; 89 len = strlen(result) + 1; 90 if (ep - bp < len) { 91 RES_SET_H_ERRNO(__res_state(), NO_RECOVERY); 92 return (-1); 93 } 94 strlcpy(bp, result, ep - bp); 95 ne->n_name = bp; 96 bp += len; 97 98 while (*cp == ' ' || *cp == '\t') 99 cp++; 100 101 ne->n_net = inet_network(cp); 102 ne->n_addrtype = AF_INET; 103 104 q = ne->n_aliases = ned->net_aliases; 105 cp = strpbrk(cp, " \t"); 106 if (cp != NULL) 107 *cp++ = '\0'; 108 while (cp && *cp) { 109 if (*cp == ' ' || *cp == '\t') { 110 cp++; 111 continue; 112 } 113 if (q > &ned->net_aliases[_MAXALIASES - 1]) 114 break; 115 p = strpbrk(cp, " \t"); 116 if (p != NULL) 117 *p++ = '\0'; 118 len = strlen(cp) + 1; 119 if (ep - bp < len) 120 break; 121 strlcpy(bp, cp, ep - bp); 122 *q++ = bp; 123 bp += len; 124 cp = p; 125 } 126 *q = NULL; 127 return (0); 128 } 129 #endif /* YP */ 130 131 int 132 _nis_getnetbyname(void *rval, void *cb_data __unused, va_list ap) 133 { 134 #ifdef YP 135 const char *name; 136 char *buffer; 137 size_t buflen; 138 int *errnop __unused; 139 int *h_errnop; 140 struct netent *nptr, ne; 141 struct netent_data *ned; 142 res_state statp; 143 144 name = va_arg(ap, const char *); 145 nptr = va_arg(ap, struct netent *); 146 buffer = va_arg(ap, char *); 147 buflen = va_arg(ap, size_t); 148 errnop = va_arg(ap, int *); 149 h_errnop = va_arg(ap, int *); 150 151 statp = __res_state(); 152 if ((ned = __netent_data_init()) == NULL) { 153 RES_SET_H_ERRNO(statp, NETDB_INTERNAL); 154 *h_errnop = statp->res_h_errno; 155 return (NS_UNAVAIL); 156 } 157 158 if (_getnetbynis(name, "networks.byname", AF_INET, &ne, ned) != 0) { 159 *h_errnop = statp->res_h_errno; 160 return (NS_NOTFOUND); 161 } 162 if (__copy_netent(&ne, nptr, buffer, buflen) != 0) { 163 *h_errnop = statp->res_h_errno; 164 return (NS_NOTFOUND); 165 } 166 *((struct netent **)rval) = nptr; 167 return (NS_SUCCESS); 168 #else 169 return (NS_UNAVAIL); 170 #endif 171 172 } 173 174 int 175 _nis_getnetbyaddr(void *rval, void *cb_data __unused, va_list ap) 176 { 177 #ifdef YP 178 uint32_t addr; 179 int af; 180 char *buffer; 181 size_t buflen; 182 int *errnop __unused; 183 int *h_errnop; 184 struct netent *nptr, ne; 185 struct netent_data *ned; 186 char *str, *cp; 187 uint32_t net2; 188 int nn; 189 unsigned int netbr[4]; 190 char buf[MAXDNAME]; 191 res_state statp; 192 193 addr = va_arg(ap, uint32_t); 194 af = va_arg(ap, int); 195 nptr = va_arg(ap, struct netent *); 196 buffer = va_arg(ap, char *); 197 buflen = va_arg(ap, size_t); 198 errnop = va_arg(ap, int *); 199 h_errnop = va_arg(ap, int *); 200 201 statp = __res_state(); 202 if ((ned = __netent_data_init()) == NULL) { 203 RES_SET_H_ERRNO(statp, NETDB_INTERNAL); 204 *h_errnop = statp->res_h_errno; 205 return (NS_UNAVAIL); 206 } 207 208 if (af != AF_INET) { 209 RES_SET_H_ERRNO(statp, NETDB_INTERNAL); 210 *h_errnop = statp->res_h_errno; 211 errno = EAFNOSUPPORT; 212 return (NS_UNAVAIL); 213 } 214 215 for (nn = 4, net2 = addr; net2; net2 >>= 8) { 216 netbr[--nn] = net2 & 0xff; 217 } 218 219 switch (nn) { 220 case 3: /* Class A */ 221 sprintf(buf, "%u", netbr[3]); 222 break; 223 case 2: /* Class B */ 224 sprintf(buf, "%u.%u", netbr[2], netbr[3]); 225 break; 226 case 1: /* Class C */ 227 sprintf(buf, "%u.%u.%u", netbr[1], netbr[2], netbr[3]); 228 break; 229 case 0: /* Class D - E */ 230 sprintf(buf, "%u.%u.%u.%u", netbr[0], netbr[1], 231 netbr[2], netbr[3]); 232 break; 233 } 234 235 str = (char *)&buf; 236 cp = str + (strlen(str) - 2); 237 238 while(!strcmp(cp, ".0")) { 239 *cp = '\0'; 240 cp = str + (strlen(str) - 2); 241 } 242 243 if (_getnetbynis(str, "networks.byaddr", af, &ne, ned) != 0) { 244 *h_errnop = statp->res_h_errno; 245 return (NS_NOTFOUND); 246 } 247 if (__copy_netent(&ne, nptr, buffer, buflen) != 0) { 248 *h_errnop = statp->res_h_errno; 249 return (NS_NOTFOUND); 250 } 251 *((struct netent **)rval) = nptr; 252 return (NS_SUCCESS); 253 #else 254 return (NS_UNAVAIL); 255 #endif /* YP */ 256 } 257