관리-도구
편집 파일: _trio.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 m Z d dlmZmZmZmZmZ G dd� de�ZG dd� de�ZG dd � d e�ZdS ) � N)� NullContext)�QUIC_MAX_DATAGRAM�AsyncQuicConnection�AsyncQuicManager�BaseQuicStream� UnexpectedEOFc sX e Zd Z� fdd�Zdd� Zddd�Zdd d �Zd d � Zd d� Zdd� Z dd� Z � Z S )�TrioQuicStreamc s t � �||� t�� | _d S �N)�super�__init__�trio� Condition�_wake_up)�self� connection� stream_id�� __class__� �2/usr/lib/python3.9/site-packages/dns/quic/_trio.pyr s zTrioQuicStream.__init__c � sl | j �|�rd S || _| j4 I d H �& | j�� I d H W d �I d H q`1 I d H sV0 Y d| _q d S )Nr )�_bufferZhaveZ _expectingr �wait)r �amountr r r �wait_for s :zTrioQuicStream.wait_forNc � s� |d u rt d �}n t�|�}|�R | �d�I d H t�d| j�d��\}| �|�I d H | j�|�W d � S 1 sx0 Y tj j �d S )N� z!H) r r � move_on_afterr �struct�unpackr �get�dns� exception�Timeout)r �timeout�context�sizer r r �receive' s *zTrioQuicStream.receiveFc � s& | � |�}| j�| j||�I d H d S r )Z _encapsulate� _connection�writeZ _stream_id)r �datagram�is_end�datar r r �send3 s zTrioQuicStream.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_input7 s zTrioQuicStream._add_inputc � s | � � d S r )Z_close�r r r r �close< s zTrioQuicStream.closec � s | S r r r. r r r � __aenter__A s zTrioQuicStream.__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__D s 4zTrioQuicStream.__aexit__)N)F) �__name__� __module__� __qualname__r r r% r+ r- r/ r0 r5 � __classcell__r r r r r s r c sR e Zd Zd� fdd� Zdd� Zdd� Zdd d �Zd d � Zdd d�Zdd� Z � Z S )�TrioQuicConnectionNc sP t � �||||||� tj�| jtjd�| _t�� | _t�� | _ d | _ d| _ d S )Nr F) r r r �socket�_afZ SOCK_DGRAM�_socket�Event�_handshake_complete� _run_done� _worker_scope� _send_pending)r r �address�port�source� source_portZmanagerr r r r L s zTrioQuicConnection.__init__c � s< �z"| j r*| j�tj�| j | j��I d H | j�| j�I d H | j �s| � d�\}}| j r^d}t j t �� | d��:| _| j�t�I d H }| j�|| jt�� � W d � n1 s�0 Y d | _| �|� | �� I d H d| _ | j�t�� �}|D ]\}}| j�|�I d H q�q>W d| _ | j�� nd| _ | j�� 0 d S )NFg )ZdeadlineT)�_sourcer= Zbindr ZinetZlow_level_address_tupler< �connect�_peer�_doneZ_get_timer_valuesrB r Z CancelScopeZ current_timerA Zrecvr r& Zreceive_datagram�timeZ _handle_timer�_handle_eventsZdatagrams_to_sendr+ r? �set)r Z expiration�intervalr( Z datagrams�_r r r �_workerT s6 � �4 �zTrioQuicConnection._workerc � 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� � � )r& Z next_event� isinstance�aioquicZquicZeventsZStreamDataReceived�_streamsr r r- r* Z end_streamZHandshakeCompletedr? rM ZConnectionTerminatedrJ r= r/ Z StreamResetr �sleep)r �countZevent�streamr r r rL x s* z!TrioQuicConnection._handle_eventsFc � s. | j �|||� d| _| jd ur*| j�� d S )NT)r& Zsend_stream_datarB rA �cancel)r rY r* r) r r r r' � s zTrioQuicConnection.writec � s` | j r d S t�� 4 I d H �"}|�| j� W d �I d H qR1 I d H sH0 Y | j�� d S r )�_closedr Z open_nursery� start_soonrP r@ rM )r �nurseryr r r �run� s 6zTrioQuicConnection.runc � s� |d u rt d �}n t�|�}|�N | j�� I d H | jr<t�| j�d�}t | |�}|| j |< |W d � S 1 st0 Y t j j �d S r1 )r r r r? r rJ r r& Z get_next_available_stream_idr rV r r r! )r r" r# r rY r r r � make_stream� s "z TrioQuicConnection.make_streamc � s^ | j sZ| j�| jd | jd � d| _ | j�� d| _| jd urJ| j�� | j � � I d H d S )Nr rR T) r[ Z_manager�closedrI r& r/ rB rA rZ r@ r r. r r r r/ � s zTrioQuicConnection.close)N)F)N) r6 r7 r8 r rP rL r' r^ r_ r/ r9 r r r r r: K s $ r: c s@ e Zd Zdejdf� fdd� Zd dd�Zd d � Zd d � Z� Z S )�TrioQuicManagerNc s t � �||t|� || _d S r )r r r: �_nursery)r r] ZconfZ verify_modeZ server_namer r r r � s zTrioQuicManager.__init__�U r Tc C s, | � |||||�\}}|r(| j�|j� |S r )Z_connectrb r\ r^ )r rC rD rE rF Zwant_session_ticketr �startr r r rH � s �zTrioQuicManager.connectc � s | S r r r. r r r r0 � s zTrioQuicManager.__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 zTrioQuicManager.__aexit__)rc Nr T) r6 r7 r8 �sslZ CERT_REQUIREDr rH r0 r5 r9 r r r r ra � s � � ra )r; rg r rK Zaioquic.quic.configurationrU Zaioquic.quic.connectionZaioquic.quic.eventsr Z dns.exceptionr Zdns.inetZdns._asyncbackendr Zdns.quic._commonr r r r r r r: ra r r r r �<module> s 2m