Made some improvements to the readability of the code and changed it to that Bob just prints the decrypted message rather than sending it to Alice to print it
This commit is contained in:
parent
b9cf6dc6b5
commit
9bd87725d3
|
@ -13,6 +13,7 @@ def main():
|
|||
|
||||
key_length = 512
|
||||
|
||||
#Bob's keys
|
||||
publicKey, privateKey = encrypt_decrypt.encrypt_decrypt.make_keys(key_length)
|
||||
|
||||
asyncio.run(websock.alice.Alice.messageEncrypt("ws://localhost:8765", publicKey, message))
|
||||
|
|
|
@ -16,7 +16,7 @@ class Alice():
|
|||
encryptedMessage = encrypt_decrypt.encrypt_decrypt.encryption(message, publicKey)
|
||||
async with connect(uri) as websocket:
|
||||
await websocket.send(encryptedMessage)
|
||||
message = await websocket.recv()
|
||||
print(message)
|
||||
#message = await websocket.recv()
|
||||
#print(message)
|
||||
|
||||
#asyncio.run(hello("ws://localhost:8765"))
|
|
@ -11,14 +11,21 @@ import encrypt_decrypt
|
|||
from websockets import serve
|
||||
|
||||
class Bob():
|
||||
async def decrypt(websocket, path, privateKey):
|
||||
def __init__(self,privateKey):
|
||||
self.pk = privateKey
|
||||
|
||||
async def decrypt(websocket, path):
|
||||
privateKey = "" #dummy variable to make code work, please change!
|
||||
async for message in websocket:
|
||||
message = await websocket.recv()
|
||||
decryptedMessage = encrypt_decrypt.encrypt_decrypt.decryption(message, privateKey)
|
||||
await websocket.send(decryptedMessage)
|
||||
#await websocket.send(decryptedMessage)
|
||||
print(decryptedMessage)
|
||||
|
||||
async def main(privateKey):
|
||||
async with serve(Bob.decrypt, "localhost", 8765, privateKey):
|
||||
async def messageDecrypt(privateKey):
|
||||
#do something with privateKey to send it to decrypt
|
||||
|
||||
async with serve(Bob.decrypt, "localhost", 8765):
|
||||
await asyncio.Future() # run forever
|
||||
|
||||
#asyncio.run(main())
|
Loading…
Reference in New Issue
Block a user