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)loggerSelectorEventLoopProactorEventLoop IocpProactorDefaultEventLoopPolicyl i i gMbP?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)_OverlappedFuturezSubclass 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)selfovr ) __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)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 Wni t k
r } zI i d d 6| d 6| d 6} | j rm | j | d )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 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 %rexc_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..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 argsshellstdinstdoutstderrbufsizerh 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
CreateIoCompletionPortINVALID_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_DELETEDConnectionResetErrorr~ )ro keyr r+ r r r finish_recv s z&IocpProactor.recv..finish_recv) _register_with_iocpr
Overlappedr[
isinstancesocketZWSARecvrt 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~ )ro r r r+ r r r finish_send s z&IocpProactor.send..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..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..accept_coror )
r _get_accept_socketfamilyr
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..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..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_BUSYminCONNECT_PIPE_MAX_DELAYr Zsleepr) r r\ )r r Zdelayr9 r+ r r r rj s zIocpProactor.connect_pipec C s | j | | d S)zWait 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..finish_wait_for_handler r )r; INFINITEmathceilr
r r[ ZRegisterWaitWithQueuer r rC r) rH r r )r r9 r Z
_is_cancelmsr 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 xt 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 }
~
XqX| j | | j j | qa qa Wx' | j D] } | j j | j d qW| j j d S) Nr znegative timeoutg @@ztimeout too bigz8GetQueuedCompletionStatus() returned an unexpected eventr# z)err=%s transferred=%s key=%#x address=%#xstatus)r
ValueErrorr r r
ZGetQueuedCompletionStatusr r popKeyErrorr) 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 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..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 sJ 0J4;] j