Your IP : 216.73.217.129
a
XC?h�� � @ s� d dl Z d dlmZ d dlmZ d dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlZd dlZd dlmZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl Z z d dl Z W n e �y dZ Y n0 z d dl!Z!W n e �y. dZ!Y n0 ej"Z"d�#d�Z$dZ%d Z&e�'� d
kZ(z d dl)Z)W n e �yx dZ)Y n0 d d � Z*d
d� Z+dd� Z,dd� Z-dd� Z.dd� Z/dd� Z0dd� Z1dd � Z2ej3d d � �Z4e+� Z5e,� Z6e-� Z7e.� Z8e/� Z9e0� Z:e1� Z;e<ed �Z=e2� Z>e�d �j?Z@G d!d"� d"e jA�ZBG d#d$� d$e jA�ZCG d%d&� d&eC�ZDG d'd(� d(�ZEG d)d*� d*e jA�ZFG d+d,� d,e jA�ZGG d-d.� d.�ZHG d/d0� d0eBeH�ZIG d1d2� d2eCeH�ZJe �Ke=d3�G d4d5� d5eDeH��ZLG d6d7� d7eFeH�ZMG d8d9� d9eGeH�ZNe �Oe!du d:�e �Ke;d;�e �Ke*� d<kd=�G d>d?� d?e jAeH����ZPG d@dA� dAeI�ZQG dBdC� dCe jAeH�ZRG dDdE� dEe jA�ZSG dFdG� dGeS�ZTG dHdI� dIeEeSeH�ZUG dJdK� dKeTeU�ZVG dLdM� dMeS�ZWG dNdO� dOeW�ZXG dPdQ� dQeS�ZYG dRdS� dSeY�ZZG dTdU� dUeY�Z[G dVdW� dWeY�Z\G dXdY� dYeY�Z]G dZd[� d[eY�Z^G d\d]� d]e^�Z_G d^d_� d_e^�Z`d`da� Zadbdc� Zbddde� ZcG dfdg� dge jA�Zde �Ke5dh�G didj� dje jA��Zee �Ke5dh�G dkdl� dleM��Zfe �Ke6dm�G dndo� doe jA��Zge �Ke7dp�G dqdr� dre jA��Zhe �Ke8ds�G dtdu� due jA��Zie �Ke8ds�G dvdw� dweN��Zje �Ke:dx�G dydz� dze jA��Zke �Oe!du d:�e �Ke;d;�G d{d|� d|e jA���Zle �Ke>d}�G d~d� de jA��ZmG d�d�� d�eQ�ZnG d�d�� d�eJ�Zoe �Ke=d3�G d�d�� d�eL��ZpG d�d�� d�eE�ZqG d�d�� d�eq�ZrG d�d�� d�eq�ZsG d�d�� d�eq�ZtG d�d�� d�eq�ZuG d�d�� d�eq�ZvG d�d�� d�eq�ZwG d�d�� d�ew�ZxG d�d�� d�ex�ZyG d�d�� d�ex�ZzG d�d�� d�eq�Z{G d�d�� d�e{�Z|G d�d�� d�e{�Z}G d�d�� d�ere{�Z~G d�d�� d�e jA�ZG d�d�� d�ew�Z�G d�d�� d�ew�Z�G d�d�� d�eseueUe[�Z�ebejd��G d�d�� d�eze���Z�ebejd��G d�d�� d�e}e���Z�ebejd��G d�d�� d�e~e���Z�G d�d�� d�eseueUe_�Z�ebejd��e �Kej�d��ecd�d��G d�d�� d�eze�����Z�ebejd��e �Kej�d��ecd�d��G d�d�� d�e}e�����Z�ebejd��e �Kej�d��ecd�d��G d�d�� d�e~e�����Z�ebejd��e �Kej�d��ebed��ecd�d��G d�d�� d�e�e������Z�ebejd��e �Kej�d��ebed��ecd�d��G d�d d�ere�e������Z�e �Ke=d3�G d�dĄ d�eseueUe\��Z�e �Ke=d3�ebejd��G d�dƄ d�eze����Z�e �Ke=d3�ebejd��G d�dȄ d�e}e����Z�e �Ke=d3�ebejd��G d�dʄ d�e~e����Z�e �Ke=d3�G d�d̄ d�eseueUe`��Z�ebejd��e �Kej�d��e �Ke=d3�ecd�d��G d�d΄ d�eze������Z�ebejd��e �Kej�d��e �Ke=d3�ecd�d��G d�dЄ d�e}e������Z�ebejd��e �Kej�d��e �Ke=d3�ecd�d��G d�d҄ d�e~e������Z�ebejd��e �Kej�d��e �Ke=d3�ebed��ecd�d��G d�dԄ d�e�e�������Z�ebejd��e �Kej�d��e �Ke=d3�ebed��ecd�d��G d�dք d�ere�e�������Z�G d�d d�eveVeZ�Z�ebejd��G d�dڄ d�eye���Z�ebejd��G d�d܄ d�e}e|e���Z�ebejd��G d�dބ d�e~e|e���Z�G d�d� d�eteveVe]�Z�ebejd��e �Oe(d�ecd�d�d�G d�d� d�eye�����Z�ebejd��e �Oe(d�ecd�d�d�G d�d� d�e}e|e�����Z�ebejd��e �Oe(d�ecd�d�d�G d�d� d�e~e|e�����Z�G d�d� d�eveVeX�Z�ebejd��ebed�G d�d� d�eye����Z�ebejd��ebed�G d�d� d�e}e|e����Z�ebejd��ebed�G d�d� d�e~e|e����Z�ebejd�d��ebed�d�d��G d�d�� d�e�e����Z�ebejd�d��ebed�d�d��G d�d�� d�ere�e����Z�G d�d�� d��Z�ebed��e �Ke<ed��� p�e<ed��d��G �d �d� �de�e[���Z�ebed��e �Ke<ed���
p(e<ed��d��G �d�d� �de�eEeTeZ���Z�G �d�d� �deI�Z�G �d�d� �deR�Z�G �d�d � �d eR�Z�G �d
�d � �d eI�Z�G �d �d
� �d
eQ�Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d�d� �de��Z�G �d �d � �d e�en�Z�G �d �d � �d e jA�Z�G �d �d!� �d!eBeH�Z�G �d"�d#� �d#eBeH�Z�G �d$�d%� �d%eB�Z�G �d&�d'� �d'eC�Z�e �Ke=d3�G �d(�d)� �d)eD��Z�G �d*�d+� �d+e jA�Z�e �Kej�d,k�d-�G �d.�d/� �d/e jA��Z�e �Ke<ed��d0�G �d1�d2� �d2e jA��Z�G �d3�d4� �d4eQ�Z��d5Z��d6Z��d7Zd8�d9� Z�e �Keà �d:�G �d;�d<� �d<e jA��Z�e �Keà �d:�G �d=�d>� �d>e jAeH��Z�G �d?�d@� �d@eI�Z�G �dA�dB� �dBe jA�Z�e �Ke<e�dC��dD�G �dE�dF� �dFe jA��Z�e �KejɐdGk�dH�e �Ke �dI�G �dJ�dK� �dKeB���Z�G �dL�dM� �dMeI�Z�e �Ke<e�dN��dO�G �dP�dQ� �dQe˃�Z�e �Ke9�dR�G �dS�dT� �dTe jA��Z�e �Kej�ΐdU��dV�G �dW�dX� �dXe jA��Z�G �dY�dZ� �dZe jA�Z�G �d[�d\� �d\e jA�Z�ebe�d]�ebe�d^�ebed�G �d_�d`� �d`e jA����ZҐda�db� Z�eԐdck�r�e �ա dS (d � N)�support)�
socket_helper)�proxyFu Michael Gilfix was hereሴ
�utf-8� �� �AIXc C s� t d u r d S ttd�sd S z>tdd�� } t �| tjd�}W d � n1 sL0 Y W n tyl Y d S 0 t�d|�d S d S )N� IOCTL_VM_SOCKETS_GET_LOCAL_CIDz
/dev/vsock�rbz �Ir ) �fcntl�hasattr�socket�open�ioctlr �OSError�struct�unpack)�f�r� r �5/opt/alt/python39/lib64/python3.9/test/test_socket.py�get_cid3 s
2 r c C s> zt � t jt jt j�} W n ttfy0 Y dS 0 | �� 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_can@ s r c C s> zt � t jt jt j�} W n ttfy0 Y dS 0 | �� dS )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_isotpJ s r# c C s> zt � t jt jt j�} W n ttfy0 Y dS 0 | �� dS )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_j1939T s r% c C s< zt � t jt jd�} W n ttfy. Y dS 0 | �� 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^ s r( c C s< zt � t jt jd�} W n ttfy. Y dS 0 | �� dS )z8Check whether AF_ALG sockets are supported on this host.r FT)r �AF_ALGr' r r r r r r r �_have_socket_algh s r* c C s< zt � t jt jd�} W n ttfy. Y dS 0 | �� dS )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_qipcrtrr s r, 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 r. c C s> zt � t jt jt j�} W n ttfy0 Y dS 0 | �� dS )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 r2 c c s8 t �� }z t �| � d V W t �|� n t �|� 0 d S �N)r �getdefaulttimeout�setdefaulttimeout)�timeoutZ old_timeoutr r r �socket_setdefaulttimeout� s
r7 �IPPROTO_UDPLITE�ic @ s e Zd Zdd� Zdd� ZdS )�
SocketTCPTestc C s. t � t jt j�| _t�| j�| _| j�� d S r3 )r �AF_INETr0 �servr � bind_port�port�listen��selfr r r �setUp� s zSocketTCPTest.setUpc C s | j �� d | _ d S r3 �r< r r@ r r r �tearDown� s
zSocketTCPTest.tearDownN��__name__�
__module__� __qualname__rB rD r r r r r: � s r: c @ s e Zd Zdd� Zdd� ZdS )�
SocketUDPTestc C s$ t � t jt j�| _t�| j�| _d S r3 )r r; r! r< r r= r> r@ r r r rB � s zSocketUDPTest.setUpc C s | j �� d | _ d S r3 rC r@ r r r rD � s
zSocketUDPTest.tearDownNrE r r r r rI � s rI c @ s e Zd Zdd� ZdS )�SocketUDPLITETestc C s( t � t jt jt j�| _t�| j�| _d S r3 )r r; r! r8 r< r r= r> r@ r r r rB � s zSocketUDPLITETest.setUpN)rF rG rH rB r r r r rJ � s rJ 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 r3 )�super�__init__� threading�RLock�
_cleanup_lock�rA �args�kwargs�� __class__r r rM � s z"ThreadSafeCleanupTestCase.__init__c s< | j �" t� j|i |��W d � S 1 s.0 Y d S r3 )rP rL �
addCleanuprQ rT r r rV � s z$ThreadSafeCleanupTestCase.addCleanupc s< | j �" t� j|i |��W d � S 1 s.0 Y d S r3 )rP rL �
doCleanupsrQ rT r r rW � s z$ThreadSafeCleanupTestCase.doCleanups)rF rG rH �__doc__rM rV rW �
__classcell__r r rT r rK � s rK c @ sH e Zd ZdZdZdZdZe�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 n" t yZ | �
d| j � Y n0 d S �N�%network interface `%s` does not exist) r r r r r rV r �bind� interfacer �skipTestr@ r r r rB s �zSocketCANTest.setUpN) rF rG rH rX rb �bufsize�
can_frame_fmtr ZcalcsizeZcan_frame_size�bcm_cmd_msg_fmtrB r r r r rZ � s
rZ 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� zt�| j�| _W n t yP | �
d� Y n0 d S )Nr zunable to bind RDS socket) r r& r' r< rV r r r= r> r rc r@ r r r rB s zSocketRDSTest.setUpN)rF rG rH rX rd rB r r r r rg s rg c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�ThreadableTesta� Threadable Test class
The ThreadableTest class makes it easy to create a threaded
client/server pair from an existing unit test. To create a
new threaded class from an existing unit test, use multiple
inheritance:
class NewClass (OldClass, ThreadableTest):
pass
This class defines two new fixture functions with obvious
purposes for overriding:
clientSetUp ()
clientTearDown ()
Any new test functions within the class must then define
tests in pairs, where the test name is preceded with a
'_' to indicate the client portion of the test. Ex:
def testFoo(self):
# Server portion
def _testFoo(self):
# Client portion
Any exceptions raised by the clients during their tests
are caught and transferred to the main thread to alert
the testing framework.
Note, the server setup function cannot call any blocking
functions that rely on the client thread during setup,
unless serverExplicitReady() is called just before
the blocking call (such as in setting up a client/server
connection and performing the accept() in setUp().
c C s | j | _| j| _ d S r3 )rB �_ThreadableTest__setUp�_setUpr@ r r r rM E 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�setr@ r r r �serverExplicitReadyJ 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�� _z0z � �� W n d� _
� Y n0 W � j�� n � j�� 0 � j �� � �� j
j� d S )Nr Fc s � j �� r� j �� �d S r3 )�queueZqsize�getr r@ r r �raise_queued_exception\ s
z5ThreadableTest._setUp.<locals>.raise_queued_exception�.�_T)r Zwait_threads_exitZ wait_threads� __enter__rV �__exit__rN �Eventrl � client_ready�donero �Queue�server_crashed�id�rfind�getattr�thread�start_new_thread� clientRunZ
client_threadrj rm �wait)rA rq Z
methodnamer9 Z test_methodr r@ r rk Q s2
�
zThreadableTest._setUpc
C s� | j �� zbz | �� W nF ty^ } z.| j�|� | �� W Y d }~W | j�� d S d }~0 0 W | j�� n | j�� 0 | j r�| �� d S t
|d�s�t d��zHz
|� W n0 ty� } z| j�|� W Y d }~n
d }~0 0 W | �� n
| �� 0 d S )N�__call__z%test_func must be a callable function) rl r� � clientSetUp�
BaseExceptionro �put�clientTearDownrw rm rz r
� TypeError)rA Z test_func�er r r r� s s(
�
$zThreadableTest.clientRunc C s t d��d S )Nz clientSetUp must be implemented.)�NotImplementedErrorr@ r r r r� � s zThreadableTest.clientSetUpc C s | j �� t�� d S r3 )rx rm r~ �exitr@ r r r r� � s
z ThreadableTest.clientTearDownN)
rF rG rH rX rM rn rk r� r� r� r r r r ri s %"ri c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedTCPSocketTest�runTestc C s t j| |d� t�| � d S �N)�
methodName�r: rM ri �rA r� r r r rM � s z ThreadedTCPSocketTest.__init__c C s t � t jt j�| _d S r3 )r r; r0 �clir@ r r r r� � s z!ThreadedTCPSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r3 �r� r ri r� r@ r r r r� � s
z$ThreadedTCPSocketTest.clientTearDownN)r� �rF rG rH rM r� r� r r r r r� � s
r� c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedUDPSocketTestr� c C s t j| |d� t�| � d S r� )rI rM ri r� r r r rM � s z ThreadedUDPSocketTest.__init__c C s t � t jt j�| _d S r3 )r r; r! r� r@ r r r r� � s z!ThreadedUDPSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r3 r� r@ r r r r� � s
z$ThreadedUDPSocketTest.clientTearDownN)r� r� r r r r r� � s
r� z'UDPLITE sockets required for this test.c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedUDPLITESocketTestr� c C s t j| |d� t�| � d S r� )rJ rM ri r� r r r rM � s z"ThreadedUDPLITESocketTest.__init__c C s t � t jt jt j�| _d S r3 )r r; r! r8 r� r@ r r r r� � s z%ThreadedUDPLITESocketTest.clientSetUpc C s | j �� d | _ t�| � d S r3 r� r@ r r r r� � s
z(ThreadedUDPLITESocketTest.clientTearDownN)r� r� r r r r r� � s
r� c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedCANSocketTestr� c C s t j| |d� t�| � d S r� )rZ rM ri r� r r r rM � s z ThreadedCANSocketTest.__init__c C sB t � t jt jt j�| _z| j�| jf� W n ty< Y n0 d S r3 )r r r r r� ra rb r r@ r r r r� � s
z!ThreadedCANSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r3 r� r@ r r r r� � s
z$ThreadedCANSocketTest.clientTearDownN)r� r� r r r r r� � s
r� c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedRDSSocketTestr� c C s t j| |d� t�| � d S r� )rg rM ri r� r r r rM � s z ThreadedRDSSocketTest.__init__c C sL t � t jt jd�| _z | j�tdf� | j�� | _W n tyF Y n0 d S �Nr ) r r& r' r� ra �HOST� getsockname�cli_addrr r@ r r r r� � s z!ThreadedRDSSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r3 r� r@ r r r r� � s
z$ThreadedRDSSocketTest.clientTearDownN)r� 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 @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )� ThreadedVSOCKSocketStreamTestr� c C s t jj| |d� t�| � d S r� ��unittest�TestCaserM ri r� r r r rM � s z&ThreadedVSOCKSocketStreamTest.__init__c C sh t � t jt j�| _| �| jj� | j�t jtf� | j� � | �
� | j� � \| _ | _
| �| j j� d S r3 )r �AF_VSOCKr0 r< rV r ra �VMADDR_CID_ANY� VSOCKPORTr? rn �accept�conn�connaddrr@ r r r rB � s
z#ThreadedVSOCKSocketStreamTest.setUpc C sX t �d� t�tjtj�| _| �| jj� t� }|tj tj
fv rDt }| j� |t
f� d S )N皙�����?)�time�sleepr r� r0 r� rV r r �VMADDR_CID_HOSTr� �VMADDR_CID_LOCAL�connectr� )rA r- r r r r� � s
z)ThreadedVSOCKSocketStreamTest.clientSetUpc
C sT z| j �d�}W n2 tyB } z| �t|�� W Y d }~n
d }~0 0 | �|t� d S �N� )r� �recv�PermissionErrorrc �repr� assertEqual�MSG)rA �msg�excr r r �
testStream s
$z(ThreadedVSOCKSocketStreamTest.testStreamc C s | j �t� | j �� d S r3 �r� �sendr� r r@ r r r � _testStream s z)ThreadedVSOCKSocketStreamTest._testStreamN)r� �rF rG rH rM rB 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 s t j| |d� d S r� )r� rM r� r r r rM s z SocketConnectedTest.__init__c C s* t �| � | �� | j�� \}}|| _d S r3 )r� rB rn r< r� �cli_conn�rA r� �addrr r r rB s
zSocketConnectedTest.setUpc C s | j �� d | _ t�| � d S r3 )r� r r� rD r@ r r r rD ! s
z SocketConnectedTest.tearDownc C s( t �| � | j�t| jf� | j| _d S r3 )r� r� r� r� r� r> � serv_connr@ r r r r� & s
z SocketConnectedTest.clientSetUpc C s | j �� d | _ t�| � d S r3 )r� r r� r� r@ r r r r� + s
z"SocketConnectedTest.clientTearDownN)r� ) rF rG rH rX rM rB rD 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� rM ri r� r< r� r r r rM 2 s
zSocketPairTest.__init__c C s t �� S r3 )r �
socketpairr@ r r r r� 8 s zSocketPairTest.socketpairc C s | � � \| _| _d S r3 )r� r< r� r@ r r r rB < s zSocketPairTest.setUpc C s | j r| j �� d | _ d S r3 rC r@ r r r rD ? s
zSocketPairTest.tearDownc C s d S r3 r r@ r r r r� D s zSocketPairTest.clientSetUpc C s$ | j r| j �� d | _ t�| � d S r3 r� r@ r r r r� G s
z SocketPairTest.clientTearDownN)r� ) rF rG rH rM r� rB rD r� r� r r r r r� 0 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 s | � � | _| �� d S r3 )� newSocketr< �
bindServerr@ r r r rB ] s
zSocketTestBase.setUpc C s | � | j� | j�� | _dS )z9Bind server socket and set self.serv_addr to its address.N)�bindSockr< r� � serv_addrr@ r r r r� a s zSocketTestBase.bindServerc C s | j �� d | _ d S r3 rC r@ r r r rD f s
zSocketTestBase.tearDownN)rF rG rH rX rB r� rD r r r r r� T s r� c s e Zd ZdZ� fdd�Z� ZS )�SocketListeningTestMixinz%Mixin to listen on the server socket.c s t � �� | j�� d S r3 )rL rB r< r? r@ rT r r rB n s
z SocketListeningTestMixin.setUp)rF rG rH rX rB rY r r rT r r� k 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 r3 )rL rM ri rQ rT r r rM { s z ThreadedSocketTestMixin.__init__c C s | � � | _| �� d S r3 )�newClientSocketr� �
bindClientr@ r r r r� s
z#ThreadedSocketTestMixin.clientSetUpc C s | � � S )z&Return a new socket for use as client.)r� r@ r r r r� � s z'ThreadedSocketTestMixin.newClientSocketc C s | � | j� | j�� | _dS )z8Bind client socket and set self.cli_addr to its address.N)r� r� r� r� r@ r r r r� � s z"ThreadedSocketTestMixin.bindClientc C s | j �� d | _ t�| � d S r3 r� r@ r r r r� � s
z&ThreadedSocketTestMixin.clientTearDown)
rF rG rH rX rM r� r� r� r� rY r r rT r r� s 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 r3 )rL rB rn r< r� r� r� rT r r rB � s
z ConnectedStreamTestMixin.setUpc s | j �� d | _ t� �� d S r3 )r� r rL rD r@ rT r r rD � s
z!ConnectedStreamTestMixin.tearDownc s$ t � �� | j�| j� | j| _d S r3 )rL r� r� r� r� r� r@ rT r r r� � s
z$ConnectedStreamTestMixin.clientSetUpc s6 z| j �� d | _ W n ty& Y n0 t� �� d S r3 )r� r r rL r� r@ rT r r r� � s
z'ConnectedStreamTestMixin.clientTearDown) rF rG rH rX rB rD r� r� rY r r rT 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 r3 )�tempfile�mkdtemp�dir_pathrV �os�rmdirrL rB r@ rT r r rB � s
zUnixSocketTestBase.setUpc C s, t j| jd�}t�||� | �tj|� d S )N)�dir)r� Zmktempr� r �bind_unix_socketrV r �unlink)rA �sock�pathr r r r� � s zUnixSocketTestBase.bindSock)rF rG rH rX rB r� rY r r rT r r� � s r� c @ s e Zd ZdZdd� ZdS )�UnixStreamBasez-Base class for Unix-domain SOCK_STREAM tests.c C s t � t jt j�S r3 )r �AF_UNIXr0 r@ r r r r� � s zUnixStreamBase.newSocketN�rF rG rH rX r� r r r r r� � s r� c s, e Zd ZdZeZ� fdd�Zdd� Z� ZS )� InetTestBasez!Base class for IPv4 socket tests.c s t � �� | jd | _d S �Nr )rL rB r� r> r@ rT r r rB � s
zInetTestBase.setUpc C s t j|| jd� d S )N��host)r r= r� �rA r� r r r r� � s zInetTestBase.bindSock) rF rG rH rX r� r� rB r� rY r r rT r r� � s r� c @ s e Zd ZdZdd� ZdS )� TCPTestBasez#Base class for TCP-over-IPv4 tests.c C s t � t jt j�S r3 )r r; r0 r@ r r r r� � s zTCPTestBase.newSocketNr� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )� UDPTestBasez#Base class for UDP-over-IPv4 tests.c C s t � t jt j�S r3 )r r; r! r@ r r r r� � s zUDPTestBase.newSocketNr� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )�UDPLITETestBasez'Base class for UDPLITE-over-IPv4 tests.c C s t � t jt jt j�S r3 )r r; r! r8 r@ r r r r� � s zUDPLITETestBase.newSocketNr� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )�SCTPStreamBasez;Base class for SCTP tests in one-to-one (SOCK_STREAM) mode.c C s t � t jt jt j�S r3 )r r; r0 � IPPROTO_SCTPr@ r r r r� � s �zSCTPStreamBase.newSocketNr� r r r r r� � s r� c @ s e Zd ZdZejZdS )�
Inet6TestBasez!Base class for IPv6 socket tests.N)rF rG rH rX r �HOSTv6r� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )� UDP6TestBasez#Base class for UDP-over-IPv6 tests.c C s t � t jt j�S r3 )r �AF_INET6r! r@ r r r r� � s zUDP6TestBase.newSocketNr� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )�UDPLITE6TestBasez'Base class for UDPLITE-over-IPv6 tests.c C s t � t jt jt j�S r3 )r r� r! r8 r@ r r r r� s zUDPLITE6TestBase.newSocketNr� r r r r r� s 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 _ s d S r3 r �rR rS r r r � client_pass s z%skipWithClientIf.<locals>.client_passc s* t ���| �}t| t�s&� fdd�|_|S )Nc s � S r3 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�s t| d�s dd� | _| S )Nr� c S s | S r3 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>% r� z requireAttrs.<locals>.<listcomp>� don't have �, c s s | ]
}|V qd S r3 r r� r r r � <genexpr>' r� z requireAttrs.<locals>.<genexpr>)r� �join)r� Z
attributes�missingr r� r � requireAttrs s �r c
G s� d}dd� | D �}|r0dd� dd� |D �� }nRdd� | D �}ztj|� }W n, tyx } zt|�}W Y d}~nd}~0 0 |�� 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� 2 s �z!requireSocket.<locals>.<listcomp>r� r� c s s | ]
}|V qd S r3 r r� r r r r 5 r� z requireSocket.<locals>.<genexpr>c S s$ g | ] }t |t�r tt|�n|�qS r )r� r r} r r r r r r� 7 s �z can't create socket({0}): {1}c s s | ]}t |�V qd S r3 )r �r� �or r r r C r� )r r r r r r� �format)rR �errr Zcallargsr r� r r r �
requireSocket* s$ � ��r
c @ sp e Zd Zdd� Zdd� Ze�edud�dd� �Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
e�ejdkd�dd� �Ze�ejdkd�e�ejd�dd� ��Zd d � Zd d � Zd d!� Ze�eed"�d#�e�eed$�d%�d&d'� ��Ze�eed(�d)�d*d+� �Z e�eed,�d-�d.d/� �Z e�eed0�d1�d2d3� �Z e�eed4�d5�d6d7� �Z d8d9� Z d:d;� Z!e"j#d<d=� �Z$d>d?� Z%d@dA� Z&e�eedB�dC�dDdE� �Z'e�eedF�dG�dHdI� �Z(e�eedF�dG�dJdK� �Z)e�eedL�dM�dNdO� �Z*e�eedL�dM�dPdQ� �Z+dRdS� Z,dTdU� Z-dVdW� Z.dXdY� Z/dZd[� Z0d\d]� Z1d^d_� Z2e�ejdkd�d`da� �Z3e�ejdkd�e�eedb�dc�ddde� ��Z4dfdg� Z5dhdi� Z6e�e"�7dj�dk�dldm� �Z8dndo� Z9dpdq� Z:drds� Z;dtdu� Z<dvdw� Z=dxdy� Z>dzd{� Z?d|d}� Z@d~d� ZAd�d�� ZBd�d�� ZCe"j#d�d�� �ZDe�ejd��d�d�� �ZEe�ejd��d�d�� �ZFe�ejd��e�Gejd�kd��e�GeHd��d�d�� ���ZIe�ejd��e�ejd�kd��d�d�� ��ZJe�ejd��e�Gejd�kd��e�GeHd��d�d�� ���ZKe�ejd��e�ejd�kd��d�d�� ��ZLd�d�� ZMd�d�� ZNd�d�� ZOe�eed��d��d�d�� �ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTd�d�� ZUe�Gejdkd��d�d�� �ZVd�d�� ZWd�d�� ZXdS )��GeneralModuleTestsc C s< dd l }| �tj|ju � t�� }| �|tj� |�� d S r� )�_socket�
assertTruer Z
SocketType�assertIsInstancer )rA r r r r r � test_SocketType_is_socketobjectK s
z2GeneralModuleTests.test_SocketType_is_socketobjectc C s� t � t jt j�}|�� | �d|�� t|�� | �dt j t|�� | �dt j t|�� | �dt|�� | �dt|�� |�d� | �dt|�� | �t|� � �t|�� W d � n1 s�0 Y | �dt|�� | �dt|�� d S ) Nzfd=%iz family=%sztype=%szproto=0Zraddr�� 127.0.0.1r Zladdrz[closed])
r r; r0 �assertIn�filenor� � assertNotInra r r� �rA r r r r � test_reprR s
6z GeneralModuleTests.test_reprNzneed _socket modulec C sz t �t jt j�}z6d|�� |j|j|jf }| �t |�|� W |�
� n
|�
� 0 d|j|j|jf }| �t |�|� d S )Nz4<socket object, fd=%s, family=%s, type=%s, proto=%s>z4<socket object, fd=-1, family=%s, type=%s, proto=%s>) r r r; r0 r �familyr� �protor� r� r )rA r �expectedr r r �test_csocket_repr` s ��z$GeneralModuleTests.test_csocket_reprc C s� t � t jt j��,}t|�}| �|�� |�� � W d � n1 sB0 Y d }t�� z |�� W n tyv Y n 0 | � d� d S )NzSocket proxy still exists)
r r; r0 r r� r r �
gc_collect�ReferenceError�fail)rA r �pr r r � test_weakrefm s 2 z GeneralModuleTests.test_weakrefc C s� d}| j t|d d�� t�W d � n1 s00 Y | j t|d d�� tj�W d � n1 sh0 Y | j t|d d�� tj�W d � n1 s�0 Y d S )Nz$Error raising socket exception (%s).r �r� z
socket.herrorzsocket.gaierror)� assertRaisesr r �herror�gaierror�rA r� r r r �testSocketErrorz s "$z"GeneralModuleTests.testSocketErrorc C s t � t jt j�}| �|j� |�d� |�� }| �t�� }|� d|� W d � n1 sZ0 Y | �
t |j �d� | �t�� }|� d|� W d � n1 s�0 Y | �
t |j �d� | �t�� }|� dd � W d � n1 s�0 Y | �
dt |j �� | �t�� }|� dd|� W d � n1 �s60 Y | �
t |j �d� | �t�� }|� dd|� W d � n1 �s�0 Y | �
t |j �d� | �t�� }|� ddd � W d � n1 �s�0 Y | �
dt |j �� | �t�� }|� dd |� W d � n1 �s0 Y | �
d
t |j �� | �t�� }|� dd d � W d � n1 �sf0 Y | �
d
t |j �� | �t��}|� d� W d � n1 �s�0 Y | �
d t |j �� | �t�� }|� dd|d � W d � n1 �s�0 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 �barzan integer is requiredz (1 given)� z (4 given))r r; r! rV r ra r� r! r� �sendtor� r � exceptionr )rA r �sockname�cmr r r �testSendtoErrors� sT
* � * � * . � . � . . . * 0z#GeneralModuleTests.testSendtoErrorsc C s@ t j t jrt j t j t j t j t j t j t j t j
d S r3 ) r r; �has_ipv6r� r0 r! r �SOCK_RDMr' �
SOL_SOCKET� SO_REUSEADDRr@ r r r �testCrucialConstants� s z'GeneralModuleTests.testCrucialConstantsc C s t j t j t jrt j d S r3 )r Z IPPROTO_TCP� IPPROTO_UDPr/ � IPPROTO_IPV6r@ r r r �testCrucialIpProtoConstants� s z.GeneralModuleTests.testCrucialIpProtoConstants�nt�Windows specificc C s4 t j t j t j t j t j t j t j t j d S r3 ) r ZIPPROTO_ICLFXBMZ
IPPROTO_STZ IPPROTO_CBTZ IPPROTO_IGPZ IPPROTO_RDPZ IPPROTO_PGMZ IPPROTO_L2TPr� r@ r r r � testWindowsSpecificConstants� s z/GeneralModuleTests.testWindowsSpecificConstants�darwinzmacOS specific test�IPv6 required for this testc C s0 h d�}|D ] }| � tt|�d|� d�� q d S )N> �
IPV6_HOPLIMITZIPV6_RECVHOPOPTSZ IPV6_NEXTHOPZIPV6_USE_MIN_MTUZ IPV6_PATHMTUZIPV6_RECVPKTINFOZIPV6_RTHDR_TYPE_0�IPV6_RECVHOPLIMITZ
IPV6_CHECKSUMZIPV6_RTHDRDSTOPTSZ IPV6_DSTOPTSZIPV6_RECVDSTOPTSZIPV6_RECVPATHMTUZIPV6_RECVRTHDRZ
IPV6_DONTFRAGZ IPV6_PKTINFO� IPV6_TCLASS�IPV6_RECVTCLASSZ IPV6_HOPOPTSZ
IPV6_RTHDRz Missing RFC3542 socket option '�')r
r
r )rA Zopts�optr r r �test3542SocketOptions� s
�z(GeneralModuleTests.test3542SocketOptionsc C s� t �� }zt �|�}W n ty2 | �d� Y n0 | �|�d�dkd� zt �|�\}}}W n tyz | �d� Y n0 ||g| }t �|�}||vr�| � d|t
|�f � d S )N�name lookup failurerr r zError resolving host to ip.z=Error testing host resolution mechanisms. (fqdn: %s, all: %s)) r � gethostname�
gethostbynamer rc r
�find�
gethostbyaddrZgetfqdnr r� )rA ZhostnameZipZhname�aliasesZipaddrsZall_host_namesZfqhnr r r �testHostnameRes� s
z"GeneralModuleTests.testHostnameResc C sL t jddfD ]}| �t�|�|� q t jfD ]}| �|t�|�d � q,d S )Nz10.0.0.1�255.255.255.255r� )r ZHOSTv4r� r rE r rG )rA r� r� r r r �test_host_resolution s z'GeneralModuleTests.test_host_resolutionc C s� d}dD ]t}| j t|d�� t�|� W d � n1 s:0 Y | j t|d�� t�|� W d � q1 sr0 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:1r )r! r r rE rG )rA Z explanationr� r r r � test_host_resolution_bad_address s �(z3GeneralModuleTests.test_host_resolution_bad_address� sethostnamez test needs socket.sethostname()rD z test needs socket.gethostname()c
C s� t �� }zt �d� W n> tyT } z&|jtjkr>| �d� n� W Y d }~n
d }~0 0 z8| �t �� d� t �d� | �t �� d� W t �|� n t �|� 0 d S )N�newztest should be run as roots barr( )r rD rM r �errnoZEPERMrc r� )rA Zoldhnr� r r r �test_sethostname s
z#GeneralModuleTests.test_sethostname� if_nameindexz$socket.if_nameindex() not available.c C s� t �� }|D ]p\}}| �|t� | �|t� | �|d� t �|�}| �|t� | �||� t �|�}| �|t� | �||� q d S r� ) r rQ r �intr �
assertGreater�if_nametoindexr� �if_indextoname)rA Z
interfaces�indexr� Z_index�_namer r r �testInterfaceNameIndex, s
z)GeneralModuleTests.testInterfaceNameIndexrU z&socket.if_indextoname() not available.c C s� | � ttjd� | � ttjd� | � ttjdd � | � ttjd� ttd�r�tt�� �}|D ]L}|d }||vr^| � ttf�� t�|� W d � q^1 s�0 Y q^dD ]D}||vr�| � ttf�� t�|� W d � q�1 s�0 Y q�d S ) Nr ���r� �� � _DEADBEEFrQ � )� �� l ���� ) r! r r rU �
OverflowErrorr� r
�dictrQ )rA �indicesrV Zindex2r r r � testInvalidInterfaceIndexToName< s
*z2GeneralModuleTests.testInvalidInterfaceIndexToNamerT z&socket.if_nametoindex() not available.c C s$ | � ttjd� | � ttjd� d S )Nr r[ )r! r� r rT r r@ r r r � testInvalidInterfaceNameToIndexO s z2GeneralModuleTests.testInvalidInterfaceNameToIndex� getrefcountz test needs sys.getrefcount()c C sJ zt �t�}t�td� W n* tyD t �t�|kr@| �d� Y n0 d S )Nr z$socket.getnameinfo loses a reference)�sysrc rF r � getnameinfor� r )rA �origr r r �testRefCountGetNameInfoU s
z*GeneralModuleTests.testRefCountGetNameInfoc C s( zt �dd� W n ty" Y n0 d S )N)r] r r r r )r re r r@ r r r �testInterpreterCrasha s z'GeneralModuleTests.testInterpreterCrashc C s� t jdt jdt jdt jdi}|�� D ]d\}}d|> d }dD ]$}| �||@ ||||@ ��|@ � q<||�}| �||@ |� | �t|d� q$d S )N� � r )r r � i ��r� igE#i2Tvl ) r �htonl�ntohl�htons�ntohs�itemsr� r! r^ )rA Zsizes�func�size�maskr9 Zswappedr r r �testNtoHi s � "zGeneralModuleTests.testNtoHc C s
dd l }g d�}|dg }ddddd> g}||jd |jd g }d |jg}|D ]}t�|� t�|� qP|D ]}t�|� t�|� qn|D ]$}| �t tj|� | �t tj|� q�|D ]$}| �t tj|� | �t tj|� q�|D ]$}| �
t tj|� | �
t tj|� q�d S )
Nr )r r r� rk r] rY ���r\ r rZ � ) � _testcapiZINT_MIN�INT_MAXr ro rn rm rl r! r^ � assertWarns�DeprecationWarning)rA rw Z
s_good_valuesZ
l_good_valuesZ l_bad_valuesZ s_bad_valuesZs_deprecated_values�kr r r �testNtoHErrorsw s.
�
z!GeneralModuleTests.testNtoHErrorsc C s | j }tj�d�s tjdv r"d}nd}|D ].}zt�|d�}W q^W q* tyV Y q*0 q*t�ttd�rtt�� dkr�t�|�}|||� zt�|d�}W n ty� d }Y n 0 |||� t j
s�|t� |�|� |t� |d�|� |d ur�|t� |d�|� | � t
tj d � | � t
tj d
� d S ) N)�freebsdZnetbsdZ gnukfreebsd)�linuxr: )�daytimeZqotd�domain)Zechor r� Ztcp�getandroidapilevel� ZudprY rv )r� rd �platform�
startswithr Z
getservbynamer r
r� r Z
is_androidZ
getservbyportr! r^ )rA �eqZservicesZservicer> Zport2Zudpportr r r �
testGetServBy� s< �
�
z GeneralModuleTests.testGetServByc C s0 | � t�� d � t�� � }| � |�� d � W d � n1 s>0 Y td��� | � t�� d� t�� � }| � |�� d� W d � n1 s�0 Y t�d � | � t�� d � t�� � }| � |�� d � W d � n1 s�0 Y W d � n1 �s0 Y | �ttjd� | �ttjd� d S )N�
rY Zspam) r� r r4 �
gettimeoutr7 r5 r! �
ValueErrorr� )rA r r� r r r �testDefaultTimeout� s
.
.
Nz%GeneralModuleTests.testDefaultTimeout� inet_atonz test needs socket.inet_aton()c C s( | � dt�d�� | � dt�d�� d S )N� �0.0.0.0� ����rJ )r� r r� r@ 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�� ts�||d� ||d� ||d� ||d� ||d� ��d|d�� ��d|d �� ��d
|d �� ��d|d�� ||d� ||d� ||d� ||d� ||d� d S )Nr )r� r� r; c s
�� | �S r3 r ��a)r; r� r r r� � r� z5GeneralModuleTests.testIPv4toString.<locals>.<lambda>c s � � ttf| |�S r3 �r! r r� �rq r� r@ r r r� � s
�r� r� s � � z 255.0.255.0s ����z170.170.170.170� �1.2.3.4r� rJ z0.0.0.z 300.0.0.0za.0.0.0z 1.2.3.4.5�::1)r r� r� r; r� r �rA r �g�
assertInvalidr )r; r� rA r �testIPv4toString� s.
z#GeneralModuleTests.testIPv4toStringc
s� z&ddl m�m� m} |s$��d� W n tyB ��d� Y n0 tjdkr�z�� d� W n8 ty� } z |j dkr���d� W Y d }~n
d }~0 0 � �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� r/ �IPv6 not available�+could not import needed symbols from socket�win32�::�&' �IPv6 might not be supportedc s
�� | �S r3 r r� )r� r� r r r� r� z5GeneralModuleTests.testIPv6toString.<locals>.<lambda>c s �� ttf� | �S r3 r� r� �r rA r r r� s
�� 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� r/ rc � ImportErrorrd r� r �winerrorr� r �rA r/ r� r� r )r� r r� rA r �testIPv6toString� sj
�����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� r; c s
�� | �S r3 r r� )r; r� r r r� B r� z5GeneralModuleTests.testStringToIPv4.<locals>.<lambda>c s � � ttf| |�S r3 r� r� r@ r r r� C s
�z1.0.1.0s z
170.85.170.85s �U�UrJ r� r� r� s s r� )r r� r� r; r� � bytearrayr� r )r; r� rA r �testStringToIPv4> s$
z#GeneralModuleTests.testStringToIPv4c
s z&ddl m�m� m} |s$��d� W n tyB ��d� Y n0 tjdkr�z�� d� W n8 ty� } z |j dkr���d� W Y d }~n
d }~0 0 � �fd d
����fd d
�}��
d �d�� ��
d
�d�� ��
d�d�� ��
d
�t d��� |d� |d� |d� d S )Nr )r� r� r/ r� r� r� r� r� r� c s
�� | �S r3 r r� )r� r� r r r� i r� z5GeneralModuleTests.testStringToIPv6.<locals>.<lambda>c s �� ttf� | �S r3 r� r� r� r r r� j s
�r� r� s z!aef:b01:506:1001:ffff:9997:55:170s
� ���� Ups s s ) r r� r� r/ rc r� rd r� r r� r� r� r� r )r� r r� rA r �testStringToIPv6X s0
�z#GeneralModuleTests.testStringToIPv6c C s� t �� }t�tjtj�}| �|j� |�d|f� |�� }zt� t�
� �}W n t yh | � d� Y n0 | �
|d d|fd|d � | �|d |� d S )Nr� rC r z
%s invalidr )r �find_unused_portr r; r0 rV r ra r� rE rD r rc r r� )rA r> 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 r; r0 rV r �
getsockoptr1 r2 � assertFalse�rA 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 r; r0 rV r �
setsockoptr1 r2 r� r� r� r r r �testSetSockOpt� s
z!GeneralModuleTests.testSetSockOptc C sN t � t jt j��}|�d� W d � n1 s00 Y | �t|jd� d S )Nr � spam)r r; r0 �
settimeoutr! r r� r� r r r �testSendAfterClose� s (z%GeneralModuleTests.testSendAfterClosec
C s� t � � }|�t jdf� t j |�� d��� z |�� W n: tyr } z"| �|jtjtj f� W Y d }~nd }~0 0 | �
d� d S )Nr �r z#close() should raise EBADF/ENOTSOCK) r ra Z
_LOCALHOSTr r r r rO �EBADF�ENOTSOCKr )rA r� r r r r �testCloseException� s ,z%GeneralModuleTests.testCloseExceptionc C s� t � t jt j��f}| �|jt j� tt d�rJ| �|jt jt jB t jf� n| �|jt j� | �|j d� W d � n1 s|0 Y d S )N� SOCK_CLOEXECr )
r r; r0 r� r r
r r� r� r 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 n: t y� } z"|j t j
ks�|dkr�� W Y d }~q\d }~0 0 q�q\d S )Nrv � )r rV r r r� r! r^ ra r� � itertools�countr rO Z
EADDRINUSE)rA r� r> Zbig_portZneg_portr9 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_VALSrY )r �d r� ) r
r
r rV r r! r� r r� r r r r �test_sock_ioctl� s z"GeneralModuleTests.test_sock_ioctl�SIO_LOOPBACK_FAST_PATHz1Loopback fast path support required for this testc
C s~ t � � }| �|j� z|�t jd� W n> tyd } z&d}|j|krN| �d� � W Y d }~n
d }~0 0 | �t |jt jd � d S )NTi=' zQSIO_LOOPBACK_FAST_PATH is defined but doesn't implemented in this Windows version)
r rV r r r� r r� rc r! r� )rA r r� Z
WSAEOPNOTSUPPr r r �test_sio_loopback_fast_path� s
z.GeneralModuleTests.test_sio_loopback_fast_pathc
C s@ zt �dd� W n> t jyN } z$|jt jkr8| �d� � W Y d }~n
d }~0 0 t �td �D ]}| �t|�d� q\t �dd� t �dd� t �d d� t j
r�t �dd� t t d�r�t �
� dkr�t �td � t �td� t �td � t �tdt jt j�}|D ]J\}}}}}| �|t j� | �t|�d
� | �|t j� | �t|�d � q�t �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 n t j�y: Y n0 d S )N� localhost�P zbuggy libc versionr� r r� r� r� Zhttp�AddressFamily.AF_INET�SocketKind.SOCK_STREAMr )r� r> �r �r� )r )�flags)r� r> r r� r r� u ��AI_NUMERICSERV)r � getaddrinfor# rO Z EAI_SERVICErc r� r� �lenr � IPV6_ENABLEDr
rd r� r; r0 r ZSOL_TCPZ
AI_PASSIVE� AF_UNSPECr! �UnicodeEncodeErrorr� )
rA r �infoZinfosr r� rs Zsocktyper� �br r r �testGetaddrinfo� s�
� � � � �z"GeneralModuleTests.testGetaddrinfoc C s | � ttjdd� d S )N)zmail.python.orgr r )r! r r re r@ r r r �test_getnameinfo; s z#GeneralModuleTests.test_getnameinfoZnetworkznetwork is not enabledc C sd t �d�� t�d� W d � n1 s*0 Y d}t�|� t�|� t�|dtjtj� d S )Nz
python.orgu! испытание.pythontest.netr )r Ztransient_internetr rE Zgethostbyname_exr� r� r0 )rA r� r r r � test_idna? s (
z GeneralModuleTests.test_idnac
s2 t td�rt td�s � �d� � fdd�}� fdd�}t�� \}}t�tj|�}z�|r`|�d� � �t��* t� d � |�
d
t j � W d � n1 s�0 Y |r�t�tj|� t� d � � �tj
|j
d
t j � W t� d � t�tj|� |�� |�� n*t� d � t�tj|� |�� |�� 0 d S ) N�alarmr� z9signal.alarm and socket.socketpair required for this testc s � � ttjd� d S r� �r! r� �mathZacosh�rR r@ r r �
ok_handlerV s z@GeneralModuleTests.check_sendall_interrupted.<locals>.ok_handlerc s � � ttjd� dd d S �Nr r r� r� r@ r r �raising_handlerX s zEGeneralModuleTests.check_sendall_interrupted.<locals>.raising_handlerg �?r � xr )r
�signalr rc r� �SIGALRMr� r! �ZeroDivisionErrorr� �sendallr �
SOCK_MAX_SIZEr6 r )rA Z with_timeoutr� r� �cr � old_alarmr r@ r �check_sendall_interruptedP s4
.
�
�
z,GeneralModuleTests.check_sendall_interruptedc C s | � d� d S �NF�r� r@ r r r �test_sendall_interruptedo s z+GeneralModuleTests.test_sendall_interruptedc C s | � d� d S )NTr� r@ r r r �%test_sendall_interrupted_with_timeoutr s z8GeneralModuleTests.test_sendall_interrupted_with_timeoutc C s� t � t jt j�}t|�}| �t�� }d }t�� W d � n1 sD0 Y | �|t |j
j d �� t � t jt j�}|� d�}t|�}d }t�� | �t�� d }t�� W d � n1 s�0 Y d S )Nr r
)
r r; r0 r� ry �ResourceWarningr r r r ZwarningrR �makefile)rA r� r r- r r r r �test_dealloc_warnu s &
z$GeneralModuleTests.test_dealloc_warnc C sV t � t jt j��2}|�d�}|�� | �t|�d� W d � n1 sH0 Y d S )Nr
z <_io.BufferedReader name=-1>)r r; r0 r� r r� r� �rA r� �fpr r r �test_name_closed_socketio� s
z,GeneralModuleTests.test_name_closed_socketioc C s� t � � �z}|jddd�}| �|�� � | �|�� � | �|�� � |�� | �t |j� | �t |j� | �t |j� W d � n1 s�0 Y d S )Nr
r )� buffering)
r r� r
�readabler� �writable�seekabler r! r� r� r r r � test_unusable_closed_socketio� s
z0GeneralModuleTests.test_unusable_closed_socketioc C s2 t � � }z�|�tdf� t �|�� � | �t�� |�d� W d � n1 sP0 Y W | �t�� |�� W d � q�1 s�0 Y n4| �t�� |�� W d � n1 s�0 Y 0 | �t�� t �d � W d � n1 s�0 Y | �t�� t �d� W d � n1 �s$0 Y d S )Nr r rY ) r ra r� r r r! r r? r� r� r r r �test_socket_close� s * (� ( ( z$GeneralModuleTests.test_socket_closec C s� dD ]�}| j |d��p t�� �H}|�|�� }| �|j|� W d � n1 sN0 Y W d � n1 sl0 Y W d � q1 s�0 Y qd S )N)r r
Zrw�w�wb��mode)�subTestr r� r� r� )rA r� r� r� r r r �test_makefile_mode� s
z%GeneralModuleTests.test_makefile_modec C s� dD ]�}| j |d��n t�� �F}| �td�� |�|� W d � n1 sL0 Y W d � n1 sj0 Y W d � q1 s�0 Y qd S )N)�rtr] �+r� r� z invalid mode)r� r �assertRaisesRegexr� r� )rA r� r� r r r �test_makefile_invalid_mode� s
z-GeneralModuleTests.test_makefile_invalid_modec C s� t � � }|�6 ttjd �D ]}| �ttj||� q W d � n1 sH0 Y ttjd �D ]H}t�t�t j|��}| � |t j� t�t�t j
|��}| � |t j
� q`d S r� ) r �range�pickleZHIGHEST_PROTOCOLr! r� �dumps�loadsr; r� r0 )rA r� Zprotocolr r� r r r � test_pickle� s 2z GeneralModuleTests.test_picklec C s� dD ]L}t � t jt j��(}|�tdf� |�|� W d � q1 sF0 Y qt � t jt j��&}|�tdf� |�� W d � n1 s�0 Y d S )N)r rY r )r r; r0 ra r� r? )rA Zbacklog�srvr r r �test_listen_backlog� s *z&GeneralModuleTests.test_listen_backlogc C s` dd l }t�tjtj��4}|�tdf� | �t|j|j d � W d � n1 sR0 Y d S r� )
rw r r; r0 ra r� r! r^ r? rx )rA rw r r r r � test_listen_backlog_overflow� s z/GeneralModuleTests.test_listen_backlog_overflow� IPv6 required for this test.c C sf | � ttjtjddfd� t�tjtj��(}| � t|jtjddf� W d � n1 sX0 Y d S )Nr r] i����) r! r^ r re r r� r� r0 ra r 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! r4 r� )rA rs �sockaddrr r r �test_getaddrinfo_ipv6_basic� s � z.GeneralModuleTests.test_getaddrinfo_ipv6_basicr� zdoes not work on Windowsz Symbolic scope id does not workc C sL t �� d \}}t �d| dt jt jt j�\�^ }}| �|ddd|f� d S )Nr �ff02::1de:c0:face:8D%r r )r rQ r� r� r! r4 r� )rA �ifindex�test_interfacers r 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�* r r r r )r r� r r� r! r4 r� )rA r rs r 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 r
r �ff02::1de:c0:face:8d%�1234)r rQ re �NI_NUMERICHOST�NI_NUMERICSERVr� )rA r r r �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 )Nr r
r r r r )r re r r r� r )rA r r r r r r �%test_getnameinfo_ipv6_scopeid_numeric
s z8GeneralModuleTests.test_getnameinfo_ipv6_scopeid_numericc C sX t � t jt j��4}| �t|j�d� | �t|j�d� W d � n1 sJ0 Y d S )Nr� r� )r r; r0 r� r r r� r 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|���}| �|jtj� |�d� | �|jtj� |�d� | �|jtj� |�d� | �|jtj� |�d� | �|jtj� W d � n1 s�0 Y d S )N�
SOCK_NONBLOCKr r� r TF)r} r r0 r; r� r� r� � setblocking)rA r r� Z sock_typer r r r � test_socket_consistent_sock_type s
z3GeneralModuleTests.test_socket_consistent_sock_typec C s� t � t jt j�}|�� }tt jj�� �d }tdd� t jj� � D ��d }t j ||d|d��>}| �
|j |� | �
|j |� | �
|jddh� W d � n1 s�0 Y d S )Nr c s s | ]\}}|d vr|V qdS )> r r� Nr )r� r� �kindr r r r : s �zEGeneralModuleTests.test_unknown_socket_family_repr.<locals>.<genexpr>r� )r r� r r r )r r; r0 �detach�maxZ
AddressFamily� __members__�valuesZ
SocketKindrp r� r r� r r )rA 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 � � ��}t�tjtj�}t�|� | �t j�� |�||�� W d � n1 sb0 Y | �t ��" |�|dd �� W d � n1 s�0 Y | �t
�� |�|d �� W d � n1 s�0 Y W d � n1 s�0 Y d S )Nc @ s e Zd Zdd� Zdd� ZdS )z<GeneralModuleTests.test__sendfile_use_sendfile.<locals>.Filec S s
|| _ d S r3 �r' )rA r' r r r rM K s zEGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.__init__c S s | j S r3 r* r@ r r r r N s zCGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.filenoN)rF rG rH rM r r r r r �FileJ s r+ r� rZ ) r r� r �curdir�O_RDONLYr r! �_GiveupOnSendfile�_sendfile_use_sendfiler^ r� )rA r+ r� r' r r r �test__sendfile_use_sendfileH s
, 0 z.GeneralModuleTests.test__sendfile_use_sendfilec C st | � |j|� | � |j|� |�� }tj|d�}| �|j� |�� | � |j|� | � |j|� | � |�� |� d S �Nr� )r� r r� r r rV r r# )rA r r �styper' �s2r r r �_test_socket_filenoZ s z&GeneralModuleTests._test_socket_filenoc C sP t � t jt j�}| �|j� |�tjdf� | �|t jt j� t t d�r�t � t jt j
�}| �|j� |�tjdf� | �|t jt j
� tj r�t � t j t j�}| �|j� |�tj
dddf� | �|t j t j� t t d��rLt�� }| �tj|� t � t jt j�}| �|j� z|�tj�|d�� W n t�y8 Y n0 | �|t jt j� d S )Nr r! r� r )r r; r0 rV r ra r r� r4 r
r! r� r� r� r� r� �shutilZrmtreer� r� r� r r� )rA r Ztmpdirr r r �test_socket_filenog s4
�z%GeneralModuleTests.test_socket_filenoc C sD | � td��$ tjtjtjdd� W d � n1 s60 Y d S )Nzinteger argument expectedg @E@r� �r r� r r; r0 r@ r r r � test_socket_fileno_rejects_float� s z3GeneralModuleTests.test_socket_fileno_rejects_floatc C sD | � td��$ tjtjtjdd� W d � n1 s60 Y d S )Nzinteger is required�foor� r7 r@ r r r �&test_socket_fileno_rejects_other_types� s z9GeneralModuleTests.test_socket_fileno_rejects_other_typesc C sD | � td��$ tjtjtjdd� W d � n1 s60 Y d S )N�negative file descriptorrY r� �r r� r r; r0 r@ r r r �)test_socket_fileno_rejects_invalid_socket� s z<GeneralModuleTests.test_socket_fileno_rejects_invalid_socketzWindows disallows -1 onlyc C sD | � td��$ tjtjtjdd� W d � n1 s60 Y d S )Nr; i���r� r< r@ r r r �#test_socket_fileno_rejects_negative� s z6GeneralModuleTests.test_socket_fileno_rejects_negativec C s� d}| � t�� }tjt�� d� W d � n1 s40 Y | �|jjtj|f� | � t��(}tjtj tj
t�� d� W d � n1 s�0 Y | �|jjtj|f� d S )Ni6' r� ) r! r r r Z make_bad_fdr r+ rO r� r; r0 )rA Z WSAENOTSOCKr- r r r �$test_socket_fileno_requires_valid_fd� s . �$z7GeneralModuleTests.test_socket_fileno_requires_valid_fdc C s� t �� ��}| �t�� tj|�� d� W d � n1 s:0 Y | �t��(}tjtjtj|�� d� W d � n1 s|0 Y | �|j j
t
j � W d � n1 s�0 Y d S r1 ) r� ZNamedTemporaryFiler! r r r r; r0 r� r+ rO r� )rA Zafiler- r r r �%test_socket_fileno_requires_socket_fd� s
. �$z8GeneralModuleTests.test_socket_fileno_requires_socket_fd)YrF rG rH r r r� �
skipUnlessr r r r% r. r3 r6 r� r� r9 rd r� r r� rB rI rK rL r
r rP rX ra rb rg rh rt 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� r� r� r� r� r r r r
r r �skipIfr r r r r r r! r( r0 r4 r6 r8 r: r= r> r? r@ r r r r r I 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 r3 )r �AF_CANr r r@ r r r r3 � s z!BasicCANTest.testCrucialConstants�CAN_BCM�&socket.CAN_BCM required for this test.c C s� 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 r3 )r rF 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_FRAMEr@ r r r �testBCMConstants� s0 z BasicCANTest.testBCMConstantsc C s8 t � t jt jt j��}W d � n1 s*0 Y d S r3 �r r r r r r r r �testCreateSocket� s z BasicCANTest.testCreateSocketc C s8 t � t jt jt j��}W d � n1 s*0 Y d S r3 )r r r! rF r r r r �testCreateBCMSocket� s z BasicCANTest.testCreateBCMSocketc C sV t � t jt jt j��.}d}|�|� | �|�� |� W d � n1 sH0 Y d S )N)r& )r r r r ra r� r� )rA r �addressr r r � testBindAny� s
zBasicCANTest.testBindAnyc C sJ t � t jt jt j��"}| �td|jd� W d � n1 s<0 Y d S )N�interface name too long)� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx)r r r r r r ra r r r r �testTooLongInterfaceName� s �z%BasicCANTest.testTooLongInterfaceName�CAN_RAW_LOOPBACKz/socket.CAN_RAW_LOOPBACK required for this test.c C sl t � t jt jt j��D}dD ].}|�t jt j|� | �||�t jt j�� qW d � n1 s^0 Y d S )N)r r ) r r r r r� � SOL_CAN_RAWrR r� r� )rA 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��R}|�tjtj|� | � ||�
tjtjd�� |�tjtjt |�� W d � n1 s�0 Y d S )N)� i z=IIr^ ) r �packr r r r r� rS rU r� r� r� )rA �can_idZcan_maskZ
can_filterr r r r �
testFilter� s �zBasicCANTest.testFilterN)rF rG rH r3 r� rA r
r rI rK rL rN rQ rT rY r r r r rD � s( �
�
�
�rD 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 s t j| |d� d S r� )r� rM r� r r r rM s zCANTest.__init__c C s& t |�}|�dd�}t�| j|||�S )zBuild a CAN frame.r^ � )r� �ljustr rW re )�clsrX �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 re )r] �framerX r_ r^ r r r �dissect_can_frame s zCANTest.dissect_can_framec C s6 | j �| j�\}}| �| j|� | �|d | j� d S r� )r �recvfromrd r� �cfrb �rA rd r� r r r �
testSendFrame s zCANTest.testSendFramec C s | � dd�| _| j�| j� d S )Nr s �r` rd r� r� r@ r r r �_testSendFrame s zCANTest._testSendFramec C s$ | j �| j�\}}| �| j|� d S r3 )r rc rd r� rd re r r r �testSendMaxFrame! s zCANTest.testSendMaxFramec C s | � dd�| _| j�| j� d S )Nr s rg r@ r r r �_testSendMaxFrame% s zCANTest._testSendMaxFramec C sD | j �| j�\}}| �| j|� | j �| j�\}}| �| j|� d S r3 )r rc rd r� �cf1�cf2re r r r �testSendMultiFrames) s zCANTest.testSendMultiFramesc C s< | � dd�| _| j�| j� | � dd�| _| j�| j� d S )N� s D3"� s �"3)r` rk r� r� rl r@ r r r �_testSendMultiFrames0 s z CANTest._testSendMultiFramesrF rG c C sP | j �| j�\}}| �| j|� | �|�\}}}| �| j|� | �| j|� d S r3 )r� rc rd r� rd rb rX r^ )rA rd r� rX r_ r^ r r r �_testBCM7 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! rF rV r r� rb rX �bytesr^ r` rd rH r rW rf r� r� r� )rA ZbcmZopcoder� r� Z
ival1_secondsZ
ival1_usecZ
ival2_secondsZ
ival2_usecZ
bcm_can_idZnframes�headerZheader_plus_frameZ
bytes_sentr r r �testBCM@ s4 �
zCANTest.testBCMN)r� )rF rG rH rM � classmethodr` rb rf rh ri rj rm rp r� rA r
r rq rx r r r r rZ s&
�
�rZ 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 s t � j|i |�� d| _d S �Nr[ �rL rM rb rQ rT r r rM c s zISOTPTest.__init__c C s t j t j t j t j d S r3 )r rE r r" r! r@ r r r r3 g s z ISOTPTest.testCrucialConstantsc C s8 t � t jt jt j��}W d � n1 s*0 Y d S r3 rJ r r r r rK m s zISOTPTest.testCreateSocketr" z(socket.CAN_ISOTP required for this test.c C s8 t � t jt jt j��}W d � n1 s*0 Y d S r3 )r r r! r" r r r r �testCreateISOTPSocketq s z ISOTPTest.testCreateISOTPSocketc C sn t � t jt jt j��F}| �td�� |�d� W d � n1 sB0 Y W d � n1 s`0 Y d S )NrO )rP r r� )r r r! r" r r ra r r r r rQ w s z"ISOTPTest.testTooLongInterfaceNamec
C s� z^t � t jt jt j��6}| jddf}|�|� | �|�� |� W d � n1 sR0 Y W nD ty� } z,|j t j
kr�| � d| j � n� W Y d }~n
d }~0 0 d S )Nrr iV r` ) r r r! r" rb ra r� r� r rO �ENODEVrc �rA r r� r� r r r �testBind} s
2 �zISOTPTest.testBind)rF rG rH rM r3 rK r� rA r
r r} rQ r� rY r r rT r rz ` s �
rz 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 s t � j|i |�� d| _d S r{ r| rQ rT r r rM � s zJ1939Test.__init__r$ z(socket.CAN_J1939 required for this test.c C s� 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 r3 )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_MAXr@ r r r �testJ1939Constants� s0 z J1939Test.testJ1939Constantsc C s8 t � t jt jt j��}W d � n1 s*0 Y d S r3 )r r r! r$ r r r r �testCreateJ1939Socket� s z J1939Test.testCreateJ1939Socketc
C s� zft � t jt jt j��>}| jt jt jt jf}|�|� | � |�
� |� W d � n1 sZ0 Y W nD t y� } z,|j t j
kr�| �d| j � n� W Y d }~n
d }~0 0 d S r_ )r r r! r$ rb r� r� r� ra r� r� r rO r~ rc r r r r r� � s
2 �zJ1939Test.testBind) rF rG rH rM r� rA r
r r� r� r� rY r r rT 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 r3 )r ZAF_RDSr& r@ r r r r3 � s z!BasicRDSTest.testCrucialConstantsc C s6 t � t jt jd��}W d � n1 s(0 Y d S r� )r r& r' r r r r rK � s z BasicRDSTest.testCreateSocketc C s^ d}t � t jt jd��4}|�t jt j|� |�t jt j|� W d � n1 sP0 Y d S )Ni @ r )r r& r' r� r1 Z SO_RCVBUFZ SO_SNDBUF)rA rd r r r r �testSocketBufferSize� s z!BasicRDSTest.testSocketBufferSizeN)rF rG rH r3 rK 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 s t j| |d� d S r� )r� rM r� r r r rM � s zRDSTest.__init__c s t � �� t�� | _d S r3 )rL rB rN rv Zevtr@ rT r r rB � s
z
RDSTest.setUpc C s2 | j �| j�\}}| �| j|� | �| j|� d S r3 )r< rc rd r� r^ r� �rA r^ r� r r r �testSendAndRecv� s zRDSTest.testSendAndRecvc C s" d| _ | j�| j dt| jf� d S �Nr� r �r^ r� r* r� r> r@ r r r �_testSendAndRecv� s zRDSTest._testSendAndRecvc C sH | j �| jtj�\}}| �| j|� | j �| j�\}}| �| j|� d S r3 )r< rc rd r �MSG_PEEKr� r^ r� r r r �testPeek� s zRDSTest.testPeekc C s" d| _ | j�| j dt| jf� d S r� r� r@ r r r � _testPeek� s zRDSTest._testPeek�recvmsgc C s( | j �| j�\}}}}| �| j|� d S r3 )r< r� rd r� r^ )rA 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� r> r@ r r r �_testSendAndRecvMsg� s zRDSTest._testSendAndRecvMsgc C sD | j �| j�\}}| �| j|� | j �| j�\}}| �| j|� d S r3 )r< rc rd r� �data1�data2r� r r r �testSendAndRecvMulti s z RDSTest.testSendAndRecvMultic C s@ d| _ | j�| j dt| jf� d| _| j�| jdt| jf� d S )Ns baconr s egg)r� r� r* r� r> r� r@ r r r �_testSendAndRecvMulti s z RDSTest._testSendAndRecvMultic C sL t � | jgg g d�\}}}| �| j|� | j�| j�\}}| �| j|� d S )Ng @)�selectr< r rc rd r� r^ )rA r r� r] r^ r� r r r �
testSelect s zRDSTest.testSelectc C s" d| _ | j�| j dt| jf� d S )Ns selectr r� r@ r r r � _testSelect s zRDSTest._testSelect)r� )rF rG rH rM rB r� r� r� r� r r r� r� r� r� r� r� rY r r rT 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 s
t j d S r3 )r r+ r@ r r r r3 s z%BasicQIPCRTRTest.testCrucialConstantsc C s4 t � t jt j��}W d � n1 s&0 Y d S r3 )r r+ r! r r r r rK ! s z!BasicQIPCRTRTest.testCreateSocketc C sH t � t jt j��$}| �|�� d d� W d � n1 s:0 Y d S �Nr r )r r+ r! r� r� r r r r � testUnbound% s z BasicQIPCRTRTest.testUnboundc C s^ t � t jt j��:}tj||�� d d� | �|�� d d� W d � n1 sP0 Y d S )Nr r� r )r r+ r! r r= r� �assertNotEqualr r r r � testBindSock) s z BasicQIPCRTRTest.testBindSockc C sH t � t jt j��$}| jttj|dd� W d � n1 s:0 Y d S )Nru r� )r r+ r! r! r r r= r r r r �testInvalidBindSock. s z$BasicQIPCRTRTest.testInvalidBindSockc C sR t � t jt j��.}|�d� | �|�� d d� W d � n1 sD0 Y d S )N)�{ r� r r )r r+ r! r� r� r� r r r r �testAutoBindSock2 s
z!BasicQIPCRTRTest.testAutoBindSockN) rF rG rH r3 rK r� r� r� r� r r r r r� s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�BasicVSOCKTestc C s
t j d S r3 )r r� r@ r r r r3 < s z#BasicVSOCKTest.testCrucialConstantsc C s4 t j t j t j t j t j t j t j t j d S r3 ) 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 r@ r r r �testVSOCKConstants? s z!BasicVSOCKTest.testVSOCKConstantsc C s4 t � t jt j��}W d � n1 s&0 Y d S r3 )r r� r0 r r r r rK I s z BasicVSOCKTest.testCreateSocketc C s� t � t jt j���}|�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 � n1 s�0 Y d S �Nr� ) r r� r0 r� r� r� r� r� r� )rA r Zorig_maxrf Zorig_minr r r r� M sD ���
�
�
�
��
��
��z#BasicVSOCKTest.testSocketBufferSizeN)rF rG rH r3 r� rK r� r r r r r� 7 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�ejdkpdej� d �d
�dd� �Z dS )�BasicBluetoothTestc C sJ t j t j t j t j tjdkrFt j t j t j tj�
d�sFt j d S )Nr� r} ) r Z
BDADDR_ANYZ BDADDR_LOCALr/ r1 rd r� � BTPROTO_HCIZSOL_HCI�
BTPROTO_L2CAPr� � BTPROTO_SCOr@ r r r �testBluetoothConstantsl s
z)BasicBluetoothTest.testBluetoothConstantsc C s8 t � t jt jt j��}W d � n1 s*0 Y d S r3 )r r/ r0 r1 r r r r �testCreateRfcommSocketz s z)BasicBluetoothTest.testCreateRfcommSocketr� z&windows does not support L2CAP socketsc C s8 t � t jt jt j��}W d � n1 s*0 Y d S r3 )r r/ r' r� r r r r �testCreateL2capSocket~ s z(BasicBluetoothTest.testCreateL2capSocketz$windows does not support HCI socketsc C s8 t � t jt jt j��}W d � n1 s*0 Y d S r3 )r r/ r r� r r r r �testCreateHciSocket� s z&BasicBluetoothTest.testCreateHciSocketr} z.windows and freebsd do not support SCO socketsc C s8 t � t jt jt j��}W d � n1 s*0 Y d S r3 )r r/ r' r� r r r r �testCreateScoSocket� s z&BasicBluetoothTest.testCreateScoSocketN)
rF rG rH r� r� r� rC rd r� r� r� r� r� r r r r r� h 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 s t j| |d� d S r� �r� rM r� r r r rM � s zBasicTCPTest.__init__c C s | j �d�}| �|t� d S r� )r� r� r� r� r$ r r r �testRecv� s zBasicTCPTest.testRecvc C s | j �t� d S r3 �r� r� r� r@ r r r � _testRecv� s zBasicTCPTest._testRecvc C s8 | j �tt�d �}| j �d�}|| }| �|t� d S �N� r� )r� r� r� r� r� )rA �seg1�seg2r� r r r �testOverFlowRecv� s z BasicTCPTest.testOverFlowRecvc C s | j �t� d S r3 r� r@ r r r �_testOverFlowRecv� s z BasicTCPTest._testOverFlowRecvc C s | j �d�\}}| �|t� d S r� )r� rc r� r� �rA r� r� r r r � testRecvFrom� s zBasicTCPTest.testRecvFromc C s | j �t� d S r3 r� r@ r r r �
_testRecvFrom� s zBasicTCPTest._testRecvFromc C s@ | j �tt�d �\}}| j �d�\}}|| }| �|t� d S r� )r� rc r� r� r� )rA r� r� r� r� r r r �testOverFlowRecvFrom� s z!BasicTCPTest.testOverFlowRecvFromc C s | j �t� d S r3 r� r@ r r r �_testOverFlowRecvFrom� s z"BasicTCPTest._testOverFlowRecvFromc C s0 d}| j �d�}|sq ||7 }q| �|d� d S )Nr� r� � ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff)r� r� r� )rA r� �readr r r � testSendAll� s
zBasicTCPTest.testSendAllc C s d}| j �|� d S )Nr� )r� r� )rA Z big_chunkr r r � _testSendAll� s zBasicTCPTest._testSendAllc C sP | j �� }t�|tjtj�}| �|j� | �|tj� |� d�}| �
|t � d S r� ) r� r r Zfromfdr; r0 rV r r r� r� r� )rA r' r� r� r r r �
testFromFd� s
zBasicTCPTest.testFromFdc C s | j �t� d S r3 r� r@ r r r � _testFromFd� s zBasicTCPTest._testFromFdc C s0 | j �� }| �|j� |�d�}| �|t� d S r� )r� �duprV r r� r� r� )rA r� r� r r r �testDup� s
zBasicTCPTest.testDupc C s | j �t� d S r3 r� r@ r r r �_testDup� s zBasicTCPTest._testDupc C s& | j �d�}| �|t� | j�� d S r� )r� r� r� r� rx r� r$ r r r � testShutdown� s zBasicTCPTest.testShutdownc C s | j �t� | j �d� d S r� )r� r� r� �shutdownr@ r r r �
_testShutdown� s zBasicTCPTest._testShutdownc C sX dd l }| j�t� | �t| jj|jd � | �t| jjd|jd � | j�d� d S )Nr r r� ) rw r� r� r� r! r^ r� rx �UINT_MAX�rA rw 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� r r# r� r
�_closedr! r r� r r r; r0 rV r� )rA r r r� r� r r r �
testDetach� s
zBasicTCPTest.testDetachc C s | j �t� d S r3 r� r@ r r r � _testDetach
s zBasicTCPTest._testDetachN)r� )rF rG rH rM r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r rB ZtestShutdown_overflowr� r� r� r r r r r� � s,
r� c @ sF e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dS )� BasicUDPTestr� c C s t j| |d� d S r� )r� rM r� r r r rM
s zBasicUDPTest.__init__c C s | j �tt��}| �|t� d S r3 �r< r� r� r� r� r$ r r r �testSendtoAndRecv
s z BasicUDPTest.testSendtoAndRecvc C s | j �tdt| jf� d S r� �r� r* r� r� r> r@ r r r �_testSendtoAndRecv
s z BasicUDPTest._testSendtoAndRecvc C s$ | j �tt��\}}| �|t� d S r3 �r< rc r� r� r� r� r r r r�
s zBasicUDPTest.testRecvFromc C s | j �tdt| jf� d S r� r� r@ r r r r�
s zBasicUDPTest._testRecvFromc C s | � t| jjd� d S �NrY �r! r� r< rc r@ r r r �testRecvFromNegative
s z!BasicUDPTest.testRecvFromNegativec C s | j �tdt| jf� d S r� r� r@ r r r �_testRecvFromNegative#
s z"BasicUDPTest._testRecvFromNegativeN)r� �
rF rG rH rM r� r� r� r� r� r� r r r r r�
s
r� c @ sF e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dS )�BasicUDPLITETestr� c C s t j| |d� d S r� )r� rM r� r r r rM +
s zBasicUDPLITETest.__init__c C s | j �tt��}| �|t� d S r3 r� r$ r r r r� .
s z"BasicUDPLITETest.testSendtoAndRecvc C s | j �tdt| jf� d S r� r� r@ r r r r� 3
s z#BasicUDPLITETest._testSendtoAndRecvc C s$ | j �tt��\}}| �|t� d S r3 r� r� r r r r� 6
s z BasicUDPLITETest.testRecvFromc C s | j �tdt| jf� d S r� r� r@ r r r r� ;
s z BasicUDPLITETest._testRecvFromc C s | � t| jjd� d S r� r� r@ r r r r� >
s z%BasicUDPLITETest.testRecvFromNegativec C s | j �tdt| jf� d S r� r� r@ r r r r� B
s z&BasicUDPLITETest._testRecvFromNegativeN)r� r� r r r r r� '
s
r� c s� e Zd ZejZ� fdd�Zdd� ZdZdd� Z dd � Z
d
d � Z d d
� Z dZ
dD ]Ze
eeed�O Z
qNdZdZdZddd�Z� ZS )�SendrecvmsgBasec s t �� | _t� �� d S r3 )rN rv �
misc_eventrL rB r@ rT r r rB g
s
zSendrecvmsgBase.setUpc C s | j �|�S r3 )�cli_sockr� r$ r r r � sendToServerk
s z SendrecvmsgBase.sendToServerr c G s | j j|| jt|�d � � S r3 )r� r� �sendmsg_to_server_defaultsr� �rA rR r r r �sendmsgToServers
s �z SendrecvmsgBase.sendmsgToServerc G s |j |g|�R � }| �|� |S r3 )r� �registerRecvmsgResult)rA r� rd rR �resultr r r � doRecvmsg{
s
zSendrecvmsgBase.doRecvmsgc C s d S r3 r �rA r� r r r r� �
s z%SendrecvmsgBase.registerRecvmsgResultc C s | � ||� d S r3 �r� �rA Zaddr1Zaddr2r r r �checkRecvmsgAddress�
s z#SendrecvmsgBase.checkRecvmsgAddressr )�
MSG_CTRUNCZMSG_OOBNc
C s� | j }| j}|r&|| jO }|| jO }n |d urB|| jO }|| jO }|| M }|| M }||O }||O }||@ | @ }|r�td�|���||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� )
rA r� �eor�checkset�
checkunset�ignoreZ
defaultsetZ defaultunsetZinbothrs r r r �
checkFlags�
s&
�zSendrecvmsgBase.checkFlags)Nr r r )rF rG rH r �LOOPBACK_TIMEOUT� fail_timeoutrB r� r� r� r� r� r� r� r� r} r r� r� r� r� rY r r rT r r� `
s
r� c @ s e Zd Zdd� ZdS )�RecvmsgIntoMixinc G sh t |�}|j|gg|�R � }| �|� | �|d d� | �|d |� t|d |d � �f|dd � S r� )r� � recvmsg_intor� �assertGreaterEqual�assertLessEqualrv )rA r� rd rR �bufr� r r r r� �
s
zRecvmsgIntoMixin.doRecvmsgN)rF rG rH r� r r r r r� �
s r� c s e Zd Ze� fdd��Z� ZS )�SendrecvmsgDgramFlagsBasec s t � jtjB S r3 )rL r� r � MSG_TRUNCr@ rT r r r� �
s z5SendrecvmsgDgramFlagsBase.msg_flags_non_eor_indicator)rF rG rH �propertyr� rY r r rT r r �
s r c s e Zd Ze� fdd��Z� ZS )�SendrecvmsgSCTPFlagsBasec s t � jtjB S r3 )rL r� r ZMSG_EORr@ rT r r r� �
s z0SendrecvmsgSCTPFlagsBase.msg_flags_eor_indicator)rF rG rH r r� rY r r rT r r �
s r c @ s8 e Zd Zedd� �Zedd� �Zedd� �Zdd� Zd S )
� SendrecvmsgConnectionlessBasec C s | j S r3 )r< r@ r r r � serv_sock�
s z'SendrecvmsgConnectionlessBase.serv_sockc C s | j S r3 )r� r@ r r r r� �
s z&SendrecvmsgConnectionlessBase.cli_sockc C s g g d| j fS r� )r� r@ r r r r� s z8SendrecvmsgConnectionlessBase.sendmsg_to_server_defaultsc C s | j �|| j�S r3 )r� r* r� r$ r r r r� s z*SendrecvmsgConnectionlessBase.sendToServerN)rF rG rH r r r� r� r� r r r r r �
s
r c @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )�SendrecvmsgConnectedBasec C s | j S r3 )r� r@ r r r r s z"SendrecvmsgConnectedBase.serv_sockc C s | j S r3 )r� r@ r r r r� s z!SendrecvmsgConnectedBase.cli_sockc C s d S r3 r r� r r r r� s z,SendrecvmsgConnectedBase.checkRecvmsgAddressN)rF rG rH r r r� r� r r r r r s
r c s e Zd Z� fdd�Z� ZS )� SendrecvmsgServerTimeoutBasec s t � �� | j�| j� d S r3 )rL rB r r� r� r@ rT r r rB s
z"SendrecvmsgServerTimeoutBase.setUp)rF rG rH rB rY r r rT r r
s r
c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zeed �d d � �Zejd d!� �Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*S )+� SendmsgTestsc C s | � | j�tt��t� d S r3 �r� r r� r� r� r@ r r r � testSendmsg( s zSendmsgTests.testSendmsgc C s | � | �tg�tt�� d S r3 �r� r� r� r� r@ r r r � _testSendmsg, s zSendmsgTests._testSendmsgc C s | � | j�tt��t� d S r3 r r@ r r r �testSendmsgDataGenerator/ s z%SendmsgTests.testSendmsgDataGeneratorc C s& | � | �dd� tfD ��tt�� d S )Nc s s | ]
}|V qd S r3 r r r r r r 4 r� z9SendmsgTests._testSendmsgDataGenerator.<locals>.<genexpr>r r@ r r r �_testSendmsgDataGenerator3 s �z&SendmsgTests._testSendmsgDataGeneratorc C s | � | j�tt��t� d S r3 r r@ r r r � testSendmsgAncillaryGenerator7 s z*SendmsgTests.testSendmsgAncillaryGeneratorc C s( | � | �tgdd� dD ��tt�� d S )Nc s s | ]
}|V qd S r3 r r r r r r < r� z>SendmsgTests._testSendmsgAncillaryGenerator.<locals>.<genexpr>r r r@ r r r � _testSendmsgAncillaryGenerator; s �z+SendmsgTests._testSendmsgAncillaryGeneratorc C s | � | j�tt��t� d S r3 r r@ r r r �testSendmsgArray? s z SendmsgTests.testSendmsgArrayc C s$ | � | �t�dt�g�tt�� d S )N�B)r� r� �arrayr� r� r@ r r r �_testSendmsgArrayC s �z SendmsgTests._testSendmsgArrayc C s | � | j�tt��t� d S r3 r r@ r r r �testSendmsgGatherG s z SendmsgTests.testSendmsgGatherc C s. | � | �td d� tdd � g�tt�� d S �Nr� r r@ r r r �_testSendmsgGatherK s z SendmsgTests._testSendmsgGatherc C s | � | j�d�d� d S �NrZ � done�r� r r� r@ r r r �testSendmsgBadArgsN s 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 )r! r� r� r� r� �objectr� r� r@ r r r �_testSendmsgBadArgsR s.
�
�
�
�
�
�
�z SendmsgTests._testSendmsgBadArgsc C s | � | j�d�d� d S r r r@ r r r �testSendmsgBadCmsgd s 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" r r �r! r� r� r� r r� r@ r r r �_testSendmsgBadCmsgh s&
�
�
�
�
�
�z SendmsgTests._testSendmsgBadCmsg�
CMSG_SPACEc C s | � | j�d�d� d S r r r@ r r r �testSendmsgBadMultiCmsgw s z$SendmsgTests.testSendmsgBadMultiCmsgc C s@ | � t| jtgg d�� | � t| jtgdt� g� | �d� d S )N�r r r� r r# r@ r r r �_testSendmsgBadMultiCmsg} s
�
�z%SendmsgTests._testSendmsgBadMultiCmsgc C s | � | j�d�d� d S r r r@ r r r �testSendmsgExcessCmsgReject� s z(SendmsgTests.testSendmsgExcessCmsgRejectc C sb t td�sT| �t��"}| �tgddg� W d � n1 s<0 Y | �|jj� | � d� d S )Nr% r' r )
r
r r! r r� r� � assertIsNoner+ rO r� �rA r- r r r � _testSendmsgExcessCmsgReject� s
0z)SendmsgTests._testSendmsgExcessCmsgRejectc C s d S r3 r r@ r r r �testSendmsgAfterClose� s z"SendmsgTests.testSendmsgAfterClosec C s | j �� | �t| jtg� d S r3 )r� r r! r r� r� r@ r r r �_testSendmsgAfterClose� s
z#SendmsgTests._testSendmsgAfterCloseN)rF rG rH r
r r r r r r r r r r r r! r$ r r r& r� r( r) r, r- r. r r r r r $ s,
r c @ sR e Zd Zdd� Zdd� Zdd� Zdd� Zeej d vd
�d d � �Z
e
j d
d� �Z dS )�SendmsgStreamTestsc C s | � | j�tt��t� d S r3 r r@ r r r �testSendmsgExplicitNoneAddr� s z.SendmsgStreamTests.testSendmsgExplicitNoneAddrc C s" | � | �tgg dd �tt�� d S r� r r@ r r r � _testSendmsgExplicitNoneAddr� s z/SendmsgStreamTests._testSendmsgExplicitNoneAddrc C s. | � | j�d�d� | �| jj| jd�� d S �NrV � aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa�r6 �r� r r� r
r� r� r� r@ r r r �testSendmsgTimeout� s z%SendmsgStreamTests.testSendmsgTimeoutc
C s� z|| j �d� z| �dg� qW nD tjy4 Y n< tyd } z|jtjkrP� W Y d }~nd }~0 0 | �d� W | j �
� n | j �
� 0 d S )N� ��Q��?r3 �socket.timeout not raised) r� r� r� r r6 r rO �ENOMEMr r� rm �rA r� r r r �_testSendmsgTimeout� s z&SendmsgStreamTests._testSendmsgTimeout> r~ z<MSG_DONTWAIT not known to work on this platform when sendingc C s. | � | j�d�d� | �| jj| jd�� d S r2 r5 r@ r r r �testSendmsgDontWait� s z&SendmsgStreamTests.testSendmsgDontWaitc C sz zh| � t��$}| �dgg tj� qW d � n1 s60 Y | �|jjtjtj tj
f� W | j � � n | j � � 0 d S �Nr3 )
r! r r� r Z MSG_DONTWAITr r+ rO ZEAGAINZ EWOULDBLOCKr9 r� rm r+ r r r �_testSendmsgDontWait� s 2
�z'SendmsgStreamTests._testSendmsgDontWaitN)
rF rG rH r0 r1 r6 r; r� rd r� r< r� r> r r r r r/ � s
�
r/ c @ s e Zd Zdd� Zdd� ZdS )�SendmsgConnectionlessTestsc C s d S r3 r r@ r r r �testSendmsgNoDestAddr� s z0SendmsgConnectionlessTests.testSendmsgNoDestAddrc C s2 | � t| jjtg� | � t| jjtgg dd � d S r� )r! r r� r� r� r@ r r r �_testSendmsgNoDestAddr� s �
�z1SendmsgConnectionlessTests._testSendmsgNoDestAddrN)rF rG rH r@ rA r r r r r? � s r? c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zeed!�d"d#� �Zejd$d%� �Zeejd&�d'd(� �Zejd)d*� �Zd+S ),�RecvmsgGenericTestsc C sR | � | jtt��\}}}}| �|t� | �|| j� | �|g � | j|dd� d S �NT�r� �r� r r� r� r� r� r� r� �rA r� r� r� r� r r r � testRecvmsg� s
z RecvmsgGenericTests.testRecvmsgc C s | � t� d S r3 �r� r� r@ r r r � _testRecvmsg� s z RecvmsgGenericTests._testRecvmsgc C sV | � | jtt�dd�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr TrD rE rF r r r �testRecvmsgExplicitDefaults� s
� z/RecvmsgGenericTests.testRecvmsgExplicitDefaultsc C s | � t� d S r3 rH r@ r r r � _testRecvmsgExplicitDefaults s z0RecvmsgGenericTests._testRecvmsgExplicitDefaultsc C sV | � | jtt�d �\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr TrD rE rF r r r �testRecvmsgShorter s
� z&RecvmsgGenericTests.testRecvmsgShorterc C s | � t� d S r3 rH r@ r r r �_testRecvmsgShorter
s z'RecvmsgGenericTests._testRecvmsgShorterc C s^ | � | jtt�d �\}}}}| �|td d� � | �|| j� | �|g � | j|dd� d S )Nr� ���FrD rE rF r r r �testRecvmsgTrunc s
� z$RecvmsgGenericTests.testRecvmsgTruncc C s | � t� d S r3 rH r@ r r r �_testRecvmsgTrunc s z%RecvmsgGenericTests._testRecvmsgTruncc C sT | � | jtt�d�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr TrD rE rF r r r � testRecvmsgShortAncillaryBuf s � z0RecvmsgGenericTests.testRecvmsgShortAncillaryBufc C s | � t� d S r3 rH r@ r r r � _testRecvmsgShortAncillaryBuf% s z1RecvmsgGenericTests._testRecvmsgShortAncillaryBufc C sT | � | jtt�d�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )N� ( TrD rE rF r r r �testRecvmsgLongAncillaryBuf( s � z/RecvmsgGenericTests.testRecvmsgLongAncillaryBufc C s | � t� d S r3 rH r@ r r r � _testRecvmsgLongAncillaryBuf1 s z0RecvmsgGenericTests._testRecvmsgLongAncillaryBufc C s" | j �� | �t| j| j d� d S r� )r r r! r r� r@ r r r �testRecvmsgAfterClose4 s
z)RecvmsgGenericTests.testRecvmsgAfterClosec C s d S r3 r r@ r r r �_testRecvmsgAfterClose9 s z*RecvmsgGenericTests._testRecvmsgAfterClosec C sF z4| j �d� | �tj| j| j tt�� W | j� � n | j� � 0 d S )Nr7 )
r r� r! r r6 r� r� r� r� rm r@ r r r �testRecvmsgTimeout< s �z&RecvmsgGenericTests.testRecvmsgTimeoutc C s | � | jj| jd�� d S �Nr4 )r
r� r� r� r@ r r r �_testRecvmsgTimeoutE s 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 rN Fr �r� r� TrD ) r� r r� r� r r� r� r� r� r� r} rF r r r �testRecvmsgPeekH s. �
�� z#RecvmsgGenericTests.testRecvmsgPeekc C s | � t� d S r3 rH r@ r r r �_testRecvmsgPeekk s z$RecvmsgGenericTests._testRecvmsgPeekr� c C s` | j �| j� | �| j tt��\}}}}| �|t� | �|| j� | �|g � | j |dd� d S rC )
r r� r� r� r� r� r� r� r� r� rF r r r �testRecvmsgFromSendmsgo s z*RecvmsgGenericTests.testRecvmsgFromSendmsgc C s. | � | �td d� tdd � g�tt�� d S r r r@ r r r �_testRecvmsgFromSendmsgz s z+RecvmsgGenericTests._testRecvmsgFromSendmsgN)rF rG rH rG rI rJ rK rL rM rO rP rQ rR rT rU rV rW rX rZ r r r\ r� r] r^ r_ r r r r rB � s0
"
rB c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�RecvmsgGenericStreamTestsc C sN | � | jd�\}}}}| �|d� | �|| j� | �|g � | j|d d� d S )Nr� r� rD )r� r r� r� r� r� rF r r r �testRecvmsgEOF� s
z(RecvmsgGenericStreamTests.testRecvmsgEOFc C s | j �� d S r3 )r� r r@ r r r �_testRecvmsgEOF� s z)RecvmsgGenericStreamTests._testRecvmsgEOFc C s� | � | jtt�d �\}}}}| �|| j� | �|g � | j|dd� | � | jd�\}}}}| �|| j� | �|g � | j|dd� || }| �|t� d S )Nr� FrD r� T)r� r r� r� r� r� r� r� )rA r� r� r� r� r� r� r r r �testRecvmsgOverflow� s
� z-RecvmsgGenericStreamTests.testRecvmsgOverflowc C s | � t� d S r3 rH r@ r r r �_testRecvmsgOverflow� s z.RecvmsgGenericStreamTests._testRecvmsgOverflowN)rF rG rH ra rb rc rd r r r r r` s r` c @ s e Zd Zdd� Zdd� ZdS )� 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 )NrY r r� TrD )
r! r� r r� r� r� r� r� r r� r� r� r� rF r r r �testRecvmsgBadArgs� s0 �
� � � � � z RecvmsgTests.testRecvmsgBadArgsc C s | � t� d S r3 rH r@ r r r �_testRecvmsgBadArgs� s z RecvmsgTests._testRecvmsgBadArgsN)rF rG rH rf rg r r r r re � s re 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 writablerY TrD )
r� r� r� r! r� r r� r r� r� r� r� r� �rA r �nbytesr� r� r� r r r �testRecvmsgIntoBadArgs� s@
� �
� � � �
�
� z'RecvmsgIntoTests.testRecvmsgIntoBadArgsc C s | � t� d S r3 rH r@ r r r �_testRecvmsgIntoBadArgs� s z(RecvmsgIntoTests._testRecvmsgIntoBadArgsc C sx t tt��}| j�dd� |fD ��\}}}}| �|tt�� | �|t t�� | �|| j� | �|g � | j|dd� d S )Nc s s | ]
}|V qd S r3 r r r r r r � r� z<RecvmsgIntoTests.testRecvmsgIntoGenerator.<locals>.<genexpr>TrD ) r� r� r� r r� r� r� r� r� ri r r r �testRecvmsgIntoGenerator� s � z)RecvmsgIntoTests.testRecvmsgIntoGeneratorc C s | � t� d S r3 rH r@ r r r �_testRecvmsgIntoGenerator� s z*RecvmsgIntoTests._testRecvmsgIntoGeneratorc C sx t � ddgtt� �}| j�|g�\}}}}| �|tt�� | �|�� t� | �|| j� | �|g � | j |dd� d S )Nr r TrD )
r r� r� r r� r� �tobytesr� r� r� ri r r r �testRecvmsgIntoArray� s z%RecvmsgIntoTests.testRecvmsgIntoArrayc C s | � t� d S r3 rH r@ r r r �_testRecvmsgIntoArray� s 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---TrD ) r� r r� �
memoryviewr� r� r� r� r� )rA Zb1Zb2Zb3rj r� r� r� r r r �testRecvmsgIntoScatter� s � z'RecvmsgIntoTests.testRecvmsgIntoScatterc C s | � d� d S )Nrs )r� r@ r r r �_testRecvmsgIntoScatter
s z(RecvmsgIntoTests._testRecvmsgIntoScatterN) rF rG rH rk rl rm rn rp rq ru rv r r r r rh � s
rh c @ s` e Zd Zz ddlZW n ey* dZY n0 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 ]2}t�|�}| �|t�d� |� | � || j � qT| �
t tjd� | �
t tj|� | �
t tjt j
� d S )Nr r � r9 r� rY )�socklen_t_limitr rx �listr rS r �itemsizer� r r! r^ rd �maxsize)rA �toobigr& �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� |�� | �||t� d� � | �
|| j � |}qX| � t tjd� | � t tj|� | � t tjt
j� d S )Nr ry r r9 r� rY )rz r r% r{ r rS r r| r rx r r! r^ rd r} )rA r~ r& Zlastr r� r r r �testCMSG_SPACE1
s
z CmsgMacroTests.testCMSG_SPACE) rF rG rH rw r� rz �minrx r r r� r� r r r r rw
s
rw 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/dfdQdR�Z0dSdT� Z1dUdV� Z2dWdX� Z3dYdZ� Z4d[d\� Z5d]d^� Z6d_d`� Z7dadb� Z8dcS )g�
SCMRightsTesti����c C s\ g }t |�D ]J}t�� \}}| �tj|� | �tj|� t�|t|�� � � |�
|� q |S r3 ) r r� ZmkstemprV r� r� r �writer �encode�append)rA r �fdsr9 r' r� r r r �newFDsR
s zSCMRightsTest.newFDsc C sB t |�D ]4\}}t�|dtj� | �t�|d�t|��� � qd S )Nr r� )� enumerater� �lseek�SEEK_SETr� r� r r� )rA r� r r' r r r �checkFDs^
s zSCMRightsTest.checkFDsc C s | � | j|� d S r3 )rV �closeRecvmsgFDsr� r r r r� e
s z#SCMRightsTest.registerRecvmsgResultc C sn |d D ]`\}}}|t jkr|t jkrt�d�}|�|d t|�t|�|j � � |D ]}t�|� qXqd S )Nr r9 ) r r1 �
SCM_RIGHTSr � frombytesr� r| r� r )rA Zrecvmsg_result�
cmsg_level� cmsg_type� cmsg_datar� r' r r r r� h
s
�
�z SCMRightsTest.closeRecvmsgFDsc
C s8 | � | �tgtjtjt�d| �|��fg�tt�� d S )Nr9 ) r� r� r� r r1 r� r r� r� )rA r r r r �createAndSendFDst
s ���z SCMRightsTest.createAndSendFDsr r c C s� |\}}}}| � |t� | �|| j� | j|dtj|d� | �|t� | � t
|�|� t � d�} |D ]`}
| �|
t � |
\} } }
| � | tj
� | � | tj� | �|
t� | � t
|
�t d� | �|
� qd| � t
| �|� | �| � d S )NT�r� r� r� r9 r )r� r� r� r� r� r r� r r{ r r� r �tupler1 r� rv �
SIZEOF_INTr� r� )rA Znumfdsr� �maxcmsgs� ignoreflagsr� r� r� r� r� �itemr� r� r� r r r �checkRecvmsgFDs~
s& �
z SCMRightsTest.checkRecvmsgFDsc C s | � d| �| jtt�d�� d S )Nr rS �r� r� r r� r� r@ r r r �testFDPassSimple�
s �z SCMRightsTest.testFDPassSimplec
C s< | � | �tgtjtjt�d| �d���� fg�t t�� d S )Nr9 r )
r� r� r� r r1 r� r r� ro r� r@ r r r �_testFDPassSimple�
s ���z SCMRightsTest._testFDPassSimplec C s | � d| �| jtt�d�� d S )Nr) rS r� r@ r r r �testMultipleFDPass�
s �z SCMRightsTest.testMultipleFDPassc C s | � d� d S �Nr) �r� r@ r r r �_testMultipleFDPass�
s z!SCMRightsTest._testMultipleFDPassr% c C s* | � d| �| jtt�t�dt ��� d S r� )r� r� r r� r� r r% r� r@ r r r �testFDPassCMSG_SPACE�
s
��z"SCMRightsTest.testFDPassCMSG_SPACEc C s | � d� d S r� r� r@ r r r �_testFDPassCMSG_SPACE�
s z#SCMRightsTest._testFDPassCMSG_SPACEc
C s0 | j d| �| jtt�t�dt ��tjd� d S )Nr r) �r� ) r� r� r r� r� r rx r� r� r@ r r r �testFDPassCMSG_LEN�
s ��z SCMRightsTest.testFDPassCMSG_LENc C s | � d� d S r� r� r@ r r r �_testFDPassCMSG_LEN�
s z!SCMRightsTest._testFDPassCMSG_LENr: zskipping, see issue #12958zskipping, see issue #22397c C s$ | j d| �| jtt�d�dd� d S )Nr� rS )r� r� r@ r r r �testFDPassSeparate�
s �z SCMRightsTest.testFDPassSeparatec C sX | � d�\}}| �| �tgtjtjt�d|g�ftjtjt�d|g�fg�tt�� d S �Nr� r9 � r� r� r� r� r r1 r� r r� �rA Zfd0�fd1r r r �_testFDPassSeparate�
s � ���z!SCMRightsTest._testFDPassSeparatec C s@ d}| j || �| jtt�t�t�t�t| � �dtj d� d S )Nr� )r� r� )
r� r� r r� r� r r% r� rx r� )rA Znum_fdsr r r �testFDPassSeparateMinSpace�
s ���z(SCMRightsTest.testFDPassSeparateMinSpacec C sX | � d�\}}| �| �tgtjtjt�d|g�ftjtjt�d|g�fg�tt�� d S r� r� r� r r r �_testFDPassSeparateMinSpace�
s � ���z)SCMRightsTest._testFDPassSeparateMinSpacec
C sf z| � |g|�}W n> tyP } z&| �|jt� | � |g�}W Y d }~n
d }~0 0 | �|t|�� d S r3 )r� r r rO rR r� r� )rA r� r� rj r� r r r �sendAncillaryIfPossible�
s "z%SCMRightsTest.sendAncillaryIfPossiblezsee issue #24725c C s& | j d| �| jtt�d�tjd� d S )Nr rS r� )r� r� r r� r� r r� r@ r r r �testFDPassEmpty s
��z SCMRightsTest.testFDPassEmptyc C s | � ttjtjdfg� d S �Nr� )r� r� r r1 r� r@ r r r �_testFDPassEmpty
s
�z SCMRightsTest._testFDPassEmptyc C s� | � | jtt�d�\}}}}| �|t� | �|| j� | j|dtj d� | �
t|�d� |D ]6\}}}| �|tj � | �|tj � | �
t|�t� q\d S )NrS Tr[ r )r� r r� r� r� r� r� r� r r� r r1 r� �
assertLessr� �rA 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 )Nr9 rY )r� r� r r1 r� r �badfdro r@ r r r �_testFDPassPartialInt s ��z#SCMRightsTest._testFDPassPartialIntc C s� | � | jtt�d�\}}}}| �|t� | �|| j� | j|dtj d� | �
t|�d� t � d�}|D ]J\}}}| �|tj � | �|tj
� |�|d t|�t|�|j � � qf| �
t|�d� | �|� d S )NrS Tr[ r� r9 )r� r r� r� r� r� r� r� r r� r r r1 r� r� r| r� ) rA r� r� r� r� r� r� r� r� r r r � testFDPassPartialIntInMiddle& s �
�
z*SCMRightsTest.testFDPassPartialIntInMiddlec
C sZ | � d�\}}| �ttjtjt�d|| jg��� d d� ftjtjt�d|g�fg� d S )Nr� r9 rY ) r� r� r� r r1 r� r r� ro r� r r r � _testFDPassPartialIntInMiddle9 s � ���z+SCMRightsTest._testFDPassPartialIntInMiddlec C sJ |\}}}}| � |t� | �|| j� | � |g � | j|dtj|d� d S )NT�r� r� r� )r� r� r� r� r� r r� )rA r� r� r� r� r� r� r r r �checkTruncatedHeaderE s �z"SCMRightsTest.checkTruncatedHeaderc C s" | j | �| jtt��tjd� d S �Nr� �r� r� r r� r� r r� r@ r r r �testCmsgTruncNoBufSizeO s �z$SCMRightsTest.testCmsgTruncNoBufSizec C s | � d� d S r� r� r@ r r r �_testCmsgTruncNoBufSizeX s z%SCMRightsTest._testCmsgTruncNoBufSizec C s$ | j | �| jtt�d�tjd� d S )Nr r� r� r@ r r r �testCmsgTrunc0[ s �z SCMRightsTest.testCmsgTrunc0c C s | � d� d S r� r� r@ r r r �_testCmsgTrunc0` s z SCMRightsTest._testCmsgTrunc0c C s | � | �| jtt�d�� d S r� )r� r� r r� r� r@ r r r �testCmsgTrunc1f s z SCMRightsTest.testCmsgTrunc1c C s | � d� d S r� r� r@ r r r �_testCmsgTrunc1i s z SCMRightsTest._testCmsgTrunc1c C s" | � | �| jtt�td �� d S r� )r� r� r r� r� r� r@ r r r �testCmsgTrunc2Intl s �z SCMRightsTest.testCmsgTrunc2Intc C s | � d� d S r� r� r@ r r r �_testCmsgTrunc2Ints s z SCMRightsTest._testCmsgTrunc2Intc C s( | � | �| jtt�t�d�d �� d S r� )r� r� r r� r� r rx r@ r r r �testCmsgTruncLen0Minus1v s �z%SCMRightsTest.testCmsgTruncLen0Minus1c C s | � d� d S r� r� r@ r r r �_testCmsgTruncLen0Minus1z s 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 r9 )r� r r� r� r� r� r� r� r r� r1 r� r r r r� r| r� ) rA �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 �r� r� �r� r rx r@ r r r �testCmsgTruncLen0� s z SCMRightsTest.testCmsgTruncLen0c C s | � d� d S r� r� r@ r r r �_testCmsgTruncLen0� s z SCMRightsTest._testCmsgTruncLen0c C s | j t�d�d dd� d S )Nr r r� r� r@ r r r �testCmsgTruncLen0Plus1� s z$SCMRightsTest.testCmsgTruncLen0Plus1c C s | � d� d S r� r� r@ r r r �_testCmsgTruncLen0Plus1� s z%SCMRightsTest._testCmsgTruncLen0Plus1c C s | j t�t�td� d S )Nr� �r� r rx r� r@ r r r �testCmsgTruncLen1� s �z SCMRightsTest.testCmsgTruncLen1c C s | � d� d S r� r� r@ r r r �_testCmsgTruncLen1� s z SCMRightsTest._testCmsgTruncLen1c C s( | j t�dt �d dt d d� d S )Nr� r r� r� r@ r r r �testCmsgTruncLen2Minus1� s
�z%SCMRightsTest.testCmsgTruncLen2Minus1c C s | � d� d S r� r� r@ r r r �_testCmsgTruncLen2Minus1� s z&SCMRightsTest._testCmsgTruncLen2Minus1N)r r )r )r )9rF rG rH r� r� r� r� r� r� r� r� r� r� r� r r r� r� r� r� r� r� rC rd 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� J
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� �Zd_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%dadGdH�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 )b�RFC3542AncillaryTestr� rY c C s4 i }|D ]&\}}}| � ||f|� ||||f< q|S r3 )r )rA r� �dr� r� r� r r r �ancillaryMapping� s
z%RFC3542AncillaryTest.ancillaryMappingrt 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 r9 )r r� r r5 r= r� rm r� r� r� r� r� r� r� r� r r� r< rv r� r r� r r ) rA �
ancbufsize�maxhopr� r� r� r� r� r� r� r� r� r r r �
checkHopLimit� s.
�
� �
z"RFC3542AncillaryTest.checkHopLimitr= r< c C s | j dd� d S �NrS �r� )r� r@ r r r �testRecvHopLimit� s z%RFC3542AncillaryTest.testRecvHopLimitc C s$ | � | jj| jd�� | �t� d S rY �r
r� r� r� r� r� r@ r r r �_testRecvHopLimit� s z&RFC3542AncillaryTest._testRecvHopLimitr% c C s | j t�t�d� d S )Nr� )r� r r% r� r@ r r r �testRecvHopLimitCMSG_SPACE� s z/RFC3542AncillaryTest.testRecvHopLimitCMSG_SPACEc C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r �_testRecvHopLimitCMSG_SPACE� s z0RFC3542AncillaryTest._testRecvHopLimitCMSG_SPACEr� c C s | j d| jd� d S �NrS )r� r� )r� � hop_limitr@ r r r �testSetHopLimit s z$RFC3542AncillaryTest.testSetHopLimitc C sL | � | jj| jd�� | �| �tgtjtj t
�
d| j g�fg�t t�� d S �Nr4 r9 )
r
r� r� r� r� r� r� r r5 r< r r� r� r@ 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� r9 r rt )r r� r r5 r= r? r� rm r� r� r� r� r� r� r� r� r� r> r� r r� r r r< ) rA r� r� r� r� r� r� r� ZancmapZtcdatar� Zhldatar r r � checkTrafficClassAndHopLimit s:
�
�
� �
z1RFC3542AncillaryTest.checkTrafficClassAndHopLimitr? r> c C s | j dd� d S r� )r� r@ r r r � testRecvTrafficClassAndHopLimit: s z4RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitc C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r � _testRecvTrafficClassAndHopLimit@ s z5RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitc C s | j t�t�d d� d S �Nr� r� )r� r r% r� r@ r r r �)testRecvTrafficClassAndHopLimitCMSG_SPACEE s �z>RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitCMSG_SPACEc C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r �*_testRecvTrafficClassAndHopLimitCMSG_SPACEM s z?RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitCMSG_SPACEc C s | j d| jd� d S r� �r� r� r@ r r r � testSetTrafficClassAndHopLimitR s �z3RFC3542AncillaryTest.testSetTrafficClassAndHopLimitc C sd | � | jj| jd�� | �| �tgtjtj t
�
d| j g�ftjtj t
�
d| j
g�fg�tt�� d S r� )r
r� r� r� r� r� r� r r5 r> r �
traffic_classr< r� r� r@ r r r � _testSetTrafficClassAndHopLimit[ s �����z4RFC3542AncillaryTest._testSetTrafficClassAndHopLimitc C s | j d| jd� d S r� r� r@ r r r �testOddCmsgSizef s �z$RFC3542AncillaryTest.testOddCmsgSizec
C s� | � | jj| jd�� zJ| �tgtjtjt � d| j
g�� � d ftjtj t � d| j
g�fg�}W n� ty� } zh| �|jt� | �tgtjtjt � d| j
g�ftjtj t � d| j
g�fg�}| �|tt�� W Y d }~n
d }~0 0 d S )Nr4 r9 r[ )r
r� r� r� r� r� r r5 r> r r� ro r< r� r r rO rR r� r� )rA rj r� r r r �_testOddCmsgSizep s2 ��������z%RFC3542AncillaryTest._testOddCmsgSizec C s� | j �tjtjd� | j�� |d u r*dn|f}| j| j tt �g|�R � \}}}}| �
|t � | � || j � | �
|g � | j
|dtj|d� d S )Nr r Tr� )r r� r r5 r= r� rm r� r� r� r� r� r� r� r� )rA r� r� rR r� r� r� r� r r r � checkHopLimitTruncatedHeader� s
�
�� �z1RFC3542AncillaryTest.checkHopLimitTruncatedHeaderc C s | j d tjd� d S )N�r� r� �r� r r� r@ r r r r� � s �z+RFC3542AncillaryTest.testCmsgTruncNoBufSizec C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r r� � s z,RFC3542AncillaryTest._testCmsgTruncNoBufSizec C s | j dtjd� d S )Nr r� r� r@ r r r �testSingleCmsgTrunc0� s �z)RFC3542AncillaryTest.testSingleCmsgTrunc0c C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r �_testSingleCmsgTrunc0� s z*RFC3542AncillaryTest._testSingleCmsgTrunc0c C s | j dd� d S )Nr r� )r� r@ r r r �testSingleCmsgTrunc1� s z)RFC3542AncillaryTest.testSingleCmsgTrunc1c C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r �_testSingleCmsgTrunc1� s z*RFC3542AncillaryTest._testSingleCmsgTrunc1c C s | j dt d� d S r� )r� r� r@ r r r �testSingleCmsgTrunc2Int� s z,RFC3542AncillaryTest.testSingleCmsgTrunc2Intc C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r �_testSingleCmsgTrunc2Int� s z-RFC3542AncillaryTest._testSingleCmsgTrunc2Intc C s | j t�d�d d� d S )Nr r r� )r� r rx r@ r r r � testSingleCmsgTruncLen0Minus1� s z2RFC3542AncillaryTest.testSingleCmsgTruncLen0Minus1c C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r � _testSingleCmsgTruncLen0Minus1� s z3RFC3542AncillaryTest._testSingleCmsgTruncLen0Minus1c C s� | j �tjtjd� | j�� | �| j tt �t�
t �d �\}}}}| � |t � | �
|| j� | j|dtjd� | �t|�d� |r�|d \}}}| � |tj� | � |tj� | �t|�t � d S )Nr Tr� r )r r� r r5 r= r� rm r� r� r� rx r� r� r� r� r� r� r r< r� r� r r r �testSingleCmsgTruncInData� s
�
� z.RFC3542AncillaryTest.testSingleCmsgTruncInDatac C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r �_testSingleCmsgTruncInData� s z/RFC3542AncillaryTest._testSingleCmsgTruncInDatac C s | j �tjtjd� | j �tjtjd� | j�� | �| j t t
�|�\}}}}| � |t
� | � || j
� | j|dtj|d� | � t |�d� |d \}}} | � |tj� | �|tjtjh� | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d d� d S )Nr Tr� r r9 rt )r r� r r5 r= r? r� rm r� r� r� r� r� r� r� r� r r> r< r� r r� r r ) rA 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� r� r@ r r r �testSecondCmsgTrunc0 s �z)RFC3542AncillaryTest.testSecondCmsgTrunc0c C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r �_testSecondCmsgTrunc0 s z*RFC3542AncillaryTest._testSecondCmsgTrunc0c C s | � t�t�d � d S r� �r r r% r� r@ r r r �testSecondCmsgTrunc1 s z)RFC3542AncillaryTest.testSecondCmsgTrunc1c C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r �_testSecondCmsgTrunc1 s z*RFC3542AncillaryTest._testSecondCmsgTrunc1c C s | � t�t�dt � d S r� r r@ r r r �testSecondCmsgTrunc2Int$ s �z,RFC3542AncillaryTest.testSecondCmsgTrunc2Intc C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r �_testSecondCmsgTrunc2Int* s z-RFC3542AncillaryTest._testSecondCmsgTrunc2Intc C s" | � t�t�t�d� d � d S r� )r r r% r� rx r@ r r r � testSecondCmsgTruncLen0Minus1/ s
��z2RFC3542AncillaryTest.testSecondCmsgTruncLen0Minus1c C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r � _testSecondCmsgTruncLen0Minus15 s z3RFC3542AncillaryTest._testSecondCmsgTruncLen0Minus1c
C sT | 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� |�rD|�d�\}}}| �|tj� |�|� | �t |�t � | �|g � d S )Nr Tr� r r9 rt ) r r� r r5 r= r? r� rm r� r� r� r% r� rx r� r� r� r� r� r> r< �pop�remover r� r r r� )
rA r� r� r� r� Z
cmsg_typesr� r� r� r� r r r �testSecomdCmsgTruncInData: s:
�
�
�
z.RFC3542AncillaryTest.testSecomdCmsgTruncInDatac C s$ | � | jj| jd�� | �t� d S rY r� r@ r r r �_testSecomdCmsgTruncInData_ s z/RFC3542AncillaryTest._testSecomdCmsgTruncInDataN)rt r )rt r )r )r )1rF rG rH r� r� r� r� r r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r
r r r r r r r r r� � s�
�
$�
�
�
�
�
�
�
�
�
#r� c @ s e Zd ZdS )�SendrecvmsgUDPTestBaseN�rF rG rH r r r r r g s r r� c @ s e Zd ZdS )�SendmsgUDPTestNr r r r r r l s r r� c @ s e Zd ZdS )�RecvmsgUDPTestNr r r r r r p s r r� c @ s e Zd ZdS )�RecvmsgIntoUDPTestNr r r r r r t s r c @ s e Zd Zdd� ZdS )�SendrecvmsgUDP6TestBasec C s | � |d d� |d d� � d S r� r� r� r r r r� } s z+SendrecvmsgUDP6TestBase.checkRecvmsgAddressN�rF rG rH r� r r r r r y s r r r� r! c @ s e Zd ZdS )�SendmsgUDP6TestNr r r r r r � s r c @ s e Zd ZdS )�RecvmsgUDP6TestNr r r r r r � s r c @ s e Zd ZdS )�RecvmsgIntoUDP6TestNr r r r r r � s r r5 c @ s e Zd ZdS )� RecvmsgRFC3542AncillaryUDP6TestNr r r r r r � s r c @ s e Zd ZdS )�#RecvmsgIntoRFC3542AncillaryUDP6TestNr r r r r r � s r c @ s e Zd ZdS )�SendrecvmsgUDPLITETestBaseNr r r r r r � s r c @ s e Zd ZdS )�SendmsgUDPLITETestNr r r r r r � s r c @ s e Zd ZdS )�RecvmsgUDPLITETestNr r r r r r � s r c @ s e Zd ZdS )�RecvmsgIntoUDPLITETestNr r r r r r � s r c @ s e Zd Zdd� ZdS )�SendrecvmsgUDPLITE6TestBasec C s | � |d d� |d d� � d S r� r� r� r r r r� � s z/SendrecvmsgUDPLITE6TestBase.checkRecvmsgAddressNr r r r r r! � s r! c @ s e Zd ZdS )�SendmsgUDPLITE6TestNr r r r r r" � s r" c @ s e Zd ZdS )�RecvmsgUDPLITE6TestNr r r r r r# � s r# c @ s e Zd ZdS )�RecvmsgIntoUDPLITE6TestNr r r r r r$ � s r$ c @ s e Zd ZdS )�#RecvmsgRFC3542AncillaryUDPLITE6TestNr r r r r r% � s r% c @ s e Zd ZdS )�'RecvmsgIntoRFC3542AncillaryUDPLITE6TestNr r r r r r& � s r& c @ s e Zd ZdS )�SendrecvmsgTCPTestBaseNr r r r r r' � s r' c @ s e Zd ZdS )�SendmsgTCPTestNr r r r r r( � s r( c @ s e Zd ZdS )�RecvmsgTCPTestNr r r r r r) s r) c @ s e Zd ZdS )�RecvmsgIntoTCPTestNr r r r r r* s r* c @ s e Zd ZdS )� SendrecvmsgSCTPStreamTestBaseNr r r r r r+ s r+ z6IPPROTO_SCTP: [Errno 62] Protocol not supported on AIXr; r0 r� c @ s e Zd ZdS )�SendmsgSCTPStreamTestNr r r r r r, s r, c s e Zd Z� fdd�Z� ZS )�RecvmsgSCTPStreamTestc
sT zt t| ��� W n< tyN } z$|jtjkr0� | �d� W Y d }~n
d }~0 0 d S �Nz4sporadic ENOTCONN (kernel issue?) - see issue #13876)rL r- ra r rO �ENOTCONNrc �rA r� rT r r ra s z$RecvmsgSCTPStreamTest.testRecvmsgEOF�rF rG rH ra rY r r rT r r- s r- c s e Zd Z� fdd�Z� ZS )�RecvmsgIntoSCTPStreamTestc
sT zt t| ��� W n< tyN } z$|jtjkr0� | �d� W Y d }~n
d }~0 0 d S r. )rL r2 ra r rO r/ rc r0 rT r r ra + s z(RecvmsgIntoSCTPStreamTest.testRecvmsgEOFr1 r r rT r r2 % s r2 c @ s e Zd ZdS )� SendrecvmsgUnixStreamTestBaseNr r r r r r3 4 s r3 r� c @ s e Zd ZdS )�SendmsgUnixStreamTestNr r r r r r4 8 s r4 c @ s e Zd ZdS )�RecvmsgUnixStreamTestNr r r r r r5 = s r5 c @ s e Zd ZdS )�RecvmsgIntoUnixStreamTestNr r r r r r6 C s r6 r1 r� c @ s e Zd ZdS )�RecvmsgSCMRightsStreamTestNr r r r r r7 I s r7 c @ s e Zd ZdS )� RecvmsgIntoSCMRightsStreamTestNr r r r r r8 N s r8 c sF e Zd Z� fdd�ZejZeed�r2dZ dd� Z
n dZ dd� Z
� Z S ) �InterruptedTimeoutBasec s2 t � �� t�tjdd� �}| �tjtj|� d S )Nc S s dd S r� r )Zsignumra r r r r� b r� z.InterruptedTimeoutBase.setUp.<locals>.<lambda>)rL rB r� r� rV )rA Zorig_alrm_handlerrT r r rB _ s
�z InterruptedTimeoutBase.setUp� setitimerg�������?c C s t �t j|� d S r3 )r� r: � ITIMER_REAL�rA Zsecondsr r r �setAlarmn s z InterruptedTimeoutBase.setAlarmr� c C s t �|� d S r3 )r� r� r< r r r r= t s ) rF rG rH rB r r� r6 r
r� �
alarm_timer= rY r r rT r r9 Z s
r9 � 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 s t � �� | j�| j� d S r3 )rL rB r< r� r6 r@ rT r r rB � s
z InterruptedRecvTimeoutTest.setUpc O sd zR| � | j� | �t�� }||i |�� W d � n1 s<0 Y W | � d� n | � d� 0 d S r� )r= r> r! r� �rA rq rR rS r- r r r �checkInterruptedRecv� s
.z/InterruptedRecvTimeoutTest.checkInterruptedRecvc C s | � | jjd� d S r� )rB r< r� r@ r r r �testInterruptedRecvTimeout� s z5InterruptedRecvTimeoutTest.testInterruptedRecvTimeoutc C s | � | jjtd�� d S r� )rB r< � recv_intor� r@ r r r � testInterruptedRecvIntoTimeout� s z9InterruptedRecvTimeoutTest.testInterruptedRecvIntoTimeoutc C s | � | jjd� d S r� )rB r< rc r@ r r r � testInterruptedRecvfromTimeout� s z9InterruptedRecvTimeoutTest.testInterruptedRecvfromTimeoutc C s | � | jjtd�� d S r� )rB r< �
recvfrom_intor� r@ r r r �"testInterruptedRecvfromIntoTimeout� s z=InterruptedRecvTimeoutTest.testInterruptedRecvfromIntoTimeoutr� c C s | � | jjd� d S r� )rB r< r� r@ r r r � testInterruptedRecvmsgTimeout� s z8InterruptedRecvTimeoutTest.testInterruptedRecvmsgTimeoutr� c C s | � | jjtd�g� d S r� )rB r< r� r� r@ r r r �!testInterruptedRecvmsgIntoTimeout� s z<InterruptedRecvTimeoutTest.testInterruptedRecvmsgIntoTimeout)rF rG rH rB rB rC rE rF rH r r rI rJ rY r r rT r r@ z 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)rL rB r� r� rV r rN �Thread� doConnect�startr< r� r� r r� r6 )rA Z
cli_threadr� rT r r rB � s
z InterruptedSendTimeoutTest.setUpc C s | j �| j� d S r3 )r� r� r� r@ r r r rN � s z$InterruptedSendTimeoutTest.doConnectc O sf zT| � t��,}| �| j� ||i |�� qW d � n1 s>0 Y W | �d� n | �d� 0 d S r� )r! r� r= r> rA r r r �checkInterruptedSend� s
0z/InterruptedSendTimeoutTest.checkInterruptedSendr� rn c C s | � | jjd� d S r= )rP r� r� r@ r r r �testInterruptedSendTimeout� s z5InterruptedSendTimeoutTest.testInterruptedSendTimeoutc C s | � | jjd| j� d S r= )rP r� r* r� r@ r r r � testInterruptedSendtoTimeout� s �z7InterruptedSendTimeoutTest.testInterruptedSendtoTimeoutr� c C s | � | jjdg� d S r= )rP r� r� r@ r r r � testInterruptedSendmsgTimeout� s z8InterruptedSendTimeoutTest.testInterruptedSendmsgTimeout)rF rG rH rB rN rP r Zrequires_mac_verrQ rR r r rS rY r r rT r rK � s
rK c @ s e Zd Zdd� Zdd� ZdS )�
TCPCloserTestc C sh | j �� \}}|�� | j}t�|gg g d�\}}}| �||g� | �|�d�d� |�� |�� d S )N� �?r r� )r< r� r r� r� r� r� )rA r� r� Zsdr� r� r r r r � testClose� s zTCPCloserTest.testClosec C s | j �t| jf� t�d� d S �NrU )r� r� r� r> r� r� r@ r r r �
_testClose� s zTCPCloserTest._testCloseN)rF rG rH rV rX r r r r rT � s
rT c @ sN e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dS )�BasicSocketPairTestr� c C s t j| |d� d S r� )r� rM r� r r r rM � s z BasicSocketPairTest.__init__c C s\ | � |tj� ttd�r*| �|jtj� n| �|jtj� | �|jtj� | �|j d� d S )Nr� r )
r r r
r� r r� r; r� r0 r r� r r r �_check_defaults� s
z#BasicSocketPairTest._check_defaultsc C s | � | j� d S r3 )rZ r� r@ r r r �
_testDefaults s z!BasicSocketPairTest._testDefaultsc C s | � | j� d S r3 )rZ r< r@ r r r � testDefaults s z BasicSocketPairTest.testDefaultsc C s | j �d�}| �|t� d S r� �r< r� r� r� r$ r r r r� s z BasicSocketPairTest.testRecvc C s | j �t� d S r3 �r� r� r� r@ r r r r� s z BasicSocketPairTest._testRecvc C s | j �t� d S r3 �r< r� r� r@ r r r �testSend s z BasicSocketPairTest.testSendc C s | j �d�}| �|t� d S r� �r� r� r� r� r$ r r r � _testSend s z BasicSocketPairTest._testSendN)r� ) rF rG rH rM rZ r[ r\ r� r� r` rb r r r r rY � s
rY c s� e Zd Zdd� Z� fdd�Z� fdd�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z e �
eed� preed� d�e �ejd�dd� ��Zdd� Zdd � Zd d � Z� ZS ) �PurePythonSocketPairTestc C s2 zt �t j�W S ty, t �t j� Y S 0 d S r3 )r r� r� r r; r@ r r r r� s z#PurePythonSocketPairTest.socketpairc s0 t td�r tj| _tjt_nd | _t� �� d S )Nr� )r
r r r� �_orig_spZ_fallback_socketpairrL rB r@ rT r r rB % s
z PurePythonSocketPairTest.setUpc s t � �� | jd ur | jt_d S r3 )rL rD rd r r� r@ rT r r rD 0 s
z!PurePythonSocketPairTest.tearDownc C s | j �d�}| �|t� d S r� r] r$ r r r � test_recv6 s z"PurePythonSocketPairTest.test_recvc C s | j �t� d S r3 r^ r@ r r r �
_test_recv: s z#PurePythonSocketPairTest._test_recvc C s | j �t� d S r3 r_ r@ r r r � test_send= s z"PurePythonSocketPairTest.test_sendc C s | j �d�}| �|t� d S r� ra r$ r r r �
_test_send@ s z#PurePythonSocketPairTest._test_sendc C s$ t �t j�\}}|�� |�� d S r3 )r r� r; r �rA r� r r r r � test_ipv4D s z"PurePythonSocketPairTest.test_ipv4c C s d S r3 r r@ r r r �
_test_ipv4I s z#PurePythonSocketPairTest._test_ipv4r5 � IPV6_V6ONLY� IPV6_V6ONLY option not supportedr; c C s$ t �t j�\}}|�� |�� d S r3 )r r� r� r ri r r r � test_ipv6L s z"PurePythonSocketPairTest.test_ipv6c C s d S r3 r r@ r r r �
_test_ipv6U s z#PurePythonSocketPairTest._test_ipv6c s� t j j�d � � �fdd�}d }}zn|t j _| �t�� t �� \}}W d � n1 sX0 Y W �t j _� rx� �� |r�|�� |r�|�� n.�t j _� r�� �� |r�|�� |r�|�� 0 d S )Nc sf �| �}� d u rbt � t jt j�� � �d� z� �|d d� � W n ttfyV Y n0 � �d� |S )NFr� T)r r; r0 r r� �BlockingIOError�InterruptedError)rA r, �Z inject_sockZorig_getsocknamer r �inject_getsocketname\ s
z[PurePythonSocketPairTest.test_injected_authentication_failure.<locals>.inject_getsocketname)r r� r! r r� r )rA rs �sock1�sock2r rr r �$test_injected_authentication_failureX s, ,
�z=PurePythonSocketPairTest.test_injected_authentication_failurec C s d S r3 r r@ r r r �%_test_injected_authentication_failurey s z>PurePythonSocketPairTest._test_injected_authentication_failure)rF rG rH r� rB rD re rf rg rh rj rk r� rC r
r rA r r� rn ro rv rw rY r r rT r rc s$
� !rc 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� )rN rv �eventr� rM r� r r r rM � s
z NonBlockingTCPTests.__init__c C sh | � | j�� |� |dk}| � |�� |� td urd|d u }t�|tjtj�}| � t|tj@ � |� d S )N� ) r� r< r� � getblockingr �F_GETFLr� �
O_NONBLOCK�bool)rA r� r6 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 )NTFrz r r� )r< r r� r� r@ r r r �testSetBlocking� s z#NonBlockingTCPTests.testSetBlockingc C s d S r3 r r@ r r r �_testSetBlocking� s z$NonBlockingTCPTests._testSetBlockingc C sb dd l }|j|jkr | �d� | j�d� | �| j�� d� | j�|jd � | �| j�� � d S )Nr zneeds UINT_MAX < ULONG_MAXFrz r ) rw r� Z ULONG_MAXrc r< r r� r� r* r� r r r �testSetBlocking_overflow� s
z,NonBlockingTCPTests.testSetBlocking_overflowr z test needs socket.SOCK_NONBLOCKr� � � c C s4 | j �� t�tjtjtjB �| _ | �| j d� d S r� )r< r r r; r0 r r� r@ r r r �testInitNonBlocking� s
�z'NonBlockingTCPTests.testInitNonBlockingc C s d S r3 r r@ r r r �_testInitNonBlocking� s z(NonBlockingTCPTests._testInitNonBlockingc C s` t d ��D | j�d� | j�� \}}| �|j� | �|�� � W d � n1 sR0 Y d S )Nr� )r7 r< r� r� rV r r* r� r� r r r �testInheritFlagsBlocking� s
z,NonBlockingTCPTests.testInheritFlagsBlockingc C s | j �t| jf� d S r3 �r� r� r� r> r@ r r r �_testInheritFlagsBlocking� s z-NonBlockingTCPTests._testInheritFlagsBlockingc C sf d}t |��F | j�d� | j�� \}}| �|j� | �|�� |� W d � n1 sX0 Y d S )Ng 4@r� )r7 r< r� r� rV r r� r� )rA Zdefault_timeoutr� r� r r r �testInheritFlagsTimeout� s
z+NonBlockingTCPTests.testInheritFlagsTimeoutc C s | j �t| jf� d S r3 r� r@ r r r �_testInheritFlagsTimeout� s z,NonBlockingTCPTests._testInheritFlagsTimeoutc C s� | j �d� t�� }| �t�� | j �� \}}W d � n1 sB0 Y t�� | }| �|d� | j� � t
�
| j gg g t j �\}}}| j |vr�| �
d� | j �� \}}| �|j� | �|�� � d S )NFrU z'Error trying to do accept after select.)r< r r� � monotonicr! rp r� r� ry rm r� r � LONG_TIMEOUTr rV r r* r� )rA Z
start_timer� r� Zdtr� r� r r r r �
testAccept� s ,
z NonBlockingTCPTests.testAcceptc C s | j �� | j�t| jf� d S r3 )ry r� r� r� r� r> r@ r r r � _testAccept� s
z NonBlockingTCPTests._testAcceptc C s� | j �� \}}| �|j� |�d� | �t�� |�tt ��}W d � n1 sR0 Y | j
� � t � |gg g t
j�\}}}||vr�| �d� |�tt ��}| �|t � d S )NFz0Error during select call to non-blocking socket.)r< r� rV r r r! rp r� r� r� ry rm r� r r� r r� )rA r� r� r� r� r� r r r r r� s
,
z NonBlockingTCPTests.testRecvc C s, | j �t| jf� | j�� | j �t� d S r3 )r� r� r� r> ry r� r� r� r@ r r r r� s
z NonBlockingTCPTests._testRecvN)r� )rF rG rH rM r� r� r� r rB r� Z_testSetBlocking_overflowr� rA r
r �requires_linux_versionr� r� r� r� r� r� r� r� r� r� r r r r rx � s*
� rx 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.
rY r �strictNr
r� r� c C s t j| |d� d S r� r� r� r r r rM 7 s z FileObjectClassTestCase.__init__c C sR dd� t d�D �\| _| _| _| _t�| � | jj| j | j
| j | j | j
d�| _d S )Nc S s g | ] }t �� �qS r )rN rv �r� r9 r r r r� ; s z1FileObjectClassTestCase.setUp.<locals>.<listcomp>r) ��encoding�errors�newline)r �evt1�evt2�
serv_finished� cli_finishedr� rB r� r� � read_moderd r� r� r� � read_filer@ r r r rB : s �
�z FileObjectClassTestCase.setUpc C s6 | j �� | j�� | �| jj� d | _t�| � d S r3 )r� rm r� r r
�closedr� rD r@ r r r rD D s
z FileObjectClassTestCase.tearDownc C s0 t �| � | jj| j| j| j| j| jd�| _ d S )Nr� )
r� r� r� r� �
write_moderd r� r� r� �
write_filer@ r r r r� K s
�z#FileObjectClassTestCase.clientSetUpc C s6 | j �� | j�� | �| jj� d | _t�| � d S r3 )r� rm r� r r
r� r� r� r@ r r r r� S s
z&FileObjectClassTestCase.clientTearDownc C sx | j �d� | j�d� | �tj| jjd� | �t�� }| j�d� W d � n1 sX0 Y | �dt |j
�� d S )Nr r� z!cannot read from timed out object) r� r� r� r� r! r r6 r r r r+ )rA �ctxr r r �testReadAfterTimeoutZ s *z,FileObjectClassTestCase.testReadAfterTimeoutc C s. | j �| jdd� � | j �� | j�� d S )Nr r� )r� r� � write_msg�flushr� r� r@ r r r �_testReadAfterTimeoutf s
z-FileObjectClassTestCase._testReadAfterTimeoutc C s< | j �t| j�d �}| j �d�}|| }| �|| j� d S r )r� r� r� �read_msgr� )rA � first_segZ
second_segr� r r r �
testSmallReadk s z%FileObjectClassTestCase.testSmallReadc C s | j �| j� | j �� d S r3 �r� r� r� r� r@ r r r �_testSmallReadr s z&FileObjectClassTestCase._testSmallReadc C s | j �� }| �|| j� d S r3 )r� r� r� r� r$ r r r � testFullReadv s
z$FileObjectClassTestCase.testFullReadc C s | j �| j� | j �� d S r3 )r� r� r� r r@ r r r �
_testFullRead{ s z%FileObjectClassTestCase._testFullReadc C s: t | j�� }| j�d�}|s q(||7 }q | �|| j� d S r� )r� r� r� r� r� )rA r �charr r r �testUnbufferedRead s
z*FileObjectClassTestCase.testUnbufferedReadc C s | j �| j� | j �� d S r3 r� r@ r r r �_testUnbufferedRead� s z+FileObjectClassTestCase._testUnbufferedReadc C s | j �� }| �|| j� d S r3 )r� �readliner� r� �rA �liner r r � testReadline� s
z$FileObjectClassTestCase.testReadlinec C s | j �| j� | j �� d S r3 r� r@ r r r �
_testReadline� s z%FileObjectClassTestCase._testReadlinec C s& | j �� | j�� }| �|| j� d S r3 )r� r r� r� r� r� r$ r r r �testCloseAfterMakefile� s
z.FileObjectClassTestCase.testCloseAfterMakefilec C s | j �| j� | j �� d S r3 r� r@ r r r �_testCloseAfterMakefile� s z/FileObjectClassTestCase._testCloseAfterMakefilec C s@ | j �� | j�tt��}t| jt�r.|� � }| �
|| j� d S r3 ) r� r r� r� r� r� r� r� r �decoder� r$ r r r � testMakefileAfterMakefileClose� s
z6FileObjectClassTestCase.testMakefileAfterMakefileClosec C s | j �| j� | j �� d S r3 r� r@ r r r � _testMakefileAfterMakefileClose� s z7FileObjectClassTestCase._testMakefileAfterMakefileClosec C s | � | jj � d S r3 )r
r� r� r@ r r r �testClosedAttr� s z&FileObjectClassTestCase.testClosedAttrc C s | � | jj � d S r3 )r
r� r� r@ r r r �_testClosedAttr� s z'FileObjectClassTestCase._testClosedAttrc C s, | � | jj| j� | � | jj| j�� � d S r3 )r� r� r� r� r� r� r r@ r r r �testAttributes� s z&FileObjectClassTestCase.testAttributesc C s, | � | jj| j� | � | jj| j�� � d S r3 )r� r� r� r� r� r� r r@ r r r �_testAttributes� s z'FileObjectClassTestCase._testAttributesc C s8 | j �� | �t| j j� | j�� | �t| jj� d S r3 )r� r r! r� r r� r r� r@ r r r �
testRealClose� s
z%FileObjectClassTestCase.testRealClosec C s d S r3 r r@ r r r �_testRealClose� s z&FileObjectClassTestCase._testRealClose)r� )&rF rG rH rX rd r� r� r� r� r� r� r� r� rM rB rD 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� ! sD
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. )r� r� r� r� r� r� r� r r r �testUnbufferedReadline� s
z8UnbufferedFileObjectClassTestCase.testUnbufferedReadlinec C s2 | j �d| j � | j �d| j � | j �� d S )Nr� r� r� r@ r r r �_testUnbufferedReadline� s z9UnbufferedFileObjectClassTestCase._testUnbufferedReadlinec C sD | j �� | j �d�}| �|| j� | j�� | �t| j jd� d S r� )r� r r� r� r� r� r! r r$ r r r �testMakefileClose� s
z3UnbufferedFileObjectClassTestCase.testMakefileClosec C s | j �| j� | j �� d S r3 r� r@ r r r �_testMakefileClose� s z4UnbufferedFileObjectClassTestCase._testMakefileClosec C s6 t �| j�}| j�� t �| j�}| �|d |� d S r� )rd rc r� r� r r� )rA Zrefcount_beforeZrefcount_afterr r r � testMakefileCloseSocketDestroy� s
z@UnbufferedFileObjectClassTestCase.testMakefileCloseSocketDestroyc C s d S r3 r r@ r r r � _testMakefileCloseSocketDestroy� s zAUnbufferedFileObjectClassTestCase._testMakefileCloseSocketDestroyc C s | j �d� | �| j�td��d � | �| j�t| j�d �d � | j �
� | j � d� | j�t| j�d �}|d u r�t
�d� | j�t| j�d �}td�}| j�|�}| �|d� ||d |� }| �|| j� | �| j�td��d � | �| j�d�d � d S )NFr� r� rU g �?rj r )r� r r� r� �readintor� r� r� r� r� rm r� r� r� r� )rA r� r r r� r r r �testSmallReadNonBlocking� s
z:UnbufferedFileObjectClassTestCase.testSmallReadNonBlockingc C s> | j �d� | j�| j� | j�� | j�� | j�d� d S )NrU � @) r� r� r� r� r� r� r� rm r� r@ r r r �_testSmallReadNonBlocking
s
z;UnbufferedFileObjectClassTestCase._testSmallReadNonBlockingc C s" | j �d� | jr | �| j� d S )Nr� )r� r� � serv_skippedrc r@ r r r �testWriteNonBlocking s z6UnbufferedFileObjectClassTestCase.testWriteNonBlockingc C st d | _ | j�d� dtj }d}| j�|�}| �|d� t|�D ](}| j�|�}|d u r\ qp| �|d� q@d| _ d S )NFr� r� r z$failed to saturate the socket buffer) r� r� r r r� r� r� rS r )rA ZBIGZLIMITr r9 r r r �_testWriteNonBlocking s
z7UnbufferedFileObjectClassTestCase._testWriteNonBlockingN)rF rG rH rX rd r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd ZdZdS )�#LineBufferedFileObjectClassTestCaser N�rF rG rH rd r r r r r� 0 s r� c @ s e Zd ZdZdS )�$SmallBufferedFileObjectClassTestCaser� Nr� r r r r r� 5 s r� 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� rF rG rH rX r� r� r� r� r� r� r� r r r r r� : s
r� c @ s* e Zd ZdZdZeZdZe�d�Z dZ
dS )�#UnicodeWriteFileObjectClassTestCaser� r
r� r r& N) rF rG rH rX r� r� r� r� r� r� r� r r r r r� D s
r� c @ s0 e Zd ZdZdZe�d�ZdZe�d�Z dZ
dS )�'UnicodeReadWriteFileObjectClassTestCaser� r r r� r& Nr� r r r r r� N s
r� c @ s e Zd ZdZdd� ZdS )�NetworkConnectionTestzProve network connection.c C s t �t| jf�| _| j| _d S r3 )r �create_connectionr� r> r� r� r@ r r r r� [ s z!NetworkConnectionTest.clientSetUpN)rF rG rH rX r� r r r r r� X s r� c @ s e Zd ZdZdS )�
BasicTCPTest2zLTests that NetworkConnection does not break existing TCP functionality.
N)rF rG rH rX r r r r r� a s r� c @ sD e Zd ZG dd� dej�Zejdd� �Zdd� Zdd� Z d d
� Z
d S ) �NetworkConnectionNoServerc @ s e Zd Zdd� ZdS )z$NetworkConnectionNoServer.MockSocketc G s t �d��d S )Nz timed out)r r6 r� r r r r� h s z,NetworkConnectionNoServer.MockSocket.connectN)rF rG rH r� r r r r �
MockSocketg s r� c c s, t j }| jt _ zdV W |t _ n|t _ 0 dS )z*Return a socket which times out on connectN)r r� )rA Z
old_socketr r r �mocked_socket_modulek s
z.NetworkConnectionNoServer.mocked_socket_modulec C sr t �� }t�tjtj�}| �|j� | �t�� }|� t
|f� W d � n1 sR0 Y | � |j j
t
j� d S r3 )r r� r r; r0 rV r r! r r� r� r� r+ rO Z ECONNREFUSED)rA r> r� r- r r r � test_connectu s ,z&NetworkConnectionNoServer.test_connectc C s\ t �� }| �t�� }t�t|f� W d � n1 s60 Y t �� }| �|j j
|� d S r3 ) r r� r! r r r� r� Z get_socket_conn_refused_errsr r+ rO )rA r> r- Zexpected_errnosr r r �test_create_connection} s
,z0NetworkConnectionNoServer.test_create_connectionc C s� | � � �x zt�tdf� W nJ tjy0 Y nB tyf } z tjsP|jtj krR� W Y d }~nd }~0 0 | �
d� W d � n1 s�0 Y d S )Nr r8 ) r� r r� r� r6 r r r� rO Z EAFNOSUPPORTr r: r r r � test_create_connection_timeout� s
z8NetworkConnectionNoServer.test_create_connection_timeoutN) rF rG rH r r� �
contextlib�contextmanagerr� r� r� r� r r r r r� e 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 s t j| |d� t�| � d S r� r� r� r r r rM � s z(NetworkConnectionAttributesTest.__init__c C s t �� | _d S r3 )r r� � source_portr@ r r r r� � s z+NetworkConnectionAttributesTest.clientSetUpc C s | j �� d | _ t�| � d S r3 r� r@ r r r r� � s
z.NetworkConnectionAttributesTest.clientTearDownc C s | j �� \}}|�� d S r3 )r< r� r r� r r r � _justAccept� s z+NetworkConnectionAttributesTest._justAcceptc C s: t jt| jftjd�| _| �| jj� | � | jj
d� d S )Nr4 r� ) r r� r� r> r r� r� rV r r� r r@ r r r � _testFamily� s
�z+NetworkConnectionAttributesTest._testFamilyc C sJ t jt| jftjd| jfd�| _| �| jj � | �
| j� � d | j� d S )Nr& )r6 Zsource_addressr ) r r� r� r> r r� r� r� rV r r� r� r@ r r r �_testSourceAddress� s �z2NetworkConnectionAttributesTest._testSourceAddressc C sn | � t�� d u � t�d� z.t�t| jf�| _| �| jj � W t�d � n t�d � 0 | �
| j� � d� d S )Nr � r
r r4 r5 r� r� r> r� rV r r� r� r@ r r r �_testTimeoutDefault� s
z3NetworkConnectionAttributesTest._testTimeoutDefaultc C sr | � t�� d u � t�d� z2tjt| jfd d�| _| �| jj � W t�d � n t�d � 0 | �
| j� � d � d S �N� r4 r� r@ r r r �_testTimeoutNone� s
z0NetworkConnectionAttributesTest._testTimeoutNonec C s, t jt| jfdd�| _| �| j�� d� d S r� )r r� r� r> r� r� r� r@ r r r �_testTimeoutValueNamed� s z6NetworkConnectionAttributesTest._testTimeoutValueNamedc C s8 t �t| jfd�| _| �| jj� | �| j�� d� d S )Nr� ) r r� r� r> r� rV r r� r� r@ r r r �_testTimeoutValueNonamed� s z8NetworkConnectionAttributesTest._testTimeoutValueNonamedN)r� )rF rG rH rM 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 s t j| |d� t�| � d S r� r� r� r r r rM � s z'NetworkConnectionBehaviourTest.__init__c C s d S r3 r r@ r r r r� � s z*NetworkConnectionBehaviourTest.clientSetUpc C s | j �� d | _ t�| � d S r3 r� r@ r r r r� � s
z-NetworkConnectionBehaviourTest.clientTearDownc C s2 | j �� \}}| �|j� t�d� |�d� d S )Nr� � done!)r< r� rV r r� r� r� r� r r r �testInsideTimeout� s
z0NetworkConnectionBehaviourTest.testInsideTimeoutc C s0 t �t| jf� | _}|�d�}| �|d� d S )Nr� r� )r r� r� r> r� r� r� )rA r� r^ r r r �_testInsideTimeout s
z1NetworkConnectionBehaviourTest._testInsideTimeoutc s4 t jt| jfdd� | _� | �t j� fdd�� d S )Nr r4 c s
� � d�S )Nr� )r� r �r� r r r� r� zDNetworkConnectionBehaviourTest._testOutsideTimeout.<locals>.<lambda>)r r� r� r> r� r! r6 r@ r r� r �_testOutsideTimeout s z2NetworkConnectionBehaviourTest._testOutsideTimeoutN)r� )
rF rG rH rM 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 s � fdd�}� � tj|d� d S )Nc s � j �d� � j �� d S rW )r< r� r� r� r@ r r �
raise_timeout
s z4TCPTimeoutTest.testTCPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (TCP)�r! r r6 �rA r� r r@ r �testTCPTimeout s
�z TCPTimeoutTest.testTCPTimeoutc C sx d}z| j �d� | j �� }W nF tjy< | �d� Y n* tyP d}Y n | �d� Y n0 |st| �d� d S )NFrz z%caught timeout instead of error (TCP)Tz!caught unexpected exception (TCP)z3accept() returned success when we did not expect it)r< r� r� r r6 r r �rA �okr9 r r r �testTimeoutZero s z TCPTimeoutTest.testTimeoutZeror� ztest needs signal.alarm()c
s | j �d� G dd� dt�� � fdd�}t�tj|�}z�z�z�zt�d� | j �� }W n\ tjyr | � d� Y nJ � y� Y n: | � dt
� � d d� t �
� f � Y n 0 | � d � W t�d
� n t�d
� 0 W n � y� | � d � Y n0 W t�tj|� nt�tj|� 0 d S ) Nr� c @ s e Zd ZdS )z4TCPTimeoutTest.testInterruptedTimeout.<locals>.AlarmNr r r r r �Alarm( s r� c s � �d S r3 r )r� ra �r� r r �
alarm_handler* s 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)r< r� r� r� r� r� r� r r6 r rd �exc_info� traceback�
format_exc)rA r r� r9 r r� r �testInterruptedTimeout! s*
� z%TCPTimeoutTest.testInterruptedTimeoutN)
rF rG rH r� r� r� rA r
r� r r r r r r�
s �r� c @ s e Zd Zdd� Zdd� ZdS )�UDPTimeoutTestc s � fdd�}� � tj|d� d S )Nc s � j �d� � j �d� d S �NrU r� �r< r� r� r� r@ r r r� F s z4UDPTimeoutTest.testUDPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (UDP)r� r� r r@ r �testUDPTimeoutE s
�z UDPTimeoutTest.testUDPTimeoutc C sz d}z | j �d� | j �d�}W nF tjy> | �d� Y n* tyR d}Y n | �d� Y n0 |sv| �d� d S )NFrz r� z%caught timeout instead of error (UDP)Tz!caught unexpected exception (UDP)�1recv() returned success when we did not expect it�r< r� r� r r6 r r r� r r r r� L s z UDPTimeoutTest.testTimeoutZeroN)rF rG rH r r� r r r r r C s r c @ s e Zd Zdd� Zdd� ZdS )�UDPLITETimeoutTestc s � fdd�}� � tj|d� d S )Nc s � j �d� � j �d� d S r r r� r@ r r r� _ s z<UDPLITETimeoutTest.testUDPLITETimeout.<locals>.raise_timeoutz.Error generating a timeout exception (UDPLITE)r� r� r r@ r �testUDPLITETimeout^ s
�z%UDPLITETimeoutTest.testUDPLITETimeoutc C sz d}z | j �d� | j �d�}W nF tjy> | �d� Y n* tyR d}Y n | �d� Y n0 |sv| �d� d S )NFrz r� z)caught timeout instead of error (UDPLITE)Tz%caught unexpected exception (UDPLITE)r r
r� r r r r� e s z"UDPLITETimeoutTest.testTimeoutZeroN)rF rG rH r r� r r r r r Z s r c @ s e Zd Zdd� Zdd� ZdS )�TestExceptionsc C sJ | � ttt�� | � ttjt�� | � ttjt�� | � ttjt�� d S r3 )r
�
issubclassr r� r r" r# r6 r@ r r r �testExceptionTreeu s z TestExceptions.testExceptionTreec C sv t � t jt jd�}t � t jt jd|�� �}|�� | �|j� | �t�� |� d� W d � n1 sh0 Y d S )Nr F)
r r; r0 r r rV r# r! r r )rA Zsock0r� r r r �test_setblocking_invalidfd{ s � z)TestExceptions.test_setblocking_invalidfdN)rF rG rH r r r r r r r
s s r
r~ �Linux specific testc @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�TestLinuxAbstractNamespace�l c
C s� d}t � t jt j���}|�|� |�� t � t jt j��j}|�|�� � |�� d �0}| �|�� |� | �|� � |� W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y d S )Ns python-test-hello �r )
r r� r0 ra r? r� r� r� r� Z getpeername)rA rM �s1r3 Zs3r r r �testLinuxAbstractNamespace� s
z5TestLinuxAbstractNamespace.testLinuxAbstractNamespacec C s` dd| j d }t�tjtj��*}|�|� | �|�� |� W d � n1 sR0 Y d S )Nr[ � hr )�
UNIX_PATH_MAXr r� r0 ra r� r� �rA rM r r r r � testMaxName� s
z&TestLinuxAbstractNamespace.testMaxNamec C sR dd| j }t�tjtj�� }| �t|j|� W d � n1 sD0 Y d S )N� �h)r r r� r0 r! r ra r r r r �testNameOverflow� s z+TestLinuxAbstractNamespace.testNameOverflowc C sF t � t jt j�}z&|�d� | �|�� d� W |�� n
|�� 0 d S )Nz
python test �
python test )r r� r0 ra r� r� r r r r r � testStrName� s
z&TestLinuxAbstractNamespace.testStrNamec C sR t � t jt j��.}|�td�� | �|�� d� W d � n1 sD0 Y d S )Nr )r r� r0 ra r� r� r� r r r r �testBytearrayName� s z,TestLinuxAbstractNamespace.testBytearrayNamec C s� d}t � t jt j���}|�d� | �|�� |� t � t jt j��>}|�d� | �|�� |� | �|�� |�� � W d � n1 s�0 Y W d � n1 s�0 Y d S )Ns
^ [0-9a-f]{5}r& )r r� r0 ra Z assertRegexr� r� )rA Zabstract_addressr r3 r r r � testAutobind� s
z'TestLinuxAbstractNamespace.testAutobindN)
rF rG rH 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 s t � t jt j�| _d S r3 )r r� r0 r� r@ r r r rB � s zTestUnixDomain.setUpc C s | j �� d S r3 )r� r r@ r r r rD � s zTestUnixDomain.tearDownc
C s: z t �|�W S ty4 | �d�|t�� �� Y n0 d S )NzBPathname {0!a} cannot be represented in file system encoding {1!r})r� �fsencoder� rc r rd �getfilesystemencoding�rA r� r r r �encoded� s ��zTestUnixDomain.encodedc
C sZ zt �||� W nD tyT } z,t|�dkr>| �d�|�� n� W Y d }~n
d }~0 0 d S )NzAF_UNIX path too longz6Pathname {0!a} is too long to serve as an AF_UNIX path)r r� r r rc r )rA r� r� r� r r r ra � s ��zTestUnixDomain.bindc C s | � | j�� d� d S )N)r& N)r r� r� r@ r r r r� � s zTestUnixDomain.testUnboundc C s@ t j�tj�}| �| j|� | �tj|� | � | j�
� |� d S r3 ) r� r� �abspathr �TESTFNra r� rV r� r� r� r$ r r r � testStrAddr� s zTestUnixDomain.testStrAddrc C sF t j�tj�}| �| j| �|�� | �tj |� | �
| j� � |� d S r3 ) r� r� r& r r' ra r� r% rV 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% ra r� r� rV r� r� r� )rA 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_UNENCODABLErc r� r� r& ra r� rV r� r� r� r$ r r r �testUnencodableAddr� s
z"TestUnixDomain.testUnencodableAddrr~ r c C s | � t| jjd� d S )Nr& )r! r r� ra r@ r r r �testEmptyAddress s z TestUnixDomain.testEmptyAddressN)rF rG rH rB rD r% ra r� r( r) r, r- r� rC rd 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 s t j| |d� d S r� r� r� r r r rM s zBufferIOTest.__init__c C sZ t � ddgtt� �}| j�|�}| �|tt�� |�� }|d tt�� }| �|t� d S �Nr r )r r� r� r� rD r� ro �rA r rj r� r r r �testRecvIntoArray s z BufferIOTest.testRecvIntoArrayc C s t t�}| j�|� d S r3 �rv r� r� r� �rA r r r r �_testRecvIntoArray s z BufferIOTest._testRecvIntoArrayc C sD t d�}| j�|�}| �|tt�� |d tt�� }| �|t� d S r� )r� r� rD r� r� r� r1 r r r �testRecvIntoBytearray s
z"BufferIOTest.testRecvIntoBytearrayc C sH t d�}| j�t|��}| �|tt�� |d tt�� }| �|t� d S r� )r� r� rD rt r� r� r� r1 r r r �testRecvIntoMemoryview' s
z#BufferIOTest.testRecvIntoMemoryviewc C s^ t � ddgtt� �}| j�|�\}}| �|tt�� |�� }|d tt�� }| �|t� d S r0 )r r� r� r� rG r� ro �rA r rj r� r� r r r �testRecvFromIntoArray0 s z"BufferIOTest.testRecvFromIntoArrayc C s t t�}| j�|� d S r3 r3 r4 r r r �_testRecvFromIntoArray8 s z#BufferIOTest._testRecvFromIntoArrayc C sH t d�}| j�|�\}}| �|tt�� |d tt�� }| �|t� d S r� )r� r� rG r� r� r� r8 r r r �testRecvFromIntoBytearray< s
z&BufferIOTest.testRecvFromIntoBytearrayc C sL t d�}| j�t|��\}}| �|tt�� |d tt�� }| �|t� d S r� )r� r� rG rt r� r� r� r8 r r r �testRecvFromIntoMemoryviewE s
z'BufferIOTest.testRecvFromIntoMemoryviewc C s t d�}| �t| jj|d� d S )Nr^ r� )r� r! r� r� rG r4 r r r �testRecvFromIntoSmallBufferN s z(BufferIOTest.testRecvFromIntoSmallBufferc C s | j �t� d S r3 r� r@ r r r � _testRecvFromIntoSmallBufferS s z)BufferIOTest._testRecvFromIntoSmallBufferc C s$ t � }| j�|� | j�|d� d S r� )r� r� rG r4 r r r �testRecvFromIntoEmptyBufferV s z(BufferIOTest.testRecvFromIntoEmptyBufferN)r� )rF rG rH rX rM r2 r5 r6 Z_testRecvIntoBytearrayr7 Z_testRecvIntoMemoryviewr9 r: r; Z_testRecvFromIntoBytearrayr< Z_testRecvFromIntoMemoryviewr= r> r? Z _testRecvFromIntoEmptyBufferr r r r r/ s$
r/ i� �� �� c
C s� t td�sdS z td�} W n tttfy4 Y dS 0 | �6 | D ] }|�d�r@ W d � dS q@W d � n1 sv0 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/modulesztipc NT)r
r r �FileNotFoundError�IsADirectoryErrorr� r� )r r� r r r �isTipcAvailableb s
2rE z6TIPC module is not loaded, please 'sudo modprobe tipc'c @ s e Zd Zdd� ZdS )�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 rB r0 rV r r� r1 r2 �TIPC_ADDR_NAMESEQ�
TIPC_STYPE�
TIPC_LOWER�
TIPC_UPPERra �TIPC_ADDR_NAMErR r* r� rc r� r� )rA r r� �srvaddrZsendaddrr� Zrecvaddrr r r �testRDMy s �
� zTIPCTest.testRDMN)rF rG rH rM r r r r rF v s rF c @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )�TIPCThreadableTestr� c C s t jj| |d� t�| � d S r� r� r� r r r rM � s 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 rB r0 r rV r r� r1 r2 rG rH rI rJ ra r? rn r� r� r� )rA rL r r r rB � 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 rB r0 r� rV r rK rH rI rR rJ r� r� �cliaddr)rA r� r r r r� � s
� z TIPCThreadableTest.clientSetUpc C s, | j �d�}| �|t� | �| j| j� d S r� )r� r� r� r� rO r� r$ r r r r� � s z TIPCThreadableTest.testStreamc C s | j �t� | j �� d S r3 r� r@ r r r r� � s z TIPCThreadableTest._testStreamN)r� r� r r r r rN � s
rN c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ContextManagersTestc C s� t � � � }| �|j� W d � n1 s*0 Y | �|j� t � � �}|�� W d � n1 sf0 Y | �|j� t � � � }| �t|jd� W d � n1 s�0 Y | �|j� d S �Nr' )r r� r� r
r r! r r� r� r r r �_testSocketClass� s
*
&
.z$ContextManagersTest._testSocketClassc C s2 | j �� \}}| �|j� |�d�}|�|� d S r� �r< r� rV r r� r� �rA r� r� r^ r r r �testCreateConnectionBase� s
z,ContextManagersTest.testCreateConnectionBasec C sl | j �� }t�|��8}| �|j� |�d� | �|�d�d� W d � n1 sR0 Y | � |j� d S )Nr' r� )
r< r� r r� r� r� r� r� r� r
�rA rM r� r r r �_testCreateConnectionBase� s
0z-ContextManagersTest._testCreateConnectionBasec C s2 | j �� \}}| �|j� |�d�}|�|� d S r� rS rT r r r �testCreateConnectionClose� s
z-ContextManagersTest.testCreateConnectionClosec C s\ | j �� }t�|��}|�� W d � n1 s20 Y | �|j� | �t|j d� d S rQ )
r< r� r r� r r
r� r! r r� rV r r r �_testCreateConnectionClose� s
& z.ContextManagersTest._testCreateConnectionCloseN)rF rG rH rR rU rW rX rY r r r r rP � s
rP 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 sX t � t jt jt jB ��.}| �|jt j� | �|�� � W d � n1 sJ0 Y d S r3 )r r; r0 r� r� r� r� �get_inheritabler r r r �test_SOCK_CLOEXEC� s
�z!InheritanceTest.test_SOCK_CLOEXECc C s@ t � � }|� | �|�� d� W d � n1 s20 Y d S r� )r r� r[ r� r r r �test_default_inheritable� s z(InheritanceTest.test_default_inheritablec C st t � � }|�T |�� }|�� |� | �|�� d� W d � n1 sH0 Y W d � n1 sf0 Y d S r� )r r� r r� r[ )rA r� Znewsockr r r �test_dup� s zInheritanceTest.test_dupc C sd t � � }|�D |�d� | �|�� d� |�d� | �|�� d� W d � n1 sV0 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 � � }|�b |�� }| �|�� d� t�|tj�}|tj M }t�|tj|� | �|�� d� W d � n1 st0 Y d S )NFT)r r r� r[ r �F_GETFD�
FD_CLOEXECZF_SETFD)rA r� r' 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 � n1 sn0 Y d S )NTr )r r r� r ra rb r_ )rA r� r' r r r � test_set_inheritable_cloexec s �
�z,InheritanceTest.test_set_inheritable_cloexecc C sH t �� \}}| �|j� | �|j� | �|�� d� | �|�� d� d S r� )r r� rV r r� r[ )rA r r3 r r r �test_socketpair s
z InheritanceTest.test_socketpair)rF rG rH r� rA r
r r r� r\ r] r^ r` rC r rc rd re r r r r rZ � s �
rZ r zSOCK_NONBLOCK not definedc @ s, e Zd Zd dd�Ze�ddd�dd � �Zd
S ) �NonblockConstantTestTrz c C s� |rj| � |jtj� | � |�� |� | �t�|tjtj �tj @ � |dkrZ| �
|� � � q�| �|� � � nL| � |jtj� | � |�� d � | �
t�|tjtj �tj @ � | �|� � � d S r� ) r� r� r r0 r� r
r r| r� r} r� r{ )rA r �nonblockr6 r r r �
checkNonblock) s � �z"NonblockConstantTest.checkNonblockr� r� r� c C s� t � t jt jt jB ���}| �|� |�d� | j|dd� |�d� | �|� |�d � | j|dd� |�d� | j|dd� |�d� | j|dd� W d � n1 s�0 Y t �� }t �d� t � � �}| �|� W d � n1 s�0 Y t �d � t � � � }| �|d� W d � n1 �s.0 Y t �d� t � � � }| j|dd� W d � n1 �sp0 Y t �d � t � � � }| �|d� W d � n1 �s�0 Y t �|� d S )NTF)rg g @r4 rz ) r r; r0 r rh r r� r4 r5 )rA r �tr r r �test_SOCK_NONBLOCKB s:
�
,
(
,
.
,z'NonblockConstantTest.test_SOCK_NONBLOCKN)Trz )rF rG rH rh r r� rj r r r r rf & s
rf r7 r8 zneed multiprocessingc @ s@ e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
S )�TestSocketSharingc C sD |� � }|� � }t�|�}|�� \}}|�|� |�� |�� d S r3 )rp r � fromsharer� r� r )r] �qZsdata�messager r3 r� r r r �remoteProcessServeri s
z%TestSocketSharing.remoteProcessServerc C s� t �� }t j| j|fd�}|�� | j�|j�}| j�� }| j� � |�
|� d}|�
|� t � |�}g }|�
d�}|szq�|�|� qj|� � d�|�}| �||� |�� d S )N�rL rR s
slapmahfror� r� )�multiprocessingry ZProcessro rO r< �share�pidr� r r� r r� r� r� r r� ) rA rm r r^ r� rn r �m�receivedr r r � testSharew s&
zTestSocketSharing.testSharec C s@ | j �t�� �}| �ttj|d d� � | �ttj|d � d S )NrY r' )r< rr r� �getpidr! r� r rl )rA r^ r r r �testShareLength� s z!TestSocketSharing.testShareLengthc C sb | � |�� d � | � |�� |�� � | � |j|j� | � |j|j� |jdkr^| � |j|j� d S r� )r� r� r r� r )rA Zorg�otherr r r �compareSockets� s
z TestSocketSharing.compareSocketsc C sD | j �t�� �}t�|�}z| �| j |� W |�� n
|�� 0 d S r3 )r< rr r� rw r rl rz r )rA r^ r r r r �testShareLocal� s
z TestSocketSharing.testShareLocalc
C s� t jt jg}t jt jg}|D ]�}|D ]�}zt � ||�}W n tyN Y q$Y n0 zH|�t�� �}t � |�}z| �
||� W |� � n
|� � 0 W |� � q$|� � 0 q$q d S r3 ) r r; r� r0 r! r rr r� rw rl rz r )rA Zfamilies�typesr ri �sourcer^ Zsharedr r r � testTypes� s
zTestSocketSharing.testTypesN)
rF rG rH ry ro rv rx rz r{ r~ r r r r rk d s
#rk 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 � rh r� c C s� dd� }d� dd� t| j�D ��}ttjd��.}|| j| j�D ]}|�|� q>W d � n1 sb0 Y ttjd��}|�� | _ W d � n1 s�0 Y d S )Nc s s& | |kr|V | |8 } q | r"| V d S r3 r )�total�stepr r r �chunks� s
z0SendfileUsingSendTest.setUpClass.<locals>.chunksr� c S s g | ]}t �tj��� �qS r )�randomZchoice�stringZ
ascii_lettersr� r� r r r r� � s �z4SendfileUsingSendTest.setUpClass.<locals>.<listcomp>r� r
)
r r �BUFSIZEr r r' �FILESIZEr� r� �FILEDATA)r] r� �chunkr Zcsizer r r �
setUpClass� s
�*
z SendfileUsingSendTest.setUpClassc C s t �t j� d S r3 )r r� r' )r] r r r �
tearDownClass� s z#SendfileUsingSendTest.tearDownClassc C s8 | j �tj� | j �� \}}|�| j� | �|j� |S r3 )r< r� r r� r� �TIMEOUTrV r r� r r r � accept_conn� s
z!SendfileUsingSendTest.accept_connc C s, g }|� | j�}|sq"|�|� qd�|�S r� )r� r� r� r )rA r� ru r� r r r � recv_data� s z SendfileUsingSendTest.recv_datac C s
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�|��f}|�B}| �|�}||�}| �|| j � | �|�
� | j � W d � n1 sn0 Y W d � n1 s�0 Y d S )Nr
� r< r� r r r' r r� r� r� r� �tell�rA rM �filer� �meth�sentr r r �_testRegularFile s
z&SendfileUsingSendTest._testRegularFilec C s6 | � � }| �|�}| �t|�| j� | �|| j� d S r3 �r� r� r� r� r� r� �rA r� r^ r r r �testRegularFile s
z%SendfileUsingSendTest.testRegularFilec C s� | j �� }t�| j�}t�|��p}|�L}|�|�}| �|| j � | �|�
� | j � | � tj |j
|� W d � n1 sx0 Y W d � n1 s�0 Y d S r3 )r< r� �io�BytesIOr� r r� r) r� r� r� r! r. r/ )rA rM r� r� r� r r r �_testNonRegularFile s
�z)SendfileUsingSendTest._testNonRegularFilec C s6 | � � }| �|�}| �t|�| j� | �|| j� d S r3 r� r� r r r �testNonRegularFile s
z(SendfileUsingSendTest.testNonRegularFilec C s� | j �� }tjd }t|d�� | �tj|� W d � n1 sB0 Y t|d�}t�|��b}|�>}| � |�}||�}| �
|d� | �
|� � d� W d � n1 s�0 Y W d � n1 s�0 Y d S )N�2r� r
r ) r< r� r r' r rV r� r r� r� r� r� )rA rM �filenamer� r� r� r� r r r �_testEmptyFileSend& s
,
z(SendfileUsingSendTest._testEmptyFileSendc C s" | � � }| �|�}| �|d� d S r� )r� r� r� r� r r r �testEmptyFileSend2 s
z'SendfileUsingSendTest.testEmptyFileSendc C s� | j �� }ttjd�}t�|��n}|�J}| �|�}||dd�}| �|| j d � | �|�
� | j � W d � n1 sv0 Y W d � n1 s�0 Y d S )Nr
� )�offsetr� r� r r r � _testOffset9 s
z!SendfileUsingSendTest._testOffsetc C sB | � � }| �|�}| �t|�| jd � | �|| jdd � � d S )Nr� r� r� r r r �
testOffsetB s
z SendfileUsingSendTest.testOffsetc C s� | j �� }ttjd�}tj|tjd�}|�j |�F d}| �|�}|||d�}| � ||� | � |�
� |� W d � n1 s|0 Y W d � n1 s�0 Y d S )Nr
r4 �GKL �r� � r< r� r r r' r r� r� r� r� r� �rA rM r� r� r� r� r� r r r �
_testCountJ s
�
z SendfileUsingSendTest._testCountc C s@ d}| � � }| �|�}| �t|�|� | �|| jd |� � d S )Nr� �r� r� r� r� r� �rA r� r� r^ r r r � testCountV s
z SendfileUsingSendTest.testCountc C s� | j �� }ttjd�}tj|tjd�}|�j |�F d}| �|�}|||d�}| � ||� | � |�
� |� W d � n1 s|0 Y W d � n1 s�0 Y d S )Nr
r4 r r� r� r� r r r �_testCountSmall_ s
�
z%SendfileUsingSendTest._testCountSmallc C s@ d}| � � }| �|�}| �t|�|� | �|| jd |� � d S r� r� r� r r r �testCountSmallk s
z$SendfileUsingSendTest.testCountSmallc C s� | j �� }ttjd�}tj|dd��p}|�L}d}| �|�}||d|d�}| �||� | �|� � |d � W d � n1 s|0 Y W d � n1 s�0 Y d S )Nr
r� r4 駆 �� )r� r� )
r< r� r r r' r r� r� r� r� r� r r r �_testCountWithOffsett s
z*SendfileUsingSendTest._testCountWithOffsetc C sD d}| � � }| �|�}| �t|�|� | �|| jd|d � � d S )Nr� r� r� r� r r r �testCountWithOffset~ s
z)SendfileUsingSendTest.testCountWithOffsetc C s� | j �� }ttjd�}t�|��f}|�B}|�d� | �|�}| � t
||� | � t
|j |� W d � n1 sn0 Y W d � n1 s�0 Y d S )Nr
F) r< r� r r r' r r� r r� r! r� r) �rA rM r� r� r� r r r �_testNonBlocking� s
z&SendfileUsingSendTest._testNonBlockingc C s | � � }|�d�r | �d� d S )Nrh z$was not supposed to receive any data)r� r� r �rA r� r r r �testNonBlocking� s
z%SendfileUsingSendTest.testNonBlockingc C s� | j �� }ttjd�}tj|tjd�}|�T |�0 | �|�}||�}| � || j
� W d � n1 sf0 Y W d � n1 s�0 Y d S )Nr
r4 ) r< r� r r r' r r� r� r� r� r� r� r r r �_testWithTimeout� s
�
z&SendfileUsingSendTest._testWithTimeoutc C s6 | � � }| �|�}| �t|�| j� | �|| j� d S r3 r� r� r r r �testWithTimeout� s
z%SendfileUsingSendTest.testWithTimeoutc C s� | j �� }ttjd��^}t�|��4}|�d� | �|�}| � tj
||� W d � n1 s\0 Y W d � n1 sz0 Y d S )Nr
g{�G�z�?) r< r� r r r' r r� r� r� r! r6 r� r r r � _testWithTimeoutTriggeredSend� s
z3SendfileUsingSendTest._testWithTimeoutTriggeredSendc C s | � � }|�d� t�d� d S )Ni�X r )r� r� r� r� r� r r r � testWithTimeoutTriggeredSend� s
z2SendfileUsingSendTest.testWithTimeoutTriggeredSendc C s d S r3 r r@ r r r � _test_errors� s z"SendfileUsingSendTest._test_errorsc C s� t tjd��X}tjtjd��*}| �|�}| �td||� W d � n1 sL0 Y W d � n1 sj0 Y t tjd��R}t�� �*}| �|�}| �td||� W d � n1 s�0 Y W d � n1 s�0 Y t tjd���}t�� �j}| �|�}| jtd||dd� | jtd||d d� | jtd||d
d� | jtd||d d� W d � n1 �sj0 Y W d � n1 �s�0 Y d S ) Nr
r� r0 r� z binary modezpositive integerr� r� r� r rY ) r r r' r r! r� r r� r� )rA r� r r� r r r � test_errors� s6
�@
�@
����z!SendfileUsingSendTest.test_errorsN)&rF rG rH rX r� r� r� r r� r� 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� r r r r r � sD
r r) z%os.sendfile() required for this test.c @ s e Zd ZdZdd� ZdS )�SendfileUsingSendfileTestzB
Test the sendfile() implementation of socket.sendfile().
c C s
t |d�S )Nr/ r� r� r r r r� � s z(SendfileUsingSendfileTest.meth_from_sockN)rF rG rH rX r� r r r r r� � s r� zAF_ALG requiredc @ s| e Zd Zdd� Ze�dd�dd� �Zdd� Ze�dd �d
d � �Ze�dd �d
d� �Z e�dd �dd� �Z
dd� Z dd� Z dS )�LinuxKernelCryptoAPIc
C sl t � t jt jd�}z|�||f� W n> tyb } z&|�� t�t|�||��W Y d }~nd }~0 0 |S d S r� ) r r) r' ra rC r r� ZSkipTestr )rA �typr� r� r� r r r �
create_alg� s (z LinuxKernelCryptoAPI.create_algr) r� c C s� t �d�}| �dd���}|�� \}}|�, |�d� | �|�d�|� W d � n1 sZ0 Y |�� \}}|�V |�dtj � |�dtj � |�dtj � |�d � | �|�d�|� W d � n1 s�0 Y W d � n1 s�0 Y d S )
NZ@ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad�hashZsha256s abcrV � a� b� cr� )
rv �fromhexr� r� r� r� r� r� r �MSG_MORE)rA r �algo�oprs r r r � test_sha256� s
0
z LinuxKernelCryptoAPI.test_sha256c C s� d}d}t �d�}| �dd��n}|�tjtj|� |�� \}}|�, |�|� | � |�
d�|� W d � n1 st0 Y W d � n1 s�0 Y d S )Ns Python loves AF_ALGs what do ya want for nothing?Z(193dbb43c6297b47ea6277ec0ce67119a3f3aa66r� z
hmac(sha1)rV ) rv r� r� r� r �SOL_ALG� ALG_SET_KEYr� r� r� r� )rA �keyr^ r r� r� rs 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 s�0 Y |�� \}}|�6 |j |gtj|d� | �
|�|�|� W d � n1 s�0 Y d } |g| }
|�� \}}|�0 |j |
tj
|d� |�|| �} W d � n1 �sX0 Y | �
t| �|| � | �
| d |� |� |�� \}}|�2 |j | gtj|d� |�|| �} W d � n1 �s�0 Y | �
t| �|| � | �
| || � W d � n1 �s0 Y d S )
NZ 06a9214036b8a15b512e03d534120006Z 3dafba429d9eb430b422da802c9fac41s Single block msgZ e353779c1079aeb82708942dbe77181aZskcipherzcbc(aes))r� �ivr� )r� r� r� )rv r� r� r� r� r r� r� r� �
sendmsg_afalg�ALG_OP_ENCRYPTr� r� r� r� �ALG_OP_DECRYPT)
rA r� r� r� Z
ciphertextZmsglenr� r� rs Z
multiplierZlongmsg�encZdecr r r � test_aes_cbc sL
�
0 �0
�. �.z!LinuxKernelCryptoAPI.test_aes_cbcrr c C s t �d�}t �d�}t �d�}t �d�}t �d�}t �d�}t|�}t|�}| �dd����} | �tjtj|� | �tjtjd |� | � � \}
} |
�� |
j
tj ||tj d � |
�
|tj � |
�
|� |
�|t|� | �} | �|| || � � | �|| | d � � W d � n1 �s0 Y | � � \}
} |
�p || }
|
j
|
gtj ||d
� |
�|t|� | �} | �|| || � � | �|| | d � � W d � n1 �s�0 Y t�d �j}| � � \}
} |
�� || }
|
�|
gtjtj|tj �gtjtj|t|��| gtjtj||�gf� |
�t|
�| �} | �|| || � � | �|| | d � � W d � n1 �sx0 Y | � � \}
} |
�X || | }
|
j
|
gtj||d
� |
�t|
�| �} | �|| |d � � W d � n1 �s�0 Y W d � n1 �s0 Y d S ) NZ c939cc13397c1d37de6ae0e1cb7c423cZb3d8cc017cbb89b39e0f67e2Z c3b3c41f113a31b73d9a5cd432103069Z 24825602bd12a984e0092d3e448eda5fZ 93fe7d9e9bfd10348a5606e5cafa7354Z 0032a1dc85f1c9786925a2e71d8272ddZaeadzgcm(aes))r� r� �assoclenr� )r� r� r� r )rv r� r� r� r� r r� r� ZALG_SET_AEAD_AUTHSIZEr� r� r� r� r� r� r� r ZStructrW r� Z
ALG_SET_OPZ
ALG_SET_IVZALG_SET_AEAD_ASSOCLENr� )rA r� r� ZplainZassocZ expected_ctZ expected_tagZtaglenr� r� r� rs �resr� Z pack_uint32r r r �test_aead_aes_gcm7 sl
�
�
6 �6 ��6 �z&LinuxKernelCryptoAPI.test_aead_aes_gcmc C s� | � dd��v}t�d�}|�tjtj|� |�� \}}|�* |�d�}| � t
|�d� W d � n1 sj0 Y W d � n1 s�0 Y d S )N�rngZdrbg_pr_sha256ri ) r� r� �urandomr� r r� r� r� r� r� r� )rA r� Z
extra_seedr� rs Zrnr r r �test_drbg_pr_sha256v s
z(LinuxKernelCryptoAPI.test_drbg_pr_sha256c C sP t � t jt jd�}|��" | �t�� |�� W d � n1 sB0 Y | �t�� |jd d� W d � n1 sx0 Y | �t�� |�d� W d � n1 s�0 Y | �t�� |jt jd d� W d � n1 s�0 Y | �t�� |jt jdd� W d � n1 �s"0 Y W d � n1 �sB0 Y d S )Nr )r� r )r� r� rY )r r) r' r! 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 sH0 Y | �t d�� |�d� W d � n1 s~0 Y | �t�� |�d� W d � n1 s�0 Y | �t d�� |�d� W d � n1 s�0 Y d S )Nr )Z
tttttttttttttr� z
type too long)Zttttttttttttttr� )r� Z?nnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnz
name too long)r� Z@nnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnn)
r r) r' rV r r! rC ra r r� r� r r r �test_length_restriction� s (( (z,LinuxKernelCryptoAPI.test_length_restrictionN)
rF rG rH r� r r� r� r� r� r� r� r� r� r r r r r� � s
(
>
r� �winzrequires Windowsc @ s e Zd Zh d�Zdd� ZdS )�TestMSWindowsTCPFlags> Z TCP_KEEPCNTZ TCP_KEEPIDLEZ TCP_FASTOPENZ TCP_NODELAYZ
TCP_MAXSEGZ
TCP_KEEPINTVLc s6 dd� t t�D �}� fdd�|D �}� �g |d� d S )Nc S s g | ]}|� d �r|�qS )ZTCP)r� �r� r r r r r� � r� z<TestMSWindowsTCPFlags.test_new_tcp_flags.<locals>.<listcomp>c s g | ]}|� j vr|�qS r )�
knownTCPFlagsr� r@ r r r� � r� zANew TCP flags were discovered. See bpo-32394 for more information)r� r r� )rA Zprovided�unknownr r@ r �test_new_tcp_flags� s
�z(TestMSWindowsTCPFlags.test_new_tcp_flagsN)rF rG rH r� r� r r r r r� � s
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��8}| �|�� d d� | �|�� d |� W d � n1 sT0 Y t jr�tjd|ftjd��8}| �|�� d d� | �|�� d |� W d � n1 s�0 Y d S )Nr r r r� r� )r r� r �
create_serverr� r� r� r� �rA r> r� r r r � test_address� s 2
�z CreateServerTest.test_addressc C s� t �d��0}| �|jt j� | �|jt j� W d � n1 s@0 Y tjr�t jdt j d��0}| �|jt j � | �|jt j� W d � n1 s�0 Y d S )Nr �r� r r� )
r r� r� r r; r� r0 r r� r� )rA r� r r r r �test_family_and_type� s .z%CreateServerTest.test_family_and_typec C s� t td�sD| �t�� tjddd� W d � q�1 s80 Y n�t�d��,}|�tjtj�}| �|d� W d � n1 s�0 Y tjddd��,}|�tjtj�}| � |d� W d � n1 s�0 Y d S )N� SO_REUSEPORT)r� r T)Z
reuse_portr )
r
r r! r� r� r� r1 r� r� r� )rA r� rA r r r �test_reuse_port� s
. *z CreateServerTest.test_reuse_portr5 rl rm r; c C s4 t jdt jd��}W d � n1 s&0 Y d S )Nr� r� )r r� r� r� r r r �test_ipv6_only_default� s z'CreateServerTest.test_ipv6_only_default� dualstack_ipv6 not supportedc C sF t jdt jdd�� }| �|jt j� W d � n1 s80 Y d S )Nr� T�r Zdualstack_ipv6)r r� r� r� r r� r r r �test_dualstack_ipv6_family� s
�z+CreateServerTest.test_dualstack_ipv6_familyN)rF rG rH r� r� r� r� rC r
r rA 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� | �� | � � \}}|�P � ��j� |�d�}|sNW d � W d � d S |�|� W d � n1 sl0 Y W d � n1 s�0 Y d S r� )r� r� r6 r� r� )r� r� rs r� �ry rA r r �run� s
z3CreateServerFunctionalTest.echo_server.<locals>.runrp ) rN rv r� r6 rM rO rV r rm )rA r� r� r~ r r� r � echo_server� s
z&CreateServerFunctionalTest.echo_serverc C sb t j |d��B}|�| j� |�|� |�d� | �|�d�d� W d � n1 sT0 Y d S )Nr� r' r� )r r� r6 r� r� r� r� )rA r� r r� r r r � echo_client s
z&CreateServerFunctionalTest.echo_clientc C sV t �� }t�d|f��,}| �|� | �d|ftj� W d � n1 sH0 Y d S )Nr& r )r r� r r� r� r� r; r� r r r � test_tcp4 s
z$CreateServerFunctionalTest.test_tcp4r; c C s\ t �� }tjd|ftjd��,}| �|� | �d|ftj� W d � n1 sN0 Y d S )Nr& r� r� �r r� r r� r� r� r� r� r r r � test_tcp6 s
�
z$CreateServerFunctionalTest.test_tcp6r� c C s^ t �� }tjd|ftjdd��,}| �|� | �d|ftj� W d � n1 sP0 Y d S )Nr& Tr� r )r r� r r� r� r� r� r; r� r r r �test_dual_stack_client_v4 s �
z4CreateServerFunctionalTest.test_dual_stack_client_v4c C s^ t �� }tjd|ftjdd��,}| �|� | �d|ftj� W d � n1 sP0 Y d S )Nr& Tr� r� r� r� r r r �test_dual_stack_client_v6% s �
z4CreateServerFunctionalTest.test_dual_stack_client_v6N)rF rG rH r r� r6 r� r� r� r� rA r r� r� rC r r� r� r� r r r r r� � s
� � r� �send_fds�recv_fdsc @ s e Zd Zdd� ZdS )�SendRecvFdsTestsc C sn dd� }dd� }dd� t d�D �}| �||� dd� |D �}t�tjtj�\}}|�v |�R t�|tg|� t�|t t�d t |�d �\}}} }
| �||� W d � n1 s�0 Y W d � n1 s�0 Y | �
|t� | �
t |�t |�� | �
| d
� t |�D ]&\} }|\} }
t �
|
t| ��� � �qt |�D ]*\} } t �| d �}| �
|t| ��� � �q>d S ) Nc S s& | D ] \}}t �|� t �|� qd S r3 �r� r )�pipesr� Zfd2r r r � close_pipes4 s
z8SendRecvFdsTests.testSendAndRecvFds.<locals>.close_pipesc S s | D ]}t �|� qd S r3 r� )r� r' r r r � close_fds9 s z6SendRecvFdsTests.testSendAndRecvFds.<locals>.close_fdsc S s g | ] }t �� �qS r )r� �pipe)r� rs 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� )r rV r r� r� r0 r� r� r� r� r� r� r� r� r r� r� )rA r� r r� r� rt ru r� Zfds2r� r� rV r r r^ r r r �testSendAndRecvFds3 s& &H z#SendRecvFdsTests.testSendAndRecvFdsN)rF rG rH r r r r r r� / s r� c C s t �� } tjt jg| �R � d S r3 )r Zthreading_setupr� ZaddModuleCleanupZthreading_cleanup)� thread_infor r r � setUpModuleY s r �__main__)�r� �testr Z test.supportr rO r� r� r r� r� r� r ro rd r� r� r r� �weakrefr r� r� r r r� r5 r� �_threadr~ rN rq r� r r� r� r� r� r� �systemr r r r r# r% r( r* r, r. r2 r� r7 ZHAVE_SOCKET_CANZHAVE_SOCKET_CAN_ISOTPZHAVE_SOCKET_CAN_J1939ZHAVE_SOCKET_RDSZHAVE_SOCKET_ALGZHAVE_SOCKET_QIPCRTRZHAVE_SOCKET_VSOCKr
ZHAVE_SOCKET_UDPLITEZHAVE_SOCKET_BLUETOOTHr| r� r� r: rI rJ rK rZ rg ri r� r� rA r� r� r� rC r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r
r rD rZ rz r� r� r� r� r� r� r� r� r� r� r� r r r r r
r r/ r? rB r` re rh rw 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- r2 r3 r4 r5 r6 r7 r8 r9 r@ rK rT rY rc rx r� r� r� r� r� r� r� r r� r� r� r� r� r� r r r
r r! r/ rH rI rJ rE rF rN rP rZ rf r� rk r r� r� r� r� r� r� r� r rF �mainr r r r �<module> sX
3q�� �$!$� �$
p
S
Z
*
=
>��.�%{ �7w w= # P; k 7 �
�
�
� � ��
�
�
� �
�
�
�
� �
� � �
�
� �
�
�
�
�
� �
� )��7#j ! $l
?G 9�7JS �� ',C�< e
� H3@
'