t/lib/io_sock.t

   1 #!./perl
   2
   3 BEGIN {
   4     unless(grep /blib/, @INC) {
   5         chdir 't' if -d 't';
   6         unshift @INC, '../lib' if -d '../lib';
   7     }
   8 }
   9
  10 use Config;
  11
  12 BEGIN {
  13     if (-d "lib" && -f "TEST") {
  14         my $reason;
  15         if (! $Config{'d_fork'}) {
  16             $reason = 'no fork';
  17         }
  18         elsif ($Config{'extensions'} !~ /\bSocket\b/) {
  19             $reason = 'Socket extension unavailable';
  20         }
  21         elsif ($Config{'extensions'} !~ /\bIO\b/) {
  22             $reason = 'IO extension unavailable';
  23         }
  24         undef $reason if $^O eq 'VMS' and $Config{d_socket};
  25         if ($reason) {
  26             print "1..0 # Skip: $reason\n";
  27             exit 0;
  28         }
  29     }
  30 }
  31
  32 $| = 1;
  33 print "1..14\n";
  34
  35 use IO::Socket;
  36
  37 $listen = IO::Socket::INET->new(Listen => 2,
  38                                 Proto => 'tcp',
  39                                 # some systems seem to need as much as 10,
  40                                 # so be generous with the timeout
  41                                 Timeout => 15,
  42                                ) or die "$!";
  43
  44 print "ok 1\n";
  45
  46 # Check if can fork with dynamic extensions (bug in CRT):
  47 if ($^O eq 'os2' and
  48     system "$^X -I../lib -MOpcode -e 'defined fork or die'  > /dev/null 2>&1") {
  49     print "ok $_ # skipped: broken fork\n" for 2..5;
  50     exit 0;
  51 }
  52
  53 $port = $listen->sockport;
  54
  55 if($pid = fork()) {
  56
  57     $sock = $listen->accept() or die "accept failed: $!";
  58     print "ok 2\n";
  59
  60     $sock->autoflush(1);
  61     print $sock->getline();
  62
  63     print $sock "ok 4\n";
  64
  65     $sock->close;
  66
  67     waitpid($pid,0);
  68
  69     print "ok 5\n";
  70
  71 } elsif(defined $pid) {
  72
  73     # This can fail if localhost is undefined or the
  74     # special 'loopback' address 127.0.0.1 is not configured
  75     # on your system. (/etc/rc.config.d/netconfig on HP-UX.)
  76     # As a shortcut (not recommended) you could change 'localhost'
  77     # here to be the name of this machine eg 'myhost.mycompany.com'.
  78
  79     $sock = IO::Socket::INET->new(PeerPort => $port,
  80                                   Proto => 'tcp',
  81                                   PeerAddr => 'localhost'
  82                                  )
  83         or die "$! (maybe your system does not have the 'localhost' address defined)";
  84
  85     $sock->autoflush(1);
  86
  87     print $sock "ok 3\n";
  88
  89     print $sock->getline();
  90
  91     $sock->close;
  92
  93     exit;
  94 } else {
  95  die;
  96 }
  97
  98 # Test various other ways to create INET sockets that should
  99 # also work.
 100 $listen = IO::Socket::INET->new(Listen => '', Timeout => 15) or die "$!";
 101 $port = $listen->sockport;
 102
 103 if($pid = fork()) {
 104   SERVER_LOOP:
 105     while (1) {
 106        last SERVER_LOOP unless $sock = $listen->accept;
 107        while (<$sock>) {
 108            last SERVER_LOOP if /^quit/;
 109            last if /^done/;
 110            print;
 111        }
 112        $sock = undef;
 113     }
 114     $listen->close;
 115 } elsif (defined $pid) {
 116     # child, try various ways to connect
 117     $sock = IO::Socket::INET->new("localhost:$port");
 118     if ($sock) {
 119         print "not " unless $sock->connected;
 120         print "ok 6\n";
 121        $sock->print("ok 7\n");
 122        sleep(1);
 123        print "ok 8\n";
 124        $sock->print("ok 9\n");
 125        $sock->print("done\n");
 126        $sock->close;
 127     }
 128     else {
 129         print "# $@\n";
 130         print "not ok 6\n";
 131         print "not ok 7\n";
 132         print "not ok 8\n";
 133         print "not ok 9\n";
 134     }
 135
 136     # some machines seem to suffer from a race condition here
 137     sleep(2);
 138
 139     $sock = IO::Socket::INET->new("127.0.0.1:$port");
 140     if ($sock) {
 141        $sock->print("ok 10\n");
 142        $sock->print("done\n");
 143        $sock->close;
 144     }
 145     else {
 146         print "# $@\n";
 147         print "not ok 10\n";
 148     }
 149
 150     # some machines seem to suffer from a race condition here
 151 #    sleep(1);
 152
 153     $sock = IO::Socket->new(Domain => AF_INET,
 154                             PeerAddr => "localhost:$port");
 155     if ($sock) {
 156        $sock->print("ok 11\n");
 157        $sock->print("quit\n");
 158     }
 159     $sock = undef;
 160     sleep(1);
 161     exit;
 162 } else {
 163     die;
 164 }
 165
 166 # Then test UDP sockets
 167 $server = IO::Socket->new(Domain => AF_INET,
 168                           Proto  => 'udp',
 169                           LocalAddr => 'localhost');
 170 $port = $server->sockport;
 171
 172 if ($pid = fork()) {
 173     my $buf;
 174     $server->recv($buf, 100);
 175     print $buf;
 176 } elsif (defined($pid)) {
 177     #child
 178     $sock = IO::Socket::INET->new(Proto => 'udp',
 179                                   PeerAddr => "localhost:$port");
 180     $sock->send("ok 12\n");
 181     sleep(1);
 182     $sock->send("ok 12\n");  # send another one to be sure
 183     exit;
 184 } else {
 185     die;
 186 }
 187
 188 print "not " unless $server->blocking;
 189 print "ok 13\n";
 190
 191 $server->blocking(0);
 192 print "not " if $server->blocking;
 193 print "ok 14\n";