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 pythonimport os,sys,subprocessdef 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()

