| Current File : //proc/self/root/lib/python3/dist-packages/twisted/test/__pycache__/iosim.cpython-312.pyc |
�
Ϫ�f�I � � � d Z ddlZ ddlmZ ddlmZmZ ddl m
Z
m Z ddl m
Z
mZ ddlmZ ddlmZmZ dd lmZ dd
lmZ G d � d � Z ee j2 � G d
� d� � Z ee j6 e j8 � G d� d� � Z d� Z d� Z G d� d� Z d d�Z!e e dddfd�Z"d� Z# G d� d� Z$d d �Z%y# e$ r Y ��w xY w) z0
Utilities and helpers for simulating a network
� N)�Error)�directlyProvides� implementer)�error�
interfaces)�TCP4ClientEndpoint�TCP4ServerEndpoint)�ConnectionRefusedError)�Factory�Protocol)�MemoryReactorClock)�Failurec �$ � e Zd Zd� Zdefd�Zd� Zy)�TLSNegotiationc �< � || _ || _ d| _ || _ y )NF)�obj� connectState�sent� readyToSend)�selfr r s �4/usr/lib/python3/dist-packages/twisted/test/iosim.py�__init__zTLSNegotiation.__init__ s � ���� (����� �'��� �returnc �" � d| j �d�S )NzTLSNegotiation(�))r �r s r �__repr__zTLSNegotiation.__repr__! s � � ���� �A�.�.r c � � | j j |j � s t � |_ |j � y y �N)r � iosimVerify�NativeOpenSSLError�disconnectReason�loseConnection)r �other�tpts r �pretendToVerifyz TLSNegotiation.pretendToVerify$ s6 � � �x�x�#�#�E�I�I�.�#5�#7�C� � � � � � /r N)�__name__�
__module__� __qualname__r �strr r' � r r r r s � �(� /�#� /�!r r c � � e Zd ZdZy)� FakeAddressz]
The default address type for the host and peer of L{FakeTransport}
connections.
N)r( r) r* �__doc__r, r r r. r. - s � �r r. c � � e Zd ZdZ e ej � fd�� ZdZdZ dZ
e j d� Z
dZdZdZd d�Zdefd�Zd � Zd
� Zd � Zd � Zd
� Zd� Zd� Zd� Zd� Z d� Z d� Z d� Z d� Z d� Z!d� Z"d!d�Z#d� Z$d� Z%d� Z&d � Z'd � Z(d � Z)d � Z*y)"�
FakeTransportz�
A wrapper around a file-like object to make it behave as a Transport.
This doesn't actually stream the file to the attached protocol,
and is thus useful mainly as a utility for debugging protocols.
c �* � t t | � � S r )�int�next)�counters r �<lambda>zFakeTransport.<lambda>> s � ��T�'�]�AS� r r zConnection doneNc � � || _ || _ g | _ | j � | _ |�
t � }|| _ |�
t � }|| _ y)a�
@param protocol: This transport will deliver bytes to this protocol.
@type protocol: L{IProtocol} provider
@param isServer: C{True} if this is the accepting side of the
connection, C{False} if it is the connecting side.
@type isServer: L{bool}
@param hostAddress: The value to return from C{getHost}. L{None}
results in a new L{FakeAddress} being created to use as the value.
@type hostAddress: L{IAddress} provider or L{None}
@param peerAddress: The value to return from C{getPeer}. L{None}
results in a new L{FakeAddress} being created to use as the value.
@type peerAddress: L{IAddress} provider or L{None}
N)�protocol�isServer�stream� _nextserial�serialr. � hostAddress� peerAddress)r r8 r9 r= r> s r r zFakeTransport.__init__G sS � �" !��
� ��
��� ��&�&�(�� � � �%�-�K�&��� � �%�-�K�&��r r c � � dj | j xr dxs d| j | j j j
� S )NzFakeTransport<{},{},{}>�S�C)�formatr9 r< r8 � __class__r( r s r r zFakeTransport.__repr__c s@ � �(�/�/� �M�M� !�c� (�S� �K�K� �M�M� #� #� ,� ,�
�
r c � � | j ry | j � | j j |� y | j j |� y r )�
disconnecting�tls�tlsbuf�appendr: )r �datas r �writezFakeTransport.writej s< � � � � � � �8�8� � �K�K� � �t� $� �K�K� � �t� $r c �l � | j r(| j s| j j � y y y r ��producer�streamingProducer�resumeProducingr s r �_checkProducerz FakeTransport._checkProducert s* � � �=�=��!7�!7� �M�M� )� )� +� "8�=r c �F � || _ || _ |s|j � yy)z.
From abstract.FileDescriptor
NrL )r rM � streamings r �registerProducerz FakeTransport.registerProducerz s' � � !��
�!*�� �� � $� $� &� r c � � d | _ y r )rM r s r �unregisterProducerz FakeTransport.unregisterProducer� s � � ��
r c �D � | j � | j � y r )rU r$ r s r �
stopConsumingzFakeTransport.stopConsuming� s � � � � �!� ��� r c �D � | j dj |� � y )Nr )rJ �join)r �iovecs r �
writeSequencezFakeTransport.writeSequence� s � � �
�
�3�8�8�E�?�#r c � � d| _ y �NT�rE r s r r$ z FakeTransport.loseConnection� s
� � !��r c � � d| _ y)zp
For the time being, this is the same as loseConnection; no buffered
data will be lost.
TNr^ r s r �abortConnectionz FakeTransport.abortConnection� s � �
"��r c � � | j � t � }n | j }| j j t |� � y r )rF r" r# r8 �connectionLostr )r �errs r �reportDisconnectz FakeTransport.reportDisconnect� s7 � � �8�8� � %�&�C��'�'�C� �
�
�$�$�W�S�\�2r c � � y)zM
Identify this transport/event source to the logging system.
�iosimr, r s r � logPrefixzFakeTransport.logPrefix� s � � r c � � | j S r )r> r s r �getPeerzFakeTransport.getPeer� � � �� � � r c � � | j S r )r= r s r �getHostzFakeTransport.getHost� rj r c � � y r r, r s r rO z FakeTransport.resumeProducing� � � � r c � � y r r, r s r �pauseProducingz FakeTransport.pauseProducing� rn r c �$ � | j � y r )r$ r s r �
stopProducingzFakeTransport.stopProducing� s � � ��� r c �R � | j |z }t ||� | _ g | _ y r )r9 r rF rG )r �contextFactory�beNormalr s r �startTLSzFakeTransport.startTLS� s' � � �}�}�x�/� �!�.�,�?����� r c �� � | j }|rg | _ dj |� S | j �4| j j r d| j _ | j S yy)z�
Get the pending writes from this transport, clearing them from the
pending buffer.
@return: the bytes written with C{transport.write}
@rtype: L{bytes}
r NT)r: rY rF r r )r r@ s r � getOutBufferzFakeTransport.getOutBuffer� sV � �
�K�K�� � �D�K��8�8�A�;� �
�X�X�
!��x�x�#�#� $����
��x�x���r c � � t |t � r�| j �J �| j j rc| j j || � d | _ | j
d c}| _ | j
|� t | t j � y d| j _
y | j j |� y r] )
�
isinstancer rF r r' rG r[ r r �
ISSLTransportr r8 � dataReceived)r �buf�bs r �bufferReceivedz FakeTransport.bufferReceived� s� � � �c�>� *��8�8�'� '�'� �x�x�}�}����(�(��d�3� ��� "&���d���4�;��"�"�1�%� ��z�'?�'?�@� (,����$� �M�M� &� &�s� +r c � � y r r, r s r �getTcpKeepAlivez FakeTransport.getTcpKeepAlive� rn r c � � y r r, r s r �
getTcpNoDelayzFakeTransport.getTcpNoDelay� rn r c � � y r r, r s r �loseWriteConnectionz!FakeTransport.loseWriteConnection� rn r c � � y r r, �r �enableds r �setTcpKeepAlivez FakeTransport.setTcpKeepAlive� rn r c � � y r r, r� s r �
setTcpNoDelayzFakeTransport.setTcpNoDelay� rn r )NN)T)+r( r) r* r/ � staticmethod� itertools�countr; �closedrE � disconnectedr �ConnectionDoner# rM rN rF r r+ r rJ rP rS rU rW r[ r$ r` rd rg ri rl rO rp rr rv rx r r� r� r� r� r� r, r r r1 r1 5 s� � �� �o�i�o�o�.?� S�T�K�
�F��M��L�+�u�+�+�,=�>���H���
�C�'�8
�#�
�%�,� '� � �$�"�"�3�� � �
�
� ���,,�&
�
�
�
�
r r1 c � � t | d�� S )z�
Create and return a new in-memory transport hooked up to the given protocol.
@param clientProtocol: The client protocol to use.
@type clientProtocol: L{IProtocol} provider
@return: The transport.
@rtype: L{FakeTransport}
F�r9 �r1 )�clientProtocols r �makeFakeClientr� � s � � � �%� 8�8r c � � t | d�� S )z�
Create and return a new in-memory transport hooked up to the given protocol.
@param serverProtocol: The server protocol to use.
@type serverProtocol: L{IProtocol} provider
@return: The transport.
@rtype: L{FakeTransport}
Tr� r� )�serverProtocols r �makeFakeServerr�
s � � � �$� 7�7r c �( � e Zd ZdZdd�Zdd�Zdd�Zy)�IOPumpz�
Utility to pump data between clients and servers for protocol testing.
Perhaps this is a utility worthy of being in protocol.py?
Nc �p � || _ || _ || _ || _ || _ |�
t � }|| _ y r )�client�server�clientIO�serverIO�debugr
�clock)r r� r� r� r� r� r� s r r zIOPump.__init__ s9 � � �� � �� � ��
� ��
���
� �=�&�(�E���
r c �b � d}t d� D ] }| j ||� rd}� |S J d� �)zk
Pump until there is no more input or output.
Returns whether any data was moved.
Fi� TzToo long)�range�pump)r r� � advanceClock�result�_s r �flushz IOPump.flush( sE � � ���t�� !�A��y�y�� �-� ��� �
� !�
!�j� �1r c �\ � |r| j j d� | j s|r t d� | j j � }| j j � }| j j � | j j � | j s|r=t d� |rt dt |� z � |rt dt |� z � |r| j j |� |r| j j |� |s|ry| j j rl| j j sV| j s|r t d� d| j _ d| j _
| j j � y| j j rl| j j sV| j s|r t d� d| j _ d| j _
| j j � yy )
z�
Move data back and forth, while also triggering any currently pending
scheduled calls (i.e. C{callLater(0, f)}).
Returns whether any data was moved.
r z
-- GLUG --�.zC: zS: Tz* Cz* SF)
r� �advancer� �printr� rx r� rP �reprr rE r� rd )r r� r� �sData�cDatas r r� z IOPump.pump8 s� � � � �J�J� � �q� !� �:�:�� �,� ��
�
�*�*�,���
�
�*�*�,�� �
�
�$�$�&� �
�
�$�$�&� �:�:�� �#�J���e�d�5�k�)�*���e�d�5�k�)�*� � �M�M� (� (�� /� � �M�M� (� (�� /� �E�� �=�=� &� &�t�}�}�/I�/I��z�z�U��e� �)-�D�M�M� &�*.�D�M�M� '� �M�M� *� *� ,�� �=�=� &� &�t�}�}�/I�/I��z�z�U��e� �)-�D�M�M� &�*.�D�M�M� '� �M�M� *� *� ,��r r )FT)r( r) r* r/ r r� r� r, r r r� r� s � �� �� *r r� FTc � � | j |� |j |� t || ||||�� }|r|j � |S )a�
Create a new L{IOPump} connecting two protocols.
@param serverProtocol: The protocol to use on the accepting side of the
connection.
@type serverProtocol: L{IProtocol} provider
@param serverTransport: The transport to associate with C{serverProtocol}.
@type serverTransport: L{FakeTransport}
@param clientProtocol: The protocol to use on the initiating side of the
connection.
@type clientProtocol: L{IProtocol} provider
@param clientTransport: The transport to associate with C{clientProtocol}.
@type clientTransport: L{FakeTransport}
@param debug: A flag indicating whether to log information about what the
L{IOPump} is doing.
@type debug: L{bool}
@param greet: Should the L{IOPump} be L{flushed <IOPump.flush>} once before
returning to put the protocols into their post-handshake or
post-server-greeting state?
@type greet: L{bool}
@param clock: An optional L{Clock}. Pumping the resulting L{IOPump} will
also increase clock time by a small increment.
@return: An L{IOPump} which connects C{serverProtocol} and
C{clientProtocol} and delivers bytes between them when it is pumped.
@rtype: L{IOPump}
�r� )�makeConnectionr� r� )r� �serverTransportr� �clientTransportr� �greetr� r� s r �connectr� e sP � �T �!�!�/�2��!�!�/�2� �����
��
�D�
� �
�
� � �Kr c �h � |� } | � } ||� } ||� }
||t ||
|| |||�� fS )a�
Connect a given server and client class to each other.
@param ServerClass: a callable that produces the server-side protocol.
@type ServerClass: 0-argument callable returning L{IProtocol} provider.
@param ClientClass: like C{ServerClass} but for the other side of the
connection.
@type ClientClass: 0-argument callable returning L{IProtocol} provider.
@param clientTransportFactory: a callable that produces the transport which
will be attached to the protocol returned from C{ClientClass}.
@type clientTransportFactory: callable taking (L{IProtocol}) and returning
L{FakeTransport}
@param serverTransportFactory: a callable that produces the transport which
will be attached to the protocol returned from C{ServerClass}.
@type serverTransportFactory: callable taking (L{IProtocol}) and returning
L{FakeTransport}
@param debug: Should this dump an escaped version of all traffic on this
connection to stdout for inspection?
@type debug: L{bool}
@param greet: Should the L{IOPump} be L{flushed <IOPump.flush>} once before
returning to put the protocols into their post-handshake or
post-server-greeting state?
@type greet: L{bool}
@param clock: An optional L{Clock}. Pumping the resulting L{IOPump} will
also increase clock time by a small increment.
@return: the client protocol, the server protocol, and an L{IOPump} which,
when its C{pump} and C{flush} methods are called, will move data
between the created client and server protocol instances.
@rtype: 3-L{tuple} of L{IProtocol}, L{IProtocol}, L{IOPump}
r� )r� ) � ServerClass� ClientClass�clientTransportFactory�serverTransportFactoryr� r� r� �c�s�cio�sios r �connectedServerAndClientr� � sE � �\ �
�A��
�A�
��
#�C�
��
#�C� �a���C��C���U�C� C�Cr c �4 � | \ }}}}}|\ }}} }
||k( r||fS y)a'
Should the client and server described by the arguments be connected to
each other, i.e. do their port numbers match?
@param clientInfo: the args for connectTCP
@type clientInfo: L{tuple}
@param serverInfo: the args for listenTCP
@type serverInfo: L{tuple}
@return: If they do match, return factories for the client and server that
should connect; otherwise return L{None}, indicating they shouldn't be
connected.
@rtype: L{None} or 2-L{tuple} of (L{ClientFactory},
L{IProtocolFactory})
Nr, ) �
clientInfo�
serverInfo�
clientHost�
clientPort�
clientFactory�
clientTimeout�clientBindAddress�
serverPort�
serverFactory�
serverBacklog�serverInterfaces r �_factoriesShouldConnectr� � s? � �. �
������BL�?�Z� �
���Z� � �m�+�+�r c �> � e Zd ZdZd� Zdd�Z e e� � fd�Zy)�ConnectionCompleterz�
A L{ConnectionCompleter} can cause synthetic TCP connections established by
L{MemoryReactor.connectTCP} and L{MemoryReactor.listenTCP} to succeed or
fail.
c � � || _ y)z�
Create a L{ConnectionCompleter} from a L{MemoryReactor}.
@param memoryReactor: The reactor to attach to.
@type memoryReactor: L{MemoryReactor}
N)�_reactor)r �
memoryReactors r r z ConnectionCompleter.__init__� s � � &��
r c � � | j }t |j � D ]� \ }}|j D ]� }t ||� }|s�|j j |� |j j |� |\ }}|j d� } |j d� }
t |
� } t | � } t |
| | | |� c c S �� y)a�
Complete a single TCP connection established on this
L{ConnectionCompleter}'s L{MemoryReactor}.
@param debug: A flag; whether to dump output from the established
connection to stdout.
@type debug: L{bool}
@return: a pump for the connection, or L{None} if no connection could
be established.
@rtype: L{IOPump} or L{None}
N) r� � enumerate�
tcpClients�
tcpServersr� �remove�
connectors�pop�
buildProtocolr� r� r� )
r r� r� � clientIdxr� r� � factoriesr� r� r� r� r� r� s
r � succeedOncez ConnectionCompleter.succeedOnce s� � � �
�
�
�%.�}�/G�/G�%H� � !�I�z� +� 6� 6�
�
� 3�J�
� K� � �!�,�,�3�3�J�?�!�,�,�0�0��;�3<�0�M�=�%2�%@�%@��%F�N�%2�%@�%@��%F�N�&4�^�&D�O�&4�^�&D�O�"�&�'�&�'� � � �
� r c � � | j j j d� d j | j j j d� |� y)z�
Fail a single TCP connection established on this
L{ConnectionCompleter}'s L{MemoryReactor}.
@param reason: the reason to provide that the connection failed.
@type reason: L{Failure}
r � N)r� r� r� �clientConnectionFailedr� )r �reasons r �failOncez ConnectionCompleter.failOnce$ sF � �
�
�
� � �$�$�Q�'��*�A�A� �M�M� $� $� (� (�� +�V�
r N�F) r( r) r* r/ r r� r r
r� r, r r r� r� � s% � �� &� �B &�&<�&>� ?�
r r� c � � t � }t |dd� }t |d� }|j t j
t � � |t |� fS )a�
Create an endpoint that can be fired on demand.
@param debug: A flag; whether to dump output from the established
connection to stdout.
@type debug: L{bool}
@return: A client endpoint, and an object that will cause one of the
L{Deferred}s returned by that client endpoint.
@rtype: 2-L{tuple} of (L{IStreamClientEndpoint}, L{ConnectionCompleter})
z0.0.0.0i� )r
r r �listenr � forProtocolr r� )r� �reactor�clientEndpoint�serverEndpoints r �connectableEndpointr� 1 sP � � !�"�G�'���D�A�N�'���6�N����'�-�-�h�7�8� �.�w�7� 7�7r )FTNr� )&r/ r� � OpenSSL.SSLr r" � ImportError�zope.interfacer r �twisted.internetr r �twisted.internet.endpointsr r �twisted.internet.errorr
�twisted.internet.protocolr r �twisted.internet.testingr
�twisted.python.failurer r �IAddressr. �
ITransport�
ITLSTransportr1 r� r� r� r� r� r� r� r� r, r r �<module>r� s� ��
� � �7� 9� .� M� 9� 7� 7� *�!� !�&
�Z�
�
�!�� � "��
�Z�
"�
"�J�$<�$<�=�D
� D
� >�D
�N
9�
8�K� K�f �
�
�7�z *�)�
�
�
�2D�j �>;
� ;
�|8��I � �� �s �C �C
� C