Skip to content

Commit

Permalink
create abupdate.conf
Browse files Browse the repository at this point in the history 
Example:

root@photon-machine [ ~ ]# cat /etc/abupdate.conf
BOOT_TYPE=BOTH
_ROOT=(f6b51480-10ec-4b87-b74c-29866643bec3 ebaa3d90-4724-48c0-8a2c-8791b1d60256 /)
EFI=(802fb52e-0d4e-4be1-ad8d-4997b8348f6a 8172187f-c4d2-4470-9b9a-2adffa11a888 /boot/efi)
SETS=(_ROOT EFI)

Change-Id: I34b352c77d655c5936368efce9f3e7c5e0000337
  • Loading branch information
@oliverkurth
oliverkurth committed Feb 23, 2023
1 parent 63edd85 commit 589e184
Showing 1 changed file with 93 additions and 24 deletions.
117 changes: 93 additions & 24 deletions photon_installer/installer.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -154,8 +154,13 @@ def configure(self, install_config, ui_config=None):

self.install_config = install_config

self.ab_present = self._is_ab_present()

self._insert_boot_partitions()

self._add_shadow_partitions()


def execute(self):
if 'setup_grub_script' in self.install_config:
self.setup_grub_command = self.install_config['setup_grub_script']
Expand Down Expand Up @@ -333,6 +338,8 @@ def _check_install_config(self, install_config):
mountpoints.append(mntpoint)
if mntpoint == '/boot' and 'lvm' in partition:
return "/boot on LVM is not supported"
elif mntpoint == '/boot/efi' and partition['filesystem'] != 'vfat':
return "/boot/efi filesystem must be vfat"
elif mntpoint == '/':
has_root = True
if not has_root:
Expand Down Expand Up @@ -360,23 +367,31 @@ def _check_install_config(self, install_config):
return None


def _is_ab_present(self):
partitions = self.install_config['partitions']
for partition in partitions:
if 'lvm' not in partition and partition.get('ab', False):
return True


def _add_shadow_partitions(self):
"""
Add shadow partitions (copy those with 'ab' = true) to list of partitions
Both will have 'ab' = True
Shadow will have 'shadow'==True, the active one will have 'shadow'==False
"""
shadow_parts = []
partitions = self.install_config['partitions']
for partition in partitions:
if 'lvm' not in partition and partition.get('ab', False):
shadow_part = copy.deepcopy(partition)
shadow_part['shadow'] = True
partition['shadow'] = False
shadow_parts.append(shadow_part)
self.ab_present = True
if self.ab_present:
shadow_parts = []
partitions = self.install_config['partitions']
for partition in partitions:
if 'lvm' not in partition and partition.get('ab', False):
shadow_part = copy.deepcopy(partition)
shadow_part['shadow'] = True
partition['shadow'] = False
shadow_parts.append(shadow_part)
self.ab_present = True

partitions.extend(shadow_parts)
partitions.extend(shadow_parts)


def _install(self, stdscreen=None):
Expand Down Expand Up @@ -451,6 +466,7 @@ def _unsafe_install(self):
self._cleanup_install_repo()
self._setup_grub()
self._create_fstab()
self._update_abupdate()
self._execute_modules(modules.commons.POST_INSTALL)
self._deactivate_network_in_chroot()
self._unmount_all()
Expand Down Expand Up @@ -587,8 +603,8 @@ def _create_fstab(self, fstab_path=None):
ptype = self._get_partition_type(partition)
if ptype == PartitionType.BIOS:
continue
# if partition.get('shadow', False):
# continue
if partition.get('shadow', False):
continue

options = 'defaults'
dump = 1
Expand Down Expand Up @@ -623,9 +639,6 @@ def _create_fstab(self, fstab_path=None):
if not mnt_src:
raise RuntimeError("Cannot get PARTUUID/UUID of: {}".format(path))

if partition.get('shadow', False):
mnt_src = "# " + mnt_src

fstab_file.write("{}\t{}\t{}\t{}\t{}\t{}\n".format(
mnt_src,
mountpoint,
Expand All @@ -641,6 +654,55 @@ def _create_fstab(self, fstab_path=None):
# Add the cdrom entry
fstab_file.write("/dev/cdrom\t/mnt/cdrom\tiso9660\tro,noauto\t0\t0\n")


def _update_abupdate(self):
if not self.ab_present:
return

abupdate_conf = os.path.join(self.photon_root, "etc/abupdate.conf")

boot_map = {'efi':'EFI', 'bios':'BIOS', 'dualboot':'BOTH'}
bootmode = self.install_config['bootmode']
if not bootmode in boot_map:
raise Exception(f"invalid boot mode '{bootmode}'")

ab_map = {}
for partition in self.install_config['partitions']:
ptype = self._get_partition_type(partition)
if ptype == PartitionType.BIOS:
continue
if partition.get('ab', False):
mntpoint = partition['mountpoint']
partuuid = self._get_partuuid(partition['path'])

if mntpoint == '/boot/efi':
name = 'EFI'
elif mntpoint == '/':
name = '_ROOT'
else:
name = mntpoint[1:].upper().replace('/', '_')

# we go through this twice - active and shadow
# only add entry once
if not name in ab_map:
ab_map[name] = {'mntpoint' : mntpoint}

if partition.get('shadow', False):
ab_map[name]['shadow'] = partuuid
else:
ab_map[name]['active'] = partuuid

# assuming a virgin file with no settings, or no file
with open(abupdate_conf, 'a') as f:
f.write(f"BOOT_TYPE={boot_map[bootmode]}\n")

for name, ab in ab_map.items():
f.write(f"{name}=({ab['active']} {ab['shadow']} {ab['mntpoint']})\n")

sets = " ".join(ab_map.keys())
f.write(f"SETS=({sets})\n")


def _generate_partitions_param(self, reverse=False):
"""
Generate partition param for mount command
Expand Down Expand Up @@ -1265,34 +1327,42 @@ def _get_partition_tree_view(self):

return ptv


def _insert_boot_partitions(self):
bios_found = False
esp_found = False
for partition in self.install_config['partitions']:
partitions = self.install_config['partitions']

for partition in partitions:
ptype = self._get_partition_type(partition)
if ptype == PartitionType.BIOS:
bios_found = True
if ptype == PartitionType.ESP:
esp_found = True
efi_partition = partition

# Adding boot partition required for ostree if already not present in partitions table
if 'ostree' in self.install_config:
mount_points = [partition['mountpoint'] for partition in self.install_config['partitions'] if 'mountpoint' in partition]
mount_points = [partition['mountpoint'] for partition in partitions if 'mountpoint' in partition]
if '/boot' not in mount_points:
boot_partition = {'size': 300, 'filesystem': 'ext4', 'mountpoint': '/boot'}
self.install_config['partitions'].insert(0, boot_partition)
partitions.insert(0, boot_partition)

bootmode = self.install_config.get('bootmode', 'bios')

# Insert efi special partition
if not esp_found and (bootmode == 'dualboot' or bootmode == 'efi'):
efi_partition = {'size': ESPSIZE, 'filesystem': 'vfat', 'mountpoint': '/boot/efi'}
self.install_config['partitions'].insert(0, efi_partition)
if bootmode == 'dualboot' or bootmode == 'efi':
# Insert efi special partition
if not esp_found:
efi_partition = {'size': ESPSIZE, 'filesystem': 'vfat', 'mountpoint': '/boot/efi'}
partitions.insert(0, efi_partition)

if self.ab_present:
efi_partition['ab'] = True

# Insert bios partition last to be very first
if not bios_found and (bootmode == 'dualboot' or bootmode == 'bios'):
bios_partition = {'size': BIOSSIZE, 'filesystem': 'bios'}
self.install_config['partitions'].insert(0, bios_partition)
partitions.insert(0, bios_partition)


def __set_ab_partition_size(self, l2entries, used_size, total_disk_size):
Expand Down Expand Up @@ -1333,7 +1403,6 @@ def _partition_disks(self):
if self.install_config['ui']:
self.progress_bar.update_message('Partitioning...')

self._insert_boot_partitions()
ptv = self._get_partition_tree_view()

self.__ptv_update_partition_sizes(ptv)
Expand Down

0 comments on commit 589e184

Please sign in to comment.