t/lib/io_sock.t

   1 #!./perl
   2
   3 BEGIN {
   4     unless(grep /blib/, @INC) {
   5         chdir 't' if -d 't';
   6         @INC = '../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..20\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     $sock = IO::Socket::INET->new(PeerPort => $port,
  74                                   Proto => 'tcp',
  75                                   PeerAddr => 'localhost'
  76                                  )
  77          || IO::Socket::INET->new(PeerPort => $port,
  78                                   Proto => 'tcp',
  79                                   PeerAddr => '127.0.0.1'
  80                                  )
  81         or die "$! (maybe your system does not have a localhost at all, 'localhost' or 127.0.0.1)";
  82
  83     $sock->autoflush(1);
  84
  85     print $sock "ok 3\n";
  86
  87     print $sock->getline();
  88
  89     $sock->close;
  90
  91     exit;
  92 } else {
  93  die;
  94 }
  95
  96 # Test various other ways to create INET sockets that should
  97 # also work.
  98 $listen = IO::Socket::INET->new(Listen => '', Timeout => 15) or die "$!";
  99 $port = $listen->sockport;
 100
 101 if($pid = fork()) {
 102   SERVER_LOOP:
 103     while (1) {
 104        last SERVER_LOOP unless $sock = $listen->accept;
 105        while (<$sock>) {
 106            last SERVER_LOOP if /^quit/;
 107            last if /^done/;
 108            print;
 109        }
 110        $sock = undef;
 111     }
 112     $listen->close;
 113 } elsif (defined $pid) {
 114     # child, try various ways to connect
 115     $sock = IO::Socket::INET->new("localhost:$port")
 116          || IO::Socket::INET->new("127.0.0.1:$port");
 117     if ($sock) {
 118         print "not " unless $sock->connected;
 119         print "ok 6\n";
 120        $sock->print("ok 7\n");
 121        sleep(1);
 122        print "ok 8\n";
 123        $sock->print("ok 9\n");
 124        $sock->print("done\n");
 125        $sock->close;
 126     }
 127     else {
 128         print "# $@\n";
 129         print "not ok 6\n";
 130         print "not ok 7\n";
 131         print "not ok 8\n";
 132         print "not ok 9\n";
 133     }
 134
 135     # some machines seem to suffer from a race condition here
 136     sleep(2);
 137
 138     $sock = IO::Socket::INET->new("127.0.0.1:$port");
 139     if ($sock) {
 140        $sock->print("ok 10\n");
 141        $sock->print("done\n");
 142        $sock->close;
 143     }
 144     else {
 145         print "# $@\n";
 146         print "not ok 10\n";
 147     }
 148
 149     # some machines seem to suffer from a race condition here
 150     sleep(1);
 151
 152     $sock = IO::Socket->new(Domain => AF_INET,
 153                             PeerAddr => "localhost:$port")
 154          || IO::Socket->new(Domain => AF_INET,
 155                             PeerAddr => "127.0.0.1:$port");
 156     if ($sock) {
 157        $sock->print("ok 11\n");
 158        $sock->print("quit\n");
 159     } else {
 160        print "not ok 11\n";
 161     }
 162     $sock = undef;
 163     sleep(1);
 164     exit;
 165 } else {
 166     die;
 167 }
 168
 169 # Then test UDP sockets
 170 $server = IO::Socket->new(Domain => AF_INET,
 171                           Proto  => 'udp',
 172                           LocalAddr => 'localhost')
 173        || IO::Socket->new(Domain => AF_INET,
 174                           Proto  => 'udp',
 175                           LocalAddr => '127.0.0.1');
 176 $port = $server->sockport;
 177
 178 if ($^O eq 'mpeix') {
 179     print("ok 12 # skipped\n")
 180 } else {
 181     if ($pid = fork()) {
 182         my $buf;
 183         $server->recv($buf, 100);
 184         print $buf;
 185     } elsif (defined($pid)) {
 186         #child
 187         $sock = IO::Socket::INET->new(Proto => 'udp',
 188                                       PeerAddr => "localhost:$port")
 189              || IO::Socket::INET->new(Proto => 'udp',
 190                                       PeerAddr => "127.0.0.1:$port");
 191         $sock->send("ok 12\n");
 192         sleep(1);
 193         $sock->send("ok 12\n");  # send another one to be sure
 194         exit;
 195     } else {
 196         die;
 197     }
 198 }
 199
 200 print "not " unless $server->blocking;
 201 print "ok 13\n";
 202
 203 $server->blocking(0);
 204 print "not " if $server->blocking;
 205 print "ok 14\n";
 206
 207 ### TEST 15
 208 ### Set up some data to be transfered between the server and
 209 ### the client. We'll use own source code ...
 210 #
 211 local @data;
 212 if( !open( SRC, "< $0")) {
 213     print "not ok 15 - $!";
 214 } else {
 215     @data = <SRC>;
 216     close( SRC);
 217 }
 218 print "ok 15\n";
 219
 220 ### TEST 16
 221 ### Start the server
 222 #
 223 my $listen = IO::Socket::INET->new( Listen => 2, Proto => 'tcp', Timeout => 15) ||
 224     print "not ";
 225 print "ok 16\n";
 226 die if( !defined( $listen));
 227 my $serverport = $listen->sockport;
 228
 229 my $server_pid = fork();
 230 if( $server_pid) {
 231
 232     ### TEST 17 Client/Server establishment
 233     #
 234     print "ok 17\n";
 235
 236     ### TEST 18
 237     ### Get data from the server using a single stream
 238     #
 239     $sock = IO::Socket::INET->new("localhost:$serverport")
 240          || IO::Socket::INET->new("127.0.0.1:$serverport");
 241
 242     if ($sock) {
 243         $sock->print("send\n");
 244
 245         my @array = ();
 246         while( <$sock>) {
 247             push( @array, $_);
 248         }
 249
 250         $sock->print("done\n");
 251         $sock->close;
 252
 253         print "not " if( @array != @data);
 254     } else {
 255         print "not ";
 256     }
 257     print "ok 18\n";
 258
 259     ### TEST 19
 260     ### Get data from the server using a stream, which is
 261     ### interrupted by eof calls.
 262     ### On perl-5.7.0@7673 this failed in a SOCKS environment, because eof
 263     ### did an getc followed by an ungetc in order to check for the streams
 264     ### end. getc(3) got replaced by the SOCKS funktion, which ended up in
 265     ### a recv(2) call on the socket, while ungetc(3) put back a character
 266     ### to an IO buffer, which never again was read.
 267     #
 268     $sock = IO::Socket::INET->new("localhost:$serverport")
 269          || IO::Socket::INET->new("127.0.0.1:$serverport");
 270
 271     if ($sock) {
 272         $sock->print("send\n");
 273
 274         my @array = ();
 275         while( !eof( $sock ) ){
 276             while( <$sock>) {
 277                 push( @array, $_);
 278                 last;
 279             }
 280         }
 281
 282         $sock->print("done\n");
 283         $sock->close;
 284
 285         print "not " if( @array != @data);
 286     } else {
 287         print "not ";
 288     }
 289     print "ok 19\n";
 290
 291     ### TEST 20
 292     ### Stop the server
 293     #
 294     $sock = IO::Socket::INET->new("localhost:$serverport")
 295          || IO::Socket::INET->new("127.0.0.1:$serverport");
 296
 297     if ($sock) {
 298         $sock->print("done\n");
 299         $sock->close;
 300
 301         print "not " if( 1 != kill 0, $server_pid);
 302     } else {
 303         print "not ";
 304     }
 305     print "ok 20\n";
 306
 307 } elsif( defined( $server_pid)) {
 308
 309     ### Child
 310     #
 311     SERVER_LOOP: while (1) {
 312         last SERVER_LOOP unless $sock = $listen->accept;
 313         while (<$sock>) {
 314             last SERVER_LOOP if /^quit/;
 315             last if /^done/;
 316             if( /^send/) {
 317                 print $sock @data;
 318                 last;
 319             }
 320             print;
 321         }
 322         $sock = undef;
 323     }
 324     $listen->close;
 325
 326 } else {
 327
 328     ### Fork failed
 329     #
 330     print "not ok 17\n";
 331     die;
 332 }
 333