use strict;
use Cwd 'abs_path';
my $usage = "usage: $0 [--use-rndc] test-directory [server-directory]";
my $use_rndc;
while (@ARGV && $ARGV[0] =~ /^-/) {
my $opt = shift @ARGV;
if ($opt eq '--use-rndc') {
$use_rndc = 1;
} else {
die "$usage\n";
}
}
my $test = $ARGV[0];
my $server = $ARGV[1];
my $errors = 0;
die "$usage\n" unless defined($test);
die "No test directory: \"$test\"\n" unless (-d $test);
die "No server directory: \"$server\"\n" if (defined($server) && !-d "$test/$server");
my $testdir = abs_path($test);
my @servers;
if (defined $server) {
@servers = ($server);
} else {
local *DIR;
opendir DIR, $testdir or die "$testdir: $!\n";
my @files = sort readdir DIR;
closedir DIR;
my @ns = grep /^ns[0-9]*$/, @files;
my @lwresd = grep /^lwresd[0-9]*$/, @files;
my @ans = grep /^ans[0-9]*$/, @files;
push @servers, @ns, @lwresd, @ans;
}
if ($use_rndc) {
foreach my $server (grep /^ns/, @servers) {
stop_rndc($server);
}
wait_for_servers(30, grep /^ns/, @servers);
}
foreach my $server (@servers) {
stop_signal($server, "TERM");
}
wait_for_servers(60, @servers);
foreach my $server (@servers) {
stop_signal($server, "ABRT");
}
exit($errors ? 1 : 0);
sub server_pid_file {
my($server) = @_;
my $pid_file;
if ($server =~ /^ns/) {
$pid_file = "named.pid";
} elsif ($server =~ /^lwresd/) {
$pid_file = "lwresd.pid";
} elsif ($server =~ /^ans/) {
$pid_file = "ans.pid";
} else {
print "I:Unknown server type $server\n";
exit 1;
}
$pid_file = "$testdir/$server/$pid_file";
}
sub read_pid {
my($pid_file) = @_;
local *FH;
my $result = open FH, "< $pid_file";
if (!$result) {
print "I:$pid_file: $!\n";
unlink $pid_file;
return;
}
my $pid = <FH>;
chomp($pid);
return $pid;
}
sub stop_rndc {
my($server) = @_;
return unless ($server =~ /^ns(\d+)$/);
my $ip = "10.53.0.$1";
system("$ENV{RNDC} -c $testdir/../common/rndc.conf -s $ip -p 9953 stop | sed 's/^/I:$server /'");
return;
}
sub stop_signal {
my($server, $sig) = @_;
my $pid_file = server_pid_file($server);
return unless -f $pid_file;
my $pid = read_pid($pid_file);
return unless defined($pid);
if ($sig eq 'ABRT') {
print "I:$server didn't die when sent a SIGTERM\n";
$errors++;
}
my $result = kill $sig, $pid;
if (!$result) {
print "I:$server died before a SIG$sig was sent\n";
unlink $pid_file;
$errors++;
}
return;
}
sub wait_for_servers {
my($timeout, @servers) = @_;
my @pid_files = grep { defined($_) }
map { server_pid_file($_) } @servers;
while ($timeout > 0 && @pid_files > 0) {
@pid_files = grep { -f $_ } @pid_files;
sleep 1 if (@pid_files > 0);
$timeout--;
}
return;
}