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.30 by pcg, Mon Mar 14 17:40:01 2005 UTC vs.
Revision 1.38 by pcg, Tue Apr 19 04:23:38 2005 UTC

… …
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
18 along with gvpe; if not, write to the Free Software 18 along with gvpe; if not, write to the Free Software
19 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 19 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20*/ 20*/
21 21
22// TODO: EDNS0 option to increase dns mtu?
23// TODO: re-write dns packet parsing/creation using a safe mem-buffer
24// to ensure no buffer overflows or similar problems.
25
22#include "config.h" 26#include "config.h"
23 27
24#if ENABLE_DNS 28#if ENABLE_DNS
25 29
26// dns processing is EXTREMELY ugly. For obvious(?) reasons. 30// dns processing is EXTREMELY ugly. For obvious(?) reasons.
… …
48 52
49#define MAX_POLL_INTERVAL 5. // how often to poll minimally when the server has no data 53#define MAX_POLL_INTERVAL 5. // how often to poll minimally when the server has no data
50#define ACTIVITY_INTERVAL 5. 54#define ACTIVITY_INTERVAL 5.
51 55
52#define INITIAL_TIMEOUT 0.1 // retry timeouts 56#define INITIAL_TIMEOUT 0.1 // retry timeouts
53#define INITIAL_SYN_TIMEOUT 10. // retry timeout for initial syn 57#define INITIAL_SYN_TIMEOUT 2. // retry timeout for initial syn
54 58
55#define MIN_SEND_INTERVAL 0.001 // wait at least this time between sending requests
56#define MAX_SEND_INTERVAL 2. // optimistic? 59#define MAX_SEND_INTERVAL 2. // optimistic?
57 60
58#define LATENCY_FACTOR 0.5 // RTT * LATENCY_FACTOR == sending rate
59#define MAX_OUTSTANDING 2 // max. outstanding requests
60#define MAX_WINDOW 1000 // max. for MAX_OUTSTANDING, and backlog 61#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 62#define MAX_BACKLOG (64*1024) // size of gvpe protocol backlog (bytes), must be > MAXSIZE
62 63
63#define MAX_DOMAIN_SIZE 240 // 255 is legal limit, but bind doesn't compress well 64#define MAX_DOMAIN_SIZE 240 // 255 is legal limit, but bind doesn't compress well
64// 240 leaves about 4 bytes of server reply data 65// 240 leaves about 4 bytes of server reply data
65// every two request bytes less give room for one reply byte 66// every request byte less give room for two reply bytes
66 67
67#define SEQNO_MASK 0x3fff 68#define SEQNO_MASK 0x3fff
68#define SEQNO_EQ(a,b) ( 0 == ( ((a) ^ (b)) & SEQNO_MASK) ) 69#define SEQNO_EQ(a,b) ( 0 == ( ((a) ^ (b)) & SEQNO_MASK) )
69 70
70#define MAX_LBL_SIZE 63 71#define MAX_LBL_SIZE 63
… …
553 double min_latency; 554 double min_latency;
554 double poll_interval, send_interval; 555 double poll_interval, send_interval;
555 556
556 vector<dns_rcv *> rcvpq; 557 vector<dns_rcv *> rcvpq;
557 558
558 byte_stream rcvdq; int rcvseq; 559 byte_stream rcvdq; int rcvseq; int repseq;
559 byte_stream snddq; int sndseq; 560 byte_stream snddq; int sndseq;
560 561
561 void time_cb (time_watcher &w); time_watcher tw; 562 void time_cb (time_watcher &w); time_watcher tw;
562 void receive_rep (dns_rcv *r); 563 void receive_rep (dns_rcv *r);
563 564
… …
720///////////////////////////////////////////////////////////////////////////// 721/////////////////////////////////////////////////////////////////////////////
721 722
722dns_connection::dns_connection (connection *c) 723dns_connection::dns_connection (connection *c)
723: c (c) 724: c (c)
724, rcvdq (MAX_BACKLOG * 2) 725, rcvdq (MAX_BACKLOG * 2)
725, snddq (MAX_BACKLOG * 2) 726, snddq (MAX_BACKLOG)
726, tw (this, &dns_connection::time_cb) 727, tw (this, &dns_connection::time_cb)
727{ 728{
728 vpn = c->vpn; 729 vpn = c->vpn;
729 730
730 established = false; 731 established = false;
731 732
732 rcvseq = sndseq = 0; 733 rcvseq = repseq = sndseq = 0;
733 734
734 last_sent = last_received = 0; 735 last_sent = last_received = 0;
735 poll_interval = 0.5; // starting here 736 poll_interval = 0.5; // starting here
736 send_interval = 0.5; // starting rate 737 send_interval = 0.5; // starting rate
737 min_latency = INITIAL_TIMEOUT; 738 min_latency = INITIAL_TIMEOUT;
… …
791 } 792 }
792 793
793 while (vpn_packet *pkt = rcvdq.get ()) 794 while (vpn_packet *pkt = rcvdq.get ())
794 { 795 {
795 sockinfo si; 796 sockinfo si;
796 si.host = 0x01010101; si.port = htons (c->conf->id); si.prot = PROT_DNSv4; 797 si.host = htonl (c->conf->id); si.port = 0; si.prot = PROT_DNSv4;
797 798
798 vpn->recv_vpn_packet (pkt, si); 799 vpn->recv_vpn_packet (pkt, si);
799 800
800 delete pkt; 801 delete pkt;
801 } 802 }
… …
855 if (0 < client && client <= conns.size ()) 856 if (0 < client && client <= conns.size ())
856 { 857 {
857 connection *c = conns [client - 1]; 858 connection *c = conns [client - 1];
858 dns_connection *dns = c->dns; 859 dns_connection *dns = c->dns;
859 dns_rcv *rcv; 860 dns_rcv *rcv;
860 bool in_seq;
861 861
862 if (dns) 862 if (dns)
863 { 863 {
864 for (vector<dns_rcv *>::iterator i = dns->rcvpq.end (); i-- != dns->rcvpq.begin (); ) 864 for (vector<dns_rcv *>::iterator i = dns->rcvpq.end (); i-- != dns->rcvpq.begin (); )
865 if (SEQNO_EQ ((*i)->seqno, seqno)) 865 if (SEQNO_EQ ((*i)->seqno, seqno))
… …
876 memcpy (pkt.at (0), r->pkt->at (0), offs = r->pkt->len); 876 memcpy (pkt.at (0), r->pkt->at (0), offs = r->pkt->len);
877 877
878 goto duplicate_request; 878 goto duplicate_request;
879 } 879 }
880 880
881 in_seq = dns->rcvseq == seqno;
882
883 // new packet, queue 881 // new packet, queue
884 rcv = new dns_rcv (seqno, data, datalen); 882 rcv = new dns_rcv (seqno, data, datalen);
885 dns->receive_rep (rcv); 883 dns->receive_rep (rcv);
886 } 884 }
887 885
… …
904 dlen -= qlen; 902 dlen -= qlen;
905 903
906 // only put data into in-order sequence packets, if 904 // only put data into in-order sequence packets, if
907 // we receive out-of-order packets we generate empty 905 // we receive out-of-order packets we generate empty
908 // replies 906 // replies
909 while (dlen > 1 && !dns->snddq.empty () && in_seq) 907 //printf ("%d - %d & %x (=%d) < %d\n", seqno, dns->repseq, SEQNO_MASK, (seqno - dns->repseq) & SEQNO_MASK, MAX_WINDOW);//D
908 if (((seqno - dns->repseq) & SEQNO_MASK) <= MAX_WINDOW)
910 { 909 {
910 dns->repseq = seqno;
911
912 while (dlen > 1 && !dns->snddq.empty ())
913 {
911 int txtlen = dlen <= 255 ? dlen - 1 : 255; 914 int txtlen = dlen <= 255 ? dlen - 1 : 255;
912 915
913 if (txtlen > dns->snddq.size ()) 916 if (txtlen > dns->snddq.size ())
914 txtlen = dns->snddq.size (); 917 txtlen = dns->snddq.size ();
915 918
916 pkt[offs++] = txtlen; 919 pkt[offs++] = txtlen;
917 memcpy (pkt.at (offs), dns->snddq.begin (), txtlen); 920 memcpy (pkt.at (offs), dns->snddq.begin (), txtlen);
918 offs += txtlen; 921 offs += txtlen;
919 dns->snddq.remove (txtlen); 922 dns->snddq.remove (txtlen);
920 923
921 dlen -= txtlen + 1; 924 dlen -= txtlen + 1;
925 }
922 } 926 }
923 927
924 // avoid empty TXT rdata 928 // avoid completely empty TXT rdata
925 if (offs == rdlen_offs) 929 if (offs == rdlen_offs)
926 pkt[offs++] = 0; 930 pkt[offs++] = 0;
927 931
928 slog (L_NOISE, "DNS: snddq %d", dns->snddq.size ()); 932 slog (L_NOISE, "DNS: snddq %d", dns->snddq.size ());
929 } 933 }
… …
1028 double latency = NOW - (*i)->sent; 1032 double latency = NOW - (*i)->sent;
1029 1033
1030 if (latency < dns->min_latency) 1034 if (latency < dns->min_latency)
1031 dns->min_latency = latency; 1035 dns->min_latency = latency;
1032 1036
1033 if (dns->send_interval > dns->min_latency * LATENCY_FACTOR) 1037 if (dns->send_interval > dns->min_latency * conf.dns_overlap_factor)
1034 dns->send_interval = dns->min_latency * LATENCY_FACTOR; 1038 dns->send_interval = dns->min_latency * conf.dns_overlap_factor;
1035 1039
1036 if (dns->send_interval < MIN_SEND_INTERVAL) 1040 if (dns->send_interval < conf.dns_send_interval)
1037 dns->send_interval = MIN_SEND_INTERVAL; 1041 dns->send_interval = conf.dns_send_interval;
1038 } 1042 }
1039 1043
1040 delete *i; 1044 delete *i;
1041 dns_sndpq.erase (i); 1045 dns_sndpq.erase (i);
1042 1046
… …
1175} 1179}
1176 1180
1177bool 1181bool
1178vpn::send_dnsv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) 1182vpn::send_dnsv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
1179{ 1183{
1180 int client = ntohs (si.port); 1184 int client = ntohl (si.host);
1181 1185
1182 assert (0 < client && client <= conns.size ()); 1186 assert (0 < client && client <= conns.size ());
1183 1187
1184 connection *c = conns [client - 1]; 1188 connection *c = conns [client - 1];
1185 1189
1186 if (!c->dns) 1190 if (!c->dns)
1187 c->dns = new dns_connection (c); 1191 c->dns = new dns_connection (c);
1188 1192
1189 if (!c->dns->snddq.put (pkt)) 1193 if (c->dns->snddq.put (pkt))
1190 return false;
1191
1192 c->dns->tw.trigger (); 1194 c->dns->tw.trigger ();
1193 1195
1196 // always return true even if the buffer overflows
1194 return true; 1197 return true;
1195} 1198}
1196 1199
1197void 1200void
1198connection::dnsv4_reset_connection () 1201connection::dnsv4_reset_connection ()
… …
1224 if (!send) 1227 if (!send)
1225 { 1228 {
1226 send = r; 1229 send = r;
1227 1230
1228 r->retry++; 1231 r->retry++;
1229 r->timeout = NOW + (r->retry * min_latency * 8.); 1232 r->timeout = NOW + (r->retry * min_latency * conf.dns_timeout_factor);
1230 1233
1231 // the following code changes the query section a bit, forcing 1234 // the following code changes the query section a bit, forcing
1232 // the forwarder to generate a new request 1235 // the forwarder to generate a new request
1233 if (r->stdhdr) 1236 if (r->stdhdr)
1234 { 1237 {
… …
1253 1256
1254 cfg.reset (THISNODE->id); 1257 cfg.reset (THISNODE->id);
1255 send->gen_syn_req (); 1258 send->gen_syn_req ();
1256 } 1259 }
1257 } 1260 }
1258 else if (vpn->dns_sndpq.size () < MAX_OUTSTANDING 1261 else if (vpn->dns_sndpq.size () < conf.dns_max_outstanding
1259 && !SEQNO_EQ (rcvseq, sndseq - (MAX_WINDOW - 1))) 1262 && !SEQNO_EQ (rcvseq, sndseq - (MAX_WINDOW - 1)))
1260 { 1263 {
1261 if (last_sent + send_interval <= NOW) 1264 if (last_sent + send_interval <= NOW)
1262 { 1265 {
1263 //printf ("sending data request etc.\n"); //D 1266 //printf ("sending data request etc.\n"); //D
… …
1267 NEXT (NOW + send_interval); 1270 NEXT (NOW + send_interval);
1268 } 1271 }
1269 1272
1270 send = new dns_snd (this); 1273 send = new dns_snd (this);
1271 send->gen_stream_req (sndseq, snddq); 1274 send->gen_stream_req (sndseq, snddq);
1272 send->timeout = NOW + min_latency * 8.; 1275 send->timeout = NOW + min_latency * conf.dns_timeout_factor;
1273 1276
1274 sndseq = (sndseq + 1) & SEQNO_MASK; 1277 sndseq = (sndseq + 1) & SEQNO_MASK;
1275 } 1278 }
1276 else 1279 else
1277 NEXT (last_sent + send_interval); 1280 NEXT (last_sent + send_interval);
… …
1293 poll_interval, send_interval, next - NOW, 1296 poll_interval, send_interval, next - NOW,
1294 vpn->dns_sndpq.size (), snddq.size (), 1297 vpn->dns_sndpq.size (), snddq.size (),
1295 rcvpq.size ()); 1298 rcvpq.size ());
1296 1299
1297 // TODO: no idea when this happens, but when next < NOW, we have a problem 1300 // TODO: no idea when this happens, but when next < NOW, we have a problem
1301 // doesn't seem to happen anymore
1298 if (next < NOW + 0.001) 1302 if (next < NOW + 0.001)
1299 next = NOW + 0.1; 1303 next = NOW + 0.1;
1300 1304
1301 w.start (next); 1305 w.start (next);
1302} 1306}

Diff Legend

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