root/indexmrtg.cgi @ 9c83a251
| 00588b29 | Hamish Coleman | #!/usr/bin/perl
|
||
#
|
||||
| 9c83a251 | Hamish Coleman | # Generate a machine readable index from mrtg.cfg inputs
|
||
| 00588b29 | Hamish Coleman | #
|
||
our $VERSION=0.5;
|
||||
use strict;
|
||||
use warnings;
|
||||
use FileHandle;
|
||||
use Data::Dumper;
|
||||
$Data::Dumper::Indent = 1;
|
||||
$Data::Dumper::Sortkeys = 1;
|
||||
use CGI ':all';
|
||||
use CGI::Carp qw(fatalsToBrowser);
|
||||
use Sys::Hostname;
|
||||
use HTML::Entities();
|
||||
# List of config files to search for
|
||||
my @config_files = (
|
||||
'/etc/mrtg.cfg',
|
||||
'/usr/share/WWW/mrtg/mrtg.cfg',
|
||||
);
|
||||
# FIXME - globals
|
||||
my $gifdone = 0; # Scan for newest graph and save info for later
|
||||
my $warnings;
|
||||
my $db = {}; # storage for all tests found
|
||||
| 9c83a251 | Hamish Coleman | $db->{mrtg}{default}{post}{xsize} = 500;
|
||
$db->{mrtg}{default}{post}{ysize} = 135;
|
||||
| 00588b29 | Hamish Coleman | |||
sub config_save_val($$$$) {
|
||||
my ($filename,$target,$var,$val) = @_;
|
||||
my $this_ysize;
|
||||
my $this_xsize;
|
||||
my $this_title;
|
||||
# HACK!
|
||||
if ( $var eq 'ysize' ) {
|
||||
$val += 35;
|
||||
} elsif ( $var eq 'xsize' ) {
|
||||
$val += 100;
|
||||
}
|
||||
# Save any default initializers
|
||||
if ($target eq '_') {
|
||||
| 9c83a251 | Hamish Coleman | $db->{mrtg}{default}{post}{$var} = $val;
|
||
| 00588b29 | Hamish Coleman | return;
|
||
}
|
||||
if ($target eq '^') {
|
||||
| 9c83a251 | Hamish Coleman | $db->{mrtg}{default}{pre}{$var} = $val;
|
||
| 00588b29 | Hamish Coleman | return;
|
||
}
|
||||
# Check for target name collision, and uniqueify
|
||||
my $unique_target = $target;
|
||||
my $unique_id = 1;
|
||||
while (defined $db->{target}->{$unique_target}
|
||||
&& $db->{target}->{$unique_target}->{_cfgfile} ne $filename) {
|
||||
$unique_target = $target . '_' . $unique_id;
|
||||
$unique_id++;
|
||||
}
|
||||
if (!defined $db->{target}->{$unique_target}) {
|
||||
# initialize a new target
|
||||
#TODO - instantiate new tests from all of the "_"/"^" defaults
|
||||
push @{$db->{sequence}}, $unique_target;
|
||||
$db->{target}->{$unique_target}->{_target} = $target;
|
||||
$db->{target}->{$unique_target}->{_cfgfile} = $filename;
|
||||
$db->{target}->{$unique_target}->{_sequence} = @{$db->{sequence}};
|
||||
| 9c83a251 | Hamish Coleman | $db->{target}{$unique_target}{xsize} = $db->{mrtg}{default}{post}{xsize};
|
||
$db->{target}{$unique_target}{ysize} = $db->{mrtg}{default}{post}{ysize};
|
||||
| 00588b29 | Hamish Coleman | |||
}
|
||||
| 9c83a251 | Hamish Coleman | $db->{target}{$unique_target}{mrtg}{$var} = $val;
|
||
| 00588b29 | Hamish Coleman | }
|
||
# Load one or more files
|
||||
#
|
||||
sub config_load_file(@) {
|
||||
while (@_) {
|
||||
my $filename = shift;
|
||||
my $fh = new FileHandle $filename,"r";
|
||||
if (!defined $fh) {
|
||||
# FIXME globals
|
||||
$warnings .= "Warning: Could not open $filename $!\n";
|
||||
return;
|
||||
}
|
||||
#TODO - multi-line values
|
||||
while(<$fh>){
|
||||
# Is this a standard MRTG setting?
|
||||
if ( $_ =~ /^([^#[][^[]+)\[(.*)\]:\s*(.+)$/ ) {
|
||||
my $var=lc $1;
|
||||
my $target=lc $2;
|
||||
my $val=$3;
|
||||
config_save_val($filename,$target,$var,$val);
|
||||
}
|
||||
# TODO - Handle extra statements:
|
||||
# Include
|
||||
# WorkDir
|
||||
# LogDir
|
||||
# ##HC Magic for baseurl
|
||||
}
|
||||
}
|
||||
}
|
||||
# Itterate through the loaded targets and read their mtime and
|
||||
# any other filesystem details for them.
|
||||
#
|
||||
sub config_read_filesystem() {
|
||||
# check and update details for all known tests;
|
||||
for my $i (keys %{$db->{target}}) {
|
||||
my $target = $db->{target}->{$i};
|
||||
my $testname = $target->{_target};
|
||||
# TODO - check workdir and validate it here
|
||||
# Default is GIF, unless a PNG file exists
|
||||
$target->{_imagetype} = 'gif';
|
||||
if ( -e "$testname-day.png" ) {
|
||||
$target->{_imagetype} = "png";
|
||||
}
|
||||
# TODO - change the filename based on the displayed period
|
||||
my $filename = $testname . "-day." . $target->{_imagetype};
|
||||
my ($dev, $ino, $mode, $nlink, $uid, $gid, $rdev, $size,
|
||||
$atime,$mtime,$ctime,$blksize,$blocks)
|
||||
= stat($filename);
|
||||
if (!defined $mtime) {
|
||||
# if we cannot stat the file, skip it
|
||||
next;
|
||||
}
|
||||
$target->{_mtime} = $mtime;
|
||||
# FIXME - global
|
||||
# find the newest file
|
||||
if ($mtime > $gifdone) {
|
||||
$gifdone = $mtime;
|
||||
}
|
||||
}
|
||||
}
|
||||
sub extract_grouphosttest($) {
|
||||
my ($name) = @_;
|
||||
# Do not classify a test with no separators in it
|
||||
if ($name !~ /,/) {
|
||||
return undef;
|
||||
}
|
||||
my ($group,$host,$test) = split ',', $name;
|
||||
# TODO cope with missing group/host/test parts.
|
||||
if (!defined $group || !defined $host || !defined $test) {
|
||||
return undef;
|
||||
}
|
||||
return ($group,$host,$test);
|
||||
}
|
||||
# Read through the list of targets and find common hostname prefixes,
|
||||
# so that we can group these hosts together
|
||||
# (a hosttype entry)
|
||||
#
|
||||
sub maketree_find_hosttypes() {
|
||||
for my $name (keys %{$db->{target}}) {
|
||||
my ($group,$host,$test) = extract_grouphosttest($name);
|
||||
if (!defined($group)) {
|
||||
next;
|
||||
}
|
||||
my $hosttype;
|
||||
# TODO - this needs to have a better way...
|
||||
# hosttype is the non digit hostname prefix - thus
|
||||
# "proxy-1" and "proxy-2" would both end up in the
|
||||
# "proxy-" hosttype
|
||||
if ($host =~ m/^(.*[^0-9])(\d+)$/) {
|
||||
$hosttype = $1.'*';
|
||||
} else {
|
||||
next;
|
||||
}
|
||||
$db->{target}->{$name}->{_hosttype} = $hosttype;
|
||||
$db->{hosttype}->{$hosttype}->{$host}++;
|
||||
}
|
||||
for my $hosttype (keys %{$db->{hosttype}}) {
|
||||
if ((keys %{$db->{hosttype}->{$hosttype}})<2) {
|
||||
delete $db->{hosttype}->{$hosttype};
|
||||
}
|
||||
}
|
||||
}
|
||||
sub maketree_grouphosttest($$) {
|
||||
my ($name,$sequence) = @_;
|
||||
my ($group,$host,$test) = extract_grouphosttest($name);
|
||||
if (!defined($group)) {
|
||||
# TODO cope with missing group/host/test parts.
|
||||
# for now, just assume it is not a group/host/test
|
||||
$db->{tree}->{other}->{$name} = $sequence;
|
||||
return;
|
||||
}
|
||||
# a valid hosttype both:
|
||||
# defined($db->{target}->{$name}->{hosttype}) &&
|
||||
# defined($db->{hosttype}->{$hosttype})
|
||||
my $hosttype = $db->{target}->{$name}->{_hosttype};
|
||||
if (defined($hosttype) && !defined($db->{hosttype}->{$hosttype})) {
|
||||
$hosttype=undef;
|
||||
}
|
||||
# TODO - determine a better description for this tree leaf
|
||||
my $testtreename = 'Graph Type';
|
||||
if (defined $hosttype) {
|
||||
$db->{tree}->{group}->{$group}->{$hosttype}->{$host}->{$test}->{$name} = $sequence;
|
||||
$db->{tree}->{group}->{$group}->{$hosttype}->{$testtreename}->{$test}->{$name} = $sequence;
|
||||
$db->{tree}->{group}->{ALL}->{$hosttype}->{$host}->{$test}->{$name} = $sequence;
|
||||
$db->{tree}->{group}->{ALL}->{$testtreename}->{$test}->{$hosttype}->{$host}->{$name} = $sequence;
|
||||
$db->{tree}->{group}->{$group}->{$testtreename}->{$test}->{$hosttype}->{$host}->{$name} = $sequence;
|
||||
} else {
|
||||
$db->{tree}->{group}->{$group}->{$host}->{$test}->{$name} = $sequence;
|
||||
$db->{tree}->{group}->{ALL}->{$host}->{$test}->{$name} = $sequence;
|
||||
$db->{tree}->{group}->{ALL}->{$testtreename}->{$test}->{$host}->{$name} = $sequence;
|
||||
$db->{tree}->{group}->{$group}->{$testtreename}->{$test}->{$host}->{$name} = $sequence;
|
||||
}
|
||||
}
|
||||
sub maketree_cfgfile($$) {
|
||||
my ($name,$sequence) = @_;
|
||||
# config files
|
||||
my $cfgfile = $db->{target}->{$name}->{_cfgfile};
|
||||
my @cfgpath = split /\//,$cfgfile;
|
||||
shift @cfgpath; # remove initial slash
|
||||
unshift @cfgpath,"config"; # ensure that the config node exists
|
||||
my $node = $db->{tree};
|
||||
while (@cfgpath ) {
|
||||
if (!defined $node->{$cfgpath[0]}) {
|
||||
$node->{$cfgpath[0]} = {};
|
||||
}
|
||||
$node = $node->{$cfgpath[0]};
|
||||
shift @cfgpath;
|
||||
}
|
||||
# assign a target name and sequence to the final node
|
||||
$node->{$name} = $sequence;
|
||||
}
|
||||
# Look through the target database and create tree entries for each cfgfile
|
||||
#
|
||||
sub maketree() {
|
||||
maketree_find_hosttypes();
|
||||
for my $name (keys %{$db->{target}}) {
|
||||
my $sequence = $db->{target}->{$name}->{_sequence};
|
||||
maketree_cfgfile($name,$sequence);
|
||||
maketree_grouphosttest($name,$sequence);
|
||||
# TODO - parse the Target setting and create trees
|
||||
# TODO - parse magic comments and create trees here
|
||||
}
|
||||
# TODO - look for nodes that have only one child and that child is
|
||||
# childless and prune the child
|
||||
}
|
||||
| 9c83a251 | Hamish Coleman | my $this_host = hostname;
|
||
| 00588b29 | Hamish Coleman | |||
| 9c83a251 | Hamish Coleman | print header,
|
||
start_html(-TITLE=>"$this_host indexmrtg (V$VERSION)"),
|
||||
"\n";
|
||||
| 00588b29 | Hamish Coleman | |||
| 9c83a251 | Hamish Coleman | config_load_file(@config_files);
|
||
config_read_filesystem();
|
||||
| 00588b29 | Hamish Coleman | |||
| 9c83a251 | Hamish Coleman | # Time the next update to occur a little while after the next interval completes
|
||
my $interval = 300; # 5 min update interval
|
||||
my $guardband = 15; # updates occur this many seconds after predicted gif completion
|
||||
my $refresh = $interval + $guardband + $gifdone - time; # predict how long until next update
|
||||
$refresh = $interval if $refresh <= $guardband;
|
||||
my $expires = gmtime (time + $interval * 2 + $guardband);
|
||||
| 00588b29 | Hamish Coleman | |||
| 9c83a251 | Hamish Coleman | print
|
||
"<meta http-equiv=\"expires\" content=\"$expires GMT\">\n",
|
||||
"<meta http-equiv=\"refresh\" content=$refresh>\n";
|
||||
| 00588b29 | Hamish Coleman | |||
| 9c83a251 | Hamish Coleman | print end_html;
|
||
| 00588b29 | Hamish Coleman | |||
if (param('debug')) {
|
||||
| 9c83a251 | Hamish Coleman | print "\n",Dumper(\$db),"\n";
|
||
| 00588b29 | Hamish Coleman | }
|
||