|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/iotop/__pycache__/ |
Upload File : |
o
�i�d+< � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z ze
e d�s5d dl m Z e e _ W n e e
fyE e j e _ Y nw ej�d�Zz d dlmZ e� ZW n eyb dZY nw dZeries�ed � ed
� eswed � es}ed � 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�pprint�pformat�__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 @ 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� )�cancelled_write_bytesi Fc C sT | j }tjD ]\}}|||d � }t�d|�d ||< qtjs(| jdkt_d S d S )N� �Qr )r r �members_offsets�struct�unpack�has_blkio_delay_totalr )r �task_stats_buffer�sd�name�offset�datar r r �__init___ s �zStats.__init__r c C s> |j }| j }|j }tjD ]\}}|| ||| ||< q dS )z3Update destination from operator(self, other_stats)N�r r r* ) r � other_stats� destination�coeff�ddr/ �od�memberr1 r r r �
accumulatej s �zStats.accumulatec C s | j ||dd�S )z*Update destination with self - other_statsr )r7 )r; )r r5 r6 r r r �deltar s z Stats.deltac C s, | j }tjD ]
\}}|| dkr dS qdS )Nr FTr4 )r r/ r0 r1 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* )�stats�stdr0 r1 r r r �build_all_zero} s
zStats.build_all_zeroN)r ) r r r r* r- r3 r; r< r= � staticmethodrA 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 )N� TASKSTATS)�optionsr r �
connectionr �
get_family_id� family_id)r rG �
controllerr r r r3 � s
zTaskStatsNetlink.__init__c C s t | jttt|�gtd�S )N)�cmd�attrs�flags)r
rJ �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|jtjkr$W Y d }~d S � d }~ww |j� � D ]\}}|t
kr=|� � } nq/d S |t j
}t|�dk rMd S t�d|d d� �d }|dks_J �t|�S )Ni �H� r rC )�task_stats_request�sendrH r
�recv�OSError�errno�ESRCHrM �items�TASKSTATS_TYPE_AGGR_PID�nested�TASKSTATS_TYPE_STATSr2 �lenr+ r, r )r �thread�reply�e� attr_type�
attr_value�taskstats_data�taskstats_versionr r r �get_single_task_stats� s* ���
z&TaskStatsNetlink.get_single_task_statsN)r r r r3 rR rg r r r r rE � s rE c C s� g | _ d}| jp g D ]8}zt|�}W n' ty9 zt�|�}W n ty3 td|tj d� d}Y nw |j
}Y nw |sB| j � |� q
|rLt� d� dS dS )zDBuild options.uids from options.users by resolving usernames to UIDsFz
Unknown user:)�fileTr N)
�uids�users�int�
ValueError�pwd�getpwnam�KeyError�print�sys�stderr�pw_uid�append�exit)rG �error�u�uid�passwdr r r � find_uids� s* ��� ��rz c C sR i }z t d| dd�D ]}|�dd�\}}|�� ||< q W |S ty( Y |S w )Nz/proc/%d/status�replace)�errors�:r )�open�split�strip�IOError)�pid� result_dict�line�key�valuer r r �parse_proc_pid_status� s � ��r� c C s> z| � d�W S ty | �d� Y S ty | Y S w )Nzutf-8�
string_escape)�decode�UnicodeDecodeError�encode�AttributeError)�sr r r �safe_utf8_decode� s �r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �
ThreadInfozStats for a single threadc C s. || _ d| _d | _t�t�| _|�|�| _d S )NT)rQ �mark� stats_totalr r> � stats_deltarR rU )r rQ �taskstats_connectionr r r r3 � s
zThreadInfo.__init__c C s t �| j�S �N)r �getrQ r r r r �
get_ioprio� s zThreadInfo.get_ioprioc C s t �t j| j||�S r� )r �
set_ioprio�IOPRIO_WHO_PROCESSrQ )r � ioprio_class� ioprio_datar r r r� � s �zThreadInfo.set_ioprioc C s& | j s|| _ |�| j | j� || _ d S r� )r� r< r� )r r? r r r � update_stats� s
zThreadInfo.update_statsN)r r r r r3 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� ) r� rx �user�threadsr rA r� � stats_accum�timer �stats_accum_timestamp)r r� r r r r3 � s
zProcessInfo.__init__c C s8 |j r|js| j|j vrdS |jr| �� |jvrdS dS )NFT)�pids� processesr� ri �get_uid)r rG r r r � is_monitored s zProcessInfo.is_monitoredc C sX | j r| j S z
t�d| j �tj }W n ty d }Y nw || j kr*d | _|| _ |S )Nz/proc/%d)rx �os�statr� �ST_UIDrX r� �r rx r r r r� s �
zProcessInfo.get_uidc C sX | � � }|d ur'| js'z tt�|�j�| _W n ttfy& t|�| _Y nw | jp+dS )Nz{none}) r� r� r� rm �getpwuid�pw_namero r� r r� r r r �get_user$ s �
zProcessInfo.get_userc C s zt d| j �}|�d�}W n
ty Y dS w t| j�}|s2|�dd�}|r.d| }|S d}|S d}t|�d| j��}|| jkrZt|��dd�}|�dd�}||krZ|d | 7 }|�d
�} | d �d �ry| d � d �d
}
| d |
d � | d < d�
| �� � }t || �S )Nz/proc/%d/cmdlinei z{no such process}�Name� z[%s]z {no name}�Tgidz [%s]� r �/r � )
r~ r� �readr� r� r� rk r �
startswith�rfind�joinr� r� ) r � proc_cmdline�cmdline� proc_status�proc_status_name�suffix�tgid� tgid_name� thread_name�parts�first_command_charr r r � get_cmdline- s6 �
�
zProcessInfo.get_cmdlinec C s4 |r| j �� S | j�� D ]
}|j�� s dS q
dS )NTF)r� r= r� �valuesr� )r � 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� r_ �pop)r �
prioritiesr r r r� T s zProcessInfo.get_ioprioc C s | j �� D ]}|�||� qd S r� )r� r� r� )r r� r� r` r r r r� Z s �zProcessInfo.set_ioprioc C s t �| �� �S r� )r �sort_keyr� r r r r �ioprio_sort_key^ s zProcessInfo.ioprio_sort_keyc C s* | j �|d �}|st||�}|| j |< |S r� )r� r� r� )r rQ r� r` r r r �
get_threada s
zProcessInfo.get_threadc C s� t �� }| j�� D ]\}}|js|�|j|� q tdd� | j�� D ��| _t| j�}|s.dS | j | _ | j
| _
|| _| j �| j| j � dS )Nc S s g | ] \}}|j s||f�qS r )r� )r� rQ r` r r r �
<listcomp>m s �z,ProcessInfo.update_stats.<locals>.<listcomp>FT) r rA r� r[ r� r; r� �dictr_ r r! r� )r r� rQ r` �
nr_threadsr r r r� h s �
�
zProcessInfo.update_statsN)r r r r r3 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� rG r� r � timestampr r �update_process_counts)r r� rG r r r r3 ~ s
zProcessList.__init__c C s8 | j �|d�}|st|�}|| j |< |�| j�r|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� rG )r r� �processr r r � get_process� s
�zProcessList.get_processc C s� | j jr| j jS t�d�}| j jrdd� |D �S g }|D ]+}d|d kr*dkrGn q z|�ttt�d| d ��� W q tyF Y q w q |S ) Nz/procc S s0 g | ]}d |d krdkrn nt |��qS )�0r �9)rk )r� r� r r r r� � s 0 z*ProcessList.list_tgids.<locals>.<listcomp>r� r r� z/proc/z/task) rG r� r� �listdirr� �extend�maprk rX )r �tgids�tidsr� r r r �
list_tgids� s"
� ��zProcessList.list_tgidsc C sh | j js|gS ztttt�d| ���}W n ty g Y S w | j jr2tt | j j��
t |���}|S )Nz
/proc/%d/task) rG r� �listr� rk r� r� rX r� r� � intersection)r r� r� r r r � list_tids� s �zProcessList.list_tidsc
C s� t �� }|| j | _|| _d }}| �� D ]6}| �|�}|s q| �|�D ]&}|�|| j�}| j� |�}|rJ|�
|� |j } || j 7 }|| j
7 }d|_q$q||f| j�� fS )Nr F)r� r r� �durationr� r� r� r� r� rg r� r� r# r% r� r r< )
r �
new_timestamp�
total_read� total_writer� r� rQ r` r? r< r r r r� � s(
�� z!ProcessList.update_process_countsc C sL | j �� D ]
}|j�� D ]}d|_q q| �� }tdd� | j �� D ��| _ |S )NTc S s g | ] \}}|� � r||f�qS r )r� )r� r� r� r r r r� � s �z1ProcessList.refresh_processes.<locals>.<listcomp>)r� r� r� r� r� r� r[ )r r� r` �total_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 r3 r� r� r� r� r� r� r r r r r� } s
r� )2�
__future__r rY r� r rm r� r+ rq r� �hasattrr � ImportError� RuntimeError�path�exists� ioaccounting� iotop.vmstatr �vmstat_fr� �vm_event_countersrp ru �iotopr r �
iotop.netlinkr r r
r �iotop.genetlinkr r
�objectr r rO rP r\ �TASKSTATS_TYPE_PIDr^ rE rz r� r� r� r� r� r r r r �<module> sf
� �
�
5*