X-Git-Url: https://git.zerfleddert.de/cgi-bin/gitweb.cgi/upsgraph/blobdiff_plain/e5da3876baf867c77cf76593d26d88c76c6c49ae..6c1071ee6b37201ba1e39703a198c35d219fc647:/upsgraph.pl

diff --git a/upsgraph.pl b/upsgraph.pl
index 19d8147..812de11 100755
--- a/upsgraph.pl
+++ b/upsgraph.pl
@@ -9,41 +9,34 @@ 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 = {};
+$UPSGRAPH::hosts = ();
 
 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;
+my $hosts = $UPSGRAPH::hosts;
 
 sub rrdcreate(@) {
-	my $rrdfile = shift;
+	my $newrrd = shift;
+	my $field = shift;
+	my $vars = shift;
 	my $start = shift;
 
-	my @cmd = ("${rrdfile}", "--step=${step}");
+	my @cmd = ("${newrrd}", "--step=${step}");
 
 	if (defined($start)) {
 		push @cmd, "--start=${start}";
 	}
 
-	foreach my $var (@fields) {
-		push @cmd, "DS:${var}:GAUGE:600:" .
-			$vars->{$var}->{'min'} . ":" .
-			$vars->{$var}->{'max'} . " ";
-	}
+	push @cmd, "DS:${field}:GAUGE:600:" .
+		$vars->{$field}->{'min'} . ":" .
+		$vars->{$field}->{'max'} . " ";
+
 	push @cmd, "RRA:AVERAGE:0.5:1:${keep}";
 
 	RRDs::create(@cmd);
@@ -53,6 +46,25 @@ sub rrdcreate(@) {
 	}
 }
 
+sub fetch_snmp(@) {
+	my $address = shift;
+	my $community = shift;
+	my $oid = shift;
+
+	(my $session, my $error) = Net::SNMP->session(Hostname => $address,
+			Community => $community);
+
+	die "session error: $error" unless ($session);
+
+	$session->translate(0);
+
+	my $val = $session->get_request($oid);
+
+	$session->close;
+
+	$val;
+}
+
 if ($> == 0) {
 	if (@ARGV != 2) {
 		print STDERR "Running as root, please provide UID as 2th argument!\n";
@@ -63,68 +75,131 @@ if ($> == 0) {
 	$< = $> = $ARGV[1];
 }
 
-if (! -e "${rrdfile}") {
-	print "Creating ${rrdfile}...\n";
-	rrdcreate("${rrdfile}");
-}
+foreach my $host (@$hosts) {
+	my $rrdfile = $host->{'rrdfile'};
 
-my $rrdinfo = RRDs::info("${rrdfile}");
-if (RRDs::error) {
-	print "Error while getting info: " . RRDs::error . "\n";
-	exit 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;
+		}
 
-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");
 
-	(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;
+		}
 
-	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}",
+					$host->{'vars'},
+					(${start}-${ostep}));
+			}
+		}
 
-	rrdcreate("${rrdfile}.new", (${start}-${ostep}));
+		my $pos = $start;
+		foreach my $line (@$data) {
+			foreach my $field (@$names) {
+				my $val = shift (@$line);
+				$val = 'U' if (!defined($val));
 
-	print "Preserving data since " . localtime($start) . "\n";
+				RRDs::update("${rrdfile}.${field}", "${pos}:${val}");
+				if (RRDs::error) {
+					print "Can't insert data: " . RRDs::error . "\n";
+					exit 1;
+				}
 
-	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";
+			$pos += $ostep;
 
-		if (RRDs::error) {
-			print "Error while updating: " . RRDs::error . "\n";
-			exit 1;
+			if ((($pos-$start)/$ostep) == $#$data) {
+				last;
+			}
 		}
-		$pos += $ostep;
 
-		if ((($pos-$start)/$ostep) == $#$data) {
-			last;
-		}
+		rename("${rrdfile}", "${rrdfile}.old") or die "Can't rename old file: $!\n";
 	}
 
-	rename("${rrdfile}", "${rrdfile}.old") or die "Can't rename old file: $!\n";
-	rename("${rrdfile}.new", "${rrdfile}") or die "Can't rename new file: $!\n";
+	foreach my $field (@{$host->{'fields'}}) {
+		if (! -e "${rrdfile}.${field}") {
+			print "Creating ${rrdfile}.${field}...\n";
+			rrdcreate("${rrdfile}.${field}",
+				"${field}",
+				$host->{'vars'});
+		}
 
-	$rrdinfo = RRDs::info("${rrdfile}");
-	if (RRDs::error) {
-		print "Error while getting info: " . RRDs::error . "\n";
-		exit 1;
-	}
+		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 "Failed!\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";
+
+			$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;
+			}
+		}
 	}
 }
 
@@ -135,101 +210,109 @@ die "fork failed!" if (!defined($child));
 exit 0 if ($child != 0);
 
 while(1) {
-	($session,$error) = Net::SNMP->session(Hostname => $host,
-			Community => $community);
+	open(HTML, ">${outdir}/index.html.new");
 
-	die "session error: $error" unless ($session);
+	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 '<body bgcolor="#ffffff">';
 
-	$session->translate(0);
+	foreach my $host (@$hosts) {
+		my $vars = $host->{'vars'};
+		my $rrdfile = $host->{'rrdfile'};
+		my $hostname = $host->{'name'};
 
-	foreach my $var (@fields) {
-		delete $vars->{$var}->{'value'};
+		foreach my $var (@{$host->{'fields'}}) {
+			delete $vars->{$var}->{'value'};
 
-		my $result = $session->get_request($vars->{$var}->{'oid'});
-		next unless (defined $result);
+			my $result = fetch_snmp($host->{'address'}, $host->{'community'}, $vars->{$var}->{'oid'});
+			next unless (defined $result);
 
-		$vars->{$var}->{'value'} = $result->{$vars->{$var}->{'oid'}};
-		if (defined($vars->{$var}->{'factor'})) {
-			$vars->{$var}->{'value'} *= $vars->{$var}->{'factor'};
+			$vars->{$var}->{'value'} = $result->{$vars->{$var}->{'oid'}};
+			if (defined($vars->{$var}->{'factor'})) {
+				$vars->{$var}->{'value'} *= $vars->{$var}->{'factor'};
+			}
 		}
-	}
-
-	$session->close;
 
-	my $vline = "N";
-	foreach my $var (@fields) {
-		if (!(defined($vars->{$var}->{'value'}))) {
-			$vars->{$var}->{'value'} = 'U';
+		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";
 		}
-		$vline .= ":" . $vars->{$var}->{'value'};
-	}
-	RRDs::update("${rrdfile}", $vline);
-	if (RRDs::error) {
-		print "Error while updating: " . RRDs::error . "\n";
-	}
 
-	open(HTML, ">${outdir}/index.html.new");
+		foreach my $var (@{$host->{'fields'}}) {
+			my @graphdef = ("-t", $hostname." - ".$vars->{$var}->{'name'}, "DEF:${var}=${rrdfile}.${var}:${var}:AVERAGE", "LINE1:${var}#FF0000");
+			(my $averages, my $width, my $height) =
+				RRDs::graph("${outdir}/${hostname}.${var}.png.new",
+						"-w", "720", @graphdef);
 
-	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 '<body bgcolor="#ffffff">';
+			if (RRDs::error) {
+				print "Error while graphing: " . RRDs::error . "\n";
+			} else {
+				rename("${outdir}/${hostname}.${var}.png.new", "${outdir}/${hostname}.${var}.png");
+			}
 
-	foreach my $var (@fields) {
-		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);
+			print HTML "<a href=\"${hostname}.${var}.html\"><img src=\"${hostname}.${var}.png\" width=\"${width}\" height=\"${height}\" border=\"0\"></a>";
 
-		if (RRDs::error) {
-			print "Error while graphing: " . RRDs::error . "\n";
-		} else {
-			rename("${outdir}/${var}.png.new", "${outdir}/${var}.png");
-		}
+			open (HTML2, ">${outdir}/${hostname}.${var}.html.new");
+			print HTML2 "<html><head><title>" . $vars->{$var}->{'name'} . "</title></head>";
+			print HTML2 '<body bgcolor="#ffffff">';
 
-		print HTML "<a href=\"${var}.html\"><img src=\"${var}.png\" width=\"${width}\" height=\"${height}\" border=\"0\"></a>";
 
-		open (HTML2, ">${outdir}/${var}.html.new");
-		print HTML2 "<html><head><title>" . $vars->{$var}->{'name'} . "</title></head>";
-		print HTML2 '<body bgcolor="#ffffff">';
+			push @graphdef, "VDEF:min=${var},MINIMUM";
+			push @graphdef, "GPRINT:min:Minimum\\: %.2lf";
 
-		($averages, $width, $height) =
-			RRDs::graph("${outdir}/${var}.long.png.new",
-			"-w", "1008", @graphdef);
+			push @graphdef, "VDEF:avg=${var},AVERAGE";
+			push @graphdef, "GPRINT:avg:Average\\: %.2lf";
 
-		if (RRDs::error) {
-			print "Error while graphing: " . RRDs::error . "\n";
-		} else {
-			rename("${outdir}/${var}.long.png.new", "${outdir}/${var}.long.png");
-		}
+			push @graphdef, "VDEF:max=${var},MAXIMUM";
+			push @graphdef, "GPRINT:max:Maximum\\: %.2lf";
 
-		print HTML2 "<img src=\"${var}.long.png\" width=\"${width}\" height=\"${height}\"><br>";
+			push @graphdef, "VDEF:cur=${var},LAST";
+			push @graphdef, "GPRINT:cur:Current\\: %.2lf";
 
-		($averages, $width, $height) =
-			RRDs::graph("${outdir}/${var}.week.png.new",
-			"-w", "1008", "-e", "now", "-s", "end-1w", @graphdef);
+			($averages, $width, $height) =
+				RRDs::graph("${outdir}/${hostname}.${var}.long.png.new",
+						"-w", "1008", @graphdef);
 
-		if (RRDs::error) {
-			print "Error while graphing: " . RRDs::error . "\n";
-		} else {
-			rename("${outdir}/${var}.week.png.new", "${outdir}/${var}.week.png");
-		}
+			if (RRDs::error) {
+				print "Error while graphing: " . RRDs::error . "\n";
+			} else {
+				rename("${outdir}/${hostname}.${var}.long.png.new", "${outdir}/${hostname}.${var}.long.png");
+			}
 
-		print HTML2 "<img src=\"${var}.week.png\" width=\"${width}\" height=\"${height}\"><br>";
+			print HTML2 "<img src=\"${hostname}.${var}.long.png\" width=\"${width}\" height=\"${height}\"><br>";
 
-		($averages, $width, $height) =
-			RRDs::graph("${outdir}/${var}.year.png.new",
-			"-w", "1008", "-e", "now", "-s", "end-1y", @graphdef);
+			($averages, $width, $height) =
+				RRDs::graph("${outdir}/${hostname}.${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}.year.png.new", "${outdir}/${var}.year.png");
-		}
+			if (RRDs::error) {
+				print "Error while graphing: " . RRDs::error . "\n";
+			} else {
+				rename("${outdir}/${hostname}.${var}.week.png.new", "${outdir}/${hostname}.${var}.week.png");
+			}
 
-		print HTML2 "<img src=\"${var}.year.png\" width=\"${width}\" height=\"${height}\"><br>";
+			print HTML2 "<img src=\"${hostname}.${var}.week.png\" width=\"${width}\" height=\"${height}\"><br>";
 
-		print HTML2 "</body></html>\n";
-		close(HTML2);
-		rename("${outdir}/${var}.html.new", "${outdir}/${var}.html");
+			($averages, $width, $height) =
+				RRDs::graph("${outdir}/${hostname}.${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}/${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";