一、程序测试

复制代码 代码如下:# 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()