def getKeyFromID(TID):
message_to_send = ['1002',TID]
enc_socket = socket.socket()
enc_socket.connect(('127.0.0.1',9099))
sendMessage(enc_socket,message_to_send)
message_received = receiveMessages(enc_socket)
print(message_received)
enc_socket.close()
def sendDataToServer(message_to_send):
server = socket.socket()
server.connect(('127.0.0.1', 9098))
sendMessage(server, message_to_send)
message_received = receiveMessages(server)
server.close()
if (message_received):
return message_received
else:
return False
def authenticateUser(secret_number):
system('cls')
returnValue = True
print(
"you are required to enter the filename or full path of the filename which contains your private key"
)
message_to_send = ['0001', secret_number]
server = socket.socket()
server.connect(('127.0.0.1', 9098))
sendMessage(server, message_to_send)
message_received = receiveMessages(server)
if (message_received[0] == '0002'):
if (message_received[1] is False):
print('no data found in corresponding to the given secret number')
exit()
puzzle_number = message_received[1]
privateKey = None
filename = input(
'your private key filename(full path of file with extension) : ')
try:
privateKey = SigningKey.from_pem(open(filename).read())
except:
print(
'(BBEMS error: 401) File does not contain any private key information'
)
sendMessage(server, ['0010', False])
server.close()
exit()
signature = privateKey.sign(str(puzzle_number).encode())
message_to_send = ['0010', signature]
sendMessage(server, message_to_send)
message_received = receiveMessages(server)
server.close()
if message_received[0] == '0020' and message_received[1]:
print('Authentication successful!!!\n')
else:
print('authentication failed !')
returnValue = False
else:
print('(BBEMS error: 0001) Failed to autenticate user')
returnValue = False
return returnValue
def askForNewKeyandTID(message_to_send,notified_socket):
enc_socket = socket.socket()
enc_socket.connect(('127.0.0.1',9099))
sendMessage(enc_socket,message_to_send)
message_received = receiveMessages(enc_socket)
data = None
if message_received[0] == '2001' and message_received[1]:
data = message_received
else:
data = ['T1001',False,()]
enc_socket.close()
sendMessage(notified_socket,data)
def askForKeyFromTID(message_to_send):
enc_socket = socket.socket()
enc_socket.connect(('127.0.0.1',9099))
sendMessage(enc_socket,message_to_send)
message_received = receiveMessages(enc_socket)
return message_received
server_socket.listen()
print(f"connection started at {IP} {PORT}")
socket_list = [server_socket,]
authenticated = {}
authenticated[000000] = True
#Running the server live
while True:
readers,_,errors = select.select(socket_list,[],socket_list)
for notified_socket in readers:
if notified_socket == server_socket:
client_socket,client_addr = server_socket.accept()
print(f"connection established with {client_socket}")
socket_list.append(client_socket)
else:
message_received = receiveMessages(notified_socket)
if message_received is False:
print(f"disconnected from {notified_socket}")
socket_list.remove(notified_socket)
else:
if message_received[0] == '1111':
print("new registration save karna hai")
secret_number = message_received[1][0]
grade = message_received[1][1]
public_key = message_received[1][2]
value = saveRegistrationDataInDatabase(secret_number,public_key,grade)
message_to_send = ['2111',value]
sendMessage(notified_socket,message_to_send)
elif message_received[0] == '1112':
secret_number = message_received[1]
value = getRegistrationDetailsFromDatabase(secret_number)