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
15 typedef struct DgramSocket {
17 int fd, script_llength;
20 void *addr_buf, *msg_buf;
21 int addr_buflen, msg_buflen;
24 IdDataTable dgram_socks= { "dgramsock" };
26 int do_dgram_socket_create(ClientData cd, Tcl_Interp *ip,
27 SockAddr_Value local, void **sock_r) {
30 const struct sockaddr *sa;
32 sa= sockaddr_addr(&local);
33 al= sockaddr_len(&local);
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");
40 sock= TALLOC(sizeof(DgramSocket));
44 sock->addr_buflen= al+1;
45 sock->addr_buf= TALLOC(sock->addr_buflen);
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;
60 hbytes_data(&data), l=hbytes_len(&data),
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);
68 static void cancel(DgramSocket *sock) {
70 Tcl_DeleteFileHandler(sock->fd);
71 Tcl_DecrRefCount(sock->script);
76 static void recv_call(ClientData sock_cd, int mask) {
77 DgramSocket *sock= (void*)sock_cd;
78 Tcl_Interp *ip= sock->ip;
80 HBytes_Value message_val;
81 SockAddr_Value peer_val;
82 Tcl_Obj *args[3], *invoke;
86 hbytes_empty(&message_val);
87 sockaddr_clear(&peer_val);
88 invoke=0; for (i=0; i<3; i++) args[i]=0;
99 mh.msg_name= sock->addr_buf;
100 mh.msg_namelen= sock->addr_buflen;
102 iov.iov_base= sock->msg_buf;
103 iov.iov_len= sock->msg_buflen;
105 sz= recvmsg(sock->fd, &mh, peek);
106 if (sz==-1) { rc=0; goto x_rc; }
108 assert(mh.msg_namelen < sock->addr_buflen);
110 if (!(mh.msg_flags & MSG_TRUNC)) {
116 TFREE(sock->msg_buf);
117 sock->msg_buflen *= 2;
118 sock->msg_buflen += 100;
119 sock->msg_buf= TALLOC(sock->msg_buflen);
122 hbytes_array(&message_val, iov.iov_base, sz);
123 sockaddr_create(&peer_val, mh.msg_name, mh.msg_namelen);
125 args[0]= ret_hb(ip, message_val); hbytes_empty(&message_val);
126 args[1]= ret_sockaddr(ip, peer_val); sockaddr_clear(&peer_val);
127 args[2]= ret_iddata(ip, sock, &dgram_socks);
128 for (i=0; i<3; i++) Tcl_IncrRefCount(args[i]);
130 invoke= Tcl_DuplicateObj(sock->script);
131 Tcl_IncrRefCount(invoke);
133 rc= Tcl_ListObjReplace(ip,invoke,sock->script_llength,0,3,args);
134 for (i=0; i<3; i++) { Tcl_DecrRefCount(args[i]); args[i]= 0; }
137 rc= Tcl_EvalObjEx(ip,invoke,TCL_EVAL_GLOBAL|TCL_EVAL_DIRECT);
140 if (invoke) Tcl_DecrRefCount(invoke);
143 Tcl_BackgroundError(ip);
146 int do_dgram_socket_on_receive(ClientData cd, Tcl_Interp *ip,
147 void *sock_v, Tcl_Obj *script) {
148 DgramSocket *sock= sock_v;
152 rc= Tcl_ListObjLength(ip, script, &sock->script_llength);
158 Tcl_IncrRefCount(script);
159 sock->script= script;
162 Tcl_CreateFileHandler(sock->fd, TCL_READABLE, recv_call, sock);
166 int do_dgram_socket_close(ClientData cd, Tcl_Interp *ip, void *sock_v) {
167 DgramSocket *sock= sock_v;
170 close(sock->fd); /* nothing useful to be done with errors */
172 TFREE(sock->addr_buf);
173 TFREE(sock->msg_buf);
175 dgram_socks.a[sockix]= 0;