shell bypass 403
o
a b � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z zej W n e
y? e�d��w dd� ZG dd � d ej�Zed
krWe�� dS dS ) � N)� cpython_only)�threading_helper)�TESTFNzselect.poll not definedc C s( g }| D ]
\}}||@ r|� |� q|S )N)�append)�ready�flag�match�fd�mode� r �7/usr/local/python-3.10/lib/python3.10/test/test_poll.py�find_ready_matching s
�r
c @ s` e Zd Zdd� Zdd� Zdd� Zdd� Zed d
� �Ze j
d d � �Z e �
ed
�e j
dd� ��ZdS )� PollTestsc C sr t �� }d}d}t|�}g }g }i }i }t|�D ]-} t�� \}
} |�|
� |�|
t j� |�| t j � |�
|
� |�
| � | ||
< |
|| < qg } |r�|�� }
t |
t j �}|sZt d��t
�|�} t�| |� |�� }
t |
t j�}|sut d��t
�|�}
t�|
|�}| �t|�|� | �
|� t�||
� t�|
� |�||
� |�|
� |�||
� |sJ| �| |g| � d S )N� s This is a test.zno pipes ready for writingzno pipes ready for reading)�select�poll�len�range�os�pipe�register�modify�POLLIN�POLLOUTr r
� RuntimeError�random�choice�write�read� assertEqual�close�
unregister�remove)�self�pZ NUM_PIPESZMSGZMSG_LEN�readersZwritersZr2wZw2r�iZrd�wrZbufsr Z
ready_writersZ
ready_readers�bufr r r �
test_poll1 sN
�zPollTests.test_poll1c
C sh t �� \}}t �|� t �|� t�� }|�|� |�� }| �|d |tjf� tt d��#}|�
� }t�� }|�|� |�� }| �|d d |� W d � n1 sSw Y |�� }| �|d |tjf� t � t � t�� }| � t
|j|� | � t
|j|� t�� }| � t|jd� t�� }G dd� d�}G dd� d�} | � t
|j|� d� | � t
|j| � d� d S )Nr �w� c @ s e Zd ZdS )z,PollTests.test_poll_unit_tests.<locals>.NopeN)�__name__�
__module__� __qualname__r r r r �Nopen s r/ c @ s e Zd Zdd� ZdS )z.PollTests.test_poll_unit_tests.<locals>.Almostc S s dS )N�filenor )r# r r r r0 r s z5PollTests.test_poll_unit_tests.<locals>.Almost.filenoN)r, r- r. r0 r r r r �Almostq s r1 )r r r r r r r �POLLNVAL�openr r0 �unlink� assertRaises� TypeErrorr! �KeyError)
r# ZFDr* r$ �r�fr �pollsterr/ r1 r r r �test_poll_unit_testsO s6
�
z PollTests.test_poll_unit_testsc
C s� d}t j|dt jdd�}|�� | �|jd d d � |j}t�� }|� |tj
� dD ]A}|�|�}|g kr5q)|d \}}|tj @ rN|� � } | dkrM| �
d� q)|tj
@ rc|� � } | s\ d S | �| d� q)| �
d | � q)d S )
Nz?for i in 0 1 2 3 4 5 6 7 8 9; do echo testing...; sleep 1; doneTr )�shell�stdout�bufsize)r i� i� i� i@ i�> ���r? r? r? r? r? r? r? r? r? � z6error: pipe seems to be closed, but still returns datas testing...
z,Unexpected return value from select.poll: %s)�
subprocess�Popen�PIPE� __enter__�
addCleanup�__exit__r= r r r r �POLLHUP�readline�failr )
r# �cmd�procr$ r: ZtoutZfdlistr �flags�liner r r �
test_poll2{ s6 �
�zPollTests.test_poll2c C s� t �� }|�d� | �t|jd� d}|dkr | �d� | �t|jdd� | �t|jdd� | �t|jdd� | �t|jdd� d S )N� l � zOverflow must have occurredr r? )r r r r5 �
OverflowErrorrI �
ValueErrorr )r# r: �xr r r �
test_poll3� s
zPollTests.test_poll3c C s~ ddl m}m}m} t�� }|�d� | �t|jd|d � | �t|j d|d � | �t|j|d � | �t|j|d � d S )Nr )� USHRT_MAX�INT_MAX�UINT_MAXrO )
Z _testcapirU rV rW r r r r5 rQ r )r# rU rV rW r: r r r �test_poll_c_limits� s
z PollTests.test_poll_c_limitsc
C s� t �� \}}| �t j|� | �t j|� g }td�D ]}t �|�}| �t j|� |�|� qt�� }|D ] }|� |tj
� q4t j |jd�}|�
� z*t�d� |D ]}|�|� qQ|� |tj� | �t|j� W t �|d� |�� d S t �|d� |�� w )N�
)�targetg �?� spam)r r rE r r �dupr r r r r � threading�Thread�start�time�sleepr! r r5 r r �join)r# r8 r* Zrfdsr& r r: �tr r r �test_threaded_poll� s.
�
z PollTests.test_threaded_pollz!Threading required for this test.c C s� dD ]H}t �� \}}t�� }|�|tj� tj|j|fd�}|�� |j dd� | �
|� � � t � |d� |� � | �
|� � � t �|� t �|� qd S )N)Ni���r? g �g��������g0��.�+�)rZ �argsg�������?)�timeoutr[ )r r r r r r r] r^ r_ rb Z
assertTrue�is_aliver Z assertFalser )r# Z
timeout_msr8 r* r: Z poll_threadr r r �!test_poll_blocks_with_negative_ms� s
�z+PollTests.test_poll_blocks_with_negative_msN)r, r- r. r) r; rN rT r rX r Z reap_threadsrd �unittestZ
skipUnlessr] rh r r r r r s 1,
r �__main__)r rA r r r] r` ri Z test.supportr r Ztest.support.os_helperr r �AttributeErrorZSkipTestr
ZTestCaser r, �mainr r r r �<module> s*
� M �