emdee/toxygen_wrapper
1
0
Fork 0
Code Issues Pull Requests Actions Projects Wiki Activity

typing

Browse Source
This commit is contained in:
emdee 2023-12-11 14:37:04 +00:00
parent ec39c353d6
commit da3602293a
2 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
wrapper/toxav.py
View File

@ -347,7 +347,9 @@ class ToxAV:
:param user_data: pointer (c_void_p) to user data :param user_data: pointer (c_void_p) to user data
""" """
if callback is None: if callback is None:
self.libtoxav.toxav_callback_audio_receive_frame(self._toxav_pointer, POINTER(None)(), user_data) self.libtoxav.toxav_callback_audio_receive_frame(self._toxav_pointer,
POINTER(None)(),
user_data)
self.audio_receive_frame_cb = None self.audio_receive_frame_cb = None
return return
LOG_DEBUG(f"toxav_callback_audio_receive_frame") LOG_DEBUG(f"toxav_callback_audio_receive_frame")

14
wrapper/toxencryptsave.py
View File

@ -16,16 +16,16 @@ class ToxEncryptSave:
def __init__(self): def __init__(self):
self.libtoxencryptsave = libtox.LibToxEncryptSave() self.libtoxencryptsave = libtox.LibToxEncryptSave()
def is_data_encrypted(self, data): def is_data_encrypted(self, data: bytes) -> bool:
""" """
Checks if given data is encrypted Checks if given data is encrypted
""" """
func = self.libtoxencryptsave.tox_is_data_encrypted func = self.libtoxencryptsave.tox_is_data_encrypted
func.restype = c_bool func.restype = c_bool
result = func(c_char_p(bytes(data))) result = func(c_char_p(bytes(data)))
return result return bool(result)
def pass_encrypt(self, data, password): def pass_encrypt(self, data: bytes, password: str) -> bytes:
""" """
Encrypts the given data with the given password. Encrypts the given data with the given password.
@ -33,9 +33,11 @@ class ToxEncryptSave:
""" """
out = create_string_buffer(len(data) + TOX_PASS_ENCRYPTION_EXTRA_LENGTH) out = create_string_buffer(len(data) + TOX_PASS_ENCRYPTION_EXTRA_LENGTH)
tox_err_encryption = c_int() tox_err_encryption = c_int()
if type(password) != bytes:
password = bytes(password, 'utf-8')
self.libtoxencryptsave.tox_pass_encrypt(c_char_p(data), self.libtoxencryptsave.tox_pass_encrypt(c_char_p(data),
c_size_t(len(data)), c_size_t(len(data)),
c_char_p(bytes(password, 'utf-8')), c_char_p(password),
c_size_t(len(password)), c_size_t(len(password)),
out, out,
byref(tox_err_encryption)) byref(tox_err_encryption))
@ -50,7 +52,7 @@ class ToxEncryptSave:
elif tox_err_encryption == TOX_ERR_ENCRYPTION['FAILED']: elif tox_err_encryption == TOX_ERR_ENCRYPTION['FAILED']:
raise RuntimeError('The encryption itself failed.') raise RuntimeError('The encryption itself failed.')
def pass_decrypt(self, data, password): def pass_decrypt(self, data, password) -> bytes:
""" """
Decrypts the given data with the given password. Decrypts the given data with the given password.
@ -66,7 +68,7 @@ class ToxEncryptSave:
byref(tox_err_decryption)) byref(tox_err_decryption))
tox_err_decryption = tox_err_decryption.value tox_err_decryption = tox_err_decryption.value
if tox_err_decryption == TOX_ERR_DECRYPTION['OK']: if tox_err_decryption == TOX_ERR_DECRYPTION['OK']:
return out[:] return bytes(out[:])
elif tox_err_decryption == TOX_ERR_DECRYPTION['NULL']: elif tox_err_decryption == TOX_ERR_DECRYPTION['NULL']:
raise ArgumentError('Some input data, or maybe the output pointer, was null.') raise ArgumentError('Some input data, or maybe the output pointer, was null.')
elif tox_err_decryption == TOX_ERR_DECRYPTION['INVALID_LENGTH']: elif tox_err_decryption == TOX_ERR_DECRYPTION['INVALID_LENGTH']: