[install.sh]

Fixed the following issues :

- The password prompt went all wonky if password was wrong
- The password prompt did not use system password cache and kept asking for password again on reruns of script
- Installation failed if the script was initially ran by "sudo" due to inability to preserve ownership of files during cp commands
- Security issue where the password prompt could be interrupted (Ctrl+C) and rerun without sleeping for few seconds first just like normal sudo failure in terminal
- The sudo command would no longer accept password from ANY commands system-wide if wrong password was given to the install script enough times!! (the only fix would be restarting the system)
This commit is contained in:
Nurverek 2020-11-10 21:48:52 +03:30
parent f4b1a85ab0
commit aa8108f74b

View File

@ -110,7 +110,6 @@ install() {
# Check for root access and proceed if it is present # Check for root access and proceed if it is present
if [ "$UID" -eq "$ROOT_UID" ]; then if [ "$UID" -eq "$ROOT_UID" ]; then
clear clear
if [[ "${custom_background}" == "custom-background" ]]; then if [[ "${custom_background}" == "custom-background" ]]; then
@ -133,28 +132,29 @@ install() {
# Copy theme # Copy theme
prompt -i "\n Installing ${name} ${icon} ${screen} theme..." prompt -i "\n Installing ${name} ${icon} ${screen} theme..."
cp -a "${REO_DIR}/common/"{*.png,*.pf2} "${THEME_DIR}/${name}" # Don't preserve ownership because the owner will be root, and that causes the script to crash if it is ran from terminal by sudo
cp -a "${REO_DIR}/config/theme-${screen}.txt" "${THEME_DIR}/${name}/theme.txt" cp -a --no-preserve=ownership "${REO_DIR}/common/"{*.png,*.pf2} "${THEME_DIR}/${name}"
cp -a --no-preserve=ownership "${REO_DIR}/config/theme-${screen}.txt" "${THEME_DIR}/${name}/theme.txt"
if [[ ${custom_background} == "background.jpg" ]] || [[ ${custom_background} == "custom-background.jpg" ]]; then if [[ ${custom_background} == "background.jpg" ]] || [[ ${custom_background} == "custom-background.jpg" ]]; then
if [[ -f "$custom_background" ]]; then if [[ -f "$custom_background" ]]; then
prompt -i "\n Using ${custom_background} as background image..." prompt -i "\n Using ${custom_background} as background image..."
cp -a "${REO_DIR}/${custom_background}" "${THEME_DIR}/${name}/background.jpg" cp -a --no-preserve=ownership "${REO_DIR}/${custom_background}" "${THEME_DIR}/${name}/background.jpg"
convert -auto-orient "${THEME_DIR}/${name}/background.jpg" "${THEME_DIR}/${name}/background.jpg" convert -auto-orient "${THEME_DIR}/${name}/background.jpg" "${THEME_DIR}/${name}/background.jpg"
else else
prompt -e "$custom_background couldn't be found, exiting" prompt -e "$custom_background couldn't be found, exiting"
exit 0 exit 0
fi fi
else else
cp -a "${REO_DIR}/backgrounds/${screen}/background-${theme}.jpg" "${THEME_DIR}/${name}/background.jpg" cp -a --no-preserve=ownership "${REO_DIR}/backgrounds/${screen}/background-${theme}.jpg" "${THEME_DIR}/${name}/background.jpg"
fi fi
if [[ ${screen} == '1080p_21:9' ]]; then if [[ ${screen} == '1080p_21:9' ]]; then
cp -a "${REO_DIR}/assets/assets-${icon}/icons-1080p" "${THEME_DIR}/${name}/icons" cp -a --no-preserve=ownership "${REO_DIR}/assets/assets-${icon}/icons-1080p" "${THEME_DIR}/${name}/icons"
cp -a "${REO_DIR}/assets/assets-${icon}/select-1080p/"*.png "${THEME_DIR}/${name}" cp -a --no-preserve=ownership "${REO_DIR}/assets/assets-${icon}/select-1080p/"*.png "${THEME_DIR}/${name}"
else else
cp -a "${REO_DIR}/assets/assets-${icon}/icons-${screen}" "${THEME_DIR}/${name}/icons" cp -a --no-preserve=ownership "${REO_DIR}/assets/assets-${icon}/icons-${screen}" "${THEME_DIR}/${name}/icons"
cp -a "${REO_DIR}/assets/assets-${icon}/select-${screen}/"*.png "${THEME_DIR}/${name}" cp -a --no-preserve=ownership "${REO_DIR}/assets/assets-${icon}/select-${screen}/"*.png "${THEME_DIR}/${name}"
fi fi
# Set theme # Set theme
@ -206,41 +206,71 @@ install() {
updating_grub updating_grub
prompt -w "\n * At the next restart of your computer you will see your new Grub theme: '$theme' " prompt -w "\n * At the next restart of your computer you will see your new Grub theme: '$theme' "
else else
# Error message #Check if password is cached (if cache timestamp not expired yet)
prompt -e "\n [ Error! ] -> Run me as root! " sudo -n true 2> /dev/null && echo
if [[ $? == 0 ]]; then
#No need to ask for password
sudo "$0" --${theme} --${icon} --${screen}
else
#Ask for password
# Persistent execution of the script as root
if [[ -n ${tui_root_login} ]] ; then if [[ -n ${tui_root_login} ]] ; then
if [[ -n "${theme}" && -n "${screen}" ]]; then if [[ -n "${theme}" && -n "${screen}" ]]; then
sudo -S <<< ${tui_root_login} $0 --${theme} --${icon} --${screen} sudo -S $0 --${theme} --${icon} --${screen} <<< ${tui_root_login}
fi fi
else else
prompt -e "\n [ Error! ] -> Run me as root! "
read -p " [ Trusted ] Specify the root password : " -t ${MAX_DELAY} -s read -p " [ Trusted ] Specify the root password : " -t ${MAX_DELAY} -s
[[ -n "$REPLY" ]] && {
if [[ -n "${theme}" && -n "${screen}" ]]; then
sudo -S <<< $REPLY "$0" --${theme} --${icon} --${screen}
fi
} || {
operation_canceled
}
fi
sudo -S echo <<< $REPLY 2> /dev/null && echo
if [[ $? == 0 ]]; then
#Correct password, use with sudo's stdin
sudo -S "$0" --${theme} --${icon} --${screen} <<< ${REPLY}
else
#block for 3 seconds before allowing another attempt
sleep 3
prompt -e "\n [ Error! ] -> Incorrect password!\n"
exit 1
fi
fi
fi
fi fi
} }
run_dialog() { run_dialog() {
if [[ -x /usr/bin/dialog ]]; then if [[ -x /usr/bin/dialog ]]; then
if [[ "$UID" -ne "$ROOT_UID" ]]; then
#Check if password is cached (if cache timestamp not expired yet)
sudo -n true 2> /dev/null && echo
if [[ $? == 0 ]]; then
#No need to ask for password
sudo $0
else
#Ask for password
tui_root_login=$(dialog --backtitle ${Project_Name} \ tui_root_login=$(dialog --backtitle ${Project_Name} \
--title "ROOT LOGIN" \ --title "ROOT LOGIN" \
--insecure \ --insecure \
--passwordbox "require root permission" 8 50 \ --passwordbox "require root permission" 8 50 \
--output-fd 1 ) --output-fd 1 )
[[ -z ${tui_root_login} ]] && exit ${UID}
sudo -S <<< $tui_root_login $0 sudo -S echo <<< $tui_root_login 2> /dev/null && echo
test $? -eq 0 || {
prompt -e "\n [ Error! ] -> wrong passwords" if [[ $? == 0 ]]; then
#Correct password, use with sudo's stdin
sudo -S "$0" <<< $tui_root_login
else
#block for 3 seconds before allowing another attempt
sleep 3
clear
prompt -e "\n [ Error! ] -> Incorrect password!\n"
exit 1 exit 1
} fi
fi
fi
tui=$(dialog --backtitle ${Project_Name} \ tui=$(dialog --backtitle ${Project_Name} \
--radiolist "Choose your Grub theme : " 15 40 5 \ --radiolist "Choose your Grub theme : " 15 40 5 \
@ -357,18 +387,30 @@ remove() {
updating_grub updating_grub
else else
# Error message #Check if password is cached (if cache timestamp not expired yet)
prompt -e "\n [ Error! ] -> Run me as root " sudo -n true 2> /dev/null && echo
# Persistent execution of the script as root if [[ $? == 0 ]]; then
read -p "[ trusted ] specify the root password : " -t${MAX_DELAY} -s #No need to ask for password
[[ -n "$REPLY" ]] && { sudo "$0" "${PROG_ARGS[@]}"
if [[ -n "${theme}" ]]; then else
sudo -S <<< $REPLY "$0" "${PROG_ARGS[@]}" #Ask for password
prompt -e "\n [ Error! ] -> Run me as root! "
read -p " [ Trusted ] Specify the root password : " -t ${MAX_DELAY} -s
sudo -S echo <<< $REPLY 2> /dev/null && echo
if [[ $? == 0 ]]; then
#Correct password, use with sudo's stdin
sudo -S "$0" "${PROG_ARGS[@]}" <<< $REPLY
else
#block for 3 seconds before allowing another attempt
sleep 3
clear
prompt -e "\n [ Error! ] -> Incorrect password!\n"
exit 1
fi
fi fi
} || {
operation_canceled
}
fi fi
} }
@ -378,16 +420,29 @@ if [[ $# -lt 1 ]] && [[ $UID -ne $ROOT_UID ]] && [[ -x /usr/bin/dialog ]] ; then
fi fi
if [[ $# -lt 1 ]] && [[ $UID -ne $ROOT_UID ]] && [[ ! -x /usr/bin/dialog ]] ; then if [[ $# -lt 1 ]] && [[ $UID -ne $ROOT_UID ]] && [[ ! -x /usr/bin/dialog ]] ; then
# Error message #Check if password is cached (if cache timestamp not expired yet)
prompt -e "\n [ Error! ] -> Run me as root! " sudo -n true 2> /dev/null && echo
# persisted execution of the script as root if [[ $? == 0 ]]; then
read -p "[ Trusted ] Specify the root password : " -t${MAX_DELAY} -s #No need to ask for password
[[ -n "$REPLY" ]]&& { exec sudo $0
exec sudo -S <<< $REPLY $0 else
}|| { #Ask for password
operation_canceled prompt -e "\n [ Error! ] -> Run me as root! "
} read -p " [ Trusted ] Specify the root password : " -t ${MAX_DELAY} -s
sudo -S echo <<< $REPLY 2> /dev/null && echo
if [[ $? == 0 ]]; then
#Correct password, use with sudo's stdin
sudo $0 <<< $REPLY
else
#block for 3 seconds before allowing another attempt
sleep 3
prompt -e "\n [ Error! ] -> Incorrect password!\n"
exit 1
fi
fi
fi fi
while [[ $# -ge 1 ]]; do while [[ $# -ge 1 ]]; do