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.8 by pcg, Mon Apr 7 01:28:56 2003 UTC vs.
Revision 1.26 by pcg, Sun Jul 25 18:11:54 2004 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-2004 Marc Lehmann <pcg@goof.com>
3 4
4 This program is free software; you can redistribute it and/or modify 5 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by 6 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 7 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version. 8 (at your option) any later version.
18 19
19#include "config.h" 20#include "config.h"
20 21
21#include <list> 22#include <list>
22 23
24#include <cstdio>
25#include <cstring>
23#include <cstdlib> 26#include <cstdlib>
24#include <cstring>
25#include <cstdio>
26 27
27#include <sys/types.h> 28#include <sys/types.h>
28#include <sys/socket.h> 29#include <sys/socket.h>
29#include <sys/poll.h>
30#include <sys/wait.h> 30#include <sys/wait.h>
31#include <netinet/in.h>
32#include <arpa/inet.h>
33#include <errno.h> 31#include <errno.h>
34#include <time.h> 32#include <time.h>
35#include <unistd.h> 33#include <unistd.h>
34#include <fcntl.h>
35#include <sys/socket.h>
36
37#include "netcompat.h"
36 38
37#include "pidfile.h" 39#include "pidfile.h"
38 40
39#include "connection.h" 41#include "connection.h"
40#include "util.h" 42#include "util.h"
41#include "vpn.h" 43#include "vpn.h"
44
45vpn network; // THE vpn (bad design...)
42 46
43///////////////////////////////////////////////////////////////////////////// 47/////////////////////////////////////////////////////////////////////////////
44 48
45const char *vpn::script_if_up () 49const char *vpn::script_if_up ()
46{ 50{
50 mtu += ETH_OVERHEAD - 6 - 6; // now we have the data portion 54 mtu += ETH_OVERHEAD - 6 - 6; // now we have the data portion
51 mtu -= mtu % EVP_CIPHER_block_size (CIPHER); // round 55 mtu -= mtu % EVP_CIPHER_block_size (CIPHER); // round
52 mtu -= ETH_OVERHEAD - 6 - 6; // and get interface mtu again 56 mtu -= ETH_OVERHEAD - 6 - 6; // and get interface mtu again
53 57
54 char *env; 58 char *env;
55 asprintf (&env, "CONFBASE=%s", confbase); 59 asprintf (&env, "CONFBASE=%s", confbase); putenv (env);
56 putenv (env);
57 asprintf (&env, "NODENAME=%s", THISNODE->nodename); 60 asprintf (&env, "NODENAME=%s", THISNODE->nodename); putenv (env);
58 putenv (env);
59 asprintf (&env, "NODEID=%d", THISNODE->id); 61 asprintf (&env, "NODEID=%d", THISNODE->id); putenv (env);
60 putenv (env);
61 asprintf (&env, "IFNAME=%s", tap->interface ()); 62 asprintf (&env, "IFNAME=%s", tap->interface ()); putenv (env);
62 putenv (env); 63 asprintf (&env, "IFTYPE=%s", IFTYPE); putenv (env);
64 asprintf (&env, "IFSUBTYPE=%s", IFSUBTYPE); putenv (env);
63 asprintf (&env, "MTU=%d", mtu); 65 asprintf (&env, "MTU=%d", mtu); putenv (env);
64 putenv (env);
65 asprintf (&env, "MAC=%02x:%02x:%02x:%02x:%02x:%02x", 66 asprintf (&env, "MAC=%02x:%02x:%02x:%02x:%02x:%02x",
66 0xfe, 0xfd, 0x80, 0x00, THISNODE->id >> 8, 67 0xfe, 0xfd, 0x80, 0x00, THISNODE->id >> 8,
67 THISNODE->id & 0xff); 68 THISNODE->id & 0xff);
68 putenv (env); 69 putenv (env);
69 70
80 ipv4_fd = socket (PF_INET, SOCK_RAW, ::conf.ip_proto); 81 ipv4_fd = socket (PF_INET, SOCK_RAW, ::conf.ip_proto);
81 82
82 if (ipv4_fd < 0) 83 if (ipv4_fd < 0)
83 return -1; 84 return -1;
84 85
85 sockinfo si (THISNODE, PROT_IPv4); 86 fcntl (ipv4_fd, F_SETFL, O_NONBLOCK);
86 87
87 if (bind (ipv4_fd, si.sav4 (), si.salenv4 ())) 88#if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
88 {
89 slog (L_ERR, _("can't bind ipv4 socket on %s: %s"), (const char *)si, strerror (errno));
90 exit (1);
91 }
92
93#ifdef IP_MTU_DISCOVER
94 // this I really consider a linux bug. I am neither connected 89 // this I really consider a linux bug. I am neither connected
95 // nor do I fragment myself. Linux still sets DF and doesn't 90 // nor do I fragment myself. Linux still sets DF and doesn't
96 // fragment for me sometimes. 91 // fragment for me sometimes.
97 { 92 {
98 int oval = IP_PMTUDISC_DONT; 93 int oval = IP_PMTUDISC_DONT;
99 setsockopt (ipv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval); 94 setsockopt (ipv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval);
100 } 95 }
101#endif 96#endif
102 97
98 sockinfo si (THISNODE, PROT_IPv4);
99
100 if (bind (ipv4_fd, si.sav4 (), si.salenv4 ()))
101 {
102 slog (L_ERR, _("can't bind ipv4 socket on %s: %s"), (const char *)si, strerror (errno));
103 exit (EXIT_FAILURE);
104 }
105
103 ipv4_ev_watcher.start (ipv4_fd, POLLIN); 106 ipv4_ev_watcher.start (ipv4_fd, EVENT_READ);
104 } 107 }
105 108
106 udpv4_fd = -1; 109 udpv4_fd = -1;
107 110
108 if (THISNODE->protocols & PROT_UDPv4 && THISNODE->udp_port) 111 if (THISNODE->protocols & PROT_UDPv4 && THISNODE->udp_port)
109 { 112 {
110 udpv4_fd = socket (PF_INET, SOCK_DGRAM, IPPROTO_UDP); 113 udpv4_fd = socket (PF_INET, SOCK_DGRAM, IPPROTO_UDP);
111 114
112 if (udpv4_fd < 0) 115 if (udpv4_fd < 0)
113 return -1; 116 return -1;
117
118 fcntl (udpv4_fd, F_SETFL, O_NONBLOCK);
114 119
115 // standard daemon practise... 120 // standard daemon practise...
116 { 121 {
117 int oval = 1; 122 int oval = 1;
118 setsockopt (udpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval); 123 setsockopt (udpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval);
119 } 124 }
120 125
121 sockinfo si (THISNODE, PROT_UDPv4); 126#if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
122
123 if (bind (udpv4_fd, si.sav4 (), si.salenv4 ()))
124 {
125 slog (L_ERR, _("can't bind udpv4 on %s: %s"), (const char *)si, strerror (errno));
126 exit (1);
127 }
128
129#ifdef IP_MTU_DISCOVER
130 // this I really consider a linux bug. I am neither connected 127 // this I really consider a linux bug. I am neither connected
131 // nor do I fragment myself. Linux still sets DF and doesn't 128 // nor do I fragment myself. Linux still sets DF and doesn't
132 // fragment for me sometimes. 129 // fragment for me sometimes.
133 { 130 {
134 int oval = IP_PMTUDISC_DONT; 131 int oval = IP_PMTUDISC_DONT;
135 setsockopt (udpv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval); 132 setsockopt (udpv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval);
136 } 133 }
137#endif 134#endif
138 135
136 sockinfo si (THISNODE, PROT_UDPv4);
137
138 if (bind (udpv4_fd, si.sav4 (), si.salenv4 ()))
139 {
140 slog (L_ERR, _("can't bind udpv4 on %s: %s"), (const char *)si, strerror (errno));
141 exit (EXIT_FAILURE);
142 }
143
139 udpv4_ev_watcher.start (udpv4_fd, POLLIN); 144 udpv4_ev_watcher.start (udpv4_fd, EVENT_READ);
145 }
146
147 icmpv4_fd = -1;
148
149#if ENABLE_ICMP
150 if (THISNODE->protocols & PROT_ICMPv4)
151 {
152 icmpv4_fd = socket (PF_INET, SOCK_RAW, IPPROTO_ICMP);
153
154 if (icmpv4_fd < 0)
155 return -1;
156
157 fcntl (icmpv4_fd, F_SETFL, O_NONBLOCK);
158
159#ifdef ICMP_FILTER
160 {
161 icmp_filter oval;
162 oval.data = 0xffffffff;
163 if (::conf.icmp_type < 32)
164 oval.data &= ~(1 << ::conf.icmp_type);
165
166 setsockopt (icmpv4_fd, SOL_RAW, ICMP_FILTER, &oval, sizeof oval);
140 } 167 }
168#endif
169
170#if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
171 // this I really consider a linux bug. I am neither connected
172 // nor do I fragment myself. Linux still sets DF and doesn't
173 // fragment for me sometimes.
174 {
175 int oval = IP_PMTUDISC_DONT;
176 setsockopt (udpv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval);
177 }
178#endif
179
180 sockinfo si (THISNODE, PROT_ICMPv4);
181
182 if (bind (icmpv4_fd, si.sav4 (), si.salenv4 ()))
183 {
184 slog (L_ERR, _("can't bind icmpv4 on %s: %s"), (const char *)si, strerror (errno));
185 exit (EXIT_FAILURE);
186 }
187
188 icmpv4_ev_watcher.start (icmpv4_fd, EVENT_READ);
189 }
190#endif
141 191
142 tcpv4_fd = -1; 192 tcpv4_fd = -1;
143 193
144#if ENABLE_TCP 194#if ENABLE_TCP
145 if (THISNODE->protocols & PROT_TCPv4 && THISNODE->tcp_port) 195 if (THISNODE->protocols & PROT_TCPv4 && THISNODE->tcp_port)
146 { 196 {
147 tcpv4_fd = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP); 197 tcpv4_fd = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP);
148 198
149 if (tcpv4_fd < 0) 199 if (tcpv4_fd < 0)
150 return -1; 200 return -1;
201
202 fcntl (tcpv4_fd, F_SETFL, O_NONBLOCK);
151 203
152 // standard daemon practise... 204 // standard daemon practise...
153 { 205 {
154 int oval = 1; 206 int oval = 1;
155 setsockopt (tcpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval); 207 setsockopt (tcpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval);
158 sockinfo si (THISNODE, PROT_TCPv4); 210 sockinfo si (THISNODE, PROT_TCPv4);
159 211
160 if (bind (tcpv4_fd, si.sav4 (), si.salenv4 ())) 212 if (bind (tcpv4_fd, si.sav4 (), si.salenv4 ()))
161 { 213 {
162 slog (L_ERR, _("can't bind tcpv4 on %s: %s"), (const char *)si, strerror (errno)); 214 slog (L_ERR, _("can't bind tcpv4 on %s: %s"), (const char *)si, strerror (errno));
163 exit (1); 215 exit (EXIT_FAILURE);
164 } 216 }
165 217
166 if (listen (tcpv4_fd, 5)) 218 if (listen (tcpv4_fd, 5))
167 { 219 {
168 slog (L_ERR, _("can't listen tcpv4 on %s: %s"), (const char *)si, strerror (errno)); 220 slog (L_ERR, _("can't listen tcpv4 on %s: %s"), (const char *)si, strerror (errno));
169 exit (1); 221 exit (EXIT_FAILURE);
222 }
223
224 tcpv4_ev_watcher.start (tcpv4_fd, EVENT_READ);
225 }
226#endif
227
228#if ENABLE_DNS
229 if (THISNODE->protocols & PROT_DNSv4)
230 {
231 dnsv4_fd = socket (PF_INET, SOCK_DGRAM, IPPROTO_UDP);
232
233 if (dnsv4_fd < 0)
234 return -1;
235
236 // standard daemon practise...
237 {
238 int oval = 1;
239 setsockopt (tcpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval);
240 }
241
242 sockinfo si (THISNODE, PROT_DNSv4);
243
244 if (bind (dnsv4_fd, si.sav4 (), si.salenv4 ()))
170 } 245 {
246 slog (L_ERR, _("can't bind dnsv4 on %s: %s"), (const char *)si, strerror (errno));
247 exit (EXIT_FAILURE);
248 }
171 249
172 tcpv4_ev_watcher.start (tcpv4_fd, POLLIN); 250 dnsv4_ev_watcher.start (dnsv4_fd, EVENT_READ);
173 } 251 }
174#endif 252#endif
175 253
176 tap = new tap_device (); 254 tap = new tap_device ();
177 if (!tap) //D this, of course, never catches 255 if (!tap) //D this, of course, never catches
178 { 256 {
179 slog (L_ERR, _("cannot create network interface '%s'"), conf.ifname); 257 slog (L_ERR, _("cannot create network interface '%s'"), conf.ifname);
180 exit (1); 258 exit (EXIT_FAILURE);
181 } 259 }
182 260
183 run_script (run_script_cb (this, &vpn::script_if_up), true); 261 run_script (run_script_cb (this, &vpn::script_if_up), true);
184 262
185 tap_ev_watcher.start (tap->fd, POLLIN); 263 tap_ev_watcher.start (tap->fd, EVENT_READ);
186 264
187 reconnect_all (); 265 reconnect_all ();
188 266
189 return 0; 267 return 0;
190} 268}
204#if ENABLE_TCP 282#if ENABLE_TCP
205 case PROT_TCPv4: 283 case PROT_TCPv4:
206 return send_tcpv4_packet (pkt, si, tos); 284 return send_tcpv4_packet (pkt, si, tos);
207#endif 285#endif
208 286
287#if ENABLE_ICMP
288 case PROT_ICMPv4:
289 return send_icmpv4_packet (pkt, si, tos);
290#endif
291
292#if ENABLE_DNS
293 case PROT_DNSv4:
294 return send_dnsv4_packet (pkt, si, tos);
295#endif
296
209 default: 297 default:
210 slog (L_CRIT, _("%s: FATAL: trying to send packet with unsupported protocol"), (const char *)si); 298 slog (L_CRIT, _("%s: FATAL: trying to send packet with unsupported protocol"), (const char *)si);
211 return false; 299 return false;
212 } 300 }
213} 301}
214 302
215bool 303bool
216vpn::send_ipv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) 304vpn::send_ipv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
217{ 305{
306#if defined(SOL_IP) && defined(IP_TOS)
218 setsockopt (ipv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos); 307 setsockopt (ipv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
308#endif
219 sendto (ipv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ()); 309 sendto (ipv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ());
220 310
221 return true; 311 return true;
222} 312}
313
314static u16
315ipv4_checksum (u16 *data, unsigned int len)
316{
317 // use 32 bit accumulator and fold back carry bits at the end
318 u32 sum = 0;
319
320 while (len > 1)
321 {
322 sum += *data++;
323 len -= 2;
324 }
325
326 // odd byte left?
327 if (len)
328 sum += *(u8 *)data;
329
330 // add back carry bits
331 sum = (sum >> 16) + (sum & 0xffff); // lo += hi
332 sum += (sum >> 16); // carry
333
334 return ~sum;
335}
336
337#if ENABLE_ICMP
338bool
339vpn::send_icmpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
340{
341#if defined(SOL_IP) && defined(IP_TOS)
342 setsockopt (icmpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
343#endif
344
345 pkt->unshift_hdr (4);
346
347 icmp_header *hdr = (icmp_header *)&((*pkt)[0]);
348 hdr->type = ::conf.icmp_type;
349 hdr->code = 255;
350 hdr->checksum = 0;
351 hdr->checksum = ipv4_checksum ((u16 *)hdr, pkt->len);
352
353 sendto (icmpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ());
354
355 return true;
356}
357#endif
223 358
224bool 359bool
225vpn::send_udpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) 360vpn::send_udpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
226{ 361{
362#if defined(SOL_IP) && defined(IP_TOS)
227 setsockopt (udpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos); 363 setsockopt (udpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
364#endif
228 sendto (udpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ()); 365 sendto (udpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ());
229 366
230 return true; 367 return true;
368}
369
370void
371vpn::inject_data_packet (tap_packet *pkt, int dst)
372{
373 if (dst)
374 {
375 // unicast
376 if (dst != THISNODE->id)
377 conns[dst - 1]->inject_data_packet (pkt);
378 }
379 else
380 {
381 // broadcast, this is ugly, but due to the security policy
382 // we have to connect to all hosts...
383 for (conns_vector::iterator c = conns.begin (); c != conns.end (); ++c)
384 if ((*c)->conf != THISNODE)
385 (*c)->inject_data_packet (pkt, true);
386 }
231} 387}
232 388
233void 389void
234vpn::recv_vpn_packet (vpn_packet *pkt, const sockinfo &rsi) 390vpn::recv_vpn_packet (vpn_packet *pkt, const sockinfo &rsi)
235{ 391{
249 (const char *)rsi, pkt->typ (), pkt->src (), pkt->dst ()); 405 (const char *)rsi, pkt->typ (), pkt->src (), pkt->dst ());
250 else 406 else
251 { 407 {
252 connection *c = conns[src - 1]; 408 connection *c = conns[src - 1];
253 409
254 if (dst == 0 && !THISNODE->routerprio) 410 if (dst == 0)
255 slog (L_WARN, _("%s(%s): received broadcast, but we are no router"), 411 slog (L_WARN, _("%s(%s): received broadcast (protocol violation)"),
256 c->conf->nodename, (const char *)rsi); 412 c->conf->nodename, (const char *)rsi);
257 else if (dst != 0 && dst != THISNODE->id) 413 else if (dst != THISNODE->id)
258 { 414 {
259 if (THISNODE->routerprio) 415 if (THISNODE->routerprio)
260 // the tos setting gets lost here. who cares. 416 // the tos setting gets lost here. who cares.
261 conns[dst - 1]->inject_vpn_packet (pkt); 417 conns[dst - 1]->inject_vpn_packet (pkt);
262 else 418 else
269 c->recv_vpn_packet (pkt, rsi); 425 c->recv_vpn_packet (pkt, rsi);
270 } 426 }
271} 427}
272 428
273void 429void
274vpn::udpv4_ev (io_watcher &w, short revents) 430vpn::ipv4_ev (io_watcher &w, short revents)
275{ 431{
276 if (revents & (POLLIN | POLLERR)) 432 if (revents & EVENT_READ)
277 { 433 {
278 vpn_packet *pkt = new vpn_packet; 434 vpn_packet *pkt = new vpn_packet;
279 struct sockaddr_in sa; 435 struct sockaddr_in sa;
280 socklen_t sa_len = sizeof (sa); 436 socklen_t sa_len = sizeof (sa);
281 int len; 437 int len;
282 438
283 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len); 439 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len);
284 440
285 sockinfo si(sa, PROT_UDPv4); 441 sockinfo si(sa, PROT_IPv4);
286 442
287 if (len > 0) 443 if (len > 0)
288 { 444 {
289 pkt->len = len; 445 pkt->len = len;
290 446
447 // raw sockets deliver the ipv4, but don't expect it on sends
448 // this is slow, but...
449 pkt->skip_hdr (IP_OVERHEAD);
450
291 recv_vpn_packet (pkt, si); 451 recv_vpn_packet (pkt, si);
292 } 452 }
293 else 453 else
294 { 454 {
295 // probably ECONNRESET or somesuch 455 // probably ECONNRESET or somesuch
296 slog (L_DEBUG, _("%s: fd %d, %s"), (const char *)si, w.fd, strerror (errno)); 456 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno));
297 } 457 }
298 458
299 delete pkt; 459 delete pkt;
300 }
301 else if (revents & POLLHUP)
302 {
303 // this cannot ;) happen on udp sockets
304 slog (L_ERR, _("FATAL: POLLHUP on udp v4 fd, terminating."));
305 exit (1);
306 } 460 }
307 else 461 else
308 { 462 {
309 slog (L_ERR, 463 slog (L_ERR,
310 _("FATAL: unknown revents %08x in socket, terminating\n"), 464 _("FATAL: unknown revents %08x in socket, terminating\n"),
311 revents); 465 revents);
312 exit (1); 466 exit (EXIT_FAILURE);
313 } 467 }
314} 468}
315 469
470#if ENABLE_ICMP
316void 471void
317vpn::ipv4_ev (io_watcher &w, short revents) 472vpn::icmpv4_ev (io_watcher &w, short revents)
318{ 473{
319 if (revents & (POLLIN | POLLERR)) 474 if (revents & EVENT_READ)
320 { 475 {
321 vpn_packet *pkt = new vpn_packet; 476 vpn_packet *pkt = new vpn_packet;
322 struct sockaddr_in sa; 477 struct sockaddr_in sa;
323 socklen_t sa_len = sizeof (sa); 478 socklen_t sa_len = sizeof (sa);
324 int len; 479 int len;
325 480
326 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len); 481 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len);
327 482
328 sockinfo si(sa, PROT_IPv4); 483 sockinfo si(sa, PROT_ICMPv4);
329 484
330 if (len > 0) 485 if (len > 0)
331 { 486 {
332 pkt->len = len; 487 pkt->len = len;
333 488
489 icmp_header *hdr = (icmp_header *)&((*pkt)[IP_OVERHEAD]);
490
491 if (hdr->type == ::conf.icmp_type
492 && hdr->code == 255)
493 {
334 // raw sockets deliver the ipv4, but don't expect it on sends 494 // raw sockets deliver the ipv4, but don't expect it on sends
335 // this is slow, but... 495 // this is slow, but...
336 pkt->skip_hdr (IP_OVERHEAD); 496 pkt->skip_hdr (ICMP_OVERHEAD);
337 497
338 recv_vpn_packet (pkt, si); 498 recv_vpn_packet (pkt, si);
499 }
339 } 500 }
340 else 501 else
341 { 502 {
342 // probably ECONNRESET or somesuch 503 // probably ECONNRESET or somesuch
343 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno)); 504 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno));
344 } 505 }
345 506
346 delete pkt; 507 delete pkt;
347 }
348 else if (revents & POLLHUP)
349 {
350 // this cannot ;) happen on udp sockets
351 slog (L_ERR, _("FATAL: POLLHUP on ipv4 fd, terminating."));
352 exit (1);
353 } 508 }
354 else 509 else
355 { 510 {
356 slog (L_ERR, 511 slog (L_ERR,
357 _("FATAL: unknown revents %08x in socket, terminating\n"), 512 _("FATAL: unknown revents %08x in socket, terminating\n"),
358 revents); 513 revents);
359 exit (1); 514 exit (EXIT_FAILURE);
515 }
516}
517#endif
518
519void
520vpn::udpv4_ev (io_watcher &w, short revents)
521{
522 if (revents & EVENT_READ)
523 {
524 vpn_packet *pkt = new vpn_packet;
525 struct sockaddr_in sa;
526 socklen_t sa_len = sizeof (sa);
527 int len;
528
529 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len);
530
531 sockinfo si(sa, PROT_UDPv4);
532
533 if (len > 0)
534 {
535 pkt->len = len;
536
537 recv_vpn_packet (pkt, si);
538 }
539 else
540 {
541 // probably ECONNRESET or somesuch
542 slog (L_DEBUG, _("%s: fd %d, %s"), (const char *)si, w.fd, strerror (errno));
543 }
544
545 delete pkt;
546 }
547 else
548 {
549 slog (L_ERR,
550 _("FATAL: unknown revents %08x in socket, terminating\n"),
551 revents);
552 exit (EXIT_FAILURE);
360 } 553 }
361} 554}
362 555
363void 556void
364vpn::tap_ev (io_watcher &w, short revents) 557vpn::tap_ev (io_watcher &w, short revents)
365{ 558{
366 if (revents & POLLIN) 559 if (revents & EVENT_READ)
367 { 560 {
368 /* process data */ 561 /* process data */
369 tap_packet *pkt; 562 tap_packet *pkt;
370 563
371 pkt = tap->recv (); 564 pkt = tap->recv ();
372 565
566 if (!pkt)
567 return;
568
569 if (pkt->len > 14)
570 {
373 int dst = mac2id (pkt->dst); 571 int dst = mac2id (pkt->dst);
374 int src = mac2id (pkt->src); 572 int src = mac2id (pkt->src);
375 573
376 if (src != THISNODE->id) 574 if (src != THISNODE->id)
377 {
378 slog (L_ERR, _("FATAL: tap packet not originating on current node received, terminating."));
379 exit (1);
380 }
381
382 if (dst == THISNODE->id)
383 {
384 slog (L_ERR, _("FATAL: tap packet destined for current node received, terminating."));
385 exit (1);
386 }
387
388 if (dst > conns.size ())
389 slog (L_ERR, _("tap packet for unknown node %d received, ignoring."), dst);
390 else
391 {
392 if (dst)
393 { 575 {
394 // unicast 576 slog (L_ERR, _("FATAL: tap packet not originating on current node received, exiting."));
395 if (dst != THISNODE->id) 577 exit (EXIT_FAILURE);
396 conns[dst - 1]->inject_data_packet (pkt);
397 } 578 }
579
580 if (dst == THISNODE->id)
581 {
582 slog (L_ERR, _("FATAL: tap packet destined for current node received, exiting."));
583 exit (EXIT_FAILURE);
584 }
585
586 if (dst > conns.size ())
587 slog (L_ERR, _("tap packet for unknown node %d received, ignoring."), dst);
398 else 588 else
399 {
400 // broadcast, first check router, then self, then english
401 connection *router = find_router ();
402
403 if (router)
404 router->inject_data_packet (pkt, true); 589 inject_data_packet (pkt, dst);
405 else
406 for (conns_vector::iterator c = conns.begin (); c != conns.end (); ++c)
407 if ((*c)->conf != THISNODE)
408 (*c)->inject_data_packet (pkt);
409 }
410 } 590 }
411 591
412 delete pkt; 592 delete pkt;
413 }
414 else if (revents & (POLLHUP | POLLERR))
415 {
416 slog (L_ERR, _("FATAL: POLLHUP or POLLERR on network device fd, terminating."));
417 exit (1);
418 } 593 }
419 else 594 else
420 abort (); 595 abort ();
421} 596}
422 597
428 if (events & EVENT_SHUTDOWN) 603 if (events & EVENT_SHUTDOWN)
429 { 604 {
430 slog (L_INFO, _("preparing shutdown...")); 605 slog (L_INFO, _("preparing shutdown..."));
431 606
432 shutdown_all (); 607 shutdown_all ();
433
434 remove_pid (pidfilename); 608 remove_pid (conf.pidfilename);
435
436 slog (L_INFO, _("terminating")); 609 slog (L_INFO, _("terminating"));
437 610 exit (EXIT_SUCCESS);
438 exit (0);
439 } 611 }
440 612
441 if (events & EVENT_RECONNECT) 613 if (events & EVENT_RECONNECT)
442 { 614 {
443 slog (L_INFO, _("forced reconnect")); 615 slog (L_INFO, _("forced reconnect"));
445 reconnect_all (); 617 reconnect_all ();
446 } 618 }
447 619
448 events = 0; 620 events = 0;
449 } 621 }
450
451 w.at = TSTAMP_CANCEL;
452} 622}
453 623
454void 624void
455vpn::shutdown_all () 625vpn::shutdown_all ()
456{ 626{
480 } 650 }
481} 651}
482 652
483connection *vpn::find_router () 653connection *vpn::find_router ()
484{ 654{
485 u32 prio = 0; 655 u32 prio = 1;
486 connection *router = 0; 656 connection *router = 0;
487 657
488 for (conns_vector::iterator i = conns.begin (); i != conns.end (); ++i) 658 for (conns_vector::iterator i = conns.begin (); i != conns.end (); ++i)
489 { 659 {
490 connection *c = *i; 660 connection *c = *i;
491 661
492 if (c->conf->routerprio > prio 662 if (c->conf->routerprio > prio
493 && c->connectmode == conf_node::C_ALWAYS 663 && c->connectmode == conf_node::C_ALWAYS // so we don't drop the connection if in use
494 && c->conf != THISNODE
495 && c->ictx && c->octx) 664 && c->ictx && c->octx
665 && c->conf != THISNODE) // redundant, since ictx==octx==0 always on thisnode
496 { 666 {
497 prio = c->conf->routerprio; 667 prio = c->conf->routerprio;
498 router = c; 668 router = c;
499 } 669 }
500 } 670 }
509 if (c) 679 if (c)
510 c->send_connect_request (id); 680 c->send_connect_request (id);
511 else 681 else
512 // no router found, aggressively connect to all routers 682 // no router found, aggressively connect to all routers
513 for (conns_vector::iterator i = conns.begin (); i != conns.end (); ++i) 683 for (conns_vector::iterator i = conns.begin (); i != conns.end (); ++i)
514 if ((*i)->conf->routerprio) 684 if ((*i)->conf->routerprio && (*i)->conf != THISNODE)
515 (*i)->establish_connection (); 685 (*i)->establish_connection ();
516} 686}
517 687
518void 688void
519connection::dump_status () 689connection::dump_status ()
537 707
538 slog (L_NOTICE, _("END status dump")); 708 slog (L_NOTICE, _("END status dump"));
539} 709}
540 710
541vpn::vpn (void) 711vpn::vpn (void)
542: event(this, &vpn::event_cb) 712: event (this, &vpn::event_cb)
543, udpv4_ev_watcher(this, &vpn::udpv4_ev) 713, udpv4_ev_watcher (this, &vpn::udpv4_ev)
544, ipv4_ev_watcher (this, &vpn::ipv4_ev) 714, ipv4_ev_watcher (this, &vpn::ipv4_ev)
545, tap_ev_watcher (this, &vpn::tap_ev)
546#if ENABLE_TCP 715#if ENABLE_TCP
547, tcpv4_ev_watcher(this, &vpn::tcpv4_ev) 716, tcpv4_ev_watcher (this, &vpn::tcpv4_ev)
548#endif 717#endif
718#if ENABLE_ICMP
719, icmpv4_ev_watcher(this, &vpn::icmpv4_ev)
720#endif
721#if ENABLE_DNS
722, dnsv4_ev_watcher (this, &vpn::dnsv4_ev)
723#endif
724, tap_ev_watcher (this, &vpn::tap_ev)
549{ 725{
550} 726}
551 727
552vpn::~vpn () 728vpn::~vpn ()
553{ 729{

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines