OwlCyberSecurity - MANAGER
Edit File: windows_events.cpython-37.pyc
B � fv � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddl m Z ddl mZ ddl mZ dd l mZ dd l mZ dd lmZ d ZdZd ZdZdZdZdZG dd� de j�Z G dd� de j�Z G dd� de �Z G dd� de �Z G dd� de �Z!G d d � d ej"�Z#G d d � d ej$�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-dS )(z.Selector and proactor event loops for Windows.� N� )�events)�base_subprocess)�futures)�proactor_events)�selector_events)�tasks)� windows_utils)�logger)�SelectorEventLoop�ProactorEventLoop� IocpProactor�DefaultEventLoopPolicy� WindowsSelectorEventLoopPolicy� WindowsProactorEventLoopPolicyl �� i� i� g����MbP?g�������?c s^ e Zd ZdZdd�� fdd� Z� fdd�Zdd � Z� 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 )N)r ���)�super�__init__�_source_traceback�_ov)�self�ovr )� __class__� �;/opt/alt/python37/lib64/python3.7/asyncio/windows_events.pyr 0 s z_OverlappedFuture.__init__c sH t � �� }| jd k rD| jjr dnd}|�dd|� d| jjd�d�� |S )N�pendingZ completedr z overlapped=<z, z#x�>)r � _repr_infor r �insert�address)r �info�state)r r r r 6 s z _OverlappedFuture._repr_infoc C sr | j d krd S y| j �� W nJ tk rf } z,d|| d�}| jrJ| j|d< | j�|� W d d }~X Y nX 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 s | � � t� �� S )N)r. r r( )r )r r r r( M s z_OverlappedFuture.cancelc s t � �|� | �� d S )N)r � set_exceptionr. )r r% )r r r r/ Q s z _OverlappedFuture.set_exceptionc s t � �|� d | _d S )N)r � set_resultr )r �result)r r r r0 U s z _OverlappedFuture.set_result) �__name__� __module__� __qualname__�__doc__r r r. r( r/ r0 � __classcell__r r )r r r * s r c sn e Zd ZdZdd�� fdd� Zdd� Z� fdd �Zd d � Zd d � Z� fdd�Z � fdd�Z � fdd�Z � Z S )�_BaseWaitHandleFuturez2Subclass of Future which represents a wait handle.N)r c s8 t � j|d� | jr | jd= || _|| _|| _d| _d S )N)r 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 �| jd�t jkS )Nr )�_winapiZWaitForSingleObjectr8 Z WAIT_OBJECT_0)r r r r �_pollk s z_BaseWaitHandleFuture._pollc sd t � �� }|�d| jd��� | jd k rB| �� r4dnd}|�|� | jd k r`|�d| jd��� |S )Nzhandle=z#xZsignaledZwaitingz wait_handle=)r r �appendr8 r> r9 )r r" r# )r r r r p s z _BaseWaitHandleFuture._repr_infoc C s d | _ d S )N)r )r �futr r r �_unregister_wait_cbz s z)_BaseWaitHandleFuture._unregister_wait_cbc C s� | j s d S d| _ | j}d | _yt�|� W nZ tk r� } z<|jtjkrtd|| d�}| jrd| j|d< | j� |� d S W d d }~X Y nX | � d � d S )NFz$Failed to unregister the wait handle)r$ r% r& r' ) r: r9 � _overlappedZUnregisterWaitr) �winerror�ERROR_IO_PENDINGr r* r+ rA )r r< r, r- r r r �_unregister_wait s" z&_BaseWaitHandleFuture._unregister_waitc s | � � t� �� S )N)rE r r( )r )r r r r( � s z _BaseWaitHandleFuture.cancelc s | � � t� �|� d S )N)rE r r/ )r r% )r r r r/ � s z#_BaseWaitHandleFuture.set_exceptionc s | � � t� �|� d S )N)rE r r0 )r r1 )r r r r0 � s z _BaseWaitHandleFuture.set_result) r2 r3 r4 r5 r r> r rA rE r( r/ r0 r6 r r )r r r7 Z s r7 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. N)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 d��d S )Nz'_WaitCancelFuture must not be cancelled)� RuntimeError)r r r r r( � s z_WaitCancelFuture.cancelc s$ t � �|� | jd k r | �| � d S )N)r r0 rG )r r1 )r r r r0 � s z _WaitCancelFuture.set_resultc s$ t � �|� | jd k r | �| � d S )N)r r/ rG )r r% )r r r r/ � s z _WaitCancelFuture.set_exception) r2 r3 r4 r5 r r( r0 r/ r6 r r )r r rF � s rF c s6 e Zd Zdd�� fdd� Z� fdd�Zdd� Z� ZS ) �_WaitHandleFutureN)r c s< t � j||||d� || _d| _t�d ddd �| _d | _d S )N)r TF)r r � _proactorZ_unregister_proactorrB Z CreateEvent�_event� _event_fut)r r r; r< �proactorr )r r r r � s z_WaitHandleFuture.__init__c sF | j d k r"t�| j � d | _ d | _| j�| j� d | _t� �|� d S )N) rL r= � CloseHandlerM rK � _unregisterr r rA )r r@ )r r r rA � s z%_WaitHandleFuture._unregister_wait_cbc C s� | j s d S d| _ | j}d | _yt�|| j� W nZ tk r� } z<|jtjkrxd|| d�}| jrh| j|d< | j � |� d S W d d }~X Y nX | j � | j| j �| _d S )NFz$Failed to unregister the wait handle)r$ r% r& r' )r: r9 rB ZUnregisterWaitExrL r) rC rD r r* r+ rK � _wait_cancelrA rM )r r< r, r- r r r rE � s$ z"_WaitHandleFuture._unregister_wait)r2 r3 r4 r rA rE r6 r r )r r rJ � s rJ 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)rW rY )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 )N)�closedr= ZPIPE_ACCESS_DUPLEXZFILE_FLAG_OVERLAPPEDZ FILE_FLAG_FIRST_PIPE_INSTANCEZCreateNamedPiperS ZPIPE_TYPE_MESSAGEZPIPE_READMODE_MESSAGEZ PIPE_WAITZPIPE_UNLIMITED_INSTANCESr ZBUFSIZEZNMPWAIT_WAIT_FOREVER�NULL� PipeHandlerV �add)r �first�flags�h�piper r r rY s z PipeServer._server_pipe_handlec C s | j d kS )N)rS )r r r r r\ s zPipeServer.closedc C sV | j d k r| j �� d | _ | jd k rRx| jD ] }|�� q,W d | _d | _| j�� d S )N)rX r( rS rV �closerW �clear)r rc r r r rd s zPipeServer.closeN) r2 r3 r4 r5 r r[ rY r\ rd �__del__r r r r rR � s rR c @ s e Zd ZdZdS )�_WindowsSelectorEventLoopz'Windows version of selector event loop.N)r2 r3 r4 r5 r r r r rg + s rg c s< e Zd ZdZd � fdd� Zdd� Zdd� Zd d d �Z� ZS ) r z2Windows version of proactor event loop using IOCP.Nc s |d krt � }t� �|� d S )N)r r r )r rN )r r r r 2 s zProactorEventLoop.__init__c � s8 | j �|�}|I d H }|� }| j||d|id�}||fS )N�addr)�extra)rK � connect_pipe�_make_duplex_pipe_transport)r �protocol_factoryr! �frc �protocol�transr r r �create_pipe_connection7 s z(ProactorEventLoop.create_pipe_connectionc � s. t � ��d� ����fdd� ����� �gS )Nc s d }yj| rL| � � }�j�|� ��� r2|�� d S �� }�j||d� id� ��� }|d kr`d S �j�|�} W n� t k r� } zF|r�|� � dkr��� d||d�� |�� n�j r�t jd|dd� W d d }~X Y n2 tjk r� |r�|�� Y nX | �_| ��� d S ) Nrh )ri r zPipe accept failed)r$ r% rc z Accept pipe failed on pipe %rT)�exc_info)r1 rV �discardr\ rd rk r[ rK � accept_piper) �filenor+ Z_debugr Zwarningr �CancelledErrorrX �add_done_callback)rm rc rn r, )r! �loop_accept_piperl r �serverr r rw B s< z>ProactorEventLoop.start_serving_pipe.<locals>.loop_accept_pipe)N)rR Z call_soon)r rl r! r )r! rw rl r rx r �start_serving_pipe? s ( z$ProactorEventLoop.start_serving_pipec � sl | � � } t| |||||||f| |d�| ��} y| I d H W n, tk rf | �� | �� I d H � Y nX | S )N)�waiterri )� create_future�_WindowsSubprocessTransport� Exceptionrd Z_wait) r rn �args�shell�stdin�stdout�stderr�bufsizeri �kwargsrz Ztranspr r r �_make_subprocess_transportm s z,ProactorEventLoop._make_subprocess_transport)N)N) r2 r3 r4 r5 r rp ry r� r6 r r )r r r / s 0r c @ s� e Zd ZdZd7dd�Zdd� Zdd� Zd d � Zd8d d �Zdd� Z d9dd�Z d:dd�Z d;dd�Z dd� Z dd� Zdd � Zd d � Zd d � Zd<d!d"�Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd=d/d0�Zd1d2� Zd3d4� Zd5d6� Z d S )>r z#Proactor implementation using IOCP.� �� c C sD d | _ g | _t�tjtd|�| _i | _t� � | _ g | _ t� � | _ d S )Nr ) r* �_resultsrB �CreateIoCompletionPort�INVALID_HANDLE_VALUEr] �_iocp�_cacherT rU r: � _unregistered�_stopped_serving)r Z concurrencyr r r r � s zIocpProactor.__init__c C s | j d krtd��d S )NzIocpProactor is closed)r� rI )r r r r � _check_closed� s zIocpProactor._check_closedc C sF dt | j� dt | j� g}| jd kr0|�d� d| jjd�|�f S )Nzoverlapped#=%sz result#=%sr\ z<%s %s>� )�lenr� r� r� r? r r2 �join)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 }g | _ |S )N)r� r> )r �timeoutrZ r r r �select� s zIocpProactor.selectc C s | j �� }|�|� |S )N)r* r{ r0 )r �valuer@ r r r �_result� s zIocpProactor._resultr c C sz | � |� t�t�}y4t|tj�r6|�|�� ||� n|�|�� |� W n t k rb | � d�S X dd� }| � |||�S )N� c S sP y|� � S tk rJ } z$|jtjtjfkr8t|j� �n� W d d }~X Y nX d S )N)� getresultr) rC rB �ERROR_NETNAME_DELETED�ERROR_OPERATION_ABORTED�ConnectionResetErrorr~ )ro �keyr r, r r r � finish_recv� s z&IocpProactor.recv.<locals>.finish_recv) �_register_with_iocprB � Overlappedr] � isinstance�socketZWSARecvrt ZReadFile�BrokenPipeErrorr� � _register)r �conn�nbytesra r r� r r r �recv� s zIocpProactor.recvc C sz | � |� t�t�}y4t|tj�r6|�|�� ||� n|�|�� |� W n t k rb | � d�S X dd� }| � |||�S )Nr� c S sP y|� � S tk rJ } z$|jtjtjfkr8t|j� �n� W d d }~X Y nX d S )N)r� r) rC rB r� r� r� r~ )ro r� r r, r r r r� � s z+IocpProactor.recv_into.<locals>.finish_recv) r� rB r� r] r� r� Z WSARecvIntort Z ReadFileIntor� r� r� )r r� �bufra r r� r r r � recv_into� s zIocpProactor.recv_intoc C sZ | � |� t�t�}t|tj�r4|�|�� ||� n|�|�� |� dd� }| � |||�S )Nc S sP y|� � S tk rJ } z$|jtjtjfkr8t|j� �n� W d d }~X Y nX d S )N)r� r) rC rB r� r� r� r~ )ro r� r r, r r r � finish_send� s z&IocpProactor.send.<locals>.finish_send) r� rB r� r] r� r� ZWSASendrt Z WriteFiler� )r r� r� ra 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� �structZpackrt � setsockoptr� � SOL_SOCKETrB ZSO_UPDATE_ACCEPT_CONTEXT� settimeoutZ gettimeoutZ getpeername)ro r� r r� )r� �listenerr r � finish_accept� s z*IocpProactor.accept.<locals>.finish_acceptc � s4 y| I d H W n t jk r. |�� � Y nX d S )N)r ru rd )r& r� r r r � accept_coro� s z(IocpProactor.accept.<locals>.accept_coro)r ) r� �_get_accept_socket�familyrB r� r] ZAcceptExrt r� r Z ensure_futurer* )r r� r r� r� r& �coror )r� r� r �accept� s zIocpProactor.acceptc s� | � � � yt�� �� � j� W nB tk rb } z$|jtjkr@� � � � d dkrR� W d d }~X Y nX t� t �}|� � �� |� � fdd�}| � |� |�S )Nr r c s |� � � �tjtjd� � S )Nr )r� r� r� r� rB ZSO_UPDATE_CONNECT_CONTEXT)ro r� r )r� r r �finish_connect s z,IocpProactor.connect.<locals>.finish_connect)r� rB Z BindLocalrt r� r) rC �errnoZ WSAEINVALZ getsocknamer� r] Z ConnectExr� )r r� r! �er r� r )r� r �connect s zIocpProactor.connectc C sb | � |� t�t�}|d@ }|d? d@ }|�|�� t�|�� �|||dd� dd� }| �|||�S )Nl �� � r c S sP y|� � S tk rJ } z$|jtjtjfkr8t|j� �n� W d d }~X Y nX d S )N)r� r) rC rB r� r� r� r~ )ro r� r r, r r r �finish_sendfile% s z.IocpProactor.sendfile.<locals>.finish_sendfile) r� rB r� r] Z TransmitFilert �msvcrtZ get_osfhandler� ) r Zsock�file�offset�countr Z offset_lowZ offset_highr� r r r �sendfile s zIocpProactor.sendfilec sJ | � � � t�t�}|�� �� �}|r0| �� �S � fdd�}| �|� |�S )Nc s |� � � S )N)r� )ro r� r )rc r r �finish_accept_pipe; s z4IocpProactor.accept_pipe.<locals>.finish_accept_pipe)r� rB r� r] ZConnectNamedPipert r� r� )r rc r Z connectedr� r )rc r rs 0 s zIocpProactor.accept_pipec � sz t }xjyt�|�}P W n0 tk rF } z|jtjkr6� W d d }~X Y nX t|d t�}tj || j d�I d H qW t � |�S )N� )r ) �CONNECT_PIPE_INIT_DELAYrB Z ConnectPiper) rC ZERROR_PIPE_BUSY�min�CONNECT_PIPE_MAX_DELAYr Zsleepr* r r^ )r r! Zdelayr; r, r r r rj A 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 r; r� r r r �wait_for_handleT s z IocpProactor.wait_for_handlec C s | � |d d�}||_|S )NT)r� rG )r rH Z done_callbackr@ r r r rQ \ s zIocpProactor._wait_cancelc s� | � � |d krtj}nt�|d �}t�t�}t�|| j |j |�}|r\t |||| j d�� nt |||| | j d�� � jr~� jd= � fdd�}� |d|f| j|j < � S )Ng @�@)r r c s � � � S )N)r> )ro r� r )rm r r �finish_wait_for_handley s z=IocpProactor._wait_for_handle.<locals>.finish_wait_for_handler )r� r= �INFINITE�math�ceilrB r� r] ZRegisterWaitWithQueuer� r! rF r* rJ r r� )r r; r� Z _is_cancel�msr r< r� r )rm r r� c s z IocpProactor._wait_for_handlec C s0 || j kr,| j �|� t�|�� | jdd� d S )Nr )r: r_ rB r� rt r� )r �objr r r r� � s z IocpProactor._register_with_iocpc C s� | � � t|| jd�}|jr$|jd= |jsry|d d |�}W n, tk rf } z|�|� W d d }~X Y n X |�|� ||||f| j|j < |S )N)r r ) r� r r* r r r) r/ r0 r� r! )r r r� �callbackrm 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� r? )r r r r r rP � s zIocpProactor._unregisterc C s t � |�}|�d� |S )Nr )r� r� )r r� �sr r r r� � s z IocpProactor._get_accept_socketc C s� |d krt }n0|dk r td��n t�|d �}|t kr>td���xt�| j|�}|d krZP d}|\}}}}y| j�|�\}} } } W nV t k r� | j � � r�| j � dd||||f d�� |dtj fkr�t�|� wBY nX | | jkr�|�� qB|�� sBy| ||| �} W n: tk �r@ } z|�| � | j�|� W d d } ~ X Y qBX |�| � | j�|� qBW x | jD ]} | j�| jd � �qdW | j�� 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� rB ZGetQueuedCompletionStatusr� r� �pop�KeyErrorr* Z get_debugr+ r� r= rO r� r( Zdoner) r/ r� r? r0 r� r! re )r r� r� r� �errZ transferredr� r! rm r r� r� r� r� r r r r> � sJ zIocpProactor._pollc C s | j �|� d S )N)r� r_ )r r� r r r � _stop_serving� s zIocpProactor._stop_servingc C s | j d krd S x�t| j�� �D ]�\}\}}}}|�� r8q t|t�rDq y |�� W q tk r� } z6| j d k r�d||d�}|j r�|j |d< | j � |� W d d }~X Y q X q W d}t � � } | | } xD| j�r| t � � kr�t�d| t � � | � t � � | } | �|� q�W g | _t�| j � d | _ d S )NzCancelling a future failed)r$ r% r& r' g �?z,%r is running after closing for %.1f seconds)r� �listr� �itemsZ cancelledr� rF r( r) r* r r+ �timeZ monotonicr �debugr> r� r= rO ) r r! r@ r r� r� r, r- Z msg_updateZ start_timeZnext_msgr r r rd � s: " zIocpProactor.closec C s | � � d S )N)rd )r r r r rf * s zIocpProactor.__del__)r� )N)r )r )r )N)N) r2 r3 r4 r5 r r� r� r� r� r� r� r� r� r� r� r� rs rj r� rQ r� r� r� rP r� r> r� rd rf r r r r r s4 " 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 )N)�_procZpollZ_process_exited)rm � returncode)r r r r� 5 s z4_WindowsSubprocessTransport._start.<locals>.callback) r �Popenr� r* rK r� �intr8 rv ) r r~ r r� r� r� r� r� r� rm r )r r �_start0 s z"_WindowsSubprocessTransport._startN)r2 r3 r4 r� r r r r r| . s r| c @ s e Zd ZeZdS )r N)r2 r3 r4 r � _loop_factoryr r r r r @ s r c @ s e Zd ZeZdS )r N)r2 r3 r4 r r� r r r r r D s r ).r5 rB r= r� r� r� r� r� r� rT � r r r r r r r �logr �__all__r] r� ZERROR_CONNECTION_REFUSEDZERROR_CONNECTION_ABORTEDr� r� ZFuturer r7 rF rJ �objectrR ZBaseSelectorEventLooprg ZBaseProactorEventLoopr r ZBaseSubprocessTransportr| r ZBaseDefaultEventLoopPolicyr r r r r r r �<module> sP 0J4;P 2