a
XC?h � � @ s� d Z ddlZejdkr ed��ddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ ddl
mZ ddl
mZ dd l
mZ dd
l
mZ dd l
mZ dd l
mZ dd
l
mZ ddlmZ dZdZdZdZdZ dZ dZ G dd� dej �Z G dd� dej �Z!G dd� de!�Z"G dd � d e!�Z#G d d � d e$�Z%G d d � d ej&�Z'G d!d"� d"ej(�Z)G d#d$� d$�Z*G d%d&� d&ej+�Z,e'Z-G d'd(� d(ej.�Z/G d)d*� d*ej.�Z0e0Z1dS )+z.Selector and proactor event loops for Windows.� NZwin32z
win32 only� )�events)�base_subprocess)�futures)�
exceptions)�proactor_events)�selector_events)�tasks)�
windows_utils)�logger)�SelectorEventLoop�ProactorEventLoop� IocpProactor�DefaultEventLoopPolicy� WindowsSelectorEventLoopPolicy� WindowsProactorEventLoopPolicy� �� i� i� g����MbP?g�������?c s` e Zd ZdZdd�� fdd�
Z� fdd�Zdd � Zd� fd
d � Z� fd d
�Z� fdd�Z � Z
S )�_OverlappedFuturez�Subclass of Future which represents an overlapped operation.
Cancelling it will immediately cancel the overlapped operation.
N��loopc s& t � j|d� | jr | jd= || _d S �Nr �����)�super�__init__�_source_traceback�_ov)�self�ovr �� __class__� �;/opt/alt/python39/lib64/python3.9/asyncio/windows_events.pyr 6 s z_OverlappedFuture.__init__c sH t � �� }| jd urD| jjr dnd}|�dd|� d| jjd�d�� |S )N�pendingZ completedr z overlapped=)r �
_repr_infor r" �insert�address�r �info�stater r r! r% < s
z _OverlappedFuture._repr_infoc
C st | j d u rd S z| j �� W nL tyh } z4d|| d�}| jrH| j|d< | j�|� W Y d }~n
d }~0 0 d | _ d S )Nz&Cancelling an overlapped future failed��message� exception�future�source_traceback)r �cancel�OSErrorr �_loop�call_exception_handler)r �exc�contextr r r! �_cancel_overlappedC s
�
"z$_OverlappedFuture._cancel_overlappedc s | � � t� j|d�S �N)�msg)r6 r r0 �r r8 r r r! r0 S s z_OverlappedFuture.cancelc s t � �|� | �� d S �N)r �
set_exceptionr6 �r r- r r r! r; W s z _OverlappedFuture.set_exceptionc s t � �|� d | _d S r: )r �
set_resultr �r �resultr r r! r= [ s z _OverlappedFuture.set_result)N) �__name__�
__module__� __qualname__�__doc__r r% r6 r0 r; r= �
__classcell__r r r r! r 0 s r c sp e Zd ZdZdd�� fdd�
Zdd� Z� fdd �Zd
d � Zd d
� Zd� fdd� Z � fdd�Z
� fdd�Z � Z S )�_BaseWaitHandleFuturez2Subclass of Future which represents a wait handle.Nr c s8 t � j|d� | jr | jd= || _|| _|| _d| _d S )Nr r T)r r r r �_handle� _wait_handle� _registered)r r �handle� wait_handler r r r! r c s z _BaseWaitHandleFuture.__init__c C s t �| jd�t jkS �Nr )�_winapiZWaitForSingleObjectrF Z
WAIT_OBJECT_0�r r r r! �_pollq s �z_BaseWaitHandleFuture._pollc sd t � �� }|�d| jd��� | jd urB| �� r4dnd}|�|� | jd ur`|�d| jd��� |S )Nzhandle=r# ZsignaledZwaitingz wait_handle=)r r% �appendrF rN rG r( r r r! r% v s
z _BaseWaitHandleFuture._repr_infoc C s
d | _ d S r: )r �r �futr r r! �_unregister_wait_cb� s z)_BaseWaitHandleFuture._unregister_wait_cbc
C s� | j s
d S d| _ | j}d | _zt�|� W nf ty� } zN|jtjkr|d|| d�}| jrb| j|d< | j� |� W Y d }~d S W Y d }~n
d }~0 0 | �
d � d S �NFz$Failed to unregister the wait handler+ r/ ) rH rG � _overlappedZUnregisterWaitr1 �winerror�ERROR_IO_PENDINGr r2 r3 rR �r rJ r4 r5 r r r! �_unregister_wait� s$ �
$z&_BaseWaitHandleFuture._unregister_waitc s | � � t� j|d�S r7 )rX r r0 r9 r r r! r0 � s z _BaseWaitHandleFuture.cancelc s | � � t� �|� d S r: )rX r r; r< r r r! r; � s z#_BaseWaitHandleFuture.set_exceptionc s | � � t� �|� d S r: )rX r r= r> r r r! r= � s z _BaseWaitHandleFuture.set_result)N)
r@ rA rB rC r rN r% rR rX r0 r; r= rD r r r r! rE ` s
rE c sF e Zd ZdZdd�� fdd�
Zdd� Z� fdd �Z� fd
d �Z� ZS ) �_WaitCancelFuturezoSubclass of Future which represents a wait for the cancellation of a
_WaitHandleFuture using an event.
Nr c s t � j||||d� d | _d S )Nr )r r �_done_callback)r r �eventrJ r r r r! r � s z_WaitCancelFuture.__init__c C s t d��d S )Nz'_WaitCancelFuture must not be cancelled)� RuntimeErrorrM r r r! r0 � s z_WaitCancelFuture.cancelc s$ t � �|� | jd ur | �| � d S r: )r r= rZ r> r r r! r= � s
z _WaitCancelFuture.set_resultc s$ t � �|� | jd ur | �| � d S r: )r r; rZ r< r r r! r; � s
z _WaitCancelFuture.set_exception) r@ rA rB rC r r0 r= r; rD r r r r! rY � s
rY c s6 e Zd Zdd�� fdd�
Z� fdd�Zdd� Z� ZS ) �_WaitHandleFutureNr c s< t � j||||d� || _d| _t�d ddd �| _d | _d S )Nr TF)r r � _proactorZ_unregister_proactorrT Z CreateEvent�_event�
_event_fut)r r rI rJ �proactorr r r r! r � s
z_WaitHandleFuture.__init__c sF | j d ur"t�| j � d | _ d | _| j�| j� d | _t� �|� d S r: ) r_ rL � CloseHandler` r^ � _unregisterr r rR rP r r r! rR � s
z%_WaitHandleFuture._unregister_wait_cbc
C s� | j s
d S d| _ | j}d | _zt�|| j� W nf ty� } zN|jtjkr�d|| d�}| jrf| j|d< | j �
|� W Y d }~d S W Y d }~n
d }~0 0 | j � | j| j
�| _d S rS )rH rG rT ZUnregisterWaitExr_ r1 rU rV r r2 r3 r^ � _wait_cancelrR r` rW r r r! rX � s( �
$
�z"_WaitHandleFuture._unregister_wait)r@ rA rB r rR rX rD r r r r! r] � s r] c @ s< e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � ZeZ d S )
�
PipeServerzXClass representing a pipe server.
This is much like a bound, listening socket.
c C s, || _ t�� | _d | _d | _| �d�| _d S �NT)�_address�weakref�WeakSet�_free_instances�_pipe�_accept_pipe_future�_server_pipe_handle)r r' r r r! r � s
zPipeServer.__init__c C s | j | �d� }| _ |S )NF)rk rm )r �tmpr r r! �_get_unconnected_pipe s z PipeServer._get_unconnected_pipec
C sr | � � r d S tjtjB }|r&|tjO }t�| j|tjtjB tj B tj
t j t j tj
tj�}t �|�}| j�|� |S r: )�closedrL ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPiperg ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr
ZBUFSIZEZNMPWAIT_WAIT_FOREVER�NULL�
PipeHandlerj �add)r �first�flags�h�piper r r! rm
s"
��
z PipeServer._server_pipe_handlec C s
| j d u S r: )rg rM r r r! rp s zPipeServer.closedc C sR | j d ur| j �� d | _ | jd urN| jD ] }|�� q*d | _d | _| j�� d S r: )rl r0 rg rj �closerk �clear)r rw r r r! rx " s
zPipeServer.closeN)
r@ rA rB rC r ro rm rp rx �__del__r r r r! re � s
re c @ s e Zd ZdZdS )�_WindowsSelectorEventLoopz'Windows version of selector event loop.N)r@ rA rB rC r r r r! r{ 1 s r{ c sH e Zd ZdZd
� fdd� Z� fdd�Zdd� Zd d
� Zdd d �Z� Z S )r
z2Windows version of proactor event loop using IOCP.Nc s |d u rt � }t� �|� d S r: )r r r )r ra r r r! r 8 s zProactorEventLoop.__init__c s� zV| � | j� t� �� W | jd ur�| jj}| j�� |d urP|jsP| j� |� d | _n>| jd ur�| jj}| j�� |d ur�|js�| j� |� d | _0 d S r: )
� call_soonZ_loop_self_readingr � run_foreverZ_self_reading_futurer r0 r" r^ rc �r r r r r! r} = s
�
z ProactorEventLoop.run_foreverc � s8 | j �|�}|I d H }|� }| j||d|id�}||fS )N�addr��extra)r^ � connect_pipe�_make_duplex_pipe_transport)r �protocol_factoryr' �frw �protocol�transr r r! �create_pipe_connectionP s
�z(ProactorEventLoop.create_pipe_connectionc � s. t � ��d� ����fdd� ����� �gS )Nc
s d }zn| rN| � � }�j�|� ��� r4|�� W d S �� }�j||d� id� ��� }|d u rdW d S �j�|�} W n� t y� } zN|r�|�
� dkr��� d||d�� |�� n�j r�t
jd|dd� W Y d }~n8d }~0 tjy� |r�|�� Y n0 | �_| ��� d S ) Nr r� r zPipe accept failed)r, r- rw z Accept pipe failed on pipe %rT)�exc_info)r? rj �discardrp rx r� ro r^ � accept_piper1 �filenor3 Z_debugr Zwarningr �CancelledErrorrl �add_done_callback)r� rw r� r4 �r' �loop_accept_piper� r Zserverr r! r� [ sB
��
�z>ProactorEventLoop.start_serving_pipe..loop_accept_pipe)N)re r| )r r� r' r r� r! �start_serving_pipeX s (
z$ProactorEventLoop.start_serving_pipec � s� | � � }
t| |||||||f|
|d�| ��} z|
I d H W n@ ttfyR � Y n* tyz | �� | �� I d H � Y n0 | S )N)�waiterr� )�
create_future�_WindowsSubprocessTransport�
SystemExit�KeyboardInterrupt�
BaseExceptionrx Z_wait) r r� �args�shell�stdin�stdout�stderr�bufsizer� �kwargsr� Ztranspr r r! �_make_subprocess_transport� s"
��� z,ProactorEventLoop._make_subprocess_transport)N)N)
r@ rA rB rC r r} r� r� r� rD r r r r! r
5 s 0 �r
c @ s� e Zd ZdZd;dd�Zdd� Zdd� Zd d
� Zddd�Z d?dd�Z d@dd�Z
dAdd�Zdd � Zd d � Zd d � Zd!d"� Zd#d$� ZdBd%d&�Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� ZdCd3d4�Zd5d6� Z d7d8� Z d9d:� Z d S )Dr z#Proactor implementation using IOCP.r c C sD d | _ g | _t�tjtd|�| _i | _t� � | _
g | _ t� � | _ d S rK )
r2 �_resultsrT �CreateIoCompletionPort�INVALID_HANDLE_VALUErq �_iocp�_cacherh ri rH �
_unregistered�_stopped_serving)r Z concurrencyr r r! r � s
�
zIocpProactor.__init__c C s | j d u rtd��d S )NzIocpProactor is closed)r� r\ rM r r r! �
_check_closed� s
zIocpProactor._check_closedc C sF dt | j� dt | j� g}| jd u r0|�d� d| jjd�|�f S )Nzoverlapped#=%sz
result#=%srp z<%s %s>� )�lenr� r� r� rO r r@ �join)r r) r r r! �__repr__� s �
zIocpProactor.__repr__c C s
|| _ d S r: )r2 )r r r r r! �set_loop� s zIocpProactor.set_loopNc C s | j s| �|� | j }g | _ |S r: )r� rN )r �timeoutrn r r r! �select� s
zIocpProactor.selectc C s | j �� }|�|� |S r: )r2 r� r= )r �valuerQ r r r! �_result� s
zIocpProactor._resultr c C s| | � |� t�t�}z4t|tj�r6|�|�� ||� n|�|�� |� W n t yd | �
d� Y S 0 dd� }| � |||�S )N� c
S sT z
|� � W S tyN } z,|jtjtjfv r8t|j� �n� W Y d }~n
d }~0 0 d S r: �� getresultr1 rU rT ZERROR_NETNAME_DELETEDZERROR_OPERATION_ABORTED�ConnectionResetErrorr� �r� �keyr r4 r r r! � finish_recv� s
� z&IocpProactor.recv..finish_recv) �_register_with_iocprT �
Overlappedrq �
isinstance�socketZWSARecvr� ZReadFile�BrokenPipeErrorr� � _register�r �conn�nbytesru r r� r r r! �recv� s
zIocpProactor.recvc C s| | � |� t�t�}z4t|tj�r6|�|�� ||� n|�|�� |� W n t yd | �
d� Y S 0 dd� }| � |||�S )Nr c
S sT z
|� � W S tyN } z,|jtjtjfv r8t|j� �n� W Y d }~n
d }~0 0 d S r: r� r� r r r! r� � s
� z+IocpProactor.recv_into..finish_recv) r� rT r� rq r� r� Z WSARecvIntor� Z ReadFileIntor� r� r� )r r� �bufru r r� r r r! � recv_into� s
zIocpProactor.recv_intoc C s^ | � |� t�t�}z|�|�� ||� W n tyF | �d� Y S 0 dd� }| �|||�S )N)r� Nc
S sT z
|� � W S tyN } z,|jtjtjfv r8t|j� �n� W Y d }~n
d }~0 0 d S r: r� r� r r r! r� � s
� z*IocpProactor.recvfrom..finish_recv) r� rT r� rq Z WSARecvFromr� r� r� r� r� r r r! �recvfrom� s
zIocpProactor.recvfromc C s> | � |� t�t�}|�|�� |||� dd� }| �|||�S )Nc
S sT z
|� � W S tyN } z,|jtjtjfv r8t|j� �n� W Y d }~n
d }~0 0 d S r: r� r� r r r! � finish_send s
� z(IocpProactor.sendto..finish_send)r� rT r� rq Z WSASendTor� r� )r r� r� ru r r r� r r r! �sendto s
zIocpProactor.sendtoc C sZ | � |� t�t�}t|tj�r4|�|�� ||� n|�|�� |� dd� }| � |||�S )Nc
S sT z
|� � W S tyN } z,|jtjtjfv r8t|j� �n� W Y d }~n
d }~0 0 d S r: r� r� r r r! r� s
� z&IocpProactor.send..finish_send)
r� rT r� rq r� r� ZWSASendr� Z WriteFiler� )r r� r� ru r r� r r r! �send s
zIocpProactor.sendc sv | � �� | ��j�� t�t�}|���� � �� � � �fdd�}dd� }| �|�|�}||� �}t j
|| j d� |S )Nc sD |� � t�d��� �}� �tjtj|� � � ��
� � � � � � fS )Nz@P) r� �structZpackr� �
setsockoptr� �
SOL_SOCKETrT ZSO_UPDATE_ACCEPT_CONTEXT�
settimeoutZ
gettimeoutZ getpeername)r� r� r r� �r� �listenerr r! �
finish_accept/ s �z*IocpProactor.accept..finish_acceptc � s2 z| I d H W n t jy, |�� � Y n0 d S r: )r r� rx )r. r� r r r! � accept_coro8 s
z(IocpProactor.accept..accept_coror ) r� �_get_accept_socket�familyrT r� rq ZAcceptExr� r� r Z
ensure_futurer2 )r r� r r� r� r. �coror r� r! �accept) s
zIocpProactor.acceptc
s� � j tjkr4t�� �� |� | j�� }|�d � |S | � � � zt�
� �� � j � W nD t y� } z,|j
tjkrr� � �� d dkr�� W Y d }~n
d }~0 0 t�t�}|�� �� |� � fdd�}| �|� |�S )Nr r c s |� � � �tjtjd� � S rK )r� r� r� r� rT ZSO_UPDATE_CONNECT_CONTEXT�r� r� r �r� r r! �finish_connect[ s
�z,IocpProactor.connect..finish_connect)�typer� Z
SOCK_DGRAMrT Z
WSAConnectr� r2 r� r= r� Z BindLocalr� r1 rU �errnoZ WSAEINVALZ getsocknamer� rq Z ConnectExr� )r r� r' rQ �er r� r r� r! �connectE s"
zIocpProactor.connectc C sb | � |� t�t�}|d@ }|d? d@ }|�|�� t�|�� �|||dd� dd� }| �|||�S )Nr � r c
S sT z
|� � W S tyN } z,|jtjtjfv r8t|j� �n� W Y d }~n
d }~0 0 d S r: r� r� r r r! �finish_sendfilen s
� z.IocpProactor.sendfile..finish_sendfile) r� rT r� rq Z TransmitFiler� �msvcrtZ
get_osfhandler� ) r Zsock�file�offset�countr Z
offset_lowZ offset_highr� r r r! �sendfiled s
� zIocpProactor.sendfilec sJ | � � � t�t�}|�� �� �}|r0| �� �S � fdd�}| �|� |�S )Nc s |� � � S r: )r� r� �rw r r! �finish_accept_pipe� s z4IocpProactor.accept_pipe..finish_accept_pipe)r� rT r� rq ZConnectNamedPiper� r� r� )r rw r Z connectedr� r r� r! r� y s
zIocpProactor.accept_pipec
� st t }zt�|�}W qjW n2 tyH } z|jtjkr4� W Y d }~n
d }~0 0 t|d t�}t� |�I d H qt
� |�S )N� ) �CONNECT_PIPE_INIT_DELAYrT Z ConnectPiper1 rU ZERROR_PIPE_BUSY�min�CONNECT_PIPE_MAX_DELAYr �sleepr
rr )r r' ZdelayrI r4 r r r! r� � s
zIocpProactor.connect_pipec C s | � ||d�S )z�Wait for a handle.
Return a Future object. The result of the future is True if the wait
completed, or False if the wait did not complete (on timeout).
F)�_wait_for_handle)r rI r� r r r! �wait_for_handle� s z IocpProactor.wait_for_handlec C s | � |d d�}||_|S rf )r� rZ )r r[ Z
done_callbackrQ r r r! rd � s zIocpProactor._wait_cancelc s� | � � |d u rtj}nt�|d �}t�t�}t�|| j |j
|�}|r\t |||| j d�� nt
|||| | j d�� � jr~� jd= � fdd�}� |d|f| j|j
< � S )N� @�@r r c s � � � S r: )rN r� �r� r r! �finish_wait_for_handle� s z=IocpProactor._wait_for_handle..finish_wait_for_handler )r� rL �INFINITE�math�ceilrT r� rq ZRegisterWaitWithQueuer� r' rY r2 r] r r� )r rI r� Z
_is_cancel�msr rJ r� r r� r! r� � s$
�
� z IocpProactor._wait_for_handlec C s0 || j vr,| j �|� t�|�� | jdd� d S rK )rH rs rT r� r� r� �r �objr r r! r� � s
z IocpProactor._register_with_iocpc
C s� | � � t|| jd�}|jr$|jd= |jstz|d d |�}W n. tyh } z|�|� W Y d }~nd }~0 0 |�|� ||||f| j|j < |S r )
r� r r2 r r" r1 r; r= r� r' )r r r� �callbackr� r� r� r r r! r� � s
zIocpProactor._registerc C s | � � | j�|� dS )a
Unregister an overlapped object.
Call this method when its future has been cancelled. The event can
already be signalled (pending in the proactor event queue). It is also
safe if the event is never signalled (because it was cancelled).
N)r� r� rO r~ r r r! rc � s zIocpProactor._unregisterc C s t � |�}|�d� |S rK )r� r� )r r� �sr r r! r� s
z IocpProactor._get_accept_socketc C s� |d u rt }n0|dk r td��n t�|d �}|t kr>td��t�| j|�}|d u rX�qZd}|\}}}}z| j�|�\}} }
} W nV t y� | j
� � r�| j
� dd||||f d�� |dtj
fvr�t�|� Y q>Y n0 |
| jv r�|�� q>|�� s>z| ||| �} W n< t�y@ }
z"|�|
� | j�|� W Y d }
~
q>d }
~
0 0 |�| � | j�|� q>| jD ]} | j�| jd � �q`| j�� d S )Nr znegative timeoutr� ztimeout too bigz8GetQueuedCompletionStatus() returned an unexpected eventz)err=%s transferred=%s key=%#x address=%#x)r, �status)r� �
ValueErrorr� r� rT ZGetQueuedCompletionStatusr� r� �pop�KeyErrorr2 Z get_debugr3 r� rL rb r� r0 Zdoner1 r; r� rO r= r� r' ry )r r� r� r� �errZ transferredr� r' r� r r� r� r� r� r r r! rN sL
��
"
zIocpProactor._pollc C s | j �|� d S r: )r� rs r� r r r! �
_stop_serving> s zIocpProactor._stop_servingc C s | j d u rd S t| j�� �D ]�\}\}}}}|�� r6q t|t�rBq z |�� W q ty� } z>| j d ur�d||d�}|j
r�|j
|d< | j � |� W Y d }~q d }~0 0 q d}t �
� } | | }
| jr�|
t �
� kr�t�d| t �
� | � t �
� | }
| �|� q�g | _t�| j � d | _ d S )NzCancelling a future failedr+ r/ g �?z,%r is running after closing for %.1f seconds)r� �listr� �itemsZ cancelledr� rY r0 r1 r2 r r3 �time� monotonicr �debugrN r� rL rb ) r r' rQ r r� r� r4 r5 Z
msg_updateZ
start_timeZnext_msgr r r! rx D s>
�
$ � zIocpProactor.closec C s | � � d S r: )rx rM r r r! rz s s zIocpProactor.__del__)r )N)r )r )r )r N)r )N)N) r@ rA rB rC r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rd r� r� r� rc r� rN r� rx rz r r r r! r � s8
"
7/r c @ s e Zd Zdd� ZdS )r� c
sP t j|f|||||d�|��� _� fdd�}� jj�t� jj��} | �|� d S )N)r� r� r� r� r� c s � j �� }� �|� d S r: )�_procZpollZ_process_exited)r� �
returncoderM r r! r� ~ s
z4_WindowsSubprocessTransport._start..callback) r
�Popenr r2 r^ r� �intrF r� )
r r� r� r� r� r� r� r� r� r� r rM r! �_starty s ��� z"_WindowsSubprocessTransport._startN)r@ rA rB r r r r r! r� w s r� c @ s e Zd ZeZdS )r N)r@ rA rB r �
_loop_factoryr r r r! r � s r c @ s e Zd ZeZdS )r N)r@ rA rB r
r r r r r! r � s r )2rC �sys�platform� ImportErrorrT rL r� r� r� r� r� r� rh � r r r r r r r r
�logr �__all__rq r� ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr� r� ZFuturer rE rY r] �objectre ZBaseSelectorEventLoopr{ ZBaseProactorEventLoopr
r ZBaseSubprocessTransportr� r ZBaseDefaultEventLoopPolicyr r r r r r r! � sX
0J4;e `