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/tun/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         int i;
  70
  71         control_fd = socket(AF_UNIX, SOCK_STREAM, 0);
  72         if (control_fd == -1 && errno != ENOENT) {
  73                 ERROR("Could not create UNIX socket");
  74         }
  75
  76         memset(&addr, 0, sizeof(struct sockaddr_un));
  77         /* Clear structure */
  78         addr.sun_family = AF_UNIX;
  79         strncpy(addr.sun_path, OVS_SOCK,
  80                         sizeof(addr.sun_path) - 1);
  81
  82         if (unlink(OVS_SOCK) == -1 && errno != ENOENT) {
  83                 ERROR("Could not unlink " OVS_SOCK " control socket");
  84         }
  85
  86         if (bind(control_fd, (struct sockaddr *) &addr,
  87                                 sizeof(struct sockaddr_un)) == -1) {
  88                 ERROR("Could not bind to " OVS_SOCK " control socket");
  89         }
  90
  91         if (listen(control_fd, 5) == -1) {
  92                 ERROR("listen on " OVS_SOCK " failed");
  93         }
  94         if (write(p, "1", 1) != 1) {
  95                 ERROR("writing on the synch pipe");
  96         }
  97         if ((accept_fd = accept(control_fd, (struct sockaddr*) &accept_addr,
  98                                                 &addr_len)) == -1) {
  99                 ERROR("accept on " OVS_SOCK " failed");
 100         }
 101         send_vif_fd(accept_fd, fd, vif_name);
 102 }
 103
 104 int main(int argc, char* argv[])
 105 {
 106         char if_name[IFNAMSIZ];
 107         int p[2]; // synchronization pipe
 108         char dummy;
 109
 110         if (pipe(p) < 0) {
 111                 ERROR("pipe");
 112         }
 113
 114         int tun_fd = tun_alloc(IFF_TAP, if_name);
 115
 116         appname = argv[0];
 117
 118         switch(fork()) {
 119         case -1:
 120                 ERROR("fork");
 121                 exit(1);
 122         case 0:
 123                 close(1);
 124                 open("/dev/null", O_WRONLY);
 125                 close(p[0]);
 126                 send_fd(p[1], tun_fd, if_name);
 127                 exit(0);
 128         default:
 129                 close(p[1]);
 130                 if (read(p[0], &dummy, 1) != 1) {
 131                         ERROR("reading from the synch pipe");
 132                 }
 133                 printf("%s\n", if_name);
 134         }
 135         return 0;
 136 }