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;
69 #define ASSIGN_ID(id) (id++)
73 handle_dns_failure(uint32_t xid
)
75 struct dnsreq
*req
= rb_dictionary_retrieve(query_dict
, RB_UINT_TO_POINTER(xid
));
78 if(req
->callback
== NULL
)
81 req
->callback("FAILED", 0, 0, req
->data
);
87 handle_dns_stat_failure(uint32_t xid
)
89 struct dnsstatreq
*req
= rb_dictionary_retrieve(stat_dict
, RB_UINT_TO_POINTER(xid
));
92 if(req
->callback
== NULL
)
95 req
->callback(1, NULL
, 2, req
->data
);
102 cancel_lookup(uint32_t xid
)
104 struct dnsreq
*req
= rb_dictionary_retrieve(query_dict
, RB_UINT_TO_POINTER(xid
));
106 req
->callback
= NULL
;
111 cancel_dns_stats(uint32_t xid
)
113 struct dnsstatreq
*req
= rb_dictionary_retrieve(stat_dict
, RB_UINT_TO_POINTER(xid
));
115 req
->callback
= NULL
;
121 lookup_hostname(const char *hostname
, int aftype
, DNSCB callback
, void *data
)
123 struct dnsreq
*req
= rb_malloc(sizeof(struct dnsreq
));
125 uint32_t rid
= ASSIGN_ID(query_id
);
129 rb_dictionary_add(query_dict
, RB_UINT_TO_POINTER(rid
), req
);
131 req
->callback
= callback
;
135 if(aftype
== AF_INET6
)
141 submit_dns(rid
, aft
== 4 ? DNS_HOST_IPV4
: DNS_HOST_IPV6
, hostname
);
146 lookup_ip(const char *addr
, int aftype
, DNSCB callback
, void *data
)
148 struct dnsreq
*req
= rb_malloc(sizeof(struct dnsreq
));
150 uint32_t rid
= ASSIGN_ID(query_id
);
154 rb_dictionary_add(query_dict
, RB_UINT_TO_POINTER(rid
), req
);
156 req
->callback
= callback
;
160 if(aftype
== AF_INET6
)
166 submit_dns(rid
, aft
== 4 ? DNS_REVERSE_IPV4
: DNS_REVERSE_IPV6
, addr
);
171 get_nameservers(DNSLISTCB callback
, void *data
)
173 struct dnsstatreq
*req
= rb_malloc(sizeof(struct dnsstatreq
));
174 uint32_t qid
= ASSIGN_ID(stat_id
);
178 rb_dictionary_add(stat_dict
, RB_UINT_TO_POINTER(qid
), req
);
180 req
->callback
= callback
;
183 submit_dns_stat(qid
);
189 dns_results_callback(const char *callid
, const char *status
, const char *type
, const char *results
)
195 long lrid
= strtol(callid
, NULL
, 16);
197 if(lrid
> UINT32_MAX
)
200 rid
= (uint32_t)lrid
;
201 req
= rb_dictionary_retrieve(query_dict
, RB_UINT_TO_POINTER(rid
));
205 st
= (*status
== 'O');
206 aft
= *type
== '6' || *type
== 'S' ? 6 : 4;
207 if(req
->callback
== NULL
)
209 /* got cancelled..oh well */
220 req
->callback(results
, st
, aft
, req
->data
);
223 rb_dictionary_delete(query_dict
, RB_UINT_TO_POINTER(rid
));
227 dns_stats_results_callback(const char *callid
, const char *status
, int resc
, const char *resv
[])
229 struct dnsstatreq
*req
;
232 long lqid
= strtol(callid
, NULL
, 16);
234 if(lqid
> UINT32_MAX
)
237 qid
= (uint32_t)lqid
;
238 req
= rb_dictionary_retrieve(stat_dict
, RB_UINT_TO_POINTER(qid
));
242 if(req
->callback
== NULL
)
258 /* Shouldn't happen... */
263 req
->callback(resc
, resv
, st
, req
->data
);
266 rb_dictionary_delete(stat_dict
, RB_UINT_TO_POINTER(qid
));
270 stats_results_callback(int resc
, const char *resv
[], int status
, void *data
)
274 rb_dlink_node
*n
, *tn
;
276 RB_DLINK_FOREACH_SAFE(n
, tn
, nameservers
.head
)
278 /* Clean up old nameservers */
280 rb_dlinkDestroy(n
, &nameservers
);
283 for(int i
= 0; i
< resc
; i
++)
284 rb_dlinkAddAlloc(rb_strdup(resv
[i
]), &nameservers
);
288 const char *error
= resc
? resv
[resc
] : "Unknown error";
289 iwarn("Error getting DNS servers: %s", error
);
297 query_dict
= rb_dictionary_create("dns queries", rb_uint32cmp
);
298 stat_dict
= rb_dictionary_create("dns stat queries", rb_uint32cmp
);
299 (void)get_nameservers(stats_results_callback
, NULL
);
303 reload_nameservers(void)
306 rb_helper_write(authd_helper
, "R D");
307 (void)get_nameservers(stats_results_callback
, NULL
);
312 submit_dns(uint32_t nid
, char type
, const char *addr
)
314 if(authd_helper
== NULL
)
316 handle_dns_failure(nid
);
319 rb_helper_write(authd_helper
, "D %x %c %s", nid
, type
, addr
);
323 submit_dns_stat(uint32_t nid
)
325 if(authd_helper
== NULL
)
327 handle_dns_stat_failure(nid
);
330 rb_helper_write(authd_helper
, "S %x D", nid
);