[master 1/1] Add ability for external templates to graft content into boot.iso

cgwalters installerbot-noreply at redhat.com
Thu Mar 19 20:50:19 UTC 2015


From: Colin Walters <walters at verbum.org>

I originally added --add-template to support doing something similar
to pungi, which injects content into the system to be used by default.
However, this causes the content to be part of the squashfs, which
means PXE installations have to download significantly more data that
they may not need (if they actually want to pull the tree data from
the network, which is not an unusual case).

What I actually need is to be able to modify *both* the runtime image
and the arch-specific content.  For the runtime, I need to change
/usr/share/anaconda/interactive-defaults.ks to point to the new
content.  (Although, potentially we could patch Anaconda itself to
auto-detect an ostree repository configured in disk image, similar to
what it does for yum repositories)

For the arch-specfic image, I want to drop my content into the ISO
root.

So this patch adds --add-arch-template and --add-arch-template-var
in order to do the latter, while preserving the --add-template
to affect the runtime image.

Further, the templates will automatically graft in a directory named
"iso-graft/" from the working directory (if it exists).

(I suggest that external templates create a subdirectory named
 "content" to avoid clashes with any future lorax work)

Thus, this will be used by the Atomic Host lorax templates to inject
content/repo, but could be used by e.g. pungi to add content/rpms as
well.

I tried to avoid code deduplication by creating a new template for the
product.img bits and this, but that broke because the parent boot.iso
code needs access to the `${imggraft}` variable.  I think a real fix
here would involve turning the product.img, content/, *and* boot.iso
into a new template.
---
 share/aarch64.tmpl         |  7 +++++++
 share/arm.tmpl             |  7 +++++++
 share/ppc.tmpl             |  7 +++++++
 share/ppc64le.tmpl         |  7 +++++++
 share/x86.tmpl             |  7 +++++++
 src/pylorax/__init__.py    | 10 ++++++++--
 src/pylorax/ltmpl.py       |  2 +-
 src/pylorax/treebuilder.py | 10 ++++++++--
 src/sbin/lorax             | 19 +++++++++++++++++--
 9 files changed, 69 insertions(+), 7 deletions(-)

diff --git a/share/aarch64.tmpl b/share/aarch64.tmpl
index 2296546..18b2f6d 100644
--- a/share/aarch64.tmpl
+++ b/share/aarch64.tmpl
@@ -47,6 +47,13 @@ mkdir ${KERNELDIR}
     %endif
 %endfor
 
+# Inherit iso-graft/ if it exists from external templates
+<%
+    import os
+    if os.path.exists(workdir + "/iso-graft"):
+        imggraft += " " + workdir + "/iso-graft"
+%>
+
 %if exists("boot/efi/EFI/*/gcdaa64.efi"):
 ## make boot.iso
 runcmd mkisofs -o ${outroot}/images/boot.iso \
diff --git a/share/arm.tmpl b/share/arm.tmpl
index e82bb32..5c4895e 100644
--- a/share/arm.tmpl
+++ b/share/arm.tmpl
@@ -123,5 +123,12 @@ treeinfo ${basearch} platforms ${platforms}
     %endif
 %endfor
 
+# Inherit iso-graft/ if it exists from external templates
+<%
+    import os
+    if os.path.exists(workdir + "/iso-graft"):
+        imggraft += " " + workdir + "/iso-graft"
+%>
+
 ## FIXME: ARM may need some extra boot config
 
diff --git a/share/ppc.tmpl b/share/ppc.tmpl
index 2cc4ec5..5d7040a 100644
--- a/share/ppc.tmpl
+++ b/share/ppc.tmpl
@@ -96,6 +96,13 @@ install ${configdir}/mapping ${BOOTDIR}
     %endif
 %endfor
 
+# Inherit iso-graft/ if it exists from external templates
+<%
+    import os
+    if os.path.exists(workdir + "/iso-graft"):
+        imggraft += " " + workdir + "/iso-graft"
+%>
+
 ## make boot.iso
 runcmd mkisofs -o ${outroot}/images/boot.iso -chrp-boot -U \
         ${prepboot} -part -hfs -T -r -l -J \
diff --git a/share/ppc64le.tmpl b/share/ppc64le.tmpl
index ccee180..14aa0ad 100644
--- a/share/ppc64le.tmpl
+++ b/share/ppc64le.tmpl
@@ -72,6 +72,13 @@ mkdir images/
     %endif
 %endfor
 
+# Inherit iso-graft/ if it exists from external templates
+<%
+    import os
+    if os.path.exists(workdir + "/iso-graft"):
+        imggraft += " " + workdir + "/iso-graft"
+%>
+
 ## make boot.iso
 runcmd mkisofs -v -U -J -R -T \
         -o ${outroot}/images/boot.iso \
diff --git a/share/x86.tmpl b/share/x86.tmpl
index 4f82dbf..7273e28 100644
--- a/share/x86.tmpl
+++ b/share/x86.tmpl
@@ -97,6 +97,13 @@ hardlink ${KERNELDIR}/initrd.img ${BOOTDIR}
     %endif
 %endfor
 
+# Inherit iso-graft/ if it exists from external templates
+<%
+    import os
+    if os.path.exists(workdir + "/iso-graft"):
+        imggraft += " " + workdir + "/iso-graft"
+%>
+
 ## make boot.iso
 runcmd mkisofs -o ${outroot}/images/boot.iso \
        -b ${BOOTDIR}/isolinux.bin -c ${BOOTDIR}/boot.cat \
diff --git a/src/pylorax/__init__.py b/src/pylorax/__init__.py
index fc489c6..3d74a4b 100644
--- a/src/pylorax/__init__.py
+++ b/src/pylorax/__init__.py
@@ -153,7 +153,10 @@ def run(self, dbo, product, version, release, variant="", bugurl="",
             installpkgs=None,
             size=2,
             add_templates=None,
-            add_template_vars=None):
+            add_template_vars=None,
+            add_arch_templates=None,
+            add_arch_template_vars=None,
+            template_tempdir=None):
 
         assert self._configured
 
@@ -310,7 +313,10 @@ def run(self, dbo, product, version, release, variant="", bugurl="",
                                   inroot=installroot, outroot=self.outputdir,
                                   runtime=runtime, isolabel=isolabel,
                                   domacboot=domacboot, doupgrade=doupgrade,
-                                  templatedir=templatedir)
+                                  templatedir=templatedir,
+                                  add_templates=add_arch_templates,
+                                  add_template_vars=add_arch_template_vars,
+                                  workdir=self.workdir)
 
         logger.info("rebuilding initramfs images")
         dracut_args = ["--xz", "--install", "/.buildstamp"]
diff --git a/src/pylorax/ltmpl.py b/src/pylorax/ltmpl.py
index 033b6d9..1abbfc8 100644
--- a/src/pylorax/ltmpl.py
+++ b/src/pylorax/ltmpl.py
@@ -184,7 +184,7 @@ def _getsize(self, *files):
     def run(self, templatefile, **variables):
         for k,v in self.defaults.items() + self.builtins.items():
             variables.setdefault(k,v)
-        logger.debug("parsing %s", templatefile)
+        logger.debug("executing {0} with variables={1}".format(templatefile, variables))
         self.templatefile = templatefile
         t = LoraxTemplate(directories=[self.templatedir])
         commands = t.parse(templatefile, variables)
diff --git a/src/pylorax/treebuilder.py b/src/pylorax/treebuilder.py
index 5fac135..c5ac269 100644
--- a/src/pylorax/treebuilder.py
+++ b/src/pylorax/treebuilder.py
@@ -184,16 +184,20 @@ def finished(self):
 class TreeBuilder(object):
     '''Builds the arch-specific boot images.
     inroot should be the installtree root (the newly-built runtime dir)'''
-    def __init__(self, product, arch, inroot, outroot, runtime, isolabel, domacboot=True, doupgrade=True, templatedir=None):
+    def __init__(self, product, arch, inroot, outroot, runtime, isolabel, domacboot=True, doupgrade=True, templatedir=None, add_templates=None, add_template_vars=None, workdir=None):
+
         # NOTE: if you pass an arg named "runtime" to a mako template it'll
         # clobber some mako internal variables - hence "runtime_img".
         self.vars = DataHolder(arch=arch, product=product, runtime_img=runtime,
                                runtime_base=basename(runtime),
                                inroot=inroot, outroot=outroot,
                                basearch=arch.basearch, libdir=arch.libdir,
-                               isolabel=isolabel, udev=udev_escape, domacboot=domacboot, doupgrade=doupgrade)
+                               isolabel=isolabel, udev=udev_escape, domacboot=domacboot, doupgrade=doupgrade,
+                               workdir=workdir)
         self._runner = LoraxTemplateRunner(inroot, outroot, templatedir=templatedir)
         self._runner.defaults = self.vars
+        self.add_templates = add_templates or []
+        self.add_template_vars = add_template_vars or {}
         self.templatedir = templatedir
         self.treeinfo_data = None
 
@@ -242,6 +246,8 @@ def rebuild_initrds(self, add_args=None, backup="", prefix=""):
 
     def build(self):
         templatefile = templatemap[self.vars.arch.basearch]
+        for tmpl in self.add_templates:
+            self._runner.run(tmpl, **self.add_template_vars)
         self._runner.run(templatefile, kernels=self.kernels)
         self.treeinfo_data = self._runner.results.treeinfo
         self.implantisomd5()
diff --git a/src/sbin/lorax b/src/sbin/lorax
index 51a32ea..bd07bb3 100755
--- a/src/sbin/lorax
+++ b/src/sbin/lorax
@@ -136,10 +136,16 @@ def main(args):
     optional.add_option("--force", default=False, action="store_true",
                         help="Run even when the destination directory exists")
     optional.add_option("--add-template", dest="add_templates",
-                        action="append", help="Additional template to execute",
+                        action="append", help="Additional template for runtime image",
                         default=[])
     optional.add_option("--add-template-var", dest="add_template_vars",
-                        action="append", help="Set variable for additional templates",
+                        action="append", help="Set variable for runtime image template",
+                        default=[])
+    optional.add_option("--add-arch-template", dest="add_arch_templates",
+                        action="append", help="Additional template for architecture-specific image",
+                        default=[])
+    optional.add_option("--add-arch-template-var", dest="add_arch_template_vars",
+                        action="append", help="Set variable for architecture-specific image",
                         default=[])
 
     # add the option groups to the parser
@@ -213,6 +219,13 @@ def main(args):
             raise ValueError("Missing '=' for key=value in " % kv)
         parsed_add_template_vars[k] = v
 
+    parsed_add_arch_template_vars = {}
+    for kv in opts.add_arch_template_vars:
+        k, t, v = kv.partition('=')
+        if t == '':
+            raise ValueError("Missing '=' for key=value in " % kv)
+        parsed_add_arch_template_vars[k] = v
+
     # run lorax
     lorax = pylorax.Lorax()
     lorax.configure(conf_file=opts.config)
@@ -224,6 +237,8 @@ def main(args):
               installpkgs=opts.installpkgs,
               add_templates=opts.add_templates,
               add_template_vars=parsed_add_template_vars,
+              add_arch_templates=opts.add_arch_templates,
+              add_arch_template_vars=parsed_add_arch_template_vars,
               remove_temp=True)
 
 


-- 
To view this commit on github, visit https://github.com/rhinstaller/lorax/commit/67217f56425ce24e74f0a75010b22f5c462de239


More information about the anaconda-patches mailing list