chiark / gitweb /
24c5446c0b1625528f98fbb5ce4dceb664845f28
[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 "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 };