diff options
| -rwxr-xr-x | projects/cros-scripts/install/cros-kernel-install | 286 | ||||
| -rw-r--r-- | projects/cros-scripts/install/install | 1 |
2 files changed, 287 insertions, 0 deletions
diff --git a/projects/cros-scripts/install/cros-kernel-install b/projects/cros-scripts/install/cros-kernel-install new file mode 100755 index 00000000..33168caa --- /dev/null +++ b/projects/cros-scripts/install/cros-kernel-install @@ -0,0 +1,286 @@ +#!/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" + +usage() { + printf "$executable [action] [storage] [kernel image|kernel modules]\n" >&2 + + printf "\nActions:\n" >&2 + printf " backup - Backup kernel image\n" >&2 + printf " image - Install kernel image\n" >&2 + printf " modules - Install kernel modules" >&2 + + usage_storage + + printf "\nEnvironment variables:\n" >&2 + printf " VBOOT_KEYS_PATH - Path to the vboot keys\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" +} + +storage_kernel_path() { + local storage_path=$1 + + cgpt find -t kernel "$storage_path" | head -n 1 +} + +storage_rootfs_path() { + local storage_path=$1 + + cgpt find -t rootfs "$storage_path" | head -n 1 +} + +backup() { + local storage_path=$1 + local kernel_image_path=$2 + + local storage_kernel_path=$( storage_kernel_path "$storage_path" ) + + if [ -z "$storage_kernel_path" ] + then + printf "No kernel partition found on storage $storage_path\n" >&2 + return 1 + fi + + cat "$storage_kernel_path" > "$kernel_image_path" + + printf "\nBacked up kernel image to $kernel_image_path\n" +} + +image() { + local storage_path=$1 + local kernel_image_path=$2 + + local storage_kernel_path=$( storage_kernel_path "$storage_path" ) + + if [ -z "$storage_kernel_path" ] + then + printf "No kernel partition found on storage $storage_path\n" >&2 + return 1 + fi + + storage_affect_confirm "$storage_path" + + cat "$kernel_image_path" > "$storage_kernel_path" + sync + + printf "\nInstalled kernel image on storage $storage_path\n" +} + +modules() { + local storage_path=$1 + local kernel_modules_path=$2 + + local storage_rootfs_path=$( storage_rootfs_path "$storage_path" ) + + if [ -z "$storage_rootfs_path" ] + then + printf "No rootfs partition found on storage $storage_path\n" >&2 + return 1 + fi + + 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" ) + + rsync -a --keep-dirlinks "$kernel_modules_path" "$storage_rootfs_mount_path/" + sync + + udisksctl unmount -b "$storage_rootfs_path" + + printf "\nInstalled kernel modules 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 "$VBOOT_TOOLS_PATH" ] + then + PATH="$PATH:$VBOOT_TOOLS_PATH" + fi + + if [ -z "$VBOOT_KEYS_PATH" ] + then + if ! [ -z "$VBOOT_TOOLS_PATH" ] && [ -d "$VBOOT_TOOLS_PATH/devkeys" ] + then + VBOOT_KEYS_PATH="$VBOOT_TOOLS_PATH/devkeys" + else + VBOOT_KEYS_PATH="/usr/share/vboot/devkeys" + fi + fi +} + +cros_kernel_install() { + local action=$1 + local storage_path=$2 + local kernel_image_path=$3 + local kernel_modules_path=$3 + + set -e + + setup "$@" + + if [ -z "$action" ] || [ -z "$storage_path" ] || [ -z "$kernel_image_path" ] || [ -z "$kernel_modules_path" ] + then + usage + exit 1 + fi + + case $action in + "backup") + requirements "cgpt" + backup "$storage_path" "$kernel_image_path" + ;; + "image") + requirements "cgpt" + image "$storage_path" "$kernel_image_path" + ;; + "modules") + requirements "cgpt" + modules "$storage_path" "$kernel_modules_path" + ;; + *) + usage + exit 1 + ;; + esac +} + +cros_kernel_install "$@" diff --git a/projects/cros-scripts/install/install b/projects/cros-scripts/install/install index 5e1e68b8..e29374d4 100644 --- a/projects/cros-scripts/install/install +++ b/projects/cros-scripts/install/install @@ -1,4 +1,5 @@ cros-boot-keys:cros-boot-keys cros-firmware-prepare:cros-firmware-prepare +cros-kernel-install:cros-kernel-install cros-kernel-prepare:cros-kernel-prepare cros-medium-setup:cros-medium-setup |