shell bypass 403
o
a b." � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlZd dl Z d dl
Z
d dlm Z dZ dZ
dZe�� Zdd � ZG d
d � d ej�ZG d d
� d
e�ZG dd� d�ZG dd� de�ZG dd� de j�ZG dd� dej�Zedkrte�� dS dS )� N)�support)�threading_helper)�
lock_tests�
� g{�G�z�?c C s< t jr t�
t| � W d � dS 1 sw Y dS dS )z2Helper function for printing out debugging output.N)r �verbose� _print_mutex�print)�arg� r �9/usr/local/python-3.10/lib/python3.10/test/test_thread.py�
verbose_print s
"��r
c @ s e Zd Zdd� ZdS )�BasicThreadTestc C sZ t �� | _| j�� t �� | _t �� | _d| _d| _d| _t �
� }| j t j g|�R � d S �Nr )
�thread�
allocate_lock�
done_mutex�acquire�
running_mutex� random_mutex�created�running�
next_identr Zthreading_setupZ
addCleanupZthreading_cleanup)�self�keyr r r �setUp s
zBasicThreadTest.setUpN)�__name__�
__module__� __qualname__r r r r r r s r c @ sV e Zd Zdd� Zdd� Zdd� Zdd� Ze�e j
d vd
�d d � �Z d
d� Z dd� Z
dS )�ThreadRunningTestsc C sv | j �. | jd7 _td| j � t�| j| jf� | jd7 _| jd7 _W d � d S 1 s4w Y d S )N� zcreating task %s)r r r
r �start_new_thread�taskr r �r r r r �newtask* s "�zThreadRunningTests.newtaskc C s� | j � t�� d }W d � n1 sw Y td|t|d �f � t�|� td| � | j�/ | jd8 _| jt krR| jdkrZ| j
� � W d � d S W d � d S W d � d S 1 sew Y d S )N� ��@�task %s will run for %sus� ��.Az task %s doner r ) r �randomr
�round�time�sleepr r r �NUMTASKSr �release)r �ident�delayr r r r" 2 s �
��"�zThreadRunningTests.taskc C s^ t �� �! tt�D ]}| �� q td� | j�� td� W d � d S 1 s(w Y d S )Nz waiting for tasks to complete...�all tasks done)r �wait_threads_exit�ranger, r$ r
r r �r �ir r r �test_starting_threads= s
"�z(ThreadRunningTests.test_starting_threadsc C s2 | � t�� dd� t�d� | � t�� dd� d S )Nr zinitial stack size is not 0z stack_size not reset to default)� assertEqualr �
stack_sizer# r r r �test_stack_sizeF s
z"ThreadRunningTests.test_stack_size)�nt�posixztest meant for nt and posixc C s zt �d� W n ty td� Y n t jy" | �d� Y nw d}dD ]}t �|� | �t �� ||| � td| � q'dD ]:}td| � d | _d | _t �
� � t t �D ]}| �
� qYtd
� | j�� td � W d � n1 sww Y qBt �d � d S ) Ni z3caught expected ValueError setting stack_size(4096)z4platform does not support changing thread stack sizez&stack_size(%d) failed - should succeed)� � r z successfully set stack_size(%d))r; r< ztrying stack_size = (%d)r z!waiting for all tasks to completer0 )r r7 �
ValueErrorr
�errorZskipTestr6 r r r r1 r2 r, r$ r r )r Zfail_msgZtssr4 r r r � test_nt_and_posix_stack_sizeM s2 �
��z/ThreadRunningTests.test_nt_and_posix_stack_sizec s� t �� }t �� ���� g ���fdd�}t�� �H t �|d� �s)t�t � �r"| �
t �� |d � �� � g � t �
|� fdd��}~� sQt�t � t�� � rF| �
t �� |� W d � d S 1 sdw Y d S )Nc s �� d � � �� � �� d S �N)�appendr r- r )�mut�startedr r r" s s
z,ThreadRunningTests.test__count.<locals>.taskr r c s
� � d �S r@ )rA )�_)�doner r �<lambda>� s
z0ThreadRunningTests.test__count.<locals>.<lambda>)r �_countr r r r1 r! r* r+ �
POLL_SLEEPr6 r- �weakref�refr �
gc_collect)r Zorigr" �wrr )rE rB rC r � test__countl s*
�
�"�z ThreadRunningTests.test__countc s� � fdd�}t �� � t�� �L}t�� � � �� t �|d� � �� W d � n1 s,w Y | �t |j
j �d� | � |j
j
|� | �|j
jd� | �|j
j� W d � d S 1 s]w Y d S )Nc s � � � td��)N� task failed)r- r= r �rC r r r" � s z:ThreadRunningTests.test_unraisable_exception.<locals>.taskr rN z&Exception ignored in thread started by)r r r Zcatch_unraisable_exceptionr r1 r r! r6 �strZ
unraisable� exc_valueZassertIs�object�err_msg�assertIsNotNone�
exc_traceback)r r" �cmr rO r �test_unraisable_exception� s
�
�"�z,ThreadRunningTests.test_unraisable_exceptionN)r r r r$ r" r5 r8 �unittestZskipIf�os�namer? rM rW r r r r r ( s
r c @ � e Zd Zdd� Zdd� ZdS )�Barrierc C s. || _ d| _t�� | _t�� | _| j�� d S r )� num_threads�waitingr r �
checkin_mutex�checkout_mutexr )r r] r r r �__init__� s
zBarrier.__init__c C s� | j �� | jd | _| j| jkr | jd | _| j�� d S | j �� | j�� | jd | _| jdkr:| j �� d S | j�� d S )Nr r )r_ r r^ r] r` r- r# r r r �enter� s
z
Barrier.enterN)r r r ra rb r r r r r\ � s r\ c @ r[ )� BarrierTestc C sv t �� �- tt�| _t| _tt�D ]
}t�| j |f� qt
d� | j � � t
d� W d � d S 1 s4w Y d S )Nzwaiting for tasks to endz
tasks done)
r r1 r\ r, �barr r2 r r! �task2r
r r r3 r r r � test_barrier� s
"�zBarrierTest.test_barrierc C s� t t�D ]F}|dkr
d}n| j� t�� d }W d � n1 s!w Y td|t|d �f � t�|� td||f � | j� � td| � q| j
� | j d8 _ | j dk}W d � n1 sew Y |rs| j �
� d S d S )Nr r% r& r' ztask %s entering %sztask %s leaving barrierr )r2 �NUMTRIPSr r( r
r) r* r+ rd rb r r r r- )r r. r4 r/ �finishedr r r re � s( ��
��zBarrierTest.task2N)r r r rf re r r r r rc � s
rc c @ s e Zd ZejZdS )� LockTestsN)r r r r r Zlocktyper r r r ri � s
ri c @ s< e Zd Zdd� Ze�eed�d�ej dd� ��Z
dd� Z d S )
�TestForkInThreadc C s t �� \| _| _d S r@ )rY �pipe�read_fd�write_fdr# r r r r � s zTestForkInThread.setUp�forkz need os.forkc s� d � � fdd�}t �� �# t�|| j| jf� | �t�| jd�d� t� | j� W d � n1 s2w Y | �
� � t j � dd� d S )Nc sD t �� � � rd S zt �| � t �|d� W t �d� d S t �d� w )N� OKr )rY rn �close�write�_exit)rl rm ��pidr r � fork_thread� s
z7TestForkInThread.test_forkinthread.<locals>.fork_thread� ro r )�exitcode)
r r1 r r! rl rm r6 rY �readrp rT r Z wait_process)r ru r rs r �test_forkinthread� s
�
z"TestForkInThread.test_forkinthreadc C sL zt �| j� W n ty Y nw z t �| j� W d S ty% Y d S w r@ )rY rp rl �OSErrorrm r# r r r �tearDown s � �zTestForkInThread.tearDownN) r r r r rX Z
skipUnless�hasattrrY r Z reap_threadsry r{ r r r r rj � s rj �__main__)rY rX r( �testr Z test.supportr �_threadr r* rI r r, rg rH r r r
ZTestCaser r r\ rc ri rj r �mainr r r r �<module> s. u%- �