Support to verify reboot for secure boot

scripts/fast-reboot

@@ -33,6 +33,7 @@ EXIT_SYNCD_SHUTDOWN=11
 EXIT_FAST_REBOOT_DUMP_FAILURE=12
 EXIT_FILTER_FDB_ENTRIES_FAILURE=13
 EXIT_NO_CONTROL_PLANE_ASSISTANT=20
+EXIT_SONIC_INSTALLER_VERIFY_REBOOT=21
 
 function error()
 {
@@ -305,10 +306,11 @@ function reboot_pre_check()
         exit ${EXIT_FILE_SYSTEM_FULL}
     fi
 
-    # Make sure that the next image exists
-    if [[ ! -d ${IMAGE_PATH} ]]; then
-        debug "Next image ${NEXT_SONIC_IMAGE} doesn't exist ..."
-        exit ${EXIT_NEXT_IMAGE_NOT_EXISTS}
+    # Verify the next image by sonic_installer
+    local message=$(sonic_installer verify_next_image 2>&1)
+    if [ $? -ne 0 ]; then
+        debug "Failed to verify next image: ${message}"
+        exit ${EXIT_SONIC_INSTALLER_VERIFY_REBOOT}
     fi
 
     # Make sure ASIC configuration has not changed between images

scripts/reboot

@@ -16,6 +16,7 @@ PLAT_REBOOT="platform_reboot"
 REBOOT_CAUSE_FILE="/host/reboot-cause/reboot-cause.txt"
 VERBOSE=no
 EXIT_NEXT_IMAGE_NOT_EXISTS=4
+EXIT_SONIC_INSTALLER_VERIFY_REBOOT=21
 
 function debug()
 {
@@ -79,10 +80,11 @@ function reboot_pre_check()
     fi
     rm ${filename}
 
-    # Make sure that the next image exists
-    if [[ ! -d ${IMAGE_PATH} ]]; then
-        VERBOSE=yes debug "Next image ${NEXT_SONIC_IMAGE} doesn't exist ..."
-        exit ${EXIT_NEXT_IMAGE_NOT_EXISTS}
+    # Verify the next image by sonic_installer
+    local message=$(sonic_installer verify_next_image 2>&1)
+    if [ $? -ne 0 ]; then
+        VERBOSE=yes debug "Failed to verify next image: ${message}"
+        exit ${EXIT_SONIC_INSTALLER_VERIFY_REBOOT}
     fi
 }
 

sonic_installer/bootloader/aboot.py

@@ -23,6 +23,7 @@
 from .bootloader import Bootloader
 
 _secureboot = None
+DEFAULT_SWI_IMAGE = 'sonic.swi'
 
 # For the signature format, see: https://github.com/aristanetworks/swi-tools/tree/master/switools
 SWI_SIG_FILE_NAME = 'swi-signature'
@@ -110,9 +111,9 @@ def remove_image(self, image):
             self._boot_config_set(SWI=image_path, SWI_DEFAULT=image_path)
             click.echo("Set next and default boot to current image %s" % current)
 
-        image_dir = image.replace(IMAGE_PREFIX, IMAGE_DIR_PREFIX)
+        image_path = self.get_image_path(image)
         click.echo('Removing image root filesystem...')
-        subprocess.call(['rm','-rf', os.path.join(HOST_PATH, image_dir)])
+        subprocess.call(['rm','-rf', image_path])
         click.echo('Image removed')
 
     def get_binary_image_version(self, image_path):
@@ -129,6 +130,13 @@ def verify_binary_image(self, image_path):
         except subprocess.CalledProcessError:
             return False
 
+    def verify_next_image(self):
+        if not super(AbootBootloader, self).verify_next_image():
+            return False
+        image = self.get_next_image()
+        image_path = os.path.join(self.get_image_path(image), DEFAULT_SWI_IMAGE)
+        return self._verify_secureboot_image(image_path)
+
     def _verify_secureboot_image(self, image_path):
         if isSecureboot():
             cert = self.getCert(image_path)

sonic_installer/bootloader/bootloader.py

@@ -2,6 +2,15 @@
 Abstract Bootloader class
 """
 
+import sys
+from os import path
+
+from ..common import (
+   HOST_PATH,
+   IMAGE_DIR_PREFIX,
+   IMAGE_PREFIX,
+)
+
 class Bootloader(object):
 
     NAME = None
@@ -43,8 +52,23 @@ def verify_binary_image(self, image_path):
         """verify that the image is supported by the bootloader"""
         raise NotImplementedError
 
+    def verify_next_image(self):
+        """verify the next image for reboot"""
+        image = self.get_next_image()
+        image_path = self.get_image_path(image)
+        if not path.exists(image_path):
+            sys.stdout.write('Next image {0} doesn\'t exist ...\n'.format(image))
+            return False
+        return True
+
     @classmethod
     def detect(cls):
         """returns True if the bootloader is in use"""
         return False
 
+    @classmethod
+    def get_image_path(cls, image):
+        """returns the image path"""
+        prefix = path.join(HOST_PATH, IMAGE_DIR_PREFIX)
+        return image.replace(IMAGE_PREFIX, prefix)
+

sonic_installer/main.py

@@ -476,5 +476,15 @@ def rollback_docker(container_name):
 
     click.echo('Done')
 
+# verify the next image
+@cli.command('verify_next_image')
+def verify_reboot():
+    """ Verify the next image for reboot"""
+    bootloader = get_bootloader()
+    if not bootloader.verify_next_image():
+        click.echo('Failed')
+        sys.exit(1)
+    click.echo('Done')
+
 if __name__ == '__main__':
     cli()