관리-도구
편집 파일: _trio.cpython-38.pyc
U /?�f� � @ 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__� �</opt/hc_python/lib/python3.8/site-packages/dns/quic/_trio.pyr s zTrioQuicStream.__init__c � sR | j �|�rd S || _| j4 I d H � | j�� I d H W 5 Q I d H R X d| _q d S )Nr )�_bufferZhaveZ _expectingr �wait)r �amountr r r �wait_for s zTrioQuicStream.wait_forNc � s� |d krt d �}n t�|�}|�P | �d�I d H t�d| j�d��\}| �|�I d H | j�|�W 5 Q R � S Q R X 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 � 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_input7 s zTrioQuicStream._add_inputc � s | � � d S r )�_close�r r r r �close< s zTrioQuicStream.closec � s | S r r r/ r r r � __aenter__A s zTrioQuicStream.__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__D s zTrioQuicStream.__aexit__)N)F) �__name__� __module__� __qualname__r r r% r+ r- r0 r1 r6 � __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�_af� SOCK_DGRAM�_socket�Event�_handshake_complete� _run_done� _worker_scope� _send_pending)r r �address�port�source� source_port�managerr r r r L s zTrioQuicConnection.__init__c � s z�| jr(| j�tj�| j| j ��I d H | j� | j �I d H | j s�| � d�\}}| j rZd}tjt�� | d��0| _| j�t�I d H }| j�|| j t�� � W 5 Q R X d | _| �|� | �� I d H d| _ | j�t�� �}|D ]\}}| j�|�I d H q�q<W 5 d| _ | j�� X d S )NTFg )�deadline)�_donerA �set�_sourcer? �bindr ZinetZlow_level_address_tupler= �connect�_peerZ_get_timer_valuesrD r Z CancelScope� current_timerC �recvr r& Zreceive_datagram�timeZ _handle_timer�_handle_eventsZdatagrams_to_sendr+ )r Z expiration�intervalr( Z datagrams�_r r r �_workerT s2 � � zTrioQuicConnection._workerc � 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� � � )r& Z next_event� isinstance�aioquicZquic�eventsZStreamDataReceived�_streamsr r r- r* Z end_streamZHandshakeCompletedrA rL ZConnectionTerminatedrK r? r0 Z StreamResetr �sleep)r �count�event�streamr r r rT x s* z!TrioQuicConnection._handle_eventsFc � s. | j �|||� d| _| jd k r*| j�� d S )NT)r& Zsend_stream_datarD rC �cancel)r rb r* r) r r r r' � s zTrioQuicConnection.writec � sF | j r d S t�� 4 I d H �}|�| j� W 5 Q I d H R X | j�� d S r )�_closedr Z open_nursery� start_soonrW rB rL )r �nurseryr r r �run� s zTrioQuicConnection.runc � s~ |d krt d �}n t�|�}|�L | j�� I d H | jr<t�| j�d�}t | |�}|| j |< |W 5 Q R � S Q R X t j j �d S r2 )r r r rA r rK r r& Z get_next_available_stream_idr r^ r r r! )r r" r# r rb r r r � make_stream� s z TrioQuicConnection.make_streamc � s^ | j sZ| j�| jd | jd � d| _ | j�� d| _| jd k rJ| j�� | j � � I d H d S )Nr rY T) rd �_manager�closedrP r& r0 rD rC rc rB r r/ r r r r0 � s zTrioQuicConnection.close)N)F)N) r7 r8 r9 r rW rT r' rg rh r0 r: 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 rf Zconf� verify_modeZ server_namer r r r � s zTrioQuicManager.__init__�U r Tc C s, | � |||||�\}}|r(| j�|j� |S r )�_connectrl re rg )r rE rF rG rH Zwant_session_ticketr �startr r r rO � s �zTrioQuicManager.connectc � s | S r r r/ r r r r1 � s zTrioQuicManager.__aenter__c � s* t | j�� �}|D ]}|�� I d H qdS r2 )�listZ _connections�valuesr0 )r r3 r4 r5 � connectionsr r r r r6 � s zTrioQuicManager.__aexit__)rn Nr T) r7 r8 r9 �ssl� CERT_REQUIREDr rO r1 r6 r: r r r r rk � s � � rk )r< rt r rS Zaioquic.quic.configurationr\ Zaioquic.quic.connectionZaioquic.quic.eventsr Z dns.exceptionr Zdns.inetZdns._asyncbackendr Zdns.quic._commonr r r r r r r; rk r r r r �<module> s 2m