python实现文件快照加密保护的方法

1215次阅读  |  发布于5年以前

本文实例讲述了python实现文件快照加密保护的方法。分享给大家供大家参考。具体如下:

这段代码可以对指定的目录进行扫描,包含子目录,对指定扩展名的文件进行SHA-1加密后存储在cvs文件,以防止文件被篡改

调用方法:python snapper.py > todayCheck.csv


    # Hello, this is a script written in Python. See http://www.pyhon.org
    #
    # Snapper 1.2p
    #
    # This script will walk a directory (and its subdirectories) and compute
    # SHA (Secure Hash Algorithm) for specific files (according to their
    # extensions) and ouput a CSV file (suited for loading into a spreadsheet
    # editor,a database or simply comparing with diff or ExamDiff.).
    #
    # You can redirect the output of this script to a file.
    # eg. python snapper.py > todayCheck.csv
    #
    # This script can be usefull to check system files tampering.
    #
    # This script is public domain. Feel free to reuse it.
    # The author is:
    #    Sebastien SAUVAGE
    #    <sebsauvage at sebsauvage dot net>
    #    http://sebsauvage.net
    #
    # More quick & dirty scripts are available at http://sebsauvage.net/python/
    #
    # Directory to scan and extensions are hardcoded below:
    directoryStart = r'c:\windows'
    extensionList=['.exe','.dll','.ini','.ocx','.cpl','.vxd','.drv','.vbx','.com','.bat','.src',
            '.sys','.386','.acm','.ax', '.bpl','.bin','.cab','.olb','.mpd','.pdr','.jar']
    import os,string,sha,stat,sys
    def snapper ( directoryStart , extensionList ) :
      os.path.walk( directoryStart, snapper_callback, extensionList )
    def snapper_callback ( extensionList , directory, files ) :
      sys.stderr.write('Scanning '+directory+'\n')
      for fileName in files:
        if os.path.isfile( os.path.join(directory,fileName) ) :
          if string.lower(os.path.splitext(fileName)[1]) in extensionList :
            filelist.append(fileSHA ( os.path.join(directory,fileName) ))
    def fileSHA ( filepath ) :
      sys.stderr.write(' Reading '+os.path.split(filepath)[1]+'\n')
      file = open(filepath,'rb')
      digest = sha.new()
      data = file.read(65536)
      while len(data) != 0:
        digest.update(data)
        data = file.read(65536)
      file.close()
      return '"'+filepath+'",'+str(os.stat(filepath)[6])+',"'+digest.hexdigest()+'"'
    sys.stderr.write('Snapper 1.1p - http://sebsauvage.net/python/\n')
    filelist = []
    snapper( directoryStart , extensionList )
    sys.stderr.write('Sorting...\n')
    filelist.sort()
    filelist.insert(0, '"File path","File size","SHA"' )
    sys.stderr.write('Printing...\n')
    for line in filelist:
     print line
    sys.stderr.write('All done.\n')

希望本文所述对大家的Python程序设计有所帮助。

Copyright© 2013-2020

All Rights Reserved 京ICP备2023019179号-8