Munin contributed plugins

#!/usr/bin/perl -w

#

# Plugin for monitoring boinc processes

#

# Parameters:

#

#	password	-- The password for RPC authentication

#			   (default: boinc_cmd will look for a file 

#			   'gui_rpc_auth.cfg' and use the password in it)

#       host 		-- the host to connect to (default: localhost)

#	port 		-- optional (default: 31416)

#

# This plugin can monitor boinc processes running on local/remote machines. 

# You can see the progress on various projects.

#

# Author: Petr Ruzicka <petr.ruzicka@gmail.com>

# GPL v3

#

#%# family=auto

#%# capabilities=autoconf

use IO::Socket;

use Digest::MD5 qw(md5_hex);

if ($ENV{'password_path'}) {

  $password = $ENV{'password_path'};

} else {

    $password = $ENV{'password'} || `cat /var/lib/boinc/gui_rpc_auth.cfg 2>/dev/null`;

  }

my $host = $ENV{'host'} || '127.0.0.1';

my $port = $ENV{'port'} || '31416';

sub autoconf {

  my $client = new IO::Socket::INET (

  PeerAddr => $host,

  PeerPort => $port,

  Proto => 'tcp' );

  if ($client) {

    print $client "<boinc_gui_rpc_request><auth1/></boinc_gui_rpc_request>\003";

    {

      local $/ = "\003";

      $reply = <$client>;

    }

    $reply =~ /<nonce>(.*)<\/nonce>/;

    $hash = md5_hex($1, $password);

    print $client "<boinc_gui_rpc_request><auth2><nonce_hash>$hash</nonce_hash></auth2></boinc_gui_rpc_request>\003";

    {

      local $/ = "\003";

      $reply = <$client>;

    }

    if ($reply =~ /<authorized\/>/) {

      print "yes\n";

      exit 0;

    }

  }

  print "no\n";

  exit 1;

}

sub config {

  my $client = IO::Socket::INET->new ( PeerAddr => $host,

    				       PeerPort => $port,

				       Proto => 'tcp' )

  or die "Can't bind : $@\n"; 

  print $client "<boinc_gui_rpc_request><auth1/></boinc_gui_rpc_request>\003";

  {	

    local $/ = "\003";

    $reply = <$client>;

  }

  $reply =~ /<nonce>(.*)<\/nonce>/;

  my $hash = md5_hex($1, $password);

  print $client "<boinc_gui_rpc_request><auth2><nonce_hash>$hash</nonce_hash></auth2></boinc_gui_rpc_request>\003";

  {

    local $/ = "\003";

    $reply = <$client>;

  }

  if ($reply !~ /<authorized\/>/) {

    die "Wrong password: $_";

  }

  print $client "<boinc_gui_rpc_request><get_state></boinc_gui_rpc_request>";

  while (chomp($reply = <$client>) && ($reply ne "</boinc_gui_rpc_reply>")) {

    if ($reply =~ /<domain_name>(.*)<\/domain_name>/) {

      print "graph_title BOINC task progress [$1]\n";

      print "graph_args -l 0\n";

      print "graph_vlabel %\n";

    }  

    if ($reply =~ /<project_name>(.*)<\/project_name>/) {

      my $boinc_munin_name=$1;

      $boinc_munin_name =~ /(\w+).*/;

      print "$1.label $boinc_munin_name\n";

    }

  }

  close ($client);

}

sub report {

  my $client = IO::Socket::INET->new ( PeerAddr => $host,

    				       PeerPort => $port,

				       Proto => 'tcp' )

  or die "Can't bind : $@\n"; 

  print $client "<boinc_gui_rpc_request><auth1/></boinc_gui_rpc_request>\003";

  {	

    local $/ = "\003";

    $reply = <$client>;

  }

  $reply =~ /<nonce>(.*)<\/nonce>/;

  my $hash = md5_hex($1, $password);

  print $client "<boinc_gui_rpc_request><auth2><nonce_hash>$hash</nonce_hash></auth2></boinc_gui_rpc_request>\003";

  {

    local $/ = "\003";

    $reply = <$client>;

  }

  if ($reply !~ /<authorized\/>/) {

    die "Wrong password: $_";

  }

  print $client "<boinc_gui_rpc_request><get_state></boinc_gui_rpc_request>";

  while (chomp($reply = <$client>) && ($reply ne "</boinc_gui_rpc_reply>")) {

    if ($reply =~ /<project_name>(\w+).*<\/project_name>/) {

      $project = $1;

      $fraction_done=0;

      while (chomp($reply = <$client>) && ($reply ne "<project>") && ($reply ne "</client_state>")) {

        if ($reply =~ /\s*<active_task_state>1<\/active_task_state>/) {

	  while (chomp($reply = <$client>) && ($reply ne "</active_task>")) {

            if ($reply =~ /<fraction_done>(.*)<\/fraction_done>/) {

	      $fraction_done+=int($1*100 + .5 * ($1*100 <=> 0));

	    }

	  }

	}

      }

      print "$project.value $fraction_done\n";

    }

  }

  close ($client);

}

if (defined $ARGV[0]) {

  my $arg = $ARGV[0];

  my %funcs = ( config => \&config,

		autoconf => \&autoconf,

		report => \&report

	      );

  if (exists $funcs{$arg}) {

    $funcs{$arg}->();

  }

  else {

    $funcs{"report"}->();

  }

} else {

    report();

  }