1 /* $NetBSD: sethostent.c,v 1.21 2023/12/07 22:33:49 kre Exp $ */ 2 3 /* 4 * Copyright (c) 1985, 1993 5 * The Regents of the University of California. All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 3. Neither the name of the University nor the names of its contributors 16 * may be used to endorse or promote products derived from this software 17 * without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 29 * SUCH DAMAGE. 30 */ 31 32 #include <sys/cdefs.h> 33 #if defined(LIBC_SCCS) && !defined(lint) 34 #if 0 35 static char sccsid[] = "@(#)sethostent.c 8.1 (Berkeley) 6/4/93"; 36 static char rcsid[] = "Id: sethostent.c,v 8.5 1996/09/28 06:51:07 vixie Exp "; 37 #else 38 __RCSID("$NetBSD: sethostent.c,v 1.21 2023/12/07 22:33:49 kre Exp $"); 39 #endif 40 #endif /* LIBC_SCCS and not lint */ 41 42 #include "namespace.h" 43 #include <sys/param.h> 44 #include <netinet/in.h> 45 #include <arpa/nameser.h> 46 #include <arpa/inet.h> 47 #include <assert.h> 48 #include <string.h> 49 #include <nsswitch.h> 50 #include <netdb.h> 51 #include <resolv.h> 52 #include <errno.h> 53 #include <stdlib.h> 54 55 #include "hostent.h" 56 57 #ifdef __weak_alias 58 __weak_alias(sethostent,_sethostent) 59 __weak_alias(endhostent,_endhostent) 60 #endif 61 62 #ifndef _REENTRANT 63 void res_close(void); 64 #endif 65 66 static struct hostent *_hf_gethtbyname2(const char *, int, struct getnamaddr *); 67 68 void 69 /*ARGSUSED*/ 70 sethostent(int stayopen) 71 { 72 #ifndef _REENTRANT 73 if ((_res.options & RES_INIT) == 0 && res_init() == -1) 74 return; 75 if (stayopen) 76 _res.options |= RES_STAYOPEN | RES_USEVC; 77 #endif 78 sethostent_r(&_h_file); 79 } 80 81 void 82 endhostent(void) 83 { 84 #ifndef _REENTRANT 85 _res.options &= ~(RES_STAYOPEN | RES_USEVC); 86 res_close(); 87 #endif 88 endhostent_r(&_h_file); 89 } 90 static const char *_h_hosts = _PATH_HOSTS; 91 92 void 93 _hf_sethostsfile(const char *f) { 94 _h_hosts = f; 95 } 96 97 void 98 sethostent_r(FILE **hf) 99 { 100 if (!*hf) 101 *hf = fopen(_h_hosts, "re"); 102 else 103 rewind(*hf); 104 } 105 106 void 107 endhostent_r(FILE **hf) 108 { 109 if (*hf) { 110 (void)fclose(*hf); 111 *hf = NULL; 112 } 113 } 114 115 /*ARGSUSED*/ 116 int 117 _hf_gethtbyname(void *rv, void *cb_data, va_list ap) 118 { 119 struct hostent *hp; 120 const char *name; 121 int af; 122 struct getnamaddr *info = rv; 123 124 _DIAGASSERT(rv != NULL); 125 126 name = va_arg(ap, char *); 127 /* NOSTRICT skip string len */(void)va_arg(ap, int); 128 af = va_arg(ap, int); 129 130 #if 0 131 { 132 res_state res = __res_get_state(); 133 if (res == NULL) 134 return NS_NOTFOUND; 135 if (res->options & RES_USE_INET6) 136 hp = _hf_gethtbyname2(name, AF_INET6, info); 137 else 138 hp = NULL; 139 if (hp == NULL) 140 hp = _hf_gethtbyname2(name, AF_INET, info); 141 __res_put_state(res); 142 } 143 #else 144 hp = _hf_gethtbyname2(name, af, info); 145 #endif 146 if (hp == NULL) { 147 *info->he = HOST_NOT_FOUND; 148 return NS_NOTFOUND; 149 } 150 return NS_SUCCESS; 151 } 152 153 struct hostent * 154 _hf_gethtbyname2(const char *name, int af, struct getnamaddr *info) 155 { 156 struct hostent *hp, hent; 157 char *buf, *ptr; 158 size_t len, anum, num, i; 159 FILE *hf; 160 char *aliases[MAXALIASES]; 161 char *addr_ptrs[MAXADDRS]; 162 163 _DIAGASSERT(name != NULL); 164 165 hf = NULL; 166 sethostent_r(&hf); 167 if (hf == NULL) { 168 errno = EINVAL; 169 *info->he = NETDB_INTERNAL; 170 return NULL; 171 } 172 173 if ((ptr = buf = malloc(len = info->buflen)) == NULL) { 174 endhostent_r(&hf); 175 *info->he = NETDB_INTERNAL; 176 return NULL; 177 } 178 179 anum = 0; /* XXX: gcc */ 180 hent.h_name = NULL; /* XXX: gcc */ 181 hent.h_addrtype = 0; /* XXX: gcc */ 182 hent.h_length = 0; /* XXX: gcc */ 183 184 for (num = 0; num < MAXADDRS;) { 185 info->hp->h_addrtype = af; 186 info->hp->h_length = 0; 187 188 hp = gethostent_r(hf, info->hp, info->buf, info->buflen, 189 info->he); 190 if (hp == NULL) 191 break; 192 193 if (strcasecmp(hp->h_name, name) != 0) { 194 char **cp; 195 for (cp = hp->h_aliases; *cp != NULL; cp++) 196 if (strcasecmp(*cp, name) == 0) 197 break; 198 if (*cp == NULL) continue; 199 } 200 201 if (num == 0) { 202 hent.h_addrtype = af = hp->h_addrtype; 203 hent.h_length = hp->h_length; 204 205 HENT_SCOPY(hent.h_name, hp->h_name, ptr, len); 206 for (anum = 0; hp->h_aliases[anum]; anum++) { 207 if (anum >= __arraycount(aliases)) 208 goto nospc; 209 HENT_SCOPY(aliases[anum], hp->h_aliases[anum], 210 ptr, len); 211 } 212 ptr = (void *)ALIGN(ptr); 213 if ((size_t)(ptr - buf) >= info->buflen) 214 goto nospc; 215 } 216 217 if (num >= __arraycount(addr_ptrs)) 218 goto nospc; 219 HENT_COPY(addr_ptrs[num], hp->h_addr_list[0], hp->h_length, ptr, 220 len); 221 num++; 222 } 223 endhostent_r(&hf); 224 225 if (num == 0) { 226 *info->he = HOST_NOT_FOUND; 227 free(buf); 228 return NULL; 229 } 230 231 hp = info->hp; 232 ptr = info->buf; 233 len = info->buflen; 234 235 hp->h_addrtype = hent.h_addrtype; 236 hp->h_length = hent.h_length; 237 238 HENT_ARRAY(hp->h_aliases, anum, ptr, len); 239 HENT_ARRAY(hp->h_addr_list, num, ptr, len); 240 241 for (i = 0; i < num; i++) 242 HENT_COPY(hp->h_addr_list[i], addr_ptrs[i], hp->h_length, ptr, 243 len); 244 hp->h_addr_list[num] = NULL; 245 246 HENT_SCOPY(hp->h_name, hent.h_name, ptr, len); 247 248 for (i = 0; i < anum; i++) 249 HENT_SCOPY(hp->h_aliases[i], aliases[i], ptr, len); 250 hp->h_aliases[anum] = NULL; 251 252 free(buf); 253 return hp; 254 nospc: 255 endhostent_r(&hf); 256 *info->he = NETDB_INTERNAL; 257 free(buf); 258 errno = ENOSPC; 259 return NULL; 260 } 261 262 /*ARGSUSED*/ 263 int 264 _hf_gethtbyaddr(void *rv, void *cb_data, va_list ap) 265 { 266 struct hostent *hp; 267 const unsigned char *addr; 268 struct getnamaddr *info = rv; 269 FILE *hf; 270 271 _DIAGASSERT(rv != NULL); 272 273 addr = va_arg(ap, unsigned char *); 274 info->hp->h_length = va_arg(ap, int); 275 info->hp->h_addrtype = va_arg(ap, int); 276 277 hf = NULL; 278 sethostent_r(&hf); 279 if (hf == NULL) { 280 *info->he = NETDB_INTERNAL; 281 return NS_UNAVAIL; 282 } 283 while ((hp = gethostent_r(hf, info->hp, info->buf, info->buflen, 284 info->he)) != NULL) 285 if (!memcmp(hp->h_addr_list[0], addr, (size_t)hp->h_length)) 286 break; 287 endhostent_r(&hf); 288 289 if (hp == NULL) { 290 *info->he = HOST_NOT_FOUND; 291 return NS_NOTFOUND; 292 } 293 return NS_SUCCESS; 294 } 295