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˜ | |¡}| j2 | 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¡ | js | d¡\}}|
|¡}|D ]\}}| ¡ qV| j |
|¡ W d ƒ n1 sŽ0 Y | ¡ | j | j t ¡ ¡}W d ƒ n1 sÌ0 Y |D ].\}}z| j |¡ W qÚ ty 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 s"
2|