vagrant-windows-update 0.0.3

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 54ea485d5e4758a19ea3a8504daf7f1e5ff17d02
+  data.tar.gz: e17fa78130d2eee869788409b26217595e9692f4
+SHA512:
+  metadata.gz: e90107938bd2ad31cdee9e9bcf85f48af5e352e6d246221ca822168654bf775bcdffb256966b3347fac7d4bfe21071bdb80e66287da7bcf73d2b6834244b8e22
+  data.tar.gz: 6cb0b04df5ad00fe868c3871e7e90db1e686fe3ff5cc22bb8491bed49b308d6631d7eb95eb6b4b3f70c4a8be211d8cd1b2323d100f73164256a27bd04d6aba57

data/lib/vagrant-windows-update.rb

@@ -0,0 +1,94 @@
+begin
+  require "vagrant"
+rescue LoadError
+  raise "The Vagrant Windows Update plugin must be run within Vagrant."
+end
+
+if Vagrant::VERSION < "1.9.0"
+  raise "The Vagrant Windows Update plugin is only compatible with Vagrant 1.9+"
+end
+
+module VagrantPlugins
+  module WindowsUpdate
+    class Plugin < Vagrant.plugin("2")
+      name "Windows Update"
+      description "Vagrant plugin to update a Windows VM as a provisioning step."
+
+      provisioner "windows-update" do
+        class Provisioner < Vagrant.plugin("2", :provisioner)
+          def initialize(machine, config)
+            super
+          end
+
+          def configure(root_config)
+          end
+
+          # see https://github.com/mitchellh/vagrant/blob/master/lib/vagrant/ui.rb
+          # see https://github.com/mitchellh/vagrant/blob/master/lib/vagrant/plugin/v2/provisioner.rb
+          # see https://github.com/mitchellh/vagrant/blob/master/lib/vagrant/plugin/v2/communicator.rb
+          # see https://github.com/mitchellh/vagrant/blob/master/plugins/provisioners/shell/provisioner.rb
+          def provision
+            remote_path = "C:/Windows/Temp/vagrant-windows-update.ps1"
+            @machine.communicate.upload(
+              File.join(File.dirname(__FILE__), "vagrant-windows-update", "windows-update.ps1"),
+              remote_path)
+            command = "PowerShell -ExecutionPolicy Bypass -OutputFormat Text -File #{remote_path}"
+            loop do
+              begin
+                until @machine.communicate.ready?
+                  sleep 10
+                end
+
+                reboot_required = false
+                @machine.communicate.sudo(command, {elevated: true, interactive: false}) do |type, data|
+                  reboot_required = true if type == :stdout && data.start_with?("Rebooting...")
+                  handle_comm(type, data)
+                end
+                break unless reboot_required
+
+                # NB we have to set an absurd high halt timeout to make sure the machine is
+                #    not turned off before the Windows updates are installed.
+                original_graceful_halt_timeout = @machine.config.vm.graceful_halt_timeout
+                @machine.config.vm.graceful_halt_timeout = 4*60*60 # 4h.
+                begin
+                  options = {}
+                  options[:provision_ignore_sentinel] = false
+                  @machine.action(:reload, options)
+                ensure
+                  @machine.config.vm.graceful_halt_timeout = original_graceful_halt_timeout
+                end
+              rescue => e
+                @machine.ui.warn("Ignoring error, hoping it is transient: #{e.class} #{e} at:\n\t#{e.backtrace.join("\n\t")}")
+                next
+              end
+            end
+          end
+
+          def cleanup
+          end
+
+          protected
+
+          # This handles outputting the communication data back to the UI
+          def handle_comm(type, data)
+            if [:stderr, :stdout].include?(type)
+              # Output the data with the proper color based on the stream.
+              color = type == :stdout ? :green : :red
+
+              # Clear out the newline since we add one
+              data = data.chomp
+              return if data.empty?
+
+              options = {}
+              options[:color] = color if !config.keep_color
+
+              @machine.ui.info(data.chomp, options)
+            end
+          end
+        end
+
+        Provisioner
+      end
+    end
+  end
+end

data/lib/vagrant-windows-update/version.rb

@@ -0,0 +1,5 @@
+module VagrantPlugins
+  module WindowsUpdate
+    VERSION = "0.0.3"
+  end
+end

data/lib/vagrant-windows-update/windows-update.ps1

@@ -0,0 +1,140 @@
+# see Using the Windows Update Agent API | Searching, Downloading, and Installing Updates
+#     at https://msdn.microsoft.com/en-us/library/windows/desktop/aa387102(v=vs.85).aspx
+# see ISystemInformation interface
+#     at https://msdn.microsoft.com/en-us/library/windows/desktop/aa386095(v=vs.85).aspx
+# see IUpdateSession interface
+#     at https://msdn.microsoft.com/en-us/library/windows/desktop/aa386854(v=vs.85).aspx
+# see IUpdateSearcher interface
+#     at https://msdn.microsoft.com/en-us/library/windows/desktop/aa386515(v=vs.85).aspx
+# see IUpdateDownloader interface
+#     at https://msdn.microsoft.com/en-us/library/windows/desktop/aa386131(v=vs.85).aspx
+# see IUpdateCollection interface
+#     at https://msdn.microsoft.com/en-us/library/windows/desktop/aa386107(v=vs.85).aspx
+# see IUpdate interface
+#     at https://msdn.microsoft.com/en-us/library/windows/desktop/aa386099(v=vs.85).aspx
+
+Set-StrictMode -Version Latest
+
+$ErrorActionPreference = 'Stop'
+
+trap {
+    Write-Output "ERROR: $_"
+    Write-Output (($_.ScriptStackTrace -split '\r?\n') -replace '^(.*)$','ERROR: $1')
+    Exit 1
+}
+
+Add-Type @'
+using System;
+using System.Runtime.InteropServices;
+
+public static class Windows
+{
+    [DllImport("kernel32", SetLastError=true)]
+    public static extern UInt64 GetTickCount64();
+
+    public static TimeSpan GetUptime()
+    {
+        return TimeSpan.FromMilliseconds(GetTickCount64());
+    }
+}
+'@
+
+function Wait-Condition {
+    param(
+      [scriptblock]$Condition,
+      [int]$DebounceSeconds=15
+    )
+    process {
+        $begin = [Windows]::GetUptime()
+        do {
+            Start-Sleep -Seconds 1
+            try {
+              $result = &$Condition
+            } catch {
+              $result = $false
+            }
+            if (-not $result) {
+                $begin = [Windows]::GetUptime()
+                continue
+            }
+        } while ((([Windows]::GetUptime()) - $begin).TotalSeconds -lt $DebounceSeconds)
+    }
+}
+
+function ExitWhenRebootRequired {
+    # check for pending Windows Updates.
+    $systemInformation = New-Object -ComObject 'Microsoft.Update.SystemInfo'
+    $rebootRequired = $systemInformation.RebootRequired
+
+    # check for pending Windows Features.
+    if (!$rebootRequired) {
+        $pendingPackagesKey = 'HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Component Based Servicing\PackagesPending'
+        $pendingPackagesCount = (Get-ChildItem -ErrorAction SilentlyContinue $pendingPackagesKey | Measure-Object).Count
+        $rebootRequired = $rebootRequired -or $pendingPackagesCount -gt 0
+    }
+
+    if ($rebootRequired) {
+        Write-Output 'Pending Reboot detected. Waiting for the Windows Modules Installer to exit...'
+        Wait-Condition {(Get-Process -ErrorAction SilentlyContinue TiWorker,TrustedInstaller | Measure-Object).Count -eq 0}
+        Write-Output 'Rebooting...'
+        Exit 0
+    }
+}
+
+ExitWhenRebootRequired
+
+$updateSession = New-Object -ComObject 'Microsoft.Update.Session'
+$updateSession.ClientApplicationID = 'vagrant-windows-update'
+
+Write-Output 'Searching for Windows updates...'
+$updateSearcher = $updateSession.CreateUpdateSearcher()
+$updatesToDownload = New-Object -ComObject 'Microsoft.Update.UpdateColl'
+$searchResult = $updateSearcher.Search("IsInstalled=0 and Type='Software' and IsHidden=0")
+if ($searchResult.Updates.Count) {
+    for ($i = 0; $i -lt $searchResult.Updates.Count; ++$i) {
+        $update = $searchResult.Updates.Item($i)
+
+        if ($update.InstallationBehavior.CanRequestUserInput) {
+            continue
+        }
+
+        $updateDate = $update.LastDeploymentChangeTime.ToString('yyyy-MM-dd')
+        $updateSize = ($update.MaxDownloadSize/1024/1024).ToString('0.##')
+        Write-Output "Found Windows update ($updateDate; $updateSize MB): $($update.Title)"
+
+        $update.AcceptEula() | Out-Null
+
+        if ($update.IsDownloaded) {
+            continue
+        }
+
+        $updatesToDownload.Add($update) | Out-Null
+    }
+
+    if ($updatesToDownload.Count) {
+        Write-Output 'Downloading Windows updates...'
+        $updateDownloader = $updateSession.CreateUpdateDownloader()
+        $updateDownloader.Updates = $updatesToDownload
+        $updateDownloader.Download() | Out-Null
+    }
+
+    $updatesToInstall = New-Object -ComObject 'Microsoft.Update.UpdateColl'
+    for ($i = 0; $i -lt $searchResult.Updates.Count; ++$i) {
+        $update = $searchResult.Updates.Item($i)
+        if ($update.IsDownloaded) {
+            $updatesToInstall.Add($update) | Out-Null
+        }
+    }
+
+    if ($updatesToInstall.Count) {
+        Write-Output 'Installing Windows updates...'
+        $updateInstaller = $updateSession.CreateUpdateInstaller()
+        $updateInstaller.Updates = $updatesToInstall
+        $installResult = $updateInstaller.Install()
+        if ($installResult.RebootRequired) {
+            ExitWhenRebootRequired
+        }
+    }
+} else {
+    Write-Output 'No Windows updates found'
+}

metadata

@@ -0,0 +1,60 @@
+--- !ruby/object:Gem::Specification
+name: vagrant-windows-update
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+platform: ruby
+authors:
+- Rui Lopes
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-04-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Vagrant plugin for installing Windows updates.
+email: rgl@ruilopes.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/vagrant-windows-update.rb
+- lib/vagrant-windows-update/version.rb
+- lib/vagrant-windows-update/windows-update.ps1
+homepage: https://github.com/rgl/vagrant-windows-update
+licenses:
+- LGPLv3
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.2
+signing_key: 
+specification_version: 4
+summary: Vagrant plugin for installing Windows updates.
+test_files: []