端口管理脚本生成器

#!/bin/bash
# ============================================================
# 端口封堵脚本 - 自动生成
# 生成时间: 2026/7/1 02:31:03
# ============================================================

RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m'

if [ "$(id -u)" -ne 0 ]; then
    echo -e "${RED}错误: 请使用 sudo 运行此脚本${NC}"
    exec sudo "$0" "$@"
fi

echo "========================================"
echo "  端口封堵脚本"
echo "========================================"
echo ""
# 备份现有规则
BACKUP_FILE="/tmp/iptables_backup_$(date +%Y%m%d_%H%M%S).rules"
iptables-save > "$BACKUP_FILE"
echo -e "${GREEN}✓ 已备份规则到 $BACKUP_FILE${NC}"
echo ""

echo -e "${YELLOW}警告: 未指定端口${NC}"

# 保存规则
iptables-save > /etc/iptables.rules
echo -e "${GREEN}✓ 规则已保存${NC}"

# 配置开机加载
if [ ! -f /etc/network/if-pre-up.d/iptables ]; then
    echo '#!/bin/sh' > /etc/network/if-pre-up.d/iptables
    echo 'iptables-restore < /etc/iptables.rules' >> /etc/network/if-pre-up.d/iptables
    chmod +x /etc/network/if-pre-up.d/iptables
    echo -e "${GREEN}✓ 已配置开机加载${NC}"
fi

echo ""
echo "操作完成！"