treebuilder.py: uses templates to create trees/images
TreeBuilder uses templates full of commands (like ramdisk.ltmpl) to create the output tree and boot images. There are 4 arch-specific templates, plus a bonus EFI template which can handle EFI image creation for any arch that implements EFI.
This commit is contained in:
Will Woods
parent
6a5851d53b
commit
1350cd028f
37
share/efi.tmpl
Normal file
37
share/efi.tmpl
Normal file
@ -0,0 +1,37 @@
|
|||||||
|
<%page args="ANABOOTDIR, KERNELDIR, efiarch"/>
|
||||||
|
<% EFIBOOTDIR="EFI/BOOT" %>
|
||||||
|
|
||||||
|
mkdir ${EFIBOOTDIR}
|
||||||
|
install boot/efi/EFI/redhat/grub.efi ${EFIBOOTDIR}/BOOT${efiarch}.bin
|
||||||
|
install boot/grub/splash.xml.gz ${EFIBOOTDIR}
|
||||||
|
|
||||||
|
## actually make the EFI images
|
||||||
|
${make_efiboot("images/efidisk.img", include_kernel=True, disk=True)}
|
||||||
|
${make_efiboot("images/efiboot.img", include_kernel=False)}
|
||||||
|
|
||||||
|
## This is kinda gross, but then... so's EFI.
|
||||||
|
<%def name="make_efiboot(img, include_kernel, disk=False)">
|
||||||
|
<%
|
||||||
|
kdir = EFIBOOTDIR if include_kernel else KERNELDIR
|
||||||
|
args = "--label=ANACONDA"
|
||||||
|
if disk: args += " --disk"
|
||||||
|
%>
|
||||||
|
%if include_kernel:
|
||||||
|
copy ${KERNELDIR}/vmlinuz ${EFIBOOTDIR}
|
||||||
|
copy ${KERNELDIR}/initrd ${EFIBOOTDIR}
|
||||||
|
%endif
|
||||||
|
install ${ANABOOTDIR}/grub.conf ${EFIBOOTDIR}/BOOT${efiarch}.conf
|
||||||
|
replace @PRODUCT@ ${product.name} ${EFIBOOTDIR}/BOOT${efiarch}.conf
|
||||||
|
replace @VERSION@ ${product.version} ${EFIBOOTDIR}/BOOT${efiarch}.conf
|
||||||
|
replace @KERNELPATH@ /${kdir}/vmlinuz ${EFIBOOTDIR}/BOOT${efiarch}.conf
|
||||||
|
replace @INITRDPATH@ /${kdir}/initrd ${EFIBOOTDIR}/BOOT${efiarch}.conf
|
||||||
|
replace @SPLASHPATH@ /EFI/BOOT/splash.xpm.gz ${EFIBOOTDIR}/BOOT${efiarch}.conf
|
||||||
|
%if efiarch == 'IA32':
|
||||||
|
copy ${EFIBOOTDIR}/BOOT${efiarch}.conf ${EFIBOOTDIR}/BOOT.conf
|
||||||
|
%endif
|
||||||
|
runcmd mkefiboot ${args} ${outroot}/${EFIBOOTDIR} ${outroot}/${img}
|
||||||
|
%if include_kernel:
|
||||||
|
remove ${EFIBOOTDIR}/vmlinuz
|
||||||
|
remove ${EFIBOOTDIR}/initrd
|
||||||
|
%endif
|
||||||
|
</%def>
|
||||||
99
share/ppc.tmpl
Normal file
99
share/ppc.tmpl
Normal file
@ -0,0 +1,99 @@
|
|||||||
|
<%
|
||||||
|
ANABOOTDIR="usr/share/anaconda/boot"
|
||||||
|
BOOTDIR="ppc"
|
||||||
|
MACDIR=BOOTDIR+"/mac"
|
||||||
|
NETBOOTDIR="images/netboot"
|
||||||
|
|
||||||
|
MKZIMAGE="usr/bin/mkzimage"
|
||||||
|
ZSTUB="usr/share/ppc64-utils/zImage.stub"
|
||||||
|
WRAPPER="usr/sbin/wrapper"
|
||||||
|
WRAPPER_A="usr/"+libdir+"/kernel-wrapper/wrapper.a"
|
||||||
|
MAPPING=ANABOOTDIR+"/mapping"
|
||||||
|
MAGIC=ANABOOTDIR+"/magic"
|
||||||
|
bitsizes = set()
|
||||||
|
prepboot = ""
|
||||||
|
macboot = ""
|
||||||
|
%>
|
||||||
|
|
||||||
|
mkdir ${BOOTDIR} ${BOOTDIR}/chrp etc
|
||||||
|
install ${ANABOOTDIR}/bootinfo.txt ${BOOTDIR}
|
||||||
|
install boot/efika.forth ${BOOTDIR}
|
||||||
|
install usr/lib/yaboot/yaboot ${BOOTDIR}/chrp
|
||||||
|
|
||||||
|
## Mac boot stuff
|
||||||
|
mkdir ${MACDIR}
|
||||||
|
install usr/lib/yaboot/yaboot ${MACDIR}
|
||||||
|
install ofboot.b ${MACDIR}
|
||||||
|
<%
|
||||||
|
macboot = "-hfs-volid {0}".format(product.version)
|
||||||
|
macboot += "-hfs-bless {0}/isopath/{1}".format(outroot,MACDIR)
|
||||||
|
%>
|
||||||
|
|
||||||
|
%for kernel in kernels:
|
||||||
|
<%
|
||||||
|
bits = 64 if kernel.arch == "ppc64" else 32
|
||||||
|
KERNELDIR=BOOTDIR+"/ppc%s" % bits
|
||||||
|
NETIMG=NETBOOTDIR+"/ppc%s.img" % bits
|
||||||
|
bitsizes.add(bits)
|
||||||
|
%>
|
||||||
|
mkdir ${KERNELDIR}
|
||||||
|
install ${ANABOOTDIR}/yaboot.conf.in ${KERNELDIR}/yaboot.conf
|
||||||
|
installkernel images-${kernel.arch} ${kernel.path} ${KERNELDIR}/vmlinuz
|
||||||
|
## Note: this used to be ramdisk.img.gz
|
||||||
|
installinitrd images-${kernel.arch} ${kernel.initrd.path} ${KERNELDIR}/initrd.img
|
||||||
|
|
||||||
|
replace %PRODUCT% ${product.name} ${KERNELDIR}/yaboot.conf
|
||||||
|
replace %VERSION% ${product.version} ${KERNELDIR}/yaboot.conf
|
||||||
|
replace %BITS% ${bits} ${KERNELDIR}/yaboot.conf
|
||||||
|
|
||||||
|
## Weirdo wrapper junk that makes the netboot combined ppc{32,64}.img
|
||||||
|
%if exists(MKZIMAGE) and exists(ZSTUB):
|
||||||
|
copy usr/${libdir}/kernel-wrapper/zImage.lds ${KERNELDIR}
|
||||||
|
runcmd chdir=${KERNELDIR} ${inroot}/${MKZIMAGE} vmlinuz no no initrd.img \
|
||||||
|
${inroot}/${ZSTUB} ${outroot}/${NETIMG}
|
||||||
|
remove ${KERNELDIR}/zImage.lds
|
||||||
|
treeinfo images-${kernel.arch} zimage ${NETIMG}
|
||||||
|
%elif exists(WRAPPER) and exists(WRAPPER_A):
|
||||||
|
runcmd chdir=${KERNELDIR} ${inroot}/${WRAPPER} -o ${outroot}/${NETIMG} \
|
||||||
|
-i initrd.img -D ${inroot}/${os.path.dirname(WRAPPER_A)} vmlinuz
|
||||||
|
treeinfo images-${kernel.arch} zimage ${NETIMG}
|
||||||
|
%endif
|
||||||
|
%if exists(NETIMG) and bits == 32:
|
||||||
|
<% prepboot="-prep-boot " + NETIMG %>
|
||||||
|
%endif
|
||||||
|
%endfor
|
||||||
|
|
||||||
|
%if not exists(NETBOOTDIR+"/*.img"):
|
||||||
|
remove ${NETBOOTDIR}
|
||||||
|
%endif
|
||||||
|
|
||||||
|
runcmd usr/lib/yaboot/addnote ${outroot}/${BOOTDIR}/chrp/yaboot
|
||||||
|
|
||||||
|
%if len(bitsizes) == 2:
|
||||||
|
## magic ppc biarch tree! we need magic ppc biarch config.
|
||||||
|
install ${ANABOOTDIR}/yaboot.conf.3264 etc/yaboot.conf
|
||||||
|
replace %PRODUCT% ${product.name} etc/yaboot.conf
|
||||||
|
replace %VERSION% ${product.version} etc/yaboot.conf
|
||||||
|
replace %BITS% 32 etc/yaboot.conf
|
||||||
|
%else:
|
||||||
|
copy ${KERNELDIR}/yaboot.conf etc/yaboot.conf
|
||||||
|
%endif
|
||||||
|
|
||||||
|
## XXX why don't we use graft-points here?
|
||||||
|
## is it because of the scary warnings in mkisofs(1)?
|
||||||
|
mkdir isopath
|
||||||
|
copy ${BOOTDIR} isopath
|
||||||
|
copy etc isopath
|
||||||
|
runcmd mkisofs -o ${outroot}/images/boot.iso -chrp-boot -U \
|
||||||
|
${prepboot} -part -hfs -T -r -l -J \
|
||||||
|
-A "${product.name} ${product.version}" -sysid PPC -V "PBOOT" \
|
||||||
|
-volset "${product.version}" -volset-size 1 -volset-seqno 1 \
|
||||||
|
${macboot} -map ${MAPPING} -magic ${MAGIC} \
|
||||||
|
-no-desktop -allow-multidot -graft-points ${outroot}/isopath
|
||||||
|
remove isopath
|
||||||
|
%if len(bitsizes) == 2:
|
||||||
|
treeinfo images-ppc boot.iso images/boot.iso
|
||||||
|
treeinfo images-ppc64 boot.iso images/boot.iso
|
||||||
|
%else:
|
||||||
|
treeinfo images-${kernel.arch} boot.iso images/boot.iso
|
||||||
|
%fi
|
||||||
26
share/s390.tmpl
Normal file
26
share/s390.tmpl
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
<%
|
||||||
|
ANABOOTDIR="usr/share/anaconda/boot"
|
||||||
|
BOOTDIR="images"
|
||||||
|
KERNELDIR=BOOTDIR
|
||||||
|
INITRD_ADDRESS="0x02000000"
|
||||||
|
MKCDBOOT="usr/libexec/anaconda/mk-s390-cdboot"
|
||||||
|
# The assumption seems to be that there is only one s390 kernel, ever
|
||||||
|
kernel = kernels[0]
|
||||||
|
%>
|
||||||
|
|
||||||
|
install ${ANABOOTDIR}/redhat.exec ${BOOTDIR}
|
||||||
|
install ${ANABOOTDIR}/generic.prm ${BOOTDIR}
|
||||||
|
install ${ANABOOTDIR}/generic.ins .
|
||||||
|
|
||||||
|
replace @INITRD_LOAD_ADDRESS@ ${INITRD_ADDRESS} generic.ins
|
||||||
|
|
||||||
|
installkernel images-${basearch} ${kernel.path} ${KERNELDIR}/kernel.img
|
||||||
|
installinitrd images-${basearch} ${kernel.initrd.path} ${KERNELDIR}/initrd.img
|
||||||
|
runcmd usr/libexec/anaconda/addrsize ${INITRD_ADDRESS} ${KERNELDIR}/initrd.img ${outroot}/${BOOTDIR}/initrd_addrsize
|
||||||
|
treeinfo images-${basearch} initrd.addrsize ${BOOTDIR}/initrd_addrsize
|
||||||
|
treeinfo images-${basearch} generic.prm ${BOOTDIR}/generic.prm
|
||||||
|
treeinfo images-${basearch} generic.ins generic.ins
|
||||||
|
|
||||||
|
runcmd ${MKCDBOOT} -i ${kernel.path} -r ${kernel.initrd.path} \
|
||||||
|
-p ${outroot}/${BOOTDIR}/generic.prm \
|
||||||
|
-o ${outroot}/${BOOTDIR}/cdboot.img
|
||||||
26
share/sparc.tmpl
Normal file
26
share/sparc.tmpl
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
<%
|
||||||
|
ANABOOTDIR="usr/share/anaconda/boot"
|
||||||
|
BOOTDIR="boot"
|
||||||
|
KERNELDIR=BOOTDIR
|
||||||
|
%>
|
||||||
|
|
||||||
|
install boot/*.b ${BOOTDIR}
|
||||||
|
install ${ANABOOTDIR}/silo.conf ${BOOTDIR}
|
||||||
|
install ${ANABOOTDIR}/boot.msg ${BOOTDIR}/boot.msg
|
||||||
|
|
||||||
|
replace %VERSION% ${product.version} ${BOOTDIR}/boot.msg
|
||||||
|
replace %PRODUCT% ${product.name} ${BOOTDIR}/boot.msg
|
||||||
|
|
||||||
|
%for kernel in kernels:
|
||||||
|
installkernel images-${basearch} ${kernel.path} ${KERNELDIR}/vmlinuz
|
||||||
|
installinitrd images-${basearch} ${kernel.initrd.path} ${KERNELDIR}/initrd.img
|
||||||
|
%endfor
|
||||||
|
|
||||||
|
runcmd mkisofs -R -J -T -G /${BOOTDIR}/isofs.b -B ... \
|
||||||
|
-s /${BOOTDIR}/silo.conf -r -V "PBOOT" \
|
||||||
|
-A "${product.name} ${product.version}" \
|
||||||
|
-x Fedora -x repodata \
|
||||||
|
-sparc-label "${product.name} ${product.version} Boot Disc" \
|
||||||
|
-o ${outroot}/images/boot.iso \
|
||||||
|
-graft-points boot=${BOOTDIR}
|
||||||
|
treeinfo images-${basearch} boot.iso images/boot.iso
|
||||||
68
share/x86.tmpl
Normal file
68
share/x86.tmpl
Normal file
@ -0,0 +1,68 @@
|
|||||||
|
<%
|
||||||
|
ANABOOTDIR="usr/share/anaconda/boot"
|
||||||
|
SYSLINUXDIR="usr/share/syslinux"
|
||||||
|
PXEBOOTDIR="images/pxeboot"
|
||||||
|
BOOTDIR="isolinux"
|
||||||
|
KERNELDIR=PXEBOOTDIR
|
||||||
|
%>
|
||||||
|
|
||||||
|
mkdir ${BOOTDIR} ${KERNELDIR}
|
||||||
|
install ${SYSLINUXDIR}/isolinux.bin ${BOOTDIR}
|
||||||
|
install ${ANABOOTDIR}/syslinux.cfg ${BOOTDIR}/isolinux.cfg
|
||||||
|
install ${ANABOOTDIR}/syslinux-vesa-splash.jpg ${BOOTDIR}/splash.jpg
|
||||||
|
install ${ANABOOTDIR}/*.msg ${BOOTDIR}
|
||||||
|
install ${SYSLINUXDIR}/vesamenu.c32 ${BOOTDIR}
|
||||||
|
install ${ANABOOTDIR}/grub.conf ${BOOTDIR}
|
||||||
|
|
||||||
|
replace @VERSION@ ${product.version} ${BOOTDIR}/*.msg ${BOOTDIR}/grub.conf
|
||||||
|
replace @PRODUCT@ ${product.name} ${BOOTDIR}/grub.conf
|
||||||
|
|
||||||
|
replace "default linux" "default vesamenu.c32" ${BOOTDIR}/isolinux.cfg
|
||||||
|
replace "prompt 1" "#prompt 1" ${BOOTDIR}/isolinux.cfg
|
||||||
|
|
||||||
|
%if exists("boot/memtest*"):
|
||||||
|
install boot/memtest* ${BOOTDIR}
|
||||||
|
append ${BOOTDIR}/isolinux.cfg "label memtest86"
|
||||||
|
append ${BOOTDIR}/isolinux.cfg " menu label ^Memory test"
|
||||||
|
append ${BOOTDIR}/isolinux.cfg " kernel memtest"
|
||||||
|
append ${BOOTDIR}/isolinux.cfg " append -"
|
||||||
|
%endif
|
||||||
|
|
||||||
|
%for kernel in kernels:
|
||||||
|
%if kernel.flavor:
|
||||||
|
installkernel images-xen ${kernel.path} ${KERNELDIR}/vmlinuz-${kernel.flavor}
|
||||||
|
installinitrd images-xen ${kernel.initrd.path} ${KERNELDIR}/initrd-${kernel.flavor}.img
|
||||||
|
%else:
|
||||||
|
installkernel images-${basearch} ${kernel.path} ${KERNELDIR}/vmlinuz
|
||||||
|
installinitrd images-${basearch} ${kernel.initrd.path} ${KERNELDIR}/initrd.img
|
||||||
|
%endif
|
||||||
|
%endfor
|
||||||
|
|
||||||
|
hardlink ${KERNELDIR}/vmlinuz ${BOOTDIR}
|
||||||
|
hardlink ${KERNELDIR}/initrd ${BOOTDIR}
|
||||||
|
%if basearch == 'x86_64':
|
||||||
|
treeinfo images-xen kernel ${KERNELDIR}/vmlinuz
|
||||||
|
treeinfo images-xen initrd ${KERNELDIR}/initrd.img
|
||||||
|
%endif
|
||||||
|
|
||||||
|
## WHeeeeeeee, EFI.
|
||||||
|
## We could remove the basearch restriction someday..
|
||||||
|
<% efiargs=""; efigraft="" %>
|
||||||
|
%if exists("boot/efi/EFI/redhat/grub.efi") and basearch != 'i386':
|
||||||
|
<%
|
||||||
|
efiarch = 'X64' if basearch=='x86_64' else 'IA32'
|
||||||
|
efiargs="-eltorito-alt-boot -e images/efiboot.img -no-emul-boot"
|
||||||
|
efigraft="BOOT/EFI={0}/BOOT/EFI".format(outroot)
|
||||||
|
%>
|
||||||
|
<%include file="efi.tmpl" args="ANABOOTDIR=ANABOOTDIR, KERNELDIR=KERNELDIR, efiarch=efiarch"/>
|
||||||
|
%endif
|
||||||
|
|
||||||
|
runcmd mkisofs -v -o ${outroot}/images/boot.iso \
|
||||||
|
-b ${BOOTDIR}/isolinux.bin -c ${BOOTDIR}/boot.cat \
|
||||||
|
-boot-load-size 4 -boot-info-table ${efiargs} \
|
||||||
|
-R -J -V '${product.name}' -T -graft-points \
|
||||||
|
${BOOTDIR}=${outroot}/${BOOTDIR} \
|
||||||
|
images=${outroot}/images \
|
||||||
|
${efigraft}
|
||||||
|
runcmd isohybrid ${outroot}/images/boot.iso
|
||||||
|
treeinfo images-${basearch} boot.iso images/boot.iso
|
||||||
268
src/pylorax/treebuilder.py
Normal file
268
src/pylorax/treebuilder.py
Normal file
@ -0,0 +1,268 @@
|
|||||||
|
# treebuilder.py - handle arch-specific tree building stuff using templates
|
||||||
|
#
|
||||||
|
# Copyright (C) 2011 Red Hat, Inc.
|
||||||
|
#
|
||||||
|
# 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 2 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/>.
|
||||||
|
#
|
||||||
|
# Author(s): Will Woods <wwoods@redhat.com>
|
||||||
|
|
||||||
|
import logging
|
||||||
|
logger = logging.getLogger("pylorax.treebuilder")
|
||||||
|
|
||||||
|
import os
|
||||||
|
import glob
|
||||||
|
from subprocess import check_call, PIPE
|
||||||
|
|
||||||
|
from sysutils import joinpaths, cpfile, replace, remove
|
||||||
|
from ltmpl import LoraxTemplate
|
||||||
|
from base import DataHolder
|
||||||
|
|
||||||
|
templatemap = {'i386': 'x86.tmpl',
|
||||||
|
'x86_64': 'x86.tmpl',
|
||||||
|
'ppc': 'ppc.tmpl',
|
||||||
|
'ppc64': 'ppc.tmpl',
|
||||||
|
'sparc': 'sparc.tmpl',
|
||||||
|
'sparc64': 'sparc.tmpl',
|
||||||
|
's390': 's390.tmpl',
|
||||||
|
's390x': 's390.tmpl',
|
||||||
|
}
|
||||||
|
|
||||||
|
def findkernels(root="/", kdir="boot"):
|
||||||
|
# To find flavors, awk '/BuildKernel/ { print $4 }' kernel.spec
|
||||||
|
flavors = ('debug', 'PAE', 'PAEdebug', 'smp', 'xen')
|
||||||
|
kre = re.compile(r"vmlinuz-(?P<version>.+?\.(?P<arch>[a-z0-9_]+)"
|
||||||
|
r"(\.(?P<flavor>{0}))?)$".format("|".join(flavors)))
|
||||||
|
kernels = []
|
||||||
|
for f in os.listdir(joinpaths(root, kdir)):
|
||||||
|
match = kre.match(f)
|
||||||
|
if match:
|
||||||
|
kernel = DataHolder(path=joinpaths(kdir, f))
|
||||||
|
kernel.update(match.groupdict()) # sets version, arch, flavor
|
||||||
|
kernels.append(kernel)
|
||||||
|
|
||||||
|
# look for associated initrd/initramfs
|
||||||
|
for kernel in kernels:
|
||||||
|
# NOTE: if both exist, the last one found will win
|
||||||
|
for imgname in ("initrd", "initramfs"):
|
||||||
|
i = kernel.path.replace("vmlinuz", imgname, 1) + ".img"
|
||||||
|
if os.path.exists(joinpaths(root, i)):
|
||||||
|
kernel.initrd = DataHolder(path=i)
|
||||||
|
|
||||||
|
return kernels
|
||||||
|
|
||||||
|
def _exists(root, p):
|
||||||
|
if p[0] != '/': p = joinpaths(root, p)
|
||||||
|
return (len(glob.glob(p)) > 0)
|
||||||
|
|
||||||
|
class BaseBuilder(object):
|
||||||
|
def __init__(self, product, arch, inroot, outroot):
|
||||||
|
self.arch = arch
|
||||||
|
self.product = product
|
||||||
|
self.inroot = inroot
|
||||||
|
self.outroot = outroot
|
||||||
|
self.runner = None
|
||||||
|
|
||||||
|
def getdefaults(self):
|
||||||
|
return dict(arch=self.arch, product=self.product,
|
||||||
|
inroot=self.inroot, outroot=self.outroot,
|
||||||
|
basearch=self.arch.basearch, libdir=self.arch.libdir,
|
||||||
|
exists=lambda p: _exists(self.inroot, p))
|
||||||
|
|
||||||
|
def runtemplate(self, templatefile, **variables):
|
||||||
|
for k,v in self.getdefaults():
|
||||||
|
variables.setdefault(k,v) # setdefault won't override existing args
|
||||||
|
t = LoraxTemplate()
|
||||||
|
logger.info("parsing %s with the following variables", templatefile)
|
||||||
|
for key, val in variables.items():
|
||||||
|
logger.info(" %s: %s", key, val)
|
||||||
|
template = t.parse(templatefile, variables)
|
||||||
|
self.runner = TemplateRunner(self.inroot, self.outroot, template)
|
||||||
|
logger.info("running template commands")
|
||||||
|
self.runner.run()
|
||||||
|
|
||||||
|
class TreeBuilder(BaseBuilder):
|
||||||
|
'''Builds the arch-specific boot images.
|
||||||
|
inroot should be the installtree root (the newly-built runtime dir)'''
|
||||||
|
def build(self):
|
||||||
|
self.runtemplate(templatemap[self.arch.basearch], kernels=self.kernels)
|
||||||
|
self.implantisomd5()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def treeinfo_data(self):
|
||||||
|
if self.runner:
|
||||||
|
return self.runner.treeinfo_data
|
||||||
|
|
||||||
|
@property
|
||||||
|
def kernels(self):
|
||||||
|
return findkernels(root=self.inroot)
|
||||||
|
|
||||||
|
def rebuild_initrds(self, add_args=[], backup=""):
|
||||||
|
'''Rebuild all the initrds in the tree. If backup is specified, each
|
||||||
|
initrd will be renamed with backup as a suffix before rebuilding.
|
||||||
|
If backup is empty, the existing initrd files will be overwritten.'''
|
||||||
|
dracut = ["/sbin/dracut", "--nomdadmconf", "--nolvmconf"] + add_args
|
||||||
|
if not backup:
|
||||||
|
dracut.append("--force")
|
||||||
|
for kernel in self.kernels:
|
||||||
|
if backup:
|
||||||
|
initrd = joinpaths(self.inroot, kernel.initrd.path)
|
||||||
|
os.rename(initrd, initrd + backup)
|
||||||
|
check_call(["chroot", self.inroot] + \
|
||||||
|
dracut + [kernel.initrd.path, kernel.version])
|
||||||
|
|
||||||
|
def initrd_append(rootdir):
|
||||||
|
'''Place the given files into a cpio archive and append that archive
|
||||||
|
to the initrds.'''
|
||||||
|
cpio = NamedTemporaryFile(prefix="lorax.")
|
||||||
|
mkcpio(rootdir, cpio, compression=None)
|
||||||
|
for kernel in self.kernels:
|
||||||
|
initrd = open(kernel.initrd.path, "ab")
|
||||||
|
cpio = open(cpio, "rb")
|
||||||
|
initrd.write(cpio.read())
|
||||||
|
|
||||||
|
def implantisomd5(self):
|
||||||
|
for section, data in self.treeinfo_data:
|
||||||
|
if 'boot.iso' in data:
|
||||||
|
iso = joinpaths(self.outputdir, data['boot.iso'])
|
||||||
|
check_call(["implantisomd5", iso])
|
||||||
|
|
||||||
|
|
||||||
|
# note: "install", "replace", "exists" allow globs
|
||||||
|
# "install" and "exist" assume their first argument is in inroot
|
||||||
|
# everything else operates on outroot
|
||||||
|
# "mkdir", "treeinfo", "runcmd", "remove", "replace" will take multiple args
|
||||||
|
|
||||||
|
# TODO: replace installtree. need glob(), find(glob), installpkg, removepkg, module
|
||||||
|
# also: run_transaction?
|
||||||
|
|
||||||
|
class TemplateRunner(object):
|
||||||
|
commands = ('install', 'mkdir', 'replace', 'append', 'treeinfo',
|
||||||
|
'installkernel', 'installinitrd', 'hardlink', 'symlink',
|
||||||
|
'copy', 'copyif', 'move', 'moveif', 'remove', 'chmod',
|
||||||
|
'runcmd', 'log')
|
||||||
|
|
||||||
|
def __init__(self, inroot, outroot, parsed_template, fatalerrors=False):
|
||||||
|
self.inroot = inroot
|
||||||
|
self.outroot = outroot
|
||||||
|
self.template = parsed_template
|
||||||
|
self.fatalerrors = fatalerrors
|
||||||
|
|
||||||
|
self.treeinfo_data = dict()
|
||||||
|
self.exists = lambda p: _exists(inroot, p)
|
||||||
|
|
||||||
|
def _out(self, path):
|
||||||
|
return joinpaths(self.outroot, path)
|
||||||
|
def _in(self, path):
|
||||||
|
return joinpaths(self.inroot, path)
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
for (num, line) in enumerate(self.template,1):
|
||||||
|
logger.debug("template line %i: %s", num, line)
|
||||||
|
(cmd, args) = (line[0], line[1:])
|
||||||
|
try:
|
||||||
|
if cmd not in self.commands:
|
||||||
|
raise ValueError, "unknown command %s" % cmd
|
||||||
|
# grab the method named in cmd and pass it the given arguments
|
||||||
|
f = getattr(self, cmd)
|
||||||
|
f(*args)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("template command error: %s", str(line))
|
||||||
|
if self.fatalerrors:
|
||||||
|
raise
|
||||||
|
logger.error(str(e))
|
||||||
|
|
||||||
|
def install(self, srcglob, dest):
|
||||||
|
sources = glob.glob(self._in(srcglob))
|
||||||
|
if not sources:
|
||||||
|
raise IOError, "couldn't find %s" % srcglob
|
||||||
|
for src in sources:
|
||||||
|
cpfile(src, self._out(dest))
|
||||||
|
|
||||||
|
def mkdir(self, *dirs):
|
||||||
|
for d in dirs:
|
||||||
|
d = self._out(d)
|
||||||
|
if not os.path.isdir(d):
|
||||||
|
os.makedirs(d)
|
||||||
|
|
||||||
|
def replace(self, pat, repl, *files):
|
||||||
|
for f in files:
|
||||||
|
replace(pat, repl, self._out(f))
|
||||||
|
|
||||||
|
def append(self, filename, data):
|
||||||
|
with open(self._out(filename), "a") as fobj:
|
||||||
|
fobj.write(data+"\n")
|
||||||
|
|
||||||
|
def treeinfo(self, section, key, *valuetoks):
|
||||||
|
if section not in self.treeinfo:
|
||||||
|
self.treeinfo_data[section] = dict()
|
||||||
|
self.treeinfo_data[section][key] = " ".join(valuetoks)
|
||||||
|
|
||||||
|
def installkernel(self, section, src, dest):
|
||||||
|
self.install(src, dest)
|
||||||
|
self.treeinfo(section, "kernel", dest)
|
||||||
|
|
||||||
|
def installinitrd(self, section, src, dest):
|
||||||
|
self.install(src, dest)
|
||||||
|
self.treeinfo(section, "initrd", dest)
|
||||||
|
|
||||||
|
def hardlink(self, src, dest):
|
||||||
|
os.link(self._out(src), self._out(dest))
|
||||||
|
|
||||||
|
def symlink(self, target, dest):
|
||||||
|
os.symlink(target, self._out(dest))
|
||||||
|
|
||||||
|
def copy(self, src, dest):
|
||||||
|
cpfile(self._out(src), self._out(dest))
|
||||||
|
|
||||||
|
def copyif(self, src, dest):
|
||||||
|
if self.exists(src):
|
||||||
|
self.copy(src, dest)
|
||||||
|
return True
|
||||||
|
|
||||||
|
def move(self, src, dest):
|
||||||
|
self.copy(src, dest)
|
||||||
|
self.remove(src)
|
||||||
|
|
||||||
|
def moveif(self, src, dest):
|
||||||
|
if self.copyif(src, dest):
|
||||||
|
self.remove(src)
|
||||||
|
return True
|
||||||
|
|
||||||
|
def remove(self, *targets):
|
||||||
|
for t in targets:
|
||||||
|
remove(self._out(t))
|
||||||
|
|
||||||
|
def chmod(self, target, mode):
|
||||||
|
os.chmod(self._out(target), int(mode,8))
|
||||||
|
|
||||||
|
def gconfset(self, path, keytype, value, outfile=None):
|
||||||
|
if outfile is None:
|
||||||
|
outfile = self._out("etc/gconf/gconf.xml.defaults")
|
||||||
|
check_call(["gconftool-2", "--direct",
|
||||||
|
"--config-source=xml:readwrite:%s" % outfile,
|
||||||
|
"--set", "--type", keytype, path, value])
|
||||||
|
|
||||||
|
def log(self, msg):
|
||||||
|
logger.info(msg)
|
||||||
|
|
||||||
|
def runcmd(self, *cmdlist):
|
||||||
|
'''Note that we need full paths for everything here'''
|
||||||
|
chdir = lambda: None
|
||||||
|
cmd = cmdlist
|
||||||
|
if cmd[0].startswith("chdir="):
|
||||||
|
dirname = cmd[0].split('=',1)[1]
|
||||||
|
chdir = lambda: os.chdir(dirname)
|
||||||
|
cmd = cmd[1:]
|
||||||
|
logger.info("runcmd: %s", cmd)
|
||||||
|
check_call(cmd, preexec_fn=chdir)
|
||||||
Loading…
Reference in New Issue
Block a user