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
98
99
100
101
102
103
104
105
106
107
|
"""
Wrapper for Extlinux support
"""
# -*- coding: utf-8 -*-
#
# extlinux.py
#
# Copyright 2015 Neil Williams <codehelp@debian.org>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import re
import os
import time
import logging
import cliapp
from vmdebootstrap.base import Base, runcmd
# pylint: disable=missing-docstring
class ExtLinux(Base):
name = 'extlinux'
def __init__(self):
super(ExtLinux, self).__init__()
def install_extlinux(self, rootdev, rootdir):
if not os.path.exists("/usr/bin/extlinux"):
self.message("extlinux not installed, skipping.")
return
self.message('Installing extlinux')
def find(pattern):
dirname = os.path.join(rootdir, 'boot')
basenames = os.listdir(dirname)
logging.debug('find: %s', basenames)
for basename in basenames:
if re.search(pattern, basename):
return os.path.join('boot', basename)
raise cliapp.AppException('Cannot find match: %s' % pattern)
try:
kernel_image = find('vmlinuz-.*')
initrd_image = find('initrd.img-.*')
except cliapp.AppException as exc:
self.message("Unable to find kernel. Not installing extlinux.")
logging.debug("No kernel found. %s. Skipping install of extlinux.", exc)
return
out = runcmd(['blkid', '-c', '/dev/null', '-o', 'value',
'-s', 'UUID', rootdev])
uuid = out.splitlines()[0].strip()
conf = os.path.join(rootdir, 'extlinux.conf')
logging.debug('configure extlinux %s', conf)
kserial = 'console=ttyS0,115200' if self.settings['serial-console'] else ''
extserial = 'serial 0 115200' if self.settings['serial-console'] else ''
msg = '''
default linux
timeout 1
label linux
kernel %(kernel)s
append initrd=%(initrd)s root=UUID=%(uuid)s ro %(kserial)s
%(extserial)s
''' % {
'kernel': kernel_image, # pylint: disable=bad-continuation
'initrd': initrd_image, # pylint: disable=bad-continuation
'uuid': uuid, # pylint: disable=bad-continuation
'kserial': kserial, # pylint: disable=bad-continuation
'extserial': extserial, # pylint: disable=bad-continuation
} # pylint: disable=bad-continuation
logging.debug("extlinux config:\n%s", msg)
# python multiline string substitution is just ugly.
# use an external file or live with the mangling, no point in
# mangling the string to remove spaces just to keep it pretty in source.
ext_f = open(conf, 'w')
ext_f.write(msg)
runcmd(['extlinux', '--install', rootdir])
runcmd(['sync'])
time.sleep(2)
def install_mbr(self):
if not self.settings['mbr'] and not self.settings['extlinux']:
return
if os.path.exists("/sbin/install-mbr"):
self.message('Installing MBR')
runcmd(['install-mbr', self.settings['image']])
else:
msg = "mbr enabled but /sbin/install-mbr not found" \
" - please install the mbr package."
raise cliapp.AppException(msg)
|