Initial start of ktest.pl

Originally named autotest.pl, but renamed to ktest.pl now because
the autotest name is used by other projects.

Signed-off-by: Steven Rostedt <rostedt@goodmis.org>
This commit is contained in:
Steven Rostedt 2010-11-02 15:01:32 -04:00 committed by Steven Rostedt
parent f6614b7bb4
commit 2545eb6198

View File

@ -0,0 +1,334 @@
#!/usr/bin/perl -w
use strict;
use IPC::Open2;
use Fcntl qw(F_GETFL F_SETFL O_NONBLOCK);
use FileHandle;
$#ARGV >= 0 || die "usage: autotest.pl config-file\n";
$| = 1;
my %opt;
#default opts
$opt{"NUM_BUILDS"} = 5;
$opt{"DEFAULT_BUILD_TYPE"} = "randconfig";
$opt{"MAKE_CMD"} = "make";
$opt{"TIMEOUT"} = 50;
$opt{"TMP_DIR"} = "/tmp/autotest";
$opt{"SLEEP_TIME"} = 60; # sleep time between tests
my $version;
my $install_mods;
my $grub_number;
my $target;
my $make;
sub read_config {
my ($config) = @_;
open(IN, $config) || die "can't read file $config";
while (<IN>) {
# ignore blank lines and comments
next if (/^\s*$/ || /\s*\#/);
if (/^\s*(\S+)\s*=\s*(.*?)\s*$/) {
my $lvalue = $1;
my $rvalue = $2;
$opt{$lvalue} = $rvalue;
}
}
close(IN);
}
sub doprint {
print @_;
if (defined($opt{"LOG_FILE"})) {
open(OUT, ">> $opt{LOG_FILE}") or die "Can't write to $opt{LOG_FILE}";
print OUT @_;
close(OUT);
}
}
sub run_command {
my ($command) = @_;
my $redirect = "";
if (defined($opt{"LOG_FILE"})) {
$redirect = " >> $opt{LOG_FILE} 2>&1";
}
doprint "$command ... ";
`$command $redirect`;
my $failed = $?;
if ($failed) {
doprint "FAILED!\n";
} else {
doprint "SUCCESS\n";
}
return $failed;
}
my $timeout = $opt{"TIMEOUT"};
sub wait_for_input
{
my ($fp, $time) = @_;
my $rin;
my $ready;
my $line;
my $ch;
if (!defined($time)) {
$time = $timeout;
}
$rin = '';
vec($rin, fileno($fp), 1) = 1;
$ready = select($rin, undef, undef, $time);
$line = "";
# try to read one char at a time
while (sysread $fp, $ch, 1) {
$line .= $ch;
last if ($ch eq "\n");
}
if (!length($line)) {
return undef;
}
return $line;
}
sub reboot {
run_command "ssh $target '(echo \"savedefault --default=$grub_number --once\" | grub --batch; reboot)'";
}
sub monitor {
my $flags;
my $booted = 0;
my $bug = 0;
my $pid;
my $doopen2 = 0;
if ($doopen2) {
$pid = open2(\*IN, \*OUT, $opt{CONSOLE});
if ($pid < 0) {
die "Failed to connect to the console";
}
} else {
$pid = open(IN, "$opt{CONSOLE} |");
}
$flags = fcntl(IN, F_GETFL, 0) or
die "Can't get flags for the socket: $!\n";
$flags = fcntl(IN, F_SETFL, $flags | O_NONBLOCK) or
die "Can't set flags for the socket: $!\n";
my $line;
my $full_line = "";
doprint "Wait for monitor to settle down.\n";
# read the monitor and wait for the system to calm down
do {
$line = wait_for_input(\*IN, 5);
} while (defined($line));
reboot;
for (;;) {
$line = wait_for_input(\*IN);
last if (!defined($line));
doprint $line;
# we are not guaranteed to get a full line
$full_line .= $line;
if ($full_line =~ /login:/) {
$booted = 1;
}
if ($full_line =~ /call trace:/i) {
$bug = 1;
}
if ($line =~ /\n/) {
$full_line = "";
}
}
doprint "kill child process $pid\n";
kill 2, $pid;
print "closing!\n";
close(IN);
if (!$booted) {
die "failed - never got a boot prompt.\n";
}
if ($bug) {
die "failed - got a bug report\n";
}
}
sub install {
if (run_command "scp $opt{OUTPUT_DIR}/$opt{BUILD_TARGET} $target:$opt{TARGET_IMAGE}") {
die "failed to copy image";
}
if ($install_mods) {
my $modlib = "/lib/modules/$version";
if (run_command "ssh $target rm -rf $modlib") {
die "failed to remove old mods: $modlib";
}
if (run_command "scp -r $opt{TMP_DIR}/lib $target:/lib/modules/$version") {
die "failed to copy modules";
}
}
}
sub build {
my ($type) = @_;
unlink "$opt{OUTPUT_DIR}/.config";
run_command "$make mrproper";
# add something to distinguish this build
open(OUT, "> $opt{OUTPUT_DIR}/localversion") or die("Can't make localversion file");
print OUT "$opt{LOCALVERSION}\n";
close(OUT);
if (run_command "$make $opt{$type}") {
die "failed make config";
}
if (defined($opt{"MIN_CONFIG"})) {
run_command "cat $opt{MIN_CONFIG} >> $opt{OUTPUT_DIR}/.config";
run_command "yes '' | $make oldconfig";
}
if (run_command "$make $opt{BUILD_OPTIONS}") {
die "failed build";
}
}
read_config $ARGV[0];
# mandatory configs
die "MACHINE not defined\n" if (!defined($opt{"MACHINE"}));
die "SSH_USER not defined\n" if (!defined($opt{"SSH_USER"}));
die "BUILD_DIR not defined\n" if (!defined($opt{"BUILD_DIR"}));
die "OUTPUT_DIR not defined\n" if (!defined($opt{"OUTPUT_DIR"}));
die "BUILD_TARGET not defined\n" if (!defined($opt{"BUILD_TARGET"}));
die "POWER_CYCLE not defined\n" if (!defined($opt{"POWER_CYCLE"}));
die "CONSOLE not defined\n" if (!defined($opt{"CONSOLE"}));
die "LOCALVERSION not defined\n" if (!defined($opt{"LOCALVERSION"}));
die "GRUB_MENU not defined\n" if (!defined($opt{"GRUB_MENU"}));
chdir $opt{"BUILD_DIR"} || die "can't change directory to $opt{BUILD_DIR}";
$target = "$opt{SSH_USER}\@$opt{MACHINE}";
doprint "\n\nSTARTING AUTOMATED TESTS\n";
doprint "Find grub menu ... ";
$grub_number = -1;
open(IN, "ssh $target cat /boot/grub/menu.lst |")
or die "unable to get menu.lst";
while (<IN>) {
if (/^\s*title\s+$opt{GRUB_MENU}\s*$/) {
$grub_number++;
last;
} elsif (/^\s*title\s/) {
$grub_number++;
}
}
close(IN);
die "Could not find '$opt{GRUB_MENU}' in /boot/grub/menu on $opt{MACHINE}"
if ($grub_number < 0);
doprint "$grub_number\n";
$make = "$opt{MAKE_CMD} O=$opt{OUTPUT_DIR}";
# First we need to do is the builds
for (my $i = 1; $i <= $opt{"NUM_BUILDS"}; $i++) {
my $type = "BUILD_TYPE[$i]";
if (!defined($opt{$type})) {
$opt{$type} = $opt{"DEFAULT_BUILD_TYPE"};
}
doprint "\n\n";
doprint "RUNNING TEST $i of $opt{NUM_BUILDS} with option $opt{$type}\n\n";
if ($opt{$type} ne "nobuild") {
build $type;
}
# get the release name
doprint "$make kernelrelease ... ";
$version = `$make kernelrelease | tail -1`;
chomp($version);
doprint "$version\n";
# should we process modules?
$install_mods = 0;
open(IN, "$opt{OUTPUT_DIR}/.config") or die("Can't read config file");
while (<IN>) {
if (/CONFIG_MODULES(=y)?/) {
$install_mods = 1 if (defined($1));
last;
}
}
close(IN);
if ($install_mods) {
if (run_command "$make INSTALL_MOD_PATH=$opt{TMP_DIR} modules_install") {
die "Failed to install modules";
}
} else {
doprint "No modules needed\n";
}
install;
monitor;
doprint "\n\n*******************************************\n";
doprint "*******************************************\n";
doprint "** SUCCESS!!!! **\n";
doprint "*******************************************\n";
doprint "*******************************************\n";
# try to reboot normally
if (run_command "ssh $target reboot") {
# nope? power cycle it.
run_command "$opt{POWER_CYCLE}";
}
sleep "$opt{SLEEP_TIME}";
}
exit 0;