File: //proc/thread-self/root/usr/local/lib/python3.10/asyncio/__pycache__/windows_events.cpython-310.pyc
o
�ir� � @ s� d Z ddlZejdkred��ddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ ddl
mZ ddl
mZ dd l
mZ dd
l
mZ dd l
mZ dd l
mZ dd
l
mZ ddlmZ dZdZdZdZdZ dZ dZ G dd� dej �Z G dd� dej �Z!G dd� de!�Z"G dd � d e!�Z#G d d � d e$�Z%G d d � d ej&�Z'G d!d"� d"ej(�Z)G d#d$� d$�Z*G d%d&� d&ej+�Z,e'Z-G d'd(� d(ej.�Z/G d)d*� d*ej.�Z0e0Z1dS )+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� g����MbP?g�������?c s` e Zd ZdZdd�� fdd�
Z� fdd�Zdd � Zd� fd
d � Z� fd d
�Z� fdd�Z � Z
S )�_OverlappedFuturez�Subclass of Future which represents an overlapped operation.
Cancelling it will immediately cancel the overlapped operation.
N��loopc s& t � j|d� | jr| jd= || _d S �Nr ���)�super�__init__�_source_traceback�_ov)�self�ovr �� __class__� �3/usr/local/lib/python3.10/asyncio/windows_events.pyr 6 s
z_OverlappedFuture.__init__c sH t � �� }| jd ur"| jjrdnd}|�dd|� d| jjd�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 rd S z| j �� W n& ty4 } zd|| d�}| jr$| j|d< | j�|� W Y d }~nd }~ww 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 Zd ZdZdd�� fdd�
Zdd� Z� fdd �Zd
d � Zd d
� Zd� fdd� Z � fdd�Z
� fdd�Z � Z S )�_BaseWaitHandleFuturez2Subclass of Future which represents a wait handle.Nr 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 c s
z _BaseWaitHandleFuture.__init__c C s t �| jd�t jkS �Nr )�_winapiZWaitForSingleObjectrI Z
WAIT_OBJECT_0�r r! r! r" �_pollq s �z_BaseWaitHandleFuture._pollc sd t � �� }|�d| jd��� | jd ur!| �� rdnd}|�|� | jd ur0|�d| jd��� |S )Nzhandle=r$ �signaledZwaitingz 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 sd S d| _ | j}d | _zt�|� W n3 tyH } z'|jtjkr>d|| d�}| jr1| j|d< | j� |� W Y d }~d S W Y d }~nd }~ww | �
d � d S �NFz$Failed to unregister the wait handler, r0 ) rK rJ � _overlappedZUnregisterWaitr2 �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 Zd ZdZdd�� fdd�
Zdd� Z� fdd �Z� fd
d �Z� ZS ) �_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 � �|� | jd ur| �| � 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 Zd Zdd�� fdd�
Z� fdd�Zdd� Z� ZS ) �_WaitHandleFutureNr c s< t � j||||d� || _d| _t�d ddd �| _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 urt�| 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 sd S d| _ | j}d | _z t�|| j� W n3 tyJ } z'|jtjkr@d|| d�}| jr3| j|d< | j �
|� W Y d }~d S W Y d }~nd }~ww | j � | j| j
�| _d S rW )rK rJ rX ZUnregisterWaitExrg 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 Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � ZeZ 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
zPipeServer.__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 | � � rd S tjtjB }|r|tjO }t�| j|tjtjB tj B tj
t j t j tj
tj�}t �|�}| j�|� |S r= )�closedrO ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPipero ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr �BUFSIZEZNMPWAIT_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 �
zPipeServer.closedc C sV | j d ur
| j �� d | _ | jd ur)| jD ]}|�� qd | _d | _| j�� d S d S r= )rt r1 ro rr �closers �clear)r r� r! r! r" r� " s
�zPipeServer.closeN)
rC rD rE rF r rw ru rx r� �__del__r! r! r! r" rm � s
rm c @ s e Zd ZdZdS )�_WindowsSelectorEventLoopz'Windows version of selector event loop.N)rC rD rE rF r! r! r! r" r� 1 � r� c sJ e Zd ZdZd
� fdd� Z� fdd�Zdd� Zd d
� Z d
d d �Z� Z S )r z2Windows version of proactor event loop using IOCP.Nc s |d u rt � }t� �|� d S r= )r r r )r ri r r! r" r 8 s zProactorEventLoop.__init__c s� z2| j d u sJ �| �| j� t� �� W | j d ur1| j j}| j �� |d ur,| j�|� d | _ d S d S | j d urN| j j}| j �� |d urK| j�|� d | _ w r= ) Z_self_reading_future� call_soonZ_loop_self_readingr � run_foreverr r1 rf rk �r r r r! r" r� = s
�
z ProactorEventLoop.run_foreverc � s: �| j �|�}|I d H }|� }| j||d|id�}||fS )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� ����fdd� ����� �gS )Nc
sT d }z7| r'| � � }�j�|� ��� r|�� W d S �� }�j||d� id� ��� }|d u r2W d S �j�|�} W nf t yT |rL|�
� dkrL|�� �� �� Y d S t y� } z-|rq|�
� dkrq��
d||d�� |�� n �jr|tjd|dd� �� �� W Y d }~d S d }~w tjy� |r�|�� Y d S Y d S w | �_| ��� d S ) Nr� r� r zPipe accept failed)r- r. r� z Accept pipe failed on pipe %rT)�exc_info)rB rr �discardrx r� r� rw rf � accept_pipe�BrokenPipeError�filenor� r2 r4 Z_debugr �warningr �CancelledErrorrt �add_done_callback)r� r� r� r5 �r( �loop_accept_piper� r Zserverr! r" r� [ sR
� �
����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 ttfy) � ty; | �� | �� I d H � w )N)�waiterr� )�
create_future�_WindowsSubprocessTransport�
SystemExit�KeyboardInterrupt�
BaseExceptionr� �_wait) r r� �args�shell�stdin�stdout�stderr�bufsizer� �kwargsr� Ztranspr! 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 5�r c @ s� e Zd ZdZd;dd�Zdd� Zdd� Zd d
� Zd<d d
�Zdd� Z d=dd�Z
d=dd�Z d=dd�Z d>dd�Z
d=dd�Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd<d%d&�Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd<d3d4�Zd5d6� Z d7d8� Z d9d:� Z d S )?r z#Proactor implementation using IOCP.r c C sD d | _ g | _t�tjtd|�| _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
�
zIocpProactor.__init__c C s | j d u r td��d S )NzIocpProactor is closed)r� rb rP r! r! r" �
_check_closed� s
�zIocpProactor._check_closedc C sF dt | j� dt | j� g}| jd u r|�d� d| jjd�|�f S )Nzoverlapped#=%sz
result#=%srx z<%s %s>� )�lenr� r� r� rS r rC �join)r r* r! r! r" �__repr__� s �
zIocpProactor.__repr__c C s
|| _ d S r= )r3 )r r r! r! r" �set_loop� r� zIocpProactor.set_loopNc C s. | j s| �|� | j }g | _ z|W d }S d }w r= )r� rQ )r �timeoutrv r! r! r" �select� s
zIocpProactor.selectc C s | j �� }|�|� |S r= )r3 r� r@ )r �valuerU r! r! r" �_result� �
zIocpProactor._resultr c C �| | � |� t�t�}zt|tj�r|�|�� ||� n|�|�� |� W n t y2 | �
d� Y S w dd� }| � |||�S )N� c
S �B z|� � W S ty } z|jtjtjfv rt|j� �� d }~ww r= �� getresultr2 rY rX ZERROR_NETNAME_DELETEDZERROR_OPERATION_ABORTED�ConnectionResetErrorr� �r� �keyr r5 r! r! r" � finish_recv� �
�
��z&IocpProactor.recv.<locals>.finish_recv) �_register_with_iocprX �
Overlappedrz �
isinstance�socketZWSARecvr� ZReadFiler� r� � _register�r �conn�nbytesr~ r r� r! r! r" �recv� �
� �
zIocpProactor.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� zIocpProactor.recv_intoc C s^ | � |� t�t�}z |�|�� ||� W n ty# | �d� Y S w dd� }| �|||�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
�
zIocpProactor.recvfromc C s> | � |� t�t�}|�|�� |||� dd� }| �|||�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
zIocpProactor.sendtoc C sZ | � |� t�t�}t|tj�r|�|�� ||� n|�|�� |� dd� }| � |||�S )Nc
S r� r= r� r� r! r! r" r� & r� z&IocpProactor.send.<locals>.finish_send)
r� rX r� rz r� r� ZWSASendr� Z WriteFiler� )r r� r� r~ r r� r! r! r" �send s
zIocpProactor.sendc sv | � �� | ��j�� t�t�}|���� � �� � � �fdd�}dd� }| �|�|�}||� �}t j
|| j d� |S )Nc sD |� � t�d��� �}� �tjtj|� � � ��
� � � � � � fS )Nz@P) r� �struct�packr� �
setsockoptr� �
SOL_SOCKETrX ZSO_UPDATE_ACCEPT_CONTEXT�
settimeout�
gettimeout� getpeername)r� r� r r� �r� �listenerr! r" �
finish_accept8 s � z*IocpProactor.accept.<locals>.finish_acceptc � s. �z| I d H W d S t jy |�� � w r= )r r� r� )r/ r� r! r! r" � accept_coroA s ��z(IocpProactor.accept.<locals>.accept_coror ) r� �_get_accept_socket�familyrX r� rz ZAcceptExr� r� r
Z
ensure_futurer3 )r r� r r� r� r/ �coror! r� r" �accept2 s
zIocpProactor.acceptc
s� � j tjkrt�� �� |� | j�� }|�d � |S | � � � z t�
� �� � j � W n" t yL } z|j
tjkr9� � �� d dkrB� W Y d }~nd }~ww t�t�}|�� �� |� � fdd�}| �|� |�S )Nr r c s |� � � �tjtjd� � S rN )r� r� r� r� rX ZSO_UPDATE_CONNECT_CONTEXT�r� r� r �r� r! r" �finish_connectd 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" �connectN s(
���
zIocpProactor.connectc C sb | � |� t�t�}|d@ }|d? d@ }|�|�� t�|�� �|||dd� dd� }| �|||�S )Nr � r c
S r� r= r� r� r! r! r" �finish_sendfilew 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" �sendfilem s
� zIocpProactor.sendfilec sJ | � � � t�t�}|�� �� �}|r| �� �S � fdd�}| �|� |�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 ZConnectNamedPiper� r� r� )r r� r Z connectedr� r! r� r" r� � s
zIocpProactor.accept_pipec
� st �t } zt�|�}W n) ty$ } z
|jtjkr� W Y d }~nd }~ww t|d t�}t� |�I d H qt
� |�S )NT� ) �CONNECT_PIPE_INIT_DELAYrX Z ConnectPiper2 rY ZERROR_PIPE_BUSY�min�CONNECT_PIPE_MAX_DELAYr
�sleepr r{ )r r( �delayrL r5 r! r! r" r� � s �
����
zIocpProactor.connect_pipec C s | � ||d�S )z�Wait 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 zIocpProactor._wait_cancelc s� | � � |d u r tj}nt�|d �}t�t�}t�|| j |j
|�}|r.t |||| j d�� n
t
|||| | j d�� � jr?� jd= � fdd�}� |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 ZRegisterWaitWithQueuer� 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 vr| j �|� t�|�� | jdd� 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|| jd�}|jr|jd= |js:z|d d |�}W n ty4 } z |�|� W Y d }~n
d }~ww |�|� ||||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 ��
zIocpProactor._registerc C s | � � | j�|� dS )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 zIocpProactor._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 rt }n|dk rtd��t�|d �}|t kr td�� t�| j|�}|d u r+n�d}|\}}}}z | j�|�\}} }
} W n) t yh | j
� � rZ| j
� dd||||f d�� |dtj
fvrft�|� Y q w |
| jv rs|�� n=|�� s�z5z| ||| �} W n ty� }
z|�|
� | j�|� W Y d }
~
nd }
~
ww |�| � | j�|� W d }nd }w q | jD ]
} | j�| jd � q�| j�� d S ) Nr znegative timeoutr ztimeout too bigTz8GetQueuedCompletionStatus() returned an unexpected eventz)err=%s transferred=%s key=%#x address=%#x)r- �status)r �
ValueErrorr r rX ZGetQueuedCompletionStatusr� 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 sZ
��
�
��
� �
)zIocpProactor._pollc C s | j �|� d S r= )r� r| r r! r! r" �
_stop_servingI s zIocpProactor._stop_servingc C s | j d u rd S t| j�� �D ]D\}\}}}}|�� rqt|t�r!qz|�� W q tyR } z | j d urHd||d�}|j
rB|j
|d< | j � |� W Y d }~qd }~ww d}t �
� } | | }
| jr|
t �
� krwt�d| t �
� | � t �
� | }
| �|� | js`g | _t�| j � d | _ d S )NzCancelling 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_timeZnext_msgr! r! r" r� O sD
�
�� �
�
zIocpProactor.closec C s | � � d S r= )r� rP r! r! r" r� ~ s zIocpProactor.__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:
"
9 /r c @ s e Zd Zdd� ZdS )r� c
sP t j|f|||||d�|��� _� fdd�}� jj�t� jj��} | �|� 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" �_start� s ��� z"_WindowsSubprocessTransport._startN)rC rD rE r! r! r! r! r" r� � s r� c @ � e Zd ZeZdS )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 ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr� r� �Futurer rH r_ re �objectrm ZBaseSelectorEventLoopr� ZBaseProactorEventLoopr r ZBaseSubprocessTransportr� r
ZBaseDefaultEventLoopPolicyr r r r! r! r! r" �<module> sZ
0J4;j f