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.
39 lines
1.4 KiB
39 lines
1.4 KiB
#!/bin/sh
|
|
#
|
|
# Copyright (C) 2018 rosysong@rosinson.com
|
|
#
|
|
|
|
. /lib/nft-qos/core.sh
|
|
|
|
qosdef_monitor_get_ip_handle() { # <family> <chain> <ip>
|
|
echo $(nft list chain $1 nft-qos-monitor $2 -a 2>/dev/null | grep $3 | awk '{print $11}')
|
|
}
|
|
|
|
qosdef_monitor_add() { # <mac> <ip> <hostname>
|
|
handle_dl=$(qosdef_monitor_get_ip_handle $NFT_QOS_INET_FAMILY download $2)
|
|
[ -z "$handle_dl" ] && nft add rule $NFT_QOS_INET_FAMILY nft-qos-monitor download ip daddr $2 counter
|
|
handle_ul=$(qosdef_monitor_get_ip_handle $NFT_QOS_INET_FAMILY upload $2)
|
|
[ -z "$handle_ul" ] && nft add rule $NFT_QOS_INET_FAMILY nft-qos-monitor upload ip saddr $2 counter
|
|
}
|
|
|
|
qosdef_monitor_del() { # <mac> <ip> <hostname>
|
|
local handle_dl handle_ul
|
|
handle_dl=$(qosdef_monitor_get_ip_handle $NFT_QOS_INET_FAMILY download $2)
|
|
handle_ul=$(qosdef_monitor_get_ip_handle $NFT_QOS_INET_FAMILY upload $2)
|
|
[ -n "$handle_dl" ] && nft delete handle $handle_dl
|
|
[ -n "$handle_ul" ] && nft delete handle $handle_ul
|
|
}
|
|
|
|
# init qos monitor
|
|
qosdef_init_monitor() {
|
|
local hook_ul="input" hook_dl="postrouting"
|
|
|
|
[ -z "$NFT_QOS_HAS_BRIDGE" ] && {
|
|
hook_ul="postrouting"
|
|
hook_dl="input"
|
|
}
|
|
|
|
nft add table $NFT_QOS_INET_FAMILY nft-qos-monitor
|
|
nft add chain $NFT_QOS_INET_FAMILY nft-qos-monitor upload { type filter hook $hook_ul priority 0\; }
|
|
nft add chain $NFT_QOS_INET_FAMILY nft-qos-monitor download { type filter hook $hook_dl priority 0\; }
|
|
}
|