◆◆0

[转]Linux系统防CC攻击自动拉黑IP增强版Shell脚本

来源： 2019-01-13 00:14:46 Linux 1927 1

前天没事写了一个防 CC 攻击的 Shell 脚本，没想到这么快就要用上了，原因是因为 360 网站卫士的缓存黑名单突然无法过滤后台，导致 WordPress 无法登录！虽然，可以通过修改本地 hosts 文件来解决这个问题，但是还是想暂时取消 CDN 加速和防护来测试下服务器的性能优化及安全防护。

前天写的 Shell 脚本是加入到 crontab 计划任务执行的，每 5 分钟执行一次，今天实际测试了下，可还是可以用的，但是感觉 5 分钟时间有点过长，无法做到严密防护。于是稍微改进了下代码，现在简单的分享下！

一、Shell 代码

Shell

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

#!/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.comnTo:收邮件地址@qq.comnSubject: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 表示并发数，可自行调整）：

Shell

1	nohup ./deny_blackip.sh 50 &

执行后会出现如下信息：

Shell

1
2
3

[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：

Shell