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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines