wcp/bin/v-add-sys-quota at c52ff93cc898c15576efba62256578ebb68b24ab · ghzhost/wcp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#!/bin/bash
# info: add system quota
# options: NONE
# labels:
#
# example: v-add-sys-quota
#
# The script enables filesystem quota on /home partition
# Some kernels do require additional packages to be installed first


#----------------------------------------------------------#
#                 Variable & Function                      #
#----------------------------------------------------------#

# Includes
source $HESTIA/func/main.sh
source $HESTIA/conf/hestia.conf


#----------------------------------------------------------#
#                    Verifications                         #
#----------------------------------------------------------#

# Ensure that quota kernel modules are installed
kernel_module_check=$(find /lib/modules/`uname -r` -type f -name '*quota_v*.ko*' | egrep '.*' && [ $? -eq 0 ])
if [ -z "$kernel_module_check" ]; then
    # Install kernel modules for quota support.
    # Requires reboot to activate updated kernel.
    echo "Installing required kernel modules for quota support..."
    reboot_req="Y"
    apt-get -qq install linux-image-extra-virtual -y
    check_result $? "kernel module installation failed" $E_UPDATE
fi

# Checking quota package
quota=$(which --skip-alias --skip-functions quota 2>/dev/null)
if [ $? -ne 0 ]; then
    if [ -d "/etc/sysconfig" ]; then
        yum -y install quota >/dev/null 2>&1
        check_result $? "quota package installation failed" $E_UPDATE
    else
        export DEBIAN_FRONTEND=noninteractive
        apt-get -y install quota >/dev/null 2>&1
        check_result $? "quota package installation failed" $E_UPDATE
    fi
fi


#----------------------------------------------------------#
#                       Action                             #
#----------------------------------------------------------#

# Adding group and user quota on /home partition
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}')
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
fi

# 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 [ ! -z "$(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

# 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 $($HESTIA/bin/v-list-sys-users plain); do
    $BIN/v-update-user-quota $user
done


#----------------------------------------------------------#
#                       Hestia                             #
#----------------------------------------------------------#

# Logging
log_history "system quota enforcement enabled"
log_event "$OK" "$ARGUMENTS"

exit