#!/usr/bin/perl -w
-if ((@ARGV != 3) && (@ARGV != 4)) {
- print STDERR "Syntax: ${0} host /path/to/file.rrd /output/directory/ [uid]\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 IO::Socket::INET;
+use RRDs;
+use File::Copy;
+use Data::Dumper;
+
+$UPSGRAPH::outdir = "";
+$UPSGRAPH::step = 60;
+$UPSGRAPH::keep = (370*24*60*60)/$UPSGRAPH::step;
+$UPSGRAPH::hosts = ();
+
+do $ARGV[0] or die "can't read config: $!";
+
+my $outdir = $UPSGRAPH::outdir;
+my $step = $UPSGRAPH::step;
+my $keep = $UPSGRAPH::keep;
+my $hosts = $UPSGRAPH::hosts;
+
+sub rrdcreate(@) {
+ my $newrrd = shift;
+ my $field = shift;
+ my $vars = 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;
+ }
+}
+
+sub fetch_snmp(@) {
+ my $address = shift;
+ my $community = shift;
+ my $oid = shift;
+
+ (my $session, my $error) = Net::SNMP->session(Hostname => $address,
+ Community => $community);
+
+ if (!$session) {
+ print STDERR "session error: $error";
+ }
+
+ $session->translate(0);
+
+ my $result = $session->get_request($oid);
+
+ $session->close;
+
+ return undef if (!defined($result));
+
+ $result->{$oid};
+}
+
+sub fetch_tcp(@) {
+ my $address = shift;
+ my $port = shift;
+
+ my $sock = IO::Socket::INET->new(PeerAddr => $address,
+ PeerPort => $port,
+ Proto => 'tcp',
+ Timeout => 1);
+
+ return undef if (!$sock);
+
+ chomp(my $value = <$sock>);
+
+ close($sock);
+
+ if (!$value) {
+ return undef;
+ }
+
+ $value=~ s/\s//g;
+
+ $value;
+}
if ($> == 0) {
- if (@ARGV != 4) {
- print STDERR "Running as root, please provide UID as 4th argument!\n";
+ if (@ARGV != 2) {
+ print STDERR "Running as root, please provide UID as 2th argument!\n";
exit(1);
}
- print "Running as root, switching to ".$ARGV[3]."\n";
- $< = $> = $ARGV[3];
+ print "Running as root, switching to ".$ARGV[1]."\n";
+ $< = $> = $ARGV[1];
}
-if (! -e "${rrdfile}") {
- my $cmd = "rrdtool create \"${rrdfile}\" ";
- foreach my $var (@fields) {
- $cmd .= "DS:${var}:GAUGE:600:" .
- $vars->{$var}->{'min'} . ":" .
- $vars->{$var}->{'max'} . " ";
+foreach my $host (@$hosts) {
+ my $rrdfile = $host->{'rrdfile'};
+
+ foreach my $var (keys(%{$host->{'vars'}})) {
+ $host->{'vars'}->{$var}->{'min'} = 'U' if (!defined($host->{'vars'}->{$var}->{'min'}));
+ $host->{'vars'}->{$var}->{'max'} = 'U' if (!defined($host->{'vars'}->{$var}->{'max'}));
}
- $cmd .= "RRA:AVERAGE:0.5:1:10080 --step ${step}";
- print "Creating ${rrdfile}...\n";
- print `${cmd}`;
-}
+ if (-e "${rrdfile}") {
+ print "Reading old ${rrdfile} to preserve data...\n";
-my $child = fork();
+ my $rrdinfo = RRDs::info("${rrdfile}");
+ if (RRDs::error) {
+ print "Error while getting info: " . RRDs::error . "\n";
+ exit 1;
+ }
-die "fork failed!" if (!defined($child));
+ (my $start, my $ostep, my $names, my $data) =
+ RRDs::fetch("${rrdfile}",
+ "-s " . (time() - ($rrdinfo->{'rra[0].rows'} * $rrdinfo->{'step'})),
+ "AVERAGE");
-exit 0 if ($child != 0);
+ if (RRDs::error) {
+ print "Error while fetching data: " . RRDs::error . "\n";
+ exit 1;
+ }
-while(1) {
- ($session,$error) = Net::SNMP->session(Hostname => $host,
- Community => $community);
+ foreach my $field (@$names) {
+ if (! -e "${rrdfile}.${field}") {
+ rrdcreate("${rrdfile}.${field}",
+ "${field}",
+ $host->{'vars'},
+ (${start}-${ostep}));
+ }
+ }
- die "session error: $error" unless ($session);
+ my $pos = $start;
+ foreach my $line (@$data) {
+ foreach my $field (@$names) {
+ my $val = shift (@$line);
+ $val = 'U' if (!defined($val));
- $session->translate(0);
+ RRDs::update("${rrdfile}.${field}", "${pos}:${val}");
+ if (RRDs::error) {
+ print "Can't insert data: " . RRDs::error . "\n";
+ exit 1;
+ }
- foreach my $var (@fields) {
- delete $vars->{$var}->{'value'};
+ }
- my $result = $session->get_request($vars->{$var}->{'oid'});
- next unless (defined $result);
+ $pos += $ostep;
- $vars->{$var}->{'value'} = $result->{$vars->{$var}->{'oid'}};
- if (defined($vars->{$var}->{'factor'})) {
- $vars->{$var}->{'value'} *= $vars->{$var}->{'factor'};
+ if ((($pos-$start)/$ostep) == $#$data) {
+ last;
+ }
}
+
+ rename("${rrdfile}", "${rrdfile}.old") or die "Can't rename old file: $!\n";
}
- $session->close;
+ foreach my $field (@{$host->{'fields'}}) {
+ if (! -e "${rrdfile}.${field}") {
+ print "Creating ${rrdfile}.${field}...\n";
+ rrdcreate("${rrdfile}.${field}",
+ "${field}",
+ $host->{'vars'});
+ }
+
+ my $rrdinfo = RRDs::info("${rrdfile}.${field}");
+ if (RRDs::error) {
+ print "Error while getting info: " . RRDs::error . "\n";
+ exit 1;
+ }
+
+ if (defined($rrdinfo->{"ds[${field}].min"})) {
+ if ($rrdinfo->{"ds[${field}].min"} ne $host->{'vars'}->{$field}->{'min'}) {
+ RRDs::tune("${rrdfile}.${field}","-i",$field.":".$host->{'vars'}->{$field}->{'min'});
+ }
+ } else {
+ if ($host->{'vars'}->{$field}->{'min'} ne 'U') {
+ RRDs::tune("${rrdfile}.${field}","-i",$field.":".$host->{'vars'}->{$field}->{'min'});
+ }
+ }
+
+ if (RRDs::error) {
+ print "Error while setting min: " . RRDs::error . "\n";
+ exit 1;
+ }
+
+ if (defined($rrdinfo->{"ds[${field}].max"})) {
+ if ($rrdinfo->{"ds[${field}].max"} ne $host->{'vars'}->{$field}->{'max'}) {
+ RRDs::tune("${rrdfile}.${field}","-a",$field.":".$host->{'vars'}->{$field}->{'max'});
+ }
+ } else {
+ if ($host->{'vars'}->{$field}->{'max'} ne 'U') {
+ RRDs::tune("${rrdfile}.${field}","-a",$field.":".$host->{'vars'}->{$field}->{'max'});
+ }
+ }
+
+ if (RRDs::error) {
+ print "Error while setting max: " . 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}",
+ $host->{'vars'},
+ (${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";
- my $cmd = "rrdtool update \"${rrdfile}\" \"N";
- foreach my $var (@fields) {
- if (!(defined($vars->{$var}->{'value'}))) {
- $vars->{$var}->{'value'} = 'U';
+ $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;
+ }
}
- $cmd .= ":" . $vars->{$var}->{'value'};
}
- $cmd .= "\"";
- print `${cmd}`;
+}
+
+my $child = fork();
+
+die "fork failed!" if (!defined($child));
+
+exit 0 if ($child != 0);
+while(1) {
open(HTML, ">${outdir}/index.html.new");
- print HTML '<html><head><meta http-equiv="refresh" content="60"/><meta http-equiv="cache-control" content="no-cache"/><meta http-equiv="pragma" content="no-cache"/><meta http_equiv="expires" content="Sat, 26 Jul 1997 05:00:00 GMT"/><title>USV status</title></head>';
+ print HTML '<html><head><meta http-equiv="refresh" content="60"/><meta http-equiv="cache-control" content="no-cache"/><meta http-equiv="pragma" content="no-cache"/><meta http_equiv="expires" content="Sat, 26 Jul 1997 05:00:00 GMT"/><title>Status</title></head>';
print HTML '<body bgcolor="#ffffff">';
- 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);
+ foreach my $host (@$hosts) {
+ print HTML "[<a href=\"#".${host}->{'name'}."\">".${host}->{'name'}."</a>] ";
+ }
+ print HTML "<br>\n";
+
+ foreach my $host (@$hosts) {
+ print HTML "<br>\n";
+ print HTML "<a name=\"".${host}->{'name'}."\"></a>\n";
+ my $vars = $host->{'vars'};
+ my $rrdfile = $host->{'rrdfile'};
+ my $hostname = $host->{'name'};
+
+ foreach my $var (@{$host->{'fields'}}) {
+ delete $vars->{$var}->{'value'};
+
+ my $result;
+
+ if ((!defined($vars->{$var}->{'proto'})) ||
+ ($vars->{$var}->{'proto'} eq '') ||
+ ($vars->{$var}->{'proto'} eq 'snmp')) {
+ $result = fetch_snmp($host->{'address'}, $host->{'community'}, $vars->{$var}->{'oid'});
+ } elsif ($vars->{$var}->{'proto'} eq 'tcp') {
+ $result = fetch_tcp($host->{'address'}, $vars->{$var}->{'port'});
+ }
- 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");
+ next unless (defined $result);
- print HTML "<a href=\"${var}.long.png\"><img src=\"${var}.png\" width=\"${width}\" height=\"${height}\" border=\"0\"></a>";
+ $vars->{$var}->{'value'} = $result;
+ if (defined($vars->{$var}->{'factor'})) {
+ $vars->{$var}->{'value'} *= $vars->{$var}->{'factor'};
+ }
+ }
+
+ foreach my $var (@{$host->{'fields'}}) {
+ if (!(defined($vars->{$var}->{'value'}))) {
+ $vars->{$var}->{'value'} = 'U';
+ }
+ RRDs::update("${rrdfile}.${var}", "N:" . $vars->{$var}->{'value'});
+ }
+ if (RRDs::error) {
+ print "Error while updating: " . RRDs::error . "\n";
+ }
+
+ foreach my $var (@{$host->{'fields'}}) {
+ my @graphdef = ('-P', "--lazy", "-t", $hostname." - ".$vars->{$var}->{'name'}, "DEF:${var}=${rrdfile}.${var}:${var}:AVERAGE", "LINE1:${var}#FF0000");
+
+ push @graphdef, "VDEF:cur=${var},LAST";
+ push @graphdef, 'GPRINT:cur:Current\\: <span foreground="#FF0000">%.2lf</span>\\r';
+
+ my $mtime;
+ $mtime=(stat("${outdir}/${hostname}.${var}.png.work"))[9];
+
+ (my $averages, my $width, my $height) =
+ RRDs::graph("${outdir}/${hostname}.${var}.png.work",
+ "-w", "720", @graphdef);
+
+ pop @graphdef;
+ pop @graphdef;
+
+ if (RRDs::error) {
+ print "Error while graphing: " . RRDs::error . "\n";
+ } else {
+ my $newmtime=(stat("${outdir}/${hostname}.${var}.png.work"))[9];
+ if ((!defined($mtime)) || ($newmtime != $mtime)) {
+ copy("${outdir}/${hostname}.${var}.png.work", "${outdir}/${hostname}.${var}.png.new");
+ rename("${outdir}/${hostname}.${var}.png.new", "${outdir}/${hostname}.${var}.png");
+ }
+ }
+
+ print HTML "<a href=\"${hostname}.${var}.html\"><img src=\"${hostname}.${var}.png\" width=\"${width}\" height=\"${height}\" border=\"0\"></a><br>\n";
+
+ open (HTML2, ">${outdir}/${hostname}.${var}.html.new");
+ print HTML2 '<html><head><meta http-equiv="refresh" content="60"/><meta http-equiv="cache-control" content="no-cache"/><meta http-equiv="pragma" content="no-cache"/><meta http_equiv="expires" content="Sat, 26 Jul 1997 05:00:00 GMT"/><title>' . $vars->{$var}->{'name'} . '</title></head>';
+ print HTML2 '<body bgcolor="#ffffff">';
+
+ push @graphdef, "VDEF:min=${var},MINIMUM";
+ push @graphdef, "GPRINT:min:Minimum\\: %.2lf";
+
+ push @graphdef, "VDEF:avg=${var},AVERAGE";
+ push @graphdef, "GPRINT:avg:Average\\: %.2lf";
+
+ push @graphdef, "VDEF:max=${var},MAXIMUM";
+ push @graphdef, "GPRINT:max:Maximum\\: %.2lf";
+
+ push @graphdef, "VDEF:cur=${var},LAST";
+ push @graphdef, "GPRINT:cur:Current\\: %.2lf";
+
+ $mtime=(stat("${outdir}/${hostname}.${var}.long.png.work"))[9];
+ ($averages, $width, $height) =
+ RRDs::graph("${outdir}/${hostname}.${var}.long.png.work",
+ "-w", "1008", @graphdef);
+
+ if (RRDs::error) {
+ print "Error while graphing: " . RRDs::error . "\n";
+ } else {
+ my $newmtime=(stat("${outdir}/${hostname}.${var}.long.png.work"))[9];
+ if ((!defined($mtime)) || ($newmtime != $mtime)) {
+ copy("${outdir}/${hostname}.${var}.long.png.work", "${outdir}/${hostname}.${var}.long.png.new");
+ rename("${outdir}/${hostname}.${var}.long.png.new", "${outdir}/${hostname}.${var}.long.png");
+ }
+ }
+
+ print HTML2 "<img src=\"${hostname}.${var}.long.png\" width=\"${width}\" height=\"${height}\"><br>";
+
+ $mtime=(stat("${outdir}/${hostname}.${var}.week.png.work"))[9];
+ ($averages, $width, $height) =
+ RRDs::graph("${outdir}/${hostname}.${var}.week.png.work",
+ "-w", "1008", "-e", "now", "-s", "end-1w", @graphdef);
+
+ if (RRDs::error) {
+ print "Error while graphing: " . RRDs::error . "\n";
+ } else {
+ my $newmtime=(stat("${outdir}/${hostname}.${var}.week.png.work"))[9];
+ if ((!defined($mtime)) || ($newmtime != $mtime)) {
+ copy("${outdir}/${hostname}.${var}.week.png.work", "${outdir}/${hostname}.${var}.week.png.new");
+ rename("${outdir}/${hostname}.${var}.week.png.new", "${outdir}/${hostname}.${var}.week.png");
+ }
+ }
+
+ print HTML2 "<img src=\"${hostname}.${var}.week.png\" width=\"${width}\" height=\"${height}\"><br>";
+
+ $mtime=(stat("${outdir}/${hostname}.${var}.year.png.work"))[9];
+ ($averages, $width, $height) =
+ RRDs::graph("${outdir}/${hostname}.${var}.year.png.work",
+ "-w", "1008", "-e", "now", "-s", "end-1y", @graphdef);
+
+ if (RRDs::error) {
+ print "Error while graphing: " . RRDs::error . "\n";
+ } else {
+ my $newmtime=(stat("${outdir}/${hostname}.${var}.year.png.work"))[9];
+ if ((!defined($mtime)) || ($newmtime != $mtime)) {
+ copy("${outdir}/${hostname}.${var}.year.png.work", "${outdir}/${hostname}.${var}.year.png.new");
+ rename("${outdir}/${hostname}.${var}.year.png.new", "${outdir}/${hostname}.${var}.year.png");
+ }
+ }
+
+ print HTML2 "<img src=\"${hostname}.${var}.year.png\" width=\"${width}\" height=\"${height}\"><br>";
+
+ print HTML2 "</body></html>\n";
+ close(HTML2);
+ rename("${outdir}/${hostname}.${var}.html.new", "${outdir}/${hostname}.${var}.html");
+ }
}
print HTML "</body></html>\n";