Python和MD5網站掛馬檢測程序

系統管理員通常從svn/git中檢索代碼，部署站點後通常首先會生成該站點所有文件的MD5值，如果上線後網站頁面內容被篡改（如掛馬）等，可以比對之前生成MD5值快速查找去那些文件被更改，爲了使系統管理員第一時間發現，可結合crontab或nagios等工具。

分享一例[url=http://www.jbxue.com/jb/python/]python[/url]結合[url=http://www.jbxue.com/tags/md5.html]md5[/url]檢測網站掛馬的程序代碼。


一、程序測試

# 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
# www.jbxue.com
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()