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.
362 lines
10 KiB
362 lines
10 KiB
#!/bin/sh
|
|
|
|
mwan3_get_iface_id()
|
|
{
|
|
let iface_count++
|
|
[ "$1" == "$INTERFACE" ] && iface_id=$iface_count
|
|
}
|
|
|
|
mwan3_set_general_iptables()
|
|
{
|
|
if ! $IPT -S mwan3_ifaces &> /dev/null; then
|
|
$IPT -N mwan3_ifaces
|
|
fi
|
|
|
|
if ! $IPT -S mwan3_rules &> /dev/null; then
|
|
$IPT -N mwan3_rules
|
|
fi
|
|
|
|
if ! $IPT -S mwan3_connected &> /dev/null; then
|
|
$IPT -N mwan3_connected
|
|
fi
|
|
|
|
if ! $IPT -S mwan3_hook &> /dev/null; then
|
|
$IPT -N mwan3_hook
|
|
$IPT -A mwan3_hook -j CONNMARK --restore-mark --nfmask 0xff00 --ctmask 0xff00
|
|
$IPT -A mwan3_hook -m mark --mark 0x0/0xff00 -j mwan3_ifaces
|
|
$IPT -A mwan3_hook -m mark --mark 0x0/0xff00 -j mwan3_connected
|
|
$IPT -A mwan3_hook -m mark --mark 0x0/0xff00 -j mwan3_rules
|
|
$IPT -A mwan3_hook -j CONNMARK --save-mark --nfmask 0xff00 --ctmask 0xff00
|
|
$IPT -A mwan3_hook -m mark ! --mark 0xff00/0xff00 -j mwan3_connected
|
|
fi
|
|
|
|
if ! $IPT -S mwan3_output_hook &> /dev/null; then
|
|
$IPT -N mwan3_output_hook
|
|
fi
|
|
|
|
if ! $IPT -S PREROUTING | grep mwan3_hook &> /dev/null; then
|
|
$IPT -A PREROUTING -j mwan3_hook
|
|
fi
|
|
|
|
if ! $IPT -S OUTPUT | grep mwan3_hook &> /dev/null; then
|
|
$IPT -A OUTPUT -j mwan3_hook
|
|
fi
|
|
|
|
if ! $IPT -S OUTPUT | grep mwan3_output_hook &> /dev/null; then
|
|
$IPT -A OUTPUT -j mwan3_output_hook
|
|
fi
|
|
|
|
$IPT -F mwan3_rules
|
|
}
|
|
|
|
mwan3_set_general_rules()
|
|
{
|
|
if [ -z "$($IP rule list | awk '$1 == "2253:"')" ]; then
|
|
$IP rule add pref 2253 fwmark 0xfd00/0xff00 blackhole
|
|
fi
|
|
|
|
if [ -z "$($IP rule list | awk '$1 == "2254:"')" ]; then
|
|
$IP rule add pref 2254 fwmark 0xfe00/0xff00 unreachable
|
|
fi
|
|
}
|
|
|
|
mwan3_set_connected_iptables()
|
|
{
|
|
local connected_networks
|
|
|
|
if $IPT -S mwan3_connected &> /dev/null; then
|
|
$IPT -F mwan3_connected
|
|
|
|
for connected_networks in $($IP route | awk '{print $1}' | egrep '[0-9]{1,3}(\.[0-9]{1,3}){3}'); do
|
|
$IPT -A mwan3_connected -d $connected_networks -j MARK --set-xmark 0xff00/0xff00
|
|
done
|
|
|
|
for connected_networks in $($IP route list table 0 | awk '{print $2}' | egrep '[0-9]{1,3}(\.[0-9]{1,3}){3}'); do
|
|
$IPT -A mwan3_connected -d $connected_networks -j MARK --set-xmark 0xff00/0xff00
|
|
done
|
|
|
|
$IPT -I mwan3_connected -d 224.0.0.0/3 -j MARK --set-xmark 0xff00/0xff00
|
|
$IPT -I mwan3_connected -d 127.0.0.0/8 -j MARK --set-xmark 0xff00/0xff00
|
|
fi
|
|
}
|
|
|
|
mwan3_set_iface_iptables()
|
|
{
|
|
local local_net local_nets
|
|
|
|
if ! $IPT -S mwan3_iface_$INTERFACE &> /dev/null; then
|
|
$IPT -N mwan3_iface_$INTERFACE
|
|
fi
|
|
|
|
$IPT -F mwan3_iface_$INTERFACE
|
|
$IPT -D mwan3_ifaces -m mark --mark 0x0/0xff00 -j mwan3_iface_$INTERFACE &> /dev/null
|
|
|
|
if [ $ACTION == "ifup" ]; then
|
|
local_nets=$($IP route list dev $DEVICE scope link | awk '{print $1}' | egrep '[0-9]{1,3}(\.[0-9]{1,3}){3}')
|
|
|
|
if [ -n "$local_nets" ]; then
|
|
for local_net in $local_nets ; do
|
|
if [ $ACTION == "ifup" ]; then
|
|
$IPT -I mwan3_iface_$INTERFACE -i $DEVICE -s $local_net -m mark --mark 0x0/0xff00 -m comment --comment "default" -j MARK --set-xmark 0xff00/0xff00
|
|
fi
|
|
done
|
|
fi
|
|
|
|
$IPT -A mwan3_iface_$INTERFACE -i $DEVICE -m mark --mark 0x0/0xff00 -m comment --comment "$INTERFACE" -j MARK --set-xmark $(($iface_id*256))/0xff00
|
|
$IPT -A mwan3_ifaces -m mark --mark 0x0/0xff00 -j mwan3_iface_$INTERFACE
|
|
fi
|
|
|
|
if [ $ACTION == "ifdown" ]; then
|
|
$IPT -X mwan3_iface_$INTERFACE
|
|
fi
|
|
}
|
|
|
|
mwan3_set_iface_route()
|
|
{
|
|
$IP route flush table $iface_id
|
|
[ $ACTION == "ifup" ] && $IP route add table $iface_id default $route_args
|
|
}
|
|
|
|
mwan3_set_iface_rules()
|
|
{
|
|
while [ -n "$($IP rule list | awk '$1 == "'$(($iface_id+1000)):'"')" ]; do
|
|
$IP rule del pref $(($iface_id+1000))
|
|
done
|
|
|
|
while [ -n "$($IP rule list | awk '$1 == "'$(($iface_id+2000)):'"')" ]; do
|
|
$IP rule del pref $(($iface_id+2000))
|
|
done
|
|
|
|
[ $ACTION == "ifup" ] && $IP rule add pref $(($iface_id+1000)) iif $DEVICE lookup main
|
|
[ $ACTION == "ifup" ] && $IP rule add pref $(($iface_id+2000)) fwmark $(($iface_id*256))/0xff00 lookup $iface_id
|
|
}
|
|
|
|
mwan3_track()
|
|
{
|
|
local track_ip track_ips reliability count timeout interval down up
|
|
|
|
mwan3_list_track_ips()
|
|
{
|
|
track_ips="$1 $track_ips"
|
|
}
|
|
config_list_foreach $INTERFACE track_ip mwan3_list_track_ips
|
|
|
|
if [ -e /var/run/mwan3track-$INTERFACE.pid ] ; then
|
|
kill $(cat /var/run/mwan3track-$INTERFACE.pid) &> /dev/null
|
|
rm /var/run/mwan3track-$INTERFACE.pid &> /dev/null
|
|
fi
|
|
|
|
if [ -n "$track_ips" ]; then
|
|
config_get reliability $INTERFACE reliability 1
|
|
config_get count $INTERFACE count 1
|
|
config_get timeout $INTERFACE timeout 4
|
|
config_get interval $INTERFACE interval 10
|
|
config_get down $INTERFACE down 5
|
|
config_get up $INTERFACE up 5
|
|
|
|
if ! $IPT -S mwan3_track_$INTERFACE &> /dev/null; then
|
|
$IPT -N mwan3_track_$INTERFACE
|
|
$IPT -A mwan3_output_hook -p icmp -m icmp --icmp-type 8 -m length --length 32 -j mwan3_track_$INTERFACE
|
|
fi
|
|
|
|
$IPT -F mwan3_track_$INTERFACE
|
|
|
|
for track_ip in $track_ips; do
|
|
$IPT -A mwan3_track_$INTERFACE -d $track_ip -j MARK --set-xmark 0xff00/0xff00
|
|
done
|
|
|
|
[ -x /usr/sbin/mwan3track ] && /usr/sbin/mwan3track $INTERFACE $DEVICE $reliability $count $timeout $interval $down $up $track_ips &
|
|
else
|
|
$IPT -D mwan3_output_hook -p icmp -m icmp --icmp-type 8 -m length --length 32 -j mwan3_track_$INTERFACE &> /dev/null
|
|
$IPT -F mwan3_track_$INTERFACE &> /dev/null
|
|
$IPT -X mwan3_track_$INTERFACE &> /dev/null
|
|
fi
|
|
}
|
|
|
|
mwan3_set_policy()
|
|
{
|
|
local iface_count iface_id INTERFACE metric probability weight
|
|
|
|
config_get INTERFACE $1 interface
|
|
config_get metric $1 metric 1
|
|
config_get weight $1 weight 1
|
|
|
|
[ -n "$INTERFACE" ] || return 0
|
|
|
|
config_foreach mwan3_get_iface_id interface
|
|
|
|
[ -n "$iface_id" ] || return 0
|
|
|
|
if $IPT -S mwan3_iface_$INTERFACE &> /dev/null; then
|
|
if [ "$metric" -lt "$lowest_metric" ]; then
|
|
|
|
total_weight=$weight
|
|
$IPT -F mwan3_policy_$policy
|
|
$IPT -A mwan3_policy_$policy -m mark --mark 0x0/0xff00 -m comment --comment "$INTERFACE $weight $weight" -j MARK --set-xmark $(($iface_id*256))/0xff00
|
|
|
|
lowest_metric=$metric
|
|
|
|
elif [ "$metric" -eq "$lowest_metric" ]; then
|
|
|
|
total_weight=$(($total_weight+$weight))
|
|
probability=$(($weight*1000/$total_weight))
|
|
|
|
if [ "$probability" -lt 10 ]; then
|
|
probability="0.00$probability"
|
|
elif [ $probability -lt 100 ]; then
|
|
probability="0.0$probability"
|
|
elif [ $probability -lt 1000 ]; then
|
|
probability="0.$probability"
|
|
else
|
|
probability="1"
|
|
fi
|
|
|
|
probability="-m statistic --mode random --probability $probability"
|
|
|
|
$IPT -I mwan3_policy_$policy -m mark --mark 0x0/0xff00 $probability -m comment --comment "$INTERFACE $weight $total_weight" -j MARK --set-xmark $(($iface_id*256))/0xff00
|
|
fi
|
|
fi
|
|
}
|
|
|
|
mwan3_set_policies_iptables()
|
|
{
|
|
local last_resort lowest_metric policy total_weight
|
|
|
|
policy=$1
|
|
|
|
config_get last_resort $1 last_resort unreachable
|
|
|
|
if [ "$policy" != $(echo "$policy" | cut -c1-15) ]; then
|
|
$LOG warn "Policy $policy exceeds max of 15 chars. Not setting policy" && return 0
|
|
fi
|
|
|
|
if ! $IPT -S mwan3_policy_$policy &> /dev/null; then
|
|
$IPT -N mwan3_policy_$policy
|
|
fi
|
|
|
|
$IPT -F mwan3_policy_$policy
|
|
|
|
case "$last_resort" in
|
|
blackhole)
|
|
$IPT -A mwan3_policy_$policy -m mark --mark 0x0/0xff00 -m comment --comment "blackhole" -j MARK --set-xmark 0xfd00/0xff00
|
|
;;
|
|
default)
|
|
$IPT -A mwan3_policy_$policy -m mark --mark 0x0/0xff00 -m comment --comment "default" -j MARK --set-xmark 0xff00/0xff00
|
|
;;
|
|
*)
|
|
$IPT -A mwan3_policy_$policy -m mark --mark 0x0/0xff00 -m comment --comment "unreachable" -j MARK --set-xmark 0xfe00/0xff00
|
|
;;
|
|
esac
|
|
|
|
lowest_metric=256
|
|
total_weight=0
|
|
|
|
config_list_foreach $policy use_member mwan3_set_policy
|
|
}
|
|
|
|
mwan3_set_user_rules_iptables()
|
|
{
|
|
local proto src_ip src_port dest_ip dest_port use_policy
|
|
|
|
config_get proto $1 proto all
|
|
config_get src_ip $1 src_ip 0.0.0.0/0
|
|
config_get src_port $1 src_port 0:65535
|
|
config_get dest_ip $1 dest_ip 0.0.0.0/0
|
|
config_get dest_port $1 dest_port 0:65535
|
|
config_get use_policy $1 use_policy
|
|
|
|
if [ -n "$use_policy" ]; then
|
|
if [ "$use_policy" == "default" ]; then
|
|
use_policy="MARK --set-xmark 0xff00/0xff00"
|
|
elif [ "$use_policy" == "unreachable" ]; then
|
|
use_policy="MARK --set-xmark 0xfe00/0xff00"
|
|
elif [ "$use_policy" == "blackhole" ]; then
|
|
use_policy="MARK --set-xmark 0xfd00/0xff00"
|
|
else
|
|
use_policy="mwan3_policy_$use_policy"
|
|
fi
|
|
|
|
case $proto in
|
|
tcp|udp)
|
|
$IPT -A mwan3_rules -p $proto -s $src_ip -d $dest_ip -m multiport --sports $src_port -m multiport --dports $dest_port -m mark --mark 0/0xff00 -m comment --comment "$1" -j $use_policy &> /dev/null
|
|
;;
|
|
*)
|
|
$IPT -A mwan3_rules -p $proto -s $src_ip -d $dest_ip -m mark --mark 0/0xff00 -m comment --comment "$1" -j $use_policy &> /dev/null
|
|
;;
|
|
esac
|
|
fi
|
|
}
|
|
|
|
mwan3_ifupdown()
|
|
{
|
|
local counter enabled iface_count iface_id route_args wan_metric
|
|
|
|
config_load mwan3
|
|
config_foreach mwan3_get_iface_id interface
|
|
|
|
[ -n "$iface_id" ] || return 0
|
|
[ "$iface_count" -le 250 ] || return 0
|
|
unset iface_count
|
|
|
|
config_get enabled $INTERFACE enabled 0
|
|
|
|
counter=0
|
|
|
|
if [ $ACTION == "ifup" ]; then
|
|
[ "$enabled" -eq 1 ] || return 0
|
|
|
|
while [ -z "$($IP route list dev $DEVICE default | head -1)" -a "$counter" -lt 10 ]; do
|
|
sleep 1
|
|
let counter++
|
|
if [ "$counter" -ge 10 ]; then
|
|
$LOG warn "Could not find gateway for interface $INTERFACE ($DEVICE)" && return 0
|
|
fi
|
|
done
|
|
|
|
route_args=$($IP route list dev $DEVICE default | head -1 | sed '/.*via \([^ ]*\) .*$/!d;s//via \1/;q' | egrep '[0-9]{1,3}(\.[0-9]{1,3}){3}')
|
|
route_args="nexthop $route_args dev $DEVICE"
|
|
fi
|
|
|
|
while [ "$(pgrep -f -o hotplug-call)" -ne $$ -a "$counter" -lt 60 ]; do
|
|
sleep 1
|
|
let counter++
|
|
if [ "$counter" -ge 60 ]; then
|
|
$LOG warn "Timeout waiting for older hotplug processes to finish. $ACTION interface $INTERFACE (${DEVICE:-unknown}) aborted" && return 0
|
|
fi
|
|
done
|
|
|
|
$LOG notice "$ACTION interface $INTERFACE (${DEVICE:-unknown})"
|
|
|
|
mwan3_set_general_iptables
|
|
mwan3_set_general_rules
|
|
mwan3_set_iface_iptables
|
|
mwan3_set_iface_route
|
|
mwan3_set_iface_rules
|
|
|
|
[ $ACTION == "ifup" ] && mwan3_track
|
|
|
|
config_foreach mwan3_set_policies_iptables policy
|
|
config_foreach mwan3_set_user_rules_iptables rule
|
|
}
|
|
|
|
[ -n "$ACTION" ] || exit 0
|
|
[ -n "$INTERFACE" ] || exit 0
|
|
|
|
if [ $ACTION == "ifup" ]; then
|
|
[ -n "$DEVICE" ] || exit 0
|
|
fi
|
|
|
|
local IP IPT LOG
|
|
|
|
IP="/usr/sbin/ip -4"
|
|
IPT="/usr/sbin/iptables -t mangle -w"
|
|
LOG="/usr/bin/logger -t mwan3 -p"
|
|
|
|
case "$ACTION" in
|
|
ifup|ifdown)
|
|
mwan3_ifupdown
|
|
mwan3_set_connected_iptables
|
|
;;
|
|
esac
|
|
|
|
exit 0
|