관리-도구
편집 파일: _sync.cpython-38.pyc
U /?�f� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Zd dl Z d dl Z d dl mZmZmZmZmZ ee d�r�e jZne jZG dd� de�ZG dd� de�ZG dd � d e�ZdS ) � N)�QUIC_MAX_DATAGRAM�BaseQuicConnection�BaseQuicManager�BaseQuicStream� UnexpectedEOF� PollSelectorc sX e Zd Z� fdd�Zdd� Zddd�Zdd d �Zd d � Zd d� Zdd� Z dd� Z � Z S )�SyncQuicStreamc s& t � �||� t�� | _t�� | _d S �N)�super�__init__� threading� Condition�_wake_up�Lock�_lock)�self� connection� stream_id�� __class__� �</opt/hc_python/lib/python3.8/site-packages/dns/quic/_sync.pyr s zSyncQuicStream.__init__c C sn | � |�}| j�& | j�|�r,W 5 Q R � d S || _W 5 Q R X | j� | j�|�sXtjj �W 5 Q R X d| _q d S )Nr ) Z_timeout_from_expirationr �_bufferZhaveZ _expectingr �wait�dns� exception�Timeout)r �amount� expiration�timeoutr r r �wait_for% s zSyncQuicStream.wait_forNc C st | � |�}| �d|� | j� t�d| j�d��\}W 5 Q R X | �||� | j� | j�|�W 5 Q R � S Q R X d S )N� z!H)Z_expiration_from_timeoutr r �struct�unpackr �get)r r r �sizer r r �receive1 s zSyncQuicStream.receiveFc C s | � |�}| j�| j||� d S r )Z _encapsulate� _connection�writeZ _stream_id)r �datagram�is_end�datar r r �send: s zSyncQuicStream.sendc C s, | � ||�r(| j� | j�� W 5 Q R X d S r )Z_common_add_inputr �notify)r r+ r* r r r � _add_input> s zSyncQuicStream._add_inputc C s | j � | �� W 5 Q R X d S r )r �_close�r r r r �closeC s zSyncQuicStream.closec C s | S r r r0 r r r � __enter__G s zSyncQuicStream.__enter__c C s( | � � | j� | j�� W 5 Q R X dS �NF)r1 r r- )r �exc_type�exc_val�exc_tbr r r �__exit__J s zSyncQuicStream.__exit__)N)F) �__name__� __module__� __qualname__r r r& r, r. r1 r2 r7 � __classcell__r r r r r s r c sl e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d � Zdd d �Zdd� Z ddd�Z � fdd�Z dd� Z � Z S )�SyncQuicConnectionc s� t � �||||||� t�| jtjd�| _| jd k rtz | j�tj � | j| j�� W n t k rr | j� � � Y nX | j� | j� t�� \| _| _| j�d� | j�d� t�� | _d | _t�� | _d S )Nr F)r r �socketZ_af� SOCK_DGRAM�_socket�_source�bindr ZinetZlow_level_address_tuple� Exceptionr1 �connect�_peer� socketpair� _send_wakeup�_receive_wakeup� setblockingr �Event�_handshake_complete�_worker_threadr r )r r �address�port�source� source_port�managerr r r r R s" � zSyncQuicConnection.__init__c C sj d}|dk rf|d7 }z| j �t�}W n tk r: Y d S X | j� | j�|| jt�� � W 5 Q R X qd S )Nr � � ) r? �recvr �BlockingIOErrorr r' Zreceive_datagramrD �time)r �countr) r r r �_reade s zSyncQuicConnection._readc C s. z| j �d� W q tk r& Y d S X q d S )N� )rG rS rT r0 r r r � _drain_wakeupp s z SyncQuicConnection._drain_wakeupc C s z�t� }|�| jtj| j � |�| j tj| j � | js�| � d�\}}|� |�}|D ]\}}|�� qR| j � | �|� W 5 Q R X | �� | j � | j�t�� �}W 5 Q R X |D ].\}}z| j�|� W q� tk r� Y q�X q�q0W 5 | j � d| _W 5 Q R X | j�� X d S )NTF)r �_donerJ �set�_selector_class�registerr? � selectors� EVENT_READrW rG rY Z_get_timer_values�selectr+ Z _handle_timer�_handle_eventsr' Zdatagrams_to_sendrU r, rT ) r �selr �interval�items�key�_Z datagramsr) r r r �_workerw s, zSyncQuicConnection._workerc C s� | j � | j�� }W 5 Q R X |d kr(d S t|tjjj�rn| j � | j� |j �}W 5 Q R X |r�|� |j |j � q t|tjjj�r�| j�� q t|tjjj�r�| j � d| _W 5 Q R X q t|tjjj�r | j � | j� |j �}W 5 Q R X |r |� dd� q d S )NT� )r r' Z next_event� isinstance�aioquicZquic�eventsZStreamDataReceived�_streamsr$ r r. r+ Z end_streamZHandshakeCompletedrJ r[ ZConnectionTerminatedrZ Z StreamReset)r �event�streamr r r ra � s& z!SyncQuicConnection._handle_eventsFc C s2 | j � | j�|||� W 5 Q R X | j�d� d S )N� )r r' Zsend_stream_datarF r, )r rn r+ r* r r r r( � s zSyncQuicConnection.writec C s( | j r d S tj| jd�| _| j�� d S )N)�target)�_closedr �Threadrg rK �startr0 r r r �run� s zSyncQuicConnection.runNc C sT | j �|�stjj�| j�0 | jr&t�| j� d�}t | |�}|| j |< W 5 Q R X |S r3 ) rJ r r r r r rZ r r' Z get_next_available_stream_idr rl )r r r rn r r r � make_stream� s z SyncQuicConnection.make_streamc s" | j � t� �|� W 5 Q R X d S r )r r � close_stream)r r r r r rv � s z SyncQuicConnection.close_streamc C sj | j �P | jr W 5 Q R � d S | j�| jd | jd � d| _| j�� | j�d� W 5 Q R X | j � � d S )Nr rR Tro ) r rq �_manager�closedrD r' r1 rF r, rK �joinr0 r r r r1 � s zSyncQuicConnection.close)F)N)r8 r9 r: r rW rY rg ra r( rt ru rv r1 r; r r r r r< Q s r< c sX e Zd Zdejdf� fdd� Zddd�Z� fd d �Z� fd d �Zd d� Z dd� Z � Z S )�SyncQuicManagerNc s t � �||t|� t�� | _d S r )r r r< r r r )r Zconf� verify_modeZ server_namer r r r � s zSyncQuicManager.__init__�U r Tc C sD | j �4 | �|||||�\}}|r*|�� |W 5 Q R � S Q R X d S r )r �_connectrt )r rL rM rN rO Zwant_session_ticketr rs r r r rC � s �zSyncQuicManager.connectc s$ | j � t� �||� W 5 Q R X d S r )r r rx )r rL rM r r r rx � s zSyncQuicManager.closedc s& | j � t� �|||� W 5 Q R X d S r )r r �save_session_ticket)r rL rM �ticketr r r r~ � s z#SyncQuicManager.save_session_ticketc C s | S r r r0 r r r r2 � s zSyncQuicManager.__enter__c C s$ t | j�� �}|D ] }|�� qdS r3 )�listZ _connections�valuesr1 )r r4 r5 r6 � connectionsr r r r r7 � s zSyncQuicManager.__exit__)r| Nr T) r8 r9 r: �ssl� CERT_REQUIREDr rC rx r~ r2 r7 r; r r r r rz � s � rz )r^ r= r� r" r rU Zaioquic.quic.configurationrj Zaioquic.quic.connectionZaioquic.quic.eventsZ dns.exceptionr Zdns.inetZdns.quic._commonr r r r r �hasattrr r\ �SelectSelectorr r< rz r r r r �<module> s" 2|