ezri/dbus-fast
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: speed up auth phase (#131)

Browse Source
This commit is contained in:
J. Nick Koston
2022-10-31 06:57:39 -05:00
committed by GitHub
parent 10e8e52cd6
commit 3eef636826
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/dbus_fast/aio/message_bus.py
View File

@@ -430,7 +430,9 @@ class MessageBus(BaseMessageBus):
async def _auth_readline(self) -> str: async def _auth_readline(self) -> str:
buf = b"" buf = b""
while buf[-2:] != b"\r\n": while buf[-2:] != b"\r\n":
buf += await self._loop.sock_recv(self._sock, 2) # The auth protocol is line based, so we can read until we get a
# newline.
buf += await self._loop.sock_recv(self._sock, 1024)
return buf[:-2].decode() return buf[:-2].decode()
async def _authenticate(self) -> None: async def _authenticate(self) -> None:
@@ -455,6 +457,9 @@ class MessageBus(BaseMessageBus):
) )
self._stream.flush() self._stream.flush()
if response == "BEGIN": if response == "BEGIN":
# The first octet received by the server after the \r\n of the BEGIN command
# from the client must be the first octet of the authenticated/encrypted stream
# of D-Bus messages.
break break
def disconnect(self) -> None: def disconnect(self) -> None: