File: //home/arjun/projects/env/lib/python3.10/site-packages/dns/quic/__pycache__/_sync.cpython-310.pyc
o
!we � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Zd dl
Z d dl Z d dl
mZmZmZmZmZ ee d�rCe 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__� �G/home/arjun/projects/env/lib/python3.10/site-packages/dns/quic/_sync.pyr s
zSyncQuicStream.__init__c C s� | � |�}| j� | j�|�r W d � d S || _W d � n1 s&w Y | j� | j�|�s9tjj �W d � n1 sCw Y d| _q)NTr )
�_timeout_from_expirationr �_buffer�have�
_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 s$w Y | �||� | j� | j�|�W d � S 1 sCw Y d S )N� z!H)�_expiration_from_timeoutr# r �struct�unpackr �get)r r" r! �sizer r r �receive1 s
�
$�zSyncQuicStream.receiveFc C s | � |�}| j�| j||� d S r )� _encapsulate� _connection�write�
_stream_id)r �datagram�is_end�datar r r �send: s
zSyncQuicStream.sendc C sF | � ||�r!| j� | j�� W d � d S 1 sw Y d S d S r )�_common_add_inputr �notify)r r1 r0 r r r �
_add_input> s
"��zSyncQuicStream._add_inputc C s4 | j �
| �� W d � d S 1 sw Y d S r )r �_close�r r r r �closeC s
"�zSyncQuicStream.closec C � | S r r r7 r r r � __enter__G � zSyncQuicStream.__enter__c C s> | � � | j� | j�� W d � dS 1 sw Y dS �NF)r8 r r4 )r �exc_type�exc_val�exc_tbr r r �__exit__J s
��zSyncQuicStream.__exit__r �F) �__name__�
__module__� __qualname__r r# r* r2 r5 r8 r: r@ �
__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�| _| j�| j� t�� \| _ | _
| j
� d� | j� d� | j d urRz| j�
tj�| j | j�� W n
tyQ | j�� � w t�� | _d | _t�� | _d S )Nr F)r
r �socket�_af�
SOCK_DGRAM�_socket�connect�_peer�
socketpair� _send_wakeup�_receive_wakeup� setblocking�_source�bindr �inet�low_level_address_tuple� Exceptionr8 r �Event�_handshake_complete�_worker_threadr r )r r �address�port�source� source_port�managerr r r r R s$
�
�
zSyncQuicConnection.__init__c C s� d}|dk rC|d7 }z| j �t�}W n
ty Y d S w | j� | j�|| jd t�� � W d � n1 s8w Y |dk sd S d S )Nr �
� ) rJ �recvr �BlockingIOErrorr r, �receive_datagramrL �time)r �countr/ r r r �_reade s � ��zSyncQuicConnection._readc C s* z| j �d� W n
ty Y d S w q)NT� )rO r` ra r7 r r r �
_drain_wakeupp s ��z SyncQuicConnection._drain_wakeupc C sV z�t � }|�| jtj| j� |�| jtj| j� | jsq| � d�\}}|�
|�}|D ]\}}|� � q)| j � | �
|� | j�t�� �}W d � n1 sMw Y |D ]\}}z| j�|� W qT tyi Y qTw | �� | jrW | j � d| _W d � n1 s�w Y | j�� d S | j � d| _W d � n1 s�w Y | j�� w )NFT)�_selector_class�registerrJ � selectors�
EVENT_READre rO rg �_done�_get_timer_values�selectr1 r �
_handle_timerr, �datagrams_to_sendrc r2 ra �_handle_eventsrW �set) r �selr! �interval�items�key�_� datagramsr/ r r r �_workerw s<
� ��� �� � zSyncQuicConnection._workerc C s | j �
| j�� }W d � n1 sw Y |d u r d S t|tjjj�rL| j � | j� |j
�}W d � n1 s<w Y |rK|� |j |j
� n4t|tjjj�rZ| j�� n&t|tjjj�sjt|tjjj�r�| j � d| _W d � n1 s{w Y q)NT)r r, �
next_event�
isinstance�aioquic�quic�events�StreamDataReceived�_streamsr( r r5 r1 �
end_stream�HandshakeCompletedrW rr �ConnectionTerminated� StreamResetrl )r �event�streamr r r rq � s0 � ��
�� ��z!SyncQuicConnection._handle_eventsFc C sF | j � | j�|||� W d � n1 sw Y | j�d� d S )N� )r r, �send_stream_datarN r2 )r r� r1 r0 r r r r- � s �zSyncQuicConnection.writec C s( | j rd S tj| jd�| _| j�� d S )N)�target)�_closedr �Threadry rX �startr7 r r r �run� s zSyncQuicConnection.runNc C sj | j �|�s
tjj�| j� | jrt�| j� d�}t
| |�}|| j |< W d � |S 1 s.w Y |S r< ) rW r r r r r rl r r, � get_next_available_stream_idr r� )r r" r r� r r r � make_stream� s
��z SyncQuicConnection.make_streamc s8 | j � t� �|� W d � d S 1 sw Y d S r )r r
� close_stream)r r r r r r� � s "�z SyncQuicConnection.close_streamc C s� | j �/ | jr W d � d S | j�| jd | jd � d| _| j�� | j�d� W d � n1 s5w Y | j �
� d S )Nr r_ Tr� ) r r� �_manager�closedrL r, r8 rN r2 rX �joinr7 r r r r8 � s �
�zSyncQuicConnection.closerA r )rB rC rD r re rg ry rq r- r� r� r� r8 rE r r r r rF Q s
rF c sL e Zd Zdejdf� fdd� Zddd�Z� fdd �Zd
d � Zd d
� Z � Z
S )�SyncQuicManagerNc s t � �||t|� t�� | _d S r )r
r rF r r r )r �conf� verify_mode� server_namer r r r � s zSyncQuicManager.__init__�U r c C sN | j � | �||||�\}}|r|�� |W d � S 1 s w Y d S r )r �_connectr� )r rY rZ r[ r\ r r� r r r rK � s $�zSyncQuicManager.connectc s: | j � t� �||� W d � d S 1 sw Y d S r )r r
r� )r rY rZ r r r r� � s "�zSyncQuicManager.closedc C r9 r r r7 r r r r: � r; zSyncQuicManager.__enter__c C s$ t | j�� �}|D ]}|�� q dS r< )�list� _connections�valuesr8 )r r= r> r? � connectionsr r r r r@ � s
zSyncQuicManager.__exit__)r� Nr ) rB rC rD �ssl�
CERT_REQUIREDr rK r� r: r@ rE r r r r r� � s
r� )rj rG r� r& r rc �aioquic.quic.configurationr| �aioquic.quic.connection�aioquic.quic.events�
dns.exceptionr �dns.inet�dns.quic._commonr r r r r �hasattrr rh �SelectSelectorr rF r� r r r r �<module> s$
2x