Blame tests/run-regression-tests.pl.in

Packit 284210
#!@PERL@
Packit 284210
#
Packit 284210
# Run regression tests.
Packit 284210
#
Packit 284210
# Syntax: run-regression-tests.pl [options] [file [N]]
Packit 284210
#
Packit 284210
#          All: run-regression-tests.pl
Packit 284210
#   All in file: run-regression-tests.pl file
Packit 284210
#   Nth in file: run-regression-tests.pl file N
Packit 284210
#
Packit 284210
use strict;
Packit 284210
use Time::HiRes qw(gettimeofday sleep);
Packit 284210
use POSIX qw(WIFEXITED WEXITSTATUS WIFSIGNALED WTERMSIG);
Packit 284210
use File::Spec qw(rel2abs);
Packit 284210
use File::Basename qw(basename dirname);
Packit 284210
use FileHandle;
Packit 284210
use IPC::Open2 qw(open2);
Packit 284210
use IPC::Open3 qw(open3);
Packit 284210
use Getopt::Std;
Packit 284210
use Data::Dumper;
Packit 284210
use IO::Socket;
Packit 284210
use LWP::UserAgent;
Packit 284210
Packit 284210
my @TYPES = qw(config misc action target rule);
Packit 284210
my $SCRIPT = basename($0);
Packit 284210
my $SCRIPT_DIR = File::Spec->rel2abs(dirname($0));
Packit 284210
my $REG_DIR = "$SCRIPT_DIR/regression";
Packit 284210
my $SROOT_DIR = "$REG_DIR/server_root";
Packit 284210
my $DATA_DIR = "$SROOT_DIR/data";
Packit 284210
my $TEMP_DIR = "$SROOT_DIR/tmp";
Packit 284210
my $UPLOAD_DIR = "$SROOT_DIR/upload";
Packit 284210
my $CONF_DIR = "$SROOT_DIR/conf";
Packit 284210
my $MODULES_DIR = q(@APXS_LIBEXECDIR@);
Packit 284210
my $FILES_DIR = "$SROOT_DIR/logs";
Packit 284210
my $PID_FILE = "$FILES_DIR/httpd.pid";
Packit 284210
my $HTTPD = q(@APXS_HTTPD@);
Packit 284210
my $PASSED = 0;
Packit 284210
my $TOTAL = 0;
Packit 284210
my $BUFSIZ = 32768;
Packit 284210
my %C = ();
Packit 284210
my %FILE = ();
Packit 284210
my $UA_NAME = "ModSecurity Regression Tests/1.2.3";
Packit 284210
my $UA = LWP::UserAgent->new;
Packit 284210
$UA->agent($UA_NAME);
Packit 284210
Packit 284210
# Hack for testing the script w/o configure
Packit 284210
if ($HTTPD eq "\@APXS_HTTPD\@") {
Packit 284210
    $HTTPD = "/usr/local/apache2/bin/httpd";
Packit 284210
    $MODULES_DIR = "/usr/local/apache2/modules";
Packit 284210
}
Packit 284210
Packit 284210
$SIG{TERM} = $SIG{INT} = \&handle_interrupt;
Packit 284210
Packit 284210
my $platform = "apache";
Packit 284210
Packit 284210
my %opt;
Packit 284210
getopts('A:E:D:C:T:H:a:p:dvh', \%opt);
Packit 284210
Packit 284210
if ($opt{d}) {
Packit 284210
    $Data::Dumper::Indent = 1;
Packit 284210
    $Data::Dumper::Terse = 1;
Packit 284210
    $Data::Dumper::Pad = "";
Packit 284210
    $Data::Dumper::Quotekeys = 0;
Packit 284210
}
Packit 284210
Packit 284210
sub usage {
Packit 284210
    print stderr <<"EOT";
Packit 284210
@_
Packit 284210
Usage: $SCRIPT [options] [file [N]]
Packit 284210
Packit 284210
 Options:
Packit 284210
  -A file   Specify ModSecurity audit log to read.
Packit 284210
  -D file   Specify ModSecurity debug log to read.
Packit 284210
  -E file   Specify Apache httpd error log to read.
Packit 284210
  -C file   Specify Apache httpd base conf file to generate/reload.
Packit 284210
  -H path   Specify Apache httpd htdocs path.
Packit 284210
  -S path   Specify Apache httpd server root path.
Packit 284210
  -a file   Specify Apache httpd binary (default: httpd)
Packit 284210
  -p port   Specify Apache httpd port (default: 8088)
Packit 284210
  -v        Enable verbose output (details on failure).
Packit 284210
  -d        Enable debugging output.
Packit 284210
  -h        This help.
Packit 284210
Packit 284210
EOT
Packit 284210
Packit 284210
    exit(1);
Packit 284210
}
Packit 284210
Packit 284210
usage() if ($opt{h});
Packit 284210
Packit 284210
### Check httpd binary
Packit 284210
if (defined $opt{a}) {
Packit 284210
    $HTTPD = $opt{a};
Packit 284210
}
Packit 284210
else {
Packit 284210
    $opt{a} = $HTTPD;
Packit 284210
}
Packit 284210
usage("Invalid Apache startup script: $HTTPD\n") unless (-e $HTTPD);
Packit 284210
Packit 284210
### Defaults
Packit 284210
$opt{A} = "$FILES_DIR/modsec_audit.log" unless (defined $opt{A});
Packit 284210
$opt{D} = "$FILES_DIR/modsec_debug.log" unless (defined $opt{D});
Packit 284210
$opt{E} = "$FILES_DIR/error.log" unless (defined $opt{E});
Packit 284210
$opt{C} = "$CONF_DIR/httpd.conf" unless (defined $opt{C});
Packit 284210
$opt{H} = "$SROOT_DIR/htdocs" unless (defined $opt{H});
Packit 284210
$opt{p} = 8088 unless (defined $opt{p});
Packit 284210
$opt{v} = 1 if ($opt{d});
Packit 284210
Packit 284210
unless (defined $opt{S}) {
Packit 284210
    my $httpd_root = `$HTTPD -V`;
Packit 284210
    ($opt{S} = $httpd_root) =~ s/.*-D HTTPD_ROOT="([^"]*)".*/$1/sm;
Packit 284210
}
Packit 284210
Packit 284210
%ENV = (
Packit 284210
    %ENV,
Packit 284210
    SERVER_ROOT => $opt{S},
Packit 284210
    SERVER_PORT => $opt{p},
Packit 284210
    SERVER_NAME => "localhost",
Packit 284210
    TEST_SERVER_ROOT => $SROOT_DIR,
Packit 284210
    DATA_DIR => $DATA_DIR,
Packit 284210
    TEMP_DIR => $TEMP_DIR,
Packit 284210
    UPLOAD_DIR => $UPLOAD_DIR,
Packit 284210
    CONF_DIR => $CONF_DIR,
Packit 284210
    MODULES_DIR => $MODULES_DIR,
Packit 284210
    LOGS_DIR => $FILES_DIR,
Packit 284210
    SCRIPT_DIR => $SCRIPT_DIR,
Packit 284210
    REGRESSION_DIR => $REG_DIR,
Packit 284210
    DIST_ROOT => File::Spec->rel2abs(dirname("$SCRIPT_DIR/../../..")),
Packit 284210
    AUDIT_LOG => $opt{A},
Packit 284210
    DEBUG_LOG => $opt{D},
Packit 284210
    ERROR_LOG => $opt{E},
Packit 284210
    HTTPD_CONF => $opt{C},
Packit 284210
    HTDOCS => $opt{H},
Packit 284210
    USER_AGENT => $UA_NAME,
Packit 284210
);
Packit 284210
Packit 284210
#dbg("OPTIONS: ", \%opt);
Packit 284210
Packit 284210
if (-e "$PID_FILE") {
Packit 284210
    msg("Shutting down previous instance: $PID_FILE");
Packit 284210
    httpd_stop();
Packit 284210
}
Packit 284210
Packit 284210
if (defined $ARGV[0]) {
Packit 284210
    runfile(dirname($ARGV[0]), basename($ARGV[0]), $ARGV[1]);
Packit 284210
    done();
Packit 284210
}
Packit 284210
Packit 284210
for my $type (@TYPES) {
Packit 284210
    my $dir = "$SCRIPT_DIR/regression/$type";
Packit 284210
    my @cfg = ();
Packit 284210
Packit 284210
    # Get test names
Packit 284210
    opendir(DIR, "$dir") or quit(1, "Failed to open \"$dir\": $!");
Packit 284210
    @cfg = grep { /\.t$/ && -f "$dir/$_" } readdir(DIR);
Packit 284210
    closedir(DIR);
Packit 284210
Packit 284210
    for my $cfg (sort @cfg) {
Packit 284210
        runfile($dir, $cfg);
Packit 284210
    }
Packit 284210
}
Packit 284210
done();
Packit 284210
Packit 284210
Packit 284210
sub runfile {
Packit 284210
    my($dir, $cfg, $testnum) = @_;
Packit 284210
    my $fn = "$dir/$cfg";
Packit 284210
    my @data = ();
Packit 284210
    my $edata;
Packit 284210
    my @C = ();
Packit 284210
    my @test = ();
Packit 284210
    my $teststr;
Packit 284210
    my $n = 0;
Packit 284210
    my $pass = 0;
Packit 284210
Packit 284210
    open(CFG, "<$fn") or quit(1, "Failed to open \"$fn\": $!");
Packit 284210
    @data = <CFG>;
Packit 284210
  
Packit 284210
    $edata = q/@C = (/ . join("", @data) . q/)/;
Packit 284210
    eval $edata;
Packit 284210
    quit(1, "Failed to read test data \"$cfg\": $@") if ($@);
Packit 284210
Packit 284210
    unless (@C) {
Packit 284210
        msg("\nNo tests defined for $fn");
Packit 284210
        return;
Packit 284210
    }
Packit 284210
Packit 284210
    msg("\nLoaded ".@C." tests from $fn");
Packit 284210
    for my $t (@C) {
Packit 284210
        $n++;
Packit 284210
        next if (defined $testnum and $n != $testnum);
Packit 284210
Packit 284210
        my $httpd_up = 0;
Packit 284210
        my %t = %{$t || {}};
Packit 284210
        my $id = sprintf("%3d", $n);
Packit 284210
        my $out = "";
Packit 284210
        my $rc = 0;
Packit 284210
        my $conf_fn;
Packit 284210
Packit 284210
        # Startup httpd with optionally included conf.
Packit 284210
        if (exists $t{conf} and defined $t{conf}) {
Packit 284210
            $conf_fn = sprintf "%s/%s_%s_%06d.conf",
Packit 284210
                         $CONF_DIR, $t{type}, $cfg, $n;
Packit 284210
            #dbg("Writing test config to: $conf_fn");
Packit 284210
            open(CONF, ">$conf_fn") or die "Failed to open conf \"$conf_fn\": $!\n";
Packit 284210
            print CONF (ref $t{conf} eq "CODE" ? eval { &{$t{conf}} } : $t{conf});
Packit 284210
            msg("$@") if ($@);
Packit 284210
            close CONF;
Packit 284210
            $httpd_up = httpd_start(\%t, "Include $conf_fn") ? 0 : 1;
Packit 284210
        }
Packit 284210
        else {
Packit 284210
            $httpd_up = httpd_start(\%t) ? 0 : 1;
Packit 284210
        }
Packit 284210
Packit 284210
        # Run any prerun setup
Packit 284210
        if ($rc == 0 and exists $t{prerun} and defined $t{prerun}) {
Packit 284210
            vrb("Executing perl prerun...");
Packit 284210
            $rc = &{$t{prerun}};
Packit 284210
            vrb("Perl prerun returned: $rc");
Packit 284210
        }
Packit 284210
Packit 284210
        if ($httpd_up) {
Packit 284210
            # Perform the request and check response
Packit 284210
            if (exists $t{request}) {
Packit 284210
                my $resp = do_request($t{request});
Packit 284210
                if (!$resp) {
Packit 284210
                    msg("invalid response");
Packit 284210
                    vrb("RESPONSE: ", $resp);
Packit 284210
                    $rc = 1;
Packit 284210
                }
Packit 284210
                else {
Packit 284210
                    for my $key (keys %{ $t{match_response} || {}}) {
Packit 284210
                        my($neg,$mtype) = ($key =~ m/^(-?)(.*)$/);
Packit 284210
                        my $m = $t{match_response}{$key};
Packit 284210
                        if (ref($m) eq "HASH") {
Packit 284210
                            if ($m->{$platform}) {
Packit 284210
                                $m = $m->{$platform};
Packit 284210
                            }
Packit 284210
                            else {
Packit 284210
                                my $ap = join(", ", keys %{$m});
Packit 284210
                                msg("Warning: trying to match: $mtype. Nothing " .
Packit 284210
                                        "to match in current platform: $platform. " .
Packit 284210
                                        "This test only contains cotent for: $ap.");
Packit 284210
                                last;
Packit 284210
                            }
Packit 284210
                        }
Packit 284210
                        my $match = match_response($mtype, $resp, $m);
Packit 284210
                        if ($neg and defined $match) {
Packit 284210
                            $rc = 1;
Packit 284210
                            msg("response $mtype matched: $m");
Packit 284210
                            vrb($resp);
Packit 284210
                            last;
Packit 284210
                        }
Packit 284210
                        elsif (!$neg and !defined $match) {
Packit 284210
                            $rc = 1;
Packit 284210
                            msg("response $mtype failed to match: $m");
Packit 284210
                            vrb($resp);
Packit 284210
                            last;
Packit 284210
                        }
Packit 284210
                    }
Packit 284210
                }
Packit 284210
            }
Packit 284210
Packit 284210
            # Run any arbitrary perl tests
Packit 284210
            if ($rc == 0 and exists $t{test} and defined $t{test}) {
Packit 284210
                dbg("Executing perl test(s)...");
Packit 284210
                $rc = eval { &{$t{test}} };
Packit 284210
                if (! defined $rc) {
Packit 284210
                    msg("Error running test: $@");
Packit 284210
                    $rc = -1;
Packit 284210
                }
Packit 284210
                dbg("Perl tests returned: $rc");
Packit 284210
            }
Packit 284210
Packit 284210
            # Search for all log matches
Packit 284210
            if ($rc == 0 and exists $t{match_log} and defined $t{match_log}) {
Packit 284210
                for my $key (keys %{ $t{match_log} || {}}) {
Packit 284210
                    my($neg,$mtype) = ($key =~ m/^(-?)(.*)$/);
Packit 284210
                    my $m = $t{match_log}{$key};
Packit 284210
                    if (ref($m) eq "HASH") {
Packit 284210
                        if ($m->{$platform}) {
Packit 284210
                            $m = $m->{$platform};
Packit 284210
                        }
Packit 284210
                        else {
Packit 284210
                            my $ap = join(", ", keys %{$m});
Packit 284210
                            msg("Warning: trying to match: $mtype. Nothing " .
Packit 284210
                                    "to match in current platform: $platform. " .
Packit 284210
                                    "This test only contains cotent for: $ap.");
Packit 284210
                            last;
Packit 284210
                        }
Packit 284210
                    }
Packit 284210
                    my $match = match_log($mtype, @{$m || []});
Packit 284210
                    if ($neg and defined $match) {
Packit 284210
                        $rc = 1;
Packit 284210
                        msg("$mtype log matched: $m->[0]");
Packit 284210
                        last;
Packit 284210
                    }
Packit 284210
                    elsif (!$neg and !defined $match) {
Packit 284210
                        $rc = 1;
Packit 284210
                        msg("$mtype log failed to match: $m->[0]");
Packit 284210
                        last;
Packit 284210
                    }
Packit 284210
                }
Packit 284210
            }
Packit 284210
Packit 284210
            # Search for all file matches
Packit 284210
            if ($rc == 0 and exists $t{match_file} and defined $t{match_file}) {
Packit 284210
                sleep 1; # Make sure the file exists
Packit 284210
                for my $key (keys %{ $t{match_file} || {}}) {
Packit 284210
                    my($neg,$fn) = ($key =~ m/^(-?)(.*)$/);
Packit 284210
                    my $m = $t{match_file}{$key};
Packit 284210
                    my $match = match_file($fn, $m);
Packit 284210
                    if ($neg and defined $match) {
Packit 284210
                        $rc = 1;
Packit 284210
                        msg("$fn file matched: $m");
Packit 284210
                        last;
Packit 284210
                    }
Packit 284210
                    elsif (!$neg and !defined $match) {
Packit 284210
                        $rc = 1;
Packit 284210
                        msg("$fn file failed match: $m");
Packit 284210
                        last;
Packit 284210
                    }
Packit 284210
                }
Packit 284210
            }
Packit 284210
        }
Packit 284210
        else {
Packit 284210
            msg("Failed to start httpd.");
Packit 284210
            $rc = 1;
Packit 284210
        }
Packit 284210
Packit 284210
        if ($rc == 0) {
Packit 284210
            $pass++;
Packit 284210
        }
Packit 284210
        else {
Packit 284210
            vrb("Test Config: $conf_fn");
Packit 284210
            vrb("Debug Log: $FILE{debug}{fn}");
Packit 284210
            dbg(escape("$FILE{debug}{buf}"));
Packit 284210
            vrb("Error Log: $FILE{error}{fn}");
Packit 284210
            dbg(escape("$FILE{error}{buf}"));
Packit 284210
        }
Packit 284210
Packit 284210
        msg(sprintf("%s) %s%s: %s%s", $id, $t{type}, (exists($t{comment}) ? " - $t{comment}" : ""), ($rc ? "failed" : "passed"), ((defined($out) && $out ne "")? " ($out)" : "")));
Packit 284210
    
Packit 284210
        if ($httpd_up) {
Packit 284210
            $httpd_up = httpd_stop(\%t) ? 0 : 1;
Packit 284210
        }
Packit 284210
Packit 284210
    }
Packit 284210
Packit 284210
    $TOTAL += $testnum ? 1 : $n;
Packit 284210
    $PASSED += $pass;
Packit 284210
Packit 284210
    msg(sprintf("Passed: %2d; Failed: %2d", $pass, $testnum ? (1 - $pass) : ($n - $pass)));
Packit 284210
}
Packit 284210
Packit 284210
# Take out any indenting and translate LF -> CRLF
Packit 284210
sub normalize_raw_request_data {
Packit 284210
    my $r = $_[0];
Packit 284210
Packit 284210
    # Allow for indenting in test file
Packit 284210
    $r =~ s/^[ \t]*\x0d?\x0a//s;
Packit 284210
    my($indention) = ($r =~ m/^([ \t]*)/s); # indention taken from first line
Packit 284210
    $r =~ s/^$indention//mg;
Packit 284210
    $r =~ s/(\x0d?\x0a)[ \t]+$/$1/s;
Packit 284210
Packit 284210
    # Translate LF to CRLF
Packit 284210
    $r =~ s/^\x0a/\x0d\x0a/mg;
Packit 284210
    $r =~ s/([^\x0d])\x0a/$1\x0d\x0a/mg;
Packit 284210
Packit 284210
    return $r;
Packit 284210
}
Packit 284210
Packit 284210
sub do_raw_request {
Packit 284210
    my $sock = new IO::Socket::INET(
Packit 284210
        Proto => "tcp",
Packit 284210
        PeerAddr => "localhost",
Packit 284210
        PeerPort => $opt{p},
Packit 284210
    ) or msg("Failed to connect to localhost:$opt{p}: $@");
Packit 284210
    return unless ($sock);
Packit 284210
Packit 284210
    # Join togeather the request
Packit 284210
    my $r = join("", @_);
Packit 284210
    dbg($r);
Packit 284210
Packit 284210
    # Write to socket
Packit 284210
    print $sock "$r";
Packit 284210
    $sock->shutdown(1);
Packit 284210
Packit 284210
    # Read from socket
Packit 284210
    my @resp = <$sock>;
Packit 284210
    $sock->close();
Packit 284210
Packit 284210
    return HTTP::Response->parse(join("", @resp));
Packit 284210
}
Packit 284210
Packit 284210
sub do_request {
Packit 284210
    my $r = $_[0];
Packit 284210
  
Packit 284210
    # Allow test to execute code
Packit 284210
    if (ref $r eq "CODE") {
Packit 284210
        $r = eval { &$r };
Packit 284210
        msg("$@") unless (defined $r);
Packit 284210
    }
Packit 284210
Packit 284210
    if (ref $r eq "HTTP::Request") {
Packit 284210
        my $resp = $UA->request($r);
Packit 284210
        dbg($resp->request()->as_string()) if ($opt{d});
Packit 284210
        return $resp
Packit 284210
    }
Packit 284210
    else {
Packit 284210
        return do_raw_request($r);
Packit 284210
    }
Packit 284210
Packit 284210
    return;
Packit 284210
}
Packit 284210
Packit 284210
Packit 284210
sub match_response {
Packit 284210
    my($name, $resp, $re) = @_;
Packit 284210
Packit 284210
    msg("Warning: Empty regular expression.") if (!defined $re or $re eq "");
Packit 284210
Packit 284210
    if ($name eq "status") {
Packit 284210
        return $& if ($resp->code =~ m/$re/);
Packit 284210
    }
Packit 284210
    elsif ($name eq "content") {
Packit 284210
        return $& if ($resp->content =~ m/$re/m);
Packit 284210
    }
Packit 284210
    elsif ($name eq "raw") {
Packit 284210
        return $& if ($resp->as_string =~ m/$re/m);
Packit 284210
    }
Packit 284210
Packit 284210
    return;
Packit 284210
}
Packit 284210
Packit 284210
sub read_log {
Packit 284210
    my($name, $timeout, $graph) = @_;
Packit 284210
    return match_log($name, undef, $timeout, $graph);
Packit 284210
}
Packit 284210
Packit 284210
sub match_log {
Packit 284210
    my($name, $re, $timeout, $graph) = @_;
Packit 284210
    my $t0 = gettimeofday;
Packit 284210
    my($fh,$rbuf) = ($FILE{$name}{fd}, \$FILE{$name}{buf});
Packit 284210
    my $n = length($$rbuf);
Packit 284210
    my $rc = undef;
Packit 284210
Packit 284210
    unless (defined $fh) {
Packit 284210
        msg("Error: File \"$name\" is not opened for matching.");
Packit 284210
        return;
Packit 284210
    }
Packit 284210
Packit 284210
    $timeout = 0 unless (defined $timeout);
Packit 284210
Packit 284210
    my $i = 0;
Packit 284210
    my $graphed = 0;
Packit 284210
    READ: {
Packit 284210
        do {
Packit 284210
            my $nbytes = $fh->sysread($$rbuf, $BUFSIZ, $n);
Packit 284210
            if (!defined($nbytes)) {
Packit 284210
                msg("Error: Could not read \"$name\" log: $!");
Packit 284210
                last;
Packit 284210
            }
Packit 284210
            elsif (!defined($re) and $nbytes == 0) {
Packit 284210
                last;
Packit 284210
            }
Packit 284210
Packit 284210
            # Remove APR pool debugging
Packit 284210
            $$rbuf =~ s/POOL DEBUG:[^\n]+PALLOC[^\n]+\n//sg;
Packit 284210
Packit 284210
            $n = length($$rbuf);
Packit 284210
Packit 284210
            #dbg("Match \"$re\" in $name \"$$rbuf\" ($n)");
Packit 284210
            if ($$rbuf =~ m/$re/m) {
Packit 284210
                $rc = $&;
Packit 284210
                last;
Packit 284210
            }
Packit 284210
            # TODO: Use select()/poll()
Packit 284210
            sleep 0.1 unless ($nbytes == $BUFSIZ);
Packit 284210
            if ($graph and $opt{d}) {
Packit 284210
                $i++;
Packit 284210
                if ($i == 10) {
Packit 284210
                    $graphed++;
Packit 284210
                    $i=0;
Packit 284210
                    print STDERR $graph if ($graphed == 1);
Packit 284210
                    print STDERR "."
Packit 284210
                }
Packit 284210
            }
Packit 284210
        } while (gettimeofday - $t0 < $timeout);
Packit 284210
    }
Packit 284210
    print STDERR "\n" if ($graphed);
Packit 284210
Packit 284210
    return $rc;
Packit 284210
}
Packit 284210
Packit 284210
sub match_file {
Packit 284210
    my($neg,$fn) = ($_[0] =~ m/^(-?)(.*)$/);
Packit 284210
    unless (exists $FILE{$fn}) {
Packit 284210
        eval {
Packit 284210
            $FILE{$fn}{fn} = $fn;
Packit 284210
            $FILE{$fn}{fd} = new FileHandle($fn, O_RDONLY) or die "$!\n";
Packit 284210
            $FILE{$fn}{fd}->blocking(0);
Packit 284210
            $FILE{$fn}{buf} = "";
Packit 284210
        };
Packit 284210
        if ($@) {
Packit 284210
            msg("Warning: Failed to open file \"$fn\": $@");
Packit 284210
            return;
Packit 284210
        }
Packit 284210
    }
Packit 284210
    return match_log($_[0], $_[1]); # timeout makes no sense
Packit 284210
}
Packit 284210
Packit 284210
sub quote_shell {
Packit 284210
    my($s) = @_;
Packit 284210
    return $s unless ($s =~ m|[^\w!%+,\-./:@^]|);
Packit 284210
    $s =~ s/(['\\])/\\$1/g;
Packit 284210
    return "'$s'";
Packit 284210
}
Packit 284210
Packit 284210
sub escape {
Packit 284210
    my @new = ();
Packit 284210
    for my $c (split(//, $_[0])) {
Packit 284210
        my $oc = ord($c);
Packit 284210
        push @new, ((($oc >= 0x20 and $oc <= 0x7e) or $oc == 0x0a or $oc == 0x0d) ? $c : sprintf("\\x%02x", ord($c)));
Packit 284210
    }
Packit 284210
    join('', @new);
Packit 284210
}
Packit 284210
Packit 284210
sub dbg {
Packit 284210
    return unless(@_ and $opt{d});
Packit 284210
    my $out = join "", map {
Packit 284210
        (ref $_ ne "" ? Dumper($_) : $_)
Packit 284210
    } @_;
Packit 284210
    $out =~ s/^/DBG: /mg;
Packit 284210
    print STDOUT "$out\n";
Packit 284210
}
Packit 284210
Packit 284210
sub vrb {
Packit 284210
    return unless(@_ and $opt{v});
Packit 284210
    msg(@_);
Packit 284210
}
Packit 284210
Packit 284210
sub msg {
Packit 284210
    return unless(@_);
Packit 284210
    my $out = join "", map {
Packit 284210
        (ref $_ ne "" ? Dumper($_) : $_)
Packit 284210
    } @_;
Packit 284210
    print STDOUT "$out\n";
Packit 284210
}
Packit 284210
Packit 284210
sub handle_interrupt {
Packit 284210
    $SIG{TERM} = $SIG{INT} = \&handle_interrupt;
Packit 284210
Packit 284210
    msg("Interrupted via SIG$_[0].  Shutting down tests...");
Packit 284210
    httpd_stop();
Packit 284210
Packit 284210
    quit(1);
Packit 284210
}
Packit 284210
Packit 284210
sub quit {
Packit 284210
    my($ec,$msg) = @_;
Packit 284210
    $ec = 0 unless (defined $_[0]);
Packit 284210
Packit 284210
    msg("$msg") if (defined $msg);
Packit 284210
Packit 284210
    exit $ec;
Packit 284210
}
Packit 284210
Packit 284210
sub done {
Packit 284210
    if ($PASSED != $TOTAL) {
Packit 284210
        quit(1, "\n$PASSED/$TOTAL tests passed.");
Packit 284210
    }
Packit 284210
Packit 284210
    quit(0, "\nAll tests passed ($TOTAL).");
Packit 284210
}
Packit 284210
Packit 284210
sub httpd_start {
Packit 284210
    my $t = shift;
Packit 284210
    httpd_reset_fd($t);
Packit 284210
    my @p = (
Packit 284210
        $HTTPD,
Packit 284210
        -d => $opt{S},
Packit 284210
        -f => $opt{C},
Packit 284210
        (map { (-c => $_) } ("Listen localhost:$opt{p}", @_)),
Packit 284210
        -k => "start",
Packit 284210
    );
Packit 284210
Packit 284210
    my $httpd_out;
Packit 284210
    my $httpd_pid = open3(undef, $httpd_out, undef, @p) or quit(1);
Packit 284210
    my $out = join("\\n", grep(!/POOL DEBUG/, (<$httpd_out>)));
Packit 284210
    close $httpd_out;
Packit 284210
    waitpid($httpd_pid, 0);
Packit 284210
Packit 284210
    my $rc = $?;
Packit 284210
    if ( WIFEXITED($rc) ) {
Packit 284210
        $rc = WEXITSTATUS($rc);
Packit 284210
        vrb("Httpd start returned with $rc.") if ($rc);
Packit 284210
    }
Packit 284210
    elsif( WIFSIGNALED($rc) ) {
Packit 284210
        msg("Httpd start failed with signal " . WTERMSIG($rc) . ".");
Packit 284210
        $rc = -1;
Packit 284210
    }
Packit 284210
    else {
Packit 284210
        msg("Httpd start failed with unknown error.");
Packit 284210
        $rc = -1;
Packit 284210
    }
Packit 284210
Packit 284210
    if (defined $out and $out ne "") {
Packit 284210
        vrb(join(" ", map { quote_shell($_) } @p));
Packit 284210
        msg("Httpd start failed with error messages:\n$out");
Packit 284210
        httpd_stop();
Packit 284210
        return -1
Packit 284210
    }
Packit 284210
Packit 284210
    # Look for startup msg
Packit 284210
    unless (defined match_log("error", qr/resuming normal operations/, 60, "Waiting on httpd to start: ")) {
Packit 284210
        vrb(join(" ", map { quote_shell($_) } @p));
Packit 284210
        vrb(match_log("error", qr/(^.*ModSecurity: .*)/sm, 10));
Packit 284210
        msg("Httpd server failed to start.");
Packit 284210
        httpd_stop();
Packit 284210
        return -1;
Packit 284210
    }
Packit 284210
Packit 284210
    return $rc;
Packit 284210
}
Packit 284210
Packit 284210
sub httpd_stop {
Packit 284210
    my $t = shift;
Packit 284210
    my @p = (
Packit 284210
        $HTTPD,
Packit 284210
        -d => $opt{S},
Packit 284210
        -f => $opt{C},
Packit 284210
        (map { (-c => $_) } ("Listen localhost:$opt{p}", @_)),
Packit 284210
        -k => "stop",
Packit 284210
    );
Packit 284210
Packit 284210
    my $httpd_out;
Packit 284210
    my $httpd_pid = open3(undef, $httpd_out, undef, @p) or quit(1);
Packit 284210
    my $out = join("\\n", grep(!/POOL DEBUG/, (<$httpd_out>)));
Packit 284210
    close $httpd_out;
Packit 284210
    waitpid($httpd_pid, 0);
Packit 284210
Packit 284210
    if (defined $out and $out ne "") {
Packit 284210
        msg("Httpd stop failed with error messages:\n$out");
Packit 284210
        return -1
Packit 284210
    }
Packit 284210
Packit 284210
    my $rc = $?;
Packit 284210
    if ( WIFEXITED($rc) ) {
Packit 284210
        $rc = WEXITSTATUS($rc);
Packit 284210
        vrb("Httpd stop returned with $rc.") if ($rc);
Packit 284210
    }
Packit 284210
    elsif( WIFSIGNALED($rc) ) {
Packit 284210
        msg("Httpd stop failed with signal " . WTERMSIG($rc) . ".");
Packit 284210
        $rc = -1;
Packit 284210
    }
Packit 284210
    else {
Packit 284210
        msg("Httpd stop failed with unknown error.");
Packit 284210
        $rc = -1;
Packit 284210
    }
Packit 284210
Packit 284210
    # Look for startup msg
Packit 284210
    unless (defined match_log("error", qr/caught SIG[A-Z]+, shutting down/, 60, "Waiting on httpd to stop: ")) {
Packit 284210
        vrb(join(" ", map { quote_shell($_) } @p));
Packit 284210
        msg("Httpd server failed to shutdown.");
Packit 284210
        sleep 0.5;
Packit 284210
        return -1;
Packit 284210
    }
Packit 284210
Packit 284210
    sleep 0.5;
Packit 284210
Packit 284210
    return $rc;
Packit 284210
}
Packit 284210
Packit 284210
sub httpd_reload {
Packit 284210
    my $t = shift;
Packit 284210
    httpd_reset_fd($t);
Packit 284210
    my @p = (
Packit 284210
        $HTTPD,
Packit 284210
        -d => $opt{S},
Packit 284210
        -f => $opt{C},
Packit 284210
        (map { (-c => $_) } ("Listen localhost:$opt{p}", @_)),
Packit 284210
        -k => "graceful",
Packit 284210
    );
Packit 284210
Packit 284210
    my $httpd_out;
Packit 284210
    my $httpd_pid = open3(undef, $httpd_out, undef, @p) or quit(1);
Packit 284210
    my $out = join("\\n", grep(!/POOL DEBUG/, (<$httpd_out>)));
Packit 284210
    close $httpd_out;
Packit 284210
    waitpid($httpd_pid, 0);
Packit 284210
Packit 284210
    if (defined $out and $out ne "") {
Packit 284210
        msg("Httpd reload failed with error messages:\n$out");
Packit 284210
        return -1
Packit 284210
    }
Packit 284210
Packit 284210
    my $rc = $?;
Packit 284210
    if ( WIFEXITED($rc) ) {
Packit 284210
        $rc = WEXITSTATUS($rc);
Packit 284210
        vrb("Httpd reload returned with $rc.") if ($rc);
Packit 284210
    }
Packit 284210
    elsif( WIFSIGNALED($rc) ) {
Packit 284210
        msg("Httpd reload failed with signal " . WTERMSIG($rc) . ".");
Packit 284210
        $rc = -1;
Packit 284210
    }
Packit 284210
    else {
Packit 284210
        msg("Httpd reload failed with unknown error.");
Packit 284210
        $rc = -1;
Packit 284210
    }
Packit 284210
Packit 284210
    # Look for startup msg
Packit 284210
    unless (defined match_log("error", qr/resuming normal operations/, 60, "Waiting on httpd to restart: ")) {
Packit 284210
        vrb(join(" ", map { quote_shell($_) } @p));
Packit 284210
        msg("Httpd server failed to reload.");
Packit 284210
        return -1;
Packit 284210
    }
Packit 284210
Packit 284210
    return $rc;
Packit 284210
}
Packit 284210
Packit 284210
sub httpd_reset_fd {
Packit 284210
    my($t) = @_;
Packit 284210
Packit 284210
    # Cleanup
Packit 284210
    for my $key (keys %FILE) {
Packit 284210
        if (exists $FILE{$key}{fd} and defined $FILE{$key}{fd}) {
Packit 284210
            $FILE{$key}{fd}->close();
Packit 284210
        }
Packit 284210
        delete $FILE{$key};
Packit 284210
    }
Packit 284210
Packit 284210
    # Error
Packit 284210
    eval {
Packit 284210
        $FILE{error}{fn} = $opt{E};
Packit 284210
        $FILE{error}{fd} = new FileHandle($opt{E}, O_RDWR|O_CREAT) or die "$!\n";
Packit 284210
        $FILE{error}{fd}->blocking(0);
Packit 284210
        $FILE{error}{fd}->sysseek(0, 2);
Packit 284210
        $FILE{error}{buf} = "";
Packit 284210
    };
Packit 284210
    if ($@) {
Packit 284210
        msg("Warning: Failed to open file \"$opt{E}\": $@");
Packit 284210
        return undef;
Packit 284210
    }
Packit 284210
Packit 284210
    # Audit
Packit 284210
    eval {
Packit 284210
        $FILE{audit}{fn} = $opt{A};
Packit 284210
        $FILE{audit}{fd} = new FileHandle($opt{A}, O_RDWR|O_CREAT) or die "$!\n";
Packit 284210
        $FILE{audit}{fd}->blocking(0);
Packit 284210
        $FILE{audit}{fd}->sysseek(0, 2);
Packit 284210
        $FILE{audit}{buf} = "";
Packit 284210
    };
Packit 284210
    if ($@) {
Packit 284210
        msg("Warning: Failed to open file \"$opt{A}\": $@");
Packit 284210
        return undef;
Packit 284210
    }
Packit 284210
Packit 284210
    # Debug
Packit 284210
    eval {
Packit 284210
        $FILE{debug}{fn} = $opt{D};
Packit 284210
        $FILE{debug}{fd} = new FileHandle($opt{D}, O_RDWR|O_CREAT) or die "$!\n";
Packit 284210
        $FILE{debug}{fd}->blocking(0);
Packit 284210
        $FILE{debug}{fd}->sysseek(0, 2);
Packit 284210
        $FILE{debug}{buf} = "";
Packit 284210
    };
Packit 284210
    if ($@) {
Packit 284210
        msg("Warning: Failed to open file \"$opt{D}\": $@");
Packit 284210
        return undef;
Packit 284210
    }
Packit 284210
Packit 284210
    # Any extras listed in "match_log"
Packit 284210
    if ($t and exists $t->{match_log}) {
Packit 284210
        for my $k (keys %{ $t->{match_log} || {} }) {
Packit 284210
            my($neg,$fn) = ($k =~ m/^(-?)(.*)$/);
Packit 284210
            next if (!$fn or exists $FILE{$fn});
Packit 284210
            eval {
Packit 284210
                $FILE{$fn}{fn} = $fn;
Packit 284210
                $FILE{$fn}{fd} = new FileHandle($fn, O_RDWR|O_CREAT) or die "$!\n";
Packit 284210
                $FILE{$fn}{fd}->blocking(0);
Packit 284210
                $FILE{$fn}{fd}->sysseek(0, 2);
Packit 284210
                $FILE{$fn}{buf} = "";
Packit 284210
            };
Packit 284210
            if ($@) {
Packit 284210
                msg("Warning: Failed to open file \"$fn\": $@");
Packit 284210
                return undef;
Packit 284210
            }
Packit 284210
        }
Packit 284210
    }
Packit 284210
}
Packit 284210
Packit 284210
sub encode_chunked {
Packit 284210
    my($data, $size) = @_;
Packit 284210
    $size = 128 unless ($size);
Packit 284210
    my $chunked = "";
Packit 284210
  
Packit 284210
    my $n = 0;
Packit 284210
    my $bytes = length($data);
Packit 284210
    while ($bytes >= $size) {
Packit 284210
        $chunked .= sprintf "%x\x0d\x0a%s\x0d\x0a", $size, substr($data, $n, $size);
Packit 284210
        $n += $size;
Packit 284210
        $bytes -= $size;
Packit 284210
    }
Packit 284210
    if ($bytes) {
Packit 284210
        $chunked .= sprintf "%x\x0d\x0a%s\x0d\x0a", $bytes, substr($data, $n, $bytes);
Packit 284210
    }
Packit 284210
    $chunked .= "0\x0d\x0a\x0d\x0a"
Packit 284210
}