Fixed stateful update in image_to_live.

image_to_live.sh

Index: image_to_live.sh
diff --git a/image_to_live.sh b/image_to_live.sh
index a6b70bee90def1614300ace56faa3655a0e96cd6..449eba89e16ee8cb1f2d10444dd3a5bfc99f809f 100755
--- a/image_to_live.sh
+++ b/image_to_live.sh
@@ -69,12 +69,22 @@ function start_dev_server {
 # from the dev server and prepares the update. chromeos_startup finishes
 # the update on next boot.
 function copy_stateful_update {
-  info "Starting stateful update."
-  local dev_dir="$(dirname $0)/../platform/dev"
+  local dev_url=$(get_devserver_url)
+  local stateful_url=""
+
+  # Assume users providing an update url are using an archive_dir path.
+  if [ -n "${FLAGS_update_url}" ]; then
+    stateful_url=$(echo ${dev_url} | sed -e "s/update/static\/archive/")
+  else
+    stateful_url=$(echo ${dev_url} | sed -e "s/update/static/")
+  fi
+
+  info "Starting stateful update using URL ${stateful_url}"
 
   # Copy over update script and run update.
-  remote_cp_to "$dev_dir/stateful_update" "/tmp"
-  remote_sh "/tmp/stateful_update"
+  local dev_dir="$(dirname $0)/../platform/dev"
+  remote_cp_to "${dev_dir}/stateful_update" "/tmp"
+  remote_sh "/tmp/stateful_update ${stateful_url}"
 }
 
 function get_update_args {