Your IP : 216.73.216.213
B
�d{T�$ � @ s` d Z dZddlZddlZddlmZmZmZmZ ddl m Z m
Z
m Z m Z m
Z
ddlmZmZmZmZ ddlmZ ddlmZ G d d
� d
e�Zd d � Zd
ZdZdZdZ dZ dZ G dd� de�Z G dd� d�Z yG dd� de!�Z"W n& e#k
�r G dd� de�Z"Y nX G dd� de"�Z$G d d � d e"�Z%G d d � d e"�Z&e"e'fZ(G d d!� d!e�Z)G d"d#� d#�Z*dS )$z�module providing:
* process information (linux specific: rely on /proc)
* a class for resource control (memory / time / cpu time)
This module doesn't work on windows platforms (only tested on linux)
:organization: Logilab
zrestructuredtext en� N)� getrlimit� setrlimit�
RLIMIT_CPU� RLIMIT_AS)�signal�SIGXCPU�SIGKILL�SIGUSR2�SIGUSR1)�Timer�
currentThread�Thread�Event)�time)�Nodec @ s e Zd ZdS )�
NoSuchProcessN)�__name__�
__module__� __qualname__� r r �D/opt/alt/python37/lib/python3.7/site-packages/logilab/common/proc.pyr ( s r c C s t j�d| �st� �dS )zTcheck the a pid is registered in /proc
raise NoSuchProcess exception if not
z/proc/%sN)�os�path�existsr )�pidr r r � proc_exists* s r � �
� � � � c @ sJ e Zd ZdZdd� Zdd� Zdd� Zdd d
�Zd d � Zd
d� Z dd� Z
dS )�ProcInfoz4provide access to process information found in /procc C sD t |�| _t�| | j� t| j� d| j | _t | �� t �| _d S )Nz
/proc/%s/stat) �intr r �__init__r �file�status�PPID�ppid)�selfr r r r r$ ; s
zProcInfo.__init__c C s* yt | �� t �S tk
r$ dS X dS )z,return the memory usage of the process in Kor N)r# r& �VSIZE�IOError)r) r r r � memory_usageB s zProcInfo.memory_usagec C s | � � tdd� | jD �� S )Nc S s g | ] }|� � �qS r )�lineage_memory_usage)�.0Zchildr r r �
<listcomp>J s z1ProcInfo.lineage_memory_usage.<locals>.<listcomp>)r, �sum�children)r) r r r r- I s z ProcInfo.lineage_memory_usager c C sD | � � }t|t �t|t � }|r@|t|t �t|t � 7 }|S )zareturn the number of jiffies that this process has been scheduled
in user and kernel mode)r& r# �UTIME�STIME�CUTIME�CSTIME)r) r1 r& r r r r r M s
z
ProcInfo.timec C s t | j��� �� S )z3return the list of fields found in /proc/<pid>/stat)�openr% �read�split)r) r r r r& V s zProcInfo.statusc C s | � � d �d�S )z:return the process name found in /proc/<pid>/stat
� z())r& �strip)r) r r r �nameZ s z
ProcInfo.namec C s t �| j�tj S )z&return the age of the process
)r �statr% �ST_MTIME)r) r r r �age_ s z ProcInfo.ageN)r ) r r r �__doc__r$ r, r- r r& r; r> r r r r r" 8 s
r" c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �ProcInfoLoaderzmanage process informationc C s
i | _ d S )N)�_loaded)r) r r r r$ g s zProcInfoLoader.__init__c c s* x$t �d�D ]}|�� r t|�V q W dS )z%return a list of existent process idsz/procN)r �listdir�isdigitr# )r) Zsubdirr r r � list_pidsj s zProcInfoLoader.list_pidsc C sD t |�}y
| j| S tk
r> t|�}| |_|| j|< |S X dS )z%get a ProcInfo object for a given pidN)r# rA �KeyErrorr" Zmanager)r) r �procinfor r r �loadp s
zProcInfoLoader.loadc C s` xZ| � � D ]N}y4| �|�}|jdkr@|jr@| �|j�}|�|� W q
tk
rV Y q
X q
W dS )z load all processes informationN)rD rG �parentr( �appendr )r) r rF Z pprocinfor r r �load_all| s
zProcInfoLoader.load_allN)r r r r? r$ rD rG rJ r r r r r@ d s
r@ c @ s e Zd ZdZdZdS )�
ResourceErrorz*Error raise when resource limit is reachedzUnknown Resource LimitN)r r r r? �limitr r r r rK � s rK c @ s e Zd ZdZdZdS )rK z*Error raise when resource limit is reachedzUnknown Resource LimitN)r r r r? rL r r r r rK � s c @ s e Zd ZdZdZdS )� XCPUErrorz+Error raised when CPU Time limit is reachedzCPU TimeN)r r r r? rL r r r r rM � s rM c @ s e Zd ZdZdZdS )�LineageMemoryErrorz\Error raised when the total amount of memory used by a process and
it's child is reachedzLineage total MemoryN)r r r r? rL r r r r rN � s rN c @ s e Zd ZdZdZdS )� TimeoutErrorz9Error raised when the process is running for to much timez Real TimeN)r r r r? rL r r r r rO � s rO c @ s0 e Zd ZdZe�� fdd�Zdd� Zdd� ZdS ) �MemorySentinelz\A class checking a process don't use too much memory in a separated
daemonic thread
c C s: t j| | jdd� || _t� | _|| _| �d� || _d S )Nz
Test.Sentinel)�targetr; T) r
r$ �_run� memory_limitr �_stop�interval� setDaemon�gpid)r) rU rS rW r r r r$ � s
zMemorySentinel.__init__c C s | j �� dS )zstop apN)rT �set)r) r r r �stop� s zMemorySentinel.stopc C sL t � }x@| j�� sF| j|�| j��� kr6t�| jt � | j�
| j � qW d S )N) r@ rT �isSetrS rG rW r- r �killpgr
�waitrU )r) Zpilr r r rR � s
zMemorySentinel._runN) r r r r? r �getpidr$ rY rR r r r r rP � s rP c @ sF e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dS )�ResourceControllerN�< c C sn t dkrtd��|| _|| _|| _|| _d | _d | _d | _d | _ d | _
d | _ d | _ d| _
d| _d | _d| _d S )N���zUnsupported platformr )r � RuntimeError�max_time�
max_memory� max_cpu_time� _reprieve�_timer�
_msentinel�_old_max_memory�_old_usr1_hdlr�_old_max_cpu_time�_old_usr2_hdlr�_old_sigxcpu_hdlr�
_limit_set�
_abort_try� _start_time� _elapse_time)r) rd rb rc Z max_repriever r r r$ � s" zResourceController.__init__c C s
t � �d S )N)rO )r) �sig�framer r r �_hangle_sig_timeout� s z&ResourceController._hangle_sig_timeoutc C s8 | j | jk r$| j d7 _ td��nt�t�� t� d S )Nr9 zMemory limit reached)rn re rN r r[ r] r )r) rq rr r r r �_hangle_sig_memory� s
z%ResourceController._hangle_sig_memoryc C s8 | j | jk r$| j d7 _ td��nt�t�� t� d S )Nr9 zSoft CPU time limit reached)rn re rM r r[ r] r )r) rq rr r r r �_handle_sigxcpu� s
z"ResourceController._handle_sigxcpuc C sb | j | jk rN| j d7 _ t�t�� t� | jdkr^td| j�| _ | j �
� nt�t�� t � d S )Nr9 r ) rn re r r[ r] r rm r � _time_outrf �startr )r) r r r rv � s
z ResourceController._time_outc C s$ t � �� dkst�t�� | jdk�r| jdk rttt| j �| _
t t dt
| j�| j �| j�| _t
t� �| _| j�� | jdk r�tt�| _t
| j�| jd f}tt| j�| _tt|� | j dk �rt dt
| j ��| _ tt �| _ tt!| j"�| _#t
| j �| j d f}tt |� | j �� | jd7 _dS )zset up the process limit�
MainThreadr Nr9 )$r �getName�AssertionErrorr �setpgrprm rb r r rs rk r �maxr# rp rv rf r ro rw rd r r rj r ru rl r rc rP rg r rh r
rt ri )r) Z cpu_limitZas_limitr r r � setup_limit� s,
z ResourceController.setup_limitc C s� | j dkr�| jdk rJ| j�� | jtt� �| j 7 _d| _tt | j
� | j dk rlt t
| j� tt| j� | jdk r�| j�� d| _t t| j� tt| j� | j d8 _ dS )z reinstall the old process limitr Nr9 )rm rb rf �cancelrp r# r ro r r rk rd r r rj r rl rc rg rY r rh r
ri )r) r r r � clean_limit s
z ResourceController.clean_limit)NNNr_ )
r r r r$ rs rt ru rv r} r r r r r r^ � s
r^ )+r? Z
__docformat__r r< �resourcer r r r r r r r r
� threadingr r r
r r Zlogilab.common.treer � Exceptionr r r' r2 r3 r4 r5 r* r" r@ �
BaseExceptionrK � NameErrorrM rN rO � MemoryErrorZRESOURCE_LIMIT_EXCEPTIONrP r^ r r r r �<module> s8 ,$