provingground/mod_test_image_for_staging.sh

#!/bin/bash

# Copyright (c) 2013 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.

# Description:
# Script modifies a chromiumos image so that it will work with test
# instances of the DM Server and GAIA in the staging environment.
#
# When run, the script will mount the specified image, make modifications to
# the flags passed to Chrome, and then re-sign the image with devkey.
#

. "/usr/lib/crosutils/common.sh" || { echo "Unable to load common.sh"; exit 1; }

assert_inside_chroot

DEFINE_string dmserver \
  "https://cros-dev.sandbox.google.com/devicemanagement/data/api" \
  "Complete URL for the DM Server" d
DEFINE_string gaiaserver "https://gaiastaging.corp.google.com" \
  "Complete URL for the GAIA server" g
DEFINE_string image "$FLAGS_image" \
  "Path and name of the chromiumos image file" i

# Parse command line.
FLAGS "$@" || exit 1
eval set -- "$FLAGS_ARGV"

set -e

FLAGS_image=$(eval readlink -f "${FLAGS_image}")
IMAGE_DIR=$(dirname "${FLAGS_image}")
IMAGE_NAME=$(basename "${FLAGS_image}")
ROOT_FS_DIR="${IMAGE_DIR}/rootfs"
CONFIG_FILE="${ROOT_FS_DIR}/etc/chrome_dev.conf"
DEVKEYS_DIR="/usr/share/vboot/devkeys"
VBOOT_DIR="${CHROOT_TRUNK_DIR}/src/platform/vboot_reference/scripts/"\
"image_signing"

cleanUp() {
  "${SCRIPTS_DIR}/mount_gpt_image.sh" -u -r "$ROOT_FS_DIR"
}

if [ ! -d "$VBOOT_DIR" ]; then
  die "The required path: $VBOOT_DIR does not exist. This directory needs"\
      "to be sync'd into your chroot.\n $ cros_workon start vboot_reference"
fi

trap cleanUp EXIT

# Mount gpt (GUID partition table) image, and sets up var, /usr/local,
# and symlinks.
"$SCRIPTS_DIR/mount_gpt_image.sh" --image="$IMAGE_NAME" --from="$IMAGE_DIR" \
  --rootfs_mountpt="$ROOT_FS_DIR"

# Instruct session_manager to make some changes to Chrome's command line.
sudo_append "${CONFIG_FILE}" <<EOF
################################################################################
# mod_test_image_for_staging.sh modifications start here.

# Delete the flag with the old URL and set a new one.
!--device-management-url=
--device-management-url=${FLAGS_dmserver}

--gaia-url=${FLAGS_gaiaserver}
--lso-url=https://test-sandbox.auth.corp.google.com
--google-apis-host=www-googleapis-test.sandbox.google.com
--oauth2-client-id=236834563817.apps.googleusercontent.com
--oauth2-client-secret=RsKv5AwFKSzNgE0yjnurkPVI
--cloud-print-url=https://cloudprint-nightly-ps.sandbox.google.com/cloudprint
--ignore-urlfetcher-cert-requests

# mod_test_image_for_staging.sh modifications end here.
################################################################################
EOF

trap - EXIT

cleanUp

TMP_BIN_PATH="${FLAGS_image}.new"
"${VBOOT_DIR}/sign_official_build.sh" usb "${FLAGS_image}" \
                                     "${DEVKEYS_DIR}" \
                                     "${TMP_BIN_PATH}"

echo "Renaming from ${TMP_BIN_PATH} to ${FLAGS_image}."
mv "${TMP_BIN_PATH}" "${FLAGS_image}"