Accept list of kernels to re-sign, instead of assuming only 2 & 4.

scripts/image_signing/make_dev_ssd.sh

 #!/bin/sh
 #
 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 #
 # This script can change key (usually developer keys) and kernel config
 # of a kernels on SSD.
 
 SCRIPT_BASE="$(dirname "$0")"
 . "$SCRIPT_BASE/common.sh"
 load_shflags || exit 1
 
 # Constants used by DEFINE_*
 VBOOT_BASE='/usr/share/vboot'
 DEFAULT_KEYS_FOLDER="$VBOOT_BASE/devkeys"
 DEFAULT_BACKUP_FOLDER='/mnt/stateful_partition/backups'
+DEFAULT_PARTITIONS='2 4'
 
 # DEFINE_string name default_value description flag
 DEFINE_string image "/dev/sda" "Path to device or image file" "i"
 DEFINE_string keys "$DEFAULT_KEYS_FOLDER" "Path to folder of dev keys" "k"
 DEFINE_boolean remove_rootfs_verification \
   $FLAGS_FALSE "Modify kernel boot config to disable rootfs verification" ""
 DEFINE_string backup_dir \
   "$DEFAULT_BACKUP_FOLDER" "Path of directory to store kernel backups" ""
 DEFINE_boolean debug $FLAGS_FALSE "Provide debug messages" "d"
 DEFINE_string save_config "" \
   "Base filename to store kernel configs to, instead of resigning." ""
 DEFINE_string set_config "" \
   "Base filename to load kernel configs from" ""
+DEFINE_string partitions "$DEFAULT_PARTITIONS" \
+ "List of partitions to examine" ""
 
 # Parse command line
 FLAGS "$@" || exit 1
 eval set -- "$FLAGS_ARGV"
 
 # Globals
 # ----------------------------------------------------------------------------
 set -e
 
 # a log file to keep the output results of executed command
@@ skipping 47 matching lines @@
     2)
       echo "Kernel A"
       ;;
     4)
       echo "Kernel B"
       ;;
     6)
       echo "Kernel C"
       ;;
     *)
-      err_die "unknown kernel index: $1"
+      echo "Partition $1"
   esac
 }
 
 # Resigns a kernel on SSD or image.
 resign_ssd_kernel() {
   # bs=512 is the fixed block size for dd and cgpt
   local bs=512
   local ssd_device="$1"
 
   # reasonable size for current kernel partition
   local min_kernel_size=32000
   local max_kernel_size=65536
   local resigned_kernels=0
 
-  for kernel_index in 2 4 6; do
+  for kernel_index in $FLAGS_partitions; do
     local old_blob="$(make_temp_file)"
     local new_blob="$(make_temp_file)"
     local name="$(cros_kernel_name $kernel_index)"
     local rootfs_index="$(($kernel_index + 1))"
 
     debug_msg "Probing $name information"
     local offset size
     offset="$(partoffset "$ssd_device" "$kernel_index")" ||
       err_die "Failed to get partition $kernel_index offset from $ssd_device"
     size="$(partsize "$ssd_device" "$kernel_index")" ||
@@ skipping 137 matching lines @@
     exit 0
   fi
 
   return $resigned_kernels
 }
 
 # Main
 # ----------------------------------------------------------------------------
 main() {
   local num_signed=0
+  local num_given=$(echo "$FLAGS_partitions" | wc -w)
   # Check parameters
   KERNEL_KEYBLOCK="$FLAGS_keys/kernel.keyblock"
   KERNEL_DATAKEY="$FLAGS_keys/kernel_data_key.vbprivk"
   KERNEL_PUBKEY="$FLAGS_keys/kernel_subkey.vbpubk"
 
   debug_msg "Prerequisite check"
   ensure_files_exist \
     "$KERNEL_KEYBLOCK" \
     "$KERNEL_DATAKEY" \
     "$KERNEL_PUBKEY" \
     "$FLAGS_image" ||
     exit 1
 
   resign_ssd_kernel "$FLAGS_image" || num_signed=$?
 
   debug_msg "Complete."
-  if [ $num_signed -gt 0 -a $num_signed -le 2 ]; then
-    # signed 1 or two kernels
-    echo "Successfully re-signed $num_signed kernel(s) on device $FLAGS_image".
+  if [ $num_signed -gt 0 -a $num_signed -le $num_given ]; then
+    # signed something at least
+    echo "Successfully re-signed $num_signed of $num_given kernel(s)" \
+      " on device $FLAGS_image".
   else
     err_die "Failed re-signing kernels."
   fi
 }
 
 main