CasperSecurity
o
`�wiF � @ s d dl mZmZ d dlmZ d dlZd dlZd dlZd dl Z d dl
T d dl
m Z m Z dd� Z
G dd � d �ZG d
d � d �Zd d
� ZeG dd� d��ZeG dd� de��Zdd� ZG dd� d�ZG dd� de�Zdd� ZG dd� d�ZG d d � d e�ZG d d � d e�ZG d d!� d!�Zd"d#� ZdS )$� )�Sequence�Iterable)�total_orderingN)�*)�_get_object_traceback� _get_tracesc C s� dD ]>}t | �dk r |dkr |rd| |f S d| |f S t | �dk s*|dkr<|r4d| |f S d | |f S | d
} qd S ) N)�B�KiB�MiB�GiB�TiB�d r z%+.1f %sz%.1f %si ( r z%+.0f %sz%.0f %si )�abs)�size�sign�unit� r �"/usr/lib/python3.10/tracemalloc.py� _format_size
s
�r c @ �D e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dS )� StatisticzS
Statistic difference on memory allocations between two Snapshot instance.
�� tracebackr �countc C s || _ || _|| _d S �Nr )�selfr r r r r r �__init__% s
zStatistic.__init__c C s t | j| j| jf�S r )�hashr r r �r r r r �__hash__* � zStatistic.__hash__c C s2 t |t�stS | j|jko| j|jko| j|jkS r )�
isinstancer �NotImplementedr r r �r �otherr r r �__eq__- s
�
�zStatistic.__eq__c C sB d| j t| jd�| jf }| jr | j| j }|dt|d� 7 }|S )Nz%s: size=%s, count=%iF� , average=%s)r r r r �r �text�averager r r �__str__4 s
�� zStatistic.__str__c C s d| j | j| jf S )Nz)<Statistic traceback=%r size=%i count=%i>r r r r r �__repr__> s �zStatistic.__repr__c C s | j | j| jfS r )r r r r r r r � _sort_keyB � zStatistic._sort_keyN� �__name__�
__module__� __qualname__�__doc__� __slots__r r r% r* r+ r, r r r r r s
r c @ r )�
StatisticDiffzd
Statistic difference on memory allocations between an old and a new
Snapshot instance.
�r r � size_diffr �
count_diffc C s" || _ || _|| _|| _|| _d S r r5 )r r r r6 r r7 r r r r M s
zStatisticDiff.__init__c C s t | j| j| j| j| jf�S r )r r r r6 r r7 r r r r r T s �zStatisticDiff.__hash__c C sJ t |t�stS | j|jko$| j|jko$| j|jko$| j|jko$| j|jkS r )r! r4 r" r r r6 r r7 r# r r r r% X s
�
�
�
�zStatisticDiff.__eq__c C sP d| j t| jd�t| jd�| j| jf }| jr&| j| j }|dt|d� 7 }|S )Nz %s: size=%s (%s), count=%i (%+i)FTr& )r r r r6 r r7 r' r r r r* a s
�� zStatisticDiff.__str__c C s d| j | j| j| j| jf S )Nz9<StatisticDiff traceback=%r size=%i (%+i) count=%i (%+i)>r5 r r r r r+ m s
��zStatisticDiff.__repr__c C s t | j�| jt | j�| j| jfS r )r r6 r r7 r r r r r r r, r s �zStatisticDiff._sort_keyNr. r r r r r4 F s r4 c C s� g }|� � D ]2\}}| �|d �}|d ur't||j|j|j |j|j|j �}n t||j|j|j|j�}|�|� q| � � D ]\}}t|d|j d|j �}|�|� q=|S �Nr )�items�popr4 r r �append)� old_group� new_group�
statisticsr �stat�previousr r r �_compare_grouped_statsx s" �� rA c @ s\ e Zd ZdZdZdd� Zedd� �Zedd� �Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dS )�Framez
Frame of a traceback.
��_framec C �
|| _ d S r rC �r �framer r r r � �
zFrame.__init__c C �
| j d S r8 rC r r r r �filename� rH zFrame.filenamec C rI �N� rC r r r r �lineno� rH z Frame.linenoc C � t |t�stS | j|jkS r �r! rB r" rD r# r r r r% � �
z Frame.__eq__c C � t |t�stS | j|jk S r rO r# r r r �__lt__� rP z Frame.__lt__c C �
t | j�S r )r rD r r r r r � �
zFrame.__hash__c C � d| j | jf S )Nz%s:%s�rJ rM r r r r r* � r- z
Frame.__str__c C rU )Nz <Frame filename=%r lineno=%r>rV r r r r r+ � r- zFrame.__repr__N)r/ r0 r1 r2 r3 r �propertyrJ rM r% rR r r* r+ r r r r rB � s
rB c @ st e Zd ZdZdZddd�Zedd� �Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd dd�ZdS ) � Tracebackz`
Sequence of Frame instances sorted from the oldest frame
to the most recent frame.
)�_frames�
_total_nframeNc C s" t �| � tt|��| _|| _d S r )r r �tuple�reversedrY rZ )r �frames� total_nframer r r r � s
zTraceback.__init__c C � | j S r )rZ r r r r r^ � � zTraceback.total_nframec C rS r )�lenrY r r r r �__len__� rT zTraceback.__len__c C �0 t |t�rtdd� | j| D ��S t| j| �S )Nc s � � | ]}t |�V qd S r )rB ��.0�tracer r r � <genexpr>� � � z(Traceback.__getitem__.<locals>.<genexpr>)r! �slicer[ rY rB �r �indexr r r � __getitem__� �
zTraceback.__getitem__c C � |j | jv S r )rD rY rF r r r � __contains__� � zTraceback.__contains__c C rS r )r rY r r r r r � rT zTraceback.__hash__c C rN r �r! rX r" rY r# r r r r% � rP zTraceback.__eq__c C rQ r rr r# r r r rR � rP zTraceback.__lt__c C s t | d �S r8 )�strr r r r r* � rq zTraceback.__str__c C s: dt | �� �}| jd u r|d7 }|S |d| j� d�7 }|S )Nz <Traceback �>z total_nframe=)r[ rZ r^ )r �sr r r r+ � s
�zTraceback.__repr__Fc C s� g }|d ur|dkr| | d � }n | d |� }n| }|r!t |�}|D ] }|�d|j|jf � t�|j|j��� }|rC|�d| � q#|S )Nr z File "%s", line %sz %s)r\ r; rJ rM � linecache�getline�strip)r �limit�most_recent_first�lines� frame_slicerG �liner r r �format� s"
��zTraceback.formatr )NF)r/ r0 r1 r2 r3 r rW r^ rb rm rp r r% rR r* r+ r~ r r r r rX � s
rX c C s t | �}|dur t|�S dS )z�
Get the traceback where the Python object *obj* was allocated.
Return a Traceback instance.
Return None if the tracemalloc module is not tracing memory allocations or
did not trace the allocation of the object.
N)r rX )�objr] r r r �get_object_traceback s r� c @ s` e Zd ZdZdZdd� Zedd� �Zedd� �Zed d
� �Z d d � Z
d
d� Z dd� Z dd� Z
dS )�Tracez"
Trace of a memory block.
��_tracec C rE r r� �r rg r r r r s
zTrace.__init__c C rI r8 r� r r r r �domain rH z Trace.domainc C rI rK r� r r r r r rH z
Trace.sizec C s t | jdd � � S )N� )rX r� r r r r r # s zTrace.tracebackc C rN r )r! r� r" r� r# r r r r% ' rP z Trace.__eq__c C rS r )r r� r r r r r , rT zTrace.__hash__c C s d| j t| jd�f S )Nz%s: %sF)r r r r r r r r* / s z
Trace.__str__c C s d| j t| jd�| jf S )Nz'<Trace domain=%s size=%s, traceback=%r>F)r� r r r r r r r r+ 2 s �zTrace.__repr__N)r/ r0 r1 r2 r3 r rW r� r r r% r r* r+ r r r r r� s
r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�_Tracesc C s t �| � || _d S r )r r �_traces)r �tracesr r r r 8 s
z_Traces.__init__c C rS r )ra r� r r r r rb = rT z_Traces.__len__c C rc )Nc s rd r )r� re r r r rh B ri z&_Traces.__getitem__.<locals>.<genexpr>)r! rj r[ r� r� rk r r r rm @ rn z_Traces.__getitem__c C ro r )r� r� r� r r r rp F rq z_Traces.__contains__c C rN r )r! r� r" r� r# r r r r% I rP z_Traces.__eq__c C s dt | � S )Nz<Traces len=%s>)ra r r r r r+ N rq z_Traces.__repr__N) r/ r0 r1 r rb rm rp r% r+ r r r r r� 7 s r� c C s& t j�| �} | �d�r| d d� } | S )Nz.pyc���)�os�path�normcase�endswith)rJ r r r �_normalize_filenameR s
r� c @ s e Zd Zdd� Zdd� ZdS )�
BaseFilterc C rE r )� inclusive)r r� r r r r Z rT zBaseFilter.__init__c C s t �r )�NotImplementedErrorr� r r r �_match] s zBaseFilter._matchN)r/ r0 r1 r r� r r r r r� Y s r� c sL e Zd Z d� fdd� Zedd� �Zdd� Zd d
� Zd d � Zd
d� Z � Z
S )�FilterNFc s2 t � �|� || _t|�| _|| _|| _|| _d S r )�superr r� r� �_filename_patternrM �
all_framesr� )r r� �filename_patternrM r� r� �� __class__r r r b s
zFilter.__init__c C r_ r )r� r r r r r� k r` zFilter.filename_patternc C s2 t |�}t�|| j�s
dS | jd u rdS || jkS )NFT)r� �fnmatchr� rM �r rJ rM r r r �_match_frame_implo s
zFilter._match_frame_implc C s | � ||�| j A S r )r� r� r� r r r � _match_framex r zFilter._match_framec sB � j rt� fdd�|D ��r� jS � j S |d \}}� �||�S )Nc 3 s � | ] \}}� � ||�V qd S r )r� )rf rJ rM r r r rh } s � �z*Filter._match_traceback.<locals>.<genexpr>r )r� �anyr� r� )r r rJ rM r r r �_match_traceback{ s � zFilter._match_tracebackc C sF |\}}}}| � |�}| jd ur!| jr|o|| jkS |p || jkS |S r )r� r� r� )r rg r� r r r^ �resr r r r� � s
z
Filter._match)NFN) r/ r0 r1 r rW r� r� r� r� r� �
__classcell__r r r� r r� a s �
r� c s0 e Zd Z� fdd�Zedd� �Zdd� Z� ZS )� DomainFilterc s t � �|� || _d S r )r� r �_domain)r r� r� r� r r r � s
zDomainFilter.__init__c C r_ r )r� r r r r r� � r` zDomainFilter.domainc C s |\}}}}|| j k| j A S r )r� r� )r rg r� r r r^ r r r r� � s zDomainFilter._match)r/ r0 r1 r rW r� r� r� r r r� r r� � s
r� c @ sX e Zd ZdZdd� Zdd� Zedd� �Zdd � Zd
d � Z d d
� Z
ddd�Z ddd�Z dS )�SnapshotzB
Snapshot of traces of memory blocks allocated by Python.
c C s t |�| _|| _d S r )r� r� �traceback_limit)r r� r� r r r r � s
zSnapshot.__init__c C s@ t |d��}t�| |tj� W d � dS 1 sw Y dS )z1
Write the snapshot into a file.
�wbN)�open�pickle�dump�HIGHEST_PROTOCOL)r rJ �fpr r r r� � s "�z
Snapshot.dumpc C s8 t | d��
}t�|�W d � S 1 sw Y dS )z.
Load a snapshot from a file.
�rbN)r� r� �load)rJ r� r r r r� � s $�z
Snapshot.loadc s@ |rt � fdd�|D ��sdS |r t � fdd�|D ��r dS dS )Nc 3 s � | ]}|� � �V qd S r �r� �rf � trace_filter�rg r r rh � s � �z)Snapshot._filter_trace.<locals>.<genexpr>Fc 3 s � | ] }|� � � V qd S r r� r� r� r r rh � s � �T)r� )r �include_filters�exclude_filtersrg r r� r �
_filter_trace� s � �zSnapshot._filter_tracec s� t |t�stdt|�j ��|r5g �g � |D ]}|jr!��|� q� �|� q� ��fdd��jjD �}n�jj� � }t
|�j �S )z�
Create a new Snapshot instance with a filtered traces sequence, filters
is a list of Filter or DomainFilter instances. If filters is an empty
list, return a new Snapshot instance with a copy of the traces.
z)filters must be a list of filters, not %sc s g | ] }�� �� |�r|�qS r )r� re �r� r� r r r �
<listcomp>� s �
�z*Snapshot.filter_traces.<locals>.<listcomp>) r! r � TypeError�typer/ r� r; r� r� �copyr� r� )r �filtersr� �
new_tracesr r� r �
filter_traces� s
� zSnapshot.filter_tracesc
C s� |dvr t d|f ��|r|dvrt d| ��i }i }|s�| jjD ]`}|\}}}} z|| }
W n, ty[ |dkr=|} n|dkrH|d d� } n |d d dff} t| �}
|
||< Y nw z||
} | j|7 _| jd7 _W q! ty� t|
|d�||
< Y q!w |S | jjD ]Y}|\}}}} |D ]N}
z||
}
W n ty� |dkr�|
f} n|
d dff} t| �}
|
||
< Y nw z||
} | j|7 _| jd7 _W q� ty� t|
|d�||
< Y q�w q�|S ) N)r rJ rM zunknown key_type: %r)rM rJ z/cumulative mode cannot by used with key type %rr rM rL r )�
ValueErrorr� r� �KeyErrorrX r r r )r �key_type�
cumulative�stats�
tracebacksrg r� r �trace_tracebackr^ r r] r? rG r r r � _group_by� sf � � � � � ��zSnapshot._group_byFc C s, | � ||�}t|�� �}|jdtjd� |S )zd
Group statistics by key_type. Return a sorted list of Statistic
instances.
T��reverse�key)r� �list�values�sortr r, )r r� r� �groupedr> r r r r> s zSnapshot.statisticsc C s6 | � ||�}|� ||�}t||�}|jdtjd� |S )z�
Compute the differences with an old snapshot old_snapshot. Get
statistics as a sorted list of StatisticDiff instances, grouped by
group_by.
Tr� )r� rA r� r4 r, )r � old_snapshotr� r� r= r< r> r r r �
compare_to s
zSnapshot.compare_toN)F)
r/ r0 r1 r2 r r� � staticmethodr� r� r� r� r> r� r r r r r� � s
3
r� c C s$ t � std��t� } t� }t| |�S )zI
Take a snapshot of traces of memory blocks allocated by Python.
zLthe tracemalloc module must be tracing memory allocations to take a snapshot)�
is_tracing� RuntimeErrorr �get_traceback_limitr� )r� r� r r r �
take_snapshot' s
r� ) �collections.abcr r � functoolsr r� rv �os.pathr� r� � _tracemallocr r r r r4 rA rB rX r� r� r� r� r� r� r� r� r� r r r r �<module> s4 (2&M'0