1 files changed, 313 insertions, 0 deletions
diff --git a/projects/cros-scripts/install/cros-medium-setup b/projects/cros-scripts/install/cros-medium-setup
new file mode 100755
index 00000000..1f8c956b
--- /dev/null
+++ b/projects/cros-scripts/install/cros-medium-setup
@@ -0,0 +1,313 @@
+#!/bin/bash
+
+# Copyright (C) 2016 Paul Kocialkowski <contact@paulk.fr>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+SYS_BLOCK_PATH="/sys/class/block"
+DEV_PATH="/dev"
+DEVICE="device"
+VENDOR="vendor"
+MODEL="model"
+NAME="name"
+KERNEL_IMAGE="kernel.img"
+KERNEL_MODULES="modules/lib"
+KERNEL_PARTITION_INDEX=1
+ROOTFS_PARTITION_INDEX=2
+
+# Size in blocks (512 kiB)
+GPT_SIZE=34
+KERNEL_SIZE=16384
+
+usage() {
+	printf "$executable [action] [storage] [rootfs tarball|kernel files]\n" >&2
+
+	printf "\nActions:\n" >&2
+	printf "  partitions - Setup partitions on storage\n" >&2
+	printf "  rootfs - Install rootfs tarball to storage\n" >&2
+	printf "  kernel - Install kernel files to storage\n" >&2
+
+	usage_storage
+
+	printf "\nEnvironment variables:\n" >&2
+	printf "  KERNEL_PATH - Path to the kernel image\n" >&2
+	printf "  VBOOT_TOOLS_PATH - Path to vboot tools\n" >&2
+}
+
+usage_storage() {
+	printf "\nStorage:\n" >&2
+
+	local nodes=$( ls "$SYS_BLOCK_PATH" )
+	local node_path
+	local name
+
+	for node in $nodes
+	do
+		node_path="$DEV_PATH/$node"
+		if ! [ -b "$node_path" ]
+		then
+			continue
+		fi
+
+		name=$( storage_name "$node_path" )
+		if [ -z "$name" ]
+		then
+			continue
+		fi
+
+		printf "  $node_path - $name\n" >&2
+	done
+}
+
+storage_affect_confirm() {
+	local storage_path=$1
+
+	local name=$( storage_name "$storage_path" )
+	local confirm
+
+	printf "This is going to affect the following storage:\n"
+	printf "  $storage_path - $name\n"
+	printf "Press enter to confirm: "
+
+	read confirm
+}
+
+storage_name() {
+	local storage_path=$1
+
+	local node=$( basename "$storage_path" )
+	local vendor_path="$SYS_BLOCK_PATH/$node/$DEVICE/$VENDOR"
+	local model_path="$SYS_BLOCK_PATH/$node/$DEVICE/$MODEL"
+	local name_path="$SYS_BLOCK_PATH/$node/$DEVICE/$NAME"
+	local vendor
+	local name
+
+	if [ -f "$model_path" ]
+	then
+		name=$( cat "$model_path" )
+	elif [ -f "$name_path" ]
+	then
+		name=$( cat "$name_path" )
+	else
+		return 0
+	fi
+
+	name=$( echo "$name" | sed -e "s/^[[:space:]]*//;s/[[:space:]]*$//" )
+
+	if [ -f "$vendor_path" ]
+	then
+		vendor=$( cat "$vendor_path" )
+		vendor=$( echo "$vendor" | sed -e "s/^[[:space:]]*//;s/[[:space:]]*$//" )
+
+		name="$vendor $name"
+	fi
+
+	echo "$name"
+}
+
+storage_partition_path() {
+	local storage_path=$1
+	local index=$2
+
+	storage_partition_path="$storage_path$index"
+
+	if ! [ -b "$storage_partition_path" ]
+	then
+		storage_partition_path="$storage_path""p$index"
+	fi
+
+	if ! [ -b "$storage_partition_path" ]
+	then
+		return 1
+	fi
+
+	echo "$storage_partition_path"
+}
+
+storage_partition_mount_path() {
+	local storage_partition_path=$1
+
+	local storage_partition_mount_path=$( udisksctl info -b "$storage_partition_path"  | grep "MountPoints" | sed "s/.*MountPoints:[[:space:]]*\(.*\)/\1/g" )
+
+	echo "$storage_partition_mount_path"
+}
+
+partitions() {
+	local storage_path=$1
+
+	local storage_rootfs_path
+	local partitions
+	local start
+
+	storage_affect_confirm "$storage_path"
+
+	partitions=$( mount | grep -P "^$storage_path" | sed "s/^\([^[:space:]]*\).*/\1/g" )
+
+	for partition in $partitions
+	do
+		# Partition may already be unmounted.
+		udisksctl unmount -b "$partition" || true
+	done
+
+	( echo "g" ; echo "w" ) | fdisk "$storage_path"
+
+	cgpt create "$storage_path"
+
+	start=$GPT_SIZE
+	size=$KERNEL_SIZE
+	cgpt add -b "$start" -s "$size" -P 1 -S 1 -t kernel -l kernel "$storage_path"
+
+	start=$(( $start + $size ))
+	size=$( cgpt show "$storage_path" | grep "Sec GPT table" | sed "s/[[:space:]]*\([0-9]*\).*/\1/g" )
+	size=$(( $size - $start ))
+	cgpt add -b "$start" -s "$size" -t rootfs -l rootfs "$storage_path"
+
+	blockdev --rereadpt "$storage_path" || partprobe "$storage_path"
+
+	storage_rootfs_path=$( storage_partition_path "$storage_path" "$ROOTFS_PARTITION_INDEX" )
+
+	mkfs.ext4 -F "$storage_rootfs_path"
+
+	printf "\nSetup partitions on storage $storage_path\n"
+}
+
+rootfs() {
+	local storage_path=$1
+	local rootfs_tarball_path=$2
+
+	local storage_rootfs_path=$( storage_partition_path "$storage_path" "$ROOTFS_PARTITION_INDEX" )
+	local storage_rootfs_mount_path
+
+	storage_affect_confirm "$storage_path"
+
+	# Partition may already be mounted.
+	udisksctl mount -b "$storage_rootfs_path" || true
+
+	storage_rootfs_mount_path=$( storage_partition_mount_path "$storage_rootfs_path" )
+
+	tar -xf "$rootfs_tarball_path" -ps -C "$storage_rootfs_mount_path"
+
+	udisksctl unmount -b "$storage_rootfs_path"
+
+	printf "\nInstalled rootfs on storage $storage_path\n"
+}
+
+kernel() {
+	local storage_path=$1
+	local kernel_files_path=$2
+
+	local storage_kernel_path=$( storage_partition_path "$storage_path" "$KERNEL_PARTITION_INDEX" )
+	local storage_rootfs_path=$( storage_partition_path "$storage_path" "$ROOTFS_PARTITION_INDEX" )
+	local kernel_image_path="$kernel_files_path/$KERNEL_IMAGE"
+	local kernel_modules_path="$kernel_files_path/$KERNEL_MODULES"
+	local storage_rootfs_mount_path
+
+	storage_affect_confirm "$storage_path"
+
+	cat "$kernel_image_path" > "$storage_kernel_path"
+	sync
+
+	# Partition may already be mounted.
+	udisksctl mount -b "$storage_rootfs_path" || true
+
+	storage_rootfs_mount_path=$( storage_partition_mount_path "$storage_rootfs_path" )
+
+	rsync -a --keep-dirlinks "$kernel_modules_path" "$storage_rootfs_mount_path/"
+	sync
+
+	udisksctl unmount -b "$storage_rootfs_path"
+
+	printf "\nInstalled kernel on storage $storage_path\n"
+}
+
+requirements() {
+	local requirement
+	local requirement_path
+
+	for requirement in "$@"
+	do
+		requirement_path=$( which "$requirement" || true )
+
+		if [ -z "$requirement_path" ]
+		then
+			printf "Missing requirement: $requirement\n" >&2
+			exit 1
+		fi
+	done
+}
+
+setup() {
+	root=$( realpath "$( dirname "$0" )" )
+	executable=$( basename "$0" )
+
+	if [ -z "$KERNEL_PATH" ]
+	then
+		KERNEL_PATH=$root
+	fi
+
+	if ! [ -z "$VBOOT_TOOLS_PATH" ]
+	then
+		PATH="$PATH:$VBOOT_TOOLS_PATH"
+	fi
+}
+
+cros_medium_setup() {
+	local action=$1
+	local storage_path=$2
+	local rootfs_tarball_path=$3
+	local kernel_files_path=$3
+
+	set -e
+
+	setup "$@"
+
+	if [ -z "$action" ] || [ -z "$storage_path" ]
+	then
+		usage
+		exit 1
+	fi
+
+	case $action in
+		"partitions")
+			requirements "udisksctl" "fdisk" "cgpt" "mkfs.ext4"
+			partitions "$storage_path"
+			;;
+		"rootfs")
+			if [ -z "$rootfs_tarball_path" ]
+			then
+				usage
+				exit 1
+			fi
+
+			requirements "udisksctl" "tar"
+			rootfs "$storage_path" "$rootfs_tarball_path"
+			;;
+		"kernel")
+			if [ -z "$kernel_files_path" ]
+			then
+				usage
+				exit 1
+			fi
+
+			requirements "udisksctl" "rsync"
+			kernel "$storage_path" "$kernel_files_path"
+			;;
+		*)
+			usage
+			exit 1
+			;;
+	esac
+}
+
+cros_medium_setup "$@"