wordlist_optimizer.py

#!/usr/bin/env python3

import sys
import os
import subprocess
import shutil
import pathlib

# Help
def usage():
  print("usage: python %s <input file list> <output directory>" % sys.argv[0])

def lineCount(file):
  try:
    outFile = open(file)
  except:
    return 0

  count = 0
  for line in outFile:
    count = count + 1
  return count

# Main guts
def main():
  try:
    if not os.path.isfile(sys.argv[1]):
      print('{0} is not a valid file.\n'.format(sys.argv[1]))
      sys.exit()
    if not os.path.isdir(sys.argv[2]):
      create_directory = input('{0} is not a directory. Do you want to create it? (Y or N)'.format(sys.argv[2]))
      if create_directory.upper() == 'Y':
        try:
          pathlib.Path(sys.argv[2]).mkdir(parents=True, exist_ok=True)
        except PermissionError:
          print('You do not have the correct permissions to receate the directory. Please try a different path or create manually')
          sys.exit()
      else:
        print('Please specify a valid directory and try again')
        sys.exit()
    input_list = open(sys.argv[1], "r")
    destination = sys.argv[2]
  except IndexError:
    usage()
    sys.exit()

  if sys.platform == 'darwin':
    splitlen_bin = "hashcat-utils/bin/splitlen.app"
    rli_bin = "hashcat-utils/bin/rli.app"
  else:
    splitlen_bin = "hashcat-utils/bin/splitlen.bin"
    rli_bin = "hashcat-utils/bin/rli.bin"

  # Get list of wordlists from <input file list> argument
  for wordlist in input_list:
    print(wordlist.strip())
    
    # Parse wordlists by password length into "optimized" <output directory>
    if len(os.listdir(destination)) == 0:
      splitlenProcess = subprocess.Popen("%s %s < %s" % (splitlen_bin, destination, wordlist), shell=True).wait()
    else:
      if not os.path.isdir("/tmp/splitlen"):
        os.mkdir("/tmp/splitlen")
      splitlenProcess = subprocess.Popen("%s /tmp/splitlen < %s" % (splitlen_bin, wordlist), shell=True).wait()

      # Copy unique passwords into "optimized" <output directory>
      for file in os.listdir("/tmp/splitlen"):
        if not os.path.isfile(destination + "/" + file):
          shutil.copyfile("/tmp/splitlen/" + file, destination + "/" + file)
        else:
          rliProcess = subprocess.Popen("%s /tmp/splitlen/%s /tmp/splitlen.out %s/%s" % (rli_bin, file, destination, file), shell=True).wait()
          if lineCount("/tmp/splitlen.out") > 0:
            destination_file = open(destination + "/" + file, "a")
            splitlen_file = open("/tmp/splitlen.out", "r")
            destination_file.write(splitlen_file.read())
            destination_file.close()
            splitlen_file.close()
            
    # Clean Up
    if os.path.isdir("/tmp/splitlen"):
      shutil.rmtree('/tmp/splitlen')
    if os.path.isfile("/tmp/splitlen.out"):
      os.remove("/tmp/splitlen.out")


# Standard boilerplate to call the main() function
if __name__ == '__main__':
  main()