update_kernel: add support for modules and firmware

update_kernel.sh

 #!/bin/bash
 
 # Copyright (c) 2009-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.
 
 # Script to update the kernel on a live running ChromiumOS instance.
 
 # Load common constants.  This should be the first executable line.
 # The path to common.sh should be relative to your script's location.
 
 . "$(dirname $0)/common.sh"
 . "$(dirname $0)/remote_access.sh"
 
 DEFINE_string board "" "Override board reported by target"
 DEFINE_string partition "" "Override kernel partition reported by target"
+DEFINE_boolean modules false "Update modules on target"
+DEFINE_boolean firmware false "Update firmware on target"
 
 function cleanup {
   cleanup_remote_access
   rm -rf "${TMP}"
 }
 
 # Ask the target what the kernel partition is
 function learn_partition() {
   [ -n "${FLAGS_partition}" ] && return
   remote_sh cat /proc/cmdline
@@ skipping 40 matching lines @@
        --vmlinuz /build/${FLAGS_board}/boot/vmlinuz"
 
   ./enter_chroot.sh -- ${cmd}
 
   learn_partition
 
   remote_cp_to new_kern.bin /tmp
 
   remote_sh dd if=/tmp/new_kern.bin of="${FLAGS_partition}"
 
+  if [[ ${FLAGS_modules} -eq ${FLAGS_TRUE} ]]; then
+    echo "copying modules"
+    cmd="tar -C /build/${FLAGS_board}/lib/modules -cjf new_modules.tar ."
+    ./enter_chroot.sh -- ${cmd}
+
+    remote_cp_to new_modules.tar /tmp/
+
+    remote_sh mount -o remount,rw /
+    remote_sh tar -C /lib/modules -xjf /tmp/new_modules.tar
+  fi
+
+  if [[ ${FLAGS_firmware} -eq ${FLAGS_TRUE} ]]; then
+    echo "copying firmware"
+    cmd="tar -C /build/${FLAGS_board}/lib/firmware -cjf new_firmware.tar ."
+    ./enter_chroot.sh -- ${cmd}
+
+    remote_cp_to new_firmware.tar /tmp/
+
+    remote_sh mount -o remount,rw /
+    remote_sh tar -C /lib/firmware -xjf /tmp/new_firmware.tar
+  fi
+
   remote_reboot
 
   remote_sh uname -r -v
 
   info "old kernel: ${old_kernel}"
 
   info "new kernel: ${REMOTE_OUT}"
 }
 
 main $@