More housekeeping.

plugins/virtualization/kvm_net

@@ -0,0 +1,141 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+# vim: set fileencoding=utf-8
+#
+# Munin plugin to show the network I/O per vm
+#
+# Copyright Igor Borodikhin
+#
+# License : GPLv3
+#
+#
+# parsed environment variables:
+# vmsuffix: part of vm name to be removed
+#
+#%# capabilities=autoconf
+#%# family=contrib
+
+import re, os, sys
+from subprocess import Popen, PIPE
+
+def config(vm_names):
+    ''' Print the plugin's config
+    @param vm_names : a list of "cleaned" vms' name
+    '''
+    base_config = """graph_title KVM Network I/O
+graph_vlabel Bytes rx(-)/tx(+) per second
+graph_category KVM
+graph_info This graph shows the network I/O of the virtual machines
+graph_args --base 1024
+    """
+    print base_config
+    for vm in vm_names:
+        print "%s_in.label %s" % (vm, vm)
+        print "%s_in.type COUNTER" % vm
+        print "%s_in.min 0" % vm
+        print "%s_in.draw LINE2" % vm
+        print "%s_out.negative %s_in" % (vm, vm)
+        print "%s_out.label %s" % (vm, vm)
+        print "%s_out.type COUNTER" % vm
+        print "%s_out.min 0" % vm
+        print "%s_out.draw LINE2" % vm
+
+def clean_vm_name(vm_name):
+    ''' Replace all special chars
+    @param vm_name : a vm's name
+    @return cleaned vm's name
+    '''
+    # suffix part defined in conf
+    suffix = os.getenv('vmsuffix')
+    if suffix:
+        vm_name = re.sub(suffix,'',vm_name)
+
+    return re.sub(r"[^a-zA-Z0-9_]", "_", vm_name)
+    
+def fetch(vms):
+    ''' Fetch values for a list of pids
+    @param dictionnary {kvm_pid: cleaned vm name}
+    '''
+    res = {}
+    for pid in vms:
+        tap = get_vm_mac(pid)
+        try:
+            f = open("/proc/net/dev", "r")
+            for line in f.readlines():
+                if tap in line:
+                    print "%s_in.value %s" % (vms[pid], re.sub(r"%s:"%tap, "", line.split()[0]))
+                    print "%s_out.value %s" % (vms[pid], line.split()[8])
+                    break
+        except Exception as inst:
+            print inst
+            continue
+
+def detect_kvm():
+    ''' Check if kvm is installed
+    '''
+    kvm = Popen("which kvm", shell=True, stdout=PIPE)
+    kvm.communicate()
+    return not bool(kvm.returncode)
+
+def find_vm_names(pids):
+    '''Find and clean vm names from pids
+    @return a dictionnary of {pids : cleaned vm name}
+    '''
+    result = {}
+    for pid in pids:
+        cmdline = open("/proc/%s/cmdline" % pid, "r")
+        result[pid] = clean_vm_name(re.sub(r"^.*-name\x00([a-zA-Z0-9.-]*)\x00\-.*$",r"\1", cmdline.readline()))        
+    return result
+    
+def get_vm_mac(pid):
+    '''Find and clean vm names from pids
+    @return the mac address for a specified pid
+    '''
+    cmdline = open("/proc/%s/cmdline" % pid, "r")
+    line = cmdline.readline()
+    mac = re.sub(r"^.*ifname=(tap[^,]+),.*$",r"\1", line)
+    return mac
+
+def list_pids():
+    ''' Find the pid of kvm processes
+    @return a list of pids from running kvm
+    '''
+    pid = Popen("pidof kvm", shell=True, stdout=PIPE)
+    return pid.communicate()[0].split()
+
+def find_vms_tap():
+    ''' Check if kvm is installed
+    @return a list of pids from running kvm
+    '''
+    result = []
+    tap = ""
+    mac = ""
+    kvm = Popen("ip a | grep -A 1 tap | awk '{print $2}' | grep -v '^$'", shell=True, stdout=PIPE)
+    res = kvm.communicate()[0].split('\n')
+    for line in res:
+        try:
+            if len(line) > 0:
+                if re.match(r"^tap.*", line):
+                    tap = re.sub(r"(tap[^:]+):", r"\1", line)
+                else:
+                    result.append(tap)
+        except Exception as inst:
+            continue
+
+    return result
+    
+if __name__ == "__main__":
+    if len(sys.argv) > 1:
+        if sys.argv[1] in ['autoconf', 'detect']:
+            if detect_kvm():
+                print "yes"
+            else:
+                print "no"
+        elif sys.argv[1] == "config":
+            config(find_vm_names(list_pids()).values())
+        else:
+            fetch(find_vm_names(list_pids()))
+    else:
+        fetch(find_vm_names(list_pids()))
+
+