194 lines
6.6 KiB
Bash
194 lines
6.6 KiB
Bash
#!/bin/bash
|
|
# 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; version 2 of the License.
|
|
#
|
|
# 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.
|
|
|
|
ignore_error() {
|
|
"$@" 2>/dev/null
|
|
return 0
|
|
}
|
|
|
|
parse_fstab(){
|
|
echo $(perl -ane 'printf("%s:%s\n", @F[0,1]) if $F[0] =~ m#^UUID=#;' $1/etc/fstab)
|
|
# perl -ane 'printf("%s:%s\n", @F[0,1]) if $F[0] =~ m#^/dev#;' $1/etc/fstab
|
|
# perl -ane 'printf("%s:%s\n", @F[0,1]) if $F[0] =~ m#^LABEL=#;' $1/etc/fstab
|
|
}
|
|
|
|
detect(){
|
|
local detected="$(os-prober | tr ' ' '_' | paste -s -d ' ')"
|
|
echo ${detected}
|
|
}
|
|
|
|
# $1: os-prober array
|
|
get_os_name(){
|
|
local str=$1
|
|
str="${str#*:}"
|
|
str="${str#*:}"
|
|
str="${str%:*}"
|
|
echo "$str"
|
|
}
|
|
|
|
get_chroot_arch(){
|
|
local elf=$(file $1/usr/bin/file)
|
|
elf=${elf//*executable,}
|
|
echo ${elf%%,*}
|
|
}
|
|
|
|
chroot_part_mount() {
|
|
info "mount: [%s]" "$2"
|
|
mount "$@" && CHROOT_ACTIVE_PART_MOUNTS=("$2" "${CHROOT_ACTIVE_PART_MOUNTS[@]}")
|
|
}
|
|
|
|
select_os(){
|
|
local detected_os_list=( $(detect) ) os_list=() count select os_str os_root
|
|
for os in ${detected_os_list[@]}; do
|
|
case ${os##*:} in
|
|
'linux') os_list+=($os)
|
|
esac
|
|
done
|
|
count=${#os_list[@]}
|
|
if [[ ${count} < 1 ]]; then
|
|
die "No Linux partitions detected!"
|
|
fi
|
|
if [[ ${count} > 1 ]]; then
|
|
msg "Detected systems:"
|
|
local i=0
|
|
for os in ${os_list[@]}; do
|
|
info "$i) $(get_os_name $os)"; i=$((i+1))
|
|
done
|
|
msg "Select system to mount [0-%s] : " "$((count-1))"
|
|
read select
|
|
else
|
|
select=0
|
|
fi
|
|
os_str=${os_list[$select]}
|
|
os_root=${os_str%%:*}
|
|
msg "Mounting (%s) [%s]" "$(get_os_name $os_str)" "$os_root"
|
|
chroot_mount_partitions "$1" "$os_root"
|
|
}
|
|
|
|
chroot_mount_partitions(){
|
|
|
|
CHROOT_ACTIVE_PART_MOUNTS=()
|
|
CHROOT_ACTIVE_MOUNTS=()
|
|
|
|
[[ $(trap -p EXIT) ]] && die 'Error! Attempting to overwrite existing EXIT trap'
|
|
trap 'trap_handler' EXIT
|
|
|
|
chroot_part_mount $2 $1
|
|
|
|
local mounts=$(parse_fstab "$1")
|
|
|
|
for entry in ${mounts[@]}; do
|
|
entry=${entry//UUID=}
|
|
local dev=${entry%:*} mp=${entry#*:}
|
|
case "${entry#*:}" in
|
|
'/'|'swap'|'none') continue ;;
|
|
*) chroot_part_mount "/dev/disk/by-uuid/${dev}" "$1${mp}" ;;
|
|
esac
|
|
done
|
|
|
|
local chroot_arch=$(get_chroot_arch $1)
|
|
[[ ${chroot_arch} == x86-64 ]] && chroot_arch=${chroot_arch/-/_}
|
|
case ${target_arch} in
|
|
i686)
|
|
if [[ ${chroot_arch} == x86_64 ]]; then
|
|
die "You can't chroot from %s host into %s!" "${target_arch}" "${chroot_arch}"
|
|
fi
|
|
;;
|
|
esac
|
|
|
|
chroot_mount_conditional "! mountpoint -q '$1'" "$1" "$1" --bind &&
|
|
chroot_mount proc "$1/proc" -t proc -o nosuid,noexec,nodev &&
|
|
chroot_mount sys "$1/sys" -t sysfs -o nosuid,noexec,nodev,ro &&
|
|
ignore_error chroot_mount_conditional "[[ -d '$1/sys/firmware/efi/efivars' ]]" \
|
|
efivarfs "$1/sys/firmware/efi/efivars" -t efivarfs -o nosuid,noexec,nodev &&
|
|
chroot_mount udev "$1/dev" -t devtmpfs -o mode=0755,nosuid &&
|
|
chroot_mount devpts "$1/dev/pts" -t devpts -o mode=0620,gid=5,nosuid,noexec &&
|
|
chroot_mount shm "$1/dev/shm" -t tmpfs -o mode=1777,nosuid,nodev &&
|
|
chroot_mount run "$1/run" -t tmpfs -o nosuid,nodev,mode=0755 &&
|
|
chroot_mount tmp "$1/tmp" -t tmpfs -o mode=1777,strictatime,nodev,nosuid
|
|
chroot_mount /etc/resolv.conf "$1/etc/resolv.conf" --bind
|
|
info "chroot_mount: active mounts [%s]" "${CHROOT_ACTIVE_MOUNTS[@]}"
|
|
info "chroot_mount: active part mounts [%s]" "${CHROOT_ACTIVE_PART_MOUNTS[@]}"
|
|
}
|
|
|
|
chroot_mount() {
|
|
info "mount: [%s]" "$2"
|
|
mount "$@" && CHROOT_ACTIVE_MOUNTS=("$2" "${CHROOT_ACTIVE_MOUNTS[@]}")
|
|
}
|
|
|
|
chroot_mount_conditional() {
|
|
local cond=$1; shift
|
|
if eval "$cond"; then
|
|
chroot_mount "$@"
|
|
fi
|
|
}
|
|
|
|
chroot_api_efi_mount() {
|
|
CHROOT_ACTIVE_MOUNTS=()
|
|
[[ $(trap -p EXIT) ]] && die 'Error! Attempting to overwrite existing EXIT trap'
|
|
trap 'chroot_api_efi_umount' EXIT
|
|
|
|
chroot_mount_conditional "! mountpoint -q '$1'" "$1" "$1" --bind &&
|
|
chroot_mount proc "$1/proc" -t proc -o nosuid,noexec,nodev &&
|
|
chroot_mount sys "$1/sys" -t sysfs -o nosuid,noexec,nodev,ro &&
|
|
ignore_error chroot_mount_conditional "[[ -d '$1/sys/firmware/efi/efivars' ]]" \
|
|
efivarfs "$1/sys/firmware/efi/efivars" -t efivarfs -o nosuid,noexec,nodev &&
|
|
chroot_mount udev "$1/dev" -t devtmpfs -o mode=0755,nosuid &&
|
|
chroot_mount devpts "$1/dev/pts" -t devpts -o mode=0620,gid=5,nosuid,noexec &&
|
|
chroot_mount shm "$1/dev/shm" -t tmpfs -o mode=1777,nosuid,nodev &&
|
|
chroot_mount run "$1/run" -t tmpfs -o nosuid,nodev,mode=0755 &&
|
|
chroot_mount tmp "$1/tmp" -t tmpfs -o mode=1777,strictatime,nodev,nosuid
|
|
info "api_efi_mount: active mounts [%s]" "${CHROOT_ACTIVE_MOUNTS[@]}"
|
|
}
|
|
|
|
chroot_api_mount() {
|
|
CHROOT_ACTIVE_MOUNTS=()
|
|
[[ $(trap -p EXIT) ]] && die 'Error! Attempting to overwrite existing EXIT trap'
|
|
trap 'chroot_api_umount' EXIT
|
|
|
|
chroot_mount_conditional "! mountpoint -q '$1'" "$1" "$1" --bind &&
|
|
chroot_mount proc "$1/proc" -t proc -o nosuid,noexec,nodev &&
|
|
chroot_mount sys "$1/sys" -t sysfs -o nosuid,noexec,nodev,ro &&
|
|
chroot_mount udev "$1/dev" -t devtmpfs -o mode=0755,nosuid &&
|
|
chroot_mount devpts "$1/dev/pts" -t devpts -o mode=0620,gid=5,nosuid,noexec &&
|
|
chroot_mount shm "$1/dev/shm" -t tmpfs -o mode=1777,nosuid,nodev &&
|
|
chroot_mount run "$1/run" -t tmpfs -o nosuid,nodev,mode=0755 &&
|
|
chroot_mount tmp "$1/tmp" -t tmpfs -o mode=1777,strictatime,nodev,nosuid
|
|
info "api_mount: active mounts [%s]" "${CHROOT_ACTIVE_MOUNTS[@]}"
|
|
}
|
|
|
|
chroot_part_umount() {
|
|
info "part_umount: active mounts [%s]" "${CHROOT_ACTIVE_PART_MOUNTS}"
|
|
mount
|
|
info "part_umount: [%s]" "${CHROOT_ACTIVE_PART_MOUNTS[@]}"
|
|
umount "${CHROOT_ACTIVE_PART_MOUNTS[@]}"
|
|
unset CHROOT_ACTIVE_PART_MOUNTS
|
|
}
|
|
|
|
chroot_api_umount() {
|
|
info "api_umount: active mounts [%s]" "${CHROOT_ACTIVE_MOUNTS[@]}"
|
|
mount
|
|
info "api_umount: [%s]" "${CHROOT_ACTIVE_MOUNTS[@]}"
|
|
umount "${CHROOT_ACTIVE_MOUNTS[@]}"
|
|
unset CHROOT_ACTIVE_MOUNTS
|
|
}
|
|
|
|
chroot_api_efi_umount() {
|
|
info "api_efi_umount: active mounts [%s]" "${CHROOT_ACTIVE_MOUNTS[@]}"
|
|
mount
|
|
info "api_efi_umount: [%s]" "${CHROOT_ACTIVE_MOUNTS[@]}"
|
|
umount "${CHROOT_ACTIVE_MOUNTS[@]}"
|
|
unset CHROOT_ACTIVE_MOUNTS
|
|
}
|
|
|
|
trap_handler(){
|
|
chroot_api_umount
|
|
chroot_part_umount
|
|
}
|