o
0i @ s d Z ddlZejdkredddlZddlZddlZddlZddlZddl 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/python310/lib64/python3.10/asyncio/windows_events.pyr 6 s
z_OverlappedFuture.__init__c sH t }| jd ur"| jjrdnd}|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 n& ty4 } zd|| d}| jr$| j|d< | j| W Y d }~nd }~ww 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 | t j|dS N)msg)r6 r r0 r r9 r r r! r0 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_resultr< ) __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 )_winapiZWaitForSingleObjectrH Z
WAIT_OBJECT_0r r r r! _pollq s z_BaseWaitHandleFuture._pollc sd t }|d| jd | jd ur!| rdnd}|| | jd ur0|d| jd |S )Nzhandle=r# ZsignaledZwaitingz wait_handle=)r r% appendrH rP rI 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 sd S d| _ | j}d | _zt| W n3 tyH } z'|jtjkr>d|| d}| jr1| j|d< | j | W Y d }~d S W Y d }~nd }~ww |
d d S NFz$Failed to unregister the wait handler+ r/ ) rJ rI _overlappedZUnregisterWaitr1 winerrorERROR_IO_PENDINGr r2 r3 rT r rL r4 r5 r r r! _unregister_wait s*
z&_BaseWaitHandleFuture._unregister_waitc r7 r8 )rZ r r0 r: r r r! r0 r; z _BaseWaitHandleFuture.cancelc | t | d S r< )rZ r r= r> r r r! r= z#_BaseWaitHandleFuture.set_exceptionc r[ r< )rZ r r? r@ r r r! r? r\ z _BaseWaitHandleFuture.set_resultr< )
rB rC rD rE r rP r% rT rZ r0 r= r? rF r r r r! rG ` s
rG 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 eventrL r r r r! r s
z_WaitCancelFuture.__init__c C s t d)Nz'_WaitCancelFuture must not be cancelled) RuntimeErrorrO r r r! r0 s z_WaitCancelFuture.cancelc ( t | | jd ur| | d S d S r< )r r? r^ r@ r r r! r?
z _WaitCancelFuture.set_resultc ra r< )r r= r^ r> r r r! r= rb z _WaitCancelFuture.set_exception) rB rC rD rE r r0 r? r= rF r r r r! r] s r] 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_proactorrV Z CreateEvent_event
_event_fut)r r rK rL proactorr r r r! r s
z_WaitHandleFuture.__init__c sF | j d urt| j d | _ d | _| j| j d | _t | d S r< ) re rN CloseHandlerf rd _unregisterr r rT rR r r r! rT s
z%_WaitHandleFuture._unregister_wait_cbc
C s | j sd S d| _ | j}d | _z t|| j W n3 tyJ } z'|jtjkr@d|| d}| jr3| j|d< | j
| W Y d }~d S W Y d }~nd }~ww | j | j| j
| _d S rU )rJ rI rV ZUnregisterWaitExre r1 rW rX r r2 r3 rd _wait_cancelrT rf rY r r r! rZ s.
z"_WaitHandleFuture._unregister_wait)rB rC rD r rT rZ rF r r r r! rc s rc 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)rq rs )r tmpr r r! _get_unconnected_pipe s z PipeServer._get_unconnected_pipec
C sr | rd S tjtjB }|r|tjO }t| j|tjtjB tj B tj
t j t j tj
tj}t |}| j| |S r< )closedrN ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPiperm ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr
ZBUFSIZEZNMPWAIT_WAIT_FOREVERNULL
PipeHandlerp add)r firstflagshpiper r r! rs
s"
z PipeServer._server_pipe_handlec C s
| j d u S r< )rm rO r r r! rv
zPipeServer.closedc C sV | j d ur
| j d | _ | jd ur)| jD ]}| qd | _d | _| j d S d S r< )rr r0 rm rp closerq clear)r r} r r r! r " s
zPipeServer.closeN)
rB rC rD rE r ru rs rv r __del__r r r r! rk s
rk c @ s e Zd ZdZdS )_WindowsSelectorEventLoopz'Windows version of selector event loop.N)rB rC rD rE r r r r! r 1 r c sJ e Zd ZdZd
fdd Z fddZdd Zd d
Z d
d 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 rg r r r! r 8 s zProactorEventLoop.__init__c s z.| | j t W | jd ur-| jj}| j |d ur(|js(| j | d | _d S d S | jd urM| jj}| j |d urJ|jsJ| j | d | _w r< )
call_soonZ_loop_self_readingr run_foreverZ_self_reading_futurer r0 r" rd ri r r r r r! r = s
z ProactorEventLoop.run_foreverc s: | j |}|I d H }| }| j||d|id}||fS )Naddrextra)rd connect_pipe_make_duplex_pipe_transport)r protocol_factoryr' fr} protocoltransr r r! create_pipe_connectionP s
z(ProactorEventLoop.create_pipe_connectionc s0 t d fdd gS )Nc
sT d }z7| r'| }j| r| W d S }j||d id }|d u r2W d S j|} W nf t yT |rL|
dkrL| Y d S t y } z-|rq|
dkrq
d||d | n jr|tjd|dd W Y d }~d S d }~w tjy |r| Y d S Y d S w | _| d S ) Nr r r zPipe accept failed)r, r- r} z Accept pipe failed on pipe %rT)exc_info)rA rp discardrv r r ru rd accept_pipeBrokenPipeErrorfilenor r1 r3 Z_debugr Zwarningr CancelledErrorrr add_done_callback)r r} r r4 r' loop_accept_piper r Zserverr r! r [ sR
z>ProactorEventLoop.start_serving_pipe..loop_accept_piper< )rk r )r r r' r r r! start_serving_pipeX s
-z$ProactorEventLoop.start_serving_pipec sx | }
t| |||||||f|
|d| } z|
I d H W | S ttfy) ty; | | I d H w )N)waiterr )
create_future_WindowsSubprocessTransport
SystemExitKeyboardInterrupt
BaseExceptionr Z_wait) r r argsshellstdinstdoutstderrbufsizer kwargsr Ztranspr r r! _make_subprocess_transport s&
z,ProactorEventLoop._make_subprocess_transportr< )
rB rC rD rE r r r r r rF r r r r! r
5 s 5r
c @ s e Zd ZdZd;ddZdd Zdd Zd d
ZdddZ
d=ddZdd Zd d Zd d Zd!d" Zd#d$ Zd )lenr r r rQ r rB join)r r) r r r! __repr__ s
zIocpProactor.__repr__c C s
|| _ d S r< )r2 )r r r r r! set_loop r~ zIocpProactor.set_loopNc C s. | j s| | | j }g | _ z|W d }S d }w r< )r rP )r timeoutrt r r r! select s
zIocpProactor.selectc C s | j }|| |S r< )r2 r r? )r valuerS r r r! _result
zIocpProactor._resultr c C | | | tt}zt|tjr|| || n|| | W n t y2 |
d Y S w dd }| |||S )N c
S B z| W S ty } z|jtjtjfv rt|j d }~ww r< getresultr1 rW rV ZERROR_NETNAME_DELETEDZERROR_OPERATION_ABORTEDConnectionResetErrorr r keyr r4 r r r! finish_recv
z&IocpProactor.recv..finish_recv) _register_with_iocprV
Overlappedrw
isinstancesocketZWSARecvr ZReadFiler r _registerr connnbytesr{ r r r r r! recv
zIocpProactor.recvc C r )Nr c
S r r< r r r r r! r r z+IocpProactor.recv_into..finish_recv) r rV r rw r r Z WSARecvIntor Z ReadFileIntor r r )r r bufr{ r r r r r! recv_into r zIocpProactor.recv_intoc C s^ | | tt}z || || W n ty# | d Y S w dd }| |||S )N)r Nc
S r r< r r r r r! r r z*IocpProactor.recvfrom..finish_recv) r rV r rw Z WSARecvFromr r r r r r r r! recvfrom s
zIocpProactor.recvfromc C s> | | tt}|| ||| dd }| |||S )Nc
S r r< r r r r r! finish_send r z(IocpProactor.sendto..finish_send)r rV r rw Z WSASendTor r )r r r r{ r r r r r r! sendto s
zIocpProactor.sendtoc C sZ | | tt}t|tjr|| || n|| | dd }| |||S )Nc
S r r< r r r r r! r & r z&IocpProactor.send..finish_send)
r rV r rw r r ZWSASendr Z WriteFiler )r r r r{ 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_SOCKETrV ZSO_UPDATE_ACCEPT_CONTEXT
settimeoutZ
gettimeoutZ getpeername)r r r r r listenerr r!
finish_accept8 s z*IocpProactor.accept..finish_acceptc s. z| I d H W d S t jy | w r< )r r r )r. r r r r! accept_coroA s z(IocpProactor.accept..accept_coror ) r _get_accept_socketfamilyrV r rw ZAcceptExr r r Z
ensure_futurer2 )r r r r r r. coror r r! accept2 s
zIocpProactor.acceptc
s j tjkrt | | j }|d |S | z t
j W n" t yL } z|j
tjkr9 d dkrB W Y d }~nd }~ww tt}| | fdd}| | |S )Nr r c s | tjtjd S rM )r r r r rV ZSO_UPDATE_CONNECT_CONTEXTr r r r r r! finish_connectd s
z,IocpProactor.connect..finish_connect)typer Z
SOCK_DGRAMrV Z
WSAConnectr r2 r r? r Z BindLocalr r1 rW errnoZ WSAEINVALZ getsocknamer rw Z ConnectExr )r r r' rS er r r r r! connectN s(
zIocpProactor.connectc C sb | | tt}|d@ }|d? d@ }|| t| |||dd dd }| |||S )Nr r c
S r r< r r r r r! finish_sendfilew r z.IocpProactor.sendfile..finish_sendfile) r rV r rw Z TransmitFiler msvcrtZ
get_osfhandler ) r Zsockfileoffsetcountr Z
offset_lowZ offset_highr r r r! sendfilem s
zIocpProactor.sendfilec sJ | tt}| }|r| S fdd}| | |S )Nc s | S r< )r r r} r r! finish_accept_pipe s z4IocpProactor.accept_pipe..finish_accept_pipe)r rV r rw ZConnectNamedPiper r r )r r} r Z connectedr r r r! r s
zIocpProactor.accept_pipec
st t } zt|}W n) ty$ } z
|jtjkr W Y d }~nd }~ww t|d t}t |I d H qt
|S )NT ) CONNECT_PIPE_INIT_DELAYrV Z ConnectPiper1 rW ZERROR_PIPE_BUSYminCONNECT_PIPE_MAX_DELAYr sleepr
rx )r r' ZdelayrK 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 rK r r r r! wait_for_handle s z IocpProactor.wait_for_handlec C s | |d d}||_|S rl )r r^ )r r_ Z
done_callbackrS r r r! rj s zIocpProactor._wait_cancelc s | |d u r tj}nt|d }tt}t|| j |j
|}|r.t |||| j d n
t
|||| | j d jr? jd= fdd} |d|f| j|j
< S )N @@r r c s S r< )rP r r r r! finish_wait_for_handle s z=IocpProactor._wait_for_handle..finish_wait_for_handler )r rN INFINITEmathceilrV r rw ZRegisterWaitWithQueuer r' r] r2 rc r r )r rK r Z
_is_cancelmsr rL r r r r! r s$
z IocpProactor._wait_for_handlec C s4 || j vr| j | t| | jdd d S d S rM )rJ ry rV r r r r objr r r! r s
z IocpProactor._register_with_iocpc
C s | t|| jd}|jr|jd= |js:z|d d |}W n ty4 } z || W Y d }~n
d }~ww || ||||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 rQ r r r r! ri s zIocpProactor._unregisterc C s t |}|d |S rM )r r )r r sr r r! r r z IocpProactor._get_accept_socketc C s |d u rt }n|dk rtdt|d }|t kr td t| j|}|d u r+nd}|\}}}}z | j|\}} }
} W n) t yh | j
rZ| j
dd||||f d |dtj
fvrft| Y q w |
| jv rs| n=| sz5z| ||| } W n ty }
z||
| j| W Y d }
~
nd }
~
ww || | j| W d }nd }w q | jD ]
} | j| jd q| j d S ) Nr znegative timeoutr ztimeout too bigTz8GetQueuedCompletionStatus() returned an unexpected eventz)err=%s transferred=%s key=%#x address=%#x)r, status)r
ValueErrorr r rV ZGetQueuedCompletionStatusr r popKeyErrorr2 Z get_debugr3 r rN rh r r0 Zdoner1 r= r rQ r? r r' r )r r r r errZ transferredr r' r r r r r r r r r! rP sZ
)zIocpProactor._pollc C s | j | d S r< )r ry r r r r!
_stop_servingI s zIocpProactor._stop_servingc C s | j d u rd S t| j D ]D\}\}}}}| rqt|tr!qz| W q tyR } z | j d urHd||d}|j
rB|j
|d< | j | W Y d }~qd }~ww d}t
} | | }
| jr|
t
krwtd| t
| t
| }
| | | js`g | _t| j d | _ d S )NzCancelling a future failedr+ r/ g ?z,%r is running after closing for %.1f seconds)r listr itemsZ cancelledr r] r0 r1 r2 r r3 time monotonicr debugrP r rN rh ) r r' rS r r r r4 r5 Z
msg_updateZ
start_timeZnext_msgr r r! r O sD
zIocpProactor.closec C s | d S r< )r rO r r r! r ~ s zIocpProactor.__del__)r r< )r )r N) rB rC rD rE r r r r r r r r r r r r r r r r r rj r r r ri r rP r r r r r r r! r s:
"
9 /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
returncoderO r r! r s
z4_WindowsSubprocessTransport._start..callback) r
Popenr r2 rd r intrH r )
r r r r r r r r r r r rO r! _start s z"_WindowsSubprocessTransport._startN)rB rC rD r r r r r! r s r c @ e Zd ZeZdS )r N)rB rC rD r
_loop_factoryr r r r! r r r c @ r )r N)rB rC rD r
r r r r r! r r r )2rE sysplatform ImportErrorrV rN r r r r r r rn r r r r r r r r
logr __all__rw r ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr r ZFuturer rG r] rc objectrk ZBaseSelectorEventLoopr ZBaseProactorEventLoopr
r ZBaseSubprocessTransportr r ZBaseDefaultEventLoopPolicyr r r r r r r! sZ
0J4;j f