You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
127 lines
4.3 KiB
127 lines
4.3 KiB
#!/bin/bash
|
|
# info: add system quota
|
|
# options: NONE
|
|
#
|
|
# example: v-add-sys-quota
|
|
#
|
|
# This function enables filesystem quota on /home partition
|
|
# Some kernels do require additional packages to be installed first
|
|
|
|
#----------------------------------------------------------#
|
|
# Variable & Function #
|
|
#----------------------------------------------------------#
|
|
|
|
# Includes
|
|
# shellcheck source=/usr/local/hestia/func/main.sh
|
|
source $HESTIA/func/main.sh
|
|
# load config file
|
|
source_conf "$HESTIA/conf/hestia.conf"
|
|
|
|
#----------------------------------------------------------#
|
|
# Verifications #
|
|
#----------------------------------------------------------#
|
|
|
|
# Checking quota package
|
|
quota=$(which --skip-alias --skip-functions quota 2> /dev/null)
|
|
if [ $? -ne 0 ]; then
|
|
dnf -y install quota > /dev/null 2>&1
|
|
check_result $? "quota package installation failed" "$E_UPDATE"
|
|
fi
|
|
|
|
# Perform verification if read-only mode is enabled
|
|
check_hestia_demo_mode
|
|
|
|
#----------------------------------------------------------#
|
|
# Action #
|
|
#----------------------------------------------------------#
|
|
|
|
# Adding group and user quota on /home partition
|
|
file_system=$(df -T /home | tail -n1 | awk '{print $2}')
|
|
mnt=$(df -P /home | awk '{print $6}' | tail -n1)
|
|
lnr=$(cat -n /etc/fstab | grep -v "#" | awk '{print $1,$3}' | grep "$mnt$" | cut -f 1 -d ' ')
|
|
opt=$(sed -n ${lnr}p /etc/fstab | awk '{print $4}')
|
|
if [ "$file_system" == "xfs" ]; then
|
|
fnd='usrquota\|grpquota'
|
|
if [ $(echo $opt | tr ',' '\n' | grep -x $fnd | wc -l) -ne 2 ]; then
|
|
old=$(echo $(echo $opt | tr ',' '\n' | grep -v 'usrquota\|grpquota') | tr ' ' ',')
|
|
new='usrquota,grpquota'
|
|
sed -i "$lnr s/$opt/$old,$new/" /etc/fstab
|
|
mount -o remount "$mnt"
|
|
systemctl daemon-reload
|
|
if [ "$mnt" == "/" ]; then
|
|
check_args=$(grubby --info=ALL | grep -oP "rootflags=\S*")
|
|
if [ -n "$check_args" ]; then
|
|
RED='\033[0;31m'
|
|
NC='\033[0m'
|
|
YELLOW='\033[1;33m'
|
|
printf "${RED}!!!!!!!Detected rootflags in grub!!!!!!!${NC}\n"
|
|
echo "${YELLOW}Add manualy 'uquota,pquota' and reboot the system${NC}\n"
|
|
else
|
|
grubby --args="rootflags=uquota,pquota" --update-kernel=ALL
|
|
fi
|
|
fi
|
|
fi
|
|
elif [ "$file_system" == "ext4" ]; then
|
|
fnd='usrquota\|grpquota\|usrjquota=aquota.user\|grpjquota=aquota.group\|jqfmt=vfsv0'
|
|
if [ $(echo $opt | tr ',' '\n' | grep -x $fnd | wc -l) -ne 5 ]; then
|
|
old=$(echo $(echo $opt | tr ',' '\n' | grep -v 'usrquota\|grpquota\|usrjquota=\|grpjquota=\|jqfmt=') | tr ' ' ',')
|
|
new='usrquota,grpquota,usrjquota=aquota.user,grpjquota=aquota.group,jqfmt=vfsv0'
|
|
sed -i "$lnr s/$opt/$old,$new/" /etc/fstab
|
|
mount -o remount "$mnt"
|
|
systemctl daemon-reload
|
|
fi
|
|
else
|
|
if [ -n "$(grep DISK_QUOTA $HESTIA/conf/hestia.conf)" ]; then
|
|
sed -i "s/DISK_QUOTA=.*/DISK_QUOTA='no'/g" $HESTIA/conf/hestia.conf
|
|
fi
|
|
$BIN/v-log-action "system" "Info" "Plugins" "System Quota not supported."
|
|
log_history "system quota not supported"
|
|
log_event "$E_DISK" "$ARGUMENTS"
|
|
exit
|
|
fi
|
|
|
|
if [ "$file_system" == "ext4" ]; then
|
|
# Adding v2 group and user quota index
|
|
if [ ! -e "$mnt/aquota.user" ] || [ ! -e "$mnt/aquota.group" ]; then
|
|
quotacheck -avcugm > /dev/null 2>&1
|
|
fi
|
|
|
|
# Adding quotacheck on reboot
|
|
touch /forcequotacheck
|
|
|
|
# Adding cron job
|
|
echo '#!/bin/bash' > /etc/cron.daily/quotacheck
|
|
echo 'touch /forcequotacheck' >> /etc/cron.daily/quotacheck
|
|
chmod a+x /etc/cron.daily/quotacheck
|
|
|
|
# Enabling group and user quota
|
|
if [ -n "$(quotaon -pa | grep " $mnt " | grep 'user\|group' | grep 'is off')" ]; then
|
|
quotaon -v $mnt
|
|
check_result $? "quota can't be enabled in $mnt" "$E_DISK"
|
|
fi
|
|
|
|
fi
|
|
|
|
# Updating hestia.conf value
|
|
if [ -z "$(grep DISK_QUOTA $HESTIA/conf/hestia.conf)" ]; then
|
|
echo "DISK_QUOTA='yes'" >> $HESTIA/conf/hestia.conf
|
|
else
|
|
sed -i "s/DISK_QUOTA=.*/DISK_QUOTA='yes'/g" $HESTIA/conf/hestia.conf
|
|
fi
|
|
|
|
# Rebuilding user quota
|
|
for user in $($BIN/v-list-sys-users plain); do
|
|
$BIN/v-update-user-quota "$user"
|
|
done
|
|
|
|
#----------------------------------------------------------#
|
|
# Hestia #
|
|
#----------------------------------------------------------#
|
|
|
|
# Logging
|
|
$BIN/v-log-action "system" "Info" "Plugins" "System Quota enforcement enabled."
|
|
log_history "system quota enforcement enabled"
|
|
log_event "$OK" "$ARGUMENTS"
|
|
|
|
exit
|