检测网站挂马程序（Python）

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

系统管理员通常从svn/git中检索代码，部署站点后通常首先会生成该站点所有文件的MD5值，如果上线后网站页面内容被篡改（如挂马）等，可以比对之前生成MD5值快速查找去那些文件被更改，为了使系统管理员第一时间发现，可结合crontab或nagios等工具。

程序测试如下：

    # python check_change.py

        Usage: python check_change.py update /home/wwwroot

               python check_change.py check /home/wwwroot

    # python check_change.py update /data/www #生成站点的md5值

    # echo ' ' > /data/www/sitemap.html #测试清空文件

    # rm -rf /data/www/sitemap.xml #测试删除文件

    # python check_change.py check /data/www  #查找那些文件被篡改

    /data/www/sitemap.xml

    /data/www/sitemap.html

代码如下（check_change.py）：

    #!/usr/bin/env python  

    import os,sys,subprocess  

    def update(path):

        f = open(file,'w')

        for root,dirs,files in os.walk(path):

            for name in files:

                line = os.path.join(root, name)

                (stdin,stderr) = subprocess.Popen(['md5sum',line],stdout=subprocess.PIPE).communicate()

                f.write(stdin)

        f.close()  

    def check(path):

        f = open(file,'r')

        for line in f:

            check_ok = """echo '%s' | md5sum -c > /dev/null 2>&1""" % line

            #print check_ok

            if not subprocess.call(check_ok, shell = True) == 0:

                abnormal = line.split()

                print abnormal[1]

        f.close()  

    def Usage():

        print '''

        Usage: python %s update /home/wwwroot

               python %s check /home/wwwroot

        ''' % (sys.argv[0],sys.argv[0])

        sys.exit()  

    if len(sys.argv) != 3:

        Usage()  

    file = 'file.key'

    model = sys.argv[1]

    path = sys.argv[2]  

    if os.path.exists(path) == False:

        print "\033[;31mThe directory or file does not exist\033[0m"

        sys.exit()

    elif model == 'update':

        update(path)

    elif model == 'check':

        check(path)

    else:

        Usage()