Real world notes: In-place OS upgrade on Server 2012 R2 ConfigMgr distribution points

# Complile mof file
Write-host "Compiling ExtendedStatus.mof file"
mofcomp "C:\Program Files\Microsoft Policy Platform\ExtendedStatus.mof"

# Stop processes that might hinder ccmeval
If (Get-Process -Name ccmrepair -ErrorAction SilentlyContinue)
{
    Write-host "Found ccmrepair process. Stopping it..."
    Stop-Process -Name ccmrepair -Force
    Start-Sleep -Seconds 5
}
If (Get-Process -Name ccmsetup -ErrorAction SilentlyContinue)
{
    Write-host "Found ccmsetup process. Stopping it..."
    Stop-Process -Name ccmsetup -Force
    Start-Sleep -Seconds 5
}

# Run ccmeval to self-remediate the broken WMI
Write-host "Starting Configuration Manager Health Evaluation to repair the ConfigMgr client"
Start-ScheduledTask -TaskName "Configuration Manager Health Evaluation" -TaskPath "\Microsoft\Configuration Manager"
$P = Get-Process -Name ccmeval
$P.WaitForExit()
Start-Sleep -Seconds 5
Write-host "Starting Configuration Manager Health Evaluation one more time"
Start-ScheduledTask -TaskName "Configuration Manager Health Evaluation" -TaskPath "\Microsoft\Configuration Manager"
$P = Get-Process -Name ccmeval
$P.WaitForExit()

# Open the logs to verify it was successful
Start-Process -FilePath C:\Windows\CCM\CMTrace.exe -ArgumentList "C:\Windows\ccmsetup\Logs\ccmsetup-ccmeval.log"
Start-Process -FilePath C:\Windows\CCM\CMTrace.exe -ArgumentList "C:\Windows\CCM\Logs\CcmEval.log"

# MEMCM database params
$script:dataSource = 'MyConfigMgrDatabaseServer' 
$script:database = 'MyConfigMgrDatabase'

# Function to query SQL server...must have db_datareader role for current user context
function Get-SQLData {
    [CmdletBinding()]
    param($Query)
    $connectionString = "Server=$dataSource;Database=$database;Integrated Security=SSPI;"
    $connection = New-Object -TypeName System.Data.SqlClient.SqlConnection
    $connection.ConnectionString = $connectionString
    $connection.Open()
    
    $command = $connection.CreateCommand()
    $command.CommandText = $Query
    $reader = $command.ExecuteReader()
    $table = New-Object -TypeName 'System.Data.DataTable'
    $table.Load($reader)
    
    # Close the connection
    $connection.Close()
    
    return $Table
}

Try
{
    Get-CimClass -Namespace root\SCCMDP -ErrorAction Stop
    Write-host "WMI namespace root\SCCMDP is present"
    Return
}
Catch
{
   If ($_.Exception.NativeErrorCode -eq "InvalidNamespace")
   {
        Write-host "WMI namespace root\SCCMDP is missing" -ForegroundColor Red
        Write-host "Performing remediations..."
   }
   else 
   {
        $_
        Return    
   }
}

# Compile DP mof file
Write-host "Compiling smsdpprov.mof file"
mofcomp "$(Get-SmbShare | where {$_.Name -match "SMS_DP"} | Select -ExpandProperty Path)\sms\bin\smsdpprov.mof"

# Query database for DP package info
$Server = Get-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\SMS\DP" -Name Server | Select -ExpandProperty Server
$Query = @"
declare @ServerName varchar (50) = '$Server'
select
  dp.PackageID
, case
 when p.ShareName <> '' 
  then '\\' + @ServerName + '\' + p.ShareName
 else ''
  end ShareName
, 'Set-WmiInstance -Path ''ROOT\SCCMDP:SMS_PackagesInContLib'' -Arguments @{PackageID="'
  + dp.PackageID + '";PackageShareLocation="' +
 case
  when p.ShareName <> '' 
   then '\\' + @ServerName + '\' + p.ShareName
  else ''
   end + '"}' PowershellCommand
from v_DistributionPoint dp
inner join v_Package p on p.PackageID = dp.PackageID
where dp.ServerNALPath like '[[]"Display=\\' + @ServerName + '%'
"@
Write-host "Querying database for DP package info"
Try
{
    $Results = Get-SQLData -Query $Query  -ErrorAction Stop
    Write-host "Found $($results.rows.count) packages"
}
Catch
{
    $_
    Return
}

# Run the POSH code to add the package back into WMI
Write-host "Restoring WMI instances to ROOT\SCCMDP:SMS_PackagesInContLib"
If ($Results)
{
    Foreach ($result in $results)
    {
        $Scriptblock = [scriptblock]::Create($Result.PowershellCommand)
        Try
        {
            $null = Invoke-Command -ScriptBlock $Scriptblock -ErrorAction Stop
        }
        Catch
        {
            $_
        }
    }
}
Else
{
    Throw "No package info found for this DP"
    Return
}

# Start the content validation task...may take some time
Write-host "Starting Content validation. Check smsdpmon.log for progress"
Start-ScheduledTask -TaskPath "\Microsoft\Configuration Manager" -TaskName "Content Validation"