o i%@sddlmZmZddlmZmZddlmZddlm Z ddlm Z dZ dZd ZeGd d d ZeGd d d ZGdddZGdddZGdddZdS)) dataclassfield)ListOptional) FfiClient)e2ee_pb2)ffi_pb2sLKFrameEncryptionKeyc@sBeZdZUdZeeed<eZeed<e Z e ed<e Z e ed<dS)KeyProviderOptionsN shared_key ratchet_saltratchet_window_sizefailure_tolerance)__name__ __module__ __qualname__r rbytes__annotations__DEFAULT_RATCHET_SALTrDEFAULT_RATCHET_WINDOW_SIZErintDEFAULT_FAILURE_TOLERANCErrrR/var/www/html/livekit_bhavya/venv/lib/python3.10/site-packages/livekit/rtc/e2ee.pyr s   r c@s4eZdZUeedZeed<ejj Z ejj ed<dS) E2EEOptions)default_factorykey_provider_optionsencryption_typeN) rrrrr rr proto_e2eeEncryptionTypeGCMr ValueTyperrrrr#s rc@seZdZdedefddZedefddZded edd fd d Z d edefd dZ d edefddZ de ded edd fddZ de d edefddZde d edefddZd S) KeyProvider room_handleoptionscCs||_||_dSN)_options _room_handleselfr%r&rrr__init__*s zKeyProvider.__init__returncC|jSr')r(r+rrrr&.zKeyProvider.optionskey key_indexNcCs6t}|j|j_||jj_||jj_tj |dS)aSets the shared encryption key. Parameters: key (bytes): The new shared key. key_index (int): The index of the key. Example: ```python key_provider.set_shared_key(b"my_shared_key", key_index=1) ``` N) proto_ffi FfiRequestr)e2eer%set_shared_keyr2r rinstancerequest)r+r1r2reqrrrr62s    zKeyProvider.set_shared_keycC6t}|j|j_||jj_tj |}|jjj }|S)a1Exports the shared encryption key. Parameters: key_index (int): The index of the key to export. Returns: bytes: The exported shared key. Example: ```python key = key_provider.export_shared_key(key_index=1) ``` ) r3r4r)r5r%get_shared_keyr2rr7r8r1)r+r2r9respr1rrrexport_shared_keyDs     zKeyProvider.export_shared_keycCr:)aJRatchets the shared encryption key to a new key. Parameters: key_index (int): The index of the key to ratchet. Returns: bytes: The new ratcheted shared key. Example: ```python new_key = key_provider.ratchet_shared_key(key_index=1) ``` ) r3r4r)r5r%ratchet_shared_keyr2rr7r8new_key)r+r2r9r<r?rrrr>Ys     zKeyProvider.ratchet_shared_keyparticipant_identitycCsFt}|j|j_||jj_||jj_||jj_||_t j |dS)aSets the encryption key for a specific participant. Parameters: participant_identity (str): The identity of the participant. key (bytes): The encryption key to set. key_index (int): The index of the key. Example: ```python key_provider.set_key("participant123", b"participant_key", key_index=2) ``` N) r3r4r)r5r%set_keyr@r2r1rr7r8)r+r@r1r2r9rrrrAps    zKeyProvider.set_keycC@t}|j|j_||jj_||jj_tj |}|jjj }|S)aExports the encryption key for a specific participant. Parameters: participant_identity (str): The identity of the participant. key_index (int): The index of the key to export. Returns: bytes: The exported key. Example: ```python key = key_provider.export_key("participant123", key_index=2) ``` ) r3r4r)r5r%get_keyr@r2rr7r8r1)r+r@r2r9r<r1rrr export_keys     zKeyProvider.export_keycCrB)aRatchets the encryption key for a specific participant to a new key. Parameters: participant_identity (str): The identity of the participant. key_index (int): The index of the key to ratchet. Returns: bytes: The new ratcheted key. Example: ```python new_key = key_provider.ratchet_key("participant123", key_index=2) ``` ) r3r4r)r5r% ratchet_keyr@r2rr7r8r?)r+r@r2r9r<r?rrrrEs     zKeyProvider.ratchet_key)rrrrr r,propertyr&rr6r=r>strrArDrErrrrr$)sr$c@seZdZdedededefddZedefdd Zedefd d Z edefd d Z deddfddZ deddfddZ dS) FrameCryptorr%r@r2enabledcCs||_||_||_||_dSr')r)_enabled_participant_identity _key_index)r+r%r@r2rIrrrr,s zFrameCryptor.__init__r-cCr.r')rKr/rrrr@r0z!FrameCryptor.participant_identitycCr.r')rLr/rrrr2r0zFrameCryptor.key_indexcCr.r'rJr/rrrrIr0zFrameCryptor.enabledNcC>||_t}|j|j_|j|jj_||jj_ t j |dS)zEnables or disables frame encryption. Parameters: enabled (bool): True to enable, False to disable. Example: ```python frame_cryptor.set_enabled(True) ``` N) rJr3r4r)r5r%rKcryptor_set_enabledr@rIrr7r8r+rIr9rrr set_enabled     zFrameCryptor.set_enabledcCrN)zSets the key index for encryption/decryption. Parameters: key_index (int): The new key index. Example: ```python frame_cryptor.set_key_index(3) ``` N) rLr3r4r)r5r%rKcryptor_set_key_indexr@r2rr7r8)r+r2r9rrr set_key_indexrRzFrameCryptor.set_key_index) rrrrrGboolr,rFr@r2rIrQrTrrrrrHsrHc@sneZdZdedeefddZedeefddZ ede fdd Z d e dd fd d Z de efddZd S) E2EEManagerr%r&cCs6||_||_|du|_|durt|j|j|_dSdSr')r&r)rJr$r _key_providerr*rrrr,s  zE2EEManager.__init__r-cCr.r')rWr/rrr key_providerr0zE2EEManager.key_providercCr.r'rMr/rrrrIr0zE2EEManager.enabledrINcCs2||_t}|j|j_||jj_tj |dS)zEnables or disables end-to-end encryption. Parameters: enabled (bool): True to enable, False to disable. Example: ```python e2ee_manager.set_enabled(True) ``` N) rJr3r4r)r5r%manager_set_enabledrIrr7r8rPrrrrQs    zE2EEManager.set_enabledcCsRt}|j|j_tj|}g}|jjj D]}| t |j|j |j |jq|S)aYRetrieves the list of frame cryptors for participants. Returns: List[FrameCryptor]: A list of FrameCryptor instances. Example: ```python cryptors = e2ee_manager.frame_cryptors() for cryptor in cryptors: print(cryptor.participant_identity) ``` )r3r4r)r5r%rr7r8manager_get_frame_cryptorsframe_cryptorsappendrHr@r2rI)r+r9r<r[ frame_cryptorrrrr[s  zE2EEManager.frame_cryptors)rrrrrrr,rFr$rXrUrIrQrrHr[rrrrrVsrVN) dataclassesrrtypingrr _ffi_clientr_protorr r r3rrrr rr$rHrVrrrrs    8