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. |
205 | struct net_rate_limiter : list<net_rateinfo> |
207 | struct 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 | |
656 | void |
661 | void |
657 | connection::send_vpn_packet (vpn_packet *pkt, const sockinfo &si, int tos) |
662 | connection::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 | |
663 | void |
694 | void |
664 | connection::send_ping (const sockinfo &si, u8 pong) |
695 | connection::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 (); |
… | |
… | |
1094 | break; |
1127 | break; |
1095 | |
1128 | |
1096 | case vpn_packet::PT_CONNECT_INFO: |
1129 | case vpn_packet::PT_CONNECT_INFO: |
1097 | if (ictx && octx && rsi == si && pkt->hmac_chk (ictx)) |
1130 | if (ictx && octx && rsi == si && pkt->hmac_chk (ictx)) |
1098 | { |
1131 | { |
1099 | connect_info_packet *p = (connect_info_packet *) pkt; |
1132 | connect_info_packet *p = (connect_info_packet *)pkt; |
1100 | |
1133 | |
1101 | 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 |
1102 | |
1135 | { |
1103 | connection *c = vpn->conns[p->id - 1]; |
1136 | connection *c = vpn->conns[p->id - 1]; |
1104 | |
1137 | |
1105 | c->conf->protocols = p->protocols; |
1138 | c->conf->protocols = p->protocols; |
1106 | 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)); |
1107 | p->si.upgrade_protocol (protocol, c->conf); |
1140 | p->si.upgrade_protocol (protocol, c->conf); |
1108 | |
1141 | |
1109 | slog (L_TRACE, "<<%d PT_CONNECT_INFO(%d,%s) (%d)", |
1142 | slog (L_TRACE, "<<%d PT_CONNECT_INFO(%d,%s) (%d)", |
1110 | 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); |
1111 | |
1144 | |
1112 | const sockinfo &dsi = forward_si (p->si); |
1145 | const sockinfo &dsi = forward_si (p->si); |
1113 | |
1146 | |
1114 | if (dsi.valid ()) |
1147 | if (dsi.valid ()) |
1115 | c->send_auth_request (dsi, true); |
1148 | c->send_auth_request (dsi, true); |
|
|
1149 | } |
1116 | } |
1150 | } |
1117 | |
1151 | |
1118 | break; |
1152 | break; |
1119 | |
1153 | |
1120 | default: |
1154 | default: |
… | |
… | |
1184 | putenv ("STATE=down"); |
1218 | putenv ("STATE=down"); |
1185 | |
1219 | |
1186 | return ::conf.script_node_up ? ::conf.script_node_down : "node-down"; |
1220 | return ::conf.script_node_up ? ::conf.script_node_down : "node-down"; |
1187 | } |
1221 | } |
1188 | |
1222 | |
1189 | connection::connection(struct vpn *vpn_) |
1223 | connection::connection (struct vpn *vpn, conf_node *conf) |
1190 | : vpn(vpn_) |
1224 | : vpn(vpn), conf(conf) |
1191 | , rekey (this, &connection::rekey_cb) |
1225 | , rekey (this, &connection::rekey_cb) |
1192 | , keepalive (this, &connection::keepalive_cb) |
1226 | , keepalive (this, &connection::keepalive_cb) |
1193 | , establish_connection (this, &connection::establish_connection_cb) |
1227 | , establish_connection (this, &connection::establish_connection_cb) |
|
|
1228 | #if ENABLE_DNS |
|
|
1229 | , dns (0) |
|
|
1230 | #endif |
1194 | { |
1231 | { |
1195 | octx = ictx = 0; |
1232 | octx = ictx = 0; |
1196 | retry_cnt = 0; |
1233 | retry_cnt = 0; |
1197 | |
1234 | |
|
|
1235 | if (!conf->protocols) // make sure some protocol is enabled |
|
|
1236 | conf->protocols = PROT_UDPv4; |
|
|
1237 | |
1198 | connectmode = conf_node::C_ALWAYS; // initial setting |
1238 | connectmode = conf_node::C_ALWAYS; // initial setting |
1199 | reset_connection (); |
1239 | reset_connection (); |
1200 | } |
1240 | } |
1201 | |
1241 | |
1202 | connection::~connection () |
1242 | connection::~connection () |