B
ӭG4+d @ 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
jZ G dd de
jZ 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)SelectorEventLoopProactorEventLoop IocpProactorDefaultEventLoopPolicy WindowsSelectorEventLoopPolicy WindowsProactorEventLoopPolicyl i i gMbP?g?c s^ e Zd ZdZdd fdd
Z fddZdd Z fd
d Z fd d
Z fddZ Z
S )_OverlappedFuturezSubclass 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 )N)r )super__init___source_traceback_ov)selfovr ) __class__ ;/opt/alt/python37/lib64/python3.7/asyncio/windows_events.pyr 0 s z_OverlappedFuture.__init__c sH t }| jd k rD| jjr dnd}|dd| d| jjdd |S )NpendingZ completedr z overlapped=)r
_repr_infor r insertaddress)r infostate)r 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 exceptionfuturesource_traceback)r cancelOSErrorr _loopcall_exception_handler)r exccontextr r r _cancel_overlapped= s
z$_OverlappedFuture._cancel_overlappedc s | t S )N)r. r r( )r )r r r r( M s z_OverlappedFuture.cancelc s t | | d S )N)r
set_exceptionr. )r r% )r r r r/ Q s z _OverlappedFuture.set_exceptionc s t | d | _d S )N)r
set_resultr )r result)r r r r0 U s z _OverlappedFuture.set_result) __name__
__module__ __qualname____doc__r r r. r( r/ r0
__classcell__r r )r r r * s r c sn e Zd ZdZdd fdd
Zdd Z fdd Zd
d Zd d
Z fddZ fddZ
fddZ Z S )_BaseWaitHandleFuturez2Subclass of Future which represents a wait handle.N)r c s8 t j|d | jr | jd= || _|| _|| _d| _d S )N)r 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 | jdt jkS )Nr )_winapiZWaitForSingleObjectr8 Z
WAIT_OBJECT_0)r r r r _pollk s z_BaseWaitHandleFuture._pollc sd t }|d| jd | jd k rB| r4dnd}|| | jd k r`|d| jd |S )Nzhandle=z#xZsignaledZwaitingz wait_handle=)r r appendr8 r> r9 )r r" r# )r r r r p s
z _BaseWaitHandleFuture._repr_infoc C s
d | _ d S )N)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 handle)r$ r% r& r' ) r: r9 _overlappedZUnregisterWaitr) winerrorERROR_IO_PENDINGr r* r+ rA )r r< r, r- r r r _unregister_wait s"
z&_BaseWaitHandleFuture._unregister_waitc s | t S )N)rE r r( )r )r r r r( s z _BaseWaitHandleFuture.cancelc s | t | d S )N)rE r r/ )r r% )r r r r/ s z#_BaseWaitHandleFuture.set_exceptionc s | t | d S )N)rE r r0 )r r1 )r r r r0 s z _BaseWaitHandleFuture.set_result)
r2 r3 r4 r5 r r> r rA rE r( r/ r0 r6 r r )r r r7 Z s
r7 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.
N)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 dd S )Nz'_WaitCancelFuture must not be cancelled) RuntimeError)r r r r r( s z_WaitCancelFuture.cancelc s$ t | | jd k r | | d S )N)r r0 rG )r r1 )r r r r0 s
z _WaitCancelFuture.set_resultc s$ t | | jd k r | | d S )N)r r/ rG )r r% )r r r r/ s
z _WaitCancelFuture.set_exception) r2 r3 r4 r5 r r( r0 r/ r6 r r )r r rF s
rF c s6 e Zd Zdd fdd
Z fddZdd Z ZS ) _WaitHandleFutureN)r c s< t j||||d || _d| _td ddd | _d | _d S )N)r TF)r r _proactorZ_unregister_proactorrB Z CreateEvent_event
_event_fut)r r r; r< 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 )N) rL r= CloseHandlerM rK _unregisterr r rA )r r@ )r r r rA 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 )NFz$Failed to unregister the wait handle)r$ r% r& r' )r: r9 rB ZUnregisterWaitExrL r) rC rD r r* r+ rK _wait_cancelrA rM )r r< r, r- r r r rE s$
z"_WaitHandleFuture._unregister_wait)r2 r3 r4 r rA rE r6 r r )r r rJ s rJ 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)_addressweakrefWeakSet_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)rW rY )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 )N)closedr= ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPiperS ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr ZBUFSIZEZNMPWAIT_WAIT_FOREVERNULL
PipeHandlerV add)r firstflagshpiper r r rY s
z PipeServer._server_pipe_handlec C s
| j d kS )N)rS )r r r r r\ s zPipeServer.closedc C sV | j d k r| j d | _ | jd k rRx| jD ] }| q,W d | _d | _| j d S )N)rX r( rS rV closerW clear)r rc r r r rd s
zPipeServer.closeN)
r2 r3 r4 r5 r r[ rY r\ rd __del__r r r r rR s
rR c @ s e Zd ZdZdS )_WindowsSelectorEventLoopz'Windows version of selector event loop.N)r2 r3 r4 r5 r r r r rg + s rg 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 )N)r
r r )r rN )r r r r 2 s zProactorEventLoop.__init__c s8 | j |}|I d H }| }| j||d|id}||fS )Naddr)extra)rK connect_pipe_make_duplex_pipe_transport)r protocol_factoryr! frc protocoltransr 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 | nj rt
jd|dd W d d }~X Y n2 tjk
r |r| Y nX | _| d S ) Nrh )ri r zPipe accept failed)r$ r% rc z Accept pipe failed on pipe %rT)exc_info)r1 rV discardr\ rd rk r[ rK accept_piper) filenor+ Z_debugr
Zwarningr CancelledErrorrX add_done_callback)rm rc rn r, )r! loop_accept_piperl r serverr r rw B s<
z>ProactorEventLoop.start_serving_pipe..loop_accept_pipe)N)rR Z call_soon)r rl r! r )r! rw rl r rx 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)waiterri )
create_future_WindowsSubprocessTransport Exceptionrd Z_wait) r rn argsshellstdinstdoutstderrbufsizeri kwargsrz Ztranspr r r _make_subprocess_transportm s
z,ProactorEventLoop._make_subprocess_transport)N)N) r2 r3 r4 r5 r rp ry r r6 r r )r r r / s
0r c @ s e Zd ZdZd7ddZdd Zdd Zd d
Zd8d d
Zdd Z d9ddZ
d:ddZ d;ddZ dd Z
dd Zdd Zd d Zd d Zdr
z#Proactor implementation using IOCP. c C sD d | _ g | _ttjtd|| _i | _t | _
g | _ t | _ d S )Nr )
r* _resultsrB CreateIoCompletionPortINVALID_HANDLE_VALUEr] _iocp_cacherT rU r:
_unregistered_stopped_serving)r Z concurrencyr r r r s
zIocpProactor.__init__c C s | j d krtdd S )NzIocpProactor is closed)r rI )r r r r
_check_closed s
zIocpProactor._check_closedc C sF dt | j dt | j g}| jd kr0|d d| jjd|f S )Nzoverlapped#=%sz
result#=%sr\ z<%s %s> )lenr r r r? r r2 join)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 }g | _ |S )N)r r> )r timeoutrZ r r r select s
zIocpProactor.selectc C s | j }|| |S )N)r* r{ r0 )r valuer@ r r r _result s
zIocpProactor._resultr c C sz | | tt}y4t|tjr6|| || n|| | W n t k
rb |
dS X dd }| |||S )N c
S sP y| S tk
rJ } z$|jtjtjfkr8t|j n W d d }~X Y nX d S )N) getresultr) rC rB ERROR_NETNAME_DELETEDERROR_OPERATION_ABORTEDConnectionResetErrorr~ )ro keyr r, r r r finish_recv s
z&IocpProactor.recv..finish_recv) _register_with_iocprB
Overlappedr]
isinstancesocketZWSARecvrt ZReadFileBrokenPipeErrorr _register)r connnbytesra r r r r r recv s
zIocpProactor.recvc C sz | | tt}y4t|tjr6|| || n|| | W n t k
rb |
dS X dd }| |||S )Nr c
S sP y| S tk
rJ } z$|jtjtjfkr8t|j n W d d }~X Y nX d S )N)r r) rC rB r r r r~ )ro r r r, r r r r s
z+IocpProactor.recv_into..finish_recv) r rB r r] r r Z WSARecvIntort Z ReadFileIntor r r )r r bufra r r r r r recv_into s
zIocpProactor.recv_intoc C sZ | | tt}t|tjr4|| || n|| | dd }| |||S )Nc
S sP y| S tk
rJ } z$|jtjtjfkr8t|j n W d d }~X Y nX d S )N)r r) rC rB r r r r~ )ro r r r, r r r finish_send s
z&IocpProactor.send..finish_send)
r rB r r] r r ZWSASendrt Z WriteFiler )r r r ra r r r r r send s
zIocpProactor.sendc sv | | j tt}| fdd}dd }| ||}|| }t j
|| j d |S )Nc sD | td } tjtj|
fS )Nz@P) r structZpackrt
setsockoptr
SOL_SOCKETrB ZSO_UPDATE_ACCEPT_CONTEXT
settimeoutZ
gettimeoutZ getpeername)ro r r r )r listenerr r
finish_accept s
z*IocpProactor.accept..finish_acceptc s4 y| I d H W n t jk
r. | Y nX d S )N)r ru rd )r& r r r r accept_coro s
z(IocpProactor.accept..accept_coro)r ) r _get_accept_socketfamilyrB r r] ZAcceptExrt r r Z
ensure_futurer* )r r r r r r& coror )r 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 )Nr )r r r r rB ZSO_UPDATE_CONNECT_CONTEXT)ro r r )r r r finish_connect s
z,IocpProactor.connect..finish_connect)r rB Z BindLocalrt r r) rC errnoZ WSAEINVALZ getsocknamer r] Z ConnectExr )r r r! er r r )r r connect s
zIocpProactor.connectc C sb | | tt}|d@ }|d? d@ }|| t| |||dd dd }| |||S )Nl r c
S sP y| S tk
rJ } z$|jtjtjfkr8t|j n W d d }~X Y nX d S )N)r r) rC rB r r r r~ )ro r r r, r r r finish_sendfile% s
z.IocpProactor.sendfile..finish_sendfile) r rB r r] Z TransmitFilert msvcrtZ
get_osfhandler ) r Zsockfileoffsetcountr Z
offset_lowZ offset_highr r r r sendfile s
zIocpProactor.sendfilec sJ | tt}| }|r0| S fdd}| | |S )Nc s | S )N)r )ro r r )rc r r finish_accept_pipe; s z4IocpProactor.accept_pipe..finish_accept_pipe)r rB r r] ZConnectNamedPipert r r )r rc r Z connectedr r )rc r rs 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
dI d H qW t |S )N )r )
CONNECT_PIPE_INIT_DELAYrB Z ConnectPiper) rC ZERROR_PIPE_BUSYminCONNECT_PIPE_MAX_DELAYr Zsleepr* r r^ )r r! Zdelayr; r, r r r rj A s
zIocpProactor.connect_pipec C s | ||dS )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 r; r r r r wait_for_handleT s z IocpProactor.wait_for_handlec C s | |d d}||_|S )NT)r rG )r rH Z
done_callbackr@ r r r rQ \ s zIocpProactor._wait_cancelc s | |d krtj}nt|d }tt}t|| j |j
|}|r\t |||| j d nt
|||| | j d jr~ jd= fdd} |d|f| j|j
< S )Ng @@)r r c s S )N)r> )ro r r )rm r r finish_wait_for_handley s z=IocpProactor._wait_for_handle..finish_wait_for_handler )r r= INFINITEmathceilrB r r] ZRegisterWaitWithQueuer r! rF r* rJ r r )r r; r Z
_is_cancelmsr r< r r )rm r r c s
z IocpProactor._wait_for_handlec C s0 || j kr,| j | t| | jdd d S )Nr )r: r_ rB r rt 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 )N)r r )
r r r* r r r) r/ r0 r r! )r r r callbackrm 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 r? )r r r r r rP s zIocpProactor._unregisterc C s t |}|d |S )Nr )r r )r r sr r r r s
z IocpProactor._get_accept_socketc C s |d krt }n0|dk r tdn t|d }|t kr>tdxt| j|}|d krZP d}|\}}}}y| j|\}} }
} W nV t k
r | j
r| j
dd||||f d |dtj
fkrt| 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 timeoutg @@ztimeout too bigz8GetQueuedCompletionStatus() returned an unexpected eventz)err=%s transferred=%s key=%#x address=%#x)r$ status)r
ValueErrorr r rB ZGetQueuedCompletionStatusr r popKeyErrorr* Z get_debugr+ r r= rO r r( Zdoner) r/ r r? r0 r r! re )r r r r errZ transferredr r! rm r r r r r r r r r> sJ
zIocpProactor._pollc C s | j | d S )N)r r_ )r r r r r
_stop_serving s zIocpProactor._stop_servingc C s | j d krd S xt| j D ]\}\}}}}| r8q t|trDq y | W q tk
r } z6| j d k rd||d}|j
r|j
|d< | j | W d d }~X Y q X q W d}t
} | | }
xD| jr|
t
krtd| t
| t
| }
| | qW g | _t| j d | _ d S )NzCancelling a future failed)r$ r% r& r' g ?z,%r is running after closing for %.1f seconds)r listr itemsZ cancelledr rF r( r) r* r r+ timeZ monotonicr
debugr> r r= rO ) r r! r@ r r r r, r- Z
msg_updateZ
start_timeZnext_msgr r r rd s:
"
zIocpProactor.closec C s | d S )N)rd )r r r r rf * s zIocpProactor.__del__)r )N)r )r )r )N)N) r2 r3 r4 r5 r r r r r r r r r r r r rs rj r rQ r r r rP r r> r rd rf r r r r r
s4
"
7/r
c @ s e Zd Zdd ZdS )r| c
sP t j|f|||||d| _ fdd} jjt jj} | | d S )N)r r r r r c s j } | d S )N)_procZpollZ_process_exited)rm
returncode)r r r r 5 s
z4_WindowsSubprocessTransport._start..callback) r Popenr r* rK r intr8 rv )
r r~ r r r r r r r rm r )r r _start0 s z"_WindowsSubprocessTransport._startN)r2 r3 r4 r r r r r r| . s r| c @ s e Zd ZeZdS )r N)r2 r3 r4 r
_loop_factoryr r r r r @ s r c @ s e Zd ZeZdS )r N)r2 r3 r4 r r r r r r r D s r ).r5 rB r= r r r r r r rT r r r r r r r logr
__all__r] r ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr r ZFuturer r7 rF rJ objectrR ZBaseSelectorEventLooprg ZBaseProactorEventLoopr r
ZBaseSubprocessTransportr| r ZBaseDefaultEventLoopPolicyr r r r r r r sP 0J4;P 2