File: //home/arjun/projects/env/lib/python3.10/site-packages/dns/quic/__pycache__/_common.cpython-310.pyc
o
!we � @ s� d dl Z d dlZd dlZd dlmZmZ d dlZd dlZd dl Z
dZ G dd� de �Z
G dd� d�ZG dd � d �ZG d
d � d �ZG d d
� d
e�ZG dd� d�ZG dd� de�ZdS )� N)�Any�Optionali c @ s e Zd ZdS )�
UnexpectedEOFN)�__name__�
__module__� __qualname__� r r �I/home/arjun/projects/env/lib/python3.10/site-packages/dns/quic/_common.pyr s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �Bufferc C s d| _ d| _d S )N� F)�_buffer� _seen_end��selfr r r �__init__ s
zBuffer.__init__c C s* | j rd S | j|7 _|rd| _ d S d S )NT)r
r �r �data�is_endr r r �put s
�z
Buffer.putc C s t | j�|kr dS | jrt�dS )NTF)�lenr r
r )r �amountr r r �have s
z Buffer.havec C � | j S �N)r
r r r r �seen_end' � zBuffer.seen_endc C s0 | � |�sJ �| jd |� }| j|d � | _|S r )r r )r r r r r r �get* s z
Buffer.getN)r r r r r r r r r r r r r
s r
c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�BaseQuicStreamc C s || _ || _t� | _d| _d S �Nr )� _connection�
_stream_idr
r �
_expecting)r �
connection� stream_idr r r r 2 s
zBaseQuicStream.__init__c C r r )r r r r r �id8 r zBaseQuicStream.idc C s |d ur t � � | }|S d }|S r )�time)r �timeout�
expirationr r r �_expiration_from_timeout; s
�z'BaseQuicStream._expiration_from_timeoutc C s& |d urt |t�� d�}|S d }|S )Ng )�maxr% )r r' r& r r r �_timeout_from_expirationB s
�z'BaseQuicStream._timeout_from_expirationc C s t |�}t�d|�| S )Nz!H)r �struct�pack)r �datagram�lr r r � _encapsulateL s zBaseQuicStream._encapsulatec C s& | j �||� | jdko| j �| j�S r )r r r! r r r r r �_common_add_inputP s z BaseQuicStream._common_add_inputc C s | j �| j� | j�dd� d S )Nr T)r � close_streamr r r r r r r �_closeT s zBaseQuicStream._closeN)
r r r r r$ r( r* r/ r0 r2 r r r r r 1 s
r c @ s2 e Zd Z d dd�Zdd� Zd
dd �Zd
d � ZdS )�BaseQuicConnectionNr c C s� d| _ || _|| _|| _d| _|| _i | _tj� |�| _
tj� ||f�| _ |d u rA|dkrA| j
t
jkr6d}n | j
t
jkr?d}nt�|rJ||f| _d S d | _d S )NFr z0.0.0.0z::)�_doner �_address�_port�_closed�_manager�_streams�dns�inet�af_for_address�_af�low_level_address_tuple�_peer�socket�AF_INET�AF_INET6�NotImplementedError�_source)r r" �address�port�source� source_port�managerr r r r Z s$
zBaseQuicConnection.__init__c C s | j |= d S r )r9 )r r# r r r r1 r s z BaseQuicConnection.close_streamTc C sL t � � }| j�� }|d u r|d }t|| d�}| jr"|r"t|d�}||fS )Ni r g�������?)r% r � get_timerr) r7 �min)r �closed_is_special�nowr' �intervalr r r �_get_timer_valuesu s
z$BaseQuicConnection._get_timer_valuesc C s$ t � � }||kr| j�|� d S d S r )r% r � handle_timer)r r' rM r r r �
_handle_timer� s �z BaseQuicConnection._handle_timer)Nr N)T)r r r r r1 rO rQ r r r r r3 Y s
�
r3 c @ s$ e Zd Zddee defdd�ZdS )�AsyncQuicConnectionNr& �returnc � s �d S r r )r r& r r r � make_stream� s �z AsyncQuicConnection.make_streamr )r r r r �floatr rT r r r r rR � s rR c @ s( e Zd Zd
dd�Zd dd�Zdd � ZdS ) �BaseQuicManagerNc C s^ i | _ || _|d u r*d }t|t�r|}d}tjjjddg||d�}|d ur*|�|� || _ d S )NT�doqzdoq-i03)�alpn_protocols� verify_mode� server_name)
� _connections�_connection_factory�
isinstance�str�aioquic�quic�
configuration�QuicConfiguration�load_verify_locations�_conf)r �confrY �connection_factoryrZ � verify_pathr r r r � s
�
zBaseQuicManager.__init__�U r c C sl | j �||f�}|d ur|dfS tjjj| jd�}|�|t�� � | � |||||| �}|| j ||f< |dfS )NF)ra T)
r[ r r_ r` r" �QuicConnectionrd �connectr% r\ )r rE rF rG rH r" �qconnr r r �_connect� s �zBaseQuicManager._connectc C s( z | j ||f= W d S ty Y d S w r )r[ �KeyError)r rE rF r r r �closed� s
�zBaseQuicManager.closedr �rh Nr )r r r r rl rn r r r r rV � s
rV c @ s e Zd Zddd�ZdS )�AsyncQuicManagerrh Nr c C s t �r )rC )r rE rF rG rH r r r rj � s zAsyncQuicManager.connectro )r r r rj r r r r rp � s rp )r@ r+ r% �typingr r �aioquic.quic.configurationr_ �aioquic.quic.connection�dns.inetr: �QUIC_MAX_DATAGRAM� Exceptionr r
r r3 rR rV rp r r r r �<module> s (/%