Fix #5187
Can execute remote windows commands Need to think to how manage errors (maybe something by commands with an exit_internal_error and msg if you want to manage it in the plugin)
This commit is contained in:
Quentin Garnier
parent
730f3be5ea
commit
34289513f2
|
|
@ -38,6 +38,7 @@ package centreon::plugins::wsman;
|
||||||
use strict;
|
use strict;
|
||||||
use warnings;
|
use warnings;
|
||||||
use openwsman;
|
use openwsman;
|
||||||
|
use MIME::Base64;
|
||||||
|
|
||||||
my %auth_method_map = (
|
my %auth_method_map = (
|
||||||
noauth => $openwsman::NO_AUTH_STR,
|
noauth => $openwsman::NO_AUTH_STR,
|
||||||
|
|
@ -112,7 +113,8 @@ sub connect {
|
||||||
|
|
||||||
if ($self->{wsman_params}->{wsman_scheme} eq 'https') {
|
if ($self->{wsman_params}->{wsman_scheme} eq 'https') {
|
||||||
# Dont verify
|
# Dont verify
|
||||||
$client->transport()->set_verify_peer(0);
|
$self->{client}->transport()->set_verify_peer(0);
|
||||||
|
$self->{client}->transport()->set_verify_host(0);
|
||||||
}
|
}
|
||||||
|
|
||||||
$self->{client}->transport()->set_auth_method($auth_method_map{$self->{wsman_params}->{wsman_auth_method}});
|
$self->{client}->transport()->set_auth_method($auth_method_map{$self->{wsman_params}->{wsman_auth_method}});
|
||||||
|
|
@ -125,6 +127,164 @@ sub connect {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
sub execute_winshell_commands {
|
||||||
|
my ($self, %options) = @_;
|
||||||
|
# $options{commands} = ref array of hash ([{label => 'myipconfig', value => 'ipconfig /all' }])
|
||||||
|
|
||||||
|
my ($dont_quit) = (defined($options{dont_quit}) && $options{dont_quit} == 1) ? 1 : 0;
|
||||||
|
$self->set_error();
|
||||||
|
|
||||||
|
my $uri = 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/cmd';
|
||||||
|
my $namespace = 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell';
|
||||||
|
$self->{shell_id} = undef;
|
||||||
|
my ($command_id);
|
||||||
|
my $command_result = {};
|
||||||
|
|
||||||
|
# Init result
|
||||||
|
foreach my $command (@{$options{commands}}) {
|
||||||
|
$command_result->{$command->{label}} = {stdout => undef, stderr => undef, exit_code => undef};
|
||||||
|
}
|
||||||
|
|
||||||
|
######
|
||||||
|
# Start Shell
|
||||||
|
my $client_options = new openwsman::ClientOptions::()
|
||||||
|
or $self->internal_exit(msg => 'Could not create client options handler');
|
||||||
|
$client_options->set_timeout(30 * 1000); # 30sec
|
||||||
|
$client_options->add_selector('Name', 'Themes');
|
||||||
|
$client_options->add_option('WINRS_NOPROFILE', 'FALSE');
|
||||||
|
$client_options->add_option('WINRS_CODEPAGE', '437'); # utf-8
|
||||||
|
|
||||||
|
my $data = new openwsman::XmlDoc::('Shell', $namespace)
|
||||||
|
or $self->internal_exit(msg => 'Could not create XmlDoc');
|
||||||
|
$data->root()->add($namespace, 'InputStreams', 'stdin');
|
||||||
|
$data->root()->add($namespace, 'OutputStreams', 'stdout stderr');
|
||||||
|
|
||||||
|
my $result = $self->{client}->create($client_options, $uri, $data->string(), length($data->string()), "utf-8");
|
||||||
|
return undef if ($self->handle_dialog_fault(result => $result, msg => 'Create failed: ', dont_quit => $dont_quit));
|
||||||
|
my $node = $result->root()->find(undef, 'Selector')
|
||||||
|
or $self->internal_exit(msg => 'No shell id returned');
|
||||||
|
$self->{shell_id} = $node->text();
|
||||||
|
|
||||||
|
foreach my $command (@{$options{commands}}) {
|
||||||
|
#######
|
||||||
|
# Issue command
|
||||||
|
$client_options = new openwsman::ClientOptions::()
|
||||||
|
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');
|
||||||
|
$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);
|
||||||
|
return undef if ($self->handle_dialog_fault(result => $result, msg => 'Invoke failed: ', dont_quit => $dont_quit));
|
||||||
|
|
||||||
|
$node = $result->root()->find(undef, 'CommandId')
|
||||||
|
or $self->internal_exit(msg => 'No command id returned');
|
||||||
|
$command_id = $node->text();
|
||||||
|
|
||||||
|
#######
|
||||||
|
# Request stdout/stderr
|
||||||
|
$client_options = new openwsman::ClientOptions::()
|
||||||
|
or $self->internal_exit(msg => 'Could not create client options handler');
|
||||||
|
$client_options->set_timeout(30 * 1000); # 30sec
|
||||||
|
$client_options->add_selector('ShellId', $self->{shell_id});
|
||||||
|
|
||||||
|
$data = new openwsman::XmlDoc::('Receive', $namespace);
|
||||||
|
$node = $data->root()->add($namespace, 'DesiredStream', 'stdout stderr')
|
||||||
|
or $self->internal_exit(msg => 'Could not create XmlDoc');
|
||||||
|
$node->attr_add(undef, 'CommandId', $command_id);
|
||||||
|
$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');
|
||||||
|
######
|
||||||
|
# Parsing Reponse
|
||||||
|
for (my $cnt = 0; $cnt < $response->size(); $cnt++) {
|
||||||
|
my $node = $response->get($cnt);
|
||||||
|
my $node_command = $node->attr_find(undef, 'CommandId')
|
||||||
|
or $self->internal_exit(msg => 'No CommandId in ReceiveResponse');
|
||||||
|
if ($node_command->value() ne $command_id) {
|
||||||
|
$self->internal_exit(msg => 'Wrong CommandId in ReceiveResponse node');
|
||||||
|
}
|
||||||
|
|
||||||
|
if ($node->name() eq 'Stream') {
|
||||||
|
my $node_tmp = $node->attr_find(undef, 'Name');
|
||||||
|
if (!defined($node_tmp)) {
|
||||||
|
next;
|
||||||
|
}
|
||||||
|
my $stream_type = $node_tmp->value();
|
||||||
|
my $output = decode_base64($node->text());
|
||||||
|
next if (!defined($output) || $output eq '');
|
||||||
|
|
||||||
|
if ($stream_type eq 'stderr') {
|
||||||
|
$command_result->{$command->{label}}->{stderr} = '' if (!defined($command_result->{$command->{label}}->{stderr}));
|
||||||
|
$command_result->{$command->{label}}->{stderr} .= $output;
|
||||||
|
}
|
||||||
|
if ($stream_type eq 'stdout') {
|
||||||
|
$command_result->{$command->{label}}->{stdout} = '' if (!defined($command_result->{$command->{label}}->{stdout}));
|
||||||
|
$command_result->{$command->{label}}->{stdout} .= $output;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if ($node->name() eq 'CommandState') {
|
||||||
|
my $node_tmp = $node->attr_find(undef, 'State');
|
||||||
|
if (!defined($node_tmp)) {
|
||||||
|
next;
|
||||||
|
}
|
||||||
|
my $state = $node_tmp->value();
|
||||||
|
if ($state eq 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/CommandState/Done') {
|
||||||
|
my $exit_code = $node->find(undef, 'ExitCode');
|
||||||
|
if (defined($exit_code)) {
|
||||||
|
$command_result->{$command->{label}}->{exit_code} = $exit_code->text();
|
||||||
|
} else {
|
||||||
|
$self->internal_exit(msg => "No exit code for 'done' command");
|
||||||
|
}
|
||||||
|
} elsif ($state eq 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/CommandState/Running') {
|
||||||
|
# mmmm should put an error
|
||||||
|
return undef if ($self->handle_dialog_fault(result => $result, msg => 'Command still running...', dont_quit => $dont_quit));
|
||||||
|
} elsif ($state eq 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/CommandState/Pending') {
|
||||||
|
# no-op
|
||||||
|
# WinRM 1.1 sends this with ExitCode:0
|
||||||
|
} else {
|
||||||
|
# unknown
|
||||||
|
$self->internal_exit(msg => 'Unknown command state: ' . $state);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
#
|
||||||
|
# terminate shell command
|
||||||
|
#
|
||||||
|
# not strictly needed for WinRM 2.0, but WinRM 1.1 requires this
|
||||||
|
#
|
||||||
|
$data = new openwsman::XmlDoc::('Signal', $namespace)
|
||||||
|
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);
|
||||||
|
return undef if ($self->handle_dialog_fault(result => $result, msg => 'Invoke failed: ', dont_quit => $dont_quit));
|
||||||
|
}
|
||||||
|
|
||||||
|
# Delete Shell resource
|
||||||
|
if (defined($self->{shell_id})) {
|
||||||
|
my $client_options = new openwsman::ClientOptions::()
|
||||||
|
or die print "[ERROR] Could not create client options handler.\n";
|
||||||
|
$client_options->set_timeout(30 * 1000); # 30sec
|
||||||
|
$client_options->add_selector('ShellId', $self->{shell_id});
|
||||||
|
$self->{shell_id} = undef;
|
||||||
|
$result = $self->{client}->invoke($client_options, $uri, 'http://schemas.xmlsoap.org/ws/2004/09/transfer/Delete', undef);
|
||||||
|
return undef if ($self->handle_dialog_fault(result => $result, msg => 'Invoke failed: ', dont_quit => $dont_quit));
|
||||||
|
}
|
||||||
|
|
||||||
|
return $command_result;
|
||||||
|
}
|
||||||
|
|
||||||
sub request {
|
sub request {
|
||||||
my ($self, %options) = @_;
|
my ($self, %options) = @_;
|
||||||
# $options{nothing_quit} = integer
|
# $options{nothing_quit} = integer
|
||||||
|
|
@ -327,7 +487,7 @@ wsman class
|
||||||
|
|
||||||
=head1 WSMAN OPTIONS
|
=head1 WSMAN OPTIONS
|
||||||
|
|
||||||
Need at least openwsman-perl version >= 2.3.0
|
Need at least openwsman-perl version >= 2.4.0
|
||||||
|
|
||||||
=over 8
|
=over 8
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue