[p5sagit/p5-mst-13.2.git] / ext / Socket / Socket.xs

#define PERL_NO_GET_CONTEXT
#include "EXTERN.h"
#include "perl.h"
#include "XSUB.h"

#ifndef VMS
# ifdef I_SYS_TYPES
#  include <sys/types.h>
# endif
# include <sys/socket.h>
# if defined(USE_SOCKS) && defined(I_SOCKS)
#   include <socks.h>
# endif
# ifdef MPE
#  define PF_INET AF_INET
#  define PF_UNIX AF_UNIX
#  define SOCK_RAW 3
# endif
# ifdef I_SYS_UN
#  include <sys/un.h>
# endif
/* XXX Configure test for <netinet/in_systm.h needed XXX */
# if defined(NeXT) || defined(__NeXT__)
#  include <netinet/in_systm.h>
# endif
# ifdef I_NETINET_IN
#  include <netinet/in.h>
# endif
# ifdef I_NETDB
#  include <netdb.h>
# endif
# ifdef I_ARPA_INET
#  include <arpa/inet.h>
# endif
# ifdef I_NETINET_TCP
#  include <netinet/tcp.h>
# endif
#else
# include "sockadapt.h"
#endif

#ifdef I_SYSUIO
# include <sys/uio.h>
#endif

#ifndef AF_NBS
# undef PF_NBS
#endif

#ifndef AF_X25
# undef PF_X25
#endif

#ifndef INADDR_NONE
# define INADDR_NONE	0xffffffff
#endif /* INADDR_NONE */
#ifndef INADDR_BROADCAST
# define INADDR_BROADCAST	0xffffffff
#endif /* INADDR_BROADCAST */
#ifndef INADDR_LOOPBACK
# define INADDR_LOOPBACK         0x7F000001
#endif /* INADDR_LOOPBACK */

#ifndef HAS_INET_ATON

/*
 * Check whether "cp" is a valid ascii representation
 * of an Internet address and convert to a binary address.
 * Returns 1 if the address is valid, 0 if not.
 * This replaces inet_addr, the return value from which
 * cannot distinguish between failure and a local broadcast address.
 */
static int
my_inet_aton(register const char *cp, struct in_addr *addr)
{
	dTHX;
	register U32 val;
	register int base;
	register char c;
	int nparts;
	const char *s;
	unsigned int parts[4];
	register unsigned int *pp = parts;

	if (!cp)
		return 0;
	for (;;) {
		/*
		 * Collect number up to ``.''.
		 * Values are specified as for C:
		 * 0x=hex, 0=octal, other=decimal.
		 */
		val = 0; base = 10;
		if (*cp == '0') {
			if (*++cp == 'x' || *cp == 'X')
				base = 16, cp++;
			else
				base = 8;
		}
		while ((c = *cp) != '\0') {
			if (isDIGIT(c)) {
				val = (val * base) + (c - '0');
				cp++;
				continue;
			}
			if (base == 16 && (s=strchr(PL_hexdigit,c))) {
				val = (val << 4) +
					((s - PL_hexdigit) & 15);
				cp++;
				continue;
			}
			break;
		}
		if (*cp == '.') {
			/*
			 * Internet format:
			 *	a.b.c.d
			 *	a.b.c	(with c treated as 16-bits)
			 *	a.b	(with b treated as 24 bits)
			 */
			if (pp >= parts + 3 || val > 0xff)
				return 0;
			*pp++ = val, cp++;
		} else
			break;
	}
	/*
	 * Check for trailing characters.
	 */
	if (*cp && !isSPACE(*cp))
		return 0;
	/*
	 * Concoct the address according to
	 * the number of parts specified.
	 */
	nparts = pp - parts + 1;	/* force to an int for switch() */
	switch (nparts) {

	case 1:				/* a -- 32 bits */
		break;

	case 2:				/* a.b -- 8.24 bits */
		if (val > 0xffffff)
			return 0;
		val |= parts[0] << 24;
		break;

	case 3:				/* a.b.c -- 8.8.16 bits */
		if (val > 0xffff)
			return 0;
		val |= (parts[0] << 24) | (parts[1] << 16);
		break;

	case 4:				/* a.b.c.d -- 8.8.8.8 bits */
		if (val > 0xff)
			return 0;
		val |= (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8);
		break;
	}
	addr->s_addr = htonl(val);
	return 1;
}

#undef inet_aton
#define inet_aton my_inet_aton

#endif /* ! HAS_INET_ATON */


static int
not_here(char *s)
{
    croak("Socket::%s not implemented on this architecture", s);
    return -1;
}

#include "constants.c"

MODULE = Socket		PACKAGE = Socket

INCLUDE: constants.xs

void
inet_aton(host)
	char *	host
	CODE:
	{
	struct in_addr ip_address;
	struct hostent * phe;
	int ok = inet_aton(host, &ip_address);

	if (!ok && (phe = gethostbyname(host))) {
		Copy( phe->h_addr, &ip_address, phe->h_length, char );
		ok = 1;
	}

	ST(0) = sv_newmortal();
	if (ok) {
		sv_setpvn( ST(0), (char *)&ip_address, sizeof ip_address );
	}
	}

void
inet_ntoa(ip_address_sv)
	SV *	ip_address_sv
	CODE:
	{
	STRLEN addrlen;
	struct in_addr addr;
	char * addr_str;
	char * ip_address = SvPV(ip_address_sv,addrlen);
	if (addrlen != sizeof(addr)) {
	    croak("Bad arg length for %s, length is %d, should be %d",
			"Socket::inet_ntoa",
			addrlen, sizeof(addr));
	}

	Copy( ip_address, &addr, sizeof addr, char );
	addr_str = inet_ntoa(addr);

	ST(0) = sv_2mortal(newSVpvn(addr_str, strlen(addr_str)));
	}

void
pack_sockaddr_un(pathname)
	char *	pathname
	CODE:
	{
#ifdef I_SYS_UN
	struct sockaddr_un sun_ad; /* fear using sun */
	STRLEN len;

	Zero( &sun_ad, sizeof sun_ad, char );
	sun_ad.sun_family = AF_UNIX;
	len = strlen(pathname);
	if (len > sizeof(sun_ad.sun_path))
	    len = sizeof(sun_ad.sun_path);
#  ifdef OS2	/* Name should start with \socket\ and contain backslashes! */
	{
	    int off;
	    char *s, *e;

	    if (pathname[0] != '/' && pathname[0] != '\\')
		croak("Relative UNIX domain socket name '%s' unsupported", pathname);
	    else if (len < 8
		     || pathname[7] != '/' && pathname[7] != '\\'
		     || !strnicmp(pathname + 1, "socket", 6))
		off = 7;
	    else
		off = 0;		/* Preserve names starting with \socket\ */
	    Copy( "\\socket", sun_ad.sun_path, off, char);
	    Copy( pathname, sun_ad.sun_path + off, len, char );

	    s = sun_ad.sun_path + off - 1;
	    e = s + len + 1;
	    while (++s < e)
		if (*s = '/')
		    *s = '\\';
	}
#  else	/* !( defined OS2 ) */
	Copy( pathname, sun_ad.sun_path, len, char );
#  endif
	if (0) not_here("dummy");
	ST(0) = sv_2mortal(newSVpvn((char *)&sun_ad, sizeof sun_ad));
#else
	ST(0) = (SV *) not_here("pack_sockaddr_un");
#endif
	
	}

void
unpack_sockaddr_un(sun_sv)
	SV *	sun_sv
	CODE:
	{
#ifdef I_SYS_UN
	struct sockaddr_un addr;
	STRLEN sockaddrlen;
	char * sun_ad = SvPV(sun_sv,sockaddrlen);
	char * e;
#   ifndef __linux__
	/* On Linux sockaddrlen on sockets returned by accept, recvfrom,
	   getpeername and getsockname is not equal to sizeof(addr). */
	if (sockaddrlen != sizeof(addr)) {
	    croak("Bad arg length for %s, length is %d, should be %d",
			"Socket::unpack_sockaddr_un",
			sockaddrlen, sizeof(addr));
	}
#   endif

	Copy( sun_ad, &addr, sizeof addr, char );

	if ( addr.sun_family != AF_UNIX ) {
	    croak("Bad address family for %s, got %d, should be %d",
			"Socket::unpack_sockaddr_un",
			addr.sun_family,
			AF_UNIX);
	}
	e = addr.sun_path;
	while (*e && e < addr.sun_path + sizeof addr.sun_path)
	    ++e;
	ST(0) = sv_2mortal(newSVpvn(addr.sun_path, e - addr.sun_path));
#else
	ST(0) = (SV *) not_here("unpack_sockaddr_un");
#endif
	}

void
pack_sockaddr_in(port,ip_address)
	unsigned short	port
	char *	ip_address
	CODE:
	{
	struct sockaddr_in sin;

	Zero( &sin, sizeof sin, char );
	sin.sin_family = AF_INET;
	sin.sin_port = htons(port);
	Copy( ip_address, &sin.sin_addr, sizeof sin.sin_addr, char );

	ST(0) = sv_2mortal(newSVpvn((char *)&sin, sizeof sin));
	}

void
unpack_sockaddr_in(sin_sv)
	SV *	sin_sv
	PPCODE:
	{
	STRLEN sockaddrlen;
	struct sockaddr_in addr;
	unsigned short	port;
	struct in_addr	ip_address;
	char *	sin = SvPV(sin_sv,sockaddrlen);
	if (sockaddrlen != sizeof(addr)) {
	    croak("Bad arg length for %s, length is %d, should be %d",
			"Socket::unpack_sockaddr_in",
			sockaddrlen, sizeof(addr));
	}
	Copy( sin, &addr,sizeof addr, char );
	if ( addr.sin_family != AF_INET ) {
	    croak("Bad address family for %s, got %d, should be %d",
			"Socket::unpack_sockaddr_in",
			addr.sin_family,
			AF_INET);
	}
	port = ntohs(addr.sin_port);
	ip_address = addr.sin_addr;

	EXTEND(SP, 2);
	PUSHs(sv_2mortal(newSViv((IV) port)));
	PUSHs(sv_2mortal(newSVpvn((char *)&ip_address,sizeof ip_address)));
	}
Commit	Line	Data
c5be433b	1	#define PERL_NO_GET_CONTEXT
a0d0e21e	2	#include "EXTERN.h"
	3	#include "perl.h"
	4	#include "XSUB.h"
	5
8e07c86e	6	#ifndef VMS
	7	# ifdef I_SYS_TYPES
	8	# include <sys/types.h>
	9	# endif
9cc6feab	10	# include <sys/socket.h>
29209bc5	11	# if defined(USE_SOCKS) && defined(I_SOCKS)
86959918	12	# include <socks.h>
a062834f	13	# endif
9cc6feab	14	# ifdef MPE
	15	# define PF_INET AF_INET
	16	# define PF_UNIX AF_UNIX
	17	# define SOCK_RAW 3
	18	# endif
	19	# ifdef I_SYS_UN
	20	# include <sys/un.h>
	21	# endif
bf9b2f8f	22	/* XXX Configure test for <netinet/in_systm.h needed XXX */
	23	# if defined(NeXT) \|\| defined(__NeXT__)
	24	# include <netinet/in_systm.h>
	25	# endif
8e07c86e	26	# ifdef I_NETINET_IN
	27	# include <netinet/in.h>
	28	# endif
86959918	29	# ifdef I_NETDB
	30	# include <netdb.h>
	31	# endif
9cc6feab	32	# ifdef I_ARPA_INET
	33	# include <arpa/inet.h>
	34	# endif
	35	# ifdef I_NETINET_TCP
	36	# include <netinet/tcp.h>
	37	# endif
8e07c86e	38	#else
9cc6feab	39	# include "sockadapt.h"
8e07c86e	40	#endif
a0d0e21e	41
6b1016b5	42	#ifdef I_SYSUIO
	43	# include <sys/uio.h>
	44	#endif
	45
a0d0e21e	46	#ifndef AF_NBS
9cc6feab	47	# undef PF_NBS
a0d0e21e	48	#endif
	49
	50	#ifndef AF_X25
9cc6feab	51	# undef PF_X25
a0d0e21e	52	#endif
a0d0e21e	53
8e07c86e	54	#ifndef INADDR_NONE
9cc6feab	55	# define INADDR_NONE 0xffffffff
8e07c86e	56	#endif /* INADDR_NONE */
7e1af8bc	57	#ifndef INADDR_BROADCAST
9cc6feab	58	# define INADDR_BROADCAST 0xffffffff
7e1af8bc	59	#endif /* INADDR_BROADCAST */
8e07c86e	60	#ifndef INADDR_LOOPBACK
9cc6feab	61	# define INADDR_LOOPBACK 0x7F000001
8e07c86e	62	#endif /* INADDR_LOOPBACK */
8e07c86e	63
7e1af8bc	64	#ifndef HAS_INET_ATON
7e1af8bc	65
a062834f	66	/*
7e1af8bc	67	* Check whether "cp" is a valid ascii representation
	68	* of an Internet address and convert to a binary address.
	69	* Returns 1 if the address is valid, 0 if not.
	70	* This replaces inet_addr, the return value from which
	71	* cannot distinguish between failure and a local broadcast address.
	72	*/
	73	static int
f0f333f4	74	my_inet_aton(register const char cp, struct in_addr addr)
7e1af8bc	75	{
c5be433b	76	dTHX;
0caed002	77	register U32 val;
7e1af8bc	78	register int base;
	79	register char c;
	80	int nparts;
	81	const char *s;
	82	unsigned int parts[4];
	83	register unsigned int *pp = parts;
	84
0caed002	85	if (!cp)
0caed002	86	return 0;
7e1af8bc	87	for (;;) {
	88	/*
	89	* Collect number up to ``.''.
	90	* Values are specified as for C:
	91	* 0x=hex, 0=octal, other=decimal.
	92	*/
	93	val = 0; base = 10;
	94	if (*cp == '0') {
	95	if (++cp == 'x' \|\| cp == 'X')
	96	base = 16, cp++;
	97	else
	98	base = 8;
	99	}
	100	while ((c = *cp) != '\0') {
	101	if (isDIGIT(c)) {
	102	val = (val * base) + (c - '0');
	103	cp++;
	104	continue;
	105	}
3280af22	106	if (base == 16 && (s=strchr(PL_hexdigit,c))) {
a062834f	107	val = (val << 4) +
3280af22	108	((s - PL_hexdigit) & 15);
7e1af8bc	109	cp++;
	110	continue;
	111	}
	112	break;
	113	}
	114	if (*cp == '.') {
	115	/*
	116	* Internet format:
	117	* a.b.c.d
	118	* a.b.c (with c treated as 16-bits)
	119	* a.b (with b treated as 24 bits)
	120	*/
	121	if (pp >= parts + 3 \|\| val > 0xff)
	122	return 0;
	123	*pp++ = val, cp++;
	124	} else
	125	break;
	126	}
	127	/*
	128	* Check for trailing characters.
	129	*/
	130	if (cp && !isSPACE(cp))
	131	return 0;
	132	/*
	133	* Concoct the address according to
	134	* the number of parts specified.
	135	*/
	136	nparts = pp - parts + 1; /* force to an int for switch() */
	137	switch (nparts) {
	138
	139	case 1: /* a -- 32 bits */
	140	break;
	141
	142	case 2: /* a.b -- 8.24 bits */
	143	if (val > 0xffffff)
	144	return 0;
	145	val \|= parts[0] << 24;
	146	break;
	147
	148	case 3: /* a.b.c -- 8.8.16 bits */
	149	if (val > 0xffff)
	150	return 0;
	151	val \|= (parts[0] << 24) \| (parts[1] << 16);
	152	break;
	153
	154	case 4: /* a.b.c.d -- 8.8.8.8 bits */
	155	if (val > 0xff)
	156	return 0;
	157	val \|= (parts[0] << 24) \| (parts[1] << 16) \| (parts[2] << 8);
	158	break;
	159	}
	160	addr->s_addr = htonl(val);
	161	return 1;
	162	}
	163
	164	#undef inet_aton
	165	#define inet_aton my_inet_aton
	166
	167	#endif /* ! HAS_INET_ATON */
	168
8e07c86e	169
a0d0e21e	170	static int
f0f333f4	171	not_here(char *s)
a0d0e21e	172	{
	173	croak("Socket::%s not implemented on this architecture", s);
	174	return -1;
	175	}
	176
ee96af8f	177	#include "constants.c"
8e07c86e	178
a0d0e21e	179	MODULE = Socket PACKAGE = Socket
a0d0e21e	180
ee96af8f	181	INCLUDE: constants.xs
8e07c86e	182
	183	void
	184	inet_aton(host)
	185	char * host
	186	CODE:
	187	{
	188	struct in_addr ip_address;
	189	struct hostent * phe;
ac9fe1c2	190	int ok = inet_aton(host, &ip_address);
8e07c86e	191
ac9fe1c2	192	if (!ok && (phe = gethostbyname(host))) {
8e07c86e	193	Copy( phe->h_addr, &ip_address, phe->h_length, char );
7e1af8bc	194	ok = 1;
8e07c86e	195	}
	196
	197	ST(0) = sv_newmortal();
7e1af8bc	198	if (ok) {
8e07c86e	199	sv_setpvn( ST(0), (char *)&ip_address, sizeof ip_address );
	200	}
	201	}
	202
	203	void
	204	inet_ntoa(ip_address_sv)
	205	SV * ip_address_sv
	206	CODE:
	207	{
	208	STRLEN addrlen;
	209	struct in_addr addr;
	210	char * addr_str;
	211	char * ip_address = SvPV(ip_address_sv,addrlen);
	212	if (addrlen != sizeof(addr)) {
	213	croak("Bad arg length for %s, length is %d, should be %d",
	214	"Socket::inet_ntoa",
	215	addrlen, sizeof(addr));
	216	}
	217
	218	Copy( ip_address, &addr, sizeof addr, char );
	219	addr_str = inet_ntoa(addr);
	220
79cb57f6	221	ST(0) = sv_2mortal(newSVpvn(addr_str, strlen(addr_str)));
8e07c86e	222	}
	223
	224	void
4633a7c4	225	pack_sockaddr_un(pathname)
	226	char * pathname
	227	CODE:
	228	{
25f94b33	229	#ifdef I_SYS_UN
4633a7c4	230	struct sockaddr_un sun_ad; /* fear using sun */
fcdb74fc	231	STRLEN len;
202975e6	232
4633a7c4	233	Zero( &sun_ad, sizeof sun_ad, char );
4633a7c4	234	sun_ad.sun_family = AF_UNIX;
20408e3c	235	len = strlen(pathname);
	236	if (len > sizeof(sun_ad.sun_path))
	237	len = sizeof(sun_ad.sun_path);
202975e6	238	# ifdef OS2 /* Name should start with \socket\ and contain backslashes! */
	239	{
	240	int off;
	241	char s, e;
	242
	243	if (pathname[0] != '/' && pathname[0] != '\\')
	244	croak("Relative UNIX domain socket name '%s' unsupported", pathname);
a062834f	245	else if (len < 8
202975e6	246	\|\| pathname[7] != '/' && pathname[7] != '\\'
	247	\|\| !strnicmp(pathname + 1, "socket", 6))
	248	off = 7;
	249	else
	250	off = 0; /* Preserve names starting with \socket\ */
	251	Copy( "\\socket", sun_ad.sun_path, off, char);
	252	Copy( pathname, sun_ad.sun_path + off, len, char );
	253
	254	s = sun_ad.sun_path + off - 1;
	255	e = s + len + 1;
	256	while (++s < e)
	257	if (*s = '/')
	258	*s = '\\';
	259	}
a062834f	260	# else /* !( defined OS2 ) */
20408e3c	261	Copy( pathname, sun_ad.sun_path, len, char );
202975e6	262	# endif
a062834f	263	if (0) not_here("dummy");
79cb57f6	264	ST(0) = sv_2mortal(newSVpvn((char *)&sun_ad, sizeof sun_ad));
25f94b33	265	#else
	266	ST(0) = (SV *) not_here("pack_sockaddr_un");
	267	#endif
	268
4633a7c4	269	}
	270
	271	void
	272	unpack_sockaddr_un(sun_sv)
	273	SV * sun_sv
f13e1b2f	274	CODE:
4633a7c4	275	{
25f94b33	276	#ifdef I_SYS_UN
4633a7c4	277	struct sockaddr_un addr;
fcdb74fc	278	STRLEN sockaddrlen;
	279	char * sun_ad = SvPV(sun_sv,sockaddrlen);
	280	char * e;
36902e12	281	# ifndef __linux__
	282	/* On Linux sockaddrlen on sockets returned by accept, recvfrom,
	283	getpeername and getsockname is not equal to sizeof(addr). */
4633a7c4	284	if (sockaddrlen != sizeof(addr)) {
	285	croak("Bad arg length for %s, length is %d, should be %d",
	286	"Socket::unpack_sockaddr_un",
	287	sockaddrlen, sizeof(addr));
	288	}
36902e12	289	# endif
4633a7c4	290
	291	Copy( sun_ad, &addr, sizeof addr, char );
	292
	293	if ( addr.sun_family != AF_UNIX ) {
	294	croak("Bad address family for %s, got %d, should be %d",
	295	"Socket::unpack_sockaddr_un",
	296	addr.sun_family,
	297	AF_UNIX);
fcdb74fc	298	}
	299	e = addr.sun_path;
	300	while (*e && e < addr.sun_path + sizeof addr.sun_path)
	301	++e;
79cb57f6	302	ST(0) = sv_2mortal(newSVpvn(addr.sun_path, e - addr.sun_path));
25f94b33	303	#else
	304	ST(0) = (SV *) not_here("unpack_sockaddr_un");
	305	#endif
4633a7c4	306	}
	307
	308	void
	309	pack_sockaddr_in(port,ip_address)
2c129a17	310	unsigned short port
8e07c86e	311	char * ip_address
	312	CODE:
	313	{
	314	struct sockaddr_in sin;
	315
	316	Zero( &sin, sizeof sin, char );
4633a7c4	317	sin.sin_family = AF_INET;
8e07c86e	318	sin.sin_port = htons(port);
	319	Copy( ip_address, &sin.sin_addr, sizeof sin.sin_addr, char );
	320
79cb57f6	321	ST(0) = sv_2mortal(newSVpvn((char *)&sin, sizeof sin));
8e07c86e	322	}
	323
	324	void
	325	unpack_sockaddr_in(sin_sv)
	326	SV * sin_sv
	327	PPCODE:
	328	{
	329	STRLEN sockaddrlen;
	330	struct sockaddr_in addr;
2c129a17	331	unsigned short port;
8e07c86e	332	struct in_addr ip_address;
	333	char * sin = SvPV(sin_sv,sockaddrlen);
	334	if (sockaddrlen != sizeof(addr)) {
	335	croak("Bad arg length for %s, length is %d, should be %d",
	336	"Socket::unpack_sockaddr_in",
	337	sockaddrlen, sizeof(addr));
	338	}
8e07c86e	339	Copy( sin, &addr,sizeof addr, char );
4633a7c4	340	if ( addr.sin_family != AF_INET ) {
	341	croak("Bad address family for %s, got %d, should be %d",
	342	"Socket::unpack_sockaddr_in",
	343	addr.sin_family,
	344	AF_INET);
a062834f	345	}
8e07c86e	346	port = ntohs(addr.sin_port);
	347	ip_address = addr.sin_addr;
	348
924508f0	349	EXTEND(SP, 2);
2c129a17	350	PUSHs(sv_2mortal(newSViv((IV) port)));
79cb57f6	351	PUSHs(sv_2mortal(newSVpvn((char *)&ip_address,sizeof ip_address)));
8e07c86e	352	}