File: //lib/python3/dist-packages/twisted/test/__pycache__/test_tcp_internals.cpython-310.pyc
o
�b�2 � @ s d Z ddlZddlZddlZzddlZW n ey dZY nw ddlmZ ddlm Z m
Z
ddl m Z m
Z
ddlmZmZ ddlmZmZmZmZmZmZmZmZmZmZm Z ddl m Z dd l m Z dd
l!m"Z" ee j#�$e
� d �G d d
� d
e"��Z%ee j#�$e
� d �G dd� de"��Z&dS )z
Whitebox tests for TCP APIs.
� N)�skipIf)�
interfaces�reactor)�
gatherResults�
maybeDeferred)�Protocol�
ServerFactory) �_ACCEPT_ERRORS�EAGAIN� ECONNABORTED� EINPROGRESS�EMFILE�ENFILE�ENOBUFS�ENOMEM�EPERM� EWOULDBLOCK�Port)�log)�platform)�TestCasez?This test only applies to reactors that implement IReactorFDsetc @ sF e Zd ZdZdZdd� Zdd� Zdd� Zee �
� d kd
�d d � �Z d
S )�PlatformAssumptionsTestsz4
Test assumptions about platform behaviors.
i c C s` g | _ td ur.ddlm} t|� �d }t�tj�| _t�tj|| jd f� |d | _ d S d S )Nr )� _listOpenFDs� � �d )
� openSockets�resource�twisted.internet.processr �len� getrlimit�
RLIMIT_NOFILE�originalFileLimit� setrlimit� socketLimit)�selfr �newLimit� r' �A/usr/lib/python3/dist-packages/twisted/test/test_tcp_internals.py�setUp4 s ��z PlatformAssumptionsTests.setUpc C s\ | j r
| j �� �� | j std ur,t�tj�d }t| jd |�}t�tj||f� d S d S )Nr r ) r �pop�closer r r! �minr" r# )r% �currentHardLimit� newSoftLimitr' r' r( �tearDownF s ��z!PlatformAssumptionsTests.tearDownc C s t � � }| j�|� |S )z|
Create and return a new socket object, also tracking it so it can be
closed in the test tear down.
)�socketr �append)r% �sr' r' r( r0 R s z PlatformAssumptionsTests.socket�win32zhWindows requires an unacceptably large amount of resources to provoke this behavior in the naive manner.c C s� | � � }|�d� |�� d }|�d� | � � }|�d� t| j�D ]%}z| � � W q" tyG } z|jd t t
fv rBW Y d}~ n � d}~ww | � d� | � |�
d|f�dtf� | �t j|j�}| � |jd t t
f� dS ) z�
Test that the platform accept(2) call fails with either L{EMFILE} or
L{ENOBUFS} when there are too many file descriptors open.
)� 127.0.0.1r r � Fr Nz7Could provoke neither EMFILE nor ENOBUFS from platform.r4 )r0 �bind� getsockname�listen� setblocking�ranger$ �OSError�argsr
r �fail�assertIn�
connect_exr � assertRaises�error�accept)r% �port�serverPortNumber�client�i�e�excr' r' r( �test_acceptOutOfFiles[ s*
��
�z.PlatformAssumptionsTests.test_acceptOutOfFilesN) �__name__�
__module__� __qualname__�__doc__r$ r) r/ r0 r r �getTyperI r' r' r' r( r ) s
�r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
e e �
� dkd�dd� �Ze e �
� dkd�dd� �Ze ej�d�dkd�dd� �Ze e �
� dkd �d d � �Zd d � Zd!S )"�SelectReactorTestsz7
Tests for select-specific failure conditions.
c C s g | _ g | _t�| jj� d S �N)�ports�messagesr � addObserverr1 �r% r' r' r( r) � s zSelectReactorTests.setUpc C s" t �| jj� tdd� | jD ��S )Nc S s g | ]}t |j��qS r' )r �
stopListening)�.0�pr' r' r( �
<listcomp>� s z/SelectReactorTests.tearDown.<locals>.<listcomp>)r �removeObserverrR r1 r rQ rT r' r' r( r/ � s zSelectReactorTests.tearDownc C s& t |||d�}|�� | j�|� |S )z
Create, start, and return a new L{Port}, also tracking it so it can
be stopped in the test tear down.
�� interface)r �startListeningrQ r1 )r% �
portNumber�factoryr[ rW r' r' r( rC � s zSelectReactorTests.portc s� G �fdd�d�}t � }| jd|dd�}| �|d|� � |�� d�tj� � � �fdd �| jD �}| �t|�dd
| jf � d S ) a�
Test behavior in the face of an exception from C{accept(2)}.
On any exception which indicates the platform is unable or unwilling
to allocate further resources to us, the existing port should remain
listening, a message should be logged, and the exception should not
propagate outward from doRead.
@param socketErrorNumber: The errno to simulate from accept.
c � e Zd ZdZ� fdd�ZdS )z9SelectReactorTests._acceptFailureTest.<locals>.FakeSocketzI
Pretend to be a socket in an overloaded system.
c s t � t�� ���rP )r; �os�strerrorrT ��socketErrorNumberr' r( rB � s z@SelectReactorTests._acceptFailureTest.<locals>.FakeSocket.acceptN�rJ rK rL rM rB r' rb r' r( �
FakeSocket� s re r r4 rZ r0 z/Could not accept new connection ({acceptError})c s( g | ]}|� d ��ko|� d�� k�qS )�
log_format� acceptError)�get)rV �msg)�expectedErrorCode�expectedFormatr' r( rX � s
� �z9SelectReactorTests._acceptFailureTest.<locals>.<listcomp>z+Log event for failed accept not found in %rN) r rC �patch�doRead�errno� errorcoderR �
assertGreaterr )r% rc re r^ rC �matchingMessagesr' )rj rk rc r( �_acceptFailureTest� s
�
�z%SelectReactorTests._acceptFailureTestc C �
| � t�S )a*
C{accept(2)} can fail with C{EMFILE} when there are too many open file
descriptors in the process. Test that this doesn't negatively impact
any other existing connections.
C{EMFILE} mainly occurs on Linux when the open file rlimit is
encountered.
)rr r
rT r' r' r( �test_tooManyFilesFromAccept� �
z.SelectReactorTests.test_tooManyFilesFromAcceptc C rs )z�
Similar to L{test_tooManyFilesFromAccept}, but test the case where
C{accept(2)} fails with C{ENOBUFS}.
This mainly occurs on Windows and FreeBSD, but may be possible on
Linux and other platforms as well.
)rr r rT r' r' r( � test_noBufferSpaceFromAccept� �
z/SelectReactorTests.test_noBufferSpaceFromAcceptc C rs )z�
Similar to L{test_tooManyFilesFromAccept}, but test the case where
C{accept(2)} fails with C{ECONNABORTED}.
It is not clear whether this is actually possible for TCP
connections on modern versions of Linux.
)rr r rT r' r' r( � test_connectionAbortedFromAccept� rw z3SelectReactorTests.test_connectionAbortedFromAcceptr3 z(Windows accept(2) cannot generate ENFILEc C rs )z�
Similar to L{test_tooManyFilesFromAccept}, but test the case where
C{accept(2)} fails with C{ENFILE}.
This can occur on Linux when the system has exhausted (!) its supply
of inodes.
)rr r rT r' r' r( �test_noFilesFromAccept� ru z)SelectReactorTests.test_noFilesFromAcceptz(Windows accept(2) cannot generate ENOMEMc C rs )a�
Similar to L{test_tooManyFilesFromAccept}, but test the case where
C{accept(2)} fails with C{ENOMEM}.
On Linux at least, this can sensibly occur, even in a Python program
(which eats memory like no ones business), when memory has become
fragmented or low memory has been filled (d_alloc calls
kmem_cache_alloc calls kmalloc - kmalloc only allocates out of low
memory).
)rr r rT r' r' r( �test_noMemoryFromAccept� s
z*SelectReactorTests.test_noMemoryFromAccept�INFRASTRUCTURE�AZUREPIPELINESz(Hangs on Azure Pipelines due to firewallc s� t � }t|_| jd|dd��| ��j� g � � fdd�}| �|� �fdd�}� �|� � d�_��� | � �jd� � �|� � ��� | �
�jd� ��� | �
�jd� d S )
z�
L{tcp.Port.doRead} increases the number of consecutive
C{accept} calls it performs if all of the previous C{accept}
calls succeed; otherwise, it reduces the number to the amount
of successful calls.
r r4 rZ c s � D ]} | � � qd S rP )r+ �rE )�clientsr' r( �closeAll s
�z7SelectReactorTests.test_acceptScaling.<locals>.closeAllc s( t � t jt j�} | �d� �� jf� | S )Nr4 )r0 �AF_INET� SOCK_STREAM�connect�getHostrC r} )rC r' r( r� s z6SelectReactorTests.test_acceptScaling.<locals>.connectr N) r r �protocolrC �
addCleanuprU r1 �
numberAcceptsrm rp � assertEqual)r% r^ r r� r' )r~ rC r( �test_acceptScaling s"
z%SelectReactorTests.test_acceptScalingz'Windows accept(2) cannot generate EPERMc s� d�dg� G � ��fdd�d�}t ��D ] }��tj|� j� q���j|� j� dg� t� }�jd|dd�}d|_�� |d|� � |�
� �� |jd� d S )
z�
C{accept(2)} returning C{EPERM} is treated as a transient
failure and the call retried no more than the maximum number
of consecutive C{accept(2)} calls.
�{ r c s e Zd ZdZ� ��fdd�ZdS )zLSelectReactorTests.test_permissionFailure.<locals>.FakeSocketWithAcceptLimit��
Pretend to be a socket in an overloaded system whose
C{accept} method can only be called
C{maximumNumberOfAccepts} times.
c s6 � d d7 < � d �kr�� d� ttt�t���)Nr r z(Maximum number of accept calls exceeded.)r= r; r r` ra ��oself�� acceptCalls�maximumNumberOfAcceptsr% r' r( rB C s
zSSelectReactorTests.test_permissionFailure.<locals>.FakeSocketWithAcceptLimit.acceptNrd r' r� r' r( �FakeSocketWithAcceptLimit< s r� r4 rZ r0 r N) r: r@ r0 rA rB �failureExceptionr rC r� rl rm � assertEquals)r% r� �_r^ rC r' r� r( �test_permissionFailure2 s z)SelectReactorTests.test_permissionFailurec s� t t�}|�tttg� tdd� |D ��d � G � fdd�d�}t� }| jd|dd�}| � |d |� � |�
� | � t j
�}| �dt|�� | �|d jjd � � d
S ) z}
A C{socket.error} raised by C{accept(2)} whose C{errno} is
unknown to the recovery logic is logged.
c s s � | ]
}t |t�r|V qd S rP )�
isinstance�int)rV rA r' r' r( � <genexpr>g s � zBSelectReactorTests.test_unknownSocketErrorRaise.<locals>.<genexpr>r c r_ )zYSelectReactorTests.test_unknownSocketErrorRaise.<locals>.FakeSocketWithUnknownAcceptErrorr� c s
t � d��)Nz unknown socket error message)r; r� ��unknownAcceptErrorr' r( rB q s
z`SelectReactorTests.test_unknownSocketErrorRaise.<locals>.FakeSocketWithUnknownAcceptError.acceptNrd r' r� r' r( � FakeSocketWithUnknownAcceptErrorj s r� r r4 rZ r0 N)�listr �extendr
r r �maxr rC rl rm �flushLoggedErrorsr0 rA r� r �valuer< )r% � knownErrorsr� r^ rC �failuresr' r� r( � test_unknownSocketErrorRaise^ s �
z/SelectReactorTests.test_unknownSocketErrorRaiseN)rJ rK rL rM r) r/ rC rr rt rv rx r r rN ry rz r` �environrh r� r� r� r' r' r' r( rO � s*
)
�
)
+rO )'rM rn r` r0 r � ImportError�unittestr �twisted.internetr r �twisted.internet.deferr r �twisted.internet.protocolr r �twisted.internet.tcpr r
r r r
r r r r r r �twisted.pythonr �twisted.python.runtimer �twisted.trial.unittestr �
IReactorFDSet�
providedByr rO r' r' r' r( �<module> s6 � 4
�[ �