File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_tcp_internals.cpython-38.pyc
U
W[=3 � @ s d Z ddlmZmZ ddlZddlZddlZz ddlZW n ek
rP dZY nX ddl m
Z
ddl m Z m
Z
ddlmZmZmZmZmZmZmZmZmZmZmZ ddlmZm Z ddl m Z dd l m Z m!Z! dd
l"m#Z#m$Z$ G d d � d e
�Z%G d
d� de
�Z&e$j'�(e#��s dZ)e)e%_*e)e&_*dS )z
Whitebox tests for TCP APIs.
� )�division�absolute_importN)�TestCase)�compat�log) �_ACCEPT_ERRORS� ECONNABORTED�EPERM�ENOMEM�ENFILE�EAGAIN�EMFILE�ENOBUFS� EINPROGRESS� EWOULDBLOCK�Port)�Protocol�
ServerFactory)�platform)�
maybeDeferred�
gatherResults)�reactor�
interfacesc @ sF e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Ze �
� d krBd e_ d
S )�PlatformAssumptionsTestsz4
Test assumptions about platform behaviors.
i c C s\ g | _ td k rXddlm} t|� �d }t�tj�| _t�tj|| jd f� |d | _ d S )Nr )� _listOpenFDs� � �d )
� openSockets�resourceZtwisted.internet.processr �len� getrlimit�
RLIMIT_NOFILE�originalFileLimit� setrlimit� socketLimit)�selfr ZnewLimit� r' �A/usr/lib/python3/dist-packages/twisted/test/test_tcp_internals.py�setUp$ s z PlatformAssumptionsTests.setUpc C sT | j r| j �� �� q td k rPt�tj�d }t| jd |�}t�tj||f� d S )Nr r ) r �pop�closer r! r" �minr# r$ )r& ZcurrentHardLimitZ newSoftLimitr' r' r( �tearDown4 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( r. A s z PlatformAssumptionsTests.socketc C s� | � � }|�d� |�� d }|�d� | � � }|�d� t| j�D ]V}z | � � W qD t jk
r� } z$|jd t t
fkr�W Y � q�n� W 5 d}~X Y qDX qD| � 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.r1 )r. ZbindZ getsocknameZlistenZ setblocking�ranger% �error�argsr
r �failZassertInZ
connect_exr � assertRaises�accept)r& �portZserverPortNumber�client�i�e�excr' r' r( �test_acceptOutOfFilesK s&
�z.PlatformAssumptionsTests.test_acceptOutOfFiles�win32zhWindows requires an unacceptably large amount of resources to provoke this behavior in the naive manner.N) �__name__�
__module__� __qualname__�__doc__r% r) r- r. r> r �getType�skipr' r' r' r( r s
% �r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z e �
� dkr^de _dd� Ze �
� dkrxde_dd� Zdd� Ze �
� dkr�de_d d � Zd S ) �SelectReactorTestsz7
Tests for select-specific failure conditions.
c C s g | _ g | _t�| jj� d S �N)�ports�messagesr Z addObserverr/ �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 ZremoveObserverrI r/ r rH rJ 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 ZstartListeningrH r/ )r& Z
portNumber�factoryrP rM r' r' r( r9 � s zSelectReactorTests.portc s� G �fdd�dt �}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 s e Zd ZdZ� fdd�ZdS )z9SelectReactorTests._acceptFailureTest.<locals>.FakeSocketzI
Pretend to be a socket in an overloaded system.
c s t �� t�� ���d S rG )r. r4 �os�strerrorrJ ��socketErrorNumberr' r( r8 � s �z@SelectReactorTests._acceptFailureTest.<locals>.FakeSocket.acceptN�r@ rA rB rC r8 r' rT r' r( �
FakeSocket� s rW r r1 rO r. z/Could not accept new connection ({acceptError})c s( g | ] }|� d ��ko"|� d�� k�qS )Z
log_formatZ acceptError)�get)rL �msg)�expectedErrorCode�expectedFormatr' r( rN � s �z9SelectReactorTests._acceptFailureTest.<locals>.<listcomp>z+Log event for failed accept not found in %rN)
�objectr r9 �patch�doRead�errnoZ errorcoderI �
assertGreaterr )r& rU rW rQ r9 ZmatchingMessagesr' )rZ r[ rU r( �_acceptFailureTest� s
� ��z%SelectReactorTests._acceptFailureTestc C s
| � 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.
)ra r
rJ r' r' r( �test_tooManyFilesFromAccept� s z.SelectReactorTests.test_tooManyFilesFromAcceptc C s
| � t�S )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.
)ra r rJ r' r' r( � test_noBufferSpaceFromAccept� s z/SelectReactorTests.test_noBufferSpaceFromAcceptc C s
| � t�S )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.
)ra r rJ r' r' r( � test_connectionAbortedFromAccept� s z3SelectReactorTests.test_connectionAbortedFromAcceptc C s
| � t�S )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.
)ra r rJ r' r' r( �test_noFilesFromAccept� s z)SelectReactorTests.test_noFilesFromAcceptr? z(Windows accept(2) cannot generate ENFILEc C s
| � t�S )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).
)ra r
rJ r' r' r( �test_noMemoryFromAccept� s z*SelectReactorTests.test_noMemoryFromAcceptz(Windows accept(2) cannot generate ENOMEMc 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 r1 rO c s � D ] } | � � qd S rG )r+ �r: )�clientsr' r( �closeAll s z7SelectReactorTests.test_acceptScaling.<locals>.closeAllc s( t � t jt j�} | �d� �� jf� | S )Nr1 )r. ZAF_INETZ SOCK_STREAM�connectZgetHostr9 rg )r9 r' r( rj
s z6SelectReactorTests.test_acceptScaling.<locals>.connectr N) r r Zprotocolr9 Z
addCleanuprK r/ �
numberAcceptsr^ r` � assertEqual)r& rQ ri rj r' )rh r9 r( �test_acceptScaling� s"
z%SelectReactorTests.test_acceptScalingc s� d�dg� G � ��fdd�dt �}t��D ]}��tj|� j� q*���j|� j� dg� t� }�jd|dd�}d|_ ��
|d|� � |� � �� |j d� 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 s< � d d7 < � d �kr&�� d� t�tt�t���d S )Nr r z(Maximum number of accept calls exceeded.)r6 r. r4 r rR rS �Zoself�Z acceptCallsZmaximumNumberOfAcceptsr& r' r( r8 2 s
zSSelectReactorTests.test_permissionFailure.<locals>.FakeSocketWithAcceptLimit.acceptNrV r' rq r' r( �FakeSocketWithAcceptLimit, s rr r1 rO r. r N)
r\ r3 r7 r. r4 r8 ZfailureExceptionr r9 rk r] r^ Z assertEquals)r& rr �_rQ r9 r' rq r( �test_permissionFailure# s" ��z)SelectReactorTests.test_permissionFailurez'Windows accept(2) cannot generate EPERMc s� t t�}|�tttg� tdd� |D ��d � G � fdd�dt�}t� }| j d|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 |ttjf�r|V qd S rG )�
isinstance�intr Zlong)rL r4 r' r' r( � <genexpr>\ s �zBSelectReactorTests.test_unknownSocketErrorRaise.<locals>.<genexpr>r c s e Zd ZdZ� fdd�ZdS )zYSelectReactorTests.test_unknownSocketErrorRaise.<locals>.FakeSocketWithUnknownAcceptErrorro c s t �� d��d S )Nz unknown socket error message)r. r4 rp �ZunknownAcceptErrorr' r( r8 g s �z`SelectReactorTests.test_unknownSocketErrorRaise.<locals>.FakeSocketWithUnknownAcceptError.acceptNrV r' rx r' r( � FakeSocketWithUnknownAcceptErrora s ry r r1 rO r. N)�listr �extendr r r �maxr\ r r9 r] r^ ZflushLoggedErrorsr. r4 rl r �valuer5 )r& Z knownErrorsry rQ r9 Zfailuresr' rx r( � test_unknownSocketErrorRaiseT s ��
z/SelectReactorTests.test_unknownSocketErrorRaiseN)r@ rA rB rC r) r- r9 ra rb rc rd re r rD rE rf rm rt r~ r' r' r' r( rF w s* % �*, �rF z?This test only applies to reactors that implement IReactorFDset)+rC Z
__future__r r r_ r. rR r � ImportErrorZtwisted.trial.unittestr Ztwisted.pythonr r Ztwisted.internet.tcpr r r r
r r r
r r r r Ztwisted.internet.protocolr r Ztwisted.python.runtimer Ztwisted.internet.deferr r Ztwisted.internetr r r rF Z
IReactorFDSetZ
providedByZskipMsgrE r' r' r' r( �<module> s*
4 Y