add pending reboot and fix scvmm

apps/hyperv/2012/local/mode/scvmmintegrationservice.pm

@@ -81,7 +81,7 @@ sub set_counters {
         { name => 'vm', type => 1, cb_prefix_output => 'prefix_vm_output', message_multiple => 'All integration services are ok' },
     ];
     $self->{maps_counters}->{vm} = [
-        { label => 'snapshot', set => {
+        { label => 'status', , threshold => 0, set => {
                 key_values => [ { name => 'vm' }, { name => 'status' }, { name => 'vmaddition' }, 
                     { name => 'operatingsystemshutdownenabled' }, { name => 'timesynchronizationenabled' }, 
                     { name => 'dataexchangeenabled' }, { name => 'heartbeatenabled' }, { name => 'backupenabled' } ],

centreon/common/powershell/windows/pendingreboot.pm

@@ -0,0 +1,140 @@
+#
+# Copyright 2017 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::pendingreboot;
+
+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"
+    
+    $ComputerName = "$env:COMPUTERNAME"
+    $CompPendRen,$PendFileRename,$Pending,$SCCM = $false,$false,$false,$false
+    
+    ## Setting CBSRebootPend to null since not all versions of Windows has this value 
+    $CBSRebootPend = $null 
+             
+    ## Querying WMI for build version 
+    $WMI_OS = Get-WmiObject -Class Win32_OperatingSystem -Property BuildNumber, CSName -ComputerName $ComputerName -ErrorAction Stop 
+ 
+    ## Making registry connection to the local/remote computer 
+    $HKLM = [UInt32] "0x80000002" 
+    $WMI_Reg = [WMIClass] "\\$ComputerName\root\default:StdRegProv" 
+
+    ## If Vista/2008 & Above query the CBS Reg Key 
+    If ([Int32]$WMI_OS.BuildNumber -ge 6001) { 
+        $RegSubKeysCBS = $WMI_Reg.EnumKey($HKLM,"SOFTWARE\Microsoft\Windows\CurrentVersion\Component Based Servicing\") 
+        $CBSRebootPend = $RegSubKeysCBS.sNames -contains "RebootPending"     
+    } 
+               
+    ## Query WUAU from the registry 
+    $RegWUAURebootReq = $WMI_Reg.EnumKey($HKLM,"SOFTWARE\Microsoft\Windows\CurrentVersion\WindowsUpdate\Auto Update\") 
+    $WUAURebootReq = $RegWUAURebootReq.sNames -contains "RebootRequired" 
+             
+    ## Query PendingFileRenameOperations from the registry 
+    $RegSubKeySM = $WMI_Reg.GetMultiStringValue($HKLM,"SYSTEM\CurrentControlSet\Control\Session Manager\","PendingFileRenameOperations") 
+    $RegValuePFRO = $RegSubKeySM.sValue 
+ 
+    ## Query ComputerName and ActiveComputerName from the registry 
+    $ActCompNm = $WMI_Reg.GetStringValue($HKLM,"SYSTEM\CurrentControlSet\Control\ComputerName\ActiveComputerName\","ComputerName")       
+    $CompNm = $WMI_Reg.GetStringValue($HKLM,"SYSTEM\CurrentControlSet\Control\ComputerName\ComputerName\","ComputerName") 
+    If ($ActCompNm -ne $CompNm) { 
+        $CompPendRen = $true 
+    } 
+             
+    ## If PendingFileRenameOperations has a value set $RegValuePFRO variable to $true 
+    If ($RegValuePFRO) { 
+        $PendFileRename = $true 
+    }
+    
+    ## Determine SCCM 2012 Client Reboot Pending Status 
+    ## To avoid nested "if" statements and unneeded WMI calls to determine if the CCM_ClientUtilities class exist, setting EA = 0 
+    $CCMClientSDK = $null 
+    $CCMSplat = @{ 
+        NameSpace="ROOT\ccm\ClientSDK" 
+        Class="CCM_ClientUtilities" 
+        Name="DetermineIfRebootPending" 
+        ComputerName=$ComputerName 
+        ErrorAction="Stop" 
+    } 
+    ## Try CCMClientSDK 
+    Try { 
+        $CCMClientSDK = Invoke-WmiMethod @CCMSplat 
+    } Catch [System.UnauthorizedAccessException] { 
+        $CcmStatus = Get-Service -Name CcmExec -ComputerName $ComputerName -ErrorAction SilentlyContinue 
+        If ($CcmStatus.Status -ne "Running") { 
+            Write-Warning "$ComputerName`: Error - CcmExec service is not running." 
+            $CCMClientSDK = $null 
+        } 
+    } Catch { 
+        $CCMClientSDK = $null 
+    } 
+ 
+    If ($CCMClientSDK) { 
+        If ($CCMClientSDK.ReturnValue -ne 0) { 
+            Write-Warning "Error: DetermineIfRebootPending returned error code $($CCMClientSDK.ReturnValue)"     
+        } 
+        If ($CCMClientSDK.IsHardRebootPending -or $CCMClientSDK.RebootPending) { 
+            $SCCM = $true 
+        } 
+    } Else { 
+        $SCCM = $null 
+    }
+    
+    Write-Host ("[CBServicing={0}]" -f $CBSRebootPend ) -NoNewline
+    Write-Host ("[WindowsUpdate={0}]" -f $WUAURebootReq ) -NoNewline
+    Write-Host ("[CCMClientSDK={0}]" -f $SCCM ) -NoNewline
+    Write-Host ("[PendComputerRename={0}]" -f $CompPendRen ) -NoNewline
+    Write-Host ("[PendFileRename={0}]" -f $PendFileRename ) -NoNewline
+    Write-Host ("[PendFileRenVal={0}]" -f $RegValuePFRO ) -NoNewline
+    Write-Host ("[RebootPending={0}]" -f ($CompPendRen -or $CBSRebootPend -or $WUAURebootReq -or $SCCM -or $PendFileRename) )    
+} Catch {
+    Write-Host $Error[0].Exception
+    exit 1
+}
+
+exit 0
+';
+
+    return centreon::plugins::misc::powershell_encoded($ps);
+}
+
+1;
+
+__END__
+
+=head1 DESCRIPTION
+
+Method to get pending reboot informations.
+
+=cut

os/windows/local/mode/pendingreboot.pm

@@ -0,0 +1,203 @@
+#
+# Copyright 2017 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::pendingreboot;
+
+use base qw(centreon::plugins::templates::counter);
+
+use strict;
+use warnings;
+use centreon::plugins::misc;
+use centreon::common::powershell::windows::pendingreboot
+
+my $instance_mode;
+
+sub custom_status_threshold {
+    my ($self, %options) = @_; 
+    my $status = 'ok';
+    my $message;
+    
+    eval {
+        local $SIG{__WARN__} = sub { $message = $_[0]; };
+        local $SIG{__DIE__} = sub { $message = $_[0]; };
+        
+        if (defined($instance_mode->{option_results}->{critical_status}) && $instance_mode->{option_results}->{critical_status} ne '' &&
+            eval "$instance_mode->{option_results}->{critical_status}") {
+            $status = 'critical';
+        } elsif (defined($instance_mode->{option_results}->{warning_status}) && $instance_mode->{option_results}->{warning_status} ne '' &&
+                 eval "$instance_mode->{option_results}->{warning_status}") {
+            $status = 'warning';
+        }
+    };
+    if (defined($message)) {
+        $self->{output}->output_add(long_msg => 'filter status issue: ' . $message);
+    }
+
+    return $status;
+}
+
+sub custom_status_output {
+    my ($self, %options) = @_;
+    
+    my $msg = 'Reboot Pending : ' . $self->{result_values}->{RebootPending};
+    return $msg;
+}
+
+sub custom_status_calc {
+    my ($self, %options) = @_;
+    
+    $self->{result_values}->{CBServicing} = $options{new_datas}->{$self->{instance} . '_CBServicing'};
+    $self->{result_values}->{RebootPending} = $options{new_datas}->{$self->{instance} . '_RebootPending'};
+    $self->{result_values}->{WindowsUpdate} = $options{new_datas}->{$self->{instance} . '_WindowsUpdate'};
+    $self->{result_values}->{CCMClientSDK} = $options{new_datas}->{$self->{instance} . '_CCMClientSDK'};
+    return 0;
+}
+
+sub set_counters {
+    my ($self, %options) = @_;
+    
+    $self->{maps_counters_type} = [
+        { name => 'vm', type => 1,  },
+    ];
+    $self->{maps_counters}->{vm} = [
+        { label => 'status', , threshold => 0, set => {
+                key_values => [ { name => 'CBServicing' }, { name => 'RebootPending' }, { name => 'WindowsUpdate' }, 
+                    { name => 'CCMClientSDK' } ],
+                closure_custom_calc => $self->can('custom_status_calc'),
+                closure_custom_output => $self->can('custom_status_output'),
+                closure_custom_perfdata => sub { return 0; },
+                closure_custom_threshold_check => $self->can('custom_status_threshold'),
+            }
+        },
+    ];
+}
+
+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 =>
+                                {
+                                  "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' },
+                                  "no-ps"               => { name => 'no_ps' },
+                                  "ps-exec-only"        => { name => 'ps_exec_only' },
+                                  "warning-status:s"    => { name => 'warning_status', default => '%{RebootPending} =~ /true/i' },
+                                  "critical-status:s"   => { name => 'critical_status', default => '' },
+                                });
+    return $self;
+}
+
+sub check_options {
+    my ($self, %options) = @_;
+    $self->SUPER::check_options(%options);
+    
+    $instance_mode = $self;
+    $self->change_macros();
+}
+
+sub change_macros {
+    my ($self, %options) = @_;
+    
+    foreach (('warning_status', 'critical_status')) {
+        if (defined($self->{option_results}->{$_})) {
+            $self->{option_results}->{$_} =~ s/%\{(.*?)\}/\$self->{result_values}->{$1}/g;
+        }
+    }
+}
+
+sub manage_selection {
+    my ($self, %options) = @_;
+    
+    my $ps = centreon::common::powershell::windows::pendingreboot::get_powershell(
+        no_ps => $self->{option_results}->{no_ps});
+    
+    $self->{option_results}->{command_options} .= " " . $ps;
+    my ($stdout) = centreon::plugins::misc::execute(output => $self->{output},
+                                                    options => $self->{option_results},
+                                                    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);
+        $self->{output}->display(nolabel => 1, force_ignore_perfdata => 1, force_long_output => 1);
+        $self->{output}->exit();
+    }
+    
+    #[CBServicing=False][WindowsUpdate=False][CCMClientSDK=][PendComputerRename=False][PendFileRename=False][PendFileRenVal=][RebootPending=False]
+    $self->{pendingreboot} = {};
+    while ($line =~ /\[(.*?)=\s*(.*?)\s*\]/mg) {
+        $self->{pendingreboot}->{$1} = $2;
+    }
+}
+
+1;
+
+__END__
+
+=head1 MODE
+
+Check windows pending reboot.
+
+=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<--warning-status>
+
+Set warning threshold for status (Default: '%{RebootPending} =~ /true/i').
+Can used special variables like: %{RebootPending}, %{CBServicing}, %{WindowsUpdate}, %{CCMClientSDK}.
+
+=item B<--critical-status>
+
+Set critical threshold for status (Default: '').
+Can used special variables like: %{RebootPending}, %{CBServicing}, %{WindowsUpdate}, %{CCMClientSDK}.
+
+=back
+
+=cut

os/windows/local/plugin.pm

@@ -31,6 +31,7 @@ sub new {
 
     $self->{version} = '0.1';
     %{$self->{modes}} = (
+                         'pending-reboot'   => 'os::windows::local::mode::pendingreboot',
                          'sessions'         => 'os::windows::local::mode::sessions',
                          'time'             => 'os::windows::local::mode::ntp',
                          );