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.21 by pcg, Sun Mar 6 21:33:40 2005 UTC vs.
Revision 1.54 by root, Tue Oct 18 11:07:28 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-2011 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 MIN_POLL_INTERVAL .02 // how often to poll minimally when the server has data 64#define MIN_POLL_INTERVAL 0.025 // poll at most this often when no data received
50#define MAX_POLL_INTERVAL 6. // how often to poll minimally when the server has no data 65#define MAX_POLL_INTERVAL 1. // how often to poll minimally when the server has no data
51#define ACTIVITY_INTERVAL 5.
52 66
53#define INITIAL_TIMEOUT 0.1 // retry timeouts 67#define INITIAL_TIMEOUT 0.1 // retry timeouts
54#define INITIAL_SYN_TIMEOUT 10. // retry timeout for initial syn 68#define INITIAL_SYN_TIMEOUT 2. // retry timeout for initial syn
55 69
56#define MIN_SEND_INTERVAL 0.01 // wait at least this time between sending requests
57#define MAX_SEND_INTERVAL 0.5 // optimistic? 70#define MAX_SEND_INTERVAL 5. // optimistic?
58 71
59#define MAX_OUTSTANDING 10 // max. outstanding requests
60#define MAX_WINDOW 1000 // max. for MAX_OUTSTANDING, and backlog 72#define MAX_WINDOW 1000 // max. for MAX_OUTSTANDING, and backlog
61#define MAX_BACKLOG (100*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
62 74
63#define MAX_DOMAIN_SIZE 220 // 255 is legal limit, but bind doesn't compress well 75#define MAX_DOMAIN_SIZE 235 // 255 is legal limit, but bind doesn't compress well
64// 240 leaves about 4 bytes of server reply data 76// 240 leaves about 4 bytes of server reply data
65// every two request bytes less give room for one reply byte 77// every request byte less give room for two reply bytes
66 78
67#define SEQNO_MASK 0x3fff 79#define SEQNO_MASK 0x3fff
68#define SEQNO_EQ(a,b) ( 0 == ( ((a) ^ (b)) & SEQNO_MASK) ) 80#define SEQNO_EQ(a,b) ( 0 == ( ((a) ^ (b)) & SEQNO_MASK) )
69 81
70#define MAX_LBL_SIZE 63 82#define MAX_LBL_SIZE 63
71#define MAX_PKT_SIZE 512 83#define MAX_PKT_SIZE 512
72 84
73#define RR_TYPE_A 1 85#define RR_TYPE_A 1
74#define RR_TYPE_NULL 10 86#define RR_TYPE_NULL 10
75#define RR_TYPE_TXT 16 87#define RR_TYPE_TXT 16
88#define RR_TYPE_AAAA 28
76#define RR_TYPE_ANY 255 89#define RR_TYPE_ANY 255
77 90
78#define RR_CLASS_IN 1 91#define RR_CLASS_IN 1
79 92
80#define CMD_IP_1 207 93#define CMD_IP_1 207
81#define CMD_IP_2 46 94#define CMD_IP_2 46
82#define CMD_IP_3 236 95#define CMD_IP_3 236
83#define CMD_IP_RST 29 96
84#define CMD_IP_SYN 113 97#define CMD_IP_RST 29 // some error, reset and retry
85#define CMD_IP_REJ 32 98#define CMD_IP_REJ 32 // do not want you
99#define CMD_IP_SYN 113 // connection established
100#define CMD_IP_CSE 213 // connection established, but likely case mismatch
101
102static bool
103is_uc (char c)
104{
105 return 'A' <= c && c <= 'Z';
106}
107
108static bool
109is_lc (char c)
110{
111 return 'a' <= c && c <= 'z';
112}
86 113
87// works for cmaps up to 255 (not 256!) 114// works for cmaps up to 255 (not 256!)
88struct charmap 115struct charmap
89{ 116{
90 enum { INVALID = (u8)255 }; 117 enum { INVALID = (u8)255 };
… …
104 memset (enc, (char) 0, 256); 131 memset (enc, (char) 0, 256);
105 memset (dec, (char)INVALID, 256); 132 memset (dec, (char)INVALID, 256);
106 133
107 for (size = 0; cmap [size]; size++) 134 for (size = 0; cmap [size]; size++)
108 { 135 {
136 char c = cmap [size];
137
109 enc [size] = cmap [size]; 138 enc [size] = c;
110 dec [(u8)enc [size]] = size; 139 dec [(u8)c] = size;
140
141 // allow lowercase/uppercase aliases if possible
142 if (is_uc (c) && dec [c + ('a' - 'A')] == INVALID) dec [c + ('a' - 'A')] = size;
143 if (is_lc (c) && dec [c - ('a' - 'A')] == INVALID) dec [c - ('a' - 'A')] = size;
111 } 144 }
112 145
113 assert (size < 256); 146 assert (size < 256);
114} 147}
115 148
… …
122{ 155{
123 charmap cmap; 156 charmap cmap;
124 unsigned int enc_len [MAX_DEC_LEN]; 157 unsigned int enc_len [MAX_DEC_LEN];
125 unsigned int dec_len [MAX_ENC_LEN]; 158 unsigned int dec_len [MAX_ENC_LEN];
126 159
127 unsigned int encode_len (unsigned int len); 160 unsigned int encode_len (unsigned int len) const;
128 unsigned int decode_len (unsigned int len); 161 unsigned int decode_len (unsigned int len) const;
129 162
130 unsigned int encode (char *dst, u8 *src, unsigned int len); 163 unsigned int encode (char *dst, u8 *src, unsigned int len) const;
131 unsigned int decode (u8 *dst, char *src, unsigned int len); 164 unsigned int decode (u8 *dst, char *src, unsigned int len) const;
132 165
133 basecoder (const char *cmap); 166 basecoder (const char *cmap);
134}; 167};
135 168
136basecoder::basecoder (const char *cmap) 169basecoder::basecoder (const char *cmap)
… …
155 enc_len [len] = n; 188 enc_len [len] = n;
156 dec_len [n] = len; 189 dec_len [n] = len;
157 } 190 }
158} 191}
159 192
193unsigned int
160unsigned int basecoder::encode_len (unsigned int len) 194basecoder::encode_len (unsigned int len) const
161{ 195{
162 return enc_len [len]; 196 return enc_len [len];
163} 197}
164 198
199unsigned int
165unsigned int basecoder::decode_len (unsigned int len) 200basecoder::decode_len (unsigned int len) const
166{ 201{
167 while (len && !dec_len [len]) 202 while (len && !dec_len [len])
168 --len; 203 --len;
169 204
170 return dec_len [len]; 205 return dec_len [len];
171} 206}
172 207
208unsigned int
173unsigned int basecoder::encode (char *dst, u8 *src, unsigned int len) 209basecoder::encode (char *dst, u8 *src, unsigned int len) const
174{ 210{
175 if (!len || len > MAX_DEC_LEN) 211 if (!len || len > MAX_DEC_LEN)
176 return 0; 212 return 0;
177 213
178 int elen = encode_len (len); 214 int elen = encode_len (len);
… …
197 *dst++ = cmap.encode [dst_ [i]]; 233 *dst++ = cmap.encode [dst_ [i]];
198 234
199 return elen; 235 return elen;
200} 236}
201 237
238unsigned int
202unsigned int basecoder::decode (u8 *dst, char *src, unsigned int len) 239basecoder::decode (u8 *dst, char *src, unsigned int len) const
203{ 240{
204 if (!len || len > MAX_ENC_LEN) 241 if (!len || len > MAX_ENC_LEN)
205 return 0; 242 return 0;
206 243
207 u8 src_ [MAX_ENC_LEN]; 244 u8 src_ [MAX_ENC_LEN];
… …
256 } 293 }
257 abort (); 294 abort ();
258} 295}
259#endif 296#endif
260 297
261//static basecoder cdc64 ("_dDpPhHzZrR06QqMmjJkKBb34TtSsvVlL81xXaAeEFf92WwGgYyoO57UucCNniI-");
262//static basecoder cdc63 ("_dDpPhHzZrR06QqMmjJkKBb34TtSsvVlL81xXaAeEFf92WwGgYyoO57UucCNniI");
263static basecoder cdc62 ("dDpPhHzZrR06QqMmjJkKBb34TtSsvVlL81xXaAeEFf92WwGgYyoO57UucCNniI"); 298static basecoder cdc62 ("dDpPhHzZrR06QqMmjJkKBb34TtSsvVlL81xXaAeEFf92WwGgYyoO57UucCNniI"); // a-zA-Z0-9
264//static basecoder cdc36 ("dphzr06qmjkb34tsvl81xaef92wgyo57ucni"); // unused as of yet 299static basecoder cdc36 ("dPhZr06QmJkB34tSvL81xAeF92wGyO57uCnI"); // a-z0-9 for case-changers
265static basecoder cdc26 ("dPhZrQmJkBtSvLxAeFwGyO"); 300static basecoder cdc26 ("dPhZrQmJkBtSvLxAeFwGyOuCnI"); // a-z
266 301
267///////////////////////////////////////////////////////////////////////////// 302/////////////////////////////////////////////////////////////////////////////
268 303
269#define HDRSIZE 6 304#define HDRSIZE 5
270 305
306inline void
271inline void encode_header (char *data, int clientid, int seqno, int retry = 0) 307encode_header (char *data, int clientid, int seqno, int retry = 0)
272{ 308{
309 assert (clientid < 256);
310
273 seqno &= SEQNO_MASK; 311 seqno &= SEQNO_MASK;
274 312
275 u8 hdr[3] = { 313 u8 hdr[3] = {
314 seqno,
315 (seqno >> 8) | (retry << 6),
276 clientid, 316 clientid,
277 (seqno >> 8) | (retry << 6),
278 seqno,
279 }; 317 };
280 318
281 assert (clientid < 256);
282
283 cdc26.encode (data, hdr, 3); 319 cdc36.encode (data, hdr, 3);
284} 320}
285 321
322inline void
286inline void decode_header (char *data, int &clientid, int &seqno) 323decode_header (char *data, int &clientid, int &seqno)
287{ 324{
288 u8 hdr[3]; 325 u8 hdr[3];
289 326
290 cdc26.decode (hdr, data, HDRSIZE); 327 cdc36.decode (hdr, data, HDRSIZE);
291 328
292 clientid = hdr[0]; 329 clientid = hdr[2];
293 seqno = ((hdr[1] << 8) | hdr[2]) & SEQNO_MASK; 330 seqno = ((hdr[1] << 8) | hdr[0]) & SEQNO_MASK;
294} 331}
295 332
296///////////////////////////////////////////////////////////////////////////// 333/////////////////////////////////////////////////////////////////////////////
297 334
298struct byte_stream 335struct byte_stream
… …
324byte_stream::~byte_stream () 361byte_stream::~byte_stream ()
325{ 362{
326 delete data; 363 delete data;
327} 364}
328 365
366void
329void byte_stream::remove (int count) 367byte_stream::remove (int count)
330{ 368{
331 if (count > fill) 369 if (count > fill)
332 assert (count <= fill); 370 assert (count <= fill);
333 371
334 memmove (data, data + count, fill -= count); 372 memmove (data, data + count, fill -= count);
335} 373}
336 374
375bool
337bool byte_stream::put (u8 *data, unsigned int datalen) 376byte_stream::put (u8 *data, unsigned int datalen)
338{ 377{
339 if (maxsize - fill < datalen) 378 if (maxsize - fill < datalen)
340 return false; 379 return false;
341 380
342 memcpy (this->data + fill, data, datalen); fill += datalen; 381 memcpy (this->data + fill, data, datalen); fill += datalen;
343 382
344 return true; 383 return true;
345} 384}
346 385
386bool
347bool byte_stream::put (vpn_packet *pkt) 387byte_stream::put (vpn_packet *pkt)
348{ 388{
349 if (maxsize - fill < pkt->len + 2) 389 if (maxsize - fill < pkt->len + 2)
350 return false; 390 return false;
351 391
352 data [fill++] = pkt->len >> 8; 392 data [fill++] = pkt->len >> 8;
… …
357 return true; 397 return true;
358} 398}
359 399
360vpn_packet *byte_stream::get () 400vpn_packet *byte_stream::get ()
361{ 401{
402 if (fill < 2)
403 return 0;
404
362 unsigned int len; 405 unsigned int len;
363 406
364 for (;;) 407 for (;;)
365 { 408 {
366 len = (data [0] << 8) | data [1]; 409 len = (data [0] << 8) | data [1];
367 410
368 if (len <= MAXSIZE || fill < 2) 411 if (len <= MAXSIZE)
369 break; 412 break;
370 413
371 // TODO: handle this better than skipping, e.g. by reset 414 // TODO: handle this better than skipping, e.g. by reset
372 slog (L_DEBUG, _("DNS: corrupted packet stream skipping a byte...")); 415 slog (L_DEBUG, _("DNS: corrupted packet (%02x %02x) stream skipping a byte..."), data [0], data [1]);
373 remove (1); 416 remove (1);
374 } 417 }
375 418
376 if (fill < len + 2) 419 if (fill < len + 2)
377 return 0; 420 return 0;
… …
408 451
409struct dns_cfg 452struct dns_cfg
410{ 453{
411 static int next_uid; 454 static int next_uid;
412 455
456 u8 chksum;
457 u8 rrtype;
458 u16 uid; // to make request unique
459
460 u8 version;
461 u8 flags;
462 u16 max_size;
463
413 u8 id1, id2, id3, id4; 464 u8 id1, id2, id3, id4;
414 u8 version; 465
415 u8 rrtype; 466 u16 client;
416 u8 flags;
417 u8 def_ttl; 467 u8 def_ttl;
418 u8 rcv_cdc; 468 u8 r0;
419 u8 snd_cdc;
420 u16 max_size;
421 u16 client;
422 u16 uid; // to make request unique
423 469
424 u8 reserved[8]; 470 u8 syn_cdc; // cdc en/decoder for syn (A?) requests
471 u8 hdr_cdc; // cdc en/decoder for regular request headers
472 u8 req_cdc; // cdc en/decoder for regular (ANY?) request data
473 u8 rep_cdc; // cdc en/decoder for regular (TXT) replies, 0 == 8 bit encoding
474
475 u8 r1, r2, r3, r4;
425 476
426 void reset (int clientid); 477 void reset (int clientid);
427 bool valid (); 478 bool valid ();
479 u8 get_chksum ();
428}; 480};
429 481
430int dns_cfg::next_uid; 482int dns_cfg::next_uid;
431 483
484void
432void dns_cfg::reset (int clientid) 485dns_cfg::reset (int clientid)
433{ 486{
487 // this ID must result in some mixed-case characters in cdc26-encoding
434 id1 = 'G'; 488 id1 = 'G';
435 id2 = 'V'; 489 id2 = 'V';
436 id3 = 'P'; 490 id3 = 'P';
437 id4 = 'E'; 491 id4 = 'E';
438 492
439 version = 1; 493 version = 2;
440 494
441 rrtype = RR_TYPE_TXT; 495 rrtype = RR_TYPE_TXT;
442 flags = 0; 496 flags = 0;
443 def_ttl = 1; 497 def_ttl = 0;
444 rcv_cdc = 0;
445 snd_cdc = 62; 498 syn_cdc = 26;
499 hdr_cdc = 36;
500 req_cdc = conf.dns_case_preserving ? 62 : 36;
501 rep_cdc = 0;
446 max_size = ntohs (MAX_PKT_SIZE); 502 max_size = htons (MAX_PKT_SIZE);
447 client = ntohs (clientid); 503 client = htons (clientid);
448 uid = next_uid++; 504 uid = ++next_uid;
449 505
450 memset (reserved, 0, 8); 506 r0 = r1 = r2 = r3 = r4 = 0;
451}
452 507
508 chksum = get_chksum ();
509}
510
511// simple but not trivial chksum
512u8
513dns_cfg::get_chksum ()
514{
515 unsigned int sum = 0xff00; // only 16 bits required
516
517 u8 old_chksum = chksum;
518 chksum = 0;
519
520 for (unsigned int i = 0; i < sizeof (*this); ++i)
521 sum += ((u8 *)this)[i] * (i + 1);
522
523 chksum = old_chksum;
524
525 return sum + (sum >> 8);
526}
527
528bool
453bool dns_cfg::valid () 529dns_cfg::valid ()
454{ 530{
531 // although the protocol itself allows for some configurability,
532 // only the following encoding/decoding settings are implemented.
455 return id1 == 'G' 533 return id1 == 'G'
456 && id2 == 'V' 534 && id2 == 'V'
457 && id3 == 'P' 535 && id3 == 'P'
458 && id4 == 'E' 536 && id4 == 'E'
459 && version == 1 537 && version == 2
460 && flags == 0
461 && rcv_cdc == 0
462 && snd_cdc == 62 538 && syn_cdc == 26
463 && max_size == ntohs (MAX_PKT_SIZE); 539 && hdr_cdc == 36
540 && (req_cdc == 36 || req_cdc == 62)
541 && rep_cdc == 0
542 && chksum == get_chksum ();
464} 543}
465 544
466struct dns_packet : net_packet 545struct dns_packet : net_packet
467{ 546{
468 u16 id; 547 u16 id;
469 u16 flags; // QR:1 Opcode:4 AA:1 TC:1 RD:1 RA:1 Z:3 RCODE:4 548 u16 flags; // QR:1 Opcode:4 AA:1 TC:1 RD:1 RA:1 Z:3 RCODE:4
470 u16 qdcount, ancount, nscount, arcount; 549 u16 qdcount, ancount, nscount, arcount;
471 550
472 u8 data[MAXSIZE - 6 * 2]; 551 u8 data [MAXSIZE - 6 * 2];
473 552
474 int decode_label (char *data, int size, int &offs); 553 int decode_label (char *data, int size, int &offs);
475}; 554};
476 555
556int
477int dns_packet::decode_label (char *data, int size, int &offs) 557dns_packet::decode_label (char *data, int size, int &offs)
478{ 558{
479 char *orig = data; 559 char *orig = data;
480 560
481 memset (data, 0, size); 561 memset (data, 0, size);
482 562
… …
508 return data - orig; 588 return data - orig;
509} 589}
510 590
511///////////////////////////////////////////////////////////////////////////// 591/////////////////////////////////////////////////////////////////////////////
512 592
593static
594u16 next_id ()
595{
513static u16 dns_id = 0; // TODO: should be per-vpn 596 static u16 dns_id = 0; // TODO: should be per-vpn
514 597
515static u16 next_id () 598#if 1
516{
517 if (!dns_id) 599 if (!dns_id)
518 dns_id = time (0); 600 dns_id = time (0);
519 601
520 // the simplest lsfr with periodicity 65535 i could find 602 // the simplest lsfr with periodicity 65535 i could find
521 dns_id = (dns_id << 1) 603 dns_id = (dns_id << 1)
… …
523 ^ (dns_id >> 2) 605 ^ (dns_id >> 2)
524 ^ (dns_id >> 4) 606 ^ (dns_id >> 4)
525 ^ (dns_id >> 15)) & 1); 607 ^ (dns_id >> 15)) & 1);
526 608
527 return dns_id; 609 return dns_id;
610#else
611 dns_id++;//D
612
613 return htons (dns_id);
614#endif
528} 615}
529 616
530struct dns_rcv; 617struct dns_rcv;
531struct dns_snd; 618struct dns_snd;
532 619
… …
536 struct vpn *vpn; 623 struct vpn *vpn;
537 624
538 dns_cfg cfg; 625 dns_cfg cfg;
539 626
540 bool established; 627 bool established;
628 const basecoder *cdc;
541 629
542 tstamp last_received; 630 tstamp last_received;
543 tstamp last_sent; 631 tstamp last_sent;
544 double last_latency; 632 double min_latency;
545 double poll_interval, send_interval; 633 double poll_interval, send_interval;
546 634
547 vector<dns_rcv *> rcvpq; 635 vector<dns_rcv *> rcvpq;
548 636
549 byte_stream rcvdq; int rcvseq; 637 byte_stream rcvdq; int rcvseq; int repseq;
550 byte_stream snddq; int sndseq; 638 byte_stream snddq; int sndseq;
551 639
552 void time_cb (time_watcher &w); time_watcher tw; 640 inline void time_cb (ev::timer &w, int revents); ev::timer tw;
553 void receive_rep (dns_rcv *r); 641 void receive_rep (dns_rcv *r);
642
643 void reset (); // quite like tcp RST
644 void set_cfg (); // to be called after any cfg changes
554 645
555 dns_connection (connection *c); 646 dns_connection (connection *c);
556 ~dns_connection (); 647 ~dns_connection ();
557}; 648};
558 649
… …
576: dns (dns) 667: dns (dns)
577{ 668{
578 timeout = 0; 669 timeout = 0;
579 retry = 0; 670 retry = 0;
580 seqno = 0; 671 seqno = 0;
581 sent = NOW; 672 sent = ev_now ();
582 stdhdr = false; 673 stdhdr = false;
583 674
584 pkt = new dns_packet; 675 pkt = new dns_packet;
585 676
586 pkt->id = next_id (); 677 pkt->id = next_id ();
… …
589dns_snd::~dns_snd () 680dns_snd::~dns_snd ()
590{ 681{
591 delete pkt; 682 delete pkt;
592} 683}
593 684
685static void
594static void append_domain (dns_packet &pkt, int &offs, const char *domain) 686append_domain (dns_packet &pkt, int &offs, const char *domain)
595{ 687{
596 // add tunnel domain 688 // add tunnel domain
597 for (;;) 689 for (;;)
598 { 690 {
599 const char *end = strchr (domain, '.'); 691 const char *end = strchr (domain, '.');
… …
612 704
613 domain = end + 1; 705 domain = end + 1;
614 } 706 }
615} 707}
616 708
709void
617void dns_snd::gen_stream_req (int seqno, byte_stream &stream) 710dns_snd::gen_stream_req (int seqno, byte_stream &stream)
618{ 711{
619 stdhdr = true; 712 stdhdr = true;
620 this->seqno = seqno; 713 this->seqno = seqno;
621 714
622 timeout = NOW + INITIAL_TIMEOUT; 715 timeout = ev_now () + INITIAL_TIMEOUT;
623 716
624 pkt->flags = htons (DEFAULT_CLIENT_FLAGS); 717 pkt->flags = htons (DEFAULT_CLIENT_FLAGS);
625 pkt->qdcount = htons (1); 718 pkt->qdcount = htons (1);
626 719
627 int offs = 6*2; 720 int offs = 6*2;
… …
630 // so we need to have space for 2*MAX_DOMAIN_SIZE + header + extra 723 // so we need to have space for 2*MAX_DOMAIN_SIZE + header + extra
631 724
632 char enc[256], *encp = enc; 725 char enc[256], *encp = enc;
633 encode_header (enc, THISNODE->id, seqno); 726 encode_header (enc, THISNODE->id, seqno);
634 727
635 int datalen = cdc62.decode_len (dlen - (dlen + MAX_LBL_SIZE - 1) / MAX_LBL_SIZE - HDRSIZE); 728 int datalen = dns->cdc->decode_len (dlen - (dlen + MAX_LBL_SIZE - 1) / MAX_LBL_SIZE - HDRSIZE);
636 729
637 if (datalen > stream.size ()) 730 if (datalen > stream.size ())
638 datalen = stream.size (); 731 datalen = stream.size ();
639 732
640 int enclen = cdc62.encode (enc + HDRSIZE, stream.begin (), datalen) + HDRSIZE; 733 int enclen = dns->cdc->encode (enc + HDRSIZE, stream.begin (), datalen) + HDRSIZE;
641 stream.remove (datalen); 734 stream.remove (datalen);
642 735
643 while (enclen) 736 while (enclen)
644 { 737 {
645 int lbllen = enclen < MAX_LBL_SIZE ? enclen : MAX_LBL_SIZE; 738 int lbllen = enclen < MAX_LBL_SIZE ? enclen : MAX_LBL_SIZE;
… …
660 (*pkt)[offs++] = RR_CLASS_IN >> 8; (*pkt)[offs++] = RR_CLASS_IN; 753 (*pkt)[offs++] = RR_CLASS_IN >> 8; (*pkt)[offs++] = RR_CLASS_IN;
661 754
662 pkt->len = offs; 755 pkt->len = offs;
663} 756}
664 757
758void
665void dns_snd::gen_syn_req () 759dns_snd::gen_syn_req ()
666{ 760{
667 timeout = NOW + INITIAL_SYN_TIMEOUT; 761 timeout = ev_now () + INITIAL_SYN_TIMEOUT;
668
669 printf ("send syn\n");//D
670 762
671 pkt->flags = htons (DEFAULT_CLIENT_FLAGS); 763 pkt->flags = htons (DEFAULT_CLIENT_FLAGS);
672 pkt->qdcount = htons (1); 764 pkt->qdcount = htons (1);
673 765
674 int offs = 6 * 2; 766 int offs = 6 * 2;
… …
713///////////////////////////////////////////////////////////////////////////// 805/////////////////////////////////////////////////////////////////////////////
714 806
715dns_connection::dns_connection (connection *c) 807dns_connection::dns_connection (connection *c)
716: c (c) 808: c (c)
717, rcvdq (MAX_BACKLOG * 2) 809, rcvdq (MAX_BACKLOG * 2)
718, snddq (MAX_BACKLOG * 2) 810, snddq (MAX_BACKLOG)
719, tw (this, &dns_connection::time_cb)
720{ 811{
812 tw.set<dns_connection, &dns_connection::time_cb> (this);
813
721 vpn = c->vpn; 814 vpn = c->vpn;
722 815
816 reset ();
817}
818
819dns_connection::~dns_connection ()
820{
821 reset ();
822}
823
824void
825dns_connection::reset ()
826{
827 while (!rcvpq.empty ())
828 {
829 delete rcvpq.back ();
830 rcvpq.pop_back ();
831 }
832
833 for (int i = vpn->dns_sndpq.size (); i--; )
834 if (vpn->dns_sndpq [i]->dns == this)
835 {
836 vpn->dns_sndpq [i] = vpn->dns_sndpq.back ();
837 vpn->dns_sndpq.pop_back ();
838 }
839
723 established = false; 840 established = false;
724 841
725 rcvseq = sndseq = 0; 842 rcvseq = repseq = sndseq = 0;
726 843
727 last_sent = last_received = 0; 844 last_sent = 0;
728 poll_interval = MIN_POLL_INTERVAL; 845 poll_interval = 0.5; // starting here
729 send_interval = 0.5; // starting rate 846 send_interval = 0.5; // starting rate
730 last_latency = INITIAL_TIMEOUT; 847 min_latency = INITIAL_TIMEOUT;
731} 848}
732 849
733dns_connection::~dns_connection () 850void
851dns_connection::set_cfg ()
734{ 852{
735 for (vector<dns_rcv *>::iterator i = rcvpq.begin (); 853 cdc = cfg.req_cdc == 36 ? &cdc36 : &cdc62;
736 i != rcvpq.end ();
737 ++i)
738 delete *i;
739} 854}
740 855
856void
741void dns_connection::receive_rep (dns_rcv *r) 857dns_connection::receive_rep (dns_rcv *r)
742{ 858{
743 if (r->datalen) 859 if (r->datalen)
744 { 860 poll_interval = max (poll_interval * (1. / 1.2), MIN_POLL_INTERVAL);
745 last_received = NOW;
746 tw.trigger ();
747
748 poll_interval = send_interval;
749 }
750 else 861 else
751 { 862 poll_interval = min (poll_interval * 1.1, MAX_POLL_INTERVAL);
752 poll_interval *= 1.5;
753 if (poll_interval > MAX_POLL_INTERVAL)
754 poll_interval = MAX_POLL_INTERVAL;
755 }
756 863
757 rcvpq.push_back (r); 864 rcvpq.push_back (r);
758 865
759 redo: 866 redo:
760 867
761 // find next packet 868 // find next packet
762 for (vector<dns_rcv *>::iterator i = rcvpq.end (); i-- != rcvpq.begin (); ) 869 for (vector<dns_rcv *>::iterator i = rcvpq.end (); i-- != rcvpq.begin (); )
763 if (SEQNO_EQ (rcvseq, (*i)->seqno)) 870 if (SEQNO_EQ (rcvseq, (*i)->seqno))
764 { 871 {
872 //printf ("seqno eq %x %x\n", rcvseq, (*i)->seqno);//D
765 // enter the packet into our input stream 873 // enter the packet into our input stream
766 r = *i; 874 r = *i;
767 875
768 // remove the oldest packet, look forward, as it's oldest first 876 // remove the oldest packet, look forward, as it's oldest first
769 for (vector<dns_rcv *>::iterator j = rcvpq.begin (); j != rcvpq.end (); ++j) 877 for (vector<dns_rcv *>::iterator j = rcvpq.begin (); j != rcvpq.end (); ++j)
770 if (SEQNO_EQ ((*j)->seqno, rcvseq - MAX_WINDOW)) 878 if (SEQNO_EQ ((*j)->seqno, rcvseq - MAX_WINDOW))
771 { 879 {
880 //printf ("seqno RR %x %x\n", (*j)->seqno, rcvseq - MAX_WINDOW);//D
772 delete *j; 881 delete *j;
773 rcvpq.erase (j); 882 rcvpq.erase (j);
774 break; 883 break;
775 } 884 }
776 885
777 rcvseq = (rcvseq + 1) & SEQNO_MASK; 886 rcvseq = (rcvseq + 1) & SEQNO_MASK;
778 887
779 if (!rcvdq.put (r->data, r->datalen)) 888 if (!rcvdq.put (r->data, r->datalen))
780 { 889 {
890 // MUST never overflow, can be caused by data corruption, TODO
781 slog (L_ERR, "DNS: !rcvdq.put (r->data, r->datalen)"); 891 slog (L_CRIT, "DNS: !rcvdq.put (r->data, r->datalen)");
782 abort (); // MUST never overflow, can be caused by data corruption, TODO 892 reset ();
893 return;
783 } 894 }
784 895
785 while (vpn_packet *pkt = rcvdq.get ()) 896 while (vpn_packet *pkt = rcvdq.get ())
786 { 897 {
787 sockinfo si; 898 sockinfo si;
788 si.host = 0x01010101; si.port = htons (c->conf->id); si.prot = PROT_DNSv4; 899 si.host = htonl (c->conf->id); si.port = 0; si.prot = PROT_DNSv4;
789 900
790 vpn->recv_vpn_packet (pkt, si); 901 vpn->recv_vpn_packet (pkt, si);
791
792 delete pkt; 902 delete pkt;
793 } 903 }
794 904
795 // check for further packets 905 // check for further packets
796 goto redo; 906 goto redo;
… …
807 pkt.flags = htons (DEFAULT_SERVER_FLAGS | FLAG_RCODE_FORMERR); 917 pkt.flags = htons (DEFAULT_SERVER_FLAGS | FLAG_RCODE_FORMERR);
808 918
809 if (0 == (flags & (FLAG_RESPONSE | FLAG_OP_MASK)) 919 if (0 == (flags & (FLAG_RESPONSE | FLAG_OP_MASK))
810 && pkt.qdcount == htons (1)) 920 && pkt.qdcount == htons (1))
811 { 921 {
812 char qname[MAXSIZE]; 922 char qname [MAXSIZE];
813 int qlen = pkt.decode_label ((char *)qname, MAXSIZE - offs, offs); 923 int qlen = pkt.decode_label ((char *)qname, MAXSIZE - offs, offs);
814 924
815 u16 qtype = pkt [offs++] << 8; qtype |= pkt [offs++]; 925 u16 qtype = pkt [offs++] << 8; qtype |= pkt [offs++];
816 u16 qclass = pkt [offs++] << 8; qclass |= pkt [offs++]; 926 u16 qclass = pkt [offs++] << 8; qclass |= pkt [offs++];
817 927
… …
824 934
825 int dlen = strlen (THISNODE->domain); 935 int dlen = strlen (THISNODE->domain);
826 936
827 if (qclass == RR_CLASS_IN 937 if (qclass == RR_CLASS_IN
828 && qlen > dlen + 1 938 && qlen > dlen + 1
829 && !memcmp (qname + qlen - dlen - 1, THISNODE->domain, dlen)) 939 && !memcmp (qname + qlen - (dlen + 1), THISNODE->domain, dlen))
830 { 940 {
831 // now generate reply 941 // now generate reply
832 pkt.ancount = htons (1); // one answer RR 942 pkt.ancount = htons (1); // one answer RR
833 pkt.flags = htons (DEFAULT_SERVER_FLAGS | FLAG_RCODE_OK); 943 pkt.flags = htons (DEFAULT_SERVER_FLAGS | FLAG_RCODE_OK);
834 944
… …
839 { 949 {
840 // correct class, domain: parse 950 // correct class, domain: parse
841 int client, seqno; 951 int client, seqno;
842 decode_header (qname, client, seqno); 952 decode_header (qname, client, seqno);
843 953
844 u8 data[MAXSIZE];
845 int datalen = cdc62.decode (data, qname + HDRSIZE, qlen - (dlen + 1 + HDRSIZE));
846
847 if (0 < client && client <= conns.size ()) 954 if (0 < client && client <= conns.size ())
848 { 955 {
849 connection *c = conns [client - 1]; 956 connection *c = conns [client - 1];
850 dns_connection *dns = c->dns; 957 dns_connection *dns = c->dns;
851 dns_rcv *rcv; 958 dns_rcv *rcv;
852 bool in_seq;
853 959
854 if (dns) 960 if (dns)
855 { 961 {
962 u8 data[MAXSIZE];
963 int datalen = dns->cdc->decode (data, qname + HDRSIZE, qlen - (dlen + 1 + HDRSIZE));
964
856 for (vector<dns_rcv *>::iterator i = dns->rcvpq.end (); i-- != dns->rcvpq.begin (); ) 965 for (vector<dns_rcv *>::iterator i = dns->rcvpq.end (); i-- != dns->rcvpq.begin (); )
857 if (SEQNO_EQ ((*i)->seqno, seqno)) 966 if (SEQNO_EQ ((*i)->seqno, seqno))
858 { 967 {
859 // already seen that request: simply reply with the cached reply 968 // already seen that request: simply reply with the cached reply
860 dns_rcv *r = *i; 969 dns_rcv *r = *i;
… …
868 memcpy (pkt.at (0), r->pkt->at (0), offs = r->pkt->len); 977 memcpy (pkt.at (0), r->pkt->at (0), offs = r->pkt->len);
869 978
870 goto duplicate_request; 979 goto duplicate_request;
871 } 980 }
872 981
873 in_seq = dns->rcvseq == seqno;
874
875 // new packet, queue 982 // new packet, queue
876 rcv = new dns_rcv (seqno, data, datalen); 983 rcv = new dns_rcv (seqno, data, datalen);
877 dns->receive_rep (rcv); 984 dns->receive_rep (rcv);
878 } 985 }
879 986
… …
886 pkt [offs++] = 0; pkt [offs++] = 0; 993 pkt [offs++] = 0; pkt [offs++] = 0;
887 pkt [offs++] = 0; pkt [offs++] = dns ? dns->cfg.def_ttl : 0; // TTL 994 pkt [offs++] = 0; pkt [offs++] = dns ? dns->cfg.def_ttl : 0; // TTL
888 995
889 int rdlen_offs = offs += 2; 996 int rdlen_offs = offs += 2;
890 997
891 int dlen = (dns ? ntohs (dns->cfg.max_size) : MAX_PKT_SIZE) - offs;
892 // bind doesn't compress well, so reduce further by one label length
893 dlen -= qlen;
894
895 if (dns) 998 if (dns)
896 { 999 {
1000 int dlen = ntohs (dns->cfg.max_size) - offs;
1001
1002 // bind doesn't compress well, so reduce further by one label length
1003 dlen -= qlen;
1004
897 // only put data into in-order sequence packets, if 1005 // only put data into in-order sequence packets, if
898 // we receive out-of-order packets we generate empty 1006 // we receive out-of-order packets we generate empty
899 // replies 1007 // replies
900 while (dlen > 1 && !dns->snddq.empty () && in_seq) 1008 //printf ("%d - %d & %x (=%d) < %d\n", seqno, dns->repseq, SEQNO_MASK, (seqno - dns->repseq) & SEQNO_MASK, MAX_WINDOW);//D
1009 if (((seqno - dns->repseq) & SEQNO_MASK) <= MAX_WINDOW)
901 { 1010 {
1011 dns->repseq = seqno;
1012
1013 while (dlen > 1 && !dns->snddq.empty ())
1014 {
902 int txtlen = dlen <= 255 ? dlen - 1 : 255; 1015 int txtlen = dlen <= 255 ? dlen - 1 : 255;
903 1016
904 if (txtlen > dns->snddq.size ()) 1017 if (txtlen > dns->snddq.size ())
905 txtlen = dns->snddq.size (); 1018 txtlen = dns->snddq.size ();
906 1019
907 pkt[offs++] = txtlen; 1020 pkt[offs++] = txtlen;
908 memcpy (pkt.at (offs), dns->snddq.begin (), txtlen); 1021 memcpy (pkt.at (offs), dns->snddq.begin (), txtlen);
909 offs += txtlen; 1022 offs += txtlen;
910 dns->snddq.remove (txtlen); 1023 dns->snddq.remove (txtlen);
911 1024
912 dlen -= txtlen + 1; 1025 dlen -= txtlen + 1;
1026 }
913 } 1027 }
914 1028
915 // avoid empty TXT rdata 1029 // avoid completely empty TXT rdata
916 if (offs == rdlen_offs) 1030 if (offs == rdlen_offs)
917 pkt[offs++] = 0; 1031 pkt[offs++] = 0;
918 1032
919 slog (L_NOISE, "DNS: snddq %d", dns->snddq.size ()); 1033 slog (L_NOISE, "DNS: snddq %d", dns->snddq.size ());
920 } 1034 }
… …
956 pkt [offs++] = RR_CLASS_IN >> 8; pkt [offs++] = RR_CLASS_IN; // class 1070 pkt [offs++] = RR_CLASS_IN >> 8; pkt [offs++] = RR_CLASS_IN; // class
957 pkt [offs++] = 0; pkt [offs++] = 0; 1071 pkt [offs++] = 0; pkt [offs++] = 0;
958 pkt [offs++] = 0; pkt [offs++] = cfg.def_ttl; // TTL 1072 pkt [offs++] = 0; pkt [offs++] = cfg.def_ttl; // TTL
959 pkt [offs++] = 0; pkt [offs++] = 4; // rdlength 1073 pkt [offs++] = 0; pkt [offs++] = 4; // rdlength
960 1074
961 slog (L_INFO, _("DNS: client %d tries to connect"), client);
962
963 pkt [offs++] = CMD_IP_1; pkt [offs++] = CMD_IP_2; pkt [offs++] = CMD_IP_3; 1075 pkt [offs++] = CMD_IP_1; pkt [offs++] = CMD_IP_2; pkt [offs++] = CMD_IP_3;
964 pkt [offs++] = CMD_IP_REJ; 1076 pkt [offs++] = CMD_IP_REJ;
965 1077
966 if (0 < client && client <= conns.size ()) 1078 if (0 < client && client <= conns.size ())
967 { 1079 {
968 connection *c = conns [client - 1]; 1080 connection *c = conns [client - 1];
969 1081
970 if (cfg.valid ()) 1082 if (cfg.valid ())
971 { 1083 {
972 pkt [offs - 1] = CMD_IP_SYN; 1084 slog (L_INFO, _("DNS: client %d connects (version %d, req_cdc %d)"), client, cfg.version, cfg.req_cdc);
1085
1086 // check for any encoding mismatches - hints at a case problem
1087 char qname2 [MAX_ENC_LEN];
1088 cdc26.encode (qname2, (u8 *)&cfg, sizeof (dns_cfg));
973 1089
974 delete c->dns; 1090 delete c->dns;
1091
1092 pkt [offs - 1] = memcmp (qname, qname2, cdc26.encode_len (sizeof (dns_cfg)))
1093 ? CMD_IP_CSE : CMD_IP_SYN;
1094
975 c->dns = new dns_connection (c); 1095 c->dns = new dns_connection (c);
976 c->dns->cfg = cfg; 1096 c->dns->cfg = cfg;
1097 c->dns->set_cfg ();
977 } 1098 }
978 } 1099 }
979 } 1100 }
980 } 1101 }
981 1102
… …
1000 { 1121 {
1001 dns_connection *dns = (*i)->dns; 1122 dns_connection *dns = (*i)->dns;
1002 connection *c = dns->c; 1123 connection *c = dns->c;
1003 int seqno = (*i)->seqno; 1124 int seqno = (*i)->seqno;
1004 u8 data[MAXSIZE], *datap = data; 1125 u8 data[MAXSIZE], *datap = data;
1126 //printf ("rcv pkt %x\n", seqno);//D
1005 1127
1006 if ((*i)->retry) 1128 if ((*i)->retry)
1007 { 1129 {
1008 dns->send_interval *= 1.01; 1130 dns->send_interval *= 1.01;
1009 if (dns->send_interval > MAX_SEND_INTERVAL) 1131 if (dns->send_interval > MAX_SEND_INTERVAL)
1010 dns->send_interval = MAX_SEND_INTERVAL; 1132 dns->send_interval = MAX_SEND_INTERVAL;
1011 } 1133 }
1012 else 1134 else
1013 { 1135 {
1014#if 1 1136#if 0
1015 dns->send_interval *= 0.999; 1137 dns->send_interval *= 0.999;
1016#endif 1138#endif
1017 if (dns->send_interval < MIN_SEND_INTERVAL)
1018 dns->send_interval = MIN_SEND_INTERVAL;
1019
1020 // the latency surely puts an upper bound on 1139 // the latency surely puts an upper bound on
1021 // the minimum send interval 1140 // the minimum send interval
1022 double latency = NOW - (*i)->sent; 1141 double latency = ev_now () - (*i)->sent;
1142
1143 if (latency < dns->min_latency)
1023 dns->last_latency = latency; 1144 dns->min_latency = latency;
1024 1145
1025 if (dns->send_interval > latency) 1146 if (dns->send_interval > dns->min_latency * conf.dns_overlap_factor)
1147 dns->send_interval = dns->min_latency * conf.dns_overlap_factor;
1148
1149 if (dns->send_interval < conf.dns_send_interval)
1026 dns->send_interval = latency; 1150 dns->send_interval = conf.dns_send_interval;
1027 } 1151 }
1028 1152
1029 delete *i; 1153 delete *i;
1030 dns_sndpq.erase (i); 1154 dns_sndpq.erase (i);
1031 1155
… …
1049 ttl |= pkt [offs++] << 16; 1173 ttl |= pkt [offs++] << 16;
1050 ttl |= pkt [offs++] << 8; 1174 ttl |= pkt [offs++] << 8;
1051 ttl |= pkt [offs++]; 1175 ttl |= pkt [offs++];
1052 u16 rdlen = pkt [offs++] << 8; rdlen |= pkt [offs++]; 1176 u16 rdlen = pkt [offs++] << 8; rdlen |= pkt [offs++];
1053 1177
1054 if (qtype == RR_TYPE_NULL || qtype == RR_TYPE_TXT) 1178 if (qtype == RR_TYPE_NULL || qtype == RR_TYPE_TXT || qtype == dns->cfg.rrtype)
1055 { 1179 {
1056 if (rdlen <= MAXSIZE - offs) 1180 if (rdlen <= MAXSIZE - offs)
1057 { 1181 {
1058 // decode bytes, finally 1182 // decode bytes, finally
1059 1183
… …
1085 { 1209 {
1086 slog (L_TRACE, _("DNS: got tunnel meta command %02x"), ip [3]); 1210 slog (L_TRACE, _("DNS: got tunnel meta command %02x"), ip [3]);
1087 1211
1088 if (ip [3] == CMD_IP_RST) 1212 if (ip [3] == CMD_IP_RST)
1089 { 1213 {
1090 slog (L_DEBUG, _("DNS: got tunnel RST request")); 1214 slog (L_DEBUG, _("DNS: got tunnel RST request."));
1091 1215
1092 delete dns; c->dns = 0; 1216 dns->reset ();
1093
1094 return; 1217 return;
1095 } 1218 }
1096 else if (ip [3] == CMD_IP_SYN) 1219 else if (ip [3] == CMD_IP_SYN)
1097 { 1220 {
1098 slog (L_DEBUG, _("DNS: got tunnel SYN reply, server likes us.")); 1221 slog (L_DEBUG, _("DNS: got tunnel SYN reply, server likes us."));
1099 dns->established = true; 1222 dns->established = true;
1100 } 1223 }
1224 else if (ip [3] == CMD_IP_CSE)
1225 {
1226 if (conf.dns_case_preserving)
1227 {
1228 slog (L_INFO, _("DNS: got tunnel CSE reply, globally downgrading to case-insensitive protocol."));
1229 conf.dns_case_preserving = false;
1230 dns->reset ();
1231 return;
1232 }
1233 else
1234 {
1235 slog (L_DEBUG, _("DNS: got tunnel CSE reply, server likes us."));
1236 dns->established = true;
1237 }
1238 }
1101 else if (ip [3] == CMD_IP_REJ) 1239 else if (ip [3] == CMD_IP_REJ)
1102 { 1240 {
1103 slog (L_DEBUG, _("DNS: got tunnel REJ reply, server does not like us, aborting.")); 1241 slog (L_ERR, _("DNS: got tunnel REJ reply, server does not like us."));
1104 abort (); 1242 dns->tw.start (60.);
1105 } 1243 }
1106 else 1244 else
1245 {
1107 slog (L_INFO, _("DNS: got unknown meta command %02x"), ip [3]); 1246 slog (L_INFO, _("DNS: got unknown meta command %02x"), ip [3]);
1247 dns->tw.start (60.);
1248 }
1108 } 1249 }
1109 else 1250 else
1110 slog (L_INFO, _("DNS: got spurious a record %d.%d.%d.%d"), 1251 slog (L_INFO, _("DNS: got spurious a record %d.%d.%d.%d"),
1111 ip [0], ip [1], ip [2], ip [3]); 1252 ip [0], ip [1], ip [2], ip [3]);
1112 1253
… …
1136 break; 1277 break;
1137 } 1278 }
1138} 1279}
1139 1280
1140void 1281void
1141vpn::dnsv4_ev (io_watcher &w, short revents) 1282vpn::dnsv4_ev (ev::io &w, int revents)
1142{ 1283{
1143 if (revents & EVENT_READ) 1284 if (revents & EV_READ)
1144 { 1285 {
1145 dns_packet *pkt = new dns_packet; 1286 dns_packet *pkt = new dns_packet;
1146 struct sockaddr_in sa; 1287 struct sockaddr_in sa;
1147 socklen_t sa_len = sizeof (sa); 1288 socklen_t sa_len = sizeof (sa);
1148 1289
1149 pkt->len = recvfrom (w.fd, pkt->at (0), MAXSIZE, 0, (sockaddr *)&sa, &sa_len); 1290 pkt->len = recvfrom (w.fd, pkt->at (0), MAXSIZE, 0, (sockaddr *)&sa, &sa_len);
1150 1291
1151 if (pkt->len > 0) 1292 if (pkt->len > 0)
1152 { 1293 {
1153 if (THISNODE->dns_port) 1294 if (ntohs (pkt->flags) & FLAG_RESPONSE)
1295 dnsv4_client (*pkt);
1296 else
1154 { 1297 {
1155 dnsv4_server (*pkt); 1298 dnsv4_server (*pkt);
1156 sendto (w.fd, pkt->at (0), pkt->len, 0, (sockaddr *)&sa, sa_len); 1299 sendto (w.fd, pkt->at (0), pkt->len, 0, (sockaddr *)&sa, sa_len);
1157 } 1300 }
1158 else
1159 dnsv4_client (*pkt);
1160 1301
1161 delete pkt; 1302 delete pkt;
1162 } 1303 }
1163 } 1304 }
1164} 1305}
1165 1306
1166bool 1307bool
1167vpn::send_dnsv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) 1308vpn::send_dnsv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
1168{ 1309{
1169 int client = ntohs (si.port); 1310 int client = ntohl (si.host);
1170 1311
1171 assert (0 < client && client <= conns.size ()); 1312 assert (0 < client && client <= conns.size ());
1172 1313
1173 connection *c = conns [client - 1]; 1314 connection *c = conns [client - 1];
1174 1315
1175 if (!c->dns) 1316 if (!c->dns)
1176 c->dns = new dns_connection (c); 1317 c->dns = new dns_connection (c);
1177 1318
1178 if (!c->dns->snddq.put (pkt)) 1319 if (c->dns->snddq.put (pkt))
1179 return false; 1320 {
1321 min_it (c->dns->poll_interval, 0.25);
1322 c->dns->tw ();
1323 }
1180 1324
1181 c->dns->tw.trigger (); 1325 // always return true even if the buffer overflows
1182
1183 return true; 1326 return true;
1184} 1327}
1185 1328
1186void 1329void
1187connection::dnsv4_reset_connection () 1330dns_connection::time_cb (ev::timer &w, int revents)
1188{
1189 //delete dns; dns = 0; //TODO
1190}
1191
1192#define NEXT(w) do { if (next > (w)) next = w; } while (0)
1193
1194void
1195dns_connection::time_cb (time_watcher &w)
1196{ 1331{
1197 // servers have to be polled 1332 // servers have to be polled
1198 if (THISNODE->dns_port) 1333 if (THISNODE->dns_port)
1199 return; 1334 return;
1200 1335
1201 // check for timeouts and (re)transmit 1336 // check for timeouts and (re)transmit
1202 tstamp next = NOW + poll_interval; 1337 tstamp next = 86400 * 365;
1203 dns_snd *send = 0; 1338 dns_snd *send = 0;
1204 1339
1205 for (vector<dns_snd *>::iterator i = vpn->dns_sndpq.begin (); 1340 for (vector<dns_snd *>::iterator i = vpn->dns_sndpq.begin ();
1206 i != vpn->dns_sndpq.end (); 1341 i != vpn->dns_sndpq.end ();
1207 ++i) 1342 ++i)
1208 { 1343 {
1209 dns_snd *r = *i; 1344 dns_snd *r = *i;
1210 1345
1211 if (r->timeout <= NOW) 1346 if (r->timeout <= ev_now ())
1212 { 1347 {
1213 if (!send) 1348 if (!send)
1214 { 1349 {
1215 send = r; 1350 send = r;
1216 1351
1217 r->retry++; 1352 r->retry++;
1218 r->timeout = NOW + (r->retry * last_latency * 8.); 1353 r->timeout = ev_now () + r->retry * min_latency * conf.dns_timeout_factor;
1354 //printf ("RETRY %x (%d, %f)\n", r->seqno, r->retry, r->timeout - ev_now ());//D
1219 1355
1220 // the following code changes the query section a bit, forcing 1356 // the following code changes the query section a bit, forcing
1221 // the forwarder to generate a new request 1357 // the forwarder to generate a new request
1222 if (r->stdhdr) 1358 if (r->stdhdr)
1223 {
1224 //printf ("reencoded header for ID %d retry %d:%d:%d\n", htons (r->pkt->id), THISNODE->id, r->seqno, r->retry);printf ("reencoded header for ID %d retry %d:%d:%d\n", htons (r->pkt->id), THISNODE->id, r->seqno, r->retry);
1225 //encode_header ((char *)r->pkt->at (6 * 2 + 1), THISNODE->id, r->seqno, r->retry); 1359 encode_header ((char *)r->pkt->at (6 * 2 + 1), THISNODE->id, r->seqno, r->retry);
1226 }
1227 } 1360 }
1228 } 1361 }
1229 else 1362 else
1230 NEXT (r->timeout); 1363 min_it (next, r->timeout - ev_now ());
1231 }
1232
1233 if (last_sent + send_interval <= NOW)
1234 { 1364 }
1365
1235 if (!send) 1366 if (!send)
1367 {
1368 // generate a new packet, if wise
1369
1370 if (!established)
1236 { 1371 {
1237 // generate a new packet, if wise 1372 if (vpn->dns_sndpq.empty ())
1238
1239 if (!established)
1240 { 1373 {
1241 if (vpn->dns_sndpq.empty ())
1242 {
1243 send = new dns_snd (this); 1374 send = new dns_snd (this);
1244 1375
1245 printf ("new conn %p %d\n", this, c->conf->id);//D
1246 cfg.reset (THISNODE->id); 1376 cfg.reset (THISNODE->id);
1377 set_cfg ();
1247 send->gen_syn_req (); 1378 send->gen_syn_req ();
1248 }
1249 } 1379 }
1250 else if (vpn->dns_sndpq.size () < MAX_OUTSTANDING 1380 }
1381 else if (vpn->dns_sndpq.size () < conf.dns_max_outstanding
1251 && !SEQNO_EQ (rcvseq, sndseq - (MAX_WINDOW - 1))) 1382 && !SEQNO_EQ (rcvseq, sndseq - (MAX_WINDOW - 1)))
1383 {
1384 if (last_sent + send_interval <= ev_now ())
1252 { 1385 {
1253 //printf ("sending data request etc.\n"); //D 1386 //printf ("sending data request etc.\n"); //D
1254 if (!snddq.empty ()) 1387 if (!snddq.empty ())
1255 {
1256 poll_interval = send_interval;
1257 NEXT (NOW + send_interval); 1388 min_it (next, send_interval);
1258 }
1259 1389
1260 send = new dns_snd (this); 1390 send = new dns_snd (this);
1261 send->gen_stream_req (sndseq, snddq); 1391 send->gen_stream_req (sndseq, snddq);
1262 send->timeout = NOW + last_latency * 8.; 1392 send->timeout = ev_now () + min_latency * conf.dns_timeout_factor;
1393 //printf ("SEND %x (%f)\n", send->seqno, send->timeout - ev_now (), min_latency, conf.dns_timeout_factor);//D
1263 1394
1264 sndseq = (sndseq + 1) & SEQNO_MASK; 1395 sndseq = (sndseq + 1) & SEQNO_MASK;
1265 } 1396 }
1266 1397 else
1267 if (send) 1398 min_it (next, last_sent + send_interval - ev_now ());
1268 vpn->dns_sndpq.push_back (send);
1269 } 1399 }
1270 1400
1271 if (send) 1401 if (send)
1272 { 1402 vpn->dns_sndpq.push_back (send);
1273 last_sent = NOW; 1403 }
1404
1405 if (send)
1406 {
1407 last_sent = ev_now ();
1274 sendto (vpn->dnsv4_fd, 1408 sendto (vpn->dnsv4_fd,
1275 send->pkt->at (0), send->pkt->len, 0, 1409 send->pkt->at (0), send->pkt->len, 0,
1276 vpn->dns_forwarder.sav4 (), vpn->dns_forwarder.salenv4 ()); 1410 vpn->dns_forwarder.sav4 (), vpn->dns_forwarder.salenv4 ());
1277 }
1278 } 1411 }
1279 else
1280 NEXT (last_sent + send_interval);
1281 1412
1413 min_it (next, last_sent + max (poll_interval, send_interval) - ev_now ());
1414
1282 slog (L_NOISE, "DNS: pi %f si %f N %f (%d:%d)", 1415 slog (L_NOISE, "DNS: pi %f si %f N %f (%d:%d %d)",
1283 poll_interval, send_interval, next - NOW, 1416 poll_interval, send_interval, next - ev_now (),
1284 vpn->dns_sndpq.size (), snddq.size ()); 1417 vpn->dns_sndpq.size (), snddq.size (),
1285 1418 rcvpq.size ());
1286 // TODO: no idea when this happens, but when next < NOW, we have a problem
1287 if (next < NOW + 0.0001)
1288 next = NOW + 0.1;
1289 1419
1290 w.start (next); 1420 w.start (next);
1291} 1421}
1292 1422
1293#endif 1423#endif

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines