Linux下用Shell指令碼防CC攻擊自動拉黑IP增強版

一、Shell代碼

#!/bin/bash
#Author:ZhangGe
#Desc:Auto Deny Black_IP Script.
#Date:2014-11-05

#取得參數$1為並發閾值，若留空則預設允許單IP最大50並發(實際測試發現，2M頻寬，十來個並發伺服器就已經無法訪問了！)
if [[ -z $1 ]];then
        num=50
else
        num=$1
fi
 
#巧妙的進入到指令碼工作目錄
cd $(cd $(dirname $BASH_SOURCE) && pwd)
 
#請求檢查、判斷及拉黑主功能函數
function check(){
        iplist=`netstat -an |grep ^tcp.*:80|egrep -v 'LISTEN|127.0.0.1'|awk -F"[ ]+|[:]" '{print $6}'|sort|uniq -c|sort -rn|awk -v str=$num '{if ($1>str){print $2}}'`
        if [[ ! -z $iplist ]];
                then
                >./black_ip.txt
                for black_ip in $iplist
                do
                        #白名單過濾中已取消IP段的判斷功能，可根據需要自行修改以下代碼(請參考前天寫的指令碼)
                        #exclude_ip=`echo $black_ip | awk -F"." '{print $1"."$2"."$3}'`
                        #grep -q $exclude_ip ./white_ip.txt
                        grep -q $black_ip ./white_ip.txt
                        if [[ $? -eq 0 ]];then
                                echo "$black_ip (white_ip)" >>./black_ip.txt
                        else
                                echo $black_ip >>./black_ip.txt    
                                iptables -nL | grep $black_ip ||(iptables -I INPUT -s $black_ip -j DROP & echo "$black_ip  `date +%Y-%m-%H:%M:%S`">>./deny.log & echo 1 >./sendmail)
                        fi
                done
                #存在並發超過閾值的單IP就發送郵件
                if [[ `cat ./sendmail` == 1 ]];then sendmsg;fi
        fi
}
 
#發郵件函數
function sendmsg(){
        netstat -nutlp | grep "sendmail" >/dev/null 2>&1 || /etc/init.d/sendmail start >/dev/null 2>&1
        echo -e "From: 發郵件地址@qq.com\nTo:收郵件地址@qq.com\nSubject:Someone Attacking your system!!\nIts Ip is" >./message
        cat ./black_ip.txt >>./message
        /usr/sbin/sendmail -f 發郵件地址@qq.com -t 收郵件地址@qq.com -i <./message
        >./sendmail
}
 
#間隔10s無限迴圈檢查函數
while true
do
        check
        #每隔10s檢查一次，時間可根據需要自訂
        sleep 10
done

二、執行指令碼

將以上代碼儲存為deny_blackip.sh之後，使用如下命令後台執行指令碼（後面的50表示並發數，可自行調整）：

nohup ./deny_blackip.sh 50 &

執行後會出現如下資訊：

[root@Mars_Server iptables]# nohup ./deny_blackip.sh 50 &
[1] 23630
[root@Mars_Server iptables]# nohup: ignoring input and appending output to `nohup.out'

表示如果指令碼產生輸出資訊，將會寫入到nohup.out檔案，可以看到目前的目錄已經產生了一個空的nohup.out：

[root@Mars_Server iptables]# ll nohup.out
-rw------- 1 root root 0 Nov  5 21:15 nohup.out

好了，現在你執行執行ps aux 應該可以找到如下進程：

root     23630  0.0  0.2   5060  1224 pts/0    S    21:15   0:00 /bin/bash ./deny_blackip.sh
root     23964  0.0  0.0   4064   508 pts/0    S    21:19   0:00 sleep 10

一切順利！每10s將檢查一次伺服器請求，如果某個IP超過50個並發，立即拉黑，並發一封郵件給你！