gergely/PySecretHandshake
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ci: Use black instead of flake8

Browse Source 
It results in mostly the same style, plus it’s configurable via pyprojects.toml.
This commit is contained in:
Gergely Polonkai
2023-10-29 09:47:57 +01:00
parent bbfc137ca8
commit 93f2d329a4
14 changed files with 221 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
secret_handshake/network.py
View File

@@ -77,13 +77,13 @@ class SHSServer(SHSEndpoint):
async def _handshake(self, reader, writer):
data = await reader.readexactly(64)
if not self.crypto.verify_challenge(data):
raise SHSClientException('Client challenge is not valid')
raise SHSClientException("Client challenge is not valid")
writer.write(self.crypto.generate_challenge())
data = await reader.readexactly(112)
if not self.crypto.verify_client_auth(data):
raise SHSClientException('Client auth is not valid')
raise SHSClientException("Client auth is not valid")
writer.write(self.crypto.generate_accept())
@@ -125,21 +125,22 @@ class SHSClient(SHSDuplexStream, SHSEndpoint):
SHSEndpoint.__init__(self)
self.host = host
self.port = port
self.crypto = SHSClientCrypto(client_kp, server_pub_key, ephemeral_key=ephemeral_key,
application_key=application_key)
self.crypto = SHSClientCrypto(
client_kp, server_pub_key, ephemeral_key=ephemeral_key, application_key=application_key
)
async def _handshake(self, reader, writer):
writer.write(self.crypto.generate_challenge())
data = await reader.readexactly(64)
if not self.crypto.verify_server_challenge(data):
raise SHSClientException('Server challenge is not valid')
raise SHSClientException("Server challenge is not valid")
writer.write(self.crypto.generate_client_auth())
data = await reader.readexactly(80)
if not self.crypto.verify_server_accept(data):
raise SHSClientException('Server accept is not valid')
raise SHSClientException("Server accept is not valid")
async def open(self):
reader, writer = await asyncio.open_connection(self.host, self.port)