class Client: def importScripts(self): self.decEncHelper = DecodingEncodingHelper() self.inputHandler = InputHandler(self.output) self.fileHelper = FileHelper(self.output) self.guiHelper = GUIHelper(self.output) def setConfig(self): Config = self.fileHelper.getConfig() self.ipV4 = Config.ip self.port = Config.port def inizializeClient(self, username, password): self.clientObject = ClientObject(username, None, self.ipV4, self.port, "Welcome_Channel") self.connected = False self.password = password def tryConnect(self): try: self.clientObject.socketObject = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self.clientObject.socketObject.connect((self.clientObject.ip, self.clientObject.port)) threading.Thread(target=ServerHandler,args=[self.clientObject,self.windows]).start() self.clientObject.socketObject.sendall(self.decEncHelper.stringToBytes("011" + self.clientObject.username + ":" + self.password)) self.connected = True except: self.connected = False def sendInput(self, message): if self.connected: if str(message).startswith("/"): self.inputHandler.handleInput(str(message[1:]), self.clientObject) else: self.output.append("you: " + message) try: self.clientObject.socketObject.sendall(self.decEncHelper.stringToBytes("001" + message)) except: self.connected = False else: self.guiHelper.printOutput("not connected") def __init__(self, username, password, windows): #Imports self.windows = windows self.output = self.windows[0].output self.importScripts() #Config self.setConfig() #Client initializations self.inizializeClient(username, password) #Client trying to establish a connection self.tryConnect()
class InputHandler: commandList = list() def importScripts(self): self.decEncHelper = DecodingEncodingHelper() self.channelManager = ChannelManager() self.clientManager = ClientManager() self.fileHelper = FileHelper() self.logHelper = LogHelper() self.mysqlHelper = MysqlHelper() def createCommand(self, name, syntax, arguments, description): command = Command(name, syntax, arguments, description) self.commandList.append(command) return command def initializeCommands(self): self.cmdListClients = self.createCommand( "listClients", "/listClients", "NONE", "Lists all connected clients with their name, ip and channel their in." ) self.cmdClear = self.createCommand("Clear", "/clear", "NONE", "Clears your interpreter console.") self.cmdHelp = self.createCommand( "Help", "/help", "NONE", "Shows a list of available commands.") self.cmdKick = self.createCommand( "Kick", "/kick <name/ip>", "<NAME/IP>", "Kicks the given IP from the server.") self.cmdBan = self.createCommand( "Ban", "/ban <name/ip> <time>", "<NAME/IP> <TIME>", "Bans the specified client for the given amount of time in minutes." ) self.cmdMonitorMode = self.createCommand("monitorMode", "/monitorMode", "NONE", "Switches to monitor mode.") self.cmdChangeRank = self.createCommand( "changeRank", "/changeRank <name/ip> <rank>", "<NAME/IP> <RANK>", "Changes the rank of the given client.") self.cmdListChannel = self.createCommand( "listChannel", "/listChannel", "NONE", "Lists all channels with their belonging clients.") self.cmdCreateChannel = self.createCommand( "createChannel", "/createChannel <name> <description> <password> <accessLevel>", "<NAME/DESCRIPTION/PASSWORD/ACCESSLEVEL>", "Creates a temporary Channel.") self.cmdRemoveChannel = self.createCommand( "removeChannel", "/removeChannel <name>", "<NAME>", "Removes the give Channel.") def __init__(self, upTime): self.upTime = upTime #Imports self.importScripts() #Create Commands self.initializeCommands() def handleInput(self, command): command = command.split() if command[0] == self.cmdClear.name: os.system('cls' if os.name == 'nt' else 'clear') elif command[0] == self.cmdListClients.name: if len(self.clientManager.clientList) < 1: self.logHelper.log("error", "No clients connected") else: self.logHelper.log("error", "Connected clients:") for clientObject in self.clientManager.clientList: self.logHelper.log( "info", clientObject.ip + " with name " + clientObject.username + " in " + clientObject.channelObject.name) elif command[0] == self.cmdHelp.name: self.logHelper.log("info", "Commands:") self.logHelper.log( "info", "----------------------------------------------------------") for command in self.commandList: self.logHelper.log( "info", command.syntax + " : " + command.description) self.logHelper.log( "info", "----------------------------------------------------------") elif command[0] == self.cmdKick.name: if len(self.clientManager.clientList) < 1: self.logHelper.log("error", "No clients connected") else: client = None try: client = command[1] except IndexError: self.logHelper.log("error", "Syntax: " + self.cmdKick.syntax) if client != None: if self.clientManager.ipExists(client): for clientObject in self.clientManager.clientList: if clientObject.ip == client: self.logHelper.log( "info", clientObject.ip + " : " + clientObject.username + " got kicked") clientObject.socketObject.sendall( self.decEncHelper.stringToBytes("402")) clientObject.socketObject.close() elif self.clientManager.usernameExists(client): for clientObject in self.clientManager.clientList: if clientObject.username.lower() == client: self.logHelper.log( "info", clientObject.ip + " : " + clientObject.username + " got kicked") clientObject.socketObject.sendall( self.decEncHelper.stringToBytes("402")) clientObject.socketObject.close() else: self.logHelper.log( "error", "Your given Ip/Name doesn't exist.") elif command[0] == self.cmdBan.name: if len(self.clientManager.clientList) < 1: self.logHelper.log("error", "No clients connected") else: client = None banTime = None try: client = command[1] banTime = int(command[2]) except IndexError: if client or banTime == None: self.logHelper.log("error", "Syntax: " + self.cmdBan.syntax) if client != None: if self.clientManager.ipExists(client): for clientObject in self.clientManager.clientList: if clientObject.ip == client: if banTime != None: if banTime == 0: self.fileHelper.addClientToBanList( clientObject.ip) self.logHelper.log( "info", clientObject.ip + " : " + clientObject.username + " got permanantly banned") clientObject.socketObject.sendall( self.decEncHelper.stringToBytes( "405" + "You got permanantly banned by the console" )) clientObject.socketObject.close() else: currentTimeStamp = datetime.datetime.now( ).timestamp() self.fileHelper.addClientToBanList( clientObject.ip + ":" + str(currentTimeStamp + int(banTime) * 60)) self.logHelper.log( "info", clientObject.ip + " : " + clientObject.username + " got banned for " + str(banTime) + "minutes") clientObject.socketObject.sendall( self.decEncHelper.stringToBytes( "405" + "You got banned for " + str(banTime) + " minutes by the console")) clientObject.socketObject.close() else: self.fileHelper.addClientToBanList( clientObject.ip) self.logHelper.log( "info", clientObject.ip + " : " + clientObject.username + " got permanantly banned") clientObject.socketObject.sendall( self.decEncHelper.stringToBytes( "405" + "You got permanantly banned by the console" )) clientObject.socketObject.close() elif self.clientManager.usernameExists(client): for clientObject in self.clientManager.clientList: if clientObject.username.lower() == client: if banTime != None: if banTime == 0: self.fileHelper.addClientToBanList( clientObject.ip) self.logHelper.log( "info", clientObject.ip + " : " + clientObject.username + " got permanantly banned") clientObject.socketObject.sendall( self.decEncHelper.stringToBytes( "405" + "You got permanantly banned by the console" )) clientObject.socketObject.close() else: currentTimeStamp = datetime.datetime.now( ).timestamp() self.fileHelper.addClientToBanList( clientObject.ip + ":" + str(currentTimeStamp + int(banTime) * 60)) self.logHelper.log( "info", clientObject.ip + " : " + clientObject.username + " got banned for " + str(banTime) + "minutes") clientObject.socketObject.sendall( self.decEncHelper.stringToBytes( "405" + "You got banned for " + str(banTime) + " minutes by the console")) clientObject.socketObject.close() else: self.fileHelper.addClientToBanList( clientObject.ip) self.logHelper.log( "info", clientObject.ip + " : " + clientObject.username + " got permanantly banned") clientObject.socketObject.sendall( self.decEncHelper.stringToBytes( "405" + "You got permanantly banned by the console" )) clientObject.socketObject.close() else: print( "[Server/Error] Your given Ip/Name doesn't exist.") elif command[0] == self.cmdListChannel.name: self.logHelper.log("info", "Channels:") self.logHelper.log( "info", "----------------------------------------------------------") for channel in self.channelManager.channelList: self.logHelper.log( "info", "-" + channel.name + " : Description:" + channel.description) self.logHelper.log("info", " Clients:") if len(channel.clientList) < 1: self.logHelper.log("info", " -channel is empty") else: for client in channel.clientList: self.logHelper.log( "info", " -" + client.ip + " : " + client.username) self.logHelper.log( "info", "----------------------------------------------------------") elif command[ 0] == self.cmdCreateChannel.name: #TODO: add things to remove user error when acces level isnst int and description contains spaces name = None description = None password = None accessLevel = None try: name = command[1] description = command[2] password = command[3] accessLevel = int(command[4]) self.channelManager.addChannel( Channel(name, description, password, accessLevel, list())) self.logHelper.log("info", "Channel " + name + " created.") except: if name or description or password or accessLevel == None: self.logHelper.log( "error", "Syntax: " + self.cmdCreateChannel.syntax) elif command[0] == self.cmdRemoveChannel.name: name = None try: name = command[1] for channel in self.channelManager.channelList: if channel.name == name: self.channelManager.removeChannel(channel) self.logHelper.log("info", "Channel " + name + " was removed.") except: if name == None: self.logHelper.log( "error", "Syntax: " + self.cmdRemoveChannel.syntax) elif command[ 0] == self.cmdChangeRank.name: #TODO: add things to remove user error for rank and check if rank isnt even a rank if len(self.clientManager.clientList) < 1: self.logHelper.log("error", "No clients connected") else: client = None rank = None try: client = command[1] rank = command[2] except IndexError: self.logHelper.log("error", "Syntax: " + self.cmdChangeRank.syntax) if client != None: if rank != None: if self.clientManager.ipExists(client): for clientObject in self.clientManager.clientList: if clientObject.ip == client: prevRank = clientObject.rank clientObject.rank = rank self.mysqlHelper.updateAccountRank( clientObject) self.logHelper.log( "info", "Changed " + clientObject.ip + ":" + str(clientObject.port) + " " + clientObject.username + " 's rank from " + prevRank + " to " + rank) elif self.clientManager.usernameExists(client): for clientObject in self.clientManager.clientList: if clientObject.username.lower() == client: prevRank = clientObject.rank clientObject.rank = rank self.mysqlHelper.updateAccountRank( clientObject) #clientObject.sendall(self.decEncHelper.stringToBytes("904" + rank))TODO: self.logHelper.log( "info", "Changed " + clientObject.ip + ":" + str(clientObject.port) + " " + clientObject.username + " 's rank from " + prevRank + " to " + rank) else: self.logHelper.log( "error", "Your given Ip/Name doesn't exist.") elif command[0] == self.cmdMonitorMode.name: monitor = True config = self.fileHelper.getConfig("Server Config") ip = config.ip + ":" + str(config.port) if len(ip) != 20: ip = " " * (20 - len(ip)) + ip while monitor: clearCount = 0 connectedClients = str(len(self.clientManager.clientList)) connectedAdmins = str(len(self.clientManager.getAdmins())) if len(connectedClients) != 3: connectedClients = "0" * ( 3 - len(connectedClients)) + connectedClients if len(connectedAdmins) != 3: connectedAdmins = "0" * ( 3 - len(connectedAdmins)) + connectedAdmins os.system('cls' if os.name == 'nt' else 'clear') try: print( "##############Monitoring##############\n#Server Ip/Port: " + ip + "#\n#Uptime: " + time.strftime( '%H:%M:%S', time.gmtime(int(time.time() - self.upTime))) + "#\n#Connected Clients: " + connectedClients + "#\n#Connected Admins: " + connectedAdmins + "#\n##############Monitoring##############") while clearCount != 5: sys.stdout.write("\x1b[1A") sys.stdout.write("\x1b[2K") clearCount = clearCount + 1 time.sleep(0.5) except KeyboardInterrupt: monitor = False os.system('cls' if os.name == 'nt' else 'clear') self.logHelper.log("info", "Exited monitor mode.") else: self.logHelper.log("error", "Unknown command: " + command[0]) self.logHelper.log("error", "type /help for a list of commands")
class Server: def importScripts(self): self.decEncHelper = DecodingEncodingHelper() self.channelManager = ChannelManager() self.clientManager = ClientManager() self.inputHandler = InputHandler(self.upTime) self.fileHelper = FileHelper() self.logHelper = LogHelper() def setConfig(self): config = self.fileHelper.getConfig("Server Config") self.port = config.port self.ipV4 = config.ip def inizializeChannel(self): self.welcomeChannel = Channel("Welcome_Channel", "welcome to the server", "No", 0, list()) self.channel1 = Channel("Channel_1", "Description of channel 1", "No", 0, list()) self.channel2 = Channel("Channel_2", "Description of channel 1", "No", 0, list()) self.channel3 = Channel("Channel_3", "Description of channel 1", "No", 0, list()) self.channelManager.addChannel(self.welcomeChannel) self.channelManager.addChannel(self.channel1) self.channelManager.addChannel(self.channel2) self.channelManager.addChannel(self.channel3) def inizializeServer(self): self.server = ServerThread((self.ipV4, self.port), ClientHandler) serverThread = threading.Thread(target=self.server.serve_forever) serverThread.daemon = True serverThread.start() self.logHelper.log("info" ,"Started server (version: " + self.fileHelper.getConfig("Version") + ") on ip: " + self.ipV4 + " port: " + str(self.port)) def askForInput(self): while True: try: command = input() except KeyboardInterrupt: self.logHelper.log("info", "Gracefully stopping server...") if len(self.clientManager.clientList) < 1: self.logHelper.log("info", "Gracefully stopped server") break else: for clientObject in self.clientManager.clientList: clientObject.socketObject.sendall(self.decEncHelper.stringToBytes("403")) self.logHelper.log("info", "Gracefully stopped server") break if str(command).startswith("/"): try: self.inputHandler.handleInput(str(command[1:]).lower()) except IndexError: self.logHelper.log("info", "type /help for a list of commands.") else: self.logHelper.log("info", "Commands always start with (/)") def __init__(self): self.upTime = time.time() #Imports self.importScripts() #Config self.setConfig() #Channel initialization self.inizializeChannel() #Server initializations self.inizializeServer() #Console Input self.askForInput()
class MysqlHelper: def __init__(self): self.fileHelper = FileHelper() self.logHelper = LogHelper() config = self.fileHelper.getConfig("Mysql Server Config") try: self.connection = mysql.connector.connect(user=config.username, password=config.password, host=config.ip, database=config.database) except: print("Couldn't establish connection to mysql database(" + config.database + ") with ip: " + config.ip) #TODO: find a way to handle with this def ExecuteCommand(self, command): result = None try: result = MysqlStatement( command, self.connection).execute().escape().fetchall().result except AttributeError: self.logHelper.log("info", "Created mysql table.") self.ExecuteCommandWithoutFetchAndResult( "CREATE TABLE accounts ( id INT NOT NULL AUTO_INCREMENT, username TEXT NOT NULL, password TEXT NOT NULL, email TEXT NOT NULL, rank TEXT NOT NULL, loggedIn TINYINT NOT NULL DEFAULT '0', PRIMARY KEY (id))" ) return result def ExecuteCommandWithoutFetchAndResult(self, command): return MysqlStatement(command, self.connection).execute().escape().commit() def tryLogin(self, clientObject, password): #TODO: give better fedback for layer 8 result = False lenght = None try: lenght = len( self.ExecuteCommand( "SELECT * FROM accounts WHERE username = '******'")) except: lenght = len( self.ExecuteCommand( "SELECT * FROM accounts WHERE username = '******'")) if lenght > 0: if self.ExecuteCommand( "select loggedIn,(case when loggedIn = 0 then 'loggedOut' when loggedIn = 1 then 'loggedIn' end) as loggedIn_status FROM accounts WHERE username = '******'")[0][1] != "loggedIn": if len( self.ExecuteCommand( "SELECT * FROM accounts WHERE username = '******' and password = '******'")) > 0: self.ExecuteCommandWithoutFetchAndResult( "UPDATE accounts SET loggedIn = 1 WHERE username = '******'") result = True return result def logoutAccount(self, clientObject): self.ExecuteCommandWithoutFetchAndResult( "UPDATE accounts SET loggedIn = 0 WHERE username = '******'") def getAccountRank(self, clientObject): return self.ExecuteCommand( "SELECT rank FROM accounts WHERE username = '******'")[0][0] def updateAccountRank(self, clientObject): self.ExecuteCommandWithoutFetchAndResult( "UPDATE accounts SET rank = '" + clientObject.rank + "' WHERE username = '******'")
class MysqlHelper: def __init__(self, announce): self.mysqlMode = 0 # 0 = Mysql; 1 = MysqlLite self.fileHelper = FileHelper() self.logHelper = LogHelper() config = self.fileHelper.getConfig("Mysql Server Config") try: self.connection = mysql.connector.connect(user=config.username, password=config.password, host=config.ip, database=config.database) except: self.mysqlMode = 1 if (announce): print( "[" + datetime.datetime.now().strftime("%H:%M:%S") + " ERROR]: Couldn't establish connection to mysql database(" + config.database + ") with ip: " + config.ip) print("[" + datetime.datetime.now().strftime("%H:%M:%S") + " INFO]: Falling back to MysqlLite.") #sqllite self.conn = self.create_connection("data/database.db") checkForAccountsTable = "SELECT * FROM accounts" result = self.executeCommandOnLite(self.conn, checkForAccountsTable) try: for row in result: if row[0] == 1: result = True else: result = False except: result = False if result == False: createTableStatement = "CREATE TABLE accounts (id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, username TEXT NOT NULL, password TEXT NOT NULL, email TEXT NOT NULL, rank TEXT NOT NULL, loggedIn TINYINT NOT NULL DEFAULT '0');" print("[" + datetime.datetime.now().strftime("%H:%M:%S") + " INFO]: Created accounts table in MysqlLite database.") self.executeCommandOnLite(self.conn, createTableStatement) def create_connection(self, db_file): try: conn = sqlite3.connect(db_file) return conn except Error as e: print(e) return e def executeCommandOnLite(self, connection, command): try: cursor = connection.cursor() result = cursor.execute(command) return result except Error as e: return e def ExecuteCommand(self, command): result = None try: result = MysqlStatement( command, self.connection).execute().escape().fetchall().result except AttributeError: self.logHelper.log("info", "Created mysql table.") self.ExecuteCommandWithoutFetchAndResult( "CREATE TABLE accounts ( id INT NOT NULL AUTO_INCREMENT, username TEXT NOT NULL, password TEXT NOT NULL, email TEXT NOT NULL, rank TEXT NOT NULL, loggedIn TINYINT NOT NULL DEFAULT '0', PRIMARY KEY (id))" ) return result def ExecuteCommandWithoutFetchAndResult(self, command): return MysqlStatement(command, self.connection).execute().escape().commit() def tryLogin(self, clientObject, password): #TODO: give better fedback for layer 8 if self.mysqlMode == 0: result = False lenght = None try: lenght = len( self.ExecuteCommand( "SELECT * FROM accounts WHERE username = '******'")) except: lenght = len( self.ExecuteCommand( "SELECT * FROM accounts WHERE username = '******'")) if lenght > 0: if self.ExecuteCommand( "select loggedIn,(case when loggedIn = 0 then 'loggedOut' when loggedIn = 1 then 'loggedIn' end) as loggedIn_status FROM accounts WHERE username = '******'")[0][1] != "loggedIn": if len( self.ExecuteCommand( "SELECT * FROM accounts WHERE username = '******' and password = '******'")) > 0: self.ExecuteCommandWithoutFetchAndResult( "UPDATE accounts SET loggedIn = 1 WHERE username = '******'") result = True return result else: checkLoggedIn = "select loggedIn,(case when loggedIn = 0 then 'loggedOut' when loggedIn = 1 then 'loggedIn' end) as loggedIn_status FROM accounts WHERE username = '******'" result = self.executeCommandOnLite(self.conn, checkLoggedIn) try: for row in result: if row[0] == 1: result = True else: result = False except: result = False if result == False: checkPW = "SELECT * FROM accounts WHERE username = '******' and password = '******'" result1 = self.executeCommandOnLite(self.conn, checkPW) result5 = False try: for row in result1: if (row[1] == clientObject.username): result5 = True else: result5 = False except: result5 = False if result5: updateStatus = "UPDATE accounts SET loggedIn = 1 WHERE username = '******'" self.executeCommandOnLite(self.conn, updateStatus) return result5 def logoutAccount(self, clientObject): if self.mysqlMode == 0: self.ExecuteCommandWithoutFetchAndResult( "UPDATE accounts SET loggedIn = 0 WHERE username = '******'") else: logoutAccount = "UPDATE accounts SET loggedIn = 0 WHERE username = '******'" self.executeCommandOnLite(self.conn, logoutAccount) def getAccountRank(self, clientObject): if self.mysqlMode == 0: result = self.ExecuteCommand( "SELECT rank FROM accounts WHERE username = '******'")[0][0] else: getRank = "SELECT rank FROM accounts WHERE username = '******'" result = self.executeCommandOnLite(self.conn, getRank) for row in result: print(row[0]) result = row[0] return result def updateAccountRank(self, clientObject): if self.mysqlMode == 0: self.ExecuteCommandWithoutFetchAndResult( "UPDATE accounts SET rank = '" + clientObject.rank + "' WHERE username = '******'") else: updateRank = "UPDATE accounts SET rank = '" + clientObject.rank + "' WHERE username = '******'" self.executeCommandOnLite(self.conn, updateRank)
class Client: def importScripts(self): self.decEncHelper = DecodingEncodingHelper() self.inputHandler = InputHandler(self.output) self.fileHelper = FileHelper(self.output) self.guiHelper = GUIHelper(self.output) def setConfig(self): Config = self.fileHelper.getConfig() self.ipV4 = Config.ip self.port = Config.port def inizializeClient(self, username, password): self.clientObject = ClientObject(username, None, self.ipV4, self.port, "Welcome_Channel") self.connected = False self.password = password def button(self): if self.mainWindow.statusButton.text() != "Offline": self.sendInput("/disconnect") self.mainWindow.statusButton.setText("Offline") self.mainWindow.output.clear() self.mainWindow.channelTree.clear() self.connected = False else: data = CustomDialog().getData() if data == ":": self.mainWindow.close() else: data = data.split(":") self.inizializeClient(data[0], data[1]) self.tryConnect() def tryConnect(self): trys = 0 while not self.connected: try: self.clientObject.socketObject = socket.socket( socket.AF_INET, socket.SOCK_STREAM) self.clientObject.socketObject.connect( (self.clientObject.ip, self.clientObject.port)) threading.Thread(target=ServerHandler, args=[self.clientObject, self.mainWindow]).start() self.clientObject.socketObject.sendall( self.decEncHelper.stringToBytes( "011" + self.clientObject.username + ":" + self.password)) self.connected = True except: trys = trys + 1 os.system('cls' if os.name == 'nt' else 'clear') self.guiHelper.printOutput( "[Client/Info] Attempting to connect to server with ip: " + self.clientObject.ip + ". Attempts: " + str(trys)) time.sleep(5) def sendInput(self, message): if self.connected: if str(message).startswith("/"): self.inputHandler.handleInput(str(message[1:]), self.clientObject) else: self.output.append("you: " + message) try: self.clientObject.socketObject.sendall( self.decEncHelper.stringToBytes("001" + message)) except: self.connected = False else: self.guiHelper.printOutput("not connected") def __init__(self, username, password, mainWindow): #Imports self.mainWindow = mainWindow self.mainWindow.statusButton.clicked.connect(self.button) self.output = mainWindow.output self.importScripts() #Config self.setConfig() #Client initializations self.inizializeClient(username, password) #Client trying to establish a connection self.tryConnect()