add-determinism 0.7.3

RPM buildroot helper to strip nondeterministic bits in files
Documentation
f

��_�F�@s�dZddlZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlZddl
mZddlmZddlmZejdd��Ze�eed�d	�Gd
d�dej��Ze�eed�d	�Gdd
�d
e��Ze�eed�d	�Gdd�de��Ze�eed�d	�Gdd�de��Ze�eed�d	�e�eed�d�Gdd�de���Ze�eed�d	�Gdd�de��ZGdd�de�Zedk�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)�	os_helper)�
socket_helperccsJ|�2z
|VWn|���Yn0Wd�n1s<0YdS)zGContext manager killing the subprocess if a Python exception is raised.N)�kill)�proc�r�4/usr/lib64/python3.10/test/eintrdata/eintr_tester.py�
kill_on_errors
r	�	setitimerzrequires setitimer()c@sHeZdZdZdZdZdZdd�Zdd�Ze	dd	��Z
d
d�Zdd
�ZdS)�
EINTRBaseTestz Base class for EINTR tests. g�������?g�������?cCs|jd7_dS�N�)�signals)�self�signum�framerrr�
sighandler3szEINTRBaseTest.sighandlercCsBd|_t�tj|j�|_t�tj|j|j�t	j
ddtjd�dS)NriXT)�exit�file)
r�signal�SIGALRMr�orig_handlerr
�ITIMER_REAL�signal_delay�
signal_period�faulthandlerZdump_traceback_later�sys�
__stderr__�rrrr�setUp6s��zEINTRBaseTest.setUpcCst�tjdd�dS�Nr)rr
rrrrr�
stop_alarmAszEINTRBaseTest.stop_alarmcCs$|��t�tj|j�t��dS�N)r!rrrrZcancel_dump_traceback_laterrrrr�tearDownEszEINTRBaseTest.tearDowncOs tjdf|}tj|fi|��S)Nz-c)r�
executable�
subprocess�Popen)r�args�kwZcmd_argsrrrr%JszEINTRBaseTest.subprocessN)
�__name__�
__module__�__qualname__�__doc__rr�
sleep_timerr�staticmethodr!r#r%rrrrr's
rc@s|eZdZdZdd�Zdd�Zdd�Ze�e	e
d�d	�d
d��Zdd
�Zdd�Z
e�e	e
d�d�dd��Zdd�Zdd�ZdS)�OSEINTRTestz  EINTR tests for the os module. cCsd|j}|�|�S)Nzimport time; time.sleep(%r))r-r%)r�coderrr�new_sleep_processSs
zOSEINTRTest.new_sleep_processcsDd}�fdd�t|�D�}t|�D]
}|�q"|D]}|��q2dS)N�csg|]}����qSr)r1)�.0�_rrr�
<listcomp>Y�z3OSEINTRTest._test_wait_multiple.<locals>.<listcomp>)�range�wait)r�	wait_func�num�	processesr4rrrr�_test_wait_multipleWszOSEINTRTest._test_wait_multiplecCs|�tj�dSr")r<�osr8rrrr�	test_wait`szOSEINTRTest.test_wait�wait3zrequires wait3()cCs|�dd��dS)NcSs
t�d�Sr )r=r?rrrr�<lambda>er6z(OSEINTRTest.test_wait3.<locals>.<lambda>)r<rrrr�
test_wait3cszOSEINTRTest.test_wait3cCs|��}||j�|��dSr")r1�pidr8)rr9rrrr�_test_wait_singlegs
zOSEINTRTest._test_wait_singlecCs|�dd��dS)NcSst�|d�Sr )r=�waitpid�rBrrrr@nr6z*OSEINTRTest.test_waitpid.<locals>.<lambda>�rCrrrr�test_waitpidmszOSEINTRTest.test_waitpid�wait4zrequires wait4()cCs|�dd��dS)NcSst�|d�Sr )r=rHrErrrr@rr6z(OSEINTRTest.test_wait4.<locals>.<lambda>rFrrrr�
test_wait4pszOSEINTRTest.test_wait4cCs�t��\}}|�tj|�gd�}d�dddd|d|jddd	d
df
�}|j|t|�|gd�}t|��Lt�|�|D]}|�	|t�
|t|���qx|�	|��d
�Wd�n1s�0YdS)N)shellosworldsspam�
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)��pass_fdsr)
r=�pipe�
addCleanup�close�joinr-r%�strr	�assertEqual�read�lenr8)r�rd�wr�datasr0r�datarrr�	test_readts*�


zOSEINTRTest.test_readcCs�t��\}}|�tj|�dtj}d�dddd|jdtjddd	d
dddd
ddddddf�}|j|t	|�|gd�}t
|��Xt�|�d}|t|�kr�|t�|t
|�|d��7}q�|�|��d�Wd�n1s�0YdS)N�xrJzimport io, os, sys, timerKzrd = int(sys.argv[1])rMzdata = 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)zvalue = read_data.getvalue()zif value != data:z0    raise Exception("read error: %s vs %s bytes"z-                    % (len(value), data_len))rOr)r=rQrRrSrZ
PIPE_MAX_SIZErTr-r%rUr	rX�write�
memoryviewrVr8)rrYrZr\r0r�writtenrrr�
test_write�s>
�

zOSEINTRTest.test_writeN)r)r*r+r,r1r<r>�unittest�
skipUnless�hasattrr=rArCrGrIr]rdrrrrr/Os	

r/c@s�eZdZdZe�eed�d�dd��Zdd�Z	e�eejd�d	�d
d��Z
dd
�Zdd�Zdd�Z
e�eejd�d�dd��Zdd�Ze�dd�e�eed�d�dd���Zdd�Ze�ejd kd!�d"d#��Zd$d%�Ze�ejd kd!�d&d'��Zd(S))�SocketEINTRTestz$ EINTR tests for the socket module. �
socketpairzneeds socketpair()c	Cs�t��\}}|�|j�gd�}d�ddddt|j�dt|j�d|d	|jdd
dddd
dddf�}|�	�}|j
|t|�|gd�}t|��H|��|D]}|�
|||t|���q�|�
|��d�Wd�n1s�0YdS)N)r^�y�zrJ�import os, socket, sys, timerK�fd = int(sys.argv[1])�family = %s�sock_type = %srLrMz)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)rOr)�socketrirRrSrT�int�family�typer-�filenor%rUr	rVrXr8)	rZ	recv_funcrYrZr[r0�fdrr\rrr�
_test_recv�s8�
zSocketEINTRTest._test_recvcCs|�tjj�dSr")rwrq�recvrrrr�	test_recv�szSocketEINTRTest.test_recv�recvmsgzneeds recvmsg()cCs|�dd��dS)NcSs|�|�dSr )rz��sockr\rrrr@�r6z.SocketEINTRTest.test_recvmsg.<locals>.<lambda>)rwrrrr�test_recvmsg�szSocketEINTRTest.test_recvmsgc
Cs(t��\}}|�|j�dtjd}d�ddddt|j�dt|j	�d	|j
d
tjddddd
dddddddddddddf�}|��}|j|t
|�|gd�}t|��h|��d}|t|�kr�||t|�|d��}	||	dur�t|�n|	7}q�|�|��d�Wd�n1�s0YdS)Nsxyzr2rJrlrKrmrnrorMzdata = b"xyz" * %sr_z)rd = socket.fromfd(fd, family, sock_type)rpzwith rd:z$    # let the parent block on send()rNz'    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))rOr)rqrirRrSrZ
SOCK_MAX_SIZErTrrrsrtr-rur%rUr	rXrbrVr8)
rZ	send_funcrYrZr\r0rvrrc�sentrrr�
_test_send�sL�
zSocketEINTRTest._test_sendcCs|�tjj�dSr")rrq�sendrrrr�	test_sendszSocketEINTRTest.test_sendcCs|�tjj�dSr")rrq�sendallrrrr�test_sendallszSocketEINTRTest.test_sendall�sendmsgzneeds sendmsg()cCs|�dd��dS)NcSs|�|g�Sr")r�r{rrrr@r6z.SocketEINTRTest.test_sendmsg.<locals>.<lambda>)rrrrr�test_sendmsgszSocketEINTRTest.test_sendmsgcCs�t�tjdf�}|�|j�|��d}d�dddtjd|d|jdd	d
ddf
�}|�	|�}t
|��4|��\}}|��|�|�
�d�Wd�n1s�0YdS)
Nrr
rJzimport socket, timerKz	host = %rz	port = %srMz# let parent block on accept()r`z,with socket.create_connection((host, port)):rN)rq�
create_serverrZHOSTrRrS�getsocknamerTr-r%r	�acceptrVr8)rr|�portr0rZclient_sockr4rrr�test_accepts(�

zSocketEINTRTest.test_accept�
r2�mkfifozneeds mkfifo()cCs�tj}t�|�zt�|�Wn2tyP}z|�d|�WYd}~n
d}~00|�tj|�d�ddd|d|j	dddd|f	�}|�
|�}t|��(||�|�|�
�d	�Wd�n1s�0YdS)
Nzos.mkfifo(): %srJ�import os, timerKz	path = %arMz# let the parent blockr`r)r�TESTFN�unlinkr=r��PermissionErrorZskipTestrRrTr-r%r	rVr8)rZdo_open_close_readerZdo_open_close_writer�filename�er0rrrr�
_test_open;s,
$�

zSocketEINTRTest._test_opencCst|d�}|��dS)N�w)�openrS)r�path�fprrr�python_openZs
zSocketEINTRTest.python_open�darwinz+hangs under macOS; see bpo-25234, bpo-35363cCs|�d|j�dS)Nzfp = open(path, 'r')
fp.close())r�r�rrrr�	test_open^s�zSocketEINTRTest.test_opencCst�|tj�}t�|�dSr")r=r��O_WRONLYrS)rr�rvrrr�os_opendszSocketEINTRTest.os_opencCs|�d|j�dS)Nz,fd = os.open(path, os.O_RDONLY)
os.close(fd))r�r�rrrr�test_os_openhs�zSocketEINTRTest.test_os_openN)r)r*r+r,rerfrgrqrwryr}rr�r�r�r�rZrequires_freebsd_versionr=r�r��skipIfr�platformr�r�r�rrrrrh�s2
#
.

�
�rhc@seZdZdZdd�ZdS)�
TimeEINTRTestz" EINTR tests for the time module. cCs:t��}t�|j�|��t��|}|�||j�dSr")�time�	monotonic�sleepr-r!�assertGreaterEqual�r�t0�dtrrr�
test_sleepss
zTimeEINTRTest.test_sleepN)r)r*r+r,r�rrrrr�osr��pthread_sigmaskzneed signal.pthread_sigmask()c@sLeZdZdZdd�Ze�eed�d�dd��Z	e�eed�d�d	d
��Z
dS)�SignalEINTRTestz$ EINTR tests for the signal module. c
Cs�tj}t��}t�|dd��}|�tj||�d�ddt��dt|�d|jdd	f�}t�tj	|g�}|�tjtj
|g�t��}|�
|�}t|��$||�t��|}	Wd�n1s�0Y|�|��d
�dS)NcWsdSr"r)r'rrrr@�r6z/SignalEINTRTest.check_sigwait.<locals>.<lambda>rJr�zpid = %szsignum = %srMr`zos.kill(pid, signum)r)r�SIGUSR1r=�getpidrRrTrrr-r��	SIG_BLOCK�SIG_UNBLOCKr�r�r%r	rVr8)
rr9rrBZold_handlerr0Zold_maskr�rr�rrr�
check_sigwait�s(

�	

*zSignalEINTRTest.check_sigwait�sigwaitinfozneed signal.sigwaitinfo()cCsdd�}|�|�dS)NcSst�|g�dSr")rr��rrrrr9�sz3SignalEINTRTest.test_sigwaitinfo.<locals>.wait_func�r��rr9rrr�test_sigwaitinfo�sz SignalEINTRTest.test_sigwaitinfo�sigtimedwaitcCsdd�}|�|�dS)NcSst�|gd�dS)Ng^@)rr�r�rrrr9�sz4SignalEINTRTest.test_sigtimedwait.<locals>.wait_funcr�r�rrr�test_sigtimedwait�sz!SignalEINTRTest.test_sigtimedwaitN)r)r*r+r,r�rerfrgrr�r�rrrrr�{s�
�r�c@s�eZdZdZdd�Ze�ejdkd�e�	e
ed�d�dd	���Ze�	e
ed
�d�dd
��Z
e�	e
ed�d�dd��Ze�	e
ed�d�dd��ZdS)�SelectEINTRTestz$ EINTR tests for the select module. cCs@t��}t�ggg|j�t��|}|��|�||j�dSr")r�r��selectr-r!r�r�rrr�test_select�s
zSelectEINTRTest.test_selectr�z(poll may fail on macOS; see issue #28087�pollzneed select.pollcCsFt��}t��}|�|jd�t��|}|��|�||j�dS�Ng@�@)r�r�r�r�r-r!r��rZpollerr�r�rrr�	test_poll�szSelectEINTRTest.test_poll�epollzneed select.epollcCsNt��}|�|j�t��}|�|j�t��|}|��|�	||j�dSr")
r�r�rRrSr�r�r�r-r!r�r�rrr�
test_epoll�szSelectEINTRTest.test_epoll�kqueuezneed select.kqueuecCsRt��}|�|j�t��}|�dd|j�t��|}|��|�	||j�dSr)
r�r�rRrSr�r�Zcontrolr-r!r�)rr�r�r�rrr�test_kqueue�szSelectEINTRTest.test_kqueue�devpollzneed select.devpollcCsRt��}|�|j�t��}|�|jd�t��|}|��|�	||j�dSr�)
r�r�rRrSr�r�r�r-r!r�r�rrr�test_devpoll�szSelectEINTRTest.test_devpollN)r)r*r+r,r�rer�rr�rfrgr�r�r�r�r�rrrrr��s�	



r�c@s8eZdZdd�Ze�e��dkd�dd��Zdd�Z	d	S)
�
FNTLEINTRTestc
Cs:|�tjtj�d�ddtjd|d|jf�}t��}|�|�}t	|���t
tjd���}t��|}|dkr|td|��z,||tj
tjB�||tj�t�d	�Wq\ty�Yq�Yq\0||tj
�t��|}|�||j�|��Wd�n1�s0Y|��Wd�n1�s,0YdS)
NrJzimport fcntl, timezwith open('%s', 'wb') as f:z   fcntl.%s(f, fcntl.LOCK_EX)z   time.sleep(%s)�wbgN@z failed to sync child in %.1f secg{�G�z�?)rRrr�r�rTr-r�r�r%r	r��	Exception�fcntl�LOCK_EX�LOCK_NB�LOCK_UNr��BlockingIOErrorr�r!r8)rZ	lock_funcZ	lock_namer0Z
start_timer�fr�rrr�_lock�s2�


(zFNTLEINTRTest._lockZAIXzAIX returns PermissionErrorcCs|�tjd�dS)N�lockf)r�r�r�rrrr�
test_lockf	szFNTLEINTRTest.test_lockfcCs|�tjd�dS)N�flock)r�r�r�rrrr�
test_flock
szFNTLEINTRTest.test_flockN)
r)r*r+r�rer�r��systemr�r�rrrrr��s 
r��__main__) r,�
contextlibrr�r=r�r�rrqr%rr�re�testrZtest.supportrr�contextmanagerr	rfrgZTestCaserr/rhr�r�r�r�r)�mainrrrr�<module>sH


'h7�/8)