1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
#
# Copyright (C) 2023 Soleta Networks <info@soleta.eu>
#
# This program is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the
# Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
import json
import logging
import os
import shlex
import subprocess
import fdisk
def _find_bootentry(entries, label):
for entry in entries:
if entry['description'] == label:
return entry
else:
raise NameError('Boot entry {label} not found')
def _strip_boot_prefix(entry):
try:
num = entry['name'][4:]
except:
raise KeyError('Unable to strip "Boot" prefix from boot entry')
return num
def _check_efibootmgr_json():
"""
Checks if efibootmgr --json option is supported.
Returns True if supported, False otherwise.
"""
supported = True
try:
subprocess.run(['efibootmgr', '--json'], stdout=subprocess.DEVNULL,
stderr=subprocess.DEVNULL, check=True)
except subprocess.CalledProcessError:
supported = False
return supported
def is_uefi_supported():
return True if os.path.exists("/sys/firmware/efi") else False
def run_efibootmgr_json():
if _check_efibootmgr_json() is False:
raise RuntimeError('efibootmgr does not support --json option')
proc = subprocess.run(['efibootmgr', '--json'], capture_output=True, text=True)
dict_json = json.loads(proc.stdout)
return dict_json
def efibootmgr_bootnext(description):
logging.info(f'Setting BootNext to value from boot entry with label {description}')
bootnext_cmd = 'efibootmgr -n {bootnum}'
boot_entries = run_efibootmgr_json().get('vars', [])
entry = _find_bootentry(boot_entries, description)
num = _strip_boot_prefix(entry) # efibootmgr output uses BootXXXX for each entry, remove the "Boot" prefix.
bootnext_cmd = bootnext_cmd.format(bootnum=num)
subprocess.run(shlex.split(bootnext_cmd), check=True,
stdout=subprocess.DEVNULL)
def efibootmgr_delete_bootentry(label):
dict_json = run_efibootmgr_json()
efibootmgr_cmd = 'efibootmgr -q -b {bootnum} -B'
for entry in dict_json.get('vars', []):
if entry['description'] == label:
num = entry['name'][4:] # Remove "Boot" prefix to extract num
efibootmgr_cmd = efibootmgr_cmd.format(bootnum=num)
subprocess.run(shlex.split(efibootmgr_cmd), check=True)
break
else:
logging.info(f'Cannot delete boot entry {label} because it was not found.')
def efibootmgr_create_bootentry(disk, part, loader, label, add_to_bootorder=True):
entries = run_efibootmgr_json().get('vars', [])
create_opt = '-c' if add_to_bootorder else '-C'
index_opt = f'-I {len(entries)}' # Requires efibootmgr version 18
efibootmgr_cmd = f'efibootmgr {create_opt} {index_opt} -d {disk} -p {part} -L {label} -l {loader}'
logging.info(f'efibootmgr command creating boot entry: {efibootmgr_cmd}')
try:
proc = subprocess.run(shlex.split(efibootmgr_cmd), check=True, text=True)
except subprocess.CalledProcessError as e:
logging.error(f'Unexpected error adding boot entry to nvram. UEFI firmware might be buggy?.')
raise e
|
