def auction_to_view(self, message_json):
username = message_json["username"]
sk = self.auction_repository.session_key_clients[username]
hm = base64.b64decode(message_json["hmac"])
cr = message_json["message"].encode()
if not HMAC_Conf.verify_integrity(hm, cr, sk):
return base64.b64encode(
"{ \"type\" : \"Tempered data\"}".encode('utf-8'))
# Decrypts the message
data = decrypt_data(sk, message_json["message"],
base64.b64decode(message_json["iv"]),
base64.b64decode(message_json["Key"]),
self.auction_repository.private_key)
auction_id = unpadd_data(data, sk)
auction = self.auction_repository.auctions[str(auction_id, "utf-8")]
blockchain = pickle.dumps(auction.blockchain)
# cipher first the blockchain
enc_bl = encrypt_message_sk(
blockchain, self.auction_repository.session_key_clients[username])
message = "{ \"type\" : \"auction_to_view\", \n"
message_int = "{\n\"auction_id\" : \"" + encrypt_message_sk(
auction_id, sk) + "\" ,\n"
message_int += "\"blockchain\" : \"" + enc_bl + "\", \n"
message_int += "\"auct_type\" : \"" + encrypt_message_sk(
auction.type, sk) + "\",\n"
message_int += "\"avail\" : \"" + encrypt_message_sk(
str(auction.open), sk) + "\"\n"
message_int += "}"
rsa_kg = RSAKGen()
client_pub = rsa_kg.load_public_key(os.getcwd() + "/Clients/" +
username)
enc_json_message = encrypt_message_complete(
base64.b64encode(message_int.encode("utf-8")), sk, client_pub)
key = enc_json_message[0]
iv = enc_json_message[2]
data = enc_json_message[1]
hmac = HMAC_Conf.integrity_control(
data.encode(),
self.auction_repository.session_key_clients[username])
message += "\"message\" : \"" + data + "\",\n"
message += "\"Key\" : \"" + str(base64.b64encode(key),
'utf-8') + "\",\n"
message += "\"hmac\" : \"" + str(base64.b64encode(hmac),
'utf-8') + "\", \n"
message += "\"iv\" : \"" + str(base64.b64encode(iv), 'utf-8') + "\"\n"
message += "}"
return base64.b64encode(message.encode("utf-8"))
def list_auctions(self, client):
hmac = HMAC_Conf.integrity_control(client.username.encode(), client.session_key_repository)
message = "{ \"type\" : \"list_auctions\" ,\n"
message += "\"username\" : \"" + client.username + "\",\n"
message += "\"hmac\" : \"" + str(base64.b64encode(hmac), 'utf-8')+ "\"\n"
message += "}"
return message, AR_ADDRESS
def list_auctions(self, message_json):
sk = self.auction_repository.session_key_clients[
message_json["username"]]
# VERIFYES THE message integrity
if not HMAC_Conf.verify_function("username", message_json, sk):
return base64.b64encode(
"{ \"type\" : \"Tempered data\"}".encode('utf-8'))
# gets the list of auctions and serializes it
auction_list = self.auction_repository.listAuctions()
serialized = pickle.dumps(auction_list)
# loads the shared secret between the server and the user
username = message_json["username"]
session_key = self.auction_repository.session_key_clients[username]
list_auc = encrypt_message_sk(serialized, session_key)
rsa_kg = RSAKGen()
client_pub = rsa_kg.load_public_key(os.getcwd() + "/Clients/" +
username)
enc_json_message = encrypt_message_complete(
base64.b64encode(list_auc.encode("utf-8")), sk, client_pub)
message = "{ \"type\" : \"list_auctions\" ,\n"
key = enc_json_message[0]
iv = enc_json_message[2]
data = enc_json_message[1]
hmac = HMAC_Conf.integrity_control(
data.encode(),
self.auction_repository.session_key_clients[username])
message += "\"list\" : \"" + data + "\", \n"
message += "\"Key\" : \"" + str(base64.b64encode(key),
'utf-8') + "\",\n"
message += "\"hmac\" : \"" + str(base64.b64encode(hmac),
'utf-8') + "\", \n"
message += "\"iv\" : \"" + str(base64.b64encode(iv), 'utf-8') + "\"\n"
message += "}"
print("auction_repos", message)
return base64.b64encode(message.encode("utf-8"))
def close_auction(self, client, message_json):
# VERIFYES THE message integrity
if not HMAC_Conf.verify_function("message", message_json, client.session_key_repository):
return base64.b64encode("{ \"type\" : \"Tempered data\"}".encode('utf-8'))
# Decrypts the message
data = decrypt_data(client.session_key_repository,
message_json["message"], base64.b64decode(message_json["iv"]),
base64.b64decode(message_json["Key"]),
client.private_key)
data_json = json.loads(data, strict="false")
something = unpadd_data(data_json["blockchain"], client.session_key_repository)
blockchain = pickle.loads(something)
rsa_kg = RSAKGen()
for bid in blockchain:
bid.second_symmetric_key = rsa_kg.decipher_with_private_key(client.auction_private_key, bid.second_symmetric_key)
bloc_chain_enc = encrypt_message_sk(pickle.dumps(blockchain), client.session_key_repository)
message = "{ \"type\" : \"close_auction\" ,\n"
message += "\"username\" : \"" + client.username + "\" ,\n"
message_interm = "{"
message_interm += "\"auction_id\" : \"" + data_json["auction_id"]+ "\",\n"
message_interm += "\"blockchain\" : \"" + bloc_chain_enc + "\"\n"
message_interm += "}"
enc_json_message = encrypt_message_complete(base64.b64encode(message_interm.encode("utf-8")), client.session_key_repository, client.server_public_key_repository)
key = enc_json_message[0]
iv = enc_json_message[2]
data = enc_json_message[1]
hmac = HMAC_Conf.integrity_control(data.encode(),client.session_key_repository)
message += "\"message\" : \"" + data + "\" ,\n"
message += "\"Key\" : \"" + str(base64.b64encode(key), 'utf-8') + "\",\n"
message += "\"iv\" : \"" + str(base64.b64encode(iv), 'utf-8') + "\",\n"
message += "\"hmac\" : \"" + str(base64.b64encode(hmac), 'utf-8') + "\"\n"
message += "}"
return message, AR_ADDRESS
def pre_bid_get_auction(self, client, auction_id):
enc_auct_id = encrypt_message_sk(auction_id, client.session_key_repository)
#print(enc_auct_id.encode())
# Cipher the auction ID complete
enc_json_message = encrypt_message_complete(base64.b64encode(enc_auct_id.encode("utf-8")),client.session_key_repository, client.server_public_key_repository)
key = enc_json_message[0]
iv = enc_json_message[2]
data = enc_json_message[1]
hmac = HMAC_Conf.integrity_control(data.encode(),client.session_key_repository)
message = "{ \"type\" : \"pre_bid_auction\" ,\n"
message += "\"username\" : \"" + client.username + "\" ,\n"
message += "\"auction_id\" : \"" + data + "\",\n"
message += "\"Key\" : \"" + str(base64.b64encode(key), 'utf-8') + "\",\n"
message += "\"iv\" : \"" + str(base64.b64encode(iv), 'utf-8') + "\",\n"
message += "\"hmac\" : \"" + str(base64.b64encode(hmac), 'utf-8') + "\"\n"
message += "}"
return message, AR_ADDRESS
def make_bid(self, client):
amount = ""
auction_id = ""
while auction_id == "":
auction_id = input("Auction ID: ")
while amount == "":
amount = input("Bid Amount: ")
if not self.pre_bid(client,auction_id):
return
citizen = CitizenCard()
certificate = citizen.load_authentication_certificate()
signature = citizen.digital_signature(amount)
message = "{ \"type\" : \"bid\" ,\n"
message += "\"username\" : \"" + client.username + "\",\n"
interm_message = "{ \"auction_id\" : \"" + encrypt_message_sk(auction_id,client.session_key_repository) + "\",\n"
interm_message += "\"amount\" : \"" + encrypt_message_sk(amount,client.session_key_repository) + "\",\n"
interm_message += "\"bidder\" : \"" + encrypt_message_sk(citizen.load_name(),client.session_key_repository) + "\",\n"
interm_message += "\"certificate\" : \"" + encrypt_message_sk(certificate, client.session_key_repository) + "\",\n"
interm_message += "\"signature\" : \"" + encrypt_message_sk(signature,client.session_key_repository) + "\"\n"
interm_message += "}"
enc_json_message = encrypt_message_complete(base64.b64encode(interm_message.encode("utf-8")),
client.session_key_repository,
client.server_public_key_repository)
key = enc_json_message[0]
iv = enc_json_message[2]
data = enc_json_message[1]
hmac = HMAC_Conf.integrity_control(data.encode(),client.session_key_repository)
message += "\"message\" : \"" + data + "\",\n"
message += "\"Key\" : \"" + str(base64.b64encode(key), 'utf-8') + "\",\n"
message += "\"hmac\" : \"" + str(base64.b64encode(hmac), 'utf-8') + "\", \n"
message += "\"iv\" : \"" + str(base64.b64encode(iv), 'utf-8') + "\"\n"
message += "}"
#print(message)
return message, AR_ADDRESS
def auction_to_view(self, client, id_auc):
message = "{\"type\" : \"auction_to_view\",\n"
message += "\"username\" :\"" + client.username +"\", \n"
idd = encrypt_message_sk(id_auc, client.session_key_repository)
inter = encrypt_message_complete(base64.b64encode(idd.encode("utf-8")), client.session_key_repository, client.server_public_key_repository)
key = inter[0]
iv = inter[2]
data = inter[1]
hmac = HMAC_Conf.integrity_control(data.encode(),client.session_key_repository)
message += "\"message\" : \"" + data + "\",\n"
message += "\"Key\" : \"" + str(base64.b64encode(key), 'utf-8') + "\",\n"
message += "\"hmac\" : \"" + str(base64.b64encode(hmac), 'utf-8') + "\", \n"
message += "\"iv\" : \"" + str(base64.b64encode(iv), 'utf-8') + "\"\n"
message += "}"
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
try:
sock.sendto(base64.b64encode(message.encode()), AR_ADDRESS)
oo = True
dee = b""
while oo:
data, server = sock.recvfrom(SOCKET_BYTES)
dee += data
try:
decoded_message = base64.b64decode(dee)
message_json = json.loads(decoded_message, strict = False)
oo = False
except:
oo = True
decoded_message = base64.b64decode(dee)
# The client should always receive a confirmation from the server
message = json.loads(decoded_message, strict = False)
if not HMAC_Conf.verify_function("message", message, client.session_key_repository):
return base64.b64encode("{ \"type\" : \"Tempered data\"}".encode('utf-8'))
# Decrypts the message
data = decrypt_data(client.session_key_repository,
message["message"], base64.b64decode(message["iv"]),
base64.b64decode(message["Key"]),
client.private_key)
data_json = json.loads(data, strict="false")
# Now decript the blockchain
auc_type = unpadd_data(data_json["auct_type"], client.session_key_repository)
dec_bloc = unpadd_data(data_json["blockchain"], client.session_key_repository)
block_chain = pickle.loads(dec_bloc)
availability = unpadd_data(data_json["avail"],client.session_key_repository)
#
# Auction is still ongoing
#
if str(availability, "utf-8") == "True":
# validate blockchain
if not Auction.validate_blockchain(block_chain, auc_type):
print("The auction is not valid! ")
return base64.b64encode("{ \"type\" : \"Tempered data\"}".encode('utf-8'))
for bid in block_chain:
print("------------------------")
print("User: "******"utf-8"))
print("Ammount: " + str(bid.amount, "utf-8"))
if bid.previous_hash == None:
print("Previous_hash: " + str(bid.previous_hash))
else:
print("Previous_hash: " + str(bid.previous_hash, "utf-8"))
print("Bid Hash: " + str(bid.hash, "utf-8"))
print("\nAuction is valid")
return base64.b64encode("{ \"type\" : \"Valid\"}".encode('utf-8'))
#
# Always validate blockchain
#
if not Auction.validate_blockchain(block_chain, auc_type):
print("The auction is not valid! ")
return base64.b64encode("{ \"type\" : \"Tempered data\"}".encode('utf-8'))
#
#Auction is closed now we check
#
if str(auc_type, "utf-8") == ENGLISH_AUCTION:
validity = True
for bid in block_chain:
sec_dec = Fernet(bid.first_symmetric_key)
print("---------------------------------")
print("Bidder: " + str(sec_dec.decrypt(bid.username), "utf-8"))
print("Amount: " + str(bid.amount, "utf-8"))
cert_bytes = sec_dec.decrypt(bid.certificate)
cert = x509.load_pem_x509_certificate(cert_bytes, default_backend())
a = CitizenCard()
if not a.validate_certificate(cert):
validity = False
sign = sec_dec.decrypt(bid.signature)
if not a.check_signature(cert, sign, bid.amount ):
validity = False
print("Signature: Not valid")
else:
print("Signature: Valid")
if bid.previous_hash == None:
print("Previous_hash: " + str(bid.previous_hash))
else:
print("Previous_hash: " + str(bid.previous_hash, "utf-8"))
print("Bid Hash: " + str(bid.hash, "utf-8"))
if validity == False:
print("The auction is not valid")
else:
validity = True
prev_bid = None #amount
bd = None #Winner bid
for bid in block_chain:
sec_dec = Fernet(bid.first_symmetric_key)
dec_amount = sec_dec.decrypt(bid.amount)
ammount = int((str(dec_amount,"utf-8")))
if prev_bid == None:
prev_bid = ammount
bd = bid
else:
if prev_bid < ammount:
prev_bid = ammount
bd = bid
print("---------------Winner--------------------")
sec_dec = Fernet(bd.first_symmetric_key)
dec_amount = sec_dec.decrypt(bd.amount)
print("Amount: " + str(dec_amount,"utf-8"))
dec_user = sec_dec.decrypt(bd.username)
print("User: "******"utf-8"))
cert_bytes = sec_dec.decrypt(bd.certificate)
cert = x509.load_pem_x509_certificate(cert_bytes, default_backend())
a = CitizenCard()
if not a.validate_certificate(cert):
validity = False
print("The auction is not valid")
print("----------------")
finally:
sock.close()
def create_auction(self, client):
message = "{ \"type\" : \"create_auction\" ,\n"
message += "\"username\" : \"" + client.username + "\",\n"
# Now ask for all the details
print("\n")
# Verify the fields
# Auction needs a name
auction_name = input("Auction name: ")
while auction_name == "":
auction_name = input("Auction name: ")
# If there is no description then put it as None
auction_description = input("Auction description: ")
if auction_description == "":
auction_description = "None"
# If no given minimum the put it at zero $
auction_min_number_bids = input("Minimum price bid: ")
if auction_min_number_bids == "":
auction_min_number_bids = "0"
# There needs to be a limit
auction_time = input("Auction time limit (Hrs): ")
while int(auction_time) < 0:
auction_time = input("Auction time limit (Hrs): ")
# If there is no max bids then 0
auction_max_number_bids = input("Auction maximum number of bids: ")
if auction_max_number_bids == "":
auction_max_number_bids = "0"
# If there is no given names then put it as None
auction_allowed_bidders = input("Auction allowed bidders (separate with [,]): ")
if auction_allowed_bidders == "":
auction_allowed_bidders = "None"
# There must have a type
auction_type = input("Auction type ((B)lind or (N)ormal): ")
if auction_type.lower() in ["b", "blind"]:
auction_type = BLIND_AUCTION
print("Blind auction is being created...")
else:
auction_type = ENGLISH_AUCTION
print("Normal auction is being created...")
# sign the session key
rsa = RSAKGen()
signature = rsa.sign_message(client.session_key_manager,client.auction_private_key)
auction_json_message = "{\"auction_name\" : \"" + encrypt_message_sk(auction_name, client.session_key_manager)+ "\" ,\n"
auction_json_message += "\"auction_description\" : \"" + encrypt_message_sk(auction_description,
client.session_key_manager) + "\" ,\n"
auction_json_message += "\"auction_min_number_bids\" : \"" + encrypt_message_sk(auction_min_number_bids,
client.session_key_manager) + "\",\n"
auction_json_message += "\"auction_time\" : \"" + encrypt_message_sk(auction_time, client.session_key_manager) + "\" ,\n"
auction_json_message += "\"auction_max_number_bids\" : \"" + encrypt_message_sk(auction_max_number_bids,
client.session_key_manager) + "\" ,\n"
auction_json_message += "\"auction_allowed_bidders\" : \"" + encrypt_message_sk(auction_allowed_bidders,
client.session_key_manager) + "\" ,\n"
auction_json_message += "\"auction_type\" : \""+ encrypt_message_sk(auction_type, client.session_key_manager) + "\" ,\n"
# now put our public auction key and a signature to prove that he has the private key
auction_json_message += "\"auction_user_key\" : \"" + encrypt_message_sk(get_public_key_bytes(client.auction_public_key),
client.session_key_manager) + "\", \n"
auction_json_message += "\"auction_signature\" : \"" + encrypt_message_sk(str(base64.b64encode(signature), 'utf-8'),
client.session_key_manager)+ "\" \n"
auction_json_message += "}"
enc_json_message = encrypt_message_complete(base64.b64encode(auction_json_message.encode("utf-8")),
client.session_key_manager,
client.server_public_key_manager)
key = enc_json_message[0]
iv = enc_json_message[2]
data = enc_json_message[1]
hmac = HMAC_Conf.integrity_control(data.encode(), client.session_key_manager)
message += "\"message\" : \"" + data + "\",\n"
message += "\"Key\" : \"" + str(base64.b64encode(key), 'utf-8') + "\",\n"
message += "\"hmac\" : \"" + str(base64.b64encode(hmac), 'utf-8')+ "\", \n"
message += "\"iv\" : \"" + str(base64.b64encode(iv), 'utf-8') + "\"\n"
message += "}"
return message, AM_ADDRESS
def make_bid(self, message_json):
sk = self.auction_repository.session_key_clients[
message_json["username"]]
# VERIFIES THE message integrity
if not HMAC_Conf.verify_function("message", message_json, sk):
return base64.b64encode(
"{ \"type\" : \"Tempered data\"}".encode('utf-8'))
# Decrypts the message
username = message_json["username"]
data = decrypt_data(
self.auction_repository.session_key_clients[username],
message_json["message"], base64.b64decode(message_json["iv"]),
base64.b64decode(message_json["Key"]),
self.auction_repository.private_key)
# Loads the message to json
message_json = json.loads(data, strict="False")
auction_id = unpadd_data(
message_json["auction_id"],
self.auction_repository.session_key_clients[username])
bidder = unpadd_data(
message_json["bidder"],
self.auction_repository.session_key_clients[username])
amount = unpadd_data(
message_json["amount"],
self.auction_repository.session_key_clients[username])
signature = unpadd_data(
message_json["signature"],
self.auction_repository.session_key_clients[username])
certificate = unpadd_data(
message_json["certificate"],
self.auction_repository.session_key_clients[username])
# validate certificate and signature
cert = x509.load_pem_x509_certificate(certificate, default_backend())
citizen = CitizenCard()
if not citizen.check_signature(cert, signature, amount):
return base64.b64encode(
"{ \"type\" : \"No valid signature\"}".encode('utf-8'))
if not citizen.validate_certificate(cert):
return base64.b64encode(
"{ \"type\" : \"No valid certificate\"}".encode('utf-8'))
auction = self.auction_repository.auctions[str(auction_id, "utf-8")]
response = auction.makeBid(bidder, amount, signature, certificate)
if not response:
return base64.b64encode(
"{ \"type\" : \"No success\"}".encode('utf-8'))
# create receipt
rsa_kg = RSAKGen()
block = auction.blockchain[-1]
server_signature = rsa_kg.sign_message(
block.hash, self.auction_repository.private_key)
server_signature = encrypt_message_sk(server_signature, sk)
uuids = encrypt_message_sk(str(uuid.uuid1()), sk)
last = encrypt_message_sk(block.hash, sk)
bidder = encrypt_message_sk(bidder, sk)
receipt = create_receipt(encrypt_message_sk(str(block.timestamp), sk),
encrypt_message_sk(str(auction_id),
sk), server_signature,
encrypt_message_sk(amount, sk),
encrypt_message_sk(signature, sk), uuids,
last, bidder)
# cipher receipt with pub_key
client_pub = rsa_kg.load_public_key(os.getcwd() + "/Clients/" +
username)
enc_json_message = encrypt_message_complete(
base64.b64encode(receipt.encode("utf-8")), sk, client_pub)
key = enc_json_message[0]
iv = enc_json_message[2]
data = enc_json_message[1]
hmac = HMAC_Conf.integrity_control(
data.encode(),
self.auction_repository.session_key_clients[username])
message = "{ \"type\" : \"receipt\", \n"
message += "\"message\" : \"" + data + "\",\n"
message += "\"Key\" : \"" + str(base64.b64encode(key),
'utf-8') + "\",\n"
message += "\"hmac\" : \"" + str(base64.b64encode(hmac),
'utf-8') + "\", \n"
message += "\"iv\" : \"" + str(base64.b64encode(iv), 'utf-8') + "\"\n"
message += "}"
return base64.b64encode(message.encode("utf-8"))
def create_auction(self, message_json, address):
username = message_json["username"]
session_key_client = self.auction_manager.session_clients[username]
# VERIFYES THE message integrity
if not HMAC_Conf.verify_function("message", message_json, session_key_client ):
return base64.b64encode("{ \"type\" : \"Tempered data\"}".encode('utf-8'))
# Decrypts the message
data = decrypt_data(session_key_client,
message_json["message"], base64.b64decode(message_json["iv"]),
base64.b64decode(message_json["Key"]),
self.auction_manager.private_key)
# Loads the messsage to json
message_json = json.loads(data,strict="False")
# Decrypt the rest of the data with the SessionKey
auction_name = unpadd_data(message_json["auction_name"],session_key_client)
auction_description = unpadd_data(message_json["auction_description"], session_key_client)
auction_min_number_bids = unpadd_data(message_json["auction_min_number_bids"],session_key_client)
auction_time = unpadd_data(message_json["auction_time"],session_key_client)
auction_max_number_bids = unpadd_data(message_json["auction_max_number_bids"], session_key_client)
auction_allowed_bidders = unpadd_data(message_json["auction_allowed_bidders"], session_key_client)
auction_type = unpadd_data(message_json["auction_type"], session_key_client)
auct_padd = unpadd_data(
message_json["auction_user_key"].encode('utf-8'),session_key_client)
auction_user_key = serialization.load_pem_public_key(auct_padd, default_backend())
auction_signature = unpadd_data(message_json["auction_signature"], session_key_client)
rsa = RSAKGen()
verification = rsa.verify_sign(base64.b64decode(auction_signature),
session_key_client, auction_user_key)
# if its not valid then the address is the client's and the message is an error
if not verification:
return base64.b64encode("{\"type\" : \"Not a valid auction signature\"}".encode("utf-8")), address
# Construct the message to send to the AR
message_final_json = "{"
message_final_json += "\"type\" : \"create_auction\", \n"
# create the encrypted message
repository_key = self.auction_manager.session_key_repository
# Create the interm message
message_interm = "{"
message_interm += "\"username\" : \"" + encrypt_message_sk(username, repository_key) + "\", \n"
message_interm += "\"auction_name\" : \"" + encrypt_message_sk(auction_name, repository_key) + "\", \n"
message_interm += "\"auction_description\" : \"" + encrypt_message_sk(auction_description, repository_key) + "\", \n"
message_interm += "\"auction_min_number_bids\" : \"" + encrypt_message_sk(auction_min_number_bids, repository_key) + "\", \n"
message_interm += "\"auction_time\" : \"" + encrypt_message_sk(auction_time, repository_key) + "\", \n"
message_interm += "\"auction_max_number_bids\" : \"" + encrypt_message_sk(auction_max_number_bids, repository_key) + "\", \n"
message_interm += "\"auction_allowed_bidders\" : \"" + encrypt_message_sk(auction_allowed_bidders, repository_key) + "\", \n"
message_interm += "\"auction_type\" : \"" + encrypt_message_sk(auction_type, repository_key) + "\", \n"
message_interm += "\"auction_user_key\" : \"" + encrypt_message_sk(auct_padd, repository_key) + "\" \n"
message_interm += "}"
# Encrypt Complete
enc_json_message = encrypt_message_complete(base64.b64encode(message_interm.encode("utf-8")),
repository_key,
self.auction_manager.public_repository_key)
key = enc_json_message[0]
iv = enc_json_message[2]
data = enc_json_message[1]
hmac = HMAC_Conf.integrity_control(data.encode(), repository_key)
message_final_json += "\"message\" : \"" + data + "\",\n"
message_final_json += "\"Key\" : \"" + str(base64.b64encode(key), 'utf-8') + "\",\n"
message_final_json += "\"iv\" : \"" + str(base64.b64encode(iv), 'utf-8') + "\",\n"
message_final_json += "\"hmac\" : \"" + str(base64.b64encode(hmac), 'utf-8') + "\"\n"
message_final_json += "\n}"
print(message_final_json)
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock.sendto(base64.b64encode(message_final_json.encode("utf-8")), AR_ADDRESS)
return base64.b64encode("{\"type\" : \"success\"}".encode("utf-8"))
def initialize_session_key(self, address, actions):
# asks for a challenge
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
message = "{\"type\" : \"ask_challenge\"," + "\n"
message += "\"username\" : \"" + self.username + "\"" + "\n"
message += "}"
sock.sendto(base64.b64encode(message.encode("utf-8")), address)
# receives the challenge
data, server = sock.recvfrom(utils_app.SOCKET_BYTES)
decoded_message = base64.b64decode(data)
sock.close()
message = json.loads(decoded_message, strict=False)
challenge = message["challenge"]
# Our parameters
parameters = dh.generate_parameters(generator=5,
key_size=utils_app.DH_KEY_SIZE,
backend=default_backend())
# Our private key and public key from DH
private_key = parameters.generate_private_key()
public_key = private_key.public_key()
pickled_params = codecs.encode(
pickle.dumps(
parameters.parameter_bytes(encoding=Encoding.PEM,
format=ParameterFormat.PKCS3)),
"base64").decode()
message = "{\"type\" : \"session_key\"," + "\n"
message += "\"params\" : \"" + pickled_params + "\"," + "\n"
message += "\"username\" : \"" + self.username + "\"," + "\n"
message += "\"pk\" : \"" + public_key.public_bytes(
encoding=serialization.Encoding.PEM,
format=serialization.PublicFormat.SubjectPublicKeyInfo).decode(
'utf-8') + "\",\n"
message += "\"public\" : \"" + self.public_key.public_bytes(
encoding=serialization.Encoding.PEM,
format=serialization.PublicFormat.SubjectPublicKeyInfo).decode(
'utf-8') + "\","
inner_message = actions.trust_server(self, challenge=challenge)
enc_json_message = None
server_pub = None
if address == utils_app.AM_ADDRESS:
server_pub = self.server_public_key_manager
else:
server_pub = self.server_public_key_repository
enc_json_message = utilities.encrypt_message_complete(
base64.b64encode(inner_message.encode("utf-8")), "", server_pub)
key = enc_json_message[0]
iv = enc_json_message[2]
data = enc_json_message[1]
rsa_kg = RSAKGen()
random_key = Fernet.generate_key()
hmac = HMAC_Conf.integrity_control(data.encode(), random_key)
random_key = rsa_kg.cipher_public_key(server_pub, random_key)
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
message += "\"message\" : \"" + data + "\",\n"
message += "\"Key\" : \"" + str(base64.b64encode(key),
'utf-8') + "\",\n"
message += "\"hmac\" : \"" + str(base64.b64encode(hmac),
'utf-8') + "\", \n"
message += "\"random_key\" : \"" + str(base64.b64encode(random_key),
'utf-8') + "\", \n"
message += "\"iv\" : \"" + str(base64.b64encode(iv), 'utf-8') + "\"\n"
message += "}"
sock.sendto(base64.b64encode(message.encode("utf-8")), address)
#print(message)
##### Sent the parameters with all the information #####
###########################################################################################################
# Receive response
data, server = sock.recvfrom(utils_app.SOCKET_BYTES)
decoded_data = base64.b64decode(data).decode()
json_message = json.loads(decoded_data, strict=False)
if json_message["type"] == "No valid signature":
print("No valid signature")
sys.exit(0)
elif json_message["type"] == "No valid certificate":
print("No valid signature")
sys.exit(0)
# 'type', 'pk' -> public dh_key , 'info' -> handshake data, 'server_key'
peer_pk = serialization.load_pem_public_key(
json_message["pk"].encode('utf-8'), default_backend())
shared_secret = private_key.exchange(peer_pk)
derived_key = hkdf.HKDF(algorithm=hashes.SHA256(), length=utils_app.DH_HKDF_KEY, salt=None,
info=json_message["info"].encode('utf-8'), backend=default_backend())\
.derive(shared_secret)
if address == utils_app.AM_ADDRESS:
self.session_key_manager = derived_key
else:
self.session_key_repository = derived_key
### Recieved the key from the server and created the public key ####
############################################################################################################
sock.close()
