← Back
Editing: proactor_events.cpython-310.pyc
o 0�i � @ sX 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� deej �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 n tjy( | j�� r&tj d|dd� Y nw d| jvrIz |� � | jd< W d S tjyH d | jd< Y d S w d S )N�socketZsocknamezgetsockname() failed on %rT��exc_info�peername) r �TransportSocket�_extraZ getsocknamer �error�_loop� get_debugr �warningZ getpeername)� transport�sock� r �>/opt/alt/python310/lib64/python3.10/asyncio/proactor_events.py�_set_socket_extra s" ��� ��r c s� e Zd ZdZ d � 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| _ d| _ | jd ur9| j�� | j�| jj| � |d urQ| j�tj|d � d S d S )Nr F)�super�__init__� _set_extra�_sock� set_protocol�_server�_buffer� _read_fut� _write_fut�_pending_write� _conn_lost�_closing�_called_connection_lost� _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 u r|�d� n| jr|�d� | jd ur(|�d| j�� � �� | jd ur6|�d| j��� | jd urD|�d| j��� | jrR|�dt | j�� �� | j rZ|�d� d� d � |��S ) N�closed�closingzfd=zread=zwrite=zwrite_bufsize=z EOF writtenz<{}>� ) r5 �__name__r �appendr( �filenor$ r% r# �lenr* �format�join)r. �infor r r �__repr__I s z#_ProactorBasePipeTransport.__repr__c C s || j d< d S )N�pipe)r �r. r r r r r [ � z%_ProactorBasePipeTransport._set_extrac C s || _ d S �N�r, )r. r0 r r r r! ^ � z'_ProactorBasePipeTransport.set_protocolc C � | j S rD rE �r. r r r � get_protocola � z'_ProactorBasePipeTransport.get_protocolc C rG rD )r( rH r r r � is_closingd rJ z%_ProactorBasePipeTransport.is_closingc C s` | j rd S d| _ | jd7 _| js | jd u r | j�| jd � | jd ur.| j�� d | _d S d S )NTr ) r( r' r# r% r r+ �_call_connection_lostr$ �cancelrH r r r �closeg s �z _ProactorBasePipeTransport.closec C s0 | j d ur|d| ��t| d� | j �� d S d S )Nzunclosed transport )�source)r �ResourceWarningrN )r. Z_warnr r r �__del__r s �z"_ProactorBasePipeTransport.__del__� Fatal error on pipe transportc C sn z0t |t�r| j�� rtjd| |dd� n| j�||| | jd�� W | �|� d S W | �|� d S | �|� w )Nz%r: %sTr )�message� exceptionr r0 ) � isinstance�OSErrorr r r �debug�call_exception_handlerr, � _force_close)r. �excrS r r r � _fatal_errorw s � �� z'_ProactorBasePipeTransport._fatal_errorc C s� | j d ur| j �� s|d u r| j �d � n| j �|� | jr#| jr#d S d| _| jd7 _| jr8| j�� d | _| j rC| j �� d | _ d| _ d | _ | j � | j|� d S )NTr r )� _empty_waiter�done� set_resultZ set_exceptionr( r) r' r% rM r$ r& r# r r+ rL )r. rZ r r r rY � s" z'_ProactorBasePipeTransport._force_closec C s� | j rd S z6| j�|� W t| jd�r!| j�� dkr!| j�tj� | j� � d | _| j }|d ur7|� � d | _ d| _ d S t| jd�rP| j�� dkrP| j�tj� | j� � d | _| j }|d urf|� � d | _ d| _ w )N�shutdown���T) r) r, Zconnection_lost�hasattrr r; r_ r Z SHUT_RDWRrN r" Z_detach)r. rZ r3 r r r rL � s, � z0_ProactorBasePipeTransport._call_connection_lostc C s" | j }| jd ur|t| j�7 }|S rD )r& r# r< )r. �sizer r r �get_write_buffer_size� s z0_ProactorBasePipeTransport.get_write_buffer_size�NNN)rR )r9 � __module__� __qualname__�__doc__r r@ r r! rI rK rN �warnings�warnrQ r[ rY rL rc � __classcell__r r r4 r r . s � r c sX e Zd ZdZ d� fdd� Zdd� Zdd � Zd d � Zd d � Zdd� Z ddd�Z � Z S )�_ProactorReadPipeTransportzTransport for read pipes.N� c sD d| _ d| _t� �||||||� t|�| _| j�| j� d| _d S )Nr` TF) �_pending_data_length�_pausedr r � bytearray�_datar r+ � _loop_reading)r. r/ r r0 r1 r2 r3 Z buffer_sizer4 r r r � s z#_ProactorReadPipeTransport.__init__c C s | j o| j S rD )rn r( rH r r r � is_reading� s z%_ProactorReadPipeTransport.is_readingc C s4 | j s| jrd S d| _| j�� rt�d| � d S d S )NTz%r pauses reading)r( rn r r r rW rH r r r � pause_reading� s �z(_ProactorReadPipeTransport.pause_readingc C s~ | j s| jsd S d| _| jd u r| j�| jd � | j}d| _|dkr0| j�| j| jd |� |� | j� � r=t � d| � d S d S )NFr` z%r resumes reading) r( rn r$ r r+ rq rm �_data_receivedrp r r rW )r. �lengthr r r �resume_reading� s �z)_ProactorReadPipeTransport.resume_readingc C s~ | j �� r t�d| � z| j�� }W n" ttfy � ty4 } z | � |d� W Y d }~d S d }~ww |s=| � � d S d S )Nz%r received EOFz1Fatal error: protocol.eof_received() call failed.) r r r rW r, Z eof_received� SystemExit�KeyboardInterrupt� BaseExceptionr[ rN )r. Z keep_openrZ r r r � _eof_received� s ��� �z(_ProactorReadPipeTransport._eof_receivedc C s� | j r| jdks J �|| _d S |dkr| �� d S t| jtj�rMz t�| j|� W d S tt fy4 � t yL } z | � |d� W Y d }~d S d }~ww | j� |� d S )Nr` r z3Fatal error: protocol.buffer_updated() call failed.) rn rm rz rU r, r ZBufferedProtocolZ _feed_data_to_buffered_protorw rx ry r[ Z data_received)r. �dataru rZ r r r rt � s( ���z)_ProactorReadPipeTransport._data_receivedc C s� d}d }z�za|d urD| j |u s| j d u r| jsJ �d | _ |�� r@|�� }|dkr8W W |dkr6| �||� d S d S | jd |� }n|�� | jrWW W |dkrU| �||� d S d S | jse| jj � | j | j�| _ W nb t y� } z | jsx| � |d� n | j�� r�tjddd� W Y d }~nId }~w ty� } z | �|� W Y d }~n3d }~w ty� } z | � |d� W Y d }~n d }~w tjy� | jsƂ Y n w | js�| j �| j� W |dkr�| �||� d S d S |dkr�| �||� w w )Nr` r z"Fatal read error on pipe transportz*Read error on pipe transport while closingTr )r$ r( r] �resultrt rp rM rn r � _proactor� recv_intor �ConnectionAbortedErrorr[ r r rW �ConnectionResetErrorrY rV r �CancelledError�add_done_callbackrq )r. �futru r{ rZ r r r rq sh �"����� ������� ��z(_ProactorReadPipeTransport._loop_reading)NNNrl rD ) r9 re rf rg r rr rs rv rz rt rq rj r r r4 r rk � s � rk 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|i |�� d | _d S rD )r r r\ �r. �args�kwr4 r r r M s z(_ProactorBaseWritePipeTransport.__init__c C s� t |tttf�stdt|�j� ���| jrtd��| j d ur"td��|s&d S | j r=| j t j kr4t �d� | j d7 _ d S | jd u rS| jd u sIJ �| jt|�d� d S | jsat|�| _| �� d S | 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 )r{ )rU �bytesro � memoryview� TypeError�typer9 r* � RuntimeErrorr\ r' r �!LOG_THRESHOLD_FOR_CONNLOST_WRITESr r r% r# � _loop_writing�_maybe_pause_protocol�extend)r. r{ r r r �writeQ s2 �� z%_ProactorBaseWritePipeTransport.writeNc C s� z�|d ur| j d u r| jrW d S || j u sJ �d | _ d| _|r#|�� |d u r-| j}d | _|sI| jr:| j�| jd � | jrD| j � t j � | � � n.| jj�| j |�| _ | j �� sp| jdks_J �t|�| _| j �| j� | �� n| j �| j� | jd ur�| j d u r�| j�d � W d S W d S W d S ty� } z | �|� W Y d }~d S d }~w ty� } z | �|d� W Y d }~d S d }~ww )Nr z#Fatal write error on pipe transport)r% r( r& r| r# r r+ rL r* r r_ r �SHUT_WR�_maybe_resume_protocolr} �sendr] r< r� r� r� r\ r^ r� rY rV r[ )r. �fr{ rZ r r r r� w sD ����z-_ProactorBaseWritePipeTransport._loop_writingc C � dS �NTr rH r r r � can_write_eof� � z-_ProactorBaseWritePipeTransport.can_write_eofc C s | � � d S rD )rN rH r r r � write_eof� s z)_ProactorBaseWritePipeTransport.write_eofc C � | � d � d S rD �rY rH r r r �abort� rC z%_ProactorBaseWritePipeTransport.abortc C s: | j d ur td��| j�� | _ | jd u r| j �d � | j S )NzEmpty waiter is already set)r\ r� r Z create_futurer% r^ rH r r r �_make_empty_waiter� s z2_ProactorBaseWritePipeTransport._make_empty_waiterc C s d | _ d S rD )r\ rH r r r �_reset_empty_waiter� rF z3_ProactorBaseWritePipeTransport._reset_empty_waiter�NN)r9 re rf rg Z_start_tls_compatibler r� r� r� r� r� r� r� rj r r r4 r r� G s &)r� c s$ e Zd Z� fdd�Zdd� Z� ZS )�_ProactorWritePipeTransportc s8 t � j|i |�� | jj�| jd�| _| j�| j� d S )N� ) r r r r} �recvr r$ r� � _pipe_closedr� r4 r r r � s z$_ProactorWritePipeTransport.__init__c C sx |� � rd S |�� dksJ �| jr| jd u sJ �d S || ju s&J || jf��d | _| jd ur6| �t� � d S | �� d S )N� )Z cancelledr| r( r$ r% rY �BrokenPipeErrorrN )r. r� r r r r� � s z(_ProactorWritePipeTransport._pipe_closed)r9 re rf r r� rj r r r4 r r� � s r� c s\ e Zd ZdZ d� 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)r1 r2 ) �_addressr\ r r � collections�dequer# r r+ rq )r. r/ r r0 �addressr1 r2 r4 r r r � s z#_ProactorDatagramTransport.__init__c C � t | |� d S rD �r rB r r r r � rC z%_ProactorDatagramTransport._set_extrac C s t dd� | jD ��S )Nc s s � | ] \}}t |�V qd S rD )r< )�.0r{ �_r r r � <genexpr>� s � zC_ProactorDatagramTransport.get_write_buffer_size.<locals>.<genexpr>)�sumr# rH r r r rc � s z0_ProactorDatagramTransport.get_write_buffer_sizec C r� rD r� rH r r r r� � rC z _ProactorDatagramTransport.abortc C s� t |tttf�stdt|���|sd S | jd ur'|d | jfvr'td| j� ���| jrA| jrA| jt j kr8t � d� | jd7 _d S | j �t|�|f� | jd u rT| �� | �� d S )Nz,data argument must be bytes-like object (%r)z!Invalid address: must be None or z!socket.sendto() raised exception.r )rU r� ro r� r� r� r� � ValueErrorr' r r� r r r# r: r% r� r� )r. r{ �addrr r r �sendto� s& � � z!_ProactorDatagramTransport.sendtoc C s" zR| j rW d S || ju sJ �d | _|r|�� | jr | j r.| jr.| jr+| j�| jd � W d S | j� � \}}| jd urE| jj � | j |�| _n | jj j | j ||d�| _W n1 tyk } z | j�|� W Y d }~d S d }~w ty� } z | �|d� W Y d }~d S d }~ww | j�| j� | �� d S )N)r� z'Fatal write error on datagram transport)r' r% r| r# r� r( r r+ rL �popleftr} r� r r� rV r, �error_received� Exceptionr[ r� r� r� )r. r� r{ r� rZ r r r r� � s< � ����� z(_ProactorDatagramTransport._loop_writingc C s� d }z�z�| j rW W |r| j�||� d S d S | j|u s%| jd u r#| js%J �d | _|d urT|�� }| jrDd }W W |rB| j�||� d S d S | jd urP|| j}}n|\}}| j rfW W |rd| j�||� d S d S | jd urw| jj� | j | j �| _n | jj� | j | j �| _W n% t y� } z | j�|� W Y d }~n d }~w tjy� | js�� Y n w | jd ur�| j�| j� W |r�| j�||� d S d S |r�| j�||� w w rD )r' r, Zdatagram_receivedr$ r( r| r� r r} r� r �max_sizeZrecvfromrV r� r r� r� rq )r. r� r{ r� �resrZ r r r rq sf $���� �� � � ����� ���z(_ProactorDatagramTransport._loop_readingrd rD ) r9 re rf r� r r rc r� r� r� rq rj r r r4 r r� � s � !r� c @ s e Zd ZdZdd� Zdd� ZdS )� _ProactorDuplexPipeTransportzTransport for duplex pipes.c C r� )NFr rH r r r r� Q r� z*_ProactorDuplexPipeTransport.can_write_eofc C s t �rD )�NotImplementedErrorrH r r r r� T r� z&_ProactorDuplexPipeTransport.write_eofN)r9 re rf rg r� r� r r r r r� L s r� c sF e Zd ZdZejjZ d � fdd� Zdd� Z dd� Z d d � Z � Z S ) �_ProactorSocketTransportz Transport for connected sockets.Nc s$ t � �||||||� t�|� d S rD )r r r Z _set_nodelayr- r4 r r r _ s z!_ProactorSocketTransport.__init__c C r� rD r� rB r r r r d rC z#_ProactorSocketTransport._set_extrac C r� r� r rH r r r r� g r� z&_ProactorSocketTransport.can_write_eofc C s6 | j s| jrd S d| _| jd u r| j�tj� d S d S r� )r( r* r% r r_ r r� rH r r r r� j s �z"_ProactorSocketTransport.write_eofrd ) r9 re rf rg r Z _SendfileModeZ TRY_NATIVEZ_sendfile_compatibler r r� r� rj r r r4 r r� X s �r� c s e Zd Z� fdd�Z d3dd�Z d4dddddd�dd �Z d3d d �Z d5d d �Z d5dd�Z d5dd�Z � fdd�Z dd� Z dd� Z dd� Z dd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd4d&d'�Zd(d)� Z * d6d+d,�Zd-d.� Zd/d0� Zd1d2� Z� ZS )7r c sl t � �� t�d|jj� || _|| _d | _i | _ |� | � | � � t � � t �� u r4t�| j�� � d S d S )NzUsing proactor: %s)r r r rW r5 r9 r} � _selector�_self_reading_future�_accept_futuresZset_loop�_make_self_pipe� threading�current_thread� main_thread�signal� set_wakeup_fd�_csockr; )r. Zproactorr4 r r r t s �z BaseProactorEventLoop.__init__Nc C � t | |||||�S rD )r� )r. r r0 r1 r2 r3 r r r �_make_socket_transport� � �z,BaseProactorEventLoop._make_socket_transportF)� server_side�server_hostnamer2 r3 �ssl_handshake_timeoutc C s0 t j| |||||| d�} t| || ||d� | jS )N)r� �r2 r3 )r Z SSLProtocolr� Z_app_transport) r. Zrawsockr0 � sslcontextr1 r� r� r2 r3 r� Z ssl_protocolr r r �_make_ssl_transport� s ��z)BaseProactorEventLoop._make_ssl_transportc C r� rD )r� )r. r r0 r� r1 r2 r r r �_make_datagram_transport� r� z.BaseProactorEventLoop._make_datagram_transportc C � t | ||||�S rD )r� �r. r r0 r1 r2 r r r �_make_duplex_pipe_transport� s �z1BaseProactorEventLoop._make_duplex_pipe_transportc C r� rD )rk r� r r r �_make_read_pipe_transport� s z/BaseProactorEventLoop._make_read_pipe_transportc C r� rD )r� r� r r r �_make_write_pipe_transport� s �z0BaseProactorEventLoop._make_write_pipe_transportc sj | � � rtd��| �� rd S t�� t�� u rt�d� | �� | � � | j � � d | _ d | _ t � � � d S )Nz!Cannot close a running event loopr` )Z is_runningr� � is_closedr� r� r� r� r� �_stop_accept_futures�_close_self_piper} rN r� r rH r4 r r rN � s zBaseProactorEventLoop.closec � � �| j �||�I d H S rD )r} r� )r. r �nr r r � sock_recv� � �z BaseProactorEventLoop.sock_recvc � r� rD )r} r~ )r. r Zbufr r r �sock_recv_into� r� z$BaseProactorEventLoop.sock_recv_intoc � r� rD )r} r� )r. r r{ r r r � sock_sendall� r� z"BaseProactorEventLoop.sock_sendallc � r� rD )r} Zconnect)r. r r� r r r � sock_connect� r� z"BaseProactorEventLoop.sock_connectc � s �| j �|�I d H S rD )r} �acceptrB r r r � sock_accept� s �z!BaseProactorEventLoop.sock_acceptc � s �z|� � }W n ttjfy } zt�d��d }~ww zt�|�j}W n t y0 t�d��w |r5|n|}|s;dS t |d�}|rIt || |�n|} t ||�}d} z. t | | |�}|dkrl| W | dkrk|� |� S S | j � ||||�I d H ||7 }| |7 } qT| dkr�|� |� w w )Nznot a regular filer l �� )r; �AttributeError�io�UnsupportedOperationr ZSendfileNotAvailableError�os�fstat�st_sizerV �min�seekr} �sendfile) r. r �file�offset�countr; �errZfsizeZ blocksizeZend_posZ total_sentr r r �_sock_sendfile_native� sD � �� � ��� �z+BaseProactorEventLoop._sock_sendfile_nativec � sn �|� � }|�� |�� I d H z| j|j|||dd�I d H W |�� |r*|�� S S |�� |r6|�� w w )NF)Zfallback)rr rs r� Z sock_sendfiler r� rv )r. Ztranspr� r� r� rv r r r �_sendfile_native� s � � �� �z&BaseProactorEventLoop._sendfile_nativec C sL | j d ur | j �� d | _ | j�� d | _| j�� d | _| jd8 _d S )Nr )r� rM �_ssockrN r� � _internal_fdsrH 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� rH r r r r� � s z%BaseProactorEventLoop._make_self_pipec C s� z|d ur |� � | j|urW d S | j�| jd�}W n/ tjy% Y d S ttfy. � t yI } z| � d|| d�� W Y d }~d S d }~ww || _|� | j � d S )Ni z.Error on reading from the event loop self pipe)rS rT r/ ) r| r� r} r� r� r r� rw rx ry rX r� �_loop_self_reading)r. r� rZ r r r r� s( ���z(BaseProactorEventLoop._loop_self_readingc C sR | j }|d u r d S z|�d� W d S ty( | jr%tjddd� Y d S Y d S w )N� z3Fail to write a null byte into the self-pipe socketTr )r� r� rV �_debugr rW )r. Zcsockr r r �_write_to_self s ���z$BaseProactorEventLoop._write_to_self�d c s( d� ������fdd� � �� � � d S )Nc sJ zE| d ur7| � � \}}�jrt�d�||� �� }�d ur,�j||�dd|i��d� n �j||d|i�d� ��� r>W d S �j���} W nQ t y� } z7�� � dkrd�� d|t � ��d�� ��� n�jrvtjd �dd � W Y d }~d S W Y d }~d S W Y d }~d S d }~w tjy� ��� Y d S w | �j�� � < | �� � d S ) Nz#%r got a new connection from %r: %rTr )r� r2 r3 r� r� r` zAccept failed on a socket)rS rT r zAccept failed on socket %rr )r| r� r rW r� r� r� r} r� rV r; rX r r rN r r� r� r� )r� Zconnr� r0 rZ �r/ �protocol_factoryr. r3 r r� r� r r r/ 6 sR ��� � �����z2BaseProactorEventLoop._start_serving.<locals>.looprD )r+ )r. r� r r� r3 Zbacklogr� r r� r �_start_serving2 s %z$BaseProactorEventLoop._start_servingc C s d S rD r )r. Z event_listr r r �_process_events] s z%BaseProactorEventLoop._process_eventsc C s& | j �� D ]}|�� q| j �� d S rD )r� �valuesrM �clear)r. �futurer r r r� a s z*BaseProactorEventLoop._stop_accept_futuresc C s6 | j �|�� d �}|r|�� | j�|� |�� d S rD )r� �popr; rM r} � _stop_servingrN )r. r r� r r r r f s z#BaseProactorEventLoop._stop_servingrd rD r� )NNr� N)r9 re rf r r� r� r� r� r� r� rN r� r� r� r� r� r� r� r� r� r� r� r� r� r� r rj r r r4 r r r sP �� � � � � � �+r )$rg �__all__r� r� r rh r� r� r� � r r r r r r r r �logr r Z_FlowControlMixinZ BaseTransportr Z ReadTransportrk ZWriteTransportr� r� ZDatagramTransportr� Z Transportr� r� Z BaseEventLoopr r r r r �<module> sX � � �n � � �
Save File
Cancel