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
  56         get_config_string(lookup_config(config_tree, "Interface"), &iface);
  57
  58         get_config_int(lookup_config(config_tree, "VDEPort"), &port);
  59
  60         get_config_string(lookup_config(config_tree, "VDEGroup"), &group);
  61
  62         device_info = "VDE socket";
  63
  64         struct vde_open_args args = {
  65                 .port = port,
  66                 .group = group,
  67                 .mode = 0700,
  68         };
  69
  70         conn = plug.vde_open(device, identname, &args);
  71
  72         if(!conn) {
  73                 logger(LOG_ERR, "Could not open VDE socket %s", device);
  74                 return false;
  75         }
  76
  77         device_fd = plug.vde_datafd(conn);
  78
  79 #ifdef FD_CLOEXEC
  80         fcntl(device_fd, F_SETFD, FD_CLOEXEC);
  81 #endif
  82
  83         logger(LOG_INFO, "%s is a %s", device, device_info);
  84
  85         if(routing_mode == RMODE_ROUTER) {
  86                 overwrite_mac = true;
  87         }
  88
  89         return true;
  90 }
  91
  92 static void close_device(void) {
  93         if(conn) {
  94                 plug.vde_close(conn);
  95         }
  96
  97         if(plug.dl_handle) {
  98                 libvdeplug_dynclose(plug);
  99         }
 100
 101         free(device);
 102
 103         free(iface);
 104 }
 105
 106 static bool read_packet(vpn_packet_t *packet) {
 107         int lenin = (ssize_t)plug.vde_recv(conn, packet->data, MTU, 0);
 108
 109         if(lenin <= 0) {
 110                 logger(LOG_ERR, "Error while reading from %s %s: %s", device_info, device, strerror(errno));
 111                 running = false;
 112                 return false;
 113         }
 114
 115         packet->len = lenin;
 116         device_total_in += packet->len;
 117         ifdebug(TRAFFIC) logger(LOG_DEBUG, "Read packet of %d bytes from %s", packet->len, device_info);
 118
 119         return true;
 120 }
 121
 122 static bool write_packet(vpn_packet_t *packet) {
 123         if((ssize_t)plug.vde_send(conn, packet->data, packet->len, 0) < 0) {
 124                 if(errno != EINTR && errno != EAGAIN) {
 125                         logger(LOG_ERR, "Can't write to %s %s: %s", device_info, device, strerror(errno));
 126                         running = false;
 127                 }
 128
 129                 return false;
 130         }
 131
 132         device_total_out += packet->len;
 133
 134         return true;
 135 }
 136
 137 static void dump_device_stats(void) {
 138         logger(LOG_DEBUG, "Statistics for %s %s:", device_info, device);
 139         logger(LOG_DEBUG, " total bytes in:  %10"PRIu64, device_total_in);
 140         logger(LOG_DEBUG, " total bytes out: %10"PRIu64, device_total_out);
 141 }
 142
 143 const devops_t vde_devops = {
 144         .setup = setup_device,
 145         .close = close_device,
 146         .read = read_packet,
 147         .write = write_packet,
 148         .dump_stats = dump_device_stats,
 149 };