File: //proc/self/root/lib/python3.9/site-packages/tracer/resources/__pycache__/processes.cpython-39.pyc
a
�|
`o# � @ s� d dl mZ d dlmZ ddlZddlZddlZddlZddlZddlm Z m
Z
ddl m Z ddl
mZ G dd � d e�ZG d
d � d e�ZG d d
� d
e�ZG dd� deee��ZG dd� de�ZdS )� )�ProcessesCollection)�FilenameCleaner� N)�PIPE�Popen)�Timer)�with_metaclassc @ s$ e Zd Zedd� �Zedd� �ZdS )� Processesc C s* z
t �� W S ty$ t �� Y S 0 d S �N)�psutil�pids�AttributeErrorZ get_pid_list� r r �>/usr/lib/python3.9/site-packages/tracer/resources/processes.pyr " s
zProcesses.pidsc C sR t � } t�� D ]>}z| �t|�� W q tjy8 Y q tjyJ Y q0 q| S r
)r r r �append�Processr Z
NoSuchProcess� AccessDenied)Z processes�pidr r r �all) s z
Processes.allN)�__name__�
__module__� __qualname__� staticmethodr r r r r r r s
r c @ s� e Zd ZdZd!dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
d"dd�Zdd� Zd d � Zd#d d �ZdS )$�ProcessWrappera�
Wrapper for ``psutil.Process class``
Library ``psutil`` is not backward compatible from version 2.x.x to 1.x.x.
Purpose of this class is cover incompatibility in ``psutil.Process`` class and
provide interface of new version. It allows using new interface even with
old version of ``psutil``.
Note that, for performance reasons, process information is cached at
object creation. To force a refresh, invoke the ``rebuild_cache()``
method.
Nc C s t �|�| _| �� d S r
)r r �_process�
rebuild_cache��selfr r r r �__init__B s zProcessWrapper.__init__c C s
t | j�S r
)�boolr �r r r r � __nonzero__F s zProcessWrapper.__nonzero__c C s | j jg d�d�| _d S )N)�name�exe�cmdline�ppid�username� create_time)�attrs)r Zas_dict� _procdictr r r r r I s z ProcessWrapper.rebuild_cachec
C s| z\| � d�dkrZ| � d�| � d�vrZt| � d��dkrZd�t�dd�| � d���d �W S W n tjyp Y n0 | � d�S ) Nr" Zsshdr# r$ r zssh-{0}-sessionz |@� )�_attr�len�format�re�split�joinr r r r r r r" L s &(zProcessWrapper.namec C s
| � d�S )Nr# �r+ r r r r r# W s zProcessWrapper.exec C s
| � d�S )Nr$ r1 r r r r r$ Z s zProcessWrapper.cmdlinec C s
| � d�S )Nr% r1 r r r r r% ] s zProcessWrapper.ppidc C s
| � d�S )N�parentr1 r r r r r2 ` s zProcessWrapper.parentc C s
| � d�S )Nr& r1 r r r r r& c s zProcessWrapper.usernamec C s
| � d�S )Nr' r1 r r r r r' f s zProcessWrapper.create_timeFc C sZ d� |�}|| jvrPz| j�|�| j|< W n$ tyN | j�|�| j|< Y n0 | j| S )Nz children-{0})r- r) r �childrenr
Z get_children)r � recursive�keyr r r r3 i s
zProcessWrapper.childrenc C sN || j vrDt| j|�}z|� | j |< W n tyB || j |< Y n0 | j | S r
)r) �getattrr � TypeError)r r" �attrr r r r+ r s
zProcessWrapper._attrc C s t | j|�S r
)r6 r )r �itemr r r � __getattr__{ s zProcessWrapper.__getattr__Tc C s^ d� |�}|| jvrTz| jj|d�| j|< W n& tyR | jj|d�| j|< Y n0 | j| S )Nzmemory_maps-{0})�grouped)r- r) r � memory_mapsr
Zget_memory_maps)r r; r5 r r r r< s
zProcessWrapper.memory_maps)N)F)T)r r r �__doc__r r! r r" r# r$ r% r2 r& r' r3 r+ r: r< r r r r r 4 s
r c s( e Zd ZdZ� fdd�Zdd� Z� ZS )� ProcessMetaz�
Caching metaclass that ensures that only one ``Process`` object is ever
instantiated for any given PID. The cache can be cleared by calling
``Process.reset_cache()``.
Based on https://stackoverflow.com/a/33458129
c s6 t t� ��|||� � fdd�}|� t� d|� d S )Nc s
i � _ d S r
)�_cacher ��clsr r � reset_cache� s z)ProcessMeta.__init__.<locals>.reset_cacherB )�superr> r �setattr)rA r" �basesZ
attributesrB �� __class__r@ r r � s zProcessMeta.__init__c O sV |d }|| j vrL| j| g|�R i |��}| j|g|�R i |�� || j |< | j | S )Nr )r? �__new__r )rA �args�kwargsr r r r r �__call__� s
zProcessMeta.__call__)r r r r= r rK �
__classcell__r r rF r r> � s r> c s� e Zd ZdZdd� Zdd� Zdd� Zed d d
��Ze d d � �Z
d
d� Z � fdd�Z d � fdd� Z
e � fdd��Ze dd� �Ze dd� �Ze dd� �Ze d d � �Z� ZS ) r aS
Represent the process instance uniquely identifiable through PID
For all class properties and methods, please see
http://pythonhosted.org/psutil/#process-class
Below listed are only reimplemented ones.
For performance reasons, instances are cached based on PID, and
multiple instantiations of a ``Process`` object with the same PID will
return the same object. To clear the cache, invoke
``Process.reset_cache()``. Additionally, as with ``ProcessWrapper``,
process information is cached at object creation. To force a refresh,
invoke the ``rebuild_cache()`` method on the object.
c C s | j |j kS )zBFor our purposes, two processes are equal when they have same name)r �r �processr r r �__eq__� s zProcess.__eq__c C s | � |� S r
)rO rM r r r �__ne__� s zProcess.__ne__c C s
t | j�S r
)�hashr r r r r �__hash__� s zProcess.__hash__� �?c C sT t dd| gttd�}t||j�}z$|�� |�� |jdkW |�� S |�� 0 dS )z�
Process arguments could be referring to files on remote filesystems and
os.path.isfile will hang forever if the shared FS is offline.
Instead, use a subprocess that we can time out if we can't reach some file.
�testz-f)�stdout�stderrr N)r r r �kill�startZ communicate�
returncode�cancel)Z file_path�timeoutrN Ztimerr r r � safe_isfile� s
�zProcess.safe_isfilec C sd g }| � � D ]}|�t�|j�� q | �� dd � D ]&}tj�|�sFq4t� |�r4|�|� q4t
|�S �Nr ) r< r r �strip�pathr$ �os�isabsr r\ �sorted)r �filesZmmap�argr r r rc � s
z
Process.filesc C s | � � rt| � � �S dS )zGThe parent process casted from ``psutil.Process`` to tracer ``Process``N)r% r r r r r r2 � s zProcess.parentc s* zt t| ��� W S ty$ Y dS 0 dS )zcThe user who owns the process. If user was deleted in the meantime,
``None`` is returned instead.N)rC r r& �KeyErrorr rF r r r& � s zProcess.usernameFc s" t t| ��|�}tdd� |D ��S )zjThe collection of process's children. Each of them casted from ``psutil.Process``
to tracer ``Process``.c S s g | ]}t |j��qS r )r r )�.0�childr r r �
<listcomp>� � z$Process.children.<locals>.<listcomp>)rC r r3 r )r r4 r3 rF r r r3 � s zProcess.childrenc sB t t| ��� }|�d�r$|dd� }d|v r>|d|�d�� }|S )zcThe absolute path to process executable. Cleaned from arbitrary strings
which appears on the end.z#newr ����;)rC r r# �endswith�index)r r# rF r r r# � s
z Process.exec C s | � � dv S )N)Zpython)r" r r r r �is_interpreted s zProcess.is_interpretedc C s8 | � � }|d u rd S | �� }|d u s0||� � kr4dS d S )NT)�terminalr2 )r ro r2 r r r �
is_session s zProcess.is_sessionc C s@ | j r8| �� dd � D ] }tj�|�rtj�|� S q| �� S r] )rn r$ r` r_ �isfile�basenamer" )r rd r r r � real_name s
zProcess.real_namec C s� t j �t�� �}t j �| �� �}|| }d}|jdkrFt|j�d }n`|jdkrhtt|jd ��d }n>|jdkr�tt|jd ��d }n |jdkr�tt|j��d }|S ) z�
The time of how long process is running. Returned as string
in format ``XX unit`` where unit is one of
``days`` | ``hours`` | ``minutes`` | ``seconds``
� r z daysi z hours�< z minutesz seconds)�datetimeZ
fromtimestamp�timer' Zdays�strZseconds�int)r Znow�startedZ started_strr r r �str_started_ago s
zProcess.str_started_ago)rS )F)r r r r= rO rP rR r r\ �propertyrc r2 r& r3 r# rn rp rs r{ rL r r rF r r � s*
r c @ s& e Zd ZdZdZddd�Zdd� ZdS )�AffectedProcessNc C s t �| |� t� | _t� | _d S r
)r r �set�packagesrc r r r r r 9 s zAffectedProcess.__init__c C s$ | j �|j �| _ | j�|j�| _d S r
)rc �unionr rM r r r �update> s zAffectedProcess.update)N)r r r r rc r r� r r r r r} 5 s
r} )� collectionsr r r rv rw r` r. �
subprocessr r � threadingr Zsixr �objectr r �typer> r r} r r r r �<module> s U