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
14 typedef struct DgramSocket {
18 ScriptToInvoke script;
19 void *addr_buf, *msg_buf;
20 int addr_buflen, msg_buflen;
23 int cht_do_dgramsocket_create(ClientData cd, Tcl_Interp *ip,
24 SockAddr_Value local, void **sock_r) {
27 const struct sockaddr *sa;
29 sa= cht_sockaddr_addr(&local);
30 al= cht_sockaddr_len(&local);
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");
37 sock= TALLOC(sizeof(DgramSocket));
40 sock->addr_buflen= al+1;
41 sock->addr_buf= TALLOC(sock->addr_buflen);
44 cht_scriptinv_init(&sock->script);
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;
57 cht_hb_data(&data), l=cht_hb_len(&data),
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);
65 static void cancel(DgramSocket *sock) {
66 if (sock->script.script) {
67 cht_scriptinv_cancel(&sock->script);
68 Tcl_DeleteFileHandler(sock->fd);
72 static void recv_call(ClientData sock_cd, int mask) {
73 DgramSocket *sock= (void*)sock_cd;
74 Tcl_Interp *ip= sock->script.ipq;
76 HBytes_Value message_val;
77 SockAddr_Value peer_val;
82 cht_hb_empty(&message_val);
83 cht_sockaddr_clear(&peer_val);
94 mh.msg_name= sock->addr_buf;
95 mh.msg_namelen= sock->addr_buflen;
97 iov.iov_base= sock->msg_buf;
98 iov.iov_len= sock->msg_buflen;
100 sz= recvmsg(sock->fd, &mh, peek);
102 if (errno == EAGAIN || errno == EWOULDBLOCK) rc=0;
103 else rc= cht_posixerr(ip,errno,"recvmsg");
107 assert(mh.msg_namelen < sock->addr_buflen);
109 if (!(mh.msg_flags & MSG_TRUNC)) {
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);
122 cht_hb_array(&message_val, iov.iov_base, sz);
123 cht_sockaddr_create(&peer_val, mh.msg_name, mh.msg_namelen);
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);
134 Tcl_BackgroundError(ip);
137 int cht_do_dgramsocket_on_receive(ClientData cd, Tcl_Interp *ip,
138 void *sock_v, Tcl_Obj *newscript) {
139 DgramSocket *sock= sock_v;
145 rc= cht_scriptinv_set(&sock->script, ip, newscript, 0);
149 Tcl_CreateFileHandler(sock->fd, TCL_READABLE, recv_call, sock);
153 static void destroy(DgramSocket *sock) {
155 close(sock->fd); /* nothing useful to be done with errors */
156 TFREE(sock->addr_buf);
157 TFREE(sock->msg_buf);
161 static void destroy_idtabcb(Tcl_Interp *ip, void *sock_v) {
165 int cht_do_dgramsocket_close(ClientData cd, Tcl_Interp *ip, void *sock_v) {
166 cht_tabledataid_disposing(ip,sock_v,&cht_dgram_socks);
171 const IdDataSpec cht_dgram_socks= {
172 "dgramsock", "dgramsock-table", destroy_idtabcb