#!/bin/bash

# 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.

# --- BEGIN COMMON.SH BOILERPLATE ---
# Load common CrOS utilities.  Inside the chroot this file is installed in
# /usr/lib/crosutils.  Outside the chroot we find it relative to the script's
# location.
find_common_sh() {
  local common_paths=(/usr/lib/crosutils $(dirname "$(readlink -f "$0")"))
  local path

  SCRIPT_ROOT=
  for path in "${common_paths[@]}"; do
    if [ -r "${path}/common.sh" ]; then
      SCRIPT_ROOT=${path}
      break
    fi
  done
}

find_common_sh
. "${SCRIPT_ROOT}/common.sh" || (echo "Unable to load common.sh" && exit 1)
# --- END COMMON.SH BOILERPLATE ---

IMAGES_DIR="${DEFAULT_BUILD_ROOT}/images"

# Flags
DEFINE_string  from   "" "Directory containing rootfs.image and mbr.image"
DEFINE_string  to     "" "Destination file for USB image."
DEFINE_integer offset 0  "Offset to write the kernel to in the destination."

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

# Die on any errors.
set -e

if [ -z "${FLAGS_from}" -o -z "${FLAGS_to}" -o -z "${FLAGS_offset}" ]; then
  echo "You must define all of from, to and offset."
  exit 1
fi

# Convert args to paths.  Need eval to un-quote the string so that shell
# chars like ~ are processed; just doing FOO=`readlink -f $FOO` won't work.
FLAGS_from=`eval readlink -f ${FLAGS_from}`
FLAGS_to=`eval readlink -f ${FLAGS_to}`

function do_cleanup {
  sync
  sudo umount -l /tmp/kernel_fetch.$$
  rmdir /tmp/kernel_fetch.$$
}

trap do_cleanup EXIT

#
# Set up loop device.  This time it is used to fetch the built kernel from the
# root image.  This kernel is then written to the fourth partition.
#
echo "Fetching kernel from root image..."
mkdir /tmp/kernel_fetch.$$
sudo mount -o ro,loop "${FLAGS_from}/rootfs.image" /tmp/kernel_fetch.$$

echo "Writing kernel to ${FLAGS_to} at ${FLAGS_offset}..."
sudo "${SCRIPTS_DIR}"/file_copy.py \
  if=/tmp/kernel_fetch.$$/boot/vmlinux.uimg \
  of="${FLAGS_to}" \
  seek_bytes="${FLAGS_offset}"
echo "Done."

do_cleanup

trap - EXIT