mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 15:36:27 +00:00
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
from base import Task
|
|
from common import phases
|
|
from common.tasks import apt
|
|
from common.fs.loopbackvolume import LoopbackVolume
|
|
|
|
|
|
class ConfigureGrub(Task):
|
|
description = 'Configuring grub'
|
|
phase = phases.system_modification
|
|
after = [apt.AptUpgrade]
|
|
|
|
def run(self, info):
|
|
import os
|
|
from common.tools import log_check_call
|
|
|
|
boot_dir = os.path.join(info.root, 'boot')
|
|
grub_dir = os.path.join(boot_dir, 'grub')
|
|
|
|
from base.fs.partitionmaps.none import NoPartitions
|
|
from base.fs.partitionmaps.gpt import GPTPartitionMap
|
|
p_map = info.volume.partition_map
|
|
|
|
def remount(fn):
|
|
# GRUB cannot deal with installing to loopback devices
|
|
# so we fake a real harddisk with dmsetup.
|
|
# Guide here: http://ebroder.net/2009/08/04/installing-grub-onto-a-disk-image/
|
|
info.volume.unmount_specials()
|
|
if hasattr(p_map, 'boot'):
|
|
boot_dir = p_map.boot.mount_dir
|
|
p_map.boot.unmount()
|
|
p_map.root.unmount()
|
|
if not isinstance(p_map, NoPartitions):
|
|
p_map.unmap(info.volume)
|
|
fn()
|
|
p_map.map(info.volume)
|
|
else:
|
|
fn()
|
|
p_map.root.device_path = info.volume.device_path
|
|
p_map.root.mount(info.root)
|
|
if hasattr(p_map, 'boot'):
|
|
p_map.boot.mount(boot_dir)
|
|
info.volume.mount_specials()
|
|
|
|
if isinstance(info.volume, LoopbackVolume):
|
|
remount(info.volume.link_dm_node)
|
|
try:
|
|
[device_path] = log_check_call(['readlink', '-f', info.volume.device_path])
|
|
device_map_path = os.path.join(grub_dir, 'device.map')
|
|
partition_prefix = 'msdos'
|
|
if isinstance(p_map, GPTPartitionMap):
|
|
partition_prefix = 'gpt'
|
|
with open(device_map_path, 'w') as device_map:
|
|
device_map.write('(hd0) {device_path}\n'.format(device_path=device_path))
|
|
if not isinstance(p_map, NoPartitions):
|
|
for idx, partition in enumerate(info.volume.partition_map.partitions):
|
|
[partition_path] = log_check_call(['readlink', '-f', partition.device_path])
|
|
device_map.write('(hd0,{prefix}{idx}) {device_path}\n'
|
|
.format(device_path=partition_path, prefix=partition_prefix, idx=idx+1))
|
|
|
|
# Install grub
|
|
log_check_call(['/usr/sbin/chroot', info.root,
|
|
'/usr/sbin/grub-install',
|
|
# '--root-directory=' + info.root,
|
|
# '--boot-directory=' + boot_dir,
|
|
device_path])
|
|
log_check_call(['/usr/sbin/chroot', info.root, '/usr/sbin/update-grub'])
|
|
except Exception as e:
|
|
if isinstance(info.volume, LoopbackVolume):
|
|
remount(info.volume.unlink_dm_node)
|
|
raise e
|
|
|
|
if isinstance(info.volume, LoopbackVolume):
|
|
remount(info.volume.unlink_dm_node)
|