1 /* $OpenBSD: res_send_async.c,v 1.32 2017/02/18 22:25:13 eric Exp $ */ 2 /* 3 * Copyright (c) 2012 Eric Faurot <eric@openbsd.org> 4 * 5 * Permission to use, copy, modify, and distribute this software for any 6 * purpose with or without fee is hereby granted, provided that the above 7 * copyright notice and this permission notice appear in all copies. 8 * 9 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES 10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 15 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 16 */ 17 18 #include <sys/types.h> 19 #include <sys/socket.h> 20 #include <sys/uio.h> 21 #include <netinet/in.h> 22 #include <arpa/nameser.h> 23 #include <netdb.h> 24 25 #include <asr.h> 26 #include <errno.h> 27 #include <fcntl.h> 28 #include <poll.h> 29 #include <resolv.h> /* for res_random */ 30 #include <stdlib.h> 31 #include <string.h> 32 #include <unistd.h> 33 34 #include "asr_private.h" 35 36 #define OP_QUERY (0) 37 38 static int res_send_async_run(struct asr_query *, struct asr_result *); 39 static int sockaddr_connect(const struct sockaddr *, int); 40 static int udp_send(struct asr_query *); 41 static int udp_recv(struct asr_query *); 42 static int tcp_write(struct asr_query *); 43 static int tcp_read(struct asr_query *); 44 static int validate_packet(struct asr_query *); 45 static int setup_query(struct asr_query *, const char *, const char *, int, int); 46 static int ensure_ibuf(struct asr_query *, size_t); 47 static int iter_ns(struct asr_query *); 48 49 #define AS_NS_SA(p) ((p)->as_ctx->ac_ns[(p)->as.dns.nsidx - 1]) 50 51 52 struct asr_query * 53 res_send_async(const unsigned char *buf, int buflen, void *asr) 54 { 55 struct asr_ctx *ac; 56 struct asr_query *as; 57 struct asr_unpack p; 58 struct asr_dns_header h; 59 struct asr_dns_query q; 60 61 DPRINT_PACKET("asr: res_send_async()", buf, buflen); 62 63 ac = _asr_use_resolver(asr); 64 if ((as = _asr_async_new(ac, ASR_SEND)) == NULL) { 65 _asr_ctx_unref(ac); 66 return (NULL); /* errno set */ 67 } 68 as->as_run = res_send_async_run; 69 70 as->as.dns.flags |= ASYNC_EXTOBUF; 71 as->as.dns.obuf = (unsigned char *)buf; 72 as->as.dns.obuflen = buflen; 73 as->as.dns.obufsize = buflen; 74 75 _asr_unpack_init(&p, buf, buflen); 76 _asr_unpack_header(&p, &h); 77 _asr_unpack_query(&p, &q); 78 if (p.err) { 79 errno = EINVAL; 80 goto err; 81 } 82 as->as.dns.reqid = h.id; 83 as->as.dns.type = q.q_type; 84 as->as.dns.class = q.q_class; 85 as->as.dns.dname = strdup(q.q_dname); 86 if (as->as.dns.dname == NULL) 87 goto err; /* errno set */ 88 89 _asr_ctx_unref(ac); 90 return (as); 91 err: 92 if (as) 93 _asr_async_free(as); 94 _asr_ctx_unref(ac); 95 return (NULL); 96 } 97 DEF_WEAK(res_send_async); 98 99 /* 100 * Unlike res_query(), this version will actually return the packet 101 * if it has received a valid one (errno == 0) even if h_errno is 102 * not NETDB_SUCCESS. So the packet *must* be freed if necessary. 103 */ 104 struct asr_query * 105 res_query_async(const char *name, int class, int type, void *asr) 106 { 107 struct asr_ctx *ac; 108 struct asr_query *as; 109 110 DPRINT("asr: res_query_async(\"%s\", %i, %i)\n", name, class, type); 111 112 ac = _asr_use_resolver(asr); 113 as = _res_query_async_ctx(name, class, type, ac); 114 _asr_ctx_unref(ac); 115 116 return (as); 117 } 118 DEF_WEAK(res_query_async); 119 120 struct asr_query * 121 _res_query_async_ctx(const char *name, int class, int type, struct asr_ctx *a_ctx) 122 { 123 struct asr_query *as; 124 125 DPRINT("asr: res_query_async_ctx(\"%s\", %i, %i)\n", name, class, type); 126 127 if ((as = _asr_async_new(a_ctx, ASR_SEND)) == NULL) 128 return (NULL); /* errno set */ 129 as->as_run = res_send_async_run; 130 131 /* This adds a "." to name if it doesn't already has one. 132 * That's how res_query() behaves (through res_mkquery"). 133 */ 134 if (setup_query(as, name, NULL, class, type) == -1) 135 goto err; /* errno set */ 136 137 return (as); 138 139 err: 140 if (as) 141 _asr_async_free(as); 142 143 return (NULL); 144 } 145 146 static int 147 res_send_async_run(struct asr_query *as, struct asr_result *ar) 148 { 149 next: 150 switch (as->as_state) { 151 152 case ASR_STATE_INIT: 153 154 if (as->as_ctx->ac_nscount == 0) { 155 ar->ar_errno = ECONNREFUSED; 156 async_set_state(as, ASR_STATE_HALT); 157 break; 158 } 159 160 async_set_state(as, ASR_STATE_NEXT_NS); 161 break; 162 163 case ASR_STATE_NEXT_NS: 164 165 if (iter_ns(as) == -1) { 166 ar->ar_errno = ETIMEDOUT; 167 async_set_state(as, ASR_STATE_HALT); 168 break; 169 } 170 171 if (as->as_ctx->ac_options & RES_USEVC || 172 as->as.dns.obuflen > PACKETSZ) 173 async_set_state(as, ASR_STATE_TCP_WRITE); 174 else 175 async_set_state(as, ASR_STATE_UDP_SEND); 176 break; 177 178 case ASR_STATE_UDP_SEND: 179 180 if (udp_send(as) == -1) { 181 async_set_state(as, ASR_STATE_NEXT_NS); 182 break; 183 } 184 async_set_state(as, ASR_STATE_UDP_RECV); 185 ar->ar_cond = ASR_WANT_READ; 186 ar->ar_fd = as->as_fd; 187 ar->ar_timeout = as->as_timeout; 188 return (ASYNC_COND); 189 break; 190 191 case ASR_STATE_UDP_RECV: 192 193 if (udp_recv(as) == -1) { 194 if (errno == ENOMEM) { 195 ar->ar_errno = errno; 196 async_set_state(as, ASR_STATE_HALT); 197 break; 198 } 199 if (errno != EOVERFLOW) { 200 /* Fail or timeout */ 201 async_set_state(as, ASR_STATE_NEXT_NS); 202 break; 203 } 204 if (as->as_ctx->ac_options & RES_IGNTC) 205 async_set_state(as, ASR_STATE_PACKET); 206 else 207 async_set_state(as, ASR_STATE_TCP_WRITE); 208 } else 209 async_set_state(as, ASR_STATE_PACKET); 210 break; 211 212 case ASR_STATE_TCP_WRITE: 213 214 switch (tcp_write(as)) { 215 case -1: /* fail or timeout */ 216 async_set_state(as, ASR_STATE_NEXT_NS); 217 break; 218 case 0: 219 async_set_state(as, ASR_STATE_TCP_READ); 220 ar->ar_cond = ASR_WANT_READ; 221 ar->ar_fd = as->as_fd; 222 ar->ar_timeout = as->as_timeout; 223 return (ASYNC_COND); 224 case 1: 225 ar->ar_cond = ASR_WANT_WRITE; 226 ar->ar_fd = as->as_fd; 227 ar->ar_timeout = as->as_timeout; 228 return (ASYNC_COND); 229 } 230 break; 231 232 case ASR_STATE_TCP_READ: 233 234 switch (tcp_read(as)) { 235 case -1: /* Fail or timeout */ 236 if (errno == ENOMEM) { 237 ar->ar_errno = errno; 238 async_set_state(as, ASR_STATE_HALT); 239 } else 240 async_set_state(as, ASR_STATE_NEXT_NS); 241 break; 242 case 0: 243 async_set_state(as, ASR_STATE_PACKET); 244 break; 245 case 1: 246 ar->ar_cond = ASR_WANT_READ; 247 ar->ar_fd = as->as_fd; 248 ar->ar_timeout = as->as_timeout; 249 return (ASYNC_COND); 250 } 251 break; 252 253 case ASR_STATE_PACKET: 254 255 memmove(&ar->ar_ns, AS_NS_SA(as), AS_NS_SA(as)->sa_len); 256 ar->ar_datalen = as->as.dns.ibuflen; 257 ar->ar_data = as->as.dns.ibuf; 258 as->as.dns.ibuf = NULL; 259 ar->ar_errno = 0; 260 ar->ar_rcode = as->as.dns.rcode; 261 async_set_state(as, ASR_STATE_HALT); 262 break; 263 264 case ASR_STATE_HALT: 265 266 if (ar->ar_errno) { 267 ar->ar_h_errno = TRY_AGAIN; 268 ar->ar_count = 0; 269 ar->ar_datalen = -1; 270 ar->ar_data = NULL; 271 } else if (as->as.dns.ancount) { 272 ar->ar_h_errno = NETDB_SUCCESS; 273 ar->ar_count = as->as.dns.ancount; 274 } else { 275 ar->ar_count = 0; 276 switch (as->as.dns.rcode) { 277 case NXDOMAIN: 278 ar->ar_h_errno = HOST_NOT_FOUND; 279 break; 280 case SERVFAIL: 281 ar->ar_h_errno = TRY_AGAIN; 282 break; 283 case NOERROR: 284 ar->ar_h_errno = NO_DATA; 285 break; 286 default: 287 ar->ar_h_errno = NO_RECOVERY; 288 } 289 } 290 return (ASYNC_DONE); 291 292 default: 293 294 ar->ar_errno = EOPNOTSUPP; 295 ar->ar_h_errno = NETDB_INTERNAL; 296 async_set_state(as, ASR_STATE_HALT); 297 break; 298 } 299 goto next; 300 } 301 302 static int 303 sockaddr_connect(const struct sockaddr *sa, int socktype) 304 { 305 int errno_save, sock; 306 307 if ((sock = socket(sa->sa_family, 308 socktype | SOCK_NONBLOCK | SOCK_DNS, 0)) == -1) 309 goto fail; 310 311 if (connect(sock, sa, sa->sa_len) == -1) { 312 /* 313 * In the TCP case, the caller will be asked to poll for 314 * POLLOUT so that we start writing the packet in tcp_write() 315 * when the connection is established, or fail there on error. 316 */ 317 if (errno == EINPROGRESS) 318 return (sock); 319 goto fail; 320 } 321 322 return (sock); 323 324 fail: 325 326 if (sock != -1) { 327 errno_save = errno; 328 close(sock); 329 errno = errno_save; 330 } 331 332 return (-1); 333 } 334 335 /* 336 * Prepare the DNS packet for the query type "type", class "class" and domain 337 * name created by the concatenation on "name" and "dom". 338 * Return 0 on success, set errno and return -1 on error. 339 */ 340 static int 341 setup_query(struct asr_query *as, const char *name, const char *dom, 342 int class, int type) 343 { 344 struct asr_pack p; 345 struct asr_dns_header h; 346 char fqdn[MAXDNAME]; 347 char dname[MAXDNAME]; 348 349 if (as->as.dns.flags & ASYNC_EXTOBUF) { 350 errno = EINVAL; 351 DPRINT("attempting to write in user packet"); 352 return (-1); 353 } 354 355 if (_asr_make_fqdn(name, dom, fqdn, sizeof(fqdn)) > sizeof(fqdn)) { 356 errno = EINVAL; 357 DPRINT("asr_make_fqdn: name too long\n"); 358 return (-1); 359 } 360 361 if (_asr_dname_from_fqdn(fqdn, dname, sizeof(dname)) == -1) { 362 errno = EINVAL; 363 DPRINT("asr_dname_from_fqdn: invalid\n"); 364 return (-1); 365 } 366 367 if (as->as.dns.obuf == NULL) { 368 as->as.dns.obufsize = PACKETSZ; 369 as->as.dns.obuf = malloc(as->as.dns.obufsize); 370 if (as->as.dns.obuf == NULL) 371 return (-1); /* errno set */ 372 } 373 as->as.dns.obuflen = 0; 374 375 memset(&h, 0, sizeof h); 376 h.id = res_randomid(); 377 if (as->as_ctx->ac_options & RES_RECURSE) 378 h.flags |= RD_MASK; 379 h.qdcount = 1; 380 if (as->as_ctx->ac_options & RES_USE_EDNS0) 381 h.arcount = 1; 382 383 _asr_pack_init(&p, as->as.dns.obuf, as->as.dns.obufsize); 384 _asr_pack_header(&p, &h); 385 _asr_pack_query(&p, type, class, dname); 386 if (as->as_ctx->ac_options & RES_USE_EDNS0) 387 _asr_pack_edns0(&p, MAXPACKETSZ); 388 if (p.err) { 389 DPRINT("error packing query"); 390 errno = EINVAL; 391 return (-1); 392 } 393 394 /* Remember the parameters. */ 395 as->as.dns.reqid = h.id; 396 as->as.dns.type = type; 397 as->as.dns.class = class; 398 if (as->as.dns.dname) 399 free(as->as.dns.dname); 400 as->as.dns.dname = strdup(dname); 401 if (as->as.dns.dname == NULL) { 402 DPRINT("strdup"); 403 return (-1); /* errno set */ 404 } 405 as->as.dns.obuflen = p.offset; 406 407 DPRINT_PACKET("asr_setup_query", as->as.dns.obuf, as->as.dns.obuflen); 408 409 return (0); 410 } 411 412 /* 413 * Create a connect UDP socket and send the output packet. 414 * 415 * Return 0 on success, or -1 on error (errno set). 416 */ 417 static int 418 udp_send(struct asr_query *as) 419 { 420 ssize_t n; 421 int save_errno; 422 #ifdef DEBUG 423 char buf[256]; 424 #endif 425 426 DPRINT("asr: [%p] connecting to %s UDP\n", as, 427 _asr_print_sockaddr(AS_NS_SA(as), buf, sizeof buf)); 428 429 as->as_fd = sockaddr_connect(AS_NS_SA(as), SOCK_DGRAM); 430 if (as->as_fd == -1) 431 return (-1); /* errno set */ 432 433 n = send(as->as_fd, as->as.dns.obuf, as->as.dns.obuflen, 0); 434 if (n == -1) { 435 save_errno = errno; 436 close(as->as_fd); 437 errno = save_errno; 438 as->as_fd = -1; 439 return (-1); 440 } 441 442 return (0); 443 } 444 445 /* 446 * Try to receive a valid packet from the current UDP socket. 447 * 448 * Return 0 if a full packet could be read, or -1 on error (errno set). 449 */ 450 static int 451 udp_recv(struct asr_query *as) 452 { 453 ssize_t n; 454 int save_errno; 455 456 if (ensure_ibuf(as, MAXPACKETSZ) == -1) { 457 save_errno = errno; 458 close(as->as_fd); 459 errno = save_errno; 460 as->as_fd = -1; 461 return (-1); 462 } 463 464 n = recv(as->as_fd, as->as.dns.ibuf, as->as.dns.ibufsize, 0); 465 save_errno = errno; 466 close(as->as_fd); 467 errno = save_errno; 468 as->as_fd = -1; 469 if (n == -1) 470 return (-1); 471 472 as->as.dns.ibuflen = n; 473 474 DPRINT_PACKET("asr_udp_recv()", as->as.dns.ibuf, as->as.dns.ibuflen); 475 476 if (validate_packet(as) == -1) 477 return (-1); /* errno set */ 478 479 return (0); 480 } 481 482 /* 483 * Write the output packet to the TCP socket. 484 * 485 * Return 0 when all bytes have been sent, 1 there is no buffer space on the 486 * socket or it is not connected yet, or -1 on error (errno set). 487 */ 488 static int 489 tcp_write(struct asr_query *as) 490 { 491 struct msghdr msg; 492 struct iovec iov[2]; 493 uint16_t len; 494 ssize_t n; 495 size_t offset; 496 int i; 497 #ifdef DEBUG 498 char buf[256]; 499 #endif 500 501 /* First try to connect if not already */ 502 if (as->as_fd == -1) { 503 DPRINT("asr: [%p] connecting to %s TCP\n", as, 504 _asr_print_sockaddr(AS_NS_SA(as), buf, sizeof buf)); 505 as->as_fd = sockaddr_connect(AS_NS_SA(as), SOCK_STREAM); 506 if (as->as_fd == -1) 507 return (-1); /* errno set */ 508 as->as.dns.datalen = 0; /* bytes sent */ 509 return (1); 510 } 511 512 i = 0; 513 514 /* Prepend de packet length if not sent already. */ 515 if (as->as.dns.datalen < sizeof(len)) { 516 offset = 0; 517 len = htons(as->as.dns.obuflen); 518 iov[i].iov_base = (char *)(&len) + as->as.dns.datalen; 519 iov[i].iov_len = sizeof(len) - as->as.dns.datalen; 520 i++; 521 } else 522 offset = as->as.dns.datalen - sizeof(len); 523 524 iov[i].iov_base = as->as.dns.obuf + offset; 525 iov[i].iov_len = as->as.dns.obuflen - offset; 526 i++; 527 528 memset(&msg, 0, sizeof msg); 529 msg.msg_iov = iov; 530 msg.msg_iovlen = i; 531 532 send_again: 533 n = sendmsg(as->as_fd, &msg, MSG_NOSIGNAL); 534 if (n == -1) { 535 if (errno == EINTR) 536 goto send_again; 537 goto close; /* errno set */ 538 } 539 540 as->as.dns.datalen += n; 541 542 if (as->as.dns.datalen == as->as.dns.obuflen + sizeof(len)) { 543 /* All sent. Prepare for TCP read */ 544 as->as.dns.datalen = 0; 545 return (0); 546 } 547 548 /* More data to write */ 549 return (1); 550 551 close: 552 close(as->as_fd); 553 as->as_fd = -1; 554 return (-1); 555 } 556 557 /* 558 * Try to read a valid packet from the current TCP socket. 559 * 560 * Return 0 if a full packet could be read, 1 if more data is needed and the 561 * socket must be read again, or -1 on error (errno set). 562 */ 563 static int 564 tcp_read(struct asr_query *as) 565 { 566 ssize_t n; 567 size_t offset, len; 568 char *pos; 569 int save_errno, nfds; 570 struct pollfd pfd; 571 572 /* We must read the packet len first */ 573 if (as->as.dns.datalen < sizeof(as->as.dns.pktlen)) { 574 575 pos = (char *)(&as->as.dns.pktlen) + as->as.dns.datalen; 576 len = sizeof(as->as.dns.pktlen) - as->as.dns.datalen; 577 578 n = read(as->as_fd, pos, len); 579 if (n == -1) 580 goto close; /* errno set */ 581 582 as->as.dns.datalen += n; 583 if (as->as.dns.datalen < sizeof(as->as.dns.pktlen)) 584 return (1); /* need more data */ 585 586 as->as.dns.ibuflen = ntohs(as->as.dns.pktlen); 587 if (ensure_ibuf(as, as->as.dns.ibuflen) == -1) 588 goto close; /* errno set */ 589 590 pfd.fd = as->as_fd; 591 pfd.events = POLLIN; 592 poll_again: 593 nfds = poll(&pfd, 1, 0); 594 if (nfds == -1) { 595 if (errno == EINTR) 596 goto poll_again; 597 goto close; /* errno set */ 598 } 599 if (nfds == 0) 600 return (1); /* no more data available */ 601 } 602 603 offset = as->as.dns.datalen - sizeof(as->as.dns.pktlen); 604 pos = as->as.dns.ibuf + offset; 605 len = as->as.dns.ibuflen - offset; 606 607 read_again: 608 n = read(as->as_fd, pos, len); 609 if (n == -1) { 610 if (errno == EINTR) 611 goto read_again; 612 goto close; /* errno set */ 613 } 614 if (n == 0) { 615 errno = ECONNRESET; 616 goto close; 617 } 618 as->as.dns.datalen += n; 619 620 /* See if we got all the advertised bytes. */ 621 if (as->as.dns.datalen != as->as.dns.ibuflen + sizeof(as->as.dns.pktlen)) 622 return (1); 623 624 DPRINT_PACKET("asr_tcp_read()", as->as.dns.ibuf, as->as.dns.ibuflen); 625 626 if (validate_packet(as) == -1) 627 goto close; /* errno set */ 628 629 errno = 0; 630 close: 631 save_errno = errno; 632 close(as->as_fd); 633 errno = save_errno; 634 as->as_fd = -1; 635 return (errno ? -1 : 0); 636 } 637 638 /* 639 * Make sure the input buffer is at least "n" bytes long, and allocate or 640 * extend it if necessary. Return 0 on success, or set errno and return -1. 641 */ 642 static int 643 ensure_ibuf(struct asr_query *as, size_t n) 644 { 645 char *t; 646 647 if (as->as.dns.ibuf == NULL) { 648 as->as.dns.ibuf = malloc(n); 649 if (as->as.dns.ibuf == NULL) 650 return (-1); /* errno set */ 651 as->as.dns.ibufsize = n; 652 return (0); 653 } 654 655 if (as->as.dns.ibufsize >= n) 656 return (0); 657 658 t = realloc(as->as.dns.ibuf, n); 659 if (t == NULL) 660 return (-1); /* errno set */ 661 as->as.dns.ibuf = t; 662 as->as.dns.ibufsize = n; 663 664 return (0); 665 } 666 667 /* 668 * Check if the received packet is valid. 669 * Return 0 on success, or set errno and return -1. 670 */ 671 static int 672 validate_packet(struct asr_query *as) 673 { 674 struct asr_unpack p; 675 struct asr_dns_header h; 676 struct asr_dns_query q; 677 struct asr_dns_rr rr; 678 int r; 679 680 _asr_unpack_init(&p, as->as.dns.ibuf, as->as.dns.ibuflen); 681 682 _asr_unpack_header(&p, &h); 683 if (p.err) 684 goto inval; 685 686 if (h.id != as->as.dns.reqid) { 687 DPRINT("incorrect reqid\n"); 688 goto inval; 689 } 690 if (h.qdcount != 1) 691 goto inval; 692 /* Should be zero, we could allow this */ 693 if ((h.flags & Z_MASK) != 0) 694 goto inval; 695 /* Actually, it depends on the request but we only use OP_QUERY */ 696 if (OPCODE(h.flags) != OP_QUERY) 697 goto inval; 698 /* Must be a response */ 699 if ((h.flags & QR_MASK) == 0) 700 goto inval; 701 702 as->as.dns.rcode = RCODE(h.flags); 703 as->as.dns.ancount = h.ancount; 704 705 _asr_unpack_query(&p, &q); 706 if (p.err) 707 goto inval; 708 709 if (q.q_type != as->as.dns.type || 710 q.q_class != as->as.dns.class || 711 strcasecmp(q.q_dname, as->as.dns.dname)) { 712 DPRINT("incorrect type/class/dname '%s' != '%s'\n", 713 q.q_dname, as->as.dns.dname); 714 goto inval; 715 } 716 717 /* Check for truncation */ 718 if (h.flags & TC_MASK && !(as->as_ctx->ac_options & RES_IGNTC)) { 719 DPRINT("truncated\n"); 720 errno = EOVERFLOW; 721 return (-1); 722 } 723 724 /* Validate the rest of the packet */ 725 for (r = h.ancount + h.nscount + h.arcount; r; r--) 726 _asr_unpack_rr(&p, &rr); 727 728 /* Report any error found when unpacking the RRs. */ 729 if (p.err) { 730 DPRINT("unpack: %s\n", strerror(p.err)); 731 errno = p.err; 732 return (-1); 733 } 734 735 if (p.offset != as->as.dns.ibuflen) { 736 DPRINT("trailing garbage\n"); 737 errno = EMSGSIZE; 738 return (-1); 739 } 740 741 return (0); 742 743 inval: 744 errno = EINVAL; 745 return (-1); 746 } 747 748 /* 749 * Set the async context nameserver index to the next nameserver, cycling 750 * over the list until the maximum retry counter is reached. Return 0 on 751 * success, or -1 if all nameservers were used. 752 */ 753 static int 754 iter_ns(struct asr_query *as) 755 { 756 for (;;) { 757 if (as->as.dns.nsloop >= as->as_ctx->ac_nsretries) 758 return (-1); 759 760 as->as.dns.nsidx += 1; 761 if (as->as.dns.nsidx <= as->as_ctx->ac_nscount) 762 break; 763 as->as.dns.nsidx = 0; 764 as->as.dns.nsloop++; 765 DPRINT("asr: iter_ns(): cycle %i\n", as->as.dns.nsloop); 766 } 767 768 as->as_timeout = 1000 * (as->as_ctx->ac_nstimeout << as->as.dns.nsloop); 769 if (as->as.dns.nsloop > 0) 770 as->as_timeout /= as->as_ctx->ac_nscount; 771 if (as->as_timeout < 1000) 772 as->as_timeout = 1000; 773 774 return (0); 775 } 776