284 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			Perl
		
	
	
	
	
	
			
		
		
	
	
			284 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			Perl
		
	
	
	
	
	
| package App::Netdisco::Web::Plugin::Device::Neighbors;
 | |
| 
 | |
| use Dancer ':syntax';
 | |
| use Dancer::Plugin::Ajax;
 | |
| use Dancer::Plugin::DBIC;
 | |
| use Dancer::Plugin::Auth::Extensible;
 | |
| 
 | |
| use List::Util 'first';
 | |
| use List::MoreUtils ();
 | |
| use App::Netdisco::Util::Permission 'check_acl_only';
 | |
| use App::Netdisco::Web::Plugin;
 | |
| 
 | |
| register_device_tab({ tag => 'netmap', label => 'Neighbors' });
 | |
| 
 | |
| ajax '/ajax/content/device/netmap' => require_login sub {
 | |
|     content_type('text/html');
 | |
|     template 'ajax/device/netmap.tt', {}, { layout => undef };
 | |
| };
 | |
| 
 | |
| ajax '/ajax/data/device/netmappositions' => require_login sub {
 | |
|     my $q = param('q');
 | |
|     my $qdev = schema('netdisco')->resultset('Device')
 | |
|       ->search_for_device($q) or send_error('Bad device', 400);
 | |
| 
 | |
|     my $p = param('positions') or send_error('Missing positions', 400);
 | |
|     my $positions = from_json($p) or send_error('Bad positions', 400);
 | |
|     send_error('Bad positions', 400) unless ref [] eq ref $positions;
 | |
| 
 | |
|     my $vlan = param('vlan');
 | |
|     undef $vlan if (defined $vlan and $vlan !~ m/^\d+$/);
 | |
| 
 | |
|     my $mapshow = param('mapshow');
 | |
|     return if !defined $mapshow or $mapshow !~ m/^(?:all|neighbors)$/;
 | |
| 
 | |
|     # list of groups selected by user and passed in param
 | |
|     my $hgroup = (ref [] eq ref param('hgroup') ? param('hgroup') : [param('hgroup')]);
 | |
|     # list of groups validated as real host groups and named host groups
 | |
|     my @hgrplist = List::MoreUtils::uniq
 | |
|                    grep { exists setting('host_group_displaynames')->{$_} }
 | |
|                    grep { exists setting('host_groups')->{$_} }
 | |
|                    grep { defined } @{ $hgroup };
 | |
| 
 | |
|     # list of locations selected by user and passed in param
 | |
|     my $lgroup = (ref [] eq ref param('lgroup') ? param('lgroup') : [param('lgroup')]);
 | |
|     my @lgrplist = List::MoreUtils::uniq grep { defined } @{ $lgroup };
 | |
| 
 | |
|     my %clean = ();
 | |
|     POSITION: foreach my $pos (@$positions) {
 | |
|       next unless ref {} eq ref $pos;
 | |
|       foreach my $k (qw/ID x y/) {
 | |
|         next POSITION unless exists $pos->{$k};
 | |
|         next POSITION unless $pos->{$k} =~ m/^[[:word:]\.-]+$/;
 | |
|       }
 | |
|       $clean{$pos->{ID}} = { x => $pos->{x}, y => $pos->{y} };
 | |
|     }
 | |
|     return unless scalar keys %clean;
 | |
| 
 | |
|     my $posrow = schema('netdisco')->resultset('NetmapPositions')->find({
 | |
|       device => (($mapshow eq 'neighbors') ? $qdev->ip : undef),
 | |
|       host_groups => \[ '= ?', [host_groups => [sort @hgrplist]] ],
 | |
|       locations   => \[ '= ?', [locations   => [sort @lgrplist]] ],
 | |
|       vlan => ($vlan || 0),
 | |
|     });
 | |
| 
 | |
|     if ($posrow) {
 | |
|       $posrow->update({ positions => to_json(\%clean) });
 | |
|     }
 | |
|     else {
 | |
|       schema('netdisco')->resultset('NetmapPositions')->create({
 | |
|         device => (($mapshow eq 'neighbors') ? $qdev->ip : undef),
 | |
|         host_groups => [sort @hgrplist],
 | |
|         locations   => [sort @lgrplist],
 | |
|         vlan => ($vlan || 0),
 | |
|         positions => to_json(\%clean),
 | |
|       });
 | |
|     }
 | |
| };
 | |
| 
 | |
| # copied from SNMP::Info to avoid introducing dependency to web frontend
 | |
| sub munge_highspeed {
 | |
|     my $speed = shift;
 | |
|     my $fmt   = "%d Mbps";
 | |
| 
 | |
|     if ( $speed > 9999999 ) {
 | |
|         $fmt = "%d Tbps";
 | |
|         $speed /= 1000000;
 | |
|     }
 | |
|     elsif ( $speed > 999999 ) {
 | |
|         $fmt = "%.1f Tbps";
 | |
|         $speed /= 1000000.0;
 | |
|     }
 | |
|     elsif ( $speed > 9999 ) {
 | |
|         $fmt = "%d Gbps";
 | |
|         $speed /= 1000;
 | |
|     }
 | |
|     elsif ( $speed > 999 ) {
 | |
|         $fmt = "%.1f Gbps";
 | |
|         $speed /= 1000.0;
 | |
|     }
 | |
|     return sprintf( $fmt, $speed );
 | |
| }
 | |
| 
 | |
| sub to_speed {
 | |
|   my $speed = shift or return '';
 | |
|   ($speed = munge_highspeed($speed / 1_000_000)) =~ s/(?:\.0 |bps$)//g;
 | |
|   return $speed;
 | |
| }
 | |
| 
 | |
| sub make_node_infostring {
 | |
|   my $node = shift or return '';
 | |
|   my $fmt = ('<b>%s</b> is %s <b>%s %s</b><br>running <b>%s %s</b><br>Serial: <b>%s</b><br>'
 | |
|     .'Uptime: <b>%s</b><br>Location: <b>%s</b><br>Contact: <b>%s</b>');
 | |
|   return sprintf $fmt, $node->ip,
 | |
|     ((($node->vendor || '') =~ m/^[aeiou]/i) ? 'an' : 'a'),
 | |
|     ucfirst($node->vendor || ''),
 | |
|     map {defined $_ ? $_ : ''}
 | |
|     map {$node->$_}
 | |
|         (qw/model os os_ver serial uptime_age location contact/);
 | |
| }
 | |
| 
 | |
| sub make_link_infostring {
 | |
|   my $link = shift or return '';
 | |
| 
 | |
|   my $domains = setting('domain_suffix');
 | |
|   (my $left_name = lc($link->{left_dns} || $link->{left_name} || $link->{left_ip})) =~ s/$domains//;
 | |
|   (my $right_name = lc($link->{right_dns} || $link->{right_name} || $link->{right_ip})) =~ s/$domains//;
 | |
| 
 | |
|   my @zipped = List::MoreUtils::zip6
 | |
|     @{$link->{left_port}}, @{$link->{left_descr}},
 | |
|     @{$link->{right_port}}, @{$link->{right_descr}};
 | |
| 
 | |
|   return join '<br><br>', map { sprintf '<b>%s:%s</b> (%s)<br><b>%s:%s</b> (%s)',
 | |
|     $left_name, $_->[0], ($_->[1] || 'no description'),
 | |
|     $right_name, $_->[2], ($_->[3] || 'no description') } @zipped;
 | |
| }
 | |
| 
 | |
| ajax '/ajax/data/device/netmap' => require_login sub {
 | |
|     my $q = param('q');
 | |
|     my $qdev = schema('netdisco')->resultset('Device')
 | |
|       ->search_for_device($q) or send_error('Bad device', 400);
 | |
| 
 | |
|     my $vlan = param('vlan');
 | |
|     undef $vlan if (defined $vlan and $vlan !~ m/^\d+$/);
 | |
| 
 | |
|     my $colorby = (param('colorby') || 'speed');
 | |
|     my $mapshow = (param('mapshow') || 'neighbors');
 | |
|     $mapshow = 'neighbors' if $mapshow !~ m/^(?:all|neighbors)$/;
 | |
|     $mapshow = 'all' unless $qdev->in_storage;
 | |
| 
 | |
|     # list of groups selected by user and passed in param
 | |
|     my $hgroup = (ref [] eq ref param('hgroup') ? param('hgroup') : [param('hgroup')]);
 | |
|     # list of groups validated as real host groups and named host groups
 | |
|     my @hgrplist = List::MoreUtils::uniq
 | |
|                    grep { exists setting('host_group_displaynames')->{$_} }
 | |
|                    grep { exists setting('host_groups')->{$_} }
 | |
|                    grep { defined } @{ $hgroup };
 | |
| 
 | |
|     # list of locations selected by user and passed in param
 | |
|     my $lgroup = (ref [] eq ref param('lgroup') ? param('lgroup') : [param('lgroup')]);
 | |
|     my @lgrplist = List::MoreUtils::uniq grep { defined } @{ $lgroup };
 | |
| 
 | |
|     my %ok_dev = ();
 | |
|     my %logvals = ();
 | |
|     my %metadata = ();
 | |
|     my %data = ( nodes => [], links => [] );
 | |
|     my $domains = setting('domain_suffix');
 | |
| 
 | |
|     # LINKS
 | |
| 
 | |
|     my %seen_link = ();
 | |
|     my $links = schema('netdisco')->resultset('Virtual::DeviceLinks')->search({
 | |
|       ($mapshow eq 'neighbors' ? ( -or => [
 | |
|           { left_ip  => $qdev->ip },
 | |
|           { right_ip => $qdev->ip },
 | |
|       ]) : ())
 | |
|     }, { result_class => 'DBIx::Class::ResultClass::HashRefInflator' });
 | |
| 
 | |
|     while (my $link = $links->next) {
 | |
|       # query is ordered by aggregate speed desc so we see highest speed
 | |
|       # first, which is hopefully the "best" if links are not symmetric
 | |
|       next if exists $seen_link{$link->{left_ip} ."\0". $link->{right_ip}}
 | |
|            or exists $seen_link{$link->{right_ip} ."\0". $link->{left_ip}};
 | |
| 
 | |
|       push @{$data{'links'}}, {
 | |
|         FROMID => $link->{left_ip},
 | |
|         TOID   => $link->{right_ip},
 | |
|         INFOSTRING => make_link_infostring($link),
 | |
|         SPEED  => to_speed($link->{aggspeed}),
 | |
|       };
 | |
| 
 | |
|       ++$ok_dev{$link->{left_ip}};
 | |
|       ++$ok_dev{$link->{right_ip}};
 | |
|       ++$seen_link{$link->{left_ip} ."\0". $link->{right_ip}};
 | |
|     }
 | |
| 
 | |
|     # DEVICES (NODES)
 | |
| 
 | |
|     my $posrow = schema('netdisco')->resultset('NetmapPositions')->find({
 | |
|       device => (($mapshow eq 'neighbors') ? $qdev->ip : undef),
 | |
|       host_groups => \[ '= ?', [host_groups => [sort @hgrplist]] ],
 | |
|       locations   => \[ '= ?', [locations   => [sort @lgrplist]] ],
 | |
|       vlan => ($vlan || 0),
 | |
|     });
 | |
|     my $pos_for = from_json( $posrow ? $posrow->positions : '{}' );
 | |
| 
 | |
|     my $devices = schema('netdisco')->resultset('Device')->search({}, {
 | |
|       '+select' => [\'floor(log(throughput.total))'], '+as' => ['log'],
 | |
|       join => 'throughput', distinct => 1,
 | |
|     })->with_times;
 | |
| 
 | |
|     # filter by vlan for all or neighbors only
 | |
|     if ($vlan) {
 | |
|       $devices = $devices->search(
 | |
|         { 'port_vlans_filter.vlan' => $vlan },
 | |
|         { join => 'port_vlans_filter' }
 | |
|       );
 | |
|     }
 | |
| 
 | |
|     DEVICE: while (my $device = $devices->next) {
 | |
|       # if in neighbors mode then use %ok_dev to filter
 | |
|       next DEVICE if ($device->ip ne $qdev->ip)
 | |
|         and ($mapshow eq 'neighbors')
 | |
|         and (not $ok_dev{$device->ip}); # showing only neighbors but no link
 | |
| 
 | |
|       # if location picked then filter
 | |
|       next DEVICE if ((scalar @lgrplist) and ((!defined $device->location)
 | |
|         or (0 == scalar grep {$_ eq $device->location} @lgrplist)));
 | |
| 
 | |
|       # if host groups picked then use ACLs to filter
 | |
|       my $first_hgrp =
 | |
|         first { check_acl_only($device, setting('host_groups')->{$_}) } @hgrplist;
 | |
|       next DEVICE if ((scalar @hgrplist) and (not $first_hgrp));
 | |
| 
 | |
|       # now reset first_hgroup to be the group matching the device, if any
 | |
|       $first_hgrp = first { check_acl_only($device, setting('host_groups')->{$_}) }
 | |
|                           keys %{ setting('host_group_displaynames') || {} };
 | |
| 
 | |
|       ++$logvals{ $device->get_column('log') || 1 };
 | |
|       (my $name = lc($device->dns || $device->name || $device->ip)) =~ s/$domains//;
 | |
| 
 | |
|       my %color_lkp = (
 | |
|         speed => (($device->get_column('log') || 1) * 1000),
 | |
|         hgroup => ($first_hgrp ?
 | |
|           setting('host_group_displaynames')->{$first_hgrp} : 'Other'),
 | |
|         lgroup => ($device->location || 'Other'),
 | |
|       );
 | |
| 
 | |
|       my $node = {
 | |
|         ID => $device->ip,
 | |
|         SIZEVALUE => (param('dynamicsize') ? $color_lkp{speed} : 3000),
 | |
|         ((exists $color_lkp{$colorby}) ? (COLORVALUE => $color_lkp{$colorby}) : ()),
 | |
|         LABEL => (param('showips') ? ($device->ip .' '. $name) : $name),
 | |
|         ORIG_LABEL => $name,
 | |
|         INFOSTRING => make_node_infostring($device),
 | |
|         LINK => uri_for('/device', {
 | |
|           tab => 'netmap',
 | |
|           q => $device->ip,
 | |
|           firstsearch => 'on',
 | |
|         })->path_query,
 | |
|       };
 | |
| 
 | |
|       if (exists $pos_for->{$device->ip}) {
 | |
|         $node->{'fixed'} = 1;
 | |
|         $node->{'x'} = $pos_for->{$device->ip}->{'x'};
 | |
|         $node->{'y'} = $pos_for->{$device->ip}->{'y'};
 | |
|       }
 | |
|       else {
 | |
|         ++$metadata{'newnodes'};
 | |
|       }
 | |
| 
 | |
|       push @{$data{'nodes'}}, $node;
 | |
|       $metadata{'centernode'} = $device->ip
 | |
|         if $qdev and $qdev->in_storage and $device->ip eq $qdev->ip;
 | |
|     }
 | |
| 
 | |
|     # to help get a sensible range of node sizes
 | |
|     $metadata{'numsizes'} = scalar keys %logvals;
 | |
| 
 | |
|     content_type('application/json');
 | |
|     to_json({ data => \%data, %metadata });
 | |
| };
 | |
| 
 | |
| true;
 |