use Fcntl qw(:flock);
use IO::File;
use File::Slurp qw(slurp);
+use Date::Format;
+use XML::Simple;
+use FS::svc_port;
+use FS::Record qw(qsearch);
+use Torrus::ConfigTree;
#$DEBUG = 0;
#$me = '[FS::NetworkMonitoringSystem::Torrus_Internal]';
our $lockfile = '/usr/local/etc/torrus/discovery/FSLOCK';
our $ddxfile = '/usr/local/etc/torrus/discovery/routers.ddx';
+sub new {
+ my $class = shift;
+ my $self = {};
+ bless $self, $class;
+ return $self;
+}
+
+sub ddx2hash {
+ my $self = shift;
+ my $ddx_xml = slurp($ddxfile);
+ my $xs = new XML::Simple(RootName=> undef, SuppressEmpty => '',
+ ForceArray => 1, );
+ return $xs->XMLin($ddx_xml);
+}
+
+sub get_router_serviceids {
+ my $self = shift;
+ my $router = shift;
+ my $find_serviceid = shift;
+ my $found_serviceid = 0;
+ my $ddx_hash = $self->ddx2hash;
+ if($ddx_hash->{host}){
+ my @hosts = @{$ddx_hash->{host}};
+ foreach my $host ( @hosts ) {
+ my $param = $host->{param};
+ if($param && $param->{'snmp-host'}
+ && (!$router || $param->{'snmp-host'}->{'value'} eq $router)
+ && $param->{'RFC2863_IF_MIB::external-serviceid'}) {
+ my $serviceids = $param->{'RFC2863_IF_MIB::external-serviceid'}->{'content'};
+ my %hash = ();
+ if($serviceids) {
+ my @serviceids = split(',',$serviceids);
+ foreach my $serviceid ( @serviceids ) {
+ $serviceid =~ s/^\s+|\s+$//g;
+ my @s = split(':',$serviceid);
+ next unless scalar(@s) == 4;
+ $hash{$s[1]} = $s[0] if $router;
+ if ($find_serviceid && $find_serviceid eq $s[0]) {
+ $hash{$param->{'snmp-host'}->{'value'}} = $s[1];
+ $found_serviceid = 1;
+ }
+ }
+ }
+ return \%hash if ($router || $found_serviceid);
+ }
+ }
+ }
+ '';
+}
+
+sub port_graphs_link {
+ # hardcoded for 'main' tree for now
+ my $self = shift;
+ my $serviceid = shift;
+ my $hash = $self->get_router_serviceids(undef,$serviceid);
+ my @keys = keys %$hash; # yeah this is weird...
+ my $host = $keys[0];
+ my $iface = $hash->{$keys[0]};
+ my $config_tree = new Torrus::ConfigTree( -TreeName => 'main' );
+ my $token = $config_tree->token("/Routers/$host/Interface_Counters/$iface/InOut_bps");
+ return $Torrus::Freeside::FSURL."/torrus/main?token=$token";
+}
+
+sub find_svc {
+ my $self = shift;
+ my $serviceid = shift;
+ return '' unless $serviceid =~ /^[0-9A-Za-z_\-.\\\/ ]+$/;
+
+ my @svc_port = qsearch('svc_port', { 'serviceid' => $serviceid });
+ return '' unless scalar(@svc_port);
+
+ # for now it's like this, later on just change to qsearchs
+
+ return $svc_port[0];
+}
+
+sub report {
+ my $self = shift;
+
+ my @ls = localtime(time);
+ my ($d,$m,$y) = ($ls[3], $ls[4]+1, $ls[5]+1900);
+ if ( $ls[3] == 1 ) {
+ $m--;
+ if ($m == 0) { $m=12; $y-- }
+ #i should have better error checking
+ system('torrus', 'report', '--report=MonthlyUsage', "--date=$y-$m-01");
+ system('torrus', 'report', '--genhtml', '--tree=main');
+ }
+
+}
+
sub add_router {
my($self, $ip) = @_;
next unless $line =~ /^\s*<param\s+name="snmp-host"\s+value="$router_ip"\/?>/i;
while ( my $hostline = shift(@ddx) ) {
- $new .= "$hostline\n";
+ $new .= "$hostline\n" unless $hostline =~ /^\s+<\/host>\s*/i;
if ( $hostline =~ /^\s*<param name="RFC2863_IF_MIB::external-serviceid"\/?>/i ) {
while ( my $paramline = shift(@ddx) ) {
- if ( $paramline =~ /^\s*</param>/ ) {
+ if ( $paramline =~ /^\s*<\/param>/ ) {
$new .= "$newline\n$paramline";
last; #paramline
} else {
$new .=
qq( <param name="RFC2863_IF_MIB::external-serviceid">\n).
qq( $newline\n").
- qq( </param>\n).
+ qq( </param>\n);
}
$new .= $hostline;
last; #hostline
}
sub _torrus_lock {
- $lock = new IO:::File ">>$lockfile" or die $!;
+ $lock = new IO::File ">>$lockfile" or die $!;
flock($lock, LOCK_EX);
}
or die "can't write to $ddxfile.new: $!";
print $new $ddx;
close $new;
- rename("$ddxfile", $ddxfile.`date +%Y%m%d%H%M%S`) or die $!;
+
+ # `date ...` created file names with weird chars in them
+ my $tmpname = $ddxfile . Date::Format::time2str('%Y%m%d%H%M%S',time);
+ rename("$ddxfile", $tmpname) or die $!;
rename("$ddxfile.new", $ddxfile) or die $!;
$self->_torrus_reload;