Gheek.net

my(@commandtable, %commands, @commands);# command lists
my(%filter_pwds);                       # password filtering mode

# This routine is used to print out the router configuration
sub ProcessHistory {

    ($new_hist_tag,$new_command,$command_string, @string) = (@_);
    if ((($new_hist_tag ne $hist_tag) || ($new_command ne $command))
        && defined %history) {
        print eval "$command \%history";
        undef %history;
    }
    if (($new_hist_tag) && ($new_command) && ($command_string)) {
        if ($history{$command_string}) {
            $history{$command_string} = "$history{$command_string}@string";
        } else {
            $history{$command_string} = "@string";
        }
    } elsif (($new_hist_tag) && ($new_command)) {
        $history{++$#history} = "@string";
    } else {
        print "@string";
    }
    $hist_tag = $new_hist_tag;
    $command = $new_command;
    1;
}

sub numerically { $a <=> $b; }

# This is a sort routine that will sort numerically on the
# keys of a hash as if it were a normal array.
sub keynsort {
    local(%lines) = @_;
    local($i) = 0;
    local(@sorted_lines);
    foreach $key (sort numerically keys(%lines)) {
        $sorted_lines[$i] = $lines{$key};
        $i++;
    }
    @sorted_lines;
}

# This is a sort routine that will sort on the
# keys of a hash as if it were a normal array.
sub keysort {
    local(%lines) = @_;
    local($i) = 0;
    local(@sorted_lines);
    foreach $key (sort keys(%lines)) {
        $sorted_lines[$i] = $lines{$key};
        $i++;
    }
    @sorted_lines;
}

# This is a sort routine that will sort on the
# values of a hash as if it were a normal array.
sub valsort{
    local(%lines) = @_;
    local($i) = 0;
    local(@sorted_lines);
    foreach $key (sort values %lines) {
        $sorted_lines[$i] = $key;
        $i++;
    }
    @sorted_lines;
}

# This is a numerical sort routine (ascending).
sub numsort {
    local(%lines) = @_;
    local($i) = 0;
    local(@sorted_lines);
    foreach $num (sort {$a <=> $b} keys %lines) {
        $sorted_lines[$i] = $lines{$num};
        $i++;
    }
    @sorted_lines;
}

# This is a sort routine that will sort on the
# ip address when the ip address is anywhere in
# the strings.
sub ipsort {
    local(%lines) = @_;
    local($i) = 0;
    local(@sorted_lines);
    foreach $addr (sort sortbyipaddr keys %lines) {
        $sorted_lines[$i] = $lines{$addr};
        $i++;
    }
    @sorted_lines;
}

# These two routines will sort based upon IP addresses
sub ipaddrval {
    my(@a) = ($_[0] =~ m#^(\d+)\.(\d+)\.(\d+)\.(\d+)$#);
    $a[3] + 256 * ($a[2] + 256 * ($a[1] +256 * $a[0]));
}
sub sortbyipaddr {
    &ipaddrval($a) <=> &ipaddrval($b);
}

# This routine parses "show config"
sub ShowConfig {
    print STDERR "    In ShowConfig: $_" if ($debug);

    $firstexit=0;

    while (<INPUT>) {
        tr/\015//d;
        tr/\020//d;

        #strip out the stupid spinning running-config progress thingy
        s/Generating current config: \.*[\|\/\-\\]//gi;
        $skipprocess=0;

        #sometimes an ‘exit’ appears at the top of the config, we don’t want them
        if ( (/^exit/) && ( ! $firstexit ) ) {
                $firstexit=1;
                $skipprocess=1;
        }

        #remove spaces left over from lame spinning progress thingy
        if ( /^\s+! ——————————/ ) {
                s/^\s+!/!/g
        }

        if (/^(read-only-community) / && $filter_pwds >= 1) {
            ProcessHistory("","","","!$1 <removed>\n"); next;
        }
        if (/^(read-write-community) / && $filter_pwds >= 1) {
            ProcessHistory("","","","!$1 <removed>\n"); next;
        }
        if (/^(trap-community-name) / && $filter_pwds >= 1) {
            ProcessHistory("","","","!$1 <removed>\n"); next;
        }
        if (/^(ntp-keys \d+ md5-key) / && $filter_pwds >= 1) {
            ProcessHistory("","","","!$1 <removed>\n"); next;
        }
        if (/^(password) / && $filter_pwds >= 1) {
            ProcessHistory("","","","!$1 <removed>\n"); next;
        }

        last if (/^$prompt/);
        next if (/^(\s*|\s*$cmd\s*)$/);
        if ( ! /^$prompt/) {
                if ( ! $skipprocess ) {
                        print STDOUT "      ShowConfig Data: $_" if ($debug);
                        ProcessHistory("","","","$_");
                }
        }
    }
    $clean_run=1;
    print STDERR "    Exiting ShowConfig: $_" if ($debug);
    return(0);
}

# This routine parses single command’s that return no required info
sub ShowVersion {
    print STDERR "    In ShowVersion: $_" if ($debug);
    ProcessHistory("","","","!\n!IPS Show Version Start\n");

    while (<INPUT>) {
        tr/\015//d;

        $skipprocess=0;

        if ( /^Sensor up-time/ ) { $skipprocess=1; }
        if ( ( /using.*bytes of available/i ) ) { $skipprocess=1; }

        last if (/^$prompt/);
        next if (/^(\s*|\s*$cmd\s*)$/);
        if ( ! /^$prompt/) {
                if ( ! $skipprocess ) {
                        print STDOUT "      ShowVersion Data: $_" if ($debug);
                        ProcessHistory("","","","! $_");
                }
        }
    }
    ProcessHistory("","","","!\n!IPS Show Version End\n");
    print STDERR "    Exiting ShowVersion: $_" if ($debug);
    return(0)
}

# This routine parses single command’s that return no required info
sub ShowUsersAll {
    print STDERR "    In ShowUsersAll: $_" if ($debug);
    ProcessHistory("","","","!\n!IPS User Database Start\n");

    while (<INPUT>) {
        tr/\015//d;

        $skipprocess=0;

        s/^    CLI ID   //g;
        s/^             //g;
        s/^\* +[0-9]+ +//g;

        last if (/^$prompt/);
        next if (/^(\s*|\s*$cmd\s*)$/);
        if ( ! /^$prompt/) {
                if ( ! $skipprocess ) {
                        print STDOUT "      ShowUsersAll Data: $_" if ($debug);
                        ProcessHistory("","","","!$_");
                }
        }
    }
    ProcessHistory("","","","!\n!IPS User Database End\n!\n!\n");
    print STDERR "    Exiting ShowUsersAll: $_" if ($debug);
    return(0)
}

# dummy function
sub DoNothing {print STDOUT;}

# Main
@commandtable = (
        {’show version’         => ‘ShowVersion’},
        {’show users all’       => ‘ShowUsersAll’},
        {’show configuration’   => ‘ShowConfig’}
);
# Use an array to preserve the order of the commands and a hash for mapping
# commands to the subroutine and track commands that have been completed.
@commands = map(keys(%$_), @commandtable);
%commands = map(%$_, @commandtable);

$cisco_cmds=join(";",@commands);
$cmds_regexp=join("|",@commands);

open(OUTPUT,">$host.new") || die "Can’t open $host.new for writing: $!\n";
select(OUTPUT);
# make OUTPUT unbuffered if debugging
if ($debug) { $| = 1; }

# The IPS doesn’t like the TERM of network so we must change it
if ( $ENV{TERM} eq ‘network’ ) {
$ENV{TERM} = ‘vt100′;
}
if ($file) {
    print STDERR "opening file $host\n" if ($debug);
    print STDOUT "opening file $host\n" if ($log);
    open(INPUT,"<$host") || die "open failed for $host: $!\n";
} else {
    print STDERR "executing clogin -t $timeo -c\"$cisco_cmds\" $host\n" if ($debug);
    print STDOUT "executing clogin -t $timeo -c\"$cisco_cmds\" $host\n" if ($log);
    if (defined($ENV{NOPIPE})) {
        system "clogin -t $timeo -c \"$cisco_cmds\" $host </dev/null > $host.raw 2>&1" || die "clogin failed for $host: $!\n";
        open(INPUT, "< $host.raw") || die "clogin failed for $host: $!\n";
    } else {
        open(INPUT,"clogin -t $timeo -c \"$cisco_cmds\" $host </dev/null |") || die "clogin failed for $host: $!\n";
    }
}
# Change the TERM back to network
if ( $ENV{TERM} eq ‘vt100′ ) {
$ENV{TERM} = ‘network’;
}

# determine password filtering mode
if ($ENV{"FILTER_PWDS"} =~ /no/i) {
        $filter_pwds = 0;
} elsif ($ENV{"FILTER_PWDS"} =~ /all/i) {
        $filter_pwds = 2;
} else {
        $filter_pwds = 1;
}

ProcessHistory("","","","!RANCID-CONTENT-TYPE: ipsrancid\n!\n");
TOP: while(<INPUT>) {
    tr/\015//d;

    #strip out the stupid spinning running-config progress thingy
    s/Generating current config: \.*[\|\/\-\\]//gi;

    if (/^.*logout$/)  {
        $clean_run=1;
        last;
    }
    if (/^Error:/) {
        print STDOUT ("$host clogin error: $_");
        print STDERR ("$host clogin error: $_") if ($debug);
        $clean_run=0;
        last;
    }
    while (/($cmds_regexp)/) {
        $cmd = $1;
        if (!defined($prompt)) {
            $prompt = ($_ =~ /^([^#]+#)/)[0];
            $prompt =~ s/([][}{)(\\])/\\$1/g;
            print STDERR ("PROMPT MATCH: $prompt\n") if ($debug);
        }
        print STDERR ("IPS COMMAND:$_") if ($debug);
        if (! defined($commands{$cmd})) {
            print STDERR "$host: found unexpected command - \"$cmd\"\n";
            $clean_run = 0;
            last TOP;
        }
        $rval = &{$commands{$cmd}};
        delete($commands{$cmd});
        if ($rval == -1) {
            $clean_run = 0;
            last TOP;
        }
    }
}
print STDOUT "Done $logincmd: $_\n" if ($log);
# Flush History
ProcessHistory("","","","");
# Cleanup
close(INPUT);
close(OUTPUT);

if (defined($ENV{NOPIPE})) {
    unlink("$host.raw") if (! $debug);
}

# check for completeness
if (scalar(%commands) || !$clean_run ) {
    if (scalar(%commands)) {
        printf(STDOUT "$host: missed cmd(s): %s\n", join(‘,’, keys(%commands)));
        printf(STDERR "$host: missed cmd(s): %s\n", join(‘,’, keys(%commands))) if ($debug);
    }
    if (!$clean_run ) {
        print STDOUT "$host: End of run not found\n";
        print STDERR "$host: End of run not found\n" if ($debug);
        system("/usr/bin/tail -1 $host.new");
    }
    unlink "$host.new" if (! $debug);
}