From e0bfe7863b4cfcf82aab8d6f6fc3fed982144171 Mon Sep 17 00:00:00 2001 From: Fabien THEPAUT Date: Wed, 10 Jan 2018 10:39:37 +0100 Subject: [PATCH] Add list-storage mode for windows NSClient --- .../common/powershell/windows/liststorages.pm | 113 ++++++++++++++ os/windows/local/mode/liststorages.pm | 143 ++++++++++++++++++ os/windows/local/plugin.pm | 1 + 3 files changed, 257 insertions(+) create mode 100644 centreon/common/powershell/windows/liststorages.pm create mode 100644 os/windows/local/mode/liststorages.pm diff --git a/centreon/common/powershell/windows/liststorages.pm b/centreon/common/powershell/windows/liststorages.pm new file mode 100644 index 000000000..7c58474a7 --- /dev/null +++ b/centreon/common/powershell/windows/liststorages.pm @@ -0,0 +1,113 @@ +# +# Copyright 2018 Centreon (http://www.centreon.com/) +# +# Centreon is a full-fledged industry-strength solution that meets +# the needs in IT infrastructure and application monitoring for +# service performance. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# + +package centreon::common::powershell::windows::liststorages; + +use strict; +use warnings; +use centreon::plugins::misc; + +sub get_powershell { + my (%options) = @_; + my $no_ps = (defined($options{no_ps})) ? 1 : 0; + + return '' if ($no_ps == 1); + + my $ps = ' +$culture = new-object "System.Globalization.CultureInfo" "en-us" +[System.Threading.Thread]::CurrentThread.CurrentUICulture = $culture +$ProgressPreference = "SilentlyContinue" + +Try { + $ErrorActionPreference = "Stop" + '; + + if (defined($options{filter})) { + $ps .= ' + $disks = Get-PSDrive -PSProvider "' . $options{filter} . '" +'; + } else { + $ps .= ' + $disks = Get-PSDrive +'; + } + + $ps .= ' +} Catch { + Write-Host $Error[0].Exception + exit 1 +}Foreach ($disk in $disks) { + Write-Host "[name=" $disk.Name "][used=" $disk.Used "][free=" $disk.Free "][provider=" $disk.Provider "][path=" $disk.Root "]" -NoNewline +} + +exit 0 +'; + + return centreon::plugins::misc::powershell_encoded($ps); +} +1; + +sub list { + my ($self, %options) = @_; + + # Following output: + #[name= c ][server= SRVI-WIN-TEST ][used= 20394858038 ][free= 509408308 ][provider= Microsoft.PowerShell.Core\FileSystem ][path= C:\ ] + #... + foreach my $line (split /\n/, $options{stdout}) { + next if ($line !~ /^\[name=(.*?)\]\[used=(.*?)\]\[free=(.*?)\]\[provider=(.*?)\]\[path=(.*?)\]/); + my ($disk, $used, $free, $provider, $path) = (centreon::plugins::misc::trim($1), centreon::plugins::misc::trim($2), + centreon::plugins::misc::trim($3), centreon::plugins::misc::trim($4), centreon::plugins::misc::trim($5)); + + $self->{output}->output_add(long_msg => "'" . $disk . "' [used = $used, free = $free, path = $path, provider = " . $provider . ']'); + + } +} +1; + + +sub disco_show { + my ($self, %options) = @_; + + # Following output: + #[name= c ][server= SRVI-WIN-TEST ][used= 20394858038 ][free= 509408308 ][provider= Microsoft.PowerShell.Core\FileSystem ][path= C:\ ] + #... + foreach my $line (split /\n/, $options{stdout}) { + next if ($line !~ /^\[name=(.*?)\]\[used=(.*?)\]\[free=(.*?)\]\[provider=(.*?)\]\[path=(.*?)\]/); + my ($disk, $used, $free, $provider, $path) = (centreon::plugins::misc::trim($1), centreon::plugins::misc::trim($2), + centreon::plugins::misc::trim($3), centreon::plugins::misc::trim($4), centreon::plugins::misc::trim($5)); + + $self->{output}->add_disco_entry(name => $disk, + used => $used, + free => $free, + provider => $provider, + path => $path); + } +} + +1; + +__END__ + +=head1 DESCRIPTION + +Method to list Windows Disks. + +=cut + diff --git a/os/windows/local/mode/liststorages.pm b/os/windows/local/mode/liststorages.pm new file mode 100644 index 000000000..fe628cdb6 --- /dev/null +++ b/os/windows/local/mode/liststorages.pm @@ -0,0 +1,143 @@ +# +# Copyright 2018 Centreon (http://www.centreon.com/) +# +# Centreon is a full-fledged industry-strength solution that meets +# the needs in IT infrastructure and application monitoring for +# service performance. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# + +package os::windows::local::mode::liststorages; + +use base qw(centreon::plugins::mode); + +use strict; +use warnings; +use centreon::plugins::misc; +use centreon::common::powershell::windows::liststorages; + +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 => + { + + "no-ps" => { name => 'no_ps', }, + "timeout:s" => { name => 'timeout', default => 50 }, + "command:s" => { name => 'command', default => 'powershell.exe' }, + "command-path:s" => { name => 'command_path' }, + "command-options:s" => { name => 'command_options', default => '-InputFormat none -NoLogo -EncodedCommand' }, + "ps-exec-only" => { name => 'ps_exec_only', }, + "filter-type:s" => { name => 'filter', default => 'FileSystem'}, + }); + return $self; +} + +sub check_options { + my ($self, %options) = @_; + $self->SUPER::init(%options); +} + +sub run { + my ($self, %options) = @_; + + my $ps = centreon::common::powershell::windows::liststorages::get_powershell(no_ps => $self->{option_results}->{no_ps}, + filter => $self->{option_results}->{filter}); + + $self->{option_results}->{command_options} .= " " . $ps; + + my ($stdout) = centreon::plugins::misc::windows_execute(output => $self->{output}, + timeout => $self->{option_results}->{timeout}, + command => $self->{option_results}->{command}, + command_path => $self->{option_results}->{command_path}, + command_options => $self->{option_results}->{command_options}); + if (defined($self->{option_results}->{ps_exec_only})) { + $self->{output}->output_add(severity => 'OK', + short_msg => $stdout); + } else { + $self->{output}->output_add(severity => 'OK', + short_msg => 'List disk:'); + centreon::common::powershell::windows::liststorages::list($self, stdout => $stdout); + } + + $self->{output}->display(nolabel => 1, force_ignore_perfdata => 1, force_long_output => 1); + $self->{output}->exit(); +} + +sub disco_format { + my ($self, %options) = @_; + + $self->{output}->add_disco_format(elements => ['name', 'used', 'free', 'provider', 'path']); +} + +sub disco_show { + my ($self, %options) = @_; + + my $ps = centreon::common::powershell::windows::liststorages::get_powershell(no_ps => $self->{option_results}->{no_ps}, + filter => $self->{option_results}->{filter}); + + $self->{option_results}->{command_options} .= " " . $ps; + my ($stdout) = centreon::plugins::misc::windows_execute(output => $self->{output}, + timeout => $self->{option_results}->{timeout}, + command => $self->{option_results}->{command}, + command_path => $self->{option_results}->{command_path}, + command_options => $self->{option_results}->{command_options}); + centreon::common::powershell::windows::liststorages::disco_show($self, stdout => $stdout); +} + +1; + +__END__ + +=head1 MODE + +List Windows disks. + +=over 8 + +=item B<--timeout> + +Set timeout time for command execution (Default: 50 sec) + +=item B<--no-ps> + +Don't encode powershell. To be used with --command and 'type' command. + +=item B<--command> + +Command to get information (Default: 'powershell.exe'). +Can be changed if you have output in a file. To be used with --no-ps option!!! + +=item B<--command-path> + +Command path (Default: none). + +=item B<--command-options> + +Command options (Default: '-InputFormat none -NoLogo -EncodedCommand'). + +=item B<--ps-exec-only> + +Print powershell output. + +=item B<--filter-type> + +Filter database (only wilcard 'FileSystem' can be used. In Powershell). + +=back + +=cut \ No newline at end of file diff --git a/os/windows/local/plugin.pm b/os/windows/local/plugin.pm index 0d428c699..c04aca28a 100644 --- a/os/windows/local/plugin.pm +++ b/os/windows/local/plugin.pm @@ -35,6 +35,7 @@ sub new { 'pending-reboot' => 'os::windows::local::mode::pendingreboot', 'sessions' => 'os::windows::local::mode::sessions', 'time' => 'os::windows::local::mode::ntp', + 'list-storages' => 'os::windows::local::mode::liststorages', ); return $self;