planetlab/pltap-ovs/pltap-ovs.c

   1 #include <sys/un.h>
   2 #include <sys/types.h>
   3 #include <sys/stat.h>
   4 #include <fcntl.h>
   5 #include <stdlib.h>
   6 #include <stdio.h>
   7 #include <string.h>
   8 #include <unistd.h>
   9 #include <sys/socket.h>
  10 #include <errno.h>
  11 #include <linux/if.h>
  12 #include <linux/if_tun.h>
  13
  14 #include "tunalloc.h"
  15
  16 #define OVS_SOCK "/var/run/pl-ovs.control"
  17
  18 char *appname;
  19
  20 #define ERROR(msg)                                                              \
  21         do {                                                                    \
  22                 fprintf(stderr, "%s: %s: %s", appname, msg, strerror(errno));   \
  23                 exit(1);                                                        \
  24         } while (0)
  25
  26
  27 int send_vif_fd(int sock_fd, int vif_fd, char *vif_name)
  28 {
  29         int retval;
  30         struct msghdr msg;
  31         struct cmsghdr *p_cmsg;
  32         struct iovec vec;
  33         size_t cmsgbuf[CMSG_SPACE(sizeof(vif_fd)) / sizeof(size_t)];
  34         int *p_fds;
  35
  36
  37         msg.msg_control = cmsgbuf;
  38         msg.msg_controllen = sizeof(cmsgbuf);
  39         p_cmsg = CMSG_FIRSTHDR(&msg);
  40         p_cmsg->cmsg_level = SOL_SOCKET;
  41         p_cmsg->cmsg_type = SCM_RIGHTS;
  42         p_cmsg->cmsg_len = CMSG_LEN(sizeof(vif_fd));
  43         p_fds = (int *) CMSG_DATA(p_cmsg);
  44         *p_fds = vif_fd;
  45         msg.msg_controllen = p_cmsg->cmsg_len;
  46         msg.msg_name = NULL;
  47         msg.msg_namelen = 0;
  48         msg.msg_iov = &vec;
  49         msg.msg_iovlen = 1;
  50         msg.msg_flags = 0;
  51
  52         /* Send the interface name as the iov */
  53         vec.iov_base = vif_name;
  54         vec.iov_len = strlen(vif_name)+1;
  55
  56         while ((retval = sendmsg(sock_fd, &msg, 0)) == -1 && errno == EINTR);
  57         if (retval == -1) {
  58                 ERROR("sending file descriptor");
  59         }
  60         return 0;
  61 }
  62
  63 void send_fd(int p, int fd, char* vif_name)
  64 {
  65         int control_fd;
  66         int accept_fd;
  67         struct sockaddr_un addr, accept_addr;
  68         socklen_t addr_len = sizeof(accept_addr);
  69
  70         control_fd = socket(AF_UNIX, SOCK_STREAM, 0);
  71         if (control_fd == -1 && errno != ENOENT) {
  72                 ERROR("Could not create UNIX socket");
  73         }
  74
  75         memset(&addr, 0, sizeof(struct sockaddr_un));
  76         /* Clear structure */
  77         addr.sun_family = AF_UNIX;
  78         strncpy(addr.sun_path, OVS_SOCK,
  79                         sizeof(addr.sun_path) - 1);
  80
  81         if (unlink(OVS_SOCK) == -1 && errno != ENOENT) {
  82                 ERROR("Could not unlink " OVS_SOCK " control socket");
  83         }
  84
  85         if (bind(control_fd, (struct sockaddr *) &addr,
  86                                 sizeof(struct sockaddr_un)) == -1) {
  87                 ERROR("Could not bind to " OVS_SOCK " control socket");
  88         }
  89
  90         if (listen(control_fd, 5) == -1) {
  91                 ERROR("listen on " OVS_SOCK " failed");
  92         }
  93         if (write(p, "1", 1) != 1) {
  94                 ERROR("writing on the synch pipe");
  95         }
  96         if ((accept_fd = accept(control_fd, (struct sockaddr*) &accept_addr,
  97                                                 &addr_len)) == -1) {
  98                 ERROR("accept on " OVS_SOCK " failed");
  99         }
 100         send_vif_fd(accept_fd, fd, vif_name);
 101 }
 102
 103 int main(int argc, char* argv[])
 104 {
 105         char if_name[IFNAMSIZ];
 106         int p[2]; // synchronization pipe
 107         char dummy;
 108
 109         if (pipe(p) < 0) {
 110                 ERROR("pipe");
 111         }
 112
 113         int tun_fd = tun_alloc(IFF_TAP, if_name);
 114
 115         appname = argv[0];
 116
 117         switch(fork()) {
 118         case -1:
 119                 ERROR("fork");
 120                 exit(1);
 121         case 0:
 122                 close(1);
 123                 open("/dev/null", O_WRONLY);
 124                 close(p[0]);
 125                 send_fd(p[1], tun_fd, if_name);
 126                 exit(0);
 127         default:
 128                 close(p[1]);
 129                 if (read(p[0], &dummy, 1) != 1) {
 130                         ERROR("reading from the synch pipe");
 131                 }
 132                 printf("%s\n", if_name);
 133         }
 134         return 0;
 135 }