vde-2/src/vdetaplib/vdetap.c

   1 /* Copyright 2004 Renzo Davoli
   2  * Reseased under the GPLv2 */
   3
   4 #define _GNU_SOURCE
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include <unistd.h>
   8 #include <stdint.h>
   9 #include <sys/select.h>
  10 #include <sys/socket.h>
  11 #include <sys/un.h>
  12 #include <pwd.h>
  13
  14 #include <config.h>
  15 #include <vde.h>
  16 #include <vdecommon.h>
  17 #include <libvdeplug.h>
  18
  19 #define SWITCH_MAGIC 0xfeedface
  20 #define BUFSIZE 2048
  21 #define MAXDESCR 128
  22
  23 VDECONN *conn;
  24
  25 static unsigned char bufin[BUFSIZE];
  26
  27 static struct pollfd pollv[]={{0,POLLIN|POLLHUP,0},{0,POLLIN|POLLHUP,0}};
  28
  29 int main(int argc,char *argv[])
  30 {
  31         int fd,fddata;
  32         struct sockaddr_un datain;
  33         /*struct sockaddr_un dataout;*/
  34         socklen_t datainsize;
  35         int result,nx;
  36         register int i;
  37         struct vde_open_args open_args={.port=0,.group=NULL,.mode=0700};
  38         char *descr;
  39         /*printf("argc = %d\n",argc);
  40         for (i=0;i<argc;i++)
  41                 printf("argv %d -%s-\n",i,argv[i]);*/
  42         if (argc < 4 && argv[0][0] != '-') {
  43                 fprintf(stderr,"vdetap must be activated by libvdetap e.g.\n"
  44                                 "   sh%% export LD_PRELOAD=%s/libvdetap.so\n"
  45                                 "   csh%% setenv LD_PRELOAD %s/libvdetap.so\n", LIBEXECDIR, LIBEXECDIR);
  46                 exit (-1);
  47         }
  48
  49         fd = atoi(argv[1]);
  50         for (i=3;i<FD_SETSIZE;i++)
  51                 if (i != fd)
  52                         close(i);
  53         if((fddata = socket(AF_UNIX, SOCK_DGRAM, 0)) < 0){
  54                 perror("socket");
  55                 exit(1);
  56         }
  57         if (argc == 8) {
  58                 open_args.port=atoi(argv[5]);
  59                 open_args.group=argv[6];
  60           sscanf(argv[7],"%i",&open_args.mode);
  61                 //fprintf(stderr,"|%d|%s|%o|\n",open_args.port,open_args.group,open_args.mode);
  62                 asprintf(&descr,"tuntaplib %s/%s",argv[4],argv[3]);
  63         } else
  64                 descr="tuntaplib";
  65         conn=vde_open(argv[2],descr,&open_args);
  66         pollv[0].fd=fd;
  67         pollv[1].fd=vde_datafd(conn);
  68         for(;;) {
  69                 result=poll(pollv,2,-1);
  70                 if (pollv[0].revents & POLLHUP || pollv[1].revents & POLLHUP)
  71                         break;
  72                 if (pollv[0].revents & POLLIN) {
  73                         nx=read(fd,bufin,sizeof(bufin));
  74                         /*fprintf(stderr,"RX from pgm %d\n",nx);*/
  75                         //send(connected_fd,bufin,nx,0);
  76                         vde_send(conn,bufin,nx,0);
  77                 }
  78                 if (pollv[1].revents & POLLIN) {
  79                         datainsize=sizeof(datain);
  80                         nx=vde_recv(conn,bufin,BUFSIZE,0);
  81                         /*fprintf(stderr,"TX to pgm %d\n",nx);*/
  82                         write(fd,bufin,nx);
  83                 }
  84         }
  85         vde_close(conn);
  86         return 0;
  87 }
  88