(was Re: [PATCH @13746] h2xs)
[p5sagit/p5-mst-13.2.git] / ext / Socket / Socket.xs
CommitLineData
c5be433b 1#define PERL_NO_GET_CONTEXT
a0d0e21e 2#include "EXTERN.h"
3#include "perl.h"
4#include "XSUB.h"
5
fc87b71c 6#include <stddef.h>
7
8e07c86e 8#ifndef VMS
9# ifdef I_SYS_TYPES
10# include <sys/types.h>
11# endif
9cc6feab 12# include <sys/socket.h>
29209bc5 13# if defined(USE_SOCKS) && defined(I_SOCKS)
86959918 14# include <socks.h>
a062834f 15# endif
9cc6feab 16# ifdef MPE
17# define PF_INET AF_INET
18# define PF_UNIX AF_UNIX
19# define SOCK_RAW 3
20# endif
21# ifdef I_SYS_UN
22# include <sys/un.h>
23# endif
bf9b2f8f 24/* XXX Configure test for <netinet/in_systm.h needed XXX */
25# if defined(NeXT) || defined(__NeXT__)
26# include <netinet/in_systm.h>
27# endif
8e07c86e 28# ifdef I_NETINET_IN
29# include <netinet/in.h>
30# endif
86959918 31# ifdef I_NETDB
32# include <netdb.h>
33# endif
9cc6feab 34# ifdef I_ARPA_INET
35# include <arpa/inet.h>
36# endif
37# ifdef I_NETINET_TCP
38# include <netinet/tcp.h>
39# endif
8e07c86e 40#else
9cc6feab 41# include "sockadapt.h"
8e07c86e 42#endif
a0d0e21e 43
2986a63f 44#ifdef NETWARE
45NETDB_DEFINE_CONTEXT
46NETINET_DEFINE_CONTEXT
47#endif
48
6b1016b5 49#ifdef I_SYSUIO
50# include <sys/uio.h>
51#endif
52
a0d0e21e 53#ifndef AF_NBS
9cc6feab 54# undef PF_NBS
a0d0e21e 55#endif
56
57#ifndef AF_X25
9cc6feab 58# undef PF_X25
a0d0e21e 59#endif
60
8e07c86e 61#ifndef INADDR_NONE
9cc6feab 62# define INADDR_NONE 0xffffffff
8e07c86e 63#endif /* INADDR_NONE */
7e1af8bc 64#ifndef INADDR_BROADCAST
9cc6feab 65# define INADDR_BROADCAST 0xffffffff
7e1af8bc 66#endif /* INADDR_BROADCAST */
8e07c86e 67#ifndef INADDR_LOOPBACK
9cc6feab 68# define INADDR_LOOPBACK 0x7F000001
8e07c86e 69#endif /* INADDR_LOOPBACK */
70
7e1af8bc 71#ifndef HAS_INET_ATON
72
a062834f 73/*
7e1af8bc 74 * Check whether "cp" is a valid ascii representation
75 * of an Internet address and convert to a binary address.
76 * Returns 1 if the address is valid, 0 if not.
77 * This replaces inet_addr, the return value from which
78 * cannot distinguish between failure and a local broadcast address.
79 */
80static int
f0f333f4 81my_inet_aton(register const char *cp, struct in_addr *addr)
7e1af8bc 82{
c5be433b 83 dTHX;
0caed002 84 register U32 val;
7e1af8bc 85 register int base;
86 register char c;
87 int nparts;
88 const char *s;
89 unsigned int parts[4];
90 register unsigned int *pp = parts;
91
dfb6a7ef 92 if (!cp || !*cp)
0caed002 93 return 0;
7e1af8bc 94 for (;;) {
95 /*
96 * Collect number up to ``.''.
97 * Values are specified as for C:
98 * 0x=hex, 0=octal, other=decimal.
99 */
100 val = 0; base = 10;
101 if (*cp == '0') {
102 if (*++cp == 'x' || *cp == 'X')
103 base = 16, cp++;
104 else
105 base = 8;
106 }
107 while ((c = *cp) != '\0') {
108 if (isDIGIT(c)) {
109 val = (val * base) + (c - '0');
110 cp++;
111 continue;
112 }
3280af22 113 if (base == 16 && (s=strchr(PL_hexdigit,c))) {
a062834f 114 val = (val << 4) +
3280af22 115 ((s - PL_hexdigit) & 15);
7e1af8bc 116 cp++;
117 continue;
118 }
119 break;
120 }
121 if (*cp == '.') {
122 /*
123 * Internet format:
124 * a.b.c.d
125 * a.b.c (with c treated as 16-bits)
126 * a.b (with b treated as 24 bits)
127 */
128 if (pp >= parts + 3 || val > 0xff)
129 return 0;
130 *pp++ = val, cp++;
131 } else
132 break;
133 }
134 /*
135 * Check for trailing characters.
136 */
137 if (*cp && !isSPACE(*cp))
138 return 0;
139 /*
140 * Concoct the address according to
141 * the number of parts specified.
142 */
143 nparts = pp - parts + 1; /* force to an int for switch() */
144 switch (nparts) {
145
146 case 1: /* a -- 32 bits */
147 break;
148
149 case 2: /* a.b -- 8.24 bits */
150 if (val > 0xffffff)
151 return 0;
152 val |= parts[0] << 24;
153 break;
154
155 case 3: /* a.b.c -- 8.8.16 bits */
156 if (val > 0xffff)
157 return 0;
158 val |= (parts[0] << 24) | (parts[1] << 16);
159 break;
160
161 case 4: /* a.b.c.d -- 8.8.8.8 bits */
162 if (val > 0xff)
163 return 0;
164 val |= (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8);
165 break;
166 }
167 addr->s_addr = htonl(val);
168 return 1;
169}
170
171#undef inet_aton
172#define inet_aton my_inet_aton
173
174#endif /* ! HAS_INET_ATON */
175
8e07c86e 176
a0d0e21e 177static int
f0f333f4 178not_here(char *s)
a0d0e21e 179{
180 croak("Socket::%s not implemented on this architecture", s);
181 return -1;
182}
183
2528d3bc 184#define PERL_IN_ADDR_S_ADDR_SIZE 4
185
186/*
187* Bad assumptions possible here.
188*
189* Bad Assumption 1: struct in_addr has no other fields
190* than the s_addr (which is the field we care about
191* in here, really). However, we can be fed either 4-byte
192* addresses (from pack("N", ...), or va.b.c.d, or ...),
193* or full struct in_addrs (from e.g. pack_sockaddr_in()),
194* which may or may not be 4 bytes in size.
195*
196* Bad Assumption 2: the s_addr field is a simple type
197* (such as an int, u_int32_t). It can be a bit field,
198* in which case using & (address-of) on it or taking sizeof()
199* wouldn't go over too well. (Those are not attempted
200* now but in case someone thinks to change the below code
201* to use addr.s_addr instead of addr, you have been warned.)
202*
203* Bad Assumption 3: the s_addr is the first field in
204* an in_addr, or that its bytes are the first bytes in
205* an in_addr.
206*
207* These bad assumptions are wrong in UNICOS which has
208* struct in_addr { struct { u_long st_addr:32; } s_da };
209* #define s_addr s_da.st_addr
210* and u_long is 64 bits.
211*
212* --jhi */
213
1cb0fb50 214#include "const-c.inc"
8e07c86e 215
a0d0e21e 216MODULE = Socket PACKAGE = Socket
217
1cb0fb50 218INCLUDE: const-xs.inc
8e07c86e 219
220void
221inet_aton(host)
222 char * host
223 CODE:
224 {
fbf77aee 225 struct in_addr ip_address;
8e07c86e 226 struct hostent * phe;
64e1b767 227 int ok =
228 (host != NULL) &&
229 (*host != '\0') &&
230 inet_aton(host, &ip_address);
8e07c86e 231
fbf77aee 232 if (!ok && (phe = gethostbyname(host))) {
233 Copy( phe->h_addr, &ip_address, phe->h_length, char );
7e1af8bc 234 ok = 1;
8e07c86e 235 }
236
237 ST(0) = sv_newmortal();
2528d3bc 238 if (ok)
fbf77aee 239 sv_setpvn( ST(0), (char *)&ip_address, sizeof ip_address );
8e07c86e 240 }
8e07c86e 241
242void
243inet_ntoa(ip_address_sv)
244 SV * ip_address_sv
245 CODE:
246 {
247 STRLEN addrlen;
248 struct in_addr addr;
249 char * addr_str;
1d68b939 250 char * ip_address;
251 if (DO_UTF8(ip_address_sv) && !sv_utf8_downgrade(ip_address_sv, 1))
b2d93eb8 252 croak("Wide character in Socket::inet_ntoa");
f780604f 253 ip_address = SvPVbyte(ip_address_sv, addrlen);
2528d3bc 254 if (addrlen == sizeof(addr) || addrlen == 4)
255 addr.s_addr =
256 (ip_address[0] & 0xFF) << 24 |
257 (ip_address[1] & 0xFF) << 16 |
258 (ip_address[2] & 0xFF) << 8 |
259 (ip_address[3] & 0xFF);
260 else
261 croak("Bad arg length for %s, length is %d, should be %d",
262 "Socket::inet_ntoa",
263 addrlen, sizeof(addr));
264 /* We could use inet_ntoa() but that is broken
265 * in HP-UX + GCC + 64bitint (returns "0.0.0.0"),
266 * so let's use this sprintf() workaround everywhere. */
1d68b939 267 New(1138, addr_str, 4 * 3 + 3 + 1, char);
268 sprintf(addr_str, "%d.%d.%d.%d",
269 ((addr.s_addr >> 24) & 0xFF),
270 ((addr.s_addr >> 16) & 0xFF),
271 ((addr.s_addr >> 8) & 0xFF),
272 ( addr.s_addr & 0xFF));
273 ST(0) = sv_2mortal(newSVpvn(addr_str, strlen(addr_str)));
274 Safefree(addr_str);
8e07c86e 275 }
276
277void
2a84dff3 278sockaddr_family(sockaddr)
279 SV * sockaddr
280 PREINIT:
281 STRLEN sockaddr_len;
f780604f 282 char *sockaddr_pv = SvPVbyte(sockaddr, sockaddr_len);
2a84dff3 283 CODE:
284 if (sockaddr_len < offsetof(struct sockaddr, sa_data)) {
285 croak("Bad arg length for %s, length is %d, should be at least %d",
286 "Socket::sockaddr_family", sockaddr_len,
287 offsetof(struct sockaddr, sa_data));
288 }
289 ST(0) = sv_2mortal(newSViv(((struct sockaddr*)sockaddr_pv)->sa_family));
290
291void
4633a7c4 292pack_sockaddr_un(pathname)
293 char * pathname
294 CODE:
295 {
25f94b33 296#ifdef I_SYS_UN
4633a7c4 297 struct sockaddr_un sun_ad; /* fear using sun */
fcdb74fc 298 STRLEN len;
202975e6 299
4633a7c4 300 Zero( &sun_ad, sizeof sun_ad, char );
301 sun_ad.sun_family = AF_UNIX;
20408e3c 302 len = strlen(pathname);
303 if (len > sizeof(sun_ad.sun_path))
304 len = sizeof(sun_ad.sun_path);
202975e6 305# ifdef OS2 /* Name should start with \socket\ and contain backslashes! */
306 {
307 int off;
308 char *s, *e;
309
310 if (pathname[0] != '/' && pathname[0] != '\\')
311 croak("Relative UNIX domain socket name '%s' unsupported", pathname);
a062834f 312 else if (len < 8
202975e6 313 || pathname[7] != '/' && pathname[7] != '\\'
314 || !strnicmp(pathname + 1, "socket", 6))
315 off = 7;
316 else
317 off = 0; /* Preserve names starting with \socket\ */
318 Copy( "\\socket", sun_ad.sun_path, off, char);
319 Copy( pathname, sun_ad.sun_path + off, len, char );
320
321 s = sun_ad.sun_path + off - 1;
322 e = s + len + 1;
323 while (++s < e)
324 if (*s = '/')
325 *s = '\\';
326 }
a062834f 327# else /* !( defined OS2 ) */
20408e3c 328 Copy( pathname, sun_ad.sun_path, len, char );
202975e6 329# endif
a062834f 330 if (0) not_here("dummy");
79cb57f6 331 ST(0) = sv_2mortal(newSVpvn((char *)&sun_ad, sizeof sun_ad));
25f94b33 332#else
333 ST(0) = (SV *) not_here("pack_sockaddr_un");
334#endif
335
4633a7c4 336 }
337
338void
339unpack_sockaddr_un(sun_sv)
340 SV * sun_sv
f13e1b2f 341 CODE:
4633a7c4 342 {
25f94b33 343#ifdef I_SYS_UN
4633a7c4 344 struct sockaddr_un addr;
fcdb74fc 345 STRLEN sockaddrlen;
f780604f 346 char * sun_ad = SvPVbyte(sun_sv,sockaddrlen);
fcdb74fc 347 char * e;
36902e12 348# ifndef __linux__
349 /* On Linux sockaddrlen on sockets returned by accept, recvfrom,
350 getpeername and getsockname is not equal to sizeof(addr). */
4633a7c4 351 if (sockaddrlen != sizeof(addr)) {
352 croak("Bad arg length for %s, length is %d, should be %d",
353 "Socket::unpack_sockaddr_un",
354 sockaddrlen, sizeof(addr));
355 }
36902e12 356# endif
4633a7c4 357
358 Copy( sun_ad, &addr, sizeof addr, char );
359
360 if ( addr.sun_family != AF_UNIX ) {
361 croak("Bad address family for %s, got %d, should be %d",
362 "Socket::unpack_sockaddr_un",
363 addr.sun_family,
364 AF_UNIX);
fcdb74fc 365 }
366 e = addr.sun_path;
367 while (*e && e < addr.sun_path + sizeof addr.sun_path)
368 ++e;
79cb57f6 369 ST(0) = sv_2mortal(newSVpvn(addr.sun_path, e - addr.sun_path));
25f94b33 370#else
371 ST(0) = (SV *) not_here("unpack_sockaddr_un");
372#endif
4633a7c4 373 }
374
375void
2528d3bc 376pack_sockaddr_in(port, ip_address_sv)
2c129a17 377 unsigned short port
2528d3bc 378 SV * ip_address_sv
8e07c86e 379 CODE:
380 {
381 struct sockaddr_in sin;
2528d3bc 382 struct in_addr addr;
383 STRLEN addrlen;
384 char * ip_address;
385 if (DO_UTF8(ip_address_sv) && !sv_utf8_downgrade(ip_address_sv, 1))
386 croak("Wide character in Socket::pack_sockaddr_in");
f780604f 387 ip_address = SvPVbyte(ip_address_sv, addrlen);
2528d3bc 388 if (addrlen == sizeof(addr) || addrlen == 4)
389 addr.s_addr =
390 (ip_address[0] & 0xFF) << 24 |
391 (ip_address[1] & 0xFF) << 16 |
392 (ip_address[2] & 0xFF) << 8 |
393 (ip_address[3] & 0xFF);
394 else
395 croak("Bad arg length for %s, length is %d, should be %d",
396 "Socket::pack_sockaddr_in",
397 addrlen, sizeof(addr));
8e07c86e 398 Zero( &sin, sizeof sin, char );
4633a7c4 399 sin.sin_family = AF_INET;
8e07c86e 400 sin.sin_port = htons(port);
2528d3bc 401 sin.sin_addr.s_addr = htonl(addr.s_addr);
79cb57f6 402 ST(0) = sv_2mortal(newSVpvn((char *)&sin, sizeof sin));
8e07c86e 403 }
404
405void
406unpack_sockaddr_in(sin_sv)
407 SV * sin_sv
408 PPCODE:
409 {
410 STRLEN sockaddrlen;
411 struct sockaddr_in addr;
2c129a17 412 unsigned short port;
2528d3bc 413 struct in_addr ip_address;
f780604f 414 char * sin = SvPVbyte(sin_sv,sockaddrlen);
8e07c86e 415 if (sockaddrlen != sizeof(addr)) {
416 croak("Bad arg length for %s, length is %d, should be %d",
417 "Socket::unpack_sockaddr_in",
418 sockaddrlen, sizeof(addr));
419 }
8e07c86e 420 Copy( sin, &addr,sizeof addr, char );
4633a7c4 421 if ( addr.sin_family != AF_INET ) {
422 croak("Bad address family for %s, got %d, should be %d",
423 "Socket::unpack_sockaddr_in",
424 addr.sin_family,
425 AF_INET);
a062834f 426 }
8e07c86e 427 port = ntohs(addr.sin_port);
428 ip_address = addr.sin_addr;
429
924508f0 430 EXTEND(SP, 2);
2c129a17 431 PUSHs(sv_2mortal(newSViv((IV) port)));
2528d3bc 432 PUSHs(sv_2mortal(newSVpvn((char *)&ip_address, sizeof ip_address)));
8e07c86e 433 }