Eventhandling: fix shutdownhandlers
[citadel.git] / citadel / modules / c-ares-dns / serv_c-ares-dns.c
1 /*
2  * Copyright (c) 1998-2012 by the citadel.org team
3  *
4  *  This program is open source software; you can redistribute it and/or modify
5  *  it under the terms of the GNU General Public License version 3.
6  *  
7  *  
8  *
9  *  This program is distributed in the hope that it will be useful,
10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  *  GNU General Public License for more details.
13  *
14  *  
15  *  
16  *  
17  *
18  *  Inspired by NodeJS.org; thanks for the MX-Parser ;-)
19  */
20
21 #include "sysdep.h"
22 #include <stdlib.h>
23 #include <unistd.h>
24 #include <stdio.h>
25 #include <termios.h>
26 #include <fcntl.h>
27 #include <signal.h>
28 #include <pwd.h>
29 #include <errno.h>
30 #include <sys/types.h>
31 #include <syslog.h>
32
33 #if TIME_WITH_SYS_TIME
34 # include <sys/time.h>
35 # include <time.h>
36 #else
37 # if HAVE_SYS_TIME_H
38 #  include <sys/time.h>
39 # else
40 #  include <time.h>
41 # endif
42 #endif
43 #include <sys/wait.h>
44 #include <ctype.h>
45 #include <string.h>
46 #include <limits.h>
47 #include <sys/socket.h>
48 #include <netinet/in.h>
49 #include <arpa/inet.h>
50
51 #include <libcitadel.h>
52 #include "citadel.h"
53 #include "server.h"
54 #include "citserver.h"
55 #include "support.h"
56
57 #include "ctdl_module.h"
58 #include "event_client.h"
59
60 int DebugCAres = 0;
61
62 extern struct ev_loop *event_base;
63
64 void SockStateCb(void *data, int sock, int read, int write);
65
66
67 static void HostByAddrCb(void *data,
68                          int status,
69                          int timeouts,
70                          struct hostent *hostent)
71 {
72         AsyncIO *IO = data;
73
74         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
75
76         EV_DNS_LOGT_STOP(DNS.timeout);
77         ev_timer_stop (event_base, &IO->DNS.timeout);
78
79         IO->DNS.Query->DNSStatus = status;
80         if  (status != ARES_SUCCESS) {
81                 StrBufPlain(IO->ErrMsg, ares_strerror(status), -1);
82                 return;
83         }
84         IO->DNS.Query->Data = hostent;
85 }
86
87 static void ParseAnswerA(AsyncIO *IO, unsigned char* abuf, int alen)
88 {
89         struct hostent* host = NULL;
90
91         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
92
93         if (IO->DNS.Query->VParsedDNSReply != NULL)
94                 IO->DNS.Query->DNSReplyFree(IO->DNS.Query->VParsedDNSReply);
95         IO->DNS.Query->VParsedDNSReply = NULL;
96
97         IO->DNS.Query->DNSStatus = ares_parse_a_reply(abuf,
98                                                       alen,
99                                                       &host,
100                                                       NULL,
101                                                       NULL);
102         if (IO->DNS.Query->DNSStatus != ARES_SUCCESS) {
103                 if (host != NULL)
104                         ares_free_hostent(host);
105                 StrBufPlain(IO->ErrMsg,
106                             ares_strerror(IO->DNS.Query->DNSStatus), -1);
107                 return;
108         }
109         IO->DNS.Query->VParsedDNSReply = host;
110         IO->DNS.Query->DNSReplyFree = (FreeDNSReply) ares_free_hostent;
111 }
112
113
114 static void ParseAnswerAAAA(AsyncIO *IO, unsigned char* abuf, int alen)
115 {
116         struct hostent* host = NULL;
117
118         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
119
120         if (IO->DNS.Query->VParsedDNSReply != NULL)
121                 IO->DNS.Query->DNSReplyFree(IO->DNS.Query->VParsedDNSReply);
122         IO->DNS.Query->VParsedDNSReply = NULL;
123
124         IO->DNS.Query->DNSStatus = ares_parse_aaaa_reply(abuf,
125                                                          alen,
126                                                          &host,
127                                                          NULL,
128                                                          NULL);
129         if (IO->DNS.Query->DNSStatus != ARES_SUCCESS) {
130                 if (host != NULL)
131                         ares_free_hostent(host);
132                 StrBufPlain(IO->ErrMsg,
133                             ares_strerror(IO->DNS.Query->DNSStatus), -1);
134                 return;
135         }
136         IO->DNS.Query->VParsedDNSReply = host;
137         IO->DNS.Query->DNSReplyFree = (FreeDNSReply) ares_free_hostent;
138 }
139
140
141 static void ParseAnswerCNAME(AsyncIO *IO, unsigned char* abuf, int alen)
142 {
143         struct hostent* host = NULL;
144
145         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
146
147         if (IO->DNS.Query->VParsedDNSReply != NULL)
148                 IO->DNS.Query->DNSReplyFree(IO->DNS.Query->VParsedDNSReply);
149         IO->DNS.Query->VParsedDNSReply = NULL;
150
151         IO->DNS.Query->DNSStatus = ares_parse_a_reply(abuf,
152                                                       alen,
153                                                       &host,
154                                                       NULL,
155                                                       NULL);
156         if (IO->DNS.Query->DNSStatus != ARES_SUCCESS) {
157                 if (host != NULL)
158                         ares_free_hostent(host);
159                 StrBufPlain(IO->ErrMsg,
160                             ares_strerror(IO->DNS.Query->DNSStatus), -1);
161                 return;
162         }
163
164         // a CNAME lookup always returns a single record but
165         IO->DNS.Query->VParsedDNSReply = host;
166         IO->DNS.Query->DNSReplyFree = (FreeDNSReply) ares_free_hostent;
167 }
168
169
170 static void ParseAnswerMX(AsyncIO *IO, unsigned char* abuf, int alen)
171 {
172         struct ares_mx_reply *mx_out = NULL;
173
174         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
175
176         if (IO->DNS.Query->VParsedDNSReply != NULL)
177                 IO->DNS.Query->DNSReplyFree(IO->DNS.Query->VParsedDNSReply);
178         IO->DNS.Query->VParsedDNSReply = NULL;
179
180         IO->DNS.Query->DNSStatus = ares_parse_mx_reply(abuf, alen, &mx_out);
181         if (IO->DNS.Query->DNSStatus != ARES_SUCCESS) {
182                 if (mx_out != NULL)
183                         ares_free_data(mx_out);
184                 StrBufPlain(IO->ErrMsg,
185                             ares_strerror(IO->DNS.Query->DNSStatus), -1);
186                 return;
187         }
188
189         IO->DNS.Query->VParsedDNSReply = mx_out;
190         IO->DNS.Query->DNSReplyFree = (FreeDNSReply) ares_free_data;
191 }
192
193
194 static void ParseAnswerNS(AsyncIO *IO, unsigned char* abuf, int alen)
195 {
196         struct hostent* host = NULL;
197
198         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
199
200         if (IO->DNS.Query->VParsedDNSReply != NULL)
201                 IO->DNS.Query->DNSReplyFree(IO->DNS.Query->VParsedDNSReply);
202         IO->DNS.Query->VParsedDNSReply = NULL;
203
204         IO->DNS.Query->DNSStatus = ares_parse_ns_reply(abuf, alen, &host);
205         if (IO->DNS.Query->DNSStatus != ARES_SUCCESS) {
206                 if (host != NULL)
207                         ares_free_hostent(host);
208                 StrBufPlain(IO->ErrMsg,
209                             ares_strerror(IO->DNS.Query->DNSStatus), -1);
210                 return;
211         }
212         IO->DNS.Query->VParsedDNSReply = host;
213         IO->DNS.Query->DNSReplyFree = (FreeDNSReply) ares_free_hostent;
214 }
215
216
217 static void ParseAnswerSRV(AsyncIO *IO, unsigned char* abuf, int alen)
218 {
219         struct ares_srv_reply *srv_out = NULL;
220
221         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
222
223         if (IO->DNS.Query->VParsedDNSReply != NULL)
224                 IO->DNS.Query->DNSReplyFree(IO->DNS.Query->VParsedDNSReply);
225         IO->DNS.Query->VParsedDNSReply = NULL;
226
227         IO->DNS.Query->DNSStatus = ares_parse_srv_reply(abuf, alen, &srv_out);
228         if (IO->DNS.Query->DNSStatus != ARES_SUCCESS) {
229                 if (srv_out != NULL)
230                         ares_free_data(srv_out);
231                 StrBufPlain(IO->ErrMsg,
232                             ares_strerror(IO->DNS.Query->DNSStatus), -1);
233                 return;
234         }
235
236         IO->DNS.Query->VParsedDNSReply = srv_out;
237         IO->DNS.Query->DNSReplyFree = (FreeDNSReply) ares_free_data;
238 }
239
240
241 static void ParseAnswerTXT(AsyncIO *IO, unsigned char* abuf, int alen)
242 {
243         struct ares_txt_reply *txt_out;
244
245         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
246
247         if (IO->DNS.Query->VParsedDNSReply != NULL)
248                 IO->DNS.Query->DNSReplyFree(IO->DNS.Query->VParsedDNSReply);
249         IO->DNS.Query->VParsedDNSReply = NULL;
250
251         IO->DNS.Query->DNSStatus = ares_parse_txt_reply(abuf, alen, &txt_out);
252         if (IO->DNS.Query->DNSStatus != ARES_SUCCESS) {
253                 if (txt_out != NULL)
254                         ares_free_data(txt_out);
255                 StrBufPlain(IO->ErrMsg,
256                             ares_strerror(IO->DNS.Query->DNSStatus), -1);
257                 return;
258         }
259         IO->DNS.Query->VParsedDNSReply = txt_out;
260         IO->DNS.Query->DNSReplyFree = (FreeDNSReply) ares_free_data;
261 }
262
263 void QueryCb(void *arg,
264              int status,
265              int timeouts,
266              unsigned char* abuf,
267              int alen)
268 {
269         AsyncIO *IO = arg;
270
271         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
272
273         EV_DNS_LOGT_STOP(DNS.timeout);
274         ev_timer_stop (event_base, &IO->DNS.timeout);
275
276         IO->DNS.Query->DNSStatus = status;
277         if (status == ARES_SUCCESS)
278                 IO->DNS.Query->DNS_CB(arg, abuf, alen);
279         else {
280                 EV_DNS_syslog(LOG_DEBUG, "C-ARES: Failed by: %s error %s\n",
281                               __FUNCTION__,
282                               ares_strerror(status));
283                 StrBufPlain(IO->ErrMsg, ares_strerror(status), -1);
284                 IO->DNS.Query->DNSStatus = status;
285         }
286
287         ev_idle_init(&IO->unwind_stack,
288                      IO_postdns_callback);
289         IO->unwind_stack.data = IO;
290         EV_DNS_LOGT_INIT(unwind_stack);
291         EV_DNS_LOGT_START(unwind_stack);
292         ev_idle_start(event_base, &IO->unwind_stack);
293 }
294
295 void QueryCbDone(AsyncIO *IO)
296 {
297         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
298
299         EV_DNS_LOGT_STOP(DNS.timeout);
300         ev_timer_stop (event_base, &IO->DNS.timeout);
301
302         EV_DNS_LOGT_STOP(unwind_stack);
303         ev_idle_stop(event_base, &IO->unwind_stack);
304 }
305
306 void DestructCAres(AsyncIO *IO)
307 {
308         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
309
310         EV_DNS_LOG_STOP(DNS.recv_event);
311         ev_io_stop(event_base, &IO->DNS.recv_event);
312
313         EV_DNS_LOG_STOP(DNS.send_event);
314         ev_io_stop(event_base, &IO->DNS.send_event);
315
316         EV_DNS_LOGT_STOP(DNS.timeout);
317         ev_timer_stop (event_base, &IO->DNS.timeout);
318
319         EV_DNS_LOGT_STOP(unwind_stack);
320         ev_idle_stop(event_base, &IO->unwind_stack);
321         ares_destroy_options(&IO->DNS.Options);
322 }
323
324
325 void InitC_ares_dns(AsyncIO *IO)
326 {
327         int optmask = 0;
328
329         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s %p\n", __FUNCTION__, IO->DNS.Channel);
330
331         if (IO->DNS.Channel == NULL) {
332                 optmask |= ARES_OPT_SOCK_STATE_CB;
333                 IO->DNS.Options.sock_state_cb = SockStateCb;
334                 IO->DNS.Options.sock_state_cb_data = IO;
335                 ares_init_options(&IO->DNS.Channel, &IO->DNS.Options, optmask);
336         }
337         IO->DNS.Query->DNSStatus = 0;
338 }
339
340 static void
341 DNStimeouttrigger_callback(struct ev_loop *loop, ev_timer *watcher, int revents)
342 {
343         AsyncIO *IO = watcher->data;
344         struct timeval tv, MaxTV;
345         struct timeval *NextTV;
346
347         memset(&MaxTV, 0, sizeof(MaxTV));
348         memset(&tv, 0, sizeof(tv));
349         MaxTV.tv_sec = 30;
350         NextTV = ares_timeout(IO->DNS.Channel, &MaxTV, &tv);
351
352         if ((NextTV->tv_sec != MaxTV.tv_sec) ||
353             (NextTV->tv_usec != MaxTV.tv_usec))
354         {
355                 fd_set readers, writers;
356                 EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s Timeout!\n", __FUNCTION__);
357
358                 FD_ZERO(&readers);
359                 FD_ZERO(&writers);
360                 ares_fds(IO->DNS.Channel, &readers, &writers);
361                 ares_process(IO->DNS.Channel, &readers, &writers);
362         }
363 }
364
365 void QueueGetHostByNameDone(void *Ctx,
366                             int status,
367                             int timeouts,
368                             struct hostent *hostent)
369 {
370         AsyncIO *IO = (AsyncIO *) Ctx;
371
372         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
373
374
375         IO->DNS.Query->DNSStatus = status;
376         IO->DNS.Query->VParsedDNSReply = hostent;
377         IO->DNS.Query->DNSReplyFree = (FreeDNSReply) ares_free_hostent;
378
379         EV_DNS_LOGT_STOP(DNS.timeout);
380         ev_timer_stop (event_base, &IO->DNS.timeout);
381
382         ev_idle_init(&IO->unwind_stack,
383                      IO_postdns_callback);
384         IO->unwind_stack.data = IO;
385         EV_DNS_LOGT_INIT(unwind_stack);
386         EV_DNS_LOGT_START(unwind_stack);
387         ev_idle_start(event_base, &IO->unwind_stack);
388
389 }
390
391 void QueueGetHostByName(AsyncIO *IO,
392                         const char *Hostname,
393                         DNSQueryParts *QueryParts,
394                         IO_CallBack PostDNS)
395 {
396
397         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
398         IO->DNS.SourcePort = 0;
399
400         IO->DNS.Query = QueryParts;
401         IO->DNS.Query->PostDNS = PostDNS;
402
403         InitC_ares_dns(IO);
404
405         ev_timer_init(&IO->DNS.timeout, DNStimeouttrigger_callback, 10, 1);
406         EV_DNS_LOGT_INIT(DNS.timeout);
407         IO->DNS.timeout.data = IO;
408         ares_gethostbyname(IO->DNS.Channel,
409                            Hostname,
410                            AF_INET6, /* it falls back to ipv4 in doubt... */
411                            QueueGetHostByNameDone,
412                            IO);
413         EV_DNS_LOGT_START(DNS.timeout);
414         ev_timer_start(event_base, &IO->DNS.timeout);
415
416 }
417
418 int QueueQuery(ns_type Type,
419                const char *name,
420                AsyncIO *IO,
421                DNSQueryParts *QueryParts,
422                IO_CallBack PostDNS)
423 {
424         int length, family;
425         char address_b[sizeof(struct in6_addr)];
426
427         IO->DNS.SourcePort = 0;
428
429         IO->DNS.Query = QueryParts;
430         IO->DNS.Query->PostDNS = PostDNS;
431         IO->DNS.Start = IO->Now;
432
433         InitC_ares_dns(IO);
434
435         ev_timer_init(&IO->DNS.timeout, DNStimeouttrigger_callback, 10, 1);
436         IO->DNS.timeout.data = IO;
437         EV_DNS_LOGT_INIT(DNS.timeout);
438
439         switch(Type) {
440         case ns_t_a:
441                 IO->DNS.Query->DNS_CB = ParseAnswerA;
442                 break;
443
444         case ns_t_aaaa:
445                 IO->DNS.Query->DNS_CB = ParseAnswerAAAA;
446                 break;
447
448         case ns_t_mx:
449                 IO->DNS.Query->DNS_CB = ParseAnswerMX;
450                 break;
451
452         case ns_t_ns:
453                 IO->DNS.Query->DNS_CB = ParseAnswerNS;
454                 break;
455
456         case ns_t_txt:
457                 IO->DNS.Query->DNS_CB = ParseAnswerTXT;
458                 break;
459
460         case ns_t_srv:
461                 IO->DNS.Query->DNS_CB = ParseAnswerSRV;
462                 break;
463
464         case ns_t_cname:
465                 IO->DNS.Query->DNS_CB = ParseAnswerCNAME;
466                 break;
467
468         case ns_t_ptr:
469
470
471                 if (inet_pton(AF_INET, name, &address_b) == 1) {
472                         length = sizeof(struct in_addr);
473                         family = AF_INET;
474                 } else if (inet_pton(AF_INET6, name, &address_b) == 1) {
475                         length = sizeof(struct in6_addr);
476                         family = AF_INET6;
477                 } else {
478                         return -1;
479                 }
480
481                 ares_gethostbyaddr(IO->DNS.Channel,
482                                    address_b,
483                                    length,
484                                    family,
485                                    HostByAddrCb,
486                                    IO);
487                 EV_DNS_LOGT_START(DNS.timeout);
488                 ev_timer_start(event_base, &IO->DNS.timeout);
489
490                 EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s X1\n", __FUNCTION__);
491
492                 return 1;
493
494         default:
495
496                 EV_DNS_syslog(LOG_DEBUG, "C-ARES: %sX2\n", __FUNCTION__);
497                 return 0;
498         }
499         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
500
501         ares_query(IO->DNS.Channel, name, ns_c_in, Type, QueryCb, IO);
502         EV_DNS_LOGT_START(DNS.timeout);
503         ev_timer_start(event_base, &IO->DNS.timeout);
504         return 1;
505 }
506
507
508
509
510
511 /*****************************************************************************
512  *                      libev / c-ares integration                           *
513  *****************************************************************************/
514 static void DNS_send_callback(struct ev_loop *loop, ev_io *watcher, int revents)
515 {
516         AsyncIO *IO = watcher->data;
517
518         IO->Now = ev_now(event_base);
519
520         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
521
522         ares_process_fd(IO->DNS.Channel,
523                         ARES_SOCKET_BAD,
524                         IO->DNS.send_event.fd);
525 }
526 static void DNS_recv_callback(struct ev_loop *loop, ev_io *watcher, int revents)
527 {
528         AsyncIO *IO = watcher->data;
529
530         IO->Now = ev_now(event_base);
531
532         EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s\n", __FUNCTION__);
533
534         ares_process_fd(IO->DNS.Channel,
535                         IO->DNS.recv_event.fd,
536                         ARES_SOCKET_BAD);
537 }
538
539 void SockStateCb(void *data, int sock, int read, int write)
540 {
541         AsyncIO *IO = data;
542 /* already inside of the event queue. */
543         if (DebugCAres)
544         {
545                 struct sockaddr_in sin = {};
546                 socklen_t slen;
547                 slen = sizeof(sin);
548                 if ((IO->DNS.SourcePort == 0) &&
549                     (getsockname(sock, &sin, &slen) == 0))
550                 {
551                         IO->DNS.SourcePort = ntohs(sin.sin_port);
552                 }
553                 EV_DNS_syslog(LOG_DEBUG, "C-ARES: %s %d|%d Sock %d port %hu\n",
554                               __FUNCTION__,
555                               read,
556                               write,
557                               sock,
558                               IO->DNS.SourcePort);
559         }
560
561         IO->Now = ev_now(event_base);
562
563         if (read) {
564                 if ((IO->DNS.recv_event.fd != sock) &&
565                     (IO->DNS.recv_event.fd != 0)) {
566                         EV_DNS_LOG_STOP(DNS.recv_event);
567                         ev_io_stop(event_base, &IO->DNS.recv_event);
568                 }
569                 IO->DNS.recv_event.fd = sock;
570                 ev_io_init(&IO->DNS.recv_event,
571                            DNS_recv_callback,
572                            IO->DNS.recv_event.fd,
573                            EV_READ);
574                 EV_DNS_LOG_INIT(DNS.recv_event);
575                 IO->DNS.recv_event.data = IO;
576                 EV_DNS_LOG_START(DNS.recv_event);
577                 ev_io_start(event_base, &IO->DNS.recv_event);
578         }
579         if (write) {
580                 if ((IO->DNS.send_event.fd != sock) &&
581                     (IO->DNS.send_event.fd != 0)) {
582                         EV_DNS_LOG_STOP(DNS.send_event);
583                         ev_io_stop(event_base, &IO->DNS.send_event);
584                 }
585                 IO->DNS.send_event.fd = sock;
586                 ev_io_init(&IO->DNS.send_event,
587                            DNS_send_callback,
588                            IO->DNS.send_event.fd,
589                            EV_WRITE);
590                 IO->DNS.send_event.data = IO;
591                 EV_DNS_LOG_INIT(DNS.send_event);
592                 EV_DNS_LOG_START(DNS.send_event);
593                 ev_io_start(event_base, &IO->DNS.send_event);
594         }
595         if ((read == 0) && (write == 0)) {
596                 EV_DNS_LOG_STOP(DNS.recv_event);
597                 EV_DNS_LOG_STOP(DNS.send_event);
598                 ev_io_stop(event_base, &IO->DNS.recv_event);
599                 ev_io_stop(event_base, &IO->DNS.send_event);
600         }
601 }
602 void EnableDebugCAres(const int n)
603 {
604         DebugCAres = n;
605 }
606
607 CTDL_MODULE_INIT(c_ares_client)
608 {
609         if (!threading)
610         {
611                 CtdlRegisterDebugFlagHook(HKEY("cares"), EnableDebugCAres, &DebugCAres);
612                 int r = ares_library_init(ARES_LIB_INIT_ALL);
613                 if (0 != r) {
614                         
615                 }
616         }
617         return "c-ares";
618 }