Code

Fix incorrectly labelled units in output (1025905, Jason Martin)
[nagiosplug.git] / plugins-scripts / check_disk_smb.pl
index d1b0b3d63901f2025d76ce04866a73b2fa95a9c9..3358f6a139d19c657b39a852cf9d5c58c7c2625e 100755 (executable)
@@ -1,4 +1,4 @@
-#! /usr/bin/perl -wT
+#!/usr/bin/perl -w
 #
 #
 # check_disk.pl <host> <share> <user> <pass> [warn] [critical] [port]
 #  Modified [warn] and [critical] parameters to accept format of nnn[M|G] to
 #  allow setting of limits in MBytes or GBytes.  Percentage settings for large
 #  drives is a pain in the butt
-
-BEGIN {
-       if ($0 =~ m/^(.*?)[\/\\]([^\/\\]+)$/) {
-               $runtimedir = $1;
-               $PROGNAME = $2;
-       }
-}
+# 2-May-2002 - SGhosh fix for embedded perl
+#
+# $Id$
+#
 
 require 5.004;
 use POSIX;
 use strict;
 use Getopt::Long;
-use vars qw($opt_V $opt_h $opt_H $opt_s $opt_W $opt_u $opt_p $opt_w $opt_c $verbose);
+use vars qw($opt_P $opt_V $opt_h $opt_H $opt_s $opt_W $opt_u $opt_p $opt_w $opt_c $verbose);
 use vars qw($PROGNAME);
-use lib $main::runtimedir;
+use lib utils.pm ;
 use utils qw($TIMEOUT %ERRORS &print_revision &support &usage);
 
 sub print_help ();
 sub print_usage ();
 
+$PROGNAME = "check_disk_smb";
+
 $ENV{'PATH'}='';
 $ENV{'BASH_ENV'}=''; 
 $ENV{'ENV'}='';
@@ -41,6 +40,7 @@ $ENV{'ENV'}='';
 Getopt::Long::Configure('bundling');
 GetOptions
        ("v"   => \$verbose, "verbose"    => \$verbose,
+        "P=s" => \$opt_P, "port=s"     => \$opt_P,
         "V"   => \$opt_V, "version"    => \$opt_V,
         "h"   => \$opt_h, "help"       => \$opt_h,
         "w=s" => \$opt_w, "warning=s"  => \$opt_w,
@@ -58,34 +58,56 @@ if ($opt_V) {
 
 if ($opt_h) {print_help(); exit $ERRORS{'OK'};}
 
-my $smbclient="/usr/bin/smbclient";
-my $smbclientoptions="";
+my $smbclient= "$utils::PATH_TO_SMBCLIENT " ;
+my $smbclientoptions= $opt_P ? "-p $opt_P " : "";
+
+
+# Options checking
 
 ($opt_H) || ($opt_H = shift) || usage("Host name not specified\n");
-my $host = $1 if ($opt_H =~ /([-_.A-Za-z0-9]+)/);
+my $host = $1 if ($opt_H =~ /^([-_.A-Za-z0-9]+\$?)$/);
 ($host) || usage("Invalid host: $opt_H\n");
 
 ($opt_s) || ($opt_s = shift) || usage("Share volume not specified\n");
-my $share = $1 if ($opt_s =~ /([-_.A-Za-z0-9]+)/);
+my $share = $1 if ($opt_s =~ /^([-_.A-Za-z0-9]+\$?)$/);
 ($share) || usage("Invalid share: $opt_s\n");
 
 ($opt_u) || ($opt_u = shift) || ($opt_u = "guest");
-my $user = $1 if ($opt_u =~ /([-_.A-Za-z0-9]+)/);
+my $user = $1 if ($opt_u =~ /^([-_.A-Za-z0-9\\]+)$/);
 ($user) || usage("Invalid user: $opt_u\n");
 
 ($opt_p) || ($opt_p = shift) || ($opt_p = "guest");
 my $pass = $1 if ($opt_p =~ /(.*)/);
 
 ($opt_w) || ($opt_w = shift) || ($opt_w = 85);
-my $warn = $1 if ($opt_w =~ /([0-9]{1,2}\%?|100\%?|[0-9]+[kmKM])+/);
+my $warn = $1 if ($opt_w =~ /^([0-9]{1,2}\%?|100\%?|[0-9]+[kMG])$/);
 ($warn) || usage("Invalid warning threshold: $opt_w\n");
 
 ($opt_c) || ($opt_c = shift) || ($opt_c = 95);
-my $crit = $1 if ($opt_c =~ /([0-9]{1,2}\%?|100\%?|[0-9]+[kmKM])/);
+my $crit = $1 if ($opt_c =~ /^([0-9]{1,2}\%?|100\%?|[0-9]+[kMG])$/);
 ($crit) || usage("Invalid critical threshold: $opt_c\n");
 
+# check if both warning and critical are percentage or size
+unless( ( ($opt_w =~ /([0-9]){1,2}$/ ) && ($opt_c =~ /([0-9]){1,2}$/ )  )|| (( $opt_w =~ /[kMG]/ ) && ($opt_c =~ /[kMG]/) )  ){
+       usage("Both warning and critical should be same type- warning: $opt_w critical: $opt_c \n");
+}
+
+# verify warning is less than critical
+if ( $opt_w =~ /[kMG]/) {
+       unless ( $warn > $crit) {
+               usage("Disk size: warning ($opt_w) should be greater than critical ($opt_c) \n");
+       }
+}else{
+       unless ( $warn < $crit) {
+               usage("Percentage: warning ($opt_w) should be less than critical ($opt_c) \n");
+       }
+}
+
 my $workgroup = $1 if (defined($opt_W) && $opt_W =~ /(.*)/);
 
+# end of options checking
+
+
 my $state = "OK";
 my $answer = undef;
 my $res = undef;
@@ -103,6 +125,7 @@ alarm($TIMEOUT);
 if (defined($workgroup)) {
        $res = qx/$smbclient \/\/$host\/$share $pass -W $workgroup -U $user $smbclientoptions -c ls/;
 } else {
+       print "$smbclient " . "\/\/$host\/$share" ." $pass -U $user $smbclientoptions -c ls\n" if ($verbose);
        $res = qx/$smbclient \/\/$host\/$share $pass -U $user $smbclientoptions -c ls/;
 }
 #Turn off alarm
@@ -128,27 +151,28 @@ if (/\s*(\d*) blocks of size (\d*)\. (\d*) blocks available/) {
        #P = Percent, K = KBytes
        my $warn_type;
        my $crit_type;
-       if ($warn =~ /^([0-9]+$)/) {
+
+       if ($opt_w =~ /^([0-9]+$)/) {
                $warn_type = "P";
-       } elsif ($warn =~ /^([0-9]+)k$/) {
-               my ($warn_type) = "K";
+       } elsif ($opt_w =~ /^([0-9]+)k$/) {
+               $warn_type = "K";
                $warn = $1;
-       } elsif ($warn =~ /^([0-9]+)M$/) {
+       } elsif ($opt_w =~ /^([0-9]+)M$/) {
                $warn_type = "K";
                $warn = $1 * 1024;
-       } elsif ($warn =~ /^([0-9]+)G$/) {
+       } elsif ($opt_w =~ /^([0-9]+)G$/) {
                $warn_type = "K";
                $warn = $1 * 1048576;
        }
-       if ($crit =~ /^([0-9]+$)/) {
+       if ($opt_c =~ /^([0-9]+$)/) {
                $crit_type = "P";
-       } elsif ($crit =~ /^([0-9]+)k$/) {
+       } elsif ($opt_c =~ /^([0-9]+)k$/) {
                $crit_type = "K";
                $crit = $1;
-       } elsif ($crit =~ /^([0-9]+)M$/) {
+       } elsif ($opt_c =~ /^([0-9]+)M$/) {
                $crit_type = "K";
                $crit = $1 * 1024;
-       } elsif ($crit =~ /^([0-9]+)G$/) {
+       } elsif ($opt_c =~ /^([0-9]+)G$/) {
                $crit_type = "K";
                $crit = $1 * 1048576;
        }
@@ -157,41 +181,42 @@ if (/\s*(\d*) blocks of size (\d*)\. (\d*) blocks available/) {
                $avail = int($avail / 1024);
                if (int($avail /1024) > 0) {
                        $avail = (int(($avail / 1024)*100))/100;
-                       $avail = $avail."G";
+                       $avail = $avail ."G";
                } else {
-                       $avail = $avail."M";
+                       $avail = $avail ."M";
                }
        } else {
-               $avail = $avail."K";
+               $avail = $avail ."K";
        }
 
 #print ":$warn:$warn_type:\n";
 #print ":$crit:$crit_type:\n";
 #print ":$avail:$avail_bytes:$capper:$mountpt:\n";
+
        if ((($warn_type eq "P") && (100 - $capper) < $warn) || (($warn_type eq "K") && ($avail_bytes > $warn))) { 
                $answer = "Disk ok - $avail ($capper%) free on $mountpt\n";
        } elsif ((($crit_type eq "P") && (100 - $capper) < $crit) || (($crit_type eq "K") && ($avail_bytes > $crit))) {
                $state = "WARNING";
-               $answer = "Only $avail ($capper%) free on $mountpt\n";
+               $answer = "WARNING: Only $avail ($capper%) free on $mountpt\n";
        } else {
                $state = "CRITICAL";
-               $answer = "Only $avail ($capper%) free on $mountpt\n";
+               $answer = "CRITICAL: Only $avail ($capper%) free on $mountpt\n";
        }
 } else {
        $answer = "Result from smbclient not suitable\n";
        $state = "UNKNOWN";
        foreach (@lines) {
-               if (/Access denied/) {
+               if (/(Access denied|NT_STATUS_LOGON_FAILURE)/) {
                        $answer = "Access Denied\n";
                        $state = "CRITICAL";
                        last;
                }
-               if (/(Unknown host \w*)/) {
+               if (/(Unknown host \w*|Connection.*failed)/) {
                        $answer = "$1\n";
                        $state = "CRITICAL";
                        last;
                }
-               if (/(You specified an invalid share name)/) {
+               if (/(You specified an invalid share name|NT_STATUS_BAD_NETWORK_NAME)/) {
                        $answer = "Invalid share name \\\\$host\\$share\n";
                        $state = "CRITICAL";
                        last;
@@ -206,7 +231,7 @@ exit $ERRORS{$state};
 
 sub print_usage () {
        print "Usage: $PROGNAME -H <host> -s <share> -u <user> -p <password> 
-      -w <warn> -c <crit> [-W <workgroup>]\n";
+      -w <warn> -c <crit> [-W <workgroup>] [-P <port>]\n";
 }
 
 sub print_help () {
@@ -228,12 +253,19 @@ Perl Check SMB Disk plugin for Nagios
    Username to log in to server. (Defaults to \"guest\")
 -p, --password=STRING
    Password to log in to server. (Defaults to \"guest\")
--w, --warning=INTEGER
+-w, --warning=INTEGER or INTEGER[kMG]
    Percent of used space at which a warning will be generated (Default: 85%)
-   
--c, --critical=INTEGER
+      
+-c, --critical=INTEGER or INTEGER[kMG]
    Percent of used space at which a critical will be generated (Defaults: 95%)
+-P, --port=INTEGER
+   Port to be used to connect to. Some Windows boxes use 139, others 445 (Defaults to smbclient default)
    
+   If thresholds are followed by either a k, M, or G then check to see if that
+   much disk space is available (kilobytes, Megabytes, Gigabytes)
+
+   Warning percentage should be less than critical
+   Warning (remaining) disk space should be greater than critical.
 
 ";
        support();