X-Git-Url: http://git.zerfleddert.de/cgi-bin/gitweb.cgi/upsgraph/blobdiff_plain/e18c93b30560e68d91bd34f298dc9d8903929b97..e5da3876baf867c77cf76593d26d88c76c6c49ae:/upsgraph.pl diff --git a/upsgraph.pl b/upsgraph.pl index 745a44c..19d8147 100755 --- a/upsgraph.pl +++ b/upsgraph.pl @@ -1,115 +1,139 @@ #!/usr/bin/perl -w -if (@ARGV != 3) { - print STDERR "Syntax: ${0} host /path/to/file.rrd /output/directory/\n"; +if ((@ARGV != 1) && (@ARGV != 2)) { + print STDERR "Syntax: ${0} configfile [uid]\n"; exit(1); } -my $host=$ARGV[0]; -my $rrdfile=$ARGV[1]; -my $outdir=$ARGV[2]; -my $community="public"; -my $step=60; - -my @fields = ( 'inputV', 'outputV', 'inputHZ', 'outputHZ', 'battT', 'battC', 'load', 'ambT', 'ambH', 'timeR', 'timeO' ); - -my $vars = { - 'inputV' => { - 'name' => 'Input Voltage', - 'oidtext' => 'PowerNet-MIB::upsAdvInputLineVoltage.0', - 'oid' => '1.3.6.1.4.1.318.1.1.1.3.2.1.0', - 'min' => '180', - 'max' => '280', - }, - 'outputV' => { - 'name' => 'Output Voltage', - 'oidtext' => 'PowerNet-MIB::upsAdvOutputVoltage.0', - 'oid' => '1.3.6.1.4.1.318.1.1.1.4.2.1.0', - 'min' => '180', - 'max' => '280', - }, - 'inputHZ' => { - 'name' => 'Input Frequency', - 'oidtext' => 'PowerNet-MIB::upsAdvInputFrequency.0', - 'oid' => '1.3.6.1.4.1.318.1.1.1.3.2.4.0', - 'min' => '40', - 'max' => '60', - }, - 'outputHZ' => { - 'name' => 'Output Frequency', - 'oidtext' => 'PowerNet-MIB::upsAdvOutputFrequency.0', - 'oid' => '1.3.6.1.4.1.318.1.1.1.4.2.2.0', - 'min' => '40', - 'max' => '60', - }, - 'battT' => { - 'name' => 'Battery Temperature', - 'oidtext' => 'PowerNet-MIB::upsAdvBatteryTemperature.0', - 'oid' => '1.3.6.1.4.1.318.1.1.1.2.2.2.0', - 'min' => '0', - 'max' => '100', - }, - 'battC' => { - 'name' => 'Battery Capacity', - 'oidtext' => 'PowerNet-MIB::upsAdvBatteryCapacity.0', - 'oid' => '1.3.6.1.4.1.318.1.1.1.2.2.1.0', - 'min' => '0', - 'max' => '110', - }, - 'load' => { - 'name' => 'UPS Load', - 'oidtext' => 'PowerNet-MIB::upsAdvOutputLoad.0', - 'oid' => '1.3.6.1.4.1.318.1.1.1.4.2.3.0', - 'min' => '0', - 'max' => '110', - }, - 'ambT' => { - 'name' => 'Ambient Temperature', - 'oidtext' => 'PowerNet-MIB::mUpsEnvironAmbientTemperature.0', - 'oid' => '1.3.6.1.4.1.318.1.1.2.1.1.0', - 'min' => '0', - 'max' => '60', - }, - 'ambH' => { - 'name' => 'Ambient Humidity', - 'oidtext' => 'PowerNet-MIB::mUpsEnvironRelativeHumidity.0', - 'oid' => '1.3.6.1.4.1.318.1.1.2.1.2.0', - 'min' => '0', - 'max' => '100', - }, - 'timeR' => { - 'name' => 'Time Remaining', - 'oidtext' => 'PowerNet-MIB::upsAdvBatteryRunTimeRemaining.0', - 'oid' => '1.3.6.1.4.1.318.1.1.1.2.2.3.0', - 'factor' => 1/6000, - 'min' => '0', - 'max' => '360', - }, - 'timeO' => { - 'name' => 'Time On Battery', - 'oidtext' => 'PowerNet-MIB::upsBasicBatteryTimeOnBattery.0', - 'oid' => '1.3.6.1.4.1.318.1.1.1.2.1.2.0', - 'factor' => 1/6000, - 'min' => '0', - 'max' => '360', - }, -}; - use Net::SNMP; +use RRDs; +use Data::Dumper; + +$UPSGRAPH::host = ""; +$UPSGRAPH::rrdfile = ""; +$UPSGRAPH::outdir = ""; +$UPSGRAPH::community = "public"; +$UPSGRAPH::step = 60; +$UPSGRAPH::keep = (370*24*60*60)/$UPSGRAPH::step; +@UPSGRAPH::fields = (); +$UPSGRAPH::vars = {}; + +do $ARGV[0] or die "can't read config: $!"; + +my $host = $UPSGRAPH::host; +my $rrdfile = $UPSGRAPH::rrdfile; +my $outdir = $UPSGRAPH::outdir; +my $community = $UPSGRAPH::community; +my $step = $UPSGRAPH::step; +my $keep = $UPSGRAPH::keep; +my @fields = @UPSGRAPH::fields; +my $vars = $UPSGRAPH::vars; + +sub rrdcreate(@) { + my $rrdfile = shift; + my $start = shift; + + my @cmd = ("${rrdfile}", "--step=${step}"); + + if (defined($start)) { + push @cmd, "--start=${start}"; + } -if (! -e "${rrdfile}") { - my $cmd = "rrdtool create \"${rrdfile}\" "; foreach my $var (@fields) { - $cmd .= "DS:${var}:GAUGE:600:" . + push @cmd, "DS:${var}:GAUGE:600:" . $vars->{$var}->{'min'} . ":" . $vars->{$var}->{'max'} . " "; } - $cmd .= "RRA:AVERAGE:0.5:1:10080 --step ${step}"; + push @cmd, "RRA:AVERAGE:0.5:1:${keep}"; + RRDs::create(@cmd); + if (RRDs::error) { + print "Error while creating: " . RRDs::error . "\n"; + exit 1; + } +} + +if ($> == 0) { + if (@ARGV != 2) { + print STDERR "Running as root, please provide UID as 2th argument!\n"; + exit(1); + } + + print "Running as root, switching to ".$ARGV[1]."\n"; + $< = $> = $ARGV[1]; +} + +if (! -e "${rrdfile}") { print "Creating ${rrdfile}...\n"; - print `${cmd}`; + rrdcreate("${rrdfile}"); +} + +my $rrdinfo = RRDs::info("${rrdfile}"); +if (RRDs::error) { + print "Error while getting info: " . RRDs::error . "\n"; + exit 1; } +if ($rrdinfo->{'rra[0].rows'} != $keep) { + print "Resizing ${rrdfile} from " . $rrdinfo->{'rra[0].rows'} . + " to ${keep} samples.\n"; + + (my $start, my $ostep, my $names, my $data) = + RRDs::fetch("${rrdfile}", + "-s " . (time() - ($rrdinfo->{'rra[0].rows'} * $rrdinfo->{'step'})), + "AVERAGE"); + + if (RRDs::error) { + print "Error while fetching data: " . RRDs::error . "\n"; + exit 1; + } + + rrdcreate("${rrdfile}.new", (${start}-${ostep})); + + print "Preserving data since " . localtime($start) . "\n"; + + my $pos = $start; + foreach my $line (@$data) { + my $vline = "${pos}"; + + foreach my $val (@$line) { + $val = 'U' if (!defined($val)); + $vline .= ":${val}"; + } + RRDs::update("${rrdfile}.new", $vline) or die "Can't insert data\n"; + + if (RRDs::error) { + print "Error while updating: " . RRDs::error . "\n"; + exit 1; + } + $pos += $ostep; + + if ((($pos-$start)/$ostep) == $#$data) { + last; + } + } + + rename("${rrdfile}", "${rrdfile}.old") or die "Can't rename old file: $!\n"; + rename("${rrdfile}.new", "${rrdfile}") or die "Can't rename new file: $!\n"; + + $rrdinfo = RRDs::info("${rrdfile}"); + if (RRDs::error) { + print "Error while getting info: " . RRDs::error . "\n"; + exit 1; + } + + if ($rrdinfo->{'rra[0].rows'} != $keep) { + print "Failed!\n"; + exit 1; + } +} + +my $child = fork(); + +die "fork failed!" if (!defined($child)); + +exit 0 if ($child != 0); + while(1) { ($session,$error) = Net::SNMP->session(Hostname => $host, Community => $community); @@ -132,16 +156,17 @@ while(1) { $session->close; - my $cmd = "rrdtool update \"${rrdfile}\" \"N"; + my $vline = "N"; foreach my $var (@fields) { if (!(defined($vars->{$var}->{'value'}))) { $vars->{$var}->{'value'} = 'U'; } - print $vars->{$var}->{'name'}.": ".$vars->{$var}->{'value'}."\n"; - $cmd .= ":" . $vars->{$var}->{'value'}; + $vline .= ":" . $vars->{$var}->{'value'}; + } + RRDs::update("${rrdfile}", $vline); + if (RRDs::error) { + print "Error while updating: " . RRDs::error . "\n"; } - $cmd .= "\""; - print `${cmd}`; open(HTML, ">${outdir}/index.html.new"); @@ -149,21 +174,67 @@ while(1) { print HTML '
'; foreach my $var (@fields) { - my $graphdef = "-t \"" . $vars->{$var}->{'name'} . "\" \"DEF:${var}=${rrdfile}:${var}:AVERAGE\" \"LINE1:${var}#FF0000\""; - my $cmd = "rrdtool graph \"${outdir}/${var}.png.new\" -w 720 ${graphdef}"; - my $size = `$cmd`; - rename("${outdir}/${var}.png.new", "${outdir}/${var}.png"); - (my $width, my $height) = split(/x/,$size); + my @graphdef = ("-t", $vars->{$var}->{'name'}, "DEF:${var}=${rrdfile}:${var}:AVERAGE", "LINE1:${var}#FF0000"); + (my $averages, my $width, my $height) = + RRDs::graph("${outdir}/${var}.png.new", + "-w", "720", @graphdef); + + if (RRDs::error) { + print "Error while graphing: " . RRDs::error . "\n"; + } else { + rename("${outdir}/${var}.png.new", "${outdir}/${var}.png"); + } + + print HTML "