dgram/dgram.c

   1 /*
   2  */
   3 /*
   4  * dgram-socket create <local>                        => <sockid>
   5  * dgram-socket close <sockid>
   6  * dgram-socket transmit <sockid> <data> <remote>
   7  * dgram-socket on-receive <sockid> [<script>]
   8  *    calls, effectively,  eval <script> [list <data> <remote-addr> <socket>]
   9  *    if script not supplied, cancel
  10  */
  11
  12 #include "tables.h"
  13 #include "hbytes.h"
  14
  15 typedef struct DgramSocket {
  16   int ix; /* first ! */
  17   int fd, script_llength;
  18   Tcl_Interp *ip;
  19   Tcl_Obj *script;
  20   void *addr_buf, *msg_buf;
  21   int addr_buflen, msg_buflen;
  22 } DgramSocket;
  23
  24 IdDataTable dgram_socks= { "dgramsock" };
  25
  26 int do_dgram_socket_create(ClientData cd, Tcl_Interp *ip,
  27                            SockAddr_Value local, void **sock_r) {
  28   int fd, al, r;
  29   DgramSocket *sock;
  30   const struct sockaddr *sa;
  31
  32   sa= sockaddr_addr(&local);
  33   al= sockaddr_len(&local);
  34
  35   fd= socket(sa->sa_family, SOCK_DGRAM, 0);
  36   if (fd<0) return posixerr(ip,errno,"socket");
  37   r= bind(fd, sa, al);  if (r) return newfdposixerr(ip,fd,"bind");
  38   r= setnonblock(fd, 1);  if (r) return newfdposixerr(ip,fd,"setnonblock");
  39
  40   sock= TALLOC(sizeof(DgramSocket));
  41   sock->ix= -1;
  42   sock->fd= fd;
  43   sock->script= 0;
  44   sock->addr_buflen= al+1;
  45   sock->addr_buf= TALLOC(sock->addr_buflen);
  46   sock->msg_buflen= 0;
  47   sock->msg_buf= 0;
  48
  49   *sock_r= sock;
  50   return TCL_OK;
  51 }
  52
  53 int do_dgram_socket_transmit(ClientData cd, Tcl_Interp *ip,
  54                              void *sock_v, HBytes_Value data,
  55                              SockAddr_Value remote) {
  56   DgramSocket *sock= sock_v;
  57   int l, r;
  58
  59   r= sendto(sock->fd,
  60             hbytes_data(&data), l=hbytes_len(&data),
  61             0,
  62             sockaddr_addr(&remote), sockaddr_len(&remote));
  63   if (r==-1) return posixerr(ip,errno,"sendto");
  64   else if (r!=l) return staticerr(ip,"sendto gave wrong answer",0);
  65   return TCL_OK;
  66 }
  67
  68 static void cancel(DgramSocket *sock) {
  69   if (sock->script) {
  70     Tcl_DeleteFileHandler(sock->fd);
  71     Tcl_DecrRefCount(sock->script);
  72     sock->script= 0;
  73   }
  74 }
  75
  76 static void recv_call(ClientData sock_cd, int mask) {
  77   DgramSocket *sock= (void*)sock_cd;
  78   Tcl_Interp *ip= sock->ip;
  79   int sz, rc, i, peek;
  80   HBytes_Value message_val;
  81   SockAddr_Value peer_val;
  82   Tcl_Obj *args[3], *invoke;
  83   struct msghdr mh;
  84   struct iovec iov;
  85
  86   hbytes_empty(&message_val);
  87   sockaddr_clear(&peer_val);
  88   invoke=0; for (i=0; i<3; i++) args[i]=0;
  89
  90   mh.msg_iov= &iov;
  91   mh.msg_iovlen= 1;
  92   mh.msg_control= 0;
  93   mh.msg_controllen= 0;
  94   mh.msg_flags= 0;
  95
  96   peek= MSG_PEEK;
  97
  98   for (;;) {
  99     mh.msg_name= sock->addr_buf;
 100     mh.msg_namelen= sock->addr_buflen;
 101
 102     iov.iov_base= sock->msg_buf;
 103     iov.iov_len= sock->msg_buflen;
 104
 105     sz= recvmsg(sock->fd, &mh, peek);
 106     if (sz==-1) { rc=0; goto x_rc; }
 107
 108     assert(mh.msg_namelen < sock->addr_buflen);
 109
 110     if (!(mh.msg_flags & MSG_TRUNC)) {
 111       if (!peek) break;
 112       peek= 0;
 113       continue;
 114     }
 115
 116     TFREE(sock->msg_buf);
 117     sock->msg_buflen *= 2;
 118     sock->msg_buflen += 100;
 119     sock->msg_buf= TALLOC(sock->msg_buflen);
 120   }
 121
 122   hbytes_array(&message_val, iov.iov_base, sz);
 123   sockaddr_create(&peer_val, mh.msg_name, mh.msg_namelen);
 124
 125   args[0]= ret_hb(ip, message_val);  hbytes_empty(&message_val);
 126   args[1]= ret_sockaddr(ip, peer_val);  sockaddr_clear(&peer_val);
 127   args[2]= ret_iddata(ip, sock, &dgram_socks);
 128   for (i=0; i<3; i++) Tcl_IncrRefCount(args[i]);
 129
 130   invoke= Tcl_DuplicateObj(sock->script);
 131   Tcl_IncrRefCount(invoke);
 132
 133   rc= Tcl_ListObjReplace(ip,invoke,sock->script_llength,0,3,args);
 134   for (i=0; i<3; i++) { Tcl_DecrRefCount(args[i]); args[i]= 0; }
 135   if (rc) goto x_rc;
 136
 137   rc= Tcl_EvalObjEx(ip,invoke,TCL_EVAL_GLOBAL|TCL_EVAL_DIRECT);
 138
 139 x_rc:
 140   if (invoke) Tcl_DecrRefCount(invoke);
 141
 142   if (rc)
 143     Tcl_BackgroundError(ip);
 144 }
 145
 146 int do_dgram_socket_on_receive(ClientData cd, Tcl_Interp *ip,
 147                                void *sock_v, Tcl_Obj *script) {
 148   DgramSocket *sock= sock_v;
 149   int rc;
 150
 151   if (script) {
 152     rc= Tcl_ListObjLength(ip, script, &sock->script_llength);
 153     if (rc) return rc;
 154   }
 155
 156   cancel(sock);
 157   if (script) {
 158     Tcl_IncrRefCount(script);
 159     sock->script= script;
 160     sock->ip= ip;
 161   }
 162   Tcl_CreateFileHandler(sock->fd, TCL_READABLE, recv_call, sock);
 163   return TCL_OK;
 164 }
 165
 166 int do_dgram_socket_close(ClientData cd, Tcl_Interp *ip, void *sock_v) {
 167   DgramSocket *sock= sock_v;
 168   int sockix;
 169   cancel(sock);
 170   close(sock->fd); /* nothing useful to be done with errors */
 171   sockix= sock->ix;
 172   TFREE(sock->addr_buf);
 173   TFREE(sock->msg_buf);
 174   TFREE(sock);
 175   dgram_socks.a[sockix]= 0;
 176   return TCL_OK;
 177 }