用python和md5实现网站挂马检测程序

一、程序测试

代码如下:

# 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()