File Coverage

blib/lib/SNMPMonitor.pm

Criterion	Covered	Total	%
statement	10	12	83.3
branch			n/a
condition			n/a
subroutine	4	4	100.0
pod			n/a
total	14	16	87.5

package SNMPMonitor;
our $VERSION = '0.10';

use 5.008008;
use common::sense;

use Module::Pluggable require => 1;

use NetSNMP::OID     (':all');
use NetSNMP::agent     (':all');
use NetSNMP::ASN     (':all');

sub new {
    my $class = shift;
    my $self = {
        running     => '1',
        debugging   => '0',
        subagent    => '0',
        valid_plugins  => '',
        plugin      => '',
        agent       => new NetSNMP::agent('dont_init_agent' => 1,'dont_init_lib' => 1),
        oid         => {},
    };

    bless $self, $class;
    $self->set_accessors;

    $self->load_plugins;
    $self->check_oid;
    $self->register;
    return $self;
}

# Create accessor and mutator methods in the symbol table.
sub set_accessors {
    my $self = shift;

    # create accessor methods for defined parameters
    for my $datum (keys %{$self}) {
        no strict "refs";

        *$datum = sub {
            my $self = shift; # Don't ignore calling class/object
            $self->{$datum} = shift if @_;
            return $self->{$datum};
        };
    }
}

# finds and attempts to load and initialize plugins
sub load_plugins {
    my $self = shift;
    my $FH = shift;

    my @plugins = $self->plugins;
    my $valid_plugins = ();
    foreach (@plugins) {
        my ($initialized, $error) = $self->evaluate_plugin($_);
        push @{$valid_plugins}, $initialized if $initialized;
        print $FH "Error in Plugin: $_\n$@\n" unless $initialized;
    }
    $self->plugin($valid_plugins);
}

# Evaluates plugins for validity,
# Returns initialized object if valid,
# Returns the error if invalid.
sub evaluate_plugin {
    my $self = shift;
    my $plugin = shift;

#    print STDERR "evaluating, " . $plugin . "\n";
    my $new_object = eval { $plugin->new; };
    return $new_object, $@;
}

# Registers All Plugins with the SNMP Agent
sub register {
    my  $self = shift;
    my $FH = shift || 'STDERR';

    foreach (@{$self->plugin}) {
        print $FH "@ Registering: " . $_->name . "\n";
        $self->print_info($_);
        $self->register_plugin($_);
    }
}

# Actually Registers a Plugin with the SNMP Agtent.
sub register_plugin {
    my $self = shift;
    my $plugin = shift;

    $self->agent->register(
        $plugin->name,
        $plugin->full_oid,
        sub {
            $plugin->set_monitor(@_)
        },
    );

}

sub dump_plugins {
    my $self = shift;
    foreach (@{$self->plugin}) {
        $self->print_info($_);
    }
}


#print some info about the plugins
sub print_info {
    my $self = shift;
    my $plugin = shift;
    my $FH = shift || 'STDERR';

    print $FH "--> Name: "      . $plugin->name . "\n";
    print $FH "--> Full Name: " . $plugin->full_name . "\n";
    print $FH "--> Name: "      . $plugin->name . "\n";
    print $FH "--> OID: "       . $plugin->plugin_oid . "\n";
#    print $FH "Root OID: "     . $plugin->root_oid . "\n";
    print $FH "--> Full OID: "  . $plugin->full_oid . "\n\n";
}

sub check_oid {
    my $self = shift;

    INCREMENT: foreach my $plugin (@{$self->plugin}) {

        #print STDOUT $self->oid;
        if ($self->oid->{$plugin->plugin_oid}) {
            $self->increment_oid($plugin);
            redo INCREMENT;
        }
        else {
            $self->oid->{$plugin->plugin_oid} = $plugin->name;
        }
    }
}

sub increment_oid {
    my $self = shift;
    my $plugin = shift;
    my $FH = shift || 'STDOUT' ;

    my @oid = split/\./, $plugin->plugin_oid;
    $oid[-1] += 1;
    my $new_oid = join '.', @oid;
    print $FH "! Created new OID: $new_oid for " . $plugin->name . "\n";

    $plugin->plugin_oid($new_oid);
    $plugin->reset_oid;
}

1;
__END__