].l @ s 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)( N )events)base_subprocess)futures)proactor_events)selector_events)tasks)
windows_utils) _overlapped) coroutine)loggerSelectorEventLoopProactorEventLoop IocpProactorDefaultEventLoopPolicyl i i gMbP?g?c s| e 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)_OverlappedFutureloopNc s3 t j d | | j r& | j d =| | _ d S)Nr r )super__init___source_traceback_ov)selfovr ) __class__ 3/opt/alt/python35/lib64/python3.5/windows_events.pyr - s
z_OverlappedFuture.__init__c sZ t j } | j d k rV | j j r0 d n d } | j d d | | j j f | S)NpendingZ completedr zoverlapped=<%s, %#x>)r
_repr_infor r insertaddress)r infostate)r r r r 3 s
z _OverlappedFuture._repr_infoc C s | j d k r d Sy | j j Wnc t k
r } zC d d d | d | i } | j rg | j | d <| j j | WYd d } ~ Xn Xd | _ d S)Nmessagez&Cancelling an overlapped future failed exceptionfuturesource_traceback)r cancelOSErrorr _loopcall_exception_handler)r exccontextr 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/ r r )r r r ' s r c s e 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)_BaseWaitHandleFuturer Nc sN t j d | | j r& | j d =| | _ | | _ | | _ 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 )_winapiZWaitForSingleObjectr5 Z
WAIT_OBJECT_0)r r r r _pollh s z_BaseWaitHandleFuture._pollc s~ t j } | j d | j | j d k rW | j rD d n d } | j | | j d k rz | j d | j | S)Nz
handle=%#xsignaledZwaitingzwait_handle=%#x)r r appendr5 r; r6 )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 | Wny t k
r } zY | j t j k r d d d | d | i } | j r | j | d <| j j | d SWYd d } ~ Xn X| j
d d S)NFr# z$Failed to unregister the wait handler$ r% r& ) r7 r6 r
ZUnregisterWaitr( winerrorERROR_IO_PENDINGr r) r* r? )r r9 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/ r r )r r r4 W s
r4 c sF e Z d Z d d f d d Z d d Z f d d Z S) _WaitCancelFuturer Nc s) t j | | | d | d | _ d S)Nr )r r _done_callback)r r eventr9 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 s3 t t | j | j d k r/ | j | d S)N)r rC _schedule_callbacksrD )r )r r r rG s z%_WaitCancelFuture._schedule_callbacks)r1 r2 r3 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 r8 r9 proactorr )r r r r s
z_WaitHandleFuture.__init__c sa | j d k r1 t j | j d | _ d | _ | 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 Wny t k
r } zY | j t j k r d d d | d | i } | j r | j | d <| j j
| d SWYd d } ~ Xn X| j j | j | j
| _ d S)NFr# z$Failed to unregister the wait handler$ r% r& )r7 r6 r
ZUnregisterWaitExrJ r( r@ rA r r) r* rI _wait_cancelr? rK )r r9 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 @ sR e Z d Z d d Z d d Z d d Z d d Z d d
Z e Z d S)
PipeServerc C s@ | | _ t j | _ d | _ d | _ | j d | _ d S)NT)_addressweakrefWeakSet_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} | r3 | t j O} 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 BUFSIZEZNMPWAIT_WAIT_FOREVERNULL
PipeHandlerT add)r firstflagshpiper 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 su | j d k r% | j j d | _ | j d k rq x | j D] } | j q> Wd | _ d | _ | j j d S)N)rV r' rQ rT closerU clear)r rb r r r rc s
zPipeServer.closeN) r1 r2 r3 r rY rW rZ rc __del__r r r r rP s
rP c @ s e Z d Z d d Z d S)_WindowsSelectorEventLoopc C s
t j S)N)r
socketpair)r r r r _socketpair& s z%_WindowsSelectorEventLoop._socketpairN)r1 r2 r3 rh r r r r rf # s rf c sj e 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 Nc s) | d k r t } 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 rg )r r r r rh 2 s z ProactorEventLoop._socketpairc c sN | j j | } | Ed H} | } | j | | d d | i } | | f S)Nextraaddr)rI connect_pipe_make_duplex_pipe_transport)r protocol_factoryr frb protocoltransr r r create_pipe_connection5 s z(ProactorEventLoop.create_pipe_connectionc sA t d f d d j g S)Nc sb d } y | rj | j } j j | j rE | j d S } j | | d d i j } | d k r d S j j | } Wn t k
r#} zh | r | j
d k r j d d d | d | i | j n j rt
j d | d d
WYd d } ~ Xn; t j k
rG| rC| j Yn X| _ | j d S) Nri rj r r# zPipe accept failedr$ rb z Accept pipe failed on pipe %rexc_infoTr )r0 rT discardrZ rc rl rY rI accept_piper( filenor* Z_debugr warningr CancelledErrorrV add_done_callback)rn rb ro r+ )r loop_accept_piperm r serverr r ry B s<
z>ProactorEventLoop.start_serving_pipe..loop_accept_pipe)rP Z call_soon)r rm r r )r ry rm r rz r start_serving_pipe> s !(
z$ProactorEventLoop.start_serving_pipec k s | j }
t | | | | | | | | d |
d | | } y
|
Ed HWn+ t k
rv } z | }
WYd d } ~ Xn Xd }
|
d k r | j | j Ed H|
| S)Nwaiterri )
create_future_WindowsSubprocessTransport Exceptionrc _wait)r ro argsshellstdinstdoutstderrbufsizeri kwargsr| Ztranspr+ errr r r _make_subprocess_transportm s
z,ProactorEventLoop._make_subprocess_transport) r1 r2 r3 r rh r rq r{ r r r )r r r * s /c @ s0 e 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. d/ Z d S)0r l c C sd d | _ g | _ t j t j t d | | _ i | _ t j | _
g | _ t j | _ d S)Nr )
r) _resultsr
CreateIoCompletionPortINVALID_HANDLE_VALUEr\ _iocp_cacherR rS r7
_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 | | j } g | _ | S)N)r r; )r timeoutrX r r r select s
zIocpProactor.selectc C s | j j } | j | | S)N)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_DELETEDConnectionResetErrorr )rp keyr r+ r r r finish_recv s z&IocpProactor.recv..finish_recv) _register_with_iocpr
Overlappedr\
isinstancesocketZWSARecvru ZReadFileBrokenPipeErrorr _register)r connnbytesr` 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 )rp r r r+ r r r finish_send s z&IocpProactor.send..finish_send)
r r
r r\ r r ZWSASendru 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 structpackru
setsockoptr
SOL_SOCKETr
ZSO_UPDATE_ACCEPT_CONTEXT
settimeout
gettimeout getpeername)rp r r r )r listenerr r
finish_accept s
z*IocpProactor.accept..finish_acceptc s s6 y
| Ed HWn" t j k
r1 | j Yn Xd S)N)r rw rc )r% r r r r accept_coro s
z(IocpProactor.accept..accept_coror )
r _get_accept_socketfamilyr
r r\ ZAcceptExru r r r Z
ensure_futurer) )r r r r r r% coror )r r r accept s
zIocpProactor.acceptc s | j y t j j j WnS t k
r } z3 | j t j k rT j d d k rm 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)rp r r )r r r finish_connect s
z,IocpProactor.connect..finish_connect)r r
Z BindLocalru r r( r@ errnoZ WSAEINVAL 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 )rp r r )rb r r finish_accept_pipe s
z4IocpProactor.accept_pipe..finish_accept_pipe)r r
r r\ ZConnectNamedPiperu r r )r rb r Z connectedr r )rb r rt s
zIocpProactor.accept_pipec c s t } x y t j | } PWn: t k
rY } z | j t j k rG WYd d } ~ Xn Xt | d t } t j | d | j
Ed Hq Wt j | S)N r )
CONNECT_PIPE_INIT_DELAYr
Z ConnectPiper( r@ ZERROR_PIPE_BUSYminCONNECT_PIPE_MAX_DELAYr sleepr) r r] )r r delayr8 r+ r r r rk s zIocpProactor.connect_pipec C s | j | | d S)NF)_wait_for_handle)r r8 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 = f d d } | d | f | j | j < S)Ng @@r r c s
j S)N)r; )rp r r )rn r r finish_wait_for_handleM s z=IocpProactor._wait_for_handle..finish_wait_for_handler r )r: INFINITEmathceilr
r r\ ZRegisterWaitWithQueuer r rC r) rH r r )r r8 r Z
_is_cancelmsr r9 r r )rn r r 9 s
z IocpProactor._wait_for_handlec C sB | | j k r> | j j | t j | j | j d d d S)Nr )r7 r^ r
r ru r )r objr r r r Y s z IocpProactor._register_with_iocpc C s t | d | j } | j r( | j d =| j s y | d d | } Wn2 t k
r{ } z | j | WYd d } ~ Xn X| j | | | | | f | j | j <| S)Nr r r ) r r) r r r( r. r/ r r )r r r callbackrn r r r r r r c s
zIocpProactor._registerc C s | j j | d S)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 } nF | d k r0 t d n+ t j | d } | t k r[ t d xut j | j | } | d k r Pd } | \ } } } } y" | j j | \ } } }
} Wnr t k
r.| j
j r| j
j d d d d | | | | f i | d t j
f k r't j | w^ Yn X|
| j k rK| j q^ | j s^ y | | | | } WnB t k
r}
z" | j |
| j j | WYd d }
~
Xq^ X| j | | j j | q^ Wx' | j D] } | j j | j d qW| j j d S) Nr znegative timeoutg @@ztimeout too bigr# z8GetQueuedCompletionStatus() returned an unexpected eventstatusz)err=%s transferred=%s key=%#x address=%#x)r
ValueErrorr r r
ZGetQueuedCompletionStatusr r popKeyErrorr) Z get_debugr* r r: rM r r' doner( r. r r= r/ r r rd )r r r r r Z transferredr r rn 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 s7 x t | j j D] \ } \ } } } } | j r= q t | t rO q y | j Wq t k
r } zR | j d k r d d d | d | i } | j r | j | d <| j j
| WYd d } ~ Xq Xq Wx) | j r| j d s t j
d q Wg | _ | j d k r3t j | j d | _ d S)Nr# zCancelling a future failedr$ r% r& r z"taking long time to close proactor)listr items cancelledr rC r' r( r) r r* r; r debugr r r: rM )r r r> r r r r+ r, r r r rc s, .
' zIocpProactor.closec C s | j d S)N)rc )r r r r re s zIocpProactor.__del__)r1 r2 r3 r r r r r r r r r rt r rk r rO r r r rN r r; r rc re 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)_procpollZ_process_exited)rn
returncode)r r r r s z4_WindowsSubprocessTransport._start..callback) r Popenr r) rI r intr5 rx )
r r r r r r r r r rn 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 ),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 Futurer r4 rC rH objectrP ZBaseSelectorEventLooprf ZBaseProactorEventLoopr r ZBaseSubprocessTransportr~ r
ZBaseDefaultEventLoopPolicyr r r r r r sH 0J4;] j