ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/gvpe/src/vpn_dns.C
(Generate patch)

Comparing gvpe/src/vpn_dns.C (file contents):
Revision 1.36 by pcg, Thu Mar 17 22:24:31 2005 UTC vs.
Revision 1.50 by root, Sun Mar 6 13:49:50 2011 UTC

1/* 1/*
2 vpn_dns.C -- handle the dns tunnel part of the protocol. 2 vpn_dns.C -- handle the dns tunnel part of the protocol.
3 Copyright (C) 2003-2005 Marc Lehmann <gvpe@schmorp.de> 3 Copyright (C) 2003-2008 Marc Lehmann <gvpe@schmorp.de>
4 4
5 This file is part of GVPE. 5 This file is part of GVPE.
6 6
7 GVPE is free software; you can redistribute it and/or modify 7 GVPE is free software; you can redistribute it and/or modify it
8 it under the terms of the GNU General Public License as published by 8 under the terms of the GNU General Public License as published by the
9 the Free Software Foundation; either version 2 of the License, or 9 Free Software Foundation; either version 3 of the License, or (at your
10 (at your option) any later version. 10 option) any later version.
11 11
12 This program is distributed in the hope that it will be useful, 12 This program is distributed in the hope that it will be useful, but
13 but WITHOUT ANY WARRANTY; without even the implied warranty of 13 WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
15 GNU General Public License for more details. 15 Public License for more details.
16 16
17 You should have received a copy of the GNU General Public License 17 You should have received a copy of the GNU General Public License along
18 along with gvpe; if not, write to the Free Software 18 with this program; if not, see <http://www.gnu.org/licenses/>.
19 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 19
20 Additional permission under GNU GPL version 3 section 7
21
22 If you modify this Program, or any covered work, by linking or
23 combining it with the OpenSSL project's OpenSSL library (or a modified
24 version of that library), containing parts covered by the terms of the
25 OpenSSL or SSLeay licenses, the licensors of this Program grant you
26 additional permission to convey the resulting work. Corresponding
27 Source for a non-source form of such a combination shall include the
28 source code for the parts of OpenSSL used as well as that of the
29 covered work.
20*/ 30*/
31
32// TODO: EDNS0 option to increase dns mtu?
33// TODO: re-write dns packet parsing/creation using a safe mem-buffer
34// to ensure no buffer overflows or similar problems.
21 35
22#include "config.h" 36#include "config.h"
23 37
24#if ENABLE_DNS 38#if ENABLE_DNS
25 39
38#include <unistd.h> 52#include <unistd.h>
39#include <fcntl.h> 53#include <fcntl.h>
40 54
41#include <map> 55#include <map>
42 56
57#include <cstdio> /* bug in libgmp: gmp.h relies on cstdio being included */
43#include <gmp.h> 58#include <gmp.h>
44 59
45#include "netcompat.h" 60#include "netcompat.h"
46 61
47#include "vpn.h" 62#include "vpn.h"
48 63
49#define MAX_POLL_INTERVAL 5. // how often to poll minimally when the server has no data 64#define MAX_POLL_INTERVAL 5. // how often to poll minimally when the server has no data
50#define ACTIVITY_INTERVAL 5. 65#define ACTIVITY_INTERVAL 5.
51 66
52#define TIMEOUT_FACTOR 8.
53
54#define INITIAL_TIMEOUT 0.1 // retry timeouts 67#define INITIAL_TIMEOUT 0.1 // retry timeouts
55#define INITIAL_SYN_TIMEOUT 2. // retry timeout for initial syn 68#define INITIAL_SYN_TIMEOUT 2. // retry timeout for initial syn
56 69
57#define MIN_SEND_INTERVAL 0.001 // wait at least this time between sending requests
58#define MAX_SEND_INTERVAL 2. // optimistic? 70#define MAX_SEND_INTERVAL 2. // optimistic?
59 71
60#define LATENCY_FACTOR 0.5 // RTT * LATENCY_FACTOR == sending rate
61#define MAX_OUTSTANDING 100 // max. outstanding requests
62#define MAX_WINDOW 1000 // max. for MAX_OUTSTANDING, and backlog 72#define MAX_WINDOW 1000 // max. for MAX_OUTSTANDING, and backlog
63#define MAX_BACKLOG (64*1024) // size of gvpe protocol backlog (bytes), must be > MAXSIZE 73#define MAX_BACKLOG (64*1024) // size of gvpe protocol backlog (bytes), must be > MAXSIZE
64 74
65#define MAX_DOMAIN_SIZE 240 // 255 is legal limit, but bind doesn't compress well 75#define MAX_DOMAIN_SIZE 240 // 255 is legal limit, but bind doesn't compress well
66// 240 leaves about 4 bytes of server reply data 76// 240 leaves about 4 bytes of server reply data
73#define MAX_PKT_SIZE 512 83#define MAX_PKT_SIZE 512
74 84
75#define RR_TYPE_A 1 85#define RR_TYPE_A 1
76#define RR_TYPE_NULL 10 86#define RR_TYPE_NULL 10
77#define RR_TYPE_TXT 16 87#define RR_TYPE_TXT 16
88#define RR_TYPE_AAAA 28
78#define RR_TYPE_ANY 255 89#define RR_TYPE_ANY 255
79 90
80#define RR_CLASS_IN 1 91#define RR_CLASS_IN 1
81 92
82#define CMD_IP_1 207 93#define CMD_IP_1 207
106 memset (enc, (char) 0, 256); 117 memset (enc, (char) 0, 256);
107 memset (dec, (char)INVALID, 256); 118 memset (dec, (char)INVALID, 256);
108 119
109 for (size = 0; cmap [size]; size++) 120 for (size = 0; cmap [size]; size++)
110 { 121 {
122 char c = cmap [size];
123
111 enc [size] = cmap [size]; 124 enc [size] = c;
112 dec [(u8)enc [size]] = size; 125 dec [(u8)c] = size;
126
127 // allow lowercase/uppercase aliases if possible
128 if (c >= 'A' && c <= 'Z' && dec [c + ('a' - 'A')] == INVALID) dec [c + ('a' - 'A')] = size;
129 if (c >= 'a' && c <= 'z' && dec [c - ('a' - 'A')] == INVALID) dec [c - ('a' - 'A')] = size;
113 } 130 }
114 131
115 assert (size < 256); 132 assert (size < 256);
116} 133}
117 134
157 enc_len [len] = n; 174 enc_len [len] = n;
158 dec_len [n] = len; 175 dec_len [n] = len;
159 } 176 }
160} 177}
161 178
179unsigned int
162unsigned int basecoder::encode_len (unsigned int len) 180basecoder::encode_len (unsigned int len)
163{ 181{
164 return enc_len [len]; 182 return enc_len [len];
165} 183}
166 184
185unsigned int
167unsigned int basecoder::decode_len (unsigned int len) 186basecoder::decode_len (unsigned int len)
168{ 187{
169 while (len && !dec_len [len]) 188 while (len && !dec_len [len])
170 --len; 189 --len;
171 190
172 return dec_len [len]; 191 return dec_len [len];
173} 192}
174 193
194unsigned int
175unsigned int basecoder::encode (char *dst, u8 *src, unsigned int len) 195basecoder::encode (char *dst, u8 *src, unsigned int len)
176{ 196{
177 if (!len || len > MAX_DEC_LEN) 197 if (!len || len > MAX_DEC_LEN)
178 return 0; 198 return 0;
179 199
180 int elen = encode_len (len); 200 int elen = encode_len (len);
199 *dst++ = cmap.encode [dst_ [i]]; 219 *dst++ = cmap.encode [dst_ [i]];
200 220
201 return elen; 221 return elen;
202} 222}
203 223
224unsigned int
204unsigned int basecoder::decode (u8 *dst, char *src, unsigned int len) 225basecoder::decode (u8 *dst, char *src, unsigned int len)
205{ 226{
206 if (!len || len > MAX_ENC_LEN) 227 if (!len || len > MAX_ENC_LEN)
207 return 0; 228 return 0;
208 229
209 u8 src_ [MAX_ENC_LEN]; 230 u8 src_ [MAX_ENC_LEN];
268 289
269///////////////////////////////////////////////////////////////////////////// 290/////////////////////////////////////////////////////////////////////////////
270 291
271#define HDRSIZE 6 292#define HDRSIZE 6
272 293
294inline void
273inline void encode_header (char *data, int clientid, int seqno, int retry = 0) 295encode_header (char *data, int clientid, int seqno, int retry = 0)
274{ 296{
275 seqno &= SEQNO_MASK; 297 seqno &= SEQNO_MASK;
276 298
277 u8 hdr[3] = { 299 u8 hdr[3] = {
278 clientid, 300 clientid,
283 assert (clientid < 256); 305 assert (clientid < 256);
284 306
285 cdc26.encode (data, hdr, 3); 307 cdc26.encode (data, hdr, 3);
286} 308}
287 309
310inline void
288inline void decode_header (char *data, int &clientid, int &seqno) 311decode_header (char *data, int &clientid, int &seqno)
289{ 312{
290 u8 hdr[3]; 313 u8 hdr[3];
291 314
292 cdc26.decode (hdr, data, HDRSIZE); 315 cdc26.decode (hdr, data, HDRSIZE);
293 316
326byte_stream::~byte_stream () 349byte_stream::~byte_stream ()
327{ 350{
328 delete data; 351 delete data;
329} 352}
330 353
354void
331void byte_stream::remove (int count) 355byte_stream::remove (int count)
332{ 356{
333 if (count > fill) 357 if (count > fill)
334 assert (count <= fill); 358 assert (count <= fill);
335 359
336 memmove (data, data + count, fill -= count); 360 memmove (data, data + count, fill -= count);
337} 361}
338 362
363bool
339bool byte_stream::put (u8 *data, unsigned int datalen) 364byte_stream::put (u8 *data, unsigned int datalen)
340{ 365{
341 if (maxsize - fill < datalen) 366 if (maxsize - fill < datalen)
342 return false; 367 return false;
343 368
344 memcpy (this->data + fill, data, datalen); fill += datalen; 369 memcpy (this->data + fill, data, datalen); fill += datalen;
345 370
346 return true; 371 return true;
347} 372}
348 373
374bool
349bool byte_stream::put (vpn_packet *pkt) 375byte_stream::put (vpn_packet *pkt)
350{ 376{
351 if (maxsize - fill < pkt->len + 2) 377 if (maxsize - fill < pkt->len + 2)
352 return false; 378 return false;
353 379
354 data [fill++] = pkt->len >> 8; 380 data [fill++] = pkt->len >> 8;
436 bool valid (); 462 bool valid ();
437}; 463};
438 464
439int dns_cfg::next_uid; 465int dns_cfg::next_uid;
440 466
467void
441void dns_cfg::reset (int clientid) 468dns_cfg::reset (int clientid)
442{ 469{
443 id1 = 'G'; 470 id1 = 'G';
444 id2 = 'V'; 471 id2 = 'V';
445 id3 = 'P'; 472 id3 = 'P';
446 id4 = 'E'; 473 id4 = 'E';
460 487
461 r3 = r4 = 0; 488 r3 = r4 = 0;
462 r4 = r5 = r6 = r7 = 0; 489 r4 = r5 = r6 = r7 = 0;
463} 490}
464 491
492bool
465bool dns_cfg::valid () 493dns_cfg::valid ()
466{ 494{
495 // although the protocol itself allows for some configurability,
496 // only the following encoding/decoding settings are implemented.
467 return id1 == 'G' 497 return id1 == 'G'
468 && id2 == 'V' 498 && id2 == 'V'
469 && id3 == 'P' 499 && id3 == 'P'
470 && id4 == 'E' 500 && id4 == 'E'
471 && seq_cdc == 26 501 && seq_cdc == 26
483 u8 data [MAXSIZE - 6 * 2]; 513 u8 data [MAXSIZE - 6 * 2];
484 514
485 int decode_label (char *data, int size, int &offs); 515 int decode_label (char *data, int size, int &offs);
486}; 516};
487 517
518int
488int dns_packet::decode_label (char *data, int size, int &offs) 519dns_packet::decode_label (char *data, int size, int &offs)
489{ 520{
490 char *orig = data; 521 char *orig = data;
491 522
492 memset (data, 0, size); 523 memset (data, 0, size);
493 524
519 return data - orig; 550 return data - orig;
520} 551}
521 552
522///////////////////////////////////////////////////////////////////////////// 553/////////////////////////////////////////////////////////////////////////////
523 554
555static
556u16 next_id ()
557{
524static u16 dns_id = 0; // TODO: should be per-vpn 558 static u16 dns_id = 0; // TODO: should be per-vpn
525 559
526static u16 next_id ()
527{
528 if (!dns_id) 560 if (!dns_id)
529 dns_id = time (0); 561 dns_id = time (0);
530 562
531 // the simplest lsfr with periodicity 65535 i could find 563 // the simplest lsfr with periodicity 65535 i could find
532 dns_id = (dns_id << 1) 564 dns_id = (dns_id << 1)
558 vector<dns_rcv *> rcvpq; 590 vector<dns_rcv *> rcvpq;
559 591
560 byte_stream rcvdq; int rcvseq; int repseq; 592 byte_stream rcvdq; int rcvseq; int repseq;
561 byte_stream snddq; int sndseq; 593 byte_stream snddq; int sndseq;
562 594
563 void time_cb (time_watcher &w); time_watcher tw; 595 inline void time_cb (ev::timer &w, int revents); ev::timer tw;
564 void receive_rep (dns_rcv *r); 596 void receive_rep (dns_rcv *r);
565 597
566 dns_connection (connection *c); 598 dns_connection (connection *c);
567 ~dns_connection (); 599 ~dns_connection ();
568}; 600};
587: dns (dns) 619: dns (dns)
588{ 620{
589 timeout = 0; 621 timeout = 0;
590 retry = 0; 622 retry = 0;
591 seqno = 0; 623 seqno = 0;
592 sent = NOW; 624 sent = ev_now ();
593 stdhdr = false; 625 stdhdr = false;
594 626
595 pkt = new dns_packet; 627 pkt = new dns_packet;
596 628
597 pkt->id = next_id (); 629 pkt->id = next_id ();
600dns_snd::~dns_snd () 632dns_snd::~dns_snd ()
601{ 633{
602 delete pkt; 634 delete pkt;
603} 635}
604 636
637static void
605static void append_domain (dns_packet &pkt, int &offs, const char *domain) 638append_domain (dns_packet &pkt, int &offs, const char *domain)
606{ 639{
607 // add tunnel domain 640 // add tunnel domain
608 for (;;) 641 for (;;)
609 { 642 {
610 const char *end = strchr (domain, '.'); 643 const char *end = strchr (domain, '.');
623 656
624 domain = end + 1; 657 domain = end + 1;
625 } 658 }
626} 659}
627 660
661void
628void dns_snd::gen_stream_req (int seqno, byte_stream &stream) 662dns_snd::gen_stream_req (int seqno, byte_stream &stream)
629{ 663{
630 stdhdr = true; 664 stdhdr = true;
631 this->seqno = seqno; 665 this->seqno = seqno;
632 666
633 timeout = NOW + INITIAL_TIMEOUT; 667 timeout = ev_now () + INITIAL_TIMEOUT;
634 668
635 pkt->flags = htons (DEFAULT_CLIENT_FLAGS); 669 pkt->flags = htons (DEFAULT_CLIENT_FLAGS);
636 pkt->qdcount = htons (1); 670 pkt->qdcount = htons (1);
637 671
638 int offs = 6*2; 672 int offs = 6*2;
671 (*pkt)[offs++] = RR_CLASS_IN >> 8; (*pkt)[offs++] = RR_CLASS_IN; 705 (*pkt)[offs++] = RR_CLASS_IN >> 8; (*pkt)[offs++] = RR_CLASS_IN;
672 706
673 pkt->len = offs; 707 pkt->len = offs;
674} 708}
675 709
710void
676void dns_snd::gen_syn_req () 711dns_snd::gen_syn_req ()
677{ 712{
678 timeout = NOW + INITIAL_SYN_TIMEOUT; 713 timeout = ev_now () + INITIAL_SYN_TIMEOUT;
679 714
680 pkt->flags = htons (DEFAULT_CLIENT_FLAGS); 715 pkt->flags = htons (DEFAULT_CLIENT_FLAGS);
681 pkt->qdcount = htons (1); 716 pkt->qdcount = htons (1);
682 717
683 int offs = 6 * 2; 718 int offs = 6 * 2;
723 758
724dns_connection::dns_connection (connection *c) 759dns_connection::dns_connection (connection *c)
725: c (c) 760: c (c)
726, rcvdq (MAX_BACKLOG * 2) 761, rcvdq (MAX_BACKLOG * 2)
727, snddq (MAX_BACKLOG) 762, snddq (MAX_BACKLOG)
728, tw (this, &dns_connection::time_cb)
729{ 763{
764 tw.set<dns_connection, &dns_connection::time_cb> (this);
765
730 vpn = c->vpn; 766 vpn = c->vpn;
731 767
732 established = false; 768 established = false;
733 769
734 rcvseq = repseq = sndseq = 0; 770 rcvseq = repseq = sndseq = 0;
745 i != rcvpq.end (); 781 i != rcvpq.end ();
746 ++i) 782 ++i)
747 delete *i; 783 delete *i;
748} 784}
749 785
786void
750void dns_connection::receive_rep (dns_rcv *r) 787dns_connection::receive_rep (dns_rcv *r)
751{ 788{
752 if (r->datalen) 789 if (r->datalen)
753 { 790 {
754 last_received = NOW; 791 last_received = ev_now ();
755 tw.trigger (); 792 tw ();
756 793
757 poll_interval = send_interval; 794 poll_interval = send_interval;
758 } 795 }
759 else 796 else
760 { 797 {
770 807
771 // find next packet 808 // find next packet
772 for (vector<dns_rcv *>::iterator i = rcvpq.end (); i-- != rcvpq.begin (); ) 809 for (vector<dns_rcv *>::iterator i = rcvpq.end (); i-- != rcvpq.begin (); )
773 if (SEQNO_EQ (rcvseq, (*i)->seqno)) 810 if (SEQNO_EQ (rcvseq, (*i)->seqno))
774 { 811 {
812 //printf ("seqno eq %x %x\n", rcvseq, (*i)->seqno);//D
775 // enter the packet into our input stream 813 // enter the packet into our input stream
776 r = *i; 814 r = *i;
777 815
778 // remove the oldest packet, look forward, as it's oldest first 816 // remove the oldest packet, look forward, as it's oldest first
779 for (vector<dns_rcv *>::iterator j = rcvpq.begin (); j != rcvpq.end (); ++j) 817 for (vector<dns_rcv *>::iterator j = rcvpq.begin (); j != rcvpq.end (); ++j)
780 if (SEQNO_EQ ((*j)->seqno, rcvseq - MAX_WINDOW)) 818 if (SEQNO_EQ ((*j)->seqno, rcvseq - MAX_WINDOW))
781 { 819 {
820 //printf ("seqno RR %x %x\n", (*j)->seqno, rcvseq - MAX_WINDOW);//D
782 delete *j; 821 delete *j;
783 rcvpq.erase (j); 822 rcvpq.erase (j);
784 break; 823 break;
785 } 824 }
786 825
787 rcvseq = (rcvseq + 1) & SEQNO_MASK; 826 rcvseq = (rcvseq + 1) & SEQNO_MASK;
788 827
789 if (!rcvdq.put (r->data, r->datalen)) 828 if (!rcvdq.put (r->data, r->datalen))
790 { 829 {
830 // MUST never overflow, can be caused by data corruption, TODO
791 slog (L_ERR, "DNS: !rcvdq.put (r->data, r->datalen)"); 831 slog (L_CRIT, "DNS: !rcvdq.put (r->data, r->datalen)");
792 abort (); // MUST never overflow, can be caused by data corruption, TODO 832 c->dnsv4_reset_connection ();
833 return;
793 } 834 }
794 835
795 while (vpn_packet *pkt = rcvdq.get ()) 836 while (vpn_packet *pkt = rcvdq.get ())
796 { 837 {
797 sockinfo si; 838 sockinfo si;
798 si.host = htonl (c->conf->id); si.port = 0; si.prot = PROT_DNSv4; 839 si.host = htonl (c->conf->id); si.port = 0; si.prot = PROT_DNSv4;
799 840
800 vpn->recv_vpn_packet (pkt, si); 841 vpn->recv_vpn_packet (pkt, si);
801
802 delete pkt; 842 delete pkt;
803 } 843 }
804 844
805 // check for further packets 845 // check for further packets
806 goto redo; 846 goto redo;
1014 { 1054 {
1015 dns_connection *dns = (*i)->dns; 1055 dns_connection *dns = (*i)->dns;
1016 connection *c = dns->c; 1056 connection *c = dns->c;
1017 int seqno = (*i)->seqno; 1057 int seqno = (*i)->seqno;
1018 u8 data[MAXSIZE], *datap = data; 1058 u8 data[MAXSIZE], *datap = data;
1059 //printf ("rcv pkt %x\n", seqno);//D
1019 1060
1020 if ((*i)->retry) 1061 if ((*i)->retry)
1021 { 1062 {
1022 dns->send_interval *= 1.01; 1063 dns->send_interval *= 1.01;
1023 if (dns->send_interval > MAX_SEND_INTERVAL) 1064 if (dns->send_interval > MAX_SEND_INTERVAL)
1028#if 0 1069#if 0
1029 dns->send_interval *= 0.999; 1070 dns->send_interval *= 0.999;
1030#endif 1071#endif
1031 // the latency surely puts an upper bound on 1072 // the latency surely puts an upper bound on
1032 // the minimum send interval 1073 // the minimum send interval
1033 double latency = NOW - (*i)->sent; 1074 double latency = ev_now () - (*i)->sent;
1034 1075
1035 if (latency < dns->min_latency) 1076 if (latency < dns->min_latency)
1036 dns->min_latency = latency; 1077 dns->min_latency = latency;
1037 1078
1038 if (dns->send_interval > dns->min_latency * LATENCY_FACTOR) 1079 if (dns->send_interval > dns->min_latency * conf.dns_overlap_factor)
1039 dns->send_interval = dns->min_latency * LATENCY_FACTOR; 1080 dns->send_interval = dns->min_latency * conf.dns_overlap_factor;
1040 1081
1041 if (dns->send_interval < MIN_SEND_INTERVAL) 1082 if (dns->send_interval < conf.dns_send_interval)
1042 dns->send_interval = MIN_SEND_INTERVAL; 1083 dns->send_interval = conf.dns_send_interval;
1043 } 1084 }
1044 1085
1045 delete *i; 1086 delete *i;
1046 dns_sndpq.erase (i); 1087 dns_sndpq.erase (i);
1047 1088
1103 1144
1104 if (ip [3] == CMD_IP_RST) 1145 if (ip [3] == CMD_IP_RST)
1105 { 1146 {
1106 slog (L_DEBUG, _("DNS: got tunnel RST request")); 1147 slog (L_DEBUG, _("DNS: got tunnel RST request"));
1107 1148
1108 delete dns; c->dns = 0; 1149 c->dnsv4_reset_connection ();
1109
1110 return;
1111 } 1150 }
1112 else if (ip [3] == CMD_IP_SYN) 1151 else if (ip [3] == CMD_IP_SYN)
1113 { 1152 {
1114 slog (L_DEBUG, _("DNS: got tunnel SYN reply, server likes us.")); 1153 slog (L_DEBUG, _("DNS: got tunnel SYN reply, server likes us."));
1115 dns->established = true; 1154 dns->established = true;
1116 } 1155 }
1117 else if (ip [3] == CMD_IP_REJ) 1156 else if (ip [3] == CMD_IP_REJ)
1118 {
1119 slog (L_DEBUG, _("DNS: got tunnel REJ reply, server does not like us, aborting.")); 1157 slog (L_ERR, _("DNS: got tunnel REJ reply, server does not like us."));
1120 abort ();
1121 }
1122 else 1158 else
1123 slog (L_INFO, _("DNS: got unknown meta command %02x"), ip [3]); 1159 slog (L_INFO, _("DNS: got unknown meta command %02x"), ip [3]);
1124 } 1160 }
1125 else 1161 else
1126 slog (L_INFO, _("DNS: got spurious a record %d.%d.%d.%d"), 1162 slog (L_INFO, _("DNS: got spurious a record %d.%d.%d.%d"),
1152 break; 1188 break;
1153 } 1189 }
1154} 1190}
1155 1191
1156void 1192void
1157vpn::dnsv4_ev (io_watcher &w, short revents) 1193vpn::dnsv4_ev (ev::io &w, int revents)
1158{ 1194{
1159 if (revents & EVENT_READ) 1195 if (revents & EV_READ)
1160 { 1196 {
1161 dns_packet *pkt = new dns_packet; 1197 dns_packet *pkt = new dns_packet;
1162 struct sockaddr_in sa; 1198 struct sockaddr_in sa;
1163 socklen_t sa_len = sizeof (sa); 1199 socklen_t sa_len = sizeof (sa);
1164 1200
1190 1226
1191 if (!c->dns) 1227 if (!c->dns)
1192 c->dns = new dns_connection (c); 1228 c->dns = new dns_connection (c);
1193 1229
1194 if (c->dns->snddq.put (pkt)) 1230 if (c->dns->snddq.put (pkt))
1195 c->dns->tw.trigger (); 1231 c->dns->tw ();
1196 1232
1197 // always return true even if the buffer overflows 1233 // always return true even if the buffer overflows
1198 return true; 1234 return true;
1199} 1235}
1200 1236
1205} 1241}
1206 1242
1207#define NEXT(w) do { if (next > (w)) next = w; } while (0) 1243#define NEXT(w) do { if (next > (w)) next = w; } while (0)
1208 1244
1209void 1245void
1210dns_connection::time_cb (time_watcher &w) 1246dns_connection::time_cb (ev::timer &w, int revents)
1211{ 1247{
1212 // servers have to be polled 1248 // servers have to be polled
1213 if (THISNODE->dns_port) 1249 if (THISNODE->dns_port)
1214 return; 1250 return;
1215 1251
1216 // check for timeouts and (re)transmit 1252 // check for timeouts and (re)transmit
1217 tstamp next = NOW + poll_interval; 1253 tstamp next = ev::now () + poll_interval;
1218 dns_snd *send = 0; 1254 dns_snd *send = 0;
1219 1255
1220 for (vector<dns_snd *>::iterator i = vpn->dns_sndpq.begin (); 1256 for (vector<dns_snd *>::iterator i = vpn->dns_sndpq.begin ();
1221 i != vpn->dns_sndpq.end (); 1257 i != vpn->dns_sndpq.end ();
1222 ++i) 1258 ++i)
1223 { 1259 {
1224 dns_snd *r = *i; 1260 dns_snd *r = *i;
1225 1261
1226 if (r->timeout <= NOW) 1262 if (r->timeout <= ev_now ())
1227 { 1263 {
1228 if (!send) 1264 if (!send)
1229 { 1265 {
1230 send = r; 1266 send = r;
1231 1267
1232 r->retry++; 1268 r->retry++;
1233 r->timeout = NOW + (r->retry * min_latency * TIMEOUT_FACTOR); 1269 r->timeout = ev_now () + (r->retry * min_latency * conf.dns_timeout_factor);
1270 //printf ("RETRY %x (%d, %f)\n", r->seqno, r->retry, r->timeout - ev_now ());//D
1234 1271
1235 // the following code changes the query section a bit, forcing 1272 // the following code changes the query section a bit, forcing
1236 // the forwarder to generate a new request 1273 // the forwarder to generate a new request
1237 if (r->stdhdr) 1274 if (r->stdhdr)
1238 {
1239 //printf ("reencoded header for ID %d retry %d:%d:%d (%p)\n", htons (r->pkt->id), THISNODE->id, r->seqno, r->retry);
1240 //encode_header ((char *)r->pkt->at (6 * 2 + 1), THISNODE->id, r->seqno, r->retry); 1275 encode_header ((char *)r->pkt->at (6 * 2 + 1), THISNODE->id, r->seqno, r->retry);
1241 }
1242 } 1276 }
1243 } 1277 }
1244 else 1278 else
1245 NEXT (r->timeout); 1279 NEXT (r->timeout);
1246 } 1280 }
1257 1291
1258 cfg.reset (THISNODE->id); 1292 cfg.reset (THISNODE->id);
1259 send->gen_syn_req (); 1293 send->gen_syn_req ();
1260 } 1294 }
1261 } 1295 }
1262 else if (vpn->dns_sndpq.size () < MAX_OUTSTANDING 1296 else if (vpn->dns_sndpq.size () < conf.dns_max_outstanding
1263 && !SEQNO_EQ (rcvseq, sndseq - (MAX_WINDOW - 1))) 1297 && !SEQNO_EQ (rcvseq, sndseq - (MAX_WINDOW - 1)))
1264 { 1298 {
1265 if (last_sent + send_interval <= NOW) 1299 if (last_sent + send_interval <= ev_now ())
1266 { 1300 {
1267 //printf ("sending data request etc.\n"); //D 1301 //printf ("sending data request etc.\n"); //D
1268 if (!snddq.empty () || last_received + 1. > NOW) 1302 if (!snddq.empty () || last_received + 1. > ev_now ())
1269 { 1303 {
1270 poll_interval = send_interval; 1304 poll_interval = send_interval;
1271 NEXT (NOW + send_interval); 1305 NEXT (ev_now () + send_interval);
1272 } 1306 }
1273 1307
1274 send = new dns_snd (this); 1308 send = new dns_snd (this);
1275 send->gen_stream_req (sndseq, snddq); 1309 send->gen_stream_req (sndseq, snddq);
1276 send->timeout = NOW + min_latency * TIMEOUT_FACTOR; 1310 send->timeout = ev_now () + min_latency * conf.dns_timeout_factor;
1311 //printf ("SEND %x (%f)\n", send->seqno, send->timeout - ev_now (), min_latency, conf.dns_timeout_factor);//D
1277 1312
1278 sndseq = (sndseq + 1) & SEQNO_MASK; 1313 sndseq = (sndseq + 1) & SEQNO_MASK;
1279 } 1314 }
1280 else 1315 else
1281 NEXT (last_sent + send_interval); 1316 NEXT (last_sent + send_interval);
1285 vpn->dns_sndpq.push_back (send); 1320 vpn->dns_sndpq.push_back (send);
1286 } 1321 }
1287 1322
1288 if (send) 1323 if (send)
1289 { 1324 {
1290 last_sent = NOW; 1325 last_sent = ev_now ();
1291 sendto (vpn->dnsv4_fd, 1326 sendto (vpn->dnsv4_fd,
1292 send->pkt->at (0), send->pkt->len, 0, 1327 send->pkt->at (0), send->pkt->len, 0,
1293 vpn->dns_forwarder.sav4 (), vpn->dns_forwarder.salenv4 ()); 1328 vpn->dns_forwarder.sav4 (), vpn->dns_forwarder.salenv4 ());
1294 } 1329 }
1295 1330
1296 slog (L_NOISE, "DNS: pi %f si %f N %f (%d:%d %d)", 1331 slog (L_NOISE, "DNS: pi %f si %f N %f (%d:%d %d)",
1297 poll_interval, send_interval, next - NOW, 1332 poll_interval, send_interval, next - ev_now (),
1298 vpn->dns_sndpq.size (), snddq.size (), 1333 vpn->dns_sndpq.size (), snddq.size (),
1299 rcvpq.size ()); 1334 rcvpq.size ());
1300 1335
1301 // TODO: no idea when this happens, but when next < NOW, we have a problem 1336 // TODO: no idea when this happens, but when next < ev_now (), we have a problem
1337 // doesn't seem to happen anymore
1302 if (next < NOW + 0.001) 1338 if (next < ev_now () + 0.001)
1303 next = NOW + 0.1; 1339 next = ev_now () + 0.1;
1304 1340
1305 w.start (next); 1341 w.start (next - ev_now ());
1306} 1342}
1307 1343
1308#endif 1344#endif
1309 1345

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines