#!/usr/bin/perl

#

# check_rsync  -  Check Rsync and modules availability

# Version: 1.02

#

# Copyright (C) 2006-2008 Thomas Guyot-Sionnest <tguyot@gmail.com>

#

# 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

#

​

use POSIX;

use strict;

use Getopt::Long;

​

use vars qw($opt_H $opt_p $opt_m);

use vars qw($PROGNAME %RSYNCMSG $cpid);

use lib "/usr/local/nagios/libexec";

use utils qw($TIMEOUT %ERRORS);

​

$PROGNAME = "check_rsync";

$ENV{'PATH'}='';

$ENV{'BASH_ENV'}='';

$ENV{'ENV'}='';

%RSYNCMSG = (

  '0' => 'Success',

  '1' => 'Syntax or usage error',

  '2' => 'Protocol incompatibility',

  '3' => 'Errors selecting input/output files, dirs',

  '4' => 'Requested action not supported',

  '5' => 'Error starting client-server protocol',

  '6' => 'Daemon unable to append to log-file',

  '10' => 'Error in socket I/O',

  '11' => 'Error in file I/O',

  '12' => 'Error in rsync protocol data stream',

  '13' => 'Errors with program diagnostics',

  '14' => 'Error in IPC code',

  '20' => 'Received SIGUSR1 or SIGINT',

  '21' => 'Some error returned by waitpid()',

  '22' => 'Error allocating core memory buffers',

  '23' => 'Partial transfer due to error',

  '24' => 'Partial transfer due to vanished source files',

  '25' => 'The --max-delete limit stopped deletions',

  '30' => 'Timeout in data send/receive',

);

$cpid = 0;

​

Getopt::Long::Configure('bundling');

GetOptions (

  "H=s" => \$opt_H, "hostname=s" => \$opt_H,

  "p=s" => \$opt_p, "port=s" => \$opt_p,

  "m=s@" => \$opt_m, "module=s@" => \$opt_m );

​

unless (defined($opt_H)){

  print "Usage: $PROGNAME -H <host> [-p <port>] [-m <module>[,<user>,<password>] [-m <module>[,<user>,<password>]...]]\n";

  exit $ERRORS{'UNKNOWN'};

}

​

my $host = $opt_H;

my $port = defined($opt_p) ? $opt_p : 873;

my $verbose = 0; # Not implemented as argument yet

​

# Create an array for each -m arguments and store them in @modules

my @modules;

if (defined($opt_m)) {

  for(@$opt_m) {

    my @tmpmod = split(/,/);

    my ($modname, $username) = @tmpmod;

    my $pass = join(',', splice(@tmpmod, 2));

    print STDERR "Adding module $modname\n" if ($verbose);

    if ($pass) {

      print STDERR "Module $modname using authentication ($username, $pass)\n" if ($verbose > 2);

      push @modules, [ $modname, $username, $pass ];

      } else {

      push @modules, [ $modname ];

    }

  }

}

​

# Just in case of problems, let's not hang Nagios

$SIG{'ALRM'} = sub {

  print "CRITICAL: Rsync timed out\n";

  kill SIGKILL, $cpid if ($cpid);

  exit $ERRORS{"CRITICAL"};

};

​

# Rsync arguments

my $source = "rsync://$host";

​

alarm($TIMEOUT);

​

# Get a list of modules to see if rsync is up

my $command = "/usr/bin/rsync --port=$port $source";

​

# Workaround to kill stale rsync processes

$cpid = open(RSYNC, "$command|") or report_error("Unable to execute rsync: $!");

my $result;

{

  local $/;

  $result = <RSYNC>;

}

close(RSYNC);

my $error_code = $?;

​

#Turn off alarm

alarm(0);

$cpid = 0;

​

my $realerr = $error_code >> 8;

report_error("Rsync command $command failed with error " . $realerr . ": " . (defined $RSYNCMSG{"$realerr"} ? $RSYNCMSG{"$realerr"} : "Unknown error")) if ($realerr != 0);

​

# If one or more -m, check if these modules exists first...

if (@modules) {

​

  my @result = split(/\n/, $result);

​

  foreach my $mod (@modules) {

    my $match = 0;

    for (@result) {

      $match = 1 if (/^$$mod[0]\s/);

    }

    report_error("Module $$mod[0] not found") if ($match == 0);

  }

} else { # else just return OK

  print "OK: Rsync is up\n";

  exit $ERRORS{'OK'};

}

​

# Check each -m aruments...

for my $arg (@modules) {

  if (defined($$arg[1]) and defined($$arg[2])) {

    $source = "rsync://$$arg[1]" . '@' . "$host/$$arg[0]";

    $ENV{'RSYNC_PASSWORD'} = $$arg[2];

  } else {

    $source = "rsync://$host/$$arg[0]";

    $ENV{'RSYNC_PASSWORD'} = '';

  }

​

  alarm($TIMEOUT);

​

  # Better safe than sorry...

  undef $error_code;

  undef $result;

  # Get a file listing of the root of the module

  $command = "/usr/bin/rsync --port=$port $source";

​

  # Workaround to kill stale rsync processes

  $cpid = open(RSYNC, "$command|") or report_error("Unable to execute rsync: $!");

  {

    local $/;

    $result = <RSYNC>;

  }

  close(RSYNC);

  $error_code = $?;

​

  #Turn off alarm

  alarm(0);

  $cpid = 0;

​

  $realerr = $error_code >> 8;

  report_error("Rsync command failed on module $$arg[0] with error " . $realerr . ": " . (defined $RSYNCMSG{$realerr} ? $RSYNCMSG{$realerr} : "Unknown error")) if ($realerr != 0);

}

​

if (@modules > 0) {

  print "OK: Rsync is up with ", scalar(@modules), " module tested\n" if (@modules == 1);

  print "OK: Rsync is up with ", scalar(@modules), " modules tested\n" if (@modules > 1);

  exit $ERRORS{'OK'};

} else { # We hould never end up here :)

  print "UNKNOWN: The unexpected occured (bug?)\n";

  exit $ERRORS{'UNKNOWN'};

}

​

# Report error passed as one string, print rsync messages to STDERR

sub report_error {

  my $report = shift;

  print "CRITICAL: $report\n";

  exit $ERRORS{'CRITICAL'};

}

​

​