]> cvs.zerfleddert.de Git - upsgraph/blobdiff - upsgraph.pl
current value
[upsgraph] / upsgraph.pl
index 3d35c2a1059304bad881cfa1a5d7c01f070804da..100c52b1f38c1a2ec2274ede6c4b6000cff672d2 100755 (executable)
@@ -6,32 +6,27 @@ if ((@ARGV != 1) && (@ARGV != 2)) {
 }
 
 use Net::SNMP;
+use IO::Socket::INET;
 use RRDs;
+use File::Copy;
 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 $newrrd = shift;
        my $field = shift;
+       my $vars = shift;
        my $start = shift;
 
        my @cmd = ("${newrrd}", "--step=${step}");
@@ -53,83 +48,82 @@ sub rrdcreate(@) {
        }
 }
 
-if ($> == 0) {
-       if (@ARGV != 2) {
-               print STDERR "Running as root, please provide UID as 2th argument!\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";
        }
 
-       print "Running as root, switching to ".$ARGV[1]."\n";
-       $< = $> = $ARGV[1];
-}
+       $session->translate(0);
 
-if (-e "${rrdfile}") {
-       print "Reading old ${rrdfile} to preserve data...\n";
+       my $result = $session->get_request($oid);
 
-       my $rrdinfo = RRDs::info("${rrdfile}");
-       if (RRDs::error) {
-               print "Error while getting info: " . RRDs::error . "\n";
-               exit 1;
-       }
+       $session->close;
 
-       (my $start, my $ostep, my $names, my $data) =
-               RRDs::fetch("${rrdfile}",
-                               "-s " . (time() - ($rrdinfo->{'rra[0].rows'} * $rrdinfo->{'step'})),
-                               "AVERAGE");
+       return undef if (!defined($result));
 
-       if (RRDs::error) {
-               print "Error while fetching data: " . RRDs::error . "\n";
-               exit 1;
-       }
+       $result->{$oid};
+}
 
-       foreach my $field (@$names) {
-               if (! -e "${rrdfile}.${field}") {
-                       rrdcreate("${rrdfile}.${field}","${field}",(${start}-${ostep}));
-               }
-       }
+sub fetch_tcp(@) {
+       my $address = shift;
+       my $port = shift;
 
-       my $pos = $start;
-       foreach my $line (@$data) {
-               foreach my $field (@$names) {
-                       my $val = shift (@$line);
-                       next if (!defined($val));
+       my $sock = IO::Socket::INET->new(PeerAddr => $address,
+                       PeerPort => $port,
+                       Proto => 'tcp',
+                       Timeout => 1);
 
-                       RRDs::update("${rrdfile}.${field}", "${pos}:${val}");
-                       if (RRDs::error) {
-                               print "Can't insert data: " . RRDs::error . "\n";
-                               exit 1;
-                       }
+       return undef if (!$sock);
 
-               }
+       chomp(my $value = <$sock>);
 
-               $pos += $ostep;
+       close($sock);
 
-               if ((($pos-$start)/$ostep) == $#$data) {
-                       last;
-               }
+       if (!$value) {
+               return undef;
        }
 
-       rename("${rrdfile}", "${rrdfile}.old") or die "Can't rename old file: $!\n";
+       $value=~ s/\s//g;
+
+       $value;
 }
 
-foreach my $field (@fields) {
-       if (! -e "${rrdfile}.${field}") {
-               print "Creating ${rrdfile}.${field}...\n";
-               rrdcreate("${rrdfile}.${field}","${field}");
+if ($> == 0) {
+       if (@ARGV != 2) {
+               print STDERR "Running as root, please provide UID as 2th argument!\n";
+               exit(1);
        }
 
-       my $rrdinfo = RRDs::info("${rrdfile}.${field}");
-       if (RRDs::error) {
-               print "Error while getting info: " . RRDs::error . "\n";
-               exit 1;
+       print "Running as root, switching to ".$ARGV[1]."\n";
+       $< = $> = $ARGV[1];
+}
+
+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'}));
        }
 
-       if ($rrdinfo->{'rra[0].rows'} != $keep) {
-               print "Resizing ${rrdfile}.${field} from " . $rrdinfo->{'rra[0].rows'} .
-                       " to ${keep} samples.\n";
+       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}.${field}",
+                       RRDs::fetch("${rrdfile}",
                                        "-s " . (time() - ($rrdinfo->{'rra[0].rows'} * $rrdinfo->{'step'})),
                                        "AVERAGE");
 
@@ -138,24 +132,29 @@ foreach my $field (@fields) {
                        exit 1;
                }
 
-               rrdcreate("${rrdfile}.${field}.new", "${field}", (${start}-${ostep}));
-
-               print "Preserving data since " . localtime($start) . "\n";
+               foreach my $field (@$names) {
+                       if (! -e "${rrdfile}.${field}") {
+                               rrdcreate("${rrdfile}.${field}",
+                                       "${field}",
+                                       $host->{'vars'},
+                                       (${start}-${ostep}));
+                       }
+               }
 
                my $pos = $start;
                foreach my $line (@$data) {
-                       my $vline = "${pos}";
-
-                       foreach my $val (@$line) {
+                       foreach my $field (@$names) {
+                               my $val = shift (@$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;
+                               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) {
@@ -163,19 +162,109 @@ foreach my $field (@fields) {
                        }
                }
 
-               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";
+               rename("${rrdfile}", "${rrdfile}.old") or die "Can't rename old 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}.${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 "Failed!\n";
+               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";
+
+                       $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;
+                       }
+               }
        }
 }
 
@@ -186,99 +275,153 @@ 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>Status</title></head>';
+       print HTML '<body bgcolor="#ffffff">';
 
-       $session->translate(0);
+       foreach my $host (@$hosts) {
+               print HTML "[<a href=\"#".${host}->{'name'}."\">".${host}->{'name'}."</a>]&nbsp;";
+       }
+       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'});
+                       }
 
-       foreach my $var (@fields) {
-               delete $vars->{$var}->{'value'};
+                       next unless (defined $result);
 
-               my $result = $session->get_request($vars->{$var}->{'oid'});
-               next unless (defined $result);
+                       $vars->{$var}->{'value'} = $result;
+                       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'};
+               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";
                }
-       }
 
-       $session->close;
+               foreach my $var (@{$host->{'fields'}}) {
+                       my @graphdef = ('-P', "--lazy", "-t", $hostname." - ".$vars->{$var}->{'name'}, "DEF:${var}=${rrdfile}.${var}:${var}:AVERAGE", "LINE1:${var}#FF0000");
 
-       foreach my $var (@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";
-       }
+                       push @graphdef, "VDEF:cur=${var},LAST";
+                       push @graphdef, 'GPRINT:cur:Current\\: <span foreground="#FF0000">%.2lf</span>\\r';
 
-       open(HTML, ">${outdir}/index.html.new");
+                       my $mtime;
+                       $mtime=(stat("${outdir}/${hostname}.${var}.png.work"))[9];
 
-       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">';
+                       (my $averages, my $width, my $height) =
+                               RRDs::graph("${outdir}/${hostname}.${var}.png.work",
+                                               "-w", "720", @graphdef);
 
-       foreach my $var (@fields) {
-               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);
+                       pop @graphdef;
+                       pop @graphdef;
 
-               if (RRDs::error) {
-                       print "Error while graphing: " . RRDs::error . "\n";
-               } else {
-                       rename("${outdir}/${var}.png.new", "${outdir}/${var}.png");
-               }
+                       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=\"${var}.html\"><img src=\"${var}.png\" width=\"${width}\" height=\"${height}\" border=\"0\"></a>";
+                       print HTML "<a href=\"${hostname}.${var}.html\"><img src=\"${hostname}.${var}.png\" width=\"${width}\" height=\"${height}\" border=\"0\"></a><br>\n";
 
-               open (HTML2, ">${outdir}/${var}.html.new");
-               print HTML2 "<html><head><title>" . $vars->{$var}->{'name'} . "</title></head>";
-               print HTML2 '<body bgcolor="#ffffff">';
+                       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">';
 
-               ($averages, $width, $height) =
-                       RRDs::graph("${outdir}/${var}.long.png.new",
-                       "-w", "1008", @graphdef);
+                       push @graphdef, "VDEF:min=${var},MINIMUM";
+                       push @graphdef, "GPRINT:min:Minimum\\: %.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:avg=${var},AVERAGE";
+                       push @graphdef, "GPRINT:avg:Average\\: %.2lf";
 
-               print HTML2 "<img src=\"${var}.long.png\" width=\"${width}\" height=\"${height}\"><br>";
+                       push @graphdef, "VDEF:max=${var},MAXIMUM";
+                       push @graphdef, "GPRINT:max:Maximum\\: %.2lf";
 
-               ($averages, $width, $height) =
-                       RRDs::graph("${outdir}/${var}.week.png.new",
-                       "-w", "1008", "-e", "now", "-s", "end-1w", @graphdef);
+                       push @graphdef, "VDEF:cur=${var},LAST";
+                       push @graphdef, "GPRINT:cur:Current\\: %.2lf";
 
-               if (RRDs::error) {
-                       print "Error while graphing: " . RRDs::error . "\n";
-               } else {
-                       rename("${outdir}/${var}.week.png.new", "${outdir}/${var}.week.png");
-               }
+                       $mtime=(stat("${outdir}/${hostname}.${var}.long.png.work"))[9];
+                       ($averages, $width, $height) =
+                               RRDs::graph("${outdir}/${hostname}.${var}.long.png.work",
+                                               "-w", "1008", @graphdef);
 
-               print HTML2 "<img src=\"${var}.week.png\" width=\"${width}\" height=\"${height}\"><br>";
+                       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");
+                               }
+                       }
 
-               ($averages, $width, $height) =
-                       RRDs::graph("${outdir}/${var}.year.png.new",
-                       "-w", "1008", "-e", "now", "-s", "end-1y", @graphdef);
+                       print HTML2 "<img src=\"${hostname}.${var}.long.png\" width=\"${width}\" height=\"${height}\"><br>";
 
-               if (RRDs::error) {
-                       print "Error while graphing: " . RRDs::error . "\n";
-               } else {
-                       rename("${outdir}/${var}.year.png.new", "${outdir}/${var}.year.png");
-               }
+                       $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);
 
-               print HTML2 "<img src=\"${var}.year.png\" width=\"${width}\" height=\"${height}\"><br>";
+                       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 "</body></html>\n";
-               close(HTML2);
-               rename("${outdir}/${var}.html.new", "${outdir}/${var}.html");
+                       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";
Impressum, Datenschutz