res_send.c revision 1.1.1.2 1 1.1.1.2 christos /* $NetBSD: res_send.c,v 1.1.1.2 2004/11/07 01:49:19 christos Exp $ */
2 1.1 christos
3 1.1 christos /*
4 1.1 christos * Copyright (c) 1985, 1989, 1993
5 1.1 christos * The Regents of the University of California. All rights reserved.
6 1.1 christos *
7 1.1 christos * Redistribution and use in source and binary forms, with or without
8 1.1 christos * modification, are permitted provided that the following conditions
9 1.1 christos * are met:
10 1.1 christos * 1. Redistributions of source code must retain the above copyright
11 1.1 christos * notice, this list of conditions and the following disclaimer.
12 1.1 christos * 2. Redistributions in binary form must reproduce the above copyright
13 1.1 christos * notice, this list of conditions and the following disclaimer in the
14 1.1 christos * documentation and/or other materials provided with the distribution.
15 1.1 christos * 3. All advertising materials mentioning features or use of this software
16 1.1 christos * must display the following acknowledgement:
17 1.1 christos * This product includes software developed by the University of
18 1.1 christos * California, Berkeley and its contributors.
19 1.1 christos * 4. Neither the name of the University nor the names of its contributors
20 1.1 christos * may be used to endorse or promote products derived from this software
21 1.1 christos * without specific prior written permission.
22 1.1 christos *
23 1.1 christos * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24 1.1 christos * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 1.1 christos * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 1.1 christos * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27 1.1 christos * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28 1.1 christos * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 1.1 christos * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 1.1 christos * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 1.1 christos * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32 1.1 christos * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 1.1 christos * SUCH DAMAGE.
34 1.1 christos */
35 1.1 christos
36 1.1 christos /*
37 1.1 christos * Portions Copyright (c) 1993 by Digital Equipment Corporation.
38 1.1 christos *
39 1.1 christos * Permission to use, copy, modify, and distribute this software for any
40 1.1 christos * purpose with or without fee is hereby granted, provided that the above
41 1.1 christos * copyright notice and this permission notice appear in all copies, and that
42 1.1 christos * the name of Digital Equipment Corporation not be used in advertising or
43 1.1 christos * publicity pertaining to distribution of the document or software without
44 1.1 christos * specific, written prior permission.
45 1.1 christos *
46 1.1 christos * THE SOFTWARE IS PROVIDED "AS IS" AND DIGITAL EQUIPMENT CORP. DISCLAIMS ALL
47 1.1 christos * WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES
48 1.1 christos * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL DIGITAL EQUIPMENT
49 1.1 christos * CORPORATION BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
50 1.1 christos * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
51 1.1 christos * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
52 1.1 christos * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
53 1.1 christos * SOFTWARE.
54 1.1 christos */
55 1.1 christos
56 1.1 christos /*
57 1.1 christos * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
58 1.1 christos * Portions Copyright (c) 1996-1999 by Internet Software Consortium.
59 1.1 christos *
60 1.1 christos * Permission to use, copy, modify, and distribute this software for any
61 1.1 christos * purpose with or without fee is hereby granted, provided that the above
62 1.1 christos * copyright notice and this permission notice appear in all copies.
63 1.1 christos *
64 1.1 christos * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
65 1.1 christos * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
66 1.1 christos * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR
67 1.1 christos * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
68 1.1 christos * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
69 1.1 christos * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
70 1.1 christos * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
71 1.1 christos */
72 1.1 christos
73 1.1 christos #if defined(LIBC_SCCS) && !defined(lint)
74 1.1 christos static const char sccsid[] = "@(#)res_send.c 8.1 (Berkeley) 6/4/93";
75 1.1.1.2 christos static const char rcsid[] = "Id: res_send.c,v 1.5.2.2.4.5 2004/08/10 02:19:56 marka Exp";
76 1.1 christos #endif /* LIBC_SCCS and not lint */
77 1.1 christos
78 1.1 christos /*
79 1.1 christos * Send query to name server and wait for reply.
80 1.1 christos */
81 1.1 christos
82 1.1 christos #include "port_before.h"
83 1.1 christos #include "fd_setsize.h"
84 1.1 christos
85 1.1 christos #include <sys/types.h>
86 1.1 christos #include <sys/param.h>
87 1.1 christos #include <sys/time.h>
88 1.1 christos #include <sys/socket.h>
89 1.1 christos #include <sys/uio.h>
90 1.1 christos
91 1.1 christos #include <netinet/in.h>
92 1.1 christos #include <arpa/nameser.h>
93 1.1 christos #include <arpa/inet.h>
94 1.1 christos
95 1.1 christos #include <errno.h>
96 1.1 christos #include <netdb.h>
97 1.1 christos #include <resolv.h>
98 1.1 christos #include <signal.h>
99 1.1 christos #include <stdio.h>
100 1.1 christos #include <stdlib.h>
101 1.1 christos #include <string.h>
102 1.1 christos #include <unistd.h>
103 1.1 christos
104 1.1 christos #include <isc/eventlib.h>
105 1.1 christos
106 1.1 christos #include "port_after.h"
107 1.1 christos
108 1.1 christos /* Options. Leave them on. */
109 1.1 christos #define DEBUG
110 1.1 christos #include "res_debug.h"
111 1.1 christos #include "res_private.h"
112 1.1 christos
113 1.1 christos #define EXT(res) ((res)->_u._ext)
114 1.1 christos
115 1.1 christos static const int highestFD = FD_SETSIZE - 1;
116 1.1 christos
117 1.1 christos /* Forward. */
118 1.1 christos
119 1.1 christos static int get_salen __P((const struct sockaddr *));
120 1.1 christos static struct sockaddr * get_nsaddr __P((res_state, size_t));
121 1.1 christos static int send_vc(res_state, const u_char *, int,
122 1.1 christos u_char *, int, int *, int);
123 1.1 christos static int send_dg(res_state, const u_char *, int,
124 1.1 christos u_char *, int, int *, int,
125 1.1 christos int *, int *);
126 1.1 christos static void Aerror(const res_state, FILE *, const char *, int,
127 1.1 christos const struct sockaddr *, int);
128 1.1 christos static void Perror(const res_state, FILE *, const char *, int);
129 1.1 christos static int sock_eq(struct sockaddr *, struct sockaddr *);
130 1.1 christos #ifdef NEED_PSELECT
131 1.1 christos static int pselect(int, void *, void *, void *,
132 1.1 christos struct timespec *,
133 1.1 christos const sigset_t *);
134 1.1 christos #endif
135 1.1 christos void res_pquery(const res_state, const u_char *, int, FILE *);
136 1.1 christos
137 1.1 christos static const int niflags = NI_NUMERICHOST | NI_NUMERICSERV;
138 1.1 christos
139 1.1 christos /* Public. */
140 1.1 christos
141 1.1 christos /* int
142 1.1 christos * res_isourserver(ina)
143 1.1 christos * looks up "ina" in _res.ns_addr_list[]
144 1.1 christos * returns:
145 1.1 christos * 0 : not found
146 1.1 christos * >0 : found
147 1.1 christos * author:
148 1.1 christos * paul vixie, 29may94
149 1.1 christos */
150 1.1 christos int
151 1.1 christos res_ourserver_p(const res_state statp, const struct sockaddr *sa) {
152 1.1 christos const struct sockaddr_in *inp, *srv;
153 1.1 christos const struct sockaddr_in6 *in6p, *srv6;
154 1.1 christos int ns;
155 1.1 christos
156 1.1 christos switch (sa->sa_family) {
157 1.1 christos case AF_INET:
158 1.1 christos inp = (const struct sockaddr_in *)sa;
159 1.1 christos for (ns = 0; ns < statp->nscount; ns++) {
160 1.1 christos srv = (struct sockaddr_in *)get_nsaddr(statp, ns);
161 1.1 christos if (srv->sin_family == inp->sin_family &&
162 1.1 christos srv->sin_port == inp->sin_port &&
163 1.1 christos (srv->sin_addr.s_addr == INADDR_ANY ||
164 1.1 christos srv->sin_addr.s_addr == inp->sin_addr.s_addr))
165 1.1 christos return (1);
166 1.1 christos }
167 1.1 christos break;
168 1.1 christos case AF_INET6:
169 1.1 christos if (EXT(statp).ext == NULL)
170 1.1 christos break;
171 1.1 christos in6p = (const struct sockaddr_in6 *)sa;
172 1.1 christos for (ns = 0; ns < statp->nscount; ns++) {
173 1.1 christos srv6 = (struct sockaddr_in6 *)get_nsaddr(statp, ns);
174 1.1 christos if (srv6->sin6_family == in6p->sin6_family &&
175 1.1 christos srv6->sin6_port == in6p->sin6_port &&
176 1.1 christos #ifdef HAVE_SIN6_SCOPE_ID
177 1.1.1.2 christos (srv6->sin6_scope_id == 0 ||
178 1.1.1.2 christos srv6->sin6_scope_id == in6p->sin6_scope_id) &&
179 1.1 christos #endif
180 1.1 christos (IN6_IS_ADDR_UNSPECIFIED(&srv6->sin6_addr) ||
181 1.1 christos IN6_ARE_ADDR_EQUAL(&srv6->sin6_addr, &in6p->sin6_addr)))
182 1.1 christos return (1);
183 1.1 christos }
184 1.1 christos break;
185 1.1 christos default:
186 1.1 christos break;
187 1.1 christos }
188 1.1 christos return (0);
189 1.1 christos }
190 1.1 christos
191 1.1 christos /* int
192 1.1 christos * res_nameinquery(name, type, class, buf, eom)
193 1.1 christos * look for (name,type,class) in the query section of packet (buf,eom)
194 1.1 christos * requires:
195 1.1 christos * buf + HFIXEDSZ <= eom
196 1.1 christos * returns:
197 1.1 christos * -1 : format error
198 1.1 christos * 0 : not found
199 1.1 christos * >0 : found
200 1.1 christos * author:
201 1.1 christos * paul vixie, 29may94
202 1.1 christos */
203 1.1 christos int
204 1.1 christos res_nameinquery(const char *name, int type, int class,
205 1.1 christos const u_char *buf, const u_char *eom)
206 1.1 christos {
207 1.1 christos const u_char *cp = buf + HFIXEDSZ;
208 1.1 christos int qdcount = ntohs(((const HEADER*)buf)->qdcount);
209 1.1 christos
210 1.1 christos while (qdcount-- > 0) {
211 1.1 christos char tname[MAXDNAME+1];
212 1.1 christos int n, ttype, tclass;
213 1.1 christos
214 1.1 christos n = dn_expand(buf, eom, cp, tname, sizeof tname);
215 1.1 christos if (n < 0)
216 1.1 christos return (-1);
217 1.1 christos cp += n;
218 1.1 christos if (cp + 2 * INT16SZ > eom)
219 1.1 christos return (-1);
220 1.1 christos ttype = ns_get16(cp); cp += INT16SZ;
221 1.1 christos tclass = ns_get16(cp); cp += INT16SZ;
222 1.1 christos if (ttype == type && tclass == class &&
223 1.1 christos ns_samename(tname, name) == 1)
224 1.1 christos return (1);
225 1.1 christos }
226 1.1 christos return (0);
227 1.1 christos }
228 1.1 christos
229 1.1 christos /* int
230 1.1 christos * res_queriesmatch(buf1, eom1, buf2, eom2)
231 1.1 christos * is there a 1:1 mapping of (name,type,class)
232 1.1 christos * in (buf1,eom1) and (buf2,eom2)?
233 1.1 christos * returns:
234 1.1 christos * -1 : format error
235 1.1 christos * 0 : not a 1:1 mapping
236 1.1 christos * >0 : is a 1:1 mapping
237 1.1 christos * author:
238 1.1 christos * paul vixie, 29may94
239 1.1 christos */
240 1.1 christos int
241 1.1 christos res_queriesmatch(const u_char *buf1, const u_char *eom1,
242 1.1 christos const u_char *buf2, const u_char *eom2)
243 1.1 christos {
244 1.1 christos const u_char *cp = buf1 + HFIXEDSZ;
245 1.1 christos int qdcount = ntohs(((const HEADER*)buf1)->qdcount);
246 1.1 christos
247 1.1 christos if (buf1 + HFIXEDSZ > eom1 || buf2 + HFIXEDSZ > eom2)
248 1.1 christos return (-1);
249 1.1 christos
250 1.1 christos /*
251 1.1 christos * Only header section present in replies to
252 1.1 christos * dynamic update packets.
253 1.1 christos */
254 1.1 christos if ((((const HEADER *)buf1)->opcode == ns_o_update) &&
255 1.1 christos (((const HEADER *)buf2)->opcode == ns_o_update))
256 1.1 christos return (1);
257 1.1 christos
258 1.1 christos if (qdcount != ntohs(((const HEADER*)buf2)->qdcount))
259 1.1 christos return (0);
260 1.1 christos while (qdcount-- > 0) {
261 1.1 christos char tname[MAXDNAME+1];
262 1.1 christos int n, ttype, tclass;
263 1.1 christos
264 1.1 christos n = dn_expand(buf1, eom1, cp, tname, sizeof tname);
265 1.1 christos if (n < 0)
266 1.1 christos return (-1);
267 1.1 christos cp += n;
268 1.1 christos if (cp + 2 * INT16SZ > eom1)
269 1.1 christos return (-1);
270 1.1 christos ttype = ns_get16(cp); cp += INT16SZ;
271 1.1 christos tclass = ns_get16(cp); cp += INT16SZ;
272 1.1 christos if (!res_nameinquery(tname, ttype, tclass, buf2, eom2))
273 1.1 christos return (0);
274 1.1 christos }
275 1.1 christos return (1);
276 1.1 christos }
277 1.1 christos
278 1.1 christos int
279 1.1 christos res_nsend(res_state statp,
280 1.1 christos const u_char *buf, int buflen, u_char *ans, int anssiz)
281 1.1 christos {
282 1.1 christos int gotsomewhere, terrno, try, v_circuit, resplen, ns, n;
283 1.1 christos char abuf[NI_MAXHOST];
284 1.1 christos
285 1.1 christos if (statp->nscount == 0) {
286 1.1 christos errno = ESRCH;
287 1.1 christos return (-1);
288 1.1 christos }
289 1.1 christos if (anssiz < HFIXEDSZ) {
290 1.1 christos errno = EINVAL;
291 1.1 christos return (-1);
292 1.1 christos }
293 1.1 christos DprintQ((statp->options & RES_DEBUG) || (statp->pfcode & RES_PRF_QUERY),
294 1.1 christos (stdout, ";; res_send()\n"), buf, buflen);
295 1.1 christos v_circuit = (statp->options & RES_USEVC) || buflen > PACKETSZ;
296 1.1 christos gotsomewhere = 0;
297 1.1 christos terrno = ETIMEDOUT;
298 1.1 christos
299 1.1 christos /*
300 1.1 christos * If the ns_addr_list in the resolver context has changed, then
301 1.1 christos * invalidate our cached copy and the associated timing data.
302 1.1 christos */
303 1.1 christos if (EXT(statp).nscount != 0) {
304 1.1 christos int needclose = 0;
305 1.1 christos struct sockaddr_storage peer;
306 1.1 christos ISC_SOCKLEN_T peerlen;
307 1.1 christos
308 1.1 christos if (EXT(statp).nscount != statp->nscount)
309 1.1 christos needclose++;
310 1.1 christos else
311 1.1 christos for (ns = 0; ns < statp->nscount; ns++) {
312 1.1 christos if (statp->nsaddr_list[ns].sin_family &&
313 1.1 christos !sock_eq((struct sockaddr *)&statp->nsaddr_list[ns],
314 1.1 christos (struct sockaddr *)&EXT(statp).ext->nsaddrs[ns])) {
315 1.1 christos needclose++;
316 1.1 christos break;
317 1.1 christos }
318 1.1 christos
319 1.1 christos if (EXT(statp).nssocks[ns] == -1)
320 1.1 christos continue;
321 1.1 christos peerlen = sizeof(peer);
322 1.1 christos if (getsockname(EXT(statp).nssocks[ns],
323 1.1 christos (struct sockaddr *)&peer, &peerlen) < 0) {
324 1.1 christos needclose++;
325 1.1 christos break;
326 1.1 christos }
327 1.1 christos if (!sock_eq((struct sockaddr *)&peer,
328 1.1 christos get_nsaddr(statp, ns))) {
329 1.1 christos needclose++;
330 1.1 christos break;
331 1.1 christos }
332 1.1 christos }
333 1.1 christos if (needclose) {
334 1.1 christos res_nclose(statp);
335 1.1 christos EXT(statp).nscount = 0;
336 1.1 christos }
337 1.1 christos }
338 1.1 christos
339 1.1 christos /*
340 1.1 christos * Maybe initialize our private copy of the ns_addr_list.
341 1.1 christos */
342 1.1 christos if (EXT(statp).nscount == 0) {
343 1.1 christos for (ns = 0; ns < statp->nscount; ns++) {
344 1.1 christos EXT(statp).nstimes[ns] = RES_MAXTIME;
345 1.1 christos EXT(statp).nssocks[ns] = -1;
346 1.1 christos if (!statp->nsaddr_list[ns].sin_family)
347 1.1 christos continue;
348 1.1 christos EXT(statp).ext->nsaddrs[ns].sin =
349 1.1 christos statp->nsaddr_list[ns];
350 1.1 christos }
351 1.1 christos EXT(statp).nscount = statp->nscount;
352 1.1 christos }
353 1.1 christos
354 1.1 christos /*
355 1.1 christos * Some resolvers want to even out the load on their nameservers.
356 1.1 christos * Note that RES_BLAST overrides RES_ROTATE.
357 1.1 christos */
358 1.1 christos if ((statp->options & RES_ROTATE) != 0U &&
359 1.1 christos (statp->options & RES_BLAST) == 0U) {
360 1.1 christos union res_sockaddr_union inu;
361 1.1 christos struct sockaddr_in ina;
362 1.1 christos int lastns = statp->nscount - 1;
363 1.1 christos int fd;
364 1.1 christos u_int16_t nstime;
365 1.1 christos
366 1.1 christos if (EXT(statp).ext != NULL)
367 1.1 christos inu = EXT(statp).ext->nsaddrs[0];
368 1.1 christos ina = statp->nsaddr_list[0];
369 1.1 christos fd = EXT(statp).nssocks[0];
370 1.1 christos nstime = EXT(statp).nstimes[0];
371 1.1 christos for (ns = 0; ns < lastns; ns++) {
372 1.1 christos if (EXT(statp).ext != NULL)
373 1.1 christos EXT(statp).ext->nsaddrs[ns] =
374 1.1 christos EXT(statp).ext->nsaddrs[ns + 1];
375 1.1 christos statp->nsaddr_list[ns] = statp->nsaddr_list[ns + 1];
376 1.1 christos EXT(statp).nssocks[ns] = EXT(statp).nssocks[ns + 1];
377 1.1 christos EXT(statp).nstimes[ns] = EXT(statp).nstimes[ns + 1];
378 1.1 christos }
379 1.1 christos if (EXT(statp).ext != NULL)
380 1.1 christos EXT(statp).ext->nsaddrs[lastns] = inu;
381 1.1 christos statp->nsaddr_list[lastns] = ina;
382 1.1 christos EXT(statp).nssocks[lastns] = fd;
383 1.1 christos EXT(statp).nstimes[lastns] = nstime;
384 1.1 christos }
385 1.1 christos
386 1.1 christos /*
387 1.1 christos * Send request, RETRY times, or until successful.
388 1.1 christos */
389 1.1 christos for (try = 0; try < statp->retry; try++) {
390 1.1 christos for (ns = 0; ns < statp->nscount; ns++) {
391 1.1 christos struct sockaddr *nsap;
392 1.1 christos int nsaplen;
393 1.1 christos nsap = get_nsaddr(statp, ns);
394 1.1 christos nsaplen = get_salen(nsap);
395 1.1 christos statp->_flags &= ~RES_F_LASTMASK;
396 1.1 christos statp->_flags |= (ns << RES_F_LASTSHIFT);
397 1.1 christos same_ns:
398 1.1 christos if (statp->qhook) {
399 1.1 christos int done = 0, loops = 0;
400 1.1 christos
401 1.1 christos do {
402 1.1 christos res_sendhookact act;
403 1.1 christos
404 1.1 christos act = (*statp->qhook)(&nsap, &buf, &buflen,
405 1.1 christos ans, anssiz, &resplen);
406 1.1 christos switch (act) {
407 1.1 christos case res_goahead:
408 1.1 christos done = 1;
409 1.1 christos break;
410 1.1 christos case res_nextns:
411 1.1 christos res_nclose(statp);
412 1.1 christos goto next_ns;
413 1.1 christos case res_done:
414 1.1 christos return (resplen);
415 1.1 christos case res_modified:
416 1.1 christos /* give the hook another try */
417 1.1 christos if (++loops < 42) /*doug adams*/
418 1.1 christos break;
419 1.1 christos /*FALLTHROUGH*/
420 1.1 christos case res_error:
421 1.1 christos /*FALLTHROUGH*/
422 1.1 christos default:
423 1.1 christos goto fail;
424 1.1 christos }
425 1.1 christos } while (!done);
426 1.1 christos }
427 1.1 christos
428 1.1 christos Dprint(((statp->options & RES_DEBUG) &&
429 1.1 christos getnameinfo(nsap, nsaplen, abuf, sizeof(abuf),
430 1.1 christos NULL, 0, niflags) == 0),
431 1.1 christos (stdout, ";; Querying server (# %d) address = %s\n",
432 1.1 christos ns + 1, abuf));
433 1.1 christos
434 1.1 christos
435 1.1 christos if (v_circuit) {
436 1.1 christos /* Use VC; at most one attempt per server. */
437 1.1 christos try = statp->retry;
438 1.1 christos n = send_vc(statp, buf, buflen, ans, anssiz, &terrno,
439 1.1 christos ns);
440 1.1 christos if (n < 0)
441 1.1 christos goto fail;
442 1.1 christos if (n == 0)
443 1.1 christos goto next_ns;
444 1.1 christos resplen = n;
445 1.1 christos } else {
446 1.1 christos /* Use datagrams. */
447 1.1 christos n = send_dg(statp, buf, buflen, ans, anssiz, &terrno,
448 1.1 christos ns, &v_circuit, &gotsomewhere);
449 1.1 christos if (n < 0)
450 1.1 christos goto fail;
451 1.1 christos if (n == 0)
452 1.1 christos goto next_ns;
453 1.1 christos if (v_circuit)
454 1.1 christos goto same_ns;
455 1.1 christos resplen = n;
456 1.1 christos }
457 1.1 christos
458 1.1 christos Dprint((statp->options & RES_DEBUG) ||
459 1.1 christos ((statp->pfcode & RES_PRF_REPLY) &&
460 1.1 christos (statp->pfcode & RES_PRF_HEAD1)),
461 1.1 christos (stdout, ";; got answer:\n"));
462 1.1 christos
463 1.1 christos DprintQ((statp->options & RES_DEBUG) ||
464 1.1 christos (statp->pfcode & RES_PRF_REPLY),
465 1.1 christos (stdout, "%s", ""),
466 1.1 christos ans, (resplen > anssiz) ? anssiz : resplen);
467 1.1 christos
468 1.1 christos /*
469 1.1 christos * If we have temporarily opened a virtual circuit,
470 1.1 christos * or if we haven't been asked to keep a socket open,
471 1.1 christos * close the socket.
472 1.1 christos */
473 1.1 christos if ((v_circuit && (statp->options & RES_USEVC) == 0U) ||
474 1.1 christos (statp->options & RES_STAYOPEN) == 0U) {
475 1.1 christos res_nclose(statp);
476 1.1 christos }
477 1.1 christos if (statp->rhook) {
478 1.1 christos int done = 0, loops = 0;
479 1.1 christos
480 1.1 christos do {
481 1.1 christos res_sendhookact act;
482 1.1 christos
483 1.1 christos act = (*statp->rhook)(nsap, buf, buflen,
484 1.1 christos ans, anssiz, &resplen);
485 1.1 christos switch (act) {
486 1.1 christos case res_goahead:
487 1.1 christos case res_done:
488 1.1 christos done = 1;
489 1.1 christos break;
490 1.1 christos case res_nextns:
491 1.1 christos res_nclose(statp);
492 1.1 christos goto next_ns;
493 1.1 christos case res_modified:
494 1.1 christos /* give the hook another try */
495 1.1 christos if (++loops < 42) /*doug adams*/
496 1.1 christos break;
497 1.1 christos /*FALLTHROUGH*/
498 1.1 christos case res_error:
499 1.1 christos /*FALLTHROUGH*/
500 1.1 christos default:
501 1.1 christos goto fail;
502 1.1 christos }
503 1.1 christos } while (!done);
504 1.1 christos
505 1.1 christos }
506 1.1 christos return (resplen);
507 1.1 christos next_ns: ;
508 1.1 christos } /*foreach ns*/
509 1.1 christos } /*foreach retry*/
510 1.1 christos res_nclose(statp);
511 1.1 christos if (!v_circuit) {
512 1.1 christos if (!gotsomewhere)
513 1.1 christos errno = ECONNREFUSED; /* no nameservers found */
514 1.1 christos else
515 1.1 christos errno = ETIMEDOUT; /* no answer obtained */
516 1.1 christos } else
517 1.1 christos errno = terrno;
518 1.1 christos return (-1);
519 1.1 christos fail:
520 1.1 christos res_nclose(statp);
521 1.1 christos return (-1);
522 1.1 christos }
523 1.1 christos
524 1.1 christos /* Private */
525 1.1 christos
526 1.1 christos static int
527 1.1 christos get_salen(sa)
528 1.1 christos const struct sockaddr *sa;
529 1.1 christos {
530 1.1 christos
531 1.1 christos #ifdef HAVE_SA_LEN
532 1.1 christos /* There are people do not set sa_len. Be forgiving to them. */
533 1.1 christos if (sa->sa_len)
534 1.1 christos return (sa->sa_len);
535 1.1 christos #endif
536 1.1 christos
537 1.1 christos if (sa->sa_family == AF_INET)
538 1.1 christos return (sizeof(struct sockaddr_in));
539 1.1 christos else if (sa->sa_family == AF_INET6)
540 1.1 christos return (sizeof(struct sockaddr_in6));
541 1.1 christos else
542 1.1 christos return (0); /* unknown, die on connect */
543 1.1 christos }
544 1.1 christos
545 1.1 christos /*
546 1.1 christos * pick appropriate nsaddr_list for use. see res_init() for initialization.
547 1.1 christos */
548 1.1 christos static struct sockaddr *
549 1.1 christos get_nsaddr(statp, n)
550 1.1 christos res_state statp;
551 1.1 christos size_t n;
552 1.1 christos {
553 1.1 christos
554 1.1 christos if (!statp->nsaddr_list[n].sin_family && EXT(statp).ext) {
555 1.1 christos /*
556 1.1 christos * - EXT(statp).ext->nsaddrs[n] holds an address that is larger
557 1.1 christos * than struct sockaddr, and
558 1.1 christos * - user code did not update statp->nsaddr_list[n].
559 1.1 christos */
560 1.1 christos return (struct sockaddr *)(void *)&EXT(statp).ext->nsaddrs[n];
561 1.1 christos } else {
562 1.1 christos /*
563 1.1 christos * - user code updated statp->nsaddr_list[n], or
564 1.1 christos * - statp->nsaddr_list[n] has the same content as
565 1.1 christos * EXT(statp).ext->nsaddrs[n].
566 1.1 christos */
567 1.1 christos return (struct sockaddr *)(void *)&statp->nsaddr_list[n];
568 1.1 christos }
569 1.1 christos }
570 1.1 christos
571 1.1 christos static int
572 1.1 christos send_vc(res_state statp,
573 1.1 christos const u_char *buf, int buflen, u_char *ans, int anssiz,
574 1.1 christos int *terrno, int ns)
575 1.1 christos {
576 1.1 christos const HEADER *hp = (const HEADER *) buf;
577 1.1 christos HEADER *anhp = (HEADER *) ans;
578 1.1 christos struct sockaddr *nsap;
579 1.1 christos int nsaplen;
580 1.1 christos int truncating, connreset, resplen, n;
581 1.1 christos struct iovec iov[2];
582 1.1 christos u_short len;
583 1.1 christos u_char *cp;
584 1.1 christos void *tmp;
585 1.1 christos
586 1.1 christos nsap = get_nsaddr(statp, ns);
587 1.1 christos nsaplen = get_salen(nsap);
588 1.1 christos
589 1.1 christos connreset = 0;
590 1.1 christos same_ns:
591 1.1 christos truncating = 0;
592 1.1 christos
593 1.1 christos /* Are we still talking to whom we want to talk to? */
594 1.1 christos if (statp->_vcsock >= 0 && (statp->_flags & RES_F_VC) != 0) {
595 1.1 christos struct sockaddr_storage peer;
596 1.1 christos ISC_SOCKLEN_T size = sizeof peer;
597 1.1 christos
598 1.1 christos if (getpeername(statp->_vcsock,
599 1.1 christos (struct sockaddr *)&peer, &size) < 0 ||
600 1.1 christos !sock_eq((struct sockaddr *)&peer, nsap)) {
601 1.1 christos res_nclose(statp);
602 1.1 christos statp->_flags &= ~RES_F_VC;
603 1.1 christos }
604 1.1 christos }
605 1.1 christos
606 1.1 christos if (statp->_vcsock < 0 || (statp->_flags & RES_F_VC) == 0) {
607 1.1 christos if (statp->_vcsock >= 0)
608 1.1 christos res_nclose(statp);
609 1.1 christos
610 1.1 christos statp->_vcsock = socket(nsap->sa_family, SOCK_STREAM, 0);
611 1.1 christos if (statp->_vcsock > highestFD) {
612 1.1 christos res_nclose(statp);
613 1.1 christos errno = ENOTSOCK;
614 1.1 christos }
615 1.1 christos if (statp->_vcsock < 0) {
616 1.1 christos switch (errno) {
617 1.1 christos case EPROTONOSUPPORT:
618 1.1 christos #ifdef EPFNOSUPPORT
619 1.1 christos case EPFNOSUPPORT:
620 1.1 christos #endif
621 1.1 christos case EAFNOSUPPORT:
622 1.1 christos Perror(statp, stderr, "socket(vc)", errno);
623 1.1 christos return (0);
624 1.1 christos default:
625 1.1 christos *terrno = errno;
626 1.1 christos Perror(statp, stderr, "socket(vc)", errno);
627 1.1 christos return (-1);
628 1.1 christos }
629 1.1 christos }
630 1.1 christos errno = 0;
631 1.1 christos if (connect(statp->_vcsock, nsap, nsaplen) < 0) {
632 1.1 christos *terrno = errno;
633 1.1 christos Aerror(statp, stderr, "connect/vc", errno, nsap,
634 1.1 christos nsaplen);
635 1.1 christos res_nclose(statp);
636 1.1 christos return (0);
637 1.1 christos }
638 1.1 christos statp->_flags |= RES_F_VC;
639 1.1 christos }
640 1.1 christos
641 1.1 christos /*
642 1.1 christos * Send length & message
643 1.1 christos */
644 1.1 christos ns_put16((u_short)buflen, (u_char*)&len);
645 1.1 christos iov[0] = evConsIovec(&len, INT16SZ);
646 1.1 christos DE_CONST(buf, tmp);
647 1.1 christos iov[1] = evConsIovec(tmp, buflen);
648 1.1 christos if (writev(statp->_vcsock, iov, 2) != (INT16SZ + buflen)) {
649 1.1 christos *terrno = errno;
650 1.1 christos Perror(statp, stderr, "write failed", errno);
651 1.1 christos res_nclose(statp);
652 1.1 christos return (0);
653 1.1 christos }
654 1.1 christos /*
655 1.1 christos * Receive length & response
656 1.1 christos */
657 1.1 christos read_len:
658 1.1 christos cp = ans;
659 1.1 christos len = INT16SZ;
660 1.1 christos while ((n = read(statp->_vcsock, (char *)cp, (int)len)) > 0) {
661 1.1 christos cp += n;
662 1.1.1.2 christos if ((len -= n) == 0)
663 1.1 christos break;
664 1.1 christos }
665 1.1 christos if (n <= 0) {
666 1.1 christos *terrno = errno;
667 1.1 christos Perror(statp, stderr, "read failed", errno);
668 1.1 christos res_nclose(statp);
669 1.1 christos /*
670 1.1 christos * A long running process might get its TCP
671 1.1 christos * connection reset if the remote server was
672 1.1 christos * restarted. Requery the server instead of
673 1.1 christos * trying a new one. When there is only one
674 1.1 christos * server, this means that a query might work
675 1.1 christos * instead of failing. We only allow one reset
676 1.1 christos * per query to prevent looping.
677 1.1 christos */
678 1.1 christos if (*terrno == ECONNRESET && !connreset) {
679 1.1 christos connreset = 1;
680 1.1 christos res_nclose(statp);
681 1.1 christos goto same_ns;
682 1.1 christos }
683 1.1 christos res_nclose(statp);
684 1.1 christos return (0);
685 1.1 christos }
686 1.1 christos resplen = ns_get16(ans);
687 1.1 christos if (resplen > anssiz) {
688 1.1 christos Dprint(statp->options & RES_DEBUG,
689 1.1 christos (stdout, ";; response truncated\n")
690 1.1 christos );
691 1.1 christos truncating = 1;
692 1.1 christos len = anssiz;
693 1.1 christos } else
694 1.1 christos len = resplen;
695 1.1 christos if (len < HFIXEDSZ) {
696 1.1 christos /*
697 1.1 christos * Undersized message.
698 1.1 christos */
699 1.1 christos Dprint(statp->options & RES_DEBUG,
700 1.1 christos (stdout, ";; undersized: %d\n", len));
701 1.1 christos *terrno = EMSGSIZE;
702 1.1 christos res_nclose(statp);
703 1.1 christos return (0);
704 1.1 christos }
705 1.1 christos cp = ans;
706 1.1 christos while (len != 0 && (n = read(statp->_vcsock, (char *)cp, (int)len)) > 0){
707 1.1 christos cp += n;
708 1.1 christos len -= n;
709 1.1 christos }
710 1.1 christos if (n <= 0) {
711 1.1 christos *terrno = errno;
712 1.1 christos Perror(statp, stderr, "read(vc)", errno);
713 1.1 christos res_nclose(statp);
714 1.1 christos return (0);
715 1.1 christos }
716 1.1 christos if (truncating) {
717 1.1 christos /*
718 1.1 christos * Flush rest of answer so connection stays in synch.
719 1.1 christos */
720 1.1 christos anhp->tc = 1;
721 1.1 christos len = resplen - anssiz;
722 1.1 christos while (len != 0) {
723 1.1 christos char junk[PACKETSZ];
724 1.1 christos
725 1.1 christos n = read(statp->_vcsock, junk,
726 1.1 christos (len > sizeof junk) ? sizeof junk : len);
727 1.1 christos if (n > 0)
728 1.1 christos len -= n;
729 1.1 christos else
730 1.1 christos break;
731 1.1 christos }
732 1.1 christos }
733 1.1 christos /*
734 1.1 christos * If the calling applicating has bailed out of
735 1.1 christos * a previous call and failed to arrange to have
736 1.1 christos * the circuit closed or the server has got
737 1.1 christos * itself confused, then drop the packet and
738 1.1 christos * wait for the correct one.
739 1.1 christos */
740 1.1 christos if (hp->id != anhp->id) {
741 1.1 christos DprintQ((statp->options & RES_DEBUG) ||
742 1.1 christos (statp->pfcode & RES_PRF_REPLY),
743 1.1 christos (stdout, ";; old answer (unexpected):\n"),
744 1.1 christos ans, (resplen > anssiz) ? anssiz: resplen);
745 1.1 christos goto read_len;
746 1.1 christos }
747 1.1 christos
748 1.1 christos /*
749 1.1 christos * All is well, or the error is fatal. Signal that the
750 1.1 christos * next nameserver ought not be tried.
751 1.1 christos */
752 1.1 christos return (resplen);
753 1.1 christos }
754 1.1 christos
755 1.1 christos static int
756 1.1 christos send_dg(res_state statp,
757 1.1 christos const u_char *buf, int buflen, u_char *ans, int anssiz,
758 1.1 christos int *terrno, int ns, int *v_circuit, int *gotsomewhere)
759 1.1 christos {
760 1.1 christos const HEADER *hp = (const HEADER *) buf;
761 1.1 christos HEADER *anhp = (HEADER *) ans;
762 1.1 christos const struct sockaddr *nsap;
763 1.1 christos int nsaplen;
764 1.1 christos struct timespec now, timeout, finish;
765 1.1 christos fd_set dsmask;
766 1.1 christos struct sockaddr_storage from;
767 1.1 christos ISC_SOCKLEN_T fromlen;
768 1.1 christos int resplen, seconds, n, s;
769 1.1 christos
770 1.1 christos nsap = get_nsaddr(statp, ns);
771 1.1 christos nsaplen = get_salen(nsap);
772 1.1 christos if (EXT(statp).nssocks[ns] == -1) {
773 1.1 christos EXT(statp).nssocks[ns] = socket(nsap->sa_family, SOCK_DGRAM, 0);
774 1.1 christos if (EXT(statp).nssocks[ns] > highestFD) {
775 1.1 christos res_nclose(statp);
776 1.1 christos errno = ENOTSOCK;
777 1.1 christos }
778 1.1 christos if (EXT(statp).nssocks[ns] < 0) {
779 1.1 christos switch (errno) {
780 1.1 christos case EPROTONOSUPPORT:
781 1.1 christos #ifdef EPFNOSUPPORT
782 1.1 christos case EPFNOSUPPORT:
783 1.1 christos #endif
784 1.1 christos case EAFNOSUPPORT:
785 1.1 christos Perror(statp, stderr, "socket(dg)", errno);
786 1.1 christos return (0);
787 1.1 christos default:
788 1.1 christos *terrno = errno;
789 1.1 christos Perror(statp, stderr, "socket(dg)", errno);
790 1.1 christos return (-1);
791 1.1 christos }
792 1.1 christos }
793 1.1 christos #ifndef CANNOT_CONNECT_DGRAM
794 1.1 christos /*
795 1.1 christos * On a 4.3BSD+ machine (client and server,
796 1.1 christos * actually), sending to a nameserver datagram
797 1.1 christos * port with no nameserver will cause an
798 1.1 christos * ICMP port unreachable message to be returned.
799 1.1 christos * If our datagram socket is "connected" to the
800 1.1 christos * server, we get an ECONNREFUSED error on the next
801 1.1 christos * socket operation, and select returns if the
802 1.1 christos * error message is received. We can thus detect
803 1.1 christos * the absence of a nameserver without timing out.
804 1.1 christos */
805 1.1 christos if (connect(EXT(statp).nssocks[ns], nsap, nsaplen) < 0) {
806 1.1 christos Aerror(statp, stderr, "connect(dg)", errno, nsap,
807 1.1 christos nsaplen);
808 1.1 christos res_nclose(statp);
809 1.1 christos return (0);
810 1.1 christos }
811 1.1 christos #endif /* !CANNOT_CONNECT_DGRAM */
812 1.1 christos Dprint(statp->options & RES_DEBUG,
813 1.1 christos (stdout, ";; new DG socket\n"))
814 1.1 christos }
815 1.1 christos s = EXT(statp).nssocks[ns];
816 1.1 christos #ifndef CANNOT_CONNECT_DGRAM
817 1.1 christos if (send(s, (const char*)buf, buflen, 0) != buflen) {
818 1.1 christos Perror(statp, stderr, "send", errno);
819 1.1 christos res_nclose(statp);
820 1.1 christos return (0);
821 1.1 christos }
822 1.1 christos #else /* !CANNOT_CONNECT_DGRAM */
823 1.1 christos if (sendto(s, (const char*)buf, buflen, 0, nsap, nsaplen) != buflen)
824 1.1 christos {
825 1.1 christos Aerror(statp, stderr, "sendto", errno, nsap, nsaplen);
826 1.1 christos res_nclose(statp);
827 1.1 christos return (0);
828 1.1 christos }
829 1.1 christos #endif /* !CANNOT_CONNECT_DGRAM */
830 1.1 christos
831 1.1 christos /*
832 1.1 christos * Wait for reply.
833 1.1 christos */
834 1.1 christos seconds = (statp->retrans << ns);
835 1.1 christos if (ns > 0)
836 1.1 christos seconds /= statp->nscount;
837 1.1 christos if (seconds <= 0)
838 1.1 christos seconds = 1;
839 1.1 christos now = evNowTime();
840 1.1 christos timeout = evConsTime(seconds, 0);
841 1.1 christos finish = evAddTime(now, timeout);
842 1.1 christos goto nonow;
843 1.1 christos wait:
844 1.1 christos now = evNowTime();
845 1.1 christos nonow:
846 1.1 christos FD_ZERO(&dsmask);
847 1.1 christos FD_SET(s, &dsmask);
848 1.1 christos if (evCmpTime(finish, now) > 0)
849 1.1 christos timeout = evSubTime(finish, now);
850 1.1 christos else
851 1.1 christos timeout = evConsTime(0, 0);
852 1.1 christos n = pselect(s + 1, &dsmask, NULL, NULL, &timeout, NULL);
853 1.1 christos if (n == 0) {
854 1.1 christos Dprint(statp->options & RES_DEBUG, (stdout, ";; timeout\n"));
855 1.1 christos *gotsomewhere = 1;
856 1.1 christos return (0);
857 1.1 christos }
858 1.1 christos if (n < 0) {
859 1.1 christos if (errno == EINTR)
860 1.1 christos goto wait;
861 1.1 christos Perror(statp, stderr, "select", errno);
862 1.1 christos res_nclose(statp);
863 1.1 christos return (0);
864 1.1 christos }
865 1.1 christos errno = 0;
866 1.1 christos fromlen = sizeof(from);
867 1.1 christos resplen = recvfrom(s, (char*)ans, anssiz,0,
868 1.1 christos (struct sockaddr *)&from, &fromlen);
869 1.1 christos if (resplen <= 0) {
870 1.1 christos Perror(statp, stderr, "recvfrom", errno);
871 1.1 christos res_nclose(statp);
872 1.1 christos return (0);
873 1.1 christos }
874 1.1 christos *gotsomewhere = 1;
875 1.1 christos if (resplen < HFIXEDSZ) {
876 1.1 christos /*
877 1.1 christos * Undersized message.
878 1.1 christos */
879 1.1 christos Dprint(statp->options & RES_DEBUG,
880 1.1 christos (stdout, ";; undersized: %d\n",
881 1.1 christos resplen));
882 1.1 christos *terrno = EMSGSIZE;
883 1.1 christos res_nclose(statp);
884 1.1 christos return (0);
885 1.1 christos }
886 1.1 christos if (hp->id != anhp->id) {
887 1.1 christos /*
888 1.1 christos * response from old query, ignore it.
889 1.1 christos * XXX - potential security hazard could
890 1.1 christos * be detected here.
891 1.1 christos */
892 1.1 christos DprintQ((statp->options & RES_DEBUG) ||
893 1.1 christos (statp->pfcode & RES_PRF_REPLY),
894 1.1 christos (stdout, ";; old answer:\n"),
895 1.1 christos ans, (resplen > anssiz) ? anssiz : resplen);
896 1.1 christos goto wait;
897 1.1 christos }
898 1.1 christos if (!(statp->options & RES_INSECURE1) &&
899 1.1 christos !res_ourserver_p(statp, (struct sockaddr *)&from)) {
900 1.1 christos /*
901 1.1 christos * response from wrong server? ignore it.
902 1.1 christos * XXX - potential security hazard could
903 1.1 christos * be detected here.
904 1.1 christos */
905 1.1 christos DprintQ((statp->options & RES_DEBUG) ||
906 1.1 christos (statp->pfcode & RES_PRF_REPLY),
907 1.1 christos (stdout, ";; not our server:\n"),
908 1.1 christos ans, (resplen > anssiz) ? anssiz : resplen);
909 1.1 christos goto wait;
910 1.1 christos }
911 1.1 christos #ifdef RES_USE_EDNS0
912 1.1 christos if (anhp->rcode == FORMERR && (statp->options & RES_USE_EDNS0) != 0U) {
913 1.1 christos /*
914 1.1 christos * Do not retry if the server do not understand EDNS0.
915 1.1 christos * The case has to be captured here, as FORMERR packet do not
916 1.1 christos * carry query section, hence res_queriesmatch() returns 0.
917 1.1 christos */
918 1.1 christos DprintQ(statp->options & RES_DEBUG,
919 1.1 christos (stdout, "server rejected query with EDNS0:\n"),
920 1.1 christos ans, (resplen > anssiz) ? anssiz : resplen);
921 1.1 christos /* record the error */
922 1.1 christos statp->_flags |= RES_F_EDNS0ERR;
923 1.1 christos res_nclose(statp);
924 1.1 christos return (0);
925 1.1 christos }
926 1.1 christos #endif
927 1.1 christos if (!(statp->options & RES_INSECURE2) &&
928 1.1 christos !res_queriesmatch(buf, buf + buflen,
929 1.1 christos ans, ans + anssiz)) {
930 1.1 christos /*
931 1.1 christos * response contains wrong query? ignore it.
932 1.1 christos * XXX - potential security hazard could
933 1.1 christos * be detected here.
934 1.1 christos */
935 1.1 christos DprintQ((statp->options & RES_DEBUG) ||
936 1.1 christos (statp->pfcode & RES_PRF_REPLY),
937 1.1 christos (stdout, ";; wrong query name:\n"),
938 1.1 christos ans, (resplen > anssiz) ? anssiz : resplen);
939 1.1 christos goto wait;
940 1.1 christos }
941 1.1 christos if (anhp->rcode == SERVFAIL ||
942 1.1 christos anhp->rcode == NOTIMP ||
943 1.1 christos anhp->rcode == REFUSED) {
944 1.1 christos DprintQ(statp->options & RES_DEBUG,
945 1.1 christos (stdout, "server rejected query:\n"),
946 1.1 christos ans, (resplen > anssiz) ? anssiz : resplen);
947 1.1 christos res_nclose(statp);
948 1.1 christos /* don't retry if called from dig */
949 1.1 christos if (!statp->pfcode)
950 1.1 christos return (0);
951 1.1 christos }
952 1.1 christos if (!(statp->options & RES_IGNTC) && anhp->tc) {
953 1.1 christos /*
954 1.1 christos * To get the rest of answer,
955 1.1 christos * use TCP with same server.
956 1.1 christos */
957 1.1 christos Dprint(statp->options & RES_DEBUG,
958 1.1 christos (stdout, ";; truncated answer\n"));
959 1.1 christos *v_circuit = 1;
960 1.1 christos res_nclose(statp);
961 1.1 christos return (1);
962 1.1 christos }
963 1.1 christos /*
964 1.1 christos * All is well, or the error is fatal. Signal that the
965 1.1 christos * next nameserver ought not be tried.
966 1.1 christos */
967 1.1 christos return (resplen);
968 1.1 christos }
969 1.1 christos
970 1.1 christos static void
971 1.1 christos Aerror(const res_state statp, FILE *file, const char *string, int error,
972 1.1 christos const struct sockaddr *address, int alen)
973 1.1 christos {
974 1.1 christos int save = errno;
975 1.1 christos char hbuf[NI_MAXHOST];
976 1.1 christos char sbuf[NI_MAXSERV];
977 1.1 christos
978 1.1 christos alen = alen;
979 1.1 christos
980 1.1 christos if ((statp->options & RES_DEBUG) != 0U) {
981 1.1 christos if (getnameinfo(address, alen, hbuf, sizeof(hbuf),
982 1.1 christos sbuf, sizeof(sbuf), niflags)) {
983 1.1 christos strncpy(hbuf, "?", sizeof(hbuf) - 1);
984 1.1 christos hbuf[sizeof(hbuf) - 1] = '\0';
985 1.1 christos strncpy(sbuf, "?", sizeof(sbuf) - 1);
986 1.1 christos sbuf[sizeof(sbuf) - 1] = '\0';
987 1.1 christos }
988 1.1 christos fprintf(file, "res_send: %s ([%s].%s): %s\n",
989 1.1 christos string, hbuf, sbuf, strerror(error));
990 1.1 christos }
991 1.1 christos errno = save;
992 1.1 christos }
993 1.1 christos
994 1.1 christos static void
995 1.1 christos Perror(const res_state statp, FILE *file, const char *string, int error) {
996 1.1 christos int save = errno;
997 1.1 christos
998 1.1 christos if ((statp->options & RES_DEBUG) != 0U)
999 1.1 christos fprintf(file, "res_send: %s: %s\n",
1000 1.1 christos string, strerror(error));
1001 1.1 christos errno = save;
1002 1.1 christos }
1003 1.1 christos
1004 1.1 christos static int
1005 1.1 christos sock_eq(struct sockaddr *a, struct sockaddr *b) {
1006 1.1 christos struct sockaddr_in *a4, *b4;
1007 1.1 christos struct sockaddr_in6 *a6, *b6;
1008 1.1 christos
1009 1.1 christos if (a->sa_family != b->sa_family)
1010 1.1 christos return 0;
1011 1.1 christos switch (a->sa_family) {
1012 1.1 christos case AF_INET:
1013 1.1 christos a4 = (struct sockaddr_in *)a;
1014 1.1 christos b4 = (struct sockaddr_in *)b;
1015 1.1 christos return a4->sin_port == b4->sin_port &&
1016 1.1 christos a4->sin_addr.s_addr == b4->sin_addr.s_addr;
1017 1.1 christos case AF_INET6:
1018 1.1 christos a6 = (struct sockaddr_in6 *)a;
1019 1.1 christos b6 = (struct sockaddr_in6 *)b;
1020 1.1 christos return a6->sin6_port == b6->sin6_port &&
1021 1.1 christos #ifdef HAVE_SIN6_SCOPE_ID
1022 1.1 christos a6->sin6_scope_id == b6->sin6_scope_id &&
1023 1.1 christos #endif
1024 1.1 christos IN6_ARE_ADDR_EQUAL(&a6->sin6_addr, &b6->sin6_addr);
1025 1.1 christos default:
1026 1.1 christos return 0;
1027 1.1 christos }
1028 1.1 christos }
1029 1.1 christos
1030 1.1 christos #ifdef NEED_PSELECT
1031 1.1 christos /* XXX needs to move to the porting library. */
1032 1.1 christos static int
1033 1.1 christos pselect(int nfds, void *rfds, void *wfds, void *efds,
1034 1.1 christos struct timespec *tsp, const sigset_t *sigmask)
1035 1.1 christos {
1036 1.1 christos struct timeval tv, *tvp;
1037 1.1 christos sigset_t sigs;
1038 1.1 christos int n;
1039 1.1 christos
1040 1.1 christos if (tsp) {
1041 1.1 christos tvp = &tv;
1042 1.1 christos tv = evTimeVal(*tsp);
1043 1.1 christos } else
1044 1.1 christos tvp = NULL;
1045 1.1 christos if (sigmask)
1046 1.1 christos sigprocmask(SIG_SETMASK, sigmask, &sigs);
1047 1.1 christos n = select(nfds, rfds, wfds, efds, tvp);
1048 1.1 christos if (sigmask)
1049 1.1 christos sigprocmask(SIG_SETMASK, &sigs, NULL);
1050 1.1 christos if (tsp)
1051 1.1 christos *tsp = evTimeSpec(tv);
1052 1.1 christos return (n);
1053 1.1 christos }
1054 1.1 christos #endif
1055