Upstream/aioesphomeapi
1
0
Fork 0
mirror of https://github.com/esphome/aioesphomeapi.git synced 2024-12-26 17:27:40 +01:00
Code Issues Projects Releases Wiki Activity

Reduce ChaCha20CipherReuseable overhead (#489)

Browse Source
This commit is contained in:
J. Nick Koston 2023-07-18 13:06:19 -05:00 committed by GitHub
parent 7196ca6ee8
commit 401ff3b61a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
aioesphomeapi/_frame_helper.py
View File

@ -5,7 +5,7 @@ from abc import abstractmethod
from enum import Enum
from functools import partial
from struct import Struct
from typing import TYPE_CHECKING, Any, Callable, Optional, Union, cast
from typing import TYPE_CHECKING, Any, Callable, Optional, Type, Union, cast
from chacha20poly1305_reuseable import ChaCha20Poly1305Reusable
from cryptography.exceptions import InvalidTag
@ -42,12 +42,11 @@ WRITE_EXCEPTIONS = (RuntimeError, ConnectionResetError, OSError)
class ChaCha20CipherReuseable(ChaCha20Cipher): # type: ignore[misc]
"""ChaCha20 cipher that can be reused."""
@property
def klass(self): # type: ignore[no-untyped-def]
return ChaCha20Poly1305Reusable
format_nonce = PACK_NONCE
def format_nonce(self, n: int) -> bytes:
return PACK_NONCE(n)
@property
def klass(self) -> Type[ChaCha20Poly1305Reusable]:
return ChaCha20Poly1305Reusable
class ESPHomeNoiseBackend(DefaultNoiseBackend): # type: ignore[misc]