immortalwrt/package/lienol/luci-app-webrestriction/root/etc/init.d/webrestriction

86 lines
2.3 KiB
Plaintext
Raw Normal View History

2019-11-02 10:02:03 +08:00
#!/bin/sh /etc/rc.common
#
# Copyright (C) 2015 OpenWrt-dist
# Copyright (C) 2016 fw867 <ffkykzs@gmail.com>
#
# This is free software, licensed under the GNU General Public License v3.
# See /LICENSE for more information.
#
START=99
CONFIG=webrestriction
limit_type=$(uci -q get webrestriction.@basic[0].limit_type)
uci_get_by_type() {
local index=0
if [ -n $4 ]; then
index=$4
fi
local ret=$(uci get $CONFIG.@$1[$index].$2 2>/dev/null)
echo ${ret:=$3}
}
is_true() {
case $1 in
1|on|true|yes|enabled) echo 0;;
*) echo 1;;
esac
}
load_config() {
ENABLED=$(uci_get_by_type basic enable)
return $(is_true $ENABLED)
}
add_rule(){
action=$1
for i in $(seq 0 100)
do
enable=$(uci_get_by_type macbind enable '' $i)
macaddr=$(uci_get_by_type macbind macaddr '' $i)
if [ -z $enable ] || [ -z $macaddr ]; then
break
fi
if [ "$enable" == "1" ]; then
iptables -t filter -A WEB_RESTRICTION -m mac --mac-source $macaddr -j $action
[ "$limit_type" == "blacklist" ] && iptables -t nat -A WEB_RESTRICTION -m mac --mac-source $macaddr -j RETURN
#unset "$macaddr"
fi
done
}
start(){
! load_config && exit 0
[ "`iptables -L FORWARD|grep -c WEB_RESTRICTION`" -gt 0 ] && exit 0;
iptables -P FORWARD DROP
iptables -t filter -N WEB_RESTRICTION
if [ "$limit_type" == "blacklist" ]; then
iptables -t nat -N WEB_RESTRICTION
add_rule DROP
else
add_rule ACCEPT
iptables -t filter -A WEB_RESTRICTION -j DROP
fi
#获取FORWARD ACCEPT规则行号
FA_INDEX=`iptables -t filter -L FORWARD|tail -n +3|sed -n -e '/^ACCEPT/='`
if [ -n "$FA_INDEX" ]; then
let FA_INDEX+=1
fi
#确保添加到FORWARD ACCEPT规则之后
iptables -t filter -I FORWARD $FA_INDEX -m comment --comment "Rule For Control" -j WEB_RESTRICTION
[ "$limit_type" == "blacklist" ] && iptables -t nat -I PREROUTING 1 -m comment --comment "Rule For Control" -j WEB_RESTRICTION
}
stop(){
[ "`iptables -t filter -L | grep -c WEB_RESTRICTION`" -gt 0 ] && {
iptables -t filter -D FORWARD -m comment --comment "Rule For Control" -j WEB_RESTRICTION
iptables -t nat -D PREROUTING -m comment --comment "Rule For Control" -j WEB_RESTRICTION
iptables -t filter -F WEB_RESTRICTION
iptables -t filter -X WEB_RESTRICTION
iptables -t nat -F WEB_RESTRICTION
iptables -t nat -X WEB_RESTRICTION
}
}