cipherkey is now guaranteed to be 32 characters strong
continuous-integration/drone/push Build is passing Details

This commit is contained in:
Torma Kristóf 2021-04-26 00:22:05 +02:00
parent ded1968f26
commit 63733a317f
Signed by: tormakris
GPG Key ID: DC83C4F2C41B1047
1 changed files with 5 additions and 1 deletions

View File

@ -2,6 +2,7 @@
import json
from base64 import b64encode, b64decode
import pyDH
from Crypto.Hash import SHA512
from Crypto.Cipher import PKCS1_OAEP
from Crypto.Cipher import ChaCha20
from Crypto.PublicKey.RSA import RsaKey
@ -62,7 +63,10 @@ class NetWrapper:
decodedmsg = json.loads(msg.decode('UTF-8'))
cipher_rsa = PKCS1_OAEP.new(self.serverPrivateKey)
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 login(self) -> bool: