ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/svn/ircd-hybrid/trunk/src/res.c
(Generate patch)

Comparing:
ircd-hybrid-7.2/src/irc_res.c (file contents), Revision 994 by michael, Fri Aug 21 09:15:31 2009 UTC vs.
ircd-hybrid/trunk/src/res.c (file contents), Revision 4408 by michael, Wed Aug 6 21:19:38 2014 UTC

# Line 1 | Line 1
1   /*
2 < * A rewrite of Darren Reeds original res.c As there is nothing
3 < * left of Darrens original code, this is now licensed by the hybrid group.
2 > *  ircd-hybrid: an advanced, lightweight Internet Relay Chat Daemon (ircd)
3 > *
4 > *  Copyright (c) 1997-2014 ircd-hybrid development team
5 > *
6 > *  This program is free software; you can redistribute it and/or modify
7 > *  it under the terms of the GNU General Public License as published by
8 > *  the Free Software Foundation; either version 2 of the License, or
9 > *  (at your option) any later version.
10 > *
11 > *  This program is distributed in the hope that it will be useful,
12 > *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13 > *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 > *  GNU General Public License for more details.
15 > *
16 > *  You should have received a copy of the GNU General Public License
17 > *  along with this program; if not, write to the Free Software
18 > *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
19 > *  USA
20 > */
21 >
22 > /*! \file res.c
23 > * \brief ircd resolver functions
24 > * \version $Id$
25 > */
26 >
27 > /*
28 > * A rewrite of Darren Reed's original res.c As there is nothing
29 > * left of Darren's original code, this is now licensed by the hybrid group.
30   * (Well, some of the function names are the same, and bits of the structs..)
31   * You can use it where it is useful, free even. Buy us a beer and stuff.
32   *
33   * The authors takes no responsibility for any damage or loss
34   * of property which results from the use of this software.
35   *
10 * $Id$
11 *
36   * July 1999 - Rewrote a bunch of stuff here. Change hostent builder code,
37   *     added callbacks and reference counting of returned hostents.
38   *     --Bleep (Thomas Helvey <tomh@inxpress.net>)
39   *
40   * This was all needlessly complicated for irc. Simplified. No more hostent
41 < * All we really care about is the IP -> hostname mappings. Thats all.
41 > * All we really care about is the IP -> hostname mappings. That's all.
42   *
43   * Apr 28, 2003 --cryogen and Dianora
44   */
45  
46   #include "stdinc.h"
23 #include "tools.h"
24 #include "client.h"
47   #include "list.h"
48 < #include "common.h"
48 > #include "client.h"
49   #include "event.h"
50   #include "irc_string.h"
29 #include "sprintf_irc.h"
51   #include "ircd.h"
52   #include "numeric.h"
32 #include "restart.h"
53   #include "rng_mt.h"
54   #include "fdlist.h"
35 #include "fileio.h" /* for fbopen / fbclose / fbputs */
55   #include "s_bsd.h"
56 < #include "s_log.h"
56 > #include "log.h"
57 > #include "misc.h"
58   #include "send.h"
59   #include "memory.h"
60 < #include "irc_res.h"
61 < #include "irc_reslib.h"
62 < #include "irc_getnameinfo.h"
60 > #include "mempool.h"
61 > #include "res.h"
62 > #include "reslib.h"
63  
64   #if (CHAR_BIT != 8)
65 < #error this code needs to be able to address individual octets
65 > #error this code needs to be able to address individual octets
66   #endif
67  
68   static PF res_readreply;
69  
70 < #define MAXPACKET      1024  /* rfc sez 512 but we expand names so ... */
71 < #define RES_MAXALIASES 35    /* maximum aliases allowed */
52 < #define RES_MAXADDRS   35    /* maximum addresses allowed */
53 < #define AR_TTL         600   /* TTL in seconds for dns cache entries */
70 > #define MAXPACKET      1024  /**< rfc says 512 but we expand names so ... */
71 > #define AR_TTL         600   /**< TTL in seconds for dns cache entries */
72  
73 < /* RFC 1104/1105 wasn't very helpful about what these fields
73 > /*
74 > * RFC 1104/1105 wasn't very helpful about what these fields
75   * should be named, so for now, we'll just name them this way.
76 < * we probably should look at what named calls them or something.
76 > * We probably should look at what named calls them or something.
77   */
78   #define TYPE_SIZE         (size_t)2
79   #define CLASS_SIZE        (size_t)2
# Line 62 | Line 81 | static PF res_readreply;
81   #define RDLENGTH_SIZE     (size_t)2
82   #define ANSWER_FIXED_SIZE (TYPE_SIZE + CLASS_SIZE + TTL_SIZE + RDLENGTH_SIZE)
83  
84 < typedef enum
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 */
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   #ifdef IPV6
90 <  REQ_AAAA,  /* Looking up an AAAA */
90 >  REQ_AAAA,  /**< Looking up an AAAA */
91   #endif
92 <  REQ_CNAME  /* We got a CNAME in response, we better get a real answer next */
92 >  REQ_CNAME  /**< We got a CNAME in response, we better get a real answer next */
93   } request_state;
94  
95 < struct reslist
95 > struct reslist
96   {
97 <  dlink_node node;
98 <  int id;
99 <  int sent;                /* number of requests sent */
100 <  request_state state;     /* State the resolver machine is in */
101 <  time_t ttl;
102 <  char type;
103 <  char retries;            /* retry counter */
104 <  char sends;              /* number of sends (>1 means resent) */
105 <  char resend;             /* send flag. 0 == dont resend */
106 <  time_t sentat;
107 <  time_t timeout;
108 <  struct irc_ssaddr addr;
109 <  char *name;
110 <  dns_callback_fnc callback;
111 <  void *callback_ctx;
97 >  dlink_node node;                       /**< Doubly linked list node. */
98 >  int id;                                /**< Request ID (from request header). */
99 >  int sent;                              /**< Number of requests sent */
100 >  request_state state;                   /**< State the resolver machine is in */
101 >  char type;                             /**< Current request type. */
102 >  char retries;                          /**< Retry counter */
103 >  unsigned int sends;                    /**< Number of sends (>1 means resent). */
104 >  char resend;                           /**< Send flag; 0 == don't resend. */
105 >  time_t sentat;                         /**< Timestamp we last sent this request. */
106 >  time_t timeout;                        /**< When this request times out. */
107 >  struct irc_ssaddr addr;                /**< Address for this request. */
108 >  char name[RFC1035_MAX_DOMAIN_LENGTH];  /**< Hostname for this request. */
109 >  size_t namelength;                     /**< Actual hostname length. */
110 >  dns_callback_fnc callback;             /**< Callback function on completion. */
111 >  void *callback_ctx;                    /**< Context pointer for callback. */
112   };
113  
114   static fde_t ResolverFileDescriptor;
115 < static dlink_list request_list = { NULL, NULL, 0 };
115 > static dlink_list request_list;
116 > static mp_pool_t *dns_pool;
117 >
118 >
119 > /*
120 > * rem_request - remove a request from the list.
121 > * This must also free any memory that has been allocated for
122 > * temporary storage of DNS results.
123 > */
124 > static void
125 > rem_request(struct reslist *request)
126 > {
127 >  dlinkDelete(&request->node, &request_list);
128 >  mp_pool_release(request);
129 > }
130  
131 < static void rem_request(struct reslist *request);
132 < static struct reslist *make_request(dns_callback_fnc callback, void *);
133 < static void do_query_name(dns_callback_fnc callback, void *,
134 <                          const char *, struct reslist *, int);
135 < static void do_query_number(dns_callback_fnc callback, void *ctx,
136 <                            const struct irc_ssaddr *,
137 <                            struct reslist *request);
105 < static void query_name(const char *name, int query_class, int query_type,
106 <                       struct reslist *request);
107 < static int send_res_msg(const char *buf, int len, int count);
108 < static void resend_query(struct reslist *request);
109 < static int proc_answer(struct reslist *request, HEADER *header, char *, char *);
110 < static struct reslist *find_id(int id);
111 < static struct DNSReply *make_dnsreply(struct reslist *request);
112 <
113 < extern struct irc_ssaddr irc_nsaddr_list[IRCD_MAXNS];
114 < extern int irc_nscount;
115 < extern char irc_domain[HOSTLEN+1];
131 > /*
132 > * make_request - Create a DNS request record for the server.
133 > */
134 > static struct reslist *
135 > make_request(dns_callback_fnc callback, void *ctx)
136 > {
137 >  struct reslist *request = mp_pool_get(dns_pool);
138  
139 +  request->sentat       = CurrentTime;
140 +  request->retries      = 2;
141 +  request->resend       = 1;
142 +  request->timeout      = 4;  /* Start at 4 and exponential inc. */
143 +  request->state        = REQ_IDLE;
144 +  request->callback     = callback;
145 +  request->callback_ctx = ctx;
146 +
147 +  dlinkAdd(request, &request->node, &request_list);
148 +  return request;
149 + }
150  
151   /*
152   * int
# Line 127 | Line 160 | extern char irc_domain[HOSTLEN+1];
160   *      revised for ircd, cryogen(stu) may03
161   */
162   static int
163 < res_ourserver(const struct irc_ssaddr *inp)
163 > res_ourserver(const struct irc_ssaddr *inp)
164   {
165   #ifdef IPV6
166    const struct sockaddr_in6 *v6;
167    const struct sockaddr_in6 *v6in = (const struct sockaddr_in6 *)inp;
168   #endif
169    const struct sockaddr_in *v4;
170 <  const struct sockaddr_in *v4in = (const struct sockaddr_in *)inp;
138 <  int ns;
170 >  const struct sockaddr_in *v4in = (const struct sockaddr_in *)inp;
171  
172 <  for (ns = 0; ns < irc_nscount; ns++)
172 >  for (unsigned int i = 0; i < irc_nscount; ++i)
173    {
174 <    const struct irc_ssaddr *srv = &irc_nsaddr_list[ns];
174 >    const struct irc_ssaddr *srv = &irc_nsaddr_list[i];
175   #ifdef IPV6
176      v6 = (const struct sockaddr_in6 *)srv;
177   #endif
178      v4 = (const struct sockaddr_in *)srv;
179  
180 <    /* could probably just memcmp(srv, inp, srv.ss_len) here
180 >    /*
181 >     * Could probably just memcmp(srv, inp, srv.ss_len) here
182       * but we'll air on the side of caution - stu
150     *
183       */
184      switch (srv->ss.ss_family)
185      {
# Line 155 | Line 187 | res_ourserver(const struct irc_ssaddr *i
187        case AF_INET6:
188          if (srv->ss.ss_family == inp->ss.ss_family)
189            if (v6->sin6_port == v6in->sin6_port)
190 <            if ((memcmp(&v6->sin6_addr.s6_addr, &v6in->sin6_addr.s6_addr,
191 <                    sizeof(struct in6_addr)) == 0) ||
192 <                (memcmp(&v6->sin6_addr.s6_addr, &in6addr_any,
161 <                        sizeof(struct in6_addr)) == 0))
162 <              return(1);
190 >            if (!memcmp(&v6->sin6_addr.s6_addr, &v6in->sin6_addr.s6_addr,
191 >                        sizeof(struct in6_addr)))
192 >              return 1;
193          break;
194   #endif
195        case AF_INET:
196          if (srv->ss.ss_family == inp->ss.ss_family)
197            if (v4->sin_port == v4in->sin_port)
198 <            if ((v4->sin_addr.s_addr == INADDR_ANY) ||
199 <                (v4->sin_addr.s_addr == v4in->sin_addr.s_addr))
170 <              return(1);
198 >            if (v4->sin_addr.s_addr == v4in->sin_addr.s_addr)
199 >              return 1;
200          break;
201        default:
202          break;
203      }
204    }
205  
206 <  return(0);
178 < }
179 <
180 < /*
181 < * timeout_query_list - Remove queries from the list which have been
182 < * there too long without being resolved.
183 < */
184 < static time_t
185 < timeout_query_list(time_t now)
186 < {
187 <  dlink_node *ptr;
188 <  dlink_node *next_ptr;
189 <  struct reslist *request;
190 <  time_t next_time = 0;
191 <  time_t timeout   = 0;
192 <
193 <  DLINK_FOREACH_SAFE(ptr, next_ptr, request_list.head)
194 <  {
195 <    request = ptr->data;
196 <    timeout = request->sentat + request->timeout;
197 <
198 <    if (now >= timeout)
199 <    {
200 <      if (--request->retries <= 0)
201 <      {
202 <        (*request->callback)(request->callback_ctx, NULL, NULL);
203 <        rem_request(request);
204 <        continue;
205 <      }
206 <      else
207 <      {
208 <        request->sentat = now;
209 <        request->timeout += request->timeout;
210 <        resend_query(request);
211 <      }
212 <    }
213 <
214 <    if ((next_time == 0) || timeout < next_time)
215 <    {
216 <      next_time = timeout;
217 <    }
218 <  }
219 <
220 <  return((next_time > now) ? next_time : (now + AR_TTL));
221 < }
222 <
223 < /*
224 < * timeout_resolver - check request list
225 < */
226 < static void
227 < timeout_resolver(void *notused)
228 < {
229 <  timeout_query_list(CurrentTime);
206 >  return 0;
207   }
208  
209   /*
# Line 245 | Line 222 | start_resolver(void)
222        return;
223  
224      /* At the moment, the resolver FD data is global .. */
225 <    comm_setselect(&ResolverFileDescriptor, COMM_SELECT_READ,
249 <        res_readreply, NULL, 0);
250 <    eventAdd("timeout_resolver", timeout_resolver, NULL, 1);
225 >    comm_setselect(&ResolverFileDescriptor, COMM_SELECT_READ, res_readreply, NULL, 0);
226    }
227   }
228  
229   /*
255 * init_resolver - initialize resolver and resolver library
256 */
257 void
258 init_resolver(void)
259 {
260  memset(&ResolverFileDescriptor, 0, sizeof(fde_t));
261  start_resolver();
262 }
263
264 /*
230   * restart_resolver - reread resolv.conf, reopen socket
231   */
232   void
233   restart_resolver(void)
234   {
235    fd_close(&ResolverFileDescriptor);
271  eventDelete(timeout_resolver, NULL); /* -ddosen */
236    start_resolver();
237   }
238  
239   /*
240 < * add_local_domain - Add the domain to hostname, if it is missing
277 < * (as suggested by eps@TOASTER.SFSU.EDU)
278 < */
279 < void
280 < add_local_domain(char *hname, size_t size)
281 < {
282 <  /* try to fix up unqualified names
283 <   */
284 <  if (strchr(hname, '.') == NULL)
285 <  {
286 <    if (irc_domain[0])
287 <    {
288 <      size_t len = strlen(hname);
289 <
290 <      if ((strlen(irc_domain) + len + 2) < size)
291 <      {
292 <        hname[len++] = '.';
293 <        strcpy(hname + len, irc_domain);
294 <      }
295 <    }
296 <  }
297 < }
298 <
299 < /*
300 < * rem_request - remove a request from the list.
301 < * This must also free any memory that has been allocated for
302 < * temporary storage of DNS results.
303 < */
304 < static void
305 < rem_request(struct reslist *request)
306 < {
307 <  dlinkDelete(&request->node, &request_list);
308 <  MyFree(request->name);
309 <  MyFree(request);
310 < }
311 <
312 < /*
313 < * make_request - Create a DNS request record for the server.
314 < */
315 < static struct reslist *
316 < make_request(dns_callback_fnc callback, void *ctx)
317 < {
318 <  struct reslist *request;
319 <
320 <  request = (struct reslist *)MyMalloc(sizeof(struct reslist));
321 <
322 <  request->sentat       = CurrentTime;
323 <  request->retries      = 3;
324 <  request->resend       = 1;
325 <  request->timeout      = 4;    /* start at 4 and exponential inc. */
326 <  request->state        = REQ_IDLE;
327 <  request->callback     = callback;
328 <  request->callback_ctx = ctx;
329 <
330 <  dlinkAdd(request, &request->node, &request_list);
331 <  return(request);
332 < }
333 <
334 < /*
335 < * delete_resolver_queries - cleanup outstanding queries
240 > * delete_resolver_queries - cleanup outstanding queries
241   * for which there no longer exist clients or conf lines.
242   */
243   void
244   delete_resolver_queries(const void *vptr)
245   {
246 <  dlink_node *ptr;
342 <  dlink_node *next_ptr;
343 <  struct reslist *request;
246 >  dlink_node *ptr = NULL, *ptr_next = NULL;
247  
248 <  DLINK_FOREACH_SAFE(ptr, next_ptr, request_list.head)
248 >  DLINK_FOREACH_SAFE(ptr, ptr_next, request_list.head)
249    {
250 <    if ((request = ptr->data) != NULL)
251 <    {
252 <      if (request->callback_ctx == vptr)
253 <        rem_request(request);
351 <    }
250 >    struct reslist *request = ptr->data;
251 >
252 >    if (request->callback_ctx == vptr)
253 >      rem_request(request);
254    }
255   }
256  
# Line 356 | Line 258 | delete_resolver_queries(const void *vptr
258   * send_res_msg - sends msg to all nameservers found in the "_res" structure.
259   * This should reflect /etc/resolv.conf. We will get responses
260   * which arent needed but is easier than checking to see if nameserver
261 < * isnt present. Returns number of messages successfully sent to
261 > * isn't present. Returns number of messages successfully sent to
262   * nameservers or -1 if no successful sends.
263   */
264   static int
265 < send_res_msg(const char *msg, int len, int rcount)
265 > send_res_msg(const char *msg, int len, unsigned int rcount)
266   {
365  int i;
267    int sent = 0;
268 <  int max_queries = IRCD_MIN(irc_nscount, rcount);
268 >  unsigned int max_queries = IRCD_MIN(irc_nscount, rcount);
269  
270    /* RES_PRIMARY option is not implemented
271     * if (res.options & RES_PRIMARY || 0 == max_queries)
# Line 372 | Line 273 | send_res_msg(const char *msg, int len, i
273    if (max_queries == 0)
274      max_queries = 1;
275  
276 <  for (i = 0; i < max_queries; i++)
276 >  for (unsigned int i = 0; i < max_queries; ++i)
277    {
278 <    if (sendto(ResolverFileDescriptor.fd, msg, len, 0,
279 <        (struct sockaddr*)&(irc_nsaddr_list[i]),
280 <        irc_nsaddr_list[i].ss_len) == len)
278 >    if (sendto(ResolverFileDescriptor.fd, msg, len, 0,
279 >        (struct sockaddr*)&(irc_nsaddr_list[i]),
280 >        irc_nsaddr_list[i].ss_len) == len)
281        ++sent;
282    }
283  
284 <  return(sent);
284 >  return sent;
285   }
286  
287   /*
# Line 389 | Line 290 | send_res_msg(const char *msg, int len, i
290   static struct reslist *
291   find_id(int id)
292   {
293 <  dlink_node *ptr;
393 <  struct reslist *request;
293 >  dlink_node *ptr = NULL;
294  
295    DLINK_FOREACH(ptr, request_list.head)
296    {
297 <    request = ptr->data;
297 >    struct reslist *request = ptr->data;
298  
299      if (request->id == id)
300 <      return(request);
300 >      return request;
301    }
302  
303 <  return(NULL);
404 < }
405 <
406 < /*
407 < * gethost_byname_type - get host address from name
408 < *
409 < */
410 < void
411 < gethost_byname_type(dns_callback_fnc callback, void *ctx, const char *name, int type)
412 < {
413 <  assert(name != 0);
414 <  do_query_name(callback, ctx, name, NULL, type);
303 >  return NULL;
304   }
305  
306   /*
307 < * gethost_byname - wrapper for _type - send T_AAAA first if IPV6 supported
307 > * query_name - generate a query based on class, type and name.
308   */
309 < void
310 < gethost_byname(dns_callback_fnc callback, void *ctx, const char *name)
309 > static void
310 > query_name(const char *name, int query_class, int type,
311 >           struct reslist *request)
312   {
313 < #ifdef IPV6
314 <  gethost_byname_type(callback, ctx, name, T_AAAA);
425 < #else
426 <  gethost_byname_type(callback, ctx, name, T_A);
427 < #endif
428 < }
313 >  char buf[MAXPACKET];
314 >  int request_len = 0;
315  
316 < /*
317 < * gethost_byaddr - get host name from address
318 < */
319 < void
320 < gethost_byaddr(dns_callback_fnc callback, void *ctx, const struct irc_ssaddr *addr)
321 < {
322 <  do_query_number(callback, ctx, addr, NULL);
316 >  memset(buf, 0, sizeof(buf));
317 >
318 >  if ((request_len = irc_res_mkquery(name, query_class, type,
319 >      (unsigned char *)buf, sizeof(buf))) > 0)
320 >  {
321 >    HEADER *header = (HEADER *)buf;
322 >
323 >    /*
324 >     * Generate an unique id.
325 >     * NOTE: we don't have to worry about converting this to and from
326 >     * network byte order, the nameserver does not interpret this value
327 >     * and returns it unchanged.
328 >     */
329 >    do
330 >      header->id = (header->id + genrand_int32()) & 0xFFFF;
331 >    while (find_id(header->id));
332 >
333 >    request->id = header->id;
334 >    ++request->sends;
335 >
336 >    request->sent += send_res_msg(buf, request_len, request->sends);
337 >  }
338   }
339  
340   /*
# Line 443 | Line 344 | static void
344   do_query_name(dns_callback_fnc callback, void *ctx, const char *name,
345                struct reslist *request, int type)
346   {
347 <  char host_name[HOSTLEN + 1];
347 >  char host_name[RFC1035_MAX_DOMAIN_LENGTH + 1];
348  
349 <  strlcpy(host_name, name, HOSTLEN + 1);
449 <  add_local_domain(host_name, HOSTLEN + 1);
349 >  strlcpy(host_name, name, sizeof(host_name));
350  
351    if (request == NULL)
352    {
353 <    request       = make_request(callback, ctx);
354 <    request->name = MyMalloc(strlen(host_name) + 1);
355 <    request->type = type;
456 <    strcpy(request->name, host_name);
353 >    request             = make_request(callback, ctx);
354 >    request->type       = type;
355 >    request->namelength = strlcpy(request->name, host_name, sizeof(request->name));
356   #ifdef IPV6
357      if (type != T_A)
358        request->state = REQ_AAAA;
# Line 474 | Line 373 | do_query_number(dns_callback_fnc callbac
373                  const struct irc_ssaddr *addr,
374                  struct reslist *request)
375   {
376 <  char ipbuf[128];
478 <  const unsigned char *cp;
376 >  char ipbuf[128] = "";
377  
378    if (addr->ss.ss_family == AF_INET)
379    {
380      const struct sockaddr_in *v4 = (const struct sockaddr_in *)addr;
381 <    cp = (const unsigned char*)&v4->sin_addr.s_addr;
381 >    const unsigned char *cp = (const unsigned char *)&v4->sin_addr.s_addr;
382  
383 <    ircsprintf(ipbuf, "%u.%u.%u.%u.in-addr.arpa.",
384 <               (unsigned int)(cp[3]), (unsigned int)(cp[2]),
385 <               (unsigned int)(cp[1]), (unsigned int)(cp[0]));
383 >    snprintf(ipbuf, sizeof(ipbuf), "%u.%u.%u.%u.in-addr.arpa.",
384 >             (unsigned int)(cp[3]), (unsigned int)(cp[2]),
385 >             (unsigned int)(cp[1]), (unsigned int)(cp[0]));
386    }
387   #ifdef IPV6
388    else if (addr->ss.ss_family == AF_INET6)
389    {
390      const struct sockaddr_in6 *v6 = (const struct sockaddr_in6 *)addr;
391 <    cp = (const unsigned char *)&v6->sin6_addr.s6_addr;
391 >    const unsigned char *cp = (const unsigned char *)&v6->sin6_addr.s6_addr;
392  
393 <    sprintf(ipbuf, "%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x."
394 <                   "%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.ip6.arpa.",
395 <                  (unsigned int)(cp[15]&0xf), (unsigned int)(cp[15]>>4),
396 <                  (unsigned int)(cp[14]&0xf), (unsigned int)(cp[14]>>4),
397 <                  (unsigned int)(cp[13]&0xf), (unsigned int)(cp[13]>>4),
398 <                  (unsigned int)(cp[12]&0xf), (unsigned int)(cp[12]>>4),
399 <                  (unsigned int)(cp[11]&0xf), (unsigned int)(cp[11]>>4),
400 <                  (unsigned int)(cp[10]&0xf), (unsigned int)(cp[10]>>4),
401 <                  (unsigned int)(cp[9]&0xf), (unsigned int)(cp[9]>>4),
402 <                  (unsigned int)(cp[8]&0xf), (unsigned int)(cp[8]>>4),
403 <                  (unsigned int)(cp[7]&0xf), (unsigned int)(cp[7]>>4),
404 <                  (unsigned int)(cp[6]&0xf), (unsigned int)(cp[6]>>4),
405 <                  (unsigned int)(cp[5]&0xf), (unsigned int)(cp[5]>>4),
406 <                  (unsigned int)(cp[4]&0xf), (unsigned int)(cp[4]>>4),
407 <                  (unsigned int)(cp[3]&0xf), (unsigned int)(cp[3]>>4),
408 <                  (unsigned int)(cp[2]&0xf), (unsigned int)(cp[2]>>4),
409 <                  (unsigned int)(cp[1]&0xf), (unsigned int)(cp[1]>>4),
410 <                  (unsigned int)(cp[0]&0xf), (unsigned int)(cp[0]>>4));
393 >    snprintf(ipbuf, sizeof(ipbuf),
394 >             "%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x."
395 >             "%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.ip6.arpa.",
396 >             (unsigned int)(cp[15] & 0xf), (unsigned int)(cp[15] >> 4),
397 >             (unsigned int)(cp[14] & 0xf), (unsigned int)(cp[14] >> 4),
398 >             (unsigned int)(cp[13] & 0xf), (unsigned int)(cp[13] >> 4),
399 >             (unsigned int)(cp[12] & 0xf), (unsigned int)(cp[12] >> 4),
400 >             (unsigned int)(cp[11] & 0xf), (unsigned int)(cp[11] >> 4),
401 >             (unsigned int)(cp[10] & 0xf), (unsigned int)(cp[10] >> 4),
402 >             (unsigned int)(cp[9] & 0xf), (unsigned int)(cp[9] >> 4),
403 >             (unsigned int)(cp[8] & 0xf), (unsigned int)(cp[8] >> 4),
404 >             (unsigned int)(cp[7] & 0xf), (unsigned int)(cp[7] >> 4),
405 >             (unsigned int)(cp[6] & 0xf), (unsigned int)(cp[6] >> 4),
406 >             (unsigned int)(cp[5] & 0xf), (unsigned int)(cp[5] >> 4),
407 >             (unsigned int)(cp[4] & 0xf), (unsigned int)(cp[4] >> 4),
408 >             (unsigned int)(cp[3] & 0xf), (unsigned int)(cp[3] >> 4),
409 >             (unsigned int)(cp[2] & 0xf), (unsigned int)(cp[2] >> 4),
410 >             (unsigned int)(cp[1] & 0xf), (unsigned int)(cp[1] >> 4),
411 >             (unsigned int)(cp[0] & 0xf), (unsigned int)(cp[0] >> 4));
412    }
413   #endif
414    if (request == NULL)
# Line 517 | Line 416 | do_query_number(dns_callback_fnc callbac
416      request       = make_request(callback, ctx);
417      request->type = T_PTR;
418      memcpy(&request->addr, addr, sizeof(struct irc_ssaddr));
520    request->name = MyMalloc(HOSTLEN + 1);
419    }
420  
421    query_name(ipbuf, C_IN, T_PTR, request);
422   }
423  
424   /*
425 < * query_name - generate a query based on class, type and name.
425 > * gethost_byname_type - get host address from name
426 > *
427   */
428 < static void
429 < query_name(const char *name, int query_class, int type,
531 <           struct reslist *request)
428 > void
429 > gethost_byname_type(dns_callback_fnc callback, void *ctx, const char *name, int type)
430   {
431 <  char buf[MAXPACKET];
432 <  int request_len = 0;
433 <
536 <  memset(buf, 0, sizeof(buf));
537 <
538 <  if ((request_len = irc_res_mkquery(name, query_class, type,
539 <      (unsigned char *)buf, sizeof(buf))) > 0)
540 <  {
541 <    HEADER *header = (HEADER *)buf;
542 <
543 <    /*
544 <     * generate an unique id
545 <     * NOTE: we don't have to worry about converting this to and from
546 <     * network byte order, the nameserver does not interpret this value
547 <     * and returns it unchanged
548 <     */
549 <    do
550 <      header->id = (header->id + genrand_int32()) & 0xffff;
551 <    while (find_id(header->id));
431 >  assert(name);
432 >  do_query_name(callback, ctx, name, NULL, type);
433 > }
434  
435 <    request->id = header->id;
436 <    ++request->sends;
435 > /*
436 > * gethost_byname - wrapper for _type - send T_AAAA first if IPV6 supported
437 > */
438 > void
439 > gethost_byname(dns_callback_fnc callback, void *ctx, const char *name)
440 > {
441 > #ifdef IPV6
442 >  gethost_byname_type(callback, ctx, name, T_AAAA);
443 > #else
444 >  gethost_byname_type(callback, ctx, name, T_A);
445 > #endif
446 > }
447  
448 <    request->sent += send_res_msg(buf, request_len, request->sends);
449 <  }
448 > /*
449 > * gethost_byaddr - get host name from address
450 > */
451 > void
452 > gethost_byaddr(dns_callback_fnc callback, void *ctx, const struct irc_ssaddr *addr)
453 > {
454 >  do_query_number(callback, ctx, addr, NULL);
455   }
456  
457   static void
# Line 572 | Line 469 | resend_query(struct reslist *request)
469        do_query_name(NULL, NULL, request->name, request, request->type);
470        break;
471   #ifdef IPV6
472 <    case T_AAAA:
576 <      /* didnt work, try A */
472 >    case T_AAAA:  /* Didn't work, try A */
473        if (request->state == REQ_AAAA)
474          do_query_name(NULL, NULL, request->name, request, T_A);
475   #endif
# Line 586 | Line 482 | resend_query(struct reslist *request)
482   * proc_answer - process name server reply
483   */
484   static int
485 < proc_answer(struct reslist *request, HEADER* header, char* buf, char* eob)
485 > proc_answer(struct reslist *request, HEADER *header, char *buf, char *eob)
486   {
487 <  char hostbuf[HOSTLEN + 100]; /* working buffer */
487 >  char hostbuf[RFC1035_MAX_DOMAIN_LENGTH + 100]; /* working buffer */
488    unsigned char *current;      /* current position in buf */
593  int query_class;             /* answer class */
489    int type;                    /* answer type */
490    int n;                       /* temp count */
491    int rd_length;
# Line 605 | Line 500 | proc_answer(struct reslist *request, HEA
500      if ((n = irc_dn_skipname(current, (unsigned char *)eob)) < 0)
501        break;
502  
503 <    current += (size_t) n + QFIXEDSZ;
503 >    current += (size_t)n + QFIXEDSZ;
504    }
505  
506    /*
507 <   * process each answer sent to us blech.
507 >   * Process each answer sent to us blech.
508     */
509    while (header->ancount > 0 && (char *)current < eob)
510    {
# Line 618 | Line 513 | proc_answer(struct reslist *request, HEA
513      n = irc_dn_expand((unsigned char *)buf, (unsigned char *)eob, current,
514          hostbuf, sizeof(hostbuf));
515  
516 <    if (n < 0)
517 <    {
623 <      /*
624 <       * broken message
625 <       */
626 <      return(0);
627 <    }
628 <    else if (n == 0)
629 <    {
630 <      /*
631 <       * no more answers left
632 <       */
633 <      return(0);
634 <    }
516 >    if (n < 0  /* Broken message */ || n == 0  /* No more answers left */)
517 >      return 0;
518  
519 <    hostbuf[HOSTLEN] = '\0';
519 >    hostbuf[RFC1035_MAX_DOMAIN_LENGTH] = '\0';
520  
521 <    /* With Address arithmetic you have to be very anal
521 >    /*
522 >     * With Address arithmetic you have to be very anal
523       * this code was not working on alpha due to that
524       * (spotted by rodder/jailbird/dianora)
525       */
526 <    current += (size_t) n;
526 >    current += (size_t)n;
527  
528      if (!(((char *)current + ANSWER_FIXED_SIZE) < eob))
529        break;
530  
531      type = irc_ns_get16(current);
532      current += TYPE_SIZE;
649
650    query_class = irc_ns_get16(current);
533      current += CLASS_SIZE;
652
653    request->ttl = irc_ns_get32(current);
534      current += TTL_SIZE;
655
535      rd_length = irc_ns_get16(current);
536      current += RDLENGTH_SIZE;
537  
538 <    /*
539 <     * Wait to set request->type until we verify this structure
538 >    /*
539 >     * Wait to set request->type until we verify this structure
540       */
541      switch (type)
542      {
543        case T_A:
544          if (request->type != T_A)
545 <          return(0);
545 >          return 0;
546  
547          /*
548 <         * check for invalid rd_length or too many addresses
548 >         * Check for invalid rd_length or too many addresses
549           */
550          if (rd_length != sizeof(struct in_addr))
551 <          return(0);
551 >          return 0;
552 >
553          v4 = (struct sockaddr_in *)&request->addr;
554          request->addr.ss_len = sizeof(struct sockaddr_in);
555          v4->sin_family = AF_INET;
556          memcpy(&v4->sin_addr, current, sizeof(struct in_addr));
557 <        return(1);
557 >        return 1;
558          break;
559   #ifdef IPV6
560        case T_AAAA:
561          if (request->type != T_AAAA)
562 <          return(0);
562 >          return 0;
563 >
564          if (rd_length != sizeof(struct in6_addr))
565 <          return(0);
565 >          return 0;
566 >
567          request->addr.ss_len = sizeof(struct sockaddr_in6);
568          v6 = (struct sockaddr_in6 *)&request->addr;
569          v6->sin6_family = AF_INET6;
570          memcpy(&v6->sin6_addr, current, sizeof(struct in6_addr));
571 <        return(1);
571 >        return 1;
572          break;
573   #endif
574        case T_PTR:
575          if (request->type != T_PTR)
576 <          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 */
576 >          return 0;
577  
578 <        strlcpy(request->name, hostbuf, HOSTLEN + 1);
578 >        n = irc_dn_expand((unsigned char *)buf, (unsigned char *)eob,
579 >                          current, hostbuf, sizeof(hostbuf));
580 >        if (n < 0  /* Broken message */ || n == 0  /* No more answers left */)
581 >          return 0;
582  
583 <        return(1);
583 >        request->namelength = strlcpy(request->name, hostbuf, sizeof(request->name));
584 >        return 1;
585          break;
586 <      case T_CNAME: /* first check we already havent started looking
587 <                       into a cname */
588 <        if (request->type != T_PTR)
709 <          return(0);
586 >      case T_CNAME:  /* First check we already haven't started looking into a cname */
587 >        if (request->type != T_PTR)
588 >          return 0;
589  
590          if (request->state == REQ_CNAME)
591          {
# Line 714 | Line 593 | proc_answer(struct reslist *request, HEA
593                              current, hostbuf, sizeof(hostbuf));
594  
595            if (n < 0)
596 <            return(0);
597 <          return(1);
596 >            return 0;
597 >          return 1;
598          }
599  
600          request->state = REQ_CNAME;
601          current += rd_length;
602          break;
603 <        
603 >
604        default:
605          /* XXX I'd rather just throw away the entire bogus thing
606           * but its possible its just a broken nameserver with still
607           * valid answers. But lets do some rudimentary logging for now...
608           */
609 <        ilog(L_ERROR, "irc_res.c bogus type %d", type);
609 >        ilog(LOG_TYPE_IRCD, "irc_res.c bogus type %d", type);
610          break;
611      }
612    }
613  
614 <  return(1);
614 >  return 1;
615   }
616  
617   /*
# Line 741 | Line 620 | proc_answer(struct reslist *request, HEA
620   static void
621   res_readreply(fde_t *fd, void *data)
622   {
623 <  char buf[sizeof(HEADER) + MAXPACKET]
745 <        /* Sparc and alpha need 16bit-alignment for accessing header->id
746 <         * (which is uint16_t). Because of the header = (HEADER*) buf;
747 <         * lateron, this is neeeded. --FaUl
748 <         */
749 < #if defined(__sparc__) || defined(__alpha__)  
750 <          __attribute__((aligned (16)))
751 < #endif
752 <          ;
623 >  char buf[sizeof(HEADER) + MAXPACKET];
624    HEADER *header;
625    struct reslist *request = NULL;
626 <  int rc;
626 >  ssize_t rc = 0;
627    socklen_t len = sizeof(struct irc_ssaddr);
628    struct irc_ssaddr lsin;
629  
630 <  rc = recvfrom(fd->fd, buf, sizeof(buf), 0, (struct sockaddr *)&lsin, &len);
631 <
632 <  /* Re-schedule a read *after* recvfrom, or we'll be registering
633 <   * interest where it'll instantly be ready for read :-) -- adrian
763 <   */
764 <  comm_setselect(fd, COMM_SELECT_READ, res_readreply, NULL, 0);
765 <
766 <  /* Better to cast the sizeof instead of rc */
767 <  if (rc <= (int)(sizeof(HEADER)))
768 <    return;
630 >  while ((rc = recvfrom(fd->fd, buf, sizeof(buf), 0, (struct sockaddr *)&lsin, &len)) != -1)
631 >  {
632 >    if (rc <= (ssize_t)sizeof(HEADER))
633 >      continue;
634  
635 <  /*
636 <   * convert DNS reply reader from Network byte order to CPU byte order.
637 <   */
638 <  header = (HEADER *)buf;
639 <  header->ancount = ntohs(header->ancount);
775 <  header->qdcount = ntohs(header->qdcount);
776 <  header->nscount = ntohs(header->nscount);
777 <  header->arcount = ntohs(header->arcount);
635 >    /*
636 >     * Check against possibly fake replies
637 >     */
638 >    if (!res_ourserver(&lsin))
639 >      continue;
640  
641 <  /*
642 <   * check against possibly fake replies
643 <   */
644 <  if (!res_ourserver(&lsin))
645 <    return;
641 >    /*
642 >     * Convert DNS reply reader from Network byte order to CPU byte order.
643 >     */
644 >    header = (HEADER *)buf;
645 >    header->ancount = ntohs(header->ancount);
646 >    header->qdcount = ntohs(header->qdcount);
647 >    header->nscount = ntohs(header->nscount);
648 >    header->arcount = ntohs(header->arcount);
649  
650 <  /*
651 <   * response for an id which we have already received an answer for
652 <   * just ignore this response.
653 <   */
654 <  if (!(request = find_id(header->id)))
655 <    return;
650 >    /*
651 >     * Response for an id which we have already received an answer for
652 >     * just ignore this response.
653 >     */
654 >    if ((request = find_id(header->id)) == NULL)
655 >      continue;
656  
657 <  if ((header->rcode != NO_ERRORS) || (header->ancount == 0))
793 <  {
794 <    if (header->rcode == SERVFAIL || header->rcode == NXDOMAIN)
657 >    if ((header->rcode != NO_ERRORS) || (header->ancount == 0))
658      {
659 <      /*
660 <       * If a bad error was returned, stop here and don't
661 <       * send any more (no retries granted).
662 <       */
663 <      (*request->callback)(request->callback_ctx, NULL, NULL);
664 <      rem_request(request);
665 <    }
659 >      if (header->rcode == SERVFAIL || header->rcode == NXDOMAIN)
660 >      {
661 >        /*
662 >         * If a bad error was returned, stop here and don't
663 >         * send any more (no retries granted).
664 >         */
665 >        (*request->callback)(request->callback_ctx, NULL, NULL, 0);
666 >        rem_request(request);
667 >      }
668   #ifdef IPV6
669 <    else
805 <    {
806 <      /*
807 <       * If we havent already tried this, and we're looking up AAAA, try A
808 <       * now
809 <       */
810 <      if (request->state == REQ_AAAA && request->type == T_AAAA)
669 >      else
670        {
671 <        request->timeout += 4;
672 <        resend_query(request);
671 >        /*
672 >         * If we havent already tried this, and we're looking up AAAA, try A now.
673 >         */
674 >        if (request->state == REQ_AAAA && request->type == T_AAAA)
675 >        {
676 >          request->timeout += 4;
677 >          resend_query(request);
678 >        }
679        }
815    }
680   #endif
681 +      continue;
682 +    }
683  
684 <    return;
685 <  }
684 >    /*
685 >     * If this fails there was an error decoding the received packet.
686 >     * We only give it one shot. If it fails, just leave the client
687 >     * unresolved.
688 >     */
689 >    if (!proc_answer(request, header, buf, buf + rc))
690 >    {
691 >      (*request->callback)(request->callback_ctx, NULL, NULL, 0);
692 >      rem_request(request);
693 >      continue;
694 >    }
695  
821  /*
822   * If this fails there was an error decoding the received packet,
823   * try it again and hope it works the next time.
824   */
825  if (proc_answer(request, header, buf, buf + rc))
826  {
696      if (request->type == T_PTR)
697      {
698 <      if (request->name == NULL)
698 >      if (request->namelength == 0)
699        {
700          /*
701 <         * got a PTR response with no name, something bogus is happening
701 >         * Got a PTR response with no name, something bogus is happening
702           * don't bother trying again, the client address doesn't resolve
703           */
704 <        (*request->callback)(request->callback_ctx, NULL, NULL);
704 >        (*request->callback)(request->callback_ctx, NULL, NULL, 0);
705          rem_request(request);
706 <        return;
706 >        continue;
707        }
708  
709        /*
710 <       * Lookup the 'authoritative' name that we were given for the
842 <       * ip#.
843 <       *
710 >       * Lookup the 'authoritative' name that we were given for the ip#.
711         */
712   #ifdef IPV6
713        if (request->addr.ss.ss_family == AF_INET6)
# Line 853 | Line 720 | res_readreply(fde_t *fd, void *data)
720      else
721      {
722        /*
723 <       * got a name and address response, client resolved
723 >       * Got a name and address response, client resolved
724         */
725 <      (*request->callback)(request->callback_ctx, &request->addr, request->name);
725 >      (*request->callback)(request->callback_ctx, &request->addr, request->name, request->namelength);
726        rem_request(request);
727      }
861  }
862  else if (!request->sent)
863  {
864    /* XXX - we got a response for a query we didn't send with a valid id?
865     * this should never happen, bail here and leave the client unresolved
866     */
867    assert(0);
728  
729 <    /* XXX don't leak it */
870 <    rem_request(request);
729 >    continue;
730    }
731 +
732 +  comm_setselect(fd, COMM_SELECT_READ, res_readreply, NULL, 0);
733   }
734  
735   void
736   report_dns_servers(struct Client *source_p)
737   {
738 <  int i;
739 <  char ipaddr[HOSTIPLEN + 1];
738 >  char ipaddr[HOSTIPLEN + 1] = "";
739 >
740 >  for (unsigned int i = 0; i < irc_nscount; ++i)
741 >  {
742 >    getnameinfo((struct sockaddr *)&(irc_nsaddr_list[i]),
743 >                irc_nsaddr_list[i].ss_len, ipaddr,
744 >                sizeof(ipaddr), NULL, 0, NI_NUMERICHOST);
745 >    sendto_one_numeric(source_p, &me, RPL_STATSALINE, ipaddr);
746 >  }
747 > }
748 >
749 > /*
750 > * timeout_query_list - Remove queries from the list which have been
751 > * there too long without being resolved.
752 > */
753 > static time_t
754 > timeout_query_list(void)
755 > {
756 >  dlink_node *ptr = NULL, *ptr_next = NULL;
757 >  struct reslist *request = NULL;
758 >  time_t next_time = 0;
759 >  time_t timeout   = 0;
760  
761 <  for (i = 0; i < irc_nscount; i++)
761 >  DLINK_FOREACH_SAFE(ptr, ptr_next, request_list.head)
762    {
763 <    irc_getnameinfo((struct sockaddr *)&(irc_nsaddr_list[i]),
764 <                    irc_nsaddr_list[i].ss_len, ipaddr,
765 <                    sizeof(ipaddr), NULL, 0, NI_NUMERICHOST);
766 <    sendto_one(source_p, form_str(RPL_STATSALINE),
767 <               me.name, source_p->name, ipaddr);
763 >    request = ptr->data;
764 >    timeout = request->sentat + request->timeout;
765 >
766 >    if (CurrentTime >= timeout)
767 >    {
768 >      if (--request->retries <= 0)
769 >      {
770 >        (*request->callback)(request->callback_ctx, NULL, NULL, 0);
771 >        rem_request(request);
772 >        continue;
773 >      }
774 >      else
775 >      {
776 >        request->sentat = CurrentTime;
777 >        request->timeout += request->timeout;
778 >        resend_query(request);
779 >      }
780 >    }
781 >
782 >    if (next_time == 0 || timeout < next_time)
783 >      next_time = timeout;
784    }
785 +
786 +  return (next_time > CurrentTime) ? next_time : (CurrentTime + AR_TTL);
787 + }
788 +
789 + /*
790 + * timeout_resolver - check request list
791 + */
792 + static void
793 + timeout_resolver(void *notused)
794 + {
795 +  timeout_query_list();
796 + }
797 +
798 + /*
799 + * init_resolver - initialize resolver and resolver library
800 + */
801 + void
802 + init_resolver(void)
803 + {
804 +  static struct event event_timeout_resolver =
805 +  {
806 +    .name = "timeout_resolver",
807 +    .handler = timeout_resolver,
808 +    .when = 1
809 +  };
810 +
811 +  dns_pool = mp_pool_new(sizeof(struct reslist), MP_CHUNK_SIZE_DNS);
812 +  memset(&ResolverFileDescriptor, 0, sizeof(fde_t));
813 +  start_resolver();
814 +
815 +  event_add(&event_timeout_resolver, NULL);
816   }

Diff Legend

Removed lines
+ Added lines
< Changed lines (old)
> Changed lines (new)