Your IP : 3.147.86.104
�
j f9l � @ s d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m
Z
d d l m Z d d l m Z d d l m
Z
d d
l m Z d d l m Z d d l m Z d d
l m Z d d d d g Z d Z d Z d Z d Z d Z d Z Gd d � d e
j � Z Gd d � d e
j � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z! Gd! d" � d" e j" � Z# Gd# d � d e j$ � Z% Gd$ d � d � Z& Gd% d& � d& e j' � Z( e# Z) Gd' d( � d( e j* � Z+ e+ Z, d S))z.Selector and proactor event loops for Windows.� 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 s� e Z d Z d Z d d � f d d � Z � f d d � Z d d � Z � f d
d � Z � f d d
� Z � f d d � Z � S)�_OverlappedFuturez�Subclass of Future which represents an overlapped operation.
Cancelling it will immediately cancel the overlapped operation.
�loopNc s6 t � j d | � | j r) | j d =n | | _ d S)Nr r ���)�super�__init__�_source_traceback�_ov)�self�ovr )� __class__� �;/opt/alt/python34/lib64/python3.4/asyncio/windows_events.pyr - s
z_OverlappedFuture.__init__c s] t � j � } | j d k rY | j j r0 d n d } | j d d | | j j f � n | 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 s� | j d k r d Sy | j j � Wni t k
r� } zI i d d 6| d 6| d 6} | j rm | j | d <n | j j | � WYd d } ~ Xn Xd | _ 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__�__doc__r r r- r' r. r/ r r )r r r ' s r c s� e Z d Z d Z d d � f d d � Z d d � Z � f d d � Z d
d � Z d d
� Z � f d d � Z � f d d � Z
� f d d � Z � S)�_BaseWaitHandleFuturez2Subclass of Future which represents a wait handle.r Nc sQ t � j d | � | j r) | j d =n | | _ | | _ | | _ d | _ d S)Nr 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 | j d � t j k S)Nr )�_winapiZWaitForSingleObjectr6 Z
WAIT_OBJECT_0)r r r r �_pollh s z_BaseWaitHandleFuture._pollc s� t � j � } | j d | j � | j d k rZ | j � rD d n d } | j | � n | j d k r� | j d | j � n | 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 Sd | _ | j } d | _ y t j | � Wn t k
r� } z_ | j t j k r� i d d 6| d 6| d 6} | j r� | j | d <n | j j | � d SWYd d } ~ Xn X| j
d � d S)NFz$Failed to unregister the wait handler# 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 r4 r r< r r? rB r' r. r/ r r )r r r5 W s
r5 c sL e Z d Z d Z d d � f d d � Z d d � Z � f d d � Z � S)
�_WaitCancelFuturezoSubclass of Future which represents a wait for the cancellation of a
_WaitHandleFuture using an event.
r Nc s) t � j | | | d | �d | _ d S)Nr )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 s6 t t | � j � | j d k r2 | j | � n d S)N)r rC �_schedule_callbacksrD )r )r r r rG � s z%_WaitCancelFuture._schedule_callbacks)r1 r2 r3 r4 r r' rG r r )r r rC � s rC c sF e Z d Z d d � f d d � Z � f d d � Z d d � Z � S) �_WaitHandleFuturer Nc sV t � j | | | d | �| | _ d | _ t j d d d d � | _ d | _ d S)Nr 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 sd | j d k r4 t j | j � d | _ d | _ n | j j | j � d | _ t � j | � d S)N) rJ r; � CloseHandlerK rI � _unregisterr r r? )r r> )r r r r? � s z%_WaitHandleFuture._unregister_wait_cbc C s� | j s
d Sd | _ | j } d | _ y t j | | j � Wn t k
r� } z_ | j t j k r� i d d 6| d 6| d 6} | j r� | j | d <n | j j
| � d SWYd d } ~ Xn X| j j | j | j
� | _ d S)NFz$Failed to unregister the wait handler# r$ r% r& )r8 r7 r
ZUnregisterWaitExrJ r( r@ rA r r) r* rI � _wait_cancelr? rK )r r: r+ r, r r r rB � s&
z"_WaitHandleFuture._unregister_wait)r1 r2 r3 r r? rB r r )r r rH � s rH c @ sX e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z e Z d S)
�
PipeServerzXClass representing a pipe server.
This is much like a bound, listening socket.
c C s@ | | _ t j � | _ d | _ d | _ | j 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 | j d � } | _ | S)NF)rU rW )r �tmpr r r �_get_unconnected_pipe� s z PipeServer._get_unconnected_pipec C s� | j � r d St j t j B} | r6 | t j O} n t j | j | t j t j Bt j Bt j
t j t j t j
t j � } t j | � } | j j | � | S)N)�closedr; ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPiperQ ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr ZBUFSIZEZNMPWAIT_WAIT_FOREVER�NULL�
PipeHandlerT �add)r �first�flags�h�piper r r rW � s
z PipeServer._server_pipe_handlec C s
| j d k S)N)rQ )r r r r rZ s zPipeServer.closedc C s{ | j d k r( | j j � d | _ n | j d k rw x | j D] } | j � qA Wd | _ d | _ | j j � n d S)N)rV r' rQ rT �closerU �clear)r ra r r r rb s
zPipeServer.closeN)
r1 r2 r3 r4 r rY rW rZ rb �__del__r r r r rP � s
rP c @ s" e Z d Z d Z d d � Z d S)�_WindowsSelectorEventLoopz'Windows version of selector event loop.c C s
t j � S)N)r �
socketpair)r r r r � _socketpair& s z%_WindowsSelectorEventLoop._socketpairN)r1 r2 r3 r4 rg r r r r re # s re c sp e Z d Z d Z d � f d d � Z d d � Z e d d � � Z e d d
� � Z e d d d � � Z � S)
r z2Windows version of proactor event loop using IOCP.Nc s, | d k r t � } n t � j | � d S)N)r r r )r rL )r r r r - s zProactorEventLoop.__init__c C s
t j � S)N)r rf )r r r r rg 2 s z ProactorEventLoop._socketpairc c sO | j j | � } | Dd H} | � } | j | | d i | d 6�} | | f S)N�extra�addr)rI � connect_pipe�_make_duplex_pipe_transport)r �protocol_factoryr �fra �protocol�transr r r �create_pipe_connection5 s z(ProactorEventLoop.create_pipe_connectionc sA t � � � d � � � � � f d d � � � j � � � g S)Nc so d } y� | rn | j � } � j j | � � j � rE | j � d S� � } � j | | d i � d 6�n � j � } | d k r� d S� j j | � } Wn� t k
r-} zn | r� | j
� d k r� � j i d d 6| d 6| d 6� | j � n" � j rt
j d | d d
�n WYd d } ~ Xn> t j k
rT| rP| j � n Yn X| � _ | j � � d S) Nrh ri r zPipe accept failedr# r$ ra z Accept pipe failed on pipe %r�exc_infoTr )r0 rT �discardrZ rb rk rY rI � accept_piper( �filenor* Z_debugr Zwarningr �CancelledErrorrV �add_done_callback)rm ra rn r+ )r �loop_accept_piperl r �serverr r rw B s<
" z>ProactorEventLoop.start_serving_pipe.<locals>.loop_accept_pipe)rP Z call_soon)r rl r r )r rw rl r rx r �start_serving_pipe> s !(
z$ProactorEventLoop.start_serving_pipec k s� t j d | � }
t | | | | | | | | d |
d | | �} y
|
Dd HWn+ t k
r| } z | }
WYd d } ~ Xn Xd }
|
d k r� | j � | j � Dd H|
� n | S)Nr �waiterrh )r �Future�_WindowsSubprocessTransport� Exceptionrb Z_wait)r rn �args�shell�stdin�stdout�stderr�bufsizerh �kwargsrz Ztranspr+ �errr r r �_make_subprocess_transportm s
z,ProactorEventLoop._make_subprocess_transport)
r1 r2 r3 r4 r rg r rp ry r� r r )r r r * s /c @ s6 e Z d Z d Z d d d � Z d d � Z d d � Z d d
d � Z d d
� Z d d d � Z d d d � Z
d d � Z d d � Z d d � Z
e d d � � Z d d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d S)1r z#Proactor implementation using IOCP.l �� c C sd d | _ g | _ t j t j t d | � | _ i | _ t j � | _
g | _ t j � | _ d S)Nr )
r) �_resultsr
�CreateIoCompletionPort�INVALID_HANDLE_VALUEr[ �_iocp�_cacherR rS r8 �
_unregistered�_stopped_serving)r Z concurrencyr r r r � s zIocpProactor.__init__c C s) d | j j t | 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 | � n | j } g | _ | S)N)r� r< )r �timeoutrX r r r �select� s
zIocpProactor.selectc C s& t j d | j � } | j | � | S)Nr )r r{ r) r/ )r �valuer> r r r �_result� s
zIocpProactor._resultr c C s� | j | � t j t � } yH t | t j � rM | j | j � | | � n | j | j � | � Wn t k
r� | j
d � SYn Xd d � } | j | | | � S)N� c S sa y | j � SWnL t k
r\ } z, | j t j k rG t | j � � n � WYd d } ~ Xn Xd S)N)� getresultr( r@ r
�ERROR_NETNAME_DELETED�ConnectionResetErrorr~ )ro �keyr r+ r r r � finish_recv� s z&IocpProactor.recv.<locals>.finish_recv) �_register_with_iocpr
�
Overlappedr[ �
isinstance�socketZWSARecvrt ZReadFile�BrokenPipeErrorr� � _register)r �conn�nbytesr_ r r� r r r �recv� s
zIocpProactor.recvc C s | j | � t j t � } t | t j � rJ | j | j � | | � n | j | j � | � d d � } | j | | | � S)Nc S sa y | j � SWnL t k
r\ } z, | j t j k rG t | j � � n � WYd d } ~ Xn Xd S)N)r� r( r@ r
r� r� r~ )ro r� r r+ r r r � finish_send� s z&IocpProactor.send.<locals>.finish_send)
r� r
r� r[ r� r� ZWSASendrt Z WriteFiler� )r r� �bufr_ r r� r r r �send� s
zIocpProactor.sendc s� | j � � | j � j � � t j t � } | j � j � � j � � � � f d d � } t d d � � } | j | � | � } | | � � } t
j | d | j �| S)Nc s^ | j � t j d � j � � } � j t j t j | � � j � j
� � � � j � f S)Nz@P) r� �structZpackrt �
setsockoptr� �
SOL_SOCKETr
ZSO_UPDATE_ACCEPT_CONTEXT�
settimeoutZ
gettimeoutZ getpeername)ro r� r r� )r� �listenerr r �
finish_accept� s
z*IocpProactor.accept.<locals>.finish_acceptc s s6 y
| Dd HWn" t j k
r1 | j � � Yn Xd S)N)r ru rb )r% r� r r r � accept_coro� s
z(IocpProactor.accept.<locals>.accept_coror )
r� �_get_accept_socket�familyr
r� r[ ZAcceptExrt r r� r Z
ensure_futurer) )r r� r r� r� r% Zcoror )r� r� r �accept� s
zIocpProactor.acceptc s� | j � � y t j � j � � j � WnY t k
r� } z9 | j t j k rW � n � j � d d k rs � n WYd d } ~ Xn Xt j
t � } | j � j � | � � f d d � } | j
| � | � S)Nr r c s' | j � � j t j t j d � � S)Nr )r� r� r� r� r
ZSO_UPDATE_CONNECT_CONTEXT)ro r� r )r� r r �finish_connect� s
z,IocpProactor.connect.<locals>.finish_connect)r� r
Z BindLocalrt r� r( r@ �errnoZ WSAEINVALZ getsocknamer� r[ Z ConnectExr� )r r� r �er r� r )r� r �connect� s
zIocpProactor.connectc si | j � � t j t � } | j � j � � } | rD | j � � S� f d d � } | j | � | � S)Nc s | j � � S)N)r� )ro r� r )ra r r �finish_accept_pipe s
z4IocpProactor.accept_pipe.<locals>.finish_accept_pipe)r� r
r� r[ ZConnectNamedPipert r� r� )r ra r Z connectedr� r )ra r rs s
zIocpProactor.accept_pipec c s� t } x� y t j | � } PWn= t k
r\ } z | j t j k rJ � n WYd d } ~ Xn Xt | d t � } t j | d | j
�Dd Hq Wt 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 rj s zIocpProactor.connect_pipec C s | j | | 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 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 rO 2 s zIocpProactor._wait_cancelc s� | d k r t j } n t j | d � } t j t � } t j | | j | j | � } | r| t
| | | d | j �� n t | | | | d | j �� � j
r� � j
d =n � f d d � } � | d | f | j | j <� S)Ng @�@r r c s
� j � S)N)r< )ro r� r )rm r r �finish_wait_for_handleM s z=IocpProactor._wait_for_handle.<locals>.finish_wait_for_handler r )r; �INFINITE�math�ceilr
r� r[ ZRegisterWaitWithQueuer� r rC r) rH r r� )r r9 r� Z
_is_cancel�msr r: r� r )rm r r� 9 s
z IocpProactor._wait_for_handlec C sE | | j k rA | j j | � t j | j � | j d d � n d S)Nr )r8 r] r
r� rt r� )r �objr r r r� Y s z IocpProactor._register_with_iocpc C s� t | d | j �} | j r+ | j d =n | j s� y | d d | � } Wn2 t k
r~ } z | j | � WYd d } ~ Xq� X| j | � n | | | | f | j | j <| S)Nr r r ) r r) r r r( r. r/ r� r )r r r� �callbackrm r� r� r r r r� c s
zIocpProactor._registerc C s | j j | � d S)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= )r r r r r rN � s zIocpProactor._unregisterc C s t j | � } | j d � | S)Nr )r� r� )r r� �sr r r r� � s
z IocpProactor._get_accept_socketc C s | d k r t } nI | d k r0 t d � � n. t j | d � } | t k r^ t d � � n x�t j | j | � } | d k r� Pn d } | \ } } } } y" | j j | � \ } } }
} Wnz t k
r<| j
j � r| j
j i d d 6d | | | | f d 6� n | d t j
f k r5t j | � n wa Yn X|
| j k rY| j � qa | j � sa y | | | | � } WnB t k
r�}
z" | j |
� | j j | � WYd d }
~
Xq�X| j | � | j j | � qa qa Wx' | j D] } | j j | j d � q�W| j j � d S) Nr znegative timeoutg @�@ztimeout too bigz8GetQueuedCompletionStatus() returned an unexpected eventr# z)err=%s transferred=%s key=%#x address=%#x�status)r� �
ValueErrorr� r� r
ZGetQueuedCompletionStatusr� r� �pop�KeyErrorr) Z get_debugr* r� r; rM r� r' �doner( r. r� r= r/ r� r rc )r r� r� r� r� Z transferredr� r rm 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 sF x� t | j j � � D]� \ } \ } } } } | j � r= q t | t � rO q y | j � Wq t k
r� } z[ | j d k r� i d d 6| d 6| d 6} | j r� | j | d <n | j j
| � n WYd d } ~ Xq Xq Wx, | j r
| j d � s� t j
d � q� q� Wg | _ | j d k rBt j | j � d | _ n d S)NzCancelling a future failedr# 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; rM )r r r> r r� r� r+ r, r r r rb � s. .
* zIocpProactor.closec C s | j � d S)N)rb )r r r r rd � s zIocpProactor.__del__)r1 r2 r3 r4 r r� r� r� r� r� r� r� r� rs r rj r� rO r� r� r� rN r� r< r� rb rd r r r r r � s.
7 c @ s e Z d Z d d � Z d S)r| c
sw t j | d | d | d | d | d | | �� _ � f d d � } � j j j t � j j � � } | j | � d S)Nr r� r� r� r� c s � j j � } � j | � d S)N)�_procZpollZ_process_exited)rm �
returncode)r r r r� � s z4_WindowsSubprocessTransport._start.<locals>.callback) r �Popenr� r) rI r� �intr6 rv )
r r~ r r� r� r� r� r� r� rm r )r r �_start� s !z"_WindowsSubprocessTransport._startN)r1 r2 r3 r� r r r r r| � s r| c @ s e Z d Z e Z d S)� _WindowsDefaultEventLoopPolicyN)r1 r2 r3 r
Z
_loop_factoryr r r r r� s r� )-r4 r; r� r� r� r� rR � r r r r r r r r
Z
coroutinesr �logr �__all__r[ r� ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr� r� r{ r r5 rC rH �objectrP ZBaseSelectorEventLoopre ZBaseProactorEventLoopr r ZBaseSubprocessTransportr| r
ZBaseDefaultEventLoopPolicyr� r r r r r �<module> sJ 0J4;]� j