U
���g � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ zd dlm Z W n$ e
k
rl d dl Z dd� Z Y nX dd� Z d d
� Z
d d � Zd
d� ZdS )� N)�
isabstract)�support)� _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.8/test/libregrtest/refleak.py� <genexpr> s z _get_dump.<locals>.<genexpr>)�set�
_abc_registry�
_abc_cache�_abc_negative_cache�_abc_negative_cache_version)�clsZregistry_weakrefsr
r
r r s
�r c '
s� ddl }ddl� ttd�s"td��t� tjdd� }|j� � }tj
� � }z ddl }W n t k
rn d}Y n X |j
� � }i } � fdd�� jjD �D ]2}
t|
�s�q�|
�� |
g D ]} t| �d | | < q�q�dd� 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 ]t\}"}#}$|"|
d� }"|$|"��rpd ||"|#t"|"�f }%t |%tj dd� t#|d!��}&t |%|&d� |&�$� W 5 Q R X d}!�qp|!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 s g | ]}t � j|��qS r
��getattr�abc�r �a�� collectionsr
r �
<listcomp>1 s 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_intZ beginningZ repetitions)�fileZ
1234567890�
� T)r! �flush�.� )�endr! r$ c S s t dd� | D ��S )Nc s s | ]}|d kV qdS )r# Nr
)r Zdeltar
r
r r | s z2dash_R.<locals>.check_rc_deltas.<locals>.<genexpr>)�all��deltasr
r
r �check_rc_deltaso s
z dash_R.<locals>.check_rc_deltasc S s t | �S r )�anyr) r
r
r �check_fd_deltas~ s z dash_R.<locals>.check_fd_deltasFZ
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 �rangeZ
huntrleaks�os�path�joinr ZSAVEDCWD�list�getallocatedblocksr �fd_count�quiet�print�stderr�dash_R_cleanup�sum�openr$ )'�nsZ test_nameZ test_funcr. �fs�ps�picr9 �zdc�abcsr r r ZnwarmupZntrackedZfnameZrepcountZ rep_rangeZ rc_deltasZ alloc_deltasZ fd_deltasrC r rD Z rc_beforeZ alloc_beforeZ fd_before�iZ alloc_afterZrc_afterZfd_afterr+ r- Zfailedr* Z item_nameZchecker�msgZrefrepr
)r r r �dash_R s�
� �
� rS c s dd l }dd l� | tjd d �< |j�� |j�|� tj�� tj�|� z dd l }W n t
k
rj Y nX |j �� |j �|� t� � � fdd�� j
jD �}tt|�}|D ]J}|�� |g D ]6} |�| t� �D ]}
|
� d k r�| �|
� � q�| �� q�q�t� d S )Nr c s g | ]}t � j|��qS r
r r r r
r r � s z"dash_R_cleanup.<locals>.<listcomp>)r. r/ r4 r5 r6 �clear�updater1 r8 r9 r: r; �_clear_type_cacher r< �filterr r= �getr
�register�_abc_caches_clear� clear_caches) rL rM rN rO rP r. r9 Z abs_classesr r r r
r r rH � s.
rH c C s^ t j�� D ]} t| d�r
| `q
t jt jt jt jfD ]}|d k r2|� � q2zt jd }W n t
k
rj Y n X |j � � t
�� zt jd }W n t
k
r� Y n X |j� � zt jd }W n t
k
r� Y n
X |�� zt jd }W n t
k
r� Y n
X |�� zt jd }W n t
k
�r( Y n
X |�� zt jd }W n t
k
�rV Y n
X |�� zt jd }W n t
k
�r� Y n X |j� � zt jd } W n t
k
�r� Y n
X | �� zt jd
}
W n t
k
�r� Y nX d |
_zt jd } W n t
k
�r Y n
X | �� zt jd } W n t
k
�r< Y nX | jD ] }
|
� �qDt�� d S )
N�__warningregistry__zdistutils.dir_util� _strptimez urllib.parsezurllib.request� linecache� mimetypes�filecmp�struct�doctest�ctypes�typing)r1 �modules�valuesr0 r\ �stdoutrG �
__stdout__�
__stderr__r$ �KeyErrorZ
_path_createdrT �re�purgeZ _regex_cacheZ clear_cacheZ
urlcleanup�
clearcacheZ_default_mime_types�_cacheZ _clearcacheZmasterZ _reset_cacheZ _cleanupsr Z
gc_collect)�mod�streamZdistutils_dir_utilr] Z urllib_parseZurllib_requestr^ r_ r` ra rb rc rd �fr
r
r r[ � s�
r[ c C sN t td��} td�D ]}| ||d � qdd� td�D � ttdd�� d S )N� r# c S s g | ] }t |��qS r
)�chr)r rQ r
r
r r s z warm_caches.<locals>.<listcomp>���i )�bytesr> rB )�srQ r
r
r r3 s
r3 )r? rk r1 r4 �inspectr Ztestr �_abcr r: r rS rH r[ r3 r
r
r
r �<module> s "` |