88 lines
2.0 KiB
Cython
88 lines
2.0 KiB
Cython
import cython
|
|
|
|
from ..connection cimport APIConnection
|
|
from .base cimport APIFrameHelper
|
|
|
|
|
|
cdef bint TYPE_CHECKING
|
|
|
|
cdef unsigned int NOISE_STATE_HELLO
|
|
cdef unsigned int NOISE_STATE_HANDSHAKE
|
|
cdef unsigned int NOISE_STATE_READY
|
|
cdef unsigned int NOISE_STATE_CLOSED
|
|
|
|
cdef bytes NOISE_HELLO
|
|
cdef object PACK_NONCE
|
|
|
|
cdef class EncryptCipher:
|
|
|
|
cdef object _nonce
|
|
cdef object _encrypt
|
|
|
|
cdef bytes encrypt(self, object frame)
|
|
|
|
cdef class DecryptCipher:
|
|
|
|
cdef object _nonce
|
|
cdef object _decrypt
|
|
|
|
cdef bytes decrypt(self, object frame)
|
|
|
|
cdef class APINoiseFrameHelper(APIFrameHelper):
|
|
|
|
cdef object _noise_psk
|
|
cdef str _expected_name
|
|
cdef unsigned int _state
|
|
cdef object _server_name
|
|
cdef object _proto
|
|
cdef EncryptCipher _encrypt_cipher
|
|
cdef DecryptCipher _decrypt_cipher
|
|
|
|
@cython.locals(
|
|
header=bytes,
|
|
preamble="unsigned char",
|
|
msg_size_high="unsigned char",
|
|
msg_size_low="unsigned char",
|
|
)
|
|
cpdef void data_received(self, object data)
|
|
|
|
@cython.locals(
|
|
msg=bytes,
|
|
type_high="unsigned char",
|
|
type_low="unsigned char",
|
|
msg_type="unsigned int",
|
|
payload=bytes
|
|
)
|
|
cdef void _handle_frame(self, bytes frame)
|
|
|
|
@cython.locals(
|
|
chosen_proto=char,
|
|
server_name_i=int
|
|
)
|
|
cdef void _handle_hello(self, bytes server_hello)
|
|
|
|
cdef void _handle_handshake(self, bytes msg)
|
|
|
|
cdef void _handle_closed(self, bytes frame)
|
|
|
|
@cython.locals(handshake_frame=bytearray, frame_len="unsigned int")
|
|
cdef void _send_hello_handshake(self)
|
|
|
|
cdef void _setup_proto(self)
|
|
|
|
@cython.locals(psk_bytes=bytes)
|
|
cdef _decode_noise_psk(self)
|
|
|
|
@cython.locals(
|
|
type_="unsigned int",
|
|
data=bytes,
|
|
data_header=bytes,
|
|
packet=tuple,
|
|
data_len=cython.uint,
|
|
frame=bytes,
|
|
frame_len=cython.uint,
|
|
)
|
|
cpdef void write_packets(self, list packets, bint debug_enabled)
|
|
|
|
cdef _error_on_incorrect_preamble(self, bytes msg)
|