Your IP : 216.73.216.213
a
XC?hlF � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ ddlmZ ejdd� �Ze �eed�d�G d d
� d
e j��Ze �eed�d�G d d � d e��Ze �eed�d�G d
d� de��Ze �eed�d�G dd� de��Ze �eed�d�e �eed�d�G dd� de���Ze �eed�d�G dd� de��ZG dd� de�Z e dk�r�e � � dS )a�
This test suite exercises some system calls subject to interruption with EINTR,
to check that it is actually handled transparently.
It is intended to be run by the main test suite within a child process, to
ensure there is no background thread running (so that signals are delivered to
the correct thread).
Signals are generated in-process using setitimer(ITIMER_REAL), which allows
sub-second periodicity (contrarily to signal()).
� N)�support)�
socket_helperc c sJ | �2 z
| V W n | � � � Y n0 W d � n1 s<0 Y dS )zGContext manager killing the subprocess if a Python exception is raised.N)�kill)�proc� r �@/opt/alt/python39/lib64/python3.9/test/eintrdata/eintr_tester.py�
kill_on_error s
r � setitimerzrequires setitimer()c @ sH e Zd ZdZdZdZdZdd� Zdd� Ze dd � �Z
d
d � Z d d
� Z dS )�
EINTRBaseTestz Base class for EINTR tests. g�������?g�������?c C s | j d7 _ d S �N� )�signals)�self�signum�framer r r �
sighandler2 s zEINTRBaseTest.sighandlerc C sB d| _ t�tj| j�| _t�tj| j| j� t j
ddt j d� d S )Nr iX T)�exit�file)
r
�signal�SIGALRMr � orig_handlerr � ITIMER_REAL� signal_delay�
signal_period� faulthandlerZdump_traceback_later�sys�
__stderr__�r r r r �setUp5 s ��zEINTRBaseTest.setUpc C s t �t jdd� d S �Nr )r r r r r r r �
stop_alarm@ s zEINTRBaseTest.stop_alarmc C s$ | � � t�tj| j� t�� d S �N)r r r r r Zcancel_dump_traceback_laterr r r r �tearDownD s zEINTRBaseTest.tearDownc O s t jdf| }tj|fi |��S )Nz-c)r �
executable�
subprocess�Popen)r �args�kwZcmd_argsr r r r$ I s zEINTRBaseTest.subprocessN)
�__name__�
__module__� __qualname__�__doc__r r �
sleep_timer r � staticmethodr r" r$ r r r r r
& s
r
c @ s| e Zd ZdZdd� Zdd� Zdd� Ze�e e
d�d �d
d � �Z d d
� Z dd� Z
e�e e
d�d�dd� �Zdd� Zdd� ZdS )� OSEINTRTestz EINTR tests for the os module. c C s d| j }| �|�S )Nzimport time; time.sleep(%r))r, r$ )r �coder r r �new_sleep_processR s
z OSEINTRTest.new_sleep_processc sD d}� fdd�t |�D �}t |�D ]
}|� q"|D ] }|�� q2d S )N� c s g | ] }� � � �qS r )r0 )�.0�_r r r �
<listcomp>X � z3OSEINTRTest._test_wait_multiple.<locals>.<listcomp>)�range�wait)r � wait_funcZnumZ processesr3 r r r r �_test_wait_multipleV s z OSEINTRTest._test_wait_multiplec C s | � tj� d S r! )r9 �osr7 r r r r � test_wait_ s zOSEINTRTest.test_wait�wait3zrequires wait3()c C s | � dd� � d S )Nc S s
t �d�S r )r: r< r r r r �<lambda>d r5 z(OSEINTRTest.test_wait3.<locals>.<lambda>)r9 r r r r �
test_wait3b s zOSEINTRTest.test_wait3c C s | � � }||j� |�� d S r! )r0 �pidr7 )r r8 r r r r �_test_wait_singlef s
z OSEINTRTest._test_wait_singlec C s | � dd� � d S )Nc S s t �| d�S r )r: �waitpid�r? r r r r= m r5 z*OSEINTRTest.test_waitpid.<locals>.<lambda>�r@ r r r r � test_waitpidl s zOSEINTRTest.test_waitpid�wait4zrequires wait4()c C s | � dd� � d S )Nc S s t �| d�S r )r: rE rB r r r r= q r5 z(OSEINTRTest.test_wait4.<locals>.<lambda>rC r r r r �
test_wait4o s zOSEINTRTest.test_wait4c C s� t �� \}}| �t j|� g d�}d�dddd| d| j ddd d
d f
�}| j|t|�|gd �}t|��L t �|� |D ] }| � |t �
|t |��� qx| � |� � d
� W d � n1 s�0 Y d S )N)s hellos worlds spam�
zimport os, sys, time� zwr = int(sys.argv[1])�
datas = %r�sleep_time = %rzfor data in datas:z$ # let the parent block on read()� time.sleep(sleep_time)z os.write(wr, data)�Zpass_fdsr )
r: �pipe�
addCleanup�close�joinr, r$ �strr � assertEqual�read�lenr7 )r �rd�wr�datasr/ r �datar r r � test_reads s* �
zOSEINTRTest.test_readc C s� t �� \}}| �t j|� dtj }d�dddd| j dtj ddd d
dd d d
ddddddf�}| j|t |�|gd�}t
|��X t �|� d}|t |�k r�|t � |t
|�|d � �7 }q�| �|�� d� W d � n1 s�0 Y d S )N� xrG zimport io, os, sys, timerH zrd = int(sys.argv[1])rJ zdata = b"x" * %s�data_len = len(data)z!# let the parent block on write()�time.sleep(sleep_time)zread_data = io.BytesIO()z+while len(read_data.getvalue()) < data_len:z% chunk = os.read(rd, 2 * data_len)z read_data.write(chunk)z value = read_data.getvalue()zif value != data:z0 raise Exception("read error: %s vs %s bytes"z- % (len(value), data_len))rL r )r: rM rN rO r Z
PIPE_MAX_SIZErP r, r$ rQ r rT �write�
memoryviewrR r7 )r rU rV rX r/ r �writtenr r r �
test_write� s>
�
zOSEINTRTest.test_writeN)r( r) r* r+ r0 r9 r; �unittest�
skipUnless�hasattrr: r> r@ rD rF rY r` r r r r r. N s
r. c @ s� e Zd ZdZe�eed�d�dd� �Zdd� Z e�eejd�d �d
d � �Z
d d
� Z dd� Z dd� Z
e�eejd�d�dd� �Zdd� Ze�dd�e�eed�d�d d � ��Zd d � Ze�ejd kd!�d"d#� �Zd$d%� Ze�ejd kd!�d&d'� �Zd(S ))�SocketEINTRTestz$ EINTR tests for the socket module. �
socketpairzneeds socketpair()c C s� t �� \}}| �|j� g d�}d�ddddt|j� dt|j� d| d | j dd
d dd d
dddf�}|� � }| j
|t |�|gd�}t |��H |�� |D ]}| �
|||t|��� q�| �
|�� d� W d � n1 s�0 Y d S )N)rZ � y� zrG � import os, socket, sys, timerH �fd = int(sys.argv[1])� family = %s�sock_type = %srI rJ z)wr = socket.fromfd(fd, family, sock_type)� os.close(fd)zwith wr:z for data in datas:z( # let the parent block on recv()z time.sleep(sleep_time)z wr.sendall(data)rL r )�socketre rN rO rP �int�family�typer, �filenor$ rQ r rR rT r7 ) r Z recv_funcrU rV rW r/ �fdr rX r r r �
_test_recv� s8 �
zSocketEINTRTest._test_recvc C s | � tjj� d S r! )rs rm Zrecvr r r r � test_recv� s zSocketEINTRTest.test_recv�recvmsgzneeds recvmsg()c C s | � dd� � d S )Nc S s | � |�d S r )ru ��sockrX r r r r= � r5 z.SocketEINTRTest.test_recvmsg.<locals>.<lambda>)rs r r r r � test_recvmsg� s z SocketEINTRTest.test_recvmsgc
C s( t �� \}}| �|j� dtjd }d�ddddt|j� dt|j � d | j
d
tjd d dd d
dddddddddddddf�}|� � }| j |t
|�|gd�}t|��h |�� d}|t|�k r�||t|�|d � �} || d u r�t|�n| 7 }q�| �|�� d� W d � n1 �s0 Y d S )Ns xyzr1 rG rh rH ri rj rk rJ zdata = b"xyz" * %sr[ z)rd = socket.fromfd(fd, family, sock_type)rl zwith rd:z$ # let the parent block on send()rK z' received_data = bytearray(data_len)z n = 0z while n < data_len:z8 n += rd.recv_into(memoryview(received_data)[n:])zif received_data != data:z0 raise Exception("recv error: %s vs %s bytes"z5 % (len(received_data), data_len))rL r )rm re rN rO r Z
SOCK_MAX_SIZErP rn ro rp r, rq r$ rQ r rT r^ rR r7 )
r Z send_funcrU rV rX r/ rr r r_ Zsentr r r �
_test_send� sL �
zSocketEINTRTest._test_sendc C s | � tjj� d S r! )ry rm �sendr r r r � test_send s zSocketEINTRTest.test_sendc C s | � tjj� d S r! )ry rm Zsendallr r r r � test_sendall s z SocketEINTRTest.test_sendall�sendmsgzneeds sendmsg()c C s | � dd� � d S )Nc S s | � |g�S r! )r} rv r r r r= r5 z.SocketEINTRTest.test_sendmsg.<locals>.<lambda>)ry r r r r � test_sendmsg s z SocketEINTRTest.test_sendmsgc C s� t �tjdf�}| �|j� |�� d }d�dddtj d| d| j dd d
d d f
�}| � |�}t
|��4 |� � \}}|�� | � |�
� d� W d � n1 s�0 Y d S )
Nr r rG zimport socket, timerH z host = %rz port = %srJ z # let parent block on accept()r\ z,with socket.create_connection((host, port)):rK )rm Z
create_serverr ZHOSTrN rO Z getsocknamerP r, r$ r ZacceptrR r7 )r rw Zportr/ r Z client_sockr3 r r r � test_accept s( �
zSocketEINTRTest.test_accept�
r1 �mkfifozneeds mkfifo()c C s� t j}t �|� zt�|� W n2 tyP } z| �d| � W Y d }~n
d }~0 0 | �t j|� d�ddd| d| j dddd|f �}| �
|�}t |��( ||� | � |�
� d � W d � n1 s�0 Y d S )
Nzos.mkfifo(): %srG �import os, timerH z path = %arJ z# let the parent blockr\ r )r �TESTFN�unlinkr: r� �PermissionErrorZskipTestrN rP r, r$ r rR r7 )r Zdo_open_close_readerZdo_open_close_writer�filename�er/ r r r r �
_test_open: s,
$�
zSocketEINTRTest._test_openc C s t |d�}|�� d S )N�w)�openrO )r �path�fpr r r � python_openY s
zSocketEINTRTest.python_open�darwinz+hangs under macOS; see bpo-25234, bpo-35363c C s | � d| j� d S )Nz fp = open(path, 'r')
fp.close())r� r� r r r r � test_open] s �zSocketEINTRTest.test_openc C s t �|t j�}t �|� d S r! )r: r� �O_WRONLYrO )r r� rr r r r �os_openc s zSocketEINTRTest.os_openc C s | � d| j� d S )Nz,fd = os.open(path, os.O_RDONLY)
os.close(fd))r� r� r r r r � test_os_openg s �z SocketEINTRTest.test_os_openN)r( r) r* r+ ra rb rc rm rs rt rx ry r{ r| r~ r r Zrequires_freebsd_versionr: r� r� �skipIfr �platformr� r� r� r r r r rd � s2
#
.
�
�rd c @ s e Zd ZdZdd� ZdS )�
TimeEINTRTestz" EINTR tests for the time module. c C s: t �� }t �| j� | �� t �� | }| �|| j� d S r! )�time� monotonic�sleepr, r �assertGreaterEqual�r �t0�dtr r r �
test_sleepr s
zTimeEINTRTest.test_sleepN)r( r) r* r+ r� r r r r r� n s r� �pthread_sigmaskz need signal.pthread_sigmask()c @ sL e Zd ZdZdd� Ze�eed�d�dd� �Z e�eed�d�d d
� �Z
d S ) �SignalEINTRTestz$ EINTR tests for the signal module. c
C s� t j}t�� }t � |dd� �}| �t j ||� d�ddt�� dt|� d| j dd f�}t �t j |g�}| �t jt j
|g� t � � }| �
|�}t|��$ ||� t � � | } W d � n1 s�0 Y | �|�� d
� d S ) Nc W s d S r! r )r& r r r r= � r5 z/SignalEINTRTest.check_sigwait.<locals>.<lambda>rG r� zpid = %sz signum = %srJ r\ zos.kill(pid, signum)r )r �SIGUSR1r: �getpidrN rP rn r, r� � SIG_BLOCK� SIG_UNBLOCKr� r� r$ r rR r7 )
r r8 r r? Z old_handlerr/ Zold_maskr� r r� r r r �
check_sigwait� s(
�
*z SignalEINTRTest.check_sigwait� sigwaitinfozneed signal.sigwaitinfo()c C s dd� }| � |� d S )Nc S s t �| g� d S r! )r r� �r r r r r8 � s z3SignalEINTRTest.test_sigwaitinfo.<locals>.wait_func�r� �r r8 r r r �test_sigwaitinfo� s z SignalEINTRTest.test_sigwaitinfo� sigtimedwaitc C s dd� }| � |� d S )Nc S s t �| gd� d S )Ng ^@)r r� r� r r r r8 � s z4SignalEINTRTest.test_sigtimedwait.<locals>.wait_funcr� r� r r r �test_sigtimedwait� s z!SignalEINTRTest.test_sigtimedwaitN) r( r) r* r+ r� ra rb rc r r� r� r r r r r� z s �
�r� c @ s� e Zd ZdZdd� Ze�ejdkd�e� e
e d�d�dd � ��Z e� e
e d
�d �d d
� �Z
e� e
e d�d�dd� �Ze� e
e d�d�dd� �ZdS )�SelectEINTRTestz$ EINTR tests for the select module. c C s@ t �� }t�g g g | j� t �� | }| �� | �|| j� d S r! )r� r� �selectr, r r� r� r r r � test_select� s
zSelectEINTRTest.test_selectr� z(poll may fail on macOS; see issue #28087�pollzneed select.pollc C sF t �� }t�� }|�| jd � t�� | }| �� | �|| j� d S �Ng @�@)r� r� r� r� r, r r� �r Zpollerr� r� r r r � test_poll� s zSelectEINTRTest.test_poll�epollzneed select.epollc C sN t �� }| �|j� t�� }|�| j� t�� | }| �� | � || j� d S r! )
r� r� rN rO r� r� r� r, r r� r� r r r �
test_epoll� s zSelectEINTRTest.test_epoll�kqueuezneed select.kqueuec C sR t �� }| �|j� t�� }|�d d| j� t�� | }| �� | � || j� d S r )
r� r� rN rO r� r� Zcontrolr, r r� )r r� r� r� r r r � test_kqueue� s zSelectEINTRTest.test_kqueue�devpollzneed select.devpollc C sR t �� }| �|j� t�� }|�| jd � t�� | }| �� | � || j� d S r� )
r� r� rN rO r� r� r� r, r r� r� r r r � test_devpoll� s z SelectEINTRTest.test_devpollN)r( r) r* r+ r� ra r� r r� rb rc r� r� r� r� r� r r r r r� � s �
r� c @ s8 e Zd Zdd� Ze�e�� dkd�dd� �Zdd� Z d S )
�
FNTLEINTRTestc
C s< | � tjtj� d�ddtj d| d| j f�}t�� }| �|�}t |��� t
tjd���}t�� | }|dkr|t d| ��z,||t j
t jB � ||t j� t�d � W q\ ty� Y q�Y q\0 q\||t j
� t�� | }| �|| j� | �� W d � n1 �s0 Y |�� W d � n1 �s.0 Y d S )
NrG zimport fcntl, timezwith open('%s', 'wb') as f:z fcntl.%s(f, fcntl.LOCK_EX)z time.sleep(%s)�wbg N@z failed to sync child in %.1f secg{�G�z�?)rN r r� r� rP r, r� r� r$ r r� � Exception�fcntlZLOCK_EXZLOCK_NBZLOCK_UNr� �BlockingIOErrorr� r r7 )r Z lock_funcZ lock_namer/ Z
start_timer �fr� r r r �_lock� s2 �
(zFNTLEINTRTest._lockZAIXzAIX returns PermissionErrorc C s | � tjd� d S )N�lockf)r� r� r� r r r r �
test_lockf s zFNTLEINTRTest.test_lockfc C s | � tjd� d S )N�flock)r� r� r� r r r r �
test_flock s zFNTLEINTRTest.test_flockN)
r( r) r* r� ra r� r� �systemr� r� r r r r r� � s
r� �__main__) r+ �
contextlibr r� r: r� r� r rm r$ r r� ra �testr Z test.supportr �contextmanagerr rb rc ZTestCaser
r. rd r� r� r� r� r( �mainr r r r �<module> sF
'h 7 �/8)