CasperSecurity
o
�bW" � @ s� d Z ddlmZ ddlmZmZ dZdZG dd� d�ZG dd � d �Z dd
l
Z
dd
l Z dd
l Z dd
l
Z
ee�G d d � d e��Zd
Zee�G dd� de��Zd
S )z�
Implements a simple polling interface for file descriptors that don't work with
select() - this is pretty much only useful on Windows.
� )� implementer)� IConsumer�
IPushProducerg��&� .>g�������?c @ s e Zd ZdZdd� Zdd� ZdS )�_PollableResourceTc C �
d| _ d S �NT��active��self� r �?/usr/lib/python3/dist-packages/twisted/internet/_pollingfile.py�activate �
z_PollableResource.activatec C r �NFr r
r r r
�
deactivate r z _PollableResource.deactivateN)�__name__�
__module__� __qualname__r r r r r r r
r s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�
_PollingTimerc C s" || _ g | _d | _t| _d| _d S r )�reactor�
_resources�
_pollTimer� MAX_TIMEOUT�_currentTimeout�_paused)r r r r r
�__init__ s
z_PollingTimer.__init__c C s | j �|� | �� d S �N)r �append�_checkPollingState)r �resr r r
�_addPollableResource' s z"_PollingTimer._addPollableResourcec C s, | j D ] }|jr| �� d S q| �� d S r )r r �
_startPolling� _stopPolling)r �resourcer r r
r + s
� z _PollingTimer._checkPollingStatec C s | j d u r | �� | _ d S d S r )r � _rescheduler
r r r
r" 3 s
�z_PollingTimer._startPollingc C s" | j d ur| j �� d | _ d S d S r )r �cancelr
r r r
r# 7 s
�z_PollingTimer._stopPollingc C r r )r r
r r r
�_pause< r z_PollingTimer._pausec C s d| _ | �� d S r )r r r
r r r
�_unpause? s z_PollingTimer._unpausec C s | j s | j�| j| j�S d S r )r r � callLaterr �
_pollEventr
r r r
r% C s �z_PollingTimer._reschedulec C s� d}g }| j D ]}|jr||�� 7 }|jr|�|� q| j}|r.| j|d }|tk r-t}n | jd }|tkr9t}|| _|rE| �� | _d S d S )Ng g �?g @) r r � checkWorkr r � MIN_TIMEOUTr r% r )r � workUnits� anyActiver$ �
newTimeoutr r r
r* G s*
��
�z_PollingTimer._pollEventN) r r r r r! r r" r# r' r( r% r* r r r r
r s r Nc @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�_PollableReadPipec C s || _ || _|| _d S r )�pipe�receivedCallback� lostCallback)r r1 r2 r3 r r r
r j s
z_PollableReadPipe.__init__c C s� d}g } z t �| jd�\}}}|sW n t�| j|d �\}}|�|� W n tjy0 d}Y nw qd�|�}|r>| � |� |rD| �
� t |�S )Nr � � ) � win32pipe�
PeekNamedPiper1 � win32file�ReadFiler �win32api�error�joinr2 �cleanup�len) r �finished� fullDataRead�buffer� bytesToRead�result�hr�data�dataBufr r r
r+ p s( ��
z_PollableReadPipe.checkWorkc C s | � � | �� d S r )r r3 r
r r r
r= � s z_PollableReadPipe.cleanupc C s* z t �| j� W d S tjy Y d S w r )r: � CloseHandler1 �
pywintypesr; r
r r r
�close� s
�z_PollableReadPipe.closec C � | � � d S r )rI r
r r r
�
stopProducing� � z _PollableReadPipe.stopProducingc C rJ r )r r
r r r
�pauseProducing� rL z _PollableReadPipe.pauseProducingc C rJ r )r r
r r r
�resumeProducing� rL z!_PollableReadPipe.resumeProducingN)
r r r r r+ r= rI rK rM rN r r r r
r0 h s r0 i c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�_PollableWritePipec C sZ d| _ d | _d| _d| _g | _|| _|| _z t�|tj d d � W d S t
j y, Y d S w )NFr ) �
disconnecting�producer�producerPaused�streamingProducer�outQueue� writePiper3 r6 �SetNamedPipeHandleState� PIPE_NOWAITrH r; )r rU r3 r r r
r � s
��z_PollableWritePipe.__init__c C r r )rP r
r r r
rI � r z_PollableWritePipe.closec C s" | j d urd| _| j �� d S d S r )rQ rR rM r
r r r
�
bufferFull� s
�z _PollableWritePipe.bufferFullc C s0 | j d ur| jr | jrd| j _| j �� dS dS )NFT)rQ rS rR rN r
r r r
� bufferEmpty� s
��
z _PollableWritePipe.bufferEmptyc C sN | j durtd|| j f ��| js|�� dS || _ || _|s%|�� dS dS )aw Register to receive data from a producer.
This sets this selectable to be a consumer for a producer. When this
selectable runs out of data on a write() call, it will ask the producer
to resumeProducing(). A producer should implement the IProducer
interface.
FileDescriptor provides some infrastructure for producer methods.
NzHCannot register producer %s, because producer %s was never unregistered.)rQ � RuntimeErrorr rK rS rN )r rQ � streamingr r r
�registerProducer� s
�� �z#_PollableWritePipe.registerProducerc C s
d| _ dS )z;Stop consuming data from a producer, without disconnecting.N)rQ r
r r r
�unregisterProducer� s
z%_PollableWritePipe.unregisterProducerc C s: | � � zt�| j� W n
tjy Y nw | �� d S r )r r: rG rU rH r; r3 r
r r r
�writeConnectionLost� s � z&_PollableWritePipe.writeConnectionLostc C s& t tt|�v r td��| j�|� dS )z�
Append a C{list} or C{tuple} of bytes to the output buffer.
@param seq: C{list} or C{tuple} of C{str} instances to be appended to
the output buffer.
@raise TypeError: If C{seq} contains C{unicode}.
�%Unicode not allowed in output buffer.N)�str�map�type� TypeErrorrT �extend)r �seqr r r
�
writeSequence� s z _PollableWritePipe.writeSequencec C sL t |t�r td��| jrdS | j�|� ttt| j��t kr$| �
� dS dS )z�
Append some bytes to the output buffer.
@param data: C{str} to be appended to the output buffer.
@type data: C{str}.
@raise TypeError: If C{data} is C{unicode} instead of C{str}.
r_ N) �
isinstancer` rc rP rT r �sumra r> �FULL_BUFFER_SIZErX )r rE r r r
�write� s
�z_PollableWritePipe.writec C s� d}| j s)| jr| �� dS z
t�| jdd � W n tjy( | �� | Y S w | j rk| j �d�}d}z t�| j|d �\}}W n t jyO | �� Y |S w ||7 }t
|�|krh| j � d||d � � |S | j s,| � � }|sx| jrx| �� |S )Nr r5 )
rT rP r^ r8 � WriteFilerU rH r; �popr: r> �insertrY )r �numBytesWrittenrE �errCode�
nBytesWritten�resumedr r r
r+ s> � � �
z _PollableWritePipe.checkWorkN)
r r r r rI rX rY r\ r] r^ rf rj r+ r r r r
rO � s
rO )�__doc__�zope.interfacer �twisted.internet.interfacesr r r, r r r rH r: r8 r6 r0 ri rO r r r r
�<module> s
E3