#!/usr/bin/env python3
#
# OTP generator for RFC 2289 / OPIE
#

import sys
import os
import os.path
import stat
import getpass
import configparser
import random
import argparse
import hashlib
import subprocess
import platform
import re

OTP_PWD_FILE = '.otp'
DEFAULT_ALGO = 'otp-md5'  # if the PWD file doesn't specify
PASSWORD_LEN = 20


def otp_path():
  "Compute the full path to the OTP password storage file."
  return os.path.expanduser(os.path.join('~', OTP_PWD_FILE))


def load_passwords():
  "Load password from the storage file."
  path = otp_path()
  try:
    s = os.lstat(path)
  except OSError:
    return { }
  if not stat.S_ISREG(s.st_mode):
    print('ERROR: not a regular file:', path)
    sys.exit(1)
  # Only user read/write. Nobody else.
  if (s.st_mode & (stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)) != 0o600:
    print('ERROR: should be mode 0600:', path)
    sys.exit(1)

  pwds = { }
  for l in open(path).readlines():
    m = RE_STORAGE.match(l)
    if m:
      if m.group(2):
        algo = m.group(2).lower()
      else:
        algo = DEFAULT_ALGO
      pwds[m.group(3).encode()] = (m.group(4).encode(), algo)

  return pwds

RE_STORAGE = re.compile(r'((otp-\w+)\s+)?(\w+)\s+(.*)')


def add_password(seed, algo):
  "Construct a new password for a seed, and store/append it."
  pwd = new_password()
  path = otp_path()
  fd = os.open(path, os.O_RDWR | os.O_APPEND | os.O_CREAT, mode=0o600)
  os.write(fd, algo.encode() + b' ' + seed + b' ' + pwd + b'\n')
  os.close(fd)
  return pwd


def new_password():
  "Generate a password."
  # Stick to printable characters.
  return ''.join(chr(random.randint(33, 126)) for _ in range(PASSWORD_LEN)).encode()


def otp_md5(begin, seq):
  "Compute a 64-bit value through repeated hashing/folding."
  value = md5_iter_fold(begin)
  for _ in range(seq):
    value = md5_iter_fold(value)
  return value


def md5_iter_fold(value):
  "Iterate the hash, then fold into 64 bits."
  value = hashlib.md5(value).digest()  # 128 bit result
  return bytes(value[i] ^ value[i+8] for i in range(8))


def to_words(value):
  "Convert a 64-bit value into words, with a parity value."
  assert len(value) == 8  # 64 bits, in an 8-byte VALUE

  # Parity is the summation of each 2-bit pair. We will mask
  # out the lowest 2 bits, so no worries about higher bits.
  parity = sum((b >> 6) + (b >> 4) + (b >> 2) + b for b in value)

  # collapse the bytes into a single 64-bit integer; fold in parity bits
  i64 = int.from_bytes(value, byteorder='big')
  i64 = (i64 << 2) | (parity & 0x03)
  
  # and map that integer into words, using the spec's WORDS
  return [WORDS[(i64 >> (i*11)) & 0x7FF] for i in (5, 4, 3, 2, 1, 0)]


class RunTests(argparse.Action):
  "Run some basic tests on the code."

  def __call__(self, parser, namespace, values, option_string=None):
    # Test the to_words() function.
    value = b'\x9E\x87\x61\x34\xD9\x04\x99\xDD'
    words = ' '.join(to_words(value))
    assert words == MD5_TESTS[0][3]
    print('CORRECT:')
    print('  ', ' '.join('%02x%02x' % (value[i*2], value[i*2+1]) for i in range(4)), '--', words)

    # Run the tests from RFC 2289.
    print('COLUMNS:','Password', 'Seed', 'Count')
    for pwd, seed, seq, result in MD5_TESTS:
      value = otp_md5(seed.lower().encode() + pwd.encode(), seq)
      print('P:'+pwd, 'S:'+seed, 'C:'+str(seq))
      words = ' '.join(to_words(value))
      print('  ', ' '.join('%02x%02x' % (value[i*2], value[i*2+1]) for i in range(4)), '--', words)
      assert words == result

    sys.exit(0)


ALGOS = {
  'otp-md4': None,
  'otp-md5': otp_md5,
  'otp-sha1': None,
  }


def main():
  cmd = os.path.basename(sys.argv[0])
  if cmd in ALGOS:
    # switch to USAGE:
    #   $ ALGO SEQUENCE SEED
    # omit the prompt, and use argv
    algo = cmd
    seq = int(sys.argv[1])
    seed = sys.argv[2].lower().encode()
    # ignore any other argv
  else:
    parser = argparse.ArgumentParser(description='Compute OTP strings.')
    parser.add_argument('--test', help='Run the test suite.', nargs=0,
                        action=RunTests)
    parser.add_argument('args', nargs=argparse.REMAINDER)
  
    # Note: this may exit, if tests are run.
    parsed = parser.parse_args()

    parts = parsed.args
    if len(parts) < 2:  # the [ALGO] SEQUENCE SEED are not on cmdline. ask.
      line = input('Challenge? ')
      parts = line.split()

    if len(parts) == 2:
      # presumably: SEQUENCE SEED
      algo = None
      seq = int(parts[0])
      seed = parts[1].lower().encode()
    elif len(parts) >= 3:
      algo = parts[0]
      seq = int(parts[1])
      seed = parts[2].lower().encode()
      # ignore anything else on line (eg. "ext")
    else:
      print('ERROR: challenge must have: [ALGO] SEQUENCE SEED ...')
      sys.exit(1)

  # Load a dictionary mapping seeds to passwords.
  pwds = load_passwords()

  if seed not in pwds:
    print('Creating new password for:', seed.decode())
    if not algo:
      algo = DEFAULT_ALGO
      print('NOTE: using "%s" algorithm' % (algo,))
    pwd = add_password(seed, algo)
  else:
    old_algo = algo
    pwd, algo = pwds[seed]
    if old_algo and algo != old_algo:
      print('NOTE: switched to "%s" algorithm' % (algo,))

  processor = ALGOS.get(algo)
  assert processor, 'Unknown/unsupported algorithm: "%s"' % (algo,)

  value = processor(seed + pwd, seq)
  response = ' '.join(to_words(value))
  print('Response:', response)

  osname = platform.system()

  # Attempt to push this to the clipboard
  if osname == 'Darwin':
    result = subprocess.run(['pbcopy', '-pboard', 'general'], input=response.encode())
  else:
    result = subprocess.run(['xclip', '-selection', 'clipboard'], input=response.encode())
  if result.returncode == 0:
    print('NOTE: copied to clipboard')


# From RFC 2289, Appendix D.
WORDS = (
  "A", "ABE", "ACE", "ACT", "AD", "ADA", "ADD",
  "AGO", "AID", "AIM", "AIR", "ALL", "ALP", "AM", "AMY",
  "AN", "ANA", "AND", "ANN", "ANT", "ANY", "APE", "APS",
  "APT", "ARC", "ARE", "ARK", "ARM", "ART", "AS", "ASH",
  "ASK", "AT", "ATE", "AUG", "AUK", "AVE", "AWE", "AWK",
  "AWL", "AWN", "AX", "AYE", "BAD", "BAG", "BAH", "BAM",
  "BAN", "BAR", "BAT", "BAY", "BE", "BED", "BEE", "BEG",
  "BEN", "BET", "BEY", "BIB", "BID", "BIG", "BIN", "BIT",
  "BOB", "BOG", "BON", "BOO", "BOP", "BOW", "BOY", "BUB",
  "BUD", "BUG", "BUM", "BUN", "BUS", "BUT", "BUY", "BY",
  "BYE", "CAB", "CAL", "CAM", "CAN", "CAP", "CAR", "CAT",
  "CAW", "COD", "COG", "COL", "CON", "COO", "COP", "COT",
  "COW", "COY", "CRY", "CUB", "CUE", "CUP", "CUR", "CUT",
  "DAB", "DAD", "DAM", "DAN", "DAR", "DAY", "DEE", "DEL",
  "DEN", "DES", "DEW", "DID", "DIE", "DIG", "DIN", "DIP",
  "DO", "DOE", "DOG", "DON", "DOT", "DOW", "DRY", "DUB",
  "DUD", "DUE", "DUG", "DUN", "EAR", "EAT", "ED", "EEL",
  "EGG", "EGO", "ELI", "ELK", "ELM", "ELY", "EM", "END",
  "EST", "ETC", "EVA", "EVE", "EWE", "EYE", "FAD", "FAN",
  "FAR", "FAT", "FAY", "FED", "FEE", "FEW", "FIB", "FIG",
  "FIN", "FIR", "FIT", "FLO", "FLY", "FOE", "FOG", "FOR",
  "FRY", "FUM", "FUN", "FUR", "GAB", "GAD", "GAG", "GAL",
  "GAM", "GAP", "GAS", "GAY", "GEE", "GEL", "GEM", "GET",
  "GIG", "GIL", "GIN", "GO", "GOT", "GUM", "GUN", "GUS",
  "GUT", "GUY", "GYM", "GYP", "HA", "HAD", "HAL", "HAM",
  "HAN", "HAP", "HAS", "HAT", "HAW", "HAY", "HE", "HEM",
  "HEN", "HER", "HEW", "HEY", "HI", "HID", "HIM", "HIP",
  "HIS", "HIT", "HO", "HOB", "HOC", "HOE", "HOG", "HOP",
  "HOT", "HOW", "HUB", "HUE", "HUG", "HUH", "HUM", "HUT",
  "I", "ICY", "IDA", "IF", "IKE", "ILL", "INK", "INN",
  "IO", "ION", "IQ", "IRA", "IRE", "IRK", "IS", "IT",
  "ITS", "IVY", "JAB", "JAG", "JAM", "JAN", "JAR", "JAW",
  "JAY", "JET", "JIG", "JIM", "JO", "JOB", "JOE", "JOG",
  "JOT", "JOY", "JUG", "JUT", "KAY", "KEG", "KEN", "KEY",
  "KID", "KIM", "KIN", "KIT", "LA", "LAB", "LAC", "LAD",
  "LAG", "LAM", "LAP", "LAW", "LAY", "LEA", "LED", "LEE",
  "LEG", "LEN", "LEO", "LET", "LEW", "LID", "LIE", "LIN",
  "LIP", "LIT", "LO", "LOB", "LOG", "LOP", "LOS", "LOT",
  "LOU", "LOW", "LOY", "LUG", "LYE", "MA", "MAC", "MAD",
  "MAE", "MAN", "MAO", "MAP", "MAT", "MAW", "MAY", "ME",
  "MEG", "MEL", "MEN", "MET", "MEW", "MID", "MIN", "MIT",
  "MOB", "MOD", "MOE", "MOO", "MOP", "MOS", "MOT", "MOW",
  "MUD", "MUG", "MUM", "MY", "NAB", "NAG", "NAN", "NAP",
  "NAT", "NAY", "NE", "NED", "NEE", "NET", "NEW", "NIB",
  "NIL", "NIP", "NIT", "NO", "NOB", "NOD", "NON", "NOR",
  "NOT", "NOV", "NOW", "NU", "NUN", "NUT", "O", "OAF",
  "OAK", "OAR", "OAT", "ODD", "ODE", "OF", "OFF", "OFT",
  "OH", "OIL", "OK", "OLD", "ON", "ONE", "OR", "ORB",
  "ORE", "ORR", "OS", "OTT", "OUR", "OUT", "OVA", "OW",
  "OWE", "OWL", "OWN", "OX", "PA", "PAD", "PAL", "PAM",
  "PAN", "PAP", "PAR", "PAT", "PAW", "PAY", "PEA", "PEG",
  "PEN", "PEP", "PER", "PET", "PEW", "PHI", "PI", "PIE",
  "PIN", "PIT", "PLY", "PO", "POD", "POE", "POP", "POT",
  "POW", "PRO", "PRY", "PUB", "PUG", "PUN", "PUP", "PUT",
  "QUO", "RAG", "RAM", "RAN", "RAP", "RAT", "RAW", "RAY",
  "REB", "RED", "REP", "RET", "RIB", "RID", "RIG", "RIM",
  "RIO", "RIP", "ROB", "ROD", "ROE", "RON", "ROT", "ROW",
  "ROY", "RUB", "RUE", "RUG", "RUM", "RUN", "RYE", "SAC",
  "SAD", "SAG", "SAL", "SAM", "SAN", "SAP", "SAT", "SAW",
  "SAY", "SEA", "SEC", "SEE", "SEN", "SET", "SEW", "SHE",
  "SHY", "SIN", "SIP", "SIR", "SIS", "SIT", "SKI", "SKY",
  "SLY", "SO", "SOB", "SOD", "SON", "SOP", "SOW", "SOY",
  "SPA", "SPY", "SUB", "SUD", "SUE", "SUM", "SUN", "SUP",
  "TAB", "TAD", "TAG", "TAN", "TAP", "TAR", "TEA", "TED",
  "TEE", "TEN", "THE", "THY", "TIC", "TIE", "TIM", "TIN",
  "TIP", "TO", "TOE", "TOG", "TOM", "TON", "TOO", "TOP",
  "TOW", "TOY", "TRY", "TUB", "TUG", "TUM", "TUN", "TWO",
  "UN", "UP", "US", "USE", "VAN", "VAT", "VET", "VIE",
  "WAD", "WAG", "WAR", "WAS", "WAY", "WE", "WEB", "WED",
  "WEE", "WET", "WHO", "WHY", "WIN", "WIT", "WOK", "WON",
  "WOO", "WOW", "WRY", "WU", "YAM", "YAP", "YAW", "YE",
  "YEA", "YES", "YET", "YOU", "ABED", "ABEL", "ABET", "ABLE",
  "ABUT", "ACHE", "ACID", "ACME", "ACRE", "ACTA", "ACTS", "ADAM",
  "ADDS", "ADEN", "AFAR", "AFRO", "AGEE", "AHEM", "AHOY", "AIDA",
  "AIDE", "AIDS", "AIRY", "AJAR", "AKIN", "ALAN", "ALEC", "ALGA",
  "ALIA", "ALLY", "ALMA", "ALOE", "ALSO", "ALTO", "ALUM", "ALVA",
  "AMEN", "AMES", "AMID", "AMMO", "AMOK", "AMOS", "AMRA", "ANDY",
  "ANEW", "ANNA", "ANNE", "ANTE", "ANTI", "AQUA", "ARAB", "ARCH",
  "AREA", "ARGO", "ARID", "ARMY", "ARTS", "ARTY", "ASIA", "ASKS",
  "ATOM", "AUNT", "AURA", "AUTO", "AVER", "AVID", "AVIS", "AVON",
  "AVOW", "AWAY", "AWRY", "BABE", "BABY", "BACH", "BACK", "BADE",
  "BAIL", "BAIT", "BAKE", "BALD", "BALE", "BALI", "BALK", "BALL",
  "BALM", "BAND", "BANE", "BANG", "BANK", "BARB", "BARD", "BARE",
  "BARK", "BARN", "BARR", "BASE", "BASH", "BASK", "BASS", "BATE",
  "BATH", "BAWD", "BAWL", "BEAD", "BEAK", "BEAM", "BEAN", "BEAR",
  "BEAT", "BEAU", "BECK", "BEEF", "BEEN", "BEER", "BEET", "BELA",
  "BELL", "BELT", "BEND", "BENT", "BERG", "BERN", "BERT", "BESS",
  "BEST", "BETA", "BETH", "BHOY", "BIAS", "BIDE", "BIEN", "BILE",
  "BILK", "BILL", "BIND", "BING", "BIRD", "BITE", "BITS", "BLAB",
  "BLAT", "BLED", "BLEW", "BLOB", "BLOC", "BLOT", "BLOW", "BLUE",
  "BLUM", "BLUR", "BOAR", "BOAT", "BOCA", "BOCK", "BODE", "BODY",
  "BOGY", "BOHR", "BOIL", "BOLD", "BOLO", "BOLT", "BOMB", "BONA",
  "BOND", "BONE", "BONG", "BONN", "BONY", "BOOK", "BOOM", "BOON",
  "BOOT", "BORE", "BORG", "BORN", "BOSE", "BOSS", "BOTH", "BOUT",
  "BOWL", "BOYD", "BRAD", "BRAE", "BRAG", "BRAN", "BRAY", "BRED",
  "BREW", "BRIG", "BRIM", "BROW", "BUCK", "BUDD", "BUFF", "BULB",
  "BULK", "BULL", "BUNK", "BUNT", "BUOY", "BURG", "BURL", "BURN",
  "BURR", "BURT", "BURY", "BUSH", "BUSS", "BUST", "BUSY", "BYTE",
  "CADY", "CAFE", "CAGE", "CAIN", "CAKE", "CALF", "CALL", "CALM",
  "CAME", "CANE", "CANT", "CARD", "CARE", "CARL", "CARR", "CART",
  "CASE", "CASH", "CASK", "CAST", "CAVE", "CEIL", "CELL", "CENT",
  "CERN", "CHAD", "CHAR", "CHAT", "CHAW", "CHEF", "CHEN", "CHEW",
  "CHIC", "CHIN", "CHOU", "CHOW", "CHUB", "CHUG", "CHUM", "CITE",
  "CITY", "CLAD", "CLAM", "CLAN", "CLAW", "CLAY", "CLOD", "CLOG",
  "CLOT", "CLUB", "CLUE", "COAL", "COAT", "COCA", "COCK", "COCO",
  "CODA", "CODE", "CODY", "COED", "COIL", "COIN", "COKE", "COLA",
  "COLD", "COLT", "COMA", "COMB", "COME", "COOK", "COOL", "COON",
  "COOT", "CORD", "CORE", "CORK", "CORN", "COST", "COVE", "COWL",
  "CRAB", "CRAG", "CRAM", "CRAY", "CREW", "CRIB", "CROW", "CRUD",
  "CUBA", "CUBE", "CUFF", "CULL", "CULT", "CUNY", "CURB", "CURD",
  "CURE", "CURL", "CURT", "CUTS", "DADE", "DALE", "DAME", "DANA",
  "DANE", "DANG", "DANK", "DARE", "DARK", "DARN", "DART", "DASH",
  "DATA", "DATE", "DAVE", "DAVY", "DAWN", "DAYS", "DEAD", "DEAF",
  "DEAL", "DEAN", "DEAR", "DEBT", "DECK", "DEED", "DEEM", "DEER",
  "DEFT", "DEFY", "DELL", "DENT", "DENY", "DESK", "DIAL", "DICE",
  "DIED", "DIET", "DIME", "DINE", "DING", "DINT", "DIRE", "DIRT",
  "DISC", "DISH", "DISK", "DIVE", "DOCK", "DOES", "DOLE", "DOLL",
  "DOLT", "DOME", "DONE", "DOOM", "DOOR", "DORA", "DOSE", "DOTE",
  "DOUG", "DOUR", "DOVE", "DOWN", "DRAB", "DRAG", "DRAM", "DRAW",
  "DREW", "DRUB", "DRUG", "DRUM", "DUAL", "DUCK", "DUCT", "DUEL",
  "DUET", "DUKE", "DULL", "DUMB", "DUNE", "DUNK", "DUSK", "DUST",
  "DUTY", "EACH", "EARL", "EARN", "EASE", "EAST", "EASY", "EBEN",
  "ECHO", "EDDY", "EDEN", "EDGE", "EDGY", "EDIT", "EDNA", "EGAN",
  "ELAN", "ELBA", "ELLA", "ELSE", "EMIL", "EMIT", "EMMA", "ENDS",
  "ERIC", "EROS", "EVEN", "EVER", "EVIL", "EYED", "FACE", "FACT",
  "FADE", "FAIL", "FAIN", "FAIR", "FAKE", "FALL", "FAME", "FANG",
  "FARM", "FAST", "FATE", "FAWN", "FEAR", "FEAT", "FEED", "FEEL",
  "FEET", "FELL", "FELT", "FEND", "FERN", "FEST", "FEUD", "FIEF",
  "FIGS", "FILE", "FILL", "FILM", "FIND", "FINE", "FINK", "FIRE",
  "FIRM", "FISH", "FISK", "FIST", "FITS", "FIVE", "FLAG", "FLAK",
  "FLAM", "FLAT", "FLAW", "FLEA", "FLED", "FLEW", "FLIT", "FLOC",
  "FLOG", "FLOW", "FLUB", "FLUE", "FOAL", "FOAM", "FOGY", "FOIL",
  "FOLD", "FOLK", "FOND", "FONT", "FOOD", "FOOL", "FOOT", "FORD",
  "FORE", "FORK", "FORM", "FORT", "FOSS", "FOUL", "FOUR", "FOWL",
  "FRAU", "FRAY", "FRED", "FREE", "FRET", "FREY", "FROG", "FROM",
  "FUEL", "FULL", "FUME", "FUND", "FUNK", "FURY", "FUSE", "FUSS",
  "GAFF", "GAGE", "GAIL", "GAIN", "GAIT", "GALA", "GALE", "GALL",
  "GALT", "GAME", "GANG", "GARB", "GARY", "GASH", "GATE", "GAUL",
  "GAUR", "GAVE", "GAWK", "GEAR", "GELD", "GENE", "GENT", "GERM",
  "GETS", "GIBE", "GIFT", "GILD", "GILL", "GILT", "GINA", "GIRD",
  "GIRL", "GIST", "GIVE", "GLAD", "GLEE", "GLEN", "GLIB", "GLOB",
  "GLOM", "GLOW", "GLUE", "GLUM", "GLUT", "GOAD", "GOAL", "GOAT",
  "GOER", "GOES", "GOLD", "GOLF", "GONE", "GONG", "GOOD", "GOOF",
  "GORE", "GORY", "GOSH", "GOUT", "GOWN", "GRAB", "GRAD", "GRAY",
  "GREG", "GREW", "GREY", "GRID", "GRIM", "GRIN", "GRIT", "GROW",
  "GRUB", "GULF", "GULL", "GUNK", "GURU", "GUSH", "GUST", "GWEN",
  "GWYN", "HAAG", "HAAS", "HACK", "HAIL", "HAIR", "HALE", "HALF",
  "HALL", "HALO", "HALT", "HAND", "HANG", "HANK", "HANS", "HARD",
  "HARK", "HARM", "HART", "HASH", "HAST", "HATE", "HATH", "HAUL",
  "HAVE", "HAWK", "HAYS", "HEAD", "HEAL", "HEAR", "HEAT", "HEBE",
  "HECK", "HEED", "HEEL", "HEFT", "HELD", "HELL", "HELM", "HERB",
  "HERD", "HERE", "HERO", "HERS", "HESS", "HEWN", "HICK", "HIDE",
  "HIGH", "HIKE", "HILL", "HILT", "HIND", "HINT", "HIRE", "HISS",
  "HIVE", "HOBO", "HOCK", "HOFF", "HOLD", "HOLE", "HOLM", "HOLT",
  "HOME", "HONE", "HONK", "HOOD", "HOOF", "HOOK", "HOOT", "HORN",
  "HOSE", "HOST", "HOUR", "HOVE", "HOWE", "HOWL", "HOYT", "HUCK",
  "HUED", "HUFF", "HUGE", "HUGH", "HUGO", "HULK", "HULL", "HUNK",
  "HUNT", "HURD", "HURL", "HURT", "HUSH", "HYDE", "HYMN", "IBIS",
  "ICON", "IDEA", "IDLE", "IFFY", "INCA", "INCH", "INTO", "IONS",
  "IOTA", "IOWA", "IRIS", "IRMA", "IRON", "ISLE", "ITCH", "ITEM",
  "IVAN", "JACK", "JADE", "JAIL", "JAKE", "JANE", "JAVA", "JEAN",
  "JEFF", "JERK", "JESS", "JEST", "JIBE", "JILL", "JILT", "JIVE",
  "JOAN", "JOBS", "JOCK", "JOEL", "JOEY", "JOHN", "JOIN", "JOKE",
  "JOLT", "JOVE", "JUDD", "JUDE", "JUDO", "JUDY", "JUJU", "JUKE",
  "JULY", "JUNE", "JUNK", "JUNO", "JURY", "JUST", "JUTE", "KAHN",
  "KALE", "KANE", "KANT", "KARL", "KATE", "KEEL", "KEEN", "KENO",
  "KENT", "KERN", "KERR", "KEYS", "KICK", "KILL", "KIND", "KING",
  "KIRK", "KISS", "KITE", "KLAN", "KNEE", "KNEW", "KNIT", "KNOB",
  "KNOT", "KNOW", "KOCH", "KONG", "KUDO", "KURD", "KURT", "KYLE",
  "LACE", "LACK", "LACY", "LADY", "LAID", "LAIN", "LAIR", "LAKE",
  "LAMB", "LAME", "LAND", "LANE", "LANG", "LARD", "LARK", "LASS",
  "LAST", "LATE", "LAUD", "LAVA", "LAWN", "LAWS", "LAYS", "LEAD",
  "LEAF", "LEAK", "LEAN", "LEAR", "LEEK", "LEER", "LEFT", "LEND",
  "LENS", "LENT", "LEON", "LESK", "LESS", "LEST", "LETS", "LIAR",
  "LICE", "LICK", "LIED", "LIEN", "LIES", "LIEU", "LIFE", "LIFT",
  "LIKE", "LILA", "LILT", "LILY", "LIMA", "LIMB", "LIME", "LIND",
  "LINE", "LINK", "LINT", "LION", "LISA", "LIST", "LIVE", "LOAD",
  "LOAF", "LOAM", "LOAN", "LOCK", "LOFT", "LOGE", "LOIS", "LOLA",
  "LONE", "LONG", "LOOK", "LOON", "LOOT", "LORD", "LORE", "LOSE",
  "LOSS", "LOST", "LOUD", "LOVE", "LOWE", "LUCK", "LUCY", "LUGE",
  "LUKE", "LULU", "LUND", "LUNG", "LURA", "LURE", "LURK", "LUSH",
  "LUST", "LYLE", "LYNN", "LYON", "LYRA", "MACE", "MADE", "MAGI",
  "MAID", "MAIL", "MAIN", "MAKE", "MALE", "MALI", "MALL", "MALT",
  "MANA", "MANN", "MANY", "MARC", "MARE", "MARK", "MARS", "MART",
  "MARY", "MASH", "MASK", "MASS", "MAST", "MATE", "MATH", "MAUL",
  "MAYO", "MEAD", "MEAL", "MEAN", "MEAT", "MEEK", "MEET", "MELD",
  "MELT", "MEMO", "MEND", "MENU", "MERT", "MESH", "MESS", "MICE",
  "MIKE", "MILD", "MILE", "MILK", "MILL", "MILT", "MIMI", "MIND",
  "MINE", "MINI", "MINK", "MINT", "MIRE", "MISS", "MIST", "MITE",
  "MITT", "MOAN", "MOAT", "MOCK", "MODE", "MOLD", "MOLE", "MOLL",
  "MOLT", "MONA", "MONK", "MONT", "MOOD", "MOON", "MOOR", "MOOT",
  "MORE", "MORN", "MORT", "MOSS", "MOST", "MOTH", "MOVE", "MUCH",
  "MUCK", "MUDD", "MUFF", "MULE", "MULL", "MURK", "MUSH", "MUST",
  "MUTE", "MUTT", "MYRA", "MYTH", "NAGY", "NAIL", "NAIR", "NAME",
  "NARY", "NASH", "NAVE", "NAVY", "NEAL", "NEAR", "NEAT", "NECK",
  "NEED", "NEIL", "NELL", "NEON", "NERO", "NESS", "NEST", "NEWS",
  "NEWT", "NIBS", "NICE", "NICK", "NILE", "NINA", "NINE", "NOAH",
  "NODE", "NOEL", "NOLL", "NONE", "NOOK", "NOON", "NORM", "NOSE",
  "NOTE", "NOUN", "NOVA", "NUDE", "NULL", "NUMB", "OATH", "OBEY",
  "OBOE", "ODIN", "OHIO", "OILY", "OINT", "OKAY", "OLAF", "OLDY",
  "OLGA", "OLIN", "OMAN", "OMEN", "OMIT", "ONCE", "ONES", "ONLY",
  "ONTO", "ONUS", "ORAL", "ORGY", "OSLO", "OTIS", "OTTO", "OUCH",
  "OUST", "OUTS", "OVAL", "OVEN", "OVER", "OWLY", "OWNS", "QUAD",
  "QUIT", "QUOD", "RACE", "RACK", "RACY", "RAFT", "RAGE", "RAID",
  "RAIL", "RAIN", "RAKE", "RANK", "RANT", "RARE", "RASH", "RATE",
  "RAVE", "RAYS", "READ", "REAL", "REAM", "REAR", "RECK", "REED",
  "REEF", "REEK", "REEL", "REID", "REIN", "RENA", "REND", "RENT",
  "REST", "RICE", "RICH", "RICK", "RIDE", "RIFT", "RILL", "RIME",
  "RING", "RINK", "RISE", "RISK", "RITE", "ROAD", "ROAM", "ROAR",
  "ROBE", "ROCK", "RODE", "ROIL", "ROLL", "ROME", "ROOD", "ROOF",
  "ROOK", "ROOM", "ROOT", "ROSA", "ROSE", "ROSS", "ROSY", "ROTH",
  "ROUT", "ROVE", "ROWE", "ROWS", "RUBE", "RUBY", "RUDE", "RUDY",
  "RUIN", "RULE", "RUNG", "RUNS", "RUNT", "RUSE", "RUSH", "RUSK",
  "RUSS", "RUST", "RUTH", "SACK", "SAFE", "SAGE", "SAID", "SAIL",
  "SALE", "SALK", "SALT", "SAME", "SAND", "SANE", "SANG", "SANK",
  "SARA", "SAUL", "SAVE", "SAYS", "SCAN", "SCAR", "SCAT", "SCOT",
  "SEAL", "SEAM", "SEAR", "SEAT", "SEED", "SEEK", "SEEM", "SEEN",
  "SEES", "SELF", "SELL", "SEND", "SENT", "SETS", "SEWN", "SHAG",
  "SHAM", "SHAW", "SHAY", "SHED", "SHIM", "SHIN", "SHOD", "SHOE",
  "SHOT", "SHOW", "SHUN", "SHUT", "SICK", "SIDE", "SIFT", "SIGH",
  "SIGN", "SILK", "SILL", "SILO", "SILT", "SINE", "SING", "SINK",
  "SIRE", "SITE", "SITS", "SITU", "SKAT", "SKEW", "SKID", "SKIM",
  "SKIN", "SKIT", "SLAB", "SLAM", "SLAT", "SLAY", "SLED", "SLEW",
  "SLID", "SLIM", "SLIT", "SLOB", "SLOG", "SLOT", "SLOW", "SLUG",
  "SLUM", "SLUR", "SMOG", "SMUG", "SNAG", "SNOB", "SNOW", "SNUB",
  "SNUG", "SOAK", "SOAR", "SOCK", "SODA", "SOFA", "SOFT", "SOIL",
  "SOLD", "SOME", "SONG", "SOON", "SOOT", "SORE", "SORT", "SOUL",
  "SOUR", "SOWN", "STAB", "STAG", "STAN", "STAR", "STAY", "STEM",
  "STEW", "STIR", "STOW", "STUB", "STUN", "SUCH", "SUDS", "SUIT",
  "SULK", "SUMS", "SUNG", "SUNK", "SURE", "SURF", "SWAB", "SWAG",
  "SWAM", "SWAN", "SWAT", "SWAY", "SWIM", "SWUM", "TACK", "TACT",
  "TAIL", "TAKE", "TALE", "TALK", "TALL", "TANK", "TASK", "TATE",
  "TAUT", "TEAL", "TEAM", "TEAR", "TECH", "TEEM", "TEEN", "TEET",
  "TELL", "TEND", "TENT", "TERM", "TERN", "TESS", "TEST", "THAN",
  "THAT", "THEE", "THEM", "THEN", "THEY", "THIN", "THIS", "THUD",
  "THUG", "TICK", "TIDE", "TIDY", "TIED", "TIER", "TILE", "TILL",
  "TILT", "TIME", "TINA", "TINE", "TINT", "TINY", "TIRE", "TOAD",
  "TOGO", "TOIL", "TOLD", "TOLL", "TONE", "TONG", "TONY", "TOOK",
  "TOOL", "TOOT", "TORE", "TORN", "TOTE", "TOUR", "TOUT", "TOWN",
  "TRAG", "TRAM", "TRAY", "TREE", "TREK", "TRIG", "TRIM", "TRIO",
  "TROD", "TROT", "TROY", "TRUE", "TUBA", "TUBE", "TUCK", "TUFT",
  "TUNA", "TUNE", "TUNG", "TURF", "TURN", "TUSK", "TWIG", "TWIN",
  "TWIT", "ULAN", "UNIT", "URGE", "USED", "USER", "USES", "UTAH",
  "VAIL", "VAIN", "VALE", "VARY", "VASE", "VAST", "VEAL", "VEDA",
  "VEIL", "VEIN", "VEND", "VENT", "VERB", "VERY", "VETO", "VICE",
  "VIEW", "VINE", "VISE", "VOID", "VOLT", "VOTE", "WACK", "WADE",
  "WAGE", "WAIL", "WAIT", "WAKE", "WALE", "WALK", "WALL", "WALT",
  "WAND", "WANE", "WANG", "WANT", "WARD", "WARM", "WARN", "WART",
  "WASH", "WAST", "WATS", "WATT", "WAVE", "WAVY", "WAYS", "WEAK",
  "WEAL", "WEAN", "WEAR", "WEED", "WEEK", "WEIR", "WELD", "WELL",
  "WELT", "WENT", "WERE", "WERT", "WEST", "WHAM", "WHAT", "WHEE",
  "WHEN", "WHET", "WHOA", "WHOM", "WICK", "WIFE", "WILD", "WILL",
  "WIND", "WINE", "WING", "WINK", "WINO", "WIRE", "WISE", "WISH",
  "WITH", "WOLF", "WONT", "WOOD", "WOOL", "WORD", "WORE", "WORK",
  "WORM", "WORN", "WOVE", "WRIT", "WYNN", "YALE", "YANG", "YANK",
  "YARD", "YARN", "YAWL", "YAWN", "YEAH", "YEAR", "YELL", "YOGA",
  "YOKE",
)


# From RFC 2289, Appendix C.
MD5_TESTS = (
  ("This is a test.", "TeSt", 0, "INCH SEA ANNE LONG AHEM TOUR"),
  ("This is a test.", "TeSt", 1, "EASE OIL FUM CURE AWRY AVIS"),
  ("This is a test.", "TeSt", 99, "BAIL TUFT BITS GANG CHEF THY"),
  ("AbCdEfGhIjK", "alpha1", 0, "FULL PEW DOWN ONCE MORT ARC"),
  ("AbCdEfGhIjK", "alpha1", 1, "FACT HOOF AT FIST SITE KENT"),
  ("AbCdEfGhIjK", "alpha1", 99, "BODE HOP JAKE STOW JUT RAP"),
  ("OTP's are good", "correct", 0, "ULAN NEW ARMY FUSE SUIT EYED"),
  ("OTP's are good", "correct", 1, "SKIM CULT LOB SLAM POE HOWL"),
  ("OTP's are good", "correct", 99, "LONG IVY JULY AJAR BOND LEE"),
)


if __name__ == '__main__':
  main()