Your IP : 216.73.216.85
U
i�f� �
@ s�
d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlmZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z z d dl Z W n e k
r� dZ Y nX z d dl Z W n e k
�r$ dZ Y nX ej Z d�!d�Z"dZ#dZ$e�%� d kZ&z d dl'Z'W n e k
�rp dZ'Y nX d
d � Z(d d
� Z)dd� Z*dd� Z+dd� Z,dd� Z-dd� Z.ej/dd� �Z0e)� Z1e*� Z2e+� Z3e,� Z4e-� Z5e.� Z6e�d�j7Z8G dd � d e j9�Z:G d d � d e j9�Z;G d d � d e j9�Z<G d!d"� d"e j9�Z=G d#d$� d$e j9�Z>G d%d&� d&�Z?G d'd(� d(e:e?�Z@G d)d*� d*e;e?�ZAG d+d,� d,e=e?�ZBG d-d.� d.e>e?�ZCe �De dkd/�e �Ee6d0�e �Ee(� d1kd2�G d3d4� d4e j9e?����ZFG d5d6� d6e@�ZGG d7d8� d8e j9e?�ZHG d9d:� d:e j9�ZIG d;d<� d<eI�ZJG d=d>� d>e<eIe?�ZKG d?d@� d@eJeK�ZLG dAdB� dBeI�ZMG dCdD� dDeM�ZNG dEdF� dFeI�ZOG dGdH� dHeO�ZPG dIdJ� dJeO�ZQG dKdL� dLeO�ZRG dMdN� dNeO�ZSG dOdP� dPeS�ZTdQdR� ZUdSdT� ZVdUdV� ZWG dWdX� dXe j9�ZXe �Ee1dY�G dZd[� d[e j9��ZYe �Ee1dY�G d\d]� d]eB��ZZe �Ee2d^�G d_d`� d`e j9��Z[e �Ee3da�G dbdc� dce j9��Z\e �Ee3da�G ddde� deeC��Z]e �Ee5df�G dgdh� dhe j9��Z^e �De dkd/�e �Ee6d0�G didj� dje j9���Z_G dkdl� dleG�Z`G dmdn� dneA�ZaG dodp� dpe<�ZbG dqdr� dreb�ZcG dsdt� dteb�ZdG dudv� dveb�ZeG dwdx� dxeb�ZfG dydz� dzeb�ZgG d{d|� d|eb�ZhG d}d~� d~eh�ZiG dd�� d�ei�ZjG d�d�� d�ei�ZkG d�d�� d�eb�ZlG d�d�� d�el�ZmG d�d�� d�el�ZnG d�d�� d�ecel�ZoG d�d�� d�e j9�ZpG d�d�� d�eh�ZqG d�d�� d�eh�ZrG d�d�� d�edefeKeQ�ZseVejd��G d�d�� d�ekes��ZteVejd��G d�d�� d�enes��ZueVejd��G d�d�� d�eoes��ZvG d�d�� d�edefeKeT�ZweVejd��e �Eejxd��eWd�d��G d�d�� d�ekew����ZyeVejd��e �Eejxd��eWd�d��G d�d�� d�enew����ZzeVejd��e �Eejxd��eWd�d��G d�d�� d�eoew����Z{eVejd��e �Eejxd��eVed��eWd�d��G d�d�� d�erew�����Z|eVejd��e �Eejxd��eVed��eWd�d��G d�d�� d�ecerew�����Z}G d�d�� d�egeLeP�Z~eVejd��G d�d�� d�eje~��ZeVejd��G d�d�� d�eneme~��Z�eVejd��G d�d�� d�eoeme~��Z�G d�d�� d�eeegeLeR�Z�eVejd��e �De&d��eWd�d�d��G d�d�� d�eje�����Z�eVejd��e �De&d��eWd�d�d��G d�d�� d�eneme�����Z�eVejd��e �De&d��eWd�d�d��G d�d�� d�eoeme�����Z�G d�d�� d�egeLeN�Z�eVejd��eVedG d�dĄ d�eje����Z�eVejd��eVedG d�dƄ d�eneme����Z�eVejd��eVedG d�dȄ d�eoeme����Z�eVejd�d��eVed�d�dʃG d�d̄ d�eqe����Z�eVejd�d��eVed�d�dʃG d�d΄ d�eceqe����Z�G d�dЄ d�e j9�Z�eVedуe �Ee�ed҃� p�e�edӃdԡG d�dք d�e�eQ���Z�eVedуe �Ee�ed҃�
p4e�edӃdԡG d�d d�e�e<eJeP���Z�G d�dڄ d�e@�Z�G d�d܄ d�eH�Z�G d�dބ d�eH�Z�G d�d� d�e@�Z�G d�d� d�eG�Z�G d�d� d�e��Z�G d�d� d�e��Z�G d�d� d�e��Z�G d�d� d�e��Z�G d�d� d�e��Z�G d�d� d�e��Z�G d�d� d�e��Z�G d�d� d�e�e`�Z�G d�d� d�e j9�Z�G d�d�� d�e:e?�Z�G d�d�� d�e:e?�Z�G d�d�� d�e:�Z�G d�d�� d�e;�Z�G d�d�� d�e j9�Z�e �Ee jd�k�d �G �d�d� �de j9��Z�e �Ee�ed�d�G �d�d� �de j9��Z�G �d�d� �deG�Z��dZ��d Z��d
Z��d �d � Z�e �Ee�� �d
�G �d�d� �de j9��Z�e �Ee�� �d
�G �d�d� �de j9e?��Z�G �d�d� �de@�Z�G �d�d� �de j9�Z�e �Ee�e�d��d�G �d�d� �de j9��Z�e �Ee
j��dk�d�e �Ee �d �G �d �d � �d e:���Z�G �d �d � �d e@�Z�e �Ee�e
�d!��d"�G �d#�d$� �d$e���Z�e �Ee4�d%�G �d&�d'� �d'e j9��Z�e �Ee j���d(��d)�G �d*�d+� �d+e j9��Z�G �d,�d-� �d-e j9�Z�G �d.�d/� �d/e j9�Z��d0�d1� Z�e��d2k�
r�e�� dS (3 � N)�support)�proxyFu Michael Gilfix was hereሴ
�utf-8g N@�� �AIXc C sp t d kr d S ttd�sd S z*tdd��} t �| tjd�}W 5 Q R X W n tk
rZ Y d S X t�d|�d S d S )N� IOCTL_VM_SOCKETS_GET_LOCAL_CIDz
/dev/vsock�rbz �Ir ) �fcntl�hasattr�socket�open�ioctlr �OSError�struct�unpack)�f�r� r �5/opt/alt/python38/lib64/python3.8/test/test_socket.py�get_cid2 s
r c C s@ zt � t jt jt j�} W n ttfk
r2 Y dS X | �� dS �NFT)r �PF_CAN�SOCK_RAW�CAN_RAW�AttributeErrorr �close��sr r r �_have_socket_can? s r c C s@ zt � t jt jt j�} W n ttfk
r2 Y dS X | �� dS r )r r �
SOCK_DGRAM� CAN_ISOTPr r r r r r r �_have_socket_can_isotpI s r" c C s> zt � t jt jd�} W n ttfk
r0 Y dS X | �� dS �Nr FT)r �PF_RDS�SOCK_SEQPACKETr r r r r r r �_have_socket_rdsS s r& c C s> zt � t jt jd�} W n ttfk
r0 Y dS X | �� dS r# )r �AF_ALGr% r r r r r r r �_have_socket_alg] s r( c C s> zt � t jt jd�} W n ttfk
r0 Y dS X | �� dS r# )r �
AF_QIPCRTRr r r r r r r r �_have_socket_qipcrtrg s r* c C s t � d k } | S �N)r )�retr r r �_have_socket_vsockq s
r- c c s. t �� }zt �| � d V W 5 t �|� X d S r+ )r �getdefaulttimeout�setdefaulttimeout)�timeoutZ old_timeoutr r r �socket_setdefaulttimeoutw s
r1 �ic @ s e Zd Zdd� Zdd� ZdS )�
SocketTCPTestc C s. t � t jt j�| _t�| j�| _| j�� d S r+ )r �AF_INET� SOCK_STREAM�servr � bind_port�port�listen��selfr r r �setUp� s zSocketTCPTest.setUpc C s | j �� d | _ d S r+ �r6 r r: r r r �tearDown� s
zSocketTCPTest.tearDownN��__name__�
__module__� __qualname__r< r> r r r r r3 � s r3 c @ s e Zd Zdd� Zdd� ZdS )�
SocketUDPTestc C s$ t � t jt j�| _t�| j�| _d S r+ )r r4 r r6 r r7 r8 r: r r r r< � s zSocketUDPTest.setUpc C s | j �� d | _ d S r+ r= r: r r r r> � s
zSocketUDPTest.tearDownNr? r r r r rC � s rC c s4 e Zd Z� fdd�Z� fdd�Z� fdd�Z� ZS )�ThreadSafeCleanupTestCasec s t � j||� t�� | _d S r+ )�super�__init__� threading�RLock�
_cleanup_lock�r; �args�kwargs�� __class__r r rF � s z"ThreadSafeCleanupTestCase.__init__c
s, | j � t� j||�W 5 Q R � S Q R X d S r+ )rI rE �
addCleanuprJ rM r r rO � s z$ThreadSafeCleanupTestCase.addCleanupc
s, | j � t� j||�W 5 Q R � S Q R X d S r+ )rI rE �
doCleanupsrJ rM r r rP � s z$ThreadSafeCleanupTestCase.doCleanups)r@ rA rB rF rO rP �
__classcell__r r rM r rD � s rD c @ sD e Zd ZdZdZdZe�e�ZdZ e de�e �d 7 Z dd� Z
d S )
�
SocketCANTest�vcan0� z=IB3x8sz@3I4l2I�x� c C sb t � t jt jt j�| _| �| jj� z| j�| jf� W n$ t k
r\ | �
d| j � Y nX d S )N�%network interface `%s` does not exist) r r r r r rO r �bind� interfacer �skipTestr: r r r r< � s �zSocketCANTest.setUpN) r@ rA rB rY �bufsize�
can_frame_fmtr ZcalcsizeZcan_frame_size�bcm_cmd_msg_fmtr< r r r r rR � s
rR c @ s e Zd ZdZdd� ZdS )�
SocketRDSTest� c C sX t � t jt jd�| _| �| jj� zt�| j�| _W n t k
rR | �
d� Y nX d S )Nr zunable to bind RDS socket) r r$ r% r6 rO r r r7 r8 r rZ r: r r r r< � s zSocketRDSTest.setUpN)r@ rA rB r[ r< r r r r r^ � s r^ c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�ThreadableTestc C s$ | j | _| j| _| j| _ | j| _d S r+ )r< �_ThreadableTest__setUpr> �_ThreadableTest__tearDown�_setUp� _tearDownr: r r r rF ! s zThreadableTest.__init__c C s | j �� d S r+ )� server_ready�setr: r r r �serverExplicitReady( s z"ThreadableTest.serverExplicitReadyc C s� t �� | _| j�� t�� | _t�� | _t�� | _t �
d�| _ d| _ | � � }|�
d�}||d d � }t| d| �}t�| j|f�| _z&z | �� W n d| _ � Y nX W 5 | j�� X | j�� d S )N� F�.�_T)r Zwait_threads_exit� wait_threads� __enter__rG �Eventre � client_ready�done�queue�Queue�server_crashed�id�rfind�getattr�thread�start_new_thread� clientRunZ
client_threadrf ra �wait)r; Z
methodnamer2 Z test_methodr r r rc / s,
� zThreadableTest._setUpc C s> | � � | j�� | j�d d d � | j�� r:| j�� }|�d S r+ )rb ro ry rk �__exit__rp Zqsize�get�r; �excr r r rd J s
zThreadableTest._tearDownc
C s� | j �� zRz | �� W n@ tk
rX } z"| j�|� | �� W Y �W �d S d }~X Y nX W 5 | j�� X | j r|| �� d S t
|d�s�t d��z>z
|� W n. tk
r� } z| j�|� W 5 d }~X Y nX W 5 | �� X d S )N�__call__z%test_func must be a callable function) re ry rn rf � clientSetUp�
BaseExceptionrp �put�clientTearDownrr r � TypeError)r; Z test_func�er r r rx S s$
"zThreadableTest.clientRunc C s t d��d S )Nz clientSetUp must be implemented.)�NotImplementedErrorr: r r r r i s zThreadableTest.clientSetUpc C s | j �� t�� d S r+ )ro rf rv �exitr: r r r r� l s
z ThreadableTest.clientTearDownN)
r@ rA rB rF rg rc rd rx r r� r r r r r` � s & r` c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedTCPSocketTest�runTestc C s t j| |d� t�| � d S �N)�
methodName�r3 rF r` �r; r� r r r rF r s z ThreadedTCPSocketTest.__init__c C s t � t jt j�| _d S r+ )r r4 r5 �clir: r r r r v s z!ThreadedTCPSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r+ �r� r r` r� r: r r r r� y s
z$ThreadedTCPSocketTest.clientTearDownN)r� �r@ rA rB rF r r� r r r r r� p s
r� c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedUDPSocketTestr� c C s t j| |d� t�| � d S r� )rC rF r` r� r r r rF � s z ThreadedUDPSocketTest.__init__c C s t � t jt j�| _d S r+ )r r4 r r� r: r r r r � s z!ThreadedUDPSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r+ r� r: r r r r� � s
z$ThreadedUDPSocketTest.clientTearDownN)r� r� r r r r r� ~ s
r� c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedCANSocketTestr� c C s t j| |d� t�| � d S r� )rR rF r` r� r r r rF � s z ThreadedCANSocketTest.__init__c C sD t � t jt jt j�| _z| j�| jf� W n tk
r> Y nX d S r+ )r r r r r� rX rY r r: r r r r � s
z!ThreadedCANSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r+ r� r: r r r r� � s
z$ThreadedCANSocketTest.clientTearDownN)r� r� r r r r r� � s
r� c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�ThreadedRDSSocketTestr� c C s t j| |d� t�| � d S r� )r^ rF r` r� r r r rF � s z ThreadedRDSSocketTest.__init__c C sN t � t jt jd�| _z | j�tdf� | j�� | _W n tk
rH Y nX d S �Nr ) r r$ r% r� rX �HOST� getsockname�cli_addrr r: r r r r � s z!ThreadedRDSSocketTest.clientSetUpc C s | j �� d | _ t�| � d S r+ r� r: r r r r� � s
z$ThreadedRDSSocketTest.clientTearDownN)r� r� r r r r r� � s
r� �
need fcntlz%VSOCK sockets required for this test.� z-This test can only be run on a virtual guest.c @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )� ThreadedVSOCKSocketStreamTestr� c C s t jj| |d� t�| � d S r� ��unittest�TestCaserF r` r� r r r rF � s z&ThreadedVSOCKSocketStreamTest.__init__c C sh t � t jt j�| _| �| jj� | j�t jtf� | j� � | �
� | j� � \| _ | _
| �| j j� d S r+ )r �AF_VSOCKr5 r6 rO r rX �VMADDR_CID_ANY� VSOCKPORTr9 rg �accept�conn�connaddrr: r r r r< � s
z#ThreadedVSOCKSocketStreamTest.setUpc C sD t �d� t�tjtj�| _| �| jj� t� }| j� |t
f� d S )N皙�����?) �time�sleepr r� r5 r� rO r r �connectr� )r; Zcidr r r r � s
z)ThreadedVSOCKSocketStreamTest.clientSetUpc C s | j �d�}| �|t� d S �N� )r� �recv� assertEqual�MSG�r; �msgr r r �
testStream� s z(ThreadedVSOCKSocketStreamTest.testStreamc C s | j �t� | j �� d S r+ �r� �sendr� r r: r r r � _testStream� s z)ThreadedVSOCKSocketStreamTest._testStreamN)r� �r@ rA rB rF r< r r� r� r r r r r� � s
r� c @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )�SocketConnectedTestr� c C s t j| |d� d S r� )r� rF r� r r r rF � s z SocketConnectedTest.__init__c C s* t �| � | �� | j�� \}}|| _d S r+ )r� r< rg r6 r� �cli_conn�r; r� �addrr r r r< � s
zSocketConnectedTest.setUpc C s | j �� d | _ t�| � d S r+ )r� r r� r> r: r r r r> � s
z SocketConnectedTest.tearDownc C s( t �| � | j�t| jf� | j| _d S r+ )r� r r� r� r� r8 � serv_connr: r r r r � s
z SocketConnectedTest.clientSetUpc C s | j �� d | _ t�| � d S r+ )r� r r� r� r: r r r r� � s
z"SocketConnectedTest.clientTearDownN)r� )r@ rA rB rF r< r> r r� r r r r r� � s
r� c @ s> e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� ZdS )�SocketPairTestr� c C s* t jj| |d� t�| � d | _d | _d S r� )r� r� rF r` r� r6 r� r r r rF � s
zSocketPairTest.__init__c C s t �� S r+ )r �
socketpairr: r r r r� s zSocketPairTest.socketpairc C s | � � \| _| _d S r+ )r� r6 r� r: r r r r< s zSocketPairTest.setUpc C s | j r| j �� d | _ d S r+ r= r: r r r r> s
zSocketPairTest.tearDownc C s d S r+ r r: r r r r s zSocketPairTest.clientSetUpc C s$ | j r| j �� d | _ t�| � d S r+ r� r: r r r r� s
z SocketPairTest.clientTearDownN)r� ) r@ rA rB rF r� r< r> r r� r r r r r� � s
r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�SocketTestBasec C s | � � | _| �� d S r+ )� newSocketr6 �
bindServerr: r r r r< ' s
zSocketTestBase.setUpc C s | � | j� | j�� | _d S r+ )�bindSockr6 r� � serv_addrr: r r r r� + s zSocketTestBase.bindServerc C s | j �� d | _ d S r+ r= r: r r r r> 0 s
zSocketTestBase.tearDownN)r@ rA rB r< r� r> r r r r r� s r� c s e Zd Z� fdd�Z� ZS )�SocketListeningTestMixinc s t � �� | j�� d S r+ )rE r< r6 r9 r: rM r r r< 8 s
z SocketListeningTestMixin.setUp�r@ rA rB r< rQ r r rM r r� 5 s r� c s< e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Z� ZS ) �ThreadedSocketTestMixinc s t � j||� t�| � d S r+ )rE rF r` rJ rM r r rF E s z ThreadedSocketTestMixin.__init__c C s | � � | _| �� d S r+ )�newClientSocketr� �
bindClientr: r r r r I s
z#ThreadedSocketTestMixin.clientSetUpc C s | � � S r+ )r� r: r r r r� M s z'ThreadedSocketTestMixin.newClientSocketc C s | � | j� | j�� | _d S r+ )r� r� r� r� r: r r r r� Q s z"ThreadedSocketTestMixin.bindClientc C s | j �� d | _ t�| � d S r+ r� r: r r r r� V s
z&ThreadedSocketTestMixin.clientTearDown) r@ rA rB rF r r� r� r� rQ r r rM r r� = s
r� c s@ e Zd Z� fdd�Z� fdd�Z� fdd�Z� fdd�Z� ZS ) �ConnectedStreamTestMixinc s* t � �� | �� | j�� \}}|| _d S r+ )rE r< rg r6 r� r� r� rM r r r< e s
z ConnectedStreamTestMixin.setUpc s | j �� d | _ t� �� d S r+ )r� r rE r> r: rM r r r> m s
z!ConnectedStreamTestMixin.tearDownc s$ t � �� | j�| j� | j| _d S r+ )rE r r� r� r� r� r: rM r r r r s
z$ConnectedStreamTestMixin.clientSetUpc s8 z| j �� d | _ W n tk
r( Y nX t� �� d S r+ )r� r r rE r� r: rM r r r� w s
z'ConnectedStreamTestMixin.clientTearDown)r@ rA rB r< r> r r� rQ r r rM r r� \ s r� c s$ e Zd Z� fdd�Zdd� Z� ZS )�UnixSocketTestBasec s( t �� | _| �tj| j� t� �� d S r+ )�tempfile�mkdtemp�dir_pathrO �os�rmdirrE r< r: rM r r r< � s
zUnixSocketTestBase.setUpc C s, t j| jd�}t�||� | �tj|� d S )N)�dir)r� Zmktempr� r �bind_unix_socketrO �unlink)r; �sock�pathr r r r� � s zUnixSocketTestBase.bindSock)r@ rA rB r< r� rQ r r rM r r� � s r� c @ s e Zd Zdd� ZdS )�UnixStreamBasec C s t � t jt j�S r+ )r �AF_UNIXr5 r: r r r r� � s zUnixStreamBase.newSocketN�r@ rA rB r� r r r r r� � s r� c s( e Zd ZeZ� fdd�Zdd� Z� ZS )� InetTestBasec s t � �� | jd | _d S �Nrh )rE r< r� r8 r: rM r r r< � s
zInetTestBase.setUpc C s t j|| jd� d S )N��host)r r7 r� �r; r� r r r r� � s zInetTestBase.bindSock)r@ rA rB r� r� r< r� rQ r r rM r r� � s r� c @ s e Zd Zdd� ZdS )� TCPTestBasec C s t � t jt j�S r+ )r r4 r5 r: r r r r� � s zTCPTestBase.newSocketNr� r r r r r� � s r� c @ s e Zd Zdd� ZdS )� UDPTestBasec C s t � t jt j�S r+ )r r4 r r: r r r r� � s zUDPTestBase.newSocketNr� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�SCTPStreamBasec C s t � t jt jt j�S r+ )r r4 r5 � IPPROTO_SCTPr: r r r r� � s �zSCTPStreamBase.newSocketNr� r r r r r� � s r� c @ s e Zd ZejZdS )�
Inet6TestBaseN)r@ rA rB r �HOSTv6r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )� UDP6TestBasec C s t � t jt j�S r+ )r �AF_INET6r r: r r r r� � s zUDP6TestBase.newSocketNr� r r r r r� � s r� c s* dd� � � �fdd�}dd� }| r&|S |S )Nc _ s d S r+ r �rK rL r r r � client_pass� s z%skipWithClientIf.<locals>.client_passc s* t ���| �}t| t�s&� fdd�|_|S )Nc s � S r+ r �r )r� r r �<lambda>� � z3skipWithClientIf.<locals>.skipdec.<locals>.<lambda>)r� �skip�
isinstance�type� client_skip)�objZretval�r� �reasonr r �skipdec� s
z!skipWithClientIf.<locals>.skipdecc S s" t | t�s t| d�s dd� | _| S )Nr� c S s | S r+ r r� r r r r� � r� z5skipWithClientIf.<locals>.noskipdec.<locals>.<lambda>)r� r� r r� �r� r r r � noskipdec� s
z#skipWithClientIf.<locals>.noskipdecr )Z conditionr� r� r� r r� r �skipWithClientIf� s r� c s0 � fdd�|D �}t |dd�dd� |D �� �S )Nc s g | ]}t � |�s|�qS r )r ��.0�namer� r r �
<listcomp>� s
z requireAttrs.<locals>.<listcomp>� don't have �, c s s | ]
}|V qd S r+ r r� r r r � <genexpr>� s z requireAttrs.<locals>.<genexpr>)r� �join)r� Z
attributes�missingr r� r � requireAttrs� s
�r� c
G s� d }dd� | D �}|r0dd� dd� |D �� }nPdd� | D �}ztj|� }W n* tk
rv } z t|�}W 5 d }~X Y n
X |�� t|d k d�d� d d� | D ��|��S )
Nc S s$ g | ] }t |t�rtt|�s|�qS r )r� �strr r �r� r� r r r r� � s
�z!requireSocket.<locals>.<listcomp>r� r� c s s | ]
}|V qd S r+ r r� r r r r� � s z requireSocket.<locals>.<genexpr>c S s$ g | ] }t |t�r tt|�n|�qS r )r� r� ru r r� r r r r� � s �z can't create socket({0}): {1}c s s | ]}t |�V qd S r+ )r� �r� �or r r r� s )r� r r r� r r� �format)rK �errr� Zcallargsr r� r r r �
requireSocket� s& � ��r c @ sH e Zd Zdd� Zdd� Ze�edk d�dd� �Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
e�ejdkd�dd� �Zdd� Zdd� Zdd � Ze�eed �d �e�eed �d �d!d"� ��Ze�eed#�d$�d%d&� �Ze�eed'�d(�d)d*� �Ze�eed+�d,�d-d.� �Ze�eed/�d0�d1d2� �Zd3d4� Z d5d6� Z e j d7d8� �Z d9d:� Z!d;d<� Z"e�eed=�d>�d?d@� �Z#e�eedA�dB�dCdD� �Z$e�eedA�dB�dEdF� �Z%e�eedG�dH�dIdJ� �Z&e�eedG�dH�dKdL� �Z'dMdN� Z(dOdP� Z)dQdR� Z*dSdT� Z+dUdV� Z,dWdX� Z-dYdZ� Z.e�ejdkd�d[d\� �Z/e�ejdkd�e�eed]�d^�d_d`� ��Z0dadb� Z1dcdd� Z2e�e �3de�df�dgdh� �Z4didj� Z5dkdl� Z6dmdn� Z7dodp� Z8dqdr� Z9dsdt� Z:dudv� Z;dwdx� Z<dydz� Z=d{d|� Z>d}d~� Z?e j dd�� �Z@e�e jAd��d�d�� �ZBe�e jAd��d�d�� �ZCe�e jAd��e�DejEd�kd��e�DeFd��d�d�� ���ZGe�e jAd��e�ejEd�kd��d�d�� ��ZHe�e jAd��e�DejEd�kd��e�DeFd��d�d�� ���ZIe�e jAd��e�ejEd�kd��d�d�� ��ZJd�d�� ZKd�d�� ZLd�d�� ZMe�eed��d��d�d�� �ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSe�Dejdkd��d�d�� �ZTd�d�� ZUd�d�� ZVdS )��GeneralModuleTestsc C s< dd l }| �tj|jk� t�� }| �|tj� |�� d S r� )�_socket�
assertTruer Z
SocketType�assertIsInstancer )r; r r r r r � test_SocketType_is_socketobject s
z2GeneralModuleTests.test_SocketType_is_socketobjectc C s� t � t jt j�}|�� | �d|�� t|�� | �dt j t|�� | �dt j t|�� | �dt|�� | �dt|�� |�d� | �dt|�� | �t|� � �t|�� W 5 Q R X | �dt|�� | �dt|�� d S ) Nzfd=%iz family=%sztype=%szproto=0Zraddr�� 127.0.0.1r Zladdrz[closed])
r r4 r5 �assertIn�fileno�repr� assertNotInrX r� r� �r; r r r r � test_repr s
"z GeneralModuleTests.test_reprNzneed _socket modulec C sr t �t jt j�}z.d|�� |j|j|jf }| � t
|�|� W 5 |�� X d|j|j|jf }| � t
|�|� d S )Nz4<socket object, fd=%s, family=%s, type=%s, proto=%s>z4<socket object, fd=-1, family=%s, type=%s, proto=%s>) r r r4 r5 r r �familyr� �protor� r
)r; r �expectedr r r �test_csocket_repr s �
�z$GeneralModuleTests.test_csocket_reprc C sl t � t jt j��"}t|�}| �|�� |�� � W 5 Q R X d }z |�� W n tk
r\ Y n X | �d� d S )NzSocket proxy still exists)r r4 r5 r r� r �ReferenceError�fail)r; r �pr r r � test_weakref+ s z GeneralModuleTests.test_weakrefc C sr d}| j t|d d��
t�W 5 Q R X | j t|d d�� tj�W 5 Q R X | j t|d d�� tj�W 5 Q R X d S )Nz$Error raising socket exception (%s).r �r� z
socket.herrorzsocket.gaierror)� assertRaisesr r �herror�gaierrorr� r r r �testSocketError7 s z"GeneralModuleTests.testSocketErrorc C sF t � t jt j�}| �|j� |�d� |�� }| �t��}|� d|� W 5 Q R X | �
t |j �d� | �t��}|� d|� W 5 Q R X | �
t |j �d� | �t��}|� dd � W 5 Q R X | �
dt |j �� | �t��}|� dd|� W 5 Q R X | �
t |j �d� | �t��}|� dd|� W 5 Q R X | �
t |j �d� | �t��}|� ddd � W 5 Q R X | �
dt |j �� | �t��}|� dd |� W 5 Q R X | �
d
t |j �� | �t��}|� dd d � W 5 Q R X | �
d
t |j �� | �t��}|� d� W 5 Q R X | �
d t |j �� | �t��}|� dd|d � W 5 Q R X | �
d
t |j �� d S )N)� r u ☠z*a bytes-like object is required, not 'str'y @z.a bytes-like object is required, not 'complex'� fooz not NoneTyper �barzan integer is requiredz (1 given)� z (4 given))r r4 r rO r rX r� r r� �sendtor� r� � exceptionr )r; r �sockname�cmr r r �testSendtoErrorsA sT
� � � � z#GeneralModuleTests.testSendtoErrorsc C s@ t j t jrt j t j t j t j t j t j t j t j
d S r+ ) r r4 �has_ipv6r� r5 r r �SOCK_RDMr% �
SOL_SOCKET� SO_REUSEADDRr: r r r �testCrucialConstantsm s z'GeneralModuleTests.testCrucialConstantsc C s t j t j t jrt j d S r+ )r Z IPPROTO_TCP� IPPROTO_UDPr$ � IPPROTO_IPV6r: r r r �testCrucialIpProtoConstantsz s z.GeneralModuleTests.testCrucialIpProtoConstants�nt�Windows specificc C s4 t j t j t j t j t j t j t j t j d S r+ ) r ZIPPROTO_ICLFXBMZ
IPPROTO_STZ IPPROTO_CBTZ IPPROTO_IGPZ IPPROTO_RDPZ IPPROTO_PGMZ IPPROTO_L2TPr� r: r r r � testWindowsSpecificConstants� s z/GeneralModuleTests.testWindowsSpecificConstantsc C s� t �� }zt �|�}W n tk
r4 | �d� Y nX | �|�d�dkd� zt �|�\}}}W n tk
r~ | �d� Y nX ||g| }t �|�}||kr�| � d|t
|�f � d S )N�name lookup failureri r zError resolving host to ip.z=Error testing host resolution mechanisms. (fqdn: %s, all: %s)) r � gethostname�
gethostbynamer rZ r �find�
gethostbyaddrZgetfqdnr r
)r; ZhostnameZipZhname�aliasesZipaddrsZall_host_namesZfqhnr r r �testHostnameRes� s
z"GeneralModuleTests.testHostnameResc C sL t jddfD ]}| �t�|�|� q t jfD ]}| �|t�|�d � q,d S )Nz10.0.0.1�255.255.255.255r� )r ZHOSTv4r� r r1 r r3 )r; r� r� r r r �test_host_resolution� s z'GeneralModuleTests.test_host_resolutionc
C sZ d}dD ]L}| j t|d�� t�|� W 5 Q R X | j t|d�� t�|� W 5 Q R X qd S )Nz[resolving an invalid IP address did not raise OSError; can be caused by a broken DNS server)z0.1.1.~1z1+.1.1.1z::1qz::1::2z1:1:1:1:1:1:1:1:1r )r r r r1 r3 )r; Z explanationr� r r r � test_host_resolution_bad_address� s �z3GeneralModuleTests.test_host_resolution_bad_address� sethostnamez test needs socket.sethostname()r0 z test needs socket.gethostname()c
C s� t �� }zt �d� W n< tk
rR } z |jtjkr@| �d� n� W 5 d }~X Y nX z.| �t �� d� t �d� | �t �� d� W 5 t �|� X d S )N�newztest should be run as roots barr )r r0 r9 r �errnoZEPERMrZ r� )r; Zoldhnr� r r r �test_sethostname� s
z#GeneralModuleTests.test_sethostname� if_nameindexz$socket.if_nameindex() not available.c C s� t �� }|D ]p\}}| �|t� | �|t� | �|d� t �|�}| �|t� | �||� t �|�}| �|t� | �||� q d S r� ) r r= r �intr� �
assertGreater�if_nametoindexr� �if_indextoname)r; Z
interfaces�indexr� Z_index�_namer r r �testInterfaceNameIndex� s
z)GeneralModuleTests.testInterfaceNameIndexrA z&socket.if_indextoname() not available.c
C s� | � ttjd� | � ttjd� | � ttjdd � | � ttjd� ttd�r�tt�� �}|D ]8}|d }||kr^| � ttf�� t�|� W 5 Q R X q^dD ]0}||kr�| � ttf�� t�|� W 5 Q R X q�d S ) Nr ���r� �� � _DEADBEEFr= � )� �� l ���� ) r r r rA �
OverflowErrorr� r �dictr= )r; �indicesrB Zindex2r r r � testInvalidInterfaceIndexToName� s
z2GeneralModuleTests.testInvalidInterfaceIndexToNamer@ z&socket.if_nametoindex() not available.c C s$ | � ttjd� | � ttjd� d S )Nr rG )r r� r r@ r r: r r r � testInvalidInterfaceNameToIndex� s z2GeneralModuleTests.testInvalidInterfaceNameToIndex� getrefcountz test needs sys.getrefcount()c C sL zt �t�}t�td� W n, tk
rF t �t�|krB| �d� Y nX d S )Nr z$socket.getnameinfo loses a reference)�sysrO r@ r � getnameinfor� r )r; �origr r r �testRefCountGetNameInfo� s
z*GeneralModuleTests.testRefCountGetNameInfoc C s* zt �dd� W n tk
r$ Y nX d S )N)rU r r r r )r rQ r r: r r r �testInterpreterCrash� s z'GeneralModuleTests.testInterpreterCrashc C s� t jdt jdt jdt jdi}|�� D ]d\}}d|> d }dD ]$}| �||@ ||||@ ��|@ � q<||�}| �||@ |� | �t|d� q$d S )N� � rh )r rh � i ��r� igE#i2Tvl ) r �htonl�ntohl�htons�ntohs�itemsr� r rJ )r; Zsizes�func�size�maskr2 Zswappedr r r �testNtoH s � "zGeneralModuleTests.testNtoHc C s dd l }ddddg}|dg }ddddd > g}||jd |jd g }d
|jg}|D ]}t�|� t�|� qT|D ]}t�|� t�|� qr|D ]$}| �t tj|� | �t tj|� q�|D ]$}| �t tj|� | �t tj|� q�|D ]$}| �
t tj|� | �
t tj|� q�d S ) Nr rh r� rW rI rE ���rH rF � ) � _testcapiZINT_MIN�INT_MAXr r[ rZ rY rX r rJ � assertWarns�DeprecationWarning)r; rc Z
s_good_valuesZ
l_good_valuesZ l_bad_valuesZ s_bad_valuesZs_deprecated_values�kr r r �testNtoHErrors s.
�
z!GeneralModuleTests.testNtoHErrorsc C s& | j }tj�d�s tjdkr"d}nd}|D ]0}zt�|d�}W q`W q* tk
rX Y q*X q*t�ttd�rvt�� dkr�t�|�}|||� zt�|d�}W n tk
r� d }Y n X |||� t j
s�|t� |�|� |t� |d�|� |d k �r|t� |d�|� | � t
tj d � | � t
tj d
� d S ) N)ZfreebsdZnetbsdZ gnukfreebsd)�linux�darwin)�daytimeZqotd�domain)Zechork rl Ztcp�getandroidapilevel� ZudprE rb )r� rP �platform�
startswithr Z
getservbynamer r rm r Z
is_androidZ
getservbyportr rJ )r; �eqZservicesZservicer8 Zport2Zudpportr r r �
testGetServBy. s< �
�
z GeneralModuleTests.testGetServByc
C s� | � t�� d � t�� �}| � |�� d � W 5 Q R X td��x | � t�� d� t�� �}| � |�� d� W 5 Q R X t�d � | � t�� d � t�� �}| � |�� d � W 5 Q R X W 5 Q R X | �ttjd� | �ttjd� d S )N�
rE Zspam) r� r r. �
gettimeoutr1 r/ r �
ValueErrorr� )r; r r� r r r �testDefaultTimeoutZ s
$z%GeneralModuleTests.testDefaultTimeout� inet_atonz test needs socket.inet_aton()c C s( | � dt�d�� | � dt�d�� d S )N� �0.0.0.0� ����r6 )r� r rw r: r r r � testIPv4_inet_aton_fourbytess s z/GeneralModuleTests.testIPv4_inet_aton_fourbytes� inet_ptonz test needs socket.inet_pton()c s* ddl m}m�m� � �fdd�}�fdd�}��d|d�� ��d|d �� ��d
|d �� ��d |d
�� ��d|d�� ts�||d� ||d� ||d� ||d� ||d� ��d|d�� ��d|d �� ��d
|d �� ��d|d�� ||d� ||d� ||d� ||d� ||d� d S )Nr )rw r| r4 c s
�� | �S r+ r ��a)r4 r| r r r� r� z5GeneralModuleTests.testIPv4toString.<locals>.<lambda>c s � � ttf| |�S r+ �r r ru �r] r~ r: r r r� � s �rx ry s � � z 255.0.255.0s ����z170.170.170.170� �1.2.3.4rz r6 z0.0.0.z 300.0.0.0za.0.0.0z 1.2.3.4.5�::1)r rw r| r4 r� r �r; r �g�
assertInvalidr )r4 r| r; r �testIPv4toString{ s.
z#GeneralModuleTests.testIPv4toStringc
s� z&ddl m�m� m} |s$��d� W n tk
rD ��d� Y nX tjdkr�z�� d� W n6 tk
r� } z|j dkr���d� W 5 d }~X Y nX � �fd d
����fd d
�}��
d �d�� ��
d �d
�� ��
d�d�� ��
d�d�� ��
d�d�� ��
d�d�� |d� |d� |d� |d� |d� |d� |d � |d � t �s`|d � ��
d �d �� ��
d!�d"�� ��
d#�d$�� |d%� |d&� |d'� |d(� |d)� |d*� d S )+Nr )r| r� r$ �IPv6 not available�+could not import needed symbols from socket�win32�::�&' �IPv6 might not be supportedc s
�� | �S r+ r r} )r� r| r r r� � r� z5GeneralModuleTests.testIPv6toString.<locals>.<lambda>c s �� ttf� | �S r+ r r} �r r; r r r� � s �� z0::0s z1::s E�v� V�� �$��z$45ef:76cb:1a:56ef:afeb:bac:1924:aeaes �B
� ' T zad42:abc::127:0:254:2s
z12:a::z0x20::z:::z::0::z1::abc::z 1::abc::defz 1:2:3:4:5:6z 1:2:3:4:5:6:z1:2:3:4:5:6:7:8:0z1:2:3:4:5:6:7:8:s �*@z::254.42.23.64s B ���*@z42::a29b:254.42.23.64s B�� �����*@z"42:a8b9:0:2:ffff:a29b:254.42.23.64z255.254.253.252z 1::260.2.3.0z 1::0.be.e.0z1:2:3:4:5:6:7:1.2.3.4z ::1.2.3.4:0z0.100.200.0:3:4:5:6:7:8) r r| r� r$ rZ � ImportErrorrP ro r �winerrorr� r �r; r$ r� r� r )r� r r| r; r �testIPv6toString� sj
�����z#GeneralModuleTests.testIPv6toString� inet_ntopz test needs socket.inet_ntop()c s ddl m}m�m� � �fdd�}�fdd�}��d|d�� ��d|d �� ��d
|d �� ��d |d
�� ||d� ||d� ||d� ��d|td ��� ��d|d�� ��d|d �� ��d
|d �� ||d� ||d� ||d� ��d|td ��� d S )Nr )� inet_ntoar� r4 c s
�� | �S r+ r r} )r4 r� r r r� � r� z5GeneralModuleTests.testStringToIPv4.<locals>.<lambda>c s � � ttf| |�S r+ r r� r: r r r� � s �z1.0.1.0s z
170.85.170.85s �U�Ur6 rz r� r� s s r� )r r� r� r4 r� � bytearrayr� r )r4 r� r; r �testStringToIPv4� s$
z#GeneralModuleTests.testStringToIPv4c
s z&ddl m�m� m} |s$��d� W n tk
rD ��d� Y nX tjdkr�z�� d� W n6 tk
r� } z|j dkr���d� W 5 d }~X Y nX � �fd d
����fd d
�}��
d �d�� ��
d
�d�� ��
d�d�� ��
d
�t d��� |d� |d� |d� d S )Nr )r� r� r$ r� r� r� r� r� r� c s
�� | �S r+ r r} )r� r� r r r� r� z5GeneralModuleTests.testStringToIPv6.<locals>.<lambda>c s �� ttf� | �S r+ r r} r� r r r� s �r� r� s z!aef:b01:506:1001:ffff:9997:55:170s
� ���� Ups s s ) r r� r� r$ rZ r� rP ro r r� r� r� r� r )r� r r� r; r �testStringToIPv6� s0
�z#GeneralModuleTests.testStringToIPv6c C s� t �� }t�tjtj�}| �|j� |�d|f� |�� }zt� t�
� �}W n t k
rj | � d� Y nX | �
|d d|fd|d � | �|d |� d S )Nry r/ r z
%s invalidrh )r �find_unused_portr r4 r5 rO r rX r� r1 r0 r rZ r r� )r; r8 r� r� Z
my_ip_addrr r r � testSockName s z GeneralModuleTests.testSockNamec C s@ t � t jt j�}| �|j� |�t jt j�}| �|dkd� d S )Nr zinitial mode is reuse) r r4 r5 rO r �
getsockoptr& r' � assertFalse�r; r� Zreuser r r �testGetSockOpt, s z!GeneralModuleTests.testGetSockOptc C sR t � t jt j�}| �|j� |�t jt jd� |�t jt j�}| � |dkd� d S )Nrh r zfailed to set reuse mode)
r r4 r5 rO r �
setsockoptr& r' r� r� r� r r r �testSetSockOpt4 s
z!GeneralModuleTests.testSetSockOptc C s: t � t jt j��}|�d� W 5 Q R X | �t|jd� d S )Nrh � spam)r r4 r5 �
settimeoutr r r� r� r r r �testSendAfterClose< s z%GeneralModuleTests.testSendAfterClosec
C s� t � � }|�t jdf� t j |�� d��� z |�� W n8 tk
rp } z| �|jtjtj f� W 5 d }~X Y n X | �
d� d S )Nr �r z#close() should raise EBADF/ENOTSOCK) r rX Z
_LOCALHOSTr r r r r; �EBADF�ENOTSOCKr )r; r� r� r r r �testCloseExceptionB s (z%GeneralModuleTests.testCloseExceptionc C sv t � t jt j��\}| �|jt j� tt d�rJ| �|jt jt jB t jf� n| �|jt j� | �|j d� W 5 Q R X d S )N� SOCK_CLOEXECr )
r r4 r5 r� r r r r� r� r r� r r r �testNewAttributesN s
��z$GeneralModuleTests.testNewAttributesc C s� t � � }| �|j� t�� }|d }|d }| �t|jt|f� | �t|jt|f� t �
� D ]\}t�� }z|�t|f� W n8 t k
r� } z|j t j
ks�|dkr�� W 5 d }~X Y q\X q�q\d S )Nrb � )r rO r r r� r rJ rX r� � itertools�countr r; Z
EADDRINUSE)r; r� r8 Zbig_portZneg_portr2 r� r r r �test_getsockaddrarg[ s z&GeneralModuleTests.test_getsockaddrargc C s� | � ttjd�� | � ttd�� | � ttd�� | � ttd�� | � ttd�� t�� }| �|j� | �t|jdd � |�tjd� d S )Nr Z
SIO_RCVALLZ RCVALL_ONZ
RCVALL_OFF�SIO_KEEPALIVE_VALSrE )rh �d r� ) r r r rO r r ru r r� r r r r �test_sock_ioctlo s z"GeneralModuleTests.test_sock_ioctl�SIO_LOOPBACK_FAST_PATHz1Loopback fast path support required for this testc
C s| t � � }| �|j� z|�t jd� W n< tk
rb } z d}|j|krP| �d� � W 5 d }~X Y nX | �t |jt jd � d S )NTi=' zQSIO_LOOPBACK_FAST_PATH is defined but doesn't implemented in this Windows version)
r rO r r r� r r� rZ r r� )r; r r} Z
WSAEOPNOTSUPPr r r �test_sio_loopback_fast_path{ s
z.GeneralModuleTests.test_sio_loopback_fast_pathc
C s@ zt �dd� W n< t jk
rL } z |jt jkr:| �d� � W 5 d }~X Y nX t �td �D ]}| �t|�d� qZt �dd� t �dd� t �d d� t j
r�t �dd� t t d�r�t �
� dkr�t �td � t �td� t �td � t �tdt jt j�}|D ]J\}}}}}| �|t j� | �t|�d
� | �|t j� | �t|�d � q�t �td d t j�}|D ] \}}}}}| �|t j� �qZt �td d d t j� t �td d d d t j� t �d d t jt jd t j� t �td �}t jtd d
�} | �|| � t �td t j�}t jtd t jd�} | �|| � t �td d t j�}t jtd t jd�} | �|| � t �td d d t j�}t jtd t jd�} | �|| � t �td d d d t j�}t jtd t jd�} | �|| � t �d d t jt jd t j�}t jd d t jt jd t jd�} | �|| � | �tt jdd� t t d��r<zt �dd d d d t j� W n t jk
�r: Y nX d S )N� localhost�P zbuggy libc versionr� r r� rm rn Zhttp�AddressFamily.AF_INET�SocketKind.SOCK_STREAMr )r� r8 �r �r� )r )�flags)r� r8 r r� r r� u ��AI_NUMERICSERV)r � getaddrinfor r; Z EAI_SERVICErZ r� r� �lenr � IPV6_ENABLEDr rP rm r4 r5 r� ZSOL_TCPZ
AI_PASSIVE� AF_UNSPECr �UnicodeEncodeErrorr� )
r; r� �infoZinfosr r� rj Zsocktyper~ �br r r �testGetaddrinfo� s�
� � � � �z"GeneralModuleTests.testGetaddrinfoc C s | � ttjdd� d S )N)zmail.python.orgr r )r r r rQ r: r r r �test_getnameinfo� s z#GeneralModuleTests.test_getnameinfoZnetworkznetwork is not enabledc C sP t �d�� t�d� W 5 Q R X d}t�|� t�|� t�|dtjtj� d S )Nz
python.orgu! испытание.pythontest.netr )r Ztransient_internetr r1 Zgethostbyname_exr� r� r5 )r; rl r r r � test_idna� s
z GeneralModuleTests.test_idnac
s� t td�rt td�s � �d� � fdd�}� fdd�}t�� \}}t�tj|�}zv|r`|�d � � � t
�� t�d
� |� d t j
� W 5 Q R X |r�t�tj|� t�d
� � � tj|j d t j
� W 5 t�d� t�tj|� |�� |�� X d S ) N�alarmr� z9signal.alarm and socket.socketpair required for this testc s � � ttjd� d S r� �r ru �mathZacosh�rK r: r r �
ok_handler� s z@GeneralModuleTests.check_sendall_interrupted.<locals>.ok_handlerc s � � ttjd� dd d S �Nr rh r� r� r: r r �raising_handler� s zEGeneralModuleTests.check_sendall_interrupted.<locals>.raising_handlerr g �?rh � x)r �signalr rZ r� �SIGALRMr� r r� r �ZeroDivisionError�sendallr �
SOCK_MAX_SIZEr0 )r; Z with_timeoutr� r� �cr � old_alarmr r: r �check_sendall_interrupted� s,
�
z,GeneralModuleTests.check_sendall_interruptedc C s | � d� d S �NF�r� r: r r r �test_sendall_interrupted
s z+GeneralModuleTests.test_sendall_interruptedc C s | � d� d S )NTr� r: r r r �%test_sendall_interrupted_with_timeout s z8GeneralModuleTests.test_sendall_interrupted_with_timeoutc C s� t � t jt j�}t|�}| �t��}d }t�� W 5 Q R X | �|t |j
j d �� t � t jt j�}|� d�}t|�}d }t�� | �t�� d }t�� W 5 Q R X d S )Nr r )
r r4 r5 r
re �ResourceWarningr Z
gc_collectr r� ZwarningrK �makefile)r; r� r r" r r r r �test_dealloc_warn s
z$GeneralModuleTests.test_dealloc_warnc C sB t � t jt j��(}|�d�}|�� | �t|�d� W 5 Q R X d S )Nr z <_io.BufferedReader name=-1>)r r4 r5 r� r r� r
�r; r� �fpr r r �test_name_closed_socketio$ s
z,GeneralModuleTests.test_name_closed_socketioc C s� t � � �p}|jddd�}| �|�� � | �|�� � | �|�� � |�� | �t |j� | �t |j� | �t |j� W 5 Q R X d S )Nr r )� buffering)
r r� r �readabler� �writable�seekabler r ru r� r r r � test_unusable_closed_socketio* s
z0GeneralModuleTests.test_unusable_closed_socketioc C s� t � � }z@|�tdf� t �|�� � | �t�� |�d� W 5 Q R X W 5 | �t�� |�� W 5 Q R X X | �t�� t �d � W 5 Q R X | �t�� t �d� W 5 Q R X d S )Nr rh rE ) r r r r rX r� r r9 r� r� r r r �test_socket_close5 s z$GeneralModuleTests.test_socket_closec C s^ dD ]T}| j |d��> t�� �*}|�|��}| �|j|� W 5 Q R X W 5 Q R X W 5 Q R X qd S )N)r r Zrw�w�wb��mode)�subTestr r� r� r� )r; r� r� r� r r r �test_makefile_modeE s
z%GeneralModuleTests.test_makefile_modec C s\ dD ]R}| j |d��< t�� �(}| �td�� |�|� W 5 Q R X W 5 Q R X W 5 Q R X qd S )N)�rtrU �+r~ r� z invalid mode)r� r �assertRaisesRegexru r� )r; r� r� r r r �test_makefile_invalid_modeL s
z-GeneralModuleTests.test_makefile_invalid_modec C s� t � � }|�, ttjd �D ]}| �ttj||� q W 5 Q R X ttjd �D ]H}t�t�t j|��}| � |t j� t�t�t j
|��}| � |t j
� qLd S r� ) r �range�pickleZHIGHEST_PROTOCOLr r� �dumps�loadsr4 r� r5 )r; r� Zprotocolr r� r r r � test_pickleS s z GeneralModuleTests.test_picklec
C st dD ]8}t � t jt j�� }|�tdf� |�|� W 5 Q R X qt � t jt j�� }|�tdf� |�� W 5 Q R X d S )N)r rE r )r r4 r5 rX r� r9 )r; Zbacklog�srvr r r �test_listen_backlog^ s z&GeneralModuleTests.test_listen_backlogc C sL dd l }t�tjtj��*}|�tdf� | �t|j|j d � W 5 Q R X d S r� )
rc r r4 r5 rX r� r rJ r9 rd )r; rc r� r r r � test_listen_backlog_overflowh s z/GeneralModuleTests.test_listen_backlog_overflow� IPv6 required for this test.c C sR | � ttjtjddfd� t�tjtj�� }| � t|jtjddf� W 5 Q R X d S )Nr rI i����) r rJ r rQ r r� r� r5 rX r r r r �
test_flowinfop s
�z GeneralModuleTests.test_flowinfoc C s0 t �ddt jt jt j�\�^ }}| �|d� d S )N�ff02::1de:c0:face:8Dr )�ff02::1de:c0:face:8dr r r )r r� r� r r) r� )r; rj �sockaddrr r r �test_getaddrinfo_ipv6_basicw s � z.GeneralModuleTests.test_getaddrinfo_ipv6_basicr� zdoes not work on Windowsz Symbolic scope id does not workc C sL t �� d \}}t �d| dt jt jt j�\�^ }}| �|ddd|f� d S )Nr �ff02::1de:c0:face:8D%r r� )r r= r� r� r r) r� )r; �ifindex�test_interfacerj r� r r r �&test_getaddrinfo_ipv6_scopeid_symbolic� s � z9GeneralModuleTests.test_getaddrinfo_ipv6_scopeid_symbolicz.Numeric scope id does not work or undocumentedc C sD d}t �dt|� dt jt jt j�\�^ }}| �|ddd|f� d S )N�* r� r r� r )r r� r� r� r r) r� )r; r� rj r� r r r �%test_getaddrinfo_ipv6_scopeid_numeric� s
� z8GeneralModuleTests.test_getaddrinfo_ipv6_scopeid_numericc C sH t �� d \}}ddd|f}t �|t jt jB �}| �|d| df� d S )Nr r� r �ff02::1de:c0:face:8d%�1234)r r= rQ �NI_NUMERICHOST�NI_NUMERICSERVr� )r; r� r� r� �nameinfor r r �&test_getnameinfo_ipv6_scopeid_symbolic� s z9GeneralModuleTests.test_getnameinfo_ipv6_scopeid_symbolicc C s@ d}ddd|f}t �|t jt jB �}| �|dt|� df� d S )Nr r� r r r r )r rQ r r r� r� )r; r� r� r r r r �%test_getnameinfo_ipv6_scopeid_numeric� s z8GeneralModuleTests.test_getnameinfo_ipv6_scopeid_numericc C sD t � t jt j��*}| �t|j�d� | �t|j�d� W 5 Q R X d S )Nr� r� )r r4 r5 r� r� r r� r r r r �test_str_for_enums� s z%GeneralModuleTests.test_str_for_enumsc C s� t tdd�}t tdd�}tj|B |B }t�tj|��~}| �|jtj� |�d� | �|jtj� |�d� | �|jtj� |�d� | �|jtj� |�d� | �|jtj� W 5 Q R X d S )N�
SOCK_NONBLOCKr r� rh TF)ru r r5 r4 r� r� r� � setblocking)r; r r� Z sock_typer r r r � test_socket_consistent_sock_type� s
z3GeneralModuleTests.test_socket_consistent_sock_typec C s� t � t jt j�}|�� }tt jj�� �d }tdd� t jj� � D ��d }t j ||d|d��4}| �
|j |� | �
|j |� | �
|jddh� W 5 Q R X d S )Nrh c s s | ]\}}|d kr|V qdS )> r r� Nr )r� r� Zkindr r r r� � s �zEGeneralModuleTests.test_unknown_socket_family_repr.<locals>.<genexpr>rn )r r� r r r )r r4 r5 �detach�maxZ
AddressFamily� __members__�valuesZ
SocketKindr\ r� r r� r r )r; r� �fdZunknown_familyZ unknown_typer r r r � test_unknown_socket_family_repr� s$
�� �z2GeneralModuleTests.test_unknown_socket_family_repr�sendfileztest needs os.sendfile()c
C s� G dd� d�}t � � ��}t�tjtj�}t�|� | �t j�� |�||�� W 5 Q R X | �t �� |�|dd �� W 5 Q R X | �t
�� |�|d �� W 5 Q R X W 5 Q R X d S )Nc @ s e Zd Zdd� Zdd� ZdS )z<GeneralModuleTests.test__sendfile_use_sendfile.<locals>.Filec S s
|| _ d S r+ �r )r; r r r r rF � s zEGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.__init__c S s | j S r+ r r: r r r r � s zCGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.filenoN)r@ rA rB rF r r r r r �File� s r r� rF ) r r� r
�curdir�O_RDONLYr r �_GiveupOnSendfile�_sendfile_use_sendfilerJ r� )r; 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 �Nr� )r� r r� r r rO r r )r; r r �styper �s2r r r �_test_socket_fileno� s z&GeneralModuleTests._test_socket_filenoc C sR t � t jt j�}| �|j� |�tjdf� | �|t jt j� t t d�r�t � t jt j
�}| �|j� |�tjdf� | �|t jt j
� tj r�t � t j t j�}| �|j� |�tj
dddf� | �|t j t j� t t d��rNt�� }| �tj|� t � t jt j�}| �|j� z|�tj�|d�� W n tk
�r: Y nX | �|t jt j� d S )Nr r r� r )r r4 r5 rO r rX r r� r r r r� r� r� r� r� �shutilZrmtreer� r� r� r� �PermissionError)r; r Ztmpdirr r r �test_socket_fileno s4
�z%GeneralModuleTests.test_socket_filenoc C s0 | � td�� tjtjtjdd� W 5 Q R X d S )Nzinteger argument expectedg @E@r� �r� r� r r4 r5 r: r r r � test_socket_fileno_rejects_float$ s z3GeneralModuleTests.test_socket_fileno_rejects_floatc C s0 | � td�� tjtjtjdd� W 5 Q R X d S )Nzinteger is required�foor� r# r: r r r �&test_socket_fileno_rejects_other_types( s z9GeneralModuleTests.test_socket_fileno_rejects_other_typesc C s0 | � td�� tjtjtjdd� W 5 Q R X d S )N�negative file descriptorrE r� �r� ru r r4 r5 r: r r r �)test_socket_fileno_rejects_invalid_socket, s z<GeneralModuleTests.test_socket_fileno_rejects_invalid_socketzWindows disallows -1 onlyc C s0 | � td�� tjtjtjdd� W 5 Q R X d S )Nr' i���r� r( r: r r r �#test_socket_fileno_rejects_negative0 s z6GeneralModuleTests.test_socket_fileno_rejects_negativec C s� d}| � t��}tjt�� d� W 5 Q R X | �|jjtj|f� | � t�� }tjtj tj
t�� d� W 5 Q R X | �|jjtj|f� d S )Ni6' r� ) r r r r Z make_bad_fdr r r; r� r4 r5 )r; Z WSAENOTSOCKr" r r r �$test_socket_fileno_requires_valid_fd5 s �z7GeneralModuleTests.test_socket_fileno_requires_valid_fdc
C s~ t �� �l}| �t�� tj|�� d� W 5 Q R X | �t�� }tjtjtj|�� d� W 5 Q R X | �|j j
t
j � W 5 Q R X d S r ) r� ZNamedTemporaryFiler r r r r4 r5 r� r r; r� )r; Zafiler" r r r �%test_socket_fileno_requires_socket_fdB s
�z8GeneralModuleTests.test_socket_fileno_requires_socket_fd)Wr@ rA rB r r
r� �
skipUnlessr r r r r# r( r+ r� r� r. r5 r7 r8 r r r< rD rM rN rP rS rT r` r � cpython_onlyrh rr rv r{ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� Zis_resource_enabledr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �skipIfro r r r r r r
r
r r r r" r$ r& r) r* r+ r, r r r r r s
,
�
�
�
�
, �
�
�
> �
�
"
�
N �
�
�
r z!SocketCan required for this test.c @ s� e Zd Zdd� Ze�eed�d�dd� �Zdd� Z e�eed�d�d d
� �Z
d d � Z d
d� Z e�eed�d�dd� �Z
e�eed�d�dd� �ZdS )� BasicCANTestc C s t j t j t j d S r+ )r �AF_CANr r r: r r r r( R s z!BasicCANTest.testCrucialConstants�CAN_BCM�&socket.CAN_BCM required for this test.c C s� t j t j t j t j t j t j t j t j t j t j
t j t j t j
t j t j t j t j t j t j t j t j t j t j t j d S r+ )r r2 ZCAN_BCM_TX_SETUPZCAN_BCM_TX_DELETEZCAN_BCM_TX_READ�CAN_BCM_TX_SENDZCAN_BCM_RX_SETUPZCAN_BCM_RX_DELETEZCAN_BCM_RX_READZCAN_BCM_TX_STATUSZCAN_BCM_TX_EXPIREDZCAN_BCM_RX_STATUSZCAN_BCM_RX_TIMEOUTZCAN_BCM_RX_CHANGEDZCAN_BCM_SETTIMERZCAN_BCM_STARTTIMERZCAN_BCM_TX_COUNTEVTZCAN_BCM_TX_ANNOUNCEZCAN_BCM_TX_CP_CAN_IDZCAN_BCM_RX_FILTER_IDZCAN_BCM_RX_CHECK_DLCZCAN_BCM_RX_NO_AUTOTIMERZCAN_BCM_RX_ANNOUNCE_RESUMEZCAN_BCM_TX_RESET_MULTI_IDXZCAN_BCM_RX_RTR_FRAMEr: r r r �testBCMConstantsW s0 z BasicCANTest.testBCMConstantsc C s$ t � t jt jt j��}W 5 Q R X d S r+ �r r r r r r r r �testCreateSocketw s z BasicCANTest.testCreateSocketc C s$ t � t jt jt j��}W 5 Q R X d S r+ )r r r r2 r r r r �testCreateBCMSocket{ s z BasicCANTest.testCreateBCMSocketc C sB t � t jt jt j��$}d}|�|� | �|�� |� W 5 Q R X d S )N)r )r r r r rX r� r� )r; r �addressr r r � testBindAny� s
zBasicCANTest.testBindAnyc C s6 t � t jt jt j��}| �td|jd� W 5 Q R X d S )N�interface name too long)� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx)r r r r r� r rX r r r r �testTooLongInterfaceName� s
�z%BasicCANTest.testTooLongInterfaceName�CAN_RAW_LOOPBACKz/socket.CAN_RAW_LOOPBACK required for this test.c C sX t � t jt jt j��:}dD ].}|�t jt j|� | �||�t jt j�� qW 5 Q R X d S )N)r rh ) r r r r r� � SOL_CAN_RAWr> r� r� )r; r Zloopbackr r r � testLoopback� s ��zBasicCANTest.testLoopback�CAN_RAW_FILTERz-socket.CAN_RAW_FILTER required for this test.c C s| d\}}t �d||�}t�tjtjtj��H}|�tjtj|� | � ||�
tjtjd�� |�tjtjt |�� W 5 Q R X d S )N)� i z=IIrV ) r �packr r r r r� r? rA r� r� r� )r; �can_idZcan_maskZ
can_filterr r r r �
testFilter� s �zBasicCANTest.testFilterN)r@ rA rB r( r� r- r r r5 r7 r8 r: r= r@ rE r r r r r0 O s( �
�
�
�r0 c @ s� e Zd Zddd�Zedd� �Zedd� �Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z e
�eed�d�dd� �Ze
�eed�d�dd� �ZdS ) �CANTestr� c C s t j| |d� d S r� )r� rF r� r r r rF � s zCANTest.__init__c C s& t |�}|�dd�}t�| j|||�S )NrV � )r� �ljustr rC r\ )�clsrD �data�can_dlcr r r �build_can_frame� s zCANTest.build_can_framec C s& t �| j|�\}}}|||d |� fS r+ )r r r\ )rI �framerD rK rJ r r r �dissect_can_frame� s zCANTest.dissect_can_framec C s6 | j �| j�\}}| �| j|� | �|d | j� d S r� )r �recvfromr[ r� �cfrY �r; rP r� r r r �
testSendFrame� s zCANTest.testSendFramec C s | � dd�| _| j�| j� d S )Nr s �rL rP r� r� r: r r r �_testSendFrame� s zCANTest._testSendFramec C s$ | j �| j�\}}| �| j|� d S r+ )r rO r[ r� rP rQ r r r �testSendMaxFrame� s zCANTest.testSendMaxFramec C s | � dd�| _| j�| j� d S )Nr s rS r: r r r �_testSendMaxFrame� s zCANTest._testSendMaxFramec C sD | j �| j�\}}| �| j|� | j �| j�\}}| �| j|� d S r+ )r rO r[ r� �cf1�cf2rQ r r r �testSendMultiFrames� s zCANTest.testSendMultiFramesc C s< | � dd�| _| j�| j� | � dd�| _| j�| j� d S )N� s D3"� s �"3)rL rW r� r� rX r: r r r �_testSendMultiFrames� s z CANTest._testSendMultiFramesr2 r3 c C sP | j �| j�\}}| �| j|� | �|�\}}}| �| j|� | �| j|� d S r+ )r� rO r[ r� rP rN rD rJ )r; rP r� rD rK rJ r r r �_testBCM� s
zCANTest._testBCMc C s� t � t jt jt j�}| �|j� |�| jf� d| _t dddg�| _
| � | j| j
�| _ t j
}d}d}d } } }}d} d}
t�| j|||||||| |
�
} | | j } |�| �}
| �|
t| �� d S )N�# � � �� r i" rh )r r r r2 rO r r� rY rD �bytesrJ rL rP r4 r rC r] r� r� r� )r; ZbcmZopcoder� r� Z
ival1_secondsZ
ival1_usecZ
ival2_secondsZ
ival2_usecZ
bcm_can_idZnframes�headerZheader_plus_frameZ
bytes_sentr r r �testBCM� s4 �
zCANTest.testBCMN)r� )r@ rA rB rF � classmethodrL rN rR rT rU rV rY r\ r� r- r r r] rd r r r r rF � s&
�
�rF z!CAN ISOTP required for this test.c sV e Zd Z� fdd�Zdd� Zdd� Ze�ee d�d�d d
� �Z
d d � Z d
d� Z � Z
S )� ISOTPTestc s t � j||� d| _d S )NrS )rE rF rY rJ rM r r rF s zISOTPTest.__init__c C s t j t j t j t j d S r+ )r r1 r r! r r: r r r r( s z ISOTPTest.testCrucialConstantsc C s$ t � t jt jt j��}W 5 Q R X d S r+ r6 r r r r r7 s zISOTPTest.testCreateSocketr! z(socket.CAN_ISOTP required for this test.c C s$ t � t jt jt j��}W 5 Q R X d S r+ )r r r r! r r r r �testCreateISOTPSocket s z ISOTPTest.testCreateISOTPSocketc
C sF t � t jt jt j��(}| �td�� |�d� W 5 Q R X W 5 Q R X d S )Nr; )r< rh r� )r r r r! r� r rX r r r r r= s z"ISOTPTest.testTooLongInterfaceNamec
C s� zJt � t jt jt j��,}| jddf}|�|� | �|�� |� W 5 Q R X W nB tk
r� } z$|j t j
krz| � d| j � n� W 5 d }~X Y nX d S )Nr^ iV rW ) r r r r! rY rX r� r� r r; ZENODEVrZ )r; r r� r� r r r �testBind s
�zISOTPTest.testBind)r@ rA rB rF r( r7 r� r- r r rg r= rh rQ r r rM r rf � s �
rf z#RDS sockets required for this test.c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� BasicRDSTestc C s t j t j d S r+ )r ZAF_RDSr$ r: r r r r( , s z!BasicRDSTest.testCrucialConstantsc C s" t � t jt jd��}W 5 Q R X d S r� )r r$ r% r r r r r7 0 s z BasicRDSTest.testCreateSocketc C sJ d}t � t jt jd��*}|�t jt j|� |�t jt j|� W 5 Q R X d S )Ni @ r )r r$ r% r� r& Z SO_RCVBUFZ SO_SNDBUF)r; r[ r r r r �testSocketBufferSize4 s z!BasicRDSTest.testSocketBufferSizeN)r@ rA rB r( r7 rj r r r r ri ) s ri c s� e Zd Zd dd�Z� fdd�Zdd� Zdd � Zd
d � Zd d
� Ze e
j
d�dd� �Z e e
j
d�dd� �Z dd� Z
dd� Zdd� Zdd� Z� ZS ) �RDSTestr� c C s t j| |d� d S r� )r� rF r� r r r rF > s zRDSTest.__init__c s t � �� t�� | _d S r+ )rE r< rG rm Zevtr: rM r r r< A s
z
RDSTest.setUpc C s2 | j �| j�\}}| �| j|� | �| j|� d S r+ )r6 rO r[ r� rJ r� �r; rJ r� r r r �testSendAndRecvE s zRDSTest.testSendAndRecvc C s" d| _ | j�| j dt| jf� d S �Nr� r �rJ r� r r� r8 r: r r r �_testSendAndRecvJ s zRDSTest._testSendAndRecvc C sH | j �| jtj�\}}| �| j|� | j �| j�\}}| �| j|� d S r+ )r6 rO r[ r �MSG_PEEKr� rJ rl r r r �testPeekN s zRDSTest.testPeekc C s" d| _ | j�| j dt| jf� d S rn ro r: r r r � _testPeekT s zRDSTest._testPeek�recvmsgc C s( | j �| j�\}}}}| �| j|� d S r+ )r6 rt r[ r� rJ )r; rJ �ancdataZ msg_flagsr� r r r �testSendAndRecvMsgX 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 )rJ r� rw r� r8 r: r r r �_testSendAndRecvMsg] s zRDSTest._testSendAndRecvMsgc C sD | j �| j�\}}| �| j|� | j �| j�\}}| �| j|� d S r+ )r6 rO r[ r� �data1�data2rl r r r �testSendAndRecvMultib s z RDSTest.testSendAndRecvMultic C s@ d| _ | j�| j dt| jf� d| _| j�| jdt| jf� d S )Ns baconr s egg)ry r� r r� r8 rz r: r r r �_testSendAndRecvMultii s z RDSTest._testSendAndRecvMultic C sL t � | jgg g d�\}}}| �| j|� | j�| j�\}}| �| j|� d S )N� @)�selectr6 r rO r[ r� rJ )r; r r� rU rJ r� r r r �
testSelectp s zRDSTest.testSelectc C s" d| _ | j�| j dt| jf� d S )Ns selectr ro r: r r r � _testSelectv s zRDSTest._testSelect)r� )r@ rA rB rF r< rm rp rr rs r� r rv rx r{ r| r r� rQ r r rM r rk ; s
rk z'QIPCRTR sockets required for this test.c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�BasicQIPCRTRTestc C s
t j d S r+ )r r) r: r r r r( ~ s z%BasicQIPCRTRTest.testCrucialConstantsc C s t � t jt j��}W 5 Q R X d S r+ )r r) r r r r r r7 � s z!BasicQIPCRTRTest.testCreateSocketc C s4 t � t jt j��}| �|�� d d� W 5 Q R X d S �Nrh r )r r) r r� r� r r r r � testUnbound� s z BasicQIPCRTRTest.testUnboundc C sJ t � t jt j��0}tj||�� d d� | �|�� d d� W 5 Q R X d S )Nr r� rh )r r) r r r7 r� �assertNotEqualr r r r � testBindSock� s z BasicQIPCRTRTest.testBindSockc C s4 t � t jt j��}| jttj|dd� W 5 Q R X d S )Nra r� )r r) r r r r r7 r r r r �testInvalidBindSock� s z$BasicQIPCRTRTest.testInvalidBindSockc C s> t � t jt j��$}|�d� | �|�� d d� W 5 Q R X d S )N)�{ r� rh r )r r) r r� r� r� r r r r �testAutoBindSock� s
z!BasicQIPCRTRTest.testAutoBindSockN) r@ rA rB r( r7 r� r� r� r� r r r r r� z s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�BasicVSOCKTestc C s
t j d S r+ )r r� r: r r r r( � s z#BasicVSOCKTest.testCrucialConstantsc C s4 t j t j t j t j t j t j t j t j d S r+ ) r �SO_VM_SOCKETS_BUFFER_SIZE� SO_VM_SOCKETS_BUFFER_MIN_SIZE� SO_VM_SOCKETS_BUFFER_MAX_SIZEr� ZVMADDR_PORT_ANYZVMADDR_CID_HOSTZVM_SOCKETS_INVALID_VERSIONr r: r r r �testVSOCKConstants� s z!BasicVSOCKTest.testVSOCKConstantsc C s t � t jt j��}W 5 Q R X d S r+ )r r� r5 r r r r r7 � s z BasicVSOCKTest.testCreateSocketc C s� t � t jt j���}|�t jt j�}|�t jt j�}|�t jt j�}|�t jt j|d � |�t jt j|d � |�t jt j|d � | �|d |�t jt j�� | �|d |�t jt j�� | �|d |�t jt j�� W 5 Q R X d S �Nr� ) r r� r5 r� r� r� r� r� r� )r; r Zorig_maxrR Zorig_minr r r rj � sJ ��� � � �
��
��
��z#BasicVSOCKTest.testSocketBufferSizeN)r@ rA rB r( r� r7 rj r r r r r� � s
r� c @ s� e Zd Zd+dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Ze�e�Zejd$d%� �Zd&d'� Zd(d)� Zd*S ),� BasicTCPTestr� c C s t j| |d� d S r� �r� rF r� r r r rF � s zBasicTCPTest.__init__c C s | j �d�}| �|t� d S r� )r� r� r� r� r� r r r �testRecv� s zBasicTCPTest.testRecvc C s | j �t� d S r+ �r� r� r� r: r r r � _testRecv� s zBasicTCPTest._testRecvc C s8 | j �tt�d �}| j �d�}|| }| �|t� d S �N� r� )r� r� r� r� r� )r; �seg1�seg2r� r r r �testOverFlowRecv� s z BasicTCPTest.testOverFlowRecvc C s | j �t� d S r+ r� r: r r r �_testOverFlowRecv� s z BasicTCPTest._testOverFlowRecvc C s | j �d�\}}| �|t� d S r� )r� rO r� r� �r; r� r� r r r � testRecvFrom� s zBasicTCPTest.testRecvFromc C s | j �t� d S r+ r� r: r r r �
_testRecvFrom� s zBasicTCPTest._testRecvFromc C s@ | j �tt�d �\}}| j �d�\}}|| }| �|t� d S r� )r� rO r� r� r� )r; r� r� r� r� r r r �testOverFlowRecvFrom� s z!BasicTCPTest.testOverFlowRecvFromc C s | j �t� d S r+ r� r: r r r �_testOverFlowRecvFrom� s z"BasicTCPTest._testOverFlowRecvFromc C s0 d}| j �d�}|sq ||7 }q| �|d� d S )Nr� r� � ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff)r� r� r� )r; r� �readr r r � testSendAll� s
zBasicTCPTest.testSendAllc C s d}| j �|� d S )Nr� )r� r� )r; Z big_chunkr r r � _testSendAll� s zBasicTCPTest._testSendAllc C sP | j �� }t�|tjtj�}| �|j� | �|tj� |� d�}| �
|t � d S r� ) r� r r Zfromfdr4 r5 rO r r r� r� r� )r; r r� r� r r r �
testFromFd� s
zBasicTCPTest.testFromFdc C s | j �t� d S r+ r� r: r r r � _testFromFd s zBasicTCPTest._testFromFdc C s0 | j �� }| �|j� |�d�}| �|t� d S r� )r� �duprO r r� r� r� )r; r� r� r r r �testDup s
zBasicTCPTest.testDupc C s | j �t� d S r+ r� r: r r r �_testDup s zBasicTCPTest._testDupc C s& | j �d�}| �|t� | j�� d S r� )r� r� r� r� ro ry r� r r r � testShutdown s zBasicTCPTest.testShutdownc C s | j �t� | j �d� d S r� )r� r� r� �shutdownr: r r r �
_testShutdown s zBasicTCPTest._testShutdownc C sX dd l }| j�t� | �t| jj|jd � | �t| jjd|jd � | j�d� d S )Nr rh r� ) rc r� r� r� r rJ r� rd �UINT_MAX�r; rc r r r �_testShutdown_overflow$ s � �z#BasicTCPTest._testShutdown_overflowc C s� | j �� }| j �� }| �||� | �| j j� | �t| j jd� | j � � t
j
t
j t
j |d�}| �
|j � |�d�}| �|t� d S )Nr� r� )r� r r r� r �_closedr r r� r r r4 r5 rO r� )r; r r r� r� r r r �
testDetach/ s
zBasicTCPTest.testDetachc C s | j �t� d S r+ r� r: r r r � _testDetach? s zBasicTCPTest._testDetachN)r� )r@ rA rB rF r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r 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 )� BasicUDPTestr� c C s t j| |d� d S r� )r� rF r� r r r rF E s zBasicUDPTest.__init__c C s | j �tt��}| �|t� d S r+ )r6 r� r� r� r� r� r r r �testSendtoAndRecvH s z BasicUDPTest.testSendtoAndRecvc C s | j �tdt| jf� d S r� �r� r r� r� r8 r: r r r �_testSendtoAndRecvM s z BasicUDPTest._testSendtoAndRecvc C s$ | j �tt��\}}| �|t� d S r+ )r6 rO r� r� r� r� r r r r� P s zBasicUDPTest.testRecvFromc C s | j �tdt| jf� d S r� r� r: r r r r� U s zBasicUDPTest._testRecvFromc C s | � t| jjd� d S �NrE )r ru r6 rO r: r r r �testRecvFromNegativeX s z!BasicUDPTest.testRecvFromNegativec C s | j �tdt| jf� d S r� r� r: r r r �_testRecvFromNegative\ s z"BasicUDPTest._testRecvFromNegativeN)r� )
r@ rA rB rF r� r� r� r� r� r� r r r r r� C s
r� c s� e Zd ZdZ� fdd�Zdd� ZdZdd� Zd d
� Zd d � Z d
d� Z
dZ dD ]Z e e
ee d�O Z qLdZdZdZddd�Z� ZS )�SendrecvmsgBaser} c s t �� | _t� �� d S r+ )rG rm �
misc_eventrE r< r: rM r r r< � s
zSendrecvmsgBase.setUpc C s | j �|�S r+ )�cli_sockr� r� r r r � sendToServer� s z SendrecvmsgBase.sendToServerr c G s | j j|| jt|�d � � S r+ )r� rw �sendmsg_to_server_defaultsr� �r; rK r r r �sendmsgToServer� s �z SendrecvmsgBase.sendmsgToServerc G s |j |f|�� }| �|� |S r+ )rt �registerRecvmsgResult)r; r� r[ rK �resultr r r � doRecvmsg� s
zSendrecvmsgBase.doRecvmsgc C s d S r+ r �r; r� r r r r� � s z%SendrecvmsgBase.registerRecvmsgResultc C s | � ||� d S r+ �r� �r; Zaddr1Zaddr2r r r �checkRecvmsgAddress� s z#SendrecvmsgBase.checkRecvmsgAddressr )�
MSG_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� )
r; r� �eor�checkset�
checkunset�ignoreZ
defaultsetZ defaultunsetZinbothr_ r r r �
checkFlags� s&
�zSendrecvmsgBase.checkFlags)Nr r r )r@ rA rB � fail_timeoutr< r� r� r� r� r� r� r� r� ru r r� r� r� r� rQ r r rM r r� z s
r� c @ s e Zd Zdd� ZdS )�RecvmsgIntoMixinc G sf t |�}|j|gf|�� }| �|� | �|d d� | �|d |� t|d |d � �f|dd � S r� )r� � recvmsg_intor� �assertGreaterEqual�assertLessEqualrb )r; r� r[ rK �bufr� r r r r� � s
zRecvmsgIntoMixin.doRecvmsgN)r@ rA rB r� r r r r r� � s r� c s e Zd Ze� fdd��Z� ZS )�SendrecvmsgDgramFlagsBasec s t � jtjB S r+ )rE r� r � MSG_TRUNCr: rM r r r�
s z5SendrecvmsgDgramFlagsBase.msg_flags_non_eor_indicator)r@ rA rB �propertyr� rQ r r rM r r� � s r� c s e Zd Ze� fdd��Z� ZS )�SendrecvmsgSCTPFlagsBasec s t � jtjB S r+ )rE r� r ZMSG_EORr: rM r r r�
s z0SendrecvmsgSCTPFlagsBase.msg_flags_eor_indicator)r@ rA rB r� r� rQ r r rM r r�
s r� c @ s8 e Zd Zedd� �Zedd� �Zedd� �Zdd� Zd S )
� SendrecvmsgConnectionlessBasec C s | j S r+ )r6 r: r r r � serv_sock
s z'SendrecvmsgConnectionlessBase.serv_sockc C s | j S r+ )r� r: r r r r�
s z&SendrecvmsgConnectionlessBase.cli_sockc C s g g d| j fS r� )r� r: r r r r�
s z8SendrecvmsgConnectionlessBase.sendmsg_to_server_defaultsc C s | j �|| j�S r+ )r� r r� r� r r r r�
s z*SendrecvmsgConnectionlessBase.sendToServerN)r@ rA rB r� r� r� r� r� r r r r r�
s
r� c @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )�SendrecvmsgConnectedBasec C s | j S r+ )r� r: r r r r� (
s z"SendrecvmsgConnectedBase.serv_sockc C s | j S r+ )r� r: r r r r� ,
s z!SendrecvmsgConnectedBase.cli_sockc C s d S r+ r r� r r r r� 0
s z,SendrecvmsgConnectedBase.checkRecvmsgAddressN)r@ rA rB r� r� r� r� r r r r r� "
s
r� c s e Zd Z� fdd�Z� ZS )� SendrecvmsgServerTimeoutBasec s t � �� | j�| j� d S r+ )rE r< r� r� r� r: rM r r r< 9
s
z"SendrecvmsgServerTimeoutBase.setUpr� r r rM r r� 6
s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zeed �d d � �Zejd d!� �Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*S )+� SendmsgTestsc C s | � | j�tt��t� d S r+ �r� r� r� r� r� r: r r r � testSendmsgB
s zSendmsgTests.testSendmsgc C s | � | �tg�tt�� d S r+ �r� r� r� r� r: r r r � _testSendmsgF
s zSendmsgTests._testSendmsgc C s | � | j�tt��t� d S r+ r� r: r r r �testSendmsgDataGeneratorI
s z%SendmsgTests.testSendmsgDataGeneratorc C s& | � | �dd� tfD ��tt�� d S )Nc s s | ]
}|V qd S r+ r r� r r r r� N
s z9SendmsgTests._testSendmsgDataGenerator.<locals>.<genexpr>r� r: r r r �_testSendmsgDataGeneratorM
s �z&SendmsgTests._testSendmsgDataGeneratorc C s | � | j�tt��t� d S r+ r� r: r r r � testSendmsgAncillaryGeneratorQ
s z*SendmsgTests.testSendmsgAncillaryGeneratorc C s( | � | �tgdd� dD ��tt�� d S )Nc s s | ]
}|V qd S r+ r r� r r r r� V
s z>SendmsgTests._testSendmsgAncillaryGenerator.<locals>.<genexpr>r r� r: r r r � _testSendmsgAncillaryGeneratorU
s �z+SendmsgTests._testSendmsgAncillaryGeneratorc C s | � | j�tt��t� d S r+ r� r: r r r �testSendmsgArrayY
s z SendmsgTests.testSendmsgArrayc C s$ | � | �t�dt�g�tt�� d S )N�B)r� r� �arrayr� r� r: r r r �_testSendmsgArray]
s �z SendmsgTests._testSendmsgArrayc C s | � | j�tt��t� d S r+ r� r: r r r �testSendmsgGathera
s z SendmsgTests.testSendmsgGatherc C s. | � | �td d� tdd � g�tt�� d S �Nr� r� r: r r r �_testSendmsgGathere
s z SendmsgTests._testSendmsgGatherc C s | � | j�d�d� d S �NrF � done�r� r� r� r: r r r �testSendmsgBadArgsh
s z SendmsgTests.testSendmsgBadArgsc C s� | � t| jj� | � t| jd� | � t| jt� � | � t| jt� g� | � t| jtt� g� | � t| jtgt� � | � t| jtgg t� � | � t| jtgg dt� � | �d� d S )Ns not in an iterabler r� )r r� r� rw r� �objectr� r� r: r r r �_testSendmsgBadArgsl
s:
�
�
�
�
�
�
�z SendmsgTests._testSendmsgBadArgsc C s | � | j�d�d� d S r� r� r: r r r �testSendmsgBadCmsg~
s z SendmsgTests.testSendmsgBadCmsgc C s� | � t| jtgt� g� | � t| jtgt� ddfg� | � t| jtgdt� dfg� | � t| jtgddt� fg� | � t| jtgdg� | � t| jtgdg� | �d� d S )Nr � data)r r )r r r� r r� �r r� r� r� r� r� r: r r r �_testSendmsgBadCmsg�
s2
�
�
�
�
�
�z SendmsgTests._testSendmsgBadCmsg�
CMSG_SPACEc C s | � | j�d�d� d S r� r� r: r r r �testSendmsgBadMultiCmsg�
s z$SendmsgTests.testSendmsgBadMultiCmsgc C sB | � t| jtgdddg� | � t| jtgdt� g� | �d� d S )Nr r� �r r r� r� r� r: r r r �_testSendmsgBadMultiCmsg�
s
�
�z%SendmsgTests._testSendmsgBadMultiCmsgc C s | � | j�d�d� d S r� r� r: r r r �testSendmsgExcessCmsgReject�
s z(SendmsgTests.testSendmsgExcessCmsgRejectc C sN t td�s@| �t��}| �tgddg� W 5 Q R X | �|jj� | � d� d S )Nr� r� r� )
r r r r r� r� � assertIsNoner r; r� �r; r" r r r � _testSendmsgExcessCmsgReject�
s
z)SendmsgTests._testSendmsgExcessCmsgRejectc C s d S r+ r r: r r r �testSendmsgAfterClose�
s z"SendmsgTests.testSendmsgAfterClosec C s | j �� | �t| jtg� d S r+ )r� r r r r� r� r: r r r �_testSendmsgAfterClose�
s
z#SendmsgTests._testSendmsgAfterCloseN)r@ rA rB r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r� r� r r r r r r r r� >
s,
r� c @ sR e Zd Zdd� Zdd� Zdd� Zdd� Zeej d kd
�d d � �Z
e
j d
d� �Z dS )�SendmsgStreamTestsc C s | � | j�tt��t� d S r+ r� r: r r r �testSendmsgExplicitNoneAddr�
s z.SendmsgStreamTests.testSendmsgExplicitNoneAddrc C s" | � | �tgg dd �tt�� d S r� r� r: r r r � _testSendmsgExplicitNoneAddr�
s z/SendmsgStreamTests._testSendmsgExplicitNoneAddrc C s. | � | j�d�d� | �| jj| jd�� d S �NrB � aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa�r0 �r� r� r� r r� ry r� r: r r r �testSendmsgTimeout�
s z%SendmsgStreamTests.testSendmsgTimeoutc
C s� zr| j�d� z| �dg� qW nD tjk
r6 Y n: tk
rd } z|jtj krT� W 5 d }~X Y n X | �
d� W 5 | j �� X d S )N� ��Q��?r �socket.timeout not raised) r� rf r� r� r� r r0 r r; �ENOMEMr r| r r r �_testSendmsgTimeout�
s z&SendmsgStreamTests._testSendmsgTimeout> ri z<MSG_DONTWAIT not known to work on this platform when sendingc C s. | � | j�d�d� | �| jj| jd�� d S r r r: r r r �testSendmsgDontWait�
s z&SendmsgStreamTests.testSendmsgDontWaitc C s\ zJ| �t��}| �dgg tj� qW 5 Q R X | �|jj t j
t j t j f� W 5 | j �� X d S �Nr )
r� rf r r r� r Z MSG_DONTWAITr r r; ZEAGAINZ EWOULDBLOCKr
r� r r r �_testSendmsgDontWait�
s
�z'SendmsgStreamTests._testSendmsgDontWaitN)
r@ rA rB r r r
r r� rP ro r r� r r r r r r �
s
�
r c @ s e Zd Zdd� Zdd� ZdS )�SendmsgConnectionlessTestsc C s d S r+ r r: r r r �testSendmsgNoDestAddr�
s z0SendmsgConnectionlessTests.testSendmsgNoDestAddrc C s2 | � t| jjtg� | � t| jjtgg dd � d S r� )r r r� rw r� r: r r r �_testSendmsgNoDestAddr�
s � �z1SendmsgConnectionlessTests._testSendmsgNoDestAddrN)r@ rA rB r r r r r r r �
s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zeed!�d"d#� �Zejd$d%� �Zeejd&�d'd(� �Zejd)d*� �Zd+S ),�RecvmsgGenericTestsc C sR | � | jtt��\}}}}| �|t� | �|| j� | �|g � | j|dd� d S �NT�r� �r� r� r� r� r� r� r� r� �r; r� ru r� r� r r r � testRecvmsg s
z RecvmsgGenericTests.testRecvmsgc C s | � t� d S r+ �r� r� r: r r r � _testRecvmsg s z RecvmsgGenericTests._testRecvmsgc C sV | � | jtt�dd�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr Tr r r r r r �testRecvmsgExplicitDefaults s � z/RecvmsgGenericTests.testRecvmsgExplicitDefaultsc C s | � t� d S r+ r r: r r r � _testRecvmsgExplicitDefaults s z0RecvmsgGenericTests._testRecvmsgExplicitDefaultsc C sV | � | jtt�d �\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nr Tr r r r r r �testRecvmsgShorter s
� z&RecvmsgGenericTests.testRecvmsgShorterc C s | � t� d S r+ r r: r r r �_testRecvmsgShorter' s z'RecvmsgGenericTests._testRecvmsgShorterc C s^ | � | jtt�d �\}}}}| �|td d� � | �|| j� | �|g � | j|dd� d S )Nr� ���Fr r r r r r �testRecvmsgTrunc* s
� z$RecvmsgGenericTests.testRecvmsgTruncc C s | � t� d S r+ r r: r r r �_testRecvmsgTrunc3 s z%RecvmsgGenericTests._testRecvmsgTruncc C sT | � | jtt�d�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )Nrh Tr r r r r r � testRecvmsgShortAncillaryBuf6 s � z0RecvmsgGenericTests.testRecvmsgShortAncillaryBufc C s | � t� d S r+ r r: r r r � _testRecvmsgShortAncillaryBuf? s z1RecvmsgGenericTests._testRecvmsgShortAncillaryBufc C sT | � | jtt�d�\}}}}| �|t� | �|| j� | �|g � | j|dd� d S )N� ( Tr r r r r r �testRecvmsgLongAncillaryBufB s � z/RecvmsgGenericTests.testRecvmsgLongAncillaryBufc C s | � t� d S r+ r r: r r r � _testRecvmsgLongAncillaryBufK s z0RecvmsgGenericTests._testRecvmsgLongAncillaryBufc C s" | j �� | �t| j| j d� d S r� )r� r r r r� r: r r r �testRecvmsgAfterCloseN s
z)RecvmsgGenericTests.testRecvmsgAfterClosec C s d S r+ r r: r r r �_testRecvmsgAfterCloseS s z*RecvmsgGenericTests._testRecvmsgAfterClosec C s< z*| j�d� | �tj| j| jtt �� W 5 | j �� X d S )Nr )
r� rf r� r� r r r0 r� r� r� r: r r r �testRecvmsgTimeoutV s �z&RecvmsgGenericTests.testRecvmsgTimeoutc C s | � | jj| jd�� d S �Nr )r r� ry r� r: r r r �_testRecvmsgTimeout_ s z'RecvmsgGenericTests._testRecvmsgTimeoutrq c C s | � | jtt�d dtj�\}}}}| �|td d� � | �|| j� | �|g � | j |dt
tdd�d� | � | jtt�dtj�\}}}}| �|t� | �|| j� | �|g � | j |dd� | � | jtt��\}}}}| �|t� | �|| j� | �|g � | j |dd� d S ) Nr� r r! Fr� �r� r� Tr ) r� r� r� r� r rq r� r� r� r� ru r r r r �testRecvmsgPeekb s2
�
� � z#RecvmsgGenericTests.testRecvmsgPeekc C s | � t� d S r+ r r: r r r �_testRecvmsgPeek� s z$RecvmsgGenericTests._testRecvmsgPeekrw c C s` | j �| j� | �| j tt��\}}}}| �|t� | �|| j� | �|g � | j |dd� d S r )
r� r� r� r� r� r� r� r� r� r� r r r r �testRecvmsgFromSendmsg� s z*RecvmsgGenericTests.testRecvmsgFromSendmsgc C s. | � | �td d� tdd � g�tt�� d S r� r� r: r r r �_testRecvmsgFromSendmsg� s z+RecvmsgGenericTests._testRecvmsgFromSendmsgN)r@ rA rB r r r r r r r" r# r$ r% r' r( r) r* r+ r- r� r r/ r� r0 r1 r2 r r r r r s0
"
r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�RecvmsgGenericStreamTestsc C sN | � | jd�\}}}}| �|d� | �|| j� | �|g � | j|d d� d S )Nr� r� r )r� r� r� r� r� r� r r r r �testRecvmsgEOF� s
z(RecvmsgGenericStreamTests.testRecvmsgEOFc C s | j �� d S r+ )r� r r: r r r �_testRecvmsgEOF� s z)RecvmsgGenericStreamTests._testRecvmsgEOFc C s� | � | jtt�d �\}}}}| �|| j� | �|g � | j|dd� | � | jd�\}}}}| �|| j� | �|g � | j|dd� || }| �|t� d S )Nr� Fr r� T)r� r� r� r� r� r� r� r� )r; r� ru r� r� r� r� r r r �testRecvmsgOverflow� s
� z-RecvmsgGenericStreamTests.testRecvmsgOverflowc C s | � t� d S r+ r r: r r r �_testRecvmsgOverflow� s z.RecvmsgGenericStreamTests._testRecvmsgOverflowN)r@ rA rB r4 r5 r6 r7 r r r r r3 � s r3 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 )NrE r rs Tr )
r r� r� rt ru r� r� r� r� r� r� r� r� r r r r �testRecvmsgBadArgs� sH � � � � � � z RecvmsgTests.testRecvmsgBadArgsc C s | � t� d S r+ r r: r r r �_testRecvmsgBadArgs� s z RecvmsgTests._testRecvmsgBadArgsN)r@ rA rB r9 r: r r r r r8 � s r8 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 writablerE Tr )
r� r� r� r r� r� r� r� ru r� r� r� r� �r; r� �nbytesru r� r� r r r �testRecvmsgIntoBadArgs� s` � � � � � � � � z'RecvmsgIntoTests.testRecvmsgIntoBadArgsc C s | � t� d S r+ r r: r r r �_testRecvmsgIntoBadArgs� s z(RecvmsgIntoTests._testRecvmsgIntoBadArgsc C sx t tt��}| j�dd� |fD ��\}}}}| �|tt�� | �|t t�� | �|| j� | �|g � | j|dd� d S )Nc s s | ]
}|V qd S r+ r r� r r r r� � s z<RecvmsgIntoTests.testRecvmsgIntoGenerator.<locals>.<genexpr>Tr ) r� r� r� r� r� r� r� r� r� r< r r r �testRecvmsgIntoGenerator� s � z)RecvmsgIntoTests.testRecvmsgIntoGeneratorc C s | � t� d S r+ r r: r r r �_testRecvmsgIntoGenerator s z*RecvmsgIntoTests._testRecvmsgIntoGeneratorc C sx t � ddgtt� �}| j�|g�\}}}}| �|tt�� | �|�� t� | �|| j� | �|g � | j |dd� d S )Nr� r Tr )
r� r� r� r� r� r� �tobytesr� r� r� r< r r r �testRecvmsgIntoArray s z%RecvmsgIntoTests.testRecvmsgIntoArrayc C s | � t� d S r+ r r: r r r �_testRecvmsgIntoArray s z&RecvmsgIntoTests._testRecvmsgIntoArrayc C s� t d�}t d�}t d�}| j�|t|�dd� |g�\}}}}| �|td�� | �|t d�� | �|t d�� | �|t d �� | �|| j� | �|g � | j|d
d � d S ) Ns ----s
0123456789s --------------r� � � Mary had a little lambs Marys
01 had a 9s little lamb---Tr ) r� r� r� �
memoryviewr� r� r� r� r� )r; Zb1Zb2Zb3r= ru r� r� r r r �testRecvmsgIntoScatter s � z'RecvmsgIntoTests.testRecvmsgIntoScatterc C s | � d� d S )NrF )r� r: r r r �_testRecvmsgIntoScatter% s z(RecvmsgIntoTests._testRecvmsgIntoScatterN) r@ rA rB r> r? r@ rA rC rD rH rI r r r r r; � s
r; c @ sb e Zd Zz ddlZW n ek
r, dZY nX edej�Zee d�dd� �Z
ee d�dd � �Z dS )
�CmsgMacroTestsr Ni����CMSG_LENc C s� | j t�d� d }ttd��tt|d |�� }| �t�d�t�d�jd � |D ]2}t�|�}| �|t�d� |� | � || j � qT| �
t tjd� | �
t tj|� | �
t tjt j
� d S )Nr rh � r2 r� rE )�socklen_t_limitr rK �listr� r? r� �itemsizer� r� r rJ rP �maxsize)r; �toobigr �nr, r r r � testCMSG_LEN6 s
zCmsgMacroTests.testCMSG_LENr� c C s� | j t�d� d }ttd��tt|d |�� }t�d�}| �|t�d�jd � |D ]T}t�|�}| �||� | �|t� |�� | �||t� d� � | �
|| j � |}qX| � t tjd� | � t tj|� | � t tjt
j� d S )Nrh rL r r2 r� rE )rM r r� rN r� r? r� rO r� rK r� r rJ rP rP )r; rQ r ZlastrR r, r r r �testCMSG_SPACEK s
z CmsgMacroTests.testCMSG_SPACE) r@ rA rB rc r� rM �minrd r� r rS rT r r r r rJ ) s
rJ c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zdddd�Z dd� Z
dd� Z dd� Z dd� Z
eed�dd� �Zejdd � �Zd d � Zd d � Ze�ejd!kd"�e�ed#�eed�d$d%� ���Zeje�ejd!kd"�e�ed#�d&d'� ���Ze�ejd!kd"�e�ed#�eed�d(d)� ���Z e je�ejd!kd"�e�ed#�d*d+� ���Z d,d-� Z e�ejd!kd.�d/d0� �Z d1d2� Z d3d4� Z!d5d6� Z"eed�d7d8� �Z#e#jd9d:� �Z$ded;d<�Z%d=d>� Z&d?d@� Z'dAdB� Z(dCdD� Z)dEdF� Z*dGdH� Z+dIdJ� Z,dKdL� Z-dMdN� Z.dOdP� Z/dfdQdR�Z0dSdT� Z1dUdV� Z2dWdX� Z3dYdZ� Z4d[d\� Z5d]d^� Z6d_d`� Z7dadb� Z8dcS )g�
SCMRightsTesti����c C s\ g }t |�D ]J}t�� \}}| �tj|� | �tj|� t�|t|�� � � |�
|� q |S r+ ) r� r� ZmkstemprO r� r� r �writer� �encode�append)r; rR �fdsr2 r r� r r r �newFDsl s zSCMRightsTest.newFDsc C sB t |�D ]4\}}t�|dtj� | �t�|d�t|��� � qd S )Nr r� )� enumerater� �lseek�SEEK_SETr� r� r� rX )r; rZ rR r r r r �checkFDsx s zSCMRightsTest.checkFDsc C s | � | j|� d S r+ )rO �closeRecvmsgFDsr� r r r r� s z#SCMRightsTest.registerRecvmsgResultc C sn |d D ]`\}}}|t jkr|t jkrt�d�}|�|d t|�t|�|j � � |D ]}t�|� qXqd S )Nrh r2 ) r r& �
SCM_RIGHTSr� � frombytesr� rO r� r )r; Zrecvmsg_result�
cmsg_level� cmsg_type� cmsg_datarZ r r r r r` � s
�
�z SCMRightsTest.closeRecvmsgFDsc
C s8 | � | �tgtjtjt�d| �|��fg�tt�� d S )Nr2 ) r� r� r� r r& ra r� r[ r� )r; rR r r r �createAndSendFDs� s ���z SCMRightsTest.createAndSendFDsrh r c C s� |\}}}}| � |t� | �|| j� | j|dtj|d� | �|t� | � t
|�|� t � d�} |D ]`}
| �|
t � |
\} } }
| � | tj
� | � | tj� | �|
t� | � t
|
�t d� | �|
� qd| � t
| �|� | �| � d S )NT�r� r� r� r2 r )r� r� r� r� r� r r� r rN r� r� r� �tupler& ra rb �
SIZEOF_INTrb r_ )r; Znumfdsr� �maxcmsgs� ignoreflagsr� ru r� r� rZ �itemrc rd re r r r �checkRecvmsgFDs� s& �
z SCMRightsTest.checkRecvmsgFDsc C s | � d| �| jtt�d�� d S )Nrh r& �rm r� r� r� r� r: r r r �testFDPassSimple� s �z SCMRightsTest.testFDPassSimplec
C s< | � | �tgtjtjt�d| �d���� fg�t t�� d S )Nr2 rh )
r� r� r� r r& ra r� r[ rB r� r: r r r �_testFDPassSimple� s ���z SCMRightsTest._testFDPassSimplec C s | � d| �| jtt�d�� d S )Nr r& rn r: r r r �testMultipleFDPass� s �z SCMRightsTest.testMultipleFDPassc C s | � d� d S �Nr �rf r: r r r �_testMultipleFDPass� s z!SCMRightsTest._testMultipleFDPassr� c C s* | � d| �| jtt�t�dt ��� d S rr )rm r� r� r� r� r r� ri r: r r r �testFDPassCMSG_SPACE� s ��z"SCMRightsTest.testFDPassCMSG_SPACEc C s | � d� d S rr rs r: r r r �_testFDPassCMSG_SPACE� s z#SCMRightsTest._testFDPassCMSG_SPACEc
C s0 | j d| �| jtt�t�dt ��tjd� d S )Nrh r �rk ) rm r� r� r� r� r rK ri r� r: r r r �testFDPassCMSG_LEN� s ��z SCMRightsTest.testFDPassCMSG_LENc C s | � d� d S r� rs r: r r r �_testFDPassCMSG_LEN� s z!SCMRightsTest._testFDPassCMSG_LENrj zskipping, see issue #12958zskipping, see issue #22397c C s$ | j d| �| jtt�d�dd� d S )Nr� r& )rj rn r: r r r �testFDPassSeparate� s �z SCMRightsTest.testFDPassSeparatec C sX | � d�\}}| �| �tgtjtjt�d|g�ftjtjt�d|g�fg�tt�� d S �Nr� r2 � r[ r� r� r� r r& ra r� r� �r; Zfd0Zfd1r r r �_testFDPassSeparate� s � ���z!SCMRightsTest._testFDPassSeparatec C s@ d}| j || �| jtt�t�t�t�t| � �dtj d� d S )Nr� )rj rk )
rm r� r� r� r� r r� ri rK r� )r; Znum_fdsr r r �testFDPassSeparateMinSpace� s �� �z(SCMRightsTest.testFDPassSeparateMinSpacec C sX | � d�\}}| �| �tgtjtjt�d|g�ftjtjt�d|g�fg�tt�� d S r{ r| r} r r r �_testFDPassSeparateMinSpace
s � ���z)SCMRightsTest._testFDPassSeparateMinSpacec
C sd z| � |g|�}W n< tk
rN } z | �|jt� | � |g�}W 5 d }~X Y nX | �|t|�� d S r+ )r� r r r; r> r� r� )r; r� ru r= r� r r r �sendAncillaryIfPossible
s z%SCMRightsTest.sendAncillaryIfPossiblezsee issue #24725c C s& | j d| �| jtt�d�tjd� d S )Nr r& rw )rm r� r� r� r� r r� r: r r r �testFDPassEmpty
s ��z SCMRightsTest.testFDPassEmptyc C s | � ttjtjdfg� d S �Nr� )r� r� r r& ra r: r r r �_testFDPassEmpty'
s
�z SCMRightsTest._testFDPassEmptyc C s� | � | jtt�d�\}}}}| �|t� | �|| j� | j|dtj d� | �
t|�d� |D ]6\}}}| �|tj � | �|tj � | �
t|�t� q\d S )Nr& Tr. rh )r� r� r� r� r� r� r� r� r r� r� r& ra �
assertLessri �r; r� ru r� r� rc rd re r r r �testFDPassPartialInt,
s � z"SCMRightsTest.testFDPassPartialIntc C s4 | � ttjtjt�d| jg��� d d� fg� d S )Nr2 rE )r� r� r r& ra r� �badfdrB r: r r r �_testFDPassPartialInt9
s ��z#SCMRightsTest._testFDPassPartialIntc C s� | � | jtt�d�\}}}}| �|t� | �|| j� | j|dtj d� | �
t|�d� t � d�}|D ]J\}}}| �|tj � | �|tj
� |�|d t|�t|�|j � � qf| �
t|�d� | �|� d S )Nr& Tr. r� r2 )r� r� r� r� r� r� r� r� r r� r� r� r& ra rb rO r_ ) r; r� ru r� r� rZ rc rd re r r r � testFDPassPartialIntInMiddle@
s" �
�
z*SCMRightsTest.testFDPassPartialIntInMiddlec
C sZ | � d�\}}| �ttjtjt�d|| jg��� d d� ftjtjt�d|g�fg� d S )Nr� r2 rE ) r[ r� r� r r& ra r� r� rB r} r r r � _testFDPassPartialIntInMiddleS
s � ���z+SCMRightsTest._testFDPassPartialIntInMiddlec C sJ |\}}}}| � |t� | �|| j� | � |g � | j|dtj|d� d S )NT�r� r� r� )r� r� r� r� r� r r� )r; r� rk r� ru r� r� r r r �checkTruncatedHeader_
s �z"SCMRightsTest.checkTruncatedHeaderc C s" | j | �| jtt��tjd� d S �Nrw �r� r� r� r� r� r r� r: r r r �testCmsgTruncNoBufSizei
s �z$SCMRightsTest.testCmsgTruncNoBufSizec C s | � d� d S r� rs r: r r r �_testCmsgTruncNoBufSizer
s z%SCMRightsTest._testCmsgTruncNoBufSizec C s$ | j | �| jtt�d�tjd� d S )Nr rw r� r: r r r �testCmsgTrunc0u
s �z SCMRightsTest.testCmsgTrunc0c C s | � d� d S r� rs r: r r r �_testCmsgTrunc0z
s z SCMRightsTest._testCmsgTrunc0c C s | � | �| jtt�d�� d S r� )r� r� r� r� r� r: r r r �testCmsgTrunc1�
s z SCMRightsTest.testCmsgTrunc1c C s | � d� d S r� rs r: r r r �_testCmsgTrunc1�
s z SCMRightsTest._testCmsgTrunc1c C s" | � | �| jtt�td �� d S r� )r� r� r� r� r� ri r: r r r �testCmsgTrunc2Int�
s �z SCMRightsTest.testCmsgTrunc2Intc C s | � d� d S r� rs r: r r r �_testCmsgTrunc2Int�
s z SCMRightsTest._testCmsgTrunc2Intc C s( | � | �| jtt�t�d�d �� d S r� )r� r� r� r� r� r rK r: r r r �testCmsgTruncLen0Minus1�
s �z%SCMRightsTest.testCmsgTruncLen0Minus1c C s | � d� d S r� rs r: r r r �_testCmsgTruncLen0Minus1�
s z&SCMRightsTest._testCmsgTruncLen0Minus1c C s� | � | jtt�|�\}}}}| �|t� | �|| j� | j|dtj d� |dkr\|g kr\d S | �t|�d� |d \}} }
| �|tj
� | �| tj � | � t|
�|� | �
t|
�|� t�d�} | �|
d t|
�t|
�| j � � | �| � d S )NT�r� r� r rh r2 )r� r� r� r� r� r� r� r� r r� r& ra r� r� r� rb rO r_ ) r; �ancbuf�maxdataZmindatar� ru r� r� rc rd re rZ r r r �checkTruncatedArray�
s( �
�z!SCMRightsTest.checkTruncatedArrayc C s | j t�d�dd� d S )Nr �r� r� �r� r rK r: r r r �testCmsgTruncLen0�
s z SCMRightsTest.testCmsgTruncLen0c C s | � d� d S r� rs r: r r r �_testCmsgTruncLen0�
s z SCMRightsTest._testCmsgTruncLen0c C s | j t�d�d dd� d S )Nr rh r� r� r: r r r �testCmsgTruncLen0Plus1�
s z$SCMRightsTest.testCmsgTruncLen0Plus1c C s | � d� d S r� rs r: r r r �_testCmsgTruncLen0Plus1�
s z%SCMRightsTest._testCmsgTruncLen0Plus1c C s | j t�t�td� d S )Nr� �r� r rK ri r: r r r �testCmsgTruncLen1�
s �z SCMRightsTest.testCmsgTruncLen1c C s | � d� d S r� rs r: r r r �_testCmsgTruncLen1�
s z SCMRightsTest._testCmsgTruncLen1c C s( | j t�dt �d dt d d� d S )Nr� rh r� r� r: r r r �testCmsgTruncLen2Minus1�
s
�z%SCMRightsTest.testCmsgTruncLen2Minus1c C s | � d� d S r� rs r: r r r �_testCmsgTruncLen2Minus1�
s z&SCMRightsTest._testCmsgTruncLen2Minus1N)rh r )r )r )9r@ rA rB r� r[ r_ r� r` rf rm ro rp rq rt r� r ru r� rv rx ry r� r/ rP ro r rz r~ r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rV d s~
rV 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� rE c C s4 i }|D ]&\}}}| � ||f|� ||||f< q|S r+ )r )r; ru �drc rd re r r r �ancillaryMapping�
s
z%RFC3542AncillaryTest.ancillaryMappingr` r c C s | j �tjtjd� | j�� | �| j tt �|�\}}}}| �
|t � | � || j � | j
|dtj|d� | �
t|�d� | �|d t� |d \}} }
| �
|tj� | �
| tj� | �|
t� | �
t|
�t� t�d�} | �|
� | �| d d� | �| d |� d S )Nrh Trg r r2 )r� r� r r* �IPV6_RECVHOPLIMITr� rf r� r� r� r� r� r� r� r� r rh �
IPV6_HOPLIMITrb ri r� rb r� r� ) r; �
ancbufsize�maxhoprk r� ru r� r� rc rd re r~ r r r �
checkHopLimit�
s2
�
� �
z"RFC3542AncillaryTest.checkHopLimitr� r� c C s | j dd� d S �Nr& �r� )r� r: r r r �testRecvHopLimit s z%RFC3542AncillaryTest.testRecvHopLimitc C s$ | � | jj| jd�� | �t� d S r, �r r� ry r� r� r� r: r r r �_testRecvHopLimit s z&RFC3542AncillaryTest._testRecvHopLimitr� c C s | j t�t�d� d S )Nr� )r� r r� ri r: r r r �testRecvHopLimitCMSG_SPACE s z/RFC3542AncillaryTest.testRecvHopLimitCMSG_SPACEc C s$ | � | jj| jd�� | �t� d S r, r� r: r r r �_testRecvHopLimitCMSG_SPACE s z0RFC3542AncillaryTest._testRecvHopLimitCMSG_SPACErw c C s | j d| jd� d S �Nr& )r� r� )r� � hop_limitr: r r r �testSetHopLimit s z$RFC3542AncillaryTest.testSetHopLimitc C sL | � | jj| jd�� | �| �tgtjtj t
�
d| j g�fg�t t�� d S �Nr r2 )
r r� ry r� r� r� r� r r* r� r� r� r� r: r r r �_testSetHopLimit' s ���z%RFC3542AncillaryTest._testSetHopLimitc C sB | j �tjtjd� | j �tjtjd� | j�� | �| j t t
�|�\}}}}| � |t
� | � || j
� | j|dtj|d� | � t |�d� | �|�}|tjtjf } | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d d� |tjtjf } | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d |� d S )Nrh Trg r� r2 r r` )r� r� r r* r� �IPV6_RECVTCLASSr� rf r� r� r� r� r� r� r� r� r� � IPV6_TCLASSri r� rb r� r� r� ) r; r� r� rk r� ru r� r� ZancmapZtcdatar~ Zhldatar r r � checkTrafficClassAndHopLimit0 s@
�
�
� �
z1RFC3542AncillaryTest.checkTrafficClassAndHopLimitr� r� c C s | j dd� d S r� )r� r: r r r � testRecvTrafficClassAndHopLimitT s z4RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitc C s$ | � | jj| jd�� | �t� d S r, r� r: r r r � _testRecvTrafficClassAndHopLimitZ s z5RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitc C s | j t�t�d d� d S �Nr� r� )r� r r� ri r: r r r �)testRecvTrafficClassAndHopLimitCMSG_SPACE_ s �z>RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitCMSG_SPACEc C s$ | � | jj| jd�� | �t� d S r, r� r: r r r �*_testRecvTrafficClassAndHopLimitCMSG_SPACEg s z?RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitCMSG_SPACEc C s | j d| jd� d S r� �r� r� r: r r r � testSetTrafficClassAndHopLimitl s �z3RFC3542AncillaryTest.testSetTrafficClassAndHopLimitc C sd | � | jj| jd�� | �| �tgtjtj t
�
d| j g�ftjtj t
�
d| j
g�fg�tt�� d S r� )r r� ry r� r� r� r� r r* r� r� �
traffic_classr� r� r� r: r r r � _testSetTrafficClassAndHopLimitu s �����z4RFC3542AncillaryTest._testSetTrafficClassAndHopLimitc C s | j d| jd� d S r� r� r: r r r �testOddCmsgSize� s �z$RFC3542AncillaryTest.testOddCmsgSizec
C s� | � | jj| jd�� zJ| �tgtjtjt � d| j
g�� � d ftjtj t � d| j
g�fg�}W n~ tk
r� } z`| �|jt� | �tgtjtjt � d| j
g�ftjtj t � d| j
g�fg�}| �|tt�� W 5 d }~X Y nX d S )Nr r2 rG )r r� ry r� r� r� r r* r� r� r� rB r� r� r r r; r> r� r� )r; r= r� r r r �_testOddCmsgSize� s2 ��������z%RFC3542AncillaryTest._testOddCmsgSizec C s� | j �tjtjd� | j�� |d kr*dn|f}| j| j tt �f|�� \}}}}| �
|t � | � || j � | �
|g � | j
|dtj|d� d S )Nrh r Tr� )r� r� r r* r� r� rf r� r� r� r� r� r� r� r� )r; r� rk rK r� ru r� r� r r r � checkHopLimitTruncatedHeader� s"
�
�� �z1RFC3542AncillaryTest.checkHopLimitTruncatedHeaderc C s | j d tjd� d S )N�r� rk �r� r r� r: r r r r� � s �z+RFC3542AncillaryTest.testCmsgTruncNoBufSizec C s$ | � | jj| jd�� | �t� d S r, r� r: r r r r� � s z,RFC3542AncillaryTest._testCmsgTruncNoBufSizec C s | j dtjd� d S )Nr r� r� r: r r r �testSingleCmsgTrunc0� s �z)RFC3542AncillaryTest.testSingleCmsgTrunc0c C s$ | � | jj| jd�� | �t� d S r, r� r: r r r �_testSingleCmsgTrunc0� s z*RFC3542AncillaryTest._testSingleCmsgTrunc0c C s | j dd� d S )Nrh r� )r� r: r r r �testSingleCmsgTrunc1� s z)RFC3542AncillaryTest.testSingleCmsgTrunc1c C s$ | � | jj| jd�� | �t� d S r, r� r: r r r �_testSingleCmsgTrunc1� s z*RFC3542AncillaryTest._testSingleCmsgTrunc1c C s | j dt d� d S r� )r� ri r: r r r �testSingleCmsgTrunc2Int� s z,RFC3542AncillaryTest.testSingleCmsgTrunc2Intc C s$ | � | jj| jd�� | �t� d S r, r� r: r r r �_testSingleCmsgTrunc2Int� s z-RFC3542AncillaryTest._testSingleCmsgTrunc2Intc C s | j t�d�d d� d S )Nr rh r� )r� r rK r: r r r � testSingleCmsgTruncLen0Minus1� s z2RFC3542AncillaryTest.testSingleCmsgTruncLen0Minus1c C s$ | � | jj| jd�� | �t� d S r, r� r: r r r � _testSingleCmsgTruncLen0Minus1� s z3RFC3542AncillaryTest._testSingleCmsgTruncLen0Minus1c C s� | j �tjtjd� | j�� | �| j tt �t�
t �d �\}}}}| � |t � | �
|| j� | j|dtjd� | �t|�d� |r�|d \}}}| � |tj� | � |tj� | �t|�t � d S )Nrh Tr� r )r� r� r r* r� r� rf r� r� r� rK ri r� r� r� r� r� r� r� r� r� r r r �testSingleCmsgTruncInData� s&
�
� z.RFC3542AncillaryTest.testSingleCmsgTruncInDatac C s$ | � | jj| jd�� | �t� d S r, r� r: r r r �_testSingleCmsgTruncInData s z/RFC3542AncillaryTest._testSingleCmsgTruncInDatac C s | j �tjtjd� | j �tjtjd� | j�� | �| j t t
�|�\}}}}| � |t
� | � || j
� | j|dtj|d� | � t |�d� |d \}}} | � |tj� | �|tjtjh� | � t | �t� t�d�}
|
�| � | �|
d d� | �|
d d� d S )Nrh Tr� r r2 r` )r� r� r r* r� r� r� rf r� r� r� r� r� r� r� r� r r� r� ri r� rb r� r� ) r; r� rk r� ru r� r� rc rd re r~ r r r �checkTruncatedSecondHeader s6
�
�
� �
z/RFC3542AncillaryTest.checkTruncatedSecondHeaderc C s | j t�t�tjd� d S r� )r� r r� ri r� r: r r r �testSecondCmsgTrunc0) s �z)RFC3542AncillaryTest.testSecondCmsgTrunc0c C s$ | � | jj| jd�� | �t� d S r, r� r: r r r �_testSecondCmsgTrunc0/ s z*RFC3542AncillaryTest._testSecondCmsgTrunc0c C s | � t�t�d � d S r� �r� r r� ri r: r r r �testSecondCmsgTrunc14 s z)RFC3542AncillaryTest.testSecondCmsgTrunc1c C s$ | � | jj| jd�� | �t� d S r, r� r: r r r �_testSecondCmsgTrunc19 s z*RFC3542AncillaryTest._testSecondCmsgTrunc1c C s | � t�t�dt � d S r� r� r: r r r �testSecondCmsgTrunc2Int> s �z,RFC3542AncillaryTest.testSecondCmsgTrunc2Intc C s$ | � | jj| jd�� | �t� d S r, r� r: r r r �_testSecondCmsgTrunc2IntD s z-RFC3542AncillaryTest._testSecondCmsgTrunc2Intc C s" | � t�t�t�d� d � d S r� )r� r r� ri rK r: r r r � testSecondCmsgTruncLen0Minus1I s
��z2RFC3542AncillaryTest.testSecondCmsgTruncLen0Minus1c C s$ | � | jj| jd�� | �t� d S r, r� r: r r r � _testSecondCmsgTruncLen0Minus1O 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 )Nrh Tr� r r2 r` ) r� r� r r* r� r� r� rf r� r� r� r� ri rK r� r� r� r� r� r� r� �pop�remover� rb r� r� r� )
r; r� ru r� r� Z
cmsg_typesrc rd re r~ r r r �testSecomdCmsgTruncInDataT s@
�
�
�
z.RFC3542AncillaryTest.testSecomdCmsgTruncInDatac C s$ | � | jj| jd�� | �t� d S r, r� r: r r r �_testSecomdCmsgTruncInDatay s z/RFC3542AncillaryTest._testSecomdCmsgTruncInDataN)r` r )r` r )r )r )1r@ rA rB r� r� r� r� r� r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� �
s�
�
$ �
�
�
�
�
�
�
�
�
#r� c @ s e Zd ZdS )�SendrecvmsgUDPTestBaseN�r@ rA rB r r r r r� � s r� rw c @ s e Zd ZdS )�SendmsgUDPTestNr� r r r r r� � s r� rt c @ s e Zd ZdS )�RecvmsgUDPTestNr� r r r r r� � s r� r� c @ s e Zd ZdS )�RecvmsgIntoUDPTestNr� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�SendrecvmsgUDP6TestBasec C s | � |d d� |d d� � d S r� r� r� r r r r� � s z+SendrecvmsgUDP6TestBase.checkRecvmsgAddressN)r@ rA rB r� r r r r r� � s r� r� r� r c @ s e Zd ZdS )�SendmsgUDP6TestNr� r r r r r� � s r� c @ s e Zd ZdS )�RecvmsgUDP6TestNr� r r r r r� � s r� c @ s e Zd ZdS )�RecvmsgIntoUDP6TestNr� r r r r r� � s r� r* c @ s e Zd ZdS )� RecvmsgRFC3542AncillaryUDP6TestNr� r r r r r� � s r� c @ s e Zd ZdS )�#RecvmsgIntoRFC3542AncillaryUDP6TestNr� r r r r r� � s r� c @ s e Zd ZdS )�SendrecvmsgTCPTestBaseNr� r r r r r� � s r� c @ s e Zd ZdS )�SendmsgTCPTestNr� r r r r r� � s r� c @ s e Zd ZdS )�RecvmsgTCPTestNr� r r r r r� � s r� c @ s e Zd ZdS )�RecvmsgIntoTCPTestNr� r r r r r� � s r� c @ s e Zd ZdS )� SendrecvmsgSCTPStreamTestBaseNr� r r r r r� � s r� z6IPPROTO_SCTP: [Errno 62] Protocol not supported on AIXr4 r5 r� c @ s e Zd ZdS )�SendmsgSCTPStreamTestNr� r r r r r� � s r� c s e Zd Z� fdd�Z� ZS )�RecvmsgSCTPStreamTestc
sR zt t| ��� W n: tk
rL } z |jtjkr2� | �d� W 5 d }~X Y nX d S �Nz4sporadic ENOTCONN (kernel issue?) - see issue #13876)rE r� r4 r r; �ENOTCONNrZ �r; r� rM r r r4 � s z$RecvmsgSCTPStreamTest.testRecvmsgEOF�r@ rA rB r4 rQ r r rM r r� � s r� c s e Zd Z� fdd�Z� ZS )�RecvmsgIntoSCTPStreamTestc
sR zt t| ��� W n: tk
rL } z |jtjkr2� | �d� W 5 d }~X Y nX d S r� )rE r� r4 r r; r� rZ r� rM r r r4 � s z(RecvmsgIntoSCTPStreamTest.testRecvmsgEOFr� r r rM r r� � s r� c @ s e Zd ZdS )� SendrecvmsgUnixStreamTestBaseNr� r r r r r� � s r� r� c @ s e Zd ZdS )�SendmsgUnixStreamTestNr� r r r r r� � s r� c @ s e Zd ZdS )�RecvmsgUnixStreamTestNr� r r r r r� s r� c @ s e Zd ZdS )�RecvmsgIntoUnixStreamTestNr� r r r r r
s r r& ra c @ s e Zd ZdS )�RecvmsgSCMRightsStreamTestNr� r r r r r s r c @ s e Zd ZdS )� RecvmsgIntoSCMRightsStreamTestNr� 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 r� r )ZsignumrM r r r r� ) r� z.InterruptedTimeoutBase.setUp.<locals>.<lambda>)rE r< r� r� rO )r; Zorig_alrm_handlerrM r r r< & s
�z InterruptedTimeoutBase.setUpg @� setitimerg�������?c C s t �t j|� d S r+ )r� r � ITIMER_REAL�r; Zsecondsr r r �setAlarm5 s z InterruptedTimeoutBase.setAlarmr� c C s t �|� d S r+ )r� r� r r r r r ; s )
r@ rA rB r< r0 r r� �
alarm_timer rQ r r rM r r ! s
r � siginterruptr� r z+Don't have signal.alarm or signal.setitimerc sl e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Ze e
j
d
�dd� �Z e e
j
d�dd� �Z � Z
S )�InterruptedRecvTimeoutTestc s t � �� | j�| j� d S r+ )rE r< r6 r� r0 r: rM r r r< H s
z InterruptedRecvTimeoutTest.setUpc O sB z0| � | j� | �t��}|||� W 5 Q R X W 5 | � d� X d S r� )r r r r� �r; r] rK rL r" r r r �checkInterruptedRecvL s
z/InterruptedRecvTimeoutTest.checkInterruptedRecvc C s | � | jjd� d S r� )r r6 r� r: r r r �testInterruptedRecvTimeoutV s z5InterruptedRecvTimeoutTest.testInterruptedRecvTimeoutc C s | � | jjtd�� d S r� )r r6 � recv_intor� r: r r r � testInterruptedRecvIntoTimeoutY s z9InterruptedRecvTimeoutTest.testInterruptedRecvIntoTimeoutc C s | � | jjd� d S r� )r r6 rO r: r r r � testInterruptedRecvfromTimeout\ s z9InterruptedRecvTimeoutTest.testInterruptedRecvfromTimeoutc C s | � | jjtd�� d S r� )r r6 �
recvfrom_intor� r: r r r �"testInterruptedRecvfromIntoTimeout_ s z=InterruptedRecvTimeoutTest.testInterruptedRecvfromIntoTimeoutrt c C s | � | jjd� d S r� )r r6 rt r: r r r � testInterruptedRecvmsgTimeoutb s z8InterruptedRecvTimeoutTest.testInterruptedRecvmsgTimeoutr� c C s | � | jjtd�g� d S r� )r r6 r� r� r: r r r �!testInterruptedRecvmsgIntoTimeoutf s z<InterruptedRecvTimeoutTest.testInterruptedRecvmsgIntoTimeout)r@ rA rB r< r r
r r r r� r r r rQ r r rM r r
A s
r
c st e Zd Z� fdd�Zdd� Zdd� Ze�dd�d d
� �Ze�dd�d d � �Z e�dd�e
e j d
�dd� ��Z � Z
S )�InterruptedSendTimeoutTestc sp t � �� | �� | _| �| jj� tj| jd�}|� � | j
� � \| _ }| �| j j� |�
� | j�| j� d S )N)�target)rE r< r� r� rO r rG �Thread� doConnect�startr6 r� r� r� r� r0 )r; Z
cli_threadr� rM r r r< v s
z InterruptedSendTimeoutTest.setUpc C s | j �| j� d S r+ )r� r� r� r: r r r r � s z$InterruptedSendTimeoutTest.doConnectc O sD z2| �t�� }| � | j� |||� qW 5 Q R X W 5 | � d� X d S r� )r r r� r r r r r �checkInterruptedSend� s
z/InterruptedSendTimeoutTest.checkInterruptedSendrs rZ c C s | � | jjd� d S r )r r� r� r: r r r �testInterruptedSendTimeout� s z5InterruptedSendTimeoutTest.testInterruptedSendTimeoutc C s | � | jjd| j� d S r )r r� r r� r: r r r � testInterruptedSendtoTimeout� s �z7InterruptedSendTimeoutTest.testInterruptedSendtoTimeoutrw c C s | � | jjdg� d S r )r r� rw r: r r r � testInterruptedSendmsgTimeout� s z8InterruptedSendTimeoutTest.testInterruptedSendmsgTimeout)r@ rA rB r< r r r Zrequires_mac_verr r r� r r rQ r r rM r r m s
r c @ s e Zd Zdd� Zdd� ZdS )�
TCPCloserTestc C sh | j �� \}}|�� | j}t�|gg g d�\}}}| �||g� | �|�d�d� |�� |�� d S )N� �?rh r� )r6 r� r r� r~ r� r� )r; r� r� Zsdr� rW r� r r r � testClose� s zTCPCloserTest.testClosec C s | j �t| jf� t�d� d S �Nr )r� r� r� r8 r� r� r: r r r �
_testClose� s zTCPCloserTest._testCloseN)r@ rA rB r r" r r r r r � s
r c @ sN e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dS )�BasicSocketPairTestr� c C s t j| |d� d S r� )r� rF r� r r r rF � s z BasicSocketPairTest.__init__c C s\ | � |tj� ttd�r*| �|jtj� n| �|jtj� | �|jtj� | �|j d� d S )Nr� r )
r r r r� r r� r4 r� r5 r r� r r r �_check_defaults� s
z#BasicSocketPairTest._check_defaultsc C s | � | j� d S r+ )r$ r� r: r r r �
_testDefaults� s z!BasicSocketPairTest._testDefaultsc C s | � | j� d S r+ )r$ r6 r: r r r � testDefaults� s z BasicSocketPairTest.testDefaultsc C s | j �d�}| �|t� d S r� �r6 r� r� r� r� r r r r� � s z BasicSocketPairTest.testRecvc C s | j �t� d S r+ �r� r� r� r: r r r r� � s z BasicSocketPairTest._testRecvc C s | j �t� d S r+ �r6 r� r� r: r r r �testSend� s z BasicSocketPairTest.testSendc C s | j �d�}| �|t� d S r� �r� r� r� r� r� r r r � _testSend� s z BasicSocketPairTest._testSendN)r� ) r@ rA rB rF r$ r% r& r� r� r* r, r r r r r# � s
r# c s� e Zd Zdd� Z� fdd�Z� fdd�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z e �
eed� preed� d�e �ejd�dd� ��Zdd� Zdd � Zd d � Z� ZS ) �PurePythonSocketPairTestc C s4 zt �t j�W S tk
r. t �t j� Y S X d S r+ )r r� r� r r4 r: r r r r� � s z#PurePythonSocketPairTest.socketpairc s0 t td�r tj| _tjt_nd | _t� �� d S )Nr� )r r r r� �_orig_spZ_fallback_socketpairrE r< r: rM r r r< � s
z PurePythonSocketPairTest.setUpc s t � �� | jd k r | jt_d S r+ )rE r> r. r r� r: rM r r r> � s
z!PurePythonSocketPairTest.tearDownc C s | j �d�}| �|t� d S r� r' r� r r r � test_recv� s z"PurePythonSocketPairTest.test_recvc C s | j �t� d S r+ r( r: r r r �
_test_recv s z#PurePythonSocketPairTest._test_recvc C s | j �t� d S r+ r) r: r r r � test_send s z"PurePythonSocketPairTest.test_sendc C s | j �d�}| �|t� d S r� r+ r� r r r �
_test_send s z#PurePythonSocketPairTest._test_sendc C s$ t �t j�\}}|�� |�� d S r+ )r r� r4 r �r; r� r� r r r � test_ipv4 s z"PurePythonSocketPairTest.test_ipv4c C s d S r+ r r: r r r �
_test_ipv4 s z#PurePythonSocketPairTest._test_ipv4r* � IPV6_V6ONLY� IPV6_V6ONLY option not supported�IPv6 required for this testc C s$ t �t j�\}}|�� |�� d S r+ )r r� r� r r3 r r r � test_ipv6 s z"PurePythonSocketPairTest.test_ipv6c C s d S r+ r r: r r r �
_test_ipv6 s z#PurePythonSocketPairTest._test_ipv6c s� t j j�d � � �fdd�}d }}z.|t j _| �t�� t �� \}}W 5 Q R X W 5 �t j _� rf� �� |rr|�� |r~|�� X d S )Nc sh �| �}� d krdt � t jt j�� � �d� z� �|d d� � W n ttfk
rX Y nX � �d� |S )NFr� T)r r4 r5 r r� �BlockingIOError�InterruptedError)r; r! �Z inject_sockZorig_getsocknamer r �inject_getsocketname# s
z[PurePythonSocketPairTest.test_injected_authentication_failure.<locals>.inject_getsocketname)r r� r r r r� )r; r> Zsock1Zsock2r r= r �$test_injected_authentication_failure s z=PurePythonSocketPairTest.test_injected_authentication_failurec C s d S r+ r r: r r r �%_test_injected_authentication_failure@ s z>PurePythonSocketPairTest._test_injected_authentication_failure)r@ rA rB r� r< r> r/ r0 r1 r2 r4 r5 r� r/ r r r- r r� r9 r: r? r@ rQ r r rM r r- � s$
� !r- c @ s� e Zd Zd&dd�Zdd� Zdd� Zdd � Zejd
d � �Z e�e�Z
e � e
ed �d
�e�ddd�dd� ��Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%S )'�NonBlockingTCPTestsr� c C s t �� | _tj| |d� d S r� )rG rm �eventr� rF r� r r r rF J s
z NonBlockingTCPTests.__init__c C sh | � | j�� |� |dk}| � |�� |� td k rd|d k}t�|tjtj�}| � t|tj@ � |� d S )N� ) r� r6 rt � getblockingr
�F_GETFLr� �
O_NONBLOCK�bool)r; r� r0 ZblockingZ fd_blocking�flagr r r �assert_sock_timeoutN s z'NonBlockingTCPTests.assert_sock_timeoutc C s� | j �d� | �| j d � | j �d� | �| j d� | j �d � | �| j d � | j �d� | �| j d� | j �d� | �| j d� | j �d� | �| j d� d S )NTFrC r rs )r6 r rI r� r: r r r �testSetBlocking_ s z#NonBlockingTCPTests.testSetBlockingc C s d S r+ r r: r r r �_testSetBlockings 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_MAXFrC rh ) rc r� Z ULONG_MAXrZ r6 r r� rt r� r� r r r �testSetBlocking_overflowv s
z,NonBlockingTCPTests.testSetBlocking_overflowr z test needs socket.SOCK_NONBLOCKr� � � c C s4 | j �� t�tjtjtjB �| _ | �| j d� d S r� )r6 r r r4 r5 r rI r: r r r �testInitNonBlocking� s
�z'NonBlockingTCPTests.testInitNonBlockingc C s d S r+ r r: r r r �_testInitNonBlocking� s z(NonBlockingTCPTests._testInitNonBlockingc C sL t d ��: | j�d� | j�� \}}| �|j� | �|�� � W 5 Q R X d S )Nrs )r1 r6 r� r� rO r r� rt r� r r r �testInheritFlagsBlocking� s
z,NonBlockingTCPTests.testInheritFlagsBlockingc C s | j �t| jf� d S r+ �r� r� r� r8 r: r r r �_testInheritFlagsBlocking� s z-NonBlockingTCPTests._testInheritFlagsBlockingc C sR d}t |��< | j�d� | j�� \}}| �|j� | �|�� |� W 5 Q R X d S )Ng 4@rs )r1 r6 r� r� rO r r� rt )r; Zdefault_timeoutr� r� r r r �testInheritFlagsTimeout� s
z+NonBlockingTCPTests.testInheritFlagsTimeoutc C s | j �t| jf� d S r+ rR r: r r r �_testInheritFlagsTimeout� s z,NonBlockingTCPTests._testInheritFlagsTimeoutc C s� | j �d� t�� }| �t�� | j �� \}}W 5 Q R X t�� | }| �|d� | j� � t
�
| j gg g t �\}}}| j |kr�| � d� | j �� \}}| �
|j� | �|�� � d S )Nr r z'Error trying to do accept after select.)r6 r r� � monotonicr r; r� r� rB rf r~ � MAIN_TIMEOUTr rO r r� rt )r; Z
start_timer� r� Zdtr� rW r� r r r �
testAccept� s
z NonBlockingTCPTests.testAcceptc C s | j �� | j�t| jf� d S r+ )rB ry r� r� r� r8 r: r r r � _testAccept� s
z NonBlockingTCPTests._testAcceptc C s� | j �� \}}| �|j� |�d� | �t�� |�tt ��}W 5 Q R X | j
� � t � |gg g t
�\}}}||kr|| �d� |�tt ��}| �|t � d S )Nr z0Error during select call to non-blocking socket.)r6 r� rO r r r r; r� r� r� rB rf r~ rW r r� )r; r� r� r� r� rW r� r r r r� � s
z NonBlockingTCPTests.testRecvc C s, | j �t| jf� | j�� | j �t� d S r+ )r� r� r� r8 rB ry r� r� r: r r r r� � s
z NonBlockingTCPTests._testRecvN)r� )r@ rA rB rF rI rJ rK r r. rL Z_testSetBlocking_overflowr� r- r r �requires_linux_versionrO rP rQ rS rT rU rX rY r� r� r r r r rA H s*
� 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 );�FileObjectClassTestCaserE r �strictNr r� r� c C s t j| |d� d S r� r� r� r r r rF � s z FileObjectClassTestCase.__init__c C sR dd� t d�D �\| _| _| _| _t�| � | jj| j | j
| j | j | j
d�| _d S )Nc S s g | ] }t �� �qS r )rG rm �r� r2 r r r r� s z1FileObjectClassTestCase.setUp.<locals>.<listcomp>r ��encoding�errors�newline)r� �evt1�evt2�
serv_finished� cli_finishedr� r< r� r� � read_moder[ r_ r` ra � read_filer: r r r r< s �
�z FileObjectClassTestCase.setUpc C s6 | j �� | j�� | �| jj� d | _t�| � d S r+ )rd rf rg r r �closedr� r> r: r r r r> s
z FileObjectClassTestCase.tearDownc C s0 t �| � | jj| j| j| j| j| jd�| _ d S )Nr^ )
r� r r� r� �
write_moder[ r_ r` ra �
write_filer: r r r r s
�z#FileObjectClassTestCase.clientSetUpc C s6 | j �� | j�� | �| jj� d | _t�| � d S r+ )re rf rj r r rh r� r� r: r r r r� s
z&FileObjectClassTestCase.clientTearDownc C sd | j �d� | j�d� | �tj| jjd� | �t��}| j�d� W 5 Q R X | �dt |j
�� d S )Nrh r� z!cannot read from timed out object) r� r� rg r� r r r0 r r r� r )r; Zctxr r r �testReadAfterTimeout! s z,FileObjectClassTestCase.testReadAfterTimeoutc C s. | j �| jdd� � | j �� | j�� d S )Nr r� )rj rW � write_msg�flushrd ry r: r r r �_testReadAfterTimeout- s
z-FileObjectClassTestCase._testReadAfterTimeoutc C s< | j �t| j�d �}| j �d�}|| }| �|| j� d S r� )rg r� r� �read_msgr� )r; � first_segZ
second_segr� r r r �
testSmallRead2 s z%FileObjectClassTestCase.testSmallReadc C s | j �| j� | j �� d S r+ �rj rW rl rm r: r r r �_testSmallRead9 s z&FileObjectClassTestCase._testSmallReadc C s | j �� }| �|| j� d S r+ )rg r� r� ro r� r r r � testFullRead= s
z$FileObjectClassTestCase.testFullReadc C s | j �| j� | j �� d S r+ )rj rW rl r r: r r r �
_testFullReadB s z%FileObjectClassTestCase._testFullReadc C s: t | j�� }| j�d�}|s q(||7 }q | �|| j� d S r� )r� ro rg r� r� )r; r� �charr r r �testUnbufferedReadF s
z*FileObjectClassTestCase.testUnbufferedReadc C s | j �| j� | j �� d S r+ rr r: r r r �_testUnbufferedReadP s z+FileObjectClassTestCase._testUnbufferedReadc C s | j �� }| �|| j� d S r+ )rg �readliner� ro �r; �liner r r � testReadlineT s
z$FileObjectClassTestCase.testReadlinec C s | j �| j� | j �� d S r+ rr r: r r r �
_testReadlineY s z%FileObjectClassTestCase._testReadlinec C s& | j �� | j�� }| �|| j� d S r+ )r� r rg r� r� ro r� r r r �testCloseAfterMakefile] s
z.FileObjectClassTestCase.testCloseAfterMakefilec C s | j �| j� | j �� d S r+ rr r: r r r �_testCloseAfterMakefiled s z/FileObjectClassTestCase._testCloseAfterMakefilec C s@ | j �� | j�tt��}t| jt�r.|� � }| �
|| j� d S r+ ) rg r r� r� r� r� r� ro r� �decoder� r� r r r � testMakefileAfterMakefileCloseh s
z6FileObjectClassTestCase.testMakefileAfterMakefileClosec C s | j �| j� | j �� d S r+ rr r: r r r � _testMakefileAfterMakefileCloseo s z7FileObjectClassTestCase._testMakefileAfterMakefileClosec C s | � | jj � d S r+ )r rg rh r: r r r �testClosedAttrs s z&FileObjectClassTestCase.testClosedAttrc C s | � | jj � d S r+ )r rj rh r: r r r �_testClosedAttrv s z'FileObjectClassTestCase._testClosedAttrc C s, | � | jj| j� | � | jj| j�� � d S r+ )r� rg r� rf r� r� r r: r r r �testAttributesy s z&FileObjectClassTestCase.testAttributesc C s, | � | jj| j� | � | jj| j�� � d S r+ )r� rj r� ri r� r� r r: r r r �_testAttributes} s z'FileObjectClassTestCase._testAttributesc C s8 | j �� | �t| j j� | j�� | �t| jj� d S r+ )rg r r ru r r� r r� r: r r r �
testRealClose� s
z%FileObjectClassTestCase.testRealClosec C s d S r+ r r: r r r �_testRealClose� s z&FileObjectClassTestCase._testRealClose)r� )%r@ rA rB r[ r_ r` ra rf r� ro ri rl rF r< r> r r� rk rn rq rs rt ru rw rx r| r} r~ r r� r� r� r� r� r� r� r� r r r r r[ � sB
r[ c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�!UnbufferedFileObjectClassTestCaser c C sL | j �� }| �|d| j � | j�dd�| _ | j �� }| �|d| j � d S )N� A. r r � B. )rg ry r� rl r� r� rz r r r �testUnbufferedReadline� s
z8UnbufferedFileObjectClassTestCase.testUnbufferedReadlinec C s2 | j �d| j � | j �d| j � | j �� d S )Nr� r� rr r: r r r �_testUnbufferedReadline� s z9UnbufferedFileObjectClassTestCase._testUnbufferedReadlinec C sD | j �� | j �d�}| �|| j� | j�� | �t| j jd� d S r� )r� r r� r� ro rg r r r� r r r �testMakefileClose� s
z3UnbufferedFileObjectClassTestCase.testMakefileClosec C s | j �| j� | j �� d S r+ rr r: r r r �_testMakefileClose� s z4UnbufferedFileObjectClassTestCase._testMakefileClosec C s6 t �| j�}| j�� t �| j�}| �|d |� d S r� )rP rO r� rg r r� )r; Zrefcount_beforeZrefcount_afterr r r � testMakefileCloseSocketDestroy� s
z@UnbufferedFileObjectClassTestCase.testMakefileCloseSocketDestroyc C s d S r+ r r: r r r � _testMakefileCloseSocketDestroy� s zAUnbufferedFileObjectClassTestCase._testMakefileCloseSocketDestroyc C s | j �d� | �| j�td��d � | �| j�t| j�d �d � | j �
� | j � d� | j�t| j�d �}|d kr�t
�d� | j�t| j�d �}td�}| j�|�}| �|d� ||d |� }| �|| j� | �| j�td��d � | �| j�d�d � d S )NFrs r� r g �?rV rh )r� r r� rg �readintor� r� r� ro rb rf rc ry r� r� )r; rp r� rR r� r r r �testSmallReadNonBlocking� s
z:UnbufferedFileObjectClassTestCase.testSmallReadNonBlockingc C s> | j �d� | j�| j� | j�� | j�� | j�d� d S )Nr � @) rb ry rj rW rl rm rc rf rd r: r r r �_testSmallReadNonBlocking� s
z;UnbufferedFileObjectClassTestCase._testSmallReadNonBlockingc C s" | j �d� | jr | �| j� d S )Nr� )re ry � serv_skippedrZ r: r r r �testWriteNonBlocking� s z6UnbufferedFileObjectClassTestCase.testWriteNonBlockingc C st d | _ | j�d� dtj }d}| j�|�}| �|d� t|�D ](}| j�|�}|d kr\ qp| �|d� q@d| _ d S )NFr� rs r z$failed to saturate the socket buffer) r� r� r r r� rj rW r? r� )r; ZBIGZLIMITrR r2 r r r �_testWriteNonBlocking� s
z7UnbufferedFileObjectClassTestCase._testWriteNonBlockingN)r@ rA rB r[ r� r� r� r� r� r� r� r� r� r� r r r r r� � s
r� c @ s e Zd ZdZdS )�#LineBufferedFileObjectClassTestCaserh N�r@ rA rB r[ r r r r r� � s r� c @ s e Zd ZdZdS )�$SmallBufferedFileObjectClassTestCaser� Nr� r r r r r� � s r� c @ s& e Zd ZdZe�d�ZdZeZdZ dS )�"UnicodeReadFileObjectClassTestCaser r r� r N�
r@ rA rB rf r� r� ro ri rl ra r r r r r� s
r� c @ s& e Zd ZdZeZdZe�d�ZdZ dS )�#UnicodeWriteFileObjectClassTestCaser r� r r N)
r@ rA rB rf r� ro ri r� rl ra r r r r r� s
r� c @ s, e Zd ZdZe�d�ZdZe�d�ZdZ dS )�'UnicodeReadWriteFileObjectClassTestCaser r r� r Nr� r r r r r� s
r� c @ s e Zd Zdd� ZdS )�NetworkConnectionTestc C s t �t| jf�| _| j| _d S r+ )r �create_connectionr� r8 r� r� r: r r r r " s z!NetworkConnectionTest.clientSetUpN)r@ rA rB r r r r r r� s r� c @ s e Zd ZdS )�
BasicTCPTest2Nr� r r r r r� ( s r� c @ sD e Zd ZG dd� dej�Zejdd� �Zdd� Zdd� Z d d
� Z
d S ) �NetworkConnectionNoServerc @ s e Zd Zdd� ZdS )z$NetworkConnectionNoServer.MockSocketc G s t �d��d S )Nz timed out)r r0 r� r r r r� / s z,NetworkConnectionNoServer.MockSocket.connectN)r@ rA rB r� r r r r �
MockSocket. s r� c c s& t j }| jt _ z
d V W 5 |t _ X d S r+ )r r� )r; Z
old_socketr r r �mocked_socket_module2 s
z.NetworkConnectionNoServer.mocked_socket_modulec C s^ t �� }t�tjtj�}| �|j� | �t��}|� t
|f� W 5 Q R X | � |j j
t
j� d S r+ )r r� r r4 r5 rO r r r r� r� r� r r; Z ECONNREFUSED)r; r8 r� r" r r r � test_connect< s z&NetworkConnectionNoServer.test_connectc C sH t �� }| �t��}t�t|f� W 5 Q R X t �� }| �|j j
|� d S r+ ) r r� r r r r� r� Z get_socket_conn_refused_errsr r r; )r; r8 r" Zexpected_errnosr r r �test_create_connectionD s
z0NetworkConnectionNoServer.test_create_connectionc C s� | � � �n zt�tdf� W nJ tjk
r2 Y n@ tk
rf } ztjsT|jtj krV� W 5 d }~X Y n X | �
d� W 5 Q R X d S )Nr r ) r� r r� r� r0 r r r� r; Z EAFNOSUPPORTr r| r r r � test_create_connection_timeout\ s
z8NetworkConnectionNoServer.test_create_connection_timeoutN) r@ rA rB r r� �
contextlib�contextmanagerr� r� r� r� r r r r r� , s
r� c @ sv e Zd Zddd�Zdd� Zdd� Zdd � ZeZd
d � ZeZ d d
� Z
eZ dd� Z eZ
dd� ZeZdd� ZeZdd� ZdS )� NetworkConnectionAttributesTestr� c C s t j| |d� t�| � d S r� r� r� r r r rF m s z(NetworkConnectionAttributesTest.__init__c C s t �� | _d S r+ )r r� � source_portr: r r r r q s z+NetworkConnectionAttributesTest.clientSetUpc C s | j �� d | _ t�| � d S r+ r� r: r r r r� t s
z.NetworkConnectionAttributesTest.clientTearDownc C s | j �� \}}|�� d S r+ )r6 r� r r� r r r � _justAccepty s z+NetworkConnectionAttributesTest._justAcceptc C s8 t jt| jfdd�| _| �| jj� | �| jjd� d S )N� r r� ) r r� r� r8 r� rO r r� r r: r r r � _testFamily~ s z+NetworkConnectionAttributesTest._testFamilyc C sH t jt| jfdd| jfd�| _| �| jj� | �| j� � d | j� d S )Nr� r )r0 Zsource_addressrh )
r r� r� r8 r� r� rO r r� r� r: r r r �_testSourceAddress� s
�z2NetworkConnectionAttributesTest._testSourceAddressc C sd | � t�� d k� t�d� z$t�t| jf�| _| �| jj � W 5 t�d � X | �
| j� � d� d S )Nr � r r r. r/ r� r� r8 r� rO r r� rt r: r r r �_testTimeoutDefault� s
z3NetworkConnectionAttributesTest._testTimeoutDefaultc C sh | � t�� d k� t�d� z(tjt| jfd d�| _| �| jj � W 5 t�d � X | �
| j� � d � d S �Nr� r r� r: r r r �_testTimeoutNone� s
z0NetworkConnectionAttributesTest._testTimeoutNonec C s, t jt| jfdd�| _| �| j�� d� d S r� )r r� r� r8 r� r� rt r: r r r �_testTimeoutValueNamed� s z6NetworkConnectionAttributesTest._testTimeoutValueNamedc C s8 t �t| jfd�| _| �| jj� | �| j�� d� d S )Nr� ) r r� r� r8 r� rO r r� rt r: r r r �_testTimeoutValueNonamed� s z8NetworkConnectionAttributesTest._testTimeoutValueNonamedN)r� )r@ rA rB rF r r� r� Z
testFamilyr� ZtestSourceAddressr� ZtestTimeoutDefaultr� ZtestTimeoutNoner� ZtestTimeoutValueNamedr� ZtestTimeoutValueNonamedr� r r r r r� k s
r� c @ sB e Zd Zddd�Zdd� Zdd� Zdd � ZeZd
d � Zd d
� Z dS )� NetworkConnectionBehaviourTestr� c C s t j| |d� t�| � d S r� r� r� r r r rF � s z'NetworkConnectionBehaviourTest.__init__c C s d S r+ r r: r r r r � s z*NetworkConnectionBehaviourTest.clientSetUpc C s | j �� d | _ t�| � d S r+ r� r: r r r r� � s
z-NetworkConnectionBehaviourTest.clientTearDownc C s2 | j �� \}}| �|j� t�d� |�d� d S )Nr� � done!)r6 r� rO r r� r� r� r� r r r �testInsideTimeout� s
z0NetworkConnectionBehaviourTest.testInsideTimeoutc C s0 t �t| jf� | _}|�d�}| �|d� d S )Nr� r� )r r� r� r8 r� r� r� )r; r� rJ r r r �_testInsideTimeout� s
z1NetworkConnectionBehaviourTest._testInsideTimeoutc s4 t jt| jfdd� | _� | �t j� fdd�� d S )Nrh r c s
� � d�S )Nr� )r� r �r� r r r� � r� zDNetworkConnectionBehaviourTest._testOutsideTimeout.<locals>.<lambda>)r r� r� r8 r� r r0 r: r r� r �_testOutsideTimeout� s z2NetworkConnectionBehaviourTest._testOutsideTimeoutN)r� )
r@ rA rB rF r r� r� ZtestOutsideTimeoutr� r� r r r r r� � s
r� c @ s6 e Zd Zdd� Zdd� Ze�eed�d�dd� �Z d S )
�TCPTimeoutTestc s � fdd�}� � tj|d� d S )Nc s � j �d� � j �� d S r! )r6 r� r� r� r: r r �
raise_timeout� s z4TCPTimeoutTest.testTCPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (TCP)�r r r0 �r; r� r r: r �testTCPTimeout� s
�z TCPTimeoutTest.testTCPTimeoutc C s| d}z| j �d� | j �� }W nJ tjk
r> | �d� Y n, tk
rT d}Y n | �d� Y nX |sx| �d� d S )NFrC z%caught timeout instead of error (TCP)Tz!caught unexpected exception (TCP)z3accept() returned success when we did not expect it)r6 r� r� r r0 r r �r; �okr% r r r �testTimeoutZero� s z TCPTimeoutTest.testTimeoutZeror� ztest needs signal.alarm()c
s
| j �d� G dd� dt�� � fdd�}t�tj|�}z�z�z�zt�d� | j �� }W n` tjk
rt | � d� Y nL � k
r� Y n: | � d t
� � d d� t �
� f � Y n X | � d
� W 5 t�d� X W n � k
r� | � d � Y nX W 5 t�tj|� X d S ) Nr� c @ s e Zd ZdS )z4TCPTimeoutTest.testInterruptedTimeout.<locals>.AlarmNr� r r r r �Alarm� s r� c s � �d S r+ r )r� rM �r� r r �
alarm_handler� s z<TCPTimeoutTest.testInterruptedTimeout.<locals>.alarm_handlerr r� z caught timeout instead of Alarmz3caught other exception instead of Alarm: %s(%s):
%sznothing caughtzgot Alarm in wrong place)r6 r� r� r� r� r� r� r r0 r rP �exc_info� traceback�
format_exc)r; r� r� r% r r� r �testInterruptedTimeout� s*
� z%TCPTimeoutTest.testInterruptedTimeoutN)
r@ rA rB r� r� r� r- r r� r� r r r r r� � s �r� c @ s e Zd Zdd� Zdd� ZdS )�UDPTimeoutTestc s � fdd�}� � tj|d� d S )Nc s � j �d� � j �d� d S )Nr r� )r6 r� r� r� r: r r r� s z4UDPTimeoutTest.testUDPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (UDP)r� r� r r: r �testUDPTimeout
s
�z UDPTimeoutTest.testUDPTimeoutc C s~ d}z | j �d� | j �d�}W nJ tjk
r@ | �d� Y n, tk
rV d}Y n | �d� Y nX |sz| �d� d S )NFrC r� z%caught timeout instead of error (UDP)Tz!caught unexpected exception (UDP)z1recv() returned success when we did not expect it)r6 r� r� r r0 r r r� r r r r� s z UDPTimeoutTest.testTimeoutZeroN)r@ rA rB r� r� r r r r r� s r� c @ s e Zd Zdd� Zdd� ZdS )�TestExceptionsc C sJ | � ttt�� | � ttjt�� | � ttjt�� | � ttjt�� d S r+ )r �
issubclassr r� r r r r0 r: r r r �testExceptionTree! s z TestExceptions.testExceptionTreec C sb t � t jt jd�}t � t jt jd|�� �}|�� | �|j� | �t�� |� d� W 5 Q R X d S )Nr F)
r r4 r5 r r rO r r r r )r; Zsock0r� r r r �test_setblocking_invalidfd' s � z)TestExceptions.test_setblocking_invalidfdN)r@ rA rB r� r� r r r r r� s r� ri zLinux specific testc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�TestLinuxAbstractNamespace�l c C s� d}t � t jt j��z}|�|� |�� t � t jt j��L}|�|�� � |�� d �&}| �|�� |� | �|� � |� W 5 Q R X W 5 Q R X W 5 Q R X d S )Ns python-test-hello �r )
r r� r5 rX r9 r� r� r� r� Z getpeername)r; r9 �s1r Zs3r r r �testLinuxAbstractNamespace9 s
z5TestLinuxAbstractNamespace.testLinuxAbstractNamespacec C sL dd| j d }t�tjtj�� }|�|� | �|�� |� W 5 Q R X d S )NrG � hrh )�
UNIX_PATH_MAXr r� r5 rX r� r� �r; r9 r r r r � testMaxNameD s
z&TestLinuxAbstractNamespace.testMaxNamec C s> dd| j }t�tjtj��}| �t|j|� W 5 Q R X d S )N� �h)r� r r� r5 r r rX r� r r r �testNameOverflowJ s z+TestLinuxAbstractNamespace.testNameOverflowc C s> t � t jt j�}z |�d� | �|�� d� W 5 |�� X d S )Nz
python test �
python test )r r� r5 r rX r� r� r r r r � testStrNameO s
z&TestLinuxAbstractNamespace.testStrNamec C s> t � t jt j��$}|�td�� | �|�� d� W 5 Q R X d S )Nr� )r r� r5 rX r� r� r� r r r r �testBytearrayNameX s z,TestLinuxAbstractNamespace.testBytearrayNameN) r@ rA rB r� r� r� r� r� r� r r r r r� 4 s r� ztest needs socket.AF_UNIXc @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestUnixDomainc C s t � t jt j�| _d S r+ )r r� r5 r� r: r r r r< a s zTestUnixDomain.setUpc C s | j �� d S r+ )r� r r: r r r r> d s zTestUnixDomain.tearDownc
C s< z t �|�W S tk
r6 | �d�|t�� �� Y nX d S )NzBPathname {0!a} cannot be represented in file system encoding {1!r})r� �fsencoder� rZ r� rP �getfilesystemencoding�r; r� r r r �encodedg s ��zTestUnixDomain.encodedc
C sX zt �||� W nB tk
rR } z$t|�dkr@| �d�|�� n� W 5 d }~X Y nX d S )NzAF_UNIX path too longz6Pathname {0!a} is too long to serve as an AF_UNIX path)r r� r r� rZ r� )r; r� r� r� r r r rX r s ��zTestUnixDomain.bindc C s | � | j�� d� d S )N)r N)r r� r� r: r r r r� ~ s zTestUnixDomain.testUnboundc C s@ t j�tj�}| �| j|� | �tj|� | � | j�
� |� d S r+ ) r� r� �abspathr �TESTFNrX r� rO r� r� r� r� r r r � testStrAddr� s zTestUnixDomain.testStrAddrc C sF t j�tj�}| �| j| �|�� | �tj |� | �
| j� � |� d S r+ ) r� r� r� r r� rX r� r� rO r� r� r� r� r r r �
testBytesAddr� s z TestUnixDomain.testBytesAddrc C sR t j�tj�}| �|�}| �| j|�dd�� | � tj
|� | � | j� � |� d S )N�ascii�surrogateescape)
r� r� r� r ZTESTFN_UNICODEr� rX r� r� rO r� r� r� )r; r� r� r r r �testSurrogateescapeBind� s
z&TestUnixDomain.testSurrogateescapeBindc C sT t jd kr| �d� tj�t j�}| �| j|� | �t j |� | �
| j� � |� d S )Nz!No unencodable filename available) r ZTESTFN_UNENCODABLErZ r� r� r� rX r� rO r� r� r� r� r r r �testUnencodableAddr� s
z"TestUnixDomain.testUnencodableAddrN) r@ rA rB r< r> r� rX r� r� r� r� r� r r r r r� ^ s r� c @ s� e Zd Zddd�Zdd� Zdd� Zdd � ZeZd
d � ZeZ d d
� Z
dd� Z dd� Z e Z
dd� Ze Zdd� Zdd� Zdd� Ze ZdS ) � BufferIOTestr� c C s t j| |d� d S r� r� r� r r r rF � s zBufferIOTest.__init__c C sZ t � ddgtt� �}| j�|�}| �|tt�� |�� }|d tt�� }| �|t� d S �Nr� r )r� r� r� r� r r� rB �r; r� r= r� r r r �testRecvIntoArray� s z BufferIOTest.testRecvIntoArrayc C s t t�}| j�|� d S r+ �rb r� r� r� �r; r� r r r �_testRecvIntoArray� s z BufferIOTest._testRecvIntoArrayc C sD t d�}| j�|�}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r r� r� r� r� r r r �testRecvIntoBytearray� s
z"BufferIOTest.testRecvIntoBytearrayc C sH t d�}| j�t|��}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r rG r� r� r� r� r r r �testRecvIntoMemoryview� s
z#BufferIOTest.testRecvIntoMemoryviewc C s^ t � ddgtt� �}| j�|�\}}| �|tt�� |�� }|d tt�� }| �|t� d S r� )r� r� r� r� r r� rB �r; r� r= r� r� r r r �testRecvFromIntoArray� s z"BufferIOTest.testRecvFromIntoArrayc C s t t�}| j�|� d S r+ r� r� r r r �_testRecvFromIntoArray� s z#BufferIOTest._testRecvFromIntoArrayc C sH t d�}| j�|�\}}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r r� r� r� r� r r r �testRecvFromIntoBytearray� s
z&BufferIOTest.testRecvFromIntoBytearrayc C sL t d�}| j�t|��\}}| �|tt�� |d tt�� }| �|t� d S r� )r� r� r rG r� r� r� r� r r r �testRecvFromIntoMemoryview� s
z'BufferIOTest.testRecvFromIntoMemoryviewc C s t d�}| �t| jj|d� d S )NrV r� )r� r ru r� r r� r r r �testRecvFromIntoSmallBuffer� s z(BufferIOTest.testRecvFromIntoSmallBufferc C s | j �t� d S r+ r� r: r r r � _testRecvFromIntoSmallBuffer� s z)BufferIOTest._testRecvFromIntoSmallBufferc C s$ t � }| j�|� | j�|d� d S r� )r� r� r r� r r r �testRecvFromIntoEmptyBuffer� s z(BufferIOTest.testRecvFromIntoEmptyBufferN)r� )r@ rA rB rF r� r� r� Z_testRecvIntoBytearrayr� Z_testRecvIntoMemoryviewr� r� r� Z_testRecvFromIntoBytearrayr� Z_testRecvFromIntoMemoryviewr� r� r� Z _testRecvFromIntoEmptyBufferr r r r r� � s"
r� i� �� �� c
C sp t td�sdS z td�} W n tttfk
r6 Y dS X | �* | D ] }|�d�rB W 5 Q R � dS qBW 5 Q R X dS )N�AF_TIPCFz
/proc/modulesztipc T)r r r
�FileNotFoundError�IsADirectoryErrorr! rp )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 )Nrh r� r r� )r r r% rO r r� r& r' �TIPC_ADDR_NAMESEQ�
TIPC_STYPE�
TIPC_LOWER�
TIPC_UPPERrX �TIPC_ADDR_NAMEr> r r� rO r� r� )r; r� r� �srvaddrZsendaddrr� Zrecvaddrr r r �testRDM s$ �
� zTIPCTest.testRDMN)r@ rA rB r r r r r r s r c @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )�TIPCThreadableTestr� c C s t jj| |d� t�| � d S r� r� r� r r r rF * s zTIPCThreadableTest.__init__c C s� t � t jt j�| _| �| jj� | j�t jt jd� t j t
t t f}| j�
|� | j�� | �� | j�� \| _| _| �| jj� d S r� )r r r5 r� rO r r� r& r' r r r r rX r9 rg r� r� r� )r; r
r r r r< . s �
zTIPCThreadableTest.setUpc C sd t �d� t�tjtj�| _| �| jj� tjt t
t t t
d � df}| j�
|� | j�� | _d S )Nr� r� r )r� r� r r r5 r� rO r r r r r> r r� r� �cliaddr)r; r� r r r r : s
� z TIPCThreadableTest.clientSetUpc C s, | j �d�}| �|t� | �| j| j� d S r� )r� r� r� r� r
r� r� r r r r� F s z TIPCThreadableTest.testStreamc C s | j �t� | j �� d S r+ r� r: r r r r� K s z TIPCThreadableTest._testStreamN)r� r� r r r r r ' s
r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ContextManagersTestc C s� t � � �}| �|j� W 5 Q R X | �|j� t � � �}|�� W 5 Q R X | �|j� t � � �}| �t|jd� W 5 Q R X | �|j� d S �Nr )r r� r� r r r r r� r� r r r �_testSocketClassR s
z$ContextManagersTest._testSocketClassc C s2 | j �� \}}| �|j� |�d�}|�|� d S r� �r6 r� rO r r� r� �r; r� r� rJ r r r �testCreateConnectionBase` s
z,ContextManagersTest.testCreateConnectionBasec C sX | j �� }t�|��.}| �|j� |�d� | �|�d�d� W 5 Q R X | � |j� d S )Nr r� )
r6 r� r r� r� r� r� r� r� r �r; r9 r� r r r �_testCreateConnectionBasef s
z-ContextManagersTest._testCreateConnectionBasec C s2 | j �� \}}| �|j� |�d�}|�|� d S r� r r r r r �testCreateConnectionClosen s
z-ContextManagersTest.testCreateConnectionClosec C sH | j �� }t�|��}|�� W 5 Q R X | �|j� | �t|j d� d S r )
r6 r� r r� r r r� r r r� r r r r �_testCreateConnectionCloset s
z.ContextManagersTest._testCreateConnectionCloseN)r@ rA rB r r r r r r r r r r P s
r c @ s� e Zd Ze�eed�d�e�ddd�dd� ��Z dd � Z
d
d � Z d d
� Z e�
edkd�dd� �Ze�
edkd�dd� �Zdd� ZdS )�InheritanceTestr� zSOCK_CLOEXEC not definedr� rM rN c C sD t � t jt jt jB ��$}| �|jt j� | �|�� � W 5 Q R X d S r+ )r r4 r5 r� r� r� r� �get_inheritabler r r r �test_SOCK_CLOEXEC} s
�z!InheritanceTest.test_SOCK_CLOEXECc C s, t � � }|� | �|�� d� W 5 Q R X d S r� )r r� r r� r r r �test_default_inheritable� s z(InheritanceTest.test_default_inheritablec
C sL t � � }|�6 |�� }|�� |� | �|�� d� W 5 Q R X W 5 Q R X d S r� )r r� r r� r )r; r� Znewsockr r r �test_dup� s zInheritanceTest.test_dupc C sP t � � }|�: |�d� | �|�� d� |�d� | �|�� d� W 5 Q R X d S )NTF)r �set_inheritabler� r r� r r r �test_set_inheritable� s
z$InheritanceTest.test_set_inheritableNr� c C sn t � � }|�X |�� }| �|�� d� t�|tj�}|tj M }t�|tj|� | �|�� d� W 5 Q R X d S r )r r r� r r
�F_GETFD�
FD_CLOEXECZF_SETFD)r; r� r r� r r r � test_get_inheritable_cloexec� s z,InheritanceTest.test_get_inheritable_cloexecc C sh t � � }|�R |�� }| �t�|tj�tj@ tj� |�d� | �t�|tj�tj@ d� W 5 Q R X d S )NTr )r r r� r
r r r )r; r� r r r r � test_set_inheritable_cloexec� s �
�z,InheritanceTest.test_set_inheritable_cloexecc C sH t �� \}}| �|j� | �|j� | �|�� d� | �|�� d� d S r� )r r� rO r r� r )r; r� r r r r �test_socketpair� s
z InheritanceTest.test_socketpair)r@ rA rB r� r- r r r rZ 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 ) �NonblockConstantTestTrC c C s� |rj| � |jtj� | � |�� |� | �t�|tjtj �tj @ � |dkrZ| �
|� � � q�| �|� � � nL| � |jtj� | � |�� d � | �
t�|tjtj �tj @ � | �|� � � d S r� ) r� r� r r5 rt r r
rE r� rF r� rD )r; r �nonblockr0 r r r �
checkNonblock� s � �z"NonblockConstantTest.checkNonblockr� rM rN c C s^ t � t jt jt jB ���}| �|� |�d� | j|dd� |�d� | �|� |�d � | j|dd� |�d� | j|dd� |�d� | j|dd� W 5 Q R X t �� }t �d� t � � �}| �|� W 5 Q R X t �d � t � � �}| �|d� W 5 Q R X t �d� t � � �}| j|dd� W 5 Q R X t �d � t � � �}| �|d� W 5 Q R X t �|� d S )Nrh F)r% r g @r rC ) r r4 r5 r r& r r� r. r/ )r; r �tr r r �test_SOCK_NONBLOCK� s:
�
z'NonblockConstantTest.test_SOCK_NONBLOCKN)TrC )r@ rA rB r& r rZ r( r r r r r$ � s
r$ r, r- zneed multiprocessingc @ s@ e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
S )�TestSocketSharingc C sD |� � }|� � }t�|�}|�� \}}|�|� |�� |�� d S r+ )r{ r � fromsharer� r� r )rI �qZsdata�messager r r� r r r �remoteProcessServer s
z%TestSocketSharing.remoteProcessServerc C s� t �� }t j| j|fd�}|�� | j�|j�}| j�� }| j� � |�
|� d}|�
|� t � |�}g }|�
d�}|szq�|�|� qj|� � d�|�}| �||� |�� d S )N�r rK s
slapmahfror� r� )�multiprocessingrq ZProcessr- r r6 �share�pidr� r r� r r� r� rY r� r� ) r; r+ r rJ r� r, r �m�receivedr r r � testShare s&
zTestSocketSharing.testSharec C s@ | j �t�� �}| �ttj|d d� � | �ttj|d � d S )NrE r )r6 r0 r� �getpidr ru r r* )r; rJ r r r �testShareLength3 s z!TestSocketSharing.testShareLengthc C sb | � |�� d � | � |�� |�� � | � |j|j� | � |j|j� |jdkr^| � |j|j� d S r� )r� rt r r� r )r; Zorg�otherr r r �compareSockets8 s
z TestSocketSharing.compareSocketsc C s< | j �t�� �}t�|�}z| �| j |� W 5 |�� X d S r+ )r6 r0 r� r5 r r* r r8 )r; rJ r r r r �testShareLocalH s
z TestSocketSharing.testShareLocalc
C s� t jt jg}t jt jg}|D ]|}|D ]r}zt � ||�}W n tk
rP Y q$Y nX z8|�t� � �}t �
|�}z| � ||� W 5 |�� X W 5 |�� X q$q d S r+ ) r r4 r� r5 r r r r0 r� r5 r* r8 )r; Zfamilies�typesr r' �sourcerJ Zsharedr r r � testTypesP s
zTestSocketSharing.testTypesN)
r@ rA rB re r- r4 r6 r8 r9 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 � r_ r� r� c C s~ dd� }d� dd� t| j�D ��}ttjd��$}|| j| j�D ]}|�|� q>W 5 Q R X ttjd��}|�� | _ W 5 Q R X d S )Nc s s& | |kr|V | |8 } q | r"| V d S r+ r )Ztotal�stepr r r �chunksp s
z0SendfileUsingSendTest.setUpClass.<locals>.chunksr� c S s g | ]}t �tj��� �qS r )�randomZchoice�stringZ
ascii_lettersrX r] r r r r� x s �z4SendfileUsingSendTest.setUpClass.<locals>.<listcomp>r� r )
r� r� �BUFSIZEr
r r� �FILESIZErW r� �FILEDATA)rI r? �chunkr Zcsizer r r �
setUpClassn s
�
z SendfileUsingSendTest.setUpClassc C s t �t j� d S r+ )r r� r� )rI r r r �
tearDownClass� s z#SendfileUsingSendTest.tearDownClassc C s6 | j �t� | j �� \}}|�| j� | �|j� |S r+ )r6 r� rW r� �TIMEOUTrO r r� r r r � accept_conn� s
z!SendfileUsingSendTest.accept_connc C s, g }|� | j�}|sq"|�|� qd�|�S r� )r� rB rY r� )r; r� r3 rE r r r � recv_data� s z SendfileUsingSendTest.recv_datac C s
t |d�S )NZ_sendfile_use_send�ru r� r r r �meth_from_sock� s z$SendfileUsingSendTest.meth_from_sockc
C sr | j �� }ttjd�}t�|��H}|�8}| �|�}||�}| �|| j � | �|�
� | j � W 5 Q R X W 5 Q R X d S )Nr � r6 r� r
r r� r r� rL r� rC �tell�r; r9 �filer� �meth�sentr r r �_testRegularFile� s
z&SendfileUsingSendTest._testRegularFilec C s6 | � � }| �|�}| �t|�| j� | �|| j� d S r+ �rI rJ r� r� rC rD �r; r� rJ r r r �testRegularFile� s
z%SendfileUsingSendTest.testRegularFilec
C s| | j �� }t�| j�}t�|��R}|�B}|�|�}| �|| j � | �|�
� | j � | � tj |j
|� W 5 Q R X W 5 Q R X d S r+ )r6 r� �io�BytesIOrD r r� r r� rC rN r r r )r; r9 rP r� rR r r r �_testNonRegularFile� s
�z)SendfileUsingSendTest._testNonRegularFilec C s6 | � � }| �|�}| �t|�| j� | �|| j� d S r+ rT rU r r r �testNonRegularFile� s
z(SendfileUsingSendTest.testNonRegularFilec
C s� | j �� }tjd }t|d�� | �tj|� W 5 Q R X t|d�}t�|��D}|�4}| � |�}||�}| �
|d� | �
|� � d� W 5 Q R X W 5 Q R X d S )N�2r� r r ) r6 r� r r� r
rO r� r r� rL r� rN )r; r9 �filenamerP r� rQ rR r r r �_testEmptyFileSend� s
z(SendfileUsingSendTest._testEmptyFileSendc C s" | � � }| �|�}| �|d� d S r� )rI rJ r� rU r r r �testEmptyFileSend� s
z'SendfileUsingSendTest.testEmptyFileSendc
C sz | j �� }ttjd�}t�|��P}|�@}| �|�}||dd�}| �|| j d � | �|�
� | j � W 5 Q R X W 5 Q R X d S )Nr � )�offsetrM rO r r r � _testOffset� s
z!SendfileUsingSendTest._testOffsetc C sB | � � }| �|�}| �t|�| jd � | �|| jdd � � d S )Nr_ rT rU r r r �
testOffset� s
z SendfileUsingSendTest.testOffsetc
C sz | j �� }ttjd�}tj|dd��L}|�<}d}| �|�}|||d�}| �||� | �|� � |� W 5 Q R X W 5 Q R X d S )Nr r� r �GKL �r� �
r6 r� r
r r� r r� rL r� rN �r; r9 rP r� r� rQ rR r r r �
_testCount� s
z SendfileUsingSendTest._testCountc C s@ d}| � � }| �|�}| �t|�|� | �|| jd |� � d S )Nrc �rI rJ r� r� rD �r; r� r� rJ r r r � testCount� s
z SendfileUsingSendTest.testCountc
C sz | j �� }ttjd�}tj|dd��L}|�<}d}| �|�}|||d�}| �||� | �|� � |� W 5 Q R X W 5 Q R X d S )Nr r� r rh rd re rf r r r �_testCountSmall� s
z%SendfileUsingSendTest._testCountSmallc C s@ d}| � � }| �|�}| �t|�|� | �|| jd |� � d S r� rh ri r r r �testCountSmall s
z$SendfileUsingSendTest.testCountSmallc
C s� | j �� }ttjd�}tj|dd��R}|�B}d}| �|�}||d|d�}| �||� | �|� � |d � W 5 Q R X W 5 Q R X d S )Nr r� r 駆 �� )r` r� re rf r r r �_testCountWithOffset s
z*SendfileUsingSendTest._testCountWithOffsetc C sD d}| � � }| �|�}| �t|�|� | �|| jd|d � � d S )Nrm rn rh ri r r r �testCountWithOffset s
z)SendfileUsingSendTest.testCountWithOffsetc
C sr | j �� }ttjd�}t�|��H}|�8}|�d� | �|�}| � t
||� | � t
|j |� W 5 Q R X W 5 Q R X d S )Nr F) r6 r� r
r r� r r� r rL r ru r �r; r9 rP r� rQ r r r �_testNonBlocking s
z&SendfileUsingSendTest._testNonBlockingc C s | � � }|�d�r | �d� d S )Nr_ z$was not supposed to receive any data)rI r� r �r; r� r r r �testNonBlocking% s
z%SendfileUsingSendTest.testNonBlockingc
C sd | j �� }ttjd�}tj|dd��6}|�&}| �|�}||�}| �|| j � W 5 Q R X W 5 Q R X d S )Nr r� r )
r6 r� r
r r� r r� rL r� rC rO r r r �_testWithTimeout, s
z&SendfileUsingSendTest._testWithTimeoutc C s6 | � � }| �|�}| �t|�| j� | �|| j� d S r+ rT rU r r r �testWithTimeout4 s
z%SendfileUsingSendTest.testWithTimeoutc
C s` | j �� }ttjd��@}t�|��*}|�d� | �|�}| � tj
||� W 5 Q R X W 5 Q R X d S )Nr g{�G�z�?) r6 r� r
r r� r r� r� rL r r0 rq r r r � _testWithTimeoutTriggeredSend< s
z3SendfileUsingSendTest._testWithTimeoutTriggeredSendc C s | � � }|�d� d S )Ni�X )rI r� rs r r r � testWithTimeoutTriggeredSendD s z2SendfileUsingSendTest.testWithTimeoutTriggeredSendc C s d S r+ r r: r r r � _test_errorsJ s z"SendfileUsingSendTest._test_errorsc
C s t tjd��:}tjtjd�� }| �|�}| �td||� W 5 Q R X W 5 Q R X t tjd��4}t�� � }| �|�}| �td||� W 5 Q R X W 5 Q R X t tjd��t}t�� �`}| �|�}| jtd||dd� | jtd||d d� | jtd||d
d� | jtd||d d� W 5 Q R X W 5 Q R X d S ) Nr r� r5 r� z binary modezpositive integerr[ rd r� r rE ) r
r r� r r rL r� ru r� )r; rP r rQ r r r � test_errorsM sR
�
�
� � � �z!SendfileUsingSendTest.test_errorsN)#r@ rA rB rC rB rD rH re rF rG rI rJ rL rS rV rY rZ r] r^ ra rb rg rj rk rl ro rp rr rt ru rv rw rx ry rz r r r r r= d sB
r= r z%os.sendfile() required for this test.c @ s e Zd Zdd� ZdS )�SendfileUsingSendfileTestc C s
t |d�S )Nr rK r� r r r rL k s z(SendfileUsingSendfileTest.meth_from_sockN)r@ rA rB rL r r r r r{ e s r{ zAF_ALG requiredc @ s| e Zd Zdd� Ze�dd�dd� �Zdd� Ze�dd �d
d � �Ze�dd �d
d� �Z e�dd �dd� �Z
dd� Z dd� Z dS )�LinuxKernelCryptoAPIc
C sj t � t jt jd�}z|�||f� W n< tk
r` } z |�� t�t|�||��W 5 d }~X Y nX |S d S r� ) r r' r% rX r r r� ZSkipTestr� )r; �typr� r� r� r r r �
create_algr s $z LinuxKernelCryptoAPI.create_algr r� c
C s� t �d�}| �dd���}|�� \}}|�" |�d� | �|�d�|� W 5 Q R X |�� \}}|�L |�dtj � |�dtj � |�dtj � |�d � | �|�d�|� W 5 Q R X W 5 Q R X d S )
NZ@ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad�hashZsha256s abcrB � a� b� cr� )
rb �fromhexr~ r� r� r� r� r� r �MSG_MORE�r; r �algo�oprj r r r � test_sha256 s
z LinuxKernelCryptoAPI.test_sha256c
C sp t �d�}| �dd��P}|�tjtjd� |�� \}}|�" |�d� | � |�
d�|� W 5 Q R X W 5 Q R X d S )NZ(effcdf6ae5eb2fa2d27416d5f184df9c259a7c79r z
hmac(sha1)s Jefes what do ya want for nothing?rB ) rb r� r~ r� r �SOL_ALG� ALG_SET_KEYr� r� r� r� r� r r r �test_hmac_sha1� s
z#LinuxKernelCryptoAPI.test_hmac_sha1r� c
C s� t �d�}t �d�}d}t �d�}t|�}| �dd���z}|�tjtj|� |�� \}}|�6 |j tj
|tj d� |� |� | �
|�|�|� W 5 Q R X |�� \}}|�, |j |gtj|d� | �
|�|�|� W 5 Q R X d } |g| }
|�� \}}|�& |j |
tj
|d� |�|| �} W 5 Q R X | �
t| �|| � | �
| d |� |� |�� \}}|�( |j | gtj|d� |�|| �} W 5 Q R X | �
t| �|| � | �
| || � W 5 Q R X d S )
NZ 06a9214036b8a15b512e03d534120006Z 3dafba429d9eb430b422da802c9fac41s Single block msgZ e353779c1079aeb82708942dbe77181aZskcipherzcbc(aes))r� �ivr� )r� r� r� )rb r� r� r~ r� r r� r� r� �
sendmsg_afalg�ALG_OP_ENCRYPTr� r� r� r� �ALG_OP_DECRYPT)
r; �keyr� r� Z
ciphertextZmsglenr� r� rj Z
multiplierZlongmsg�encZdecr r r � test_aes_cbc� sR
�
�
� �z!LinuxKernelCryptoAPI.test_aes_cbcrE c C s� t �d�}t �d�}t �d�}t �d�}t �d�}t �d�}t|�}t|�}| �dd���N} | �tjtj|� | �tjtjd |� | � � \}
} |
�v |
j
tj ||tj d � |
�
|tj � |
�
|� |
�|t|� | �} | �|| || � � | �|| | d � � W 5 Q R X | � � \}
} |
�f || }
|
j
|
gtj ||d
� |
�|t|� | �} | �|| || � � | �|| | d � � W 5 Q R X t�d �j}| � � \}
} |
�� || }
|
�|
gtjtj|tj �gtjtj|t|��| gtjtj||�gf� |
�t|
�| �} | �|| || � � | �|| | d � � W 5 Q R X | � � \}
} |
�N || | }
|
j
|
gtj||d
� |
�t|
�| �} | �|| |d � � W 5 Q R X W 5 Q R X d S ) NZ c939cc13397c1d37de6ae0e1cb7c423cZb3d8cc017cbb89b39e0f67e2Z c3b3c41f113a31b73d9a5cd432103069Z 24825602bd12a984e0092d3e448eda5fZ 93fe7d9e9bfd10348a5606e5cafa7354Z 0032a1dc85f1c9786925a2e71d8272ddZaeadzgcm(aes))r� r� �assoclenr� )r� r� r� r )rb r� r� r~ r� r r� r� ZALG_SET_AEAD_AUTHSIZEr� r� r� r� r� r� r� r ZStructrC rw Z
ALG_SET_OPZ
ALG_SET_IVZALG_SET_AEAD_ASSOCLENr� )r; r� r� ZplainZassocZ expected_ctZ expected_tagZtaglenr� r� r� rj �resr� Z pack_uint32r r r �test_aead_aes_gcm� sp
�
�
� �� �z&LinuxKernelCryptoAPI.test_aead_aes_gcmc
C sn | � dd��X}t�d�}|�tjtj|� |�� \}}|� |�d�}| � t
|�d� W 5 Q R X W 5 Q R X d S )N�rngZdrbg_pr_sha256rU ) r~ r� �urandomr� r r� r� r� r� r� r� )r; r� Z
extra_seedr� rj Zrnr r r �test_drbg_pr_sha256 s
z(LinuxKernelCryptoAPI.test_drbg_pr_sha256c
C s� t � t jt jd�}|�� | �t�� |�� W 5 Q R X | �t�� |jd d� W 5 Q R X | �t�� |�d� W 5 Q R X | �t�� |jt jd d� W 5 Q R X | �t�� |jt jdd� W 5 Q R X W 5 Q R X d S )Nr )r� rh )r� r� rE )r r' r% r r� r� r� r� r r r �test_sendmsg_afalg_args s z,LinuxKernelCryptoAPI.test_sendmsg_afalg_argsc C s� t � t jt jd�}| �|j� | �t�� |�d� W 5 Q R X | �t d�� |�d� W 5 Q R X | �t�� |�d� W 5 Q R X | �t d�� |�d� W 5 Q R X d S )Nr )Z
tttttttttttttr� z
type too long)Zttttttttttttttr� )r� Z?nnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnz
name too long)r� Z@nnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnn)
r r' r% rO r r r rX r� ru r� r r r �test_length_restriction! s z,LinuxKernelCryptoAPI.test_length_restrictionN)
r@ rA rB r~ r rZ r� r� r� r� r� r� r� r r r r r| o s
(
>
r| �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)rp �r� r r r r r� C s
z<TestMSWindowsTCPFlags.test_new_tcp_flags.<locals>.<listcomp>c s g | ]}|� j kr|�qS r )�
knownTCPFlagsr� r: r r r� D s
zANew TCP flags were discovered. See bpo-32394 for more information)r� r r� )r; Zprovided�unknownr r: r �test_new_tcp_flagsB s
�z(TestMSWindowsTCPFlags.test_new_tcp_flagsN)r@ rA rB r� r� r r r r r� 3 s �
r� c @ s� e Zd Zdd� Zdd� Zdd� Ze�ee d� p:ee d� d �e�
e j d
�d d � ��Z
e�e�� d
�e�
e j d
�dd� ��ZdS )�CreateServerTestc C s� t �� }t�d|f��.}| �|�� d d� | �|�� d |� W 5 Q R X t jr�tjd|ftjd��.}| �|�� d d� | �|�� d |� W 5 Q R X d S )Nr r rh r� r� )r r� r �
create_serverr� r� r� r� �r; r8 r� r r r � test_addressL s
�z CreateServerTest.test_addressc C s| t �d��&}| �|jt j� | �|jt j� W 5 Q R X tjrxt jdt j d��&}| �|jt j � | �|jt j� W 5 Q R X d S )Nr �r� r r� )
r r� r� r r4 r� r5 r r� r� )r; r� r r r r �test_family_and_typeW s z%CreateServerTest.test_family_and_typec C s� t td�s0| �t�� tjddd� W 5 Q R X nht�d��"}|�tjtj�}| �|d� W 5 Q R X tjddd��"}|�tjtj�}| � |d� W 5 Q R X d S )N� SO_REUSEPORT)r� r T)Z
reuse_portr )
r r r ru r� r� r& r� r� r� )r; r� �optr r r �test_reuse_port` s
z CreateServerTest.test_reuse_portr* r6 r7 r8 c C s t jdt jd��}W 5 Q R X d S )Nr� r� )r r� r� r� r r r �test_ipv6_only_defaultl s z'CreateServerTest.test_ipv6_only_default� dualstack_ipv6 not supportedc C s2 t jdt jdd��}| �|jt j� W 5 Q R X d S )Nr� T�r Zdualstack_ipv6)r r� r� r� r r� r r r �test_dualstack_ipv6_familyt s
�z+CreateServerTest.test_dualstack_ipv6_familyN)r@ rA rB r� r� r� r� r/ r r r- r r� r� r �has_dualstack_ipv6r� r r r r r� J s
� � r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Ze �
e j d �d
d� �Z
e �e�� d�e �
e j d �dd� ��Ze �e�� d�e �
e j d �dd� ��ZdS )�CreateServerFunctionalTestr� c C s
d | _ d S r+ )rv r: r r r r< � s z CreateServerFunctionalTest.setUpc C s | j d k r| j �| j� d S r+ )rv r� r0 r: r r r r> � s
z#CreateServerFunctionalTest.tearDownc sJ � �fdd�}t �� � |��j� t j||fd��_�j�� � �� d S )Nc
sl | �^ | � � \}}|�B � ��j� |�d�}|sJW 5 Q R � W 5 Q R � d S |�|� W 5 Q R X W 5 Q R X d S r� )r� ry r0 r� r� )r� r� rj r� �rB r; r r �run� s
z3CreateServerFunctionalTest.echo_server.<locals>.runr. )rG rm r� r0 r rv r rf )r; r� r� r r� r � echo_server� s
z&CreateServerFunctionalTest.echo_serverc C sN t j |d��8}|�| j� |�|� |�d� | �|�d�d� W 5 Q R X d S )Nr� r r� )r r� r0 r� r� r� r� )r; r� r r� r r r � echo_client� s
z&CreateServerFunctionalTest.echo_clientc C sB t �� }t�d|f��"}| �|� | �d|ftj� W 5 Q R X d S )Nr r )r r� r r� r� r� r4 r� r r r � test_tcp4� s
z$CreateServerFunctionalTest.test_tcp4r8 c C sH t �� }tjd|ftjd��"}| �|� | �d|ftj� W 5 Q R X d S )Nr r� r� �r r� r r� r� r� r� r� r r r � test_tcp6� s
�
z$CreateServerFunctionalTest.test_tcp6r� c C sJ t �� }tjd|ftjdd��"}| �|� | �d|ftj� W 5 Q R X d S )Nr Tr� r )r r� r r� r� r� r� r4 r� r r r �test_dual_stack_client_v4� s �
z4CreateServerFunctionalTest.test_dual_stack_client_v4c C sJ t �� }tjd|ftjdd��"}| �|� | �d|ftj� W 5 Q R X d S )Nr Tr� r� r� r� r r r �test_dual_stack_client_v6� s �
z4CreateServerFunctionalTest.test_dual_stack_client_v6N)r@ rA rB r0 r< r> r� r� r� r� r- r r� r� r/ r r� r� r� r r r r r� } s$
� � r� c C s t ttttttttt t
g } | � t t
ttttttttttttg� | �t� | �t � | �t � | � t t g� | � t t!g� | � t"t#g� | �t$� | �t%� | � t&t'g� | � t(t)t*t+t,t-t.t/t0t1t2t3t4t5t6t7t8t9t:t;t<t=t>t?t@g� | �tA� tB�C� }tBjD| � tBjE|� d S r+ )Fr r� r r� r� r� r� r� r� r� r� �extendrA r[ r� r� r� r� r� r� r� r� r� r r r$ rY r# r� r� r r r0 rF ri rk r| r� r� r� rJ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r
r r) r= r{ r� r Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r � test_main� s� ��
��
r� �__main__)�r� Ztestr r; rW r� r r~ r� r� r� rp rP r� ro r� r� �weakrefr r� r� r� r r@ r rA �_threadrv rG r/ r� r
r� rX r� rW r� �systemr r r r r" r& r( r* r- r� r1 ZHAVE_SOCKET_CANZHAVE_SOCKET_CAN_ISOTPZHAVE_SOCKET_RDSZHAVE_SOCKET_ALGZHAVE_SOCKET_QIPCRTRZHAVE_SOCKET_VSOCKrO ri r� r3 rC rD rR r^ r` r� r� r� r� r/ r- r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r0 rF rf ri rk r� r� r� r� r� r� r� r� r� r� r� r� r r r r3 r8 r; rJ 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 r# r- rA r[ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r$ r� r) r= r{ r| rp r� r� r� r� r@ r r r r �<module> s�
3u� � !$� �$
P
S
Z
*
>��.{7w w= # P; k 7 �
�
�
� �
�
� �
�
�
�
�
� �
�)� �7#j ! $l
?E 9)ES �� ',C�< e � D3G C