summaryrefslogtreecommitdiffstats
path: root/src/utils/uefi.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/utils/uefi.py')
-rw-r--r--src/utils/uefi.py52
1 files changed, 52 insertions, 0 deletions
diff --git a/src/utils/uefi.py b/src/utils/uefi.py
index 9bc2e09..5561a6a 100644
--- a/src/utils/uefi.py
+++ b/src/utils/uefi.py
@@ -11,6 +11,10 @@ import logging
import os
import shlex
import subprocess
+import shutil
+from src.utils.disk import *
+from src.utils.fs import *
+from src.utils.probe import *
import fdisk
@@ -96,3 +100,51 @@ def efibootmgr_create_bootentry(disk, part, loader, label, add_to_bootorder=True
proc = subprocess.run(shlex.split(efibootmgr_cmd), check=True, text=True)
except OSError as e:
raise OSError(f'Unexpected error adding boot entry to nvram. UEFI firmware might be buggy') from e
+
+
+def copy_windows_efi_bootloader(disk, partition):
+ device = get_partition_device(disk, partition)
+ mountpoint = device.replace('dev', 'mnt')
+ if not mount_mkdir(device, mountpoint):
+ raise RuntimeError(f'Cannot probe OS family. Unable to mount {device} at {esp_mountpoint}')
+
+ os_family = get_os_family(mountpoint)
+ is_uefi = is_uefi_supported()
+
+ if not is_uefi or os_family != OSFamily.WINDOWS:
+ return
+
+ bootlabel = f'Part-{disk:02d}-{partition:02d}'
+ esp, esp_disk, esp_part_number = get_efi_partition(disk, enforce_gpt=True)
+ esp_mountpoint = esp.replace('dev', 'mnt')
+ if not mount_mkdir(esp, esp_mountpoint):
+ umount(mountpoint)
+ raise RuntimeError(f'Unable to mount detected EFI System Partition at {esp} into {esp_mountpoint}')
+
+ loader_paths = [f'{esp_mountpoint}/EFI/{bootlabel}/Boot/bootmgfw.efi',
+ f'{esp_mountpoint}/EFI/Microsoft/Boot/bootmgfw.efi']
+ try:
+ for efi_app in loader_paths:
+ logging.info(f'Searching for {efi_app}')
+ if os.path.exists(efi_app):
+ loader = efi_app
+ logging.info(f'Found bootloader at ESP partition: {loader}')
+ break
+ else:
+ raise RuntimeError(f'Unable to locate Windows EFI bootloader bootmgfw.efi')
+
+ loader_dir = os.path.dirname(loader)
+ destination_dir = f'{mountpoint}/ogBoot'
+ if os.path.exists(destination_dir):
+ try:
+ shutil.rmtree(destination_dir)
+ except Exception as e:
+ raise OSError(f'Failed to delete {destination_dir}: {e}') from e
+ logging.info(f'Copying {loader_dir} into {destination_dir}')
+ try:
+ shutil.copytree(loader_dir, destination_dir)
+ except Exception as e:
+ raise OSError(f'Failed to copy {loader_dir} into {destination_dir}: {e}') from e
+ finally:
+ umount(mountpoint)
+ umount(esp_mountpoint)