a
XC?h @ s d dl Z e jdkredd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z ddl m
Z
ddl mZ ddl mZ ddl mZ dd l mZ dd
l mZ dd l mZ dd l mZ dd
lmZ dZd ZdZdZdZdZ dZ G dd dej Z G dd dej Z G dd de Z!G dd de 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'e
j-Z.G d(d) d)e
j-Z/e/Z0dS )* 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 Zdd fdd
Z fddZdd Zd fd d
Z fd d Z fd
dZ Z S )_OverlappedFutureNloopc 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__r r% r6 r0 r; r=
__classcell__r r r r! r 0 s r c sl e Zd Zdd fdd
Zdd Z fddZd d
Zd d Zd fd
d Z fddZ fddZ
Z S )_BaseWaitHandleFutureNr 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 )_winapiZWaitForSingleObjectrE 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% appendrE rM rF 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/ ) rG rF _overlappedZUnregisterWaitr1 winerrorERROR_IO_PENDINGr r2 r3 rQ r rI r4 r5 r r r! _unregister_wait s$
$z&_BaseWaitHandleFuture._unregister_waitc s | t j|dS r7 )rW r r0 r9 r r r! r0 s z _BaseWaitHandleFuture.cancelc s | t | d S r: )rW r r; r< r r r! r; s z#_BaseWaitHandleFuture.set_exceptionc s | t | d S r: )rW r r= r> r r r! r= s z _BaseWaitHandleFuture.set_result)N) r@ rA rB r rM r% rQ rW r0 r; r= rC r r r r! rD ` s
rD c sB e Zd Zdd fdd
Zdd Z fddZ fd d
Z ZS ) _WaitCancelFutureNr c s t j||||d d | _d S )Nr )r r _done_callback)r r eventrI r r r r! r s z_WaitCancelFuture.__init__c C s t dd S )Nz'_WaitCancelFuture must not be cancelled) RuntimeErrorrL r r r! r0 s z_WaitCancelFuture.cancelc s$ t | | jd ur | | d S r: )r r= rY r> r r r! r= s
z _WaitCancelFuture.set_resultc s$ t | | jd ur | | d S r: )r r; rY r< r r r! r; s
z _WaitCancelFuture.set_exception)r@ rA rB r r0 r= r; rC r r r r! rX s rX 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_proactorrS Z CreateEvent_event
_event_fut)r r rH rI 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^ rK CloseHandler_ r] _unregisterr r rQ rO r r r! rQ 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 rR )rG rF rS ZUnregisterWaitExr^ r1 rT rU r r2 r3 r] _wait_cancelrQ r_ rV r r r! rW s(
$
z"_WaitHandleFuture._unregister_wait)r@ rA rB r rQ rW rC r r r r! r\ s r\ c @ s8 e Zd Zdd Zdd Zdd Zdd Zd d
ZeZd S )
PipeServerc 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)rj rl )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: )closedrK ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPiperf ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr
ZBUFSIZEZNMPWAIT_WAIT_FOREVERNULL
PipeHandleri add)r firstflagshpiper r r! rl
s"
z PipeServer._server_pipe_handlec C s
| j d u S r: )rf rL r r r! rp s zPipeServer.closedc C sR | j d ur| j d | _ | jd urN| jD ] }| q*d | _d | _| j d S r: )rk r0 rf ri closerj clear)r rw r r r! rx " s
zPipeServer.closeN) r@ rA rB r ro rl rp rx __del__r r r r! rd s
rd c @ s e Zd ZdS )_WindowsSelectorEventLoopN)r@ rA rB r r r r! r{ 1 s r{ c sD e Zd Zd fdd Z fddZdd Zdd Zd
d
d Z ZS )r
Nc s |d u rt }t | d S r: )r r r )r r` 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] rb 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? ri discardrp rx r ro r] accept_piper1 filenor3 Z_debugr Zwarningr CancelledErrorrk 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)rd 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 r r} r r r rC r r r r! r
5 s 0 r
c @ s e Zd Zd:ddZdd Zdd Zdd Zd;d d Zd
d ZdddZ d?ddZ d@ddZ
dd Zd d Zd d Zd d! Zd"d# ZdAd$d%Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 ZdBd2d3Zd4d5 Zd6d7 Z d8d9 Z d
S )Cr r c C sD d | _ g | _ttjtd|| _i | _t | _
g | _ t | _ d S rJ )
r2 _resultsrS CreateIoCompletionPortINVALID_HANDLE_VALUErq _iocp_cacherg rh rG
_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[ rL 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 rN 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 rM )r timeoutrn r r r! select s
zIocpProactor.selectc C s | j }|| |S r: )r2 r r= )r valuerP 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 rT rS ZERROR_NETNAME_DELETEDZERROR_OPERATION_ABORTEDConnectionResetErrorr r keyr r4 r r r! finish_recv s
z&IocpProactor.recv..finish_recv) _register_with_iocprS
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 rS 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 rS 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 rS 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 rS 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_SOCKETrS 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_socketfamilyrS 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 rJ )r r r r rS ZSO_UPDATE_CONNECT_CONTEXTr r r r r r! finish_connect[ s
z,IocpProactor.connect..finish_connect)typer Z
SOCK_DGRAMrS Z
WSAConnectr r2 r r= r Z BindLocalr r1 rT errnoZ WSAEINVALZ getsocknamer rq Z ConnectExr )r r r' rP 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 rS 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 rS 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_DELAYrS Z ConnectPiper1 rT ZERROR_PIPE_BUSYminCONNECT_PIPE_MAX_DELAYr sleepr
rr )r r' ZdelayrH r4 r r r! r s
zIocpProactor.connect_pipec C s | ||dS rm )_wait_for_handle)r rH r r r r! wait_for_handle s z IocpProactor.wait_for_handlec C s | |d d}||_|S re )r rY )r rZ Z
done_callbackrP r r r! rc 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: )rM r r r r! finish_wait_for_handle s z=IocpProactor._wait_for_handle..finish_wait_for_handler )r rK INFINITEmathceilrS r rq ZRegisterWaitWithQueuer r' rX r2 r\ r r )r rH r Z
_is_cancelmsr rI r r r r! r s$
z IocpProactor._wait_for_handlec C s0 || j vr,| j | t| | jdd d S rJ )rG rs rS 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| d S r: )r r rN r~ r r r! rb s zIocpProactor._unregisterc C s t |}|d |S rJ )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 rS ZGetQueuedCompletionStatusr r popKeyErrorr2 Z get_debugr3 r rK ra r r0 Zdoner1 r; r rN r= r r' ry )r r r r errZ transferredr r' r r r r r r r r r! rM 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 rX r0 r1 r2 r r3 time monotonicr debugrM r rK ra ) r r' rP r r r r4 r5 Z
msg_updateZ
start_timeZnext_msgr r r! rx D s>
$ zIocpProactor.closec C s | d S r: )rx rL r r r! rz s s zIocpProactor.__del__)r )N)r )r )r )r N)r )N)N) r@ rA rB r r r r r r r r r r r r r r r r r rc r r r rb r rM r rx rz r r r r! r s6
"
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
returncoderL r r! r ~ s
z4_WindowsSubprocessTransport._start..callback) r
Popenr r2 r] r intrE r )
r r r r r r r r r r r rL 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 )1sysplatform ImportErrorrS rK r r r r r r rg r r r r r r r r
logr __all__rq r ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr r ZFuturer rD rX r\ objectrd ZBaseSelectorEventLoopr{ ZBaseProactorEventLoopr
r ZBaseSubprocessTransportr r ZBaseDefaultEventLoopPolicyr r r r r r r! sV
0J4;e `