Python实例分享:快速查找出被挂马的文件_python教程-查字典教程网
Python实例分享:快速查找出被挂马的文件
Python实例分享:快速查找出被挂马的文件
发布时间:2017-01-07 来源:查字典编辑
摘要:思路需要实现准备一份未受感染的源代码和一份可能受感染的源代码,然后运行以下脚本,就能找出到底哪些文件被挂马了。其中,主要是根据比对2份文件的...

思路

需要实现准备一份未受感染的源代码和一份可能受感染的源代码,然后运行以下脚本,就能找出到底哪些文件被挂马了。

其中,主要是根据比对2份文件的md5值来过滤可能被挂马的文件(确切的说应该是被修改过的文件)

Python脚本

复制代码 代码如下:

__author__ = 'Flying'

#coding:utf-8

#Date:2014.6.5

#检测修改过的文件

import os,sys,hashlib,datetime

global_DirOld = ""

global_DirNew = ""

global_FilesList = []

#输入要比对的文件路径

def InputDirPath():

global global_DirOld,global_DirNew

global_DirOld = unicode(raw_input("请输入备份文件所在目录:"),"utf-8")

while not os.path.exists(global_DirOld):

print u"指定的路径不存在,请重新输入"

global_DirOld = unicode(raw_input("请输入备份文件所在目录:"),"utf-8")

global_DirNew = unicode(raw_input("请输入要检测文件的目录:"),"utf-8")

while not os.path.exists(global_DirNew):

print u"指定的路径不存在,请重新输入"

global_DirNew = unicode(raw_input("请输入要检测文件的目录:"),"utf-8")

#将数据保存到文件中

def SaveToFile(filePath,content):

try:

f = open(filePath,"a+")

f.write(content.encode("utf-8") + "n")

f.close()

except Exception,ex:

print "Error:" + str(ex)

#计算文件的MD5值

def CalcMD5(filepath):

try:

#以二进制的形式打开

with open(filepath,'rb') as f:

md5obj = hashlib.md5()

md5obj.update(f.read())

hash = md5obj.hexdigest()

return hash

except Exception,ex:

print "Error:" + str(ex)

return None

#遍历目录下的所有文件

def GetAllSubFiles():

global global_FilesList

for dir in os.walk(global_DirNew):

for file in dir[2]:

filePath = dir[0] + os.sep + file

global_FilesList.append(filePath[len(global_DirNew)+1:])

#列出新增文件和变动的文件

def ListChangedFiles():

global global_DirOld,global_DirNew,global_FilesList

print u"变动或新增的文件:"

for file in global_FilesList:

filePathOld = global_DirOld + os.sep + file

filePathNew = global_DirNew + os.sep + file

if not os.path.exists(filePathOld) or CalcMD5(filePathOld)!=CalcMD5(filePathNew):

content = "[" + datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')+ "]" + filePathNew

print content

SaveToFile("ChangedFiles.txt",content)

if __name__=="__main__":

InputDirPath()

GetAllSubFiles()

ListChangedFiles()

脚本执行结果

相关阅读
推荐文章
猜你喜欢
附近的人在看
推荐阅读
拓展阅读
  • 大家都在看
  • 小编推荐
  • 猜你喜欢
  • 最新python学习
    热门python学习
    脚本专栏子分类