tuntap/tuntap.c

   1 /*
   2  */
   3 /*
   4  * tuntap-socket create-ptp <local-addr> <peer-addr> <mtu> [<ifname>]
   5  *               => <sockid>
   6  * tuntap-socket close <sockid>
   7  * tuntap-socket ifname <sockid>
   8  * tuntap-socket receive <sockid> <data>
   9  * tuntap-socket on-transmit <sockid> [<script>]
  10  *    calls, effectively,  eval <script> [list <data> <socket>]
  11  *    if script not supplied, cancel
  12  */
  13
  14 #include "tables.h"
  15 #include "hbytes.h"
  16
  17 typedef struct TunSocket {
  18   int ix, fd, script_llength;
  19   Tcl_Interp *ip;
  20   Tcl_Obj *script;
  21   void msg_buf;
  22   int mtu;
  23   char *ifname;
  24 } TuntapSocket;
  25
  26 IdDataTable tuntap_socks= { "tuntap" };
  27
  28 static int sockfail(Tcl_Interp *ip, int fd, const char *m) {
  29   int e;
  30   e= errno;
  31   close(fd);
  32   return posixerr(ip,e,m);
  33 }
  34
  35 int do_tuntap_socket_create_ptp(ClientData cd, Tcl_Interp *ip,
  36                                 SockAddr_Value local, SockAddr_Value peer,
  37                                 long mtu, const char *ifname,
  38                                 void **sock_r) {
  39   int fd, local_al, peer_al, r;
  40   struct ifreq ifr;
  41   DgramSocket *sock;
  42   const struct sockaddr *local_sa, *peer_sa;
  43
  44   local_sa= sockaddr_addr(&local);
  45   local_al= sockaddr_len(&local);
  46
  47   peer_sa= sockaddr_addr(&peer);
  48   peer_al= sockaddr_len(&peer);
  49
  50   if (local_sa != AF_INET || local_al != sizeof(struct in_addr) ||
  51       peer_sa != AF_INET || peer_al != sizeof(struct in_addr))
  52     return staticerr(ip,"tuntap not IPv4");
  53
  54   memset(&ifr,0,sizeof(ifr));
  55   ifr.ifr_flags= IFF_TUN | IFF_NO_PI;
  56
  57   if (ifname) {
  58     if (strlen(ifname) > IFNAMSIZ-1)
  59       return staticerr(ip,"tun interface name too long");
  60     strcpy(ifr.ifr_name, ifname);
  61   }
  62
  63   fd= open("/dev/net/tun", O_RDWR);
  64   if (fd<0) return posixerr(ip,errno,"open /dev/net/tun");
  65
  66   r= ioctl(fd, TUNSETIFF, (void*)&ifr);
  67   if (r) return posixerr(ip,errno,"ioctl TUNSETIFF");
  68
  69
  70
  71   r=
  72   r= bind(fd, sa, al);  if (r) return sockfail(ip,fd,"bind");
  73   r= setnonblock(fd, 1);  if (r) return sockfail(ip,fd,"setnonblock");
  74
  75   sock= TALLOC(sizeof(DgramSocket));
  76   sock->ix= -1;
  77   sock->fd= fd;
  78   sock->script= 0;
  79   sock->addr_buflen= al+1;
  80   sock->addr_buf= TALLOC(sock->addr_buflen);
  81   sock->msg_buflen= 0;
  82   sock->msg_buf= 0;
  83
  84   *sock_r= sock;
  85   return TCL_OK;
  86 }
  87
  88 int do_tuntap_socket_transmit(ClientData cd, Tcl_Interp *ip,
  89                              TuntapSocket *sock, HBytes_Value data,
  90                              SockAddr_Value remote) {
  91   int l, r;
  92
  93   r= sendto(sock->fd,
  94             hbytes_data(&data), l=hbytes_len(&data),
  95             0,
  96             sockaddr_addr(&remote), sockaddr_len(&remote));
  97   if (r==-1) return posixerr(ip,errno,"sendto");
  98   else if (r!=l) return staticerr(ip,"sendto gave wrong answer",0);
  99   return TCL_OK;
 100 }
 101
 102 static void cancel(TuntapSocket *sock) {
 103   if (sock->script) {
 104     Tcl_DeleteFileHandler(sock->fd);
 105     Tcl_DecrRefCount(sock->script);
 106     sock->script= 0;
 107   }
 108 }
 109
 110 static void recv_call(ClientData sock_cd, int mask) {
 111   TuntapSocket *sock= (void*)sock_cd;
 112   Tcl_Interp *ip= sock->ip;
 113   int sz, rc, i, peek;
 114   HBytes_Value message_val;
 115   SockAddr_Value peer_val;
 116   Tcl_Obj *args[3], *invoke;
 117   struct msghdr mh;
 118   struct iovec iov;
 119
 120   hbytes_empty(&message_val);
 121   sockaddr_clear(&peer_val);
 122   invoke=0; for (i=0; i<3; i++) args[i]=0;
 123
 124   mh.msg_iov= &iov;
 125   mh.msg_iovlen= 1;
 126   mh.msg_control= 0;
 127   mh.msg_controllen= 0;
 128   mh.msg_flags= 0;
 129
 130   peek= MSG_PEEK;
 131
 132   for (;;) {
 133     mh.msg_name= sock->addr_buf;
 134     mh.msg_namelen= sock->addr_buflen;
 135
 136     iov.iov_base= sock->msg_buf;
 137     iov.iov_len= sock->msg_buflen;
 138
 139     sz= recvmsg(sock->fd, &mh, peek);
 140     if (sz==-1) { rc=0; goto x_rc; }
 141
 142     assert(mh.msg_namelen < sock->addr_buflen);
 143
 144     if (!(mh.msg_flags & MSG_TRUNC)) {
 145       if (!peek) break;
 146       peek= 0;
 147       continue;
 148     }
 149
 150     TFREE(sock->msg_buf);
 151     sock->msg_buflen *= 2;
 152     sock->msg_buflen += 100;
 153     sock->msg_buf= TALLOC(sock->msg_buflen);
 154   }
 155
 156   hbytes_array(&message_val, iov.iov_base, sz);
 157   sockaddr_create(&peer_val, mh.msg_name, mh.msg_namelen);
 158
 159   args[0]= ret_hb(ip, message_val);  hbytes_empty(&message_val);
 160   args[1]= ret_sockaddr(ip, peer_val);  sockaddr_clear(&peer_val);
 161   args[2]= ret_sockid(ip, sock);
 162   for (i=0; i<3; i++) Tcl_IncrRefCount(args[i]);
 163
 164   invoke= Tcl_DuplicateObj(sock->script);
 165   Tcl_IncrRefCount(invoke);
 166
 167   rc= Tcl_ListObjReplace(ip,invoke,sock->script_llength,0,3,args);
 168   for (i=0; i<3; i++) { Tcl_DecrRefCount(args[i]); args[i]= 0; }
 169   if (rc) goto x_rc;
 170
 171   rc= Tcl_EvalObjEx(ip,invoke,TCL_EVAL_GLOBAL|TCL_EVAL_DIRECT);
 172
 173 x_rc:
 174   if (invoke) Tcl_DecrRefCount(invoke);
 175
 176   if (rc)
 177     Tcl_BackgroundError(ip);
 178 }
 179
 180 int do_tuntap_socket_on_receive(ClientData cd, Tcl_Interp *ip,
 181                                TuntapSocket *sock, Tcl_Obj *script) {
 182   int rc;
 183
 184   if (script) {
 185     rc= Tcl_ListObjLength(ip, script, &sock->script_llength);
 186     if (rc) return rc;
 187   }
 188
 189   cancel(sock);
 190   if (script) {
 191     Tcl_IncrRefCount(script);
 192     sock->script= script;
 193     sock->ip= ip;
 194   }
 195   Tcl_CreateFileHandler(sock->fd, TCL_READABLE, recv_call, sock);
 196   return TCL_OK;
 197 }
 198
 199 int do_tuntap_socket_close(ClientData cd, Tcl_Interp *ip, TuntapSocket *sock) {
 200   int sockix;
 201   cancel(sock);
 202   close(sock->fd); /* nothing useful to be done with errors */
 203   sockix= sock->ix;
 204   TFREE(sock->addr_buf);
 205   TFREE(sock->msg_buf);
 206   TFREE(sock);
 207   socks[sockix]= 0;
 208   return TCL_OK;
 209 }
 210
 211 /* Arg parsing */
 212
 213 int pat_sockid(Tcl_Interp *ip, Tcl_Obj *o, TuntapSocket **val) {
 214   int rc, sockix;
 215   TuntapSocket *sock;
 216
 217   rc= Tcl_ConvertToType(ip,o,&tuntapsockid_type);
 218   if (rc) return rc;
 219
 220   sockix= o->internalRep.longValue;
 221   if (sockix >= n_socks || !(sock= socks[sockix]))
 222     return staticerr(ip,"tuntap socket not open",0);
 223
 224   assert(socks[sockix]->ix == sockix);
 225
 226   *val= sock;
 227   return TCL_OK;
 228 }
 229
 230 Tcl_Obj *ret_sockid(Tcl_Interp *ip, TuntapSocket *val) {
 231   Tcl_Obj *o;
 232
 233   o= Tcl_NewObj();
 234   Tcl_InvalidateStringRep(o);
 235   o->internalRep.longValue= val->ix;
 236   o->typePtr= &tuntapsockid_type;
 237   return o;
 238 }
 239
 240 static void sockid_t_free(Tcl_Obj *o) { }
 241
 242 static void sockid_t_dup(Tcl_Obj *src, Tcl_Obj *dup) {
 243   dup->internalRep= src->internalRep;
 244   dup->typePtr= &tuntapsockid_type;
 245 }
 246
 247 static void sockid_t_ustr(Tcl_Obj *o) {
 248   char buf[75];
 249
 250   snprintf(buf,sizeof(buf), "%d", (int)o->internalRep.longValue);
 251   obj_updatestr_vstringls(o,
 252                           "tuntapsock",9,
 253                           buf, strlen(buf),
 254                           (char*)0);
 255 }
 256
 257 static int sockid_t_sfa(Tcl_Interp *ip, Tcl_Obj *o) {
 258   unsigned long ul;
 259   char *ep, *str;
 260
 261   str= Tcl_GetStringFromObj(o,0);
 262   if (memcmp(str,"tuntapsock",9)) return staticerr(ip,"bad tuntap socket id",0);
 263   errno=0; ul=strtoul(str+9,&ep,10);
 264   if (errno || *ep) return staticerr(ip,"bad tuntap socket id number",0);
 265   if (ul > INT_MAX) return staticerr(ip,"out of range tuntap socket id",0);
 266
 267   objfreeir(o);
 268   o->internalRep.longValue= ul;
 269   o->typePtr= &tuntapsockid_type;
 270   return TCL_OK;
 271 }
 272
 273 Tcl_ObjType tuntapsockid_type = {
 274   "tuntapsockid",
 275   sockid_t_free, sockid_t_dup, sockid_t_ustr, sockid_t_sfa
 276 };