use fdisk to create partitions
This commit is contained in:
ading2210
parent
d3d7b6caff
commit
0c7003eb34
2
build.sh
2
build.sh
|
|
@ -14,7 +14,7 @@ print_help() {
|
|||
}
|
||||
|
||||
check_deps() {
|
||||
local needed_commands="cpio binwalk pcregrep"
|
||||
local needed_commands="cpio binwalk pcregrep cgpt realpath"
|
||||
for command in $needed_commands; do
|
||||
if ! command -v $command &> /dev/null; then
|
||||
echo $command
|
||||
|
|
|
|||
|
|
@ -0,0 +1,68 @@
|
|||
#!/bin/bash
|
||||
|
||||
set -e
|
||||
if [ "$DEBUG" ]; then
|
||||
set -x
|
||||
fi
|
||||
|
||||
make_mountable() {
|
||||
sh lib/ssd_util.sh --no_resign_kernel --remove_rootfs_verification -i $1
|
||||
printf '\000' | dd of=$1 seek=$((0x464 + 3)) conv=notrunc count=1 bs=1
|
||||
}
|
||||
|
||||
partition_disk() {
|
||||
local image_path=$(realpath "${1}")
|
||||
local bootloader_size=${2}
|
||||
|
||||
#create partitions
|
||||
(
|
||||
echo g #new gpt disk label
|
||||
|
||||
#create 1MB stateful
|
||||
echo n #new partition
|
||||
echo #accept default parition number
|
||||
echo #accept default first sector
|
||||
echo +1M #partition size is 1M
|
||||
|
||||
#create 32MB kernel partition
|
||||
echo n
|
||||
echo #accept default parition number
|
||||
echo #accept default first sector
|
||||
echo +32M #partition size is 32M
|
||||
echo t #change partition type
|
||||
echo #accept default parition number
|
||||
echo FE3A2A5D-4F32-41A7-B725-ACCC3285A309 #chromeos kernel type
|
||||
|
||||
#create bootloader partition
|
||||
echo n
|
||||
echo #accept default parition number
|
||||
echo #accept default first sector
|
||||
echo "+${bootloader_size}M" #set partition size
|
||||
echo t #change partition type
|
||||
echo #accept default parition number
|
||||
echo 3CB8E202-3B7E-47DD-8A3C-7FF2A13CFCEC #chromeos rootfs type
|
||||
|
||||
#create rootfs partition
|
||||
echo n
|
||||
echo #accept default parition number
|
||||
echo #accept default first sector
|
||||
echo #accept default size to fill rest of image
|
||||
|
||||
#write changes
|
||||
echo w
|
||||
) | fdisk $image_path
|
||||
}
|
||||
|
||||
create_image() {
|
||||
local image_path=$(realpath "${1}")
|
||||
local bootloader_size=${2}
|
||||
local rootfs_size=${3}
|
||||
|
||||
#stateful + kernel + bootloader + rootfs
|
||||
local total_size=$((1 + 32 + $bootloader_size + $rootfs_size))
|
||||
dd if=/dev/zero of=$image_path bs=1M oflag=sync count=$total_size status=progress
|
||||
|
||||
partition_disk $image_path $bootloader_size
|
||||
}
|
||||
|
||||
create_image ./test.bin 20 200
|
||||
|
|
@ -0,0 +1,664 @@
|
|||
#!/bin/sh
|
||||
#original: https://github.com/MercuryWorkshop/sh1mmer/blob/beautifulworld/wax/lib/ssd_util.sh
|
||||
|
||||
# this is a modified version of make_dev_ssd from cros. it allows the script to be run on any linux device and adds extra utility
|
||||
|
||||
#
|
||||
# Copyright (c) 2012 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.
|
||||
#
|
||||
# This script can change key (usually developer keys) and kernel config
|
||||
# of kernels on an disk image (usually for SSD but also works for USB).
|
||||
|
||||
SCRIPT_BASE="$(dirname "$0")"
|
||||
. "$SCRIPT_BASE/common_minimal.sh"
|
||||
load_shflags || exit 1
|
||||
|
||||
# Constants used by DEFINE_*
|
||||
VBOOT_BASE='/usr/share/vboot'
|
||||
DEFAULT_KEYS_FOLDER="$VBOOT_BASE/devkeys"
|
||||
DEFAULT_PARTITIONS='2 4'
|
||||
|
||||
# Only store backup in stateful partition if available.
|
||||
if [ -d /mnt/stateful_partition ]; then
|
||||
DEFAULT_BACKUP_FOLDER="/mnt/stateful_partition"
|
||||
else
|
||||
DEFAULT_BACKUP_FOLDER="$(pwd)"
|
||||
fi
|
||||
DEFAULT_BACKUP_FOLDER="${DEFAULT_BACKUP_FOLDER}/cros_sign_backups"
|
||||
|
||||
# TODO(hungte) The default image selection is no longer a SSD, so the script
|
||||
# works more like "make_dev_image". We may change the file name in future.
|
||||
ROOTDEV="$(rootdev -s 2>/dev/null)"
|
||||
ROOTDEV_PARTITION="$(echo $ROOTDEV | sed -n 's/.*[^0-9]\([0-9][0-9]*\)$/\1/p')"
|
||||
ROOTDEV_DISK="$(rootdev -s -d 2>/dev/null)"
|
||||
ROOTDEV_KERNEL="$((ROOTDEV_PARTITION - 1))"
|
||||
|
||||
# DEFINE_string name default_value description flag
|
||||
DEFINE_string image "$ROOTDEV_DISK" "Path to device or image file" "i"
|
||||
DEFINE_string keys "$DEFAULT_KEYS_FOLDER" "Path to folder of dev keys" "k"
|
||||
DEFINE_boolean remove_rootfs_verification \
|
||||
"${FLAGS_FALSE}" "Modify kernel boot config to disable rootfs verification" \
|
||||
"r"
|
||||
DEFINE_boolean unlock_arch \
|
||||
"${FLAGS_FALSE}" "Unlock arch"
|
||||
DEFINE_boolean lock_root \
|
||||
"${FLAGS_FALSE}" "lock rootfs"
|
||||
DEFINE_boolean lock_arch \
|
||||
"${FLAGS_FALSE}" "lock arch"
|
||||
DEFINE_boolean enable_earlycon "${FLAGS_FALSE}" \
|
||||
"Enable earlycon from stdout-path (ARM/ARM64) or SPCR (x86)." ""
|
||||
DEFINE_boolean disable_earlycon "${FLAGS_FALSE}" \
|
||||
"Disable earlycon." ""
|
||||
DEFINE_boolean enable_console "${FLAGS_FALSE}" \
|
||||
"Enable serial console." ""
|
||||
DEFINE_boolean disable_console "${FLAGS_FALSE}" \
|
||||
"Disable serial console." ""
|
||||
DEFINE_string backup_dir \
|
||||
"$DEFAULT_BACKUP_FOLDER" "Path of directory to store kernel backups" ""
|
||||
DEFINE_string save_config "" \
|
||||
"Base filename to store kernel configs to, instead of resigning." ""
|
||||
DEFINE_string set_config "" \
|
||||
"Base filename to load kernel configs from" ""
|
||||
DEFINE_boolean edit_config "${FLAGS_FALSE}" \
|
||||
"Edit kernel config in-place." ""
|
||||
DEFINE_string partitions "" \
|
||||
"List of partitions to examine (default: $DEFAULT_PARTITIONS)" ""
|
||||
DEFINE_boolean recovery_key "$FLAGS_FALSE" \
|
||||
"Use recovery key to sign image (to boot from USB)" ""
|
||||
DEFINE_boolean force "$FLAGS_FALSE" \
|
||||
"Skip validity checks and make the change" "f"
|
||||
DEFINE_boolean default_rw_root "${FLAGS_TRUE}" \
|
||||
"When --remove_rootfs_verification is set, change root mount option to RW." ""
|
||||
DEFINE_boolean no_resign_kernel "${FLAGS_FALSE}" \
|
||||
"make changes without resigning the kernel" ""
|
||||
# Parse command line
|
||||
FLAGS "$@" || exit 1
|
||||
ORIGINAL_CMD="$0"
|
||||
ORIGINAL_PARAMS="$@"
|
||||
eval set -- "$FLAGS_ARGV"
|
||||
ORIGINAL_PARTITIONS="$FLAGS_partitions"
|
||||
: ${FLAGS_partitions:=$DEFAULT_PARTITIONS}
|
||||
|
||||
# Globals
|
||||
# ----------------------------------------------------------------------------
|
||||
set -e
|
||||
|
||||
# a log file to keep the output results of executed command
|
||||
EXEC_LOG="$(make_temp_file)"
|
||||
|
||||
# Functions
|
||||
# ----------------------------------------------------------------------------
|
||||
|
||||
# Removes rootfs verification from kernel boot parameter
|
||||
# And strip out bootcache args if it exists
|
||||
remove_rootfs_verification() {
|
||||
local new_root="PARTUUID=%U/PARTNROFF=1"
|
||||
# the first line in sed is to strip out bootcache details
|
||||
local rw_root_opt="s| ro | rw |"
|
||||
if [ "${FLAGS_default_rw_root}" = "${FLAGS_FALSE}" ]; then
|
||||
rw_root_opt="s| rw | ro |"
|
||||
fi
|
||||
|
||||
echo "$*" | sed '
|
||||
s| dm=\"2 [^"]*bootcache[^"]* vroot | dm=\"1 vroot |
|
||||
s| root=/dev/dm-[0-9] | root='"$new_root"' |
|
||||
s| dm_verity.dev_wait=1 | dm_verity.dev_wait=0 |
|
||||
s| payload=PARTUUID=%U/PARTNROFF=1 | payload=ROOT_DEV |
|
||||
s| hashtree=PARTUUID=%U/PARTNROFF=1 | hashtree=HASH_DEV |
|
||||
'"${rw_root_opt}"
|
||||
}
|
||||
|
||||
remove_legacy_boot_rootfs_verification() {
|
||||
# See src/scripts/create_legacy_bootloader_templates
|
||||
local image="$1"
|
||||
local mount_point="$(make_temp_dir)"
|
||||
local config_file
|
||||
debug_msg "Removing rootfs verification for legacy boot configuration."
|
||||
mount_image_partition "$image" 12 "$mount_point" || return $FLAGS_FALSE
|
||||
config_file="$mount_point/efi/boot/grub.cfg"
|
||||
[ ! -f "${config_file}" ] ||
|
||||
sudo sed -i -e 's/^ *defaultA=2 *$/defaultA=0/g' \
|
||||
-e 's/^ *defaultB=3 *$/defaultB=1/g' "${config_file}"
|
||||
config_file="$mount_point/syslinux/default.cfg"
|
||||
[ ! -f "$config_file" ] ||
|
||||
sudo sed -i 's/-vusb/-usb/g; s/-vhd/-hd/g' "$config_file"
|
||||
sudo umount "$mount_point"
|
||||
}
|
||||
|
||||
# Enable/Disable earlycon or serial console
|
||||
insert_parameter() {
|
||||
local cmdline="$1"
|
||||
local param="$2"
|
||||
|
||||
if [ -n "${cmdline##*${param}*}" ]; then
|
||||
cmdline="${param} ${cmdline}"
|
||||
fi
|
||||
|
||||
echo "${cmdline}"
|
||||
}
|
||||
|
||||
remove_parameter() {
|
||||
local cmdline="$1"
|
||||
local param="$2"
|
||||
|
||||
cmdline=$(echo "${cmdline}" | sed '
|
||||
s/'"${param} "'//g')
|
||||
|
||||
echo "${cmdline}"
|
||||
}
|
||||
|
||||
# Wrapped version of dd
|
||||
mydd() {
|
||||
# oflag=sync is safer, but since we need bs=512, syncing every block would be
|
||||
# very slow.
|
||||
dd "$@" >"$EXEC_LOG" 2>&1 ||
|
||||
die "Failed in [dd $*], Message: $(cat "${EXEC_LOG}")"
|
||||
}
|
||||
|
||||
# Prints a more friendly name from kernel index number
|
||||
cros_kernel_name() {
|
||||
case $1 in
|
||||
2)
|
||||
echo "Kernel A"
|
||||
;;
|
||||
4)
|
||||
echo "Kernel B"
|
||||
;;
|
||||
6)
|
||||
echo "Kernel C"
|
||||
;;
|
||||
*)
|
||||
echo "Partition $1"
|
||||
esac
|
||||
}
|
||||
|
||||
find_valid_kernel_partitions() {
|
||||
local part_id
|
||||
local valid_partitions=""
|
||||
for part_id in $*; do
|
||||
local name="$(cros_kernel_name $part_id)"
|
||||
local kernel_part="$(make_partition_dev "$FLAGS_image" "$part_id")"
|
||||
info "futil: $FUTILITY"
|
||||
if [ -z "$(${FUTILITY} dump_kernel_config "$kernel_part" 2>"$EXEC_LOG")" ]; then
|
||||
info "${name}: no kernel boot information, ignored." >&2
|
||||
else
|
||||
[ -z "$valid_partitions" ] &&
|
||||
valid_partitions="$part_id" ||
|
||||
valid_partitions="$valid_partitions $part_id"
|
||||
continue
|
||||
fi
|
||||
done
|
||||
debug_msg "find_valid_kernel_partitions: [$*] -> [$valid_partitions]"
|
||||
echo "$valid_partitions"
|
||||
}
|
||||
|
||||
# Resigns a kernel on SSD or image.
|
||||
resign_ssd_kernel() {
|
||||
local ssd_device="$1"
|
||||
local bs="$(blocksize "${ssd_device}")"
|
||||
|
||||
# reasonable size for current kernel partition
|
||||
local min_kernel_size=$((8000 * 1024 / bs))
|
||||
local resigned_kernels=0
|
||||
|
||||
for kernel_index in $FLAGS_partitions; do
|
||||
local old_blob="$(make_temp_file)"
|
||||
local new_blob="$(make_temp_file)"
|
||||
local name="$(cros_kernel_name $kernel_index)"
|
||||
local rootfs_index="$(($kernel_index + 1))"
|
||||
|
||||
debug_msg "Probing $name information"
|
||||
local offset size
|
||||
offset="$(partoffset "$ssd_device" "$kernel_index")" ||
|
||||
die "Failed to get partition ${kernel_index} offset from ${ssd_device}"
|
||||
size="$(partsize "$ssd_device" "$kernel_index")" ||
|
||||
die "Failed to get partition ${kernel_index} size from ${ssd_device}"
|
||||
if [ ! $size -gt $min_kernel_size ]; then
|
||||
info "${name} seems too small (${size}), ignored."
|
||||
continue
|
||||
fi
|
||||
|
||||
debug_msg "Reading $name from partition $kernel_index"
|
||||
mydd if="$ssd_device" of="$old_blob" bs=$bs skip=$offset count=$size
|
||||
|
||||
debug_msg "Checking if $name is valid"
|
||||
|
||||
local kernel_config
|
||||
if ! kernel_config="$(${FUTILITY} dump_kernel_config "$old_blob" 2>"$EXEC_LOG")"; then
|
||||
debug_msg "dump_kernel_config error message: $(cat "$EXEC_LOG")"
|
||||
info "${name}: no kernel boot information, ignored."
|
||||
continue
|
||||
fi
|
||||
|
||||
|
||||
if [ "$FLAGS_no_resign_kernel" = "$FLAGS_TRUE" ]; then
|
||||
info "Skipping resign for ${FLAGS_image}"
|
||||
if [ ${FLAGS_remove_rootfs_verification} = $FLAGS_TRUE ]; then
|
||||
local root_offset_sector=$(partoffset "$ssd_device" $rootfs_index)
|
||||
local root_offset_bytes=$((root_offset_sector * bs))
|
||||
info "dev: ${ssd_device} $root_offset_bytes"
|
||||
if ! is_ext2 "$ssd_device" "$root_offset_bytes"; then
|
||||
debug_msg "Non-ext2 partition: $ssd_device$rootfs_index, skip."
|
||||
elif ! rw_mount_disabled "$ssd_device" "$root_offset_bytes"; then
|
||||
debug_msg "Root file system is writable. No need to modify."
|
||||
else
|
||||
# disable the RO ext2 hack
|
||||
info "Disabling rootfs ext2 RO bit hack"
|
||||
enable_rw_mount "$ssd_device" "$root_offset_bytes" >"$EXEC_LOG" 2>&1 ||
|
||||
die "Failed turning off rootfs RO bit. OS may be corrupted. " \
|
||||
"Message: $(cat "${EXEC_LOG}")"
|
||||
fi
|
||||
fi
|
||||
return
|
||||
fi
|
||||
|
||||
if [ -n "${FLAGS_save_config}" ]; then
|
||||
# Save current kernel config
|
||||
local old_config_file
|
||||
old_config_file="${FLAGS_save_config}.$kernel_index"
|
||||
info "Saving ${name} config to ${old_config_file}"
|
||||
echo "$kernel_config" > "$old_config_file"
|
||||
# Just save; don't resign
|
||||
continue
|
||||
fi
|
||||
|
||||
if [ -n "${FLAGS_set_config}" ]; then
|
||||
# Set new kernel config from file
|
||||
local new_config_file
|
||||
new_config_file="${FLAGS_set_config}.$kernel_index"
|
||||
kernel_config="$(cat "$new_config_file")" ||
|
||||
die "Failed to read new kernel config from ${new_config_file}"
|
||||
debug_msg "New kernel config: $kernel_config)"
|
||||
info "${name}: Replaced config from ${new_config_file}"
|
||||
fi
|
||||
|
||||
if [ "${FLAGS_edit_config}" = ${FLAGS_TRUE} ]; then
|
||||
debug_msg "Editing kernel config file."
|
||||
local new_config_file="$(make_temp_file)"
|
||||
echo "${kernel_config}" >"${new_config_file}"
|
||||
local old_md5sum="$(md5sum "${new_config_file}")"
|
||||
local editor="${VISUAL:-${EDITOR:-vi}}"
|
||||
info "${name}: Editing kernel config:"
|
||||
# On ChromiumOS, some builds may come with broken EDITOR that refers to
|
||||
# nano so we want to check again if the editor really exists.
|
||||
if type "${editor}" >/dev/null 2>&1; then
|
||||
"${editor}" "${new_config_file}"
|
||||
else
|
||||
# This script runs under dash but we want readline in bash to support
|
||||
# editing in in console.
|
||||
bash -c "read -e -i '${kernel_config}' &&
|
||||
echo \"\${REPLY}\" >${new_config_file}" ||
|
||||
die "Failed to run editor. Please specify editor name by VISUAL."
|
||||
fi
|
||||
kernel_config="$(cat "${new_config_file}")"
|
||||
if [ "$(md5sum "${new_config_file}")" = "${old_md5sum}" ]; then
|
||||
info "${name}: Config not changed."
|
||||
else
|
||||
debug_msg "New kernel config: ${kernel_config})"
|
||||
info "${name}: Config updated"
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ ${FLAGS_remove_rootfs_verification} = $FLAGS_FALSE ]; then
|
||||
debug_msg "Bypassing rootfs verification check"
|
||||
else
|
||||
debug_msg "Changing boot parameter to remove rootfs verification"
|
||||
kernel_config="$(remove_rootfs_verification "$kernel_config")"
|
||||
debug_msg "New kernel config: $kernel_config"
|
||||
info "${name}: Disabled rootfs verification."
|
||||
remove_legacy_boot_rootfs_verification "$ssd_device"
|
||||
fi
|
||||
|
||||
if [ "${FLAGS_enable_earlycon}" = "${FLAGS_TRUE}" ]; then
|
||||
debug_msg "Enabling earlycon"
|
||||
kernel_config="$(insert_parameter "${kernel_config}" "earlycon")"
|
||||
debug_msg "New kernel config: ${kernel_config}"
|
||||
elif [ "${FLAGS_disable_earlycon}" = "${FLAGS_TRUE}" ]; then
|
||||
debug_msg "Disabling earlycon"
|
||||
kernel_config="$(remove_parameter "${kernel_config}" "earlycon")"
|
||||
debug_msg "New kernel config: ${kernel_config}"
|
||||
fi
|
||||
|
||||
if [ "${FLAGS_enable_console}" = "${FLAGS_TRUE}" ]; then
|
||||
debug_msg "Enabling serial console"
|
||||
kernel_config="$(remove_parameter "${kernel_config}" "console=")"
|
||||
debug_msg "New kernel config: ${kernel_config}"
|
||||
elif [ "${FLAGS_disable_console}" = "${FLAGS_TRUE}" ]; then
|
||||
debug_msg "Disabling serial console"
|
||||
kernel_config="$(insert_parameter "${kernel_config}" "console=")"
|
||||
debug_msg "New kernel config: ${kernel_config}"
|
||||
fi
|
||||
|
||||
local new_kernel_config_file="$(make_temp_file)"
|
||||
echo -n "$kernel_config" >"$new_kernel_config_file"
|
||||
|
||||
|
||||
|
||||
|
||||
debug_msg "Re-signing $name from $old_blob to $new_blob"
|
||||
debug_msg "Using key: $KERNEL_DATAKEY"
|
||||
vbutil_kernel \
|
||||
--repack "$new_blob" \
|
||||
--keyblock "$KERNEL_KEYBLOCK" \
|
||||
--config "$new_kernel_config_file" \
|
||||
--signprivate "$KERNEL_DATAKEY" \
|
||||
--oldblob "$old_blob" >"$EXEC_LOG" 2>&1 ||
|
||||
die "Failed to resign ${name}. Message: $(cat "${EXEC_LOG}")"
|
||||
|
||||
debug_msg "Creating new kernel image (vboot+code+config)"
|
||||
local new_kern="$(make_temp_file)"
|
||||
cp "$old_blob" "$new_kern"
|
||||
mydd if="$new_blob" of="$new_kern" conv=notrunc
|
||||
|
||||
if is_debug_mode; then
|
||||
debug_msg "for debug purposes, check *.dbgbin"
|
||||
cp "$old_blob" old_blob.dbgbin
|
||||
cp "$new_blob" new_blob.dbgbin
|
||||
cp "$new_kern" new_kern.dbgbin
|
||||
fi
|
||||
|
||||
debug_msg "Verifying new kernel and keys"
|
||||
vbutil_kernel \
|
||||
--verify "$new_kern" \
|
||||
--signpubkey "$KERNEL_PUBKEY" --verbose >"$EXEC_LOG" 2>&1 ||
|
||||
die "Failed to verify new ${name}. Message: $(cat "${EXEC_LOG}")"
|
||||
|
||||
debug_msg "Backup old kernel blob"
|
||||
local backup_date_time="$(date +'%Y%m%d_%H%M%S')"
|
||||
local backup_name="$(echo "$name" | sed 's/ /_/g; s/^K/k/')"
|
||||
local backup_file_name="${backup_name}_${backup_date_time}.bin"
|
||||
local backup_file_path="$FLAGS_backup_dir/$backup_file_name"
|
||||
if mkdir -p "$FLAGS_backup_dir" &&
|
||||
cp -f "$old_blob" "$backup_file_path"; then
|
||||
info "Backup of ${name} is stored in: ${backup_file_path}"
|
||||
else
|
||||
warn "Cannot create file in ${FLAGS_backup_dir} ... Ignore backups."
|
||||
fi
|
||||
|
||||
debug_msg "Writing $name to partition $kernel_index"
|
||||
mydd \
|
||||
if="$new_kern" \
|
||||
of="$ssd_device" \
|
||||
seek=$offset \
|
||||
bs=$bs \
|
||||
count=$size \
|
||||
conv=notrunc
|
||||
resigned_kernels=$(($resigned_kernels + 1))
|
||||
|
||||
debug_msg "Make the root file system writable if needed."
|
||||
# TODO(hungte) for safety concern, a more robust way would be to:
|
||||
# (1) change kernel config to ro
|
||||
# (2) check if we can enable rw mount
|
||||
# (3) change kernel config to rw
|
||||
|
||||
if [ ${FLAGS_lock_root} = $FLAGS_TRUE ]; then
|
||||
local root_offset_sector=$(partoffset "$ssd_device" $rootfs_index)
|
||||
local root_offset_bytes=$((root_offset_sector * bs))
|
||||
# enable the RO ext2 hack
|
||||
if ! is_ext2 "$ssd_device" "$root_offset_bytes"; then
|
||||
debug_msg "Non-ext2 partition: $ssd_device$rootfs_index, skip."
|
||||
else
|
||||
echo "Re-enabling the ext2 hack :trolley:"
|
||||
disable_rw_mount "$ssd_device" "$root_offset_bytes" >"$EXEC_LOG" 2>&1 ||
|
||||
die "Failed turning off rootfs RO bit. OS may be corrupted. " \
|
||||
"Message: $(cat "${EXEC_LOG}")"
|
||||
fi
|
||||
fi
|
||||
if [ ${FLAGS_lock_arch} = $FLAGS_TRUE ]; then
|
||||
local root_offset_sector=$(partoffset "$ssd_device" 13)
|
||||
local root_offset_bytes=$((root_offset_sector * bs))
|
||||
# enable the RO ext2 hack
|
||||
if ! is_ext2 "$ssd_device" "$root_offset_bytes"; then
|
||||
debug_msg "Non-ext2 partition: $ssd_device$rootfs_index, skip."
|
||||
else
|
||||
echo "Locking arch partition"
|
||||
disable_rw_mount "$ssd_device" "$root_offset_bytes" >"$EXEC_LOG" 2>&1 ||
|
||||
die "Failed turning off rootfs RO bit. OS may be corrupted. " \
|
||||
"Message: $(cat "${EXEC_LOG}")"
|
||||
fi
|
||||
fi
|
||||
if [ ${FLAGS_unlock_arch} = $FLAGS_TRUE ]; then
|
||||
local root_offset_sector=$(partoffset "$ssd_device" 13)
|
||||
local root_offset_bytes=$((root_offset_sector * bs))
|
||||
# enable the RO ext2 hack
|
||||
if ! is_ext2 "$ssd_device" "$root_offset_bytes"; then
|
||||
echo "Non-ext2 partition: $ssd_device$rootfs_index, skip."
|
||||
else
|
||||
echo "unlocking arch partition"
|
||||
enable_rw_mount "$ssd_device" "$root_offset_bytes" >"$EXEC_LOG" 2>&1 ||
|
||||
die "Failed turning off rootfs RO bit. OS may be corrupted. " \
|
||||
"Message: $(cat "${EXEC_LOG}")"
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ ${FLAGS_remove_rootfs_verification} = $FLAGS_TRUE ]; then
|
||||
local root_offset_sector=$(partoffset "$ssd_device" $rootfs_index)
|
||||
local root_offset_bytes=$((root_offset_sector * bs))
|
||||
if ! is_ext2 "$ssd_device" "$root_offset_bytes"; then
|
||||
debug_msg "Non-ext2 partition: $ssd_device$rootfs_index, skip."
|
||||
elif ! rw_mount_disabled "$ssd_device" "$root_offset_bytes"; then
|
||||
debug_msg "Root file system is writable. No need to modify."
|
||||
else
|
||||
# disable the RO ext2 hack
|
||||
info "Disabling rootfs ext2 RO bit hack"
|
||||
enable_rw_mount "$ssd_device" "$root_offset_bytes" >"$EXEC_LOG" 2>&1 ||
|
||||
die "Failed turning off rootfs RO bit. OS may be corrupted. " \
|
||||
"Message: $(cat "${EXEC_LOG}")"
|
||||
fi
|
||||
fi
|
||||
|
||||
# Sometimes doing "dump_kernel_config" or other I/O now (or after return to
|
||||
# shell) will get the data before modification. Not a problem now, but for
|
||||
# safety, let's try to sync more.
|
||||
sync; sync; sync
|
||||
|
||||
info "${name}: Re-signed with developer keys successfully."
|
||||
done
|
||||
|
||||
# If we saved the kernel config, exit now so we don't print an error
|
||||
if [ -n "${FLAGS_save_config}" ]; then
|
||||
info "(Kernels have not been resigned.)"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
return $resigned_kernels
|
||||
}
|
||||
|
||||
validity_check_crossystem_flags() {
|
||||
debug_msg "crossystem validity check"
|
||||
if [ -n "${FLAGS_save_config}" ]; then
|
||||
debug_msg "not resigning kernel."
|
||||
return
|
||||
fi
|
||||
|
||||
if [ "$(crossystem dev_boot_signed_only)" = "0" ]; then
|
||||
debug_msg "dev_boot_signed_only not set - safe."
|
||||
return
|
||||
fi
|
||||
|
||||
echo "
|
||||
ERROR: YOUR FIRMWARE WILL ONLY BOOT SIGNED IMAGES.
|
||||
|
||||
Modifying the kernel or root filesystem will result in an unusable system. If
|
||||
you really want to make this change, allow the firmware to boot self-signed
|
||||
images by running:
|
||||
|
||||
sudo crossystem dev_boot_signed_only=0
|
||||
|
||||
before re-executing this command.
|
||||
"
|
||||
return $FLAGS_FALSE
|
||||
}
|
||||
|
||||
validity_check_live_partitions() {
|
||||
debug_msg "Partition validity check"
|
||||
if [ "$FLAGS_partitions" = "$ROOTDEV_KERNEL" ]; then
|
||||
debug_msg "only for current active partition - safe."
|
||||
return
|
||||
fi
|
||||
if [ "$ORIGINAL_PARTITIONS" != "" ]; then
|
||||
debug_msg "user has assigned partitions - provide more info."
|
||||
info "Making change to ${FLAGS_partitions} on ${FLAGS_image}."
|
||||
return
|
||||
fi
|
||||
echo "
|
||||
ERROR: YOU ARE TRYING TO MODIFY THE LIVE SYSTEM IMAGE $FLAGS_image.
|
||||
|
||||
The system may become unusable after that change, especially when you have
|
||||
some auto updates in progress. To make it safer, we suggest you to only
|
||||
change the partition you have booted with. To do that, re-execute this command
|
||||
as:
|
||||
|
||||
sudo $ORIGINAL_CMD $ORIGINAL_PARAMS --partitions $ROOTDEV_KERNEL
|
||||
|
||||
If you are sure to modify other partition, please invoke the command again and
|
||||
explicitly assign only one target partition for each time (--partitions N )
|
||||
"
|
||||
return $FLAGS_FALSE
|
||||
}
|
||||
|
||||
validity_check_live_firmware() {
|
||||
debug_msg "Firmware compatibility validity check"
|
||||
if [ "$(crossystem mainfw_type)" = "developer" ]; then
|
||||
debug_msg "developer type firmware in active."
|
||||
return
|
||||
fi
|
||||
debug_msg "Loading firmware to check root key..."
|
||||
local bios_image="$(make_temp_file)"
|
||||
local rootkey_file="$(make_temp_file)"
|
||||
info "checking system firmware..."
|
||||
sudo flashrom -p host -i GBB -r "$bios_image" >/dev/null 2>&1
|
||||
futility gbb -g --rootkey="$rootkey_file" "$bios_image" >/dev/null 2>&1
|
||||
if [ ! -s "$rootkey_file" ]; then
|
||||
debug_msg "failed to read root key from system firmware..."
|
||||
else
|
||||
# The magic 130 is counted by "od dev-rootkey" for the lines until the body
|
||||
# of key is reached. Trailing bytes (0x00 or 0xFF - both may appear, and
|
||||
# that's why we need to skip them) are started at line 131.
|
||||
# TODO(hungte) compare with rootkey in $VBOOT_BASE directly.
|
||||
local rootkey_hash="$(od "$rootkey_file" |
|
||||
head -130 | md5sum |
|
||||
sed 's/ .*$//' )"
|
||||
if [ "$rootkey_hash" = "a13642246ef93daaf75bd791446fec9b" ]; then
|
||||
debug_msg "detected DEV root key in firmware."
|
||||
return
|
||||
else
|
||||
debug_msg "non-devkey hash: $rootkey_hash"
|
||||
fi
|
||||
fi
|
||||
|
||||
echo "
|
||||
ERROR: YOU ARE NOT USING DEVELOPER FIRMWARE, AND RUNNING THIS COMMAND MAY
|
||||
THROW YOUR CHROMEOS DEVICE INTO UN-BOOTABLE STATE.
|
||||
|
||||
You need to either install developer firmware, or change system root key.
|
||||
|
||||
- To install developer firmware: type command
|
||||
sudo chromeos-firmwareupdate --mode=todev
|
||||
|
||||
- To change system rootkey: disable firmware write protection (a hardware
|
||||
switch) and then type command:
|
||||
sudo $SCRIPT_BASE/make_dev_firmware.sh
|
||||
|
||||
If you are sure that you want to make such image without developer
|
||||
firmware or you've already changed system root keys, please run this
|
||||
command again with --force paramemeter:
|
||||
|
||||
sudo $ORIGINAL_CMD --force $ORIGINAL_PARAMS
|
||||
"
|
||||
return $FLAGS_FALSE
|
||||
}
|
||||
|
||||
validity_check() {
|
||||
validity_check_live_partitions || return $FLAGS_FALSE
|
||||
|
||||
# Remaining checks depend on firmware; skip if device is running in a VM.
|
||||
if crossystem 'inside_vm?1'; then
|
||||
debug_msg "Device is a VM, skipping firmware checks"
|
||||
return $FLAGS_TRUE
|
||||
fi
|
||||
|
||||
validity_check_live_firmware || return $FLAGS_FALSE
|
||||
validity_check_crossystem_flags || return $FLAGS_FALSE
|
||||
return $FLAGS_TRUE
|
||||
}
|
||||
|
||||
# Main
|
||||
# ----------------------------------------------------------------------------
|
||||
main() {
|
||||
local num_signed=0
|
||||
local num_given=$(echo "$FLAGS_partitions" | wc -w)
|
||||
local kaltdir=$(dirname $0)/keys
|
||||
if [ -d "$kaltdir" ]; then
|
||||
FLAGS_keys=$kaltdir
|
||||
fi
|
||||
# Check parameters
|
||||
if [ "$FLAGS_recovery_key" = "$FLAGS_TRUE" ]; then
|
||||
KERNEL_KEYBLOCK="$FLAGS_keys/recovery_kernel.keyblock"
|
||||
KERNEL_DATAKEY="$FLAGS_keys/recovery_kernel_data_key.vbprivk"
|
||||
KERNEL_PUBKEY="$FLAGS_keys/recovery_key.vbpubk"
|
||||
else
|
||||
KERNEL_KEYBLOCK="$FLAGS_keys/kernel.keyblock"
|
||||
KERNEL_DATAKEY="$FLAGS_keys/kernel_data_key.vbprivk"
|
||||
KERNEL_PUBKEY="$FLAGS_keys/kernel_subkey.vbpubk"
|
||||
fi
|
||||
|
||||
debug_msg "Prerequisite check"
|
||||
ensure_files_exist \
|
||||
"$KERNEL_KEYBLOCK" \
|
||||
"$KERNEL_DATAKEY" \
|
||||
"$KERNEL_PUBKEY" \
|
||||
"$FLAGS_image" ||
|
||||
exit 1
|
||||
|
||||
# checks for running on a live system image.
|
||||
if [ "$FLAGS_image" = "$ROOTDEV_DISK" ]; then
|
||||
debug_msg "check valid kernel partitions for live system"
|
||||
local valid_partitions="$(find_valid_kernel_partitions $FLAGS_partitions)"
|
||||
[ -n "$valid_partitions" ] ||
|
||||
die "No valid kernel partitions on ${FLAGS_image} (${FLAGS_partitions})."
|
||||
FLAGS_partitions="$valid_partitions"
|
||||
|
||||
# Validity checks
|
||||
if [ "$FLAGS_force" = "$FLAGS_TRUE" ]; then
|
||||
echo "
|
||||
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
! INFO: ALL VALIDITY CHECKS WERE BYPASSED. YOU ARE ON YOUR OWN. !
|
||||
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||
" >&2
|
||||
local i
|
||||
for i in $(seq 5 -1 1); do
|
||||
echo -n "\rStart in $i second(s) (^C to abort)... " >&2
|
||||
sleep 1
|
||||
done
|
||||
echo ""
|
||||
elif ! validity_check; then
|
||||
die "IMAGE ${FLAGS_image} IS NOT MODIFIED."
|
||||
fi
|
||||
fi
|
||||
|
||||
resign_ssd_kernel "$FLAGS_image" || num_signed=$?
|
||||
|
||||
debug_msg "Complete."
|
||||
if [ $num_signed -gt 0 -a $num_signed -le $num_given ]; then
|
||||
# signed something at least
|
||||
info "Successfully re-signed ${num_signed} of ${num_given} kernel(s)" \
|
||||
" on device ${FLAGS_image}."
|
||||
info "Please remember to reboot before updating the kernel on this device."
|
||||
else
|
||||
if [ "$FLAGS_no_resign_kernel" = "$FLAGS_FALSE" ]; then
|
||||
die "Failed re-signing kernels."
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
# People using this to process images may forget to add "-i",
|
||||
# so adding parameter check is safer.
|
||||
if [ "$#" -gt 0 ]; then
|
||||
flags_help
|
||||
die "Unknown parameters: $*"
|
||||
fi
|
||||
|
||||
main
|
||||
Loading…
Reference in New Issue