use strict;
use warnings;
-use POSIX qw/SIGINT SIGTERM SIGKILL EXIT_SUCCESS EXIT_FAILURE WIFEXITED WEXITSTATUS/;
+use POSIX qw/SIGINT SIGTERM SIGKILL SIGHUP EXIT_FAILURE/;
use lib qw{blib/lib};
-use IPC::MorseSignals qw/msend mrecv/;
+use IPC::MorseSignals qw/msend mrecv mreset/;
+
+my $lives = 100;
+
+sub spawn {
+ --$lives;
+ die 'forked too many times' if $lives < 0;
+ pipe my $rdr, my $wtr or die "pipe() failed: $!";
+ my $pid = fork;
+ if (!defined $pid) {
+ die "fork() failed: $!";
+ } elsif ($pid == 0) {
+ close $rdr or die "close() failed: $!";
+ my $s = mrecv local %SIG, cb => sub {
+ select $wtr; $| = 1;
+ print $wtr $_[1], "\n";
+ select $wtr; $| = 1;
+ };
+ $SIG{'HUP'} = sub { mreset $s };
+ 1 while 1;
+ exit EXIT_FAILURE;
+ }
+ close $wtr or die "close() failed: $!";
+ return ($pid, $rdr);
+}
+
+sub slaughter {
+ my ($pid) = @_;
+ kill SIGINT => $pid;
+ kill SIGTERM => $pid;
+ kill SIGKILL => $pid;
+ waitpid $pid, 0;
+}
my @res;
-sub tryspeed {
+my ($pid, $rdr) = spawn;
+
+sub tryspeed {
my ($l, $n) = @_;
my $speed = 2 ** 16;
my $ok = 0;
- my $desc;
-SPEED:
- while (($speed > 1) && ($ok < $n)) {
- $speed /= 2;
- $desc = "$n sends of $l bytes at $speed bits/s";
- $ok = 0;
- print STDERR "try $desc";
+ my @alpha = ('a' .. 'z');
+ my $msg = join '', map { $alpha[rand @alpha] } 1 .. $l;
+ while (($ok < $n) && (($speed /= 2) >= 1)) {
+ print STDERR "$n sends of $l bytes at $speed bits/s";
+TRY:
for (1 .. $n) {
- print STDERR ".";
- my @alpha = ('a' .. 'z');
- my $msg = join '', map { $alpha[rand @alpha] } 1 .. $l;
- my $pid = fork;
- if (!defined $pid) {
- die "$desc: fork() failed : $!";
- } elsif ($pid == 0) {
- local @SIG{qw/USR1 USR2/} = mrecv sub {
- exit(($msg eq $_[0]) ? EXIT_SUCCESS : EXIT_FAILURE);
- };
- 1 while 1;
- exit EXIT_FAILURE;
- }
- my $next = 0;
+ print STDERR '.';
+ my $r = '';
eval {
- local $SIG{ALRM} = sub { die 'timeout' };
+ local $SIG{ALRM} = sub { print STDERR "timeout\n"; die };
my $a = (int(100 * (3 * $l) / $speed) || 1);
$a = 10 if $a > 10;
alarm $a;
msend $msg => $pid, speed => $speed;
- waitpid $pid, 0;
- if (WIFEXITED($?) && (WEXITSTATUS($?) == EXIT_SUCCESS)) {
+ $r = <$rdr>;
+ };
+ kill SIGHUP => $pid if $@;
+ alarm 0;
+ if (!defined $r) { # Something bad happened, respawn
+ print STDERR "oops\n";
+ close $rdr or die "close() failed: $!";
+ slaughter $pid;
+ ($pid, $rdr) = spawn;
+ redo TRY; # Retry this send
+ } else {
+ chomp $r;
+ if ($r eq $msg) {
++$ok;
} else {
- print STDERR " transfer error\n";
- $next = 1;
+ print STDERR "transfer error\n";
+ kill SIGHUP => $pid;
+ last TRY;
}
- };
- alarm 0;
- if ($@) {
- kill SIGINT, $pid;
- kill SIGTERM, $pid;
- kill SIGKILL, $pid;
- print STDERR " timeout\n";
- $next = 1;
}
- next SPEED if $next;
}
}
- $desc = "$l bytes sent $n times";
+ my $desc = "$l bytes sent $n times";
if ($speed >= 1) {
print STDERR " OK\n\n";
push @res, "$desc at $speed bits/s";
}
}
-tryspeed 4, 1;
-tryspeed 4, 4;
-tryspeed 4, 16;
-tryspeed 4, 64;
-tryspeed 4, 256;
-tryspeed 16, 1;
-tryspeed 16, 4;
-tryspeed 16, 16;
-tryspeed 16, 64;
-tryspeed 64, 1;
-tryspeed 64, 4;
-tryspeed 64, 16;
-tryspeed 256, 1;
-tryspeed 256, 4;
+tryspeed 4, 1;
+tryspeed 4, 4;
+tryspeed 4, 16;
+tryspeed 4, 64;
+tryspeed 4, 256;
+tryspeed 16, 1;
+tryspeed 16, 4;
+tryspeed 16, 16;
+tryspeed 16, 64;
+tryspeed 64, 1;
+tryspeed 64, 4;
+tryspeed 64, 16;
+tryspeed 256, 1;
+tryspeed 256, 4;
tryspeed 1024, 1;
print STDERR "=== Summary ===\n";