shell bypass 403
o
a b�� � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl Z d dl Z d dl Z zd dl!Z!W n e"y� dZ!Y nw zd dl#Z#W n e"y� dZ#Y nw ej$Z$d�%d �Z&d
Z'e�(� d kZ)zd dl*Z*W n e"y� dZ*Y nw d d
� Z+dd� Z,dd� Z-dd� Z.dd� Z/dd� Z0dd� Z1dd� Z2d d � Z3ej4d d � �Z5e,� Z6e-� Z7e.� Z8e/� Z9e0� Z:e1� Z;e2� Z<e=e
d �Z>e3� Z?e�d!�j@ZAG d"d#� d#e jB�ZCG d$d%� d%e jB�ZDG d&d'� d'eD�ZEG d(d)� d)�ZFG d*d+� d+e jB�ZGG d,d-� d-e jB�ZHG d.d/� d/�ZIG d0d1� d1eCeI�ZJG d2d3� d3eDeI�ZKe �Le>d4�G d5d6� d6eEeI��ZMG d7d8� d8eGeI�ZNG d9d:� d:eHeI�ZOe �Pe#du d;�e �Le<d<�e �Le+� d=kd>�G d?d@� d@e jBeI����ZQG dAdB� dBeJ�ZRG dCdD� dDe jBeI�ZSG dEdF� dFe jB�ZTG dGdH� dHeT�ZUG dIdJ� dJeFeTeI�ZVG dKdL� dLeUeV�ZWG dMdN� dNeT�ZXG dOdP� dPeX�ZYG dQdR� dReT�ZZG dSdT� dTeZ�Z[G dUdV� dVeZ�Z\G dWdX� dXeZ�Z]G dYdZ� dZeZ�Z^G d[d\� d\eZ�Z_G d]d^� d^e_�Z`G d_d`� d`e_�Zadadb� Zbdcdd� Zcdedf� ZdG dgdh� dhe jB�Zee �Le6di�G djdk� dke jB��Zfe �Le6di�G dldm� dmeN��Zge �Le7dn�G dodp� dpe jB��Zhe �Le8dq�G drds� dse jB��Zie �Le9dt�G dudv� dve jB��Zje �Le9dt�G dwdx� dxeO��Zke �Le;dy�G dzd{� d{e jB��Zle �Pe#du d;�e �Le<d<�G d|d}� d}e jB���Zme �Le?d~�G dd�� d�e jB��ZnG d�d�� d�eR�ZoG d�d�� d�eK�Zpe �Le>d4�G d�d�� d�eM��ZqG d�d�� d�eF�ZrG d�d�� d�er�ZsG d�d�� d�er�ZtG d�d�� d�er�ZuG d�d�� d�er�ZvG d�d�� d�er�ZwG d�d�� d�er�ZxG d�d�� d�ex�ZyG d�d�� d�ey�ZzG d�d�� d�ey�Z{G d�d�� d�er�Z|G d�d�� d�e|�Z}G d�d�� d�e|�Z~G d�d�� d�ese|�ZG d�d�� d�e jB�Z�G d�d�� d�ex�Z�G d�d�� d�ex�Z�G d�d�� d�eteveVe\�Z�ece
j
d��G d�d�� d�e{e���Z�ece
j
d��G d�d�� d�e~e���Z�ece
j
d��G d�d�� d�ee���Z�G d�d�� d�eteveVe`�Z�ece
j
d��e �Lej�d��edd�d��G d�d�� d�e{e�����Z�ece
j
d��e �Lej�d��edd�d��G d�d�� d�e~e�����Z�ece
j
d��e �Lej�d��edd�d��G d�d�� d�ee�����Z�ece
j
d��e �Lej�d��ece
d��edd�d��G d�d�� d�e�e������Z�ece
j
d��e �Lej�d��ece
d��edd�d��G d�dÄ d�ese�e������Z�e �Le>d4�G d�dń d�eteveVe]��Z�e �Le>d4�ece
j
d��G d�dDŽ d�e{e����Z�e �Le>d4�ece
j
d��G d�dɄ d�e~e����Z�e �Le>d4�ece
j
d��G d�d˄ d�ee����Z�e �Le>d4�G d�d̈́ d�eteveVea��Z�ece
j
d��e �Lej�d��e �Le>d4�edd�d��G d�dτ d�e{e������Z�ece
j
d��e �Lej�d��e �Le>d4�edd�d��G d�dф d�e~e������Z�ece
j
d��e �Lej�d��e �Le>d4�edd�d��G d�dӄ d�ee������Z�ece
j
d��e �Lej�d��e �Le>d4�ece
d��edd�d��G d�dՄ d�e�e�������Z�ece
j
d��e �Lej�d��e �Le>d4�ece
d��edd�d��G d�dׄ d�ese�e�������Z�G d�dل d�eweWe[�Z�ece
j
d��G d�dۄ d�eze���Z�ece
j
d��G d�d݄ d�e~e}e���Z�ece
j
d��G d�d߄ d�ee}e���Z�G d�d� d�eueweWe^�Z�ece
j
d��e �Pe)d�edd�d�d�G d�d� d�eze�����Z�ece
j
d��e �Pe)d�edd�d�d�G d�d� d�e~e}e�����Z�ece
j
d��e �Pe)d�edd�d�d�G d�d� d�ee}e�����Z�G d�d� d�eweWeY�Z�ece
j
d��ece
d�G d�d� d�eze����Z�ece
j
d��ece
d�G d�d� d�e~e}e����Z�ece
j
d��ece
d�G d�d� d�ee}e����Z�ece
j
d�d��ece
d�d�d��G d�d�� d�e�e����Z�ece
j
d�d��ece
d�d�d��G d�d�� d�ese�e����Z�G d�d�� d��Z�eced��e �Le=ed���p{e=ed���d �G �d�d� �de�e\���Z�eced��e �Le=ed���p�e=ed���d �G �d�d� �de�eFeUe[���Z�G �d�d� �deJ�Z�G �d�d� �deS�Z�G �d �d
� �d
eJ�Z�G �d �d � �d eR�Z�G �d
�d� �de��Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d � �d e�eo�Z�G �d �d � �d e jB�Z�G �d �d � �d eCeI�Z�G �d!�d"� �d"eCeI�Z�G �d#�d$� �d$eC�Z�G �d%�d&� �d&eD�Z�e �Le>d4�G �d'�d(� �d(eE��Z�G �d)�d*� �d*e jB�Z�e �Lej�d+k�d,�G �d-�d.� �d.e jB��Z�e �Le=e
d��d/�G �d0�d1� �d1e jB��Z�G �d2�d3� �d3eR�Z��d4Z��d5Z��d6Zd7�d8� Z�e �Leà �d9�G �d:�d;� �d;e jB��Z�e �Leà �d9�G �d<�d=� �d=e jBeI��Z�G �d>�d?� �d?eJ�Z�G �d@�dA� �dAe jB�Z�e �Le=e
�dB��dC�G �dD�dE� �dEe jB��Z�e �LejɐdFk�dG�e �Le!�dH�G �dI�dJ� �dJeC���Z�G �dK�dL� �dLeJ�Z�e �Le=e�dM��dN�G �dO�dP� �dPe˃�Z�e �Le:�dQ�G �dR�dS� �dSe jB��Z�e �Lej�dTk�dU�G �dV�dW� �dWe jB��Z�e �Lej�ϐdX��dY�G �dZ�d[� �d[e jB��Z�G �d\�d]� �d]e jB�Z�G �d^�d_� �d_e jB�Z�ece
�d`�ece
�da�ece
d�G �db�dc� �dce jB����ZӐdd�de� Z�eՐdfk�r�e �֡ dS dS (g � N)�support)� os_helper)�
socket_helper)�threading_helper)�proxyFu Michael Gilfix was hereሴ
�utf-8�� �AIXc C s~ t d u rd S ttd�s
d S z tdd��} t �| tjd�}W d � n1 s&w Y W n
ty6 Y d S w t�d|�d S )N� IOCTL_VM_SOCKETS_GET_LOCAL_CIDz
/dev/vsock�rbz �Ir ) �fcntl�hasattr�socket�open�ioctlr
�OSError�struct�unpack)�f�r� r �9/usr/local/python-3.10/lib/python3.10/test/test_socket.py�get_cid4 s
�� �r c C �> z t � t jt jt j�} W n ttfy Y dS w | �� dS )z5Check whether CAN sockets are supported on this host.FT)r �PF_CAN�SOCK_RAW�CAN_RAW�AttributeErrorr �close��sr r r �_have_socket_canA � �r" c C r )z;Check whether CAN ISOTP sockets are supported on this host.FT)r r �
SOCK_DGRAM� CAN_ISOTPr r r r r r r �_have_socket_can_isotpK r# r& c C r )z;Check whether CAN J1939 sockets are supported on this host.FT)r r r$ � CAN_J1939r r r r r r r �_have_socket_can_j1939U r# r( c C �< z t � t jt jd�} W n ttfy Y dS w | �� dS )z5Check whether RDS sockets are supported on this host.r FT)r �PF_RDS�SOCK_SEQPACKETr r r r r r r �_have_socket_rds_ � �r, c C r) )z8Check whether AF_ALG sockets are supported on this host.r FT)r �AF_ALGr+ r r r r r r r �_have_socket_algi r- r/ c C r) )z<Check whether AF_QIPCRTR sockets are supported on this host.r FT)r �
AF_QIPCRTRr$ r r r r r r r �_have_socket_qipcrtrs r- r1 c C s t � du} | S )z:Check whether AF_VSOCK sockets are supported on this host.N)r )�retr r r �_have_socket_vsock} s
r3 c C r )z>Check whether AF_BLUETOOTH sockets are supported on this host.FT)r � AF_BLUETOOTH� SOCK_STREAM�BTPROTO_RFCOMMr r r r r r r �_have_socket_bluetooth� s �r7 c c s8 � t �� }zt �| � d V W t �|� d S t �|� w �N)r �getdefaulttimeout�setdefaulttimeout)�timeoutZ old_timeoutr r r �socket_setdefaulttimeout� s �
r< �IPPROTO_UDPLITE�ic @ � e Zd Zdd� Zdd� ZdS )�
SocketTCPTestc C s. t � t jt j�| _t�| j�| _| j�� d S r8 )r �AF_INETr5 �servr � bind_port�port�listen��selfr r r �setUp� s zSocketTCPTest.setUpc C � | j �� d | _ d S r8 �rB r rF r r r �tearDown� �
zSocketTCPTest.tearDownN��__name__�
__module__� __qualname__rH rK r r r r r@ � s r@ c @ r? )�
SocketUDPTestc C s$ t � t jt j�| _t�| j�| _d S r8 )r rA r$ rB r rC rD rF r r r rH � � zSocketUDPTest.setUpc C rI r8 rJ rF r r r rK � rL zSocketUDPTest.tearDownNrM r r r r rQ � s rQ c @ � e Zd Zdd� ZdS )�SocketUDPLITETestc C s( t � t jt jt j�| _t�| j�| _d S r8 )r rA r$ r= rB r rC rD rF r r r rH � s zSocketUDPLITETest.setUpN)rN rO rP rH r r r r rT � � rT c s8 e Zd ZdZ� fdd�Z� fdd�Z� fdd�Z� ZS )�ThreadSafeCleanupTestCasez�Subclass of unittest.TestCase with thread-safe cleanup methods.
This subclass protects the addCleanup() and doCleanups() methods
with a recursive lock.
c s t � j|i |�� t�� | _d S r8 )�super�__init__� threading�RLock�
_cleanup_lock�rG �args�kwargs�� __class__r r rX � � z"ThreadSafeCleanupTestCase.__init__c �< | j � t� j|i |��W d � S 1 sw Y d S r8 )r[ rW �
addCleanupr\ r_ r r rc � � $�z$ThreadSafeCleanupTestCase.addCleanupc rb r8 )r[ rW �
doCleanupsr\ r_ r r re � rd z$ThreadSafeCleanupTestCase.doCleanups)rN rO rP �__doc__rX rc re �
__classcell__r r r_ r rV � s
rV c @ sL e Zd ZdZdZdZ dZe�e�Z dZ
e
de�e
�d 7 Z
dd � Z d
S ) �
SocketCANTestz�To be able to run this test, a `vcan0` CAN interface can be created with
the following commands:
# modprobe vcan
# ip link add dev vcan0 type vcan
# ip link set up vcan0
�vcan0� z=IB3x8sz@3I4l2I�x� c C s` t � t jt jt j�| _| �| jj� z | j�| jf� W d S t y/ | �
d| j � Y d S w �N�%network interface `%s` does not exist) r r r r r! rc r �bind� interfacer �skipTestrF r r r rH s ��zSocketCANTest.setUpN) rN rO rP rf rp �bufsize�
can_frame_fmtr �calcsizeZcan_frame_size�bcm_cmd_msg_fmtrH r r r r rh � s
rh c @ s e Zd ZdZdZdd� ZdS )�
SocketRDSTestz\To be able to run this test, the `rds` kernel module must be loaded:
# modprobe rds
� c C sV t � t jt jd�| _| �| jj� z
t�| j�| _W d S t y* | �
d� Y d S w )Nr zunable to bind RDS socket) r r* r+ rB rc r r rC rD r rq rF r r r rH s �zSocketRDSTest.setUpN)rN rO rP rf rr rH r r r r rv s rv c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�ThreadableTesta� Threadable Test class
The ThreadableTest class makes it easy to create a threaded
client/server pair from an existing unit test. To create a
new threaded class from an existing unit test, use multiple
inheritance:
class NewClass (OldClass, ThreadableTest):
pass
This class defines two new fixture functions with obvious
purposes for overriding:
clientSetUp ()
clientTearDown ()
Any new test functions within the class must then define
tests in pairs, where the test name is preceded with a
'_' to indicate the client portion of the test. Ex:
def testFoo(self):
# Server portion
def _testFoo(self):
# Client portion
Any exceptions raised by the clients during their tests
are caught and transferred to the main thread to alert
the testing framework.
Note, the server setup function cannot call any blocking
functions that rely on the client thread during setup,
unless serverExplicitReady() is called just before
the blocking call (such as in setting up a client/server
connection and performing the accept() in setUp().
c C s | j | _| j| _ d S r8 )rH �_ThreadableTest__setUp�_setUprF r r r rX F s zThreadableTest.__init__c C s | j �� dS )z�This method allows the server to explicitly indicate that
it wants the client thread to proceed. This is useful if the
server is about to execute a blocking routine that is
dependent upon the client thread during its setup routine.N)� server_ready�setrF r r r �serverExplicitReadyK s z"ThreadableTest.serverExplicitReadyc s t �� � _� j�� � �� jjd d d � t�� � _t�� � _ t�� � _
t � d�� _ d� _
� fdd�}� �|� � �� }|�d�}||d d � }t� d| �}t�� j|f�� _zz� �� W n d� _
� W � j�� n� j�� w � j �� � �� j
j� d S )N� Fc s � j �� r
� j �� �d S r8 )�queue�qsize�getr rF r r �raise_queued_exception] s
�z5ThreadableTest._setUp.<locals>.raise_queued_exception�.�_T)r Zwait_threads_exitZ wait_threads� __enter__rc �__exit__rY �Eventr{ � client_ready�doner �Queue�server_crashed�id�rfind�getattr�thread�start_new_thread� clientRunZ
client_threadry r| �wait)rG r� Z
methodnamer> Z test_methodr rF r rz R s4
� �
zThreadableTest._setUpc
C s | j �� z1z| �� W n# ty/ } z| j�|� | �� W Y d }~W | j�� d S d }~ww W | j�� n| j�� w | j rF| �� d S t
|d�sOt d��z,z|� W n tym } z | j�|� W Y d }~n d }~ww W | �� d S W | �� d S | �� w )N�__call__z%test_func must be a callable function) r{ r� � clientSetUp�
BaseExceptionr �put�clientTearDownr� r| r� r � TypeError)rG Z test_func�er r r r� t s6
���
��� �zThreadableTest.clientRunc C � t d��)Nz clientSetUp must be implemented.)�NotImplementedErrorrF r r r r� � � zThreadableTest.clientSetUpc C s | j �� t�� d S r8 )r� r| r� �exitrF r r r r� � �
z ThreadableTest.clientTearDownN)
rN rO rP rf rX r} rz r� r� r� r r r r rx s %" rx c @ �& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedTCPSocketTest�runTestc C � t j| |d� t�| � d S �N)�
methodName�r@ rX rx �rG r� r r r rX � � z ThreadedTCPSocketTest.__init__c C � t � t jt j�| _d S r8 )r rA r5 �clirF r r r r� � � z!ThreadedTCPSocketTest.clientSetUpc C � | j �� d | _ t�| � d S r8 �r� r rx r� rF r r r r� � �
z$ThreadedTCPSocketTest.clientTearDownN�r� �rN rO rP rX r� r� r r r r r� � �
r� c @ r� )
�ThreadedUDPSocketTestr� c C r� r� )rQ rX rx r� r r r rX � r� z ThreadedUDPSocketTest.__init__c C r� r8 )r rA r$ r� rF r r r r� � r� z!ThreadedUDPSocketTest.clientSetUpc C r� r8 r� rF r r r r� � r� z$ThreadedUDPSocketTest.clientTearDownNr� r� r r r r r� � r� r� z'UDPLITE sockets required for this test.c @ r� )
�ThreadedUDPLITESocketTestr� c C r� r� )rT rX rx r� r r r rX � r� z"ThreadedUDPLITESocketTest.__init__c C s t � t jt jt j�| _d S r8 )r rA r$ r= r� rF r r r r� � � z%ThreadedUDPLITESocketTest.clientSetUpc C r� r8 r� rF r r r r� � r� z(ThreadedUDPLITESocketTest.clientTearDownNr� r� r r r r r� � s
r� c @ r� )
�ThreadedCANSocketTestr� c C r� r� )rh rX rx r� r r r rX � r� z ThreadedCANSocketTest.__init__c C sB t � t jt jt j�| _z | j�| jf� W d S ty Y d S w r8 )r r r r r� ro rp r rF r r r r� � s �z!ThreadedCANSocketTest.clientSetUpc C r� r8 r� rF r r r r� � r� z$ThreadedCANSocketTest.clientTearDownNr� r� r r r r r� � s
r� c @ r� )
�ThreadedRDSSocketTestr� c C r� r� )rv rX rx r� r r r rX � r� z ThreadedRDSSocketTest.__init__c C sL t � t jt jd�| _z| j�tdf� | j�� | _W d S ty% Y d S w �Nr ) r r* r+ r� ro �HOST� getsockname�cli_addrr rF r r r r� � s �z!ThreadedRDSSocketTest.clientSetUpc C r� r8 r� rF r r r r� � r� z$ThreadedRDSSocketTest.clientTearDownNr� r� r r r r r� � s
r� �
need fcntlz%VSOCK sockets required for this test.� z-This test can only be run on a virtual guest.c @ �6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )� ThreadedVSOCKSocketStreamTestr� c C � t jj| |d� t�| � d S r� ��unittest�TestCaserX rx r� r r r rX � � z&ThreadedVSOCKSocketStreamTest.__init__c C sh t � t jt j�| _| �| jj� | j�t jtf� | j� � | �
� | j� � \| _ | _
| �| j j� d S r8 )r �AF_VSOCKr5 rB rc r ro �VMADDR_CID_ANY� VSOCKPORTrE r} �accept�conn�connaddrrF r r r rH � s
z#ThreadedVSOCKSocketStreamTest.setUpc C sD t �d� t�tjtj�| _| �| jj� t� }| j� |t
f� d S )N皙�����?) �time�sleepr r� r5 r� rc r r �connectr� )rG Zcidr r r r� � s
z)ThreadedVSOCKSocketStreamTest.clientSetUpc C � | j �d�}| �|t� d S �N� )r� �recv� assertEqual�MSG�rG �msgr r r �
testStream � z(ThreadedVSOCKSocketStreamTest.testStreamc C � | j �t� | j �� d S r8 �r� �sendr� r rF r r r � _testStream � z)ThreadedVSOCKSocketStreamTest._testStreamNr� �rN rO rP rX rH r� r� r� r r r r r� � s
r� c @ s: e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
S )�SocketConnectedTestz�Socket tests for client-server connection.
self.cli_conn is a client socket connected to the server. The
setUp() method guarantees that it is connected to the server.
r� c C � t j| |d� d S r� )r� rX r� r r r rX � z SocketConnectedTest.__init__c C s* t �| � | �� | j�� \}}|| _d S r8 )r� rH r} rB r� �cli_conn�rG r� �addrr r r rH �
zSocketConnectedTest.setUpc C r� r8 )r� r r� rK rF r r r rK r� z SocketConnectedTest.tearDownc C s( t �| � | j�t| jf� | j| _d S r8 )r� r� r� r� r� rD � serv_connrF r r r r� ! s
z SocketConnectedTest.clientSetUpc C r� r8 )r� r r� r� rF r r r r� & r� z"SocketConnectedTest.clientTearDownNr� ) rN rO rP rf rX rH rK r� r� r r r r r�
s
r� c @ r� )�SocketPairTestr� c C r� r� r� r� r r r rX - r� zSocketPairTest.__init__c C s t �� \| _| _d S r8 )r �
socketpairrB r� rF r r r rH 1 � zSocketPairTest.setUpc C rI r8 rJ rF r r r rK 4 rL zSocketPairTest.tearDownc C � d S r8 r rF r r r r� 8 � zSocketPairTest.clientSetUpc C r� r8 r� rF r r r r� ; r� z SocketPairTest.clientTearDownNr� )rN rO rP rX rH rK r� r� r r r r r� + s
r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �SocketTestBasez�A base class for socket tests.
Subclasses must provide methods newSocket() to return a new socket
and bindSock(sock) to bind it to an unused address.
Creates a socket self.serv and sets self.serv_addr to its address.
c C � | � � | _| �� d S r8 )� newSocketrB �
bindServerrF r r r rH P r� zSocketTestBase.setUpc C � | � | j� | j�� | _dS )z9Bind server socket and set self.serv_addr to its address.N)�bindSockrB r� � serv_addrrF r r r r� T � zSocketTestBase.bindServerc C rI r8 rJ rF r r r rK Y rL zSocketTestBase.tearDownN)rN rO rP rf rH r� rK r r r r r� G s
r� c s e Zd ZdZ� fdd�Z� ZS )�SocketListeningTestMixinz%Mixin to listen on the server socket.c s t � �� | j�� d S r8 )rW rH rB rE rF r_ r r rH a �
z SocketListeningTestMixin.setUp)rN rO rP rf rH rg r r r_ r r� ^ s r� c s@ e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Z� Z S ) �ThreadedSocketTestMixinz�Mixin to add client socket and allow client/server tests.
Client socket is self.cli and its address is self.cli_addr. See
ThreadableTest for usage information.
c s t � j|i |�� t�| � d S r8 )rW rX rx r\ r_ r r rX n ra z ThreadedSocketTestMixin.__init__c C r� r8 )�newClientSocketr� �
bindClientrF r r r r� r r� z#ThreadedSocketTestMixin.clientSetUpc C s | � � S )z&Return a new socket for use as client.)r� rF r r r r� v s z'ThreadedSocketTestMixin.newClientSocketc C r� )z8Bind client socket and set self.cli_addr to its address.N)r� r� r� r� rF r r r r� z r� z"ThreadedSocketTestMixin.bindClientc C r� r8 r� rF r r r r� r� z&ThreadedSocketTestMixin.clientTearDown)
rN rO rP rf rX r� r� r� r� rg r r r_ r r� f s r� c sD e Zd ZdZ� fdd�Z� fdd�Z� fdd�Z� fdd �Z� ZS )
�ConnectedStreamTestMixinz�Mixin to allow client/server stream tests with connected client.
Server's socket representing connection to client is self.cli_conn
and client's connection to server is self.serv_conn. (Based on
SocketConnectedTest.)
c s* t � �� | �� | j�� \}}|| _d S r8 )rW rH r} rB r� r� r� r_ r r rH � r� z ConnectedStreamTestMixin.setUpc s | j �� d | _ t� �� d S r8 )r� r rW rK rF r_ r r rK � r� z!ConnectedStreamTestMixin.tearDownc s$ t � �� | j�| j� | j| _d S r8 )rW r� r� r� r� r� rF r_ r r r� � s
z$ConnectedStreamTestMixin.clientSetUpc s6 z
| j �� d | _ W n ty Y nw t� �� d S r8 )r� r r rW r� rF r_ r r r� � s
�z'ConnectedStreamTestMixin.clientTearDown) rN rO rP rf rH rK r� r� rg r r r_ r r� � s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�UnixSocketTestBasez(Base class for Unix-domain socket tests.c s( t �� | _| �tj| j� t� �� d S r8 )�tempfile�mkdtemp�dir_pathrc �os�rmdirrW rH rF r_ r r rH � s
zUnixSocketTestBase.setUpc C s, t j| jd�}t�||� | �tj|� d S )N)�dir)r� �mktempr� r �bind_unix_socketrc r �unlink)rG �sock�pathr r r r� � s zUnixSocketTestBase.bindSock)rN rO rP rf rH r� rg r r r_ r r� � s r� c @ � e Zd ZdZdd� ZdS )�UnixStreamBasez-Base class for Unix-domain SOCK_STREAM tests.c C � t � t jt j�S r8 )r �AF_UNIXr5 rF r r r r� � � zUnixStreamBase.newSocketN�rN rO rP rf r� r r r r r � � r c s, e Zd ZdZeZ� fdd�Zdd� Z� ZS )� InetTestBasez!Base class for IPv4 socket tests.c s t � �� | jd | _d S �Nr~ )rW rH r� rD rF r_ r r rH � s
zInetTestBase.setUpc C s t j|| jd� d S )N��host)r rC r �rG r r r r r� � r� zInetTestBase.bindSock) rN rO rP rf r� r rH r� rg r r r_ r r � s
r c @ r )� TCPTestBasez#Base class for TCP-over-IPv4 tests.c C r r8 )r rA r5 rF r r r r� � r zTCPTestBase.newSocketNr r r r r r � r
r c @ r )� UDPTestBasez#Base class for UDP-over-IPv4 tests.c C r r8 )r rA r$ rF r r r r� � r zUDPTestBase.newSocketNr r r r r r � r
r c @ r )�UDPLITETestBasez'Base class for UDPLITE-over-IPv4 tests.c C � t � t jt jt j�S r8 )r rA r$ r= rF r r r r� � r� zUDPLITETestBase.newSocketNr r r r r r � r
r c @ r )�SCTPStreamBasez;Base class for SCTP tests in one-to-one (SOCK_STREAM) mode.c C r r8 )r rA r5 � IPPROTO_SCTPrF r r r r� � s �zSCTPStreamBase.newSocketNr r r r r r � r
r c @ s e Zd ZdZejZdS )�
Inet6TestBasez!Base class for IPv6 socket tests.N)rN rO rP rf r �HOSTv6r r r r r r � s
r c @ r )� UDP6TestBasez#Base class for UDP-over-IPv6 tests.c C r r8 )r �AF_INET6r$ rF r r r r� � r zUDP6TestBase.newSocketNr r r r r r � r
r c @ r )�UDPLITE6TestBasez'Base class for UDPLITE-over-IPv6 tests.c C r r8 )r r r$ r= rF r r r r� � r� zUDPLITE6TestBase.newSocketNr r r r r r � r
r c s* dd� � � �fdd�}dd� }| r|S |S )a Skip decorated test if condition is true, add client_skip decorator.
If the decorated object is not a class, sets its attribute
"client_skip" to a decorator which will return an empty function
if the test is to be skipped, or the original function if it is
not. This can be used to avoid running the client part of a
skipped test when using ThreadableTest.
c _ r� r8 r �r] r^ r r r � client_pass r� z%skipWithClientIf.<locals>.client_passc s* t ���| �}t| t�s� fdd�|_|S )Nc s � S r8 r �r )r r r �<lambda>
� z3skipWithClientIf.<locals>.skipdec.<locals>.<lambda>)r� �skip�
isinstance�type� client_skip)�objZretval�r �reasonr r �skipdec s
z!skipWithClientIf.<locals>.skipdecc S s" t | t�st| d�sdd� | _| S )Nr# c S s | S r8 r r r r r r r z5skipWithClientIf.<locals>.noskipdec.<locals>.<lambda>)r! r" r r# �r$ r r r � noskipdec s
z#skipWithClientIf.<locals>.noskipdecr )Z conditionr& r' r) r r% r �skipWithClientIf� s r* c s0 � fdd�|D �}t |dd�dd� |D �� �S )z�Skip decorated test if obj is missing any of the given attributes.
Sets client_skip attribute as skipWithClientIf() does.
c s g | ] }t � |�s|�qS r )r ��.0�namer( r r �
<listcomp> � z requireAttrs.<locals>.<listcomp>� don't have �, c s � � | ]}|V qd S r8 r r+ r r r � <genexpr> � � z requireAttrs.<locals>.<genexpr>)r* �join)r$ Z
attributes�missingr r( r � requireAttrs s �r7 c
G s� d}dd� | D �}|rdd� dd� |D �� }n)dd� | D �}ztj|� }W n ty< } z
t|�}W Y d}~n d}~ww |�� t|dud �d� d
d� | D ��|��S ) a Skip decorated test if a socket cannot be created with given arguments.
When an argument is given as a string, will use the value of that
attribute of the socket module, or skip the test if it doesn't
exist. Sets client_skip attribute as skipWithClientIf() does.
Nc S s$ g | ]}t |t�rtt|�s|�qS r )r! �strr r �r, r$ r r r r. % s
�
�z!requireSocket.<locals>.<listcomp>r0 r1 c s r2 r8 r r+ r r r r3 ( r4 z requireSocket.<locals>.<genexpr>c S s$ g | ]}t |t�rtt|�n|�qS r )r! r8 r� r r9 r r r r. * s �z can't create socket({0}): {1}c s s � | ]}t |�V qd S r8 )r8 �r, �or r r r3 6 s � )r5 r r r8 r r* �format)r] �errr6 Zcallargsr! r� r r r �
requireSocket s( �����r> c @ s� e Zd Zdd� Zdd� Ze�edud�dd� �Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
e�ejdkd�dd� �Ze�ejdkd�e�ejd�dd� ��Zd d � Zd d � Zd d!� Ze�eed"�d#�e�eed$�d%�d&d'� ��Ze�eed(�d)�d*d+� �Z e�eed,�d-�d.d/� �Z e�eed0�d1�d2d3� �Z e�eed4�d5�d6d7� �Z d8d9� Z d:d;� Z!e"j#d<d=� �Z$d>d?� Z%d@dA� Z&e�eedB�dC�dDdE� �Z'e�eedF�dG�dHdI� �Z(e�eedF�dG�dJdK� �Z)e�eedL�dM�dNdO� �Z*e�eedL�dM�dPdQ� �Z+dRdS� Z,dTdU� Z-dVdW� Z.dXdY� Z/dZd[� Z0d\d]� Z1d^d_� Z2e�ejdkd�d`da� �Z3e�ejdkd�e�eedb�dc�ddde� ��Z4dfdg� Z5dhdi� Z6e�e"�7dj�dk�dldm� �Z8dndo� Z9dpdq� Z:drds� Z;dtdu� Z<dvdw� Z=dxdy� Z>dzd{� Z?d|d}� Z@d~d� ZAd�d�� ZBd�d�� ZCe"j#d�d�� �ZDe�ejd��d�d�� �ZEe�ejd��d�d�� �ZFe�ejd��e�Gejd�kd��e�GeHd��e�eed(�d��d�d�� ����ZIe�ejd��e�ejd�kd��d�d�� ��ZJe�ejd��e�Gejd�kd��e�GeHd��e�eed(�d��d�d�� ����ZKe�ejd��e�ejd�kd��d�d�� ��ZLd�d�� ZMd�d�� ZNd�d�� ZOe�eed��d��d�d�� �ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTd�d�� ZUe�Gejdkd��d�d�� �ZVd�d�� ZWd�d�� ZXdS )��GeneralModuleTestsc C s< dd l }| �tj|ju � t�� }| �|tj� |�� d S r� )�_socket�
assertTruer �
SocketType�assertIsInstancer )rG r@ r! r r r � test_SocketType_is_socketobject> s
z2GeneralModuleTests.test_SocketType_is_socketobjectc C s� t � t jt j�}|�S | �d|�� t|�� | �dt j t|�� | �dt j t|�� | �dt|�� | �dt|�� |�d� | �dt|�� | �t|� � �t|�� W d � n1 s`w Y | �dt|�� | �dt|�� d S ) Nzfd=%iz family=%sztype=%szproto=0�raddr�� 127.0.0.1r �laddrz[closed])
r rA r5 �assertIn�fileno�repr� assertNotInro r8 r� �rG r! r r r � test_reprE s
� z GeneralModuleTests.test_reprNzneed _socket modulec C sz t �t jt j�}zd|�� |j|j|jf }| �t |�|� W |�
� n|�
� w d|j|j|jf }| �t |�|� d S )Nz4<socket object, fd=%s, family=%s, type=%s, proto=%s>z4<socket object, fd=-1, family=%s, type=%s, proto=%s>) r@ r rA r5 rJ �familyr" �protor� rK r )rG r! �expectedr r r �test_csocket_reprS s ��z$GeneralModuleTests.test_csocket_reprc C s� t � t jt j��}t|�}| �|�� |�� � W d � n1 s!w Y d }t�� z|�� W n
ty< Y d S w | � d� d S )NzSocket proxy still exists)
r rA r5 r r� rJ r �
gc_collect�ReferenceError�fail)rG r! �pr r r � test_weakref` s � �z GeneralModuleTests.test_weakrefc C s� d}| j t|d d�� t�1 sw Y | j t|d d�� tj�1 s&w Y | j t|d d�� tj�1 s;w Y d S )Nz$Error raising socket exception (%s).r �r� z
socket.herrorzsocket.gaierror)� assertRaisesr r �herror�gaierrorr� r r r �testSocketErrorm s ���z"GeneralModuleTests.testSocketErrorc C s� t � t jt j�}| �|j� |�d� |�� }| �t��}|� d|� W d � n1 s-w Y | �
t |j �d� | �t��}|� d|� W d � n1 sQw Y | �
t |j �d� | �t��}|� dd � W d � n1 suw Y | �
dt |j �� | �t��}|� dd|� W d � n1 s�w Y | �
t |j �d� | �t��}|� dd|� W d � n1 s�w Y | �
t |j �d� | �t��}|� ddd � W d � n1 s�w Y | �
dt |j �� | �t��}|� dd |� W d � n 1 �s
w Y | �t��}|� dd d � W d � n 1 �s'w Y | �t��
}|� d� W d � n 1 �sBw Y | �
d
t |j �� | �t��}|� dd|d � W d � n 1 �siw Y | �
d t |j �� d S )
N)� r u ☠z*a bytes-like object is required, not 'str'y @z.a bytes-like object is required, not 'complex'� fooz not NoneTyper �barz (1 given)� z (4 given))r rA r$ rc r ro r� rY r� �sendtor� r8 � exceptionrI )rG r! Zsockname�cmr r r �testSendtoErrorsw sd
� � � � � � � � � � � � � �z#GeneralModuleTests.testSendtoErrorsc C s@ t j t jr t j t j t j t j t j t j t j t j
d S r8 ) r rA �has_ipv6r r5 r$ r �SOCK_RDMr+ �
SOL_SOCKET� SO_REUSEADDRrF r r r �testCrucialConstants� s
z'GeneralModuleTests.testCrucialConstantsc C s t j t j t jrt j d S d S r8 )r � IPPROTO_TCP� IPPROTO_UDPre � IPPROTO_IPV6rF r r r �testCrucialIpProtoConstants� s
�z.GeneralModuleTests.testCrucialIpProtoConstants�nt�Windows specificc C �4 t j t j t j t j t j t j t j t j d S r8 ) r ZIPPROTO_ICLFXBMZ
IPPROTO_STZ IPPROTO_CBTZ IPPROTO_IGPZ IPPROTO_RDPZ IPPROTO_PGMZ IPPROTO_L2TPr rF r r r � testWindowsSpecificConstants� s
z/GeneralModuleTests.testWindowsSpecificConstants�darwin�macOS specific test�IPv6 required for this testc C s0 h d�}|D ]}| � tt|�d|� d�� qd S )N> �IPV6_RECVPATHMTU�IPV6_RECVTCLASS�IPV6_RECVPKTINFO�IPV6_RECVRTHDR�IPV6_RECVHOPOPTS�
IPV6_HOPLIMIT�IPV6_RECVDSTOPTS�IPV6_RTHDRDSTOPTS�
IPV6_CHECKSUMZIPV6_USE_MIN_MTU�
IPV6_RTHDR�IPV6_RECVHOPLIMIT� IPV6_TCLASS� IPV6_DSTOPTS� IPV6_NEXTHOP� IPV6_PATHMTU�
IPV6_DONTFRAG�IPV6_RTHDR_TYPE_0� IPV6_HOPOPTS� IPV6_PKTINFOz Missing RFC3542 socket option '�')rA r r )rG Zopts�optr r r �test3542SocketOptions� s ��z(GeneralModuleTests.test3542SocketOptionsc C s� t �� }zt �|�}W n ty | �d� Y nw | �|�d�dkd� z
t �|�\}}}W n ty= | �d� Y nw ||g| }t �|�}||vrZ| � d|t
|�f � d S d S )N�name lookup failurer� r zError resolving host to ip.z=Error testing host resolution mechanisms. (fqdn: %s, all: %s)) r � gethostname�
gethostbynamer rq rA �find�
gethostbyaddr�getfqdnrU rK )rG �hostnameZipZhname�aliases�ipaddrsZall_host_namesZfqhnr r r �testHostnameRes� s" � �
�z"GeneralModuleTests.testHostnameResc C sL t jddfD ] }| �t�|�|� qt jfD ]
}| �|t�|�d � qd S )Nz10.0.0.1�255.255.255.255r� )r ZHOSTv4r� r r� rI r� )rG r� r r r r �test_host_resolution� s
�z'GeneralModuleTests.test_host_resolutionc C s� d}dD ]:}| j t|d��
t�|� W d � n1 s w Y | j t|d��
t�|� W d � n1 s9w Y qd S )Nz[resolving an invalid IP address did not raise OSError; can be caused by a broken DNS server)z0.1.1.~1z1+.1.1.1z::1qz::1::2z1:1:1:1:1:1:1:1:1rX )rY r r r� r� )rG Z explanationr� r r r � test_host_resolution_bad_address� s � � ���z3GeneralModuleTests.test_host_resolution_bad_address� sethostnamez test needs socket.sethostname()r� z test needs socket.gethostname()c
C s� t �� }zt �d� W n ty* } z|jtjkr | �d� n� W Y d }~nd }~ww z | �t �� d� t �d� | �t �� d� W t �|� d S t �|� w )N�newztest should be run as roots barr_ )r r� r� r �errno�EPERMrq r� )rG Zoldhnr� r r r �test_sethostname s ���
z#GeneralModuleTests.test_sethostname� if_nameindexz$socket.if_nameindex() not available.c C s� t �� }|D ]8\}}| �|t� | �|t� | �|d� t �|�}| �|t� | �||� t �|�}| �|t� | �||� qd S r� ) r r� rC �intr8 �
assertGreater�if_nametoindexr� �if_indextoname)rG Z
interfaces�indexr- �_index�_namer r r �testInterfaceNameIndex s
�z)GeneralModuleTests.testInterfaceNameIndexr� z&socket.if_indextoname() not available.c C �$ | � ttjd� | � ttjd� d S �Nr Z _DEADBEEF)rY r r r� r� rF r r r � testInvalidInterfaceIndexToName- � z2GeneralModuleTests.testInvalidInterfaceIndexToNamer� z&socket.if_nametoindex() not available.c C r� r� )rY r� r r� r rF r r r � testInvalidInterfaceNameToIndex3 r� z2GeneralModuleTests.testInvalidInterfaceNameToIndex� getrefcountz test needs sys.getrefcount()c C sP zt �t�}t�td� W d S ty' t �t�|kr$| �d� Y d S Y d S w )Nr z$socket.getnameinfo loses a reference)�sysr� rN r � getnameinfor� rU )rG �origr r r �testRefCountGetNameInfo9 s
��z*GeneralModuleTests.testRefCountGetNameInfoc C s( z t �dd� W d S ty Y d S w )N)rk r r r r )r r� r rF r r r �testInterpreterCrashE s
�z'GeneralModuleTests.testInterpreterCrashc C s� t jdt jdt jdt jdi}|�� D ]2\}}d|> d }dD ]}| �||@ ||||@ ��|@ � q ||�}| �||@ |� | �t|d� qd S )N� � r~ )r r~ � i ��r� igE#i2Tvl ) r �htonl�ntohl�htons�ntohs�itemsr� rY �
OverflowError)rG Zsizes�func�size�maskr> Zswappedr r r �testNtoHM s � "�zGeneralModuleTests.testNtoHc C s� dd l }g d�}|dg }ddddd> g}||jd |jd g d |jg }|D ] }t�|� t�|� q(|D ] }t�|� t�|� q7|D ]}| �t tj|� | �t tj|� qF|D ]}| �t tj|� | �t tj|� q[d S )
Nr )r r~ r� r� � �� ������l r~ �� � )
� _testcapiZINT_MIN�INT_MAXr r� r� r� r� rY r� )rG r� Z
s_good_valuesZ
l_good_valuesZ l_bad_valuesZ s_bad_values�kr r r �testNtoHErrors[ s.
���
�z!GeneralModuleTests.testNtoHErrorsc C s | j }tj�d�stjdv rd}nd}|D ]}z t�|d�}W q, ty) Y qw t�ttd�r7t�� dkrAt�|�}|||� zt�|d�}W n tyT d }Y nw |||� t j
se|t� |�|� |t� |d�|� |d ur{|t� |d�|� | � t
tj d � | � t
tj d
� d S ) N)�freebsdZnetbsdZ gnukfreebsd)�linuxrr )�daytimeZqotd�domain)Zechor� r� Ztcp�getandroidapilevel� Zudpr� r� )r� r� �platform�
startswithr �
getservbynamer r r� r Z
is_android�
getservbyportrY r� )rG �eqZservicesZservicerD Zport2Zudpportr r r �
testGetServBys s@ � �
�
�
z GeneralModuleTests.testGetServByc C s. | � t�� d � t�� �}| � |�� d � W d � n1 s w Y td��U | � t�� d� t�� �}| � |�� d� W d � n1 sHw Y t�d � | � t�� d � t�� �}| � |�� d � W d � n1 sqw Y W d � n1 s�w Y | �ttjd� | �ttjd� d S )N�
r� Zspam) r� r r9 �
gettimeoutr< r: rY �
ValueErrorr� )rG r! r r r r �testDefaultTimeout� s$
�
�
�� � z%GeneralModuleTests.testDefaultTimeout� inet_atonz test needs socket.inet_aton()c C s( | � dt�d�� | � dt�d�� d S )N� �0.0.0.0� ����r� )r� r r� rF r r r � testIPv4_inet_aton_fourbytes� s z/GeneralModuleTests.testIPv4_inet_aton_fourbytes� inet_ptonz test needs socket.inet_pton()c s* ddl m}m�m� � �fdd�}�fdd�}��d|d�� ��d|d �� ��d
|d �� ��d |d
�� ��d|d�� tsF||d� ||d� ||d� ||d� ||d� ��d|d�� ��d|d �� ��d
|d �� ��d|d�� ||d� ||d� ||d� ||d� ||d� d S )Nr )r� r� rA c �
�� | �S r8 r ��a)rA r� r r r � �
z5GeneralModuleTests.testIPv4toString.<locals>.<lambda>c � � � ttf| |�S r8 �rY r r� �r� r� rF r r r � �
�r� r� s � � z 255.0.255.0s ����z170.170.170.170� �1.2.3.4r� r� z0.0.0.z 300.0.0.0za.0.0.0z 1.2.3.4.5�::1)r r� r� rA r� r �rG r �g�
assertInvalidr )rA r� rG r �testIPv4toString� s.
z#GeneralModuleTests.testIPv4toStringc
s� zddl m�m� m} |s��d� W n ty! ��d� Y nw tjdkrKz�� d� W n tyJ } z|j dkr@��d� W Y d }~nd }~ww � �fd d
����fd d
�}��
d �d�� ��
d �d
�� ��
d�d�� ��
d�d�� ��
d�d�� ��
d�d�� |d� |d� |d� |d� |d� |d� |d � |d � t s�|d � ��
d �d �� ��
d!�d"�� ��
d#�d$�� |d%� |d&� |d'� |d(� |d)� |d*� d S )+Nr )r� r re �IPv6 not available�+could not import needed symbols from socket�win32�::�&' �IPv6 might not be supportedc r� r8 r r� )r r� r r r � r� z5GeneralModuleTests.testIPv6toString.<locals>.<lambda>c � �� ttf� | �S r8 r� r� �r rG r r r � r� � z0::0s z1::s E�v� V�� �$��z$45ef:76cb:1a:56ef:afeb:bac:1924:aeaes �B
� ' T zad42:abc::127:0:254:2s
z12:a::z0x20::z:::z::0::z1::abc::z 1::abc::defz 1:2:3:4:5:6z 1:2:3:4:5:6:z1:2:3:4:5:6:7:8:0z1:2:3:4:5:6:7:8:s �*@z::254.42.23.64s B ���*@z42::a29b:254.42.23.64s B�� �����*@z"42:a8b9:0:2:ffff:a29b:254.42.23.64z255.254.253.252z 1::260.2.3.0z 1::0.be.e.0z1:2:3:4:5:6:7:1.2.3.4z ::1.2.3.4:0z0.100.200.0:3:4:5:6:7:8) r r� r re rq � ImportErrorr� r� r �winerrorr� r �rG re r� r� r )r r r� rG r �testIPv6toString� sr
� �
������� z#GeneralModuleTests.testIPv6toString� inet_ntopz test needs socket.inet_ntop()c s ddl m}m�m� � �fdd�}�fdd�}��d|d�� ��d|d �� ��d
|d �� ��d |d
�� ||d� ||d� ||d� ��d|td ��� ��d|d�� ��d|d �� ��d
|d �� ||d� ||d� ||d� ��d|td ��� d S )Nr )� inet_ntoar� rA c r� r8 r r� )rA r� r r r % r� z5GeneralModuleTests.testStringToIPv4.<locals>.<lambda>c r� r8 r� r� rF r r r & r� z1.0.1.0s z
170.85.170.85s �U�Ur� r� r� r� s s r� )r r� r� rA r� � bytearrayr� r )rA r� rG r �testStringToIPv4! s$
z#GeneralModuleTests.testStringToIPv4c
s zddl m�m� m} |s��d� W n ty! ��d� Y nw tjdkrKz�� d� W n tyJ } z|j dkr@��d� W Y d }~nd }~ww � �fd d
����fd d
�}��
d �d�� ��
d
�d�� ��
d�d�� ��
d
�t d��� |d� |d� |d� d S )Nr )r� r re r� r� r� r� r� r� c r� r8 r r� )r r� r r r L r� z5GeneralModuleTests.testStringToIPv6.<locals>.<lambda>c r� r8 r� r� r� r r r M r� r� r� s z!aef:b01:506:1001:ffff:9997:55:170s
� ���� Ups s s ) r r� r re rq r� r� r� r r� r� r� r� r )r r r� rG r �testStringToIPv6; s8
� �
��� z#GeneralModuleTests.testStringToIPv6c C s� t �� }t�tjtj�}| �|j� |�d|f� |�� }z t� t�
� �}W n t y4 | � d� Y nw | �
|d d|fd|d � | �|d |� d S )Nr� r� r z
%s invalidr~ )r �find_unused_portr rA r5 rc r ro r� r� r� r rq rI r� )rG rD r r- Z
my_ip_addrr r r � testSockName_ s � z GeneralModuleTests.testSockNamec C s@ t � t jt j�}| �|j� |�t jt j�}| �|dkd� d S )Nr zinitial mode is reuse) r rA r5 rc r �
getsockoptrg rh � assertFalse�rG r Zreuser r r �testGetSockOptq s z!GeneralModuleTests.testGetSockOptc C sR t � t jt j�}| �|j� |�t jt jd� |�t jt j�}| � |dkd� d S )Nr~ r zfailed to set reuse mode)
r rA r5 rc r �
setsockoptrg rh r r r r r r �testSetSockOpty s
z!GeneralModuleTests.testSetSockOptc C sN t � t jt j��
}|�d� W d � n1 sw Y | �t|jd� d S )Nr~ � spam)r rA r5 �
settimeoutrY r r� r r r r �testSendAfterClose� s �z%GeneralModuleTests.testSendAfterClosec
C s� t � � }|�t jdf� t j |�� d��� z|�� W n ty: } z| �|jtjtj f� W Y d }~d S d }~ww | �
d� d S )Nr �rJ z#close() should raise EBADF/ENOTSOCK) r ro �
_LOCALHOSTrJ r r rI r� �EBADF�ENOTSOCKrU )rG r r= r r r �testCloseException� s $��z%GeneralModuleTests.testCloseExceptionc C s� t � t jt j��4}| �|jt j� tt d�r%| �|jt jt jB t jf� n| �|jt j� | �|j d� W d � d S 1 s?w Y d S )N� SOCK_CLOEXECr )
r rA r5 r� rO r rI r" r rP r r r r �testNewAttributes� s
��"�z$GeneralModuleTests.testNewAttributesc C s� t � � }| �|j� t�� }|d }|d }| �t|jt|f� | �t|jt|f� t �
� D ].}t�� }z |�t|f� W d S t y\ } z|j t j
ksQ|dkrR� W Y d }~q.d }~ww d S )Nr� � )r rc r r r� rY r� ro r� � itertools�countr r� Z
EADDRINUSE)rG r rD Zbig_portZneg_portr> r� r r r �test_getsockaddrarg� s& � ����z&GeneralModuleTests.test_getsockaddrargc C s� | � ttjd�� | � ttd�� | � ttd�� | � ttd�� | � ttd�� t�� }| �|j� | �t|jdd � |�tjd� d S )Nr Z
SIO_RCVALLZ RCVALL_ONZ
RCVALL_OFF�SIO_KEEPALIVE_VALSr� )r~ �d r ) rA r r rc r rY r� r r rM r r r �test_sock_ioctl� s z"GeneralModuleTests.test_sock_ioctl�SIO_LOOPBACK_FAST_PATHz1Loopback fast path support required for this testc
C sr t � � }| �|j� z |�t jd� W n ty, } z
d}|j|kr'| �d� � d }~ww | �t |jt jd � d S )NTi=' zQSIO_LOOPBACK_FAST_PATH is defined but doesn't implemented in this Windows version)
r rc r r r r r� rq rY r� )rG r! �excZ
WSAEOPNOTSUPPr r r �test_sio_loopback_fast_path� s
��z.GeneralModuleTests.test_sio_loopback_fast_pathc
C s6 zt �dd� W n t jy! } z |jt jkr | �d� � d }~ww t �td �D ]
}| �t|�d� q(t �dd� t �dd� t �d d� t j
rNt �dd� t t d�rYt �
� dkr_t �td � t �td� t �td � t �tdt jt j�}|D ]%\}}}}}| �|t j� | �t|�d
� | �|t j� | �t|�d � qwt �td d t j�}|D ]\}}}}}| �|t j� q�t �td d d t j� t �td d d d t j� t �d d t jt jd t j� t �td �}t jtd d
�} | �|| � t �td t j�}t jtd t jd�} | �|| � t �td d t j�}t jtd t jd�} | �|| � t �td d d t j�}t jtd t jd�} | �|| � t �td d d d t j�}t jtd t jd�} | �|| � t �d d t jt jd t j�}t jd d t jt jd t jd�} | �|| � | �tt jdd� t t d��r�zt �dd d d d t j� W d S t j�y� Y d S w d S )N� localhost�P zbuggy libc versionr rG r� r� r� �http�AddressFamily.AF_INET�SocketKind.SOCK_STREAMr )r rD �rO �r" )rP )�flags)r rD rO r" rP r! u ��AI_NUMERICSERV)r � getaddrinfor[ r� � EAI_SERVICErq r� r� �lenr � IPV6_ENABLEDr r� r� rA r5 r8 �SOL_TCP�
AI_PASSIVE� AF_UNSPECrY �UnicodeEncodeErrorr" )
rG r= �infoZinfosrO r" r� �socktyper� �br r r �testGetaddrinfo� s�
��
� � � �
���z"GeneralModuleTests.testGetaddrinfoc C s | � ttjdd� d S )N)zmail.python.orgr r )rY r r r� rF r r r �test_getnameinfo � z#GeneralModuleTests.test_getnameinfoZnetworkznetwork is not enabledc C sd t �d��
t�d� W d � n1 sw Y d}t�|� t�|� t�|dtjtj� d S )Nz
python.orgu! испытание.pythontest.netr )r Ztransient_internetr r� �gethostbyname_exr# r) r5 )rG r� r r r � test_idna" s �
z GeneralModuleTests.test_idnac
s\ t td�r
t td�s� �d� � fdd�}� fdd�}t�� \}}t�tj|�}zp|r0|�d� � �t�� t� d � |�
d
t j � W d � n1 sMw Y |r�t�tj|� t� d � � �t
|j
d
t j � W t� d � t�tj|� |�� |�� d S W t� d � t�tj|� |�� |�� d S t� d � t�tj|� |�� |�� w ) N�alarmr� z9signal.alarm and socket.socketpair required for this testc s � � ttjd� d S r� �rY r� �math�acosh�r] rF r r �
ok_handler9 r� z@GeneralModuleTests.check_sendall_interrupted.<locals>.ok_handlerc s � � ttjd� dd d S �Nr r~ r4 r7 rF r r �raising_handler; s zEGeneralModuleTests.check_sendall_interrupted.<locals>.raising_handlerg �?r~ � xr )r �signalr rq r� �SIGALRMr rY �ZeroDivisionErrorr3 �sendallr �
SOCK_MAX_SIZE� TimeoutErrorr )rG Z with_timeoutr8 r: �cr! � old_alarmr rF r �check_sendall_interrupted3 s@
�
�
�
�
z,GeneralModuleTests.check_sendall_interruptedc C � | � d� d S �NF�rD rF r r r �test_sendall_interruptedR � z+GeneralModuleTests.test_sendall_interruptedc C rE )NTrG rF r r r �%test_sendall_interrupted_with_timeoutU rI z8GeneralModuleTests.test_sendall_interrupted_with_timeoutc C s� t � t jt j�}t|�}| �t��}d }t�� W d � n1 s"w Y | �|t |j
j d �� t � t jt j�}|� d�}t|�}d }t�� | �t�� d }t�� W d � d S 1 saw Y d S )Nr r )
r rA r5 rK Z assertWarns�ResourceWarningr rS rI r8 �warningr] �makefile)rG r r rc r r r r �test_dealloc_warnX s
�
"�z$GeneralModuleTests.test_dealloc_warnc C sX t � t jt j��}|�d�}|�� | �t|�d� W d � d S 1 s%w Y d S )Nr z <_io.BufferedReader name=-1>)r rA r5 rM r r� rK �rG r �fpr r r �test_name_closed_socketioi s
"�z,GeneralModuleTests.test_name_closed_socketioc C s� t � � �>}|jddd�}| �|�� � | �|�� � | �|�� � |�� | �t |j� | �t |j� | �t |j� W d � d S 1 sEw Y d S )Nr r )� buffering)
r rM rA �readabler �writable�seekabler rY r� rO r r r � test_unusable_closed_socketioo s
"�z0GeneralModuleTests.test_unusable_closed_socketioc C s2 t � � }zC|�tdf� t �|�� � | �t��
|�d� W d � n1 s(w Y W | �t�� |�� W d � n1 sBw Y n| �t�� |�� W d � w 1 s\w Y w | �t��
t �d � W d � n1 sww Y | �t�� t �d� W d � d S 1 s�w Y d S )Nr r~ r� ) r ro r� r rJ rY r rE r� r r r r �test_socket_closez s* ��
��
� � "�z$GeneralModuleTests.test_socket_closec C s� dD ]R}| j |d��B t�� �.}d|v rd nd}|j||d��}| �|j|� W d � n1 s1w Y W d � n1 s@w Y W d � n1 sOw Y qd S )N)r r �rw�w�wb��moder- r ��encoding)�subTestr rM r� r\ )rG r\ r r^ rP r r r �test_makefile_mode� s
�� �� ���z%GeneralModuleTests.test_makefile_modec C s� dD ]G}| j |d��7 t�� �#}| �td��
|�|� W d � n1 s&w Y W d � n1 s5w Y W d � n1 sDw Y qd S )N)Zrtrk �+r� r[ z invalid mode)r_ r �assertRaisesRegexr� rM )rG r\ r r r r �test_makefile_invalid_mode� s
�� �� ���z-GeneralModuleTests.test_makefile_invalid_modec C s� t � � }|� ttjd �D ] }| �ttj||� qW d � n1 s$w Y ttjd �D ]$}t�t�t j|��}| � |t j� t�t�t j
|��}| � |t j
� q0d S r ) r �range�pickle�HIGHEST_PROTOCOLrY r� �dumps�loadsrA r� r5 )rG r �protocolrO r" r r r � test_pickle� s � ��z GeneralModuleTests.test_picklec C s� dD ]&}t � t jt j��}|�tdf� |�|� W d � n1 s#w Y qt � t jt j��}|�tdf� |�� W d � d S 1 sHw Y d S )N)r r� r )r rA r5 ro r� rE )rG �backlog�srvr r r �test_listen_backlog� s ��
"�z&GeneralModuleTests.test_listen_backlogc C sb dd l }t�tjtj��}|�tdf� | �t|j|j d � W d � d S 1 s*w Y d S r9 )
r� r rA r5 ro r� rY r� rE r� )rG r� rl r r r � test_listen_backlog_overflow� s
"�z/GeneralModuleTests.test_listen_backlog_overflow� IPv6 required for this test.c C sh | � ttjtjddfd� t�tjtj��}| � t|jtjddf� W d � d S 1 s-w Y d S )Nr r� i����) rY r� r r� r r r r5 ro rM r r r �
test_flowinfo� s
�"�z GeneralModuleTests.test_flowinfoc C s0 t �ddt jt jt j�\�^ }}| �|d� d S )N�ff02::1de:c0:face:8Dr )�ff02::1de:c0:face:8dr r r )r r# r r$ rk r� )rG r� �sockaddrr r r �test_getaddrinfo_ipv6_basic� s �z.GeneralModuleTests.test_getaddrinfo_ipv6_basicr� zdoes not work on Windowsz Symbolic scope id does not workz test needs socket.if_nameindex()c C sL t �� d \}}t �d| dt jt jt j�\�^ }}| �|ddd|f� d S )Nr �ff02::1de:c0:face:8D%r rr )r r� r# r r$ rk r� )rG �ifindex�test_interfacer� rs r r r �&test_getaddrinfo_ipv6_scopeid_symbolic� s �z9GeneralModuleTests.test_getaddrinfo_ipv6_scopeid_symbolicz.Numeric scope id does not work or undocumentedc C sD d}t �dt|� dt jt jt j�\�^ }}| �|ddd|f� d S )N�* ru r rr r )r r# r8 r r$ rk r� )rG rv r� rs r r r �%test_getaddrinfo_ipv6_scopeid_numeric� s
�z8GeneralModuleTests.test_getaddrinfo_ipv6_scopeid_numericc C sH t �� d \}}ddd|f}t �|t jt jB �}| �|d| df� d S )Nr rq r �ff02::1de:c0:face:8d%�1234)r r� r� �NI_NUMERICHOST�NI_NUMERICSERVr� )rG rv rw rs �nameinfor r r �&test_getnameinfo_ipv6_scopeid_symbolic� s z9GeneralModuleTests.test_getnameinfo_ipv6_scopeid_symbolicc C s@ d}ddd|f}t �|t jt jB �}| �|dt|� df� d S )Nry rq r r r{ r| )r r� r} r~ r� r8 )rG rv rs r r r r �%test_getnameinfo_ipv6_scopeid_numeric� s z8GeneralModuleTests.test_getnameinfo_ipv6_scopeid_numericc C sZ t � t jt j��}| �t|j�d� | �t|j�d� W d � d S 1 s&w Y d S )Nr r )r rA r5 r� r8 rO r" rM r r r �test_str_for_enums� s "�z%GeneralModuleTests.test_str_for_enumsc C s� t tdd�}t tdd�}tj|B |B }t�tj|��E}| �|jtj� |�d� | �|jtj� |�d� | �|jtj� |�d� | �|jtj� |�d� | �|jtj� W d � d S 1 sbw Y d S )N�
SOCK_NONBLOCKr r r~ TF)r� r r5 rA r� r" r � setblocking)rG r� r Z sock_typer! r r r � test_socket_consistent_sock_type s
"�z3GeneralModuleTests.test_socket_consistent_sock_typec C s� t � t jt j�}|�� }tt jj�� �d }tdd� t jj� � D ��d }t j ||d|d�� }| �
|j |� | �
|j |� | �
|jddh� W d � d S 1 sQw Y d S )Nr~ c s s � | ] \}}|d vr|V qdS )> r r� Nr )r, r- �kindr r r r3 s � �
�zEGeneralModuleTests.test_unknown_socket_family_repr.<locals>.<genexpr>r� )rO r" rP rJ r )r rA r5 �detach�max�
AddressFamily� __members__�values�
SocketKindr� r� rO r" rI rP )rG r �fdZunknown_familyZ unknown_typer! r r r � test_unknown_socket_family_repr s"
���"�z2GeneralModuleTests.test_unknown_socket_family_repr�sendfileztest needs os.sendfile()c C s G dd� d�}t � � �u}t�tjtj�}t�|� | �t j�� |�||�� W d � n1 s1w Y | �t �� |�|dd �� W d � n1 sOw Y | �t
�� |�|d �� W d � n1 skw Y W d � d S W d � d S 1 s�w Y d S )Nc @ r? )z<GeneralModuleTests.test__sendfile_use_sendfile.<locals>.Filec S s
|| _ d S r8 �r� )rG r� r r r rX 1 �
zEGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.__init__c S � | j S r8 r� rF r r r rJ 4 s zCGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.filenoN)rN rO rP rX rJ r r r r �File0 s r� r� r� ) r r� r �curdir�O_RDONLYr rY �_GiveupOnSendfile�_sendfile_use_sendfiler� r� )rG r� r r� r r r �test__sendfile_use_sendfile. s
� � ��"�z.GeneralModuleTests.test__sendfile_use_sendfilec C st | � |j|� | � |j|� |�� }tj|d�}| �|j� |�� | � |j|� | � |j|� | � |�� |� d S �Nr )r� rO r" rJ r rc r r� )rG r! rO �styper� �s2r r r �_test_socket_fileno@ s z&GeneralModuleTests._test_socket_filenoc C sR t � t jt j�}| �|j� |�tjdf� | �|t jt j� t t d�rCt � t jt j
�}| �|j� |�tjdf� | �|t jt j
� tj rgt � t j t j�}| �|j� |�tj
dddf� | �|t j t j� t t d�r�t�� }| �tj|� t � t jt j�}| �|j� z |�tj�|d�� W n
ty� Y d S w | �|t jt j� d S d S )Nr r$ r r )r rA r5 rc r ro r r� r� r r$ r& r r r� r� �shutil�rmtreer r� r r5 �PermissionError)rG r! Ztmpdirr r r �test_socket_filenoM s8
�
��z%GeneralModuleTests.test_socket_filenoc C �D | � t�� tjtjtjdd� W d � d S 1 sw Y d S )Ng @E@r �rY r� r rA r5 rF r r r � test_socket_fileno_rejects_floatl � "�z3GeneralModuleTests.test_socket_fileno_rejects_floatc C r� )N�foor r� rF r r r �&test_socket_fileno_rejects_other_typesp r� z9GeneralModuleTests.test_socket_fileno_rejects_other_typesc C �F | � td�� tjtjtjdd� W d � d S 1 s w Y d S )N�negative file descriptorr� r �rb r� r rA r5 rF r r r �)test_socket_fileno_rejects_invalid_sockett s "�z<GeneralModuleTests.test_socket_fileno_rejects_invalid_socketzWindows disallows -1 onlyc C r� )Nr� i���r r� rF r r r �#test_socket_fileno_rejects_negativex s "�z6GeneralModuleTests.test_socket_fileno_rejects_negativec C s� d}| � t��}tjt�� d� W d � n1 sw Y | �|jjtj|f� | � t��}tjtj tj
t�� d� W d � n1 sFw Y | �|jjtj|f� d S )Ni6' r ) rY r r r Z make_bad_fdrI rb r� r rA r5 )rG Z WSAENOTSOCKrc r r r �$test_socket_fileno_requires_valid_fd} s � � �z7GeneralModuleTests.test_socket_fileno_requires_valid_fdc C s� t �� �P}| �t�� tj|�� d� W d � n1 s w Y | �t��}tjtjtj|�� d� W d � n1 s>w Y | �|j j
t
j � W d � d S 1 sWw Y d S r� ) r� �NamedTemporaryFilerY r r rJ rA r5 r� rb r� r )rG Zafilerc r r r �%test_socket_fileno_requires_socket_fd� s
� � �"�z8GeneralModuleTests.test_socket_fileno_requires_socket_fd)YrN rO rP rD rN r� �
skipUnlessr@ rR rW r\ rd ri rm r� r- rq r� r� r r& r� r� r� r� r r r� r� r� r� r� r� r� r � cpython_onlyr� r� r� r� r� r� r� r� r� r r r r
r r r r r. r/ Zis_resource_enabledr2 rD rH rJ rN rQ rV rW r` rc rj rm rn rp rt �skipIfr rx rz r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r? < s
*
�
�
�
�
, �
�
�
>�
�
"
�
N �
�
�
r? z!SocketCan required for this test.c @ s� e Zd Zdd� Ze�eed�d�dd� �Zdd� Z e�eed�d�d d
� �Z
d d � Z d
d� Z e�eed�d�dd� �Z
e�eed�d�dd� �ZdS )� BasicCANTestc C s t j t j t j d S r8 )r �AF_CANr r rF r r r ri � s
z!BasicCANTest.testCrucialConstants�CAN_BCM�&socket.CAN_BCM required for this test.c C � t j t j t j t j t j t j t j t j t j t j
t j t j t j
t j t j t j t j t j t j t j t j t j t j t j d S r8 )r r� �CAN_BCM_TX_SETUP�CAN_BCM_TX_DELETE�CAN_BCM_TX_READ�CAN_BCM_TX_SEND�CAN_BCM_RX_SETUP�CAN_BCM_RX_DELETE�CAN_BCM_RX_READ�CAN_BCM_TX_STATUS�CAN_BCM_TX_EXPIRED�CAN_BCM_RX_STATUS�CAN_BCM_RX_TIMEOUT�CAN_BCM_RX_CHANGED�CAN_BCM_SETTIMER�CAN_BCM_STARTTIMER�CAN_BCM_TX_COUNTEVT�CAN_BCM_TX_ANNOUNCE�CAN_BCM_TX_CP_CAN_ID�CAN_BCM_RX_FILTER_ID�CAN_BCM_RX_CHECK_DLC�CAN_BCM_RX_NO_AUTOTIMER�CAN_BCM_RX_ANNOUNCE_RESUME�CAN_BCM_TX_RESET_MULTI_IDX�CAN_BCM_RX_RTR_FRAMErF r r r �testBCMConstants� s0
z BasicCANTest.testBCMConstantsc C �: t � t jt jt j�� }W d � d S 1 sw Y d S r8 �r r r r rM r r r �testCreateSocket� � "�z BasicCANTest.testCreateSocketc C r� r8 )r r r$ r� rM r r r �testCreateBCMSocket� � "�z BasicCANTest.testCreateBCMSocketc C sX t � t jt jt j��}d}|�|� | �|�� |� W d � d S 1 s%w Y d S )N)r] )r r r r ro r� r� )rG r! �addressr r r � testBindAny� s
"�zBasicCANTest.testBindAnyc C sL t � t jt jt j��}| �td|jd� W d � d S 1 s w Y d S )N�interface name too long)� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx)r r r r rb r ro rM r r r �testTooLongInterfaceName� s
�"�z%BasicCANTest.testTooLongInterfaceName�CAN_RAW_LOOPBACKz/socket.CAN_RAW_LOOPBACK required for this test.c C sn t � t jt jt j��#}dD ]}|�t jt j|� | �||�t jt j�� q
W d � d S 1 s0w Y d S )N)r r~ ) r r r r r � SOL_CAN_RAWr� r� r )rG r! Zloopbackr r r � testLoopback� s ���"�zBasicCANTest.testLoopback�CAN_RAW_FILTERz-socket.CAN_RAW_FILTER required for this test.c C s� d\}}t �d||�}t�tjtjtj��*}|�tjtj|� | � ||�
tjtjd�� |�tjtjt |�� W d � d S 1 sBw Y d S )N)� i z=IIrl ) r �packr r r r r r� r� r� r r� )rG �can_idZcan_maskZ
can_filterr! r r r �
testFilter� s �"�zBasicCANTest.testFilterN)rN rO rP ri r� r� r r r� r� r� r� r� r� r� r r r r r� � s* �
�
�
�r� c @ s� e Zd Zddd�Zedd� �Zedd� �Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z e
�eed�d�dd� �Ze
�eed�d�dd� �ZdS ) �CANTestr� c C r� r� )r� rX r� r r r rX � r� zCANTest.__init__c C s& t |�}|�dd�}t�| j|||�S )zBuild a CAN frame.rl � )r% �ljustr r� rs )�clsr� �data�can_dlcr r r �build_can_frame� s zCANTest.build_can_framec C s& t �| j|�\}}}|||d|� fS )zDissect a CAN frame.N)r r rs )r� �framer� r� r� r r r �dissect_can_frame� s zCANTest.dissect_can_framec C s6 | j �| j�\}}| �| j|� | �|d | j� d S r� )r! �recvfromrr r� �cfrp �rG r� r� r r r �
testSendFrame� s zCANTest.testSendFramec C � | � dd�| _| j�| j� d S )Nr s �r� r� r� r� rF r r r �_testSendFrame � zCANTest._testSendFramec C s$ | j �| j�\}}| �| j|� d S r8 )r! r� rr r� r� r� r r r �testSendMaxFrame rR zCANTest.testSendMaxFramec C r� )Nr s r� rF r r r �_testSendMaxFrame r� zCANTest._testSendMaxFramec C �D | j �| j�\}}| �| j|� | j �| j�\}}| �| j|� d S r8 )r! r� rr r� �cf1�cf2r� r r r �testSendMultiFrames � zCANTest.testSendMultiFramesc C s< | � dd�| _| j�| j� | � dd�| _| j�| j� d S )N� s D3"� s �"3)r� r� r� r� r� rF r r r �_testSendMultiFrames s z CANTest._testSendMultiFramesr� r� c C sP | j �| j�\}}| �| j|� | �|�\}}}| �| j|� | �| j|� d S r8 )r� r� rr r� r� r� r� r� )rG r� r� r� r� r� r r r �_testBCM s
zCANTest._testBCMc C s� t � t jt jt j�}| �|j� |�| jf� d| _t g d��| _
| � | j| j
�| _ t j
}d}d}d } } }}d} d}
t| j �dksFJ �t�| j|||||||| |
�
} | | j } |�| �}
| �|
t| �� d S )N�# )� � �� r i" r~ r� )r r r$ r� rc r r� rp r� �bytesr� r� r� r� r% r r� ru r� r� )rG ZbcmZopcoder! r Z
ival1_secondsZ
ival1_usecZ
ival2_secondsZ
ival2_usecZ
bcm_can_idZnframes�headerZheader_plus_frameZ
bytes_sentr r r �testBCM& s6 �
zCANTest.testBCMNr� )rN rO rP rX � classmethodr� r� r� r� r� r� r� r� r� r� r r r� r r r r r r� � s(
�
�r� z!CAN ISOTP required for this test.c sV e Zd Z� fdd�Zdd� Zdd� Ze�ee d�d�d d
� �Z
d d � Z d
d� Z � Z
S )� ISOTPTestc � t � j|i |�� d| _d S �Nri �rW rX rp r\ r_ r r rX I �
zISOTPTest.__init__c C s t j t j t j t j d S r8 )r r� r r% r$ rF r r r ri M s
z ISOTPTest.testCrucialConstantsc C r� r8 r� rM r r r r� S r� zISOTPTest.testCreateSocketr% z(socket.CAN_ISOTP required for this test.c C r� r8 )r r r$ r% rM r r r �testCreateISOTPSocketW r� z ISOTPTest.testCreateISOTPSocketc C s� t � t jt jt j��,}| �td��
|�d� W d � n1 s!w Y W d � d S W d � d S 1 s9w Y d S )Nr� )r� r~ r� )r r r$ r% rb r ro rM r r r r� ] s ��"�z"ISOTPTest.testTooLongInterfaceNamec
C s� z2t � t jt jt j�� }| jddf}|�|� | �|�� |� W d � W d S 1 s+w Y W d S tyU } z|j t j
krI| � d| j � n� W Y d }~d S d }~ww )Nr� iV rn ) r r r$ r% rp ro r� r� r r� �ENODEVrq �rG r! r� r� r r r �testBindc s
&� ����zISOTPTest.testBind)rN rO rP rX ri r� r� r� r r r r� r rg r r r_ r r F s �
r z!CAN J1939 required for this test.c sX e Zd Z� fdd�Ze�eed�d�dd� �Ze�eed�d�dd� �Z d d
� Z
� Z S ) � J1939Testc r r r r\ r_ r r rX t r
zJ1939Test.__init__r' z(socket.CAN_J1939 required for this test.c C r� r8 )r r' ZJ1939_MAX_UNICAST_ADDRZJ1939_IDLE_ADDR�
J1939_NO_ADDR�
J1939_NO_NAMEZJ1939_PGN_REQUESTZJ1939_PGN_ADDRESS_CLAIMEDZJ1939_PGN_ADDRESS_COMMANDEDZJ1939_PGN_PDU1_MAXZ
J1939_PGN_MAX� J1939_NO_PGNZSO_J1939_FILTERZSO_J1939_PROMISCZSO_J1939_SEND_PRIOZSO_J1939_ERRQUEUEZSCM_J1939_DEST_ADDRZSCM_J1939_DEST_NAMEZSCM_J1939_PRIOZSCM_J1939_ERRQUEUEZ
J1939_NLA_PADZJ1939_NLA_BYTES_ACKEDZJ1939_EE_INFO_NONEZJ1939_EE_INFO_TX_ABORTZJ1939_FILTER_MAXrF r r r �testJ1939Constantsx s0
z J1939Test.testJ1939Constantsc C r� r8 )r r r$ r' rM r r r �testCreateJ1939Socket� r� z J1939Test.testCreateJ1939Socketc
C s� z6t � t jt jt j��!}| jt jt jt jf}|�|� | � |�
� |� W d � W d S 1 s/w Y W d S t yY } z|j t j
krM| �d| j � n� W Y d }~d S d }~ww rm )r r r$ r' rp r r r ro r� r� r r� r rq r
r r r r � s
&� ����zJ1939Test.testBind) rN rO rP rX r� r� r r r r r rg r r r_ r r q s �
!�
r z#RDS sockets required for this test.c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� BasicRDSTestc C s t j t j d S r8 )r �AF_RDSr* rF r r r ri � s
z!BasicRDSTest.testCrucialConstantsc C s8 t � t jt jd�� }W d � d S 1 sw Y d S r� )r r* r+ rM r r r r� � s "�z BasicRDSTest.testCreateSocketc C s` d}t � t jt jd��}|�t jt j|� |�t jt j|� W d � d S 1 s)w Y d S )Ni @ r )r r* r+ r rg � SO_RCVBUF� SO_SNDBUF)rG rr r! r r r �testSocketBufferSize� s
"�z!BasicRDSTest.testSocketBufferSizeN)rN rO rP ri r� r r r r r r � s r c s� e Zd Zd dd�Z� fdd�Zdd� Zdd � Zd
d � Zd d
� Ze e
j
d�dd� �Z e e
j
d�dd� �Z dd� Z
dd� Zdd� Zdd� Z� ZS ) �RDSTestr� c C r� r� )r� rX r� r r r rX � r� zRDSTest.__init__c s t � �� t�� | _d S r8 )rW rH rY r� ZevtrF r_ r r rH � r� z
RDSTest.setUpc C s2 | j �| j�\}}| �| j|� | �| j|� d S r8 )rB r� rr r� r� r� �rG r� r� r r r �testSendAndRecv� s zRDSTest.testSendAndRecvc C �" d| _ | j�| j dt| jf� d S �Nr r �r� r� ra r� rD rF r r r �_testSendAndRecv� � zRDSTest._testSendAndRecvc C sH | j �| jtj�\}}| �| j|� | j �| j�\}}| �| j|� d S r8 )rB r� rr r �MSG_PEEKr� r� r r r r �testPeek� s zRDSTest.testPeekc C r r r rF r r r � _testPeek� r! zRDSTest._testPeek�recvmsgc C s( | j �| j�\}}}}| �| j|� d S r8 )rB r% rr r� r� )rG r� �ancdataZ msg_flagsr� r r r �testSendAndRecvMsg� s zRDSTest.testSendAndRecvMsg�sendmsgc C s& d| _ | j�| j gddt| jf� d S )Ns< hello hello hello hello hello hello hello hello hello hello r r )r� r� r( r� rD rF r r r �_testSendAndRecvMsg� s zRDSTest._testSendAndRecvMsgc C r� r8 )rB r� rr r� �data1�data2r r r r �testSendAndRecvMulti� r� z RDSTest.testSendAndRecvMultic C s@ d| _ | j�| j dt| jf� d| _| j�| jdt| jf� d S )Ns baconr s egg)r* r� ra r� rD r+ rF r r r �_testSendAndRecvMulti� s z RDSTest._testSendAndRecvMultic C sL t � | jgg g d�\}}}| �| j|� | j�| j�\}}| �| j|� d S )Ng @)�selectrB rI r� rr r� r� )rG r rY rk r� r� r r r �
testSelect� s zRDSTest.testSelectc C r )Ns selectr r rF r r r � _testSelect� r! zRDSTest._testSelectr� )rN rO rP rX rH r r r# r$ r7 r r' r) r, r- r/ r0 rg r r r_ r r � s
r z'QIPCRTR sockets required for this test.c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�BasicQIPCRTRTestc C �
t j d S r8 )r r0 rF r r r ri r� z%BasicQIPCRTRTest.testCrucialConstantsc C �6 t � t jt j�� }W d � d S 1 sw Y d S r8 )r r0 r$ rM r r r r� � "�z!BasicQIPCRTRTest.testCreateSocketc C sJ t � t jt j��}| �|�� d d� W d � d S 1 s w Y d S �Nr~ r )r r0 r$ r� r� rM r r r � testUnbound � "�z BasicQIPCRTRTest.testUnboundc C s` t � t jt j�� }tj||�� d d� | �|�� d d� W d � d S 1 s)w Y d S )Nr r
r~ )r r0 r$ r rC r� �assertNotEqualrM r r r � testBindSock s "�z BasicQIPCRTRTest.testBindSockc C sJ t � t jt j��}| jttj|dd� W d � d S 1 s w Y d S )Nr� r
)r r0 r$ rY r r rC rM r r r �testInvalidBindSock r7 z$BasicQIPCRTRTest.testInvalidBindSockc C sT t � t jt j��}|�d� | �|�� d d� W d � d S 1 s#w Y d S )N)�{ r; r~ r )r r0 r$ r� r8 r� rM r r r �testAutoBindSock s
"�z!BasicQIPCRTRTest.testAutoBindSockN) rN rO rP ri r� r6 r9 r: r< r r r r r1 s r1 c @ �, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�BasicVSOCKTestc C r2 r8 )r r� rF r r r ri " r� z#BasicVSOCKTest.testCrucialConstantsc C rp r8 ) r �SO_VM_SOCKETS_BUFFER_SIZE� SO_VM_SOCKETS_BUFFER_MIN_SIZE� SO_VM_SOCKETS_BUFFER_MAX_SIZEr� �VMADDR_PORT_ANY�VMADDR_CID_HOST�VM_SOCKETS_INVALID_VERSIONr
rF r r r �testVSOCKConstants% s
z!BasicVSOCKTest.testVSOCKConstantsc C r3 r8 )r r� r5 rM r r r r� / r4 z BasicVSOCKTest.testCreateSocketc C s� t � t jt j��l}|�t jt j�}|�t jt j�}|�t jt j�}|�t jt j|d � |�t jt j|d � |�t jt j|d � | �|d |�t jt j�� | �|d |�t jt j�� | �|d |�t jt j�� W d � d S 1 sww Y d S �Nr� ) r r� r5 r rA r? r@ r r� )rG r! Zorig_maxr� Zorig_minr r r r 3 sF ���
�
�
�
��
��
��"�z#BasicVSOCKTest.testSocketBufferSizeN)rN rO rP ri rE r� r r r r r r> s
r> z)Bluetooth sockets required for this test.c @ sv e Zd Zdd� Zdd� Ze�ejdkd�dd� �Z e�ejdkd �d
d � �Z
e�ejdkp2ej� d �d
�dd� �Z dS )�BasicBluetoothTestc C sR t j t j t j t j tjdkr%t j t j t j tj�
d�s't j d S d S d S )Nr� r� ) r Z
BDADDR_ANYZ BDADDR_LOCALr4 r6 r� r� � BTPROTO_HCIZSOL_HCI�
BTPROTO_L2CAPr� � BTPROTO_SCOrF r r r �testBluetoothConstantsR s
�z)BasicBluetoothTest.testBluetoothConstantsc C r� r8 )r r4 r5 r6 rM r r r �testCreateRfcommSocket` r� z)BasicBluetoothTest.testCreateRfcommSocketr� z&windows does not support L2CAP socketsc C r� r8 )r r4 r+ rI rM r r r �testCreateL2capSocketd � "�z(BasicBluetoothTest.testCreateL2capSocketz$windows does not support HCI socketsc C r� r8 )r r4 r rH rM r r r �testCreateHciSocketi rN z&BasicBluetoothTest.testCreateHciSocketr� z.windows and freebsd do not support SCO socketsc C r� r8 )r r4 r+ rJ rM r r r �testCreateScoSocketn r� z&BasicBluetoothTest.testCreateScoSocketN)
rN rO rP rK rL r� r� r� r� rM rO r� rP r r r r rG N s
�rG c @ s� e Zd Zd+dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Ze�e�Zejd$d%� �Zd&d'� Zd(d)� Zd*S ),� BasicTCPTestr� c C r� r� �r� rX r� r r r rX w r� zBasicTCPTest.__init__c C r� r� )r� r� r� r� r� r r r �testRecvz r� zBasicTCPTest.testRecvc C � | j �t� d S r8 �r� r� r� rF r r r � _testRecv r zBasicTCPTest._testRecvc C s8 | j �tt�d �}| j �d�}|| }| �|t� d S �N� r� )r� r� r% r� r� )rG �seg1�seg2r� r r r �testOverFlowRecv� s z BasicTCPTest.testOverFlowRecvc C rT r8 rU rF r r r �_testOverFlowRecv� r z BasicTCPTest._testOverFlowRecvc C s | j �d�\}}| �|t� d S r� )r� r� r� r� �rG r� r� r r r � testRecvFrom� � zBasicTCPTest.testRecvFromc C rT r8 rU rF r r r �
_testRecvFrom� r zBasicTCPTest._testRecvFromc C s@ | j �tt�d �\}}| j �d�\}}|| }| �|t� d S rW )r� r� r% r� r� )rG rY r� rZ r� r r r �testOverFlowRecvFrom� s z!BasicTCPTest.testOverFlowRecvFromc C rT r8 rU rF r r r �_testOverFlowRecvFrom� r z"BasicTCPTest._testOverFlowRecvFromc C s2 d} | j �d�}|s q||7 }q| �|d� d S )N� r~ r� � ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff)r� r� r� )rG r� �readr r r � testSendAll� s �zBasicTCPTest.testSendAllc C s d}| j �|� d S )Nrd )r� r? )rG Z big_chunkr r r � _testSendAll� s zBasicTCPTest._testSendAllc C sP | j �� }t�|tjtj�}| �|j� | �|tj� |� d�}| �
|t � d S r� ) r� rJ r �fromfdrA r5 rc r rC r� r� r� )rG r� r r� r r r �
testFromFd� s
zBasicTCPTest.testFromFdc C rT r8 rU rF r r r � _testFromFd� r zBasicTCPTest._testFromFdc C s0 | j �� }| �|j� |�d�}| �|t� d S r� )r� �duprc r r� r� r� )rG r r� r r r �testDup� s
zBasicTCPTest.testDupc C rT r8 rU rF r r r �_testDup� r zBasicTCPTest._testDupc C s& | j �d�}| �|t� | j�� d S r� )r� r� r� r� r� r� r� r r r � testShutdown� s zBasicTCPTest.testShutdownc C s | j �t� | j �d� d S rF )r� r� r� �shutdownrF r r r �
_testShutdown� r� zBasicTCPTest._testShutdownc C sX dd l }| j�t� | �t| jj|jd � | �t| jjd|jd � | j�d� d S )Nr r~ r� ) r� r� r� r� rY r� ro r� �UINT_MAX�rG r� r r r �_testShutdown_overflow� s � �z#BasicTCPTest._testShutdown_overflowc C s� | j �� }| j �� }| �||� | �| j j� | �t| j jd� | j � � t
j
t
j t
j |d�}| �
|j � |�d�}| �|t� d S )Nr� r )r� rJ r� r� rA �_closedrY r r� r r rA r5 rc r� )rG rJ r r r� r r r �
testDetach� s
zBasicTCPTest.testDetachc C rT r8 rU rF r r r � _testDetach� r zBasicTCPTest._testDetachNr� )rN rO rP rX rS rV r[ r\ r^ r` ra rb rf rg ri rj rl rm rn rp r r� ZtestShutdown_overflowrs ru rv r r r r rQ u s.
rQ c @ �F e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dS )� BasicUDPTestr� c C r� r� )r� rX r� r r r rX � r� zBasicUDPTest.__init__c C � | j �tt��}| �|t� d S r8 �rB r� r% r� r� r� r r r �testSendtoAndRecv� r_ z BasicUDPTest.testSendtoAndRecvc C � | j �tdt| jf� d S r� �r� ra r� r� rD rF r r r �_testSendtoAndRecv� r� z BasicUDPTest._testSendtoAndRecvc C �$ | j �tt��\}}| �|t� d S r8 �rB r� r% r� r� r] r r r r^ � � zBasicUDPTest.testRecvFromc C r| r� r} rF r r r r`
r� zBasicUDPTest._testRecvFromc C � | � t| jjd� d S �Nr� �rY r� rB r� rF r r r �testRecvFromNegative
r0 z!BasicUDPTest.testRecvFromNegativec C r| r� r} rF r r r �_testRecvFromNegative
r� z"BasicUDPTest._testRecvFromNegativeNr� �
rN rO rP rX r{ r~ r^ r` r� r� r r r r rx � s
rx c @ rw )�BasicUDPLITETestr� c C r� r� )r� rX r� r r r rX
r� zBasicUDPLITETest.__init__c C ry r8 rz r� r r r r{
r_ z"BasicUDPLITETest.testSendtoAndRecvc C r| r� r} rF r r r r~
r� z#BasicUDPLITETest._testSendtoAndRecvc C r r8 r� r] r r r r^
r� z BasicUDPLITETest.testRecvFromc C r| r� r} rF r r r r` !
r� z BasicUDPLITETest._testRecvFromc C r� r� r� rF r r r r� $
r0 z%BasicUDPLITETest.testRecvFromNegativec C r| r� r} rF r r r r� (
r� z&BasicUDPLITETest._testRecvFromNegativeNr� r� r r r r r�
s
r� c s� e Zd ZejZ� fdd�Zdd� ZdZdd� Z dd � Z
d
d � Z d d
� Z dZ
dD ]
Ze
eeed�O Z
q'dZdZdZddd�Z� ZS )�SendrecvmsgBasec s t �� | _t� �� d S r8 )rY r� �
misc_eventrW rH rF r_ r r rH M
r� zSendrecvmsgBase.setUpc C s | j �|�S r8 )�cli_sockr� r� r r r � sendToServerQ
s z SendrecvmsgBase.sendToServerr c G s | j j|| jt|�d � � S r8 )r� r( �sendmsg_to_server_defaultsr% �rG r] r r r �sendmsgToServerY
s �z SendrecvmsgBase.sendmsgToServerc G s |j |g|�R � }| �|� |S r8 )r% �registerRecvmsgResult)rG r rr r] �resultr r r � doRecvmsga
s
zSendrecvmsgBase.doRecvmsgc C r� r8 r �rG r� r r r r� k
s z%SendrecvmsgBase.registerRecvmsgResultc C s | � ||� d S r8 �r� �rG Zaddr1Zaddr2r r r �checkRecvmsgAddressq
� z#SendrecvmsgBase.checkRecvmsgAddressr )�
MSG_CTRUNC�MSG_OOBNc
C s� | j }| j}|r|| jO }|| jO }n|d ur!|| jO }|| jO }|| M }|| M }||O }||O }||@ | @ }|rCtd�|���||B | @ } | �|| @ || @ � d S )Nz6contradictory set, unset requirements for flags {0:#x})�msg_flags_common_set�msg_flags_common_unset�msg_flags_eor_indicator�msg_flags_non_eor_indicator� Exceptionr< r� )
rG r! �eor�checkset�
checkunset�ignoreZ
defaultsetZ defaultunsetZinbothr� r r r �
checkFlags�
s&
�zSendrecvmsgBase.checkFlags)Nr r r )rN rO rP r �LOOPBACK_TIMEOUT� fail_timeoutrH r� r� r� r� r� r� r� r- r� r r� r� r� r� rg r r r_ r r� F
s
r� c @ rS )�RecvmsgIntoMixinc G sh t |�}|j|gg|�R � }| �|� | �|d d� | �|d |� t|d |d � �f|dd � S r9 )r� � recvmsg_intor� �assertGreaterEqual�assertLessEqualr )rG r rr r] �bufr� r r r r� �
s
"zRecvmsgIntoMixin.doRecvmsgN)rN rO rP r� r r r r r� �
� r� c � e Zd Ze� fdd��Z� ZS )�SendrecvmsgDgramFlagsBasec � t � jtjB S r8 )rW r� r � MSG_TRUNCrF r_ r r r� �
� z5SendrecvmsgDgramFlagsBase.msg_flags_non_eor_indicator)rN rO rP �propertyr� rg r r r_ r r� �
� r� c r� )�SendrecvmsgSCTPFlagsBasec r� r8 )rW r� r �MSG_EORrF r_ r r r� �
r� z0SendrecvmsgSCTPFlagsBase.msg_flags_eor_indicator)rN rO rP r� r� rg r r r_ r r� �
r� r� c @ s8 e Zd Zedd� �Zedd� �Zedd� �Zdd� Zd S )
� SendrecvmsgConnectionlessBasec C r� r8 )rB rF r r r � serv_sock�
� z'SendrecvmsgConnectionlessBase.serv_sockc C r� r8 )r� rF r r r r� �
r� z&SendrecvmsgConnectionlessBase.cli_sockc C s g g d| j fS r� )r� rF r r r r� �
r� z8SendrecvmsgConnectionlessBase.sendmsg_to_server_defaultsc C s | j �|| j�S r8 )r� ra r� r� r r r r� �
r z*SendrecvmsgConnectionlessBase.sendToServerN)rN rO rP r� r� r� r� r� r r r r r� �
s
r� c @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )�SendrecvmsgConnectedBasec C r� r8 )r� rF r r r r� �
r� z"SendrecvmsgConnectedBase.serv_sockc C r� r8 )r� rF r r r r� �
r� z!SendrecvmsgConnectedBase.cli_sockc C r� r8 r r� r r r r� �
� z,SendrecvmsgConnectedBase.checkRecvmsgAddressN)rN rO rP r� r� r� r� r r r r r� �
s
r� c � e Zd Z� fdd�Z� ZS )� SendrecvmsgServerTimeoutBasec � t � �� | j�| j� d S r8 )rW rH r� r r� rF r_ r r rH �
z"SendrecvmsgServerTimeoutBase.setUp)rN rO rP rH rg r r r_ r r� s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zeed �d d � �Zejd d!� �Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*S )+� SendmsgTestsc C � | � | j�tt��t� d S r8 �r� r� r� r% r� rF r r r � testSendmsg � zSendmsgTests.testSendmsgc C s | � | �tg�tt�� d S r8 �r� r� r� r% rF r r r � _testSendmsg � zSendmsgTests._testSendmsgc C r� r8 r� rF r r r �testSendmsgDataGenerator r� z%SendmsgTests.testSendmsgDataGeneratorc C s& | � | �dd� tfD ��tt�� d S )Nc s r2 r8 r r: r r r r3 r4 z9SendmsgTests._testSendmsgDataGenerator.<locals>.<genexpr>r� rF r r r �_testSendmsgDataGenerator s �z&SendmsgTests._testSendmsgDataGeneratorc C r� r8 r� rF r r r � testSendmsgAncillaryGenerator r� z*SendmsgTests.testSendmsgAncillaryGeneratorc C s( | � | �tgdd� dD ��tt�� d S )Nc s r2 r8 r r: r r r r3 " r4 z>SendmsgTests._testSendmsgAncillaryGenerator.<locals>.<genexpr>r r� rF r r r � _testSendmsgAncillaryGenerator! s �z+SendmsgTests._testSendmsgAncillaryGeneratorc C r� r8 r� rF r r r �testSendmsgArray% r� z SendmsgTests.testSendmsgArrayc C s$ | � | �t�dt�g�tt�� d S )N�B)r� r� �arrayr� r% rF r r r �_testSendmsgArray) s �z SendmsgTests._testSendmsgArrayc C r� r8 r� rF r r r �testSendmsgGather- r� z SendmsgTests.testSendmsgGatherc C �. | � | �td d� tdd � g�tt�� d S �NrX r� rF r r r �_testSendmsgGather1 s .z SendmsgTests._testSendmsgGatherc C � | � | j�d�d� d S �Nr� � done�r� r� r� rF r r r �testSendmsgBadArgs4 � z SendmsgTests.testSendmsgBadArgsc C s� | � t| jj� | � t| jd� | � t| jt� � | � t| jt� g� | � t| jtt� g� | � t| jtgt� � | � t| jtgg t� � | � t| jtgg dt� � | �d� d S )Ns not in an iterabler r� )rY r� r� r( r� �objectr� r� rF r r r �_testSendmsgBadArgs8 s.
�
�
�
�
�
�
�z SendmsgTests._testSendmsgBadArgsc C r� r� r� rF r r r �testSendmsgBadCmsgJ r� z SendmsgTests.testSendmsgBadCmsgc C s� | � t| jtgt� g� | � t| jtgt� ddfg� | � t| jtgdt� dfg� | � t| jtgddt� fg� | � t| jtgdg� | � t| jtgdg� | �d� d S )Nr � data)r r )r r r� ry r� �rY r� r� r� r� r� rF r r r �_testSendmsgBadCmsgN s&
�
�
�
�
�
�z SendmsgTests._testSendmsgBadCmsg�
CMSG_SPACEc C r� r� r� rF r r r �testSendmsgBadMultiCmsg] s z$SendmsgTests.testSendmsgBadMultiCmsgc C s@ | � t| jtgg d�� | � t| jtgdt� g� | �d� d S )N�r r rc r� r� rF r r r �_testSendmsgBadMultiCmsgc s
�
�z%SendmsgTests._testSendmsgBadMultiCmsgc C r� r� r� rF r r r �testSendmsgExcessCmsgRejectk � z(SendmsgTests.testSendmsgExcessCmsgRejectc C sb t td�s*| �t��}| �tgddg� W d � n1 s w Y | �|jj� | � d� d S )Nr� r� r� )
r r rY r r� r� � assertIsNonerb r� r� �rG rc r r r � _testSendmsgExcessCmsgRejectp s
�z)SendmsgTests._testSendmsgExcessCmsgRejectc C r� r8 r rF r r r �testSendmsgAfterClosex s z"SendmsgTests.testSendmsgAfterClosec C s | j �� | �t| jtg� d S r8 )r� r rY r r� r� rF r r r �_testSendmsgAfterClose| s
z#SendmsgTests._testSendmsgAfterCloseN)rN rO rP r� r� r� r� r� r� r� r� r� r� r� r� r� r� r7 r r� r# r� r� r� r� r� r r r r r�
s.
r� c @ sR e Zd Zdd� Zdd� Zdd� Zdd� Zeej d vd
�d d � �Z
e
j d
d� �Z dS )�SendmsgStreamTestsc C r� r8 r� rF r r r �testSendmsgExplicitNoneAddr� r� z.SendmsgStreamTests.testSendmsgExplicitNoneAddrc C s" | � | �tgg dd �tt�� d S r� r� rF r r r � _testSendmsgExplicitNoneAddr� s "z/SendmsgStreamTests._testSendmsgExplicitNoneAddrc C �. | � | j�d�d� | �| jj| jd�� d S �Nr� � aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa�r; �r� r� r� rA r� r� r� rF r r r �testSendmsgTimeout� s z%SendmsgStreamTests.testSendmsgTimeoutc
C s� z@| j �d� z | �dg� q ty Y n ty0 } z
|jtjkr&� W Y d }~n
d }~ww W | j� � d S W | j� � d S | j� � w )N� ��Q��?Tr� )
r� r r� rA r r� �ENOMEMrU r� r| �rG r r r r �_testSendmsgTimeout� s$ � ���� �z&SendmsgStreamTests._testSendmsgTimeout> r� z<MSG_DONTWAIT not known to work on this platform when sendingc C r� r� r� rF r r r �testSendmsgDontWait� s z&SendmsgStreamTests.testSendmsgDontWaitc C sl z/| � t�� } | �dgg tj� q1 sw Y | �|jjtjtj tj
f� W | j � � d S | j � � w )NTr� )
rY r r� r � MSG_DONTWAITrI rb r� �EAGAIN� EWOULDBLOCKr� r� r| r� r r r �_testSendmsgDontWait� s ��
�z'SendmsgStreamTests._testSendmsgDontWaitN)
rN rO rP r� r� r� r� r* r� r� r� r# r� r r r r r� � s
�
r� c @ r? )�SendmsgConnectionlessTestsc C r� r8 r rF r r r �testSendmsgNoDestAddr� r� z0SendmsgConnectionlessTests.testSendmsgNoDestAddrc C s2 | � t| jjtg� | � t| jjtgg dd � d S r� )rY r r� r( r� rF r r r �_testSendmsgNoDestAddr� s �
�z1SendmsgConnectionlessTests._testSendmsgNoDestAddrN)rN rO rP r� r� r r r r r� � s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zeed!�d"d#� �Zejd$d%� �Zeejd&�d'd(� �Zejd)d*� �Zd+S ),�RecvmsgGenericTestsc C sR | � | jtt��\}}}}| �|t� | �|| j� | �|g � | j|dd� d S �NT�r� �r� r� r% r� r� r� r� r� �rG r� r& r! r� r r r � testRecvmsg� s
z RecvmsgGenericTests.testRecvmsgc C � | � t� d S r8 �r� r� rF r r r � _testRecvmsg� rI z RecvmsgGenericTests._testRecvmsgc C sV | � | jtt�dd�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr Tr r r r r r �testRecvmsgExplicitDefaults� �
� z/RecvmsgGenericTests.testRecvmsgExplicitDefaultsc C r r8 r rF r r r � _testRecvmsgExplicitDefaults� rI z0RecvmsgGenericTests._testRecvmsgExplicitDefaultsc C sV | � | jtt�d �\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nry Tr r r r r r �testRecvmsgShorter� r z&RecvmsgGenericTests.testRecvmsgShorterc C r r8 r rF r r r �_testRecvmsgShorter� rI z'RecvmsgGenericTests._testRecvmsgShorterc C s^ | � | jtt�d �\}}}}| �|td d� � | �|| j� | �|g � | j|dd� d S )NrX ���Fr r r r r r �testRecvmsgTrunc� s
� z$RecvmsgGenericTests.testRecvmsgTruncc C r r8 r rF r r r �_testRecvmsgTrunc� rI z%RecvmsgGenericTests._testRecvmsgTruncc C �T | � | jtt�d�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr~ Tr r r r r r � testRecvmsgShortAncillaryBuf � � z0RecvmsgGenericTests.testRecvmsgShortAncillaryBufc C r r8 r rF r r r � _testRecvmsgShortAncillaryBuf rI z1RecvmsgGenericTests._testRecvmsgShortAncillaryBufc C r )N� ( Tr r r r r r �testRecvmsgLongAncillaryBuf r z/RecvmsgGenericTests.testRecvmsgLongAncillaryBufc C r r8 r rF r r r � _testRecvmsgLongAncillaryBuf rI z0RecvmsgGenericTests._testRecvmsgLongAncillaryBufc C s" | j �� | �t| j| j d� d S r� )r� r rY r r� rF r r r �testRecvmsgAfterClose s
z)RecvmsgGenericTests.testRecvmsgAfterClosec C r� r8 r rF r r r �_testRecvmsgAfterClose r� z*RecvmsgGenericTests._testRecvmsgAfterClosec C sB z| j �d� | �t| j| j tt�� W | j�� d S | j�� w )Nr� ) r� r rY rA r� r% r� r� r| rF r r r �testRecvmsgTimeout" s �z&RecvmsgGenericTests.testRecvmsgTimeoutc C s | � | jj| jd�� d S �Nr� )rA r� r� r� rF r r r �_testRecvmsgTimeout+ r� z'RecvmsgGenericTests._testRecvmsgTimeoutr" c C s | � | jtt�d dtj�\}}}}| �|td d� � | �|| j� | �|g � | j |dt
tdd�d� | � | jtt�dtj�\}}}}| �|t� | �|| j� | �|g � | j |dd� | � | jtt��\}}}}| �|t� | �|| j� | �|g � | j |dd� d S ) NrX r r Fr� �r� r� Tr ) r� r� r% r� r r" r� r� r� r� r� r r r r �testRecvmsgPeek. s. �
� � z#RecvmsgGenericTests.testRecvmsgPeekc C r r8 r rF r r r �_testRecvmsgPeekQ r� z$RecvmsgGenericTests._testRecvmsgPeekr( c C s` | j �| j� | �| j tt��\}}}}| �|t� | �|| j� | �|g � | j |dd� d S r� )
r� r r� r� r% r� r� r� r� r� r r r r �testRecvmsgFromSendmsgU s z*RecvmsgGenericTests.testRecvmsgFromSendmsgc C r� r� r� rF r r r �_testRecvmsgFromSendmsg` s .z+RecvmsgGenericTests._testRecvmsgFromSendmsgN)rN rO rP r r r r r
r r
r r r r r r r r r r7 r r r# r r r r r r r r� � s2
"
r� c @ r= )
�RecvmsgGenericStreamTestsc C sN | � | jd�\}}}}| �|d� | �|| j� | �|g � | j|d d� d S )Nr� rc r )r� r� r� r� r� r� r r r r �testRecvmsgEOFi s
z(RecvmsgGenericStreamTests.testRecvmsgEOFc C � | j �� d S r8 )r� r rF r r r �_testRecvmsgEOFq rI z)RecvmsgGenericStreamTests._testRecvmsgEOFc C s� | � | jtt�d �\}}}}| �|| j� | �|g � | j|dd� | � | jd�\}}}}| �|| j� | �|g � | j|dd� || }| �|t� d S )NrX Fr r� T)r� r� r% r� r� r� r� r� )rG rY r& r! r� rZ r� r r r �testRecvmsgOverflowt s
� z-RecvmsgGenericStreamTests.testRecvmsgOverflowc C r r8 r rF r r r �_testRecvmsgOverflow� rI z.RecvmsgGenericStreamTests._testRecvmsgOverflowN)rN rO rP r! r# r$ r% r r r r r e s
r c @ r? )� RecvmsgTestsc C s | � t| jj� | � t| jjddd� | � t| jjtt�dd� | � t| jjtd�gdd� | � t| jjt� dd� | � t| jjtt�t� d� | � t| jjtt�dt� � | j�tt�dd�\}}}}| � |t� | �
|| j � | � |g � | j |dd� d S )Nr� r r� Tr )
rY r� r� r% r� r% r� r� r� r� r� r� r� r r r r �testRecvmsgBadArgs� s0 �
� � � � � z RecvmsgTests.testRecvmsgBadArgsc C r r8 r rF r r r �_testRecvmsgBadArgs� rI z RecvmsgTests._testRecvmsgBadArgsN)rN rO rP r' r( r r r r r& � s r& c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�RecvmsgIntoTestsc C sL t tt��}| �t| jj� | �t| jjtt�dd� | �t| jj|dd� | �t| jjt� gdd� | �t| jjdgdd� | �t| jj|t� gdd� | �t| jj|gdd� | �t| jj|gt� d� | �t| jj|gdt� � | j�|gdd�\}}}}| � |tt�� | � |t t�� | �
|| j � | � |g � | j |dd� d S )Nr s I'm not writabler� Tr )
r� r% r� rY r� r� r� r� r� r� r� r� r� �rG r� �nbytesr& r! r� r r r �testRecvmsgIntoBadArgs� s@
� �
� � � �
�
� z'RecvmsgIntoTests.testRecvmsgIntoBadArgsc C r r8 r rF r r r �_testRecvmsgIntoBadArgs� rI z(RecvmsgIntoTests._testRecvmsgIntoBadArgsc C sx t tt��}| j�dd� |fD ��\}}}}| �|tt�� | �|t t�� | �|| j� | �|g � | j|dd� d S )Nc s r2 r8 r r: r r r r3 � r4 z<RecvmsgIntoTests.testRecvmsgIntoGenerator.<locals>.<genexpr>Tr ) r� r% r� r� r� r� r� r� r� r* r r r �testRecvmsgIntoGenerator� s � z)RecvmsgIntoTests.testRecvmsgIntoGeneratorc C r r8 r rF r r r �_testRecvmsgIntoGenerator� rI z*RecvmsgIntoTests._testRecvmsgIntoGeneratorc C sx t � ddgtt� �}| j�|g�\}}}}| �|tt�� | �|�� t� | �|| j� | �|g � | j |dd� d S )Nr� r Tr )
r� r% r� r� r� r� �tobytesr� r� r� r* r r r �testRecvmsgIntoArray� s z%RecvmsgIntoTests.testRecvmsgIntoArrayc C r r8 r rF r r r �_testRecvmsgIntoArray� rI z&RecvmsgIntoTests._testRecvmsgIntoArrayc C s� t d�}t d�}t d�}| j�|t|�dd� |g�\}}}}| �|td�� | �|t d�� | �|t d�� | �|t d �� | �|| j� | �|g � | j|d
d � d S ) Ns ----s
0123456789s --------------r� � � Mary had a little lambs Marys
01 had a 9s little lamb---Tr ) r� r� r� �
memoryviewr� r% r� r� r� )rG �b1�b2Zb3r+ r& r! r� r r r �testRecvmsgIntoScatter� s � z'RecvmsgIntoTests.testRecvmsgIntoScatterc C rE )Nr4 )r� rF r r r �_testRecvmsgIntoScatter� rI z(RecvmsgIntoTests._testRecvmsgIntoScatterN) rN rO rP r, r- r. r/ r1 r2 r8 r9 r r r r r) � s
r) c @ s` e Zd ZzddlZW n ey dZY nw edej�Zee d�dd� �Z
ee d�dd � �Z dS )
�CmsgMacroTestsr Ni����CMSG_LENc C s� | j t�d� d }ttd��tt|d |�� }| �t�d�t�d�jd � |D ]}t�|�}| �|t�d� |� | � || j � q*| �
t tjd� | �
t tj|� | �
t tjt j
� d S )Nr r~ � r> r� r� )�socklen_t_limitr r; �listrd r� r� �itemsizer� r� rY r� r� �maxsize)rG �toobigr� �nr2 r r r � testCMSG_LEN
s
zCmsgMacroTests.testCMSG_LENr� c C s� | j t�d� d }ttd��tt|d |�� }t�d�}| �|t�d�jd � |D ]*}t�|�}| �||� | �|t� |�� | �||t� d� � | �
|| j � |}q,| � t tjd� | � t tj|� | � t tjt
j� d S )Nr~ r< r r> r� r� )r= r r� r> rd r� r� r? r� r; r� rY r� r� r@ )rG rA r� ZlastrB r2 r r r �testCMSG_SPACE
s
z CmsgMacroTests.testCMSG_SPACE) rN rO rP r� r� r= �minr� r7 r rC rD r r r r r: � s �
r: c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zdddd�Z dd� Z
dd� Z dd� Z dd� Z
eed�dd� �Zejdd � �Zd d � Zd d � Ze�ejd!kd"�e�ed#�eed�d$d%� ���Zeje�ejd!kd"�e�ed#�d&d'� ���Ze�ejd!kd"�e�ed#�eed�d(d)� ���Z e je�ejd!kd"�e�ed#�d*d+� ���Z d,d-� Z e�ejd!kd.�d/d0� �Z d1d2� Z d3d4� Z!d5d6� Z"eed�d7d8� �Z#e#jd9d:� �Z$ded;d<�Z%d=d>� Z&d?d@� Z'dAdB� Z(dCdD� Z)dEdF� Z*dGdH� Z+dIdJ� Z,dKdL� Z-dMdN� Z.dOdP� Z/dedQdR�Z0dSdT� Z1dUdV� Z2dWdX� Z3dYdZ� Z4d[d\� Z5d]d^� Z6d_d`� Z7dadb� Z8dcS )f�
SCMRightsTesti����c C s\ g }t |�D ]%}t�� \}}| �tj|� | �tj|� t�|t|�� � � |�
|� q|S r8 ) rd r� �mkstemprc r� r r �writer8 �encode�append)rG rB �fdsr> r� r r r r �newFDs8
s zSCMRightsTest.newFDsc C sB t |�D ]\}}t�|dtj� | �t�|d�t|��� � qd S )Nr r� )� enumerater� �lseek�SEEK_SETr� re r8 rI )rG rK rB r� r r r �checkFDsD
s �zSCMRightsTest.checkFDsc C s | � | j|� d S r8 )rc �closeRecvmsgFDsr� r r r r� K
r� z#SCMRightsTest.registerRecvmsgResultc C sn |d D ]0\}}}|t jkr4|t jkr4t�d�}|�|d t|�t|�|j � � |D ]}t�|� q,qd S )Nr~ r> ) r rg �
SCM_RIGHTSr� � frombytesr% r? r� r )rG Zrecvmsg_result�
cmsg_level� cmsg_type� cmsg_datarK r� r r r rQ N
s
�
� ��z SCMRightsTest.closeRecvmsgFDsc
C s8 | � | �tgtjtjt�d| �|��fg�tt�� d S )Nr> ) r� r� r� r rg rR r� rL r% )rG rB r r r �createAndSendFDsZ
s ���z SCMRightsTest.createAndSendFDsr~ r c C s� |\}}}}| � |t� | �|| j� | j|dtj|d� | �|t� | � t
|�|� t � d�} |D ]0}
| �|
t � |
\} } }
| � | tj
� | � | tj� | �|
t� | � t
|
�t d� | �|
� q2| � t
| �|� | �| � d S )NT�r� r� r� r> r )r� r� r� r� r� r r� rC r> r� r% r� �tuplerg rR r �
SIZEOF_INTrS rP )rG Znumfdsr� �maxcmsgs� ignoreflagsr� r& r! r� rK �itemrT rU rV r r r �checkRecvmsgFDsd
s& �
z SCMRightsTest.checkRecvmsgFDsc C � | � d| �| jtt�d�� d S )Nr~ r �r^ r� r� r% r� rF r r r �testFDPassSimple
�
�z SCMRightsTest.testFDPassSimplec
C s< | � | �tgtjtjt�d| �d���� fg�t t�� d S )Nr> r~ )
r� r� r� r rg rR r� rL r0 r% rF r r r �_testFDPassSimple�
s ���z SCMRightsTest._testFDPassSimplec C r_ )Nr` r r` rF r r r �testMultipleFDPass�
rb z SCMRightsTest.testMultipleFDPassc C rE �Nr` �rW rF r r r �_testMultipleFDPass�
rI z!SCMRightsTest._testMultipleFDPassr� c C s* | � d| �| jtt�t�dt ��� d S re )r^ r� r� r% r� r r� rZ rF r r r �testFDPassCMSG_SPACE�
s
��z"SCMRightsTest.testFDPassCMSG_SPACEc C rE re rf rF r r r �_testFDPassCMSG_SPACE�
r� z#SCMRightsTest._testFDPassCMSG_SPACEc
C s0 | j d| �| jtt�t�dt ��tjd� d S )Nr~ r` �r\ ) r^ r� r� r% r� r r; rZ r� rF r r r �testFDPassCMSG_LEN�
s �
�z SCMRightsTest.testFDPassCMSG_LENc C rE r rf rF r r r �_testFDPassCMSG_LEN�
rI z!SCMRightsTest._testFDPassCMSG_LENrr zskipping, see issue #12958zskipping, see issue #22397c C s$ | j d| �| jtt�d�dd� d S )Nr� r )r[ r` rF r r r �testFDPassSeparate�
s
�z SCMRightsTest.testFDPassSeparatec C �X | � d�\}}| �| �tgtjtjt�d|g�ftjtjt�d|g�fg�tt�� d S �Nr� r> � rL r� r� r� r rg rR r� r% �rG Zfd0�fd1r r r �_testFDPassSeparate�
� � ���z!SCMRightsTest._testFDPassSeparatec C s@ d}| j || �| jtt�t�t�t�t| � �dtj d� d S )Nr� )r[ r\ )
r^ r� r� r% r� r r� rZ r; r� )rG Znum_fdsr r r �testFDPassSeparateMinSpace�
s ��
�z(SCMRightsTest.testFDPassSeparateMinSpacec C rn ro rp rq r r r �_testFDPassSeparateMinSpace�
rt z)SCMRightsTest._testFDPassSeparateMinSpacec
C sf z | � |g|�}W n ty( } z| �|jt� | � |g�}W Y d }~nd }~ww | �|t|�� d S r8 )r� r rC r� r� r� r% )rG r� r& r+ r� r r r �sendAncillaryIfPossible�
s ��z%SCMRightsTest.sendAncillaryIfPossiblezsee issue #24725c C s& | j d| �| jtt�d�tjd� d S )Nr r rj )r^ r� r� r% r� r r� rF r r r �testFDPassEmpty�
s
�
�z SCMRightsTest.testFDPassEmptyc C s | � ttjtjdfg� d S �Nrc )rw r� r rg rR rF r r r �_testFDPassEmpty�
s
�z SCMRightsTest._testFDPassEmptyc C s� | � | jtt�d�\}}}}| �|t� | �|| j� | j|dtj d� | �
t|�d� |D ]\}}}| �|tj � | �|tj � | �
t|�t� q.d S )Nr Tr r~ )r� r� r% r� r� r� r� r� r r� r� rg rR �
assertLessrZ �rG r� r& r! r� rT rU rV r r r �testFDPassPartialInt�
s � �z"SCMRightsTest.testFDPassPartialIntc C s4 | � ttjtjt�d| jg��� d d� fg� d S )Nr> r� )rw r� r rg rR r� �badfdr0 rF r r r �_testFDPassPartialInt s ��z#SCMRightsTest._testFDPassPartialIntc C s� | � | jtt�d�\}}}}| �|t� | �|| j� | j|dtj d� | �
t|�d� t � d�}|D ]%\}}}| �|tj � | �|tj
� |�|d t|�t|�|j � � q3| �
t|�d� | �|� d S )Nr Tr r� r> )r� r� r% r� r� r� r� r� r r� r� r� rg rR rS r? rP ) rG r� r& r! r� rK rT rU rV r r r � testFDPassPartialIntInMiddle s �
�z*SCMRightsTest.testFDPassPartialIntInMiddlec
C sZ | � d�\}}| �ttjtjt�d|| jg��� d d� ftjtjt�d|g�fg� d S )Nr� r> r� ) rL rw r� r rg rR r� r~ r0 rq r r r � _testFDPassPartialIntInMiddle s � ���z+SCMRightsTest._testFDPassPartialIntInMiddlec C sJ |\}}}}| � |t� | �|| j� | � |g � | j|dtj|d� d S )NT�r� r� r� )r� r� r� r� r� r r� )rG r� r\ r� r& r! r� r r r �checkTruncatedHeader+ s
�z"SCMRightsTest.checkTruncatedHeaderc C s" | j | �| jtt��tjd� d S �Nrj �r� r� r� r% r� r r� rF r r r �testCmsgTruncNoBufSize5 s
�z$SCMRightsTest.testCmsgTruncNoBufSizec C rE r rf rF r r r �_testCmsgTruncNoBufSize> rI z%SCMRightsTest._testCmsgTruncNoBufSizec C s$ | j | �| jtt�d�tjd� d S )Nr rj r� rF r r r �testCmsgTrunc0A s
�z SCMRightsTest.testCmsgTrunc0c C rE r rf rF r r r �_testCmsgTrunc0F rI z SCMRightsTest._testCmsgTrunc0c C s | � | �| jtt�d�� d S r )r� r� r� r% r� rF r r r �testCmsgTrunc1L s z SCMRightsTest.testCmsgTrunc1c C rE r rf rF r r r �_testCmsgTrunc1O rI z SCMRightsTest._testCmsgTrunc1c C s" | � | �| jtt�td �� d S rF )r� r� r� r% r� rZ rF r r r �testCmsgTrunc2IntR s
�z SCMRightsTest.testCmsgTrunc2Intc C rE r rf rF r r r �_testCmsgTrunc2IntY rI z SCMRightsTest._testCmsgTrunc2Intc C s( | � | �| jtt�t�d�d �� d S r9 )r� r� r� r% r� r r; rF r r r �testCmsgTruncLen0Minus1\ s
�z%SCMRightsTest.testCmsgTruncLen0Minus1c C rE r rf rF r r r �_testCmsgTruncLen0Minus1` rI z&SCMRightsTest._testCmsgTruncLen0Minus1c C s� | � | jtt�|�\}}}}| �|t� | �|| j� | j|dtj d� |dkr.|g kr.d S | �t|�d� |d \}} }
| �|tj
� | �| tj � | � t|
�|� | �
t|
�|� t�d�} | �|
d t|
�t|
�| j � � | �| � d S )NT�r� r� r r~ r> )r� r� r% r� r� r� r� r� r r� rg rR r� r� r� rS r? rP ) rG �ancbuf�maxdataZmindatar� r& r! r� rT rU rV rK r r r �checkTruncatedArrayf s& �
�z!SCMRightsTest.checkTruncatedArrayc C s | j t�d�dd� d S )Nr �r� r� �r� r r; rF r r r �testCmsgTruncLen0~ � z SCMRightsTest.testCmsgTruncLen0c C rE r rf rF r r r �_testCmsgTruncLen0� rI z SCMRightsTest._testCmsgTruncLen0c C s | j t�d�d dd� d S )Nr r~ r� r� rF r r r �testCmsgTruncLen0Plus1� r� z$SCMRightsTest.testCmsgTruncLen0Plus1c C rE rF rf rF r r r �_testCmsgTruncLen0Plus1� rI z%SCMRightsTest._testCmsgTruncLen0Plus1c C s | j t�t�td� d S )Nr� �r� r r; rZ rF r r r �testCmsgTruncLen1� s
�z SCMRightsTest.testCmsgTruncLen1c C rE rF rf rF r r r �_testCmsgTruncLen1� rI z SCMRightsTest._testCmsgTruncLen1c C s( | j t�dt �d dt d d� d S )Nr� r~ r� r� rF r r r �testCmsgTruncLen2Minus1� s
�z%SCMRightsTest.testCmsgTruncLen2Minus1c C rE rF rf rF r r r �_testCmsgTruncLen2Minus1� rI z&SCMRightsTest._testCmsgTruncLen2Minus1N)r~ r �r )9rN rO rP r~ rL rP r� rQ rW r^ ra rc rd rg r7 r rh r# ri rk rl r� r� r� r� r rm rs ru rv rw rx rz r} r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rF 0
s�
rF c @ s e Zd ZdZdZdd� Zd^dd�Zeed d
�d d � �Z e j
d
d� �Z eedd d
�dd� �Z e j
dd� �Z
eejd�eed d
�dd� ��Zej
dd� �Z d^dd�Zeed d
dd �d d � �Zej
d d � �Zeedd d
dd �d!d"� �Zej
d#d$� �Zeejd�eedd d
dd �d%d&� ��Zej
d'd(� �Zeejd�eedd d
dd �d)d*� ��Zej
d+d,� �Zd_d-d.�Zeed d
�d/d0� �Zej
d1d2� �Zeed d
�d3d4� �Z e j
d5d6� �Z eed d
�d7d8� �Z e j
d9d:� �Z eed d
�d;d<� �Z e j
d=d>� �Z!eed d
�d?d@� �Z"e"j
dAdB� �Z#eed d
�dCdD� �Z$e$j
dEdF� �Z%d_dGdH�Z&eedd d
dd �dIdJ� �Z'e'j
dKdL� �Z(eedd d
dd �dMdN� �Z)e)j
dOdP� �Z*eedd d
dd �dQdR� �Z+e+j
dSdT� �Z,eedd d
dd �dUdV� �Z-e-j
dWdX� �Z.eedd d
dd �dYdZ� �Z/e/j
d[d\� �Z0d]S )`�RFC3542AncillaryTestr� r� c C s4 i }|D ]\}}}| � ||f|� ||||f< q|S r8 )rL )rG r& �drT rU rV r r r �ancillaryMapping� s
z%RFC3542AncillaryTest.ancillaryMappingr r c C s | j �tjtjd� | j�� | �| j tt �|�\}}}}| �
|t � | � || j � | j
|dtj|d� | �
t|�d� | �|d t� |d \}} }
| �
|tj� | �
| tj� | �|
t� | �
t|
�t� t�d�} | �|
� | �| d d� | �| d |� d S )Nr~ TrX r r> )r� r r rl r r� r| r� r% r� r� r� r� r� r� rC rY rz r rZ r� rS r� r� ) rG �
ancbufsize�maxhopr\ r� r& r! r� rT rU rV r� r r r �
checkHopLimit� s.
�
� �
z"RFC3542AncillaryTest.checkHopLimitr rz c C � | j dd� d S �Nr �r� )r� rF r r r �testRecvHopLimit� r� z%RFC3542AncillaryTest.testRecvHopLimitc C �$ | � | jj| jd�� | �t� d S r �rA r� r� r� r� r� rF r r r �_testRecvHopLimit� s z&RFC3542AncillaryTest._testRecvHopLimitr� c C s | j t�t�d� d S )Nr� )r� r r� rZ rF r r r �testRecvHopLimitCMSG_SPACE� � z/RFC3542AncillaryTest.testRecvHopLimitCMSG_SPACEc C r� r r� rF r r r �_testRecvHopLimitCMSG_SPACE� � z0RFC3542AncillaryTest._testRecvHopLimitCMSG_SPACEr( c C � | j d| jd� d S �Nr )r� r� )r� � hop_limitrF r r r �testSetHopLimit� s z$RFC3542AncillaryTest.testSetHopLimitc C sL | � | jj| jd�� | �| �tgtjtj t
�
d| j g�fg�t t�� d S �Nr� r> )
rA r� r� r� r� r� r� r rl rz r� r� r% rF r r r �_testSetHopLimit� s ���z%RFC3542AncillaryTest._testSetHopLimitc C sB | j �tjtjd� | j �tjtjd� | j�� | �| j t t
�|�\}}}}| � |t
� | � || j
� | j|dtj|d� | � t |�d� | �|�}|tjtjf } | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d d� |tjtjf } | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d |� d S )Nr~ TrX r� r> r r )r� r r rl r rv r� r| r� r% r� r� r� r� r� r� r� r� rZ r� rS r� r� rz ) rG r� r� r\ r� r& r! r� ZancmapZtcdatar� Zhldatar r r � checkTrafficClassAndHopLimit� s:
�
�
� �
z1RFC3542AncillaryTest.checkTrafficClassAndHopLimitrv r� c C r� r� )r� rF r r r � testRecvTrafficClassAndHopLimit s z4RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitc C r� r r� rF r r r � _testRecvTrafficClassAndHopLimit& r� z5RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitc C s | j t�t�d d� d S �Nr� r� )r� r r� rZ rF r r r �)testRecvTrafficClassAndHopLimitCMSG_SPACE+ s
�z>RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitCMSG_SPACEc C r� r r� rF r r r �*_testRecvTrafficClassAndHopLimitCMSG_SPACE3 r� z?RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitCMSG_SPACEc C r� r� �r� r� rF r r r � testSetTrafficClassAndHopLimit8 s
�z3RFC3542AncillaryTest.testSetTrafficClassAndHopLimitc C sd | � | jj| jd�� | �| �tgtjtj t
�
d| j g�ftjtj t
�
d| j
g�fg�tt�� d S r� )rA r� r� r� r� r� r� r rl r� r� �
traffic_classrz r� r% rF r r r � _testSetTrafficClassAndHopLimitA s �����z4RFC3542AncillaryTest._testSetTrafficClassAndHopLimitc C r� r� r� rF r r r �testOddCmsgSizeL s
�z$RFC3542AncillaryTest.testOddCmsgSizec
C s� | � | jj| jd�� z&| �tgtjtjt � d| j
g�� � d ftjtj t � d| j
g�fg�}W d S tyr } z5| �|jt� | �tgtjtjt � d| j
g�ftjtj t � d| j
g�fg�}| �|tt�� W Y d }~d S d }~ww )Nr� r> r� )rA r� r� r� r� r� r rl r� r� r� r0 rz r� r rC r� r� r� r% )rG r+ r� r r r �_testOddCmsgSizeV s6 ���
����� ��z%RFC3542AncillaryTest._testOddCmsgSizec C s� | j �tjtjd� | j�� |d u rdn|f}| j| j tt �g|�R � \}}}}| �
|t � | � || j � | �
|g � | j
|dtj|d� d S )Nr~ r Tr� )r� r r rl r r� r| r� r% r� r� r� r� r� r� )rG r� r\ r] r� r& r! r� r r r � checkHopLimitTruncatedHeaderl s
�
��
�z1RFC3542AncillaryTest.checkHopLimitTruncatedHeaderc C s | j d tjd� d S )N�r� r\ �r� r r� rF r r r r� � s
�z+RFC3542AncillaryTest.testCmsgTruncNoBufSizec C r� r r� rF r r r r� � r� z,RFC3542AncillaryTest._testCmsgTruncNoBufSizec C s | j dtjd� d S )Nr r� r� rF r r r �testSingleCmsgTrunc0� s
�z)RFC3542AncillaryTest.testSingleCmsgTrunc0c C r� r r� rF r r r �_testSingleCmsgTrunc0� r� z*RFC3542AncillaryTest._testSingleCmsgTrunc0c C r� )Nr~ r� )r� rF r r r �testSingleCmsgTrunc1� � z)RFC3542AncillaryTest.testSingleCmsgTrunc1c C r� r r� rF r r r �_testSingleCmsgTrunc1� r� z*RFC3542AncillaryTest._testSingleCmsgTrunc1c C s | j dt d� d S r� )r� rZ rF r r r �testSingleCmsgTrunc2Int� � z,RFC3542AncillaryTest.testSingleCmsgTrunc2Intc C r� r r� rF r r r �_testSingleCmsgTrunc2Int� r� z-RFC3542AncillaryTest._testSingleCmsgTrunc2Intc C s | j t�d�d d� d S )Nr r~ r� )r� r r; rF r r r � testSingleCmsgTruncLen0Minus1� � z2RFC3542AncillaryTest.testSingleCmsgTruncLen0Minus1c C r� r r� rF r r r � _testSingleCmsgTruncLen0Minus1� r� z3RFC3542AncillaryTest._testSingleCmsgTruncLen0Minus1c C s� | j �tjtjd� | j�� | �| j tt �t�
t �d �\}}}}| � |t � | �
|| j� | j|dtjd� | �t|�d� |ra|d \}}}| � |tj� | � |tj� | �t|�t � d S d S )Nr~ Tr� r )r� r r rl r r� r| r� r% r� r; rZ r� r� r� r� r� r� rz r{ r| r r r �testSingleCmsgTruncInData� s"
�
� �z.RFC3542AncillaryTest.testSingleCmsgTruncInDatac C r� r r� rF r r r �_testSingleCmsgTruncInData� r� z/RFC3542AncillaryTest._testSingleCmsgTruncInDatac C s | j �tjtjd� | j �tjtjd� | j�� | �| j t t
�|�\}}}}| � |t
� | � || j
� | j|dtj|d� | � t |�d� |d \}}} | � |tj� | �|tjtjh� | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d d� d S )Nr~ Tr� r r> r )r� r r rl r rv r� r| r� r% r� r� r� r� r� r� rI r� rz rZ r� rS r� r� ) rG r� r\ r� r& r! r� rT rU rV r� r r r �checkTruncatedSecondHeader� s0
�
�
� �
z/RFC3542AncillaryTest.checkTruncatedSecondHeaderc C s | j t�t�tjd� d S r� )r� r r� rZ r� rF r r r �testSecondCmsgTrunc0� s
�z)RFC3542AncillaryTest.testSecondCmsgTrunc0c C r� r r� rF r r r �_testSecondCmsgTrunc0� r� z*RFC3542AncillaryTest._testSecondCmsgTrunc0c C s | � t�t�d � d S r �r� r r� rZ rF r r r �testSecondCmsgTrunc1 r� z)RFC3542AncillaryTest.testSecondCmsgTrunc1c C r� r r� rF r r r �_testSecondCmsgTrunc1 r� z*RFC3542AncillaryTest._testSecondCmsgTrunc1c C s | � t�t�dt � d S rF r� rF r r r �testSecondCmsgTrunc2Int
s
�z,RFC3542AncillaryTest.testSecondCmsgTrunc2Intc C r� r r� rF r r r �_testSecondCmsgTrunc2Int r� z-RFC3542AncillaryTest._testSecondCmsgTrunc2Intc C s" | � t�t�t�d� d � d S r9 )r� r r� rZ r; rF r r r � testSecondCmsgTruncLen0Minus1 s
�
�z2RFC3542AncillaryTest.testSecondCmsgTruncLen0Minus1c C r� r r� rF r r r � _testSecondCmsgTruncLen0Minus1 r� z3RFC3542AncillaryTest._testSecondCmsgTruncLen0Minus1c
C sR | j �tjtjd� | j �tjtjd� | j�� | �| j t t
�t� t �t�
t � d �\}}}}| �|t
� | �|| j� | j|dtjd� tjtjh}|�d�\}}}| �|tj� |�|� | �t |�t � t�d�} | �|� | �| d d� | �| d d� |r�|�d�\}}}| �|tj� |�|� | �t |�t � | �|g � d S )Nr~ Tr� r r> r ) r� r r rl r rv r� r| r� r% r� r� rZ r; r� r� r� r� r� r� rz �pop�remover� rS r� r� r{ )
rG r� r& r! r� Z
cmsg_typesrT rU rV r� r r r �testSecondCmsgTruncInData s:
�
�
�
z.RFC3542AncillaryTest.testSecondCmsgTruncInDatac C r� r r� rF r r r �_testSecondCmsgTruncInDataE r� z/RFC3542AncillaryTest._testSecondCmsgTruncInDataN)r r r� )1rN rO rP r� r� r� r� r7 r r� r# r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s�
�$�
�
�
�
�
�
�
�
�
#r� c @ � e Zd ZdS )�SendrecvmsgUDPTestBaseN�rN rO rP r r r r r� M � r� r( c @ r� )�SendmsgUDPTestNr� r r r r r� R � r� r% c @ r� )�RecvmsgUDPTestNr� r r r r r� V r� r� r� c @ r� )�RecvmsgIntoUDPTestNr� r r r r r� Z r� r� c @ rS )�SendrecvmsgUDP6TestBasec C � | � |d d� |d d� � d S r� r� r� r r r r� c � z+SendrecvmsgUDP6TestBase.checkRecvmsgAddressN�rN rO rP r� r r r r r� _ � r� ro r r$ c @ r� )�SendmsgUDP6TestNr� r r r r r� h � r� c @ r� )�RecvmsgUDP6TestNr� r r r r r� n r� r� c @ r� )�RecvmsgIntoUDP6TestNr� r r r r r� t r� r� rl c @ r� )� RecvmsgRFC3542AncillaryUDP6TestNr� r r r r r� z � r� c @ r� )�#RecvmsgIntoRFC3542AncillaryUDP6TestNr� r r r r r� � s r� c @ r� )�SendrecvmsgUDPLITETestBaseNr� r r r r r� � s r� c @ r� )�SendmsgUDPLITETestNr� r r r r r� � r� r� c @ r� )�RecvmsgUDPLITETestNr� r r r r r� � r� r� c @ r� )�RecvmsgIntoUDPLITETestNr� r r r r r� � r� r� c @ rS )�SendrecvmsgUDPLITE6TestBasec C r� r� r� r� r r r r� � r� z/SendrecvmsgUDPLITE6TestBase.checkRecvmsgAddressNr� r r r r r� � s r� c @ r� )�SendmsgUDPLITE6TestNr� r r r r r� � r� r� c @ r� )�RecvmsgUDPLITE6TestNr� r r r r r� � r� r� c @ r� )�RecvmsgIntoUDPLITE6TestNr� r r r r r� � r� r� c @ r� )�#RecvmsgRFC3542AncillaryUDPLITE6TestNr� r r r r r� � s r� c @ r� )�'RecvmsgIntoRFC3542AncillaryUDPLITE6TestNr� r r r r r� � s r� c @ r� )�SendrecvmsgTCPTestBaseNr� r r r r r � r� r c @ r� )�SendmsgTCPTestNr� r r r r r � r� r c @ r� )�RecvmsgTCPTestNr� r r r r r � r� r c @ r� )�RecvmsgIntoTCPTestNr� r r r r r � r� r c @ r� )� SendrecvmsgSCTPStreamTestBaseNr� r r r r r � r� r z6IPPROTO_SCTP: [Errno 62] Protocol not supported on AIXrA r5 r c @ r� )�SendmsgSCTPStreamTestNr� r r r r r � r� r c r� )�RecvmsgSCTPStreamTestc
�T z
t t| ��� W d S ty) } z|jtjkr� | �d� W Y d }~d S d }~ww �Nz4sporadic ENOTCONN (kernel issue?) - see issue #13876)rW r r! r r� �ENOTCONNrq �rG r� r_ r r r! � ��z$RecvmsgSCTPStreamTest.testRecvmsgEOF�rN rO rP r! rg r r r_ r r � � r c r� )�RecvmsgIntoSCTPStreamTestc
r r )rW r r! r r� r rq r
r_ r r r! r z(RecvmsgIntoSCTPStreamTest.testRecvmsgEOFr r r r_ r r r
r c @ r� )� SendrecvmsgUnixStreamTestBaseNr� r r r r r r� r r c @ r� )�SendmsgUnixStreamTestNr� r r r r r r� r c @ r� )�RecvmsgUnixStreamTestNr� r r r r r # r� r c @ r� )�RecvmsgIntoUnixStreamTestNr� r r r r r ) r� r rg rR c @ r� )�RecvmsgSCMRightsStreamTestNr� r r r r r / r� r c @ r� )� RecvmsgIntoSCMRightsStreamTestNr� r r r r r 4 r� r c sL e Zd Z� fdd�ZejZeed�r dZ dd� Z
� Z S dZ dd� Z
� Z S ) �InterruptedTimeoutBasec s2 t � �� t�tjdd� �}| �tjtj|� d S )Nc S s dd S r5 r )�signumr� r r r r H s z.InterruptedTimeoutBase.setUp.<locals>.<lambda>)rW rH r< r= rc )rG Zorig_alrm_handlerr_ r r rH E s
�z InterruptedTimeoutBase.setUp� setitimerg�������?c C s t �t j|� d S r8 )r< r � ITIMER_REAL�rG Zsecondsr r r �setAlarmT r� z InterruptedTimeoutBase.setAlarmr� c C s t �|� d S r8 )r< r3 r r r r r Z rI ) rN rO rP rH r r� r; r r< �
alarm_timer rg r r r_ r r @ s
r � siginterruptr3 r z+Don't have signal.alarm or signal.setitimerc sl e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Ze e
j
d
�dd� �Z e e
j
d�dd� �Z � Z
S )�InterruptedRecvTimeoutTestc r� r8 )rW rH rB r r; rF r_ r r rH g r� z InterruptedRecvTimeoutTest.setUpc O sr z2| � | j� | �t��}||i |�� W d � n1 s w Y W | � d� d S W | � d� d S | � d� w r� )r r rY r> �rG r� r] r^ rc r r r �checkInterruptedRecvk s ��z/InterruptedRecvTimeoutTest.checkInterruptedRecvc C � | � | jjd� d S r� )r rB r� rF r r r �testInterruptedRecvTimeoutu r� z5InterruptedRecvTimeoutTest.testInterruptedRecvTimeoutc C � | � | jjtd�� d S r� )r rB � recv_intor� rF r r r � testInterruptedRecvIntoTimeoutx r� z9InterruptedRecvTimeoutTest.testInterruptedRecvIntoTimeoutc C r r� )r rB r� rF r r r � testInterruptedRecvfromTimeout{ r� z9InterruptedRecvTimeoutTest.testInterruptedRecvfromTimeoutc C r" r� )r rB �
recvfrom_intor� rF r r r �"testInterruptedRecvfromIntoTimeout~ r� z=InterruptedRecvTimeoutTest.testInterruptedRecvfromIntoTimeoutr% c C r r� )r rB r% rF r r r � testInterruptedRecvmsgTimeout� r� z8InterruptedRecvTimeoutTest.testInterruptedRecvmsgTimeoutr� c C s | � | jjtd�g� d S r� )r rB r� r� rF r r r �!testInterruptedRecvmsgIntoTimeout� r� z<InterruptedRecvTimeoutTest.testInterruptedRecvmsgIntoTimeout)rN rO rP rH r r! r$ r% r' r7 r r( r) rg r r r_ r r ` s
r c st e Zd Z� fdd�Zdd� Zdd� Ze�dd�d d
� �Ze�dd�d d � �Z e�dd�e
e j d
�dd� ��Z � Z
S )�InterruptedSendTimeoutTestc sp t � �� | �� | _| �| jj� tj| jd�}|� � | j
� � \| _ }| �| j j� |�
� | j�| j� d S )N)�target)rW rH r� r� rc r rY �Thread� doConnect�startrB r� r� r5 r r; )rG Z
cli_threadr� r_ r r rH � s
z InterruptedSendTimeoutTest.setUpc C s | j �| j� d S r8 )r� r� r� rF r r r r- � r� z$InterruptedSendTimeoutTest.doConnectc O sX z%| � t��} | �| j� ||i |�� q1 sw Y W | �d� d S | �d� w �NTr )rY r> r r r r r r �checkInterruptedSend� s ��z/InterruptedSendTimeoutTest.checkInterruptedSendr� r� c C r �Nr� )r0 r� r� rF r r r �testInterruptedSendTimeout� r� z5InterruptedSendTimeoutTest.testInterruptedSendTimeoutc C s | � | jjd| j� d S r1 )r0 r� ra r� rF r r r � testInterruptedSendtoTimeout� s �z7InterruptedSendTimeoutTest.testInterruptedSendtoTimeoutr( c C s | � | jjdg� d S r1 )r0 r� r( rF r r r � testInterruptedSendmsgTimeout� r� z8InterruptedSendTimeoutTest.testInterruptedSendmsgTimeout)rN rO rP rH r- r0 r Zrequires_mac_verr2 r3 r7 r r4 rg r r r_ r r* � s
r* c @ r? )�
TCPCloserTestc C sh | j �� \}}|�� | j}t�|gg g d�\}}}| �||g� | �|�d�d� |�� |�� d S )N� �?r~ rc )rB r� r r� r. r� r� )rG r� r� �sdre rH r= r r r � testClose� s zTCPCloserTest.testClosec C s | j �t| jf� t�d� d S �Nr6 )r� r� r� rD r� r� rF r r r �
_testClose� ra zTCPCloserTest._testCloseN)rN rO rP r8 r: r r r r r5 � �
r5 c @ sN e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dS )�BasicSocketPairTestr� c C r� r� )r� rX r� r r r rX � r� z BasicSocketPairTest.__init__c C s\ | � |tj� ttd�r| �|jtj� n| �|jtj� | �|jtj� | �|j d� d S )Nr r )
rC r r r� rO r rA r" r5 rP r r r r �_check_defaults� s
z#BasicSocketPairTest._check_defaultsc C � | � | j� d S r8 )r= r� rF r r r �
_testDefaults� r z!BasicSocketPairTest._testDefaultsc C r> r8 )r= rB rF r r r � testDefaults� r z BasicSocketPairTest.testDefaultsc C r� r� )rB r� r� r� r� r r r rS � r� z BasicSocketPairTest.testRecvc C rT r8 )r� r� r� rF r r r rV � r z BasicSocketPairTest._testRecvc C rT r8 )rB r� r� rF r r r �testSend� r z BasicSocketPairTest.testSendc C r� r� )r� r� r� r� r� r r r � _testSend� r� z BasicSocketPairTest._testSendNr� ) rN rO rP rX r= r? r@ rS rV rA rB r r r r r< � s
r< c @ s� e Zd Zd&dd�Zdd� Zdd� Zdd � Zejd
d � �Z e�e�Z
e � e
ed �d
�e�ddd�dd� ��Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%S )'�NonBlockingTCPTestsr� c C s t �� | _tj| |d� d S r� )rY r� �eventr� rX r� r r r rX � r� z NonBlockingTCPTests.__init__c C sl | � | j�� |� |dk}| � |�� |� td ur4|d u }t�|tjtj�}| � t|tj@ � |� d S d S )N� ) r� rB r� � getblockingr
�F_GETFLr� �
O_NONBLOCK�bool)rG r r; �blockingZ fd_blocking�flagr r r �assert_sock_timeout s �z'NonBlockingTCPTests.assert_sock_timeoutc C s� | j �d� | �| j d � | j �d� | �| j d� | j �d � | �| j d � | j �d� | �| j d� | j �d� | �| j d� | j �d� | �| j d� d S )NTFrE r r� )rB r� rL r rF r r r �testSetBlocking s z#NonBlockingTCPTests.testSetBlockingc C r� r8 r rF r r r �_testSetBlocking( r� z$NonBlockingTCPTests._testSetBlockingc C sb dd l }|j|jkr| �d� | j�d� | �| j�� d� | j�|jd � | �| j�� � d S )Nr zneeds UINT_MAX < ULONG_MAXFrE r~ ) r� rq Z ULONG_MAXrq rB r� r� r� r� rr r r r �testSetBlocking_overflow+ s
z,NonBlockingTCPTests.testSetBlocking_overflowr� z test needs socket.SOCK_NONBLOCKr� � � c C s4 | j �� t�tjtjtjB �| _ | �| j d� d S r� )rB r r rA r5 r� rL rF r r r �testInitNonBlocking: s
�z'NonBlockingTCPTests.testInitNonBlockingc C r� r8 r rF r r r �_testInitNonBlockingD r� z(NonBlockingTCPTests._testInitNonBlockingc C sb t d ��# | j�d� | j�� \}}| �|j� | �|�� � W d � d S 1 s*w Y d S )Nr� )r< rB r r� rc r r� r� r� r r r �testInheritFlagsBlockingG s
"�z,NonBlockingTCPTests.testInheritFlagsBlockingc C � | j �t| jf� d S r8 �r� r� r� rD rF r r r �_testInheritFlagsBlockingP r� z-NonBlockingTCPTests._testInheritFlagsBlockingc C sh d}t |��$ | j�d� | j�� \}}| �|j� | �|�� |� W d � d S 1 s-w Y d S )Ng 4@r� )r< rB r r� rc r r� r� )rG Zdefault_timeoutr� r� r r r �testInheritFlagsTimeoutS s
"�z+NonBlockingTCPTests.testInheritFlagsTimeoutc C rU r8 rV rF r r r �_testInheritFlagsTimeout^ r� z,NonBlockingTCPTests._testInheritFlagsTimeoutc C s� | j �d� t�� }| �t�� | j �� \}}W d � n1 s!w Y t�� | }| �|d� | j� � t
�
| j gg g t j �\}}}| j |vrO| �
d� | j �� \}}| �|j� | �|�� � d S )NFr6 z'Error trying to do accept after select.)rB r� r� � monotonicrY �BlockingIOErrorr� r{ rD r| r. r � LONG_TIMEOUTrU rc r r� r� )rG Z
start_timer� r� Zdtre rH r= r r r �
testAccepta s �
z NonBlockingTCPTests.testAcceptc C s | j �� | j�t| jf� d S r8 )rD r� r� r� r� rD rF r r r � _testAcceptw s
z NonBlockingTCPTests._testAcceptc C s� | j �� \}}| �|j� |�d� | �t�� |�tt ��}W d � n1 s)w Y | j
� � t � |gg g t
j�\}}}||vrI| �d� |�tt ��}| �|t � d S )NFz0Error during select call to non-blocking socket.)rB r� rc r r� rY r[ r� r% r� rD r| r. r r\ rU r� )rG r� r� r� re rH r= r r r rS ~ s
�
z NonBlockingTCPTests.testRecvc C s, | j �t| jf� | j�� | j �t� d S r8 )r� r� r� rD rD r� r? r� rF r r r rV � s
z NonBlockingTCPTests._testRecvNr� )rN rO rP rX rL rM rN r r� rO Z_testSetBlocking_overflowr� r� r r �requires_linux_versionrR rS rT rW rX rY r] r^ rS rV r r r r rC � s,
� rC c @ s� e Zd ZdZdZdZdZdZdZe Z
dZ e Z d;d d
�Z
d d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Z d)d*� Z d+d,� Z d-d.� Z d/d0� Z d1d2� Z!d3d4� Z"d5d6� Z#d7d8� Z$d9d:� Z%dS )<�FileObjectClassTestCaseao Unit tests for the object returned by socket.makefile()
self.read_file is the io object returned by makefile() on
the client connection. You can read from this file to
get output from the server.
self.write_file is the io object returned by makefile() on the
server connection. You can write to this file to send output
to the client.
r� r �strictNr rZ r� c C r� r� rR r� r r r rX � r� z FileObjectClassTestCase.__init__c C sR dd� t d�D �\| _| _| _| _t�| � | jj| j | j
| j | j | j
d�| _d S )Nc S � g | ]}t �� �qS r )rY r� �r, r> r r r r. � s �z1FileObjectClassTestCase.setUp.<locals>.<listcomp>r` �r^ �errors�newline)rd �evt1�evt2�
serv_finished� cli_finishedr� rH r� rM � read_moderr r^ re rf � read_filerF r r r rH � s �
�z FileObjectClassTestCase.setUpc C �6 | j �� | j�� | �| jj� d | _t�| � d S r8 )ri r| rl r rA �closedr� rK rF r r r rK � �
z FileObjectClassTestCase.tearDownc C s0 t �| � | jj| j| j| j| j| jd�| _ d S )Nrd )
r� r� r� rM �
write_moderr r^ re rf �
write_filerF r r r r� � s
�z#FileObjectClassTestCase.clientSetUpc C rm r8 )rj r| rq r rA rn r� r� rF r r r r� � ro z&FileObjectClassTestCase.clientTearDownc C sv | j �d� | j�d� | �t| jjd� | �t��}| j�d� W d � n1 s+w Y | �dt|j �� d S )Nr~ rX z!cannot read from timed out object)
r� r rl re rY rA r rI r8 rb )rG �ctxr r r �testReadAfterTimeout� s �z,FileObjectClassTestCase.testReadAfterTimeoutc C s. | j �| jdd� � | j �� | j�� d S )Nr rX )rq rH � write_msg�flushri r� rF r r r �_testReadAfterTimeout� s
z-FileObjectClassTestCase._testReadAfterTimeoutc C s< | j �t| j�d �}| j �d�}|| }| �|| j� d S r� )rl re r% �read_msgr� )rG � first_segZ
second_segr� r r r �
testSmallRead� s z%FileObjectClassTestCase.testSmallReadc C � | j �| j� | j �� d S r8 �rq rH rt ru rF r r r �_testSmallRead� r� z&FileObjectClassTestCase._testSmallReadc C � | j �� }| �|| j� d S r8 )rl re r� rw r� r r r � testFullRead� �
z$FileObjectClassTestCase.testFullReadc C rz r8 )rq rH rt r rF r r r �
_testFullRead� r� z%FileObjectClassTestCase._testFullReadc C s< t | j�� } | j�d�}|sq||7 }q| �|| j� d S r )r" rw rl re r� )rG r� �charr r r �testUnbufferedRead� s �z*FileObjectClassTestCase.testUnbufferedReadc C rz r8 r{ rF r r r �_testUnbufferedRead r� z+FileObjectClassTestCase._testUnbufferedReadc C r} r8 )rl �readliner� rw �rG �liner r r � testReadline r z$FileObjectClassTestCase.testReadlinec C rz r8 r{ rF r r r �
_testReadline r� z%FileObjectClassTestCase._testReadlinec C s& | j �� | j�� }| �|| j� d S r8 )r� r rl re r� rw r� r r r �testCloseAfterMakefile s
z.FileObjectClassTestCase.testCloseAfterMakefilec C rz r8 r{ rF r r r �_testCloseAfterMakefile r� z/FileObjectClassTestCase._testCloseAfterMakefilec C s@ | j �� | j�tt��}t| jt�r|� � }| �
|| j� d S r8 ) rl r r� r� r% r� r! rw r8 �decoder� r� r r r � testMakefileAfterMakefileClose s
z6FileObjectClassTestCase.testMakefileAfterMakefileClosec C rz r8 r{ rF r r r � _testMakefileAfterMakefileClose$ r� z7FileObjectClassTestCase._testMakefileAfterMakefileClosec C � | � | jj � d S r8 )rA rl rn rF r r r �testClosedAttr( r� z&FileObjectClassTestCase.testClosedAttrc C r� r8 )rA rq rn rF r r r �_testClosedAttr+ r� z'FileObjectClassTestCase._testClosedAttrc C �, | � | jj| j� | � | jj| j�� � d S r8 )r� rl r\ rk r- r� rJ rF r r r �testAttributes. � z&FileObjectClassTestCase.testAttributesc C r� r8 )r� rq r\ rp r- r� rJ rF r r r �_testAttributes2 r� z'FileObjectClassTestCase._testAttributesc C s8 | j �� | �t| j j� | j�� | �t| jj� d S r8 )rl r rY r� rJ r� r r� rF r r r �
testRealClose6 s
z%FileObjectClassTestCase.testRealClosec C r� r8 r rF r r r �_testRealClose< r� z&FileObjectClassTestCase._testRealCloser� )&rN rO rP rf rr r^ re rf rk r� rw rp rt rX rH rK r� r� rs rv ry r| r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r` � sF
r` c @ sd e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� ZdS )�!UnbufferedFileObjectClassTestCasea� Repeat the tests from FileObjectClassTestCase with bufsize==0.
In this case (and in this case only), it should be possible to
create a file object, read a line from it, create another file
object, read another line from it, without loss of data in the
first file object's buffer. Note that http.client relies on this
when reading multiple requests from the same socket.r c C sL | j �� }| �|d| j � | j�dd�| _ | j �� }| �|d| j � d S )N� A. r r � B. )rl r� r� rt r� rM r� r r r �testUnbufferedReadlineL s
z8UnbufferedFileObjectClassTestCase.testUnbufferedReadlinec C s2 | j �d| j � | j �d| j � | j �� d S )Nr� r� r{ rF r r r �_testUnbufferedReadlineT s z9UnbufferedFileObjectClassTestCase._testUnbufferedReadlinec C sD | j �� | j �d�}| �|| j� | j�� | �t| j jd� d S r� )r� r r� r� rw rl rY r r� r r r �testMakefileCloseY s
z3UnbufferedFileObjectClassTestCase.testMakefileClosec C rz r8 r{ rF r r r �_testMakefileCloseb r� z4UnbufferedFileObjectClassTestCase._testMakefileClosec C s6 t �| j�}| j�� t �| j�}| �|d |� d S r )r� r� r� rl r r� )rG Zrefcount_beforeZrefcount_afterr r r � testMakefileCloseSocketDestroyf s
z@UnbufferedFileObjectClassTestCase.testMakefileCloseSocketDestroyc C r� r8 r rF r r r � _testMakefileCloseSocketDestroyl r� zAUnbufferedFileObjectClassTestCase._testMakefileCloseSocketDestroyc C s | j �d� | �| j�td��d � | �| j�t| j�d �d � | j �
� | j � d� | j�t| j�d �}|d u rKt
�d� | j�t| j�d �}td�}| j�|�}| �|d� ||d |� }| �|| j� | �| j�td��d � | �| j�d�d � d S )NFr� rX r6 g �?r� r~ )r� r� r� rl �readintor� re r% rw rg r| rh r� r� r� )rG rx r� rB r� r r r �testSmallReadNonBlockings s
z:UnbufferedFileObjectClassTestCase.testSmallReadNonBlockingc C s> | j �d� | j�| j� | j�� | j�� | j�d� d S )Nr6 � @) rg r� rq rH rt ru rh r| ri rF r r r �_testSmallReadNonBlocking� s
z;UnbufferedFileObjectClassTestCase._testSmallReadNonBlockingc C s& | j �d� | jr| �| j� d S d S )Nr� )rj r� � serv_skippedrq rF r r r �testWriteNonBlocking� s �z6UnbufferedFileObjectClassTestCase.testWriteNonBlockingc C sv d | _ | j�d� dtj }d}| j�|�}| �|d� t|�D ]}| j�|�}|d u r/ d S | �|d� q d| _ d S )NFr; r� r z$failed to saturate the socket buffer) r� r� r� r r@ rq rH r� rd )rG ZBIGZLIMITrB r> r r r �_testWriteNonBlocking� s
z7UnbufferedFileObjectClassTestCase._testWriteNonBlockingN)rN rO rP rf rr r� r� r� r� r� r� r� r� r� r� r r r r r� @ s r� c @ � e Zd ZdZdS )�#LineBufferedFileObjectClassTestCaser~ N�rN rO rP rr r r r r r� � � r� c @ r� )�$SmallBufferedFileObjectClassTestCaser� Nr� r r r r r� � r� r� c @ s* e Zd ZdZdZe�d�ZdZeZ dZ
dS )�"UnicodeReadFileObjectClassTestCase�=Tests for socket.makefile() in text mode (rather than binary)r r rZ r] N� rN rO rP rf rk r� r� rw rp rt rf r r r r r� � s
r� c @ s* e Zd ZdZdZeZdZe�d�Z dZ
dS )�#UnicodeWriteFileObjectClassTestCaser� r rY r r] N) rN rO rP rf rk r� rw rp r� rt rf r r r r r� � s
r� c @ s0 e Zd ZdZdZe�d�ZdZe�d�Z dZ
dS )�'UnicodeReadWriteFileObjectClassTestCaser� r r rY r] Nr� r r r r r� � s
r� c @ r )�NetworkConnectionTestzProve network connection.c C s t �t| jf�| _| j| _d S r8 )r �create_connectionr� rD r� r� rF r r r r� � s z!NetworkConnectionTest.clientSetUpN)rN rO rP rf r� r r r r r� � r
r� c @ r� )�
BasicTCPTest2zLTests that NetworkConnection does not break existing TCP functionality.
N)rN rO rP rf r r r r r� � s r� c @ sD e Zd ZG dd� dej�Zejdd� �Zdd� Zdd� Z d d
� Z
d S ) �NetworkConnectionNoServerc @ rS )z$NetworkConnectionNoServer.MockSocketc G r� )Nz timed out)rA r� r r r r� � r� z,NetworkConnectionNoServer.MockSocket.connectN)rN rO rP r� r r r r �
MockSocket� s r� c c s, � t j }| jt _ z dV W |t _ dS |t _ w )z*Return a socket which times out on connectN)r r� )rG Z
old_socketr r r �mocked_socket_module� s �z.NetworkConnectionNoServer.mocked_socket_modulec C sr t �� }t�tjtj�}| �|j� | �t��}|� t
|f� W d � n1 s)w Y | � |j j
t
j� d S r8 )r r� r rA r5 rc r rY r r� r� r� rb r� Z ECONNREFUSED)rG rD r� rc r r r � test_connect� s �z&NetworkConnectionNoServer.test_connectc C s\ t �� }| �t��}t�t|f� W d � n1 sw Y t �� }| �|j j
|� d S r8 ) r r� rY r r r� r� Z get_socket_conn_refused_errsrI rb r� )rG rD rc Zexpected_errnosr r r �test_create_connection� s �z0NetworkConnectionNoServer.test_create_connectionc C s� | � � �L z t�tdf� W n$ ty Y n) ty2 } ztjs'|jtj kr(� W Y d }~nd }~ww | �
d� W d � d S W d � d S W d � d S 1 sSw Y d S )Nr zTimeoutError not raised) r� r r� r� rA r r r& r� Z EAFNOSUPPORTrU r� r r r � test_create_connection_timeout s"
��� ��"�z8NetworkConnectionNoServer.test_create_connection_timeoutN) rN rO rP r r� �
contextlib�contextmanagerr� r� r� r� r r r r r� � s
r� c @ sv e Zd Zddd�Zdd� Zdd� Zdd � ZeZd
d � ZeZ d d
� Z
eZ dd� Z eZ
dd� ZeZdd� ZeZdd� ZdS )� NetworkConnectionAttributesTestr� c C r� r� r� r� r r r rX " r� z(NetworkConnectionAttributesTest.__init__c C s t �� | _d S r8 )r r� � source_portrF r r r r� & rI z+NetworkConnectionAttributesTest.clientSetUpc C r� r8 r� rF r r r r� ) r� z.NetworkConnectionAttributesTest.clientTearDownc C s | j �� \}}|�� d S r8 )rB r� r r� r r r � _justAccept. s z+NetworkConnectionAttributesTest._justAcceptc C s: t jt| jftjd�| _| �| jj� | � | jj
d� d S )Nr� r� ) r r� r� rD r r� r� rc r r� rO rF r r r � _testFamily3 s
�z+NetworkConnectionAttributesTest._testFamilyc C sJ t jt| jftjd| jfd�| _| �| jj � | �
| j� � d | j� d S )Nr] )r; �source_addressr~ ) r r� r� rD r r� r� r� rc r r� r� rF r r r �_testSourceAddress: s � z2NetworkConnectionAttributesTest._testSourceAddressc C sn | � t�� d u � t�d� zt�t| jf�| _| �| jj � W t�d � nt�d � w | �
| j� � d� d S )Nry � rA r r9 r: r� r� rD r� rc r r� r� rF r r r �_testTimeoutDefaultD s
z3NetworkConnectionAttributesTest._testTimeoutDefaultc C sr | � t�� d u � t�d� ztjt| jfd d�| _| �| jj � W t�d � nt�d � w | �
| j� � d � d S �N� r� r� rF r r r �_testTimeoutNoneP s
z0NetworkConnectionAttributesTest._testTimeoutNonec C s, t jt| jfdd�| _| �| j�� d� d S r� )r r� r� rD r� r� r� rF r r r �_testTimeoutValueNamed\ s z6NetworkConnectionAttributesTest._testTimeoutValueNamedc C s8 t �t| jfd�| _| �| jj� | �| j�� d� d S )Nr� ) r r� r� rD r� rc r r� r� rF r r r �_testTimeoutValueNonameda s z8NetworkConnectionAttributesTest._testTimeoutValueNonamedNr� )rN rO rP rX r� r� r� Z
testFamilyr� ZtestSourceAddressr� ZtestTimeoutDefaultr� ZtestTimeoutNoner� ZtestTimeoutValueNamedr� ZtestTimeoutValueNonamedr� r r r r r� s"
r� c @ sB e Zd Zddd�Zdd� Zdd� Zdd � ZeZd
d � Zd d
� Z dS )� NetworkConnectionBehaviourTestr� c C r� r� r� r� r r r rX i r� z'NetworkConnectionBehaviourTest.__init__c C r� r8 r rF r r r r� m r� z*NetworkConnectionBehaviourTest.clientSetUpc C r� r8 r� rF r r r r� p r� z-NetworkConnectionBehaviourTest.clientTearDownc C s2 | j �� \}}| �|j� t�d� |�d� d S )NrX � done!)rB r� rc r r� r� r� r� r r r �testInsideTimeoutu �
z0NetworkConnectionBehaviourTest.testInsideTimeoutc C s0 t �t| jf� | _}|�d�}| �|d� d S )Nr r� )r r� r� rD r� r� r� )rG r r� r r r �_testInsideTimeout| s
z1NetworkConnectionBehaviourTest._testInsideTimeoutc s2 t jt| jfdd� | _� | �t� fdd�� d S )Nr~ r� c s
� � d�S )Nr )r� r �r r r r � r� zDNetworkConnectionBehaviourTest._testOutsideTimeout.<locals>.<lambda>)r r� r� rD r� rY rA rF r r� r �_testOutsideTimeout� s z2NetworkConnectionBehaviourTest._testOutsideTimeoutNr� )
rN rO rP rX r� r� r� ZtestOutsideTimeoutr� r� r r r r r� g s
r� c @ s6 e Zd Zdd� Zdd� Ze�eed�d�dd� �Z d S )
�TCPTimeoutTestc � � fdd�}� � t|d� d S )Nc s � j �d� � j �� d S r9 )rB r r� r rF r r �
raise_timeout� r� z4TCPTimeoutTest.testTCPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (TCP)�rY rA �rG r� r rF r �testTCPTimeout� � �z TCPTimeoutTest.testTCPTimeoutc C sv d}z
| j �d� | j �� }W n ty | �d� Y n ty' d}Y n | �d� Y |s9| �d� d S d S )NFrE z%caught timeout instead of error (TCP)Tz!caught unexpected exception (TCP)z3accept() returned success when we did not expect it)rB r r� rA rU r �rG �okr� r r r �testTimeoutZero� s �z TCPTimeoutTest.testTimeoutZeror3 ztest needs signal.alarm()c
s* | j �d� G dd� dt�� � fdd�}t�tj|�}zqzNzEz t�d� | j �� }W n, ty8 | �d� Y n$ � y@ Y n | �dt �
� d d� t � � f � Y n| �d � W t�d
� nt�d
� w W n � yx | �d � Y n w W t�tj|� d S W t�tj|� d S t�tj|� w ) Nr� c @ r� )z4TCPTimeoutTest.testInterruptedTimeout.<locals>.AlarmNr� r r r r �Alarm� s r� c s � �r8 r )r< r� �r� r r �
alarm_handler� r� z<TCPTimeoutTest.testInterruptedTimeout.<locals>.alarm_handlerr� z caught timeout instead of Alarmz3caught other exception instead of Alarm: %s(%s):
%sznothing caughtr zgot Alarm in wrong place)
rB r r� r< r= r3 r� rA rU r� �exc_info� traceback�
format_exc)rG r� rC r� r r� r �testInterruptedTimeout� s4
�
� ���"z%TCPTimeoutTest.testInterruptedTimeoutN)
rN rO rP r� r� r� r� r r< r� r r r r r� � s �r� c @ r? )�UDPTimeoutTestc r� )Nc � � j �d� � j �d� d S �Nr6 r� �rB r r� r rF r r r� � r� z4UDPTimeoutTest.testUDPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (UDP)r� r� r rF r �testUDPTimeout� r� z UDPTimeoutTest.testUDPTimeoutc C �x d}z| j �d� | j �d�}W n ty | �d� Y n ty( d}Y n | �d� Y |s:| �d� d S d S )NFrE r� z%caught timeout instead of error (UDP)Tz!caught unexpected exception (UDP)�1recv() returned success when we did not expect it�rB r r� rA rU r r� r r r r� � � �z UDPTimeoutTest.testTimeoutZeroN)rN rO rP r� r� r r r r r� � s r� c @ r? )�UDPLITETimeoutTestc r� )Nc r� r� r� r rF r r r� � r� z<UDPLITETimeoutTest.testUDPLITETimeout.<locals>.raise_timeoutz.Error generating a timeout exception (UDPLITE)r� r� r rF r �testUDPLITETimeout� r� z%UDPLITETimeoutTest.testUDPLITETimeoutc C r� )NFrE r� z)caught timeout instead of error (UDPLITE)Tz%caught unexpected exception (UDPLITE)r� r� r� r r r r� � r� z"UDPLITETimeoutTest.testTimeoutZeroN)rN rO rP r� r� r r r r r� � s r� c @ r? )�TestExceptionsc C sf | � ttt�� | � ttjt�� | � ttjt�� | � ttjt�� | �tj t� | �tjt
� d S r8 ) rA �
issubclassr r� r rZ r[ r; ZassertIs�errorrA rF r r r �testExceptionTree� s z TestExceptions.testExceptionTreec C sx t � t jt jd�}t � t jt jd|�� �}|�� | �|j� | �t�� |� d� W d � d S 1 s5w Y d S )Nr F)
r rA r5 rJ r rc r� rY r r� )rG Zsock0r r r r �test_setblocking_invalidfd� s � "�z)TestExceptions.test_setblocking_invalidfdN)rN rO rP r� r� r r r r r� � s r� r� zLinux specific testc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�TestLinuxAbstractNamespace�l c
C s� d}t � t jt j��_}|�|� |�� t � t jt j��5}|�|�� � |�� d �}| �|�� |� | �|� � |� W d � n1 sEw Y W d � n1 sTw Y W d � d S W d � d S 1 slw Y d S )Ns python-test-hello �r )
r r r5 ro rE r� r� r� r� � getpeername)rG r� �s1r� Zs3r r r �testLinuxAbstractNamespace s
�� ��"�z5TestLinuxAbstractNamespace.testLinuxAbstractNamespacec C sb dd| j d }t�tjtj��}|�|� | �|�� |� W d � d S 1 s*w Y d S )Nr� � hr~ )�
UNIX_PATH_MAXr r r5 ro r� r� �rG r� r! r r r � testMaxName s
"�z&TestLinuxAbstractNamespace.testMaxNamec C sT dd| j }t�tjtj��}| �t|j|� W d � d S 1 s#w Y d S )N� �h)r� r r r5 rY r ro r� r r r �testNameOverflow s "�z+TestLinuxAbstractNamespace.testNameOverflowc C sD t � t jt j�}z|�d� | �|�� d� W |�� d S |�� w )Nz
python test �
python test )r r r5 ro r� r� r rM r r r � testStrName! s
z&TestLinuxAbstractNamespace.testStrNamec C sT t � t jt j��}|�td�� | �|�� d� W d � d S 1 s#w Y d S )Nr� )r r r5 ro r� r� r� rM r r r �testBytearrayName* s "�z,TestLinuxAbstractNamespace.testBytearrayNameN) rN rO rP r� r� r� r� r� r� r r r r r� s r� ztest needs socket.AF_UNIXc @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestUnixDomainc C r� r8 )r r r5 r rF r r r rH 3 r� zTestUnixDomain.setUpc C r" r8 )r r rF r r r rK 6 rI zTestUnixDomain.tearDownc
C s8 zt �|�W S ty | �d�|t�� �� Y d S w )NzBPathname {0!a} cannot be represented in file system encoding {1!r})r� �fsencoder* rq r< r� �getfilesystemencoding�rG r r r r �encoded9 s �
��zTestUnixDomain.encodedc
C sZ z t �||� W d S ty, } zt|�dkr | �d�|�� n� W Y d }~d S d }~ww )NzAF_UNIX path too longz6Pathname {0!a} is too long to serve as an AF_UNIX path)r r r r8 rq r< )rG r r r� r r r ro D s ����zTestUnixDomain.bindc C s | � | j�� d� d S )N)r] N)rI r r� rF r r r r6 P r0 zTestUnixDomain.testUnboundc C s@ t j�tj�}| �| j|� | �tj|� | � | j�
� |� d S r8 ) r� r �abspathr �TESTFNro r rc r r� r� r r r r � testStrAddrT s zTestUnixDomain.testStrAddrc C sF t j�tj�}| �| j| �|�� | �tj |� | �
| j� � |� d S r8 ) r� r r r r ro r r rc r r� r� r r r r �
testBytesAddr[ s z TestUnixDomain.testBytesAddrc C sR t j�tj�}| �|�}| �| j|�dd�� | � tj
|� | � | j� � |� d S )N�ascii�surrogateescape)
r� r r r ZTESTFN_UNICODEr ro r r� rc r r� r� )rG r r- r r r �testSurrogateescapeBindb s
z&TestUnixDomain.testSurrogateescapeBindc C sT t jd u r
| �d� tj�t j�}| �| j|� | �t j |� | �
| j� � |� d S )Nz!No unencodable filename available) r ZTESTFN_UNENCODABLErq r� r r ro r rc r r� r� r r r r �testUnencodableAddrk s
z"TestUnixDomain.testUnencodableAddrN) rN rO rP rH rK r ro r6 r r r r r r r r r 0 s r c @ s� e Zd ZdZd dd�Zdd� Zdd� Zd d
� ZeZd d � Z eZ
d
d� Z dd� Z dd� Z
e Zdd� Ze Zdd� Zdd� Zdd� Ze ZdS ) � BufferIOTestzF
Test the buffer versions of socket.recv() and socket.send().
r� c C r� r� rR r� r r r rX z r� zBufferIOTest.__init__c C sZ t � ddgtt� �}| j�|�}| �|tt�� |�� }|d tt�� }| �|t� d S �Nr� r )r� r% r� r� r# r� r0 �rG r� r+ r� r r r �testRecvIntoArray} s z BufferIOTest.testRecvIntoArrayc C � t t�}| j�|� d S r8 �r r� r� r� �rG r� r r r �_testRecvIntoArray� � z BufferIOTest._testRecvIntoArrayc C sD t d�}| j�|�}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r# r� r% r� r r r r �testRecvIntoBytearray� s
z"BufferIOTest.testRecvIntoBytearrayc C sH t d�}| j�t|��}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r# r5 r� r% r� r r r r �testRecvIntoMemoryview� �
z#BufferIOTest.testRecvIntoMemoryviewc C s^ t � ddgtt� �}| j�|�\}}| �|tt�� |�� }|d tt�� }| �|t� d S r )r� r% r� r� r& r� r0 �rG r� r+ r� r� r r r �testRecvFromIntoArray� s z"BufferIOTest.testRecvFromIntoArrayc C r r8 r r r r r �_testRecvFromIntoArray� r z#BufferIOTest._testRecvFromIntoArrayc C sH t d�}| j�|�\}}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r& r� r% r� r r r r �testRecvFromIntoBytearray� r z&BufferIOTest.testRecvFromIntoBytearrayc C sL t d�}| j�t|��\}}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r& r5 r� r% r� r r r r �testRecvFromIntoMemoryview� s
z'BufferIOTest.testRecvFromIntoMemoryviewc C s t d�}| �t| jj|d� d S )Nrl r� )r� rY r� r� r& r r r r �testRecvFromIntoSmallBuffer� s z(BufferIOTest.testRecvFromIntoSmallBufferc C rT r8 rU rF r r r � _testRecvFromIntoSmallBuffer� r z)BufferIOTest._testRecvFromIntoSmallBufferc C s$ t � }| j�|� | j�|d� d S r� )r� r� r& r r r r �testRecvFromIntoEmptyBuffer� s z(BufferIOTest.testRecvFromIntoEmptyBufferNr� )rN rO rP rf rX r r r Z_testRecvIntoBytearrayr Z_testRecvIntoMemoryviewr r r Z_testRecvFromIntoBytearrayr Z_testRecvFromIntoMemoryviewr r r Z _testRecvFromIntoEmptyBufferr r r r r
v s&
r
i� �� �� c
C s� t td�sdS ztddd�} W n
tttfy Y dS w | � | D ]}|�d�r2 W d � dS q"W d � dS 1 s>w Y dS ) z�Check if the TIPC module is loaded
The TIPC module is not loaded automatically on Ubuntu and probably
other Linux distros.
�AF_TIPCFz
/proc/modulesr r] ztipc NT)r r r �FileNotFoundError�IsADirectoryErrorr� r� )r r� r r r �isTipcAvailable� s$
�
��
��r& z6TIPC module is not loaded, please 'sudo modprobe tipc'c @ rS )�TIPCTestc C s� t � t jt j�}t � t jt j�}| �|j� | �|j� |�t jt jd� t jt t
t f}|� |� t j
t t
tt t
d � df}|�t|� |�d�\}}| �|�� |� | �|t� d S )Nr~ r� r r� )r r# rf rc r r rg rh �TIPC_ADDR_NAMESEQ�
TIPC_STYPE�
TIPC_LOWER�
TIPC_UPPERro �TIPC_ADDR_NAMEr� ra r� r� r� r� )rG rl r� �srvaddrZsendaddrr� Zrecvaddrr r r �testRDM� s �
� zTIPCTest.testRDMN)rN rO rP r. r r r r r' � r� r' c @ r� )�TIPCThreadableTestr� c C r� r� r� r� r r r rX � r� zTIPCThreadableTest.__init__c C s� t � t jt j�| _| �| jj� | j�t jt jd� t j t
t t f}| j�
|� | j�� | �� | j�� \| _| _| �| jj� d S r )r r# r5 rl rc r r rg rh r( r) r* r+ ro rE r} r� r� r� )rG r- r r r rH s �
zTIPCThreadableTest.setUpc C sd t �d� t�tjtj�| _| �| jj� tjt t
t t t
d � df}| j�
|� | j�� | _d S )Nr� r� r )r� r� r r# r5 r� rc r r, r) r* r� r+ r� r� �cliaddr)rG r� r r r r� s
� z TIPCThreadableTest.clientSetUpc C s, | j �d�}| �|t� | �| j| j� d S r� )r� r� r� r� r0 r� r� r r r r� s z TIPCThreadableTest.testStreamc C r� r8 r� rF r r r r� r� z TIPCThreadableTest._testStreamNr� r� r r r r r/ � s
r/ c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ContextManagersTestc C s� t � � �}| �|j� W d � n1 sw Y | �|j� t � � � }|�� W d � n1 s3w Y | �|j� t � � �}| �t|jd� W d � n1 sUw Y | �|j� d S �Nr^ )r r rt rA r rY r r? r r r r �_testSocketClass$ s
�
�
�z$ContextManagersTest._testSocketClassc C �2 | j �� \}}| �|j� |�d�}|�|� d S r� �rB r� rc r r� r? �rG r� r� r� r r r �testCreateConnectionBase2 r� z,ContextManagersTest.testCreateConnectionBasec C sl | j �� }t�|�� }| �|j� |�d� | �|�d�d� W d � n1 s)w Y | � |j� d S )Nr^ r� )
rB r� r r� r rt r? r� r� rA �rG r� r r r r �_testCreateConnectionBase8 s
�z-ContextManagersTest._testCreateConnectionBasec C r4 r� r5 r6 r r r �testCreateConnectionClose@ r� z-ContextManagersTest.testCreateConnectionClosec C s\ | j �� }t�|�� }|�� W d � n1 sw Y | �|j� | �t|j d� d S r2 )
rB r� r r� r rA rt rY r r? r8 r r r �_testCreateConnectionCloseF s
� z.ContextManagersTest._testCreateConnectionCloseN)rN rO rP r3 r7 r9 r: r; r r r r r1 " s r1 c @ s� e Zd Ze�eed�d�e�ddd�dd� ��Z dd � Z
d
d � Z d d
� Z e�
edu d�dd� �Ze�
edu d�dd� �Zdd� ZdS )�InheritanceTestr zSOCK_CLOEXEC not definedr� rP rQ c C sZ t � t jt jt jB ��}| �|jt j� | �|�� � W d � d S 1 s&w Y d S r8 )r rA r5 r r� r" r �get_inheritablerM r r r �test_SOCK_CLOEXECO s
�"�z!InheritanceTest.test_SOCK_CLOEXECc C sB t � � }|� | �|�� d� W d � d S 1 sw Y d S rF )r r� r= r r r r �test_default_inheritableX s "�z(InheritanceTest.test_default_inheritablec C s� t � � }|�3 |�� }|�� |� | �|�� d� W d � n1 s$w Y W d � d S W d � d S 1 s<w Y d S rF )r rk r r� r= )rG r Znewsockr r r �test_dup] s ��"�zInheritanceTest.test_dupc C sf t � � }|�# |�d� | �|�� d� |�d� | �|�� d� W d � d S 1 s,w Y d S )NTF)r �set_inheritabler� r= r r r r �test_set_inheritablee s
"�z$InheritanceTest.test_set_inheritableNr� c C s� t � � }|�2 |�� }| �|�� d� t�|tj�}|tj M }t�|tj|� | �|�� d� W d � d S 1 s;w Y d S )NFT)r rJ r� r= r
�F_GETFD�
FD_CLOEXEC�F_SETFD)rG r r� r! r r r � test_get_inheritable_cloexecn s "�z,InheritanceTest.test_get_inheritable_cloexecc C s~ t � � }|�/ |�� }| �t�|tj�tj@ tj� |�d� | �t�|tj�tj@ d� W d � d S 1 s8w Y d S r/ )r rJ r� r
rC rD rA )rG r r� r r r � test_set_inheritable_cloexec| s �
�"�z,InheritanceTest.test_set_inheritable_cloexecc C sH t �� \}}| �|j� | �|j� | �|�� d� | �|�� d� d S rF )r r� rc r r� r= )rG r� r� r r r �test_socketpair� s
z InheritanceTest.test_socketpair)rN rO rP r� r� r r r r_ r> r? r@ rB r� r
rF rG rH r r r r r< N s �
r< r� zSOCK_NONBLOCK not definedc @ s, e Zd Zd dd�Ze�ddd�dd � �Zd
S ) �NonblockConstantTestTrE c C s� |r7| � |jtj� | � |�� |� | �t�|tjtj �tj @ � |dkr.| �
|� � � d S | �|� � � d S | � |jtj� | � |�� d � | �
t�|tjtj �tj @ � | �|� � � d S r� ) r� r" r r5 r� rA r
rG r� rH r rF )rG r! �nonblockr; r r r �
checkNonblock� s � �z"NonblockConstantTest.checkNonblockr� rP rQ c C s� t � t jt jt jB ��G}| �|� |�d� | j|dd� |�d� | �|� |�d � | j|dd� |�d� | j|dd� |�d� | j|dd� W d � n1 sUw Y t �� }t �d� t � � �
}| �|� W d � n1 sww Y t �d � t � � �}| �|d� W d � n1 s�w Y t �d� t � � �}| j|dd� W d � n1 s�w Y t �d � t � � �}| �|d� W d � n1 s�w Y t �|� d S )NTF)rJ g @r� rE ) r rA r5 r� rK r� r r9 r: )rG r! �tr r r �test_SOCK_NONBLOCK� sD
�
�
�
�
�
�z'NonblockConstantTest.test_SOCK_NONBLOCKN)TrE )rN rO rP rK r r_ rM r r r r rI � s
rI rn ro zneed multiprocessingc @ s@ e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
S )�TestSocketSharingc C sD |� � }|� � }t�|�}|�� \}}|�|� |�� |�� d S r8 )r� r � fromsharer� r? r )r� �qZsdata�messager! r� rB r r r �remoteProcessServer� s
z%TestSocketSharing.remoteProcessServerc C s� t �� }t j| j|fd�}|�� | j�|j�}| j�� }| j� � |�
|� d}|�
|� t � |�}g } |�
d�}|s>qD|�|� q6|� � d�|�}| �||� |�� d S )N�r+ r] s
slapmahfroTr rc )�multiprocessingr� �ProcessrR r. rB �share�pidr� r r� r r� r� rJ r5 r� ) rG rP rV r� r� rQ r! �m�receivedr r r � testShare� s*
�
zTestSocketSharing.testSharec C s@ | j �t�� �}| �ttj|d d� � | �ttj|d � d S )Nr� r^ )rB rV r� �getpidrY r� r rO )rG r� r r r �testShareLength s z!TestSocketSharing.testShareLengthc C sf | � |�� d � | � |�� |�� � | � |j|j� | � |j|j� |jdkr1| � |j|j� d S d S r� )r� r� rO r" rP )rG Zorg�otherr r r �compareSockets
s
�z TestSocketSharing.compareSocketsc C sB | j �t�� �}t�|�}z| �| j |� W |�� d S |�� w r8 )rB rV r� r[ r rO r^ r )rG r� r! r r r �testShareLocal s
z TestSocketSharing.testShareLocalc
C s� t jt jg}t jt jg}|D ]B}|D ]=}zt � ||�}W n ty% Y qw z$|�t�� �}t � |�}z | �
||� W |� � n|� � w W |� � q|� � w qd S r8 ) r rA r r5 r$ r rV r� r[ rO r^ r )rG �families�typesr rL �sourcer� Zsharedr r r � testTypes" s$ �
��zTestSocketSharing.testTypesN)
rN rO rP r rR rZ r\ r^ r_ rc r r r r rN � s
# rN c @ s e Zd ZdZdZdZdZejZ e
dd� �Z e
dd� �Z d d
� Z
d d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Z d)d*� Z d+d,� Z d-d.� Z d/d0� Z d1d2� Z!d3d4� Z"d5d6� Z#d7d8� Z$d9d:� Z%d;S )<�SendfileUsingSendTestz>
Test the send() implementation of socket.sendfile().
i � rw rc c C s� dd� }d� dd� t| j�D ��}ttjd��}|| j| j�D ]}|�|� q W d � n1 s1w Y ttjd��}|�� | _ t
| j �| jksLJ �W d � d S 1 sWw Y d S )Nc s s> � | |ksJ �| |kr|V | |8 } | |ks | r | V d S d S r8 r )�total�stepr r r �chunksB s � �
�z0SendfileUsingSendTest.setUpClass.<locals>.chunksrc c S s g | ]
}t �tj��� �qS r )�random�choice�string�
ascii_lettersrI rc r r r r. J s �z4SendfileUsingSendTest.setUpClass.<locals>.<listcomp>rZ r ) r5 rd �BUFSIZEr r r �FILESIZErH re �FILEDATAr% )r� rg �chunkr Zcsizer r r �
setUpClass@ s
� � �
"�z SendfileUsingSendTest.setUpClassc C s t �t j� d S r8 )r r r )r� r r r �
tearDownClassS r� z#SendfileUsingSendTest.tearDownClassc C s8 | j �tj� | j �� \}}|�| j� | �|j� |S r8 )rB r r r\ r� �TIMEOUTrc r r� r r r � accept_connW s
z!SendfileUsingSendTest.accept_connc C s. g } |� | j�}|s q|�|� qd�|�S )NTrc )r� rl rJ r5 )rG r� rY ro r r r � recv_data^ s
�
z SendfileUsingSendTest.recv_datac C �
t |d�S )N�_sendfile_use_send�r� r r r r �meth_from_sockg s
z$SendfileUsingSendTest.meth_from_sockc C s� | j �� }ttjd�}t�|��<}|�!}| �|�}||�}| �|| j � | �|�
� | j � W d � n1 s7w Y W d � d S W d � d S 1 sOw Y d S )Nr � rB r� r r r r r� rx r� rm �tell�rG r� �filer �meth�sentr r r �_testRegularFilen s
P�z&SendfileUsingSendTest._testRegularFilec C �6 | � � }| �|�}| �t|�| j� | �|| j� d S r8 �rs rt r� r% rm rn �rG r� r� r r r �testRegularFilew �
z%SendfileUsingSendTest.testRegularFilec C s� | j �� }t�| j�}t�|��A}|�&}|�|�}| �|| j � | �|�
� | j � | � tj |j
|� W d � n1 s<w Y W d � d S W d � d S 1 sTw Y d S r8 )rB r� �io�BytesIOrn r r� r� r� rm rz rY r� r� )rG r� r| r r~ r r r �_testNonRegularFile s
�P�z)SendfileUsingSendTest._testNonRegularFilec C r� r8 r� r� r r r �testNonRegularFile� r� z(SendfileUsingSendTest.testNonRegularFilec C s� | j �� }tjd }t|d�� | �tj|� W d � n1 s!w Y t|d�}t�|��:}|� }| � |�}||�}| �
|d� | �
|� � d� W d � n1 sUw Y W d � d S W d � d S 1 smw Y d S )N�2rZ r r ) rB r� r r r rc r r r� rx r� rz )rG r� �filenamer| r r} r~ r r r �_testEmptyFileSend� s
�
P�z(SendfileUsingSendTest._testEmptyFileSendc C s" | � � }| �|�}| �|d� d S ry )rs rt r� r� r r r �testEmptyFileSend� s
z'SendfileUsingSendTest.testEmptyFileSendc C s� | j �� }ttjd�}t�|��@}|�%}| �|�}||dd�}| �|| j d � | �|�
� | j � W d � n1 s;w Y W d � d S W d � d S 1 sSw Y d S )Nr � )�offsetry r{ r r r � _testOffset� s
P�z!SendfileUsingSendTest._testOffsetc C sB | � � }| �|�}| �t|�| jd � | �|| jdd � � d S )Nr� r� r� r r r �
testOffset� s
z SendfileUsingSendTest.testOffsetc C � | j �� }ttjd�}tj|tjd�}|�> |�# d}| � |�}|||d�}| �
||� | �
|� � |� W d � n1 s>w Y W d � d S W d � d S 1 sVw Y d S )Nr r� �GKL �r � rB r� r r r r r� r r� rx r� rz �rG r� r| r r r} r~ r r r �
_testCount� �
�
P�z SendfileUsingSendTest._testCountc C �@ d}| � � }| �|�}| �t|�|� | �|| jd |� � d S )Nr� �rs rt r� r% rn �rG r r� r� r r r � testCount� �
z SendfileUsingSendTest.testCountc C r� )Nr r� r~ r� r� r� r r r �_testCountSmall� r� z%SendfileUsingSendTest._testCountSmallc C r� r r� r� r r r �testCountSmall� r� z$SendfileUsingSendTest.testCountSmallc C s� | j �� }ttjd�}tj|dd��A}|�&}d}| �|�}||d|d�}| �||� | �|� � |d � W d � n1 s>w Y W d � d S W d � d S 1 sVw Y d S )Nr r� r� 駆 �� )r� r )
rB r� r r r r r� rx r� rz r� r r r �_testCountWithOffset� s
P�z*SendfileUsingSendTest._testCountWithOffsetc C sD d}| � � }| �|�}| �t|�|� | �|| jd|d � � d S )Nr� r� r� r� r r r �testCountWithOffset� s
z)SendfileUsingSendTest.testCountWithOffsetc C s� | j �� }ttjd�}t�|��<}|�!}|�d� | �|�}| � t
||� | � t
|j |� W d � n1 s7w Y W d � d S W d � d S 1 sOw Y d S )Nr F) rB r� r r r r r� r� rx rY r� r� �rG r� r| r r} r r r �_testNonBlocking� s
P�z&SendfileUsingSendTest._testNonBlockingc C s$ | � � }|�d�r| �d� d S d S )Nrw z$was not supposed to receive any data)rs r� rU �rG r� r r r �testNonBlocking� s
�z%SendfileUsingSendTest.testNonBlockingc C s� | j �� }ttjd�}tj|tjd�}|�3 |� | � |�}||�}| �
|| j � W d � n1 s3w Y W d � d S W d � d S 1 sKw Y d S )Nr r� ) rB r� r r r r r� r r� rx r� rm r{ r r r �_testWithTimeout s
�
P�z&SendfileUsingSendTest._testWithTimeoutc C r� r8 r� r� r r r �testWithTimeout r� z%SendfileUsingSendTest.testWithTimeoutc C s� | j �� }ttjd��7}t�|��}|�d� | �|�}| � t
||� W d � n1 s-w Y W d � d S W d � d S 1 sEw Y d S )Nr g{�G�z�?) rB r� r r r r r� r rx rY rA r� r r r � _testWithTimeoutTriggeredSend s
��"�z3SendfileUsingSendTest._testWithTimeoutTriggeredSendc C s | � � }|�d� t�d� d S )Ni�X r~ )rs r� r� r� r� r r r � testWithTimeoutTriggeredSend s
z2SendfileUsingSendTest.testWithTimeoutTriggeredSendc C r� r8 r rF r r r � _test_errors# r� z"SendfileUsingSendTest._test_errorsc C s� t tjd��,}tjtjd��}| �|�}| �td||� W d � n1 s&w Y W d � n1 s5w Y t tjdd��)}t�� �}| �|�}| �td||� W d � n1 s^w Y W d � n1 smw Y t tjd��R}t�� �5}| �|�}| jtd||dd � | jtd||d
d � | jtd||d d � | jtd||d d � W d � n1 s�w Y W d � d S W d � d S 1 s�w Y d S )
Nr r r5 r r] z binary modezpositive integerr� r� r� r r� ) r r r r r$ rx rb r� r� )rG r| r! r} r r r � test_errors& sJ
� �� �
� �� �
���� ��"�z!SendfileUsingSendTest.test_errorsN)&rN rO rP rf rm rl rn r r� rr r rp rq rs rt rx r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rd 6 sF
rd r� z%os.sendfile() required for this test.c @ r )�SendfileUsingSendfileTestzB
Test the sendfile() implementation of socket.sendfile().
c C ru )Nr� rw r r r r rx D r� z(SendfileUsingSendfileTest.meth_from_sockN)rN rO rP rf rx r r r r r� > s r� zAF_ALG requiredc @ s| e Zd Zdd� Ze�dd�dd� �Zdd� Ze�dd �d
d � �Ze�dd �d
d� �Z e�dd �dd� �Z
dd� Z dd� Z dS )�LinuxKernelCryptoAPIc
C sZ t � t jt jd�}z
|�||f� W |S ty, } z
|�� t�t|�||��d }~ww r� ) r r. r+ ro r$ r r� ZSkipTestr8 )rG �typr- r r� r r r �
create_algK s ���z LinuxKernelCryptoAPI.create_algr` r c C s t �d�}| �dd��r}|�� \}}|� |�d� | �|�d�|� W d � n1 s-w Y |�� \}}|�+ |�dtj � |�dtj � |�dtj � |�d � | �|�d�|� W d � n1 shw Y W d � d S W d � d S 1 s�w Y d S )
NZ@ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad�hashZsha256s abcr� � a� b� crc )
r �fromhexr� r� r? r� r� r� r �MSG_MORE�rG rQ �algo�opr� r r r � test_sha256X s$
�
��"�z LinuxKernelCryptoAPI.test_sha256c C s� t �d�}| �dd��@}|�tjtjd� |�� \}}|� |�d� | � |�
d�|� W d � n1 s6w Y W d � d S W d � d S 1 sNw Y d S )NZ(effcdf6ae5eb2fa2d27416d5f184df9c259a7c79r� z
hmac(sha1)s Jefes what do ya want for nothing?r� ) r r� r� r r �SOL_ALG� ALG_SET_KEYr� r? r� r� r� r r r �test_hmac_sha1j s
��"�z#LinuxKernelCryptoAPI.test_hmac_sha1rX c
C s" t �d�}t �d�}d}t �d�}t|�}| �dd���}|�tjtj|� |�� \}}|� |j tj
|tj d� |� |� | �
|�|�|� W d � n1 sPw Y |�� \}}|� |j |gtj|d� | �
|�|�|� W d � n1 s{w Y d } |g| }
|�� \}}|� |j |
tj
|d� |�|| �} W d � n1 s�w Y | �
t| �|| � | �
| d |� |� |�� \}}|� |j | gtj|d� |�|| �} W d � n1 s�w Y | �
t| �|| � | �
| || � W d � d S 1 �s
w Y d S )
NZ 06a9214036b8a15b512e03d534120006Z 3dafba429d9eb430b422da802c9fac41s Single block msgZ e353779c1079aeb82708942dbe77181aZskcipherzcbc(aes))r� �ivr! )r� r� r� )r r� r% r� r r r� r� r� �
sendmsg_afalg�ALG_OP_ENCRYPTr� r? r� r� �ALG_OP_DECRYPT)
rG �keyr� r� Z
ciphertextZmsglenr� r� r� Z
multiplierZlongmsg�encZdecr r r � test_aes_cbcu sV
�
� � �
� � � �$�z!LinuxKernelCryptoAPI.test_aes_cbcr3 c C s. t �d�}t �d�}t �d�}t �d�}t �d�}t �d�}t|�}t|�}| �dd���_} | �tjtj|� | �tjtjd |� | � � \}
} |
�@ |
j
tj ||tj d � |
�
|tj � |
�
|� |
�|t|� | �} | �|| || � � | �|| | d � � W d � n1 s�w Y | � � \}
} |
�8 || }
|
j
|
gtj ||d
� |
�|t|� | �} | �|| || � � | �|| | d � � W d � n1 s�w Y t�d �j}| � � \}
} |
�O || }
|
�|
gtjtj|tj �gtjtj|t|��| gtjtj||�gf� |
�t|
�| �} | �|| || � � | �|| | d � � W d � n 1 �s:w Y | � � \}
} |
�, || | }
|
j
|
gtj||d
� |
�t|
�| �} | �|| |d � � W d � n1 �sww Y W d � d S W d � d S 1 �s�w Y d S ) NZ c939cc13397c1d37de6ae0e1cb7c423cZb3d8cc017cbb89b39e0f67e2Z c3b3c41f113a31b73d9a5cd432103069Z 24825602bd12a984e0092d3e448eda5fZ 93fe7d9e9bfd10348a5606e5cafa7354Z 0032a1dc85f1c9786925a2e71d8272ddZaeadzgcm(aes))r� r� �assoclenr! )r� r� r� r )r r� r% r� r r r� r� �ALG_SET_AEAD_AUTHSIZEr� r� r� r� r? r� r� r �Structr� r( �
ALG_SET_OP�
ALG_SET_IV�ALG_SET_AEAD_ASSOCLENr� )rG r� r� ZplainZassocZ expected_ctZ expected_tagZtaglenr� r� r� r� �resr� Z pack_uint32r r r �test_aead_aes_gcm� sz
�
�
�
� � �� � � ��,$�z&LinuxKernelCryptoAPI.test_aead_aes_gcmc C s� | � dd��D}t�d�}|�tjtj|� |�� \}}|� |�d�}| � t
|�d� W d � n1 s5w Y W d � d S W d � d S 1 sMw Y d S )N�rngZdrbg_pr_sha256r� ) r� r� �urandomr r r� r� r� r� r� r% )rG r� Z
extra_seedr� r� Zrnr r r �test_drbg_pr_sha256� s
��"�z(LinuxKernelCryptoAPI.test_drbg_pr_sha256c C s\ t � t jt jd�}|�� | �t�� |�� W d � n1 s w Y | �t�� |jd d� W d � n1 s;w Y | �t��
|�d� W d � n1 sUw Y | �t�� |jt jd d� W d � n1 srw Y | �t�� |jt jdd� W d � n1 s�w Y W d � d S W d � d S 1 s�w Y d S )Nr )r� r~ )r� r� r� )r r. r+ rY r� r� r� r r r r �test_sendmsg_afalg_args� s(
� � � � ��
"�z,LinuxKernelCryptoAPI.test_sendmsg_afalg_argsc C s� t � t jt jd�}| �|j� | �t��
|�d� W d � n1 s$w Y | �t d��
|�d� W d � n1 s?w Y | �t��
|�d� W d � n1 sYw Y | �t d�� |�d� W d � d S 1 suw Y d S )Nr )Z
tttttttttttttr- z
type too long)Zttttttttttttttr- )r" Z?nnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnz
name too long)r" Z@nnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnn)
r r. r+ rc r rY r$ ro rb r� r r r r �test_length_restriction� s � � � "�z,LinuxKernelCryptoAPI.test_length_restrictionN)
rN rO rP r� r r_ r� r� r� r� r� r� r� r r r r r� H s
(
>
r� rr rs c @ rS )�TestMacOSTCPFlagsc C s | � tj� d S r8 )rA r Z
TCP_KEEPALIVErF r r r �test_tcp_keepalive r z$TestMacOSTCPFlags.test_tcp_keepaliveN)rN rO rP r� r r r r r� rU r� �winzrequires Windowsc @ s e Zd Zh d�Zdd� ZdS )�TestMSWindowsTCPFlags> � TCP_FASTOPEN�
TCP_KEEPINTVL� TCP_KEEPIDLE�
TCP_MAXSEG� TCP_KEEPCNT� TCP_NODELAYc s6 dd� t t�D �}� fdd�|D �}� �g |d� d S )Nc S s g | ] }|� d �r|�qS )ZTCP)r� �r, r! r r r r. " r/ z<TestMSWindowsTCPFlags.test_new_tcp_flags.<locals>.<listcomp>c s g | ] }|� j vr|�qS r )�
knownTCPFlagsr� rF r r r. # r/ zANew TCP flags were discovered. See bpo-32394 for more information)r� r r� )rG Zprovided�unknownr rF r �test_new_tcp_flags! s
�z(TestMSWindowsTCPFlags.test_new_tcp_flagsN)rN rO rP r� r� r r r r r� r; r� c @ s� e Zd Zdd� Zdd� Zdd� Ze�ee d� p ee d� d �e�
e j d
�d d � ��Z
e�e�� d
�e�
e j d
�dd� ��ZdS )�CreateServerTestc C s� t �� }t�d|f�� }| �|�� d d� | �|�� d |� W d � n1 s*w Y t jrctjd|ftjd�� }| �|�� d d� | �|�� d |� W d � d S 1 s\w Y d S d S )NrG r r~ r� r )r r� r �
create_serverr� r� r& r �rG rD r r r r � test_address+ s �
�"��z CreateServerTest.test_addressc C s� t �d��}| �|jt j� | �|jt j� W d � n1 s w Y tjrSt jdt j d��}| �|jt j � | �|jt j� W d � d S 1 sLw Y d S d S )NrF �r� r r )
r r� r� rO rA r" r5 r r& r )rG r r! r r r �test_family_and_type6 s �"��z%CreateServerTest.test_family_and_typec C s� t td�s$| �t�� tjddd� W d � d S 1 s w Y d S t�d��}|�tjtj�}| �|d� W d � n1 sBw Y tjddd��}|�tjtj�}| � |d� W d � d S 1 shw Y d S )N� SO_REUSEPORT)r r T)�
reuse_portr )
r r rY r� r� r rg r� r� r8 )rG r r� r r r �test_reuse_port? s
"� �"�z CreateServerTest.test_reuse_portrl � IPV6_V6ONLYz IPV6_V6ONLY option not supportedrt c C sJ t jdt jd��}|�t jt j�sJ �W d � d S 1 s w Y d S )Nr� r )r r� r r rl r� r r r r �test_ipv6_only_defaultK s "�z'CreateServerTest.test_ipv6_only_default� dualstack_ipv6 not supportedc C sH t jdt jdd��}| �|jt j� W d � d S 1 s w Y d S )Nr� T�rO �dualstack_ipv6)r r� r r� rO r r r r �test_dualstack_ipv6_familyS s
�"�z+CreateServerTest.test_dualstack_ipv6_familyN)rN rO rP r� r� r� r� r� r r@ r� r r& r� r �has_dualstack_ipv6r� r r r r r� ) s
� � r� c @ s� e Zd ZejZdd� Zdd� Zdd� Ze �
e j d�dd � �Z
e �e�� d
�e �
e j d�d d � ��Ze �e�� d
�e �
e j d�d
d� ��ZdS )�CreateServerFunctionalTestc sV � �fdd�}t �� � |��j� t j||fd�}|�� ��|j�j� � �� d S )Nc s� | �J | � � \}}|�) � ��j� |�d�}|s( W d � W d � d S |�|� W d � n1 s7w Y W d � d S W d � d S 1 sOw Y d S r� )r� r� r; r� r? )r r� r� r� �rD rG r r �run` s
�� ��"�z3CreateServerFunctionalTest.echo_server.<locals>.runrS ) rY r� r r; r, r. rc r5 r| )rG r r� r� r r� r � echo_server_ s
z&CreateServerFunctionalTest.echo_serverc C sd t j |d��"}|�| j� |�|� |�d� | �|�d�d� W d � d S 1 s+w Y d S )Nr r^ r� )r r r; r� r? r� r� )rG r� rO r r r r � echo_clientq s
"�z&CreateServerFunctionalTest.echo_clientc C sX t �� }t�d|f��}| �|� | �d|ftj� W d � d S 1 s%w Y d S )Nr] rG )r r� r r� r� r� rA r� r r r � test_tcp4x s
"�z$CreateServerFunctionalTest.test_tcp4rt c C s^ t �� }tjd|ftjd��}| �|� | �d|ftj� W d � d S 1 s(w Y d S )Nr] r r� �r r� r r� r r� r� r� r r r � test_tcp6~ s
�
"�z$CreateServerFunctionalTest.test_tcp6r� c C s` t �� }tjd|ftjdd��}| �|� | �d|ftj� W d � d S 1 s)w Y d S )Nr] Tr� rG )r r� r r� r r� r� rA r� r r r �test_dual_stack_client_v4� � �
"�z4CreateServerFunctionalTest.test_dual_stack_client_v4c C s` t �� }tjd|ftjdd��}| �|� | �d|ftj� W d � d S 1 s)w Y d S )Nr] Tr� r� r� r� r r r �test_dual_stack_client_v6� r� z4CreateServerFunctionalTest.test_dual_stack_client_v6N)rN rO rP r r� r; r� r� r� r� r� r r& r� r� r r� r� r� r r r r r� \ s"
� � r� �send_fds�recv_fdsc @ rS )�SendRecvFdsTestsc C sj dd� }dd� }dd� t d�D �}| �||� dd� |D �}t�tjtj�\}}|�; |�) t�|tg|� t�|t t�d t |�d �\}}} }
| �||� W d � n1 sYw Y W d � n1 shw Y | �
|t� | �
t |�t |�� | �
| d
� t |�D ]\} }|\} }
t �
|
t| ��� � q�t |�D ]\} } t �| d �}| �
|t| ��� � q�d S ) Nc S s& | D ]\}}t �|� t �|� qd S r8 �r� r )�pipesrr �fd2r r r � close_pipes� s
�z8SendRecvFdsTests.testSendAndRecvFds.<locals>.close_pipesc S s | D ]}t �|� qd S r8 r� )rK r� r r r � close_fds� s �z6SendRecvFdsTests.testSendAndRecvFds.<locals>.close_fdsc S rb r )r� �pipe)r, r� r r r r. � � z7SendRecvFdsTests.testSendAndRecvFds.<locals>.<listcomp>r� c S s g | ]\}}|�qS r r )r, �rfd�wfdr r r r. � r r� r r )rd rc r r� r r5 r� r� r� r% r� rM r� rH r8 rI re )rG r r r� rK Zsock1Zsock2r� Zfds2r! r� r� r r r� r r r �testSendAndRecvFds� s. & �� �z#SendRecvFdsTests.testSendAndRecvFdsN)rN rO rP r r r r r r� � r� r� c C s t �� } tjt jg| �R � d S r8 )r Zthreading_setupr� ZaddModuleCleanupZthreading_cleanup)� thread_infor r r � setUpModule� s r �__main__)�r� �testr Z test.supportr r r r� r� r r r. r� r� r� r r� r� r� r� r� �weakrefr r< r5 re r rh r� rj �_threadr� rY rT r� r
r� rI r� r� �systemr r@ r r" r& r( r, r/ r1 r3 r7 r� r< ZHAVE_SOCKET_CANZHAVE_SOCKET_CAN_ISOTPZHAVE_SOCKET_CAN_J1939ZHAVE_SOCKET_RDSZHAVE_SOCKET_ALGZHAVE_SOCKET_QIPCRTRZHAVE_SOCKET_VSOCKr ZHAVE_SOCKET_UDPLITEZHAVE_SOCKET_BLUETOOTHr? rZ r� r@ rQ rT rV rh rv rx r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r* r7 r> r? r� r� r r r r r1 r> rG rQ rx r� r� r� r� r� r� r� r� r� r� r� r� r r& r) r: rF r� r� r� r� r� r� r& r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r r r r r r r r* r5 r< rC r` r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r
r) r* r+ r& r' r/ r1 r< rI r- rN rd r� r� r� r� r� r� r� r� r rN �mainr r r r �<module> sf � �
�
3q�� � ! � �$
c
S
Z
*
=
>��.�%{ �7w w= # P; k 7�
�
�
�� ��
�
�
� �
�
�
�
� �
� � �
�
� �
�
� �
�
� �
� )��7# ! $l
?G 9�)ES �� ',C�< e
� D3
@
' �