plugin-vmware_gsx_: vmware_gsx_

#!/usr/bin/perl -w
#
# Copyright (C) Rune Nordbøe Skillingstad - rune.skillingstad@ntnu.no
#
# 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; version 2 dated June,
# 1991.
#
# 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 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
#
# $Log$
#
#%# family=auto
#%# capabilities=autoconf suggest

use strict;

my @ret = ();

if(!eval "require VMware::Control::Server;") {
    push(@ret, "VMware::Control::Server not found");
}
if(!eval "require VMware::Control::VM;") {
    push(@ret, "VMware::Control::VM not found");
}

my $host   = $ENV{'host'}   || undef;
my $port   = $ENV{'port'}   || 902;
my $user   = $ENV{'user'}   || undef;
my $passwd = $ENV{'passwd'} || undef;
my $debug  = $ENV{'debug'}  || undef; 

my $server = undef;;
my %result;
my %names;

if($ARGV[0] and $ARGV[0] eq "autoconf") {
    if(@ret) {
        print "no (" . join(", ", @ret) . ")\n";
        exit(1);
    }
    my $ret = &connect(1);
    if($ret) {
        print "no ($ret)\n";
        exit(1);
    }
    print "yes\n";
    exit(0);
}

if($ARGV[0] and $ARGV[0] eq "suggest") {
    print join("\n", qw(disk net memory)) . "\n";
    exit(0);
}

&connect();
&enumerate();

if($debug) {
    use Data::Dumper;
    print Dumper(%result);
}

my $type = $0;
$type =~ s/.*_([^\W\d_]+)$/$1/;

if($ARGV[0] and $ARGV[0] eq "config") {
    if($type eq "net") {
        &config_net();
    } elsif($type eq "disk") {
        &config_disk();
    } elsif($type eq "memory") {
        &config_memory();
    } else {
        exit(2);
    }
}

if($type eq "net") {
    &fetch_net();
} elsif($type eq "disk") {
    &fetch_disk();
} elsif($type eq "memory") {
    &fetch_memory();
} else {
    exit(2);
}


## ==================================================================

&connect();

sub fetch_cpu {
    foreach my $host (sort(keys(%result))) {
        print &rrd_name($host) . ".value " .
            $result{$host}{vm}{cpuusage} . "\n";
    }
}

sub fetch_memory {
    foreach my $host (sort(keys(%result))) {
        if(defined $result{$host}{vm}{ram}) {
            print &rrd_name($host) . "_used.value " .
                $result{$host}{vm}{ram} . "\n";
            print &rrd_name($host) . "_free.value " .
                (($result{$host}{config}{memsize} * 1048576) - $result{$host}{vm}{ram}) . "\n";
        }
    }
}

sub fetch_net {
    foreach my $host (sort(keys(%result))) {
        foreach my $dev (grep(/ethernet\d+/, sort(keys(%{$result{$host}})))) {
            print &rrd_name($host."_".$dev) . "_in.value " . $result{$host}{$dev}{bytesread} . "\n";
            print &rrd_name($host."_".$dev) . "_out.value " . $result{$host}{$dev}{byteswritten} . "\n";
        }
    }
}

sub fetch_disk {
    foreach my $host (sort(keys(%result))) {
        foreach my $dev (grep(/(?:scsi|ide)\d+:\d/, sort(keys(%{$result{$host}})))) {
            print &rrd_name($host."_".$dev) . "_read.value " . $result{$host}{$dev}{bytesread} . "\n";
            print &rrd_name($host."_".$dev) . "_write.value " . $result{$host}{$dev}{byteswritten} . "\n";
        }
    }
}

sub config_cpu {
    print "graph_args --base 1000 -r --lower-limit 0\n"; # --upper-limit 100\n";
    print "graph_title CPU usage pr VM\n";
    print "graph_scale no\n";
    print "graph_category vmware\n";
    print "graph_info This graph shows CPU usage pr VM\n";
    print "graph_order " . join(" ", map{&rrd_name($_)}sort(keys(%result))) . "\n";

    my $first = 1;
    foreach $host (sort(keys(%result))) {
        if($first) {
            print &rrd_name($host) . ".draw AREA\n";
            $first = 0;
        } else {
            print &rrd_name($host) . ".draw STACK\n";
        }
        print &rrd_name($host) . ".min 0\n";
        print &rrd_name($host) . ".max 100\n";
        print &rrd_name($host) . ".label $host cpu\n";
        print &rrd_name($host) . ".type DERIVE\n";
        print &rrd_name($host) . ".info % CPU usage on $host\n";
    }
    print "idle.min 0\n";
    print "idle.max 100\n";
    print "idle.label idle cpu\n";
    print "idle.type DERIVE\n";
    print "idle.draw STACK\n";
    print "idle.info % idle CPU usage \n";
    print "idle.cdef 100," . join(",-,", map{&rrd_name($_);}keys(%result)) . ",-\n";
    exit(0);
}

sub config_memory {
    my $total = 0;
    foreach my $host (keys(%result)) {
        if(defined $result{$host}{vm}{ram}) {
            $total += ($result{$host}{config}{memsize} * 1048576);
        }
    }
    print "graph_args --base 1024 -l 0 --vertical-label Bytes --upper-limit $total\n";
    print "graph_title Memory usage pr VM\n";
    print "graph_category vmware\n";
    print "graph_info This graph shows memory usage and limits pr VM\n";
    print "graph_total Total\n";

    print "graph_order ";
    foreach my $host (sort(keys(%result))) {
        if(defined $result{$host}{vm}{ram}) {
            print &rrd_name($host) . "_used " . &rrd_name($host) . "_free ";
        }
    }
    print "\n";

    my $first = 1;
    foreach $host (sort(keys(%result))) {
        if(defined $result{$host}{vm}{ram}) {
            if($first) {
                print &rrd_name($host) . "_used.draw AREA\n";
                $first = 0;
            } else {
                print &rrd_name($host) . "_used.draw STACK\n";
            }
            print &rrd_name($host) . "_used.min 0\n";
            print &rrd_name($host) . "_used.label $host used\n";
            print &rrd_name($host) . "_used.info Used memory on $host\n";
            print &rrd_name($host) . "_free.draw STACK\n";
            print &rrd_name($host) . "_free.min 0\n";
            print &rrd_name($host) . "_free.label $host free\n";
            print &rrd_name($host) . "_free.info Free memory on $host\n";
        }
    }
    exit(0);
}

sub config_net {
    print "graph_args --base 1000\n";
    print "graph_vlabel read (-) / written (+)\n";
    print "graph_title Traffic usage pr VM\n";
    print "graph_category vmware\n";
    print "graph_info This graph shows the network traffic pr VM\n";
    
    print "graph_order ";
    foreach my $host (sort(keys(%result))) {
        foreach my $dev (grep(/ethernet\d+/, sort(keys(%{$result{$host}})))) {
            print &rrd_name($host."_".$dev) . "_in " . &rrd_name($host."_".$dev) . "_out ";
        }
    }
    print "\n";
    
    foreach $host (sort(keys(%result))) {
        foreach my $dev (grep(/ethernet\d+/, sort(keys(%{$result{$host}})))) {
            print &rrd_name($host."_".$dev) . "_in.label $host $dev\n";
            print &rrd_name($host."_".$dev) . "_in.type DERIVE\n";
            print &rrd_name($host."_".$dev) . "_in.max 900000\n";
            print &rrd_name($host."_".$dev) . "_in.min 0\n";
            print &rrd_name($host."_".$dev) . "_in.graph no\n";
            print &rrd_name($host."_".$dev) . "_out.label $host $dev\n";
            print &rrd_name($host."_".$dev) . "_out.info Traffic on device $dev for host $host\n";
            print &rrd_name($host."_".$dev) . "_out.type DERIVE\n";
            print &rrd_name($host."_".$dev) . "_out.max 900000\n";
            print &rrd_name($host."_".$dev) . "_out.min 0\n";
            print &rrd_name($host."_".$dev) . "_out.negative " . &rrd_name($host."_".$dev) . "_in\n";
        }
    }
    exit(0);
}

sub config_disk {
    print "graph_args --base 1024 -l 0\n";
    print "graph_vlabel read (-) / written (+)\n";
    print "graph_title IOstat usage pr VM\n";
    print "graph_category vmware\n";
    print "graph_info This graph shows the I/O to and from block devices pr VM\n";

    print "graph_order ";
    foreach my $host (sort(keys(%result))) {
        foreach my $dev (grep(/(?:scsi|ide)\d+:\d/, sort(keys(%{$result{$host}})))) {
            print &rrd_name($host."_".$dev) . "_read " . &rrd_name($host."_".$dev) . "_write ";
        }
    }
    print "\n";

    foreach $host (sort(keys(%result))) {
        foreach my $dev (grep(/(?:scsi|ide)\d+:\d/, sort(keys(%{$result{$host}})))) {
            print &rrd_name($host."_".$dev) . "_read.label $host $dev\n";
            print &rrd_name($host."_".$dev) . "_read.type DERIVE\n";
            print &rrd_name($host."_".$dev) . "_read.max 900000\n";
            print &rrd_name($host."_".$dev) . "_read.min 0\n";
            print &rrd_name($host."_".$dev) . "_read.graph no\n";
            print &rrd_name($host."_".$dev) . "_write.label $host $dev\n";
            print &rrd_name($host."_".$dev) . "_write.info I/O on device $dev for host $host\n";
            print &rrd_name($host."_".$dev) . "_write.type DERIVE\n";
            print &rrd_name($host."_".$dev) . "_write.max 900000\n";
            print &rrd_name($host."_".$dev) . "_write.min 0\n";
            print &rrd_name($host."_".$dev) . "_write.negative " . &rrd_name($host."_".$dev) . "_read\n";
        }
    }
    exit(0);
}

sub rrd_name {
    my $name = shift;
    if(!defined $names{$name}) {
        print "rrd_name on $name running\n" if $debug;
        my $new = lc($name);
        $new =~ s/\s/_/g;
        $new =~ s/:/_/g;
        $names{$name} = $new;
    }
    return $names{$name};
}

sub enumerate {
    my @hosts = $server->enumerate();

    if(!@hosts) {
        my ($errno, $errstr) = $server->get_last_error();
        die "Could not get list of VMs from server: Error $errno: $errstr\n";
    }
    
    foreach my $host (@hosts) {
        &check_host($host);
    }
}

sub connect {
    my $test = shift;
    if($server) {
        return $server;
    }
    $server = VMware::Control::Server::new($host, $port, $user, $passwd);
    if(!$server->connect()) {
        my ($errno, $errstr) = $server->get_last_error();
        $server = undef;
        if(!$test) {
            die "Could not connect to server: Error $errno: $errstr\n";
        } else {
            return "$errno: $errstr";
        }
    }
    if($test) {
        my $conf = ($server->enumerate())[0];
        my $vm = VMware::Control::VM::new($server, $conf);
        if(!$vm->connect()) {
            my ($errno, $errstr) = $server->get_last_error();
            return "Could not check version";
        }
        my $vers = $vm->get("Status.version");
        print "Checking \"$vers\" for GSX" if $debug;
        if($vers !~ /GSX/) {
            return "Not GSX";
        }
    }
        
    print "Connected to server\n" if $debug; 
    return 0;
}



sub check_host {
    $host = shift;
    print "Checking $host\n" if $debug;
    my $vm = VMware::Control::VM::new($server, $host);
    if(!$vm->connect()) {
        my ($errno, $errstr) = $server->get_last_error();
        warn "Could not connect to server: Error $errno: $errstr\n";
    }
    
    if($vm->get("Status.power") eq "on") {
        my $name = $vm->get("Config.displayName");
        print "Host $name is on\n" if $debug;
        my $mem  = $vm->get("Config.memsize");
        $result{$name}{"config"}{"memsize"} = $mem;
        my $devs = ($vm->get("Status.stats.devices"));
        print "Found devices: " . join(" ", @{$devs}) . "\n" if $debug;
        foreach my $dev (@{$devs}) {
            my $stats = $vm->get("Status.stats.$dev.stats");
            if(!$stats) {
                next;
            }
            print "Found status for $dev: " . join(" ", @{$stats}) . "\n" if $debug;
            for my $stat (@{$stats}) {
                my $val = $vm->get("Status.stats.$dev.$stat");
                $result{$name}{$dev}{$stat} = $val;
                print "Got value for $stat on $dev: $val\n" if $debug;
            }
        }
    }
    $vm->disconnect();
}

undef $server;