bwmon/__init__.py

   1 # -*- coding: utf-8 -*-
   2
   3 from __future__ import absolute_import
   4
   5 from bwmon import proc
   6
   7 import collections
   8 import time
   9 import sys
  10 import curses
  11 import copy
  12
  13 def clear():
  14     curses.setupterm()
  15     sys.stdout.write(curses.tigetstr("clear"))
  16     sys.stdout.flush()
  17
  18 class Monitor(object):
  19     def __init__(self):
  20         self.fd_map = {}
  21         self.conntrack = {}
  22         self.last_conntrack = {}
  23         self.connections = {}
  24         self.tracking = {}
  25
  26     def update(self):
  27         self.fd_map.update(proc.get_fd_map())
  28         self.last_conntrack = copy.deepcopy(self.conntrack)
  29         self.conntrack.update(proc.parse_ip_conntrack())
  30         self.connections.update(proc.get_connections())
  31
  32     def convert(self):
  33         for con in self.connections.itervalues():
  34             inode = con.get('inode', None)
  35             process = self.fd_map.get(inode, None)
  36
  37             if process is None:
  38                 continue
  39
  40             key_in  = proc.ip_hash(con['remote'], con['local'])
  41             key_out = proc.ip_hash(con['local'], con['remote'])
  42             keys = {'in': key_in, 'out': key_out}
  43             new_byte = {'in': 0, 'out': 0}
  44
  45             for direction in ('in', 'out'):
  46                 k = keys[direction]
  47                 if k in self.conntrack:
  48                     if key_in in self.last_conntrack:
  49                         new_byte[direction] = int(self.conntrack[k]['bytes']) - int(self.last_conntrack[k]['bytes'])
  50                     else:
  51                         new_byte[direction] = int(self.conntrack[k]['bytes'])
  52
  53             if process['cmd'] in self.tracking:
  54                 old_in, old_out = self.tracking[process['cmd']]
  55             else:
  56                 old_in = 0
  57                 old_out = 0
  58
  59             self.tracking[process['cmd']] = (old_in + new_byte['in'], old_out + new_byte['out'])
  60
  61     def output(self):
  62         def compare(a, b):
  63             return cmp(a[1], b[1])
  64
  65         clear()
  66         for cmd, bytes in sorted(self.tracking.iteritems(), cmp=compare):
  67             if len(cmd) > 60:
  68                 cmd = cmd[:57] + '...'
  69             print '%10d / %10d -- %s' % (bytes[0], bytes[1], cmd)
  70             sys.stdout.flush()
  71
  72     def loop(self):
  73         while True:
  74             self.update()
  75             self.convert()
  76             self.output()
  77             time.sleep(1)
  78
  79 #def loop():
  80 #    """
  81 #    prints a list of processes
  82 #    """
  83 #
  84 #    while True:
  85 #        for con in connections.itervalues():
  86 #
  87 #            if not con['inode']:
  88 #                continue
  89 #
  90 #            if not con['inode'] in fd_map:
  91 #                print 'inode %s not im map' % con['inode']
  92 #                continue
  93 #
  94 #            process = fd_map[con['inode']]
  95 #
  96 #
  97 #            key1 = proc.ip_hash(con['local'], con['remote'])
  98 #            key2 = proc.ip_hash(con['remote'], con['local'])
  99 #            key = None
 100 #
 101 #            if key1 in conntrack:
 102 #                key = key1
 103 #            elif key2 in conntrack:
 104 #                key = key2
 105 #
 106 #            if key and 'bytes' in conntrack[key]:
 107 #                byte_count = conntrack[key]['bytes']
 108 #            else:
 109 #                byte_count = '?'
 110 #
 111 #            print '%d: %s: %s -> %s (%s Bytes)' % (process['pid'], process['cmd'], con['local'], con['remote'], byte_count)
 112 #        time.sleep(1)
 113 #
 114 #
 115