root/n2rrd.pl

#!@BIN_PERL@ -w

#
#    Short description: Creates/Updates RRA archives, from nagios performance data
#    Copyright (C) 2006  Badri Prasad Pillai
#
#    This program is free software; you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation; either version 2 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program; if not, write to the Free Software
#    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA

#
#    Contact details:
#               Badri(at)DigLinks.COM
#   



use strict;

use RRDs;
use Getopt::Std;

sub log_msg($);
sub usage();

my $opt = {};
my $rrderror = "";

#options
getopts("c:dhH:T:o:s:", $opt);

usage  if $opt->{h};

#
# Exit if there is no configuration file
usage  if ! $opt->{c};

#
# If can't read configuration file then just abort
#
open(CONF, "<$opt->{c}")
  or die "Can't open configuration file -> \"$opt->{c}\":$!\n";

my %config = ();
my $line   = 0;
#
# Parse configuration file
while (<CONF>) {
    chop;    # Remove newline character
    $line++;
    s/ //g;    # remove spaces
    next if /^#/;    # Skip comments
    next if /^$/;    # Skip empty lines
    s/#.*//;         # Remove partial comments

    if (/^(.*)=(.*)$/) {
        if ($1 eq "CONF_DIR") {
            $config{$1} = $2;
        }
        elsif ($1 eq "RRA_DIR") {
            $config{$1} = $2;
        }
        elsif ($1 eq "TEMPLATES_DIR") {
            $config{$1} = $2;
        }
        elsif ($1 eq "SERVICE_NAME_MAPS") {
            $config{$1} = $2;
        }
        elsif ($1 eq "LOGFILE") {
            $config{$1} = $2;
        }
        else {    # A warning should be ok
            log_msg "WARNING: Unknown global variable declaration in line $line:\"$1 = $2\"\n";
        }
    }
}
close(CONF);


#
# Sanity check: if RRA directory exists
log_msg "Direcotry rra_dir = \"$config{CONF_DIR}\" doesn't exists" 
    and exit 1
            if ! -d $config{CONF_DIR};

#
# Sanity check: if RRA directory exists
log_msg "Direcotry templates_dir = \"$config{TEMPLATES_DIR}\" doesn't exists" 
    and exit 1
            if ! -d $config{TEMPLATES_DIR};
#
# All three arguments are required
#
usage if ! $opt->{H};
usage if ! $opt->{s};
usage if ! $opt->{o};

if ( $opt->{d} ) {
        my $s = "Host = $opt->{H}, Service name = $opt->{s}, Check result = $opt->{o}\n";
        log_msg $s if $opt->{d};
}
#
# assume delimiter between service name is '_'
my @parse_service_str = split /_/, $opt->{s};
my $service = "";

if ( $#parse_service_str <= 0 ) {
   open S_MAPS,"< $config{SERVICE_NAME_MAPS}" 
                or log_msg "Can't open service maps file \"$config{SERVICE_NAME_MAPS}\"\n"
                        and exit 1;

   log_msg "Searching map in file \"$config{SERVICE_NAME_MAPS}\" for service \"$opt->{s}\" \n"
            if $opt->{d};

   while(<S_MAPS>) {
        next if /^#/;    # Skip comments
        next if /^$/;    # Skip empty lines
        s/#.*//;         # Remove partial comments
        chomp;  

        $service = $1 if /$opt->{s}:\s+(\S+)/i;
   }
}
else {
    #
    # The last word after terminator carachter "_"
        $service = $parse_service_str[$#parse_service_str];
}
#
# assume delimiter between ds_name and ds_value is '='

my @data = ();
# HOST check, works for PING and FPING output
if ( $opt->{o} =~ /PING.*?(\d+)%.+?([.\d]+)\sms/) {
        push @data, "loss=$1";
        push @data, "rta=$2";
}
else {
        #
        # remove spaces before and/or after "=" character
        $opt->{o} =~ s/\s+=/=/g;
        $opt->{o} =~ s/=\s+/=/g;
        @data = ($opt->{o} =~ /(\S+=[0-9\.]+)/g);
}

log_msg "Missing ds_name in: $opt->{o}\n" and exit 1 if ! @data;

my $ds_name  = "";
my $ds_value = "";

foreach my $kv (@data) {
        my ($key, $val) = split /=/, $kv;
        $ds_name       .= "$key:" if $key;
        $ds_value      .= "$val:";
}
$ds_name     =~ s/:$//;
$ds_value    =~ s/:$//;

my %ds_rewrite = ();

my $rrd_file =
        "$config{RRA_DIR}/$opt->{H}_${service}.rrd";

#
# Check if a rewrite rules exist
#
if ( -f "$config{TEMPLATES_DIR}/rewrite/$opt->{H}_${service}_rewrite" ) {

# 
#       print "Rewrite file detected for: $opt->{H}_${service}\n";
        open REWRITE, "<$config{TEMPLATES_DIR}/rewrite/$opt->{H}_${service}_rewrite"
                        or log_msg "Can't open rewrite rules file for $opt->{H}" 
                                and exit 1;

        my $ds_name_tmp = "";

        while(<REWRITE>) {
               next if /^#/;    # Skip comments
               next if /^$/;    # Skip empty lines
               s/#.*//;         # Remove partial comments
               chomp;   
        
               $rrd_file = $1 if /^rrd_file\s+(\S+)/;
               $ds_name_tmp .= "$2:" and $ds_rewrite{$1} = $2 if /^ds_name\s+(\S+)\s+(\S+)/;    
        }
        #print "rrd_file = $rrd_file\n";
        if ($ds_name_tmp) {
                $ds_name_tmp =~ s/:$//;
                $ds_name = $ds_name_tmp;
        }
}

unless ( -e $rrd_file ) {
        #
        # First check for template with hostname
        my $rrd_create_template = 
                "$config{TEMPLATES_DIR}/rra/$opt->{H}_${service}.t";

        if ( ! -f $rrd_create_template ) {

                #
                # Else look for default template
                $rrd_create_template = "$config{TEMPLATES_DIR}/rra/${service}.t";
                if ( ! -f $rrd_create_template ) {
                        log_msg "Missing template \"$rrd_create_template\"\n";
                        exit 1;
                }
        }

        open T, "<$rrd_create_template" 
                or log_msg "can't open file \"$rrd_create_template\""
        and exit 1;

        my @t_params = ();

        while (<T>){
                next if /^#/;    # Skip comments
                next if /^$/;    # Skip empty lines
                s/#.*//;         # Remove partial comments

                chomp;

            #
            # if there are ds_name rewrites needed
            foreach my $k (keys %ds_rewrite) {
                s/:$k:/:$ds_rewrite{$k}:/;
            }

                push @t_params, "$_";
        }
        log_msg "Creating RRD file: $rrd_file\n" if $opt->{d};
        RRDs::create(
            $rrd_file,
            @t_params
        );
        $rrderror = RRDs::error;
        log_msg "Problem while creating rrd: $rrderror\n" and exit 1 if ($rrderror);
    
            exit 0;
}

#
# If everything went well until now, then try to update rrd database

#
# If not time given 
my $time = $opt->{T} || "N";

log_msg "Filtered ds_names: $ds_name, ds_values: $ds_value\n" if $opt->{d};
RRDs::update ("$rrd_file", "--template", "$ds_name", "$time:$ds_value");
$rrderror = RRDs::error;
log_msg "ERROR while updating $rrd_file: $rrderror\n" and exit 1 if $rrderror;



exit 0;

sub log_msg ($) {
        my $str = shift;

        open L,">>$config{LOGFILE}" or die "Can't open logfile \"$config{LOGFILE}\"";

        print L "$str";

        close L;
}

sub usage () {
        print "$0 [-h] -c <path>/config-file-name -T time -H hostname -s servicename -o service_check_output\n";
        exit 0;
}