chiark / gitweb /
dgram sock references fixed. can close
[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  */
10
11 #include "tables.h"
12 #include "hbytes.h"
13
14 typedef struct DgramSocket {
15   int fd;
16 } DgramSocket;
17
18 static int n_socks;
19 static DgramSocket *socks;
20
21 static int sockfail(Tcl_Interp *ip, int fd, const char *m) {
22   int e;
23   e= errno;
24   close(fd);
25   return posixerr(ip,e,m);
26 }
27
28 int do_dgram_socket_create(ClientData cd, Tcl_Interp *ip,
29                           SockAddr_Value local, int *sock_r) {
30   int fd, al, r, sock;
31   const struct sockaddr *sa;
32
33   for (sock=0; sock<n_socks && socks[sock].fd>=0; sock++);
34   if (sock>=n_socks) {
35     n_socks += 2;
36     n_socks *= 2;
37     socks= (void*)Tcl_Realloc((void*)socks, n_socks*sizeof(*socks));
38     while (sock<n_socks) socks[sock++].fd=-1;
39     sock--;
40   }
41
42   sa= sockaddr_addr(&local);
43   al= sockaddr_len(&local);
44
45   fd= socket(sa->sa_family, SOCK_DGRAM, 0);
46   if (fd<0) return posixerr(ip,errno,"socket");
47   r= bind(fd, sa, al);  if (r) return sockfail(ip,fd,"bind");
48   r= setnonblock(fd, 1);  if (r) return sockfail(ip,fd,"setnonblock");
49
50   socks[sock].fd= fd;
51   *sock_r= sock;
52   return TCL_OK;
53 }
54
55 int do_dgram_socket_close(ClientData cd, Tcl_Interp *ip, int sock) {
56   close(socks[sock].fd); /* nothing useful to be done with errors */
57   socks[sock].fd= -1;
58   return TCL_OK;
59 }
60
61 /* Arg parsing */
62
63 int pat_sockid(Tcl_Interp *ip, Tcl_Obj *o, int *val) {
64   int rc, sock;
65   
66   rc= Tcl_ConvertToType(ip,o,&sockid_type);
67   if (rc) return rc;
68
69   sock= o->internalRep.longValue;
70   if (sock >= n_socks || socks[sock].fd==-1)
71     return staticerr(ip,"dgram socket not open");
72
73   *val= sock;
74   return TCL_OK;
75 }
76
77 Tcl_Obj *ret_sockid(Tcl_Interp *ip, int val) {
78   Tcl_Obj *o;
79
80   o= Tcl_NewObj();
81   Tcl_InvalidateStringRep(o);
82   o->internalRep.longValue= val;
83   o->typePtr= &sockid_type;
84   return o;
85 }
86
87 static void sockid_t_free(Tcl_Obj *o) { }
88
89 static void sockid_t_dup(Tcl_Obj *src, Tcl_Obj *dup) {
90   dup->internalRep= src->internalRep;
91 }
92
93 static void sockid_t_ustr(Tcl_Obj *o) {
94   char buf[100];
95   int l;
96
97   snprintf(buf,sizeof(buf),"dgramsock%d", (int)o->internalRep.longValue);
98   l= o->length= strlen(buf);
99   o->bytes= TALLOC(l+1);
100   strcpy(o->bytes, buf);
101 }
102
103 static int sockid_t_sfa(Tcl_Interp *ip, Tcl_Obj *o) {
104   unsigned long ul;
105   char *ep, *str;
106   
107   str= Tcl_GetStringFromObj(o,0);
108   if (memcmp(str,"dgramsock",9)) return staticerr(ip,"bad dgram socket id");
109   errno=0; ul=strtoul(str+9,&ep,10);
110   if (errno || *ep) return staticerr(ip,"bad dgram socket id number");
111   if (ul > INT_MAX) return staticerr(ip,"out of range dgram socket id");
112   o->internalRep.longValue= ul;
113   return TCL_OK;
114 }
115
116 Tcl_ObjType sockid_type = {
117   "sockid-nearly",
118   sockid_t_free, sockid_t_dup, sockid_t_ustr, sockid_t_sfa
119 };