Your IP : 216.73.216.213
��YfI? � @ s
d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l m Z e j
d d � � Z e
j e e d � d � Gd d � d e
j � � Z e
j e e d � d � Gd
d � d e � � Z e
j e e d � d � Gd d
� d
e � � Z e
j e e d � d � Gd d � d e � � Z e
j e e d � d � e
j e e d � d � Gd d � d e � � � Z e
j e e d � d � Gd d � d e � � Z d d � Z e d k re � d S)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)�supportc c s3 | �& y | VWn | j � � Yn XWd QRXd S)zGContext manager killing the subprocess if a Python exception is raised.N)�kill)�proc� r �@/opt/alt/python35/lib64/python3.5/test/eintrdata/eintr_tester.py�
kill_on_error s
r � setitimerzrequires setitimer()c @ sj e Z d Z d Z d Z d Z d Z e d d � � Z e d d � � Z e d d � � Z
d
d � Z d S)
�
EINTRBaseTestz Base class for EINTR tests. g�������?g�������?c C s` t j t j d d � � | _ t j t j | j | j � t t d � r\ t j d d d �d S) Nc W s d S)Nr )�argsr r r �<lambda>1 s z*EINTRBaseTest.setUpClass.<locals>.<lambda>�dump_traceback_later�
�< �exitTiX )
�signal�SIGALRM� orig_handlerr � ITIMER_REAL� signal_delay�
signal_period�hasattr� faulthandlerr )�clsr r r �
setUpClass/ s
zEINTRBaseTest.setUpClassc C s t j t j d d � d S)Nr )r r r )r r r r �
stop_alarm9 s zEINTRBaseTest.stop_alarmc C s= | j � t j t j | j � t t d � r9 t j � d S)N�cancel_dump_traceback_later)r r r r r r r )r r r r �
tearDownClass= s
zEINTRBaseTest.tearDownClassc O s# t j d f | } t j | | � S)Nz-c)�sys�
executable�
subprocess�Popen)�selfr
�kwZcmd_argsr r r r D s zEINTRBaseTest.subprocessN) �__name__�
__module__� __qualname__�__doc__r r �
sleep_time� classmethodr r r r r r r r r # s
r c @ s� e Z d Z d Z d d � Z d d � Z d d � Z e j e e
d � d � d
d � � Z d d
� Z d d � Z
e j e e
d � d � d d � � Z d d � Z d d � Z d S)� OSEINTRTestz EINTR tests for the os module. c C s d | j } | j | � S)Nzimport time; time.sleep(%r))r' r )r! �coder r r �new_sleep_processM s
z OSEINTRTest.new_sleep_processc sb d } � f d d � t | � D� } x t | � D]
} | � q2 Wx | D] } | j � qJ Wd S)N� c s g | ] } � j � � q Sr )r+ )�.0�_)r! r r �
<listcomp>S s z3OSEINTRTest._test_wait_multiple.<locals>.<listcomp>)�range�wait)r! � wait_funcZnumZ processesr. r r )r! r �_test_wait_multipleQ s
z OSEINTRTest._test_wait_multiplec C s | j t j � d S)N)r3 �osr1 )r! r r r � test_waitZ s zOSEINTRTest.test_wait�wait3zrequires wait3()c C s | j d d � � d S)Nc S s
t j d � S)Nr )r4 r6 r r r r r _ s z(OSEINTRTest.test_wait3.<locals>.<lambda>)r3 )r! r r r �
test_wait3] s zOSEINTRTest.test_wait3c C s' | j � } | | j � | j � d S)N)r+ �pidr1 )r! r2 r r r r �_test_wait_singlea s
z OSEINTRTest._test_wait_singlec C s | j d d � � d S)Nc S s t j | d � S)Nr )r4 �waitpid)r8 r r r r h s z*OSEINTRTest.test_waitpid.<locals>.<lambda>)r9 )r! r r r � test_waitpidg s zOSEINTRTest.test_waitpid�wait4zrequires wait4()c C s | j d d � � d S)Nc S s t j | d � S)Nr )r4 r< )r8 r r r r l s z(OSEINTRTest.test_wait4.<locals>.<lambda>)r9 )r! r r r �
test_wait4j s zOSEINTRTest.test_wait4c C s� t j � \ } } | j t j | � d d d g } d j d d d d | d | j d d
d d d
f
� } | j | t | � d | g �} t | � �[ t j | � x0 | D]( } | j | t j
| t | � � � q� W| j | j � d � Wd QRXd S)Ns hellos worlds spam�
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 )
r4 �pipe�
addCleanup�close�joinr' r �strr � assertEqual�read�lenr1 )r! �rd�wr�datasr* r �datar r r � test_readn s(
!
&zOSEINTRTest.test_readc C s) t j � \ } } | j t j | � d t j } d j d d d d | j d t j d d d d
d d d d
d d d d d d f � } | j | t | � d | g �} t
| � �m t j | � d } x< | t | � k r| t j | t
| � | d � � 7} q� W| j | j � d � Wd QRXd S)N� xr> zimport io, os, sys, timer? zrd = 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)z value = read_data.getvalue()zif value != data:z0 raise Exception("read error: %s vs %s bytes"z- % (len(value), data_len))r@ r )r4 rA rB rC r Z
PIPE_MAX_SIZErD r' r rE r rH �write�
memoryviewrF r1 )r! rI rJ rL r* r �writtenr r r �
test_write� s<
!
*zOSEINTRTest.test_writeN)r# r$ r% r& r+ r3 r5 �unittest�
skipUnlessr r4 r7 r9 r; r= rM rR r r r r r) I s ' ' r) c @ sl e Z d Z d Z e j e e d � d � d d � � Z d d � Z e j e e j d � d � d
d � � Z
d d
� Z d d � Z d d � Z
e j e e j d � d � d d � � Z d d � Z e j d d � e j e e d � d � d d � � � Z d d � Z d d! � Z e j e j d" k d# � d$ d% � � Z e j e j d" k d# � d& d' � � Z d( S))�SocketEINTRTestz$ EINTR tests for the socket module. �
socketpairzneeds socketpair()c C s* t j � \ } } | j | j � d d d g } d j d d d d t | j � d t | j � d
| d | j d d d
d d d d d d f � } | j � } | j
| t | � d | g �} t | � �U | j � x- | D]% } | j
| | | t | � � � q� W| j
| j � d � Wd QRXd S)NrN � y� zr> z import os, socket, sys, timer? zfd = int(sys.argv[1])z family = %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)r@ r )�socketrV rB rC rD �int�family�typer' �filenor rE r rF rH r1 ) r! Z recv_funcrI rJ rK r* �fdr rL r r r �
_test_recv� s6
!
#zSocketEINTRTest._test_recvc C s | j t j j � d S)N)r_ rY Zrecv)r! r r r � test_recv� s zSocketEINTRTest.test_recv�recvmsgzneeds recvmsg()c C s | j d d � � d S)Nc S s | j | � d S)Nr )ra )�sockrL r r r r � s z.SocketEINTRTest.test_recvmsg.<locals>.<lambda>)r_ )r! r r r � test_recvmsg� s z SocketEINTRTest.test_recvmsgc
C s{ t j � \ } } | j | j � d t j d } d j d d d d t | j � d t | j � d | j
d
t j d d d d d
d d d d d d d d d d d d d f � } | j � } | j | t
| � d | g �} t | � �� | j � d } xW | t | � k rY| | t | � | d � � } | | d k rOt | � n | 7} qW| j | j � d � Wd QRXd S)Ns xyzr, r> z import os, socket, sys, timer? zfd = int(sys.argv[1])z family = %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))r@ r )rY rV rB rC r Z
SOCK_MAX_SIZErD rZ r[ r\ r' r] r rE r rH rP rF r1 )
r! Z send_funcrI rJ rL r* r^ r rQ Zsentr r r �
_test_send� sJ
!
&zSocketEINTRTest._test_sendc C s | j t j j � d S)N)rd rY �send)r! r r r � test_send s zSocketEINTRTest.test_sendc C s | j t j j � d S)N)rd rY Zsendall)r! r r r � test_sendall s z SocketEINTRTest.test_sendall�sendmsgzneeds sendmsg()c C s | j d d � � d S)Nc S s | j | g � S)N)rh )rb rL r r r r s z.SocketEINTRTest.test_sendmsg.<locals>.<lambda>)rd )r! r r r � test_sendmsg s z SocketEINTRTest.test_sendmsgc C s� t j t j t j � } | j | j � | j t j d f � | j � d } | j � d j
d d d t j d | d | j d d d
d d f
� } | j | � } t
| � �7 | j � \ } } | j � | j | j � d � Wd QRXd S)
Nr � r> zimport socket, timer? z 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))rY ZAF_INETZ SOCK_STREAMrB rC Zbindr ZHOSTZ getsocknameZlistenrD r' r r ZacceptrF r1 )r! rb Zportr* r Z client_sockr. r r r � test_accept s*
zSocketEINTRTest.test_acceptr
r, �mkfifozneeds mkfifo()c
C s� t j } t j | � t j | � | j t j | � d j d d d | d | j d d d d | f � } | j | � } t | � �% | | � | j
| j � d � Wd QRXd S) Nr> zimport os, timer? z path = %azsleep_time = %rz# let the parent blockztime.sleep(sleep_time)r ) r ZTESTFN�unlinkr4 rl rB rD r' r r rF r1 )r! Zdo_open_close_readerZdo_open_close_writer�filenamer* r r r r �
_test_open8 s$
zSocketEINTRTest._test_openc C s t | d � } | j � d S)N�w)�openrC )r! �path�fpr r r � python_openT s zSocketEINTRTest.python_openc C s | j d | j � d S)Nz fp = open(path, 'r')
fp.close())ro rt )r! r r r � test_openX s zSocketEINTRTest.test_open�darwinz"hangs under OS X; see issue #25234c C s& t j | t j � } t j | � d S)N)r4 rq �O_WRONLYrC )r! rr r^ r r r �os_open\ s zSocketEINTRTest.os_openc C s | j d | j � d S)Nz,fd = os.open(path, os.O_RDONLY)
os.close(fd))ro rx )r! r r r � test_os_opena s z SocketEINTRTest.test_os_openN)r# r$ r% r& rS rT r rY r_ r` rc rd rf rg ri rk r Zrequires_freebsd_versionr4 ro rt ru �skipIfr �platformrx ry r r r r rU � s '$ * . * * 'rU c @ s" e Z d Z d Z d d � Z d S)�
TimeEINTRTestz" EINTR tests for the time module. c C sM t j � } t j | j � | j � t j � | } | j | | j � d S)N)�time� monotonicZsleepr' r �assertGreaterEqual)r! �t0�dtr r r �
test_sleepk s
zTimeEINTRTest.test_sleepN)r# r$ r% r& r� r r r r r| g s r| �pthread_sigmaskz need signal.pthread_sigmask()c @ sp e Z d Z d Z d d � Z e j e e d � d � d d � � Z e j e e d � d � d d
� � Z
d S) �SignalEINTRTestz$ EINTR tests for the signal module. c
C s t j } t j � } t j | d d � � } | j t j | | � d j d d t j � d t | � d | j d d f � } t j t j | g � } | j t j t j
| g � t j � } | j
| � } t | � � | | � t j � | } Wd QRX| j | j � d
� d S) Nc W s d S)Nr )r
r r r r s z/SignalEINTRTest.check_sigwait.<locals>.<lambda>r> zimport os, timezpid = %sz signum = %szsleep_time = %rztime.sleep(sleep_time)zos.kill(pid, signum)r )r �SIGUSR1r4 �getpidrB rD rZ r' r� � SIG_BLOCK� SIG_UNBLOCKr} r~ r r rF r1 )
r! r2 �signumr8 Z old_handlerr* Zold_maskr� r r� r r r �
check_sigwait{ s&
z SignalEINTRTest.check_sigwait� sigwaitinfozneed signal.sigwaitinfo()c C s d d � } | j | � d S)Nc S s t j | g � d S)N)r r� )r� r r r r2 � s z3SignalEINTRTest.test_sigwaitinfo.<locals>.wait_func)r� )r! r2 r r r �test_sigwaitinfo� s z SignalEINTRTest.test_sigwaitinfo� sigtimedwaitc C s d d � } | j | � d S)Nc S s t j | g d � d S)Ng ^@)r r� )r� r r r r2 � s z4SignalEINTRTest.test_sigtimedwait.<locals>.wait_func)r� )r! r2 r r r �test_sigtimedwait� s z!SignalEINTRTest.test_sigtimedwaitN) r# r$ r% r& r� rS rT r r r� r� r r r r r� s s r� c @ s� e Z d Z d Z d d � Z e j e j d k d � e j e
e d � d � d d � � � Z e j e
e d
� d � d d
� � Z
e j e
e d � d � d d � � Z e j e
e d � d � d d � � Z d S)�SelectEINTRTestz$ EINTR tests for the select module. c C sV t j � } t j g g g | j � t j � | } | j � | j | | j � d S)N)r} r~ �selectr' r r )r! r� r� r r r � test_select� s
zSelectEINTRTest.test_selectrv z(poll may fail on macOS; see issue #28087�pollzneed select.pollc C s] t j � } t j � } | j | j d � t j � | } | j � | j | | j � d S)Ng @�@)r� r� r} r~ r' r r )r! �pollerr� r� r r r � test_poll� s
zSelectEINTRTest.test_poll�epollzneed select.epollc C si t j � } | j | j � t j � } | j | j � t j � | } | j � | j | | j � d S)N)
r� r� rB rC r} r~ r� r' r r )r! r� r� r� r r r �
test_epoll� s
zSelectEINTRTest.test_epoll�kqueuezneed select.kqueuec C so t j � } | j | j � t j � } | j d d | j � t j � | } | j � | j | | j � d S)Nrj )
r� r� rB rC r} r~ Zcontrolr' r r )r! r� r� r� r r r � test_kqueue� s
zSelectEINTRTest.test_kqueue�devpollzneed select.devpollc C sm t j � } | j | j � t j � } | j | j d � t j � | } | j � | j | | j � d S)Ng @�@)
r� r� rB rC r} r~ r� r' r r )r! r� r� r� r r r � test_devpoll� s
z SelectEINTRTest.test_devpollN)r# r$ r% r& r� rS rz r r{ rT r r� r� r� r� r� r r r r r� � s *
' ' r� c C s t j t t t t t � d S)N)r Z run_unittestr) rU r| r� r� r r r r � test_main� s r� �__main__)r& �
contextlibr r4 r� r rY r r r} rS Ztestr �contextmanagerr rT r ZTestCaser r) rU r| r� r� r� r# r r r r �<module> s: %%% h% �% /% 8