本文实例讲述了python实现文件快照加密保护的方法。分享给大家供大家参考。具体如下:
这段代码可以对指定的目录进行扫描,包含子目录,对指定扩展名的文件进行SHA-1加密后存储在cvs文件,以防止文件被篡改
调用方法:python snapper.py > todayCheck.csv
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
# 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程序设计有所帮助。