Compartir a través de

Agregar un paquete de actualización a Windows RE

  • Artículo

Puede aplicar una actualización, ya sea una actualización de LCU o una actualización dinámica del sistema operativo seguro (DU) a una imagen de Windows RE. Los pasos para actualizar una imagen de Windows RE son diferentes en función de si va a actualizar una imagen sin conexión o una imagen en un sistema en ejecución.

  • Al agregar una LCU: tanto antes de agregar una actualización como después de haberla agregado, verifique el número de versión de WinRE para asegurarse de que la actualización se haya aplicado correctamente.
  • Al agregar un paquete DU: después de agregar el paquete, compruebe la lista de paquetes de la imagen para asegurarse de que el paquete se agregó correctamente a la imagen.

Actualizar una imagen sin conexión

En los pasos siguientes se explica cómo agregar un paquete actualizado a una imagen de Windows RE sin conexión.

Montar la imagen de Windows

La imagen de Windows RE (winre.wim) se incluye dentro de la imagen de Windows (install.wim) y finalmente se copia en la partición de herramientas de Windows RE en el equipo o dispositivo de destino. Para modificar la imagen de WinRE, primero debe montar la imagen de Windows y, a continuación, montar la imagen de Windows RE. Una vez montadas ambas imágenes, puede realizar los cambios, desmontar la imagen de WinRE y desmontar la imagen de Windows.

  • Crear un directorio de montaje y montar install.wim:

    md C:\mount\windows
Dism /Mount-Wim /WimFile:C:\images\install.wim /index:1 /MountDir:C:\mount\windows

Monte la imagen de WinRE

  • Monte el archivo de imagen de Windows RE desde su imagen montada.

    Md c:\mount\winre 
Dism /Mount-Wim /WimFile:C:\mount\windows\Windows\System32\Recovery\winre.wim /index:1 /MountDir:C:\mount\winre

    Sugerencia

    Si no puede ver winre.wim en el directorio especificado, use el siguiente comando para definir el archivo visible:

    attrib -h -a -s C:\mount\windows\Windows\System32\Recovery\winre.wim

Agregar el paquete de actualización a WinRE

Con la imagen de recuperación montada, puede agregar la actualización para corregir el problema.

  1. Descargue el paquete de actualización necesario y agréguelo a la imagen de recuperación.

    Dism /Add-Package /Image:C:\mount\winre /PackagePath:"c:\downloadedupdate\update.msu"

    or

    Dism /Add-Package /Image:C:\mount\ /PackagePath:"c:\downloadedupdate\dynamicupdate.cab"

  2. Si agregó un paquete de actualización dinámica, compruebe que el paquete que agregó está en la imagen:

     Dism /Image:C:\mount\ /Get-Packages

    Compruebe la lista del paquete de actualización que acaba de agregar. El paquete debe incluir el número de KB y mostrará el State como Installed si se ha agregado correctamente:

    Package Identity : Package_for_KB5021041~31bf3856ad364e35~amd64~~22621.815.1.0
State : Installed
Release Type : Update

  3. Limpie la imagen de recuperación y restablezca la base de los componentes reemplazados:

    dism /image:C:\mount /cleanup-image /StartComponentCleanup /ResetBase

Desmontar las imágenes

  1. Cierre todas las aplicaciones que puedan acceder a los archivos de la imagen, incluida la Explorador de archivos.

  2. Confirma los cambios y desmonta la imagen de Windows RE:

    Dism /Unmount-Image /MountDir:"C:\mount\winre" /Commit

    El proceso puede tardar unos minutos.

  3. Confirma los cambios y desmonta la imagen de Windows:

    Dism /Unmount-Image /MountDir:"C:\mount\windows" /Commit

    Donde C es la letra de unidad de la unidad que contiene la imagen. Este proceso puede tardar varios minutos.

La imagen de Windows RE ahora se actualiza y forma parte de la imagen de Windows más grande.

Aplicar la actualización a un equipo en ejecución

Puede aplicar la corrección a la imagen de recuperación de Windows en un equipo en ejecución. Los pasos siguientes le guiarán a través del montaje de una imagen de recuperación de Windows, la aplicación de una actualización y, a continuación, el guardado de los cambios:

  1. Abrir un aviso de comando como administrador.

  2. Cree un directorio donde pueda montar la imagen de recuperación para realizar cambios en ella.

    md c:\mount

  3. Monte la imagen de recuperación con REAgentC:

    ReAgentC.exe /mountre /path c:\mount

  4. Con la imagen de recuperación montada, puede descargar y agregar el paquete de actualización a la imagen de Windows RE.

    Dism /Add-Package /Image:C:\mount\ /PackagePath:"c:\downloadedupdate\update.msu"

    or

    Dism /Add-Package /Image:C:\mount\ /PackagePath:"c:\downloadedupdate\dynamicupdate.cab"

  5. Si agregó un paquete de actualización dinámica, compruebe que el paquete que agregó está en la imagen:

     Dism /Image:C:\mount\ /Get-Packages

    Compruebe la lista del paquete de actualización que acaba de agregar. El paquete debe incluir el número de KB y mostrará el State como Installed si se ha agregado correctamente:

    Package Identity : Package_for_KB5021041~31bf3856ad364e35~amd64~~22621.815.1.0
State : Installed
Release Type : Update

  6. Limpie la imagen de recuperación y restablezca la base de los componentes reemplazados:

    dism /image:C:\mount /cleanup-image /StartComponentCleanup /ResetBase

  7. Desmontar la imagen de recuperación de Windows:

    ReAgentC.exe /unmountre /path c:\mount /commit

  8. Si el disco del equipo está protegido por BitLocker o el cifrado de dispositivos: Use ReagentC para deshabilitar y volver a habilitar Windows RE. Esto garantiza que la imagen de Windows RE actualizada esté activada y configurada correctamente para la instalación de Windows:

    reagentc /disable
reagentc /enable

La imagen de Windows RE en el equipo en ejecución ahora se actualiza.

Comprobar la versión de la imagen de WinRE

Puede comprobar el número de versión de la imagen de WinRE en instalaciones de Windows en línea y sin conexión. Esto puede resultar útil para comprobar la versión de WinRE antes de agregar una actualización y, después, después de aplicar una actualización para comprobar que la actualización se agregó correctamente.

Nota:

El número de versión de WinRE solo cambiará después de agregar una LCU. Si agrega un paquete DU, use DISM /get-packages como se describe en los pasos anteriores para asegurarse de que el paquete se ha agregado a la imagen.

Validar la versión de Windows RE en un sistema operativo Windows en línea

  1. Use ReagentC para buscar la ubicación de la imagen WinRE instalada ejecutando el siguiente comando:

    reagentc /info

    Este comando devuelve la ubicación Windows RE similar al ejemplo siguiente:

    \\?\GLOBALROOT\device\harddisk0\partition4\Recovery\WindowsRE

  2. Use DISM para obtener información de versión sobre la imagen winre.wim ubicada en la ruta de acceso devuelta por ReagentC en el paso anterior. Asegúrese de agregar el nombre de la imagen WinRE y el número de índice de la imagen a la ruta cuando ejecute el comando:

    Dism /Get-ImageInfo /ImageFile:\\?\GLOBALROOT\device\harddisk0\partition4\Recovery\WindowsRE\winre.wim /index:1

    Observe la versión notificada. Por ejemplo:

    Version : 10.0.22621
ServicePack Build : 1
ServicePack Level : 0

    Asegúrese de que ServicePackBuild sea mayor o igual que UBR para la actualización que agregó. Por ejemplo, para Windows 11, versión 22H2, la actualización de seguridad de noviembre se mostraría 819 como SerivcePack Build, ya que el número de versión completo de esa actualización es 22621.819.

    • Si la versión notificada es una versión anterior, indica que la imagen de Windows RE no está actualizada.
    • Si la versión notificada es la misma o posterior, no se necesita ninguna acción.

Validar la versión de Windows RE en un sistema operativo Windows sin conexión

  1. Monte la imagen de Windows en una carpeta local, por ejemplo C:\mount\windows:

    mkdir C:\mount\windows
Dism /Mount-Image /ImageFile:C:\mount\install.wim /Index:1 /MountDir:C:\mount\windows

  2. Use DISM para obtener información de versión sobre la imagen winre.wim ubicada en la carpeta Windows\system32\recovery. Asegúrese de especificar el número de índice de la imagen al ejecutar el comando:

    Dism /Get-ImageInfo /ImageFile:c:\mount\windows\windows\system32\recovery\winre.wim /index:1

    Observe la versión notificada. Por ejemplo:

    Version : 10.0.22621
ServicePack Build : 1
ServicePack Level : 0

    Asegúrese de que ServicePackBuild sea mayor o igual que UBR para la actualización que agregó. Por ejemplo, para Windows 11, versión 22H2, la actualización de seguridad de noviembre se mostraría 819 como SerivcePack Build, ya que el número de versión completo de esa actualización es 22621.819.

    • Si la versión notificada es una versión anterior, indica que la imagen de Windows RE no está actualizada.
    • Si la versión notificada es la misma o posterior, no se necesita ninguna acción.

Ampliar la partición de Windows RE

El siguiente script de ejemplo se puede usar para aumentar el tamaño de la partición de recuperación para actualizar correctamente el entorno de recuperación de Windows (WinRE). Se recomienda tener 250 MB de espacio libre en la partición de recuperación para que las actualizaciones de WinRE se instalen correctamente. En los dispositivos que pueden no tener suficiente espacio libre en la partición de recuperación, se puede usar el script de ejemplo siguiente para ampliar la partición de recuperación en 250 MB.

Reinicie la máquina antes de ejecutar el script. Esto es fundamental, ya que puede haber operaciones de partición pendientes almacenadas provisionalmente en el equipo que deban finalizarse antes de que el script pueda aumentar de forma segura el tamaño de la partición WinRE. Después de que la máquina se reinicie, abra Powershell como administrador y ejecute la ruta mkdir< en el nuevo directorio de respaldo> para crear un directorio de respaldo que el script pueda usar en caso de que no se pueda restaurar la partición original. Tenga en cuenta la ubicación de este directorio de copia de seguridad, ya que el script solicitará la ruta de acceso a la copia de seguridad. Si va a implementar esto a escala, puede omitir la solicitud del script mediante los parámetros

"<Archivo de script>: SkipConfirmation $true -BackupFolder <carpeta para copia de seguridad>"

Por ejemplo,

Resize_script.ps1 -SkipConfirmation $true -BackupFolder c:\winre_backup

Param (
[Parameter(Mandatory=$false,HelpMessage="Skip confirmation")][bool]$SkipConfirmation=$false,
[Parameter(Mandatory=$true,HelpMessage="Path to backup old WinRE partition content to")][string]$BackupFolder
)
# ------------------------------------
# Helper functions
# ------------------------------------
# Log message
function LogMessage([string]$message)
{
  $message = "$message"
  Write-Host $message
}
# Extract numbers from string
function ExtractNumbers([string]$str)
{
  $cleanString = $str -replace "[^0-9]"
  return [long]$cleanString
}
# Display partition info using fsutil
# Return an array, the first value is total size and the second value is free space
function DisplayPartitionInfo([string[]]$partitionPath)
{
  $volume = Get-WmiObject -Class Win32_Volume | Where-Object { $partitionPath -contains $_.DeviceID }
  LogMessage("  Partition capacity: " +  $volume.Capacity)
  LogMessage("  Partition free space: " + $volume.FreeSpace)
  return $volume.Capacity, $volume.FreeSpace
} 
# Display WinRE status
function DisplayWinREStatus()
{
  # Get WinRE partition info
  $WinREInfo = Reagentc /info
  foreach ($line in $WinREInfo)
  {
    $params = $line.Split(':')
    if ($params.Count -lt 2)
    {
      continue
    }
    if (($params[1].Trim() -ieq "Enabled") -Or (($params[1].Trim() -ieq "Disabled")))
    {
      $Status = $params[1].Trim() -ieq "Enabled"
      LogMessage($line.Trim())
    }
    if ($params[1].Trim() -like "\\?\GLOBALROOT*")
    {
      $Location = $params[1].Trim()
      LogMessage($line.Trim())
    }
  }
  
  return $Status, $Location
}
# ------------------------------------
# Main execution
# ------------------------------------
# Clear the error
$Error.Clear()
# ------------------------------------
# Examining the system to collect required info 
# for the execution
# Need to check WinRE status, collect OS and WinRE
# partition info
# ------------------------------------
LogMessage("Start time: $([DateTime]::Now)")
LogMessage("Examining the system...")
$NeedShrink = $true
$NeedCreateNew = $false
$NeedBackup = $false
# Get WinRE partition info
$InitialWinREStatus = DisplayWinREStatus
$WinREStatus = $InitialWinREStatus[0]
$WinRELocation = $InitialWinREStatus[1]
if (!$WinREStatus)
{
  LogMessage("Error: WinRE Disabled")
  exit 1
}
# Get System directory and ReAgent xml file
$system32Path = [System.Environment]::SystemDirectory
LogMessage("System directory: " + $system32Path)
$ReAgentXmlPath = Join-Path -Path $system32Path -ChildPath "\Recovery\ReAgent.xml"
LogMessage("ReAgent xml: " + $ReAgentXmlPath)
if (!(Test-Path  $ReAgentXmlPath))
{
  LogMessage("Error: ReAgent.xml cannot be found")
  exit 1
}
# Get OS partition
LogMessage("")
LogMessage("Collecting OS and WinRE partition info...")
$OSDrive = $system32Path.Substring(0,1)
$OSPartition = Get-Partition -DriveLetter $OSDrive
# Get WinRE partition
$WinRELocationItems = $WinRELocation.Split('\\')
foreach ($item in $WinRELocationItems)
{
    if ($item -like "harddisk*")
  {
    $OSDiskIndex = ExtractNumbers($item)
  }
  if ($item -like "partition*")
  {
    $WinREPartitionIndex = ExtractNumbers($item)
  }
}
LogMessage("OS Disk: " + $OSDiskIndex)
LogMessage("OS Partition: " + $OSPartition.PartitionNumber)
LogMessage("WinRE Partition: " + $WinREPartitionIndex)
$WinREPartition = Get-Partition -DiskNumber $OSDiskIndex -PartitionNumber $WinREPartitionIndex
$diskInfo = Get-Disk -number $OSDiskIndex
$diskType = $diskInfo.PartitionStyle
LogMessage("Disk PartitionStyle: " + $diskType)
# Display WinRE partition size info
LogMessage("WinRE partition size info")
$WinREPartitionSizeInfo = DisplayPartitionInfo($WinREPartition.AccessPaths)
LogMessage("WinRE Partition Offset: " + $WinREPartition.Offset)
LogMessage("WinRE Partition Type: " + $WinREPartition.Type)
LogMessage("OS partition size: " + $OSPartition.Size)
LogMessage("OS partition Offset: " + $OSPartition.Offset)
$OSPartitionEnds = $OSPartition.Offset + $OSPartition.Size
LogMessage("OS partition ends at: " + $OSPartitionEnds)
LogMessage("WinRE partition starts at: " + $WinREPartition.Offset)
$WinREIsOnSystemPartition = $false
if ($diskType -ieq "MBR")
{
  if ($WinREPartition.IsActive)
  {
    LogMessage("WinRE is on System partition")
    $WinREIsOnSystemPartition = $true
  }
}
if ($diskType -ieq "GPT")
{
  if ($WinREPartition.Type -ieq "System")
  {
    LogMessage("WinRE is on System partition")
    $WinREIsOnSystemPartition = $true
  }
}
# Checking the BackupFolder parameter
if ($PSBoundParameters.ContainsKey('BackupFolder'))
{
  LogMessage("")
  LogMessage("Backup Directory: [" + $BackupFolder + "]")
  
  $Needbackup = $true
  
  if ($WinREIsOnSystemPartition)
  {
    $Needbackup = $false
    LogMessage("WinRE is on System partition which will be preserved. No need to backup content")
  }
  else
  {
    if (Test-path $BackupFolder)
    {
      $items = Get-ChildItem -Path $BackupFolder
      if ($items)
      {
        LogMessage("Error: Existing backup directory is not empty")
        exit 1
      }
    }
    else
    {
      LogMessage("Creating backup directory...")
      try 
      {
        $item = New-Item -Path $BackupFolder -ItemType Directory -ErrorAction Stop
        if ($item)
        {
          LogMessage("Backup directory created")
        }
        else
        {
          LogMessage("Error: Failed to create backup directory [" + $BackupFolder + "]")
          exit 1
        }
      } catch 
      {
        LogMessage("Error: An error occurred: $_")
        exit 1
      }
    }
  }
}
# ------------------------------------
# Verify whether we meet requirements of execution
# - WinRE cannot be on OS partition for the extension
# - WinRE partition must be the next partition after OS partition
# - If WinRE partition already have >=250MB free space, no need to do repartition
# - If there is enough unallocated space to grow the WinRE partition size, skip shrinking OS
# 
# However, if the WinRE partition is before the OS partition, there is no chance to extend it
# As a result, it's better to create a new WinRE partition after the OS partition
# ------------------------------------
# Perform a few checks
LogMessage("")
LogMessage("Verifying if the WinRE partition needs to be extended or not...")
if (!(($diskType -ieq "MBR") -Or ($diskType -ieq "GPT")))
{
  LogMessage("Error: Got an unexpected disk partition style: " +$diskType)
  exit 1
}
# WinRE partition must be after OS partition for the repartition
if ($WinREPartitionIndex -eq $OSPartition.PartitionNumber)
{
  LogMessage("WinRE and OS are on the same partition, should not perform extension")
  exit 0
}
$supportedSize = Get-PartitionSupportedSize -DriveLetter $OSDrive
# if there is enough free space, skip extension
if ($WinREPartitionSizeInfo[1] -ge 250MB)
{
  LogMessage("More than 250 MB of free space was detected in the WinRE partition, there is no need to extend the partition")
  exit 0
}
if ($WinREPartition.Offset -lt $OSPartitionEnds)
{
  LogMessage("WinRE partition is not after OS partition, cannot perform extension")
  LogMessage("Need to create a new WinRE partition after OS partition")
  $NeedCreateNew = $true
  $NeedShrink = $true
  
  # Calculate the size of repartition
  # Will create a new WinRE partition with current WinRE partition size + 250 MB
  # The OS partition size will be shrunk by the new WinRE partition size
  $targetWinREPartitionSize = $WinREPartitionSizeInfo[0] + 250MB
  $shrinkSize = [Math]::Ceiling($targetWinREPartitionSize / 1MB) * 1MB
  $targetOSPartitionSize = $OSPartition.Size - $shrinkSize
  if ($targetOSPartitionSize -lt $supportedSize.SizeMin)
  {
    LogMessage("Error: The target OS partition size after shrinking is smaller than the supported minimum size, cannot perform the repartition")
    exit 1
  }
}
else
{
  if ($WinREIsOnSystemPartition)
  {
    LogMessage("WinRE parititon is after the OS partition and it's also System partition")
    LogMessage("Error: Got unexpected disk layout, cannot proceed")
    exit 1
  }
  if (!($WinREPartitionIndex -eq ($OSPartition.PartitionNumber + 1)))
  {
    LogMessage("Error: WinRE partition is not right after the OS partition, cannot extend WinRE partition")
    exit 1
  }
  # Calculate the size of repartition
  # Will shrink OS partitition by 250 MB
  $shrinkSize = 250MB
  $targetOSPartitionSize = $OSPartition.Size - $shrinkSize
  $targetWinREPartitionSize = $WinREPartitionSizeInfo[0] + 250MB
  $UnallocatedSpace = $WinREPartition.Offset - $OSPartitionEnds;
  # If there is unallocated space, consider using it
  if ($UnallocatedSpace -ge 250MB)
  {
    $UnallocatedSpace = $WinREPartition.Offset - $OSPartitionEnds;
    LogMessage("Found unallocated space between OS and WinRE partition: " + $UnallocatedSpace)
    LogMessage("There is already enough space to extend WinRE partition without shrinking the OS partition")
    $NeedShrink = $false
    $targetOSPartitionSize = 0
  }
  else
  {
    $shrinkSize = [Math]::Ceiling((250MB - $UnallocatedSpace)/ 1MB) * 1MB
    if ($shrinkSize > 250MB)
    {
      $shrinkSize = 250MB
    }
    $targetOSPartitionSize = $OSPartition.Size - $shrinkSize
    if ($targetOSPartitionSize -lt $supportedSize.SizeMin)
    {
      LogMessage("Error: The target OS partition size after shrinking is smaller than the supported minimum size, cannot perform the repartition")
      exit 1
    }
  }
}
# ------------------------------------
# Report execution plan and ask for user confirmation to continue
# ------------------------------------
# Report the changes planned to be executed, waiting for user confirmation
LogMessage("")
LogMessage("Summary of proposed changes")
if ($NeedCreateNew)
{
  LogMessage("Note: WinRE partition is before OS partition, need to create a new WinRE partition after OS partition")
  LogMessage("Will shrink OS partition by " + $shrinkSize)
  LogMessage("  Current OS partition size: " + $OSPartition.Size)
  LogMessage("  Target OS partition size after shrinking: " + $targetOSPartitionSize)
  LogMessage("New WinRE partition will be created with size: ", $targetWinREPartitionSize)
  if ($WinREIsOnSystemPartition)
  {
    LogMessage("Existing WinRE partition is also system partition, it will be preserved")
  }
  else
  {
    LogMessage("Existing WinRE partition will be deleted")
    LogMessage("  WinRE partition: Disk [" + $OSDiskIndex + "] Partition [" + $WinREPartitionIndex + "]")
    LogMessage("  Current WinRE partition size: " + $WinREPartitionSizeInfo[0])
  }
}
else
{
  if ($NeedShrink)
  {
    LogMessage("Will shrink OS partition by " + $shrinkSize)
    LogMessage("  Current OS partition size: " + $OSPartition.Size)
    LogMessage("  Target OS partition size after shrinking: " + $targetOSPartitionSize)
    if ($UnallocatedSpace -ge 0)
    {
      LogMessage("Unallocated space between OS and WinRE partition that will be used towards the new WinRE partition: " + $UnallocatedSpace)
    }
  }
  else
  {
    LogMessage("Will use 250MB from unallocated space between OS and WinRE partition")
  }
  LogMessage("Will extend WinRE partition size by 250MB")
  LogMessage("  WinRE partition: Disk [" + $OSDiskIndex + "] Partition [" + $WinREPartitionIndex + "]")
  LogMessage("  Current WinRE partition size: " + $WinREPartitionSizeInfo[0])
  LogMessage("  New WinRE partition size:     " + $targetWinREPartitionSize)
  LogMessage("WinRE will be temporarily disabled before extending the WinRE partition and enabled automatically in the end")
  if ($UnallocatedSpace -ge 100MB)
  {
    LogMessage("Warning: More than 100MB of unallocated space was detected between the OS and WinRE partitions")
    LogMessage("Would you like to proceed by using the unallocated space between the OS and the WinRE partitions?")
  }
}
if ($Needbackup)
{
  LogMessage("")
  LogMessage("The contents of the old WinRE partition will be backed up to [" + $BackupFolder + "]")
}
LogMessage("")
LogMessage("Please reboot the device before running this script to ensure any pending partition actions are finalized")
LogMessage("")
if ($SkipConfirmation)
{
  LogMessage("User chose to skip confirmation")
  LogMessage("Proceeding with changes...")
}
else
{
  $userInput = Read-Host -Prompt "Would you like to proceed? Y for Yes and N for No"
    
  if ($userInput -ieq "Y")
  {
    LogMessage("Proceeding with changes...")
  }
  elseif ($userInput -ieq "N")
  {
    LogMessage("Canceling based on user request, no changes were made to the system")
    exit 0
  }
  else
  {
    LogMessage("Error: Unexpected user input: [" + $userInput + "]") 
    exit 0
  }
}
LogMessage("")
LogMessage("Note: To prevent unexpected results, please do not interrupt the execution or restart your system")
# ------------------------------------
# Do the actual execution
# The main flow is:
# 1. Check whether ReAgent.xml has stage location and clear it for repartiion
# 2. Disable WinRE as WinRE partition will be deleted
# 3. Perform the repartition to create a larger WinRE partition
# 4. Re-enable WinRE
# ------------------------------------
LogMessage("")
# Load ReAgent.xml to clear Stage location
LogMessage("Loading [" + $ReAgentXmlPath + "] ...")
$xml = [xml](Get-Content -Path $ReAgentXmlPath)
$node = $xml.WindowsRE.ImageLocation
if (($node.path -eq "") -And ($node.guid -eq "{00000000-0000-0000-0000-000000000000}") -And ($node.offset -eq "0") -And ($node.id -eq "0"))
{
  LogMessage("Stage location info is empty")
}
else
{
  LogMessage("Clearing stage location info...")
  $node.path = ""
  $node.offset = "0"
  $node.guid= "{00000000-0000-0000-0000-000000000000}"
  $node.id="0"
  # Save the change
  LogMessage("Saving changes to [" + $ReAgentXmlPath + "]...")
  $xml.Save($ReAgentXmlPath)
}
# Disable WinRE
LogMessage("Disabling WinRE...")
reagentc /disable
if (!($LASTEXITCODE -eq 0))
{
  LogMessage("Warning: encountered an error when disabling WinRE: " + $LASTEXITCODE)
  exit $LASTEXITCODE
}
# Verify WinRE is under C:\Windows\System32\Recovery\WinRE.wim
$disableWinREPath = Join-Path -Path $system32Path -ChildPath "\Recovery\WinRE.wim"
LogMessage("Verifying that WinRE wim exists in downlevel at default location")
if (!(Test-Path $disableWinREPath))
{
  LogMessage("Error: Cannot find " + $disableWinREPath)
  
  # Re-enable WinRE
  LogMessage("Re-enabling WinRE on error...")
  reagentc /enable
  if (!($LASTEXITCODE -eq 0))
  {
    LogMessage("Warning: encountered an error when enabling WinRE: " + $LASTEXITCODE)
  }
  exit 1
}
# ------------------------------------
# Perform the repartition
# 1. Resize the OS partition
# 2. Delete the WinRE partition
# 3. Create a new WinRE partition
# ------------------------------------
LogMessage("Performing repartition to extend the WinRE partition ...")
# 1. Resize the OS partition
if ($NeedShrink)
{
  LogMessage("Shrinking the OS partition to create a larger WinRE partition")
  LogMessage("Resizing the OS partition to: [" + $targetOSPartitionSize + "]...")
  Resize-Partition -DriveLetter $OSDrive -Size $targetOSPartitionSize
  if ($Error.Count -gt 0) {
    LogMessage("Error: Resize-Partition encountered errors: " + $Error[0].Exception.Message)
    
    # Re-enable WinRE
    LogMessage("Re-enabling WinRE on error...")
    reagentc /enable
    if (!($LASTEXITCODE -eq 0))
    {
      LogMessage("Warning: encountered an error when enabling WinRE: " + $LASTEXITCODE)
    }
    exit 1
  }
  $OSPartitionAfterShrink = Get-Partition -DriveLetter $OSDrive
  LogMessage("Target partition size: " + $targetOSPartitionSize)
  LogMessage("Size of OS partition after shrinking: " + $OSPartitionAfterShrink.Size)
}
# 2. Delete the WinRE partition
LogMessage("")
if ($WinREIsOnSystemPartition)
{
  LogMessage("Existing WinRE partition is System partition, skipping deletion")
}
else
{  
  # If requested by user, backup rest of the content on WinRE partition to backup directory
  if ($Needbackup)
  {
    $sourcePath = $WinREPartition.AccessPaths[0] 
    LogMessage("Copying content on WinRE partition from [" + $sourcePath + "] to [" + $BackupFolder + "]...")
    
    # Copy-Item may have access issue with certain system folders, enumerate the children items and exlcude them
    $items = Get-ChildItem -LiteralPath $sourcePath -Force
    foreach ($item in $items)
    {
      if ($item.Name -ieq "System Volume Information")
      {
        continue
      }
      $sourceItemPath = Join-Path -Path $sourcePath -ChildPath $item.Name
      $destItemPath = Join-Path -Path $BackupFolder -ChildPath $item.Name
      try 
      {
        LogMessage("Copying [" + $sourceItemPath + "] to [" + $destItemPath + "]...")
        Copy-Item -LiteralPath $sourceItemPath -Destination $destItemPath -Recurse -Force
      } catch 
      {
        LogMessage("Error: An error occurred during copy: $_")
        exit 1
      }
    }
    
    LogMessage("Backup completed")
    LogMessage("")
  }
  LogMessage("Deleting WinRE partition: Disk [" + $OSDiskIndex + "] Partition [" + $WinREPartitionIndex + "]...")
  Remove-Partition -DiskNumber $OSDiskIndex -PartitionNumber $WinREPartitionIndex -Confirm:$false
  if ($Error.Count -gt 0) {
    LogMessage("Error: Remove-Partition encountered errors: " + $Error[0].Exception.Message)
    exit 1
  }
}
# A short sleep for the partition change
Sleep 5
# 3. Create a new WinRE partition
LogMessage("")
LogMessage("Creating new WinRE partition...")
LogMessage("Target size: " + $targetWinREPartitionSize)
if ($diskType -ieq "GPT")
{
  $partition = New-Partition -DiskNumber $OSDiskIndex -Size $targetWinREPartitionSize -GptType "{de94bba4-06d1-4d40-a16a-bfd50179d6ac}"
  
  $newPartitionIndex = $partition.PartitionNumber
  # A short sleep to make sure the partition is ready for formatting
  Sleep 2
  LogMessage("Formating the partition...")
  $result = Format-Volume -Partition $partition -FileSystem NTFS -Confirm:$false
  if ($Error.Count -gt 0) {
    LogMessage("Error: Format-Volume encountered errors: " + $Error[0].Exception.Message)
    exit 1
  }
}
else
{
  #$partition = New-Partition -DiskNumber $OSDiskIndex -Size $targetWinREPartitionSize -MbrType 0x27
  $targetWinREPartitionSizeInMb = [int]($targetWinREPartitionSize/1MB)
  $diskpartScript = 
@"
select disk $OSDiskIndex
create partition primary size=$targetWinREPartitionSizeInMb id=27
format quick fs=ntfs label="Recovery"
set id=27
"@
  $TempPath = $env:Temp
  $diskpartSciptFile = Join-Path -Path $TempPath -ChildPath "\ExtendWinRE_MBR_PowershellScript.txt"
  
  LogMessage("Creating temporary diskpart script to create Recovery partition on MBR disk...")
  LogMessage("Temporary diskpart script file: " + $diskpartSciptFile)
  $diskpartScript | Out-File -FilePath $diskpartSciptFile -Encoding ascii
  
  LogMessage("Executing diskpart script...")
  try 
  {
    $diskpartOutput = diskpart /s $diskpartSciptFile
    
    if ($diskpartOutput -match "DiskPart successfully") 
    {
      LogMessage("Diskpart script executed successfully")
    }
    else
    {
      LogMessage("Error executing diskpart script:" + $diskpartOutput)
      exit 1
    }
    LogMessage("Deleting temporary diskpart script file...")
    Remove-Item $diskpartSciptFile
  }
  catch 
  {
    LogMessage("Error executing diskpart script: $_")
    exit 1
  }
  
  $vol = Get-Volume -FileSystemLabel "Recovery"
  $newPartitionIndex = (Get-Partition | Where-Object { $_.AccessPaths -contains $vol.Path } ).PartitionNumber
}
if ($Error.Count -gt 0) 
{
  LogMessage("Error: New-Partition encountered errors: " + $Error[0].Exception.Message)
  exit 1
}
LogMessage("New Partition index: " + $newPartitionIndex)
# Re-enable WinRE
LogMessage("Re-enabling WinRE...")
reagentc /enable
if (!($LASTEXITCODE -eq 0))
{
  LogMessage("Warning: encountered an error when enabling WinRE: " + $LASTEXITCODE)
  exit $LASTEXITCODE
}
# In the end, Display WinRE status to verify WinRE is enabled correctly
LogMessage("")
LogMessage("WinRE Information:")
$FinalWinREStatus = DisplayWinREStatus
$WinREStatus = $FinalWinREStatus[0]
$WinRELocation = $FinalWinREStatus[1]
if (!$WinREStatus)
{
  LogMessage("Warning: WinRE Disabled")
}
$WinRELocationItems = $WinRELocation.Split('\\')
foreach ($item in $WinRELocationItems)
{
  if ($item -like "partition*")
  {
    $WinREPartitionIndex = ExtractNumbers($item)
  }
}
LogMessage("WinRE Partition Index: " + $WinREPartitionIndex)
$WinREPartition = Get-Partition -DiskNumber $OSDiskIndex -PartitionNumber $WinREPartitionIndex
$WinREPartitionSizeInfoAfter = DisplayPartitionInfo($WinREPartition.AccessPaths)
LogMessage("")
LogMessage("OS Information:")
$OSPartition = Get-Partition -DriveLetter $OSDrive
LogMessage("OS partition size: " + $OSPartition.Size)
LogMessage("OS partition Offset: " + $OSPartition.Offset)
if (!($WinREPartitionIndex -eq $newPartitionIndex))
{
  LogMessage("Warning: WinRE is installed to partition [" + $WinREPartitionIndex +"], but the newly created Recovery partition is [" + $newPartitionIndex + "]")
}
LogMessage("End time: $([DateTime]::Now)")
if ($NeedBackup)
{
  LogMessage("")
  LogMessage("The contents of the old WinRE partition has been backed up to [" + $BackupFolder + "]")
}
LogMessage("")
LogMessage("Successfully completed the operation")