o ]iW@sbddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZmZmZddlZddlmZmZmZddlmZmZmZddlmZeeZdZdZ d Z!d Z"e#Z$e#Z%e &Z'Gd d d ej(Z)d e*dej+fddZ,d e*ddfddZ-dedede.de/fddZ0de.de.de1e2fddZ3de/de gdfde4e2e/fd ee2d!ee2d"e.d#e2de4edffd$d%Z5d&dd'e4e2e/fde4edffd(d)Z6d*e1d+de.ddfd,d-Z7d.e2ddfd/d0Z8d1edefd2d3Z9d.e2ddfd4d5Z:Gd6d+d+Z;Gd7ddej<Z=Gd8d9d9Z>Gd:d;d;e>Z?GdN)Callable)OptionalUnioncast)mdnsstunturn) Candidatecandidate_foundationcandidate_priority) random_stringc@seZdZdZ dZdS)TransportPolicyrrN)__name__ __module__ __qualname__ALLRELAYrr>/home/ubuntu/.local/lib/python3.10/site-packages/aioice/ice.pyr#s r subscriberreturnc sttdstt_dt_tt_tj4IdH tjdur't IdHt_tj |WdIdHtjS1IdHs?wYtjSNlock) hasattr_mdnsasyncioLockrprotocolset subscribersrcreate_mdns_protocoladdrrrrget_or_create_mdns_protocol1s   r'c s~ttdr=tj4IdH#tj|tjr%tjs%tjIdHdt_WdIdHdS1IdHs6wYdSdSr)rrrr#discardr!closer&rrrunref_mdns_protocol=s   .r*localremoteice_controllingcCsL|r|jp|j}|r |jp|j}dt||dt||||kr#dp$dS)zJ See RFC 5245 - 5.7.2. Computing Pair Priority and Ordering Pairs lrrr)priorityminmax)r+r,r-GDrrrcandidate_pair_priorityFs,r3use_ipv4use_ipv6cCszg}tD]4}|jD].}t|jtr!|r!|jdkr!||jq |r9|jddkr9|jddkr9||jdq q|S)z! Get local IP addresses. z 127.0.0.1rz::1r)ifaddr get_adaptersips isinstanceipstrappend)r4r5 addressesadapterr:rrrget_host_addressesQs   r? componentprotocol_factory StunProtocol turn_server turn_username turn_passwordturn_sslturn_transportc stj||||||dIdH\}}|jd} |jd} ttdd| d|dt|d| d| dd| d| dd |_|j|fS) zA Connect to a TURN server to obtain a relayed candidate. ) server_addrusernamepasswordssl transportNsocknamerelated_addressrelayudprr foundationr@rLr.hostporttyperN related_port)r create_turn_endpointrLget_extra_infor r r local_candidate) r@rArCrDrErFrG_r!candidate_addressrNrrrrelayed_candidate_s.   r\r! stun_serverc st}|dtj|dIdH|df}tjtjjtj j d}| ||IdH\}}|j }t tdd|j|j|jt|jd|jdd|jddd|j|jd dfS) zC Query STUN server to obtain a server-reflexive candidate. Nrrmessage_method message_classsrflxrPXOR-MAPPED-ADDRESSrQ)rget_event_looprun_in_executorsocket gethostbynamerMessageMethodBINDINGClassREQUESTrequestrYr r rSr@rLr attributesrT)r!r]looprlresponserZrYrrrserver_reflexive_candidates.      rppairs CandidatePaircs&dtdtffdd }|j|ddS)z) Sort a list of candidate pairs. pairrcst|j|j SN)r3rYremote_candidate)rsr-rr pair_prioritys z+sort_candidate_pairs..pair_prioritykeyN)rrintsort)rqr-rwrrvrsort_candidate_pairssr|valuecCtd|s tddS)zk Check the password is well-formed. See RFC 5245 - 15.4. "ice-ufrag" and "ice-pwd" Attributes z^[a-zA-Z0-9+/]{22,256}$z$Password must satisfy 22*256ice-charNrematch ValueErrorr}rrrvalidate_password r candidatecCs(|jdvr td|jt|j|S)z2 Check the remote candidate is supported. )rSrOrazUnexpected candidate type "%s")rUr ipaddress ip_addressrS)rrrrvalidate_remote_candidates  rcCr~)zk Check the username is well-formed. See RFC 5245 - 15.4. "ice-ufrag" and "ice-pwd" Attributes z^[a-zA-Z0-9+/]{4,256}$z#Username must satisfy 4*256ice-charNrrrrrvalidate_usernamerrc@seZdZdddeddfddZdefdd Zedefd d Z ede eeffd d Z edefddZ ede eeffddZ GdddejZdS)rrr!rBrurNcCs,d|_d|_||_||_d|_tjj|_dS)NF) task nominatedr!ruremote_nominatedrrStateFROZENstate)selfr!rurrr__init__s zCandidatePair.__init__cCsd|j|jfS)NzCandidatePair(%s -> %s)) local_addr remote_addrrrrr__repr__zCandidatePair.__repr__cC|jjSrt)rYr@rrrrr@zCandidatePair.componentcC|jj|jjfSrt)rYrSrTrrrrrzCandidatePair.local_addrcCrrt)r!rYrrrrrYrzCandidatePair.local_candidatecCrrt)rurSrTrrrrrrzCandidatePair.remote_addrc@s eZdZdZdZdZdZdZdS)zCandidatePair.StaterrrN)rrrrWAITING IN_PROGRESS SUCCEEDEDFAILEDrrrrrs r)rrrr rr;rpropertyrzr@tuplerrYrenumEnumrrrrrrrsc@s eZdZd%ddZdeddfdd Zd ejddfd d Zd e e e fde ddfddZ deddfddZd&ddZ  d'dejde e efdee deede eje e efff ddZd e de e efddfddZdejde e efddfddZde d eddfd!d"Zde fd#d$ZdS)(rBreceiver ConnectionrNcCs0t|_tt|_d|_||_d|_i|_ dSrt) rFuture_StunProtocol__closednext protocol_ididrYrrL transactions)rrrrrrs   zStunProtocol.__init__exccCs8|d||js|jdd|jddSdS)Nzconnection_lost(%s)T)_StunProtocol__log_debugrdoner data_received set_resultrrrrrconnection_losts  zStunProtocol.connection_lostrLcCs|d|ttj||_dS)Nzconnection_made(%s))rrrDatagramTransportrL)rrLrrrconnection_mades zStunProtocol.connection_madedataaddrcCs|d|df}tt|}zt|}|d||Wnty.|j||jj YdSw|j tj j ks=|j tj j krQ|j|jvrQ|j|j}|||dS|j tj jkrc|j||||dSdS)Nrrz< %s %s)rbytesr parse_messagerrrrrYr@r`rjRESPONSEERRORtransaction_idrresponse_receivedrkrequest_received)rrrmessage transactionrrrdatagram_received s"     zStunProtocol.datagram_receivedcCs|d|dS)Nzerror_received(%s))rrrrrerror_receivedrzStunProtocol.error_receivedcs|j|jIdHdSrt)rLr)rrrrrr)#s zStunProtocol.closerl integrity_keyretransmissionscsj|j|jvs J|dur||tj||||d}||j|j<z |IdHW|j|j=S|j|j=w)zE Execute a STUN transaction and return the response. N)r)rradd_message_integrityr Transactionrun)rrlrrrrrrrrl's   zStunProtocol.requestcs|j||dSrt)rLsendto)rrrrrr send_data?szStunProtocol.send_datarcCs$|d|||jt||dS)z& Send a STUN message. z> %s %sN)rrLrr)rrrrrr send_stunBszStunProtocol.send_stunmsgargscGs tjd||j|g|RdS)Nz%s %s )loggerdebugrrrrrrr __log_debugIs zStunProtocol.__log_debugcC d|jS)Nz protocol(%s))rrrrrrL zStunProtocol.__repr__)rrrNrNNN)rrrr Exceptionrr BaseTransportrrrr;rrrr)rrgrzrrlrrobjectrrrrrrrBs0     c@ eZdZdS)ConnectionEventNrrrrrrrrPrc@r)ConnectionClosedNrrrrrrTrrc@sfeZdZdZdddddddddejddf deded ee e efd ee e efd ee d ee d ede dedededee dee ddfddZ e de efddZe de fddZe de fddZe de efddZdeeddfd d!Zdqd"d#Zd$edeefd%d&Zdqd'd(Zdqd)d*Zdeefd+d,Zdefd-d.Zde eeffd/d0Zd1eddfd2d3Zd1ed$eddfd4d5Zd$ed6e d7e ddfd8d9Zd:e d;ede!j"fdd?Z$d@e!j"dAe e efdBe%ddfdCdDZ&defdEdFZ'd:e ddfdGdHZ(d:e ddfdIdJZ)d:e dKe j*ddfdLdMZ+dNeddfdOdPZ,dBe%dedee fdQdRZ- Sdrd$edTe e dUede efdVdWZ.dqdXdYZ/dqdZd[Z0d1eed$eeddfd\d]Z1d@e!j"dAe e efdBe%d^eddf d_d`Z2dae!j"dAe e efdBe%dbe ee fddf dcddZ3dqdedfZ4deddfdgdhZ5dqdidjZ6dke dle7ddfdmdnZ8de fdodpZ9dS)sra An ICE connection for a single media stream. :param ice_controlling: Whether the local peer has the controlling role. :param components: The number of components. :param stun_server: The address of the STUN server or `None`. :param turn_server: The address of the TURN server or `None`. :param turn_username: The username for the TURN server. :param turn_password: The password for the TURN server. :param turn_ssl: Whether to use TLS for the TURN server. :param turn_transport: The transport for TURN server, `"udp"` or `"tcp"`. :param use_ipv4: Whether to use IPv4 candidates. :param use_ipv6: Whether to use IPv6 candidates. :param transport_policy: Transport policy. :param local_username: An optional local username, otherwise a random one will be generated. :param local_password: An optional local password, otherwise a random one will be generated. rNFrPTr- componentsr]rCrDrErFrGr4r5transport_policylocal_usernamelocal_passwordrcCsN||_| dur td} nt| | durtd} nt| d|_d|_d|_||_||_||_ ||_ ||_ ||_ d|_ ttd|d|_g|_d|_t|_g|_d|_d|_tt|_g|_d|_d|_| |_| |_ i|_!t|_"g|_#g|_$d|_%d|_&t|_'t()d|_*| |_+| |_,|dur|dur| t-j.krt/d| |_0dS)NrFr@z:Relay transport policy requires a STUN and/or TURN server.)1r-r rrremote_is_literemote_usernameremote_passwordr]rCrDrErFrG_closedr"range _components _check_list_check_list_donerQueue_check_list_state _early_checks_early_checks_done _event_waiterr connection_id_id_local_candidates_local_candidates_end_local_candidates_start_local_password_local_username _nominated _nominating _protocols_remote_candidates_remote_candidates_end_query_consent_task_queuesecretsrandbits _tie_breaker _use_ipv4 _use_ipv6rrr_transport_policy)rr-rr]rCrDrErFrGr4r5rrrrrrrmsb       zConnection.__init__cC|jddS)zS Local candidates, automatically set by :meth:`gather_candidates`. N)rrrrrlocal_candidatesszConnection.local_candidatescC|jS)z; Local password, set at construction time. )rrrrrrzConnection.local_passwordcCr)z; Local username, set at construction time. )rrrrrrrzConnection.local_usernamecCr)zk Remote candidates, which you need to populate using :meth:`add_remote_candidate`. N)rrrrrremote_candidatesszConnection.remote_candidatesruc sP|jrtd|dur|d|_dSt|jrYt|IdH}||jIdH}|dur;|d|jddS|d|jd|t |}||_| |IdHdSzt |Wnty~}z|d|jd|WYd}~dSd}~ww|j ||jD]}|j|r|||st||}|j |q|dS)z Add a remote candidate or signal end-of-candidates. To signal end-of-candidates, pass `None`. :param remote_candidate: A :class:`Candidate` instance or `None`. z4Cannot add remote candidate after end-of-candidates.NTzRemote candidate "z" could not be resolvedz" resolved to z" is not valid: )rr_prune_componentsris_mdns_hostnamerSr'resolve_Connection__log_infocopyadd_remote_candidaterrr<rrY can_pair_with _find_pairrrrsort_check_list)rru mdns_protocolrcopy_candidateer!rsrrrrsT          zConnection.add_remote_candidatecsfjs1d_tjjdfddjD}tj|IdHD] }j|7_q"d_dSdS)zt Gather local candidates. You **must** call this coroutine before calling :meth:`connect`. T)r4r5csg|] }j|dqS))r@r=)get_component_candidates).0r@r=rrr s z0Connection.gather_candidates..N) rr?rrrrgatherrr)rcoros candidatesrrrgather_candidatess  zConnection.gather_candidatesr@cCs.t|jdddD] }|j|kr|Sq dS)z Get the default local candidate for the specified component. :param component: The component whose default candidate is requested. cSrrt)r.xrrr,z2Connection.get_default_candidate..rxN)sortedrr@)rr@rrrrget_default_candidate&s  z Connection.get_default_candidatecs|jstd|jdus|jdurtd|jD]}|jD]}|j|r7|||s7t ||}|j |qq| | |jD]}|j|qDg|_d|_ |sXn tdIdHqS|j rm|jIdH}nt}|j D] }|jr||jqr|tkrtdt||_dS)z Perform ICE handshake. This coroutine returns if a candidate pair was successfuly nominated and raises an exception otherwise. z,Local candidates gathering was not performedNz&Remote username or password is missingTg{Gz?zICE negotiation failed)rConnectionErrorrrrrrYrrrrrr<r_unfreeze_initialrcheck_incomingrcheck_periodicrsleeprget ICE_FAILEDrcancel ICE_COMPLETED create_task query_consentr)rrur!rs early_checkrescheckrrrconnect1sJ         zConnection.connectcs|jr!|js!|jz|jIdHWn tjy Ynw|jr-|js-|jt t |IdH|j |j D] }|IdHq<|j |j |js^|td|_dSdS)z' Close the connection. NT)rrr+rCancelledErrorrrr put_nowaitr*r*rclearrr)rr _emit_eventr)rr!rrrr)gs(         zConnection.closecsB|jdus Jd|jrdSt}||_t|jIdHS)z Return the next `ConnectionEvent` or `None` if the connection is already closed. This method may only be called once at a time. Nzalready awaiting event)rrrrc create_futureshield)rrnrrr get_events zConnection.get_eventcs|IdH\}}|S)z Receive the next datagram. The return value is a `bytes` object representing the data received. If the connection is not established, a `ConnectionError` is raised. N)recvfromrrr@rrrrecvszConnection.recvcs<t|js td|jIdH}|ddurtd|S)aD Receive the next datagram. The return value is a `(bytes, component)` tuple where `bytes` is a bytes object representing the data received and `component` is the component on which the data was received. If the connection is not established, a `ConnectionError` is raised. z"Cannot receive data, not connectedNrz$Connection lost while receiving data)lenrr$rr))rresultrrrr:s  zConnection.recvfromrcs||dIdHdS)z Send a datagram on the first component. If the connection is not established, a `ConnectionError` is raised. :param data: The data to be sent. rN)r)rrrrrsendszConnection.sendcs4|j|}|r|j||jIdHdStd)z Send a datagram on the specified component. If the connection is not established, a `ConnectionError` is raised. :param data: The data to be sent. :param component: The component on which to send the data. NzCannot send data, not connected)rr)r!rrr$)rrr@ active_pairrrrrs  zConnection.sendtolocal_foundationremote_foundationcCsxd}|jD]}|jj|kr|jj|kr|}nqd}|jD]}|j|kr+|j|kr+|}q|r0|s2Jt|||j|<dS)z Force the selected candidate pair. If the remote party does not support ICE, you should using this instead of calling :meth:`connect`. N)rrYr@rRrrrr)rr@rArBr!prucrrrset_selected_pairs     zConnection.set_selected_pairrsnominatecCsvd|j|jf}tjtjjtjjd}||jd<t |j d|jd<|j r3|j |jd<|r1d|jd<|S|j |jd<|S) N%s:%sr^USERNAMEprflxPRIORITYICE-CONTROLLING USE-CANDIDATEICE-CONTROLLED) rrrrgrhrirjrkrmr r@r-r)rrsrF tx_usernamerlrrr build_requests     zConnection.build_requestcCsPd|_|jtjjkrl|jr2||j|j<|jD]}|j|jkr1|jtjj tjj fvr1| |tjj qt |jt |jkrO|jsM|d|jtd|_dS|jD]}|jj|jjkrk|jtjj krk| |tjj qR|jD]}|jtjjtjj fvrdSqo|js|jD] }|jtjjkrdSq|js|d|jtd|_dSdS)Nz ICE completedTz ICE failed)rrrrrrrrr@rrr check_staterr=rrr rr4r,rYrRr-r*)rrsrCrrrcheck_completesN          zConnection.check_completerrr!c Cs2|jj}d}|jD]}|j|dkr$|j|dkr$|}|j|ks"Jnq |durIttd|d|jd|d|ddd}|j|| d || ||}|durgt ||}t j j |_|j|||jt j j t j jfvrw||d |jvr|jsd |_|jt j jkrd |_||dSdSdSdS) z4 Handle a succesful incoming check. Nrr rPrJrIrRr@rLr.rSrTrUz&Discovered peer reflexive candidate %srLT)rYr@rrSrTr r rmr<r rrrrrrrrrcheck_start_taskr-rrrrQ)rrrr!r@rurDrsrrrr&,sF       zConnection.check_incomingcCsf|jD]}|jtjjkr||dSq|jD]}|jtjjkr)||dSq|js1|j SdS)NTF) rrrrrrrTrrrrrsrrrr']s    zConnection.check_periodicc s ||tjj|jo|j }|j||d}z|jj||j |j ddIdH\}}WnUt j y}zH|jri|jjddddkrid |jvrP|jd d n d |jvr[|jd d ||IdHWYd}~S||tjj||WYd}~dSd}~ww||j kr|d|||tjj||dS|s|jrd |_nR|jr|j|jvr|d||j|j|j|d d}z|jj||j |j ddIdHWnt j y|d|||tjj||YdSwd |_||tjj||dS)z! Starts a check. rFutf8rN ERROR-CODErrrKFrvrMTz)Check %s failed : source address mismatchzCheck %s nominating pairz)Check %s failed : could not nominate pair)rPrrrrr-rrOr!rlrrencoderTransactionErrorrormr) switch_role check_startrrQr rrr@rr%r)rrsrFrlrorrrrrr^psd             zConnection.check_startcCs$|jdurt|||_dSdS)zC Starts a check in a task, unless already started. N)rrr-r^rUrrrrTs zConnection.check_start_taskrcCs|d||j|||_dS)z/ Updates the state of a check. zCheck %s %s -> %sN)r r)rrsrrrrrPs zConnection.check_stateeventcCs(|jdur|j}d|_||dSdSrt)rr)rr_waiterrrrr6s zConnection._emit_eventcCs,|jD]}|j|kr|j|kr|SqdS)z: Find a candidate pair in the check list. N)rr!ru)rr!rursrrrrs zConnection._find_pairrr=timeoutc sg}t}g}|D]s}z'|jfdd|dfdIdH\}} |d} | dur3| tjtjtj Wnt yM} z d|| WYd} ~ q d} ~ ww| | | j d} ttdd | d|d t|d| d| d dd | _jtjkr~| | jq j|7_g} jr|D]} t| jjjd kr| tt| jqjr| tt|fd djjj j!j"dt#| rtj$| |dIdH\}}|D]}|%dur|&\}} | || durj | q|D]}|'q|S)NctSrtrBrrrrr z5Connection.get_component_candidates..r)rrezCould not bind to %s - %srMrSrPrrSrcrbrtrcrrrrr rd)r@rArCrDrErFrG)ra)(rrccreate_datagram_endpointrX setsockoptre SOL_SOCKET SO_RCVBUFr UDP_SOCKET_BUFFER_SIZEOSErrorr r<rLr r r rYrrrrr]rrrSversionr-rprCr\rDrErFrGr=wait exceptionr>r+)rr@r=rarrnhost_protocolsaddressrLr!sockrr[tasksrpendingrrrrrrs             z#Connection.get_component_candidatescCs>ttdd|j}|j|}|r|d|||_dSdS)z Remove components for which the remote party did not provide any candidates. This can only be determined after end-of-candidates. cSrrt)r@rrrrr #r!z.Connection._prune_components..z%Components %s have no candidate pairsN)r"maprrr )rseen_componentsmissing_componentsrrrr s  zConnection._prune_componentsc sd} ttddtIdH|jD]C}|j|dd}z|jj||j |j ddd IdHd}Wnt j yF|d 7}Ynw|tkr\|d d|_|IdHSqq) z8 Periodically check consent (RFC 7675). rTg?g?NFrVrW)rrrzConsent to send expired)rr(CONSENT_INTERVALrandomrvaluesrOr!rlrrr[rr\CONSENT_FAILURESr rr))rfailuresrsrlrrrr.+s.     zConnection.query_consentcCs|j||fdSrt)rr4r;rrrrEszConnection.data_receivedraw_datac Cs|jtjjkr||||ddSz&tj||jdd|jdur6d|j |jf}|j d|kr6t dWnt yI||||dYdSw|j rpd|j vrp|d|j|j dkri||||d dS|jd d n%|j sd |j vr|d |j|j d kr||||d dS|jdd tjtjjtjj|jd}||j d<||jd||||js|js|j|||fdS||||dS)N)iz Bad RequestrWrXrGrHzWrong usernamerKz)Role conflict, expected to be controlling)rZz Role ConflictFrvrMz(Role conflict, expected to be controlledTr_r`rrb)r_rrhri respond_errorrrr[rrrmr)rr-r rr]rgrjrrrrrrrr<r&)rrrr!r{ rx_usernamerorrrrHsN         zConnection.request_receivedrl error_codecCsDtj|jtjj|jd}||jd<||j d| ||dS)Nr|rYrW) rrgr_rjrrrmrrr[r)rrlrr!rrorrrr}}s zConnection.respond_errorcCst|j|jdSrt)r|rr-rrrrrszConnection.sort_check_listcCs&|d|rdpd||_|dS)NzSwitching to %s role controlling controlled)r r-r)rr-rrrr]s   zConnection.switch_rolecCsd}|jD]}|jt|jkr|}nq|durdS|jtjjkr)||tjj t |j j }|jD]$}|j|jkrV|j j |vrV|jtjjkrV||tjj | |j j q2dSrt)rr@r/rrrrrrrPrr"rYrRr%)r first_pairrsseen_foundationsrrrr%s&     zConnection._unfreeze_initialrrcGstjd||g|RdS)Nz%s )rinforrrr __log_infoszConnection.__log_infocCr)NzConnection(%s))rrrrrrrzConnection.__repr__r)r):rrr__doc__rrboolrzrrr;rrlistr rrrr rrr#r2r)rr9rr<r:r?rrErrrrgrOrQrBr&r'r^rTrrPr6rrr r.rrr}rr]r%rr rrrrrrXs       S 6 6   6  1=   Q   5    r)Arrrr itertoolsloggingrwrrre threadingcollections.abcrtypingrrrr6rrr rr r r utilsr getLoggerrrr,r*ryrvcountrrr+rrrr MDnsProtocolr'r*rrzr3rr;r?rr\rpr|rrrrrDatagramProtocolrBrrrrrrrs           '   !    $Z