o
     JAf                     @   s  d Z ddlZejdkredddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlZddlmZ ddlmZ ddlmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ dZdZdZdZdZdZdZG dd dejZ G dd dejZ!G dd de!Z"G dd de!Z#G dd de$Z%G dd  d ej&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.Z0e0Z1dS )+z.Selector and proactor event loops for Windows.    Nwin32z
win32 only   )events)base_subprocess)futures)
exceptions)proactor_events)selector_events)tasks)windows_utils)logger)SelectorEventLoopProactorEventLoopIocpProactorDefaultEventLoopPolicyWindowsSelectorEventLoopPolicyWindowsProactorEventLoopPolicy    i  i  gMbP?g?c                       s`   e Zd ZdZdd fdd
Z fddZdd	 Zd fd
d	Z fddZ fddZ	  Z
S )_OverlappedFuturezSubclass of Future which represents an overlapped operation.

    Cancelling it will immediately cancel the overlapped operation.
    Nloopc                   s&   t  j|d | jr| jd= || _d S Nr   )super__init___source_traceback_ov)selfovr   	__class__ -/usr/lib/python3.10/asyncio/windows_events.pyr   6   s   
z_OverlappedFuture.__init__c                    sH   t   }| jd ur"| jjrdnd}|dd| d| jjdd |S )NpendingZ	completedr   zoverlapped=<z, #x>)r   
_repr_infor   r#   insertaddressr   infostater   r!   r"   r&   <   s
   

 z_OverlappedFuture._repr_infoc              
   C   st   | j d u rd S z| j   W n& ty4 } zd|| d}| jr$| j|d< | j| W Y d }~nd }~ww d | _ d S )Nz&Cancelling an overlapped future failedmessage	exceptionfuturesource_traceback)r   cancelOSErrorr   _loopcall_exception_handler)r   exccontextr!   r!   r"   _cancel_overlappedC   s   


	z$_OverlappedFuture._cancel_overlappedc                       |    t j|dS N)msg)r7   r   r1   r   r:   r   r!   r"   r1   S      z_OverlappedFuture.cancelc                    s   t  | |   d S N)r   set_exceptionr7   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&   r7   r1   r>   r@   __classcell__r!   r!   r   r"   r   0   s    r   c                       sp   e Zd ZdZdd fdd
Zdd Z fdd	Zd
d Zdd Zd fdd	Z	 fddZ
 fddZ  ZS )_BaseWaitHandleFuturez2Subclass of Future which represents a wait handle.Nr   c                   s8   t  j|d | jr| jd= || _|| _|| _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 | jdt jkS Nr   )_winapiZWaitForSingleObjectrI   ZWAIT_OBJECT_0r   r!   r!   r"   _pollq   s   z_BaseWaitHandleFuture._pollc                    sd   t   }|d| jd | jd ur!|  rdnd}|| | jd ur0|d| jd |S )Nzhandle=r$   signaledZwaitingzwait_handle=)r   r&   appendrI   rQ   rJ   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 | _zt| W n3 tyH } z'|jtjkr>d|| d}| jr1| j|d< | j	| W Y d }~d S W Y d }~nd }~ww | 
d  d S NFz$Failed to unregister the wait handler,   r0   )rK   rJ   _overlappedZUnregisterWaitr2   winerrorERROR_IO_PENDINGr   r3   r4   rV   r   rM   r5   r6   r!   r!   r"   _unregister_wait   s*   
z&_BaseWaitHandleFuture._unregister_waitc                    r8   r9   )r\   r   r1   r;   r   r!   r"   r1      r<   z_BaseWaitHandleFuture.cancelc                       |    t | d S r=   )r\   r   r>   r?   r   r!   r"   r>         z#_BaseWaitHandleFuture.set_exceptionc                    r]   r=   )r\   r   r@   rA   r   r!   r"   r@      r^   z _BaseWaitHandleFuture.set_resultr=   )rC   rD   rE   rF   r   rQ   r&   rV   r\   r1   r>   r@   rG   r!   r!   r   r"   rH   `   s    
rH   c                       sF   e Zd ZdZdd fdd
Zdd Z fdd	Z fd
dZ  ZS )_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   eventrM   r   r   r!   r"   r      s   
z_WaitCancelFuture.__init__c                 C   s   t d)Nz'_WaitCancelFuture must not be cancelled)RuntimeErrorrP   r!   r!   r"   r1      s   z_WaitCancelFuture.cancelc                    (   t  | | jd ur| |  d S d S r=   )r   r@   r`   rA   r   r!   r"   r@         
z_WaitCancelFuture.set_resultc                    rc   r=   )r   r>   r`   r?   r   r!   r"   r>      rd   z_WaitCancelFuture.set_exception)	rC   rD   rE   rF   r   r1   r@   r>   rG   r!   r!   r   r"   r_      s    r_   c                       s6   e Zd Zdd fdd
Z fddZdd Z  ZS )	_WaitHandleFutureNr   c                   s<   t  j||||d || _d| _td ddd | _d | _d S )Nr   TF)r   r   	_proactorZ_unregister_proactorrX   ZCreateEvent_event
_event_fut)r   r   rL   rM   proactorr   r   r!   r"   r      s
   
z_WaitHandleFuture.__init__c                    sF   | j d urt| j  d | _ d | _| j| j d | _t | d S r=   )	rg   rO   CloseHandlerh   rf   _unregisterr   r   rV   rT   r   r!   r"   rV      s   
	z%_WaitHandleFuture._unregister_wait_cbc              
   C   s   | j sd S d| _ | j}d | _z	t|| j W n3 tyJ } z'|jtjkr@d|| d}| jr3| j|d< | j	
| W Y d }~d S W Y d }~nd }~ww | j| j| j| _d S rW   )rK   rJ   rX   ZUnregisterWaitExrg   r2   rY   rZ   r   r3   r4   rf   _wait_cancelrV   rh   r[   r!   r!   r"   r\      s.   


z"_WaitHandleFuture._unregister_wait)rC   rD   rE   r   rV   r\   rG   r!   r!   r   r"   re      s    re   c                   @   s<   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d ZeZ	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)_addressweakrefWeakSet_free_instances_pipe_accept_pipe_future_server_pipe_handle)r   r(   r!   r!   r"   r      s
   
zPipeServer.__init__c                 C   s   | j | d}| _ |S )NF)rs   ru   )r   tmpr!   r!   r"   _get_unconnected_pipe  s   z PipeServer._get_unconnected_pipec              
   C   sr   |   rd S tjtjB }|r|tjO }t| j|tjtjB tj	B tj
tjtjtjtj}t|}| j| |S r=   )closedrO   ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZFILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPipero   ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ	PIPE_WAITZPIPE_UNLIMITED_INSTANCESr   BUFSIZEZNMPWAIT_WAIT_FOREVERNULL
PipeHandlerr   add)r   firstflagshpiper!   r!   r"   ru     s"   


zPipeServer._server_pipe_handlec                 C   s
   | j d u S r=   )ro   rP   r!   r!   r"   rx        
zPipeServer.closedc                 C   sV   | j d ur| j   d | _ | jd ur)| jD ]}|  qd | _d | _| j  d S d S r=   )rt   r1   ro   rr   closers   clear)r   r   r!   r!   r"   r   "  s   




zPipeServer.closeN)
rC   rD   rE   rF   r   rw   ru   rx   r   __del__r!   r!   r!   r"   rm      s    
rm   c                   @   s   e Zd ZdZdS )_WindowsSelectorEventLoopz'Windows version of selector event loop.N)rC   rD   rE   rF   r!   r!   r!   r"   r   1      r   c                       sJ   e Zd ZdZd fdd	Z fddZdd Zd	d
 Z	dddZ  Z	S )r   z2Windows version of proactor event loop using IOCP.Nc                    s   |d u rt  }t | d S r=   )r   r   r   )r   ri   r   r!   r"   r   8  s   zProactorEventLoop.__init__c              	      s   z+|  | j t   W | jd ur*| jj}| j  |d ur%| j| d | _d S d S | jd urG| jj}| j  |d urD| j| d | _w r=   )		call_soonZ_loop_self_readingr   run_foreverZ_self_reading_futurer   r1   rf   rk   r   r   r   r!   r"   r   =  s   



zProactorEventLoop.run_foreverc                    s:   | j |}|I d H }| }| j||d|id}||fS )Naddrextra)rf   connect_pipe_make_duplex_pipe_transport)r   protocol_factoryr(   fr   protocoltransr!   r!   r"   create_pipe_connectionP  s   
z(ProactorEventLoop.create_pipe_connectionc                    s0   t  d fdd	 gS )Nc              
      s2  d }z7| r'|   }j|  r|  W d S  }j||d id  }|d u r2W d S j|} W nU t	y{ } z6|rW|
 dkrWd||d |  njritjd|dd W Y d }~d S W Y d }~d S W Y d }~d S d }~w tjy   |r|  Y d S Y d S w | _|  d S )	Nr   r   r   zPipe accept failed)r-   r.   r   zAccept pipe failed on pipe %rT)exc_info)rB   rr   discardrx   r   r   rw   rf   accept_piper2   filenor4   Z_debugr   warningr   CancelledErrorrt   add_done_callback)r   r   r   r5   r(   loop_accept_piper   r   Zserverr!   r"   r   [  sL   

z>ProactorEventLoop.start_serving_pipe.<locals>.loop_accept_piper=   )rm   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  ttfy)     ty;   |  | I d H   w )N)waiterr   )create_future_WindowsSubprocessTransport
SystemExitKeyboardInterruptBaseExceptionr   _wait)r   r   argsshellstdinstdoutstderrbufsizer   kwargsr   Ztranspr!   r!   r"   _make_subprocess_transport  s&   
z,ProactorEventLoop._make_subprocess_transportr=   )
rC   rD   rE   rF   r   r   r   r   r   rG   r!   r!   r   r"   r   5  s    0r   c                   @   s   e Zd ZdZd;ddZdd Zdd Zd	d
 Zd<ddZdd Z	d=ddZ
d=ddZd=ddZd>ddZd=ddZdd Zdd Zdd  Zd!d" Zd#d$ Zd<d%d&Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd<d3d4Zd5d6 Zd7d8 Zd9d: ZdS )?r   z#Proactor implementation using IOCP.r   c                 C   sD   d | _ g | _ttjtd|| _i | _t	 | _
g | _t	 | _d S rN   )r3   _resultsrX   CreateIoCompletionPortINVALID_HANDLE_VALUErz   _iocp_cacherp   rq   rK   _unregistered_stopped_serving)r   Zconcurrencyr!   r!   r"   r     s   

zIocpProactor.__init__c                 C   s   | j d u r	tdd S )NzIocpProactor is closed)r   rb   rP   r!   r!   r"   _check_closed  s   
zIocpProactor._check_closedc                 C   sF   dt | j dt | j g}| jd u r|d d| jjd|f S )Nzoverlapped#=%sz
result#=%srx   z<%s %s> )lenr   r   r   rS   r    rC   join)r   r*   r!   r!   r"   __repr__  s   

zIocpProactor.__repr__c                 C   s
   || _ d S r=   )r3   )r   r   r!   r!   r"   set_loop  r   zIocpProactor.set_loopNc                 C   s    | j s| | | j }g | _ |S r=   )r   rQ   )r   timeoutrv   r!   r!   r"   select  s
   
zIocpProactor.selectc                 C   s   | j  }|| |S r=   )r3   r   r@   )r   valuerU   r!   r!   r"   _result     

zIocpProactor._resultr   c                 C   |   |  | tt}zt|tjr|| || n|| | W n t	y2   | 
d Y S w dd }| |||S )N    c              
   S   B   z|  W S  ty  } z|jtjtjfv rt|j  d }~ww r=   	getresultr2   rY   rX   ZERROR_NETNAME_DELETEDZERROR_OPERATION_ABORTEDConnectionResetErrorr   r   keyr   r5   r!   r!   r"   finish_recv     

z&IocpProactor.recv.<locals>.finish_recv)_register_with_iocprX   
Overlappedrz   
isinstancesocketZWSARecvr   ZReadFileBrokenPipeErrorr   	_registerr   connnbytesr~   r   r   r!   r!   r"   recv     


zIocpProactor.recvc                 C   r   )Nr   c              
   S   r   r=   r   r   r!   r!   r"   r     r   z+IocpProactor.recv_into.<locals>.finish_recv)r   rX   r   rz   r   r   ZWSARecvIntor   ZReadFileIntor   r   r   )r   r   bufr~   r   r   r!   r!   r"   	recv_into  r   zIocpProactor.recv_intoc                 C   s^   |  | tt}z|| || W n ty#   | d Y S w dd }| |||S )N)r   Nc              
   S   r   r=   r   r   r!   r!   r"   r     r   z*IocpProactor.recvfrom.<locals>.finish_recv)	r   rX   r   rz   ZWSARecvFromr   r   r   r   r   r!   r!   r"   recvfrom  s   


zIocpProactor.recvfromc                 C   s>   |  | tt}|| ||| dd }| |||S )Nc              
   S   r   r=   r   r   r!   r!   r"   finish_send	  r   z(IocpProactor.sendto.<locals>.finish_send)r   rX   r   rz   Z	WSASendTor   r   )r   r   r   r~   r   r   r   r!   r!   r"   sendto  s
   


zIocpProactor.sendtoc                 C   sZ   |  | tt}t|tjr|| || n|| | dd }| 	|||S )Nc              
   S   r   r=   r   r   r!   r!   r"   r     r   z&IocpProactor.send.<locals>.finish_send)
r   rX   r   rz   r   r   ZWSASendr   Z	WriteFiler   )r   r   r   r~   r   r   r!   r!   r"   send  s   


zIocpProactor.sendc                    sv   |   | j tt}|     fdd}dd }| ||}|| }t	j
|| jd |S )Nc                    sD   |   td } tjtj|  	
     fS )Nz@P)r   structpackr   
setsockoptr   
SOL_SOCKETrX   ZSO_UPDATE_ACCEPT_CONTEXT
settimeout
gettimeoutgetpeername)r   r   r   r   r   listenerr!   r"   finish_accept/  s   z*IocpProactor.accept.<locals>.finish_acceptc                    s.   z| I d H  W d S  t jy   |   w r=   )r   r   r   )r/   r   r!   r!   r"   accept_coro8  s   z(IocpProactor.accept.<locals>.accept_coror   )r   _get_accept_socketfamilyrX   r   rz   ZAcceptExr   r   r
   Zensure_futurer3   )r   r   r   r   r   r/   coror!   r   r"   accept)  s   

	
zIocpProactor.acceptc              
      s    j tjkrt  | | j }|d  |S | 	  zt
   j W n" tyL } z|jtjkr9   d dkrB W Y d }~nd }~ww tt}|  |  fdd}| | |S )Nr   r   c                    s   |    tjtjd  S rN   )r   r   r   r   rX   ZSO_UPDATE_CONNECT_CONTEXTr   r   r   r   r!   r"   finish_connect[  s
   z,IocpProactor.connect.<locals>.finish_connect)typer   
SOCK_DGRAMrX   Z
WSAConnectr   r3   r   r@   r   Z	BindLocalr   r2   rY   errnoZ	WSAEINVALgetsocknamer   rz   Z	ConnectExr   )r   r   r(   rU   er   r   r!   r   r"   connectE  s(   



zIocpProactor.connectc           	   	   C   sb   |  | tt}|d@ }|d? d@ }|| t| |||dd dd }| |||S )Nr       r   c              
   S   r   r=   r   r   r!   r!   r"   finish_sendfilen  r   z.IocpProactor.sendfile.<locals>.finish_sendfile)	r   rX   r   rz   ZTransmitFiler   msvcrtZget_osfhandler   )	r   sockfileoffsetcountr   Z
offset_lowZoffset_highr   r!   r!   r"   sendfiled  s   


	zIocpProactor.sendfilec                    sJ   |    tt}|  }|r|  S  fdd}| | |S )Nc                    s   |    S r=   )r   r   r   r!   r"   finish_accept_pipe  s   z4IocpProactor.accept_pipe.<locals>.finish_accept_pipe)r   rX   r   rz   ZConnectNamedPiper   r   r   )r   r   r   Z	connectedr   r!   r   r"   r   y  s   


zIocpProactor.accept_pipec              
      st   t }	 zt|}W n) ty$ } z|jtjkr W Y d }~nd }~ww t|d t}t	|I d H  qt
|S )NT   )CONNECT_PIPE_INIT_DELAYrX   ZConnectPiper2   rY   ZERROR_PIPE_BUSYminCONNECT_PIPE_MAX_DELAYr
   sleepr   r{   )r   r(   delayrL   r5   r!   r!   r"   r     s    

zIocpProactor.connect_pipec                 C   s   |  ||dS )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   rL   r   r!   r!   r"   wait_for_handle  s   zIocpProactor.wait_for_handlec                 C   s   |  |d d}||_|S rn   )r  r`   )r   ra   Zdone_callbackrU   r!   r!   r"   rl     s   zIocpProactor._wait_cancelc                    s   |    |d u rtj}nt|d }tt}t|| j	|j
|}|r.t|||| jd n
t|||| | jd  jr? jd=  fdd} |d|f| j|j
<  S )N     @@r   r   c                    s      S r=   )rQ   r   r   r!   r"   finish_wait_for_handle  s   z=IocpProactor._wait_for_handle.<locals>.finish_wait_for_handler   )r   rO   INFINITEmathceilrX   r   rz   ZRegisterWaitWithQueuer   r(   r_   r3   re   r   r   )r   rL   r   Z
_is_cancelmsr   rM   r  r!   r  r"   r    s$   

	zIocpProactor._wait_for_handlec                 C   s4   || j vr| j | t| | jdd d S d S rN   )rK   r|   rX   r   r   r   r   objr!   r!   r"   r     s   
z IocpProactor._register_with_iocpc              
   C   s   |    t|| jd}|jr|jd= |js:z|d d |}W n ty4 } z|| W Y d }~n
d }~ww || ||||f| j|j	< |S r   )
r   r   r3   r   r#   r2   r>   r@   r   r(   )r   r   r  callbackr   r   r   r!   r!   r"   r     s   

zIocpProactor._registerc                 C   s   |    | j| dS )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   rS   r   r!   r!   r"   rk     s   zIocpProactor._unregisterc                 C   s   t  |}|d |S rN   )r   r   )r   r   sr!   r!   r"   r     r   zIocpProactor._get_accept_socketc                 C   s|  |d u rt }n|dk rtdt|d }|t krtd	 t| j|}|d u r+n~d}|\}}}}z| j|\}}	}
}W n) t	yh   | j
 rZ| j
dd||||f d |dtjfvrft| Y qw |
| jv rs|  n5| sz||||	}W n ty } z|| | j| W Y d }~nd }~ww || | j| q| jD ]
}	| j|	jd  q| j  d S )	Nr   znegative timeoutr  ztimeout too bigTz8GetQueuedCompletionStatus() returned an unexpected eventz)err=%s transferred=%s key=%#x address=%#x)r-   status)r  
ValueErrorr  r	  rX   ZGetQueuedCompletionStatusr   r   popKeyErrorr3   Z	get_debugr4   r   rO   rj   r   r1   doner2   r>   r   rS   r@   r   r(   r   )r   r   r
  r  errZtransferredr   r(   r   r   r  r  r   r   r!   r!   r"   rQ     sV   

	





'zIocpProactor._pollc                 C   s   | j | d S r=   )r   r|   r  r!   r!   r"   _stop_serving>  s   zIocpProactor._stop_servingc                 C   s  | j d u rd S t| j D ]D\}\}}}}| rqt|tr!qz|  W q tyR } z| j	d urHd||d}|j
rB|j
|d< | j	| W Y d }~qd }~ww d}t }	|	| }
| jr|
t krwtd| t |	  t | }
| | | js`g | _t| j  d | _ d S )NzCancelling a future failedr,   r0   g      ?z,%r is running after closing for %.1f seconds)r   listr   items	cancelledr   r_   r1   r2   r3   r   r4   time	monotonicr   debugrQ   r   rO   rj   )r   r(   rU   r   r  r  r5   r6   Z
msg_updateZ
start_timeZnext_msgr!   r!   r"   r   D  sD   




	
zIocpProactor.closec                 C   s   |    d S r=   )r   rP   r!   r!   r"   r   s  s   zIocpProactor.__del__)r   r=   )r   )r   N)rC   rD   rE   rF   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  rl   r  r   r   rk   r   rQ   r  r   r   r!   r!   r!   r"   r     s:    








"
 

7/r   c                   @   s   e Zd Zdd ZdS )r   c           
         sP   t j|f|||||d| _ fdd} jjt jj}	|	| d S )N)r   r   r   r   r   c                    s    j  } | d S r=   )_procpollZ_process_exited)r   
returncoderP   r!   r"   r  ~  s   
z4_WindowsSubprocessTransport._start.<locals>.callback)	r   Popenr  r3   rf   r  intrI   r   )
r   r   r   r   r   r   r   r   r  r   r!   rP   r"   _starty  s   z"_WindowsSubprocessTransport._startN)rC   rD   rE   r!  r!   r!   r!   r"   r   w  s    r   c                   @      e Zd ZeZdS )r   N)rC   rD   rE   r   _loop_factoryr!   r!   r!   r"   r     r   r   c                   @   r"  )r   N)rC   rD   rE   r   r#  r!   r!   r!   r"   r     r   r   )2rF   sysplatformImportErrorrX   rO   r   r  r   r   r   r  rp    r   r   r   r   r   r	   r
   r   logr   __all__rz   r  ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr   r   Futurer   rH   r_   re   objectrm   ZBaseSelectorEventLoopr   ZBaseProactorEventLoopr   r   ZBaseSubprocessTransportr   r   ZBaseDefaultEventLoopPolicyr   r   r   r!   r!   r!   r"   <module>   sZ    
0J4;e   `