Your IP : 3.133.155.253
3
�bf�b � @ sz d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
mZmZmZmZ d dl mZ d dlmZ d dlmZ d dlmZ dd d
d d d
dgZdd� Zdd� Zdd� Z dd� Z dd� Z e � Z G dd� de �Z!G dd � 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
j+d Z,d,d-� Z-dS ).� )�print_functionN)� find_uids�TaskStatsNetlink� ProcessList�Stats�sysctl_task_delayacct)�
ThreadInfo)�VERSION)�ioprio)�IoprioSetError�B�K�M�G�T�P�Ec C s` | dkrd}n| dk r"d}| } ndS t tj| 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.6/ui.py�
human_size- s r c C s | j rd|d S t|�S )Nz%.2f Kg �@)� kilobytesr )�options�bytesr r r � format_size9 s r$ c C s t | |ot|�| �d S )Nz/s)r$ r )r" r �durationr r r �format_bandwidth> s r& c s� � fdd�}�j r4|j}�fdd�}tj� |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 �
delay2percentD s z#format_stats.<locals>.delay2percentc s
t � | �S )N)r$ )r r% )r" r r �<lambda>H s z format_stats.<locals>.<lambda>c s t � | |�S )N)r& )r r% )r" r r r) L s r ) � accumulated� stats_accum�timeZstats_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_delayr0 Z
written_bytesr1 r )r% r" r � format_statsA s
r7 c C sB yt td�j� j� �S tk
r< } z t| � dS d } ~ X nX d S )Nz/proc/sys/kernel/pid_max� )�len�open�read�strip� Exception�print)�er r r �get_max_pid_widthV s
r@ c @ s e Zd ZdS )� CancelInputN)�__name__�
__module__� __qualname__r r r r rA d s rA c @ s e Zd ZdS )�
InvalidIntN)rB rC rD r r r r rE e s rE c @ s e Zd ZdS )�
InvalidPidN)rB rC rD r r r r rF f s rF c @ s e Zd ZdS )�
InvalidTidN)rB rC rD r r r r rG g s rG c @ s e Zd ZdS )�InvalidIoprioDataN)rB rC rD r r r r rH h s rH c @ s� e Zd Zdd� dfdd� dfdd� dfdd� dfdd� dfd d� dfd
d� dfd d� dfgZd d
� Zdd� Zdd� Zdd� Zdd� Zd0dd�Z d1dd�Z
dd � Z d d � Z d d � Z
d!d"� Zd2d#d$�Zd3d&d'�Zd(d)� Zd*d+� Zd,d-� Zd.d/� ZdS )4�IOTopUIc C s | j S )N)�pid)�p�sr r r r) q s zIOTopUI.<lambda>Fc C s | j � S )N)Zioprio_sort_key)rK rL r r r r) r s c C s | j � S )N)�get_user)rK rL r r r r) s s c C s |j S )N)r0 )rK rL r r r r) t s Tc C s |j |j S )N)r1 r2 )rK rL r r r r) u s c C s |j S )N)r/ )rK rL r r r r) v s c C s |j pt|jp|j �S )N)r. r r0 r1 )rK rL r r r r) y s c C s | j � S )N)� get_cmdline)rK rL r r r r) { s c C st || _ || _d| _tj| j d | _| jjsp|| _| j� y t j
� t j � t j d� W n t j
k
rn Y nX d S )N� r r )� process_listr" � sorting_keyrI � sorting_keys�sorting_reverse�batch�win�resize�cursesZuse_default_colorsZ start_color�curs_set�error)�selfrU rP r" r r r �__init__~ s zIOTopUI.__init__c C s | j j� \| _| _d S )N)rU Zgetmaxyx�height�width)rZ r r r rV � s zIOTopUI.resizec
C s^ d}t j� }| jjs.|jtjj� t jt j B � �x(| jj
d ksL|| jj
k �rX| j j � \}}| j
|dk||| j j� | jj
d k r�|d7 }|| jj
kr�P n |dkr�d}y|j| jjd �}W nB t jk
r� } z$|jr�|jd tjkr�g }n� W Y d d }~X nX x.|D ]&\}}|t jt jB @ �rtjd� �qW | jj�s>| j� |r2| jj� } | j| � q2W d S )Nr r g @�@)�select�pollr" rT �register�sys�stdin�filenoZPOLLINZPOLLPRI�
iterationsrP �refresh_processes�refresh_displayr% �
delay_secondsrY �args�errnoZEINTRZPOLLERRZPOLLHUP�exitrV rU �getch�
handle_key)
rZ rd r_ �total�actualZeventsr? �fdZevent�keyr r r �run� s:
z IOTopUI.runc C s | j | _ d S )N)rS )rZ r r r �reverse_sorting� s zIOTopUI.reverse_sortingc C s� | j }| j |7 _ td| j �| _ tttj�d | j �| _ | jsh| j dkrh|dkrZd| _ n|dkrhd| _ || j kr�tj| j d | _d S )Nr r r8 rO � � )r8 rO )rQ r3 r' r9 rI rR �
has_swapin_iorS )rZ ZdeltaZorig_sorting_keyr r r �adjust_sorting_key� s
zIOTopUI.adjust_sorting_keyNc C s� | j jddtd�tjB | j� | j jdd|tj� | j j� tj � tj
d� | j j dt |��}tj
d� tj
� |dkr~|S |r�t� �|S )Nr r � r )Nr )rU �hline�ordrW �A_NORMALr] �addstr�A_BOLD�refreshZechorX Zgetstrr9 ZnoechorA )rZ �prompt�default�empty_is_cancel�inpr r r �
prompt_str� s
zIOTopUI.prompt_strc C s6 | j |||�}yt|�S tk
r0 t� �Y nX d S )N)r� r �
ValueErrorrE )rZ r~ r r� r� r r r �
prompt_int� s
zIOTopUI.prompt_intc C s> y
| j d�S tk
r$ t� �Y n tk
r8 � Y nX d S )NzPID to ionice: )r� rE rF rA )rZ r r r �
prompt_pid� s
zIOTopUI.prompt_pidc C s> y
| j d�S tk
r$ t� �Y n tk
r8 � Y nX d S )NzTID to ionice: )r� rE rG rA )rZ r r r �
prompt_tid� s
zIOTopUI.prompt_tidc C sd y.|d k r | j d| |d�}n| j 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 rt )r� rE rH )rZ � ioprio_datar� r r r � prompt_data� s zIOTopUI.prompt_datac C sv y|j |�}W n tk
r& d }Y nX t|�d }�x:| jjddtd�tjB | j� | jj dd|tj
� t|�}xPt |�D ]D\}}d| } ||kr�tj }
ntj}
| jj d|| |
� |t| �7 }q|W x�| jj
� } | tjtd�fkr�|dkr�|d8 }P q�| tjtd�fk�r ||k �r |d7 }P q�| tjtd�td�fk�rB|| S | d tjtjtjtd
�td �fkr�t� �q�W q8W d S )
Nr r rw z %s �l�r�
�
� �q�Q���)�indexr� r9 rU rx ry rW rz r] �insstrr| � enumerate� A_REVERSErk �KEY_LEFT� KEY_RIGHTZ KEY_ENTERZ
KEY_CANCELZ KEY_CLOSEZKEY_EXITrA ) rZ r~ Z display_listZret_listZselectedZset_len�offset�i�itemZdisplay�attrrp r r r �
prompt_set� s:
zIOTopUI.prompt_setc C s8 d}dddg}dddg}|d kr$d}| j ||||�}|S ) NzI/O priority class: z Real-timez Best-effortZIdleZrtZbe�idler )r� )rZ � ioprio_classr~ Zclasses_promptZ classes_retr� r r r � prompt_class s
zIOTopUI.prompt_class�Error!c C sN | j jddtd�tjB | j� | j jddd| tj� | j j� t j
d� d S )Nr r rw z %s ) rU rx ry rW rz r] r� r� r} r, Zsleep)rZ rY r r r � prompt_error" s
zIOTopUI.prompt_errorc C s, | j jddtd�tjB | j� | j j� d S )Nr r rw )rU rx ry rW rz r] r} )rZ r r r � prompt_clear( 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}|j|d d � �}|� d S ) Nc s � j jdN _d S )NT)r" r* r )rZ r r �toggle_accumulated- s z.IOTopUI.handle_key.<locals>.toggle_accumulatedc s � j jdN _d S )NT)r" �onlyr )rZ r r �toggle_only_io/ s z*IOTopUI.handle_key.<locals>.toggle_only_ioc s( � j jdN _� jj� � jj� d S )NT)r" � processesrP �clearre r )rZ r r �toggle_processes1 s
z,IOTopUI.handle_key.<locals>.toggle_processesc s` y�� j jr � j� } � jj| �}n� j� }t|� jj�}|j� }t j
|�\}}� j |�}|dkrdd}n
� j |�}|j
||� � jj� � jj� W n� tk
r� } z� jd|j � W Y d d }~X n� tk
r� � jd� Y n~ tk
r� � jd� Y nb tk
�r � jd� Y nD tk
�r6 � jd� Y n& tk
�rR � j� Y n
X � j� 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� rP Z get_processr� r �taskstats_connection�
get_ioprior
Zto_class_and_datar� r� Z
set_ioprior� re r r� �errrF rG rH rE rA r� )rJ Z exec_unit�tidZ ioprio_valuer� r� r? )rZ r r �ionice5 s:
z"IOTopUI.handle_key.<locals>.ionicer� c S s
t jd�S )Nr )ra rj r r r r r) Z s z$IOTopUI.handle_key.<locals>.<lambda>r� c S s
t jd�S )Nr )ra rj r r r r r) \ s r� c s � j � S )N)rr r )rZ r r r) ^ s �Rc s � j � S )N)rr r )rZ r r r) ` s �a�A�o�OrK r r� �Ic s
� j d�S )Nr r� )rv r )rZ r r r) r s c s
� j d�S )Nr )rv r )rZ r r r) t s c s � j ttj� �S )N)rv r9 rI rR r )rZ r r r) v s c s � j ttj��S )N)rv r9 rI rR r )rZ r r r) x s c S s d S )Nr r r r r r) { s )ry rW r� r� ZKEY_HOMEZKEY_END�get)rZ rp r� r� r� r� Z key_bindings�actionr )rZ r rl , sL $
zIOTopUI.handle_keyc s� �fdd�}�fdd�}t t|�jjj� ��}tj�j d � �jj rPdd� �ndd� �|j
� �fd d��j d
� �jj s�|�j
d d �= t t||��S ) Nc
sR t � j| � jj�}|\}}}}dt }| jf}|d7 }|| j� f7 }|d7 }|| j� d d� f7 }|d7 }|||f7 }� jr�|d7 }|||f7 }n� jj r�|d7 }|d
7 }|d 7 }|| }| j
� } � jj �s.� j t |� }
d
|
k o�t | �k n �r.|
d d
} |
| d } | d | � d | | d � } || 7 }� jj �sN|d � j � }|S )Nz%%%ddz %4sz %-8s� z
%11s %11sz %7s %7sz %s �
?unavailable?rw r r �~)r� )
r7 r" rP r% �
MAX_PID_WIDTHrJ r� rM ru rT rN r] r9 )
rK r4 r5 r6 r0 r1 �formatZparams�lineZcmdlineZremaining_lengthZlen1Zoffset2)rZ r r r� s:
z IOTopUI.get_data.<locals>.formatc s � j j p| j� j j�S )N)r" r� Z did_some_ior* )rK )rZ r r �
should_format� s
z'IOTopUI.get_data.<locals>.should_formatr c S s | j S )N)r+ )rK r r r r) � s z"IOTopUI.get_data.<locals>.<lambda>c S s | j S )N)r- )rK r r r r) � s c s � | �| ��S )Nr )rK )rp � stats_lambdar r r) � s )rp �reverser )�list�filterrP r� �valuesrI rR rQ r" r* �sortrS rT r\ �map)rZ r� r� r� r )rp rZ r� r �get_data~ s
zIOTopUI.get_datac s� dt �j|d |�jd�t �j|d |�jd�f dt �j|d |�jd�t �j|d |�jd�f g�tdtd �d }�jjr�|d7 }n|d 7 }|d
d d d
dddg}tj�_�jr�t � dkr�d�_�j
d� �j � }�jj �rdg| }t j
d�� � fdd�|D �}� fdd��D ���jj�r��jjdk�rjx�D ]}t|� �q8W �jjt|�k�rjtdj|�� x |D ]} t| jd�� �qpW tjj� �n�jj� x0t��D ]$\�}�jj�d|d �j� � �q�W �jj t ��dt d�t j B �j� �j}
x�t!t |��D ]���j �r$�d!k�r$�qt j } |� } ��j"k�rJ| dd � } ��j"k�rv| t j#O } | �j$�rpd �prd 7 } | d |
� } |
t | �8 }
�jj| | � �qW �jdk �r�d }
nd }
t%t |��j&d tt'|
�� �}xzt!|�D ]n�yN���fd d �}y||� � W n( t(k
�r0 ||� jd�� Y nX W n t j)k
�rL Y nX �q�W |
�rx�jj*�j&t �� d|
t j#� �jj+� d S )"Nz,Total DISK READ : %s | Total DISK WRITE : %sr � r z,Actual DISK READ: %s | Actual DISK WRITE: %s� rw �PIDZTIDz PRIOz USERz DISK READz DISK WRITEz SWAPINz IOz COMMANDFz 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� rL )r� r r r� � s r r zutf-8r8 rO �>�<zoCONFIG_TASK_DELAY_ACCT and kernel.task_delayacct sysctl not enabled in kernel, cannot determine SWAPIN and IO %c s �j j� t�� d d| � d S )Nr r )rU r{ r9 )r� )r� rZ �summaryr r �
print_line� s z+IOTopUI.refresh_display.<locals>.print_line)r8 rO ),r& r" �rjustr3 r� r� r Zhas_blkio_delay_totalru r rv r� r, ZstrftimerT �quietr> r �join�encodera �stdout�flushrU Zeraser� r{ r] rx r9 ry rW r� �rangerQ r| rS r' r\ �bool�UnicodeEncodeErrorrY r� r} )rZ Z
first_timerm rn r% rJ Ztitles�linesrL r� Zremaining_colsr� �titleZ
status_msgZ num_linesr� r )r� r� rZ r� r rf � s�
zIOTopUI.refresh_display)NT)NT)N)r� )rB rC rD rR r[ rV rq rr rv r� r� r� r� r� r� r� r� r� rl r� rf r r r r rI n s4
!
R1rI c C sf |j rtjtjtj� n$dd� }tjtj|� tjtj|� t|�}t||�}t| ||�}|j � d S )Nc _ s t jd� d S )Nr )ra rj )rh �kwargsr r r �
clean_exit s z$run_iotop_window.<locals>.clean_exit)
rT �signal�SIGPIPE�SIG_DFL�SIGINT�SIGTERMr r rI rq )rU r" r� r� rP Zuir r r �run_iotop_window s
r� c C s� y | j rtd | �S tjt| �S W n� tk
rx } z<|jtjkrft|tj d� tdtj d� tj
d� n� W Y d d }~X nZ tj k
r� } z<t |�}|j
d�dkr�|j
d�dks�|j
d�dkr�n� W Y d d }~X nX d S )N)�filea�
The Linux kernel interfaces that iotop relies on now require root priviliges
or the NET_ADMIN capability. This change occured 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 ZERRr z
nocbreak()zendwin())rT r� rW �wrapper�OSErrorri ZEPERMr> ra �stderrrj rY �str�find)r" r? Zstrer r r � run_iotop s"
r� c C s� d}y:dd l }dd l}td� |jdt� t� |� |j|�}W nX tk
r� dd l}dd l }|j
|dd�}td� |j | � |j � |j
j|�}Y nX |j� |jdd � |jd
� |jd
� tj|� d S ) Nz
iotop.profr zProfiling using cProfilezcontinuation()r )Z
lineeventszProfiling using hotshotr, Zcalls�2 )�cProfile�pstatsr> Zrunctx�globals�localsr � ImportError�hotshotZ
hotshot.statsZProfileZruncall�closer4 �loadZ
strip_dirsZ
sort_statsZ print_statsZ
print_callees�os�remove)Z continuationZ prof_filer� r� r4 r� Zprofr r r �_profile6 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� yt jt 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 � \�}|�rt| j dEdFj
|� � t�� �j�p�g �_�j�p��j�p��j�_�fdGdH�� �j�ṙ fdIdJ�}t|� n� � d S )KNr 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 �helpz-bz--batchrT znon-interactive mode)r� r� r� z-nz--iterr rd ZNUMz-number of iterations before ending [infinite])�typer� �metavarr� z-dz--delayr rg z#delay between iterations [1 second]ZSECr )r� r� r� r� r z-pz--pid�pids�appendz"processes/threads to monitor [all]r� )r� r� r� r� r� z-uz--userr� Zuserszusers 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--timer, z.add a timestamp on each line (implies --batch)z-qz--quiet�countr� r z/suppress some lines of header (implies --batch)z --profile�profilezUnexpected arguments: rw c s t � �S )N)r� r )r" r r r) � s zmain.<locals>.<lambda>c s y
� � W n Y nX d S )Nr r )� main_loopr r �safe_main_loop� s
z main.<locals>.safe_main_loop)�locale� setlocale�LC_ALL�Errorr> �optparseZ OptionParser�USAGEr Z
add_optionZ
SUPPRESS_HELP�
parse_argsrY r� r r� rT r, r� r� r� )�parserrh r� r )r� r" r �main\ s\
r ).Z
__future__r rW ri r r r r� r^ r� ra r, Z
iotop.datar r r r r r Z
iotop.versionr Ziotopr
Z iotop.ioprior r r r$ r& r7 r@ r� r= rA rE rF rG rH �objectrI r� r� r� �argvr r r r r r �<module> sH
$$