import string

import md5

from sys import argv

def MDPrint(str):

    outstr = ''

    for i in str:

        o = ord(i)

        outstr = (outstr

                  + string.hexdigits[(o >> 4) & 0xF]

                  + string.hexdigits[o & 0xF])

    print outstr,

from time import time

def makestr(start, end):

    result = ''

    for i in range(start, end + 1):

        result = result + chr(i)

    return result

def MDTimeTrial():

    TEST_BLOCK_SIZE = 1000

    TEST_BLOCKS = 10000

    TEST_BYTES = TEST_BLOCK_SIZE * TEST_BLOCKS

    # initialize test data, need temporary string filler

    filsiz = 1 << 8

    filler = makestr(0, filsiz-1)

    data = filler * (TEST_BLOCK_SIZE // filsiz)

    data = data + filler[:(TEST_BLOCK_SIZE % filsiz)]

    del filsiz, filler

    # start timer

    print 'MD5 time trial. Processing', TEST_BYTES, 'characters...'

    t1 = time()

    mdContext = md5.new()

    for i in range(TEST_BLOCKS):

        mdContext.update(data)

    str = mdContext.digest()

    t2 = time()

    MDPrint(str)

    print 'is digest of test input.'

    print 'Seconds to process test input:', t2 - t1

    print 'Characters processed per second:', TEST_BYTES / (t2 - t1)

def MDString(str):

    MDPrint(md5.new(str).digest())

    print '"' + str + '"'

def MDFile(filename):

    f = open(filename, 'rb')

    mdContext = md5.new()

    while 1:

        data = f.read(1024)

        if not data:

            break

        mdContext.update(data)

    MDPrint(mdContext.digest())

    print filename

import sys

def MDFilter():

    mdContext = md5.new()

    while 1:

        data = sys.stdin.read(16)

        if not data:

            break

        mdContext.update(data)

    MDPrint(mdContext.digest())

    print

def MDTestSuite():

    print 'MD5 test suite results:'

    MDString('')

    MDString('a')

    MDString('abc')

    MDString('message digest')

    MDString(makestr(ord('a'), ord('z')))

    MDString(makestr(ord('A'), ord('Z'))

              + makestr(ord('a'), ord('z'))

              + makestr(ord('0'), ord('9')))

    MDString((makestr(ord('1'), ord('9')) + '0') * 8)

    # Contents of file foo are "abc"

    MDFile('foo')

# I don't wanna use getopt(), since I want to use the same i/f...

def main():

    if len(argv) == 1:

        MDFilter()

    for arg in argv[1:]:

        if arg[:2] == '-s':

            MDString(arg[2:])

        elif arg == '-t':

            MDTimeTrial()

        elif arg == '-x':

            MDTestSuite()

        else:

            MDFile(arg)

main()