Your IP : 216.73.216.172
U
�Z�d+< � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z z e
e d�sjd dl m Z e e _ W n e e
fk
r� e j e _ Y nX ej�d�Zzd dlmZ e� ZW n ek
r� dZY nX dZer�e�s ed � ed
� es�ed � e�sed � e�d
� d dlmZmZ d dlm Z m Z m Z m Z d dl m!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-d d � Z.G d d � d e$�Z/G d!d"� d"e$�Z0G d#d$� d$e$�Z1dS )%� )�print_functionN� monotonic)r z
/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)z0 - VM event counters (CONFIG_VM_EVENT_COUNTERS)� )�ioprio�vmstat)�
Connection�NETLINK_GENERIC�U32Attr�
NLM_F_REQUEST)�
Controller� GeNlMessagec @ s e Zd ZdZdd� ZdS )�DumpableObjectzBBase class for objects that allows easy introspection when printedc C s$ dt t| ��d d� t�| j�f S )Nz%s: %s>���)�str�type�pprintZpformat�__dict__��self� r �,/usr/lib/python3/dist-packages/iotop/data.py�__repr__K s
�zDumpableObject.__repr__N)�__name__�
__module__� __qualname__�__doc__r r r r r r I s r c @ sL e Zd ZdddddgZdZdd� Zdd
d �Zd d
� Zdd� Ze dd� �Z
dS )�Stats)�blkio_delay_total�( )�swapin_delay_total�8 )�
read_bytes� )� write_bytes� )Zcancelled_write_bytesi Fc C sP | j }tjD ],\}}|||d � }t�d|�d ||< q tjsL| jdkt_d S )N� �Qr )r r �members_offsets�struct�unpack�has_blkio_delay_totalr )r Ztask_stats_buffer�sd�name�offset�datar r r �__init___ s zStats.__init__r c C s> |j }| j }|j }tjD ] \}}|| ||| ||< qdS )z3Update destination from operator(self, other_stats)N�r r r( ) r � other_stats� destination�coeffZddr, Zod�memberr. r r r �
accumulatej s
zStats.accumulatec C s | j ||dd�S )z*Update destination with self - other_statsr )r4 )r6 )r r2 r3 r r r �deltar s z Stats.deltac C s, | j }tjD ]\}}|| dkr dS q dS )Nr FTr1 )r r, r- r. r r r � is_all_zerov s
zStats.is_all_zeroc C s, t �t �} | j}t jD ]\}}d||< q| S )Nr )r �__new__r r( )�statsZstdr- r. r r r �build_all_zero} s
zStats.build_all_zeroN)r ) r r r r( r+ r0 r6 r7 r8 � staticmethodr; r r r r r T s �
r � � c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TaskStatsNetlinkc C s* || _ tt�| _t| j�}|�d�| _d S )NZ TASKSTATS)�optionsr r �
connectionr Z
get_family_id� family_id)r r@ Z
controllerr r r r0 � 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� zt�| j�}W n: tk
rX } z |jtjkrFW Y �
d S � W 5 d }~X Y nX |j� � D ] \}}|t
krd|� � } q�qdd S |t j
}t|�dk r�d S t�d|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_STATSr/ �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 r0 rI rX r r r r r? � s r? c C s� g | _ d}| jpg D ]t}z t|�}W nR tk
rv zt�|�}W n( tk
rj td|tj d� d}Y nX |j
}Y nX |s| j � |� q|r�t� 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 }z6t d| dd�D ] }|�dd�\}}|�� ||< qW n tk
rN Y nX |S )Nz/proc/%d/status�replace)�errors�:r )�open�split�strip�IOError)�pidZ result_dict�line�key�valuer r r �parse_proc_pid_status� s rt c C sF z | � d�W S tk
r* | �d� Y S tk
r@ | Y S X d S )Nzutf-8Z
string_escape)�decode�UnicodeDecodeError�encode�AttributeError)�sr r r �safe_utf8_decode� s rz c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �
ThreadInfozStats for a single threadc C s. || _ d| _d | _t�t�| _|�|�| _d S )NT)rH �mark� stats_totalr r9 � stats_deltarI rL )r rH �taskstats_connectionr r r r0 � s
zThreadInfo.__init__c C s t �| j�S �N)r �getrH r r r r �
get_ioprio� s zThreadInfo.get_ioprioc C s t �t j| j||�S r� )r �
set_ioprioZIOPRIO_WHO_PROCESSrH )r � ioprio_class� ioprio_datar r r r� � s �zThreadInfo.set_ioprioc C s& | j s || _ |�| j | j� || _ d S r� )r} r7 r~ )r r: r r r � update_stats� s zThreadInfo.update_statsN)r r r r r0 r� r� r� r r r r r{ � s
r{ 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 | _t�� | _t�� | _t� � | _
d S r� ) rp rg �user�threadsr r; r~ � stats_accum�timer Zstats_accum_timestamp)r rp r r r r0 � s
zProcessInfo.__init__c C s8 |j r |js | j|j kr dS |jr4| �� |jkr4dS dS )NFT)�pids� processesrp rZ �get_uid)r r@ r r r � is_monitored s
�zProcessInfo.is_monitoredc C sZ | j r | j S zt�d| j �tj }W n tk
r> d }Y nX || j krVd | _|| _ |S )Nz/proc/%d)rg �os�statrp �ST_UIDrN r� �r rg r r r r� s
zProcessInfo.get_uidc C sZ | � � }|d k rP| jsPztt�|�j�| _W n" ttfk
rN t|�| _Y nX | jpXdS )Nz{none}) r� r� rz r] �getpwuidZpw_namer_ rx r r� r r r �get_user$ s zProcessInfo.get_userc C s z t d| j �}|�d�}W n tk
r2 Y dS X t| j�}|sd|�dd�}|r\d| }nd}|S d}t|�d| j��}|| jkr�t|��dd�}|�dd�}||kr�|d | 7 }|�d
�} | d �d �r�| d � d �d
}
| d |
d � | d < d�
| �� � }t || �S )Nz/proc/%d/cmdlinei z{no such process}�Name� z[%s]z {no name}ZTgidz [%s]� r �/r � )
rl rp �readro rt r� r[ rm �
startswith�rfind�joinrn rz ) 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 s4 |r| j �� S | j�� D ]}|j�� s dS qdS )NTF)r� r8 r� �valuesr~ )r Z accumulated�tr r r � did_some_ioL s
zProcessInfo.did_some_ioc C s0 t dd� | j�� D ��}t|�dkr,|�� S dS )Nc s s | ]}|� � V qd S r� )r� )�.0r� r r r � <genexpr>U s z)ProcessInfo.get_ioprio.<locals>.<genexpr>r z?dif)�setr� r� rT �pop)r Z
prioritiesr r r r� T s zProcessInfo.get_ioprioc C s | j �� D ]}|�||� q
d S r� )r� r� r� )r r� r� rV r r r r� Z s zProcessInfo.set_ioprioc C s t �| �� �S r� )r Zsort_keyr� r r r r �ioprio_sort_key^ s zProcessInfo.ioprio_sort_keyc C s* | j �|d �}|s&t||�}|| j |< |S r� )r� r� r{ )r rH r rV r r r �
get_threada s
zProcessInfo.get_threadc C s� t �� }| j�� D ] \}}|js|�|j|� qtdd� | j�� D ��| _t| j�}|s\dS | j | _ | j
| _
|| _| j �| j| j � dS )Nc S s g | ]\}}|j s||f�qS r )r| )r� rH rV r r r �
<listcomp>m s �z,ProcessInfo.update_stats.<locals>.<listcomp>FT) r r; r� rP r| r6 r~ �dictrT r r r� )r r~ rH rV Z
nr_threadsr r r r� h s �
zProcessInfo.update_statsN)r r r r r0 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 | _ || _|| _t�� | _t�� | _| �� d S r� ) r� r r@ r� r � timestampr r �update_process_counts)r r r@ r r r r0 ~ s
zProcessList.__init__c C s8 | j �|d�}|s$t|�}|| 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 rp �processr r r � get_process� s
zProcessList.get_processc C s� | j jr| j jS t�d�}| j jr0dd� |D �S g }|D ]X}d|d krTdkr8n q8z"|�ttt�d| d ��� W q8 tk
r� Y q8X q8|S ) Nz/procc S s0 g | ](}d |d kr dkrn qt |��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 sj | j js|gS z tttt�d| ���}W n tk
rB g Y S X | j jrftt | 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 | _|| _d }}| �� D ]l}| �|�}|s>q*| �|�D ]L}|�|| j�}| j� |�}|rH|�
|� |j } || j 7 }|| j
7 }d|_qHq*||f| j�� fS )Nr F)r� r r� Zdurationr� r� r� r� r rX r� r~ r" r$ r| 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 sL | j �� D ]}|j�� D ]
}d|_qq
| �� }tdd� | j �� D ��| _ |S )NTc S s g | ]\}}|� � r||f�qS r )r� )r� rp r� r r r r� � s �z1ProcessList.refresh_processes.<locals>.<listcomp>)r� r� r� r| r� r� rP )r r� rV Ztotal_read_and_writer r r �refresh_processes� s
�
z ProcessList.refresh_processesc C s
i | _ d S r� )r� r r r r �clear� s zProcessList.clearN)
r r r r0 r� r� r� r� r� r� r r r r r� } s
r� )2Z
__future__r rO r� r r] r� r) ra r� �hasattrr � ImportError� RuntimeError�path�existsZ ioaccountingZ iotop.vmstatr Zvmstat_fro Zvm_event_countersr` rd Ziotopr r Z
iotop.netlinkr r r
r Ziotop.genetlinkr r
�objectr r rF rG rQ ZTASKSTATS_TYPE_PIDrS r? rh rt rz r{ r� r� r r r r �<module> s^
5*