X-Git-Url: https://git.zerfleddert.de/cgi-bin/gitweb.cgi/upsgraph/blobdiff_plain/863e62dca19c7c24de77bcee3258c2f4c7afdb66..21377f4360914189ced6c9dfcf609372914aa65d:/upsgraph.pl diff --git a/upsgraph.pl b/upsgraph.pl index aa77721..3d35c2a 100755 --- a/upsgraph.pl +++ b/upsgraph.pl @@ -1,113 +1,182 @@ #!/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; -if (! -e "${rrdfile}") { - my $cmd = "rrdtool create \"${rrdfile}\" "; - foreach my $var (@fields) { - $cmd .= "DS:${var}:GAUGE:600:" . - $vars->{$var}->{'min'} . ":" . - $vars->{$var}->{'max'} . " "; +$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 $newrrd = shift; + my $field = shift; + my $start = shift; + + my @cmd = ("${newrrd}", "--step=${step}"); + + if (defined($start)) { + push @cmd, "--start=${start}"; + } + + push @cmd, "DS:${field}:GAUGE:600:" . + $vars->{$field}->{'min'} . ":" . + $vars->{$field}->{'max'} . " "; + + 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 "Reading old ${rrdfile} to preserve data...\n"; + + my $rrdinfo = RRDs::info("${rrdfile}"); + if (RRDs::error) { + print "Error while getting info: " . RRDs::error . "\n"; + exit 1; + } + + (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; + } + + foreach my $field (@$names) { + if (! -e "${rrdfile}.${field}") { + rrdcreate("${rrdfile}.${field}","${field}",(${start}-${ostep})); + } + } + + my $pos = $start; + foreach my $line (@$data) { + foreach my $field (@$names) { + my $val = shift (@$line); + next if (!defined($val)); + + RRDs::update("${rrdfile}.${field}", "${pos}:${val}"); + if (RRDs::error) { + print "Can't insert data: " . RRDs::error . "\n"; + exit 1; + } + + } + + $pos += $ostep; + + if ((($pos-$start)/$ostep) == $#$data) { + last; + } } - $cmd .= "RRA:AVERAGE:0.5:1:10080 --step ${step}"; - print "Creating ${rrdfile}...\n"; - print `${cmd}`; + rename("${rrdfile}", "${rrdfile}.old") or die "Can't rename old file: $!\n"; +} + +foreach my $field (@fields) { + if (! -e "${rrdfile}.${field}") { + print "Creating ${rrdfile}.${field}...\n"; + rrdcreate("${rrdfile}.${field}","${field}"); + } + + my $rrdinfo = RRDs::info("${rrdfile}.${field}"); + if (RRDs::error) { + print "Error while getting info: " . RRDs::error . "\n"; + exit 1; + } + + if ($rrdinfo->{'rra[0].rows'} != $keep) { + print "Resizing ${rrdfile}.${field} from " . $rrdinfo->{'rra[0].rows'} . + " to ${keep} samples.\n"; + + (my $start, my $ostep, my $names, my $data) = + RRDs::fetch("${rrdfile}.${field}", + "-s " . (time() - ($rrdinfo->{'rra[0].rows'} * $rrdinfo->{'step'})), + "AVERAGE"); + + if (RRDs::error) { + print "Error while fetching data: " . RRDs::error . "\n"; + exit 1; + } + + rrdcreate("${rrdfile}.${field}.new", "${field}", (${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}.${field}.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}.${field}", "${rrdfile}.${field}.old") or die "Can't rename old file: $!\n"; + rename("${rrdfile}.${field}.new", "${rrdfile}.${field}") or die "Can't rename new file: $!\n"; + + $rrdinfo = RRDs::info("${rrdfile}.${field}"); + 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(); @@ -138,15 +207,15 @@ while(1) { $session->close; - my $cmd = "rrdtool update \"${rrdfile}\" \"N"; foreach my $var (@fields) { if (!(defined($vars->{$var}->{'value'}))) { $vars->{$var}->{'value'} = 'U'; } - $cmd .= ":" . $vars->{$var}->{'value'}; + RRDs::update("${rrdfile}.${var}", "N:" . $vars->{$var}->{'value'}); + } + if (RRDs::error) { + print "Error while updating: " . RRDs::error . "\n"; } - $cmd .= "\""; - print `${cmd}`; open(HTML, ">${outdir}/index.html.new"); @@ -154,17 +223,62 @@ 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}:${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 ""; + + open (HTML2, ">${outdir}/${var}.html.new"); + print HTML2 "" . $vars->{$var}->{'name'} . ""; + print HTML2 ''; + + ($averages, $width, $height) = + RRDs::graph("${outdir}/${var}.long.png.new", + "-w", "1008", @graphdef); + + if (RRDs::error) { + print "Error while graphing: " . RRDs::error . "\n"; + } else { + rename("${outdir}/${var}.long.png.new", "${outdir}/${var}.long.png"); + } + + print HTML2 "
"; + + ($averages, $width, $height) = + RRDs::graph("${outdir}/${var}.week.png.new", + "-w", "1008", "-e", "now", "-s", "end-1w", @graphdef); + + if (RRDs::error) { + print "Error while graphing: " . RRDs::error . "\n"; + } else { + rename("${outdir}/${var}.week.png.new", "${outdir}/${var}.week.png"); + } + + print HTML2 "
"; + + ($averages, $width, $height) = + RRDs::graph("${outdir}/${var}.year.png.new", + "-w", "1008", "-e", "now", "-s", "end-1y", @graphdef); + + if (RRDs::error) { + print "Error while graphing: " . RRDs::error . "\n"; + } else { + rename("${outdir}/${var}.year.png.new", "${outdir}/${var}.year.png"); + } - my $cmd2 = "rrdtool graph \"${outdir}/${var}.long.png.new\" -w 1008 -e now -s end-1w ${graphdef}"; - my $size2 = `$cmd2`; - rename("${outdir}/${var}.long.png.new", "${outdir}/${var}.long.png"); + print HTML2 "
"; - print HTML ""; + print HTML2 "\n"; + close(HTML2); + rename("${outdir}/${var}.html.new", "${outdir}/${var}.html"); } print HTML "\n";