blob: ffb4d894e613ba73fbf481394ca1b6b6fc31e991 (
plain) (
tree)
|
|
#!/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="kernel"
IMG="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] [medium]\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 medium=$3
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-$medium.$IMG"
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
local medium=$4
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" ] || [ -z "$medium" ]
then
usage
exit 1
fi
requirements "udisksctl" "rsync"
kernel "$storage_path" "$kernel_files_path" "$medium"
;;
*)
usage
exit 1
;;
esac
}
cros_medium_setup "$@"
|