Your IP : 216.73.216.213
���]I? � @ s 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 r e � d S)� N)�supportc c s3 | �& y | VWn | j � � Yn XWd QRXd S)N)�kill)�proc� r �1/opt/alt/python35/lib64/python3.5/eintr_tester.py�
kill_on_error s
r � setitimerzrequires setitimer()c @ sd e 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) �
EINTRBaseTestg�������?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__r r �
sleep_time� classmethodr r r r r r r r r # s
r c @ s� e 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)� OSEINTRTestc 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_func�num� processesr- r r )r! r �_test_wait_multipleQ s
z OSEINTRTest._test_wait_multiplec C s | j t j � d S)N)r4 �osr0 )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 )r5 r7 r r r r r _ s z(OSEINTRTest.test_wait3.<locals>.<lambda>)r4 )r! r r r �
test_wait3] s zOSEINTRTest.test_wait3c C s' | j � } | | j � | j � d S)N)r* �pidr0 )r! r1 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 )r5 �waitpid)r9 r r r r h s z*OSEINTRTest.test_waitpid.<locals>.<lambda>)r: )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 )r5 r= )r9 r r r r l s z(OSEINTRTest.test_wait4.<locals>.<lambda>)r: )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 )
r5 �pipe�
addCleanup�close�joinr& r �strr � assertEqual�read�lenr0 )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))rA r )r5 rB rC rD r Z
PIPE_MAX_SIZErE r& r rF r rI �write�
memoryviewrG r0 )r! rJ rK rM r) r �writtenr r r �
test_write� s<
!
*zOSEINTRTest.test_writeN)r# r$ r% r* r4 r6 �unittest�
skipUnlessr r5 r8 r: r< r> rN rS r r r r r( I s ' ' r( c @ sf e 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)(�SocketEINTRTest�
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)NrO � 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)rA r )�socketrW rC rD rE �int�family�typer& �filenor rF r rG rI r0 ) r! Z recv_funcrJ rK rL r) �fdr rM r r r �
_test_recv� s6
!
#zSocketEINTRTest._test_recvc C s | j t j j � d S)N)r` rZ �recv)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 )rc )�sockrM 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))rA r )rZ rW rC rD r Z
SOCK_MAX_SIZErE r[ r\ r] r& r^ r rF r rI rQ rG r0 )
r! Z send_funcrJ rK rM r) r_ r rR �sentr r r �
_test_send� sJ
!
&zSocketEINTRTest._test_sendc C s | j t j j � d S)N)rg rZ �send)r! r r r � test_send s zSocketEINTRTest.test_sendc C s | j t j j � d S)N)rg rZ �sendall)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)rl )rd rM r r r r s z.SocketEINTRTest.test_sendmsg.<locals>.<lambda>)rg )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))rZ �AF_INET� SOCK_STREAMrC rD �bindr ZHOST� getsockname�listenrE r& r r �acceptrG r0 )r! rd �portr) 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�unlinkr5 rw rC rE r& r r rG r0 )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)�openrD )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())rz r )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)r5 r| �O_WRONLYrD )r! r} 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))rz r� )r! r r r � test_os_opena s z SocketEINTRTest.test_os_openN)r# r$ r% rT rU r rZ r` rb re rg ri rk rm rv r Zrequires_freebsd_versionr5 rz r r� �skipIfr �platformr� r� r r r r rV � s '$ * . * * 'rV c @ s e Z d Z d d � Z d S)�
TimeEINTRTestc C sM t j � } t j | j � | j � t j � | } | j | | j � d S)N)�time� monotonic�sleepr& r �assertGreaterEqual)r! �t0�dtr r r �
test_sleepk s
zTimeEINTRTest.test_sleepN)r# r$ r% r� r r r r r� g s r� �pthread_sigmaskz need signal.pthread_sigmask()c @ sj e 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) �SignalEINTRTestc
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 �SIGUSR1r5 �getpidrC rE r[ r& r� � SIG_BLOCK� SIG_UNBLOCKr� r� r r rG r0 )
r! r1 �signumr9 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 r1 � s z3SignalEINTRTest.test_sigwaitinfo.<locals>.wait_func)r� )r! r1 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 r1 � s z4SignalEINTRTest.test_sigtimedwait.<locals>.wait_func)r� )r! r1 r r r �test_sigtimedwait� s z!SignalEINTRTest.test_sigtimedwaitN)
r# r$ r% r� rT rU r r r� r� r r r r r� s s
r� c @ s� e 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)�SelectEINTRTestc 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_selectr� 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� rC rD 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)Nrn )
r� r� rC rD 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� rC rD r� r� r� r& r r� )r! r� r� r� r r r � test_devpoll� s
z SelectEINTRTest.test_devpollN)r# r$ r% r� rT r� r r� rU 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( rV r� r� r� r r r r � test_main� s r� �__main__)�
contextlibr r5 r� r rZ r r r� rT Ztestr �contextmanagerr rU r ZTestCaser r( rV r� r� r� r� r# r r r r �<module> s8 %%% h% �% /% 8