File: //lib/python3.9/site-packages/dns/quic/__pycache__/_trio.cpython-39.opt-1.pyc
a
�M�e� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z
d dl Z
d dl m
Z
d dlmZmZmZmZmZ G dd� de�ZG dd� de�ZG dd � d e�ZdS )
� N)� NullContext)�QUIC_MAX_DATAGRAM�AsyncQuicConnection�AsyncQuicManager�BaseQuicStream�
UnexpectedEOFc sX e Zd Z� fdd�Zdd� Zddd�Zdd d
�Zd d � Zd
d� Zdd� Z dd� Z
� Z S )�TrioQuicStreamc s t � �||� t�� | _d S �N)�super�__init__�trio� Condition�_wake_up)�self�
connection� stream_id�� __class__� �2/usr/lib/python3.9/site-packages/dns/quic/_trio.pyr s zTrioQuicStream.__init__c � sl | j �|�rd S || _| j4 I d H �& | j�� I d H W d �I d H q`1 I d H sV0 Y d| _q d S )Nr )�_bufferZhaveZ
_expectingr �wait)r �amountr r r �wait_for s :zTrioQuicStream.wait_forNc � s� |d u rt d �}n
t�|�}|�R | �d�I d H t�d| j�d��\}| �|�I d H | j�|�W d � S 1 sx0 Y tj j
�d S )N� z!H) r r �
move_on_afterr �struct�unpackr �get�dns� exception�Timeout)r �timeout�context�sizer r r �receive' s
*zTrioQuicStream.receiveFc � s& | � |�}| j�| j||�I d H d S r )Z _encapsulate� _connection�writeZ
_stream_id)r �datagram�is_end�datar r r �send3 s
zTrioQuicStream.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_input7 s zTrioQuicStream._add_inputc � s | � � d S r )Z_close�r r r r �close< s zTrioQuicStream.closec � s | S r r r. r r r �
__aenter__A s zTrioQuicStream.__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__D s 4zTrioQuicStream.__aexit__)N)F) �__name__�
__module__� __qualname__r r r% r+ r- r/ r0 r5 �
__classcell__r r r r r s
r c sR e Zd Zd� fdd� Zdd� Zdd� Zdd d
�Zd d � Zdd
d�Zdd� Z � Z
S )�TrioQuicConnectionNc sP t � �||||||� tj�| jtjd�| _t�� | _t�� | _ d | _
d| _ d S )Nr F) r
r r �socket�_afZ
SOCK_DGRAM�_socket�Event�_handshake_complete� _run_done�
_worker_scope�
_send_pending)r r �address�port�source� source_portZmanagerr r r r L s
zTrioQuicConnection.__init__c � s< �z"| j r*| j�tj�| j | j��I d H | j�| j�I d H | j �s| �
d�\}}| j r^d}t j
t �� | d��:| _| j�t�I d H }| j�|| jt�� � W d � n1 s�0 Y d | _| �|� | �� I d H d| _ | j�t�� �}|D ]\}}| j�|�I d H q�q>W d| _ | j�� nd| _ | j�� 0 d S )NFg )ZdeadlineT)�_sourcer= Zbindr ZinetZlow_level_address_tupler< �connect�_peer�_doneZ_get_timer_valuesrB r Z CancelScopeZ current_timerA Zrecvr r& Zreceive_datagram�timeZ
_handle_timer�_handle_eventsZdatagrams_to_sendr+ r? �set)r Z
expiration�intervalr( Z datagrams�_r r r �_workerT s6 �
�4
�zTrioQuicConnection._workerc � 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� � �
)r& Z
next_event�
isinstance�aioquicZquicZeventsZStreamDataReceived�_streamsr r r- r* Z
end_streamZHandshakeCompletedr? rM ZConnectionTerminatedrJ r= r/ Z StreamResetr �sleep)r �countZevent�streamr r r rL x s*
z!TrioQuicConnection._handle_eventsFc � s. | j �|||� d| _| jd ur*| j�� d S )NT)r& Zsend_stream_datarB rA �cancel)r rY r* r) r r r r' � s
zTrioQuicConnection.writec � s` | j r
d S t�� 4 I d H �"}|�| j� W d �I d H qR1 I d H sH0 Y | j�� d S r )�_closedr Z open_nursery�
start_soonrP r@ rM )r �nurseryr r r �run� s
6zTrioQuicConnection.runc � s� |d u rt d �}n
t�|�}|�N | j�� I d H | jr<t�| j�d�}t | |�}|| j
|< |W d � S 1 st0 Y t j j
�d S r1 )r r r r? r rJ r r& Z get_next_available_stream_idr rV r r r! )r r" r# r rY r r r � make_stream� s
"z TrioQuicConnection.make_streamc � s^ | j sZ| j�| jd | jd � d| _ | j�� d| _| jd urJ| j�� | j �
� I d H d S )Nr rR T) r[ Z_manager�closedrI r& r/ rB rA rZ r@ r r. r r r r/ � s
zTrioQuicConnection.close)N)F)N) r6 r7 r8 r rP rL r' r^ r_ r/ r9 r r r r r: K s $
r: c s@ e Zd Zdejdf� fdd� Zd
dd�Zd d
� Zd d � Z� Z S )�TrioQuicManagerNc s t � �||t|� || _d S r )r
r r: �_nursery)r r] ZconfZ verify_modeZ server_namer r r r � s zTrioQuicManager.__init__�U r Tc C s, | � |||||�\}}|r(| j�|j� |S r )Z_connectrb r\ r^ )r rC rD rE rF Zwant_session_ticketr �startr r r rH � s
�zTrioQuicManager.connectc � s | S r r r. r r r r0 � s zTrioQuicManager.__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 zTrioQuicManager.__aexit__)rc Nr T)
r6 r7 r8 �sslZ
CERT_REQUIREDr rH r0 r5 r9 r r r r ra � s � �
ra )r; rg r rK Zaioquic.quic.configurationrU Zaioquic.quic.connectionZaioquic.quic.eventsr Z
dns.exceptionr Zdns.inetZdns._asyncbackendr Zdns.quic._commonr r r r r r r: ra r r r r �<module> s 2m