1#!/usr/bin/perl
2
3use v5.14;
4use warnings;
5
6use Test::More;
7
8use IO::Socket::IP;
9use Socket qw( inet_pton inet_ntop pack_sockaddr_in6 unpack_sockaddr_in6 IN6ADDR_LOOPBACK );
10
11my $AF_INET6 = eval { Socket::AF_INET6() } or
12   plan skip_all => "No AF_INET6";
13
14# Some odd locations like BSD jails might not like IN6ADDR_LOOPBACK. We'll
15# establish a baseline first to test against
16my $IN6ADDR_LOOPBACK = eval {
17   socket my $sockh, Socket::PF_INET6(), SOCK_STREAM, 0 or die "Cannot socket(PF_INET6) - $!";
18   bind $sockh, pack_sockaddr_in6( 0, inet_pton( $AF_INET6, "::1" ) ) or die "Cannot bind() - $!";
19   ( unpack_sockaddr_in6( getsockname $sockh ) )[1];
20} or plan skip_all => "Unable to bind to ::1 - $@";
21my $IN6ADDR_LOOPBACK_HOST = inet_ntop( $AF_INET6, $IN6ADDR_LOOPBACK );
22if( $IN6ADDR_LOOPBACK ne IN6ADDR_LOOPBACK ) {
23   diag( "Testing with IN6ADDR_LOOPBACK=$IN6ADDR_LOOPBACK_HOST; this may be because of odd networking" );
24}
25my $IN6ADDR_LOOPBACK_HEX = unpack "H*", $IN6ADDR_LOOPBACK;
26
27# Unpack just ip6_addr and port because other fields might not match end to end
28sub unpack_sockaddr_in6_addrport {
29   return ( Socket::unpack_sockaddr_in6( shift ) )[0,1];
30}
31
32foreach my $socktype (qw( SOCK_STREAM SOCK_DGRAM )) {
33   my $testserver = IO::Socket::IP->new(
34      ( $socktype eq "SOCK_STREAM" ? ( Listen => 1 ) : () ),
35      LocalHost => "::1",
36      LocalPort => "0",
37      Type      => Socket->$socktype,
38      GetAddrInfoFlags => 0, # disable AI_ADDRCONFIG
39   );
40
41   ok( defined $testserver, "IO::Socket::IP->new constructs a $socktype socket" ) or
42      diag( "  error was $IO::Socket::errstr" );
43
44   is( $testserver->sockdomain, $AF_INET6,         "\$testserver->sockdomain for $socktype" );
45   is( $testserver->socktype,   Socket->$socktype, "\$testserver->socktype for $socktype" );
46
47   is( $testserver->sockhost, $IN6ADDR_LOOPBACK_HOST, "\$testserver->sockhost for $socktype" );
48   like( $testserver->sockport, qr/^\d+$/,            "\$testserver->sockport for $socktype" );
49
50   ok( eval { $testserver->peerport; 1 }, "\$testserver->peerport does not die for $socktype" )
51      or do { chomp( my $e = $@ ); diag( "Exception was: $e" ) };
52
53   my $socket = IO::Socket->new;
54   $socket->socket( $AF_INET6, Socket->$socktype, 0 )
55      or die "Cannot socket() - $!";
56
57   my ( $err, $ai ) = Socket::getaddrinfo( "::1", $testserver->sockport, { family => $AF_INET6, socktype => Socket->$socktype } );
58   die "getaddrinfo() - $err" if $err;
59
60   $socket->connect( $ai->{addr} ) or die "Cannot connect() - $!";
61
62   my $testclient = ( $socktype eq "SOCK_STREAM" ) ?
63      $testserver->accept :
64      do { $testserver->connect( $socket->sockname ); $testserver };
65
66   ok( defined $testclient, "accepted test $socktype client" );
67   isa_ok( $testclient, "IO::Socket::IP", "\$testclient for $socktype" );
68
69   is( $testclient->sockdomain, $AF_INET6,         "\$testclient->sockdomain for $socktype" );
70   is( $testclient->socktype,   Socket->$socktype, "\$testclient->socktype for $socktype" );
71
72   is_deeply( [ unpack_sockaddr_in6_addrport( $socket->sockname ) ],
73              [ unpack_sockaddr_in6_addrport( $testclient->peername ) ],
74              "\$socket->sockname for $socktype" );
75
76   is_deeply( [ unpack_sockaddr_in6_addrport( $socket->peername ) ],
77              [ unpack_sockaddr_in6_addrport( $testclient->sockname ) ],
78              "\$socket->peername for $socktype" );
79
80   my $peerport = ( Socket::unpack_sockaddr_in6 $socket->peername )[0];
81   my $sockport = ( Socket::unpack_sockaddr_in6 $socket->sockname )[0];
82
83   is( $testclient->sockport, $peerport, "\$testclient->sockport for $socktype" );
84   is( $testclient->peerport, $sockport, "\$testclient->peerport for $socktype" );
85
86   # Unpack just so it pretty prints without wrecking the terminal if it fails
87   is( unpack("H*", $testclient->peeraddr), $IN6ADDR_LOOPBACK_HEX, "\$testclient->peeraddr for $socktype" );
88   if( $socktype eq "SOCK_STREAM" ) {
89      # Some OSes don't update sockaddr with a local bind() on SOCK_DGRAM sockets
90      is( unpack("H*", $testclient->sockaddr), $IN6ADDR_LOOPBACK_HEX, "\$testclient->sockaddr for $socktype" );
91   }
92}
93
94done_testing;
95