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

Comparing gvpe/src/vpn.C (file contents):
Revision 1.4 by pcg, Fri Apr 4 05:26:45 2003 UTC vs.
Revision 1.39 by pcg, Sat Nov 10 05:14:22 2007 UTC

1/* 1/*
2 vpn.C -- handle the protocol, encryption, handshaking etc. 2 vpn.C -- handle the protocol, encryption, handshaking etc.
3 Copyright (C) 2003-2005 Marc Lehmann <gvpe@schmorp.de>
3 4
5 This file is part of GVPE.
6
4 This program is free software; you can redistribute it and/or modify 7 GVPE is free software; you can redistribute it and/or modify
5 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
6 the Free Software Foundation; either version 2 of the License, or 9 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version. 10 (at your option) any later version.
8 11
9 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,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of 13 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details. 15 GNU General Public License for more details.
13 16
14 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
15 along with this program; if not, write to the Free Software 18 along with gvpe; if not, write to the Free Software
16 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 19 Foundation, Inc. 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17*/ 20*/
18 21
19#include "config.h" 22#include "config.h"
20 23
21#include <list> 24#include <list>
22 25
26#include <cstdio>
27#include <cstring>
23#include <cstdlib> 28#include <cstdlib>
24#include <cstring>
25#include <cstdio>
26 29
27#include <sys/types.h> 30#include <sys/types.h>
28#include <sys/socket.h> 31#include <sys/socket.h>
29#include <sys/poll.h>
30#include <sys/wait.h> 32#include <sys/wait.h>
31#include <netinet/in.h>
32#include <arpa/inet.h>
33#include <errno.h> 33#include <errno.h>
34#include <time.h> 34#include <time.h>
35#include <unistd.h> 35#include <unistd.h>
36#include <fcntl.h>
37#include <sys/socket.h>
38
39#include "netcompat.h"
36 40
37#include "pidfile.h" 41#include "pidfile.h"
38 42
39#include "connection.h" 43#include "connection.h"
40#include "util.h" 44#include "util.h"
41#include "vpn.h" 45#include "vpn.h"
42 46
43#if ENABLE_TCP 47vpn network; // THE vpn (bad design...)
44# include <map>
45# include <unistd.h>
46# include <fcntl.h>
47# include <sys/poll.h>
48#endif
49 48
50///////////////////////////////////////////////////////////////////////////// 49/////////////////////////////////////////////////////////////////////////////
51 50
52const char *vpn::script_if_up () 51void
52vpn::script_init_env ()
53{ 53{
54 // the tunnel device mtu should be the physical mtu - overhead 54 // the tunnel device mtu should be the physical mtu - overhead
55 // the tricky part is rounding to the cipher key blocksize 55 // the tricky part is rounding to the cipher key blocksize
56 int mtu = conf.mtu - ETH_OVERHEAD - VPE_OVERHEAD - MAX_OVERHEAD; 56 int mtu = conf.mtu - ETH_OVERHEAD - VPE_OVERHEAD - MAX_OVERHEAD;
57 mtu += ETH_OVERHEAD - 6 - 6; // now we have the data portion 57 mtu += ETH_OVERHEAD - 6 - 6; // now we have the data portion
58 mtu -= mtu % EVP_CIPHER_block_size (CIPHER); // round 58 mtu -= mtu % EVP_CIPHER_block_size (CIPHER); // round
59 mtu -= ETH_OVERHEAD - 6 - 6; // and get interface mtu again 59 mtu -= ETH_OVERHEAD - 6 - 6; // and get interface mtu again
60 60
61 char *env; 61 char *env;
62 asprintf (&env, "CONFBASE=%s", confbase); 62 asprintf (&env, "CONFBASE=%s", confbase); putenv (env);
63 putenv (env);
64 asprintf (&env, "NODENAME=%s", THISNODE->nodename);
65 putenv (env);
66 asprintf (&env, "NODEID=%d", THISNODE->id);
67 putenv (env);
68 asprintf (&env, "IFNAME=%s", tap->interface ()); 63 asprintf (&env, "IFNAME=%s", tap->interface ()); putenv (env);
69 putenv (env); 64 asprintf (&env, "IFTYPE=%s", IFTYPE); putenv (env);
65 asprintf (&env, "IFSUBTYPE=%s", IFSUBTYPE); putenv (env);
70 asprintf (&env, "MTU=%d", mtu); 66 asprintf (&env, "MTU=%d", mtu); putenv (env);
71 putenv (env); 67 asprintf (&env, "NODES=%d", conns.size ()); putenv (env);
72 asprintf (&env, "MAC=%02x:%02x:%02x:%02x:%02x:%02x", 68 asprintf (&env, "NODEID=%d", THISNODE->id); putenv (env);
73 0xfe, 0xfd, 0x80, 0x00, THISNODE->id >> 8,
74 THISNODE->id & 0xff);
75 putenv (env);
76 69
70 conns [THISNODE->id - 1]->script_init_env ("");
71
72 for (conns_vector::iterator c = conns.begin (); c != conns.end (); ++c)
73 {
74 char ext[16];
75 snprintf (ext, 16, "_%d", (*c)->conf->id);
76 (*c)->script_init_env (ext);
77 }
78}
79
80const char *vpn::script_if_init ()
81{
82 script_init_env ();
83
84 return tap->if_up ();
85}
86
87const char *vpn::script_if_up ()
88{
89 script_init_env ();
90
91 char *filename;
92 asprintf (&filename,
93 "%s/%s",
94 confbase,
77 return ::conf.script_if_up ? ::conf.script_if_up : "if-up"; 95 ::conf.script_if_up ? ::conf.script_if_up : "if-up");
96
97 return filename;
78} 98}
79 99
80int 100int
81vpn::setup () 101vpn::setup ()
82{ 102{
103 ipv4_fd = -1;
104
105 if (THISNODE->protocols & PROT_IPv4 && ::conf.ip_proto)
106 {
107 ipv4_fd = socket (PF_INET, SOCK_RAW, ::conf.ip_proto);
108
109 if (ipv4_fd < 0)
110 return -1;
111
112 fcntl (ipv4_fd, F_SETFL, O_NONBLOCK);
113 fcntl (ipv4_fd, F_SETFD, FD_CLOEXEC);
114
115#if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
116 // this I really consider a linux bug. I am neither connected
117 // nor do I fragment myself. Linux still sets DF and doesn't
118 // fragment for me sometimes.
119 {
120 int oval = IP_PMTUDISC_DONT;
121 setsockopt (ipv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval);
122 }
123#endif
124
125 sockinfo si (THISNODE, PROT_IPv4);
126
127 if (bind (ipv4_fd, si.sav4 (), si.salenv4 ()))
128 {
129 slog (L_ERR, _("can't bind ipv4 socket on %s: %s"), (const char *)si, strerror (errno));
130 exit (EXIT_FAILURE);
131 }
132
133 ipv4_ev_watcher.start (ipv4_fd, EV_READ);
134 }
135
83 udpv4_fd = -1; 136 udpv4_fd = -1;
84 137
85 if (THISNODE->protocols & PROT_UDPv4) 138 if (THISNODE->protocols & PROT_UDPv4 && THISNODE->udp_port)
86 { 139 {
87 udpv4_fd = socket (PF_INET, SOCK_DGRAM, IPPROTO_UDP); 140 udpv4_fd = socket (PF_INET, SOCK_DGRAM, IPPROTO_UDP);
88 141
89 if (udpv4_fd < 0) 142 if (udpv4_fd < 0)
90 return -1; 143 return -1;
144
145 fcntl (udpv4_fd, F_SETFL, O_NONBLOCK);
146 fcntl (udpv4_fd, F_SETFD, FD_CLOEXEC);
91 147
92 // standard daemon practise... 148 // standard daemon practise...
93 { 149 {
94 int oval = 1; 150 int oval = 1;
95 setsockopt (udpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval); 151 setsockopt (udpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval);
96 } 152 }
97 153
98 sockinfo si (THISNODE, PROT_UDPv4); 154#if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
99
100 if (bind (udpv4_fd, si.sav4 (), si.salenv4 ()))
101 {
102 slog (L_ERR, _("can't bind udpv4 on %s: %s"), (const char *)si, strerror (errno));
103 exit (1);
104 }
105
106#ifdef IP_MTU_DISCOVER
107 // this I really consider a linux bug. I am neither connected 155 // this I really consider a linux bug. I am neither connected
108 // nor do I fragment myself. Linux still sets DF and doesn't 156 // nor do I fragment myself. Linux still sets DF and doesn't
109 // fragment for me sometimes. 157 // fragment for me sometimes.
110 { 158 {
111 int oval = IP_PMTUDISC_DONT; 159 int oval = IP_PMTUDISC_DONT;
112 setsockopt (udpv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval); 160 setsockopt (udpv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval);
113 } 161 }
114#endif 162#endif
115 163
164 sockinfo si (THISNODE, PROT_UDPv4);
165
166 if (bind (udpv4_fd, si.sav4 (), si.salenv4 ()))
167 {
168 slog (L_ERR, _("can't bind udpv4 on %s: %s"), (const char *)si, strerror (errno));
169 exit (EXIT_FAILURE);
170 }
171
116 udpv4_ev_watcher.start (udpv4_fd, POLLIN); 172 udpv4_ev_watcher.start (udpv4_fd, EV_READ);
117 } 173 }
118 174
119 ipv4_fd = -1; 175 icmpv4_fd = -1;
176
177#if ENABLE_ICMP
120 if (THISNODE->protocols & PROT_IPv4) 178 if (THISNODE->protocols & PROT_ICMPv4)
121 { 179 {
122 ipv4_fd = socket (PF_INET, SOCK_RAW, ::conf.ip_proto); 180 icmpv4_fd = socket (PF_INET, SOCK_RAW, IPPROTO_ICMP);
123 181
124 if (ipv4_fd < 0) 182 if (icmpv4_fd < 0)
125 return -1; 183 return -1;
126 184
127 sockinfo si (THISNODE, PROT_IPv4); 185 fcntl (icmpv4_fd, F_SETFL, O_NONBLOCK);
186 fcntl (icmpv4_fd, F_SETFD, FD_CLOEXEC);
128 187
129 if (bind (ipv4_fd, si.sav4 (), si.salenv4 ())) 188#ifdef ICMP_FILTER
130 { 189 {
131 slog (L_ERR, _("can't bind ipv4 socket on %s: %s"), (const char *)si, strerror (errno)); 190 icmp_filter oval;
132 exit (1); 191 oval.data = 0xffffffff;
192 if (::conf.icmp_type < 32)
193 oval.data &= ~(1 << ::conf.icmp_type);
194
195 setsockopt (icmpv4_fd, SOL_RAW, ICMP_FILTER, &oval, sizeof oval);
133 } 196 }
197#endif
134 198
135#ifdef IP_MTU_DISCOVER 199#if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
136 // this I really consider a linux bug. I am neither connected 200 // this I really consider a linux bug. I am neither connected
137 // nor do I fragment myself. Linux still sets DF and doesn't 201 // nor do I fragment myself. Linux still sets DF and doesn't
138 // fragment for me sometimes. 202 // fragment for me sometimes.
139 { 203 {
140 int oval = IP_PMTUDISC_DONT; 204 int oval = IP_PMTUDISC_DONT;
141 setsockopt (ipv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval); 205 setsockopt (icmpv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval);
142 } 206 }
143#endif 207#endif
144 208
209 sockinfo si (THISNODE, PROT_ICMPv4);
210
211 if (bind (icmpv4_fd, si.sav4 (), si.salenv4 ()))
212 {
213 slog (L_ERR, _("can't bind icmpv4 on %s: %s"), (const char *)si, strerror (errno));
214 exit (EXIT_FAILURE);
215 }
216
145 ipv4_ev_watcher.start (ipv4_fd, POLLIN); 217 icmpv4_ev_watcher.start (icmpv4_fd, EV_READ);
146 } 218 }
219#endif
220
221 tcpv4_fd = -1;
147 222
148#if ENABLE_TCP 223#if ENABLE_TCP
149 if (THISNODE->protocols & PROT_TCPv4) 224 if (THISNODE->protocols & PROT_TCPv4 && THISNODE->tcp_port)
150 { 225 {
151 tcpv4_fd = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP); 226 tcpv4_fd = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP);
152 227
153 if (tcpv4_fd < 0) 228 if (tcpv4_fd < 0)
154 return -1; 229 return -1;
230
231 fcntl (tcpv4_fd, F_SETFL, O_NONBLOCK);
232 fcntl (tcpv4_fd, F_SETFD, FD_CLOEXEC);
155 233
156 // standard daemon practise... 234 // standard daemon practise...
157 { 235 {
158 int oval = 1; 236 int oval = 1;
159 setsockopt (tcpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval); 237 setsockopt (tcpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval);
162 sockinfo si (THISNODE, PROT_TCPv4); 240 sockinfo si (THISNODE, PROT_TCPv4);
163 241
164 if (bind (tcpv4_fd, si.sav4 (), si.salenv4 ())) 242 if (bind (tcpv4_fd, si.sav4 (), si.salenv4 ()))
165 { 243 {
166 slog (L_ERR, _("can't bind tcpv4 on %s: %s"), (const char *)si, strerror (errno)); 244 slog (L_ERR, _("can't bind tcpv4 on %s: %s"), (const char *)si, strerror (errno));
167 exit (1); 245 exit (EXIT_FAILURE);
168 } 246 }
169 247
170 if (listen (tcpv4_fd, 5)) 248 if (listen (tcpv4_fd, 5))
171 { 249 {
172 slog (L_ERR, _("can't listen tcpv4 on %s: %s"), (const char *)si, strerror (errno)); 250 slog (L_ERR, _("can't listen tcpv4 on %s: %s"), (const char *)si, strerror (errno));
173 exit (1); 251 exit (EXIT_FAILURE);
252 }
253
254 tcpv4_ev_watcher.start (tcpv4_fd, EV_READ);
255 }
256#endif
257
258#if ENABLE_DNS
259 if (THISNODE->protocols & PROT_DNSv4)
260 {
261 dns_forwarder.set (::conf.dns_forw_host, ::conf.dns_forw_port, PROT_DNSv4);
262
263 dnsv4_fd = socket (PF_INET, SOCK_DGRAM, IPPROTO_UDP);
264
265 if (dnsv4_fd < 0)
266 return -1;
267
268 fcntl (dnsv4_fd, F_SETFL, O_NONBLOCK);
269 fcntl (dnsv4_fd, F_SETFD, FD_CLOEXEC);
270
271# if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
272 // this I really consider a linux bug. I am neither connected
273 // nor do I fragment myself. Linux still sets DF and doesn't
274 // fragment for me sometimes.
275 {
276 int oval = IP_PMTUDISC_DONT;
277 setsockopt (dnsv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval);
174 } 278 }
175
176 tcpv4_ev_watcher.start (tcpv4_fd, POLLIN);
177 }
178#endif 279# endif
280
281 // standard daemon practise...
282 {
283 int oval = 1;
284 setsockopt (dnsv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval);
285 }
286
287 sockinfo si (THISNODE->dns_hostname,
288 THISNODE->dns_hostname ? THISNODE->dns_port : 0,
289 PROT_DNSv4);
290
291 if (bind (dnsv4_fd, si.sav4 (), si.salenv4 ()))
292 {
293 slog (L_ERR, _("can't bind dnsv4 on %s: %s"), (const char *)si, strerror (errno));
294 exit (EXIT_FAILURE);
295 }
296
297 dnsv4_ev_watcher.start (dnsv4_fd, EV_READ);
298 }
299#endif
300
301 /////////////////////////////////////////////////////////////////////////////
302
303 reconnect_all ();
304
305 /////////////////////////////////////////////////////////////////////////////
179 306
180 tap = new tap_device (); 307 tap = new tap_device ();
181 if (!tap) //D this, of course, never catches 308 if (!tap) //D this, of course, never catches
182 { 309 {
183 slog (L_ERR, _("cannot create network interface '%s'"), conf.ifname); 310 slog (L_ERR, _("cannot create network interface '%s'"), conf.ifname);
184 exit (1); 311 exit (EXIT_FAILURE);
185 } 312 }
186 313
314 fcntl (tap->fd, F_SETFD, FD_CLOEXEC);
315
316 if (tap->if_up () &&
317 !run_script (run_script_cb (this, &vpn::script_if_init), true))
318 {
319 slog (L_ERR, _("interface initialization command '%s' failed, exiting."),
320 tap->if_up ());
321 exit (EXIT_FAILURE);
322 }
323
187 run_script (run_script_cb (this, &vpn::script_if_up), true); 324 if (!run_script (run_script_cb (this, &vpn::script_if_up), true))
325 {
326 slog (L_ERR, _("if-up command execution failed, exiting."));
327 exit (EXIT_FAILURE);
328 }
188 329
189 tap_ev_watcher.start (tap->fd, POLLIN); 330 tap_ev_watcher.start (tap->fd, EV_READ);
190
191 reconnect_all ();
192 331
193 return 0; 332 return 0;
194} 333}
195 334
196void 335bool
197vpn::send_ipv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) 336vpn::send_ipv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
198{ 337{
338#if defined(SOL_IP) && defined(IP_TOS)
199 setsockopt (ipv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos); 339 setsockopt (ipv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
340#endif
200 sendto (ipv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ()); 341 sendto (ipv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ());
201}
202 342
203void 343 return true;
344}
345
346static u16
347ipv4_checksum (u16 *data, unsigned int len)
348{
349 // use 32 bit accumulator and fold back carry bits at the end
350 u32 sum = 0;
351
352 while (len > 1)
353 {
354 sum += *data++;
355 len -= 2;
356 }
357
358 // odd byte left?
359 if (len)
360 sum += *(u8 *)data;
361
362 // add back carry bits
363 sum = (sum >> 16) + (sum & 0xffff); // lo += hi
364 sum += (sum >> 16); // carry
365
366 return ~sum;
367}
368
369#if ENABLE_ICMP
370bool
371vpn::send_icmpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
372{
373#if defined(SOL_IP) && defined(IP_TOS)
374 setsockopt (icmpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
375#endif
376
377 pkt->unshift_hdr (4);
378
379 icmp_header *hdr = (icmp_header *)&((*pkt)[0]);
380 hdr->type = ::conf.icmp_type;
381 hdr->code = 255;
382 hdr->checksum = 0;
383 hdr->checksum = ipv4_checksum ((u16 *)hdr, pkt->len);
384
385 sendto (icmpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ());
386
387 return true;
388}
389#endif
390
391bool
204vpn::send_udpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) 392vpn::send_udpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
205{ 393{
394#if defined(SOL_IP) && defined(IP_TOS)
206 setsockopt (udpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos); 395 setsockopt (udpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
396#endif
207 sendto (udpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ()); 397 sendto (udpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ());
398
399 return true;
400}
401
402void
403vpn::inject_data_packet (tap_packet *pkt, int dst)
404{
405 if (dst)
406 {
407 // unicast
408 if (dst != THISNODE->id)
409 conns[dst - 1]->inject_data_packet (pkt);
410 }
411 else
412 {
413 // broadcast, this is ugly, but due to the security policy
414 // we have to connect to all hosts...
415 for (conns_vector::iterator c = conns.begin (); c != conns.end (); ++c)
416 if ((*c)->conf != THISNODE)
417 (*c)->inject_data_packet (pkt, true);
418 }
208} 419}
209 420
210void 421void
211vpn::recv_vpn_packet (vpn_packet *pkt, const sockinfo &rsi) 422vpn::recv_vpn_packet (vpn_packet *pkt, const sockinfo &rsi)
212{ 423{
219 if (src == 0 || src > conns.size () 430 if (src == 0 || src > conns.size ()
220 || dst > conns.size () 431 || dst > conns.size ()
221 || pkt->typ () >= vpn_packet::PT_MAX) 432 || pkt->typ () >= vpn_packet::PT_MAX)
222 slog (L_WARN, _("(%s): received corrupted packet type %d (src %d, dst %d)"), 433 slog (L_WARN, _("(%s): received corrupted packet type %d (src %d, dst %d)"),
223 (const char *)rsi, pkt->typ (), pkt->src (), pkt->dst ()); 434 (const char *)rsi, pkt->typ (), pkt->src (), pkt->dst ());
435 else if (dst > conns.size ())
436 slog (L_WARN, _("(%s): received corrupted packet type %d (src %d, dst %d)"),
437 (const char *)rsi, pkt->typ (), pkt->src (), pkt->dst ());
224 else 438 else
225 { 439 {
226 connection *c = conns[src - 1]; 440 connection *c = conns[src - 1];
227 441
228 if (dst == 0 && !THISNODE->routerprio) 442 if (dst == 0)
229 slog (L_WARN, _("%s(%s): received broadcast, but we are no router"), 443 slog (L_WARN, _("%s(%s): received broadcast (protocol violation)"),
230 c->conf->nodename, (const char *)rsi); 444 c->conf->nodename, (const char *)rsi);
231 else if (dst != 0 && dst != THISNODE->id) 445 else if (dst != THISNODE->id)
232 // FORWARDING NEEDED ;) 446 {
447 if (THISNODE->routerprio)
448 // the tos setting gets lost here. who cares.
449 conns[dst - 1]->inject_vpn_packet (pkt);
450 else
233 slog (L_WARN, 451 slog (L_WARN,
234 _("received frame for node %d ('%s') from %s, but this is node %d ('%s')"), 452 _("%s(%s): forwarding request (=> %s), but we are no router"),
453 c->conf->nodename, (const char *)rsi,
235 dst, conns[dst - 1]->conf->nodename, 454 conns[dst - 1]->conf->nodename);
236 (const char *)rsi, 455 }
237 THISNODE->id, THISNODE->nodename);
238 else 456 else
239 c->recv_vpn_packet (pkt, rsi); 457 c->recv_vpn_packet (pkt, rsi);
240 } 458 }
241} 459}
242 460
243void 461bool
244vpn::udpv4_ev (io_watcher &w, short revents) 462vpn::send_vpn_packet (vpn_packet *pkt, const sockinfo &si, int tos)
245{ 463{
246 if (revents & (POLLIN | POLLERR)) 464 switch (si.prot)
465 {
466 case PROT_IPv4:
467 return send_ipv4_packet (pkt, si, tos);
468 case PROT_UDPv4:
469 return send_udpv4_packet (pkt, si, tos);
470#if ENABLE_TCP
471 case PROT_TCPv4:
472 return send_tcpv4_packet (pkt, si, tos);
473#endif
474#if ENABLE_ICMP
475 case PROT_ICMPv4:
476 return send_icmpv4_packet (pkt, si, tos);
477#endif
478#if ENABLE_DNS
479 case PROT_DNSv4:
480 return send_dnsv4_packet (pkt, si, tos);
481#endif
482
483 default:
484 slog (L_CRIT, _("%s: FATAL: trying to send packet with unsupported protocol"), (const char *)si);
485 }
486
487 return false;
488}
489
490void
491vpn::ipv4_ev (ev::io &w, int revents)
492{
493 if (revents & EV_READ)
247 { 494 {
248 vpn_packet *pkt = new vpn_packet; 495 vpn_packet *pkt = new vpn_packet;
249 struct sockaddr_in sa; 496 struct sockaddr_in sa;
250 socklen_t sa_len = sizeof (sa); 497 socklen_t sa_len = sizeof (sa);
251 int len; 498 int len;
252 499
253 len = recvfrom (w.p->fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len); 500 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len);
254 501
255 sockinfo si(sa); 502 sockinfo si(sa, PROT_IPv4);
256 503
257 if (len > 0) 504 if (len > 0)
258 { 505 {
259 pkt->len = len; 506 pkt->len = len;
507
508 // raw sockets deliver the ipv4 header, but don't expect it on sends
509 pkt->skip_hdr (IP_OVERHEAD);
260 510
261 recv_vpn_packet (pkt, si); 511 recv_vpn_packet (pkt, si);
262 } 512 }
263 else 513 else
264 { 514 {
266 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno)); 516 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno));
267 } 517 }
268 518
269 delete pkt; 519 delete pkt;
270 } 520 }
271 else if (revents & POLLHUP)
272 {
273 // this cannot ;) happen on udp sockets
274 slog (L_ERR, _("FATAL: POLLHUP on udp v4 fd, terminating."));
275 exit (1);
276 }
277 else 521 else
278 { 522 {
279 slog (L_ERR, 523 slog (L_ERR,
280 _("FATAL: unknown revents %08x in socket, terminating\n"), 524 _("FATAL: unknown revents %08x in socket, terminating\n"),
281 revents); 525 revents);
282 exit (1); 526 exit (EXIT_FAILURE);
283 } 527 }
284} 528}
285 529
530#if ENABLE_ICMP
286void 531void
287vpn::ipv4_ev (io_watcher &w, short revents) 532vpn::icmpv4_ev (ev::io &w, int revents)
288{ 533{
289 if (revents & (POLLIN | POLLERR)) 534 if (revents & EV_READ)
290 { 535 {
291 vpn_packet *pkt = new vpn_packet; 536 vpn_packet *pkt = new vpn_packet;
292 struct sockaddr_in sa; 537 struct sockaddr_in sa;
293 socklen_t sa_len = sizeof (sa); 538 socklen_t sa_len = sizeof (sa);
294 int len; 539 int len;
295 540
296 len = recvfrom (w.p->fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len); 541 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len);
297 542
298 sockinfo si(sa, PROT_IPv4); 543 sockinfo si(sa, PROT_ICMPv4);
299 544
300 if (len > 0) 545 if (len > 0)
301 { 546 {
302 pkt->len = len; 547 pkt->len = len;
303 548
549 icmp_header *hdr = (icmp_header *)&((*pkt)[IP_OVERHEAD]);
550
551 if (hdr->type == ::conf.icmp_type
552 && hdr->code == 255)
553 {
304 // raw sockets deliver the ipv4, but don't expect it on sends 554 // raw sockets deliver the ipv4, but don't expect it on sends
305 // this is slow, but... 555 // this is slow, but...
306 pkt->skip_hdr (IP_OVERHEAD); 556 pkt->skip_hdr (ICMP_OVERHEAD);
307 557
308 recv_vpn_packet (pkt, si); 558 recv_vpn_packet (pkt, si);
559 }
309 } 560 }
310 else 561 else
311 { 562 {
312 // probably ECONNRESET or somesuch 563 // probably ECONNRESET or somesuch
313 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno)); 564 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno));
314 } 565 }
315 566
316 delete pkt; 567 delete pkt;
317 }
318 else if (revents & POLLHUP)
319 {
320 // this cannot ;) happen on udp sockets
321 slog (L_ERR, _("FATAL: POLLHUP on ipv4 fd, terminating."));
322 exit (1);
323 } 568 }
324 else 569 else
325 { 570 {
326 slog (L_ERR, 571 slog (L_ERR,
327 _("FATAL: unknown revents %08x in socket, terminating\n"), 572 _("FATAL: unknown revents %08x in socket, terminating\n"),
328 revents); 573 revents);
329 exit (1); 574 exit (EXIT_FAILURE);
330 }
331}
332
333#if ENABLE_TCP
334
335struct tcp_connection;
336
337struct lt_sockinfo
338{
339 bool operator()(const sockinfo *a, const sockinfo *b) const
340 {
341 return *a < *b;
342 }
343};
344
345struct tcp_si_map : public map<const sockinfo *, tcp_connection *, lt_sockinfo> {
346 void cleaner_cb (time_watcher &w); time_watcher cleaner;
347
348 tcp_si_map ()
349 : cleaner(this, &tcp_si_map::cleaner_cb)
350 {
351 cleaner.start (0);
352 }
353} tcp_si;
354
355struct tcp_connection : io_watcher {
356 tstamp last_activity;
357 const sockinfo si;
358 vpn &v;
359 bool ok;
360
361 vpn_packet *r_pkt;
362 u32 r_len, r_ofs;
363
364 void tcpv4_ev (io_watcher &w, short revents);
365
366 operator tcp_si_map::value_type()
367 {
368 return tcp_si_map::value_type (&si, this);
369 }
370
371 tcp_connection (int fd, const sockinfo &si_, vpn &v_)
372 : v(v_), si(si_), io_watcher(this, &tcp_connection::tcpv4_ev)
373 {
374 last_activity = NOW;
375 ok = false;
376 r_pkt = 0;
377 start (fd, POLLOUT);
378 }
379
380 ~tcp_connection () { if (p) close (p->fd); }
381};
382
383void tcp_si_map::cleaner_cb (time_watcher &w)
384{
385 w.at = NOW + 600;
386 tstamp to = NOW - ::conf.keepalive - 30;
387
388 for (iterator i = begin (); i != end(); )
389 if (i->second->last_activity >= to)
390 ++i;
391 else
392 {
393 erase (i);
394 i = begin ();
395 } 575 }
396} 576}
577#endif
397 578
398void 579void
399vpn::send_tcpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) 580vpn::udpv4_ev (ev::io &w, int revents)
400{ 581{
401 tcp_si_map::iterator info = tcp_si.find (&si); 582 if (revents & EV_READ)
402
403 if (info == tcp_si.end ())
404 {
405 // woaw, the first lost packet ;)
406 int fd = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP);
407
408 if (fd >= 0)
409 {
410 fcntl (fd, F_SETFL, O_NONBLOCK);
411
412 if (connect (fd, si.sav4 (), si.salenv4 ()) >= 0
413 || errno == EINPROGRESS)
414 {
415 tcp_connection *i = new tcp_connection (fd, si, *this);
416
417 tcp_si.insert (*i);
418 }
419 else
420 close (fd);
421 }
422 } 583 {
423 else
424 {
425 tcp_connection *i = info->second;
426
427 i->last_activity = NOW;
428
429 if (i->ok)
430 {
431 setsockopt (i->p->fd, SOL_IP, IP_TOS, &tos, sizeof tos);
432
433 // we use none of the advantages of tcp
434 write (i->p->fd, (void *)pkt, pkt->len + sizeof (u32)) != pkt->len + sizeof (u32);
435 }
436 }
437
438#if 0
439 setsockopt (udpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
440 sendto (udpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ());
441#endif
442}
443
444void
445tcp_connection::tcpv4_ev (io_watcher &w, short revents)
446{
447 last_activity = NOW;
448
449 if (!ok) // just established?
450 {
451 ok = true;
452 set (POLLIN);
453 }
454
455 if (revents & (POLLIN | POLLERR))
456 {
457 u32 len;
458
459 if (sizeof (len) == read (p->fd, &len, sizeof (len)))
460 {
461 vpn_packet *pkt = new vpn_packet; 584 vpn_packet *pkt = new vpn_packet;
462
463 if (len == read (p->fd, &((*pkt)[0]), len))
464 {
465 pkt->len = len;
466
467 v.recv_vpn_packet (pkt, si);
468 return;
469 }
470
471 delete pkt;
472 }
473
474 tcp_si.erase (&si);
475
476 set (0);//D
477 }
478}
479
480void
481vpn::tcpv4_ev (io_watcher &w, short revents)
482{
483 if (revents & (POLLIN | POLLERR))
484 {
485 struct sockaddr_in sa; 585 struct sockaddr_in sa;
486 socklen_t sa_len = sizeof (sa); 586 socklen_t sa_len = sizeof (sa);
487 int len; 587 int len;
488 588
489 int fd = accept (w.p->fd, (sockaddr *)&sa, &sa_len); 589 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len);
490 590
491 if (fd >= 0)
492 {
493 fcntl (fd, F_SETFL, O_NONBLOCK);
494
495 sockinfo si(sa, PROT_TCPv4); 591 sockinfo si(sa, PROT_UDPv4);
496 tcp_connection *i = new tcp_connection (fd, si, *this);
497 592
498 slog (L_DEBUG, _("accepted tcp connection from %s\n"), (const char *)si);//D 593 if (len > 0)
499
500 tcp_si.insert (*i);
501 } 594 {
595 pkt->len = len;
596
597 recv_vpn_packet (pkt, si);
598 }
599 else
600 {
601 // probably ECONNRESET or somesuch
602 slog (L_DEBUG, _("%s: fd %d, %s"), (const char *)si, w.fd, strerror (errno));
603 }
604
605 delete pkt;
606 }
607 else
502 } 608 {
609 slog (L_ERR,
610 _("FATAL: unknown revents %08x in socket, terminating\n"),
611 revents);
612 exit (EXIT_FAILURE);
613 }
503} 614}
504 615
505#endif
506
507void 616void
508vpn::tap_ev (io_watcher &w, short revents) 617vpn::tap_ev (ev::io &w, int revents)
509{ 618{
510 if (revents & POLLIN) 619 if (revents & EV_READ)
511 { 620 {
512 /* process data */ 621 /* process data */
513 tap_packet *pkt; 622 tap_packet *pkt;
514 623
515 pkt = tap->recv (); 624 pkt = tap->recv ();
516 625
626 if (!pkt)
627 return;
628
629 if (pkt->len > 14)
630 {
517 int dst = mac2id (pkt->dst); 631 int dst = mac2id (pkt->dst);
518 int src = mac2id (pkt->src); 632 int src = mac2id (pkt->src);
519 633
520 if (src != THISNODE->id) 634 if (src != THISNODE->id)
521 {
522 slog (L_ERR, _("FATAL: tap packet not originating on current node received, terminating."));
523 exit (1);
524 }
525
526 if (dst == THISNODE->id)
527 {
528 slog (L_ERR, _("FATAL: tap packet destined for current node received, terminating."));
529 exit (1);
530 }
531
532 if (dst > conns.size ())
533 slog (L_ERR, _("tap packet for unknown node %d received, ignoring."), dst);
534 else
535 {
536 if (dst)
537 { 635 {
538 // unicast 636 slog (L_ERR, _("FATAL: tap packet not originating on current node received (if-up script not working properly?), exiting."));
539 if (dst != THISNODE->id) 637 exit (EXIT_FAILURE);
540 conns[dst - 1]->inject_data_packet (pkt);
541 } 638 }
639
640 if (dst == THISNODE->id)
641 {
642 slog (L_ERR, _("FATAL: tap packet destined for current node received, exiting."));
643 exit (EXIT_FAILURE);
644 }
645
646 if (dst > conns.size ())
647 slog (L_ERR, _("tap packet for unknown node %d received, ignoring."), dst);
542 else 648 else
543 {
544 // broadcast, first check router, then self, then english
545 connection *router = find_router ();
546
547 if (router)
548 router->inject_data_packet (pkt, true); 649 inject_data_packet (pkt, dst);
549 else
550 for (conns_vector::iterator c = conns.begin (); c != conns.end (); ++c)
551 if ((*c)->conf != THISNODE)
552 (*c)->inject_data_packet (pkt);
553 }
554 } 650 }
555 651
556 delete pkt; 652 delete pkt;
557 }
558 else if (revents & (POLLHUP | POLLERR))
559 {
560 slog (L_ERR, _("FATAL: POLLHUP or POLLERR on network device fd, terminating."));
561 exit (1);
562 } 653 }
563 else 654 else
564 abort (); 655 abort ();
565} 656}
566 657
567void 658void
568vpn::event_cb (time_watcher &w) 659vpn::event_cb (ev::timer &w, int)
569{ 660{
570 if (events) 661 if (events)
571 { 662 {
572 if (events & EVENT_SHUTDOWN) 663 if (events & EVENT_SHUTDOWN)
573 { 664 {
574 slog (L_INFO, _("preparing shutdown...")); 665 slog (L_INFO, _("preparing shutdown..."));
575 666
576 shutdown_all (); 667 shutdown_all ();
577
578 remove_pid (pidfilename); 668 remove_pid (conf.pidfilename);
579
580 slog (L_INFO, _("terminating")); 669 slog (L_INFO, _("terminating"));
581 670 exit (EXIT_SUCCESS);
582 exit (0);
583 } 671 }
584 672
585 if (events & EVENT_RECONNECT) 673 if (events & EVENT_RECONNECT)
586 { 674 {
587 slog (L_INFO, _("forced reconnect")); 675 slog (L_INFO, _("forced reconnect"));
589 reconnect_all (); 677 reconnect_all ();
590 } 678 }
591 679
592 events = 0; 680 events = 0;
593 } 681 }
594
595 w.at = TSTAMP_CANCEL;
596} 682}
597 683
598void 684void
599vpn::shutdown_all () 685vpn::shutdown_all ()
600{ 686{
613 connection_init (); 699 connection_init ();
614 700
615 for (configuration::node_vector::iterator i = conf.nodes.begin (); 701 for (configuration::node_vector::iterator i = conf.nodes.begin ();
616 i != conf.nodes.end (); ++i) 702 i != conf.nodes.end (); ++i)
617 { 703 {
618 connection *conn = new connection (this); 704 connection *conn = new connection (this, *i);
619
620 conn->conf = *i;
621 conns.push_back (conn); 705 conns.push_back (conn);
622
623 conn->establish_connection (); 706 conn->establish_connection ();
624 } 707 }
625} 708}
626 709
627connection *vpn::find_router () 710connection *vpn::find_router ()
628{ 711{
629 u32 prio = 0; 712 u32 prio = 1;
630 connection *router = 0; 713 connection *router = 0;
631 714
632 for (conns_vector::iterator i = conns.begin (); i != conns.end (); ++i) 715 for (conns_vector::iterator i = conns.begin (); i != conns.end (); ++i)
633 { 716 {
634 connection *c = *i; 717 connection *c = *i;
635 718
636 if (c->conf->routerprio > prio 719 if (c->conf->routerprio > prio
637 && c->connectmode == conf_node::C_ALWAYS 720 && c->connectmode == conf_node::C_ALWAYS // so we don't drop the connection if in use
638 && c->conf != THISNODE
639 && c->ictx && c->octx) 721 && c->ictx && c->octx
722 && c->conf != THISNODE) // redundant, since ictx==octx==0 always on thisnode
640 { 723 {
641 prio = c->conf->routerprio; 724 prio = c->conf->routerprio;
642 router = c; 725 router = c;
643 } 726 }
644 } 727 }
645 728
646 return router; 729 return router;
647} 730}
648 731
649void vpn::connect_request (int id) 732void vpn::send_connect_request (int id)
650{ 733{
651 connection *c = find_router (); 734 connection *c = find_router ();
652 735
653 if (c) 736 if (c)
654 c->connect_request (id); 737 c->send_connect_request (id);
655 //else // does not work, because all others must connect to the same router 738 else
656 // // no router found, aggressively connect to all routers 739 // no router found, aggressively connect to all routers
657 // for (conns_vector::iterator i = conns.begin (); i != conns.end (); ++i) 740 for (conns_vector::iterator i = conns.begin (); i != conns.end (); ++i)
658 // if ((*i)->conf->routerprio) 741 if ((*i)->conf->routerprio && (*i)->conf != THISNODE)
659 // (*i)->establish_connection (); 742 (*i)->establish_connection ();
660} 743}
661 744
662void 745void
663connection::dump_status () 746connection::dump_status ()
664{ 747{
672} 755}
673 756
674void 757void
675vpn::dump_status () 758vpn::dump_status ()
676{ 759{
677 slog (L_NOTICE, _("BEGIN status dump (%ld)"), (long)NOW); 760 slog (L_NOTICE, _("BEGIN status dump (%ld)"), (long)ev::ev_now ());
678 761
679 for (conns_vector::iterator c = conns.begin (); c != conns.end (); ++c) 762 for (conns_vector::iterator c = conns.begin (); c != conns.end (); ++c)
680 (*c)->dump_status (); 763 (*c)->dump_status ();
681 764
682 slog (L_NOTICE, _("END status dump")); 765 slog (L_NOTICE, _("END status dump"));
683} 766}
684 767
685vpn::vpn (void) 768vpn::vpn (void)
686: event(this, &vpn::event_cb) 769: event (this, &vpn::event_cb)
687, udpv4_ev_watcher(this, &vpn::udpv4_ev) 770, udpv4_ev_watcher (this, &vpn::udpv4_ev)
688, ipv4_ev_watcher (this, &vpn::ipv4_ev) 771, ipv4_ev_watcher (this, &vpn::ipv4_ev)
689, tap_ev_watcher (this, &vpn::tap_ev)
690#if ENABLE_TCP 772#if ENABLE_TCP
691, tcpv4_ev_watcher(this, &vpn::tcpv4_ev) 773, tcpv4_ev_watcher (this, &vpn::tcpv4_ev)
692#endif 774#endif
775#if ENABLE_ICMP
776, icmpv4_ev_watcher(this, &vpn::icmpv4_ev)
777#endif
778#if ENABLE_DNS
779, dnsv4_ev_watcher (this, &vpn::dnsv4_ev)
780#endif
781, tap_ev_watcher (this, &vpn::tap_ev)
693{ 782{
694} 783}
695 784
696vpn::~vpn () 785vpn::~vpn ()
697{ 786{

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines