U
e5di @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZddl m
Z
ddl m Z ddl m Z ddl m
Z
ddl mZ ddl mZ dd l mZ dd
l mZ dd lmZ d Zd Zd
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 )( N )events)base_subprocess)futures)
exceptions)proactor_events)selector_events)tasks)
windows_utils)logger)SelectorEventLoopProactorEventLoop IocpProactorDefaultEventLoopPolicy WindowsSelectorEventLoopPolicy WindowsProactorEventLoopPolicy i i gMbP?g?c sZ e Zd Zdd fdd
Z fddZdd Z 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__ ./usr/lib64/python3.8/asyncio/windows_events.pyr 1 s z_OverlappedFuture.__init__c sH t }| jd k rD| jjr dnd}|dd| d| jjdd |S )NpendingZ completedr z overlapped=)r
_repr_infor r" insertaddressr infostater r r! r% 7 s
z _OverlappedFuture._repr_infoc
C sr | j d krd S z| j W nJ tk
rf } z,d|| d}| jrJ| j|d< | j| W 5 d }~X Y nX d | _ d S )Nz&Cancelling an overlapped future failedmessage exceptionfuturesource_traceback)r cancelOSErrorr _loopcall_exception_handler)r exccontextr r r! _cancel_overlapped> s
z$_OverlappedFuture._cancel_overlappedc s | t S N)r6 r r0 r r r r! r0 N s z_OverlappedFuture.cancelc s t | | d S r7 )r
set_exceptionr6 r r- r r r! r9 R s z _OverlappedFuture.set_exceptionc s t | d | _d S r7 )r
set_resultr r resultr r r! r; V s z _OverlappedFuture.set_result)
__name__
__module__ __qualname__r r% r6 r0 r9 r;
__classcell__r r r r! r + s r c sj e Zd Zdd fdd
Zdd Z fddZd d
Zd d Z fd
dZ 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 ^ s z _BaseWaitHandleFuture.__init__c C s t | jdt jkS Nr )_winapiZWaitForSingleObjectrC Z
WAIT_OBJECT_0r8 r r r! _polll s z_BaseWaitHandleFuture._pollc sd t }|d| jd | jd k rB| r4dnd}|| | jd k r`|d| jd |S )Nzhandle=r# ZsignaledZwaitingz wait_handle=)r r% appendrC rJ rD r( r r r! r% q s
z _BaseWaitHandleFuture._repr_infoc C s
d | _ d S r7 )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 n` tk
r } zB|jtjkrzd|| d}| jrd| j|d< | j | W Y d S W 5 d }~X Y nX |
d d S NFz$Failed to unregister the wait handler+ r/ ) rE rD _overlappedZUnregisterWaitr1 winerrorERROR_IO_PENDINGr r2 r3 rN r rG r4 r5 r r r! _unregister_wait s$
z&_BaseWaitHandleFuture._unregister_waitc s | t S r7 )rT r r0 r8 r r r! r0 s z _BaseWaitHandleFuture.cancelc s | t | d S r7 )rT r r9 r: r r r! r9 s z#_BaseWaitHandleFuture.set_exceptionc s | t | d S r7 )rT r r; r< r r r! r; s z _BaseWaitHandleFuture.set_result) r> r? r@ r rJ r% rN rT r0 r9 r; rA r r r r! rB [ s
rB 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 eventrG r r r r! r s z_WaitCancelFuture.__init__c C s t dd S )Nz'_WaitCancelFuture must not be cancelled) RuntimeErrorr8 r r r! r0 s z_WaitCancelFuture.cancelc s$ t | | jd k r | | d S r7 )r r; rV r< r r r! r; s
z _WaitCancelFuture.set_resultc s$ t | | jd k r | | d S r7 )r r9 rV r: r r r! r9 s
z _WaitCancelFuture.set_exception)r> r? r@ r r0 r; r9 rA r r r r! rU s rU 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_proactorrP Z CreateEvent_event
_event_fut)r r rF rG 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 r7 ) r[ rI CloseHandler\ rZ _unregisterr r rN rL r r r! rN s
z%_WaitHandleFuture._unregister_wait_cbc
C s | j s
d S d| _ | j}d | _zt|| j W n` tk
r } zB|jtjkr~d|| d}| jrh| j|d< | j
| W Y d S W 5 d }~X Y nX | j | j| j
| _d S rO )rE rD rP ZUnregisterWaitExr[ r1 rQ rR r r2 r3 rZ _wait_cancelrN r\ rS r r r! rT s(
z"_WaitHandleFuture._unregister_wait)r> r? r@ r rN rT rA r r r r! rY s rY 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)rg ri )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 r7 )closedrI ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPiperc ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr
ZBUFSIZEZNMPWAIT_WAIT_FOREVERNULL
PipeHandlerf add)r firstflagshpiper r r! ri s(
z PipeServer._server_pipe_handlec C s
| j d kS r7 )rc r8 r r r! rm s zPipeServer.closedc C sR | j d k r| j d | _ | jd k rN| jD ] }| q*d | _d | _| j d S r7 )rh r0 rc rf closerg clear)r rt r r r! ru s
zPipeServer.closeN) r> r? r@ r rl ri rm ru __del__r r r r! ra s
ra c @ s e Zd ZdS )_WindowsSelectorEventLoopN)r> r? r@ r r r r! rx , s rx c sD e Zd Zd fdd Z fddZdd Zdd Zd
d
d Z ZS )r
Nc s |d krt }t | d S r7 )r r r )r r] r r r! r 3 s zProactorEventLoop.__init__c sX z| | j t W 5 | j d k rR| j j}| j |d k rL| j| d | _ X d S r7 ) Z_self_reading_futurer r0 rZ r_ call_soonZ_loop_self_readingr run_foreverr r r r r! rz 8 s
z ProactorEventLoop.run_foreverc s8 | j |}|I d H }| }| j||d|id}||fS )Naddrextra)rZ connect_pipe_make_duplex_pipe_transport)r protocol_factoryr' frt protocoltransr r r! create_pipe_connectionK 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 krdW d S j|} W n t k
r } zF|r|
dkr d||d | nj rt
jd|dd W 5 d }~X Y n2 tjk
r |r| Y nX | _| d S ) Nr| r} r zPipe accept failed)r, r- rt z Accept pipe failed on pipe %rT)exc_info)r= rf discardrm ru r rl rZ accept_piper1 filenor3 Z_debugr Zwarningr CancelledErrorrh add_done_callback)r rt r r4 r' loop_accept_piper r Zserverr r! r V sH
z>ProactorEventLoop.start_serving_pipe..loop_accept_pipe)N)ra ry )r r r' r r r! start_serving_pipeS s (
z$ProactorEventLoop.start_serving_pipec s | }
t| |||||||f|
|d| } z|
I d H W nD ttfk
rT Y n, tk
r~ | | I d H Y nX | S )N)waiterr~ )
create_future_WindowsSubprocessTransport
SystemExitKeyboardInterrupt
BaseExceptionru Z_wait) r r argsshellstdinstdoutstderrbufsizer~ kwargsr Ztranspr r r! _make_subprocess_transport s*
z,ProactorEventLoop._make_subprocess_transport)N)N) r> r? r@ r rz r r r rA r r r r! r
0 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 rH )
r2 _resultsrP CreateIoCompletionPortINVALID_HANDLE_VALUErn _iocp_cacherd re rE
_unregistered_stopped_serving)r Z concurrencyr r r! r s
zIocpProactor.__init__c C s | j d krtdd S )NzIocpProactor is closed)r rX r8 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#=%srm z<%s %s> )lenr r r rK r r> join)r r) r r r! __repr__ s
zIocpProactor.__repr__c C s
|| _ d S r7 )r2 )r r r r r! set_loop s zIocpProactor.set_loopNc C s | j s| | | j }g | _ |S r7 )r rJ )r timeoutrk r r r! select s
zIocpProactor.selectc C s | j }|| |S r7 )r2 r r; )r valuerM r r r! _result s
zIocpProactor._resultr c C s~ | | tt}z4t|tjr6|| || n|| | W n t k
rf |
d Y S X dd }| |||S )N c
S sR z
| W S tk
rL } z$|jtjtjfkr:t|j n W 5 d }~X Y nX d S r7 getresultr1 rQ rP ZERROR_NETNAME_DELETEDZERROR_OPERATION_ABORTEDConnectionResetErrorr r keyr r4 r r r! finish_recv s
z&IocpProactor.recv..finish_recv) _register_with_iocprP
Overlappedrn
isinstancesocketZWSARecvr ZReadFileBrokenPipeErrorr _registerr connnbytesrr r r r r r! recv s
zIocpProactor.recvc C s~ | | tt}z4t|tjr6|| || n|| | W n t k
rf |
d Y S X dd }| |||S )Nr c
S sR z
| W S tk
rL } z$|jtjtjfkr:t|j n W 5 d }~X Y nX d S r7 r r r r r! r s
z+IocpProactor.recv_into..finish_recv) r rP r rn r r Z WSARecvIntor Z ReadFileIntor r r )r r bufrr r r r r r! recv_into s
zIocpProactor.recv_intoc C s` | | tt}z|| || W n tk
rH | d Y S X dd }| |||S )N)r Nc
S sR z
| W S tk
rL } z$|jtjtjfkr:t|j n W 5 d }~X Y nX d S r7 r r r r r! r s
z*IocpProactor.recvfrom..finish_recv) r rP r rn Z WSARecvFromr r r r r r r r! recvfrom s
zIocpProactor.recvfromc C s> | | tt}|| ||| dd }| |||S )Nc
S sR z
| W S tk
rL } z$|jtjtjfkr:t|j n W 5 d }~X Y nX d S r7 r r r r r! finish_send s
z(IocpProactor.sendto..finish_send)r rP r rn Z WSASendTor r )r r r rr r| r r r r r! sendto s
zIocpProactor.sendtoc C sZ | | tt}t|tjr4|| || n|| | dd }| |||S )Nc
S sR z
| W S tk
rL } z$|jtjtjfkr:t|j n W 5 d }~X Y nX d S r7 r r r r r! r s
z&IocpProactor.send..finish_send)
r rP r rn r r ZWSASendr Z WriteFiler )r r r rr 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_SOCKETrP ZSO_UPDATE_ACCEPT_CONTEXT
settimeoutZ
gettimeoutZ getpeername)r r r r r listenerr r!
finish_accept* s z*IocpProactor.accept..finish_acceptc s4 z| I d H W n t jk
r. | Y nX d S r7 )r r ru )r. r r r r! accept_coro3 s
z(IocpProactor.accept..accept_coror ) r _get_accept_socketfamilyrP r rn 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 nB t k
r } z$|j
tjkrt d dkr W 5 d }~X Y nX tt}| | fdd}| | |S )Nr r c s | tjtjd S rH )r r r r rP ZSO_UPDATE_CONNECT_CONTEXTr r r r r r! finish_connectV s z,IocpProactor.connect..finish_connect)typer Z
SOCK_DGRAMrP Z
WSAConnectr r2 r r; r Z BindLocalr r1 rQ errnoZ WSAEINVALZ getsocknamer rn Z ConnectExr )r r r' rM er r r r r! connect@ s"
zIocpProactor.connectc C sb | | tt}|d@ }|d? d@ }|| t| |||dd dd }| |||S )Nr r c
S sR z
| W S tk
rL } z$|jtjtjfkr:t|j n W 5 d }~X Y nX d S r7 r r r r r! finish_sendfilei s
z.IocpProactor.sendfile..finish_sendfile) r rP r rn Z TransmitFiler 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 r7 )r r rt r r! finish_accept_pipe s z4IocpProactor.accept_pipe..finish_accept_pipe)r rP r rn ZConnectNamedPiper r r )r rt r Z connectedr r r r! r t s
zIocpProactor.accept_pipec
sr t }zt|}W qhW n0 tk
rF } z|jtjkr6 W 5 d }~X Y nX t|d t}t |I d H qt
|S )N ) CONNECT_PIPE_INIT_DELAYrP Z ConnectPiper1 rQ ZERROR_PIPE_BUSYminCONNECT_PIPE_MAX_DELAYr sleepr
ro )r r' ZdelayrF r4 r r r! r s
zIocpProactor.connect_pipec C s | ||dS rj )_wait_for_handle)r rF r r r r! wait_for_handle s z IocpProactor.wait_for_handlec C s | |d d}||_|S rb )r rV )r rW Z
done_callbackrM r r r! r` 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 )N @@r r c s S r7 )rJ r r r r! finish_wait_for_handle s z=IocpProactor._wait_for_handle..finish_wait_for_handler )r rI INFINITEmathceilrP r rn ZRegisterWaitWithQueuer r' rU r2 rY r r )r rF r Z
_is_cancelmsr rG r r r r! r s*
z IocpProactor._wait_for_handlec C s0 || j kr,| j | t| | jdd d S rH )rE rp rP r r r r objr r r! r s
z IocpProactor._register_with_iocpc
C s | t|| jd}|jr$|jd= |jsrz|d d |}W n, tk
rf } z|| W 5 d }~X Y n X || ||||f| j|j < |S r )
r r r2 r r" r1 r9 r; r r' )r r r callbackr r r r r r! r s
zIocpProactor._registerc C s | | j| d S r7 )r r rK r{ r r r! r_ s zIocpProactor._unregisterc C s t |}|d |S rH )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>tdt| j|}|d krXqZd}|\}}}}z| j|\}} }
} W nX t k
r | j
r| j
dd||||f d |dtj
fkrt| Y q>Y nX |
| jkr| q>| s>z| ||| } W n: tk
r@ }
z||
| j| W 5 d }
~
X Y q>X || | 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 rP ZGetQueuedCompletionStatusr r popKeyErrorr2 Z get_debugr3 r rI r^ r r0 Zdoner1 r9 r rK r; r r' rv )r r r r errZ transferredr r' r r r r r r r r r! rJ sL
zIocpProactor._pollc C s | j | d S r7 )r rp r r r r!
_stop_serving9 s zIocpProactor._stop_servingc C s | j d krd S t| j D ]\}\}}}}| r6q t|trBq z | W q tk
r } z6| j d k rd||d}|j
r|j
|d< | j | W 5 d }~X Y q X 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 rU r0 r1 r2 r r3 time monotonicr debugrJ r rI r^ ) r r' rM r r r r4 r5 Z
msg_updateZ
start_timeZnext_msgr r r! ru ? s@
zIocpProactor.closec C s | d S r7 )ru r8 r r r! rw n s zIocpProactor.__del__)r )N)r )r )r )r N)r )N)N) r> r? r@ r r r r r r r r r r r r r r r r r r` r r r r_ r rJ r ru rw 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 r7 )_procZpollZ_process_exited)r
returncoder8 r r! r y s
z4_WindowsSubprocessTransport._start..callback) r
Popenr r2 rZ r intrC r )
r r r r r r r r r r r r8 r! _startt s z"_WindowsSubprocessTransport._startN)r> r? r@ r r r r r! r r s r c @ s e Zd ZeZdS )r N)r> r? r@ r
_loop_factoryr r r r! r s r c @ s e Zd ZeZdS )r N)r> r? r@ r
r r r r r! r s r ).rP rI r r r r r r rd r r r r r r r r
logr __all__rn r ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr r ZFuturer rB rU rY objectra ZBaseSelectorEventLooprx ZBaseProactorEventLoopr
r ZBaseSubprocessTransportr r ZBaseDefaultEventLoopPolicyr r r r r r r! sP 0J4;e `