1#!/usr/local/bin/perl 2# 3# 2007-06-26 4# Written by Ghost 5# 6# 2008-04-16 7# Update: Wildcard version 8# 9# 2008-11-12 10# Update: Perl RCON system 11# 12# Configuration variables 13# 14# srcdspass - RCON password 15# srcdssamples - Number of samples to take (optional, default: 5) 16# 17# Magic markers - optional - used by installation scripts and 18# munin-config: 19# 20#%# family=contrib 21#%# capabilities=autoconf 22 23use strict; 24 25# Set library path correctly 26use File::Basename; 27if (-l $0) { 28 push(@INC, dirname(readlink($0))); 29} 30push(@INC, dirname($0)); 31 32# Load Rcon module or exit with failure message 33if (!eval "require Rcon") { 34 print "Failed to load Rcon module. "; 35 print "Make sure Rcon.pm is copied to Munin plugins directory.\n"; 36 exit 1; 37} 38 39# Parse hostname and port from the plugin filename 40my ($HOST, $PORT) = $0 =~ m/.*_([^:]+)_(\d+)$/; 41if (!defined($HOST) || !defined($PORT)) { 42 print "Could not parse server address from filename.\n"; 43 exit 1; 44} 45 46# Load config variables or use default values 47my $PASS = $ENV{srcdspass} || ""; 48my $SAMPLES = $ENV{srcdssamples} || 5; 49 50# Print config or do plugin test if asked 51my $arg = shift(); 52if ($arg eq 'config') { 53 print_config(); 54} elsif ($arg eq 'autoconf') { 55 test_service(); 56} 57 58 59# 60# Main program starts here 61# 62 63my $sock = Rcon::sock_connect($HOST, $PORT); 64if (!$sock) { 65 print "Could not open socket to $HOST:$PORT.\n"; 66 exit 1; 67} 68if (!Rcon::rcon_auth($sock, $PASS)) { 69 print "Could not authenticate.\n"; 70 exit 1; 71} 72 73my $in_avg = 0; 74my @in_samples; 75my $out_avg = 0; 76my @out_samples; 77 78for (my $i = 0; $i < $SAMPLES; $i++) { 79 my $reply = Rcon::rcon_command($sock, "stats"); 80 if (!defined($reply)) { 81 print "Did not receive reply from server (sample $i).\n"; 82 next; 83 } 84 my @reply = split(/\n/, $reply); 85 86 foreach my $statline (@reply) { 87 next if ($statline !~ m/\s*[\w+\.]+\s+[\w+\.]+\s+[\w+\.]+\s+\d+\s+\d+\s+[\w+\.]+\s+\d+/); 88 my ($cpu, $in, $out, $uptime, $users, $fps, $players) = ($statline =~ m/^\s*([\w+\.]+)\s+([\w+\.]+)\s+([\w+\.]+)\s+(\d+)\s+(\d+)\s+([\w+\.]+)\s+(\d+)/); 89 90 if (defined($in) && defined($out)) { 91 $in = int($in); 92 $out = int($out); 93 push(@in_samples, $in); 94 push(@out_samples, $out); 95 } 96 } 97 select(undef, undef, undef, 0.2); # Wait moment before next sample 98} 99 100# MEAN 101if (@in_samples && @out_samples) { 102 foreach (@in_samples) { 103 $in_avg += int($_); 104 } 105 $in_avg /= ($#in_samples+1); 106 $in_avg = int($in_avg); 107 108 foreach (@out_samples) { 109 $out_avg += int($_); 110 } 111 $out_avg /= ($#out_samples+1); 112 $out_avg = int($out_avg); 113 114 print "in.value $in_avg\n"; 115 print "out.value $out_avg\n"; 116} 117 118 119 120sub print_config { 121 print("graph_title Server traffic at $HOST:$PORT\n", 122 "graph_args --base 1000\n", 123 "graph_vlabel bits in (-) / out (+) per second\n", 124 "graph_category games\n", 125 "graph_info The traffic of incoming and outgoing data from Source game server, such as HL2, CS:S and DoD:S.\n"); 126 127 print ( 128 "in.type GAUGE\n", 129 "out.type GAUGE\n", 130 "in.min 0\n", 131 "out.min 0\n", 132 "in.label bps\n", 133 "out.label bps\n", 134 "in.graph no\n", 135 "in.cdef in,8,*\n", 136 "out.cdef out,8,*\n", 137 "out.negative in\n" 138 ); 139} 140 141 142sub test_service { 143 my $sock = Rcon::sock_connect($HOST, $PORT); 144 if (!$sock) { 145 print "no (could not open socket to $HOST:$PORT)\n"; 146 exit 0; 147 } 148 if (!Rcon::rcon_auth($sock, $PASS)) { 149 print "no (could not authenticate)\n"; 150 exit 0; 151 } 152 if (!defined(Rcon::rcon_command($sock, "stats"))) { 153 print "no (did not receive reply from server)\n"; 154 exit 0; 155 } 156 157 print "yes\n"; 158 exit 0; 159} 160