3
\l @ s d dl Z d dlZd dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z ddlm
Z
ddlm Z ddlm Z dd lm
Z
dd
lmZ dd lmZ dd lmZ d
dddgZd ZdZdZdZdZdZG dd de jZG dd de jZ G dd de Z G d d d e Z G d d d e Z G d d! d!e j!Z"G d"d de
j#Z$G d#d dZ%G d$d% d%ej&Z'e"Z(G d&d' d'ej)Z*e*Z+dS )( N )events)base_subprocess)futures)proactor_events)selector_events)tasks)
windows_utils) _overlapped) coroutine)loggerSelectorEventLoopProactorEventLoop IocpProactorDefaultEventLoopPolicyl 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 r )super__init___source_traceback_ov)selfovr ) __class__ ./usr/lib64/python3.6/asyncio/windows_events.pyr - s z_OverlappedFuture.__init__c s@ t j }| jd k r<| jjr dnd}|jdd|| jjf |S )NpendingZ completedr zoverlapped=<%s, %#x>)r
_repr_infor r insertaddress)r infostate)r r r r 3 s
z _OverlappedFuture._repr_infoc C sr | j d krd S y| j j W nJ tk
rf } z.d|| d}| jrJ| j|d< | jj| W Y d d }~X 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 | j t j S )N)r- r r' )r )r r r r' J s z_OverlappedFuture.cancelc s t j| | j d S )N)r
set_exceptionr- )r r$ )r r r r. N s z _OverlappedFuture.set_exceptionc s t j| d | _d S )N)r
set_resultr )r result)r r r r/ R s z _OverlappedFuture.set_result)
__name__
__module__ __qualname__r r r- r' r. 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 )_BaseWaitHandleFutureN)r c s8 t j|d | jr | jd= || _|| _|| _d| _d S )N)r r Tr )r r r r _handle _wait_handle _registered)r r handle wait_handler )r r r r Z s z _BaseWaitHandleFuture.__init__c C s t j| jdt jkS )Nr )_winapiZWaitForSingleObjectr6 Z
WAIT_OBJECT_0)r r r r _pollh s z_BaseWaitHandleFuture._pollc s\ t j }|jd| j | jd k r>| j r0dnd}|j| | jd k rX|jd| j |S )Nz
handle=%#xZsignaledZwaitingzwait_handle=%#x)r r appendr6 r< r7 )r r! r" )r r r r m s
z _BaseWaitHandleFuture._repr_infoc C s
d | _ d S )N)r )r futr r r _unregister_wait_cbw s z)_BaseWaitHandleFuture._unregister_wait_cbc C s | j s
d S d| _ | j}d | _ytj| W nZ tk
r } z>|jtjkrtd|| d}| jrd| j|d< | jj | d S W Y d d }~X nX | j
d d S )NFz$Failed to unregister the wait handle)r# r$ r% r& ) r8 r7 r
ZUnregisterWaitr( winerrorERROR_IO_PENDINGr r) r* r? )r r: r+ r, r r r _unregister_wait| s"
z&_BaseWaitHandleFuture._unregister_waitc s | j t j S )N)rB r r' )r )r r r r' s z _BaseWaitHandleFuture.cancelc s | j t j| d S )N)rB r r. )r r$ )r r r r. s z#_BaseWaitHandleFuture.set_exceptionc s | j t j| d S )N)rB r r/ )r r0 )r r r r/ s z _BaseWaitHandleFuture.set_result) r1 r2 r3 r r< r r? rB r' r. r/ r4 r r )r r r5 W s
r5 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 j| | jd k r | j| d S )N)r r/ rD )r r0 )r r r r/ s
z _WaitCancelFuture.set_resultc s$ t j| | jd k r | j| d S )N)r r. rD )r r$ )r r r r. s
z _WaitCancelFuture.set_exception)r1 r2 r3 r r' r/ r. r4 r r )r r rC s rC c s6 e Zd Zdd fdd
Z fddZdd Z ZS ) _WaitHandleFutureN)r c s< t j||||d || _d| _tjd ddd | _d | _d S )N)r TF)r r _proactorZ_unregister_proactorr
Z CreateEvent_event
_event_fut)r r r9 r: proactorr )r r r r s
z_WaitHandleFuture.__init__c sF | j d k r"tj| j d | _ d | _| jj| j d | _t j| d S )N) rI r; CloseHandlerJ rH _unregisterr r r? )r r> )r r r r? s
z%_WaitHandleFuture._unregister_wait_cbc C s | j s
d S d| _ | j}d | _ytj|| j W nZ tk
r } z>|jtjkrxd|| d}| jrh| j|d< | j j
| d S W Y d d }~X nX | j j | j| j
| _d S )NFz$Failed to unregister the wait handle)r# r$ r% r& )r8 r7 r
ZUnregisterWaitExrI r( r@ rA r r) r* rH _wait_cancelr? rJ )r r: r+ r, r r r rB s$
z"_WaitHandleFuture._unregister_wait)r1 r2 r3 r r? rB r4 r r )r r rG s rG c @ s8 e Zd Zdd Zdd Zdd Zdd Zd d
ZeZd S )
PipeServerc C s, || _ tj | _d | _d | _| jd| _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 | jd }| _ |S )NF)rT rV )r tmpr r r _get_unconnected_pipe s z PipeServer._get_unconnected_pipec C sr | j r d S tjtjB }|r&|tjO }tj| j|tjtjB tj B tj
t j t j tj
tj}t j|}| jj| |S )N)closedr; ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPiperP ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr ZBUFSIZEZNMPWAIT_WAIT_FOREVERNULL
PipeHandlerS add)r firstflagshpiper r r rV s
z PipeServer._server_pipe_handlec C s
| j d kS )N)rP )r r r r rY s zPipeServer.closedc C sV | j d k r| j j d | _ | jd k rRx| jD ] }|j q,W d | _d | _| jj d S )N)rU r' rP rS closerT clear)r r` r r r ra s
zPipeServer.closeN) r1 r2 r3 r rX rV rY ra __del__r r r r rO s
rO c @ s e Zd Zdd ZdS )_WindowsSelectorEventLoopc C s t j S )N)r
socketpair)r r r r _socketpair+ s z%_WindowsSelectorEventLoop._socketpairN)r1 r2 r3 rf r r r r rd ( s rd c sL e Zd Zd fdd Zdd Zedd Zedd Zed
d
d Z Z S )r Nc s |d krt }t j| d S )N)r r r )r rK )r r r r 2 s zProactorEventLoop.__init__c C s t j S )N)r re )r r r r rf 7 s z ProactorEventLoop._socketpairc c s8 | j j|}|E d H }| }| j||d|id}||fS )Naddr)extra)rH connect_pipe_make_duplex_pipe_transport)r protocol_factoryr fr` protocoltransr r r create_pipe_connection: s
z(ProactorEventLoop.create_pipe_connectionc s. t d fdd j gS )Nc s d }yj| rL| j }jj| j r2|j d S }j||d id j }|d kr`d S jj|} W n t k
r } zH|r|j
d krj d||d |j nj rt
jd|dd W Y d d }~X n2 tjk
r |r|j Y nX | _| j d S )
Nrg )rh r zPipe accept failed)r# r$ r` z Accept pipe failed on pipe %rT)exc_infor )r0 rS discardrY ra rj rX rH accept_piper( filenor* Z_debugr Zwarningr CancelledErrorrU add_done_callback)rl r` rm r+ )r loop_accept_piperk r serverr r rv G s<
z>ProactorEventLoop.start_serving_pipe..loop_accept_pipe)N)rO Z call_soon)r rk r r )r rv rk r rw r start_serving_pipeC s (
z$ProactorEventLoop.start_serving_pipec k s | j }
t| |||||||f|
|d| } y|
E d H W n& tk
r` } z
| }
W Y d d } ~ X nX d }
|
d k r| j | j E d H |
| S )N)waiterrh )
create_future_WindowsSubprocessTransport Exceptionra Z_wait)r rm argsshellstdinstdoutstderrbufsizerh kwargsry Ztranspr+ errr r r _make_subprocess_transportr s
z,ProactorEventLoop._make_subprocess_transport)N)N)
r1 r2 r3 r rf r ro rx r r4 r r )r r r / s /c @ s e Zd Zd0ddZdd Zdd Zd1d d
Zd d Zd2ddZd3ddZ dd Z
dd Z dd Z e
dd Zd4ddZd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd5d(d)Zd*d+ Zd,d- Zd.d/ ZdS )6r c C sD d | _ g | _tjtjtd|| _i | _tj | _
g | _ tj | _ d S )Nr )
r) _resultsr
CreateIoCompletionPortINVALID_HANDLE_VALUErZ _iocp_cacherQ rR r8
_unregistered_stopped_serving)r Z concurrencyr r r r s
zIocpProactor.__init__c C s d| j jt| jt| jf S )Nz <%s overlapped#=%s result#=%s>)r r1 lenr 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| | j }g | _ |S )N)r r< )r timeoutrW r r r select s
zIocpProactor.selectc C s | j j }|j| |S )N)r) rz r/ )r valuer> r r r _result s
zIocpProactor._resultr c C sz | j | tjt}y4t|tjr6|j|j || n|j|j | W n t k
rb | j
dS X dd }| j |||S )N c S sJ y|j S tk
rD } z |jtjkr2t|j n W Y d d }~X nX d S )N) getresultr( r@ r
ERROR_NETNAME_DELETEDConnectionResetErrorr} )rn keyr r+ r r r finish_recv s z&IocpProactor.recv..finish_recv) _register_with_iocpr
OverlappedrZ
isinstancesocketZWSARecvrs ZReadFileBrokenPipeErrorr _register)r connnbytesr^ r r r r r recv s
zIocpProactor.recvc C sZ | j | tjt}t|tjr4|j|j || n|j|j | dd }| j |||S )Nc S sJ y|j S tk
rD } z |jtjkr2t|j n W Y d d }~X nX d S )N)r r( r@ r
r r r} )rn r r r+ r r r finish_send s z&IocpProactor.send..finish_send)
r r
r rZ r r ZWSASendrs Z WriteFiler )r r bufr^ r r r r r send s
zIocpProactor.sendc sz | j | jj tjt}|jj j fdd}tdd }| j ||}|| }t
j || j d |S )Nc sD |j tjdj } jtjtj| j j
j fS )Nz@P) r structZpackrs
setsockoptr
SOL_SOCKETr
ZSO_UPDATE_ACCEPT_CONTEXT
settimeoutZ
gettimeoutZ getpeername)rn r r r )r listenerr r
finish_accept s
z*IocpProactor.accept..finish_acceptc s s4 y| E d H W n t jk
r. |j Y nX d S )N)r rt ra )r% r r r r accept_coro s
z(IocpProactor.accept..accept_coro)r )
r _get_accept_socketfamilyr
r rZ ZAcceptExrs r r r Z
ensure_futurer) )r r r r r r% coror )r r r accept s
zIocpProactor.acceptc s | j ytj j j W nB tk
rb } z&|jtjkr@ j d dkrR W Y d d }~X nX tj
t }|j j | fdd}| j
| |S )Nr r c s |j jtjtjd S )Nr )r r r r r
ZSO_UPDATE_CONNECT_CONTEXT)rn r r )r r r finish_connect s
z,IocpProactor.connect..finish_connect)r r
Z BindLocalrs r r( r@ errnoZ WSAEINVALZ getsocknamer rZ Z ConnectExr )r r r er r r )r r connect s
zIocpProactor.connectc sJ | j tjt}|j j }|r0| j S fdd}| j| |S )Nc s |j S )N)r )rn r r )r` r r finish_accept_pipe s z4IocpProactor.accept_pipe..finish_accept_pipe)r r
r rZ ZConnectNamedPipers r r )r r` r Z connectedr r )r` r rr
s
zIocpProactor.accept_pipec c sz t }xjytj|}P W n0 tk
rF } z|jtjkr6 W Y d d }~X nX t|d t}tj || j
dE d H qW t j |S )N )r )
CONNECT_PIPE_INIT_DELAYr
Z ConnectPiper( r@ ZERROR_PIPE_BUSYminCONNECT_PIPE_MAX_DELAYr Zsleepr) r r[ )r r Zdelayr9 r+ r r r ri s
zIocpProactor.connect_pipec C s | j ||dS )NF)_wait_for_handle)r r9 r r r r wait_for_handle/ s z IocpProactor.wait_for_handlec C s | j |d d}||_|S )NT)r rD )r rE Z
done_callbackr> r r r rN 7 s zIocpProactor._wait_cancelc s |d krt j}ntj|d }tjt}tj|| j|j |}|rTt
|||| j d nt |||| | j d j
rv j
d= fdd} |d|f| j|j < S )Ng @@)r r c s j S )N)r< )rn r r )rl r r finish_wait_for_handleR s z=IocpProactor._wait_for_handle..finish_wait_for_handler r )r; INFINITEmathceilr
r rZ ZRegisterWaitWithQueuer r rC r) rG r r )r r9 r Z
_is_cancelmsr r: r r )rl r r > s
z IocpProactor._wait_for_handlec C s0 || j kr,| j j| tj|j | jdd d S )Nr )r8 r\ r
r rs r )r objr r r r ^ s
z IocpProactor._register_with_iocpc C s t || jd}|jr |jd= |jsjy|d d |}W n, tk
r^ } z|j| W Y d d }~X n X |j| ||||f| j|j< |S )N)r r r ) r r) r r r( r. r/ r r )r r r callbackrl r r r r r r h s
zIocpProactor._registerc C s | j j| d S )N)r r= )r r r r r rM s zIocpProactor._unregisterc C s t j |}|jd |S )Nr )r r )r r sr r r r s
z IocpProactor._get_accept_socketc C s |d krt }n0|dk r tdn tj|d }|t kr>tdxtj| j|}|d krZP d}|\}}}}y| jj|\}} }
} W nV t k
r | j
j r| j
j dd||||f d |dtj
fkrtj| wBY nX |
| jkr|j qB|j sBy| ||| } W n: tk
r@ }
z |j|
| jj| W Y d d }
~
X qBX |j| | jj| qBW x | jD ]} | jj| jd qdW | jj 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 r
ZGetQueuedCompletionStatusr r popKeyErrorr) Z get_debugr* r r; rL r r' doner( r. r r= r/ r r rb )r r r r r Z transferredr r rl r r r r r r r r r< sJ
zIocpProactor._pollc C s | j j| d S )N)r r\ )r r r r r
_stop_serving s zIocpProactor._stop_servingc C s xt | jj D ]\}\}}}}|j r*qt|tr6qy |j W q tk
r } z8| jd k rd||d}|j rz|j |d< | jj
| W Y d d }~X qX qW x | jr| j dst j
d qW g | _| jd k rtj| j d | _d S )NzCancelling a future failed)r# r$ r% r& r z"taking long time to close proactor)listr itemsZ cancelledr rC r' r( r) r r* r< r debugr r r; rL )r r r> r r r r+ r, r r r ra s,
"
zIocpProactor.closec C s | j d S )N)ra )r r r r rc s zIocpProactor.__del__)r )N)r )r )N)N)r1 r2 r3 r r r r r r r r r rr r ri r rN r r r rM r r< r ra rc r r r r r s,
7 c @ s e Zd Zdd ZdS )r{ c
sP t j|f|||||d| _ fdd} jjjt jj} | j| d S )N)r~ r r r r c s j j } j| d S )N)_procZpollZ_process_exited)rl
returncode)r r r r s
z4_WindowsSubprocessTransport._start..callback) r Popenr r) rH r intr6 ru )
r r} r~ r r r r r r rl r )r r _start s z"_WindowsSubprocessTransport._startN)r1 r2 r3 r r r r r r{ s r{ c @ s e Zd ZeZdS ) _WindowsDefaultEventLoopPolicyN)r1 r2 r3 r
Z
_loop_factoryr r r r r s r ),r; r r r r rQ r r r r r r r r
Z
coroutinesr logr __all__rZ r ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr r ZFuturer r5 rC rG objectrO ZBaseSelectorEventLooprd ZBaseProactorEventLoopr r ZBaseSubprocessTransportr{ r
ZBaseDefaultEventLoopPolicyr r r r r r sJ 0J4;] k