Add a benchmark for the SPTPS protocol.
[tinc] / src / vde_device.c
1 /*
2     device.c -- VDE plug
3     Copyright (C) 2013 Guus Sliepen <guus@tinc-vpn.org>
4
5     This program is free software; you can redistribute it and/or modify
6     it under the terms of the GNU General Public License as published by
7     the Free Software Foundation; either version 2 of the License, or
8     (at your option) any later version.
9
10     This program is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13     GNU General Public License for more details.
14
15     You should have received a copy of the GNU General Public License along
16     with this program; if not, write to the Free Software Foundation, Inc.,
17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18 */
19
20 #include "system.h"
21
22 #include <libvdeplug_dyn.h>
23
24 #include "conf.h"
25 #include "device.h"
26 #include "names.h"
27 #include "net.h"
28 #include "logger.h"
29 #include "utils.h"
30 #include "route.h"
31 #include "xalloc.h"
32
33 static struct vdepluglib plug;
34 static struct vdeconn *conn = NULL;
35 static int port = 0;
36 static char *group = NULL;
37 static char *device_info;
38
39 static uint64_t device_total_in = 0;
40 static uint64_t device_total_out = 0;
41
42 static bool setup_device(void) {
43         libvdeplug_dynopen(plug);
44
45         if(!plug.dl_handle) {
46                 logger(DEBUG_ALWAYS, LOG_ERR, "Could not open libvdeplug library!");
47                 return false;
48         }
49
50         if(!get_config_string(lookup_config(config_tree, "Device"), &device))
51                 xasprintf(&device, LOCALSTATEDIR "/run/vde.ctl");
52
53         get_config_string(lookup_config(config_tree, "Interface"), &iface);
54
55         get_config_int(lookup_config(config_tree, "VDEPort"), &port);
56
57         get_config_string(lookup_config(config_tree, "VDEGroup"), &group);
58
59         device_info = "VDE socket";
60
61         struct vde_open_args args = {
62                 .port = port,
63                 .group = group,
64                 .mode = 0700,
65         };
66
67         conn = plug.vde_open(device, identname, &args);
68         if(!conn) {
69                 logger(DEBUG_ALWAYS, LOG_ERR, "Could not open VDE socket %s", device);
70                 return false;
71         }
72
73         device_fd = plug.vde_datafd(conn);
74
75 #ifdef FD_CLOEXEC
76         fcntl(device_fd, F_SETFD, FD_CLOEXEC);
77 #endif
78
79         logger(DEBUG_ALWAYS, LOG_INFO, "%s is a %s", device, device_info);
80
81         if(routing_mode == RMODE_ROUTER)
82                 overwrite_mac = true;
83
84         return true;
85 }
86
87 static void close_device(void) {
88         if(conn)
89                 plug.vde_close(conn);
90
91         if(plug.dl_handle)
92                 libvdeplug_dynclose(plug);
93
94         free(device);
95
96         free(iface);
97 }
98
99 static bool read_packet(vpn_packet_t *packet) {
100         int lenin = (ssize_t)plug.vde_recv(conn, packet->data, MTU, 0);
101         if(lenin <= 0) {
102                 logger(DEBUG_ALWAYS, LOG_ERR, "Error while reading from %s %s: %s", device_info, device, strerror(errno));
103                 event_exit();
104                 return false;
105         }
106
107         packet->len = lenin;
108         device_total_in += packet->len;
109         logger(DEBUG_TRAFFIC, LOG_DEBUG, "Read packet of %d bytes from %s", packet->len, device_info);
110
111         return true;
112 }
113
114 static bool write_packet(vpn_packet_t *packet) {
115         if((ssize_t)plug.vde_send(conn, packet->data, packet->len, 0) < 0) {
116                 if(errno != EINTR && errno != EAGAIN) {
117                         logger(DEBUG_ALWAYS, LOG_ERR, "Can't write to %s %s: %s", device_info, device, strerror(errno));
118                         event_exit();
119                 }
120
121                 return false;
122         }
123
124         device_total_out += packet->len;
125
126         return true;
127 }
128
129 static void dump_device_stats(void) {
130         logger(DEBUG_ALWAYS, LOG_DEBUG, "Statistics for %s %s:", device_info, device);
131         logger(DEBUG_ALWAYS, LOG_DEBUG, " total bytes in:  %10"PRIu64, device_total_in);
132         logger(DEBUG_ALWAYS, LOG_DEBUG, " total bytes out: %10"PRIu64, device_total_out);
133 }
134
135 const devops_t vde_devops = {
136         .setup = setup_device,
137         .close = close_device,
138         .read = read_packet,
139         .write = write_packet,
140         .dump_stats = dump_device_stats,
141 };