X-Git-Url: https://git.zerfleddert.de/cgi-bin/gitweb.cgi/upsgraph/blobdiff_plain/21377f4360914189ced6c9dfcf609372914aa65d..2c148a8ed56c09cb34ba4c44f5d6b84c06afb50c:/upsgraph.pl diff --git a/upsgraph.pl b/upsgraph.pl index 3d35c2a..24d196a 100755 --- a/upsgraph.pl +++ b/upsgraph.pl @@ -6,32 +6,26 @@ if ((@ARGV != 1) && (@ARGV != 2)) { } use Net::SNMP; +use IO::Socket::INET; 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 $newrrd = shift; my $field = shift; + my $vars = shift; my $start = shift; my @cmd = ("${newrrd}", "--step=${step}"); @@ -53,83 +47,77 @@ 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'}; - 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 +126,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,18 +156,78 @@ 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"; - exit 1; + 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 +239,125 @@ 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 'Status'; + print HTML ''; - $session->translate(0); + foreach my $host (@$hosts) { + print HTML "[{'name'}."\">".${host}->{'name'}."] "; + } + print HTML "
\n"; + + foreach my $host (@$hosts) { + print HTML "
\n"; + print HTML "{'name'}."\">\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 = ("-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); - 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"; - } + if (RRDs::error) { + print "Error while graphing: " . RRDs::error . "\n"; + } else { + rename("${outdir}/${hostname}.${var}.png.new", "${outdir}/${hostname}.${var}.png"); + } - open(HTML, ">${outdir}/index.html.new"); + print HTML "
\n"; - print HTML 'USV status'; - print HTML ''; + open (HTML2, ">${outdir}/${hostname}.${var}.html.new"); + print HTML2 "" . $vars->{$var}->{'name'} . ""; + print HTML2 ''; - 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); - if (RRDs::error) { - print "Error while graphing: " . RRDs::error . "\n"; - } else { - rename("${outdir}/${var}.png.new", "${outdir}/${var}.png"); - } + push @graphdef, "VDEF:min=${var},MINIMUM"; + push @graphdef, "GPRINT:min:Minimum\\: %.2lf"; - print HTML ""; + push @graphdef, "VDEF:avg=${var},AVERAGE"; + push @graphdef, "GPRINT:avg:Average\\: %.2lf"; - open (HTML2, ">${outdir}/${var}.html.new"); - print HTML2 "" . $vars->{$var}->{'name'} . ""; - print HTML2 ''; + push @graphdef, "VDEF:max=${var},MAXIMUM"; + push @graphdef, "GPRINT:max:Maximum\\: %.2lf"; - ($averages, $width, $height) = - RRDs::graph("${outdir}/${var}.long.png.new", - "-w", "1008", @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}.long.png.new", "${outdir}/${var}.long.png"); - } + ($averages, $width, $height) = + RRDs::graph("${outdir}/${hostname}.${var}.long.png.new", + "-w", "1008", @graphdef); - print HTML2 "
"; + if (RRDs::error) { + print "Error while graphing: " . RRDs::error . "\n"; + } else { + rename("${outdir}/${hostname}.${var}.long.png.new", "${outdir}/${hostname}.${var}.long.png"); + } - ($averages, $width, $height) = - RRDs::graph("${outdir}/${var}.week.png.new", - "-w", "1008", "-e", "now", "-s", "end-1w", @graphdef); + print HTML2 "
"; - if (RRDs::error) { - print "Error while graphing: " . RRDs::error . "\n"; - } else { - rename("${outdir}/${var}.week.png.new", "${outdir}/${var}.week.png"); - } + ($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}/${hostname}.${var}.week.png.new", "${outdir}/${hostname}.${var}.week.png"); + } - print HTML2 "
"; + print HTML2 "
"; - ($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}.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"); - } + 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 "
"; + print HTML2 "
"; - print HTML2 "\n"; - close(HTML2); - rename("${outdir}/${var}.html.new", "${outdir}/${var}.html"); + print HTML2 "\n"; + close(HTML2); + rename("${outdir}/${hostname}.${var}.html.new", "${outdir}/${hostname}.${var}.html"); + } } print HTML "\n";