当前位置 博文首页 > 用Python和MD5实现网站挂马检测程序

    用Python和MD5实现网站挂马检测程序

    作者:admin 时间:2021-06-29 08:58

    一、程序测试

    复制代码 代码如下:
    # 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()

    js