diff --git a/client/netwrapper.py b/client/netwrapper.py index dfb8219..a179909 100644 --- a/client/netwrapper.py +++ b/client/netwrapper.py @@ -4,6 +4,7 @@ import string import json from base64 import b64encode, b64decode import pyDH +from Crypto.Hash import SHA512 from Crypto.Cipher import ChaCha20, PKCS1_OAEP from Crypto.PublicKey.RSA import RsaKey from Crypto.Random import get_random_bytes @@ -61,7 +62,10 @@ class NetWrapper: decodedmsg = json.loads(msg.decode('UTF-8')) cipher_rsa = PKCS1_OAEP.new(self.privateKey) serverpubkey = int(cipher_rsa.decrypt(b64decode(decodedmsg['message'])).decode('UTF-8')) - self.cipherkey = dh.gen_shared_key(serverpubkey).encode('UTF-8') + cipherkey = dh.gen_shared_key(serverpubkey).encode('UTF-8') + hasher = SHA512.new() + hasher.update(cipherkey) + self.cipherkey = hasher.hexdigest()[:32] def authenticate(self, password: str):