|
Server : LiteSpeed System : Linux server51.dnsbootclub.com 4.18.0-553.62.1.lve.el8.x86_64 #1 SMP Mon Jul 21 17:50:35 UTC 2025 x86_64 User : nandedex ( 1060) PHP Version : 8.1.33 Disable Function : NONE Directory : /opt/cppython/lib/python3.8/test/__pycache__/ |
U
>��g|� �
@ s�
d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlmZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z z d dl Z W n e k
r� dZ Y nX z d dl Z W n e k
�r$ dZ Y nX ej Z d�!d�Z"dZ#dZ$e�%� d kZ&z d dl'Z'W n e k
�rp dZ'Y nX d
d � Z(d d
� Z)dd� Z*dd� Z+dd� Z,dd� Z-dd� Z.ej/dd� �Z0e)� Z1e*� Z2e+� Z3e,� Z4e-� Z5e.� Z6e�d�j7Z8G dd � d e j9�Z:G d d � d e j9�Z;G d d � d e j9�Z<G d!d"� d"e j9�Z=G d#d$� d$e j9�Z>G d%d&� d&�Z?G d'd(� d(e:e?�Z@G d)d*� d*e;e?�ZAG d+d,� d,e=e?�ZBG d-d.� d.e>e?�ZCe �De dkd/�e �Ee6d0�e �Ee(� d1kd2�G d3d4� d4e j9e?����ZFG d5d6� d6e@�ZGG d7d8� d8e j9e?�ZHG d9d:� d:e j9�ZIG d;d<� d<eI�ZJG d=d>� d>e<eIe?�ZKG d?d@� d@eJeK�ZLG dAdB� dBeI�ZMG dCdD� dDeM�ZNG dEdF� dFeI�ZOG dGdH� dHeO�ZPG dIdJ� dJeO�ZQG dKdL� dLeO�ZRG dMdN� dNeO�ZSG dOdP� dPeS�ZTdQdR� ZUdSdT� ZVdUdV� ZWG dWdX� dXe j9�ZXe �Ee1dY�G dZd[� d[e j9��ZYe �Ee1dY�G d\d]� d]eB��ZZe �Ee2d^�G d_d`� d`e j9��Z[e �Ee3da�G dbdc� dce j9��Z\e �Ee3da�G ddde� deeC��Z]e �Ee5df�G dgdh� dhe j9��Z^e �De dkd/�e �Ee6d0�G didj� dje j9���Z_G dkdl� dleG�Z`G dmdn� dneA�ZaG dodp� dpe<�ZbG dqdr� dreb�ZcG dsdt� dteb�ZdG dudv� dveb�ZeG dwdx� dxeb�ZfG dydz� dzeb�ZgG d{d|� d|eb�ZhG d}d~� d~eh�ZiG dd�� d�ei�ZjG d�d�� d�ei�ZkG d�d�� d�eb�ZlG d�d�� d�el�ZmG d�d�� d�el�ZnG d�d�� d�ecel�ZoG d�d�� d�e j9�ZpG d�d�� d�eh�ZqG d�d�� d�eh�ZrG d�d�� d�edefeKeQ�ZseVejd��G d�d�� d�ekes��ZteVejd��G d�d�� d�enes��ZueVejd��G d�d�� d�eoes��ZvG d�d�� d�edefeKeT�ZweVejd��e �Eejxd��eWd�d��G d�d�� d�ekew����ZyeVejd��e �Eejxd��eWd�d��G d�d�� d�enew����ZzeVejd��e �Eejxd��eWd�d��G d�d�� d�eoew����Z{eVejd��e �Eejxd��eVed��eWd�d��G d�d�� d�erew�����Z|eVejd��e �Eejxd��eVed��eWd�d��G d�d�� d�ecerew�����Z}G d�d�� d�egeLeP�Z~eVejd��G d�d�� d�eje~��ZeVejd��G d�d�� d�eneme~��Z�eVejd��G d�d�� d�eoeme~��Z�G d�d�� d�eeegeLeR�Z�eVejd��e �De&d��eWd�d�d��G d�d�� d�eje�����Z�eVejd��e �De&d��eWd�d�d��G d�d�� d�eneme�����Z�eVejd��e �De&d��eWd�d�d��G d�d�� d�eoeme�����Z�G d�d�� d�egeLeN�Z�eVejd��eVedG d�dĄ d�eje����Z�eVejd��eVedG d�dƄ d�eneme����Z�eVejd��eVedG d�dȄ d�eoeme����Z�eVejd�d��eVed�d�dʃG d�d̄ d�eqe����Z�eVejd�d��eVed�d�dʃG d�d΄ d�eceqe����Z�G d�dЄ d�e j9�Z�eVedуe �Ee�ed҃� p�e�edӃdԡG d�dք d�e�eQ���Z�eVedуe �Ee�ed҃�
p4e�edӃdԡG d�d d�e�e<eJeP���Z�G d�dڄ d�e@�Z�G d�d܄ d�eH�Z�G d�dބ d�e@�Z�G d�d� d�eG�Z�G d�d� d�e��Z�G d�d� d�e��Z�G d�d� d�e��Z�G d�d� d�e��Z�G d�d� d�e��Z�G d�d� d�e��Z�G d�d� d�e��Z�G d�d� d�e�e`�Z�G d�d� d�e j9�Z�G d�d� d�e:e?�Z�G d�d�� d�e:e?�Z�G d�d�� d�e:�Z�G d�d�� d�e;�Z�G d�d�� d�e j9�Z�e �Ee jd�kd��G d��d � �d e j9��Z�e �Ee�ed�d�G �d�d� �de j9��Z�G �d�d� �deG�Z��dZ��dZ��dZ��d �d
� Z�e �Ee�� �d �G �d �d
� �d
e j9��Z�e �Ee�� �d �G �d�d� �de j9e?��Z�G �d�d� �de@�Z�G �d�d� �de j9�Z�e �Ee�e�d��d�G �d�d� �de j9��Z�e �Ee
j��dk�d�e �Ee �d�G �d�d � �d e:���Z�G �d �d � �d e@�Z�e �Ee�e
�d ��d �G �d!�d"� �d"e���Z�e �Ee4�d#�G �d$�d%� �d%e j9��Z�e �Ee j���d&��d'�G �d(�d)� �d)e j9��Z�G �d*�d+� �d+e j9�Z�G �d,�d-� �d-e j9�Z��d.�d/� Z�e��d0k�
r�e�� dS (1 � N)�support)�proxyFu Michael Gilfix was hereሴ
�utf-8g N@�� �AIXc C sp t d kr d S ttd�sd S z*tdd��} t �| tjd�}W 5 Q R X W n tk
rZ Y d S X t�d|�d S d S )N� IOCTL_VM_SOCKETS_GET_LOCAL_CIDz
/dev/vsock�rbz �Ir ) �fcntl�hasattr�socket�open�ioctlr �OSError�struct�unpack)�f�r� r �//opt/cppython/lib/python3.8/test/test_socket.py�get_cid2 s
r c C s@ zt � t jt jt j�} W n ttfk
r2 Y dS X | �� dS �NFT)r �PF_CAN�SOCK_RAW�CAN_RAW�AttributeErrorr �close��sr r r �_have_socket_can? s r c C s@ zt � t jt jt j�} W n ttfk
r2 Y dS X | �� dS r )r r �
SOCK_DGRAM� CAN_ISOTPr r r r r r r �_have_socket_can_isotpI s r" c C s> zt � t jt jd�} W n ttfk
r0 Y dS X | �� dS �Nr FT)r �PF_RDS�SOCK_SEQPACKETr r r r r r r �_have_socket_rdsS s r& c C s> zt � t jt jd�} W n ttfk
r0 Y dS X | �� dS r# )r �AF_ALGr% r r r r r r r �_have_socket_alg] s r( c C s> zt � t jt jd�} W n ttfk
r0 Y dS X | �� dS r# )r �
AF_QIPCRTRr r r r r r r r �_have_socket_qipcrtrg s r* c C s t � d k } | S �N)r )�retr r r �_have_socket_vsockq s
r- c c s. t �� }zt �| � d V W 5 t �|� X d S r+ )r �getdefaulttimeout�setdefaulttimeout)�timeoutZ old_timeoutr r r �socket_setdefaulttimeoutw s
r1 �ic @ s e Zd Zdd� Zdd� ZdS )�
SocketTCPTestc C s. t � t jt j�| _t�| j�| _| j�� d S r+ )r �AF_INET� SOCK_STREAM�servr � bind_port�port�listen��selfr r r �setUp� s zSocketTCPTest.setUpc C s | j �� d | _ d S r+ �r6 r r: r r r �tearDown� s
zSocketTCPTest.tearDownN��__name__�
__module__� __qualname__r< r> r r r r r3 � s r3 c @ s e Zd Zdd� Zdd� ZdS )�
SocketUDPTestc C s$ t � t jt j�| _t�| j�| _d S r+ )r r4 r r6 r r7 r8 r: r r r r< � s zSocketUDPTest.setUpc C s | j �� d | _ d S r+ r= r: r r r r> � s
zSocketUDPTest.tearDownNr? r r r r rC � s rC c s4 e Zd Z� fdd�Z� fdd�Z� fdd�Z� ZS )�ThreadSafeCleanupTestCasec s t � j||� t�� | _d S r+ )�super�__init__� threading�RLock�
_cleanup_lock�r; �args�kwargs�� __class__r r rF � s z"ThreadSafeCleanupTestCase.__init__c
s, | j � t� j||�W 5 Q R � S Q R X d S r+ )rI rE �
addCleanuprJ rM r r rO � s z$ThreadSafeCleanupTestCase.addCleanupc
s, | j � t� j||�W 5 Q R � S Q R X d S r+ )rI rE �
doCleanupsrJ rM r r rP � s z$ThreadSafeCleanupTestCase.doCleanups)r@ rA rB rF rO rP �
__classcell__r r rM r rD � s rD c @ sD e Zd ZdZdZdZe�e�ZdZ e de�e �d 7 Z dd� Z
d S )
�
SocketCANTest�vcan0� z=IB3x8sz@3I4l2I�x� c C sb t � t jt jt j�| _| �| jj� z| j�| jf� W n$ t k
r\ | �
d| j � Y nX d S )N�%network interface `%s` does not exist) r r r r r rO r �bind� interfacer �skipTestr: r r r r< � s �zSocketCANTest.setUpN) r@ rA rB rY �bufsize�
can_frame_fmtr �calcsizeZcan_frame_size�bcm_cmd_msg_fmtr< r r r r rR � s
rR c @ s e Zd ZdZdd� ZdS )�
SocketRDSTest� c C sX t � t jt jd�| _| �| jj� zt�| j�| _W n t k
rR | �
d� Y nX d S )Nr zunable to bind RDS socket) r r$ r% r6 rO r r r7 r8 r rZ r: r r r r< � s zSocketRDSTest.setUpN)r@ rA rB r[ r< r r r r r_ � s r_ c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�ThreadableTestc C s$ | j | _| j| _| j| _ | j| _d S r+ )r< �_ThreadableTest__setUpr> �_ThreadableTest__tearDown�_setUp� _tearDownr: r r r rF ! s zThreadableTest.__init__c C s | j �� d S r+ )� server_ready�setr: r r r �serverExplicitReady( s z"ThreadableTest.serverExplicitReadyc C s� t �� | _| j�� t�� | _t�� | _t�� | _t �
d�| _ d| _ | � � }|�
d�}||d d � }t| d| �}t�| j|f�| _z&z | �� W n d| _ � Y nX W 5 | j�� X | j�� d S )N� F�.�_T)r Zwait_threads_exit� wait_threads� __enter__rG �Eventrf � client_ready�done�queue�Queue�server_crashed�id�rfind�getattr�thread�start_new_thread� clientRunZ
client_threadrg rb �wait)r; Z
methodnamer2 Z test_methodr r r rd / s,
� zThreadableTest._setUpc C s> | � � | j�� | j�d d d � | j�� r:| j�� }|�d S r+ )rc rp rz rl �__exit__rq �qsize�get�r; �excr r r re J s
zThreadableTest._tearDownc
C s� | j �� zRz | �� W n@ tk
rX } z"| j�|� | �� W Y �W �d S d }~X Y nX W 5 | j�� X | j r|| �� d S t
|d�s�t d��z>z
|� W n. tk
r� } z| j�|� W 5 d }~X Y nX W 5 | �� X d S )N�__call__z%test_func must be a callable function) rf rz ro rg � clientSetUp�
BaseExceptionrq �put�clientTearDownrs r � TypeError)r; Z test_func�er r r ry S s$
"zThreadableTest.clientRunc C s t d��d S )Nz clientSetUp must be implemented.)�NotImplementedErrorr: r r r r� i s zThreadableTest.clientSetUpc C s | j �� t�� d S r+ )rp rg rw �exitr: r r r r� l s
z ThreadableTest.clientTearDownN)
r@ rA rB rF rh rd re ry r� r� r r r r ra � s & ra c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedTCPSocketTest�runTestc C s t j| |d� t�| � d S �N)�
methodName�r3 rF ra �r; r� r r r rF r s z ThreadedTCPSocketTest.__init__c C s t � t jt j�| _d S r+ )r r4 r5 �clir: r r r r� v s z!ThreadedTCPSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r+ �r� r ra r� r: r r r r� y s
z$ThreadedTCPSocketTest.clientTearDownN)r� �r@ rA rB rF r� r� r r r r r� p s
r� c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedUDPSocketTestr� c C s t j| |d� t�| � d S r� )rC rF ra r� r r r rF � s z ThreadedUDPSocketTest.__init__c C s t � t jt j�| _d S r+ )r r4 r r� r: r r r r� � s z!ThreadedUDPSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r+ r� r: r r r r� � s
z$ThreadedUDPSocketTest.clientTearDownN)r� r� r r r r r� ~ s
r� c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedCANSocketTestr� c C s t j| |d� t�| � d S r� )rR rF ra r� r r r rF � s z ThreadedCANSocketTest.__init__c C sD t � t jt jt j�| _z| j�| jf� W n tk
r> Y nX d S r+ )r r r r r� rX rY r r: r r r r� � s
z!ThreadedCANSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r+ r� r: r r r r� � s
z$ThreadedCANSocketTest.clientTearDownN)r� r� r r r r r� � s
r� c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedRDSSocketTestr� c C s t j| |d� t�| � d S r� )r_ rF ra r� r r r rF � s z ThreadedRDSSocketTest.__init__c C sN t � t jt jd�| _z | j�tdf� | j�� | _W n tk
rH Y nX d S �Nr ) r r$ r% r� rX �HOST� getsockname�cli_addrr r: r r r r� � s z!ThreadedRDSSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r+ r� r: r r r r� � s
z$ThreadedRDSSocketTest.clientTearDownN)r� r� r r r r r� � s
r� �
need fcntlz%VSOCK sockets required for this test.� z-This test can only be run on a virtual guest.c @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )� ThreadedVSOCKSocketStreamTestr� c C s t jj| |d� t�| � d S r� ��unittest�TestCaserF ra r� r r r rF � s z&ThreadedVSOCKSocketStreamTest.__init__c C sh t � t jt j�| _| �| jj� | j�t jtf� | j� � | �
� | j� � \| _ | _
| �| j j� d S r+ )r �AF_VSOCKr5 r6 rO r rX �VMADDR_CID_ANY� VSOCKPORTr9 rh �accept�conn�connaddrr: r r r r< � s
z#ThreadedVSOCKSocketStreamTest.setUpc C sD t �d� t�tjtj�| _| �| jj� t� }| j� |t
f� d S )N皙�����?) �time�sleepr r� r5 r� rO r r �connectr� )r; Zcidr r r r� � s
z)ThreadedVSOCKSocketStreamTest.clientSetUpc C s | j �d�}| �|t� d S �N� )r� �recv� assertEqual�MSG�r; �msgr r r �
testStream� s z(ThreadedVSOCKSocketStreamTest.testStreamc C s | j �t� | j �� d S r+ �r� �sendr� r r: r r r � _testStream� s z)ThreadedVSOCKSocketStreamTest._testStreamN)r� �r@ rA rB rF r< r� r� r� r r r r r� � s
r� c @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )�SocketConnectedTestr� c C s t j| |d� d S r� )r� rF r� r r r rF � s z SocketConnectedTest.__init__c C s* t �| � | �� | j�� \}}|| _d S r+ )r� r< rh r6 r� �cli_conn�r; r� �addrr r r r< � s
zSocketConnectedTest.setUpc C s | j �� d | _ t�| � d S r+ )r� r r� r> r: r r r r> � s
z SocketConnectedTest.tearDownc C s( t �| � | j�t| jf� | j| _d S r+ )r� r� r� r� r� r8 � serv_connr: r r r r� � s
z SocketConnectedTest.clientSetUpc C s | j �� d | _ t�| � d S r+ )r� r r� r� r: r r r r� � s
z"SocketConnectedTest.clientTearDownN)r� �r@ rA rB rF r< r> r� r� r r r r r� � s
r� c @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )�SocketPairTestr� c C s t jj| |d� t�| � d S r� r� r� r r r rF � s zSocketPairTest.__init__c C s t �� \| _| _d S r+ )r �
socketpairr6 r� r: r r r r< s zSocketPairTest.setUpc C s | j �� d | _ d S r+ r= r: r r r r> s
zSocketPairTest.tearDownc C s d S r+ r r: r r r r� s zSocketPairTest.clientSetUpc C s | j �� d | _ t�| � d S r+ r� r: r r r r�
s
z SocketPairTest.clientTearDownN)r� r� r r r r r� � s
r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�SocketTestBasec C s | � � | _| �� d S r+ )� newSocketr6 �
bindServerr: r r r r< s
zSocketTestBase.setUpc C s | � | j� | j�� | _d S r+ )�bindSockr6 r� � serv_addrr: r r r r� # s zSocketTestBase.bindServerc C s | j �� d | _ d S r+ r= r: r r r r> ( s
zSocketTestBase.tearDownN)r@ rA rB r< r� r> r r r r r� s r� c s e Zd Z� fdd�Z� ZS )�SocketListeningTestMixinc s t � �� | j�� d S r+ )rE r< r6 r9 r: rM r r r< 0 s
z SocketListeningTestMixin.setUp�r@ rA rB r< rQ r r rM r r� - s r� c s< e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Z� ZS ) �ThreadedSocketTestMixinc s t � j||� t�| � d S r+ )rE rF ra rJ rM r r rF = s z ThreadedSocketTestMixin.__init__c C s | � � | _| �� d S r+ )�newClientSocketr� �
bindClientr: r r r r� A s
z#ThreadedSocketTestMixin.clientSetUpc C s | � � S r+ )r� r: r r r r� E s z'ThreadedSocketTestMixin.newClientSocketc C s | � | j� | j�� | _d S r+ )r� r� r� r� r: r r r r� I s z"ThreadedSocketTestMixin.bindClientc C s | j �� d | _ t�| � d S r+ r� r: r r r r� N s
z&ThreadedSocketTestMixin.clientTearDown) r@ rA rB rF r� r� r� r� rQ r r rM r r� 5 s
r� c s@ e Zd Z� fdd�Z� fdd�Z� fdd�Z� fdd�Z� ZS ) �ConnectedStreamTestMixinc s* t � �� | �� | j�� \}}|| _d S r+ )rE r< rh r6 r� r� r� rM r r r< ] s
z ConnectedStreamTestMixin.setUpc s | j �� d | _ t� �� d S r+ )r� r rE r> r: rM r r r> e s
z!ConnectedStreamTestMixin.tearDownc s$ t � �� | j�| j� | j| _d S r+ )rE r� r� r� r� r� r: rM r r r� j s
z$ConnectedStreamTestMixin.clientSetUpc s8 z| j �� d | _ W n tk
r( Y nX t� �� d S r+ )r� r r rE r� r: rM r r r� o s
z'ConnectedStreamTestMixin.clientTearDown)r@ rA rB r< r> r� r� rQ r r rM r r� T s r� c s$ e Zd Z� fdd�Zdd� Z� ZS )�UnixSocketTestBasec s( t �� | _| �tj| j� t� �� d S r+ )�tempfile�mkdtemp�dir_pathrO �os�rmdirrE r< r: rM r r r< � s
zUnixSocketTestBase.setUpc C s, t j| jd�}t�||� | �tj|� d S )N)�dir)r� �mktempr� r �bind_unix_socketrO �unlink)r; �sock�pathr r r r� � s zUnixSocketTestBase.bindSock)r@ rA rB r< r� rQ r r rM r r� x s r� c @ s e Zd Zdd� ZdS )�UnixStreamBasec C s t � t jt j�S r+ )r �AF_UNIXr5 r: r r r r� � s zUnixStreamBase.newSocketN�r@ rA rB r� r r r r r� � s r� c s( e Zd ZeZ� fdd�Zdd� Z� ZS )� InetTestBasec s t � �� | jd | _d S �Nri )rE r< r� r8 r: rM r r r< � s
zInetTestBase.setUpc C s t j|| jd� d S )N��host)r r7 r� �r; r� r r r r� � s zInetTestBase.bindSock)r@ rA rB r� r� r< r� rQ r r rM r r� � s r� c @ s e Zd Zdd� ZdS )� TCPTestBasec C s t � t jt j�S r+ )r r4 r5 r: r r r r� � s zTCPTestBase.newSocketNr� r r r r r� � s r� c @ s e Zd Zdd� ZdS )� UDPTestBasec C s t � t jt j�S r+ )r r4 r r: r r r r� � s zUDPTestBase.newSocketNr� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�SCTPStreamBasec C s t � t jt jt j�S r+ )r r4 r5 � IPPROTO_SCTPr: r r r r� � s �zSCTPStreamBase.newSocketNr� r r r r r� � s r� c @ s e Zd ZejZdS )�
Inet6TestBaseN)r@ rA rB r �HOSTv6r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )� UDP6TestBasec C s t � t jt j�S r+ )r �AF_INET6r r: r r r r� � s zUDP6TestBase.newSocketNr� r r r r r� � s r� c s* dd� � � �fdd�}dd� }| r&|S |S )Nc _ s d S r+ r �rK rL r r r � client_pass� s z%skipWithClientIf.<locals>.client_passc s* t ���| �}t| t�s&� fdd�|_|S )Nc s � S r+ r �r )r� r r �<lambda>� � z3skipWithClientIf.<locals>.skipdec.<locals>.<lambda>)r� �skip�
isinstance�type� client_skip)�objZretval�r� �reasonr r �skipdec� s
z!skipWithClientIf.<locals>.skipdecc S s" t | t�s t| d�s dd� | _| S )Nr� c S s | S r+ 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 )Nc s g | ]}t � |�s|�qS r )r ��.0�namer� r r �
<listcomp>� s
z requireAttrs.<locals>.<listcomp>� don't have �, c s s | ]
}|V qd S r+ r r� r r r � <genexpr>� s z requireAttrs.<locals>.<genexpr>)r� �join)r� Z
attributes�missingr r� r � requireAttrs� s
�r� c
G s� d }dd� | D �}|r0dd� dd� |D �� }nPdd� | D �}ztj|� }W n* tk
rv } z t|�}W 5 d }~X Y n
X |�� t|d k d�d� d d� | D ��|��S )
Nc S s$ g | ] }t |t�rtt|�s|�qS r )r� �strr r �r� r� r r r r� � s
�z!requireSocket.<locals>.<listcomp>r� r� c s s | ]
}|V qd S r+ r r� r r r r� � s z requireSocket.<locals>.<genexpr>c S s$ g | ] }t |t�r tt|�n|�qS r )r� r� rv r r� r r r r� � s �z can't create socket({0}): {1}c s s | ]}t |�V qd S r+ )r� �r� �or r r r� � s )r� r r r� r r� �format)rK �errr� Zcallargsr r� r r r �
requireSocket� s& � ��r c @ sH e Zd Zdd� Zdd� Ze�edk d�dd� �Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
e�ejdkd�dd� �Zdd� Zdd� Zdd � Ze�eed �d �e�eed �d �d!d"� ��Ze�eed#�d$�d%d&� �Ze�eed'�d(�d)d*� �Ze�eed+�d,�d-d.� �Ze�eed/�d0�d1d2� �Zd3d4� Z d5d6� Z e j d7d8� �Z d9d:� Z!d;d<� Z"e�eed=�d>�d?d@� �Z#e�eedA�dB�dCdD� �Z$e�eedA�dB�dEdF� �Z%e�eedG�dH�dIdJ� �Z&e�eedG�dH�dKdL� �Z'dMdN� Z(dOdP� Z)dQdR� Z*dSdT� Z+dUdV� Z,dWdX� Z-dYdZ� Z.e�ejdkd�d[d\� �Z/e�ejdkd�e�eed]�d^�d_d`� ��Z0dadb� Z1dcdd� Z2e�e �3de�df�dgdh� �Z4didj� Z5dkdl� Z6dmdn� Z7dodp� Z8dqdr� Z9dsdt� Z:dudv� Z;dwdx� Z<dydz� Z=d{d|� Z>d}d~� Z?e j dd�� �Z@e�e jAd��d�d�� �ZBe�e jAd��d�d�� �ZCe�e jAd��e�DejEd�kd��e�DeFd��d�d�� ���ZGe�e jAd��e�ejEd�kd��d�d�� ��ZHe�e jAd��e�DejEd�kd��e�DeFd��d�d�� ���ZIe�e jAd��e�ejEd�kd��d�d�� ��ZJd�d�� ZKd�d�� ZLd�d�� ZMe�eed��d��d�d�� �ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSe�Dejdkd��d�d�� �ZTd�d�� ZUd�d�� ZVdS )��GeneralModuleTestsc C s< dd l }| �tj|jk� t�� }| �|tj� |�� d S r� )�_socket�
assertTruer �
SocketType�assertIsInstancer )r; r r r r r � test_SocketType_is_socketobject s
z2GeneralModuleTests.test_SocketType_is_socketobjectc C s� t � t jt j�}|�� | �d|�� t|�� | �dt j t|�� | �dt j t|�� | �dt|�� | �dt|�� |�d� | �dt|�� | �t|� � �t|�� W 5 Q R X | �dt|�� | �dt|�� d S ) Nzfd=%iz family=%sztype=%szproto=0�raddr�� 127.0.0.1r �laddrz[closed])
r r4 r5 �assertIn�fileno�repr� assertNotInrX r� r� �r; r r r r � test_repr s
"z GeneralModuleTests.test_reprNzneed _socket modulec C sr t �t jt j�}z.d|�� |j|j|jf }| � t
|�|� W 5 |�� X d|j|j|jf }| � t
|�|� d S )Nz4<socket object, fd=%s, family=%s, type=%s, proto=%s>z4<socket object, fd=-1, family=%s, type=%s, proto=%s>) r r r4 r5 r r �familyr� �protor� r )r; r �expectedr r r �test_csocket_repr s �
�z$GeneralModuleTests.test_csocket_reprc C sl t � t jt j��"}t|�}| �|�� |�� � W 5 Q R X d }z |�� W n tk
r\ Y n X | �d� d S )NzSocket proxy still exists)r r4 r5 r r� r �ReferenceError�fail)r; r �pr r r � test_weakref# s z GeneralModuleTests.test_weakrefc C sr d}| j t|d d��
t�W 5 Q R X | j t|d d�� tj�W 5 Q R X | j t|d d�� tj�W 5 Q R X d S )Nz$Error raising socket exception (%s).r �r� z
socket.herrorzsocket.gaierror)� assertRaisesr r �herror�gaierrorr� r r r �testSocketError/ s z"GeneralModuleTests.testSocketErrorc C sF t � t jt j�}| �|j� |�d� |�� }| �t��}|� d|� W 5 Q R X | �
t |j �d� | �t��}|� d|� W 5 Q R X | �
t |j �d� | �t��}|� dd � W 5 Q R X | �
dt |j �� | �t��}|� dd|� W 5 Q R X | �
t |j �d� | �t��}|� dd|� W 5 Q R X | �
t |j �d� | �t��}|� ddd � W 5 Q R X | �
dt |j �� | �t��}|� dd |� W 5 Q R X | �
d
t |j �� | �t��}|� dd d � W 5 Q R X | �
d
t |j �� | �t��}|� d� W 5 Q R X | �
d t |j �� | �t��}|� dd|d � W 5 Q R X | �
d
t |j �� d S )N)� r u ☠z*a bytes-like object is required, not 'str'y @z.a bytes-like object is required, not 'complex'� fooz not NoneTyper �barzan integer is requiredz (1 given)� z (4 given))r r4 r rO r rX r� r r� �sendtor� r� � exceptionr )r; r Zsockname�cmr r r �testSendtoErrors9 sT
� � � � z#GeneralModuleTests.testSendtoErrorsc C s@ t j t jrt j t j t j t j t j t j t j t j
d S r+ ) r r4 �has_ipv6r� r5 r r �SOCK_RDMr% �
SOL_SOCKET� SO_REUSEADDRr: r r r �testCrucialConstantse s z'GeneralModuleTests.testCrucialConstantsc C s t j t j t jrt j d S r+ )r � IPPROTO_TCP� IPPROTO_UDPr* � IPPROTO_IPV6r: r r r �testCrucialIpProtoConstantsr s z.GeneralModuleTests.testCrucialIpProtoConstants�nt�Windows specificc C s4 t j t j t j t j t j t j t j t j d S r+ ) r ZIPPROTO_ICLFXBMZ
IPPROTO_STZ IPPROTO_CBTZ IPPROTO_IGPZ IPPROTO_RDPZ IPPROTO_PGMZ IPPROTO_L2TPr� r: r r r � testWindowsSpecificConstantsx s z/GeneralModuleTests.testWindowsSpecificConstantsc C s� t �� }zt �|�}W n tk
r4 | �d� Y nX | �|�d�dkd� zt �|�\}}}W n tk
r~ | �d� Y nX ||g| }t �|�}||kr�| � d|t
|�f � d S )N�name lookup failurerj r zError resolving host to ip.z=Error testing host resolution mechanisms. (fqdn: %s, all: %s)) r � gethostname�
gethostbynamer rZ r �find�
gethostbyaddr�getfqdnr r )r; �hostnameZipZhname�aliases�ipaddrsZall_host_namesZfqhnr r r �testHostnameRes� s
z"GeneralModuleTests.testHostnameResc C sL t jddfD ]}| �t�|�|� q t jfD ]}| �|t�|�d � q,d S )Nz10.0.0.1�255.255.255.255r� )r ZHOSTv4r� r r8 r r: )r; r� r� r r r �test_host_resolution� s z'GeneralModuleTests.test_host_resolutionc
C sZ d}dD ]L}| j t|d�� t�|� W 5 Q R X | j t|d�� t�|� W 5 Q R X qd S )Nz[resolving an invalid IP address did not raise OSError; can be caused by a broken DNS server)z0.1.1.~1z1+.1.1.1z::1qz::1::2z1:1:1:1:1:1:1:1:1r )r r r r8 r: )r; Z explanationr� r r r � test_host_resolution_bad_address� s �z3GeneralModuleTests.test_host_resolution_bad_address� sethostnamez test needs socket.sethostname()r7 z test needs socket.gethostname()c
C s� t �� }zt �d� W n< tk
rR } z |jtjkr@| �d� n� W 5 d }~X Y nX z.| �t �� d� t �d� | �t �� d� W 5 t �|� X d S )N�newztest should be run as roots barr$ )r r7 rC r �errno�EPERMrZ r� )r; Zoldhnr� r r r �test_sethostname� s
z#GeneralModuleTests.test_sethostname� if_nameindexz$socket.if_nameindex() not available.c C s� t �� }|D ]p\}}| �|t� | �|t� | �|d� t �|�}| �|t� | �||� t �|�}| �|t� | �||� q d S r� ) r rH r �intr� �
assertGreater�if_nametoindexr� �if_indextoname)r; Z
interfaces�indexr� Z_index�_namer r r �testInterfaceNameIndex� s
z)GeneralModuleTests.testInterfaceNameIndexrL z&socket.if_indextoname() not available.c
C s� | � ttjd� | � ttjd� | � ttjdd � | � ttjd� ttd�r�tt�� �}|D ]8}|d }||kr^| � ttf�� t�|� W 5 Q R X q^dD ]0}||kr�| � ttf�� t�|� W 5 Q R X q�d S ) Nr ���r� �� � _DEADBEEFrH � )� �� l ���� ) r r r rL �
OverflowErrorr� r �dictrH )r; �indicesrM Zindex2r r r � testInvalidInterfaceIndexToName� s
z2GeneralModuleTests.testInvalidInterfaceIndexToNamerK z&socket.if_nametoindex() not available.c C s$ | � ttjd� | � ttjd� d S )Nr rR )r r� r rK r r: r r r � testInvalidInterfaceNameToIndex� s z2GeneralModuleTests.testInvalidInterfaceNameToIndex� getrefcountz test needs sys.getrefcount()c C sL zt �t�}t�td� W n, tk
rF t �t�|krB| �d� Y nX d S )Nr z$socket.getnameinfo loses a reference)�sysrZ r@ r � getnameinfor� r )r; �origr r r �testRefCountGetNameInfo� s
z*GeneralModuleTests.testRefCountGetNameInfoc C s* zt �dd� W n tk
r$ Y nX d S )N)rU r r r r )r r\ r r: r r r �testInterpreterCrash� s z'GeneralModuleTests.testInterpreterCrashc C s� t jdt jdt jdt jdi}|�� D ]d\}}d|> d }dD ]$}| �||@ ||||@ ��|@ � q<||�}| �||@ |� | �t|d� q$d S )N� � ri )r ri � i ��r� igE#i2Tvl ) r �htonl�ntohl�htons�ntohs�itemsr� r rU )r; Zsizes�func�size�maskr2 Zswappedr r r �testNtoH� s � "zGeneralModuleTests.testNtoHc C s dd l }ddddg}|dg }ddddd > g}||jd |jd g }d
|jg}|D ]}t�|� t�|� qT|D ]}t�|� t�|� qr|D ]$}| �t tj|� | �t tj|� q�|D ]$}| �t tj|� | �t tj|� q�|D ]$}| �
t tj|� | �
t tj|� q�d S ) Nr ri r� rb rT rP ���rS rQ � ) � _testcapiZINT_MIN�INT_MAXr rf re rd rc r rU � assertWarns�DeprecationWarning)r; rn Z
s_good_valuesZ
l_good_valuesZ l_bad_valuesZ s_bad_valuesZs_deprecated_values�kr r r �testNtoHErrors
s.
�
z!GeneralModuleTests.testNtoHErrorsc C s& | j }tj�d�s tjdkr"d}nd}|D ]0}zt�|d�}W q`W q* tk
rX Y q*X q*t�ttd�rvt�� dkr�t�|�}|||� zt�|d�}W n tk
r� d }Y n X |||� t j
s�|t� |�|� |t� |d�|� |d k �r|t� |d�|� | � t
tj d � | � t
tj d
� d S ) N)ZfreebsdZnetbsdZ gnukfreebsd)�linux�darwin)�daytimeZqotd�domain)Zechorv rw Ztcp�getandroidapilevel� ZudprP rm )r� r[ �platform�
startswithr �
getservbynamer r rx r Z
is_android�
getservbyportr rU )r; �eqZservicesZservicer8 Zport2Zudpportr r r �
testGetServBy& s< �
�
z GeneralModuleTests.testGetServByc
C s� | � t�� d � t�� �}| � |�� d � W 5 Q R X td��x | � t�� d� t�� �}| � |�� d� W 5 Q R X t�d � | � t�� d � t�� �}| � |�� d � W 5 Q R X W 5 Q R X | �ttjd� | �ttjd� d S )N�
rP Zspam) r� r r. �
gettimeoutr1 r/ r �
ValueErrorr� )r; r r� r r r �testDefaultTimeoutR 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� r: r r r � testIPv4_inet_aton_fourbytesk s z/GeneralModuleTests.testIPv4_inet_aton_fourbytes� inet_ptonz test needs socket.inet_pton()c s* ddl m}m�m� � �fdd�}�fdd�}��d|d�� ��d|d �� ��d
|d �� ��d |d
�� ��d|d�� ts�||d� ||d� ||d� ||d� ||d� ��d|d�� ��d|d �� ��d
|d �� ��d|d�� ||d� ||d� ||d� ||d� ||d� d S )Nr )r� r� r4 c s
�� | �S r+ r ��a)r4 r� r r r� w r� z5GeneralModuleTests.testIPv4toString.<locals>.<lambda>c s � � ttf| |�S r+ �r r r� �rh r� r: r r r� y s �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� r4 r� r �r; r �g�
assertInvalidr )r4 r� r; r �testIPv4toStrings s.
z#GeneralModuleTests.testIPv4toStringc
s� z&ddl m�m� m} |s$��d� W n tk
rD ��d� Y nX tjdkr�z�� d� W n6 tk
r� } z|j dkr���d� W 5 d }~X Y nX � �fd d
����fd d
�}��
d �d�� ��
d �d
�� ��
d�d�� ��
d�d�� ��
d�d�� ��
d�d�� |d� |d� |d� |d� |d� |d� |d � |d � t �s`|d � ��
d �d �� ��
d!�d"�� ��
d#�d$�� |d%� |d&� |d'� |d(� |d)� |d*� d S )+Nr )r� r� r* �IPv6 not available�+could not import needed symbols from socket�win32�::�&' �IPv6 might not be supportedc s
�� | �S r+ r r� )r� r� r r r� � r� z5GeneralModuleTests.testIPv6toString.<locals>.<lambda>c s �� ttf� | �S r+ r� r� �r r; r r r� � s �� z0::0s z1::s E�v� V�� �$��z$45ef:76cb:1a:56ef:afeb:bac:1924:aeaes �B
� ' T zad42:abc::127:0:254:2s
z12:a::z0x20::z:::z::0::z1::abc::z 1::abc::defz 1:2:3:4:5:6z 1:2:3:4:5:6:z1:2:3:4:5:6:7:8:0z1:2:3:4:5:6:7:8:s �*@z::254.42.23.64s B ���*@z42::a29b:254.42.23.64s B�� �����*@z"42:a8b9:0:2:ffff:a29b:254.42.23.64z255.254.253.252z 1::260.2.3.0z 1::0.be.e.0z1:2:3:4:5:6:7:1.2.3.4z ::1.2.3.4:0z0.100.200.0:3:4:5:6:7:8) r r� r� r* rZ � ImportErrorr[ rz r �winerrorr� r �r; r* r� r� r )r� r r� r; r �testIPv6toString� sj
�����z#GeneralModuleTests.testIPv6toString� inet_ntopz test needs socket.inet_ntop()c s ddl m}m�m� � �fdd�}�fdd�}��d|d�� ��d|d �� ��d
|d �� ��d |d
�� ||d� ||d� ||d� ��d|td ��� ��d|d�� ��d|d �� ��d
|d �� ||d� ||d� ||d� ��d|td ��� d S )Nr )� inet_ntoar� r4 c s
�� | �S r+ r r� )r4 r� r r r� � r� z5GeneralModuleTests.testStringToIPv4.<locals>.<lambda>c s � � ttf| |�S r+ r� r� r: r r r� � s �z1.0.1.0s z
170.85.170.85s �U�Ur@ r� r� r� s s r� )r r� r� r4 r� � bytearrayr� r )r4 r� r; r �testStringToIPv4� s$
z#GeneralModuleTests.testStringToIPv4c
s z&ddl m�m� m} |s$��d� W n tk
rD ��d� Y nX tjdkr�z�� d� W n6 tk
r� } z|j dkr���d� W 5 d }~X Y nX � �fd d
����fd d
�}��
d �d�� ��
d
�d�� ��
d�d�� ��
d
�t d��� |d� |d� |d� d S )Nr )r� r� r* r� r� r� r� r� r� c s
�� | �S r+ r r� )r� r� r r r� � r� z5GeneralModuleTests.testStringToIPv6.<locals>.<lambda>c s �� ttf� | �S r+ r� r� r� r r r� s �r� r� s z!aef:b01:506:1001:ffff:9997:55:170s
� ���� Ups s s ) r r� r� r* rZ r� r[ rz r r� r� r� r� r )r� r r� r; r �testStringToIPv6� s0
�z#GeneralModuleTests.testStringToIPv6c C s� t �� }t�tjtj�}| �|j� |�d|f� |�� }zt� t�
� �}W n t k
rj | � d� Y nX | �
|d d|fd|d � | �|d |� d S )Nr� r6 r z
%s invalidri )r �find_unused_portr r4 r5 rO r rX r� r8 r7 r rZ r r� )r; r8 r� r� Z
my_ip_addrr r r � testSockName s z GeneralModuleTests.testSockNamec C s@ t � t jt j�}| �|j� |�t jt j�}| �|dkd� d S )Nr zinitial mode is reuse) r r4 r5 rO r �
getsockoptr, r- � assertFalse�r; r� Zreuser r r �testGetSockOpt$ s z!GeneralModuleTests.testGetSockOptc C sR t � t jt j�}| �|j� |�t jt jd� |�t jt j�}| � |dkd� d S )Nri r zfailed to set reuse mode)
r r4 r5 rO r �
setsockoptr, r- r� r� r� r r r �testSetSockOpt, s
z!GeneralModuleTests.testSetSockOptc C s: t � t jt j��}|�d� W 5 Q R X | �t|jd� d S )Nri � spam)r r4 r5 �
settimeoutr r r� r� r r r �testSendAfterClose4 s z%GeneralModuleTests.testSendAfterClosec
C s� t � � }|�t jdf� t j |�� d��� z |�� W n8 tk
rp } z| �|jtjtj f� W 5 d }~X Y n X | �
d� d S )Nr �r z#close() should raise EBADF/ENOTSOCK) r rX �
_LOCALHOSTr r r r rE �EBADF�ENOTSOCKr )r; r� r r r r �testCloseException: s (z%GeneralModuleTests.testCloseExceptionc C sv t � t jt j��\}| �|jt j� tt d�rJ| �|jt jt jB t jf� n| �|jt j� | �|j d� W 5 Q R X d S )N� SOCK_CLOEXECr )
r r4 r5 r� r r r r� r� r r� r r r �testNewAttributesF s
��z$GeneralModuleTests.testNewAttributesc C s� t � � }| �|j� t�� }|d }|d }| �t|jt|f� | �t|jt|f� t �
� D ]\}t�� }z|�t|f� W n8 t k
r� } z|j t j
ks�|dkr�� W 5 d }~X Y q\X q�q\d S )Nrm � )r rO r r r� r rU rX r� � itertools�countr rE Z
EADDRINUSE)r; r� r8 Zbig_portZneg_portr2 r� r r r �test_getsockaddrargS s z&GeneralModuleTests.test_getsockaddrargc C s� | � ttjd�� | � ttd�� | � ttd�� | � ttd�� | � ttd�� t�� }| �|j� | �t|jdd � |�tjd� d S )Nr Z
SIO_RCVALLZ RCVALL_ONZ
RCVALL_OFF�SIO_KEEPALIVE_VALSrP )ri �d r� ) r r r rO r r r� r r� r r r r �test_sock_ioctlg s z"GeneralModuleTests.test_sock_ioctl�SIO_LOOPBACK_FAST_PATHz1Loopback fast path support required for this testc
C s| t � � }| �|j� z|�t jd� W n< tk
rb } z d}|j|krP| �d� � W 5 d }~X Y nX | �t |jt jd � d S )NTi=' zQSIO_LOOPBACK_FAST_PATH is defined but doesn't implemented in this Windows version)
r rO r r r� r r� rZ r r� )r; r r Z
WSAEOPNOTSUPPr r r �test_sio_loopback_fast_paths s
z.GeneralModuleTests.test_sio_loopback_fast_pathc
C s@ zt �dd� W n< t jk
rL } z |jt jkr:| �d� � W 5 d }~X Y nX t �td �D ]}| �t|�d� qZt �dd� t �dd� t �d d� t j
r�t �dd� t t d�r�t �
� dkr�t �td � t �td� t �td � t �tdt jt j�}|D ]J\}}}}}| �|t j� | �t|�d
� | �|t j� | �t|�d � q�t �td d t j�}|D ] \}}}}}| �|t j� �qZt �td d d t j� t �td d d d t j� t �d d t jt jd t j� t �td �}t jtd d
�} | �|| � t �td t j�}t jtd t jd�} | �|| � t �td d t j�}t jtd t jd�} | �|| � t �td d d t j�}t jtd t jd�} | �|| � t �td d d d t j�}t jtd t jd�} | �|| � t �d d t jt jd t j�}t jd d t jt jd t jd�} | �|| � | �tt jdd� t t d��r<zt �dd d d d t j� W n t jk
�r: Y nX d S )N� localhost�P zbuggy libc versionr� r
r� rx ry Zhttp�AddressFamily.AF_INET�SocketKind.SOCK_STREAMr )r� r8 �r �r� )r )�flags)r� r8 r r� r r� u ��AI_NUMERICSERV)r � getaddrinfor rE � EAI_SERVICErZ r� r� �lenr � IPV6_ENABLEDr r[ rx r4 r5 r� �SOL_TCP�
AI_PASSIVE� AF_UNSPECr �UnicodeEncodeErrorr� )
r; r �infoZinfosr r� rk �socktyper� �br r r �testGetaddrinfo� s�
� � � � �z"GeneralModuleTests.testGetaddrinfoc C s | � ttjdd� d S )N)zmail.python.orgr r )r r r r\ r: r r r �test_getnameinfo� s z#GeneralModuleTests.test_getnameinfoZnetworkznetwork is not enabledc C sP t �d�� t�d� W 5 Q R X d}t�|� t�|� t�|dtjtj� d S )Nz
python.orgu! испытание.pythontest.netr )r Ztransient_internetr r8 �gethostbyname_exr� r� r5 )r; rw r r r � test_idna� s
z GeneralModuleTests.test_idnac
s� t td�rt td�s � �d� � fdd�}� fdd�}t�� \}}t�tj|�}zv|r`|�d � � � t
�� t�d
� |� d t j
� W 5 Q R X |r�t�tj|� t�d
� � � tj|j d t j
� W 5 t�d� t�tj|� |�� |�� X d S ) N�alarmr� z9signal.alarm and socket.socketpair required for this testc s � � ttjd� d S r� �r r� �math�acosh�rK r: r r �
ok_handler� s z@GeneralModuleTests.check_sendall_interrupted.<locals>.ok_handlerc s � � ttjd� dd d S �Nr ri r� r� r: r r �raising_handler� s zEGeneralModuleTests.check_sendall_interrupted.<locals>.raising_handlerr g �?ri � x)r �signalr rZ r� �SIGALRMr� r r� r �ZeroDivisionError�sendallr �
SOCK_MAX_SIZEr0 )r; Z with_timeoutr� r� �cr � old_alarmr r: r �check_sendall_interrupted� s,
�
z,GeneralModuleTests.check_sendall_interruptedc C s | � d� d S �NF�r� r: r r r �test_sendall_interrupted s z+GeneralModuleTests.test_sendall_interruptedc C s | � d� d S )NTr� r: r r r �%test_sendall_interrupted_with_timeout s z8GeneralModuleTests.test_sendall_interrupted_with_timeoutc C s� t � t jt j�}t|�}| �t��}d }t�� W 5 Q R X | �|t |j
j d �� t � t jt j�}|� d�}t|�}d }t�� | �t�� d }t�� W 5 Q R X d S )Nr r )
r r4 r5 r rp �ResourceWarningr �
gc_collectr r� �warningrK �makefile)r; r� r r( r r r r �test_dealloc_warn s
z$GeneralModuleTests.test_dealloc_warnc C sB t � t jt j��(}|�d�}|�� | �t|�d� W 5 Q R X d S )Nr z <_io.BufferedReader name=-1>)r r4 r5 r� r r� r �r; r� �fpr r r �test_name_closed_socketio s
z,GeneralModuleTests.test_name_closed_socketioc C s� t � � �p}|jddd�}| �|�� � | �|�� � | �|�� � |�� | �t |j� | �t |j� | �t |j� W 5 Q R X d S )Nr r )� buffering)
r r� r �readabler� �writable�seekabler r r� r� r r r � test_unusable_closed_socketio" s
z0GeneralModuleTests.test_unusable_closed_socketioc C s� t � � }z@|�tdf� t �|�� � | �t�� |�d� W 5 Q R X W 5 | �t�� |�� W 5 Q R X X | �t�� t �d � W 5 Q R X | �t�� t �d� W 5 Q R X d S )Nr ri rP ) r r r r rX r� r r9 r� r� r r r �test_socket_close- s z$GeneralModuleTests.test_socket_closec C s^ dD ]T}| j |d��> t�� �*}|�|��}| �|j|� W 5 Q R X W 5 Q R X W 5 Q R X qd S )N)r r �rw�w�wb��mode)�subTestr r� r� r� )r; r� r� r� r r r �test_makefile_mode= s
z%GeneralModuleTests.test_makefile_modec C s\ dD ]R}| j |d��< t�� �(}| �td�� |�|� W 5 Q R X W 5 Q R X W 5 Q R X qd S )N)�rtrU �+r� r� z invalid mode)r r �assertRaisesRegexr� r� )r; r� r� r r r �test_makefile_invalid_modeD s
z-GeneralModuleTests.test_makefile_invalid_modec C s� t � � }|�, ttjd �D ]}| �ttj||� q W 5 Q R X ttjd �D ]H}t�t�t j|��}| � |t j� t�t�t j
|��}| � |t j
� qLd S r� ) r �range�pickle�HIGHEST_PROTOCOLr r� �dumps�loadsr4 r� r5 )r; r� �protocolr r� r r r � test_pickleK s z GeneralModuleTests.test_picklec
C st dD ]8}t � t jt j�� }|�tdf� |�|� W 5 Q R X qt � t jt j�� }|�tdf� |�� W 5 Q R X d S )N)r rP r )r r4 r5 rX r� r9 )r; �backlog�srvr r r �test_listen_backlogV s z&GeneralModuleTests.test_listen_backlogc C sL dd l }t�tjtj��*}|�tdf� | �t|j|j d � W 5 Q R X d S r� )
rn r r4 r5 rX r� r rU r9 ro )r; rn r r r r � test_listen_backlog_overflow` s z/GeneralModuleTests.test_listen_backlog_overflow� IPv6 required for this test.c C sR | � ttjtjddfd� t�tjtj�� }| � t|jtjddf� W 5 Q R X d S )Nr rT i����) r rU r r\ r r� r� r5 rX r r r r �
test_flowinfoh 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 r0 r� )r; rk �sockaddrr r r �test_getaddrinfo_ipv6_basico s � z.GeneralModuleTests.test_getaddrinfo_ipv6_basicr� zdoes not work on Windowsz Symbolic scope id does not workc C sL t �� d \}}t �d| dt jt jt j�\�^ }}| �|ddd|f� d S )Nr �ff02::1de:c0:face:8D%r r )r rH r� r� r r0 r� )r; �ifindex�test_interfacerk r r r r �&test_getaddrinfo_ipv6_scopeid_symbolicy s � z9GeneralModuleTests.test_getaddrinfo_ipv6_scopeid_symbolicz.Numeric scope id does not work or undocumentedc C sD d}t �dt|� dt jt jt j�\�^ }}| �|ddd|f� d S )N�* r r r r )r r� r� r� r r0 r� )r; r rk r r r r �%test_getaddrinfo_ipv6_scopeid_numeric� s
� z8GeneralModuleTests.test_getaddrinfo_ipv6_scopeid_numericc C sH t �� d \}}ddd|f}t �|t jt jB �}| �|d| df� d S )Nr r r �ff02::1de:c0:face:8d%�1234)r rH r\ �NI_NUMERICHOST�NI_NUMERICSERVr� )r; r r r �nameinfor r r �&test_getnameinfo_ipv6_scopeid_symbolic� s z9GeneralModuleTests.test_getnameinfo_ipv6_scopeid_symbolicc C s@ d}ddd|f}t �|t jt jB �}| �|dt|� df� d S )Nr r r r r r )r r\ r r r� r� )r; r r r! r r r �%test_getnameinfo_ipv6_scopeid_numeric� s z8GeneralModuleTests.test_getnameinfo_ipv6_scopeid_numericc C sD t � t jt j��*}| �t|j�d� | �t|j�d� W 5 Q R X d S )Nr� r� )r r4 r5 r� r� r r� r r r r �test_str_for_enums� s z%GeneralModuleTests.test_str_for_enumsc C s� t tdd�}t tdd�}tj|B |B }t�tj|��~}| �|jtj� |�d� | �|jtj� |�d� | �|jtj� |�d� | �|jtj� |�d� | �|jtj� W 5 Q R X d S )N�
SOCK_NONBLOCKr r� ri TF)rv r r5 r4 r� r� r� � setblocking)r; 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��4}| �
|j |� | �
|j |� | �
|jddh� W 5 Q R X d S )Nri c s s | ]\}}|d kr|V qdS )> r% r� Nr )r� r� �kindr r r r� � s �zEGeneralModuleTests.test_unknown_socket_family_repr.<locals>.<genexpr>ry )r r� r r r )r r4 r5 �detach�max�
AddressFamily� __members__�values�
SocketKindrg r� r r� r r )r; r� �fdZunknown_familyZ unknown_typer r r r � test_unknown_socket_family_repr� s$
�� �z2GeneralModuleTests.test_unknown_socket_family_repr�sendfileztest needs os.sendfile()c
C s� G dd� d�}t � � ��}t�tjtj�}t�|� | �t j�� |�||�� W 5 Q R X | �t �� |�|dd �� W 5 Q R X | �t
�� |�|d �� W 5 Q R X W 5 Q R X d S )Nc @ s e Zd Zdd� Zdd� ZdS )z<GeneralModuleTests.test__sendfile_use_sendfile.<locals>.Filec S s
|| _ d S r+ �r/ )r; r/ r r r rF � s zEGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.__init__c S s | j S r+ r2 r: r r r r � s zCGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.filenoN)r@ rA rB rF r r r r r �File� s r3 r� rQ ) r r� r
�curdir�O_RDONLYr r �_GiveupOnSendfile�_sendfile_use_sendfilerU r� )r; r3 r� r/ r r r �test__sendfile_use_sendfile� s
z.GeneralModuleTests.test__sendfile_use_sendfilec C st | � |j|� | � |j|� |�� }tj|d�}| �|j� |�� | � |j|� | � |j|� | � |�� |� d S �Nr� )r� r r� r r rO r r) )r; r r �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�r�t � t jt j
�}| �|j� |�tjdf� | �|t jt j
� tj r�t � t j t j�}| �|j� |�tj
dddf� | �|t j t j� t t d��rNt�� }| �tj|� t � t jt j�}| �|j� z|�tj�|d�� W n tk
�r: Y nX | �|t jt j� d S )Nr r r� r )r r4 r5 rO r rX r r� r< r r r� r� r� r� r� �shutil�rmtreer� r� r� r� �PermissionError)r; r Ztmpdirr r r �test_socket_fileno� s4
�z%GeneralModuleTests.test_socket_filenoc C s0 | � td�� tjtjtjdd� W 5 Q R X d S )Nzinteger argument expectedg @E@r� �r r� r r4 r5 r: r r r � test_socket_fileno_rejects_float s z3GeneralModuleTests.test_socket_fileno_rejects_floatc C s0 | � td�� tjtjtjdd� W 5 Q R X d S )Nzinteger is required�foor� rA r: r r r �&test_socket_fileno_rejects_other_types s z9GeneralModuleTests.test_socket_fileno_rejects_other_typesc C s0 | � td�� tjtjtjdd� W 5 Q R X d S )N�negative file descriptorrP r� �r r� r r4 r5 r: r r r �)test_socket_fileno_rejects_invalid_socket$ s z<GeneralModuleTests.test_socket_fileno_rejects_invalid_socketzWindows disallows -1 onlyc C s0 | � td�� tjtjtjdd� W 5 Q R X d S )NrE i���r� rF r: r r r �#test_socket_fileno_rejects_negative( s z6GeneralModuleTests.test_socket_fileno_rejects_negativec C s� d}| � t��}tjt�� d� W 5 Q R X | �|jjtj|f� | � t�� }tjtj tj
t�� d� W 5 Q R X | �|jjtj|f� d S )Ni6' r� ) r r r r Z make_bad_fdr r' rE r� r4 r5 )r; Z WSAENOTSOCKr( r r r �$test_socket_fileno_requires_valid_fd- s �z7GeneralModuleTests.test_socket_fileno_requires_valid_fdc
C s~ t �� �l}| �t�� tj|�� d� W 5 Q R X | �t�� }tjtjtj|�� d� W 5 Q R X | �|j j
t
j � W 5 Q R X d S r9 ) r� �NamedTemporaryFiler r r r r4 r5 r� r' rE r� )r; Zafiler( r r r �%test_socket_fileno_requires_socket_fd: s
�z8GeneralModuleTests.test_socket_fileno_requires_socket_fd)Wr@ rA rB r
r r� �
skipUnlessr r r r! r) r. r2 r� r� r5 r? rA rB r r rG rO rX rY r[ r^ r_ rk r � cpython_onlyrs r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� Zis_resource_enabledr� r� r� r� r� r� r� r� r r r r r r� r r �skipIfrz r r r r" r# r$ r' r0 r8 r< r@ rB rD rG rH rI rK 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 r+ )r �AF_CANr r r: r r r r. J s z!BasicCANTest.testCrucialConstants�CAN_BCM�&socket.CAN_BCM required for this test.c C s� t j t j t j t j t j t j t j t j t j t j
t j t j t j
t j t j t j t j t j t j t j t j t j t j t j d S r+ )r rQ �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_FRAMEr: r r r �testBCMConstantsO s0 z BasicCANTest.testBCMConstantsc C s$ t � t jt jt j��}W 5 Q R X d S r+ �r r r r r r r r �testCreateSocketo s z BasicCANTest.testCreateSocketc C s$ t � t jt jt j��}W 5 Q R X d S r+ )r r r rQ r r r r �testCreateBCMSockets s z BasicCANTest.testCreateBCMSocketc C sB t � t jt jt j��$}d}|�|� | �|�� |� W 5 Q R X d S )N)r" )r r r r rX r� r� )r; r �addressr r r � testBindAnyy s
zBasicCANTest.testBindAnyc C s6 t � t jt jt j��}| �td|jd� W 5 Q R X d S )N�interface name too long)� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx)r r r r r r rX r r r r �testTooLongInterfaceName s
�z%BasicCANTest.testTooLongInterfaceName�CAN_RAW_LOOPBACKz/socket.CAN_RAW_LOOPBACK required for this test.c C sX t � t jt jt j��:}dD ].}|�t jt j|� | �||�t jt j�� qW 5 Q R X d S )N)r ri ) r r r r r� � SOL_CAN_RAWrs r� r� )r; r Zloopbackr r r � testLoopback� s ��zBasicCANTest.testLoopback�CAN_RAW_FILTERz-socket.CAN_RAW_FILTER required for this test.c C s| d\}}t �d||�}t�tjtjtj��H}|�tjtj|� | � ||�
tjtjd�� |�tjtjt |�� W 5 Q R X d S )N)� i z=IIrV ) r �packr r r r r� rt rv r� r� r� )r; �can_idZcan_maskZ
can_filterr r r r �
testFilter� s �zBasicCANTest.testFilterN)r@ rA rB r. r� rL r r rj rl rm ro rr ru rz r r r r rO G s( �
�
�
�rO c @ s� e Zd Zddd�Zedd� �Zedd� �Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z e
�eed�d�dd� �Ze
�eed�d�dd� �ZdS ) �CANTestr� c C s t j| |d� d S r� )r� rF r� r r r rF � s zCANTest.__init__c C s& t |�}|�dd�}t�| j|||�S )NrV � )r� �ljustr rx r\ )�clsry �data�can_dlcr r r �build_can_frame� s zCANTest.build_can_framec C s& t �| j|�\}}}|||d |� fS r+ )r r r\ )r~ �framery r� r r r r �dissect_can_frame� s zCANTest.dissect_can_framec C s6 | j �| j�\}}| �| j|� | �|d | j� d S r� )r �recvfromr[ r� �cfrY �r; r� r� r r r �
testSendFrame� s zCANTest.testSendFramec C s | � dd�| _| j�| j� d S )Nr s �r� r� r� r� r: r r r �_testSendFrame� s zCANTest._testSendFramec C s$ | j �| j�\}}| �| j|� d S r+ )r r� r[ r� r� r� r r r �testSendMaxFrame� s zCANTest.testSendMaxFramec C s | � dd�| _| j�| j� d S )Nr s r� r: r r r �_testSendMaxFrame� s zCANTest._testSendMaxFramec C sD | j �| j�\}}| �| j|� | j �| j�\}}| �| j|� d S r+ )r r� r[ r� �cf1�cf2r� r r r �testSendMultiFrames� s zCANTest.testSendMultiFramesc C s< | � dd�| _| j�| j� | � dd�| _| j�| j� d S )N� s D3"� s �"3)r� r� r� r� r� r: r r r �_testSendMultiFrames� s z CANTest._testSendMultiFramesrQ rR c C sP | j �| j�\}}| �| j|� | �|�\}}}| �| j|� | �| j|� d S r+ )r� r� r[ r� r� r� ry r )r; r� r� ry r� r r r r �_testBCM� s
zCANTest._testBCMc C s� t � t jt jt j�}| �|j� |�| jf� d| _t dddg�| _
| � | j| j
�| _ t j
}d}d}d } } }}d} d}
t�| j|||||||| |
�
} | | j } |�| �}
| �|
t| �� d S )N�# � � �� r i" ri )r r r rQ rO r r� rY ry �bytesr r� r� rV r rx r^ r� r� r� )r; ZbcmZopcoder� r� Z
ival1_secondsZ
ival1_usecZ
ival2_secondsZ
ival2_usecZ
bcm_can_idZnframes�headerZheader_plus_frameZ
bytes_sentr r r �testBCM� s4 �
zCANTest.testBCMN)r� )r@ rA rB rF � classmethodr� r� r� r� r� r� r� r� r� rL r r r� r� r r r r r{ � s&
�
�r{ z!CAN ISOTP required for this test.c sV e Zd Z� fdd�Zdd� Zdd� Ze�ee d�d�d d
� �Z
d d � Z d
d� Z � Z
S )� ISOTPTestc s t � j||� d| _d S )NrS )rE rF rY rJ rM r r rF � s zISOTPTest.__init__c C s t j t j t j t j d S r+ )r rP r r! r r: r r r r. � s z ISOTPTest.testCrucialConstantsc C s$ t � t jt jt j��}W 5 Q R X d S r+ rk r r r r rl s zISOTPTest.testCreateSocketr! z(socket.CAN_ISOTP required for this test.c C s$ t � t jt jt j��}W 5 Q R X d S r+ )r r r r! r r r r �testCreateISOTPSocket s z ISOTPTest.testCreateISOTPSocketc
C sF t � t jt jt j��(}| �td�� |�d� W 5 Q R X W 5 Q R X d S )Nrp )rq ri r� )r r r r! r r rX r r r r rr
s z"ISOTPTest.testTooLongInterfaceNamec
C s� zJt � t jt jt j��,}| jddf}|�|� | �|�� |� W 5 Q R X W nB tk
r� } z$|j t j
krz| � d| j � n� W 5 d }~X Y nX d S )Nr� iV rW ) r r r r! rY rX r� r� r rE ZENODEVrZ )r; r r� r� r r r �testBind s
�zISOTPTest.testBind)r@ rA rB rF r. rl r� rL r r r� rr r� rQ r r rM 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 r+ )r �AF_RDSr$ r: r r r r. $ s z!BasicRDSTest.testCrucialConstantsc C s" t � t jt jd��}W 5 Q R X d S r� )r r$ r% r r r r rl ( s z BasicRDSTest.testCreateSocketc C sJ d}t � t jt jd��*}|�t jt j|� |�t jt j|� W 5 Q R X d S )Ni @ r )r r$ r% r� r, � SO_RCVBUF� SO_SNDBUF)r; r[ r r r r �testSocketBufferSize, s z!BasicRDSTest.testSocketBufferSizeN)r@ rA rB r. rl r� r r r r r� ! s r� c s� e Zd Zd dd�Z� fdd�Zdd� Zdd � Zd
d � Zd d
� Ze e
j
d�dd� �Z e e
j
d�dd� �Z dd� Z
dd� Zdd� Zdd� Z� ZS ) �RDSTestr� c C s t j| |d� d S r� )r� rF r� r r r rF 6 s zRDSTest.__init__c s t � �� t�� | _d S r+ )rE r< rG rn Zevtr: rM r r r< 9 s
z
RDSTest.setUpc C s2 | j �| j�\}}| �| j|� | �| j|� d S r+ )r6 r� r[ r� r r� �r; r r� r r r �testSendAndRecv= s zRDSTest.testSendAndRecvc C s" d| _ | j�| j dt| jf� d S �Nr� r �r r� r& r� r8 r: r r r �_testSendAndRecvB s zRDSTest._testSendAndRecvc C sH | j �| jtj�\}}| �| j|� | j �| j�\}}| �| j|� d S r+ )r6 r� r[ r �MSG_PEEKr� r r� r r r �testPeekF s zRDSTest.testPeekc C s" d| _ | j�| j dt| jf� d S r� r� r: r r r � _testPeekL s zRDSTest._testPeek�recvmsgc C s( | j �| j�\}}}}| �| j|� d S r+ )r6 r� r[ r� r )r; r �ancdataZ msg_flagsr� r r r �testSendAndRecvMsgP 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� r8 r: r r r �_testSendAndRecvMsgU s zRDSTest._testSendAndRecvMsgc C sD | j �| j�\}}| �| j|� | j �| j�\}}| �| j|� d S r+ )r6 r� r[ r� �data1�data2r� r r r �testSendAndRecvMultiZ s z RDSTest.testSendAndRecvMultic C s@ d| _ | j�| j dt| jf� d| _| j�| jdt| jf� d S )Ns baconr s egg)r� r� r& r� r8 r� r: r r r �_testSendAndRecvMultia s z RDSTest._testSendAndRecvMultic C sL t � | jgg g d�\}}}| �| j|� | j�| j�\}}| �| j|� d S )N� @)�selectr6 r r� r[ r� r )r; r r� rU r r� r r r �
testSelecth s zRDSTest.testSelectc C s" d| _ | j�| j dt| jf� d S )Ns selectr r� r: r r r � _testSelectn s zRDSTest._testSelect)r� )r@ rA rB rF r< r� r� r� r� r� r r� r� r� r� r� r� rQ r r rM r r� 3 s
r� z'QIPCRTR sockets required for this test.c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�BasicQIPCRTRTestc C s
t j d S r+ )r r) r: r r r r. v s z%BasicQIPCRTRTest.testCrucialConstantsc C s t � t jt j��}W 5 Q R X d S r+ )r r) r r r r r rl y s z!BasicQIPCRTRTest.testCreateSocketc C s4 t � t jt j��}| �|�� d d� W 5 Q R X d S �Nri r )r r) r r� r� r r r r � testUnbound} s z BasicQIPCRTRTest.testUnboundc C sJ t � t jt j��0}tj||�� d d� | �|�� d d� W 5 Q R X d S )Nr r� ri )r r) r r r7 r� �assertNotEqualr r r r � testBindSock� s z BasicQIPCRTRTest.testBindSockc C s4 t � t jt j��}| jttj|dd� W 5 Q R X d S )Nrl r� )r r) r r r r r7 r r r r �testInvalidBindSock� s z$BasicQIPCRTRTest.testInvalidBindSockc C s> t � t jt j��$}|�d� | �|�� d d� W 5 Q R X d S )N)�{ r� ri r )r r) r r� r� r� r r r r �testAutoBindSock� s
z!BasicQIPCRTRTest.testAutoBindSockN) r@ rA rB r. rl r� r� r� r� r r r r r� r s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�BasicVSOCKTestc C s
t j d S r+ )r r� r: r r r r. � s z#BasicVSOCKTest.testCrucialConstantsc C s4 t j t j t j t j t j t j t j t j d S r+ ) 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 r: r r r �testVSOCKConstants� s z!BasicVSOCKTest.testVSOCKConstantsc C s t � t jt j��}W 5 Q R X d S r+ )r r� r5 r r r r rl � s z BasicVSOCKTest.testCreateSocketc C s� t � t jt j���}|�t jt j�}|�t jt j�}|�t jt j�}|�t jt j|d � |�t jt j|d � |�t jt j|d � | �|d |�t jt j�� | �|d |�t jt j�� | �|d |�t jt j�� W 5 Q R X d S �Nr� ) r r� r5 r� r� r� r� r� r� )r; r Zorig_maxr] Zorig_minr r r r� � sJ ��� � � �
��
��
��z#BasicVSOCKTest.testSocketBufferSizeN)r@ rA rB r. r� rl r� r r r r r� � s
r� c @ s� e Zd Zd+dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Ze�e�Zejd$d%� �Zd&d'� Zd(d)� Zd*S ),� BasicTCPTestr� c C s t j| |d� d S r� �r� rF r� r r r rF � s zBasicTCPTest.__init__c C s | j �d�}| �|t� d S r� )r� r� r� r� r� r r r �testRecv� s zBasicTCPTest.testRecvc C s | j �t� d S r+ �r� r� r� r: r r r � _testRecv� s zBasicTCPTest._testRecvc C s8 | j �tt�d �}| j �d�}|| }| �|t� d S �N� r� )r� r� r� r� r� )r; �seg1�seg2r� r r r �testOverFlowRecv� s z BasicTCPTest.testOverFlowRecvc C s | j �t� d S r+ r� r: r r r �_testOverFlowRecv� s z BasicTCPTest._testOverFlowRecvc C s | j �d�\}}| �|t� d S r� )r� r� r� r� �r; r� r� r r r � testRecvFrom� s zBasicTCPTest.testRecvFromc C s | j �t� d S r+ r� r: r r r �
_testRecvFrom� s zBasicTCPTest._testRecvFromc C s@ | j �tt�d �\}}| j �d�\}}|| }| �|t� d S r� )r� r� r� r� r� )r; r� r� r� r� r r r �testOverFlowRecvFrom� s z!BasicTCPTest.testOverFlowRecvFromc C s | j �t� d S r+ r� r: r r r �_testOverFlowRecvFrom� s z"BasicTCPTest._testOverFlowRecvFromc C s0 d}| j �d�}|sq ||7 }q| �|d� d S )Nr� r� � ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff)r� r� r� )r; r� �readr r r � testSendAll� s
zBasicTCPTest.testSendAllc C s d}| j �|� d S )Nr� )r� r� )r; Z big_chunkr r r � _testSendAll� s zBasicTCPTest._testSendAllc C sP | j �� }t�|tjtj�}| �|j� | �|tj� |� d�}| �
|t � d S r� ) r� r r �fromfdr4 r5 rO r r r� r� r� )r; r/ r� r� r r r �
testFromFd� s
zBasicTCPTest.testFromFdc C s | j �t� d S r+ r� r: r r r � _testFromFd s zBasicTCPTest._testFromFdc C s0 | j �� }| �|j� |�d�}| �|t� d S r� )r� �duprO r r� r� r� )r; r� r� r r r �testDup s
zBasicTCPTest.testDupc C s | j �t� d S r+ r� r: r r r �_testDup
s zBasicTCPTest._testDupc C s& | j �d�}| �|t� | j�� d S r� )r� r� r� r� rp rz r� r r r � testShutdown
s zBasicTCPTest.testShutdownc C s | j �t� | j �d� d S r� )r� r� r� �shutdownr: r r r �
_testShutdown s zBasicTCPTest._testShutdownc C sX dd l }| j�t� | �t| jj|jd � | �t| jjd|jd � | j�d� d S )Nr ri r� ) rn r� r� r� r rU r� ro �UINT_MAX�r; rn r r r �_testShutdown_overflow s � �z#BasicTCPTest._testShutdown_overflowc C s� | j �� }| j �� }| �||� | �| j j� | �t| j jd� | j � � t
j
t
j t
j |d�}| �
|j � |�d�}| �|t� d S )Nr� r� )r� r r) r� r �_closedr r r� r r r4 r5 rO r� )r; r r r� r� r r r �
testDetach' s
zBasicTCPTest.testDetachc C s | j �t� d S r+ r� r: r r r � _testDetach7 s zBasicTCPTest._testDetachN)r� )r@ rA rB rF r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r rM ZtestShutdown_overflowr� r� r� r r r r r� � s,
r� c @ sF e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dS )� BasicUDPTestr� c C s t j| |d� d S r� )r� rF r� r r r rF = s zBasicUDPTest.__init__c C s | j �tt��}| �|t� d S r+ )r6 r� r� r� r� r� r r r �testSendtoAndRecv@ s z BasicUDPTest.testSendtoAndRecvc C s | j �tdt| jf� d S r� �r� r& r� r� r8 r: r r r �_testSendtoAndRecvE s z BasicUDPTest._testSendtoAndRecvc C s$ | j �tt��\}}| �|t� d S r+ )r6 r� r� r� r� r� r r r r� H s zBasicUDPTest.testRecvFromc C s | j �tdt| jf� d S r� r� r: r r r r� M s zBasicUDPTest._testRecvFromc C s | � t| jjd� d S �NrP )r r� r6 r� r: r r r �testRecvFromNegativeP s z!BasicUDPTest.testRecvFromNegativec C s | j �tdt| jf� d S r� r� r: r r r �_testRecvFromNegativeT s z"BasicUDPTest._testRecvFromNegativeN)r� )
r@ rA rB rF r� r� r� r� r� r� r r r r r� ; s
r� c s� e Zd ZdZ� fdd�Zdd� ZdZdd� Zd d
� Zd d � Z d
d� Z
dZ dD ]Z e e
ee d�O Z qLdZdZdZddd�Z� ZS )�SendrecvmsgBaser� c s t �� | _t� �� d S r+ )rG rn �
misc_eventrE r< r: rM r r r< y s
zSendrecvmsgBase.setUpc C s | j �|�S r+ )�cli_sockr� r� r r r � sendToServer} s z SendrecvmsgBase.sendToServerr c G s | j j|| jt|�d � � S r+ )r� r� �sendmsg_to_server_defaultsr� �r; rK r r r �sendmsgToServer� s �z SendrecvmsgBase.sendmsgToServerc G s |j |f|�� }| �|� |S r+ )r� �registerRecvmsgResult)r; r� r[ rK �resultr r r � doRecvmsg� s
zSendrecvmsgBase.doRecvmsgc C s d S r+ r �r; r� r r r r� � s z%SendrecvmsgBase.registerRecvmsgResultc C s | � ||� d S r+ �r� �r; Zaddr1Zaddr2r r r �checkRecvmsgAddress� s z#SendrecvmsgBase.checkRecvmsgAddressr )�
MSG_CTRUNC�MSG_OOBNc
C s� | j }| j}|r&|| jO }|| jO }n |d k rB|| jO }|| jO }|| M }|| M }||O }||O }||@ | @ }|r�td�|���||B | @ } | �|| @ || @ � d S )Nz6contradictory set, unset requirements for flags {0:#x})�msg_flags_common_set�msg_flags_common_unset�msg_flags_eor_indicator�msg_flags_non_eor_indicator� Exceptionr r� )
r; r� �eor�checkset�
checkunset�ignoreZ
defaultsetZ defaultunsetZinbothrj r r r �
checkFlags� s&
�zSendrecvmsgBase.checkFlags)Nr r r )r@ rA rB � fail_timeoutr< r� r� r� r� r� r r r� rv r r r r r
rQ r r rM r r� r s
r� c @ s e Zd Zdd� ZdS )�RecvmsgIntoMixinc G sf t |�}|j|gf|�� }| �|� | �|d d� | �|d |� t|d |d � �f|dd � S r� )r� � recvmsg_intor� �assertGreaterEqual�assertLessEqualr� )r; r� r[ rK �bufr� r r r r� � s
zRecvmsgIntoMixin.doRecvmsgN)r@ rA rB r� r r r r r � s r c s e Zd Ze� fdd��Z� ZS )�SendrecvmsgDgramFlagsBasec s t � jtjB S r+ )rE r r � MSG_TRUNCr: rM r r r � s z5SendrecvmsgDgramFlagsBase.msg_flags_non_eor_indicator)r@ rA rB �propertyr rQ r r rM r r � s r c s e Zd Ze� fdd��Z� ZS )�SendrecvmsgSCTPFlagsBasec s t � jtjB S r+ )rE r r �MSG_EORr: rM r r r
s z0SendrecvmsgSCTPFlagsBase.msg_flags_eor_indicator)r@ rA rB r r rQ r r rM r r � s r c @ s8 e Zd Zedd� �Zedd� �Zedd� �Zdd� Zd S )
� SendrecvmsgConnectionlessBasec C s | j S r+ )r6 r: r r r � serv_sock
s z'SendrecvmsgConnectionlessBase.serv_sockc C s | j S r+ )r� r: r r r r�
s z&SendrecvmsgConnectionlessBase.cli_sockc C s g g d| j fS r� )r� r: r r r r�
s z8SendrecvmsgConnectionlessBase.sendmsg_to_server_defaultsc C s | j �|| j�S r+ )r� r& r� r� r r r r�
s z*SendrecvmsgConnectionlessBase.sendToServerN)r@ rA rB r r r� r� r� r r r r r
s
r c @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )�SendrecvmsgConnectedBasec C s | j S r+ )r� r: r r r r
s z"SendrecvmsgConnectedBase.serv_sockc C s | j S r+ )r� r: r r r r� $
s z!SendrecvmsgConnectedBase.cli_sockc C s d S r+ r r r r r r (
s z,SendrecvmsgConnectedBase.checkRecvmsgAddressN)r@ rA rB r r r� r r r r r r
s
r c s e Zd Z� fdd�Z� ZS )� SendrecvmsgServerTimeoutBasec s t � �� | j�| j� d S r+ )rE r< r r� r r: rM r r r< 1
s
z"SendrecvmsgServerTimeoutBase.setUpr� r r rM r r .
s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zeed �d d � �Zejd d!� �Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*S )+� SendmsgTestsc C s | � | j�tt��t� d S r+ �r� r r� r� r� r: r r r � testSendmsg:
s zSendmsgTests.testSendmsgc C s | � | �tg�tt�� d S r+ �r� r� r� r� r: r r r � _testSendmsg>
s zSendmsgTests._testSendmsgc C s | � | j�tt��t� d S r+ r r: r r r �testSendmsgDataGeneratorA
s z%SendmsgTests.testSendmsgDataGeneratorc C s& | � | �dd� tfD ��tt�� d S )Nc s s | ]
}|V qd S r+ r r r r r r� F
s z9SendmsgTests._testSendmsgDataGenerator.<locals>.<genexpr>r r: r r r �_testSendmsgDataGeneratorE
s �z&SendmsgTests._testSendmsgDataGeneratorc C s | � | j�tt��t� d S r+ r r: r r r � testSendmsgAncillaryGeneratorI
s z*SendmsgTests.testSendmsgAncillaryGeneratorc C s( | � | �tgdd� dD ��tt�� d S )Nc s s | ]
}|V qd S r+ r r r r r r� N
s z>SendmsgTests._testSendmsgAncillaryGenerator.<locals>.<genexpr>r r r: r r r � _testSendmsgAncillaryGeneratorM
s �z+SendmsgTests._testSendmsgAncillaryGeneratorc C s | � | j�tt��t� d S r+ r r: r r r �testSendmsgArrayQ
s z SendmsgTests.testSendmsgArrayc C s$ | � | �t�dt�g�tt�� d S )N�B)r� r� �arrayr� r� r: r r r �_testSendmsgArrayU
s �z SendmsgTests._testSendmsgArrayc C s | � | j�tt��t� d S r+ r r: r r r �testSendmsgGatherY
s z SendmsgTests.testSendmsgGatherc C s. | � | �td d� tdd � g�tt�� d S �Nr� r r: r r r �_testSendmsgGather]
s z SendmsgTests._testSendmsgGatherc C s | � | j�d�d� d S �NrQ � done�r� r r� r: r r r �testSendmsgBadArgs`
s z SendmsgTests.testSendmsgBadArgsc C s� | � t| jj� | � t| jd� | � t| jt� � | � t| jt� g� | � t| jtt� g� | � t| jtgt� � | � t| jtgg t� � | � t| jtgg dt� � | �d� d S )Ns not in an iterabler r. )r r� r� r� r� �objectr� r� r: r r r �_testSendmsgBadArgsd
s:
�
�
�
�
�
�
�z SendmsgTests._testSendmsgBadArgsc C s | � | j�d�d� d S r- r/ r: r r r �testSendmsgBadCmsgv
s z SendmsgTests.testSendmsgBadCmsgc C s� | � t| jtgt� g� | � t| jtgt� ddfg� | � t| jtgdt� dfg� | � t| jtgddt� fg� | � t| jtgdg� | � t| jtgdg� | �d� d S )Nr � data)r r )r r r4 r r. �r r� r� r� r1 r� r: r r r �_testSendmsgBadCmsgz
s2
�
�
�
�
�
�z SendmsgTests._testSendmsgBadCmsg�
CMSG_SPACEc C s | � | j�d�d� d S r- r/ r: r r r �testSendmsgBadMultiCmsg�
s z$SendmsgTests.testSendmsgBadMultiCmsgc C sB | � t| jtgdddg� | � t| jtgdt� g� | �d� d S )Nr r� �r r r� r. r5 r: r r r �_testSendmsgBadMultiCmsg�
s
�
�z%SendmsgTests._testSendmsgBadMultiCmsgc C s | � | j�d�d� d S r- r/ r: r r r �testSendmsgExcessCmsgReject�
s z(SendmsgTests.testSendmsgExcessCmsgRejectc C sN t td�s@| �t��}| �tgddg� W 5 Q R X | �|jj� | � d� d S )Nr7 r9 r. )
r r r r r� r� � assertIsNoner' rE r� �r; r( r r r � _testSendmsgExcessCmsgReject�
s
z)SendmsgTests._testSendmsgExcessCmsgRejectc C s d S r+ r r: r r r �testSendmsgAfterClose�
s z"SendmsgTests.testSendmsgAfterClosec C s | j �� | �t| jtg� d S r+ )r� r r r r� r� r: r r r �_testSendmsgAfterClose�
s
z#SendmsgTests._testSendmsgAfterCloseN)r@ rA rB r r! r" r# r$ r% r& r) r* r, r0 r2 r3 r6 r� r r8 r� r: r; r> r? r@ r r r r r 6
s,
r c @ sR e Zd Zdd� Zdd� Zdd� Zdd� Zeej d kd
�d d � �Z
e
j d
d� �Z dS )�SendmsgStreamTestsc C s | � | j�tt��t� d S r+ r r: r r r �testSendmsgExplicitNoneAddr�
s z.SendmsgStreamTests.testSendmsgExplicitNoneAddrc C s" | � | �tgg dd �tt�� d S r� r r: r r r � _testSendmsgExplicitNoneAddr�
s z/SendmsgStreamTests._testSendmsgExplicitNoneAddrc C s. | � | j�d�d� | �| jj| jd�� d S �Nrw � aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa�r0 �r� r r� r r� rz r r: r r r �testSendmsgTimeout�
s z%SendmsgStreamTests.testSendmsgTimeoutc
C s� zr| j�d� z| �dg� qW nD tjk
r6 Y n: tk
rd } z|jtj krT� W 5 d }~X Y n X | �
d� W 5 | j �� X d S )N� ��Q��?rE �socket.timeout not raised) r� rg r� r� r� r r0 r rE �ENOMEMr r~ r r r �_testSendmsgTimeout�
s z&SendmsgStreamTests._testSendmsgTimeout> rt z<MSG_DONTWAIT not known to work on this platform when sendingc C s. | � | j�d�d� | �| jj| jd�� d S rD rG r: r r r �testSendmsgDontWait�
s z&SendmsgStreamTests.testSendmsgDontWaitc C s\ zJ| �t��}| �dgg tj� qW 5 Q R X | �|jj t j
t j t j f� W 5 | j �� X d S �NrE )
r� rg r r r� r � MSG_DONTWAITr r' rE �EAGAIN� EWOULDBLOCKrK r= r r r �_testSendmsgDontWait�
s
�z'SendmsgStreamTests._testSendmsgDontWaitN)
r@ rA rB rB rC rH rL r� r[ rz rM r� rR r r r r rA �
s
�
rA c @ s e Zd Zdd� Zdd� ZdS )�SendmsgConnectionlessTestsc C s d S r+ r r: r r r �testSendmsgNoDestAddr�
s z0SendmsgConnectionlessTests.testSendmsgNoDestAddrc C s2 | � t| jjtg� | � t| jjtgg dd � d S r� )r r r� r� r� r: r r r �_testSendmsgNoDestAddr�
s � �z1SendmsgConnectionlessTests._testSendmsgNoDestAddrN)r@ rA rB rT rU r r r r rS �
s rS 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
�r; r� r� r� r� r r r � testRecvmsg�
s
z RecvmsgGenericTests.testRecvmsgc C s | � t� d S r+ �r� r� r: r r r � _testRecvmsg s z RecvmsgGenericTests._testRecvmsgc C sV | � | jtt�dd�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr TrX rY rZ r r r �testRecvmsgExplicitDefaults
s � z/RecvmsgGenericTests.testRecvmsgExplicitDefaultsc C s | � t� d S r+ r\ r: r r r � _testRecvmsgExplicitDefaults s z0RecvmsgGenericTests._testRecvmsgExplicitDefaultsc C sV | � | jtt�d �\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr TrX rY rZ r r r �testRecvmsgShorter s
� z&RecvmsgGenericTests.testRecvmsgShorterc C s | � t� d S r+ r\ r: r r r �_testRecvmsgShorter s z'RecvmsgGenericTests._testRecvmsgShorterc C s^ | � | jtt�d �\}}}}| �|td d� � | �|| j� | �|g � | j|dd� d S )Nr� ���FrX rY rZ r r r �testRecvmsgTrunc" s
� z$RecvmsgGenericTests.testRecvmsgTruncc C s | � t� d S r+ r\ r: r r r �_testRecvmsgTrunc+ s z%RecvmsgGenericTests._testRecvmsgTruncc C sT | � | jtt�d�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nri TrX rY rZ r r r � testRecvmsgShortAncillaryBuf. s � z0RecvmsgGenericTests.testRecvmsgShortAncillaryBufc C s | � t� d S r+ r\ r: r r r � _testRecvmsgShortAncillaryBuf7 s z1RecvmsgGenericTests._testRecvmsgShortAncillaryBufc C sT | � | jtt�d�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )N� ( TrX rY rZ r r r �testRecvmsgLongAncillaryBuf: s � z/RecvmsgGenericTests.testRecvmsgLongAncillaryBufc C s | � t� d S r+ r\ r: r r r � _testRecvmsgLongAncillaryBufC s z0RecvmsgGenericTests._testRecvmsgLongAncillaryBufc C s" | j �� | �t| j| j d� d S r� )r r r r r� r: r r r �testRecvmsgAfterCloseF s
z)RecvmsgGenericTests.testRecvmsgAfterClosec C s d S r+ r r: r r r �_testRecvmsgAfterCloseK s z*RecvmsgGenericTests._testRecvmsgAfterClosec C s< z*| j�d� | �tj| j| jtt �� W 5 | j �� X d S )NrI )
r� rg r r� r r r0 r� r� r� r: r r r �testRecvmsgTimeoutN s �z&RecvmsgGenericTests.testRecvmsgTimeoutc C s | � | jj| jd�� d S �NrF )r r� rz r r: r r r �_testRecvmsgTimeoutW s z'RecvmsgGenericTests._testRecvmsgTimeoutr� c C s | � | jtt�d dtj�\}}}}| �|td d� � | �|| j� | �|g � | j |dt
tdd�d� | � | jtt�dtj�\}}}}| �|t� | �|| j� | �|g � | j |dd� | � | jtt��\}}}}| �|t� | �|| j� | �|g � | j |dd� d S ) Nr� r rb Fr �r r TrX ) r� r r� r� r r� r� r r� r
rv rZ r r r �testRecvmsgPeekZ s2
�
� � z#RecvmsgGenericTests.testRecvmsgPeekc C s | � t� d S r+ r\ r: r r r �_testRecvmsgPeek} s z$RecvmsgGenericTests._testRecvmsgPeekr� c C s` | j �| j� | �| j tt��\}}}}| �|t� | �|| j� | �|g � | j |dd� d S rW )
r r� r r� r� r� r� r r� r
rZ r r r �testRecvmsgFromSendmsg� s z*RecvmsgGenericTests.testRecvmsgFromSendmsgc C s. | � | �td d� tdd � g�tt�� d S r+ r r: r r r �_testRecvmsgFromSendmsg� s z+RecvmsgGenericTests._testRecvmsgFromSendmsgN)r@ rA rB r[ r] r^ r_ r` ra rc rd re rf rh ri rj rk rl rn r� r rp r� rq rr rs r r r r rV �
s0
"
rV c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�RecvmsgGenericStreamTestsc C sN | � | jd�\}}}}| �|d� | �|| j� | �|g � | j|d d� d S )Nr� r� rX )r� r r� r r� r
rZ r r r �testRecvmsgEOF� s
z(RecvmsgGenericStreamTests.testRecvmsgEOFc C s | j �� d S r+ )r� r r: r r r �_testRecvmsgEOF� s z)RecvmsgGenericStreamTests._testRecvmsgEOFc C s� | � | jtt�d �\}}}}| �|| j� | �|g � | j|dd� | � | jd�\}}}}| �|| j� | �|g � | j|dd� || }| �|t� d S )Nr� FrX r� T)r� r r� r� r r� r� r
)r; r� r� r� r� r� r� r r r �testRecvmsgOverflow� s
� z-RecvmsgGenericStreamTests.testRecvmsgOverflowc C s | � t� d S r+ r\ r: r r r �_testRecvmsgOverflow� s z.RecvmsgGenericStreamTests._testRecvmsgOverflowN)r@ rA rB ru rv rw rx r r r r rt � s rt c @ s e Zd Zdd� Zdd� ZdS )� RecvmsgTestsc C s | � t| jj� | � t| jjddd� | � t| jjtt�dd� | � t| jjtd�gdd� | � t| jjt� dd� | � t| jjtt�t� d� | � t| jjtt�dt� � | j�tt�dd�\}}}}| � |t� | �
|| j � | � |g � | j |dd� d S )NrP r r� TrX )
r r� r r� r� r� r� r� r1 r� r r� r
rZ r r r �testRecvmsgBadArgs� sH � � � � � � z RecvmsgTests.testRecvmsgBadArgsc C s | � t� d S r+ r\ r: r r r �_testRecvmsgBadArgs� s z RecvmsgTests._testRecvmsgBadArgsN)r@ rA rB rz r{ r r r r ry � s ry c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�RecvmsgIntoTestsc C sL t tt��}| �t| jj� | �t| jjtt�dd� | �t| jj|dd� | �t| jjt� gdd� | �t| jjdgdd� | �t| jj|t� gdd� | �t| jj|gdd� | �t| jj|gt� d� | �t| jj|gdt� � | j�|gdd�\}}}}| � |tt�� | � |t t�� | �
|| j � | � |g � | j |dd� d S )Nr s I'm not writablerP TrX )
r� r� r� r r� r r r1 r� r� r r� r
�r; r �nbytesr� r� r� r r r �testRecvmsgIntoBadArgs� s` � � � � � � � � z'RecvmsgIntoTests.testRecvmsgIntoBadArgsc C s | � t� d S r+ r\ r: r r r �_testRecvmsgIntoBadArgs� s z(RecvmsgIntoTests._testRecvmsgIntoBadArgsc C sx t tt��}| j�dd� |fD ��\}}}}| �|tt�� | �|t t�� | �|| j� | �|g � | j|dd� d S )Nc s s | ]
}|V qd S r+ r r r r r r� � s z<RecvmsgIntoTests.testRecvmsgIntoGenerator.<locals>.<genexpr>TrX ) r� r� r� r r r� r r� r
r} r r r �testRecvmsgIntoGenerator� s � z)RecvmsgIntoTests.testRecvmsgIntoGeneratorc C s | � t� d S r+ r\ r: r r r �_testRecvmsgIntoGenerator� s z*RecvmsgIntoTests._testRecvmsgIntoGeneratorc C sx t � ddgtt� �}| j�|g�\}}}}| �|tt�� | �|�� t� | �|| j� | �|g � | j |dd� d S )Nr' r TrX )
r( r� r� r r r� �tobytesr r� r
r} r r r �testRecvmsgIntoArray s z%RecvmsgIntoTests.testRecvmsgIntoArrayc C s | � t� d S r+ r\ r: r r r �_testRecvmsgIntoArray s z&RecvmsgIntoTests._testRecvmsgIntoArrayc C s� t d�}t d�}t d�}| j�|t|�dd� |g�\}}}}| �|td�� | �|t d�� | �|t d�� | �|t d �� | �|| j� | �|g � | j|d
d � d S ) Ns ----s
0123456789s --------------r� � � Mary had a little lambs Marys
01 had a 9s little lamb---TrX ) r� r r �
memoryviewr� r� r r� r
)r; Zb1Zb2Zb3r~ r� r� r� r r r �testRecvmsgIntoScatter s � z'RecvmsgIntoTests.testRecvmsgIntoScatterc C s | � d� d S )Nr� )r� r: r r r �_testRecvmsgIntoScatter s z(RecvmsgIntoTests._testRecvmsgIntoScatterN) r@ rA rB r r� r� r� r� r� r� r� r r r r r| � s
r| c @ sb e Zd Zz ddlZW n ek
r, dZY nX edej�Zee d�dd� �Z
ee d�dd � �Z dS )
�CmsgMacroTestsr Ni����CMSG_LENc C s� | j t�d� d }ttd��tt|d |�� }| �t�d�t�d�jd � |D ]2}t�|�}| �|t�d� |� | � || j � qT| �
t tjd� | �
t tj|� | �
t tjt j
� d S )Nr ri � r2 r� rP )�socklen_t_limitr r� �listr rJ r( �itemsizer� r r rU r[ �maxsize)r; �toobigr- �nr, r r r � testCMSG_LEN. s
zCmsgMacroTests.testCMSG_LENr7 c C s� | j t�d� d }ttd��tt|d |�� }t�d�}| �|t�d�jd � |D ]T}t�|�}| �||� | �|t� |�� | �||t� d� � | �
|| j � |}qX| � t tjd� | � t tj|� | � t tjt
j� d S )Nri r� r r2 r� rP )r� r r7 r� r rJ r( r� r r� r r rU r[ r� )r; r� r- Zlastr� r, r r r �testCMSG_SPACEC s
z CmsgMacroTests.testCMSG_SPACE) r@ rA rB rn r� r� �minro r� r r� r� r r r r r� ! s
r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zdddd�Z dd� Z
dd� Z dd� Z dd� Z
eed�dd� �Zejdd � �Zd d � Zd d � Ze�ejd!kd"�e�ed#�eed�d$d%� ���Zeje�ejd!kd"�e�ed#�d&d'� ���Ze�ejd!kd"�e�ed#�eed�d(d)� ���Z e je�ejd!kd"�e�ed#�d*d+� ���Z d,d-� Z e�ejd!kd.�d/d0� �Z d1d2� Z d3d4� Z!d5d6� Z"eed�d7d8� �Z#e#jd9d:� �Z$ded;d<�Z%d=d>� Z&d?d@� Z'dAdB� Z(dCdD� Z)dEdF� Z*dGdH� Z+dIdJ� Z,dKdL� Z-dMdN� Z.dOdP� Z/dfdQdR�Z0dSdT� Z1dUdV� Z2dWdX� Z3dYdZ� Z4d[d\� Z5d]d^� Z6d_d`� Z7dadb� Z8dcS )g�
SCMRightsTesti����c C s\ g }t |�D ]J}t�� \}}| �tj|� | �tj|� t�|t|�� � � |�
|� q |S r+ ) r r� �mkstemprO r� r� r �writer� �encode�append)r; r� �fdsr2 r/ r� r r r �newFDsd s zSCMRightsTest.newFDsc C sB t |�D ]4\}}t�|dtj� | �t�|d�t|��� � qd S )Nr r� )� enumerater� �lseek�SEEK_SETr� r� r� r� )r; r� r� r/ r r r �checkFDsp s zSCMRightsTest.checkFDsc C s | � | j|� d S r+ )rO �closeRecvmsgFDsr� r r r r� w s z#SCMRightsTest.registerRecvmsgResultc C sn |d D ]`\}}}|t jkr|t jkrt�d�}|�|d t|�t|�|j � � |D ]}t�|� qXqd S )Nri r2 ) r r, �
SCM_RIGHTSr( � frombytesr� r� r� r )r; Zrecvmsg_result�
cmsg_level� cmsg_type� cmsg_datar� r/ r r r r� z s
�
�z SCMRightsTest.closeRecvmsgFDsc
C s8 | � | �tgtjtjt�d| �|��fg�tt�� d S )Nr2 ) r� r� r� r r, r� r( r� r� )r; r� r r r �createAndSendFDs� s ���z SCMRightsTest.createAndSendFDsri r c C s� |\}}}}| � |t� | �|| j� | j|dtj|d� | �|t� | � t
|�|� t � d�} |D ]`}
| �|
t � |
\} } }
| � | tj
� | � | tj� | �|
t� | � t
|
�t d� | �|
� qd| � t
| �|� | �| � d S )NT�r r r r2 r )r� r� r r� r
r r r r� r r� r( �tupler, r� r� �
SIZEOF_INTr� r� )r; Znumfdsr� �maxcmsgs� ignoreflagsr� r� r� r� r� �itemr� r� r� r r r �checkRecvmsgFDs� s& �
z SCMRightsTest.checkRecvmsgFDsc C s | � d| �| jtt�d�� d S )Nri rg �r� r� r r� r� r: r r r �testFDPassSimple� s �z SCMRightsTest.testFDPassSimplec
C s< | � | �tgtjtjt�d| �d���� fg�t t�� d S )Nr2 ri )
r� r� r� r r, r� r( r� r� r� r: r r r �_testFDPassSimple� s ���z SCMRightsTest._testFDPassSimplec C s | � d| �| jtt�d�� d S )Nr% rg r� r: r r r �testMultipleFDPass� s �z SCMRightsTest.testMultipleFDPassc C s | � d� d S �Nr% �r� r: r r r �_testMultipleFDPass� s z!SCMRightsTest._testMultipleFDPassr7 c C s* | � d| �| jtt�t�dt ��� d S r� )r� r� r r� r� r r7 r� r: r r r �testFDPassCMSG_SPACE� s ��z"SCMRightsTest.testFDPassCMSG_SPACEc C s | � d� d S r� r� r: r r r �_testFDPassCMSG_SPACE� s z#SCMRightsTest._testFDPassCMSG_SPACEc
C s0 | j d| �| jtt�t�dt ��tjd� d S )Nri r% �r� ) r� r� r r� r� r r� r� r r: r r r �testFDPassCMSG_LEN� s ��z SCMRightsTest.testFDPassCMSG_LENc C s | � d� d S r� r� r: r r r �_testFDPassCMSG_LEN� s z!SCMRightsTest._testFDPassCMSG_LENru zskipping, see issue #12958zskipping, see issue #22397c C s$ | j d| �| jtt�d�dd� d S )Nr� rg )r� r� r: r r r �testFDPassSeparate� s �z SCMRightsTest.testFDPassSeparatec C sX | � d�\}}| �| �tgtjtjt�d|g�ftjtjt�d|g�fg�tt�� d S �Nr� r2 � r� r� r� r� r r, r� r( r� �r; Zfd0�fd1r r r �_testFDPassSeparate� s � ���z!SCMRightsTest._testFDPassSeparatec C s@ d}| j || �| jtt�t�t�t�t| � �dtj d� d S )Nr� )r� r� )
r� r� r r� r� r r7 r� r� r )r; Znum_fdsr r r �testFDPassSeparateMinSpace� s �� �z(SCMRightsTest.testFDPassSeparateMinSpacec C sX | � d�\}}| �| �tgtjtjt�d|g�ftjtjt�d|g�fg�tt�� d S r� r� r� r r r �_testFDPassSeparateMinSpace� s � ���z)SCMRightsTest._testFDPassSeparateMinSpacec
C sd z| � |g|�}W n< tk
rN } z | �|jt� | � |g�}W 5 d }~X Y nX | �|t|�� d S r+ )r� r r rE rI r� r� )r; r� r� r~ r� r r r �sendAncillaryIfPossible
s z%SCMRightsTest.sendAncillaryIfPossiblezsee issue #24725c C s& | j d| �| jtt�d�tjd� d S )Nr rg r� )r� r� r r� r� r r r: r r r �testFDPassEmpty
s ��z SCMRightsTest.testFDPassEmptyc C s | � ttjtjdfg� d S �Nr� )r� r� r r, r� r: r r r �_testFDPassEmpty
s
�z SCMRightsTest._testFDPassEmptyc C s� | � | jtt�d�\}}}}| �|t� | �|| j� | j|dtj d� | �
t|�d� |D ]6\}}}| �|tj � | �|tj � | �
t|�t� q\d S )Nrg Tro ri )r� r r� r� r� r r� r
r r r r, r� �
assertLessr� �r; r� r� r� r� r� r� r� r r r �testFDPassPartialInt$
s � z"SCMRightsTest.testFDPassPartialIntc C s4 | � ttjtjt�d| jg��� d d� fg� d S )Nr2 rP )r� r� r r, r� r( �badfdr� r: r r r �_testFDPassPartialInt1
s ��z#SCMRightsTest._testFDPassPartialIntc C s� | � | jtt�d�\}}}}| �|t� | �|| j� | j|dtj d� | �
t|�d� t � d�}|D ]J\}}}| �|tj � | �|tj
� |�|d t|�t|�|j � � qf| �
t|�d� | �|� d S )Nrg Tro r� r2 )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 � testFDPassPartialIntInMiddle8
s" �
�
z*SCMRightsTest.testFDPassPartialIntInMiddlec
C sZ | � d�\}}| �ttjtjt�d|| jg��� d d� ftjtjt�d|g�fg� d S )Nr� r2 rP ) r� r� r� r r, r� r( r� r� r� r r r � _testFDPassPartialIntInMiddleK
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 )r; r� r� r� r� r� r� r r r �checkTruncatedHeaderW
s �z"SCMRightsTest.checkTruncatedHeaderc C s" | j | �| jtt��tjd� d S �Nr� �r� r� r r� r� r r r: r r r �testCmsgTruncNoBufSizea
s �z$SCMRightsTest.testCmsgTruncNoBufSizec C s | � d� d S r� r� r: r r r �_testCmsgTruncNoBufSizej
s z%SCMRightsTest._testCmsgTruncNoBufSizec C s$ | j | �| jtt�d�tjd� d S )Nr r� r� r: r r r �testCmsgTrunc0m
s �z SCMRightsTest.testCmsgTrunc0c C s | � d� d S r� r� r: r r r �_testCmsgTrunc0r
s z SCMRightsTest._testCmsgTrunc0c C s | � | �| jtt�d�� d S r� )r� r� r r� r� r: r r r �testCmsgTrunc1x
s z SCMRightsTest.testCmsgTrunc1c C s | � d� d S r� r� r: r r r �_testCmsgTrunc1{
s z SCMRightsTest._testCmsgTrunc1c C s" | � | �| jtt�td �� d S r� )r� r� r r� r� r� r: r r r �testCmsgTrunc2Int~
s �z SCMRightsTest.testCmsgTrunc2Intc C s | � d� d S r� r� r: r r r �_testCmsgTrunc2Int�
s z SCMRightsTest._testCmsgTrunc2Intc C s( | � | �| jtt�t�d�d �� d S r� )r� r� r r� r� r r� r: r r r �testCmsgTruncLen0Minus1�
s �z%SCMRightsTest.testCmsgTruncLen0Minus1c C s | � d� d S r� r� r: r r r �_testCmsgTruncLen0Minus1�
s z&SCMRightsTest._testCmsgTruncLen0Minus1c C s� | � | jtt�|�\}}}}| �|t� | �|| j� | j|dtj d� |dkr\|g kr\d S | �t|�d� |d \}} }
| �|tj
� | �| tj � | � t|
�|� | �
t|
�|� t�d�} | �|
d t|
�t|
�| j � � | �| � d S )NT�r r
r ri r2 )r� r r� r� r� r r� r
r r r, r� r r r( r� r� r� ) r; �ancbuf�maxdataZmindatar� r� r� r� r� r� r� r� r r r �checkTruncatedArray�
s( �
�z!SCMRightsTest.checkTruncatedArrayc C s | j t�d�dd� d S )Nr �r� r� �r� r r� r: r r r �testCmsgTruncLen0�
s z SCMRightsTest.testCmsgTruncLen0c C s | � d� d S r� r� r: r r r �_testCmsgTruncLen0�
s z SCMRightsTest._testCmsgTruncLen0c C s | j t�d�d dd� d S )Nr ri r� r� r: r r r �testCmsgTruncLen0Plus1�
s z$SCMRightsTest.testCmsgTruncLen0Plus1c C s | � d� d S r� r� r: r r r �_testCmsgTruncLen0Plus1�
s z%SCMRightsTest._testCmsgTruncLen0Plus1c C s | j t�t�td� d S )Nr� �r� r r� r� r: r r r �testCmsgTruncLen1�
s �z SCMRightsTest.testCmsgTruncLen1c C s | � d� d S r� r� r: r r r �_testCmsgTruncLen1�
s z SCMRightsTest._testCmsgTruncLen1c C s( | j t�dt �d dt d d� d S )Nr� ri r� r� r: r r r �testCmsgTruncLen2Minus1�
s
�z%SCMRightsTest.testCmsgTruncLen2Minus1c C s | � d� d S r� r� r: r r r �_testCmsgTruncLen2Minus1�
s z&SCMRightsTest._testCmsgTruncLen2Minus1N)ri r )r )r )9r@ rA rB r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r� r� r� r� rN r[ 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� r r r r r� \ s~
r� c @ s e Zd ZdZdZdd� Zd^dd�Zeed d
�d d � �Z e j
d
d� �Z eedd d
�dd� �Z e j
dd� �Z
eejd�eed d
�dd� ��Zej
dd� �Zd_dd�Zeed d
dd �d d � �Zej
d d � �Zeedd d
dd �d!d"� �Zej
d#d$� �Zeejd�eedd d
dd �d%d&� ��Zej
d'd(� �Zeejd�eedd d
dd �d)d*� ��Zej
d+d,� �Zd`d-d.�Zeed d
�d/d0� �Zej
d1d2� �Zeed d
�d3d4� �Z e j
d5d6� �Z eed d
�d7d8� �Z e j
d9d:� �Z eed d
�d;d<� �Z e j
d=d>� �Z!eed d
�d?d@� �Z"e"j
dAdB� �Z#eed d
�dCdD� �Z$e$j
dEdF� �Z%dadGdH�Z&eedd d
dd �dIdJ� �Z'e'j
dKdL� �Z(eedd d
dd �dMdN� �Z)e)j
dOdP� �Z*eedd d
dd �dQdR� �Z+e+j
dSdT� �Z,eedd d
dd �dUdV� �Z-e-j
dWdX� �Z.eedd d
dd �dYdZ� �Z/e/j
d[d\� �Z0d]S )b�RFC3542AncillaryTestr� rP c C s4 i }|D ]&\}}}| � ||f|� ||||f< q|S r+ )r )r; r� �dr� r� r� r r r �ancillaryMapping�
s
z%RFC3542AncillaryTest.ancillaryMappingr� r c C s | j �tjtjd� | j�� | �| j tt �|�\}}}}| �
|t � | � || j � | j
|dtj|d� | �
t|�d� | �|d t� |d \}} }
| �
|tj� | �
| tj� | �|
t� | �
t|
�t� t�d�} | �|
� | �| d d� | �| d |� d S )Nri Tr� r r2 )r r� r r1 �IPV6_RECVHOPLIMITr� rg r� r� r� r� r r� r
r r r� �
IPV6_HOPLIMITr� r� r( r� r r ) r; �
ancbufsize�maxhopr� r� r� r� r� r� r� r� r� r r r �
checkHopLimit�
s2
�
� �
z"RFC3542AncillaryTest.checkHopLimitr� r� c C s | j dd� d S �Nrg �r� )r� r: r r r �testRecvHopLimit�
s z%RFC3542AncillaryTest.testRecvHopLimitc C s$ | � | jj| jd�� | �t� d S rm �r r� rz r r� r� r: r r r �_testRecvHopLimit s z&RFC3542AncillaryTest._testRecvHopLimitr7 c C s | j t�t�d� d S )Nr� )r� r r7 r� r: r r r �testRecvHopLimitCMSG_SPACE s z/RFC3542AncillaryTest.testRecvHopLimitCMSG_SPACEc C s$ | � | jj| jd�� | �t� d S rm r� r: r r r �_testRecvHopLimitCMSG_SPACE s z0RFC3542AncillaryTest._testRecvHopLimitCMSG_SPACEr� c C s | j d| jd� d S �Nrg )r� r� )r� � hop_limitr: r r r �testSetHopLimit s z$RFC3542AncillaryTest.testSetHopLimitc C sL | � | jj| jd�� | �| �tgtjtj t
�
d| j g�fg�t t�� d S �NrF r2 )
r r� rz r r� r� r� r r1 r� r( r� r� r: r r r �_testSetHopLimit s ���z%RFC3542AncillaryTest._testSetHopLimitc C sB | j �tjtjd� | j �tjtjd� | j�� | �| j t t
�|�\}}}}| � |t
� | � || j
� | j|dtj|d� | � t |�d� | �|�}|tjtjf } | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d d� |tjtjf } | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d |� d S )Nri Tr� r� r2 r r� )r r� r r1 r� �IPV6_RECVTCLASSr� rg r� r� r� r� r r� r
r r� � IPV6_TCLASSr� r( r� r r r� ) r; r� r� r� r� r� r� r� ZancmapZtcdatar� Zhldatar r r � checkTrafficClassAndHopLimit( s@
�
�
� �
z1RFC3542AncillaryTest.checkTrafficClassAndHopLimitr r c C s | j dd� d S r� )r r: r r r � testRecvTrafficClassAndHopLimitL s z4RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitc C s$ | � | jj| jd�� | �t� d S rm r� r: r r r � _testRecvTrafficClassAndHopLimitR s z5RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitc C s | j t�t�d d� d S �Nr� r� )r r r7 r� r: r r r �)testRecvTrafficClassAndHopLimitCMSG_SPACEW s �z>RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitCMSG_SPACEc C s$ | � | jj| jd�� | �t� d S rm r� r: r r r �*_testRecvTrafficClassAndHopLimitCMSG_SPACE_ s z?RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitCMSG_SPACEc C s | j d| jd� d S r� �r r� r: r r r � testSetTrafficClassAndHopLimitd s �z3RFC3542AncillaryTest.testSetTrafficClassAndHopLimitc C sd | � | jj| jd�� | �| �tgtjtj t
�
d| j g�ftjtj t
�
d| j
g�fg�tt�� d S r� )r r� rz r r� r� r� r r1 r r( �
traffic_classr� r� r� r: r r r � _testSetTrafficClassAndHopLimitm s �����z4RFC3542AncillaryTest._testSetTrafficClassAndHopLimitc C s | j d| jd� d S r� r r: r r r �testOddCmsgSizex s �z$RFC3542AncillaryTest.testOddCmsgSizec
C s� | � | jj| jd�� zJ| �tgtjtjt � d| j
g�� � d ftjtj t � d| j
g�fg�}W n~ tk
r� } z`| �|jt� | �tgtjtjt � d| j
g�ftjtj t � d| j
g�fg�}| �|tt�� W 5 d }~X Y nX d S )NrF r2 r| )r r� rz r r� r� r r1 r r( r
r� r� r� r r rE rI r� r� )r; r~ r� r r r �_testOddCmsgSize� s2 ��������z%RFC3542AncillaryTest._testOddCmsgSizec C s� | j �tjtjd� | j�� |d kr*dn|f}| j| j tt �f|�� \}}}}| �
|t � | � || j � | �
|g � | j
|dtj|d� d S )Nri r Tr� )r r� r r1 r� r� rg r� r� r� r� r r� r
r )r; r� r� rK r� r� r� r� r r r � checkHopLimitTruncatedHeader� s"
�
�� �z1RFC3542AncillaryTest.checkHopLimitTruncatedHeaderc C s | j d tjd� d S )N�r� r� �r r r r: r r r r� � s �z+RFC3542AncillaryTest.testCmsgTruncNoBufSizec C s$ | � | jj| jd�� | �t� d S rm r� r: r r r r� � s z,RFC3542AncillaryTest._testCmsgTruncNoBufSizec C s | j dtjd� d S )Nr r r r: r r r �testSingleCmsgTrunc0� s �z)RFC3542AncillaryTest.testSingleCmsgTrunc0c C s$ | � | jj| jd�� | �t� d S rm r� r: r r r �_testSingleCmsgTrunc0� s z*RFC3542AncillaryTest._testSingleCmsgTrunc0c C s | j dd� d S )Nri r� )r r: r r r �testSingleCmsgTrunc1� s z)RFC3542AncillaryTest.testSingleCmsgTrunc1c C s$ | � | jj| jd�� | �t� d S rm r� r: r r r �_testSingleCmsgTrunc1� s z*RFC3542AncillaryTest._testSingleCmsgTrunc1c C s | j dt d� d S r )r r� r: r r r �testSingleCmsgTrunc2Int� s z,RFC3542AncillaryTest.testSingleCmsgTrunc2Intc C s$ | � | jj| jd�� | �t� d S rm r� r: r r r �_testSingleCmsgTrunc2Int� s z-RFC3542AncillaryTest._testSingleCmsgTrunc2Intc C s | j t�d�d d� d S )Nr ri r� )r r r� r: r r r � testSingleCmsgTruncLen0Minus1� s z2RFC3542AncillaryTest.testSingleCmsgTruncLen0Minus1c C s$ | � | jj| jd�� | �t� d S rm r� r: r r r � _testSingleCmsgTruncLen0Minus1� s z3RFC3542AncillaryTest._testSingleCmsgTruncLen0Minus1c C s� | j �tjtjd� | j�� | �| j tt �t�
t �d �\}}}}| � |t � | �
|| j� | j|dtjd� | �t|�d� |r�|d \}}}| � |tj� | � |tj� | �t|�t � d S )Nri Tr� r )r r� r r1 r� r� rg r� r� r� r� r� r� r r� r
r r r� r� r� r r r �testSingleCmsgTruncInData� s&
�
� z.RFC3542AncillaryTest.testSingleCmsgTruncInDatac C s$ | � | jj| jd�� | �t� d S rm r� r: r r r �_testSingleCmsgTruncInData� s z/RFC3542AncillaryTest._testSingleCmsgTruncInDatac C s | j �tjtjd� | j �tjtjd� | j�� | �| j t t
�|�\}}}}| � |t
� | � || j
� | j|dtj|d� | � t |�d� |d \}}} | � |tj� | �|tjtjh� | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d d� d S )Nri Tr� r r2 r� )r r� r r1 r� r r� rg r� r� r� r� r r� r
r r r r� r� r( r� r r ) r; r� r� r� r� r� r� r� r� r� r� r r r �checkTruncatedSecondHeader s6
�
�
� �
z/RFC3542AncillaryTest.checkTruncatedSecondHeaderc C s | j t�t�tjd� d S r� )r r r7 r� r r: r r r �testSecondCmsgTrunc0! s �z)RFC3542AncillaryTest.testSecondCmsgTrunc0c C s$ | � | jj| jd�� | �t� d S rm r� r: r r r �_testSecondCmsgTrunc0' s z*RFC3542AncillaryTest._testSecondCmsgTrunc0c C s | � t�t�d � d S r� �r r r7 r� r: r r r �testSecondCmsgTrunc1, s z)RFC3542AncillaryTest.testSecondCmsgTrunc1c C s$ | � | jj| jd�� | �t� d S rm r� r: r r r �_testSecondCmsgTrunc11 s z*RFC3542AncillaryTest._testSecondCmsgTrunc1c C s | � t�t�dt � d S r� r r: r r r �testSecondCmsgTrunc2Int6 s �z,RFC3542AncillaryTest.testSecondCmsgTrunc2Intc C s$ | � | jj| jd�� | �t� d S rm r� r: r r r �_testSecondCmsgTrunc2Int< s z-RFC3542AncillaryTest._testSecondCmsgTrunc2Intc C s" | � t�t�t�d� d � d S r� )r r r7 r� r� r: r r r � testSecondCmsgTruncLen0Minus1A s
��z2RFC3542AncillaryTest.testSecondCmsgTruncLen0Minus1c C s$ | � | jj| jd�� | �t� d S rm r� r: r r r � _testSecondCmsgTruncLen0Minus1G s z3RFC3542AncillaryTest._testSecondCmsgTruncLen0Minus1c
C sT | j �tjtjd� | j �tjtjd� | j�� | �| j t t
�t� t �t�
t � d �\}}}}| �|t
� | �|| j� | j|dtjd� tjtjh}|�d�\}}}| �|tj� |�|� | �t |�t � t�d�} | �|� | �| d d� | �| d d� |�rD|�d�\}}}| �|tj� |�|� | �t |�t � | �|g � d S )Nri Tr� r r2 r� ) r r� r r1 r� r r� rg r� r� r� r7 r� r� r� r r� r
r r r� �pop�remover( r� r r r� )
r; r� r� r� r� Z
cmsg_typesr� r� r� r� r r r �testSecomdCmsgTruncInDataL s@
�
�
�
z.RFC3542AncillaryTest.testSecomdCmsgTruncInDatac C s$ | � | jj| jd�� | �t� d S rm r� r: r r r �_testSecomdCmsgTruncInDataq s z/RFC3542AncillaryTest._testSecomdCmsgTruncInDataN)r� r )r� r )r )r )1r@ rA rB r� r
r� r� r� r r� r� r� r� r� r� r� r r r r r r r r r
r r� r� r r r r r r r r r r r r r r r r! r" r# r$ r' r( r r r r r� �
s�
�
$ �
�
�
�
�
�
�
�
�
#r� c @ s e Zd ZdS )�SendrecvmsgUDPTestBaseN�r@ rA rB r r r r r) y s r) r� c @ s e Zd ZdS )�SendmsgUDPTestNr* r r r r r+ ~ s r+ r� c @ s e Zd ZdS )�RecvmsgUDPTestNr* r r r r r, � s r, r c @ s e Zd ZdS )�RecvmsgIntoUDPTestNr* r r r r r- � s r- c @ s e Zd Zdd� ZdS )�SendrecvmsgUDP6TestBasec C s | � |d d� |d d� � d S r� r� r r r r r � s z+SendrecvmsgUDP6TestBase.checkRecvmsgAddressN)r@ rA rB r r r r r r. � s r. r r� r c @ s e Zd ZdS )�SendmsgUDP6TestNr* r r r r r/ � s r/ c @ s e Zd ZdS )�RecvmsgUDP6TestNr* r r r r r0 � s r0 c @ s e Zd ZdS )�RecvmsgIntoUDP6TestNr* r r r r r1 � s r1 r1 c @ s e Zd ZdS )� RecvmsgRFC3542AncillaryUDP6TestNr* r r r r r2 � s r2 c @ s e Zd ZdS )�#RecvmsgIntoRFC3542AncillaryUDP6TestNr* r r r r r3 � s r3 c @ s e Zd ZdS )�SendrecvmsgTCPTestBaseNr* r r r r r4 � s r4 c @ s e Zd ZdS )�SendmsgTCPTestNr* r r r r r5 � s r5 c @ s e Zd ZdS )�RecvmsgTCPTestNr* r r r r r6 � s r6 c @ s e Zd ZdS )�RecvmsgIntoTCPTestNr* r r r r r7 � s r7 c @ s e Zd ZdS )� SendrecvmsgSCTPStreamTestBaseNr* r r r r r8 � s r8 z6IPPROTO_SCTP: [Errno 62] Protocol not supported on AIXr4 r5 r� c @ s e Zd ZdS )�SendmsgSCTPStreamTestNr* r r r r r9 � s r9 c s e Zd Z� fdd�Z� ZS )�RecvmsgSCTPStreamTestc
sR zt t| ��� W n: tk
rL } z |jtjkr2� | �d� W 5 d }~X Y nX d S �Nz4sporadic ENOTCONN (kernel issue?) - see issue #13876)rE r: ru r rE �ENOTCONNrZ �r; r� rM r r ru � s z$RecvmsgSCTPStreamTest.testRecvmsgEOF�r@ rA rB ru rQ r r rM r r: � s r: c s e Zd Z� fdd�Z� ZS )�RecvmsgIntoSCTPStreamTestc
sR zt t| ��� W n: tk
rL } z |jtjkr2� | �d� W 5 d }~X Y nX d S r; )rE r? ru r rE r< rZ r= rM r r ru � s z(RecvmsgIntoSCTPStreamTest.testRecvmsgEOFr> r r rM r r? � s r? c @ s e Zd ZdS )� SendrecvmsgUnixStreamTestBaseNr* r r r r r@ � s r@ r� c @ s e Zd ZdS )�SendmsgUnixStreamTestNr* r r r r rA � s rA c @ s e Zd ZdS )�RecvmsgUnixStreamTestNr* r r r r rB � s rB c @ s e Zd ZdS )�RecvmsgIntoUnixStreamTestNr* r r r r rC s rC r, r� c @ s e Zd ZdS )�RecvmsgSCMRightsStreamTestNr* r r r r rD s rD c @ s e Zd ZdS )� RecvmsgIntoSCMRightsStreamTestNr* r r r r rE
s rE c sD e Zd Z� fdd�ZdZeed�r0dZdd� Zn dZd d� Z� Z S )
�InterruptedTimeoutBasec s2 t � �� t�tjdd� �}| �tjtj|� d S )Nc S s dd S r� r )�signumr� r r r r� ! r� z.InterruptedTimeoutBase.setUp.<locals>.<lambda>)rE r< r� r� rO )r; Zorig_alrm_handlerrM r r r< s
�z InterruptedTimeoutBase.setUpg @� setitimerg�������?c C s t �t j|� d S r+ )r� rH � ITIMER_REAL�r; Zsecondsr r r �setAlarm- s z InterruptedTimeoutBase.setAlarmr� c C s t �|� d S r+ )r� r� rJ r r r rK 3 s )
r@ rA rB r< r0 r r� �
alarm_timerK rQ r r rM r rF s
rF � siginterruptr� rH z+Don't have signal.alarm or signal.setitimerc sl e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Ze e
j
d
�dd� �Z e e
j
d�dd� �Z � Z
S )�InterruptedRecvTimeoutTestc s t � �� | j�| j� d S r+ )rE r< r6 r� r0 r: rM r r r< @ s
z InterruptedRecvTimeoutTest.setUpc O sB z0| � | j� | �t��}|||� W 5 Q R X W 5 | � d� X d S r� )rK rL r r� �r; rh rK rL r( r r r �checkInterruptedRecvD s
z/InterruptedRecvTimeoutTest.checkInterruptedRecvc C s | � | jjd� d S r� )rP r6 r� r: r r r �testInterruptedRecvTimeoutN s z5InterruptedRecvTimeoutTest.testInterruptedRecvTimeoutc C s | � | jjtd�� d S r� )rP r6 � recv_intor� r: r r r � testInterruptedRecvIntoTimeoutQ s z9InterruptedRecvTimeoutTest.testInterruptedRecvIntoTimeoutc C s | � | jjd� d S r� )rP r6 r� r: r r r � testInterruptedRecvfromTimeoutT s z9InterruptedRecvTimeoutTest.testInterruptedRecvfromTimeoutc C s | � | jjtd�� d S r� )rP r6 �
recvfrom_intor� r: r r r �"testInterruptedRecvfromIntoTimeoutW s z=InterruptedRecvTimeoutTest.testInterruptedRecvfromIntoTimeoutr� c C s | � | jjd� d S r� )rP r6 r� r: r r r � testInterruptedRecvmsgTimeoutZ s z8InterruptedRecvTimeoutTest.testInterruptedRecvmsgTimeoutr c C s | � | jjtd�g� d S r� )rP r6 r r� r: r r r �!testInterruptedRecvmsgIntoTimeout^ s z<InterruptedRecvTimeoutTest.testInterruptedRecvmsgIntoTimeout)r@ rA rB r< rP rQ rS rT rV r� r rW rX rQ r r rM r rN 9 s
rN 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)rE r< r� r� rO r rG �Thread� doConnect�startr6 r� r� r� r� r0 )r; Z
cli_threadr� rM r r r< n s
z InterruptedSendTimeoutTest.setUpc C s | j �| j� d S r+ )r� r� r� r: r r r r\ | s z$InterruptedSendTimeoutTest.doConnectc O sD z2| �t�� }| � | j� |||� qW 5 Q R X W 5 | � d� X d S r� )rK r r� rL rO r r r �checkInterruptedSend s
z/InterruptedSendTimeoutTest.checkInterruptedSendr� r� c C s | � | jjd� d S rN )r^ r� r� r: r r r �testInterruptedSendTimeout� s z5InterruptedSendTimeoutTest.testInterruptedSendTimeoutc C s | � | jjd| j� d S rN )r^ r� r& r� r: r r r � testInterruptedSendtoTimeout� s �z7InterruptedSendTimeoutTest.testInterruptedSendtoTimeoutr� c C s | � | jjdg� d S rN )r^ r� r� r: r r r � testInterruptedSendmsgTimeout� s z8InterruptedSendTimeoutTest.testInterruptedSendmsgTimeout)r@ rA rB r< r\ r^ r Zrequires_mac_verr_ r` r� r ra rQ r r rM r rY e s
rY c @ s e Zd Zdd� Zdd� ZdS )�
TCPCloserTestc C sh | j �� \}}|�� | j}t�|gg g d�\}}}| �||g� | �|�d�d� |�� |�� d S )N� �?ri r� )r6 r� r r� r� r� r� )r; r� r� �sdr� r� r r r r � testClose� s zTCPCloserTest.testClosec C s | j �t| jf� t�d� d S �Nrc )r� r� r� r8 r� r� r: r r r �
_testClose� s zTCPCloserTest._testCloseN)r@ rA rB re rg r r r r rb � s
rb c @ sN e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dS )�BasicSocketPairTestr� c C s t j| |d� d S r� )r� rF r� r r r rF � s z BasicSocketPairTest.__init__c C s\ | � |tj� ttd�r*| �|jtj� n| �|jtj� | �|jtj� | �|j d� d S )Nr� r )
r r r r� r r� r4 r� r5 r r� r r r �_check_defaults� s
z#BasicSocketPairTest._check_defaultsc C s | � | j� d S r+ )ri r� r: r r r �
_testDefaults� s z!BasicSocketPairTest._testDefaultsc C s | � | j� d S r+ )ri r6 r: r r r � testDefaults� s z BasicSocketPairTest.testDefaultsc C s | j �d�}| �|t� d S r� )r6 r� r� r� r� r r r r� � s z BasicSocketPairTest.testRecvc C s | j �t� d S r+ )r� r� r� r: r r r r� � s z BasicSocketPairTest._testRecvc C s | j �t� d S r+ )r6 r� r� r: r r r �testSend� s z BasicSocketPairTest.testSendc C s | j �d�}| �|t� d S r� )r� r� r� r� r� r r r � _testSend� s z BasicSocketPairTest._testSendN)r� ) r@ rA rB rF ri rj rk r� r� rl rm r r r r rh � s
rh 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� )rG rn �eventr� rF r� r r r rF � s
z NonBlockingTCPTests.__init__c C sh | � | j�� |� |dk}| � |�� |� td k rd|d k}t�|tjtj�}| � t|tj@ � |� d S )N� ) r� r6 r� � getblockingr
�F_GETFLr� �
O_NONBLOCK�bool)r; r� r0 �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 )NTFrp r r� )r6 r& rw r� r: r r r �testSetBlocking� s z#NonBlockingTCPTests.testSetBlockingc C s d S r+ r r: r r r �_testSetBlocking s z$NonBlockingTCPTests._testSetBlockingc C sb dd l }|j|jkr | �d� | j�d� | �| j�� d� | j�|jd � | �| j�� � d S )Nr zneeds UINT_MAX < ULONG_MAXFrp ri ) rn r� Z ULONG_MAXrZ r6 r& r� r� r< r� r r r �testSetBlocking_overflow s
z,NonBlockingTCPTests.testSetBlocking_overflowr% z test needs socket.SOCK_NONBLOCKr� � � c C s4 | j �� t�tjtjtjB �| _ | �| j d� d S r� )r6 r r r4 r5 r% rw r: r r r �testInitNonBlocking s
�z'NonBlockingTCPTests.testInitNonBlockingc C s d S r+ r r: r r r �_testInitNonBlocking s z(NonBlockingTCPTests._testInitNonBlockingc C sL t d ��: | j�d� | j�� \}}| �|j� | �|�� � W 5 Q R X d S )Nr� )r1 r6 r� r� rO r r< r� r� r r r �testInheritFlagsBlocking s
z,NonBlockingTCPTests.testInheritFlagsBlockingc C s | j �t| jf� d S r+ �r� r� r� r8 r: r r r �_testInheritFlagsBlocking) s z-NonBlockingTCPTests._testInheritFlagsBlockingc C sR d}t |��< | j�d� | j�� \}}| �|j� | �|�� |� W 5 Q R X d S )Ng 4@r� )r1 r6 r� r� rO r r� r� )r; Zdefault_timeoutr� r� r r r �testInheritFlagsTimeout, s
z+NonBlockingTCPTests.testInheritFlagsTimeoutc C s | j �t| jf� d S r+ r� r: r r r �_testInheritFlagsTimeout7 s z,NonBlockingTCPTests._testInheritFlagsTimeoutc C s� | j �d� t�� }| �t�� | j �� \}}W 5 Q R X t�� | }| �|d� | j� � t
�
| j gg g t �\}}}| j |kr�| � d� | j �� \}}| �
|j� | �|�� � d S )Nr rc z'Error trying to do accept after select.)r6 r& r� � monotonicr �BlockingIOErrorr� r� ro rg r� � MAIN_TIMEOUTr rO r r< r� )r; Z
start_timer� r� Zdtr� r� r r r r �
testAccept: s
z NonBlockingTCPTests.testAcceptc C s | j �� | j�t| jf� d S r+ )ro rz r� r� r� r8 r: r r r � _testAcceptP s
z NonBlockingTCPTests._testAcceptc C s� | j �� \}}| �|j� |�d� | �t�� |�tt ��}W 5 Q R X | j
� � t � |gg g t
�\}}}||kr|| �d� |�tt ��}| �|t � d S )Nr z0Error during select call to non-blocking socket.)r6 r� rO r r& r r� r� r� r� ro rg r� r� r r� )r; r� r� r� r� r� r r r r r� W s
z NonBlockingTCPTests.testRecvc C s, | j �t| jf� | j�� | j �t� d S r+ )r� r� r� r8 ro rz r� r� r: r r r r� k s
z NonBlockingTCPTests._testRecvN)r� )r@ rA rB rF rw rx ry r rM rz Z_testSetBlocking_overflowr� rL r r �requires_linux_versionr} r~ r r� r� r� r� r� r� r� r r r r rn � s*
� rn c @ s� e Zd ZdZdZdZdZdZeZ dZ
eZ d:dd �Z d
d � Z
d d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Z d*d+� Z d,d-� Z d.d/� Z d0d1� Z d2d3� Z!d4d5� Z"d6d7� Z#d8d9� Z$dS );�FileObjectClassTestCaserP r �strictNr r� r� c C s t j| |d� d S r� r� r� r r r rF � s z FileObjectClassTestCase.__init__c C sR dd� t d�D �\| _| _| _| _t�| � | jj| j | j
| j | j | j
d�| _d S )Nc S s g | ] }t �� �qS r )rG rn �r� r2 r r r r� � s z1FileObjectClassTestCase.setUp.<locals>.<listcomp>r% ��encoding�errors�newline)r �evt1�evt2�
serv_finished� cli_finishedr� r< r� r� � read_moder[ r� r� r� � read_filer: r r r r< � s �
�z FileObjectClassTestCase.setUpc C s6 | j �� | j�� | �| jj� d | _t�| � d S r+ )r� rg r� r r �closedr� r> r: r r r r> � s
z FileObjectClassTestCase.tearDownc C s0 t �| � | jj| j| j| j| j| jd�| _ d S )Nr� )
r� r� r� r� �
write_moder[ r� r� r� �
write_filer: r r r r� � s
�z#FileObjectClassTestCase.clientSetUpc C s6 | j �� | j�� | �| jj� d | _t�| � d S r+ )r� rg r� r r r� r� r� r: r r r r� � s
z&FileObjectClassTestCase.clientTearDownc C sd | j �d� | j�d� | �tj| jjd� | �t��}| j�d� W 5 Q R X | �dt |j
�� d S )Nri r� z!cannot read from timed out object) r� r� r� r� r r r0 r r r� r' )r; �ctxr r r �testReadAfterTimeout� s z,FileObjectClassTestCase.testReadAfterTimeoutc C s. | j �| jdd� � | j �� | j�� d S )Nr r� )r� r� � write_msg�flushr� rz r: r r r �_testReadAfterTimeout� s
z-FileObjectClassTestCase._testReadAfterTimeoutc C s< | j �t| j�d �}| j �d�}|| }| �|| j� d S r+ )r� r� r� �read_msgr� )r; � first_segZ
second_segr� r r r �
testSmallRead� s z%FileObjectClassTestCase.testSmallReadc C s | j �| j� | j �� d S r+ �r� r� r� r� r: r r r �_testSmallRead� s z&FileObjectClassTestCase._testSmallReadc C s | j �� }| �|| j� d S r+ )r� r� r� r� r� r r r � testFullRead� s
z$FileObjectClassTestCase.testFullReadc C s | j �| j� | j �� d S r+ )r� r� r� r r: r r r �
_testFullRead� s z%FileObjectClassTestCase._testFullReadc C s: t | j�� }| j�d�}|s q(||7 }q | �|| j� d S r� )r� r� r� r� r� )r; r �charr r r �testUnbufferedRead� s
z*FileObjectClassTestCase.testUnbufferedReadc C s | j �| j� | j �� d S r+ r� r: r r r �_testUnbufferedRead� s z+FileObjectClassTestCase._testUnbufferedReadc C s | j �� }| �|| j� d S r+ )r� �readliner� r� �r; �liner r r � testReadline� s
z$FileObjectClassTestCase.testReadlinec C s | j �| j� | j �� d S r+ r� r: r r r �
_testReadline� s z%FileObjectClassTestCase._testReadlinec C s& | j �� | j�� }| �|| j� d S r+ )r� r r� r� r� r� r� r r r �testCloseAfterMakefile� s
z.FileObjectClassTestCase.testCloseAfterMakefilec C s | j �| j� | j �� d S r+ r� r: r r r �_testCloseAfterMakefile� s z/FileObjectClassTestCase._testCloseAfterMakefilec C s@ | j �� | j�tt��}t| jt�r.|� � }| �
|| j� d S r+ ) r� r r� r� r� r� r� r� r� �decoder� r� r r r � testMakefileAfterMakefileClose� s
z6FileObjectClassTestCase.testMakefileAfterMakefileClosec C s | j �| j� | j �� d S r+ r� r: r r r � _testMakefileAfterMakefileClose� s z7FileObjectClassTestCase._testMakefileAfterMakefileClosec C s | � | jj � d S r+ )r r� r� r: r r r �testClosedAttr s z&FileObjectClassTestCase.testClosedAttrc C s | � | jj � d S r+ )r r� r� r: r r r �_testClosedAttr s z'FileObjectClassTestCase._testClosedAttrc C s, | � | jj| j� | � | jj| j�� � d S r+ )r� r� r� r� r� r� r r: r r r �testAttributes s z&FileObjectClassTestCase.testAttributesc C s, | � | jj| j� | � | jj| j�� � d S r+ )r� r� r� r� r� r� r r: r r r �_testAttributes s z'FileObjectClassTestCase._testAttributesc C s8 | j �� | �t| j j� | j�� | �t| jj� d S r+ )r� r r r� r r� r r� r: r r r �
testRealClose s
z%FileObjectClassTestCase.testRealClosec C s d S r+ r r: r r r �_testRealClose s z&FileObjectClassTestCase._testRealClose)r� )%r@ rA rB r[ r� r� r� r� r� r� r� r� rF r< r> r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� v sB
r� c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�!UnbufferedFileObjectClassTestCaser c C sL | j �� }| �|d| j � | j�dd�| _ | j �� }| �|d| j � d S )N� A. r r � B. )r� r� r� r� r� r� r� r r r �testUnbufferedReadline% s
z8UnbufferedFileObjectClassTestCase.testUnbufferedReadlinec C s2 | j �d| j � | j �d| j � | j �� d S )Nr� r� r� r: r r r �_testUnbufferedReadline- s z9UnbufferedFileObjectClassTestCase._testUnbufferedReadlinec C sD | j �� | j �d�}| �|| j� | j�� | �t| j jd� d S r� )r� r r� r� r� r� r r r� r r r �testMakefileClose2 s
z3UnbufferedFileObjectClassTestCase.testMakefileClosec C s | j �| j� | j �� d S r+ r� r: r r r �_testMakefileClose; s z4UnbufferedFileObjectClassTestCase._testMakefileClosec C s6 t �| j�}| j�� t �| j�}| �|d |� d S r� )r[ rZ r� r� r r� )r; Zrefcount_beforeZrefcount_afterr r r � testMakefileCloseSocketDestroy? s
z@UnbufferedFileObjectClassTestCase.testMakefileCloseSocketDestroyc C s d S r+ r r: r r r � _testMakefileCloseSocketDestroyE s zAUnbufferedFileObjectClassTestCase._testMakefileCloseSocketDestroyc C s | j �d� | �| j�td��d � | �| j�t| j�d �d � | j �
� | j � d� | j�t| j�d �}|d kr�t
�d� | j�t| j�d �}td�}| j�|�}| �|d� ||d |� }| �|| j� | �| j�td��d � | �| j�d�d � d S )NFr� r� rc g �?ra ri )r� r& r� r� �readintor� r� r� r� r� rg r� rz r� r� )r; r� r r� r� r r r �testSmallReadNonBlockingL s
z:UnbufferedFileObjectClassTestCase.testSmallReadNonBlockingc C s> | j �d� | j�| j� | j�� | j�� | j�d� d S )Nrc � @) r� rz r� r� r� r� r� rg r� r: r r r �_testSmallReadNonBlocking_ s
z;UnbufferedFileObjectClassTestCase._testSmallReadNonBlockingc C s" | j �d� | jr | �| j� d S )Nr� )r� rz � serv_skippedrZ r: r r r �testWriteNonBlockingh s z6UnbufferedFileObjectClassTestCase.testWriteNonBlockingc C st d | _ | j�d� dtj }d}| j�|�}| �|d� t|�D ](}| j�|�}|d kr\ qp| �|d� q@d| _ d S )NFr� r� r z$failed to saturate the socket buffer) r� r� r& r r� r� r� rJ r )r; ZBIGZLIMITr� r2 r r r �_testWriteNonBlockingo s
z7UnbufferedFileObjectClassTestCase._testWriteNonBlockingN)r@ rA rB r[ r� r� r� r� r� r� r� r� r� r� r r r r r� s
r� c @ s e Zd ZdZdS )�#LineBufferedFileObjectClassTestCaseri N�r@ rA rB r[ r r r r r� � s r� c @ s e Zd ZdZdS )�$SmallBufferedFileObjectClassTestCaser� Nr� r r r r r� � s r� c @ s& e Zd ZdZe�d�ZdZeZdZ dS )�"UnicodeReadFileObjectClassTestCaser r r� r" N�
r@ rA rB r� r� r� r� r� r� r� r r r r r� � s
r� c @ s& e Zd ZdZeZdZe�d�ZdZ dS )�#UnicodeWriteFileObjectClassTestCaser r� r r" N)
r@ rA rB r� r� r� r� r� r� r� r r r r r� � s
r� c @ s, e Zd ZdZe�d�ZdZe�d�ZdZ dS )�'UnicodeReadWriteFileObjectClassTestCaser r r� r" Nr� r r r r r� � s
r� c @ s e Zd Zdd� ZdS )�NetworkConnectionTestc C s t �t| jf�| _| j| _d S r+ )r �create_connectionr� r8 r� r� r: r r r r� � s z!NetworkConnectionTest.clientSetUpN)r@ rA rB r� r r r r r� � s r� c @ s e Zd ZdS )�
BasicTCPTest2Nr* r r r r r� � s r� c @ sD e Zd ZG dd� dej�Zejdd� �Zdd� Zdd� Z d d
� Z
d S ) �NetworkConnectionNoServerc @ s e Zd Zdd� ZdS )z$NetworkConnectionNoServer.MockSocketc G s t �d��d S )Nz timed out)r r0 r� r r r r� � s z,NetworkConnectionNoServer.MockSocket.connectN)r@ rA rB r� r r r r �
MockSocket� s r� c c s& t j }| jt _ z
d V W 5 |t _ X d S r+ )r r� )r; Z
old_socketr r r �mocked_socket_module� s
z.NetworkConnectionNoServer.mocked_socket_modulec C s^ t �� }t�tjtj�}| �|j� | �t��}|� t
|f� W 5 Q R X | � |j j
t
j� d S r+ )r r� r r4 r5 rO r r r r� r� r� r' rE Z ECONNREFUSED)r; r8 r� r( r r r � test_connect� s z&NetworkConnectionNoServer.test_connectc C sH t �� }| �t��}t�t|f� W 5 Q R X t �� }| �|j j
|� d S r+ ) r r� r r r r� r� Z get_socket_conn_refused_errsr r' rE )r; r8 r( Zexpected_errnosr r r �test_create_connection� s
z0NetworkConnectionNoServer.test_create_connectionc C s� | � � �n zt�tdf� W nJ tjk
r2 Y n@ tk
rf } ztjsT|jtj krV� W 5 d }~X Y n X | �
d� W 5 Q R X d S )Nr rJ ) r� r r� r� r0 r r r� rE Z EAFNOSUPPORTr r~ r r r � test_create_connection_timeout� s
z8NetworkConnectionNoServer.test_create_connection_timeoutN) r@ rA rB r r� �
contextlib�contextmanagerr� r� r� r� r r r r r� � s
r� c @ sv e Zd Zddd�Zdd� Zdd� Zdd � ZeZd
d � ZeZ d d
� Z
eZ dd� Z eZ
dd� ZeZdd� ZeZdd� ZdS )� NetworkConnectionAttributesTestr� c C s t j| |d� t�| � d S r� r� r� r r r rF � s z(NetworkConnectionAttributesTest.__init__c C s t �� | _d S r+ )r r� � source_portr: r r r r� � s z+NetworkConnectionAttributesTest.clientSetUpc C s | j �� d | _ t�| � d S r+ r� r: r r r r� s
z.NetworkConnectionAttributesTest.clientTearDownc C s | j �� \}}|�� d S r+ )r6 r� r r� r r r � _justAccept s z+NetworkConnectionAttributesTest._justAcceptc C s8 t jt| jfdd�| _| �| jj� | �| jjd� d S )N� rF r� ) r r� r� r8 r� rO r r� r r: r r r � _testFamily s z+NetworkConnectionAttributesTest._testFamilyc C sH t jt| jfdd| jfd�| _| �| jj� | �| j� � d | j� d S )Nr� r" )r0 �source_addressri )
r r� r� r8 r� r� rO r r� r� r: r r r �_testSourceAddress s
�z2NetworkConnectionAttributesTest._testSourceAddressc C sd | � t�� d k� t�d� z$t�t| jf�| _| �| jj � W 5 t�d � X | �
| j� � d� d S )Nr � r r r. r/ r� r� r8 r� rO r r� r� r: r r r �_testTimeoutDefault s
z3NetworkConnectionAttributesTest._testTimeoutDefaultc C sh | � t�� d k� t�d� z(tjt| jfd d�| _| �| jj � W 5 t�d � X | �
| j� � d � d S �Nr� rF r� r: r r r �_testTimeoutNone' s
z0NetworkConnectionAttributesTest._testTimeoutNonec C s, t jt| jfdd�| _| �| j�� d� d S r� )r r� r� r8 r� r� r� r: r r r �_testTimeoutValueNamed3 s z6NetworkConnectionAttributesTest._testTimeoutValueNamedc C s8 t �t| jfd�| _| �| jj� | �| j�� d� d S )Nr� ) r r� r� r8 r� rO r r� r� r: r r r �_testTimeoutValueNonamed8 s z8NetworkConnectionAttributesTest._testTimeoutValueNonamedN)r� )r@ rA rB rF r� r� r� Z
testFamilyr� ZtestSourceAddressr� ZtestTimeoutDefaultr� ZtestTimeoutNoner� ZtestTimeoutValueNamedr� ZtestTimeoutValueNonamedr� r r r r r� � s
r� c @ sB e Zd Zddd�Zdd� Zdd� Zdd � ZeZd
d � Zd d
� Z dS )� NetworkConnectionBehaviourTestr� c C s t j| |d� t�| � d S r� r� r� r r r rF @ s z'NetworkConnectionBehaviourTest.__init__c C s d S r+ r r: r r r r� D s z*NetworkConnectionBehaviourTest.clientSetUpc C s | j �� d | _ t�| � d S r+ r� r: r r r r� G s
z-NetworkConnectionBehaviourTest.clientTearDownc C s2 | j �� \}}| �|j� t�d� |�d� d S )Nr� � done!)r6 r� rO r r� r� r� r� r r r �testInsideTimeoutL s
z0NetworkConnectionBehaviourTest.testInsideTimeoutc C s0 t �t| jf� | _}|�d�}| �|d� d S )Nr� r� )r r� r� r8 r� r� r� )r; r� r r r r �_testInsideTimeoutS s
z1NetworkConnectionBehaviourTest._testInsideTimeoutc s4 t jt| jfdd� | _� | �t j� fdd�� d S )Nri rF c s
� � d�S )Nr� )r� r �r� r r r� Z r� zDNetworkConnectionBehaviourTest._testOutsideTimeout.<locals>.<lambda>)r r� r� r8 r� r r0 r: r r� r �_testOutsideTimeoutX s z2NetworkConnectionBehaviourTest._testOutsideTimeoutN)r� )
r@ rA rB rF r� r� r� ZtestOutsideTimeoutr� r� r r r r r� > s
r� c @ s6 e Zd Zdd� Zdd� Ze�eed�d�dd� �Z d S )
�TCPTimeoutTestc s � fdd�}� � tj|d� d S )Nc s � j �d� � j �� d S rf )r6 r� r� r� r: r r �
raise_timeout` s z4TCPTimeoutTest.testTCPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (TCP)�r r r0 �r; r� r r: r �testTCPTimeout_ s
�z TCPTimeoutTest.testTCPTimeoutc C s| d}z| j �d� | j �� }W nJ tjk
r> | �d� Y n, tk
rT d}Y n | �d� Y nX |sx| �d� d S )NFrp z%caught timeout instead of error (TCP)Tz!caught unexpected exception (TCP)z3accept() returned success when we did not expect it)r6 r� r� r r0 r r �r; �okrC r r r �testTimeoutZerof s z TCPTimeoutTest.testTimeoutZeror� ztest needs signal.alarm()c
s
| j �d� G dd� dt�� � fdd�}t�tj|�}z�z�z�zt�d� | j �� }W n` tjk
rt | � d� Y nL � k
r� Y n: | � d t
� � d d� t �
� f � Y n X | � d
� W 5 t�d� X W n � k
r� | � d � Y nX W 5 t�tj|� X d S ) Nr� c @ s e Zd ZdS )z4TCPTimeoutTest.testInterruptedTimeout.<locals>.AlarmNr* r r r r �Alarm{ s r� c s � �d S r+ r )r� r� �r� r r �
alarm_handler} s z<TCPTimeoutTest.testInterruptedTimeout.<locals>.alarm_handlerr r� z caught timeout instead of Alarmz3caught other exception instead of Alarm: %s(%s):
%sznothing caughtzgot Alarm in wrong place)r6 r� r r� r� r� r� r r0 r r[ �exc_info� traceback�
format_exc)r; r� r� rC r r� r �testInterruptedTimeoutt s*
� z%TCPTimeoutTest.testInterruptedTimeoutN)
r@ rA rB r� r� r� rL r r� r� r r r r r� ] s �r� c @ s e Zd Zdd� Zdd� ZdS )�UDPTimeoutTestc s � fdd�}� � tj|d� d S )Nc s � j �d� � j �d� d S )Nrc r� )r6 r� r� r� r: r r r� � s z4UDPTimeoutTest.testUDPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (UDP)r� r� r r: r �testUDPTimeout� s
�z UDPTimeoutTest.testUDPTimeoutc C s~ d}z | j �d� | j �d�}W nJ tjk
r@ | �d� Y n, tk
rV d}Y n | �d� Y nX |sz| �d� d S )NFrp r� z%caught timeout instead of error (UDP)Tz!caught unexpected exception (UDP)z1recv() returned success when we did not expect it)r6 r� r� r r0 r r r� r r r r� � s z UDPTimeoutTest.testTimeoutZeroN)r@ rA rB r� r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�TestExceptionsc C sJ | � ttt�� | � ttjt�� | � ttjt�� | � ttjt�� d S r+ )r �
issubclassr r r r r r0 r: r r r �testExceptionTree� s z TestExceptions.testExceptionTreec C sb t � t jt jd�}t � t jt jd|�� �}|�� | �|j� | �t�� |� d� W 5 Q R X d S )Nr F)
r r4 r5 r r rO r) r r r& )r; Zsock0r� r r r �test_setblocking_invalidfd� s � z)TestExceptions.test_setblocking_invalidfdN)r@ rA rB r r r r r r r� � s r� rt zLinux specific testc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�TestLinuxAbstractNamespace�l c C s� d}t � t jt j��z}|�|� |�� t � t jt j��L}|�|�� � |�� d �&}| �|�� |� | �|� � |� W 5 Q R X W 5 Q R X W 5 Q R X d S )Ns python-test-hello �r )
r r� r5 rX r9 r� r� r� r� � getpeername)r; rn �s1r; Zs3r r r �testLinuxAbstractNamespace� s
z5TestLinuxAbstractNamespace.testLinuxAbstractNamespacec C sL dd| j d }t�tjtj�� }|�|� | �|�� |� W 5 Q R X d S )Nr| � hri )�
UNIX_PATH_MAXr r� r5 rX r� r� �r; rn r r r r � testMaxName� s
z&TestLinuxAbstractNamespace.testMaxNamec C s> dd| j }t�tjtj��}| �t|j|� W 5 Q R X d S )N� �h)r r r� r5 r r rX r
r r r �testNameOverflow� s z+TestLinuxAbstractNamespace.testNameOverflowc C s> t � t jt j�}z |�d� | �|�� d� W 5 |�� X d S )Nz
python test �
python test )r r� r5 r rX r� r� r r r r � testStrName� s
z&TestLinuxAbstractNamespace.testStrNamec C s> t � t jt j��$}|�td�� | �|�� d� W 5 Q R X d S )Nr )r r� r5 rX r� r� r� r r r r �testBytearrayName� s z,TestLinuxAbstractNamespace.testBytearrayNameN) r@ rA rB r r r r r r r r r r r � s r ztest needs socket.AF_UNIXc @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestUnixDomainc C s t � t jt j�| _d S r+ )r r� r5 r� r: r r r r< � s zTestUnixDomain.setUpc C s | j �� d S r+ )r� r r: r r r r> � s zTestUnixDomain.tearDownc
C s< z t �|�W S tk
r6 | �d�|t�� �� Y nX d S )NzBPathname {0!a} cannot be represented in file system encoding {1!r})r� �fsencoder� rZ r r[ �getfilesystemencoding�r; r� r r r �encoded� s ��zTestUnixDomain.encodedc
C sX zt �||� W nB tk
rR } z$t|�dkr@| �d�|�� n� W 5 d }~X Y nX d S )NzAF_UNIX path too longz6Pathname {0!a} is too long to serve as an AF_UNIX path)r r� r r� rZ r )r; r� r� r� r r r rX s ��zTestUnixDomain.bindc C s | � | j�� d� d S )N)r" N)r r� r� r: r r r r� s zTestUnixDomain.testUnboundc C s@ t j�tj�}| �| j|� | �tj|� | � | j�
� |� d S r+ ) r� r� �abspathr �TESTFNrX r� rO r� r� r� r r r r � testStrAddr s zTestUnixDomain.testStrAddrc C sF t j�tj�}| �| j| �|�� | �tj |� | �
| j� � |� d S r+ ) r� r� r r r rX r� r rO 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 rX r� r� rO r� r� r� )r; r� r� r r r �testSurrogateescapeBind s
z&TestUnixDomain.testSurrogateescapeBindc C sT t jd kr| �d� tj�t j�}| �| j|� | �t j |� | �
| j� � |� d S )Nz!No unencodable filename available) r ZTESTFN_UNENCODABLErZ r� r� r rX r� rO r� r� r� r r r r �testUnencodableAddr' s
z"TestUnixDomain.testUnencodableAddrN) r@ rA rB r< r> r rX r� r r r r r r r r r � s r c @ s� e Zd Zddd�Zdd� Zdd� Zdd � ZeZd
d � ZeZ d d
� Z
dd� Z dd� Z e Z
dd� Ze Zdd� Zdd� Zdd� Ze ZdS ) � BufferIOTestr� c C s t j| |d� d S r� r� r� r r r rF 6 s zBufferIOTest.__init__c C sZ t � ddgtt� �}| j�|�}| �|tt�� |�� }|d tt�� }| �|t� d S �Nr' r )r( r� r� r� rR r� r� �r; r r~ r� r r r �testRecvIntoArray9 s z BufferIOTest.testRecvIntoArrayc C s t t�}| j�|� d S r+ �r� r� r� r� �r; r r r r �_testRecvIntoArrayA s z BufferIOTest._testRecvIntoArrayc C sD t d�}| j�|�}| �|tt�� |d tt�� }| �|t� d S r� )r� r� rR r� r� r� r! r r r �testRecvIntoBytearrayE s
z"BufferIOTest.testRecvIntoBytearrayc C sH t d�}| j�t|��}| �|tt�� |d tt�� }| �|t� d S r� )r� r� rR r� r� r� r� r! r r r �testRecvIntoMemoryviewN s
z#BufferIOTest.testRecvIntoMemoryviewc C s^ t � ddgtt� �}| j�|�\}}| �|tt�� |�� }|d tt�� }| �|t� d S r )r( r� r� r� rU r� r� �r; r r~ r� r� r r r �testRecvFromIntoArrayW s z"BufferIOTest.testRecvFromIntoArrayc C s t t�}| j�|� d S r+ r# r$ r r r �_testRecvFromIntoArray_ s z#BufferIOTest._testRecvFromIntoArrayc C sH t d�}| j�|�\}}| �|tt�� |d tt�� }| �|t� d S r� )r� r� rU r� r� r� r( r r r �testRecvFromIntoBytearrayc s
z&BufferIOTest.testRecvFromIntoBytearrayc C sL t d�}| j�t|��\}}| �|tt�� |d tt�� }| �|t� d S r� )r� r� rU r� r� r� r� r( r r r �testRecvFromIntoMemoryviewl s
z'BufferIOTest.testRecvFromIntoMemoryviewc C s t d�}| �t| jj|d� d S )NrV r� )r� r r� r� rU r$ r r r �testRecvFromIntoSmallBufferu s z(BufferIOTest.testRecvFromIntoSmallBufferc C s | j �t� d S r+ r� r: r r r � _testRecvFromIntoSmallBufferz s z)BufferIOTest._testRecvFromIntoSmallBufferc C s$ t � }| j�|� | j�|d� d S r� )r� r� rU r$ r r r �testRecvFromIntoEmptyBuffer} s z(BufferIOTest.testRecvFromIntoEmptyBufferN)r� )r@ rA rB rF r" r% r& Z_testRecvIntoBytearrayr' Z_testRecvIntoMemoryviewr) r* r+ Z_testRecvFromIntoBytearrayr, Z_testRecvFromIntoMemoryviewr- r. r/ Z _testRecvFromIntoEmptyBufferr r r r r 2 s"
r i� �� �� c
C sp t td�sdS z td�} W n tttfk
r6 Y dS X | �* | D ] }|�d�rB W 5 Q R � dS qBW 5 Q R X dS )N�AF_TIPCFz
/proc/modulesztipc T)r r r
�FileNotFoundError�IsADirectoryErrorr? r{ )r r� r r r �isTipcAvailable� s
r5 z6TIPC module is not loaded, please 'sudo modprobe tipc'c @ s e Zd Zdd� ZdS )�TIPCTestc C s� t � t jt j�}t � t jt j�}| �|j� | �|j� |�t jt jd� t jt t
t f}|� |� t j
t t
tt t
d � df}|�t|� |�d�\}}| �|�� |� | �|t� d S )Nri r� r r� )r r2 r+ rO r r� r, r- �TIPC_ADDR_NAMESEQ�
TIPC_STYPE�
TIPC_LOWER�
TIPC_UPPERrX �TIPC_ADDR_NAMErI r& r� r� r� r� )r; r r� �srvaddrZsendaddrr� Zrecvaddrr r r �testRDM� s$ �
� zTIPCTest.testRDMN)r@ rA rB r= r r r r r6 � s r6 c @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )�TIPCThreadableTestr� c C s t jj| |d� t�| � d S r� r� r� r r r rF � s zTIPCThreadableTest.__init__c C s� t � t jt j�| _| �| jj� | j�t jt jd� t j t
t t f}| j�
|� | j�� | �� | j�� \| _| _| �| jj� d S r� )r r2 r5 r rO r r� r, r- r7 r8 r9 r: rX r9 rh r� r� r� )r; r< r r r r< � 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 r2 r5 r� rO r r; r8 r9 rI r: r� r� �cliaddr)r; r� r r r r� � s
� z TIPCThreadableTest.clientSetUpc C s, | j �d�}| �|t� | �| j| j� d S r� )r� r� r� r� r? r� r� r r r r� � s z TIPCThreadableTest.testStreamc C s | j �t� | j �� d S r+ r� r: r r r r� � s z TIPCThreadableTest._testStreamN)r� r� r r r r r> � s
r> c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ContextManagersTestc C s� t � � �}| �|j� W 5 Q R X | �|j� t � � �}|�� W 5 Q R X | �|j� t � � �}| �t|jd� W 5 Q R X | �|j� d S �Nr# )r r� r� r r r r r� r� r r r �_testSocketClass� s
z$ContextManagersTest._testSocketClassc C s2 | j �� \}}| �|j� |�d�}|�|� d S r� �r6 r� rO r r� r� �r; r� r� r r r r �testCreateConnectionBase� s
z,ContextManagersTest.testCreateConnectionBasec C sX | j �� }t�|��.}| �|j� |�d� | �|�d�d� W 5 Q R X | � |j� d S )Nr# r� )
r6 r� r r� r� r� r� r� r� r �r; rn r� r r r �_testCreateConnectionBase� s
z-ContextManagersTest._testCreateConnectionBasec C s2 | j �� \}}| �|j� |�d�}|�|� d S r� rC rD r r r �testCreateConnectionClose� s
z-ContextManagersTest.testCreateConnectionClosec C sH | j �� }t�|��}|�� W 5 Q R X | �|j� | �t|j d� d S rA )
r6 r� r r� r r r� r r r� rF r r r �_testCreateConnectionClose s
z.ContextManagersTest._testCreateConnectionCloseN)r@ rA rB rB rE rG rH rI r r r r r@ � s
r@ c @ s� e Zd Ze�eed�d�e�ddd�dd� ��Z dd � Z
d
d � Z d d
� Z e�
edkd�dd� �Ze�
edkd�dd� �Zdd� ZdS )�InheritanceTestr� zSOCK_CLOEXEC not definedr� r{ r| c C sD t � t jt jt jB ��$}| �|jt j� | �|�� � W 5 Q R X d S r+ )r r4 r5 r� r� r� r� �get_inheritabler r r r �test_SOCK_CLOEXEC s
�z!InheritanceTest.test_SOCK_CLOEXECc C s, t � � }|� | �|�� d� W 5 Q R X d S r� )r r� rK r� r r r �test_default_inheritable s z(InheritanceTest.test_default_inheritablec
C sL t � � }|�6 |�� }|�� |� | �|�� d� W 5 Q R X W 5 Q R X d S r� )r r� r r� rK )r; r� Znewsockr r r �test_dup s zInheritanceTest.test_dupc C sP t � � }|�: |�d� | �|�� d� |�d� | �|�� d� W 5 Q R X d S )NTF)r �set_inheritabler� rK r� r r r �test_set_inheritable! s
z$InheritanceTest.test_set_inheritableNr� c C sn t � � }|�X |�� }| �|�� d� t�|tj�}|tj M }t�|tj|� | �|�� d� W 5 Q R X d S r )r r r� rK r
�F_GETFD�
FD_CLOEXECZF_SETFD)r; r� r/ r� r r r � test_get_inheritable_cloexec* s z,InheritanceTest.test_get_inheritable_cloexecc C sh t � � }|�R |�� }| �t�|tj�tj@ tj� |�d� | �t�|tj�tj@ d� W 5 Q R X d S )NTr )r r r� r
rQ rR rO )r; r� r/ r r r � test_set_inheritable_cloexec8 s �
�z,InheritanceTest.test_set_inheritable_cloexecc C sH t �� \}}| �|j� | �|j� | �|�� d� | �|�� d� d S r� )r r� rO r r� rK )r; r r; r r r �test_socketpairE s
z InheritanceTest.test_socketpair)r@ rA rB r� rL r r r r� rL rM rN rP rN r
rS rT rU r r r r rJ
s �
rJ r% zSOCK_NONBLOCK not definedc @ s, e Zd Zd dd�Ze�ddd�dd � �Zd
S ) �NonblockConstantTestTrp c C s� |rj| � |jtj� | � |�� |� | �t�|tjtj �tj @ � |dkrZ| �
|� � � q�| �|� � � nL| � |jtj� | � |�� d � | �
t�|tjtj �tj @ � | �|� � � d S r� ) r� r� r r5 r� r r
rr r� rs r� rq )r; r �nonblockr0 r r r �
checkNonblockP s � �z"NonblockConstantTest.checkNonblockr� r{ r| c C s^ t � t jt jt jB ���}| �|� |�d� | j|dd� |�d� | �|� |�d � | j|dd� |�d� | j|dd� |�d� | j|dd� W 5 Q R X t �� }t �d� t � � �}| �|� W 5 Q R X t �d � t � � �}| �|d� W 5 Q R X t �d� t � � �}| j|dd� W 5 Q R X t �d � t � � �}| �|d� W 5 Q R X t �|� d S )Nri F)rW r g @rF rp ) r r4 r5 r% rX r& r� r. r/ )r; r �tr r r �test_SOCK_NONBLOCKi s:
�
z'NonblockConstantTest.test_SOCK_NONBLOCKN)Trp )r@ rA rB rX r r� rZ r r r r rV M s
rV r3 r4 zneed multiprocessingc @ s@ e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
S )�TestSocketSharingc C sD |� � }|� � }t�|�}|�� \}}|�|� |�� |�� d S r+ )r} r � fromsharer� r� r )r~ �qZsdata�messager r; r� r r r �remoteProcessServer� s
z%TestSocketSharing.remoteProcessServerc C s� t �� }t j| j|fd�}|�� | j�|j�}| j�� }| j� � |�
|� d}|�
|� t � |�}g }|�
d�}|szq�|�|� qj|� � d�|�}| �||� |�� d S )N�rZ rK s
slapmahfror� r� )�multiprocessingrr �Processr_ r] r6 �share�pidr� r r� r r� r� r� r� r� ) r; r] r r r� r^ r �m�receivedr r r � testShare� s&
zTestSocketSharing.testSharec C s@ | j �t�� �}| �ttj|d d� � | �ttj|d � d S )NrP r# )r6 rc r� �getpidr r� r r\ )r; r r r r �testShareLength� s z!TestSocketSharing.testShareLengthc C sb | � |�� d � | � |�� |�� � | � |j|j� | � |j|j� |jdkr^| � |j|j� d S r� )r� r� r r� r )r; Zorg�otherr r r �compareSockets� s
z TestSocketSharing.compareSocketsc C s< | j �t�� �}t�|�}z| �| j |� W 5 |�� X d S r+ )r6 rc r� rh r r\ r rk )r; r r r r r �testShareLocal� s
z TestSocketSharing.testShareLocalc
C s� t jt jg}t jt jg}|D ]|}|D ]r}zt � ||�}W n tk
rP Y q$Y nX z8|�t� � �}t �
|�}z| � ||� W 5 |�� X W 5 |�� X q$q d S r+ ) r r4 r� r5 r r r rc r� rh r\ rk )r; �families�typesr rY �sourcer Zsharedr r r � testTypes� s
zTestSocketSharing.testTypesN)
r@ rA rB r� r_ rg ri rk rl rp r r r r r[ � s
#r[ c @ s� e Zd ZdZdZdZdZedd� �Zedd� �Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!d9d:� Z"d;S )<�SendfileUsingSendTesti � r` r� r� c C s~ dd� }d� dd� t| j�D ��}ttjd��$}|| j| j�D ]}|�|� q>W 5 Q R X ttjd��}|�� | _ W 5 Q R X d S )Nc s s& | |kr|V | |8 } q | r"| V d S r+ r )�total�stepr r r �chunks� s
z0SendfileUsingSendTest.setUpClass.<locals>.chunksr� c S s g | ]}t �tj��� �qS r )�random�choice�string�
ascii_lettersr� r� r r r r� s �z4SendfileUsingSendTest.setUpClass.<locals>.<listcomp>r� r )
r� r �BUFSIZEr
r r �FILESIZEr� r� �FILEDATA)r~ rt �chunkr Zcsizer r r �
setUpClass� s
�
z SendfileUsingSendTest.setUpClassc C s t �t j� d S r+ )r r� r )r~ r r r �
tearDownClass s z#SendfileUsingSendTest.tearDownClassc C s6 | j �t� | j �� \}}|�| j� | �|j� |S r+ )r6 r� r� r� �TIMEOUTrO r r� r r r � accept_conn s
z!SendfileUsingSendTest.accept_connc C s, g }|� | j�}|sq"|�|� qd�|�S r� )r� ry r� r� )r; r� rf r| r r r � recv_data s z SendfileUsingSendTest.recv_datac C s
t |d�S )N�_sendfile_use_send�rv r� r r r �meth_from_sock# s z$SendfileUsingSendTest.meth_from_sockc
C sr | j �� }ttjd�}t�|��H}|�8}| �|�}||�}| �|| j � | �|�
� | j � W 5 Q R X W 5 Q R X d S )Nr � r6 r� r
r r r r� r� r� rz �tell�r; rn �filer� �meth�sentr r r �_testRegularFile* s
z&SendfileUsingSendTest._testRegularFilec C s6 | � � }| �|�}| �t|�| j� | �|| j� d S r+ �r� r� r� r� rz r{ �r; r� r r r r �testRegularFile3 s
z%SendfileUsingSendTest.testRegularFilec
C s| | j �� }t�| j�}t�|��R}|�B}|�|�}| �|| j � | �|�
� | j � | � tj |j
|� W 5 Q R X W 5 Q R X d S r+ )r6 r� �io�BytesIOr{ r r� r1 r� rz r� r r6 r7 )r; rn r� r� r� r r r �_testNonRegularFile; s
�z)SendfileUsingSendTest._testNonRegularFilec C s6 | � � }| �|�}| �t|�| j� | �|| j� d S r+ r� r� r r r �testNonRegularFileE s
z(SendfileUsingSendTest.testNonRegularFilec
C s� | j �� }tjd }t|d�� | �tj|� W 5 Q R X t|d�}t�|��D}|�4}| � |�}||�}| �
|d� | �
|� � d� W 5 Q R X W 5 Q R X d S )N�2r� r r ) r6 r� r r r
rO r� r r� r� r� r� )r; rn �filenamer� r� r� r� r r r �_testEmptyFileSendM s
z(SendfileUsingSendTest._testEmptyFileSendc C s" | � � }| �|�}| �|d� d S r� )r� r� r� r� r r r �testEmptyFileSendY s
z'SendfileUsingSendTest.testEmptyFileSendc
C sz | j �� }ttjd�}t�|��P}|�@}| �|�}||dd�}| �|| j d � | �|�
� | j � W 5 Q R X W 5 Q R X d S )Nr � )�offsetr� r� r r r � _testOffset` s
z!SendfileUsingSendTest._testOffsetc C sB | � � }| �|�}| �t|�| jd � | �|| jdd � � d S )Nr� r� r� r r r �
testOffseti s
z SendfileUsingSendTest.testOffsetc
C sz | j �� }ttjd�}tj|dd��L}|�<}d}| �|�}|||d�}| �||� | �|� � |� W 5 Q R X W 5 Q R X d S )Nr r� rF �GKL �r� �
r6 r� r
r r r r� r� r� r� �r; rn r� r� r� r� r� r r r �
_testCountq s
z SendfileUsingSendTest._testCountc C s@ d}| � � }| �|�}| �t|�|� | �|| jd |� � d S )Nr� �r� r� r� r� r{ �r; r� r� r r r r � testCount{ s
z SendfileUsingSendTest.testCountc
C sz | j �� }ttjd�}tj|dd��L}|�<}d}| �|�}|||d�}| �||� | �|� � |� W 5 Q R X W 5 Q R X d S )Nr r� rF ri r� r� r� r r r �_testCountSmall� s
z%SendfileUsingSendTest._testCountSmallc C s@ d}| � � }| �|�}| �t|�|� | �|| jd |� � d S r� r� r� r r r �testCountSmall� s
z$SendfileUsingSendTest.testCountSmallc
C s� | j �� }ttjd�}tj|dd��R}|�B}d}| �|�}||d|d�}| �||� | �|� � |d � W 5 Q R X W 5 Q R X d S )Nr r� rF 駆 �� )r� r� r� r� r r r �_testCountWithOffset� s
z*SendfileUsingSendTest._testCountWithOffsetc C sD d}| � � }| �|�}| �t|�|� | �|| jd|d � � d S )Nr� r� r� r� r r r �testCountWithOffset� s
z)SendfileUsingSendTest.testCountWithOffsetc
C sr | j �� }ttjd�}t�|��H}|�8}|�d� | �|�}| � t
||� | � t
|j |� W 5 Q R X W 5 Q R X d S )Nr F) r6 r� r
r r r r� r& r� r r� r1 �r; rn r� r� r� r r r �_testNonBlocking� s
z&SendfileUsingSendTest._testNonBlockingc C s | � � }|�d�r | �d� d S )Nr` z$was not supposed to receive any data)r� r� r �r; r� r r r �testNonBlocking� s
z%SendfileUsingSendTest.testNonBlockingc
C sd | j �� }ttjd�}tj|dd��6}|�&}| �|�}||�}| �|| j � W 5 Q R X W 5 Q R X d S )Nr r� rF )
r6 r� r
r r r r� r� r� rz r� r r r �_testWithTimeout� s
z&SendfileUsingSendTest._testWithTimeoutc C s6 | � � }| �|�}| �t|�| j� | �|| j� d S r+ r� r� r r r �testWithTimeout� s
z%SendfileUsingSendTest.testWithTimeoutc
C s` | j �� }ttjd��@}t�|��*}|�d� | �|�}| � tj
||� W 5 Q R X W 5 Q R X d S )Nr g{�G�z�?) r6 r� r
r r r r� r� r� r r0 r� r r r � _testWithTimeoutTriggeredSend� s
z3SendfileUsingSendTest._testWithTimeoutTriggeredSendc C s | � � }|�d� d S )Ni�X )r� r� r� r r r � testWithTimeoutTriggeredSend� s z2SendfileUsingSendTest.testWithTimeoutTriggeredSendc C s d S r+ r r: r r r � _test_errors� s z"SendfileUsingSendTest._test_errorsc
C s t tjd��:}tjtjd�� }| �|�}| �td||� W 5 Q R X W 5 Q R X t tjd��4}t�� � }| �|�}| �td||� W 5 Q R X W 5 Q R X t tjd��t}t�� �`}| �|�}| jtd||dd� | jtd||d d� | jtd||d
d� | jtd||d d� W 5 Q R X W 5 Q R X d S ) Nr r� r5 r z binary modezpositive integerr� r� r� r rP ) r
r r r r r� r r� r� )r; r� r r� r r r � test_errors� sR
�
�
� � � �z!SendfileUsingSendTest.test_errorsN)#r@ rA rB rz ry r{ r r� r} r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rq � sB
rq r1 z%os.sendfile() required for this test.c @ s e Zd Zdd� ZdS )�SendfileUsingSendfileTestc C s
t |d�S )Nr7 r� r� r r r r� � s z(SendfileUsingSendfileTest.meth_from_sockN)r@ rA rB r� r r r r r� � s r� zAF_ALG requiredc @ s| e Zd Zdd� Ze�dd�dd� �Zdd� Ze�dd �d
d � �Ze�dd �d
d� �Z e�dd �dd� �Z
dd� Z dd� Z dS )�LinuxKernelCryptoAPIc
C sj t � t jt jd�}z|�||f� W n< tk
r` } z |�� t�t|�||��W 5 d }~X Y nX |S d S r� ) r r' r% rX r3 r r� ZSkipTestr� )r; �typr� r� r� r r r �
create_alg s $z LinuxKernelCryptoAPI.create_algr% r� c
C s� t �d�}| �dd���}|�� \}}|�" |�d� | �|�d�|� W 5 Q R X |�� \}}|�L |�dtj � |�dtj � |�dtj � |�d � | �|�d�|� W 5 Q R X W 5 Q R X d S )
NZ@ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad�hashZsha256s abcrw � a� b� cr� )
r� �fromhexr� r� r� r� r� r� r �MSG_MORE�r; r �algo�oprk r r r � test_sha256
s
z LinuxKernelCryptoAPI.test_sha256c
C sp t �d�}| �dd��P}|�tjtjd� |�� \}}|�" |�d� | � |�
d�|� W 5 Q R X W 5 Q R X d S )NZ(effcdf6ae5eb2fa2d27416d5f184df9c259a7c79r� z
hmac(sha1)s Jefes what do ya want for nothing?rw ) r� r� r� r� r �SOL_ALG� ALG_SET_KEYr� r� 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���z}|�tjtj|� |�� \}}|�6 |j tj
|tj d� |� |� | �
|�|�|� W 5 Q R X |�� \}}|�, |j |gtj|d� | �
|�|�|� W 5 Q R X d } |g| }
|�� \}}|�& |j |
tj
|d� |�|| �} W 5 Q R X | �
t| �|| � | �
| d |� |� |�� \}}|�( |j | gtj|d� |�|| �} W 5 Q R X | �
t| �|| � | �
| || � W 5 Q R X d S )
NZ 06a9214036b8a15b512e03d534120006Z 3dafba429d9eb430b422da802c9fac41s Single block msgZ e353779c1079aeb82708942dbe77181aZskcipherzcbc(aes))r� �ivr� )r� r� r� )r� r� r� r� r� r r� r� r� �
sendmsg_afalg�ALG_OP_ENCRYPTr� r� r� r� �ALG_OP_DECRYPT)
r; �keyr� r� Z
ciphertextZmsglenr� r� rk Z
multiplierZlongmsg�encZdecr r r � test_aes_cbc* sR
�
�
� �z!LinuxKernelCryptoAPI.test_aes_cbcr� c C s� t �d�}t �d�}t �d�}t �d�}t �d�}t �d�}t|�}t|�}| �dd���N} | �tjtj|� | �tjtjd |� | � � \}
} |
�v |
j
tj ||tj d � |
�
|tj � |
�
|� |
�|t|� | �} | �|| || � � | �|| | d � � W 5 Q R X | � � \}
} |
�f || }
|
j
|
gtj ||d
� |
�|t|� | �} | �|| || � � | �|| | d � � W 5 Q R X t�d �j}| � � \}
} |
�� || }
|
�|
gtjtj|tj �gtjtj|t|��| gtjtj||�gf� |
�t|
�| �} | �|| || � � | �|| | d � � W 5 Q R X | � � \}
} |
�N || | }
|
j
|
gtj||d
� |
�t|
�| �} | �|| |d � � W 5 Q R X W 5 Q R X d S ) NZ c939cc13397c1d37de6ae0e1cb7c423cZb3d8cc017cbb89b39e0f67e2Z c3b3c41f113a31b73d9a5cd432103069Z 24825602bd12a984e0092d3e448eda5fZ 93fe7d9e9bfd10348a5606e5cafa7354Z 0032a1dc85f1c9786925a2e71d8272ddZaeadzgcm(aes))r� r� �assoclenr� )r� r� r� r )r� r� r� r� r� r r� r� �ALG_SET_AEAD_AUTHSIZEr� r� r� r� r� r� r� r �Structrx r� �
ALG_SET_OP�
ALG_SET_IV�ALG_SET_AEAD_ASSOCLENr� )r; r� r� ZplainZassocZ expected_ctZ expected_tagZtaglenr� r� r� rk �resr� Z pack_uint32r r r �test_aead_aes_gcmS sp
�
�
� �� �z&LinuxKernelCryptoAPI.test_aead_aes_gcmc
C sn | � dd��X}t�d�}|�tjtj|� |�� \}}|� |�d�}| � t
|�d� W 5 Q R X W 5 Q R X d S )N�rngZdrbg_pr_sha256r` ) r� r� �urandomr� r r� r� r� r� r� r� )r; r� Z
extra_seedr� rk Zrnr r r �test_drbg_pr_sha256� s
z(LinuxKernelCryptoAPI.test_drbg_pr_sha256c
C s� t � t jt jd�}|�� | �t�� |�� W 5 Q R X | �t�� |jd d� W 5 Q R X | �t�� |�d� W 5 Q R X | �t�� |jt jd d� W 5 Q R X | �t�� |jt jdd� W 5 Q R X W 5 Q R X d S )Nr )r� ri )r� r� rP )r r' r% r r� r� r� r� r r r �test_sendmsg_afalg_args� s z,LinuxKernelCryptoAPI.test_sendmsg_afalg_argsc C s� t � t jt jd�}| �|j� | �t�� |�d� W 5 Q R X | �t d�� |�d� W 5 Q R X | �t�� |�d� W 5 Q R X | �t d�� |�d� W 5 Q R X d S )Nr )Z
tttttttttttttr� z
type too long)Zttttttttttttttr� )r� Z?nnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnz
name too long)r� Z@nnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnn)
r r' r% rO r r r3 rX r r� r� r r r �test_length_restriction� s z,LinuxKernelCryptoAPI.test_length_restrictionN)
r@ rA rB r� r r� r� r� r� r� r� r� r� r r r r r� � s
(
>
r� �winzrequires Windowsc @ s$ e Zd ZddddddhZdd� Zd S )
�TestMSWindowsTCPFlags�
TCP_MAXSEG� TCP_NODELAY� TCP_FASTOPEN� TCP_KEEPCNT� TCP_KEEPIDLE�
TCP_KEEPINTVLc s6 dd� t t�D �}� fdd�|D �}� �g |d� d S )Nc S s g | ]}|� d �r|�qS )ZTCP)r{ �r� r r r r r� � s
z<TestMSWindowsTCPFlags.test_new_tcp_flags.<locals>.<listcomp>c s g | ]}|� j kr|�qS r )�
knownTCPFlagsr� r: r r r� � s
zANew TCP flags were discovered. See bpo-32394 for more information)r� r r� )r; Zprovided�unknownr r: r �test_new_tcp_flags� s
�z(TestMSWindowsTCPFlags.test_new_tcp_flagsN)r@ rA rB r� r� r r r r r� � s �
r� c @ s� e Zd Zdd� Zdd� Zdd� Ze�ee d� p:ee d� d �e�
e j d
�d d � ��Z
e�e�� d
�e�
e j d
�dd� ��ZdS )�CreateServerTestc C s� t �� }t�d|f��.}| �|�� d d� | �|�� d |� W 5 Q R X t jr�tjd|ftjd��.}| �|�� d d� | �|�� d |� W 5 Q R X d S )Nr
r ri r� r� )r r� r �
create_serverr� r� r� r� �r; r8 r� r r r � test_address� s
�z CreateServerTest.test_addressc C s| t �d��&}| �|jt j� | �|jt j� W 5 Q R X tjrxt jdt j d��&}| �|jt j � | �|jt j� W 5 Q R X d S )Nr �r� r r� )
r r� r� r r4 r� r5 r r� r� )r; r� r r r r �test_family_and_type� s z%CreateServerTest.test_family_and_typec C s� t td�s0| �t�� tjddd� W 5 Q R X nht�d��"}|�tjtj�}| �|d� W 5 Q R X tjddd��"}|�tjtj�}| � |d� W 5 Q R X d S )N� SO_REUSEPORT)r� r T)�
reuse_portr )
r r r r� r� r� r, r� r� r� )r; r� �optr r r �test_reuse_port� s
z CreateServerTest.test_reuse_portr1 � IPV6_V6ONLYz IPV6_V6ONLY option not supported�IPv6 required for this testc C s t jdt jd��}W 5 Q R X d S )Nr� r� )r r� r� r� r r r �test_ipv6_only_default� s z'CreateServerTest.test_ipv6_only_default� dualstack_ipv6 not supportedc C s2 t jdt jdd��}| �|jt j� W 5 Q R X d S )Nr� T�r �dualstack_ipv6)r r� r� r� r r� r r r �test_dualstack_ipv6_family s
�z+CreateServerTest.test_dualstack_ipv6_familyN)r@ rA rB r� r� r� r� rN r r rL r r� r� r �has_dualstack_ipv6r� r r r r r� � s
� � r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Ze �
e j d �d
d� �Z
e �e�� d�e �
e j d �dd� ��Ze �e�� d�e �
e j d �dd� ��ZdS )�CreateServerFunctionalTestr� c C s
d | _ d S r+ )rw r: r r r r< s z CreateServerFunctionalTest.setUpc C s | j d k r| j �| j� d S r+ )rw r� r0 r: r r r r> s
z#CreateServerFunctionalTest.tearDownc sJ � �fdd�}t �� � |��j� t j||fd��_�j�� � �� d S )Nc
sl | �^ | � � \}}|�B � ��j� |�d�}|sJW 5 Q R � W 5 Q R � d S |�|� W 5 Q R X W 5 Q R X d S r� )r� rz r0 r� r� )r� r� rk r� �ro r; r r �run s
z3CreateServerFunctionalTest.echo_server.<locals>.runr` )rG rn r� r0 r[ rw r] rg )r; r� r� r r� r � echo_server s
z&CreateServerFunctionalTest.echo_serverc C sN t j |d��8}|�| j� |�|� |�d� | �|�d�d� W 5 Q R X d S )Nr� r# r� )r r� r0 r� r� r� r� )r; r� r r� r r r � echo_client& s
z&CreateServerFunctionalTest.echo_clientc C sB t �� }t�d|f��"}| �|� | �d|ftj� W 5 Q R X d S )Nr" r
)r r� r r� r� r� r4 r� r r r � test_tcp4- s
z$CreateServerFunctionalTest.test_tcp4r� c C sH t �� }tjd|ftjd��"}| �|� | �d|ftj� W 5 Q R X d S )Nr" r� r� �r r� r r� r� r� r� r� r r r � test_tcp63 s
�
z$CreateServerFunctionalTest.test_tcp6r� c C sJ t �� }tjd|ftjdd��"}| �|� | �d|ftj� W 5 Q R X d S )Nr" Tr� r
)r r� r r� r� r� r� r4 r� r r r �test_dual_stack_client_v4= s �
z4CreateServerFunctionalTest.test_dual_stack_client_v4c C sJ t �� }tjd|ftjdd��"}| �|� | �d|ftj� W 5 Q R X d S )Nr" Tr� r� r� r� r r r �test_dual_stack_client_v6G s �
z4CreateServerFunctionalTest.test_dual_stack_client_v6N)r@ rA rB r0 r< r> r� r� r� r� rL r r� r� rN r r� r� r� r r r r r� s$
� � r� c C s t ttttttttt t
g } | � t t
ttttttttttttg� | �t� | �t � | �t � | � t t g� | � t t!g� | � t"t#g� | �t$� | �t%� | � t&t'g� | � t(t)t*t+t,t-t.t/t0t1t2t3t4t5t6t7t8t9t:t;t<t=t>t?t@g� | �tA� tB�C� }tBjD| � tBjE|� d S r+ )Fr r� rb r� r� r r� r� r� r� r� �extendrn r� r� r� r� r� r� r� r� r� r� r@ rJ rV r� rh r r r6 r> rO r{ r� r� r� r� r� r� r� r+ r, r- r/ r0 r2 r3 r1 r5 r6 r7 r9 r: r? rA rB rC rD rE rN rY r[ rq r� r� r Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_mainR s� ��
��
r �__main__)�r� �testr rE r� r� r r� r� r� r� rq r[ r� rz r( r� �weakrefr r� r� r r ru r= rw �_threadrw rG ra r� r
r� r� r� r� r� �systemr r r r r" r& r( r* r- r� r1 ZHAVE_SOCKET_CANZHAVE_SOCKET_CAN_ISOTPZHAVE_SOCKET_RDSZHAVE_SOCKET_ALGZHAVE_SOCKET_QIPCRTRZHAVE_SOCKET_VSOCKr� r� r� r3 rC rD rR r_ ra r� r� r� r� rN rL r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r rO r{ r� r� r� r� r� r� r� r� r r r r r r r rA rS rV rt ry r| r� r� r� r) r+ r, r- r. r� r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r? r@ rA rB rC rD rE rF r rN rY rb rh rn r� r� r� r� r� r� r� r1 r� r� r� r� r� r� r� r� r r r r8 r9 r: r5 r6 r> r@ rJ rV r� r[ rq r� r� r{ r� r� r� r r@ r r r r �<module> s�
3u� � ! � �$
P
S
Z
*
>��.{7w w= # P; k 7 �
�
�
� �
�
� �
�
�
�
�
� �
�)� �7# ! $l
?E 9)ES �� ',C�< e � D3G C