File: //proc/self/root/lib/python3/dist-packages/twisted/protocols/__pycache__/policies.cpython-38.pyc
U
W[OS � @ s> d Z ddlmZmZ ddlZddlmZmZ ddlm Z m
Z
m Z ddl m
Z
ddlmZ ddlmZ d d
� ZG d d � d e
�ZG d
d� de �ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de �ZG dd � d e�ZG d d � d e�Z G d d � d e�Z G d!d"� d"e�Z G d#d$� d$�Z dS )%z]
Resource limiting policies.
@seealso: See also L{twisted.protocols.htb} for rate limiting.
� )�division�absolute_importN)�directlyProvides�
providedBy)�
ServerFactory�Protocol�
ClientFactory)�error)�ILoggingContext)�logc C s, t �|�r|�� }n|jj}d|| jjf S )zY
Compute a log prefix for a wrapper and the object it wraps.
@rtype: C{str}
z%s (%s))r
r � logPrefix� __class__�__name__)�wrapper�wrappedr � r �</usr/lib/python3/dist-packages/twisted/protocols/policies.py�_wrappedLogPrefix s
r c @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd S ) �ProtocolWrappera�
Wraps protocol instances and acts as their transport as well.
@ivar wrappedProtocol: An L{IProtocol<twisted.internet.interfaces.IProtocol>}
provider to which L{IProtocol<twisted.internet.interfaces.IProtocol>}
method calls onto this L{ProtocolWrapper} will be proxied.
@ivar factory: The L{WrappingFactory} which created this
L{ProtocolWrapper}.
r c C s || _ || _d S �N)�wrappedProtocol�factory)�selfr r r r r �__init__5 s zProtocolWrapper.__init__c C s t | | j�S )zo
Use a customized log prefix mentioning both the wrapped protocol and
the current one.
)r r �r r r r r : s zProtocolWrapper.logPrefixc C s6 t | t|�� t�| |� | j�| � | j�| � dS )z�
When a connection is made, register this wrapper with its factory,
save the real transport, and connect the wrapped protocol to this
L{ProtocolWrapper} to intercept any transport calls it makes.
N)r r r �makeConnectionr �registerProtocolr )r � transportr r r r B s z ProtocolWrapper.makeConnectionc C s | j �|� d S r )r �write�r �datar r r r P s zProtocolWrapper.writec C s | j �|� d S r )r �
writeSequencer r r r r! T s z ProtocolWrapper.writeSequencec C s d| _ | j�� d S �N� )�
disconnectingr �loseConnectionr r r r r% X s z ProtocolWrapper.loseConnectionc C s
| j �� S r )r �getPeerr r r r r& ] s zProtocolWrapper.getPeerc C s
| j �� S r )r �getHostr r r r r' a s zProtocolWrapper.getHostc C s | j �||� d S r )r �registerProducer�r �producerZ streamingr r r r( e s z ProtocolWrapper.registerProducerc C s | j �� d S r )r �unregisterProducerr r r r r+ i s z"ProtocolWrapper.unregisterProducerc C s | j �� d S r )r �
stopConsumingr r r r r, m s z ProtocolWrapper.stopConsumingc C s t | j|�S r )�getattrr �r �namer r r � __getattr__q s zProtocolWrapper.__getattr__c C s | j �|� d S r )r � dataReceivedr r r r r1 w s z ProtocolWrapper.dataReceivedc C s" | j �| � | j�|� d | _d S r )r �unregisterProtocolr �connectionLost�r �reasonr r r r3 { s z ProtocolWrapper.connectionLostN)r �
__module__� __qualname__�__doc__r$ r r r r r! r% r& r' r( r+ r, r0 r1 r3 r r r r r ' s r c @ sd e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� ZdS )�WrappingFactoryzE
Wraps a factory and its protocols, and keeps track of them.
c C s || _ i | _d S r )�wrappedFactory� protocols)r r: r r r r � s zWrappingFactory.__init__c C s t | | j�S )zY
Generate a log prefix mentioning both the wrapped factory and this one.
)r r: r r r r r � s zWrappingFactory.logPrefixc C s | j �� t�| � d S r )r: �doStartr r r r r r< � s
zWrappingFactory.doStartc C s | j �� t�| � d S r )r: �doStopr r r r r r= � s
zWrappingFactory.doStopc C s | j �|� d S r )r: �startedConnecting)r � connectorr r r r> � s z!WrappingFactory.startedConnectingc C s | j �||� d S r )r: �clientConnectionFailed�r r? r5 r r r r@ � s z&WrappingFactory.clientConnectionFailedc C s | j �||� d S r )r: �clientConnectionLostrA r r r rB � s z$WrappingFactory.clientConnectionLostc C s | � | | j�|��S r )�protocolr: �
buildProtocol�r �addrr r r rD � s z WrappingFactory.buildProtocolc C s d| j |< dS )z8
Called by protocol to register itself.
r# N�r; �r �pr r r r � s z WrappingFactory.registerProtocolc C s | j |= dS )z8
Called by protocols when they go away.
NrG rH r r r r2 � s z"WrappingFactory.unregisterProtocolN)r r6 r7 r8 r rC r r r< r= r> r@ rB rD r r2 r r r r r9 � s r9 c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )�ThrottlingProtocolz,
Protocol for L{ThrottlingFactory}.
c C s | j �t|�� t�| |� d S r )r �registerWritten�lenr r r r r r r � s zThrottlingProtocol.writec C s& | j �ttt|��� t�| |� d S r )r rK �sum�maprL r r! �r �seqr r r r! � s z ThrottlingProtocol.writeSequencec C s | j �t|�� t�| |� d S r )r � registerReadrL r r1 r r r r r1 � s z ThrottlingProtocol.dataReceivedc C s || _ t�| ||� d S r )r* r r( r) r r r r( � s z#ThrottlingProtocol.registerProducerc C s | ` t�| � d S r )r* r r+ r r r r r+ � s z%ThrottlingProtocol.unregisterProducerc C s | j �� d S r )r �pauseProducingr r r r �
throttleReads� s z ThrottlingProtocol.throttleReadsc C s | j �� d S r )r �resumeProducingr r r r �unthrottleReads� s z"ThrottlingProtocol.unthrottleReadsc C s t | d�r| j�� d S �Nr* )�hasattrr* rR r r r r �throttleWrites� s
z!ThrottlingProtocol.throttleWritesc C s t | d�r| j�� d S rV )rW r* rT r r r r �unthrottleWrites� s
z#ThrottlingProtocol.unthrottleWritesN)
r r6 r7 r8 r r! r1 r( r+ rS rU rX rY r r r r rJ � s rJ c @ s~ e Zd ZdZeZejddfdd�Zdd� Z dd� Z
d d
� Z d d � Z d
d� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )�ThrottlingFactoryz�
Throttles bandwidth and number of connections.
Write bandwidth will only be throttled if there is a producer
registered.
Nc C sL t �| |� d| _|| _|| _|| _d| _d| _d | _d | _ d | _
d | _ d S )Nr ) r9 r �connectionCount�maxConnectionCount� readLimit�
writeLimit�readThisSecond�writtenThisSecond�unthrottleReadsID�checkReadBandwidthID�unthrottleWritesID�checkWriteBandwidthID)r r: r\ r] r^ r r r r � s zThrottlingFactory.__init__c C s ddl m} |�||�S �z�
Wrapper around
L{reactor.callLater<twisted.internet.interfaces.IReactorTime.callLater>}
for test purpose.
r )�reactor��twisted.internetrf � callLater�r �period�funcrf r r r ri s zThrottlingFactory.callLaterc C s | j |7 _ dS )zH
Called by protocol to tell us more bytes were written.
N)r` �r Zlengthr r r rK s z!ThrottlingFactory.registerWrittenc C s | j |7 _ dS )zE
Called by protocol to tell us more bytes were read.
N)r_ rm r r r rQ s z ThrottlingFactory.registerReadc C sR | j | jkr8| �� t| j �| j d }| �|| j�| _d| _ | �d| j�| _dS )z:
Checks if we've passed bandwidth limits.
� �?r r# N) r_ r] rS �floatri rU ra �checkReadBandwidthrb �r Z throttleTimer r r rp # s �z$ThrottlingFactory.checkReadBandwidthc C sR | j | jkr8| �� t| j �| j d }| �|| j�| _d| _ | �d| j�| _d S )Nrn r r# ) r` r^ rX ro ri rY rc �checkWriteBandwidthrd rq r r r rr 0 s �z%ThrottlingFactory.checkWriteBandwidthc C s* t �d| � | j�� D ] }|�� qdS )z2
Throttle reads on all protocols.
zThrottling reads on %sN)r �msgr; �keysrS rH r r r rS ; s z ThrottlingFactory.throttleReadsc C s0 d| _ t�d| � | j�� D ] }|�� q dS )z9
Stop throttling reads on all protocols.
Nz Stopped throttling reads on %s)ra r rs r; rt rU rH r r r rU D s z!ThrottlingFactory.unthrottleReadsc C s* t �d| � | j�� D ] }|�� qdS )z3
Throttle writes on all protocols.
zThrottling writes on %sN)r rs r; rt rX rH r r r rX N s z ThrottlingFactory.throttleWritesc C s0 d| _ t�d| � | j�� D ] }|�� q dS )z:
Stop throttling writes on all protocols.
Nz Stopped throttling writes on %s)rc r rs r; rt rY rH r r r rY W s z"ThrottlingFactory.unthrottleWritesc C sf | j dkr.| jd k r | �� | jd k r.| �� | j | jk rT| j d7 _ t�| |�S t� d� d S d S )Nr r# z Max connection count reached!)
r[ r] rp r^ rr r\ r9 rD r rs rE r r r rD a s
z ThrottlingFactory.buildProtocolc C sx t �| |� | jd8 _| jdkrt| jd k r8| j�� | jd k rL| j�� | jd k r`| j�� | jd k rt| j�� d S �Nr# r )r9 r2 r[ ra �cancelrb rc rd rH r r r r2 p s
z$ThrottlingFactory.unregisterProtocol)r r6 r7 r8 rJ rC �sys�maxsizer ri rK rQ rp rr rS rU rX rY rD r2 r r r r rZ � s" �
rZ c @ s e Zd Zdd� Zdd� ZdS )�SpewingProtocolc C s t �d| � t�| |� d S )Nz Received: %r)r rs r r1 r r r r r1 � s z SpewingProtocol.dataReceivedc C s t �d| � t�| |� d S )Nz Sending: %r)r rs r r r r r r r � s zSpewingProtocol.writeN)r r6 r7 r1 r r r r r ry s ry c @ s e Zd ZeZdS )�SpewingFactoryN)r r6 r7 ry rC r r r r rz � s rz c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �LimitConnectionsByPeer� c C s
i | _ d S r )�peerConnectionsr r r r � startFactory� s z#LimitConnectionsByPeer.startFactoryc C s> |d }| j �|d�}|| jkr$d S |d | j |< t�| |�S )Nr r# )r} �get�maxConnectionsPerPeerr9 rD )r rF �peerHostr[ r r r rD � s
z$LimitConnectionsByPeer.buildProtocolc C s8 |� � d }| j| d8 < | j| dkr4| j|= d S ru )r& r} )r rI r� r r r r2 � s z)LimitConnectionsByPeer.unregisterProtocolN)r r6 r7 r� r~ rD r2 r r r r r{ � s r{ c @ s4 e Zd ZdZdZdZdZdd� Zdd� Zdd � Z dS )
� LimitTotalConnectionsFactorya�
Factory that limits the number of simultaneous connections.
@type connectionCount: C{int}
@ivar connectionCount: number of current connections.
@type connectionLimit: C{int} or L{None}
@cvar connectionLimit: maximum number of connections.
@type overflowProtocol: L{Protocol} or L{None}
@cvar overflowProtocol: Protocol to use for new connections when
connectionLimit is exceeded. If L{None} (the default value), excess
connections will be closed immediately.
r Nc C sX | j d ks| j| j k r | �� }n| jd kr.d S | �� }| |_t| |�}| jd7 _|S r" )�connectionLimitr[ rC �overflowProtocolr r )r rF r rC r r r rD � s
�
z*LimitTotalConnectionsFactory.buildProtocolc C s d S r r rH r r r r � s z-LimitTotalConnectionsFactory.registerProtocolc C s | j d8 _ d S r" )r[ rH r r r r2 � s z/LimitTotalConnectionsFactory.unregisterProtocol)
r r6 r7 r8 r[ r� r� rD r r2 r r r r r� � s r� c @ sZ e Zd ZdZdd� Zddd�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�TimeoutProtocolzN
Protocol that automatically disconnects when the connection is idle.
c C s( t �| ||� d| _d| _| �|� dS )z�
Constructor.
@param factory: An L{TimeoutFactory}.
@param wrappedProtocol: A L{Protocol} to wrapp.
@param timeoutPeriod: Number of seconds to wait for activity before
timing out.
N)r r � timeoutCall�
timeoutPeriod�
setTimeout)r r r r� r r r r � s zTimeoutProtocol.__init__Nc C s. | � � || _|dk r*| j�| j| j�| _dS )z�
Set a timeout.
This will cancel any existing timeouts.
@param timeoutPeriod: If not L{None}, change the timeout period.
Otherwise, use the existing value.
N)�
cancelTimeoutr� r ri � timeoutFuncr� )r r� r r r r� � s zTimeoutProtocol.setTimeoutc C sB d| _ | jr>z| j�� W n tjtjfk
r6 Y nX d| _dS )zg
Cancel the timeout.
If the timeout was already cancelled, this does nothing.
N)r� r� rv r �
AlreadyCalled�AlreadyCancelledr r r r r� � s z TimeoutProtocol.cancelTimeoutc C s | j r| j �| j� dS )zQ
Reset the timeout, usually because some activity just happened.
N)r� �resetr� r r r r � resetTimeout s z TimeoutProtocol.resetTimeoutc C s | � � t�| |� d S r )r� r r r r r r r s zTimeoutProtocol.writec C s | � � t�| |� d S r )r� r r! rO r r r r! s z TimeoutProtocol.writeSequencec C s | � � t�| |� d S r )r� r r1 r r r r r1 s z TimeoutProtocol.dataReceivedc C s | � � t�| |� d S r )r� r r3 r4 r r r r3 s z TimeoutProtocol.connectionLostc C s | � � dS )z�
This method is called when the timeout is triggered.
By default it calls I{loseConnection}. Override this if you want
something else to happen.
N)r% r r r r r� s zTimeoutProtocol.timeoutFunc)N)
r r6 r7 r8 r r� r� r� r r! r1 r3 r� r r r r r� � s
r� c @ s. e Zd ZdZeZd
dd�Zdd� Zdd� Zd S ) �TimeoutFactoryz%
Factory for TimeoutWrapper.
� c C s || _ t�| |� d S r )r� r9 r )r r: r� r r r r / s zTimeoutFactory.__init__c C s | j | | j�|�| jd�S )N)r� )rC r: rD r� rE r r r rD 4 s �z TimeoutFactory.buildProtocolc C s ddl m} |�||�S re rg rj r r r ri 9 s zTimeoutFactory.callLaterN)r� ) r r6 r7 r8 r� rC r rD ri r r r r r� ( s
r� c @ sV e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dS )�TrafficLoggingProtocolNr c C s$ t �| ||� || _|| _|| _dS )a
@param factory: factory which created this protocol.
@type factory: L{protocol.Factory}.
@param wrappedProtocol: the underlying protocol.
@type wrappedProtocol: C{protocol.Protocol}.
@param logfile: file opened for writing used to write log messages.
@type logfile: C{file}
@param lengthLimit: maximum size of the datareceived logged.
@type lengthLimit: C{int}
@param number: identifier of the connection.
@type number: C{int}.
N)r r �logfile� lengthLimit�_number)r r r r� r� Znumberr r r r F s z TrafficLoggingProtocol.__init__c C s | j �|d � | j �� d S )N�
)r� r �flush)r �liner r r �_logZ s zTrafficLoggingProtocol._logc C s. | j r*t|�| j kr*|d | j d � d }|S )N� z <... elided>)r� rL r r r r �
_mungeData_ s z!TrafficLoggingProtocol._mungeDatac C s | � d� t�| �S )N�*)r� r �connectionMader r r r r� f s
z%TrafficLoggingProtocol.connectionMadec C s& | � d| j| �|�f � t�| |�S �NzC %d: %r)r� r� r� r r1 r r r r r1 k s z#TrafficLoggingProtocol.dataReceivedc C s | � d| j|f � t�| |�S r� )r� r� r r3 r4 r r r r3 p s z%TrafficLoggingProtocol.connectionLostc C s& | � d| j| �|�f � t�| |�S )NzS %d: %r)r� r� r� r r r r r r r v s z TrafficLoggingProtocol.writec s. � � d� j� fdd�|D �f � t�� |�S )Nz SV %d: %rc s g | ]}� � |��qS r )r� )�.0�dr r r �
<listcomp>| s z8TrafficLoggingProtocol.writeSequence.<locals>.<listcomp>)r� r� r r! )r Ziovecr r r r! { s "z$TrafficLoggingProtocol.writeSequencec C s | � d| jf � t�| �S )NzS %d: *)r� r� r r% r r r r r% � s z%TrafficLoggingProtocol.loseConnection)Nr ) r r6 r7 r r� r� r� r1 r3 r r! r% r r r r r� D s �
r� c @ s6 e Zd ZeZdZd dd�Zdd� Zdd� Zd d
� Z dS ) �TrafficLoggingFactoryr Nc C s || _ || _t�| |� d S r )�
logfilePrefixr� r9 r )r r: r� r� r r r r � s z TrafficLoggingFactory.__init__c C s
t |d�S )N�w)�openr. r r r r� � s zTrafficLoggingFactory.openc C sF | j d7 _ | �| jd t| j � �}| �| | j�|�|| j| j �S )Nr# �-)�_counterr� r� �strrC r: rD r� )r rF r� r r r rD � s �z#TrafficLoggingFactory.buildProtocolc C s
d| _ dS )zN
Reset the value of the counter used to identify connections.
r N)r� r r r r � resetCounter� s z"TrafficLoggingFactory.resetCounter)N)
r r6 r7 r� rC r� r r� rD r� r r r r r� � s
r� c @ s@ e Zd ZdZdZdZdd� Zdd� Zdd� Zd d
� Z d d � Z
dS )
� TimeoutMixinaL
Mixin for protocols which wish to timeout connections.
Protocols that mix this in have a single timeout, set using L{setTimeout}.
When the timeout is hit, L{timeoutConnection} is called, which, by
default, closes the connection.
@cvar timeOut: The number of seconds after which to timeout the connection.
Nc C s ddl m} |�||�S re rg rj r r r ri � s zTimeoutMixin.callLaterc C s& | j dk r"| jdk r"| j �| j� dS )a�
Reset the timeout count down.
If the connection has already timed out, then do nothing. If the
timeout has been cancelled (probably using C{setTimeout(None)}), also
do nothing.
It's often a good idea to call this when the protocol has received
some meaningful input from the other end of the connection. "I've got
some data, they're still there, reset the timeout".
N)�_TimeoutMixin__timeoutCall�timeOutr� r r r r r� � s zTimeoutMixin.resetTimeoutc C s| | j }|| _ | jdk r`|dkrRz| j�� W n tjtjfk
rH Y nX d| _qx| j�|� n|dk rx| �|| j�| _|S )z�
Change the timeout period
@type period: C{int} or L{None}
@param period: The period, in seconds, to change the timeout to, or
L{None} to disable the timeout.
N) r� r� rv r r� r� r� ri �_TimeoutMixin__timedOut)r rk �prevr r r r� � s
zTimeoutMixin.setTimeoutc C s d | _ | �� d S r )r� �timeoutConnectionr r r r Z
__timedOut� s zTimeoutMixin.__timedOutc C s | j �� dS )z�
Called when the connection times out.
Override to define behavior other than dropping the connection.
N)r r% r r r r r� � s z TimeoutMixin.timeoutConnection) r r6 r7 r8 r� r� ri r� r� r� r� r r r r r� � s
r� ) r8 Z
__future__r r rw Zzope.interfacer r Ztwisted.internet.protocolr r r rh r Ztwisted.internet.interfacesr
Ztwisted.pythonr r r r9 rJ rZ ry rz r{ r� r� r� r� r� r� r r r r �<module> s, \<3 *Y B