a
XC?h @ s d Z ddlZejdkr edddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ ddl
mZ ddl
mZ dd l
mZ dd
l
mZ dd l
mZ dd l
mZ dd
l
mZ ddlmZ dZdZdZdZdZ dZ dZ G dd dej Z G dd dej Z!G dd de!Z"G dd d e!Z#G d d d e$Z%G d d d ej&Z'G d!d" d"ej(Z)G d#d$ d$Z*G d%d& d&ej+Z,e'Z-G d'd( d(ej.Z/G d)d* d*ej.Z0e0Z1dS )+z.Selector and proactor event loops for Windows. NZwin32z
win32 only )events)base_subprocess)futures)
exceptions)proactor_events)selector_events)tasks)
windows_utils)logger)SelectorEventLoopProactorEventLoop IocpProactorDefaultEventLoopPolicy WindowsSelectorEventLoopPolicy WindowsProactorEventLoopPolicy i i gMbP?g?c s` e Zd ZdZdd fdd
Z fddZdd Zd 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.
Nloopc s& t j|d | jr | jd= || _d S Nr )super__init___source_traceback_ov)selfovr __class__ ;/opt/alt/python39/lib64/python3.9/asyncio/windows_events.pyr 6 s z_OverlappedFuture.__init__c sH t }| jd urD| jjr dnd}|dd| d| jjdd |S )NpendingZ completedr z overlapped=)r
_repr_infor r" insertaddressr infostater r r! r% < s
z _OverlappedFuture._repr_infoc
C st | j d u rd S z| j W nL tyh } z4d|| d}| jrH| j|d< | j| W Y d }~n
d }~0 0 d | _ d S )Nz&Cancelling an overlapped future failedmessage exceptionfuturesource_traceback)r cancelOSErrorr _loopcall_exception_handler)r exccontextr r r! _cancel_overlappedC s
"z$_OverlappedFuture._cancel_overlappedc s | t j|dS N)msg)r6 r r0 r r8 r r r! r0 S s z_OverlappedFuture.cancelc s t | | d S N)r
set_exceptionr6 r r- r r r! r; W s z _OverlappedFuture.set_exceptionc s t | d | _d S r: )r
set_resultr r resultr r r! r= [ s z _OverlappedFuture.set_result)N) __name__
__module__ __qualname____doc__r r% r6 r0 r; r=
__classcell__r r r r! r 0 s r c sp e Zd ZdZdd fdd
Zdd Z fdd Zd
d Zd d
Zd fdd Z fddZ
fddZ Z S )_BaseWaitHandleFuturez2Subclass of Future which represents a wait handle.Nr c s8 t j|d | jr | jd= || _|| _|| _d| _d S )Nr r T)r r r r _handle _wait_handle _registered)r r handle wait_handler r r r! r c s z _BaseWaitHandleFuture.__init__c C s t | jdt jkS Nr )_winapiZWaitForSingleObjectrF Z
WAIT_OBJECT_0r r r r! _pollq s z_BaseWaitHandleFuture._pollc sd t }|d| jd | jd urB| r4dnd}|| | jd ur`|d| jd |S )Nzhandle=r# ZsignaledZwaitingz wait_handle=)r r% appendrF rN rG r( r r r! r% v s
z _BaseWaitHandleFuture._repr_infoc C s
d | _ d S r: )r r futr r r! _unregister_wait_cb s z)_BaseWaitHandleFuture._unregister_wait_cbc
C s | j s
d S d| _ | j}d | _zt| W nf ty } zN|jtjkr|d|| d}| jrb| j|d< | j | W Y d }~d S W Y d }~n
d }~0 0 |
d d S NFz$Failed to unregister the wait handler+ r/ ) rH rG _overlappedZUnregisterWaitr1 winerrorERROR_IO_PENDINGr r2 r3 rR r rJ r4 r5 r r r! _unregister_wait s$
$z&_BaseWaitHandleFuture._unregister_waitc s | t j|dS r7 )rX r r0 r9 r r r! r0 s z _BaseWaitHandleFuture.cancelc s | t | d S r: )rX r r; r< r r r! r; s z#_BaseWaitHandleFuture.set_exceptionc s | t | d S r: )rX r r= r> r r r! r= s z _BaseWaitHandleFuture.set_result)N)
r@ rA rB rC r rN r% rR rX r0 r; r= rD r r r r! rE ` s
rE 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.
Nr c s t j||||d d | _d S )Nr )r r _done_callback)r r eventrJ r r r r! r s z_WaitCancelFuture.__init__c C s t dd S )Nz'_WaitCancelFuture must not be cancelled) RuntimeErrorrM r r r! r0 s z_WaitCancelFuture.cancelc s$ t | | jd ur | | d S r: )r r= rZ r> r r r! r= s
z _WaitCancelFuture.set_resultc s$ t | | jd ur | | d S r: )r r; rZ r< r r r! r; s
z _WaitCancelFuture.set_exception) r@ rA rB rC r r0 r= r; rD r r r r! rY s
rY c s6 e Zd Zdd fdd
Z fddZdd Z ZS ) _WaitHandleFutureNr c s< t j||||d || _d| _td ddd | _d | _d S )Nr TF)r r _proactorZ_unregister_proactorrT Z CreateEvent_event
_event_fut)r r rI rJ proactorr r r r! r s
z_WaitHandleFuture.__init__c sF | j d ur"t| j d | _ d | _| j| j d | _t | d S r: ) r_ rL CloseHandler` r^ _unregisterr r rR rP r r r! rR s
z%_WaitHandleFuture._unregister_wait_cbc
C s | j s
d S d| _ | j}d | _zt|| j W nf ty } zN|jtjkrd|| d}| jrf| j|d< | j
| W Y d }~d S W Y d }~n
d }~0 0 | j | j| j
| _d S rS )rH rG rT ZUnregisterWaitExr_ r1 rU rV r r2 r3 r^ _wait_cancelrR r` rW r r r! rX s(
$
z"_WaitHandleFuture._unregister_wait)r@ rA rB r rR rX rD r r r r! r] s r] 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)rk rm )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 r: )closedrL ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPiperg ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr
ZBUFSIZEZNMPWAIT_WAIT_FOREVERNULL
PipeHandlerj add)r firstflagshpiper r r! rm
s"
z PipeServer._server_pipe_handlec C s
| j d u S r: )rg rM r r r! rp s zPipeServer.closedc C sR | j d ur| j d | _ | jd urN| jD ] }| q*d | _d | _| j d S r: )rl r0 rg rj closerk clear)r rw r r r! rx " s
zPipeServer.closeN)
r@ rA rB rC r ro rm rp rx __del__r r r r! re s
re c @ s e Zd ZdZdS )_WindowsSelectorEventLoopz'Windows version of selector event loop.N)r@ rA rB rC r r r r! r{ 1 s r{ c sH e Zd ZdZd
fdd Z fddZdd Zd d
Zdd d Z Z S )r
z2Windows version of proactor event loop using IOCP.Nc s |d u rt }t | d S r: )r r r )r ra r r r! r 8 s zProactorEventLoop.__init__c s zV| | j t W | jd ur| jj}| j |d urP|jsP| j | d | _n>| jd ur| jj}| j |d ur|js| j | d | _0 d S r: )
call_soonZ_loop_self_readingr run_foreverZ_self_reading_futurer r0 r" r^ rc r r r r r! r} = s
z ProactorEventLoop.run_foreverc s8 | j |}|I d H }| }| j||d|id}||fS )Naddrextra)r^ connect_pipe_make_duplex_pipe_transport)r protocol_factoryr' frw protocoltransr r r! create_pipe_connectionP s
z(ProactorEventLoop.create_pipe_connectionc s. t d fdd gS )Nc
s d }zn| rN| }j| r4| W d S }j||d id }|d u rdW d S j|} W n t y } zN|r|
dkr d||d | nj rt
jd|dd W Y d }~n8d }~0 tjy |r| Y n0 | _| d S ) Nr r r zPipe accept failed)r, r- rw z Accept pipe failed on pipe %rT)exc_info)r? rj discardrp rx r ro r^ accept_piper1 filenor3 Z_debugr Zwarningr CancelledErrorrl add_done_callback)r rw r r4 r' loop_accept_piper r Zserverr r! r [ sB
z>ProactorEventLoop.start_serving_pipe..loop_accept_pipe)N)re r| )r r r' r r r! start_serving_pipeX s (
z$ProactorEventLoop.start_serving_pipec s | }
t| |||||||f|
|d| } z|
I d H W n@ ttfyR Y n* tyz | | I d H Y n0 | S )N)waiterr )
create_future_WindowsSubprocessTransport
SystemExitKeyboardInterrupt
BaseExceptionrx Z_wait) r r argsshellstdinstdoutstderrbufsizer kwargsr Ztranspr r r! _make_subprocess_transport s"
z,ProactorEventLoop._make_subprocess_transport)N)N)
r@ rA rB rC r r} r r r rD r r r r! r
5 s 0 r
c @ s e Zd ZdZd;ddZdd Zdd Zd d
ZdddZ d?ddZ d@ddZ
dAddZdd Zd d Zd d Zd!d" Zd#d$ ZdBd%d&Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 ZdCd3d4Zd5d6 Z d7d8 Z d9d: Z d S )Dr z#Proactor implementation using IOCP.r c C sD d | _ g | _ttjtd|| _i | _t | _
g | _ t | _ d S rK )
r2 _resultsrT CreateIoCompletionPortINVALID_HANDLE_VALUErq _iocp_cacherh ri rH
_unregistered_stopped_serving)r Z concurrencyr r r! r s
zIocpProactor.__init__c C s | j d u rtdd S )NzIocpProactor is closed)r r\ rM r r r!
_check_closed s
zIocpProactor._check_closedc C sF dt | j dt | j g}| jd u r0|d d| jjd|f S )Nzoverlapped#=%sz
result#=%srp z<%s %s> )lenr r r rO r r@ join)r r) r r r! __repr__ s
zIocpProactor.__repr__c C s
|| _ d S r: )r2 )r r r r r! set_loop s zIocpProactor.set_loopNc C s | j s| | | j }g | _ |S r: )r rN )r timeoutrn r r r! select s
zIocpProactor.selectc C s | j }|| |S r: )r2 r r= )r valuerQ r r r! _result s
zIocpProactor._resultr c C s| | | tt}z4t|tjr6|| || n|| | W n t yd |
d Y S 0 dd }| |||S )N c
S sT z
| W S tyN } z,|jtjtjfv r8t|j n W Y d }~n
d }~0 0 d S r: getresultr1 rU rT ZERROR_NETNAME_DELETEDZERROR_OPERATION_ABORTEDConnectionResetErrorr r keyr r4 r r r! finish_recv s
z&IocpProactor.recv..finish_recv) _register_with_iocprT
Overlappedrq
isinstancesocketZWSARecvr ZReadFileBrokenPipeErrorr _registerr connnbytesru r r r r r! recv s
zIocpProactor.recvc C s| | | tt}z4t|tjr6|| || n|| | W n t yd |
d Y S 0 dd }| |||S )Nr c
S sT z
| W S tyN } z,|jtjtjfv r8t|j n W Y d }~n
d }~0 0 d S r: r r r r r! r s
z+IocpProactor.recv_into..finish_recv) r rT r rq r r Z WSARecvIntor Z ReadFileIntor r r )r r bufru r r r r r! recv_into s
zIocpProactor.recv_intoc C s^ | | tt}z|| || W n tyF | d Y S 0 dd }| |||S )N)r Nc
S sT z
| W S tyN } z,|jtjtjfv r8t|j n W Y d }~n
d }~0 0 d S r: r r r r r! r s
z*IocpProactor.recvfrom..finish_recv) r rT r rq Z WSARecvFromr r r r r r r r! recvfrom s
zIocpProactor.recvfromc C s> | | tt}|| ||| dd }| |||S )Nc
S sT z
| W S tyN } z,|jtjtjfv r8t|j n W Y d }~n
d }~0 0 d S r: r r r r r! finish_send s
z(IocpProactor.sendto..finish_send)r rT r rq Z WSASendTor r )r r r ru r r r r r r! sendto s
zIocpProactor.sendtoc C sZ | | tt}t|tjr4|| || n|| | dd }| |||S )Nc
S sT z
| W S tyN } z,|jtjtjfv r8t|j n W Y d }~n
d }~0 0 d S r: r r r r r! r s
z&IocpProactor.send..finish_send)
r rT r rq r r ZWSASendr Z WriteFiler )r r r ru 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 structZpackr
setsockoptr
SOL_SOCKETrT ZSO_UPDATE_ACCEPT_CONTEXT
settimeoutZ
gettimeoutZ getpeername)r r r r r listenerr r!
finish_accept/ s z*IocpProactor.accept..finish_acceptc s2 z| I d H W n t jy, | Y n0 d S r: )r r rx )r. r r r r! accept_coro8 s
z(IocpProactor.accept..accept_coror ) r _get_accept_socketfamilyrT r rq ZAcceptExr r r Z
ensure_futurer2 )r r r r r r. coror r r! accept) s
zIocpProactor.acceptc
s j tjkr4t | | j }|d |S | zt
j W nD t y } z,|j
tjkrr d dkr W Y d }~n
d }~0 0 tt}| | fdd}| | |S )Nr r c s | tjtjd S rK )r r r r rT ZSO_UPDATE_CONNECT_CONTEXTr r r r r r! finish_connect[ s
z,IocpProactor.connect..finish_connect)typer Z
SOCK_DGRAMrT Z
WSAConnectr r2 r r= r Z BindLocalr r1 rU errnoZ WSAEINVALZ getsocknamer rq Z ConnectExr )r r r' rQ er r r r r! connectE s"
zIocpProactor.connectc C sb | | tt}|d@ }|d? d@ }|| t| |||dd dd }| |||S )Nr r c
S sT z
| W S tyN } z,|jtjtjfv r8t|j n W Y d }~n
d }~0 0 d S r: r r r r r! finish_sendfilen s
z.IocpProactor.sendfile..finish_sendfile) r rT r rq Z TransmitFiler msvcrtZ
get_osfhandler ) r Zsockfileoffsetcountr Z
offset_lowZ offset_highr r r r! sendfiled s
zIocpProactor.sendfilec sJ | tt}| }|r0| S fdd}| | |S )Nc s | S r: )r r rw r r! finish_accept_pipe s z4IocpProactor.accept_pipe..finish_accept_pipe)r rT r rq ZConnectNamedPiper r r )r rw r Z connectedr r r r! r y s
zIocpProactor.accept_pipec
st t }zt|}W qjW n2 tyH } z|jtjkr4 W Y d }~n
d }~0 0 t|d t}t |I d H qt
|S )N ) CONNECT_PIPE_INIT_DELAYrT Z ConnectPiper1 rU ZERROR_PIPE_BUSYminCONNECT_PIPE_MAX_DELAYr sleepr
rr )r r' ZdelayrI r4 r r r! r 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 rI r r r r! wait_for_handle s z IocpProactor.wait_for_handlec C s | |d d}||_|S rf )r rZ )r r[ Z
done_callbackrQ r r r! rd s zIocpProactor._wait_cancelc s | |d u rtj}nt|d }tt}t|| j |j
|}|r\t |||| j d nt
|||| | j d jr~ jd= fdd} |d|f| j|j
< S )N @@r r c s S r: )rN r r r r! finish_wait_for_handle s z=IocpProactor._wait_for_handle..finish_wait_for_handler )r rL INFINITEmathceilrT r rq ZRegisterWaitWithQueuer r' rY r2 r] r r )r rI r Z
_is_cancelmsr rJ r r r r! r s$
z IocpProactor._wait_for_handlec C s0 || j vr,| j | t| | jdd d S rK )rH rs rT r r r r objr r r! r s
z IocpProactor._register_with_iocpc
C s | t|| jd}|jr$|jd= |jstz|d d |}W n. tyh } z|| W Y d }~nd }~0 0 || ||||f| j|j < |S r )
r r r2 r r" r1 r; r= r r' )r r r callbackr 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 rO r~ r r r! rc s zIocpProactor._unregisterc C s t |}|d |S rK )r r )r r sr r r! r s
z IocpProactor._get_accept_socketc C s |d u rt }n0|dk r tdn t|d }|t kr>tdt| j|}|d u rXqZd}|\}}}}z| j|\}} }
} W nV t y | j
r| j
dd||||f d |dtj
fvrt| Y q>Y n0 |
| jv r| q>| s>z| ||| } W n< ty@ }
z"||
| j| W Y d }
~
q>d }
~
0 0 || | j| q>| jD ]} | j| jd q`| j d S )Nr znegative timeoutr ztimeout too bigz8GetQueuedCompletionStatus() returned an unexpected eventz)err=%s transferred=%s key=%#x address=%#x)r, status)r
ValueErrorr r rT ZGetQueuedCompletionStatusr r popKeyErrorr2 Z get_debugr3 r rL rb r r0 Zdoner1 r; r rO r= r r' ry )r r r r errZ transferredr r' r r r r r r r r r! rN sL
"
zIocpProactor._pollc C s | j | d S r: )r rs r r r r!
_stop_serving> s zIocpProactor._stop_servingc C s | j d u rd S t| j D ]\}\}}}}| r6q t|trBq z | W q ty } z>| j d urd||d}|j
r|j
|d< | j | W Y d }~q d }~0 0 q d}t
} | | }
| jr|
t
krtd| t
| t
| }
| | qg | _t| j d | _ d S )NzCancelling a future failedr+ r/ g ?z,%r is running after closing for %.1f seconds)r listr itemsZ cancelledr rY r0 r1 r2 r r3 time monotonicr debugrN r rL rb ) r r' rQ r r r r4 r5 Z
msg_updateZ
start_timeZnext_msgr r r! rx D s>
$ zIocpProactor.closec C s | d S r: )rx rM r r r! rz s s zIocpProactor.__del__)r )N)r )r )r )r N)r )N)N) r@ rA rB rC r r r r r r r r r r r r r r r r r rd r r r rc r rN r rx rz r r r r! r s8
"
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 r: )_procZpollZ_process_exited)r
returncoderM r r! r ~ s
z4_WindowsSubprocessTransport._start..callback) r
Popenr r2 r^ r intrF r )
r r r r r r r r r r r rM r! _starty s z"_WindowsSubprocessTransport._startN)r@ rA rB r r r r r! r w s r c @ s e Zd ZeZdS )r N)r@ rA rB r
_loop_factoryr r r r! r s r c @ s e Zd ZeZdS )r N)r@ rA rB r
r r r r r! r s r )2rC sysplatform ImportErrorrT rL r r r r r r rh r r r r r r r r
logr __all__rq r ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr r ZFuturer rE rY r] objectre ZBaseSelectorEventLoopr{ ZBaseProactorEventLoopr
r ZBaseSubprocessTransportr r ZBaseDefaultEventLoopPolicyr r r r r r r! sX
0J4;e `