Demo/md5test/md5driver.py

   1 import string
   2 import md5
   3 from sys import argv
   4
   5 def MDPrint(str):
   6     outstr = ''
   7     for i in str:
   8         o = ord(i)
   9         outstr = (outstr
  10                   + string.hexdigits[(o >> 4) & 0xF]
  11                   + string.hexdigits[o & 0xF])
  12     print outstr,
  13
  14
  15 from time import time
  16
  17 def makestr(start, end):
  18     result = ''
  19     for i in range(start, end + 1):
  20         result = result + chr(i)
  21
  22     return result
  23
  24
  25 def MDTimeTrial():
  26     TEST_BLOCK_SIZE = 1000
  27     TEST_BLOCKS = 10000
  28
  29     TEST_BYTES = TEST_BLOCK_SIZE * TEST_BLOCKS
  30
  31     # initialize test data, need temporary string filler
  32
  33     filsiz = 1 << 8
  34     filler = makestr(0, filsiz-1)
  35     data = filler * (TEST_BLOCK_SIZE // filsiz)
  36     data = data + filler[:(TEST_BLOCK_SIZE % filsiz)]
  37
  38     del filsiz, filler
  39
  40
  41     # start timer
  42     print 'MD5 time trial. Processing', TEST_BYTES, 'characters...'
  43     t1 = time()
  44
  45     mdContext = md5.new()
  46
  47     for i in range(TEST_BLOCKS):
  48         mdContext.update(data)
  49
  50     str = mdContext.digest()
  51     t2 = time()
  52
  53     MDPrint(str)
  54     print 'is digest of test input.'
  55     print 'Seconds to process test input:', t2 - t1
  56     print 'Characters processed per second:', TEST_BYTES / (t2 - t1)
  57
  58
  59 def MDString(str):
  60     MDPrint(md5.new(str).digest())
  61     print '"' + str + '"'
  62
  63
  64 def MDFile(filename):
  65     f = open(filename, 'rb')
  66     mdContext = md5.new()
  67
  68     while 1:
  69         data = f.read(1024)
  70         if not data:
  71             break
  72         mdContext.update(data)
  73
  74     MDPrint(mdContext.digest())
  75     print filename
  76
  77
  78 import sys
  79
  80 def MDFilter():
  81     mdContext = md5.new()
  82
  83     while 1:
  84         data = sys.stdin.read(16)
  85         if not data:
  86             break
  87         mdContext.update(data)
  88
  89     MDPrint(mdContext.digest())
  90     print
  91
  92
  93 def MDTestSuite():
  94     print 'MD5 test suite results:'
  95     MDString('')
  96     MDString('a')
  97     MDString('abc')
  98     MDString('message digest')
  99     MDString(makestr(ord('a'), ord('z')))
 100     MDString(makestr(ord('A'), ord('Z'))
 101               + makestr(ord('a'), ord('z'))
 102               + makestr(ord('0'), ord('9')))
 103     MDString((makestr(ord('1'), ord('9')) + '0') * 8)
 104
 105     # Contents of file foo are "abc"
 106     MDFile('foo')
 107
 108
 109 # I don't wanna use getopt(), since I want to use the same i/f...
 110 def main():
 111     if len(argv) == 1:
 112         MDFilter()
 113     for arg in argv[1:]:
 114         if arg[:2] == '-s':
 115             MDString(arg[2:])
 116         elif arg == '-t':
 117             MDTimeTrial()
 118         elif arg == '-x':
 119             MDTestSuite()
 120         else:
 121             MDFile(arg)
 122
 123 main()