File: //usr/lib/python3.6/site-packages/tracer/resources/__pycache__/processes.cpython-36.pyc
3
�|
`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& yt j� S tk
r t j� S X d S )N)�psutil�pids�AttributeErrorZ get_pid_list� r
r
� /usr/lib/python3.6/processes.pyr " s zProcesses.pidsc C sZ t � } xNtj� D ]B}y| jt|�� W q tjk
r< Y q tjk
rP Y qX qW | S )N)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 j|�| _| j� d S )N)r
r �_process�
rebuild_cache)�selfr r
r
r �__init__B s zProcessWrapper.__init__c C s
t | j�S )N)�boolr )r r
r
r � __nonzero__F s zProcessWrapper.__nonzero__c C s | j jddddddgd�| _d S )N�name�exe�cmdline�ppid�username� create_time)Zattrs)r Zas_dict� _procdict)r r
r
r r I s z ProcessWrapper.rebuild_cachec C s| yZ| j d�dkrX| j d�| j d�krXt| j d��dkrXdjtjddj| j d���d �S W n tjk
rp Y nX | j 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
| j d�S )Nr )r' )r r
r
r r W s zProcessWrapper.exec C s
| j d�S )Nr! )r' )r r
r
r r! Z s zProcessWrapper.cmdlinec C s
| j d�S )Nr" )r' )r r
r
r r" ] s zProcessWrapper.ppidc C s
| j d�S )N�parent)r' )r r
r
r r- ` s zProcessWrapper.parentc C s
| j d�S )Nr# )r' )r r
r
r r# c s zProcessWrapper.usernamec C s
| j d�S )Nr$ )r' )r r
r
r r$ f s zProcessWrapper.create_timeFc C s\ dj |�}|| jkrRy| jj|�| j|< W n& tk
rP | jj|�| j|< Y nX | j| S )Nz children-{0})r) r% r �childrenr Z get_children)r � recursive�keyr
r
r r. i s
zProcessWrapper.childrenc C sP || j krFt| j|�}y|� | j |< W n tk
rD || j |< Y nX | j | S )N)r% �getattrr � TypeError)r r �attrr
r
r r' r s
zProcessWrapper._attrc C s t | j|�S )N)r1 r )r �itemr
r
r � __getattr__{ s zProcessWrapper.__getattr__Tc C s` dj |�}|| jkrVy| jj|d�| j|< W n( tk
rT | jj|d�| j|< Y nX | j| S )Nzmemory_maps-{0})�grouped)r) r% r � memory_mapsr Zget_memory_maps)r r6 r0 r
r
r r7 s
zProcessWrapper.memory_maps)N)F)T)r r r �__doc__r r r r r r! r" r- r# r$ r. r' r5 r7 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� �j|||� � fdd�}|� t� d|� d S )Nc s
i � _ d S )N)�_cacher
)�clsr
r � reset_cache� s z)ProcessMeta.__init__.<locals>.reset_cacher< )�superr9 r �setattr)r; r �basesZ
attributesr< )� __class__)r; r r � s zProcessMeta.__init__c O sJ |d }|| j kr@| j| f|�|�}| j|f|�|� || j |< | j | S )Nr )r: �__new__r )r; �args�kwargsr r r
r
r �__call__� s
zProcessMeta.__call__)r r r r8 r rD �
__classcell__r
r
)r@ r r9 � s r9 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 | j |� S )N)rG )r rF r
r
r �__ne__� s zProcess.__ne__c C s
t | j�S )N)�hashr )r r
r
r �__hash__� s zProcess.__hash__� �?c
C sJ t dd| gttd�}t||j�}z|j� |j� |jdkS |j� X 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.
Ztestz-f)�stdout�stderrr N)r r r �kill�startZ communicate�
returncodeZcancel)Z file_pathZtimeoutrF Ztimerr
r
r � safe_isfile� s
zProcess.safe_isfilec C sl g }x"| j � D ]}|jtj|j�� qW x:| j� dd � D ]&}tjj|�sLq:tj |�r:|j|� q:W t
|�S )Nr ) r7 r r �strip�pathr! �os�isabsr rQ �sorted)r �filesZmmap�argr
r
r rW � s
z
Process.filesc C s | j � rt| j � �S dS )zGThe parent process casted from ``psutil.Process`` to tracer ``Process``N)r" r )r r
r
r r- � s zProcess.parentc s( yt t| �j� S tk
r" dS X dS )zcThe user who owns the process. If user was deleted in the meantime,
``None`` is returned instead.N)r= r r# �KeyError)r )r@ r
r r# � s zProcess.usernameFc s" t t| �j|�}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 )�.0Zchildr
r
r �
<listcomp>� s z$Process.children.<locals>.<listcomp>)r= r r. r )r r/ r. )r@ r
r r. � s zProcess.childrenc sB t t| �j� }|jd�r$|dd� }d|kr>|d|jd�� }|S )zcThe absolute path to process executable. Cleaned from arbitrary strings
which appears on the end.z#newr � �;���)r= r r �endswith�index)r r )r@ r
r r � s
z Process.exec C s | j � dkS )N�python)ra )r )r r
r
r �is_interpreted s zProcess.is_interpretedc C s8 | j � }|d krd S | j� }|d ks0||j � kr4dS d S )NT)�terminalr- )r rc r- r
r
r �
is_session s zProcess.is_sessionc C s@ | j r8x0| j� dd � D ] }tjj|�rtjj|�S qW | j� S )Nr )rb r! rT rS �isfile�basenamer )r rX r
r
r � real_name s
zProcess.real_namec C s� t j jtj� �}t j j| 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 days�< z hoursz minutesz secondsi i )�datetimeZ
fromtimestamp�timer$ Zdays�strZseconds�int)r ZnowZstartedZ started_strr
r
r �str_started_ago s
zProcess.str_started_ago)rK )F)r r r r8 rG rH rJ r rQ �propertyrW r- r# r. r rb rd rg rn rE r
r
)r@ r r � s
r c @ s& e Zd ZdZdZddd�Zdd� ZdS )�AffectedProcessNc C s t j| |� t� | _t� | _d S )N)r r �set�packagesrW )r r r
r
r r 9 s zAffectedProcess.__init__c C s$ | j j|j �| _ | jj|j�| _d S )N)rW �unionrr )r rF r
r
r �update> s zAffectedProcess.update)N)r r r rr rW r rt r
r
r
r rp 5 s
rp )� collectionsr r r
rj rk rT r* �
subprocessr r Z threadingr Zsixr �objectr r �typer9 r rp r
r
r
r �<module> s U