def openConnection():
HOST = "localhost"
PORT = 000 #invalid port number for security reasons
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
s.bind((HOST, PORT))
except socket.error as err:
print('Bind failed. Error Code : '.format(err))
s.listen(10)
#conn, addr = s.accept()
open = True
try:
while (open):
conn, addr = s.accept()
#conn.send(bytes("Message1"+"\r\n",'UTF-8'))
message = conn.recv(1024)
msg = message.decode(encoding='UTF-8')
if (msg == "filePaths"):
paths = []
while (msg != "done"):
message = conn.recv(1024)
msg = message.decode(encoding='UTF-8')
paths.append(msg)
for p in paths:
pa = p.replace('/', '\\')
classification = PSD.convertAndClassify(pa)
conn.send(bytes(classification + "\r", 'UTF-8'))
elif (msg != "close"):
credentials = msg.split()
username = credentials[0]
password = hash_sha1(credentials[1])
event = credentials[2]
if (event == "login"):
# p = collection.find_one({'username': username})
# print(p['password'])
# print("\n\n")
# print(password)
if (collection.count_documents({
'username': username,
'password': password
}) == 1):
#if(verifyPass(p['password'], password)):
conn.send(bytes("ok" + "\r\n", 'UTF-8'))
else:
conn.send(bytes("notok" + "\r\n", 'UTF-8'))
if (event == "register"):
if (collection.count_documents({'username': username}) >=
1):
conn.send(bytes("taken" + "\r\n", 'UTF-8'))
else:
conn.send(bytes("ok" + "\r\n", 'UTF-8'))
collection.insert_one({
'username': username,
'password': password
})
# print(message.decode(encoding='UTF-8'))
if (message.decode(encoding='UTF-8') == "close"):
print("Connection gracefully closed")
open = False
except ConnectionResetError:
s.close()
print("Connection terminated")
s.close()
