#! /usr/bin/perl -w
#
#
# License Information:
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
#
############################################################################

use POSIX;
use strict;
use Getopt::Long;
use vars qw($host $opt_V $opt_h $opt_v $verbose $PROGNAME $opt_w $opt_e $opt_c $opt_t $opt_H $status $state $msg $msg_q $MAILQ $SHELL $device $used $avail $percent $fs $blocks $CMD $systype);
use lib  "/usr/local/nagios/libexec" ;
use utils qw(%ERRORS &print_revision &support &usage );


sub print_help ();
sub print_usage ();
sub process_arguments ();

$ENV{'PATH'}='';
$ENV{'BASH_ENV'}=''; 
$ENV{'ENV'}='';
$PROGNAME = "check_netstat_remote";

Getopt::Long::Configure('bundling');
$status = process_arguments();

if ($status){
	print "ERROR: processing arguments\n";
	exit $ERRORS{"UNKNOWN"};
}

$SIG{'ALRM'} = sub {
	print ("ERROR: timed out waiting for $CMD on $host\n");
	exit $ERRORS{"WARNING"};
};

$host = $opt_H;

alarm($opt_t);

my $systype = `$SHELL $host uname`;
chomp($systype);

#
# These might need to be modified for your OS....
#

SWITCH: {
  if ($systype eq "SunOS") {
	$CMD = "/usr/bin/netstat | /usr/bin/wc -l";
	last SWITCH;
  }
  if ($systype eq "Linux") {
	$CMD = "/usr/bin/netstat | /usr/bin/wc -l";
	last SWITCH;
  }
  if ($systype eq "AIX") {
	$CMD = "/usr/bin/netstat | /usr/bin/wc -l";
	last SWITCH;
  }
  if ($systype eq "HP-UX") {
	$CMD = "/usr/bin/netstat | /usr/bin/wc -l";
	last SWITCH;
  }
}#end switch

alarm($opt_t);

## get cmd output from remote system

if (! open (OUTPUT, "$SHELL $host $CMD| " ) ) {
	print "ERROR: could not open $CMD on $host\n";
	exit $ERRORS{'UNKNOWN'};
}

my $perfdata = "";
my $state = "-1";

# only first line is relevant in this iteration.
while (<OUTPUT>) {
	print $_ if $verbose;
		chomp;	
		my $val = $_;
		$val =~ s/^\s+//;
		$val =~ s/\s+$//;
		$perfdata = $perfdata . "sockets=$val;;;0;10000";
			print "sockets = $val warn=$opt_w crit=$opt_c\n" if $verbose;

			if ($val < $opt_w) {
				if ( $state <= $ERRORS{'OK'} ) {
					$state = $ERRORS{'OK'};
					$msg = "OK: Number of open TCP sockets ($val) are below threshold ($opt_w/$opt_c)";
				}
			} elsif ($val >= $opt_w  && $val < $opt_c) {
                                if ( $state <= $ERRORS{'WARNING'} ) {
					$state = $ERRORS{'WARNING'};
					$msg = "WARNING: number of sockets high $val (threshold w = $opt_w)";
				}
			} else {
                                if ( $state <= $ERRORS{'CRITICAL'} ) {
					$state = $ERRORS{'CRITICAL'};
					$msg = "CRITICAL: number of sockets is critical $val (threshold c = $opt_c)";
				}
			}
}

close (OUTPUT); 
# declare an error if we also get a non-zero return code from load
# unless already set to critical
if ( $? ) {
	print "stderr = $? : $! \n" if $verbose;
	$state = $state == $ERRORS{"CRITICAL"} ? $ERRORS{"CRITICAL"} : $ERRORS{"UNKNOWN"}  ;
	print "error: $!\n" if $verbose;
}
## close cmd

# Perfdata support
print "$msg | $perfdata\n";
exit $state;


#####################################
#### subs


sub process_arguments(){
	GetOptions
		("V"   => \$opt_V, "version"	=> \$opt_V,
		 "v"   => \$opt_v, "verbose"	=> \$opt_v,
		 "h"   => \$opt_h, "help"	=> \$opt_h,
		 "e=s" => \$opt_e, "shell=s"	=> \$opt_e,
		 "w=f" => \$opt_w, "warning=f"  => \$opt_w,   # warning if above this number
		 "c=f" => \$opt_c, "critical=f" => \$opt_c,   # critical if above this number
		 "t=i" => \$opt_t, "timeout=i"  => \$opt_t,
	         "H=s" => \$opt_H, "hostname=s" => \$opt_H
		 );

	if ($opt_V) {
		print_revision($PROGNAME,'$Revision: 1.0 $ ');
		exit $ERRORS{'OK'};
	}

	if ($opt_h) {
		print_help();
		exit $ERRORS{'OK'};
	}

	if (defined $opt_v ){
		$verbose = $opt_v;
	}

        if (defined $opt_e ){
		if ( $opt_e eq "ssh" ) {
		    if (-x "/usr/local/bin/ssh") {
			$SHELL = "/usr/local/bin/ssh";
		    } elsif ( -x "/usr/bin/ssh" ) {
			$SHELL = "/usr/bin/ssh"; 
		    } else {
                        print_usage();
                        exit $ERRORS{'UNKNOWN'};
                    }
	
		} elsif ( $opt_e eq "rsh" ) {
			$SHELL = "/usr/bin/rsh";
		} else {
			print_usage();
                	exit $ERRORS{'UNKNOWN'};
		}
        } else {
           print_usage();
           exit $ERRORS{'UNKNOWN'};
        }


	unless (defined $opt_t) {
		$opt_t = $utils::TIMEOUT ;	# default timeout
	}

	unless (defined $opt_H) {
		print_usage();
		exit $ERRORS{'UNKNOWN'};
	}

	unless (  defined $opt_w &&  defined $opt_c ) {
		print_usage();
		exit $ERRORS{'UNKNOWN'};
	}

	if ( $opt_w >= $opt_c) {
		print "Warning cannot be greater than Critical!\n";
		exit $ERRORS{'UNKNOWN'};
	}

	return $ERRORS{'OK'};
}

sub print_usage () {
	print "Usage: $PROGNAME -e <shell> -H <hostname> [-w <warn>] [-c <crit>] [-t <timeout>] [-v verbose]\n";
}

sub print_help () {
	print_revision($PROGNAME,'$Revision: 1.1 $');
	print "Copyright (c) 2004 jeff scott\n";
	print "\n";
	print_usage();
	print "\n";
	print "   Checks the five minute load average\n";
	print "   I chose five minute average to ignore spikes in cpu usage\n";
	print "-e (--shell)     = ssh or rsh (required)\n";
	print "-H (--hostname)  = remote server name (required)";
	print "-w (--warning)   = Five minute average to generate warning\n";
	print "-c (--critical)  = Five minute average to generate critical alert ( w < c )\n";
	print "-t (--timeout)   = Plugin timeout in seconds (default = $utils::TIMEOUT)\n";
	print "-h (--help)\n";
	print "-V (--version)\n";
	print "-v (--verbose)   = debugging output\n";
	print "\n\n";
	support();
}