JFIF $
d d 7
Viewing File: /usr/lib/python3.9/site-packages/dns/quic/__pycache__/_asyncio.cpython-39.pyc
a
�M�e� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z
d dl Z
d dl Z
d dl
mZmZmZmZmZ G dd� de�ZG dd� de�ZG dd� de�ZdS ) � N)�QUIC_MAX_DATAGRAM�AsyncQuicConnection�AsyncQuicManager�BaseQuicStream�
UnexpectedEOFc s` e Zd Z� fdd�Zdd� Zdd� Zddd �Zdd d �Zd
d� Zdd� Z dd� Z
dd� Z � Z S )�AsyncioQuicStreamc s t � �||� t�� | _d S �N)�super�__init__�asyncio� Condition�_wake_up)�self�
connection� stream_id�� __class__� �5/usr/lib/python3.9/site-packages/dns/quic/_asyncio.pyr
s zAsyncioQuicStream.__init__c � sN | j 4 I d H �& | j �� I d H W d �I d H qJ1 I d H s@0 Y d S r )r
�wait�r r r r �_wait_for_wake_up s z#AsyncioQuicStream._wait_for_wake_upc � sb | � |�}| j�|�rd S || _zt�| �� |�I d H W n tyT tj j
�Y n0 d| _q d S )Nr ) Z_timeout_from_expiration�_bufferZhaveZ
_expectingr �wait_forr � TimeoutError�dns� exception�Timeout)r �amount�
expiration�timeoutr r r r " s
zAsyncioQuicStream.wait_forNc � sP | � |�}| �d|�I d H t�d| j�d��\}| �||�I d H | j�|�S )N� z!H)Z_expiration_from_timeoutr �struct�unpackr �get)r r r �sizer r r �receive. s
zAsyncioQuicStream.receiveFc � s& | � |�}| j�| j||�I d H d S r )Z _encapsulate� _connection�writeZ
_stream_id)r �datagram�is_end�datar r r �send5 s
zAsyncioQuicStream.sendc � sT | � ||�rP| j4 I d H � | j�� W d �I d H qP1 I d H sF0 Y d S r )Z_common_add_inputr
�notify)r r+ r* r r r �
_add_input9 s z AsyncioQuicStream._add_inputc � s | � � d S r )Z_closer r r r �close> s zAsyncioQuicStream.closec � s | S r r r r r r �
__aenter__C s z AsyncioQuicStream.__aenter__c � sV | � � I d H | j4 I d H � | j�� W d �I d H qR1 I d H sH0 Y dS �NF)r/ r
r- )r �exc_type�exc_val�exc_tbr r r � __aexit__F s 4zAsyncioQuicStream.__aexit__)N)F)
�__name__�
__module__� __qualname__r
r r r& r, r. r/ r0 r5 �
__classcell__r r r r r s
r c sb e Zd Zd� fdd� Zdd� Zdd� Zdd � Zd
d � Zdd
d�Zdd� Z ddd�Z
dd� Z � Z S )�AsyncioQuicConnectionNc sJ t � �||||||� d | _t�� | _t�� | _t�� | _d | _ d | _
d S r ) r r
�_socketr �Event�_handshake_complete�_socket_createdr � _wake_timer�_receiver_task� _sender_task)r r �address�port�source� source_portZmanagerr r r r
N s
z AsyncioQuicConnection.__init__c � s �z��z6t j�| j�}t j�d�}|�|tjd| j | j
d | j
d f�I d H | _ | j �
� | j 4 I d H �� | j�s | j �td �I d H \}}|d | j
d ksj|d | j
d kr�qj| j�||t�� � | j4 I d H � | j�� W d �I d H qj1 I d H �s 0 Y qjW d �I d H �q:1 I d H �s00 Y W n t�yP Y n0 W d| _| j4 I d H �" | j�� W d �I d H �q�1 I d H �s�0 Y | j�
� nZd| _| j4 I d H �" | j�� W d �I d H �q�1 I d H �s�0 Y | j�
� 0 d S )Nr r � T)r ZinetZaf_for_addressZ_addressZ asyncbackendZ get_backendZ make_socket�socketZ
SOCK_DGRAM�_source�_peerr; r> �set�_doneZrecvfromr r' Zreceive_datagram�timer? �
notify_all� Exceptionr= )r �afZbackendr) rB r r r � _receiverW s6 �
�$j8 �8z AsyncioQuicConnection._receiverc � sN | j 4 I d H �& | j �� I d H W d �I d H qJ1 I d H s@0 Y d S r )r? r r r r r �_wait_for_wake_timeru s z*AsyncioQuicConnection._wait_for_wake_timerc � s� | j �� I d H | js�| j�t�� �}|D ].\}}|| jks@J �| j�|| jd �I d H q*| � � \}}zt
� | � � |�I d H W n t
y� Y n0 | �|� | �� I d H qd S r )r> r rK r' Zdatagrams_to_sendrL rI r; ZsendtoZ_get_timer_valuesr r rQ rN Z
_handle_timer�_handle_events)r Z datagramsr) rB r �intervalr r r �_sendery s
z AsyncioQuicConnection._senderc � s� d}| j �� }|d u rd S t|tjjj�rT| j�|j �}|r�|�
|j |j �I d H nrt|tjjj
�rp| j�� nVt|tjjj�r�d| _| j�� n4t|tjjj�r�| j�|j �}|r�|�
dd�I d H |d7 }|dkrd}t�d�I d H qd S )Nr T� rF �
)r' Z
next_event�
isinstance�aioquicZquicZeventsZStreamDataReceived�_streamsr$ r r. r+ Z
end_streamZHandshakeCompletedr= rJ ZConnectionTerminatedrK r@ �cancelZ StreamResetr �sleep)r �countZevent�streamr r r rR � s*
z$AsyncioQuicConnection._handle_eventsFc � sX | j �|||� | j4 I d H � | j�� W d �I d H qT1 I d H sJ0 Y d S r )r' Zsend_stream_datar? rM )r r] r+ r* r r r r( � s zAsyncioQuicConnection.writec C s. | j r
d S t�| �� �| _t�| �� �| _d S r )�_closedr ZTaskrP r@ rT rA r r r r �run� s zAsyncioQuicConnection.runc � sf z t �| j�� |�I d H W n ty6 tjj�Y n0 | jrBt �| j
� d�}t | |�}|| j
|< |S r1 )r r r= r r r r r rK r r' Z get_next_available_stream_idr rY )r r r r] r r r � make_stream� s
z!AsyncioQuicConnection.make_streamc � s� | j s�| j�| jd | jd � d| _ | j�� | j�� | j4 I d H � | j� � W d �I d H q~1 I d H st0 Y z| j
I d H W n t j y� Y n0 z| j
I d H W n t j y� Y n0 | j�� I d H d S )Nr rF T)r^ Z_manager�closedrI r' r/ r> rJ r? rM r@ r ZCancelledErrorrA r; r r r r r/ � s
4zAsyncioQuicConnection.close)N)F)N)
r6 r7 r8 r
rP rQ rT rR r( r_ r` r/ r9 r r r r r: M s
r: c s@ e Zd Zdejdf� fdd� Zd
dd�Zd d
� Zd d � Z� Z S )�AsyncioQuicManagerNc s t � �||t|� d S r )r r
r: )r ZconfZ verify_modeZ server_namer r r r
� s zAsyncioQuicManager.__init__�U r Tc C s& | � |||||�\}}|r"|�� |S r )Z_connectr_ )r rB rC rD rE Zwant_session_ticketr �startr r r �connect� s
�zAsyncioQuicManager.connectc � s | S r r r r r r r0 � s z AsyncioQuicManager.__aenter__c � s* t | j�� �}|D ]}|�� I d H qdS r1 )�listZ _connections�valuesr/ )r r2 r3 r4 Z connectionsr r r r r5 � s z AsyncioQuicManager.__aexit__)rc Nr T)
r6 r7 r8 �sslZ
CERT_REQUIREDr
re r0 r5 r9 r r r r rb � s
�
rb )r rG rh r" rL Zaioquic.quic.configurationrX Zaioquic.quic.connectionZaioquic.quic.eventsZdns.asyncbackendr Z
dns.exceptionZdns.inetZdns.quic._commonr r r r r r r: rb r r r r �<module> s 4 Back to Directory
nL+D550H?Mx ,D"v]qv;6*Zqn)ZP 0 !1 A "#a$2Qr D8 a Ri[f\mIykIw0 cuFcRı?lO7к_f˓[C$殷WF<_W ԣs KcëIzyQy /_LKℂ;C",pFA: /]=H ~,ls/9ć:[=/#f;)x {ٛEQ )~ =𘙲r*2~ a _V=' kumF D}KYYC)({ *g&f`툪ry`=^cJ.I](*`wq 1dđ#̩͑0;H]u搂@:~וKL Nsh}OIR*8:2 !lDJVo( 3=M(zȰ+i*NA r6KnSl )!JJӁ* %݉?|D}d5:eP0R;{$X'xF@.ÊB {,WJuQɲRI;9QE琯62fT.DUJ;*cP A\ILNj!J۱+O\ ͔]ޒ SJȧc%ANolՎprULZԛerE2=XDXgVQeӓk yP7U*omQIs,K`)6\G3t?pgjrmۛجwluGtf h9uyP0D;Uڽ"OXlif$)&|ML0Zrm1[HXPlPR0'G=i2N +0e2]] 9VTPO7h(F*癈 '=QVZDF,d߬~TX
G[`le69CR(!S2!P
<0x < !1AQ "Raq02Br#SCTb
?Ζ"]mH5WR7k.ۛ!}Q~+yԏz|@T20S~Kek *zFf^2X*(@8r?CIuI|֓>^ExLgNUY+{.RѪ
τVYTD I62'8Y2 7'\TP.6d&˦@Vqi|8- OΕ]ʔ U=TL8=;6c| !qfF3aů&~$l}'NWUs$Uk^SV:U# 6w++s&r+nڐ{@29 gL
u"TÙM=6 (^"7r} =6YݾlCuhquympǦ
G jhsǜNlɻ}o7 #S6aw4!OSrD57%|?x>L
|/nD6?/8w#[)L7+6〼T ATg ! %5MmZ/c-{1_Je"|^$'O&ޱմ Trb$ w)R$&
N1EtdU3Uȉ1pM" N*(DN yd96.(jQ)X
5cQɎMyW? Q*!R>6=7)Xj5`J]e 8%t!+'!1Q 5 !1 AQaqё#2"0BRb ? Gt^## .llQT $v,,m 㵜5 ubV =sY+@d{N! dnO< .-B; _wJt6;QJd.Qc%p{ 1,sNDdFHI0ГoXшe黅XۢF:)[FGXƹ/w_cMeD,ʡcc. WDtA$j@:) -# u
c1<@ۗ9F)KJ-hpP]_x[qBlbp ʖw q"LFGd ƶ*s+ډ_Zc"? %t[IP 6J]#=ɺVvvCGsGh1 >)6|ey?Lӣm,4GWUi` ]uJVoV DG< SB 6ϏQ@ TiUly OU0kfV~~}SZ@*WUUi##; s/[ =!7}" WN]'(L! ~y5g9T̅JkbM' +s:S + B)v@Mj
e Cf jE 0Y\QnzG1д~Wo{T 9?`Rmyhsy3!HAD]mc1~2L Su7xT;j$`}4->L#vzŏ ILS ֭ T { rjGKC; bpU= -`BsK.SFw4Mq]ZdHS0)tLg