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()