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

Refs #5196

This commit is contained in:
Quentin Garnier 2014-02-12 18:16:27 +01:00
parent 6f2f880db0
commit bfe6c6181c
1 changed files with 70 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
centreon/plugins/wsman.pm
View File

@ -201,6 +201,15 @@ sub execute_winshell_commands {
$node = $data->root()->add($namespace, 'DesiredStream', 'stdout stderr') $node = $data->root()->add($namespace, 'DesiredStream', 'stdout stderr')
or $self->internal_exit(msg => 'Could not create XmlDoc'); or $self->internal_exit(msg => 'Could not create XmlDoc');
$node->attr_add(undef, 'CommandId', $command_id); $node->attr_add(undef, 'CommandId', $command_id);
my $timeout_global = 30; #seconds
my $wait_timeout_done = 0;
my $loop_out = 1;
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', $result = $self->{client}->invoke($client_options, $uri, 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/Receive',
$data); $data);
return undef if ($self->handle_dialog_fault(result => $result, msg => 'Invoke failed: ', dont_quit => $dont_quit)); return undef if ($self->handle_dialog_fault(result => $result, msg => 'Invoke failed: ', dont_quit => $dont_quit));
@ -226,12 +235,10 @@ sub execute_winshell_commands {
next if (!defined($output) || $output eq ''); next if (!defined($output) || $output eq '');
if ($stream_type eq 'stderr') { if ($stream_type eq 'stderr') {
$command_result->{$command->{label}}->{stderr} = '' if (!defined($command_result->{$command->{label}}->{stderr})); $current_stderr .= $output;
$command_result->{$command->{label}}->{stderr} .= $output;
} }
if ($stream_type eq 'stdout') { if ($stream_type eq 'stdout') {
$command_result->{$command->{label}}->{stdout} = '' if (!defined($command_result->{$command->{label}}->{stdout})); $current_stdout .= $output;
$command_result->{$command->{label}}->{stdout} .= $output;
} }
} }
if ($node->name() eq 'CommandState') { if ($node->name() eq 'CommandState') {
@ -247,12 +254,15 @@ sub execute_winshell_commands {
} else { } else {
$self->internal_exit(msg => "No exit code for 'done' command"); $self->internal_exit(msg => "No exit code for 'done' command");
} }
$loop_out = 0;
} elsif ($state eq 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/CommandState/Running') { } elsif ($state eq 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/CommandState/Running') {
# mmmm should put an error # we wait
return undef if ($self->handle_dialog_fault(result => $result, msg => 'Command still running...', dont_quit => $dont_quit)); $wait_timeout_done += 3;
sleep 3;
} elsif ($state eq 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/CommandState/Pending') { } elsif ($state eq 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/CommandState/Pending') {
# no-op # no-op
# WinRM 1.1 sends this with ExitCode:0 # WinRM 1.1 sends this with ExitCode:0
$loop_out = 0;
} else { } else {
# unknown # unknown
$self->internal_exit(msg => 'Unknown command state: ' . $state); $self->internal_exit(msg => 'Unknown command state: ' . $state);
@ -261,6 +271,14 @@ sub execute_winshell_commands {
} }
}
$current_stderr =~ s/\r//mg;
$current_stdout =~ s/\r//mg;
$command_result->{$command->{label}}->{stderr} = $current_stderr if ($current_stderr ne '');
$command_result->{$command->{label}}->{stdout} = $current_stdout if ($current_stdout ne '');
# #
# terminate shell command # terminate shell command
# #