o ]i@sddlmZddlmZddlmZmZgdZdZgdZ dZ d Z Gd d d e Z d d ZGdddZGdddZedS))htonl)Optional)ffilib)ErrorPolicySession __version__z1.0.0)znothing to reportzunspecified failurezunsupported parameterzcouldn't allocate memoryzcouldn't deallocate properlyzcouldn't initializez'can't process as much data as requestedzauthentication failurezcipher failurezreplay check failed (bad index)z#replay check failed (index too old)zalgorithm failed test routinezunsupported operationzno appropriate context foundz$unable to perform desired validationzcan't use key any morezerror in use of socketzerror in use POSIX signalsznonce check failedzcouldn't read datazcouldn't write datazerror parsing datazerror encoding datazerror while using semaphoreszerror while using pfkeyz&error MKI present in packet is invalidz#packet index is too old to considerz#packet index advanced, reset neededc@seZdZdZdS)rz: Error that occurred making a `libsrtp` API call. N)__name__ __module__ __qualname____doc__rrF/home/ubuntu/.local/lib/python3.10/site-packages/pylibsrtp/__init__.pyr1srcCs|tjkr tt|dSN)rsrtp_err_status_okrERRORS)rcrrr _srtp_assert9s  rc @sXeZdZdZejZejZej Z ej Z ej ZejZejZejZdedefdeededededdf d d Zedefd d Zejd eddfdd ZedeefddZejdeeddfddZedefddZedefddZejdeddfddZedefddZ e jdeddfddZ edefddZ!e!jdeddfddZ!dS)rz( Policy for single SRTP stream. Nrkey ssrc_type ssrc_value srtp_profilereturncCs\td|_||_ttt|jj|tt t|jj |||_ ||_ ||_ dS)Nzsrtp_policy_t *)rnew_policy _srtp_profilerr+srtp_crypto_policy_set_from_profile_for_rtp addressofrtp,srtp_crypto_policy_set_from_profile_for_rtcprtcprrr)selfrrrrrrr__init__Us  zPolicy.__init__cCs |jjdkS)z_ Whether retransmissions of packets with the same sequence number are allowed. rrallow_repeat_txr%rrrr(ns zPolicy.allow_repeat_txr(cCs|rdnd|j_dS)Nrrr')r%r(rrrr(uscCs|jdurdSt|jS)z4 The SRTP master key + master salt. N)_Policy__cdatarbufferr)rrrrys  z Policy.keycCs|durd|_tj|j_dSt|jt|j}t |t s#t dt ||kr/t d|tdt ||_||jdt |<|j|j_dS)Nzkey must be bytesz"key must contain at least %d byteszunsigned char[]r)r*rNULLrrr"srtp_profile_get_master_key_lengthr#srtp_profile_get_master_salt_length isinstancebytes TypeErrorlen ValueErrorr)r%rexpected_lengthrrrrs      cCs|jS)z# The SRTP profile. )rr)rrrrszPolicy.srtp_profilecC |jjjS)z The SSRC type. rssrctyper)rrrr zPolicy.ssrc_typecC||jj_dSrr6)r%rrrrrcCr5)z: The SSRC value, if it is not a wildcard. rr7valuer)rrrrr9zPolicy.ssrc_valuecCr:rr<)r%rrrrrr;cCs|jjS)z? The window size to use for replay protection. r window_sizer)rrrr?szPolicy.window_sizer?cCs ||j_dSrr>)r%r?rrrr?s )"r rrrrsrtp_profile_aes128_cm_sha1_80SRTP_PROFILE_AES128_CM_SHA1_80srtp_profile_aes128_cm_sha1_32SRTP_PROFILE_AES128_CM_SHA1_32srtp_profile_aead_aes_128_gcmSRTP_PROFILE_AEAD_AES_128_GCMsrtp_profile_aead_aes_256_gcmSRTP_PROFILE_AEAD_AES_256_GCMssrc_undefinedSSRC_UNDEFINED ssrc_specific SSRC_SPECIFICssrc_any_inboundSSRC_ANY_INBOUNDssrc_any_outboundSSRC_ANY_OUTBOUNDrr0intr&propertyboolr(setterrrrrr?rrrrr>s^ rc@seZdZdZddeeddfddZdeddfddZd eddfd d Z d e de fd dZ d e de fddZ d e de fddZ d e de fddZdddZdS)r z SRTP session, which may comprise several streams. If `policy` is not specified, streams should be added later using the :func:`add_stream` method. NpolicyrcCsbtd}|dur tj}n|j}tt||tdd|_t|j|_ t |dd|_ dS)Nzsrtp_t *zchar[]icSst|dS)Nr)r srtp_dealloc)xrrrsz"Session.__init__..) rrr,rrr srtp_create_cdatar+_buffergc_srtp)r%rTsrtprrrrr&s zSession.__init__cCstt|jd|jdS)z Add a stream to the SRTP session, applying the given `policy` to the stream. :param policy: :class:`Policy` rN)rrsrtp_add_streamr\r)r%rTrrr add_streamszSession.add_streamr7cCstt|jdt|dS)zs Remove the stream with the given `ssrc` from the SRTP session. :param ssrc: :class:`int` rN)rrsrtp_remove_streamr\r)r%r7rrr remove_streamszSession.remove_streampacketcC||tjtS)z Apply SRTP protection to the RTP `packet`. :param packet: :class:`bytes` :rtype: :class:`bytes` )_Session__processr srtp_protectSRTP_MAX_TRAILER_LENr%rbrrrprotectzSession.protectcCrc)z Apply SRTCP protection to the RTCP `packet`. :param packet: :class:`bytes` :rtype: :class:`bytes` )rdrsrtp_protect_rtcpSRTP_MAX_SRTCP_TRAILER_LENrgrrr protect_rtcprizSession.protect_rtcpcC||tjS)z Verify SRTP protection of the SRTP packet. :param packet: :class:`bytes` :rtype: :class:`bytes` )rdrsrtp_unprotectrgrrr unprotectzSession.unprotectcCrm)z Verify SRTCP protection of the SRTCP packet. :param packet: :class:`bytes` :rtype: :class:`bytes` )rdrsrtp_unprotect_rtcprgrrrunprotect_rtcprpzSession.unprotect_rtcprcCst|ts tdt|t|j|krtdtd}t||d<||jdt|<t ||j d|j||jd|dS)Nzpacket must be byteszpacket is too longzint *r) r/r0r1r2rYr3rrrZrr\)r%datafunctrailerlen_prrr __process s   zSession.__processr)r)r rrrrrr&r_rPrar0rhrlrorrrdrrrrr s      r N)socketrtypingr_bindingrr__all__r rrfrk Exceptionrrrr srtp_initrrrrs ! W