File manager - Edit - /usr/lib/python3/dist-packages/twisted/test/__pycache__/iosim.cpython-310.pyc
Back
o �b�F � @ s8 d Z ddlZzddlmZ W n ey Y nw ddlmZmZ ddl m Z m Z ddl m Z mZ ddlmZ ddlmZmZ dd lmZ dd lmZ G d d � d �Zee j�G d d� d��Zee je j �G dd� d��Z dd� Z dd� Z G dd� d�Z d#dd�Z!e e ddfdd �Z"d d � Z#G d d � d �Z$d$d!d"�Z%dS )%z0 Utilities and helpers for simulating a network � N)�Error)�directlyProvides� implementer)�error� interfaces)�TCP4ClientEndpoint�TCP4ServerEndpoint)�ConnectionRefusedError)�Factory�Protocol)�MemoryReactorClock)�Failurec @ s* e Zd Zdd� Zdefdd�Zdd� ZdS ) �TLSNegotiationc C s || _ || _d| _|| _d S )NF)�obj� connectState�sent� readyToSend)�selfr r � r �4/usr/lib/python3/dist-packages/twisted/test/iosim.py�__init__ s zTLSNegotiation.__init__�returnc C s d| j �d�S )NzTLSNegotiation(�))r �r r r r �__repr__" s zTLSNegotiation.__repr__c C s& | j �|j �st� |_|�� d S d S �N)r � iosimVerify�NativeOpenSSLError�disconnectReason�loseConnection)r �other�tptr r r �pretendToVerify% s �z TLSNegotiation.pretendToVerifyN)�__name__� __module__� __qualname__r �strr r" r r r r r s r c @ s e Zd ZdZdS )� 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' c @ s e Zd ZdZee�� fdd��ZdZdZ dZ e � d�Z dZdZdZd;dd�Zd efd d �Zd d � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Z d d � Z d d!� Z d"d#� Z d$d%� Z d&d'� Z!d(d)� Z"d<d+d,�Z#d-d.� Z$d/d0� Z%d1d2� Z&d3d4� Z'd5d6� Z(d7d8� Z)d9d:� Z*dS )=� 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 C s t t| ��S r )�int�next)�counterr r r �<lambda>? s zFakeTransport.<lambda>r zConnection doneNc C sH || _ || _g | _| �� | _|du rt� }|| _|du r t� }|| _dS )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 r. r/ r3 r4 r r r r H s zFakeTransport.__init__r c C s d� | jrdpd| j| jjj�S )NzFakeTransport<{},{},{}>�S�C)�formatr/ r2 r. � __class__r# r r r r r d s �zFakeTransport.__repr__c C s4 | j rd S | jd ur| j�|� d S | j�|� d S r )� disconnecting�tls�tlsbuf�appendr0 )r �datar r r �writek s zFakeTransport.writec C s" | j r | js| j �� d S d S d S r ��producer�streamingProducer�resumeProducingr r r r �_checkProduceru s �z FakeTransport._checkProducerc C s || _ || _|s|�� dS dS )z. From abstract.FileDescriptor Nr? )r r@ � streamingr r r �registerProducer{ s �z FakeTransport.registerProducerc C s d | _ d S r )r@ r r r r �unregisterProducer� � z FakeTransport.unregisterProducerc C s | � � | �� d S r )rF r r r r r � stopConsuming� s zFakeTransport.stopConsumingc C s | � d�|�� d S )N� )r> �join)r �iovecr r r � writeSequence� s zFakeTransport.writeSequencec C s d| _ d S �NT�r9 r r r r r � rG z FakeTransport.loseConnectionc C s d| _ dS )zp For the time being, this is the same as loseConnection; no buffered data will be lost. TNrN r r r r �abortConnection� s z FakeTransport.abortConnectionc C s, | j d ur t� }n| j}| j�t|�� d S r )r: r r r. �connectionLostr )r �errr r r �reportDisconnect� s z FakeTransport.reportDisconnectc C s dS )zM Identify this transport/event source to the logging system. �iosimr r r r r � logPrefix� s zFakeTransport.logPrefixc C � | j S r )r4 r r r r �getPeer� � zFakeTransport.getPeerc C rU r )r3 r r r r �getHost� rW zFakeTransport.getHostc C � d S r r r r r r rB � � z FakeTransport.resumeProducingc C rY r r r r r r �pauseProducing� rZ z FakeTransport.pauseProducingc C s | � � d S r )r r r r r � stopProducing� s zFakeTransport.stopProducingTc C s | j |A }t||�| _g | _d S r )r/ r r: r; )r �contextFactory�beNormalr r r r �startTLS� s zFakeTransport.startTLSc C sB | j }|r g | _ d�|�S | jdur | jjr d| j_| jS dS dS )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} rI NT)r0 rJ r: r r )r r5 r r r � getOutBuffer� s zFakeTransport.getOutBufferc C sx t |t�r4| jd us J �| jjr.| j�|| � d | _| jd }| _| �|� t| tj � d S d| j_ d S | j � |� d S rM ) � isinstancer r: r r" r; rL r r � ISSLTransportr r. � dataReceived)r �buf�br r r �bufferReceived� s z FakeTransport.bufferReceivedc C rY r r r r r r �getTcpKeepAlive� rZ z FakeTransport.getTcpKeepAlivec C rY r r r r r r � getTcpNoDelay� rZ zFakeTransport.getTcpNoDelayc C rY r r r r r r �loseWriteConnection� rZ z!FakeTransport.loseWriteConnectionc C rY r r �r �enabledr r r �setTcpKeepAlive� rZ z FakeTransport.setTcpKeepAlivec C rY r r rj r r r � setTcpNoDelay� rZ zFakeTransport.setTcpNoDelay)NN)T)+r# r$ r% r( � staticmethod� itertools�countr1 �closedr9 � disconnectedr �ConnectionDoner r@ rA r: r r& r r> rC rE rF rH rL r rO rR rT rV rX rB r[ r\ r_ r` rf rg rh ri rl rm r r r r r) 6 sF r) c C � t | dd�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�r/ �r) )�clientProtocolr r r �makeFakeClient� � rx c C rt )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} Tru rv )�serverProtocolr r r �makeFakeServer ry r{ c @ s, e Zd ZdZdd� Zd dd�Zd dd�Zd S ) �IOPumpz� Utility to pump data between clients and servers for protocol testing. Perhaps this is a utility worthy of being in protocol.py? c C s" || _ || _|| _|| _|| _d S r )�client�server�clientIO�serverIO�debug)r r} r~ r r� r� r r r r s zIOPump.__init__Fc C s. d}t d�D ] }| �|�rd}q |S J d��)zk Pump until there is no more input or output. Returns whether any data was moved. Fi� Tr zToo long)�range�pump)r r� �result�xr r r �flush& s �z IOPump.flushc C s$ | j s|r td� | j�� }| j�� }| j�� | j�� | j s"|r:td� |r0tdt|� � |r:tdt|� � |rB| j�|� |rJ| j�|� |sN|rPdS | jjrp| jj sp| j s]|ratd� d| j_ d| j_| j� � dS | jjr�| jj s�| j s}|r�td� d| j_ d| j_| j� � dS dS ) zX Move data back and forth. Returns whether any data was moved. z -- GLUG --�.zC: zS: Tz* Cz* SF) r� �printr� r` r rC �reprrf r9 rr rR )r r� �sData�cDatar r r r� 6 sB z IOPump.pumpN�F)r# r$ r% r( r r� r� r r r r r| s r| FTc C s4 | � |� |� |� t|| |||�}|r|�� |S )aN 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} @return: An L{IOPump} which connects C{serverProtocol} and C{clientProtocol} and delivers bytes between them when it is pumped. @rtype: L{IOPump} )�makeConnectionr| r� )rz �serverTransportrw �clientTransportr� �greetr� r r r �connect` s & �r� c C s4 |� }| � }||�}||�} ||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} @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� ) � ServerClass� ClientClass�clientTransportFactory�serverTransportFactoryr� r� �c�s�cio�sior r r �connectedServerAndClient� s *r� c C s. | \}}}}}|\}}} } ||kr||fS dS )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�serverInterfacer r r �_factoriesShouldConnect� s � r� c @ s4 e Zd ZdZdd� Zd dd�Zee� �fdd�Zd S ) �ConnectionCompleterz� A L{ConnectionCompleter} can cause synthetic TCP connections established by L{MemoryReactor.connectTCP} and L{MemoryReactor.listenTCP} to succeed or fail. c C s || _ dS )z� Create a L{ConnectionCompleter} from a L{MemoryReactor}. @param memoryReactor: The reactor to attach to. @type memoryReactor: L{MemoryReactor} N)�_reactor)r � memoryReactorr r r r � s z ConnectionCompleter.__init__Fc C s� | j }t|j�D ]?\}}|jD ]7}t||�}|rF|j�|� |j�|� |\}}|�d�} |�d�} t | �} t | �} t | | | | |� S qqdS )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{ rx r� ) r r� r� � clientIdxr� r� � factoriesr� r� rw rz r� r� r r r � succeedOnce� s, ����z ConnectionCompleter.succeedOncec C s( | j j�d�d �| j j�d�|� dS )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 �reasonr r r �failOnce s �z ConnectionCompleter.failOnceNr� ) r# r$ r% r( r r� r r r� r r r r r� � s !r� c C s8 t � }t|dd�}t|d�}|�t�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�serverEndpointr r r �connectableEndpoint s r� )FTr� )&r( ro � 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� ITLSTransportr) rx r{ r| r� r� r� r� r� r r r r �<module> sD � H M �4 �1 >
| ver. 1.4 |
Github
|
.
| PHP 8.2.28 | Generation time: 0.02 |
proxy
|
phpinfo
|
Settings