|
Packit |
8bbd3c |
#!/usr/bin/perl
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
use strict;
|
|
Packit |
8bbd3c |
use warnings;
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
use Test::More;
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
use IO::Socket::IP;
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
use IO::Socket::INET;
|
|
Packit |
8bbd3c |
use Socket qw( inet_aton inet_ntoa pack_sockaddr_in unpack_sockaddr_in );
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
# Some odd locations like BSD jails might not like INADDR_LOOPBACK. We'll
|
|
Packit |
8bbd3c |
# establish a baseline first to test against
|
|
Packit |
8bbd3c |
my $INADDR_LOOPBACK = do {
|
|
Packit |
8bbd3c |
socket my $sockh, PF_INET, SOCK_STREAM, 0 or die "Cannot socket(PF_INET) - $!";
|
|
Packit |
8bbd3c |
bind $sockh, pack_sockaddr_in( 0, inet_aton( "127.0.0.1" ) ) or die "Cannot bind() - $!";
|
|
Packit |
8bbd3c |
( unpack_sockaddr_in( getsockname $sockh ) )[1];
|
|
Packit |
8bbd3c |
};
|
|
Packit |
8bbd3c |
my $INADDR_LOOPBACK_HOST = inet_ntoa( $INADDR_LOOPBACK );
|
|
Packit |
8bbd3c |
if( $INADDR_LOOPBACK ne INADDR_LOOPBACK ) {
|
|
Packit |
8bbd3c |
diag( "Testing with INADDR_LOOPBACK=$INADDR_LOOPBACK_HOST; this may be because of odd networking" );
|
|
Packit |
8bbd3c |
}
|
|
Packit |
8bbd3c |
my $INADDR_LOOPBACK_HEX = unpack "H*", $INADDR_LOOPBACK;
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
foreach my $socktype (qw( SOCK_STREAM SOCK_DGRAM )) {
|
|
Packit |
8bbd3c |
my $testserver = IO::Socket::INET->new(
|
|
Packit |
8bbd3c |
( $socktype eq "SOCK_STREAM" ? ( Listen => 1 ) : () ),
|
|
Packit |
8bbd3c |
LocalHost => "127.0.0.1",
|
|
Packit |
8bbd3c |
Type => Socket->$socktype,
|
|
Packit |
8bbd3c |
Proto => ( $socktype eq "SOCK_STREAM" ? "tcp" : "udp" ), # Because IO::Socket::INET is stupid and always presumes tcp
|
|
Packit |
8bbd3c |
) or die "Cannot listen on PF_INET - $@";
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
my $socket = IO::Socket::IP->new(
|
|
Packit |
8bbd3c |
PeerHost => "127.0.0.1",
|
|
Packit |
8bbd3c |
PeerService => $testserver->sockport,
|
|
Packit |
8bbd3c |
Type => Socket->$socktype,
|
|
Packit |
8bbd3c |
);
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
ok( defined $socket, "IO::Socket::IP->new constructs a $socktype socket" ) or
|
|
Packit |
8bbd3c |
diag( " error was $@" );
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
is( $socket->sockdomain, AF_INET, "\$socket->sockdomain for $socktype" );
|
|
Packit |
8bbd3c |
is( $socket->socktype, Socket->$socktype, "\$socket->socktype for $socktype" );
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
my $testclient = ( $socktype eq "SOCK_STREAM" ) ?
|
|
Packit |
8bbd3c |
$testserver->accept :
|
|
Packit |
8bbd3c |
do { $testserver->connect( $socket->sockname ); $testserver };
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
ok( defined $testclient, "accepted test $socktype client" );
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
ok( $socket->connected, "\$socket is connected for $socktype" );
|
|
Packit |
8bbd3c |
ok( $socket->blocking, "\$socket is in blocking mode after connect for $socktype" );
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
is_deeply( [ unpack_sockaddr_in $socket->sockname ],
|
|
Packit |
8bbd3c |
[ unpack_sockaddr_in $testclient->peername ],
|
|
Packit |
8bbd3c |
"\$socket->sockname for $socktype" );
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
is_deeply( [ unpack_sockaddr_in $socket->peername ],
|
|
Packit |
8bbd3c |
[ unpack_sockaddr_in $testclient->sockname ],
|
|
Packit |
8bbd3c |
"\$socket->peername for $socktype" );
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
is( $socket->peerhost, $INADDR_LOOPBACK_HOST, "\$socket->peerhost for $socktype" );
|
|
Packit |
8bbd3c |
is( $socket->peerport, $testserver->sockport, "\$socket->peerport for $socktype" );
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
# Unpack just so it pretty prints without wrecking the terminal if it fails
|
|
Packit |
8bbd3c |
is( unpack("H*", $socket->sockaddr), $INADDR_LOOPBACK_HEX, "\$socket->sockaddr for $socktype" );
|
|
Packit |
8bbd3c |
is( unpack("H*", $socket->peeraddr), $INADDR_LOOPBACK_HEX, "\$socket->peeraddr for $socktype" );
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
# Can't easily test the non-numeric versions without relying on the system's
|
|
Packit |
8bbd3c |
# ability to resolve the name "localhost"
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
$socket->close;
|
|
Packit |
8bbd3c |
ok( !$socket->connected, "\$socket not connected after close for $socktype" );
|
|
Packit |
8bbd3c |
}
|
|
Packit |
8bbd3c |
|
|
Packit |
8bbd3c |
done_testing;
|