X-Git-Url: https://git.zerfleddert.de/cgi-bin/gitweb.cgi/upsgraph/blobdiff_plain/e26b065f3d496edd6845b58c5a3598793315876f..d036d16a336b8b31db2c7b09ea40012164d96a83:/upsgraph.pl

diff --git a/upsgraph.pl b/upsgraph.pl
index 04a4f8c..cade9b2 100755
--- a/upsgraph.pl
+++ b/upsgraph.pl
@@ -1,5 +1,9 @@
 #!/usr/bin/perl -w
 
+#Due to memory leak in Debian squeeze (Bug #545519)
+my $use_rrds = 0;
+my $rrd_result = 0;
+
 if ((@ARGV != 1) && (@ARGV != 2)) {
 	print STDERR "Syntax: ${0} configfile [uid]\n";
 	exit(1);
@@ -7,22 +11,66 @@ if ((@ARGV != 1) && (@ARGV != 2)) {
 
 use Net::SNMP;
 use IO::Socket::INET;
+use IO::Select;
 use RRDs;
 use File::Copy;
 use Data::Dumper;
 
 $UPSGRAPH::outdir = "";
+$UPSGRAPH::daysCovered = 370;
 $UPSGRAPH::step = 60;
-$UPSGRAPH::keep = (370*24*60*60)/$UPSGRAPH::step;
+$UPSGRAPH::stepsPerHour = (60 * 60) / $UPSGRAPH::step;
+$UPSGRAPH::keep = ($UPSGRAPH::daysCovered*24*60*60)/$UPSGRAPH::step;
+$UPSGRAPH::keepHours = ($UPSGRAPH::daysCovered*24*60*60)/$UPSGRAPH::stepsPerHour/$UPSGRAPH::step;
 $UPSGRAPH::hosts = ();
+$UPSGRAPH::regenerateOnStart = 1; #when set, regenerate graphs on script startup
 
 do $ARGV[0] or die "can't read config: $!";
 
 my $outdir = $UPSGRAPH::outdir;
 my $step = $UPSGRAPH::step;
+my $stepsPerHour = $UPSGRAPH::stepsPerHour;
 my $keep = $UPSGRAPH::keep;
+my $keepHours = $UPSGRAPH::keepHours;
 my $hosts = $UPSGRAPH::hosts;
 
+sub rrd_update(@) {
+	my @args = @_;
+
+	if ($use_rrds == 1) {
+		RRDs::update(@args);
+		$rrd_result = RRDs::error;
+	} else {
+		$rrd_result = system("rrdtool", "update", @args);
+	}
+}
+
+sub rrd_graph(@) {
+	my @args = @_;
+	my @rrd_out = ();
+
+	if ($use_rrds == 1) {
+		@rrd_out = RRDs::graph(@args);
+		$rrd_result = RRDs::error;
+	} else {
+		my $rrd_stdout;
+
+		open(RRDFD, '-|', 'rrdtool', 'graph', @args);
+		while(<RRDFD>) {
+			chomp;
+			$rrd_stdout = $_;
+		}
+		close(RRDFD);
+		$rrd_result = $?;
+		if ($rrd_result == 0) {
+			push @rrd_out, 0;
+			push @rrd_out, split(/x/, $rrd_stdout);
+		}
+	}
+
+	return @rrd_out;
+}
+
 sub rrdcreate(@) {
 	my $newrrd = shift;
 	my $field = shift;
@@ -37,9 +85,11 @@ sub rrdcreate(@) {
 
 	push @cmd, "DS:${field}:GAUGE:600:" .
 		$vars->{$field}->{'min'} . ":" .
-		$vars->{$field}->{'max'} . " ";
+		$vars->{$field}->{'max'};
 
 	push @cmd, "RRA:AVERAGE:0.5:1:${keep}";
+	push @cmd, "RRA:MIN:0.4:${stepsPerHour}:${keepHours}";
+	push @cmd, "RRA:MAX:0.4:${stepsPerHour}:${keepHours}";
 
 	RRDs::create(@cmd);
 	if (RRDs::error) {
@@ -58,6 +108,7 @@ sub fetch_snmp(@) {
 
 	if (!$session) {
 		print STDERR "session error: $error";
+		return undef;
 	}
 
 	$session->translate(0);
@@ -82,7 +133,13 @@ sub fetch_tcp(@) {
 
 	return undef if (!$sock);
 
-	chomp(my $value = <$sock>);
+	my $select = IO::Select->new($sock);
+
+	my $value = undef;
+
+	if ($select->can_read(1)) {
+		chomp($value) if (sysread($sock, $value, 4096) > 0);
+	}
 
 	close($sock);
 
@@ -95,6 +152,95 @@ sub fetch_tcp(@) {
 	$value;
 }
 
+sub fetch_tcp_multi(@) {
+	my $address = shift;
+	my $port = shift;
+	my $delimiter = shift;
+	my %values;
+
+	my $sock = IO::Socket::INET->new(PeerAddr => $address,
+			PeerPort => $port,
+			Proto => 'tcp',
+			Timeout => 1);
+
+	return undef if (!$sock);
+
+	my $select = IO::Select->new($sock);
+
+	while($select->can_read(1)) {
+		if (sysread($sock, my $buf, 16384) > 0) {
+			$buf=~s/\r//g;
+			foreach my $line (split(/\n/, $buf)) {
+				(my $key, my $value) = split(/${delimiter}/, $line);
+				$value=~ s/\s//g;
+				$values{$key} = $value;
+			}
+		} else {
+			last;
+		}
+	}
+
+	close($sock);
+
+	%values;
+}
+
+sub dayGraphFunc {
+	my $dataSrc = shift;
+	my $mode = shift;
+	my $color = shift;
+	my $label = shift;
+	my $dataPoints = shift;
+	my @args = ();
+	push @args, "CDEF:prev${mode}1=PREV(${dataSrc})";
+	for (my $i = 1; $i < $dataPoints - 1; ++$i) {
+		my $prev = $i;
+		my $next = $i+1;
+		push @args, "CDEF:prev${mode}${next}=PREV(prev${mode}${prev})";
+	}
+	my $dayCons = '';
+	my $consFunc = '';
+	for (my $i = 1; $i < $dataPoints; ++$i) {
+		$dayCons .= "prev${mode}${i},";
+		$consFunc .= ",${mode}";
+	}
+	push @args, "CDEF:day${mode}=${dayCons}${dataSrc}${consFunc}";
+	push @args, "CDEF:fillCalDay${mode}0=COUNT,${dataPoints},%,0,EQ,day${mode},UNKN,IF";
+	for (my $i = 1; $i < $dataPoints; ++$i) {
+		my $prev = $i-1;
+		my $next = $i;
+		push @args, "CDEF:fillCalDay${mode}${next}=PREV(fillCalDay${mode}${prev})";
+	}
+	my $fillPoint = '';
+	my $if = '';
+	for (my $i = 0; $i < $dataPoints; ++$i) {
+		$fillPoint .= "COUNT,${dataPoints},%,${i},EQ,fillCalDay${mode}${i},";
+		$if .= ",IF";
+	}
+	push @args, "CDEF:${mode}Curve=${fillPoint}UNKN${if}";
+	my $forwardShift = (24*60*60) * ($dataPoints - 1) / $dataPoints;
+	push @args, "SHIFT:${mode}Curve:-${forwardShift}";
+	push @args, "LINE1:${mode}Curve#${color}:${label}";
+	return \@args;
+}
+
+sub cfgToGraphDef {
+	my $cfg = shift;
+	my $varname = shift;
+	my $dpPerDay= shift;
+	my @graphDef = ();
+	foreach my $subGraph (@$cfg) {
+		if ($subGraph eq 'avg') {
+			push @graphDef, "LINE1:${varname}-avg#FF0000";
+		} elsif ($subGraph eq 'day-min') {
+			push @graphDef, @{dayGraphFunc("${varname}-min", 'MIN', '0000ff', 'Day Minimum Temperature', $dpPerDay)};
+		} elsif ($subGraph eq 'day-max') {
+			push @graphDef, @{dayGraphFunc("${varname}-max", 'MAX', '00ff00', 'Day Maximum Temperature', $dpPerDay)};
+		}
+	}
+	return \@graphDef;
+}
+
 if ($> == 0) {
 	if (@ARGV != 2) {
 		print STDERR "Running as root, please provide UID as 2th argument!\n";
@@ -179,12 +325,15 @@ foreach my $host (@$hosts) {
 			exit 1;
 		}
 
+		my $limitsChanged = 0;
 		if (defined($rrdinfo->{"ds[${field}].min"})) {
 			if ($rrdinfo->{"ds[${field}].min"} ne $host->{'vars'}->{$field}->{'min'}) {
+				$limitsChanged = 1;
 				RRDs::tune("${rrdfile}.${field}","-i",$field.":".$host->{'vars'}->{$field}->{'min'});
 			}
 		} else {
 			if ($host->{'vars'}->{$field}->{'min'} ne 'U') {
+				$limitsChanged = 1;
 				RRDs::tune("${rrdfile}.${field}","-i",$field.":".$host->{'vars'}->{$field}->{'min'});
 			}
 		}
@@ -196,10 +345,12 @@ foreach my $host (@$hosts) {
 
 		if (defined($rrdinfo->{"ds[${field}].max"})) {
 			if ($rrdinfo->{"ds[${field}].max"} ne $host->{'vars'}->{$field}->{'max'}) {
+				$limitsChanged = 1;
 				RRDs::tune("${rrdfile}.${field}","-a",$field.":".$host->{'vars'}->{$field}->{'max'});
 			}
 		} else {
 			if ($host->{'vars'}->{$field}->{'max'} ne 'U') {
+				$limitsChanged = 1;
 				RRDs::tune("${rrdfile}.${field}","-a",$field.":".$host->{'vars'}->{$field}->{'max'});
 			}
 		}
@@ -209,7 +360,10 @@ foreach my $host (@$hosts) {
 			exit 1;
 		}
 		    
-		if ($rrdinfo->{'rra[0].rows'} != $keep) {
+		if ($rrdinfo->{'rra[0].rows'} != $keep ||
+			!defined($rrdinfo->{'rra[1].rows'}) || $rrdinfo->{'rra[1].rows'} != $keepHours ||
+			$limitsChanged == 1) {
+
 			print "Resizing ${rrdfile}.${field} from " . $rrdinfo->{'rra[0].rows'} .
 				" to ${keep} samples.\n";
 
@@ -228,7 +382,7 @@ foreach my $host (@$hosts) {
 				$host->{'vars'},
 				(${start}-${ostep}));
 
-			print "Preserving data since " . localtime($start) . "\n";
+			print "Preserving data in file ${rrdfile}.${field} since " . localtime($start) . "\n";
 
 			my $pos = $start;
 			foreach my $line (@$data) {
@@ -274,6 +428,7 @@ die "fork failed!" if (!defined($child));
 
 exit 0 if ($child != 0);
 
+my $first = $UPSGRAPH::regenerateOnStart;
 while(1) {
 	open(HTML, ">${outdir}/index.html.new");
 
@@ -291,6 +446,7 @@ while(1) {
 		my $vars = $host->{'vars'};
 		my $rrdfile = $host->{'rrdfile'};
 		my $hostname = $host->{'name'};
+		my %multi_values = ();
 
 		foreach my $var (@{$host->{'fields'}}) {
 			delete $vars->{$var}->{'value'};
@@ -303,6 +459,14 @@ while(1) {
 				$result = fetch_snmp($host->{'address'}, $host->{'community'}, $vars->{$var}->{'oid'});
 			} elsif ($vars->{$var}->{'proto'} eq 'tcp') {
 				$result = fetch_tcp($host->{'address'}, $vars->{$var}->{'port'});
+			} elsif ($vars->{$var}->{'proto'} eq 'tcp_multi') {
+				if (defined($multi_values{$vars->{$var}->{'multi_id'}})) {
+					$result = $multi_values{$vars->{$var}->{'multi_id'}}
+				} else {
+					my %values = fetch_tcp_multi($host->{'address'}, $vars->{$var}->{'port'}, $vars->{$var}->{'multi_delimiter'});
+					@multi_values{keys %values} = values %values;
+					$result = $multi_values{$vars->{$var}->{'multi_id'}};
+				}
 			}
 
 			next unless (defined $result);
@@ -317,78 +481,140 @@ while(1) {
 			if (!(defined($vars->{$var}->{'value'}))) {
 				$vars->{$var}->{'value'} = 'U';
 			}
-			RRDs::update("${rrdfile}.${var}", "N:" . $vars->{$var}->{'value'});
+			rrd_update("${rrdfile}.${var}", "N:" . $vars->{$var}->{'value'});
 		}
-		if (RRDs::error) {
-			print "Error while updating: " . RRDs::error . "\n";
+		if ($rrd_result) {
+			print "Error while updating: " . $rrd_result . "\n";
 		}
 
 		foreach my $var (@{$host->{'fields'}}) {
-			my @graphdef = ("--lazy", "-t", $hostname." - ".$vars->{$var}->{'name'}, "DEF:${var}=${rrdfile}.${var}:${var}:AVERAGE", "LINE1:${var}#FF0000");
+			my $graphWidth = 365 * 3;
+			my $graphConfig;
+			if (defined $vars->{$var}->{'graph'}) {
+				$graphConfig = $vars->{$var}->{'graph'};
+				foreach my $subGraph (qw(day week year)) {
+					if (!defined($graphConfig->{$subGraph})) {
+						$graphConfig->{$subGraph} = [ 'avg'];
+					}
+				}
+			} else {
+				$graphConfig = {
+					'day' =>  [ 'avg' ],
+					'week' => [ 'avg' ],
+					'year' => [ 'avg' ],
+				};
+			}
+
+			my @graphdef = ('-P', "-t", $hostname." - ".$vars->{$var}->{'name'});
+			push @graphdef, "--lazy" if !$first;
+
+			push @graphdef, "DEF:${var}-avg=${rrdfile}.${var}:${var}:AVERAGE";
+			push @graphdef, "DEF:${var}-min=${rrdfile}.${var}:${var}:MIN";
+			push @graphdef, "DEF:${var}-max=${rrdfile}.${var}:${var}:MAX";
+			push @graphdef, "LINE1:${var}-avg#FF0000";
+			push @graphdef, "VDEF:cur=${var}-avg,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.new",
+				rrd_graph("${outdir}/${hostname}.${var}.png.work",
 						"-w", "720", @graphdef);
 
-			if (RRDs::error) {
-				print "Error while graphing: " . RRDs::error . "\n";
+			pop @graphdef;
+			pop @graphdef;
+			pop @graphdef;
+
+			if ($rrd_result) {
+				print "Error while graphing: " . $rrd_result . "\n";
 			} else {
-				copy("${outdir}/${hostname}.${var}.png.new", "${outdir}/${hostname}.${var}.png");
+				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><title>" . $vars->{$var}->{'name'} . "</title></head>";
+			print HTML2 '<html><head><meta http-equiv="Content-Type" content="text/html; charset=utf-8"><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, "VDEF:min=${var}-min,MINIMUM";
 			push @graphdef, "GPRINT:min:Minimum\\: %.2lf";
 
-			push @graphdef, "VDEF:avg=${var},AVERAGE";
+			push @graphdef, "VDEF:avg=${var}-avg,AVERAGE";
 			push @graphdef, "GPRINT:avg:Average\\: %.2lf";
 
-			push @graphdef, "VDEF:max=${var},MAXIMUM";
+			push @graphdef, "VDEF:max=${var}-max,MAXIMUM";
 			push @graphdef, "GPRINT:max:Maximum\\: %.2lf";
 
-			push @graphdef, "VDEF:cur=${var},LAST";
+			push @graphdef, "VDEF:cur=${var}-avg,LAST";
 			push @graphdef, "GPRINT:cur:Current\\: %.2lf";
 
+			my @dayGraphDef = @graphdef;
+			push @dayGraphDef, @{cfgToGraphDef($graphConfig->{'day'}, ${var}, 24)};
+
+			$mtime=(stat("${outdir}/${hostname}.${var}.long.png.work"))[9];
 			($averages, $width, $height) =
-				RRDs::graph("${outdir}/${hostname}.${var}.long.png.new",
-						"-w", "1008", @graphdef);
+				rrd_graph("${outdir}/${hostname}.${var}.long.png.work",
+						"-w", $graphWidth, @dayGraphDef);
 
-			if (RRDs::error) {
-				print "Error while graphing: " . RRDs::error . "\n";
+			if ($rrd_result) {
+				print "Error while graphing: " . $rrd_result . "\n";
 			} else {
-				copy("${outdir}/${hostname}.${var}.long.png.new", "${outdir}/${hostname}.${var}.long.png");
+				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>";
+			print HTML2 "<img src=\"${hostname}.${var}.long.png\" width=\"${width}\" height=\"${height}\"><br>\n";
+
+			$mtime=(stat("${outdir}/${hostname}.${var}.week.png.work"))[9];
+
+			my @weekGraphDef = @graphdef;
+			push @weekGraphDef, @{cfgToGraphDef($graphConfig->{'week'}, ${var}, 24)};
 
 			($averages, $width, $height) =
-				RRDs::graph("${outdir}/${hostname}.${var}.week.png.new",
-						"-w", "1008", "-e", "now", "-s", "end-1w", @graphdef);
+				rrd_graph("${outdir}/${hostname}.${var}.week.png.work",
+						"-w", "$graphWidth", "-e", "now", "-s", "00:00-8d", @weekGraphDef);
 
-			if (RRDs::error) {
-				print "Error while graphing: " . RRDs::error . "\n";
+			if ($rrd_result) {
+				print "Error while graphing: " . $rrd_result . "\n";
 			} else {
-				copy("${outdir}/${hostname}.${var}.week.png.new", "${outdir}/${hostname}.${var}.week.png");
+				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>";
+			print HTML2 "<img src=\"${hostname}.${var}.week.png\" width=\"${width}\" height=\"${height}\"><br>\n";
+
+			$mtime=(stat("${outdir}/${hostname}.${var}.year.png.work"))[9];
+
+			my @yearGraphDef = @graphdef;
+			push @yearGraphDef, @{cfgToGraphDef($graphConfig->{'year'}, ${var}, 3)};
 
 			($averages, $width, $height) =
-				RRDs::graph("${outdir}/${hostname}.${var}.year.png.new",
-						"-w", "1008", "-e", "now", "-s", "end-1y", @graphdef);
+				rrd_graph("${outdir}/${hostname}.${var}.year.png.work",
+						"-w", "$graphWidth", "-e", "00:00", "-s", "end-365d", @yearGraphDef);
 
-			if (RRDs::error) {
-				print "Error while graphing: " . RRDs::error . "\n";
+			if ($rrd_result) {
+				print "Error while graphing: " . $rrd_result . "\n";
 			} else {
-				copy("${outdir}/${hostname}.${var}.year.png.new", "${outdir}/${hostname}.${var}.year.png");
+				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 "<img src=\"${hostname}.${var}.year.png\" width=\"${width}\" height=\"${height}\"><br>\n";
 
 			print HTML2 "</body></html>\n";
 			close(HTML2);
@@ -405,4 +631,5 @@ while(1) {
 	rename("${outdir}/index.html.new", "${outdir}/index.html");
 
 	sleep(${step}/2);
+	$first = 0;
 }