CasperSecurity
o
�bV � @ s� d Z ddlZddlmZmZmZmZmZmZ m
Z
ddl m Z ddl
mZ ddlmZ ddlmZ e e�G dd � d ejej��Zd
d � Zd d gZdS ) z�
A poll() based implementation of the twisted main loop.
To install the event loop (and you should do this before any connections,
listeners or connectors are added)::
from twisted.internet import pollreactor
pollreactor.install()
� N)�POLLERR�POLLHUP�POLLIN�POLLNVAL�POLLOUT�error�poll)� implementer)� posixbase)�
IReactorFDSet)�logc @ s� e Zd ZdZeeB eB ZeZ e
Z dd� Z dd� Z
dd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� ZeZdd� Zdd� ZdS )� PollReactora^
A reactor that uses poll(2).
@ivar _poller: A L{select.poll} which will be used to check for I/O
readiness.
@ivar _selectables: A dictionary mapping integer file descriptors to
instances of L{FileDescriptor} which have been registered with the
reactor. All L{FileDescriptor}s which are currently receiving read or
write readiness notifications will be present as values in this
dictionary.
@ivar _reads: A dictionary mapping integer file descriptors to arbitrary
values (this is essentially a set). Keys in this dictionary will be
registered with C{_poller} for read readiness notifications which will
be dispatched to the corresponding L{FileDescriptor} instances in
C{_selectables}.
@ivar _writes: A dictionary mapping integer file descriptors to arbitrary
values (this is essentially a set). Keys in this dictionary will be
registered with C{_poller} for write readiness notifications which will
be dispatched to the corresponding L{FileDescriptor} instances in
C{_selectables}.
c C s* t � | _i | _i | _i | _tj�| � dS )zo
Initialize polling object, file descriptor tracking dictionaries, and
the base class.
N)r �_poller� _selectables�_reads�_writesr
�PosixReactorBase�__init__��self� r �>/usr/lib/python3/dist-packages/twisted/internet/pollreactor.pyr C s
zPollReactor.__init__c C s� z| j �|� W n ty Y nw d}|| jv r |tB }|| jv r&|tB }|dkr3| j �||� dS || jv r>| j|= dS dS )z*Register/unregister an fd with the poller.r N) r �
unregister�KeyErrorr r r r �registerr )r �fd�maskr r r �_updateRegistrationN s �
�z PollReactor._updateRegistrationc C sn z
|� � }|| W n ty& | j�� D ]
\}}||u r nqY d S Y nw ||v r5||= | �|� d S d S )N)�fileno�
BaseExceptionr �itemsr )r �
selectable�mdictr �fdesr r r � _dictRemove` s ���
�zPollReactor._dictRemovec C �8 |� � }|| jvr|| j|< d| j|< | �|� dS dS )z@Add a FileDescriptor for notification of data available to read.� N)r r r r )r �readerr r r r � addReaderu �
�zPollReactor.addReaderc C r% )zAAdd a FileDescriptor for notification of data available to write.r& N)r r r r )r �writerr r r r � addWriter} r) zPollReactor.addWriterc C � | � || j�S )z?Remove a Selectable for notification of data available to read.)r$ r )r r' r r r � removeReader� � zPollReactor.removeReaderc C r, )z@Remove a Selectable for notification of data available to write.)r$ r )r r* r r r � removeWriter� r. zPollReactor.removeWriterc s, � � � fdd�� jD �� fdd�� jD ��S )zD
Remove all selectables, and return a list of them.
c � g | ]}� j | �qS r �r ��.0r r r r �
<listcomp>� � z)PollReactor.removeAll.<locals>.<listcomp>c r0 r r1 r2 r r r r4 � r5 )�
_removeAllr r r r r r � removeAll� s �zPollReactor.removeAllc
C s� |dur
t |d �}z| j�|�}W n ty. } z|jd tjkr)W Y d}~dS � d}~ww | j}|D ] \}}z| j| }W n t yH Y q4w t
� |||||� q4dS )z Poll the poller for new events.Ni� r ) �intr r � SelectError�args�errno�EINTR�_doReadOrWriter r r �callWithLogger)r �timeout�l�e�_drdwr �eventr! r r r �doPoll� s& �� ��zPollReactor.doPollc � � fdd�� j D �S )Nc r0 r r1 r2 r r r r4 � r5 z*PollReactor.getReaders.<locals>.<listcomp>)r r r r r �
getReaders� � zPollReactor.getReadersc rE )Nc r0 r r1 r2 r r r r4 � r5 z*PollReactor.getWriters.<locals>.<listcomp>)r r r r r �
getWriters� rG zPollReactor.getWritersN)�__name__�
__module__� __qualname__�__doc__r r r �_POLL_DISCONNECTEDr �_POLL_INr � _POLL_OUTr r r$ r( r+ r- r/ r7 rD � doIterationrF rH r r r r r
$ s" r
c C s t � } ddlm} || � dS )zInstall the poll() reactor.r )�installReactorN)r
�twisted.internet.mainrQ )�prQ r r r �install� s rT )rL r; �selectr r r r r r r9 r �zope.interfacer �twisted.internetr
�twisted.internet.interfacesr �twisted.pythonr r �_PollLikeMixinr
rT �__all__r r r r �<module> s $