File: //lib/python3.9/site-packages/iotop/__pycache__/data.cpython-39.opt-1.pyc
a
��Jc,; � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z ej
� d�Z zd dl
mZ e� ZW n dZY n0 dZe r�es�ed� ed� e s�ed � es�ed
� e�d � 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� t�| j�f S )Nz%s: %s>���)�str�type�pprintZpformat�__dict__��self� r �./usr/lib/python3.9/site-packages/iotop/data.py�__repr__> s zDumpableObject.__repr__N)�__name__�
__module__� __qualname__�__doc__r r r r r r
< s r
c @ sF e Zd Zg d�ZdZdd� Zddd�Zdd � Zd
d � Ze d d
� �Z
dS )�Stats))�blkio_delay_total�( )�swapin_delay_total�8 )�
read_bytes� )� write_bytes� )Zcancelled_write_bytesi Nc C sL | j }tjD ],\}}|||d � }t�d|�d ||< q tjsHt� 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 s> |j }| j }|j }tjD ] \}}|| ||| ||< qdS )z3Update destination from operator(self, other_stats)N�r r r' ) r � other_stats� destination�coeff�ddr, Zod�memberr. r r r �
accumulate\ s
zStats.accumulatec C s | j ||dd�S )z*Update destination with self - other_statsr )r4 )r7 )r r2 r3 r r r �deltad s z Stats.deltac C s, | j }tjD ]\}}|| dkr dS q dS )Nr FTr1 )r r, r- r. r r r � is_all_zeroh 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_zeroo s
zStats.build_all_zero)r ) r r r r' r* r0 r7 r8 r9 � staticmethodr< r r r r r F 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 rA Z
controllerr r r r0 � s
zTaskStatsNetlink.__init__c C s t | jttt|�gtd�S )N)�cmd�attrs�flags)r rC �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@ ty^ } z(|jtjkrHW Y d }~d S � W Y d }~n
d }~0 0 |j� � D ] \}}|t
krj|� � } q�qjd S |t j
}t|�dk r�d S t�d|d d� �d }t|�S )Ni �H� r )�task_stats_request�sendrB r Zrecv�OSError�errnoZESRCHrE �items�TASKSTATS_TYPE_AGGR_PID�nested�TASKSTATS_TYPE_STATSr/ �lenr( r) r )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 rJ rX r r r r r@ � s r@ c C s� g | _ d}| jpg D ]p}z t|�}W nN tyr zt�|�}W n& tyf td|tj d� d}Y n0 |j
}Y n0 |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�stderr�pw_uid�append�exit)rA �error�u�uid�passwdr r r � find_uids� s
rj c C sN i }z2t d| �D ] }|�dd�\}}|�� ||< qW n tyH Y n0 |S )Nz/proc/%d/statusz: r )�open�split�strip�IOError)�pidZ result_dict�line�key�valuer r r �parse_proc_pid_status� s rs c C sB z | � d�W S ty( | �d� Y S ty< | Y S 0 d S )Nzutf-8Z
string_escape)�decode�UnicodeDecodeError�encode�AttributeError)�sr r r �safe_utf8_decode� s ry c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �
ThreadInfozStats for a single threadc C s. || _ d| _d | _t�t�| _|�|�| _d S )NT)rI �mark� stats_totalr r: � stats_deltarJ rM )r rI �taskstats_connectionr r r r0 � s
zThreadInfo.__init__c C s t �| j�S �N)r �getrI r r r r �
get_ioprio� s zThreadInfo.get_ioprioc C s t �t j| j||�S r )r �
set_ioprioZIOPRIO_WHO_PROCESSrI )r � ioprio_class� ioprio_datar r r r� � s �zThreadInfo.set_ioprioc C s& | j s || _ |�| j | j� || _ d S r )r| r8 r} )r r; r r r � update_stats� s zThreadInfo.update_statsN)r r r r r0 r� r� r� r r r r rz � s
rz 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 )
ro rh �user�threadsr r< r} � stats_accum�timeZstats_accum_timestamp)r ro r r r r0 � s
zProcessInfo.__init__c C s8 |j r |js | j|j vr dS |jr4| �� |jvr4dS dS )NFT)�pids� processesro rZ �get_uid)r rA r r r � is_monitored� s
�zProcessInfo.is_monitoredc C sX | j r | j S zt�d| j �tj }W n ty< d }Y n0 || j krTd | _|| _ |S )Nz/proc/%d)rh �os�statro �ST_UIDrO r� �r rh r r r r� s
zProcessInfo.get_uidc C sX | � � }|d urN| jsNztt�|�j�| _W n ttfyL t|�| _Y n0 | jpVdS )Nz{none}) r� r� ry r] �getpwuid�pw_namer_ rw r r� r r r �get_user s zProcessInfo.get_userc C s
z t d| j �}|�d�}W n ty0 Y dS 0 t| j�}|sb|�dd�}|rZd| }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 � )
rk ro �readrn rs r� r[ rl �
startswith�rfind�joinrm ry ) 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� r9 r� �valuesr} )r Z accumulated�tr r r � did_some_io; 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>D � z)ProcessInfo.get_ioprio.<locals>.<genexpr>r z?dif)�setr� r� rU �pop)r Z
prioritiesr r r r� C s zProcessInfo.get_ioprioc C s | j �� D ]}|�||� q
d S r )r� r� r� )r r� r� rV r r r r� I s zProcessInfo.set_ioprioc C s t �| �� �S r )r Zsort_keyr� r r r r �ioprio_sort_keyM s zProcessInfo.ioprio_sort_keyc C s* | j �|d �}|s&t||�}|| j |< |S r )r� r� rz )r rI r~ rV r r r �
get_threadP 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� rI rV r r r �
<listcomp>\ s �z,ProcessInfo.update_stats.<locals>.<listcomp>FT) r r< r� rQ r{ r7 r} �dictrU r r r� )r r} rI rV Z
nr_threadsr r r r� W 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~ rA r� � timestampr r �update_process_counts)r r~ rA r r r r0 l 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� rA )r ro �processr r r � get_processw s
zProcessList.get_processc C s� | j jr| j jS t�d�}| j jr0dd� |D �S g }|D ]V}d|d krTdkr8n q8z"|�ttt�d| d ��� W q8 ty� Y q80 q8|S ) Nz/procc S s0 g | ](}d |d kr dkrn qt |��qS )�0r �9)r[ )r� r� r r r r� � r� z*ProcessList.list_tgids.<locals>.<listcomp>r� r r� z/proc/z/task) rA r� r� �listdirr� �extend�mapr[ rO )r Ztgids�tidsr� r r r �
list_tgids� s
" zProcessList.list_tgidsc C sh | j js|gS z tttt�d| ���}W n ty@ g Y S 0 | j jrdtt | j j��
t |���}|S )Nz
/proc/%d/task) rA r� �listr� r[ r� r� rO 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� Zdurationr� r� r� r� r~ rX r� r} r! r# r{ r r8 )
r Z
new_timestampZ
total_readZ total_writer� r� rI rV r; r8 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� ro r� r r r r� � s �z1ProcessList.refresh_processes.<locals>.<listcomp>)r� r� r� r{ r� r� rQ )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� k s
r� c C s\ zBt d��&} tt| �� �� ��W d � W S 1 s60 Y W n tyV Y d S 0 d S )Nz /proc/sys/kernel/task_delayacct)rk �boolr[ r� rm �FileNotFoundError)�fr r r r+ � s
8 r+ ).Z
__future__r rP r� r r] r� r( ra r� �path�existsZ ioaccountingZ iotop.vmstatr Zvmstat_fZvm_event_countersr` re Ziotopr r Z
iotop.netlinkr r r r
Ziotop.genetlinkr r �objectr
r rG rH rR ZTASKSTATS_TYPE_PIDrT r@ rj rs ry rz r� r� r+ r r r r �<module> sT
5) ^