chiark / gitweb /
adns compiles ish, working on transferring the rest
[chiark-tcl.git] / dgram / sockaddr.c
1 /*
2  * struct sockaddr
3  *
4  * syntaxes:
5  *    ddd.ddd.ddd.ddd,nnnn       IPv4 address and port
6  *                                 ... host or port may be `*'
7  *    /abs/path/to/socket        AF_UNIX
8  *    ./rel/path/to/socket       AF_UNIX
9  */
10
11 #include "hbytes.h"
12 #include "tables.h"
13
14 #define SOCKADDR_LEN(sa) ((sa)->end - (sa)->begin)
15
16 /* parsing */
17
18 int pat_sockaddr(Tcl_Interp *ip, Tcl_Obj *o, SockAddr_Value *val) {
19   int rc;
20
21   rc= Tcl_ConvertToType(ip,o,&sockaddr_type);
22   if (rc) return rc;
23
24   *val= *OBJ_SOCKADDR(o);
25   return TCL_OK;
26 }
27   
28 Tcl_Obj *ret_sockaddr(Tcl_Interp *ip, SockAddr_Value val) {
29   Tcl_Obj *o;
30
31   o= Tcl_NewObj();
32   Tcl_InvalidateStringRep(o);
33   *OBJ_SOCKADDR(o)= val;
34   o->typePtr= &sockaddr_type;
35   return o;
36 }
37
38 /* native type methods */
39
40 void sockaddr_clear(SockAddr_Value *v) { v->begin= v->end= 0; }
41
42 void sockaddr_create(SockAddr_Value *v, const struct sockaddr *addr, int len) {
43   Byte *begin;
44   
45   v->begin= begin= TALLOC(len);
46   memcpy(begin, addr, len);
47   v->end= begin + len;
48 }  
49
50 int sockaddr_len(const SockAddr_Value *v) {
51   return SOCKADDR_LEN(v);
52 }
53
54 const struct sockaddr *sockaddr_addr(const SockAddr_Value *v) {
55   return (const void*)v->begin;
56 }
57   
58 void sockaddr_free(const SockAddr_Value *v) {
59   TFREE(v->begin);
60 }
61
62 /* Sockaddr Tcl type */
63
64 static void sockaddr_t_dup(Tcl_Obj *src, Tcl_Obj *dup) {
65   sockaddr_create(OBJ_SOCKADDR(dup),
66                   sockaddr_addr(OBJ_SOCKADDR(src)),
67                   sockaddr_len(OBJ_SOCKADDR(src)));
68   dup->typePtr= &hbytes_type;
69 }
70
71 static void sockaddr_t_free(Tcl_Obj *o) {
72   sockaddr_free(OBJ_SOCKADDR(o));
73 }
74
75 static void sockaddr_t_ustr(Tcl_Obj *o) {
76   const struct sockaddr *sa;
77   char i46buf[INET6_ADDRSTRLEN], portbuf[50];
78   const struct sockaddr_in *sin;
79   int al;
80   const char *string, *prepend;
81   
82   sa= sockaddr_addr(OBJ_SOCKADDR(o));
83   al= sockaddr_len(OBJ_SOCKADDR(o));
84     
85   switch (sa->sa_family) {
86   case AF_INET:
87   case AF_INET6:
88     assert(sizeof(i46buf) >= INET_ADDRSTRLEN);
89     assert(al >= sizeof(struct sockaddr_in));
90     sin= (const void*)sa;
91     inet_ntop(sa->sa_family, &sin->sin_addr, i46buf, al);
92     snprintf(portbuf,sizeof(portbuf),",%d",(int)ntohs(sin->sin_port));
93     prepend= i46buf;
94     string= portbuf;
95     break;
96
97   case AF_UNIX:
98     string= ((const struct sockaddr_un*)sa)->sun_path;
99     prepend= "";
100     if (!string[0]) string="//"; 
101     else if (string[0] != '/' || string[1] == '/') prepend= "./";
102     break;
103
104   default: /* ouch ! */
105     obj_updatestr_array_prefix(o,(const void*)sa,al,"?");
106     return;
107   }
108
109   obj_updatestr_vstringls(o,
110                           prepend, strlen(prepend),
111                           string, strlen(string),
112                           (char*)0);
113 }
114
115 static int sockaddr_t_sfa(Tcl_Interp *ip, Tcl_Obj *o) {
116   union {
117     struct sockaddr_un sun;
118     struct sockaddr_in sin;
119   } s;
120   unsigned long port_l;
121   
122   char *str, *ep, *copy;
123   int sl, pl, iprv;
124   const char *comma, *path;
125
126   str= Tcl_GetStringFromObj(o,0);  assert(str);
127   objfreeir(o);
128   memset(&s,0,sizeof(s));
129
130   if (str[0]=='/' || (str[0]=='.' && str[1]=='/')) {
131     
132     sl= sizeof(s.sun);
133     s.sun.sun_family= AF_UNIX;
134
135     if (!strcmp(str,"//")) path= "";
136     else if (!memcmp(str,"./",2) && str[2]) path= str+2;
137     else { assert(str[0]=='/' && str[1]!='/'); path=str; }
138
139     if (strlen(str) >= sizeof(s.sun.sun_path))
140       return staticerr(ip, "AF_UNIX path too long", "SOCKADDR AFUNIX LENGTH");
141
142     strcpy(s.sun.sun_path, path);
143
144   } else if ((comma= strchr(str, ','))) {
145
146     sl= sizeof(s.sin);
147     s.sin.sin_family= AF_INET;
148
149     pl= comma - str;
150     copy= TALLOC(pl+1);
151     memcpy(copy, str, pl);
152     copy[pl]= 0;
153
154     if (!strcmp(copy,"*")) {
155       s.sin.sin_addr.s_addr= INADDR_ANY;
156       iprv= 1;
157     } else {
158       iprv= inet_pton(AF_INET, copy, &s.sin.sin_addr);
159     }
160     TFREE(copy);
161
162     if (!iprv)
163       return staticerr(ip, "bad IPv4 address syntax", "SOCKADDR SYNTAX IPV4");
164
165     comma++;
166     if (!strcmp(comma,"*")) {
167       s.sin.sin_port= 0;
168     } else {
169       errno=0; port_l=strtoul(comma,&ep,10);
170       if (errno || *ep)
171         return staticerr(ip, "bad IPv4 port", "SOCKADDR SYNTAX IPV4");
172       if (port_l > 65535)
173         return staticerr(ip, "IPv4 port out of range", "SOCKADDR SYNTAX IPV4");
174       s.sin.sin_port= htons(port_l);
175     }
176
177   } else {
178
179     return staticerr(ip, "bad socket address syntax", "SOCKADDR SYNTAX OTHER");
180
181   }
182
183   sockaddr_create(OBJ_SOCKADDR(o), (void*)&s, sl);
184   
185   o->typePtr = &sockaddr_type;
186   return TCL_OK;
187 }
188
189 Tcl_ObjType sockaddr_type = {
190   "sockaddr",
191   sockaddr_t_free, sockaddr_t_dup, sockaddr_t_ustr, sockaddr_t_sfa
192 };