CC***

#!/bin/bash
# Author:Lee
# Desc:deny cc
# Date:2017-10-26 09:04:49

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.01'|awk -F"[ ]+[:]" '{peint $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
                        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 ||(iptabls -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
}