src/vde_device.c

   1 /*
   2     device.c -- VDE plug
   3     Copyright (C) 2012 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 "net.h"
  27 #include "logger.h"
  28 #include "utils.h"
  29 #include "route.h"
  30 #include "xalloc.h"
  31
  32 static struct vdepluglib plug;
  33 static struct vdeconn *conn = NULL;
  34 static int port = 0;
  35 static char *group = NULL;
  36 static char *device_info;
  37
  38 extern char *identname;
  39 extern volatile bool running;
  40
  41 static uint64_t device_total_in = 0;
  42 static uint64_t device_total_out = 0;
  43
  44 static bool setup_device(void) {
  45         libvdeplug_dynopen(plug);
  46
  47         if(!plug.dl_handle) {
  48                 logger(LOG_ERR, "Could not open libvdeplug library!");
  49                 return false;
  50         }
  51
  52         if(!get_config_string(lookup_config(config_tree, "Device"), &device))
  53                 xasprintf(&device, LOCALSTATEDIR "/run/vde.ctl");
  54
  55         get_config_string(lookup_config(config_tree, "Interface"), &iface);
  56
  57         get_config_int(lookup_config(config_tree, "VDEPort"), &port);
  58
  59         get_config_string(lookup_config(config_tree, "VDEGroup"), &group);
  60
  61         device_info = "VDE socket";
  62
  63         struct vde_open_args args = {
  64                 .port = port,
  65                 .group = group,
  66                 .mode = 0700,
  67         };
  68
  69         conn = plug.vde_open(device, identname, &args);
  70         if(!conn) {
  71                 logger(LOG_ERR, "Could not open VDE socket %s", device);
  72                 return false;
  73         }
  74
  75         device_fd = plug.vde_datafd(conn);
  76
  77 #ifdef FD_CLOEXEC
  78         fcntl(device_fd, F_SETFD, FD_CLOEXEC);
  79 #endif
  80
  81         logger(LOG_INFO, "%s is a %s", device, device_info);
  82
  83         if(routing_mode == RMODE_ROUTER)
  84                 overwrite_mac = true;
  85
  86         return true;
  87 }
  88
  89 static void close_device(void) {
  90         if(conn)
  91                 plug.vde_close(conn);
  92
  93         if(plug.dl_handle)
  94                 libvdeplug_dynclose(plug);
  95
  96         free(device);
  97
  98         free(iface);
  99 }
 100
 101 static bool read_packet(vpn_packet_t *packet) {
 102         int lenin = plug.vde_recv(conn, packet->data, MTU, 0);
 103         if(lenin <= 0) {
 104                 logger(LOG_ERR, "Error while reading from %s %s: %s", device_info, device, strerror(errno));
 105                 running = false;
 106                 return false;
 107         }
 108
 109         packet->len = lenin;
 110         device_total_in += packet->len;
 111         ifdebug(TRAFFIC) logger(LOG_DEBUG, "Read packet of %d bytes from %s", packet->len, device_info);
 112
 113         return true;
 114 }
 115
 116 static bool write_packet(vpn_packet_t *packet) {
 117         if(plug.vde_send(conn, packet->data, packet->len, 0) < 0) {
 118                 if(errno != EINTR && errno != EAGAIN) {
 119                         logger(LOG_ERR, "Can't write to %s %s: %s", device_info, device, strerror(errno));
 120                         running = false;
 121                 }
 122
 123                 return false;
 124         }
 125
 126         device_total_out += packet->len;
 127
 128         return true;
 129 }
 130
 131 static void dump_device_stats(void) {
 132         logger(LOG_DEBUG, "Statistics for %s %s:", device_info, device);
 133         logger(LOG_DEBUG, " total bytes in:  %10"PRIu64, device_total_in);
 134         logger(LOG_DEBUG, " total bytes out: %10"PRIu64, device_total_out);
 135 }
 136
 137 const devops_t vde_devops = {
 138         .setup = setup_device,
 139         .close = close_device,
 140         .read = read_packet,
 141         .write = write_packet,
 142         .dump_stats = dump_device_stats,
 143 };