tyler.durden/centreon-plugins
1
0
Fork 0
mirror of https://github.com/centreon/centreon-plugins.git synced 2025-07-29 16:45:04 +02:00
Code Issues Packages Projects Releases Wiki Activity

add(wsman): powershell chunk management (#3456)

Browse Source
This commit is contained in:
qgarnier 2022-02-02 15:55:02 +01:00 committed by GitHub
parent 7cc0dfc5a1
commit 85ba046869
3 changed files with 294 additions and 89 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

115
centreon-plugins/centreon/plugins/wsman.pm
View File

@ -31,16 +31,13 @@ my %auth_method_map = (
digest => $openwsman::DIGEST_AUTH_STR,
pass => $openwsman::PASS_AUTH_STR,
ntlm => $openwsman::NTLM_AUTH_STR,
gssnegotiate => $openwsman::GSSNEGOTIATE_AUTH_STR,
gssnegotiate => $openwsman::GSSNEGOTIATE_AUTH_STR
);
sub new {
my ($class, %options) = @_;
my $self = {};
bless $self, $class;
# $options{options} = options object
# $options{output} = output object
# $options{exit_value} = integer
if (!defined($options{output})) {
print "Class wsman: Need to specify 'output' argument.\n";
@ -51,20 +48,20 @@ sub new {
$options{output}->option_exit();
}
$options{options}->add_options(arguments =>
{ "hostname|host:s" => { name => 'host' },
"wsman-port:s" => { name => 'wsman_port', default => 5985 },
"wsman-path:s" => { name => 'wsman_path', default => '/wsman' },
"wsman-scheme:s" => { name => 'wsman_scheme', default => 'http' },
"wsman-username:s" => { name => 'wsman_username' },
"wsman-password:s" => { name => 'wsman_password' },
"wsman-timeout:s" => { name => 'wsman_timeout', default => 30 },
"wsman-proxy-url:s" => { name => 'wsman_proxy_url', },
"wsman-proxy-username:s" => { name => 'wsman_proxy_username', },
"wsman-proxy-password:s" => { name => 'wsman_proxy_password', },
"wsman-debug" => { name => 'wsman_debug', },
"wsman-auth-method:s" => { name => 'wsman_auth_method', default => 'basic' },
"wsman-errors-exit:s" => { name => 'wsman_errors_exit', default => 'unknown' },
$options{options}->add_options(arguments => {
'hostname|host:s' => { name => 'host' },
'wsman-port:s' => { name => 'wsman_port', default => 5985 },
'wsman-path:s' => { name => 'wsman_path', default => '/wsman' },
'wsman-scheme:s' => { name => 'wsman_scheme', default => 'http' },
'wsman-username:s' => { name => 'wsman_username' },
'wsman-password:s' => { name => 'wsman_password' },
'wsman-timeout:s' => { name => 'wsman_timeout', default => 30 },
'wsman-proxy-url:s' => { name => 'wsman_proxy_url', },
'wsman-proxy-username:s' => { name => 'wsman_proxy_username', },
'wsman-proxy-password:s' => { name => 'wsman_proxy_password', },
'wsman-debug' => { name => 'wsman_debug', },
'wsman-auth-method:s' => { name => 'wsman_auth_method', default => 'basic' },
'wsman-errors-exit:s' => { name => 'wsman_errors_exit', default => 'unknown' }
});
$options{options}->add_help(package => __PACKAGE__, sections => 'WSMAN OPTIONS');
@ -88,9 +85,11 @@ sub connect {
}
openwsman::set_debug(1) if (defined($self->{wsman_params}->{wsman_debug}));
$self->{client} = new openwsman::Client::($self->{wsman_params}->{host}, $self->{wsman_params}->{wsman_port},
$self->{client} = new openwsman::Client::(
$self->{wsman_params}->{host}, $self->{wsman_params}->{wsman_port},
$self->{wsman_params}->{wsman_path}, $self->{wsman_params}->{wsman_scheme},
$self->{wsman_params}->{wsman_username}, $self->{wsman_params}->{wsman_password});
$self->{wsman_params}->{wsman_username}, $self->{wsman_params}->{wsman_password}
);
if (!defined($self->{client})) {
$self->{output}->add_option_msg(short_msg => 'Could not create client handler');
$self->{output}->option_exit(exit_litteral => $self->{wsman_errors_exit});
@ -161,14 +160,20 @@ sub execute_winshell_commands {
or $self->internal_exit(msg => 'Could not create client options handler');
$client_options->set_timeout(30 * 1000); # 30sec
$client_options->add_option('WINRS_CONSOLEMODE_STDIN', 'TRUE');
if (defined($options{skip_cmd})) {
$client_options->add_option('WINRS_SKIP_CMD_SHELL', 'TRUE');
} else {
$client_options->add_option('WINRS_SKIP_CMD_SHELL', 'FALSE');
}
$client_options->add_selector('ShellId', $self->{shell_id});
$data = new openwsman::XmlDoc::('CommandLine', $namespace)
or $self->internal_exit(msg => 'Could not create XmlDoc');
$data->root()->add($namespace, 'Command', $command->{value});
$result = $self->{client}->invoke($client_options, $uri, 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/Command',
$data);
$result = $self->{client}->invoke(
$client_options, $uri, 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/Command',
$data
);
return undef if ($self->handle_dialog_fault(result => $result, msg => 'Invoke failed: ', dont_quit => $dont_quit));
$node = $result->root()->find(undef, 'CommandId')
@ -193,10 +198,10 @@ sub execute_winshell_commands {
my ($current_stdout, $current_stderr) = ('', '');
while ($loop_out == 1 && $wait_timeout_done < $timeout_global) {
$result = $self->{client}->invoke($client_options, $uri, 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/Receive',
$data);
$result = $self->{client}->invoke(
$client_options, $uri, 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/Receive',
$data
);
return undef if ($self->handle_dialog_fault(result => $result, msg => 'Invoke failed: ', dont_quit => $dont_quit));
my $response = $result->root()->find($namespace, 'ReceiveResponse')
or $self->internal_exit(msg => 'No ReceiveResponse');
@ -253,9 +258,7 @@ sub execute_winshell_commands {
$self->internal_exit(msg => 'Unknown command state: ' . $state);
}
}
}
}
if ($loop_out == 1) {
@ -277,8 +280,10 @@ sub execute_winshell_commands {
or $self->internal_exit(msg => 'Could not create XmlDoc');
$data->root()->attr_add(undef, 'CommandId', $command_id);
$data->root()->add($namespace, 'Code', 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/signal/terminate');
$result = $self->{client}->invoke($client_options, $uri, 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/Signal',
$data);
$result = $self->{client}->invoke(
$client_options, $uri, 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/Signal',
$data
);
return undef if ($self->handle_dialog_fault(result => $result, msg => 'Invoke failed: ', dont_quit => $dont_quit));
}
@ -296,6 +301,49 @@ sub execute_winshell_commands {
return $command_result;
}
sub execute_powershell {
my ($self, %options) = @_;
my $chunk = 8000;
my $filename = $options{label} . '-' . sprintf("%08X", rand(0xFFFFFFFF)) . '.bat';
my $commands = [
{
label => 'prev1-' . $options{label},
value => 'if not exist "C:\\temp\\" mkdir C:\\temp'
},
{
label => 'prev2-' . $options{label},
value => '(echo @ECHO OFF) >> ' . $filename
},
{
label => 'prev3-' . $options{label},
value => 'echo|set /P="powershell -encodedcommand ">> ' . $filename
}
];
my $i = 0;
foreach (unpack('(A' . $chunk . ')*', $options{content})) {
push @$commands, {
label => 'chunk-' . $i . '-' . $options{label},
value => 'echo|set /P="' . $_ . '">>' . $filename
};
}
push @$commands,
{
label => $options{label},
value => $filename
},
{
label => 'del-' . $options{label},
value => 'del /f ' . $filename
};
return $self->execute_winshell_commands(
commands => $commands,
keep_open => 1
);
}
sub request {
my ($self, %options) = @_;
# $options{nothing_quit} = integer
@ -362,11 +410,9 @@ sub request {
$hash_return->{$row_return->{$options{hash_key}}} = $row_return if ($result_type eq 'hash');
}
$context = $result->context()
or last;
$context = $result->context() or last;
$result = $self->{client}->pull($client_options, $filter, $options{uri}, $context)
or last;
}
# Release context.
@ -473,8 +519,7 @@ sub error {
sub get_hostname {
my ($self) = @_;
my $host = $self->{wsman_params}->{host};
return $host;
return $self->{wsman_params}->{host};
}
sub get_port {

159
centreon-plugins/os/windows/wsman/mode/pendingreboot.pm Normal file
View File

@ -0,0 +1,159 @@
#
# Copyright 2022 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::wsman::mode::pendingreboot;
use base qw(centreon::plugins::templates::counter);
use strict;
use warnings;
use centreon::plugins::misc;
use centreon::common::powershell::windows::pendingreboot;
use centreon::plugins::templates::catalog_functions qw(catalog_status_threshold_ng);
use JSON::XS;
sub custom_status_output {
my ($self, %options) = @_;
return sprintf(
"'%s': reboot pending is %s [Windows Update: %s][Component Based Servicing: %s][SCCM Client: %s][File Rename Operations: %s][Computer Name Change: %s]",
$self->{result_values}->{WindowsVersion},
$self->{result_values}->{RebootPending},
$self->{result_values}->{WindowsUpdate},
$self->{result_values}->{CBServicing},
$self->{result_values}->{CCMClientSDK},
$self->{result_values}->{PendFileRename},
$self->{result_values}->{PendComputerRename}
);
}
sub set_counters {
my ($self, %options) = @_;
$self->{maps_counters_type} = [
{ name => 'pendingreboot', type => 0 },
];
$self->{maps_counters}->{pendingreboot} = [
{ label => 'status', type => 2, warning_status => '%{RebootPending} =~ /true/i', set => {
key_values => [
{ name => 'WindowsVersion' }, { name => 'CBServicing' }, { name => 'RebootPending' }, { name => 'WindowsUpdate' },
{ name => 'CCMClientSDK' }, { name => 'PendComputerRename' }, { name => 'PendFileRename' }
],
closure_custom_output => $self->can('custom_status_output'),
closure_custom_perfdata => sub { return 0; },
closure_custom_threshold_check => \&catalog_status_threshold_ng
}
}
];
}
sub new {
my ($class, %options) = @_;
my $self = $class->SUPER::new(package => __PACKAGE__, %options);
bless $self, $class;
$options{options}->add_options(arguments => {
'ps-exec-only' => { name => 'ps_exec_only' },
'ps-display' => { name => 'ps_display' }
});
return $self;
}
sub manage_selection {
my ($self, %options) = @_;
my $ps = centreon::common::powershell::windows::pendingreboot::get_powershell();
if (defined($self->{option_results}->{ps_display})) {
$self->{output}->output_add(
severity => 'OK',
short_msg => $ps
);
$self->{output}->display(nolabel => 1, force_ignore_perfdata => 1, force_long_output => 1);
$self->{output}->exit();
}
my $result = $options{wsman}->execute_powershell(
label => 'pendingreboot',
content => centreon::plugins::misc::powershell_encoded($ps)
);
if (defined($self->{option_results}->{ps_exec_only})) {
$self->{output}->output_add(
severity => 'OK',
short_msg => $result->{pendingreboot}->{stdout}
);
$self->{output}->display(nolabel => 1, force_ignore_perfdata => 1, force_long_output => 1);
$self->{output}->exit();
}
my $decoded;
eval {
$decoded = JSON::XS->new->decode($self->{output}->decode($result->{pendingreboot}->{stdout}));
};
if ($@) {
$self->{output}->add_option_msg(short_msg => "Cannot decode json response: $@");
$self->{output}->option_exit();
}
#{ WindowsVersion: "Microsoft Windows 2003 Server", CBServicing: false, WindowsUpdate: false, CCMClientSDK: null, PendComputerRename: false, PendFileRename: false, PendFileRenVal: null, RebootPending: false }
foreach (keys %$decoded) {
$decoded->{$_} = '-' if (!defined($decoded->{$_}));
$decoded->{$_} = 'true' if ($decoded->{$_} =~ /^(?:true|1)$/i);
$decoded->{$_} = 'false' if ($decoded->{$_} =~ /^(?:false|0)$/i);
}
$self->{pendingreboot} = $decoded;
}
1;
__END__
=head1 MODE
Check windows pending reboot.
=over 8
=item B<--ps-display>
Display powershell script.
=item B<--ps-exec-only>
Print powershell output.
=item B<--warning-status>
Set warning threshold for status (Default: '%{RebootPending} =~ /true/i').
Can used special variables like: %{RebootPending}, %{WindowsUpdate}, %{CBServicing}, %{CCMClientSDK},
%{PendFileRename}, %{PendComputerRename}.
=item B<--critical-status>
Set critical threshold for status (Default: '').
Can used special variables like: %{RebootPending}, %{WindowsUpdate}, %{CBServicing}, %{CCMClientSDK},
%{PendFileRename}, %{PendComputerRename}.
=back
=cut

1
centreon-plugins/os/windows/wsman/plugin.pm
View File

@ -40,6 +40,7 @@ sub new {
'list-storages' => 'os::windows::wsman::mode::liststorages',
'memory' => 'os::windows::wsman::mode::memory',
'pages' => 'os::windows::wsman::mode::pages',
'pending-reboot' => 'os::windows::wsman::mode::pendingreboot',
'processes' => 'os::windows::wsman::mode::processes',
'services' => 'os::windows::wsman::mode::services',
'storages' => 'os::windows::wsman::mode::storages',