관리-도구
편집 파일: _asyncio.cpython-39.pyc
a �M�e� � @ 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__� �5/usr/lib/python3.9/site-packages/dns/quic/_asyncio.pyr s zAsyncioQuicStream.__init__c � sN | j 4 I d H �& | j �� I d H W d �I d H qJ1 I d H s@0 Y d S r )r �wait�r r r r �_wait_for_wake_up s z#AsyncioQuicStream._wait_for_wake_upc � sb | � |�}| j�|�rd S || _zt�| �� |�I d H W n tyT tj j �Y n0 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 � sT | � ||�rP| j4 I d H � | j�� W d �I d H qP1 I d H sF0 Y 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 )Z_closer r r r �close> s zAsyncioQuicStream.closec � s | S r r r r r r � __aenter__C s z AsyncioQuicStream.__aenter__c � sV | � � I d H | j4 I d H � | j�� W d �I d H qR1 I d H sH0 Y dS �NF)r/ r r- )r �exc_type�exc_val�exc_tbr r r � __aexit__F s 4zAsyncioQuicStream.__aexit__)N)F) �__name__� __module__� __qualname__r r r r& r, r. r/ r0 r5 � __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_portZmanagerr r r r N s z AsyncioQuicConnection.__init__c � s �z��z6t j�| j�}t j�d�}|�|tjd| j | j d | j d f�I d H | _ | j � � | j 4 I d H �� | j�s | j �td �I d H \}}|d | j d ksj|d | j d kr�qj| j�||t�� � | j4 I d H � | j�� W d �I d H qj1 I d H �s 0 Y qjW d �I d H �q:1 I d H �s00 Y W n t�yP Y n0 W d| _| j4 I d H �" | j�� W d �I d H �q�1 I d H �s�0 Y | j� � nZd| _| j4 I d H �" | j�� W d �I d H �q�1 I d H �s�0 Y | j� � 0 d S )Nr r � T)r ZinetZaf_for_addressZ_addressZ asyncbackendZ get_backendZ make_socket�socketZ SOCK_DGRAM�_source�_peerr; r> �set�_doneZrecvfromr r' Zreceive_datagram�timer? � notify_all� Exceptionr= )r �afZbackendr) rB r r r � _receiverW s6 � �$j8 �8z AsyncioQuicConnection._receiverc � sN | j 4 I d H �& | j �� I d H W d �I d H qJ1 I d H s@0 Y 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@J �| j�|| jd �I d H q*| � � \}}zt � | � � |�I d H W n t y� Y n0 | �|� | �� I d H qd S r )r> r rK r' Zdatagrams_to_sendrL rI r; ZsendtoZ_get_timer_valuesr r rQ rN Z _handle_timer�_handle_events)r Z datagramsr) rB r �intervalr r r �_sendery s z AsyncioQuicConnection._senderc � s� d}| j �� }|d u rd 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� rF � )r' Z next_event� isinstance�aioquicZquicZeventsZStreamDataReceived�_streamsr$ r r. r+ Z end_streamZHandshakeCompletedr= rJ ZConnectionTerminatedrK r@ �cancelZ StreamResetr �sleep)r �countZevent�streamr r r rR � s* z$AsyncioQuicConnection._handle_eventsFc � sX | j �|||� | j4 I d H � | j�� W d �I d H qT1 I d H sJ0 Y d S r )r' Zsend_stream_datar? rM )r r] r+ r* r r r r( � s zAsyncioQuicConnection.writec C s. | j r d S t�| �� �| _t�| �� �| _d S r )�_closedr ZTaskrP r@ rT rA r r r r �run� s zAsyncioQuicConnection.runc � sf z t �| j�� |�I d H W n ty6 tjj�Y n0 | jrBt �| j � d�}t | |�}|| j |< |S r1 )r r r= r r r r r rK r r' Z get_next_available_stream_idr rY )r r r r] 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 d �I d H q~1 I d H st0 Y z| j I d H W n t j y� Y n0 z| j I d H W n t j y� Y n0 | j�� I d H d S )Nr rF T)r^ Z_manager�closedrI r' r/ r> rJ r? rM r@ r ZCancelledErrorrA r; r r r r r/ � s 4zAsyncioQuicConnection.close)N)F)N) r6 r7 r8 r rP rQ rT rR r( r_ r` r/ r9 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 ZconfZ verify_modeZ server_namer r r r � s zAsyncioQuicManager.__init__�U r Tc C s& | � |||||�\}}|r"|�� |S r )Z_connectr_ )r rB rC rD rE Zwant_session_ticketr �startr r r �connect� s �zAsyncioQuicManager.connectc � s | S r r r r r r r0 � s z AsyncioQuicManager.__aenter__c � s* t | j�� �}|D ]}|�� I d H qdS r1 )�listZ _connections�valuesr/ )r r2 r3 r4 Z connectionsr r r r r5 � s z AsyncioQuicManager.__aexit__)rc Nr T) r6 r7 r8 �sslZ CERT_REQUIREDr re r0 r5 r9 r r r r rb � s � rb )r rG rh r" rL Zaioquic.quic.configurationrX Zaioquic.quic.connectionZaioquic.quic.eventsZdns.asyncbackendr Z dns.exceptionZdns.inetZdns.quic._commonr r r r r r r: rb r r r r �<module> s 4