fstream/fstream-server.py

#!/usr/bin/env python3
import sys

oldexcepthook = sys.excepthook
def newexcepthook(type,value,traceback):
	oldexcepthook(type,value,traceback)
	#input("Press ENTER to quit.")
sys.excepthook = newexcepthook

import os
p = os.path.join
pUp = os.path.dirname
s = False
if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
	s = os.path.realpath(sys.executable)
else:
	s = os.path.realpath(__file__)
sp = pUp(s)

# script start
import threading
import queue
import socket
import subprocess
import configparser

outThreads = {}
inThreads = {}
threadId = 0
threadsLock = threading.Lock()
fileLock = threading.Lock()

serverAddr = ("127.0.0.1",12000)
serverSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

bufferSize = 1000 # buffer size in bytes

def commandToList(cmd):
	args = []
	cArg = ""
	escape = False
	quoted = False
	for letter in cmd:
		if escape == True:
			cArg += letter
			escape = False
			continue
		
		if letter == "\\":
			escape = True
			continue
		
		#if quoted == False and letter == ",":
		if letter == ",":
			if cArg == "": continue
			args.append(cArg)
			cArg = ""
			continue
		
		#if letter == '"':
		#	quoted = not quoted
		#	continue
		
		cArg += letter
	
	args.append(cArg)
	
	return args

class outThread(threading.Thread):
	def __init__(self,threadId,connection,address,user):
		threading.Thread.__init__(self)
		self.threadId = threadId
		self.queue = queue.Queue()
		self.connection = connection
		self.address = address
		self.user = user
	
	def closeThread(self):
		with threadsLock:
			try:
				self.connection.close()
			except Exception as e:
				print("closing a connection failed: " +str(e))
			del outThreads[str(self.threadId)]
	
	def run(self):
		try:
			while True:
				data = self.queue.get()
				self.connection.sendall(data)
			self.closeThread()
		except:
			self.closeThread()
			raise

class inThread(threading.Thread):
	def __init__(self,threadId,connection,address):
		threading.Thread.__init__(self)
		self.threadId = threadId
		self.connection = connection
		self.address = address
	
	def closeThread(self,closeConnection = True):
		with threadsLock:
			if closeConnection:
				for thread in outThreads:
					thread = outThreads[thread]
					if thread.user == self.user:
						try:
							thread.connection.close()
						except Exception as e:
							print("closing a connection failed: " +str(e))
							pass
				
				try:
					self.connection.close()
				except Exception as e:
					print("closing a connection failed: " +str(e))
					pass
			del inThreads[str(self.threadId)]
	
	def run(self):
		try:
			global threadId
			data = self.connection.recv(1000)
			if data == b"":
				self.closeThread()
				return
			data = data.decode("utf-8")
			while data[-1] == " ": data = data[:-1]
			args = commandToList(data)
			cmd = args.pop(0)
			
			user = args[0]
			self.user = user
			password = False
			if len(args) > 1:
				password = args[1]
			
			userPath = p(sp,"users",user+ ".ini")
			with fileLock:
				if not os.path.isfile(userPath):
					self.closeThread()
					return
			
				config = configparser.ConfigParser()
				config.read(userPath)
			
			if cmd == "watch":
				if cmd in config and "pass" in config[cmd] and config[cmd]["pass"] != "" and config[cmd]["pass"] != password:
					self.closeThread()
					return
				
				with threadsLock:
					thread = outThread(self.threadId,self.connection,self.address,self.user)
					outThreads[str(self.threadId)] = thread
					thread.start()
				
				self.closeThread(False)
				return
			
			if cmd == "broadcast":
				if cmd in config and "pass" in config[cmd] and config[cmd]["pass"] != "" and config[cmd]["pass"] != password:
					self.closeThread()
					return
				
				while True:
					data = self.connection.recv(bufferSize)
					if data == b"":
						self.closeThread()
						return
					with threadsLock:
						for thread in outThreads:
							thread = outThreads[thread]
							if thread.user == user:
								thread.queue.put(data)
			
			self.closeThread()
			return
		except:
			self.closeThread()
			raise

class debugThread(threading.Thread):
	def __init__(self):
		threading.Thread.__init__(self)
	
	def run(self):
		while True:
			with threadsLock:
				print("Threads - IN:  " +str(len(inThreads)))
				print("Threads - OUT: " +str(len(outThreads)))
			time.sleep(10)

def main():
	global threadId
	serverSocket.bind(serverAddr)
	serverSocket.listen(1024)
	
	# DEBUG
	global time
	import time
	debug = debugThread()
	debug.start()
	# DEBUG END
	
	while True:
		connection, address = serverSocket.accept()
		connection.settimeout(15)
		with threadsLock:
			threadId += 1
			
			thread = inThread(threadId,connection,address)
			inThreads[str(threadId)] = thread
			thread.start()

if __name__ == '__main__':
	main()