mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-22 09:50:37 +00:00
a5e8aae7b0
In each of these cases we have an un-escaped \n or \t that gets passed to a regex. That means that the regexes actually contain tabs or newlines, respectively. After this change an actual '\t' and '\n' are passed in instead. I have not tested this! I am writing this patch based on the assumption that /surely/ the original authors meant to pass in \t and \n to the regex and not actual tab or newline characters.
84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
from bootstrapvz.base import Task
|
|
from bootstrapvz.common import phases
|
|
from bootstrapvz.common.tasks import grub
|
|
from . import assets
|
|
import os
|
|
from bootstrapvz.common.tools import log_check_call
|
|
|
|
|
|
class AddXenGrubConsoleOutputDevice(Task):
|
|
description = 'Adding XEN `hvc0\' as output device for grub'
|
|
phase = phases.system_modification
|
|
successors = [grub.WriteGrubConfig]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
info.grub_config['GRUB_CMDLINE_LINUX_DEFAULT'].append('console=hvc0')
|
|
|
|
|
|
class UpdateGrubConfig(Task):
|
|
description = 'Updating the grub config'
|
|
phase = phases.system_modification
|
|
successors = [grub.WriteGrubConfig]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
log_check_call(['chroot', info.root, 'update-grub'])
|
|
|
|
|
|
class CreatePVGrubCustomRule(Task):
|
|
description = 'Creating special rule for PVGrub'
|
|
phase = phases.system_modification
|
|
successors = [UpdateGrubConfig]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
import stat
|
|
x_all = stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH
|
|
|
|
grubd = os.path.join(info.root, 'etc/grub.d')
|
|
for cfg in [os.path.join(grubd, f) for f in os.listdir(grubd)]:
|
|
os.chmod(cfg, os.stat(cfg).st_mode & ~ x_all)
|
|
|
|
from shutil import copy
|
|
script_src = os.path.join(assets, 'grub.d/40_custom')
|
|
script_dst = os.path.join(info.root, 'etc/grub.d/40_custom')
|
|
copy(script_src, script_dst)
|
|
os.chmod(script_dst, 0o755)
|
|
|
|
from bootstrapvz.base.fs.partitionmaps.none import NoPartitions
|
|
if not isinstance(info.volume.partition_map, NoPartitions):
|
|
from bootstrapvz.common.tools import sed_i
|
|
root_idx = info.volume.partition_map.root.get_index()
|
|
grub_device = 'GRUB_DEVICE=/dev/xvda' + str(root_idx)
|
|
sed_i(script_dst, '^GRUB_DEVICE=/dev/xvda$', grub_device)
|
|
grub_root = '\troot (hd0,{idx})'.format(idx=root_idx - 1)
|
|
sed_i(script_dst, r'^\troot \(hd0\)$', grub_root)
|
|
|
|
if info.manifest.volume['backing'] == 's3':
|
|
from bootstrapvz.common.tools import sed_i
|
|
sed_i(script_dst, '^GRUB_DEVICE=/dev/xvda$', 'GRUB_DEVICE=/dev/xvda1')
|
|
|
|
|
|
class ConfigurePVGrub(Task):
|
|
description = 'Configuring PVGrub'
|
|
phase = phases.system_modification
|
|
successors = [UpdateGrubConfig]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
info.grub_config['GRUB_CMDLINE_LINUX'].extend([
|
|
'consoleblank=0',
|
|
'elevator=noop',
|
|
])
|
|
|
|
|
|
class LinkGrubConfig(Task):
|
|
description = 'Linking the grub config to /boot/grub/menu.lst'
|
|
phase = phases.system_modification
|
|
predecessors = [UpdateGrubConfig]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
log_check_call(['chroot', info.root,
|
|
'ln', '--symbolic', '/boot/grub/grub.cfg', '/boot/grub/menu.lst'])
|