X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ian/git?p=chiark-tcl.git;a=blobdiff_plain;f=dgram%2Fdgram.c;h=b2661a4fe82611792a359e02fb25a12ee3c713ff;hp=ecdd2f7355173ca677f860630a12004b4bac6777;hb=2ca964a890a31ab4f14f17cfbd359271cd8c4ea3;hpb=d23a32272b954579fe15c78d6ea605e087d0a512

diff --git a/dgram/dgram.c b/dgram/dgram.c
index ecdd2f7..b2661a4 100644
--- a/dgram/dgram.c
+++ b/dgram/dgram.c
@@ -4,23 +4,29 @@
  * dgram-socket create <local>                        => <sockid>
  * dgram-socket close <sockid>
  * dgram-socket transmit <sockid> <data> <remote>
- * dgram-socket on-receive <sockid> <script>
+ * dgram-socket on-receive <sockid> [<script>]
  *    calls, effectively,  eval <script> [list <data> <remote-addr> <socket>]
+ *    if script not supplied, cancel
  */
 
 #include "tables.h"
 #include "hbytes.h"
 
-static int sockfail(Tcl_Interp *ip, int fd, const char *m) {
-  int e;
-  e= errno;
-  close(fd);
-  return posixerr(ip,e,m);
-}
+typedef struct DgramSocket {
+  int ix; /* first ! */
+  int fd;
+  Tcl_Interp *ip;
+  ScriptToInvoke script;
+  void *addr_buf, *msg_buf;
+  int addr_buflen, msg_buflen;
+} DgramSocket;
+
+IdDataTable dgram_socks= { "dgramsock" };
 
 int do_dgram_socket_create(ClientData cd, Tcl_Interp *ip,
-			  SockAddr_Value local, int *result) {
+			   SockAddr_Value local, void **sock_r) {
   int fd, al, r;
+  DgramSocket *sock;
   const struct sockaddr *sa;
 
   sa= sockaddr_addr(&local);
@@ -28,64 +34,133 @@ int do_dgram_socket_create(ClientData cd, Tcl_Interp *ip,
 
   fd= socket(sa->sa_family, SOCK_DGRAM, 0);
   if (fd<0) return posixerr(ip,errno,"socket");
-  r= bind(fd, sa, al);  if (r) return sockfail(ip,fd,"bind");
-  r= setnonblock(fd, 1);  if (r) return sockfail(ip,fd,"setnonblock");
-  *result= fd;
+  r= bind(fd, sa, al);  if (r) return newfdposixerr(ip,fd,"bind");
+  r= setnonblock(fd, 1);  if (r) return newfdposixerr(ip,fd,"setnonblock");
+
+  sock= TALLOC(sizeof(DgramSocket));
+  sock->ix= -1;
+  sock->fd= fd;
+  sock->addr_buflen= al+1;
+  sock->addr_buf= TALLOC(sock->addr_buflen);
+  sock->msg_buflen= 0;
+  sock->msg_buf= 0;
+  scriptinv_init(&sock->script);
+
+  *sock_r= sock;
   return TCL_OK;
 }
 
-/* Arg parsing */
-
-int pat_sockid(Tcl_Interp *ip, Tcl_Obj *o, int *val) {
-  int rc;
-  
-  rc= Tcl_ConvertToType(ip,o,&sockid_type);
-  if (rc) return rc;
-
-  *val= o->internalRep.longValue;
+int do_dgram_socket_transmit(ClientData cd, Tcl_Interp *ip,
+			     void *sock_v, HBytes_Value data,
+			     SockAddr_Value remote) {
+  DgramSocket *sock= sock_v;
+  int l, r;
+
+  r= sendto(sock->fd,
+	    hbytes_data(&data), l=hbytes_len(&data),
+	    0,
+	    sockaddr_addr(&remote), sockaddr_len(&remote));
+  if (r==-1) return posixerr(ip,errno,"sendto");
+  else if (r!=l) return staticerr(ip,"sendto gave wrong answer",0);
   return TCL_OK;
 }
 
-Tcl_Obj *ret_sockid(Tcl_Interp *ip, int val) {
-  Tcl_Obj *o;
-
-  o= Tcl_NewObj();
-  Tcl_InvalidateStringRep(o);
-  o->internalRep.longValue= val;
-  o->typePtr= &sockid_type;
-  return o;
-}
-
-static void sockid_t_free(Tcl_Obj *o) { }
-
-static void sockid_t_dup(Tcl_Obj *src, Tcl_Obj *dup) {
-  dup->internalRep= src->internalRep;
+static void cancel(DgramSocket *sock) {
+  if (sock->script.obj) {
+    scriptinv_cancel(&sock->script);
+    Tcl_DeleteFileHandler(sock->fd);
+  }
 }
 
-static void sockid_t_ustr(Tcl_Obj *o) {
-  char buf[100];
-  int l;
-
-  snprintf(buf,sizeof(buf),"dgramsock%d", (int)o->internalRep.longValue);
-  l= o->length= strlen(buf);
-  o->bytes= TALLOC(l+1);
-  strcpy(o->bytes, buf);
+static void recv_call(ClientData sock_cd, int mask) {
+  DgramSocket *sock= (void*)sock_cd;
+  Tcl_Interp *ip= sock->script.ip;
+  int sz, rc, peek;
+  HBytes_Value message_val;
+  SockAddr_Value peer_val;
+  Tcl_Obj *args[3];
+  struct msghdr mh;
+  struct iovec iov;
+
+  hbytes_empty(&message_val);
+  sockaddr_clear(&peer_val);
+
+  mh.msg_iov= &iov;
+  mh.msg_iovlen= 1;
+  mh.msg_control= 0;
+  mh.msg_controllen= 0;
+  mh.msg_flags= 0;
+
+  peek= MSG_PEEK;
+  
+  for (;;) {
+    mh.msg_name= sock->addr_buf;
+    mh.msg_namelen= sock->addr_buflen;
+
+    iov.iov_base= sock->msg_buf;
+    iov.iov_len= sock->msg_buflen;
+
+    sz= recvmsg(sock->fd, &mh, peek);
+    if (sz==-1) {
+      if (errno == EAGAIN || errno == EWOULDBLOCK) rc=0;
+      else rc= posixerr(ip,errno,"recvmsg");
+      goto x_rc;
+    }
+
+    assert(mh.msg_namelen < sock->addr_buflen);
+
+    if (!(mh.msg_flags & MSG_TRUNC)) {
+      if (!peek) break;
+      peek= 0;
+      continue;
+    }
+
+    TFREE(sock->msg_buf);
+    sock->msg_buflen *= 2;
+    sock->msg_buflen += 100;
+    sock->msg_buf= TALLOC(sock->msg_buflen);
+  }
+
+  hbytes_array(&message_val, iov.iov_base, sz);
+  sockaddr_create(&peer_val, mh.msg_name, mh.msg_namelen);
+
+  args[0]= ret_hb(ip, message_val);  hbytes_empty(&message_val);
+  args[1]= ret_sockaddr(ip, peer_val);  sockaddr_clear(&peer_val);
+  args[2]= ret_iddata(ip, sock, &dgram_socks);
+  scriptinv_invoke(&sock->script,3,args);
+
+  rc= 0;
+  
+x_rc:
+  if (rc)
+    Tcl_BackgroundError(ip);
 }
 
-static int sockid_t_sfa(Tcl_Interp *ip, Tcl_Obj *o) {
-  unsigned long ul;
-  char *ep, *str;
+int do_dgram_socket_on_receive(ClientData cd, Tcl_Interp *ip,
+			       void *sock_v, Tcl_Obj *newscript) {
+  DgramSocket *sock= sock_v;
+  int rc;
   
-  str= Tcl_GetStringFromObj(o,0);
-  if (memcmp(str,"dgramsock",9)) return staticerr(ip,"bad dgram socket id");
-  errno=0; ul=strtoul(str+9,&ep,10);
-  if (errno || *ep) return staticerr(ip,"bad dgram socket id number");
-  if (ul > INT_MAX) return staticerr(ip,"out of range dgram socket id");
-  o->internalRep.longValue= ul;
+  cancel(sock);
+  
+  if (newscript) {
+    rc= scriptinv_set(&sock->script, ip, newscript);
+    if (rc) return rc;
+  }
+  
+  Tcl_CreateFileHandler(sock->fd, TCL_READABLE, recv_call, sock);
   return TCL_OK;
 }
 
-Tcl_ObjType sockid_type = {
-  "sockid-nearly",
-  sockid_t_free, sockid_t_dup, sockid_t_ustr, sockid_t_sfa
-};
+int do_dgram_socket_close(ClientData cd, Tcl_Interp *ip, void *sock_v) {
+  DgramSocket *sock= sock_v;
+  int sockix;
+  cancel(sock);
+  close(sock->fd); /* nothing useful to be done with errors */
+  sockix= sock->ix;
+  TFREE(sock->addr_buf);
+  TFREE(sock->msg_buf);
+  TFREE(sock);
+  dgram_socks.a[sockix]= 0;
+  return TCL_OK;
+}