Add a common utility function to create an SNMP session, that also supports SNMPv3.

1 files changed, 34 insertions, 1 deletions

diff --git a/include/nms.pm b/include/nms.pm
index 9567bde..047c243 100644
--- a/include/nms.pm
+++ b/include/nms.pm
@@ -4,10 +4,10 @@ use warnings;
 use DBI;
 use Net::Telnet;
 use Data::Dumper;
+use Net::SNMP;
 use FileHandle;
 package nms;
 
-
 use base 'Exporter';
 our @EXPORT = qw(switch_disconnect switch_connect switch_exec switch_timeout db_connect);
 
@@ -101,4 +101,37 @@ sub switch_disconnect {
 	$conn->close;
 }
 
+sub snmp_open_session {
+	my ($ip, $community) = @_;
+
+	my $domain = ($ip =~ /:/) ? 'udp6' : 'udp4';
+	my $version;
+	my %options = (
+		-hostname => $ip,
+		-domain => $domain,
+	);
+
+	if ($community =~ /^snmpv3:(.*)$/) {
+		my ($username, $authprotocol, $authpassword, $privprotocol, $privpassword) = split /\//, $1;
+
+		$options{'-username'} = $username;
+		$options{'-authprotocol'} = $authprotocol;
+		$options{'-authpassword'} = $authpassword;
+
+		if (defined($privprotocol) && defined($privpassword)) {
+			$options{'-privprotocol'} = $privprotocol;
+			$options{'-privpassword'} = $privpassword;
+		}
+
+		$options{'-version'} = 3;
+	} else {
+		$options{'-version'} = 2;
+	}
+
+	my ($session, $error) = Net::SNMP->session(%options);
+	die "SNMP session failed: " . $error if (!defined($session));
+
+	return $session;
+}
+
 1;