| Current File : /home/mmdealscpanel/yummmdeals.com/iotop.tar |
__init__.py 0000644 00000000000 15035135563 0006651 0 ustar 00 data.py 0000644 00000036132 15035135563 0006042 0 ustar 00 # This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Library General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
#
# See the COPYING file for license information.
#
# Copyright (c) 2007 Guillaume Chazarain <guichaz@gmail.com>
# Allow printing with same syntax in Python 2/3
from __future__ import print_function
import errno
import os
import pprint
import pwd
import stat
import struct
import sys
import time
from procfs import cmdline
#
# Check for requirements:
# o Linux >= 2.6.20 with I/O accounting and VM event counters
#
ioaccounting = os.path.exists('/proc/self/io')
try:
from iotop.vmstat import VmStat
vmstat_f = VmStat()
except:
vm_event_counters = False
else:
vm_event_counters = True
if not ioaccounting or not vm_event_counters:
print('Could not run iotop as some of the requirements are not met:')
print('- Linux >= 2.6.20 with')
if not ioaccounting:
print(' - I/O accounting support ' \
'(CONFIG_TASKSTATS, CONFIG_TASK_DELAY_ACCT, ' \
'CONFIG_TASK_IO_ACCOUNTING, kernel.task_delayacct sysctl)')
if not vm_event_counters:
print(' - VM event counters (CONFIG_VM_EVENT_COUNTERS)')
sys.exit(1)
from iotop import ioprio, vmstat
from iotop.netlink import Connection, NETLINK_GENERIC, U32Attr, NLM_F_REQUEST
from iotop.genetlink import Controller, GeNlMessage
class DumpableObject(object):
"""Base class for all objects that allows easy introspection when printed"""
def __repr__(self):
return '%s: %s>' % (str(type(self))[:-1], pprint.pformat(self.__dict__))
#
# Interesting fields in a taskstats output
#
class Stats(DumpableObject):
members_offsets = [
('blkio_delay_total', 40),
('swapin_delay_total', 56),
('read_bytes', 248),
('write_bytes', 256),
('cancelled_write_bytes', 264)
]
has_blkio_delay_total = None
def __init__(self, task_stats_buffer):
sd = self.__dict__
for name, offset in Stats.members_offsets:
data = task_stats_buffer[offset:offset + 8]
sd[name] = struct.unpack('Q', data)[0]
# This is a heuristic to detect if CONFIG_TASK_DELAY_ACCT is enabled in
# the kernel.
if not Stats.has_blkio_delay_total:
Stats.has_blkio_delay_total = sysctl_task_delayacct()
def accumulate(self, other_stats, destination, coeff=1):
"""Update destination from operator(self, other_stats)"""
dd = destination.__dict__
sd = self.__dict__
od = other_stats.__dict__
for member, offset in Stats.members_offsets:
dd[member] = sd[member] + coeff * od[member]
def delta(self, other_stats, destination):
"""Update destination with self - other_stats"""
return self.accumulate(other_stats, destination, coeff=-1)
def is_all_zero(self):
sd = self.__dict__
for name, offset in Stats.members_offsets:
if sd[name] != 0:
return False
return True
@staticmethod
def build_all_zero():
stats = Stats.__new__(Stats)
std = stats.__dict__
for name, offset in Stats.members_offsets:
std[name] = 0
return stats
#
# Netlink usage for taskstats
#
TASKSTATS_CMD_GET = 1
TASKSTATS_CMD_ATTR_PID = 1
TASKSTATS_TYPE_AGGR_PID = 4
TASKSTATS_TYPE_PID = 1
TASKSTATS_TYPE_STATS = 3
class TaskStatsNetlink(object):
# Keep in sync with format_stats() and pinfo.did_some_io()
def __init__(self, options):
self.options = options
self.connection = Connection(NETLINK_GENERIC)
controller = Controller(self.connection)
self.family_id = controller.get_family_id('TASKSTATS')
def build_request(self, tid):
return GeNlMessage(self.family_id, cmd=TASKSTATS_CMD_GET,
attrs=[U32Attr(TASKSTATS_CMD_ATTR_PID, tid)],
flags=NLM_F_REQUEST)
def get_single_task_stats(self, thread):
thread.task_stats_request.send(self.connection)
try:
reply = GeNlMessage.recv(self.connection)
except OSError as e:
if e.errno == errno.ESRCH:
# OSError: Netlink error: No such process (3)
return
raise
for attr_type, attr_value in reply.attrs.items():
if attr_type == TASKSTATS_TYPE_AGGR_PID:
reply = attr_value.nested()
break
else:
return
taskstats_data = reply[TASKSTATS_TYPE_STATS].data
if len(taskstats_data) < 272:
# Short reply
return
taskstats_version = struct.unpack('H', taskstats_data[:2])[0]
assert taskstats_version >= 4
return Stats(taskstats_data)
#
# PIDs manipulations
#
def find_uids(options):
"""Build options.uids from options.users by resolving usernames to UIDs"""
options.uids = []
error = False
for u in options.users or []:
try:
uid = int(u)
except ValueError:
try:
passwd = pwd.getpwnam(u)
except KeyError:
print('Unknown user:', u, file=sys.stderr)
error = True
else:
uid = passwd.pw_uid
if not error:
options.uids.append(uid)
if error:
sys.exit(1)
def parse_proc_pid_status(pid):
result_dict = {}
try:
for line in open('/proc/%d/status' % pid):
key, value = line.split(':\t', 1)
result_dict[key] = value.strip()
except IOError:
pass # No such process
return result_dict
def safe_utf8_decode(s):
try:
return s.decode('utf-8')
except UnicodeDecodeError:
return s.encode('string_escape')
except AttributeError:
return s
class ThreadInfo(DumpableObject):
"""Stats for a single thread"""
def __init__(self, tid, taskstats_connection):
self.tid = tid
self.mark = True
self.stats_total = None
self.stats_delta = Stats.__new__(Stats)
self.task_stats_request = taskstats_connection.build_request(tid)
def get_ioprio(self):
return ioprio.get(self.tid)
def set_ioprio(self, ioprio_class, ioprio_data):
return ioprio.set_ioprio(ioprio.IOPRIO_WHO_PROCESS, self.tid,
ioprio_class, ioprio_data)
def update_stats(self, stats):
if not self.stats_total:
self.stats_total = stats
stats.delta(self.stats_total, self.stats_delta)
self.stats_total = stats
class ProcessInfo(DumpableObject):
"""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."""
def __init__(self, pid):
self.pid = pid
self.uid = None
self.user = None
self.threads = {} # {tid: ThreadInfo}
self.stats_delta = Stats.build_all_zero()
self.stats_accum = Stats.build_all_zero()
self.stats_accum_timestamp = time.time()
def is_monitored(self, options):
if (options.pids and not options.processes and
self.pid not in options.pids):
# We only monitor some threads, not this one
return False
if options.uids and self.get_uid() not in options.uids:
# We only monitor some users, not this one
return False
return True
def get_uid(self):
if self.uid:
return self.uid
# uid in (None, 0) means either we don't know the UID yet or the process
# runs as root so it can change its UID. In both cases it means we have
# to find out its current UID.
try:
uid = os.stat('/proc/%d' % self.pid)[stat.ST_UID]
except OSError:
# The process disappeared
uid = None
if uid != self.uid:
# Maybe the process called setuid()
self.user = None
self.uid = uid
return uid
def get_user(self):
uid = self.get_uid()
if uid is not None and not self.user:
try:
self.user = safe_utf8_decode(pwd.getpwuid(uid).pw_name)
except (KeyError, AttributeError):
self.user = str(uid)
return self.user or '{none}'
def get_cmdline(self):
# A process may exec, so we must always reread its cmdline
try:
proc_cmdline = open('/proc/%d/cmdline' % self.pid)
cmdline = proc_cmdline.read(4096)
except IOError:
return '{no such process}'
proc_status = parse_proc_pid_status(self.pid)
if not cmdline:
# Probably a kernel thread, get its name from /proc/PID/status
proc_status_name = proc_status.get('Name', '')
if proc_status_name:
proc_status_name = '[%s]' % proc_status_name
else:
proc_status_name = '{no name}'
return proc_status_name
suffix = ''
tgid = int(proc_status.get('Tgid', self.pid))
if tgid != self.pid:
# Not the main thread, maybe it has a custom name
tgid_name = parse_proc_pid_status(tgid).get('Name', '')
thread_name = proc_status.get('Name', '')
if thread_name != tgid_name:
suffix += ' [%s]' % thread_name
parts = cmdline.split('\0')
if parts[0].startswith('/'):
first_command_char = parts[0].rfind('/') + 1
parts[0] = parts[0][first_command_char:]
cmdline = ' '.join(parts).strip()
return safe_utf8_decode(cmdline + suffix)
def did_some_io(self, accumulated):
if accumulated:
return not self.stats_accum.is_all_zero()
for t in self.threads.values():
if not t.stats_delta.is_all_zero():
return True
return False
def get_ioprio(self):
priorities = set(t.get_ioprio() for t in self.threads.values())
if len(priorities) == 1:
return priorities.pop()
return '?dif'
def set_ioprio(self, ioprio_class, ioprio_data):
for thread in self.threads.values():
thread.set_ioprio(ioprio_class, ioprio_data)
def ioprio_sort_key(self):
return ioprio.sort_key(self.get_ioprio())
def get_thread(self, tid, taskstats_connection):
thread = self.threads.get(tid, None)
if not thread:
thread = ThreadInfo(tid, taskstats_connection)
self.threads[tid] = thread
return thread
def update_stats(self):
stats_delta = Stats.build_all_zero()
for tid, thread in self.threads.items():
if not thread.mark:
stats_delta.accumulate(thread.stats_delta, stats_delta)
self.threads = dict([(tid, thread) for tid, thread in
self.threads.items() if not thread.mark])
nr_threads = len(self.threads)
if not nr_threads:
return False
stats_delta.blkio_delay_total /= nr_threads
stats_delta.swapin_delay_total /= nr_threads
self.stats_delta = stats_delta
self.stats_accum.accumulate(self.stats_delta, self.stats_accum)
return True
class ProcessList(DumpableObject):
def __init__(self, taskstats_connection, options):
# {pid: ProcessInfo}
self.processes = {}
self.taskstats_connection = taskstats_connection
self.options = options
self.timestamp = time.time()
self.vmstat = vmstat.VmStat()
# A first time as we are interested in the delta
self.update_process_counts()
def get_process(self, pid):
"""Either get the specified PID from self.processes or build a new
ProcessInfo if we see this PID for the first time"""
process = self.processes.get(pid, None)
if not process:
process = ProcessInfo(pid)
self.processes[pid] = process
if process.is_monitored(self.options):
return process
def list_tgids(self):
if self.options.pids:
return self.options.pids
tgids = os.listdir('/proc')
if self.options.processes:
return [int(tgid) for tgid in tgids if '0' <= tgid[0] <= '9']
tids = []
for tgid in tgids:
if '0' <= tgid[0] <= '9':
try:
tids.extend(map(int, os.listdir('/proc/' + tgid + '/task')))
except OSError:
# The PID went away
pass
return tids
def list_tids(self, tgid):
if not self.options.processes:
return [tgid]
try:
tids = list(map(int, os.listdir('/proc/%d/task' % tgid)))
except OSError:
return []
if self.options.pids:
tids = list(set(self.options.pids).intersection(set(tids)))
return tids
def update_process_counts(self):
new_timestamp = time.time()
self.duration = new_timestamp - self.timestamp
self.timestamp = new_timestamp
total_read = total_write = 0
for tgid in self.list_tgids():
process = self.get_process(tgid)
if not process:
continue
for tid in self.list_tids(tgid):
thread = process.get_thread(tid, self.taskstats_connection)
stats = self.taskstats_connection.get_single_task_stats(thread)
if stats:
thread.update_stats(stats)
delta = thread.stats_delta
total_read += delta.read_bytes
total_write += delta.write_bytes
thread.mark = False
return (total_read, total_write), self.vmstat.delta()
def refresh_processes(self):
for process in self.processes.values():
for thread in process.threads.values():
thread.mark = True
total_read_and_write = self.update_process_counts()
self.processes = dict([(pid, process) for pid, process in
self.processes.items() if
process.update_stats()])
return total_read_and_write
def clear(self):
self.processes = {}
def sysctl_task_delayacct():
""" 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
"""
return 'nodelayacct' not in cmdline().keys()
genetlink.py 0000644 00000003774 15035135563 0007117 0 ustar 00 '''
Netlink message generation/parsing
Copyright 2007 Johannes Berg <johannes@sipsolutions.net>
GPLv2+; See copying for details.
'''
import struct
from iotop.netlink import NLM_F_REQUEST, NLMSG_MIN_TYPE, Message, parse_attributes
from iotop.netlink import NulStrAttr, Connection, NETLINK_GENERIC
CTRL_CMD_UNSPEC = 0
CTRL_CMD_NEWFAMILY = 1
CTRL_CMD_DELFAMILY = 2
CTRL_CMD_GETFAMILY = 3
CTRL_CMD_NEWOPS = 4
CTRL_CMD_DELOPS = 5
CTRL_CMD_GETOPS = 6
CTRL_ATTR_UNSPEC = 0
CTRL_ATTR_FAMILY_ID = 1
CTRL_ATTR_FAMILY_NAME = 2
CTRL_ATTR_VERSION = 3
CTRL_ATTR_HDRSIZE = 4
CTRL_ATTR_MAXATTR = 5
CTRL_ATTR_OPS = 6
class GenlHdr:
def __init__(self, cmd, version = 0):
self.cmd = cmd
self.version = version
def _dump(self):
return struct.pack("BBxx", self.cmd, self.version)
def _genl_hdr_parse(data):
return GenlHdr(*struct.unpack("BBxx", data))
GENL_ID_CTRL = NLMSG_MIN_TYPE
class GeNlMessage(Message):
def __init__(self, family, cmd, attrs=[], flags=0):
self.cmd = cmd
self.attrs = attrs
self.family = family
Message.__init__(self, family, flags=flags,
payload=[GenlHdr(self.cmd)]+attrs)
@staticmethod
def recv(conn):
msg = conn.recv()
packet = msg.payload
hdr = _genl_hdr_parse(packet[:4])
genlmsg = GeNlMessage(msg.type, hdr.cmd, [], msg.flags)
genlmsg.attrs = parse_attributes(packet[4:])
genlmsg.version = hdr.version
return genlmsg
class Controller:
def __init__(self, conn):
self.conn = conn
def get_family_id(self, family):
a = NulStrAttr(CTRL_ATTR_FAMILY_NAME, family)
m = GeNlMessage(GENL_ID_CTRL, CTRL_CMD_GETFAMILY,
flags=NLM_F_REQUEST, attrs=[a])
m.send(self.conn)
m = GeNlMessage.recv(self.conn)
return m.attrs[CTRL_ATTR_FAMILY_ID].u16()
connection = Connection(NETLINK_GENERIC)
controller = Controller(connection)
__pycache__/genetlink.cpython-36.opt-1.pyc 0000644 00000005263 15035135563 0014335 0 ustar 00 3
B��Q� � @ s� d Z ddlZddlmZmZmZmZ ddlmZmZm Z dZ
dZ dZ dZ
dZd Zd
ZdZdZdZdZdZd Zd
ZG d d � d �Zd
d� ZeZG dd� de�ZG dd� d�Z ee �Z e e �Z dS )z�
Netlink message generation/parsing
Copyright 2007 Johannes Berg <johannes@sipsolutions.net>
GPLv2+; See copying for details.
� N)�
NLM_F_REQUEST�NLMSG_MIN_TYPE�Message�parse_attributes)�
NulStrAttr�
Connection�NETLINK_GENERIC� � � � � � c @ s e Zd Zddd�Zdd� ZdS )�GenlHdrr c C s || _ || _d S )N)�cmd�version)�selfr r � r � /usr/lib/python3.6/genetlink.py�__init__ s zGenlHdr.__init__c C s t jd| j| j�S )N�BBxx)�struct�packr r )r r r r �_dump! s z
GenlHdr._dumpN)r )�__name__�
__module__� __qualname__r r r r r r r s
r c C s t tjd| �� S )Nr )r r �unpack)�datar r r �_genl_hdr_parse$ s r c @ s& e Zd Zg dfdd�Zedd� �ZdS )� GeNlMessager c C s4 || _ || _|| _tj| ||t| j �g| d� d S )N)�flags�payload)r �attrs�familyr r r )r r$ r r# r! r r r r * s
zGeNlMessage.__init__c C sP | j � }|j}t|d d� �}t|j|jg |j�}t|dd � �|_|j |_ |S )Nr )
�recvr" r r �typer r! r r# r )�conn�msgZpacketZhdrZgenlmsgr r r r% 1 s zGeNlMessage.recvN)r r r r � staticmethodr% r r r r r ) s r c @ s e Zd Zdd� Zdd� ZdS )�
Controllerc C s
|| _ d S )N)r' )r r' r r r r > s zController.__init__c C sB t t|�}tttt|gd�}|j| j� tj| j�}|j t
j � S )N)r! r# ) r �CTRL_ATTR_FAMILY_NAMEr � GENL_ID_CTRL�CTRL_CMD_GETFAMILYr �sendr' r% r# �CTRL_ATTR_FAMILY_ID�u16)r r$ �a�mr r r �
get_family_id@ s
zController.get_family_idN)r r r r r3 r r r r r* = s r* ) �__doc__r Z
iotop.netlinkr r r r r r r ZCTRL_CMD_UNSPECZCTRL_CMD_NEWFAMILYZCTRL_CMD_DELFAMILYr- ZCTRL_CMD_NEWOPSZCTRL_CMD_DELOPSZCTRL_CMD_GETOPSZCTRL_ATTR_UNSPECr/ r+ ZCTRL_ATTR_VERSIONZCTRL_ATTR_HDRSIZEZCTRL_ATTR_MAXATTRZ
CTRL_ATTR_OPSr r r, r r* Z
connectionZ
controllerr r r r �<module> s0 __pycache__/data.cpython-36.pyc 0000644 00000032206 15035135563 0012324 0 ustar 00 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) ^ __pycache__/data.cpython-36.opt-1.pyc 0000644 00000032143 15035135563 0013263 0 ustar 00 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 }t|�S )Ni �H� r )�task_stats_request�sendrB r
Zrecv�OSError�errnoZESRCHrE �items�TASKSTATS_TYPE_AGGR_PID�nested�TASKSTATS_TYPE_STATSr1 �lenr* r+ r )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 rX 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
rh 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 rq 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
rw 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)rz 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 rx � s
rx 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)
rm rg �user�threadsr r< r{ � stats_accum�timeZstats_accum_timestamp)r rm r r r r2 � s
zProcessInfo.__init__c C s: |j r |j r | j|j kr dS |jr6| j� |jkr6dS dS )NFT)�pids� processesrm rZ �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)rg �os�statrm �ST_UIDrO r� )r rg 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� rw r] �getpwuidZpw_namer_ ru r )r rg 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 � )
ri rm �readrl rq r} r[ rj �
startswith�rfind�joinrk rw ) 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� rV 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} rx )r rI r| rV 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 )ry )r� rI rV r r r �
<listcomp>] s z,ProcessInfo.update_stats.<locals>.<listcomp>FT) r r< r� rQ ry r7 r{ �dictrU r r r� )r r{ rI rV 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 rm �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| rX r� r{ r! r# ry r r8 )
r Z
new_timestampZ
total_readZ total_writer� r� rI rV 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� rm r� r r r r� � s z1ProcessList.refresh_processes.<locals>.<listcomp>)r� r� r� ry r� r� rQ )r r� rV 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* ra r� Zprocfsr �path�existsZ ioaccountingZ iotop.vmstatr Zvmstat_fZvm_event_countersr` rd Ziotopr r Z
iotop.netlinkr r r
r Ziotop.genetlinkr r
�objectr r rG rH rR ZTASKSTATS_TYPE_PIDrT r@ rh rq rw rx r� r� r- r r r r �<module> sV
5) ^ __pycache__/version.cpython-36.opt-1.pyc 0000644 00000000177 15035135563 0014041 0 ustar 00 3
B��Q � @ s d Z dS )z0.6N)�VERSION� r r � /usr/lib/python3.6/version.py�<module> s __pycache__/version.cpython-36.pyc 0000644 00000000177 15035135563 0013102 0 ustar 00 3
B��Q � @ s d Z dS )z0.6N)�VERSION� r r � /usr/lib/python3.6/version.py�<module> s __pycache__/netlink.cpython-36.pyc 0000644 00000022075 15035135563 0013062 0 ustar 00 3
B��Q� � @ s� d Z ddlZddlZddlZyDejejejd�Zejd8� [dd� Zdd� Z dd� Z
d9d
d �Z W n� ej k
�r y.ddl
Z
d d� Zd
d� Z dd� Z
d:dd �Z W n\ ek
� r� ddlZejd�ZG dd� dej�Zdd� Zdd� Z dd� Z
d;dd �Z Y nX Y nX dZdZdZdZdZdZdZdZdZ G d d � d �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"dZ#dZ$dZ%dZ&dZ'd(Z(d)Z)d*Z*dZ+d+Z,d,Z-d-Z.d.Z/d/Z0d0Z1d1Z2dZ3G d2d3� d3�Z4G d4d5� d5�Z5d6d7� Z6dS )<z�
Netlink message generation/parsing
Copyright 2007 Johannes Berg <johannes@sipsolutions.net>
GPLv2+; See copying for details.
� Nc C s | j |� d S )N)�bind)�
descriptor�addr� r � /usr/lib/python3.6/netlink.py�_nl_bind s r c C s | j � S )N)� getsockname)r r r r �_nl_getsockname s r c C s | j |� d S )N)�send)r �msgr r r �_nl_send s r � @ c C s
| j |�S )N)�recvfrom)r �bufsr r r �_nl_recv s r c C s t j| j� |d � d S )N� )�_netlinkr �fileno)r r r r r r s c C s t j| j� �S )N)r r r )r r r r r s c C s t j| j� |� d S )N)r r
r )r r r r r r " s c C s t j| j� |�S )N)r r r )r r r r r r $ s c @ s0 e Zd ZdejfdejfdejfdejfgZdS )� SOCKADDR_NLZ nl_familyZnl_pad�nl_pid� nl_groupsN)�__name__�
__module__� __qualname__�ctypesZc_ushort�c_intZ_fields_r r r r r , s r c C s2 t tjdtj� d�}tj| j� tj |�tj
|��S )Nr ) r �socket�
AF_NETLINK�os�getpid�libcr r r �pointer�sizeof)r r r r r r 2 s
c C sH t dddd�}tjtj|��}tj| j� tj|�tj|�� |j|j fS )Nr )
r r r r" r r r r! r r )r r �lenr r r r 8 s
c C s t j| j� |t|�d�S )Nr )r r
r r# )r r r r r r @ s c C sn t dddd�}tjtj|��}tj|�}tj| j� ||dtj|�tj|��}tj tj|�|�}||j
|j ffS )Nr ) r r r r" Zcreate_string_bufferr r r r! Z string_atr r )r r r r# Zbuf�r�retr r r r C s
r � � � � � c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�Attrc G s, || _ t|�r"tj|f|�� | _n|| _d S )N)�typer# �struct�pack�data)�self� attr_typer/ �valuesr r r �__init__^ s z
Attr.__init__c C sJ t jdt| j�d | j�}t| j�}|d d d@ | }|| j d| S )N�HHr' r r) � ���)r- r. r# r/ r, )r0 �hdr�lengthZpadr r r �_dumpe s
z
Attr._dumpc C s d| j t| j�f S )Nz<Attr type %d, data "%s">)r, �reprr/ )r0 r r r �__repr__k s z
Attr.__repr__c C s t jd| j�d S )N�Hr )r- �unpackr/ )r0 r r r �u16n s zAttr.u16c C s t jd| j�d S )N�hr )r- r= r/ )r0 r r r �s16p s zAttr.s16c C s t jd| j�d S )N�Ir )r- r= r/ )r0 r r r �u32r s zAttr.u32c C s t jd| j�d S )N�ir )r- r= r/ )r0 r r r �s32t s zAttr.s32c C s | j S )N)r/ )r0 r r r �strv s zAttr.strc C s | j jd�d S )N� r )r/ �split)r0 r r r �nulstrx s z Attr.nulstrc C s
t | j�S )N)�parse_attributesr/ )r0 r r r �nestedz s z Attr.nestedN)
r r r r3 r9 r; r> r@ rB rD rE rH rJ r r r r r+ ] s r+ c @ s e Zd Zdd� ZdS )�StrAttrc C s" t j| |dt|� |jd�� d S )Nz%dszutf-8)r+ r3 r# �encode)r0 r1 r/ r r r r3 ~ s zStrAttr.__init__N)r r r r3 r r r r rK } s rK c @ s e Zd Zdd� ZdS )�
NulStrAttrc C s$ t j| |dt|� |jd�d� d S )Nz%dsBzutf-8r )r+ r3 r# rL )r0 r1 r/ r r r r3 � s zNulStrAttr.__init__N)r r r r3 r r r r rM � s rM c @ s e Zd Zdd� ZdS )�U32Attrc C s t j| |d|� d S )NrA )r+ r3 )r0 r1 �valr r r r3 � s zU32Attr.__init__N)r r r r3 r r r r rN � s rN c @ s e Zd Zdd� ZdS )�U8Attrc C s t j| |d|� d S )N�B)r+ r3 )r0 r1 rO r r r r3 � s zU8Attr.__init__N)r r r r3 r r r r rP � s rP c @ s e Zd Zdd� Zdd� ZdS )�Nestedc C s || _ || _d S )N)�attrsr, )r0 r1 rS r r r r3 � s zNested.__init__c C sP g }x | j D ]}|j|j� � q W dj|�}t|�}tjd|d | j�}|| S )N� r4 r' )rS �appendr9 �joinr# r- r. r, )r0 �contents�attrr8 r7 r r r r9 � s
z Nested._dumpN)r r r r3 r9 r r r r rR � s rR � � � � �
� � �
� � c @ s& e Zd Zd dd�Zdd� Zdd � ZdS ) �Messager r Nc C sb || _ || _|| _d| _|p g }t|t�rXg }x|D ]}|j|j� � q4W dj|�| _ n|| _ d S )Nr � ���)
r, �flags�seq�pid�
isinstance�listrU r9 rV �payload)r0 �msg_typerf rg rk rW rX r r r r3 � s
zMessage.__init__c C sZ | j dkr|j � | _ |j| _t| j�}tjd|d | j| j| j | j�}|j|| j � d S )Nr �IHHIIr' re r* ) rg rh r# rk r- r. r, rf r
)r0 Zconnr8 r7 r r r r
� s
z Message.sendc C s d| j | j| j| jt| j�f S )Nz:<netlink.Message type=%d, pid=%d, seq=%d, flags=0x%x "%s">)r, rh rg rf r: rk )r0 r r r r; � s zMessage.__repr__re )r re N)r r r r3 r
r; r r r r rc � s
rc c @ s. e Zd Zd dd�Zdd� Zdd� Zd d
� ZdS ) �
Connectionr Nc C sn t j t jt j|�| _| jjt jt jd� | jjt jt jd� t| jd|f� t | j�\| _
| _ d| _ || _
d S )Ni r )r r �SOCK_RAWr Z
setsockoptZ
SOL_SOCKETZ SO_SNDBUFZ SO_RCVBUFr r rh �groups�_seqZ
unexpected)r0 Znltyperp Zunexpected_msg_handlerr r r r3 � s zConnection.__init__c C s t | j|� d S )N)r r )r0 r r r r r
� s zConnection.sendc C s� t | j�\}\}}tjd|d d� �\}}}}}t||||dd � �} || _| jtkr�tjd| jd d� �d }
|
dkr�t dt
j |
�|
f �} |
| _ | �| S )Nrm r* rC r' r zNetlink error: %s (%d))
r r r- r= rc rh r, � NLMSG_ERRORrk �OSErrorr �strerror�errno) r0 rW ZnlpidZnlgrpsZmsglenrl rf rg rh r ru �errr r r �recv� s
zConnection.recvc C s | j d7 _ | j S )Nr )rq )r0 r r r rg � s zConnection.seq)r N)r r r r3 r
rw rg r r r r rn � s
rn c C s` i }xVt | �rZtjd| d d� �\}}t|| d|� �||< |d d d@ }| |d � } qW |S )Nr4 r' r r) r6 )r# r- r= r+ )r/ rS Zattr_lenr1 r r r rI � s
rI )r r )r
)r
)r
)7�__doc__r r r- r ro Z
_dummysockr r r r r �errorr � ImportErrorr ZCDLLr Z Structurer Z
NLM_F_REQUESTZ NLM_F_MULTIZ NLM_F_ACKZ
NLM_F_ECHOZ
NLMSG_NOOPrr Z
NLMSG_DONEZ
NLMSG_OVERRUNZNLMSG_MIN_TYPEr+ rK rM rN rP rR Z
NETLINK_ROUTEZNETLINK_UNUSEDZNETLINK_USERSOCKZNETLINK_FIREWALLZNETLINK_INET_DIAGZ
NETLINK_NFLOGZ NETLINK_XFRMZNETLINK_SELINUXZ
NETLINK_ISCSIZ
NETLINK_AUDITZNETLINK_FIB_LOOKUPZNETLINK_CONNECTORZNETLINK_NETFILTERZNETLINK_IP6_FWZNETLINK_DNRTMSGZNETLINK_KOBJECT_UEVENTZNETLINK_GENERICrc rn rI r r r r �<module> sz
__pycache__/netlink.cpython-36.opt-1.pyc 0000644 00000022075 15035135563 0014021 0 ustar 00 3
B��Q� � @ s� d Z ddlZddlZddlZyDejejejd�Zejd8� [dd� Zdd� Z dd� Z
d9d
d �Z W n� ej k
�r y.ddl
Z
d d� Zd
d� Z dd� Z
d:dd �Z W n\ ek
� r� ddlZejd�ZG dd� dej�Zdd� Zdd� Z dd� Z
d;dd �Z Y nX Y nX dZdZdZdZdZdZdZdZdZ G d d � d �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"dZ#dZ$dZ%dZ&dZ'd(Z(d)Z)d*Z*dZ+d+Z,d,Z-d-Z.d.Z/d/Z0d0Z1d1Z2dZ3G d2d3� d3�Z4G d4d5� d5�Z5d6d7� Z6dS )<z�
Netlink message generation/parsing
Copyright 2007 Johannes Berg <johannes@sipsolutions.net>
GPLv2+; See copying for details.
� Nc C s | j |� d S )N)�bind)�
descriptor�addr� r � /usr/lib/python3.6/netlink.py�_nl_bind s r c C s | j � S )N)� getsockname)r r r r �_nl_getsockname s r c C s | j |� d S )N)�send)r �msgr r r �_nl_send s r � @ c C s
| j |�S )N)�recvfrom)r �bufsr r r �_nl_recv s r c C s t j| j� |d � d S )N� )�_netlinkr �fileno)r r r r r r s c C s t j| j� �S )N)r r r )r r r r r s c C s t j| j� |� d S )N)r r
r )r r r r r r " s c C s t j| j� |�S )N)r r r )r r r r r r $ s c @ s0 e Zd ZdejfdejfdejfdejfgZdS )� SOCKADDR_NLZ nl_familyZnl_pad�nl_pid� nl_groupsN)�__name__�
__module__� __qualname__�ctypesZc_ushort�c_intZ_fields_r r r r r , s r c C s2 t tjdtj� d�}tj| j� tj |�tj
|��S )Nr ) r �socket�
AF_NETLINK�os�getpid�libcr r r �pointer�sizeof)r r r r r r 2 s
c C sH t dddd�}tjtj|��}tj| j� tj|�tj|�� |j|j fS )Nr )
r r r r" r r r r! r r )r r �lenr r r r 8 s
c C s t j| j� |t|�d�S )Nr )r r
r r# )r r r r r r @ s c C sn t dddd�}tjtj|��}tj|�}tj| j� ||dtj|�tj|��}tj tj|�|�}||j
|j ffS )Nr ) r r r r" Zcreate_string_bufferr r r r! Z string_atr r )r r r r# Zbuf�r�retr r r r C s
r � � � � � c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�Attrc G s, || _ t|�r"tj|f|�� | _n|| _d S )N)�typer# �struct�pack�data)�self� attr_typer/ �valuesr r r �__init__^ s z
Attr.__init__c C sJ t jdt| j�d | j�}t| j�}|d d d@ | }|| j d| S )N�HHr' r r) � ���)r- r. r# r/ r, )r0 �hdr�lengthZpadr r r �_dumpe s
z
Attr._dumpc C s d| j t| j�f S )Nz<Attr type %d, data "%s">)r, �reprr/ )r0 r r r �__repr__k s z
Attr.__repr__c C s t jd| j�d S )N�Hr )r- �unpackr/ )r0 r r r �u16n s zAttr.u16c C s t jd| j�d S )N�hr )r- r= r/ )r0 r r r �s16p s zAttr.s16c C s t jd| j�d S )N�Ir )r- r= r/ )r0 r r r �u32r s zAttr.u32c C s t jd| j�d S )N�ir )r- r= r/ )r0 r r r �s32t s zAttr.s32c C s | j S )N)r/ )r0 r r r �strv s zAttr.strc C s | j jd�d S )N� r )r/ �split)r0 r r r �nulstrx s z Attr.nulstrc C s
t | j�S )N)�parse_attributesr/ )r0 r r r �nestedz s z Attr.nestedN)
r r r r3 r9 r; r> r@ rB rD rE rH rJ r r r r r+ ] s r+ c @ s e Zd Zdd� ZdS )�StrAttrc C s" t j| |dt|� |jd�� d S )Nz%dszutf-8)r+ r3 r# �encode)r0 r1 r/ r r r r3 ~ s zStrAttr.__init__N)r r r r3 r r r r rK } s rK c @ s e Zd Zdd� ZdS )�
NulStrAttrc C s$ t j| |dt|� |jd�d� d S )Nz%dsBzutf-8r )r+ r3 r# rL )r0 r1 r/ r r r r3 � s zNulStrAttr.__init__N)r r r r3 r r r r rM � s rM c @ s e Zd Zdd� ZdS )�U32Attrc C s t j| |d|� d S )NrA )r+ r3 )r0 r1 �valr r r r3 � s zU32Attr.__init__N)r r r r3 r r r r rN � s rN c @ s e Zd Zdd� ZdS )�U8Attrc C s t j| |d|� d S )N�B)r+ r3 )r0 r1 rO r r r r3 � s zU8Attr.__init__N)r r r r3 r r r r rP � s rP c @ s e Zd Zdd� Zdd� ZdS )�Nestedc C s || _ || _d S )N)�attrsr, )r0 r1 rS r r r r3 � s zNested.__init__c C sP g }x | j D ]}|j|j� � q W dj|�}t|�}tjd|d | j�}|| S )N� r4 r' )rS �appendr9 �joinr# r- r. r, )r0 �contents�attrr8 r7 r r r r9 � s
z Nested._dumpN)r r r r3 r9 r r r r rR � s rR � � � � �
� � �
� � c @ s& e Zd Zd dd�Zdd� Zdd � ZdS ) �Messager r Nc C sb || _ || _|| _d| _|p g }t|t�rXg }x|D ]}|j|j� � q4W dj|�| _ n|| _ d S )Nr � ���)
r, �flags�seq�pid�
isinstance�listrU r9 rV �payload)r0 �msg_typerf rg rk rW rX r r r r3 � s
zMessage.__init__c C sZ | j dkr|j � | _ |j| _t| j�}tjd|d | j| j| j | j�}|j|| j � d S )Nr �IHHIIr' re r* ) rg rh r# rk r- r. r, rf r
)r0 Zconnr8 r7 r r r r
� s
z Message.sendc C s d| j | j| j| jt| j�f S )Nz:<netlink.Message type=%d, pid=%d, seq=%d, flags=0x%x "%s">)r, rh rg rf r: rk )r0 r r r r; � s zMessage.__repr__re )r re N)r r r r3 r
r; r r r r rc � s
rc c @ s. e Zd Zd dd�Zdd� Zdd� Zd d
� ZdS ) �
Connectionr Nc C sn t j t jt j|�| _| jjt jt jd� | jjt jt jd� t| jd|f� t | j�\| _
| _ d| _ || _
d S )Ni r )r r �SOCK_RAWr Z
setsockoptZ
SOL_SOCKETZ SO_SNDBUFZ SO_RCVBUFr r rh �groups�_seqZ
unexpected)r0 Znltyperp Zunexpected_msg_handlerr r r r3 � s zConnection.__init__c C s t | j|� d S )N)r r )r0 r r r r r
� s zConnection.sendc C s� t | j�\}\}}tjd|d d� �\}}}}}t||||dd � �} || _| jtkr�tjd| jd d� �d }
|
dkr�t dt
j |
�|
f �} |
| _ | �| S )Nrm r* rC r' r zNetlink error: %s (%d))
r r r- r= rc rh r, � NLMSG_ERRORrk �OSErrorr �strerror�errno) r0 rW ZnlpidZnlgrpsZmsglenrl rf rg rh r ru �errr r r �recv� s
zConnection.recvc C s | j d7 _ | j S )Nr )rq )r0 r r r rg � s zConnection.seq)r N)r r r r3 r
rw rg r r r r rn � s
rn c C s` i }xVt | �rZtjd| d d� �\}}t|| d|� �||< |d d d@ }| |d � } qW |S )Nr4 r' r r) r6 )r# r- r= r+ )r/ rS Zattr_lenr1 r r r rI � s
rI )r r )r
)r
)r
)7�__doc__r r r- r ro Z
_dummysockr r r r r �errorr � ImportErrorr ZCDLLr Z Structurer Z
NLM_F_REQUESTZ NLM_F_MULTIZ NLM_F_ACKZ
NLM_F_ECHOZ
NLMSG_NOOPrr Z
NLMSG_DONEZ
NLMSG_OVERRUNZNLMSG_MIN_TYPEr+ rK rM rN rP rR Z
NETLINK_ROUTEZNETLINK_UNUSEDZNETLINK_USERSOCKZNETLINK_FIREWALLZNETLINK_INET_DIAGZ
NETLINK_NFLOGZ NETLINK_XFRMZNETLINK_SELINUXZ
NETLINK_ISCSIZ
NETLINK_AUDITZNETLINK_FIB_LOOKUPZNETLINK_CONNECTORZNETLINK_NETFILTERZNETLINK_IP6_FWZNETLINK_DNRTMSGZNETLINK_KOBJECT_UEVENTZNETLINK_GENERICrc rn rI r r r r �<module> sz
__pycache__/genetlink.cpython-36.pyc 0000644 00000005263 15035135563 0013376 0 ustar 00 3
B��Q� � @ s� d Z ddlZddlmZmZmZmZ ddlmZmZm Z dZ
dZ dZ dZ
dZd Zd
ZdZdZdZdZdZd Zd
ZG d d � d �Zd
d� ZeZG dd� de�ZG dd� d�Z ee �Z e e �Z dS )z�
Netlink message generation/parsing
Copyright 2007 Johannes Berg <johannes@sipsolutions.net>
GPLv2+; See copying for details.
� N)�
NLM_F_REQUEST�NLMSG_MIN_TYPE�Message�parse_attributes)�
NulStrAttr�
Connection�NETLINK_GENERIC� � � � � � c @ s e Zd Zddd�Zdd� ZdS )�GenlHdrr c C s || _ || _d S )N)�cmd�version)�selfr r � r � /usr/lib/python3.6/genetlink.py�__init__ s zGenlHdr.__init__c C s t jd| j| j�S )N�BBxx)�struct�packr r )r r r r �_dump! s z
GenlHdr._dumpN)r )�__name__�
__module__� __qualname__r r r r r r r s
r c C s t tjd| �� S )Nr )r r �unpack)�datar r r �_genl_hdr_parse$ s r c @ s& e Zd Zg dfdd�Zedd� �ZdS )� GeNlMessager c C s4 || _ || _|| _tj| ||t| j �g| d� d S )N)�flags�payload)r �attrs�familyr r r )r r$ r r# r! r r r r * s
zGeNlMessage.__init__c C sP | j � }|j}t|d d� �}t|j|jg |j�}t|dd � �|_|j |_ |S )Nr )
�recvr" r r �typer r! r r# r )�conn�msgZpacketZhdrZgenlmsgr r r r% 1 s zGeNlMessage.recvN)r r r r � staticmethodr% r r r r r ) s r c @ s e Zd Zdd� Zdd� ZdS )�
Controllerc C s
|| _ d S )N)r' )r r' r r r r > s zController.__init__c C sB t t|�}tttt|gd�}|j| j� tj| j�}|j t
j � S )N)r! r# ) r �CTRL_ATTR_FAMILY_NAMEr � GENL_ID_CTRL�CTRL_CMD_GETFAMILYr �sendr' r% r# �CTRL_ATTR_FAMILY_ID�u16)r r$ �a�mr r r �
get_family_id@ s
zController.get_family_idN)r r r r r3 r r r r r* = s r* ) �__doc__r Z
iotop.netlinkr r r r r r r ZCTRL_CMD_UNSPECZCTRL_CMD_NEWFAMILYZCTRL_CMD_DELFAMILYr- ZCTRL_CMD_NEWOPSZCTRL_CMD_DELOPSZCTRL_CMD_GETOPSZCTRL_ATTR_UNSPECr/ r+ ZCTRL_ATTR_VERSIONZCTRL_ATTR_HDRSIZEZCTRL_ATTR_MAXATTRZ
CTRL_ATTR_OPSr r r, r r* Z
connectionZ
controllerr r r r �<module> s0 __pycache__/vmstat.cpython-36.pyc 0000644 00000002056 15035135563 0012731 0 ustar 00 3
B��Q � @ s G d d� de �ZdS )c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�VmStatc C s t d�| _| j� | _d S )Nz /proc/vmstat)�open� vmstat_file�read�vmstat)�self� r � /usr/lib/python3.6/vmstat.py�__init__ s
zVmStat.__init__c C sd dd� }x"| j D ]}|jd�r||�}P qW x"| j D ]}|jd�r4||�}P q4W | j jd� ||fS )Nc S s t | j� d �d S )N� i )�int�split)�liner r r �extract s z VmStat.read.<locals>.extractzpgpgin zpgpgout � )r �
startswith�seek)r r r
ZpgpginZpgpgoutr r r r s
z VmStat.readc C s6 | j � }|d | jd |d | jd f}|| _|S )Nr r
)r r )r Znow�deltar r r r ) s $z VmStat.deltaN)�__name__�
__module__� __qualname__r r r r r r r r s r N)�objectr r r r r �<module> s __pycache__/ui.cpython-36.pyc 0000644 00000050456 15035135563 0012037 0 ustar 00 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
$$ __pycache__/ioprio.cpython-36.pyc 0000644 00000010102 15035135563 0012703 0 ustar 00 3
B��Qw � @ s� d dl Z d dlZd dlZd dlZdBdCdDdEdFdGdHdIdJdKdLg ZdMdNdOdPdQdRdSdTdUdVdWg Zd"d#� ZG d$d%� d%e�Zee�Z ee�Z
ye j dd&d'�Z W n e
k
� r� e j d�Z Y nX e jZdd(d)d*gZd+Zd,Zd+e> d+ Zd-d.� Zd/d0� Zd1d2� Ze jZed3�\ZZZZZ Z e j Z d Z d4d5� Z d6d7� Z!d8d9� Z"d:d;� Z#d<d=� Z$e%d>k�r�d dl&Z&e'e&j(�d?k�rle)e&j(d+ �Z*nej+� Z*e,d@e*� e,dAe!e*�� dS )X� N�alpha�*� �arm*�; �i*86�" �ia64*� �parisc*� �powerpc*� �s390*� �sparc*�� �sh*�! �x86_64*�32bit�64bit� � �: � � � � �� � � c C sL t j� d }tj� d }x.| D ]&\}}}tj||�r tj||�r |S q W d S )N� r )�os�uname�platformZ architecture�fnmatch)Z syscall_listZarch�bitsZcandidate_archZcandidate_bitsZ
syscall_nr� r( � /usr/lib/python3.6/ioprio.py�find_ioprio_syscall_number9 s r* c @ s e Zd Zdd� ZdS )�IoprioSetErrorc C s0 yt j|�| _W n tk
r* || _Y nX d S )N)r# �strerror�err� TypeError)�selfr- r( r( r) �__init__C s zIoprioSetError.__init__N)�__name__�
__module__� __qualname__r0 r( r( r( r) r+ B s r+ T)� use_errnoZrtZbe�idle� �
c C s: yt j| �} W n tk
r, t jd �} Y nX | t> |B S )N)�PRIORITY_CLASSES�index�
ValueError�IOPRIO_CLASS_SHIFT)� ioprio_class� ioprio_datar( r( r) � ioprio_valueY s
r> c C s t | t? S )N)r8 r; )�ioprior( r( r) r< ` s r< c C s | t @ S )N)�IOPRIO_PRIO_MASK)r? r( r( r) r= c s r= � c C sJ t | �}tt| �}|d d }|ttfkr2d| S |tkr>dS d| S d S )N� � zrt/%dr5 zbe/%d)�sched_getscheduler� getpriority� PRIO_PROCESS�
SCHED_FIFO�SCHED_RR�
SCHED_IDLE)�pidZ scheduler�niceZ ioprio_nicer( r( r) �get_ioprio_from_schedl s
rL c C sT t d kr dS tt t| �}|dk r$dS t|�}|s8t| �S |dkrD|S d|t|�f S )Nz?sysr z?errr5 z%s/%d)�__NR_ioprio_get�syscall�IOPRIO_WHO_PROCESSr< rL r= )rJ r? Z
prio_classr( r( r) �getx s rP c C sf t d krtd��t||�}tt | ||dd�}|dk rby tj� }W n tk
rX d}Y nX t|��d S )NzNo ioprio_set syscall foundT)r4 r z<Unknown error (errno support not available before Python2.6))�__NR_ioprio_setr+ r> rN �ctypesZ get_errno�AttributeError)ZwhichZwhor< r= Z
ioprio_val�retr- r( r( r) �
set_ioprio� s
rU c C st | d dkrt | d � S d| krT| jd�r2d}n| jd�r@d}t| jd�d �}n| dkrdd}d}d|t > | S ) Nr �?r6 �/zrt/zbe/r5 � )�ord�
startswith�int�splitr; )�keyZshiftZprior( r( r) �sort_key� s
r^ c C s6 d| kr&| j d�}|d t|d �fS | dkr2dS dS )NrW r r6 r5 )r5 r )NN)r\ r[ )Z
ioprio_strr\ r( r( r) �to_class_and_data� s
r_ �__main__rX zpid:zioprio:)r r r )r r r )r r r )r r r
)r r r )r
r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r
r r )r r r )r r r )r r r )r r r )r r r! )-rR r& r# r% ZIOPRIO_GET_ARCH_SYSCALLZIOPRIO_SET_ARCH_SYSCALLr* � Exceptionr+ rM rQ ZCDLLZ
ctypes_handler. rN r8 rO r; r@ r> r< r= rD �range� SCHED_OTHERrG rH � SCHED_BATCHZ SCHED_ISOrI rE rF rL rP rU r^ r_ r1 �sys�len�argvr[ rJ �getpid�printr( r( r( r) �<module> sr
__pycache__/__init__.cpython-36.pyc 0000644 00000000161 15035135563 0013145 0 ustar 00 3
B��Q � @ s d S )N� r r r � /usr/lib/python3.6/__init__.py�<module> s __pycache__/__init__.cpython-36.opt-1.pyc 0000644 00000000161 15035135563 0014104 0 ustar 00 3
B��Q � @ s d S )N� r r r � /usr/lib/python3.6/__init__.py�<module> s __pycache__/ui.cpython-36.opt-1.pyc 0000644 00000050456 15035135563 0012776 0 ustar 00 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
$$ __pycache__/vmstat.cpython-36.opt-1.pyc 0000644 00000002056 15035135563 0013670 0 ustar 00 3
B��Q � @ s G d d� de �ZdS )c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�VmStatc C s t d�| _| j� | _d S )Nz /proc/vmstat)�open� vmstat_file�read�vmstat)�self� r � /usr/lib/python3.6/vmstat.py�__init__ s
zVmStat.__init__c C sd dd� }x"| j D ]}|jd�r||�}P qW x"| j D ]}|jd�r4||�}P q4W | j jd� ||fS )Nc S s t | j� d �d S )N� i )�int�split)�liner r r �extract s z VmStat.read.<locals>.extractzpgpgin zpgpgout � )r �
startswith�seek)r r r
ZpgpginZpgpgoutr r r r s
z VmStat.readc C s6 | j � }|d | jd |d | jd f}|| _|S )Nr r
)r r )r Znow�deltar r r r ) s $z VmStat.deltaN)�__name__�
__module__� __qualname__r r r r r r r r s r N)�objectr r r r r �<module> s __pycache__/ioprio.cpython-36.opt-1.pyc 0000644 00000010102 15035135563 0013642 0 ustar 00 3
B��Qw � @ s� d dl Z d dlZd dlZd dlZdBdCdDdEdFdGdHdIdJdKdLg ZdMdNdOdPdQdRdSdTdUdVdWg Zd"d#� ZG d$d%� d%e�Zee�Z ee�Z
ye j dd&d'�Z W n e
k
� r� e j d�Z Y nX e jZdd(d)d*gZd+Zd,Zd+e> d+ Zd-d.� Zd/d0� Zd1d2� Ze jZed3�\ZZZZZ Z e j Z d Z d4d5� Z d6d7� Z!d8d9� Z"d:d;� Z#d<d=� Z$e%d>k�r�d dl&Z&e'e&j(�d?k�rle)e&j(d+ �Z*nej+� Z*e,d@e*� e,dAe!e*�� dS )X� N�alpha�*� �arm*�; �i*86�" �ia64*� �parisc*� �powerpc*� �s390*� �sparc*�� �sh*�! �x86_64*�32bit�64bit� � �: � � � � �� � � c C sL t j� d }tj� d }x.| D ]&\}}}tj||�r tj||�r |S q W d S )N� r )�os�uname�platformZ architecture�fnmatch)Z syscall_listZarch�bitsZcandidate_archZcandidate_bitsZ
syscall_nr� r( � /usr/lib/python3.6/ioprio.py�find_ioprio_syscall_number9 s r* c @ s e Zd Zdd� ZdS )�IoprioSetErrorc C s0 yt j|�| _W n tk
r* || _Y nX d S )N)r# �strerror�err� TypeError)�selfr- r( r( r) �__init__C s zIoprioSetError.__init__N)�__name__�
__module__� __qualname__r0 r( r( r( r) r+ B s r+ T)� use_errnoZrtZbe�idle� �
c C s: yt j| �} W n tk
r, t jd �} Y nX | t> |B S )N)�PRIORITY_CLASSES�index�
ValueError�IOPRIO_CLASS_SHIFT)� ioprio_class� ioprio_datar( r( r) � ioprio_valueY s
r> c C s t | t? S )N)r8 r; )�ioprior( r( r) r< ` s r< c C s | t @ S )N)�IOPRIO_PRIO_MASK)r? r( r( r) r= c s r= � c C sJ t | �}tt| �}|d d }|ttfkr2d| S |tkr>dS d| S d S )N� � zrt/%dr5 zbe/%d)�sched_getscheduler� getpriority� PRIO_PROCESS�
SCHED_FIFO�SCHED_RR�
SCHED_IDLE)�pidZ scheduler�niceZ ioprio_nicer( r( r) �get_ioprio_from_schedl s
rL c C sT t d kr dS tt t| �}|dk r$dS t|�}|s8t| �S |dkrD|S d|t|�f S )Nz?sysr z?errr5 z%s/%d)�__NR_ioprio_get�syscall�IOPRIO_WHO_PROCESSr< rL r= )rJ r? Z
prio_classr( r( r) �getx s rP c C sf t d krtd��t||�}tt | ||dd�}|dk rby tj� }W n tk
rX d}Y nX t|��d S )NzNo ioprio_set syscall foundT)r4 r z<Unknown error (errno support not available before Python2.6))�__NR_ioprio_setr+ r> rN �ctypesZ get_errno�AttributeError)ZwhichZwhor< r= Z
ioprio_val�retr- r( r( r) �
set_ioprio� s
rU c C st | d dkrt | d � S d| krT| jd�r2d}n| jd�r@d}t| jd�d �}n| dkrdd}d}d|t > | S ) Nr �?r6 �/zrt/zbe/r5 � )�ord�
startswith�int�splitr; )�keyZshiftZprior( r( r) �sort_key� s
r^ c C s6 d| kr&| j d�}|d t|d �fS | dkr2dS dS )NrW r r6 r5 )r5 r )NN)r\ r[ )Z
ioprio_strr\ r( r( r) �to_class_and_data� s
r_ �__main__rX zpid:zioprio:)r r r )r r r )r r r )r r r
)r r r )r
r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r
r r )r r r )r r r )r r r )r r r )r r r! )-rR r& r# r% ZIOPRIO_GET_ARCH_SYSCALLZIOPRIO_SET_ARCH_SYSCALLr* � Exceptionr+ rM rQ ZCDLLZ
ctypes_handler. rN r8 rO r; r@ r> r< r= rD �range� SCHED_OTHERrG rH � SCHED_BATCHZ SCHED_ISOrI rE rF rL rP rU r^ r_ r1 �sys�len�argvr[ rJ �getpid�printr( r( r( r) �<module> sr
ui.py 0000644 00000061261 15035135563 0005547 0 ustar 00 # This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Library General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
#
# See the COPYING file for license information.
#
# Copyright (c) 2007 Guillaume Chazarain <guichaz@gmail.com>
# Allow printing with same syntax in Python 2/3
from __future__ import print_function
import curses
import errno
import locale
import math
import optparse
import os
import select
import signal
import sys
import time
from iotop.data import find_uids, TaskStatsNetlink, ProcessList, Stats, sysctl_task_delayacct
from iotop.data import ThreadInfo
from iotop.version import VERSION
from iotop import ioprio
from iotop.ioprio import IoprioSetError
#
# Utility functions for the UI
#
UNITS = ['B', 'K', 'M', 'G', 'T', 'P', 'E']
def human_size(size):
if size > 0:
sign = ''
elif size < 0:
sign = '-'
size = -size
else:
return '0.00 B'
expo = int(math.log(size / 2, 2) / 10)
return '%s%.2f %s' % (sign, (float(size) / (1 << (10 * expo))), UNITS[expo])
def format_size(options, bytes):
if options.kilobytes:
return '%.2f K' % (bytes / 1024.0)
return human_size(bytes)
def format_bandwidth(options, size, duration):
return format_size(options, size and float(size) / duration) + '/s'
def format_stats(options, process, duration):
# Keep in sync with TaskStatsNetlink.members_offsets and
# IOTopUI.get_data(self)
def delay2percent(delay): # delay in ns, duration in s
return '%.2f %%' % min(99.99, delay / (duration * 10000000.0))
if options.accumulated:
stats = process.stats_accum
display_format = lambda size, duration: format_size(options, size)
duration = time.time() - process.stats_accum_timestamp
else:
stats = process.stats_delta
display_format = lambda size, duration: format_bandwidth(
options, size, duration)
io_delay = delay2percent(stats.blkio_delay_total)
swapin_delay = delay2percent(stats.swapin_delay_total)
read_bytes = display_format(stats.read_bytes, duration)
written_bytes = stats.write_bytes - stats.cancelled_write_bytes
written_bytes = max(0, written_bytes)
write_bytes = display_format(written_bytes, duration)
return io_delay, swapin_delay, read_bytes, write_bytes
def get_max_pid_width():
try:
return len(open('/proc/sys/kernel/pid_max').read().strip())
except Exception as e:
print(e)
# Reasonable default in case something fails
return 5
MAX_PID_WIDTH = get_max_pid_width()
#
# UI Exceptions
#
class CancelInput(Exception): pass
class InvalidInt(Exception): pass
class InvalidPid(Exception): pass
class InvalidTid(Exception): pass
class InvalidIoprioData(Exception): pass
#
# The UI
#
class IOTopUI(object):
# key, reverse
sorting_keys = [
(lambda p, s: p.pid, False),
(lambda p, s: p.ioprio_sort_key(), False),
(lambda p, s: p.get_user(), False),
(lambda p, s: s.read_bytes, True),
(lambda p, s: s.write_bytes - s.cancelled_write_bytes, True),
(lambda p, s: s.swapin_delay_total, True),
# The default sorting (by I/O % time) should show processes doing
# only writes, without waiting on them
(lambda p, s: s.blkio_delay_total or
int(not(not(s.read_bytes or s.write_bytes))), True),
(lambda p, s: p.get_cmdline(), False),
]
def __init__(self, win, process_list, options):
self.process_list = process_list
self.options = options
self.sorting_key = 6
self.sorting_reverse = IOTopUI.sorting_keys[self.sorting_key][1]
if not self.options.batch:
self.win = win
self.resize()
try:
curses.use_default_colors()
curses.start_color()
curses.curs_set(0)
except curses.error:
# This call can fail with misconfigured terminals, for example
# TERM=xterm-color. This is harmless
pass
def resize(self):
self.height, self.width = self.win.getmaxyx()
def run(self):
iterations = 0
poll = select.poll()
if not self.options.batch:
poll.register(sys.stdin.fileno(), select.POLLIN|select.POLLPRI)
while self.options.iterations is None or \
iterations < self.options.iterations:
total, actual = self.process_list.refresh_processes()
self.refresh_display(iterations == 0, total, actual,
self.process_list.duration)
if self.options.iterations is not None:
iterations += 1
if iterations >= self.options.iterations:
break
elif iterations == 0:
iterations = 1
try:
events = poll.poll(self.options.delay_seconds * 1000.0)
except select.error as e:
if e.args and e.args[0] == errno.EINTR:
events = []
else:
raise
for (fd, event) in events:
if event & (select.POLLERR | select.POLLHUP):
sys.exit(1)
if not self.options.batch:
self.resize()
if events:
key = self.win.getch()
self.handle_key(key)
def reverse_sorting(self):
self.sorting_reverse = not self.sorting_reverse
def adjust_sorting_key(self, delta):
orig_sorting_key = self.sorting_key
self.sorting_key += delta
self.sorting_key = max(0, self.sorting_key)
self.sorting_key = min(len(IOTopUI.sorting_keys) - 1, self.sorting_key)
if not self.has_swapin_io:
if self.sorting_key in (5, 6):
if delta <= 0:
self.sorting_key = 4
elif delta > 0:
self.sorting_key = 7
if orig_sorting_key != self.sorting_key:
self.sorting_reverse = IOTopUI.sorting_keys[self.sorting_key][1]
# I wonder if switching to urwid for the display would be better here
def prompt_str(self, prompt, default=None, empty_is_cancel=True):
self.win.hline(1, 0, ord(' ') | curses.A_NORMAL, self.width)
self.win.addstr(1, 0, prompt, curses.A_BOLD)
self.win.refresh()
curses.echo()
curses.curs_set(1)
inp = self.win.getstr(1, len(prompt))
curses.curs_set(0)
curses.noecho()
if inp not in (None, ''):
return inp
if empty_is_cancel:
raise CancelInput()
return default
def prompt_int(self, prompt, default = None, empty_is_cancel = True):
inp = self.prompt_str(prompt, default, empty_is_cancel)
try:
return int(inp)
except ValueError:
raise InvalidInt()
def prompt_pid(self):
try:
return self.prompt_int('PID to ionice: ')
except InvalidInt:
raise InvalidPid()
except CancelInput:
raise
def prompt_tid(self):
try:
return self.prompt_int('TID to ionice: ')
except InvalidInt:
raise InvalidTid()
except CancelInput:
raise
def prompt_data(self, ioprio_data):
try:
if ioprio_data is not None:
inp = self.prompt_int('I/O priority data (0-7, currently %s): '
% ioprio_data, ioprio_data, False)
else:
inp = self.prompt_int('I/O priority data (0-7): ', None, False)
except InvalidInt:
raise InvalidIoprioData()
if inp < 0 or inp > 7:
raise InvalidIoprioData()
return inp
def prompt_set(self, prompt, display_list, ret_list, selected):
try:
selected = ret_list.index(selected)
except ValueError:
selected = -1
set_len = len(display_list) - 1
while True:
self.win.hline(1, 0, ord(' ') | curses.A_NORMAL, self.width)
self.win.insstr(1, 0, prompt, curses.A_BOLD)
offset = len(prompt)
for i, item in enumerate(display_list):
display = ' %s ' % item
if i is selected:
attr = curses.A_REVERSE
else:
attr = curses.A_NORMAL
self.win.insstr(1, offset, display, attr)
offset += len(display)
while True:
key = self.win.getch()
if key in (curses.KEY_LEFT, ord('l')) and selected > 0:
selected -= 1
break
elif key in (curses.KEY_RIGHT, ord('r')) and selected < set_len:
selected += 1
break
elif key in (curses.KEY_ENTER, ord('\n'), ord('\r')):
return ret_list[selected]
elif key in (27, curses.KEY_CANCEL, curses.KEY_CLOSE,
curses.KEY_EXIT, ord('q'), ord('Q')):
raise CancelInput()
def prompt_class(self, ioprio_class=None):
prompt = 'I/O priority class: '
classes_prompt = ['Real-time', 'Best-effort', 'Idle']
classes_ret = ['rt', 'be', 'idle']
if ioprio_class is None:
ioprio_class = 2
inp = self.prompt_set(prompt, classes_prompt, classes_ret, ioprio_class)
return inp
def prompt_error(self, error = 'Error!'):
self.win.hline(1, 0, ord(' ') | curses.A_NORMAL, self.width)
self.win.insstr(1, 0, ' %s ' % error, curses.A_REVERSE)
self.win.refresh()
time.sleep(1)
def prompt_clear(self):
self.win.hline(1, 0, ord(' ') | curses.A_NORMAL, self.width)
self.win.refresh()
def handle_key(self, key):
def toggle_accumulated():
self.options.accumulated ^= True
def toggle_only_io():
self.options.only ^= True
def toggle_processes():
self.options.processes ^= True
self.process_list.clear()
self.process_list.refresh_processes()
def ionice():
try:
if self.options.processes:
pid = self.prompt_pid()
exec_unit = self.process_list.get_process(pid)
else:
tid = self.prompt_tid()
exec_unit = ThreadInfo(tid,
self.process_list.taskstats_connection)
ioprio_value = exec_unit.get_ioprio()
(ioprio_class, ioprio_data) = \
ioprio.to_class_and_data(ioprio_value)
ioprio_class = self.prompt_class(ioprio_class)
if ioprio_class == 'idle':
ioprio_data = 0
else:
ioprio_data = self.prompt_data(ioprio_data)
exec_unit.set_ioprio(ioprio_class, ioprio_data)
self.process_list.clear()
self.process_list.refresh_processes()
except IoprioSetError as e:
self.prompt_error('Error setting I/O priority: %s' % e.err)
except InvalidPid:
self.prompt_error('Invalid process id!')
except InvalidTid:
self.prompt_error('Invalid thread id!')
except InvalidIoprioData:
self.prompt_error('Invalid I/O priority data!')
except InvalidInt:
self.prompt_error('Invalid integer!')
except CancelInput:
self.prompt_clear()
else:
self.prompt_clear()
key_bindings = {
ord('q'):
lambda: sys.exit(0),
ord('Q'):
lambda: sys.exit(0),
ord('r'):
lambda: self.reverse_sorting(),
ord('R'):
lambda: self.reverse_sorting(),
ord('a'):
toggle_accumulated,
ord('A'):
toggle_accumulated,
ord('o'):
toggle_only_io,
ord('O'):
toggle_only_io,
ord('p'):
toggle_processes,
ord('P'):
toggle_processes,
ord('i'):
ionice,
ord('I'):
ionice,
curses.KEY_LEFT:
lambda: self.adjust_sorting_key(-1),
curses.KEY_RIGHT:
lambda: self.adjust_sorting_key(1),
curses.KEY_HOME:
lambda: self.adjust_sorting_key(-len(IOTopUI.sorting_keys)),
curses.KEY_END:
lambda: self.adjust_sorting_key(len(IOTopUI.sorting_keys))
}
action = key_bindings.get(key, lambda: None)
action()
def get_data(self):
def format(p):
stats = format_stats(self.options, p, self.process_list.duration)
io_delay, swapin_delay, read_bytes, write_bytes = stats
format = '%%%dd' % MAX_PID_WIDTH
params = p.pid,
format += ' %4s'
params += p.get_ioprio(),
format += ' %-8s'
params += p.get_user()[:8],
format += ' %11s %11s'
params += read_bytes, write_bytes
if self.has_swapin_io:
format += ' %7s %7s'
params += swapin_delay, io_delay
elif self.options.batch:
format += ' %s '
params += '?unavailable?',
format += ' '
line = format % (params)
cmdline = p.get_cmdline()
if not self.options.batch:
remaining_length = self.width - len(line)
if 2 < remaining_length < len(cmdline):
len1 = (remaining_length - 1) // 2
offset2 = -(remaining_length - len1 - 1)
cmdline = cmdline[:len1] + '~' + cmdline[offset2:]
line += cmdline
if not self.options.batch:
line = line[:self.width]
return line
def should_format(p):
return not self.options.only or \
p.did_some_io(self.options.accumulated)
processes = list(filter(should_format,
self.process_list.processes.values()))
key = IOTopUI.sorting_keys[self.sorting_key][0]
if self.options.accumulated:
stats_lambda = lambda p: p.stats_accum
else:
stats_lambda = lambda p: p.stats_delta
processes.sort(key=lambda p: key(p, stats_lambda(p)),
reverse=self.sorting_reverse)
if not self.options.batch:
del processes[self.height - 2:]
return list(map(format, processes))
def refresh_display(self, first_time, total, actual, duration):
summary = [
'Total DISK READ : %s | Total DISK WRITE : %s' % (
format_bandwidth(self.options, total[0], duration).rjust(14),
format_bandwidth(self.options, total[1], duration).rjust(14)),
'Actual DISK READ: %s | Actual DISK WRITE: %s' % (
format_bandwidth(self.options, actual[0], duration).rjust(14),
format_bandwidth(self.options, actual[1], duration).rjust(14))
]
pid = max(0, (MAX_PID_WIDTH - 3)) * ' '
if self.options.processes:
pid += 'PID'
else:
pid += 'TID'
titles = [pid, ' PRIO', ' USER', ' DISK READ', ' DISK WRITE',
' SWAPIN', ' IO', ' COMMAND']
self.has_swapin_io = Stats.has_blkio_delay_total
if self.has_swapin_io:
# Linux kernels without the sysctl return None and
# iotop just uses the heuristic for those versions.
# Linux kernels with the sysctl return True or False
# and iotop then uses the sysctl value instead.
if sysctl_task_delayacct() == False:
self.has_swapin_io = False
self.adjust_sorting_key(0)
lines = self.get_data()
if self.options.time:
titles = [' TIME'] + titles
current_time = time.strftime('%H:%M:%S ')
lines = [current_time + l for l in lines]
summary = [current_time + s for s in summary]
if self.options.batch:
if self.options.quiet <= 2:
for s in summary:
print(s)
if self.options.quiet <= int(first_time):
print(''.join(titles))
for l in lines:
print(l.encode('utf-8'))
sys.stdout.flush()
else:
self.win.erase()
for i, s in enumerate(summary):
self.win.addstr(i, 0, s[:self.width])
self.win.hline(len(summary), 0, ord(' ') | curses.A_REVERSE,
self.width)
remaining_cols = self.width
for i in range(len(titles)):
if not self.has_swapin_io and i in (5, 6):
continue
attr = curses.A_REVERSE
title = titles[i]
if i == self.sorting_key:
title = title[1:]
if i == self.sorting_key:
attr |= curses.A_BOLD
title += self.sorting_reverse and '>' or '<'
title = title[:remaining_cols]
remaining_cols -= len(title)
self.win.addstr(title, attr)
if self.has_swapin_io is not False:
status_msg = None
else:
status_msg = ('CONFIG_TASK_DELAY_ACCT '
'and kernel.task_delayacct sysctl '
'not enabled in kernel, '
'cannot determine SWAPIN and IO %')
num_lines = min(len(lines), self.height - 2 - int(bool(status_msg)))
for i in range(num_lines):
try:
def print_line(line):
self.win.addstr(i + len(summary) + 1, 0, line)
try:
print_line(lines[i])
except UnicodeEncodeError:
# Python2: 'ascii' codec can't encode character ...
# http://bugs.debian.org/708252
print_line(lines[i].encode('utf-8'))
except curses.error:
pass
if status_msg:
self.win.insstr(self.height - len(summary), 0, status_msg,
curses.A_BOLD)
self.win.refresh()
def run_iotop_window(win, options):
if options.batch:
signal.signal(signal.SIGPIPE, signal.SIG_DFL)
else:
def clean_exit(*args, **kwargs):
sys.exit(0)
signal.signal(signal.SIGINT, clean_exit)
signal.signal(signal.SIGTERM, clean_exit)
taskstats_connection = TaskStatsNetlink(options)
process_list = ProcessList(taskstats_connection, options)
ui = IOTopUI(win, process_list, options)
ui.run()
def run_iotop(options):
try:
if options.batch:
return run_iotop_window(None, options)
else:
return curses.wrapper(run_iotop_window, options)
except OSError as e:
if e.errno == errno.EPERM:
print(e, file=sys.stderr)
print('''
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.''', file=sys.stderr)
sys.exit(1)
else:
raise
except curses.error as e:
stre = str(e)
if stre.find('ERR')>=0 and (
stre.find('nocbreak()')>=0 or stre.find('endwin()')>=0
):
pass
# endwin and nocbreak can cause error (and raise hard to catch
# exception) if iotop was running in the terminal and that
# terminal got closed while iotop was still running
else:
raise
#
# Profiling
#
def _profile(continuation):
prof_file = 'iotop.prof'
try:
import cProfile
import pstats
print('Profiling using cProfile')
cProfile.runctx('continuation()', globals(), locals(), prof_file)
stats = pstats.Stats(prof_file)
except ImportError:
import hotshot
import hotshot.stats
prof = hotshot.Profile(prof_file, lineevents=1)
print('Profiling using hotshot')
prof.runcall(continuation)
prof.close()
stats = hotshot.stats.load(prof_file)
stats.strip_dirs()
stats.sort_stats('time', 'calls')
stats.print_stats(50)
stats.print_callees(50)
os.remove(prof_file)
#
# Main program
#
USAGE = '''%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.''' % sys.argv[0]
def main():
try:
locale.setlocale(locale.LC_ALL, '')
except locale.Error:
print('unable to set locale, falling back to the default locale')
parser = optparse.OptionParser(usage=USAGE, version='iotop ' + VERSION)
parser.add_option('-o', '--only', action='store_true',
dest='only', default=False,
help='only show processes or threads actually doing I/O')
parser.add_option('-b', '--batch', action='store_true', dest='batch',
help='non-interactive mode')
parser.add_option('-n', '--iter', type='int', dest='iterations',
metavar='NUM',
help='number of iterations before ending [infinite]')
parser.add_option('-d', '--delay', type='float', dest='delay_seconds',
help='delay between iterations [1 second]',
metavar='SEC', default=1)
parser.add_option('-p', '--pid', type='int', dest='pids', action='append',
help='processes/threads to monitor [all]', metavar='PID')
parser.add_option('-u', '--user', type='str', dest='users', action='append',
help='users to monitor [all]', metavar='USER')
parser.add_option('-P', '--processes', action='store_true',
dest='processes', default=False,
help='only show processes, not all threads')
parser.add_option('-a', '--accumulated', action='store_true',
dest='accumulated', default=False,
help='show accumulated I/O instead of bandwidth')
parser.add_option('-k', '--kilobytes', action='store_true',
dest='kilobytes', default=False,
help='use kilobytes instead of a human friendly unit')
parser.add_option('-t', '--time', action='store_true', dest='time',
help='add a timestamp on each line (implies --batch)')
parser.add_option('-q', '--quiet', action='count', dest='quiet', default=0,
help='suppress some lines of header (implies --batch)')
parser.add_option('--profile', action='store_true', dest='profile',
default=False, help=optparse.SUPPRESS_HELP)
options, args = parser.parse_args()
if args:
parser.error('Unexpected arguments: ' + ' '.join(args))
find_uids(options)
options.pids = options.pids or []
options.batch = options.batch or options.time or options.quiet
main_loop = lambda: run_iotop(options)
if options.profile:
def safe_main_loop():
try:
main_loop()
except:
pass
_profile(safe_main_loop)
else:
main_loop()
netlink.py 0000644 00000017323 15035135563 0006576 0 ustar 00 '''
Netlink message generation/parsing
Copyright 2007 Johannes Berg <johannes@sipsolutions.net>
GPLv2+; See copying for details.
'''
import os
import socket
import struct
try:
# try to use python 2.5's netlink support
_dummysock = socket.socket(socket.AF_NETLINK, socket.SOCK_RAW, 0)
_dummysock.bind((0, 0))
del _dummysock
def _nl_bind(descriptor, addr):
descriptor.bind(addr)
def _nl_getsockname(descriptor):
return descriptor.getsockname()
def _nl_send(descriptor, msg):
descriptor.send(msg)
def _nl_recv(descriptor, bufs=16384):
return descriptor.recvfrom(bufs)
except socket.error:
# or fall back to the _netlink C module
try:
import _netlink
def _nl_bind(descriptor, addr):
_netlink.bind(descriptor.fileno(), addr[1])
def _nl_getsockname(descriptor):
return _netlink.getsockname(descriptor.fileno())
def _nl_send(descriptor, msg):
_netlink.send(descriptor.fileno(), msg)
def _nl_recv(descriptor, bufs=16384):
return _netlink.recvfrom(descriptor.fileno(), bufs)
except ImportError:
# or fall back to the ctypes module
import ctypes
libc = ctypes.CDLL(None)
class SOCKADDR_NL(ctypes.Structure):
_fields_ = [("nl_family", ctypes.c_ushort),
("nl_pad", ctypes.c_ushort),
("nl_pid", ctypes.c_int),
("nl_groups", ctypes.c_int)]
def _nl_bind(descriptor, addr):
addr = SOCKADDR_NL(socket.AF_NETLINK, 0, os.getpid(), 0)
return libc.bind(descriptor.fileno(),
ctypes.pointer(addr),
ctypes.sizeof(addr))
def _nl_getsockname(descriptor):
addr = SOCKADDR_NL(0, 0, 0, 0)
len = ctypes.c_int(ctypes.sizeof(addr));
libc.getsockname(descriptor.fileno(),
ctypes.pointer(addr),
ctypes.pointer(len))
return addr.nl_pid, addr.nl_groups;
def _nl_send(descriptor, msg):
return libc.send(descriptor.fileno(), msg, len(msg), 0);
def _nl_recv(descriptor, bufs=16384):
addr = SOCKADDR_NL(0, 0, 0, 0)
len = ctypes.c_int(ctypes.sizeof(addr))
buf = ctypes.create_string_buffer(bufs)
r = libc.recvfrom(descriptor.fileno(),
buf, bufs, 0,
ctypes.pointer(addr), ctypes.pointer(len))
ret = ctypes.string_at(ctypes.pointer(buf), r)
return ret, (addr.nl_pid, addr.nl_groups)
# flags
NLM_F_REQUEST = 1
NLM_F_MULTI = 2
NLM_F_ACK = 4
NLM_F_ECHO = 8
# types
NLMSG_NOOP = 1
NLMSG_ERROR = 2
NLMSG_DONE = 3
NLMSG_OVERRUN = 4
NLMSG_MIN_TYPE = 0x10
class Attr:
def __init__(self, attr_type, data, *values):
self.type = attr_type
if len(values):
self.data = struct.pack(data, *values)
else:
self.data = data
def _dump(self):
hdr = struct.pack("HH", len(self.data)+4, self.type)
length = len(self.data)
pad = ((length + 4 - 1) & ~3 ) - length
return hdr + self.data + b'\0' * pad
def __repr__(self):
return '<Attr type %d, data "%s">' % (self.type, repr(self.data))
def u16(self):
return struct.unpack('H', self.data)[0]
def s16(self):
return struct.unpack('h', self.data)[0]
def u32(self):
return struct.unpack('I', self.data)[0]
def s32(self):
return struct.unpack('i', self.data)[0]
def str(self):
return self.data
def nulstr(self):
return self.data.split('\0')[0]
def nested(self):
return parse_attributes(self.data)
class StrAttr(Attr):
def __init__(self, attr_type, data):
Attr.__init__(self, attr_type, "%ds" % len(data), data.encode('utf-8'))
class NulStrAttr(Attr):
def __init__(self, attr_type, data):
Attr.__init__(self, attr_type, "%dsB" % len(data), data.encode('utf-8'), 0)
class U32Attr(Attr):
def __init__(self, attr_type, val):
Attr.__init__(self, attr_type, "I", val)
class U8Attr(Attr):
def __init__(self, attr_type, val):
Attr.__init__(self, attr_type, "B", val)
class Nested(Attr):
def __init__(self, attr_type, attrs):
self.attrs = attrs
self.type = attr_type
def _dump(self):
contents = []
for attr in self.attrs:
contents.append(attr._dump())
contents = ''.join(contents)
length = len(contents)
hdr = struct.pack("HH", length+4, self.type)
return hdr + contents
NETLINK_ROUTE = 0
NETLINK_UNUSED = 1
NETLINK_USERSOCK = 2
NETLINK_FIREWALL = 3
NETLINK_INET_DIAG = 4
NETLINK_NFLOG = 5
NETLINK_XFRM = 6
NETLINK_SELINUX = 7
NETLINK_ISCSI = 8
NETLINK_AUDIT = 9
NETLINK_FIB_LOOKUP = 10
NETLINK_CONNECTOR = 11
NETLINK_NETFILTER = 12
NETLINK_IP6_FW = 13
NETLINK_DNRTMSG = 14
NETLINK_KOBJECT_UEVENT = 15
NETLINK_GENERIC = 16
class Message:
def __init__(self, msg_type, flags=0, seq=-1, payload=None):
self.type = msg_type
self.flags = flags
self.seq = seq
self.pid = -1
payload = payload or []
if isinstance(payload, list):
contents = []
for attr in payload:
contents.append(attr._dump())
self.payload = b''.join(contents)
else:
self.payload = payload
def send(self, conn):
if self.seq == -1:
self.seq = conn.seq()
self.pid = conn.pid
length = len(self.payload)
hdr = struct.pack("IHHII", length + 4*4, self.type,
self.flags, self.seq, self.pid)
conn.send(hdr + self.payload)
def __repr__(self):
return '<netlink.Message type=%d, pid=%d, seq=%d, flags=0x%x "%s">' % (
self.type, self.pid, self.seq, self.flags, repr(self.payload))
class Connection:
def __init__(self, nltype, groups=0, unexpected_msg_handler=None):
self.descriptor = socket.socket(socket.AF_NETLINK,
socket.SOCK_RAW, nltype)
self.descriptor.setsockopt(socket.SOL_SOCKET, socket.SO_SNDBUF, 65536)
self.descriptor.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, 65536)
_nl_bind(self.descriptor, (0, groups))
self.pid, self.groups = _nl_getsockname(self.descriptor)
self._seq = 0
self.unexpected = unexpected_msg_handler
def send(self, msg):
_nl_send(self.descriptor, msg)
def recv(self):
contents, (nlpid, nlgrps) = _nl_recv(self.descriptor)
# XXX: python doesn't give us message flags, check
# len(contents) vs. msglen for TRUNC
msglen, msg_type, flags, seq, pid = struct.unpack("IHHII",
contents[:16])
msg = Message(msg_type, flags, seq, contents[16:])
msg.pid = pid
if msg.type == NLMSG_ERROR:
errno = -struct.unpack("i", msg.payload[:4])[0]
if errno != 0:
err = OSError("Netlink error: %s (%d)" % (
os.strerror(errno), errno))
err.errno = errno
raise err
return msg
def seq(self):
self._seq += 1
return self._seq
def parse_attributes(data):
attrs = {}
while len(data):
attr_len, attr_type = struct.unpack("HH", data[:4])
attrs[attr_type] = Attr(attr_type, data[4:attr_len])
attr_len = ((attr_len + 4 - 1) & ~3 )
data = data[attr_len:]
return attrs
ioprio.py 0000644 00000012567 15035135563 0006440 0 ustar 00 # This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Library General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
#
# See the COPYING file for license information.
#
# Copyright (c) 2007 Guillaume Chazarain <guichaz@gmail.com>
import ctypes
import fnmatch
import os
import platform
# From http://git.kernel.org/?p=utils/util-linux-ng/util-linux-ng.git;a=blob;
# f=configure.ac;h=770eb45ae85d32757fc3cff1d70a7808a627f9f7;hb=HEAD#l354
# i386 bit userspace under an x86_64 kernel will have its uname() appear as
# 'x86_64' but it will use the i386 syscall number, that's why we consider both
# the architecture name and the word size.
IOPRIO_GET_ARCH_SYSCALL = [
('alpha', '*', 443),
('arm*', '*', 315),
('i*86', '*', 290),
('ia64*', '*', 1275),
('parisc*', '*', 268),
('powerpc*', '*', 274),
('s390*', '*', 283),
('sparc*', '*', 218),
('sh*', '*', 289),
('x86_64*', '32bit', 290),
('x86_64*', '64bit', 252),
]
IOPRIO_SET_ARCH_SYSCALL = [
('alpha', '*', 442),
('arm*', '*', 314),
('i*86', '*', 289),
('ia64*', '*', 1274),
('parisc*', '*', 267),
('powerpc*', '*', 273),
('s390*', '*', 282),
('sparc*', '*', 196),
('sh*', '*', 288),
('x86_64*', '32bit', 289),
('x86_64*', '64bit', 251),
]
def find_ioprio_syscall_number(syscall_list):
arch = os.uname()[4]
bits = platform.architecture()[0]
for candidate_arch, candidate_bits, syscall_nr in syscall_list:
if fnmatch.fnmatch(arch, candidate_arch) and \
fnmatch.fnmatch(bits, candidate_bits):
return syscall_nr
class IoprioSetError(Exception):
def __init__(self, err):
try:
self.err = os.strerror(err)
except TypeError:
self.err = err
__NR_ioprio_get = find_ioprio_syscall_number(IOPRIO_GET_ARCH_SYSCALL)
__NR_ioprio_set = find_ioprio_syscall_number(IOPRIO_SET_ARCH_SYSCALL)
try:
ctypes_handle = ctypes.CDLL(None, use_errno=True)
except TypeError:
ctypes_handle = ctypes.CDLL(None)
syscall = ctypes_handle.syscall
PRIORITY_CLASSES = [None, 'rt', 'be', 'idle']
IOPRIO_WHO_PROCESS = 1
IOPRIO_CLASS_SHIFT = 13
IOPRIO_PRIO_MASK = (1 << IOPRIO_CLASS_SHIFT) - 1
def ioprio_value(ioprio_class, ioprio_data):
try:
ioprio_class = PRIORITY_CLASSES.index(ioprio_class)
except ValueError:
ioprio_class = PRIORITY_CLASSES.index(None)
return (ioprio_class << IOPRIO_CLASS_SHIFT) | ioprio_data
def ioprio_class(ioprio):
return PRIORITY_CLASSES[ioprio >> IOPRIO_CLASS_SHIFT]
def ioprio_data(ioprio):
return ioprio & IOPRIO_PRIO_MASK
sched_getscheduler = ctypes_handle.sched_getscheduler
SCHED_OTHER, SCHED_FIFO, SCHED_RR, SCHED_BATCH, SCHED_ISO, SCHED_IDLE = range(6)
getpriority = ctypes_handle.getpriority
PRIO_PROCESS = 0
def get_ioprio_from_sched(pid):
scheduler = sched_getscheduler(pid)
nice = getpriority(PRIO_PROCESS, pid)
ioprio_nice = (nice + 20) / 5
if scheduler in (SCHED_FIFO, SCHED_RR):
return 'rt/%d' % ioprio_nice
elif scheduler == SCHED_IDLE:
return 'idle'
else:
return 'be/%d' % ioprio_nice
def get(pid):
if __NR_ioprio_get is None:
return '?sys'
ioprio = syscall(__NR_ioprio_get, IOPRIO_WHO_PROCESS, pid)
if ioprio < 0:
return '?err'
prio_class = ioprio_class(ioprio)
if not prio_class:
return get_ioprio_from_sched(pid)
if prio_class == 'idle':
return prio_class
return '%s/%d' % (prio_class, ioprio_data(ioprio))
def set_ioprio(which, who, ioprio_class, ioprio_data):
if __NR_ioprio_set is None:
raise IoprioSetError('No ioprio_set syscall found')
ioprio_val = ioprio_value(ioprio_class, ioprio_data)
ret = syscall(__NR_ioprio_set, which, who, ioprio_val, use_errno=True)
if ret < 0:
try:
err = ctypes.get_errno()
except AttributeError:
err = 'Unknown error (errno support not available before Python2.6)'
raise IoprioSetError(err)
def sort_key(key):
if key[0] == '?':
return -ord(key[1])
if '/' in key:
if key.startswith('rt/'):
shift = 0
elif key.startswith('be/'):
shift = 1
prio = int(key.split('/')[1])
elif key == 'idle':
shift = 2
prio = 0
return (1 << (shift * IOPRIO_CLASS_SHIFT)) + prio
def to_class_and_data(ioprio_str):
if '/' in ioprio_str:
split = ioprio_str.split('/')
return (split[0], int(split[1]))
elif ioprio_str == 'idle':
return ('idle', 0)
return (None, None)
if __name__ == '__main__':
import sys
if len(sys.argv) == 2:
pid = int(sys.argv[1])
else:
pid = os.getpid()
print('pid:', pid)
print('ioprio:', get(pid))
version.py 0000644 00000000020 15035135563 0006601 0 ustar 00 VERSION = '0.6'
vmstat.py 0000644 00000003027 15035135563 0006444 0 ustar 00 # This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Library General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
#
# See the COPYING file for license information.
#
# Copyright (c) 2007 Guillaume Chazarain <guichaz@gmail.com>
class VmStat(object):
def __init__(self):
self.vmstat_file = open('/proc/vmstat')
self.vmstat = self.read()
def read(self):
def extract(line):
return int(line.split()[1]) * 1024
for line in self.vmstat_file:
if line.startswith('pgpgin '):
pgpgin = extract(line)
break
for line in self.vmstat_file:
if line.startswith('pgpgout '):
pgpgout = extract(line)
break
self.vmstat_file.seek(0)
return pgpgin, pgpgout
def delta(self):
now = self.read()
delta = now[0] - self.vmstat[0], now[1] - self.vmstat[1]
self.vmstat = now
return delta