shell bypass 403
o
a b�+ � @ s� d Z ddlZddlZddlmZ ddlmZ e�d� Zddl Z ddl
Z
ddl Z e� � dd� �Z
G dd � d ej�ZG d
d � d ej�ZG d d
� d
e�ZG dd� de�Zdd� Zedkr`e�� dS dS )z&Unit tests for socket timeout feature.� N)�support)�
socket_helper�networkc C sJ t �| �� t�| |tjtj�d d W d � S 1 s w Y dS )z�Resolve an (host, port) to an address.
We must perform name resolution before timeout tests, otherwise it will be
performed by connect().
r � N)r �transient_internet�socket� getaddrinfo�AF_INET� SOCK_STREAM)�host�port� r
�:/usr/local/python-3.10/lib/python3.10/test/test_timeout.py�resolve_address s ���$�r c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )�CreationTestCasez9Test case for socket.gettimeout() and socket.settimeout()c C � t � t jt j�| _d S �N)r r r
�sock��selfr
r
r �setUp � zCreationTestCase.setUpc C � | j �� d S r �r �closer r
r
r �tearDown" � zCreationTestCase.tearDownc C s | � | j�� d d� d S )Nz timeout not disabled by default)� assertEqualr �
gettimeoutr r
r
r �testObjectCreation% s �z#CreationTestCase.testObjectCreationc C s^ | j �d� | �| j �� d� | j �d� | �| j �� d� | j �d � | �| j �� d � d S )Ng�z�Ga @� )r �
settimeoutr r r r
r
r �testFloatReturnValue* s z%CreationTestCase.testFloatReturnValuec C sP | j �d� | �t| j �� �td�� | j �d� | �t| j �� �td�� d S )N� � �?g333333@)r r! r �typer r r
r
r �testReturnType5 s z CreationTestCase.testReturnTypec C s� | j �d� | j �d� | j �d� | j �d � | �t| j jd� | �t| j jd� | �t| j jd� | �t| j jg � | �t| j ji � | �t| j jd� d S )Nr � � r
y )r r! � assertRaises� TypeErrorr r
r
r �
testTypeCheck= s z CreationTestCase.testTypeCheckc C s: | � t| jjd� | � t| jjd� | � t| jjd� d S )N���g �)r) �
ValueErrorr r! r r
r
r �testRangeCheckJ s z CreationTestCase.testRangeCheckc C s� | j �d� | j �d� | �| j �� d � | j �d� | �| j �� d� | j �d� | j �d� | �| j �� d� | j �d� | �| j �� d � d S )N�
TFr' )r r! � setblockingr r r r
r
r �testTimeoutThenBlockingP s z(CreationTestCase.testTimeoutThenBlockingc C sX | j �d� | j �d� | �| j �� d� | j �d� | j �d� | �| j �� d� d S )NFr# T)r r0 r! r r r r
r
r �testBlockingThenTimeout^ s z(CreationTestCase.testBlockingThenTimeoutN)
�__name__�
__module__� __qualname__�__doc__r r r r" r& r+ r. r1 r2 r
r
r
r r s
r c @ s* e Zd ZdZejZdd� ZeZdd� Z dS )�TimeoutTestCaseg @c C s t � �r )�NotImplementedErrorr r
r
r r t s zTimeoutTestCase.setUpc G s� | j �|� t| j |�}t|�D ]%}t�� }z||� W q ty5 } z
t�� | }W Y d}~ q;d}~ww | �d� | �||| j � | �
||d � dS )z�
Test the specified socket method.
The method is run at most `count` times and must raise a TimeoutError
within `timeout` + self.fuzz seconds.
NzTimeoutError was not raisedr$ ) r r! �getattr�range�time� monotonic� TimeoutErrorZfailZ
assertLess�fuzzZ
assertGreater) r �count�timeout�method�args�i�t1�eZdeltar
r
r �_sock_operationy s ��
z TimeoutTestCase._sock_operationN)
r3 r4 r5 r> r ZHOST� localhostr r rF r
r
r
r r7 i s r7 c @ s\ e Zd ZdZdd� Zdd� Ze�dd�dd � �Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dS )�TCPTimeoutTestCasez3TCP test case for socket.socket() timeout functionsc C s" t � t jt j�| _tdd�| _d S )Nzwww.python.org.�P )r r r
r r � addr_remoter r
r
r r � s zTCPTimeoutTestCase.setUpc C r r r r r
r
r r � r zTCPTimeoutTestCase.tearDownTz*need to replace these hosts; see bpo-35518c
C s" t dd�}t dd�}d}t�tjtj�}tj}|�|� z1z|�|� W n" ty- Y n t yF } z|j
t
j kr<d}W Y d }~nd }~ww W |� � ~n|� � ~w |rj| �
d�|d |d ||d |d �� || _t�| jd �� | �d d
d | j� W d � d S 1 s�w Y d S ) Nzblackhole.snakebite.netiZ� zwhitehole.snakebite.neti[� TFz�We didn't receive a connection reset (RST) packet from {}:{} within {} seconds, so we're unable to test connect timeout against the corresponding {}:{} (which is configured to silently drop packets).r r# g����MbP?�connect)r r r r
r ZLOOPBACK_TIMEOUTr! rK r= �OSError�errnoZ ECONNREFUSEDr ZskipTest�formatrJ r r rF )r Z blackholeZ whitehole�skipr r@ �errr
r
r �testConnectTimeout� sF
������"�z%TCPTimeoutTestCase.testConnectTimeoutc C sT t �| jd �� | j�| j� | �dddd� W d � d S 1 s#w Y d S )Nr r# � �?�recv� )r r rJ r rK rF r r
r
r �testRecvTimeout� s "�z"TCPTimeoutTestCase.testRecvTimeoutc C s, t �| j| j� | j�� | �ddd� d S )Nr# rR �accept)r � bind_portr rG �listenrF r r
r
r �testAcceptTimeout� s
z$TCPTimeoutTestCase.testAcceptTimeoutc C �p t � t jt j��&}t�|| j� |�� | j�|� � � | �
ddddd � W d � d S 1 s1w Y d S )N�d rR �send� X�@
� r r r
r rW rG rX r rK � getsocknamerF �r Zservr
r
r �testSend� � "�zTCPTimeoutTestCase.testSendc C sv t � t jt j��)}t�|| j� |�� | j�|� � � | �
ddddd |� � � W d � d S 1 s4w Y d S )Nr[ rR �sendtor] r^ r_ ra r
r
r �
testSendto s �"�z TCPTimeoutTestCase.testSendtoc C rZ )Nr[ rR �sendallr] r^ r_ ra r
r
r � testSendall rc z TCPTimeoutTestCase.testSendallN)r3 r4 r5 r6 r r �unittestZskipIfrQ rU rY rb re rg r
r
r
r rH � s
Q
rH c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �UDPTimeoutTestCasez3UDP test case for socket.socket() timeout functionsc C r r )r r �
SOCK_DGRAMr r r
r
r r r zUDPTimeoutTestCase.setUpc C r r r r r
r
r r r zUDPTimeoutTestCase.tearDownc C s$ t �| j| j� | �dddd� d S )Nr# rR �recvfromrT )r rW r rG rF r r
r
r �testRecvfromTimeout s z&UDPTimeoutTestCase.testRecvfromTimeoutN)r3 r4 r5 r6 r r rl r
r
r
r ri s
ri c C s t �d� d S )Nr )r Zrequiresr
r
r
r � setUpModule% r rm �__main__)r6 � functoolsrh �testr Z test.supportr Zis_resource_enabledZ
skip_expectedr; rM r � lru_cacher ZTestCaser r7 rH ri rm r3 �mainr
r
r
r �<module> s(
M' �