Current File : //usr/lib/python3.9/site-packages/tracer/resources/__pycache__/processes.cpython-39.pyc
a
" eq# � @ s� d dl mZ d dlmZ ddlZddlZddlZddlZddlmZm Z ddl
m Z ddl m
Z
G dd � d e�ZG d
d � d e�ZG d d
� d
e�ZG dd� de
ee��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__A s zProcessWrapper.__init__c C s
t | j�S r
)�boolr �r r r r � __nonzero__E 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 H s z ProcessWrapper.rebuild_cachec C s| z\| � d�dkrZ| � d�| � d�vrZt| � d��dkrZ| � d�d �d�d }d�|�W S W n tjyp Y n0 | � d�S ) Nr" Zsshdr# r$ r �@r zssh-{0}-session)�_attr�len�split�formatr r )r r& r r r r" K s &zProcessWrapper.namec C s
| � d�S )Nr# �r+ r r r r r# W s zProcessWrapper.exec C s
| � d�S )Nr$ r/ r r r r r$ Z s zProcessWrapper.cmdlinec C s
| � d�S )Nr% r/ r r r r r% ] s zProcessWrapper.ppidc C s
| � d�S )N�parentr/ r r r r r0 ` s zProcessWrapper.parentc C s
| � d�S )Nr& r/ r r r r r& c s zProcessWrapper.usernamec C s
| � d�S )Nr' r/ 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 r1 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
)r4 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 r9 r3 r r r r: s
zProcessWrapper.memory_maps)N)F)T)r r r �__doc__r r! r r" r# r$ r% r0 r& r' r1 r+ r8 r: r r r r r 3 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_cacher@ )�superr< r �setattr)r? r" �basesZ
attributesr@ �� __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 )r? �args�kwargsr r r r r �__call__� s
zProcessMeta.__call__)r r r r; r rI �
__classcell__r r rD 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
)rM rK 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�timeoutrL 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 rZ �sorted)r �filesZmmap�argr r r ra � 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 r0 � 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)rA r r& �KeyErrorr rD 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>)rA r r1 r )r r2 r1 rD r r r1 � 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 ����;)rA r r# �endswith�index)r r# rD 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)�terminalr0 )r rm r0 r r r �
is_session s zProcess.is_sessionc C s@ | j r8| �� dd � D ] }tj�|�rtj�|� S q| �� S r[ )rl r$ r^ r] �isfile�basenamer" )r rb 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)rQ )F)r r r r; rM rN rP r rZ �propertyra r0 r& r1 r# rl rn rq ry rJ r r rD r r � s*
r c @ s& e Zd ZdZdZddd�Zdd� ZdS )�AffectedProcessNc C s t �| |� t� | _t� | _d S r
)r r �set�packagesra r r r r r 9 s zAffectedProcess.__init__c C s$ | j �|j �| _ | j�|j�| _d S r
)ra �unionr} rK r r r �update> s zAffectedProcess.update)N)r r r r} ra r r r r r r r{ 5 s
r{ )� collectionsr r r rt ru r^ �
subprocessr r � threadingr Zsixr �objectr r �typer< r r{ r r r r �<module> s V
Mini Shell