B
ӭG4+d @ 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
lmZ d Zd Zd Zd
ZdZdZdZG dd de jZG dd de jZ G dd de Z G dd de Z G dd de Z G dd d ej!Z"G d d d e
j#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)proactor_events)selector_events)tasks)
windows_utils)logger)SelectorEventLoopProactorEventLoop IocpProactorDefaultEventLoopPolicy WindowsSelectorEventLoopPolicy WindowsProactorEventLoopPolicyl i i gMbP?g?c sZ e Zd Zdd fdd
Z fddZdd Z fd d
Z fd d Z fd
dZ Z S )_OverlappedFutureN)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__r r r. r( r/ r0
__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 )_BaseWaitHandleFutureN)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 )_winapiZWaitForSingleObjectr7 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 appendr7 r= r8 )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' ) r9 r8 _overlappedZUnregisterWaitr) winerrorERROR_IO_PENDINGr r* r+ r@ )r r; r, r- r r r _unregister_wait s"
z&_BaseWaitHandleFuture._unregister_waitc s | t S )N)rD r r( )r )r r r r( s z _BaseWaitHandleFuture.cancelc s | t | d S )N)rD r r/ )r r% )r r r r/ s z#_BaseWaitHandleFuture.set_exceptionc s | t | d S )N)rD r r0 )r r1 )r r r r0 s z _BaseWaitHandleFuture.set_result) r2 r3 r4 r r= r r@ rD r( r/ r0 r5 r r )r r r6 Z s
r6 c sB e Zd Zdd fdd
Zdd Z fddZ fd d
Z ZS ) _WaitCancelFutureN)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 rF )r r1 )r r r r0 s
z _WaitCancelFuture.set_resultc s$ t | | jd k r | | d S )N)r r/ rF )r r% )r r r r/ s
z _WaitCancelFuture.set_exception)r2 r3 r4 r r( r0 r/ r5 r r )r r rE s rE 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_proactorrA 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) rK r< CloseHandlerL rJ _unregisterr r r@ )r r? )r r r r@ 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' )r9 r8 rA ZUnregisterWaitExrK r) rB rC r r* r+ rJ _wait_cancelr@ rL )r r; r, r- r r r rD s$
z"_WaitHandleFuture._unregister_wait)r2 r3 r4 r r@ rD r5 r r )r r rI s rI 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)rV rX )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_INSTANCEZCreateNamedPiperR ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr ZBUFSIZEZNMPWAIT_WAIT_FOREVERNULL
PipeHandlerU add)r firstflagshpiper r r rX s
z PipeServer._server_pipe_handlec C s
| j d kS )N)rR )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)rW r( rR rU closerV clear)r rb r r r rc s
zPipeServer.closeN) r2 r3 r4 r rZ rX r[ rc __del__r r r r rQ s
rQ c @ s e Zd ZdS )_WindowsSelectorEventLoopN)r2 r3 r4 r r r r rf + s rf c s8 e Zd Zd
fdd Zdd Zdd Zd dd Z ZS ) r Nc s |d krt }t | d S )N)r
r r )r rM )r r r r 2 s zProactorEventLoop.__init__c s8 | j |}|I d H }| }| j||d|id}||fS )Naddr)extra)rJ connect_pipe_make_duplex_pipe_transport)r protocol_factoryr! frb 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 ) Nrg )rh r zPipe accept failed)r$ r% rb z Accept pipe failed on pipe %rT)exc_info)r1 rU discardr[ rc rj rZ rJ accept_piper) filenor+ Z_debugr
Zwarningr CancelledErrorrW add_done_callback)rl rb rm r, )r! loop_accept_piperk r serverr r rv B s<
z>ProactorEventLoop.start_serving_pipe..loop_accept_pipe)N)rQ Z call_soon)r rk r! r )r! rv rk r rw 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)waiterrh )
create_future_WindowsSubprocessTransport Exceptionrc Z_wait) r rm argsshellstdinstdoutstderrbufsizerh kwargsry Ztranspr r r _make_subprocess_transportm s
z,ProactorEventLoop._make_subprocess_transport)N)N)r2 r3 r4 r ro rx r r5 r r )r r r / s 0r c @ s e Zd Zd6ddZdd Zdd Zdd Zd7d d Zd
d Zd8ddZ d9ddZ
d:ddZ dd Z dd Z
dd Zd d Zd d Zd;d d!Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd )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 timeoutrY r r r select s
zIocpProactor.selectc C s | j }|| |S )N)r* rz 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) rB rA ERROR_NETNAME_DELETEDERROR_OPERATION_ABORTEDConnectionResetErrorr} )rn keyr r, r r r finish_recv s
z&IocpProactor.recv..finish_recv) _register_with_iocprA
Overlappedr\
isinstancesocketZWSARecvrs ZReadFileBrokenPipeErrorr _register)r connnbytesr` 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) rB rA r r r r} )rn r r r, r r r r s
z+IocpProactor.recv_into..finish_recv) r rA r r\ r r Z WSARecvIntors Z ReadFileIntor r r )r r bufr` 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) rB rA r r r r} )rn r r r, r r r finish_send s
z&IocpProactor.send..finish_send)
r rA r r\ r r ZWSASendrs 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 structpackrs
setsockoptr
SOL_SOCKETrA ZSO_UPDATE_ACCEPT_CONTEXT
settimeoutZ
gettimeoutZ getpeername)rn 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 rt rc )r& r r r r accept_coro s
z(IocpProactor.accept..accept_coro)r ) r _get_accept_socketfamilyrA r r\ ZAcceptExrs 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 rA ZSO_UPDATE_CONNECT_CONTEXT)rn r r )r r r finish_connect s
z,IocpProactor.connect..finish_connect)r rA Z BindLocalrs r r) rB 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) rB rA r r r r} )rn r r r, r r r finish_sendfile% s
z.IocpProactor.sendfile..finish_sendfile) r rA r r\ Z TransmitFilers 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 )rn r r )rb r r finish_accept_pipe; s z4IocpProactor.accept_pipe..finish_accept_pipe)r rA r r\ ZConnectNamedPipers r r )r rb r Z connectedr r )rb r rr 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_DELAYrA Z ConnectPiper) rB ZERROR_PIPE_BUSYminCONNECT_PIPE_MAX_DELAYr Zsleepr* r r] )r r! Zdelayr: r, r r r ri A s
zIocpProactor.connect_pipec C s | ||dS )NF)_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 rF )r rG Z
done_callbackr? r r r rP \ 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= )rn r r )rl r r finish_wait_for_handley s z=IocpProactor._wait_for_handle..finish_wait_for_handler )r r< INFINITEmathceilrA r r\ ZRegisterWaitWithQueuer r! rE r* rI r r )r r: r Z
_is_cancelmsr r; r r )rl r r c s
z IocpProactor._wait_for_handlec C s0 || j kr,| j | t| | jdd d S )Nr )r9 r^ rA r rs 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 callbackrl r r r r r r s
zIocpProactor._registerc C s | | j| d S )N)r r r> )r r r r r rO 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 rA ZGetQueuedCompletionStatusr r popKeyErrorr* Z get_debugr+ r r< rN r r( Zdoner) r/ r r> r0 r r! rd )r r r r errZ transferredr r! rl 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 rE r( r) r* r r+ timeZ monotonicr
debugr= r r< rN ) r r! r? r r r r, r- Z
msg_updateZ
start_timeZnext_msgr r r rc s:
"
zIocpProactor.closec C s | d S )N)rc )r r r r re * s zIocpProactor.__del__)r )N)r )r )r )N)N) r2 r3 r4 r r r r r r r r r r r r rr ri r rP r r r rO r r= r rc re r r r r r
s2
"
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)rl
returncode)r r r r 5 s
z4_WindowsSubprocessTransport._start..callback) r Popenr r* rJ r intr7 ru )
r r} r~ r r r r r r rl 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 )-rA r< r r r r r r rS r r r r r r r logr
__all__r\ r ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr r ZFuturer r6 rE rI objectrQ ZBaseSelectorEventLooprf ZBaseProactorEventLoopr r
ZBaseSubprocessTransportr{ r ZBaseDefaultEventLoopPolicyr r r r r r r sN 0J4;P 2