JezK
Edit File: windows_events.cpython-39.pyc
a �i � � @ s� d Z ddlZejdkr ed��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__� �./usr/lib64/python3.9/asyncio/windows_events.pyr 6 s z_OverlappedFuture.__init__c sH t � �� }| jd urD| jjr dnd}|�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 nL tyh } z4d|| d�}| jrH| j|d< | j�|� W Y d }~n d }~0 0 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 s | � � t� j|d�S �N)�msg)r7 r r1 �r r9 r r! r" r1 S 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_result)N) �__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 )�_winapiZWaitForSingleObjectrG Z WAIT_OBJECT_0�r r! r! r" �_pollq s �z_BaseWaitHandleFuture._pollc sd t � �� }|�d| jd��� | jd urB| �� r4dnd}|�|� | jd ur`|�d| jd��� |S )Nzhandle=r$ �signaledZwaitingz wait_handle=)r r& �appendrG rO rH 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 | _zt�|� W nf ty� } zN|jtjkr|d|| d�}| jrb| j|d< | j� |� W Y d }~d S W Y d }~n d }~0 0 | � d � d S �NFz$Failed to unregister the wait handler, r0 ) rI rH � _overlappedZUnregisterWaitr2 �winerror�ERROR_IO_PENDINGr r3 r4 rT �r rK r5 r6 r! r! r" �_unregister_wait� s$ � $z&_BaseWaitHandleFuture._unregister_waitc s | � � t� j|d�S r8 )rZ r r1 r: r r! r" r1 � s z _BaseWaitHandleFuture.cancelc s | � � t� �|� d S r; )rZ r r< r= r r! r" r< � s z#_BaseWaitHandleFuture.set_exceptionc s | � � t� �|� d S r; )rZ r r> r? r r! r" r> � s z _BaseWaitHandleFuture.set_result)N) rA rB rC rD r rO r& rT rZ r1 r< r> rE r! r! r r" rF ` s rF 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 �eventrK r r r! r" r � s z_WaitCancelFuture.__init__c C s t d��d S )Nz'_WaitCancelFuture must not be cancelled)� RuntimeErrorrN r! r! r" r1 � s z_WaitCancelFuture.cancelc s$ t � �|� | jd ur | �| � d S r; )r r> r\ r? r r! r" r> � s z _WaitCancelFuture.set_resultc s$ t � �|� | jd ur | �| � d S r; )r r< r\ r= r r! r" r< � s z _WaitCancelFuture.set_exception) rA rB rC rD r r1 r> r< rE 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_proactorrV Z CreateEvent�_event� _event_fut)r r rJ rK �proactorr r r! r" r � s z_WaitHandleFuture.__init__c sF | j d ur"t�| j � d | _ d | _| j�| j� d | _t� �|� d S r; ) ra rM � CloseHandlerb r` � _unregisterr r rT rR r r! r" rT � s z%_WaitHandleFuture._unregister_wait_cbc C s� | j s d S d| _ | j}d | _zt�|| j� W nf ty� } zN|jtjkr�d|| d�}| jrf| j|d< | j � |� W Y d }~d S W Y d }~n d }~0 0 | j � | j| j �| _d S rU )rI rH rV ZUnregisterWaitExra r2 rW rX r r3 r4 r` � _wait_cancelrT rb rY r! r! r" rZ � s( � $ �z"_WaitHandleFuture._unregister_wait)rA rB rC r rT rZ rE r! r! r r" r_ � s r_ 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)rm ro )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 r; )�closedrM ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPiperi ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr �BUFSIZEZNMPWAIT_WAIT_FOREVER�NULL� PipeHandlerl �add)r �first�flags�h�piper! r! r" ro s" �� z PipeServer._server_pipe_handlec C s | j d u S r; )ri rN r! r! r" rr s zPipeServer.closedc C sR | j d ur| j �� d | _ | jd urN| jD ] }|�� q*d | _d | _| j�� d S r; )rn r1 ri rl �closerm �clear)r rz r! r! r" r{ " s zPipeServer.closeN) rA rB rC rD r rq ro rr r{ �__del__r! r! r! r" rg � s rg c @ s e Zd ZdZdS )�_WindowsSelectorEventLoopz'Windows version of selector event loop.N)rA rB rC rD r! r! r! r" r~ 1 s r~ c sH e Zd ZdZd � fdd� Z� fdd�Zdd� Zd d � Zdd 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 rc r r! r" r 8 s zProactorEventLoop.__init__c s� zd| j d u sJ �| �| j� t� �� W | j d ur�| j j}| j �� |d ur^|js^| j� |� d | _ n>| j d ur�| j j}| j �� |d ur�|js�| j� |� d | _ 0 d S r; ) Z_self_reading_future� call_soonZ_loop_self_readingr � run_foreverr r1 r# r` re �r r r r! r" r� = s � z ProactorEventLoop.run_foreverc � s8 | j �|�}|I d H }|� }| j||d|id�}||fS )N�addr��extra)r` � connect_pipe�_make_duplex_pipe_transport)r �protocol_factoryr( �frz �protocol�transr! r! r" �create_pipe_connectionP s �z(ProactorEventLoop.create_pipe_connectionc � s. t � ��d� ����fdd� ����� �gS )Nc s d }zn| rN| � � }�j�|� ��� r4|�� W d S �� }�j||d� id� ��� }|d u rdW d S �j�|�} W n� t y� } zN|r�|� � dkr��� d||d�� |�� n�j r�t jd|dd� W Y d }~n8d }~0 tjy� |r�|�� Y n0 | �_| ��� d S ) Nr� r� r zPipe accept failed)r- r. rz z Accept pipe failed on pipe %rT)�exc_info)r@ rl �discardrr r{ r� rq r` � accept_piper2 �filenor4 Z_debugr �warningr �CancelledErrorrn �add_done_callback)r� rz r� r5 �r( �loop_accept_piper� r Zserverr! r" r� [ sB �� �z>ProactorEventLoop.start_serving_pipe.<locals>.loop_accept_pipe)N)rg r )r r� r( r! r� r" �start_serving_pipeX s ( z$ProactorEventLoop.start_serving_pipec � s� | � � } t| |||||||f| |d�| ��} z| I d H W n@ ttfyR � Y n* tyz | �� | �� I d H � Y n0 | S )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_transport)N)N) rA rB rC rD r r� r� r� r� rE r! r! r r" r 5 s 0 �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 dAdd�Zdd � Zd d � Zd d � Zd!d"� Zd#d$� ZdBd%d&�Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� ZdCd3d4�Zd5d6� Z d7d8� Z d9d:� Z d S )Dr z#Proactor implementation using IOCP.r c C sD d | _ g | _t�tjtd|�| _i | _t� � | _ g | _ t� � | _ d S rL ) r3 �_resultsrV �CreateIoCompletionPort�INVALID_HANDLE_VALUErt �_iocp�_cacherj rk rI � _unregistered�_stopped_serving)r Z concurrencyr! r! r" r � s � zIocpProactor.__init__c C s | j d u rtd��d S )NzIocpProactor is closed)r� r^ rN r! r! r" � _check_closed� s zIocpProactor._check_closedc C sF dt | j� dt | j� g}| jd u r0|�d� d| jjd�|�f S )Nzoverlapped#=%sz result#=%srr z<%s %s>� )�lenr� r� r� rQ r rA �join)r r* r! r! r" �__repr__� s � zIocpProactor.__repr__c C s || _ d S r; )r3 )r r r! r! r" �set_loop� s zIocpProactor.set_loopNc C s | j s| �|� | j }g | _ |S r; )r� rO )r �timeoutrp r! r! r" �select� s zIocpProactor.selectc C s | j �� }|�|� |S r; )r3 r� r> )r �valuerS r! r! r" �_result� s zIocpProactor._resultr c C s| | � |� t�t�}z4t|tj�r6|�|�� ||� n|�|�� |� W n t yd | � d� Y S 0 dd� }| � |||�S )N� c S sT z |� � W S tyN } z,|jtjtjfv r8t|j� �n� W Y d }~n d }~0 0 d S r; �� getresultr2 rW rV ZERROR_NETNAME_DELETEDZERROR_OPERATION_ABORTED�ConnectionResetErrorr� �r� �keyr r5 r! r! r" � finish_recv� s � z&IocpProactor.recv.<locals>.finish_recv) �_register_with_iocprV � Overlappedrt � isinstance�socketZWSARecvr� ZReadFile�BrokenPipeErrorr� � _register�r �conn�nbytesrx r r� r! r! r" �recv� s zIocpProactor.recvc C s| | � |� t�t�}z4t|tj�r6|�|�� ||� n|�|�� |� W n t yd | � d� Y S 0 dd� }| � |||�S )Nr c S sT z |� � W S tyN } z,|jtjtjfv r8t|j� �n� W Y d }~n d }~0 0 d S r; r� r� r! r! r" r� � s � z+IocpProactor.recv_into.<locals>.finish_recv) r� rV r� rt r� r� Z WSARecvIntor� Z ReadFileIntor� r� r� )r r� �bufrx r r� r! r! r" � recv_into� s zIocpProactor.recv_intoc C s^ | � |� t�t�}z|�|�� ||� W n tyF | �d� Y S 0 dd� }| �|||�S )N)r� Nc S sT z |� � W S tyN } z,|jtjtjfv r8t|j� �n� W Y d }~n d }~0 0 d S r; r� r� r! r! r" r� � s � z*IocpProactor.recvfrom.<locals>.finish_recv) r� rV r� rt Z WSARecvFromr� r� r� r� r� r! r! r" �recvfrom� s zIocpProactor.recvfromc C s> | � |� t�t�}|�|�� |||� dd� }| �|||�S )Nc S sT z |� � W S tyN } z,|jtjtjfv r8t|j� �n� W Y d }~n d }~0 0 d S r; r� r� r! r! r" � finish_send s � z(IocpProactor.sendto.<locals>.finish_send)r� rV r� rt Z WSASendTor� r� )r r� r� rx r� r r� r! r! r" �sendto s zIocpProactor.sendtoc C sZ | � |� t�t�}t|tj�r4|�|�� ||� n|�|�� |� dd� }| � |||�S )Nc S sT z |� � W S tyN } z,|jtjtjfv r8t|j� �n� W Y d }~n d }~0 0 d S r; r� r� r! r! r" r� s � z&IocpProactor.send.<locals>.finish_send) r� rV r� rt r� r� ZWSASendr� Z WriteFiler� )r r� r� rx 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_SOCKETrV ZSO_UPDATE_ACCEPT_CONTEXT� settimeout� gettimeout� getpeername)r� r� r r� �r� �listenerr! r" � finish_accept/ s �z*IocpProactor.accept.<locals>.finish_acceptc � s2 z| I d H W n t jy, |�� � Y n0 d S r; )r r� r{ )r/ r� r! r! r" � accept_coro8 s z(IocpProactor.accept.<locals>.accept_coror ) r� �_get_accept_socket�familyrV r� rt ZAcceptExr� r� r Z ensure_futurer3 )r r� r r� r� r/ �coror! r� r" �accept) s zIocpProactor.acceptc s� � j tjkr4t�� �� |� | j�� }|�d � |S | � � � zt� � �� � j � W nD t y� } z,|j tjkrr� � �� d dkr�� W Y d }~n d }~0 0 t�t�}|�� �� |� � fdd�}| �|� |�S )Nr r c s |� � � �tjtjd� � S rL )r� r� r� r� rV ZSO_UPDATE_CONNECT_CONTEXT�r� r� r �r� r! r" �finish_connect[ s �z,IocpProactor.connect.<locals>.finish_connect)�typer� � SOCK_DGRAMrV Z WSAConnectr� r3 r� r> r� Z BindLocalr� r2 rW �errnoZ WSAEINVAL� getsocknamer� rt Z ConnectExr� )r r� r( rS �er r� r! r� r" �connectE s" zIocpProactor.connectc C sb | � |� t�t�}|d@ }|d? d@ }|�|�� t�|�� �|||dd� dd� }| �|||�S )Nr � r c S sT z |� � W S tyN } z,|jtjtjfv r8t|j� �n� W Y d }~n d }~0 0 d S r; r� r� r! r! r" �finish_sendfilen s � z.IocpProactor.sendfile.<locals>.finish_sendfile) r� rV r� rt Z TransmitFiler� �msvcrtZ get_osfhandler� ) r �sock�file�offset�countr Z offset_lowZ offset_highr� r! r! r" �sendfiled s � zIocpProactor.sendfilec sJ | � � � t�t�}|�� �� �}|r0| �� �S � fdd�}| �|� |�S )Nc s |� � � S r; )r� r� �rz r! r" �finish_accept_pipe� s z4IocpProactor.accept_pipe.<locals>.finish_accept_pipe)r� rV r� rt ZConnectNamedPiper� r� r� )r rz r Z connectedr� r! r� r" r� y s zIocpProactor.accept_pipec � st t }zt�|�}W qjW n2 tyH } z|jtjkr4� W Y d }~n d }~0 0 t|d t�}t� |�I d H qt � |�S )N� ) �CONNECT_PIPE_INIT_DELAYrV Z ConnectPiper2 rW ZERROR_PIPE_BUSY�min�CONNECT_PIPE_MAX_DELAYr �sleepr ru )r r( �delayrJ 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 rJ r� r! r! r" �wait_for_handle� s z IocpProactor.wait_for_handlec C s | � |d d�}||_|S rh )r� r\ )r r] Z done_callbackrS r! r! r" rf � s zIocpProactor._wait_cancelc s� | � � |d u rtj}nt�|d �}t�t�}t�|| j |j |�}|r\t |||| j d�� nt |||| | j d�� � jr~� jd= � fdd�}� |d|f| j|j < � S )N� @�@r r c s � � � S r; )rO r� �r� r! r" �finish_wait_for_handle� s z=IocpProactor._wait_for_handle.<locals>.finish_wait_for_handler )r� rM �INFINITE�math�ceilrV r� rt ZRegisterWaitWithQueuer� r( r[ r3 r_ r r� )r rJ r� Z _is_cancel�msr rK r� r! r� r" r� � s$ � � z IocpProactor._wait_for_handlec C s0 || j vr,| j �|� t�|�� | jdd� d S rL )rI rv rV r� r� r� �r �objr! r! r" r� � s z IocpProactor._register_with_iocpc C s� | � � t|| jd�}|jr$|jd= |jstz|d d |�}W n. tyh } z|�|� W Y d }~nd }~0 0 |�|� ||||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� rQ r� r! r! r" re � s zIocpProactor._unregisterc C s t � |�}|�d� |S rL )r� r� )r r� �sr! r! r" r� s z IocpProactor._get_accept_socketc C s� |d u rt }n0|dk r td��n t�|d �}|t kr>td��t�| j|�}|d u rX�qZd}|\}}}}z| j�|�\}} } } W nV t y� | j � � r�| j � dd||||f d�� |dtj fvr�t�|� Y q>Y n0 | | jv r�|�� q>|�� s>z| ||| �} W n< t�y@ } z"|�| � | j�|� W Y d } ~ q>d } ~ 0 0 |�| � | j�|� q>| jD ]} | j�| jd � �q`| j�� d S )Nr znegative timeoutr� ztimeout too bigz8GetQueuedCompletionStatus() returned an unexpected eventz)err=%s transferred=%s key=%#x address=%#x)r- �status)r� � ValueErrorr� r� rV ZGetQueuedCompletionStatusr� r� �pop�KeyErrorr3 Z get_debugr4 r� rM rd r� r1 �doner2 r< r� rQ r> r� r( r| )r r� r� r �errZ transferredr� r( r� r r� r r� r� r! r! r" rO sL �� " zIocpProactor._pollc C s | j �|� d S r; )r� rv r� r! r! r" � _stop_serving> s zIocpProactor._stop_servingc C s | j d u rd S t| j�� �D ]�\}\}}}}|�� r6q t|t�rBq z |�� W q ty� } z>| j d ur�d||d�}|j r�|j |d< | j � |� W Y d }~q d }~0 0 q d}t � � } | | } | jr�| t � � kr�t�d| t � � | � t � � | } | �|� q�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 �debugrO r� rM rd ) r r( rS r r� r r5 r6 Z msg_updateZ start_timeZnext_msgr! r! r" r{ D s> � $ � zIocpProactor.closec C s | � � d S r; )r{ rN r! r! r" r} s s zIocpProactor.__del__)r )N)r )r )r )r N)r )N)N) rA rB rC rD r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rf r� r� r� re r� rO r r{ r} r! r! r! r" r � s8 " 7/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� � returncoderN r! r" r ~ s z4_WindowsSubprocessTransport._start.<locals>.callback) r �Popenr r3 r` r� �intrG r� ) r r� r� r� r� r� r� r� r r� r! rN r" �_starty s ��� z"_WindowsSubprocessTransport._startN)rA rB rC r r! r! r! r" r� w s r� c @ s e Zd ZeZdS )r N)rA rB rC r � _loop_factoryr! r! r! r" r � s r c @ s e Zd ZeZdS )r N)rA rB rC r r r! r! r! r" r � s r )2rD �sys�platform� ImportErrorrV rM r� r� r� r� r� r rj � r r r r r r r r �logr �__all__rt r� ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr� r� �Futurer rF r[ r_ �objectrg ZBaseSelectorEventLoopr~ ZBaseProactorEventLoopr r ZBaseSubprocessTransportr� r ZBaseDefaultEventLoopPolicyr r r r! r! r! r" �<module> sX 0J4;e `