380 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			Perl
		
	
	
	
	
	
			
		
		
	
	
			380 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			Perl
		
	
	
	
	
	
package App::Netdisco::Worker::Plugin::Discover::Neighbors;
 | 
						|
 | 
						|
use Dancer ':syntax';
 | 
						|
use App::Netdisco::Worker::Plugin;
 | 
						|
use aliased 'App::Netdisco::Worker::Status';
 | 
						|
 | 
						|
use App::Netdisco::Transport::SNMP ();
 | 
						|
use App::Netdisco::Util::Device qw/get_device is_discoverable/;
 | 
						|
use App::Netdisco::Util::Permission 'check_acl_no';
 | 
						|
use App::Netdisco::JobQueue 'jq_insert';
 | 
						|
use Dancer::Plugin::DBIC 'schema';
 | 
						|
use List::MoreUtils ();
 | 
						|
use NetAddr::IP::Lite ();
 | 
						|
use NetAddr::MAC;
 | 
						|
use Encode;
 | 
						|
use Try::Tiny;
 | 
						|
 | 
						|
=head2 discover_new_neighbors( )
 | 
						|
 | 
						|
Given a Device database object, and a working SNMP connection, discover and
 | 
						|
store the device's port neighbors information.
 | 
						|
 | 
						|
Entries in the Topology database table will override any discovered device
 | 
						|
port relationships.
 | 
						|
 | 
						|
The Device database object can be a fresh L<DBIx::Class::Row> object which is
 | 
						|
not yet stored to the database.
 | 
						|
 | 
						|
Any discovered neighbor unknown to Netdisco will have a C<discover> job
 | 
						|
immediately queued (subject to the filtering by the C<discover_*> settings).
 | 
						|
 | 
						|
=cut
 | 
						|
 | 
						|
register_worker({ phase => 'main', driver => 'snmp' }, sub {
 | 
						|
  my ($job, $workerconf) = @_;
 | 
						|
 | 
						|
  my $device = $job->device;
 | 
						|
  return unless $device->in_storage;
 | 
						|
  my $snmp = App::Netdisco::Transport::SNMP->reader_for($device)
 | 
						|
    or return Status->defer("discover failed: could not SNMP connect to $device");
 | 
						|
 | 
						|
  my @to_discover = store_neighbors($device);
 | 
						|
  my (%seen_id, %seen_ip) = ((), ());
 | 
						|
 | 
						|
  # only enqueue if device is not already discovered,
 | 
						|
  # discover_* config permits the discovery
 | 
						|
  foreach my $neighbor (@to_discover) {
 | 
						|
      my ($ip, $remote_id) = @$neighbor;
 | 
						|
      if ($seen_ip{ $ip }++) {
 | 
						|
          debug sprintf
 | 
						|
            ' queue - skip: IP %s is already queued from %s',
 | 
						|
            $ip, $device->ip;
 | 
						|
          next;
 | 
						|
      }
 | 
						|
 | 
						|
      if ($remote_id and $seen_id{ $remote_id }++) {
 | 
						|
          debug sprintf
 | 
						|
            ' queue - skip: %s with ID [%s] already queued from %s',
 | 
						|
            $ip, $remote_id, $device->ip;
 | 
						|
          next;
 | 
						|
      }
 | 
						|
 | 
						|
      my $newdev = get_device($ip);
 | 
						|
      next if $newdev->in_storage;
 | 
						|
 | 
						|
      # risk of things going wrong...?
 | 
						|
      # https://quickview.cloudapps.cisco.com/quickview/bug/CSCur12254
 | 
						|
 | 
						|
      jq_insert({
 | 
						|
        device => $ip,
 | 
						|
        action => 'discover',
 | 
						|
        subaction => 'with-nodes',
 | 
						|
        ($remote_id ? (device_key => $remote_id) : ()),
 | 
						|
      });
 | 
						|
 | 
						|
      vars->{'queued'}->{$ip} = true;
 | 
						|
      debug sprintf ' [%s] queue - queued %s for discovery (ID: [%s])',
 | 
						|
        $device, $ip, ($remote_id || '');
 | 
						|
  }
 | 
						|
 | 
						|
  return Status->info(sprintf ' [%s] neigh - processed %s neighbors',
 | 
						|
       $device->ip, scalar @to_discover);
 | 
						|
});
 | 
						|
 | 
						|
=head2 store_neighbors( $device )
 | 
						|
 | 
						|
returns: C<@to_discover>
 | 
						|
 | 
						|
Given a Device database object, and a working SNMP connection, discover and
 | 
						|
store the device's port neighbors information.
 | 
						|
 | 
						|
Entries in the Topology database table will override any discovered device
 | 
						|
port relationships.
 | 
						|
 | 
						|
The Device database object can be a fresh L<DBIx::Class::Row> object which is
 | 
						|
not yet stored to the database.
 | 
						|
 | 
						|
A list of discovered neighbors will be returned as [C<$ip>, C<$type>] tuples.
 | 
						|
 | 
						|
=cut
 | 
						|
 | 
						|
sub store_neighbors {
 | 
						|
  my $device = shift;
 | 
						|
  my @to_discover = ();
 | 
						|
 | 
						|
  my $snmp = App::Netdisco::Transport::SNMP->reader_for($device)
 | 
						|
    or return (); # already checked!
 | 
						|
 | 
						|
  # first allow any manually configured topology to be set
 | 
						|
  # and do this before we cache the rows in vars->{'device_ports'}
 | 
						|
  set_manual_topology($device);
 | 
						|
 | 
						|
  if (!defined $snmp->has_topo) {
 | 
						|
      debug sprintf ' [%s] neigh - neighbor protocols are not enabled', $device->ip;
 | 
						|
      return @to_discover;
 | 
						|
  }
 | 
						|
 | 
						|
  my $interfaces = $snmp->interfaces;
 | 
						|
  my $c_if       = $snmp->c_if;
 | 
						|
  my $c_port     = $snmp->c_port;
 | 
						|
  my $c_id       = $snmp->c_id;
 | 
						|
  my $c_platform = $snmp->c_platform;
 | 
						|
  my $c_cap      = $snmp->c_cap;
 | 
						|
 | 
						|
  # cache the device ports to save hitting the database for many single rows
 | 
						|
  vars->{'device_ports'} =
 | 
						|
    { map {($_->port => $_)} $device->ports->reset->all };
 | 
						|
  my $device_ports = vars->{'device_ports'};
 | 
						|
 | 
						|
  # v4 and v6 neighbor tables
 | 
						|
  my $c_ip = ($snmp->c_ip || {});
 | 
						|
  my %c_ipv6 = %{ ($snmp->can('hasLLDP') and $snmp->hasLLDP)
 | 
						|
    ? ($snmp->lldp_ipv6 || {}) : {} };
 | 
						|
 | 
						|
  # remove keys with undef values, as c_ip does
 | 
						|
  delete @c_ipv6{ grep { not defined $c_ipv6{$_} } keys %c_ipv6 };
 | 
						|
  # now combine them, v6 wins
 | 
						|
  $c_ip = { %$c_ip, %c_ipv6 };
 | 
						|
 | 
						|
  foreach my $entry (sort (List::MoreUtils::uniq( keys %$c_ip ))) {
 | 
						|
      if (!defined $c_if->{$entry} or !defined $interfaces->{ $c_if->{$entry} }) {
 | 
						|
          debug sprintf ' [%s] neigh - port for IID:%s not resolved, skipping',
 | 
						|
            $device->ip, $entry;
 | 
						|
          next;
 | 
						|
      }
 | 
						|
 | 
						|
      # WRT #475 this is SAFE because we check against known ports below
 | 
						|
      my $port = $interfaces->{ $c_if->{$entry} } or next;
 | 
						|
      my $portrow = $device_ports->{$port};
 | 
						|
 | 
						|
      if (!defined $portrow) {
 | 
						|
          debug sprintf ' [%s] neigh - local port %s already skipped, ignoring',
 | 
						|
            $device->ip, $port;
 | 
						|
          next;
 | 
						|
      }
 | 
						|
 | 
						|
      if (ref $c_ip->{$entry}) {
 | 
						|
          debug sprintf ' [%s] neigh - port %s has multiple neighbors - skipping',
 | 
						|
            $device->ip, $port;
 | 
						|
          next;
 | 
						|
      }
 | 
						|
 | 
						|
      if ($portrow->manual_topo) {
 | 
						|
          debug sprintf ' [%s] neigh - %s has manually defined topology',
 | 
						|
            $device->ip, $port;
 | 
						|
          next;
 | 
						|
      }
 | 
						|
 | 
						|
      my $remote_ip   = $c_ip->{$entry};
 | 
						|
      my $remote_port = undef;
 | 
						|
      my $remote_type = Encode::decode('UTF-8', $c_platform->{$entry} || '');
 | 
						|
      my $remote_id   = Encode::decode('UTF-8', $c_id->{$entry});
 | 
						|
      my $remote_cap  = $c_cap->{$entry} || [];
 | 
						|
 | 
						|
      next unless $remote_ip;
 | 
						|
      my $r_netaddr = NetAddr::IP::Lite->new($remote_ip);
 | 
						|
 | 
						|
      if ($r_netaddr and ($r_netaddr->addr ne $remote_ip)) {
 | 
						|
        debug sprintf ' [%s] neigh - IP on %s: using %s as canonical form of %s',
 | 
						|
          $device->ip, $port, $r_netaddr->addr, $remote_ip;
 | 
						|
        $remote_ip = $r_netaddr->addr;
 | 
						|
      }
 | 
						|
 | 
						|
      # a bunch of heuristics to search known devices if we don't have a
 | 
						|
      # useable remote IP...
 | 
						|
 | 
						|
      if ((! $r_netaddr) or ($remote_ip eq '0.0.0.0') or
 | 
						|
        check_acl_no($remote_ip, 'group:__LOCAL_ADDRESSES__')) {
 | 
						|
 | 
						|
          if ($remote_id) {
 | 
						|
              my $devices = schema('netdisco')->resultset('Device');
 | 
						|
 | 
						|
              debug sprintf
 | 
						|
                ' [%s] neigh - bad address %s on port %s, searching for %s instead',
 | 
						|
                $device->ip, $remote_ip, $port, $remote_id;
 | 
						|
              my $neigh_rs = $devices->search_rs({name => $remote_id});
 | 
						|
              my $neigh = ($neigh_rs->count == 1 ? $neigh_rs->first : undef);
 | 
						|
 | 
						|
              if (!defined $neigh and $neigh_rs->count) {
 | 
						|
                  debug sprintf ' [%s] neigh - multiple devices claim to be %s (port %s) - skipping',
 | 
						|
                    $device->ip, $remote_id, $port;
 | 
						|
                  next;
 | 
						|
              }
 | 
						|
 | 
						|
              if (!defined $neigh) {
 | 
						|
                  my $mac = NetAddr::MAC->new(mac => ($remote_id || ''));
 | 
						|
                  if ($mac and not $mac->errstr) {
 | 
						|
                      $neigh = $devices->single({mac => $mac->as_ieee});
 | 
						|
                  }
 | 
						|
              }
 | 
						|
 | 
						|
              # some HP switches send 127.0.0.1 as remote_ip if no ip address
 | 
						|
              # on default vlan for HP switches remote_ip looks like
 | 
						|
              # "myswitchname(012345-012345)"
 | 
						|
              if (!defined $neigh) {
 | 
						|
                  (my $tmpid = $remote_id) =~ s/.*\(([0-9a-f]{6})-([0-9a-f]{6})\).*/$1$2/;
 | 
						|
                  my $mac = NetAddr::MAC->new(mac => ($tmpid || ''));
 | 
						|
                  if ($mac and not $mac->errstr) {
 | 
						|
                      debug sprintf
 | 
						|
                        ' [%s] neigh - trying to find neighbor %s by MAC %s',
 | 
						|
                        $device->ip, $remote_id, $mac->as_ieee;
 | 
						|
                      $neigh = $devices->single({mac => $mac->as_ieee});
 | 
						|
                  }
 | 
						|
              }
 | 
						|
 | 
						|
              if (!defined $neigh) {
 | 
						|
                  (my $shortid = $remote_id) =~ s/\..*//;
 | 
						|
                  $neigh = $devices->single({name => { -ilike => "${shortid}%" }});
 | 
						|
              }
 | 
						|
 | 
						|
              if ($neigh) {
 | 
						|
                  $remote_ip = $neigh->ip;
 | 
						|
                  debug sprintf ' [%s] neigh - found %s with IP %s',
 | 
						|
                    $device->ip, $remote_id, $remote_ip;
 | 
						|
              }
 | 
						|
              else {
 | 
						|
                  debug sprintf ' [%s] neigh - could not find %s, skipping',
 | 
						|
                    $device->ip, $remote_id;
 | 
						|
                  next;
 | 
						|
              }
 | 
						|
          }
 | 
						|
          else {
 | 
						|
              debug sprintf ' [%s] neigh - skipping unuseable address %s on port %s',
 | 
						|
                $device->ip, $remote_ip, $port;
 | 
						|
              next;
 | 
						|
          }
 | 
						|
      }
 | 
						|
 | 
						|
      # what we came here to do.... discover the neighbor
 | 
						|
      debug sprintf ' [%s] neigh - %s with ID [%s] on %s',
 | 
						|
        $device->ip, $remote_ip, ($remote_id || ''), $port;
 | 
						|
 | 
						|
      if (is_discoverable($remote_ip, $remote_type, $remote_cap)) {
 | 
						|
          push @to_discover, [$remote_ip, $remote_id];
 | 
						|
      }
 | 
						|
      else {
 | 
						|
          debug sprintf
 | 
						|
            ' [%s] neigh - skip: %s of type [%s] excluded by discover_* config',
 | 
						|
            $device->ip, $remote_ip, ($remote_type || '');
 | 
						|
      }
 | 
						|
 | 
						|
      $remote_port = $c_port->{$entry};
 | 
						|
      if (defined $remote_port) {
 | 
						|
          # clean weird characters
 | 
						|
          $remote_port =~ s/[^\d\s\/\.,"()\w:-]+//gi;
 | 
						|
      }
 | 
						|
      else {
 | 
						|
          debug sprintf ' [%s] neigh - no remote port found for port %s at %s',
 | 
						|
            $device->ip, $port, $remote_ip;
 | 
						|
      }
 | 
						|
 | 
						|
      $portrow = $portrow->update({
 | 
						|
          remote_ip   => $remote_ip,
 | 
						|
          remote_port => $remote_port,
 | 
						|
          remote_type => $remote_type,
 | 
						|
          remote_id   => $remote_id,
 | 
						|
          is_uplink   => \"true",
 | 
						|
          manual_topo => \"false",
 | 
						|
      })->discard_changes();
 | 
						|
 | 
						|
      # update master of our aggregate to be a neighbor of
 | 
						|
      # the master on our peer device (a lot of iffs to get there...).
 | 
						|
      # & cannot use ->neighbor prefetch because this is the port insert!
 | 
						|
      if (defined $portrow->slave_of) {
 | 
						|
 | 
						|
          my $peer_device = get_device($remote_ip);
 | 
						|
          my $master = schema('netdisco')->resultset('DevicePort')->single({
 | 
						|
            ip => $device->ip,
 | 
						|
            port => $portrow->slave_of
 | 
						|
          });
 | 
						|
 | 
						|
          if ($peer_device and $peer_device->in_storage and $master
 | 
						|
              and not ($portrow->is_master or defined $master->slave_of)) {
 | 
						|
 | 
						|
              my $peer_port = schema('netdisco')->resultset('DevicePort')->single({
 | 
						|
                ip   => $peer_device->ip,
 | 
						|
                port => $portrow->remote_port,
 | 
						|
              });
 | 
						|
 | 
						|
              $master->update({
 | 
						|
                  remote_ip => ($peer_device->ip || $remote_ip),
 | 
						|
                  remote_port => ($peer_port ? $peer_port->slave_of : undef ),
 | 
						|
                  is_uplink => \"true",
 | 
						|
                  is_master => \"true",
 | 
						|
                  manual_topo => \"false",
 | 
						|
              });
 | 
						|
          }
 | 
						|
      }
 | 
						|
  }
 | 
						|
 | 
						|
  return @to_discover;
 | 
						|
}
 | 
						|
 | 
						|
# take data from the topology table and update remote_ip and remote_port
 | 
						|
# in the devices table. only use root_ips and skip any bad topo entries.
 | 
						|
sub set_manual_topology {
 | 
						|
  my $device = shift;
 | 
						|
  my $snmp = App::Netdisco::Transport::SNMP->reader_for($device) or return;
 | 
						|
 | 
						|
  schema('netdisco')->txn_do(sub {
 | 
						|
    # clear manual topology flags
 | 
						|
    schema('netdisco')->resultset('DevicePort')
 | 
						|
      ->search({ip => $device->ip})->update({manual_topo => \'false'});
 | 
						|
 | 
						|
    # clear outdated manual topology links
 | 
						|
    my $old_links = schema('netdisco')->resultset('Topology')->search({
 | 
						|
      -or => [
 | 
						|
        { dev1 => $device->ip,
 | 
						|
          port1 => { '-not_in' => $device->ports->get_column('port')->as_query } },
 | 
						|
        { dev2 => $device->ip,
 | 
						|
          port2 => { '-not_in' => $device->ports->get_column('port')->as_query } },
 | 
						|
      ],
 | 
						|
    })->delete;
 | 
						|
    debug sprintf ' [%s] neigh - removed %d outdated manual topology links',
 | 
						|
      $device->ip, $old_links;
 | 
						|
 | 
						|
    my $topo_links = schema('netdisco')->resultset('Topology')
 | 
						|
      ->search({-or => [dev1 => $device->ip, dev2 => $device->ip]});
 | 
						|
    debug sprintf ' [%s] neigh - setting manual topology links', $device->ip;
 | 
						|
 | 
						|
    while (my $link = $topo_links->next) {
 | 
						|
        # could fail for broken topo, but we ignore to try the rest
 | 
						|
        try {
 | 
						|
            schema('netdisco')->txn_do(sub {
 | 
						|
              # only work on root_ips
 | 
						|
              my $left  = get_device($link->dev1);
 | 
						|
              my $right = get_device($link->dev2);
 | 
						|
 | 
						|
              # skip bad entries
 | 
						|
              return unless ($left->in_storage and $right->in_storage);
 | 
						|
 | 
						|
              $left->ports
 | 
						|
                ->single({port => $link->port1})
 | 
						|
                ->update({
 | 
						|
                  remote_ip => $right->ip,
 | 
						|
                  remote_port => $link->port2,
 | 
						|
                  remote_type => undef,
 | 
						|
                  remote_id   => undef,
 | 
						|
                  is_uplink   => \"true",
 | 
						|
                  manual_topo => \"true",
 | 
						|
                });
 | 
						|
 | 
						|
              $right->ports
 | 
						|
                ->single({port => $link->port2})
 | 
						|
                ->update({
 | 
						|
                  remote_ip => $left->ip,
 | 
						|
                  remote_port => $link->port1,
 | 
						|
                  remote_type => undef,
 | 
						|
                  remote_id   => undef,
 | 
						|
                  is_uplink   => \"true",
 | 
						|
                  manual_topo => \"true",
 | 
						|
                });
 | 
						|
            });
 | 
						|
        };
 | 
						|
    }
 | 
						|
  });
 | 
						|
}
 | 
						|
 | 
						|
true;
 |