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)�.0�childr 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) r2 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# rB �KeyErrorr" �manager)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)rE rI �parentr( �appendr )r) r rH Z pprocinfor r r �load_all| s
zProcInfoLoader.load_allN)r r r r@ r$ rE rI rL r r r r rA d s
rA c @ s e Zd ZdZdZdS )�
ResourceErrorz*Error raise when resource limit is reachedzUnknown Resource LimitN)r r r r@ �limitr r r r rM � s rM c @ s e Zd ZdZdZdS )rM z*Error raise when resource limit is reachedzUnknown Resource LimitN)r r r r@ rN r r r r rM � s c @ s e Zd ZdZdZdS )� XCPUErrorz+Error raised when CPU Time limit is reachedzCPU TimeN)r r r r@ rN r r r r rO � s rO 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@ rN r r r r rP � s rP c @ s e Zd ZdZdZdS )� TimeoutErrorz9Error raised when the process is running for to much timez Real TimeN)r r r r@ rN r r r r rQ � s rQ 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) rW rU rY r r r r$ � s
zMemorySentinel.__init__c C s | j �� dS )zstop apN)rV �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) rA rV �isSetrU rI rY r- r �killpgr
�waitrW )r) Zpilr r r rT � s
zMemorySentinel._runN) r r r r@ r �getpidr$ r[ rT r r r r rR � s rR 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) rf rd re Z max_repriever r r r$ � s" zResourceController.__init__c C s
t � �d S )N)rQ )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 )Nr: zMemory limit reached)rp rg rP r r] r_ r )r) rs rt 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 )Nr: zSoft CPU time limit reached)rp rg rO r r] r_ r )r) rs rt 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 )Nr: r ) rp rg r r] r_ r ro r � _time_outrh �startr )r) r r r rx � 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 Nr: )$r �getName�AssertionErrorr �setpgrpro rd r r ru rm r �maxr# rr rx rh r rq ry rf r r rl r rw rn r re rR ri r rj r
rv rk )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 Nr: )ro rd rh �cancelrr r# r rq r r rm rf r r rl r rn re ri r[ r rj r
rk )r) r r r � clean_limit s
z ResourceController.clean_limit)NNNra )
r r r r$ ru rv rw rx r r� r r r r r` � s
r` )+r@ �
__docformat__r r= �resourcer r r r r r r r r
� threadingr r r
r r Zlogilab.common.treer � Exceptionr r r' r3 r4 r5 r6 r* r" rA �
BaseExceptionrM � NameErrorrO rP rQ � MemoryErrorZRESOURCE_LIMIT_EXCEPTIONrR r` r r r r �<module> s8 ,$