File: //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