관리-도구
편집 파일: proactor_events.cpython-38.pyc
U e5d<} � @ sT d Z dZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl m Z ddl m Z dd l mZ dd l mZ dd l mZ dd l mZ dd lmZ dd� ZG dd� dejej�ZG dd� deej�ZG dd� deej�ZG dd� de�Z G dd� de�Z G dd� deeej �Z G d d � d eeej �Z G d d � d e j!�Z"dS ) z�Event loop using a proactor and related classes. A proactor is a "notify-on-completion" multiplexer. Currently a proactor is only implemented on Windows with IOCP. )�BaseProactorEventLoop� N� )� base_events)� constants)�futures)� exceptions)� protocols)�sslproto)� transports)�trsock)�loggerc C s� t �|�| jd< z|�� | jd< W n0 tjk rR | j�� rNtj d|dd� Y nX d| jkr�z|� � | jd< W n tjk r� d | jd< Y nX d S )N�socketZsocknamezgetsockname() failed on %rT��exc_info�peername) r �TransportSocket�_extraZ getsocknamer �error�_loop� get_debugr �warningZ getpeername)� transport�sock� r �//usr/lib64/python3.8/asyncio/proactor_events.py�_set_socket_extra s � r c s� e Zd ZdZd � fdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z e j fdd�Z d dd�Zdd� Zdd� Zdd� Z� ZS ) �_ProactorBasePipeTransportz*Base class for pipe and socket transports.Nc s� t � �||� | �|� || _| �|� || _d | _d | _d | _d| _ d| _ d| _ d| _ | jd k rl| j� � | j�| jj| � |d k r�| j�tj|d � d S )Nr F)�super�__init__� _set_extra�_sock� set_protocol�_server�_buffer� _read_fut� _write_fut�_pending_write� _conn_lost�_closing� _eof_writtenZ_attachr � call_soon� _protocolZconnection_mader Z _set_result_unless_cancelled��self�loopr �protocol�waiter�extra�server�� __class__r r r 2 s( �z#_ProactorBasePipeTransport.__init__c C s� | j jg}| jd kr |�d� n| jr0|�d� | jd k rP|�d| j�� � �� | jd k rl|�d| j��� | jd k r�|�d| j��� | jr�|�dt | j�� �� | j r�|�d� d� d � |��S ) N�closed�closingzfd=zread=zwrite=zwrite_bufsize=z EOF writtenz<{}>� ) r4 �__name__r �appendr( �filenor$ r% r# �lenr) �format�join)r- �infor r r �__repr__H s z#_ProactorBasePipeTransport.__repr__c C s || j d< d S )N�pipe)r �r- r r r r r Z s z%_ProactorBasePipeTransport._set_extrac C s || _ d S �N�r+ )r- r/ r r r r! ] s z'_ProactorBasePipeTransport.set_protocolc C s | j S rB rC �r- r r r � get_protocol` s z'_ProactorBasePipeTransport.get_protocolc C s | j S rB )r( rD r r r � is_closingc s z%_ProactorBasePipeTransport.is_closingc C s\ | j r d S d| _ | jd7 _| js>| jd kr>| j�| jd � | jd k rX| j�� d | _d S )NTr ) r( r' r# r% r r* �_call_connection_lostr$ �cancelrD r r r �closef s z _ProactorBasePipeTransport.closec C s* | j d k r&|d| ��t| d� | �� d S )Nzunclosed transport )�source)r �ResourceWarningrI )r- Z_warnr r r �__del__q s z"_ProactorBasePipeTransport.__del__� Fatal error on pipe transportc C sV zDt|t�r*| j�� rBtjd| |dd� n| j�||| | jd�� W 5 | � |� X d S )Nz%r: %sTr )�message� exceptionr r/ ) � _force_close� isinstance�OSErrorr r r �debug�call_exception_handlerr+ )r- �excrN r r r � _fatal_errorv s � z'_ProactorBasePipeTransport._fatal_errorc C s� | j d k r6| j �� s6|d kr*| j �d � n | j �|� | jr@d S d| _| jd7 _| jrj| j�� d | _| jr�| j�� d | _d| _ d | _ | j � | j |� d S )NTr r )� _empty_waiter�done� set_resultZ set_exceptionr( r' r% rH r$ r& r# r r* rG )r- rU r r r rP � s" z'_ProactorBasePipeTransport._force_closec C s^ z| j� |� W 5 t | jd�r,| j�tj� | j�� d | _| j}|d k rX|�� d | _X d S )N�shutdown) �hasattrr rZ r Z SHUT_RDWRrI r" Z_detachr+ Zconnection_lost)r- rU r2 r r r rG � s z0_ProactorBasePipeTransport._call_connection_lostc C s" | j }| jd k r |t| j�7 }|S rB )r& r# r; )r- �sizer r r �get_write_buffer_size� s z0_ProactorBasePipeTransport.get_write_buffer_size)NNN)rM )r8 � __module__� __qualname__�__doc__r r? r r! rE rF rI �warnings�warnrL rV rP rG r] � __classcell__r r r3 r r . s � r c sT e Zd ZdZd� fdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z ddd�Z � Z S )�_ProactorReadPipeTransportzTransport for read pipes.Nc s: d | _ d| _t� �||||||� | j�| j� d| _d S )NTF)� _pending_data�_pausedr r r r* � _loop_readingr, r3 r r r � s z#_ProactorReadPipeTransport.__init__c C s | j o| j S rB )rf r( rD r r r � is_reading� s z%_ProactorReadPipeTransport.is_readingc C s0 | j s | jrd S d| _| j�� r,t�d| � d S )NTz%r pauses reading)r( rf r r r rS rD r r r � pause_reading� s z(_ProactorReadPipeTransport.pause_readingc C sn | j s | jsd S d| _| jd kr0| j�| jd � | j}d | _|d k rT| j�| j|� | j�� rjt � d| � d S )NFz%r resumes reading) r( rf r$ r r* rg re �_data_receivedr r rS �r- �datar r r �resume_reading� s z)_ProactorReadPipeTransport.resume_readingc C s� | j �� rt�d| � z| j�� }W nL ttfk r> � Y n4 tk rp } z| � |d� W Y �d S d }~X Y nX |s~| � � d S )Nz%r received EOFz1Fatal error: protocol.eof_received() call failed.) r r r rS r+ Z eof_received� SystemExit�KeyboardInterrupt� BaseExceptionrV rI )r- Z keep_openrU r r r � _eof_received� s �z(_ProactorReadPipeTransport._eof_receivedc C s� | j r | jd kst�|| _d S |s.| �� d S t| jtj�r�zt�| j|� W q� t t fk rh � Y q� t k r� } z| � |d� W Y �d S d }~X Y q�X n | j� |� d S )Nz3Fatal error: protocol.buffer_updated() call failed.)rf re �AssertionErrorrq rQ r+ r ZBufferedProtocolZ _feed_data_to_buffered_protorn ro rp rV Z data_received)r- rl rU r r r rj � s$ �z)_ProactorReadPipeTransport._data_receivedc C s� d }�zrz�|d k rP| j|ks0| jd kr,| js0t�d | _|�� rH|�� }n|�� | jrfd }W W ��d S |dkrzW W ��d S | js�| jj � | j d�| _W n� t k r� } z0| js�| � |d� n| j�� r�tjddd� W 5 d }~X Y n� tk �r } z| �|� W 5 d }~X Y nf tk �r> } z| � |d� W 5 d }~X Y n8 tjk �r^ | j�sZ� Y nX | j�sv| j�| j� W 5 |d k �r�| � |� X d S )N� i � z"Fatal read error on pipe transportz*Read error on pipe transport while closingTr )rj r$ r( rr rX �resultrH rf r � _proactor�recvr �ConnectionAbortedErrorrV r r rS �ConnectionResetErrorrP rR r �CancelledError�add_done_callbackrg )r- �futrl rU r r r rg sF � � z(_ProactorReadPipeTransport._loop_reading)NNN)N) r8 r^ r_ r` r rh ri rm rq rj rg rc r r r3 r rd � s � rd c s^ e Zd ZdZdZ� fdd�Zdd� Zddd �Zd d � Zd d � Z dd� Z dd� Z dd� Z � Z S )� _ProactorBaseWritePipeTransportzTransport for write pipes.Tc s t � j||� d | _d S rB )r r rW �r- �args�kwr3 r r r G s z(_ProactorBaseWritePipeTransport.__init__c C s� t |tttf�s$tdt|�j� ���| jr2td��| j d k rDtd��|sLd S | j rz| j t j krht �d� | j d7 _ d S | jd kr�| jd ks�t�| jt|�d� n.| js�t|�| _| �� n| j�|� | �� d S )Nz/data argument must be a bytes-like object, not zwrite_eof() already calledz(unable to write; sendfile is in progressz socket.send() raised exception.r )rl )rQ �bytes� bytearray� memoryview� TypeError�typer8 r) � RuntimeErrorrW r' r �!LOG_THRESHOLD_FOR_CONNLOST_WRITESr r r% r# rr � _loop_writing�_maybe_pause_protocol�extendrk r r r �writeK s. � z%_ProactorBaseWritePipeTransport.writeNc C sx �z|d k r"| j d kr"| jr"W d S || j ks0t�d | _ d| _|rH|�� |d kr\| j}d | _|s�| jrv| j�| jd � | j r�| j � t j � | �� n\| jj�| j |�| _ | j �� s�| jdks�t�t|�| _| j �| j� | �� n| j �| j� | jd k �r| j d k�r| j�d � W n\ tk �rD } z| �|� W 5 d }~X Y n0 tk �rr } z| �|d� W 5 d }~X Y nX d S )Nr z#Fatal write error on pipe transport) r% r( rr r& rt r# r r* rG r) r rZ r �SHUT_WR�_maybe_resume_protocolru �sendrX r; rz r� r� rW rY rx rP rR rV )r- �frl rU r r r r� q s< z-_ProactorBaseWritePipeTransport._loop_writingc C s dS �NTr rD r r r � can_write_eof� s z-_ProactorBaseWritePipeTransport.can_write_eofc C s | � � d S rB )rI rD r r r � write_eof� s z)_ProactorBaseWritePipeTransport.write_eofc C s | � d � d S rB �rP rD r r r �abort� s z%_ProactorBaseWritePipeTransport.abortc C s: | j d k rtd��| j�� | _ | jd kr4| j �d � | j S )NzEmpty waiter is already set)rW r� r Z create_futurer% rY rD r r r �_make_empty_waiter� s z2_ProactorBaseWritePipeTransport._make_empty_waiterc C s d | _ d S rB )rW rD r r r �_reset_empty_waiter� s z3_ProactorBaseWritePipeTransport._reset_empty_waiter)NN)r8 r^ r_ r` Z_start_tls_compatibler r� r� r� r� r� r� r� rc r r r3 r r| A s & )r| c s$ e Zd Z� fdd�Zdd� Z� ZS )�_ProactorWritePipeTransportc s4 t � j||� | jj�| jd�| _| j�| j� d S )N� ) r r r ru rv r r$ rz � _pipe_closedr} r3 r r r � s z$_ProactorWritePipeTransport.__init__c C sv |� � r d S |�� dks t�| jr4| jd ks0t�d S || jksLt|| jf��d | _| jd k rj| �t� � n| �� d S )Nrs ) Z cancelledrt rr r( r$ r% rP �BrokenPipeErrorrI )r- r{ r r r r� � s z(_ProactorWritePipeTransport._pipe_closed)r8 r^ r_ r r� rc r r r3 r r� � s r� c sX e Zd ZdZd� fdd� Zdd� Zdd� Zd d � Zdd d �Zdd d�Z ddd�Z � Z S )�_ProactorDatagramTransporti Nc s> || _ d | _t� j|||||d� t�� | _| j�| j � d S )N)r0 r1 ) �_addressrW r r � collections�dequer# r r* rg )r- r. r r/ �addressr0 r1 r3 r r r � s z#_ProactorDatagramTransport.__init__c C s t | |� d S rB �r rA r r r r � s z%_ProactorDatagramTransport._set_extrac C s t dd� | jD ��S )Nc s s | ]\}}t |�V qd S rB )r; )�.0rl �_r r r � <genexpr>� s zC_ProactorDatagramTransport.get_write_buffer_size.<locals>.<genexpr>)�sumr# rD r r r r] � s z0_ProactorDatagramTransport.get_write_buffer_sizec C s | � d � d S rB r� rD r r r r� � s z _ProactorDatagramTransport.abortc C s� t |tttf�s tdt|���|s&d S | jd k rN|d | jfkrNtd| j� ���| jr�| jr�| jt j krpt � d� | jd7 _d S | j �t|�|f� | jd kr�| �� | �� d S )Nz,data argument must be bytes-like object (%r)z!Invalid address: must be None or z!socket.sendto() raised exception.r )rQ r� r� r� r� r� r� � ValueErrorr' r r� r r r# r9 r% r� r� )r- rl �addrr r r �sendto� s& � � z!_ProactorDatagramTransport.sendtoc C s z�| j rW d S || jks t�d | _|r.|�� | jr@| j r\| jr\| jrV| j�| j d � W d S | j� � \}}| jd k r�| jj � | j |�| _n| jj j| j ||d�| _W nZ tk r� } z| j�|� W 5 d }~X Y nD tk r� } z| �|d� W 5 d }~X Y nX | j�| j� | �� d S )N)r� z'Fatal write error on datagram transport)r' r% rr rt r# r� r( r r* rG �popleftru r� r r� rR r+ �error_received� ExceptionrV rz r� r� )r- r{ rl r� rU r r r r� � s4 � � z(_ProactorDatagramTransport._loop_writingc C sV d }�z4z�| jr W W ��$d S | j|ks:| jd kr6| js:t�d | _|d k r�|�� }| jrdd }W W ��d S | jd k r||| j }}n|\}}| jr�W W ��d S | jd k r�| jj � | j | j �| _n| jj � | j | j �| _W nN tk r� } z| j �|� W 5 d }~X Y n< tjk �r | j�s� Y n X | jd k �r8| j�| j� W 5 |�rP| j �||� X d S rB )r+ Zdatagram_receivedr' r$ r( rr rt r� r ru rv r �max_sizeZrecvfromrR r� r ry rz rg )r- r{ rl r� �resrU r r r rg sD � � � z(_ProactorDatagramTransport._loop_reading)NNN)N)N)N) r8 r^ r_ r� r r r] r� r� r� rg rc r r r3 r r� � s � !r� c @ s e Zd ZdZdd� Zdd� ZdS )� _ProactorDuplexPipeTransportzTransport for duplex pipes.c C s dS )NFr rD r r r r� J s z*_ProactorDuplexPipeTransport.can_write_eofc C s t �d S rB )�NotImplementedErrorrD r r r r� M s z&_ProactorDuplexPipeTransport.write_eofN)r8 r^ r_ r` r� r� r r r r r� E s r� c sB e Zd ZdZejjZd � fdd� Zdd� Z dd� Z d d � Z � Z S ) �_ProactorSocketTransportz Transport for connected sockets.Nc s$ t � �||||||� t�|� d S rB )r r r Z _set_nodelayr, r3 r r r X s z!_ProactorSocketTransport.__init__c C s t | |� d S rB r� rA r r r r ] s z#_ProactorSocketTransport._set_extrac C s dS r� r rD r r r r� ` s z&_ProactorSocketTransport.can_write_eofc C s2 | j s | jrd S d| _| jd kr.| j�tj� d S r� )r( r) r% r rZ r r� rD r r r r� c s z"_ProactorSocketTransport.write_eof)NNN) r8 r^ r_ r` r Z _SendfileModeZ TRY_NATIVEZ_sendfile_compatibler r r� r� rc r r r3 r r� Q s �r� c s� e Zd Z� fdd�Zd3dd�Zd4dddddd�dd �Zd5d d �Zd6d d �Zd7dd�Zd8dd�Z � fdd�Z dd� Z dd� Z dd� Z dd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd9d&d'�Zd(d)� Zd:d+d,�Zd-d.� Zd/d0� Zd1d2� Z� ZS );r c sh t � �� t�d|jj� || _|| _d | _i | _ |� | � | � � t � � t �� krdt�| j�� � d S )NzUsing proactor: %s)r r r rS r4 r8 ru � _selector�_self_reading_future�_accept_futuresZset_loop�_make_self_pipe� threading�current_thread� main_thread�signal� set_wakeup_fd�_csockr: )r- Zproactorr3 r r r m s z BaseProactorEventLoop.__init__Nc C s t | |||||�S rB )r� )r- r r/ r0 r1 r2 r r r �_make_socket_transportz s �z,BaseProactorEventLoop._make_socket_transportF)� server_side�server_hostnamer1 r2 �ssl_handshake_timeoutc C s0 t j| |||||| d�} t| || ||d� | jS )N)r� �r1 r2 )r Z SSLProtocolr� Z_app_transport) r- Zrawsockr/ � sslcontextr0 r� r� r1 r2 r� Z ssl_protocolr r r �_make_ssl_transport s � �z)BaseProactorEventLoop._make_ssl_transportc C s t | |||||�S rB )r� )r- r r/ r� r0 r1 r r r �_make_datagram_transport� s �z.BaseProactorEventLoop._make_datagram_transportc C s t | ||||�S rB )r� �r- r r/ r0 r1 r r r �_make_duplex_pipe_transport� s �z1BaseProactorEventLoop._make_duplex_pipe_transportc C s t | ||||�S rB )rd r� r r r �_make_read_pipe_transport� s z/BaseProactorEventLoop._make_read_pipe_transportc C s t | ||||�S rB )r� r� r r r �_make_write_pipe_transport� s �z0BaseProactorEventLoop._make_write_pipe_transportc sj | � � rtd��| �� r d S t�� t�� kr6t�d� | �� | � � | j � � d | _ d | _ t � � � d S )Nz!Cannot close a running event loop���)Z is_runningr� � is_closedr� r� r� r� r� �_stop_accept_futures�_close_self_piperu rI r� r rD r3 r r rI � s zBaseProactorEventLoop.closec � s | j �||�I d H S rB )ru rv )r- r �nr r r � sock_recv� s z BaseProactorEventLoop.sock_recvc � s | j �||�I d H S rB )ru Z recv_into)r- r Zbufr r r �sock_recv_into� s z$BaseProactorEventLoop.sock_recv_intoc � s | j �||�I d H S rB )ru r� )r- r rl r r r � sock_sendall� s z"BaseProactorEventLoop.sock_sendallc � s | j �||�I d H S rB )ru Zconnect)r- r r� r r r � sock_connect� s z"BaseProactorEventLoop.sock_connectc � s | j �|�I d H S rB )ru �acceptrA r r r � sock_accept� s z!BaseProactorEventLoop.sock_acceptc � s( z |� � }W n2 ttjfk r> } zt�d��W 5 d }~X Y nX zt�|�j}W n, t k r| } zt�d��W 5 d }~X Y nX |r�|n|}|s�dS t |d�}|r�t || |�n|} t ||�}d} zLt | | |�}|dkr�| W �0S | j � ||||�I d H ||7 }| |7 } q�W 5 | dk�r"|� |� X d S )Nznot a regular filer l �� )r: �AttributeError�io�UnsupportedOperationr ZSendfileNotAvailableError�os�fstat�st_sizerR �min�seekru �sendfile) r- r �file�offset�countr: �errZfsizeZ blocksizeZend_posZ total_sentr r r �_sock_sendfile_native� s0 z+BaseProactorEventLoop._sock_sendfile_nativec � sZ |� � }|�� |�� I d H z | j|j|||dd�I d H W �S |�� |rT|�� X d S )NF)Zfallback)rh ri r� r� rm Z sock_sendfiler )r- Ztranspr� r� r� rm r r r �_sendfile_native� s �z&BaseProactorEventLoop._sendfile_nativec C sL | j d k r| j �� d | _ | j�� d | _| j�� d | _| jd8 _d S )Nr )r� rH �_ssockrI r� � _internal_fdsrD r r r r� � s z&BaseProactorEventLoop._close_self_pipec C s: t �� \| _| _| j�d� | j�d� | jd7 _d S )NFr )r Z socketpairr� r� Z setblockingr� rD r r r r� � s z%BaseProactorEventLoop._make_self_pipec C s� z4|d k r|� � | j|k r"W d S | j�| jd�}W nb tjk rL Y d S ttfk rd � Y nF t k r� } z| � d|| d�� W 5 d }~X Y nX || _|� | j � d S )Ni z.Error on reading from the event loop self pipe)rN rO r. ) rt r� ru rv r� r ry rn ro rp rT rz �_loop_self_reading)r- r� rU r r r r� � s$ �z(BaseProactorEventLoop._loop_self_readingc C sN | j }|d krd S z|�d� W n( tk rH | jrDtjddd� Y nX d S )N� z3Fail to write a null byte into the self-pipe socketTr )r� r� rR �_debugr rS )r- Zcsockr r r �_write_to_self s �z$BaseProactorEventLoop._write_to_self�d c s( d� ������fdd� � �� � � d S )Nc s, z�| d k rn| � � \}}�jr,t�d�||� �� }�d k rX�j||�dd|i��d� n�j||d|i�d� ��� r|W d S �j���} W n� t k r� } zH�� � dkrʈ� d|t � ��d�� ��� n�jr�tjd �dd � W 5 d }~X Y n8 tjk �r ��� Y nX | �j�� � < | �� � d S ) Nz#%r got a new connection from %r: %rTr )r� r1 r2 r� r� r� zAccept failed on a socket)rN rO r zAccept failed on socket %rr )rt r� r rS r� r� r� ru r� rR r: rT r r rI r ry r� rz )r� Zconnr� r/ rU �r. �protocol_factoryr- r2 r r� r� r r r. / s\ � � � � �z2BaseProactorEventLoop._start_serving.<locals>.loop)N)r* )r- r� r r� r2 Zbacklogr� r r� r �_start_serving+ s %z$BaseProactorEventLoop._start_servingc C s d S rB r )r- Z event_listr r r �_process_eventsV s z%BaseProactorEventLoop._process_eventsc C s& | j �� D ] }|�� q | j �� d S rB )r� �valuesrH �clear)r- �futurer r r r� Z s z*BaseProactorEventLoop._stop_accept_futuresc C s6 | j �|�� d �}|r |�� | j�|� |�� d S rB )r� �popr: rH ru � _stop_servingrI )r- r r� r r r r� _ s z#BaseProactorEventLoop._stop_serving)NNN)N)NNN)NN)NN)NN)N)NNr� N)r8 r^ r_ r r� r� r� r� r� r� rI r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rc r r r3 r r k s\ � � � � � � � � +r )#r` �__all__r� r� r ra r� r� r� � r r r r r r r r �logr r Z_FlowControlMixinZ BaseTransportr Z ReadTransportrd ZWriteTransportr| r� r� Z Transportr� r� Z BaseEventLoopr r r r r �<module> sR � � �n � �