grs/ISOIt.py: add in build for initramfs and iso.

1 files changed, 83 insertions, 21 deletions

diff --git a/grs/ISOIt.py b/grs/ISOIt.py
index 9b3a920..2a0be81 100644
--- a/grs/ISOIt.py
+++ b/grs/ISOIt.py
@@ -39,31 +39,93 @@ class ISOIt(HashIt):
         self.digest_name = '%s.DIGESTS' % self.medium_name
 
 
+    def initramfs(self, isoboot_dir):
+        """ TODO """
+        # Paths to where we'll build busybox and the initramfs.
+        busybox_root     = os.path.join(self.workdir, 'busybox')
+        busybox_path     = os.path.join(busybox_root, 'bin/busybox')
+        savedconfig_path = os.path.join(busybox_root, 'etc/portage/savedconfig/sys-apps/busybox')
+        busybox_config   = os.path.join(self.libdir, 'scripts/busybox-config')
+
+        # Remove any old busybox build directory and prepare new one.
+        shutil.rmtree(busybox_root, ignore_errors=True)
+        os.makedirs(savedconfig_dir, mode=0o755, exist_ok=True)
+        shutil.copyfile(busybox_config, savedconfig_path)
+
+        # Emerge busybox.
+        cmd = 'emerge --nodeps -1 busybox'
+        emerge_env = { 'USE' : '-* savedconfig', 'ROOT' : busybox_root }
+        Execute(cmd, timeout=600, extra_env=emerge_env, logfile=self.logfile)
+
+        # Remove any old initramfs root and prepare a new one.
+        initramfs_root = os.path.join(self.workdir, 'initramfs')
+        shutil.rmtree(initramfs_root, ignore_errors=True)
+        root_paths = ['bin', 'dev', 'etc', 'mnt/cdrom', 'mnt/squashfs', 'mnt/tmpfs',
+            'proc', 'sbin', 'sys', 'tmp', 'usr/bin', 'usr/sbin', 'var', 'var/run']
+        for p in root_paths:
+            d = os.path.join(initramfs_root, p)
+            os.makdirs(d, mode=0o755, exist_ok=True)
+
+        # Copy the static busybox to the initramfs root.
+        # TODO: we are assuming a static busybox, so we should check.
+        shutil.copyfile(busybox_path, os.join.path(initramfs_root, 'bin/busybox'))
+        cmd = 'chroot %s /bin/busybox --install -s' % initramfs_root
+        Execute(cmd, timeout=60, logfile=self.logfile)
+        initscript_path = os.path.join(self.libdir, 'scripts/initramfs-init')
+        shutil.copy(initscript_path, initramfs_root)
+
+        # TODO: we are assuming a static kernel and so not copying in
+        # any modules.  This is where we should copy in modules.
+
+        # cpio-gzip the initramfs root to the iso boot dir
+        initramfs_path = os.path.join(isoboot_dir, 'initramfs')
+        cwd = os.getcwd()
+        os.chdir(initramfs_root)
+        cmd = 'find . | cpio -H newc -o | gzip -9 > %s' % initramfs_path
+        Execute(cmd, timeout=600, logfile=self.logfile)
+        os.chdir(cwd)
+
+
     def isoit(self, alt_name = None):
         # Create the ISO with the default name unless an alt_name is given.
         if alt_name:
             self.medium_name = '%s-%s%s%s.iso' % (alt_name, self.year, self.month, self.day)
             self.digest_name = '%s.DIGESTS' % self.medium_name
-        iso_path = os.path.join(self.workdir, 'iso')
-        grub_path = os.path.join(iso_path, 'boot/grub')
-        os.makedirs(grub_path, mode=0o755, exist_ok=False)
-        #
+        iso_dir     = os.path.join(self.workdir, 'iso')
+        isoboot_dir = os.path.join(iso_dir, 'boot')
+        isogrub_dir = os.path.join(isoboot_dir, 'grub')
+        os.makedirs(isogrub_dir, mode=0o755, exist_ok=False)
+
         # 1. build initramfs image and copy it in
-        #    locate a build script for the initramfs in self.libdir/scripts
-        #    locate a busybox config script in self.libdir/scripts
-        #    locate an init scripts in self.libdir/scripts
-        #    copy in any kernel modules(?)
-        #    find . | cpio -H newc -o | gzip -9 > iso/boot/initramfs.igz
-        #
-        # 2. make the squashfs image and copy it into the iso/boot
-        squashfs_path = os.path.join(iso_path, 'rootfs')
+        self.initramfs(isoboot_dir)
+
+        # 2. Move the kernel image into the iso/boot directory.
+        # TODO: we are assuming a static kernel
+        kernelimage_dir  = os.path.join(self.portage_configroot, 'boot')
+        kernelimage_path = os.path.join(kernelimage_dir, 'kernel')
+        shutil.copy(kernelimage_path, isoboot_dir)
+        shutil.rmtree(kernelimage_dir, ignore_errors=True)
+
+        # 3. make the squashfs image and copy it into the iso/boot
+        squashfs_path = os.path.join(iso_dir, 'rootfs')
         cmd = 'mksquashfs %s %s -xattrs -comp xz' % (self.portage_configroot, squashfs_path)
-        Execute(cmd, timeout=None, logfile=self.logfile)
-        #
-        # 3. prepare the grub bootloader
-        #    copy in stage2_eltorito into iso/boot/grub
-        #    copy in menu.lst into iso/boot/grub
-        #
-        # 4. create the iso image
-        #    mkisofs -R -b boot/grub/stage2_eltorito -no-emul-boot -boot-load-size 4 \
-        #        -boot-info-table -o medium_pathname.iso iso
+        Execute(cmd, timeout=600, logfile=self.logfile)
+
+        # 4. Emerge grub:0 to grab stage2_eltorito
+        grub_root     = os.path.join(self.workdir, 'grub')
+        eltorito_path = os.path.join(grub_root, 'boot/grub/stage2_eltorito')
+        menulst_path  = os.path.join(self.libdir, 'scripts/menu.lst')
+        cmd = 'emerge --nodeps -1 grub:0'
+        emerge_env = { 'USE' : '-* savedconfig', 'ROOT' : grub_root }
+        Execute(cmd, timeout=600, extra_env=emerge_env, logfile=self.logfile)
+        shutil.copyfile(eltorito_path, isogrub_dir)
+        shutil.copyfile(menulst_path, isogrub_dir)
+
+        # 5. create the iso image
+        args  = '-R '                           # Rock Ridge protocol
+        args += '-b boot/grub/stage2_eltorito ' # El Torito boot image
+        args += '-no-emul-boot '                # No disk emulation for El Torito
+        args += '-boot-load-size 4 '            # 4x512-bit sectors for no-emulation mode
+        args += '-boot-info-table '             # Create El Torito boot info table
+        cmd = 'mkisofs %s -o %s %s' % (args, self.medium_pathname, iso_dir)
+        Execute(cmd, timeout=600, logfile=self.logfile)