Code

Fix Debian bug #478906: Failure when run via ePN
[nagiosplug.git] / plugins-scripts / check_disk_smb.pl
index 8c275c8ce25f95409d3cd7fb0674204812e66ce2..ca593d468480bb9384324914d530cef495339992 100755 (executable)
 #  drives is a pain in the butt
 # 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 $opt_a $verbose);
 use vars qw($PROGNAME);
 use lib utils.pm ;
 use utils qw($TIMEOUT %ERRORS &print_revision &support &usage);
@@ -40,6 +39,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,
@@ -48,62 +48,103 @@ GetOptions
         "u=s" => \$opt_u, "username=s" => \$opt_u,
         "s=s" => \$opt_s, "share=s"    => \$opt_s,
         "W=s" => \$opt_W, "workgroup=s" => \$opt_W,
-        "H=s" => \$opt_H, "hostname=s" => \$opt_H);
+        "H=s" => \$opt_H, "hostname=s" => \$opt_H,
+        "a=s" => \$opt_a, "address=s" => \$opt_a);
 
 if ($opt_V) {
-       print_revision($PROGNAME,'$Revision$'); #'
+       print_revision($PROGNAME,'@NP_VERSION@'); #'
        exit $ERRORS{'OK'};
 }
 
 if ($opt_h) {print_help(); exit $ERRORS{'OK'};}
 
 my $smbclient= "$utils::PATH_TO_SMBCLIENT " ;
-my $smbclientoptions="";
+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]+\$?)/);
+($opt_H) || ($opt_H = shift @ARGV) || usage("Host name not specified\n");
+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]+)/);
+($opt_s) || ($opt_s = shift @ARGV) || usage("Share volume not specified\n");
+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]+)/);
+($opt_u) || ($opt_u = shift @ARGV) || ($opt_u = "guest");
+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");
+($opt_p) || ($opt_p = shift @ARGV) || ($opt_p = "");
 my $pass = $1 if ($opt_p =~ /(.*)/);
+$pass = "-N" if ($opt_p eq "");
 
-($opt_w) || ($opt_w = shift) || ($opt_w = 85);
-my $warn = $1 if ($opt_w =~ /([0-9]{1,2}\%?|100\%?|[0-9]+[kMG])/);
+($opt_w) || ($opt_w = shift @ARGV) || ($opt_w = 85);
+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]+[kMG])/);
+($opt_c) || ($opt_c = shift @ARGV) || ($opt_c = 95);
+my $crit = $1 if ($opt_c =~ /^([0-9]{1,2}\%?|100\%?|[0-9]+[kMG])$/);
 ($crit) || usage("Invalid critical threshold: $opt_c\n");
 
+# split the type from the unit value
+#Check $warn and $crit for type (%/M/G) and set up for tests
+#P = Percent, K = KBytes
+my $warn_type;
+my $crit_type;
+
+if ($opt_w =~ /^([0-9]+)\%?$/) {
+       $warn = "$1";
+       $warn_type = "P";
+} elsif ($opt_w =~ /^([0-9]+)k$/) {
+       $warn_type = "K";
+       $warn = $1;
+} elsif ($opt_w =~ /^([0-9]+)M$/) {
+       $warn_type = "K";
+       $warn = $1 * 1024;
+} elsif ($opt_w =~ /^([0-9]+)G$/) {
+       $warn_type = "K";
+       $warn = $1 * 1048576;
+}
+if ($opt_c =~ /^([0-9]+)\%?$/) {
+       $crit = "$1";
+       $crit_type = "P";
+} elsif ($opt_c =~ /^([0-9]+)k$/) {
+       $crit_type = "K";
+       $crit = $1;
+} elsif ($opt_c =~ /^([0-9]+)M$/) {
+       $crit_type = "K";
+       $crit = $1 * 1024;
+} elsif ($opt_c =~ /^([0-9]+)G$/) {
+       $crit_type = "K";
+       $crit = $1 * 1048576;
+}
+
 # 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]/) )  ){
+unless( ( $warn_type eq "P" && $crit_type eq "P" ) || ( $warn_type ne "P" && $crit_type ne "P" ) ){
+       $opt_w =~ s/\%/\%\%/g;
+       $opt_c =~ s/\%/\%\%/g;
        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]/) {
+if ( $warn_type eq "K") {
        unless ( $warn > $crit) {
                usage("Disk size: warning ($opt_w) should be greater than critical ($opt_c) \n");
        }
 }else{
        unless ( $warn < $crit) {
+               $opt_w =~ s/\%/\%\%/g;
+               $opt_c =~ s/\%/\%\%/g;
                usage("Percentage: warning ($opt_w) should be less than critical ($opt_c) \n");
        }
 }
 
 my $workgroup = $1 if (defined($opt_W) && $opt_W =~ /(.*)/);
 
+my $address = $1 if (defined($opt_a) && $opt_a =~ /(.*)/);
+
 # end of options checking
 
 
@@ -122,10 +163,21 @@ alarm($TIMEOUT);
 # Execute an "ls" on the share using smbclient program
 # get the results into $res
 if (defined($workgroup)) {
-       $res = qx/$smbclient \/\/$host\/$share $pass -W $workgroup -U $user $smbclientoptions -c ls/;
+       if (defined($address)) {
+               print "$smbclient " . "\/\/$host\/$share" ." $pass -W $workgroup -U $user $smbclientoptions -I $address -c ls\n" if ($verbose);
+               $res = qx/$smbclient "\/\/$host\/$share" $pass -W $workgroup -U $user $smbclientoptions -I $address -c ls/;
+       } else {
+               print "$smbclient " . "\/\/$host\/$share" ." $pass -W $workgroup -U $user $smbclientoptions -c ls\n" if ($verbose);
+               $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/;
+       if (defined($address)) {
+               print "$smbclient " . "\/\/$host\/$share" ." $pass -U $user $smbclientoptions -I $address -c ls\n" if ($verbose);
+               $res = qx/$smbclient "\/\/$host\/$share" $pass -U $user $smbclientoptions -I $address -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
 alarm(0);
@@ -146,35 +198,6 @@ if (/\s*(\d*) blocks of size (\d*)\. (\d*) blocks available/) {
        my ($capper) = int(($3/$1)*100);
        my ($mountpt) = "\\\\$host\\$share";
 
-       #Check $warn and $crit for type (%/M/G) and set up for tests
-       #P = Percent, K = KBytes
-       my $warn_type;
-       my $crit_type;
-
-       if ($opt_w =~ /^([0-9]+$)/) {
-               $warn_type = "P";
-       } elsif ($opt_w =~ /^([0-9]+)k$/) {
-               $warn_type = "K";
-               $warn = $1;
-       } elsif ($opt_w =~ /^([0-9]+)M$/) {
-               $warn_type = "K";
-               $warn = $1 * 1024;
-       } elsif ($opt_w =~ /^([0-9]+)G$/) {
-               $warn_type = "K";
-               $warn = $1 * 1048576;
-       }
-       if ($opt_c =~ /^([0-9]+$)/) {
-               $crit_type = "P";
-       } elsif ($opt_c =~ /^([0-9]+)k$/) {
-               $crit_type = "K";
-               $crit = $1;
-       } elsif ($opt_c =~ /^([0-9]+)M$/) {
-               $crit_type = "K";
-               $crit = $1 * 1024;
-       } elsif ($opt_c =~ /^([0-9]+)G$/) {
-               $crit_type = "K";
-               $crit = $1 * 1048576;
-       }
 
        if (int($avail / 1024) > 0) {
                $avail = int($avail / 1024);
@@ -205,18 +228,17 @@ if (/\s*(\d*) blocks of size (\d*)\. (\d*) blocks available/) {
        $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*)/) {
-                       $answer = "$1\n";_
-
+               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;
@@ -231,11 +253,11 @@ 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>] [-a <IP>]\n";
 }
 
 sub print_help () {
-       print_revision($PROGNAME,'$Revision$');
+       print_revision($PROGNAME,'@NP_VERSION@');
        print "Copyright (c) 2000 Michael Anthon/Karl DeBisschop
 
 Perl Check SMB Disk plugin for Nagios
@@ -249,15 +271,19 @@ Perl Check SMB Disk plugin for Nagios
    Share name to be tested
 -W, --workgroup=STRING
    Workgroup or Domain used (Defaults to \"WORKGROUP\")
+-a, --address=IP
+   IP-address of HOST (only necessary if HOST is in another network)
 -u, --user=STRING
    Username to log in to server. (Defaults to \"guest\")
 -p, --password=STRING
-   Password to log in to server. (Defaults to \"guest\")
+   Password to log in to server. (Defaults to an empty password)
 -w, --warning=INTEGER or INTEGER[kMG]
    Percent of used space at which a warning will be generated (Default: 85%)
       
 -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)