前提

导入的py脚本为《数据处理 | pandas处理数据后输出为Excel表格》

🐍Python🐍

逻辑

  1. 遍历文件夹
  2. 获取PE文件的ImpHash

2.1 判断是否是PE文件
2.1.1 是：获取ImpHash

  1. 获取ImpHash一致的文件MD5

3.1 对比指定的ImpHash
3.1.1一致：获取文件MD5

代码

  1. import os
  2. import re
  3. from 遍历文件夹提取文件名andMD5andImpHash import PE_isPE
  4. from 遍历文件夹提取文件名andMD5andImpHash import PE_GetImpHash
  5. from 遍历文件夹提取文件名andMD5andImpHash import GetFileMD5
  7. msgStart = r"——————————Python脚本开始——————————"
  8. msgEnd = r"——————————Python脚本结束——————————"
  9. strNewLine = "\r\n"
  11. def EnumFile2CmpImphash(pathDir, theImpHash):
  12.     print("——————————遍历开始——————————")
  13.     for home, dirs, files in os.walk(pathDir):
  14.         for file in files:
  15.             pathFile = os.path.join(home, file)
  16.             # 查看是否是PE文件，是则获取其ImpHash
  17.             isPE = PE_isPE(pathFile)
  18.             if isPE:
  19.                 # 获取ImpHash
  20.                 fileImpHash = PE_GetImpHash(pathFile)
  21.             else:
  22.                 # 提示非PE
  23.                 fileImpHash = "非PE"
  24.             try:
  25.                 # 忽略大小写比较ImpHash字符串
  26.                 if (re.search(fileImpHash, theImpHash, re.IGNORECASE)):
  27.                     # ImpHash相同的PE获取其MD5
  28.                     fileMD5 = GetFileMD5(pathFile)
  29.                     print("ImpHash相同的文件信息", strNewLine + "路径：" + pathFile + strNewLine + "MD5：", fileMD5 + strNewLine)
  30.             except:
  31.                 continue
  32.     print("——————————遍历完成——————————")
  34. # 输入需要扫描的文件夹路径和指定的ImpHash
  35. pathDir = (r'D:\')
  36. theImpHashStopV2 = "FB36740C126064E8D052AD24F396D94D"
  38. if __name__ == '__main__':
  39.     print(msgStart)
  40.     EnumFile2CmpImphash(pathDir, theImpHashStopV2)
  41.     print(msgEnd)

运行结果

image.png