관리-도구
편집 파일: _common.cpython-38.pyc
U /?�fr � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dlZ d dl Z d dl Z dZ dZed ZG dd� de�ZG dd � d �ZG d d � d �ZG d d � d �ZG dd� de�ZG dd� d�ZG dd� de�ZdS )� N)�Any�Optionali � � c @ s e Zd ZdS )� UnexpectedEOFN)�__name__� __module__� __qualname__� r r �>/opt/hc_python/lib/python3.8/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 r d S | j|7 _|r"d| _ d S )NT)r r �r �dataZis_endr r r �put s z Buffer.putc C s t | j�|krdS | jr t�dS )NTF)�lenr r r )r �amountr r r �have& s z Buffer.havec C s | j S �N)r r r r r �seen_end- s zBuffer.seen_endc C s0 | � |�st�| jd |� }| j|d � | _|S r )r �AssertionErrorr )r r r r r r �get0 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 8 s zBaseQuicStream.__init__c C s | j S r )r r r r r �id> s zBaseQuicStream.idc C s |d k rt � � | }nd }|S r )�time)r �timeout� expirationr r r �_expiration_from_timeoutA s z'BaseQuicStream._expiration_from_timeoutc C s$ |d k r t |t�� d�}nd }|S )Ng )�maxr% )r r' r& r r r �_timeout_from_expirationH s z'BaseQuicStream._timeout_from_expirationc C s t |�}t�d|�| S )Nz!H)r �struct�pack)r Zdatagram�lr r r � _encapsulateR s zBaseQuicStream._encapsulatec C sD | j �||� z| jdko&| j �| j�W S tk r> Y dS X d S )Nr T)r r r! r r r r r r �_common_add_inputV 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 �_close] s zBaseQuicStream._closeN) r r r r r$ r( r* r. r/ r1 r r r r r 7 s r c @ s0 e Zd Zd dd�Zdd� Zd dd �Zd d � ZdS )�BaseQuicConnectionNr c C s� d| _ || _|| _|| _d| _|| _i | _tj� |�| _ tj� ||f�| _ |d kr�|dkr�| j t jkrld}n| j t jkr~d}nt�|r�||f| _nd | _d S )NFr z0.0.0.0z::)Z_doner Z_addressZ_port�_closed�_manager�_streams�dns�inetZaf_for_addressZ_af�low_level_address_tupleZ_peer�socket�AF_INET�AF_INET6�NotImplementedError�_source)r r" �address�port�source� source_port�managerr r r r c s$ zBaseQuicConnection.__init__c C s | j |= d S r )r5 )r r# r r r r0 { s z BaseQuicConnection.close_streamTc C sL t � � }| j�� }|d kr"|d }t|| d�}| jrD|rDt|d�}||fS )Ni r g�������?)r% r Z get_timerr) r3 �min)r Zclosed_is_special�nowr' �intervalr r r �_get_timer_values~ s z$BaseQuicConnection._get_timer_valuesc C s t � � }||kr | j�|� d S r )r% r Z handle_timer)r r' rD r r r � _handle_timer� s z BaseQuicConnection._handle_timer)Nr N)T)r r r r r0 rF rG r r r r r2 b s � r2 c @ s" e Zd Zdee ed�dd�ZdS )�AsyncQuicConnectionN)r& �returnc � s d S r r )r r& r r r � make_stream� s z AsyncQuicConnection.make_stream)N)r r r r �floatr rJ r r r r rH � s rH c @ s0 e Zd Zd dd�Zddd�Zd d � Zd d � ZdS )�BaseQuicManagerNc C sd i | _ || _i | _|d krZd }t|t�r0|}d}tjjjddg||d�}|d k rZ|� |� || _ d S )NTZdoqzdoq-i03)�alpn_protocols� verify_mode� server_name) � _connections�_connection_factory�_session_tickets� isinstance�str�aioquic�quic� configurationZQuicConfiguration�load_verify_locations�_conf)r �confrN �connection_factoryrO Z verify_pathr r r r � s � zBaseQuicManager.__init__�U r Tc C s� | j �||f�}|d k r |dfS | j}|rvz$| j�||f�}t�|�}||_W n tk rb Y nX t� | j ||�} nd } t j j j|| d�} tj�||f�} | �| t�� � | �| ||||| �}|| j ||f< |dfS )NF)rW �session_ticket_handlerT)rP r rY rR �pop�copy�session_ticket�KeyError� functools�partial�save_session_ticketrU rV r" ZQuicConnectionr6 r7 r8 �connectr% rQ ) r r> r? r@ rA Zwant_session_ticketr" rZ r` r] ZqconnZ lladdressr r r �_connect� sB �� �zBaseQuicManager._connectc C s* z| j ||f= W n tk r$ Y nX d S r )rP ra )r r> r? r r r �closed� s zBaseQuicManager.closedc C sL t | j�}|tkr:t| j�� �dt� }|D ] }| j|= q,|| j||f< d S r )r rR �MAX_SESSION_TICKETS�list�keys�SESSIONS_TO_DELETE)r r> r? �ticketr- Zkeys_to_delete�keyr r r rd � s z#BaseQuicManager.save_session_ticket)N)r\ Nr T)r r r r rf rg rd r r r r rL � s � #rL c @ s e Zd Zddd�ZdS )�AsyncQuicManagerr\ Nr c C s t �d S r )r< )r r> r? r@ rA r r r re � s zAsyncQuicManager.connect)r\ Nr )r r r re r r r r rn � s rn )r_ rb r9 r+ r% �typingr r Zaioquic.quic.configurationrU Zaioquic.quic.connectionZdns.inetr6 ZQUIC_MAX_DATAGRAMrh rk � Exceptionr r r r2 rH rL rn r r r r �<module> s$ +/H