File: //home/arjun/projects/env/lib/python3.10/site-packages/dns/quic/__pycache__/_trio.cpython-310.pyc
o
!we� � @ 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__� �G/home/arjun/projects/env/lib/python3.10/site-packages/dns/quic/_trio.pyr s zTrioQuicStream.__init__c � sl � | j �|�r
d S || _| j4 I d H � | j�� I d H W d �I d H n 1 I d H s-w Y d| _q)NTr )�_buffer�have�
_expectingr �wait)r �amountr r r �wait_for s � (��zTrioQuicStream.wait_forNc � s� �|d u r
t d �}nt�|�}|�) | �d�I d H t�d| j�d��\}| �|�I d H | j�|�W d � S 1 s=w Y tj j
�)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 )� _encapsulate� _connection�write�
_stream_id)r �datagram�is_end�datar r r �send3 s �
zTrioQuicStream.sendc � s\ �| � ||�r,| j4 I d H � | j�� W d �I d H d S 1 I d H s%w Y d S d S r )�_common_add_inputr �notify)r r. r- r r r �
_add_input7 s � .��zTrioQuicStream._add_inputc � s �| � � d S r )�_close�r r r r �close< s � zTrioQuicStream.closec � � �| S r r r4 r r r �
__aenter__A � �zTrioQuicStream.__aenter__c � sZ �| � � I d H | j4 I d H � | j�� W d �I d H dS 1 I d H s&w Y dS �NF)r5 r r1 )r �exc_type�exc_val�exc_tbr r r � __aexit__D s � ��zTrioQuicStream.__aexit__r �F) �__name__�
__module__� __qualname__r r r' r/ r2 r5 r7 r= �
__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 sj t � �||||||� tj�| jtjd�| _| jr&tj�t j
� | j| j�� t� � | _
t� � | _d | _d S )Nr )r
r r �socket�_af�
SOCK_DGRAM�_socket�_source�bindr! �inet�low_level_address_tuple�Event�_handshake_complete� _run_done�
_worker_scope)r r �address�port�source� source_port�managerr r r r L s
zTrioQuicConnection.__init__c � s �z�| j �| j�I d H | js|| �d�\}}tjt�� | d�� | _| j � t
�I d H }| j � || jd t
�
� � W d � n1 sBw Y d | _| �|� | j �t
�
� �}|D ]
\}}| j �|�I d H qY| �� I d H | jrW d| _| j�� d S W d| _| j�� d S d| _| j�� w )NF)�deadliner T)rG �connect�_peer�_done�_get_timer_valuesr � CancelScope� current_timerO �recvr r) �receive_datagram�time�
_handle_timer�datagrams_to_sendr/ �_handle_eventsrM �set)r �
expiration�intervalr, � datagrams�_r r r �_workerU s8 �
�� �
��� zTrioQuicConnection._workerc � s� �d} | j �� }|d u rd S t|tjjj�r,| j�|j �}|r+|�
|j |j �I d H n&t|tjjj
�r:| j�� nt|tjjj�sJt|tjjj�rRd| _| j�� |d7 }|dkrdd}t�d�I d H q)Nr T� �
)r) �
next_event�
isinstance�aioquic�quic�events�StreamDataReceived�_streamsr r r2 r. �
end_stream�HandshakeCompletedrM rb �ConnectionTerminated� StreamResetrX rG r5 r �sleep)r �count�event�streamr r r ra k s2 �
�
��
�z!TrioQuicConnection._handle_eventsFc � s. �| j �|||� | jd ur| j�� d S d S r )r) �send_stream_datarO �cancel)r rx r. r- r r r r* � s
�
�zTrioQuicConnection.writec � sb �| j rd S t�� 4 I d H �}|�| j� W d �I d H n 1 I d H s%w Y | j�� d S r )�_closedr � open_nursery�
start_soonrg rN rb )r �nurseryr r r �run� s �(�zTrioQuicConnection.runc � s� �|d u r
t d �}nt�|�}|�' | j�� I d H | jr t�| j�d�}t | |�}|| j
|< |W d � S 1 s;w Y t j j
�r9 )r r r rM r rX r r) � get_next_available_stream_idr rp r! r" r# )r r$ r% r rx r r r � make_stream� s �
�z TrioQuicConnection.make_streamc � s^ �| j s-| j�| jd | jd � d| _ | j�� | jd ur#| j�� | j� � I d H d S d S )Nr rh T)
r{ �_manager�closedrW r) r5 rO rz rN r r4 r r r r5 � s �
�zTrioQuicConnection.closer r> ) r? r@ rA r rg ra r* r r� r5 rB r r r r rC K s
rC c s@ e Zd Zdejdf� fdd� Zd dd�Zdd � Zd
d � Z� Z S )
�TrioQuicManagerNc s t � �||t|� || _d S r )r
r rC �_nursery)r r~ �conf� verify_mode� server_namer r r r � s
zTrioQuicManager.__init__�U r c C s* | � ||||�\}}|r| j�|j� |S r )�_connectr� r} r )r rP rQ rR rS r �startr r r rV � s zTrioQuicManager.connectc � r6 r r r4 r r r r7 � r8 zTrioQuicManager.__aenter__c � s, �t | j�� �}|D ] }|�� I d H q
dS r9 )�list� _connections�valuesr5 )r r: r; r< � connectionsr r r r r= � s
�zTrioQuicManager.__aexit__)r� Nr )
r? r@ rA �ssl�
CERT_REQUIREDr rV r7 r= rB r r r r r� � s �
r� )rD r� r r^ �aioquic.quic.configurationrl �aioquic.quic.connection�aioquic.quic.eventsr �
dns.exceptionr! �dns.inet�dns._asyncbackendr �dns.quic._commonr r r r r r rC r� r r r r �<module> s 2\