Add appropriate MIN/MAX RRAs to rrd files on creation and trigger recreation, if...
[upsgraph] / upsgraph.pl
1 #!/usr/bin/perl -w
2
3 #Due to memory leak in Debian squeeze (Bug #545519)
4 my $use_rrds = 0;
5 my $rrd_result = 0;
6
7 if ((@ARGV != 1) && (@ARGV != 2)) {
8 print STDERR "Syntax: ${0} configfile [uid]\n";
9 exit(1);
10 }
11
12 use Net::SNMP;
13 use IO::Socket::INET;
14 use IO::Select;
15 use RRDs;
16 use File::Copy;
17 use Data::Dumper;
18
19 $UPSGRAPH::outdir = "";
20 $UPSGRAPH::daysCovered = 370;
21 $UPSGRAPH::step = 60;
22 $UPSGRAPH::stepsPerHour = (60 * 60) / $UPSGRAPH::step;
23 $UPSGRAPH::keep = ($UPSGRAPH::daysCovered*24*60*60)/$UPSGRAPH::step;
24 $UPSGRAPH::keepHours = ($UPSGRAPH::daysCovered*24*60*60)/$UPSGRAPH::stepsPerHour/$UPSGRAPH::step;
25 $UPSGRAPH::hosts = ();
26
27 do $ARGV[0] or die "can't read config: $!";
28
29 my $outdir = $UPSGRAPH::outdir;
30 my $step = $UPSGRAPH::step;
31 my $stepsPerHour = $UPSGRAPH::stepsPerHour;
32 my $keep = $UPSGRAPH::keep;
33 my $keepHours = $UPSGRAPH::keepHours;
34 my $hosts = $UPSGRAPH::hosts;
35
36 sub rrd_update(@) {
37 my @args = @_;
38
39 if ($use_rrds == 1) {
40 RRDs::update(@args);
41 $rrd_result = RRDs::error;
42 } else {
43 $rrd_result = system("rrdtool", "update", @args);
44 }
45 }
46
47 sub rrd_graph(@) {
48 my @args = @_;
49 my @rrd_out = ();
50
51 if ($use_rrds == 1) {
52 @rrd_out = RRDs::graph(@args);
53 $rrd_result = RRDs::error;
54 } else {
55 my $rrd_stdout;
56
57 open(RRDFD, '-|', 'rrdtool', 'graph', @args);
58 while(<RRDFD>) {
59 chomp;
60 $rrd_stdout = $_;
61 }
62 close(RRDFD);
63 $rrd_result = $?;
64 if ($rrd_result == 0) {
65 push @rrd_out, 0;
66 push @rrd_out, split(/x/, $rrd_stdout);
67 }
68 }
69
70 return @rrd_out;
71 }
72
73 sub rrdcreate(@) {
74 my $newrrd = shift;
75 my $field = shift;
76 my $vars = shift;
77 my $start = shift;
78
79 my @cmd = ("${newrrd}", "--step=${step}");
80
81 if (defined($start)) {
82 push @cmd, "--start=${start}";
83 }
84
85 push @cmd, "DS:${field}:GAUGE:600:" .
86 $vars->{$field}->{'min'} . ":" .
87 $vars->{$field}->{'max'};
88
89 push @cmd, "RRA:AVERAGE:0.5:1:${keep}";
90 push @cmd, "RRA:MIN:0.4:${stepsPerHour}:${keepHours}";
91 push @cmd, "RRA:MAX:0.4:${stepsPerHour}:${keepHours}";
92
93 RRDs::create(@cmd);
94 if (RRDs::error) {
95 print "Error while creating: " . RRDs::error . "\n";
96 exit 1;
97 }
98 }
99
100 sub fetch_snmp(@) {
101 my $address = shift;
102 my $community = shift;
103 my $oid = shift;
104
105 (my $session, my $error) = Net::SNMP->session(Hostname => $address,
106 Community => $community);
107
108 if (!$session) {
109 print STDERR "session error: $error";
110 return undef;
111 }
112
113 $session->translate(0);
114
115 my $result = $session->get_request($oid);
116
117 $session->close;
118
119 return undef if (!defined($result));
120
121 $result->{$oid};
122 }
123
124 sub fetch_tcp(@) {
125 my $address = shift;
126 my $port = shift;
127
128 my $sock = IO::Socket::INET->new(PeerAddr => $address,
129 PeerPort => $port,
130 Proto => 'tcp',
131 Timeout => 1);
132
133 return undef if (!$sock);
134
135 my $select = IO::Select->new($sock);
136
137 my $value = undef;
138
139 if ($select->can_read(1)) {
140 chomp($value) if (sysread($sock, $value, 4096) > 0);
141 }
142
143 close($sock);
144
145 if (!$value) {
146 return undef;
147 }
148
149 $value=~ s/\s//g;
150
151 $value;
152 }
153
154 sub fetch_tcp_multi(@) {
155 my $address = shift;
156 my $port = shift;
157 my $delimiter = shift;
158 my %values;
159
160 my $sock = IO::Socket::INET->new(PeerAddr => $address,
161 PeerPort => $port,
162 Proto => 'tcp',
163 Timeout => 1);
164
165 return undef if (!$sock);
166
167 my $select = IO::Select->new($sock);
168
169 while($select->can_read(1)) {
170 if (sysread($sock, my $buf, 16384) > 0) {
171 $buf=~s/\r//g;
172 foreach my $line (split(/\n/, $buf)) {
173 (my $key, my $value) = split(/${delimiter}/, $line);
174 $value=~ s/\s//g;
175 $values{$key} = $value;
176 }
177 } else {
178 last;
179 }
180 }
181
182 close($sock);
183
184 %values;
185 }
186
187 if ($> == 0) {
188 if (@ARGV != 2) {
189 print STDERR "Running as root, please provide UID as 2th argument!\n";
190 exit(1);
191 }
192
193 print "Running as root, switching to ".$ARGV[1]."\n";
194 $< = $> = $ARGV[1];
195 }
196
197 foreach my $host (@$hosts) {
198 my $rrdfile = $host->{'rrdfile'};
199
200 foreach my $var (keys(%{$host->{'vars'}})) {
201 $host->{'vars'}->{$var}->{'min'} = 'U' if (!defined($host->{'vars'}->{$var}->{'min'}));
202 $host->{'vars'}->{$var}->{'max'} = 'U' if (!defined($host->{'vars'}->{$var}->{'max'}));
203 }
204
205 if (-e "${rrdfile}") {
206 print "Reading old ${rrdfile} to preserve data...\n";
207
208 my $rrdinfo = RRDs::info("${rrdfile}");
209 if (RRDs::error) {
210 print "Error while getting info: " . RRDs::error . "\n";
211 exit 1;
212 }
213
214 (my $start, my $ostep, my $names, my $data) =
215 RRDs::fetch("${rrdfile}",
216 "-s " . (time() - ($rrdinfo->{'rra[0].rows'} * $rrdinfo->{'step'})),
217 "AVERAGE");
218
219 if (RRDs::error) {
220 print "Error while fetching data: " . RRDs::error . "\n";
221 exit 1;
222 }
223
224 foreach my $field (@$names) {
225 if (! -e "${rrdfile}.${field}") {
226 rrdcreate("${rrdfile}.${field}",
227 "${field}",
228 $host->{'vars'},
229 (${start}-${ostep}));
230 }
231 }
232
233 my $pos = $start;
234 foreach my $line (@$data) {
235 foreach my $field (@$names) {
236 my $val = shift (@$line);
237 $val = 'U' if (!defined($val));
238
239 RRDs::update("${rrdfile}.${field}", "${pos}:${val}");
240 if (RRDs::error) {
241 print "Can't insert data: " . RRDs::error . "\n";
242 exit 1;
243 }
244
245 }
246
247 $pos += $ostep;
248
249 if ((($pos-$start)/$ostep) == $#$data) {
250 last;
251 }
252 }
253
254 rename("${rrdfile}", "${rrdfile}.old") or die "Can't rename old file: $!\n";
255 }
256
257 foreach my $field (@{$host->{'fields'}}) {
258 if (! -e "${rrdfile}.${field}") {
259 print "Creating ${rrdfile}.${field}...\n";
260 rrdcreate("${rrdfile}.${field}",
261 "${field}",
262 $host->{'vars'});
263 }
264
265 my $rrdinfo = RRDs::info("${rrdfile}.${field}");
266 if (RRDs::error) {
267 print "Error while getting info: " . RRDs::error . "\n";
268 exit 1;
269 }
270
271 if (defined($rrdinfo->{"ds[${field}].min"})) {
272 if ($rrdinfo->{"ds[${field}].min"} ne $host->{'vars'}->{$field}->{'min'}) {
273 RRDs::tune("${rrdfile}.${field}","-i",$field.":".$host->{'vars'}->{$field}->{'min'});
274 }
275 } else {
276 if ($host->{'vars'}->{$field}->{'min'} ne 'U') {
277 RRDs::tune("${rrdfile}.${field}","-i",$field.":".$host->{'vars'}->{$field}->{'min'});
278 }
279 }
280
281 if (RRDs::error) {
282 print "Error while setting min: " . RRDs::error . "\n";
283 exit 1;
284 }
285
286 if (defined($rrdinfo->{"ds[${field}].max"})) {
287 if ($rrdinfo->{"ds[${field}].max"} ne $host->{'vars'}->{$field}->{'max'}) {
288 RRDs::tune("${rrdfile}.${field}","-a",$field.":".$host->{'vars'}->{$field}->{'max'});
289 }
290 } else {
291 if ($host->{'vars'}->{$field}->{'max'} ne 'U') {
292 RRDs::tune("${rrdfile}.${field}","-a",$field.":".$host->{'vars'}->{$field}->{'max'});
293 }
294 }
295
296 if (RRDs::error) {
297 print "Error while setting max: " . RRDs::error . "\n";
298 exit 1;
299 }
300
301 if ($rrdinfo->{'rra[0].rows'} != $keep ||
302 !defined($rrdinfo->{'rra[1].rows'}) || $rrdinfo->{'rra[1].rows'} != $keepHours) {
303
304 print "Resizing ${rrdfile}.${field} from " . $rrdinfo->{'rra[0].rows'} .
305 " to ${keep} samples.\n";
306
307 (my $start, my $ostep, my $names, my $data) =
308 RRDs::fetch("${rrdfile}.${field}",
309 "-s " . (time() - ($rrdinfo->{'rra[0].rows'} * $rrdinfo->{'step'})),
310 "AVERAGE");
311
312 if (RRDs::error) {
313 print "Error while fetching data: " . RRDs::error . "\n";
314 exit 1;
315 }
316
317 rrdcreate("${rrdfile}.${field}.new",
318 "${field}",
319 $host->{'vars'},
320 (${start}-${ostep}));
321
322 print "Preserving data in file ${rrdfile}.${field} since " . localtime($start) . "\n";
323
324 my $pos = $start;
325 foreach my $line (@$data) {
326 my $vline = "${pos}";
327
328 foreach my $val (@$line) {
329 $val = 'U' if (!defined($val));
330 $vline .= ":${val}";
331 }
332 RRDs::update("${rrdfile}.${field}.new", $vline) or die "Can't insert data\n";
333
334 if (RRDs::error) {
335 print "Error while updating: " . RRDs::error . "\n";
336 exit 1;
337 }
338 $pos += $ostep;
339
340 if ((($pos-$start)/$ostep) == $#$data) {
341 last;
342 }
343 }
344
345 rename("${rrdfile}.${field}", "${rrdfile}.${field}.old") or die "Can't rename old file: $!\n";
346 rename("${rrdfile}.${field}.new", "${rrdfile}.${field}") or die "Can't rename new file: $!\n";
347
348 $rrdinfo = RRDs::info("${rrdfile}.${field}");
349 if (RRDs::error) {
350 print "Error while getting info: " . RRDs::error . "\n";
351 exit 1;
352 }
353
354 if ($rrdinfo->{'rra[0].rows'} != $keep) {
355 print "Failed!\n";
356 exit 1;
357 }
358 }
359 }
360 }
361
362 my $child = fork();
363
364 die "fork failed!" if (!defined($child));
365
366 exit 0 if ($child != 0);
367
368 while(1) {
369 open(HTML, ">${outdir}/index.html.new");
370
371 print HTML '<html><head><meta http-equiv="refresh" content="60"/><meta http-equiv="cache-control" content="no-cache"/><meta http-equiv="pragma" content="no-cache"/><meta http_equiv="expires" content="Sat, 26 Jul 1997 05:00:00 GMT"/><title>Status</title></head>';
372 print HTML '<body bgcolor="#ffffff">';
373
374 foreach my $host (@$hosts) {
375 print HTML "[<a href=\"#".${host}->{'name'}."\">".${host}->{'name'}."</a>]&nbsp;";
376 }
377 print HTML "<br>\n";
378
379 foreach my $host (@$hosts) {
380 print HTML "<br>\n";
381 print HTML "<a name=\"".${host}->{'name'}."\"></a>\n";
382 my $vars = $host->{'vars'};
383 my $rrdfile = $host->{'rrdfile'};
384 my $hostname = $host->{'name'};
385 my %multi_values = ();
386
387 foreach my $var (@{$host->{'fields'}}) {
388 delete $vars->{$var}->{'value'};
389
390 my $result;
391
392 if ((!defined($vars->{$var}->{'proto'})) ||
393 ($vars->{$var}->{'proto'} eq '') ||
394 ($vars->{$var}->{'proto'} eq 'snmp')) {
395 $result = fetch_snmp($host->{'address'}, $host->{'community'}, $vars->{$var}->{'oid'});
396 } elsif ($vars->{$var}->{'proto'} eq 'tcp') {
397 $result = fetch_tcp($host->{'address'}, $vars->{$var}->{'port'});
398 } elsif ($vars->{$var}->{'proto'} eq 'tcp_multi') {
399 if (defined($multi_values{$vars->{$var}->{'multi_id'}})) {
400 $result = $multi_values{$vars->{$var}->{'multi_id'}}
401 } else {
402 my %values = fetch_tcp_multi($host->{'address'}, $vars->{$var}->{'port'}, $vars->{$var}->{'multi_delimiter'});
403 @multi_values{keys %values} = values %values;
404 $result = $multi_values{$vars->{$var}->{'multi_id'}};
405 }
406 }
407
408 next unless (defined $result);
409
410 $vars->{$var}->{'value'} = $result;
411 if (defined($vars->{$var}->{'factor'})) {
412 $vars->{$var}->{'value'} *= $vars->{$var}->{'factor'};
413 }
414 }
415
416 foreach my $var (@{$host->{'fields'}}) {
417 if (!(defined($vars->{$var}->{'value'}))) {
418 $vars->{$var}->{'value'} = 'U';
419 }
420 rrd_update("${rrdfile}.${var}", "N:" . $vars->{$var}->{'value'});
421 }
422 if ($rrd_result) {
423 print "Error while updating: " . $rrd_result . "\n";
424 }
425
426 foreach my $var (@{$host->{'fields'}}) {
427 my @graphdef = ('-P', "--lazy", "-t", $hostname." - ".$vars->{$var}->{'name'}, "DEF:${var}=${rrdfile}.${var}:${var}:AVERAGE", "LINE1:${var}#FF0000");
428
429 push @graphdef, "DEF:${var}-min=${rrdfile}.${var}:${var}:MIN", "LINE1:${var}-min#0000FF";
430 push @graphdef, "DEF:${var}-max=${rrdfile}.${var}:${var}:MAX", "LINE1:${var}-max#00FF00";
431 push @graphdef, "VDEF:cur=${var},LAST";
432 push @graphdef, 'GPRINT:cur:Current\\: <span foreground="#FF0000">%.2lf</span>\\r';
433
434 my $mtime;
435 $mtime=(stat("${outdir}/${hostname}.${var}.png.work"))[9];
436
437 (my $averages, my $width, my $height) =
438 rrd_graph("${outdir}/${hostname}.${var}.png.work",
439 "-w", "720", @graphdef);
440
441 pop @graphdef;
442 pop @graphdef;
443
444 if ($rrd_result) {
445 print "Error while graphing: " . $rrd_result . "\n";
446 } else {
447 my $newmtime=(stat("${outdir}/${hostname}.${var}.png.work"))[9];
448 if ((!defined($mtime)) || ($newmtime != $mtime)) {
449 copy("${outdir}/${hostname}.${var}.png.work", "${outdir}/${hostname}.${var}.png.new");
450 rename("${outdir}/${hostname}.${var}.png.new", "${outdir}/${hostname}.${var}.png");
451 }
452 }
453
454 print HTML "<a href=\"${hostname}.${var}.html\"><img src=\"${hostname}.${var}.png\" width=\"${width}\" height=\"${height}\" border=\"0\"></a><br>\n";
455
456 open (HTML2, ">${outdir}/${hostname}.${var}.html.new");
457 print HTML2 '<html><head><meta http-equiv="Content-Type" content="text/html; charset=utf-8"><meta http-equiv="refresh" content="60"/><meta http-equiv="cache-control" content="no-cache"/><meta http-equiv="pragma" content="no-cache"/><meta http_equiv="expires" content="Sat, 26 Jul 1997 05:00:00 GMT"/><title>' . $vars->{$var}->{'name'} . '</title></head>';
458 print HTML2 '<body bgcolor="#ffffff">';
459
460 push @graphdef, "VDEF:min=${var}-min,MINIMUM";
461 push @graphdef, "GPRINT:min:Minimum\\: %.2lf";
462
463 push @graphdef, "VDEF:avg=${var},AVERAGE";
464 push @graphdef, "GPRINT:avg:Average\\: %.2lf";
465
466 push @graphdef, "VDEF:max=${var}-max,MAXIMUM";
467 push @graphdef, "GPRINT:max:Maximum\\: %.2lf";
468
469 push @graphdef, "VDEF:cur=${var},LAST";
470 push @graphdef, "GPRINT:cur:Current\\: %.2lf";
471
472 $mtime=(stat("${outdir}/${hostname}.${var}.long.png.work"))[9];
473 ($averages, $width, $height) =
474 rrd_graph("${outdir}/${hostname}.${var}.long.png.work",
475 "-w", "1008", @graphdef);
476
477 if ($rrd_result) {
478 print "Error while graphing: " . $rrd_result . "\n";
479 } else {
480 my $newmtime=(stat("${outdir}/${hostname}.${var}.long.png.work"))[9];
481 if ((!defined($mtime)) || ($newmtime != $mtime)) {
482 copy("${outdir}/${hostname}.${var}.long.png.work", "${outdir}/${hostname}.${var}.long.png.new");
483 rename("${outdir}/${hostname}.${var}.long.png.new", "${outdir}/${hostname}.${var}.long.png");
484 }
485 }
486
487 print HTML2 "<img src=\"${hostname}.${var}.long.png\" width=\"${width}\" height=\"${height}\"><br>\n";
488
489 $mtime=(stat("${outdir}/${hostname}.${var}.week.png.work"))[9];
490 ($averages, $width, $height) =
491 rrd_graph("${outdir}/${hostname}.${var}.week.png.work",
492 "-w", "1008", "-e", "now", "-s", "end-1w", @graphdef);
493
494 if ($rrd_result) {
495 print "Error while graphing: " . $rrd_result . "\n";
496 } else {
497 my $newmtime=(stat("${outdir}/${hostname}.${var}.week.png.work"))[9];
498 if ((!defined($mtime)) || ($newmtime != $mtime)) {
499 copy("${outdir}/${hostname}.${var}.week.png.work", "${outdir}/${hostname}.${var}.week.png.new");
500 rename("${outdir}/${hostname}.${var}.week.png.new", "${outdir}/${hostname}.${var}.week.png");
501 }
502 }
503
504 print HTML2 "<img src=\"${hostname}.${var}.week.png\" width=\"${width}\" height=\"${height}\"><br>\n";
505
506 $mtime=(stat("${outdir}/${hostname}.${var}.year.png.work"))[9];
507 ($averages, $width, $height) =
508 rrd_graph("${outdir}/${hostname}.${var}.year.png.work",
509 "-w", "1008", "-e", "now", "-s", "end-1y", @graphdef);
510
511 if ($rrd_result) {
512 print "Error while graphing: " . $rrd_result . "\n";
513 } else {
514 my $newmtime=(stat("${outdir}/${hostname}.${var}.year.png.work"))[9];
515 if ((!defined($mtime)) || ($newmtime != $mtime)) {
516 copy("${outdir}/${hostname}.${var}.year.png.work", "${outdir}/${hostname}.${var}.year.png.new");
517 rename("${outdir}/${hostname}.${var}.year.png.new", "${outdir}/${hostname}.${var}.year.png");
518 }
519 }
520
521 print HTML2 "<img src=\"${hostname}.${var}.year.png\" width=\"${width}\" height=\"${height}\"><br>\n";
522
523 print HTML2 "</body></html>\n";
524 close(HTML2);
525 rename("${outdir}/${hostname}.${var}.html.new", "${outdir}/${hostname}.${var}.html");
526 }
527 }
528
529 print HTML "</body></html>\n";
530 print HTML "<br>Generated on: " . localtime(time());
531 print HTML ' by <a href="http://git.zerfleddert.de/cgi-bin/gitweb.cgi/upsgraph">upsgraph</a>.';
532
533 close(HTML);
534
535 rename("${outdir}/index.html.new", "${outdir}/index.html");
536
537 sleep(${step}/2);
538 }
Impressum, Datenschutz