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