blob: 79da57f3023b734d19e4cd358587db0469241293 [file] [log] [blame]
#!/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
$opt{"BUILD_NOCLEAN"} = 0;
$opt{"REBOOT_ON_ERROR"} = 0;
$opt{"POWEROFF_ON_ERROR"} = 0;
$opt{"POWEROFF_ON_SUCCESS"} = 0;
$opt{"BUILD_OPTIONS"} = "";
my $version;
my $install_mods;
my $grub_number;
my $target;
my $make;
my $noclean;
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 dodie {
doprint "CRITICAL FAILURE... ", @_;
if ($opt{"REBOOT_ON_ERROR"}) {
doprint "REBOOTING\n";
`$opt{"POWER_CYCLE"}`;
} elsif ($opt{"POWEROFF_ON_ERROR"} && defined($opt{"POWER_OFF"})) {
doprint "POWERING OFF\n";
`$opt{"POWER_OFF"}`;
}
die @_;
}
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_to {
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;
my $skip_call_trace = 0;
if ($doopen2) {
$pid = open2(\*IN, \*OUT, $opt{"CONSOLE"});
if ($pid < 0) {
dodie "Failed to connect to the console";
}
} else {
$pid = open(IN, "$opt{CONSOLE} |");
}
$flags = fcntl(IN, F_GETFL, 0) or
dodie "Can't get flags for the socket: $!\n";
$flags = fcntl(IN, F_SETFL, $flags | O_NONBLOCK) or
dodie "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_to;
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 =~ /\[ backtrace testing \]/) {
$skip_call_trace = 1;
}
if ($full_line =~ /call trace:/i) {
$bug = 1 if (!$skip_call_trace);
}
if ($full_line =~ /\[ end of backtrace testing \]/) {
$skip_call_trace = 0;
}
if ($full_line =~ /Kernel panic -/) {
$bug = 1;
}
if ($line =~ /\n/) {
$full_line = "";
}
}
doprint "kill child process $pid\n";
kill 2, $pid;
print "closing!\n";
close(IN);
if (!$booted) {
dodie "failed - never got a boot prompt.\n";
}
if ($bug) {
dodie "failed - got a bug report\n";
}
}
sub install {
if (run_command "scp $opt{OUTPUT_DIR}/$opt{BUILD_TARGET} $target:$opt{TARGET_IMAGE}") {
dodie "failed to copy image";
}
if ($install_mods) {
my $modlib = "/lib/modules/$version";
my $modtar = "autotest-mods.tar.bz2";
if (run_command "ssh $target rm -rf $modlib") {
dodie "failed to remove old mods: $modlib";
}
# would be nice if scp -r did not follow symbolic links
if (run_command "cd $opt{TMP_DIR} && tar -cjf $modtar lib/modules/$version") {
dodie "making tarball";
}
if (run_command "scp $opt{TMP_DIR}/$modtar $target:/tmp") {
dodie "failed to copy modules";
}
unlink "$opt{TMP_DIR}/$modtar";
if (run_command "ssh $target '(cd / && tar xf /tmp/$modtar)'") {
dodie "failed to tar modules";
}
run_command "ssh $target rm -f /tmp/$modtar";
}
}
sub build {
my ($type) = @_;
my $defconfig = "";
my $append = "";
if ($type =~ /^useconfig:(.*)/) {
if (run_command "cp $1 $opt{OUTPUT_DIR}/.config") {
dodie "could not copy $1 to .config";
}
$type = "oldconfig";
}
# old config can ask questions
if ($type eq "oldconfig") {
$append = "yes ''|";
# allow for empty configs
run_command "touch $opt{OUTPUT_DIR}/.config";
if (run_command "mv $opt{OUTPUT_DIR}/.config $opt{OUTPUT_DIR}/config_temp") {
dodie "moving .config";
}
if (!$noclean && run_command "$make mrproper") {
dodie "make mrproper";
}
if (run_command "mv $opt{OUTPUT_DIR}/config_temp $opt{OUTPUT_DIR}/.config") {
dodie "moving config_temp";
}
} elsif (!$noclean) {
unlink "$opt{OUTPUT_DIR}/.config";
if (run_command "$make mrproper") {
dodie "make mrproper";
}
}
# add something to distinguish this build
open(OUT, "> $opt{OUTPUT_DIR}/localversion") or dodie("Can't make localversion file");
print OUT "$opt{LOCALVERSION}\n";
close(OUT);
if (defined($opt{"MIN_CONFIG"})) {
$defconfig = "KCONFIG_ALLCONFIG=$opt{MIN_CONFIG}";
}
if (run_command "$defconfig $append $make $type") {
dodie "failed make config";
}
if (run_command "$make $opt{BUILD_OPTIONS}") {
dodie "failed build";
}
}
sub reboot {
# try to reboot normally
if (run_command "ssh $target reboot") {
# nope? power cycle it.
run_command "$opt{POWER_CYCLE}";
}
}
sub halt {
if ((run_command "ssh $target halt") or defined($opt{"POWER_OFF"})) {
# nope? the zap it!
run_command "$opt{POWER_OFF}";
}
}
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 "TARGET_IMAGE not defined\n" if (!defined($opt{"TARGET_IMAGE"}));
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{"BUILD_NOCLEAN[$i]"}) &&
$opt{"BUILD_NOCLEAN[$i]"} != 0) {
$noclean = 1;
} else {
$noclean = $opt{"BUILD_NOCLEAN"};
}
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 $opt{$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 dodie("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") {
dodie "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";
if ($i != $opt{"NUM_BUILDS"}) {
reboot;
sleep "$opt{SLEEP_TIME}";
}
}
if ($opt{"POWEROFF_ON_SUCCESS"}) {
halt;
} else {
reboot;
}
exit 0;