#!/bin/sh
# Author: mario manno <projects@manno.name>
# Description: measure traffic for all xen hosts
#
# Changelog:
# Also handle domUs with named network interfaces.
# Ward Vandewege (ward@gnu.org), 2011-04-20
#
#%# family=auto
#%# capabilities=autoconf

if [ "$1" = "autoconf" ]; then
    if which xm > /dev/null ; then
        if [ -r /proc/net/dev ]; then
            echo yes
        else
            echo "no (/proc/net/dev not found)"
        fi
    else
        echo "no (xm not found)"
    fi
    exit 0
fi

if [ "$1" = "config" ]; then
    echo 'graph_title Xen Traffic'
    # shellcheck disable=SC2016
    echo 'graph_vlabel bits received (-) / sent (+) per ${graph_period}'
    echo 'graph_args --base 1024 -l 0'
    echo 'graph_category Virtualization'
    DOMAINS=$(xm list | awk '{print $1}' | grep -v -E "^(Name|Domain-0)")
    for dom in $DOMAINS; do
        name=$( echo "$dom" | tr .- __ )
        #echo $name"UP.label $name"
        #echo $name"DOWN.label $name"

        echo "${name}Down.label received"
        echo "${name}Down.type COUNTER"
        echo "${name}Down.graph no"
        echo "${name}Down.cdef ${name}Down,8,*"
        echo "${name}Up.label ${name}"
        echo "${name}Up.type COUNTER"
        echo "${name}Up.negative ${name}Down"
        echo "${name}Up.cdef ${name}Up,8,*"
    done
    exit 0
fi

DOMAINS=$(xm list | awk '{print $1}' | grep -v -E "^(Name|Domain-0)")
for dom in $DOMAINS; do
    dev=$( xm list "$dom" --long | awk '/vifname / { print $2 }' | sed 's/)//' )
    if [ -z "$dev" ]; then
        dev=$( xm network-list "$dom" | grep '^[0-9]\+' | sed 's@^.*vif/\([0-9]*\)/\([0-9]*\).*$@vif\1.\2@')
    fi
    name=$( echo "$dom" | tr .- __ )
    #awk -v name="$name" -v interface="$dev" -F'[: \t]+' \
    #'{ sub(/^ */,""); if ($1 == interface) \
    #print name"DOWN.value "$2"\n"name"UP.value "$10; }' /proc/net/dev
    awk -v name="$name" -v interface="$dev" \
        'BEGIN { gsub(/\./, "\\.", interface) }
        $1 ~ "^" interface ":" {
            split($0, a, /: */); $0 = a[2];
            print name"Down.value " $1 "\n"name"Up.value " $9;
        }' \
        /proc/net/dev
done