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.
hestiacp/bin/v-update-firewall

224 lines
6.4 KiB

#!/bin/bash
# info: update system firewall rules
# options: no opts
#
# example: v-update-firewall
#
# This function updates iptables rules
#----------------------------------------------------------#
# Variables & Functions #
#----------------------------------------------------------#
# Defining absolute path for iptables and modprobe
iptables="/sbin/iptables"
modprobe="/sbin/modprobe"
sysctl="/sbin/sysctl"
# Includes
source /etc/profile.d/hestia.sh
# shellcheck source=/etc/hestiacp/hestia.conf
source /etc/hestiacp/hestia.conf
# shellcheck source=/usr/local/hestia/func/main.sh
source $HESTIA/func/main.sh
# shellcheck source=/usr/local/hestia/func/firewall.sh
source $HESTIA/func/firewall.sh
# load config file
source_conf "$HESTIA/conf/hestia.conf"
#----------------------------------------------------------#
# Verifications #
#----------------------------------------------------------#
is_system_enabled "$FIREWALL_SYSTEM" 'FIREWALL_SYSTEM'
#----------------------------------------------------------#
# Action #
#----------------------------------------------------------#
# Self heal iptables links
heal_iptables_links
# Checking local IPv4 rules
rules="$HESTIA/data/firewall/rules.conf"
if [ ! -e "$rules" ]; then
exit
fi
if [ -f /etc/redhat-release ]; then
conntrack='yes'
else
# Checking conntrack module avaiabilty
$modprobe nf_conntrack > /dev/null 2>&1
if [ $? -ne 0 ]; then
$sysctl net.netfilter.nf_conntrack_max > /dev/null 2>&1
if [ $? -ne 0 ]; then
conntrack='no'
fi
fi
fi
if [ -f /etc/redhat-release ]; then
conntrack_ftp='yes'
else
$modprobe nf_conntrack_ftp > /dev/null 2>&1
if [ $? -ne 0 ]; then
conntrack_ftp='no'
fi
fi
# Checking custom OpenSSH port
sshport=$(grep '^Port ' /etc/ssh/sshd_config | head -1 | cut -d ' ' -f 2)
if [[ "$sshport" =~ ^[0-9]+$ ]] && [ "$sshport" -ne "22" ]; then
sed -i "s/PORT='22'/PORT=\'$sshport\'/" $rules
fi
# Load ipset lists before adding Hestia iptables rules
$BIN/v-update-firewall-ipset load
# Creating temporary file
tmp="$(mktemp)"
# Flushing INPUT chain
echo "$iptables -P INPUT ACCEPT" >> $tmp
echo "$iptables -F INPUT" >> $tmp
# Enabling stateful support
if [ "$conntrack" != 'no' ] || grep --quiet container=lxc /proc/1/environ; then
str="$iptables -A INPUT -m state"
str="$str --state ESTABLISHED,RELATED -j ACCEPT"
echo "$str" >> $tmp
fi
ips="$(ls $HESTIA/data/ips)"
# Handling local traffic
for ip in $ips; do
echo "$iptables -A INPUT -s $ip -j ACCEPT" >> $tmp
done
echo "$iptables -A INPUT -s 127.0.0.1 -j ACCEPT" >> $tmp
# Pasring iptables rules
IFS=$'\n'
for line in $(sort -r -n -k 2 -t \' $rules); do
parse_object_kv_list "$line"
if [ "$SUSPENDED" = 'no' ]; then
proto="-p $PROTOCOL"
port="--dport $PORT"
state=""
action="-j $ACTION"
if [[ "$IP" =~ ^ipset: ]]; then
ipset_name="${IP#ipset:}"
$(v-list-firewall-ipset plain | grep "^$ipset_name\s" > /dev/null) || log_event $E_NOTEXIST "IPset IP list ($ipset_name) not found"
ip="-m set --match-set '${ipset_name}' src"
else
ip="-s $IP"
fi
# Adding multiport module
if [[ "$PORT" =~ ,|-|: ]]; then
port="-m multiport --dports ${PORT//-/:}"
fi
# Accepting all dst ports
if [[ "$PORT" = "0" ]] || [ "$PROTOCOL" = 'ICMP' ]; then
port=""
fi
# Checking FTP for contrack module
if [ "$TYPE" = "FTP" ] || [ "$PORT" = '21' ]; then
if [ "$conntrack_ftp" != 'no' ]; then
state="-m conntrack --ctstate NEW"
else
port="-m multiport --dports 20,21,12000:12100"
fi
ftp="yes"
fi
# Adding firewall rule
echo "$iptables -A INPUT $proto $port $ip $state $action" >> $tmp
fi
done
# Switching chain policy to DROP
echo "$iptables -P INPUT DROP" >> $tmp
# Adding hestia chain
echo "$iptables -N hestia" >> $tmp
# Applying rules
bash $tmp 2> /dev/null
# Deleting temporary file
rm -f $tmp
# Checking custom trigger
if [ -x "$HESTIA/data/firewall/custom.sh" ]; then
bash $HESTIA/data/firewall/custom.sh
fi
# Checking fail2ban support
if [ -n "$FIREWALL_EXTENSION" ]; then
for chain in $(cat $HESTIA/data/firewall/chains.conf 2> /dev/null); do
parse_object_kv_list "$chain"
if [[ "$PORT" =~ ,|-|: ]]; then
port="-m multiport --dports $PORT"
else
port="--dport $PORT"
fi
echo "$iptables -N fail2ban-$CHAIN" >> $tmp
echo "$iptables -F fail2ban-$CHAIN" >> $tmp
echo "$iptables -I fail2ban-$CHAIN -s 0.0.0.0/0 -j RETURN" >> $tmp
echo "$iptables -I INPUT -p $PROTOCOL $port -j fail2ban-$CHAIN" >> $tmp
done
bash $tmp 2> /dev/null
rm -f $tmp
for ban in $(cat $HESTIA/data/firewall/banlist.conf 2> /dev/null); do
parse_object_kv_list "$ban"
echo -n "$iptables -I fail2ban-$CHAIN 1 -s $IP" >> $tmp
echo " -j REJECT --reject-with icmp-port-unreachable" >> $tmp
done
bash $tmp 2> /dev/null
rm -f $tmp
fi
# Clean up and saving rules to the master iptables file
if [ -d "/etc/sysconfig" ] && [ ! -e /etc/redhat-release ]; then
/sbin/iptables-save | sed -e 's/[[0-9]\+:[0-9]\+]/[0:0]/g' -e '/^-A fail2ban-[A-Z]\+ -s .\+$/d' > /etc/sysconfig/iptables
else
/sbin/iptables-save | sed -e 's/[[0-9]\+:[0-9]\+]/[0:0]/g' -e '/^-A fail2ban-[A-Z]\+ -s .\+$/d' > /etc/iptables.rules
iptablesversion="$(iptables --version | head -1 | awk '{print $2}' | cut -f -2 -d .)"
sd_unit="/lib/systemd/system/hestia-iptables.service"
if [ ! -e "$sd_unit" ]; then
echo "[Unit]" >> $sd_unit
echo "Description=Loading Hestia firewall rules" >> $sd_unit
echo "DefaultDependencies=no" >> $sd_unit
echo "Wants=network-pre.target local-fs.target" >> $sd_unit
echo "Before=network-pre.target" >> $sd_unit
echo "After=local-fs.target" >> $sd_unit
echo "" >> $sd_unit
echo "[Service]" >> $sd_unit
echo "Type=oneshot" >> $sd_unit
echo "RemainAfterExit=yes" >> $sd_unit
echo "ExecStartPre=-${HESTIA}/bin/v-update-firewall-ipset load" >> $sd_unit
if [ "$iptablesversion" = "v1.6" ]; then
echo "ExecStart=/sbin/iptables-restore /etc/iptables.rules" >> $sd_unit
else
echo "ExecStart=/sbin/iptables-restore --wait=10 /etc/iptables.rules" >> $sd_unit
fi
echo "" >> $sd_unit
echo "[Install]" >> $sd_unit
echo "WantedBy=multi-user.target" >> $sd_unit
systemctl -q daemon-reload
fi
systemctl -q is-enabled hestia-iptables 2> /dev/null || systemctl -q enable hestia-iptables
fi
#----------------------------------------------------------#
# Hestia #
#----------------------------------------------------------#
exit