9 require 'SNMP_Session.pm';
14 my $password = 'removed';
17 my $dbh = nms::db_connect();
18 $dbh->{AutoCommit} = 0;
20 # normal mode: fetch switches from the database
21 # instant mode: poll the switches specified on the command line
22 my $instant = (defined($ARGV[0]));
25 $qualification = "sysname=?";
27 $qualification = <<"EOF";
28 (last_updated IS NULL OR now() - last_updated > poll_frequency)
29 AND (locked='f' OR now() - last_updated > '15 minutes'::interval)
33 my $qswitch = $dbh->prepare(<<"EOF")
36 DATE_TRUNC('second', now() - last_updated - poll_frequency) AS overdue
39 NATURAL LEFT JOIN switchtypes
45 FOR UPDATE OF switches
47 or die "Couldn't prepare qswitch";
48 my $qlock = $dbh->prepare("UPDATE switches SET locked='t', last_updated=now() WHERE switch=?")
49 or die "Couldn't prepare qlock";
50 my $qunlock = $dbh->prepare("UPDATE switches SET locked='f', last_updated=now() WHERE switch=?")
51 or die "Couldn't prepare qunlock";
52 my $qpoll = $dbh->prepare("INSERT INTO polls (time, switch, port, bytes_in, bytes_out) VALUES (timeofday()::timestamp,?,?,?,?)")
53 or die "Couldn't prepare qpoll";
54 my $qtemppoll = $dbh->prepare("INSERT INTO temppoll (time, switch, temp) VALUES (timeofday()::timestamp,?::text::int,?::text::float)")
55 or die "Couldn't prepare qtemppoll";
56 my $qcpupoll = $dbh->prepare("INSERT INTO cpuloadpoll (time, switch, entity, value) VALUES (timeofday()::timestamp,?::text::int,?,?)")
57 or die "Couldn't prepare qtemppoll";
62 $sysname = shift @ARGV;
63 exit if (!defined($sysname));
64 $qswitch->execute($sysname)
65 or die "Couldn't get switch";
67 # Find a switch to grab
69 or die "Couldn't get switch";
71 my $switch = $qswitch->fetchrow_hashref();
73 if (!defined($switch)) {
77 mylog("No such switch $sysname available, quitting.");
80 mylog("No available switches in pool, sleeping.");
86 $qlock->execute($switch->{'switch'})
87 or die "Couldn't lock switch";
90 if ($switch->{'locked'}) {
91 mylog("WARNING: Lock timed out on $switch->{'ip'}, breaking lock");
95 if (defined($switch->{'overdue'})) {
96 $msg = sprintf "Polling ports %s on %s (%s), %s overdue.",
97 $switch->{'ports'}, $switch->{'ip'}, $switch->{'sysname'}, $switch->{'overdue'};
99 $msg = sprintf "Polling ports %s on %s (%s), never polled before.",
100 $switch->{'ports'}, $switch->{'ip'}, $switch->{'sysname'};
104 my $ip = $switch->{'ip'};
106 if ($ip eq '127.0.0.1') {
107 mylog("Polling disabled for this switch, skipping.");
108 $qunlock->execute($switch->{'switch'})
109 or die "Couldn't unlock switch";
114 my $community = $switch->{'community'};
115 my $start = [Time::HiRes::gettimeofday];
118 if ($switch->{'wide_counters'}) {
119 $session = SNMPv2c_Session->open($ip, $community, 161)
120 or die "Couldn't talk to switch";
122 $session = SNMP_Session->open($ip, $community, 161)
123 or die "Couldn't talk to switch";
125 my @ports = expand_ports($switch->{'ports'});
127 for my $port (@ports) {
128 my $in = fetch_data($session, $port, 0, $switch->{'wide_counters'});
129 my $out = fetch_data($session, $port, 1, $switch->{'wide_counters'});
131 $qpoll->execute($switch->{'switch'}, $port, $in, $out);
133 my $conn = switch_connect($ip);
134 if (!defined($conn)) {
135 print "Could not connect to switch ".$switch->{'switch'}."\n";
136 } elsif ($switch->{'switchtype'} eq 'es3024') {
137 my @data = switch_exec('sys monitor status', $conn);
138 my @fields = split(/\s+/, $data[2]);
139 # The temp fields are 6, 7, 8
140 print "$fields[7] + $fields[8] + $fields[9]\n";
141 my $avgtemp = ($fields[7] + $fields[8] + $fields[9]) / 3;
142 print $avgtemp." avgtemp\n";
143 $qtemppoll->execute($switch->{'switch'},
144 $avgtemp) or die "Could not exec qtemppoll";
145 } elsif ($switch->{'switchtype'} eq 'cisco6509') {
148 my $oid = BER::encode_oid(1, 3, 6, 1, 4, 1, 9, 9, 109, 1, 1, 1, 1, 2, $i);
149 my $entity = fetch_snmp($session, $oid);
151 next if (!defined($entity));
153 # last-minute load for the given entity
154 $oid = BER::encode_oid(1, 3, 6, 1, 4, 1, 9, 9, 109, 1, 1, 1, 1, 4, $i);
155 my $value = fetch_snmp($session, $oid);
157 $qcpupoll->execute($switch->{'switch'}, $entity, $value)
158 or die "Could not exec qcpupoll";
164 mylog("ERROR: $@ (during poll of $ip)");
168 my $elapsed = Time::HiRes::tv_interval($start);
169 $msg = sprintf "Polled $switch->{'ip'} in %5.3f seconds.", $elapsed;
172 $qunlock->execute($switch->{'switch'})
173 or die "Couldn't unlock switch";
178 my ($session, $port, $out, $wide_counters) = @_;
181 if ($wide_counters) {
183 $oid = BER::encode_oid(1, 3, 6, 1, 2, 1, 31, 1, 1, 1, 10, $port); # interfaces.ifTable.ifEntry.ifHCOutOctets
185 $oid = BER::encode_oid(1, 3, 6, 1, 2, 1, 31, 1, 1, 1, 6, $port); # interfaces.ifTable.ifEntry.ifHCInOctets
189 $oid = BER::encode_oid(1, 3, 6, 1, 2, 1, 2, 2, 1, 16, $port); # interfaces.ifTable.ifEntry.ifOutOctets
191 $oid = BER::encode_oid(1, 3, 6, 1, 2, 1, 2, 2, 1, 10, $port); # interfaces.ifTable.ifEntry.ifInOctets
195 return fetch_snmp($session, $oid);
199 my ($session, $oid) = @_;
201 if ($session->get_request_response($oid)) {
202 my ($bindings) = $session->decode_get_response ($session->{pdu_buffer});
204 while ($bindings ne '') {
205 ($binding,$bindings) = &decode_sequence ($bindings);
206 my ($oid,$value) = &decode_by_template ($binding, "%O%@");
207 return BER::pretty_print($value);
210 die "Couldn't get info from switch";
215 my @ranges = split /,/, $in;
218 for my $range (@ranges) {
219 if ($range =~ /^\d+$/) {
221 } elsif ($range =~ /^(\d+)-(\d+)$/) {
226 die "Couldn't understand '$range' in ports";
230 return (sort { $a <=> $b } @ret);
235 my $time = POSIX::ctime(time);
237 printf STDERR "[%s] %s\n", $time, $msg;
241 my ($cmd, $conn) = @_;
243 # Send the command and get data from switch
244 # $conn->dump_log(*STDOUT);
245 my @data = $conn->cmd($cmd);
247 foreach my $line (@data) {
248 # Remove escape-7 sequence
249 $line =~ s/\x1b\x37//g;
259 my $conn = new Net::Telnet( Timeout => $timeout,
260 Dump_Log => '/tmp/dumplog-tempfetch',
262 Prompt => '/es-3024|e(\-)?\d+\-\dsw>/i');
263 my $ret = $conn->open( Host => $ip);
264 if (!$ret || $ret != 1) {
267 # XXX: Just send the password as text, I did not figure out how to
268 # handle authentication with only password through $conn->login().
269 #$conn->login( Prompt => '/password[: ]*$/i',
271 # Password => $password);
272 my @data = $conn->cmd($password);