OwlCyberSecurity - MANAGER
Edit File: ui.cpython-38.pyc
U $�DYOk � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl Z d dl Z z e e d�szd dl m Z e e _ W n eefk r� e j e _ Y nX d dlmZ d dlmZmZmZmZ d dlmZ d dlmZ d d lmZ d d l m Z d d d ddddgZ dd� Z dd� Z dd� Z!dd� Z"dd� Z#e#� Z$G d d � d e%�Z&G d d � d e%�Z'G d d!� d!e%�Z(G d"d#� d#e%�Z)G d$d%� d%e%�Z*G d&d'� d'e+�Z,d(d)� Z-d*d+� Z.d,d-� Z/d.e j0d Z1d/d0� Z2dS )1� )�print_functionN� monotonic)r )� OrderedDict)� find_uids�TaskStatsNetlink� ProcessList�Stats)� ThreadInfo)�VERSION)�ioprio)�IoprioSetError�B�K�M�G�T�P�Ec C s` | dkrd}n| dk r"d}| } ndS t t�| d d�d �}d|t| �dd| > t| f S ) Nr � �-z0.00 B� � z %s%.2f %s� )�int�math�log�float�UNITS)�sizeZsignZexpo� r �*/usr/lib/python3/dist-packages/iotop/ui.py� human_size; s �r! c C s | j rd|d S t|�S )Nz%.2f Kg �@)� kilobytesr! )�options�bytesr r r � format_sizeI s r% c C s t | |ot|�| �d S )Nz/s)r% r )r# r �durationr r r �format_bandwidthO s r' c s� � fdd�}�j r4|j}�fdd�}t�� |j � n|j}�fdd�}||j�}||j�}||j� �}|j |j } t d| �} || � �} |||| fS )Nc s dt d| � d � S )Nz%.2f %%g��(\�X@g �cA)�min)Zdelay)r& r r � delay2percentV s z#format_stats.<locals>.delay2percentc s t � | �S �N)r% �r r& �r# r r �<lambda>Z � z format_stats.<locals>.<lambda>c s t � | |�S r* )r' r+ r, r r r- ^ s �r ) � accumulated� stats_accum�timer Zstats_accum_timestamp� stats_delta�blkio_delay_total�swapin_delay_total� read_bytes� write_bytes�cancelled_write_bytes�max) r# Zprocessr& r) �statsZdisplay_format�io_delay� swapin_delayr5 Z written_bytesr6 r )r& r# r � format_statsS s r< c C sL zt td��� �� �W S tk rF } zt| � W Y �dS d } ~ X Y nX d S )Nz/proc/sys/kernel/pid_max� )�len�open�read�strip� Exception�print)�er r r �get_max_pid_widthi s rE c @ s e Zd ZdS )� CancelInputN��__name__� __module__� __qualname__r r r r rF x s rF c @ s e Zd ZdS )� InvalidIntNrG r r r r rK | s rK c @ s e Zd ZdS )� InvalidPidNrG r r r r rL � s rL c @ s e Zd ZdS )� InvalidTidNrG r r r r rM � s rM c @ s e Zd ZdS )�InvalidIoprioDataNrG r r r r rN � s rN c @ s� e Zd Zdd� dfdd� dfdd� dfdd� dfdd� dfd d� dfd d� dfd d� dfgZd d � Zdd� Zdd� Zdd� Zdd� Zdd� Z d2dd�Z d3dd �Z d d � Z d d � Z d!d"� Zd#d$� Zd4d%d&�Zd5d(d)�Zd*d+� Zd,d-� Zd.d/� Zd0d1� ZdS )6�IOTopUIc C s | j S r* )�pid��p�sr r r r- � r. zIOTopUI.<lambda>Fc C s | � � S r* )Zioprio_sort_keyrQ r r r r- � r. c C s | � � S r* )�get_userrQ r r r r- � r. c C s |j S r* )r5 rQ r r r r- � r. Tc C s |j |j S r* )r6 r7 rQ r r r r- � r. c C s |j S r* )r4 rQ r r r r- � r. c C s |j pt|jp|j �S r* )r3 r r5 r6 rQ r r r r- � s c C s | � � S r* )� get_cmdlinerQ r r r r- � r. c C st || _ || _d| _tj| j d | _| jjsp|| _| �� z t � � t � � t � d� W n t j k rn Y nX d S )N� r r )� process_listr# � sorting_keyrO � sorting_keys�sorting_reverse�batch�win�resize�cursesZuse_default_colorsZ start_color�curs_set�error)�selfr\ rW r# r r r �__init__� s zIOTopUI.__init__c C s | j �� \| _| _d S r* )r\ Zgetmaxyx�height�width�ra r r r r] � s zIOTopUI.resizec C sR d}t �� }| jjs.|�tj�� t jt j B � | jj d ksH|| jj k �rN| j � � \}}| � |dk||| j j� | jj d k r�|d7 }|| jj kr��qNn |dkr�d}z|�| jjd �}W nB t jk r� } z"|jr�|jd tjkr�g }n� W 5 d }~X Y nX |D ]"\}}|t jt jB @ r�t�d� q�| jj�s4| �� |r.| j�� } | �| � q.d S )Nr r g @�@)�select�pollr# r[ �register�sys�stdin�filenoZPOLLINZPOLLPRI� iterationsrW �refresh_processes�refresh_displayr& � delay_secondsr` �args�errnoZEINTRZPOLLERRZPOLLHUP�exitr] r\ �getch� handle_key) ra rl rg �total�currentZeventsrD �fdZevent�keyr r r �run� s> �� z IOTopUI.runc C s | j | _ d S r* )rZ re r r r �reverse_sorting� s zIOTopUI.reverse_sortingc C s2 | j }| �|�| _ || j kr.tj| j d | _d S �Nr )rX �get_sorting_keyrO rY rZ )ra �deltaZorig_sorting_keyr r r �adjust_sorting_key� s zIOTopUI.adjust_sorting_keyc C s0 | j }||7 }td|�}tttj�d |�}|S )Nr r )rX r8 r( r> rO rY )ra r} Znew_sorting_keyr r r r| � s zIOTopUI.get_sorting_keyNc C s� | j �ddtd�tjB | j� | j �dd|tj� | j �� t� � t� d� | j � dt |��}t� d� t� � |dkr~|S |r�t� �|S )Nr r � )Nr )r\ �hline�ordr^ �A_NORMALrd �addstr�A_BOLD�refreshZechor_ Zgetstrr> ZnoechorF �ra �prompt�defaultZempty_is_cancel�inpr r r � prompt_str� s zIOTopUI.prompt_strc C s8 | � |||�}z t|�W S tk r2 t� �Y nX d S r* )r� r � ValueErrorrK r� r r r � prompt_int� s zIOTopUI.prompt_intc C s@ z | � d�W S tk r& t� �Y n tk r: � Y nX d S )NzPID to ionice: )r� rK rL rF re r r r � prompt_pid� s zIOTopUI.prompt_pidc C s@ z | � d�W S tk r& t� �Y n tk r: � Y nX d S )NzTID to ionice: )r� rK rM rF re r r r � prompt_tid s zIOTopUI.prompt_tidc C sd z.|d k r | � d| |d�}n| � dd d�}W n tk rH t� �Y nX |dk sZ|dkr`t� �|S )Nz'I/O priority data (0-7, currently %s): FzI/O priority data (0-7): r � )r� rK rN )ra � ioprio_datar� r r r � prompt_data s � � zIOTopUI.prompt_datac C sh z|� |�}W n tk r& d}Y nX t|�d }| j�ddtd�tjB | j� | j� dd|tj � t|�}t |�D ]D\}}d| } ||kr�tj } ntj} | j� d|| | � |t| �7 }qv| j� � } | tjtd�fkr�|dkr�|d8 }q4q�| tjtd�fk�r||k �r|d7 }q4q�| tjtd�td �fk�r8|| S | d tjtjtjtd �td �fkr�t� �q�q4d S ) N���r r r z %s �l�r� � � �q�Q)�indexr� r> r\ r� r� r^ r� rd �insstrr� � enumerate� A_REVERSErs �KEY_LEFT� KEY_RIGHTZ KEY_ENTERZ KEY_CANCELZ KEY_CLOSEZKEY_EXITrF ) ra r� Z display_listZret_listZselectedZset_len�offset�i�itemZdisplay�attrrx r r r � prompt_set s@ � �zIOTopUI.prompt_setc C s8 d}dddg}dddg}|d kr$d}| � ||||�}|S ) NzI/O priority class: z Real-timez Best-effortZIdleZrtZbe�idler )r� )ra � ioprio_classr� Zclasses_promptZ classes_retr� r r r � prompt_class; s �zIOTopUI.prompt_class�Error!c C sN | j �ddtd�tjB | j� | j �ddd| tj� | j �� t � d� d S )Nr r r z %s ) r\ r� r� r^ r� rd r� r� r� r1 �sleep)ra r` r r r � prompt_errorE s zIOTopUI.prompt_errorc C s, | j �ddtd�tjB | j� | j �� d S )Nr r r )r\ r� r� r^ r� rd r� re r r r � prompt_clearK s zIOTopUI.prompt_clearc " s� � fdd�}� fdd�}� fdd�}� fdd�}t d �d d � t d �d d � t d�� fdd �t d�� fdd �t d�|t d�|t d�|t d�|t d�|t d�|t d�|t d�|tj� fdd �tj� fdd �tj� fd d �tj� fd d �i}|�|d d � �}|� d S ) Nc s � j jdN _d S �NT)r# r/ r re r r �toggle_accumulatedP s z.IOTopUI.handle_key.<locals>.toggle_accumulatedc s � j jdN _d S r� )r# �onlyr re r r �toggle_only_ioS s z*IOTopUI.handle_key.<locals>.toggle_only_ioc s( � j jdN _� j�� � j�� d S r� )r# � processesrW �clearrm r re r r �toggle_processesV s z,IOTopUI.handle_key.<locals>.toggle_processesc s` z�� j jr � �� } � j�| �}n� �� }t|� jj�}|�� }t � |�\}}� � |�}|dkrdd}n � � |�}|� ||� � j�� � j�� W n� tk r� } z� �d|j � W 5 d }~X Y n� tk r� � �d� Y n~ tk r� � �d� Y nb tk �r � �d� Y nD tk �r6 � �d� Y n& tk �rR � �� Y n X � �� d S )Nr� r z Error setting I/O priority: %szInvalid process id!zInvalid thread id!zInvalid I/O priority data!zInvalid integer!)r# r� r� rW Z get_processr� r �taskstats_connection� get_ioprior Zto_class_and_datar� r� Z set_ioprior� rm r r� �errrL rM rN rK rF r� )rP Z exec_unit�tidZ ioprio_valuer� r� rD re r r �ionice[ s> �� z"IOTopUI.handle_key.<locals>.ionicer� c S s t �d�S �Nr �ri rr r r r r r- � r. z$IOTopUI.handle_key.<locals>.<lambda>r� c S s t �d�S r� r� r r r r r- � r. r� c s � � � S r* �rz r re r r r- � r. �Rc s � � � S r* r� r re r r r- � r. �a�A�o�OrR r r� �Ic s � � d�S )Nr� �r~ r re r r r- � r. c s � � d�S r{ r� r re r r r- � r. c s � � ttj� �S r* �r~ r> rO rY r re r r r- � r. c s � � ttj��S r* r� r re r r r- � r. c S s d S r* r r r r r r- � r. )r� r^ r� r� ZKEY_HOMEZKEY_END�get)ra rx r� r� r� r� Z key_bindings�actionr re r rt O sN $ �#zIOTopUI.handle_keyc s� �fdd�}�fdd�}t t|�jj�� ��}tj�j d � �jj rPdd� �ndd� �|j � �fd d��j d � t t ||��S ) Nc s� t � j| � jj�}|\}}}}tjr2d||f }nd}dt }|d | j| �� | � � d d� |||f }| � � } � jj s؈ j t |� } d| k r�t | �k r�n n6| d d } | | d } | d | � d | | d � } || 7 }� jj s�|d � j � }|S ) Nz%7s %7s z ?unavailable? z%%%ddz %4s %-8s %11s %11s %s� r r �~)r< r# rW r& r �has_blkio_delay_total� MAX_PID_WIDTHrP r� rT rU r[ rd r> ) rR r9 r: r; r5 r6 Z delay_statsZ pid_format�lineZcmdlineZremaining_lengthZlen1Zoffset2re r r �format� s2 � z IOTopUI.get_data.<locals>.formatc s � j j p| �� j j�S r* )r# r� Z did_some_ior/ �rR re r r � should_format� s �z'IOTopUI.get_data.<locals>.should_formatr c S s | j S r* )r0 r� r r r r- � r. z"IOTopUI.get_data.<locals>.<lambda>c S s | j S r* )r2 r� r r r r- � r. c s � | �| ��S r* r r� )rx � stats_lambdar r r- � r. )rx �reverse) �list�filterrW r� �valuesrO rY rX r# r/ �sortrZ �map)ra r� r� r� r )rx ra r� r �get_data� s � �zIOTopUI.get_datac s� dt �j|d |��d�t �j|d |��d�f dt �j|d |��d�t �j|d |��d�f g}tdtd �d }�jjr�|d7 }n|d 7 }|d d d d dddg}��� }�jjr�dg| }t�d�� � fdd�|D �}� fdd�|D �}�jj �r^�jj dk�r<|D ]} t | � �q �jj t |�k�r<t d� |�� |D ]} t | � �q@tj�� �n��j�� tj�rvd } nd} g } g } �jj�r��d�}��d�}tddd d d �jj�r�d nd fd!�jj�r�d"nd#fd$�jj�r�d%nd&fd'd(�j�r�d)nd*fd+|| �� fd,|| �� fd-|d �� fd.|d �� fg �}d}|� � D ]�\}}|�r|d/|d0|g}d|d1k�rndnt j ddg}nd/|d2g}dddg}| �r�|�r�t d� | | �d� |� ��j k�r�| �!|� | �!|� |d7 }n | | |7 < | | |7 < �qDt |��t t"|���t t"| ��}t | �}�j#� � | | }|d3k �rRg }d�|d4k �rdg }d�|d5k �rvd } d}|d6k �r�g } g } d}�j#� � | | }t$t |�|�}t%|�D ]$\�} �j�&�d| d �j � � �q�|�r�j�'�dt(d�t j)B �j � d}�j }t*t |��D ]��t j)}|� } ��j+k�rB| dd � } ��j+k�rn|t j,O }| �j�rhd7�pjd87 } | d |� } |t | �8 }| �r��j�&�|| |� |t | �7 }�qt*|�D ]p�zP����fd9d:�} z| |� � W n( t-k �r | |� �.d;�� Y nX W n t j/k �r" Y nX �q�t*|�D ]�} �j#| | | } �j�'| dt(d�t j)B �j � d}t*t | | ��D ]B��j�0| || | � t j)| | � B � |t | | � �7 }�qx�q0| �r�j�0�j#d d| t j,� �j�1� d S )<Nz.Total DISK READ: %s | Total DISK WRITE: %sr � r z.Current DISK READ: %s | Current DISK WRITE: %s� r �PIDZTIDz PRIOz USERz DISK READz DISK WRITEz SWAPINz IOz COMMANDz TIMEz %H:%M:%S c s g | ] }� | �qS r r )�.0r� �� current_timer r � <listcomp>� s z+IOTopUI.refresh_display.<locals>.<listcomp>c s g | ] }� | �qS r r )r� rS r� r r r� � s r r zNCONFIG_TASK_DELAY_ACCT not enabled in kernel, cannot determine SWAPIN and IO %r� )�keysr )�anyr� )r� �quit)r� r� r� �allZactiverR ZthreadsZprocsr� Z bandwidthZaccum)r� r r� ZascZdesc�left�right�home�endz z: r� �:r= rV r� r� �>�<c s �j �� � � d| � d S r� )r\ r� )r� )r� � len_summary� len_titlesra r r � print_lineB s z+IOTopUI.refresh_display.<locals>.print_linezutf-8)2r' r# �rjustr8 r� r� r� r1 �strftimer[ �quietrC r �joinri �stdout�flushr\ Zeraser r� �helpr| r r� r/ rZ rA �itemsr^ Z A_UNDERLINEr> rd �append�boolrc r( r� r� r� r� r� �rangerX r� �UnicodeEncodeError�encoder` r� r� ) ra Z first_timeru rv r& ZsummaryrP Ztitles�linesrS r� Z status_msgZ help_linesZ help_attrs�prev�nextr� Z help_linerx Z help_itemZ help_attrZlen_status_msgZlen_helpZ max_linesZ num_lines�posZremaining_colsr� �titler� Zlnr� r )r� r� r� r� ra r rn � s ��� � � 0 ( �zIOTopUI.refresh_display)NT)NT)N)r� )rH rI rJ rY rb r] ry rz r~ r| r� r� r� r� r� r� r� r� r� rt r� rn r r r r rO � s: � �! ! U'rO c C sf |j rt�tjtj� n$dd� }t�tj|� t�tj|� t|�}t||�}t| ||�}|� � d S )Nc _ s t �d� d S r� r� )rp �kwargsr r r � clean_exit] s z$run_iotop_window.<locals>.clean_exit) r[ �signal�SIGPIPE�SIG_DFL�SIGINT�SIGTERMr r rO ry )r\ r# r� r� rW Zuir r r �run_iotop_windowY s r c C s� z$| j rtd | �W S t�t| �W S W nX tk r| } z:|jtjkrjt|tj d� tdtj d� t� d� n� W 5 d }~X Y nX d S )N)�filea� The Linux kernel interfaces that iotop relies on now require root privileges or the NET_ADMIN capability. This change occurred because a security issue (CVE-2011-2494) was found that allows leakage of sensitive data across user boundaries. If you require the ability to run iotop as a non-root user, please configure sudo to allow you to run iotop as root. Please do not file bugs on iotop about this.r ) r[ r r^ �wrapper�OSErrorrq ZEPERMrC ri �stderrrr )r# rD r r r � run_iotopg s � r c C s� d}z:dd l }dd l}td� |�dt� t� |� |�|�}W nX tk r� dd l}dd l }|j |dd�}td� |� | � |� � |j �|�}Y nX |�� |�dd � |�d � |�d � t�|� d S ) Nz iotop.profr zProfiling using cProfilezcontinuation()r )Z lineeventszProfiling using hotshotr1 Zcalls�2 )�cProfile�pstatsrC Zrunctx�globals�localsr � ImportError�hotshotZ hotshot.statsZProfileZruncall�closer9 �loadZ strip_dirsZ sort_statsZ print_statsZ print_callees�os�remove)Z continuationZ prof_filer r r9 r Zprofr r r �_profile� s( r aF %s [OPTIONS] DISK READ and DISK WRITE are the block I/O bandwidth used during the sampling period. SWAPIN and IO are the percentages of time the thread spent respectively while swapping in and waiting on I/O more generally. PRIO is the I/O priority at which the thread is running (set using the ionice command). Controls: left and right arrows to change the sorting column, r to invert the sorting order, o to toggle the --only option, p to toggle the --processes option, a to toggle the --accumulated option, i to change I/O priority, q to quit, any other key to force a refresh.c s� zt �t jd� W n t jk r0 td� Y nX tjtdt d�} | j ddddd d d � | j d d dddd� | j ddddddd� | j ddddd d d d � | j d d!dd"d#d$d%d&� | j d'd(d)d*d#d+d,d&� | j d-d.dd/d d0d � | j d1d2dd3d d4d � | j d5d6dd7d d8d � | j d9d:dd;d<d� | j d=d>d?d@dAdBd � | j dCddDd tj d � | j dEdFdGdHdId � | � � \�}|�r�| � dJdK� |� � t�� �j�p�g �_�j�p��j�p��j�_�fdLdM�� �j�r� fdNdO�}t|� n� � d S )PNr z8unable to set locale, falling back to the default localeziotop )Zusage�versionz-oz--only� store_truer� Fz1only show processes or threads actually doing I/O)r� �destr� r� z-bz--batchr[ znon-interactive mode)r� r r� z-nz--iterr rl ZNUMz-number of iterations before ending [infinite])�typer �metavarr� z-dz--delayr ro z#delay between iterations [1 second]ZSECr )r r r� r r� z-pz--pid�pidsr� z"processes/threads to monitor [all]r� )r r r� r� r z-uz--user�strZuserszusers to monitor [all]ZUSERz-Pz --processesr� z$only show processes, not all threadsz-az --accumulatedr/ z)show accumulated I/O instead of bandwidthz-kz --kilobytesr" z.use kilobytes instead of a human friendly unitz-tz--timer1 z.add a timestamp on each line (implies --batch)z-qz--quiet�countr� r z/suppress some lines of header (implies --batch)z --profile�profilez --no-helpZ store_falser� Tz suppress listing of shortcutszUnexpected arguments: r c s t � �S r* )r r r, r r r- � r. zmain.<locals>.<lambda>c s$ z � � W n t k r Y nX d S r* )rB r )� main_loopr r �safe_main_loop� s z main.<locals>.safe_main_loop)�locale� setlocale�LC_ALL�ErrorrC �optparseZ OptionParser�USAGEr Z add_optionZ SUPPRESS_HELP� parse_argsr` r� r r r[ r1 r� r r )�parserrp r r )r r# r �main� s� � � � � � � � � � �� � � r( )3Z __future__r r^ rq r r r$ r rf r� ri r1 �hasattrr r � RuntimeError� collectionsr Z iotop.datar r r r r Z iotop.versionr Ziotopr Z iotop.ioprior r r! r% r' r< rE r� rB rF rK rL rM rN �objectrO r r r �argvr% r( r r r r �<module> sZ L �