1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
|
#!/usr/bin/perl
use strict;
use warnings;
use DBI;
use POSIX;
use Time::HiRes;
use SNMP;
use Data::Dumper;
use lib '../include';
use nms;
our $dbh = nms::db_connect();
$dbh->{AutoCommit} = 0;
$dbh->{RaiseError} = 1;
#my $qualification = 'sysname LIKE \'e71%\'';
my $qualification = <<"EOF";
(last_updated IS NULL OR now() - last_updated > poll_frequency)
AND (locked='f' OR now() - last_updated > '5 minutes'::interval)
AND ip is not null
EOF
# Borrowed from snmpfetch.pl
our $qswitch = $dbh->prepare(<<"EOF")
SELECT
*,
DATE_TRUNC('second', now() - last_updated - poll_frequency) AS overdue
FROM
switches
NATURAL LEFT JOIN switchtypes
WHERE
$qualification
ORDER BY
priority DESC,
overdue DESC
LIMIT 10
FOR UPDATE OF switches
EOF
or die "Couldn't prepare qswitch";
our $qlock = $dbh->prepare("UPDATE switches SET locked='t', last_updated=now() WHERE switch=?")
or die "Couldn't prepare qlock";
our $qunlock = $dbh->prepare("UPDATE switches SET locked='f', last_updated=now() WHERE switch=?")
or die "Couldn't prepare qunlock";
my @switches = ();
our $temppoll = $dbh->prepare("INSERT INTO switch_temp (switch,temp,time) VALUES((select switch from switches where sysname = ?),?,now())")
or die "Couldn't prepare temppoll";
sub mylog
{
my $msg = shift;
my $time = POSIX::ctime(time);
$time =~ s/\n.*$//;
printf STDERR "[%s] %s\n", $time, $msg;
}
sub populate_switches
{
@switches = ();
$qswitch->execute()
or die "Couldn't get switch";
while (my $ref = $qswitch->fetchrow_hashref()) {
push @switches, {
'sysname' => $ref->{'sysname'},
'id' => $ref->{'switch'},
'mgtip' => $ref->{'ip'},
'community' => $ref->{'community'}
};
}
$dbh->commit;
}
sub inner_loop
{
mylog("Starting run");
populate_switches();
for my $refswitch (@switches) {
my %switch = %{$refswitch};
mylog( "START: Polling $switch{'sysname'} ($switch{'mgtip'}) ");
$switch{'start'} = time;
$qlock->execute($switch{'id'})
or die "Couldn't lock switch";
$dbh->commit;
my $s = new SNMP::Session(DestHost => $switch{'mgtip'},
Community => $switch{'community'},
Version => '2');
my @vars = ();
push @vars, [ "sysName", 0];
push @vars, [ "sysDescr", 0];
push @vars, [ "1.3.6.1.4.1.2636.3.1.13.1.7.7.1.0", 0];
my $varlist = SNMP::VarList->new(@vars);
$s->get($varlist, [ \&ckcall, \%switch ]);
$s->gettable('ifXTable',callback => [\&callback, \%switch]);
}
mylog( "Added " . @switches . " ");
SNMP::MainLoop(5);
}
sub ckcall
{
my %switch = %{$_[0]};
my $vars = $_[1];
my ($sysname,$sysdescr,$temp) = (undef,undef,undef);
for my $var (@$vars) {
if ($var->[0] eq "sysName") {
$sysname = $var->[2];
} elsif ($var->[0] eq "sysDescr") {
$sysdescr = $var->[2];
} elsif ($var->[0] eq "enterprises.2636.3.1.13.1.7.7.1.0.0") {
$temp = $var->[2];
}
}
if (defined $temp && $temp =~ /^\d+$/) {
$temppoll->execute($switch{'sysname'},$temp);
} else {
warn "Couldn't read temp for " . $switch{'sysname'} . ", got " . (defined $temp ? $temp : "undef");
}
$dbh->commit;
}
my @values = ('ifName','ifHighSpeed','ifHCOutOctets','ifHCInOctets');
my $query = "INSERT INTO polls2 (switch,time";
foreach my $val (@values) {
$query .= ",$val";
}
$query .= ") VALUES(?,timeofday()::timestamp";
foreach my $val (@values) {
$query .= ",?";
}
$query .= ");";
our $qpoll = $dbh->prepare($query)
or die "Couldn't prepare qpoll";
sub callback
{
my %switch = %{$_[0]};
my $table = $_[1];
my %ifs = ();
foreach my $key (keys %{$table}) {
my $descr = $table->{$key}->{'ifName'};
if ($descr =~ m/(ge|xe|et)-/ && $descr !~ m/\./) {
$ifs{$descr} = $table->{$key};
}
}
foreach my $key (keys %ifs) {
my @vals = ();
foreach my $val (@values) {
if (!defined($ifs{$key}{$val})) {
die "Missing data";
}
push @vals, $ifs{$key}{$val};
}
$qpoll->execute($switch{'id'},@vals) || die "ops";
}
mylog( "STOP: Polling $switch{'sysname'} took " . (time - $switch{'start'}) . "s");
$qunlock->execute($switch{'id'})
or die "Couldn't unlock switch";
$dbh->commit;
}
print $query . "\n";
while (1) {
inner_loop();
}
|
