def __init__(self, path):
"""
Create a client instance given a configuration file
@param path Path to the configuration file of the client
to create. Must be a valid path.
"""
self.init = False
# Let's parse the configuration file
self.cfg = ClientParser(path).parse()
self.nick = self.cfg["general"]["nick"]
self.server = (self.cfg["server"]["addr"], self.cfg["server"]["port"])
self.cid = self.cfg["comptoir"]["cid"]
# Default IV for AES keys
self.iv = str.encode("*#hello,world!#*")
# Derive an AES key from a passphrase
self.__key = PBKDF2(self.cfg["comptoir"]["key"], self.iv)
# Client socket
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# Wrapping socket for ssl
self.sock = ssl.wrap_socket(self.sock, ciphers="HIGH")
self.connected = False
self.ui = CommandLineUI(self.nick, self.cid)
self.init = True
class Client(object):
def __init__(self, path):
"""
Create a client instance given a configuration file
@param path Path to the configuration file of the client
to create. Must be a valid path.
"""
self.init = False
# Let's parse the configuration file
self.cfg = ClientParser(path).parse()
self.nick = self.cfg["general"]["nick"]
self.server = (self.cfg["server"]["addr"], self.cfg["server"]["port"])
self.cid = self.cfg["comptoir"]["cid"]
# Default IV for AES keys
self.iv = str.encode("*#hello,world!#*")
# Derive an AES key from a passphrase
self.__key = PBKDF2(self.cfg["comptoir"]["key"], self.iv)
# Client socket
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# Wrapping socket for ssl
self.sock = ssl.wrap_socket(self.sock, ciphers="HIGH")
self.connected = False
self.ui = CommandLineUI(self.nick, self.cid)
self.init = True
def connect(self):
self.sock.connect((self.server))
self.sock.send(self.nick)
# TODO HANDLE ACK CORRECTLY
self.sock.recv(1024)
self.connected = True
def disconnect(self):
self.sock.close()
self.ui.close()
self.connected = False
@property
def keyhash(self):
return sha3_256(self.__key).hexdigest()
def join(self, cid=None):
"""
Require connected
"""
if not self.connected:
raise NotConnectedException
self.sock.send(self.cid)
self.sock.send(self.keyhash)
# TODO other way to check the ACK
self.sock.recv(1024)
def send(self):
while True:
msg = self.ui.get_input()
if msg == "/quit":
pkt = QuitPacket().pack()
else:
pkt = MessagePacket(msg=msg, key=self.__key, keyhash=self.keyhash).pack()
with open("./log", "a") as f:
f.write("SYN: ")
f.write(str(pkt))
f.write("\n")
self.sock.send(pkt)
def close(self):
self.ui.close()
def idle(self):
"""
Require connected
"""
if not self.connected:
raise NotConnectedException
t = thread.start_new_thread(self.send, ())
try:
while True:
data = self.sock.recv(1024)
if len(data) == 0:
#TODO CHANGE THIS
raise NotConnectedException
type = Packet.get_type(data)
if type == PKT_MESSAGE:
user, msg = MessagePacket(data=data, key=self.__key).unpack()
self.ui.new_msg("[{0}] {1}".format(user, msg))
elif type == PKT_CONNECTED:
user = ConnectedPacket(data=data).unpack()
self.ui.new_msg("{0} just appeared".format(user))
elif type == PKT_DISCONNECTED:
user = DisconnectedPacket(data=data).unpack()
self.ui.new_msg("{0} ran away".format(user))
else:
raise NotImplemented
except NotConnectedException:
self.disconnect()
def __del__(self):
if self.init:
self.ui.close()
