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

Comparing gvpe/src/connection.C (file contents):
Revision 1.32 by pcg, Sun Mar 21 13:15:42 2004 UTC vs.
Revision 1.43 by pcg, Fri Mar 4 04:52:38 2005 UTC

1/* 1/*
2 connection.C -- manage a single connection 2 connection.C -- manage a single connection
3 Copyright (C) 2003-2004 Marc Lehmann <pcg@goof.com> 3 Copyright (C) 2003-2005 Marc Lehmann <gvpe@schmorp.de>
4 4
5 This file is part of GVPE.
6
5 This program is free software; you can redistribute it and/or modify 7 GVPE is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by 8 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 2 of the License, or 9 the Free Software Foundation; either version 2 of the License, or
8 (at your option) any later version. 10 (at your option) any later version.
9 11
10 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,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of 13 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details. 15 GNU General Public License for more details.
14 16
15 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
16 along with this program; if not, write to the Free Software 18 along with gvpe; if not, write to the Free Software
17 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
18*/ 20*/
19 21
20#include "config.h" 22#include "config.h"
21 23
… …
202// only do action once every x seconds per host whole allowing bursts. 204// only do action once every x seconds per host whole allowing bursts.
203// this implementation ("splay list" ;) is inefficient, 205// this implementation ("splay list" ;) is inefficient,
204// but low on resources. 206// but low on resources.
205struct net_rate_limiter : list<net_rateinfo> 207struct net_rate_limiter : list<net_rateinfo>
206{ 208{
207 static const double ALPHA = 1. - 1. / 600.; // allow bursts 209# define NRL_ALPHA (1. - 1. / 600.) // allow bursts
208 static const double CUTOFF = 10.; // one event every CUTOFF seconds 210# define NRL_CUTOFF 10. // one event every CUTOFF seconds
209 static const double EXPIRE = CUTOFF * 30.; // expire entries after this time 211# define NRL_EXPIRE (NRL_CUTOFF * 30.) // expire entries after this time
210 static const double MAXDIF = CUTOFF * (1. / (1. - ALPHA)); // maximum diff /count value 212# define NRL_MAXDIF (NRL_CUTOFF * (1. / (1. - NRL_ALPHA))) // maximum diff /count value
211 213
212 bool can (const sockinfo &si) { return can((u32)si.host); } 214 bool can (const sockinfo &si) { return can((u32)si.host); }
213 bool can (u32 host); 215 bool can (u32 host);
214}; 216};
215 217
… …
220 iterator i; 222 iterator i;
221 223
222 for (i = begin (); i != end (); ) 224 for (i = begin (); i != end (); )
223 if (i->host == host) 225 if (i->host == host)
224 break; 226 break;
225 else if (i->last < NOW - EXPIRE) 227 else if (i->last < NOW - NRL_EXPIRE)
226 i = erase (i); 228 i = erase (i);
227 else 229 else
228 i++; 230 i++;
229 231
230 if (i == end ()) 232 if (i == end ())
231 { 233 {
232 net_rateinfo ri; 234 net_rateinfo ri;
233 235
234 ri.host = host; 236 ri.host = host;
235 ri.pcnt = 1.; 237 ri.pcnt = 1.;
236 ri.diff = MAXDIF; 238 ri.diff = NRL_MAXDIF;
237 ri.last = NOW; 239 ri.last = NOW;
238 240
239 push_front (ri); 241 push_front (ri);
240 242
241 return true; 243 return true;
… …
243 else 245 else
244 { 246 {
245 net_rateinfo ri (*i); 247 net_rateinfo ri (*i);
246 erase (i); 248 erase (i);
247 249
248 ri.pcnt = ri.pcnt * ALPHA; 250 ri.pcnt = ri.pcnt * NRL_ALPHA;
249 ri.diff = ri.diff * ALPHA + (NOW - ri.last); 251 ri.diff = ri.diff * NRL_ALPHA + (NOW - ri.last);
250 252
251 ri.last = NOW; 253 ri.last = NOW;
252 254
253 double dif = ri.diff / ri.pcnt; 255 double dif = ri.diff / ri.pcnt;
254 256
255 bool send = dif > CUTOFF; 257 bool send = dif > NRL_CUTOFF;
256 258
257 if (dif > MAXDIF) 259 if (dif > NRL_MAXDIF)
258 { 260 {
259 ri.pcnt = 1.; 261 ri.pcnt = 1.;
260 ri.diff = MAXDIF; 262 ri.diff = NRL_MAXDIF;
261 } 263 }
262 else if (send) 264 else if (send)
263 ri.pcnt++; 265 ri.pcnt++;
264 266
265 push_front (ri); 267 push_front (ri);
… …
589{ 591{
590 if (ictx && octx) 592 if (ictx && octx)
591 { 593 {
592 connectmode = conf->connectmode; 594 connectmode = conf->connectmode;
593 595
596 // make sure rekeying timeouts are slightly asymmetric
594 rekey.start (NOW + ::conf.rekey); 597 rekey.start (NOW + ::conf.rekey
598 + (conf->id > THISNODE->id ? 10 : 0));
595 keepalive.start (NOW + ::conf.keepalive); 599 keepalive.start (NOW + ::conf.keepalive);
596 600
597 // send queued packets 601 // send queued packets
598 if (ictx && octx) 602 if (ictx && octx)
599 { 603 {
… …
625 protocol = best_protocol (THISNODE->protocols & conf->protocols); 629 protocol = best_protocol (THISNODE->protocols & conf->protocols);
626 630
627 // mask out protocols we cannot establish 631 // mask out protocols we cannot establish
628 if (!conf->udp_port) protocol &= ~PROT_UDPv4; 632 if (!conf->udp_port) protocol &= ~PROT_UDPv4;
629 if (!conf->tcp_port) protocol &= ~PROT_TCPv4; 633 if (!conf->tcp_port) protocol &= ~PROT_TCPv4;
634 if (!conf->dns_port) protocol &= ~PROT_DNSv4;
630 635
631 si.set (conf, protocol); 636 si.set (conf, protocol);
632} 637}
633 638
634// ensure sockinfo is valid, forward if necessary 639// ensure sockinfo is valid, forward if necessary
… …
654} 659}
655 660
656void 661void
657connection::send_vpn_packet (vpn_packet *pkt, const sockinfo &si, int tos) 662connection::send_vpn_packet (vpn_packet *pkt, const sockinfo &si, int tos)
658{ 663{
659 if (!vpn->send_vpn_packet (pkt, si, tos)) 664 bool ok;
665
666 switch (si.prot)
667 {
668 case PROT_IPv4:
669 ok = vpn->send_ipv4_packet (pkt, si, tos); break;
670 case PROT_UDPv4:
671 ok = vpn->send_udpv4_packet (pkt, si, tos); break;
672#if ENABLE_TCP
673 case PROT_TCPv4:
674 ok = vpn->send_tcpv4_packet (pkt, si, tos); break;
675#endif
676#if ENABLE_ICMP
677 case PROT_ICMPv4:
678 ok = vpn->send_icmpv4_packet (pkt, si, tos); break;
679#endif
680#if ENABLE_DNS
681 case PROT_DNSv4:
682 ok = send_dnsv4_packet (pkt, si, tos); break;
683#endif
684
685 default:
686 slog (L_CRIT, _("%s: FATAL: trying to send packet with unsupported protocol"), (const char *)si);
687 ok = false;
688 }
689
690 if (!ok)
660 reset_connection (); 691 reset_connection ();
661} 692}
662 693
663void 694void
664connection::send_ping (const sockinfo &si, u8 pong) 695connection::send_ping (const sockinfo &si, u8 pong)
… …
742 && connectmode != conf_node::C_DISABLED 773 && connectmode != conf_node::C_DISABLED
743 && NOW > w.at) 774 && NOW > w.at)
744 { 775 {
745 double retry_int = double (retry_cnt & 3 ? (retry_cnt & 3) : 1 << (retry_cnt >> 2)) * 0.6; 776 double retry_int = double (retry_cnt & 3 ? (retry_cnt & 3) : 1 << (retry_cnt >> 2)) * 0.6;
746 777
747 if (retry_int < 3600 * 8) 778 if (retry_int < conf->max_retry)
748 retry_cnt++; 779 retry_cnt++;
780 else
781 retry_int = conf->max_retry;
749 782
750 w.start (NOW + retry_int); 783 w.start (NOW + retry_int);
751 784
752 reset_si (); 785 reset_si ();
753 786
… …
781 } 814 }
782 815
783 delete ictx; ictx = 0; 816 delete ictx; ictx = 0;
784 delete octx; octx = 0; 817 delete octx; octx = 0;
785 818
786 si.host= 0; 819 si.host = 0;
787 820
788 last_activity = 0; 821 last_activity = 0;
789 retry_cnt = 0; 822 retry_cnt = 0;
790 823
791 rekey.stop (); 824 rekey.stop ();
… …
1059 conf->nodename, (const char *)si, (const char *)rsi); 1092 conf->nodename, (const char *)si, (const char *)rsi);
1060 } 1093 }
1061 } 1094 }
1062 1095
1063 delete d; 1096 delete d;
1064
1065 break; 1097 break;
1066 } 1098 }
1067 } 1099 }
1068 1100
1069 send_reset (rsi); 1101 send_reset (rsi);
… …
1095 break; 1127 break;
1096 1128
1097 case vpn_packet::PT_CONNECT_INFO: 1129 case vpn_packet::PT_CONNECT_INFO:
1098 if (ictx && octx && rsi == si && pkt->hmac_chk (ictx)) 1130 if (ictx && octx && rsi == si && pkt->hmac_chk (ictx))
1099 { 1131 {
1100 connect_info_packet *p = (connect_info_packet *) pkt; 1132 connect_info_packet *p = (connect_info_packet *)pkt;
1101 1133
1102 assert (p->id > 0 && p->id <= vpn->conns.size ()); // hmac-auth does not mean we accept anything 1134 if (p->id > 0 && p->id <= vpn->conns.size ()) // hmac-auth does not mean we accept anything
1103 1135 {
1104 connection *c = vpn->conns[p->id - 1]; 1136 connection *c = vpn->conns[p->id - 1];
1105 1137
1106 c->conf->protocols = p->protocols; 1138 c->conf->protocols = p->protocols;
1107 protocol = best_protocol (c->conf->protocols & THISNODE->protocols & p->si.supported_protocols (c->conf)); 1139 protocol = best_protocol (c->conf->protocols & THISNODE->protocols & p->si.supported_protocols (c->conf));
1108 p->si.upgrade_protocol (protocol, c->conf); 1140 p->si.upgrade_protocol (protocol, c->conf);
1109 1141
1110 slog (L_TRACE, "<<%d PT_CONNECT_INFO(%d,%s) (%d)", 1142 slog (L_TRACE, "<<%d PT_CONNECT_INFO(%d,%s) (%d)",
1111 conf->id, p->id, (const char *)p->si, !c->ictx && !c->octx); 1143 conf->id, p->id, (const char *)p->si, !c->ictx && !c->octx);
1112 1144
1113 const sockinfo &dsi = forward_si (p->si); 1145 const sockinfo &dsi = forward_si (p->si);
1114 1146
1115 if (dsi.valid ()) 1147 if (dsi.valid ())
1116 c->send_auth_request (dsi, true); 1148 c->send_auth_request (dsi, true);
1149 }
1117 } 1150 }
1118 1151
1119 break; 1152 break;
1120 1153
1121 default: 1154 default:
… …
1185 putenv ("STATE=down"); 1218 putenv ("STATE=down");
1186 1219
1187 return ::conf.script_node_up ? ::conf.script_node_down : "node-down"; 1220 return ::conf.script_node_up ? ::conf.script_node_down : "node-down";
1188} 1221}
1189 1222
1190connection::connection(struct vpn *vpn_) 1223connection::connection (struct vpn *vpn, conf_node *conf)
1191: vpn(vpn_) 1224: vpn(vpn), conf(conf)
1192, rekey (this, &connection::rekey_cb) 1225, rekey (this, &connection::rekey_cb)
1193, keepalive (this, &connection::keepalive_cb) 1226, keepalive (this, &connection::keepalive_cb)
1194, establish_connection (this, &connection::establish_connection_cb) 1227, establish_connection (this, &connection::establish_connection_cb)
1228#if ENABLE_DNS
1229, dns (0)
1230#endif
1195{ 1231{
1196 octx = ictx = 0; 1232 octx = ictx = 0;
1197 retry_cnt = 0; 1233 retry_cnt = 0;
1198 1234
1235 if (!conf->protocols) // make sure some protocol is enabled
1236 conf->protocols = PROT_UDPv4;
1237
1199 connectmode = conf_node::C_ALWAYS; // initial setting 1238 connectmode = conf_node::C_ALWAYS; // initial setting
1200 reset_connection (); 1239 reset_connection ();
1201} 1240}
1202 1241
1203connection::~connection () 1242connection::~connection ()

Diff Legend

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