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.7 by pcg, Sun Apr 6 18:12:18 2003 UTC vs.
Revision 1.27 by pcg, Tue Mar 1 06:27:20 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-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
71} 72}
72 73
73int 74int
74vpn::setup () 75vpn::setup ()
75{ 76{
77 ipv4_fd = -1;
78
79 if (THISNODE->protocols & PROT_IPv4 && ::conf.ip_proto)
80 {
81 ipv4_fd = socket (PF_INET, SOCK_RAW, ::conf.ip_proto);
82
83 if (ipv4_fd < 0)
84 return -1;
85
86 fcntl (ipv4_fd, F_SETFL, O_NONBLOCK);
87
88#if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
89 // this I really consider a linux bug. I am neither connected
90 // nor do I fragment myself. Linux still sets DF and doesn't
91 // fragment for me sometimes.
92 {
93 int oval = IP_PMTUDISC_DONT;
94 setsockopt (ipv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval);
95 }
96#endif
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
106 ipv4_ev_watcher.start (ipv4_fd, EVENT_READ);
107 }
108
76 udpv4_fd = -1; 109 udpv4_fd = -1;
77 110
78 if (THISNODE->protocols & PROT_UDPv4) 111 if (THISNODE->protocols & PROT_UDPv4 && THISNODE->udp_port)
79 { 112 {
80 udpv4_fd = socket (PF_INET, SOCK_DGRAM, IPPROTO_UDP); 113 udpv4_fd = socket (PF_INET, SOCK_DGRAM, IPPROTO_UDP);
81 114
82 if (udpv4_fd < 0) 115 if (udpv4_fd < 0)
83 return -1; 116 return -1;
117
118 fcntl (udpv4_fd, F_SETFL, O_NONBLOCK);
84 119
85 // standard daemon practise... 120 // standard daemon practise...
86 { 121 {
87 int oval = 1; 122 int oval = 1;
88 setsockopt (udpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval); 123 setsockopt (udpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval);
89 } 124 }
90 125
91 sockinfo si (THISNODE, PROT_UDPv4); 126#if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
92
93 if (bind (udpv4_fd, si.sav4 (), si.salenv4 ()))
94 {
95 slog (L_ERR, _("can't bind udpv4 on %s: %s"), (const char *)si, strerror (errno));
96 exit (1);
97 }
98
99#ifdef IP_MTU_DISCOVER
100 // this I really consider a linux bug. I am neither connected 127 // this I really consider a linux bug. I am neither connected
101 // 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
102 // fragment for me sometimes. 129 // fragment for me sometimes.
103 { 130 {
104 int oval = IP_PMTUDISC_DONT; 131 int oval = IP_PMTUDISC_DONT;
105 setsockopt (udpv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval); 132 setsockopt (udpv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval);
106 } 133 }
107#endif 134#endif
108 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
109 udpv4_ev_watcher.start (udpv4_fd, POLLIN); 144 udpv4_ev_watcher.start (udpv4_fd, EVENT_READ);
110 } 145 }
111 146
112 ipv4_fd = -1; 147 icmpv4_fd = -1;
148
149#if ENABLE_ICMP
113 if (THISNODE->protocols & PROT_IPv4) 150 if (THISNODE->protocols & PROT_ICMPv4)
114 { 151 {
115 ipv4_fd = socket (PF_INET, SOCK_RAW, ::conf.ip_proto); 152 icmpv4_fd = socket (PF_INET, SOCK_RAW, IPPROTO_ICMP);
116 153
117 if (ipv4_fd < 0) 154 if (icmpv4_fd < 0)
118 return -1; 155 return -1;
119 156
120 sockinfo si (THISNODE, PROT_IPv4); 157 fcntl (icmpv4_fd, F_SETFL, O_NONBLOCK);
121 158
122 if (bind (ipv4_fd, si.sav4 (), si.salenv4 ())) 159#ifdef ICMP_FILTER
123 { 160 {
124 slog (L_ERR, _("can't bind ipv4 socket on %s: %s"), (const char *)si, strerror (errno)); 161 icmp_filter oval;
125 exit (1); 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);
126 } 167 }
168#endif
127 169
128#ifdef IP_MTU_DISCOVER 170#if defined(SOL_IP) && defined(IP_MTU_DISCOVER)
129 // this I really consider a linux bug. I am neither connected 171 // this I really consider a linux bug. I am neither connected
130 // nor do I fragment myself. Linux still sets DF and doesn't 172 // nor do I fragment myself. Linux still sets DF and doesn't
131 // fragment for me sometimes. 173 // fragment for me sometimes.
132 { 174 {
133 int oval = IP_PMTUDISC_DONT; 175 int oval = IP_PMTUDISC_DONT;
134 setsockopt (ipv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval); 176 setsockopt (udpv4_fd, SOL_IP, IP_MTU_DISCOVER, &oval, sizeof oval);
135 } 177 }
136#endif 178#endif
137 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
138 ipv4_ev_watcher.start (ipv4_fd, POLLIN); 188 icmpv4_ev_watcher.start (icmpv4_fd, EVENT_READ);
139 } 189 }
190#endif
191
192 tcpv4_fd = -1;
140 193
141#if ENABLE_TCP 194#if ENABLE_TCP
142 if (THISNODE->protocols & PROT_TCPv4) 195 if (THISNODE->protocols & PROT_TCPv4 && THISNODE->tcp_port)
143 { 196 {
144 tcpv4_fd = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP); 197 tcpv4_fd = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP);
145 198
146 if (tcpv4_fd < 0) 199 if (tcpv4_fd < 0)
147 return -1; 200 return -1;
201
202 fcntl (tcpv4_fd, F_SETFL, O_NONBLOCK);
148 203
149 // standard daemon practise... 204 // standard daemon practise...
150 { 205 {
151 int oval = 1; 206 int oval = 1;
152 setsockopt (tcpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval); 207 setsockopt (tcpv4_fd, SOL_SOCKET, SO_REUSEADDR, &oval, sizeof oval);
155 sockinfo si (THISNODE, PROT_TCPv4); 210 sockinfo si (THISNODE, PROT_TCPv4);
156 211
157 if (bind (tcpv4_fd, si.sav4 (), si.salenv4 ())) 212 if (bind (tcpv4_fd, si.sav4 (), si.salenv4 ()))
158 { 213 {
159 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));
160 exit (1); 215 exit (EXIT_FAILURE);
161 } 216 }
162 217
163 if (listen (tcpv4_fd, 5)) 218 if (listen (tcpv4_fd, 5))
164 { 219 {
165 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));
166 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 ()))
167 } 245 {
246 slog (L_ERR, _("can't bind dnsv4 on %s: %s"), (const char *)si, strerror (errno));
247 exit (EXIT_FAILURE);
248 }
168 249
169 tcpv4_ev_watcher.start (tcpv4_fd, POLLIN); 250 dnsv4_ev_watcher.start (dnsv4_fd, EVENT_READ);
170 } 251 }
171#endif 252#endif
172 253
173 tap = new tap_device (); 254 tap = new tap_device ();
174 if (!tap) //D this, of course, never catches 255 if (!tap) //D this, of course, never catches
175 { 256 {
176 slog (L_ERR, _("cannot create network interface '%s'"), conf.ifname); 257 slog (L_ERR, _("cannot create network interface '%s'"), conf.ifname);
177 exit (1); 258 exit (EXIT_FAILURE);
178 } 259 }
179 260
180 run_script (run_script_cb (this, &vpn::script_if_up), true); 261 run_script (run_script_cb (this, &vpn::script_if_up), true);
181 262
182 tap_ev_watcher.start (tap->fd, POLLIN); 263 tap_ev_watcher.start (tap->fd, EVENT_READ);
183 264
184 reconnect_all (); 265 reconnect_all ();
185 266
186 return 0; 267 return 0;
187}
188
189// send a vpn packet out to other hosts
190bool
191vpn::send_vpn_packet (vpn_packet *pkt, const sockinfo &si, int tos)
192{
193 switch (si.prot)
194 {
195 case PROT_IPv4:
196 return send_ipv4_packet (pkt, si, tos);
197
198 case PROT_UDPv4:
199 return send_udpv4_packet (pkt, si, tos);
200
201#if ENABLE_TCP
202 case PROT_TCPv4:
203 return send_tcpv4_packet (pkt, si, tos);
204#endif
205
206 default:
207 slog (L_CRIT, _("%s: FATAL: trying to send packet with unsupported protocol"), (const char *)si);
208 return false;
209 }
210} 268}
211 269
212bool 270bool
213vpn::send_ipv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) 271vpn::send_ipv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
214{ 272{
273#if defined(SOL_IP) && defined(IP_TOS)
215 setsockopt (ipv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos); 274 setsockopt (ipv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
275#endif
216 sendto (ipv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ()); 276 sendto (ipv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ());
217 277
218 return true; 278 return true;
219} 279}
280
281static u16
282ipv4_checksum (u16 *data, unsigned int len)
283{
284 // use 32 bit accumulator and fold back carry bits at the end
285 u32 sum = 0;
286
287 while (len > 1)
288 {
289 sum += *data++;
290 len -= 2;
291 }
292
293 // odd byte left?
294 if (len)
295 sum += *(u8 *)data;
296
297 // add back carry bits
298 sum = (sum >> 16) + (sum & 0xffff); // lo += hi
299 sum += (sum >> 16); // carry
300
301 return ~sum;
302}
303
304#if ENABLE_ICMP
305bool
306vpn::send_icmpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
307{
308#if defined(SOL_IP) && defined(IP_TOS)
309 setsockopt (icmpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
310#endif
311
312 pkt->unshift_hdr (4);
313
314 icmp_header *hdr = (icmp_header *)&((*pkt)[0]);
315 hdr->type = ::conf.icmp_type;
316 hdr->code = 255;
317 hdr->checksum = 0;
318 hdr->checksum = ipv4_checksum ((u16 *)hdr, pkt->len);
319
320 sendto (icmpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ());
321
322 return true;
323}
324#endif
220 325
221bool 326bool
222vpn::send_udpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) 327vpn::send_udpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos)
223{ 328{
329#if defined(SOL_IP) && defined(IP_TOS)
224 setsockopt (udpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos); 330 setsockopt (udpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos);
331#endif
225 sendto (udpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ()); 332 sendto (udpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ());
226 333
227 return true; 334 return true;
335}
336
337void
338vpn::inject_data_packet (tap_packet *pkt, int dst)
339{
340 if (dst)
341 {
342 // unicast
343 if (dst != THISNODE->id)
344 conns[dst - 1]->inject_data_packet (pkt);
345 }
346 else
347 {
348 // broadcast, this is ugly, but due to the security policy
349 // we have to connect to all hosts...
350 for (conns_vector::iterator c = conns.begin (); c != conns.end (); ++c)
351 if ((*c)->conf != THISNODE)
352 (*c)->inject_data_packet (pkt, true);
353 }
228} 354}
229 355
230void 356void
231vpn::recv_vpn_packet (vpn_packet *pkt, const sockinfo &rsi) 357vpn::recv_vpn_packet (vpn_packet *pkt, const sockinfo &rsi)
232{ 358{
246 (const char *)rsi, pkt->typ (), pkt->src (), pkt->dst ()); 372 (const char *)rsi, pkt->typ (), pkt->src (), pkt->dst ());
247 else 373 else
248 { 374 {
249 connection *c = conns[src - 1]; 375 connection *c = conns[src - 1];
250 376
251 if (dst == 0 && !THISNODE->routerprio) 377 if (dst == 0)
252 slog (L_WARN, _("%s(%s): received broadcast, but we are no router"), 378 slog (L_WARN, _("%s(%s): received broadcast (protocol violation)"),
253 c->conf->nodename, (const char *)rsi); 379 c->conf->nodename, (const char *)rsi);
254 else if (dst != 0 && dst != THISNODE->id) 380 else if (dst != THISNODE->id)
255 { 381 {
256 if (THISNODE->routerprio) 382 if (THISNODE->routerprio)
257 // the tos setting gets lost here. who cares. 383 // the tos setting gets lost here. who cares.
258 conns[dst - 1]->inject_vpn_packet (pkt); 384 conns[dst - 1]->inject_vpn_packet (pkt);
259 else 385 else
266 c->recv_vpn_packet (pkt, rsi); 392 c->recv_vpn_packet (pkt, rsi);
267 } 393 }
268} 394}
269 395
270void 396void
271vpn::udpv4_ev (io_watcher &w, short revents) 397vpn::ipv4_ev (io_watcher &w, short revents)
272{ 398{
273 if (revents & (POLLIN | POLLERR)) 399 if (revents & EVENT_READ)
274 { 400 {
275 vpn_packet *pkt = new vpn_packet; 401 vpn_packet *pkt = new vpn_packet;
276 struct sockaddr_in sa; 402 struct sockaddr_in sa;
277 socklen_t sa_len = sizeof (sa); 403 socklen_t sa_len = sizeof (sa);
278 int len; 404 int len;
279 405
280 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len); 406 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len);
281 407
282 sockinfo si(sa, PROT_UDPv4); 408 sockinfo si(sa, PROT_IPv4);
283 409
284 if (len > 0) 410 if (len > 0)
285 { 411 {
286 pkt->len = len; 412 pkt->len = len;
287 413
414 // raw sockets deliver the ipv4, but don't expect it on sends
415 // this is slow, but...
416 pkt->skip_hdr (IP_OVERHEAD);
417
288 recv_vpn_packet (pkt, si); 418 recv_vpn_packet (pkt, si);
289 } 419 }
290 else 420 else
291 { 421 {
292 // probably ECONNRESET or somesuch 422 // probably ECONNRESET or somesuch
293 slog (L_DEBUG, _("%s: fd %d, %s"), (const char *)si, w.fd, strerror (errno)); 423 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno));
294 } 424 }
295 425
296 delete pkt; 426 delete pkt;
297 }
298 else if (revents & POLLHUP)
299 {
300 // this cannot ;) happen on udp sockets
301 slog (L_ERR, _("FATAL: POLLHUP on udp v4 fd, terminating."));
302 exit (1);
303 } 427 }
304 else 428 else
305 { 429 {
306 slog (L_ERR, 430 slog (L_ERR,
307 _("FATAL: unknown revents %08x in socket, terminating\n"), 431 _("FATAL: unknown revents %08x in socket, terminating\n"),
308 revents); 432 revents);
309 exit (1); 433 exit (EXIT_FAILURE);
310 } 434 }
311} 435}
312 436
437#if ENABLE_ICMP
313void 438void
314vpn::ipv4_ev (io_watcher &w, short revents) 439vpn::icmpv4_ev (io_watcher &w, short revents)
315{ 440{
316 if (revents & (POLLIN | POLLERR)) 441 if (revents & EVENT_READ)
317 { 442 {
318 vpn_packet *pkt = new vpn_packet; 443 vpn_packet *pkt = new vpn_packet;
319 struct sockaddr_in sa; 444 struct sockaddr_in sa;
320 socklen_t sa_len = sizeof (sa); 445 socklen_t sa_len = sizeof (sa);
321 int len; 446 int len;
322 447
323 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len); 448 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len);
324 449
325 sockinfo si(sa, PROT_IPv4); 450 sockinfo si(sa, PROT_ICMPv4);
326 451
327 if (len > 0) 452 if (len > 0)
328 { 453 {
329 pkt->len = len; 454 pkt->len = len;
330 455
456 icmp_header *hdr = (icmp_header *)&((*pkt)[IP_OVERHEAD]);
457
458 if (hdr->type == ::conf.icmp_type
459 && hdr->code == 255)
460 {
331 // raw sockets deliver the ipv4, but don't expect it on sends 461 // raw sockets deliver the ipv4, but don't expect it on sends
332 // this is slow, but... 462 // this is slow, but...
333 pkt->skip_hdr (IP_OVERHEAD); 463 pkt->skip_hdr (ICMP_OVERHEAD);
334 464
335 recv_vpn_packet (pkt, si); 465 recv_vpn_packet (pkt, si);
466 }
336 } 467 }
337 else 468 else
338 { 469 {
339 // probably ECONNRESET or somesuch 470 // probably ECONNRESET or somesuch
340 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno)); 471 slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno));
341 } 472 }
342 473
343 delete pkt; 474 delete pkt;
344 }
345 else if (revents & POLLHUP)
346 {
347 // this cannot ;) happen on udp sockets
348 slog (L_ERR, _("FATAL: POLLHUP on ipv4 fd, terminating."));
349 exit (1);
350 } 475 }
351 else 476 else
352 { 477 {
353 slog (L_ERR, 478 slog (L_ERR,
354 _("FATAL: unknown revents %08x in socket, terminating\n"), 479 _("FATAL: unknown revents %08x in socket, terminating\n"),
355 revents); 480 revents);
356 exit (1); 481 exit (EXIT_FAILURE);
482 }
483}
484#endif
485
486void
487vpn::udpv4_ev (io_watcher &w, short revents)
488{
489 if (revents & EVENT_READ)
490 {
491 vpn_packet *pkt = new vpn_packet;
492 struct sockaddr_in sa;
493 socklen_t sa_len = sizeof (sa);
494 int len;
495
496 len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len);
497
498 sockinfo si(sa, PROT_UDPv4);
499
500 if (len > 0)
501 {
502 pkt->len = len;
503
504 recv_vpn_packet (pkt, si);
505 }
506 else
507 {
508 // probably ECONNRESET or somesuch
509 slog (L_DEBUG, _("%s: fd %d, %s"), (const char *)si, w.fd, strerror (errno));
510 }
511
512 delete pkt;
513 }
514 else
515 {
516 slog (L_ERR,
517 _("FATAL: unknown revents %08x in socket, terminating\n"),
518 revents);
519 exit (EXIT_FAILURE);
357 } 520 }
358} 521}
359 522
360void 523void
361vpn::tap_ev (io_watcher &w, short revents) 524vpn::tap_ev (io_watcher &w, short revents)
362{ 525{
363 if (revents & POLLIN) 526 if (revents & EVENT_READ)
364 { 527 {
365 /* process data */ 528 /* process data */
366 tap_packet *pkt; 529 tap_packet *pkt;
367 530
368 pkt = tap->recv (); 531 pkt = tap->recv ();
369 532
533 if (!pkt)
534 return;
535
536 if (pkt->len > 14)
537 {
370 int dst = mac2id (pkt->dst); 538 int dst = mac2id (pkt->dst);
371 int src = mac2id (pkt->src); 539 int src = mac2id (pkt->src);
372 540
373 if (src != THISNODE->id) 541 if (src != THISNODE->id)
374 {
375 slog (L_ERR, _("FATAL: tap packet not originating on current node received, terminating."));
376 exit (1);
377 }
378
379 if (dst == THISNODE->id)
380 {
381 slog (L_ERR, _("FATAL: tap packet destined for current node received, terminating."));
382 exit (1);
383 }
384
385 if (dst > conns.size ())
386 slog (L_ERR, _("tap packet for unknown node %d received, ignoring."), dst);
387 else
388 {
389 if (dst)
390 { 542 {
391 // unicast 543 slog (L_ERR, _("FATAL: tap packet not originating on current node received, exiting."));
392 if (dst != THISNODE->id) 544 exit (EXIT_FAILURE);
393 conns[dst - 1]->inject_data_packet (pkt);
394 } 545 }
546
547 if (dst == THISNODE->id)
548 {
549 slog (L_ERR, _("FATAL: tap packet destined for current node received, exiting."));
550 exit (EXIT_FAILURE);
551 }
552
553 if (dst > conns.size ())
554 slog (L_ERR, _("tap packet for unknown node %d received, ignoring."), dst);
395 else 555 else
396 {
397 // broadcast, first check router, then self, then english
398 connection *router = find_router ();
399
400 if (router)
401 router->inject_data_packet (pkt, true); 556 inject_data_packet (pkt, dst);
402 else
403 for (conns_vector::iterator c = conns.begin (); c != conns.end (); ++c)
404 if ((*c)->conf != THISNODE)
405 (*c)->inject_data_packet (pkt);
406 }
407 } 557 }
408 558
409 delete pkt; 559 delete pkt;
410 }
411 else if (revents & (POLLHUP | POLLERR))
412 {
413 slog (L_ERR, _("FATAL: POLLHUP or POLLERR on network device fd, terminating."));
414 exit (1);
415 } 560 }
416 else 561 else
417 abort (); 562 abort ();
418} 563}
419 564
425 if (events & EVENT_SHUTDOWN) 570 if (events & EVENT_SHUTDOWN)
426 { 571 {
427 slog (L_INFO, _("preparing shutdown...")); 572 slog (L_INFO, _("preparing shutdown..."));
428 573
429 shutdown_all (); 574 shutdown_all ();
430
431 remove_pid (pidfilename); 575 remove_pid (conf.pidfilename);
432
433 slog (L_INFO, _("terminating")); 576 slog (L_INFO, _("terminating"));
434 577 exit (EXIT_SUCCESS);
435 exit (0);
436 } 578 }
437 579
438 if (events & EVENT_RECONNECT) 580 if (events & EVENT_RECONNECT)
439 { 581 {
440 slog (L_INFO, _("forced reconnect")); 582 slog (L_INFO, _("forced reconnect"));
442 reconnect_all (); 584 reconnect_all ();
443 } 585 }
444 586
445 events = 0; 587 events = 0;
446 } 588 }
447
448 w.at = TSTAMP_CANCEL;
449} 589}
450 590
451void 591void
452vpn::shutdown_all () 592vpn::shutdown_all ()
453{ 593{
466 connection_init (); 606 connection_init ();
467 607
468 for (configuration::node_vector::iterator i = conf.nodes.begin (); 608 for (configuration::node_vector::iterator i = conf.nodes.begin ();
469 i != conf.nodes.end (); ++i) 609 i != conf.nodes.end (); ++i)
470 { 610 {
471 connection *conn = new connection (this); 611 connection *conn = new connection (this, *i);
472
473 conn->conf = *i;
474 conns.push_back (conn); 612 conns.push_back (conn);
475
476 conn->establish_connection (); 613 conn->establish_connection ();
477 } 614 }
478} 615}
479 616
480connection *vpn::find_router () 617connection *vpn::find_router ()
481{ 618{
482 u32 prio = 0; 619 u32 prio = 1;
483 connection *router = 0; 620 connection *router = 0;
484 621
485 for (conns_vector::iterator i = conns.begin (); i != conns.end (); ++i) 622 for (conns_vector::iterator i = conns.begin (); i != conns.end (); ++i)
486 { 623 {
487 connection *c = *i; 624 connection *c = *i;
488 625
489 if (c->conf->routerprio > prio 626 if (c->conf->routerprio > prio
490 && c->connectmode == conf_node::C_ALWAYS 627 && c->connectmode == conf_node::C_ALWAYS // so we don't drop the connection if in use
491 && c->conf != THISNODE
492 && c->ictx && c->octx) 628 && c->ictx && c->octx
629 && c->conf != THISNODE) // redundant, since ictx==octx==0 always on thisnode
493 { 630 {
494 prio = c->conf->routerprio; 631 prio = c->conf->routerprio;
495 router = c; 632 router = c;
496 } 633 }
497 } 634 }
506 if (c) 643 if (c)
507 c->send_connect_request (id); 644 c->send_connect_request (id);
508 else 645 else
509 // no router found, aggressively connect to all routers 646 // no router found, aggressively connect to all routers
510 for (conns_vector::iterator i = conns.begin (); i != conns.end (); ++i) 647 for (conns_vector::iterator i = conns.begin (); i != conns.end (); ++i)
511 if ((*i)->conf->routerprio) 648 if ((*i)->conf->routerprio && (*i)->conf != THISNODE)
512 (*i)->establish_connection (); 649 (*i)->establish_connection ();
513} 650}
514 651
515void 652void
516connection::dump_status () 653connection::dump_status ()
534 671
535 slog (L_NOTICE, _("END status dump")); 672 slog (L_NOTICE, _("END status dump"));
536} 673}
537 674
538vpn::vpn (void) 675vpn::vpn (void)
539: event(this, &vpn::event_cb) 676: event (this, &vpn::event_cb)
540, udpv4_ev_watcher(this, &vpn::udpv4_ev) 677, udpv4_ev_watcher (this, &vpn::udpv4_ev)
541, ipv4_ev_watcher (this, &vpn::ipv4_ev) 678, ipv4_ev_watcher (this, &vpn::ipv4_ev)
542, tap_ev_watcher (this, &vpn::tap_ev)
543#if ENABLE_TCP 679#if ENABLE_TCP
544, tcpv4_ev_watcher(this, &vpn::tcpv4_ev) 680, tcpv4_ev_watcher (this, &vpn::tcpv4_ev)
545#endif 681#endif
682#if ENABLE_ICMP
683, icmpv4_ev_watcher(this, &vpn::icmpv4_ev)
684#endif
685#if ENABLE_DNS
686, dnsv4_ev_watcher (this, &vpn::dnsv4_ev)
687#endif
688, tap_ev_watcher (this, &vpn::tap_ev)
546{ 689{
547} 690}
548 691
549vpn::~vpn () 692vpn::~vpn ()
550{ 693{

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines