File: //proc/thread-self/root/usr/local/lib/python3.7/test/__pycache__/test_socket.cpython-37.opt-2.pyc
B
��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 dlmZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZy d dl Z W n e k
r� dZ Y nX y d dl Z W n e k
�r dZ Y nX ej Z d� d�Z!dZ"dZ#y d dl$Z$W n e k
�r\ dZ$Y nX d d
� Z%d d � Z&d
d� Z'dd� Z(dd� Z)dd� Z*dd� Z+e&� Z,e'� Z-e(� Z.e)� Z/e*� Z0e�d�j1Z2G dd� de j3�Z4G dd� de j3�Z5G d d � d e j3�Z6G d d � d e j3�Z7G d d!� d!e j3�Z8G d"d#� d#�Z9G d$d%� d%e4e9�Z:G d&d'� d'e5e9�Z;G d(d)� d)e7e9�Z<G d*d+� d+e8e9�Z=e �>e dkd,�e �?e0d-�e �?e%� d.kd/�G d0d1� d1e j3e9����Z@G d2d3� d3e:�ZAG d4d5� d5e j3e9�ZBG d6d7� d7e j3�ZCG d8d9� d9eC�ZDG d:d;� d;e6eCe9�ZEG d<d=� d=eDeE�ZFG d>d?� d?eC�ZGG d@dA� dAeG�ZHG dBdC� dCeC�ZIG dDdE� dEeI�ZJG dFdG� dGeI�ZKG dHdI� dIeI�ZLG dJdK� dKeI�ZMG dLdM� dMeM�ZNdNdO� ZOdPdQ� ZPdRdS� ZQG dTdU� dUe j3�ZRe �?e,dV�G dWdX� dXe j3��ZSe �?e,dV�G dYdZ� dZe<��ZTe �?e-d[�G d\d]� d]e j3��ZUe �?e.d^�G d_d`� d`e j3��ZVe �?e.d^�G dadb� dbe=��ZWe �>e dkd,�e �?e0d-�G dcdd� dde j3���ZXG dedf� dfeA�ZYG dgdh� dhe;�ZZG didj� dje6�Z[G dkdl� dle[�Z\G dmdn� dne[�Z]G dodp� dpe[�Z^G dqdr� dre[�Z_G dsdt� dte[�Z`G dudv� dve[�ZaG dwdx� dxea�ZbG dydz� dzeb�ZcG d{d|� d|eb�ZdG d}d~� d~e[�ZeG dd�� d�ee�ZfG d�d�� d�ee�ZgG d�d�� d�e\ee�ZhG d�d�� d�e j3�ZiG d�d�� d�ea�ZjG d�d�� d�ea�ZkG d�d�� d�e]e_eEeK�ZlePejd��G d�d�� d�edel��ZmePejd��G d�d�� d�egel��ZnePejd��G d�d�� d�ehel��ZoG d�d�� d�e]e_eEeN�ZpePejd��e �?ejqd��eQd�d��G d�d�� d�edep����ZrePejd��e �?ejqd��eQd�d��G d�d�� d�egep����ZsePejd��e �?ejqd��eQd�d��G d�d�� d�ehep����ZtePejd��e �?ejqd��ePed��eQd�d��G d�d�� d�ekep�����ZuePejd��e �?ejqd��ePed��eQd�d��G d�d�� d�e\ekep�����ZvG d�d�� d�e`eFeJ�ZwePejd��G d�d�� d�ecew��ZxePejd��G d�d�� d�egefew��ZyePejd��G d�d�� d�ehefew��ZzG d�d�� d�e^e`eFeL�Z{ePejd��eQd�d�d��G d�d�� d�ece{���Z|ePejd��eQd�d�d��G d�d�� d�egefe{���Z}ePejd��eQd�d�d��G d�d�� d�ehefe{���Z~G d�d�� d�e`eFeH�ZePejd��ePed��G d�d�� d�ece���Z�ePejd��ePed��G d�d�� d�egefe���Z�ePejd��ePed��G d�d�� d�ehefe���Z�ePejd�d��ePed�d�dÃG d�dń d�eje���Z�ePejd�d��ePed�d�dÃG d�dDŽ d�e\eje���Z�G d�dɄ d�e j3�Z�ePedʃe �?e�ed˃� p�e�ed̃d͡G d�dτ d�e�eK���Z�ePedʃe �?e�ed˃� p�e�ed̃d͡G d�dф d�e�e6eDeJ���Z�G d�dӄ d�e:�Z�G d�dՄ d�eB�Z�G d�dׄ d�e:�Z�G d�dل d�eA�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�eY�Z�G d�d� d�e j3�Z�G d�d� d�e4e9�Z�G d�d� d�e4e9�Z�G d�d� d�e4�Z�G d�d� d�e5�Z�G d�d�� d�e j3�Z�e �?e j�d�kd��G d�d�� d�e j3��Z�e �?e�ed��d��G d�d�� d�e j3��Z�G d�d�� d�eA�Z�d�Z��d Z��dZ��d�d� Z�e �?e�� �d�G �d�d� �de j3��Z�e �?e�� �d�G �d�d� �de j3e9��Z�G �d �d
� �d
e:�Z�G �d �d � �d e j3�Z�e �?e�e�d
��d�G �d�d� �de j3��Z�e �?e
j��dk�d�e �?e �d�G �d�d� �de4���Z�G �d�d� �de:�Z�e �?e�e
�d��d�G �d�d� �de���Z�e �?e/�d �G �d �d � �d e j3��Z�e �?e j����d ��d �G �d!�d"� �d"e j3��Z��d#�d$� Z�e��d%k�
r$e�� dS (& � N)�support)�proxyFu Michael Gilfix was hereሴ
zutf-8g N@i� c C sn t d kr d S ttd�sd S y*tdd��} t �| tjd�}W d Q R X W n tk
rX 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 �,/usr/local/lib/python3.7/test/test_socket.py�get_cid/ s
r c C s> yt � t jt jt j�} W n ttfk
r0 dS X | �� dS )NFT)r �PF_CAN�SOCK_RAW�CAN_RAW�AttributeErrorr �close)�sr r r �_have_socket_can< s r c C s> yt � t jt jt j�} W n ttfk
r0 dS X | �� dS )NFT)r r �
SOCK_DGRAM� CAN_ISOTPr r r )r r r r �_have_socket_can_isotpF s r c C s< yt � t jt jd�} W n ttfk
r. dS X | �� dS )Nr FT)r �PF_RDS�SOCK_SEQPACKETr r r )r r r r �_have_socket_rdsP s r c C s< yt � t jt jd�} W n ttfk
r. dS X | �� dS )Nr FT)r �AF_ALGr r r r )r r r r �_have_socket_algZ s r" c C s t � d k } | S )N)r )�retr r r �_have_socket_vsockd s
r$ c C s t t�| tjtj�tj@ � S )N)�boolr �F_GETFL�os�
O_NONBLOCK)�sockr r r �_is_fd_in_blocking_modej s r* �ic @ s e Zd Zdd� Zdd� ZdS )�
SocketTCPTestc C s. t � t jt j�| _t�| j�| _| j�� d S )N)r �AF_INET� SOCK_STREAM�servr � bind_port�port�listen)�selfr r r �setUp~ s zSocketTCPTest.setUpc C s | j �� d | _ d S )N)r/ r )r3 r r r �tearDown� s
zSocketTCPTest.tearDownN)�__name__�
__module__� __qualname__r4 r5 r r r r r, | s r, c @ s e Zd Zdd� Zdd� ZdS )�
SocketUDPTestc C s$ t � t jt j�| _t�| j�| _d S )N)r r- r r/ r r0 r1 )r3 r r r r4 � s zSocketUDPTest.setUpc C s | j �� d | _ d S )N)r/ r )r3 r r r r5 � s
zSocketUDPTest.tearDownN)r6 r7 r8 r4 r5 r r r r r9 � s r9 c s4 e Zd Z� fdd�Z� fdd�Z� fdd�Z� ZS )�ThreadSafeCleanupTestCasec s t � j||� t�� | _d S )N)�super�__init__� threading�RLock�
_cleanup_lock)r3 �args�kwargs)� __class__r r r< � s z"ThreadSafeCleanupTestCase.__init__c s | j � t� j||�S Q R X d S )N)r? r; �
addCleanup)r3 r@ rA )rB r r rC � s z$ThreadSafeCleanupTestCase.addCleanupc s | j � t� j||�S Q R X d S )N)r? r; �
doCleanups)r3 r@ rA )rB r r rD � s z$ThreadSafeCleanupTestCase.doCleanups)r6 r7 r8 r< rC rD �
__classcell__r r )rB r r: � s r: 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� y| j�| jf� W n$ t k
r\ | �
d| j � Y nX d S )Nz%network interface `%s` does not exist) r r r r r rC r �bind� interfacer �skipTest)r3 r r r r4 � s zSocketCANTest.setUpN) r6 r7 r8 rL �bufsize�
can_frame_fmtr
�calcsizeZcan_frame_size�bcm_cmd_msg_fmtr4 r r r r rF � s
rF c @ s e Zd ZdZdd� ZdS )�
SocketRDSTesti c C sX t � t jt jd�| _| �| jj� yt�| j�| _W n t k
rR | �
d� Y nX d S )Nr zunable to bind RDS socket) r r r r/ rC r r r0 r1 r rM )r3 r r r r4 � s zSocketRDSTest.setUpN)r6 r7 r8 rN r4 r r r r rR � s rR 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 )N)r4 �_ThreadableTest__setUpr5 �_ThreadableTest__tearDown�_setUp� _tearDown)r3 r r r r<
s zThreadableTest.__init__c C s | j �� d S )N)� server_ready�set)r3 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&y | �� W n d| _ � Y nX W d | j�� X | j�� d S )N� F�.�_T)r Zwait_threads_exit� wait_threads� __enter__r= �EventrX � client_ready�done�queue�Queue�server_crashed�id�rfind�getattr�thread�start_new_thread� clientRunZ
client_threadrT rY �wait)r3 Z
methodnamer+ Z test_methodr r r rV s(
zThreadableTest._setUpc C s> | � � | j�� | j�d d d � | j�� r:| j�� }|�d S )N)rU rb rl r^ �__exit__rc Zqsize�get)r3 �excr r r rW 6 s
zThreadableTest._tearDownc
C s� | j �� zHy | �� W n6 tk
rN } z| j�|� | �� d S d }~X Y nX W d | j�� X | j rr| �� d S t
|d�s�t d��z>y
|� W n. tk
r� } z| j�|� W d d }~X Y nX W d | �� X d S )N�__call__z%test_func must be a callable function) rX rl � clientSetUp�
BaseExceptionrc �put�clientTearDownra rY re r � TypeError)r3 Z test_func�er r r rk ? s$
"zThreadableTest.clientRunc C s t d��d S )Nz clientSetUp must be implemented.)�NotImplementedError)r3 r r r rq U s zThreadableTest.clientSetUpc C s | j �� t�� d S )N)rb rY ri �exit)r3 r r r rt X s
z ThreadableTest.clientTearDownN)
r6 r7 r8 r< rZ rV rW rk rq rt r r r r rS � s & rS c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedTCPSocketTest�runTestc C s t j| |d� t�| � d S )N)�
methodName)r, r< rS )r3 r{ r r r r< ^ s z ThreadedTCPSocketTest.__init__c C s t � t jt j�| _d S )N)r r- r. �cli)r3 r r r rq b s z!ThreadedTCPSocketTest.clientSetUpc C s | j �� d | _ t�| � d S )N)r| r rS rt )r3 r r r rt e s
z$ThreadedTCPSocketTest.clientTearDownN)rz )r6 r7 r8 r< rq rt r r r r ry \ s
ry c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedUDPSocketTestrz c C s t j| |d� t�| � d S )N)r{ )r9 r< rS )r3 r{ r r r r< l s z ThreadedUDPSocketTest.__init__c C s t � t jt j�| _d S )N)r r- r r| )r3 r r r rq p s z!ThreadedUDPSocketTest.clientSetUpc C s | j �� d | _ t�| � d S )N)r| r rS rt )r3 r r r rt s s
z$ThreadedUDPSocketTest.clientTearDownN)rz )r6 r7 r8 r< rq rt r r r r r} j s
r} c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedCANSocketTestrz c C s t j| |d� t�| � d S )N)r{ )rF r< rS )r3 r{ r r r r< z s z ThreadedCANSocketTest.__init__c C sD t � t jt jt j�| _y| j�| jf� W n tk
r> Y nX d S )N)r r r r r| rK rL r )r3 r r r rq ~ s
z!ThreadedCANSocketTest.clientSetUpc C s | j �� d | _ t�| � d S )N)r| r rS rt )r3 r r r rt � s
z$ThreadedCANSocketTest.clientTearDownN)rz )r6 r7 r8 r< rq rt r r r r r~ x s
r~ c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedRDSSocketTestrz c C s t j| |d� t�| � d S )N)r{ )rR r< rS )r3 r{ r r r r< � s z ThreadedRDSSocketTest.__init__c C sN t � t jt jd�| _y | j�tdf� | j�� | _W n tk
rH Y nX d S )Nr ) r r r r| rK �HOST� getsockname�cli_addrr )r3 r r r rq � s z!ThreadedRDSSocketTest.clientSetUpc C s | j �� d | _ t�| � d S )N)r| r rS rt )r3 r r r rt � s
z$ThreadedRDSSocketTest.clientTearDownN)rz )r6 r7 r8 r< rq rt r r r r r � s
r z
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 )� ThreadedVSOCKSocketStreamTestrz c C s t jj| |d� t�| � d S )N)r{ )�unittest�TestCaser< rS )r3 r{ r r r r< � s z&ThreadedVSOCKSocketStreamTest.__init__c C sh t � t jt j�| _| �| jj� | j�t jtf� | j� � | �
� | j� � \| _ | _
| �| j j� d S )N)r �AF_VSOCKr. r/ rC r rK �VMADDR_CID_ANY� VSOCKPORTr2 rZ �accept�conn�connaddr)r3 r r r r4 � s
z#ThreadedVSOCKSocketStreamTest.setUpc C sD t �d� t�tjtj�| _| �| jj� t� }| j� |t
f� d S )Ng�������?) �time�sleepr r� r. r| rC r r �connectr� )r3 Zcidr r r rq � s
z)ThreadedVSOCKSocketStreamTest.clientSetUpc C s | j �d�}| �|t� d S )Ni )r� �recv� assertEqual�MSG)r3 �msgr r r �
testStream� s z(ThreadedVSOCKSocketStreamTest.testStreamc C s | j �t� | j �� d S )N)r| �sendr� r )r3 r r r � _testStream� s z)ThreadedVSOCKSocketStreamTest._testStreamN)rz )r6 r7 r8 r< r4 rq r� r� r r r r r� � s
r� c @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )�SocketConnectedTestrz c C s t j| |d� d S )N)r{ )ry r< )r3 r{ r r r r< � s z SocketConnectedTest.__init__c C s* t �| � | �� | j�� \}}|| _d S )N)ry r4 rZ r/ r� �cli_conn)r3 r� �addrr r r r4 � s
zSocketConnectedTest.setUpc C s | j �� d | _ t�| � d S )N)r� r ry r5 )r3 r r r r5 � s
z SocketConnectedTest.tearDownc C s( t �| � | j�t| jf� | j| _d S )N)ry rq r| r� r� r1 � serv_conn)r3 r r r rq � s
z SocketConnectedTest.clientSetUpc C s | j �� d | _ t�| � d S )N)r� r ry rt )r3 r r r rt � s
z"SocketConnectedTest.clientTearDownN)rz )r6 r7 r8 r< r4 r5 rq rt r r r r r� � s
r� c @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )�SocketPairTestrz c C s t jj| |d� t�| � d S )N)r{ )r� r� r< rS )r3 r{ r r r r< � s zSocketPairTest.__init__c C s t �� \| _| _d S )N)r �
socketpairr/ r| )r3 r r r r4 � s zSocketPairTest.setUpc C s | j �� d | _ d S )N)r/ r )r3 r r r r5 � s
zSocketPairTest.tearDownc C s d S )Nr )r3 r r r rq � s zSocketPairTest.clientSetUpc C s | j �� d | _ t�| � d S )N)r| r rS rt )r3 r r r rt � s
z SocketPairTest.clientTearDownN)rz )r6 r7 r8 r< r4 r5 rq rt r r r r r� � s
r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�SocketTestBasec C s | � � | _| �� d S )N)� newSocketr/ �
bindServer)r3 r r r r4 s
zSocketTestBase.setUpc C s | � | j� | j�� | _d S )N)�bindSockr/ r� � serv_addr)r3 r r r r� s zSocketTestBase.bindServerc C s | j �� d | _ d S )N)r/ r )r3 r r r r5 s
zSocketTestBase.tearDownN)r6 r7 r8 r4 r� r5 r r r r r� s r� c s e Zd Z� fdd�Z� ZS )�SocketListeningTestMixinc s t � �� | j�� d S )N)r; r4 r/ r2 )r3 )rB r r r4 s
z SocketListeningTestMixin.setUp)r6 r7 r8 r4 rE r r )rB r r� s r� c s< e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Z� ZS ) �ThreadedSocketTestMixinc s t � j||� t�| � d S )N)r; r< rS )r3 r@ rA )rB r r r< ) s z ThreadedSocketTestMixin.__init__c C s | � � | _| �� d S )N)�newClientSocketr| �
bindClient)r3 r r r rq - s
z#ThreadedSocketTestMixin.clientSetUpc C s | � � S )N)r� )r3 r r r r� 1 s z'ThreadedSocketTestMixin.newClientSocketc C s | � | j� | j�� | _d S )N)r� r| r� r� )r3 r r r r� 5 s z"ThreadedSocketTestMixin.bindClientc C s | j �� d | _ t�| � d S )N)r| r rS rt )r3 r r r rt : s
z&ThreadedSocketTestMixin.clientTearDown) r6 r7 r8 r< rq r� r� rt rE r r )rB r r� ! s
r� c s@ e Zd Z� fdd�Z� fdd�Z� fdd�Z� fdd�Z� ZS ) �ConnectedStreamTestMixinc s* t � �� | �� | j�� \}}|| _d S )N)r; r4 rZ r/ r� r� )r3 r� r� )rB r r r4 I s
z ConnectedStreamTestMixin.setUpc s | j �� d | _ t� �� d S )N)r� r r; r5 )r3 )rB r r r5 Q s
z!ConnectedStreamTestMixin.tearDownc s$ t � �� | j�| j� | j| _d S )N)r; rq r| r� r� r� )r3 )rB r r rq V s
z$ConnectedStreamTestMixin.clientSetUpc s8 y| j �� d | _ W n tk
r( Y nX t� �� d S )N)r� r r r; rt )r3 )rB r r rt [ s
z'ConnectedStreamTestMixin.clientTearDown)r6 r7 r8 r4 r5 rq rt rE r r )rB r r� @ s r� c s$ e Zd Z� fdd�Zdd� Z� ZS )�UnixSocketTestBasec s( t �� | _| �tj| j� t� �� d S )N)�tempfile�mkdtemp�dir_pathrC r' �rmdirr; r4 )r3 )rB r r r4 l s
zUnixSocketTestBase.setUpc C s, t j| jd�}t�||� | �tj|� d S )N)�dir)r� Zmktempr� r �bind_unix_socketrC �unlink)r3 r) �pathr r r r� q s zUnixSocketTestBase.bindSock)r6 r7 r8 r4 r� rE r r )rB r r� d s r� c @ s e Zd Zdd� ZdS )�UnixStreamBasec C s t � t jt j�S )N)r �AF_UNIXr. )r3 r r r r� y s zUnixStreamBase.newSocketN)r6 r7 r8 r� r r r r r� v s r� c s( e Zd ZeZ� fdd�Zdd� Z� ZS )� InetTestBasec s t � �� | jd | _d S )Nr[ )r; r4 r� r1 )r3 )rB r r r4 � s
zInetTestBase.setUpc C s t j|| jd� d S )N)�host)r r0 r� )r3 r) r r r r� � s zInetTestBase.bindSock)r6 r7 r8 r� r� r4 r� rE r r )rB r r� } s r� c @ s e Zd Zdd� ZdS )� TCPTestBasec C s t � t jt j�S )N)r r- r. )r3 r r r r� � s zTCPTestBase.newSocketN)r6 r7 r8 r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )� UDPTestBasec C s t � t jt j�S )N)r r- r )r3 r r r r� � s zUDPTestBase.newSocketN)r6 r7 r8 r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�SCTPStreamBasec C s t � t jt jt j�S )N)r r- r. � IPPROTO_SCTP)r3 r r r r� � s zSCTPStreamBase.newSocketN)r6 r7 r8 r� r r r r r� � s r� c @ s e Zd ZejZdS )�
Inet6TestBaseN)r6 r7 r8 r �HOSTv6r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )� UDP6TestBasec C s t � t jt j�S )N)r �AF_INET6r )r3 r r r r� � s zUDP6TestBase.newSocketN)r6 r7 r8 r� r r r r r� � s r� c s* dd� � � �fdd�}dd� }| r&|S |S )Nc _ s d S )Nr )r@ rA r r r � client_pass� s z%skipWithClientIf.<locals>.client_passc s* t ���| �}t| t�s&� fdd�|_|S )Nc s � S )Nr )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 )Nr )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� r �skipWithClientIf� s r� c s0 � fdd�|D �}t |dd�dd� |D �� �S )Nc s g | ]}t � |�s|�qS r )r )�.0�name)r� r r �
<listcomp>� s z requireAttrs.<locals>.<listcomp>z don't have z, c s s | ]
}|V qd S )Nr )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 �}ytj|� }W n* tk
rv } z t|�}W d 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>z don't have z, c s s | ]
}|V qd S )Nr )r� r� r r r r� � s z requireSocket.<locals>.<genexpr>c S s$ g | ] }t |t�r tt|�n|�qS r )r� r� rh r )r� r� r r r r� � s z can't create socket({0}): {1}c s s | ]}t |�V qd S )N)r� )r� �or r r r� � s )r� r r r� r r� �format)r@ �errr� Zcallargsr rv r r r �
requireSocket� s r� c @ s� e Zd Zdd� Zdd� Ze�edk d�dd� �Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Ze�eed�d�e�eed�d�dd � ��Ze�eed �d �d d � �Ze�eed �d �d!d"� �Ze�eed#�d$�d%d&� �Zd'd(� Zd)d*� Zejd+d,� �Zd-d.� Z d/d0� Z e�eed1�d2�d3d4� �Z e�eed5�d6�d7d8� �Z e�eed5�d6�d9d:� �Z e�eed;�d<�d=d>� �Z!e�eed;�d<�d?d@� �Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)e�e*j+dOkdP�dQdR� �Z,e�e*j+dOkdP�e�eedS�dT�dUdV� ��Z-dWdX� Z.dYdZ� Z/e�e�0d[�d\�d]d^� �Z1d_d`� Z2dadb� Z3dcdd� Z4dedf� Z5dgdh� Z6didj� Z7dkdl� Z8dmdn� Z9dodp� Z:dqdr� Z;dsdt� Z<ejdudv� �Z=e�ej>dw�dxdy� �Z?e�ej>dw�dzd{� �Z@e�ej>dw�e�eed �d|�d}d~� ��ZAe�ej>dw�e�ejBdkd��d�d�� ��ZCe�ej>dw�e�eed �d|�d�d�� ��ZDe�ej>dw�e�ejBdkd��d�d�� ��ZEd�d�� ZFd�d�� ZGe�He*j+dOkd��d�d�� �ZIe�ee*d��d��d�d�� �ZJd�d�� ZKd�d�� ZLdS )��GeneralModuleTestsc C s< dd l }| �tj|jk� t�� }| �|tj� |�� d S )Nr )�_socket�
assertTruer Z
SocketType�assertIsInstancer )r3 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 d Q R X | �dt|�� | �dt|�� d S ) Nzfd=%iz family=%sztype=%szproto=0Zraddr)z 127.0.0.1r Zladdrz[closed])
r r- r. �assertIn�fileno�repr� assertNotInrK r� r� )r3 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 d |�
� 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 r- r. r� �familyr� �protor� r� r )r3 r �expectedr r r �test_csocket_repr s
z$GeneralModuleTests.test_csocket_reprc C sh t � t jt j�}t|�}| �|�� |�� � |�� d }y |�� W n tk
rX Y n X | �d� d S )NzSocket proxy still exists) r r- r. r r� r� r �ReferenceError�fail)r3 r �pr r r � test_weakref s z GeneralModuleTests.test_weakrefc C sr d}| j t|d d��
t�W d Q R X | j t|d d�� tj�W d Q R X | j t|d d�� tj�W d Q R X d S )Nz$Error raising socket exception (%s).r )r� z
socket.herrorzsocket.gaierror)� assertRaisesr r �herror�gaierror)r3 r� r r r �testSocketError s z"GeneralModuleTests.testSocketErrorc C sF t � t jt j�}| �|j� |�d� |�� }| �t��}|� d|� W d Q R X | �
t |j �d� | �t��}|� d|� W d Q R X | �
t |j �d� | �t��}|� dd � W d Q R X | �
dt |j �� | �t��}|� dd|� W d Q R X | �
t |j �d� | �t��}|� dd|� W d Q R X | �
t |j �d� | �t��}|� ddd � W d Q R X | �
dt |j �� | �t��}|� dd |� W d Q R X | �
d
t |j �� | �t��}|� dd d � W d Q R X | �
d
t |j �� | �t��}|� d� W d Q R X | �
d t |j �� | �t��}|� dd|d � W d 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's fooz not NoneTyper �barzan integer is requiredz (1 given)� z (4 given))r r- r rC r rK r� r� ru �sendtor� r� � exceptionr� )r3 r Zsockname�cmr r r �testSendtoErrors& sL
z#GeneralModuleTests.testSendtoErrorsc C s4 t j t j t j t j t j t j t j t j d S )N) r r- r. r r �SOCK_RDMr �
SOL_SOCKET� SO_REUSEADDR)r3 r r r �testCrucialConstantsR s z'GeneralModuleTests.testCrucialConstantsc C s� t �� }yt �|�}W n tk
r4 | �d� Y nX | �|�d�dkd� yt �|�\}}}W n tk
r~ | �d� Y nX ||g| }t �|�}||kr�| � d|t
|�f � d S )Nzname lookup failurer\ r zError resolving host to ip.z=Error testing host resolution mechanisms. (fqdn: %s, all: %s)) r � gethostname�
gethostbynamer rM r� �find�
gethostbyaddrZgetfqdnr� r� )r3 ZhostnameZipZhname�aliasesZipaddrsZall_host_namesZfqhnr r r �testHostnameRes] s
z"GeneralModuleTests.testHostnameResc C sT x&t jddgD ]}| �t�|�|� qW x&t jgD ]}| �|t�|�d � q2W d S )Nz10.0.0.1z255.255.255.255r� )r ZHOSTv4r� r r� r� r� )r3 r� r� r r r �test_host_resolutionp s z'GeneralModuleTests.test_host_resolutionc
C s^ d}xTdD ]L}| j t|d�� t�|� W d Q R X | j t|d�� t�|� W d Q R X q
W d 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:1)r� )r� r r r� r� )r3 Z explanationr� r r r � test_host_resolution_bad_addressy s
z3GeneralModuleTests.test_host_resolution_bad_address� sethostnamez test needs socket.sethostname()r� z test needs socket.gethostname()c
C s� t �� }yt �d� W n< tk
rR } z |jtjkr@| �d� n� W d d }~X Y nX z.| �t �� d� t �d� | �t �� d� W d t �|� X d S )N�newztest should be run as roots barr� )r r� r� r �errnoZEPERMrM r� )r3 Zoldhnrv r r r �test_sethostname� s
z#GeneralModuleTests.test_sethostname� if_nameindexz$socket.if_nameindex() not available.c C s� t �� }xx|D ]p\}}| �|t� | �|t� | �|d� t �|�}| �|t� | �||� t �|�}| �|t� | �||� qW d S )Nr ) r r r� �intr� �
assertGreater�if_nametoindexr� �if_indextoname)r3 Z
interfaces�indexr� Z_index�_namer r r �testInterfaceNameIndex� s
z)GeneralModuleTests.testInterfaceNameIndexc C sD | � ttjd� | � ttjd� | � ttjd� | � ttjd� d S )Nr Z _DEADBEEF)r� r r r r ru )r3 r r r � testInvalidInterfaceNameIndex� s z0GeneralModuleTests.testInvalidInterfaceNameIndex� getrefcountz test needs sys.getrefcount()c C sL yt �t�}t�td� W n, tk
rF t �t�|krB| �d� Y nX d S )Nr z$socket.getnameinfo loses a reference)�sysr
r6 r � getnameinforu r� )r3 �origr r r �testRefCountGetNameInfo� s
z*GeneralModuleTests.testRefCountGetNameInfoc C s* yt �dd� W n tk
r$ Y nX d S )N)rI r r r r )r r r )r3 r r r �testInterpreterCrash� s z'GeneralModuleTests.testInterpreterCrashc C s� t jdt jdt jdt jdi}xt|�� D ]h\}}d|> d }x,dD ]$}| �||@ ||||@ ��|@ � q@W ||�}| �||@ |� | �t|d� q&W d S )N� � r[ )r r[ i�� i ��r� igE#i2Tvl ) r �htonl�ntohl�htons�ntohs�itemsr� r� �
OverflowError)r3 Zsizes�func�size�maskr+ Zswappedr r r �testNtoH� s
$zGeneralModuleTests.testNtoHc C s" dd l }ddddg}|dg }ddddd > g}||jd |jd g }d
|jg}x |D ]}t�|� t�|� qVW x |D ]}t�|� t�|� qxW x,|D ]$}| �t tj|� | �t tj|� q�W x,|D ]$}| �t tj|� | �t tj|� q�W x,|D ]$}| �
t tj|� | �
t tj|� q�W d S ) Nr r[ r� i�� l �� ������l i� i ) � _testcapiZINT_MIN�INT_MAXr r r r r r� r � assertWarns�DeprecationWarning)r3 r 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}x8|D ],}yt�|d�}P W q, tk
rV Y q,X q,W t�ttd�rvt�� dkr�t�|�}|||� yt�|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)Zechor& r' Ztcp�getandroidapilevel� Zudpr i )r� r �platform�
startswithr Z
getservbynamer r r( r Z
is_androidZ
getservbyportr� r )r3 �eqZservicesZservicer1 Zport2Zudpportr r r �
testGetServBy� s8
z GeneralModuleTests.testGetServByc C s� | � t�� d � t�� }| � |�� d � |�� t�d� | � t�� d� t�� }| � |�� d� |�� t�d � | � t�� d � t�� }| � |�� d � |�� | �ttjd� | �ttjd� d S )N�
r Zspam) r� r �getdefaulttimeout�
gettimeoutr �setdefaulttimeoutr� �
ValueErrorru )r3 r r r r �testDefaultTimeout s
z%GeneralModuleTests.testDefaultTimeout� inet_atonz test needs socket.inet_aton()c C s( | � dt�d�� | � dt�d�� d S )Ns z0.0.0.0s ����z255.255.255.255)r� r r4 )r3 r r r � testIPv4_inet_aton_fourbytes9 s z/GeneralModuleTests.testIPv4_inet_aton_fourbytes� inet_ptonz test needs socket.inet_pton()c s2 ddl m}m�m� � �fdd�}�fdd�}��d|d�� ��d|d �� ��d
|d �� ��d |d
�� ��d|d�� tj�d�s�||d� ||d� ||d� ||d� ||d� ��d|d�� ��d|d �� ��d
|d �� ��d|d�� ||d� ||d� ||d� ||d� ||d� d S )Nr )r4 r6 r- c s
�� | �S )Nr )�a)r- r6 r r r� E r� z5GeneralModuleTests.testIPv4toString.<locals>.<lambda>c s � � ttf| |�S )N)r� r r2 )r r7 )r3 r r r� G s s z0.0.0.0s � � z 255.0.255.0s ����z170.170.170.170s z1.2.3.4s ����z255.255.255.255�aixz0.0.0.z 300.0.0.0za.0.0.0z 1.2.3.4.5z::1)r r4 r6 r- r� r r* r+ )r3 r �g�
assertInvalidr )r- r6 r3 r �testIPv4toStringA s.
z#GeneralModuleTests.testIPv4toStringc
s� y&ddl m�m� m} |s$��d� W n tk
rD ��d� Y nX tjdkr�y�� d� W n6 tk
r� } z|j dkr���d� W d 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 � tj� d ��sh|d � |d � ��
d �d!�� ��
d"�d#�� ��
d$�d%�� |d&� |d'� |d(� |d)� |d*� |d+� d S ),Nr )r6 r� �has_ipv6zIPv6 not availablez+could not import needed symbols from socket�win32z::i&' zIPv6 might not be supportedc s
�� | �S )Nr )r7 )r� r6 r r r� s r� z5GeneralModuleTests.testIPv6toString.<locals>.<lambda>c s �� ttf� | �S )N)r� r r2 )r7 )r r3 r r r� t s 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:6z1:2:3:4:5:6:7:8:0r8 z 1:2:3:4:5:6:z1: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 r6 r� r< rM � ImportErrorr r* r �winerrorr� r+ )r3 r< rv r: r )r� r r6 r3 r �testIPv6toStringb s`
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_ntoarA r- c s
�� | �S )Nr )r7 )r- rA r r r� � r� z5GeneralModuleTests.testStringToIPv4.<locals>.<lambda>c s � � ttf| |�S )N)r� r r2 )r r7 )r3 r r r� � s z1.0.1.0s z
170.85.170.85s �U�Uz255.255.255.255s ����z1.2.3.4s s s s )r rB rA r- r� � bytearray)r3 r r9 r: r )r- rA r3 r �testStringToIPv4� s$
z#GeneralModuleTests.testStringToIPv4c
s y&ddl m�m� m} |s$��d� W n tk
rD ��d� Y nX tjdkr�y�� d� W n6 tk
r� } z|j dkr���d� W d d }~X Y nX � �fd d
����fd d
�}��
d �d�� ��
d
�d�� ��
d�d�� ��
d
�t d��� |d� |d� |d� d S )Nr )rA r� r< zIPv6 not availablez+could not import needed symbols from socketr= s i&' zIPv6 might not be supportedc s
�� | �S )Nr )r7 )r� rA r r r� � r� z5GeneralModuleTests.testStringToIPv6.<locals>.<lambda>c s �� ttf� | �S )N)r� r r2 )r7 )r r3 r r r� � s z::z::1s z!aef:b01:506:1001:ffff:9997:55:170s
� ���� Ups s s ) r rA r� r< rM r> r r* r r? r� rC )r3 r< rv r: r )r� r rA r3 r �testStringToIPv6� s.
z#GeneralModuleTests.testStringToIPv6c C s� t �� }t�tjtj�}| �|j� |�d|f� |�� }yt� t�
� �}W n t k
rj | � d� Y nX | �
|d d|fd|d � | �|d |� d S )Nz0.0.0.0zname lookup failurer z
%s invalidr[ )r �find_unused_portr r- r. rC r rK r� r� r� r rM r� r� )r3 r1 r) r� Z
my_ip_addrr r r � testSockName� s z GeneralModuleTests.testSockNamec C s@ t � t jt j�}| �|j� |�t jt j�}| �|dkd� d S )Nr zinitial mode is reuse) r r- r. rC r �
getsockoptr� r� � assertFalse)r3 r) �reuser r r �testGetSockOpt� s z!GeneralModuleTests.testGetSockOptc C sR t � t jt j�}| �|j� |�t jt jd� |�t jt j�}| � |dkd� d S )Nr[ r zfailed to set reuse mode)
r r- r. rC r �
setsockoptr� r� rH rI )r3 r) rJ r r r �testSetSockOpt� s
z!GeneralModuleTests.testSetSockOptc C s6 t � t jt j�}|�d� |�� | �t|jd� d S )Nr[ s spam)r r- r. �
settimeoutr r� r r� )r3 r) r r r �testSendAfterClose s
z%GeneralModuleTests.testSendAfterClosec
C s� t � � }|�t jdf� t j |�� d��� y |�� W n8 tk
rp } z| �|jtjtj f� W d d }~X Y n X | �
d� d S )Nr )r� z#close() should raise EBADF/ENOTSOCK) r rK Z
_LOCALHOSTr� r r r� r� ZEBADFZENOTSOCKr� )r3 r) r� r r r �testCloseException s (z%GeneralModuleTests.testCloseExceptionc C sr t � t jt j�}| �|jt j� tt d�rH| �|jt jt jB t jf� n| �|jt j� | �|j d� |�
� d S )N� SOCK_CLOEXECr ) r r- r. r� r� r r� r� rQ r� r )r3 r) r r r �testNewAttributes s
z$GeneralModuleTests.testNewAttributesc C s� t � � }| �|j� t�� }|d }|d }| �t|jt|f� | �t|jt|f� xft �
� D ]Z}t�� }y|�t|f� W n8 t k
r� } z|j t j
ks�|dkr�� W d d }~X Y q^X P q^W d S )Ni � )r rC r r rF r� r rK r� � itertools�countr r� Z
EADDRINUSE)r3 r) r1 Zbig_portZneg_portr+ rv r r r �test_getsockaddrarg# s z&GeneralModuleTests.test_getsockaddrarg�ntzWindows specificc C s� | � ttjd�� | � ttd�� | � ttd�� | � ttd�� | � ttd�� t�� }| �|j� | �t|jdd � |�tjd� d S )Nr Z
SIO_RCVALLZ RCVALL_ONZ
RCVALL_OFF�SIO_KEEPALIVE_VALSr )r[ �d rY ) r� r r rC r r� r2 r rX )r3 r r r r �test_sock_ioctl7 s z"GeneralModuleTests.test_sock_ioctl�SIO_LOOPBACK_FAST_PATHz1Loopback fast path support required for this testc
C s| t � � }| �|j� y|�t jd� W n< tk
rb } z d}|j|krP| �d� � W d 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 rC r r r[ r r? rM r� ru )r3 r ro Z
WSAEOPNOTSUPPr r r �test_sio_loopback_fast_pathC s
z.GeneralModuleTests.test_sio_loopback_fast_pathc
C sL yt �dd� W n< t jk
rL } z |jt jkr:| �d� � W d d }~X Y nX x$t �td �D ]}| �t|�d� q\W t �dd� t �dd� t �d d� t j
r�t �dd� t t d�r�t �
� dkr�t �td � t �td� t �td � t �tdt jt j�}xR|D ]J\}}}}}| �|t j� | �t|�d
� | �|t j� | �t|�d � q�W t �td d t j�}x&|D ] \}}}}}| �|t j� �qdW t �td d d t j� t �td d d d t j� t �d d t jt jd t j� t �td �}t jtd d
�} | �|| � t �td t j�}t jtd t jd�} | �|| � t �td d t j�}t jtd t jd�} | �|| � t �td d d t j�}t jtd t jd�} | �|| � t �td d d d t j�}t jtd t jd�} | �|| � t �d d t jt jd t j�}t jd d t jt jd t jd�} | �|| � | �tt jdd� t t d��rHyt �dd d d d t j� W n t jk
�rF Y nX d S )NZ localhost�P zbuggy libc versionrS z 127.0.0.1z::1r( r) ZhttpzAddressFamily.AF_INETzSocketKind.SOCK_STREAMr )r� r1 )r� )r� )r� )�flags)r� r1 r� r� r� r^ u ��AI_NUMERICSERV)r � getaddrinfor� r� Z EAI_SERVICErM r� r� �lenr � IPV6_ENABLEDr r r( r- r. r� ZSOL_TCPZ
AI_PASSIVE� AF_UNSPECr� �UnicodeEncodeErrorr_ )
r3 r� �infoZinfosr� r� r] Zsocktyper7 �br r r �testGetaddrinfoS sv
z"GeneralModuleTests.testGetaddrinfoc C s | � ttjdd� d S )N)zmail.python.orgr r )r� r r r )r3 r r r �test_getnameinfo� s z#GeneralModuleTests.test_getnameinfoZnetworkznetwork is not enabledc C sP t �d�� t�d� W d Q R X d}t�|� t�|� t�|dtjtj� d S )Nz
python.orgu! испытание.pythontest.netr )r Ztransient_internetr r� Zgethostbyname_exr` rc r. )r3 r' 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 d Q R X |r�t�tj|� t� d � � �tj
|j
d
t j � W d t� d � t�tj|� |�� |�� X d S ) N�alarmr� z9signal.alarm and socket.socketpair required for this testc s � � ttjd� d S )Nr )r� r2 �math�acosh)r@ )r3 r r �
ok_handler� s z@GeneralModuleTests.check_sendall_interrupted.<locals>.ok_handlerc s � � ttjd� dd d S )Nr r[ )r� r2 rk rl )r@ )r3 r r �raising_handler� s zEGeneralModuleTests.check_sendall_interrupted.<locals>.raising_handlerg �?r[ � xr )r �signalr rM r� �SIGALRMrN r� �ZeroDivisionErrorrj �sendallr �
SOCK_MAX_SIZE�timeoutr )r3 Z with_timeoutrm rn �cr � old_alarmr )r3 r �check_sendall_interrupted� s*
z,GeneralModuleTests.check_sendall_interruptedc C s | � d� d S )NF)rx )r3 r r r �test_sendall_interrupted� s z+GeneralModuleTests.test_sendall_interruptedc C s | � d� d S )NT)rx )r3 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 d Q R X | �|t |j
j d �� t � t jt j�}|� d�}t|�}d }t�� | �t�� d }t�� W d Q R X d S )Nr r )
r r- r. r� r �ResourceWarningr Z
gc_collectr� r� Zwarningr@ �makefile)r3 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 d Q R X d S )Nr z <_io.BufferedReader name=-1>)r r- r. r| r r� r� )r3 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 d Q R X d S )Nr r )� buffering)
r r| r� �readablerI �writable�seekabler r� r2 )r3 r) r~ r r r � test_unusable_closed_socketio� s
z0GeneralModuleTests.test_unusable_closed_socketioc C s� t � � }z@|�tdf� t �|�� � | �t�� |�d� W d Q R X W d | �t�� |�� W d Q R X X | �t�� t �d � W d Q R X | �t�� t �d� W d Q R X d S )Nr r[ r ) r rK r� r r� r� r r2 ru )r3 r) r r r �test_socket_close� s z$GeneralModuleTests.test_socket_closec C sb x\dD ]T}| j |d��> t�� �*}|�|��}| �|j|� W d Q R X W d Q R X W d Q R X qW d S )N)r r �rw�w�wb)�mode)�subTestr r| r� r� )r3 r� r) r~ r r r �test_makefile_mode
s
z%GeneralModuleTests.test_makefile_modec C s` xZdD ]R}| j |d��< t�� �(}| �td�� |�|� W d Q R X W d Q R X W d Q R X qW d S )N)�rtrI �+r7 )r� z invalid mode)r� r �assertRaisesRegexr2 r| )r3 r� r) r r r �test_makefile_invalid_mode s
z-GeneralModuleTests.test_makefile_invalid_modec C s� t � � }|�0 x(ttjd �D ]}| �ttj||� q W W d Q R X xZttjd �D ]H}t�t�t j|��}| � |t j� t�t�t j
|��}| � |t j
� qRW d S )Nr[ ) r �range�pickleZHIGHEST_PROTOCOLr� ru �dumps�loadsr- r� r. )r3 r) Zprotocolr� r� r r r � test_pickle s z GeneralModuleTests.test_picklec
C sx x@dD ]8}t � t jt j�� }|�tdf� |�|� W d Q R X qW t � t jt j�� }|�tdf� |�� W d Q R X d S )N)r r r )r r- r. rK r� r2 )r3 Zbacklog�srvr r r �test_listen_backlog& s
z&GeneralModuleTests.test_listen_backlogc C sH dd l }t�tjtj�}|�tdf� | �t|j|j d � |�
� d S )Nr r[ ) r r r- r. rK r� r� r r2 r r )r3 r r� r r r � test_listen_backlog_overflow0 s
z/GeneralModuleTests.test_listen_backlog_overflowz IPv6 required for this test.c C sR | � ttjtjddfd� t�tjtj�� }| � t|jtjddf� W d Q R X d S )Nr l �� i����) r� r r r r r� r� r. rK )r3 r r r r �
test_flowinfo9 s
z GeneralModuleTests.test_flowinfoc C s0 t �ddt jt jt j�\�^ }}| �|d� d S )Nzff02::1de:c0:face:8Di� )zff02::1de:c0:face:8di� r r )r r` r� r � IPPROTO_UDPr� )r3 r] �sockaddrr r r �test_getaddrinfo_ipv6_basic@ s z.GeneralModuleTests.test_getaddrinfo_ipv6_basicz if_nameindex is not supportedc C sL t �� d \}}t �d| dt jt jt j�\�^ }}| �|ddd|f� d S )Nr zff02::1de:c0:face:8D%i� zff02::1de:c0:face:8d)r r r` r� r r� r� )r3 �ifindex�test_interfacer] r� r r r �&test_getaddrinfo_ipv6_scopeid_symbolicJ s z9GeneralModuleTests.test_getaddrinfo_ipv6_scopeid_symbolicr= z.Numeric scope id does not work or undocumentedc C sD d}t �dt|� dt jt jt j�\�^ }}| �|ddd|f� d S )N�* zff02::1de:c0:face:8D%i� zff02::1de:c0:face:8dr )r r` r� r� r r� r� )r3 r� r] r� r r r �%test_getaddrinfo_ipv6_scopeid_numericZ s
z8GeneralModuleTests.test_getaddrinfo_ipv6_scopeid_numericc C sH t �� d \}}ddd|f}t �|t jt jB �}| �|d| df� d S )Nr zff02::1de:c0:face:8Di� zff02::1de:c0:face:8d%�1234)r r r �NI_NUMERICHOST�NI_NUMERICSERVr� )r3 r� r� r� �nameinfor r r �&test_getnameinfo_ipv6_scopeid_symbolick s z9GeneralModuleTests.test_getnameinfo_ipv6_scopeid_symbolicc C s@ d}ddd|f}t �|t jt jB �}| �|dt|� df� d S )Nr� zff02::1de:c0:face:8Di� r zff02::1de:c0:face:8d%r� )r r r� r� r� r� )r3 r� r� r� r r r �%test_getnameinfo_ipv6_scopeid_numericv s z8GeneralModuleTests.test_getnameinfo_ipv6_scopeid_numericc C sD t � t jt j��*}| �t|j�d� | �t|j�d� W d Q R X d S )NzAddressFamily.AF_INETzSocketKind.SOCK_STREAM)r r- r. r� r� r� r� )r3 r r r r �test_str_for_enums� s z%GeneralModuleTests.test_str_for_enumsc C s� t tdd�}t tdd�}tj|B |B }t�tj|��~}| �|jtj� |�d� | �|jtj� |�d� | �|jtj� |�d� | �|jtj� |�d� | �|jtj� W d Q R X d S )N�
SOCK_NONBLOCKr rQ r[ TF)rh r r. r- r� r� rN � setblocking)r3 r� rQ Z sock_typer r r r � test_socket_consistent_sock_type� s
z3GeneralModuleTests.test_socket_consistent_sock_typezWill not work on Windowsc C s� t �� \}}| �tj|� ttjj� � �d }tdd� tj
j� � D ��d }tj||d|d��4}| � |j
|� | � |j|� | �|jddh� W d Q R X d S )Nr[ c s s | ]\}}|d kr|V qdS )> rQ r� Nr )r� r� �kindr r r r� � s zEGeneralModuleTests.test_unknown_socket_family_repr.<locals>.<genexpr>r) )r� r� r� r� r )r� �mkstemprC r' r� �maxr Z
AddressFamily� __members__�valuesZ
SocketKindr r� r� r� r� r� )r3 �fdr� Zunknown_familyZ unknown_typer r r r � test_unknown_socket_family_repr� s
z2GeneralModuleTests.test_unknown_socket_family_repr�sendfileztest needs os.sendfile()c
C s� G dd� d�}t � � ��}t�tjtj�}t�|� | �t j�� |�||�� W d Q R X | �t �� |�|dd �� W d Q R X | �t
�� |�|d �� W d Q R X W d Q R X d S )Nc @ s e Zd Zdd� Zdd� ZdS )z<GeneralModuleTests.test__sendfile_use_sendfile.<locals>.Filec S s
|| _ d S )N)r� )r3 r� r r r r< � s zEGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.__init__c S s | j S )N)r� )r3 r r r r� � s zCGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.filenoN)r6 r7 r8 r< r� r r r r �File� s r� r� i� ) r r' r
�curdir�O_RDONLYr r� �_GiveupOnSendfile�_sendfile_use_sendfiler ru )r3 r� 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 )N)r� )r� r� r� r� r rC r �detach)r3 r r� �styper� �s2r r r �_test_socket_fileno� s z&GeneralModuleTests._test_socket_filenoc C s6 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��r2t�� }| �tj|� t � t jt j�}| �|j� |�tj�|d�� | �|t jt j� d S )Nr r r� r )r r- r. rC r rK r r� r� r r rb r� r� r� r� �shutilZrmtreer� r' r� r� )r3 r Ztmpdirr r r �test_socket_fileno� s*
z%GeneralModuleTests.test_socket_fileno)Mr6 r7 r8 r� r� r� �
skipUnlessr� r� r� r� r� r� r� r� r� r r r r r r r r r r � cpython_onlyr# r- r3 r5 r; r@ rD rE rG rK rM rO rP rR rV r' r� rZ r\ rg rh Zis_resource_enabledri rx ry rz r} r r� r� r� r� r� r� r� rb r� r� r� r* r� r� r� r� r� �skipIfr� r� r� r� r r r r r� � s�
, , ? # N
r� z!SocketCan required for this test.c @ s� e Zd Zdd� Ze�eed�d�dd� �Zdd� Z e�eed�d�d d
� �Z
d d � Z d
d� Z e�eed�d�dd� �Z
e�eed�d�dd� �ZdS )� BasicCANTestc C s t j t j t j d S )N)r �AF_CANr r )r3 r r r r� � s z!BasicCANTest.testCrucialConstants�CAN_BCMz&socket.CAN_BCM required for this test.c C sR 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 )N)r r� ZCAN_BCM_TX_SETUPZCAN_BCM_TX_DELETEZCAN_BCM_TX_READ�CAN_BCM_TX_SENDZCAN_BCM_RX_SETUPZCAN_BCM_RX_DELETEZCAN_BCM_RX_READZCAN_BCM_TX_STATUSZCAN_BCM_TX_EXPIREDZCAN_BCM_RX_STATUSZCAN_BCM_RX_TIMEOUTZCAN_BCM_RX_CHANGED)r3 r r r �testBCMConstants� s z BasicCANTest.testBCMConstantsc C s$ t � t jt jt j��}W d Q R X d S )N)r r r r )r3 r r r r �testCreateSocket
s z BasicCANTest.testCreateSocketc C s$ t � t jt jt j��}W d Q R X d S )N)r r r r� )r3 r r r r �testCreateBCMSocket s z BasicCANTest.testCreateBCMSocketc C s. t � t jt jt j��}|�d� W d Q R X d S )N)r� )r r r r rK )r3 r r r r � testBindAny s zBasicCANTest.testBindAnyc C s6 t � t jt jt j��}| �td|jd� W d Q R X d S )Nzinterface name too long)� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx)r r r r r� r rK )r3 r r r r �testTooLongInterfaceName s z%BasicCANTest.testTooLongInterfaceName�CAN_RAW_LOOPBACKz/socket.CAN_RAW_LOOPBACK required for this test.c C s\ t � t jt jt j��>}x6dD ].}|�t jt j|� | �||�t jt j�� q W W d Q R X d S )N)r r[ ) r r r r rL � SOL_CAN_RAWr� r� rH )r3 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 d Q R X d S )N)i i z=IIrJ ) r
�packr r r r rL r� r� r� rH rC )r3 �can_idZcan_maskZ
can_filterr r r r �
testFilter( s zBasicCANTest.testFilterN)r6 r7 r8 r� r� r� r r r� r� r� r� r� r� r� r r r r r� � s r� c @ s� e Zd Zddd�Zedd� �Zedd� �Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z e
�eed�d�dd� �Ze
�eed�d�dd� �ZdS ) �CANTestrz c C s t j| |d� d S )N)r{ )r~ r< )r3 r{ r r r r< 7 s zCANTest.__init__c C s& t |�}|�dd�}t�| j|||�S )NrJ � )ra �ljustr
r� rO )�clsr� �data�can_dlcr r r �build_can_frame: s zCANTest.build_can_framec C s& t �| j|�\}}}|||d |� fS )N)r
r rO )r� �framer� r� r� r r r �dissect_can_frameA s zCANTest.dissect_can_framec C s2 | j �| j�\}}| �| j|� | �|| j� d S )N)r �recvfromrN r� �cfrL )r3 r� r� r r r �
testSendFrameG s zCANTest.testSendFramec C s | � dd�| _| j�| j� d S )Nr s )r� r� r| r� )r3 r r r �_testSendFrameO s zCANTest._testSendFramec C s$ | j �| j�\}}| �| j|� d S )N)r r� rN r� r� )r3 r� r� r r r �testSendMaxFrameS s zCANTest.testSendMaxFramec C s | � dd�| _| j�| j� d S )Nr s )r� r� r| r� )r3 r r r �_testSendMaxFrameW s zCANTest._testSendMaxFramec C sD | j �| j�\}}| �| j|� | j �| j�\}}| �| j|� d S )N)r r� rN r� �cf1�cf2)r3 r� r� 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� )r3 r r r �_testSendMultiFramesb s z CANTest._testSendMultiFramesr� z&socket.CAN_BCM required for this test.c C sP | j �| j�\}}| �| j|� | �|�\}}}| �| j|� | �| j|� d S )N)r| r� rN r� r� r� r� r� )r3 r� r� r� r� r� r r r �_testBCMi 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 )Ni# � � �� r i" r[ )r r r r� rC r r� rL r� �bytesr� r� r� r� r
r� rQ r� r� ra )r3 ZbcmZopcoder^ rU Z
ival1_secondsZ
ival1_usecZ
ival2_secondsZ
ival2_usecZ
bcm_can_idZnframes�headerZheader_plus_frameZ
bytes_sentr r r �testBCMr s2
zCANTest.testBCMN)rz )r6 r7 r8 r< � classmethodr� r� r� r� r� r� r� r� r� r� r r r� r� r r r r r� 4 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 )NrG )r; r< rL )r3 r@ rA )rB r r r< � s zISOTPTest.__init__c C s t j t j t j t j d S )N)r r� r r r )r3 r r r r� � s z ISOTPTest.testCrucialConstantsc C s$ t � t jt jt j��}W d Q R X d S )N)r r r r )r3 r r r r r� � s zISOTPTest.testCreateSocketr z(socket.CAN_ISOTP required for this test.c C s$ t � t jt jt j��}W d Q R X d S )N)r r r r )r3 r r r r �testCreateISOTPSocket� s z ISOTPTest.testCreateISOTPSocketc
C sF t � t jt jt j��(}| �td�� |�d� W d Q R X W d Q R X d S )Nzinterface name too long)r� r[ r� )r r r r r� r rK )r3 r r r r r� � s z"ISOTPTest.testTooLongInterfaceNamec
C s� yJt � t jt jt j��,}| jddf}|�|� | �|�� |� W d Q R X W nB tk
r� } z$|j t j
krz| � d| j � n� W d d }~X Y nX d S )Ni# iV z%network interface `%s` does not exist) r r r r rL rK r� r� r r� ZENODEVrM )r3 r r� rv r r r �testBind� s
zISOTPTest.testBind)r6 r7 r8 r< r� r� r� r� r r r� r� r� rE r r )rB 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 )N)r ZAF_RDSr )r3 r r r r� � s z!BasicRDSTest.testCrucialConstantsc C s" t � t jt jd��}W d Q R X d S )Nr )r r r )r3 r r r r r� � s z BasicRDSTest.testCreateSocketc C sJ d}t � t jt jd��*}|�t jt j|� |�t jt j|� W d Q R X d S )Ni @ r )r r r rL r� Z SO_RCVBUFZ SO_SNDBUF)r3 rN r r r r �testSocketBufferSize� s z!BasicRDSTest.testSocketBufferSizeN)r6 r7 r8 r� r� r� r r r r r� � s r� c s� e Zd Zd dd�Z� fdd�Zdd� Zdd � Zd
d � Zd d
� Ze e
j
d�dd� �Z e e
j
d�dd� �Z dd� Z
dd� Zdd� Zdd� Z� ZS ) �RDSTestrz c C s t j| |d� d S )N)r{ )r r< )r3 r{ r r r r< � s zRDSTest.__init__c s t � �� t�� | _d S )N)r; r4 r= r` Zevt)r3 )rB r r r4 � s
z
RDSTest.setUpc C s2 | j �| j�\}}| �| j|� | �| j|� d S )N)r/ r� rN r� r� r� )r3 r� r� r r r �testSendAndRecv� s zRDSTest.testSendAndRecvc C s" d| _ | j�| j dt| jf� d S )Ns spamr )r� r| r� r� r1 )r3 r r r �_testSendAndRecv� s zRDSTest._testSendAndRecvc C sH | j �| jtj�\}}| �| j|� | j �| j�\}}| �| j|� d S )N)r/ r� rN r �MSG_PEEKr� r� )r3 r� r� r r r �testPeek� s zRDSTest.testPeekc C s" d| _ | j�| j dt| jf� d S )Ns spamr )r� r| r� r� r1 )r3 r r r � _testPeek� s zRDSTest._testPeek�recvmsgc C s( | j �| j�\}}}}| �| j|� d S )N)r/ r� rN r� r� )r3 r� �ancdataZ msg_flagsr� r r r �testSendAndRecvMsg� s zRDSTest.testSendAndRecvMsg�sendmsgc C s& d| _ | j�| j gddt| jf� d S )Ns< hello hello hello hello hello hello hello hello hello hello r r )r� r| r� r� r1 )r3 r r r �_testSendAndRecvMsg� s zRDSTest._testSendAndRecvMsgc C sD | j �| j�\}}| �| j|� | j �| j�\}}| �| j|� d S )N)r/ r� rN r� �data1�data2)r3 r� r� r r r �testSendAndRecvMulti� s z RDSTest.testSendAndRecvMultic C s@ d| _ | j�| j dt| jf� d| _| j�| jdt| jf� d S )Ns baconr s egg)r r| r� r� r1 r )r3 r r r �_testSendAndRecvMulti� s z RDSTest._testSendAndRecvMultic C sL t � | jgg g d�\}}}| �| j|� | j�| j�\}}| �| j|� d S )Ng @)�selectr/ r� r� rN r� r� )r3 r r� rI r� r� r r r �
testSelect s zRDSTest.testSelectc C s" d| _ | j�| j dt| jf� d S )Ns selectr )r� r| r� r� r1 )r3 r r r � _testSelect
s zRDSTest._testSelect)rz )r6 r7 r8 r< r4 r� r� r� r� r� r r� r� r r r r rE r r )rB r r� � s
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�BasicVSOCKTestc C s
t j d S )N)r r� )r3 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 )N) r �SO_VM_SOCKETS_BUFFER_SIZE� SO_VM_SOCKETS_BUFFER_MIN_SIZE� SO_VM_SOCKETS_BUFFER_MAX_SIZEr� ZVMADDR_PORT_ANYZVMADDR_CID_HOSTZVM_SOCKETS_INVALID_VERSIONr )r3 r r r �testVSOCKConstants s z!BasicVSOCKTest.testVSOCKConstantsc C s t � t jt j��}W d Q R X d S )N)r r� r. )r3 r r r r r� ! s z BasicVSOCKTest.testCreateSocketc C s� t � t jt j���}|�t jt j�}|�t jt j�}|�t jt j�}|�t jt j|d � |�t jt j|d � |�t jt j|d � | �|d |�t jt j�� | �|d |�t jt j�� | �|d |�t jt j�� W d Q R X d S )Nr� ) r r� r. rH r
r r rL r� )r3 r Zorig_maxr
Zorig_minr r r r� % s,
z#BasicVSOCKTest.testSocketBufferSizeN)r6 r7 r8 r� r r� 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 ),� BasicTCPTestrz c C s t j| |d� d S )N)r{ )r� r< )r3 r{ r r r r< B s zBasicTCPTest.__init__c C s | j �d�}| �|t� d S )Ni )r� r� r� r� )r3 r� r r r �testRecvE s zBasicTCPTest.testRecvc C s | j �t� d S )N)r� r� r� )r3 r r r � _testRecvJ s zBasicTCPTest._testRecvc C s8 | j �tt�d �}| j �d�}|| }| �|t� d S )N� i )r� r� ra r� r� )r3 �seg1�seg2r� r r r �testOverFlowRecvM s z BasicTCPTest.testOverFlowRecvc C s | j �t� d S )N)r� r� r� )r3 r r r �_testOverFlowRecvT s z BasicTCPTest._testOverFlowRecvc C s | j �d�\}}| �|t� d S )Ni )r� r� r� r� )r3 r� r� r r r � testRecvFromW s zBasicTCPTest.testRecvFromc C s | j �t� d S )N)r� r� r� )r3 r r r �
_testRecvFrom\ s zBasicTCPTest._testRecvFromc C s@ | j �tt�d �\}}| j �d�\}}|| }| �|t� d S )Nr i )r� r� ra r� r� )r3 r r� r r� r r r �testOverFlowRecvFrom_ s z!BasicTCPTest.testOverFlowRecvFromc C s | j �t� d S )N)r� r� r� )r3 r r r �_testOverFlowRecvFromf s z"BasicTCPTest._testOverFlowRecvFromc C s4 d}x | j �d�}|sP ||7 }qW | �|d� d S )Nr� i s ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff)r� r� r� )r3 r� �readr r r � testSendAlli s zBasicTCPTest.testSendAllc C s d}| j �|� d S )Ns ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff)r� rs )r3 Z big_chunkr r r � _testSendAlls s zBasicTCPTest._testSendAllc C sP | j �� }t�|tjtj�}| �|j� | �|tj� |� d�}| �
|t � d S )Ni ) r� r� r Zfromfdr- r. rC r r� r� r� r� )r3 r� r) r� r r r �
testFromFdw s
zBasicTCPTest.testFromFdc C s | j �t� d S )N)r� r� r� )r3 r r r � _testFromFd� s zBasicTCPTest._testFromFdc C s0 | j �� }| �|j� |�d�}| �|t� d S )Ni )r� �duprC r r� r� r� )r3 r) r� r r r �testDup� s
zBasicTCPTest.testDupc C s | j �t� d S )N)r� r� r� )r3 r r r �_testDup� s zBasicTCPTest._testDupc C s& | j �d�}| �|t� | j�� d S )Ni )r� r� r� r� rb rl )r3 r� r r r � testShutdown� s zBasicTCPTest.testShutdownc C s | j �t� | j �d� d S )Nr� )r� r� r� �shutdown)r3 r r r �
_testShutdown� s zBasicTCPTest._testShutdownc C sX dd l }| j�t� | �t| jj|jd � | �t| jjd|jd � | j�d� d S )Nr r[ r� ) r r� r� r� r� r r! r �UINT_MAX)r3 r r r r �_testShutdown_overflow� s z#BasicTCPTest._testShutdown_overflowc C s� | j �� }| j �� }| �||� | �| j j� | �t| j jd� | j � � t
j
t
j t
j |d�}| �
|j � |�d�}| �|t� d S )Ni )r� )r� r� r� r� r� �_closedr� r r� r r r- r. rC r� )r3 r� r r) r� r r r �
testDetach� s
zBasicTCPTest.testDetachc C s | j �t� d S )N)r� r� r� )r3 r r r � _testDetach� s zBasicTCPTest._testDetachN)rz )r6 r7 r8 r< r
r r r r r r r r r r r r r r r" r r� 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 )� BasicUDPTestrz c C s t j| |d� d S )N)r{ )r} r< )r3 r{ r r r r< � s zBasicUDPTest.__init__c C s | j �tt��}| �|t� d S )N)r/ r� ra r� r� )r3 r� r r r �testSendtoAndRecv� s z BasicUDPTest.testSendtoAndRecvc C s | j �tdt| jf� d S )Nr )r| r� r� r� r1 )r3 r r r �_testSendtoAndRecv� s z BasicUDPTest._testSendtoAndRecvc C s$ | j �tt��\}}| �|t� d S )N)r/ r� ra r� r� )r3 r� r� r r r r � s zBasicUDPTest.testRecvFromc C s | j �tdt| jf� d S )Nr )r| r� r� r� r1 )r3 r r r r � s zBasicUDPTest._testRecvFromc C s | � t| jjd� d S )Nr )r� r2 r/ r� )r3 r r r �testRecvFromNegative� s z!BasicUDPTest.testRecvFromNegativec C s | j �tdt| jf� d S )Nr )r| r� r� r� r1 )r3 r r r �_testRecvFromNegative� s z"BasicUDPTest._testRecvFromNegativeN)rz )
r6 r7 r8 r< 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 x dD ]Z e e
ee d�O Z qNW dZdZdZddd�Z� ZS )�SendrecvmsgBaseg @c s t �� | _t� �� d S )N)r= r` �
misc_eventr; r4 )r3 )rB r r r4 � s
zSendrecvmsgBase.setUpc C s | j �|�S )N)�cli_sockr� )r3 r� r r r � sendToServer� s z SendrecvmsgBase.sendToServerr c G s | j j|| jt|�d � � S )N)r/ r� �sendmsg_to_server_defaultsra )r3 r@ r r r �sendmsgToServer s z SendrecvmsgBase.sendmsgToServerc G s |j |f|�� }| �|� |S )N)r� �registerRecvmsgResult)r3 r) rN r@ �resultr r r � doRecvmsg
s
zSendrecvmsgBase.doRecvmsgc C s d S )Nr )r3 r4 r r r r3 s z%SendrecvmsgBase.registerRecvmsgResultc C s | � ||� d S )N)r� )r3 �addr1�addr2r r r �checkRecvmsgAddress s z#SendrecvmsgBase.checkRecvmsgAddressr )�
MSG_CTRUNCZMSG_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� )
r3 r^ �eor�checkset�
checkunset�ignoreZ
defaultsetZ defaultunsetZinbothr r r r �
checkFlags2 s$
zSendrecvmsgBase.checkFlags)Nr r r )r6 r7 r8 � fail_timeoutr4 r0 r1 r2 r5 r3 r8 r; r� rh r r: r< r= rC rE r r )rB 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 )Nr r[ )rC � recvmsg_intor3 �assertGreaterEqual�assertLessEqualr� )r3 r) rN r@ �bufr4 r r r r5 l s
zRecvmsgIntoMixin.doRecvmsgN)r6 r7 r8 r5 r r r r rE i s rE c s e Zd Ze� fdd��Z� ZS )�SendrecvmsgDgramFlagsBasec s t � jtjB S )N)r; r= r � MSG_TRUNC)r3 )rB r r r= x s z5SendrecvmsgDgramFlagsBase.msg_flags_non_eor_indicator)r6 r7 r8 �propertyr= rE r r )rB r rJ u s rJ c s e Zd Ze� fdd��Z� ZS )�SendrecvmsgSCTPFlagsBasec s t � jtjB S )N)r; r< r ZMSG_EOR)r3 )rB r r r< � s z0SendrecvmsgSCTPFlagsBase.msg_flags_eor_indicator)r6 r7 r8 rL r< rE r r )rB r rM } s rM c @ s8 e Zd Zedd� �Zedd� �Zedd� �Zdd� Zd S )
� SendrecvmsgConnectionlessBasec C s | j S )N)r/ )r3 r r r � serv_sock� s z'SendrecvmsgConnectionlessBase.serv_sockc C s | j S )N)r| )r3 r r r r/ � s z&SendrecvmsgConnectionlessBase.cli_sockc C s g g d| j fS )Nr )r� )r3 r r r r1 � s z8SendrecvmsgConnectionlessBase.sendmsg_to_server_defaultsc C s | j �|| j�S )N)r/ r� r� )r3 r� r r r r0 � s z*SendrecvmsgConnectionlessBase.sendToServerN)r6 r7 r8 rL rO r/ r1 r0 r r r r rN � s rN c @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )�SendrecvmsgConnectedBasec C s | j S )N)r� )r3 r r r rO � s z"SendrecvmsgConnectedBase.serv_sockc C s | j S )N)r� )r3 r r r r/ � s z!SendrecvmsgConnectedBase.cli_sockc C s d S )Nr )r3 r6 r7 r r r r8 � s z,SendrecvmsgConnectedBase.checkRecvmsgAddressN)r6 r7 r8 rL rO r/ r8 r r r r rP � s rP c s e Zd Z� fdd�Z� ZS )� SendrecvmsgServerTimeoutBasec s t � �� | j�| j� d S )N)r; r4 rO rN rD )r3 )rB r r r4 � s
z"SendrecvmsgServerTimeoutBase.setUp)r6 r7 r8 r4 rE r r )rB r rQ � s rQ 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 )N)r� rO r� ra r� )r3 r r r � testSendmsg� s zSendmsgTests.testSendmsgc C s | � | �tg�tt�� d S )N)r� r2 r� ra )r3 r r r � _testSendmsg� s zSendmsgTests._testSendmsgc C s | � | j�tt��t� d S )N)r� rO r� ra r� )r3 r r r �testSendmsgDataGenerator� s z%SendmsgTests.testSendmsgDataGeneratorc C s& | � | �dd� tgD ��tt�� d S )Nc s s | ]
}|V qd S )Nr )r� r� r r r r� � s z9SendmsgTests._testSendmsgDataGenerator.<locals>.<genexpr>)r� r2 r� ra )r3 r r r �_testSendmsgDataGenerator� s z&SendmsgTests._testSendmsgDataGeneratorc C s | � | j�tt��t� d S )N)r� rO r� ra r� )r3 r r r � testSendmsgAncillaryGenerator� s z*SendmsgTests.testSendmsgAncillaryGeneratorc C s( | � | �tgdd� dD ��tt�� d S )Nc s s | ]
}|V qd S )Nr )r� r� r r r r� � s z>SendmsgTests._testSendmsgAncillaryGenerator.<locals>.<genexpr>r )r� r2 r� ra )r3 r r r � _testSendmsgAncillaryGenerator� s z+SendmsgTests._testSendmsgAncillaryGeneratorc C s | � | j�tt��t� d S )N)r� rO r� ra r� )r3 r r r �testSendmsgArray� s z SendmsgTests.testSendmsgArrayc C s$ | � | �t�dt�g�tt�� d S )N�B)r� r2 �arrayr� ra )r3 r r r �_testSendmsgArray� s z SendmsgTests._testSendmsgArrayc C s | � | j�tt��t� d S )N)r� rO r� ra r� )r3 r r r �testSendmsgGather� s z SendmsgTests.testSendmsgGatherc C s. | � | �td d� tdd � g�tt�� d S )Nr )r� r2 r� ra )r3 r r r �_testSendmsgGather� s z SendmsgTests._testSendmsgGatherc C s | � | j�d�d� d S )Ni� s done)r� rO r� )r3 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 s done)r� ru r/ r� r2 �objectr� r0 )r3 r r r �_testSendmsgBadArgs� s
z SendmsgTests._testSendmsgBadArgsc C s | � | j�d�d� d S )Ni� s done)r� rO r� )r3 r r r �testSendmsgBadCmsg� s z SendmsgTests.testSendmsgBadCmsgc C s� | � t| jtgt� g� | � t| jtgt� ddfg� | � t| jtgdt� dfg� | � t| jtgddt� fg� | � t| jtgdg� | � t| jtgdg� | �d� d S )Nr s data)r r )r r s datar� s done)r� ru r2 r� r` r0 )r3 r r r �_testSendmsgBadCmsg� s
z SendmsgTests._testSendmsgBadCmsg�
CMSG_SPACEc C s | � | j�d�d� d S )Ni� s done)r� rO r� )r3 r r r �testSendmsgBadMultiCmsg
s z$SendmsgTests.testSendmsgBadMultiCmsgc C sB | � t| jtgdddg� | � t| jtgdt� g� | �d� d S )Nr r� )r r r� s done)r� ru r2 r� r` r0 )r3 r r r �_testSendmsgBadMultiCmsg
s
z%SendmsgTests._testSendmsgBadMultiCmsgc C s | � | j�d�d� d S )Ni� s done)r� rO r� )r3 r r r �testSendmsgExcessCmsgReject
s z(SendmsgTests.testSendmsgExcessCmsgRejectc C sN t td�s@| �t��}| �tgddg� W d Q R X | �|jj� | � d� d S )Nrd )r r r� s done)
r r r� r r2 r� � assertIsNoner� r� r0 )r3 r� r r r � _testSendmsgExcessCmsgReject
s
z)SendmsgTests._testSendmsgExcessCmsgRejectc C s d S )Nr )r3 r r r �testSendmsgAfterClose$
s z"SendmsgTests.testSendmsgAfterClosec C s | j �� | �t| jtg� d S )N)r/ r r� r r2 r� )r3 r r r �_testSendmsgAfterClose(
s
z#SendmsgTests._testSendmsgAfterCloseN)r6 r7 r8 rS rT rU rV rW rX rY r\ r] r^ r_ ra rb rc r� r re r� rf rg ri rj rk r r r r rR � s( rR 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 )N)r� rO r� ra r� )r3 r r r �testSendmsgExplicitNoneAddr1
s z.SendmsgStreamTests.testSendmsgExplicitNoneAddrc C s" | � | �tgg dd �tt�� d S )Nr )r� r2 r� ra )r3 r r r � _testSendmsgExplicitNoneAddr5
s z/SendmsgStreamTests._testSendmsgExplicitNoneAddrc C s. | � | j�d�d� | �| jj| jd�� d S )Ni s aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)ru )r� rO r� r� r. rl rD )r3 r r r �testSendmsgTimeout8
s z%SendmsgStreamTests.testSendmsgTimeoutc
C s� zv| j �d� yx| �dg� qW W nD tjk
r: Y n: tk
rh } z|jtjkrX� W d d }~X Y n X | �d� W d | j �
� X d S )Ng� ��Q��?s aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaazsocket.timeout not raised) r/ rN r2 r ru r r� �ENOMEMr� r. rY )r3 ro r r r �_testSendmsgTimeout=
s z&SendmsgStreamTests._testSendmsgTimeout> r$ z<MSG_DONTWAIT not known to work on this platform when sendingc C s. | � | j�d�d� | �| jj| jd�� d S )Ni s aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)ru )r� rO r� r� r. rl rD )r3 r r r �testSendmsgDontWaitT
s z&SendmsgStreamTests.testSendmsgDontWaitc C s` zN| � t�� }x| �dgg tj� qW W d Q R X | �|jjtjtj tj
f� W d | j � � X d S )Ns aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)
r� r r2 r Z MSG_DONTWAITr� r� r� ZEAGAINZ EWOULDBLOCKrp r. rY )r3 r� r r r �_testSendmsgDontWait\
s
z'SendmsgStreamTests._testSendmsgDontWaitN)
r6 r7 r8 rm rn ro rq r� r r* rr r� rs r r r r rl -
s
rl c @ s e Zd Zdd� Zdd� ZdS )�SendmsgConnectionlessTestsc C s d S )Nr )r3 r r r �testSendmsgNoDestAddro
s z0SendmsgConnectionlessTests.testSendmsgNoDestAddrc C s2 | � t| jjtg� | � t| jjtgg dd � d S )Nr )r� r r/ r� r� )r3 r r r �_testSendmsgNoDestAddrt
s z1SendmsgConnectionlessTests._testSendmsgNoDestAddrN)r6 r7 r8 ru rv r r r r rt j
s rt 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? )r5 rO ra r� r� r8 r� rC )r3 r� r� r^ r� r r r � testRecvmsg
s
z RecvmsgGenericTests.testRecvmsgc C s | � t� d S )N)r0 r� )r3 r r r � _testRecvmsg�
s z RecvmsgGenericTests._testRecvmsgc C sV | � | jtt�dd�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr T)r? )r5 rO ra r� r� r8 r� rC )r3 r� r� r^ r� r r r �testRecvmsgExplicitDefaults�
s z/RecvmsgGenericTests.testRecvmsgExplicitDefaultsc C s | � t� d S )N)r0 r� )r3 r r r � _testRecvmsgExplicitDefaults�
s z0RecvmsgGenericTests._testRecvmsgExplicitDefaultsc C sV | � | jtt�d �\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr� T)r? )r5 rO ra r� r� r8 r� rC )r3 r� r� r^ r� r r r �testRecvmsgShorter�
s z&RecvmsgGenericTests.testRecvmsgShorterc C s | � t� d S )N)r0 r� )r3 r r r �_testRecvmsgShorter�
s z'RecvmsgGenericTests._testRecvmsgShorterc C s^ | � | jtt�d �\}}}}| �|td d� � | �|| j� | �|g � | j|dd� d S )Nr ���F)r? )r5 rO ra r� r� r8 r� rC )r3 r� r� r^ r� r r r �testRecvmsgTrunc�
s z$RecvmsgGenericTests.testRecvmsgTruncc C s | � t� d S )N)r0 r� )r3 r r r �_testRecvmsgTrunc�
s z%RecvmsgGenericTests._testRecvmsgTruncc C sT | � | jtt�d�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr[ T)r? )r5 rO ra r� r� r8 r� rC )r3 r� r� r^ r� r r r � testRecvmsgShortAncillaryBuf�
s z0RecvmsgGenericTests.testRecvmsgShortAncillaryBufc C s | � t� d S )N)r0 r� )r3 r r r � _testRecvmsgShortAncillaryBuf�
s z1RecvmsgGenericTests._testRecvmsgShortAncillaryBufc C sT | � | jtt�d�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Ni ( T)r? )r5 rO ra r� r� r8 r� rC )r3 r� r� r^ r� r r r �testRecvmsgLongAncillaryBuf�
s z/RecvmsgGenericTests.testRecvmsgLongAncillaryBufc C s | � t� d S )N)r0 r� )r3 r r r � _testRecvmsgLongAncillaryBuf�
s z0RecvmsgGenericTests._testRecvmsgLongAncillaryBufc C s" | j �� | �t| j| j d� d S )Ni )rO r r� r r5 )r3 r r r �testRecvmsgAfterClose�
s
z)RecvmsgGenericTests.testRecvmsgAfterClosec C s d S )Nr )r3 r r r �_testRecvmsgAfterClose�
s z*RecvmsgGenericTests._testRecvmsgAfterClosec C s< z*| j �d� | �tj| j| j tt�� W d | j� � X d S )Ng� ��Q��?)
rO rN r� r ru r5 ra r� r. rY )r3 r r r �testRecvmsgTimeout�
s
z&RecvmsgGenericTests.testRecvmsgTimeoutc C s | � | jj| jd�� d S )N)ru )r� r. rl rD )r3 r r r �_testRecvmsgTimeout�
s z'RecvmsgGenericTests._testRecvmsgTimeoutr� c C s | � | jtt�d dtj�\}}}}| �|td d� � | �|| j� | �|g � | j |dt
tdd�d� | � | jtt�dtj�\}}}}| �|t� | �|| j� | �|g � | j |dd� | � | jtt��\}}}}| �|t� | �|| j� | �|g � | j |dd� d S ) Nr r r~ FrK )r? rB T)r? ) r5 rO ra r� r r� r� r8 r� rC rh )r3 r� r� r^ r� r r r �testRecvmsgPeek�
s( z#RecvmsgGenericTests.testRecvmsgPeekc C s | � t� d S )N)r0 r� )r3 r r r �_testRecvmsgPeek�
s z$RecvmsgGenericTests._testRecvmsgPeekr� c C s` | j �| j� | �| j tt��\}}}}| �|t� | �|| j� | �|g � | j |dd� d S )NT)r? )
rO rN rD r5 ra r� r� r8 r� rC )r3 r� r� r^ r� r r r �testRecvmsgFromSendmsg s z*RecvmsgGenericTests.testRecvmsgFromSendmsgc C s. | � | �td d� tdd � g�tt�� d S )Nr )r� r2 r� ra )r3 r r r �_testRecvmsgFromSendmsg s z+RecvmsgGenericTests._testRecvmsgFromSendmsgN)r6 r7 r8 rx ry rz r{ r| r} r r� r� r� r� r� r� r� r� r� r� r r� r� r� r� r� r r r r rw {
s( # rw c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�RecvmsgGenericStreamTestsc C sN | � | jd�\}}}}| �|d� | �|| j� | �|g � | j|d d� d S )Ni r� )r? )r5 rO r� r8 r� rC )r3 r� r� r^ r� r r r �testRecvmsgEOF s
z(RecvmsgGenericStreamTests.testRecvmsgEOFc C s | j �� d S )N)r/ r )r3 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 F)r? i T)r5 rO ra r� r8 r� r� rC )r3 r r� r^ r� r r� r r r �testRecvmsgOverflow s z-RecvmsgGenericStreamTests.testRecvmsgOverflowc C s | � t� d S )N)r0 r� )r3 r r r �_testRecvmsgOverflow0 s z.RecvmsgGenericStreamTests._testRecvmsgOverflowN)r6 r7 r8 r� r� r� r� r r r r r� s r� c @ s e Zd Zdd� Zdd� ZdS )� RecvmsgTestsc C s | � t| jj� | � t| jjddd� | � t| jjtt�dd� | � t| jjtd�gdd� | � t| jjt� dd� | � t| jjtt�t� d� | � t| jjtt�dt� � | j�tt�dd�\}}}}| � |t� | �
|| j � | � |g � | j |dd� d S )Nr r r. T)r? )
r� ru rO r� r2 ra r� rC r` r� r8 r� rC )r3 r� r� r^ r� r r r �testRecvmsgBadArgs7 s$
z RecvmsgTests.testRecvmsgBadArgsc C s | � t� d S )N)r0 r� )r3 r r r �_testRecvmsgBadArgsM s z RecvmsgTests._testRecvmsgBadArgsN)r6 r7 r8 r� r� r r r r r� 4 s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�RecvmsgIntoTestsc C sL t tt��}| �t| jj� | �t| jjtt�dd� | �t| jj|dd� | �t| jjt� gdd� | �t| jjdgdd� | �t| jj|t� gdd� | �t| jj|gdd� | �t| jj|gt� d� | �t| jj|gdt� � | j�|gdd�\}}}}| � |tt�� | � |t t�� | �
|| j � | � |g � | j |dd� d S )Nr s I'm not writabler T)r? )
rC ra r� r� ru rO rF r` r2 r� r8 r� rC )r3 rI �nbytesr� r^ r� r r r �testRecvmsgIntoBadArgsT s0
z'RecvmsgIntoTests.testRecvmsgIntoBadArgsc C s | � t� d S )N)r0 r� )r3 r r r �_testRecvmsgIntoBadArgsp s z(RecvmsgIntoTests._testRecvmsgIntoBadArgsc C sx t tt��}| j�dd� |gD ��\}}}}| �|tt�� | �|t t�� | �|| j� | �|g � | j|dd� d S )Nc s s | ]
}|V qd S )Nr )r� r� r r r r� w s z<RecvmsgIntoTests.testRecvmsgIntoGenerator.<locals>.<genexpr>T)r? ) rC ra r� rO rF r� r8 r� rC )r3 rI r� r� r^ r� r r r �testRecvmsgIntoGenerators s z)RecvmsgIntoTests.testRecvmsgIntoGeneratorc C s | � t� d S )N)r0 r� )r3 r r r �_testRecvmsgIntoGenerator~ s z*RecvmsgIntoTests._testRecvmsgIntoGeneratorc C sx t � ddgtt� �}| j�|g�\}}}}| �|tt�� | �|�� t� | �|| j� | �|g � | j |dd� d S )NrZ r T)r? )
r[ ra r� rO rF r� �tobytesr8 r� rC )r3 rI r� r� r^ r� r r r �testRecvmsgIntoArray� s z%RecvmsgIntoTests.testRecvmsgIntoArrayc C s | � t� d S )N)r0 r� )r3 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� � s Mary had a little lambs Marys
01 had a 9s little lamb---T)r? ) rC rO rF �
memoryviewr� ra r8 r� rC )r3 Zb1Zb2Zb3r� r� r^ r� r r r �testRecvmsgIntoScatter� s z'RecvmsgIntoTests.testRecvmsgIntoScatterc C s | � d� d S )Ns Mary had a little lamb)r0 )r3 r r r �_testRecvmsgIntoScatter� s z(RecvmsgIntoTests._testRecvmsgIntoScatterN) r6 r7 r8 r� r� r� r� r� r� r� r� r r r r r� Q s
r� c @ sb e Zd Zy 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 � x:|D ]2}t�|�}| �|t�d� |� | � || j � qVW | �
t tjd� | �
t tj|� | �
t tjt j
� d S )Nr r[ i r+ r� r )�socklen_t_limitr r� �listr� r r[ �itemsizer� rH r� r r �maxsize)r3 �toobigr� �nr# r r r � testCMSG_LEN� s
zCmsgMacroTests.testCMSG_LENrd c C s� | j t�d� d }ttd��tt|d |�� }t�d�}| �|t�d�jd � x\|D ]T}t�|�}| �||� | �|t� |�� | �||t� d� � | �
|| j � |}qZW | � t tjd� | � t tj|� | � t tjt
j� d S )Nr[ i r r+ r� r )r� r rd r� r� r r[ r� rG r� rH r� r r r� )r3 r� r� Zlastr� r# r r r �testCMSG_SPACE� s
z CmsgMacroTests.testCMSG_SPACE) r6 r7 r8 r r> r� �minr r� r r� r� r r r r r� � s
r� c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zdedd�Z dd� Z
dd� Z dd� Z dd� Z
eed�dd� �Zejdd � �Zd d � Zd d � Ze�ejd!kd"�e�ej�d#�d$�eed�d%d&� ���Zeje�ejd!kd"�e�ej�d#�d$�d'd(� ���Ze�ejd!kd"�e�ej�d#�d$�eed�d)d*� ���Z e je�ejd!kd"�e�ej�d#�d$�d+d,� ���Z d-d.� Z e�ejd!kd/�d0d1� �Z d2d3� Z d4d5� Z!d6d7� Z"eed�d8d9� �Z#e#jd:d;� �Z$dfd<d=�Z%d>d?� Z&d@dA� Z'dBdC� Z(dDdE� Z)dFdG� Z*dHdI� Z+dJdK� Z,dLdM� Z-dNdO� Z.dPdQ� Z/dgdRdS�Z0dTdU� Z1dVdW� Z2dXdY� Z3dZd[� Z4d\d]� Z5d^d_� Z6d`da� Z7dbdc� Z8ddS )h�
SCMRightsTesti����c C s` g }xVt |�D ]J}t�� \}}| �tj|� | �tj|� t�|t|�� � � |�
|� qW |S )N) r� r� r� rC r' r� r �writer� �encode�append)r3 r� �fdsr+ r� r� r r r �newFDs� s zSCMRightsTest.newFDsc C sF x@t |�D ]4\}}t�|dtj� | �t�|d�t|��� � q
W d S )Nr i )� enumerater' �lseek�SEEK_SETr� r r� r� )r3 r� r� r� r r r �checkFDs� s zSCMRightsTest.checkFDsc C s | � | j|� d S )N)rC �closeRecvmsgFDs)r3 r4 r r r r3 � s z#SCMRightsTest.registerRecvmsgResultc C sv xp|d D ]d\}}}|t jkr
|t jkr
t�d�}|�|d t|�t|�|j � � x|D ]}t�|� q\W q
W d S )Nr[ r+ ) r r� �
SCM_RIGHTSr[ � frombytesra r� r' r )r3 Zrecvmsg_result�
cmsg_level� cmsg_type� cmsg_datar� r� r r r r� � s
z SCMRightsTest.closeRecvmsgFDsc
C s8 | � | �tgtjtjt�d| �|��fg�tt�� d S )Nr+ ) r� r2 r� r r� r� r[ r� ra )r3 r� r r r �createAndSendFDs s z SCMRightsTest.createAndSendFDsr[ r c C s� |\}}}}| � |t� | �|| j� | j|dtj|d� | �|t� | � t
|�|� t � d�} xh|D ]`}
| �|
t � |
\} } }
| � | tj
� | � | tj� | �|
t� | � t
|
�t d� | �|
� qfW | � t
| �|� | �| � d S )NT)r? rA rB r+ r )r� r� r8 r� rC r r9 r� r� rH ra r[ �tupler� r� r� �
SIZEOF_INTr� r� )r3 Znumfdsr4 �maxcmsgs� ignoreflagsr� r� r^ r� r� �itemr� r� r� r r r �checkRecvmsgFDs s$
z SCMRightsTest.checkRecvmsgFDsc C s | � d| �| jtt�d�� d S )Nr[ i ( )r� r5 rO ra r� )r3 r r r �testFDPassSimple+ s z SCMRightsTest.testFDPassSimplec
C s< | � | �tgtjtjt�d| �d���� fg�t t�� d S )Nr+ r[ )
r� r2 r� r r� r� r[ r� r� ra )r3 r r r �_testFDPassSimple0 s z SCMRightsTest._testFDPassSimplec C s | � d| �| jtt�d�� d S )Nr� i ( )r� r5 rO ra r� )r3 r r r �testMultipleFDPass9 s z SCMRightsTest.testMultipleFDPassc C s | � d� d S )Nr� )r� )r3 r r r �_testMultipleFDPass> s z!SCMRightsTest._testMultipleFDPassrd c C s* | � d| �| jtt�t�dt ��� d S )Nr� )r� r5 rO ra r� r rd r� )r3 r r r �testFDPassCMSG_SPACEA s z"SCMRightsTest.testFDPassCMSG_SPACEc C s | � d� d S )Nr� )r� )r3 r r r �_testFDPassCMSG_SPACEH s z#SCMRightsTest._testFDPassCMSG_SPACEc
C s0 | j d| �| jtt�t�dt ��tjd� d S )Nr[ r� )r� ) r� r5 rO ra r� r r� r� r9 )r3 r r r �testFDPassCMSG_LENL s z SCMRightsTest.testFDPassCMSG_LENc C s | � d� d S )Nr[ )r� )r3 r r r �_testFDPassCMSG_LENV s z!SCMRightsTest._testFDPassCMSG_LENr% zskipping, see issue #12958r8 zskipping, see issue #22397c C s$ | j d| �| jtt�d�dd� d S )Nr� i ( )r� )r� r5 rO ra r� )r3 r r r �testFDPassSeparateY s z SCMRightsTest.testFDPassSeparatec C sX | � d�\}}| �| �tgtjtjt�d|g�ftjtjt�d|g�fg�tt�� d S )Nr� r+ ) r� r� r2 r� r r� r� r[ ra )r3 �fd0�fd1r r r �_testFDPassSeparatec s z!SCMRightsTest._testFDPassSeparatec C s@ d}| j || �| jtt�t�t�t�t| � �dtj d� d S )Nr� )r� r� )
r� r5 rO ra r� r rd r� r� r9 )r3 Znum_fdsr r r �testFDPassSeparateMinSpaceq s z(SCMRightsTest.testFDPassSeparateMinSpacec C sX | � d�\}}| �| �tgtjtjt�d|g�ftjtjt�d|g�fg�tt�� d S )Nr� r+ ) r� r� r2 r� r r� r� r[ ra )r3 r� r� r r r �_testFDPassSeparateMinSpace~ s z)SCMRightsTest._testFDPassSeparateMinSpacec
C sd y| � |g|�}W n< tk
rN } z | �|jt� | � |g�}W d d }~X Y nX | �|t|�� d S )N)r2 r r� r� r r� ra )r3 r� r� r� rv r r r �sendAncillaryIfPossible� s z%SCMRightsTest.sendAncillaryIfPossiblezsee issue #24725c C s& | j d| �| jtt�d�tjd� d S )Nr i ( )r� )r� r5 rO ra r� r r9 )r3 r r r �testFDPassEmpty� s
z SCMRightsTest.testFDPassEmptyc C s | � ttjtjdfg� d S )Nr� )r� r� r r� r� )r3 r r r �_testFDPassEmpty� s
z SCMRightsTest._testFDPassEmptyc C s� | � | jtt�d�\}}}}| �|t� | �|| j� | j|dtj d� | �
t|�d� x>|D ]6\}}}| �|tj � | �|tj � | �
t|�t� q^W d S )Ni ( T)r? rB r[ )r5 rO ra r� r� r8 r� rC r r9 rH r� r� �
assertLessr� )r3 r� r� r^ r� r� r� r� r r r �testFDPassPartialInt� s z"SCMRightsTest.testFDPassPartialIntc C s4 | � ttjtjt�d| jg��� d d� fg� d S )Nr+ r )r� r� r r� r� r[ �badfdr� )r3 r r r �_testFDPassPartialInt� s
z#SCMRightsTest._testFDPassPartialIntc C s� | � | jtt�d�\}}}}| �|t� | �|| j� | j|dtj d� | �
t|�d� t � d�}xR|D ]J\}}}| �|tj � | �|tj
� |�|d t|�t|�|j � � qhW | �
t|�d� | �|� d S )Ni ( T)r? rB r� r+ )r5 rO ra r� r� r8 r� rC r r9 rH r[ r� r� r� r� r� ) r3 r� r� r^ r� r� r� r� r� r r r � testFDPassPartialIntInMiddle� s
z*SCMRightsTest.testFDPassPartialIntInMiddlec
C sZ | � d�\}}| �ttjtjt�d|| jg��� d d� ftjtjt�d|g�fg� d S )Nr� r+ r ) r� r� r� r r� r� r[ r� r� )r3 r� r� r r r � _testFDPassPartialIntInMiddle� s z+SCMRightsTest._testFDPassPartialIntInMiddlec C sJ |\}}}}| � |t� | �|| j� | � |g � | j|dtj|d� d S )NT)r? r@ rB )r� r� r8 r� rC r r9 )r3 r4 r� r� r� r^ r� r r r �checkTruncatedHeader� s z"SCMRightsTest.checkTruncatedHeaderc C s" | j | �| jtt��tjd� d S )N)r� )r� r5 rO ra r� r r9 )r3 r r r �testCmsgTruncNoBufSize� s z$SCMRightsTest.testCmsgTruncNoBufSizec C s | � d� d S )Nr[ )r� )r3 r r r �_testCmsgTruncNoBufSize� s z%SCMRightsTest._testCmsgTruncNoBufSizec C s$ | j | �| jtt�d�tjd� d S )Nr )r� )r� r5 rO ra r� r r9 )r3 r r r �testCmsgTrunc0� s z SCMRightsTest.testCmsgTrunc0c C s | � d� d S )Nr[ )r� )r3 r r r �_testCmsgTrunc0� s z SCMRightsTest._testCmsgTrunc0c C s | � | �| jtt�d�� d S )Nr[ )r� r5 rO ra r� )r3 r r r �testCmsgTrunc1� s z SCMRightsTest.testCmsgTrunc1c C s | � d� d S )Nr[ )r� )r3 r r r �_testCmsgTrunc1� s z SCMRightsTest._testCmsgTrunc1c C s" | � | �| jtt�td �� d S )Nr� )r� r5 rO ra r� r� )r3 r r r �testCmsgTrunc2Int� s z SCMRightsTest.testCmsgTrunc2Intc C s | � d� d S )Nr[ )r� )r3 r r r �_testCmsgTrunc2Int
s z SCMRightsTest._testCmsgTrunc2Intc C s( | � | �| jtt�t�d�d �� d S )Nr r[ )r� r5 rO ra r� r r� )r3 r r r �testCmsgTruncLen0Minus1
s z%SCMRightsTest.testCmsgTruncLen0Minus1c C s | � d� d S )Nr[ )r� )r3 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 r[ r+ )r5 rO ra r� r� r8 r� rC r r9 r� r� rG rH r[ r� r� r� ) r3 �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� )r3 r r r �testCmsgTruncLen0*
s z SCMRightsTest.testCmsgTruncLen0c C s | � d� d S )Nr[ )r� )r3 r r r �_testCmsgTruncLen0-
s z SCMRightsTest._testCmsgTruncLen0c C s | j t�d�d dd� d S )Nr r[ )r� r� )r� r r� )r3 r r r �testCmsgTruncLen0Plus10
s z$SCMRightsTest.testCmsgTruncLen0Plus1c C s | � d� d S )Nr� )r� )r3 r r r �_testCmsgTruncLen0Plus13
s z%SCMRightsTest._testCmsgTruncLen0Plus1c C s | j t�t�td� d S )N)r� r� )r� r r� r� )r3 r r r �testCmsgTruncLen16
s z SCMRightsTest.testCmsgTruncLen1c C s | � d� d S )Nr� )r� )r3 r r r �_testCmsgTruncLen1:
s z SCMRightsTest._testCmsgTruncLen1c C s( | j t�dt �d dt d d� d S )Nr� r[ )r� r� )r� r r� r� )r3 r r r �testCmsgTruncLen2Minus1=
s z%SCMRightsTest.testCmsgTruncLen2Minus1c C s | � d� d S )Nr� )r� )r3 r r r �_testCmsgTruncLen2Minus1A
s z&SCMRightsTest._testCmsgTruncLen2Minus1N)r[ r )r )r )9r6 r7 r8 r� r� r� r3 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 r r r r� � sl
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� r c C s8 i }x.|D ]&\}}}| � ||f|� ||||f< q
W |S )N)r� )r3 r� �dr� r� r� r r r �ancillaryMappingV
s
z%RFC3542AncillaryTest.ancillaryMappingr� r c C s | j �tjtjd� | j�� | �| j tt �|�\}}}}| �
|t � | � || j � | j
|dtj|d� | �
t|�d� | �|d t� |d \}} }
| �
|tj� | �
| tj� | �|
t� | �
t|
�t� t�d�} | �|
� | �| d d� | �| d |� d S )Nr[ T)r? rA rB r r+ )rO rL r � IPPROTO_IPV6�IPV6_RECVHOPLIMITr. rY r5 ra r� r� r8 r� rC r9 r� r� �
IPV6_HOPLIMITr� r� r[ r� rG rH ) r3 �
ancbufsize�maxhopr� r� r� r^ r� r� r� r� r7 r r r �
checkHopLimit`
s(
z"RFC3542AncillaryTest.checkHopLimitr� r� c C s | j dd� d S )Ni ( )r� )r� )r3 r r r �testRecvHopLimit|
s z%RFC3542AncillaryTest.testRecvHopLimitc C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r �_testRecvHopLimit�
s z&RFC3542AncillaryTest._testRecvHopLimitrd c C s | j t�t�d� d S )N)r� )r� r rd r� )r3 r r r �testRecvHopLimitCMSG_SPACE�
s z/RFC3542AncillaryTest.testRecvHopLimitCMSG_SPACEc C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r �_testRecvHopLimitCMSG_SPACE�
s z0RFC3542AncillaryTest._testRecvHopLimitCMSG_SPACEr� c C s | j d| jd� d S )Ni ( )r� r� )r� � hop_limit)r3 r r r �testSetHopLimit�
s z$RFC3542AncillaryTest.testSetHopLimitc C sL | � | jj| jd�� | �| �tgtjtj t
�
d| j g�fg�t t�� d S )N)ru r+ )
r� r. rl rD r� r2 r� r r� r� r[ r� ra )r3 r r r �_testSetHopLimit�
s z%RFC3542AncillaryTest._testSetHopLimitc C sB | j �tjtjd� | j �tjtjd� | j�� | �| j t t
�|�\}}}}| � |t
� | � || j
� | j|dtj|d� | � t |�d� | �|�}|tjtjf } | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d d� |tjtjf } | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d |� d S )Nr[ T)r? rA rB r� r+ r r� )rO rL r r� r� �IPV6_RECVTCLASSr. rY r5 ra r� r� r8 r� rC r9 r� � IPV6_TCLASSr� r[ r� rG rH r� ) r3 r� r� r� r� r� r^ r� ZancmapZtcdatar7 Zhldatar r r � checkTrafficClassAndHopLimit�
s2
z1RFC3542AncillaryTest.checkTrafficClassAndHopLimitr r c C s | j dd� d S )Ni ( )r� )r )r3 r r r � testRecvTrafficClassAndHopLimit�
s z4RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitc C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r � _testRecvTrafficClassAndHopLimit�
s z5RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitc C s | j t�t�d d� d S )Nr� )r� )r r rd r� )r3 r r r �)testRecvTrafficClassAndHopLimitCMSG_SPACE�
s z>RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitCMSG_SPACEc C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r �*_testRecvTrafficClassAndHopLimitCMSG_SPACE�
s z?RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitCMSG_SPACEc C s | j d| jd� d S )Ni ( )r� r� )r r� )r3 r r r � testSetTrafficClassAndHopLimit�
s z3RFC3542AncillaryTest.testSetTrafficClassAndHopLimitc C sd | � | jj| jd�� | �| �tgtjtj t
�
d| j g�ftjtj t
�
d| j
g�fg�tt�� d S )N)ru r+ )r� r. rl rD r� r2 r� r r� r r[ �
traffic_classr� r� ra )r3 r r r � _testSetTrafficClassAndHopLimit�
s z4RFC3542AncillaryTest._testSetTrafficClassAndHopLimitc C s | j d| jd� d S )Ni ( )r� r� )r r� )r3 r r r �testOddCmsgSize�
s z$RFC3542AncillaryTest.testOddCmsgSizec
C s� | � | jj| jd�� yJ| �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 d d }~X Y nX d S )N)ru r+ r� )r� r. rl rD r2 r� r r� r r[ r r� r� r� r r� r� r r� ra )r3 r� rv r r r �_testOddCmsgSize s" z%RFC3542AncillaryTest._testOddCmsgSizec C s� | j �tjtjd� | j�� |d kr*dn|f}| j| j tt �f|�� \}}}}| �
|t � | � || j � | �
|g � | j
|dtj|d� d S )Nr[ r T)r? r@ rB )rO rL r r� r� r. rY r5 ra r� r� r8 r� rC r9 )r3 r� r� r@ r� r� r^ r� r r r � checkHopLimitTruncatedHeader s
z1RFC3542AncillaryTest.checkHopLimitTruncatedHeaderc C s | j d tjd� d S )N)r� r� )r
r r9 )r3 r r r r� , s z+RFC3542AncillaryTest.testCmsgTruncNoBufSizec C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r r� 7 s z,RFC3542AncillaryTest._testCmsgTruncNoBufSizec C s | j dtjd� d S )Nr )r� r� )r
r r9 )r3 r r r �testSingleCmsgTrunc0< s z)RFC3542AncillaryTest.testSingleCmsgTrunc0c C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r �_testSingleCmsgTrunc0C s z*RFC3542AncillaryTest._testSingleCmsgTrunc0c C s | j dd� d S )Nr[ )r� )r
)r3 r r r �testSingleCmsgTrunc1K s z)RFC3542AncillaryTest.testSingleCmsgTrunc1c C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r �_testSingleCmsgTrunc1O s z*RFC3542AncillaryTest._testSingleCmsgTrunc1c C s | j dt d� d S )Nr� )r� )r
r� )r3 r r r �testSingleCmsgTrunc2IntT s z,RFC3542AncillaryTest.testSingleCmsgTrunc2Intc C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r �_testSingleCmsgTrunc2IntX s z-RFC3542AncillaryTest._testSingleCmsgTrunc2Intc C s | j t�d�d d� d S )Nr r[ )r� )r
r r� )r3 r r r � testSingleCmsgTruncLen0Minus1] s z2RFC3542AncillaryTest.testSingleCmsgTruncLen0Minus1c C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r � _testSingleCmsgTruncLen0Minus1a s z3RFC3542AncillaryTest._testSingleCmsgTruncLen0Minus1c C s� | j �tjtjd� | j�� | �| j tt �t�
t �d �\}}}}| � |t � | �
|| j� | j|dtjd� | �t|�d� |r�|d \}}}| � |tj� | � |tj� | �t|�t � d S )Nr[ T)r? r@ r )rO rL r r� r� r. rY r5 ra r� r� r� r� r8 r� rC r9 rH r� r� )r3 r� r� r^ r� r� r� r� r r r �testSingleCmsgTruncInDataf s
" z.RFC3542AncillaryTest.testSingleCmsgTruncInDatac C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r �_testSingleCmsgTruncInData| s z/RFC3542AncillaryTest._testSingleCmsgTruncInDatac C s | j �tjtjd� | j �tjtjd� | j�� | �| j t t
�|�\}}}}| � |t
� | � || j
� | j|dtj|d� | � t |�d� |d \}}} | � |tj� | �|tjtjh� | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d d� d S )Nr[ T)r? r@ rB r r+ r� )rO rL r r� r� r r. rY r5 ra r� r� r8 r� rC r9 r� r r� r� r[ r� rG rH ) r3 r� r� r� r� r^ r� r� r� r� r7 r r r �checkTruncatedSecondHeader� s(
z/RFC3542AncillaryTest.checkTruncatedSecondHeaderc C s | j t�t�tjd� d S )N)r� )r r rd r� r9 )r3 r r r �testSecondCmsgTrunc0� s z)RFC3542AncillaryTest.testSecondCmsgTrunc0c C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r �_testSecondCmsgTrunc0� s z*RFC3542AncillaryTest._testSecondCmsgTrunc0c C s | � t�t�d � d S )Nr[ )r r rd r� )r3 r r r �testSecondCmsgTrunc1� s z)RFC3542AncillaryTest.testSecondCmsgTrunc1c C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r �_testSecondCmsgTrunc1� s z*RFC3542AncillaryTest._testSecondCmsgTrunc1c C s | � t�t�dt � d S )Nr� )r r rd r� )r3 r r r �testSecondCmsgTrunc2Int� s z,RFC3542AncillaryTest.testSecondCmsgTrunc2Intc C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r �_testSecondCmsgTrunc2Int� s z-RFC3542AncillaryTest._testSecondCmsgTrunc2Intc C s" | � t�t�t�d� d � d S )Nr r[ )r r rd r� r� )r3 r r r � testSecondCmsgTruncLen0Minus1� s z2RFC3542AncillaryTest.testSecondCmsgTruncLen0Minus1c C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r � _testSecondCmsgTruncLen0Minus1� s z3RFC3542AncillaryTest._testSecondCmsgTruncLen0Minus1c
C sT | j �tjtjd� | j �tjtjd� | j�� | �| j t t
�t� t �t�
t � d �\}}}}| �|t
� | �|| j� | j|dtjd� tjtjh}|�d�\}}}| �|tj� |�|� | �t |�t � t�d�} | �|� | �| d d� | �| d d� |�rD|�d�\}}}| �|tj� |�|� | �t |�t � | �|g � d S )Nr[ T)r? r@ r r+ r� ) rO rL r r� r� r r. rY r5 ra r� rd r� r� r� r8 r� rC r9 r r� �pop�remover[ r� rG rH r� )
r3 r� r� r^ r� Z
cmsg_typesr� r� r� r7 r r r �testSecomdCmsgTruncInData� s4
"
z.RFC3542AncillaryTest.testSecomdCmsgTruncInDatac C s$ | � | jj| jd�� | �t� d S )N)ru )r� r. rl rD r0 r� )r3 r r r �_testSecomdCmsgTruncInData� s z/RFC3542AncillaryTest._testSecomdCmsgTruncInDataN)r� r )r� r )r )r )1r6 r7 r8 r� r r� r� r� r r� r� r� r� r� r� r r r r r r r r
r r r
r� r� r r r r r r r r r r r r r r r r r r r r# r$ r r r r r� E
sp
#
$r� c @ s e Zd ZdS )�SendrecvmsgUDPTestBaseN)r6 r7 r8 r r r r r% � s r% r� c @ s e Zd ZdS )�SendmsgUDPTestN)r6 r7 r8 r r r r r& � s r& r� c @ s e Zd ZdS )�RecvmsgUDPTestN)r6 r7 r8 r r r r r' s r' rF c @ s e Zd ZdS )�RecvmsgIntoUDPTestN)r6 r7 r8 r r r r r( s r( c @ s e Zd Zdd� ZdS )�SendrecvmsgUDP6TestBasec C s | � |d d� |d d� � d S )Nr )r� )r3 r6 r7 r r r r8 s z+SendrecvmsgUDP6TestBase.checkRecvmsgAddressN)r6 r7 r8 r8 r r r r r) s r) z IPv6 required for this test.r� r c @ s e Zd ZdS )�SendmsgUDP6TestN)r6 r7 r8 r r r r r* s r* c @ s e Zd ZdS )�RecvmsgUDP6TestN)r6 r7 r8 r r r r r+ s r+ c @ s e Zd ZdS )�RecvmsgIntoUDP6TestN)r6 r7 r8 r r r r r, s r, r� c @ s e Zd ZdS )� RecvmsgRFC3542AncillaryUDP6TestN)r6 r7 r8 r r r r r- & s r- c @ s e Zd ZdS )�#RecvmsgIntoRFC3542AncillaryUDP6TestN)r6 r7 r8 r r r r r. . s r. c @ s e Zd ZdS )�SendrecvmsgTCPTestBaseN)r6 r7 r8 r r r r r/ 8 s r/ c @ s e Zd ZdS )�SendmsgTCPTestN)r6 r7 r8 r r r r r0 < s r0 c @ s e Zd ZdS )�RecvmsgTCPTestN)r6 r7 r8 r r r r r1 @ s r1 c @ s e Zd ZdS )�RecvmsgIntoTCPTestN)r6 r7 r8 r r r r r2 E s r2 c @ s e Zd ZdS )� SendrecvmsgSCTPStreamTestBaseN)r6 r7 r8 r r r r r3 K s r3 r- r. r� c @ s e Zd ZdS )�SendmsgSCTPStreamTestN)r6 r7 r8 r r r r r4 P s r4 c s e Zd Z� fdd�Z� ZS )�RecvmsgSCTPStreamTestc
sR yt t| ��� W n: tk
rL } z |jtjkr2� | �d� W d d }~X Y nX d S )Nz4sporadic ENOTCONN (kernel issue?) - see issue #13876)r; r5 r� r r� �ENOTCONNrM )r3 rv )rB r r r� Z s z$RecvmsgSCTPStreamTest.testRecvmsgEOF)r6 r7 r8 r� rE r r )rB r r5 U s r5 c s e Zd Z� fdd�Z� ZS )�RecvmsgIntoSCTPStreamTestc
sR yt t| ��� W n: tk
rL } z |jtjkr2� | �d� W d d }~X Y nX d S )Nz4sporadic ENOTCONN (kernel issue?) - see issue #13876)r; r7 r� r r� r6 rM )r3 rv )rB r r r� g s z(RecvmsgIntoSCTPStreamTest.testRecvmsgEOF)r6 r7 r8 r� rE r r )rB r r7 b s r7 c @ s e Zd ZdS )� SendrecvmsgUnixStreamTestBaseN)r6 r7 r8 r r r r r8 p s r8 r� c @ s e Zd ZdS )�SendmsgUnixStreamTestN)r6 r7 r8 r r r r r9 t s r9 c @ s e Zd ZdS )�RecvmsgUnixStreamTestN)r6 r7 r8 r r r r r: y s r: c @ s e Zd ZdS )�RecvmsgIntoUnixStreamTestN)r6 r7 r8 r r r r r; s r; r� r� c @ s e Zd ZdS )�RecvmsgSCMRightsStreamTestN)r6 r7 r8 r r r r r< � s r< c @ s e Zd ZdS )� RecvmsgIntoSCMRightsStreamTestN)r6 r7 r8 r r r r r= � s r= 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 )Nr[ r r )Zsignumr� r r r r� � r� z.InterruptedTimeoutBase.setUp.<locals>.<lambda>)r; r4 rp rq rC )r3 Zorig_alrm_handler)rB r r r4 � s
z InterruptedTimeoutBase.setUpg @� setitimerg�������?c C s t �t j|� d S )N)rp r? � ITIMER_REAL)r3 �secondsr r r �setAlarm� s z InterruptedTimeoutBase.setAlarmr� c C s t �|� d S )N)rp rj )r3 rA r r r rB � s )
r6 r7 r8 r4 ru r rp �
alarm_timerB rE r r )rB r r> � s
r> � siginterruptrj r? z+Don't have signal.alarm or signal.setitimerc sl e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Ze e
j
d
�dd� �Z e e
j
d�dd� �Z � Z
S )�InterruptedRecvTimeoutTestc s t � �� | j�| j� d S )N)r; r4 r/ rN ru )r3 )rB r r r4 � s
z InterruptedRecvTimeoutTest.setUpc O sB z0| � | j� | �t��}|||� W d Q R X W d | � d� X d S )Nr )rB rC r� rr )r3 r r@ rA r� r r r �checkInterruptedRecv� s
z/InterruptedRecvTimeoutTest.checkInterruptedRecvc C s | � | jjd� d S )Ni )rF r/ r� )r3 r r r �testInterruptedRecvTimeout� s z5InterruptedRecvTimeoutTest.testInterruptedRecvTimeoutc C s | � | jjtd�� d S )Ni )rF r/ � recv_intorC )r3 r r r � testInterruptedRecvIntoTimeout� s z9InterruptedRecvTimeoutTest.testInterruptedRecvIntoTimeoutc C s | � | jjd� d S )Ni )rF r/ r� )r3 r r r � testInterruptedRecvfromTimeout� s z9InterruptedRecvTimeoutTest.testInterruptedRecvfromTimeoutc C s | � | jjtd�� d S )Ni )rF r/ �
recvfrom_intorC )r3 r r r �"testInterruptedRecvfromIntoTimeout� s z=InterruptedRecvTimeoutTest.testInterruptedRecvfromIntoTimeoutr� c C s | � | jjd� d S )Ni )rF r/ r� )r3 r r r � testInterruptedRecvmsgTimeout� s z8InterruptedRecvTimeoutTest.testInterruptedRecvmsgTimeoutrF c C s | � | jjtd�g� d S )Ni )rF r/ rF rC )r3 r r r �!testInterruptedRecvmsgIntoTimeout� s z<InterruptedRecvTimeoutTest.testInterruptedRecvmsgIntoTimeout)r6 r7 r8 r4 rF rG rI rJ rL r� r rM rN rE r r )rB r rE � s
rE 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)r; r4 r� r� rC r r= ZThread� doConnect�startr/ r� r� r� rN ru )r3 Z
cli_threadr� )rB r r r4 � s
z InterruptedSendTimeoutTest.setUpc C s | j �| j� d S )N)r� r� r� )r3 r r r rQ � s z$InterruptedSendTimeoutTest.doConnectc O sH z6| � t��"}x| �| j� |||� qW W d Q R X W d | �d� X d S )Nr )r� rr rB rC )r3 r r@ rA r� r r r �checkInterruptedSend� s z/InterruptedSendTimeoutTest.checkInterruptedSendr. r� c C s | � | jjd� d S )Ns aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)rS r� r� )r3 r r r �testInterruptedSendTimeout s z5InterruptedSendTimeoutTest.testInterruptedSendTimeoutc C s | � | jjd| j� d S )Ns aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)rS r� r� r� )r3 r r r � testInterruptedSendtoTimeout
s z7InterruptedSendTimeoutTest.testInterruptedSendtoTimeoutr� c C s | � | jjdg� d S )Ns aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)rS r� r� )r3 r r r � testInterruptedSendmsgTimeout s z8InterruptedSendTimeoutTest.testInterruptedSendmsgTimeout)r6 r7 r8 r4 rQ rS r Zrequires_mac_verrT rU r� r rV rE r r )rB r rO � s
rO c @ s e Zd Zdd� Zdd� ZdS )�
TCPCloserTestc C sh | j �� \}}|�� | j}t�|gg g d�\}}}| �||g� | �|�d�d� |�� |�� d S )Ng �?r[ r� )r/ r� r r| r r� r� )r3 r� r� �sdr r� r� r r r � testClose s zTCPCloserTest.testClosec C s | j �t| jf� t�d� d S )Ng �?)r| r� r� r1 r� r� )r3 r r r �
_testClose+ s zTCPCloserTest._testCloseN)r6 r7 r8 rY rZ r r r r rW s
rW c @ sN e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dS )�BasicSocketPairTestrz c C s t j| |d� d S )N)r{ )r� r< )r3 r{ r r r r< 2 s z BasicSocketPairTest.__init__c C s\ | � |tj� ttd�r*| �|jtj� n| �|jtj� | �|jtj� | �|j d� d S )Nr� r )
r� r r r� r� r� r- r� r. r� )r3 r) r r r �_check_defaults5 s
z#BasicSocketPairTest._check_defaultsc C s | � | j� d S )N)r\ r| )r3 r r r �
_testDefaults> s z!BasicSocketPairTest._testDefaultsc C s | � | j� d S )N)r\ r/ )r3 r r r � testDefaultsA s z BasicSocketPairTest.testDefaultsc C s | j �d�}| �|t� d S )Ni )r/ r� r� r� )r3 r� r r r r
D s z BasicSocketPairTest.testRecvc C s | j �t� d S )N)r| r� r� )r3 r r r r H s z BasicSocketPairTest._testRecvc C s | j �t� d S )N)r/ r� r� )r3 r r r �testSendK s z BasicSocketPairTest.testSendc C s | j �d�}| �|t� d S )Ni )r| r� r� r� )r3 r� r r r � _testSendN s z BasicSocketPairTest._testSendN)rz ) r6 r7 r8 r< r\ r] r^ r
r r_ r` r r r r r[ 0 s
r[ c @ s� e Zd Zd$dd�Zdd� Zdd� Zejdd � �Ze�e�Z e
� e e
d
�d �e�d d
d�dd� ��Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )%�NonBlockingTCPTestsrz c C s t �� | _tj| |d� d S )N)r{ )r= r` �eventry r< )r3 r{ r r r r< U s
z NonBlockingTCPTests.__init__c C s� | j �d� | �| j �� � | �| j �� � tr@| �t| j �� | j �d� | �| j �� d� | � | j �� � tr�| � t| j �� | j �
d � | �| j �� � tr�| �t| j �� | j �
d� | � | j �� � | �| j �� d� tr�| � t| j �� | j �
d� | �| j �� � | �| j �� d� t�r8| � t| j �� | j �
d� | � | j �� � t�rj| � t| j �� t � � }y| j � � W n t
k
�r� Y nX t � � }| �|| dk d� d S )NTFg r r. g �?z Error setting non-blocking mode.)r/ r� rh r0 r� � getblockingr r* r� rI rN r� r� r )r3 rR �endr r r �testSetBlockingY sF z#NonBlockingTCPTests.testSetBlockingc C s d S )Nr )r3 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_MAXFg r[ ) r r# Z ULONG_MAXrM r/ r� r� r0 rh )r3 r r r r �testSetBlocking_overflow� s
z,NonBlockingTCPTests.testSetBlocking_overflowr� z test needs socket.SOCK_NONBLOCKr� � � c C s� | j �� t�tjtjtjB �| _ | �| j �� � | �| j � � d� t
� | j �| _ | j �
� t�� }y| j �� W n tk
r� Y nX t�� }| �|| dk d� d S )Nr g �?z&Error creating with non-blocking mode.)r/ r r r- r. r� rI rc r� r0 r r0 r1 r2 r� r� r r� )r3 rR rd r r r �testInitNonBlocking� s
z'NonBlockingTCPTests.testInitNonBlockingc C s d S )Nr )r3 r r r �_testInitNonBlocking� s z(NonBlockingTCPTests._testInitNonBlockingc C sH | j �d� z | j �� \}}|�tt��}W d |�� | j �d � X d S )Nr. )r/ rN r� r� ra r� r )r3 r� r� �messager r r �testInheritFlags� s z$NonBlockingTCPTests.testInheritFlagsc C s6 t �d� | j�t| jf� t �d� | j�t� d S )Ng�������?g �?)r� r� r| r� r� r1 r� r� )r3 r r r �_testInheritFlags� s
z%NonBlockingTCPTests._testInheritFlagsc C s� | j �d� | �t�� | j �� \}}W d Q R X | j�� t�| j gg g t�\}}}| j |krh| � d� | j �� \}}| �
|j � | � |�
� � d S )Nr z'Error trying to do accept after select.)r/ r� r� �BlockingIOErrorr� rb rY r � MAIN_TIMEOUTr� rC r rh r0 )r3 r� r� r r� r� r r r �
testAccept� s
z NonBlockingTCPTests.testAcceptc C s | j �� | j�t| jf� d S )N)rb rl r| r� r� r1 )r3 r r r � _testAccept� s
z NonBlockingTCPTests._testAcceptc C s | j �� \}}|�� d S )N)r/ r� r )r3 r� r� r r r � testConnect� s z NonBlockingTCPTests.testConnectc C s" | j �d� | j �t| jf� d S )Nr. )r| rN r� r� r1 )r3 r r r � _testConnect� s z NonBlockingTCPTests._testConnectc C s� | j �� \}}| �|j� |�d� | �t�� |�tt ��}W d Q R X | j
� � t � |gg g t
�\}}}||kr|| �d� |�tt ��}| �|t � d S )Nr z0Error during select call to non-blocking socket.)r/ r� rC r r� r� ro r� ra r� rb rY r rp r� r� )r3 r� r� r� r r� r� r r r r
� s
z NonBlockingTCPTests.testRecvc C s, | j �t| jf� | j�� | j �t� d S )N)r| r� r� r1 rb rl rs r� )r3 r r r r � s
z NonBlockingTCPTests._testRecvN)rz )r6 r7 r8 r< re rf r r� rg Z_testSetBlocking_overflowr� r� r r �requires_linux_versionrj rk rm rn rq rr rs rt r
r r r r r ra S s"
2
ra 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 );�FileObjectClassTestCaser zutf-8�strictNr r� rz c C s t j| |d� d S )N)r{ )r� r< )r3 r{ r r r r< 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 )r= r` )r� r+ r r r r� ! s z1FileObjectClassTestCase.setUp.<locals>.<listcomp>r� )�encoding�errors�newline)r� �evt1�evt2�
serv_finished� cli_finishedr� r4 r� r| � read_moderN rx ry rz � read_file)r3 r r r r4 s "
z FileObjectClassTestCase.setUpc C s6 | j �� | j�� | �| jj� d | _t�| � d S )N)r} rY r� r r� �closedr� r5 )r3 r r r r5 ) s
z FileObjectClassTestCase.tearDownc C s0 t �| � | jj| j| j| j| j| jd�| _ d S )N)rx ry rz )
r� rq r� r| �
write_moderN rx ry rz �
write_file)r3 r r r rq 0 s
z#FileObjectClassTestCase.clientSetUpc C s6 | j �� | j�� | �| jj� d | _t�| � d S )N)r~ rY r� r r� r� r� rt )r3 r r r rt 8 s
z&FileObjectClassTestCase.clientTearDownc C sd | j �d� | j�d� | �tj| jjd� | �t��}| j�d� W d Q R X | �dt |j
�� d S )Nr[ r z!cannot read from timed out object) r� rN r� r r� r ru r r� r� r� )r3 Zctxr r r �testReadAfterTimeout? s z,FileObjectClassTestCase.testReadAfterTimeoutc C s. | j �| jdd� � | j �� | j�� d S )Nr r )r� r� � write_msg�flushr} rl )r3 r r r �_testReadAfterTimeoutK s
z-FileObjectClassTestCase._testReadAfterTimeoutc C s< | j �t| j�d �}| j �d�}|| }| �|| j� d S )Nr )r� r ra �read_msgr� )r3 � first_segZ
second_segr� r r r �
testSmallReadP s z%FileObjectClassTestCase.testSmallReadc C s | j �| j� | j �� d S )N)r� r� r� r� )r3 r r r �_testSmallReadW s z&FileObjectClassTestCase._testSmallReadc C s | j �� }| �|| j� d S )N)r� r r� r� )r3 r� r r r � testFullRead[ s
z$FileObjectClassTestCase.testFullReadc C s | j �| j� | j �� d S )N)r� r� r� r )r3 r r r �
_testFullRead` s z%FileObjectClassTestCase._testFullReadc C s> t | j�� }x | j�d�}|s P ||7 }qW | �|| j� d S )Nr[ )r� r� r� r r� )r3 rI �charr r r �testUnbufferedReadd s z*FileObjectClassTestCase.testUnbufferedReadc C s | j �| j� | j �� d S )N)r� r� r� r� )r3 r r r �_testUnbufferedReadn s z+FileObjectClassTestCase._testUnbufferedReadc C s | j �� }| �|| j� d S )N)r� �readliner� r� )r3 �liner r r � testReadliner s
z$FileObjectClassTestCase.testReadlinec C s | j �| j� | j �� d S )N)r� r� r� r� )r3 r r r �
_testReadlinew s z%FileObjectClassTestCase._testReadlinec C s& | j �� | j�� }| �|| j� d S )N)r� r r� r r� r� )r3 r� r r r �testCloseAfterMakefile{ s
z.FileObjectClassTestCase.testCloseAfterMakefilec C s | j �| j� | j �� d S )N)r� r� r� r� )r3 r r r �_testCloseAfterMakefile� s z/FileObjectClassTestCase._testCloseAfterMakefilec C s@ | j �� | j�tt��}t| jt�r.|� � }| �
|| j� d S )N) r� r r� r� ra r� r� r� r� �decoder� )r3 r� r r r � testMakefileAfterMakefileClose� s
z6FileObjectClassTestCase.testMakefileAfterMakefileClosec C s | j �| j� | j �� d S )N)r� r� r� r� )r3 r r r � _testMakefileAfterMakefileClose� s z7FileObjectClassTestCase._testMakefileAfterMakefileClosec C s | � | jj � d S )N)r� r� r� )r3 r r r �testClosedAttr� s z&FileObjectClassTestCase.testClosedAttrc C s | � | jj � d S )N)r� r� r� )r3 r r r �_testClosedAttr� s z'FileObjectClassTestCase._testClosedAttrc C s, | � | jj| j� | � | jj| j�� � d S )N)r� r� r� r r� r� r� )r3 r r r �testAttributes� s z&FileObjectClassTestCase.testAttributesc C s, | � | jj| j� | � | jj| j�� � d S )N)r� r� r� r� r� r� r� )r3 r r r �_testAttributes� s z'FileObjectClassTestCase._testAttributesc C s8 | j �� | �t| j j� | j�� | �t| jj� d S )N)r� r r� r2 r� r� r r� )r3 r r r �
testRealClose� s
z%FileObjectClassTestCase.testRealClosec C s d S )Nr )r3 r r r �_testRealClose� s z&FileObjectClassTestCase._testRealClose)rz )%r6 r7 r8 rN rx ry rz r r� r� r� r� r< r4 r5 rq rt r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rv sB
rv 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 )Ns A. r r s B. )r� r� r� r� r� r| )r3 r� r r r �testUnbufferedReadline� s
z8UnbufferedFileObjectClassTestCase.testUnbufferedReadlinec C s2 | j �d| j � | j �d| j � | j �� d S )Ns A. s B. )r� r� r� r� )r3 r r r �_testUnbufferedReadline� s z9UnbufferedFileObjectClassTestCase._testUnbufferedReadlinec C sD | j �� | j �d�}| �|| j� | j�� | �t| j jd� d S )Ni )r� r r� r� r� r� r� r )r3 r� r r r �testMakefileClose� s
z3UnbufferedFileObjectClassTestCase.testMakefileClosec C s | j �| j� | j �� d S )N)r� r� r� r� )r3 r r r �_testMakefileClose� s z4UnbufferedFileObjectClassTestCase._testMakefileClosec C s6 t �| j�}| j�� t �| j�}| �|d |� d S )Nr[ )r r
r� r� r r� )r3 Zrefcount_beforeZrefcount_afterr r r � testMakefileCloseSocketDestroy� s
z@UnbufferedFileObjectClassTestCase.testMakefileCloseSocketDestroyc C s d S )Nr )r3 r r r � _testMakefileCloseSocketDestroy� s zAUnbufferedFileObjectClassTestCase._testMakefileCloseSocketDestroyc C s | j �d� | �| j�td��d � | �| j�t| j�d �d � | j �
� | j � d� | j�t| j�d �}|d kr�t
�d� | j�t| j�d �}td�}| j�|�}| �|d� ||d |� }| �|| j� | �| j�td��d � | �| j�d�d � d S )NFr. r g �?g �?r r[ )r� r� r� r� �readintorC r ra r� r{ rY r| rl r� r� )r3 r� rI r� r� r r r �testSmallReadNonBlocking� s
z:UnbufferedFileObjectClassTestCase.testSmallReadNonBlockingc C s> | j �d� | j�| j� | j�� | j�� | j�d� d S )Ng �?g @) r{ rl r� r� r� r� r| rY r} )r3 r r r �_testSmallReadNonBlocking� s
z;UnbufferedFileObjectClassTestCase._testSmallReadNonBlockingc C s" | j �d� | jr | �| j� d S )Ng @)r~ rl � serv_skippedrM )r3 r r r �testWriteNonBlocking� s z6UnbufferedFileObjectClassTestCase.testWriteNonBlockingc C sv d | _ | j�d� dtj }d}| j�|�}| �|d� x8t|�D ]&}| j�|�}|d kr\P | �|d� qBW d| _ d S )NFro r. r z$failed to saturate the socket buffer) r� r� r� r rt r� r� r r� )r3 ZBIGZLIMITr� r+ r r r �_testWriteNonBlocking� s
z7UnbufferedFileObjectClassTestCase._testWriteNonBlockingN)r6 r7 r8 rN r� r� r� r� r� r� r� r� r� r� r r r r r� � s
r� c @ s e Zd ZdZdS )�#LineBufferedFileObjectClassTestCaser[ N)r6 r7 r8 rN r r r r r� s r� c @ s e Zd ZdZdS )�$SmallBufferedFileObjectClassTestCaser� N)r6 r7 r8 rN r r r r r� s r� c @ s& e Zd ZdZe�d�ZdZeZdZ dS )�"UnicodeReadFileObjectClassTestCaser zutf-8r� r� N)
r6 r7 r8 r r� r� r� r� r� rz r r r r r� s
r� c @ s& e Zd ZdZeZdZe�d�ZdZ dS )�#UnicodeWriteFileObjectClassTestCaser r� zutf-8r� N)
r6 r7 r8 r r� r� r� r� r� rz r r r r r� ) s
r� c @ s, e Zd ZdZe�d�ZdZe�d�ZdZ dS )�'UnicodeReadWriteFileObjectClassTestCaser zutf-8r� r� N)
r6 r7 r8 r r� r� r� r� r� rz r r r r r� 3 s
r� c @ s e Zd Zdd� ZdS )�NetworkConnectionTestc C s t �t| jf�| _| j| _d S )N)r �create_connectionr� r1 r| r� )r3 r r r rq @ s z!NetworkConnectionTest.clientSetUpN)r6 r7 r8 rq r r r r r� = s r� c @ s e Zd ZdS )�
BasicTCPTest2N)r6 r7 r8 r r r r r� F 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 ru )r3 r@ r r r r� M s z,NetworkConnectionNoServer.MockSocket.connectN)r6 r7 r8 r� r r r r �
MockSocketL s r� c c s& t j }| jt _ z
d V W d |t _ X d S )N)r r� )r3 Z
old_socketr r r �mocked_socket_moduleP s
z.NetworkConnectionNoServer.mocked_socket_modulec C s^ t �� }t�tjtj�}| �|j� | �t��}|� t
|f� W d Q R X | � |j j
t
j� d S )N)r rF r r- r. rC r r� r r� r� r� r� r� Z ECONNREFUSED)r3 r1 r| r� r r r � test_connectZ s z&NetworkConnectionNoServer.test_connectc C sH t �� }| �t��}t�t|f� W d Q R X t �� }| �|j j
|� d S )N) r rF r� r r r� r� Z get_socket_conn_refused_errsr� r� r� )r3 r1 r� Zexpected_errnosr r r �test_create_connectionb s
z0NetworkConnectionNoServer.test_create_connectionc
C s> | � � �, | �tj�� t�tdf� W d Q R X W d Q R X d S )Ni� )r� r� r ru r� r� )r3 r r r � test_create_connection_timeoutz s
z8NetworkConnectionNoServer.test_create_connection_timeoutN) r6 r7 r8 r r� �
contextlib�contextmanagerr� r� r� r� r r r r r� J s
r� c @ sv e Zd Zddd�Zdd� Zdd� Zdd � ZeZd
d � ZeZ d d
� Z
eZ dd� Z eZ
dd� ZeZdd� ZeZdd� ZdS )� NetworkConnectionAttributesTestrz c C s t j| |d� t�| � d S )N)r{ )r, r< rS )r3 r{ r r r r< � s z(NetworkConnectionAttributesTest.__init__c C s t �� | _d S )N)r rF � source_port)r3 r r r rq � s z+NetworkConnectionAttributesTest.clientSetUpc C s | j �� d | _ t�| � d S )N)r| r rS rt )r3 r r r rt � s
z.NetworkConnectionAttributesTest.clientTearDownc C s | j �� \}}|�� d S )N)r/ r� r )r3 r� r� r r r � _justAccept� s z+NetworkConnectionAttributesTest._justAcceptc C s8 t jt| jfdd�| _| �| jj� | �| jjd� d S )N� )ru r� ) r r� r� r1 r| rC r r� r� )r3 r r r � _testFamily� s z+NetworkConnectionAttributesTest._testFamilyc C sH t jt| jfdd| jfd�| _| �| jj� | �| j� � d | j� d S )Nr� r� )ru Zsource_addressr[ )
r r� r� r1 r� r| rC r r� r� )r3 r r r �_testSourceAddress� s z2NetworkConnectionAttributesTest._testSourceAddressc C sd | � t�� d k� t�d� z$t�t| jf�| _| �| jj � W d t�d � X | �
| j� � d� d S )Nr� ) r� r r/ r1 r� r� r1 r| rC r r� r0 )r3 r r r �_testTimeoutDefault� s
z3NetworkConnectionAttributesTest._testTimeoutDefaultc C sh | � t�� d k� t�d� z(tjt| jfd d�| _| �| jj � W d t�d � X | �
| j� � d � d S )Nr� )ru ) r� r r/ r1 r� r� r1 r| rC r r� r0 )r3 r r r �_testTimeoutNone� s
z0NetworkConnectionAttributesTest._testTimeoutNonec C s, t jt| jfdd�| _| �| j�� d� d S )Nr� )ru )r r� r� r1 r| r� r0 )r3 r r r �_testTimeoutValueNamed� s z6NetworkConnectionAttributesTest._testTimeoutValueNamedc C s8 t �t| jfd�| _| �| jj� | �| j�� d� d S )Nr� ) r r� r� r1 r| rC r r� r0 )r3 r r r �_testTimeoutValueNonamed� s z8NetworkConnectionAttributesTest._testTimeoutValueNonamedN)rz )r6 r7 r8 r< rq rt 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 )� NetworkConnectionBehaviourTestrz c C s t j| |d� t�| � d S )N)r{ )r, r< rS )r3 r{ r r r r< � s z'NetworkConnectionBehaviourTest.__init__c C s d S )Nr )r3 r r r rq � s z*NetworkConnectionBehaviourTest.clientSetUpc C s | j �� d | _ t�| � d S )N)r| r rS rt )r3 r r r rt � s
z-NetworkConnectionBehaviourTest.clientTearDownc C s2 | j �� \}}| �|j� t�d� |�d� d S )Nr s done!)r/ r� rC r r� r� r� )r3 r� r� r r r �testInsideTimeout� s
z0NetworkConnectionBehaviourTest.testInsideTimeoutc C s0 t �t| jf� | _}|�d�}| �|d� d S )NrS s done!)r r� r� r1 r| r� r� )r3 r) r� r r r �_testInsideTimeout� s
z1NetworkConnectionBehaviourTest._testInsideTimeoutc s4 t jt| jfdd� | _� | �t j� fdd�� d S )Nr[ )ru c s
� � d�S )NrS )r� r )r) r r r� � r� zDNetworkConnectionBehaviourTest._testOutsideTimeout.<locals>.<lambda>)r r� r� r1 r| r� ru )r3 r )r) r �_testOutsideTimeout� s z2NetworkConnectionBehaviourTest._testOutsideTimeoutN)rz )
r6 r7 r8 r< rq rt 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 )Ng �?)r/ rN r� )r@ rA )r3 r r �
raise_timeout� s z4TCPTimeoutTest.testTCPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (TCP))r� r ru )r3 r� r )r3 r �testTCPTimeout� s
z TCPTimeoutTest.testTCPTimeoutc C s| d}y| j �d� | j �� }W nJ tjk
r> | �d� Y n, tk
rT d}Y n | �d� Y nX |sx| �d� d S )NFg z%caught timeout instead of error (TCP)Tz!caught unexpected exception (TCP)z3accept() returned success when we did not expect it)r/ rN r� r ru r� r )r3 �ok�foor r r �testTimeoutZero� s z TCPTimeoutTest.testTimeoutZerorj ztest needs signal.alarm()c
s
| j �d� G dd� dt�� � fdd�}t�tj|�}z�y�z�yt�d� | j �� }W n` tjk
rt | � d� Y nL � k
r� Y n: | � dt
� � d d� t �
� f � Y n X | � d � W d t�d
� X W n � k
r� | � d � Y nX W d t�tj|� X d S ) Ng @c @ s e Zd ZdS )z4TCPTimeoutTest.testInterruptedTimeout.<locals>.AlarmN)r6 r7 r8 r r r r �Alarm s r� c s � �d S )Nr )rp r� )r� r r �
alarm_handler s z<TCPTimeoutTest.testInterruptedTimeout.<locals>.alarm_handlerr� z caught timeout instead of Alarmz3caught other exception instead of Alarm: %s(%s):
%sznothing caughtr zgot Alarm in wrong place)r/ rN r> rp rq rj r� r ru r� r �exc_info� traceback�
format_exc)r3 r� rw r� r )r� r �testInterruptedTimeout� s(
$z%TCPTimeoutTest.testInterruptedTimeoutN)
r6 r7 r8 r� r� r� r� r rp 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 )Ng �?i )r/ rN r� )r@ rA )r3 r r r� " s z4UDPTimeoutTest.testUDPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (UDP))r� r ru )r3 r� r )r3 r �testUDPTimeout! s
z UDPTimeoutTest.testUDPTimeoutc C s~ d}y | j �d� | j �d�}W nJ tjk
r@ | �d� Y n, tk
rV d}Y n | �d� Y nX |sz| �d� d S )NFg i z%caught timeout instead of error (UDP)Tz!caught unexpected exception (UDP)z1recv() returned success when we did not expect it)r/ rN r� r ru r� r )r3 r� r� r r r r� ( s z UDPTimeoutTest.testTimeoutZeroN)r6 r7 r8 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 )N)r� �
issubclassr r> r r� r� ru )r3 r r r �testExceptionTree8 s z TestExceptions.testExceptionTreec C sb t � t jt jd�}t � t jt jd|�� �}|�� | �|j� | �t�� |� d� W d Q R X d S )Nr F)
r r- r. r� r rC r� r� r r� )r3 Zsock0r) r r r �test_setblocking_invalidfd> s z)TestExceptions.test_setblocking_invalidfdN)r6 r7 r8 r� r� r r r r r� 6 s r� r$ 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 d Q R X W d Q R X W d Q R X d S )Ns python-test-hello �r )
r r� r. rK r2 r� r� r� r� Z getpeername)r3 �address�s1r� Zs3r r r �testLinuxAbstractNamespaceP s
z5TestLinuxAbstractNamespace.testLinuxAbstractNamespacec C sL dd| j d }t�tjtj�� }|�|� | �|�� |� W d Q R X d S )Nr� � hr[ )�
UNIX_PATH_MAXr r� r. rK r� r� )r3 r� r r r r � testMaxName[ s
z&TestLinuxAbstractNamespace.testMaxNamec C s> dd| j }t�tjtj��}| �t|j|� W d Q R X d S )N� �h)r� r r� r. r� r rK )r3 r� r r r r �testNameOverflowa s z+TestLinuxAbstractNamespace.testNameOverflowc C s> t � t jt j�}z |�d� | �|�� d� W d |�� X d S )Nz
python test s
python test )r r� r. rK r� r� r )r3 r r r r � testStrNamef s
z&TestLinuxAbstractNamespace.testStrNamec C s> t � t jt j��$}|�td�� | �|�� d� W d Q R X d S )Ns
python test )r r� r. rK rC r� r� )r3 r r r r �testBytearrayNameo s z,TestLinuxAbstractNamespace.testBytearrayNameN) r6 r7 r8 r� r� r� r� r� r� r r r r r� K 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 )N)r r� r. r) )r3 r r r r4 x s zTestUnixDomain.setUpc C s | j �� d S )N)r) r )r3 r r r r5 { s zTestUnixDomain.tearDownc
C s: y
t �|�S tk
r4 | �d�|t�� �� Y nX d S )NzBPathname {0!a} cannot be represented in file system encoding {1!r})r' �fsencoderd rM r� r �getfilesystemencoding)r3 r� r r r �encoded~ s
zTestUnixDomain.encodedc
C sX yt �||� W nB tk
rR } z$t|�dkr@| �d�|�� n� W d 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� rM r� )r3 r) r� rv r r r rK � s
zTestUnixDomain.bindc C s | � | j�� d� d S )N)r� N)r� r) r� )r3 r r r � testUnbound� s zTestUnixDomain.testUnboundc C s@ t j�tj�}| �| j|� | �tj|� | � | j�
� |� d S )N) r' r� �abspathr �TESTFNrK r) rC r� r� r� )r3 r� r r r � testStrAddr� s zTestUnixDomain.testStrAddrc C sF t j�tj�}| �| j| �|�� | �tj |� | �
| j� � |� d S )N) r' r� r� r r� rK r) r� rC r� r� r� )r3 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� rK r) r� rC r� r� r� )r3 r� rf 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_UNENCODABLErM r' r� r� rK r) rC r� r� r� )r3 r� r r r �testUnencodableAddr� s
z"TestUnixDomain.testUnencodableAddrN) r6 r7 r8 r4 r5 r� rK r� r� r� r� r� r r r r r� u 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 ) � BufferIOTestrz c C s t j| |d� d S )N)r{ )r� r< )r3 r{ r r r r< � s zBufferIOTest.__init__c C sZ t � ddgtt� �}| j�|�}| �|tt�� |�� }|d tt�� }| �|t� d S )NrZ r )r[ ra r� r� rH r� r� )r3 rI r� r� r r r �testRecvIntoArray� s z BufferIOTest.testRecvIntoArrayc C s t t�}| j�|� d S )N)r� r� r� r� )r3 rI r r r �_testRecvIntoArray� s z BufferIOTest._testRecvIntoArrayc C sD t d�}| j�|�}| �|tt�� |d tt�� }| �|t� d S )Ni )rC r� rH r� ra r� )r3 rI r� r� r r r �testRecvIntoBytearray� s
z"BufferIOTest.testRecvIntoBytearrayc C sH t d�}| j�t|��}| �|tt�� |d tt�� }| �|t� d S )Ni )rC r� rH r� r� ra r� )r3 rI r� r� r r r �testRecvIntoMemoryview� s
z#BufferIOTest.testRecvIntoMemoryviewc C s^ t � ddgtt� �}| j�|�\}}| �|tt�� |�� }|d tt�� }| �|t� d S )NrZ r )r[ ra r� r� rK r� r� )r3 rI r� r� r� r r r �testRecvFromIntoArray� s z"BufferIOTest.testRecvFromIntoArrayc C s t t�}| j�|� d S )N)r� r� r� r� )r3 rI r r r �_testRecvFromIntoArray� s z#BufferIOTest._testRecvFromIntoArrayc C sH t d�}| j�|�\}}| �|tt�� |d tt�� }| �|t� d S )Ni )rC r� rK r� ra r� )r3 rI r� r� r� r r r �testRecvFromIntoBytearray� s
z&BufferIOTest.testRecvFromIntoBytearrayc C sL t d�}| j�t|��\}}| �|tt�� |d tt�� }| �|t� d S )Ni )rC r� rK r� r� ra r� )r3 rI r� r� r� r r r �testRecvFromIntoMemoryview� s
z'BufferIOTest.testRecvFromIntoMemoryviewc C s t d�}| �t| jj|d� d S )NrJ i )rC r� r2 r� rK )r3 rI r r r �testRecvFromIntoSmallBuffer� s z(BufferIOTest.testRecvFromIntoSmallBufferc C s | j �t� d S )N)r� r� r� )r3 r r r � _testRecvFromIntoSmallBuffer s z)BufferIOTest._testRecvFromIntoSmallBufferc C s$ t � }| j�|� | j�|d� d S )Nr )rC r� rK )r3 rI r r r �testRecvFromIntoEmptyBuffer s z(BufferIOTest.testRecvFromIntoEmptyBufferN)rz )r6 r7 r8 r< r� r� r� Z_testRecvIntoBytearrayr� Z_testRecvIntoMemoryviewr� r� r� Z_testRecvFromIntoBytearrayr� Z_testRecvFromIntoMemoryviewr r r Z _testRecvFromIntoEmptyBufferr r r r r� � s"
r� i� �� �� c
C sf t td�sdS y td�} W n tttfk
r4 dS X | �" x| D ]}|�d�rBdS qBW W d Q R X dS )N�AF_TIPCFz
/proc/modulesztipc T)r r r
�FileNotFoundError�IsADirectoryError�PermissionErrorr+ )r r� r r r �isTipcAvailable s
r z6TIPC module is not loaded, please 'sudo modprobe tipc'c @ s e Zd Zdd� ZdS )�TIPCTestc C s� t � t jt j�}t � t jt j�}| �|j� | �|j� |�t jt jd� t jt t
t f}|� |� t j
t t
tt t
d � df}|�t|� |�d�\}}| �|�� |� | �|t� d S )Nr[ r� r i )r r r� rC r rL r� r� �TIPC_ADDR_NAMESEQ�
TIPC_STYPE�
TIPC_LOWER�
TIPC_UPPERrK �TIPC_ADDR_NAMEr r� r� r� r� r� )r3 r� r| �srvaddrZsendaddrr� Zrecvaddrr r r �testRDM) s
zTIPCTest.testRDMN)r6 r7 r8 r r r r r r
& s r
c @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )�TIPCThreadableTestrz c C s t jj| |d� t�| � d S )N)r{ )r� r� r< rS )r3 r{ r r r r< A 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 )Nr[ )r r r. r� rC r rL r� r� r r r
r rK r2 rZ r� r� r� )r3 r r r r r4 E s
zTIPCThreadableTest.setUpc C sd t �d� t�tjtj�| _| �| jj� tjt t
t t t
d � df}| j�
|� | j�� | _d S )Ng�������?r� r )r� r� r r r. r| rC r r r r
r r r� r� �cliaddr)r3 r� r r r rq Q s
z TIPCThreadableTest.clientSetUpc C s, | j �d�}| �|t� | �| j| j� d S )Ni )r� r� r� r� r r� )r3 r� r r r r� ] s z TIPCThreadableTest.testStreamc C s | j �t� | j �� d S )N)r| r� r� r )r3 r r r r� b s z TIPCThreadableTest._testStreamN)rz )r6 r7 r8 r< r4 rq r� r� r r r r r > s
r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ContextManagersTestc C s� t � � �}| �|j� W d Q R X | �|j� t � � �}|�� W d Q R X | �|j� t � � �}| �t|jd� W d Q R X | �|j� d S )Ns foo)r rI r% r� r r� r rs )r3 r) r r r �_testSocketClassi s
z$ContextManagersTest._testSocketClassc C s2 | j �� \}}| �|j� |�d�}|�|� d S )Ni )r/ r� rC r r� rs )r3 r� r� r� r r r �testCreateConnectionBasew s
z,ContextManagersTest.testCreateConnectionBasec C sX | j �� }t�|��.}| �|j� |�d� | �|�d�d� W d Q R X | � |j� d S )Ns fooi )
r/ r� r r� rI r% rs r� r� r� )r3 r� r) r r r �_testCreateConnectionBase} s
z-ContextManagersTest._testCreateConnectionBasec C s2 | j �� \}}| �|j� |�d�}|�|� d S )Ni )r/ r� rC r r� rs )r3 r� r� r� r r r �testCreateConnectionClose� s
z-ContextManagersTest.testCreateConnectionClosec C sH | j �� }t�|��}|�� W d Q R X | �|j� | �t|j d� d S )Ns foo)
r/ r� r r� r r� r% r� r rs )r3 r� r) r r r �_testCreateConnectionClose� s
z.ContextManagersTest._testCreateConnectionCloseN)r6 r7 r8 r r r r r r r r r r g 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 )�InheritanceTestrQ zSOCK_CLOEXEC not definedr� rh ri c C sD t � t jt jt jB ��$}| �|jt j� | �|�� � W d Q R X d S )N)r r- r. rQ r� r� rI �get_inheritable)r3 r r r r �test_SOCK_CLOEXEC� s z!InheritanceTest.test_SOCK_CLOEXECc C s, t � � }|� | �|�� d� W d Q R X d S )NF)r r� r )r3 r) r r r �test_default_inheritable� s z(InheritanceTest.test_default_inheritablec
C sL t � � }|�6 |�� }|�� |� | �|�� d� W d Q R X W d Q R X d S )NF)r r r r� r )r3 r) Znewsockr r r �test_dup� s zInheritanceTest.test_dupc C sP t � � }|�: |�d� | �|�� d� |�d� | �|�� d� W d Q R X d S )NTF)r �set_inheritabler� r )r3 r) r r r �test_set_inheritable� s
z$InheritanceTest.test_set_inheritableNz
need fcntlc C sn t � � }|�X |�� }| �|�� d� t�|tj�}|tj M }t�|tj|� | �|�� d� W d Q R X d S )NFT)r r� r� r r �F_GETFD�
FD_CLOEXECZF_SETFD)r3 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 d Q R X d S )NTr )r r� r� r r! r" r )r3 r) r� r r r � test_set_inheritable_cloexec� s
z,InheritanceTest.test_set_inheritable_cloexecc C sH t �� \}}| �|j� | �|j� | �|�� d� | �|�� d� d S )NF)r r� rC r r� r )r3 r� r� r r r �test_socketpair� s
z InheritanceTest.test_socketpair)r6 r7 r8 r� r� r r r ru r r r r r� r r# r$ r% r r r r r � s
r r� zSOCK_NONBLOCK not definedc @ s, e Zd Zd dd�Ze�ddd�dd � �Zd
S ) �NonblockConstantTestT� c C s� |rj| � |jtj� | � |�� |� | �t�|tjtj �tj @ � |dkrZ| �
|� � � q�| �|� � � nL| � |jtj� | � |�� d � | �
t�|tjtj �tj @ � | �|� � � d S )Nr ) r� r� r r. r0 r� r r& r' r( rI rc )r3 r �nonblockru r r r �
checkNonblock� s z"NonblockConstantTest.checkNonblockr� rh ri c C s^ t � t jt jt jB ���}| �|� |�d� | j|dd� |�d� | �|� |�d � | j|dd� |�d� | j|dd� |�d� | j|dd� W d Q R X t �� }t �d� t � � �}| �|� W d Q R X t �d � t � � �}| �|d� W d Q R X t �d� t � � �}| j|dd� W d Q R X t �d � t � � �}| �|d� W d Q R X t �|� d S )Nr[ F)r( r g @)ru g ) r r- r. r� r) r� rN r/ r1 )r3 r �tr r r �test_SOCK_NONBLOCK� s6
z'NonblockConstantTest.test_SOCK_NONBLOCKN)Tr' )r6 r7 r8 r) r ru r+ r r r r r& � s
r& rW zWindows specificzneed multiprocessingc @ s@ e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
S )�TestSocketSharingc C sD |� � }|� � }t�|�}|�� \}}|�|� |�� |�� d S )N)rn r � fromsharer� rs r )r� �qZsdatarl r r� rv r r r �remoteProcessServer s
z%TestSocketSharing.remoteProcessServerc C s� t �� }t j| j|fd�}|�� | j�|j�}| j�� }| j� � |�
|� d}|�
|� t � |�}g }x |�
d�}|s|P |�|� qlW |� � d�|�}| �||� |�� d S )N)rP r@ s
slapmahfrorY r� )�multiprocessingrd ZProcessr/ rR r/ �share�pidr� r rs r r� r� r� r� r� ) r3 r. r� r� r� rl r �m�receivedr r r � testShare' s(
zTestSocketSharing.testSharec C s@ | j �t�� �}| �ttj|d d� � | �ttj|d � d S )Nr s foo)r/ r1 r' �getpidr� r2 r r- )r3 r� r r r �testShareLengthJ s z!TestSocketSharing.testShareLengthc C sb | � |�� d � | � |�� |�� � | � |j|j� | � |j|j� |jdkr^| � |j|j� d S )Nr )r� r0 r� r� r� )r3 Zorg�otherr r r �compareSocketsO s
z TestSocketSharing.compareSocketsc C s< | j �t�� �}t�|�}z| �| j |� W d |�� X d S )N)r/ r1 r' r6 r r- r9 r )r3 r� r r r r �testShareLocal_ s
z TestSocketSharing.testShareLocalc
C s� t jt jg}t jt jg}x�|D ]~}xx|D ]p}yt � ||�}W n tk
rR w(Y nX z8|�t�� �}t � |�}z| �
||� W d |� � X W d |� � X q(W q W d S )N) r r- r� r. r r r1 r' r6 r- r9 r )r3 Zfamilies�typesr r* �sourcer� Zsharedr r r � testTypesg s
zTestSocketSharing.testTypesN)
r6 r7 r8 r� r/ r5 r7 r9 r: r= 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 � i r� r� c C s� dd� }d� dd� t| j�D ��}ttjd��(}x || j| j�D ]}|�|� q@W W d Q R X ttjd��}|�� | _ W d Q R X d S )Nc s s* x| |kr|V | |8 } qW | r&| V d S )Nr )Ztotal�stepr r r �chunks� s
z0SendfileUsingSendTest.setUpClass.<locals>.chunksr� c S s g | ]}t �tj��� �qS r )�random�choice�stringZ
ascii_lettersr� )r� r+ r r r r� � s z4SendfileUsingSendTest.setUpClass.<locals>.<listcomp>r� r )
r� r� �BUFSIZEr
r r� �FILESIZEr� r �FILEDATA)r� r@ �chunkr Zcsizer r r �
setUpClass� s
z SendfileUsingSendTest.setUpClassc C s t �t j� d S )N)r r� r� )r� r r r �
tearDownClass� s z#SendfileUsingSendTest.tearDownClassc C s6 | j �t� | j �� \}}|�| j� | �|j� |S )N)r/ rN rp r� �TIMEOUTrC r )r3 r� r� r r r � accept_conn� s
z!SendfileUsingSendTest.accept_connc C s0 g }x |� | j�}|sP |�|� qW d�|�S )Nr� )r� rD r� r� )r3 r� r4 rG r r r � recv_data� s z SendfileUsingSendTest.recv_datac C s
t |d�S )NZ_sendfile_use_send)rh )r3 r) r r r �meth_from_sock� s z$SendfileUsingSendTest.meth_from_sockc
C sr | j �� }ttjd�}t�|��H}|�8}| �|�}||�}| �|| j � | �|�
� | j � W d Q R X W d Q R X d S )Nr ) r/ r� r
r r� r r� rM r� rE �tell)r3 r� �filer) �meth�sentr r r �_testRegularFile� s
z&SendfileUsingSendTest._testRegularFilec C s6 | � � }| �|�}| �t|�| j� | �|| j� d S )N)rK rL r� ra rE rF )r3 r� r� r r r �testRegularFile� s
z%SendfileUsingSendTest.testRegularFilec
C s| | j �� }t�| j�}t�|��R}|�B}|�|�}| �|| j � | �|�
� | j � | � tj |j
|� W d Q R X W d Q R X d S )N)r/ r� �io�BytesIOrF r r� r� r� rE rN r� r� r� )r3 r� rO r) rQ r r r �_testNonRegularFile� s
z)SendfileUsingSendTest._testNonRegularFilec C s6 | � � }| �|�}| �t|�| j� | �|| j� d S )N)rK rL r� ra rE rF )r3 r� r� r r r �testNonRegularFile� s
z(SendfileUsingSendTest.testNonRegularFilec
C s� | j �� }tjd }t|d�� | �tj|� W d Q R X t|d�}t�|��D}|�4}| � |�}||�}| �
|d� | �
|� � d� W d Q R X W d Q R X d S )N�2r� r r ) r/ r� r r� r
rC r� r r� rM r� rN )r3 r� �filenamerO r) rP rQ r r r �_testEmptyFileSend� s
z(SendfileUsingSendTest._testEmptyFileSendc C s" | � � }| �|�}| �|d� d S )Nr� )rK rL r� )r3 r� r� r r r �testEmptyFileSend� s
z'SendfileUsingSendTest.testEmptyFileSendc
C sz | j �� }ttjd�}t�|��P}|�@}| �|�}||dd�}| �|| j d � | �|�
� | j � W d Q R X W d Q R X d S )Nr i� )�offset) r/ r� r
r r� r r� rM r� rE rN )r3 r� rO r) rP rQ r r r � _testOffset� s
z!SendfileUsingSendTest._testOffsetc C sB | � � }| �|�}| �t|�| jd � | �|| jdd � � d S )Ni� )rK rL r� ra rE rF )r3 r� r� r r r �
testOffset� s
z SendfileUsingSendTest.testOffsetc
C sz | j �� }ttjd�}tj|dd��L}|�<}d}| �|�}|||d�}| �||� | �|� � |� W d Q R X W d Q R X d S )Nr r� )ru iGKL )rU )
r/ r� r
r r� r r� rM r� rN )r3 r� rO r) rU rP rQ r r r �
_testCount� s
z SendfileUsingSendTest._testCountc C s@ d}| � � }| �|�}| �t|�|� | �|| jd |� � d S )NiGKL )rK rL r� ra rF )r3 rU r� r� r r r � testCount s
z SendfileUsingSendTest.testCountc
C sz | j �� }ttjd�}tj|dd��L}|�<}d}| �|�}|||d�}| �||� | �|� � |� W d Q R X W d Q R X d S )Nr r� )ru r[ )rU )
r/ r� r
r r� r r� rM r� rN )r3 r� rO r) rU rP rQ r r r �_testCountSmall
s
z%SendfileUsingSendTest._testCountSmallc C s@ d}| � � }| �|�}| �t|�|� | �|| jd |� � d S )Nr[ )rK rL r� ra rF )r3 rU r� r� r r r �testCountSmall s
z$SendfileUsingSendTest.testCountSmallc
C s� | j �� }ttjd�}tj|dd��R}|�B}d}| �|�}||d|d�}| �||� | �|� � |d � W d Q R X W d Q R X d S )Nr r� )ru i�� i� )r\ rU )
r/ r� r
r r� r r� rM r� rN )r3 r� rO r) rU rP rQ r r r �_testCountWithOffset s
z*SendfileUsingSendTest._testCountWithOffsetc C sD d}| � � }| �|�}| �t|�|� | �|| jd|d � � d S )Ni�� i� )rK rL r� ra rF )r3 rU r� r� r r r �testCountWithOffset* s
z)SendfileUsingSendTest.testCountWithOffsetc
C sr | j �� }ttjd�}t�|��H}|�8}|�d� | �|�}| � t
||� | � t
|j |� W d Q R X W d Q R X d S )Nr F) r/ r� r
r r� r r� r� rM r� r2 r� )r3 r� rO r) rP r r r �_testNonBlocking3 s
z&SendfileUsingSendTest._testNonBlockingc C s | � � }|�d�r | �d� d S )Ni z$was not supposed to receive any data)rK r� r� )r3 r� r r r �testNonBlocking< s
z%SendfileUsingSendTest.testNonBlockingc
C sd | j �� }ttjd�}tj|dd��6}|�&}| �|�}||�}| �|| j � W d Q R X W d Q R X d S )Nr r� )ru )
r/ r� r
r r� r r� rM r� rE )r3 r� rO r) rP rQ r r r �_testWithTimeoutC s
z&SendfileUsingSendTest._testWithTimeoutc C s6 | � � }| �|�}| �t|�| j� | �|| j� d S )N)rK rL r� ra rE rF )r3 r� r� r r r �testWithTimeoutK s
z%SendfileUsingSendTest.testWithTimeoutc
C s` | j �� }ttjd��@}t�|��*}|�d� | �|�}| � tj
||� W d Q R X W d Q R X d S )Nr g{�G�z�?) r/ r� r
r r� r r� rN rM r� ru )r3 r� rO r) rP r r r � _testWithTimeoutTriggeredSendS s
z3SendfileUsingSendTest._testWithTimeoutTriggeredSendc C s | � � }|�d� d S )Ni�X )rK r� )r3 r� r r r � testWithTimeoutTriggeredSend[ s z2SendfileUsingSendTest.testWithTimeoutTriggeredSendc C s d S )Nr )r3 r r r � _test_errorsa s z"SendfileUsingSendTest._test_errorsc
C s t tjd��:}tjtjd�� }| �|�}| �td||� W d Q R X W d Q R X t tjd��4}t�� � }| �|�}| �td||� W d Q R X W d Q R X t tjd��t}t�� �`}| �|�}| jtd||dd� | jtd||d d� | jtd||d
d� | jtd||d d� W d Q R X W d Q R X d S ) Nr )r� r. r� z binary modezpositive integerrX )rU g�������?r r ) r
r r� r r rM r� r2 ru )r3 rO r rP r r r � test_errorsd s*
z!SendfileUsingSendTest.test_errorsN)#r6 r7 r8 rE rD rF rJ r� rH rI rK rL rM rR rS rV rW rZ r[ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl r r r r r> { s>
r> r� z%os.sendfile() required for this test.c @ s e Zd Zdd� ZdS )�SendfileUsingSendfileTestc C s
t |d�S )Nr� )rh )r3 r) r r r rM � s z(SendfileUsingSendfileTest.meth_from_sockN)r6 r7 r8 rM r r r r rm | s rm 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�}y|�||f� W n< tk
r` } z |�� t�t|�||��W d d }~X Y nX |S d S )Nr ) r r! r rK r r r� ZSkipTestr� )r3 �typr� r) rv r r r �
create_alg� s $z LinuxKernelCryptoAPI.create_algr� rS c
C s� t �d�}| �dd���}|�� \}}|�" |�d� | �|�d�|� W d Q R X |�� \}}|�L |�dtj � |�dtj � |�dtj � |�d � | �|�d�|� W d Q R X W d Q R X d S )
NZ@ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad�hashZsha256s abci � a� b� cr� )
r� �fromhexrp r� rs r� r� r� r �MSG_MORE)r3 r� �algo�opr] r r r � test_sha256� s
z LinuxKernelCryptoAPI.test_sha256c
C sp t �d�}| �dd��P}|�tjtjd� |�� \}}|�" |�d� | � |�
d�|� W d Q R X W d Q R X d S )NZ(effcdf6ae5eb2fa2d27416d5f184df9c259a7c79rq z
hmac(sha1)s Jefes what do ya want for nothing?i ) r� ru rp rL r �SOL_ALG� ALG_SET_KEYr� rs r� r� )r3 r� rw rx 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 d Q R X |�� \}}|�, |j |gtj|d� | �
|�|�|� W d Q R X d } |g| }
|�� \}}|�& |j |
tj
|d� |�|| �} W d Q R X | �
t| �|| � | �
| d |� |� |�� \}}|�( |j | gtj|d� |�|| �} W d Q R X | �
t| �|| � | �
| || � W d Q R X d S )
NZ 06a9214036b8a15b512e03d534120006Z 3dafba429d9eb430b422da802c9fac41s Single block msgZ e353779c1079aeb82708942dbe77181aZskcipherzcbc(aes))rx �ivr^ )rx r} i )r� ru ra rp rL r rz r{ r� �
sendmsg_afalg�ALG_OP_ENCRYPTrv rs r� r� �ALG_OP_DECRYPT)
r3 �keyr} r� Z
ciphertextZmsglenrw rx r] Z
multiplierZlongmsg�encZdecr r r � test_aes_cbc� sD
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 d Q R X | � � \}
} |
�f || }
|
j
|
gtj ||d
� |
�|t|� | �} | �|| || � � | �|| | d � � W d Q R X t�d �j}| � � \}
} |
�� || }
|
�|
gtjtj|tj �gtjtj|t|��| gtjtj||�gf� |
�t|
�| �} | �|| || � � | �|| | d � � W d Q R X | � � \}
} |
�N || | }
|
j
|
gtj||d
� |
�t|
�| �} | �|| |d � � W d Q R X W d Q R X d S ) NZ c939cc13397c1d37de6ae0e1cb7c423cZb3d8cc017cbb89b39e0f67e2Z c3b3c41f113a31b73d9a5cd432103069Z 24825602bd12a984e0092d3e448eda5fZ 93fe7d9e9bfd10348a5606e5cafa7354Z 0032a1dc85f1c9786925a2e71d8272ddZaeadzgcm(aes))rx r} �assoclenr^ )rx r} r� r )r� ru ra rp rL r rz r{ ZALG_SET_AEAD_AUTHSIZEr� r~ r rv rs r� r� r
�Structr� r� Z
ALG_SET_OPZ
ALG_SET_IVZALG_SET_AEAD_ASSOCLENr� )r3 r� r} ZplainZassocZ expected_ctZ expected_tagZtaglenr� rw rx r] Zresr� Z pack_uint32r r r �test_aead_aes_gcm� s`
z&LinuxKernelCryptoAPI.test_aead_aes_gcmc
C sn | � dd��X}t�d�}|�tjtj|� |�� \}}|� |�d�}| � t
|�d� W d Q R X W d Q R X d S )N�rngZdrbg_pr_sha256r ) rp r' �urandomrL r rz r{ r� r� r� ra )r3 rw Z
extra_seedrx r] Zrnr r r �test_drbg_pr_sha256 s
z(LinuxKernelCryptoAPI.test_drbg_pr_sha256c
C s� t � t jt jd�}|�� | �t�� |�� W d Q R X | �t�� |jd d� W d Q R X | �t�� |�d� W d Q R X | �t�� |jt jd d� W d Q R X | �t�� |jt jdd� W d Q R X W d Q R X d S )Nr )rx r[ )rx r� r )r r! r r� ru r~ r )r3 r) r r r �test_sendmsg_afalg_args& s z,LinuxKernelCryptoAPI.test_sendmsg_afalg_argsc C s� t � t jt jd�}| �|j� | �t�� |�d� W d Q R X | �t d�� |�d� W d Q R X | �t�� |�d� W d Q R X | �t d�� |�d� W d 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 rC r r� r rK r� r2 )r3 r) r r r �test_length_restriction8 s z,LinuxKernelCryptoAPI.test_length_restrictionN)
r6 r7 r8 rp r ru ry r| r� r� r� r� r� r r r r rn � s
)? rn �winzrequires Windowsc @ s$ e Zd ZddddddhZdd� Zd S )
�TestMSWindowsTCPFlagsZ
TCP_MAXSEGZ TCP_NODELAYZ TCP_FASTOPENZ TCP_KEEPCNTZ TCP_KEEPIDLEZ
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� Z s z<TestMSWindowsTCPFlags.test_new_tcp_flags.<locals>.<listcomp>c s g | ]}|� j kr|�qS r )�
knownTCPFlags)r� r )r3 r r r� [ s zANew TCP flags were discovered. See bpo-32394 for more information)r� r r� )r3 Zprovided�unknownr )r3 r �test_new_tcp_flagsY s z(TestMSWindowsTCPFlags.test_new_tcp_flagsN)r6 r7 r8 r� r� r r r r r� J s r� c C s t ttttttttg } | � t
t t t
ttttttttttg� | �t� | �t� | �t� | � t t g� | � t t g� | � t t!g� | �t"� | � t#t$g� | � t%t&t't(t)t*t+t,t-t.t/t0t1t2t3t4t5t6t7t8t9t:t;t<t=g� | �t>� t?�@� }t?jA| � t?jB|� d S )N)Cr� r rW r� r� r� r� r( r� �extendra rv r� r� r� r� r� r� r� r� r� r r r& r� r[ r� r� r
r r� r� r� r� rn r r� r� r& r' r( r* r+ r- r. r, r0 r1 r2 r4 r5 r7 r9 r: r; r< r= rE rO r, r> rm r� r Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_main` sr
r� �__main__)�r� Ztestr r� rT rT r r r� r� r� rc r r' r[ r� �weakrefr rp rk r� r
rA r� rC �_threadri r= r0 r> r r� r� r� rp r� r� r r r r r" r$ r* ZHAVE_SOCKET_CANZHAVE_SOCKET_CAN_ISOTPZHAVE_SOCKET_RDSZHAVE_SOCKET_ALGZHAVE_SOCKET_VSOCKr� r� r� r, r9 r: rF rR rS 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- rE rJ rM rN rP rQ rR rl rt rw r� r� r� r� r� r� r% r& r' r( r) rb r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r7 r8 r9 r: r; r< r= r> r rE rO rW r[ ra rv 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> rm rn r+ r� r� r6 r r r r �<module> s:
3u ! #
D]
*
? .{7w w= # P; k 7
) 5# 4 $l
8E 9
)
ES
',C
<
e
D&
A