Your IP : 216.73.216.213

Current Path : /opt/alt/python36/lib64/python3.6/test/eintrdata/__pycache__/
Upload File :
Current File : //opt/alt/python36/lib64/python3.6/test/eintrdata/__pycache__/eintr_tester.cpython-36.opt-1.pyc

3

� fF�@s|dZddlZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlm
Z
ejdd��Zejeed�d�Gdd	�d	ej��Zejeed�d�Gd
d�de��Zejeed�d�Gdd
�d
e��Zejeed�d�Gdd�de��Zejeed�d�ejeed�d�Gdd�de���Zejeed�d�Gdd�de��ZGdd�de�Zedk�rxej�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)�supportccs6|�(y
|VWn|j��YnXWdQRXdS)zGContext manager killing the subprocess if a Python exception is raised.N)�kill)�proc�r�@/opt/alt/python36/lib64/python3.6/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�
sighandler0szEINTRBaseTest.sighandlercCsLd|_tjtj|j�|_tjtj|j|j�t	t
d�rHt
jddtj
d�dS)Nr�dump_traceback_later�
�<T)�exit�fileiX)r�signal�SIGALRMr�orig_handlerr�ITIMER_REAL�signal_delay�
signal_period�hasattr�faulthandlerr�sys�
__stderr__)rrrr�setUp3s
zEINTRBaseTest.setUpcCstjtjdd�dS)Nr)rrrrrrr�
stop_alarm?szEINTRBaseTest.stop_alarmcCs.|j�tjtj|j�ttd�r*tj�dS)N�cancel_dump_traceback_later)r rrrrrr!)rrrr�tearDownCs
zEINTRBaseTest.tearDowncOstjdf|}tj|f|�S)Nz-c)r�
executable�
subprocess�Popen)r�args�kwZcmd_argsrrrr$IszEINTRBaseTest.subprocessN)
�__name__�
__module__�__qualname__�__doc__rr�
sleep_timerr�staticmethodr r"r$rrrrr	$sr	c@s|eZdZdZdd�Zdd�Zdd�Zeje	e
d�d	�d
d��Zdd
�Zdd�Z
eje	e
d�d�dd��Zdd�Zdd�ZdS)�OSEINTRTestz  EINTR tests for the os module. cCsd|j}|j|�S)Nzimport time; time.sleep(%r))r,r$)r�coderrr�new_sleep_processRs
zOSEINTRTest.new_sleep_processcsLd}�fdd�t|�D�}xt|�D]
}|�q$Wx|D]}|j�q8WdS)N�csg|]}�j��qSr)r0)�.0�_)rrr�
<listcomp>Xsz3OSEINTRTest._test_wait_multiple.<locals>.<listcomp>)�range�wait)r�	wait_funcZnumZ	processesr3rr)rr�_test_wait_multipleVs

zOSEINTRTest._test_wait_multiplecCs|jtj�dS)N)r8�osr6)rrrr�	test_wait_szOSEINTRTest.test_wait�wait3zrequires wait3()cCs|jdd��dS)NcSs
tjd�S)Nr)r9r;rrrr�<lambda>dsz(OSEINTRTest.test_wait3.<locals>.<lambda>)r8)rrrr�
test_wait3bszOSEINTRTest.test_wait3cCs|j�}||j�|j�dS)N)r0�pidr6)rr7rrrr�_test_wait_singlefs
zOSEINTRTest._test_wait_singlecCs|jdd��dS)NcSstj|d�S)Nr)r9�waitpid)r>rrrr<msz*OSEINTRTest.test_waitpid.<locals>.<lambda>)r?)rrrr�test_waitpidlszOSEINTRTest.test_waitpid�wait4zrequires wait4()cCs|jdd��dS)NcSstj|d�S)Nr)r9rB)r>rrrr<qsz(OSEINTRTest.test_wait4.<locals>.<lambda>)r?)rrrr�
test_wait4oszOSEINTRTest.test_wait4cCs�tj�\}}|jtj|�dddg}djdddd|d	|jdd
ddd
f
�}|j|t|�|gd�}t|��Ftj|�x$|D]}|j	|tj
|t|���q|W|j	|j�d�WdQRXdS)Nshellosworldsspam�
zimport os, sys, time�zwr = int(sys.argv[1])z
datas = %rzsleep_time = %rzfor data in datas:z$    # let the parent block on read()z    time.sleep(sleep_time)z    os.write(wr, data))�pass_fdsr)
r9�pipe�
addCleanup�close�joinr,r$�strr�assertEqual�read�lenr6)r�rd�wr�datasr/r�datarrr�	test_readss(



zOSEINTRTest.test_readcCs�tj�\}}|jtj|�dtj}djdddd|jdtjddd	d
dddd
ddddddf�}|j|t	|�|gd�}t
|��Rtj|�d}x,|t|�kr�|tj|t
|�|d��7}q�W|j|j�d�WdQRXdS)N�xrDzimport io, os, sys, timerEzrd = int(sys.argv[1])zsleep_time = %rzdata = b"x" * %szdata_len = len(data)z!# let the parent block on write()ztime.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)))rFr)r9rGrHrIrZ
PIPE_MAX_SIZErJr,r$rKrrN�write�
memoryviewrLr6)rrOrPrRr/r�writtenrrr�
test_write�s<


 zOSEINTRTest.test_writeN)r(r)r*r+r0r8r:�unittest�
skipUnlessrr9r=r?rArCrSrXrrrrr.Ns	r.c@s�eZdZdZejeed�d�dd��Zdd�Z	ejeejd�d	�d
d��Z
dd
�Zdd�Zdd�Z
ejeejd�d�dd��Zdd�Zejdd�ejeed�d�dd���Zdd�Zejejd kd!�d"d#��Zd$d%�Zejejd kd!�d&d'��Zd(S))�SocketEINTRTestz$ EINTR tests for the socket module. �
socketpairzneeds socketpair()c	Cs�tj�\}}|j|j�dddg}djddddt|j�d	t|j�d
|d|jddd
ddddddf�}|j	�}|j
|t|�|gd�}t|��B|j�x"|D]}|j
|||t|���q�W|j
|j�d�WdQRXdS)NrT�y�zrDzimport os, socket, sys, timerEzfd = int(sys.argv[1])zfamily = %szsock_type = %sz
datas = %rzsleep_time = %rz)wr = socket.fromfd(fd, family, sock_type)z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))rFr)�socketr\rHrIrJ�int�family�typer,�filenor$rKrrLrNr6)	rZ	recv_funcrOrPrQr/�fdrrRrrr�
_test_recv�s6


zSocketEINTRTest._test_recvcCs|jtjj�dS)N)rer_Zrecv)rrrr�	test_recv�szSocketEINTRTest.test_recv�recvmsgzneeds recvmsg()cCs|jdd��dS)NcSs|j|�dS)Nr)rg)�sockrRrrrr<�sz.SocketEINTRTest.test_recvmsg.<locals>.<lambda>)re)rrrr�test_recvmsg�szSocketEINTRTest.test_recvmsgc
Cstj�\}}|j|j�dtjd}djddddt|j�dt|j	�d	|j
d
tjddddd
dddddddddddddf�}|j�}|j|t
|�|gd�}t|��b|j�d}x>|t|�kr�||t|�|d��}	||	dkr�t|�n|	7}q�W|j|j�d�WdQRXdS)Nsxyzr1rDzimport os, socket, sys, timerEzfd = int(sys.argv[1])zfamily = %szsock_type = %szsleep_time = %rzdata = b"xyz" * %szdata_len = len(data)z)rd = socket.fromfd(fd, family, sock_type)zos.close(fd)zwith rd:z$    # let the parent block on send()z    time.sleep(sleep_time)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)))rFr)r_r\rHrIrZ
SOCK_MAX_SIZErJr`rarbr,rcr$rKrrNrVrLr6)
rZ	send_funcrOrPrRr/rdrrWZsentrrr�
_test_send�sJ
zSocketEINTRTest._test_sendcCs|jtjj�dS)N)rjr_�send)rrrr�	test_sendszSocketEINTRTest.test_sendcCs|jtjj�dS)N)rjr_Zsendall)rrrr�test_sendallszSocketEINTRTest.test_sendall�sendmsgzneeds sendmsg()cCs|jdd��dS)NcSs|j|g�S)N)rn)rhrRrrrr<sz.SocketEINTRTest.test_sendmsg.<locals>.<lambda>)rj)rrrr�test_sendmsgszSocketEINTRTest.test_sendmsgcCs�tjtjtj�}|j|j�|jtjdf�|j�d}|j	�dj
dddtjd|d|jdd	d
ddf
�}|j|�}t
|��*|j�\}}|j�|j|j�d�WdQRXdS)
Nrr
rDzimport socket, timerEz	host = %rz	port = %szsleep_time = %rz# let parent block on accept()ztime.sleep(sleep_time)z,with socket.create_connection((host, port)):z    time.sleep(sleep_time))r_ZAF_INETZSOCK_STREAMrHrIZbindrZHOSTZgetsocknameZlistenrJr,r$rZacceptrLr6)rrhZportr/rZclient_sockr3rrr�test_accepts*

zSocketEINTRTest.test_acceptrr1�mkfifozneeds mkfifo()cCs�tj}tj|�ytj|�Wn0tk
rN}z|jd|�WYdd}~XnX|jtj|�djddd|d|j	dddd|f	�}|j
|�}t|��||�|j|j
�d	�WdQRXdS)
Nzos.mkfifo(): %srDzimport os, timerEz	path = %azsleep_time = %rz# let the parent blockztime.sleep(sleep_time)r)r�TESTFN�unlinkr9rq�PermissionErrorZskipTestrHrJr,r$rrLr6)rZdo_open_close_readerZdo_open_close_writer�filename�er/rrrr�
_test_open=s*
 

zSocketEINTRTest._test_opencCst|d�}|j�dS)N�w)�openrI)r�path�fprrr�python_open\s
zSocketEINTRTest.python_open�darwinz+hangs under macOS; see bpo-25234, bpo-35363cCs|jd|j�dS)Nzfp = open(path, 'r')
fp.close())rwr|)rrrr�	test_open`szSocketEINTRTest.test_opencCstj|tj�}tj|�dS)N)r9ry�O_WRONLYrI)rrzrdrrr�os_openfszSocketEINTRTest.os_opencCs|jd|j�dS)Nz,fd = os.open(path, os.O_RDONLY)
os.close(fd))rwr�)rrrr�test_os_openjszSocketEINTRTest.test_os_openN)r(r)r*r+rYrZrr_rerfrirjrlrmrorprZrequires_freebsd_versionr9rwr|�skipIfr�platformr~r�r�rrrrr[�s"$.
r[c@seZdZdZdd�ZdS)�
TimeEINTRTestz" EINTR tests for the time module. cCs:tj�}tj|j�|j�tj�|}|j||j�dS)N)�time�	monotonic�sleepr,r �assertGreaterEqual)r�t0�dtrrr�
test_sleepus
zTimeEINTRTest.test_sleepN)r(r)r*r+r�rrrrr�qsr��pthread_sigmaskzneed signal.pthread_sigmask()c@sLeZdZdZdd�Zejeed�d�dd��Z	ejeed�d�d	d
��Z
dS)�SignalEINTRTestz$ EINTR tests for the signal module. c

Cs�tj}tj�}tj|dd��}|jtj||�djddtj�dt|�d|jdd	f�}tjtj	|g�}|jtjtj
|g�tj�}|j
|�}t|��||�tj�|}	WdQRX|j|j�d
�dS)NcWsdS)Nr)r&rrrr<�sz/SignalEINTRTest.check_sigwait.<locals>.<lambda>rDzimport os, timezpid = %szsignum = %szsleep_time = %rztime.sleep(sleep_time)zos.kill(pid, signum)r)r�SIGUSR1r9�getpidrHrJr`r,r��	SIG_BLOCK�SIG_UNBLOCKr�r�r$rrLr6)
rr7r
r>Zold_handlerr/Zold_maskr�rr�rrr�
check_sigwait�s&



zSignalEINTRTest.check_sigwait�sigwaitinfozneed signal.sigwaitinfo()cCsdd�}|j|�dS)NcSstj|g�dS)N)rr�)r
rrrr7�sz3SignalEINTRTest.test_sigwaitinfo.<locals>.wait_func)r�)rr7rrr�test_sigwaitinfo�sz SignalEINTRTest.test_sigwaitinfo�sigtimedwaitcCsdd�}|j|�dS)NcSstj|gd�dS)Ng^@)rr�)r
rrrr7�sz4SignalEINTRTest.test_sigtimedwait.<locals>.wait_func)r�)rr7rrr�test_sigtimedwait�sz!SignalEINTRTest.test_sigtimedwaitN)r(r)r*r+r�rYrZrrr�r�rrrrr�}sr�c@s�eZdZdZdd�Zejejdkd�ej	e
ed�d�dd	���Zej	e
ed
�d�dd
��Z
ej	e
ed�d�dd��Zej	e
ed�d�dd��ZdS)�SelectEINTRTestz$ EINTR tests for the select module. cCs@tj�}tjggg|j�tj�|}|j�|j||j�dS)N)r�r��selectr,r r�)rr�r�rrr�test_select�s
zSelectEINTRTest.test_selectr}z(poll may fail on macOS; see issue #28087�pollzneed select.pollcCsFtj�}tj�}|j|jd�tj�|}|j�|j||j�dS)Ng@�@)r�r�r�r�r,r r�)r�pollerr�r�rrr�	test_poll�szSelectEINTRTest.test_poll�epollzneed select.epollcCsNtj�}|j|j�tj�}|j|j�tj�|}|j�|j	||j�dS)N)
r�r�rHrIr�r�r�r,r r�)rr�r�r�rrr�
test_epoll�szSelectEINTRTest.test_epoll�kqueuezneed select.kqueuecCsRtj�}|j|j�tj�}|jdd|j�tj�|}|j�|j	||j�dS)Nr
)
r�r�rHrIr�r�Zcontrolr,r r�)rr�r�r�rrr�test_kqueue�szSelectEINTRTest.test_kqueue�devpollzneed select.devpollcCsRtj�}|j|j�tj�}|j|jd�tj�|}|j�|j	||j�dS)Ng@�@)
r�r�rHrIr�r�r�r,r r�)rr�r�r�rrr�test_devpoll�szSelectEINTRTest.test_devpollN)r(r)r*r+r�rYr�rr�rZrr�r�r�r�r�rrrrr��s
r�c@s$eZdZdd�Zdd�Zdd�ZdS)�
FNTLEINTRTestcCs|jtjtj�djddtjd|d|jf�}tj�}|j|�}t	|���t
tjd���}xhtj�|}|dkr~td|��y,||tj
tjB�||tj�tjd	�Wq^tk
r�PYq^Xq^W||tj
�tj�|}|j||j�|j�WdQRX|j�WdQRXdS)
NrDzimport 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�?)rHrrsrrrJr,r�r�r$rry�	Exception�fcntlZLOCK_EXZLOCK_NBZLOCK_UNr��BlockingIOErrorr�r r6)rZ	lock_funcZ	lock_namer/Z
start_timer�fr�rrr�_lock�s2

zFNTLEINTRTest._lockcCs|jtjd�dS)N�lockf)r�r�r�)rrrr�
test_lockf	szFNTLEINTRTest.test_lockfcCs|jtjd�dS)N�flock)r�r�r�)rrrr�
test_flockszFNTLEINTRTest.test_flockN)r(r)r*r�r�r�rrrrr��sr��__main__)r+�
contextlibrr�r9r�rr_r$rr�rYZtestr�contextmanagerrrZrZTestCaser	r.r[r�r�r�r�r(�mainrrrr�<module>	s>
)h:
/8&