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 "dgram.h"
  13
  14 typedef struct DgramSocket {
  15   int ix; /* first ! */
  16   int fd;
  17   Tcl_Interp *ip;
  18   ScriptToInvoke script;
  19   void *addr_buf, *msg_buf;
  20   int addr_buflen, msg_buflen;
  21 } DgramSocket;
  22
  23 int cht_do_dgramsocket_create(ClientData cd, Tcl_Interp *ip,
  24                            SockAddr_Value local, void **sock_r) {
  25   int fd, al, r;
  26   DgramSocket *sock;
  27   const struct sockaddr *sa;
  28
  29   sa= cht_sockaddr_addr(&local);
  30   al= cht_sockaddr_len(&local);
  31
  32   fd= socket(sa->sa_family, SOCK_DGRAM, 0);
  33   if (fd<0) return cht_posixerr(ip,errno,"socket");
  34   r= bind(fd, sa, al);  if (r) return cht_newfdposixerr(ip,fd,"bind");
  35   r= cht_setnonblock(fd, 1);  if (r) return cht_newfdposixerr(ip,fd,"setnonblock");
  36
  37   sock= TALLOC(sizeof(DgramSocket));
  38   sock->ix= -1;
  39   sock->fd= fd;
  40   sock->addr_buflen= al+1;
  41   sock->addr_buf= TALLOC(sock->addr_buflen);
  42   sock->msg_buflen= 0;
  43   sock->msg_buf= 0;
  44   cht_scriptinv_init(&sock->script);
  45
  46   *sock_r= sock;
  47   return TCL_OK;
  48 }
  49
  50 int cht_do_dgramsocket_transmit(ClientData cd, Tcl_Interp *ip,
  51                              void *sock_v, HBytes_Value data,
  52                              SockAddr_Value remote) {
  53   DgramSocket *sock= sock_v;
  54   int l, r;
  55
  56   r= sendto(sock->fd,
  57             cht_hb_data(&data), l=cht_hb_len(&data),
  58             0,
  59             cht_sockaddr_addr(&remote), cht_sockaddr_len(&remote));
  60   if (r==-1) return cht_posixerr(ip,errno,"sendto");
  61   else if (r!=l) return cht_staticerr(ip,"sendto gave wrong answer",0);
  62   return TCL_OK;
  63 }
  64
  65 static void cancel(DgramSocket *sock) {
  66   if (sock->script.script) {
  67     cht_scriptinv_cancel(&sock->script);
  68     Tcl_DeleteFileHandler(sock->fd);
  69   }
  70 }
  71
  72 static void recv_call(ClientData sock_cd, int mask) {
  73   DgramSocket *sock= (void*)sock_cd;
  74   Tcl_Interp *ip= sock->script.ipq;
  75   int sz, rc, peek;
  76   HBytes_Value message_val;
  77   SockAddr_Value peer_val;
  78   Tcl_Obj *args[3];
  79   struct msghdr mh;
  80   struct iovec iov;
  81
  82   cht_hb_empty(&message_val);
  83   cht_sockaddr_clear(&peer_val);
  84
  85   mh.msg_iov= &iov;
  86   mh.msg_iovlen= 1;
  87   mh.msg_control= 0;
  88   mh.msg_controllen= 0;
  89   mh.msg_flags= 0;
  90
  91   peek= MSG_PEEK;
  92
  93   for (;;) {
  94     mh.msg_name= sock->addr_buf;
  95     mh.msg_namelen= sock->addr_buflen;
  96
  97     iov.iov_base= sock->msg_buf;
  98     iov.iov_len= sock->msg_buflen;
  99
 100     sz= recvmsg(sock->fd, &mh, peek);
 101     if (sz==-1) {
 102       if (errno == EAGAIN || errno == EWOULDBLOCK) rc=0;
 103       else rc= cht_posixerr(ip,errno,"recvmsg");
 104       goto x_rc;
 105     }
 106
 107     assert(mh.msg_namelen < sock->addr_buflen);
 108
 109     if (!(mh.msg_flags & MSG_TRUNC)) {
 110       if (!peek) break;
 111       peek= 0;
 112       continue;
 113     }
 114
 115     TFREE(sock->msg_buf);
 116     assert(sock->msg_buflen < INT_MAX/4);
 117     sock->msg_buflen *= 2;
 118     sock->msg_buflen += 100;
 119     sock->msg_buf= TALLOC(sock->msg_buflen);
 120   }
 121
 122   cht_hb_array(&message_val, iov.iov_base, sz);
 123   cht_sockaddr_create(&peer_val, mh.msg_name, mh.msg_namelen);
 124
 125   args[0]= cht_ret_hb(ip, message_val);  cht_hb_empty(&message_val);
 126   args[1]= cht_ret_sockaddr(ip, peer_val);  cht_sockaddr_clear(&peer_val);
 127   args[2]= cht_ret_iddata(ip, sock, &cht_dgram_socks);
 128   cht_scriptinv_invoke(&sock->script,3,args);
 129
 130   rc= 0;
 131
 132 x_rc:
 133   if (rc)
 134     Tcl_BackgroundError(ip);
 135 }
 136
 137 int cht_do_dgramsocket_on_receive(ClientData cd, Tcl_Interp *ip,
 138                                void *sock_v, Tcl_Obj *newscript) {
 139   DgramSocket *sock= sock_v;
 140   int rc;
 141
 142   cancel(sock);
 143
 144   if (newscript) {
 145     rc= cht_scriptinv_set(&sock->script, ip, newscript, 0);
 146     if (rc) return rc;
 147   }
 148
 149   Tcl_CreateFileHandler(sock->fd, TCL_READABLE, recv_call, sock);
 150   return TCL_OK;
 151 }
 152
 153 static void destroy(DgramSocket *sock) {
 154   cancel(sock);
 155   close(sock->fd); /* nothing useful to be done with errors */
 156   TFREE(sock->addr_buf);
 157   TFREE(sock->msg_buf);
 158   TFREE(sock);
 159 }
 160
 161 static void destroy_idtabcb(Tcl_Interp *ip, void *sock_v) {
 162   destroy(sock_v);
 163 }
 164
 165 int cht_do_dgramsocket_close(ClientData cd, Tcl_Interp *ip, void *sock_v) {
 166   cht_tabledataid_disposing(ip,sock_v,&cht_dgram_socks);
 167   destroy(sock_v);
 168   return TCL_OK;
 169 }
 170
 171 const IdDataSpec cht_dgram_socks= {
 172   "dgramsock", "dgramsock-table", destroy_idtabcb
 173 };