Your IP : 216.73.216.196
o
0�i�� � @ 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'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� Z0dd� Z1dd� Z2dd � Z3d d � Z4ej5d d � �Z6e-� Z7e.� Z8e/� Z9e0� Z:e1� Z;e2� Z<e3� Z=e>e
d!�Z?e4� Z@e�d"�jAZBG d#d$� d$e jC�ZDG d%d&� d&e jC�ZEG d'd(� d(eE�ZFG d)d*� d*�ZGG d+d,� d,e jC�ZHG d-d.� d.e jC�ZIG d/d0� d0�ZJG d1d2� d2eDeJ�ZKG d3d4� d4eEeJ�ZLe �Me?d5�G d6d7� d7eFeJ��ZNG d8d9� d9eHeJ�ZOG d:d;� d;eIeJ�ZPe �Qe#du d<�e �Me=d=�e �Me,� d>kd?�G d@dA� dAe jCeJ����ZRG dBdC� dCeK�ZSG dDdE� dEe jCeJ�ZTG dFdG� dGe jC�ZUG dHdI� dIeU�ZVG dJdK� dKeGeUeJ�ZWG dLdM� dMeVeW�ZXG dNdO� dOeU�ZYG dPdQ� dQeY�ZZG dRdS� dSeU�Z[G dTdU� dUe[�Z\G dVdW� dWe[�Z]G dXdY� dYe[�Z^G dZd[� d[e[�Z_G d\d]� d]e[�Z`G d^d_� d_e`�ZaG d`da� dae`�Zbdbdc� Zcddde� Zddfdg� ZeG dhdi� die jC�Zfe �Me7dj�G dkdl� dle jC��Zge �Me7dj�G dmdn� dneO��Zhe �Me8do�G dpdq� dqe jC��Zie �Me9dr�G dsdt� dte jC��Zje �Me:du�G dvdw� dwe jC��Zke �Me:du�G dxdy� dyeP��Zle �Me<dz�G d{d|� d|e jC��Zme �Qe#du d<�e �Me=d=�G d}d~� d~e jC���Zne �Me@d�G d�d�� d�e jC��ZoG d�d�� d�eS�ZpG d�d�� d�eL�Zqe �Me?d5�G d�d�� d�eN��ZrG d�d�� d�eG�ZsG d�d�� d�es�ZtG d�d�� d�es�ZuG d�d�� d�es�ZvG d�d�� d�es�ZwG d�d�� d�es�ZxG d�d�� d�es�ZyG d�d�� d�ey�ZzG d�d�� d�ez�Z{G d�d�� d�ez�Z|G d�d�� d�es�Z}G d�d�� d�e}�Z~G d�d�� d�e}�ZG d�d�� d�ete}�Z�G d�d�� d�e jC�Z�G d�d�� d�ey�Z�G d�d�� d�ey�Z�G d�d�� d�eueweWe]�Z�ede
j
d��G d�d�� d�e|e���Z�ede
j
d��G d�d�� d�ee���Z�ede
j
d��G d�d�� d�e�e���Z�G d�d�� d�eueweWea�Z�ede
j
d��e �Mej�d��eed�d��G d�d�� d�e|e�����Z�ede
j
d��e �Mej�d��eed�d��G d�d�� d�ee�����Z�ede
j
d��e �Mej�d��eed�d��G d�d�� d�e�e�����Z�ede
j
d��e �Mej�d��ede
d��eed�d��G d�d d�e�e������Z�ede
j
d��e �Mej�d��ede
d��eed�d��G d�dĄ d�ete�e������Z�e �Me?d5�G d�dƄ d�eueweWe^��Z�e �Me?d5�ede
j
d��G d�dȄ d�e|e����Z�e �Me?d5�ede
j
d��G d�dʄ d�ee����Z�e �Me?d5�ede
j
d��G d�d̄ d�e�e����Z�e �Me?d5�G d�d΄ d�eueweWeb��Z�ede
j
d��e �Mej�d��e �Me?d5�eed�d��G d�dЄ d�e|e������Z�ede
j
d��e �Mej�d��e �Me?d5�eed�d��G d�d҄ d�ee������Z�ede
j
d��e �Mej�d��e �Me?d5�eed�d��G d�dԄ d�e�e������Z�ede
j
d��e �Mej�d��e �Me?d5�ede
d��eed�d��G d�dք d�e�e�������Z�ede
j
d��e �Mej�d��e �Me?d5�ede
d��eed�d��G d�d d�ete�e�������Z�G d�dڄ d�exeXe\�Z�ede
j
d��G d�d܄ d�e{e���Z�ede
j
d��G d�dބ d�ee~e���Z�ede
j
d��G d�d� d�e�e~e���Z�G d�d� d�evexeXe_�Z�ede
j
d��e �Qe*d�eed�d�d�G d�d� d�e{e�����Z�ede
j
d��e �Qe*d�eed�d�d�G d�d� d�ee~e�����Z�ede
j
d��e �Qe*d�eed�d�d�G d�d� d�e�e~e�����Z�G d�d� d�exeXeZ�Z�ede
j
d��ede
d�G d�d� d�e{e����Z�ede
j
d��ede
d�G d�d� d�ee~e����Z�ede
j
d��ede
d�G d�d�� d�e�e~e����Z�ede
j
d�d��ede
d�d�d��G d�d�� d�e�e����Z�ede
j
d�d��ede
d�d�d��G d�d�� d�ete�e����Z�G d�d�� d��Z�eded��e �Me>ed���p~e>e�d ��d�G �d�d� �de�e]���Z�eded��e �Me>ed���p�e>e�d ��d�G �d�d� �de�eGeVe\���Z�G �d�d� �deK�Z�G �d�d � �d eT�Z�G �d
�d � �d eT�Z�G �d �d
� �d
eK�Z�G �d�d� �deS�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��Z�G �d �d � �d e�ep�Z�G �d �d!� �d!e jC�Z�G �d"�d#� �d#eDeJ�Z�G �d$�d%� �d%eDeJ�Z�G �d&�d'� �d'eD�Z�G �d(�d)� �d)eE�Z�e �Me?d5�G �d*�d+� �d+eF��Z�G �d,�d-� �d-e jC�Z�e �Mej�d.k�d/�G �d0�d1� �d1e jC��Z�e �Me>e
d��d2�G �d3�d4� �d4e jC��Z�G �d5�d6� �d6eS�Z��d7Zd8ZÐd9ZĐd:�d;� Z�e �MeŃ �d<�G �d=�d>� �d>e jC��Z�e �MeŃ �d<�G �d?�d@� �d@e jCeJ��Z�G �dA�dB� �dBeK�Z�G �dC�dD� �dDe jC�Z�e �Me>e
�dE��dF�G �dG�dH� �dHe jC��Z�e �MejːdIk�dJ�e �Me!�dK�G �dL�dM� �dMeD���Z�G �dN�dO� �dOeK�Z�e �Me>e�dP��dQ�G �dR�dS� �dSe̓�Z�e �Me;�dT�G �dU�dV� �dVe jC��Z�e �Mej�dWk�dX�G �dY�dZ� �dZe jC��Z�e �Mej�ѐd[��d\�G �d]�d^� �d^e jC��Z�G �d_�d`� �d`e jC�Z�G �da�db� �dbe jC�Z�ede
�dc�ede
�dd�ede
d�G �de�df� �dfe jC����ZՐdg�dh� Z�eאdik�r�e �ء dS dS (j � 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 �7/opt/alt/python310/lib64/python3.10/test/test_socket.py�get_cid5 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_canB � �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_isotpL 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_j1939V 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_algj r. r0 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_qipcrtrt r. r2 c C s t � } | duS )z:Check whether AF_VSOCK sockets are supported on this host.N)r )�cidr r r �_have_socket_vsock~ s r4 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 �r8 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 r9 )r �AF_INETr6 �servr � bind_port�port�listen��selfr r r �setUp� s zSocketTCPTest.setUpc C � | j �� d | _ d S r9 �rC r rG r r r �tearDown� �
zSocketTCPTest.tearDownN��__name__�
__module__� __qualname__rI rL r r r r rA � s rA c @ r@ )�
SocketUDPTestc C s$ t � t jt j�| _t�| j�| _d S r9 )r rB r% rC r rD rE rG r r r rI � � zSocketUDPTest.setUpc C rJ r9 rK rG r r r rL � rM zSocketUDPTest.tearDownNrN r r r r rR � s rR c @ � e Zd Zdd� ZdS )�SocketUDPLITETestc C s( t � t jt jt j�| _t�| j�| _d S r9 )r rB r% r> rC r rD rE rG r r r rI � s zSocketUDPLITETest.setUpN)rO rP rQ rI r r r r rU � � rU 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 r9 )�super�__init__� threading�RLock�
_cleanup_lock�rH �args�kwargs�� __class__r r rY � � z"ThreadSafeCleanupTestCase.__init__c �< | j � t� j|i |��W d � S 1 sw Y d S r9 )r\ rX �
addCleanupr] r` r r rd � � $�z$ThreadSafeCleanupTestCase.addCleanupc rc r9 )r\ rX �
doCleanupsr] r` r r rf � re z$ThreadSafeCleanupTestCase.doCleanups)rO rP rQ �__doc__rY rd rf �
__classcell__r r r` r rW � s
rW 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" rd r �bind� interfacer �skipTestrG r r r rI s ��zSocketCANTest.setUpN) rO rP rQ rg rq �bufsize�
can_frame_fmtr ZcalcsizeZcan_frame_size�bcm_cmd_msg_fmtrI r r r r ri � s
ri 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, rC rd r r rD rE r rr rG r r r rI s �zSocketRDSTest.setUpN)rO rP rQ rg rs rI r r r r rv s rv c @ �@ 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 r9 )rI �_ThreadableTest__setUp�_setUprG r r r rY G 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�setrG r r r �serverExplicitReadyL 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 )Nr Fc s � j �� r
� j �� �d S r9 )�queueZqsize�getr rG r r �raise_queued_exception^ s
�z5ThreadableTest._setUp.<locals>.raise_queued_exception�.�_T)r Zwait_threads_exitZ wait_threads� __enter__rd �__exit__rZ �Eventr| � client_ready�doner �Queue�server_crashed�id�rfind�getattr�thread�start_new_thread� clientRunZ
client_threadrz r} �wait)rH r� Z
methodnamer? Z test_methodr rG r r{ S 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)rH Z test_func�er r r r� u s6
���
��� �zThreadableTest.clientRunc C � t d��)Nz clientSetUp must be implemented.)�NotImplementedErrorrG r r r r� � � zThreadableTest.clientSetUpc C s | j �� t�� d S r9 )r� r} r� �exitrG r r r r� � �
z ThreadableTest.clientTearDownN)
rO rP rQ rg rY r~ r{ r� r� r� r r r r ry ! s %" ry c @ �& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedTCPSocketTest�runTestc C � t j| |d� t�| � d S �N)�
methodName�rA rY ry �rH r� r r r rY � � z ThreadedTCPSocketTest.__init__c C � t � t jt j�| _d S r9 )r rB r6 �clirG r r r r� � � z!ThreadedTCPSocketTest.clientSetUpc C � | j �� d | _ t�| � d S r9 �r� r ry r� rG r r r r� � �
z$ThreadedTCPSocketTest.clientTearDownN�r� �rO rP rQ rY r� r� r r r r r� � �
r� c @ r� )
�ThreadedUDPSocketTestr� c C r� r� )rR rY ry r� r r r rY � r� z ThreadedUDPSocketTest.__init__c C r� r9 )r rB r% r� rG r r r r� � r� z!ThreadedUDPSocketTest.clientSetUpc C r� r9 r� rG 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� )rU rY ry r� r r r rY � r� z"ThreadedUDPLITESocketTest.__init__c C s t � t jt jt j�| _d S r9 )r rB r% r> r� rG r r r r� � � z%ThreadedUDPLITESocketTest.clientSetUpc C r� r9 r� rG r r r r� � r� z(ThreadedUDPLITESocketTest.clientTearDownNr� r� r r r r r� � s
r� c @ r� )
�ThreadedCANSocketTestr� c C r� r� )ri rY ry r� r r r rY � r� z ThreadedCANSocketTest.__init__c C sB t � t jt jt j�| _z | j�| jf� W d S ty Y d S w r9 )r r r r r� rp rq r rG r r r r� � s �z!ThreadedCANSocketTest.clientSetUpc C r� r9 r� rG r r r r� � r� z$ThreadedCANSocketTest.clientTearDownNr� r� r r r r r� � s
r� c @ r� )
�ThreadedRDSSocketTestr� c C r� r� )rv rY ry r� r r r rY � 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� rp �HOST� getsockname�cli_addrr rG r r r r� � s �z!ThreadedRDSSocketTest.clientSetUpc C r� r9 r� rG 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�TestCaserY ry r� r r r rY � � z&ThreadedVSOCKSocketStreamTest.__init__c C sh t � t jt j�| _| �| jj� | j�t jtf� | j� � | �
� | j� � \| _ | _
| �| j j� d S r9 )r �AF_VSOCKr6 rC rd r rp �VMADDR_CID_ANY� VSOCKPORTrF r~ �accept�conn�connaddrrG r r r rI � s
z#ThreadedVSOCKSocketStreamTest.setUpc C sX t �d� t�tjtj�| _| �| jj� t� }|tj tj
fv r"t }| j� |t
f� d S )N皙�����?)�time�sleepr r� r6 r� rd r r �VMADDR_CID_HOSTr� �VMADDR_CID_LOCAL�connectr� )rH r3 r r r r� � s
z)ThreadedVSOCKSocketStreamTest.clientSetUpc
C sT z| j �d�}W n ty! } z
| �t|�� W Y d }~nd }~ww | �|t� d S �N� )r� �recv�PermissionErrorrr �repr� assertEqual�MSG)rH �msg�excr r r �
testStream s ��z(ThreadedVSOCKSocketStreamTest.testStreamc C � | j �t� | j �� d S r9 �r� �sendr� r rG r r r � _testStream
� z)ThreadedVSOCKSocketStreamTest._testStreamNr� �rO rP rQ rY rI 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� rY r� r r r rY � z SocketConnectedTest.__init__c C s* t �| � | �� | j�� \}}|| _d S r9 )r� rI r~ rC r� �cli_conn�rH r� �addrr r r rI �
zSocketConnectedTest.setUpc C r� r9 )r� r r� rL rG r r r rL # r� z SocketConnectedTest.tearDownc C s( t �| � | j�t| jf� | j| _d S r9 )r� r� r� r� r� rE � serv_connrG r r r r� ( s
z SocketConnectedTest.clientSetUpc C r� r9 )r� r r� r� rG r r r r� - r� z"SocketConnectedTest.clientTearDownNr� ) rO rP rQ rg rY rI rL r� r� r r r r r� s
r� c @ s> e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� ZdS )�SocketPairTestr� c C s* t jj| |d� t�| � d | _d | _d S r� )r� r� rY ry r� rC r� r r r rY 4 s
zSocketPairTest.__init__c C s t �� S r9 )r �
socketpairrG r r r r� : � zSocketPairTest.socketpairc C s | � � \| _| _d S r9 )r� rC r� rG r r r rI > � zSocketPairTest.setUpc C s | j r| j �� d | _ d S r9 rK rG r r r rL A s
zSocketPairTest.tearDownc C � d S r9 r rG r r r r� F � zSocketPairTest.clientSetUpc C s$ | j r| j �� d | _ t�| � d S r9 r� rG r r r r� I s
z SocketPairTest.clientTearDownNr� ) rO rP rQ rY r� rI rL r� r� r r r r r� 2 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 r9 )� newSocketrC �
bindServerrG r r r rI _ r� zSocketTestBase.setUpc C � | � | j� | j�� | _dS )z9Bind server socket and set self.serv_addr to its address.N)�bindSockrC r� � serv_addrrG r r r r� c � zSocketTestBase.bindServerc C rJ r9 rK rG r r r rL h rM zSocketTestBase.tearDownN)rO rP rQ rg rI r� rL r r r r r� V s
r� c s e Zd ZdZ� fdd�Z� ZS )�SocketListeningTestMixinz%Mixin to listen on the server socket.c s t � �� | j�� d S r9 )rX rI rC rF rG r` r r rI p �
z SocketListeningTestMixin.setUp)rO rP rQ rg rI rh r r r` r r� m 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 r9 )rX rY ry r] r` r r rY } rb z ThreadedSocketTestMixin.__init__c C r� r9 )�newClientSocketr� �
bindClientrG r r r r� � r� z#ThreadedSocketTestMixin.clientSetUpc C s | � � S )z&Return a new socket for use as client.)r� rG r r r r� � r� z'ThreadedSocketTestMixin.newClientSocketc C r� )z8Bind client socket and set self.cli_addr to its address.N)r� r� r� r� rG r r r r� � r� z"ThreadedSocketTestMixin.bindClientc C r� r9 r� rG r r r r� � r� z&ThreadedSocketTestMixin.clientTearDown)
rO rP rQ rg rY r� r� r� r� rh r r r` r r� u 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 r9 )rX rI r~ rC r� r� r� r` r r rI � r� z ConnectedStreamTestMixin.setUpc s | j �� d | _ t� �� d S r9 )r� r rX rL rG r` r r rL � r� z!ConnectedStreamTestMixin.tearDownc s$ t � �� | j�| j� | j| _d S r9 )rX r� r� r� r� r� rG r` r r r� � s
z$ConnectedStreamTestMixin.clientSetUpc s6 z
| j �� d | _ W n ty Y nw t� �� d S r9 )r� r r rX r� rG r` r r r� � s
�z'ConnectedStreamTestMixin.clientTearDown) rO rP rQ rg rI rL r� r� rh 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 r9 )�tempfile�mkdtemp�dir_pathrd �os�rmdirrX rI rG r` r r rI � s
zUnixSocketTestBase.setUpc C s, t j| jd�}t�||� | �tj|� d S )N)�dir)r� Zmktempr� r �bind_unix_socketrd r �unlink)rH �sock�pathr r r r� � s zUnixSocketTestBase.bindSock)rO rP rQ rg rI r� rh 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 r9 )r �AF_UNIXr6 rG r r r r� � � zUnixStreamBase.newSocketN�rO rP rQ rg 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 )rX rI r� rE rG r` r r rI � s
zInetTestBase.setUpc C s t j|| jd� d S )N��host)r rD r �rH r r r r r� � r� zInetTestBase.bindSock) rO rP rQ rg r� r rI r� rh r r r` r r � s
r c @ r )� TCPTestBasez#Base class for TCP-over-IPv4 tests.c C r r9 )r rB r6 rG 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 r9 )r rB r% rG 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 r9 )r rB r% r> rG 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 r9 )r rB r6 � IPPROTO_SCTPrG 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)rO rP rQ rg r �HOSTv6r r r r r r � s
r c @ r )� UDP6TestBasez#Base class for UDP-over-IPv6 tests.c C r r9 )r �AF_INET6r% rG 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 r9 )r r r% r> rG 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� r9 r �r^ r_ r r r � client_pass r� z%skipWithClientIf.<locals>.client_passc s* t ���| �}t| t�s� fdd�|_|S )Nc s � S r9 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 r9 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 r9 r r, r r r � <genexpr>) � � z requireAttrs.<locals>.<genexpr>)r+ �join)r% Z
attributes�missingr r) r � requireAttrs" s �r8 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/ 4 s
�
�z!requireSocket.<locals>.<listcomp>r1 r2 c s r3 r9 r r, r r r r4 7 r5 z requireSocket.<locals>.<genexpr>c S s$ g | ]}t |t�rtt|�n|�qS r )r" r9 r� r r: r r r r/ 9 s �z can't create socket({0}): {1}c s s � | ]}t |�V qd S r9 )r9 �r- �or r r r4 E s � )r6 r r r9 r r+ �format)r^ �errr7 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�� �ZFd�d�� ZGe�ejd��e�Hejd�kd��e�HeId��e�eed(�d��d�d�� ����ZJe�ejd��e�ejd�kd��d�d�� ��ZKe�ejd��e�Hejd�kd��e�HeId��e�eed(�d��d�d�� ����ZLe�ejd��e�ejd�kd��d�d�� ��ZMd�d�� ZNd�d�� ZOd�d�� ZPe�eed��d��d�d�� �ZQd�d�� ZRd�d�� ZSd�d�� ZTd�d�� ZUd�d�� ZVe�Hejdkd��d�d�� �ZWd�d�� ZXd�d�� ZYdS )��GeneralModuleTestsc C s< dd l }| �tj|ju � t�� }| �|tj� |�� d S r� )�_socket�
assertTruer Z
SocketType�assertIsInstancer )rH rA r" r r r � test_SocketType_is_socketobjectM 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=0Zraddr�� 127.0.0.1r Zladdrz[closed])
r rB r6 �assertIn�filenor� � assertNotInrp r9 r� �rH r" r r r � test_reprT 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>) rA r rB r6 rH �familyr# �protor� r� r )rH r" �expectedr r r �test_csocket_reprb 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 rB r6 r r� rH r �
gc_collect�ReferenceError�fail)rH r" �pr r r � test_weakrefo 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�gaierror�rH r� r r r �testSocketError| 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 rB r% rd r rp r� rV r� �sendtor� r9 � exceptionrG )rH r" �sockname�cmr r r �testSendtoErrors� 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 r9 ) r rB �has_ipv6r r6 r% r �SOCK_RDMr, �
SOL_SOCKET� SO_REUSEADDRrG r r r �testCrucialConstants� s
z'GeneralModuleTests.testCrucialConstantsc C s t j t j t jrt j d S d S r9 )r Z IPPROTO_TCP� IPPROTO_UDPrd � IPPROTO_IPV6rG 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 r9 ) r ZIPPROTO_ICLFXBMZ
IPPROTO_STZ IPPROTO_CBTZ IPPROTO_IGPZ IPPROTO_RDPZ IPPROTO_PGMZ IPPROTO_L2TPr rG 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> Z
IPV6_CHECKSUMZ
IPV6_RTHDRZIPV6_USE_MIN_MTU� IPV6_TCLASSZIPV6_RECVDSTOPTSZ IPV6_PKTINFOZIPV6_RECVRTHDRZIPV6_RECVHOPOPTSZ IPV6_NEXTHOPZ IPV6_DSTOPTSZ IPV6_PATHMTU�
IPV6_HOPLIMIT�IPV6_RECVHOPLIMITZIPV6_RECVPKTINFOZIPV6_RTHDR_TYPE_0Z
IPV6_DONTFRAG�IPV6_RECVTCLASSZIPV6_RTHDRDSTOPTSZ IPV6_HOPOPTSZIPV6_RECVPATHMTUz Missing RFC3542 socket option '�')rB r r )rH 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 rr rB �find�
gethostbyaddr�getfqdnrR r� )rH ZhostnameZipZhname�aliasesZipaddrsZall_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| rG r~ )rH 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:1rU )rV r r r| r~ )rH 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 �errnoZEPERMrr r� )rH 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 �intr9 �
assertGreater�if_nametoindexr� �if_indextoname)rH Z
interfaces�indexr. Z_index�_namer r r �testInterfaceNameIndex, s
�z)GeneralModuleTests.testInterfaceNameIndexr� z&socket.if_indextoname() not available.c C s� | � ttjd� | � ttjd� | � ttjdd � | � ttjd� ttd�r{tt�� �}|D ]&}|d }||vrU| � ttf��
t�|� W d � n1 sPw Y q/dD ]$}||vrz| � ttf��
t�|� W d � n1 suw Y qXd S d S ) Nr ���r� �� � _DEADBEEFr� � )� �� l ���� ) rV r r r� �
OverflowErrorr� r �dictr� )rH �indicesr� Zindex2r r r � testInvalidInterfaceIndexToName< s*
�� ���z2GeneralModuleTests.testInvalidInterfaceIndexToNamer� z&socket.if_nametoindex() not available.c C s$ | � ttjd� | � ttjd� d S )Nr r� )rV r� r r� r rG r r r � testInvalidInterfaceNameToIndexO s 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� rO r � getnameinfor� rR )rH �origr r r �testRefCountGetNameInfoU s
��z*GeneralModuleTests.testRefCountGetNameInfoc C s( z t �dd� W d S ty Y d S w )N)rl r r r r )r r� r rG r r r �testInterpreterCrasha 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� rV r� )rH Zsizes�func�size�maskr? Zswappedr r r �testNtoHi 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� r� r� ���r� r r� � )
� _testcapiZINT_MIN�INT_MAXr r� r� r� r� rV r� )rH r� Z
s_good_valuesZ
l_good_valuesZ l_bad_valuesZ s_bad_values�kr r r �testNtoHErrorsw s.
���
�z!GeneralModuleTests.testNtoHErrorsc C s | j }tj�d�stjdv rd}nd}|D ]}z t�|d�}W n 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)�linuxrp )�daytimeZqotd�domain)Zechor� r� Ztcp�getandroidapilevel� Zudpr� r� )r� r� �platform�
startswithr Z
getservbynamer r r� r Z
is_androidZ
getservbyportrV r� )rH �eqZservicesZservicerE Zport2Zudpportr r r �
testGetServBy� 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 r: �
gettimeoutr= r; rV �
ValueErrorr� )rH 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� rG 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� rB c �
�� | �S r9 r ��a)rB r� r r r � �
z5GeneralModuleTests.testIPv4toString.<locals>.<lambda>c � � � ttf| |�S r9 �rV r r� �r� r� rG 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� rB r� r
�rH r �g�
assertInvalidr )rB r� rH 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 rd �IPv6 not available�+could not import needed symbols from socket�win32�::�&' �IPv6 might not be supportedc r� r9 r r� )r r� r r r r� z5GeneralModuleTests.testIPv6toString.<locals>.<lambda>c � �� ttf� | �S r9 r� r� �r rH 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 rd rr � ImportErrorr� r� r �winerrorr� r
�rH rd r� r� r )r r r� rH 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� rB c r� r9 r r� )rB r� r r r A r� z5GeneralModuleTests.testStringToIPv4.<locals>.<lambda>c r� r9 r� r� rG r r r B r� z1.0.1.0s z
170.85.170.85s �U�Ur� r� r� r� s s r� )r r� r� rB r� � bytearrayr� r )rB r� rH 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 rd r� r� r� r� r� r� c r� r9 r r� )r r� r r r h r� z5GeneralModuleTests.testStringToIPv6.<locals>.<lambda>c r� r9 r� r� r� r r r i r� r� r� s z!aef:b01:506:1001:ffff:9997:55:170s
� ���� Ups s s ) r r� r rd rr r� r� r� r r� r� r� r� r )r r r� rH r �testStringToIPv6W s8
� �
��� z#GeneralModuleTests.testStringToIPv6c C s� t � t jt j�}| �|j� t�� D ]-}t�� }z |� d|f� W n t
y> } z|j t j ks3|dkr4� W Y d }~qd }~ww |�
� }z t �t �� �}W n t
y[ | �d� Y nw | �|d d|fd|d � | �|d |� d S )Nr� � rz r z
%s invalidr )r rB r6 rd r � itertools�countr �find_unused_portrp r r� �
EADDRINUSEr� r| r{ rr rG r� )rH r r? rE 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 rB r6 rd r �
getsockoptrf rg � assertFalse�rH r Zreuser r r �testGetSockOpt� 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 rB r6 rd r �
setsockoptrf rg r� r� r� r r r �testSetSockOpt� s
z!GeneralModuleTests.testSetSockOptc C sN t � t jt j��
}|�d� W d � n1 sw Y | �t|jd� d S )Nr � spam)r rB r6 �
settimeoutrV 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 �rH z#close() should raise EBADF/ENOTSOCK) r rp Z
_LOCALHOSTrH r r rG r� �EBADF�ENOTSOCKrR )rH 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 rB r6 r� rL r rG r# r� rM 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� )r rd r r r� rV r� rp r� r� r� r r� r� )rH r rE 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� ) rB r r rd r rV r� r r� rJ 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 rd r r r r r� rr rV r� )rH r" r� Z
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� rF r� r� r� Zhttp�AddressFamily.AF_INET�SocketKind.SOCK_STREAMr )r rE �rL �r# )rM )�flags)r rE rL r# rM r u ��AI_NUMERICSERV)r � getaddrinforX r� Z EAI_SERVICErr r� r� �lenr � IPV6_ENABLEDr r� r� rB r6 r9 ZSOL_TCPZ
AI_PASSIVE� AF_UNSPECrV �UnicodeEncodeErrorr
)
rH r> �infoZinfosrL r# r� Zsocktyper� �br r r �testGetaddrinfo� s�
��
� � �
���z"GeneralModuleTests.testGetaddrinfoc C s | � ttjdd� d S )N)zmail.python.orgr r )rV r r r� rG r r r �test_getnameinfoE � 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| Zgethostbyname_exr r r6 )rH r� r r r � test_idnaI 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� �rV r� �mathZacosh�r^ rG r r �
ok_handler` r� z@GeneralModuleTests.check_sendall_interrupted.<locals>.ok_handlerc s � � ttjd� dd d S �Nr r r r rG r r �raising_handlerb s zEGeneralModuleTests.check_sendall_interrupted.<locals>.raising_handlerg �?r � xr )r �signalr rr r� �SIGALRMr� rV �ZeroDivisionErrorr �sendallr �
SOCK_MAX_SIZE� TimeoutErrorr )rH Z with_timeoutr r �cr" � old_alarmr rG r �check_sendall_interruptedZ s@
�
�
�
�
z,GeneralModuleTests.check_sendall_interruptedc C � | � d� d S �NF�r& rG r r r �test_sendall_interruptedy � z+GeneralModuleTests.test_sendall_interruptedc C r' )NTr) rG r r r �%test_sendall_interrupted_with_timeout| r+ 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 rB r6 r� Z assertWarns�ResourceWarningr rP rG r9 Zwarningr^ �makefile)rH r r rb r r r r �test_dealloc_warn 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 rB r6 r. r r� r� �rH r �fpr r r �test_name_closed_socketio� 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 r. rB �readabler� �writable�seekabler rV r� r0 r r r � test_unusable_closed_socketio� 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 rp r� r rH rV r rF r� r r r r �test_socket_close� 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 Zrw�w�wb��moder r ��encoding)�subTestr r. r� r< )rH r< r r> r1 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)Zrtrl �+r� r; z invalid mode)r? r �assertRaisesRegexr� r. )rH 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�pickleZHIGHEST_PROTOCOLrV r� �dumps�loadsrB r� r6 )rH r ZprotocolrL 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 rB r6 rp r� rF )rH Zbacklog�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 r )
r� r rB r6 rp r� rV r� rF r� )rH r� rI 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����) rV r� r r� r r r r6 rp rJ 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% ri r� )rH r� �sockaddrr r r �test_getaddrinfo_ipv6_basic� s �z.GeneralModuleTests.test_getaddrinfo_ipv6_basicc C s0 | � t�� t�d�� | � t�� t�d�� d S )Nr� r� )r� r r rG r r r � test_getfqdn_filter_localhost� s z0GeneralModuleTests.test_getfqdn_filter_localhostr� 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 rO )r r� r r r% ri r� )rH �ifindex�test_interfacer� rP 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�* rS r rO r )r r r9 r r% ri r� )rH rT r� rP 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 rN r �ff02::1de:c0:face:8d%�1234)r r� r� �NI_NUMERICHOST�NI_NUMERICSERVr� )rH rT rU rP �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 )NrW rN r r rY rZ )r r� r[ r\ r� r9 )rH rT rP 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 rB r6 r� r9 rL r# rJ 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 r6 rB r� r# r� � setblocking)rH ra r� Z sock_typer" r r r � test_socket_consistent_sock_type0 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 )> ra r� Nr )r- r. �kindr r r r4 K s � ��zEGeneralModuleTests.test_unknown_socket_family_repr.<locals>.<genexpr>r� )rL r# rM rH r )r rB r6 �detach�maxZ
AddressFamily� __members__�valuesZ
SocketKindr� r� rL r# rG rM )rH 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 r9 �ri )rH ri r r r rY \ �
zEGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.__init__c S � | j S r9 rl rG r r r rH _ s zCGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.filenoN)rO rP rQ rY rH r r r r �File[ s ro r� r� ) r r� r �curdir�O_RDONLYr rV �_GiveupOnSendfile�_sendfile_use_sendfiler� r� )rH ro r ri r r r �test__sendfile_use_sendfileY s
� � ��"�z.GeneralModuleTests.test__sendfile_use_sendfilec C st | � |j|� | � |j|� |�� }tj|d�}| �|j� |�� | � |j|� | � |j|� | � |�� |� d S �Nr� )r� rL r# rH r rd r re )rH r" rL �styperi �s2r r r �_test_socket_filenok 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 rB r6 rd r rp r r� rx r r% r
r r r� r� �shutilZrmtreer r� r r6 r� )rH r" Ztmpdirr r r �test_socket_filenox s8
�
��z%GeneralModuleTests.test_socket_filenoc C �D | � t�� tjtjtjdd� W d � d S 1 sw Y d S )Ng @E@r� �rV r� r rB r6 rG r r r � test_socket_fileno_rejects_float� � "�z3GeneralModuleTests.test_socket_fileno_rejects_floatc C r{ )N�foor� r| rG r r r �&test_socket_fileno_rejects_other_types� 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 rB r6 rG r r r �)test_socket_fileno_rejects_invalid_socket� s "�z<GeneralModuleTests.test_socket_fileno_rejects_invalid_socketzWindows disallows -1 onlyc C r� )Nr� i���r� r� rG r r r �#test_socket_fileno_rejects_negative� 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� ) rV r r r Z make_bad_fdrG r` r� r� rB r6 )rH Z WSAENOTSOCKrb 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 ru ) r� ZNamedTemporaryFilerV r r rH rB r6 r� r` r� r� )rH Zafilerb r r r �%test_socket_fileno_requires_socket_fd� s
� � �"�z8GeneralModuleTests.test_socket_fileno_requires_socket_fd)ZrO rP rQ rD rK r� �
skipUnlessrA rO rT rZ rc rh rk r� r. ro r� r� r r
ry 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_enabledr r& r* r, r/ r2 r7 r8 r@ rC rH rJ rK rM rQ rR �skipIfr
rV rX r^ r_ r` rc rj rt rx rz r} r� r� r� r� r� r r r r r@ K 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 r9 )r �AF_CANr r rG r r r rh � 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 r9 )r r� ZCAN_BCM_TX_SETUPZCAN_BCM_TX_DELETEZCAN_BCM_TX_READ�CAN_BCM_TX_SENDZCAN_BCM_RX_SETUPZCAN_BCM_RX_DELETEZCAN_BCM_RX_READZCAN_BCM_TX_STATUSZCAN_BCM_TX_EXPIREDZCAN_BCM_RX_STATUSZCAN_BCM_RX_TIMEOUTZCAN_BCM_RX_CHANGEDZCAN_BCM_SETTIMERZCAN_BCM_STARTTIMERZCAN_BCM_TX_COUNTEVTZCAN_BCM_TX_ANNOUNCEZCAN_BCM_TX_CP_CAN_IDZCAN_BCM_RX_FILTER_IDZCAN_BCM_RX_CHECK_DLCZCAN_BCM_RX_NO_AUTOTIMERZCAN_BCM_RX_ANNOUNCE_RESUMEZCAN_BCM_TX_RESET_MULTI_IDXZCAN_BCM_RX_RTR_FRAMErG r r r �testBCMConstants� s0
z BasicCANTest.testBCMConstantsc C �: t � t jt jt j�� }W d � d S 1 sw Y d S r9 �r r r r rJ r r r �testCreateSocket� � "�z BasicCANTest.testCreateSocketc C r� r9 )r r r% r� rJ 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 rp r� r� )rH 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 rp rJ 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� )rH 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=IIrm ) r �packr r r r r� r� r� r� r� r� )rH �can_idZcan_maskZ
can_filterr" r r r �
testFilter
s �"�zBasicCANTest.testFilterN)rO rP rQ rh 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� rY r� r r r rY r� zCANTest.__init__c C s& t |�}|�dd�}t�| j|||�S )zBuild a CAN frame.rm � )r �ljustr r� rt )�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 rt )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" �recvfromrs r� �cfrq �rH r� r� r r r �
testSendFrame) s zCANTest.testSendFramec C � | � dd�| _| j�| j� d S )Nr s �r� r� r� r� rG r r r �_testSendFrame. � zCANTest._testSendFramec C s$ | j �| j�\}}| �| j|� d S r9 )r" r� rs r� r� r� r r r �testSendMaxFrame2 rS zCANTest.testSendMaxFramec C r� )Nr s r� rG r r r �_testSendMaxFrame6 r� zCANTest._testSendMaxFramec C �D | j �| j�\}}| �| j|� | j �| j�\}}| �| j|� d S r9 )r" r� rs 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� rG r r r �_testSendMultiFramesA s z CANTest._testSendMultiFramesr� r� c C sP | j �| j�\}}| �| j|� | �|�\}}}| �| j|� | �| j|� d S r9 )r� r� rs r� r� r� r� r� )rH r� r� r� r� r� r r r �_testBCMH 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|||||||| |
�
} | | j } |�| �}
| �|
t| �� d S )N�# )� � �� r i" r )r r r% r� rd r r� rq r� �bytesr� r� r� r� r r� ru r� r� r )rH ZbcmZopcoder r� Z
ival1_secondsZ
ival1_usecZ
ival2_secondsZ
ival2_usecZ
bcm_can_idZnframes�headerZheader_plus_frameZ
bytes_sentr r r �testBCMQ s4 �
zCANTest.testBCMNr� )rO rP rQ rY � 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 �Nrj �rX rY rq r] r` r r rY t �
zISOTPTest.__init__c C s t j t j t j t j d S r9 )r r� r r& r% rG r r r rh x s
z ISOTPTest.testCrucialConstantsc C r� r9 r� rJ r r r r� ~ r� zISOTPTest.testCreateSocketr& z(socket.CAN_ISOTP required for this test.c C r� r9 )r r r% r& rJ r r r �testCreateISOTPSocket� 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 rp rJ 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 ro ) r r r% r& rq rp r� r� r r� �ENODEVrr �rH r" r� r� r r r �testBind� s
&� ����zISOTPTest.testBind)rO rP rQ rY rh r� r� r� r r r� r� r� rh r r r` r r� q 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 rY � r� zJ1939Test.__init__r( z(socket.CAN_J1939 required for this test.c C r� r9 )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_MAXrG r r r �testJ1939Constants� s0
z J1939Test.testJ1939Constantsc C r� r9 )r r r% r( rJ 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 rn )r r r% r( rq r� r� r� rp r� r� r r� r� rr r� r r r r� � s
&� ����zJ1939Test.testBind) rO rP rQ rY r� r� r r r� r� r� rh r r r` r r� � 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 r9 )r ZAF_RDSr+ rG r r r rh � s
z!BasicRDSTest.testCrucialConstantsc C s8 t � t jt jd�� }W d � d S 1 sw Y d S r� )r r+ r, rJ 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� rf Z SO_RCVBUFZ SO_SNDBUF)rH rs r" r r r �testSocketBufferSize� s
"�z!BasicRDSTest.testSocketBufferSizeN)rO rP rQ rh 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� rY r� r r r rY � r� zRDSTest.__init__c s t � �� t�� | _d S r9 )rX rI rZ r� ZevtrG r` r r rI � r� z
RDSTest.setUpc C s2 | j �| j�\}}| �| j|� | �| j|� d S r9 )rC r� rs r� r� r� �rH r� r� r r r �testSendAndRecv� s zRDSTest.testSendAndRecvc C �" d| _ | j�| j dt| jf� d S �Nr� r �r� r� r_ r� rE rG r r r �_testSendAndRecv� � zRDSTest._testSendAndRecvc C sH | j �| jtj�\}}| �| j|� | j �| j�\}}| �| j|� d S r9 )rC r� rs r �MSG_PEEKr� r� r� r r r �testPeek� s zRDSTest.testPeekc C r� r� r� rG r r r � _testPeek r� zRDSTest._testPeek�recvmsgc C s( | j �| j�\}}}}| �| j|� d S r9 )rC r� rs r� r� )rH 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� rE rG r r r �_testSendAndRecvMsg s zRDSTest._testSendAndRecvMsgc C r� r9 )rC r� rs 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� r_ r� rE r� rG r r r �_testSendAndRecvMulti s z RDSTest._testSendAndRecvMultic C sL t � | jgg g d�\}}}| �| j|� | j�| j�\}}| �| j|� d S )Ng @)�selectrC rG r� rs r� r� )rH r r9 rl r� r� r r r �
testSelect! s zRDSTest.testSelectc C r� )Ns selectr r� rG r r r � _testSelect' r� zRDSTest._testSelectr� )rO rP rQ rY rI r� r� r� r� r8 r r� r� r� r� r� r� rh 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 r9 )r r1 rG r r r rh / rm z%BasicQIPCRTRTest.testCrucialConstantsc C �6 t � t jt j�� }W d � d S 1 sw Y d S r9 )r r1 r% rJ r r r r� 2 � "�z!BasicQIPCRTRTest.testCreateSocketc C sJ t � t jt j��}| �|�� d d� W d � d S 1 s w Y d S �Nr r )r r1 r% r� r� rJ r r r � testUnbound6 � "�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 r1 r% r rD r� �assertNotEqualrJ 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 r1 r% rV r r rD rJ r r r �testInvalidBindSock? r� 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 r1 r% r� r� r� rJ r r r �testAutoBindSockC s
"�z!BasicQIPCRTRTest.testAutoBindSockN) rO rP rQ rh r� r� r� r� r� r r r r r� + s r� c @ �, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�BasicVSOCKTestc C r� r9 )r r� rG r r r rh M rm z#BasicVSOCKTest.testCrucialConstantsc C rn r9 ) r �SO_VM_SOCKETS_BUFFER_SIZE� SO_VM_SOCKETS_BUFFER_MIN_SIZE� SO_VM_SOCKETS_BUFFER_MAX_SIZEr� ZVMADDR_PORT_ANYr� ZVM_SOCKETS_INVALID_VERSIONr rG r r r �testVSOCKConstantsP s
z!BasicVSOCKTest.testVSOCKConstantsc C r� r9 )r r� r6 rJ r r r r� Z r� 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� r6 r� r r� r r� r� )rH r" Zorig_maxr� Zorig_minr r r r� ^ sF ���
�
�
�
��
��
��"�z#BasicVSOCKTest.testSocketBufferSizeN)rO rP rQ rh r r� r� r r r r r� H 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_LOCALr5 r7 r� r� � BTPROTO_HCIZSOL_HCI�
BTPROTO_L2CAPr� � BTPROTO_SCOrG r r r �testBluetoothConstants} s
�z)BasicBluetoothTest.testBluetoothConstantsc C r� r9 )r r5 r6 r7 rJ r r r �testCreateRfcommSocket� r� z)BasicBluetoothTest.testCreateRfcommSocketr� z&windows does not support L2CAP socketsc C r� r9 )r r5 r, r rJ r r r �testCreateL2capSocket� � "�z(BasicBluetoothTest.testCreateL2capSocketz$windows does not support HCI socketsc C r� r9 )r r5 r r rJ r r r �testCreateHciSocket� r z&BasicBluetoothTest.testCreateHciSocketr� z.windows and freebsd do not support SCO socketsc C r� r9 )r r5 r, r rJ r r r �testCreateScoSocket� r� z&BasicBluetoothTest.testCreateScoSocketN)
rO rP rQ r r r� r� r� r� r
r r� r
r r r r r y s
�r 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� rY r� r r r rY � r� zBasicTCPTest.__init__c C � | j �d�}| �|t� d S r� )r� r� r� r� rY r r r �testRecv� r� zBasicTCPTest.testRecvc C � | j �t� d S r9 �r� r� r� rG r r r � _testRecv� r zBasicTCPTest._testRecvc C s8 | j �tt�d �}| j �d�}|| }| �|t� d S �N� r� )r� r� r r� r� )rH �seg1�seg2r� r r r �testOverFlowRecv� s z BasicTCPTest.testOverFlowRecvc C r r9 r rG r r r �_testOverFlowRecv� r z BasicTCPTest._testOverFlowRecvc C s | j �d�\}}| �|t� d S r� )r� r� r� r� �rH r� r� r r r � testRecvFrom� � zBasicTCPTest.testRecvFromc C r r9 r rG r r r �
_testRecvFrom� r zBasicTCPTest._testRecvFromc C s@ | j �tt�d �\}}| j �d�\}}|| }| �|t� d S r )r� r� r r� r� )rH r r� r r� r r r �testOverFlowRecvFrom� s z!BasicTCPTest.testOverFlowRecvFromc C r r9 r rG r r r �_testOverFlowRecvFrom� r z"BasicTCPTest._testOverFlowRecvFromc C s2 d} | j �d�}|s n||7 }q| �|d� d S )N� r r� � ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff)r� r� r� )rH r� �readr r r � testSendAll� s �zBasicTCPTest.testSendAllc C s d}| j �|� d S )Nr" )r� r! )rH Z big_chunkr r r � _testSendAll� s zBasicTCPTest._testSendAllc C sP | j �� }t�|tjtj�}| �|j� | �|tj� |� d�}| �
|t � d S r� ) r� rH r ZfromfdrB r6 rd r rC r� r� r� )rH ri r r� r r r �
testFromFd� s
zBasicTCPTest.testFromFdc C r r9 r rG r r r � _testFromFd� r zBasicTCPTest._testFromFdc C s0 | j �� }| �|j� |�d�}| �|t� d S r� )r� �duprd r r� r� r� )rH r r� r r r �testDup� s
zBasicTCPTest.testDupc C r r9 r rG r r r �_testDup� r zBasicTCPTest._testDupc C s& | j �d�}| �|t� | j�� d S r� )r� r� r� r� r� r� rY r r r � testShutdown� s zBasicTCPTest.testShutdownc C s | j �t� | j �d� d S r )r� r� r� �shutdownrG r r r �
_testShutdown� � 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� rV r� r, r� �UINT_MAX�rH 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� rH re r� rB �_closedrV r r� r r rB r6 rd r� )rH rH r r r� r r r �
testDetach
s
zBasicTCPTest.testDetachc C r r9 r rG r r r � _testDetach
r zBasicTCPTest._testDetachNr� )rO rP rQ rY r r r r r r r r r$ r% r& r' r) r* r+ r- r r� ZtestShutdown_overflowr1 r3 r4 r r r r r � s.
r c @ �F e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dS )� BasicUDPTestr� c C r� r� )r� rY r� r r r rY
r� zBasicUDPTest.__init__c C � | j �tt��}| �|t� d S r9 �rC r� r r� r� rY r r r �testSendtoAndRecv
r z BasicUDPTest.testSendtoAndRecvc C � | j �tdt| jf� d S r� �r� r_ r� r� rE rG r r r �_testSendtoAndRecv%
r� z BasicUDPTest._testSendtoAndRecvc C �$ | j �tt��\}}| �|t� d S r9 �rC r� r r� r� r r r r r (
� zBasicUDPTest.testRecvFromc C r: r� r; rG r r r r -
r� zBasicUDPTest._testRecvFromc C � | � t| jjd� d S �Nr� �rV r� rC r� rG r r r �testRecvFromNegative0
r z!BasicUDPTest.testRecvFromNegativec C r: r� r; rG r r r �_testRecvFromNegative4
r� z"BasicUDPTest._testRecvFromNegativeNr� �
rO rP rQ rY r9 r< r r rC rD r r r r r6
s
r6 c @ r5 )�BasicUDPLITETestr� c C r� r� )r� rY r� r r r rY <
r� zBasicUDPLITETest.__init__c C r7 r9 r8 rY r r r r9 ?
r z"BasicUDPLITETest.testSendtoAndRecvc C r: r� r; rG r r r r< D
r� z#BasicUDPLITETest._testSendtoAndRecvc C r= r9 r> r r r r r G
r? z BasicUDPLITETest.testRecvFromc C r: r� r; rG r r r r L
r� z BasicUDPLITETest._testRecvFromc C r@ rA rB rG r r r rC O
r z%BasicUDPLITETest.testRecvFromNegativec C r: r� r; rG r r r rD S
r� z&BasicUDPLITETest._testRecvFromNegativeNr� rE r r r r rF 8
s
rF 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 r9 )rZ r� �
misc_eventrX rI rG r` r r rI x
r� zSendrecvmsgBase.setUpc C s | j �|�S r9 )�cli_sockr� rY r r r � sendToServer|
s z SendrecvmsgBase.sendToServerr c G s | j j|| jt|�d � � S r9 )rI r� �sendmsg_to_server_defaultsr �rH r^ r r r �sendmsgToServer�
s �z SendrecvmsgBase.sendmsgToServerc G s |j |g|�R � }| �|� |S r9 )r� �registerRecvmsgResult)rH r rs r^ �resultr r r � doRecvmsg�
s
zSendrecvmsgBase.doRecvmsgc C r� r9 r �rH rO r r r rN �
s z%SendrecvmsgBase.registerRecvmsgResultc C s | � ||� d S r9 �r� �rH Zaddr1Zaddr2r r r �checkRecvmsgAddress�
� z#SendrecvmsgBase.checkRecvmsgAddressr )�
MSG_CTRUNCZMSG_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� )
rH r �eor�checkset�
checkunset�ignoreZ
defaultsetZ defaultunsetZinbothr� r r r �
checkFlags�
s&
�zSendrecvmsgBase.checkFlags)Nr r r )rO rP rQ r �LOOPBACK_TIMEOUT� fail_timeoutrI rJ rK rM rP rN rT rX r. r� r rW rY rZ r` rh r r r` r rG q
s
rG c @ rT )�RecvmsgIntoMixinc G sh t |�}|j|gg|�R � }| �|� | �|d d� | �|d |� t|d |d � �f|dd � S r )r� � recvmsg_intorN �assertGreaterEqual�assertLessEqualr� )rH r rs r^ �bufrO r r r rP �
s
"zRecvmsgIntoMixin.doRecvmsgN)rO rP rQ rP r r r r rc �
� rc c � e Zd Ze� fdd��Z� ZS )�SendrecvmsgDgramFlagsBasec � t � jtjB S r9 )rX rZ r � MSG_TRUNCrG r` r r rZ �
� z5SendrecvmsgDgramFlagsBase.msg_flags_non_eor_indicator)rO rP rQ �propertyrZ rh r r r` r rj �
� rj c ri )�SendrecvmsgSCTPFlagsBasec rk r9 )rX rY r ZMSG_EORrG r` r r rY �
rm z0SendrecvmsgSCTPFlagsBase.msg_flags_eor_indicator)rO rP rQ rn rY rh r r r` r rp �
ro rp c @ s8 e Zd Zedd� �Zedd� �Zedd� �Zdd� Zd S )
� SendrecvmsgConnectionlessBasec C rn r9 )rC rG r r r � serv_sock � z'SendrecvmsgConnectionlessBase.serv_sockc C rn r9 )r� rG r r r rI
rs z&SendrecvmsgConnectionlessBase.cli_sockc C s g g d| j fS r� )r� rG r r r rK rm z8SendrecvmsgConnectionlessBase.sendmsg_to_server_defaultsc C s | j �|| j�S r9 )rI r_ r� rY r r r rJ r z*SendrecvmsgConnectionlessBase.sendToServerN)rO rP rQ rn rr rI rK rJ r r r r rq s
rq c @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )�SendrecvmsgConnectedBasec C rn r9 )r� rG r r r rr rs z"SendrecvmsgConnectedBase.serv_sockc C rn r9 )r� rG r r r rI # rs z!SendrecvmsgConnectedBase.cli_sockc C r� r9 r rS r r r rT ' � z,SendrecvmsgConnectedBase.checkRecvmsgAddressN)rO rP rQ rn rr rI rT r r r r rt s
rt c � e Zd Z� fdd�Z� ZS )� SendrecvmsgServerTimeoutBasec � t � �� | j�| j� d S r9 )rX rI rr r� rb rG r` r r rI 0 �
z"SendrecvmsgServerTimeoutBase.setUp)rO rP rQ rI rh r r r` r rw - s rw 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 r9 �r� rr r� r r� rG r r r � testSendmsg9 � zSendmsgTests.testSendmsgc C s | � | �tg�tt�� d S r9 �r� rM r� r rG r r r � _testSendmsg= � zSendmsgTests._testSendmsgc C r{ r9 r| rG r r r �testSendmsgDataGenerator@ r~ z%SendmsgTests.testSendmsgDataGeneratorc C s& | � | �dd� tfD ��tt�� d S )Nc s r3 r9 r r; r r r r4 E r5 z9SendmsgTests._testSendmsgDataGenerator.<locals>.<genexpr>r rG r r r �_testSendmsgDataGeneratorD s �z&SendmsgTests._testSendmsgDataGeneratorc C r{ r9 r| rG r r r � testSendmsgAncillaryGeneratorH r~ z*SendmsgTests.testSendmsgAncillaryGeneratorc C s( | � | �tgdd� dD ��tt�� d S )Nc s r3 r9 r r; r r r r4 M r5 z>SendmsgTests._testSendmsgAncillaryGenerator.<locals>.<genexpr>r r rG r r r � _testSendmsgAncillaryGeneratorL s �z+SendmsgTests._testSendmsgAncillaryGeneratorc C r{ r9 r| rG r r r �testSendmsgArrayP r~ z SendmsgTests.testSendmsgArrayc C s$ | � | �t�dt�g�tt�� d S )N�B)r� rM �arrayr� r rG r r r �_testSendmsgArrayT s �z SendmsgTests._testSendmsgArrayc C r{ r9 r| rG r r r �testSendmsgGatherX r~ z SendmsgTests.testSendmsgGatherc C �. | � | �td d� tdd � g�tt�� d S �Nr r rG r r r �_testSendmsgGather\ s .z SendmsgTests._testSendmsgGatherc C � | � | j�d�d� d S �Nr� � done�r� rr r� rG r r r �testSendmsgBadArgs_ � 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� )rV r� rI r� rM �objectr� rJ rG r r r �_testSendmsgBadArgsc s.
�
�
�
�
�
�
�z SendmsgTests._testSendmsgBadArgsc C r� r� r� rG r r r �testSendmsgBadCmsgu 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� rW r� �rV r� rM r� r� rJ rG r r r �_testSendmsgBadCmsgy s&
�
�
�
�
�
�z SendmsgTests._testSendmsgBadCmsg�
CMSG_SPACEc C r� r� r� rG r r r �testSendmsgBadMultiCmsg� s z$SendmsgTests.testSendmsgBadMultiCmsgc C s@ | � t| jtgg d�� | � t| jtgdt� g� | �d� d S )N�r r r! r� r� rG r r r �_testSendmsgBadMultiCmsg� s
�
�z%SendmsgTests._testSendmsgBadMultiCmsgc C r� r� r� rG r r r �testSendmsgExcessCmsgReject� � 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 rV r rM r� � assertIsNoner` r� rJ �rH rb r r r � _testSendmsgExcessCmsgReject� s
�z)SendmsgTests._testSendmsgExcessCmsgRejectc C r� r9 r rG r r r �testSendmsgAfterClose� s z"SendmsgTests.testSendmsgAfterClosec C s | j �� | �t| jtg� d S r9 )rI r rV r rM r� rG r r r �_testSendmsgAfterClose� s
z#SendmsgTests._testSendmsgAfterCloseN)rO rP rQ r} r� r� r� r� r� r� r� r� r� r� r� r� r� r8 r r� r$ r� r� r� r� r� r r r r rz 5 s.
rz c @ sR e Zd Zdd� Zdd� Zdd� Zdd� Zeej d vd
�d d � �Z
e
j d
d� �Z dS )�SendmsgStreamTestsc C r{ r9 r| rG r r r �testSendmsgExplicitNoneAddr� r~ z.SendmsgStreamTests.testSendmsgExplicitNoneAddrc C s" | � | �tgg dd �tt�� d S r� r rG r r r � _testSendmsgExplicitNoneAddr� s "z/SendmsgStreamTests._testSendmsgExplicitNoneAddrc C �. | � | j�d�d� | �| jj| jd�� d S �Nr� � aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa�r< �r� rr r� rB rH r� rb rG 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� )
rI r� rM r# r r� �ENOMEMrR rH r} �rH 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� rG 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� )
rV r rM r Z MSG_DONTWAITrG r` r� ZEAGAINZ EWOULDBLOCKr� rH r} r� r r r �_testSendmsgDontWait� s ��
�z'SendmsgStreamTests._testSendmsgDontWaitN)
rO rP rQ r� r� r� r� r+ r� r� r� r$ r� r r r r r� � s
�
r� c @ r@ )�SendmsgConnectionlessTestsc C r� r9 r rG r r r �testSendmsgNoDestAddr� ru z0SendmsgConnectionlessTests.testSendmsgNoDestAddrc C s2 | � t| jjtg� | � t| jjtgg dd � d S r� )rV r rI r� r� rG r r r �_testSendmsgNoDestAddr� s �
�z1SendmsgConnectionlessTests._testSendmsgNoDestAddrN)rO rP rQ 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\ �rP rr r r� r� rT r� r` �rH r� r� r r� r r r � testRecvmsg� s
z RecvmsgGenericTests.testRecvmsgc C � | � t� d S r9 �rJ r� rG r r r � _testRecvmsg r+ 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� r9 r� rG r r r � _testRecvmsgExplicitDefaults r+ z0RecvmsgGenericTests._testRecvmsgExplicitDefaultsc C sV | � | jtt�d �\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )NrW Tr� r� r� r r r �testRecvmsgShorter r� z&RecvmsgGenericTests.testRecvmsgShorterc C r� r9 r� rG r r r �_testRecvmsgShorter r+ z'RecvmsgGenericTests._testRecvmsgShorterc C s^ | � | jtt�d �\}}}}| �|td d� � | �|| j� | �|g � | j|dd� d S )Nr ���Fr� r� r� r r r �testRecvmsgTrunc! s
� z$RecvmsgGenericTests.testRecvmsgTruncc C r� r9 r� rG r r r �_testRecvmsgTrunc* r+ 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� r9 r� rG r r r � _testRecvmsgShortAncillaryBuf6 r+ z1RecvmsgGenericTests._testRecvmsgShortAncillaryBufc C r� )N� ( Tr� r� r� r r r �testRecvmsgLongAncillaryBuf9 r� z/RecvmsgGenericTests.testRecvmsgLongAncillaryBufc C r� r9 r� rG r r r � _testRecvmsgLongAncillaryBufB r+ z0RecvmsgGenericTests._testRecvmsgLongAncillaryBufc C s" | j �� | �t| j| j d� d S r� )rr r rV r rP rG r r r �testRecvmsgAfterCloseE s
z)RecvmsgGenericTests.testRecvmsgAfterClosec C r� r9 r rG r r r �_testRecvmsgAfterCloseJ r� z*RecvmsgGenericTests._testRecvmsgAfterClosec C sB z| j �d� | �t| j| j tt�� W | j�� d S | j�� w )Nr� ) rr r� rV r# rP r r� rH r} rG r r r �testRecvmsgTimeoutM s �z&RecvmsgGenericTests.testRecvmsgTimeoutc C s | � | jj| jd�� d S �Nr� )rB rH r� rb rG r r r �_testRecvmsgTimeoutV 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 ) Nr r r� Frl �r\ r_ Tr� ) rP rr r r� r r� r� rT r� r` r� r� r r r �testRecvmsgPeekY s. �
� � z#RecvmsgGenericTests.testRecvmsgPeekc C r� r9 r� rG r r r �_testRecvmsgPeek| rm z$RecvmsgGenericTests._testRecvmsgPeekr� c C s` | j �| j� | �| j tt��\}}}}| �|t� | �|| j� | �|g � | j |dd� d S r� )
rr r� rb rP r r� r� rT r� r` r� r r r �testRecvmsgFromSendmsg� s z*RecvmsgGenericTests.testRecvmsgFromSendmsgc C r� r� r rG r r r �_testRecvmsgFromSendmsg� s .z+RecvmsgGenericTests._testRecvmsgFromSendmsgN)rO rP rQ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r8 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� r! r� )rP rr r� rT r� r` r� r r r �testRecvmsgEOF� s
z(RecvmsgGenericStreamTests.testRecvmsgEOFc C � | j �� d S r9 )rI r rG r r r �_testRecvmsgEOF� r+ z)RecvmsgGenericStreamTests._testRecvmsgEOFc C s� | � | jtt�d �\}}}}| �|| j� | �|g � | j|dd� | � | jd�\}}}}| �|| j� | �|g � | j|dd� || }| �|t� d S )Nr Fr� r� T)rP rr r r� rT r� r� r` )rH r r� r r� r r� r r r �testRecvmsgOverflow� s
� z-RecvmsgGenericStreamTests.testRecvmsgOverflowc C r� r9 r� rG r r r �_testRecvmsgOverflow� r+ z.RecvmsgGenericStreamTests._testRecvmsgOverflowN)rO rP rQ r� r� r� r� r r r r r� � 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� )
rV r� rr r� r� r r� r� r� r� rT r� r` r� r r r �testRecvmsgBadArgs� s0 �
� � � � � z RecvmsgTests.testRecvmsgBadArgsc C r� r9 r� rG r r r �_testRecvmsgBadArgs� r+ z RecvmsgTests._testRecvmsgBadArgsN)rO rP rQ 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� rV r� rr rd r� r� r� rT r� r` �rH rg �nbytesr� r r� r r r �testRecvmsgIntoBadArgs� s@
� �
� � � �
�
� z'RecvmsgIntoTests.testRecvmsgIntoBadArgsc C r� r9 r� rG r r r �_testRecvmsgIntoBadArgs� r+ z(RecvmsgIntoTests._testRecvmsgIntoBadArgsc C sx t tt��}| j�dd� |fD ��\}}}}| �|tt�� | �|t t�� | �|| j� | �|g � | j|dd� d S )Nc s r3 r9 r r; r r r r4 � r5 z<RecvmsgIntoTests.testRecvmsgIntoGenerator.<locals>.<genexpr>Tr� ) r� r r� rr rd r� rT r� r` r� r r r �testRecvmsgIntoGenerator� s � z)RecvmsgIntoTests.testRecvmsgIntoGeneratorc C r� r9 r� rG r r r �_testRecvmsgIntoGenerator� r+ z*RecvmsgIntoTests._testRecvmsgIntoGeneratorc C sx t � ddgtt� �}| j�|g�\}}}}| �|tt�� | �|�� t� | �|| j� | �|g � | j |dd� d S )Nr� r Tr� )
r� r r� rr rd r� �tobytesrT r� r` r� r r r �testRecvmsgIntoArray
s z%RecvmsgIntoTests.testRecvmsgIntoArrayc C r� r9 r� rG r r r �_testRecvmsgIntoArray
r+ 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� rr rd �
memoryviewr� r rT r� r` )rH Zb1Zb2Zb3r� r� r r� r r r �testRecvmsgIntoScatter
s � z'RecvmsgIntoTests.testRecvmsgIntoScatterc C r' )Nr� )rJ rG r r r �_testRecvmsgIntoScatter
r+ z(RecvmsgIntoTests._testRecvmsgIntoScatterN) rO rP rQ r� r� r� r� r� r� r� r� 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� rf rV r� r� �maxsize)rH �toobigrh �n�retr 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� re r� rf rV r� r� r� )rH r� rh Zlastr� r� r r r �testCMSG_SPACEB
s
z CmsgMacroTests.testCMSG_SPACE) rO rP rQ r� r� r� �minr� r8 r r� r� 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 r9 ) rD r� Zmkstemprd r� r r �writer9 �encode�append)rH r� �fdsr? ri r r r r �newFDsc
s zSCMRightsTest.newFDsc C sB t |�D ]\}}t�|dtj� | �t�|d�t|��� � qd S )Nr r� )� enumerater� �lseek�SEEK_SETr� r# r9 r )rH r r� ri r r r �checkFDso
s �zSCMRightsTest.checkFDsc C s | � | j|� d S r9 )rd �closeRecvmsgFDsrQ r r r rN v
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 rf �
SCM_RIGHTSr� � frombytesr r� r� r )rH Zrecvmsg_result�
cmsg_level� cmsg_type� cmsg_datar ri r r r r y
s
� ��z SCMRightsTest.closeRecvmsgFDsc
C s8 | � | �tgtjtjt�d| �|��fg�tt�� d S )Nr? ) r� rM r� r rf r r� r r )rH r� r r r �createAndSendFDs�
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� rT r� r` r rV rC r� rf r r� �tuplerf r r� �
SIZEOF_INTr
r )rH ZnumfdsrO �maxcmsgs� ignoreflagsr� r� r r� r �itemr r r
r r r �checkRecvmsgFDs�
s& �
z SCMRightsTest.checkRecvmsgFDsc C � | � d| �| jtt�d�� d S )Nr r� �r rP rr r r� rG r r r �testFDPassSimple�
�
�z SCMRightsTest.testFDPassSimplec
C s< | � | �tgtjtjt�d| �d���� fg�t t�� d S )Nr? r )
r� rM r� r rf r r� r r� r rG r r r �_testFDPassSimple�
s ���z SCMRightsTest._testFDPassSimplec C r )Nr^ r� r rG r r r �testMultipleFDPass�
r z SCMRightsTest.testMultipleFDPassc C r' �Nr^ �r rG r r r �_testMultipleFDPass�
r+ z!SCMRightsTest._testMultipleFDPassr� c C s* | � d| �| jtt�t�dt ��� d S r )r rP rr r r� r r� r rG r r r �testFDPassCMSG_SPACE�
s
��z"SCMRightsTest.testFDPassCMSG_SPACEc C r' r r rG r r r �_testFDPassCMSG_SPACE�
rm z#SCMRightsTest._testFDPassCMSG_SPACEc
C s0 | j d| �| jtt�t�dt ��tjd� d S )Nr r^ �r ) r rP rr r r� r r� r rV rG r r r �testFDPassCMSG_LEN�
s �
�z SCMRightsTest.testFDPassCMSG_LENc C r' r
r rG r r r �_testFDPassCMSG_LEN�
r+ z!SCMRightsTest._testFDPassCMSG_LENrp zskipping, see issue #12958zskipping, see issue #22397c C s$ | j d| �| jtt�d�dd� d S )Nr� r� )r r rG r r r �testFDPassSeparate�
s
�z SCMRightsTest.testFDPassSeparatec C �X | � d�\}}| �| �tgtjtjt�d|g�ftjtjt�d|g�fg�tt�� d S �Nr� r? � r r� rM r� r rf r r� r �rH 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 rP rr r r� r r� r r� rV )rH Znum_fdsr r r �testFDPassSeparateMinSpace�
s ��
�z(SCMRightsTest.testFDPassSeparateMinSpacec C r% r& r' r( r r r �_testFDPassSeparateMinSpace�
r+ z)SCMRightsTest._testFDPassSeparateMinSpacec
C sf z | � |g|�}W n ty( } z| �|jt� | � |g�}W Y d }~nd }~ww | �|t|�� d S r9 )rM r rC r� r� r� r )rH 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� r! )r rP rr r r� r rV rG r r r �testFDPassEmpty s
�
�z SCMRightsTest.testFDPassEmptyc C s | � ttjtjdfg� d S �Nr! )r. r� r rf r rG 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 )rP rr r r� r� rT r� r` r rV rf rf r �
assertLessr �rH r� r� r r� r r r
r r r �testFDPassPartialInt# s � �z"SCMRightsTest.testFDPassPartialIntc C s4 | � ttjtjt�d| jg��� d d� fg� d S )Nr? r� )r. r� r rf r r� �badfdr� rG r r r �_testFDPassPartialInt0 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? )rP rr r r� r� rT r� r` r rV rf r� rf r r
r� r ) rH r� r� r r� r r r r
r r r � testFDPassPartialIntInMiddle7 s �
�z*SCMRightsTest.testFDPassPartialIntInMiddlec
C sZ | � d�\}}| �ttjtjt�d|| jg��� d d� ftjtjt�d|g�fg� d S )Nr� r? r� ) r r. r� r rf r r� r5 r� r( r r r � _testFDPassPartialIntInMiddleJ s � ���z+SCMRightsTest._testFDPassPartialIntInMiddlec C sJ |\}}}}| � |t� | �|| j� | � |g � | j|dtj|d� d S )NT�r\ r] r_ )r� r� rT r� r` r rV )rH rO r r� r� r r� r r r �checkTruncatedHeaderV s
�z"SCMRightsTest.checkTruncatedHeaderc C s" | j | �| jtt��tjd� d S �Nr! �r: rP rr r r� r rV rG r r r �testCmsgTruncNoBufSize` s
�z$SCMRightsTest.testCmsgTruncNoBufSizec C r' r
r rG r r r �_testCmsgTruncNoBufSizei r+ z%SCMRightsTest._testCmsgTruncNoBufSizec C s$ | j | �| jtt�d�tjd� d S )Nr r! r< rG r r r �testCmsgTrunc0l s
�z SCMRightsTest.testCmsgTrunc0c C r' r
r rG r r r �_testCmsgTrunc0q r+ z SCMRightsTest._testCmsgTrunc0c C s | � | �| jtt�d�� d S r
)r: rP rr r r� rG r r r �testCmsgTrunc1w s z SCMRightsTest.testCmsgTrunc1c C r' r
r rG r r r �_testCmsgTrunc1z r+ z SCMRightsTest._testCmsgTrunc1c C s" | � | �| jtt�td �� d S r )r: rP rr r r� r rG r r r �testCmsgTrunc2Int} s
�z SCMRightsTest.testCmsgTrunc2Intc C r' r
r rG r r r �_testCmsgTrunc2Int� r+ z SCMRightsTest._testCmsgTrunc2Intc C s( | � | �| jtt�t�d�d �� d S r )r: rP rr r r� r r� rG r r r �testCmsgTruncLen0Minus1� s
�z%SCMRightsTest.testCmsgTruncLen0Minus1c C r' r
r rG r r r �_testCmsgTruncLen0Minus1� r+ 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? )rP rr r r� r� rT r� r` r rV rf r re rf r� r
r� r ) rH �ancbuf�maxdataZmindatar� r� r r� r r r
r r r r �checkTruncatedArray� s& �
�z!SCMRightsTest.checkTruncatedArrayc C s | j t�d�dd� d S )Nr �rH rI �rJ r r� rG r r r �testCmsgTruncLen0� � z SCMRightsTest.testCmsgTruncLen0c C r' r
r rG r r r �_testCmsgTruncLen0� r+ z SCMRightsTest._testCmsgTruncLen0c C s | j t�d�d dd� d S )Nr r rK rL rG r r r �testCmsgTruncLen0Plus1� r� z$SCMRightsTest.testCmsgTruncLen0Plus1c C r' r r rG r r r �_testCmsgTruncLen0Plus1� r+ z%SCMRightsTest._testCmsgTruncLen0Plus1c C s | j t�t�td� d S )NrK �rJ r r� r rG r r r �testCmsgTruncLen1� s
�z SCMRightsTest.testCmsgTruncLen1c C r' r r rG r r r �_testCmsgTruncLen1� r+ z SCMRightsTest._testCmsgTruncLen1c C s( | j t�dt �d dt d d� d S )Nr� r rK rR rG r r r �testCmsgTruncLen2Minus1� s
�z%SCMRightsTest.testCmsgTruncLen2Minus1c C r' r r rG r r r �_testCmsgTruncLen2Minus1� r+ z&SCMRightsTest._testCmsgTruncLen2Minus1N)r r �r )9rO rP rQ r5 r r rN r r r r r r r r8 r r r$ r r" r# r� r� r� r� r
r$ r* r, r- r. r/ r1 r4 r6 r7 r8 r: r= r> r? r@ rA rB rC rD rE rF rJ rM rO rP rQ rS rT rU rV r r r r r� [
s�
r� 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 r9 )rI )rH r� �dr r r
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 Tr r r? )rr r� r rj ru rH r} rP r r� r� rT r� r` rV rC r rt r� r r� r
re rf ) rH �
ancbufsize�maxhopr r� r� r r� r r r
r� r r r �
checkHopLimit� s.
�
� �
z"RFC3542AncillaryTest.checkHopLimitru rt c C � | j dd� d S �Nr� �r[ )r] rG r r r �testRecvHopLimit� rU z%RFC3542AncillaryTest.testRecvHopLimitc C �$ | � | jj| jd�� | �t� d S r� �rB rH r� rb rJ r� rG r r r �_testRecvHopLimit s z&RFC3542AncillaryTest._testRecvHopLimitr� c C s | j t�t�d� d S )Nr` )r] r r� r rG r r r �testRecvHopLimitCMSG_SPACE � z/RFC3542AncillaryTest.testRecvHopLimitCMSG_SPACEc C rb r� rc rG r r r �_testRecvHopLimitCMSG_SPACE
� z0RFC3542AncillaryTest._testRecvHopLimitCMSG_SPACEr� c C � | j d| jd� d S �Nr� )r[ r\ )r] � hop_limitrG r r r �testSetHopLimit s z$RFC3542AncillaryTest.testSetHopLimitc C sL | � | jj| jd�� | �| �tgtjtj t
�
d| j g�fg�t t�� d S �Nr� r? )
rB rH r� rb r� rM r� r rj rt r� rk r rG 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 Tr r� r? r r� )rr r� r rj ru rv rH r} rP r r� r� rT r� r` rV rZ rs r r� r
re rf rt ) rH r[ r\ r r� r� r r� ZancmapZtcdatar� Zhldatar r r � checkTrafficClassAndHopLimit' s:
�
�
� �
z1RFC3542AncillaryTest.checkTrafficClassAndHopLimitrv rs c C r^ r_ )ro rG r r r � testRecvTrafficClassAndHopLimitK s z4RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitc C rb r� rc rG r r r � _testRecvTrafficClassAndHopLimitQ rh z5RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitc C s | j t�t�d d� d S �Nr� r` )ro r r� r rG r r r �)testRecvTrafficClassAndHopLimitCMSG_SPACEV s
�z>RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitCMSG_SPACEc C rb r� rc rG r r r �*_testRecvTrafficClassAndHopLimitCMSG_SPACE^ rh z?RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitCMSG_SPACEc C ri rj �ro rk rG r r r � testSetTrafficClassAndHopLimitc s
�z3RFC3542AncillaryTest.testSetTrafficClassAndHopLimitc C sd | � | jj| jd�� | �| �tgtjtj t
�
d| j g�ftjtj t
�
d| j
g�fg�tt�� d S rm )rB rH r� rb r� rM r� r rj rs r� �
traffic_classrt rk r rG r r r � _testSetTrafficClassAndHopLimitl s �����z4RFC3542AncillaryTest._testSetTrafficClassAndHopLimitc C ri rj ru rG r r r �testOddCmsgSizew 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� )rB rH r� rb rM r� r rj rs r� rw r� rt rk r rC r� r� r� r )rH r� r� r r r �_testOddCmsgSize� 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 Tr9 )rr r� r rj ru rH r} rP r r� r� rT r� r` rV )rH r[ r r^ r� r� r r� r r r � checkHopLimitTruncatedHeader� s
�
��
�z1RFC3542AncillaryTest.checkHopLimitTruncatedHeaderc C s | j d tjd� d S )N�r[ r �r{ r rV rG r r r r= � s
�z+RFC3542AncillaryTest.testCmsgTruncNoBufSizec C rb r� rc rG r r r r> � rh z,RFC3542AncillaryTest._testCmsgTruncNoBufSizec C s | j dtjd� d S )Nr r| r} rG r r r �testSingleCmsgTrunc0� s
�z)RFC3542AncillaryTest.testSingleCmsgTrunc0c C rb r� rc rG r r r �_testSingleCmsgTrunc0� rh z*RFC3542AncillaryTest._testSingleCmsgTrunc0c C r^ )Nr r` )r{ rG r r r �testSingleCmsgTrunc1� � z)RFC3542AncillaryTest.testSingleCmsgTrunc1c C rb r� rc rG r r r �_testSingleCmsgTrunc1� rh z*RFC3542AncillaryTest._testSingleCmsgTrunc1c C s | j dt d� d S rr )r{ r rG r r r �testSingleCmsgTrunc2Int� � z,RFC3542AncillaryTest.testSingleCmsgTrunc2Intc C rb r� rc rG r r r �_testSingleCmsgTrunc2Int� rh z-RFC3542AncillaryTest._testSingleCmsgTrunc2Intc C s | j t�d�d d� d S )Nr r r` )r{ r r� rG r r r � testSingleCmsgTruncLen0Minus1� � z2RFC3542AncillaryTest.testSingleCmsgTruncLen0Minus1c C rb r� rc rG r r r � _testSingleCmsgTruncLen0Minus1� rh 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 TrG r )rr r� r rj ru rH r} rP r r� r� r r� rT r� r` rV rf rt r2 r3 r r r �testSingleCmsgTruncInData� s"
�
� �z.RFC3542AncillaryTest.testSingleCmsgTruncInDatac C rb r� rc rG r r r �_testSingleCmsgTruncInData� rh 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 Tr9 r r? r� )rr r� r rj ru rv rH r} rP r r� r� rT r� r` rV rG rs rt r r� r
re rf ) rH r[ r r� r� r r� r r r
r� r r r �checkTruncatedSecondHeader s0
�
�
� �
z/RFC3542AncillaryTest.checkTruncatedSecondHeaderc C s | j t�t�tjd� d S r; )r� r r� r rV rG r r r �testSecondCmsgTrunc0 s
�z)RFC3542AncillaryTest.testSecondCmsgTrunc0c C rb r� rc rG r r r �_testSecondCmsgTrunc0& rh z*RFC3542AncillaryTest._testSecondCmsgTrunc0c C s | � t�t�d � d S r
�r� r r� r rG r r r �testSecondCmsgTrunc1+ r� z)RFC3542AncillaryTest.testSecondCmsgTrunc1c C rb r� rc rG r r r �_testSecondCmsgTrunc10 rh z*RFC3542AncillaryTest._testSecondCmsgTrunc1c C s | � t�t�dt � d S r r� rG r r r �testSecondCmsgTrunc2Int5 s
�z,RFC3542AncillaryTest.testSecondCmsgTrunc2Intc C rb r� rc rG r r r �_testSecondCmsgTrunc2Int; rh z-RFC3542AncillaryTest._testSecondCmsgTrunc2Intc C s" | � t�t�t�d� d � d S r )r� r r� r r� rG r r r � testSecondCmsgTruncLen0Minus1@ s
�
�z2RFC3542AncillaryTest.testSecondCmsgTruncLen0Minus1c C rb r� rc rG r r r � _testSecondCmsgTruncLen0Minus1F rh 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 TrG r r? r� ) rr r� r rj ru rv rH r} rP r r� r� r r� r� rT r� r` rV rs rt �pop�remover� r
re rf r2 )
rH r� r� r r� Z
cmsg_typesr r r
r� r r r �testSecondCmsgTruncInDataK s:
�
�
�
z.RFC3542AncillaryTest.testSecondCmsgTruncInDatac C rb r� rc rG r r r �_testSecondCmsgTruncInDatap rh z/RFC3542AncillaryTest._testSecondCmsgTruncInDataN)r� r rW )1rO rP rQ rk rw rZ r] r8 r ra r$ rd re rg rl rn ro rp rq rs rt rv rx ry 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 rX � s�
�$�
�
�
�
�
�
�
�
�
#rX c @ � e Zd ZdS )�SendrecvmsgUDPTestBaseN�rO rP rQ r r r r r� x � r� r� c @ r� )�SendmsgUDPTestNr� r r r r r� } � r� r� c @ r� )�RecvmsgUDPTestNr� r r r r r� � r� r� rd c @ r� )�RecvmsgIntoUDPTestNr� r r r r r� � r� r� c @ rT )�SendrecvmsgUDP6TestBasec C � | � |d d� |d d� � d S rA rR rS r r r rT � � z+SendrecvmsgUDP6TestBase.checkRecvmsgAddressN�rO rP rQ rT r r r r r� � � r� rL r r% c @ r� )�SendmsgUDP6TestNr� r r r r r� � � r� c @ r� )�RecvmsgUDP6TestNr� r r r r r� � r� r� c @ r� )�RecvmsgIntoUDP6TestNr� r r r r r� � r� r� rj c @ r� )� RecvmsgRFC3542AncillaryUDP6TestNr� r r r r r� � � 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 @ rT )�SendrecvmsgUDPLITE6TestBasec C r� rA rR rS r r r rT � 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 AIXrB r6 r c @ r� )�SendmsgSCTPStreamTestNr� r r r r r� " r� r� c rv )�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)rX r� r� r r� �ENOTCONNrr �rH r� r` r r r� . � ��z$RecvmsgSCTPStreamTest.testRecvmsgEOF�rO rP rQ r� rh r r r` r r� ( � r� c rv )�RecvmsgIntoSCTPStreamTestc
r� r� )rX r� r� r r� r� rr r� r` r r r� < r� z(RecvmsgIntoSCTPStreamTest.testRecvmsgEOFr� r r r` r r� 6 r� r� c @ r� )� SendrecvmsgUnixStreamTestBaseNr� r r r r r� E r� r� r c @ r� )�SendmsgUnixStreamTestNr� r r r r r� I r� r� c @ r� )�RecvmsgUnixStreamTestNr� r r r r r� N r� r� c @ r� )�RecvmsgIntoUnixStreamTestNr� r r r r r� T r� r� rf r c @ r� )�RecvmsgSCMRightsStreamTestNr� r r r r r� Z r� r� c @ r� )� RecvmsgIntoSCMRightsStreamTestNr� r r r r r� _ 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 r� r )Zsignumr� r r r r s s z.InterruptedTimeoutBase.setUp.<locals>.<lambda>)rX rI r r rd )rH Zorig_alrm_handlerr` r r rI p s
�z InterruptedTimeoutBase.setUp� setitimerg�������?c C s t �t j|� d S r9 )r r� � ITIMER_REAL�rH Zsecondsr r r �setAlarm r� z InterruptedTimeoutBase.setAlarmr� c C s t �|� d S r9 )r r r� r r r r� � r+ ) rO rP rQ rI r ra r< r r �
alarm_timer� rh r r r` r r� k s
r� � siginterruptr 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 rx r9 )rX rI rC r� r< rG r` r r rI � ry 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� rV r �rH r� r^ r_ rb r r r �checkInterruptedRecv� s ��z/InterruptedRecvTimeoutTest.checkInterruptedRecvc C � | � | jjd� d S r� )r� rC r� rG r r r �testInterruptedRecvTimeout� r� z5InterruptedRecvTimeoutTest.testInterruptedRecvTimeoutc C � | � | jjtd�� d S r� )r� rC � recv_intor� rG r r r � testInterruptedRecvIntoTimeout� rN z9InterruptedRecvTimeoutTest.testInterruptedRecvIntoTimeoutc C r� r� )r� rC r� rG r r r � testInterruptedRecvfromTimeout� r� z9InterruptedRecvTimeoutTest.testInterruptedRecvfromTimeoutc C r� r� )r� rC �
recvfrom_intor� rG r r r �"testInterruptedRecvfromIntoTimeout� rN z=InterruptedRecvTimeoutTest.testInterruptedRecvfromIntoTimeoutr� c C r� r� )r� rC r� rG r r r � testInterruptedRecvmsgTimeout� r� z8InterruptedRecvTimeoutTest.testInterruptedRecvmsgTimeoutrd c C s | � | jjtd�g� d S r� )r� rC rd r� rG r r r �!testInterruptedRecvmsgIntoTimeout� r� z<InterruptedRecvTimeoutTest.testInterruptedRecvmsgIntoTimeout)rO rP rQ rI r� r� r� r� r� r8 r r� r� rh 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)rX rI r� r� rd r rZ �Thread� doConnect�startrC r� r� r6 r� r< )rH Z
cli_threadr� r` r r rI � s
z InterruptedSendTimeoutTest.setUpc C s | j �| j� d S r9 )r� r� r� rG r r r r� � r� z$InterruptedSendTimeoutTest.doConnectc O sX z%| � t��} | �| j� ||i |�� q1 sw Y W | �d� d S | �d� w �NTr )rV r r� r� r� r r r �checkInterruptedSend� s ��z/InterruptedSendTimeoutTest.checkInterruptedSendr� r� c C r� �Nr� )r� r� r� rG r r r �testInterruptedSendTimeout� r� z5InterruptedSendTimeoutTest.testInterruptedSendTimeoutc C s | � | jjd| j� d S r� )r� r� r_ r� rG r r r � testInterruptedSendtoTimeout� s �z7InterruptedSendTimeoutTest.testInterruptedSendtoTimeoutr� c C s | � | jjdg� d S r� )r� r� r� rG r r r � testInterruptedSendmsgTimeout� rf z8InterruptedSendTimeoutTest.testInterruptedSendmsgTimeout)rO rP rQ rI r� r� r Zrequires_mac_verr� r� r8 r r� rh 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 r! )rC r� r r� r� r� r� )rH r� r� Zsdr# r� r> r r r � testClose� s zTCPCloserTest.testClosec C s | j �t| jf� t�d� d S �Nr� )r� r� r� rE r� r� rG r r r �
_testClose rb zTCPCloserTest._testCloseN)rO rP rQ r� r� r r r r r� � �
r� c @ sN e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dS )�BasicSocketPairTestr� c C r� r� )r� rY r� r r r rY r� z BasicSocketPairTest.__init__c C s\ | � |tj� ttd�r| �|jtj� n| �|jtj� | �|jtj� | �|j d� d S )Nr r )
rC r r r� rL r rB r# r6 rM r r r r �_check_defaults
s
z#BasicSocketPairTest._check_defaultsc C � | � | j� d S r9 )r� r� rG r r r �
_testDefaults r z!BasicSocketPairTest._testDefaultsc C r� r9 )r� rC rG r r r � testDefaults r z BasicSocketPairTest.testDefaultsc C r r� �rC r� r� r� rY r r r r r. z BasicSocketPairTest.testRecvc C r r9 �r� r� r� rG r r r r r z BasicSocketPairTest._testRecvc C r r9 �rC r� r� rG r r r �testSend r z BasicSocketPairTest.testSendc C r r� �r� r� r� r� rY r r r � _testSend# r. z BasicSocketPairTest._testSendNr� ) rO rP rQ rY r� r� r� r r r� r� r r r r r� s
r� c s� e Zd Zdd� Z� fdd�Z� fdd�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z e �
eed� p9eed� d�e �ejd�dd� ��Zdd� Zdd � Zd d � Z� ZS ) �PurePythonSocketPairTestc C s. zt �t j�W S ty t �t j� Y S w r9 )r r� r r rB rG r r r r� - s
�z#PurePythonSocketPairTest.socketpairc s0 t td�rtj| _tjt_nd | _t� �� d S )Nr� )r rA r r� �_orig_spZ_fallback_socketpairrX rI rG r` r r rI 6 s
z PurePythonSocketPairTest.setUpc s$ t � �� | jd ur| jt_d S d S r9 )rX rL r� r r� rG r` r r rL A s
�z!PurePythonSocketPairTest.tearDownc C r r� r� rY r r r � test_recvG r. z"PurePythonSocketPairTest.test_recvc C r r9 r� rG r r r �
_test_recvK r z#PurePythonSocketPairTest._test_recvc C r r9 r� rG r r r � test_sendN r z"PurePythonSocketPairTest.test_sendc C r r� r� rY r r r �
_test_sendQ r. z#PurePythonSocketPairTest._test_sendc C �$ t �t j�\}}|�� |�� d S r9 )r r� rB r �rH r� rI r r r � test_ipv4U s z"PurePythonSocketPairTest.test_ipv4c C r� r9 r rG r r r �
_test_ipv4Z r� z#PurePythonSocketPairTest._test_ipv4rj � IPV6_V6ONLY� IPV6_V6ONLY option not supportedrr c C r r9 )r r� r r r r r r � test_ipv6] s z"PurePythonSocketPairTest.test_ipv6c C r� r9 r rG r r r �
_test_ipv6f r� z#PurePythonSocketPairTest._test_ipv6c s� t j j�d � � �fdd�}d }}z:|t j _| �t�� t �� \}}W d � n1 s,w Y W �t j _� r<� �� |rB|�� |rJ|�� d S d S �t j _� rV� �� |r\|�� |rc|�� w w )Nc sf �| �}� d u r1t � t jt j�� � �d� z � �|d d� � W n ttfy+ Y nw � �d� |S )NFr� T)r rB r6 rb r� �BlockingIOError�InterruptedError)rH ra �Z inject_sockZorig_getsocknamer r �inject_getsocketnamem s
�
z[PurePythonSocketPairTest.test_injected_authentication_failure.<locals>.inject_getsocketname)r r� rV r r� r )rH r
�sock1�sock2r r r �$test_injected_authentication_failurei s4 �� ��
�z=PurePythonSocketPairTest.test_injected_authentication_failurec C r� r9 r rG r r r �%_test_injected_authentication_failure� s z>PurePythonSocketPairTest._test_injected_authentication_failure)rO rP rQ r� rI rL r� r� r r r r r� r� r rA r� r r
r r r r rh 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� )rZ r� �eventr� rY r� r r r rY � ry z NonBlockingTCPTests.__init__c C sl | � | j�� |� |dk}| � |�� |� td ur4|d u }t�|tjtj�}| � t|tj@ � |� d S d S )N� ) r� rC r� � getblockingr �F_GETFLr� �
O_NONBLOCK�bool)rH r r< ZblockingZ 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 )NTFr r r� )rC rb r r� rG r r r �testSetBlocking� s z#NonBlockingTCPTests.testSetBlockingc C r� r9 r rG 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_MAXFr r ) r� r/ Z ULONG_MAXrr rC rb r� r� r� r0 r r r �testSetBlocking_overflow� s
z,NonBlockingTCPTests.testSetBlocking_overflowra z test needs socket.SOCK_NONBLOCKr� � � c C s4 | j �� t�tjtjtjB �| _ | �| j d� d S r� )rC r r rB r6 ra r rG r r r �testInitNonBlocking� s
�z'NonBlockingTCPTests.testInitNonBlockingc C r� r9 r rG r r r �_testInitNonBlocking� r� z(NonBlockingTCPTests._testInitNonBlockingc C sb t d ��# | j�d� | j�� \}}| �|j� | �|�� � W d � d S 1 s*w Y d S )Nr� )r= rC r� r� rd r r� r� r� r r r �testInheritFlagsBlocking� s
"�z,NonBlockingTCPTests.testInheritFlagsBlockingc C � | j �t| jf� d S r9 �r� r� r� rE rG r r r �_testInheritFlagsBlocking� 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= rC r� r� rd r r� r� )rH Zdefault_timeoutr� r� r r r �testInheritFlagsTimeout� s
"�z+NonBlockingTCPTests.testInheritFlagsTimeoutc C r# r9 r$ rG 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 )NFr� z'Error trying to do accept after select.)rC rb r� � monotonicrV r
r� r2 r r} r� r � LONG_TIMEOUTrR rd r r� r� )rH Z
start_timer� r� Zdtr# r� r> r r r �
testAccept� s �
z NonBlockingTCPTests.testAcceptc C s | j �� | j�t| jf� d S r9 )r r� r� r� r� rE rG r r r � _testAccept 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.)rC r� rd r rb rV r
r� r r� r r} r� r r) rR r� )rH r� r� r� r# r� r> r r r r s
�
z NonBlockingTCPTests.testRecvc C s, | j �t| jf� | j�� | j �t� d S r9 )r� r� r� rE r r� r! r� rG r r r r ' s
z NonBlockingTCPTests._testRecvNr� )rO rP rQ rY r r r r r� r Z_testSetBlocking_overflowr� r� r r �requires_linux_versionr r! r" r% r& r' r* r+ r r r r r r r � s,
� r 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 r: r� c C r� r� r r� r r r rY H 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 )rZ r� �r- r? r r r r/ L s �z1FileObjectClassTestCase.setUp.<locals>.<listcomp>r^ �r> �errors�newline)rD �evt1�evt2�
serv_finished� cli_finishedr� rI r� r. � read_moders r> r2 r3 � read_filerG r r r rI K s �
�z FileObjectClassTestCase.setUpc C �6 | j �� | j�� | �| jj� d | _t�| � d S r9 )r6 r} r9 r rB �closedr� rL rG r r r rL U �
z FileObjectClassTestCase.tearDownc C s0 t �| � | jj| j| j| j| j| jd�| _ d S )Nr1 )
r� r� r� r. �
write_moders r> r2 r3 �
write_filerG r r r r� \ s
�z#FileObjectClassTestCase.clientSetUpc C r: r9 )r7 r} r> r rB r; r� r� rG r r r r� d r< 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 r z!cannot read from timed out object)
r� r� r9 r# rV r# r rG r9 r` )rH �ctxr r r �testReadAfterTimeoutk s �z,FileObjectClassTestCase.testReadAfterTimeoutc C s. | j �| jdd� � | j �� | j�� d S )Nr r )r> r� � write_msg�flushr6 r� rG r r r �_testReadAfterTimeoutw s
z-FileObjectClassTestCase._testReadAfterTimeoutc C s< | j �t| j�d �}| j �d�}|| }| �|| j� d S r� )r9 r# r �read_msgr� )rH � first_segZ
second_segr� r r r �
testSmallRead| s z%FileObjectClassTestCase.testSmallReadc C � | j �| j� | j �� d S r9 �r> r� rA rB rG r r r �_testSmallRead� r� z&FileObjectClassTestCase._testSmallReadc C � | j �� }| �|| j� d S r9 )r9 r# r� rD rY r r r � testFullRead� �
z$FileObjectClassTestCase.testFullReadc C rG r9 )r> r� rA r rG r r r �
_testFullRead� r� z%FileObjectClassTestCase._testFullReadc C s< t | j�� } | j�d�}|sn||7 }q| �|| j� d S r
)r# rD r9 r# r� )rH rg �charr r r �testUnbufferedRead� s �z*FileObjectClassTestCase.testUnbufferedReadc C rG r9 rH rG r r r �_testUnbufferedRead� r� z+FileObjectClassTestCase._testUnbufferedReadc C rJ r9 )r9 �readliner� rD �rH �liner r r � testReadline� rL z$FileObjectClassTestCase.testReadlinec C rG r9 rH rG r r r �
_testReadline� r� z%FileObjectClassTestCase._testReadlinec C s& | j �� | j�� }| �|| j� d S r9 )r� r r9 r# r� rD rY r r r �testCloseAfterMakefile� s
z.FileObjectClassTestCase.testCloseAfterMakefilec C rG r9 rH rG r r r �_testCloseAfterMakefile� r� z/FileObjectClassTestCase._testCloseAfterMakefilec C s@ | j �� | j�tt��}t| jt�r|� � }| �
|| j� d S r9 ) r9 r r� r� r r� r" rD r9 �decoder� rY r r r � testMakefileAfterMakefileClose� s
z6FileObjectClassTestCase.testMakefileAfterMakefileClosec C rG r9 rH rG r r r � _testMakefileAfterMakefileClose� r� z7FileObjectClassTestCase._testMakefileAfterMakefileClosec C � | � | jj � d S r9 )rB r9 r; rG r r r �testClosedAttr� r� z&FileObjectClassTestCase.testClosedAttrc C r[ r9 )rB r> r; rG r r r �_testClosedAttr� r� z'FileObjectClassTestCase._testClosedAttrc C �, | � | jj| j� | � | jj| j�� � d S r9 )r� r9 r< r8 r. r� rH rG r r r �testAttributes� � z&FileObjectClassTestCase.testAttributesc C r^ r9 )r� r> r< r= r. r� rH rG r r r �_testAttributes� r` z'FileObjectClassTestCase._testAttributesc C s8 | j �� | �t| j j� | j�� | �t| jj� d S r9 )r9 r rV r� rH r� r r� rG r r r �
testRealClose� s
z%FileObjectClassTestCase.testRealClosec C r� r9 r rG r r r �_testRealClose� r� z&FileObjectClassTestCase._testRealCloser� )&rO rP rQ rg rs r> r2 r3 r8 r� rD r= rA rY rI rL r� r� r@ rC rF rI rK rM rO rP rT rU rV rW rY rZ r\ r] r_ ra rb rc r r r r r- 2 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. )r9 rQ r� rA r� r. rR r r r �testUnbufferedReadline� s
z8UnbufferedFileObjectClassTestCase.testUnbufferedReadlinec C s2 | j �d| j � | j �d| j � | j �� d S )Nre rf rH rG r r r �_testUnbufferedReadline� s z9UnbufferedFileObjectClassTestCase._testUnbufferedReadlinec C sD | j �� | j �d�}| �|| j� | j�� | �t| j jd� d S r� )r� r r� r� rD r9 rV r rY r r r �testMakefileClose� s
z3UnbufferedFileObjectClassTestCase.testMakefileClosec C rG r9 rH rG r r r �_testMakefileClose� r� z4UnbufferedFileObjectClassTestCase._testMakefileClosec C s6 t �| j�}| j�� t �| j�}| �|d |� d S r
)r� r� r� r9 r r� )rH Zrefcount_beforeZrefcount_afterr r r � testMakefileCloseSocketDestroy� s
z@UnbufferedFileObjectClassTestCase.testMakefileCloseSocketDestroyc C r� r9 r rG r r r � _testMakefileCloseSocketDestroy 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� r r� g �?r� r )r� rb r� r9 �readintor� r# r rD r4 r} r5 r� r� r� )rH rE rg r� r� r r r �testSmallReadNonBlocking s
z:UnbufferedFileObjectClassTestCase.testSmallReadNonBlockingc C s> | j �d� | j�| j� | j�� | j�� | j�d� d S )Nr� � @) r4 r� r> r� rA rB r5 r} r6 rG r r r �_testSmallReadNonBlocking s
z;UnbufferedFileObjectClassTestCase._testSmallReadNonBlockingc C s& | j �d� | jr| �| j� d S d S )Nro )r7 r� � serv_skippedrr rG 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) rq r� rb r r" r> r� r� rD )rH ZBIGZLIMITr� r? r r r �_testWriteNonBlocking+ s
z7UnbufferedFileObjectClassTestCase._testWriteNonBlockingN)rO rP rQ rg rs rg rh ri rj rk rl rn rp rr rs r r r r rd � s rd c @ � e Zd ZdZdS )�#LineBufferedFileObjectClassTestCaser N�rO rP rQ rs r r r r ru A � ru c @ rt )�$SmallBufferedFileObjectClassTestCaser� Nrv r r r r rx F rw rx c @ s* e Zd ZdZdZe�d�ZdZeZ dZ
dS )�"UnicodeReadFileObjectClassTestCase�=Tests for socket.makefile() in text mode (rather than binary)r r r: r[ N� rO rP rQ rg r8 r� rX rD r= rA r3 r r r r ry K s
ry c @ s* e Zd ZdZdZeZdZe�d�Z dZ
dS )�#UnicodeWriteFileObjectClassTestCaserz r r9 r r[ N) rO rP rQ rg r8 r� rD r= rX rA r3 r r r r r| U s
r| c @ s0 e Zd ZdZdZe�d�ZdZe�d�Z dZ
dS )�'UnicodeReadWriteFileObjectClassTestCaserz r r r9 r[ Nr{ r r r r r} _ s
r} c @ r )�NetworkConnectionTestzProve network connection.c C s t �t| jf�| _| j| _d S r9 )r �create_connectionr� rE r� r� rG r r r r� l s z!NetworkConnectionTest.clientSetUpN)rO rP rQ rg r� r r r r r~ i r r~ c @ rt )�
BasicTCPTest2zLTests that NetworkConnection does not break existing TCP functionality.
N)rO rP rQ rg r r r r r� r s r� c @ sD e Zd ZG dd� dej�Zejdd� �Zdd� Zdd� Z d d
� Z
d S ) �NetworkConnectionNoServerc @ rT )z$NetworkConnectionNoServer.MockSocketc G r� )Nz timed out)r# rL r r r r� y r� z,NetworkConnectionNoServer.MockSocket.connectN)rO rP rQ r� r r r r �
MockSocketx 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� )rH 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 r9 )r r� r rB r6 rd r rV r r� r� r� r` r� Z ECONNREFUSED)rH rE r� rb 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 r9 ) r r� rV r r r r� Z get_socket_conn_refused_errsrG r` r� )rH rE rb 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� r# r r r
r� Z EAFNOSUPPORTrR r� r r r � test_create_connection_timeout� s"
��� ��"�z8NetworkConnectionNoServer.test_create_connection_timeoutN) rO rP rQ r r� �
contextlib�contextmanagerr� r� r� r� r r r r r� v 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 rY � r� z(NetworkConnectionAttributesTest.__init__c C s t �� | _d S r9 )r r� � source_portrG r r r r� � r+ z+NetworkConnectionAttributesTest.clientSetUpc C r� r9 r� rG r r r r� � r� z.NetworkConnectionAttributesTest.clientTearDownc C s | j �� \}}|�� d S r9 )rC 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� rE r ra r� rd r r� rL rG r r r � _testFamily� s
�z+NetworkConnectionAttributesTest._testFamilyc C sJ t jt| jftjd| jfd�| _| �| jj � | �
| j� � d | j� d S )Nr[ )r< Zsource_addressr ) r r r� rE r ra r� r� rd r r� r� rG 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 )NrW � rB r r: r; r r� rE r� rd r r� r� rG r r r �_testTimeoutDefault� 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� rG r r r �_testTimeoutNone� s
z0NetworkConnectionAttributesTest._testTimeoutNonec C s, t jt| jfdd�| _| �| j�� d� d S r� )r r r� rE r� r� r� rG r r r �_testTimeoutValueNamed� s z6NetworkConnectionAttributesTest._testTimeoutValueNamedc C s8 t �t| jfd�| _| �| jj� | �| j�� d� d S )Nr� ) r r r� rE r� rd r r� r� rG r r r �_testTimeoutValueNonamed� s z8NetworkConnectionAttributesTest._testTimeoutValueNonamedNr� )rO rP rQ rY 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 rY � r� z'NetworkConnectionBehaviourTest.__init__c C r� r9 r rG r r r r� r� z*NetworkConnectionBehaviourTest.clientSetUpc C r� r9 r� rG r r r r� r� z-NetworkConnectionBehaviourTest.clientTearDownc C s2 | j �� \}}| �|j� t�d� |�d� d S )Nr � done!)rC r� rd r r� r� r� r� r r r �testInsideTimeout
�
z0NetworkConnectionBehaviourTest.testInsideTimeoutc C s0 t �t| jf� | _}|�d�}| �|d� d S )Nr� r� )r r r� rE r� r� r� )rH 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� rE r� rV r# rG r r� r �_testOutsideTimeout s z2NetworkConnectionBehaviourTest._testOutsideTimeoutNr� )
rO rP rQ rY r� r� r� ZtestOutsideTimeoutr� r� r r r r r� � 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 r� )rC r� r� r rG r r �
raise_timeout r� z4TCPTimeoutTest.testTCPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (TCP)�rV r# �rH r� r rG 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 )NFr z%caught timeout instead of error (TCP)Tz!caught unexpected exception (TCP)z3accept() returned success when we did not expect it)rC r� r� r# rR r �rH �okr r r r �testTimeoutZero$ s �z TCPTimeoutTest.testTimeoutZeror 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 ) Nro c @ r� )z4TCPTimeoutTest.testInterruptedTimeout.<locals>.AlarmNr� r r r r �Alarm9 s r� c s � �r9 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)
rC r� r[ r r r r� r# rR r� �exc_info� traceback�
format_exc)rH r� r% r r r� r �testInterruptedTimeout2 s4
�
� ���"z%TCPTimeoutTest.testInterruptedTimeoutN)
rO rP rQ 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 �Nr� r� �rC r� r� r rG r r r� W r. z4UDPTimeoutTest.testUDPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (UDP)r� r� r rG r �testUDPTimeoutV 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 )NFr r� z%caught timeout instead of error (UDP)Tz!caught unexpected exception (UDP)�1recv() returned success when we did not expect it�rC r� r� r# rR r r� r r r r� ] � �z UDPTimeoutTest.testTimeoutZeroN)rO rP rQ r� r� r r r r r� T s r� c @ r@ )�UDPLITETimeoutTestc r� )Nc r� r� r� r rG r r r� p r. z<UDPLITETimeoutTest.testUDPLITETimeout.<locals>.raise_timeoutz.Error generating a timeout exception (UDPLITE)r� r� r rG r �testUDPLITETimeouto r� z%UDPLITETimeoutTest.testUDPLITETimeoutc C r� )NFr r� z)caught timeout instead of error (UDPLITE)Tz%caught unexpected exception (UDPLITE)r� r� r� r r r r� v r� z"UDPLITETimeoutTest.testTimeoutZeroN)rO rP rQ r� r� r r r r r� k s r� c @ r@ )�TestExceptionsc C sf | � ttt�� | � ttjt�� | � ttjt�� | � ttjt�� | �tj t� | �tjt
� d S r9 ) rB �
issubclassr r[ r rW rX r< ZassertIs�errorr# rG 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 rB r6 rH r rd re rV r rb )rH Zsock0r r r r �test_setblocking_invalidfd� s � "�z)TestExceptions.test_setblocking_invalidfdN)rO rP rQ r� r� r r r r r� � s r� r� �Linux specific testc @ rx )�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 r6 rp rF r� r� r� r� Z getpeername)rH r� �s1rw 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 r6 rp r� r� �rH 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 r6 rV r rp 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 r6 rp r� r� r rJ 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 r6 rp r� r� r� rJ r r r �testBytearrayName� s "�z,TestLinuxAbstractNamespace.testBytearrayNamec C s� d}t � t jt j��M}|�d� | �|�� |� t � t jt j�� }|�d� | �|�� |� | �|�� |�� � W d � n1 sBw Y W d � d S W d � d S 1 sZw Y d S )Ns
^ [0-9a-f]{5}r[ )r r r6 rp Z assertRegexr� r� )rH Zabstract_addressr� rw r r r � testAutobind� s
��"�z'TestLinuxAbstractNamespace.testAutobindN)
rO rP rQ r� r� r� r� r� r� r� r r r r r� � s r� ztest needs socket.AF_UNIXc @ sn e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z e �
ejdkd�dd� �ZdS )�TestUnixDomainc C r� r9 )r r r6 r rG r r r rI � r� zTestUnixDomain.setUpc C r� r9 )r r rG r r r rL � r+ 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 rr r= r� �getfilesystemencoding�rH r r r r �encoded� 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 r9 rr r= )rH r r r� r r r rp � s ����zTestUnixDomain.bindc C s | � | j�� d� d S )N)r[ N)rG r r� rG r r r r� � r zTestUnixDomain.testUnboundc C s@ t j�tj�}| �| j|� | �tj|� | � | j�
� |� d S r9 ) r� r �abspathr �TESTFNrp r rd r r� r� r� r r r � testStrAddr� s zTestUnixDomain.testStrAddrc C sF t j�tj�}| �| j| �|�� | �tj |� | �
| j� � |� d S r9 ) r� r r� r r� rp r r� rd 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� rp r rX rd r r� r� )rH r r r r r �testSurrogateescapeBind 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_UNENCODABLErr r� r r� rp r rd r r� r� r� r r r �testUnencodableAddr s
z"TestUnixDomain.testUnencodableAddrr� r� c C r@ )Nr[ )rV r r rp rG r r r �testEmptyAddress rf z TestUnixDomain.testEmptyAddressN)rO rP rQ rI rL r� rp r� r� r� r� r� r� r� r� r� r� r r r r r� � 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� r r� r r r rY " r� zBufferIOTest.__init__c C sZ t � ddgtt� �}| j�|�}| �|tt�� |�� }|d tt�� }| �|t� d S �Nr� r )r� r r� r� r� r� r� �rH rg r� r� r r r �testRecvIntoArray% s z BufferIOTest.testRecvIntoArrayc C � t t�}| j�|� d S r9 �r� r� r� r� �rH rg 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 �testRecvIntoBytearray1 s
z"BufferIOTest.testRecvIntoBytearrayc C sH t d�}| j�t|��}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r� r� 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� r� �rH rg r� r� r� r r r �testRecvFromIntoArrayC s z"BufferIOTest.testRecvFromIntoArrayc C r� r9 r� r� r r r �_testRecvFromIntoArrayK 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 �testRecvFromIntoBytearrayO r� z&BufferIOTest.testRecvFromIntoBytearrayc C sL t d�}| j�t|��\}}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r� r� r� r r� r� r r r �testRecvFromIntoMemoryviewX s
z'BufferIOTest.testRecvFromIntoMemoryviewc C s t d�}| �t| jj|d� d S )Nrm r� )r� rV r� r� r� r� r r r �testRecvFromIntoSmallBuffera s z(BufferIOTest.testRecvFromIntoSmallBufferc C r r9 r rG r r r � _testRecvFromIntoSmallBufferf r z)BufferIOTest._testRecvFromIntoSmallBufferc C s$ t � }| j�|� | j�|d� d S r� )r� r� r� r� r r r �testRecvFromIntoEmptyBufferi s z(BufferIOTest.testRecvFromIntoEmptyBufferNr� )rO rP rQ rg rY r� r� r� Z_testRecvIntoBytearrayr� Z_testRecvIntoMemoryviewr� r� r� Z_testRecvFromIntoBytearrayr� Z_testRecvFromIntoMemoryviewr� r� r� Z _testRecvFromIntoEmptyBufferr r r r r� 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 rS r r r �isTipcAvailableu s$
�
��
��r� z6TIPC module is not loaded, please 'sudo modprobe tipc'c @ rT )�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� re rd r r� rf rg �TIPC_ADDR_NAMESEQ�
TIPC_STYPE�
TIPC_LOWER�
TIPC_UPPERrp �TIPC_ADDR_NAMEr� r_ r� r� r� r� )rH rI r� �srvaddrZsendaddrr� Zrecvaddrr r r �testRDM� s �
� zTIPCTest.testRDMN)rO rP rQ r� r r r r r� � rh r� c @ r� )�TIPCThreadableTestr� c C r� r� r� r� r r r rY � 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� r6 rI rd r r� rf rg r� r� r� r� rp rF r~ r� r� r� )rH r� r r r rI � 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� r6 r� rd r r� r� r� r� r� r� r� �cliaddr)rH r� r r r r� � s
� z TIPCThreadableTest.clientSetUpc C s, | j �d�}| �|t� | �| j| j� d S r� )r� r� r� r� r� r� rY r r r r� � s z TIPCThreadableTest.testStreamc C r� r9 r� rG 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� r2 rB r rV r r! r r r r �_testSocketClass� s
�
�
�z$ContextManagersTest._testSocketClassc C �2 | j �� \}}| �|j� |�d�}|�|� d S r� �rC r� rd r r� r! �rH r� r� r� r r r �testCreateConnectionBase� r� z,ContextManagersTest.testCreateConnectionBasec C sl | j �� }t�|�� }| �|j� |�d� | �|�d�d� W d � n1 s)w Y | � |j� d S )Nr\ r� )
rC r� r r r� r2 r! r� r� rB �rH r� r r r r �_testCreateConnectionBase� s
�z-ContextManagersTest._testCreateConnectionBasec C r r� r r r r r �testCreateConnectionClose� r� z-ContextManagersTest.testCreateConnectionClosec C s\ | j �� }t�|�� }|�� W d � n1 sw Y | �|j� | �t|j d� d S r )
rC r� r r r rB r2 rV r r! r r r r �_testCreateConnectionClose� s
� z.ContextManagersTest._testCreateConnectionCloseN)rO rP rQ r r r r r r r r r r� � s r� 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� r r c C sZ t � t jt jt jB ��}| �|jt j� | �|�� � W d � d S 1 s&w Y d S r9 )r rB r6 r� r� r# r� �get_inheritablerJ r r r �test_SOCK_CLOEXEC� s
�"�z!InheritanceTest.test_SOCK_CLOEXECc C sB t � � }|� | �|�� d� W d � d S 1 sw Y d S r( )r r� r r r r r �test_default_inheritable 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 r( )r r( r r� r )rH 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_inheritable
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 rH r� r r �F_GETFD�
FD_CLOEXECZF_SETFD)rH r ri r r r r � test_get_inheritable_cloexec 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 rH r� r r r r )rH r ri r r r � test_set_inheritable_cloexec$ s �
�"�z,InheritanceTest.test_set_inheritable_cloexecc C sH t �� \}}| �|j� | �|j� | �|�� d� | �|�� d� d S r( )r r� rd r r� r )rH r� rw r r r �test_socketpair1 s
z InheritanceTest.test_socketpair)rO rP rQ r� r� r r r r, r r
r r r� r r r r r r r r r
� s �
r
ra zSOCK_NONBLOCK not definedc @ s, e Zd Zd dd�Ze�ddd�dd � �Zd
S ) �NonblockConstantTestTr c C s� |r7| � |jtj� | � |�� |� | �t�|tjtj �tj @ � |dkr.| �
|� � � d S | �|� � � d S | � |jtj� | � |�� d � | �
t�|tjtj �tj @ � | �|� � � d S r� ) r� r# r r6 r� rB r r r� r r� r )rH r" �nonblockr< r r r �
checkNonblock< s � �z"NonblockConstantTest.checkNonblockr� r r 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)r g @r� r ) r rB r6 ra r rb r� r: r; )rH r" �tr r r �test_SOCK_NONBLOCKU sD
�
�
�
�
�
�z'NonblockConstantTest.test_SOCK_NONBLOCKN)Tr )rO rP rQ r r r, r r r r r r 9 s
r rl rm zneed multiprocessingc @ s@ e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
S )�TestSocketSharingc C sD |� � }|� � }t�|�}|�� \}}|�|� |�� |�� d S r9 )r� r � fromsharer� r! r )r� �qZsdata�messager" rw r$ r r r �remoteProcessServer| s
z%TestSocketSharing.remoteProcessServerc C s� t �� }t j| j|fd�}|�� | j�|j�}| j�� }| j� � |�
|� d}|�
|� t � |�}g } |�
d�}|s>n|�|� q6|� � d�|�}| �||� |�� d S )N�r� r^ s
slapmahfroTr� r! )�multiprocessingr� ZProcessr r� rC �share�pidr� r r� r r r� r r6 r� ) rH r rS r� r� r r" �m�receivedr r r � testShare� s*
�
zTestSocketSharing.testSharec C s@ | j �t�� �}| �ttj|d d� � | �ttj|d � d S )Nr� r\ )rC r" r� �getpidrV r� r r )rH 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� rL r# rM )rH Zorg�otherr r r �compareSockets� s
�z TestSocketSharing.compareSocketsc C sB | j �t�� �}t�|�}z| �| j |� W |�� d S |�� w r9 )rC r" r� r' r r r* r )rH 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 r9 ) r rB r r6 r% r r" r� r' r r* r )rH Zfamilies�typesr r �sourcer� Zsharedr r r � testTypes� s$ �
��zTestSocketSharing.testTypesN)
rO rP rQ r� r r& r( r* r+ r. r r r r r w s
# r 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 r! c C s� dd� }d� dd� t| j�D ��}ttjd��}|| j| j�D ]}|�|� q W d � n1 s1w Y ttjd��}|�� | _ W d � d S 1 sMw Y d S )Nc s s2 � | |kr|V | |8 } | |ks| r| V d S d S r9 r )�total�stepr r r �chunks� s ��
�z0SendfileUsingSendTest.setUpClass.<locals>.chunksr! c S s g | ]
}t �tj��� �qS r )�randomZchoice�stringZ
ascii_lettersr r0 r r r r/ � s �z4SendfileUsingSendTest.setUpClass.<locals>.<listcomp>r: r )
r6 rD �BUFSIZEr r r� �FILESIZEr� r# �FILEDATA)r� r2 �chunkr Zcsizer r r �
setUpClass� s
� � � "�z SendfileUsingSendTest.setUpClassc C s t �t j� d S r9 )r r r� )r� r r r �
tearDownClass� r� z#SendfileUsingSendTest.tearDownClassc C s8 | j �tj� | j �� \}}|�| j� | �|j� |S r9 )rC r� r r) r� �TIMEOUTrd r r� r r r � accept_conn� s
z!SendfileUsingSendTest.accept_connc C s. g } |� | j�}|s n|�|� qd�|�S )NTr! )r� r5 r r6 )rH r� r% r8 r r r � recv_data s
�
z SendfileUsingSendTest.recv_datac C �
t |d�S )NZ_sendfile_use_send�r� r r r r �meth_from_sock 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 � rC r� r r r� r r r@ r� r6 �tell�rH r� �filer �meth�sentr r r �_testRegularFile s
P�z&SendfileUsingSendTest._testRegularFilec C �6 | � � }| �|�}| �t|�| j� | �|| j� d S r9 �r< r= r� r r6 r7 �rH r� r� r r r �testRegularFile �
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 r9 )rC r� �io�BytesIOr7 r r rk r� r6 rB rV rr rs )rH r� rD r rF r r r �_testNonRegularFile' s
�P�z)SendfileUsingSendTest._testNonRegularFilec C rH r9 rI rJ r r r �testNonRegularFile1 rL 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�2r: r r ) rC r� r r� r rd r r r r@ r� rB )rH r� �filenamerD r rE rF r r r �_testEmptyFileSend9 s
�
P�z(SendfileUsingSendTest._testEmptyFileSendc C s" | � � }| �|�}| �|d� d S r0 )r< r= r� rJ r r r �testEmptyFileSendE 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 � )�offsetrA rC r r r � _testOffsetL s
P�z!SendfileUsingSendTest._testOffsetc C sB | � � }| �|�}| �t|�| jd � | �|| jdd � � d S )NrU rI rJ r r r �
testOffsetU 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� � rC r� r r r� r r r ra r@ r� rB �rH r� rD r r� rE rF r r r �
_testCount] �
�
P�z SendfileUsingSendTest._testCountc C �@ d}| � � }| �|�}| �t|�|� | �|| jd |� � d S )NrZ �r< r= r� r r7 �rH r� r� r� r r r � testCounti �
z SendfileUsingSendTest.testCountc C rY )Nr r� r r[ r\ r] r r r �_testCountSmallr r_ z%SendfileUsingSendTest._testCountSmallc C r` r
ra rb r r r �testCountSmall~ rd 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� 駆 �� )rV r� )
rC r� r r r� r r r@ r� rB r] r r r �_testCountWithOffset� s
P�z*SendfileUsingSendTest._testCountWithOffsetc C sD d}| � � }| �|�}| �t|�|� | �|| jd|d � � d S )Nrg rh ra rb 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) rC r� r r r� r r rb r@ rV r� rk �rH r� rD r rE 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)r< r� rR �rH 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� ) rC r� r r r� r r r ra r@ r� r6 rC r r r �_testWithTimeout� s
�
P�z&SendfileUsingSendTest._testWithTimeoutc C rH r9 rI rJ r r r �testWithTimeout� rL 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�?) rC r� r r r� r r r� r@ rV r# rk r r r � _testWithTimeoutTriggeredSend� s
��"�z3SendfileUsingSendTest._testWithTimeoutTriggeredSendc C s | � � }|�d� t�d� d S )Ni�X r )r< r� r� r� rm r r r � testWithTimeoutTriggeredSend� s
z2SendfileUsingSendTest.testWithTimeoutTriggeredSendc C r� r9 r rG 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 r6 r r= z binary modezpositive integerrQ r[ r� r r� ) r r r� r r% r@ rB r� r� )rH rD r" rE r r r � test_errors� sJ
� �� �
� �� �
���� ��"�z!SendfileUsingSendTest.test_errorsN)&rO rP rQ rg r6 r5 r7 r ra r; r� r9 r: r< r= r@ rG rK rO rP rS rT rW rX r^ rc re rf ri rj rl rn ro rp rq rr rs rt r r r r r/ � sF
r/ rk z%os.sendfile() required for this test.c @ r )�SendfileUsingSendfileTestzB
Test the sendfile() implementation of socket.sendfile().
c C r> )Nrs r? r r r r r@ � rm z(SendfileUsingSendfileTest.meth_from_sockN)rO rP rQ rg r@ r r r r ru � s ru 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, rp r� r r� ZSkipTestr9 )rH �typr. r r� r r r �
create_alg� 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� cr! )
r� �fromhexrx r� r! r� r� r� r �MSG_MORE)rH rN �algo�opr� r r r � test_sha256 s$
�
��"�z LinuxKernelCryptoAPI.test_sha256c C s� d}d}t �d�}| �dd��@}|�tjtj|� |�� \}}|� |�|� | � |�
d�|� W d � n1 s:w Y W d � d S W d � d S 1 sRw Y d S )Ns Python loves AF_ALGs what do ya want for nothing?Z(193dbb43c6297b47ea6277ec0ce67119a3f3aa66ry z
hmac(sha1)r� ) r� r} rx r� r �SOL_ALG� ALG_SET_KEYr� r! r� r� )rH �keyr� rN r r� r� r r r �test_hmac_sha1 s
��"�z#LinuxKernelCryptoAPI.test_hmac_sha1r 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 rx r� r r� r� r� �
sendmsg_afalg�ALG_OP_ENCRYPTr~ r! r� r� �ALG_OP_DECRYPT)
rH r� r� r� Z
ciphertextZmsglenr r� r� Z
multiplierZlongmsg�encZdecr r r � test_aes_cbc! sV
�
� � �
� � � �$�z!LinuxKernelCryptoAPI.test_aes_cbcr� c C sZ t �d�}|d ur|dkr|dk r| �d� t�d�}t�d�}t�d�}t�d�}t�d �}t�d
�}t|�}t|�} | �d d ���_}
|
�tj tj
|� |
�tj tj d |� |
� � \} } | �@ | j
tj|| tjd
� | �|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 || }| �|gtj tj|tj�gtj tj|t|��| gtj tj|| �gf� | �t|�| �}
| �||
| | � � | �||
| d � � W d � n 1 �sPw Y |
� � \} } | �, || | }| j
|gtj|| d� | �t|�| �}
| �||
| d � � W d � n1 �s�w Y W d � d S W d � d S 1 �s�w Y d S )NZLinux)r r� )r r� zupstream Linux kernel issueZ c939cc13397c1d37de6ae0e1cb7c423cZb3d8cc017cbb89b39e0f67e2Z c3b3c41f113a31b73d9a5cd432103069Z 24825602bd12a984e0092d3e448eda5fZ 93fe7d9e9bfd10348a5606e5cafa7354Z 0032a1dc85f1c9786925a2e71d8272ddZaeadzgcm(aes))r� r� �assoclenr )r� r� r� r
)r Z_get_kernel_versionrr r� r} r rx r� r r� r� ZALG_SET_AEAD_AUTHSIZEr� r� r� r~ r! r� r� r ZStructr� r� Z
ALG_SET_OPZ
ALG_SET_IVZALG_SET_AEAD_ASSOCLENr� )rH Zkernel_versionr� r� ZplainZassocZ expected_ctZ expected_tagZtaglenr� r r� r� �resr� Z pack_uint32r r r �test_aead_aes_gcmJ s�
�
�
�
� � �� � � ��,$�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� ) rx r� �urandomr� r r� r� r� r� r� r )rH 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, rV 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, rd r rV r� rp rB r� r r r r �test_length_restriction� s � � � "�z,LinuxKernelCryptoAPI.test_length_restrictionN)
rO rP rQ rx r r, r� r� r� r� r� r� r� r r r r rv � s
(
D
rv rp rq c @ rT )�TestMacOSTCPFlagsc C s | � tj� d S r9 )rB r Z
TCP_KEEPALIVErG r r r �test_tcp_keepalive� r z$TestMacOSTCPFlags.test_tcp_keepaliveN)rO rP rQ r� r r r r r� � rV r� �winzrequires Windowsc @ s e Zd Zh d�Zdd� ZdS )�TestMSWindowsTCPFlags> Z TCP_NODELAYZ TCP_KEEPCNTZ TCP_FASTOPENZ
TCP_MAXSEGZ
TCP_KEEPINTVLZ TCP_KEEPIDLEc 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/ � r0 z<TestMSWindowsTCPFlags.test_new_tcp_flags.<locals>.<listcomp>c s g | ] }|� j vr|�qS r )�
knownTCPFlagsr� rG r r r/ � r0 zANew TCP flags were discovered. See bpo-32394 for more information)r r r� )rH Zprovided�unknownr rG r �test_new_tcp_flags� s
�z(TestMSWindowsTCPFlags.test_new_tcp_flagsN)rO rP rQ 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 )NrF r r r� r )r r� r �
create_serverr� r� r
r �rH rE 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 )NrE �r� r r )
r r� r� rL rB r# r6 r r
r )rH r r" r r r �test_family_and_type� 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)Z
reuse_portr )
r r rV r� r� r� rf r� r� r� )rH r rx r r r �test_reuse_port� s
"� �"�z CreateServerTest.test_reuse_portrj r r rr c C s6 t jdt jd�� }W d � d S 1 sw Y d S )Nr� r )r r� r r r r r �test_ipv6_only_default� s 6z'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�rL Zdualstack_ipv6)r r� r r� rL r r r r �test_dualstack_ipv6_family s
�"�z+CreateServerTest.test_dualstack_ipv6_familyN)rO rP rQ r� r� r� r� r� r rA 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� �r rH r r �run s
�� ��"�z3CreateServerFunctionalTest.echo_server.<locals>.runr ) rZ r� r� r< r� r� rd r6 r} )rH 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� )rH r� rL r r r r � echo_client# s
"�z&CreateServerFunctionalTest.echo_clientc C sX t �� }t�d|f��}| �|� | �d|ftj� W d � d S 1 s%w Y d S )Nr[ rF )r r� r r� r� r� rB r� r r r � test_tcp4* s
"�z$CreateServerFunctionalTest.test_tcp4rr 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_tcp60 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� rF )r r� r r� r r� r� rB 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_v6D r� z4CreateServerFunctionalTest.test_dual_stack_client_v6N)rO rP rQ r ra 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 @ rT )�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 r9 �r� r )�pipesr) Zfd2r r r � close_pipesS s
�z8SendRecvFdsTests.testSendAndRecvFds.<locals>.close_pipesc S s | D ]}t �|� qd S r9 r� )r ri r r r � close_fdsX s �z6SendRecvFdsTests.testSendAndRecvFds.<locals>.close_fdsc S r/ 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 rd r r� r r6 r� r� r� r r� r r� r� r9 r r# )rH r� r� r� r r r r� Zfds2r r� r� r� r� r� r r r �testSendAndRecvFdsR s. & �� �z#SendRecvFdsTests.testSendAndRecvFdsN)rO rP rQ r� r r r r r� N r� r� c C s t �� } tjt jg| �R � d S r9 )r Zthreading_setupr� ZaddModuleCleanupZthreading_cleanup)� thread_infor r r � setUpModulex s r� �__main__)�r� �testr Z test.supportr r r r� rM r� r r� r� r� r� r r� r� r� r� r� �weakrefr r r rE r r3 ry r4 �_threadr� rZ r! r� r r� r r� r� r� �systemr
rA r r# r' r) r- r0 r2 r4 r8 r� r= ZHAVE_SOCKET_CANZHAVE_SOCKET_CAN_ISOTPZHAVE_SOCKET_CAN_J1939ZHAVE_SOCKET_RDSZHAVE_SOCKET_ALGZHAVE_SOCKET_QIPCRTRZHAVE_SOCKET_VSOCKr ZHAVE_SOCKET_UDPLITEZHAVE_SOCKET_BLUETOOTHr� r r� rA rR rU rW ri rv ry 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+ r8 r? r@ r� r� r� r� r� r� r� r� r r r6 rF rG rc rj rp rq rt rw rz r� r� r� r� r� r� r� r� 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� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r- rd ru rx ry 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/ ru rv r� r� r� r� r� r� r� rO �mainr r r r �<module> sj � �
�
3q�� �$!$� �$
S
Z
*
=
>��.�%{ �7w w= # P; k 7�
�
�
�� ��
�
�
� �
�
�
�
� �
� � �
�
� �
�
� �
�
� �
� )��7#j ! $l
?G 9�7JS �� ',C�< e
� N3
@
' �