move handlers into a module directory

1 files changed, 138 insertions, 0 deletions

diff --git a/vmdebootstrap/uefi.py b/vmdebootstrap/uefi.py
new file mode 100644
index 0000000..82bf1eb
--- /dev/null
+++ b/vmdebootstrap/uefi.py
@@ -0,0 +1,138 @@
+"""
+  Wrapper for UEFI operations
+"""
+# -*- coding: utf-8 -*-
+#
+#  uefi.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 os
+import cliapp
+import logging
+from vmdebootstrap.base import Base, runcmd
+
+# pylint: disable=missing-docstring
+
+
+arch_table = {  # pylint: disable=invalid-name
+    'amd64': {
+        'removable': '/EFI/boot/bootx64.efi',  # destination location
+        'install': '/EFI/debian/grubx64.efi',  # package location
+        'package': 'grub-efi-amd64',  # bootstrap package
+        'bin_package': 'grub-efi-amd64-bin',  # binary only
+        'extra': 'i386',  # architecture to add binary package
+        'exclusive': False,  # only EFI supported for this arch.
+        'target': 'x86_64-efi',  # grub target name
+    },
+    'i386': {
+        'removable': '/EFI/boot/bootia32.efi',
+        'install': '/EFI/debian/grubia32.efi',
+        'package': 'grub-efi-ia32',
+        'bin_package': 'grub-efi-ia32-bin',
+        'extra': None,
+        'exclusive': False,
+        'target': 'i386-efi',
+    },
+    'arm64': {
+        'removable': '/EFI/boot/bootaa64.efi',
+        'install': '/EFI/debian/grubaa64.efi',
+        'package': 'grub-efi-arm64',
+        'bin_package': 'grub-efi-arm64-bin',
+        'extra': None,
+        'exclusive': True,
+        'target': 'arm64-efi',
+    }
+}
+
+
+class Uefi(Base):
+
+    name = 'uefi'
+
+    def __init__(self):
+        super(Uefi, self).__init__()
+        self.bootdir = None
+
+    def efi_packages(self):
+        packages = []
+        pkg = arch_table[self.settings['arch']]['package']
+        self.message("Adding %s" % pkg)
+        packages.append(pkg)
+        extra = arch_table[self.settings['arch']]['extra']
+        if extra and isinstance(extra, str):
+            bin_pkg = arch_table[str(extra)]['bin_package']
+            self.message("Adding support for %s using %s" % (extra, bin_pkg))
+            packages.append(bin_pkg)
+        return packages
+
+    def copy_efi_binary(self, efi_removable, efi_install):
+        logging.debug("using bootdir=%s", self.bootdir)
+        logging.debug("moving %s to %s", efi_removable, efi_install)
+        if efi_removable.startswith('/'):
+            efi_removable = efi_removable[1:]
+        if efi_install.startswith('/'):
+            efi_install = efi_install[1:]
+        efi_output = os.path.join(self.bootdir, efi_removable)
+        efi_input = os.path.join(self.bootdir, efi_install)
+        if not os.path.exists(efi_input):
+            logging.warning("%s does not exist (%s)", efi_input, efi_install)
+            raise cliapp.AppException("Missing %s" % efi_install)
+        if not os.path.exists(os.path.dirname(efi_output)):
+            os.makedirs(os.path.dirname(efi_output))
+        logging.debug(
+            'Moving UEFI support: %s -> %s', efi_input, efi_output)
+        if os.path.exists(efi_output):
+            os.unlink(efi_output)
+        os.rename(efi_input, efi_output)
+
+    def configure_efi(self):
+        """
+        Copy the bootloader file from the package into the location
+        so needs to be after grub and kernel already installed.
+        """
+        self.message('Configuring EFI')
+        efi_removable = str(arch_table[self.settings['arch']]['removable'])
+        efi_install = str(arch_table[self.settings['arch']]['install'])
+        self.message('Installing UEFI support binary')
+        self.copy_efi_binary(efi_removable, efi_install)
+
+    def configure_extra_efi(self):
+        extra = str(arch_table[self.settings['arch']]['extra'])
+        if extra:
+            efi_removable = str(arch_table[extra]['removable'])
+            efi_install = str(arch_table[extra]['install'])
+            self.message('Copying UEFI support binary for %s' % extra)
+            self.copy_efi_binary(efi_removable, efi_install)
+
+    def partition_esp(self):
+        espsize = self.settings['esp-size'] / (1024 * 1024)
+        self.message("Using ESP size: %smib %s bytes" % (espsize, self.settings['esp-size']))
+        runcmd(['parted', '-s', self.settings['image'],
+                'mkpart', 'primary', 'fat32',
+                '1', str(espsize)])
+        runcmd(['parted', '-s', self.settings['image'],
+                'set', '1', 'boot', 'on'])
+        runcmd(['parted', '-s', self.settings['image'],
+                'set', '1', 'esp', 'on'])
+
+    def prepare_esp(self, rootdir, bootdev):
+        bootdir = '%s/%s/%s' % (rootdir, 'boot', 'efi')
+        logging.debug("bootdir:%s", self.bootdir)
+        self.mkfs(bootdev, fstype='vfat')
+        os.makedirs(bootdir)
+        return bootdir