관리-도구
편집 파일: _asyncio.cpython-38.pyc
U /?�f� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl Z d dl Z d dl mZmZmZmZmZ G dd� de�ZG dd� de�ZG dd� de�ZdS ) � N)�QUIC_MAX_DATAGRAM�AsyncQuicConnection�AsyncQuicManager�BaseQuicStream� UnexpectedEOFc s` e Zd Z� fdd�Zdd� Zdd� Zddd �Zdd d �Zd d� Zdd� Z dd� Z dd� Z � Z S )�AsyncioQuicStreamc s t � �||� t�� | _d S �N)�super�__init__�asyncio� Condition�_wake_up)�self� connection� stream_id�� __class__� �?/opt/hc_python/lib/python3.8/site-packages/dns/quic/_asyncio.pyr s zAsyncioQuicStream.__init__c � s4 | j 4 I d H � | j �� I d H W 5 Q I d H R X d S r )r �wait�r r r r �_wait_for_wake_up s z#AsyncioQuicStream._wait_for_wake_upc � sd | � |�}| j�|�rd S || _zt�| �� |�I d H W n tk rV tj j �Y nX d| _q d S )Nr ) Z_timeout_from_expiration�_bufferZhaveZ _expectingr �wait_forr � TimeoutError�dns� exception�Timeout)r �amount� expiration�timeoutr r r r " s zAsyncioQuicStream.wait_forNc � sP | � |�}| �d|�I d H t�d| j�d��\}| �||�I d H | j�|�S )N� z!H)Z_expiration_from_timeoutr �struct�unpackr �get)r r r �sizer r r �receive. s zAsyncioQuicStream.receiveFc � s& | � |�}| j�| j||�I d H d S r )Z _encapsulate� _connection�writeZ _stream_id)r �datagram�is_end�datar r r �send5 s zAsyncioQuicStream.sendc � s: | � ||�r6| j4 I d H � | j�� W 5 Q I d H R X d S r )Z_common_add_inputr �notify)r r+ r* r r r � _add_input9 s z AsyncioQuicStream._add_inputc � s | � � d S r )�_closer r r r �close> s zAsyncioQuicStream.closec � s | S r r r r r r � __aenter__C s z AsyncioQuicStream.__aenter__c � s< | � � I d H | j4 I d H � | j�� W 5 Q I d H R X dS �NF)r0 r r- )r �exc_type�exc_val�exc_tbr r r � __aexit__F s zAsyncioQuicStream.__aexit__)N)F) �__name__� __module__� __qualname__r r r r& r, r. r0 r1 r6 � __classcell__r r r r r s r c sb e Zd Zd� fdd� Zdd� Zdd� Zdd � Zd d � Zdd d�Zdd� Z ddd�Z dd� Z � Z S )�AsyncioQuicConnectionNc sJ t � �||||||� d | _t�� | _t�� | _t�� | _d | _ d | _ d S r ) r r �_socketr �Event�_handshake_complete�_socket_createdr � _wake_timer�_receiver_task� _sender_task)r r �address�port�source� source_port�managerr r r r N s z AsyncioQuicConnection.__init__c � sZ �zz�tj�| j�}tj � d�}|� |t j d| j| jd | jd f�I d H | _| j�� | j4 I d H �� | j s�| j�td �I d H \}}|d | jd ksh|d | jd kr�qh| j�||t�� � | j4 I d H � | j�� W 5 Q I d H R X qhW 5 Q I d H R X W n tk �r Y nX W 5 d| _ | j4 I d H � | j�� W 5 Q I d H R X | j�� X d S )NTr r � )�_doner@ � notify_allr> �setr ZinetZaf_for_addressZ_addressZ asyncbackend� get_backendZ make_socket�socket� SOCK_DGRAM�_source�_peerr< r? �recvfromr r' Zreceive_datagram�time� Exception)r �af�backendr) rC r r r � _receiverW s8 � �$0 z AsyncioQuicConnection._receiverc � s4 | j 4 I d H � | j �� I d H W 5 Q I d H R X d S r )r@ r r r r r �_wait_for_wake_timeru s z*AsyncioQuicConnection._wait_for_wake_timerc � s� | j �� I d H | js�| j�t�� �}|D ].\}}|| jks@t�| j� || jd �I d H q*| � � \}}zt � | � � |�I d H W n tk r� Y nX | �|� | �� I d H qd S r )r? r rI r' Zdatagrams_to_sendrR rP �AssertionErrorr< �sendtoZ_get_timer_valuesr r rW rS Z _handle_timer�_handle_events)r Z datagramsr) rC r �intervalr r r �_sendery s z AsyncioQuicConnection._senderc � s� d}| j �� }|d krd 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� rH � )r' Z next_event� isinstance�aioquicZquic�eventsZStreamDataReceived�_streamsr$ r r. r+ Z end_streamZHandshakeCompletedr> rK ZConnectionTerminatedrI rA �cancelZ StreamResetr �sleep)r �count�event�streamr r r rZ � s* z$AsyncioQuicConnection._handle_eventsFc � s> | j �|||� | j4 I d H � | j�� W 5 Q I d H R X d S r )r' Zsend_stream_datar@ rJ )r rg r+ r* r r r r( � s zAsyncioQuicConnection.writec C s. | j r d S t�| �� �| _t�| �� �| _d S r )�_closedr �TaskrV rA r\ rB r r r r �run� s zAsyncioQuicConnection.runc � sh z t �| j�� |�I d H W n tk r8 tjj�Y nX | jrDt �| j � d�}t | |�}|| j |< |S r2 )r r r> r r r r r rI r r' Z get_next_available_stream_idr rb )r r r rg r r r � make_stream� s z!AsyncioQuicConnection.make_streamc � s� | j s�| j�| jd | jd � d| _ | j�� | j�� | j4 I d H � | j� � W 5 Q I d H R X z| j I d H W n t j k r� Y nX z| j I d H W n t j k r� Y nX | j�� I d H d S )Nr rH T)rh �_manager�closedrP r' r0 r? rK r@ rJ rA r ZCancelledErrorrB r< r r r r r0 � s zAsyncioQuicConnection.close)N)F)N) r7 r8 r9 r rV rW r\ rZ r( rj rk r0 r: r r r r r; M s r; c s@ e Zd Zdejdf� fdd� Zd dd�Zd d � Zd d � Z� Z S )�AsyncioQuicManagerNc s t � �||t|� d S r )r r r; )r Zconf� verify_modeZ server_namer r r r � s zAsyncioQuicManager.__init__�U r Tc C s& | � |||||�\}}|r"|�� |S r )�_connectrj )r rC rD rE rF Zwant_session_ticketr �startr r r �connect� s �zAsyncioQuicManager.connectc � s | S r r r r r r r1 � s z AsyncioQuicManager.__aenter__c � s* t | j�� �}|D ]}|�� I d H qdS r2 )�listZ _connections�valuesr0 )r r3 r4 r5 � connectionsr r r r r6 � s z AsyncioQuicManager.__aexit__)rp Nr T) r7 r8 r9 �ssl� CERT_REQUIREDr rs r1 r6 r: r r r r rn � s � rn )r rM rw r" rR Zaioquic.quic.configurationr` Zaioquic.quic.connectionZaioquic.quic.eventsZdns.asyncbackendr Z dns.exceptionZdns.inetZdns.quic._commonr r r r r r r; rn r r r r �<module> s 4