3 files changed, 37 insertions, 34 deletions

diff --git a/classes/override_grub-efi.inc b/classes/override_grub-efi-cfg.inc
index 15a7063..42d89b7 100644
--- a/classes/override_grub-efi.inc
+++ b/classes/override_grub-efi-cfg.inc
@@ -1,4 +1,6 @@
+## ENEA_start ##
 GRUB_GRAPHICS ?= "console=tty0"
+## ENEA_end ##
 
 python build_efi_cfg() {
     import sys
@@ -48,8 +50,10 @@ python build_efi_cfg() {
         bb.fatal('GRUB_ROOT not defined')
 
     if gfxserial == "1":
+        ## ENEA_start ##
         btypes = [ [ " serial console", d.getVar('GRUB_SERIAL') or "" ],
             [ " graphics console", d.getVar('GRUB_GRAPHICS') or "" ] ]
+        ## ENEA_end ##
     else:
         btypes = [ [ "", "" ] ]
 
@@ -60,14 +64,15 @@ python build_efi_cfg() {
         if not overrides:
             bb.fatal('OVERRIDES not defined')
 
-        for btype in btypes:
-            localdata.setVar('OVERRIDES', label + ':' + overrides)
+        localdata.setVar('OVERRIDES', 'grub_' + label + ':' + overrides)
 
+        for btype in btypes:
             cfgfile.write('\nmenuentry \'%s%s\'{\n' % (label, btype[0]))
             lb = label
             if label == "install":
                 lb = "install-efi"
-            cfgfile.write('linux /vmlinuz LABEL=%s' % (lb))
+            kernel = localdata.getVar('KERNEL_IMAGETYPE')
+            cfgfile.write('linux /%s LABEL=%s' % (kernel, lb))
 
             cfgfile.write(' %s' % replace_rootfs_uuid(d, root))
 
diff --git a/classes/override_grub-efi_secureboot.inc b/classes/override_grub-efi_secureboot.inc
deleted file mode 100644
index b944ec8..0000000
--- a/classes/override_grub-efi_secureboot.inc
+++ /dev/null
@@ -1,31 +0,0 @@
-efi_populate() {
-        # DEST must be the root of the image so that EFIDIR is not
-        # nested under a top level directory.
-        DEST=$1
-
-        install -d ${DEST}${EFIDIR}
-
-        GRUB_IMAGE="grub-efi-bootia32.efi"
-        DEST_IMAGE="bootia32.efi"
-        if [ "${TARGET_ARCH}" = "x86_64" ]; then
-                GRUB_IMAGE="grub-efi-bootx64.efi"
-                DEST_IMAGE="bootx64.efi"
-        fi
-
-        #Install EFI binaries
-        install -m 0644 ${IMAGE_ROOTFS}/boot/efi${EFIDIR}/*.* ${DEST}${EFIDIR}
-
-        #Install kernel and initramfs sig
-        install -m 0644 ${DEPLOY_DIR_IMAGE}/${KERNEL_IMAGETYPE}.p7b ${DEST}/vmlinuz.p7b
-        install -m 0644 ${INITRD_LIVE}.p7b ${DEST}/initrd.p7b
-
-        EFIPATH=$(echo "${EFIDIR}" | sed 's/\//\\/g')
-        printf 'fs0:%s\%s\n' "$EFIPATH" "$DEST_IMAGE" >${DEST}/startup.nsh
-
-        install -m 0644 ${GRUB_CFG} ${DEST}${EFIDIR}/grub.cfg
-
-        #Install UEFI keys
-        install -d ${DEST}/uefi_sb_keys
-        install -m 0644 ${DEPLOY_DIR_IMAGE}/user-keys/uefi_sb_keys/*.crt ${DEST}/uefi_sb_keys/
-        install -m 0644 ${DEPLOY_DIR_IMAGE}/user-keys/uefi_sb_keys/*.esl ${DEST}/uefi_sb_keys/
-}
diff --git a/classes/override_live-vm-common.inc b/classes/override_live-vm-common.inc
new file mode 100644
index 0000000..dc8e076
--- /dev/null
+++ b/classes/override_live-vm-common.inc
@@ -0,0 +1,29 @@
+# efi_populate_common DEST BOOTLOADER
+efi_populate_common() {
+        # DEST must be the root of the image so that EFIDIR is not
+        # nested under a top level directory.
+        DEST=$1
+
+        install -d ${DEST}${EFIDIR}
+
+        ## ENEA_start ##
+        # install -m 0644 ${DEPLOY_DIR_IMAGE}/$2-${EFI_BOOT_IMAGE} ${DEST}${EFIDIR}/${EFI_BOOT_IMAGE}
+        ## ENEA_end ##
+
+        EFIPATH=$(echo "${EFIDIR}" | sed 's/\//\\/g')
+        printf 'fs0:%s\%s\n' "$EFIPATH" "${EFI_BOOT_IMAGE}" >${DEST}/startup.nsh
+
+        ## ENEA_start ##
+        # Install EFI binaries
+        install -m 0644 ${IMAGE_ROOTFS}/boot/efi${EFIDIR}/*.* ${DEST}${EFIDIR}
+
+        # Install kernel and initramfs sig
+        install -m 0644 ${DEPLOY_DIR_IMAGE}/${KERNEL_IMAGETYPE}${SB_FILE_EXT} ${DEST}/${KERNEL_IMAGETYPE}${SB_FILE_EXT}
+        install -m 0644 ${INITRD_LIVE}${SB_FILE_EXT} ${DEST}/initrd${SB_FILE_EXT}
+
+        # Install UEFI keys
+        install -d ${DEST}/uefi_sb_keys
+        install -m 0644 ${DEPLOY_DIR_IMAGE}/user-keys/uefi_sb_keys/*.crt ${DEST}/uefi_sb_keys/
+        install -m 0644 ${DEPLOY_DIR_IMAGE}/user-keys/uefi_sb_keys/*.esl ${DEST}/uefi_sb_keys/
+        ## ENEA_end ##
+}