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

