CasperSecurity
o
`�wi� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ dd� Zdd� Z dd� Z
ej dd d
��Z dd d �Z
ej dd
d��ZG dd� d�ZdS )� N)�supportc C s t �� tj�� fS �N)�_thread�_count� threading� _dangling�copy� r r �4/usr/lib/python3.10/test/support/threading_helper.py�threading_setup s r c
G s� d}t |�D ]L}t�� tjf}|| kr d S |sGdt_|d }t�d|d | d � d|d � dt|�� d�� |D ]
}t�d |��� q:d }d }t �
d
� t� � qd S ) N�d T� z&threading_cleanup() failed to cleanup r z threads (count: z , dangling: �)zDangling thread: �{�G�z�?) �ranger r r r r �environment_altered�
print_warning�len�time�sleep�
gc_collect)�original_values�
_MAX_COUNT�count�values�dangling_threads�threadr r r
�threading_cleanup s, ��
�
�r c s t �� �� fdd��}|S )z�Use this function when threads are being used. This will
ensure that the threads are cleaned up even when the test fails.
c s$ t � }z � | � W t|� S t|� w r )r r )�args�key��funcr r
� decorator: s z reap_threads.<locals>.decorator)� functools�wraps)r! r" r r r
� reap_threads6 s r% c c s � | du rt j} t�� }zCdV W t�� }|| } t�� }||kr$dS t�� |krFt�� | }d|| � d|d�d|� d|� d� }t|��t�d � t �� qt�� }|| } t�� }||krc w t�� |kr�t�� | }d|| � d|d�d|� d|� d� }t|��t�d � t �� qY)
aH
bpo-31234: Context manager to wait until all threads created in the with
statement exit.
Use _thread.count() to check if threads exited. Indirectly, wait until
threads exit the internal t_bootstrap() C function of the _thread module.
threading_setup() and threading_cleanup() are designed to emit a warning
if a test leaves running threads in the background. This context manager
is designed to cleanup threads started by the _thread.start_new_thread()
which doesn't allow to wait for thread exit, whereas thread.Thread has a
join() method.
NTz!wait_threads() failed to cleanup z threads after �.1fz seconds (count: z
, old count: r r ) r �
SHORT_TIMEOUTr r r � monotonic�AssertionErrorr r )�timeout� old_count�
start_time�deadliner �dt�msgr r r
�wait_threads_exitD sZ � ���
��� ���
�r0 c C s: |du rt j}| �|� | �� rd|d�d�}t|��dS )zcJoin a thread. Raise an AssertionError if the thread is still alive
after timeout seconds.
Nz failed to join the thread in r& z seconds)r r' �join�is_aliver) )r r* r/ r r r
� join_threadi s
�r3 c c s, � dd l }t| �} g }z�z| D ] }|�� |�|� qW n tjr/tdt| �t|�f � � d V W zX|r:|� t� � }t
dd�D ]/}|d7 }|D ]}|� t |t� � d�� qKdd� |D �}|se ntjrrtd t|�|f � qCW d
d� |D �}|r�|�
tj� td t|� ��d S d
d� |D �}|r�|�
tj� td t|� ��w zW|r�|� t� � }t
dd�D ]/}|d7 }|D ]}|� t |t� � d�� q�dd� |D �}|s� ntjr�td t|�|f � q�W d
d� |D �}|r�|�
tj� td t|� ��w d
d� |D �}|�r|�
tj� td t|� ��w ) Nr z/Can't start %d threads, only %d threads startedr
� �< r c S � g | ]}|� � r|�qS r �r2 ��.0�tr r r
�
<listcomp>� � z!start_threads.<locals>.<listcomp>z7Unable to join %d threads during a period of %d minutesc S r6 r r7 r8 r r r
r; � r< zUnable to join %d threads)� faulthandler�list�start�appendr �verbose�printr r r( r r1 �max�dump_traceback�sys�stdoutr) )�threads�unlockr= �startedr: �endtimer* r r r
�
start_threadsu s� � ��
�� �� �
�� � rK c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �catch_threading_exceptiona
Context manager catching threading.Thread exception using
threading.excepthook.
Attributes set when an exception is caught:
* exc_type
* exc_value
* exc_traceback
* thread
See threading.excepthook() documentation for these attributes.
These attributes are deleted at the context manager exit.
Usage:
with threading_helper.catch_threading_exception() as cm:
# code spawning a thread which raises an exception
...
# check the thread exception, use cm attributes:
# exc_type, exc_value, exc_traceback, thread
...
# exc_type, exc_value, exc_traceback, thread attributes of cm no longer
# exists at this point
# (to avoid reference cycles)
c C s" d | _ d | _d | _d | _d | _d S r )�exc_type� exc_value�
exc_tracebackr � _old_hook��selfr r r
�__init__� s
z"catch_threading_exception.__init__c C s$ |j | _ |j| _|j| _|j| _d S r )rM rN rO r )rR r r r r
�_hook� s z catch_threading_exception._hookc C s t j| _| jt _| S r )r �
excepthookrP rT rQ r r r
� __enter__� s z#catch_threading_exception.__enter__c G s | j t_| `| `| `| `d S r )rP r rU rM rN rO r )rR �exc_infor r r
�__exit__� s
z"catch_threading_exception.__exit__N)�__name__�
__module__� __qualname__�__doc__rS rT rV rX r r r r
rL � s rL r )r �
contextlibr# rE r r �testr r r r% �contextmanagerr0 r3 rK rL r r r r
�<module> s
$ %