summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/utils/boot.py39
1 files changed, 33 insertions, 6 deletions
diff --git a/src/utils/boot.py b/src/utils/boot.py
index 5a02ec6..cd66357 100644
--- a/src/utils/boot.py
+++ b/src/utils/boot.py
@@ -43,6 +43,30 @@ def _boot_uefi_windows(disk, part, mountpoint):
finally:
umount(esp_mountpoint)
+def _boot_uefi_linux(disk, part, mountpoint):
+ logging.info(f'Booting Linux system')
+ bootlabel = f'Part-{disk:02d}-{part:02d}'
+ esp, esp_disk, esp_part_number = get_efi_partition(disk)
+ esp_mountpoint = esp.replace('dev', 'mnt')
+ if not mount_mkdir(esp, esp_mountpoint):
+ raise RuntimeError(f'Unable to mount detected EFI System Partition at {esp} into {esp_mountpoint}')
+
+ loader_paths = [f'{esp_mountpoint}/EFI/{bootlabel}/Boot/shimx64.efi',
+ f'{esp_mountpoint}/EFI/ubuntu/shimx64.efi']
+ try:
+ for efi_app in loader_paths:
+ if os.path.exists(efi_app):
+ loader = efi_app[len(esp_mountpoint):]
+ logging.info(f'Found bootloader at ESP partition: {loader}')
+ break
+ else:
+ raise RuntimeError(f'Unable to locate Linux EFI bootloader shimx64.efi')
+
+ efibootmgr_delete_bootentry(bootlabel)
+ efibootmgr_create_bootentry(esp_disk, esp_part_number, loader, bootlabel)
+ efibootmgr_bootnext(bootlabel)
+ finally:
+ umount(esp_mountpoint)
def boot_os_at(disk, part):
if not is_uefi_supported():
@@ -55,9 +79,12 @@ def boot_os_at(disk, part):
logging.info(f'Booting system at {device}. Probing OS family...')
os_family = get_os_family(mountpoint)
- if os_family == OSFamily.WINDOWS:
- _boot_uefi_windows(disk, part, mountpoint)
- elif os_family == OSFamily.LINUX:
- raise NotImplementedError('WIP: Linux UEFI boot not implemented yet.')
- else:
- raise RuntimeError('Unknown OS family')
+ try:
+ if os_family == OSFamily.WINDOWS:
+ _boot_uefi_windows(disk, part, mountpoint)
+ elif os_family == OSFamily.LINUX:
+ _boot_uefi_linux(disk, part, mountpoint)
+ else:
+ raise RuntimeError('Unknown OS family')
+ finally:
+ umount(mountpoint)