quil-docker/xs.sh
2024-11-26 23:56:47 +08:00

60 lines
2.1 KiB
Bash

#!/bin/bash
# 定义网络接口和限速参数
INTERFACE="eno1"
LIMIT_RATE="2mbit"
add_limit() {
echo "Applying bandwidth limit for ports 8337 and 8338..."
# 加载 ifb 模块并确保 ifb0 存在且启用
modprobe ifb 2>/dev/null
ip link show ifb0 >/dev/null 2>&1 || ip link add ifb0 type ifb
ip link set ifb0 up 2>/dev/null
# 清理旧规则
tc qdisc del dev "$INTERFACE" root 2>/dev/null || true
tc qdisc del dev ifb0 root 2>/dev/null || true
# 添加上行规则
tc qdisc add dev "$INTERFACE" root handle 1: htb
tc class add dev "$INTERFACE" parent 1: classid 1:1 htb rate "$LIMIT_RATE" ceil "$LIMIT_RATE"
tc filter add dev "$INTERFACE" protocol ip parent 1:0 prio 1 u32 match ip dport 8337 0xffff flowid 1:1
tc filter add dev "$INTERFACE" protocol ip parent 1:0 prio 1 u32 match ip dport 8338 0xffff flowid 1:1
# 添加下行规则
tc qdisc add dev ifb0 root handle 1: htb
tc class add dev ifb0 parent 1: classid 1:1 htb rate "$LIMIT_RATE" ceil "$LIMIT_RATE"
tc filter add dev ifb0 protocol ip parent 1:0 prio 1 u32 match ip sport 8337 0xffff flowid 1:1
tc filter add dev ifb0 protocol ip parent 1:0 prio 1 u32 match ip sport 8338 0xffff flowid 1:1
# 将 ingress 流量重定向到 ifb0
tc qdisc add dev "$INTERFACE" ingress
tc filter add dev "$INTERFACE" parent ffff: protocol ip prio 1 u32 match ip sport 8337 0xffff action mirred egress redirect dev ifb0
tc filter add dev "$INTERFACE" parent ffff: protocol ip prio 1 u32 match ip sport 8338 0xffff action mirred egress redirect dev ifb0
echo "Bandwidth limit applied successfully."
}
delete_limit() {
echo "Removing bandwidth limit..."
tc qdisc del dev "$INTERFACE" root 2>/dev/null || true
tc qdisc del dev "$INTERFACE" ingress 2>/dev/null || true
tc qdisc del dev ifb0 root 2>/dev/null || true
ip link set ifb0 down 2>/dev/null || true
echo "Bandwidth limit removed successfully."
}
case "$1" in
add)
add_limit
;;
del)
delete_limit
;;
*)
echo "Usage: $0 {add|del}"
exit 1
;;
esac