* if script not supplied, cancel
*/
-#include "tables.h"
-#include "hbytes.h"
+#include "dgram.h"
typedef struct DgramSocket {
- int ix, fd, script_llength;
+ int ix; /* first ! */
+ int fd;
Tcl_Interp *ip;
- Tcl_Obj *script;
+ ScriptToInvoke script;
void *addr_buf, *msg_buf;
int addr_buflen, msg_buflen;
} DgramSocket;
-static int n_socks;
-static DgramSocket **socks;
-
-static int sockfail(Tcl_Interp *ip, int fd, const char *m) {
- int e;
- e= errno;
- close(fd);
- return posixerr(ip,e,m);
-}
-
-int do_dgram_socket_create(ClientData cd, Tcl_Interp *ip,
- SockAddr_Value local, DgramSockID *sock_r) {
- int fd, al, r, sockix;
+int cht_do_dgramsocket_create(ClientData cd, Tcl_Interp *ip,
+ SockAddr_Value local, void **sock_r) {
+ int fd, al, r;
DgramSocket *sock;
const struct sockaddr *sa;
- for (sockix=0; sockix<n_socks && socks[sockix]; sockix++);
- if (sockix>=n_socks) {
- n_socks += 2;
- n_socks *= 2;
- socks= (void*)Tcl_Realloc((void*)socks, n_socks*sizeof(*socks));
- while (sockix<n_socks) socks[sockix++]=0;
- sockix--;
- }
-
- sa= sockaddr_addr(&local);
- al= sockaddr_len(&local);
+ sa= cht_sockaddr_addr(&local);
+ al= cht_sockaddr_len(&local);
fd= socket(sa->sa_family, SOCK_DGRAM, 0);
- if (fd<0) return posixerr(ip,errno,"socket");
- r= bind(fd, sa, al); if (r) return sockfail(ip,fd,"bind");
- r= setnonblock(fd, 1); if (r) return sockfail(ip,fd,"setnonblock");
+ if (fd<0) return cht_posixerr(ip,errno,"socket");
+ r= bind(fd, sa, al); if (r) return cht_newfdposixerr(ip,fd,"bind");
+ r= cht_setnonblock(fd, 1); if (r) return cht_newfdposixerr(ip,fd,"setnonblock");
- socks[sockix]= sock= TALLOC(sizeof(DgramSocket));
+ sock= TALLOC(sizeof(DgramSocket));
+ sock->ix= -1;
sock->fd= fd;
- sock->ix= sockix;
- sock->script= 0;
sock->addr_buflen= al+1;
sock->addr_buf= TALLOC(sock->addr_buflen);
sock->msg_buflen= 0;
sock->msg_buf= 0;
+ cht_scriptinv_init(&sock->script);
*sock_r= sock;
return TCL_OK;
}
-int do_dgram_socket_transmit(ClientData cd, Tcl_Interp *ip,
- DgramSocket *sock, HBytes_Value data,
+int cht_do_dgramsocket_transmit(ClientData cd, Tcl_Interp *ip,
+ void *sock_v, HBytes_Value data,
SockAddr_Value remote) {
+ DgramSocket *sock= sock_v;
int l, r;
r= sendto(sock->fd,
- hbytes_data(&data), l=hbytes_len(&data),
+ cht_hb_data(&data), l=cht_hb_len(&data),
0,
- sockaddr_addr(&remote), sockaddr_len(&remote));
- if (r==-1) return posixerr(ip,errno,"sendto");
- else if (r!=l) return staticerr(ip,"sendto gave wrong answer",0);
+ cht_sockaddr_addr(&remote), cht_sockaddr_len(&remote));
+ if (r==-1) return cht_posixerr(ip,errno,"sendto");
+ else if (r!=l) return cht_staticerr(ip,"sendto gave wrong answer",0);
return TCL_OK;
}
static void cancel(DgramSocket *sock) {
- if (sock->script) {
+ if (sock->script.script) {
+ cht_scriptinv_cancel(&sock->script);
Tcl_DeleteFileHandler(sock->fd);
- Tcl_DecrRefCount(sock->script);
- sock->script= 0;
}
}
static void recv_call(ClientData sock_cd, int mask) {
DgramSocket *sock= (void*)sock_cd;
- Tcl_Interp *ip= sock->ip;
- int sz, rc, i, peek;
+ Tcl_Interp *ip= sock->script.ipq;
+ int sz, rc, peek;
HBytes_Value message_val;
SockAddr_Value peer_val;
- Tcl_Obj *args[3], *invoke;
+ Tcl_Obj *args[3];
struct msghdr mh;
struct iovec iov;
- hbytes_empty(&message_val);
- sockaddr_clear(&peer_val);
- invoke=0; for (i=0; i<3; i++) args[i]=0;
+ cht_hb_empty(&message_val);
+ cht_sockaddr_clear(&peer_val);
mh.msg_iov= &iov;
mh.msg_iovlen= 1;
iov.iov_len= sock->msg_buflen;
sz= recvmsg(sock->fd, &mh, peek);
- if (sz==-1) { rc=0; goto x_rc; }
+ if (sz==-1) {
+ if (errno == EAGAIN || errno == EWOULDBLOCK) rc=0;
+ else rc= cht_posixerr(ip,errno,"recvmsg");
+ goto x_rc;
+ }
assert(mh.msg_namelen < sock->addr_buflen);
}
TFREE(sock->msg_buf);
+ assert(sock->msg_buflen < INT_MAX/4);
sock->msg_buflen *= 2;
sock->msg_buflen += 100;
sock->msg_buf= TALLOC(sock->msg_buflen);
}
- hbytes_array(&message_val, iov.iov_base, sz);
- sockaddr_create(&peer_val, mh.msg_name, mh.msg_namelen);
-
- args[0]= ret_hb(ip, message_val); hbytes_empty(&message_val);
- args[1]= ret_sockaddr(ip, peer_val); sockaddr_clear(&peer_val);
- args[2]= ret_sockid(ip, sock);
- for (i=0; i<3; i++) Tcl_IncrRefCount(args[i]);
-
- invoke= Tcl_DuplicateObj(sock->script);
- Tcl_IncrRefCount(invoke);
+ cht_hb_array(&message_val, iov.iov_base, sz);
+ cht_sockaddr_create(&peer_val, mh.msg_name, mh.msg_namelen);
- rc= Tcl_ListObjReplace(ip,invoke,sock->script_llength,0,3,args);
- for (i=0; i<3; i++) { Tcl_DecrRefCount(args[i]); args[i]= 0; }
- if (rc) goto x_rc;
-
- rc= Tcl_EvalObjEx(ip,invoke,TCL_EVAL_GLOBAL|TCL_EVAL_DIRECT);
+ args[0]= cht_ret_hb(ip, message_val); cht_hb_empty(&message_val);
+ args[1]= cht_ret_sockaddr(ip, peer_val); cht_sockaddr_clear(&peer_val);
+ args[2]= cht_ret_iddata(ip, sock, &cht_dgram_socks);
+ cht_scriptinv_invoke(&sock->script,3,args);
+ rc= 0;
+
x_rc:
- if (invoke) Tcl_DecrRefCount(invoke);
-
if (rc)
Tcl_BackgroundError(ip);
}
-int do_dgram_socket_on_receive(ClientData cd, Tcl_Interp *ip,
- DgramSocket *sock, Tcl_Obj *script) {
+int cht_do_dgramsocket_sockname(ClientData cd, Tcl_Interp *ip,
+ void *sock_v, SockAddr_Value *result) {
+ DgramSocket *sock= sock_v;
+ int r;
+
+ socklen_t salen = sock->addr_buflen;
+ r= getsockname(sock->fd, sock->addr_buf, &salen);
+ if (r) return cht_posixerr(ip,errno,"getsockname");
+ cht_sockaddr_create(result, sock->addr_buf, salen);
+ return TCL_OK;
+}
+
+int cht_do_dgramsocket_on_receive(ClientData cd, Tcl_Interp *ip,
+ void *sock_v, Tcl_Obj *newscript) {
+ DgramSocket *sock= sock_v;
int rc;
- if (script) {
- rc= Tcl_ListObjLength(ip, script, &sock->script_llength);
+ cancel(sock);
+
+ if (newscript) {
+ rc= cht_scriptinv_set(&sock->script, ip, newscript, 0);
if (rc) return rc;
}
- cancel(sock);
- if (script) {
- Tcl_IncrRefCount(script);
- sock->script= script;
- sock->ip= ip;
- }
Tcl_CreateFileHandler(sock->fd, TCL_READABLE, recv_call, sock);
return TCL_OK;
}
-int do_dgram_socket_close(ClientData cd, Tcl_Interp *ip, DgramSocket *sock) {
- int sockix;
+static void destroy(DgramSocket *sock) {
cancel(sock);
close(sock->fd); /* nothing useful to be done with errors */
- sockix= sock->ix;
TFREE(sock->addr_buf);
TFREE(sock->msg_buf);
TFREE(sock);
- socks[sockix]= 0;
- return TCL_OK;
}
-/* Arg parsing */
-
-int pat_sockid(Tcl_Interp *ip, Tcl_Obj *o, DgramSocket **val) {
- int rc, sockix;
- DgramSocket *sock;
-
- rc= Tcl_ConvertToType(ip,o,&dgramsockid_type);
- if (rc) return rc;
-
- sockix= o->internalRep.longValue;
- if (sockix >= n_socks || !(sock= socks[sockix]))
- return staticerr(ip,"dgram socket not open",0);
-
- assert(socks[sockix]->ix == sockix);
-
- *val= sock;
- return TCL_OK;
+static void destroy_idtabcb(Tcl_Interp *ip, void *sock_v) {
+ destroy(sock_v);
}
-Tcl_Obj *ret_sockid(Tcl_Interp *ip, DgramSocket *val) {
- Tcl_Obj *o;
-
- o= Tcl_NewObj();
- Tcl_InvalidateStringRep(o);
- o->internalRep.longValue= val->ix;
- o->typePtr= &dgramsockid_type;
- return o;
-}
-
-static void sockid_t_free(Tcl_Obj *o) { }
-
-static void sockid_t_dup(Tcl_Obj *src, Tcl_Obj *dup) {
- dup->internalRep= src->internalRep;
- dup->typePtr= &dgramsockid_type;
-}
-
-static void sockid_t_ustr(Tcl_Obj *o) {
- char buf[75];
-
- snprintf(buf,sizeof(buf), "%d", (int)o->internalRep.longValue);
- obj_updatestr_vstringls(o,
- "dgramsock",9,
- buf, strlen(buf),
- (char*)0);
-}
-
-static int sockid_t_sfa(Tcl_Interp *ip, Tcl_Obj *o) {
- unsigned long ul;
- char *ep, *str;
-
- str= Tcl_GetStringFromObj(o,0);
- if (memcmp(str,"dgramsock",9)) return staticerr(ip,"bad dgram socket id",0);
- errno=0; ul=strtoul(str+9,&ep,10);
- if (errno || *ep) return staticerr(ip,"bad dgram socket id number",0);
- if (ul > INT_MAX) return staticerr(ip,"out of range dgram socket id",0);
-
- objfreeir(o);
- o->internalRep.longValue= ul;
- o->typePtr= &dgramsockid_type;
+int cht_do_dgramsocket_close(ClientData cd, Tcl_Interp *ip, void *sock_v) {
+ cht_tabledataid_disposing(ip,sock_v,&cht_dgram_socks);
+ destroy(sock_v);
return TCL_OK;
}
-Tcl_ObjType dgramsockid_type = {
- "dgramsockid",
- sockid_t_free, sockid_t_dup, sockid_t_ustr, sockid_t_sfa
+const IdDataSpec cht_dgram_socks= {
+ "dgramsock", "dgramsock-table", destroy_idtabcb
};
~ian / chiark-tcl.git / blobdiff