3
\�l � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z ddlm
Z
ddlm Z ddlm Z dd lm
Z
dd
lmZ dd lmZ dd lmZ d
dddgZd ZdZdZdZdZdZG dd� de j�ZG dd� de j�Z G dd� de �Z G d d � d e �Z G d d � d e �Z G d d!� d!e j!�Z"G d"d� de
j#�Z$G d#d� d�Z%G d$d%� d%ej&�Z'e"Z(G d&d'� d'ej)�Z*e*Z+dS )(� N� )�events)�base_subprocess)�futures)�proactor_events)�selector_events)�tasks)�
windows_utils)� _overlapped)� coroutine)�logger�SelectorEventLoop�ProactorEventLoop� IocpProactor�DefaultEventLoopPolicyl �� i� i� g����MbP?g�������?c sZ e Zd Zdd�� fdd�
Z� fdd�Zdd� Z� fd d
�Z� fd d �Z� fd
d�Z� Z S )�_OverlappedFutureN)�loopc s& t � j|d� | jr | jd= || _d S )N)r r �����)�super�__init__�_source_traceback�_ov)�self�ovr )� __class__� �./usr/lib64/python3.6/asyncio/windows_events.pyr - s z_OverlappedFuture.__init__c s@ t � j� }| jd k r<| jjr dnd}|jdd|| jjf � |S )N�pendingZ completedr zoverlapped=<%s, %#x>)r �
_repr_infor r �insert�address)r �info�state)r r r r 3 s
z _OverlappedFuture._repr_infoc C sr | j d krd S y| j j� W nJ tk
rf } z.d|| d�}| jrJ| j|d< | jj|� W Y d d }~X 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 s | j � t� j� S )N)r- r r' )r )r r r r' J s z_OverlappedFuture.cancelc s t � j|� | j� d S )N)r �
set_exceptionr- )r r$ )r r r r. N s z _OverlappedFuture.set_exceptionc s t � j|� d | _d S )N)r �
set_resultr )r �result)r r r r/ R s z _OverlappedFuture.set_result)
�__name__�
__module__� __qualname__r r r- r' r. r/ �
__classcell__r r )r r r ' s r c sj e Zd Zdd�� fdd�
Zdd� Z� fdd�Zd d
� Zd d � Z� fd
d�Z� fdd�Z � fdd�Z
� Z S )�_BaseWaitHandleFutureN)r c s8 t � j|d� | jr | jd= || _|| _|| _d| _d S )N)r r Tr )r r r r �_handle� _wait_handle� _registered)r r �handle� wait_handler )r r r r Z s z _BaseWaitHandleFuture.__init__c C s t j| jd�t jkS )Nr )�_winapiZWaitForSingleObjectr6 Z
WAIT_OBJECT_0)r r r r �_pollh s z_BaseWaitHandleFuture._pollc s\ t � j� }|jd| j � | jd k r>| j� r0dnd}|j|� | jd k rX|jd| j � |S )Nz
handle=%#xZsignaledZwaitingzwait_handle=%#x)r r �appendr6 r< r7 )r r! r" )r r r r m s
z _BaseWaitHandleFuture._repr_infoc C s
d | _ d S )N)r )r �futr r r �_unregister_wait_cbw s z)_BaseWaitHandleFuture._unregister_wait_cbc C s� | j s
d S d| _ | j}d | _ytj|� W nZ tk
r� } z>|jtjkrtd|| d�}| jrd| j|d< | jj |� d S W Y d d }~X nX | j
d � d S )NFz$Failed to unregister the wait handle)r# r$ r% r& ) r8 r7 r
ZUnregisterWaitr( �winerror�ERROR_IO_PENDINGr r) r* r? )r r: r+ r, r r r �_unregister_wait| s"
z&_BaseWaitHandleFuture._unregister_waitc s | j � t� j� S )N)rB r r' )r )r r r r' � s z _BaseWaitHandleFuture.cancelc s | j � t� j|� d S )N)rB r r. )r r$ )r r r r. � s z#_BaseWaitHandleFuture.set_exceptionc s | j � t� j|� d S )N)rB r r/ )r r0 )r r r r/ � s z _BaseWaitHandleFuture.set_result) r1 r2 r3 r r< r r? rB r' r. r/ r4 r r )r r r5 W s
r5 c sB e Zd Zdd�� fdd�
Zdd� Z� fdd�Z� fd d
�Z� ZS ) �_WaitCancelFutureN)r c s t � j||||d� d | _d S )N)r )r r �_done_callback)r r �eventr: r )r r r r � s z_WaitCancelFuture.__init__c C s t d��d S )Nz'_WaitCancelFuture must not be cancelled)� RuntimeError)r r r r r' � s z_WaitCancelFuture.cancelc s$ t � j|� | jd k r | j| � d S )N)r r/ rD )r r0 )r r r r/ � s
z _WaitCancelFuture.set_resultc s$ t � j|� | jd k r | j| � d S )N)r r. rD )r r$ )r r r r. � s
z _WaitCancelFuture.set_exception)r1 r2 r3 r r' r/ r. r4 r r )r r rC � s rC c s6 e Zd Zdd�� fdd�
Z� fdd�Zdd� Z� ZS ) �_WaitHandleFutureN)r c s< t � j||||d� || _d| _tjd ddd �| _d | _d S )N)r TF)r r � _proactorZ_unregister_proactorr
Z CreateEvent�_event�
_event_fut)r r r9 r: �proactorr )r r r r � s
z_WaitHandleFuture.__init__c sF | j d k r"tj| j � d | _ d | _| jj| j� d | _t� j|� d S )N) rI r; � CloseHandlerJ rH � _unregisterr r r? )r r> )r r r r? � s
z%_WaitHandleFuture._unregister_wait_cbc C s� | j s
d S d| _ | j}d | _ytj|| j� W nZ tk
r� } z>|jtjkrxd|| d�}| jrh| j|d< | j j
|� d S W Y d d }~X nX | j j | j| j
�| _d S )NFz$Failed to unregister the wait handle)r# r$ r% r& )r8 r7 r
ZUnregisterWaitExrI r( r@ rA r r) r* rH � _wait_cancelr? rJ )r r: r+ r, r r r rB � s$
z"_WaitHandleFuture._unregister_wait)r1 r2 r3 r r? rB r4 r r )r r rG � s rG c @ s8 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� ZeZd S ) �
PipeServerc C s, || _ tj� | _d | _d | _| jd�| _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 | jd� }| _ |S )NF)rT rV )r �tmpr r r �_get_unconnected_pipe� s z PipeServer._get_unconnected_pipec C sr | j � r d S tjtjB }|r&|tjO }tj| j|tjtjB tj B tj
t j t j tj
tj�}t j|�}| jj|� |S )N)�closedr; ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPiperP ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr ZBUFSIZEZNMPWAIT_WAIT_FOREVER�NULL�
PipeHandlerS �add)r �first�flags�h�piper r r rV s
z PipeServer._server_pipe_handlec C s
| j d kS )N)rP )r r r r rY s zPipeServer.closedc C sV | j d k r| j j� d | _ | jd k rRx| jD ] }|j� q,W d | _d | _| jj� d S )N)rU r' rP rS �closerT �clear)r r` r r r ra s
zPipeServer.closeN) r1 r2 r3 r rX rV rY ra �__del__r r r r rO � s
rO c @ s e Zd Zdd� ZdS )�_WindowsSelectorEventLoopc C s t j� S )N)r �
socketpair)r r r r � _socketpair+ s z%_WindowsSelectorEventLoop._socketpairN)r1 r2 r3 rf r r r r rd ( s rd c sL e Zd Zd � fdd� Zdd� Zedd� �Zedd � �Zed
d
d ��Z� Z S )r Nc s |d krt � }t� j|� d S )N)r r r )r rK )r r r r 2 s zProactorEventLoop.__init__c C s t j� S )N)r re )r r r r rf 7 s z ProactorEventLoop._socketpairc c s8 | j j|�}|E d H }|� }| j||d|id�}||fS )N�addr)�extra)rH � connect_pipe�_make_duplex_pipe_transport)r �protocol_factoryr �fr` �protocol�transr r r �create_pipe_connection: s
z(ProactorEventLoop.create_pipe_connectionc s. t � ��d� ����fdd� ��j�� �gS )Nc s d }yj| rL| j � }�jj|� �j� r2|j� d S �� }�j||d� id� �j� }|d kr`d S �jj|�} W n� t k
r� } zH|r�|j
� d kr��j d||d�� |j� n�j r�t
jd|dd� W Y d d }~X n2 tjk
r� |r�|j� Y nX | �_| j�� d S )
Nrg )rh r zPipe accept failed)r# r$ r` z Accept pipe failed on pipe %rT)�exc_infor )r0 rS �discardrY ra rj rX rH � accept_piper( �filenor* Z_debugr Zwarningr �CancelledErrorrU �add_done_callback)rl r` rm r+ )r �loop_accept_piperk r �serverr r rv G s<
z>ProactorEventLoop.start_serving_pipe..loop_accept_pipe)N)rO Z call_soon)r rk r r )r rv rk r rw r �start_serving_pipeC s (
z$ProactorEventLoop.start_serving_pipec k s� | j � }
t| |||||||f|
|d�| ��} y|
E d H W n& tk
r` } z
| }
W Y d d } ~ X nX d }
|
d k r�| j� | j� E d H |
�| S )N)�waiterrh )�
create_future�_WindowsSubprocessTransport� Exceptionra Z_wait)r rm �args�shell�stdin�stdout�stderr�bufsizerh �kwargsry Ztranspr+ �errr r r �_make_subprocess_transportr s
z,ProactorEventLoop._make_subprocess_transport)N)N)
r1 r2 r3 r rf r ro rx r� r4 r r )r r r / s /c @ s� e Zd Zd0dd�Zdd� Zdd� Zd1d d
�Zd d � Zd2dd�Zd3dd�Z dd� Z
dd� Z dd� Z e
dd� �Zd4dd�Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd5d(d)�Zd*d+� Zd,d-� Zd.d/� ZdS )6r � �� c C sD d | _ g | _tjtjtd|�| _i | _tj � | _
g | _ tj � | _ d S )Nr )
r) �_resultsr
�CreateIoCompletionPort�INVALID_HANDLE_VALUErZ �_iocp�_cacherQ rR r8 �
_unregistered�_stopped_serving)r Z concurrencyr r r r � s
zIocpProactor.__init__c C s d| j jt| j�t| j�f S )Nz <%s overlapped#=%s result#=%s>)r r1 �lenr� r� )r r r r �__repr__� s zIocpProactor.__repr__c C s
|| _ d S )N)r) )r r r r r �set_loop� s zIocpProactor.set_loopNc C s | j s| j|� | j }g | _ |S )N)r� r< )r �timeoutrW r r r �select� s
zIocpProactor.selectc C s | j j� }|j|� |S )N)r) rz r/ )r �valuer> r r r �_result� s
zIocpProactor._resultr c C sz | j |� tjt�}y4t|tj�r6|j|j� ||� n|j|j� |� W n t k
rb | j
d�S X dd� }| j |||�S )N� c S sJ y|j � S tk
rD } z |jtjkr2t|j� �n� W Y d d }~X nX d S )N)� getresultr( r@ r
�ERROR_NETNAME_DELETED�ConnectionResetErrorr} )rn �keyr r+ r r r � finish_recv� s z&IocpProactor.recv..finish_recv) �_register_with_iocpr
�
OverlappedrZ �
isinstance�socketZWSARecvrs ZReadFile�BrokenPipeErrorr� � _register)r �conn�nbytesr^ r r� r r r �recv� s
zIocpProactor.recvc C sZ | j |� tjt�}t|tj�r4|j|j� ||� n|j|j� |� dd� }| j |||�S )Nc S sJ y|j � S tk
rD } z |jtjkr2t|j� �n� W Y d d }~X nX d S )N)r� r( r@ r
r� r� r} )rn r� r r+ r r r � finish_send� s z&IocpProactor.send..finish_send)
r� r
r� rZ r� r� ZWSASendrs Z WriteFiler� )r r� �bufr^ r r� r r r �send� s
zIocpProactor.sendc sz | j �� | j�j�� tjt�}|j�j� � j� � � �fdd�}tdd� �}| j |�|�}||� �}t
j || j d� |S )Nc sD |j � tjd�j� �}� jtjtj|� � j �j
� � � � j � fS )Nz@P) r� �structZpackrs �
setsockoptr� �
SOL_SOCKETr
ZSO_UPDATE_ACCEPT_CONTEXT�
settimeoutZ
gettimeoutZ getpeername)rn r� r r� )r� �listenerr r �
finish_accept� s
z*IocpProactor.accept..finish_acceptc s s4 y| E d H W n t jk
r. |j� � Y nX d S )N)r rt ra )r% r� r r r � accept_coro� s
z(IocpProactor.accept..accept_coro)r )
r� �_get_accept_socket�familyr
r� rZ ZAcceptExrs r r� r Z
ensure_futurer) )r r� r r� r� r% �coror )r� r� r �accept� s
zIocpProactor.acceptc s� | j � � ytj� j� � j� W nB tk
rb } z&|jtjkr@� � j � d dkrR� W Y d d }~X nX tj
t �}|j � j� |� � fdd�}| j
|� |�S )Nr r c s |j � � jtjtjd� � S )Nr )r� r� r� r� r
ZSO_UPDATE_CONNECT_CONTEXT)rn r� r )r� r r �finish_connect s
z,IocpProactor.connect..finish_connect)r� r
Z BindLocalrs r� r( r@ �errnoZ WSAEINVALZ getsocknamer� rZ Z ConnectExr� )r r� r �er r� r )r� r �connect� s
zIocpProactor.connectc sJ | j � � tjt�}|j� j� �}|r0| j� �S � fdd�}| j|� |�S )Nc s |j � � S )N)r� )rn r� r )r` r r �finish_accept_pipe s z4IocpProactor.accept_pipe..finish_accept_pipe)r� r
r� rZ ZConnectNamedPipers r� r� )r r` r Z connectedr� r )r` r rr
s
zIocpProactor.accept_pipec c sz t }xjytj|�}P W n0 tk
rF } z|jtjkr6� W Y d d }~X nX t|d t�}tj || j
d�E d H qW t j |�S )N� )r )
�CONNECT_PIPE_INIT_DELAYr
Z ConnectPiper( r@ ZERROR_PIPE_BUSY�min�CONNECT_PIPE_MAX_DELAYr Zsleepr) r r[ )r r Zdelayr9 r+ r r r ri s
zIocpProactor.connect_pipec C s | j ||d�S )NF)�_wait_for_handle)r r9 r� r r r �wait_for_handle/ s z IocpProactor.wait_for_handlec C s | j |d d�}||_|S )NT)r� rD )r rE Z
done_callbackr> r r r rN 7 s zIocpProactor._wait_cancelc s� |d krt j}ntj|d �}tjt�}tj|| j|j |�}|rTt
|||| j d�� nt |||| | j d�� � j
rv� j
d= � fdd�}� |d|f| j|j < � S )Ng @�@)r r c s � j � S )N)r< )rn r� r )rl r r �finish_wait_for_handleR s z=IocpProactor._wait_for_handle..finish_wait_for_handler r )r; �INFINITE�math�ceilr
r� rZ ZRegisterWaitWithQueuer� r rC r) rG r r� )r r9 r� Z
_is_cancel�msr r: r� r )rl r r� > s
z IocpProactor._wait_for_handlec C s0 || j kr,| j j|� tj|j� | jdd� d S )Nr )r8 r\ r
r� rs r� )r �objr r r r� ^ s
z IocpProactor._register_with_iocpc C s� t || jd�}|jr |jd= |jsjy|d d |�}W n, tk
r^ } z|j|� W Y d d }~X n X |j|� ||||f| j|j< |S )N)r r r ) r r) r r r( r. r/ r� r )r r r� �callbackrl r� r� r r r r� h s
zIocpProactor._registerc C s | j j|� d S )N)r� r= )r r r r r rM � s zIocpProactor._unregisterc C s t j |�}|jd� |S )Nr )r� r� )r r� �sr r r r� � s
z IocpProactor._get_accept_socketc C s� |d krt }n0|dk r td��n tj|d �}|t kr>td���xtj| j|�}|d krZP d}|\}}}}y| jj|�\}} }
} W nV t k
r� | j
j � r�| j
j dd||||f d�� |dtj
fkr�tj|� wBY nX |
| jkr�|j� qB|j� sBy| ||| �} W n: tk
�r@ }
z |j|
� | jj|� W Y d d }
~
X qBX |j| � | jj|� qBW x | jD ]} | jj| jd � �qdW | jj� d S )Nr znegative timeoutg @�@ztimeout too bigz8GetQueuedCompletionStatus() returned an unexpected eventz)err=%s transferred=%s key=%#x address=%#x)r# �status)r� �
ValueErrorr� r� r
ZGetQueuedCompletionStatusr� r� �pop�KeyErrorr) Z get_debugr* r� r; rL r� r' �doner( r. r� r= r/ r� r rb )r r� r� r� r� Z transferredr� r rl r r� r� r� r� r r r r< � sJ
zIocpProactor._pollc C s | j j|� d S )N)r� r\ )r r� r r r �
_stop_serving� s zIocpProactor._stop_servingc C s� x�t | jj� �D ]�\}\}}}}|j� r*qt|t�r6qy |j� W q tk
r� } z8| jd k r�d||d�}|j rz|j |d< | jj
|� W Y d d }~X qX qW x | jr�| j d�s�t j
d� q�W g | _| jd k r�tj| j� d | _d S )NzCancelling a future failed)r# r$ r% r& r z"taking long time to close proactor)�listr� �itemsZ cancelledr� rC r' r( r) r r* r< r �debugr� r� r; rL )r r r> r r� r� r+ r, r r r ra � s,
"
zIocpProactor.closec C s | j � d S )N)ra )r r r r rc � s zIocpProactor.__del__)r� )N)r )r )N)N)r1 r2 r3 r r� r� r� r� r� r� r� r� rr r ri r� rN r� r� r� rM r� r< r� ra rc r r r r r � s,
7 c @ s e Zd Zdd� ZdS )r{ c
sP t j|f|||||d�|��� _� fdd�}� jjjt� jj��} | j|� d S )N)r~ r r� r� r� c s � j j� }� j|� d S )N)�_procZpollZ_process_exited)rl �
returncode)r r r r� � s
z4_WindowsSubprocessTransport._start..callback) r �Popenr� r) rH r� �intr6 ru )
r r} r~ r r� r� r� r� r� rl r )r r �_start� s z"_WindowsSubprocessTransport._startN)r1 r2 r3 r� r r r r r{ � s r{ c @ s e Zd ZeZdS )� _WindowsDefaultEventLoopPolicyN)r1 r2 r3 r
Z
_loop_factoryr r r r r� s r� ),r; r� r� r� r� rQ � r r r r r r r r
Z
coroutinesr �logr �__all__rZ r� ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr� r� ZFuturer r5 rC rG �objectrO ZBaseSelectorEventLooprd ZBaseProactorEventLoopr r ZBaseSubprocessTransportr{ r
ZBaseDefaultEventLoopPolicyr� r r r r r � sJ 0J4;] k