Munin contributed plugins

#!/usr/bin/perl

#

# 2007-06-26

# Written by Ghost

#

# 2008-04-16

# Update: Wildcard version

#

# 2008-11-12

# Update: Perl RCON system

#

# Configuration variables

#

#   srcdspass       - RCON password

#   srcdssamples    - Number of samples to take (optional, default: 5)

#

# Magic markers - optional - used by installation scripts and

# munin-config:

#

#%# family=contrib

#%# capabilities=autoconf

use strict;

# Set library path correctly

use File::Basename;

if (-l $0) {

    push(@INC, dirname(readlink($0)));

}

push(@INC, dirname($0));

# Load Rcon module or exit with failure message

if (!eval "require Rcon") {

    print "Failed to load Rcon module. ";

    print "Make sure Rcon.pm is copied to Munin plugins directory.\n";

    exit 1;

}

# Parse hostname and port from the plugin filename

my ($HOST, $PORT) = $0 =~ m/.*_([^:]+)_(\d+)$/;

if (!defined($HOST) || !defined($PORT)) {

    print "Could not parse server address from filename.\n";

    exit 1;

}

# Load config variables or use default values

my $PASS    = $ENV{srcdspass}       || "";

my $SAMPLES = $ENV{srcdssamples}    || 5;

# Print config or do plugin test if asked

my $arg = shift();

if ($arg eq 'config') {

    print_config();

} elsif ($arg eq 'autoconf') {

    test_service();

}

#

# Main program starts here

#

my $sock = Rcon::sock_connect($HOST, $PORT);

if (!$sock) {

    print "Could not open socket to $HOST:$PORT.\n";

    exit 1;

}

if (!Rcon::rcon_auth($sock, $PASS)) {

    print "Could not authenticate.\n";

    exit 1;

}

my $in_avg = 0;

my @in_samples;

my $out_avg = 0;

my @out_samples;

for (my $i = 0; $i < $SAMPLES; $i++) {

    my $reply = Rcon::rcon_command($sock, "stats");

    if (!defined($reply)) {

	print "Did not receive reply from server (sample $i).\n";

	next;

    }

    my @reply = split(/\n/, $reply);

    foreach my $statline (@reply) {

	next if ($statline !~ m/\s*[\w+\.]+\s+[\w+\.]+\s+[\w+\.]+\s+\d+\s+\d+\s+[\w+\.]+\s+\d+/);

	my ($cpu, $in, $out, $uptime, $users, $fps, $players) = ($statline =~ m/^\s*([\w+\.]+)\s+([\w+\.]+)\s+([\w+\.]+)\s+(\d+)\s+(\d+)\s+([\w+\.]+)\s+(\d+)/);

	if (defined($in) && defined($out)) {

	    $in = int($in);

	    $out = int($out);

	    push(@in_samples, $in);

	    push(@out_samples, $out);

	}

    }

    select(undef, undef, undef, 0.2); # Wait moment before next sample

}

# MEAN

if (@in_samples && @out_samples) {

    foreach (@in_samples) {

        $in_avg += int($_);

    }

    $in_avg /= ($#in_samples+1);

    $in_avg = int($in_avg);

    foreach (@out_samples) {

        $out_avg += int($_);

    }

    $out_avg /= ($#out_samples+1);

    $out_avg = int($out_avg);

    print "in.value $in_avg\n";

    print "out.value $out_avg\n";

}

sub print_config {

    print("graph_title Server traffic at $HOST:$PORT\n",

	  "graph_args --base 1000\n",

	  "graph_vlabel bits in (-) / out (+) per second\n",

	  "graph_category games\n",

	  "graph_info The traffic of incoming and outgoing data from Source game server, such as HL2, CS:S and DoD:S.\n");

    print (

	   "in.type GAUGE\n",

	   "out.type GAUGE\n",

	   "in.min 0\n",

	   "out.min 0\n",

	   "in.label bps\n",

	   "out.label bps\n",

	   "in.graph no\n",

	   "in.cdef in,8,*\n",

	   "out.cdef out,8,*\n",

	   "out.negative in\n"

	   );

}

sub test_service {

    my $sock = Rcon::sock_connect($HOST, $PORT);

    if (!$sock) {

	print "no (could not open socket to $HOST:$PORT)\n";

	exit 1;

    }

    if (!Rcon::rcon_auth($sock, $PASS)) {

	print "no (could not authenticate)\n";

	exit 1;

    }

    if (!defined(Rcon::rcon_command($sock, "stats"))) {

	print "no (did not receive reply from server)\n";

	exit 1;

    }

    print "yes\n";

    exit 0;

}