LinuxSir.cn,穿越时空的Linuxsir!

 找回密码
 注册
搜索
热搜: shell linux mysql
查看: 1425|回复: 3

通过 iptables 限制连接数量的另一种方法

[复制链接]
发表于 2009-7-14 22:46:36 | 显示全部楼层 |阅读模式
通过 iptables 限制连接数量的另一种方法
作者:bixjun

首先建立目录:

mkdir /tmp/blockip
mkdir /tmp/blockip/ip
mkdir /tmp/blockip/mac

然后建立一个 shell 脚本程序(内容如下),运行即可。可以限制内网机器的连接数量,超出连接数量即被阻断,待连接数降回设定值之后即可自动恢复连接。可用于限制内网中个别机器的多线程连接。


#!/bin/bash

rm -f /tmp/blockip/ip/*
rm -f /tmp/blockip/mac/*

while true
do
touch 2>/dev/null `netstat-nat -n |awk '{print $2;}'| \
grep 192.168 | \
awk ' FS=":" { print $1}'| \
sort | \
awk ' BEGIN {
        COUNT=0
        IP=""
      }
      {
        if ( $1 != IP ) {
          if (COUNT >= 200) {
            print "/tmp/blockip/ip/"IP
          }
            COUNT=0
            IP=$1
          } else {
            COUNT++
        }
      }'`
cd /tmp/blockip/ip
for ip in *
do
   if [ -e $ip ] && [ ! -e ../mac/$ip ]
      then
          touch 2>/dev/null ../mac/$ip
          iptables -I FORWARD -s $ip -j DROP
          iptables -I FORWARD -d $ip -j DROP
          echo `date "+%Y-%m-%d %H: %M: %S "`$ip was blocked for `netstat-nat -n -s $ip|wc|awk '{print $1}'` connections
   fi
done

cd /tmp/blockip/mac

for ip in *
do
if [ -e $ip ]
  then
  netstat-nat -n -s $ip | wc | \
  awk -v IP=$ip ' {
        if ( $1 < 100 ) {
          print "iptables -D FORWARD -s " IP " -j DROP"
          print "iptables -D FORWARD -d " IP " -j DROP"
          print "rm " IP
          print "rm ../ip/" IP
          print "echo `date \"+%Y-%m-%d %H: %M: %S \"`\"" IP " was unblocked\""
        }
      } ' >> unblock
fi
done
if [ -e unblock ]
then
  chmod 700 unblock
  ./unblock
  rm unblock -f
fi
sleep 6
done
发表于 2009-7-15 14:40:52 | 显示全部楼层
收下 可能以后有用
回复 支持 反对

使用道具 举报

发表于 2009-7-15 17:19:46 | 显示全部楼层
收下 可能以后有用
回复 支持 反对

使用道具 举报

 楼主| 发表于 2009-7-15 18:02:00 | 显示全部楼层
做了一点改进,把连接数太多的受限IP插在已经建立好的链路后面,这样可以保证可限制后面的连接,而前面已经建立好的连接可以继续使用。只有关闭了前面的连接以后或者连接数降下来以后才可以建立新的连接。
回复 支持 反对

使用道具 举报

您需要登录后才可以回帖 登录 | 注册

本版积分规则

快速回复 返回顶部 返回列表