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 {
16 int ix, fd, script_llength;
19 void *addr_buf, *msg_buf;
20 int addr_buflen, msg_buflen;
24 static DgramSocket **socks;
26 static int sockfail(Tcl_Interp *ip, int fd, const char *m) {
30 return posixerr(ip,e,m);
33 int do_dgram_socket_create(ClientData cd, Tcl_Interp *ip,
34 SockAddr_Value local, DgramSockID *sock_r) {
35 int fd, al, r, sockix;
37 const struct sockaddr *sa;
39 for (sockix=0; sockix<n_socks && socks[sockix]; sockix++);
40 if (sockix>=n_socks) {
43 socks= (void*)Tcl_Realloc((void*)socks, n_socks*sizeof(*socks));
44 while (sockix<n_socks) socks[sockix++]=0;
48 sa= sockaddr_addr(&local);
49 al= sockaddr_len(&local);
51 fd= socket(sa->sa_family, SOCK_DGRAM, 0);
52 if (fd<0) return posixerr(ip,errno,"socket");
53 r= bind(fd, sa, al); if (r) return sockfail(ip,fd,"bind");
54 r= setnonblock(fd, 1); if (r) return sockfail(ip,fd,"setnonblock");
56 socks[sockix]= sock= TALLOC(sizeof(DgramSocket));
60 sock->addr_buflen= al+1;
61 sock->addr_buf= TALLOC(sock->addr_buflen);
69 int do_dgram_socket_transmit(ClientData cd, Tcl_Interp *ip,
70 DgramSocket *sock, HBytes_Value data,
71 SockAddr_Value remote) {
75 hbytes_data(&data), l=hbytes_len(&data),
77 sockaddr_addr(&remote), sockaddr_len(&remote));
78 if (r==-1) return posixerr(ip,errno,"sendto");
79 else if (r!=l) return staticerr(ip,"sendto gave wrong answer");
83 static void cancel(DgramSocket *sock) {
85 Tcl_DeleteFileHandler(sock->fd);
86 Tcl_DecrRefCount(sock->script);
91 static void recv_call(ClientData sock_cd, int mask) {
92 DgramSocket *sock= (void*)sock_cd;
93 Tcl_Interp *ip= sock->ip;
95 HBytes_Value message_val;
96 SockAddr_Value peer_val;
97 Tcl_Obj *args[3], *invoke;
101 hbytes_empty(&message_val);
102 sockaddr_clear(&peer_val);
103 invoke=0; for (i=0; i<3; i++) args[i]=0;
108 mh.msg_controllen= 0;
114 mh.msg_name= sock->addr_buf;
115 mh.msg_namelen= sock->addr_buflen;
117 iov.iov_base= sock->msg_buf;
118 iov.iov_len= sock->msg_buflen;
120 sz= recvmsg(sock->fd, &mh, peek);
121 if (sz==-1) { rc=0; goto x_rc; }
123 assert(mh.msg_namelen < sock->addr_buflen);
125 if (!(mh.msg_flags & MSG_TRUNC)) {
131 TFREE(sock->msg_buf);
132 sock->msg_buflen *= 2;
133 sock->msg_buflen += 100;
134 sock->msg_buf= TALLOC(sock->msg_buflen);
137 hbytes_array(&message_val, iov.iov_base, sz);
138 sockaddr_create(&peer_val, mh.msg_name, mh.msg_namelen);
140 args[0]= ret_hb(ip, message_val); hbytes_empty(&message_val);
141 args[1]= ret_sockaddr(ip, peer_val); sockaddr_clear(&peer_val);
142 args[2]= ret_sockid(ip, sock);
143 for (i=0; i<3; i++) Tcl_IncrRefCount(args[i]);
145 invoke= Tcl_DuplicateObj(sock->script);
146 Tcl_IncrRefCount(invoke);
148 rc= Tcl_ListObjReplace(ip,invoke,sock->script_llength,0,3,args);
151 rc= Tcl_EvalObjEx(ip,invoke,TCL_EVAL_GLOBAL|TCL_EVAL_DIRECT);
153 for (i=0; i<3; i++) Tcl_DecrRefCount(args[i]);
154 Tcl_DecrRefCount(invoke);
156 hbytes_free(&message_val);
157 sockaddr_free(&peer_val);
161 Tcl_BackgroundError(ip);
164 int do_dgram_socket_on_receive(ClientData cd, Tcl_Interp *ip,
165 DgramSocket *sock, Tcl_Obj *script) {
169 rc= Tcl_ListObjLength(ip, script, &sock->script_llength);
175 Tcl_IncrRefCount(script);
176 sock->script= script;
179 Tcl_CreateFileHandler(sock->fd, TCL_READABLE, recv_call, sock);
183 int do_dgram_socket_close(ClientData cd, Tcl_Interp *ip, DgramSocket *sock) {
186 close(sock->fd); /* nothing useful to be done with errors */
188 TFREE(sock->addr_buf);
189 TFREE(sock->msg_buf);
197 int pat_sockid(Tcl_Interp *ip, Tcl_Obj *o, DgramSocket **val) {
201 rc= Tcl_ConvertToType(ip,o,&dgramsockid_type);
204 sockix= o->internalRep.longValue;
205 if (sockix >= n_socks || !(sock= socks[sockix]))
206 return staticerr(ip,"dgram socket not open");
208 assert(socks[sockix]->ix == sockix);
214 Tcl_Obj *ret_sockid(Tcl_Interp *ip, DgramSocket *val) {
218 Tcl_InvalidateStringRep(o);
219 o->internalRep.longValue= val->ix;
220 o->typePtr= &dgramsockid_type;
224 static void sockid_t_free(Tcl_Obj *o) { }
226 static void sockid_t_dup(Tcl_Obj *src, Tcl_Obj *dup) {
227 dup->internalRep= src->internalRep;
230 static void sockid_t_ustr(Tcl_Obj *o) {
233 snprintf(buf,sizeof(buf), "%d", (int)o->internalRep.longValue);
234 obj_updatestr_vstringls(o,
240 static int sockid_t_sfa(Tcl_Interp *ip, Tcl_Obj *o) {
244 str= Tcl_GetStringFromObj(o,0);
245 if (memcmp(str,"dgramsock",9)) return staticerr(ip,"bad dgram socket id");
246 errno=0; ul=strtoul(str+9,&ep,10);
247 if (errno || *ep) return staticerr(ip,"bad dgram socket id number");
248 if (ul > INT_MAX) return staticerr(ip,"out of range dgram socket id");
251 o->internalRep.longValue= ul;
252 o->typePtr= &dgramsockid_type;
256 Tcl_ObjType dgramsockid_type = {
258 sockid_t_free, sockid_t_dup, sockid_t_ustr, sockid_t_sfa