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