1 files changed, 98 insertions, 0 deletions
diff --git a/clients/ipv6-stats-v2.pl b/clients/ipv6-stats-v2.pl
new file mode 100644
index 0000000..ab76cc9
--- /dev/null
+++ b/clients/ipv6-stats-v2.pl
@@ -0,0 +1,98 @@
+#! /usr/bin/perl
+use strict;
+use warnings;
+use lib '../include';
+use nms;
+use Data::Validate::IP qw(is_public_ipv6 is_public_ipv4 is_private_ipv4);
+use Net::MAC qw(mac_is_unicast);
+
+my $dbh = nms::db_connect();
+$dbh->{AutoCommit} = 0;
+
+while (1) {
+	my $coregws = $dbh->prepare("SELECT ip, community, sysname FROM switches WHERE switchtype <> 'dlink3100'")
+		or die "Can't prepare query: $!";
+	$coregws->execute;
+	
+	my %seen = ();
+	my $num_v4 = 0;
+	my $num_v6 = 0;
+	while (my $ref = $coregws->fetchrow_hashref) {
+		print STDERR "Querying ".$ref->{'sysname'}." ...\n";
+		my $snmp;
+		eval {
+			$snmp = nms::snmp_open_session($ref->{'ip'}, $ref->{'community'});
+		};
+		warn $@ if $@;
+		next if not $snmp;
+		
+		# Pull in old media table that does not support ipv6.
+		my $ip_phys_table = fetch($snmp, ('ipNetToMediaNetAddress', 'ipNetToMediaPhysAddress'));
+		for my $entry (values %$ip_phys_table) {
+			my $ip_addr = $entry->{'ipNetToMediaNetAddress'};
+			my $mac = Net::MAC->new(
+				mac => nms::convert_mac($entry->{'ipNetToMediaPhysAddress'}),
+				die => 0,
+			);
+		
+			next if $mac->get_base() != 16 || $mac->get_mac() eq ''; # We only support base 16 addresses
+			next if (!is_public_ipv4($ip_addr) && !is_private_ipv4($ip_addr)); # We consider RFC1918 public
+	
+			$seen{$ip_addr} = $mac->get_mac();
+			$num_v4++;
+		}
+		
+		# Pull in new media table with IPv6 support
+		$ip_phys_table = $snmp->gettable('ipNetToPhysicalTable');
+		for my $entry (values %$ip_phys_table) {
+			my $type = $entry->{'ipNetToPhysicalNetAddressType'};
+			my $ip_addr = undef;
+			my $mac = Net::MAC->new(
+				mac => nms::convert_mac($entry->{'ipNetToPhysicalPhysAddress'}),
+				die => 0,
+			);
+			
+			if ($type != 2) {
+				warn "$ip_addr is of unexpected type $type (should be 2)! Tell Berge.\n";
+				next;
+			}
+		
+			$ip_addr = nms::convert_ipv6($entry->{'ipNetToPhysicalNetAddress'});
+		
+			next if $mac->get_base() != 16 || $mac->get_mac() eq ''; # We only support base 16 addresses
+			next if not is_public_ipv6($ip_addr);
+	
+			$seen{$ip_addr} = $mac->get_mac();
+			$num_v6++;
+		}
+	
+	}
+	
+	# Populate database
+	my $i = 0;
+	foreach my $ip_addr (keys %seen) {
+		$i++;
+		my $q = $dbh->do('INSERT INTO seen_mac (address, mac) VALUES (?, ?)', undef, $ip_addr, $seen{$ip_addr})
+			or die "Can't execute query: $!";
+	}
+	
+	$dbh->commit;
+	print "Collected $num_v6 IPv6 addresses and $num_v4 IPv4 addresses. $i unique addresses.\n";
+	print "Sleeping for 60 seconds ...\n";
+	sleep(60);
+}
+
+
+# Fetch provided fields from a single table returning {iid => {tag => val}}
+sub fetch {
+	my $session = shift;
+	my @vars    = map { new SNMP::Varbind([$_]) } @_;
+	my $data    = {};
+	foreach my $result (@{$session->bulkwalk(0, 8, new SNMP::VarList(@vars))}) {
+		foreach my $entry (@$result) {
+			$data->{$entry->iid}->{$entry->tag} = $entry->val;
+		}
+	}
+	return $data;
+}
+