B
�/g^v c @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl
m Z ddl
m
Z
ddl
mZ ddl
mZ dd l
mZ dd
l
mZ dd lmZ d ZdZd
ZdZdZdZdZG dd� de
j�Z G dd� de
j�Z G dd� de �Z G dd� de �Z G dd� de �Z!G d d � d ej"�Z#G d d � d ej$�Z%G d d!� d!�Z&G d"d#� d#e j'�Z(e#Z)G d$d%� d%e j*�Z+G d&d'� d'e j*�Z,e+Z-dS )(z.Selector and proactor event loops for Windows.� N� )�events)�base_subprocess)�futures)�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 � Z� 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__� �./usr/lib64/python3.7/asyncio/windows_events.pyr 0 s z_OverlappedFuture.__init__c sH t � �� }| jd k rD| jjr dnd}|�dd|� d| jjd�d�� |S )N�pendingZ completedr z overlapped=<z, �#xz>)r �
_repr_infor r Zinsert�address�r �infoZstater r r r" 6 s
z _OverlappedFuture._repr_infoc
C sr | j d krd S y| j �� W nJ tk
rf } z,d|| d�}| jrJ| j|d< | j�|� W d d }~X Y nX 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_overlapped= s
z$_OverlappedFuture._cancel_overlappedc � | � � t� �� S �N)r1 r r+ �r r r r r+ M � z_OverlappedFuture.cancelc s t � �|� | �� d S r3 )r �
set_exceptionr1 �r r( r r r r6 Q � z _OverlappedFuture.set_exceptionc s t � �|� d | _d S r3 )r �
set_resultr �r �resultr r r r9 U r8 z _OverlappedFuture.set_result) �__name__�
__module__� __qualname__�__doc__r r" r1 r+ r6 r9 �
__classcell__r r r r r * s r c sn e Zd ZdZdd�� fdd�
Zdd� Z� fdd �Zd
d � Zd d
� Z� 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 ] s z _BaseWaitHandleFuture.__init__c C s t �| jd�t jkS �Nr )�_winapiZWaitForSingleObjectrB Z
WAIT_OBJECT_0r4 r r r �_pollk s z_BaseWaitHandleFuture._pollc sd t � �� }|�d| jd��� | jd k rB| �� r4dnd}|�|� | jd k r`|�d| jd��� |S )Nzhandle=r! ZsignaledZwaitingz wait_handle=)r r" �appendrB rI rC r$ r r r r" p s
z _BaseWaitHandleFuture._repr_infoc C s
d | _ d S r3 )r �r �futr r r �_unregister_wait_cbz s z)_BaseWaitHandleFuture._unregister_wait_cbc
C s� | j s
d S d| _ | j}d | _yt�|� W nZ tk
r� } z<|jtjkrtd|| d�}| jrd| j|d< | j� |� d S W d d }~X Y nX | �
d � d S �NFz$Failed to unregister the wait handler& r* ) rD rC � _overlappedZUnregisterWaitr, �winerror�ERROR_IO_PENDINGr r- r. rM �r rF r/ r0 r r r �_unregister_wait s"
z&_BaseWaitHandleFuture._unregister_waitc r2 r3 )rS r r+ r4 r r r r+ � r5 z _BaseWaitHandleFuture.cancelc � | � � t� �|� d S r3 )rS r r6 r7 r r r r6 � r5 z#_BaseWaitHandleFuture.set_exceptionc rT r3 )rS r r9 r: r r r r9 � r5 z _BaseWaitHandleFuture.set_result)
r< r= r> r? r rI r" rM rS r+ r6 r9 r@ r r r r rA Z s
rA 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 �eventrF r r r r r � s z_WaitCancelFuture.__init__c C s t d��d S )Nz'_WaitCancelFuture must not be cancelled)� RuntimeErrorr4 r r r r+ � � z_WaitCancelFuture.cancelc �$ t � �|� | jd k r | �| � d S r3 )r r9 rV r: r r r r9 � �
z _WaitCancelFuture.set_resultc rZ r3 )r r6 rV r7 r r r r6 � r[ z _WaitCancelFuture.set_exception) r< r= r> r? r r+ r9 r6 r@ r r r r rU � s
rU 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_proactorrO Z CreateEvent�_event�
_event_fut)r r rE rF �proactorr r r r r � s
z_WaitHandleFuture.__init__c sF | j d k r"t�| j � d | _ d | _| j�| j� d | _t� �|� d S r3 ) r^ rH � CloseHandler_ r] � _unregisterr r rM rK r r r rM � s
z%_WaitHandleFuture._unregister_wait_cbc
C s� | j s
d S d| _ | j}d | _yt�|| j� W nZ tk
r� } z<|jtjkrxd|| d�}| jrh| j|d< | j �
|� d S W d d }~X Y nX | j � | j| j
�| _d S rN )rD rC rO ZUnregisterWaitExr^ r, rP rQ r r- r. r] � _wait_cancelrM r_ rR r r r rS � s$
z"_WaitHandleFuture._unregister_wait)r< r= r> r rM rS r@ 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)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 r3 )�closedrH 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 Zfirst�flagsZh�piper r r rl s
z PipeServer._server_pipe_handlec C s
| j d kS r3 )rf r4 r r r ro rY zPipeServer.closedc C sV | j d k r| j �� d | _ | jd k rRx| jD ] }|�� q,W d | _d | _| j�� d S r3 )rk r+ rf ri �closerj �clear)r rt r r r ru s
zPipeServer.closeN)
r< r= r> r? r rn rl ro ru �__del__r r r r rd � s
rd c @ s e Zd ZdZdS )�_WindowsSelectorEventLoopz'Windows version of selector event loop.N)r< r= r> r? r r r r rx + � rx c s< e Zd ZdZd � fdd� Zdd� Zdd� Zd d d
�Z� ZS )
r z2Windows version of proactor event loop using IOCP.Nc s |d krt � }t� �|� d S r3 )r r r )r r` r r r r 2 s zProactorEventLoop.__init__c � s8 | j �|�}|I d H }|� }| j||d|id�}||fS )N�addr��extra)r] � connect_pipe�_make_duplex_pipe_transport)r �protocol_factoryr# �frt �protocol�transr r r �create_pipe_connection7 s
z(ProactorEventLoop.create_pipe_connectionc � s. t � ��d� ����fdd� ����� �gS )Nc
s d }yj| rL| � � }�j�|� ��� r2|�� d S �� }�j||d� id� ��� }|d kr`d S �j�|�} W n� t k
r� } zF|r�|�
� dkr��� d||d�� |�� n�j r�t
jd|dd� W d d }~X Y n2 tjk
r� |r�|�� Y nX | �_| ��� d S ) Nrz r{ r zPipe accept failed)r' r( rt z Accept pipe failed on pipe %rT)Zexc_info)r; ri Zdiscardro ru r~ rn r] � accept_piper, �filenor. Z_debugr Zwarningr �CancelledErrorrk �add_done_callback)r� rt r� r/ �r# �loop_accept_piper r Zserverr r r� B s<
z>ProactorEventLoop.start_serving_pipe.<locals>.loop_accept_piper3 )rd Z call_soon)r r r# r r� r �start_serving_pipe? s (
z$ProactorEventLoop.start_serving_pipec � sl | � � }
t| |||||||f|
|d�| ��} y|
I d H W n, tk
rf | �� | �� I d H � Y nX | S )N)�waiterr| )�
create_future�_WindowsSubprocessTransportZ Exceptionru Z_wait) r r� �args�shell�stdin�stdout�stderr�bufsizer| �kwargsr� Ztranspr r r �_make_subprocess_transportm s
z,ProactorEventLoop._make_subprocess_transportr3 r3 ) r< r= r> r? r r� r� r� r@ r r r r r / s
0r c @ s� e Zd ZdZd7dd�Zdd� Zdd� Zd d
� Zd8d d
�Zdd� Z d9dd�Z
d:dd�Z d;dd�Z dd� Z
dd� Zdd � Zd d � Zd d � Zd<d!d"�Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd=d/d0�Zd1d2� Zd3d4� Zd5d6� Z d S )>r z#Proactor implementation using IOCP.r c C sD d | _ g | _t�tjtd|�| _i | _t� � | _
g | _ t� � | _ d S rG )
r- �_resultsrO �CreateIoCompletionPort�INVALID_HANDLE_VALUErp �_iocp�_cacherg rh rD �
_unregistered�_stopped_serving)r Z concurrencyr r r r � s
zIocpProactor.__init__c C s | j d krtd��d S )NzIocpProactor is closed)r� rX r4 r r r �
_check_closed� �
zIocpProactor._check_closedc C sF dt | j� dt | j� g}| jd kr0|�d� d| jjd�|�f S )Nzoverlapped#=%sz
result#=%sro z<%s %s>z )Zlenr� r� r� rJ r r< Zjoin)r r% r r r �__repr__� s
zIocpProactor.__repr__c C s
|| _ d S r3 )r- )r r r r r �set_loop� rY zIocpProactor.set_loopNc C s | j s| �|� | j }g | _ |S r3 )r� rI )r �timeoutrm r r r �select� s
zIocpProactor.selectc C s | j �� }|�|� |S r3 )r- r� r9 )r �valuerL r r r �_result� �
zIocpProactor._resultr c C �z | � |� t�t�}y4t|tj�r6|�|�� ||� n|�|�� |� W n t k
rb | �
d�S X dd� }| � |||�S )N� c
S �P y|� � S tk
rJ } z$|jtjtjfkr8t|j� �n� W d d }~X Y nX d S r3 �� getresultr, rP rO ZERROR_NETNAME_DELETEDZERROR_OPERATION_ABORTEDZConnectionResetErrorr� �r� �keyr r/ r r r � finish_recv� �
z&IocpProactor.recv.<locals>.finish_recv) �_register_with_iocprO �
Overlappedrp �
isinstance�socketZWSARecvr� ZReadFile�BrokenPipeErrorr� � _register)r �connZnbytesrs r r� r r r �recv� �
zIocpProactor.recvc C r� )Nr� c
S r� r3 r� r� r r r r� � r� z+IocpProactor.recv_into.<locals>.finish_recv) r� rO r� rp r� r� Z WSARecvIntor� Z ReadFileIntor� r� r� )r r� �bufrs r r� r r r � recv_into� r� zIocpProactor.recv_intoc C sZ | � |� t�t�}t|tj�r4|�|�� ||� n|�|�� |� dd� }| � |||�S )Nc
S r� r3 r� r� r r r � finish_send� r� z&IocpProactor.send.<locals>.finish_send)
r� rO r� rp r� r� ZWSASendr� Z WriteFiler� )r r� r� rs 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_SOCKETrO ZSO_UPDATE_ACCEPT_CONTEXT�
settimeoutZ
gettimeoutZ getpeername)r� r� r r� �r� �listenerr r �
finish_accept� s
z*IocpProactor.accept.<locals>.finish_acceptc � s4 y| I d H W n t jk
r. |�� � Y nX d S r3 )r r� ru )r) r� r r r � accept_coro� s
z(IocpProactor.accept.<locals>.accept_coror ) r� �_get_accept_socket�familyrO r� rp ZAcceptExr� r� r Z
ensure_futurer- )r r� r r� r� r) Zcoror r� r �accept� s
zIocpProactor.acceptc
s� | � � � yt�� �� � j� W nB tk
rb } z$|jtjkr@� � � � d dkrR� W d d }~X Y nX t�
t �}|� � �� |� � fdd�}| �
|� |�S )Nr r c s |� � � �tjtjd� � S rG )r� r� r� r� rO ZSO_UPDATE_CONNECT_CONTEXT�r� r� r �r� r r �finish_connect s
z,IocpProactor.connect.<locals>.finish_connect)r� rO Z BindLocalr� r� r, rP �errnoZ WSAEINVALZ getsocknamer� rp Z ConnectExr� )r r� r# �er r� r r� r �connect s
zIocpProactor.connectc C sb | � |� t�t�}|d@ }|d? d@ }|�|�� t�|�� �|||dd� dd� }| �|||�S )Nr i r c
S r� r3 r� r� r r r �finish_sendfile% r� z.IocpProactor.sendfile.<locals>.finish_sendfile) r� rO r� rp Z TransmitFiler� �msvcrtZ
get_osfhandler� ) r ZsockZfileZoffsetZcountr Z
offset_lowZ offset_highr� r r r �sendfile s
zIocpProactor.sendfilec sJ | � � � t�t�}|�� �� �}|r0| �� �S � fdd�}| �|� |�S )Nc s |� � � S r3 )r� r� �rt r r �finish_accept_pipe; r5 z4IocpProactor.accept_pipe.<locals>.finish_accept_pipe)r� rO r� rp ZConnectNamedPiper� r� r� )r rt r Z connectedr� r r� r r� 0 s
zIocpProactor.accept_pipec
� sz t }xjyt�|�}P W n0 tk
rF } z|jtjkr6� W d d }~X Y nX t|d t�}tj || j
d�I d H qW t � |�S )Ni r )
�CONNECT_PIPE_INIT_DELAYrO Z ConnectPiper, rP ZERROR_PIPE_BUSYZmin�CONNECT_PIPE_MAX_DELAYr Zsleepr- r rq )r r# ZdelayrE r/ r r r r} A 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 rE r� r r r �wait_for_handleT s z IocpProactor.wait_for_handlec C s | � |d d�}||_|S re )r� rV )r rW Z
done_callbackrL r r r rc \ s zIocpProactor._wait_cancelc s� | � � |d krtj}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 r3 )rI r� �r� r r �finish_wait_for_handley s z=IocpProactor._wait_for_handle.<locals>.finish_wait_for_handler )r� rH �INFINITE�math�ceilrO r� rp ZRegisterWaitWithQueuer� r# rU r- r\ r r� )r rE r� Z
_is_cancel�msr rF r� r r� r r� c s
z IocpProactor._wait_for_handlec C s0 || j kr,| j �|� t�|�� | jdd� d S rG )rD rr rO r� r� r� �r �objr r r r� � s
z IocpProactor._register_with_iocpc
C s� | � � t|| jd�}|jr$|jd= |jsry|d d |�}W n, tk
rf } z|�|� W d d }~X Y n X |�|� ||||f| j|j < |S r )
r� r r- r r r, r6 r9 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� rJ )r r r r r rb � s zIocpProactor._unregisterc C s t � |�}|�d� |S rG )r� r� )r r� Zsr r r r� � r� z IocpProactor._get_accept_socketc C s� |d krt }n0|dk r td��n t�|d �}|t kr>td���xt�| j|�}|d krZP d}|\}}}}y| j�|�\}} }
} W nV t k
r� | j
� � r�| j
� dd||||f d�� |dtj
fkr�t�|� wBY nX |
| jkr�|�� qB|�� sBy| ||| �} W n: tk
�r@ }
z|�|
� | j�|� W d d }
~
X Y qBX |�| � | j�|� qBW x | jD ]} | j�| jd � �qdW | j�� d S )Nr znegative timeoutr� ztimeout too bigz8GetQueuedCompletionStatus() returned an unexpected eventz)err=%s transferred=%s key=%#x address=%#x)r' �status)r� Z
ValueErrorr� r� rO ZGetQueuedCompletionStatusr� r� ZpopZKeyErrorr- Z get_debugr. r� rH ra r� r+ Zdoner, r6 r� rJ r9 r� r# rv )r r� r� r� ZerrZ transferredr� r# r� r r� r� r� r� r r r rI � sJ
zIocpProactor._pollc C s | j �|� d S r3 )r� rr r� r r r �
_stop_serving� s zIocpProactor._stop_servingc C s | j d krd S x�t| j�� �D ]�\}\}}}}|�� r8q t|t�rDq y |�� W q tk
r� } z6| j d k r�d||d�}|j
r�|j
|d< | j � |� W d d }~X Y q X q W d}t �
� } | | }
xD| j�r|
t �
� kr�t�d| t �
� | � t �
� | }
| �|� q�W g | _t�| j � d | _ d S )NzCancelling a future failedr& r* g �?z,%r is running after closing for %.1f seconds)r� Zlistr� ZitemsZ cancelledr� rU r+ r, r- r r. �timeZ monotonicr ZdebugrI r� rH ra ) r r# rL r r� r� r/ r0 Z
msg_updateZ
start_timeZnext_msgr r r ru � s:
"
zIocpProactor.closec C s | � � d S r3 )ru r4 r r r rw * rY zIocpProactor.__del__)r r3 �r r� r� r3 r3 ) r< r= r> r? r r� r� r� r� r� r� r� r� r� r� r� r� r} r� rc r� r� r� rb r� rI r� ru rw r r r r r s4
"
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 r3 )�_procZpollZ_process_exited)r� Z
returncoder4 r r r� 5 r� z4_WindowsSubprocessTransport._start.<locals>.callback) r ZPopenr� r- r] r� ZintrB r� )
r r� r� r� r� r� r� r� r� r� r r4 r �_start0 s z"_WindowsSubprocessTransport._startN)r< r= r> r� r r r r r� . s r� c @ � e Zd ZeZdS )r N)r< r= r> r
�
_loop_factoryr r r r r @ ry r c @ r� )r N)r< r= r> r r� r r r r r D ry r ).r? rO rH r� r� r� r� r� r� rg Z r r r r r r r Zlogr Z__all__rp r� ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr� r� ZFuturer rA rU r\ Zobjectrd ZBaseSelectorEventLooprx ZBaseProactorEventLoopr r ZBaseSubprocessTransportr� r
ZBaseDefaultEventLoopPolicyr r r
r r r r Z<module> sP 0J4;P 2