class ServerListenCommand(Command):
def __init__(self, port, sharedSecret):
self.host = "0.0.0.0"
self.port = port
self.kex = KeyExchanger()
self.auth = Authenticator(sharedSecret)
def execute(self):
self.sock = socket.socket()
logger.info("Server socket created")
try:
self.sock.bind((self.host, self.port))
logger.info("Server listening on port: {}".format(self.port))
except socket.error as msg:
logger.error("Server socket bind failed")
self.sock.close()
return
logger.info("Server socket bind complete")
t = Thread(target=self.listenThread)
t.start()
def listenThread(self):
logger.info("Listening for connection on separate thread")
self.sock.listen(NUM_CLIENTS)
try:
while True:
conn, addr = self.sock.accept()
logger.info("Connected with {} : {}".format(addr[0], addr[1]))
self.handleClient(conn)
except:
logger.info("Connection to client closed unexpectedly")
traceback.print_exc()
self.sock.close()
def handleClient(self, conn):
try:
conn = ConnectionWrapper(conn)
self.kex.exchangeKey(KeyExchanger.SERVER, conn)
self.auth.authenticate(conn)
t = Thread(target=packetListener, args=(conn, ))
t.start()
while True:
try:
msg = Q.get(timeout=0.2)
if not msg:
break
if msg.mtype == Message.SEND:
conn.send(msg.bytes)
Q.task_done()
except Empty:
pass
except AuthError as err:
logger.error(str(err))
finally:
logger.info("Connection to client closed")
conn.close()
class ClientConnectCommand(Command):
def __init__(self, host, port, sharedSecret):
self.host = host
self.port = port
self.kex = KeyExchanger()
self.auth = Authenticator(sharedSecret)
def execute(self):
self.sock = socket.socket()
logger.info("Client socket created")
t = Thread(target=self.connectThread)
t.start()
def connectThread(self):
conn = None
try:
self.sock.connect((self.host, self.port))
logger.info("Client connected")
conn = ConnectionWrapper(self.sock)
# use connection from now on to talk
self.kex.exchangeKey(KeyExchanger.CLIENT, conn)
self.auth.authenticate(conn)
t = Thread(target=packetListener, args=(conn, ))
t.start()
while True:
try:
msg = Q.get(timeout=0.2)
if not msg:
break
if msg.mtype == Message.SEND:
conn.send(msg.bytes)
Q.task_done()
except Empty:
pass
except AuthError as err:
logger.error(str(err))
finally:
logger.info("Connection to server closed")
if conn:
conn.close()