… | |
… | |
38 | |
38 | |
39 | #include "connection.h" |
39 | #include "connection.h" |
40 | #include "util.h" |
40 | #include "util.h" |
41 | #include "vpn.h" |
41 | #include "vpn.h" |
42 | |
42 | |
43 | #if ENABLE_TCP |
|
|
44 | # include <map> |
|
|
45 | # include <unistd.h> |
|
|
46 | # include <fcntl.h> |
|
|
47 | # include <sys/poll.h> |
|
|
48 | #endif |
|
|
49 | |
|
|
50 | ///////////////////////////////////////////////////////////////////////////// |
43 | ///////////////////////////////////////////////////////////////////////////// |
51 | |
44 | |
52 | const char *vpn::script_if_up () |
45 | const char *vpn::script_if_up () |
53 | { |
46 | { |
54 | // the tunnel device mtu should be the physical mtu - overhead |
47 | // the tunnel device mtu should be the physical mtu - overhead |
… | |
… | |
191 | reconnect_all (); |
184 | reconnect_all (); |
192 | |
185 | |
193 | return 0; |
186 | return 0; |
194 | } |
187 | } |
195 | |
188 | |
|
|
189 | // send a vpn packet out to other hosts |
|
|
190 | void |
|
|
191 | vpn::send_vpn_packet (vpn_packet *pkt, const sockinfo &si, int tos) |
|
|
192 | { |
|
|
193 | switch (si.prot) |
|
|
194 | { |
|
|
195 | case PROT_IPv4: |
|
|
196 | send_ipv4_packet (pkt, si, tos); |
|
|
197 | break; |
|
|
198 | |
|
|
199 | case PROT_UDPv4: |
|
|
200 | send_udpv4_packet (pkt, si, tos); |
|
|
201 | break; |
|
|
202 | |
|
|
203 | #if ENABLE_TCP |
|
|
204 | case PROT_TCPv4: |
|
|
205 | send_tcpv4_packet (pkt, si, tos); |
|
|
206 | break; |
|
|
207 | #endif |
|
|
208 | |
|
|
209 | default: |
|
|
210 | slog (L_CRIT, _("%s: FATAL: trying to send packet with unsupported protocol"), (const char *)si); |
|
|
211 | abort (); |
|
|
212 | } |
|
|
213 | } |
|
|
214 | |
196 | void |
215 | void |
197 | vpn::send_ipv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) |
216 | vpn::send_ipv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) |
198 | { |
217 | { |
199 | setsockopt (ipv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos); |
218 | setsockopt (ipv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos); |
200 | sendto (ipv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ()); |
219 | sendto (ipv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ()); |
… | |
… | |
248 | vpn_packet *pkt = new vpn_packet; |
267 | vpn_packet *pkt = new vpn_packet; |
249 | struct sockaddr_in sa; |
268 | struct sockaddr_in sa; |
250 | socklen_t sa_len = sizeof (sa); |
269 | socklen_t sa_len = sizeof (sa); |
251 | int len; |
270 | int len; |
252 | |
271 | |
253 | len = recvfrom (w.p->fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len); |
272 | len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len); |
254 | |
273 | |
255 | sockinfo si(sa); |
274 | sockinfo si(sa, PROT_UDPv4); |
256 | |
275 | |
257 | if (len > 0) |
276 | if (len > 0) |
258 | { |
277 | { |
259 | pkt->len = len; |
278 | pkt->len = len; |
260 | |
279 | |
261 | recv_vpn_packet (pkt, si); |
280 | recv_vpn_packet (pkt, si); |
262 | } |
281 | } |
263 | else |
282 | else |
264 | { |
283 | { |
265 | // probably ECONNRESET or somesuch |
284 | // probably ECONNRESET or somesuch |
266 | slog (L_DEBUG, _("%s: %s"), (const char *)si, strerror (errno)); |
285 | slog (L_DEBUG, _("%s: fd %d, %s"), (const char *)si, w.fd, strerror (errno)); |
267 | } |
286 | } |
268 | |
287 | |
269 | delete pkt; |
288 | delete pkt; |
270 | } |
289 | } |
271 | else if (revents & POLLHUP) |
290 | else if (revents & POLLHUP) |
… | |
… | |
291 | vpn_packet *pkt = new vpn_packet; |
310 | vpn_packet *pkt = new vpn_packet; |
292 | struct sockaddr_in sa; |
311 | struct sockaddr_in sa; |
293 | socklen_t sa_len = sizeof (sa); |
312 | socklen_t sa_len = sizeof (sa); |
294 | int len; |
313 | int len; |
295 | |
314 | |
296 | len = recvfrom (w.p->fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len); |
315 | len = recvfrom (w.fd, &((*pkt)[0]), MAXSIZE, 0, (sockaddr *)&sa, &sa_len); |
297 | |
316 | |
298 | sockinfo si(sa, PROT_IPv4); |
317 | sockinfo si(sa, PROT_IPv4); |
299 | |
318 | |
300 | if (len > 0) |
319 | if (len > 0) |
301 | { |
320 | { |
… | |
… | |
327 | _("FATAL: unknown revents %08x in socket, terminating\n"), |
346 | _("FATAL: unknown revents %08x in socket, terminating\n"), |
328 | revents); |
347 | revents); |
329 | exit (1); |
348 | exit (1); |
330 | } |
349 | } |
331 | } |
350 | } |
332 | |
|
|
333 | #if ENABLE_TCP |
|
|
334 | |
|
|
335 | struct tcp_connection; |
|
|
336 | |
|
|
337 | struct lt_sockinfo |
|
|
338 | { |
|
|
339 | bool operator()(const sockinfo *a, const sockinfo *b) const |
|
|
340 | { |
|
|
341 | return *a < *b; |
|
|
342 | } |
|
|
343 | }; |
|
|
344 | |
|
|
345 | struct 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 | |
|
|
355 | struct tcp_connection : io_watcher { |
|
|
356 | tstamp last_activity; |
|
|
357 | const sockinfo si; |
|
|
358 | vpn &v; |
|
|
359 | bool ok; |
|
|
360 | |
|
|
361 | vpn_packet *r_pkt; |
|
|
362 | u32 r_len, r_ofs; |
|
|
363 | |
|
|
364 | void tcpv4_ev (io_watcher &w, short revents); |
|
|
365 | |
|
|
366 | operator tcp_si_map::value_type() |
|
|
367 | { |
|
|
368 | return tcp_si_map::value_type (&si, this); |
|
|
369 | } |
|
|
370 | |
|
|
371 | tcp_connection (int fd, const sockinfo &si_, vpn &v_) |
|
|
372 | : v(v_), si(si_), io_watcher(this, &tcp_connection::tcpv4_ev) |
|
|
373 | { |
|
|
374 | last_activity = NOW; |
|
|
375 | ok = false; |
|
|
376 | r_pkt = 0; |
|
|
377 | start (fd, POLLOUT); |
|
|
378 | } |
|
|
379 | |
|
|
380 | ~tcp_connection () { if (p) close (p->fd); } |
|
|
381 | }; |
|
|
382 | |
|
|
383 | void tcp_si_map::cleaner_cb (time_watcher &w) |
|
|
384 | { |
|
|
385 | w.at = NOW + 600; |
|
|
386 | tstamp to = NOW - ::conf.keepalive - 30; |
|
|
387 | |
|
|
388 | for (iterator i = begin (); i != end(); ) |
|
|
389 | if (i->second->last_activity >= to) |
|
|
390 | ++i; |
|
|
391 | else |
|
|
392 | { |
|
|
393 | erase (i); |
|
|
394 | i = begin (); |
|
|
395 | } |
|
|
396 | } |
|
|
397 | |
|
|
398 | void |
|
|
399 | vpn::send_tcpv4_packet (vpn_packet *pkt, const sockinfo &si, int tos) |
|
|
400 | { |
|
|
401 | tcp_si_map::iterator info = tcp_si.find (&si); |
|
|
402 | |
|
|
403 | if (info == tcp_si.end ()) |
|
|
404 | { |
|
|
405 | // woaw, the first lost packet ;) |
|
|
406 | int fd = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP); |
|
|
407 | |
|
|
408 | if (fd >= 0) |
|
|
409 | { |
|
|
410 | fcntl (fd, F_SETFL, O_NONBLOCK); |
|
|
411 | |
|
|
412 | if (connect (fd, si.sav4 (), si.salenv4 ()) >= 0 |
|
|
413 | || errno == EINPROGRESS) |
|
|
414 | { |
|
|
415 | tcp_connection *i = new tcp_connection (fd, si, *this); |
|
|
416 | |
|
|
417 | tcp_si.insert (*i); |
|
|
418 | } |
|
|
419 | else |
|
|
420 | close (fd); |
|
|
421 | } |
|
|
422 | } |
|
|
423 | else |
|
|
424 | { |
|
|
425 | tcp_connection *i = info->second; |
|
|
426 | |
|
|
427 | i->last_activity = NOW; |
|
|
428 | |
|
|
429 | if (i->ok) |
|
|
430 | { |
|
|
431 | setsockopt (i->p->fd, SOL_IP, IP_TOS, &tos, sizeof tos); |
|
|
432 | |
|
|
433 | // we use none of the advantages of tcp |
|
|
434 | write (i->p->fd, (void *)pkt, pkt->len + sizeof (u32)) != pkt->len + sizeof (u32); |
|
|
435 | } |
|
|
436 | } |
|
|
437 | |
|
|
438 | #if 0 |
|
|
439 | setsockopt (udpv4_fd, SOL_IP, IP_TOS, &tos, sizeof tos); |
|
|
440 | sendto (udpv4_fd, &((*pkt)[0]), pkt->len, 0, si.sav4 (), si.salenv4 ()); |
|
|
441 | #endif |
|
|
442 | } |
|
|
443 | |
|
|
444 | void |
|
|
445 | tcp_connection::tcpv4_ev (io_watcher &w, short revents) |
|
|
446 | { |
|
|
447 | last_activity = NOW; |
|
|
448 | |
|
|
449 | if (!ok) // just established? |
|
|
450 | { |
|
|
451 | ok = true; |
|
|
452 | set (POLLIN); |
|
|
453 | } |
|
|
454 | |
|
|
455 | if (revents & (POLLIN | POLLERR)) |
|
|
456 | { |
|
|
457 | u32 len; |
|
|
458 | |
|
|
459 | if (sizeof (len) == read (p->fd, &len, sizeof (len))) |
|
|
460 | { |
|
|
461 | vpn_packet *pkt = new vpn_packet; |
|
|
462 | |
|
|
463 | if (len == read (p->fd, &((*pkt)[0]), len)) |
|
|
464 | { |
|
|
465 | pkt->len = len; |
|
|
466 | |
|
|
467 | v.recv_vpn_packet (pkt, si); |
|
|
468 | return; |
|
|
469 | } |
|
|
470 | |
|
|
471 | delete pkt; |
|
|
472 | } |
|
|
473 | |
|
|
474 | tcp_si.erase (&si); |
|
|
475 | |
|
|
476 | set (0);//D |
|
|
477 | } |
|
|
478 | } |
|
|
479 | |
|
|
480 | void |
|
|
481 | vpn::tcpv4_ev (io_watcher &w, short revents) |
|
|
482 | { |
|
|
483 | if (revents & (POLLIN | POLLERR)) |
|
|
484 | { |
|
|
485 | struct sockaddr_in sa; |
|
|
486 | socklen_t sa_len = sizeof (sa); |
|
|
487 | int len; |
|
|
488 | |
|
|
489 | int fd = accept (w.p->fd, (sockaddr *)&sa, &sa_len); |
|
|
490 | |
|
|
491 | if (fd >= 0) |
|
|
492 | { |
|
|
493 | fcntl (fd, F_SETFL, O_NONBLOCK); |
|
|
494 | |
|
|
495 | sockinfo si(sa, PROT_TCPv4); |
|
|
496 | tcp_connection *i = new tcp_connection (fd, si, *this); |
|
|
497 | |
|
|
498 | slog (L_DEBUG, _("accepted tcp connection from %s\n"), (const char *)si);//D |
|
|
499 | |
|
|
500 | tcp_si.insert (*i); |
|
|
501 | } |
|
|
502 | } |
|
|
503 | } |
|
|
504 | |
|
|
505 | #endif |
|
|
506 | |
351 | |
507 | void |
352 | void |
508 | vpn::tap_ev (io_watcher &w, short revents) |
353 | vpn::tap_ev (io_watcher &w, short revents) |
509 | { |
354 | { |
510 | if (revents & POLLIN) |
355 | if (revents & POLLIN) |