tools/jailhouse-cell-stats

   1 #!/usr/bin/python
   2
   3 # Jailhouse, a Linux-based partitioning hypervisor
   4 #
   5 # Copyright (c) Siemens AG, 2014
   6 #
   7 # Authors:
   8 #  Jan Kiszka <jan.kiszka@siemens.com>
   9 #
  10 # This work is licensed under the terms of the GNU GPL, version 2.  See
  11 # the COPYING file in the top-level directory.
  12
  13 from __future__ import print_function
  14 import curses
  15 import datetime
  16 import os
  17 import sys
  18
  19 stats_dir = "/sys/devices/jailhouse/cells/%s/statistics"
  20
  21
  22 def main(stdscr, cell, stats_names):
  23     curses.use_default_colors()
  24     curses.noecho()
  25     curses.curs_set(0)
  26     curses.halfdelay(10)
  27     value = dict.fromkeys(stats_names)
  28     old_value = dict.fromkeys(stats_names, None)
  29     while True:
  30         now = datetime.datetime.now()
  31
  32         for name in stats_names:
  33             f = open((stats_dir + "/%s") % (cell, name), "r")
  34             value[name] = int(f.read())
  35
  36         def sortkey(name):
  37             if old_value[name] is None:
  38                 return (-value[name], name)
  39             else:
  40                 return (old_value[name] - value[name], -value[name], name)
  41
  42         stdscr.erase()
  43         stdscr.addstr(0, 0, "Statistics for %s cell" % cell)
  44         (height, width) = stdscr.getmaxyx()
  45         stdscr.hline(2, 0, " ", width, curses.A_REVERSE)
  46         stdscr.addstr(2, 0, "COUNTER", curses.A_REVERSE)
  47         stdscr.addstr(2, 30, "%10s" % "SUM", curses.A_REVERSE)
  48         stdscr.addstr(2, 40, "%10s" % "PER SEC", curses.A_REVERSE)
  49         line = 3
  50         for name in sorted(stats_names, key=sortkey):
  51             stdscr.addstr(line, 0, name)
  52             stdscr.addstr(line, 30, "%10u" % value[name])
  53             if not old_value[name] is None:
  54                 dt = (now - last_refresh).total_seconds()
  55                 delta_per_sec = (value[name] - old_value[name]) / dt
  56                 stdscr.addstr(line, 40, "%10u" % round(delta_per_sec))
  57             old_value[name] = value[name]
  58             line += 1
  59         stdscr.hline(height - 1, 0, " ", width, curses.A_REVERSE)
  60         stdscr.addstr(height - 1, 1, "Q - Quit", curses.A_REVERSE)
  61         stdscr.refresh()
  62
  63         last_refresh = now
  64
  65         try:
  66             if stdscr.getch() == ord('q'):
  67                 break
  68             curses.halfdelay(40)
  69         except KeyboardInterrupt:
  70             break
  71         except curses.error:
  72             continue
  73
  74
  75 def usage(exit_code):
  76     prog = os.path.basename(sys.argv[0]).replace('-', ' ')
  77     print("usage: %s { ID | [--name] NAME }" % prog)
  78     exit(exit_code)
  79
  80
  81 argc = len(sys.argv)
  82 use_name = argc >= 2 and sys.argv[1] == "--name"
  83
  84 if argc < 2 or argc > 3 or (not use_name and argc > 2):
  85     usage(1)
  86 if sys.argv[1] in ("--help", "-h"):
  87     usage(0)
  88
  89 try:
  90     if use_name:
  91         cell_name = sys.argv[2]
  92     else:
  93         cell_name = sys.argv[1]
  94         try:
  95             cell_id = int(sys.argv[1])
  96             for cell in os.listdir("/sys/devices/jailhouse/cells"):
  97                 f = open("/sys/devices/jailhouse/cells/%s/id" % cell, "r")
  98                 if int(f.read()) == cell_id:
  99                     cell_name = cell
 100                     break
 101         except ValueError:
 102             pass
 103
 104     stats_names = os.listdir(stats_dir % cell_name)
 105 except OSError as e:
 106     print("reading stats: %s" % e.strerror, file=sys.stderr)
 107     exit(1)
 108
 109 curses.wrapper(main, cell_name, stats_names)