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

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines