109 lines
2.7 KiB
Bash
109 lines
2.7 KiB
Bash
#!/bin/bash
|
|
|
|
# matter-chroot | Matterlinux Chroot Script
|
|
# Copyright (C) 2023 Matterlinux
|
|
|
|
# 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 <https://www.gnu.org/licenses/>.
|
|
|
|
############################
|
|
## logging functions/vars ##
|
|
############################
|
|
RED="\e[31m"
|
|
BOLD="\e[1m"
|
|
RESET="\e[0m"
|
|
|
|
error() {
|
|
echo -e "$BOLD$RED>>>$RESET$BOLD $1$RESET"
|
|
exit 1
|
|
}
|
|
|
|
####################
|
|
## util functions ##
|
|
####################
|
|
linkresolv(){
|
|
if [ ! -f "$TARGET/etc/resolv.conf" ]; then
|
|
ln -sf /run/systemd/resolve/resolv.conf "$TARGET/etc/resolv.conf"
|
|
fi
|
|
}
|
|
|
|
chrt() {
|
|
mount -t proc proc "$TARGET/proc"
|
|
mount -t sysfs sysfs "$TARGET/sys"
|
|
if [[ -d "$TARGET/sys/firmware/efi/efivars" ]]; then
|
|
mount -t efivarfs efivarfs "$TARGET/sys/firmware/efi/efivars"
|
|
fi
|
|
|
|
mount -o bind /dev "$TARGET/dev"
|
|
mount -t devpts none "$TARGET/dev/pts"
|
|
mount --bind /run "$TARGET/run"
|
|
linkresolv
|
|
|
|
if [ -h $TARGET/dev/shm ]; then
|
|
mkdir -p $TARGET/$(readlink $TARGET/dev/shm)
|
|
else
|
|
mount -t tmpfs -o nosuid,nodev tmpfs "$TARGET/dev/shm"
|
|
fi
|
|
|
|
|
|
local prompt='\['$BOLD'\['$RED'(chroot)\['$RESET'\['$BOLD' \u@\h:\w#\['$RESET' '
|
|
chroot "$TARGET" /usr/bin/env -i \
|
|
HOME=/root \
|
|
TERM="$TERM" \
|
|
PS1="$prompt" \
|
|
PATH=/usr/bin:/usr/sbin \
|
|
"$@"
|
|
RET_CODE=$?
|
|
|
|
# kill procs that may prevent umount
|
|
killall -9 dirmngr 2> /dev/null
|
|
killall -9 gpg-agent 2> /dev/null
|
|
|
|
umount "$TARGET/proc"
|
|
mountpoint -q "$TARGET/sys/firmware/efi/efivars" && umount "$TARGET/sys/firmware/efi/efivars"
|
|
umount "$TARGET/sys"
|
|
|
|
umount "$TARGET/dev/pts"
|
|
mountpoint -q "$TARGET/dev/shm" && umount "$TARGET/dev/shm"
|
|
umount "$TARGET/dev"
|
|
umount "$TARGET/run"
|
|
}
|
|
|
|
#################
|
|
## main script ##
|
|
#################
|
|
if [ "$EUID" -ne 0 ]; then
|
|
error "Cannot chroot without root"
|
|
fi
|
|
|
|
if [ $# -eq 0 ]; then
|
|
error "Please specify a directory"
|
|
fi
|
|
|
|
TARGET=$(realpath $1)
|
|
if [ -f $TARGET ]; then
|
|
error "$TARGET is a file"
|
|
fi
|
|
|
|
if [ ! -d $TARGET ]; then
|
|
error "$TARGET does not exist"
|
|
fi
|
|
|
|
if [ $# -gt 1 ]; then
|
|
chrt ${@:2}
|
|
exit $RET_CODE
|
|
fi
|
|
|
|
chrt bash --login
|
|
exit $?
|