o
    g                     @   s  d 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	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+Z-e-Z.dS ))z.Selector and proactor event loops for Windows.    N   )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__ L/Users/admin/.pyenv/versions/3.10.0/lib/python3.10/asyncio/windows_events.pyr   1   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%   7   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_overlapped>   s   


	z$_OverlappedFuture._cancel_overlappedc                       |    t j|dS N)msg)r6   r   r0   r   r9   r   r    r!   r0   N      z_OverlappedFuture.cancelc                    s   t  | |   d S N)r   set_exceptionr6   r   r-   r   r    r!   r=   R   s   z_OverlappedFuture.set_exceptionc                    s   t  | d | _d S r<   )r   
set_resultr   r   resultr   r    r!   r?   V   s   
z_OverlappedFuture.set_resultr<   )__name__
__module____qualname____doc__r   r%   r6   r0   r=   r?   __classcell__r    r    r   r!   r   +   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   ^   s   
z_BaseWaitHandleFuture.__init__c                 C   s   t | jdt jkS Nr   )_winapiZWaitForSingleObjectrH   ZWAIT_OBJECT_0r   r    r    r!   _polll   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%   appendrH   rP   rI   r(   r   r    r!   r%   q   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+   r/   )rJ   rI   _overlappedZUnregisterWaitr1   winerrorERROR_IO_PENDINGr   r2   r3   rU   r   rL   r4   r5   r    r    r!   _unregister_wait   s*   
z&_BaseWaitHandleFuture._unregister_waitc                    r7   r8   )r[   r   r0   r:   r   r    r!   r0      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?   r@   r   r    r!   r?      r]   z _BaseWaitHandleFuture.set_resultr<   )rB   rC   rD   rE   r   rP   r%   rU   r[   r0   r=   r?   rF   r    r    r   r!   rG   [   s    
rG   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   eventrL   r   r   r    r!   r      s   
z_WaitCancelFuture.__init__c                 C   s   t d)Nz'_WaitCancelFuture must not be cancelled)RuntimeErrorrO   r    r    r!   r0      s   z_WaitCancelFuture.cancelc                    (   t  | | jd ur| |  d S d S r<   )r   r?   r_   r@   r   r    r!   r?         
z_WaitCancelFuture.set_resultc                    rb   r<   )r   r=   r_   r>   r   r    r!   r=      rc   z_WaitCancelFuture.set_exception)	rB   rC   rD   rE   r   r0   r?   r=   rF   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_proactorrW   ZCreateEvent_event
_event_fut)r   r   rK   rL   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<   )	rf   rN   CloseHandlerg   re   _unregisterr   r   rU   rS   r   r    r!   rU      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 rV   )rJ   rI   rW   ZUnregisterWaitExrf   r1   rX   rY   r   r2   r3   re   _wait_cancelrU   rg   rZ   r    r    r!   r[      s.   


z"_WaitHandleFuture._unregister_wait)rB   rC   rD   r   rU   r[   rF   r    r    r   r!   rd      s    rd   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)rr   rt   )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<   )closedrN   ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZFILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPipern   ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ	PIPE_WAITZPIPE_UNLIMITED_INSTANCESr
   BUFSIZEZNMPWAIT_WAIT_FOREVERNULL
PipeHandlerq   add)r   firstflagshpiper    r    r!   rt     s"   


zPipeServer._server_pipe_handlec                 C   s
   | j d u S r<   )rn   rO   r    r    r!   rw        
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<   )rs   r0   rn   rq   closerr   clear)r   r   r    r    r!   r     s   




zPipeServer.closeN)
rB   rC   rD   rE   r   rv   rt   rw   r   __del__r    r    r    r!   rl      s    
rl   c                   @   s   e Zd ZdZdS )_WindowsSelectorEventLoopz'Windows version of selector event loop.N)rB   rC   rD   rE   r    r    r    r!   r   ,      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   rh   r   r    r!   r   3  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   r0   re   rj   r   r   r   r    r!   r   8  s   



zProactorEventLoop.run_foreverc                    s:   | j |}|I d H }| }| j||d|id}||fS )Naddrextra)re   connect_pipe_make_duplex_pipe_transport)r   protocol_factoryr'   fr   protocoltransr    r    r!   create_pipe_connectionK  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)rA   rq   discardrw   r   r   rv   re   accept_piper1   filenor3   Z_debugr   warningr   CancelledErrorrs   add_done_callback)r   r   r   r4   r'   loop_accept_piper   r   Zserverr    r!   r   V  sL   

z>ProactorEventLoop.start_serving_pipe.<locals>.loop_accept_piper<   )rl   r   )r   r   r'   r    r   r!   start_serving_pipeS  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<   )
rB   rC   rD   rE   r   r   r   r   r   rF   r    r    r   r!   r   0  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 rM   )r2   _resultsrW   CreateIoCompletionPortINVALID_HANDLE_VALUEry   _iocp_cachero   rp   rJ   _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   ra   rO   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#=%srw   z<%s %s> )lenr   r   r   rR   r   rB   join)r   r)   r    r    r!   __repr__  s   

zIocpProactor.__repr__c                 C   s
   || _ d S r<   )r2   )r   r   r    r    r!   set_loop  r   zIocpProactor.set_loopNc                 C   s    | j s| | | j }g | _ |S r<   )r   rP   )r   timeoutru   r    r    r!   select  s
   
zIocpProactor.selectc                 C   s   | j  }|| |S r<   )r2   r   r?   )r   valuerT   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<   	getresultr1   rX   rW   ZERROR_NETNAME_DELETEDZERROR_OPERATION_ABORTEDConnectionResetErrorr   r   keyr   r4   r    r    r!   finish_recv     

z&IocpProactor.recv.<locals>.finish_recv)_register_with_iocprW   
Overlappedry   
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   rW   r   ry   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   rW   r   ry   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   rW   r   ry   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   rW   r   ry   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_SOCKETrW   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_coro3  s   z(IocpProactor.accept.<locals>.accept_coror   )r   _get_accept_socketfamilyrW   r   ry   ZAcceptExr   r   r	   Zensure_futurer2   )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 rM   )r   r   r   r   rW   ZSO_UPDATE_CONNECT_CONTEXTr   r   r   r   r    r!   finish_connectV  s
   z,IocpProactor.connect.<locals>.finish_connect)typer   
SOCK_DGRAMrW   Z
WSAConnectr   r2   r   r?   r   Z	BindLocalr   r1   rX   errnoZ	WSAEINVALgetsocknamer   ry   Z	ConnectExr   )r   r   r'   rT   er   r   r    r   r!   connect@  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_sendfilei  r   z.IocpProactor.sendfile.<locals>.finish_sendfile)	r   rW   r   ry   ZTransmitFiler   msvcrtZget_osfhandler   )	r   sockfileoffsetcountr   Z
offset_lowZoffset_highr   r    r    r!   sendfile_  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   rW   r   ry   ZConnectNamedPiper   r   r   )r   r   r   Z	connectedr   r    r   r!   r   t  s   


zIocpProactor.accept_pipec              
      st   t }	 zt|}W q5 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_DELAYrW   ZConnectPiper1   rX   ZERROR_PIPE_BUSYminCONNECT_PIPE_MAX_DELAYr	   sleepr
   rz   )r   r'   delayrK   r4   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).
        FN)_wait_for_handle)r   rK   r   r    r    r!   wait_for_handle  s   zIocpProactor.wait_for_handlec                 C   s   |  |d d}||_|S rm   )r  r_   )r   r`   Zdone_callbackrT   r    r    r!   rk     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<   )rP   r   r   r    r!   finish_wait_for_handle  s   z=IocpProactor._wait_for_handle.<locals>.finish_wait_for_handler   )r   rN   INFINITEmathceilrW   r   ry   ZRegisterWaitWithQueuer   r'   r^   r2   rd   r   r   )r   rK   r   Z
_is_cancelmsr   rL   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 rM   )rJ   r{   rW   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   r2   r   r"   r1   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   rR   r   r    r    r!   rj     s   zIocpProactor._unregisterc                 C   s   t  |}|d |S rM   )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+qd}|\}}}}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  rW   ZGetQueuedCompletionStatusr   r   popKeyErrorr2   Z	get_debugr3   r   rN   ri   r   r0   doner1   r=   r   rR   r?   r   r'   r   )r   r   r	  r  errZtransferredr   r'   r   r   r  r  r   r   r    r    r!   rP     sV   

	





'zIocpProactor._pollc                 C   s   | j | d S r<   )r   r{   r
  r    r    r!   _stop_serving9  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+   r/   g      ?z,%r is running after closing for %.1f seconds)r   listr   items	cancelledr   r^   r0   r1   r2   r   r3   time	monotonicr   debugrP   r   rN   ri   )r   r'   rT   r   r  r  r4   r5   Z
msg_updateZ
start_timeZnext_msgr    r    r!   r   ?  sD   




	
zIocpProactor.closec                 C   s   |    d S r<   )r   rO   r    r    r!   r   n  s   zIocpProactor.__del__)r   r<   )r   )r   N)rB   rC   rD   rE   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  rk   r  r   r   rj   r   rP   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   
returncoderO   r    r!   r  y  s   
z4_WindowsSubprocessTransport._start.<locals>.callback)	r
   Popenr  r2   re   r  intrH   r   )
r   r   r   r   r   r   r   r   r  r   r    rO   r!   _startt  s   z"_WindowsSubprocessTransport._startN)rB   rC   rD   r   r    r    r    r!   r   r  s    r   c                   @      e Zd ZeZdS )r   N)rB   rC   rD   r   _loop_factoryr    r    r    r!   r     r   r   c                   @   r!  )r   N)rB   rC   rD   r   r"  r    r    r    r!   r     r   r   )/rE   rW   rN   r   r  r   r   r   r  ro    r   r   r   r   r   r   r	   r
   logr   __all__ry   r  ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr   r   Futurer   rG   r^   rd   objectrl   ZBaseSelectorEventLoopr   ZBaseProactorEventLoopr   r   ZBaseSubprocessTransportr   r   ZBaseDefaultEventLoopPolicyr   r   r   r    r    r    r!   <module>   sT    0J4;e   `