summaryrefslogtreecommitdiffstats
path: root/Tools/scripts/sum5.py
blob: 09bc6863b2037ed296f4c013f52c889ced6f8f65 (plain)
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#! /usr/bin/env python

# print md5 checksum for files

bufsize = 8096
fnfilter = None
rmode = 'r'

usage = """
usage: sum5 [-b] [-t] [-l] [-s bufsize] [file ...]
-b        : read files in binary mode
-t        : read files in text mode (default)
-l        : print last pathname component only
-s bufsize: read buffer size (default %d)
file ...  : files to sum; '-' or no files means stdin
""" % bufsize

import sys
import string
import os
import md5
import regsub

StringType = type('')
FileType = type(sys.stdin)

def sum(*files):
	sts = 0
	if files and type(files[-1]) == FileType:
		out, files = files[-1], files[:-1]
	else:
		out = sys.stdout
	if len(files) == 1 and type(files[0]) != StringType:
		files = files[0]
	for f in files:
		if type(f) == StringType:
			if f == '-':
				sts = printsumfp(sys.stdin, '<stdin>', out) or sts
			else:
				sts = printsum(f, out) or sts
		else:
			sts = sum(f, out) or sts
	return sts

def printsum(file, out = sys.stdout):
	try:
		fp = open(file, rmode)
	except IOError, msg:
		sys.stderr.write('%s: Can\'t open: %s\n' % (file, msg))
		return 1
	if fnfilter:
		file = fnfilter(file)
	sts = printsumfp(fp, file, out)
	fp.close()
	return sts

def printsumfp(fp, file, out = sys.stdout):
	m = md5.md5()
	try:
		while 1:
			data = fp.read(bufsize)
			if not data: break
			m.update(data)
	except IOError, msg:
		sys.stderr.write('%s: I/O error: %s\n' % (file, msg))
		return 1
	out.write('%s %s\n' % (hexify(m.digest()), file))
	return 0

def hexify(s):
	res = ''
	for c in s:
		res = res + '%02x' % ord(c)
	return res

def main(args = sys.argv[1:], out = sys.stdout):
	global fnfilter, rmode, bufsize
	import getopt
	try:
		opts, args = getopt.getopt(args, 'blts:')
	except getopt.error, msg:
		sys.stderr.write('%s: %s\n%s' % (sys.argv[0], msg, usage))
		return 2
	for o, a in opts:
		if o == '-l':
			fnfilter = os.path.basename
		if o == '-b':
			rmode = 'rb'
		if o == '-t':
			rmode = 'r'
		if o == '-s':
			bufsize = string.atoi(a)
	if not args: args = ['-']
	return sum(args, out)

if __name__ == '__main__' or __name__ == sys.argv[0]:
	sys.exit(main(sys.argv[1:], sys.stdout))