a
XC?h � � @ s� d dl Z e jdkred��d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z ddl m
Z
ddl mZ ddl mZ ddl mZ dd l mZ dd
l mZ dd l mZ dd l mZ dd
lmZ dZd ZdZdZdZdZ dZ G dd� dej �Z G dd� dej �Z G dd� de �Z!G dd� de �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'e
j-�Z.G d(d)� d)e
j-�Z/e/Z0dS )*� 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 Zdd�� fdd�
Z� fdd�Zdd� Zd� fd d
� Z� fd d �Z� fd
d�Z� Z S )�_OverlappedFutureN��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__r r% r6 r0 r; r= �
__classcell__r r r r! r 0 s r c sl e Zd Zdd�� fdd�
Zdd� Z� fdd�Zd d
� Zd d � Zd� fd
d� Z� fdd�Z � fdd�Z
� Z S )�_BaseWaitHandleFutureNr 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 )�_winapiZWaitForSingleObjectrE 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% �appendrE rM rF 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/ ) rG rF � _overlappedZUnregisterWaitr1 �winerror�ERROR_IO_PENDINGr r2 r3 rQ �r rI r4 r5 r r r! �_unregister_wait� s$ �
$z&_BaseWaitHandleFuture._unregister_waitc s | � � t� j|d�S r7 )rW r r0 r9 r r r! r0 � s z _BaseWaitHandleFuture.cancelc s | � � t� �|� d S r: )rW r r; r< r r r! r; � s z#_BaseWaitHandleFuture.set_exceptionc s | � � t� �|� d S r: )rW r r= r> r r r! r= � s z _BaseWaitHandleFuture.set_result)N) r@ rA rB r rM r% rQ rW r0 r; r= rC r r r r! rD ` s
rD c sB e Zd Zdd�� fdd�
Zdd� Z� fdd�Z� fd d
�Z� ZS ) �_WaitCancelFutureNr c s t � j||||d� d | _d S )Nr )r r �_done_callback)r r �eventrI r r r r! r � s z_WaitCancelFuture.__init__c C s t d��d S )Nz'_WaitCancelFuture must not be cancelled)� RuntimeErrorrL r r r! r0 � s z_WaitCancelFuture.cancelc s$ t � �|� | jd ur | �| � d S r: )r r= rY r> r r r! r= � s
z _WaitCancelFuture.set_resultc s$ t � �|� | jd ur | �| � d S r: )r r; rY r< r r r! r; � s
z _WaitCancelFuture.set_exception)r@ rA rB r r0 r= r; rC r r r r! rX � s rX 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_proactorrS Z CreateEvent�_event�
_event_fut)r r rH rI �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^ rK � CloseHandler_ r] � _unregisterr r rQ rO r r r! rQ � 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 rR )rG rF rS ZUnregisterWaitExr^ r1 rT rU r r2 r3 r] � _wait_cancelrQ r_ rV r r r! rW � s( �
$
�z"_WaitHandleFuture._unregister_wait)r@ rA rB r rQ rW rC r r r r! r\ � s r\ c @ s8 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� ZeZd S ) �
PipeServerc 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)rj rl )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: )�closedrK ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPiperf ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr
ZBUFSIZEZNMPWAIT_WAIT_FOREVER�NULL�
PipeHandleri �add)r �first�flags�h�piper r r! rl
s"
��
z PipeServer._server_pipe_handlec C s
| j d u S r: )rf rL r r r! rp s zPipeServer.closedc C sR | j d ur| j �� d | _ | jd urN| jD ] }|�� q*d | _d | _| j�� d S r: )rk r0 rf ri �closerj �clear)r rw r r r! rx " s
zPipeServer.closeN) r@ rA rB r ro rl rp rx �__del__r r r r! rd � s
rd c @ s e Zd ZdS )�_WindowsSelectorEventLoopN)r@ rA rB r r r r! r{ 1 s r{ c sD e Zd Zd � fdd� Z� fdd�Zdd� Zdd � Zd
d
d �Z� ZS )r
Nc s |d u rt � }t� �|� d S r: )r r r )r r` 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] rb �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? ri �discardrp rx r� ro r] � accept_piper1 �filenor3 Z_debugr Zwarningr �CancelledErrorrk �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)rd 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 r r} r� r� r� rC r r r r! r
5 s 0 �r
c @ s� e Zd Zd:dd�Zdd� Zdd� Zdd � Zd;d d �Zd
d� Zddd�Z d?dd�Z d@dd�Z
dd� Zd d � Zd d � Zd d!� Zd"d#� ZdAd$d%�Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� ZdBd2d3�Zd4d5� Zd6d7� Z d8d9� Z d
S )Cr r c C sD d | _ g | _t�tjtd|�| _i | _t� � | _
g | _ t� � | _ d S rJ )
r2 �_resultsrS �CreateIoCompletionPort�INVALID_HANDLE_VALUErq �_iocp�_cacherg rh rG �
_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[ rL 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� rN 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� rM )r �timeoutrn r r r! �select� s
zIocpProactor.selectc C s | j �� }|�|� |S r: )r2 r� r= )r �valuerP 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 rT rS ZERROR_NETNAME_DELETEDZERROR_OPERATION_ABORTED�ConnectionResetErrorr� �r� �keyr r4 r r r! � finish_recv� s
� z&IocpProactor.recv..finish_recv) �_register_with_iocprS �
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� rS 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� rS 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� rS 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� rS 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_SOCKETrS 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�familyrS 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 rJ )r� r� r� r� rS ZSO_UPDATE_CONNECT_CONTEXT�r� r� r �r� r r! �finish_connect[ s
�z,IocpProactor.connect..finish_connect)�typer� Z
SOCK_DGRAMrS Z
WSAConnectr� r2 r� r= r� Z BindLocalr� r1 rT �errnoZ WSAEINVALZ getsocknamer� rq Z ConnectExr� )r r� r' rP �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� rS 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� rS 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_DELAYrS Z ConnectPiper1 rT ZERROR_PIPE_BUSY�min�CONNECT_PIPE_MAX_DELAYr �sleepr
rr )r r' ZdelayrH r4 r r r! r� � s
zIocpProactor.connect_pipec C s | � ||d�S rm )�_wait_for_handle)r rH r� r r r! �wait_for_handle� s z IocpProactor.wait_for_handlec C s | � |d d�}||_|S re )r� rY )r rZ Z
done_callbackrP r r r! rc � 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: )rM r� �r� r r! �finish_wait_for_handle� s z=IocpProactor._wait_for_handle..finish_wait_for_handler )r� rK �INFINITE�math�ceilrS r� rq ZRegisterWaitWithQueuer� r' rX r2 r\ r r� )r rH r� Z
_is_cancel�msr rI r� r r� r! r� � s$
�
� z IocpProactor._wait_for_handlec C s0 || j vr,| j �|� t�|�� | jdd� d S rJ )rG rs rS 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�|� d S r: )r� r� rN r~ r r r! rb � s zIocpProactor._unregisterc C s t � |�}|�d� |S rJ )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� rS ZGetQueuedCompletionStatusr� r� �pop�KeyErrorr2 Z get_debugr3 r� rK ra r� r0 Zdoner1 r; r� rN r= r� r' ry )r r� r� r� �errZ transferredr� r' r� r r� r� r� r� r r r! rM 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� rX r0 r1 r2 r r3 �time� monotonicr �debugrM r� rK ra ) r r' rP r r� r� r4 r5 Z
msg_updateZ
start_timeZnext_msgr r r! rx D s>
�
$ � zIocpProactor.closec C s | � � d S r: )rx rL r r r! rz s s zIocpProactor.__del__)r )N)r )r )r )r N)r )N)N) r@ rA rB r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rc r� r� r� rb r� rM r� rx rz r r r r! r � s6
"
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� �
returncoderL r r! r� ~ s
z4_WindowsSubprocessTransport._start..callback) r
�Popenr r2 r] r� �intrE r� )
r r� r� r� r� r� r� r� r� r� r rL 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 )1�sys�platform� ImportErrorrS rK r� r� r� r� r� r� rg � r r r r r r r r
�logr �__all__rq r� ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr� r� ZFuturer rD rX r\ �objectrd ZBaseSelectorEventLoopr{ ZBaseProactorEventLoopr
r ZBaseSubprocessTransportr� r ZBaseDefaultEventLoopPolicyr r r r r r r! � sV
0J4;e `