chiark / gitweb /
08f5f45ace92aea45c11cca8ec7c9204c0a6cf7e
[chiark-tcl.git] / 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;
18   Tcl_Interp *ip;
19   ScriptToInvoke 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->addr_buflen= al+1;
44   sock->addr_buf= TALLOC(sock->addr_buflen);
45   sock->msg_buflen= 0;
46   sock->msg_buf= 0;
47   scriptinv_init(&sock->script);
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.obj) {
70     scriptinv_cancel(&sock->script);
71     Tcl_DeleteFileHandler(sock->fd);
72   }
73 }
74
75 static void recv_call(ClientData sock_cd, int mask) {
76   DgramSocket *sock= (void*)sock_cd;
77   Tcl_Interp *ip= sock->script.ip;
78   int sz, rc, peek;
79   HBytes_Value message_val;
80   SockAddr_Value peer_val;
81   Tcl_Obj *args[3];
82   struct msghdr mh;
83   struct iovec iov;
84
85   hbytes_empty(&message_val);
86   sockaddr_clear(&peer_val);
87
88   mh.msg_iov= &iov;
89   mh.msg_iovlen= 1;
90   mh.msg_control= 0;
91   mh.msg_controllen= 0;
92   mh.msg_flags= 0;
93
94   peek= MSG_PEEK;
95   
96   for (;;) {
97     mh.msg_name= sock->addr_buf;
98     mh.msg_namelen= sock->addr_buflen;
99
100     iov.iov_base= sock->msg_buf;
101     iov.iov_len= sock->msg_buflen;
102
103     sz= recvmsg(sock->fd, &mh, peek);
104     if (sz==-1) {
105       if (errno == EAGAIN || errno == EWOULDBLOCK) rc=0;
106       else rc= posixerr(ip,errno,"recvmsg");
107       goto x_rc;
108     }
109
110     assert(mh.msg_namelen < sock->addr_buflen);
111
112     if (!(mh.msg_flags & MSG_TRUNC)) {
113       if (!peek) break;
114       peek= 0;
115       continue;
116     }
117
118     TFREE(sock->msg_buf);
119     sock->msg_buflen *= 2;
120     sock->msg_buflen += 100;
121     sock->msg_buf= TALLOC(sock->msg_buflen);
122   }
123
124   hbytes_array(&message_val, iov.iov_base, sz);
125   sockaddr_create(&peer_val, mh.msg_name, mh.msg_namelen);
126
127   args[0]= ret_hb(ip, message_val);  hbytes_empty(&message_val);
128   args[1]= ret_sockaddr(ip, peer_val);  sockaddr_clear(&peer_val);
129   args[2]= ret_iddata(ip, sock, &dgram_socks);
130   scriptinv_invoke(&sock->script,3,args);
131
132   rc= 0;
133   
134 x_rc:
135   if (rc)
136     Tcl_BackgroundError(ip);
137 }
138
139 int do_dgram_socket_on_receive(ClientData cd, Tcl_Interp *ip,
140                                void *sock_v, Tcl_Obj *newscript) {
141   DgramSocket *sock= sock_v;
142   int rc;
143   
144   cancel(sock);
145   
146   if (newscript) {
147     rc= scriptinv_set(&sock->script, ip, newscript);
148     if (rc) return rc;
149   }
150   
151   Tcl_CreateFileHandler(sock->fd, TCL_READABLE, recv_call, sock);
152   return TCL_OK;
153 }
154
155 int do_dgram_socket_close(ClientData cd, Tcl_Interp *ip, void *sock_v) {
156   DgramSocket *sock= sock_v;
157
158   cancel(sock);
159   close(sock->fd); /* nothing useful to be done with errors */
160   tabledataid_disposing(sock,&dgram_socks);
161   TFREE(sock->addr_buf);
162   TFREE(sock->msg_buf);
163   TFREE(sock);
164   return TCL_OK;
165 }