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