dsc/ext/PsGallery/xWindowsUpdate.2.7.0.0/DscResources/MSFT_xWindowsUpdateAgent/MSFT_xWindowsUpdateAgent.psm1
|
Set-StrictMode -Version latest $script:WuaSearchString = 'IsAssigned=1 and IsHidden=0 and IsInstalled=0' function Get-WuaServiceManager { return (New-Object -ComObject Microsoft.Update.ServiceManager) } function Add-WuaService { param( [Parameter(Mandatory=$true)] [string] $ServiceId, [int] $Flags = 7, [string] $AuthorizationCabPath = [string]::Empty ) $wuaServiceManager = Get-WuaServiceManager $wuaServiceManager.AddService2($ServiceId, $Flags, $AuthorizationCabPath) } function Remove-WuaService { param( [Parameter(Mandatory=$true)] [string] $ServiceId ) $wuaServiceManager = Get-WuaServiceManager $wuaServiceManager.RemoveService($ServiceId) } function Get-WuaSearchString { param( [switch] $security, [switch] $optional, [switch] $important ) $securityCategoryId = "'0FA1201D-4330-4FA8-8AE9-B877473B6441'" <# # invalid, would not install anything - not security and not optional and not important #> # security and optional and important # not security and optional and important if($optional -and $important) { # Installing everything not hidden and not already installed return 'IsHidden=0 and IsInstalled=0' } # security and optional and not important elseif ($security -and $optional) { # or can only be used at the top most boolean expression return "(IsAssigned=0 and IsHidden=0 and IsInstalled=0) or (CategoryIds contains $securityCategoryId and IsHidden=0 and IsInstalled=0)" } # security and not optional and important elseif($security -and $important ){ # Installing everything not hidden, # not optional (optional are not assigned) and not already installed return 'IsAssigned=1 and IsHidden=0 and IsInstalled=0' } elseif ($optional -and $important) { # Installing everything not hidden, # not optional (optional are not assigned) and not already installed return 'IsHidden=0 and IsInstalled=0' } # security and not optional and not important elseif ($security) { # Installing everything that is security and not hidden, # and not already installed return "CategoryIds contains $securityCategoryId and IsHidden=0 and IsInstalled=0" } # not security and not optional and important elseif ($important) { # Installing everything that is not hidden, # is assigned (not optional) and not already installed # not valid cannot do not contains or a boolean not # Note important updates will include security updates return "IsAssigned=1 and IsHidden=0 and IsInstalled=0" } # not security and optional and not important elseif ($optional) { # Installing everything that is not hidden, # is not assigned (is optional) and not already installed # not valid cannot do not contains or a boolean not # Note optional updates may include security updates return "IsAssigned=0 and IsHidden=0 and IsInstalled=0" } return "CategoryIds contains $securityCategoryId and IsHidden=0 and IsInstalled=0" } function Get-WuaAu { return (New-Object -ComObject 'Microsoft.Update.AutoUpdate') } function Get-WuaAuSettings { return (Get-WuaAu).Settings } function Get-WuaWrapper { param( [ScriptBlock] $tryBlock, [object[]] $argumentList, [Parameter(ParameterSetName='OneValue')] [object] $ExceptionReturnValue = $null ) try { return Invoke-Command -ScriptBlock $tryBlock -NoNewScope -ArgumentList $argumentList } catch [System.Runtime.InteropServices.COMException] { switch($_.Exception.HResult) { # 0x8024001e -2145124322 WU_E_SERVICE_STOP Operation did not complete because the service or system was being shut down. wuerror.h -2145124322 { Write-Warning 'Got an error that WU service is stopping. Handling the error.' return $ExceptionReturnValue } # 0x8024402c -2145107924 WU_E_PT_WINHTTP_NAME_NOT_RESOLVED Same as ERROR_WINHTTP_NAME_NOT_RESOLVED - the proxy server or target server name cannot be resolved. wuerror.h -2145107924 { # TODO: add retry for this error Write-Warning 'Got an error that WU could not resolve the name of the update service. Handling the error.' return $ExceptionReturnValue } # 0x8024401c -2145107940 WU_E_PT_HTTP_STATUS_REQUEST_TIMEOUT Same as HTTP status 408 - the server timed out waiting for the request. wuerror.h -2145107940 { # TODO: add retry for this error Write-Warning 'Got an error a request timed out (http status 408 or equivalent) when WU was communicating with the update service. Handling the error.' return $ExceptionReturnValue } # 0x8024402f -2145107921 WU_E_PT_ECP_SUCCEEDED_WITH_ERRORS External cab file processing completed with some errors. wuerror.h -2145107921 { # No retry needed Write-Warning 'Got an error that CAB processing completed with some errors.' return $ExceptionReturnValue } default { throw } } } } function Get-WuaAuNotificationLevel { return Get-WuaWrapper -tryBlock { switch ((Get-WuaAuSettings).NotificationLevel) { 0 { return 'Not Configured' } 1 { return 'Disabled' } 2 { return 'Notify before download' } 3 { return 'Notify before installation' } 4 { return 'Scheduled installation' } default { return 'Reserved'} } } -ExceptionReturnValue [string]::Empty } function Invoke-WuaDownloadUpdates { param( [Parameter(Mandatory=$true)] [ValidateNotNullOrEmpty()] [object] $UpdateCollection ) $downloader = (Get-WuaSession).CreateUpdateDownloader() $downloader.Updates = $UpdateCollection Write-Verbose -Message 'Downloading updates...' -Verbose $downloadResult = $downloader.Download() } function Invoke-WuaInstallUpdates { param( [Parameter(Mandatory=$true)] [ValidateNotNullOrEmpty()] [object] $UpdateCollection ) $installer = (Get-WuaSession).CreateUpdateInstaller() $installer.Updates = $UpdateCollection Write-Verbose -Message 'Installing updates...' -Verbose $installResult = $installer.Install() } function Set-WuaAuNotificationLevel { param( [ValidateSet('Not Configured','Disabled','Notify before download','Notify before installation','Scheduled installation','ScheduledInstallation')] [string] $notificationLevel ) $intNotificationLevel = Get-WuaAuNotificationLevelInt -notificationLevel $notificationLevel $settings = Get-WuaAuSettings $settings.NotificationLevel = $intNotificationLevel $settings.Save() } function Get-WuaAuNotificationLevelInt { param( [ValidateSet('Not Configured','Disabled','Notify before download','Notify before installation','Scheduled installation','ScheduledInstallation')] [string] $notificationLevel ) $intNotificationLevel =0 switch -Regex ($notificationLevel) { '^Not\s*Configured$' { $intNotificationLevel = 0 } '^Disabled$' { $intNotificationLevel = 1 } '^Notify\s*before\s*download$' { $intNotificationLevel = 2 } '^Notify\s*before\s*installation$' { $intNotificationLevel = 3 } '^Scheduled\s*installation$' { $intNotificationLevel = 4 } default { throw 'Invalid notification level'} } return $intNotificationLevel } function Get-WuaSystemInfo { return (New-Object -ComObject 'Microsoft.Update.SystemInfo') } function Get-WuaRebootRequired { return Get-WuaWrapper -tryBlock { Write-Verbose -Message 'TryGet RebootRequired...' -Verbose $rebootRequired = (Get-WuaSystemInfo).rebootRequired Write-Verbose -Message "Got rebootRequired: $rebootRequired" -Verbose return $rebootRequired } -ExceptionReturnValue $true } function Get-WuaSession { return (New-Object -ComObject 'Microsoft.Update.Session') } function get-WuaSearcher { [cmdletbinding(DefaultParameterSetName='category')] param( [parameter(Mandatory = $true, ParameterSetName = 'searchString')] [System.String] $SearchString, [parameter(ParameterSetName = 'category')] [ValidateSet("Security","Important","Optional")] [AllowEmptyCollection()] [AllowNull()] [System.String[]] $Category= @('Security') ) $memberSearchString = $SearchString if($PSCmdlet.ParameterSetName -eq 'category') { $searchStringParams = @{} foreach($CategoryItem in $Category) { $searchStringParams[$CategoryItem]=$true } $memberSearchString = (Get-WuaSearchString @searchStringParams) } return Get-WuaWrapper -tryBlock { param( [parameter(Mandatory = $true)] [System.String] $memberSearchString ) Write-Verbose -Message "Searching for updating using: $memberSearchString" -Verbose return ((Get-WuaSession).CreateUpdateSearcher()).Search($memberSearchString) } -argumentList @($memberSearchString) } function Test-SearchResult { [CmdletBinding()] [OutputType([bool])] param ( [parameter(Mandatory = $true)] [ValidateNotNull()] [System.Object] $SearchResult ) if(!(@($SearchResult | get-member |Select-Object -ExpandProperty Name) -contains 'Updates')) { Write-Verbose 'Did not find updates on SearchResult' return $false } if(!(@(Get-Member -InputObject $SearchResult.Updates |Select-Object -ExpandProperty Name) -contains 'Count')) { Write-Verbose 'Did not find count on updates on SearchResult' return $false } return $true } function Get-TargetResource { [CmdletBinding()] [OutputType([System.Collections.Hashtable])] param ( [parameter(Mandatory = $true)] [ValidateSet("Yes")] [System.String] $IsSingleInstance, [ValidateSet("Security","Important","Optional")] [AllowEmptyCollection()] [AllowNull()] [System.String[]] $Category= @('Security'), [ValidateSet("Disabled","ScheduledInstallation")] [System.String] $Notifications, [parameter(Mandatory = $true)] [ValidateSet("WindowsUpdate","MicrosoftUpdate","WSUS")] [System.String] $Source, [bool] $UpdateNow = $false ) Test-TargetResourceProperties @PSBoundParameters $totalUpdatesNotInstalled = $null $UpdateNowReturn = $null $rebootRequired = $null if($UpdateNow) { $rebootRequired = Get-WuaRebootRequired $SearchResult = (get-WuaSearcher -Category $Category) $totalUpdatesNotInstalled = 0 if($SearchResult -and (Test-SearchResult -SearchResult $SearchResult)) { $totalUpdatesNotInstalled = $SearchResult.Updates.Count } $UpdateNowReturn = $false if($totalUpdatesNotInstalled -eq 0 -and !$rebootRequired) { $UpdateNowReturn = $true } } $notificationLevel = (Get-WuaAuNotificationLevel) $CategoryReturn = $Category $SourceReturn = 'WindowsUpdate' $UpdateServices = (Get-WuaServiceManager).Services #Check if the microsoft update service is registered $defaultService = @($UpdateServices).where{$_.IsDefaultAuService} Write-Verbose -Message "Get default search service: $($defaultService.ServiceId)" if($defaultService.ServiceId -eq '7971f918-a847-4430-9279-4a52d1efe18d') { $SourceReturn = 'MicrosoftUpdate' } elseif ($defaultService.IsManaged) { $SourceReturn = 'WSUS' } $returnValue = @{ IsSingleInstance = 'Yes' Category = $CategoryReturn AutomaticUpdatesNotificationSetting = $notificationLevel TotalUpdatesNotInstalled = $totalUpdatesNotInstalled RebootRequired = $rebootRequired Notifications = $notificationLevel Source = $SourceReturn UpdateNow = $UpdateNowReturn } $returnValue } function Set-TargetResource { # should be [Diagnostics.CodeAnalysis.SuppressMessageAttribute("PSAvoidGlobalVars", "DSCMachineStatus")], but it doesn't work [Diagnostics.CodeAnalysis.SuppressMessageAttribute("PSAvoidGlobalVars", "")] [CmdletBinding(SupportsShouldProcess=$true)] param ( [parameter(Mandatory = $true)] [ValidateSet("Yes")] [System.String] $IsSingleInstance, [ValidateSet("Security","Important","Optional")] [System.String[]] $Category= @('Security'), [ValidateSet("Disabled","ScheduledInstallation")] [System.String] $Notifications, [parameter(Mandatory = $true)] [ValidateSet("WindowsUpdate","MicrosoftUpdate","WSUS")] [System.String] $Source, [bool] $UpdateNow = $false ) Test-TargetResourceProperties @PSBoundParameters $Get = Get-TargetResource @PSBoundParameters $updateCompliant = ($UpdateNow -eq $false -or $Get.UpdateNow -eq $UpdateNow) Write-Verbose "updateNow compliant: $updateCompliant" $notificationCompliant = (!$Notifications -or $Notifications -eq $Get.Notifications) Write-Verbose "notifications compliant: $notificationCompliant" $SourceCompliant = (!$Source -or $Source -eq $Get.Source) Write-Verbose "service compliant: $notificationCompliant" If(!$updateCompliant -and $PSCmdlet.ShouldProcess("Install Updates")) { $SearchResult = (get-WuaSearcher -Category $Category) if($SearchResult -and $SearchResult.Updates.Count -gt 0) { Write-Verbose -Verbose -Message 'Installing updates...' #Write Results foreach($update in $SearchResult.Updates) { $title = $update.Title Write-Verbose -Message "Found update: $Title" -Verbose } Invoke-WuaDownloadUpdates -UpdateCollection $SearchResult.Updates Invoke-WuaInstallUpdates -UpdateCollection $SearchResult.Updates } else { Write-Verbose -Verbose -Message 'No updates' } Write-Verbose -Verbose -Message 'Checking for a reboot...' $rebootRequired = (Get-WuaRebootRequired) if($rebootRequired) { Write-Verbose -Verbose -Message 'A reboot was required' $global:DSCMachineStatus = 1 } else { Write-Verbose -Verbose -Message 'A reboot was NOT required' } } If(!$notificationCompliant -and $PSCmdlet.ShouldProcess("Set notifications to: $notifications")) { Try { #TODO verify that group policy is not overriding this settings # if it is throw an error, if it conflicts Set-WuaAuNotificationLevel -notificationLevel $Notifications } Catch { $ErrorMsg = $_.Exception.Message Write-Verbose $ErrorMsg } } If(!$SourceCompliant ) { if($Source -eq 'MicrosoftUpdate' -and $PSCmdlet.ShouldProcess("Enable Microsoft Update Service")) { Write-Verbose "Enable the Microsoft Update setting" Add-WuaService -ServiceId '7971f918-a847-4430-9279-4a52d1efe18d' Restart-Service wuauserv -ErrorAction SilentlyContinue } elseif($PSCmdlet.ShouldProcess("Disable Microsoft Update Service")) { Write-Verbose "Disable the Microsoft Update setting" Remove-WuaService -ServiceId '7971f918-a847-4430-9279-4a52d1efe18d' } } } function Test-TargetResource { [CmdletBinding()] [OutputType([System.Boolean])] param ( [parameter(Mandatory = $true)] [ValidateSet("Yes")] [System.String] $IsSingleInstance, [ValidateSet("Security","Important","Optional")] [System.String[]] $Category= @('Security'), [ValidateSet("Disabled","ScheduledInstallation")] [System.String] $Notifications, [parameter(Mandatory = $true)] [ValidateSet("WindowsUpdate","MicrosoftUpdate","WSUS")] [System.String] $Source, [bool] $UpdateNow = $false ) Test-TargetResourceProperties @PSBoundParameters #Output the result of Get-TargetResource function. $Get = Get-TargetResource @PSBoundParameters $updateCompliant = ($UpdateNow -eq $false -or $Get.UpdateNow -eq $UpdateNow) Write-Verbose "updateNow compliant: $updateCompliant" $notificationCompliant = (!$Notifications -or $Notifications -eq $Get.Notifications) Write-Verbose "notifications compliant: $notificationCompliant" $SourceCompliant = (!$Source -or $Source -eq $Get.Source) Write-Verbose "service compliant: $notificationCompliant" If($updateCompliant -and $notificationCompliant -and $SourceCompliant) { return $true } Else { return $false } } function Test-TargetResourceProperties { param ( [parameter(Mandatory = $true)] [ValidateSet("Yes")] [System.String] $IsSingleInstance, [ValidateSet("Security","Important","Optional")] [AllowEmptyCollection()] [AllowNull()] [System.String[]] $Category, [ValidateSet("Disabled","ScheduledInstallation")] [System.String] $Notifications, [parameter(Mandatory = $true)] [ValidateSet("WindowsUpdate","MicrosoftUpdate","WSUS")] [System.String] $Source, [bool] $UpdateNow ) $searchStringParams = @{} foreach($CategoryItem in $Category) { $searchStringParams[$CategoryItem.ToLowerInvariant()]=$true } if($UpdateNow -and (!$Category -or $Category.Count -eq 0)) { Write-Warning 'Defaulting to updating to security updates only. Please specify Category to avoid this warning.' } elseif ($searchStringParams.ContainsKey('important') -and !$searchStringParams.ContainsKey('security') ) { Write-Warning "Important updates will include security updates. Please include Security in category to avoid this warning." } elseif ($searchStringParams.ContainsKey('optional') -and !$searchStringParams.ContainsKey('security') ) { Write-Verbose "Optional updates may include security updates." } if($Source -eq 'WSUS') { throw 'The WSUS service option is not implemented.' } } Export-ModuleMember -Function *-TargetResource |