JFIF $
d d 7
Viewing File: /usr/lib/python3.9/site-packages/dns/quic/__pycache__/_sync.cpython-39.opt-1.pyc
a
�M�e� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Zd dl
Z d dl Z d dl
mZmZmZmZmZ ee d�r�e jZne jZG dd� de�ZG dd� de�ZG dd � d e�ZdS )
� N)�QUIC_MAX_DATAGRAM�BaseQuicConnection�BaseQuicManager�BaseQuicStream�
UnexpectedEOF� PollSelectorc sX e Zd Z� fdd�Zdd� Zddd�Zdd d
�Zd d � Zd
d� Zdd� Z dd� Z
� Z S )�SyncQuicStreamc s& t � �||� t�� | _t�� | _d S �N)�super�__init__� threading� Condition�_wake_up�Lock�_lock)�self�
connection� stream_id�� __class__� �2/usr/lib/python3.9/site-packages/dns/quic/_sync.pyr s
zSyncQuicStream.__init__c C s� | � |�}| j�2 | j�|�r.W d � d S || _W d � n1 sH0 Y | j�$ | j�|�sntjj �W d � n1 s�0 Y d| _q d S )Nr )
Z_timeout_from_expirationr �_bufferZhaveZ
_expectingr �wait�dns� exception�Timeout)r �amount�
expiration�timeoutr r r �wait_for% s
$ &zSyncQuicStream.wait_forNc C s� | � |�}| �d|� | j�& t�d| j�d��\}W d � n1 sH0 Y | �||� | j� | j�|�W d � S 1 s�0 Y d S )N� z!H)Z_expiration_from_timeoutr r �struct�unpackr �get)r r r �sizer r r �receive1 s
4 zSyncQuicStream.receiveFc C s | � |�}| j�| j||� d S r )Z _encapsulate� _connection�writeZ
_stream_id)r �datagram�is_end�datar r r �send: s
zSyncQuicStream.sendc C s@ | � ||�r<| j� | j�� W d � n1 s20 Y d S r )Z_common_add_inputr �notify)r r+ r* r r r �
_add_input> s zSyncQuicStream._add_inputc C s2 | j � | �� W d � n1 s$0 Y d S r )r Z_close�r r r r �closeC s zSyncQuicStream.closec C s | S r r r/ r r r � __enter__G s zSyncQuicStream.__enter__c C s< | � � | j� | j�� W d � n1 s.0 Y dS �NF)r0 r r- )r �exc_type�exc_val�exc_tbr r r �__exit__J s (zSyncQuicStream.__exit__)N)F) �__name__�
__module__� __qualname__r r r& r, r. r0 r1 r6 �
__classcell__r r r r r s
r c sl e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zdd d
�Zdd� Z ddd�Z
� fdd�Z dd� Z � Z
S )�SyncQuicConnectionc s� t � �||||||� t�| jtjd�| _| jd urrz | j�tj �
| j| j�� W n t yp | j� � � Y n0 | j�
| j� t�� \| _| _| j�d� | j�d� t�� | _d | _t�� | _d S )Nr F)r
r �socketZ_afZ
SOCK_DGRAM�_socket�_sourceZbindr ZinetZlow_level_address_tuple� Exceptionr0 �connect�_peerZ
socketpair� _send_wakeup�_receive_wakeupZ setblockingr �Event�_handshake_complete�_worker_threadr r )r r �address�port�source� source_portZmanagerr r r r R s"
�
zSyncQuicConnection.__init__c C s| d}|dk rx|d7 }z| j �t�}W n ty8 Y d S 0 | j�& | j�|| jt�� � W d � q1 sl0 Y qd S )Nr �
� ) r= �recvr �BlockingIOErrorr r' Zreceive_datagramrA �time)r �countr) r r r �_reade s zSyncQuicConnection._readc C s, z| j �d� W q ty$ Y d S 0 q d S )N� )rC rM rN r/ r r r �
_drain_wakeupp s z SyncQuicConnection._drain_wakeupc C s� �zDt � }|�| jtj| j� |�| jtj| j� | j�s | � d�\}}|�
|�}|D ]\}}|� � qV| j � | �
|� W d � n1 s�0 Y | �� | j � | j�t�� �}W d � n1 s�0 Y |D ].\}}z| j�|� W q� t�y Y q�0 q�q2W | j � d| _W d � n1 �s20 Y | j�� n:| j � d| _W d � n1 �sl0 Y | j�� 0 d S )NFT)�_selector_class�registerr= � selectorsZ
EVENT_READrQ rC rS �_doneZ_get_timer_valuesZselectr+ r Z
_handle_timer�_handle_eventsr' Zdatagrams_to_sendrO r, rN rE �set) r Zselr �interval�items�key�_Z datagramsr) r r r �_workerw s2
(. & �&zSyncQuicConnection._workerc C sL | j � | j�� }W d � n1 s&0 Y |d u r<d S t|tjjj�r�| j � | j� |j
�}W d � n1 sv0 Y |r�|� |j |j
� q t|tjjj�r�| j�� q t|tjjj�r�| j � d| _W d � n1 s�0 Y q t|tjjj�r | j � | j� |j
�}W d � n1 �s,0 Y |r |� dd� q d S )NT� )r r' Z
next_event�
isinstance�aioquicZquicZeventsZStreamDataReceived�_streamsr$ r r. r+ Z
end_streamZHandshakeCompletedrE rY ZConnectionTerminatedrW Z StreamReset)r Zevent�streamr r r rX � s& (, &.z!SyncQuicConnection._handle_eventsFc C sF | j � | j�|||� W d � n1 s,0 Y | j�d� d S )N� )r r' Zsend_stream_datarB r, )r rc r+ r* r r r r( � s .zSyncQuicConnection.writec C s( | j r
d S tj| jd�| _| j�� d S )N)�target)�_closedr �Threadr^ rF �startr/ r r r �run� s zSyncQuicConnection.runNc C sh | j �|�stjj�| j�: | jr&t�| j� d�}t
| |�}|| j |< W d � n1 sZ0 Y |S r2 ) rE r r r r r rW r r' Z get_next_available_stream_idr rb )r r r rc r r r � make_stream� s
(z SyncQuicConnection.make_streamc s6 | j � t� �|� W d � n1 s(0 Y d S r )r r
� close_stream)r r r r r rk � s z SyncQuicConnection.close_streamc C s� | j �\ | jr W d � d S | j�| jd | jd � d| _| j�� | j�d� W d � n1 sh0 Y | j �
� d S )Nr rL Trd ) r rf Z_manager�closedrA r' r0 rB r, rF �joinr/ r r r r0 � s
*zSyncQuicConnection.close)F)N)r7 r8 r9 r rQ rS r^ rX r( ri rj rk r0 r: r r r r r; Q s
r; c sX e Zd Zdejdf� fdd� Zddd�Z� fd d
�Z� fd d �Zd
d� Z dd� Z
� Z S )�SyncQuicManagerNc s t � �||t|� t�� | _d S r )r
r r; r r r )r ZconfZ verify_modeZ server_namer r r r � s zSyncQuicManager.__init__�U r Tc C sP | j �6 | �|||||�\}}|r*|�� |W d � S 1 sB0 Y d S r )r Z_connectri )r rG rH rI rJ Zwant_session_ticketr rh r r r r@ � s
�zSyncQuicManager.connectc s8 | j � t� �||� W d � n1 s*0 Y d S r )r r
rl )r rG rH r r r rl � s zSyncQuicManager.closedc s: | j � t� �|||� W d � n1 s,0 Y d S r )r r
�save_session_ticket)r rG rH Zticketr r r rp � s z#SyncQuicManager.save_session_ticketc C s | S r r r/ r r r r1 � s zSyncQuicManager.__enter__c C s$ t | j�� �}|D ] }|�� qdS r2 )�listZ _connections�valuesr0 )r r3 r4 r5 Z connectionsr r r r r6 � s
zSyncQuicManager.__exit__)ro Nr T) r7 r8 r9 �sslZ
CERT_REQUIREDr r@ rl rp r1 r6 r: r r r r rn � s �
rn )rV r< rs r" r rO Zaioquic.quic.configurationra Zaioquic.quic.connectionZaioquic.quic.eventsZ
dns.exceptionr Zdns.inetZdns.quic._commonr r r r r �hasattrr rT ZSelectSelectorr r; rn r r r r �<module> s"
2|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