Create a new command update_kernel.sh

remote_access.sh

 # Copyright (c) 2009 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.
 
 # Library for setting up remote access and running remote commands.
 
 DEFAULT_PRIVATE_KEY="${GCLIENT_ROOT}/src/scripts/mod_for_test_scripts/\
 ssh_keys/testing_rsa"
 
 DEFINE_string remote "" "remote hostname/IP of running Chromium OS instance"
@@ skipping 50 matching lines @@
   [ -n "${FLAGS_board}" ] && return
   remote_sh grep CHROMEOS_RELEASE_BOARD /etc/lsb-release
   FLAGS_board=$(echo "${REMOTE_OUT}" | cut -d '=' -f 2)
   if [ -z "${FLAGS_board}" ]; then
     error "Board required"
     exit 1
   fi
   info "Target reports board is ${FLAGS_board}"
 }
 
+function remote_reboot {
+  info "Rebooting."
+  remote_sh "touch /tmp/awaiting_reboot; reboot"
+  local output_file
+  output_file="${TMP}/output"
+
+  while true; do
+    REMOTE_OUT=""
+    # This may fail while the machine is down so generate output and a
+    # boolean result to distinguish between down/timeout and real failure
+    ! remote_sh_allow_changed_host_key \
+        "echo 0; [ -e /tmp/awaiting_reboot ] && echo '1'; true"
+    echo "${REMOTE_OUT}" > "${output_file}"
+    if grep -q "0" "${output_file}"; then
+      if grep -q "1" "${output_file}"; then
+        info "Not yet rebooted"
+      else
+        info "Rebooted and responding"
+        break
+      fi
+    fi
+    sleep .5
+  done
+}
+
 function cleanup_remote_access() {
   # Call this function from the exit trap of the main script.
   # Iff we started ssh-agent, be nice and clean it up.
   # Note, only works if called from the main script - no subshells.
   if [[ 1 -eq ${OWN_SSH_AGENT} ]]
   then
     kill ${SSH_AGENT_PID} 2>/dev/null
     unset SSH_AGENT_PID SSH_AUTH_SOCK
   fi
 }
 
 function remote_access_init() {
   TMP_PRIVATE_KEY=$TMP/private_key
   TMP_KNOWN_HOSTS=$TMP/known_hosts
   if [ -z "$FLAGS_remote" ]; then
     echo "Please specify --remote=<IP-or-hostname> of the Chromium OS instance"
     exit 1
   fi
   set_up_remote_access
 }