土小帽 发表于 2023-11-3 20:34:58

检测网站挂马程序(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
[*]    f.close()
[*]
[*]def Usage():
[*]    print '''
[*]    Usage: python %s update /home/wwwroot
[*]         python %s check /home/wwwroot
[*]    ''' % (sys.argv,sys.argv)
[*]    sys.exit()
[*]
[*]if len(sys.argv) != 3:
[*]    Usage()
[*]
[*]file = 'file.key'
[*]model = sys.argv
[*]path = sys.argv
[*]
[*]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()复制代码
页: [1]
查看完整版本: 检测网站挂马程序(Python)