2 * dns.c: An interface to the resolver module in authd
3 * (based somewhat on ircd-ratbox dns.c)
5 * Copyright (C) 2005 Aaron Sethman <androsyn@ratbox.org>
6 * Copyright (C) 2005-2012 ircd-ratbox development team
7 * Copyright (C) 2016 William Pitcock <nenolod@dereferenced.org>
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301
28 #include "ircd_defs.h"
40 #define DNS_HOST_IPV4 ((char)'4')
41 #define DNS_HOST_IPV6 ((char)'6')
42 #define DNS_REVERSE_IPV4 ((char)'R')
43 #define DNS_REVERSE_IPV6 ((char)'S')
45 static void submit_dns(uint32_t uid
, char type
, const char *addr
);
46 static void submit_dns_stat(uint32_t uid
);
60 /* These serve as a form of sparse array */
61 static rb_dictionary
*query_dict
;
62 static rb_dictionary
*stat_dict
;
64 rb_dlink_list nameservers
;
66 static uint32_t query_id
= 0;
67 static uint32_t stat_id
= 0;
70 static inline uint32_t
71 assign_id(uint32_t *id
)
80 handle_dns_failure(uint32_t xid
)
82 struct dnsreq
*req
= rb_dictionary_retrieve(query_dict
, RB_UINT_TO_POINTER(xid
));
85 if(req
->callback
== NULL
)
88 req
->callback("FAILED", 0, 0, req
->data
);
94 handle_dns_stat_failure(uint32_t xid
)
96 struct dnsstatreq
*req
= rb_dictionary_retrieve(stat_dict
, RB_UINT_TO_POINTER(xid
));
99 if(req
->callback
== NULL
)
102 req
->callback(1, NULL
, 2, req
->data
);
103 req
->callback
= NULL
;
109 cancel_lookup(uint32_t xid
)
111 struct dnsreq
*req
= rb_dictionary_retrieve(query_dict
, RB_UINT_TO_POINTER(xid
));
113 req
->callback
= NULL
;
118 cancel_dns_stats(uint32_t xid
)
120 struct dnsstatreq
*req
= rb_dictionary_retrieve(stat_dict
, RB_UINT_TO_POINTER(xid
));
122 req
->callback
= NULL
;
128 lookup_hostname(const char *hostname
, int aftype
, DNSCB callback
, void *data
)
130 struct dnsreq
*req
= rb_malloc(sizeof(struct dnsreq
));
132 uint32_t rid
= assign_id(&query_id
);
136 rb_dictionary_add(query_dict
, RB_UINT_TO_POINTER(rid
), req
);
138 req
->callback
= callback
;
142 if(aftype
== AF_INET6
)
148 submit_dns(rid
, aft
== 4 ? DNS_HOST_IPV4
: DNS_HOST_IPV6
, hostname
);
153 lookup_ip(const char *addr
, int aftype
, DNSCB callback
, void *data
)
155 struct dnsreq
*req
= rb_malloc(sizeof(struct dnsreq
));
157 uint32_t rid
= assign_id(&query_id
);
161 rb_dictionary_add(query_dict
, RB_UINT_TO_POINTER(rid
), req
);
163 req
->callback
= callback
;
167 if(aftype
== AF_INET6
)
173 submit_dns(rid
, aft
== 4 ? DNS_REVERSE_IPV4
: DNS_REVERSE_IPV6
, addr
);
178 get_nameservers(DNSLISTCB callback
, void *data
)
180 struct dnsstatreq
*req
= rb_malloc(sizeof(struct dnsstatreq
));
181 uint32_t qid
= assign_id(&stat_id
);
185 rb_dictionary_add(stat_dict
, RB_UINT_TO_POINTER(qid
), req
);
187 req
->callback
= callback
;
190 submit_dns_stat(qid
);
196 dns_results_callback(const char *callid
, const char *status
, const char *type
, const char *results
)
202 long lrid
= strtol(callid
, NULL
, 16);
204 if(lrid
> UINT32_MAX
)
207 rid
= (uint32_t)lrid
;
208 req
= rb_dictionary_retrieve(query_dict
, RB_UINT_TO_POINTER(rid
));
212 st
= (*status
== 'O');
213 aft
= *type
== '6' || *type
== 'S' ? 6 : 4;
214 if(req
->callback
== NULL
)
216 /* got cancelled..oh well */
227 req
->callback(results
, st
, aft
, req
->data
);
230 rb_dictionary_delete(query_dict
, RB_UINT_TO_POINTER(rid
));
234 dns_stats_results_callback(const char *callid
, const char *status
, int resc
, const char *resv
[])
236 struct dnsstatreq
*req
;
239 long lqid
= strtol(callid
, NULL
, 16);
241 if(lqid
> UINT32_MAX
)
244 qid
= (uint32_t)lqid
;
245 req
= rb_dictionary_retrieve(stat_dict
, RB_UINT_TO_POINTER(qid
));
249 if(req
->callback
== NULL
)
265 /* Shouldn't happen... */
270 req
->callback(resc
, resv
, st
, req
->data
);
273 rb_dictionary_delete(stat_dict
, RB_UINT_TO_POINTER(qid
));
277 stats_results_callback(int resc
, const char *resv
[], int status
, void *data
)
281 rb_dlink_node
*n
, *tn
;
283 RB_DLINK_FOREACH_SAFE(n
, tn
, nameservers
.head
)
285 /* Clean up old nameservers */
287 rb_dlinkDestroy(n
, &nameservers
);
290 for(int i
= 0; i
< resc
; i
++)
291 rb_dlinkAddAlloc(rb_strdup(resv
[i
]), &nameservers
);
295 const char *error
= resc
? resv
[resc
] : "Unknown error";
296 iwarn("Error getting DNS servers: %s", error
);
304 query_dict
= rb_dictionary_create("dns queries", rb_uint32cmp
);
305 stat_dict
= rb_dictionary_create("dns stat queries", rb_uint32cmp
);
306 (void)get_nameservers(stats_results_callback
, NULL
);
310 reload_nameservers(void)
313 rb_helper_write(authd_helper
, "R D");
314 (void)get_nameservers(stats_results_callback
, NULL
);
319 submit_dns(uint32_t nid
, char type
, const char *addr
)
321 if(authd_helper
== NULL
)
323 handle_dns_failure(nid
);
326 rb_helper_write(authd_helper
, "D %x %c %s", nid
, type
, addr
);
330 submit_dns_stat(uint32_t nid
)
332 if(authd_helper
== NULL
)
334 handle_dns_stat_failure(nid
);
337 rb_helper_write(authd_helper
, "S %x D", nid
);