#!/usr/bin/perl -wT

require 5.004;
use strict;
use lib utils.pm ;
use Plugin;
use Plugin::Parameter qw(:DEFAULT $configfileparameter);

use vars qw($opt_e $opt_c $opt_d $opt_t $PROGNAME);

use utils qw(%ERRORS);

use DBI;

my $c_opt = $configfileparameter;
$c_opt->flags([ 'c', 'config-file' ]);
$c_opt->default("/etc/nagios/cgi.cfg");
my $e_opt = new Plugin::Parameter(-name => "expire", -flags => [ 'e', 'expire' ],
				  -optional => "yes", -valueoptional => "no", -type => "INTEGER",
				  -default => 5,
				  -description => "Timeframe within which Nagios must have updated its status - given in minutes");
my $d_opt = new Plugin::Parameter(-name => "driver", -flags => [ 'd', 'driver' ],
				  -optional => "yes", -valueoptional => "yes", -type => "INTEGER",
				  -default => "mysql",
				  -description => "Driver to use to talk to the backend database - mysql, postgresql, ... - (See DBI)");
my $plugin = new Plugin(-revision => '$Revision: 1.1 $',
			-copyright => "Anonymous, 2004 Howard Wilkinson <howard\@cohtech.com>",
			-shortcomment => "Check to see if Nagios is running when configured to use Database storage",
			-parameterlists => [ [ $e_opt, $c_opt, $d_opt, $t_opt ], $h_opts, $V_opts ]);

$ENV{"PATH"} = "/usr/bin:/usr/sbin:/bin";
$ENV{"BASH_ENV"} = "";

$plugin->init();

my $QUERY = "select *, UNIX_TIMESTAMP(last_update) as ut from programstatus;";

my $PROCCNT = 0;

use constant OK => 1;
use constant WARN => 2;

my $STAT = 'UNKNOWN';

unless ( -f $opt_c ) {
  $plugin->usage();
  usage("$PROGNAME UNKNOWN: no config file '$opt_c': $!\n");
}

$plugin->start_timeout($opt_t, "Timed out while checking status");

my ($dbhost, $dbport, $dbuser, $dbpass, $dbname);

open(F, "< $opt_c");
while ( <F> ) {
  if (/^xsddb_host=(.+)/) { $dbhost = $1; next; };
  if (/^xsddb_port=(.+)/) { $dbport = $1; next; };
  if (/^xsddb_database=(.+)/) { $dbname = $1; next; };
  if (/^xsddb_username=(.+)/) { $dbuser = $1; next; };
  if (/^xsddb_password=(.+)/) { $dbpass = $1; next; };
}
close(F);

my $dsn = "DBI:$opt_d:database=$dbname;host=$dbhost;port=$dbport";
my $dbh = DBI->connect($dsn, $dbuser, $dbpass, {'RaiseError' => 1});

my $sth = $dbh->prepare($QUERY);
if (!$sth) {
  print "$PROGNAME UNKNOWN: " . $dbh->errstr . "\n";
  exit $ERRORS{'UNKNOWN'};
}
$sth->execute;
if (!$sth->execute) {
  print "$PROGNAME UNKNOWN: " . $sth->errstr . "\n";
  exit $ERRORS{'UNKNOWN'};
}

my %status = ();

my $names = $sth->{'NAME'};
my $numFields = $sth->{'NUM_OF_FIELDS'};
my $ref = $sth->fetchrow_arrayref;
for (my $i = 0;  $i < $numFields;  $i++) {
  $status{"$$names[$i]"} = $$ref[$i];
}

my $lastupdated = time() - $status{"ut"};
if ( $lastupdated < ($opt_e*60) ) { ## convert $opt_e to seconds
  $STAT = 'OK';
} else {
  $STAT = 'CRITICAL';
}

open(PS, "ps -eaf | grep $status{nagios_pid} | grep -v grep | ");
$PROCCNT = 0;
while(<PS>) {
  $PROCCNT++;
}
close(PS);

$plugin->stop_timeout();

print "$PROGNAME $STAT: located $PROCCNT processes, program status updated $lastupdated seconds ago\n";
exit $ERRORS{$STAT};
