Your IP : 216.73.216.213
U
i�f�N � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dl mZ d dl mZ d dlmZ d dl m
Z
d dl m Z
z d dlZW n ek
r� dZY nX dd� ZG d d
� d
e j�ZG d d � d e j�ZG d
d� d�ZG dd� de�ZG dd� de�ZG dd� dee�Zejdk�r4G dd� dee
j�ZG dd� dee
j�Zn|d dl Z e e d��r\G dd � d ee
j�Z e e d ��r|G d d � d ee
j�Z e e d ��r�G d!d"� d"ee
j�Z G d#d� dee
j�ZdS )$� N)� base_events)� constants)�mock)�support)�utilsc C s t �d � d S �N)�asyncioZset_event_loop_policy� r r �D/opt/alt/python38/lib64/python3.8/test/test_asyncio/test_sendfile.py�tearDownModule s r c s> e Zd Zd
dd�Zdd� Zdd� Zd d
� Z� fd d �Z� ZS )�MySendfileProtoNr c C s@ d | _ d| _d| _|d k r.|�� | _|�� | _t� | _|| _d S )NZINITIALr ) � transport�state�nbytes�
create_future� connected�done� bytearray�data� close_after)�self�loopr r r r
�__init__ s
zMySendfileProto.__init__c C s" || _ d| _| jr | j�d � d S )NZ CONNECTED)r
r r �
set_result�r r
r r r
�connection_made% s z MySendfileProto.connection_madec C s
d| _ d S )NZEOF)r �r r r r
� eof_received, s z MySendfileProto.eof_receivedc C s d| _ | jr| j�d � d S )NZCLOSED)r r r �r �excr r r
�connection_lost0 s z MySendfileProto.connection_lostc sJ | j t|�7 _ | j�|� t� �|� | jrF| j | jkrF| j�� d S r ) r �lenr �extend�super�
data_receivedr r
�close�r r �� __class__r r
r$ 6 s
z MySendfileProto.data_received)Nr ) �__name__�
__module__� __qualname__r r r r r$ �
__classcell__r r r' r
r s
r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �MyProtoc C s( d| _ d| _t� | _|�� | _d | _d S �NF)�started�closedr r r �futr
)r r r r r
r A s
zMyProto.__init__c C s d| _ || _d S �NT)r/ r
r r r r
r H s zMyProto.connection_madec C s | j �|� d S r )r r" r& r r r
r$ L s zMyProto.data_receivedc C s d| _ | j�d � d S r2 )r0 r1 r r r r r
r O s zMyProto.connection_lostc � s | j I d H d S r )r1 r r r r
� wait_closedS s zMyProto.wait_closedN)r) r* r+ r r r$ r r3 r r r r
r- ? s
r- c sd e Zd Zdd ZdZdd� Ze� fdd��Ze� fdd ��Z� fd
d �Z � fd d
�Z
dd� Z � Z S )� SendfileBases SendfileBaseDatai � c C s t �d S r )�NotImplementedErrorr r r r
�create_event_loop_ s z SendfileBase.create_event_loopc s2 t tjd��}|�| j� W 5 Q R X t� �� d S )N�wb)�openr �TESTFN�write�DATAr# �
setUpClass)�cls�fpr' r r
r= b s zSendfileBase.setUpClassc s t �t j� t� �� d S r )r �unlinkr: r# �
tearDownClass�r> r' r r
rA h s zSendfileBase.tearDownClassc s@ t tjd�| _| �| jj� | �� | _| �| j� t � �
� d S )N�rb) r9 r r: �file�
addCleanupr% r7 r Zset_event_loopr# �setUpr r' r r
rF m s
zSendfileBase.setUpc s4 | j �� st�| j � | �� t�� t� �� d S r ) r Z is_closed�
test_utilsZ run_brieflyZ
doCleanupsr Z
gc_collectr# �tearDownr r' r r
rH t s
zSendfileBase.tearDownc C s | j �|�S r )r �run_until_complete)r �coror r r
�run_loop} s zSendfileBase.run_loop)
r) r* r+ r< �BUF_SIZEr7 � classmethodr= rA rF rH rK r, r r r' r
r4 W s r4 c st e Zd Ze� fdd��Ze� fdd��Zddd�Zdd � Zdd d �Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
� ZS )�SockSendfileMixinc s t j| _dt _t� �� d S )Ni @ )r �!SENDFILE_FALLBACK_READBUFFER_SIZE� _SockSendfileMixin__old_bufsizer# r= rB r' r r
r= � s z SockSendfileMixin.setUpClassc s | j t_t� �� d S r )rP r rO r# rA rB r' r r
rA � s z SockSendfileMixin.tearDownClassTc C s. t � t jt j�}|�d� |r*| �|j� |S r. )�socket�AF_INET� SOCK_STREAMZ setblockingrE r% )r �cleanup�sockr r r
� make_socket� s
z SockSendfileMixin.make_socketc C s |� tjtj| j� d S r )�
setsockoptrQ �
SOL_SOCKETZ SO_RCVBUFrL )r rU r r r
�reduce_receive_buffer_size� s z,SockSendfileMixin.reduce_receive_buffer_sizeNc C s. |� tjtj| j� |d k r*|j| jd� d S )N)Zhigh)rW rQ rX Z SO_SNDBUFrL Zset_write_buffer_limits)r rU r
r r r
�reduce_send_buffer_size� s z)SockSendfileMixin.reduce_send_buffer_sizec s� t �j�� t�� }�jdd�}|�tj|f� ���jj� fdd�|d����� |� ��� }���j�
|d|f�� �� |� � ��fdd�}�� |� |� fS ) NF)rT c s � S r r r )�protor r
�<lambda>� � z8SockSendfileMixin.prepare_socksendfile.<locals>.<lambda>)rU z 127.0.0.1c s< � j d k r"� j �� ��� �� � ��� ����� � d S r )r
r% rK r3 r �r[ r �serverr r
rT � s
z7SockSendfileMixin.prepare_socksendfile.<locals>.cleanup)
r- r r �find_unused_portrV �bind�HOSTrK �
create_serverrY Z sock_connectrZ rE )r �port�srv_sockrU rT r r^ r
�prepare_socksendfile� s
�
z&SockSendfileMixin.prepare_socksendfilec C sv | � � \}}| �| j�|| j��}|�� | �|�� � | �|t| j �� | �|j
| j � | �| j� � t| j �� d S r ) rf rK r �
sock_sendfilerD r% r3 � assertEqualr! r< r �tell�r rU r[ �retr r r
�test_sock_sendfile_success� s z,SockSendfileMixin.test_sock_sendfile_successc C sv | � � \}}| �| j�|| jdd��}|�� | �|�� � | �|j| j dd� � | �| j�
� d� | �|d� d S )N�� i� i� ) rf rK r rg rD r% r3 rh r r< ri rj r r r
�(test_sock_sendfile_with_offset_and_count� s �z:SockSendfileMixin.test_sock_sendfile_with_offset_and_countc C sp | � � \}}t�� � }| �| j�||dd ��}W 5 Q R X |�� | �|�� � | �|d� | �| j �
� d� d S )Nr ) rf �tempfileZ
TemporaryFilerK r rg r% r3 rh rD ri )r rU r[ �frk r r r
� test_sock_sendfile_zero_size� s
� z.SockSendfileMixin.test_sock_sendfile_zero_sizec C s� dd }| � � \}}| �| j�||�� | �| j�|| j��}| �| j�||�� |�� | �|�� � | �|t | j
�� || j
| }| �|j |� | �| j� � t | j
�� d S )Ns mix_regular_sendr5 )
rf rK r Z sock_sendallrg rD r% r3 rh r! r< r ri )r ZbufrU r[ rk Zexpectedr r r
�(test_sock_sendfile_mix_with_regular_send� s z:SockSendfileMixin.test_sock_sendfile_mix_with_regular_send)T)N)r) r* r+ rM r= rA rV rY rZ rf rl rn rq rr r, r r r' r
rN � s
rN c @ s� e Zd Zddd�dd�Ze�ejdkd�dd � �Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Ze�eed$� d%�d&d'� �Zd(d)� Zd*S )+�
SendfileMixinFr ��is_sslr c s t �� }t�j|d��|r:ts(��d� t�� }t�� }nd }d }t � t j
t j �}|� t j
|f� ���jj�fdd�||d�����|� |r�t j
}nd }t � t j
t j �}|�t j
|f� t�jd�� ���jj� fdd�|||d��\} }
�j|| d � � ���fd
d �} ��| � �� fS ) N)r r z
No ssl modulec s � S r r r )� srv_protor r
r\ r] z0SendfileMixin.prepare_sendfile.<locals>.<lambda>)rU �ssl)r c s � S r r r )� cli_protor r
r\
r] )rU rw �server_hostname)r
c sF �j �� � j �� ���j� ��� j� ��� ����� � d S r )r
r% rK r r3 r �rx r r_ rv r r
rT s
z/SendfileMixin.prepare_sendfile.<locals>.cleanup)r r` r r rw �skipTestrG Zsimple_server_sslcontextZsimple_client_sslcontextrQ rR rS ra rb rK rc rY ZconnectZcreate_connectionrZ rE ) r ru r rd Zsrv_ctxZcli_ctxre ry Zcli_sock�tr�prrT r rz r
�prepare_sendfile� sF �
�
�
z SendfileMixin.prepare_sendfile�win32z UDP sockets are not supportedc C sr | � | jjtjtjd��\}}zD| �t d�� | � | j�
|| j �� W 5 Q R X | � d| j �
� � W 5 |�� X d S )N)Zfamily�
not supportedr )rK r Zcreate_datagram_endpointr ZDatagramProtocolrQ rR r% �assertRaisesRegex� RuntimeError�sendfilerD rh ri )r r| r} r r r
�test_sendfile_not_supported s ���z)SendfileMixin.test_sendfile_not_supportedc C s� | � � \}}| �| j�|j| j��}|j�� | �|j� | �|t | j
�� | �|j t | j
�� | �|j | j
� | �| j�
� t | j
�� d S r �r~ rK r r� r
rD r% r rh r! r< r r ri �r rv rx rk r r r
�
test_sendfile, s �
zSendfileMixin.test_sendfilec s� � � � \}}� fdd�}|� j_� �� j�|j� j��}|j�� � �|j� � � |t
� j �� � � |j t
� j �� � � |j
� j � � � � j�� t
� j �� d S )Nc s t j�� j| |||�S r �r Z
BaseEventLoop�_sendfile_nativer �ZtransprD �offset�countr r r
�sendfile_native: s �zCSendfileMixin.test_sendfile_force_fallback.<locals>.sendfile_native)r~ r r� rK r� r
rD r% r rh r! r< r r ri )r rv rx r� rk r r r
� test_sendfile_force_fallback7 s �
z*SendfileMixin.test_sendfile_force_fallbackc s� t jdkr"t� jtj�r"� �d� � �� \}}� fdd�}|� j_� � tj
d��" � � � jj |j
� jdd�� W 5 Q R X |j
�� � � |j� � �|jd� � �� j�� d� d S ) Nr z Fails on proactor event loopc s t j�� j| |||�S r r� r� r r r
r� P s �zMSendfileMixin.test_sendfile_force_unsupported_native.<locals>.sendfile_nativer� F�Zfallbackr )�sys�platform�
isinstancer r �ProactorEventLoopr{ r~ r� r� ZSendfileNotAvailableErrorrK r� r
rD r% r rh r ri )r rv rx r� r r r
�&test_sendfile_force_unsupported_nativeJ s$
���
z4SendfileMixin.test_sendfile_force_unsupported_nativec C s� | j dd�\}}| �| j�|j| j��}|j�� | �|j� | �|t | j
�� | �|j t | j
�� | �|j | j
� | �| j�
� t | j
�� d S )NT�ru r� r� r r r
�test_sendfile_sslb s �
z SendfileMixin.test_sendfile_sslc C sv | � � \}}|j�� | �td�� | �| j�|j| j�� W 5 Q R X | �|j � | �
|j d� | �
| j� � d� d S )Nz
is closingr )
r~ r
r% r� r� rK r r� rD r rh r ri �r rv rx r r r
� test_sendfile_for_closing_transpm s
" z.SendfileMixin.test_sendfile_for_closing_transpc C s� | � � \}}dd }dd }|j�|� | �| j�|j| j��}|j�|� |j�� | �|j� | � |t
| j �� | � |j || j | � | � | j�
� t
| j �� d S )Ns PREFIX__� s --SUFFIX�r~ r
r; rK r r� rD r% r rh r! r< r ri �r rv rx ZPREFIXZSUFFIXrk r r r
� test_sendfile_pre_and_post_datav s �
z-SendfileMixin.test_sendfile_pre_and_post_datac C s� | j dd�\}}dd }dd }|j�|� | �| j�|j| j��}|j�|� |j�� | �|j� | � |t
| j �� | � |j || j | � | � | j�
� t
| j �� d S )NTr� s zxcvbnmr� s
0987654321r� r� r r r
�#test_sendfile_ssl_pre_and_post_data� s �
z1SendfileMixin.test_sendfile_ssl_pre_and_post_datac C s� | � � \}}| �| j�|j| jdd��}|j�� | �|j� | �|d� | �|j d� | �|j
| j dd� � | �| j� � d� d S )Nrm �d �L �
r~ rK r r� r
rD r% r rh r r r< ri r� r r r
�test_sendfile_partial� s �
z#SendfileMixin.test_sendfile_partialc C s� | j dd�\}}| �| j�|j| jdd��}|j�� | �|j� | �|d� | �|j d� | �|j
| j dd� � | �| j� � d� d S )NTr� rm r� r� r� r� r r r
�test_sendfile_ssl_partial� s �
z'SendfileMixin.test_sendfile_ssl_partialc C s� | j t| j�d�\}}| �| j�|j| j��}|j�� | �|j � | �
|t| j�� | �
|j t| j�� | �
|j | j� | �
| j�
� t| j�� d S )N�r )r~ r! r< rK r r� r
rD r% r rh r r ri r� r r r
�(test_sendfile_close_peer_after_receiving� s �
�
z6SendfileMixin.test_sendfile_close_peer_after_receivingc C s� | j dt| j�d�\}}| �| j�|j| j��}| �|j� | � |t| j�� | � |j
t| j�� | � |j | j� | � | j� � t| j�� d S )NTrt )
r~ r! r< rK r r� r
rD r rh r r ri r� r r r
�,test_sendfile_ssl_close_peer_after_receiving� s �
� z:SendfileMixin.test_sendfile_ssl_close_peer_after_receivingc C s� | j dd�\}}| �t�� | �| j�|j| j�� W 5 Q R X | �|j� | � d|j
koht | j �k n |j
� | � d| j�
� ko�t | j �k n | j�
� � | � |j�� � d S )Nr� r� )r~ � assertRaises�ConnectionErrorrK r r� r
rD r �
assertTruer r! r< ri �
is_closingr� r r r
�3test_sendfile_close_peer_in_the_middle_of_receiving� s � "�&�zASendfileMixin.test_sendfile_close_peer_in_the_middle_of_receivingc s� � fdd�}|� j _� jdd�\}}� �t�� � �� j �|j� j�� W 5 Q R X � �|j � � �
d|j ko|t � j
�k n |j � � �
d� j�� ko�t � j
�k n � j�� � d S )Nc s t j�� j| |||�S r r� r� r r r
r� � s �zcSendfileMixin.test_sendfile_fallback_close_peer_in_the_middle_of_receiving.<locals>.sendfile_nativer� r� )r r� r~ r� r� rK r� r
rD r r� r r! r< ri )r r� rv rx r r r
�<test_sendfile_fallback_close_peer_in_the_middle_of_receiving� s � "�&�zJSendfileMixin.test_sendfile_fallback_close_peer_in_the_middle_of_receivingr� z"Don't have native sendfile supportc s� �� � \}� �j�� �� ��fdd�}�j�|� �}���� ��td�� � j�d� W 5 Q R X ��|�}�� |t
�j �� d S )Nc � s" �� d � �j�� j�j�I d H S r )r r r� r
rD r �rx r1 r r r
rJ � s
z=SendfileMixin.test_sendfile_prevents_bare_write.<locals>.corozsendfile is in progresss data) r~ r r Z create_taskrK r� r� r
r; rh r! r< )r rv rJ �trk r r� r
�!test_sendfile_prevents_bare_write� s
�
z/SendfileMixin.test_sendfile_prevents_bare_writec C sT t �� }dd� |j_tjj|_| �t d�� | j
� | j
j |d dd�� W 5 Q R X d S )Nc S s dS r. r r r r r
r\ � r] zPSendfileMixin.test_sendfile_no_fallback_for_fallback_transport.<locals>.<lambda>zfallback is disabledFr� )
r ZMockr� Z side_effectr Z
_SendfileModeZFALLBACKZ_sendfile_compatibler� r� r rI r� r r r r
�0test_sendfile_no_fallback_for_fallback_transport� s
�z>SendfileMixin.test_sendfile_no_fallback_for_fallback_transportN)r) r* r+ r~ �unittestZskipIfr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �hasattr�osr� r� r r r r
rs � s* +
�
rs c @ s e Zd ZdS )�SendfileTestsBaseN)r) r* r+ r r r r
r� � s r� r c @ s e Zd Zdd� ZdS )�SelectEventLoopTestsc C s t �� S r )r �SelectorEventLoopr r r r
r7 s �&SelectEventLoopTests.create_event_loopN�r) r* r+ r7 r r r r
r� s r� c @ s e Zd Zdd� ZdS )�ProactorEventLoopTestsc C s t �� S r )r r� r r r r
r7
s z(ProactorEventLoopTests.create_event_loopNr� r r r r
r� s r� �KqueueSelectorc @ s e Zd Zdd� ZdS )�KqueueEventLoopTestsc C s t �t�� �S r )r r� � selectorsr� r r r r
r7 s �z&KqueueEventLoopTests.create_event_loopNr� r r r r
r� s r� �
EpollSelectorc @ s e Zd Zdd� ZdS )�EPollEventLoopTestsc C s t �t�� �S r )r r� r� r� r r r r
r7 s z%EPollEventLoopTests.create_event_loopNr� r r r r
r� s r� � PollSelectorc @ s e Zd Zdd� ZdS )�PollEventLoopTestsc C s t �t�� �S r )r r� r� r� r r r r
r7 # s z$PollEventLoopTests.create_event_loopNr� r r r r
r� s r� c @ s e Zd Zdd� ZdS )r� c C s t �t�� �S r )r r� r� ZSelectSelectorr r r r
r7 * s r� Nr� r r r r
r� ' s )!r r� rQ r� ro r� r r r Ztestr Ztest.test_asyncior rG rw � ImportErrorr ZProtocolr r- r4 rN rs r� r� ZTestCaser� r� r� r� r� r� r� r r r r
�<module> s\
&*m � � � � � �