238 lines
8.2 KiB
Perl
238 lines
8.2 KiB
Perl
###############################################################################
|
|
# Copyright 2005-2014 MERETHIS
|
|
# Centreon is developped by : Julien Mathis and Romain Le Merlus under
|
|
# GPL Licence 2.0.
|
|
#
|
|
# 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.
|
|
#
|
|
# 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, see <http://www.gnu.org/licenses>.
|
|
#
|
|
# Linking this program statically or dynamically with other modules is making a
|
|
# combined work based on this program. Thus, the terms and conditions of the GNU
|
|
# General Public License cover the whole combination.
|
|
#
|
|
# As a special exception, the copyright holders of this program give MERETHIS
|
|
# permission to link this program with independent modules to produce an timeelapsedutable,
|
|
# regardless of the license terms of these independent modules, and to copy and
|
|
# distribute the resulting timeelapsedutable under terms of MERETHIS choice, provided that
|
|
# MERETHIS also meet, for each linked independent module, the terms and conditions
|
|
# of the license of that module. An independent module is a module which is not
|
|
# derived from this program. If you modify this program, you may extend this
|
|
# exception to your version of the program, but you are not obliged to do so. If you
|
|
# do not wish to do so, delete this exception statement from your version.
|
|
#
|
|
# For more information : contact@centreon.com
|
|
# Author : Simon Bomm <sbomm@merethis.com>
|
|
#
|
|
####################################################################################
|
|
|
|
package apps::protocols::ftp::mode::filescount;
|
|
|
|
use base qw(centreon::plugins::mode);
|
|
|
|
use strict;
|
|
use warnings;
|
|
use apps::protocols::ftp::lib::ftp;
|
|
use File::Basename;
|
|
|
|
# How much arguments i need and commands manages
|
|
my %map_commands = (
|
|
ls => { ssl => { name => 'list' }, nossl => { name => 'dir'} },
|
|
);
|
|
|
|
sub new {
|
|
my ($class, %options) = @_;
|
|
my $self = $class->SUPER::new(package => __PACKAGE__, %options);
|
|
bless $self, $class;
|
|
|
|
$self->{version} = '1.0';
|
|
$options{options}->add_options(arguments =>
|
|
{
|
|
"hostname:s" => { name => 'hostname' },
|
|
"port:s" => { name => 'port', },
|
|
"ssl" => { name => 'use_ssl' },
|
|
"ftp-options:s@" => { name => 'ftp_options' },
|
|
"directory:s@" => { name => 'directory' },
|
|
"max-depth:s" => { name => 'max_depth', default => 0 },
|
|
"username:s" => { name => 'username' },
|
|
"password:s" => { name => 'password' },
|
|
"warning:s" => { name => 'warning' },
|
|
"critical:s" => { name => 'critical' },
|
|
"filter-file:s" => { name => 'filter_file' },
|
|
"timeout:s" => { name => 'timeout', default => '30' },
|
|
});
|
|
return $self;
|
|
}
|
|
|
|
sub check_options {
|
|
my ($self, %options) = @_;
|
|
$self->SUPER::init(%options);
|
|
|
|
if (($self->{perfdata}->threshold_validate(label => 'warning', value => $self->{option_results}->{warning})) == 0) {
|
|
$self->{output}->add_option_msg(short_msg => "Wrong warning threshold '" . $self->{option_results}->{warning} . "'.");
|
|
$self->{output}->option_exit();
|
|
}
|
|
if (($self->{perfdata}->threshold_validate(label => 'critical', value => $self->{option_results}->{critical})) == 0) {
|
|
$self->{output}->add_option_msg(short_msg => "Wrong critical threshold '" . $self->{option_results}->{critical} . "'.");
|
|
$self->{output}->option_exit();
|
|
}
|
|
if (!defined($self->{option_results}->{hostname})) {
|
|
$self->{output}->add_option_msg(short_msg => "Please set the hostname option");
|
|
$self->{output}->option_exit();
|
|
}
|
|
$self->{ssl_or_not} = 'nossl';
|
|
if (defined($self->{option_results}->{use_ssl})) {
|
|
$self->{ssl_or_not} = 'ssl';
|
|
}
|
|
}
|
|
|
|
sub run {
|
|
my ($self, %options) = @_;
|
|
my $cpt;
|
|
my @files;
|
|
my @array;
|
|
|
|
apps::protocols::ftp::lib::ftp::connect($self);
|
|
my $count = $self->countFiles();
|
|
apps::protocols::ftp::lib::ftp::quit();
|
|
|
|
my $exit_code = $self->{perfdata}->threshold_check(value => $count,
|
|
threshold => [ { label => 'critical', exit_litteral => 'critical' }, { label => 'warning', exit_litteral => 'warning' } ]);
|
|
|
|
$self->{output}->output_add(severity => $exit_code,
|
|
short_msg => sprintf("Number of files : %s", $count));
|
|
$self->{output}->perfdata_add(label => 'files',
|
|
value => $count,
|
|
warning => $self->{perfdata}->get_perfdata_for_output(label => 'warning'),
|
|
critical => $self->{perfdata}->get_perfdata_for_output(label => 'critical'),
|
|
min => 0);
|
|
$self->{output}->display();
|
|
$self->{output}->exit();
|
|
}
|
|
|
|
sub countFiles {
|
|
my ($self) = @_;
|
|
my @listings;
|
|
my $count = 0;
|
|
|
|
if (!defined($self->{option_results}->{directory}) || scalar(@{$self->{option_results}->{directory}}) == 0) {
|
|
push @listings, [ { name => '.', level => 0 } ];
|
|
} else {
|
|
foreach my $dir (@{$self->{option_results}->{directory}}) {
|
|
push @listings, [ { name => $dir, level => 0 } ];
|
|
}
|
|
}
|
|
|
|
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 $line (@files) {
|
|
# IIS: 05-13-15 10:59AM 1184403 test.jpg
|
|
next if ($line !~ /(\S+)\s+\S+\s+\S+\s+\S+\s+\S+\s+\S+\s+\S+\s+\S+\s+(.*)/ &&
|
|
$line !~ /^\s*\S+\s*\S+\s*(\S+)\s+(.*)/);
|
|
my ($rights, $filename) = ($1, $2);
|
|
my $bname = basename($filename);
|
|
next if ($bname eq '.' || $bname eq '..');
|
|
my $name = $dir . '/' . $bname;
|
|
|
|
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 ($rights =~ /^(d|<DIR>)/i) {
|
|
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;
|
|
|
|
__END__
|
|
|
|
=head1 MODE
|
|
|
|
Count files in an FTP directory (can be recursive).
|
|
|
|
=over 8
|
|
|
|
=item B<--hostname>
|
|
|
|
IP Addr/FQDN of the ftp host
|
|
|
|
=item B<--port>
|
|
|
|
Port used
|
|
|
|
=item B<--ssl>
|
|
|
|
Use SSL connection
|
|
Need Perl 'Net::FTPSSL' module
|
|
|
|
=item B<--ftp-options>
|
|
|
|
Add custom ftp options.
|
|
Example: --ftp-options='Debug=1" --ftp-options='useSSL=1'
|
|
|
|
=item B<--username>
|
|
|
|
Specify username for authentification
|
|
|
|
=item B<--password>
|
|
|
|
Specify password for authentification
|
|
|
|
=item B<--timeout>
|
|
|
|
Connection timeout in seconds (Default: 30)
|
|
|
|
=item B<--warning>
|
|
|
|
Threshold warning (number of files)
|
|
|
|
=item B<--critical>
|
|
|
|
Threshold critical (number of files)
|
|
|
|
=item B<--directory>
|
|
|
|
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
|
|
|
|
=cut
|