Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2025-10-24 08:18:01 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Ensure login token is only sent if aes state is ESTABLISHED (#702)

Browse Source
This commit is contained in:
J. Nick Koston
2024-01-24 09:43:42 -10:00
committed by GitHub
parent aecf0ecd8a
commit 3df837cc82
2 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
kasa/aestransport.py
View File

@@ -151,7 +151,7 @@ class AesTransport(BaseTransport):
async def send_secure_passthrough(self, request: str) -> Dict[str, Any]:
"""Send encrypted message as passthrough."""
url = f"http://{self._host}/app"
if self._login_token:
if self._state is TransportState.ESTABLISHED and self._login_token:
url += f"?token={self._login_token}"
encrypted_payload = self._encryption_session.encrypt(request.encode()) # type: ignore
@@ -250,6 +250,7 @@ class AesTransport(BaseTransport):
_LOGGER.debug("Will perform handshaking...")
self._key_pair = None
self._login_token = None
self._session_expire_at = None
self._session_cookie = None
@@ -284,9 +285,7 @@ class AesTransport(BaseTransport):
handshake_key = resp_dict["result"]["key"]
if (
cookie := http_client.get_cookie( # type: ignore
self.SESSION_COOKIE_NAME
)
cookie := http_client.get_cookie(self.SESSION_COOKIE_NAME) # type: ignore
) or (
cookie := http_client.get_cookie("SESSIONID") # type: ignore
):