Sindbad~EG File Manager
a
��g�� � @ 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%e�&� d kZ'z d dl(Z(W n e �yt dZ(Y n0 d
d � Z)d d
� Z*dd� Z+dd� Z,dd� Z-dd� Z.dd� Z/dd� Z0dd� Z1ej2d d � �Z3e*� Z4e+� Z5e,� Z6e-� Z7e.� Z8e/� Z9e0� Z:e;ed �Z<e1� Z=e�d �j>Z?G d d!� d!e j@�ZAG d"d#� d#e j@�ZBG d$d%� d%eB�ZCG d&d'� d'�ZDG d(d)� d)e j@�ZEG d*d+� d+e j@�ZFG d,d-� d-�ZGG d.d/� d/eAeG�ZHG d0d1� d1eBeG�ZIe �Je<d2�G d3d4� d4eCeG��ZKG d5d6� d6eEeG�ZLG d7d8� d8eFeG�ZMe �Ne!du d9�e �Je:d:�e �Je)� d;kd<�G d=d>� d>e j@eG����ZOG d?d@� d@eH�ZPG dAdB� dBe j@eG�ZQG dCdD� dDe j@�ZRG dEdF� dFeR�ZSG dGdH� dHeDeReG�ZTG dIdJ� dJeSeT�ZUG dKdL� dLeR�ZVG dMdN� dNeV�ZWG dOdP� dPeR�ZXG dQdR� dReX�ZYG dSdT� dTeX�ZZG dUdV� dVeX�Z[G dWdX� dXeX�Z\G dYdZ� dZeX�Z]G d[d\� d\e]�Z^G d]d^� d^e]�Z_d_d`� Z`dadb� Zadcdd� ZbG dedf� dfe j@�Zce �Je4dg�G dhdi� die j@��Zde �Je4dg�G djdk� dkeL��Zee �Je5dl�G dmdn� dne j@��Zfe �Je6do�G dpdq� dqe j@��Zge �Je7dr�G dsdt� dte j@��Zhe �Je7dr�G dudv� dveM��Zie �Je9dw�G dxdy� dye j@��Zje �Ne!du d9�e �Je:d:�G dzd{� d{e j@���Zke �Je=d|�G d}d~� d~e j@��ZlG dd�� d�eP�ZmG d�d�� d�eI�Zne �Je<d2�G d�d�� d�eK��ZoG d�d�� d�eD�ZpG d�d�� d�ep�ZqG d�d�� d�ep�ZrG d�d�� d�ep�ZsG d�d�� d�ep�ZtG d�d�� d�ep�ZuG d�d�� d�ep�ZvG d�d�� d�ev�ZwG d�d�� d�ew�ZxG d�d�� d�ew�ZyG d�d�� d�ep�ZzG d�d�� d�ez�Z{G d�d�� d�ez�Z|G d�d�� d�eqez�Z}G d�d�� d�e j@�Z~G d�d�� d�ev�ZG d�d�� d�ev�Z�G d�d�� d�ereteTeZ�Z�eaejd��G d�d�� d�eye���Z�eaejd��G d�d�� d�e|e���Z�eaejd��G d�d�� d�e}e���Z�G d�d�� d�ereteTe^�Z�eaejd��e �Jej�d��ebd�d��G d�d�� d�eye�����Z�eaejd��e �Jej�d��ebd�d��G d�d�� d�e|e�����Z�eaejd��e �Jej�d��ebd�d��G d�d�� d�e}e�����Z�eaejd��e �Jej�d��eaed��ebd�d��G d�d�� d�e�e������Z�eaejd��e �Jej�d��eaed��ebd�d��G d�d�� d�eqe�e������Z�e �Je<d2�G d�dÄ d�ereteTe[��Z�e �Je<d2�eaejd��G d�dń d�eye����Z�e �Je<d2�eaejd��G d�dDŽ d�e|e����Z�e �Je<d2�eaejd��G d�dɄ d�e}e����Z�e �Je<d2�G d�d˄ d�ereteTe_��Z�eaejd��e �Jej�d��e �Je<d2�ebd�d��G d�d̈́ d�eye������Z�eaejd��e �Jej�d��e �Je<d2�ebd�d��G d�dτ d�e|e������Z�eaejd��e �Jej�d��e �Je<d2�ebd�d��G d�dф d�e}e������Z�eaejd��e �Jej�d��e �Je<d2�eaed��ebd�d��G d�dӄ d�e�e�������Z�eaejd��e �Jej�d��e �Je<d2�eaed��ebd�d��G d�dՄ d�eqe�e�������Z�G d�dׄ d�eueUeY�Z�eaejd��G d�dل d�exe���Z�eaejd��G d�dۄ d�e|e{e���Z�eaejd��G d�d݄ d�e}e{e���Z�G d�d߄ d�eseueUe\�Z�eaejd��e �Ne'd�ebd�d�d�G d�d� d�exe�����Z�eaejd��e �Ne'd�ebd�d�d�G d�d� d�e|e{e�����Z�eaejd��e �Ne'd�ebd�d�d�G d�d� d�e}e{e�����Z�G d�d� d�eueUeW�Z�eaejd��eaed�G d�d� d�exe����Z�eaejd��eaed�G d�d� d�e|e{e����Z�eaejd��eaed�G d�d� d�e}e{e����Z�eaejd�d��eaed�d�d�G d�d�� d�ee����Z�eaejd�d��eaed�d�d�G d�d�� d�eqee����Z�G d�d�� d��Z�eaed��e �Je;ed��� p�e;ed��d��G d��d � �d e�eZ���Z�eaed��e �Je;ed���
p"e;ed��d��G �d�d� �de�eDeSeY���Z�G �d�d� �deH�Z�G �d�d� �deQ�Z�G �d�d� �deQ�Z�G �d �d
� �d
eH�Z�G �d �d � �d eP�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�em�Z�G �d �d � �d e j@�Z�G �d �d � �d eAeG�Z�G �d!�d"� �d"eAeG�Z�G �d#�d$� �d$eA�Z�G �d%�d&� �d&eB�Z�e �Je<d2�G �d'�d(� �d(eC��Z�G �d)�d*� �d*e j@�Z�e �Jej�d+k�d,�G �d-�d.� �d.e j@��Z�e �Je;ed��d/�G �d0�d1� �d1e j@��Z�G �d2�d3� �d3eP�Z��d4Z��d5Z��d6Z��d7�d8� Z�e �Je �d9�G �d:�d;� �d;e j@��Z�e �Je �d9�G �d<�d=� �d=e j@eG��Z�G �d>�d?� �d?eH�Z�G �d@�dA� �dAe j@�Z�e �Je;e�dB��dC�G �dD�dE� �dEe j@��Z�e �JejȐdFk�dG�e �Je �dH�G �dI�dJ� �dJeA���Z�G �dK�dL� �dLeH�Z�e �Je;e�dM��dN�G �dO�dP� �dPeʃ�Z�e �Je8�dQ�G �dR�dS� �dSe j@��Z�e �Jej�͐dT��dU�G �dV�dW� �dWe j@��Z�G �dX�dY� �dYe j@�Z�G �dZ�d[� �d[e j@�Z�eae�d\�eae�d]�eaed�G �d^�d_� �d_e j@����Zѐd`�da� Z�eӐdbk�r�e �ԡ dS (c � 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 �,/usr/local/lib/python3.9/test/test_socket.py�get_cid2 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_isotpI 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_j1939S 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_algg 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_qipcrtrq s r+ c C s t � du} | S )z:Check whether AF_VSOCK sockets are supported on this host.N)r )�retr r r �_have_socket_vsock{ s
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 r1 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
r6 �IPPROTO_UDPLITE�ic @ s e Zd Zdd� Zdd� ZdS )�
SocketTCPTestc C s. t � t jt j�| _t�| j�| _| j�� d S r2 )r
�AF_INETr/ �servr � bind_port�port�listen��selfr r r �setUp� s zSocketTCPTest.setUpc C s | j �� d | _ d S r2 �r; r r? r r r �tearDown� s
zSocketTCPTest.tearDownN��__name__�
__module__� __qualname__rA rC r r r r r9 � s r9 c @ s e Zd Zdd� Zdd� ZdS )�
SocketUDPTestc C s$ t � t jt j�| _t�| j�| _d S r2 )r
r: r r; r r< r= r? r r r rA � s zSocketUDPTest.setUpc C s | j �� d | _ d S r2 rB r? r r r rC � s
zSocketUDPTest.tearDownNrD r r r r rH � s rH c @ s e Zd Zdd� ZdS )�SocketUDPLITETestc C s( t � t jt jt j�| _t�| j�| _d S r2 )r
r: r r7 r; r r< r= r? r r r rA � s zSocketUDPLITETest.setUpN)rE rF rG rA r r r r rI � s rI 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 r2 )�super�__init__� threading�RLock�
_cleanup_lock�r@ �args�kwargs�� __class__r r rL � s z"ThreadSafeCleanupTestCase.__init__c s< | j �" t� j|i |��W d � S 1 s.0 Y d S r2 )rO rK �
addCleanuprP rS r r rU � s z$ThreadSafeCleanupTestCase.addCleanupc s< | j �" t� j|i |��W d � S 1 s.0 Y d S r2 )rO rK �
doCleanupsrP rS r r rV � s z$ThreadSafeCleanupTestCase.doCleanups)rE rF rG �__doc__rL rU rV �
__classcell__r r rS r rJ � s rJ 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 rU r �bind� interfacer �skipTestr? r r r rA s �zSocketCANTest.setUpN) rE rF rG rW ra �bufsize�
can_frame_fmtr �calcsizeZcan_frame_size�bcm_cmd_msg_fmtrA r r r r rY � s
rY 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; rU r r r< r= r rb r? r r r rA s zSocketRDSTest.setUpN)rE rF rG rW rc rA 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 r2 )rA �_ThreadableTest__setUp�_setUpr? r r r rL D 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 �serverExplicitReadyI 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 )N� Fc s � j �� r� j �� �d S r2 )�queue�qsize�getr r? r r �raise_queued_exception[ s
z5ThreadableTest._setUp.<locals>.raise_queued_exception�.�_T)r Zwait_threads_exitZ wait_threads� __enter__rU �__exit__rM �Eventrl � client_ready�donerp �Queue�server_crashed�id�rfind�getattr�thread�start_new_thread� clientRunZ
client_threadrj rm �wait)r@ rs Z
methodnamer8 Z test_methodr r? r rk P 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�
BaseExceptionrp �put�clientTearDownry rm r| r � TypeError)r@ Z test_func�er r r r� r 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 r2 )rz rm r� �exitr? r r r r� � s
z ThreadableTest.clientTearDownN)
rE rF rG rW rL 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�r9 rL ri �r@ r� r r r rL � s z ThreadedTCPSocketTest.__init__c C s t � t jt j�| _d S r2 )r
r: r/ �clir? r r r r� � s z!ThreadedTCPSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r2 �r� r ri r� r? r r r r� � s
z$ThreadedTCPSocketTest.clientTearDownN)r� �rE rF rG rL 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� )rH rL ri r� r r r rL � s z ThreadedUDPSocketTest.__init__c C s t � t jt j�| _d S r2 )r
r: r r� r? r r r r� � s z!ThreadedUDPSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r2 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� )rI rL ri r� r r r rL � s z"ThreadedUDPLITESocketTest.__init__c C s t � t jt jt j�| _d S r2 )r
r: r r7 r� r? r r r r� � s z%ThreadedUDPLITESocketTest.clientSetUpc C s | j �� d | _ t�| � d S r2 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� )rY rL ri r� r r r rL � s z ThreadedCANSocketTest.__init__c C sB t � t jt jt j�| _z| j�| jf� W n ty< Y n0 d S r2 )r
r r r r� r` ra r r? r r r r� � s
z!ThreadedCANSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r2 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 rL ri r� r r r rL � 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� r` �HOST� getsockname�cli_addrr r? r r r r� � s z!ThreadedRDSSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r2 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�TestCaserL ri r� r r r rL � s z&ThreadedVSOCKSocketStreamTest.__init__c C sh t � t jt j�| _| �| jj� | j�t jtf� | j� � | �
� | j� � \| _ | _
| �| j j� d S r2 )r
�AF_VSOCKr/ r; rU r r` �VMADDR_CID_ANY� VSOCKPORTr> rn �accept�conn�connaddrr? r r r rA � s
z#ThreadedVSOCKSocketStreamTest.setUpc C sD t �d� t�tjtj�| _| �| jj� t� }| j� |t
f� d S )N皙�����?) �time�sleepr
r� r/ r� rU r r �connectr� )r@ Zcidr r r r� � s
z)ThreadedVSOCKSocketStreamTest.clientSetUpc C s | j �d�}| �|t� d S �N� )r� �recv� assertEqual�MSG�r@ �msgr r r �
testStream s z(ThreadedVSOCKSocketStreamTest.testStreamc C s | j �t� | j �� d S r2 �r� �sendr� r r? r r r � _testStream s z)ThreadedVSOCKSocketStreamTest._testStreamN)r� �rE rF rG rL rA 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� rL r� r r r rL s z SocketConnectedTest.__init__c C s* t �| � | �� | j�� \}}|| _d S r2 )r� rA rn r; r� �cli_conn�r@ r� �addrr r r rA s
zSocketConnectedTest.setUpc C s | j �� d | _ t�| � d S r2 )r� r r� rC r? r r r rC s
z SocketConnectedTest.tearDownc C s( t �| � | j�t| jf� | j| _d S r2 )r� r� r� r� r� r= � serv_connr? r r r r� s
z SocketConnectedTest.clientSetUpc C s | j �� d | _ t�| � d S r2 )r� r r� r� r? r r r r� $ s
z"SocketConnectedTest.clientTearDownN)r� ) rE rF rG rW rL rA rC 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� rL ri r� r; r� r r r rL + s
zSocketPairTest.__init__c C s t �� S r2 )r
�
socketpairr? r r r r� 1 s zSocketPairTest.socketpairc C s | � � \| _| _d S r2 )r� r; r� r? r r r rA 5 s zSocketPairTest.setUpc C s | j r| j �� d | _ d S r2 rB r? r r r rC 8 s
zSocketPairTest.tearDownc C s d S r2 r r? r r r r� = s zSocketPairTest.clientSetUpc C s$ | j r| j �� d | _ t�| � d S r2 r� r? r r r r� @ s
z SocketPairTest.clientTearDownN)r� ) rE rF rG rL r� rA rC r� r� r r r r r� ) s
r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �SocketTestBasez�A base class for socket tests.
Subclasses must provide methods newSocket() to return a new socket
and bindSock(sock) to bind it to an unused address.
Creates a socket self.serv and sets self.serv_addr to its address.
c C s | � � | _| �� d S r2 )� newSocketr; �
bindServerr? r r r rA V 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� Z s zSocketTestBase.bindServerc C s | j �� d | _ d S r2 rB r? r r r rC _ s
zSocketTestBase.tearDownN)rE rF rG rW rA r� rC r r r r r� M s r� c s e Zd ZdZ� fdd�Z� ZS )�SocketListeningTestMixinz%Mixin to listen on the server socket.c s t � �� | j�� d S r2 )rK rA r; r> r? rS r r rA g s
z SocketListeningTestMixin.setUp)rE rF rG rW rA rX r r rS r r� d 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 r2 )rK rL ri rP rS r r rL t s z ThreadedSocketTestMixin.__init__c C s | � � | _| �� d S r2 )�newClientSocketr� �
bindClientr? r r r r� x 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 r2 r� r? r r r r� � s
z&ThreadedSocketTestMixin.clientTearDown)
rE rF rG rW rL r� r� r� r� rX r r rS r r� l 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 r2 )rK rA rn r; r� r� r� rS r r rA � s
z ConnectedStreamTestMixin.setUpc s | j �� d | _ t� �� d S r2 )r� r rK rC r? rS r r rC � s
z!ConnectedStreamTestMixin.tearDownc s$ t � �� | j�| j� | j| _d S r2 )rK r� r� r� r� r� r? rS r r r� � s
z$ConnectedStreamTestMixin.clientSetUpc s6 z| j �� d | _ W n ty& Y n0 t� �� d S r2 )r� r r rK r� r? rS r r r� � s
z'ConnectedStreamTestMixin.clientTearDown) rE rF rG rW rA rC r� r� rX r r rS 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 r2 )�tempfile�mkdtemp�dir_pathrU �os�rmdirrK rA r? rS r r rA � s
zUnixSocketTestBase.setUpc C s, t j| jd�}t�||� | �tj|� d S )N)�dir)r� �mktempr� r �bind_unix_socketrU r �unlink)r@ �sock�pathr r r r� � s zUnixSocketTestBase.bindSock)rE rF rG rW rA r� rX r r rS 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 r2 )r
�AF_UNIXr/ r? r r r r� � s zUnixStreamBase.newSocketN�rE rF rG rW 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 �Nro )rK rA r� r= r? rS r r rA � s
zInetTestBase.setUpc C s t j|| jd� d S )N��host)r r< r� �r@ r� r r r r� � s zInetTestBase.bindSock) rE rF rG rW r� r� rA r� rX r r rS 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 r2 )r
r: r/ 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 r2 )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 r2 )r
r: r r7 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 r2 )r
r: r/ � 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)rE rF rG rW 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 r2 )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 r2 )r
r� r r7 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 r2 r �rQ rR r r r � client_pass s z%skipWithClientIf.<locals>.client_passc s* t ���| �}t| t�s&� fdd�|_|S )Nc s � S r2 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 r2 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 r2 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� + s �z!requireSocket.<locals>.<listcomp>r� r� c s s | ]
}|V qd S r2 r r� r r r r� . 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� 0 s �z can't create socket({0}): {1}c s s | ]}t |�V qd S r2 )r �r� �or r r r� < r� )r r
r r r r� �format)rQ �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
�
SocketType�assertIsInstancer )r@ r r r r r � test_SocketType_is_socketobjectD 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=0�raddr�� 127.0.0.1r �laddrz[closed])
r
r: r/ �assertIn�fileno�repr� assertNotInr` r r� �r@ r r r r � test_reprK 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: r/ r �familyr� �protor� r r )r@ r �expectedr r r �test_csocket_reprY 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: r/ r r� r r �
gc_collect�ReferenceError�fail)r@ r �pr r r � test_weakreff 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�gaierrorr� r r r �testSocketErrors 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 rU r r` r� r$ r� �sendtor� r � exceptionr )r@ 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 r2 ) r
r: �has_ipv6r� r/ 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 r2 )r
� IPPROTO_TCP� IPPROTO_UDPr1 � 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 r2 ) 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_RECVPKTINFO�IPV6_RTHDR_TYPE_0�IPV6_RECVDSTOPTS�
IPV6_DONTFRAG�
IPV6_CHECKSUM�IPV6_RECVPATHMTU� IPV6_PATHMTU� IPV6_TCLASS� IPV6_PKTINFO� IPV6_NEXTHOP�IPV6_RECVHOPLIMIT� IPV6_HOPOPTS�IPV6_RECVTCLASS�IPV6_RECVRTHDR�IPV6_RECVHOPOPTS�
IPV6_RTHDR�
IPV6_HOPLIMIT� IPV6_DSTOPTS�IPV6_USE_MIN_MTU�IPV6_RTHDRDSTOPTSz Missing RFC3542 socket option '�')r r r
)r@ 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 failurert r zError resolving host to ip.z=Error testing host resolution mechanisms. (fqdn: %s, all: %s)) r
� gethostname�
gethostbynamer rb r �find�
gethostbyaddr�getfqdnr r )r@ �hostname�ipZhname�aliases�ipaddrsZall_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
rX r rZ )r@ 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
rX rZ )r@ Z explanationr� r r r � test_host_resolution_bad_address s �(z3GeneralModuleTests.test_host_resolution_bad_address� sethostnamez test needs socket.sethostname()rW 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
rW rd r �errnoZEPERMrb r� )r@ 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
rh r �intr �
assertGreater�if_nametoindexr� �if_indextoname)r@ Z
interfaces�indexr� Z_index�_namer r r �testInterfaceNameIndex% s
z)GeneralModuleTests.testInterfaceNameIndexrl 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� �� � _DEADBEEFrh � )� �� l ���� ) r$ r r
rl �
OverflowErrorr� r �dictrh )r@ �indicesrm �index2r r r � testInvalidInterfaceIndexToName5 s
*z2GeneralModuleTests.testInvalidInterfaceIndexToNamerk z&socket.if_nametoindex() not available.c C s$ | � ttjd� | � ttjd� d S )Nr rr )r$ r� r
rk r r? r r r � testInvalidInterfaceNameToIndexH 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)�sysr{ rE r
� getnameinfor� r )r@ �origr r r �testRefCountGetNameInfoN s
z*GeneralModuleTests.testRefCountGetNameInfoc C s( zt �dd� W n ty" Y n0 d S )N)r\ r r r r )r
r} r r? r r r �testInterpreterCrashZ s z'GeneralModuleTests.testInterpreterCrashc C s� t jdt jdt jdt jdi}|�� D ]d\}}d|> d }dD ]$}| �||@ ||||@ ��|@ � q<||�}| �||@ |� | �t|d� q$d S )N� � ro )r ro � i ��r� igE#i2Tvl ) r
�htonl�ntohl�htons�ntohs�itemsr� r$ ru )r@ Zsizes�func�size�maskr8 Zswappedr r r �testNtoHb 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 ro r� r� rt rp ���rs ro rq � ) � _testcapiZINT_MIN�INT_MAXr
r� r� r� r� r$ ru � assertWarns�DeprecationWarning)r@ r� Z
s_good_valuesZ
l_good_valuesZ l_bad_valuesZ s_bad_valuesZs_deprecated_values�kr r r �testNtoHErrorsp 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� Zudprp r� )r� r| �platform�
startswithr
�
getservbynamer r r� r Z
is_android�
getservbyportr$ ru )r@ �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�
rp Zspam) r� r
r3 �
gettimeoutr6 r4 r$ �
ValueErrorr� )r@ 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� ����ra )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 r2 r ��a)r: r� r r r� � r� z5GeneralModuleTests.testIPv4toString.<locals>.<lambda>c s � � ttf| |�S r2 �r$ r r� �r� r� r? r r r� � s
�r� r� s � � z 255.0.255.0s ����z170.170.170.170� �1.2.3.4r� ra z0.0.0.z 300.0.0.0za.0.0.0z 1.2.3.4.5�::1)r
r� r� r: r� r �r@ r �g�
assertInvalidr )r: r� r@ 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� r1 �IPv6 not available�+could not import needed symbols from socket�win32�::�&' �IPv6 might not be supportedc s
�� | �S r2 r r� )r� r� r r r� r� z5GeneralModuleTests.testIPv6toString.<locals>.<lambda>c s �� ttf� | �S r2 r� r� �r r@ 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� r1 rb � ImportErrorr| r� r �winerrorr� r �r@ r1 r� r� r )r� r r� r@ 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 r2 r r� )r: r� r r r� ; r� z5GeneralModuleTests.testStringToIPv4.<locals>.<lambda>c s � � ttf| |�S r2 r� r� r? r r r� < s
�z1.0.1.0s z
170.85.170.85s �U�Ura r� r� r� s s r� )r
r� r� r: r� � bytearrayr� r )r: r� r@ r �testStringToIPv47 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� r1 r� r� r� r� r� r� c s
�� | �S r2 r r� )r� r� r r r� b r� z5GeneralModuleTests.testStringToIPv6.<locals>.<lambda>c s �� ttf� | �S r2 r� r� r� r r r� c s
�r� r� s z!aef:b01:506:1001:ffff:9997:55:170s
� ���� Ups s s ) r
r� r� r1 rb r� r| r� r r� r� r� r� r )r� r r� r@ r �testStringToIPv6Q 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� rV r z
%s invalidro )r �find_unused_portr
r: r/ rU r r` r� rX rW r rb r r� )r@ r= r� r� Z
my_ip_addrr r r � testSockNameu s z GeneralModuleTests.testSockNamec C s@ t � t jt j�}| �|j� |�t jt j�}| �|dkd� d S )Nr zinitial mode is reuse) r
r: r/ rU r �
getsockoptr3 r4 � assertFalse�r@ 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 )Nro r zfailed to set reuse mode)
r
r: r/ rU r �
setsockoptr3 r4 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 )Nro � spam)r
r: r/ �
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
r` �
_LOCALHOSTr r r r rf �EBADF�ENOTSOCKr )r@ 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: r/ 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 )Nr� � )r
rU r r r� r$ ru r` r� � itertools�countr rf Z
EADDRINUSE)r@ r� r= Zbig_portZneg_portr8 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_VALSrp )ro �d r� ) r r r
rU 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
rU r r r� r r� rb r$ r� )r@ r �excZ
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� �http�AddressFamily.AF_INET�SocketKind.SOCK_STREAMr )r� r= �r �r� )r )�flags)r� r= r r� r r� u ��AI_NUMERICSERV)r
� getaddrinfor& rf � EAI_SERVICErb r� r� �lenr � IPV6_ENABLEDr r| r� r: r/ r �SOL_TCP�
AI_PASSIVE� AF_UNSPECr$ �UnicodeEncodeErrorr� )
r@ r �infoZinfosr r� ru �socktyper� �br r r �testGetaddrinfo� s�
� � � � �z"GeneralModuleTests.testGetaddrinfoc C s | � ttjdd� d S )N)zmail.python.orgr r )r$ r r
r} r? r r r �test_getnameinfo4 s z#GeneralModuleTests.test_getnameinfo�networkznetwork 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
rX �gethostbyname_exr� r� r/ )r@ r� r r r � test_idna8 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� �math�acosh�rQ r? r r �
ok_handlerO s z@GeneralModuleTests.check_sendall_interrupted.<locals>.ok_handlerc s � � ttjd� dd d S �Nr ro r� r r? r r �raising_handlerQ s zEGeneralModuleTests.check_sendall_interrupted.<locals>.raising_handlerg �?ro � xr )r �signalr
rb r� �SIGALRMr� r$ �ZeroDivisionErrorr� �sendallr �
SOCK_MAX_SIZEr5 r )r@ Z with_timeoutr r �cr � old_alarmr r? r �check_sendall_interruptedI s4
.
�
�
z,GeneralModuleTests.check_sendall_interruptedc C s | � d� d S �NF�r r? r r r �test_sendall_interruptedh s z+GeneralModuleTests.test_sendall_interruptedc C s | � d� d S )NTr r? r r r �%test_sendall_interrupted_with_timeoutk 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: r/ r r� �ResourceWarningr r r r �warningrQ �makefile)r@ r� r r/ r r r r �test_dealloc_warnn 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: r/ r r r� r �r@ 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 ro rp ) r
r` 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 �rw�w�wb��mode)�subTestr
r r� r" )r@ 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 )r@ 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�pickle�HIGHEST_PROTOCOLr$ r� �dumps�loadsr: r� r/ )r@ r� �protocolr 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 rp r )r
r: r/ r` r� r> )r@ �backlog�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 )
r� r
r: r/ r` r� r$ ru r> r� )r@ r� r1 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 rt i����) r$ ru r
r} r r� r� r/ r` 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 r7 r� )r@ ru �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 r7 )r
rh r� r� r r7 r� )r@ �ifindex�test_interfaceru r8 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 r7 r )r
r� r r� r r7 r� )r@ r; ru r8 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 r6 r �ff02::1de:c0:face:8d%�1234)r
rh r} �NI_NUMERICHOST�NI_NUMERICSERVr� )r@ r; r< r8 �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> r6 r r r@ rA )r
r} rB rC r� r )r@ r; r8 rD 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: r/ 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� ro TF)r r
r/ r: r� r� r� � setblocking)r@ rH 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 )Nro c s s | ]\}}|d vr|V qdS )> rH r� Nr )r� r� �kindr r r r� 3 s �zEGeneralModuleTests.test_unknown_socket_family_repr.<locals>.<genexpr>r� )r r� r r r )r
r: r/ �detach�max�
AddressFamily� __members__�values�
SocketKindr� r� r r� r r )r@ 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 r2 �rR )r@ rR r r r rL D s zEGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.__init__c S s | j S r2 rU r? r r r r G s zCGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.filenoN)rE rF rG rL r r r r r �FileC s rV r� rq ) r
r� r �curdir�O_RDONLYr r$ �_GiveupOnSendfile�_sendfile_use_sendfileru r� )r@ rV r� rR r r r �test__sendfile_use_sendfileA 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
rU r rL )r@ r r �styperR �s2r r r �_test_socket_filenoS 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: r/ rU r r` r r� r_ r r r� r� r� r� r� �shutil�rmtreer� r� r� r �PermissionError)r@ r Ztmpdirr r r �test_socket_fileno` 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: r/ 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� rd 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 descriptorrp r� �r' r� r
r: r/ 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 )Nrh i���r� ri 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- rf r� r: r/ )r@ 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 r\ ) r� �NamedTemporaryFiler$ r r
r r: r/ r� r- rf r� )r@ Zafiler/ r r r �%test_socket_fileno_requires_socket_fd� s
. �$z8GeneralModuleTests.test_socket_fileno_requires_socket_fd)YrE rF rG r r r� �
skipUnlessr r r" r' r0 r5 r9 r� r� r< r| r� r r� rU r` rb rc r r
rg ro ry rz r r� r� r � cpython_onlyr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� Zis_resource_enabledr� r r r r r r r r$ r( r/ r2 r3 r5 r9 �skipIfr r= r? rE rF rG rJ rS r[ r_ rc re rg rj rk rl rn r r r r r
B 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 r2 )r
�AF_CANr r r? r r r r5 � 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 r2 )r
rt 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 r2 �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 r2 )r
r r rt 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 r` r� r� )r@ 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 r` 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 ro ) r
r r r r� � SOL_CAN_RAWr� r� r� )r@ 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� r� r� r� r� r� )r@ �can_idZcan_maskZ
can_filterr r r r �
testFilter� s �zBasicCANTest.testFilterN)rE rF rG r5 r� ro r r
rw ry rz r| r r� r� r r r r rr � s( �
�
�
�rr 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� rL r� r r r rL s zCANTest.__init__c C s& t |�}|�dd�}t�| j|||�S )zBuild a CAN frame.r] � )r� �ljustr r� rd )�clsr� �data�can_dlcr r r �build_can_frame s zCANTest.build_can_framec C s& t �| j|�\}}}|||d|� fS )zDissect a CAN frame.N)r r rd )r� �framer� r� r� r r r �dissect_can_frame s zCANTest.dissect_can_framec C s6 | j �| j�\}}| �| j|� | �|d | j� d S r� )r �recvfromrc r� �cfra �r@ r� r� r r r �
testSendFrame s zCANTest.testSendFramec C s | � dd�| _| j�| j� d S )Nr s �r� r� r� r� r? r r r �_testSendFrame s zCANTest._testSendFramec C s$ | j �| j�\}}| �| j|� d S r2 )r r� rc r� r� r� r r r �testSendMaxFrame s zCANTest.testSendMaxFramec C s | � dd�| _| j�| j� d S )Nr s r� r? r r r �_testSendMaxFrame s zCANTest._testSendMaxFramec C sD | j �| j�\}}| �| j|� | j �| j�\}}| �| j|� d S r2 )r r� rc r� �cf1�cf2r� r r r �testSendMultiFrames" s zCANTest.testSendMultiFramesc C s< | � dd�| _| j�| j� | � dd�| _| j�| j� d S )N� s D3"� s �"3)r� r� r� r� r� r? r r r �_testSendMultiFrames) s z CANTest._testSendMultiFramesrt ru c C sP | j �| j�\}}| �| j|� | �|�\}}}| �| j|� | �| j|� d S r2 )r� r� rc r� r� r� r� r� )r@ r� r� r� r� r� r r r �_testBCM0 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 �dks�J �t�| j|||||||| |
�
} | | j } |�| �}
| �|
t| �� d S )N�# )� � �� r i" ro r� )r
r r rt rU r r� ra r� �bytesr� r� r� rv r� r r� rf r� r� )r@ ZbcmZopcoder� r� Z
ival1_secondsZ
ival1_usecZ
ival2_secondsZ
ival2_usecZ
bcm_can_idZnframes�headerZheader_plus_frameZ
bytes_sentr r r �testBCM9 s6 �
zCANTest.testBCMN)r� )rE rF rG rL � classmethodr� r� r� r� r� r� r� r� r� ro r r
r� r� r r r r r� � s&
�
�r� z!CAN ISOTP required for this test.c sV e Zd Z� fdd�Zdd� Zdd� Ze�ee d�d�d d
� �Z
d d � Z d
d� Z � Z
S )� ISOTPTestc s t � j|i |�� d| _d S �NrZ �rK rL ra rP rS r r rL \ s zISOTPTest.__init__c C s t j t j t j t j d S r2 )r
rs r r! r r? r r r r5 ` s z ISOTPTest.testCrucialConstantsc C s8 t � t jt jt j��}W d � n1 s*0 Y d S r2 rx r r r r ry f 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 r2 )r
r r r! r r r r �testCreateISOTPSocketj 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 )Nr} )r~ ro r� )r
r r r! r' r r` r r r r r p 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 )Nr� iV r_ ) r
r r r! ra r` r� r� r rf �ENODEVrb �r@ r r� r� r r r �testBindv s
2 �zISOTPTest.testBind)rE rF rG rL r5 ry r� ro r r
r� r r� rX r r rS r r� Y s �
r� z!CAN J1939 required for this test.c sX e Zd Z� fdd�Ze�eed�d�dd� �Ze�eed�d�dd� �Z d d
� Z
� Z S ) � J1939Testc s t � j|i |�� d| _d S r� r� rP rS r r rL � 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 r2 )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 r2 )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# ra r� r� r� r` r� r� r rf r� rb r� r r r r� � s
2 �zJ1939Test.testBind) rE rF rG rL r� ro r r
r� r� r� rX r r rS 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 r2 )r
ZAF_RDSr% r? r r r r5 � 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 ry � 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� r3 � SO_RCVBUF� SO_SNDBUF)r@ rc r r r r �testSocketBufferSize� s z!BasicRDSTest.testSocketBufferSizeN)rE rF rG r5 ry 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� rL r� r r r rL � s zRDSTest.__init__c s t � �� t�� | _d S r2 )rK rA rM rx Zevtr? rS r r rA � s
z
RDSTest.setUpc C s2 | j �| j�\}}| �| j|� | �| j|� d S r2 )r; r� rc r� r� r� �r@ 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 r2 )r; r� rc 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 r2 )r; r� rc r� r� )r@ 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 r2 )r; r� rc 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 r� rc r� r� )r@ 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� )rE rF rG rL rA r� r� r� r� r r
r� r� r� r� r� r� rX r r rS 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 r2 )r
r* r? r r r r5 s z%BasicQIPCRTRTest.testCrucialConstantsc C s4 t � t jt j��}W d � n1 s&0 Y d S r2 )r
r* r r r r r ry s z!BasicQIPCRTRTest.testCreateSocketc C sH t � t jt j��$}| �|�� d d� W d � n1 s:0 Y d S �Nro 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� ro )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 )Nr� 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� ro r )r
r* r r� r� r� r r r r �testAutoBindSock+ s
z!BasicQIPCRTRTest.testAutoBindSockN) rE rF rG r5 ry 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 r2 )r
r� r? r r r r5 5 s z#BasicVSOCKTest.testCrucialConstantsc C s4 t j t j t j t j t j t j t j t j d S r2 ) r
�SO_VM_SOCKETS_BUFFER_SIZE� SO_VM_SOCKETS_BUFFER_MIN_SIZE� SO_VM_SOCKETS_BUFFER_MAX_SIZEr� ZVMADDR_PORT_ANYZVMADDR_CID_HOSTZVM_SOCKETS_INVALID_VERSIONr r? r r r �testVSOCKConstants8 s z!BasicVSOCKTest.testVSOCKConstantsc C s4 t � t jt j��}W d � n1 s&0 Y d S r2 )r
r� r/ r r r r ry B 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� r/ r� r� r� r� r� r� )r@ r Zorig_maxr~ Zorig_minr r r r� F sD ���
�
�
�
��
��
��z#BasicVSOCKTest.testSocketBufferSizeN)rE rF rG r5 r� ry r� r r r r r� 0 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
�
BDADDR_ANY� BDADDR_LOCALr. r0 r| r� � BTPROTO_HCI�SOL_HCI�
BTPROTO_L2CAPr� � BTPROTO_SCOr? r r r �testBluetoothConstantse s
z)BasicBluetoothTest.testBluetoothConstantsc C s8 t � t jt jt j��}W d � n1 s*0 Y d S r2 )r
r. r/ r0 r r r r �testCreateRfcommSockets 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 r2 )r
r. r& r� r r r r �testCreateL2capSocketw 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 r2 )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 r2 )r
r. r& r� r r r r �testCreateScoSocket� s z&BasicBluetoothTest.testCreateScoSocketN)
rE rF rG r� r� r� rq r| r� r� r� r� r� r r r r r� a 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� rL r� r r r rL � 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 r2 �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� )r@ �seg1�seg2r� r r r �testOverFlowRecv� s z BasicTCPTest.testOverFlowRecvc C s | j �t� d S r2 r� r? r r r �_testOverFlowRecv� s z BasicTCPTest._testOverFlowRecvc C s | j �d�\}}| �|t� d S r� )r� r� r� r� �r@ r� r� r r r � testRecvFrom� s zBasicTCPTest.testRecvFromc C s | j �t� d S r2 r� r? r r r �
_testRecvFrom� s zBasicTCPTest._testRecvFromc C s@ | j �tt�d �\}}| j �d�\}}|| }| �|t� d S r� )r� r� r� r� r� )r@ r� r� r� r� r r r �testOverFlowRecvFrom� s z!BasicTCPTest.testOverFlowRecvFromc C s | j �t� d S r2 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� )r@ r� �readr r r � testSendAll� s
zBasicTCPTest.testSendAllc C s d}| j �|� d S )Nr� )r� r )r@ Z big_chunkr r r � _testSendAll� s zBasicTCPTest._testSendAllc C sP | j �� }t�|tjtj�}| �|j� | �|tj� |� d�}| �
|t � d S r� ) r� r r
�fromfdr: r/ rU r r r� r� r� )r@ rR r� r� r r r �
testFromFd� s
zBasicTCPTest.testFromFdc C s | j �t� d S r2 r� r? r r r � _testFromFd� s zBasicTCPTest._testFromFdc C s0 | j �� }| �|j� |�d�}| �|t� d S r� )r� �duprU r r� r� r� )r@ r� r� r r r �testDup� s
zBasicTCPTest.testDupc C s | j �t� d S r2 r� r? r r r �_testDup� s zBasicTCPTest._testDupc C s& | j �d�}| �|t� | j�� d S r� )r� r� r� r� rz 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 ro r� ) r� r� r� r� r$ ru r r� �UINT_MAX�r@ r� r r r �_testShutdown_overflow� s � �z#BasicTCPTest._testShutdown_overflowc C s� | j �� }| j �� }| �||� | �| j j� | �t| j jd� | j � � t
j
t
j t
j |d�}| �
|j � |�d�}| �|t� d S )Nr� r� )r� r rL r� r �_closedr$ r r� r r
r: r/ rU r� )r@ r r r� r� r r r �
testDetach� s
zBasicTCPTest.testDetachc C s | j �t� d S r2 r� r? r r r � _testDetach� s zBasicTCPTest._testDetachN)r� )rE rF rG rL r� r� r� r� r� r� r� r� r� r� r� r� r r r r r rp 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� rL r� r r r rL
s zBasicUDPTest.__init__c C s | j �tt��}| �|t� d S r2 �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 r2 �r; r� 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 �Nrp �r$ r� r; r� 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� �
rE rF rG rL 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� rL r� r r r rL $
s zBasicUDPLITETest.__init__c C s | j �tt��}| �|t� d S r2 r r� r r r r
'
s z"BasicUDPLITETest.testSendtoAndRecvc C s | j �tdt| jf� d S r� r r? r r r r ,
s z#BasicUDPLITETest._testSendtoAndRecvc C s$ | j �tt��\}}| �|t� d S r2 r r� r r r r� /
s z BasicUDPLITETest.testRecvFromc C s | j �tdt| jf� d S r� r r? r r r r� 4
s z BasicUDPLITETest._testRecvFromc C s | � t| jjd� d S r r r? r r r r 7
s z%BasicUDPLITETest.testRecvFromNegativec C s | j �tdt| jf� d S r� r r? r r r r ;
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 r2 )rM rx �
misc_eventrK rA r? rS r r rA `
s
zSendrecvmsgBase.setUpc C s | j �|�S r2 )�cli_sockr� r� r r r � sendToServerd
s z SendrecvmsgBase.sendToServerr c G s | j j|| jt|�d � � S r2 )r r� �sendmsg_to_server_defaultsr� �r@ rQ r r r �sendmsgToServerl
s �z SendrecvmsgBase.sendmsgToServerc G s |j |g|�R � }| �|� |S r2 )r� �registerRecvmsgResult)r@ r� rc rQ �resultr r r � doRecvmsgt
s
zSendrecvmsgBase.doRecvmsgc C s d S r2 r �r@ r r r r r ~
s z%SendrecvmsgBase.registerRecvmsgResultc C s | � ||� d S r2 �r� �r@ Zaddr1Zaddr2r r r �checkRecvmsgAddress�
s z#SendrecvmsgBase.checkRecvmsgAddressr )�
MSG_CTRUNC�MSG_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� )
r@ r� �eor�checkset�
checkunset�ignoreZ
defaultsetZ defaultunsetZinbothr� r r r �
checkFlags�
s&
�zSendrecvmsgBase.checkFlags)Nr r r )rE rF rG r �LOOPBACK_TIMEOUT� fail_timeoutrA r r r r r r$ r( r� r r
r' r) r* r0 rX r r rS r r Y
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�assertLessEqualr� )r@ r� rc rQ �bufr r r r r �
s
zRecvmsgIntoMixin.doRecvmsgN)rE rF rG r r r r r r3 �
s r3 c s e Zd Ze� fdd��Z� ZS )�SendrecvmsgDgramFlagsBasec s t � jtjB S r2 )rK r* r
� MSG_TRUNCr? rS r r r* �
s z5SendrecvmsgDgramFlagsBase.msg_flags_non_eor_indicator)rE rF rG �propertyr* rX r r rS r r8 �
s r8 c s e Zd Ze� fdd��Z� ZS )�SendrecvmsgSCTPFlagsBasec s t � jtjB S r2 )rK r) r
�MSG_EORr? rS r r r) �
s z0SendrecvmsgSCTPFlagsBase.msg_flags_eor_indicator)rE rF rG r: r) rX r r rS r r; �
s r; c @ s8 e Zd Zedd� �Zedd� �Zedd� �Zdd� Zd S )
� SendrecvmsgConnectionlessBasec C s | j S r2 )r; r? r r r � serv_sock�
s z'SendrecvmsgConnectionlessBase.serv_sockc C s | j S r2 )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 r2 )r r, r� r� r r r r �
s z*SendrecvmsgConnectionlessBase.sendToServerN)rE rF rG 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 r2 )r� r? r r r r> s z"SendrecvmsgConnectedBase.serv_sockc C s | j S r2 )r� r? r r r r s z!SendrecvmsgConnectedBase.cli_sockc C s d S r2 r r# r r r r$ s z,SendrecvmsgConnectedBase.checkRecvmsgAddressN)rE rF rG 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 r2 )rK rA r> r� r2 r? rS r r rA s
z"SendrecvmsgServerTimeoutBase.setUp)rE rF rG rA rX r r rS 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 r2 �r� r> r� r� r� r? r r r � testSendmsg! s zSendmsgTests.testSendmsgc C s | � | �tg�tt�� d S r2 �r� r r� r� r? r r r � _testSendmsg% s zSendmsgTests._testSendmsgc C s | � | j�tt��t� d S r2 rB r? r r r �testSendmsgDataGenerator( s z%SendmsgTests.testSendmsgDataGeneratorc C s& | � | �dd� tfD ��tt�� d S )Nc s s | ]
}|V qd S r2 r r r r r r� - r� z9SendmsgTests._testSendmsgDataGenerator.<locals>.<genexpr>rD r? r r r �_testSendmsgDataGenerator, s �z&SendmsgTests._testSendmsgDataGeneratorc C s | � | j�tt��t� d S r2 rB r? r r r � testSendmsgAncillaryGenerator0 s z*SendmsgTests.testSendmsgAncillaryGeneratorc C s( | � | �tgdd� dD ��tt�� d S )Nc s s | ]
}|V qd S r2 r r r r r r� 5 r� z>SendmsgTests._testSendmsgAncillaryGenerator.<locals>.<genexpr>r rD r? r r r � _testSendmsgAncillaryGenerator4 s �z+SendmsgTests._testSendmsgAncillaryGeneratorc C s | � | j�tt��t� d S r2 rB r? r r r �testSendmsgArray8 s z SendmsgTests.testSendmsgArrayc C s$ | � | �t�dt�g�tt�� d S )N�B)r� r �arrayr� r� r? r r r �_testSendmsgArray< s �z SendmsgTests._testSendmsgArrayc C s | � | j�tt��t� d S r2 rB r? r r r �testSendmsgGather@ s z SendmsgTests.testSendmsgGatherc C s. | � | �td d� tdd � g�tt�� d S �Nr� rD r? r r r �_testSendmsgGatherD s z SendmsgTests._testSendmsgGatherc C s | � | j�d�d� d S �Nrq � done�r� r> r� r? r r r �testSendmsgBadArgsG 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 rR )r$ r� r r� r �objectr� r r? r r r �_testSendmsgBadArgsK s.
�
�
�
�
�
�
�z SendmsgTests._testSendmsgBadArgsc C s | � | j�d�d� d S rQ rS r? r r r �testSendmsgBadCmsg] 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 rX r> rR �r$ r� r r� rU r r? r r r �_testSendmsgBadCmsga s&
�
�
�
�
�
�z SendmsgTests._testSendmsgBadCmsg�
CMSG_SPACEc C s | � | j�d�d� d S rQ rS r? r r r �testSendmsgBadMultiCmsgp s z$SendmsgTests.testSendmsgBadMultiCmsgc C s@ | � t| jtgg d�� | � t| jtgdt� g� | �d� d S )N�r r r� rR rY r? r r r �_testSendmsgBadMultiCmsgv s
�
�z%SendmsgTests._testSendmsgBadMultiCmsgc C s | � | j�d�d� d S rQ rS 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] rR )
r r
r$ r r r� � assertIsNoner- rf r �r@ r/ r r r � _testSendmsgExcessCmsgReject� s
0z)SendmsgTests._testSendmsgExcessCmsgRejectc C s d S r2 r r? r r r �testSendmsgAfterClose� s z"SendmsgTests.testSendmsgAfterClosec C s | j �� | �t| jtg� d S r2 )r r r$ r r r� r? r r r �_testSendmsgAfterClose� s
z#SendmsgTests._testSendmsgAfterCloseN)rE rF rG rC rE rF rG rH rI rJ rM rN rP rT rV rW rZ r r
r\ r� r^ r_ rb rc rd r r r r rA s,
rA 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 r2 rB r? r r r �testSendmsgExplicitNoneAddr� s z.SendmsgStreamTests.testSendmsgExplicitNoneAddrc C s" | � | �tgg dd �tt�� d S r� rD r? r r r � _testSendmsgExplicitNoneAddr� s z/SendmsgStreamTests._testSendmsgExplicitNoneAddrc C s. | � | j�d�d� | �| jj| jd�� d S �Nr� � aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa�r5 �r� r> r� r r r� r2 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��?ri �socket.timeout not raised) r r� r r
r5 r rf �ENOMEMr r rm �r@ 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 rh rk 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 �Nri )
r$ r r r
� MSG_DONTWAITr r- rf �EAGAIN� EWOULDBLOCKro r rm ra r r r �_testSendmsgDontWait� s 2
�z'SendmsgStreamTests._testSendmsgDontWaitN)
rE rF rG rf rg rl rq r� r| r� rr r� rw r r r r re � s
�
re c @ s e Zd Zdd� Zdd� ZdS )�SendmsgConnectionlessTestsc C s d S r2 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)rE rF rG ry rz r r r r rx � s rx 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� r0 �r@ r� r� r� r� r r r � testRecvmsg� s
z RecvmsgGenericTests.testRecvmsgc C s | � t� d S r2 �r r� r? r r r � _testRecvmsg� s z RecvmsgGenericTests._testRecvmsgc C sV | � | jtt�dd�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr Tr} r~ r r r r �testRecvmsgExplicitDefaults� s
� z/RecvmsgGenericTests.testRecvmsgExplicitDefaultsc C s | � t� d S r2 r� r? r r r � _testRecvmsgExplicitDefaults� s z0RecvmsgGenericTests._testRecvmsgExplicitDefaultsc C sV | � | jtt�d �\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr> Tr} r~ r r r r �testRecvmsgShorter� s
� z&RecvmsgGenericTests.testRecvmsgShorterc C s | � t� d S r2 r� r? r r r �_testRecvmsgShorter s z'RecvmsgGenericTests._testRecvmsgShorterc C s^ | � | jtt�d �\}}}}| �|td d� � | �|| j� | �|g � | j|dd� d S )Nr� ���Fr} r~ r r r r �testRecvmsgTrunc s
� z$RecvmsgGenericTests.testRecvmsgTruncc C s | � t� d S r2 r� r? r r r �_testRecvmsgTrunc s z%RecvmsgGenericTests._testRecvmsgTruncc C sT | � | jtt�d�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nro Tr} r~ r r r r � testRecvmsgShortAncillaryBuf s � z0RecvmsgGenericTests.testRecvmsgShortAncillaryBufc C s | � t� d S r2 r� r? r r r � _testRecvmsgShortAncillaryBuf s z1RecvmsgGenericTests._testRecvmsgShortAncillaryBufc C sT | � | jtt�d�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )N� ( Tr} r~ r r r r �testRecvmsgLongAncillaryBuf! s � z/RecvmsgGenericTests.testRecvmsgLongAncillaryBufc C s | � t� d S r2 r� r? r r r � _testRecvmsgLongAncillaryBuf* s z0RecvmsgGenericTests._testRecvmsgLongAncillaryBufc C s" | j �� | �t| j| j d� d S r� )r> r r$ r r r? r r r �testRecvmsgAfterClose- s
z)RecvmsgGenericTests.testRecvmsgAfterClosec C s d S r2 r r? r r r �_testRecvmsgAfterClose2 s z*RecvmsgGenericTests._testRecvmsgAfterClosec C sF z4| j �d� | �tj| j| j tt�� W | j� � n | j� � 0 d S )Nrm )
r> r� r$ r
r5 r r� r� r rm r? r r r �testRecvmsgTimeout5 s �z&RecvmsgGenericTests.testRecvmsgTimeoutc C s | � | jj| jd�� d S �Nrj )r r r� r2 r? r r r �_testRecvmsgTimeout> 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 r� Fr9 �r, r/ Tr} ) r r> r� r� r
r� r� r$ r� r0 r r r r r �testRecvmsgPeekA s. �
�� z#RecvmsgGenericTests.testRecvmsgPeekc C s | � t� d S r2 r� r? r r r �_testRecvmsgPeekd s z$RecvmsgGenericTests._testRecvmsgPeekr� c C s` | j �| j� | �| j tt��\}}}}| �|t� | �|| j� | �|g � | j |dd� d S r| )
r> r� r2 r r� r� r� r$ r� r0 r r r r �testRecvmsgFromSendmsgh s z*RecvmsgGenericTests.testRecvmsgFromSendmsgc C s. | � | �td d� tdd � g�tt�� d S rO rD r? r r r �_testRecvmsgFromSendmsgs s z+RecvmsgGenericTests._testRecvmsgFromSendmsgN)rE rF rG 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{ � s0
"
r{ c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�RecvmsgGenericStreamTestsc C sN | � | jd�\}}}}| �|d� | �|| j� | �|g � | j|d d� d S )Nr� r� r} )r r> r� r$ r� r0 r r r r �testRecvmsgEOF| s
z(RecvmsgGenericStreamTests.testRecvmsgEOFc C s | j �� d S r2 )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� Fr} r� T)r r> r� r� r$ r� r� r0 )r@ r� r� r� r� r� r� r r r �testRecvmsgOverflow� s
� z-RecvmsgGenericStreamTests.testRecvmsgOverflowc C s | � t� d S r2 r� r? r r r �_testRecvmsgOverflow� s z.RecvmsgGenericStreamTests._testRecvmsgOverflowN)rE rF rG r� r� r� r� r r r r r� x 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 )Nrp r r� Tr} )
r$ r� r> r� r� r� r� r� rU r� r$ r� r0 r r r r �testRecvmsgBadArgs� s0 �
� � � � � z RecvmsgTests.testRecvmsgBadArgsc C s | � t� d S r2 r� r? r r r �_testRecvmsgBadArgs� s z RecvmsgTests._testRecvmsgBadArgsN)rE rF rG r� r� r r r r r� � s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�RecvmsgIntoTestsc C sL t tt��}| �t| jj� | �t| jjtt�dd� | �t| jj|dd� | �t| jjt� gdd� | �t| jjdgdd� | �t| jj|t� gdd� | �t| jj|gdd� | �t| jj|gt� d� | �t| jj|gdt� � | j�|gdd�\}}}}| � |tt�� | � |t t�� | �
|| j � | � |g � | j |dd� d S )Nr s I'm not writablerp Tr} )
r� r� r� r$ r� r> r4 rU r� r� r$ r� r0 �r@ r7 �nbytesr� r� r� r r r �testRecvmsgIntoBadArgs� s@
� �
� � � �
�
� z'RecvmsgIntoTests.testRecvmsgIntoBadArgsc C s | � t� d S r2 r� 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 r2 r r r r r r� � r� z<RecvmsgIntoTests.testRecvmsgIntoGenerator.<locals>.<genexpr>Tr} ) r� r� r� r> r4 r� r$ r� r0 r� r r r �testRecvmsgIntoGenerator� s � z)RecvmsgIntoTests.testRecvmsgIntoGeneratorc C s | � t� d S r2 r� r? r r r �_testRecvmsgIntoGenerator� s z*RecvmsgIntoTests._testRecvmsgIntoGeneratorc C sx t � ddgtt� �}| j�|g�\}}}}| �|tt�� | �|�� t� | �|| j� | �|g � | j |dd� d S )NrK r Tr} )
rL r� r� r> r4 r� �tobytesr$ r� r0 r� r r r �testRecvmsgIntoArray� s z%RecvmsgIntoTests.testRecvmsgIntoArrayc C s | � t� d S r2 r� 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---Tr} ) r� r> r4 �
memoryviewr� r� r$ r� r0 )r@ �b1�b2Zb3r� r� r� r� r r r �testRecvmsgIntoScatter� s � z'RecvmsgIntoTests.testRecvmsgIntoScatterc C s | � d� d S )Nr� )r r? r r r �_testRecvmsgIntoScatter
s z(RecvmsgIntoTests._testRecvmsgIntoScatterN) rE rF rG r� r� r� r� r� r� r� r� r r r r r� � s
r� 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 ro � r8 r� rp )�socklen_t_limitr
r� �listr) rj rL �itemsizer� r6 r$ ru r| �maxsize)r@ �toobigrP �nr, r r r � testCMSG_LEN
s
zCmsgMacroTests.testCMSG_LENr[ c C s� | j t�d� d }ttd��tt|d |�� }t�d�}| �|t�d�jd � |D ]T}t�|�}| �||� | �|t� |�� | �||t� d� � | �
|| j � |}qX| � t tjd� | � t tj|� | � t tjt
j� d S )Nro r� r r8 r� rp )r� r
r[ r� r) rj rL r� r5 r� r6 r$ ru r| r� )r@ r� rP �lastr� r, r r r �testCMSG_SPACE*
s
z CmsgMacroTests.testCMSG_SPACE) rE rF rG r� r� r� �minr� r r
r� r� r r r r r�
s
r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zdddd�Z dd� Z
dd� Z dd� Z dd� Z
eed�dd� �Zejdd � �Zd d � Zd d � Ze�ejd!kd"�e�ed#�eed�d$d%� ���Zeje�ejd!kd"�e�ed#�d&d'� ���Ze�ejd!kd"�e�ed#�eed�d(d)� ���Z e je�ejd!kd"�e�ed#�d*d+� ���Z d,d-� Z e�ejd!kd.�d/d0� �Z d1d2� Z d3d4� Z!d5d6� Z"eed�d7d8� �Z#e#jd9d:� �Z$ded;d<�Z%d=d>� Z&d?d@� Z'dAdB� Z(dCdD� Z)dEdF� Z*dGdH� Z+dIdJ� Z,dKdL� Z-dMdN� Z.dOdP� Z/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 r2 ) r) r� �mkstemprU r� r� r �writer �encode�append)r@ r� �fdsr8 rR r� r r r �newFDsK
s zSCMRightsTest.newFDsc C sB t |�D ]4\}}t�|dtj� | �t�|d�t|��� � qd S )Nr r� )� enumerater� �lseek�SEEK_SETr� r� r r� )r@ r� r� rR r r r �checkFDsW
s zSCMRightsTest.checkFDsc C s | � | j|� d S r2 )rU �closeRecvmsgFDsr! r r r r ^
s z#SCMRightsTest.registerRecvmsgResultc C sn |d D ]`\}}}|t jkr|t jkrt�d�}|�|d t|�t|�|j � � |D ]}t�|� qXqd S )Nro r8 ) r
r3 �
SCM_RIGHTSrL � frombytesr� r� r� r )r@ Zrecvmsg_result�
cmsg_level� cmsg_type� cmsg_datar� rR r r r r� a
s
�
�z SCMRightsTest.closeRecvmsgFDsc
C s8 | � | �tgtjtjt�d| �|��fg�tt�� d S )Nr8 ) r� r r� r
r3 r� rL r� r� )r@ r� r r r �createAndSendFDsm
s ���z SCMRightsTest.createAndSendFDsro 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/ r8 r )r� r� r$ r� r0 r
r% r r� r6 r� rL �tupler3 r� r� �
SIZEOF_INTr� r� )r@ Znumfdsr �maxcmsgs� ignoreflagsr� r� r� r� r� �itemr� r� r� r r r �checkRecvmsgFDsw
s& �
z SCMRightsTest.checkRecvmsgFDsc C s | � d| �| jtt�d�� d S )Nro r� �r� r r> r� r� r? r r r �testFDPassSimple�
s �z SCMRightsTest.testFDPassSimplec
C s< | � | �tgtjtjt�d| �d���� fg�t t�� d S )Nr8 ro )
r� r r� r
r3 r� rL r� r� r� r? r r r �_testFDPassSimple�
s ���z SCMRightsTest._testFDPassSimplec C s | � d| �| jtt�d�� d S )Nr+ r� 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 )Nro r+ �r� ) r� r r> r� r� r
r� 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� r� )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� r8 � r� r� r r� r
r3 r� rL r� �r@ 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� r� r% )r@ 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 r2 )r r r rf ri r� r� )r@ r� r� r� r� r r r �sendAncillaryIfPossible�
s "z%SCMRightsTest.sendAncillaryIfPossiblezsee issue #24725c C s& | j d| �| jtt�d�tjd� d S )Nr r� r� )r� r r> r� r� r
r% r? r r r �testFDPassEmpty�
s
��z SCMRightsTest.testFDPassEmptyc C s | � ttjtjdfg� d S �Nr� )r� r� r
r3 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 )Nr� Tr� ro )r r> r� r� r� r$ r� r0 r
r% r6 r3 r� �
assertLessr� �r@ 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 )Nr8 rp )r� r� r
r3 r� rL �badfdr� 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 )Nr� Tr� r� r8 )r r> r� r� r� r$ r� r0 r
r% r6 rL r3 r� r� r� r� ) r@ 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� r8 rp ) r� r� r� r
r3 r� rL r� r� r� r r r � _testFDPassPartialIntInMiddle2 s � ���z+SCMRightsTest._testFDPassPartialIntInMiddlec C sJ |\}}}}| � |t� | �|| j� | � |g � | j|dtj|d� d S )NT�r, r- r/ )r� r� r$ r� r0 r
r% )r@ r r� r� r� r� r� r r r �checkTruncatedHeader> s �z"SCMRightsTest.checkTruncatedHeaderc C s" | j | �| jtt��tjd� d S �Nr� �r� r r> r� r� r
r% r? r r r �testCmsgTruncNoBufSizeH s �z$SCMRightsTest.testCmsgTruncNoBufSizec C s | � d� d S r� r� r? r r r �_testCmsgTruncNoBufSizeQ s z%SCMRightsTest._testCmsgTruncNoBufSizec C s$ | j | �| jtt�d�tjd� d S )Nr r� r� r? r r r �testCmsgTrunc0T s �z SCMRightsTest.testCmsgTrunc0c C s | � d� d S r� r� r? r r r �_testCmsgTrunc0Y s z SCMRightsTest._testCmsgTrunc0c C s | � | �| jtt�d�� d S r� )r� r r> r� r� r? r r r �testCmsgTrunc1_ s z SCMRightsTest.testCmsgTrunc1c C s | � d� d S r� r� r? r r r �_testCmsgTrunc1b s z SCMRightsTest._testCmsgTrunc1c C s" | � | �| jtt�td �� d S r� )r� r r> r� r� r� r? r r r �testCmsgTrunc2Inte s �z SCMRightsTest.testCmsgTrunc2Intc C s | � d� d S r� r� r? r r r �_testCmsgTrunc2Intl s z SCMRightsTest._testCmsgTrunc2Intc C s( | � | �| jtt�t�d�d �� d S r )r� r r> r� r� r
r� r? r r r �testCmsgTruncLen0Minus1o s �z%SCMRightsTest.testCmsgTruncLen0Minus1c C s | � d� d S r� r� r? r r r �_testCmsgTruncLen0Minus1s 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 ro r8 )r r> r� r� r� r$ r� r0 r
r% r3 r� r5 r6 rL r� r� r� ) r@ �ancbuf�maxdataZmindatar� r� r� r� r� r� r� r� r r r �checkTruncatedArrayy s& �
�z!SCMRightsTest.checkTruncatedArrayc C s | j t�d�dd� d S )Nr �r r �r r
r� 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 ro 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
r� 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� ro 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)ro r )r )r )9rE rF rG r� r� r� r r� r� r� r� r� r� r� r r
r� r� r� r� r� r� rq 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� C
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� rp c C s4 i }|D ]&\}}}| � ||f|� ||||f< q|S r2 )r )r@ r� �dr� r� r� r r r �ancillaryMapping� s
z%RFC3542AncillaryTest.ancillaryMappingr� r c C s | j �tjtjd� | j�� | �| j tt �|�\}}}}| �
|t � | � || j � | j
|dtj|d� | �
t|�d� | �|d t� |d \}} }
| �
|tj� | �
| tj� | �|
t� | �
t|
�t� t�d�} | �|
� | �| d d� | �| d |� d S )Nro Tr� r r8 )r> r� r
r8 rI r rm r r� r� r� r$ r� r0 r% r r� rO r� r� rL r� r5 r6 ) r@ �
ancbufsize�maxhopr� r� r� r� r� r� r� r� r� r r r �
checkHopLimit� s.
�
� �
z"RFC3542AncillaryTest.checkHopLimitrI rO c C s | j dd� d S �Nr� �r )r r? r r r �testRecvHopLimit� s z%RFC3542AncillaryTest.testRecvHopLimitc C s$ | � | jj| jd�� | �t� d S r� �r r r� r2 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 r� r r? r r r �_testRecvHopLimitCMSG_SPACE� s z0RFC3542AncillaryTest._testRecvHopLimitCMSG_SPACEr� c C s | j d| jd� d S �Nr� )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 �Nrj r8 )
r r r� r2 r� r r� r
r8 rO rL 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 )Nro Tr� r� r8 r r� )r> r� r
r8 rI rK r rm r r� r� r� r$ r� r0 r% r rF r� rL r� r5 r6 rO ) r@ r r r� r� r� r� r� ZancmapZtcdatar� Zhldatar r r � checkTrafficClassAndHopLimit s:
�
�
� �
z1RFC3542AncillaryTest.checkTrafficClassAndHopLimitrK rF c C s | j dd� d S r )r& r? r r r � testRecvTrafficClassAndHopLimit3 s z4RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitc C s$ | � | jj| jd�� | �t� d S r� r r? r r r � _testRecvTrafficClassAndHopLimit9 s z5RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitc C s | j t�t�d d� d S �Nr� r )r& r
r[ r� r? r r r �)testRecvTrafficClassAndHopLimitCMSG_SPACE> s �z>RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitCMSG_SPACEc C s$ | � | jj| jd�� | �t� d S r� r r? r r r �*_testRecvTrafficClassAndHopLimitCMSG_SPACEF s z?RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitCMSG_SPACEc C s | j d| jd� d S r! �r& r" r? r r r � testSetTrafficClassAndHopLimitK 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� r2 r� r r� r
r8 rF rL �
traffic_classrO r" r� r? r r r � _testSetTrafficClassAndHopLimitT s �����z4RFC3542AncillaryTest._testSetTrafficClassAndHopLimitc C s | j d| jd� d S r! r, r? r r r �testOddCmsgSize_ 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 )Nrj r8 r� )r r r� r2 r r� r
r8 rF rL r. r� rO r" r r rf ri r� r� )r@ r� r� r r r �_testOddCmsgSizei 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 )Nro r Tr� )r> r� r
r8 rI r rm r r� r� r� r$ r� r0 r% )r@ r r� rQ r� r� r� r� r r r � checkHopLimitTruncatedHeader s
�
�� �z1RFC3542AncillaryTest.checkHopLimitTruncatedHeaderc C s | j d tjd� d S )N�r r� �r2 r
r% r? r r r r� � s �z+RFC3542AncillaryTest.testCmsgTruncNoBufSizec C s$ | � | jj| jd�� | �t� d S r� r r? r r r r� � s z,RFC3542AncillaryTest._testCmsgTruncNoBufSizec C s | j dtjd� d S )Nr r3 r4 r? r r r �testSingleCmsgTrunc0� s �z)RFC3542AncillaryTest.testSingleCmsgTrunc0c C s$ | � | jj| jd�� | �t� d S r� r r? r r r �_testSingleCmsgTrunc0� s z*RFC3542AncillaryTest._testSingleCmsgTrunc0c C s | j dd� d S )Nro r )r2 r? r r r �testSingleCmsgTrunc1� s z)RFC3542AncillaryTest.testSingleCmsgTrunc1c C s$ | � | jj| jd�� | �t� d S r� r r? r r r �_testSingleCmsgTrunc1� s z*RFC3542AncillaryTest._testSingleCmsgTrunc1c C s | j dt d� d S r) )r2 r� r? r r r �testSingleCmsgTrunc2Int� s z,RFC3542AncillaryTest.testSingleCmsgTrunc2Intc C s$ | � | jj| jd�� | �t� d S r� r r? r r r �_testSingleCmsgTrunc2Int� s z-RFC3542AncillaryTest._testSingleCmsgTrunc2Intc C s | j t�d�d d� d S )Nr ro r )r2 r
r� r? r r r � testSingleCmsgTruncLen0Minus1� s z2RFC3542AncillaryTest.testSingleCmsgTruncLen0Minus1c C s$ | � | jj| jd�� | �t� d S r� 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 )Nro Tr r )r> r� r
r8 rI r rm r r� r� r� r� r� r$ r� r0 r% r6 rO r� r� r r r �testSingleCmsgTruncInData� s
�
� z.RFC3542AncillaryTest.testSingleCmsgTruncInDatac C s$ | � | jj| jd�� | �t� d S r� 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 )Nro Tr� r r8 r� )r> r� r
r8 rI rK r rm r r� r� r� r$ r� r0 r% r rF rO r� rL r� r5 r6 ) r@ 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 r� 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 r� r r? r r r �_testSecondCmsgTrunc1 s z*RFC3542AncillaryTest._testSecondCmsgTrunc1c C s | � t�t�dt � d S r� rB r? r r r �testSecondCmsgTrunc2Int s �z,RFC3542AncillaryTest.testSecondCmsgTrunc2Intc C s$ | � | jj| jd�� | �t� d S r� r r? r r r �_testSecondCmsgTrunc2Int# s z-RFC3542AncillaryTest._testSecondCmsgTrunc2Intc C s" | � t�t�t�d� d � d S r )r? r
r[ r� r� r? r r r � testSecondCmsgTruncLen0Minus1( s
��z2RFC3542AncillaryTest.testSecondCmsgTruncLen0Minus1c C s$ | � | jj| jd�� | �t� d S r� r r? r r r � _testSecondCmsgTruncLen0Minus1. 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 )Nro Tr r r8 r� ) r> r� r
r8 rI rK r rm r r� r� r[ r� r� r� r$ r� r0 r% rF rO �pop�removerL r� r5 r6 r� )
r@ r� r� r� r� Z
cmsg_typesr� r� r� r� r r r �testSecomdCmsgTruncInData3 s:
�
�
�
z.RFC3542AncillaryTest.testSecomdCmsgTruncInDatac C s$ | � | jj| jd�� | �t� d S r� r r? r r r �_testSecomdCmsgTruncInDataX s z/RFC3542AncillaryTest._testSecomdCmsgTruncInDataN)r� r )r� r )r )r )1rE rF rG r" r. r r r r
r r� r r r r# r% r& r' r( r* r+ r- r/ r0 r1 r2 r� r� r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA rC rD rE rF rG rH rK rL r r r r r � s�
�
$�
�
�
�
�
�
�
�
�
#r c @ s e Zd ZdS )�SendrecvmsgUDPTestBaseN�rE rF rG r r r r rM ` s rM r� c @ s e Zd ZdS )�SendmsgUDPTestNrN r r r r rO e s rO r� c @ s e Zd ZdS )�RecvmsgUDPTestNrN r r r r rP i s rP r4 c @ s e Zd ZdS )�RecvmsgIntoUDPTestNrN r r r r rQ m s rQ c @ s e Zd Zdd� ZdS )�SendrecvmsgUDP6TestBasec C s | � |d d� |d d� � d S r r" r# r r r r$ v s z+SendrecvmsgUDP6TestBase.checkRecvmsgAddressN�rE rF rG r$ r r r r rR r s rR r4 r� r c @ s e Zd ZdS )�SendmsgUDP6TestNrN r r r r rT { s rT c @ s e Zd ZdS )�RecvmsgUDP6TestNrN r r r r rU � s rU c @ s e Zd ZdS )�RecvmsgIntoUDP6TestNrN r r r r rV � s rV r8 c @ s e Zd ZdS )� RecvmsgRFC3542AncillaryUDP6TestNrN r r r r rW � s rW c @ s e Zd ZdS )�#RecvmsgIntoRFC3542AncillaryUDP6TestNrN r r r r rX � s rX c @ s e Zd ZdS )�SendrecvmsgUDPLITETestBaseNrN r r r r rY � s rY c @ s e Zd ZdS )�SendmsgUDPLITETestNrN r r r r rZ � s rZ c @ s e Zd ZdS )�RecvmsgUDPLITETestNrN r r r r r[ � s r[ c @ s e Zd ZdS )�RecvmsgIntoUDPLITETestNrN 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.checkRecvmsgAddressNrS r r r r r] � s r] c @ s e Zd ZdS )�SendmsgUDPLITE6TestNrN r r r r r^ � s r^ c @ s e Zd ZdS )�RecvmsgUDPLITE6TestNrN r r r r r_ � s r_ c @ s e Zd ZdS )�RecvmsgIntoUDPLITE6TestNrN r r r r r` � s r` c @ s e Zd ZdS )�#RecvmsgRFC3542AncillaryUDPLITE6TestNrN r r r r ra � s ra c @ s e Zd ZdS )�'RecvmsgIntoRFC3542AncillaryUDPLITE6TestNrN r r r r rb � s rb c @ s e Zd ZdS )�SendrecvmsgTCPTestBaseNrN r r r r rc � s rc c @ s e Zd ZdS )�SendmsgTCPTestNrN r r r r rd � s rd c @ s e Zd ZdS )�RecvmsgTCPTestNrN r r r r re � s re c @ s e Zd ZdS )�RecvmsgIntoTCPTestNrN r r r r rf � s rf c @ s e Zd ZdS )� SendrecvmsgSCTPStreamTestBaseNrN r r r r rg s rg z6IPPROTO_SCTP: [Errno 62] Protocol not supported on AIXr: r/ r� c @ s e Zd ZdS )�SendmsgSCTPStreamTestNrN r r r r rh
s rh 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)rK ri r� r rf �ENOTCONNrb �r@ r� rS r r r� s z$RecvmsgSCTPStreamTest.testRecvmsgEOF�rE rF rG r� rX r r rS r ri s ri 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 rj )rK rn r� r rf rk rb rl rS r r r� $ s z(RecvmsgIntoSCTPStreamTest.testRecvmsgEOFrm r r rS r rn s rn c @ s e Zd ZdS )� SendrecvmsgUnixStreamTestBaseNrN r r r r ro - s ro r� c @ s e Zd ZdS )�SendmsgUnixStreamTestNrN r r r r rp 1 s rp c @ s e Zd ZdS )�RecvmsgUnixStreamTestNrN r r r r rq 6 s rq c @ s e Zd ZdS )�RecvmsgIntoUnixStreamTestNrN r r r r rr < s rr r3 r� c @ s e Zd ZdS )�RecvmsgSCMRightsStreamTestNrN r r r r rs B s rs c @ s e Zd ZdS )� RecvmsgIntoSCMRightsStreamTestNrN r r r r rt G s rt 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 )�signumr� r r r r� [ r� z.InterruptedTimeoutBase.setUp.<locals>.<lambda>)rK rA r r rU )r@ Zorig_alrm_handlerrS r r rA X s
�z InterruptedTimeoutBase.setUp� setitimerg�������?c C s t �t j|� d S r2 )r rw � ITIMER_REAL�r@ Zsecondsr r r �setAlarmg s z InterruptedTimeoutBase.setAlarmr� c C s t �|� d S r2 )r r� ry r r r rz m s ) rE rF rG rA r r1 r5 r r �
alarm_timerz rX r r rS r ru S s
ru � siginterruptr� rw 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 r2 )rK rA r; r� r5 r? rS r r rA z s
z InterruptedRecvTimeoutTest.setUpc O sd zR| � | j� | �t�� }||i |�� W d � n1 s<0 Y W | � d� n | � d� 0 d S r� )rz r{ r$ r �r@ r� rQ rR r/ r r r �checkInterruptedRecv~ s
.z/InterruptedRecvTimeoutTest.checkInterruptedRecvc C s | � | jjd� d S r� )r r; r� r? r r r �testInterruptedRecvTimeout� s z5InterruptedRecvTimeoutTest.testInterruptedRecvTimeoutc C s | � | jjtd�� d S r� )r r; � recv_intor� r? r r r � testInterruptedRecvIntoTimeout� s z9InterruptedRecvTimeoutTest.testInterruptedRecvIntoTimeoutc C s | � | jjd� d S r� )r r; r� r? r r r � testInterruptedRecvfromTimeout� s z9InterruptedRecvTimeoutTest.testInterruptedRecvfromTimeoutc C s | � | jjtd�� d S r� )r r; �
recvfrom_intor� r? r r r �"testInterruptedRecvfromIntoTimeout� s z=InterruptedRecvTimeoutTest.testInterruptedRecvfromIntoTimeoutr� c C s | � | jjd� d S r� )r r; r� r? r r r � testInterruptedRecvmsgTimeout� s z8InterruptedRecvTimeoutTest.testInterruptedRecvmsgTimeoutr4 c C s | � | jjtd�g� d S r� )r r; r4 r� r? r r r �!testInterruptedRecvmsgIntoTimeout� s z<InterruptedRecvTimeoutTest.testInterruptedRecvmsgIntoTimeout)rE rF rG rA r r� r� r� r� r r
r� r� rX r r rS r r} s 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)rK rA r� r� rU r rM �Thread� doConnect�startr; r� r� r r� r5 )r@ Z
cli_threadr� rS r r rA � s
z InterruptedSendTimeoutTest.setUpc C s | j �| j� d S r2 )r� r� r� r? r r r r� � 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 rz r{ r~ r r r �checkInterruptedSend� s
0z/InterruptedSendTimeoutTest.checkInterruptedSendr� r� c C s | � | jjd� d S rs )r� r� r� r? r r r �testInterruptedSendTimeout� s z5InterruptedSendTimeoutTest.testInterruptedSendTimeoutc C s | � | jjd| j� d S rs )r� r� r, r� r? r r r � testInterruptedSendtoTimeout� s �z7InterruptedSendTimeoutTest.testInterruptedSendtoTimeoutr� c C s | � | jjdg� d S rs )r� r� r� r? r r r � testInterruptedSendmsgTimeout� s z8InterruptedSendTimeoutTest.testInterruptedSendmsgTimeout)rE rF rG rA r� r� r Zrequires_mac_verr� r� r r
r� rX r r rS r r� � s
r� c @ s e Zd Zdd� Zdd� ZdS )�
TCPCloserTestc C sh | j �� \}}|�� | j}t�|gg g d�\}}}| �||g� | �|�d�d� |�� |�� d S )N� �?ro r� )r; r� r r� r� r� r� )r@ r� r� �sdr� r� r r r r � testClose� s zTCPCloserTest.testClosec C s | j �t| jf� t�d� d S �Nr� )r� r� r� r= r� r� r? r r r �
_testClose� s zTCPCloserTest._testCloseN)rE rF rG r� r� r r r r r� � s
r� 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� rL r� r r r rL � 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� r/ r r� r r r �_check_defaults� s
z#BasicSocketPairTest._check_defaultsc C s | � | j� d S r2 )r� r� r? r r r �
_testDefaults� s z!BasicSocketPairTest._testDefaultsc C s | � | j� d S r2 )r� 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 r2 �r� r� r� r? r r r r� s z BasicSocketPairTest._testRecvc C s | j �t� d S r2 �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� ) rE rF rG rL r� r� r� r� r� r� r� r r r r r� � s
r� c s� e Zd Zdd� Z� fdd�Z� fdd�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z e �
eed� 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 r2 )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_sp�_fallback_socketpairrK rA r? rS r r rA s
z PurePythonSocketPairTest.setUpc s t � �� | jd ur | jt_d S r2 )rK rC r� r
r� r? rS r r rC ) s
z!PurePythonSocketPairTest.tearDownc C s | j �d�}| �|t� d S r� r� r� r r r � test_recv/ s z"PurePythonSocketPairTest.test_recvc C s | j �t� d S r2 r� r? r r r �
_test_recv3 s z#PurePythonSocketPairTest._test_recvc C s | j �t� d S r2 r� r? r r r � test_send6 s z"PurePythonSocketPairTest.test_sendc C s | j �d�}| �|t� d S r� r� r� r r r �
_test_send9 s z#PurePythonSocketPairTest._test_sendc C s$ t �t j�\}}|�� |�� d S r2 )r
r� r: r �r@ r� r1 r r r � test_ipv4= s z"PurePythonSocketPairTest.test_ipv4c C s d S r2 r r? r r r �
_test_ipv4B s z#PurePythonSocketPairTest._test_ipv4r8 � IPV6_V6ONLY� IPV6_V6ONLY option not supportedr> c C s$ t �t j�\}}|�� |�� d S r2 )r
r� r� r r� r r r � test_ipv6E s z"PurePythonSocketPairTest.test_ipv6c C s d S r2 r r? r r r �
_test_ipv6N 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: r/ rI r� �BlockingIOError�InterruptedError)r@ r. �Z inject_sockZorig_getsocknamer r �inject_getsocketnameU s
z[PurePythonSocketPairTest.test_injected_authentication_failure.<locals>.inject_getsocketname)r
r� r$ r r� r )r@ r� �sock1�sock2r r� r �$test_injected_authentication_failureQ s, ,
�z=PurePythonSocketPairTest.test_injected_authentication_failurec C s d S r2 r r? r r r �%_test_injected_authentication_failurer s z>PurePythonSocketPairTest._test_injected_authentication_failure)rE rF rG r� rA rC r� r� r� r� r� r� r� rq r r ro r r� r� r� r� r� rX r r rS r r� s$
� !r� c @ s� e Zd Zd&dd�Zdd� Zdd� Zdd � Zejd
d � �Z e�e�Z
e � e
ed �d
�e�ddd�dd� ��Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%S )'�NonBlockingTCPTestsr� c C s t �� | _tj| |d� d S r� )rM rx �eventr� rL r� r r r rL | 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)r@ r� r5 �blockingZ fd_blocking�flagr r r �assert_sock_timeout� s z'NonBlockingTCPTests.assert_sock_timeoutc C s� | j �d� | �| j d � | j �d� | �| j d� | j �d � | �| j d � | j �d� | �| j d� | j �d� | �| j d� | j �d� | �| j d� d S )NTFr� r r� )r; rI r� r� r? r r r �testSetBlocking� s z#NonBlockingTCPTests.testSetBlockingc C s d S r2 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_MAXFr� ro ) r� r Z ULONG_MAXrb r; rI r� r� r` r r r r �testSetBlocking_overflow� s
z,NonBlockingTCPTests.testSetBlocking_overflowrH z test needs socket.SOCK_NONBLOCKr� � � c C s4 | j �� t�tjtjtjB �| _ | �| j d� d S r� )r; r r
r: r/ rH r� r? r r r �testInitNonBlocking� s
�z'NonBlockingTCPTests.testInitNonBlockingc C s d S r2 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� )r6 r; r� r� rU r r` r� r� r r r �testInheritFlagsBlocking� s
z,NonBlockingTCPTests.testInheritFlagsBlockingc C s | j �t| jf� d S r2 �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� )r6 r; r� r� rU r r� r� )r@ Zdefault_timeoutr� r� r r r �testInheritFlagsTimeout� s
z+NonBlockingTCPTests.testInheritFlagsTimeoutc C s | j �t| jf� d S r2 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 )NFr� z'Error trying to do accept after select.)r; rI r� � monotonicr$ r� r� r� r� rm r� r � LONG_TIMEOUTr rU r r` r� )r@ Z
start_timer� r� Zdtr� r� r r r r �
testAccept� s ,
z NonBlockingTCPTests.testAcceptc C s | j �� | j�t| jf� d S r2 )r� 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� rU r rI r$ r� r� r� r� r� rm r� r r� r r� )r@ r� r� r� r� r� r r r r r� � s
,
z NonBlockingTCPTests.testRecvc C s, | j �t| jf� | j�� | j �t� d S r2 )r� r� r� r= r� r� r r� r? r r r r� s
z NonBlockingTCPTests._testRecvN)r� )rE rF rG rL r� r� r� r rp r� Z_testSetBlocking_overflowr� ro r r
�requires_linux_versionr� r� r� r� r� r� r� r� r� r� r r r r r� z s*
� r� c @ s� e Zd ZdZdZdZdZdZdZe Z
dZ e Z d;d d
�Z
d d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Z d)d*� Z d+d,� Z d-d.� Z d/d0� Z d1d2� Z!d3d4� Z"d5d6� Z#d7d8� Z$d9d:� Z%dS )<�FileObjectClassTestCaseao Unit tests for the object returned by socket.makefile()
self.read_file is the io object returned by makefile() on
the client connection. You can read from this file to
get output from the server.
self.write_file is the io object returned by makefile() on the
server connection. You can write to this file to send output
to the client.
rp r �strictNr r r� c C s t j| |d� d S r� r� r� r r r rL 0 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 )rM rx �r� r8 r r r r� 4 s z1FileObjectClassTestCase.setUp.<locals>.<listcomp>r+ ��encoding�errors�newline)r) �evt1�evt2�
serv_finished� cli_finishedr� rA r� r � read_moderc r� r� r� � read_filer? r r r rA 3 s �
�z FileObjectClassTestCase.setUpc C s6 | j �� | j�� | �| jj� d | _t�| � d S r2 )r� rm r� r r �closedr� rC r? r r r rC = s
z FileObjectClassTestCase.tearDownc C s0 t �| � | jj| j| j| j| j| jd�| _ d S )Nr� )
r� r� r� r �
write_moderc r� r� r� �
write_filer? r r r r� D s
�z#FileObjectClassTestCase.clientSetUpc C s6 | j �� | j�� | �| jj� d | _t�| � d S r2 )r� rm r� r r r� r� r� r? r r r r� L s
z&FileObjectClassTestCase.clientTearDownc C sx | j �d� | j�d� | �tj| jjd� | �t�� }| j�d� W d � n1 sX0 Y | �dt |j
�� d S )Nro r� z!cannot read from timed out object) r� r� r� r� r$ r
r5 r r r r- )r@ �ctxr r r �testReadAfterTimeoutS s *z,FileObjectClassTestCase.testReadAfterTimeoutc C s. | j �| jdd� � | j �� | j�� d S )Nr r� )r� r� � write_msg�flushr� r� r? r r r �_testReadAfterTimeout_ s
z-FileObjectClassTestCase._testReadAfterTimeoutc C s< | j �t| j�d �}| j �d�}|| }| �|| j� d S rO )r� r� r� �read_msgr� )r@ � first_segZ
second_segr� r r r �
testSmallReadd s z%FileObjectClassTestCase.testSmallReadc C s | j �| j� | j �� d S r2 �r� r� r� r� r? r r r �_testSmallReadk s z&FileObjectClassTestCase._testSmallReadc C s | j �� }| �|| j� d S r2 )r� r� r� r� r� r r r � testFullReado s
z$FileObjectClassTestCase.testFullReadc C s | j �| j� | j �� d S r2 )r� r� r� r r? r r r �
_testFullReadt s z%FileObjectClassTestCase._testFullReadc C s: t | j�� }| j�d�}|s q(||7 }q | �|| j� d S r� )r� r� r� r� r� )r@ r7 �charr r r �testUnbufferedReadx s
z*FileObjectClassTestCase.testUnbufferedReadc C s | j �| j� | j �� d S r2 r� r? r r r �_testUnbufferedRead� s z+FileObjectClassTestCase._testUnbufferedReadc C s | j �� }| �|| j� d S r2 )r� �readliner� r� �r@ �liner r r � testReadline� s
z$FileObjectClassTestCase.testReadlinec C s | j �| j� | j �� d S r2 r� r? r r r �
_testReadline� s z%FileObjectClassTestCase._testReadlinec C s& | j �� | j�� }| �|| j� d S r2 )r� r r� r� r� r� r� r r r �testCloseAfterMakefile� s
z.FileObjectClassTestCase.testCloseAfterMakefilec C s | j �| j� | j �� d S r2 r� r? r r r �_testCloseAfterMakefile� s z/FileObjectClassTestCase._testCloseAfterMakefilec C s@ | j �� | j�tt��}t| jt�r.|� � }| �
|| j� d S r2 ) 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 r2 r� r? r r r � _testMakefileAfterMakefileClose� s z7FileObjectClassTestCase._testMakefileAfterMakefileClosec C s | � | jj � d S r2 )r r� r� r? r r r �testClosedAttr� s z&FileObjectClassTestCase.testClosedAttrc C s | � | jj � d S r2 )r r� r� r? r r r �_testClosedAttr� s z'FileObjectClassTestCase._testClosedAttrc C s, | � | jj| j� | � | jj| j�� � d S r2 )r� r� r" r� r� r� r r? r r r �testAttributes� s z&FileObjectClassTestCase.testAttributesc C s, | � | jj| j� | � | jj| j�� � d S r2 )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 r2 )r� r r$ r� r r� r r� r? r r r �
testRealClose� s
z%FileObjectClassTestCase.testRealClosec C s d S r2 r r? r r r �_testRealClose� s z&FileObjectClassTestCase._testRealClose)r� )&rE rF rG rW rc r� r� r� r� r� r� r� r� rL rA rC r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r 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 r2 r� r? r r r �_testMakefileClose� s z4UnbufferedFileObjectClassTestCase._testMakefileClosec C s6 t �| j�}| j�� t �| j�}| �|d |� d S r� )r| r{ r� r� r r� )r@ Zrefcount_beforeZrefcount_afterr r r � testMakefileCloseSocketDestroy� s
z@UnbufferedFileObjectClassTestCase.testMakefileCloseSocketDestroyc C s d S r2 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� r� g �?r� ro )r� rI r� r� �readintor� r� r� r� r� rm r� r� r� r� )r@ r� r7 r� r� r r r �testSmallReadNonBlocking� s
z:UnbufferedFileObjectClassTestCase.testSmallReadNonBlockingc C s> | j �d� | j�| j� | j�� | j�� | j�d� d S )Nr� � @) 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_skippedrb 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� rI r r r� r� rj r) )r@ ZBIGZLIMITr� r8 r r r �_testWriteNonBlocking s
z7UnbufferedFileObjectClassTestCase._testWriteNonBlockingN)rE rF rG rW rc r r r r r r r r
r r r r r r r � s r c @ s e Zd ZdZdS )�#LineBufferedFileObjectClassTestCasero N�rE rF rG rc r r r r r ) s r c @ s e Zd ZdZdS )�$SmallBufferedFileObjectClassTestCaser� Nr r r r r r . 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� rE rF rG rW r� r� r� r� r� r� r� r r r r r 3 s
r c @ s* e Zd ZdZdZeZdZe�d�Z dZ
dS )�#UnicodeWriteFileObjectClassTestCaser r r r r( N) rE rF rG rW r� r� r� r� r� r� r� r r r r r = s
r c @ s0 e Zd ZdZdZe�d�ZdZe�d�Z dZ
dS )�'UnicodeReadWriteFileObjectClassTestCaser r r r r( Nr r r r r r G s
r c @ s e Zd ZdZdd� ZdS )�NetworkConnectionTestzProve network connection.c C s t �t| jf�| _| j| _d S r2 )r
�create_connectionr� r= r� r� r? r r r r� T s z!NetworkConnectionTest.clientSetUpN)rE rF rG rW r� r r r r r Q s r c @ s e Zd ZdZdS )�
BasicTCPTest2zLTests that NetworkConnection does not break existing TCP functionality.
N)rE rF rG rW r r r r r Z 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
r5 r r r r r� a s z,NetworkConnectionNoServer.MockSocket.connectN)rE rF rG r� r r r r �
MockSocket` 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 )r@ Z
old_socketr r r �mocked_socket_moduled 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 r2 )r r� r
r: r/ rU r r$ r r� r� r� r- rf Z ECONNREFUSED)r@ r= r� r/ r r r � test_connectn s ,z&NetworkConnectionNoServer.test_connectc C s\ t �� }| �t�� }t�t|f� W d � n1 s60 Y t �� }| �|j j
|� d S r2 ) r r� r$ r r
r r� Z get_socket_conn_refused_errsr r- rf )r@ r= r/ Zexpected_errnosr r r �test_create_connectionv 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 rn ) r r
r r� r5 r r r� rf Z EAFNOSUPPORTr rp r r r � test_create_connection_timeout� s
z8NetworkConnectionNoServer.test_create_connection_timeoutN) rE rF rG r
r �
contextlib�contextmanagerr r r r! r r r r r ^ s
r c @ sv e Zd Zddd�Zdd� Zdd� Zdd � ZeZd
d � ZeZ d d
� Z
eZ dd� Z eZ
dd� ZeZdd� ZeZdd� ZdS )� NetworkConnectionAttributesTestr� c C s t j| |d� t�| � d S r� r� r� r r r rL � s z(NetworkConnectionAttributesTest.__init__c C s t �� | _d S r2 )r r� � source_portr? r r r r� � s z+NetworkConnectionAttributesTest.clientSetUpc C s | j �� d | _ t�| � d S r2 r� r? r r r r� � s
z.NetworkConnectionAttributesTest.clientTearDownc C s | j �� \}}|�� d S r2 )r; r� r r� r r r � _justAccept� s z+NetworkConnectionAttributesTest._justAcceptc C s: t jt| jftjd�| _| �| jj� | � | jj
d� d S )Nrj r� ) r
r r� r= r r1 r� rU 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( )r5 �source_addressro ) r
r r� r= r r1 r% r� rU 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
r3 r4 r r� r= r� rU 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� rj 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� rU r r� r� r? r r r �_testTimeoutValueNonamed� s z8NetworkConnectionAttributesTest._testTimeoutValueNonamedN)r� )rE rF rG rL r� r� r& Z
testFamilyr' ZtestSourceAddressr) ZtestTimeoutDefaultr+ ZtestTimeoutNoner. ZtestTimeoutValueNamedr/ ZtestTimeoutValueNonamedr0 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 rL � s z'NetworkConnectionBehaviourTest.__init__c C s d S r2 r r? r r r r� � s z*NetworkConnectionBehaviourTest.clientSetUpc C s | j �� d | _ t�| � d S r2 r� r? r r r r� � s
z-NetworkConnectionBehaviourTest.clientTearDownc C s2 | j �� \}}| �|j� t�d� |�d� d S )Nr� � done!)r; r� rU r r� r� r� r� r r r �testInsideTimeout� s
z0NetworkConnectionBehaviourTest.testInsideTimeoutc C s0 t �t| jf� | _}|�d�}| �|d� d S )Nr� r2 )r
r r� r= r� r� r� )r@ r� r� r r r �_testInsideTimeout� s
z1NetworkConnectionBehaviourTest._testInsideTimeoutc s4 t jt| jfdd� | _� | �t j� fdd�� d S )Nro rj c s
� � d�S )Nr� )r� r �r� r r r� r� zDNetworkConnectionBehaviourTest._testOutsideTimeout.<locals>.<lambda>)r
r r� r= r� r$ r5 r? r r5 r �_testOutsideTimeout� s z2NetworkConnectionBehaviourTest._testOutsideTimeoutN)r� )
rE rF rG rL r� r� r3 ZtestOutsideTimeoutr4 r6 r r r r r1 � s
r1 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 r� )r; r� r� r� r? r r �
raise_timeout s z4TCPTimeoutTest.testTCPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (TCP)�r$ r
r5 �r@ r8 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 )NFr� 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
r5 r r �r@ �okrf 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>.AlarmNrN r r r r �Alarm! s r? c s � �d S r2 r )r r� �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
r5 r r| �exc_info� traceback�
format_exc)r@ rA r rf r r@ r �testInterruptedTimeout s*
� z%TCPTimeoutTest.testInterruptedTimeoutN)
rE rF rG r; r> r� ro r r rE r r r r r7 s �r7 c @ s e Zd Zdd� Zdd� ZdS )�UDPTimeoutTestc s � fdd�}� � tj|d� d S )Nc s � j �d� � j �d� d S �Nr� r� �r; r� r� r� r? r r r8 ? s z4UDPTimeoutTest.testUDPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (UDP)r9 r: r r? r �testUDPTimeout> 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 )NFr� 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
r5 r r r< r r r r> E s z UDPTimeoutTest.testTimeoutZeroN)rE rF rG rI r> r r r r rF < s rF c @ s e Zd Zdd� Zdd� ZdS )�UDPLITETimeoutTestc s � fdd�}� � tj|d� d S )Nc s � j �d� � j �d� d S rG rH r� r? r r r8 X s z<UDPLITETimeoutTest.testUDPLITETimeout.<locals>.raise_timeoutz.Error generating a timeout exception (UDPLITE)r9 r: r r? r �testUDPLITETimeoutW 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 )NFr� r� z)caught timeout instead of error (UDPLITE)Tz%caught unexpected exception (UDPLITE)rJ rK r< r r r r> ^ s z"UDPLITETimeoutTest.testTimeoutZeroN)rE rF rG rM r> r r r r rL S s rL c @ s e Zd Zdd� Zdd� ZdS )�TestExceptionsc C sJ | � ttt�� | � ttjt�� | � ttjt�� | � ttjt�� d S r2 )r �
issubclassr r+ r
r% r& r5 r? r r r �testExceptionTreen 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: r/ r r rU rL r$ r rI )r@ Zsock0r� r r r �test_setblocking_invalidfdt s � z)TestExceptions.test_setblocking_invalidfdN)rE rF rG rP rQ r r r r rN l s rN 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� r/ r` r> r� r� r� r� � getpeername)r@ r{ �s1r^ Zs3r r r �testLinuxAbstractNamespace� s
z5TestLinuxAbstractNamespace.testLinuxAbstractNamespacec C s` dd| j d }t�tjtj��*}|�|� | �|�� |� W d � n1 sR0 Y d S )Nr� � hro )�
UNIX_PATH_MAXr
r� r/ r` r� r� �r@ r{ 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)rY r
r� r/ r$ r r` rZ 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� r/ r` 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� r/ r` 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� r/ r` Z assertRegexr� r� )r@ Zabstract_addressrV r^ r r r � testAutobind� s
z'TestLinuxAbstractNamespace.testAutobindN)
rE rF rG rY rW r[ r^ r` ra rb r r r r rS � s rS 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 r2 )r
r� r/ r� r? r r r rA � s zTestUnixDomain.setUpc C s | j �� d S r2 )r� r r? r r r rC � 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� rb r r| �getfilesystemencoding�r@ 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 rb r )r@ r� r� r� r r r r` � 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 r2 ) r� r� �abspathr �TESTFNr` r� rU r� r� r� rf r r r � testStrAddr� s zTestUnixDomain.testStrAddrc C sF t j�tj�}| �| j| �|�� | �tj |� | �
| j� � |� d S r2 ) r� r� rh r ri r` r� rg rU r� r� r� rf r r r �
testBytesAddr� s z TestUnixDomain.testBytesAddrc C sR t j�tj�}| �|�}| �| j|�dd�� | � tj
|� | � | j� � |� d S )N�ascii�surrogateescape)
r� r� rh r ZTESTFN_UNICODErg r` r� r� rU r� r� r� )r@ 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_UNENCODABLErb r� r� rh r` r� rU r� r� r� rf r r r �testUnencodableAddr� s
z"TestUnixDomain.testUnencodableAddrr� rR c C s | � t| jjd� d S )Nr( )r$ r r� r` r? r r r �testEmptyAddress� s z TestUnixDomain.testEmptyAddressN)rE rF rG rA rC rg r` r� rj rk rn ro r� rq r| r� rp r r r r rc � s
rc 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 rL s zBufferIOTest.__init__c C sZ t � ddgtt� �}| j�|�}| �|tt�� |�� }|d tt�� }| �|t� d S �NrK r )rL r� r� r� r� r� r� �r@ r7 r� r� r r r �testRecvIntoArray s z BufferIOTest.testRecvIntoArrayc C s t t�}| j�|� d S r2 �r� r� r� r� �r@ r7 r r r �_testRecvIntoArray s z BufferIOTest._testRecvIntoArrayc C sD t d�}| j�|�}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r� r� r� r� rs r r r �testRecvIntoBytearray s
z"BufferIOTest.testRecvIntoBytearrayc C sH t d�}| j�t|��}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r� r� r� r� r� rs r r r �testRecvIntoMemoryview s
z#BufferIOTest.testRecvIntoMemoryviewc C s^ t � ddgtt� �}| j�|�\}}| �|tt�� |�� }|d tt�� }| �|t� d S rr )rL r� r� r� r� r� r� �r@ r7 r� r� r� r r r �testRecvFromIntoArray) s z"BufferIOTest.testRecvFromIntoArrayc C s t t�}| j�|� d S r2 ru rv r r r �_testRecvFromIntoArray1 s z#BufferIOTest._testRecvFromIntoArrayc C sH t d�}| j�|�\}}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r� r� r� r� rz r r r �testRecvFromIntoBytearray5 s
z&BufferIOTest.testRecvFromIntoBytearrayc C sL t d�}| j�t|��\}}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r� r� r� r� r� rz r r r �testRecvFromIntoMemoryview> s
z'BufferIOTest.testRecvFromIntoMemoryviewc C s t d�}| �t| jj|d� d S )Nr] r� )r� r$ r� r� r� rv r r r �testRecvFromIntoSmallBufferG s z(BufferIOTest.testRecvFromIntoSmallBufferc C s | j �t� d S r2 r� r? r r r � _testRecvFromIntoSmallBufferL s z)BufferIOTest._testRecvFromIntoSmallBufferc C s$ t � }| j�|� | j�|d� d S r� )r� r� r� rv r r r �testRecvFromIntoEmptyBufferO s z(BufferIOTest.testRecvFromIntoEmptyBufferN)r� )rE rF rG rW rL rt rw rx Z_testRecvIntoBytearrayry Z_testRecvIntoMemoryviewr{ r| r} Z_testRecvFromIntoBytearrayr~ Z_testRecvFromIntoMemoryviewr r� r� Z _testRecvFromIntoEmptyBufferr r r r rq s$
rq 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�IsADirectoryErrorrb r� )r r� r r r �isTipcAvailable[ s
2r� 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 )Nro r� r r� )r
r� r2 rU r r� r3 r4 �TIPC_ADDR_NAMESEQ�
TIPC_STYPE�
TIPC_LOWER�
TIPC_UPPERr` �TIPC_ADDR_NAMEri r, r� r� r� r� )r@ r1 r� �srvaddrZsendaddrr� Zrecvaddrr r r �testRDMr s �
� zTIPCTest.testRDMN)rE rF rG r� r r r r r� o s r� 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 rL � 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
r� r/ r1 rU r r� r3 r4 r� r� r� r� r` r> rn r� r� r� )r@ r� r r r rA � s �
zTIPCThreadableTest.setUpc C sd t �d� t�tjtj�| _| �| jj� tjt t
t t t
d � df}| j�
|� | j�� | _d S )Nr� r� r )r� r� r
r� r/ r� rU r r� r� r� ri r� r� r� �cliaddr)r@ r� r r r r� � s
� z TIPCThreadableTest.clientSetUpc C s, | j �d�}| �|t� | �| j| j� d S r� )r� r� r� r� r� r� r� r r r r� � s z TIPCThreadableTest.testStreamc C s | j �t� | j �� d S r2 r� r? r r r r� � s z TIPCThreadableTest._testStreamN)r� r� r r r r r� � s
r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ContextManagersTestc C s� t � � � }| �|j� W d � n1 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� rU r r� r �r@ 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 �r@ r{ r� r r r �_testCreateConnectionBase� s
0z-ContextManagersTest._testCreateConnectionBasec C s2 | j �� \}}| �|j� |�d�}|�|� d S r� r� r� r r r �testCreateConnectionClose� s
z-ContextManagersTest.testCreateConnectionClosec C s\ | j �� }t�|��}|�� W d � n1 s20 Y | �|j� | �t|j d� d S r� )
r; r� r
r r r r r$ r r r� r r r �_testCreateConnectionClose� s
& z.ContextManagersTest._testCreateConnectionCloseN)rE rF rG r� r� r� r� r� r r r r r� � s
r� c @ s� e Zd Ze�eed�d�e�ddd�dd� ��Z dd � Z
d
d � Z d d
� Z e�
edu d�dd� �Ze�
edu d�dd� �Zdd� ZdS )�InheritanceTestr� zSOCK_CLOEXEC not definedr� r� r� c C sX t � t jt jt jB ��.}| �|jt j� | �|�� � W d � n1 sJ0 Y d S r2 )r
r: r/ 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� )r@ 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)r@ r� rR 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 r� r� r� )r@ r� rR 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� rU r r� r� )r@ rV r^ r r r �test_socketpair s
z InheritanceTest.test_socketpair)rE rF rG r� ro r r
r r� r� r� r� r� rq r r� r� r� r r r r r� � s �
r� rH zSOCK_NONBLOCK not definedc @ s, e Zd Zd dd�Ze�ddd�dd � �Zd
S ) �NonblockConstantTestTr� c C s� |rj| � |jtj� | � |�� |� | �t�|tjtj �tj @ � |dkrZ| �
|� � � q�| �|� � � nL| � |jtj� | � |�� d � | �
t�|tjtj �tj @ � | �|� � � d S r� ) r� r� r
r/ r� r r r� r� r� r� r� )r@ r �nonblockr5 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)r� g @rj r� ) r
r: r/ rH r� rI r� r3 r4 )r@ r �tr r r �test_SOCK_NONBLOCK; s:
�
,
(
,
.
,z'NonblockConstantTest.test_SOCK_NONBLOCKN)Tr� )rE rF rG r� r r� r� r r r r r� s
r� r: r; zneed multiprocessingc @ s@ e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
S )�TestSocketSharingc C sD |� � }|� � }t�|�}|�� \}}|�|� |�� |�� d S r2 )rr r
� fromsharer� r r )r� �qZsdata�messager r^ r
r r r �remoteProcessServerb 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�r� rQ s
slapmahfror� r� )�multiprocessingr{ �Processr� r� r; �share�pidr� r r� r
r r� r� r r� ) r@ r� r! r� r� r� r �m�receivedr r r � testSharep s&
zTestSocketSharing.testSharec C s@ | j �t�� �}| �ttj|d d� � | �ttj|d � d S )Nrp r) )r; r� r� �getpidr$ r� r
r� )r@ 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 )r@ Zorg�otherr r r �compareSockets� s
z TestSocketSharing.compareSocketsc C sD | j �t�� �}t�|�}z| �| j |� W |�� n
|�� 0 d S r2 )r; r� r� r� r
r� r� r )r@ 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 r2 ) r
r: r� r/ r r r� r� r� r� r� r )r@ �families�typesr r� �sourcer� Zsharedr r r � testTypes� s
zTestSocketSharing.testTypesN)
rE rF rG r� r� r� r� r� r� r� r r r r r� ] s
#r� c @ s e Zd ZdZdZdZdZejZ e
dd� �Z e
dd� �Z d d
� Z
d d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Z d)d*� Z d+d,� Z d-d.� Z d/d0� Z d1d2� Z!d3d4� Z"d5d6� Z#d7d8� Z$d9d:� Z%d;S )<�SendfileUsingSendTestz>
Test the send() implementation of socket.sendfile().
i � rh r� c C s� dd� }d� dd� t| j�D ��}ttjd��.}|| j| j�D ]}|�|� q>W d � n1 sb0 Y ttjd��.}|�� | _ t
| j �| jks�J �W d � n1 s�0 Y d S )Nc s s2 | |ks J �| |kr$|V | |8 } q | r.| V d S r2 r )�total�stepr r r �chunks� s
z0SendfileUsingSendTest.setUpClass.<locals>.chunksr� c S s g | ]}t �tj��� �qS r )�random�choice�string�
ascii_lettersr� r� r r r r� � s �z4SendfileUsingSendTest.setUpClass.<locals>.<listcomp>r r ) r r) �BUFSIZEr r ri �FILESIZEr� r� �FILEDATAr� )r� r� �chunkr Zcsizer r r �
setUpClass� s
�*
z SendfileUsingSendTest.setUpClassc C s t �t j� d S r2 )r r� ri )r� r r r �
tearDownClass� s z#SendfileUsingSendTest.tearDownClassc C s8 | j �tj� | j �� \}}|�| j� | �|j� |S r2 )r; r� r r� r� �TIMEOUTrU r r� r r r � accept_conn� s
z!SendfileUsingSendTest.accept_connc C s, g }|� | j�}|sq"|�|� qd�|�S r� )r� r� r� r )r@ r� r� r� r r r � recv_data� s z SendfileUsingSendTest.recv_datac C s
t |d�S )N�_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 ri r
r r� r� r� �tell�r@ r{ �filer� �meth�sentr r r �_testRegularFile� s
z&SendfileUsingSendTest._testRegularFilec C s6 | � � }| �|�}| �t|�| j� | �|| j� d S r2 �r� r� r� r� r� r� �r@ 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 r2 )r; r� �io�BytesIOr� r
r rT r� r� r� r$ rY rZ )r@ r{ r� r� r� r r r �_testNonRegularFile
s
�z)SendfileUsingSendTest._testNonRegularFilec C s6 | � � }| �|�}| �t|�| j� | �|| j� d S r2 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 ri r rU r� r
r r� r� r� )r@ r{ �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 �testEmptyFileSend+ 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 � _testOffset2 s
z!SendfileUsingSendTest._testOffsetc C sB | � � }| �|�}| �t|�| jd � | �|| jdd � � d S )Nr� r� r� r r r �
testOffset; 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 rj �GKL �r� � r; r� r r ri r
r r1 r� r� r� �r@ r{ r� r� r� r� r� r r r �
_testCountC s
�
z SendfileUsingSendTest._testCountc C s@ d}| � � }| �|�}| �t|�|� | �|| jd |� � d S )Nr� �r� r� r� r� r� �r@ r� r� r� r r r � testCountO 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 rj ro r� r� r� r r r �_testCountSmallX s
�
z%SendfileUsingSendTest._testCountSmallc C s@ d}| � � }| �|�}| �t|�|� | �|| jd |� � d S r� r� r� r r r �testCountSmalld 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� rj 駆 �� )r� r� )
r; r� r r ri r
r r� r� r� r� r r r �_testCountWithOffsetm s
z*SendfileUsingSendTest._testCountWithOffsetc C sD d}| � � }| �|�}| �t|�|� | �|| jd|d � � d S )Nr� r� r� r� r r r �testCountWithOffsetw 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 ri r
r rI r� r$ r� rT �r@ r{ 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 �r@ 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 rj ) r; r� r r ri r
r r1 r� r� r� r� r r r �_testWithTimeout� s
�
z&SendfileUsingSendTest._testWithTimeoutc C s6 | � � }| �|�}| �t|�| j� | �|| j� d S r2 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 ri r
r r� r� r$ r5 r� r r r � _testWithTimeoutTriggeredSend� s
z3SendfileUsingSendTest._testWithTimeoutTriggeredSendc C s | � � }|�d� t�d� d S )Ni�X ro )r� r� r� r� r� r r r � testWithTimeoutTriggeredSend� s
z2SendfileUsingSendTest.testWithTimeoutTriggeredSendc C s d S r2 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� r/ r% z binary modezpositive integerr� r� r� r rp ) r r ri r
r r� r' r� r� )r@ r� r r� r r r � test_errors� s6
�@
�@
����z!SendfileUsingSendTest.test_errorsN)&rE rF rG rW r� r� r� r r1 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� � sD
r� rT 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 )NrZ r� r� r r r r� � s z(SendfileUsingSendfileTest.meth_from_sockN)rE rF rG rW 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& r` r� r r� ZSkipTestr )r@ �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 abcr� � a� b� cr� )
r� �fromhexr r� r r� r� r� r
�MSG_MORE)r@ r �algo�opru 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)r� ) r� r
r r� r
�SOL_ALG� ALG_SET_KEYr� r r� r� )r@ �keyr� r r r ru 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� )r� r
r� r r� r
r r r� �
sendmsg_afalg�ALG_OP_ENCRYPTr r r� r� �ALG_OP_DECRYPT)
r@ r r r� Z
ciphertextZmsglenr r ru Z
multiplierZlongmsg�encZdecr r r � test_aes_cbc sL
�
0 �0
�. �.z!LinuxKernelCryptoAPI.test_aes_cbcr� 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
)r� r
r� r r� r
r r ZALG_SET_AEAD_AUTHSIZEr� r r r r r� r� r �Structr� r� Z
ALG_SET_OPZ
ALG_SET_IVZALG_SET_AEAD_ASSOCLENr )r@ r r ZplainZassocZ expected_ctZ expected_tagZtaglenr r r ru �resr� Z pack_uint32r r r �test_aead_aes_gcm0 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_sha256r� ) r r� �urandomr� r
r r r� r� r� r� )r@ r Z
extra_seedr ru Zrnr r r �test_drbg_pr_sha256o 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 ro )r r rp )r
r( r& r$ r� r r r� r r r �test_sendmsg_afalg_argsz 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& rU r r$ r� r` r' r� r� r r r �test_length_restriction� s (( (z,LinuxKernelCryptoAPI.test_length_restrictionN)
rE rF rG 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> �
TCP_MAXSEG� TCP_KEEPIDLE� TCP_FASTOPEN�
TCP_KEEPINTVL� TCP_NODELAY� TCP_KEEPCNTc 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� )r@ Zprovided�unknownr r? r �test_new_tcp_flags� s
�z(TestMSWindowsTCPFlags.test_new_tcp_flagsN)rE rF rG r. r0 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 ro r� r� )r r� r
�
create_serverr� r� r� r� �r@ 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
r2 r� r r: r� r/ r r� r� )r@ 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)�
reuse_portr )
r r
r$ r� r2 r� r3 r7 r� r� )r@ r� rT r r r �test_reuse_port� s
. *z CreateServerTest.test_reuse_portr8 r� r� r> c C sH t jdt jd��$}|�t jt j�s&J �W d � n1 s:0 Y d S )Nr5 r� )r
r2 r� r� r8 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 )Nr5 T�r �dualstack_ipv6)r
r2 r� r� r r� r r r �test_dualstack_ipv6_family� s
�z+CreateServerTest.test_dualstack_ipv6_familyN)rE rF rG r4 r6 r9 r� rq r r ro r r� r: r
�has_dualstack_ipv6r> r r r r r1 � s
� � r1 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� r5 r� r )r� r� ru r� �r� r@ r r �run� s
z3CreateServerFunctionalTest.echo_server.<locals>.runr� ) rM rx r� r5 r� r� rU r rm )r@ r� rB r� r rA 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� r5 r� r r� r� )r@ 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
r2 rC rD r: r3 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
r2 r� rC rD r3 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
r2 r� rC rD r: r3 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� rF r3 r r r �test_dual_stack_client_v6 s �
z4CreateServerFunctionalTest.test_dual_stack_client_v6N)rE rF rG r r1 r5 rC rD rE r� ro r r� rG rq r
r? rH rI 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 r2 �r� r )�pipesr� �fd2r r r � close_pipes- s
z8SendRecvFdsTests.testSendAndRecvFds.<locals>.close_pipesc S s | D ]}t �|� qd S r2 rM )r� rR r r r � close_fds2 s z6SendRecvFdsTests.testSendAndRecvFds.<locals>.close_fdsc S s g | ] }t �� �qS r )r� �pipe)r� ru r r r r� 7 r� z7SendRecvFdsTests.testSendAndRecvFds.<locals>.<listcomp>r� c S s g | ] \}}|�qS r r )r� �rfd�wfdr r r r� 9 r� r� r r� )r) rU r
r� r� r/ rJ r� rK r� r� r� r� r� r r� r� )r@ rP rQ rN r� r� r� r� Zfds2r� r� rm rS rT r� r r r �testSendAndRecvFds, s& &H z#SendRecvFdsTests.testSendAndRecvFdsN)rE rF rG rU r r r r rL ( s rL c C s t �� } tjt jg| �R � d S r2 )r Zthreading_setupr� ZaddModuleCleanupZthreading_cleanup)� thread_infor r r � setUpModuleR s rW �__main__)�r� �testr Z test.supportr rf r� r� r
r� r� r� rC rp r| r� r� rL r" �weakrefr r r� r* r r� r` r� �_threadr� rM r� r� r r� r� r� r� �systemr r r r r" r$ r' r) r+ r- r1 r# r6 ZHAVE_SOCKET_CANZHAVE_SOCKET_CAN_ISOTPZHAVE_SOCKET_CAN_J1939ZHAVE_SOCKET_RDSZHAVE_SOCKET_ALGZHAVE_SOCKET_QIPCRTRZHAVE_SOCKET_VSOCKr ZHAVE_SOCKET_UDPLITEZHAVE_SOCKET_BLUETOOTHr� r� r� r9 rH rI rJ rY rg ri r� r� ro r� r� r� rq r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r
rr r� r� r� r� r� r� r� r� r� r r r r3 r8 r; r= r? r@ rA re rx r{ r� r� r� r� r� r rM rO rP rQ rR r� rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rn ro rp rq rr rs rt ru r} r� r� r� r� r� r� r r r r r r rU r r r r$ r1 r7 rF rL rN rS rc rq r� r� r� r� r� r� r� r� r� r� r� r� r r r� r& r1 r@ rL rW rE �mainr r r r �<module> sV
3q�� � !$� �$
p
S
Z
*
=
>��.�%{ �7w w= # P; k 7 �
�
�
� � ��
�
�
� �
�
�
�
� �
� � �
�
� �
�
�
�
�
� �
�)��7#j ! $l
?G 9�7JS �� ',C�< e
� H3@
'
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists