OwlCyberSecurity - MANAGER
Edit File: data.cpython-36.pyc
3 �bfZ< � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl m Z ej j d�Zyd dlmZ e� ZW n dZY nX dZe s�e r�ed� ed � es�ed � es�ed � ejd � d d lmZmZ d dlmZmZmZm 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!�Z-G d"d#� d#e!�Z.d$d%� Z/dS )&� )�print_functionN)�cmdlinez /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__R 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�memberr0 r r r � accumulate] s zStats.accumulatec C s | j ||dd�S )z*Update destination with self - other_statsr )r5 r )r7 )r r3 r4 r r r �deltae s z Stats.deltac C s. | j }x"tjD ]\}}|| dkrdS qW dS )Nr FT)r r r) )r r. r/ r0 r r r � is_all_zeroi s zStats.is_all_zeroc C s0 t jt �} | j}xt jD ]\}}d||< qW | S )Nr )r �__new__r r) )�statsZstdr/ r0 r r r �build_all_zerop s zStats.build_all_zero)r r )r r )r! r" )r# r$ )r% r& )r ) r r r r) r, r2 r7 r8 r9 � staticmethodr< r r r r r G 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 rA Z controllerr r r r2 � 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| 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�sendrB r Zrecv�OSError�errnoZESRCHrE �items�TASKSTATS_TYPE_AGGR_PID�nested�TASKSTATS_TYPE_STATSr1 �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 r2 rJ rY 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)rA �error�u�uidZpasswdr r r � find_uids� s ri 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 rr 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 rx c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � ThreadInfozStats for a single threadc C s. || _ d| _d | _tjt�| _|j|�| _d S )NT)rI �mark� stats_totalr r: � stats_deltarJ rM )r rI �taskstats_connectionr r r r2 � s zThreadInfo.__init__c C s t j| j�S )N)r �getrI )r r r r � get_ioprio� s zThreadInfo.get_ioprioc C s t jt j| j||�S )N)r � set_ioprioZIOPRIO_WHO_PROCESSrI )r � ioprio_class� ioprio_datar r r r� � s zThreadInfo.set_ioprioc C s& | j s || _ |j| j | j� || _ d S )N)r{ r8 r| )r r; r r r � update_stats� s zThreadInfo.update_statsN)r r r r r2 r r� r� r r r r ry � s ry 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) rn rh �user�threadsr r< r| � stats_accum�timeZstats_accum_timestamp)r rn r r r r2 � s zProcessInfo.__init__c C s: |j r |j r | j|j kr dS |jr6| j� |jkr6dS dS )NFT)�pids� processesrn r[ �get_uid)r rA 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)rh �os�statrn �ST_UIDrO r� )r rh 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� rx r^ �getpwuidZpw_namer` rv r )r rh 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 � ) rj rn �readrm rr r~ r\ rk � startswith�rfind�joinrl rx ) r Z proc_cmdliner Z 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� r9 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>E s z)ProcessInfo.get_ioprio.<locals>.<genexpr>r z?dif)�setr� r� rU �pop)r Z prioritiesr r r r D s zProcessInfo.get_ioprioc C s$ x | j j� D ]}|j||� q W d S )N)r� r� r� )r r� r� rW r r r r� J s zProcessInfo.set_ioprioc C s t j| j� �S )N)r Zsort_keyr )r r r r �ioprio_sort_keyN s zProcessInfo.ioprio_sort_keyc C s* | j j|d �}|s&t||�}|| j |< |S )N)r� r~ ry )r rI r} rW r r r � get_threadQ 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 )rz )r� rI rW r r r � <listcomp>] s z,ProcessInfo.update_stats.<locals>.<listcomp>FT) r r< r� rQ rz r7 r| �dictrU r r r� )r r| rI rW Z nr_threadsr r r r� X s zProcessInfo.update_statsN)r r r r r2 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} rA r� � timestampr r �update_process_counts)r r} rA r r r r2 m 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� rA )r rn �processr r r � get_processx 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) rA r� r� �listdirr� �extend�mapr\ rO )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) 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 � }|| 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} rY r� r| r! r# rz r r8 ) r Z new_timestampZ total_readZ total_writer� r� rI rW r; r8 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� rn r� r r r r� � s z1ProcessList.refresh_processes.<locals>.<listcomp>)r� r� r� rz r� r� rQ )r r� rW 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 r2 r� r� r� r� r� r� r r r r r� l s r� c C s dt � j� kS )a| WAS: try: with open('/proc/sys/kernel/task_delayacct') as f: return bool(int(f.read().strip())) except FileNotFoundError: return None Because /proc/sys/kernel/task_delayacct doesn't exist on RHEL8, it always returns None, which is equivalent to False in the end. On RHEL8, delayacct_on kernel variable is enabled by default Z nodelayacct)r �keysr r r r r- � s r- )0Z __future__r rP r� r r^ r� r* rb r� Zprocfsr �path�existsZ ioaccountingZ iotop.vmstatr Zvmstat_fZvm_event_countersra re Ziotopr r Z iotop.netlinkr r r r Ziotop.genetlinkr r �objectr r rG rH rR ZTASKSTATS_TYPE_PIDrT r@ ri rr rx ry r� r� r- r r r r �<module> sV 5) ^