ircd/ircd_res.c

   1 /*
   2  * A rewrite of Darren Reed's original res.c As there is nothing
   3  * left of Darren's original code, this is now licensed by the hybrid group.
   4  * (Well, some of the function names are the same, and bits of the structs..)
   5  * You can use it where it is useful, free even. Buy us a beer and stuff.
   6  *
   7  * The authors takes no responsibility for any damage or loss
   8  * of property which results from the use of this software.
   9  *
  10  * July 1999 - Rewrote a bunch of stuff here. Change hostent builder code,
  11  *     added callbacks and reference counting of returned hostents.
  12  *     --Bleep (Thomas Helvey <tomh@inxpress.net>)
  13  *
  14  * This was all needlessly complicated for irc. Simplified. No more hostent
  15  * All we really care about is the IP -> hostname mappings. Thats all.
  16  *
  17  * Apr 28, 2003 --cryogen and Dianora
  18  */
  19 /** @file
  20  * @brief IRC resolver functions.
  21  * @version $Id$
  22  */
  23
  24 #include "client.h"
  25 #include "ircd_alloc.h"
  26 #include "ircd_log.h"
  27 #include "ircd_osdep.h"
  28 #include "ircd_reply.h"
  29 #include "ircd_string.h"
  30 #include "ircd_snprintf.h"
  31 #include "ircd.h"
  32 #include "numeric.h"
  33 #include "fileio.h" /* for fbopen / fbclose / fbputs */
  34 #include "random.h"
  35 #include "s_bsd.h"
  36 #include "s_debug.h"
  37 #include "s_stats.h"
  38 #include "send.h"
  39 #include "sys.h"
  40 #include "res.h"
  41 #include "ircd_reslib.h"
  42
  43 /* #include <assert.h> -- Now using assert in ircd_log.h */
  44 #include <string.h>
  45 #include <sys/time.h>
  46 #include <sys/socket.h>
  47 #include <time.h>
  48
  49 #if (CHAR_BIT != 8)
  50 #error this code needs to be able to address individual octets
  51 #endif
  52
  53 /** IPv4 resolver UDP socket. */
  54 static struct Socket res_socket_v4;
  55 /** IPv6 resolver UDP socket. */
  56 static struct Socket res_socket_v6;
  57 /** Next DNS lookup timeout. */
  58 static struct Timer res_timeout;
  59 /** Check for whether the resolver has been initialized yet. */
  60 #define resolver_started() (request_list.next != NULL)
  61
  62 /** Maximum DNS packet length.
  63  * RFC says 512, but we add extra for expanded names.
  64  */
  65 #define MAXPACKET      1024
  66 #define AR_TTL         600   /**< TTL in seconds for dns cache entries */
  67
  68 /* RFC 1104/1105 wasn't very helpful about what these fields
  69  * should be named, so for now, we'll just name them this way.
  70  * we probably should look at what named calls them or something.
  71  */
  72 /** Size of TYPE field of a DNS RR header. */
  73 #define TYPE_SIZE         (size_t)2
  74 /** Size of CLASS field of a DNS RR header. */
  75 #define CLASS_SIZE        (size_t)2
  76 /** Size of TTL field of a DNS RR header. */
  77 #define TTL_SIZE          (size_t)4
  78 /** Size of RDLENGTH field of a DNS RR header. */
  79 #define RDLENGTH_SIZE     (size_t)2
  80 /** Size of fixed-format part of a DNS RR header. */
  81 #define ANSWER_FIXED_SIZE (TYPE_SIZE + CLASS_SIZE + TTL_SIZE + RDLENGTH_SIZE)
  82
  83 /** Current request state. */
  84 typedef enum
  85 {
  86   REQ_IDLE,  /**< We're doing not much at all. */
  87   REQ_PTR,   /**< Looking up a PTR. */
  88   REQ_A,     /**< Looking up an A, possibly because AAAA failed. */
  89   REQ_AAAA,  /**< Looking up an AAAA. */
  90   REQ_CNAME, /**< We got a CNAME in response, we better get a real answer next. */
  91   REQ_INT    /**< ip6.arpa failed, falling back to ip6.int. */
  92 } request_state;
  93
  94 /** Doubly linked list node. */
  95 struct dlink
  96 {
  97     struct dlink *prev; /**< Previous element in list. */
  98     struct dlink *next; /**< Next element in list. */
  99 };
 100
 101 /** A single resolver request.
 102  * (Do not be fooled by the "list" in the name.)
 103  */
 104 struct reslist
 105 {
 106   struct dlink node;       /**< Doubly linked list node. */
 107   int id;                  /**< Request ID (from request header). */
 108   int sent;                /**< Number of requests sent. */
 109   request_state state;     /**< State the resolver machine is in. */
 110   char type;               /**< Current request type. */
 111   char retries;            /**< Retry counter. */
 112   char sends;              /**< Number of sends (>1 means resent). */
 113   char resend;             /**< Send flag; 0 == don't resend. */
 114   time_t sentat;           /**< Timestamp we last sent this request. */
 115   time_t timeout;          /**< When this request times out. */
 116   struct irc_in_addr addr; /**< Address for this request. */
 117   char *name;              /**< Hostname for this request. */
 118   struct DNSQuery query;   /**< Query callback for this request. */
 119 };
 120
 121 /** Base of request list. */
 122 static struct dlink request_list;
 123
 124 static void rem_request(struct reslist *request);
 125 static struct reslist *make_request(const struct DNSQuery *query);
 126 static void do_query_name(const struct DNSQuery *query,
 127                           const char* name, struct reslist *request, int);
 128 static void do_query_number(const struct DNSQuery *query,
 129                             const struct irc_in_addr *,
 130                             struct reslist *request);
 131 static void query_name(const char *name, int query_class, int query_type,
 132                        struct reslist *request);
 133 static int send_res_msg(const char *buf, int len, int count);
 134 static void resend_query(struct reslist *request);
 135 static int proc_answer(struct reslist *request, HEADER *header, char *, char *);
 136 static struct reslist *find_id(int id);
 137 static struct DNSReply *make_dnsreply(struct reslist *request);
 138 static void res_readreply(struct Event *ev);
 139 static void timeout_resolver(struct Event *notused);
 140
 141 extern struct irc_sockaddr irc_nsaddr_list[IRCD_MAXNS];
 142 extern int irc_nscount;
 143 extern char irc_domain[HOSTLEN];
 144
 145 /** Check whether \a inp is a nameserver we use.
 146  * @param[in] inp Nameserver address.
 147  * @return Non-zero if we trust \a inp; zero if not.
 148  */
 149 static int
 150 res_ourserver(const struct irc_sockaddr *inp)
 151 {
 152   int ns;
 153
 154   for (ns = 0;  ns < irc_nscount;  ns++)
 155     if (!irc_in_addr_cmp(&inp->addr, &irc_nsaddr_list[ns].addr)
 156         && inp->port == irc_nsaddr_list[ns].port)
 157       return 1;
 158
 159   return(0);
 160 }
 161
 162 /** Start (or re-start) resolver.
 163  * This means read resolv.conf, initialize the list of pending
 164  * requests, open the resolver socket and initialize its timeout.
 165  */
 166 void
 167 restart_resolver(void)
 168 {
 169   irc_res_init();
 170
 171   if (!request_list.next)
 172     request_list.next = request_list.prev = &request_list;
 173
 174   if (!s_active(&res_socket_v4))
 175   {
 176     int fd = os_socket(&VirtualHost_v4, SOCK_DGRAM, "Resolver UDPv4 socket");
 177     if (fd >= 0)
 178       socket_add(&res_socket_v4, res_readreply, NULL,
 179                  SS_DATAGRAM, SOCK_EVENT_READABLE, fd);
 180   }
 181
 182   if (!s_active(&res_socket_v6))
 183   {
 184     int fd = os_socket(&VirtualHost_v6, SOCK_DGRAM, "Resolver UDPv6 socket");
 185     if (fd >= 0)
 186       socket_add(&res_socket_v6, res_readreply, NULL,
 187                  SS_DATAGRAM, SOCK_EVENT_READABLE, fd);
 188   }
 189
 190   if (s_active(&res_socket_v4) || s_active(&res_socket_v6))
 191     timer_init(&res_timeout);
 192 }
 193
 194 /** Append local domain to hostname if needed.
 195  * If \a hname does not contain any '.'s, append #irc_domain to it.
 196  * @param[in,out] hname Hostname to check.
 197  * @param[in] size Length of \a hname buffer.
 198  */
 199 void
 200 add_local_domain(char* hname, size_t size)
 201 {
 202   /* try to fix up unqualified names
 203    */
 204   if (strchr(hname, '.') == NULL)
 205   {
 206     if (irc_domain[0])
 207     {
 208       size_t len = strlen(hname);
 209
 210       if ((strlen(irc_domain) + len + 2) < size)
 211       {
 212         hname[len++] = '.';
 213         strcpy(hname + len, irc_domain);
 214       }
 215     }
 216   }
 217 }
 218
 219 /** Add a node to a doubly linked list.
 220  * @param[in,out] node Node to add to list.
 221  * @param[in,out] next Add \a node before this one.
 222  */
 223 static void
 224 add_dlink(struct dlink *node, struct dlink *next)
 225 {
 226     node->prev = next->prev;
 227     node->next = next;
 228     node->prev->next = node;
 229     node->next->prev = node;
 230 }
 231
 232 /** Remove a request from the list and free it.
 233  * @param[in] request Node to free.
 234  */
 235 static void
 236 rem_request(struct reslist *request)
 237 {
 238   /* remove from dlist */
 239   request->node.prev->next = request->node.next;
 240   request->node.next->prev = request->node.prev;
 241   /* free memory */
 242   MyFree(request->name);
 243   MyFree(request);
 244 }
 245
 246 /** Create a DNS request record for the server.
 247  * @param[in] query Callback information for caller.
 248  * @return Newly allocated and linked-in reslist.
 249  */
 250 static struct reslist *
 251 make_request(const struct DNSQuery* query)
 252 {
 253   struct reslist *request;
 254
 255   if (!resolver_started())
 256     restart_resolver();
 257
 258   request = (struct reslist *)MyMalloc(sizeof(struct reslist));
 259   memset(request, 0, sizeof(struct reslist));
 260
 261   request->state   = REQ_IDLE;
 262   request->sentat  = CurrentTime;
 263   request->retries = feature_int(FEAT_IRCD_RES_RETRIES);
 264   request->resend  = 1;
 265   request->timeout = feature_int(FEAT_IRCD_RES_TIMEOUT);
 266   memset(&request->addr, 0, sizeof(request->addr));
 267   memcpy(&request->query, query, sizeof(request->query));
 268
 269   add_dlink(&request->node, &request_list);
 270   return(request);
 271 }
 272
 273 /** Make sure that a timeout event will happen by the given time.
 274  * @param[in] when Latest time for timeout to run.
 275  */
 276 static void
 277 check_resolver_timeout(time_t when)
 278 {
 279   if (when > CurrentTime + AR_TTL)
 280     when = CurrentTime + AR_TTL;
 281   /* TODO after 2.10.12: Rewrite the timer API because there should be
 282    * no need for clients to know this kind of implementation detail. */
 283   if (when > t_expire(&res_timeout))
 284     /* do nothing */;
 285   else if (t_onqueue(&res_timeout) && !(res_timeout.t_header.gh_flags & GEN_MARKED))
 286     timer_chg(&res_timeout, TT_ABSOLUTE, when);
 287   else
 288     timer_add(&res_timeout, timeout_resolver, NULL, TT_ABSOLUTE, when);
 289 }
 290
 291 /** Drop pending DNS lookups which have timed out.
 292  * @param[in] ev Timer event data (ignored).
 293  */
 294 static void
 295 timeout_resolver(struct Event *ev)
 296 {
 297   struct dlink *ptr, *next_ptr;
 298   struct reslist *request;
 299   time_t next_time = 0;
 300   time_t timeout   = 0;
 301
 302   if (ev_type(ev) != ET_EXPIRE)
 303     return;
 304
 305   for (ptr = request_list.next; ptr != &request_list; ptr = next_ptr)
 306   {
 307     next_ptr = ptr->next;
 308     request = (struct reslist*)ptr;
 309     timeout = request->sentat + request->timeout;
 310
 311     if (CurrentTime >= timeout)
 312     {
 313       if (--request->retries <= 0)
 314       {
 315         Debug((DEBUG_DNS, "Request %p out of retries; destroying", request));
 316         (*request->query.callback)(request->query.vptr, 0);
 317         rem_request(request);
 318         continue;
 319       }
 320       else
 321       {
 322         request->sentat = CurrentTime;
 323         request->timeout += request->timeout;
 324         resend_query(request);
 325       }
 326     }
 327
 328     if ((next_time == 0) || timeout < next_time)
 329     {
 330       next_time = timeout;
 331     }
 332   }
 333
 334   if (next_time <= CurrentTime)
 335     next_time = CurrentTime + AR_TTL;
 336   check_resolver_timeout(next_time);
 337 }
 338
 339 /** Drop queries that are associated with a particular pointer.
 340  * This is used to clean up lookups for clients or conf blocks
 341  * that went away.
 342  * @param[in] vptr User callback pointer to search for.
 343  */
 344 void
 345 delete_resolver_queries(const void *vptr)
 346 {
 347   struct dlink *ptr, *next_ptr;
 348   struct reslist *request;
 349
 350   if (request_list.next) {
 351     for (ptr = request_list.next; ptr != &request_list; ptr = next_ptr)
 352     {
 353       next_ptr = ptr->next;
 354       request = (struct reslist*)ptr;
 355       if (vptr == request->query.vptr) {
 356         Debug((DEBUG_DNS, "Removing request %p with vptr %p", request, vptr));
 357         rem_request(request);
 358       }
 359     }
 360   }
 361 }
 362
 363 /** Send a message to all of our nameservers.
 364  * @param[in] msg Message to send.
 365  * @param[in] len Length of message.
 366  * @param[in] rcount Maximum number of servers to ask.
 367  * @return Number of servers that were successfully asked.
 368  */
 369 static int
 370 send_res_msg(const char *msg, int len, int rcount)
 371 {
 372   int i;
 373   int sent = 0;
 374   int max_queries = IRCD_MIN(irc_nscount, rcount);
 375
 376   /* RES_PRIMARY option is not implemented
 377    * if (res.options & RES_PRIMARY || 0 == max_queries)
 378    */
 379   if (max_queries == 0)
 380     max_queries = 1;
 381
 382   for (i = 0; i < max_queries; i++) {
 383     int fd = irc_in_addr_is_ipv4(&irc_nsaddr_list[i].addr) ? s_fd(&res_socket_v4) : s_fd(&res_socket_v6);
 384     if (os_sendto_nonb(fd, msg, len, NULL, 0, &irc_nsaddr_list[i]) == IO_SUCCESS)
 385       ++sent;
 386   }
 387
 388   return(sent);
 389 }
 390
 391 /** Find a DNS request by ID.
 392  * @param[in] id Identifier to find.
 393  * @return Matching DNS request, or NULL if none are found.
 394  */
 395 static struct reslist *
 396 find_id(int id)
 397 {
 398   struct dlink *ptr;
 399   struct reslist *request;
 400
 401   for (ptr = request_list.next; ptr != &request_list; ptr = ptr->next)
 402   {
 403     request = (struct reslist*)ptr;
 404
 405     if (request->id == id) {
 406       Debug((DEBUG_DNS, "find_id(%d) -> %p", id, request));
 407       return(request);
 408     }
 409   }
 410
 411   Debug((DEBUG_DNS, "find_id(%d) -> NULL", id));
 412   return(NULL);
 413 }
 414
 415 /** Try to look up address for a hostname, trying IPv6 (T_AAAA) first.
 416  * @param[in] name Hostname to look up.
 417  * @param[in] query Callback information.
 418  */
 419 void
 420 gethost_byname(const char *name, const struct DNSQuery *query)
 421 {
 422   do_query_name(query, name, NULL, T_AAAA);
 423 }
 424
 425 /** Try to look up hostname for an address.
 426  * @param[in] addr Address to look up.
 427  * @param[in] query Callback information.
 428  */
 429 void
 430 gethost_byaddr(const struct irc_in_addr *addr, const struct DNSQuery *query)
 431 {
 432   do_query_number(query, addr, NULL);
 433 }
 434
 435 /** Send a query to look up the address for a name.
 436  * @param[in] query Callback information.
 437  * @param[in] name Hostname to look up.
 438  * @param[in] request DNS lookup structure (may be NULL).
 439  * @param[in] type Preferred request type.
 440  */
 441 static void
 442 do_query_name(const struct DNSQuery *query, const char *name,
 443               struct reslist *request, int type)
 444 {
 445   char host_name[HOSTLEN + 1];
 446
 447   ircd_strncpy(host_name, name, HOSTLEN);
 448   add_local_domain(host_name, HOSTLEN);
 449
 450   if (request == NULL)
 451   {
 452     request       = make_request(query);
 453     DupString(request->name, host_name);
 454 #ifdef IPV6
 455     if (type != T_A)
 456       request->state = REQ_AAAA;
 457     else
 458 #endif
 459     request->state = REQ_A;
 460   }
 461
 462   request->type = type;
 463   Debug((DEBUG_DNS, "Requesting DNS %s %s as %p", (request->state == REQ_AAAA ? "AAAA" : "A"), host_name, request));
 464   query_name(host_name, C_IN, type, request);
 465 }
 466
 467 /** Send a query to look up the name for an address.
 468  * @param[in] query Callback information.
 469  * @param[in] addr Address to look up.
 470  * @param[in] request DNS lookup structure (may be NULL).
 471  */
 472 static void
 473 do_query_number(const struct DNSQuery *query, const struct irc_in_addr *addr,
 474                 struct reslist *request)
 475 {
 476   char ipbuf[128];
 477   const unsigned char *cp;
 478
 479   if (irc_in_addr_is_ipv4(addr))
 480   {
 481     cp = (const unsigned char*)&addr->in6_16[6];
 482     ircd_snprintf(NULL, ipbuf, sizeof(ipbuf), "%u.%u.%u.%u.in-addr.arpa.",
 483                   (unsigned int)(cp[3]), (unsigned int)(cp[2]),
 484                   (unsigned int)(cp[1]), (unsigned int)(cp[0]));
 485   }
 486   else
 487   {
 488     const char *intarpa;
 489
 490     if (request != NULL && request->state == REQ_INT)
 491       intarpa = "int";
 492     else
 493       intarpa = "arpa";
 494
 495     cp = (const unsigned char *)&addr->in6_16[0];
 496     ircd_snprintf(NULL, ipbuf, sizeof(ipbuf),
 497                   "%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x."
 498                   "%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.ip6.%s.",
 499                   (unsigned int)(cp[15]&0xf), (unsigned int)(cp[15]>>4),
 500                   (unsigned int)(cp[14]&0xf), (unsigned int)(cp[14]>>4),
 501                   (unsigned int)(cp[13]&0xf), (unsigned int)(cp[13]>>4),
 502                   (unsigned int)(cp[12]&0xf), (unsigned int)(cp[12]>>4),
 503                   (unsigned int)(cp[11]&0xf), (unsigned int)(cp[11]>>4),
 504                   (unsigned int)(cp[10]&0xf), (unsigned int)(cp[10]>>4),
 505                   (unsigned int)(cp[9]&0xf), (unsigned int)(cp[9]>>4),
 506                   (unsigned int)(cp[8]&0xf), (unsigned int)(cp[8]>>4),
 507                   (unsigned int)(cp[7]&0xf), (unsigned int)(cp[7]>>4),
 508                   (unsigned int)(cp[6]&0xf), (unsigned int)(cp[6]>>4),
 509                   (unsigned int)(cp[5]&0xf), (unsigned int)(cp[5]>>4),
 510                   (unsigned int)(cp[4]&0xf), (unsigned int)(cp[4]>>4),
 511                   (unsigned int)(cp[3]&0xf), (unsigned int)(cp[3]>>4),
 512                   (unsigned int)(cp[2]&0xf), (unsigned int)(cp[2]>>4),
 513                   (unsigned int)(cp[1]&0xf), (unsigned int)(cp[1]>>4),
 514                   (unsigned int)(cp[0]&0xf), (unsigned int)(cp[0]>>4), intarpa);
 515   }
 516   if (request == NULL)
 517   {
 518     request       = make_request(query);
 519     request->state= REQ_PTR;
 520     request->type = T_PTR;
 521     memcpy(&request->addr, addr, sizeof(request->addr));
 522     request->name = (char *)MyMalloc(HOSTLEN + 1);
 523   }
 524   Debug((DEBUG_DNS, "Requesting DNS PTR %s as %p", ipbuf, request));
 525   query_name(ipbuf, C_IN, T_PTR, request);
 526 }
 527
 528 /** Generate a query based on class, type and name.
 529  * @param[in] name Domain name to look up.
 530  * @param[in] query_class Query class (see RFC 1035).
 531  * @param[in] type Query type (see RFC 1035).
 532  * @param[in] request DNS request structure.
 533  */
 534 static void
 535 query_name(const char *name, int query_class, int type,
 536            struct reslist *request)
 537 {
 538   char buf[MAXPACKET];
 539   int request_len = 0;
 540
 541   memset(buf, 0, sizeof(buf));
 542
 543   if ((request_len = irc_res_mkquery(name, query_class, type,
 544       (unsigned char *)buf, sizeof(buf))) > 0)
 545   {
 546     HEADER *header = (HEADER *)buf;
 547
 548     /*
 549      * generate an unique id
 550      * NOTE: we don't have to worry about converting this to and from
 551      * network byte order, the nameserver does not interpret this value
 552      * and returns it unchanged
 553      */
 554     do
 555     {
 556       header->id = (header->id + ircrandom()) & 0xffff;
 557     } while (find_id(header->id));
 558     request->id = header->id;
 559     ++request->sends;
 560
 561     request->sent += send_res_msg(buf, request_len, request->sends);
 562     check_resolver_timeout(request->sentat + request->timeout);
 563   }
 564 }
 565
 566 /** Send a failed DNS lookup request again.
 567  * @param[in] request Request to resend.
 568  */
 569 static void
 570 resend_query(struct reslist *request)
 571 {
 572   if (request->resend == 0)
 573     return;
 574
 575   switch(request->type)
 576   {
 577     case T_PTR:
 578       do_query_number(NULL, &request->addr, request);
 579       break;
 580     case T_A:
 581       do_query_name(NULL, request->name, request, request->type);
 582       break;
 583     case T_AAAA:
 584       /* didn't work, try A */
 585       if (request->state == REQ_AAAA)
 586         do_query_name(NULL, request->name, request, T_A);
 587     default:
 588       break;
 589   }
 590 }
 591
 592 /** Process the answer for a lookup request.
 593  * @param[in] request DNS request that got an answer.
 594  * @param[in] header Header of DNS response.
 595  * @param[in] buf DNS response body.
 596  * @param[in] eob Pointer to end of DNS response.
 597  * @return Number of answers read from \a buf.
 598  */
 599 static int
 600 proc_answer(struct reslist *request, HEADER* header, char* buf, char* eob)
 601 {
 602   char hostbuf[HOSTLEN + 100]; /* working buffer */
 603   unsigned char *current;      /* current position in buf */
 604   int query_class;             /* answer class */
 605   int type;                    /* answer type */
 606   int n;                       /* temp count */
 607   int rd_length;
 608
 609   current = (unsigned char *)buf + sizeof(HEADER);
 610
 611   for (; header->qdcount > 0; --header->qdcount)
 612   {
 613     if ((n = irc_dn_skipname(current, (unsigned char *)eob)) < 0)
 614       break;
 615
 616     current += (size_t) n + QFIXEDSZ;
 617   }
 618
 619   /*
 620    * process each answer sent to us blech.
 621    */
 622   while (header->ancount > 0 && (char *)current < eob)
 623   {
 624     header->ancount--;
 625
 626     n = irc_dn_expand((unsigned char *)buf, (unsigned char *)eob, current,
 627         hostbuf, sizeof(hostbuf));
 628
 629     if (n < 0)
 630     {
 631       /*
 632        * broken message
 633        */
 634       return(0);
 635     }
 636     else if (n == 0)
 637     {
 638       /*
 639        * no more answers left
 640        */
 641       return(0);
 642     }
 643
 644     hostbuf[HOSTLEN] = '\0';
 645
 646     /* With Address arithmetic you have to be very anal
 647      * this code was not working on alpha due to that
 648      * (spotted by rodder/jailbird/dianora)
 649      */
 650     current += (size_t) n;
 651
 652     if (!(((char *)current + ANSWER_FIXED_SIZE) < eob))
 653       break;
 654
 655     type = irc_ns_get16(current);
 656     current += TYPE_SIZE;
 657
 658     query_class = irc_ns_get16(current);
 659     current += CLASS_SIZE;
 660
 661     current += TTL_SIZE;
 662
 663     rd_length = irc_ns_get16(current);
 664     current += RDLENGTH_SIZE;
 665
 666     /*
 667      * Wait to set request->type until we verify this structure
 668      */
 669     switch (type)
 670     {
 671       case T_A:
 672         if (request->type != T_A)
 673           return(0);
 674
 675         /*
 676          * check for invalid rd_length or too many addresses
 677          */
 678         if (rd_length != sizeof(struct in_addr))
 679           return(0);
 680         memset(&request->addr, 0, sizeof(request->addr));
 681         memcpy(&request->addr.in6_16[6], current, sizeof(struct in_addr));
 682         return(1);
 683         break;
 684       case T_AAAA:
 685         if (request->type != T_AAAA)
 686           return(0);
 687         if (rd_length != sizeof(struct irc_in_addr))
 688           return(0);
 689         memcpy(&request->addr, current, sizeof(struct irc_in_addr));
 690         return(1);
 691         break;
 692       case T_PTR:
 693         if (request->type != T_PTR)
 694           return(0);
 695         n = irc_dn_expand((unsigned char *)buf, (unsigned char *)eob,
 696             current, hostbuf, sizeof(hostbuf));
 697         if (n < 0)
 698           return(0); /* broken message */
 699         else if (n == 0)
 700           return(0); /* no more answers left */
 701
 702         ircd_strncpy(request->name, hostbuf, HOSTLEN);
 703
 704         return(1);
 705         break;
 706       case T_CNAME: /* first check we already haven't started looking
 707                        into a cname */
 708         if (request->type != T_PTR)
 709           return(0);
 710
 711         if (request->state == REQ_CNAME)
 712         {
 713           n = irc_dn_expand((unsigned char *)buf, (unsigned char *)eob,
 714                             current, hostbuf, sizeof(hostbuf));
 715
 716           if (n < 0)
 717             return(0);
 718           return(1);
 719         }
 720
 721         request->state = REQ_CNAME;
 722         current += rd_length;
 723         break;
 724
 725       default:
 726         /* XXX I'd rather just throw away the entire bogus thing
 727          * but its possible its just a broken nameserver with still
 728          * valid answers. But lets do some rudimentary logging for now...
 729          */
 730         log_write(LS_RESOLVER, L_ERROR, 0, "irc_res.c bogus type %d", type);
 731
 732         if ((char*)current + rd_length >= (char*)current)
 733           current += rd_length;
 734         else
 735           return(0);
 736
 737         break;
 738     }
 739   }
 740
 741   return(1);
 742 }
 743
 744 /** Read a DNS reply from the nameserver and process it.
 745  * @param[in] ev I/O activity event for resolver socket.
 746  */
 747 static void
 748 res_readreply(struct Event *ev)
 749 {
 750   struct irc_sockaddr lsin;
 751   struct Socket *sock;
 752   char buf[sizeof(HEADER) + MAXPACKET];
 753   HEADER *header;
 754   struct reslist *request = NULL;
 755   struct DNSReply *reply  = NULL;
 756   unsigned int rc;
 757   int answer_count;
 758
 759   assert((ev_socket(ev) == &res_socket_v4) || (ev_socket(ev) == &res_socket_v6));
 760   sock = ev_socket(ev);
 761
 762   if (IO_SUCCESS != os_recvfrom_nonb(s_fd(sock), buf, sizeof(buf), &rc, &lsin)
 763       || (rc <= sizeof(HEADER)))
 764     return;
 765
 766   /*
 767    * check against possibly fake replies
 768    */
 769   if (!res_ourserver(&lsin))
 770     return;
 771
 772   /*
 773    * convert DNS reply reader from Network byte order to CPU byte order.
 774    */
 775   header = (HEADER *)buf;
 776   header->ancount = ntohs(header->ancount);
 777   header->qdcount = ntohs(header->qdcount);
 778   header->nscount = ntohs(header->nscount);
 779   header->arcount = ntohs(header->arcount);
 780
 781   /*
 782    * response for an id which we have already received an answer for
 783    * just ignore this response.
 784    */
 785   if (0 == (request = find_id(header->id)))
 786     return;
 787
 788   if ((header->rcode != NO_ERRORS) || (header->ancount == 0))
 789   {
 790     if (SERVFAIL == header->rcode)
 791       resend_query(request);
 792     else
 793     {
 794       /*
 795        * If we haven't already tried this, and we're looking up AAAA, try A
 796        * now
 797        */
 798
 799       if (request->state == REQ_AAAA && request->type == T_AAAA)
 800       {
 801         request->timeout += feature_int(FEAT_IRCD_RES_TIMEOUT);
 802         resend_query(request);
 803       }
 804       else if (request->type == T_PTR && request->state != REQ_INT &&
 805                !irc_in_addr_is_ipv4(&request->addr))
 806       {
 807         request->state = REQ_INT;
 808         request->timeout += feature_int(FEAT_IRCD_RES_TIMEOUT);
 809         resend_query(request);
 810       }
 811       else
 812       {
 813         /*
 814          * If a bad error was returned, we stop here and don't send
 815          * send any more (no retries granted).
 816          */
 817         Debug((DEBUG_DNS, "Request %p has bad response (state %d type %d rcode %d)", request, request->state, request->type, header->rcode));
 818         (*request->query.callback)(request->query.vptr, 0);
 819         rem_request(request);
 820       }
 821     }
 822
 823     return;
 824   }
 825   /*
 826    * If this fails there was an error decoding the received packet,
 827    * try it again and hope it works the next time.
 828    */
 829   answer_count = proc_answer(request, header, buf, buf + rc);
 830
 831   if (answer_count)
 832   {
 833     if (request->type == T_PTR)
 834     {
 835       if (request->name == NULL)
 836       {
 837         /*
 838          * got a PTR response with no name, something bogus is happening
 839          * don't bother trying again, the client address doesn't resolve
 840          */
 841         Debug((DEBUG_DNS, "Request %p PTR had empty name", request));
 842         (*request->query.callback)(request->query.vptr, reply);
 843         rem_request(request);
 844         return;
 845       }
 846
 847       /*
 848        * Lookup the 'authoritative' name that we were given for the
 849        * ip#.
 850        */
 851 #ifdef IPV6
 852       if (!irc_in_addr_is_ipv4(&request->addr))
 853         do_query_name(&request->query, request->name, NULL, T_AAAA);
 854       else
 855 #endif
 856       do_query_name(&request->query, request->name, NULL, T_A);
 857       Debug((DEBUG_DNS, "Request %p switching to forward resolution", request));
 858       rem_request(request);
 859     }
 860     else
 861     {
 862       /*
 863        * got a name and address response, client resolved
 864        */
 865       reply = make_dnsreply(request);
 866       (*request->query.callback)(request->query.vptr, (reply) ? reply : 0);
 867       Debug((DEBUG_DNS, "Request %p got forward resolution", request));
 868       rem_request(request);
 869     }
 870   }
 871   else if (!request->sent)
 872   {
 873     /* XXX - we got a response for a query we didn't send with a valid id?
 874      * this should never happen, bail here and leave the client unresolved
 875      */
 876     assert(0);
 877
 878     /* XXX don't leak it */
 879     Debug((DEBUG_DNS, "Request %p was unexpected(!)", request));
 880     rem_request(request);
 881   }
 882 }
 883
 884 /** Build a DNSReply for a completed request.
 885  * @param[in] request Completed DNS request.
 886  * @return Newly allocated DNSReply containing host name and address.
 887  */
 888 static struct DNSReply *
 889 make_dnsreply(struct reslist *request)
 890 {
 891   struct DNSReply *cp;
 892   assert(request != 0);
 893
 894   cp = (struct DNSReply *)MyMalloc(sizeof(struct DNSReply));
 895
 896   DupString(cp->h_name, request->name);
 897   memcpy(&cp->addr, &request->addr, sizeof(cp->addr));
 898   return(cp);
 899 }
 900
 901 /** Statistics callback to list DNS servers.
 902  * @param[in] source_p Client requesting statistics.
 903  * @param[in] sd Stats descriptor for request (ignored).
 904  * @param[in] param Extra parameter from user (ignored).
 905  */
 906 void
 907 report_dns_servers(struct Client *source_p, const struct StatDesc *sd, char *param)
 908 {
 909   int i;
 910   char ipaddr[128];
 911
 912   for (i = 0; i < irc_nscount; i++)
 913   {
 914     ircd_ntoa_r(ipaddr, &irc_nsaddr_list[i].addr);
 915     send_reply(source_p, RPL_STATSALINE, ipaddr);
 916   }
 917 }
 918
 919 /** Report memory usage to a client.
 920  * @param[in] sptr Client requesting information.
 921  * @return Total memory used by pending requests.
 922  */
 923 size_t
 924 cres_mem(struct Client* sptr)
 925 {
 926   struct dlink *dlink;
 927   struct reslist *request;
 928   size_t request_mem   = 0;
 929   int    request_count = 0;
 930
 931   if (request_list.next) {
 932     for (dlink = request_list.next; dlink != &request_list; dlink = dlink->next) {
 933       request = (struct reslist*)dlink;
 934       request_mem += sizeof(*request);
 935       if (request->name)
 936         request_mem += strlen(request->name) + 1;
 937       ++request_count;
 938     }
 939   }
 940
 941   send_reply(sptr, SND_EXPLICIT | RPL_STATSDEBUG,
 942              ":Resolver: requests %d(%d)", request_count, request_mem);
 943   return request_mem;
 944 }