CasperSecurity
o
�b�$ � @ sd U d Z ddlZddlZddlZddlmZmZ ddlmZ ddl m
Z
m Z m Z m
Z
ddlmZ ddlmZmZmZ ddlmZ dd lmZ dd
lmZmZ zdd l m Z W n e yr dZ d Z eee j! d
f e"d< e�#d� Y nw e Z e j!fZ dZ$dZ%e&d�\Z'Z(e �)d�Z*e �)d�Z+ee j,e j-e j.e j/ge �R � G dd� de
j0e
j1e��Z2dd� Z3ddgZ4dS )z'
Reactor that uses IO completion ports
� N)�Tuple�Type)� implementer)�base�error�
interfaces�main)�Process)� iocpsupport�tcp�udp)� WAIT_TIMEOUT)�_ThreadedWin32EventsMixin)�failure�log)�TLSMemoryBIOFactory� .�_extraInterfacesz|pyOpenSSL 0.10 or newer is required for SSL support in iocpreactor. It is missing, so the reactor will not support SSL APIs.i� �� � z#Handler has no getFileHandle methodzFiledescriptor went awayc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z d-dd�Z
d.dd�ZedurDd-dd �Z d.d d �Zn d-d d �Z d.d d �Zd/d"d#�Z $d0d%d&�Zd'i dddd(dfd)d*�Zd+d,� ZdS )1� IOCPReactorNc C s" t j�| � t�� | _t� | _d S �N)r � ReactorBase�__init__�_iocp�CompletionPort�port�set�handles��selfr r �F/usr/lib/python3/dist-packages/twisted/internet/iocpreactor/reactor.pyr ? s
zIOCPReactor.__init__c C � | j �|� d S r )r �add�r �handler r r! �addActiveHandleD � zIOCPReactor.addActiveHandlec C r" r )r �discardr$ r r r! �removeActiveHandleG r' z IOCPReactor.removeActiveHandlec C s� d}|du r t }n tt td| ��}| j�|�\}}}} |tkr#dS |tkr<|tks-J �t� |j
| j |||� |d7 }|t krBdS | j�d�\}}}}q )z=
Poll the IO completion port for new events.
r Nr � )
� MAX_TIMEOUT�min�intr �getEventr
�
KEY_WAKEUP�
KEY_NORMALr �callWithLogger�owner�_callEventCallback�EVENTS_PER_LOOP)r �timeout�processed_events�rc�numBytes�key�evtr r r! � doIterationJ s$ ��zIOCPReactor.doIterationc C s� |j }d }z |�|||� t|dd �}|st}n|� dkr t}|r$W d S W n ty8 t�� d }t� � Y nw |rE|�
t � |�� d S d S )N�
getFileHandle���r* )
r2 �callback�getattr�
_NO_GETHANDLE� _NO_FILEDESC�
BaseException�sys�exc_infor �err�loseConnectionr �Failure)r r7 r8 r: r2 �why�handfnr r r! r3 v s&
� ��z IOCPReactor._callEventCallbackc C s d S r r r r r r! � installWaker� s zIOCPReactor.installWakerc C s | j �dtd � d S )Nr )r � postEventr/ r r r r! �wakeUp� s zIOCPReactor.wakeUpc C s | j �|t� d S r )r � addHandler0 r$ r r r! �registerHandle� s zIOCPReactor.registerHandlec C s t � ||�}| �|�� � |S r )�socketrN �fileno)r �af�stype�sktr r r! � createSocket� s zIOCPReactor.createSocket�2 � c C � t �||||| �}|�� |S )zI
@see: twisted.internet.interfaces.IReactorTCP.listenTCP
)r �Port�startListening)r r �factory�backlog� interface�pr r r! � listenTCP� s zIOCPReactor.listenTCP� c C s t �|||||| �}|�� |S )zJ
@see: twisted.internet.interfaces.IReactorTCP.connectTCP
)r � Connector�connect)r �hostr rZ r5 � bindAddress�cr r r! �
connectTCP� s zIOCPReactor.connectTCPc C s" | � |t|d|�||�}d|_|S )zQ
@see: twisted.internet.interfaces.IReactorSSL.listenSSL
F�TLS)r^ r �_type�r r rZ �contextFactoryr[ r\ r r r! � listenSSL� s
�zIOCPReactor.listenSSLc C s | � ||t|d|�||�S )zR
@see: twisted.internet.interfaces.IReactorSSL.connectSSL
T)re r �r rb r rZ ri r5 rc r r r! �
connectSSL� s
�zIOCPReactor.connectSSLc C � t d��)z�
Non-implementation of L{IReactorSSL.listenSSL}. Some dependency
is not satisfied. This implementation always raises
L{NotImplementedError}.
�|pyOpenSSL 0.10 or newer is required for SSL support in iocpreactor. It is missing, so the reactor does not support SSL APIs.��NotImplementedErrorrh r r r! rj � s �c C rm )z�
Non-implementation of L{IReactorSSL.connectSSL}. Some dependency
is not satisfied. This implementation always raises
L{NotImplementedError}.
rn ro rk r r r! rl � s �� c C rW )z�
Connects a given L{DatagramProtocol} to the given numeric UDP port.
@returns: object conforming to L{IListeningPort}.
)r rX rY )r r �protocolr\ �
maxPacketSizer] r r r! � listenUDP� s zIOCPReactor.listenUDPFc C s t �||||| |�}|�� |S )z�
Connects a given DatagramProtocol to the given numeric UDP port.
EXPERIMENTAL.
@returns: object conforming to IListeningPort.
)r �
MulticastPortrY )r r rr r\ rs �listenMultipler] r r r! �listenMulticast� s
�zIOCPReactor.listenMulticastr r c
C sN |durt d��|durt d��|rt d��| dur t d��t| |||||�S )z"
Spawn a process.
Nz,Setting UID is unsupported on this platform.z,Setting GID is unsupported on this platform.z&PTYs are unsupported on this platform.zGCustom child file descriptor mappings are unsupported on this platform.)�
ValueErrorr )
r �processProtocol�
executable�args�env�path�uid�gid�usePTY�childFDsr r r! � spawnProcess� s �zIOCPReactor.spawnProcessc C s t | j�}| j�� |S r )�listr �clear)r �resr r r! � removeAll s
zIOCPReactor.removeAll)rU rV )r_ N)rV rq )rV rq F)�__name__�
__module__� __qualname__r r r& r) r; r3 rJ rL rN rT r^ re r rj rl rt rw r� r� r r r r! r 2 s@ ,
�
�
�
� r c C s t � } t�| � d S r )r r �installReactor)�rr r r! �install s r� )5�__doc__rO rC �warnings�typingr r �zope.interfacer �twisted.internetr r r r � twisted.internet._dumbwin32procr � twisted.internet.iocpreactorr
r r r �"twisted.internet.iocpreactor.constr
�"twisted.internet.win32eventreactorr �twisted.pythonr r �twisted.protocols.tlsr �_TLSMemoryBIOFactory� ImportErrorr � IReactorSSL�__annotations__�warnr+ r4 �ranger0 r/ �ConnectionFdescWentAwayr@ rA � IReactorTCP� IReactorUDP�IReactorMulticast�IReactorProcess�_SignalReactorMixinr r r� �__all__r r r r! �<module> sT ��
��
� c