OwlCyberSecurity - MANAGER
Edit File: data.cpython-38.pyc
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*