111 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			Perl
		
	
	
	
	
	
			
		
		
	
	
			111 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			Perl
		
	
	
	
	
	
package App::Netdisco::Worker::Plugin::Discover::Wireless;
 | 
						|
 | 
						|
use Dancer ':syntax';
 | 
						|
use App::Netdisco::Worker::Plugin;
 | 
						|
use aliased 'App::Netdisco::Worker::Status';
 | 
						|
 | 
						|
use App::Netdisco::Transport::SNMP ();
 | 
						|
use Dancer::Plugin::DBIC 'schema';
 | 
						|
 | 
						|
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 $ssidlist = $snmp->i_ssidlist;
 | 
						|
  return unless scalar keys %$ssidlist;
 | 
						|
 | 
						|
  # cache the device ports to save hitting the database for many single rows
 | 
						|
  my $device_ports = vars->{'device_ports'}
 | 
						|
    || { map {($_->port => $_)} $device->ports->all };
 | 
						|
 | 
						|
  my $interfaces = $snmp->interfaces;
 | 
						|
  my $ssidbcast  = $snmp->i_ssidbcast;
 | 
						|
  my $ssidmac    = $snmp->i_ssidmac;
 | 
						|
  my $channel    = $snmp->i_80211channel;
 | 
						|
  my $power      = $snmp->dot11_cur_tx_pwr_mw;
 | 
						|
 | 
						|
  # build device ssid list suitable for DBIC
 | 
						|
  my (%ssidseen, @ssids);
 | 
						|
  foreach my $entry (keys %$ssidlist) {
 | 
						|
      (my $iid = $entry) =~ s/\.\d+$//;
 | 
						|
      my $port = $interfaces->{$iid};
 | 
						|
 | 
						|
      if (not $port) {
 | 
						|
          debug sprintf ' [%s] wireless - ignoring %s (no port mapping)',
 | 
						|
            $device->ip, $iid;
 | 
						|
          next;
 | 
						|
      }
 | 
						|
 | 
						|
      if (!defined $device_ports->{$port}) {
 | 
						|
          debug sprintf ' [%s] wireless - local port %s already skipped, ignoring',
 | 
						|
            $device->ip, $port;
 | 
						|
          next;
 | 
						|
      }
 | 
						|
 | 
						|
      next unless $ssidmac->{$entry};
 | 
						|
 | 
						|
      if (exists $ssidseen{$port}{ $ssidmac->{$entry} }) {
 | 
						|
          debug sprintf ' [%s] wireless - duplicate bssid %s on port %s',
 | 
						|
            $device->ip, $ssidmac->{$entry}, $port;
 | 
						|
          next;
 | 
						|
      }
 | 
						|
      ++$ssidseen{$port}{ $ssidmac->{$entry} };
 | 
						|
 | 
						|
      push @ssids, {
 | 
						|
          port      => $port,
 | 
						|
          ssid      => $ssidlist->{$entry},
 | 
						|
          broadcast => $ssidbcast->{$entry},
 | 
						|
          bssid     => $ssidmac->{$entry},
 | 
						|
      };
 | 
						|
  }
 | 
						|
 | 
						|
  schema('netdisco')->txn_do(sub {
 | 
						|
    my $gone = $device->ssids->delete;
 | 
						|
    debug sprintf ' [%s] wireless - removed %d SSIDs',
 | 
						|
      $device->ip, $gone;
 | 
						|
    $device->ssids->populate(\@ssids);
 | 
						|
    debug sprintf ' [%s] wireless - added %d new SSIDs',
 | 
						|
      $device->ip, scalar @ssids;
 | 
						|
  });
 | 
						|
 | 
						|
  # build device channel list suitable for DBIC
 | 
						|
  my @channels;
 | 
						|
  foreach my $entry (keys %$channel) {
 | 
						|
      my $port = $interfaces->{$entry};
 | 
						|
 | 
						|
      if (not $port) {
 | 
						|
          debug sprintf ' [%s] wireless - ignoring %s (no port mapping)',
 | 
						|
            $device->ip, $entry;
 | 
						|
          next;
 | 
						|
      }
 | 
						|
 | 
						|
      if (!defined $device_ports->{$port}) {
 | 
						|
          debug sprintf ' [%s] wireless - local port %s already skipped, ignoring',
 | 
						|
            $device->ip, $port;
 | 
						|
          next;
 | 
						|
      }
 | 
						|
 | 
						|
      push @channels, {
 | 
						|
          port    => $port,
 | 
						|
          channel => $channel->{$entry},
 | 
						|
          power   => $power->{$entry},
 | 
						|
      };
 | 
						|
  }
 | 
						|
 | 
						|
  schema('netdisco')->txn_do(sub {
 | 
						|
    my $gone = $device->wireless_ports->delete;
 | 
						|
    debug sprintf ' [%s] wireless - removed %d wireless channels',
 | 
						|
      $device->ip, $gone;
 | 
						|
    $device->wireless_ports->populate(\@channels);
 | 
						|
 | 
						|
    return Status->info(sprintf ' [%s] wireless - added %d new wireless channels',
 | 
						|
      $device->ip, scalar @channels);
 | 
						|
  });
 | 
						|
});
 | 
						|
 | 
						|
true;
 |