tuntap/tuntap.c

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