Home | History | Annotate | Download | only in md5test 
      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()
    124