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.13 by pcg, Sun Apr 13 16:53:36 2003 UTC vs.
Revision 1.31 by pcg, Sat Mar 5 19:13:16 2005 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. 59 Temple Place, Suite 330, Boston, MA 02111-1307 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 <errno.h> 33#include <errno.h>
32#include <time.h> 34#include <time.h>
33#include <unistd.h> 35#include <unistd.h>
34#include <fcntl.h> 36#include <fcntl.h>
35#include <arpa/inet.h> 37#include <sys/socket.h>
36#include <netinet/in.h> 38
37#include <netinet/ip_icmp.h> 39#include "netcompat.h"
38 40
39#include "pidfile.h" 41#include "pidfile.h"
40 42
41#include "connection.h" 43#include "connection.h"
42#include "util.h" 44#include "util.h"
43#include "vpn.h" 45#include "vpn.h"
46
47vpn network; // THE vpn (bad design...)
44 48
45///////////////////////////////////////////////////////////////////////////// 49/////////////////////////////////////////////////////////////////////////////
46 50
47const char *vpn::script_if_up () 51const char *vpn::script_if_up ()
48{ 52{
52 mtu += ETH_OVERHEAD - 6 - 6; // now we have the data portion 56 mtu += ETH_OVERHEAD - 6 - 6; // now we have the data portion
53 mtu -= mtu % EVP_CIPHER_block_size (CIPHER); // round 57 mtu -= mtu % EVP_CIPHER_block_size (CIPHER); // round
54 mtu -= ETH_OVERHEAD - 6 - 6; // and get interface mtu again 58 mtu -= ETH_OVERHEAD - 6 - 6; // and get interface mtu again
55 59
56 char *env; 60 char *env;
57 asprintf (&env, "CONFBASE=%s", confbase); 61 asprintf (&env, "CONFBASE=%s", confbase); putenv (env);
58 putenv (env);
59 asprintf (&env, "NODENAME=%s", THISNODE->nodename); 62 asprintf (&env, "NODENAME=%s", THISNODE->nodename); putenv (env);
60 putenv (env);
61 asprintf (&env, "NODEID=%d", THISNODE->id); 63 asprintf (&env, "NODEID=%d", THISNODE->id); putenv (env);
62 putenv (env);
63 asprintf (&env, "IFNAME=%s", tap->interface ()); 64 asprintf (&env, "IFNAME=%s", tap->interface ()); putenv (env);
64 putenv (env); 65 asprintf (&env, "IFTYPE=%s", IFTYPE); putenv (env);
66 asprintf (&env, "IFSUBTYPE=%s", IFSUBTYPE); putenv (env);
65 asprintf (&env, "MTU=%d", mtu); 67 asprintf (&env, "MTU=%d", mtu); putenv (env);
66 putenv (env);
67 asprintf (&env, "MAC=%02x:%02x:%02x:%02x:%02x:%02x", 68 asprintf (&env, "MAC=%02x:%02x:%02x:%02x:%02x:%02x",
68 0xfe, 0xfd, 0x80, 0x00, THISNODE->id >> 8, 69 0xfe, 0xfd, 0x80, 0x00, THISNODE->id >> 8,
69 THISNODE->id & 0xff); 70 THISNODE->id & 0xff);
70 putenv (env); 71 putenv (env);
71 72
84 if (ipv4_fd < 0) 85 if (ipv4_fd < 0)
85 return -1; 86 return -1;
86 87
87 fcntl (ipv4_fd, F_SETFL, O_NONBLOCK); 88 fcntl (ipv4_fd, F_SETFL, O_NONBLOCK);
88 89
89#ifdef IP_MTU_DISCOVER 90#if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
90 // this I really consider a linux bug. I am neither connected 91 // this I really consider a linux bug. I am neither connected
91 // nor do I fragment myself. Linux still sets DF and doesn't 92 // nor do I fragment myself. Linux still sets DF and doesn't
92 // fragment for me sometimes. 93 // fragment for me sometimes.
93 { 94 {
94 int oval = IP_PMTUDISC_DONT; 95 int oval = IP_PMTUDISC_DONT;
99 sockinfo si (THISNODE, PROT_IPv4); 100 sockinfo si (THISNODE, PROT_IPv4);
100 101
101 if (bind (ipv4_fd, si.sav4 (), si.salenv4 ())) 102 if (bind (ipv4_fd, si.sav4 (), si.salenv4 ()))
102 { 103 {
103 slog (L_ERR, _("can't bind ipv4 socket on %s: %s"), (const char *)si, strerror (errno)); 104 slog (L_ERR, _("can't bind ipv4 socket on %s: %s"), (const char *)si, strerror (errno));
104 exit (1); 105 exit (EXIT_FAILURE);
105 } 106 }
106 107
107 ipv4_ev_watcher.start (ipv4_fd, POLLIN); 108 ipv4_ev_watcher.start (ipv4_fd, EVENT_READ);
108 } 109 }
109 110
110 udpv4_fd = -1; 111 udpv4_fd = -1;
111 112
112 if (THISNODE->protocols & PROT_UDPv4 && THISNODE->udp_port) 113 if (THISNODE->protocols & PROT_UDPv4 && THISNODE->udp_port)
122 { 123 {
123 int oval = 1; 124 int oval = 1;
124 setsockopt (udpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval); 125 setsockopt (udpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval);
125 } 126 }
126 127
127#ifdef IP_MTU_DISCOVER 128#if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
128 // this I really consider a linux bug. I am neither connected 129 // this I really consider a linux bug. I am neither connected
129 // nor do I fragment myself. Linux still sets DF and doesn't 130 // nor do I fragment myself. Linux still sets DF and doesn't
130 // fragment for me sometimes. 131 // fragment for me sometimes.
131 { 132 {
132 int oval = IP_PMTUDISC_DONT; 133 int oval = IP_PMTUDISC_DONT;
137 sockinfo si (THISNODE, PROT_UDPv4); 138 sockinfo si (THISNODE, PROT_UDPv4);
138 139
139 if (bind (udpv4_fd, si.sav4 (), si.salenv4 ())) 140 if (bind (udpv4_fd, si.sav4 (), si.salenv4 ()))
140 { 141 {
141 slog (L_ERR, _("can't bind udpv4 on %s: %s"), (const char *)si, strerror (errno)); 142 slog (L_ERR, _("can't bind udpv4 on %s: %s"), (const char *)si, strerror (errno));
142 exit (1); 143 exit (EXIT_FAILURE);
143 } 144 }
144 145
145 udpv4_ev_watcher.start (udpv4_fd, POLLIN); 146 udpv4_ev_watcher.start (udpv4_fd, EVENT_READ);
146 } 147 }
147 148
148 icmpv4_fd = -1; 149 icmpv4_fd = -1;
149 150
150#if ENABLE_ICMP 151#if ENABLE_ICMP
166 167
167 setsockopt (icmpv4_fd, SOL_RAW, ICMP_FILTER, &oval, sizeof oval); 168 setsockopt (icmpv4_fd, SOL_RAW, ICMP_FILTER, &oval, sizeof oval);
168 } 169 }
169#endif 170#endif
170 171
171#ifdef IP_MTU_DISCOVER 172#if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
172 // this I really consider a linux bug. I am neither connected 173 // this I really consider a linux bug. I am neither connected
173 // nor do I fragment myself. Linux still sets DF and doesn't 174 // nor do I fragment myself. Linux still sets DF and doesn't
174 // fragment for me sometimes. 175 // fragment for me sometimes.
175 { 176 {
176 int oval = IP_PMTUDISC_DONT; 177 int oval = IP_PMTUDISC_DONT;
181 sockinfo si (THISNODE, PROT_ICMPv4); 182 sockinfo si (THISNODE, PROT_ICMPv4);
182 183
183 if (bind (icmpv4_fd, si.sav4 (), si.salenv4 ())) 184 if (bind (icmpv4_fd, si.sav4 (), si.salenv4 ()))
184 { 185 {
185 slog (L_ERR, _("can't bind icmpv4 on %s: %s"), (const char *)si, strerror (errno)); 186 slog (L_ERR, _("can't bind icmpv4 on %s: %s"), (const char *)si, strerror (errno));
186 exit (1); 187 exit (EXIT_FAILURE);
187 } 188 }
188 189
189 icmpv4_ev_watcher.start (icmpv4_fd, POLLIN); 190 icmpv4_ev_watcher.start (icmpv4_fd, EVENT_READ);
190 } 191 }
191#endif 192#endif
192 193
193 tcpv4_fd = -1; 194 tcpv4_fd = -1;
194 195
211 sockinfo si (THISNODE, PROT_TCPv4); 212 sockinfo si (THISNODE, PROT_TCPv4);
212 213
213 if (bind (tcpv4_fd, si.sav4 (), si.salenv4 ())) 214 if (bind (tcpv4_fd, si.sav4 (), si.salenv4 ()))
214 { 215 {
215 slog (L_ERR, _("can't bind tcpv4 on %s: %s"), (const char *)si, strerror (errno)); 216 slog (L_ERR, _("can't bind tcpv4 on %s: %s"), (const char *)si, strerror (errno));
216 exit (1); 217 exit (EXIT_FAILURE);
217 } 218 }
218 219
219 if (listen (tcpv4_fd, 5)) 220 if (listen (tcpv4_fd, 5))
220 { 221 {
221 slog (L_ERR, _("can't listen tcpv4 on %s: %s"), (const char *)si, strerror (errno)); 222 slog (L_ERR, _("can't listen tcpv4 on %s: %s"), (const char *)si, strerror (errno));
222 exit (1); 223 exit (EXIT_FAILURE);
224 }
225
226 tcpv4_ev_watcher.start (tcpv4_fd, EVENT_READ);
227 }
228#endif
229
230#if ENABLE_DNS
231 if (THISNODE->protocols & PROT_DNSv4)
232 {
233 dns_forwarder.set (::conf.dns_forw_host, ::conf.dns_forw_port, PROT_DNSv4);
234
235 dnsv4_fd = socket (PF_INET, SOCK_DGRAM, IPPROTO_UDP);
236
237 if (dnsv4_fd < 0)
238 return -1;
239
240 // standard daemon practise...
241 {
242 int oval = 1;
243 setsockopt (dnsv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval);
244 }
245
246 sockinfo si (THISNODE->dns_hostname,
247 THISNODE->dns_hostname ? THISNODE->dns_port : 0,
248 PROT_DNSv4);
249
250 if (bind (dnsv4_fd, si.sav4 (), si.salenv4 ()))
223 } 251 {
252 slog (L_ERR, _("can't bind dnsv4 on %s: %s"), (const char *)si, strerror (errno));
253 exit (EXIT_FAILURE);
254 }
224 255
225 tcpv4_ev_watcher.start (tcpv4_fd, POLLIN); 256 dnsv4_ev_watcher.start (dnsv4_fd, EVENT_READ);
226 } 257 }
227#endif 258#endif
228 259
229 tap = new tap_device (); 260 tap = new tap_device ();
230 if (!tap) //D this, of course, never catches 261 if (!tap) //D this, of course, never catches
231 { 262 {
232 slog (L_ERR, _("cannot create network interface '%s'"), conf.ifname); 263 slog (L_ERR, _("cannot create network interface '%s'"), conf.ifname);
233 exit (1); 264 exit (EXIT_FAILURE);
234 } 265 }
235 266
236 run_script (run_script_cb (this, &vpn::script_if_up), true); 267 run_script (run_script_cb (this, &vpn::script_if_up), true);
237 268
238 tap_ev_watcher.start (tap->fd, POLLIN); 269 tap_ev_watcher.start (tap->fd, EVENT_READ);
239 270
240 reconnect_all (); 271 reconnect_all ();
241 272
242 return 0; 273 return 0;
243}
244
245// send a vpn packet out to other hosts
246bool
247vpn::send_vpn_packet (vpn_packet *pkt, const sockinfo &si, int tos)
248{
249 switch (si.prot)
250 {
251 case PROT_IPv4:
252 return send_ipv4_packet (pkt, si, tos);
253
254 case PROT_UDPv4:
255 return send_udpv4_packet (pkt, si, tos);
256
257#if ENABLE_TCP
258 case PROT_TCPv4:
259 return send_tcpv4_packet (pkt, si, tos);
260#endif
261
262#if ENABLE_ICMP
263 case PROT_ICMPv4:
264 return send_icmpv4_packet (pkt, si, tos);
265#endif
266
267 default:
268 slog (L_CRIT, _("%s: FATAL: trying to send packet with unsupported protocol"), (const char *)si);
269 return false;
270 }
271} 274}
272 275
273bool 276bool
274vpn::send_ipv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) 277vpn::send_ipv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
275{ 278{
279#if defined(SOL_IP) && defined(IP_TOS)
276 setsockopt (ipv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos); 280 setsockopt (ipv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
281#endif
277 sendto (ipv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ()); 282 sendto (ipv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ());
278 283
279 return true; 284 return true;
280} 285}
281 286
304 309
305#if ENABLE_ICMP 310#if ENABLE_ICMP
306bool 311bool
307vpn::send_icmpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) 312vpn::send_icmpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
308{ 313{
314#if defined(SOL_IP) && defined(IP_TOS)
309 setsockopt (icmpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos); 315 setsockopt (icmpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
316#endif
310 317
311 pkt->unshift_hdr (4); 318 pkt->unshift_hdr (4);
312 319
313 icmphdr *hdr = (icmphdr *)&((*pkt)[0]); 320 icmp_header *hdr = (icmp_header *)&((*pkt)[0]);
314 hdr->type = ::conf.icmp_type; 321 hdr->type = ::conf.icmp_type;
315 hdr->code = 255; 322 hdr->code = 255;
316 hdr->checksum = 0; 323 hdr->checksum = 0;
317 hdr->checksum = ipv4_checksum ((u16 *)hdr, pkt->len); 324 hdr->checksum = ipv4_checksum ((u16 *)hdr, pkt->len);
318 325
323#endif 330#endif
324 331
325bool 332bool
326vpn::send_udpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) 333vpn::send_udpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
327{ 334{
335#if defined(SOL_IP) && defined(IP_TOS)
328 setsockopt (udpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos); 336 setsockopt (udpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
337#endif
329 sendto (udpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ()); 338 sendto (udpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ());
330 339
331 return true; 340 return true;
341}
342
343void
344vpn::inject_data_packet (tap_packet *pkt, int dst)
345{
346 if (dst)
347 {
348 // unicast
349 if (dst != THISNODE->id)
350 conns[dst - 1]->inject_data_packet (pkt);
351 }
352 else
353 {
354 // broadcast, this is ugly, but due to the security policy
355 // we have to connect to all hosts...
356 for (conns_vector::iterator c = conns.begin (); c != conns.end (); ++c)
357 if ((*c)->conf != THISNODE)
358 (*c)->inject_data_packet (pkt, true);
359 }
332} 360}
333 361
334void 362void
335vpn::recv_vpn_packet (vpn_packet *pkt, const sockinfo &rsi) 363vpn::recv_vpn_packet (vpn_packet *pkt, const sockinfo &rsi)
336{ 364{
353 connection *c = conns[src - 1]; 381 connection *c = conns[src - 1];
354 382
355 if (dst == 0) 383 if (dst == 0)
356 slog (L_WARN, _("%s(%s): received broadcast (protocol violation)"), 384 slog (L_WARN, _("%s(%s): received broadcast (protocol violation)"),
357 c->conf->nodename, (const char *)rsi); 385 c->conf->nodename, (const char *)rsi);
358 else if (dst != 0 && dst != THISNODE->id) 386 else if (dst != THISNODE->id)
359 { 387 {
360 if (THISNODE->routerprio) 388 if (THISNODE->routerprio)
361 // the tos setting gets lost here. who cares. 389 // the tos setting gets lost here. who cares.
362 conns[dst - 1]->inject_vpn_packet (pkt); 390 conns[dst - 1]->inject_vpn_packet (pkt);
363 else 391 else
369 else 397 else
370 c->recv_vpn_packet (pkt, rsi); 398 c->recv_vpn_packet (pkt, rsi);
371 } 399 }
372} 400}
373 401
402bool
403vpn::send_vpn_packet (vpn_packet *pkt, const sockinfo &si, int tos)
404{
405 switch (si.prot)
406 {
407 case PROT_IPv4:
408 return send_ipv4_packet (pkt, si, tos);
409 case PROT_UDPv4:
410 return send_udpv4_packet (pkt, si, tos);
411#if ENABLE_TCP
412 case PROT_TCPv4:
413 return send_tcpv4_packet (pkt, si, tos);
414#endif
415#if ENABLE_ICMP
416 case PROT_ICMPv4:
417 return send_icmpv4_packet (pkt, si, tos);
418#endif
419#if ENABLE_DNS
420 case PROT_DNSv4:
421 return send_dnsv4_packet (pkt, si, tos);
422#endif
423
424 default:
425 slog (L_CRIT, _("%s: FATAL: trying to send packet with unsupported protocol"), (const char *)si);
426 }
427
428 return false;
429}
430
374void 431void
375vpn::ipv4_ev (io_watcher &w, short revents) 432vpn::ipv4_ev (io_watcher &w, short revents)
376{ 433{
377 if (revents & (POLLIN | POLLERR)) 434 if (revents & EVENT_READ)
378 { 435 {
379 vpn_packet *pkt = new vpn_packet; 436 vpn_packet *pkt = new vpn_packet;
380 struct sockaddr_in sa; 437 struct sockaddr_in sa;
381 socklen_t sa_len = sizeof (sa); 438 socklen_t sa_len = sizeof (sa);
382 int len; 439 int len;
387 444
388 if (len > 0) 445 if (len > 0)
389 { 446 {
390 pkt->len = len; 447 pkt->len = len;
391 448
392 // raw sockets deliver the ipv4, but don't expect it on sends 449 // raw sockets deliver the ipv4 header, but don't expect it on sends
393 // this is slow, but...
394 pkt->skip_hdr (IP_OVERHEAD); 450 pkt->skip_hdr (IP_OVERHEAD);
395 451
396 recv_vpn_packet (pkt, si); 452 recv_vpn_packet (pkt, si);
397 } 453 }
398 else 454 else
401 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno)); 457 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno));
402 } 458 }
403 459
404 delete pkt; 460 delete pkt;
405 } 461 }
406 else if (revents & POLLHUP)
407 {
408 // this cannot ;) happen on udp sockets
409 slog (L_ERR, _("FATAL: POLLHUP on ipv4 fd, terminating."));
410 exit (1);
411 }
412 else 462 else
413 { 463 {
414 slog (L_ERR, 464 slog (L_ERR,
415 _("FATAL: unknown revents %08x in socket, terminating\n"), 465 _("FATAL: unknown revents %08x in socket, terminating\n"),
416 revents); 466 revents);
417 exit (1); 467 exit (EXIT_FAILURE);
418 } 468 }
419} 469}
420 470
421#if ENABLE_ICMP 471#if ENABLE_ICMP
422void 472void
423vpn::icmpv4_ev (io_watcher &w, short revents) 473vpn::icmpv4_ev (io_watcher &w, short revents)
424{ 474{
425 if (revents & (POLLIN | POLLERR)) 475 if (revents & EVENT_READ)
426 { 476 {
427 vpn_packet *pkt = new vpn_packet; 477 vpn_packet *pkt = new vpn_packet;
428 struct sockaddr_in sa; 478 struct sockaddr_in sa;
429 socklen_t sa_len = sizeof (sa); 479 socklen_t sa_len = sizeof (sa);
430 int len; 480 int len;
435 485
436 if (len > 0) 486 if (len > 0)
437 { 487 {
438 pkt->len = len; 488 pkt->len = len;
439 489
440 icmphdr *hdr = (icmphdr *)&((*pkt)[IP_OVERHEAD]); 490 icmp_header *hdr = (icmp_header *)&((*pkt)[IP_OVERHEAD]);
441 491
442 if (hdr->type == ::conf.icmp_type 492 if (hdr->type == ::conf.icmp_type
443 && hdr->code == 255) 493 && hdr->code == 255)
444 { 494 {
445 // raw sockets deliver the ipv4, but don't expect it on sends 495 // raw sockets deliver the ipv4, but don't expect it on sends
455 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno)); 505 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno));
456 } 506 }
457 507
458 delete pkt; 508 delete pkt;
459 } 509 }
460 else if (revents & POLLHUP)
461 {
462 // this cannot ;) happen on udp sockets
463 slog (L_ERR, _("FATAL: POLLHUP on icmpv4 fd, terminating."));
464 exit (1);
465 }
466 else 510 else
467 { 511 {
468 slog (L_ERR, 512 slog (L_ERR,
469 _("FATAL: unknown revents %08x in socket, terminating\n"), 513 _("FATAL: unknown revents %08x in socket, terminating\n"),
470 revents); 514 revents);
471 exit (1); 515 exit (EXIT_FAILURE);
472 } 516 }
473} 517}
474#endif 518#endif
475 519
476void 520void
477vpn::udpv4_ev (io_watcher &w, short revents) 521vpn::udpv4_ev (io_watcher &w, short revents)
478{ 522{
479 if (revents & (POLLIN | POLLERR)) 523 if (revents & EVENT_READ)
480 { 524 {
481 vpn_packet *pkt = new vpn_packet; 525 vpn_packet *pkt = new vpn_packet;
482 struct sockaddr_in sa; 526 struct sockaddr_in sa;
483 socklen_t sa_len = sizeof (sa); 527 socklen_t sa_len = sizeof (sa);
484 int len; 528 int len;
499 slog (L_DEBUG, _("%s: fd %d, %s"), (const char *)si, w.fd, strerror (errno)); 543 slog (L_DEBUG, _("%s: fd %d, %s"), (const char *)si, w.fd, strerror (errno));
500 } 544 }
501 545
502 delete pkt; 546 delete pkt;
503 } 547 }
504 else if (revents & POLLHUP)
505 {
506 // this cannot ;) happen on udp sockets
507 slog (L_ERR, _("FATAL: POLLHUP on udp v4 fd, terminating."));
508 exit (1);
509 }
510 else 548 else
511 { 549 {
512 slog (L_ERR, 550 slog (L_ERR,
513 _("FATAL: unknown revents %08x in socket, terminating\n"), 551 _("FATAL: unknown revents %08x in socket, terminating\n"),
514 revents); 552 revents);
515 exit (1); 553 exit (EXIT_FAILURE);
516 } 554 }
517} 555}
518 556
519void 557void
520vpn::tap_ev (io_watcher &w, short revents) 558vpn::tap_ev (io_watcher &w, short revents)
521{ 559{
522 if (revents & POLLIN) 560 if (revents & EVENT_READ)
523 { 561 {
524 /* process data */ 562 /* process data */
525 tap_packet *pkt; 563 tap_packet *pkt;
526 564
527 pkt = tap->recv (); 565 pkt = tap->recv ();
528 566
567 if (!pkt)
568 return;
569
570 if (pkt->len > 14)
571 {
529 int dst = mac2id (pkt->dst); 572 int dst = mac2id (pkt->dst);
530 int src = mac2id (pkt->src); 573 int src = mac2id (pkt->src);
531 574
532 if (src != THISNODE->id) 575 if (src != THISNODE->id)
533 {
534 slog (L_ERR, _("FATAL: tap packet not originating on current node received, terminating."));
535 exit (1);
536 }
537
538 if (dst == THISNODE->id)
539 {
540 slog (L_ERR, _("FATAL: tap packet destined for current node received, terminating."));
541 exit (1);
542 }
543
544 if (dst > conns.size ())
545 slog (L_ERR, _("tap packet for unknown node %d received, ignoring."), dst);
546 else
547 {
548 if (dst)
549 { 576 {
550 // unicast 577 slog (L_ERR, _("FATAL: tap packet not originating on current node received (if-up script not working properly?), exiting."));
551 if (dst != THISNODE->id) 578 exit (EXIT_FAILURE);
552 conns[dst - 1]->inject_data_packet (pkt);
553 } 579 }
580
581 if (dst == THISNODE->id)
582 {
583 slog (L_ERR, _("FATAL: tap packet destined for current node received, exiting."));
584 exit (EXIT_FAILURE);
585 }
586
587 if (dst > conns.size ())
588 slog (L_ERR, _("tap packet for unknown node %d received, ignoring."), dst);
554 else 589 else
555 {
556 // broadcast, this is ugly, but due to the security policy
557 // we have to connect to all hosts...
558 for (conns_vector::iterator c = conns.begin (); c != conns.end (); ++c)
559 if ((*c)->conf != THISNODE)
560 (*c)->inject_data_packet (pkt); 590 inject_data_packet (pkt, dst);
561 }
562 } 591 }
563 592
564 delete pkt; 593 delete pkt;
565 }
566 else if (revents & (POLLHUP | POLLERR))
567 {
568 slog (L_ERR, _("FATAL: POLLHUP or POLLERR on network device fd, terminating."));
569 exit (1);
570 } 594 }
571 else 595 else
572 abort (); 596 abort ();
573} 597}
574 598
580 if (events & EVENT_SHUTDOWN) 604 if (events & EVENT_SHUTDOWN)
581 { 605 {
582 slog (L_INFO, _("preparing shutdown...")); 606 slog (L_INFO, _("preparing shutdown..."));
583 607
584 shutdown_all (); 608 shutdown_all ();
585
586 remove_pid (pidfilename); 609 remove_pid (conf.pidfilename);
587
588 slog (L_INFO, _("terminating")); 610 slog (L_INFO, _("terminating"));
589 611 exit (EXIT_SUCCESS);
590 exit (0);
591 } 612 }
592 613
593 if (events & EVENT_RECONNECT) 614 if (events & EVENT_RECONNECT)
594 { 615 {
595 slog (L_INFO, _("forced reconnect")); 616 slog (L_INFO, _("forced reconnect"));
597 reconnect_all (); 618 reconnect_all ();
598 } 619 }
599 620
600 events = 0; 621 events = 0;
601 } 622 }
602
603 w.at = TSTAMP_CANCEL;
604} 623}
605 624
606void 625void
607vpn::shutdown_all () 626vpn::shutdown_all ()
608{ 627{
621 connection_init (); 640 connection_init ();
622 641
623 for (configuration::node_vector::iterator i = conf.nodes.begin (); 642 for (configuration::node_vector::iterator i = conf.nodes.begin ();
624 i != conf.nodes.end (); ++i) 643 i != conf.nodes.end (); ++i)
625 { 644 {
626 connection *conn = new connection (this); 645 connection *conn = new connection (this, *i);
627
628 conn->conf = *i;
629 conns.push_back (conn); 646 conns.push_back (conn);
630
631 conn->establish_connection (); 647 conn->establish_connection ();
632 } 648 }
633} 649}
634 650
635connection *vpn::find_router () 651connection *vpn::find_router ()
698, tcpv4_ev_watcher (this, &vpn::tcpv4_ev) 714, tcpv4_ev_watcher (this, &vpn::tcpv4_ev)
699#endif 715#endif
700#if ENABLE_ICMP 716#if ENABLE_ICMP
701, icmpv4_ev_watcher(this, &vpn::icmpv4_ev) 717, icmpv4_ev_watcher(this, &vpn::icmpv4_ev)
702#endif 718#endif
719#if ENABLE_DNS
720, dnsv4_ev_watcher (this, &vpn::dnsv4_ev)
721#endif
703, tap_ev_watcher (this, &vpn::tap_ev) 722, tap_ev_watcher (this, &vpn::tap_ev)
704{ 723{
705} 724}
706 725
707vpn::~vpn () 726vpn::~vpn ()

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines