tyler.durden
/
centreon-plugins
mirror of https://github.com/centreon/centreon-plugins.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix #5812

This commit is contained in:
Quentin Garnier 2014-09-12 15:58:02 +02:00
parent 2479fe890c
commit d18a19fdd0
1 changed files with 64 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

103
centreon-plugins/apps/protocols/ftp/mode/filescount.pm
View File

@ -40,6 +40,7 @@ use base qw(centreon::plugins::mode);
use strict; use strict;
use warnings; use warnings;
use apps::protocols::ftp::lib::ftp; use apps::protocols::ftp::lib::ftp;
use File::Basename;
# How much arguments i need and commands manages # How much arguments i need and commands manages
my %map_commands = ( my %map_commands = (
@ -60,12 +61,13 @@ sub new {
"ssl" => { name => 'use_ssl' }, "ssl" => { name => 'use_ssl' },
"ftp-options:s@" => { name => 'ftp_options' }, "ftp-options:s@" => { name => 'ftp_options' },
"directory:s@" => { name => 'directory' }, "directory:s@" => { name => 'directory' },
"recursive" => { name => 'recursive' }, "max-depth:s" => { name => 'max_depth', default => 0 },
"username:s" => { name => 'username' }, "username:s" => { name => 'username' },
"password:s" => { name => 'password' }, "password:s" => { name => 'password' },
"warning:s" => { name => 'warning' }, "warning:s" => { name => 'warning' },
"critical:s" => { name => 'critical' }, "critical:s" => { name => 'critical' },
"timeout:s" => { name => 'timeout', default => '30' }, "filter-file:s" => { name => 'filter_file' },
"timeout:s" => { name => 'timeout', default => '30' },
}); });
return $self; return $self;
} }
@ -88,7 +90,7 @@ sub check_options {
} }
$self->{ssl_or_not} = 'nossl'; $self->{ssl_or_not} = 'nossl';
if (defined($self->{option_results}->{use_ssl})) { if (defined($self->{option_results}->{use_ssl})) {
$self->{ssl_or_not} = 'ssl'; $self->{ssl_or_not} = 'ssl';
} }
} }
@ -98,26 +100,17 @@ sub run {
my @files; my @files;
my @array; my @array;
apps::protocols::ftp::lib::ftp::connect($self); apps::protocols::ftp::lib::ftp::connect($self);
my ($ref_array, $globalCount, $flag) = $self->countFiles(@{$self->{option_results}->{directory}}); my $count = $self->countFiles();
@array = @$ref_array;
if (defined($self->{option_results}->{recursive})) {
while ($flag == 1) {
($ref_array, $cpt, $flag) = $self->countFiles(@array);
$globalCount = $globalCount + $cpt;
@array = @$ref_array;
}
}
apps::protocols::ftp::lib::ftp::quit(); apps::protocols::ftp::lib::ftp::quit();
my $exit_code = $self->{perfdata}->threshold_check(value => $globalCount, my $exit_code = $self->{perfdata}->threshold_check(value => $count,
threshold => [ { label => 'critical', exit_litteral => 'critical' }, { label => 'warning', exit_litteral => 'warning' } ]); threshold => [ { label => 'critical', exit_litteral => 'critical' }, { label => 'warning', exit_litteral => 'warning' } ]);
$self->{output}->output_add(severity => $exit_code, $self->{output}->output_add(severity => $exit_code,
short_msg => sprintf("Number of files : %s", $globalCount)); short_msg => sprintf("Number of files : %s", $count));
$self->{output}->perfdata_add(label => 'files', $self->{output}->perfdata_add(label => 'files',
value => $globalCount, value => $count,
warning => $self->{perfdata}->get_perfdata_for_output(label => 'warning'), warning => $self->{perfdata}->get_perfdata_for_output(label => 'warning'),
critical => $self->{perfdata}->get_perfdata_for_output(label => 'critical'), critical => $self->{perfdata}->get_perfdata_for_output(label => 'critical'),
min => 0); min => 0);
@ -126,30 +119,54 @@ sub run {
} }
sub countFiles { sub countFiles {
my ($self, @array) = @_; my ($self) = @_;
my @files; my @listings;
my @subdirs; my $count = 0;
my $size;
my $cpt = 0; if (!defined($self->{option_results}->{directory}) || scalar(@{$self->{option_results}->{directory}}) == 0) {
my $time_result; push @listings, [ { name => '.', level => 0 } ];
my $flag = 0; } else {
foreach my $dir (@{$self->{option_results}->{directory}}) {
foreach my $dir (@array) { push @listings, [ { name => $dir, level => 0 } ];
if (!(@files = apps::protocols::ftp::lib::ftp::execute($self, command => $map_commands{ls}->{$self->{ssl_or_not}}->{name}, command_args => [$dir]))) {
$flag = 0;
} }
foreach my $file (@files) {
if (!($time_result = apps::protocols::ftp::lib::ftp::execute($self, command => $map_commands{mdtm}->{$self->{ssl_or_not}}->{name}, command_args => [$file]))) {
push(@subdirs, $file);
$flag = 1;
}
$cpt++;
}
$size = @subdirs;
} }
$cpt = $cpt - $size;
return \@subdirs, $cpt, $flag; my @build_name = ();
foreach my $list (@listings) {
while (@$list) {
my @files;
my $hash = pop @$list;
my $dir = $hash->{name};
my $level = $hash->{level};
if (!(@files = apps::protocols::ftp::lib::ftp::execute($self, command => $map_commands{ls}->{$self->{ssl_or_not}}->{name}, command_args => [$dir]))) {
# Cannot list we skip
next;
}
foreach my $file (@files) {
my $name = $dir . '/' . basename($file);
if (defined($self->{option_results}->{filter_file}) && $self->{option_results}->{filter_file} ne '' &&
$name !~ /$self->{option_results}->{filter_file}/) {
$self->{output}->output_add(long_msg => sprintf("Skipping '%s'", $name));
next;
}
if (!(my $time_result = apps::protocols::ftp::lib::ftp::execute($self,
command => $map_commands{mdtm}->{$self->{ssl_or_not}}->{name},
command_args => [$name]))) {
if (defined($self->{option_results}->{max_depth}) && $level + 1 <= $self->{option_results}->{max_depth}) {
push @$list, { name => $name, level => $level + 1};
}
} else {
$self->{output}->output_add(long_msg => sprintf("Match '%s'", $name));
$count++;
}
}
}
}
return $count;
} }
1; 1;
@ -204,6 +221,14 @@ Threshold critical (number of files)
Check files in the directory (Multiple option) Check files in the directory (Multiple option)
=item B<--max-depth>
Don't check fewer levels (Default: '0'. Means current dir only).
=item B<--filter-file>
Filter files (can be a regexp. Directory is in the name).
=back =back
=cut =cut