#!./perl -w

use strict;

BEGIN {

    unless(grep /blib/, @INC) {

	chdir 't' if -d 't';

	@INC = '../lib';

    }

}

use Config;

BEGIN {

    if (-d "lib" && -f "TEST") {

	my $reason;

	if (! $Config{'d_fork'}) {

	    $reason = 'no fork';

	}

	elsif ($Config{'extensions'} !~ /\bSocket\b/) {

	    $reason = 'Socket extension unavailable';

	}

	elsif ($Config{'extensions'} !~ /\bIO\b/) {

	    $reason = 'IO extension unavailable';

	}

	if ($reason) {

	    print "1..0 # Skip: $reason\n";

	    exit 0;

        }

    }

    if ($^O eq 'MSWin32') {

        print "1..0 # Skip: accept() fails for IPv6 under MSWin32\n";

        exit 0;

    }

}

$| = 1;

print "1..20\n";

eval {

    $SIG{ALRM} = sub { die; };

    alarm 120;

};

use IO::Socket::INET6;

my $listen = IO::Socket::INET6->new(Listen => 2,

				Proto => 'tcp',

				# some systems seem to need as much as 10,

				# so be generous with the timeout

				Timeout => 15,

			       ) or die "$@";

print "ok 1\n";

# Check if can fork with dynamic extensions (bug in CRT):

if ($^O eq 'os2' and

    system "$^X -I../lib -MOpcode -e 'defined fork or die'  > /dev/null 2>&1") {

    print "ok $_ # skipped: broken fork\n" for 2..5;

    exit 0;

}

my $port = $listen->sockport;

my ($pid, $sock);

if($pid = fork()) {

    $sock = $listen->accept() or die "accept failed: $!";

    print "ok 2\n";

    $sock->autoflush(1);

    print $sock->getline();

    print $sock "ok 4\n";

    $sock->close;

    waitpid($pid,0);

    print "ok 5\n";

} elsif(defined $pid) {

    $sock = IO::Socket::INET6->new(PeerPort => $port,

				  Proto => 'tcp',

				  PeerAddr => 'localhost'

				 )

         || IO::Socket::INET6->new(PeerPort => $port,

				  Proto => 'tcp',

				  PeerAddr => '::1'

				 )

	or die "$! (maybe your system does not have a localhost at all, 'localhost' or ::1)";

    $sock->autoflush(1);

    print $sock "ok 3\n";

    print $sock->getline();

    $sock->close;

    exit;

} else {

 die;

}

# Test various other ways to create INET sockets that should

# also work.

$listen = IO::Socket::INET6->new(Listen => '', Timeout => 15) or die "$@";

$port = $listen->sockport;

if($pid = fork()) {

  SERVER_LOOP:

    while (1) {

       last SERVER_LOOP unless $sock = $listen->accept;

       while (<$sock>) {

           last SERVER_LOOP if /^quit/;

           last if /^done/;

           print;

       }

       $sock = undef;

    }

    $listen->close;

} elsif (defined $pid) {

    # child, try various ways to connect

    $sock = IO::Socket::INET6->new("localhost:$port")

         || IO::Socket::INET6->new("[::1]:$port");

    if ($sock) {

	print "not " unless $sock->connected;

	print "ok 6\n";

       $sock->print("ok 7\n");

       sleep(1);

       print "ok 8\n";

       $sock->print("ok 9\n");

       $sock->print("done\n");

       $sock->close;

    }

    else {

	print "# $@\n";

	print "not ok 6\n";

	print "not ok 7\n";

	print "not ok 8\n";

	print "not ok 9\n";

    }

    # some machines seem to suffer from a race condition here

    sleep(2);

    $sock = IO::Socket::INET6->new("[::1]:$port");

    if ($sock) {

       $sock->print("ok 10\n");

       $sock->print("done\n");

       $sock->close;

    }

    else {

	print "# $@\n";

	print "not ok 10\n";

    }

    # some machines seem to suffer from a race condition here

    sleep(1);

    $sock = IO::Socket->new(Domain => AF_INET6,

                            PeerAddr => "localhost:$port")

         || IO::Socket->new(Domain => AF_INET6,

                            PeerAddr => "[::1]:$port");

    if ($sock) {

       $sock->print("ok 11\n");

       $sock->print("quit\n");

    } else {

       print "not ok 11\n";

    }

    $sock = undef;

    sleep(1);

    exit;

} else {

    die;

}

# Then test UDP sockets

# Test the numeric address directly because "localhost" usually maps

# to an IPv4 address.

my $server = IO::Socket->new(

			Domain => AF_INET6,

			 Proto  => 'udp',

                          LocalAddr => '::1');

$port = $server->sockport;

if ($^O eq 'mpeix') {

    print("ok 12 # skipped\n")

} else {

    if ($pid = fork()) {

        my $buf;

        $server->recv($buf, 100);

        print $buf;

    } elsif (defined($pid)) {

        #child

        # Test the numeric address directly because "localhost" usually maps

        # to an IPv4 address.

        $sock = IO::Socket::INET6->new(Proto => 'udp',

					Domain => AF_INET6,

                                     	PeerAddr => "[::1]:$port");

        $sock->send("ok 12\n");

        sleep(1);

        $sock->send("ok 12\n");

        exit;

    } else {

        die;

    }

}

print "not " unless $server->blocking;

print "ok 13\n";

if ( $^O eq 'qnx' ) {

  # QNX library bug: Can set non-blocking on socket, but

  # cannot return that status.

  print "ok 14 # skipped\n";

} else {

  $server->blocking(0);

  print "not " if $server->blocking;

  print "ok 14\n";

}

### TEST 15

### Set up some data to be transfered between the server and

### the client. We'll use own source code ...

#

my @data;

if( !open( SRC, "< $0")) {

    print "not ok 15 - $!";

} else {

    @data = <SRC>;

    close( SRC);

}

print "ok 15\n";

### TEST 16

### Start the server

#

$listen = IO::Socket::INET6->new( Listen => 2, Proto => 'tcp', Timeout => 15) ||

    print "not ";

print "ok 16\n";

die if( !defined( $listen));

my $serverport = $listen->sockport;

my $server_pid = fork();

if( $server_pid) {

    ### TEST 17 Client/Server establishment

    #

    print "ok 17\n";

    ### TEST 18

    ### Get data from the server using a single stream

    #

    $sock = IO::Socket::INET6->new("localhost:$serverport")

         || IO::Socket::INET6->new("[::1]:$serverport");

    if ($sock) {

	$sock->print("send\n");

	my @array = ();

	while( <$sock>) {

	    push( @array, $_);

	}

	$sock->print("done\n");

	$sock->close;

	print "not " if( @array != @data);

    } else {

	print "not ";

    }

    print "ok 18\n";

    ### TEST 19

    ### Get data from the server using a stream, which is

    ### interrupted by eof calls.

    ### On perl-5.7.0@7673 this failed in a SOCKS environment, because eof

    ### did an getc followed by an ungetc in order to check for the streams

    ### end. getc(3) got replaced by the SOCKS funktion, which ended up in

    ### a recv(2) call on the socket, while ungetc(3) put back a character

    ### to an IO buffer, which never again was read.

    #

    if ($^O eq 'mpeix') {

	print "ok 19 # skipped: broken on MPE/iX\n";

    } else {

    $sock = IO::Socket::INET6->new("localhost:$serverport")

         || IO::Socket::INET6->new("[::1]:$serverport");

    if ($sock) {

	$sock->print("send\n");

	my @array = ();

	while( !eof( $sock ) ){

	    while( <$sock>) {

		push( @array, $_);

		last;

	    }

	}

	$sock->print("done\n");

	$sock->close;

	print "not " if( @array != @data);

    } else {

	print "not ";

    }

    print "ok 19\n";

    }

    ### TEST 20

    ### Stop the server

    #

    $sock = IO::Socket::INET6->new("localhost:$serverport")

         || IO::Socket::INET6->new("[::1]:$serverport");

    if ($sock) {

	$sock->print("done\n");

	$sock->close;

	print "not " if( 1 != kill 0, $server_pid);

    } else {

	print "not ";

    }

    print "ok 20\n";

} elsif( defined( $server_pid)) {

    ### Child

    #

    SERVER_LOOP: while (1) {

	last SERVER_LOOP unless $sock = $listen->accept;

	while (<$sock>) {

	    last SERVER_LOOP if /^quit/;

	    last if /^done/;

	    if( /^send/) {

		print $sock @data;

		last;

	    }

	    print;

	}

	$sock = undef;

    }

    $listen->close;

} else {

    ### Fork failed

    #

    print "not ok 17\n";

    die;

}