o
#T_i � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dl
m Z zd dl m
Z
W n eyA d dlZdd� Z
Y nw d d
� Zd d � Zd
d� ZdS )� N)�
isabstract)�support)� os_helper)� clear_caches)� _get_dumpc C s& t dd� | jD ��}|| j| j| jfS )Nc s s � | ]}t �|�V qd S �N)�weakref�ref)�.0�obj� r �//usr/lib/python3.10/test/libregrtest/refleak.py� <genexpr> s � z _get_dump.<locals>.<genexpr>)�set�
_abc_registry�
_abc_cache�_abc_negative_cache�_abc_negative_cache_version)�cls�registry_weakrefsr r r
r s �r c ' s� ddl }ddl� ttd�std��t� tjdd� }|j� � }tj
� � }zddl }W n t y6 d}Y nw |j
� � }i } � fdd�� jjD �D ]}
t|
�sPqI|
�� |
g D ]
} t| �d | | < qWqIdd� td d
�D ���fd d �} | j\}
}}tj�tj|�}|
| }tt|��}dg| }dg| }dg| }tj}tj }tj }d } }}| j s�t d
|dtj d� t d|d d d|� tj dd� t!||||| � |D ]@}|� t!||||| � |� } |� } |� } | j s�t ddtj dd� | | | �||< | | | �||< | | | �||< | }| }| }q�| j �s t tj d� dd� } dd� } d }!|d | f|d | f|d | ffD ]E\}"}#}$|"|
d� }"|$|"��rxd ||"|#t"|"�f }%t |%tj dd� t#|d!��}&t |%|&d� |&�$� W d � n 1 �sqw Y d}!�q4|!S )"z�Run a test multiple times, looking for reference leaks.
Returns:
False if the test didn't leak references; True if we detected refleaks.
r N�gettotalrefcountz9Tracking reference leaks requires a debug build of Pythonc � g | ]}t � j|��qS r ��getattr�abc�r
�a�� collectionsr r
�
<listcomp>4 � zdash_R.<locals>.<listcomp>c S s i | ]}||�qS r r )r
�valuer r r
�
<dictcomp>> s zdash_R.<locals>.<dictcomp>i���i� c s � � | | �S r )�
setdefault)r! )�int_poolr r
�get_pooled_int? s z dash_R.<locals>.get_pooled_int� beginning� repetitions)�file�
1234567890�
� T)r( �flush�.� )�endr( r, c S s t dd� | D ��S )Nc s s � | ]}|d kV qdS )r+ Nr )r
�deltar r r
r s � z2dash_R.<locals>.check_rc_deltas.<locals>.<genexpr>)�all��deltasr r r
�check_rc_deltasr s
z dash_R.<locals>.check_rc_deltasc S s t | �S r )�anyr2 r r r
�check_fd_deltas� s z dash_R.<locals>.check_fd_deltasF�
referencesz
memory blockszfile descriptorsz%s leaked %s %s, sum=%sr )%�copyreg�collections.abc�hasattr�sys� Exception� warm_caches�warnings�filters�dispatch_table�copy�path_importer_cache� zipimport� ImportError�_zip_directory_cacher �__all__r �__subclasses__r �range�
huntrleaks�os�path�joinr �SAVEDCWD�list�getallocatedblocksr �fd_count�quiet�print�stderr�dash_R_cleanup�sum�openr, )'�ns� test_name� test_funcr8 �fs�ps�picrC �zdc�abcsr r r% �nwarmup�ntracked�fname�repcount� rep_range� rc_deltas� alloc_deltas� fd_deltasrO r rP � rc_before� alloc_before� fd_before�i� alloc_after�rc_after�fd_afterr4 r6 �failedr3 � item_name�checker�msg�refrepr )r r$ r
�dash_R s�
�
�
� �
�
��rs c s dd l }dd l� | tjd d �< |j�� |j�|� tj�� tj�|� zdd l }W n t
y4 Y n w |j �� |j �|� t� � � fdd�� j
jD �}tt|�}|D ]%}|�� |g D ]} |�| t� �D ]
}
|
� d uru| �|
� � qh| �� q_qVt� d S )Nr c r r r r r r r
r � r z"dash_R_cleanup.<locals>.<listcomp>)r8 r9 r> r? r@ �clear�updater; rB rC rD rE �_clear_type_cacher rF �filterr rG �getr �register�_abc_caches_clearr ) rZ r[ r\ r] r^ r8 rC � abs_classesr r r r r r
rT � s4
�
�
�
rT c C sN t td��} td�D ]
}| ||d � q
dd� td�D � ttdd�� d S )N� r+ c S s g | ]}t |��qS r )�chr)r
rj r r r
r � s z warm_caches.<locals>.<listcomp>���i )�bytesrH rN )�srj r r r
r= � s
r= )rJ �rer; r> �inspectr �testr � test.supportr �test.libregrtest.utilsr �_abcr rD r rs rT r= r r r r
�<module> s$ � "