Initial Chef repository

cookbooks/windows/files/default/handlers/windows_reboot_handler.rb

@@ -0,0 +1,76 @@
+#
+# Author:: Seth Chisamore (<schisamo@chef.io>)
+# Copyright:: Copyright (c) 2011 Chef Software, Inc
+# License:: Apache License, Version 2.0
+#
+# 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.
+#
+
+class WindowsRebootHandler < Chef::Handler
+  include Chef::Mixin::ShellOut
+
+  def initialize(allow_pending_reboots = true, timeout = 60, reason = "Chef Software Chef initiated reboot")
+    @allow_pending_reboots = allow_pending_reboots
+    @timeout = timeout
+    @reason = reason
+  end
+
+  def report
+    log_message, reboot = begin
+      if reboot_requested?
+        ["chef_handler[#{self.class}] requested reboot will occur in #{timeout} seconds", true]
+      elsif reboot_pending?
+        if @allow_pending_reboots
+          ["chef_handler[#{self.class}] reboot pending - automatic reboot will occur in #{timeout} seconds", true]
+        else
+          ["chef_handler[#{self.class}] reboot pending but handler not configured to act on pending reboots - please reboot node manually", false]
+        end
+      else
+        ["chef_handler[#{self.class}] no reboot requested or pending", false]
+      end
+    end
+
+    Chef::Log.warn(log_message)
+    shell_out!("shutdown /r /t #{timeout} /c \"#{reason}\"") if reboot
+  end
+
+  private
+  # reboot cause CHEF says so:
+  # reboot explicitly requested in our cookbook code
+  def reboot_requested?
+    node.run_state[:reboot_requested] == true
+  end
+
+  # reboot cause WIN says so:
+  # reboot pending because of some configuration action we performed
+  def reboot_pending?
+    # Any files listed here means reboot needed
+    (Registry.key_exists?('HKLM\SYSTEM\CurrentControlSet\Control\Session Manager\PendingFileRenameOperations') &&
+      Registry.get_value('HKLM\SYSTEM\CurrentControlSet\Control\Session Manager','PendingFileRenameOperations').any?) ||
+    # 1 for any value means reboot pending
+    # "9306cdfc-c4a1-4a22-9996-848cb67eddc3"=1
+    (Registry.key_exists?('HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\WindowsUpdate\Auto Update\RebootRequired') &&
+      Registry.get_values('HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\WindowsUpdate\Auto Update\RebootRequired').select{|v| v[2] == 1 }.any?) ||
+    # 1 or 2 for 'Flags' value means reboot pending
+    (Registry.key_exists?('HKLM\SOFTWARE\Microsoft\Updates\UpdateExeVolatile') &&
+      [1,2].include?(Registry::get_value('HKLM\SOFTWARE\Microsoft\Updates\UpdateExeVolatile','Flags')))
+  end
+
+  def timeout
+    node.run_state[:reboot_timeout] || node['windows']['reboot_timeout'] || @timeout
+  end
+
+  def reason
+    node.run_state[:reboot_reason] || @reason
+  end
+end