| Current File : //proc/self/root/lib/python3.6/site-packages/iotop/__pycache__/data.cpython-36.pyc |
3
q�]b,; �
@ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z ej
j d�Z yd dl
mZ e� ZW n dZY nX dZe s�e r�ed� ed� e s�ed � es�ed
� ejd � d d lmZmZ d d
lmZmZmZmZ d dlm Z m Z G dd� de �Z G dd� de �Z d Z!d Z"dZ#d Z$dZ%G dd� de �Z&dd� Z'dd� Z(dd � Z)G d d � d e �Z*G d d � d e �Z+G d!d"� d"e �Z,d#d$� Z-dS )%� )�print_functionNz
/proc/self/io)�VmStatFTz<Could not run iotop as some of the requirements are not met:z- Linux >= 2.6.20 withz~ - I/O accounting support (CONFIG_TASKSTATS, CONFIG_TASK_DELAY_ACCT, CONFIG_TASK_IO_ACCOUNTING, kernel.task_delayacct sysctl)z0 - VM event counters (CONFIG_VM_EVENT_COUNTERS)� )�ioprio�vmstat)�
Connection�NETLINK_GENERIC�U32Attr�
NLM_F_REQUEST)�
Controller� GeNlMessagec @ s e Zd ZdZdd� ZdS )�DumpableObjectzFBase class for all objects that allows easy introspection when printedc C s$ dt t| ��d d� tj| j�f S )Nz%s: %s>r ���)�str�type�pprintZpformat�__dict__)�self� r �/usr/lib/python3.6/data.py�__repr__> s zDumpableObject.__repr__N)�__name__�
__module__� __qualname__�__doc__r r r r r r
< s r
c @ sL e Zd ZdddddgZd Zd d
� Zd dd�Zdd� Zdd� Ze dd� �Z
d S ) �Stats�blkio_delay_total�( �swapin_delay_total�8 �
read_bytes� � write_bytes� �cancelled_write_bytes� Nc C sP | j }x6tjD ],\}}|||d � }tjd|�d ||< qW tjsLt� t_d S )N� �Qr )r r �members_offsets�struct�unpack�has_blkio_delay_total�sysctl_task_delayacct)r Ztask_stats_buffer�sd�name�offset�datar r r �__init__Q s zStats.__init__r c C sB |j }| j }|j }x*tjD ] \}}|| ||| ||< qW dS )z3Update destination from operator(self, other_stats)N)r r r( ) r � other_stats� destination�coeffZddr- Zod�memberr/ r r r �
accumulate\ s
zStats.accumulatec C s | j ||dd�S )z*Update destination with self - other_statsr )r4 r )r6 )r r2 r3 r r r �deltad s z Stats.deltac C s. | j }x"tjD ]\}}|| dkrdS qW dS )Nr FT)r r r( )r r- r. r/ r r r � is_all_zeroh s
zStats.is_all_zeroc C s0 t jt �} | j}xt jD ]\}}d||< qW | S )Nr )r �__new__r r( )�statsZstdr. r/ r r r �build_all_zeroo s
zStats.build_all_zero)r r )r r )r r! )r" r# )r$ r% )r ) r r r r( r+ r1 r6 r7 r8 � staticmethodr; r r r r r F s
r � � c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TaskStatsNetlinkc C s* || _ tt�| _t| j�}|jd�| _d S )NZ TASKSTATS)�optionsr r �
connectionr Z
get_family_id� family_id)r r@ Z
controllerr r r r1 � s
zTaskStatsNetlink.__init__c C s t | jttt|�gtd�S )N)�cmd�attrs�flags)r rB �TASKSTATS_CMD_GETr �TASKSTATS_CMD_ATTR_PIDr
)r �tidr r r �
build_request� s
z TaskStatsNetlink.build_requestc C s� |j j| j� ytj| j�}W n4 tk
rR } z|jtjkr@d S � W Y d d }~X nX x,|jj � D ]\}}|t
kr`|j � }P q`W d S |t j
}t|�dk r�d S tjd|d d� �d }|dks�t�t|�S )Ni �H� r r= )�task_stats_request�sendrA r Zrecv�OSError�errnoZESRCHrD �items�TASKSTATS_TYPE_AGGR_PID�nested�TASKSTATS_TYPE_STATSr0 �lenr) r* �AssertionErrorr )r �threadZreply�eZ attr_typeZ
attr_valueZtaskstats_dataZtaskstats_versionr r r �get_single_task_stats� s$
z&TaskStatsNetlink.get_single_task_statsN)r r r r1 rI rX r r r r r? � s r? c C s� g | _ d}x�| jpg D ]t}y t|�}W nR tk
rx ytj|�}W n( tk
rl td|tj d� d}Y nX |j
}Y nX |s| j j |� qW |r�tj d� dS )zDBuild options.uids from options.users by resolving usernames to UIDsFz
Unknown user:)�fileTr N)
�uidsZusers�int�
ValueError�pwd�getpwnam�KeyError�print�sys�stderrZpw_uid�append�exit)r@ �error�u�uidZpasswdr r r � find_uids� s
rh c C sT i }y6x0t d| �D ] }|jdd�\}}|j� ||< qW W n tk
rN Y nX |S )Nz/proc/%d/statusz: r )�open�split�strip�IOError)�pidZ result_dict�line�key�valuer r r �parse_proc_pid_status� s rq c C s< y
| j d�S tk
r$ | jd�S tk
r6 | S X d S )Nzutf-8Z
string_escape)�decode�UnicodeDecodeError�encode�AttributeError)�sr r r �safe_utf8_decode� s
rw c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �
ThreadInfozStats for a single threadc C s. || _ d| _d | _tjt�| _|j|�| _d S )NT)rH �mark� stats_totalr r9 � stats_deltarI rL )r rH �taskstats_connectionr r r r1 � s
zThreadInfo.__init__c C s t j| j�S )N)r �getrH )r r r r �
get_ioprio� s zThreadInfo.get_ioprioc C s t jt j| j||�S )N)r �
set_ioprioZIOPRIO_WHO_PROCESSrH )r � ioprio_class� ioprio_datar r r r � s zThreadInfo.set_ioprioc C s& | j s || _ |j| j | j� || _ d S )N)rz r7 r{ )r r: r r r � update_stats� s zThreadInfo.update_statsN)r r r r r1 r~ r r� r r r r rx � s
rx c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )� ProcessInfoz�Stats for a single process (a single line in the output): if
options.processes is set, it is a collection of threads, otherwise a single
thread.c C s: || _ d | _d | _i | _tj� | _tj� | _tj� | _ d S )N)
rm rg �user�threadsr r; r{ � stats_accum�timeZstats_accum_timestamp)r rm r r r r1 � s
zProcessInfo.__init__c C s: |j r |j r | j|j kr dS |jr6| j� |jkr6dS dS )NFT)�pids� processesrm rZ �get_uid)r r@ r r r � is_monitored� s zProcessInfo.is_monitoredc C sZ | j r | j S ytjd| j �tj }W n tk
r> d }Y nX || j krVd | _|| _ |S )Nz/proc/%d)rg �os�statrm �ST_UIDrN r� )r rg r r r r� s
zProcessInfo.get_uidc C s\ | j � }|d k rR| j rRyttj|�j�| _W n" ttfk
rP t|�| _Y nX | jpZdS )Nz{none}) r� r� rw r] �getpwuidZpw_namer_ ru r )r rg r r r �get_user s zProcessInfo.get_userc C s
y t d| j �}|jd�}W n tk
r0 dS X t| j�}|sb|jdd�}|rZd| }nd}|S d}t|jd| j��}|| jkr�t|�jdd�}|jdd�}||kr�|d | 7 }|jd
�} | d jd �r�| d j d �d
}
| d |
d � | d < dj
| �j � }t || �S )Nz/proc/%d/cmdlinei z{no such process}�Name� z[%s]z {no name}ZTgidz [%s]� r �/r � )
ri rm �readrl rq r} r[ rj �
startswith�rfind�joinrk rw ) r Z proc_cmdlineZcmdlineZ proc_statusZproc_status_name�suffix�tgidZ tgid_nameZ thread_name�partsZfirst_command_charr r r � get_cmdline s2
zProcessInfo.get_cmdlinec C s6 |r| j j� S x | jj� D ]}|jj� s dS q W dS )NTF)r� r8 r� �valuesr{ )r Z accumulated�tr r r � did_some_io; s
zProcessInfo.did_some_ioc C s0 t dd� | jj� D ��}t|�dkr,|j� S dS )Nc s s | ]}|j � V qd S )N)r~ )�.0r� r r r � <genexpr>D s z)ProcessInfo.get_ioprio.<locals>.<genexpr>r z?dif)�setr� r� rT �pop)r Z
prioritiesr r r r~ C s zProcessInfo.get_ioprioc C s$ x | j j� D ]}|j||� q W d S )N)r� r� r )r r� r� rV r r r r I s zProcessInfo.set_ioprioc C s t j| j� �S )N)r Zsort_keyr~ )r r r r �ioprio_sort_keyM s zProcessInfo.ioprio_sort_keyc C s* | j j|d �}|s&t||�}|| j |< |S )N)r� r} rx )r rH r| rV r r r �
get_threadP s
zProcessInfo.get_threadc C s� t j� }x*| jj� D ] \}}|js|j|j|� qW tdd� | jj� D ��| _t| j�}|s`dS | j | _ | j
| _
|| _| j j| j| j � dS )Nc S s g | ]\}}|j s||f�qS r )ry )r� rH rV r r r �
<listcomp>\ s z,ProcessInfo.update_stats.<locals>.<listcomp>FT) r r; r� rP ry r6 r{ �dictrT r r r� )r r{ rH rV Z
nr_threadsr r r r� W s
zProcessInfo.update_statsN)r r r r r1 r� r� r� r� r� r~ r r� r� r� r r r r r� � s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� ProcessListc C s2 i | _ || _|| _tj� | _tj� | _| j� d S )N)r� r| r@ r� � timestampr r �update_process_counts)r r| r@ r r r r1 l s
zProcessList.__init__c C s8 | j j|d�}|s$t|�}|| j |< |j| j�r4|S dS )zyEither get the specified PID from self.processes or build a new
ProcessInfo if we see this PID for the first timeN)r� r} r� r� r@ )r rm �processr r r � get_processw s
zProcessList.get_processc
C s� | j jr| j jS tjd�}| j jr0dd� |D �S g }x`|D ]X}d|d koTdkn r:y"|jtttjd| d ��� W q: tk
r� Y q:X q:W |S ) Nz/procc S s0 g | ](}d |d ko dkn rt |��qS )�0r �9)r[ )r� r� r r r r� � s z*ProcessList.list_tgids.<locals>.<listcomp>r� r r� z/proc/z/task) r@ r� r� �listdirr� �extend�mapr[ rN )r Ztgids�tidsr� r r r �
list_tgids� s
"
zProcessList.list_tgidsc C sf | j js|gS y ttttjd| ���}W n tk
r> g S X | j jrbtt | j j�j
t |���}|S )Nz
/proc/%d/task) r@ r� �listr� r[ r� r� rN r� r� � intersection)r r� r� r r r � list_tids� s zProcessList.list_tidsc
C s� t j � }|| j | _|| _d }}x|| j� D ]p}| j|�}|s@q,xZ| j|�D ]L}|j|| j�}| jj|�}|rL|j |� |j
} || j 7 }|| j 7 }d|_
qLW q,W ||f| jj� fS )Nr F)r� r� Zdurationr� r� r� r� r| rX r� r{ r r" ry r r7 )
r Z
new_timestampZ
total_readZ total_writer� r� rH rV r: r7 r r r r� � s$
z!ProcessList.update_process_countsc C sT x,| j j� D ] }x|jj� D ]
}d|_q W q W | j� }tdd� | j j� D ��| _ |S )NTc S s g | ]\}}|j � r||f�qS r )r� )r� rm r� r r r r� � s z1ProcessList.refresh_processes.<locals>.<listcomp>)r� r� r� ry r� r� rP )r r� rV Ztotal_read_and_writer r r �refresh_processes� s z ProcessList.refresh_processesc C s
i | _ d S )N)r� )r r r r �clear� s zProcessList.clearN)
r r r r1 r� r� r� r� r� r� r r r r r� k s
r� c C sB y(t d��} tt| j� j� ��S Q R X W n tk
r< d S X d S )Nz /proc/sys/kernel/task_delayacct)ri �boolr[ r� rk �FileNotFoundError)�fr r r r, � s
r, ).Z
__future__r rO r� r r] r� r) ra r� �path�existsZ ioaccountingZ iotop.vmstatr Zvmstat_fZvm_event_countersr` rd Ziotopr r Z
iotop.netlinkr r r r
Ziotop.genetlinkr r �objectr
r rF rG rQ ZTASKSTATS_TYPE_PIDrS r? rh rq rw rx r� r� r, r r r r �<module> sT
5) ^